From 929b690f4e1878d2271646f810d53a34c1a329a9 Mon Sep 17 00:00:00 2001 From: xinyuan Date: Thu, 19 Jun 2025 00:11:27 +0800 Subject: [PATCH] add_ms_dualpipe_forward_step_warp --- mindspeed_llm/mindspore/mindspore_adaptor.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mindspeed_llm/mindspore/mindspore_adaptor.py b/mindspeed_llm/mindspore/mindspore_adaptor.py index a140ee171..8b87a91fb 100644 --- a/mindspeed_llm/mindspore/mindspore_adaptor.py +++ b/mindspeed_llm/mindspore/mindspore_adaptor.py @@ -281,6 +281,9 @@ class MindSporeAdaptation(MegatronAdaptationABC): MindSporeAdaptation.register('mindspeed.core.pipeline_parallel.fb_overlap.modules.token_dispatcher.overlap_stream', overlap_stream) + from mindspeed_llm.core.pipeline_parallel.dualpipe.adaptor import dualpipe_forward_step_wrapper + MindSporeAdaptation.register('mindspeed.mindspore.core.pipeline_parallel.dualpipev.dualpipev_schedules.forward_step_with_model_graph', + dualpipe_forward_step_wrapper) if args.swap_optimizer: -- Gitee