From a0781b319505cf86e6bc7fff540a2444c6b88d16 Mon Sep 17 00:00:00 2001 From: qu_yueze Date: Thu, 31 Jul 2025 20:41:06 +0800 Subject: [PATCH] refactor about mc2 --- mindspeed_llm/features_manager/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspeed_llm/features_manager/__init__.py b/mindspeed_llm/features_manager/__init__.py index b90ec815c..73188c156 100644 --- a/mindspeed_llm/features_manager/__init__.py +++ b/mindspeed_llm/features_manager/__init__.py @@ -14,6 +14,7 @@ from mindspeed.features_manager.moe.fb_overlap import MoEFwdBwdOverlapFeature from mindspeed.features_manager.memory.reuse_fp32_param import ReuseFP32Param from mindspeed.features_manager.pipeline_parallel import NoopLayersFeature from mindspeed.features_manager.recompute.norm_function import RecomputeNormFeature +from mindspeed.features_manager.tensor_parallel.mc2 import MC2Feature from mindspeed_llm.features_manager.common.training import TrainingDefaultFeature @@ -96,6 +97,7 @@ def add_fusions_features(features_list: List[MindSpeedFeature]): def add_tensor_parallel_features(features_list: List[MindSpeedFeature]): features_list.extend([ AscendCocFeature(), + MC2Feature() ]) -- Gitee