Skip to content

Commit b1d9a36

Browse files
committed
typo
1 parent e5f34a1 commit b1d9a36

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

aten/cuda/cluster_kernel.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ at::Tensor grid_cuda(at::Tensor pos, at::Tensor size, at::Tensor start,
3232
const int threads = 1024;
3333
const dim3 blocks((num_nodes + threads - 1) / threads);
3434

35-
AT_DISPATCH_ALL_TYPES(pos.type(), "unique", [&] {
35+
AT_DISPATCH_ALL_TYPES(pos.type(), "grid_cuda_kernel", [&] {
3636
auto cluster_data = cluster.data<int64_t>();
3737
auto pos_info = at::cuda::detail::getTensorInfo<scalar_t, int>(pos);
3838
auto size_data = size.data<scalar_t>();

0 commit comments

Comments
 (0)