cudss.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
 
// includes, FEAT
#include <kernel/base_header.hpp>
#include <kernel/solver/base.hpp>
#include <kernel/lafem/saddle_point_matrix.hpp>
#include <kernel/lafem/sparse_matrix_bcsr.hpp>
#include <kernel/lafem/sparse_matrix_csr.hpp>
#include <kernel/lafem/mean_filter.hpp>
#include <kernel/lafem/tuple_vector.hpp>
#include <kernel/lafem/dense_vector.hpp>
#include <kernel/lafem/dense_vector_blocked.hpp>
 
namespace FEAT
{
  namespace Solver
  {
#if defined(FEAT_HAVE_CUDSS) || defined(DOXYGEN)
    typedef double cudss_native_data_type;
    typedef std::uint32_t cudss_native_index_type;
 
    class CUDSS :
      public SolverBase<LAFEM::DenseVector<double, Index>>
    {
    public:
      typedef LAFEM::SparseMatrixCSR<double, Index> MatrixType;
      typedef LAFEM::DenseVector<double, Index> VectorType;
 
      typedef SolverBase<VectorType> BaseClass;
 
    private:
      const MatrixType& _system_matrix;
      void* _cudss_core;
 
    public:
      explicit CUDSS(const MatrixType& system_matrix);
 
      virtual ~CUDSS();
 
      virtual String name() const override
      {
        return "CUDSS";
      }
 
      virtual void init_symbolic() override;
      virtual void done_symbolic() override;
      virtual void init_numeric() override;
      virtual void done_numeric() override;
 
      virtual Status apply(VectorType& vec_sol, const VectorType& vec_rhs) override;
    }; // class CUDSS
 
    inline std::shared_ptr<CUDSS> new_cudss(const LAFEM::SparseMatrixCSR<double, Index>& matrix)
    {
      return std::make_shared<CUDSS>(matrix);
    }
#endif // defined(FEAT_HAVE_CUDSS) || defined(DOXYGEN)
  } // namespace Solver
} // namespace FEAT