preconditioner_multielimination.hpp

Go to the documentation of this file.

#ifndef PARALUTION_PRECONDITIONER_MULTIELIMINATION_HPP_
#define PARALUTION_PRECONDITIONER_MULTIELIMINATION_HPP_

#include "../solver.hpp"
#include "preconditioner.hpp"
#include "../../base/local_vector.hpp"

#include <vector>

namespace paralution {

template <class OperatorType, class VectorType, typename ValueType>
class MultiElimination : public Preconditioner<OperatorType, VectorType, ValueType> {

public:

  MultiElimination();
  virtual ~MultiElimination();

  inline int get_size_diag_block(void) const { return this->size_; }

  inline int get_level(void) const { return this->level_; }

  virtual void Print(void) const;
  virtual void Clear(void);

  virtual void Set(Solver<OperatorType, VectorType, ValueType> &AA_Solver,
                   const int level, const double drop_off = double(0.0));

  virtual void SetPrecondMatrixFormat(const unsigned int mat_format);

  virtual void Build(void);

  virtual void Solve(const VectorType &rhs,
                     VectorType *x);

protected:

  OperatorType A_;
  OperatorType D_;
  OperatorType E_,F_,C_;
  OperatorType AA_;

  int AA_nrow_, AA_nnz_;

  bool op_mat_format_;
  unsigned int precond_mat_format_;

  VectorType x_;
  VectorType x_1_;
  VectorType x_2_;

  VectorType rhs_;
  VectorType rhs_1_;
  VectorType rhs_2_;

  MultiElimination<OperatorType, VectorType, ValueType> *AA_me_;
  Solver<OperatorType, VectorType, ValueType> *AA_solver_;

  bool diag_solver_init_;
  int level_;
  double drop_off_;

  VectorType inv_vec_D_;
  VectorType vec_D_;
  LocalVector<int> permutation_;
  int size_;

  virtual void MoveToHostLocalData_(void);
  virtual void MoveToAcceleratorLocalData_(void);

};


}

#endif // PARALUTION_PRECONDITIONER_MULTIELIMINATION_HPP_