Browse Source

mark attr distance

pull/1573/head
lichun 4 years ago
parent
commit
82a85bc254
1 changed files with 7 additions and 7 deletions
  1. +7
    -7
      ge/graph/build/memory/graph_mem_assigner.cc

+ 7
- 7
ge/graph/build/memory/graph_mem_assigner.cc View File

@@ -1974,7 +1974,7 @@ void GraphMemoryAssigner::UpdatePrevNodeInputDesc(const NodePtr &prev_node,
auto input_desc = prev_node_op_desc->GetInputDesc(prev_node_input_index); auto input_desc = prev_node_op_desc->GetInputDesc(prev_node_input_index);
vector<int64_t> prev_next_distances; vector<int64_t> prev_next_distances;
if (!ge::AttrUtils::GetListInt(input_desc, ATTR_NAME_DATA_VISIT_DISTANCE, 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_op_desc->GetName().c_str(),
prev_node_input_index); prev_node_input_index);
continue; continue;
@@ -1987,19 +1987,19 @@ void GraphMemoryAssigner::UpdatePrevNodeInputDesc(const NodePtr &prev_node,
continue; continue;
} }
if (!ge::AttrUtils::SetListInt(input_desc, ATTR_NAME_DATA_VISIT_DISTANCE, prev_next_distances)) { 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_op_desc->GetName().c_str(),
prev_node_input_index); prev_node_input_index);
continue; continue;
} }


if (prev_node_op_desc->UpdateInputDesc(prev_node_input_index, input_desc) != GRAPH_SUCCESS) { 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_op_desc->GetName().c_str(),
prev_node_input_index); prev_node_input_index);
continue; 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, distance,
prev_node->GetName().c_str(), prev_node->GetName().c_str(),
prev_node_input_index); prev_node_input_index);
@@ -2016,18 +2016,18 @@ void GraphMemoryAssigner::UpdateCurNodeInputDesc(const NodePtr &cur_node,
vector<int64_t> prev_next_distances{distance, -1}; vector<int64_t> prev_next_distances{distance, -1};


if (!ge::AttrUtils::SetListInt(input_desc, ATTR_NAME_DATA_VISIT_DISTANCE, prev_next_distances)) { 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->GetOpDesc()->GetName().c_str(),
cur_node_input_index); cur_node_input_index);
return; return;
} }
if (cur_node->GetOpDesc()->UpdateInputDesc(cur_node_input_index, input_desc) != GRAPH_SUCCESS) { 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->GetOpDesc()->GetName().c_str(),
cur_node_input_index); cur_node_input_index);
return; 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, distance,
cur_node->GetName().c_str(), cur_node->GetName().c_str(),
cur_node_input_index); cur_node_input_index);


Loading…
Cancel
Save