Fix

sql-plugin/src/main/scala/org/apache/spark/sql/rapids/aggregate/GpuHyperLogLogPlusPlus.scala

@@ -68,16 +68,16 @@ case class GpuHyperLogLogPlusPlusEvaluation(children: Seq[Expression],
 
   override def columnarEval(batch: ColumnarBatch): GpuColumnVector = {
 
-  // TODO
-//    withResourceIfAllowed(childExpr.columnarEval(batch)) { sketches =>
-//      val distinctValues = HLL.estimateDistinctValueFromSketches(
+    // TODO
+    //    withResourceIfAllowed(childExpr.columnarEval(batch)) { sketches =>
+    //      val distinctValues = HLL.estimateDistinctValueFromSketches(
     //      sketches.getBase, numRegistersPerSketch)
-//      GpuColumnVector.from(distinctValues, LongType)
-//    }
+    //      GpuColumnVector.from(distinctValues, LongType)
+    //    }
 
     val numRows = batch.numRows()
     val zeros = Array.fill(numRows)(0L)
-    val longCv = ColumnVector.fromLongs(zeros : _*)
+    val longCv = ColumnVector.fromLongs(zeros: _*)
     GpuColumnVector.from(longCv, LongType)
   }
 }
@@ -140,15 +140,10 @@ case class GpuHyperLogLogPlusPlus(childExpr: Expression, relativeSD: Double)
   /**
    * Convert Struct<long, ..., long> column to long columns
    */
-  private def extractChildren: Seq[Expression] = Seq.tabulate(hllppHelper.numWords) {
+  override lazy val postUpdate: Seq[Expression] = Seq.tabulate(hllppHelper.numWords) {
     i => GpuGetStructField(postUpdateAttr.head, i)
   }
 
-  /**
-   * Convert to long columns
-   */
-  override lazy val postUpdate: Seq[Expression] = extractChildren
-
   /**
    * convert to Struct<long, ..., long>
    */
@@ -158,9 +153,11 @@ case class GpuHyperLogLogPlusPlus(childExpr: Expression, relativeSD: Double)
     Seq(CudfMergeHLLPP(cuDFBufferType, numRegistersPerSketch))
 
   /**
-   * Convert to long columns
+   * Convert Struct<long, ..., long> column to long columns
    */
-  override lazy val postMerge: Seq[Expression] = extractChildren
+  override lazy val postMerge: Seq[Expression] = Seq.tabulate(hllppHelper.numWords) {
+    i => GpuGetStructField(postMergeAttr.head, i)
+  }
 
   override lazy val evaluateExpression: Expression =
     GpuHyperLogLogPlusPlusEvaluation(aggBufferAttributes, numRegistersPerSketch)