Skip to content

GitLab

Commit 388a2e2b authored by rusty1s's avatar rusty1s
Browse files

typo

parent 4f7761ae
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
csrc/cuda/nearest_cuda.cu
View file @ 388a2e2b
...@@ -73,7 +73,7 @@ torch::Tensor nearest_cuda(torch::Tensor x, torch::Tensor y, ...@@ -73,7 +73,7 @@ torch::Tensor nearest_cuda(torch::Tensor x, torch::Tensor y,
x = x.view({x.size(0), -1}).contiguous(); x = x.view({x.size(0), -1}).contiguous();
y = y.view({y.size(0), -1}).contiguous(); y = y.view({y.size(0), -1}).contiguous();
auto out = torch::empty({x.size(0), ptr_x.options()}); auto out = torch::empty({x.size(0)}, ptr_x.options());
auto stream = at::cuda::getCurrentCUDAStream(); auto stream = at::cuda::getCurrentCUDAStream();
AT_DISPATCH_FLOATING_TYPES(x.scalar_type(), "nearest_kernel", [&] { AT_DISPATCH_FLOATING_TYPES(x.scalar_type(), "nearest_kernel", [&] {
......
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