[VL][uniffle] Correct the write wait duration log (apache#6994)

sharkdtu · Oct 14, 2024 · 59ed58c · 59ed58c
1 parent 5919a70
commit 59ed58c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...elox/src/main/java/org/apache/spark/shuffle/writer/VeloxUniffleColumnarShuffleWriter.java b/...elox/src/main/java/org/apache/spark/shuffle/writer/VeloxUniffleColumnarShuffleWriter.java
@@ -226,11 +226,11 @@ public long spill(MemoryTarget self, Spiller.Phase phase, long size) {
     // clear all
     sendRestBlockAndWait();
     sendCommit();
-    long writeDurationMs = System.nanoTime() - pushMergedDataTime;
-    shuffleWriteMetrics.incWriteTime(writeDurationMs);
+    long writeDurationNanos = System.nanoTime() - pushMergedDataTime;
+    shuffleWriteMetrics.incWriteTime(writeDurationNanos);
     LOG.info(
         "Finish write shuffle with rest write {} ms",
-        TimeUnit.MILLISECONDS.toNanos(writeDurationMs));
+        TimeUnit.NANOSECONDS.toMillis(writeDurationNanos));
   }
 
   @Override