mpqsolver::pattern::PatternSolver Class Reference

#include <PatternSolver.h>

Collaboration diagram for mpqsolver::pattern::PatternSolver:

Public Member Functions
	PatternSolver (const mpqsolver::core::Quantizer::Context &ctx, const std::vector< QuantizationPattern > &patterns)
	construct PatternSolver using qunatization context and patterns to apply
std::unique_ptr< luci::Module >	run (const std::string &module_path) override
	run solver for recorded float module at module_path
Public Member Functions inherited from mpqsolver::MPQSolver
	MPQSolver (const core::Quantizer::Context &ctx)
virtual	~MPQSolver ()=default
void	setSaveIntermediate (const std::string &save_path)
	set all intermediate artifacts to be saved

Additional Inherited Members
Protected Member Functions inherited from mpqsolver::MPQSolver
std::unique_ptr< luci::Module >	readModule (const std::string &path)
Protected Attributes inherited from mpqsolver::MPQSolver
std::string	_input_quantization
std::string	_output_quantization
std::unique_ptr< core::Quantizer >	_quantizer
std::unique_ptr< core::DumpingHooks >	_hooks

Detailed Description

Definition at line 48 of file PatternSolver.h.

Constructor & Destructor Documentation

PatternSolver()

PatternSolver::PatternSolver	(	const mpqsolver::core::Quantizer::Context &	ctx,
		const std::vector< QuantizationPattern > &	patterns
	)

construct PatternSolver using qunatization context and patterns to apply

Definition at line 29 of file PatternSolver.cpp.

  : MPQSolver(ctx)
{
  MPQOptions options{patterns};
  setMPQOptions(options);
}

Member Function Documentation

run()

std::unique_ptr< luci::Module > PatternSolver::run ( const std::string &  module_path )

overridevirtual

run solver for recorded float module at module_path

Implements mpqsolver::MPQSolver.

Definition at line 37 of file PatternSolver.cpp.

{
  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");
  }
 
  return module;
}

References mpqsolver::MPQSolver::_hooks, and mpqsolver::MPQSolver::_quantizer.

Referenced by package.infer.session::inference().

---

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

• compiler/circle-mpqsolver/src/pattern/PatternSolver.h
• compiler/circle-mpqsolver/src/pattern/PatternSolver.cpp