Commit b1d9a365 authored by rusty1s's avatar rusty1s
Browse files

typo

parent e5f34a12
...@@ -32,7 +32,7 @@ at::Tensor grid_cuda(at::Tensor pos, at::Tensor size, at::Tensor start, ...@@ -32,7 +32,7 @@ at::Tensor grid_cuda(at::Tensor pos, at::Tensor size, at::Tensor start,
const int threads = 1024; const int threads = 1024;
const dim3 blocks((num_nodes + threads - 1) / threads); const dim3 blocks((num_nodes + threads - 1) / threads);
AT_DISPATCH_ALL_TYPES(pos.type(), "unique", [&] { AT_DISPATCH_ALL_TYPES(pos.type(), "grid_cuda_kernel", [&] {
auto cluster_data = cluster.data<int64_t>(); auto cluster_data = cluster.data<int64_t>();
auto pos_info = at::cuda::detail::getTensorInfo<scalar_t, int>(pos); auto pos_info = at::cuda::detail::getTensorInfo<scalar_t, int>(pos);
auto size_data = size.data<scalar_t>(); auto size_data = size.data<scalar_t>();
......
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