Browse Source

!2709 [r0.5]Fix bug of paramter is the second input of control depend

Merge pull request !2709 from chenfei_mindspore/parameter-should-not-act-as-the-second-input-of-control-depend-when-depend-mode-is-0
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
bb3e05c317
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/session/kernel_graph.cc

+ 2
- 2
mindspore/ccsrc/session/kernel_graph.cc View File

@@ -548,8 +548,8 @@ void KernelGraph::UpdateControlDependRelations(const std::vector<AnfNodePtr> &de
if (prior_node->isa<Parameter>() && depend_mode == 1) {
prior_nodes = GetOutputNodes(prior_node);
}
if (depend_node->isa<Parameter>() && depend_mode == 1) {
depend_nodes = GetOutputNodes(depend_node);
if (depend_node->isa<Parameter>()) {
depend_nodes = depend_mode == 1 ? GetOutputNodes(depend_node) : std::vector<AnfNodePtr>();
}
for (auto &first_node : prior_nodes) {
if (AnfAlgo::CheckPrimitiveType(first_node, prim::kPrimControlDepend)) {


Loading…
Cancel
Save