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-spline-conv
Commits
ddd591d6
Commit
ddd591d6
authored
Apr 01, 2020
by
rusty1s
Browse files
clean up
parent
cd7b1988
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
13 deletions
+4
-13
.travis.yml
.travis.yml
+1
-1
torch_spline_conv/__init__.py
torch_spline_conv/__init__.py
+3
-12
No files found.
.travis.yml
View file @
ddd591d6
...
...
@@ -85,6 +85,6 @@ deploy:
acl
:
public_read
on
:
repo
:
rusty1s/pytorch_spline_conv
tags
:
true
condition
:
$TRAVIS_TAG != "" || $TRAVIS_COMMIT_MESSAGE =~ ci-deploy
notifications
:
email
:
false
torch_spline_conv/__init__.py
View file @
ddd591d6
...
...
@@ -6,18 +6,9 @@ import torch
__version__
=
'1.2.0'
expected_torch_version
=
(
1
,
4
)
try
:
for
library
in
[
'_version'
,
'_basis'
,
'_weighting'
]:
torch
.
ops
.
load_library
(
importlib
.
machinery
.
PathFinder
().
find_spec
(
library
,
[
osp
.
dirname
(
__file__
)]).
origin
)
except
OSError
as
e
:
major
,
minor
=
[
int
(
x
)
for
x
in
torch
.
__version__
.
split
(
'.'
)[:
2
]]
t_major
,
t_minor
=
expected_torch_version
if
major
!=
t_major
or
(
major
==
t_major
and
minor
!=
t_minor
):
raise
RuntimeError
(
f
'Expected PyTorch version
{
t_major
}
.
{
t_minor
}
but found '
f
'version
{
major
}
.
{
minor
}
.'
)
raise
OSError
(
e
)
for
library
in
[
'_version'
,
'_basis'
,
'_weighting'
]:
torch
.
ops
.
load_library
(
importlib
.
machinery
.
PathFinder
().
find_spec
(
library
,
[
osp
.
dirname
(
__file__
)]).
origin
)
if
torch
.
version
.
cuda
is
not
None
:
# pragma: no cover
cuda_version
=
torch
.
ops
.
torch_spline_conv
.
cuda_version
()
...
...
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