Mark stageStarted for ShuffleReduceOp