Reference documentation for deal.II version Git 840703ecf3 2019-10-14 17:15:20 -0400
\(\newcommand{\dealcoloneq}{\mathrel{\vcenter{:}}=}\)
cuda_solver_direct.h
1 // ---------------------------------------------------------------------
2 //
3 // Copyright (C) 2018 - 2019 by the deal.II authors
4 //
5 // This file is part of the deal.II library.
6 //
7 // The deal.II library is free software; you can use it, redistribute
8 // it, and/or modify it under the terms of the GNU Lesser General
9 // Public License as published by the Free Software Foundation; either
10 // version 2.1 of the License, or (at your option) any later version.
11 // The full text of the license can be found in the file LICENSE.md at
12 // the top level directory of deal.II.
13 //
14 // ---------------------------------------------------------------------
15 
16 #ifndef dealii_cuda_solver_direct_h
17 #define dealii_cuda_solver_direct_h
18 
19 #include <deal.II/base/config.h>
20 
21 #ifdef DEAL_II_COMPILER_CUDA_AWARE
22 # include <deal.II/base/cuda.h>
23 
24 # include <deal.II/lac/cuda_sparse_matrix.h>
25 # include <deal.II/lac/cuda_vector.h>
26 # include <deal.II/lac/solver_control.h>
27 
28 DEAL_II_NAMESPACE_OPEN
29 
30 namespace CUDAWrappers
31 {
42  template <typename Number>
44  {
45  public:
50  {
54  explicit AdditionalData(const std::string &solver_type = "LU_dense");
55 
66  std::string solver_type;
67  };
68 
73  SolverControl & cn,
74  const AdditionalData & data = AdditionalData());
75 
79  virtual ~SolverDirect() = default;
80 
84  void
85  solve(const SparseMatrix<Number> & A,
88 
93  control() const;
94 
95  private:
100 
108 
113  };
114 } // namespace CUDAWrappers
115 
116 DEAL_II_NAMESPACE_CLOSE
117 
118 #endif
119 
120 #endif
const AdditionalData additional_data
virtual ~SolverDirect()=default
SolverDirect(const Utilities::CUDA::Handle &handle, SolverControl &cn, const AdditionalData &data=AdditionalData())
const Utilities::CUDA::Handle & cuda_handle
SolverControl & control() const
void solve(const SparseMatrix< Number > &A, LinearAlgebra::CUDAWrappers::Vector< Number > &x, const LinearAlgebra::CUDAWrappers::Vector< Number > &b)
AdditionalData(const std::string &solver_type="LU_dense")