From 118b191f2df77cbaaa429ee3f893f6e1971ff632 Mon Sep 17 00:00:00 2001
From: wangxiaotian22 <wangxiaotian4@huawei.com>
Date: Wed, 27 Jan 2021 16:21:16 +0800
Subject: [PATCH] add single op compile in hccl_continuous_memcpy_pass

---
 ge/graph/passes/hccl_continuous_memcpy_pass.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/ge/graph/passes/hccl_continuous_memcpy_pass.cc b/ge/graph/passes/hccl_continuous_memcpy_pass.cc
index 7dd2fb06..cc928479 100644
--- 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.");