deftdawg/bitsandbytes-rocm
1
0
Fork 0
forked from mrq/bitsandbytes-rocm
Code Pull Requests Activity

Fixed deployment script to check for LD_LIBRARY_PATH.

Browse Source
This commit is contained in:
Tim Dettmers 2022-07-27 05:57:50 -07:00
parent 389f66ca5a
commit bd515328d7
1 changed files with 37 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
deploy_from_slurm.sh
View File

@ -1,28 +1,37 @@
#!/bin/bash #!/bin/bash
BASE_PATH=$1 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 cuda
module unload gcc module unload gcc
#rm -rf dist build rm -rf dist build
#make clean make clean
#make cleaneggs make cleaneggs
#export CUDA_HOME= export CUDA_HOME=
#make cpuonly make cpuonly
#
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist. # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2 echo "Compilation unsuccessul!" 1>&2
# exit 64 exit 64
#fi fi
#CUDA_VERSION=cpu python -m build CUDA_VERSION=cpu python -m build
#python -m twine upload dist/* --verbose --repository testpypi python -m twine upload dist/* --verbose --repository testpypi
rm -rf dist build rm -rf dist build
make clean make clean
make cleaneggs make cleaneggs
export CUDA_HOME=$BASE_PATH/cuda-11.0 export CUDA_HOME=$BASE_PATH/cuda-11.0
make cuda110 make cuda110
if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# Control will enter here if $DIRECTORY doesn't exist. # Control will enter here if $DIRECTORY doesn't exist.
@ -102,20 +111,20 @@ fi
CUDA_VERSION=115 python -m build CUDA_VERSION=115 python -m build
python -m twine upload dist/* --verbose --repository testpypi python -m twine upload dist/* --verbose --repository testpypi
#rm -rf dist build rm -rf dist build
#make clean make clean
#make cleaneggs make cleaneggs
#export CUDA_HOME=$BASE_PATH/cuda-11.6 export CUDA_HOME=$BASE_PATH/cuda-11.6
#
#make cuda11x make cuda11x
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist. # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2 echo "Compilation unsuccessul!" 1>&2
# exit 64 exit 64
#fi fi
#CUDA_VERSION=116 python -m build CUDA_VERSION=116 python -m build
#python -m twine upload dist/* --verbose --repository testpypi python -m twine upload dist/* --verbose --repository testpypi
#
rm -rf dist build rm -rf dist build
make clean make clean
make cleaneggs make cleaneggs
@ -257,5 +266,4 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=117-nomatmul python -m build CUDA_VERSION=117-nomatmul python -m build
python -m twine upload dist/* --verbose
python -m twine upload dist/* --verbose --repository testpypi python -m twine upload dist/* --verbose --repository testpypi