From 1ced47c5043ed88b78c288f55f43ec3e66a0f765 Mon Sep 17 00:00:00 2001
From: Tim Dettmers <tim.dettmers@gmail.com>
Date: Tue, 9 Aug 2022 20:02:47 -0700
Subject: Added CUDA version warning and fixed cuda_install for 9.2/10.2.

---
 bitsandbytes/cuda_setup/main.py | 8 ++++++++
 cuda_install.sh                 | 2 +-
 deploy_from_slurm.sh            | 2 +-
 setup.py                        | 2 +-
 4 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py
index f1c845c..1f2ceb4 100644
--- a/bitsandbytes/cuda_setup/main.py
+++ b/bitsandbytes/cuda_setup/main.py
@@ -45,6 +45,9 @@ def get_cuda_version(cuda, cudart_path):
     major = version//1000
     minor = (version-(major*1000))//10
 
+    if major < 11:
+       print('CUDA SETUP: CUDA version lower than 11 are currenlty not supported!')
+
     return f'{major}{minor}'
 
 
@@ -110,6 +113,10 @@ def get_compute_capability(cuda):
 
 
 def evaluate_cuda_setup():
+    print('')
+    print('='*35 + 'BUG REPORT' + '='*35)
+    print('Welcome to bitsandbytes. For bug reports, please use this form: https://docs.google.com/forms/d/e/1FAIpQLScPB8emS3Thkp66nvqwmjTEgxp8Y9ufuWTzFyr9kJ5AoI47dQ/viewform?usp=sf_link')
+    print('='*80)
     binary_name = "libbitsandbytes_cpu.so"
     cudart_path = determine_cuda_runtime_lib_path()
     if cudart_path is None:
@@ -121,6 +128,7 @@ def evaluate_cuda_setup():
     print(f"CUDA SETUP: CUDA path found: {cudart_path}")
     cuda = get_cuda_lib_handle()
     cc = get_compute_capability(cuda)
+    print(f"CUDA SETUP: Highest compute capability among GPUs detected: {cc}")
     cuda_version_string = get_cuda_version(cuda, cudart_path)
 
 
diff --git a/cuda_install.sh b/cuda_install.sh
index 856cbe5..4508f4a 100644
--- a/cuda_install.sh
+++ b/cuda_install.sh
@@ -65,7 +65,7 @@ if [[ -n "$CUDA_VERSION" ]]; then
   echo $URL
   echo $FILE
   wget $URL
-  bash $FILE --no-drm --no-man-page --override --installpath=~/local --librarypath=$BASE_PATH/lib --toolkitpath=$BASE_PATH/$FOLDER/ --toolkit --silent
+  bash $FILE --no-drm --no-man-page --override --toolkitpath=$BASE_PATH/$FOLDER/ --toolkit --silent
   echo "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$BASE_PATH/$FOLDER/lib64/" >> ~/.bashrc
   echo "export PATH=$PATH:$BASE_PATH/$FOLDER/bin/" >> ~/.bashrc
   source ~/.bashrc
diff --git a/deploy_from_slurm.sh b/deploy_from_slurm.sh
index c6ee84d..bc53869 100644
--- a/deploy_from_slurm.sh
+++ b/deploy_from_slurm.sh
@@ -202,4 +202,4 @@ if [ ! -f "./bitsandbytes/libbitsandbytes_cuda117_nocublaslt.so" ]; then
 fi
 
 python -m build
-python -m twine upload dist/* --verbose --repository testpypi
+python -m twine upload dist/* --verbose
diff --git a/setup.py b/setup.py
index 9fb1889..61a5d05 100644
--- a/setup.py
+++ b/setup.py
@@ -18,7 +18,7 @@ def read(fname):
 
 setup(
     name=f"bitsandbytes",
-    version=f"0.31.7",
+    version=f"0.31.8",
     author="Tim Dettmers",
     author_email="dettmers@cs.washington.edu",
     description="8-bit optimizers and matrix multiplication routines.",
-- 
cgit v1.2.3