From cf7d7249f238bbf1509a861293bbe9bee0ce9c3e Mon Sep 17 00:00:00 2001 From: Ankita Victor-Levi Date: Fri, 13 Mar 2026 12:28:00 +0530 Subject: [PATCH] Close batch --- .../apache/gluten/execution/ColumnarCollectLimitExec.scala | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backends-velox/src/main/scala/org/apache/gluten/execution/ColumnarCollectLimitExec.scala b/backends-velox/src/main/scala/org/apache/gluten/execution/ColumnarCollectLimitExec.scala index 61c42715ea53..a14e97490c7f 100644 --- a/backends-velox/src/main/scala/org/apache/gluten/execution/ColumnarCollectLimitExec.scala +++ b/backends-velox/src/main/scala/org/apache/gluten/execution/ColumnarCollectLimitExec.scala @@ -70,6 +70,7 @@ case class ColumnarCollectLimitExec( if (rowsToSkip >= batchSize) { rowsToSkip -= batchSize + batch.close() } else { val startIndex = rowsToSkip val leftoverAfterSkip = batchSize - startIndex @@ -82,7 +83,9 @@ case class ColumnarCollectLimitExec( ColumnarBatches.retain(batch) batch } else { - VeloxColumnarBatches.slice(batch, startIndex, needed) + val sliced = VeloxColumnarBatches.slice(batch, startIndex, needed) + batch.close() + sliced } rowsToCollect -= needed