deftdawg/bitsandbytes-rocm
1
0
Fork 0
forked from mrq/bitsandbytes-rocm
Code Pull Requests Activity

Merge pull request #37 from tomaarsen/hotfix/colab_just_cpu

Browse Source 
Perform check using implicit list length
This commit is contained in:
Tim Dettmers 2022-10-09 16:43:58 -07:00 committed by GitHub
commit 76699b4a8d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bitsandbytes/cuda_setup/main.py
View File

@ -103,7 +103,7 @@ def get_compute_capability(cuda):
None. None.
""" """
ccs = get_compute_capabilities(cuda) ccs = get_compute_capabilities(cuda)
if ccs is not None: if ccs:
# TODO: handle different compute capabilities; for now, take the max # TODO: handle different compute capabilities; for now, take the max
return ccs[-1] return ccs[-1]
return None return None