otterc commented on a change in pull request #30691: URL: https://github.com/apache/spark/pull/30691#discussion_r640374790
########## File path: core/src/main/scala/org/apache/spark/scheduler/DAGScheduler.scala ########## @@ -2136,9 +2137,24 @@ private[spark] class DAGScheduler( } } - private[scheduler] def handleShuffleMergeFinalized(stage: ShuffleMapStage): Unit = { - stage.shuffleDep.markShuffleMergeFinalized - processShuffleMapStageCompletion(stage) + private[scheduler] def handleRegisterMergeStatuses( + stage: ShuffleMapStage, + mergeStatuses: Seq[(Int, MergeStatus)]): Unit = { + // Register merge statuses if the stage is still running and shuffle merge is not finalized yet. + if (runningStages.contains(stage) && !stage.shuffleDep.shuffleMergeFinalized) { Review comment: In that case, currently there is no need to even post these messages to `eventProcessLoop` after `ShuffleMergeFinalized` message is posted. So should `RegisterMergeStatuses` be posted only when `!timedOut.get()`? Anyways, these messages are just getting ignored so why to even deserialize them. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org