forked from mrq/bitsandbytes-rocm
Merge pull request #37 from tomaarsen/hotfix/colab_just_cpu
Perform check using implicit list length
This commit is contained in:
commit
76699b4a8d
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue
Block a user