From 399d72874be3379f9b7d707ef1b1cb8b1ef2a14c Mon Sep 17 00:00:00 2001 From: limingqi107 Date: Tue, 21 Apr 2020 10:37:08 +0800 Subject: [PATCH] fix visit kernel missing the return_types --- mindspore/ccsrc/session/anf_runtime_algorithm.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mindspore/ccsrc/session/anf_runtime_algorithm.cc b/mindspore/ccsrc/session/anf_runtime_algorithm.cc index 2591f763c5..525ff44dd8 100644 --- a/mindspore/ccsrc/session/anf_runtime_algorithm.cc +++ b/mindspore/ccsrc/session/anf_runtime_algorithm.cc @@ -111,12 +111,12 @@ KernelWithIndex AnfRuntimeAlgorithm::VisitKernelWithReturnType(const AnfNodePtr MS_EXCEPTION_IF_NULL(value_node); int item_idx = GetValue(value_node->value()); return VisitKernelWithReturnType(cnode->input(kRealInputNodeIndexInTupleGetItem), IntToSize(item_idx), - visit_nop_node); + visit_nop_node, return_types); } else if (IsPrimitive(input0, prim::kPrimDepend) || IsPrimitive(input0, prim::kPrimControlDepend)) { - return VisitKernelWithReturnType(cnode->input(kRealInputIndexInDepend), 0, visit_nop_node); + return VisitKernelWithReturnType(cnode->input(kRealInputIndexInDepend), 0, visit_nop_node, return_types); } else if (opt::IsNopNode(cnode) && visit_nop_node) { if (cnode->inputs().size() == 2) { - return VisitKernelWithReturnType(cnode->input(1), 0, visit_nop_node); + return VisitKernelWithReturnType(cnode->input(1), 0, visit_nop_node, return_types); } else { MS_LOG(EXCEPTION) << cnode->DebugString() << "Invalid nop node"; }