dpcg.hpp 2.35 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
// *************************************************************************
//
//    This code is developed and maintained by TU Delft. 
//    The deflation solver (files src/solvers/deflation/dpcg.hpp and
//    src/solvers/deflation/dpcg.cpp) are released under GNU LESSER GENERAL 
//    PUBLIC LICENSE (LGPL v3)
//
//    Copyright (C) 2013 Kees Vuik (TU Delft)
//    Delft University of Technology, the Netherlands
//
//    This program is free software: you can redistribute it and/or modify
//    it under the terms of the GNU Lesser General Public License as 
//    published by the Free Software Foundation, either version 3 of the 
//    License, or (at your option) any later version.
//
//    This program is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU Lesser General Public 
//    License along with this program.  
//    If not, see <http://www.gnu.org/licenses/>.
//
// *************************************************************************

#ifndef PARALUTION_DEFLATION_DPCG_HPP_
#define PARALUTION_DEFLATION_DPCG_HPP_

#include "../solver.hpp"

namespace paralution {

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

  DPCG();
  virtual ~DPCG();

  virtual void Print(void) const;

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

  virtual void SetNVectors(const int novecni);

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:

  void MakeZ_COO(void);
  void MakeZ_CSR(void);

  OperatorType L_, LT_;
  OperatorType AZ_, ZT_;
  OperatorType Z_, AZT_;
  OperatorType E_;

  VectorType r_, w_;
  VectorType p_, q_;

  VectorType Dinv_;
  VectorType hat_, intmed_;
  VectorType Qb_, Ptx_;
  VectorType LLtx_, LLtx2_;

  int novecni_;
};


}

#endif // PARALUTION_DEFLATION_DPCG_HPP_