finalize/close can double close broadcast in GpuBroadcastExchangeExec

NVIDIA · Jan 13, 2023 · c123170 · c123170
1 parent f861823
commit c123170
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...lugin/src/main/scala/org/apache/spark/sql/rapids/execution/GpuBroadcastExchangeExec.scala b/...lugin/src/main/scala/org/apache/spark/sql/rapids/execution/GpuBroadcastExchangeExec.scala
@@ -226,7 +226,9 @@ class SerializeConcatHostBuffersDeserializeBatch(
 
   override def close(): Unit = this.synchronized {
     buffers.safeClose()
+    buffers = Array.empty
     Option(batchInternal).foreach(_.close())
+    batchInternal = null
   }
 
   @scala.annotation.nowarn("msg=method finalize in class Object is deprecated")