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-scatter
Commits
47745860
Commit
47745860
authored
Feb 14, 2020
by
rusty1s
Browse files
fix doc build
parent
8ca6d861
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
95 additions
and
8 deletions
+95
-8
.coveragerc
.coveragerc
+1
-0
torch_scatter/__init__.py
torch_scatter/__init__.py
+32
-8
torch_scatter/placeholder.py
torch_scatter/placeholder.py
+62
-0
No files found.
.coveragerc
View file @
47745860
[run]
[run]
source=torch_scatter
source=torch_scatter
omit=torch_scatter/placeholder.py
[report]
[report]
exclude_lines =
exclude_lines =
pragma: no cover
pragma: no cover
...
...
torch_scatter/__init__.py
View file @
47745860
# flake8: noqa
import
os
import
os
import
importlib
import
importlib
import
os.path
as
osp
import
os.path
as
osp
...
@@ -10,8 +8,9 @@ __version__ = '2.0.3'
...
@@ -10,8 +8,9 @@ __version__ = '2.0.3'
expected_torch_version
=
(
1
,
4
)
expected_torch_version
=
(
1
,
4
)
try
:
try
:
torch
.
ops
.
load_library
(
importlib
.
machinery
.
PathFinder
().
find_spec
(
for
library
in
[
'_version'
,
'_scatter'
,
'_segment_csr'
,
'_segment_coo'
]:
'_version'
,
[
osp
.
dirname
(
__file__
)]).
origin
)
torch
.
ops
.
load_library
(
importlib
.
machinery
.
PathFinder
().
find_spec
(
library
,
[
osp
.
dirname
(
__file__
)]).
origin
)
except
OSError
as
e
:
except
OSError
as
e
:
if
'undefined symbol'
in
str
(
e
):
if
'undefined symbol'
in
str
(
e
):
major
,
minor
=
[
int
(
x
)
for
x
in
torch
.
__version__
.
split
(
'.'
)[:
2
]]
major
,
minor
=
[
int
(
x
)
for
x
in
torch
.
__version__
.
split
(
'.'
)[:
2
]]
...
@@ -25,6 +24,31 @@ except AttributeError as e:
...
@@ -25,6 +24,31 @@ except AttributeError as e:
if
os
.
getenv
(
'BUILD_DOCS'
,
'0'
)
!=
'1'
:
if
os
.
getenv
(
'BUILD_DOCS'
,
'0'
)
!=
'1'
:
raise
AttributeError
(
e
)
raise
AttributeError
(
e
)
from
.placeholder
import
cuda_version_placeholder
torch
.
ops
.
torch_scatter
.
cuda_version
=
cuda_version_placeholder
from
.placeholder
import
scatter_arg_placeholder
torch
.
ops
.
torch_scatter
.
scatter_min
=
scatter_arg_placeholder
torch
.
ops
.
torch_scatter
.
scatter_max
=
scatter_arg_placeholder
from
.placeholder
import
segment_csr_placeholder
from
.placeholder
import
segment_csr_arg_placeholder
from
.placeholder
import
gather_csr_placeholder
torch
.
ops
.
torch_scatter
.
segment_sum_csr
=
segment_csr_placeholder
torch
.
ops
.
torch_scatter
.
segment_mean_csr
=
segment_csr_placeholder
torch
.
ops
.
torch_scatter
.
segment_min_csr
=
segment_csr_arg_placeholder
torch
.
ops
.
torch_scatter
.
segment_max_csr
=
segment_csr_arg_placeholder
torch
.
ops
.
torch_scatter
.
gather_csr
=
gather_csr_placeholder
from
.placeholder
import
segment_coo_placeholder
from
.placeholder
import
segment_coo_arg_placeholder
from
.placeholder
import
gather_coo_placeholder
torch
.
ops
.
torch_scatter
.
segment_sum_coo
=
segment_coo_placeholder
torch
.
ops
.
torch_scatter
.
segment_mean_coo
=
segment_coo_placeholder
torch
.
ops
.
torch_scatter
.
segment_min_coo
=
segment_coo_arg_placeholder
torch
.
ops
.
torch_scatter
.
segment_max_coo
=
segment_coo_arg_placeholder
torch
.
ops
.
torch_scatter
.
gather_coo
=
gather_coo_placeholder
if
torch
.
version
.
cuda
is
not
None
:
# pragma: no cover
if
torch
.
version
.
cuda
is
not
None
:
# pragma: no cover
cuda_version
=
torch
.
ops
.
torch_scatter
.
cuda_version
()
cuda_version
=
torch
.
ops
.
torch_scatter
.
cuda_version
()
...
@@ -45,15 +69,15 @@ if torch.version.cuda is not None: # pragma: no cover
...
@@ -45,15 +69,15 @@ if torch.version.cuda is not None: # pragma: no cover
f
'matches your PyTorch install.'
)
f
'matches your PyTorch install.'
)
from
.scatter
import
(
scatter_sum
,
scatter_add
,
scatter_mean
,
scatter_min
,
from
.scatter
import
(
scatter_sum
,
scatter_add
,
scatter_mean
,
scatter_min
,
scatter_max
,
scatter
)
scatter_max
,
scatter
)
# noqa: E402
from
.segment_csr
import
(
segment_sum_csr
,
segment_add_csr
,
segment_mean_csr
,
from
.segment_csr
import
(
segment_sum_csr
,
segment_add_csr
,
segment_mean_csr
,
segment_min_csr
,
segment_max_csr
,
segment_csr
,
segment_min_csr
,
segment_max_csr
,
segment_csr
,
gather_csr
)
gather_csr
)
# noqa: E402
from
.segment_coo
import
(
segment_sum_coo
,
segment_add_coo
,
segment_mean_coo
,
from
.segment_coo
import
(
segment_sum_coo
,
segment_add_coo
,
segment_mean_coo
,
segment_min_coo
,
segment_max_coo
,
segment_coo
,
segment_min_coo
,
segment_max_coo
,
segment_coo
,
gather_coo
)
gather_coo
)
# noqa: E402
from
.composite
import
(
scatter_std
,
scatter_logsumexp
,
scatter_softmax
,
from
.composite
import
(
scatter_std
,
scatter_logsumexp
,
scatter_softmax
,
scatter_log_softmax
)
scatter_log_softmax
)
# noqa: E402
__all__
=
[
__all__
=
[
'scatter_sum'
,
'scatter_sum'
,
...
...
torch_scatter/placeholder.py
0 → 100644
View file @
47745860
from
typing
import
Optional
,
Tuple
import
torch
def
cuda_version_placeholder
()
->
int
:
return
-
1
def
scatter_placeholder
(
src
:
torch
.
Tensor
,
index
:
torch
.
Tensor
,
dim
:
int
,
out
:
Optional
[
torch
.
Tensor
],
dim_size
:
Optional
[
int
])
->
torch
.
Tensor
:
raise
ImportError
return
src
def
scatter_arg_placeholder
(
src
:
torch
.
Tensor
,
index
:
torch
.
Tensor
,
dim
:
int
,
out
:
Optional
[
torch
.
Tensor
],
dim_size
:
Optional
[
int
]
)
->
Tuple
[
torch
.
Tensor
,
torch
.
Tensor
]:
raise
ImportError
return
src
,
index
def
segment_csr_placeholder
(
src
:
torch
.
Tensor
,
indptr
:
torch
.
Tensor
,
out
:
Optional
[
torch
.
Tensor
])
->
torch
.
Tensor
:
raise
ImportError
return
src
def
segment_csr_arg_placeholder
(
src
:
torch
.
Tensor
,
indptr
:
torch
.
Tensor
,
out
:
Optional
[
torch
.
Tensor
]
)
->
Tuple
[
torch
.
Tensor
,
torch
.
Tensor
]:
raise
ImportError
return
src
,
indptr
def
gather_csr_placeholder
(
src
:
torch
.
Tensor
,
indptr
:
torch
.
Tensor
,
out
:
Optional
[
torch
.
Tensor
])
->
torch
.
Tensor
:
raise
ImportError
return
src
def
segment_coo_placeholder
(
src
:
torch
.
Tensor
,
index
:
torch
.
Tensor
,
out
:
Optional
[
torch
.
Tensor
],
dim_size
:
Optional
[
int
])
->
torch
.
Tensor
:
raise
ImportError
return
src
def
segment_coo_arg_placeholder
(
src
:
torch
.
Tensor
,
index
:
torch
.
Tensor
,
out
:
Optional
[
torch
.
Tensor
],
dim_size
:
Optional
[
int
]
)
->
Tuple
[
torch
.
Tensor
,
torch
.
Tensor
]:
raise
ImportError
return
src
,
index
def
gather_coo_placeholder
(
src
:
torch
.
Tensor
,
index
:
torch
.
Tensor
,
out
:
Optional
[
torch
.
Tensor
])
->
torch
.
Tensor
:
raise
ImportError
return
src
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