| @@ -264,7 +264,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) { | |||||
| OpDescPtr sub_sub_graph_while_cond_data_op_desc = CreateOpDesc("cond_data", DATA); | OpDescPtr sub_sub_graph_while_cond_data_op_desc = CreateOpDesc("cond_data", DATA); | ||||
| NodePtr sub_sub_graph_while_cond_data_node = sub_sub_graph1->AddNode(sub_sub_graph_while_cond_data_op_desc); | NodePtr sub_sub_graph_while_cond_data_node = sub_sub_graph1->AddNode(sub_sub_graph_while_cond_data_op_desc); | ||||
| ComputeGraphPtr sub_sub_graph2 = std::make_shared<ComputeGraph>("while body"); | |||||
| ComputeGraphPtr sub_sub_graph2 = std::make_shared<ComputeGraph>("while_body"); | |||||
| /*OpDescPtr sub_sub_graph_while_body_const_op_desc = CreateOpDesc("body_const", CONSTANT); | /*OpDescPtr sub_sub_graph_while_body_const_op_desc = CreateOpDesc("body_const", CONSTANT); | ||||
| NodePtr sub_sub_graph_while_body_const_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_const_op_desc);*/ | NodePtr sub_sub_graph_while_body_const_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_const_op_desc);*/ | ||||
| OpDescPtr sub_sub_graph_while_body_data_op_desc = CreateOpDesc("body_data", DATA); | OpDescPtr sub_sub_graph_while_body_data_op_desc = CreateOpDesc("body_data", DATA); | ||||
| @@ -278,7 +278,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) { | |||||
| ComputeGraphPtr sub_graph = std::make_shared<ComputeGraph>("sub_graph"); | ComputeGraphPtr sub_graph = std::make_shared<ComputeGraph>("sub_graph"); | ||||
| OpDescPtr sub_graph_while_op_desc = CreateOpDesc("while", WHILE); | OpDescPtr sub_graph_while_op_desc = CreateOpDesc("while", WHILE); | ||||
| NodePtr sub_graph_while_node = sub_graph->AddNode(sub_graph_while_op_desc); | NodePtr sub_graph_while_node = sub_graph->AddNode(sub_graph_while_op_desc); | ||||
| sub_graph->SetGraphUnknownFlag(false); | |||||
| sub_graph->SetGraphUnknownFlag(true); | |||||
| sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_cond"); | sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_cond"); | ||||
| sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_body"); | sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_body"); | ||||
| sub_graph_while_node->GetOpDesc()->SetSubgraphInstanceName(0, "while_cond"); | sub_graph_while_node->GetOpDesc()->SetSubgraphInstanceName(0, "while_cond"); | ||||
| @@ -290,14 +290,16 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) { | |||||
| partitioned_call_node->GetOpDesc()->AddSubgraphName("sub_graph"); | partitioned_call_node->GetOpDesc()->AddSubgraphName("sub_graph"); | ||||
| partitioned_call_node->GetOpDesc()->SetSubgraphInstanceName(0, "sub_graph"); | partitioned_call_node->GetOpDesc()->SetSubgraphInstanceName(0, "sub_graph"); | ||||
| root_graph->AddSubGraph(sub_sub_graph1); | root_graph->AddSubGraph(sub_sub_graph1); | ||||
| root_graph->AddSubGraph(sub_sub_graph2); | root_graph->AddSubGraph(sub_sub_graph2); | ||||
| sub_sub_graph1->SetParentGraph(root_graph); | sub_sub_graph1->SetParentGraph(root_graph); | ||||
| sub_sub_graph2->SetParentGraph(root_graph); | sub_sub_graph2->SetParentGraph(root_graph); | ||||
| sub_sub_graph1->SetParentNode(sub_graph_while_node); | |||||
| sub_sub_graph2->SetParentNode(sub_graph_while_node); | |||||
| root_graph->AddSubGraph(sub_graph); | root_graph->AddSubGraph(sub_graph); | ||||
| sub_graph->SetParentNode(partitioned_call_node); | sub_graph->SetParentNode(partitioned_call_node); | ||||
| sub_graph->SetParentGraph(root_graph); | |||||
| GeRootModelPtr root_model = MakeShared<ge::GeRootModel>(root_graph); | GeRootModelPtr root_model = MakeShared<ge::GeRootModel>(root_graph); | ||||
| HybridModel hybrid_model(root_model); | HybridModel hybrid_model(root_model); | ||||