Browse Source

!5950 set depend abstract

Merge pull request !5950 from Margaret_wangrui/depend_abstract
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
47ca1e3128
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

@@ -857,6 +857,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