diff options
| author | Tim Dettmers <TimDettmers@users.noreply.github.com> | 2022-10-09 16:43:58 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-09 16:43:58 -0700 |
| commit | 76699b4a8d8dd55ab7da8bcf68a849c723a0c0ab (patch) | |
| tree | 862b0b8a4e2cc050f53f2b1c661e7d2398f5e138 /bitsandbytes/cuda_setup | |
| parent | 7740c6e9c959f164b6836768295f661c76a39c06 (diff) | |
| parent | 58aa7c53f679a30e05da57939ed72a7512c854c6 (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.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 |