RemoveStageBarrier: support partial map outputs