[circle-mpqsolver] Dump config (Samsung#12285)

This commit dumps final qconfig in PatternSolver if save_intermediate option is on. ONE-DCO-1.0-Signed-off-by: s.malakhov <[email protected]>
glistening · Dec 14, 2023 · f64b740 · f64b740
1 parent ab0cee9
commit f64b740
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/compiler/circle-mpqsolver/src/pattern/PatternSolver.cpp b/compiler/circle-mpqsolver/src/pattern/PatternSolver.cpp
@@ -19,6 +19,7 @@
 #include "PatternResolver.h"
 
 #include <iostream>
+#include <cmath>
 
 using namespace mpqsolver::pattern;
 
@@ -38,10 +39,21 @@ std::unique_ptr<luci::Module> PatternSolver::run(const std::string &module_path)
   auto module = readModule(module_path);
   assert(module != nullptr);
 
+  _quantizer->setHook(_hooks.get());
+  if (_hooks)
+  {
+    _hooks->onBeginSolver(module_path, NAN, NAN);
+  }
+
   resolvePatterns(module.get());
 
   auto layer_params = getFrozenParams();
 
+  if (_hooks)
+  {
+    _hooks->onEndSolver(layer_params, _quantizer->getContext().output_model_dtype, NAN);
+  }
+
   if (!_quantizer->quantize(module.get(), layer_params))
   {
     throw std::runtime_error("Failed to quantize model");