Browse Source

Bugfix: support unknown while op subgraph

pull/1342/head
lichun 4 years ago
parent
commit
d88ffe46e0
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      ge/hybrid/model/hybrid_model_builder.cc

+ 3
- 2
ge/hybrid/model/hybrid_model_builder.cc View File

@@ -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());


Loading…
Cancel
Save