Browse Source

add single op compile in hccl_continuous_memcpy_pass

tags/v1.2.0
wangxiaotian22 3 years ago
parent
commit
118b191f2d
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      ge/graph/passes/hccl_continuous_memcpy_pass.cc

+ 5
- 0
ge/graph/passes/hccl_continuous_memcpy_pass.cc View File

@@ -372,6 +372,11 @@ NodePtr HcclContinuousMemcpyPass::CreateAssignNode(const ComputeGraphPtr &graph,
}
GELOGI("Create Assign op:%s.", op_desc->GetName().c_str());

if (!AttrUtils::SetBool(op_desc, ATTR_NEED_COMPILE, true)) {
GELOGE(INTERNAL_ERROR, "Set ATTR_NEED_COMPILE Attr for node:%s fail.", op_desc->GetName().c_str());
return nullptr;
}

graphStatus ret = op_desc->AddInputDesc("ref", pre_op_desc->GetOutputDesc(out_data_anchor->GetIdx()));
if (ret != GRAPH_SUCCESS) {
GELOGE(INTERNAL_ERROR, "Create Assign op: add ref input desc fail.");


Loading…
Cancel
Save