|
|
|
@@ -670,6 +670,7 @@ Status HybridModelBuilder::UnfoldSubgraphs(ComputeGraphPtr &root_graph, ComputeG |
|
|
|
GE_CHK_GRAPH_STATUS_RET(merged_graph->AddSubgraph(remained_subgraph), |
|
|
|
"Failed to add subgraph [%s]", |
|
|
|
remained_subgraph->GetName().c_str()); |
|
|
|
remained_subgraph->SetParentGraph(merged_graph); |
|
|
|
} |
|
|
|
|
|
|
|
return SUCCESS; |
|
|
|
@@ -709,7 +710,7 @@ Status HybridModelBuilder::UnfoldSubgraph(ComputeGraphPtr &root_graph, |
|
|
|
for (size_t i = 0; i < sub_node->GetOpDesc()->GetSubgraphInstanceNames().size(); ++i) { |
|
|
|
auto sub_sub_graph = NodeUtils::GetSubgraph(*sub_node, i); |
|
|
|
GE_CHECK_NOTNULL(sub_sub_graph); |
|
|
|
sub_sub_graph->SetParentGraph(root_graph); |
|
|
|
sub_sub_graph->SetParentGraph(parent_graph); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
@@ -718,7 +719,7 @@ Status HybridModelBuilder::UnfoldSubgraph(ComputeGraphPtr &root_graph, |
|
|
|
sub_graph.GetName().c_str(), |
|
|
|
sub_node->GetName().c_str(), |
|
|
|
parent_graph->GetName().c_str()); |
|
|
|
sub_node->SetOwnerComputeGraph(root_graph); |
|
|
|
sub_node->SetOwnerComputeGraph(parent_graph); |
|
|
|
} |
|
|
|
|
|
|
|
GELOGD("[%s] Done merging subgraph. remove it from root graph.", sub_graph.GetName().c_str()); |
|
|
|
|