cuda_solver_direct.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2018 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, 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 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE.md at
// the top level directory of deal.II.
//
// ---------------------------------------------------------------------

#ifndef dealii_cuda_solver_direct_h
#define dealii_cuda_solver_direct_h

#include <deal.II/base/config.h>

#ifdef DEAL_II_WITH_CUDA
#  include <deal.II/base/cuda.h>

#  include <deal.II/lac/cuda_sparse_matrix.h>
#  include <deal.II/lac/cuda_vector.h>
#  include <deal.II/lac/solver_control.h>

DEAL_II_NAMESPACE_OPEN

namespace CUDAWrappers
{
  template <typename Number>
  class SolverDirect
  {
  public:
    struct AdditionalData
    {
      explicit AdditionalData(const std::string &solver_type = "LU_dense");

      std::string solver_type;
    };

    SolverDirect(const Utilities::CUDA::Handle &handle,
                 SolverControl &                cn,
                 const AdditionalData &         data = AdditionalData());

    virtual ~SolverDirect() = default;

    void
    solve(const SparseMatrix<Number> &                       A,
          LinearAlgebra::CUDAWrappers::Vector<Number> &      x,
          const LinearAlgebra::CUDAWrappers::Vector<Number> &b);

    SolverControl &
    control() const;

  private:
    const Utilities::CUDA::Handle &cuda_handle;

    SolverControl &solver_control;

    const AdditionalData additional_data;
  };
} // namespace CUDAWrappers

DEAL_II_NAMESPACE_CLOSE

#endif

#endif