diff --git a/compiler/luci/pass/src/QuantizeOnnxFakeQuantModelPass.cpp b/compiler/luci/pass/src/QuantizeOnnxFakeQuantModelPass.cpp index b95c7ea150c..c0826201a87 100644 --- a/compiler/luci/pass/src/QuantizeOnnxFakeQuantModelPass.cpp +++ b/compiler/luci/pass/src/QuantizeOnnxFakeQuantModelPass.cpp @@ -22,6 +22,7 @@ #include "QuantizeActivation.h" #include "QuantizationUtils.h" +#include #include #include #include @@ -89,8 +90,11 @@ bool QuantizeOnnxFakeQuantModelPass::run(loco::Graph *g) // Quantize nodes using their predecessors' qparams { - QuantizeWithPredecessorPass pass; - pass.run(g); + logo::Phase phase; + phase.emplace_back(std::make_unique()); + + logo::PhaseRunner phase_runner{g}; + phase_runner.run(phase); } // Backward propagation of activation qparam