diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-04 09:16:00 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-04 09:16:00 -0700 |
| commit | 8f84674d6774c351b1e69dfede2c11a370e334b9 (patch) | |
| tree | 2312dde5180f471cd94b9ade5941c0e374386615 /bitsandbytes/cextension.py | |
| parent | 758c7175a24df307c40b743b1def8b4c34f68674 (diff) | |
Fixed bugs in cuda setup.
Diffstat (limited to 'bitsandbytes/cextension.py')
| -rw-r--r-- | bitsandbytes/cextension.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/bitsandbytes/cextension.py b/bitsandbytes/cextension.py index 66c79d8..e0f280a 100644 --- a/bitsandbytes/cextension.py +++ b/bitsandbytes/cextension.py @@ -17,12 +17,13 @@ class CUDALibrary_Singleton(object): binary_path = package_dir / binary_name if not binary_path.exists(): - print(f"TODO: compile library for specific version: {binary_name}") + print(f"CUDA_SETUP: TODO: compile library for specific version: {binary_name}") legacy_binary_name = "libbitsandbytes.so" - print(f"Defaulting to {legacy_binary_name}...") + print(f"CUDA_SETUP: Defaulting to {legacy_binary_name}...") self.lib = ct.cdll.LoadLibrary(package_dir / legacy_binary_name) else: - self.lib = ct.cdll.LoadLibrary(package_dir / binary_name) + print(f"CUDA_SETUP: Loading binary {binary_path}...") + self.lib = ct.cdll.LoadLibrary(binary_path) @classmethod def get_instance(cls): |