From 1e9a360a8cfb673d0b9dac9f69850da5cdf2f8e1 Mon Sep 17 00:00:00 2001 From: lichun Date: Tue, 23 Mar 2021 14:41:07 +0800 Subject: [PATCH] Bugfix: support unknown control op subgraph --- tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc index c13a9003..242ca91d 100644 --- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc +++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc @@ -266,11 +266,11 @@ 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(op_desc); + NodePtr sub_graph_while_node = sub_graph->AddNode(sub_graph_while_op_desc); ComputeGraphPtr sub_sub_graph1 = std::make_shared("while_cond"); - OpDescPtr sub_sub_graph_while_cond_const_op_desc = CreateOpDesc("cond_const", CONST); - NodePtr sub_sub_graph_while_cond_const_node = sub_sub_graph1->AddNode(sub_sub_graph_while_cond_const_op_desc); + 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); ComputeGraphPtr sub_sub_graph2 = std::make_shared("while body"); OpDescPtr sub_sub_graph_while_body_const_op_desc = CreateOpDesc("body_const", CONST);