support osx

Author: wkpark

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)

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)