add single op compile in hccl_continuous_memcpy_pass

4 years ago · 118b191f2d
parent 5cdca396f0
commit 118b191f2d
1 changed files with 5 additions and 0 deletions
--- a/ge/graph/passes/hccl_continuous_memcpy_pass.cc
+++ b/ge/graph/passes/hccl_continuous_memcpy_pass.cc
@ -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.");