Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
https://gitcode.com/cann/ops-transformer/blob/master/moe/moe_gating_top_k_softmax/README.md
yout的dtyep不对， yout的dtype应该同x， 是浮点类型，而非int32
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Document Link（文档链接）
https://gitcode.com/cann/ops-transformer/blob/master/moe/moe_gating_top_k_softmax/README.md
Issues Section（问题文档片段）
Existing Issues（存在的问题）
yout的dtyep不对， yout的dtype应该同x， 是浮点类型，而非int32