Fix bus error

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
5 years ago · f5d726b0f9
parent b967997554
commit f5d726b0f9
1 changed files with 3 additions and 0 deletions
--- a/mindspore/ccsrc/session/ascend_session.cc
+++ b/mindspore/ccsrc/session/ascend_session.cc
@ -293,6 +293,9 @@ static void RecurseToUpdateCallRealInput(NotNull<KernelGraphPtr> graph,
 void InsertMakeTupleForEmptyGraph(NotNull<KernelGraphPtr> graph) {
  auto return_node = graph->get_return();
  MS_EXCEPTION_IF_NULL(return_node);
+  if (return_node->size() <= kReturnDataIndex) {
+    return;
+  }
  auto origin_output = return_node->input(kReturnDataIndex);
  MS_EXCEPTION_IF_NULL(origin_output);
  std::vector<AnfNodePtr> make_tuple_input{