Browse Source

Bugfix: support unknown control op subgraph

pull/1322/head
lichun 4 years ago
parent
commit
8a83fbf1a6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

+ 1
- 1
tests/ut/ge/hybrid/ge_hybrid_unittest.cc View File

@@ -273,7 +273,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
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", CONST);
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);
NodePtr sub_sub_graph_while_body_data_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_data_op_desc); NodePtr sub_sub_graph_while_body_data_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_data_op_desc);


Loading…
Cancel
Save