From f8e0e3944c18d67d7dd188a9f347a1804f3d322e Mon Sep 17 00:00:00 2001 From: wangjialin Date: Mon, 9 Jun 2025 19:53:30 +0800 Subject: [PATCH] =?UTF-8?q?bugfix,=E4=BF=AE=E5=A4=8Dpatch=E7=9A=84?= =?UTF-8?q?=E6=96=B9=E6=B3=95=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- mindspeed_llm/mindspore/mindspore_adaptor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspeed_llm/mindspore/mindspore_adaptor.py b/mindspeed_llm/mindspore/mindspore_adaptor.py index 467a40c5c..27066e0e4 100644 --- a/mindspeed_llm/mindspore/mindspore_adaptor.py +++ b/mindspeed_llm/mindspore/mindspore_adaptor.py @@ -323,8 +323,8 @@ class MindSporeAdaptation(MegatronAdaptationABC): zerc_alltoall_token_unperm2) from mindspeed.mindspore.core.pipeline_parallel.fb_overlap.modules.token_dispatcher import PackProb, UnpackProb - MindSporeAdaptation.register('mindspeed.mindspore.core.pipeline_parallel.fb_overlap.modules.token_dispatcher.PackProb', PackProb) - MindSporeAdaptation.register('mindspeed.mindspore.core.pipeline_parallel.fb_overlap.modules.token_dispatcher.UnpackProb', UnpackProb) + MindSporeAdaptation.register('mindspeed.core.pipeline_parallel.fb_overlap.modules.token_dispatcher.PackProb', PackProb) + MindSporeAdaptation.register('mindspeed.core.pipeline_parallel.fb_overlap.modules.token_dispatcher.UnpackProb', UnpackProb) if args.gradient_accumulation_fusion: from mindspeed.mindspore.ops.npu_matmul_add import npu_matmul_add_fp32 -- Gitee