Unverified Commit 76699b4a authored by Tim Dettmers's avatar Tim Dettmers Committed by GitHub
Browse files

Merge pull request #37 from tomaarsen/hotfix/colab_just_cpu

Perform check using implicit list length
parents 7740c6e9 58aa7c53
...@@ -103,7 +103,7 @@ def get_compute_capability(cuda): ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment