From 31ef751bea48eeee2e0e95aca79df8e59b4c25c4 Mon Sep 17 00:00:00 2001 From: Ben Sima Date: Wed, 21 Dec 2022 12:36:48 -0500 Subject: Log binary_path when CUDA setup fails This way we can debug the build paths if it fails. --- bitsandbytes/cextension.py | 1 + 1 file changed, 1 insertion(+) (limited to 'bitsandbytes') diff --git a/bitsandbytes/cextension.py b/bitsandbytes/cextension.py index 264e899..afad6b8 100644 --- a/bitsandbytes/cextension.py +++ b/bitsandbytes/cextension.py @@ -73,6 +73,7 @@ class CUDASetup(object): try: if not binary_path.exists(): self.add_log_entry(f"CUDA SETUP: Required library version not found: {binary_name}. Maybe you need to compile it from source?") + self.add_log_entry(f"CUDA SETUP: looked in {binary_path}") legacy_binary_name = "libbitsandbytes.so" self.add_log_entry(f"CUDA SETUP: Defaulting to {legacy_binary_name}...") binary_path = package_dir / legacy_binary_name -- cgit v1.2.3