deftdawg/bitsandbytes-rocm
1
0
Fork 0
forked from mrq/bitsandbytes-rocm
Code Pull Requests Activity

Changed setup.py; deployed on test pypi.

Browse Source
This commit is contained in:
Tim Dettmers 2022-07-26 10:32:22 -07:00
parent 953b7285dd
commit dc8c9efdb3
2 changed files with 46 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

85
deploy_from_slurm.sh
View File

@ -4,19 +4,19 @@ BASE_PATH=$1
module unload cuda module unload cuda
module unload gcc module unload gcc
rm -rf dist build #rm -rf dist build
make clean #make clean
make cleaneggs #make cleaneggs
export CUDA_HOME= #export CUDA_HOME=
make cpuonly #make cpuonly
#
if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then #if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# Control will enter here if $DIRECTORY doesn't exist. # # Control will enter here if $DIRECTORY doesn't exist.
echo "Compilation unsuccessul!" 1>&2 # echo "Compilation unsuccessul!" 1>&2
exit 64 # exit 64
fi #fi
CUDA_VERSION=cpu python -m build #CUDA_VERSION=cpu python -m build
python -m twine upload dist/* --verbose #python -m twine upload dist/* --verbose --repository testpypi
rm -rf dist build rm -rf dist build
make clean make clean
@ -30,7 +30,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=110 python -m build CUDA_VERSION=110 python -m build
python -m twine upload dist/* --verbose python -m twine upload dist/* --verbose --repository testpypi
rm -rf dist build rm -rf dist build
make clean make clean
@ -44,7 +44,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=111 python -m build CUDA_VERSION=111 python -m build
python -m twine upload dist/* --verbose python -m twine upload dist/* --verbose --repository testpypi
rm -rf dist build rm -rf dist build
make clean make clean
@ -58,7 +58,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=112 python -m build CUDA_VERSION=112 python -m build
python -m twine upload dist/* --verbose python -m twine upload dist/* --verbose --repository testpypi
rm -rf dist build rm -rf dist build
make clean make clean
@ -72,7 +72,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=113 python -m build CUDA_VERSION=113 python -m build
python -m twine upload dist/* --verbose python -m twine upload dist/* --verbose --repository testpypi
rm -rf dist build rm -rf dist build
make clean make clean
@ -86,7 +86,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=114 python -m build CUDA_VERSION=114 python -m build
python -m twine upload dist/* --verbose python -m twine upload dist/* --verbose --repository testpypi
rm -rf dist build rm -rf dist build
make clean make clean
@ -100,22 +100,22 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=115 python -m build CUDA_VERSION=115 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
#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 rm -rf dist build
make clean make clean
make cleaneggs make cleaneggs
@ -128,7 +128,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=117 python -m build CUDA_VERSION=117 python -m build
python -m twine upload dist/* --verbose python -m twine upload dist/* --verbose --repository testpypi
rm -rf dist build rm -rf dist build
@ -143,7 +143,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=102-nomatmul python -m build 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 rm -rf dist build
@ -158,7 +158,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=110-nomatmul python -m build 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 rm -rf dist build
@ -173,7 +173,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=111-nomatmul python -m build 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 rm -rf dist build
make clean make clean
@ -187,7 +187,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=112-nomatmul python -m build 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 rm -rf dist build
make clean make clean
@ -201,7 +201,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=113-nomatmul python -m build 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 rm -rf dist build
make clean make clean
@ -215,7 +215,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=114-nomatmul python -m build 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 rm -rf dist build
make clean make clean
@ -229,7 +229,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=115-nomatmul python -m build 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 rm -rf dist build
make clean make clean
@ -243,7 +243,7 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
exit 64 exit 64
fi fi
CUDA_VERSION=116-nomatmul python -m build 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 rm -rf dist build
make clean make clean
@ -258,3 +258,4 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
fi fi
CUDA_VERSION=117-nomatmul python -m build CUDA_VERSION=117-nomatmul python -m build
python -m twine upload dist/* --verbose python -m twine upload dist/* --verbose
python -m twine upload dist/* --verbose --repository testpypi

5
setup.py
View File

@ -11,13 +11,14 @@ def read(fname):
version = os.getenv("CUDA_VERSION", "cpu") version = os.getenv("CUDA_VERSION", "cpu")
prefix = '' if version == 'cpu' else 'cuda'
setup( setup(
name="bitsandbytes", name=f"bitsandbytes-{prefix}{version}",
version=f"0.30.0", version=f"0.30.0",
author="Tim Dettmers", author="Tim Dettmers",
author_email="dettmers@cs.washington.edu", author_email="dettmers@cs.washington.edu",
description="8-bit optimizers and quantization routines.", description="8-bit optimizers and matrix multiplication routines.",
license="MIT", license="MIT",
keywords="gpu optimizers optimization 8-bit quantization compression", keywords="gpu optimizers optimization 8-bit quantization compression",
url="http://packages.python.org/bitsandbytes", url="http://packages.python.org/bitsandbytes",