cusolver.hpp

// FEAT3: Finite Element Analysis Toolbox, Version 3
// Copyright (C) 2010 by Stefan Turek & the FEAT group
// FEAT3 is released under the GNU General Public License version 3,
// see the file 'copyright.txt' in the top level directory for details.
 
#pragma once
#include <kernel/base_header.hpp>
 
#include <kernel/solver/base.hpp>
#include <kernel/lafem/dense_vector.hpp>
#include <kernel/lafem/sparse_matrix_csr.hpp>
 
namespace FEAT
{
  namespace Solver
  {
    namespace Intern
    {
        int cuda_lu(int n, int nnzA, const double * csrValA, const int * csrRowPtrA, const int * csrColIndA,
        const double * b, double * x);
 
        int cuda_qr(int m, int nnz, const double * csrValA, const int * csrRowPtrA, const int * csrColIndA,
        const double * b, double * x);
    }
 
    class CuSolverLU :
      public SolverBase<LAFEM::DenseVector<double, unsigned int>>
    {
      private:
        const LAFEM::SparseMatrixCSR<double, unsigned int> & _system_matrix;
 
      public:
        explicit CuSolverLU(const LAFEM::SparseMatrixCSR<double, unsigned int> & system_matrix) :
          _system_matrix(system_matrix)
          {
          }
 
        virtual String name() const override
        {
          return "CuSolverLU";
        }
 
        virtual Status apply(LAFEM::DenseVector<double, unsigned int> & x, const LAFEM::DenseVector<double, unsigned int> & b) override
          {
            int status = Intern::cuda_lu((int)x.size(), (int)_system_matrix.used_elements(), _system_matrix.val(), (const int*) _system_matrix.row_ptr(), (const int*) _system_matrix.col_ind(), b.elements(), x.elements());
 
            return (status == 0) ? Status::success :  Status::aborted;
          }
    };
 
    class CuSolverQR :
      public SolverBase<LAFEM::DenseVector<double, unsigned int>>
    {
      private:
        const LAFEM::SparseMatrixCSR<double, unsigned int> & _system_matrix;
      public:
        explicit CuSolverQR(const LAFEM::SparseMatrixCSR<double, unsigned int> & system_matrix) :
          _system_matrix(system_matrix)
        {
        }
 
        virtual String name() const override
        {
          return "CuSolverQR";
        }
 
        virtual Status apply(LAFEM::DenseVector<double, unsigned int> & x, const LAFEM::DenseVector<double, unsigned int> & b) override
        {
          int status = Intern::cuda_qr((int)x.size(), (int)_system_matrix.used_elements(), _system_matrix.val(), (const int*) _system_matrix.row_ptr(), (const int*) _system_matrix.col_ind(), b.elements(), x.elements());
 
          return (status == 0) ? Status::success :  Status::aborted;
        }
    };
  } // namespace LAFEM
} // namespace FEAT