Browse Source

Bugfix:fix error of get null subgraph

tags/v1.3.0
lichun 3 years ago
parent
commit
0f5de61a80
1 changed files with 7 additions and 7 deletions
  1. +7
    -7
      ge/hybrid/model/hybrid_model_builder.cc

+ 7
- 7
ge/hybrid/model/hybrid_model_builder.cc View File

@@ -270,7 +270,7 @@ Status HybridModelBuilder::ParseForceInfershapeNodes(const NodePtr &node, NodeIt
GE_CHECK_NOTNULL(op_desc);
// not care result, if no this attr, stand for the op does not need force infershape
(void)AttrUtils::GetBool(op_desc, kForceInfershape, node_item.is_need_force_infershape);
GELOGD("node [%s] is need do infershape , flag is %d",
GELOGD("node [%s] is need do infershape, flag is %d",
op_desc->GetName().c_str(),
node_item.is_need_force_infershape);
return SUCCESS;
@@ -537,7 +537,7 @@ Status HybridModelBuilder::MergeNetOutputNode(ComputeGraph &graph) {
const auto &parent_node = graph.GetParentNode();
const NodePtr &net_output_node = graph.FindFirstNodeMatchType(NETOUTPUT);
if (net_output_node == nullptr) {
GELOGD("Graph has no netoutput no need to merge.");
GELOGD("Graph has no netoutput no need to merge");
return SUCCESS;
}
const auto &net_output_desc = net_output_node->GetOpDesc();
@@ -670,7 +670,7 @@ Status HybridModelBuilder::UnfoldSubgraph(ComputeGraphPtr &root_graph,
GE_CHK_STATUS_RET(MergeNetOutputNode(sub_graph),
"[%s] Failed to merge net output nodes for subgraph",
sub_graph.GetName().c_str());
GELOGD("[%s] Done merging subgraph inputs and outputs successfully.", sub_graph.GetName().c_str());
GELOGD("[%s] Done merging subgraph inputs and outputs successfully", sub_graph.GetName().c_str());

for (auto &sub_node : sub_graph.GetDirectNode()) {
auto sub_op_type = sub_node->GetType();
@@ -703,7 +703,7 @@ Status HybridModelBuilder::UnfoldSubgraph(ComputeGraphPtr &root_graph,
sub_node->SetOwnerComputeGraph(parent_graph);
}

GELOGD("[%s] Done merging subgraph. remove it from root graph.", sub_graph.GetName().c_str());
GELOGD("[%s] Done merging subgraph. remove it from root graph", sub_graph.GetName().c_str());
root_graph->RemoveSubgraph(sub_graph.GetName());
return SUCCESS;
}
@@ -1174,11 +1174,11 @@ Status HybridModelBuilder::IndexTaskDefs(const ComputeGraphPtr &sub_graph, const
GELOGD("Skip task type: %d", static_cast<int>(task_type));
continue;
}
GELOGD("op_index = %u, task_type = %d.", op_index, task_type);
GELOGD("op_index = %u, task_type = %d", op_index, task_type);

auto iter = node_map.find(op_index);
if (iter == node_map.end()) {
GELOGE(INTERNAL_ERROR, "Failed to get node by op_index = %u.", op_index);
GELOGE(INTERNAL_ERROR, "Failed to get node by op_index = %u", op_index);
return INTERNAL_ERROR;
}

@@ -1187,7 +1187,7 @@ Status HybridModelBuilder::IndexTaskDefs(const ComputeGraphPtr &sub_graph, const
ge_model->GetTBEKernelStore().LoadTBEKernelBinToOpDesc(node->GetOpDesc());
}

GELOGD("Task loaded for node: %s, task type = %d, op_index = %u.", node->GetName().c_str(), task_type, op_index);
GELOGD("Task loaded for node: %s, task type = %d, op_index = %u", node->GetName().c_str(), task_type, op_index);
hybrid_model_.task_defs_[node].emplace_back(task_def);
}



Loading…
Cancel
Save