| @@ -34,6 +34,11 @@ AnfNodePtr ConvertTupleInputToMakeTuple(const FuncGraphPtr &graph, const AnfNode | |||||
| return tuple_anf; | return tuple_anf; | ||||
| } | } | ||||
| auto kernel_graph = graph->cast<KernelGraphPtr>(); | 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)) { | if (kernel_graph->FindTupleParameterToMakeTupleMap(tuple_anf)) { | ||||
| return 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; | 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) { | if (kernel_graph == nullptr || !cnode_input_changed) { | ||||
| return nullptr; | return nullptr; | ||||
| } | } | ||||