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
2520670a
Commit
2520670a
authored
Jan 29, 2020
by
rusty1s
Browse files
typo
parent
62fc6106
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
cuda/scatter_kernel.cu
cuda/scatter_kernel.cu
+2
-2
No files found.
cuda/scatter_kernel.cu
View file @
2520670a
...
@@ -147,8 +147,8 @@ scatter_min_kernel(at::cuda::detail::TensorInfo<scalar_t, int64_t> src,
...
@@ -147,8 +147,8 @@ scatter_min_kernel(at::cuda::detail::TensorInfo<scalar_t, int64_t> src,
}
}
}
}
void
scatter_min_cuda
(
a
t
::
Tensor
src
,
a
t
::
Tensor
index
,
a
t
::
Tensor
out
,
void
scatter_min_cuda
(
t
orch
::
Tensor
src
,
t
orch
::
Tensor
index
,
t
orch
::
Tensor
out
,
a
t
::
Tensor
arg
,
int64_t
dim
)
{
t
orch
::
Tensor
arg
,
int64_t
dim
)
{
cudaSetDevice
(
src
.
get_device
());
cudaSetDevice
(
src
.
get_device
());
AT_DISPATCH_ALL_TYPES
(
src
.
scalar_type
(),
"scatter_min_kernel"
,
[
&
]
{
AT_DISPATCH_ALL_TYPES
(
src
.
scalar_type
(),
"scatter_min_kernel"
,
[
&
]
{
auto
src_info
=
at
::
cuda
::
detail
::
getTensorInfo
<
scalar_t
,
int64_t
>
(
src
);
auto
src_info
=
at
::
cuda
::
detail
::
getTensorInfo
<
scalar_t
,
int64_t
>
(
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