diff --git a/mindspore/ccsrc/backend/session/ascend_session.cc b/mindspore/ccsrc/backend/session/ascend_session.cc index c71534d54aa83cc6b7401e667009ee41df071cae..adb35257e1860c660b88290fab6a72b3a961ed36 100644 --- a/mindspore/ccsrc/backend/session/ascend_session.cc +++ b/mindspore/ccsrc/backend/session/ascend_session.cc @@ -857,6 +857,7 @@ void AscendSession::CreateMultiBranchOutput(NotNull graph, NotNu std::vector depend_inputs = { graph->NewValueNode(NewValueNode(std::make_shared(prim::kPrimDepend->name()))), output_param, node}; auto depend = graph->NewCNode(depend_inputs); + depend->set_abstract(output_param->abstract()); need_replace_list.emplace(node, depend); MS_LOG(INFO) << "Create parameter " << output_param->DebugString() << " for call node " << node->DebugString() << ", depend node is " << depend->DebugString();