Skip to content

GitLab

Unverified Commit fbe0990e authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Merge pull request #149 from mpariente/cuda_check 

 Check if CUDA is available before checking CUDA version
parents 54d23137 4945965c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torch_scatter/__init__.py
View file @ fbe0990e
...@@ -39,7 +39,7 @@ except AttributeError as e: ...@@ -39,7 +39,7 @@ except AttributeError as e:
torch.ops.torch_scatter.segment_max_coo = segment_coo_arg_placeholder torch.ops.torch_scatter.segment_max_coo = segment_coo_arg_placeholder
torch.ops.torch_scatter.gather_coo = gather_coo_placeholder torch.ops.torch_scatter.gather_coo = gather_coo_placeholder
if torch.version.cuda is not None: # pragma: no cover if torch.version.cuda is not None and torch.cuda.is_available(): # pragma: no cover
cuda_version = torch.ops.torch_scatter.cuda_version() cuda_version = torch.ops.torch_scatter.cuda_version()
if cuda_version == -1: if cuda_version == -1:
......
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