Browse Source

!15856 fix tuple to maketuple bug

From: @Margaret_wangrui
Reviewed-by: @hwhewei,@zh_qh
Signed-off-by: @zh_qh
pull/15856/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
71ff08c3d6
1 changed files with 7 additions and 1 deletions
  1. +7
    -1
      mindspore/ccsrc/backend/optimizer/pass/convert_tuple_output_to_maketuple.cc

+ 7
- 1
mindspore/ccsrc/backend/optimizer/pass/convert_tuple_output_to_maketuple.cc View File

@@ -34,6 +34,11 @@ AnfNodePtr ConvertTupleInputToMakeTuple(const FuncGraphPtr &graph, const AnfNode
return tuple_anf;
}
auto kernel_graph = graph->cast<KernelGraphPtr>();
FuncGraphPtr anf_graph = tuple_anf->func_graph();
if (anf_graph != nullptr) {
kernel_graph = anf_graph->cast<KernelGraphPtr>();
}

if (kernel_graph->FindTupleParameterToMakeTupleMap(tuple_anf)) {
return kernel_graph->FindTupleParameterToMakeTupleMap(tuple_anf);
}
@@ -77,7 +82,8 @@ const AnfNodePtr ConvertTupleOutputToMaketuple::Process(const FuncGraphPtr &func
cnode_input_changed = true;
}
}
auto kernel_graph = func_graph->cast<KernelGraphPtr>();
FuncGraphPtr graph = node->func_graph();
auto kernel_graph = graph->cast<KernelGraphPtr>();
if (kernel_graph == nullptr || !cnode_input_changed) {
return nullptr;
}


Loading…
Cancel
Save