We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e5f34a1 commit b1d9a36Copy full SHA for b1d9a36
aten/cuda/cluster_kernel.cu
@@ -32,7 +32,7 @@ at::Tensor grid_cuda(at::Tensor pos, at::Tensor size, at::Tensor start,
32
const int threads = 1024;
33
const dim3 blocks((num_nodes + threads - 1) / threads);
34
35
- AT_DISPATCH_ALL_TYPES(pos.type(), "unique", [&] {
+ AT_DISPATCH_ALL_TYPES(pos.type(), "grid_cuda_kernel", [&] {
36
auto cluster_data = cluster.data<int64_t>();
37
auto pos_info = at::cuda::detail::getTensorInfo<scalar_t, int>(pos);
38
auto size_data = size.data<scalar_t>();
0 commit comments