diff --git a/mindspore/ccsrc/backend/session/ascend_session.cc b/mindspore/ccsrc/backend/session/ascend_session.cc index fd240d41cbb..63e27a07f90 100644 --- a/mindspore/ccsrc/backend/session/ascend_session.cc +++ b/mindspore/ccsrc/backend/session/ascend_session.cc @@ -856,6 +856,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();