From b01018ce89e5a1be74c2f171cda0a0e834bb75b7 Mon Sep 17 00:00:00 2001 From: zhangxiaokun Date: Tue, 23 Mar 2021 10:01:37 +0800 Subject: [PATCH] Add UT delete_need_infer_again --- .../ge/graph/passes/infershape_pass_unittest.cc | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/tests/ut/ge/graph/passes/infershape_pass_unittest.cc b/tests/ut/ge/graph/passes/infershape_pass_unittest.cc index 8fa5b34e..a7628b2e 100644 --- a/tests/ut/ge/graph/passes/infershape_pass_unittest.cc +++ b/tests/ut/ge/graph/passes/infershape_pass_unittest.cc @@ -26,12 +26,9 @@ #include "graph/operator_factory.h" #include "graph/operator_reg.h" #include "graph_builder_utils.h" -#undef protected -#undef private using namespace std; using namespace testing; -using namespace ge; namespace ge { class UtestGraphInfershapePass : public testing::Test { protected: @@ -52,4 +49,17 @@ TEST_F(UtestGraphInfershapePass, infershape_pass_failed) { InferShapePass infershape_pass; EXPECT_EQ(infershape_pass.Run(addn_node), GE_GRAPH_INFERSHAPE_FAILED); } + +TEST_F(UtestGraphInfershapePass, delete_need_infer_again) { + auto graph = std::make_shared("test"); + + auto no_op_desc = std::make_shared("No", "NoOp"); + auto no_op_node = graph->AddNode(no_op_desc); + AttrUtils::SetBool(no_op_desc, "_need_infer_again", false); + + InferShapePass infershape_pass; + infershape_pass.options_[kOptimizeAfterSubGraph] = "yes"; + EXPECT_EQ(infershape_pass.Run(no_op_node), SUCCESS); +} + } // namespace ge