diff options
Diffstat (limited to 'setup.py')
| -rw-r--r-- | setup.py | 22 |
1 files changed, 7 insertions, 15 deletions
@@ -3,28 +3,20 @@ # This source code is licensed under the MIT license found in the # LICENSE file in the root directory of this source tree. import os +import glob from setuptools import setup, find_packages -import os - -def all_libs(directory): - paths = [] - for (path, directories, filenames) in os.walk(directory): - for filename in filenames: - paths.append(os.path.join('..', path, filename)) - return paths +libs = list(glob.glob('./bitsandbytes/libbitsandbytes*.so')) +libs = [os.path.basename(p) for p in libs] +print('libs:', libs) def read(fname): return open(os.path.join(os.path.dirname(__file__), fname)).read() - -version = os.getenv("CUDA_VERSION", "cpu") -prefix = '' if version == 'cpu' else 'cuda' - setup( - name=f"bitsandbytes-{prefix}{version}", - version=f"0.30.2", + name=f"bitsandbytes", + version=f"0.31.0", author="Tim Dettmers", author_email="dettmers@cs.washington.edu", description="8-bit optimizers and matrix multiplication routines.", @@ -35,7 +27,7 @@ setup( entry_points={ "console_scripts": ["debug_cuda = bitsandbytes.debug_cli:cli"], }, - package_data={'': ['libbitsandbytes*.so']}, + package_data={'': libs}, long_description=read('README.md'), long_description_content_type='text/markdown', classifiers=[ |