add ext_modules

bitsandbytes-foundation · Dec 11, 2023 · 260eb33 · 260eb33
1 parent 10535ee
commit 260eb33
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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",