Commit 17a36cd8 authored by rusty1s's avatar rusty1s
Browse files

fix version

parent e1db3288
...@@ -85,6 +85,6 @@ deploy: ...@@ -85,6 +85,6 @@ deploy:
acl: public_read acl: public_read
on: on:
repo: rusty1s/pytorch_sparse repo: rusty1s/pytorch_sparse
tags: true branch: master
notifications: notifications:
email: false email: false
...@@ -18,4 +18,4 @@ int64_t cuda_version() { ...@@ -18,4 +18,4 @@ int64_t cuda_version() {
} }
static auto registry = static auto registry =
torch::RegisterOperators().op("torch_scatter::cuda_version", &cuda_version); torch::RegisterOperators().op("torch_sparse::cuda_version", &cuda_version);
...@@ -21,7 +21,7 @@ except OSError as e: ...@@ -21,7 +21,7 @@ except OSError as e:
f'version {major}.{minor}.') f'version {major}.{minor}.')
raise OSError(e) raise OSError(e)
cuda_version = torch.ops.torch_scatter.cuda_version() cuda_version = torch.ops.torch_sparse.cuda_version()
if cuda_version != -1 and torch.version.cuda is not None: # pragma: no cover if cuda_version != -1 and torch.version.cuda is not None: # 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