summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--bitsandbytes/cuda_setup.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/bitsandbytes/cuda_setup.py b/bitsandbytes/cuda_setup.py
index 08f3ed2..2fb5049 100644
--- a/bitsandbytes/cuda_setup.py
+++ b/bitsandbytes/cuda_setup.py
@@ -131,8 +131,7 @@ def evaluate_cuda_setup():
cc = get_compute_capability()
binary_name = 'libbitsandbytes_cpu.so'
- has_gpu = cc != ''
- if not has_gpu:
+ if not (has_gpu := bool(cc)):
print('WARNING: No GPU detected! Check our CUDA paths. Processing to load CPU-only library...')
return binary_name