setup.py

# Copyright (c) Facebook, Inc. and its affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
import glob
import os

from setuptools import Extension, find_packages, setup

libs = list(glob.glob("./bitsandbytes/libbitsandbytes*.so"))
libs += list(glob.glob("./bitsandbytes/libbitsandbytes*.dll"))
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()


setup(
    name="bitsandbytes",
    version="0.43.0.dev0",
    author="Tim Dettmers",
    author_email="dettmers@cs.washington.edu",
    description="k-bit optimizers and matrix multiplication routines.",
    license="MIT",
    keywords="gpu optimizers optimization 8-bit quantization compression",
    url="https://github.com/TimDettmers/bitsandbytes",
    packages=find_packages(),
    package_data={"": libs},
    install_requires=['torch', 'numpy'],
    extras_require={
        'benchmark': ['pandas', 'matplotlib'],
        'test': ['scipy'],
    },
    long_description=read("README.md"),
    long_description_content_type="text/markdown",
    # HACK: pretend we have a native extension module so the wheel is tagged
    #       correctly with a platform tag (e.g. `-linux_x86_64.whl`).
    ext_modules=[Extension("bitsandbytes", sources=[], language="c")],
    classifiers=[
        "Development Status :: 4 - Beta",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
    ],
)