mpqsolver::core::DumpingHooks Class Reference

DumpingHooks is intended to save intermediate results. More...

#include <DumpingHooks.h>

Collaboration diagram for mpqsolver::core::DumpingHooks:

Public Member Functions
	DumpingHooks (const std::string &save_path, const Quantizer::Context &ctx)
	DumpingHooks constructor.
virtual void	onQuantized (luci::Module *module) const override
	called on successfull quantization
virtual void	onBeginSolver (const std::string &model_path, float q8error, float q16error) override
	called on the start of mpq search
virtual void	onBeginIteration () override
	called on the start of current iteration
virtual void	onEndIteration (const LayerParams &layers, const std::string &def_dtype, float error) override
	called at the end of current iteration
virtual void	onEndSolver (const LayerParams &layers, const std::string &def_dtype, float qerror) override
	called at the end of mpq search

Detailed Description

DumpingHooks is intended to save intermediate results.

Definition at line 36 of file DumpingHooks.h.

Constructor & Destructor Documentation

DumpingHooks()

DumpingHooks::DumpingHooks	(	const std::string &	save_path,
		const Quantizer::Context &	ctx
	)

DumpingHooks constructor.

Parameters

save_path

directory where all intermediate data will be saved

Definition at line 22 of file DumpingHooks.cpp.

  : _save_path(save_path), _dumper(_save_path), _ctx(ctx)
{
}

Member Function Documentation

onBeginIteration()

void DumpingHooks::onBeginIteration ( )

overridevirtual

called on the start of current iteration

Implements mpqsolver::core::SolverHooks.

Definition at line 45 of file DumpingHooks.cpp.

{
  _in_iterations = true;
  _num_of_iterations += 1;
}

onBeginSolver()

void DumpingHooks::onBeginSolver ( const std::string &  model_path, float  q8error, float  q16error  )

overridevirtual

called on the start of mpq search

Implements mpqsolver::core::SolverHooks.

Definition at line 27 of file DumpingHooks.cpp.

{
  _model_path = model_path;
  _dumper.setModelPath(_model_path);
  if (!std::isnan(q8error) || !std::isnan(q16error))
  {
    _dumper.prepareForErrorDumping();
  }
  if (!std::isnan(q8error))
  {
    _dumper.dumpQ8Error(q8error);
  }
  if (!std::isnan(q16error))
  {
    _dumper.dumpQ16Error(q16error);
  }
}

References mpqsolver::core::Dumper::dumpQ16Error(), mpqsolver::core::Dumper::dumpQ8Error(), mpqsolver::core::Dumper::prepareForErrorDumping(), and mpqsolver::core::Dumper::setModelPath().

onEndIteration()

void DumpingHooks::onEndIteration ( const LayerParams &  layers, const std::string &  def_dtype, float  error  )

overridevirtual

called at the end of current iteration

Implements mpqsolver::core::SolverHooks.

Definition at line 51 of file DumpingHooks.cpp.

{
  _dumper.dumpMPQConfiguration(layers, def_type, _ctx.granularity, _num_of_iterations);
  _dumper.dumpMPQError(error, _num_of_iterations);
  _in_iterations = false;
}

References mpqsolver::core::Dumper::dumpMPQConfiguration(), mpqsolver::core::Dumper::dumpMPQError(), and mpqsolver::core::Quantizer::Context::granularity.

onEndSolver()

void DumpingHooks::onEndSolver ( const LayerParams &  layers, const std::string &  def_dtype, float  qerror  )

overridevirtual

called at the end of mpq search

Implements mpqsolver::core::SolverHooks.

Definition at line 59 of file DumpingHooks.cpp.

{
  _dumper.dumpFinalMPQ(layers, def_dtype, _ctx.granularity);
  if (!std::isnan(qerror))
  {
    _dumper.dumpMPQError(qerror);
  }
}

References mpqsolver::core::Dumper::dumpFinalMPQ(), mpqsolver::core::Dumper::dumpMPQError(), and mpqsolver::core::Quantizer::Context::granularity.

onQuantized()

void DumpingHooks::onQuantized ( luci::Module *  module ) const

overridevirtual

called on successfull quantization

Implements mpqsolver::core::QuantizerHook.

Definition at line 69 of file DumpingHooks.cpp.

{
  if (_in_iterations)
  {
    _dumper.dumpQuantized(module, _num_of_iterations);
  }
}

References mpqsolver::core::Dumper::dumpQuantized().

---

The documentation for this class was generated from the following files:

• compiler/circle-mpqsolver/src/core/DumpingHooks.h
• compiler/circle-mpqsolver/src/core/DumpingHooks.cpp