Skip to content

GitLab

Commit bdb05cec authored by rusty1s's avatar rusty1s
Browse files

backward coverage can not be identified

parent a10a29f4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torch_scatter/functions/scatter.py
View file @ bdb05cec
...@@ -39,7 +39,7 @@ def _scatter(name, dim, *data): ...@@ -39,7 +39,7 @@ def _scatter(name, dim, *data):
return (data[0], data[3]) if has_arg(name) else data[0] 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', '') typename = type(grad).__name__.replace('Tensor', '')
cuda = 'cuda_' if grad.is_cuda else '' cuda = 'cuda_' if grad.is_cuda else ''
func = getattr(ffi, 'index_backward_{}{}'.format(cuda, typename)) func = getattr(ffi, 'index_backward_{}{}'.format(cuda, typename))
...@@ -72,7 +72,7 @@ class _Scatter(Function): ...@@ -72,7 +72,7 @@ class _Scatter(Function):
self.save_for_backward(data[1]) self.save_for_backward(data[1])
return data[0] return data[0]
def backward(self, *data): def backward(self, *data): # pragma: no cover
grad_output = grad_input = None grad_output = grad_input = None
if self.needs_input_grad[0]: if self.needs_input_grad[0]:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment