Skip to content

N
ngraph
Commit 00c4830e authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Scott Cyphers
Browse files
Options

remove visualiazeTree passes in AlgebraicSimplification tests (#998)

parent ddf3f4f0
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 19 deletions
test/algebraic_simplification.cpp
View file @ 00c4830e
...@@ -43,7 +43,6 @@ ...@@ -43,7 +43,6 @@
#include "ngraph/pass/algebraic_simplification.hpp" #include "ngraph/pass/algebraic_simplification.hpp"
#include "ngraph/pass/graph_rewrite.hpp" #include "ngraph/pass/graph_rewrite.hpp"
#include "ngraph/pass/manager.hpp" #include "ngraph/pass/manager.hpp"
#include "ngraph/pass/visualize_tree.hpp"
#include "ngraph/pattern/matcher.hpp" #include "ngraph/pattern/matcher.hpp"
#include "ngraph/pattern/op/label.hpp" #include "ngraph/pattern/op/label.hpp"
#include "ngraph/pattern/op/skip.hpp" #include "ngraph/pattern/op/skip.hpp"
...@@ -63,9 +62,7 @@ TEST(algebraic_simplification, add_types_shapes) ...@@ -63,9 +62,7 @@ TEST(algebraic_simplification, add_types_shapes)
for (auto shape : shapes) for (auto shape : shapes)
{ {
pass::Manager pass_manager; pass::Manager pass_manager;
pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
pass_manager.register_pass<pass::AlgebraicSimplification>(); pass_manager.register_pass<pass::AlgebraicSimplification>();
pass_manager.register_pass<pass::VisualizeTree>("after.pdf");
auto a = make_shared<op::Parameter>(type, shape); auto a = make_shared<op::Parameter>(type, shape);
auto b = make_shared<op::Parameter>(type, shape); auto b = make_shared<op::Parameter>(type, shape);
...@@ -95,9 +92,7 @@ TEST(algebraic_simplification, add_broadcast) ...@@ -95,9 +92,7 @@ TEST(algebraic_simplification, add_broadcast)
{ {
Shape shape{2, 2}; Shape shape{2, 2};
pass::Manager pass_manager; pass::Manager pass_manager;
pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
pass_manager.register_pass<pass::AlgebraicSimplification>(); pass_manager.register_pass<pass::AlgebraicSimplification>();
pass_manager.register_pass<pass::VisualizeTree>("after.pdf");
auto a = make_shared<op::Parameter>(element::i32, shape); auto a = make_shared<op::Parameter>(element::i32, shape);
auto b = make_shared<op::Parameter>(element::i32, shape); auto b = make_shared<op::Parameter>(element::i32, shape);
...@@ -126,9 +121,7 @@ TEST(algebraic_simplification, multiply_broadcast) ...@@ -126,9 +121,7 @@ TEST(algebraic_simplification, multiply_broadcast)
{ {
Shape shape{2, 2}; Shape shape{2, 2};
pass::Manager pass_manager; pass::Manager pass_manager;
pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
pass_manager.register_pass<pass::AlgebraicSimplification>(); pass_manager.register_pass<pass::AlgebraicSimplification>();
pass_manager.register_pass<pass::VisualizeTree>("after.pdf");
auto a = make_shared<op::Parameter>(element::i32, shape); auto a = make_shared<op::Parameter>(element::i32, shape);
auto b = make_shared<op::Parameter>(element::i32, shape); auto b = make_shared<op::Parameter>(element::i32, shape);
...@@ -158,9 +151,7 @@ TEST(algebraic_simplification, zero_plus_zero_commutativity) ...@@ -158,9 +151,7 @@ TEST(algebraic_simplification, zero_plus_zero_commutativity)
Shape shape{}; Shape shape{};
auto type = element::f32; auto type = element::f32;
pass::Manager pass_manager; pass::Manager pass_manager;
pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
pass_manager.register_pass<pass::AlgebraicSimplification>(); pass_manager.register_pass<pass::AlgebraicSimplification>();
pass_manager.register_pass<pass::VisualizeTree>("after.pdf");
auto a = make_shared<op::Parameter>(type, shape); auto a = make_shared<op::Parameter>(type, shape);
auto b = make_shared<op::Parameter>(type, shape); auto b = make_shared<op::Parameter>(type, shape);
...@@ -184,9 +175,7 @@ TEST(algebraic_simplification, zero_multiply_zero_one) ...@@ -184,9 +175,7 @@ TEST(algebraic_simplification, zero_multiply_zero_one)
Shape shape{}; Shape shape{};
auto type = element::f32; auto type = element::f32;
pass::Manager pass_manager; pass::Manager pass_manager;
pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
pass_manager.register_pass<pass::AlgebraicSimplification>(); pass_manager.register_pass<pass::AlgebraicSimplification>();
pass_manager.register_pass<pass::VisualizeTree>("after.pdf");
auto a = make_shared<op::Parameter>(type, shape); auto a = make_shared<op::Parameter>(type, shape);
auto b = make_shared<op::Parameter>(type, shape); auto b = make_shared<op::Parameter>(type, shape);
...@@ -209,9 +198,7 @@ TEST(algebraic_simplification, add_negative_tests) ...@@ -209,9 +198,7 @@ TEST(algebraic_simplification, add_negative_tests)
Shape shape{}; Shape shape{};
auto type = element::f32; auto type = element::f32;
pass::Manager pass_manager; pass::Manager pass_manager;
pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
pass_manager.register_pass<pass::AlgebraicSimplification>(); pass_manager.register_pass<pass::AlgebraicSimplification>();
pass_manager.register_pass<pass::VisualizeTree>("after.pdf");
auto a = make_shared<op::Parameter>(type, shape); auto a = make_shared<op::Parameter>(type, shape);
auto b = make_shared<op::Parameter>(type, shape); auto b = make_shared<op::Parameter>(type, shape);
...@@ -240,9 +227,7 @@ TEST(algebraic_simplification, multiply_negative_tests) ...@@ -240,9 +227,7 @@ TEST(algebraic_simplification, multiply_negative_tests)
Shape shape{}; Shape shape{};
auto type = element::f32; auto type = element::f32;
pass::Manager pass_manager; pass::Manager pass_manager;
pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
pass_manager.register_pass<pass::AlgebraicSimplification>(); pass_manager.register_pass<pass::AlgebraicSimplification>();
pass_manager.register_pass<pass::VisualizeTree>("after.pdf");
auto a = make_shared<op::Parameter>(type, shape); auto a = make_shared<op::Parameter>(type, shape);
auto b = make_shared<op::Parameter>(type, shape); auto b = make_shared<op::Parameter>(type, shape);
...@@ -293,9 +278,7 @@ TEST(algebraic_simplification, multiply_prod_scalar_one) ...@@ -293,9 +278,7 @@ TEST(algebraic_simplification, multiply_prod_scalar_one)
auto prod_fconst1 = std::make_shared<op::Product>(broadcast, AxisSet{0, 1}); auto prod_fconst1 = std::make_shared<op::Product>(broadcast, AxisSet{0, 1});
pass::Manager pass_manager; pass::Manager pass_manager;
pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
pass_manager.register_pass<pass::AlgebraicSimplification>(); pass_manager.register_pass<pass::AlgebraicSimplification>();
pass_manager.register_pass<pass::VisualizeTree>("after.pdf");
auto f = std::make_shared<Function>(ngraph::NodeVector{prod_fconst1}, op::ParameterVector{}); auto f = std::make_shared<Function>(ngraph::NodeVector{prod_fconst1}, op::ParameterVector{});
pass_manager.run_passes(f); pass_manager.run_passes(f);
...@@ -329,9 +312,7 @@ TEST(algebraic_simplification, multiply_sum_scalar_one) ...@@ -329,9 +312,7 @@ TEST(algebraic_simplification, multiply_sum_scalar_one)
auto sum_fconst1 = std::make_shared<op::Sum>(broadcast, AxisSet{0, 1}); auto sum_fconst1 = std::make_shared<op::Sum>(broadcast, AxisSet{0, 1});
pass::Manager pass_manager; pass::Manager pass_manager;
pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
pass_manager.register_pass<pass::AlgebraicSimplification>(); pass_manager.register_pass<pass::AlgebraicSimplification>();
pass_manager.register_pass<pass::VisualizeTree>("after.pdf");
auto f = std::make_shared<Function>(ngraph::NodeVector{sum_fconst1}, op::ParameterVector{}); auto f = std::make_shared<Function>(ngraph::NodeVector{sum_fconst1}, op::ParameterVector{});
pass_manager.run_passes(f); pass_manager.run_passes(f);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment