Commit 5ab67263 authored by rusty1s's avatar rusty1s
Browse files

typo

parent 04f797ea
...@@ -20,7 +20,7 @@ for library in [ ...@@ -20,7 +20,7 @@ for library in [
raise ImportError(f"Could not find module '{library}_cpu' in " raise ImportError(f"Could not find module '{library}_cpu' in "
f"{osp.dirname(__file__)}") f"{osp.dirname(__file__)}")
cuda_version = torch.ops.torch_sparse.cuda_version() cuda_version = torch.ops.torch_cluster.cuda_version()
if torch.cuda.is_available() and cuda_version != -1: # pragma: no cover if torch.cuda.is_available() and cuda_version != -1: # pragma: no cover
if cuda_version < 10000: if cuda_version < 10000:
major, minor = int(str(cuda_version)[0]), int(str(cuda_version)[2]) major, minor = int(str(cuda_version)[0]), int(str(cuda_version)[2])
......
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