Address half->int build error (#2240)

Summary: Pull Request resolved: #2240 Reviewed By: sryap Differential Revision: D52469670 fbshipit-source-id: ebad4580a4b653967cbf0fcd15c8ebd4908aa80d
pytorch · Jan 1, 2024 · 905ff66 · 905ff66
1 parent cfdc6b8
commit 905ff66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fbgemm_gpu/src/jagged_tensor_ops/dense_to_jagged_forward.cu b/fbgemm_gpu/src/jagged_tensor_ops/dense_to_jagged_forward.cu
@@ -49,8 +49,8 @@ Tensor dense_to_jagged_forward(
       values.scalar_type(),
       "dense_to_jagged_gpu_op_forward",
       DISPATCH_DENSE_TO_JAGGED_CASE(at::ScalarType::Half)
-      DISPATCH_DENSE_TO_JAGGED_CASE(at::ScalarType::Int)
-      AT_DISPATCH_CASE_FLOATING_TYPES_AND2(
+      AT_DISPATCH_CASE_FLOATING_TYPES_AND3(
+          at::ScalarType::Int,
           at::ScalarType::Long,
           at::ScalarType::BFloat16,
           [&] {