Changed setup.py; deployed on test pypi.

bitsandbytes-foundation · Jul 26, 2022 · dc8c9ef · dc8c9ef
1 parent 953b728
commit dc8c9ef
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 45 deletions.
diff --git a/deploy_from_slurm.sh b/deploy_from_slurm.sh
@@ -4,19 +4,19 @@ BASE_PATH=$1
 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
+#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
@@ -30,7 +30,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=110 python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -44,7 +44,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=111 python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -58,7 +58,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=112 python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -72,7 +72,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=113 python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -86,7 +86,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=114 python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -100,22 +100,22 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=115 python -m build
-python -m twine upload dist/* --verbose
-
-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
-
+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
@@ -128,7 +128,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=117 python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 
 rm -rf dist build
@@ -143,7 +143,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=102-nomatmul python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 
 rm -rf dist build
@@ -158,7 +158,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=110-nomatmul python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 
 rm -rf dist build
@@ -173,7 +173,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=111-nomatmul python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -187,7 +187,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=112-nomatmul python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -201,7 +201,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=113-nomatmul python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -215,7 +215,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=114-nomatmul python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -229,7 +229,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=115-nomatmul python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -243,7 +243,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
   exit 64
 fi
 CUDA_VERSION=116-nomatmul python -m build
-python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
 
 rm -rf dist build
 make clean
@@ -258,3 +258,4 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
 fi
 CUDA_VERSION=117-nomatmul python -m build
 python -m twine upload dist/* --verbose
+python -m twine upload dist/* --verbose --repository testpypi
diff --git a/setup.py b/setup.py
@@ -11,13 +11,14 @@ def read(fname):
 
 
 version = os.getenv("CUDA_VERSION", "cpu")
+prefix = '' if version == 'cpu' else 'cuda'
 
 setup(
-    name="bitsandbytes",
+    name=f"bitsandbytes-{prefix}{version}",
     version=f"0.30.0",
     author="Tim Dettmers",
     author_email="[email protected]",
-    description="8-bit optimizers and quantization routines.",
+    description="8-bit optimizers and matrix multiplication routines.",
     license="MIT",
     keywords="gpu optimizers optimization 8-bit quantization compression",
     url="http://packages.python.org/bitsandbytes",