[spmd_types] fix MoE routing bias TP reduce (#4154)