From b5fda09ac9ab65af10f6c966d0809184332f79ab Mon Sep 17 00:00:00 2001
From: MithunR <mythrocks@gmail.com>
Date: Wed, 6 Dec 2023 18:35:28 -0800
Subject: [PATCH] Removed erroneous batching.

This commit fixes the batching.  The new exec should not have to receive
batched input.
---
 .../nvidia/spark/rapids/GpuBatchedBoundedWindowExec.scala | 8 ++------
 1 file changed, 2 insertions(+), 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
index e2141ef3ff8..5f31009ed41 100644
--- 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)