Merge pull request #10482 from NVIDIA/branch-24.02

[auto-merge] branch-24.02 to branch-24.04 [skip ci] [bot]
NVIDIA · Feb 23, 2024 · 4976e8d · 4976e8d
2 parents 5be63f9 + cf32321
commit 4976e8d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/window/GpuWindowExpression.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/window/GpuWindowExpression.scala
@@ -24,6 +24,7 @@ import com.nvidia.spark.Retryable
 import com.nvidia.spark.rapids._
 import com.nvidia.spark.rapids.Arm.{closeOnExcept, withResource}
 import com.nvidia.spark.rapids.GpuOverrides.wrapExpr
+import com.nvidia.spark.rapids.RapidsPluginImplicits._
 import com.nvidia.spark.rapids.shims.{GpuWindowUtil, ShimExpression}
 import scala.util.{Left, Right}
 
@@ -1715,8 +1716,9 @@ class RankFixer extends BatchedRunningWindowFixer with Logging {
   }
 
   override def close(): Unit = {
-    previousRank.foreach(_.close())
+    previousRank.foreach(_.safeClose())
     previousRank = None
+    rowNumFixer.close()
   }
 }