Fix signature

Author: rui-mo

velox/functions/sparksql/aggregates/AverageAggregate.cpp

@@ -429,7 +429,9 @@ exec::AggregateRegistrationResult registerAverage(
                 auto inputScale = inputType->asShortDecimal().scale();
                 auto sumType =
                     DECIMAL(std::min(38, inputPrecision + 10), inputScale);
-                if (exec::isPartialOutput(step)) {
+                if (exec::isPartialOutput(step) ||
+                    (step == core::AggregationNode::Step::kSingle &&
+                     resultType->isRow())) {
                   return std::make_unique<
                       DecimalAverageAggregate<int64_t, int64_t>>(
                       resultType, sumType);

velox/functions/sparksql/aggregates/SumAggregate.cpp

@@ -29,7 +29,13 @@ using SumAggregate = SumAggregateBase<TInput, TAccumulator, ResultType, true>;
 TypePtr getDecimalSumType(
     const TypePtr& resultType,
     core::AggregationNode::Step step) {
-  return exec::isPartialOutput(step) ? resultType->childAt(0) : resultType;
+  if (exec::isPartialOutput(step)) {
+    return resultType->childAt(0);
+  }
+  if (step == core::AggregationNode::Step::kSingle && resultType->isRow()) {
+    return resultType->childAt(0);
+  }
+  return resultType;
 }
 } // namespace
 

velox/functions/sparksql/tests/DecimalUtilTest.cpp

@@ -30,7 +30,7 @@ class DecimalUtilTest : public testing::Test {
       R expectedResult,
       bool expectedOverflow) {
     R r;
-    bool overflow;
+    bool overflow = false;
     DecimalUtil::divideWithRoundUp<R, A, B>(r, a, b, aRescale, overflow);
     ASSERT_EQ(overflow, expectedOverflow);
     ASSERT_EQ(r, expectedResult);