Browse Source

!920 bug fix

From: @dimitri_rose
Reviewed-by: @wqtshg,@xchu42
Signed-off-by: @wqtshg
r1.1.1
mindspore-ci-bot Gitee 4 years ago
parent
commit
49aa7bb17b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/subgraph_pass.cc

+ 1
- 1
ge/graph/passes/subgraph_pass.cc View File

@@ -443,7 +443,7 @@ Status SubgraphPass::InsertMemcpyNode(const ComputeGraphPtr &graph, const OutDat
.AddOutput("y", in_node->GetOpDesc()->GetOutputDesc(0))
.Build();
(void)AttrUtils::SetBool(op_desc, ATTR_NO_NEED_CONSTANT_FOLDING, false);
if (graph->GetParentNode() != nullptr || !graph->GetParentNode()->GetOwnerComputeGraph()->GetGraphUnknownFlag()) {
if (graph->GetParentNode() != nullptr && !graph->GetGraphUnknownFlag()) {
(void)AttrUtils::SetBool(op_desc, "_zero_copy_identity_reserved", true);
}
if (GraphUtils::InsertNodeAfter(out_anchor, in_anchors, graph->AddNode(op_desc)) != GRAPH_SUCCESS) {


Loading…
Cancel
Save