mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!35805 [AutoParallel]Fix pipeline parallel cycle issue master 

Merge pull request !35805 from lichen/fix_pipeline_parallel_cycle_for_master
This commit is contained in:
i-robot 2022-06-14 18:22:49 +00:00 committed by Gitee
parent 5ebbae9659 b3e6985ed8
commit b644d3a485
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/frontend/parallel/graph_util/pipeline_split_utils.cc
View File

@ -454,12 +454,6 @@ void ReorderForParams(const std::vector<AnfNodePtr> &backward_params, const std:
InsertDepend(prior_node, post_node, manager, root);
}
if (!backward_params.empty()) {
if (!allreduce_params.empty()) {
for (auto &node : allreduce_params) {
auto post_node1 = backward_params_pair.first[0];
InsertDepend(node, post_node1, manager, root);
}
}
auto prior_node2 = backward_end.back();
auto post_node2 = backward_params[0];
InsertDepend(prior_node2, post_node2, manager, root);