#include "Canonicalizer.h"
#include "Knob.h"
-#include "Phase.h"
-#include "Transforms.h"
+#include "ProgressReporter.h"
#include "Transforms/ShapeInferencePass.h"
#include "Transforms/TypeInferencePass.h"
#include "Dialect/TFDialect.h"
#include "Dialect/TFNodes.h"
+#include <logo/Phase.h>
+
#include <stdex/Memory.h>
#include <cassert>
void Canonicalizer::canonicalize(loco::Graph *g) const
{
- moco::tf::Phase phase;
+ logo::Phase phase;
/* TRANSFORM DECLARATION BEGIN */
// Run shape and type inference at the top
phase.emplace_back(stdex::make_unique<TanhCanonicalizer>());
/* TRANSFORM DECLARATION END */
- moco::tf::PhaseRunner<moco::tf::PhaseStrategy::Restart> phase_runner{g};
+ ProgressReporter prog(g, logo::PhaseStrategy::Restart);
+ logo::PhaseRunner<logo::PhaseStrategy::Restart> phase_runner{g};
+ phase_runner.attach(&prog);
phase_runner.run(phase);
// Assert if graph has TF dialect nodes