Fix type

Signed-off-by: Nghia Truong <[email protected]>
NVIDIA · Nov 27, 2024 · c17f007 · c17f007
1 parent d2ae155
commit c17f007
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/aggregate/aggregateFunctions.scala b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/aggregate/aggregateFunctions.scala
@@ -64,7 +64,8 @@ class CudfSum(override val dataType: DataType) extends CudfAggregate {
 
   override val reductionAggregate: cudf.ColumnVector => cudf.Scalar =
     (col: cudf.ColumnVector) => col.reduce(ReductionAggregation.hostUDF(
-      AggregationUtils.createTestHostUDF(AggregationUtils.AggregationType.Reduction)), DType.INT64)
+      AggregationUtils.createTestHostUDF(AggregationUtils.AggregationType.Reduction)), 
+      DType.FLOAT64)
   override lazy val groupByAggregate: GroupByAggregation =
     GroupByAggregation.hostUDF(
     AggregationUtils.createTestHostUDF(AggregationUtils.AggregationType.GroupByAggregation))