Merge pull request #37 from tomaarsen/hotfix/colab_just_cpu

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

View File

@ -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