solvergpu_agm.h

Go to the documentation of this file.

#ifndef  _MY_SolverGPU_AGM_
#define _MY_SolverGPU_AGM_
#include "globals.h"
#include <lac/sparse_matrix.h>
#include "linearsolver.h"
class SolverGPU_AGM  : public LinearSolver
{
 private:
  void convertSparseMatrix(const SparseMatrix<double> &M,std::vector<int> &cnt,std::vector<int> &col, std::vector<double> &ele);


  void gpu_agm(std::vector<int> &cnt,std::vector<int> &col, std::vector<double> &ele, std::vector<double> &b, std::vector<double> &x,double tol,unsigned nIt); 
  void add_diagonal(std::vector<int> &cnt,std::vector<int> &col, std::vector<double> &ele,const VecDouble &addElem);
  void gpu_agmII(std::vector<int> &cnt,std::vector<int> &col, std::vector<double> &ele_agm,std::vector<double> &ele_cg, std::vector<double> &b, std::vector<double> &x,double tol,unsigned nIt); 

  

  void gpu_agm_compressible(std::vector<int> &cnt,std::vector<int> &col, std::vector<double> &ele, std::vector<double> &b, std::vector<double> &x,double tol,unsigned nIt); 

 protected:
  double m_tol;
  unsigned m_nIt;
 public:
  SolverGPU_AGM(double tol,unsigned nIt);
  virtual ~SolverGPU_AGM(){}

  virtual void solve(const SparseMatrix<double> &M,VecDouble &sol,const VecDouble &rhs);
  
  virtual void solveAgain(const SparseMatrix<double> &M,VecDouble &sol,const VecDouble &rhs);



  void solve(const SparseMatrix<double> &A,VecDouble &sol,const VecDouble &vRHS,double tol,unsigned nIt); 


  



  void gpu_agmSchur(std::vector<int> &Acnt,std::vector<int> &Acol, std::vector<double> &Aele,
                    std::vector<int> &Bcnt,std::vector<int> &Bcol, std::vector<double> &Bele,
                    std::vector<int> &tcnt,std::vector<int> &tcol, std::vector<double> &tele,
                    std::vector<double> &_f, std::vector<double> &_g,std::vector<double> &_u, std::vector<double> &_p);


//  void solveII(const SparseMatrix<double> &A,VecDouble &sol,const VecDouble &vRHS,const VecDouble &addDiag); 

};

#endif

---