Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-sparse
Commits
17a36cd8
Commit
17a36cd8
authored
Feb 13, 2020
by
rusty1s
Browse files
fix version
parent
e1db3288
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
3 deletions
+3
-3
.travis.yml
.travis.yml
+1
-1
csrc/version.cpp
csrc/version.cpp
+1
-1
torch_sparse/__init__.py
torch_sparse/__init__.py
+1
-1
No files found.
.travis.yml
View file @
17a36cd8
...
@@ -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
csrc/version.cpp
View file @
17a36cd8
...
@@ -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_s
catter
::cuda_version"
,
&
cuda_version
);
torch
::
RegisterOperators
().
op
(
"torch_s
parse
::cuda_version"
,
&
cuda_version
);
torch_sparse/__init__.py
View file @
17a36cd8
...
@@ -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_s
catter
.
cuda_version
()
cuda_version
=
torch
.
ops
.
torch_s
parse
.
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
])
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment