Skip to content

GitLab

Commit 57fa6462 authored by Titus von Koeller's avatar Titus von Koeller
Browse files

minor refactor to more concise syntax

parent 4a6ea7e2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
bitsandbytes/cuda_setup.py
View file @ 57fa6462
...@@ -131,8 +131,7 @@ def evaluate_cuda_setup(): ...@@ -131,8 +131,7 @@ def evaluate_cuda_setup():
cc = get_compute_capability() cc = get_compute_capability()
binary_name = 'libbitsandbytes_cpu.so' binary_name = 'libbitsandbytes_cpu.so'
has_gpu = cc != '' if not (has_gpu := bool(cc)):
if not has_gpu:
print('WARNING: No GPU detected! Check our CUDA paths. Processing to load CPU-only library...') print('WARNING: No GPU detected! Check our CUDA paths. Processing to load CPU-only library...')
return binary_name return binary_name
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment