summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--cuda_install.sh4
-rw-r--r--deploy.sh (renamed from deploy_from_slurm.sh)20
2 files changed, 24 insertions, 0 deletions
diff --git a/cuda_install.sh b/cuda_install.sh
index 4508f4a..b6c553b 100644
--- a/cuda_install.sh
+++ b/cuda_install.sh
@@ -10,6 +10,7 @@ URL114=https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installer
URL115=https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux.run
URL116=https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
URL117=https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
+URL118=https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
CUDA_VERSION=$1
@@ -52,6 +53,9 @@ if [[ -n "$CUDA_VERSION" ]]; then
elif [[ "$CUDA_VERSION" -eq "117" ]]; then
URL=$URL117
FOLDER=cuda-11.7
+ elif [[ "$CUDA_VERSION" -eq "118" ]]; then
+ URL=$URL118
+ FOLDER=cuda-11.8
else
echo "argument error: No cuda version passed as input. Choose among: {111, 115}"
fi
diff --git a/deploy_from_slurm.sh b/deploy.sh
index bc53869..7d4f1a0 100644
--- a/deploy_from_slurm.sh
+++ b/deploy.sh
@@ -108,6 +108,16 @@ if [ ! -f "./bitsandbytes/libbitsandbytes_cuda117.so" ]; then
exit 64
fi
+make clean
+export CUDA_HOME=$BASE_PATH/cuda-11.8
+make cuda11x CUDA_VERSION=118
+
+if [ ! -f "./bitsandbytes/libbitsandbytes_cuda118.so" ]; then
+ # Control will enter here if $DIRECTORY doesn't exist.
+ echo "Compilation unsuccessul!" 1>&2
+ exit 64
+fi
+
make clean
export CUDA_HOME=$BASE_PATH/cuda-10.2
@@ -201,5 +211,15 @@ if [ ! -f "./bitsandbytes/libbitsandbytes_cuda117_nocublaslt.so" ]; then
exit 64
fi
+make clean
+export CUDA_HOME=$BASE_PATH/cuda-11.8
+make cuda11x_nomatmul CUDA_VERSION=118
+
+if [ ! -f "./bitsandbytes/libbitsandbytes_cuda118_nocublaslt.so" ]; then
+ # Control will enter here if $DIRECTORY doesn't exist.
+ echo "Compilation unsuccessul!" 1>&2
+ exit 64
+fi
+
python -m build
python -m twine upload dist/* --verbose