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

fix version

parent e1db3288
......@@ -85,6 +85,6 @@ deploy:
acl: public_read
on:
repo: rusty1s/pytorch_sparse
tags: true
branch: master
notifications:
email: false
......@@ -18,4 +18,4 @@ int64_t cuda_version() {
}
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:
f'version {major}.{minor}.')
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 < 10000:
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