support osx

bitsandbytes-foundation · Feb 3, 2024 · 24f91ce · 24f91ce
1 parent 35344df
commit 24f91ce
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
@@ -348,7 +348,7 @@ def get_compute_capabilities():
 
 def evaluate_cuda_setup():
     cuda_setup = CUDASetup.get_instance()
-    suffix = ".so" if os.name != "nt" else ".dll"
+    suffix = { "Darwin": ".dylib", "Windows":".dll" }.get(platform.system(), ".so")
     if 'BITSANDBYTES_NOWELCOME' not in os.environ or str(os.environ['BITSANDBYTES_NOWELCOME']) == '0':
         cuda_setup.add_log_entry('')
         cuda_setup.add_log_entry('='*35 + 'BUG REPORT' + '='*35)

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)