From d88ffe46e0f62d4e71cf12f36f5e16ba87914898 Mon Sep 17 00:00:00 2001 From: lichun Date: Sat, 27 Mar 2021 19:05:11 +0800 Subject: [PATCH] Bugfix: support unknown while op subgraph --- ge/hybrid/model/hybrid_model_builder.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ge/hybrid/model/hybrid_model_builder.cc b/ge/hybrid/model/hybrid_model_builder.cc index def32766..25751bc7 100755 --- a/ge/hybrid/model/hybrid_model_builder.cc +++ b/ge/hybrid/model/hybrid_model_builder.cc @@ -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());