Enable running setup-env from non-bash shells (#319)

Signed-off-by: Tal Ben-Nun <talbn@llnl.gov>
ROCm · Jan 10, 2024 · 5b42702 · 5b42702
1 parent 608287d
commit 5b42702
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cmake/Templates/setup-env.sh.in b/cmake/Templates/setup-env.sh.in
@@ -1,6 +1,11 @@
 #!/usr/bin/env bash
 
-BASEDIR=$(dirname ${BASH_SOURCE[0]})
+if [ -z "$BASH_SOURCE" ]; then
+    # If not running bash, try to obtain directory with $0
+    BASEDIR="$( cd "$(dirname "$0")"; pwd -P )"
+else
+    BASEDIR=$(dirname ${BASH_SOURCE[0]})
+fi
 command -v realpath &> /dev/null && BASEDIR=$(realpath ${BASEDIR}/../..) || BASEDIR=$(cd ${BASEDIR}/../.. && pwd)
 
 if [ ! -d "${BASEDIR}" ]; then