diff --git a/tests/ut/parser/testcase/graph_optimizer_testcase/graph_optimizer_unittest.cc b/tests/ut/parser/testcase/graph_optimizer_testcase/graph_optimizer_unittest.cc index 0627af8..94d201d 100644 --- a/tests/ut/parser/testcase/graph_optimizer_testcase/graph_optimizer_unittest.cc +++ b/tests/ut/parser/testcase/graph_optimizer_testcase/graph_optimizer_unittest.cc @@ -9,58 +9,54 @@ #define private public #include "tensorflow/graph_optimizer.h" #undef private -namespace ge -{ - class UtestGraphOptimizer : public testing::Test - { +namespace ge { + class UtestGraphOptimizer : public testing::Test { protected: void SetUp() {} void TearDown() {} }; - namespace - { - ComputeGraphPtr MakeGraph() { - ge::ut::GraphBuilder builder("graph"); - std::string name = "graph"; - std::string original_type; - original_type = "IteratorV2"; // - auto data1 = builder.AddNode(name + "_" + original_type, ge::parser::FRAMEWORKOP, 1, 1); - ge::AttrUtils::SetStr(data1->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_ORIGINAL_TYPE, original_type); - original_type = "IteratorGetNext"; - auto data2 = builder.AddNode(name + "_" + original_type + "2", ge::parser::FRAMEWORKOP, 1, 2); - ge::AttrUtils::SetStr(data2->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_ORIGINAL_TYPE, original_type); - string nodefStr; - AttrUtils::SetZeroCopyBytes( - data2->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_NODE_DEF, - Buffer::CopyFrom(reinterpret_cast(nodefStr.data()), nodefStr.length())); - original_type = "IteratorGetNext"; - auto data3 = builder.AddNode(name + "_" + original_type + "3", ge::parser::FRAMEWORKOP, 2, 1); - ge::AttrUtils::SetStr(data3->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_ORIGINAL_TYPE, original_type); - AttrUtils::SetZeroCopyBytes( - data3->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_NODE_DEF, - Buffer::CopyFrom(reinterpret_cast(nodefStr.data()), nodefStr.length())); +namespace { + ComputeGraphPtr MakeGraph() { + ge::ut::GraphBuilder builder("graph"); + std::string name = "graph"; + std::string original_type; + original_type = "IteratorV2"; // + auto data1 = builder.AddNode(name + "_" + original_type, ge::parser::FRAMEWORKOP, 1, 1); + ge::AttrUtils::SetStr(data1->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_ORIGINAL_TYPE, original_type); + original_type = "IteratorGetNext"; + auto data2 = builder.AddNode(name + "_" + original_type + "2", ge::parser::FRAMEWORKOP, 1, 2); + ge::AttrUtils::SetStr(data2->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_ORIGINAL_TYPE, original_type); + string nodefStr; + AttrUtils::SetZeroCopyBytes( + data2->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_NODE_DEF, + Buffer::CopyFrom(reinterpret_cast(nodefStr.data()), nodefStr.length())); + original_type = "IteratorGetNext"; + auto data3 = builder.AddNode(name + "_" + original_type + "3", ge::parser::FRAMEWORKOP, 2, 1); + ge::AttrUtils::SetStr(data3->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_ORIGINAL_TYPE, original_type); + AttrUtils::SetZeroCopyBytes( + data3->GetOpDesc(), ge::ATTR_NAME_FRAMEWORK_NODE_DEF, + Buffer::CopyFrom(reinterpret_cast(nodefStr.data()), nodefStr.length())); - builder.AddDataEdge(data1, 0, data2, 0); - builder.AddDataEdge(data2, 0, data3, 0); - builder.AddDataEdge(data2, 1, data3, 1); - return builder.GetGraph(); - } + builder.AddDataEdge(data1, 0, data2, 0); + builder.AddDataEdge(data2, 0, data3, 0); + builder.AddDataEdge(data2, 1, data3, 1); + return builder.GetGraph(); } +} +TEST_F(UtestGraphOptimizer, graph_optimizer) { + ge::ComputeGraphPtr graph = MakeGraph(); + ge::IteratorFusionPass iteratorFusionPass(domi::TENSORFLOW); + EXPECT_NE(iteratorFusionPass.Run(graph), ge::SUCCESS); +} +TEST_F(UtestGraphOptimizer, graph_optimizer_output) { + ge::ComputeGraphPtr graph = MakeGraph(); + domi::FrameworkType type = domi::TENSORFLOW; + ge::ParserGraphOptimizer parserGraphOptimizer(graph, type); - TEST_F(UtestGraphOptimizer, graph_optimizer) { - ge::ComputeGraphPtr graph = MakeGraph(); - ge::IteratorFusionPass iteratorFusionPass(domi::TENSORFLOW); - EXPECT_NE(iteratorFusionPass.Run(graph), ge::SUCCESS); - } - TEST_F(UtestGraphOptimizer, graph_optimizer_output) { - ge::ComputeGraphPtr graph = MakeGraph(); - domi::FrameworkType type = domi::TENSORFLOW; - ge::ParserGraphOptimizer parserGraphOptimizer(graph, type); - - vector input_anchors; - vector output_anchors; - ge::OpDescPtr fusion_op_desc; - EXPECT_NE(parserGraphOptimizer.RebuildInputAnchors(input_anchors, fusion_op_desc), ge::SUCCESS); - EXPECT_NE(parserGraphOptimizer.RebuildOutputAnchors(output_anchors, fusion_op_desc), ge::SUCCESS); - } + vector input_anchors; + vector output_anchors; + ge::OpDescPtr fusion_op_desc; + EXPECT_NE(parserGraphOptimizer.RebuildInputAnchors(input_anchors, fusion_op_desc), ge::SUCCESS); + EXPECT_NE(parserGraphOptimizer.RebuildOutputAnchors(output_anchors, fusion_op_desc), ge::SUCCESS); +} } \ No newline at end of file