Bugfix: support unknown control op subgraph

pull/1322/head
lichun 4 years ago
parent bb81990dd3
commit 1e9a360a8c

@ -266,11 +266,11 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
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(op_desc);
NodePtr sub_graph_while_node = sub_graph->AddNode(sub_graph_while_op_desc);
ComputeGraphPtr sub_sub_graph1 = std::make_shared<ComputeGraph>("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<ComputeGraph>("while body");
OpDescPtr sub_sub_graph_while_body_const_op_desc = CreateOpDesc("body_const", CONST);

Loading…
Cancel
Save