diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc index 88141d97..e8e8e196 100644 --- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc +++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc @@ -269,6 +269,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) { 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); NodePtr sub_sub_graph_while_body_data_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_data_op_desc); + sub_sub_graph2->SetGraphUnknownFlag(true); /*OpDescPtr sub_sub_graph_while_body_add_op_desc = CreateOpDesc("body_add", ADD); NodePtr sub_sub_graph_while_body_add_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_add_node); sub_sub_graph_while_body_add_node->AddLinkFrom(sub_sub_graph_while_body_data_node); @@ -277,7 +278,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) { ComputeGraphPtr sub_graph = std::make_shared("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); + sub_graph->SetGraphUnknownFlag(false); sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_cond"); sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_body"); sub_graph_while_node->GetOpDesc()->SetSubgraphInstanceName(0, "while_cond"); @@ -289,6 +290,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) { partitioned_call_node->GetOpDesc()->AddSubgraphName("sub_graph"); partitioned_call_node->GetOpDesc()->SetSubgraphInstanceName(0, "sub_graph"); + root_graph->AddSubGraph(sub_sub_graph1); root_graph->AddSubGraph(sub_sub_graph2); sub_sub_graph1->SetParentGraph(root_graph);