mridulm commented on code in PR #36601: URL: https://github.com/apache/spark/pull/36601#discussion_r877361726
########## core/src/main/scala/org/apache/spark/scheduler/DAGScheduler.scala: ########## @@ -1885,6 +1885,14 @@ private[spark] class DAGScheduler( mapOutputTracker. unregisterMergeResult(shuffleId, reduceId, bmAddress, Option(mapIndex)) } + } else { + // Unregister the merge result of <shuffleId, reduceId> if + // the FetchFailed event contains a mapIndex of -1, and is not a + // MetaDataFetchException which is signified by bmAddress being null + if (bmAddress != null && pushBasedShuffleEnabled) { Review Comment: I am trying to understand, can we have a case where `mapIndex == -1` and `bmAddress != null` for non-push based shuffle ? If no, we should add an assertion there (not drop the check entirely). Note, L1882 has the check because the earlier check is for `mapIndex != -1` - which can be satisfied for all shuffles -- 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. To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org 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