Clamp capacity to num_tokens in MoE gating drop branches (complete #5353) (#8155)