diff --git a/setup.py b/setup.py index 3065740f8..9a35f6d9e 100644 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ import glob import os -from setuptools import find_packages, setup +from setuptools import find_packages, setup, Extension libs = list(glob.glob("./bitsandbytes/libbitsandbytes*.so")) libs = [os.path.basename(p) for p in libs] @@ -29,6 +29,7 @@ def read(fname): package_data={"": libs}, long_description=read("README.md"), long_description_content_type="text/markdown", + ext_modules=[Extension("bitsandbytes", sources=[], language="c")], classifiers=[ "Development Status :: 4 - Beta", "Topic :: Scientific/Engineering :: Artificial Intelligence",