|
|
@@ -1974,7 +1974,7 @@ void GraphMemoryAssigner::UpdatePrevNodeInputDesc(const NodePtr &prev_node, |
|
|
|
auto input_desc = prev_node_op_desc->GetInputDesc(prev_node_input_index); |
|
|
|
vector<int64_t> prev_next_distances; |
|
|
|
if (!ge::AttrUtils::GetListInt(input_desc, ATTR_NAME_DATA_VISIT_DISTANCE, prev_next_distances)) { |
|
|
|
GELOGW("Get [%s] input [%lld] ATTR_NAME_DATA_VISIT_DISTANCE failed", |
|
|
|
GELOGW("Get [%s] input [%ld] ATTR_NAME_DATA_VISIT_DISTANCE failed", |
|
|
|
prev_node_op_desc->GetName().c_str(), |
|
|
|
prev_node_input_index); |
|
|
|
continue; |
|
|
@@ -1987,19 +1987,19 @@ void GraphMemoryAssigner::UpdatePrevNodeInputDesc(const NodePtr &prev_node, |
|
|
|
continue; |
|
|
|
} |
|
|
|
if (!ge::AttrUtils::SetListInt(input_desc, ATTR_NAME_DATA_VISIT_DISTANCE, prev_next_distances)) { |
|
|
|
GELOGW("Set [%s] input [%lld] ATTR_NAME_DATA_VISIT_DISTANCE failed.", |
|
|
|
GELOGW("Set [%s] input [%ld] ATTR_NAME_DATA_VISIT_DISTANCE failed.", |
|
|
|
prev_node_op_desc->GetName().c_str(), |
|
|
|
prev_node_input_index); |
|
|
|
continue; |
|
|
|
} |
|
|
|
|
|
|
|
if (prev_node_op_desc->UpdateInputDesc(prev_node_input_index, input_desc) != GRAPH_SUCCESS) { |
|
|
|
GELOGW("Update [%s] input [%lld] ATTR_NAME_DATA_VISIT_DISTANCE failed.", |
|
|
|
GELOGW("Update [%s] input [%ld] ATTR_NAME_DATA_VISIT_DISTANCE failed.", |
|
|
|
prev_node_op_desc->GetName().c_str(), |
|
|
|
prev_node_input_index); |
|
|
|
continue; |
|
|
|
} |
|
|
|
GELOGD("Set the next distance[%lld] to node[%s], input index[%lld]", |
|
|
|
GELOGD("Set the next distance[%ld] to node[%s], input index[%ld]", |
|
|
|
distance, |
|
|
|
prev_node->GetName().c_str(), |
|
|
|
prev_node_input_index); |
|
|
@@ -2016,18 +2016,18 @@ void GraphMemoryAssigner::UpdateCurNodeInputDesc(const NodePtr &cur_node, |
|
|
|
vector<int64_t> prev_next_distances{distance, -1}; |
|
|
|
|
|
|
|
if (!ge::AttrUtils::SetListInt(input_desc, ATTR_NAME_DATA_VISIT_DISTANCE, prev_next_distances)) { |
|
|
|
GELOGW("Set [%s] input[%lld] ATTR_NAME_DATA_VISIT_DISTANCE failed.", |
|
|
|
GELOGW("Set [%s] input[%ld] ATTR_NAME_DATA_VISIT_DISTANCE failed.", |
|
|
|
cur_node->GetOpDesc()->GetName().c_str(), |
|
|
|
cur_node_input_index); |
|
|
|
return; |
|
|
|
} |
|
|
|
if (cur_node->GetOpDesc()->UpdateInputDesc(cur_node_input_index, input_desc) != GRAPH_SUCCESS) { |
|
|
|
GELOGW("Update [%s] input[%lld] ATTR_NAME_DATA_VISIT_DISTANCE failed.", |
|
|
|
GELOGW("Update [%s] input[%ld] ATTR_NAME_DATA_VISIT_DISTANCE failed.", |
|
|
|
cur_node->GetOpDesc()->GetName().c_str(), |
|
|
|
cur_node_input_index); |
|
|
|
return; |
|
|
|
} |
|
|
|
GELOGD("Set the prev distance[%lld] to node[%s], input index[%lld]", |
|
|
|
GELOGD("Set the prev distance[%ld] to node[%s], input index[%ld]", |
|
|
|
distance, |
|
|
|
cur_node->GetName().c_str(), |
|
|
|
cur_node_input_index); |
|
|
|