Browse Source

modified: tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc

tags/v1.2.0
zhaoxinxin 3 years ago
parent
commit
0646b4b6ef
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc

+ 6
- 0
tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc View File

@@ -74,6 +74,12 @@ TEST_F(UtestReplaceWithEmptyConstPass, replace_whith_empty_const_success) {
EXPECT_EQ(ret, SUCCESS); EXPECT_EQ(ret, SUCCESS);
EXPECT_EQ(graph->GetDirectNodesSize(),5); EXPECT_EQ(graph->GetDirectNodesSize(),5);


// run pass on const1, graph still has 5 nodes
auto const1 = graph->FindNode("const1");
Status ret = replace_with_empty_const_pass.Run(const1);
EXPECT_EQ(ret, SUCCESS);
EXPECT_EQ(graph->GetDirectNodesSize(),5);

auto cast1 = graph->FindNode("cast1"); auto cast1 = graph->FindNode("cast1");
ret = replace_with_empty_const_pass.Run(cast1); ret = replace_with_empty_const_pass.Run(cast1);
EXPECT_EQ(cast1->GetOutAllNodes().size(),0); EXPECT_EQ(cast1->GetOutAllNodes().size(),0);


Loading…
Cancel
Save