diff --git a/mindspeed_llm/core/parallel_state.py b/mindspeed_llm/core/parallel_state.py index 20fda45b3b242a97aa4b6b79ba530f33a43da3b8..7d15b7e4062211d5682db46fdeffff1396420232 100644 --- a/mindspeed_llm/core/parallel_state.py +++ b/mindspeed_llm/core/parallel_state.py @@ -319,6 +319,7 @@ def destroy_model_parallel_decorator(destroy_model_parallel): return wrapper +# test-0418 def get_pipeline_model_parallel_node_info(): return _PIPELINE_MODEL_PARALLEL_NODE_INFO