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
bdb05cec
Commit
bdb05cec
authored
Dec 28, 2017
by
rusty1s
Browse files
backward coverage can not be identified
parent
a10a29f4
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
torch_scatter/functions/scatter.py
torch_scatter/functions/scatter.py
+2
-2
No files found.
torch_scatter/functions/scatter.py
View file @
bdb05cec
...
...
@@ -39,7 +39,7 @@ def _scatter(name, dim, *data):
return
(
data
[
0
],
data
[
3
])
if
has_arg
(
name
)
else
data
[
0
]
def
index_backward
(
dim
,
index
,
grad
,
arg
):
def
index_backward
(
dim
,
index
,
grad
,
arg
):
# pragma: no cover
typename
=
type
(
grad
).
__name__
.
replace
(
'Tensor'
,
''
)
cuda
=
'cuda_'
if
grad
.
is_cuda
else
''
func
=
getattr
(
ffi
,
'index_backward_{}{}'
.
format
(
cuda
,
typename
))
...
...
@@ -72,7 +72,7 @@ class _Scatter(Function):
self
.
save_for_backward
(
data
[
1
])
return
data
[
0
]
def
backward
(
self
,
*
data
):
def
backward
(
self
,
*
data
):
# pragma: no cover
grad_output
=
grad_input
=
None
if
self
.
needs_input_grad
[
0
]:
...
...
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