Skip to content

Commit

Permalink
[BLAZE-736] Write time should increment for mapperEnd in CelebornPart…
Browse files Browse the repository at this point in the history
…itionWriter (#739)
  • Loading branch information
HYBG-1126 authored Jan 3, 2025
1 parent 7b15dd2 commit d0a6fea
Showing 1 changed file with 2 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,9 @@ class CelebornPartitionWriter(
override def flush(): Unit = {}

override def close(): Unit = {
val waitStartTime = System.nanoTime();
shuffleClient.mapperEnd(shuffleId, mapId, encodedAttemptId, numMappers)
metrics.incWriteTime(System.nanoTime() - waitStartTime);
}

override def getPartitionLengthMap: Array[Long] =
Expand Down

0 comments on commit d0a6fea

Please sign in to comment.