[Question] grouped_gemm.ops.permute/unpermute is broken?

[zzhhjjj]

I had an extremely large gradient norm during MoE training. I replaced them with the native PyTorch version, and the problem went away

[JayeeeZ]

Have you solved this problem? I found I can't use grouped_gemm.ops.permute …