RemoveStageBarrier: partialForShuffle may cause infinite loop