Skip to content

Commit 17a36cd

Browse files
committed
fix version
1 parent e1db328 commit 17a36cd

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

.travis.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,6 @@ deploy:
8585
acl: public_read
8686
on:
8787
repo: rusty1s/pytorch_sparse
88-
tags: true
88+
branch: master
8989
notifications:
9090
email: false

csrc/version.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,4 +18,4 @@ int64_t cuda_version() {
1818
}
1919

2020
static auto registry =
21-
torch::RegisterOperators().op("torch_scatter::cuda_version", &cuda_version);
21+
torch::RegisterOperators().op("torch_sparse::cuda_version", &cuda_version);

torch_sparse/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
f'version {major}.{minor}.')
2222
raise OSError(e)
2323

24-
cuda_version = torch.ops.torch_scatter.cuda_version()
24+
cuda_version = torch.ops.torch_sparse.cuda_version()
2525
if cuda_version != -1 and torch.version.cuda is not None: # pragma: no cover
2626
if cuda_version < 10000:
2727
major, minor = int(str(cuda_version)[0]), int(str(cuda_version)[2])

0 commit comments

Comments
 (0)