Update sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuPartition…

…ing.scala Co-authored-by: Jason Lowe <[email protected]>
NVIDIA · Dec 18, 2024 · 85e84eb · 85e84eb
1 parent 00580a0
commit 85e84eb
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuPartitioning.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuPartitioning.scala
@@ -260,11 +260,6 @@ trait GpuPartitioning extends Partitioning {
    * be called at the query planning stage. Therefore, this method is NOT thread safe.
    */
   def setupDebugMetrics(metrics: Map[String, GpuMetric]): Unit = {
-    // Check and set GpuPartitioning.CopyToHostTime
-    if (memCopyTime.isEmpty) {
-      metrics.get(GpuPartitioning.CopyToHostTime).foreach { metric =>
-        memCopyTime = Some(metric)
-      }
-    }
+    memCopyTime = metrics.get(GpuPartitioning.CopyToHostTime).getOrElse(NoopMetric)
   }
 }