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