cg.hpp

Go to the documentation of this file.

#ifndef PARALUTION_KRYLOV_CG_HPP_
#define PARALUTION_KRYLOV_CG_HPP_

#include "../solver.hpp"

#include <vector>

namespace paralution {

template <class OperatorType, class VectorType, typename ValueType>
class CG : public IterativeLinearSolver<OperatorType, VectorType, ValueType> {

public:

  CG();
  virtual ~CG();

  virtual void Print(void) const;

  virtual void Build(void);
  virtual void Clear(void);

protected:

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

  virtual void PrintStart_(void) const;
  virtual void PrintEnd_(void) const;

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

private:

  VectorType r_, z_;
  VectorType p_, q_;

};


}

#endif // PARALUTION_KRYLOV_CG_HPP_