SubstitutionUnit.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/api.h>
 
#include <tvm/Range.h>
#include <tvm/constraint/BasicLinearConstraint.h>
#include <tvm/function/BasicLinearFunction.h>
#include <tvm/hint/Substitution.h>
#include <tvm/hint/abstract/SubstitutionCalculatorImpl.h>
 
#include <Eigen/Core>
 
#include <vector>
 
namespace tvm
{
 
namespace hint
{
 
namespace internal
{
class TVM_DLLAPI SubstitutionUnit
{
public:
  SubstitutionUnit(const std::vector<Substitution> & substitutionPool,
                   const std::vector<std::vector<size_t>> & groups,
                   const std::vector<size_t> order);
 
  void update();
 
  const std::vector<VariablePtr> & variables() const;
  const std::vector<std::shared_ptr<function::BasicLinearFunction>> & variableSubstitutions() const;
  const std::vector<VariablePtr> & additionalVariables() const;
  const std::vector<std::shared_ptr<constraint::BasicLinearConstraint>> & additionalConstraints() const;
  const std::vector<VariablePtr> & otherVariables() const;
 
private:
  void extractSubstitutions(const std::vector<Substitution> & substitutionPool,
                            const std::vector<std::vector<size_t>> & groups,
                            const std::vector<size_t> order);
  void scanSubstitutions();
  void computeDependencies();
  void initializeMatrices();
  void createFunctions();
 
  int m_;
  std::vector<Substitution> substitutions_;
  std::vector<std::shared_ptr<abstract::SubstitutionCalculatorImpl>> calculators_;
  std::vector<LinearConstraintPtr> constraints_;
  VariableVector x_;
  VariableVector y_;
  VariableVector z_;
  std::vector<Range> substitutionMRanges_;
  std::vector<Range> substitutionNRanges_;
 
  std::vector<std::vector<size_t>> sub2cstr_;
  std::vector<std::vector<size_t>> sub2x_;
  std::vector<size_t> x2sub_;
  std::vector<Range> xRange_;
 
  std::vector<std::vector<int>> constraintsY_;
  std::vector<std::vector<int>> CXdependencies_;
  std::vector<std::vector<int>> XYdependencies_;
  std::vector<std::vector<int>> XZdependencies_;
  std::vector<std::vector<int>> SYdependencies_;
  std::vector<std::vector<int>> SZdependencies_;
 
  std::vector<std::shared_ptr<function::BasicLinearFunction>> varSubstitutions_;
  std::vector<std::shared_ptr<constraint::BasicLinearConstraint>> remaining_;
 
  Eigen::MatrixXd B_;
  Eigen::MatrixXd Z_;
  Eigen::VectorXd c_;
  std::vector<bool> cIsZero_;
  Eigen::MatrixXd M_;
  Eigen::MatrixXd AsZ_;
  Eigen::VectorXd u_;
  std::vector<bool> uIsZero_;
  std::vector<Eigen::MatrixXd> StB_;
  std::vector<Eigen::MatrixXd> StZ_;
  std::vector<Eigen::VectorXd> Stc_;
 
  std::vector<bool> firstY_;
  std::vector<bool> firstZ_;
};
 
} // namespace internal
 
} // namespace hint
 
} // namespace tvm