Skip to content

Commit 51e4b6f

Browse files
wkparkakx
andcommitted
add ext_modules
* add a comment suggested by @akx (2024/01/30) Co-authored-by: Aarni Koskela <[email protected]>
1 parent 926b2fe commit 51e4b6f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

setup.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import glob
66
import os
77

8-
from setuptools import find_packages, setup
8+
from setuptools import Extension, find_packages, setup
99

1010
libs = list(glob.glob("./bitsandbytes/libbitsandbytes*.so"))
1111
libs += list(glob.glob("./bitsandbytes/libbitsandbytes*.dll"))
@@ -35,6 +35,9 @@ def read(fname):
3535
},
3636
long_description=read("README.md"),
3737
long_description_content_type="text/markdown",
38+
# HACK: pretend we have a native extension module so the wheel is tagged
39+
# correctly with a platform tag (e.g. `-linux_x86_64.whl`).
40+
ext_modules=[Extension("bitsandbytes", sources=[], language="c")],
3841
classifiers=[
3942
"Development Status :: 4 - Beta",
4043
"Topic :: Scientific/Engineering :: Artificial Intelligence",

0 commit comments

Comments
 (0)