|
|
|
@@ -260,13 +260,10 @@ TEST_F(UtestGeHybrid, hybrid_model_executor) { |
|
|
|
TEST_F(UtestGeHybrid, unfold_subgraphs_success) { |
|
|
|
ComputeGraphPtr merged_graph = nullptr; |
|
|
|
|
|
|
|
ComputeGraphPtr root_graph = std::make_shared<ComputeGraph>("root_graph"); |
|
|
|
auto partitioned_call_op_desc = MakeShared<OpDesc>("partitioned_call", PARTITIONEDCALL); |
|
|
|
auto partitioned_call_node = root_graph->AddNode(partitioned_call_op_desc); |
|
|
|
|
|
|
|
ComputeGraphPtr sub_graph = std::make_shared<ComputeGraph>("sub_graph"); |
|
|
|
OpDescPtr sub_graph_while_op_desc = CreateOpDesc("while", WHILE); |
|
|
|
NodePtr sub_graph_while_node = sub_graph->AddNode(sub_graph_while_op_desc); |
|
|
|
sub_graph->SetGraphUnknownFlag(True); |
|
|
|
|
|
|
|
ComputeGraphPtr sub_sub_graph1 = std::make_shared<ComputeGraph>("while_cond"); |
|
|
|
OpDescPtr sub_sub_graph_while_cond_data_op_desc = CreateOpDesc("cond_data", DATA); |
|
|
|
@@ -282,6 +279,11 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) { |
|
|
|
sub_sub_graph_while_body_add_node->AddLinkFrom(sub_sub_graph_while_body_data_node); |
|
|
|
sub_sub_graph_while_body_add_node->AddLinkFrom(sub_sub_graph_while_body_const_node); |
|
|
|
|
|
|
|
ComputeGraphPtr root_graph = std::make_shared<ComputeGraph>("root_graph"); |
|
|
|
auto partitioned_call_op_desc = MakeShared<OpDesc>("partitioned_call", PARTITIONEDCALL); |
|
|
|
auto partitioned_call_node = root_graph->AddNode(partitioned_call_op_desc); |
|
|
|
partitioned_call_node->GetOpDesc()->AddSubgraphName("sub_graph"); |
|
|
|
|
|
|
|
sub_sub_graph1->AddSubGraph(sub_sub_graph1); |
|
|
|
sub_sub_graph2->AddSubGraph(sub_sub_graph2); |
|
|
|
|
|
|
|
|