if parameter is the second input of control depend and depend mode is 0,this control relation is invalid

5 years ago · 1ecaf912f9
parent fafdfa9a03
commit 1ecaf912f9
1 changed files with 2 additions and 2 deletions
--- a/mindspore/ccsrc/session/kernel_graph.cc
+++ b/mindspore/ccsrc/session/kernel_graph.cc
@ -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;