|
@@ -382,11 +382,18 @@ graphStatus ge::GraphPartitioner::AddPlaceHolderEndInSrcDstGraph(const AnchorPtr |
|
|
GELOGW("SetInt anchorIndex failed");) |
|
|
GELOGW("SetInt anchorIndex failed");) |
|
|
GE_IF_BOOL_EXEC(!pld_op_desc->SetExtAttr("parentNode", src_node), |
|
|
GE_IF_BOOL_EXEC(!pld_op_desc->SetExtAttr("parentNode", src_node), |
|
|
GELOGW("SetPldExtAttr parentNode failed");) |
|
|
GELOGW("SetPldExtAttr parentNode failed");) |
|
|
|
|
|
|
|
|
OpDescPtr src_node_op_desc = src_node->GetOpDesc(); |
|
|
|
|
|
GE_CHECK_NOTNULL(src_node_op_desc); |
|
|
|
|
|
GE_IF_BOOL_EXEC(!AttrUtils::SetStr(pld_op_desc, ATTR_NAME_PLD_FRONT_NODE_ENGINE_NAME, |
|
|
GE_IF_BOOL_EXEC(!AttrUtils::SetStr(pld_op_desc, ATTR_NAME_PLD_FRONT_NODE_ENGINE_NAME, |
|
|
src_node_op_desc->GetOpEngineName()), GELOGW("SetStr frontNodeEngineName failed");) |
|
|
|
|
|
|
|
|
src_node_opdesc->GetOpEngineName()), GELOGW("SetStr frontNodeEngineName failed");) |
|
|
|
|
|
std::string l2_info_attr; |
|
|
|
|
|
if (AttrUtils::GetStr(src_node_opdesc, "_task_L2FusionInfo", l2_info_attr)) { |
|
|
|
|
|
GE_IF_BOOL_EXEC(!AttrUtils::SetStr(pld_op_desc, "_task_L2FusionInfo", l2_info_attr), |
|
|
|
|
|
GELOGW("SetStr l2_info_attr failed");) |
|
|
|
|
|
} |
|
|
|
|
|
int64_t anchor_index_for_lxfusion; |
|
|
|
|
|
if (AttrUtils::GetInt(src_node_opdesc, "_data_anchor_index_for_lxfusion", anchor_index_for_lxfusion)) { |
|
|
|
|
|
GE_IF_BOOL_EXEC(!AttrUtils::SetInt(pld_op_desc, "_data_anchor_index_for_lxfusion", anchor_index_for_lxfusion), |
|
|
|
|
|
GELOGW("SetInt anchor_index_for_lxfusion failed");) |
|
|
|
|
|
} |
|
|
// do not care over flow |
|
|
// do not care over flow |
|
|
graph_info_.num_of_pld_end_++; |
|
|
graph_info_.num_of_pld_end_++; |
|
|
// replace output_desc of pld with input node's output desc |
|
|
// replace output_desc of pld with input node's output desc |
|
|