!1028 add single node compile in hccl_continuous_memcpy_pass

From: @wangxiaotian22
Reviewed-by: @ji_chen,@sheng-nan
Signed-off-by: @ji_chen
pull/1028/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit d8fdf1740b

@ -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