add dylib for osx

bitsandbytes-foundation · Feb 5, 2024 · 4069a3b · 4069a3b
1 parent 3b3d32a
commit 4069a3b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py
@@ -37,7 +37,7 @@
     # we have libcudart.so.11.0 which causes a lot of errors before
     # not sure if libcudart.so.12.0 exists in pytorch installs, but it does not hurt
     CUDA_RUNTIME_LIBS = ["libcudart.so", "libcudart.so.11.0", "libcudart.so.12.0", "libcudart.so.12.1", "libcudart.so.12.2"]
-    DYNAMIC_LIBRARY_SUFFIX = ".so"
+    DYNAMIC_LIBRARY_SUFFIX = { "Darwin": ".dylib", "Windows":".dll" }.get(platform.system(), ".so")
 
 
 class CUDASetup:

diff --git a/setup.py b/setup.py
@@ -9,6 +9,7 @@
 
 libs = list(glob.glob("./bitsandbytes/libbitsandbytes*.so"))
 libs += list(glob.glob("./bitsandbytes/libbitsandbytes*.dll"))
+libs += list(glob.glob("./bitsandbytes/libbitsandbytes*.dylib"))
 libs = [os.path.basename(p) for p in libs]
 print("libs:", libs)