diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py index 0db9df343..54de60d24 100644 --- 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 index c493b8b62..37fdf5958 100644 --- 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)