|
|
@ -3090,6 +3090,15 @@ Status GraphManager::OptimizeSubgraph(const GraphNodePtr &graph_node, ComputeGra
|
|
|
|
sub_graph->SetSessionID(session_id);
|
|
|
|
sub_graph->SetSessionID(session_id);
|
|
|
|
sub_graph->SetGraphID(graph_node->GetGraphId());
|
|
|
|
sub_graph->SetGraphID(graph_node->GetGraphId());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool off_superkernel = false;
|
|
|
|
|
|
|
|
if (AttrUtils::GetBool(compute_graph, ATTR_NAME_OFF_SUPERKERNEL_ATTR, off_superkernel)) {
|
|
|
|
|
|
|
|
GELOGI("Compute graph %s get superkernel flag %d.", compute_graph->GetName().c_str(), off_superkernel);
|
|
|
|
|
|
|
|
if (!AttrUtils::SetBool(merged_compute_graph, ATTR_NAME_OFF_SUPERKERNEL_ATTR, off_superkernel)) {
|
|
|
|
|
|
|
|
GELOGE(FAILED, "Compute graph %s set superkernel flag %d failed", merged_compute_graph->GetName().c_str(),
|
|
|
|
|
|
|
|
off_superkernel);
|
|
|
|
|
|
|
|
return FAILED;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
GE_TIMESTAMP_EVENT_END(MergeSubgraph, "OptimizeSubgraph::MergeSubGraph");
|
|
|
|
GE_TIMESTAMP_EVENT_END(MergeSubgraph, "OptimizeSubgraph::MergeSubGraph");
|
|
|
|
GE_DUMP(merged_compute_graph, "mergedComputeGraph");
|
|
|
|
GE_DUMP(merged_compute_graph, "mergedComputeGraph");
|
|
|
|
compute_graph = merged_compute_graph;
|
|
|
|
compute_graph = merged_compute_graph;
|
|
|
|