[moco-tf] Use logo for Optimizer (#7120)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Wed, 4 Sep 2019 01:44:24 +0000 (10:44 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Wed, 4 Sep 2019 01:44:24 +0000 (10:44 +0900)
This will revise Optimizer to use logo Phase and ProgressReporter

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
compiler/moco-tf/src/Optimizer.cpp

index fee4d02..f33b410 100644 (file)
 #include "Optimizer.h"
 
 #include "Knob.h"
-#include "Phase.h"
+#include "ProgressReporter.h"
 #include "Transforms.h"
 
+#include <logo/Phase.h>
+
 #include <stdex/Memory.h>
 
 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<moco::tf::PhaseStrategy::Saturate> phase_runner{g};
+  ProgressReporter prog(g, logo::PhaseStrategy::Saturate);
+  logo::PhaseRunner<logo::PhaseStrategy::Saturate> phase_runner{g};
+  phase_runner.attach(&prog);
   phase_runner.run(phase);
 }