summaryrefslogtreecommitdiff
path: root/bitsandbytes/cuda_setup
diff options
context:
space:
mode:
authorTim Dettmers <TimDettmers@users.noreply.github.com>2022-10-09 16:43:58 -0700
committerGitHub <noreply@github.com>2022-10-09 16:43:58 -0700
commit76699b4a8d8dd55ab7da8bcf68a849c723a0c0ab (patch)
tree862b0b8a4e2cc050f53f2b1c661e7d2398f5e138 /bitsandbytes/cuda_setup
parent7740c6e9c959f164b6836768295f661c76a39c06 (diff)
parent58aa7c53f679a30e05da57939ed72a7512c854c6 (diff)
Merge pull request #37 from tomaarsen/hotfix/colab_just_cpu
Perform check using implicit list length
Diffstat (limited to 'bitsandbytes/cuda_setup')
-rw-r--r--bitsandbytes/cuda_setup/main.py2
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