deftdawg/bitsandbytes-rocm
1
0
Fork 0
forked from mrq/bitsandbytes-rocm
Code Pull Requests Activity

minor refactor to more concise syntax

Browse Source
This commit is contained in:
Titus von Koeller 2022-08-01 03:22:12 -07:00
parent 4a6ea7e24b
commit 57fa64628f
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
bitsandbytes/cuda_setup.py
View File

@ -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