Skip to content

Commit 2520670

Browse files
committed
typo
1 parent 62fc610 commit 2520670

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cuda/scatter_kernel.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,8 @@ scatter_min_kernel(at::cuda::detail::TensorInfo<scalar_t, int64_t> src,
147147
}
148148
}
149149

150-
void scatter_min_cuda(at::Tensor src, at::Tensor index, at::Tensor out,
151-
at::Tensor arg, int64_t dim) {
150+
void scatter_min_cuda(torch::Tensor src, torch::Tensor index, torch::Tensor out,
151+
torch::Tensor arg, int64_t dim) {
152152
cudaSetDevice(src.get_device());
153153
AT_DISPATCH_ALL_TYPES(src.scalar_type(), "scatter_min_kernel", [&] {
154154
auto src_info = at::cuda::detail::getTensorInfo<scalar_t, int64_t>(src);

0 commit comments

Comments
 (0)