From dc8c9efdb33130f960adc864916b67d0cb744dbb Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Tue, 26 Jul 2022 10:32:22 -0700 Subject: Changed setup.py; deployed on test pypi. --- deploy_from_slurm.sh | 87 ++++++++++++++++++++++++++-------------------------- setup.py | 5 +-- 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", -- cgit v1.2.3