fix scatter integer types

rapidsai · Sep 27, 2023 · 3618fa3 · 3618fa3
1 parent 2fd6872
commit 3618fa3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpp/src/wholememory_ops/functions/scatter_func_impl_integer_data_int64_indices.cu b/cpp/src/wholememory_ops/functions/scatter_func_impl_integer_data_int64_indices.cu
@@ -37,8 +37,8 @@ void scatter_integer_int64_temp_func(const void* input,
 
 REGISTER_DISPATCH_TWO_TYPES(ScatterFuncIntegerInt64,
                             scatter_integer_int64_temp_func,
-                            HALF_FLOAT_DOUBLE,
-                            HALF_FLOAT_DOUBLE)
+                            ALLSINT,
+                            ALLSINT)
 
 wholememory_error_code_t scatter_integer_int64_func(const void* input,
                                                     wholememory_matrix_description_t input_desc,