Browse Source

set depend abstract

tags/v1.0.0
Margaret_wangrui 5 years ago
parent
commit
4ebe9f4dab
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/backend/session/ascend_session.cc

+ 1
- 0
mindspore/ccsrc/backend/session/ascend_session.cc View File

@@ -856,6 +856,7 @@ void AscendSession::CreateMultiBranchOutput(NotNull<KernelGraphPtr> graph, NotNu
std::vector<AnfNodePtr> depend_inputs = {
graph->NewValueNode(NewValueNode(std::make_shared<Primitive>(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();


Loading…
Cancel
Save