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
2f2dd12f
Commit
2f2dd12f
authored
Jan 15, 2018
by
rusty1s
Browse files
typo
parent
cc6f8862
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
torch_scatter/kernel/kernel.cu
torch_scatter/kernel/kernel.cu
+3
-1
No files found.
torch_scatter/kernel/kernel.cu
View file @
2f2dd12f
...
...
@@ -75,7 +75,9 @@ __global__ void indexBackwardKernel(TensorInfo<Real> output, TensorInfo<int64_t>
KERNEL_LOOP
(
i
,
n
)
{
int
outputOffset
=
0
;
int
indexOffset
=
0
;
int
gradOffset
=
0
;
int
argOffset
=
0
;
IndexToScatterOffsets4
<
Real
,
Real
,
int64_t
,
Dims
>::
compute
(
i
,
dim
,
index
,
&
indexOffset
,
output
,
&
outputOffset
,
grad
,
&
gradOffset
,
arg
,
&
argOffset
);
if
(
arg
.
data
[
argOffset
]
==
(
outputOffset
/
output
.
stride
[
dim
])
%
output
.
size
[
dim
])
output
.
data
[
outputOffset
]
=
grad
.
data
[
gradOffset
];
if
(
arg
.
data
[
argOffset
]
==
(
outputOffset
/
output
.
stride
[
dim
])
%
output
.
size
[
dim
])
{
output
.
data
[
outputOffset
]
=
grad
.
data
[
gradOffset
];
}
}
}
...
...
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