From 7a0cfa5b230cafceccd8e941e7a410e94d4669de Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 4 Sep 2019 10:44:24 +0900 Subject: [PATCH] [moco-tf] Use logo for Optimizer (#7120) This will revise Optimizer to use logo Phase and ProgressReporter Signed-off-by: SaeHie Park --- compiler/moco-tf/src/Optimizer.cpp | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/compiler/moco-tf/src/Optimizer.cpp b/compiler/moco-tf/src/Optimizer.cpp index fee4d02..f33b410 100644 --- a/compiler/moco-tf/src/Optimizer.cpp +++ b/compiler/moco-tf/src/Optimizer.cpp @@ -17,9 +17,11 @@ #include "Optimizer.h" #include "Knob.h" -#include "Phase.h" +#include "ProgressReporter.h" #include "Transforms.h" +#include + #include namespace moco @@ -29,7 +31,7 @@ namespace tf void Optimizer::optimize(loco::Graph *g) const { - moco::tf::Phase phase; + logo::Phase phase; /* TRANSFORM DECLARATION BEGIN */ // Shape inference is required for ResolveRedundantReshape @@ -78,7 +80,9 @@ void Optimizer::optimize(loco::Graph *g) const } /* TRANSFORM DECLARATION END */ - moco::tf::PhaseRunner phase_runner{g}; + ProgressReporter prog(g, logo::PhaseStrategy::Saturate); + logo::PhaseRunner phase_runner{g}; + phase_runner.attach(&prog); phase_runner.run(phase); } -- 2.7.4