Browse Source

!1028 add single node compile in hccl_continuous_memcpy_pass

From: @wangxiaotian22
Reviewed-by: @ji_chen,@sheng-nan
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
d8fdf1740b
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()); 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())); graphStatus ret = op_desc->AddInputDesc("ref", pre_op_desc->GetOutputDesc(out_data_anchor->GetIdx()));
if (ret != GRAPH_SUCCESS) { if (ret != GRAPH_SUCCESS) {
GELOGE(INTERNAL_ERROR, "Create Assign op: add ref input desc fail."); GELOGE(INTERNAL_ERROR, "Create Assign op: add ref input desc fail.");


Loading…
Cancel
Save