diff options
Diffstat (limited to 'bitsandbytes/cuda_setup')
| -rw-r--r-- | bitsandbytes/cuda_setup/main.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py index 78a2844..f11b430 100644 --- a/bitsandbytes/cuda_setup/main.py +++ b/bitsandbytes/cuda_setup/main.py @@ -103,7 +103,7 @@ def get_compute_capability(cuda): None. """ ccs = get_compute_capabilities(cuda) - if ccs is not None: + if ccs: # TODO: handle different compute capabilities; for now, take the max return ccs[-1] return None |