[sync] PR-40312: Update moe token unpermute DTensor support-pytorch-AtomGit

Star854

Star854

已合并

[sync] PR-40312: Update moe token unpermute DTensor support #42530

ascend-robot创建于 7月23日从ascend-ds-bot/pytorch： sync-pr40312-feature/moe_token_permute_memory_opt-to-v2.9.0 合入到 Ascend/pytorch： v2.9.0

[sync] PR-40312: Update moe token unpermute DTensor support #42530

已合并

ascend-robot创建于 7月23日

从ascend-ds-bot/pytorch sync-pr40312-feature/moe_token_permute_memory_opt-to-v2.9.0合入到Ascend/pytorch v2.9.0