From 4ebe9f4dab44194f7e8a34592f0330ca2115f906 Mon Sep 17 00:00:00 2001 From: Margaret_wangrui Date: Wed, 9 Sep 2020 12:48:34 +0800 Subject: [PATCH] set depend abstract --- mindspore/ccsrc/backend/session/ascend_session.cc | 1 + 1 file changed, 1 insertion(+) 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();