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