summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTim Dettmers <tim.dettmers@gmail.com>2022-07-27 05:57:50 -0700
committerTim Dettmers <tim.dettmers@gmail.com>2022-07-27 05:57:50 -0700
commitbd515328d70f344f935075f359c5aefc616878d5 (patch)
tree54d0b0743e38c5ad140d943d3b2c6d640dddf076
parent389f66ca5a737eb7f22f22fed420274ff622623e (diff)
Fixed deployment script to check for LD_LIBRARY_PATH.
-rw-r--r--deploy_from_slurm.sh66
1 files changed, 37 insertions, 29 deletions
diff --git a/deploy_from_slurm.sh b/deploy_from_slurm.sh
index 664d40e..37311bc 100644
--- a/deploy_from_slurm.sh
+++ b/deploy_from_slurm.sh
@@ -1,28 +1,37 @@
#!/bin/bash
BASE_PATH=$1
+echo "MAKE SURE LD_LIBRARY_PATH IS EMPTY!"
+echo $LD_LIBRARY_PATH
+
+if [[ ! -z "${LD_LIBRARY_PATH}" ]]; then
+ echo "Compilation unsuccessul!" 1>&2
+ exit 64
+fi
+
+
module unload cuda
module unload gcc
-#rm -rf dist build
-#make clean
-#make cleaneggs
-#export CUDA_HOME=
-#make cpuonly
-#
-#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
-# # Control will enter here if $DIRECTORY doesn't exist.
-# echo "Compilation unsuccessul!" 1>&2
-# exit 64
-#fi
-#CUDA_VERSION=cpu python -m build
-#python -m twine upload dist/* --verbose --repository testpypi
+rm -rf dist build
+make clean
+make cleaneggs
+export CUDA_HOME=
+make cpuonly
+
+if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
+ # Control will enter here if $DIRECTORY doesn't exist.
+ echo "Compilation unsuccessul!" 1>&2
+ exit 64
+fi
+CUDA_VERSION=cpu python -m build
+python -m twine upload dist/* --verbose --repository testpypi
rm -rf dist build
make clean
make cleaneggs
export CUDA_HOME=$BASE_PATH/cuda-11.0
-make cuda110
+make cuda110
if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# Control will enter here if $DIRECTORY doesn't exist.
@@ -102,20 +111,20 @@ fi
CUDA_VERSION=115 python -m build
python -m twine upload dist/* --verbose --repository testpypi
-#rm -rf dist build
-#make clean
-#make cleaneggs
-#export CUDA_HOME=$BASE_PATH/cuda-11.6
-#
-#make cuda11x
-#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
-# # Control will enter here if $DIRECTORY doesn't exist.
-# echo "Compilation unsuccessul!" 1>&2
-# exit 64
-#fi
-#CUDA_VERSION=116 python -m build
-#python -m twine upload dist/* --verbose --repository testpypi
-#
+rm -rf dist build
+make clean
+make cleaneggs
+export CUDA_HOME=$BASE_PATH/cuda-11.6
+
+make cuda11x
+if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
+ # Control will enter here if $DIRECTORY doesn't exist.
+ echo "Compilation unsuccessul!" 1>&2
+ exit 64
+fi
+CUDA_VERSION=116 python -m build
+python -m twine upload dist/* --verbose --repository testpypi
+
rm -rf dist build
make clean
make cleaneggs
@@ -257,5 +266,4 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64
fi
CUDA_VERSION=117-nomatmul python -m build
-python -m twine upload dist/* --verbose
python -m twine upload dist/* --verbose --repository testpypi