From e6b2933150a14fecfd8c349514e0eb90ed8c18e6 Mon Sep 17 00:00:00 2001 From: mengyuanli Date: Mon, 6 Sep 2021 19:46:03 +0800 Subject: [PATCH] fix bug of control flow mindIR import --- .../lite/tools/converter/import/mindir_control_flow_adjust.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/tools/converter/import/mindir_control_flow_adjust.cc b/mindspore/lite/tools/converter/import/mindir_control_flow_adjust.cc index e70dc73516b..517db397cd3 100644 --- a/mindspore/lite/tools/converter/import/mindir_control_flow_adjust.cc +++ b/mindspore/lite/tools/converter/import/mindir_control_flow_adjust.cc @@ -68,7 +68,7 @@ std::vector MindIRControlFlowAdjust::GetFgOutput(const FuncGraphPtr auto output_cnode = output_node->cast(); if (output_cnode == nullptr) { MS_LOG(INFO) << "graph output is not cnode."; - return {}; + return {output_node}; } if (!IsMakeTuple(output_node)) { MS_LOG(INFO) << "graph is single output.";