Browse Source

!2703 Fix bug of paramter is the second input of control depend

Merge pull request !2703 from chenfei_mindspore/parameter-should-not-act-as-the-second-input-of-control-depend-when-depend-mode-is-0
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
28b0f5286c
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

@@ -592,8 +592,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>{};
}

std::vector<AnfNodePtr> real_prior_nodes;


Loading…
Cancel
Save