p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!24142 [AutoParallel]fix_stage_broadcast_bug 

Merge pull request !24142 from lichen/fix_pipeline_stage_broadcast_bug
This commit is contained in:
i-robot 2021-09-26 09:32:53 +00:00 committed by Gitee
parent 6e38f98c4e d2fc8edfac
commit 3da1229a66
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/frontend/parallel/pipeline_transformer/pipeline_transformer.cc
View File

@ -248,6 +248,8 @@ void PipelineTransformer::BroadCastColoring() {
auto user_node = user_pair.first->cast<CNodePtr>();
auto user_stage_info = user_node->user_data<NodeStageInfo>();
if (user_stage_info == nullptr) {
user_node->set_user_data<NodeStageInfo>(std::make_shared<NodeStageInfo>(stage));
need_coloring = true;
continue;
}
auto user_node_stage = user_stage_info->stage();