From 0ed60257fa97bd687e6457ba6df619302616395e Mon Sep 17 00:00:00 2001 From: kpy Date: Tue, 1 Sep 2020 14:35:59 +0800 Subject: [PATCH] fix tuple return bug --- mindspore/ccsrc/pipeline/pynative/pynative_execute.cc | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc index ee6d2742c..ded685ed2 100644 --- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc +++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc @@ -1151,10 +1151,13 @@ void PynativeExecutor::EndGraphInner(const py::object &cell, const py::object &o auto cnode = curr_g_->NewCNode(args); for (int i = 0; i < tuple_size; i++) { args.push_back(GetInput(tuple[i], false)); + } + cnode->set_inputs(args); + + for (int i = 0; i < tuple_size; i++) { set_obj_node_map(curr_g_, GetId(tuple[i]), cnode, i); SetTupleOutput(tuple[i], cnode, std::vector{i}); } - cnode->set_inputs(args); set_obj_node_map(curr_g_, out_id, cnode); } else { MS_LOG(DEBUG) << "Set ValueNode as output for graph, out id: " << out_id; -- GitLab