Removed erroneous batching.

This commit fixes the batching. The new exec should not have to receive batched input.
NVIDIA · Dec 7, 2023 · b5fda09 · b5fda09
1 parent 0ba8be6
commit b5fda09
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuBatchedBoundedWindowExec.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuBatchedBoundedWindowExec.scala
@@ -235,13 +235,9 @@ class GpuBatchedBoundedWindowExec(
   override def otherCopyArgs: Seq[AnyRef] =
     cpuPartitionSpec :: cpuOrderSpec :: minPreceding :: maxFollowing :: Nil
 
-  override def childrenCoalesceGoal: Seq[CoalesceGoal] = Seq(outputBatching)
+  override def childrenCoalesceGoal: Seq[CoalesceGoal] = Seq.fill(children.size)(null)
 
-  override def outputBatching: CoalesceGoal = if (gpuPartitionSpec.isEmpty) {
-    RequireSingleBatch
-  } else {
-    BatchedByKey(gpuPartitionOrdering)(cpuPartitionOrdering)
-  }
+  override def outputBatching: CoalesceGoal = null
 
   override protected def internalDoExecuteColumnar(): RDD[ColumnarBatch] = {
     val numOutputBatches = gpuLongMetric(GpuMetric.NUM_OUTPUT_BATCHES)