diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/SpillableColumnarBatch.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/SpillableColumnarBatch.scala index c8a7dbcbb87..4894fc82527 100644 --- a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/SpillableColumnarBatch.scala +++ b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/SpillableColumnarBatch.scala @@ -139,9 +139,12 @@ class SpillableColumnarBatchImpl ( if (refCount == 0) { // closing my reference handle.close() - } else if (refCount < 0) { - throw new IllegalStateException("Double free on SpillableColumnarBatchImpl") } + // TODO this is causing problems so we need to look into this + // https://github.com/NVIDIA/spark-rapids/issues/10161 +// else if (refCount < 0) { +// throw new IllegalStateException("Double free on SpillableColumnarBatchImpl") +// } } } diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/window/GpuWindowExecMeta.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/window/GpuWindowExecMeta.scala index 5ab6546e9c0..77e8ae679d3 100644 --- a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/window/GpuWindowExecMeta.scala +++ b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/window/GpuWindowExecMeta.scala @@ -325,7 +325,7 @@ case class BatchedOps(running: Seq[NamedExpression], } if (hasBounded) { - getBatchedBoundedWindowExec(gpuPartitionSpec, gpuOrderSpec, currentPlan, + currentPlan = getBatchedBoundedWindowExec(gpuPartitionSpec, gpuOrderSpec, currentPlan, cpuPartitionSpec, cpuOrderSpec) } currentPlan.asInstanceOf[GpuExec]