summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTim Dettmers <tim.dettmers@gmail.com>2022-07-26 10:32:22 -0700
committerTim Dettmers <tim.dettmers@gmail.com>2022-07-26 10:32:22 -0700
commitdc8c9efdb33130f960adc864916b67d0cb744dbb (patch)
tree5fc5063cb5f4cacefdb3f81185fb8b5e5da66cba
parent953b7285ddf55913732bc9f137953dd00ac64c35 (diff)
Changed setup.py; deployed on test pypi.
-rw-r--r--deploy_from_slurm.sh87
-rw-r--r--setup.py5
2 files changed, 47 insertions, 45 deletions
diff --git a/deploy_from_slurm.sh b/deploy_from_slurm.sh
index d58aa38..664d40e 100644
--- 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
index 6cc091b..6275ddd 100644
--- 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="dettmers@cs.washington.edu",
- 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",