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. --- setup.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'setup.py') 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