cuda_solver_direct.h

Go to the documentation of this file.

// ------------------------------------------------------------------------
//
// SPDX-License-Identifier: LGPL-2.1-or-later
// Copyright (C) 2018 - 2024 by the deal.II authors
//
// This file is part of the deal.II library.
//
// Part of the source code is dual licensed under Apache-2.0 WITH
// LLVM-exception OR LGPL-2.1-or-later. Detailed license information
// governing the source code and code contributions can be found in
// LICENSE.md and CONTRIBUTING.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