Amesos2 - Direct Sparse Solver Interfaces  Version of the Day
Amesos2_Kokkos_Impl.hpp
Go to the documentation of this file.
1 // @HEADER
2 //
3 // ***********************************************************************
4 //
5 // Amesos2: Templated Direct Sparse Solver Package
6 // Copyright 2011 Sandia Corporation
7 //
8 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
9 // the U.S. Government retains certain rights in this software.
10 //
11 // Redistribution and use in source and binary forms, with or without
12 // modification, are permitted provided that the following conditions are
13 // met:
14 //
15 // 1. Redistributions of source code must retain the above copyright
16 // notice, this list of conditions and the following disclaimer.
17 //
18 // 2. Redistributions in binary form must reproduce the above copyright
19 // notice, this list of conditions and the following disclaimer in the
20 // documentation and/or other materials provided with the distribution.
21 //
22 // 3. Neither the name of the Corporation nor the names of the
23 // contributors may be used to endorse or promote products derived from
24 // this software without specific prior written permission.
25 //
26 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
27 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
30 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
31 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
32 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
33 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
34 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
35 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
36 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37 //
38 // Questions? Contact Michael A. Heroux (maherou@sandia.gov)
39 //
40 // ***********************************************************************
41 //
42 // @HEADER
43 
52 #ifndef AMESOS2_KOKKOS_IMPL_HPP
53 #define AMESOS2_KOKKOS_IMPL_HPP
54 
55 #include <type_traits>
57 #include <Kokkos_Core.hpp>
58 #include <KokkosSparse_CrsMatrix.hpp>
59 
60 #define AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(S,LO,EXEC_SPACE) \
61  template class Amesos2::AMESOS2_KOKKOS_IMPL_SOLVER_NAME<KokkosSparse::CrsMatrix<S, LO, \
62  typename EXEC_SPACE::device_type>, \
63  Kokkos::View<S**, Kokkos::LayoutLeft, typename EXEC_SPACE::device_type> >;
64 
65 #ifdef KOKKOS_ENABLE_CUDA_UVM
66 #define AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER_UVM_OFF(S,LO) \
67  template class Amesos2::AMESOS2_KOKKOS_IMPL_SOLVER_NAME<KokkosSparse::CrsMatrix<S, LO, \
68  Kokkos::Device<Kokkos::Cuda,Kokkos::CudaSpace>>, \
69  Kokkos::View<S**, Kokkos::LayoutLeft, Kokkos::Device<Kokkos::Cuda,Kokkos::CudaSpace>> >;
70 #else
71 #define AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER_UVM_OFF(S,LO)
72 #endif
73 
74 
75 #if defined(KOKKOS_ENABLE_SERIAL)
76 #ifdef HAVE_TPETRA_INST_FLOAT
77  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(float, int, Kokkos::Serial)
78 #endif
79 #ifdef HAVE_TPETRA_INST_DOUBLE
80  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(double, int, Kokkos::Serial)
81 #endif
82 #ifdef HAVE_TPETRA_INST_COMPLEX_FLOAT
83  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<float>, int, Kokkos::Serial)
84 #endif
85 #ifdef HAVE_TPETRA_INST_COMPLEX_DOUBLE
86  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<double>, int, Kokkos::Serial)
87 #endif
88 #endif
89 
90 #if defined(KOKKOS_ENABLE_THREADS)
91 #define EXEC_SPACE Kokkos::Threads
92 #ifdef HAVE_TPETRA_INST_FLOAT
93  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(float, int, Kokkos::Threads)
94 #endif
95 #ifdef HAVE_TPETRA_INST_DOUBLE
96  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(double, int, Kokkos::Threads)
97 #endif
98 #ifdef HAVE_TPETRA_INST_COMPLEX_FLOAT
99  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<float>, int, Kokkos::Threads)
100 #endif
101 #ifdef HAVE_TPETRA_INST_COMPLEX_DOUBLE
102  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<double>, int, Kokkos::Threads)
103 #endif
104 #endif // KOKKOS_ENABLE_THREADS
105 
106 #if defined(KOKKOS_ENABLE_OPENMP)
107 #ifdef HAVE_TPETRA_INST_FLOAT
108  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(float, int, Kokkos::OpenMP)
109 #endif
110 #ifdef HAVE_TPETRA_INST_DOUBLE
111  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(double, int, Kokkos::OpenMP)
112 #endif
113 #ifdef HAVE_TPETRA_INST_COMPLEX_FLOAT
114  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<float>, int, Kokkos::OpenMP)
115 #endif
116 #ifdef HAVE_TPETRA_INST_COMPLEX_DOUBLE
117  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<double>, int, Kokkos::OpenMP)
118 #endif
119 #endif // KOKKOS_ENABLE_OPENMP
120 
121 #if defined(KOKKOS_ENABLE_CUDA)
122 #ifdef HAVE_TPETRA_INST_FLOAT
123  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(float, int, Kokkos::Cuda)
124  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER_UVM_OFF(float, int)
125 #endif
126 #ifdef HAVE_TPETRA_INST_DOUBLE
127  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(double, int, Kokkos::Cuda)
128  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER_UVM_OFF(double, int)
129 #endif
130 #ifdef HAVE_TPETRA_INST_COMPLEX_FLOAT
131  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<float>, int, Kokkos::Cuda)
132  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER_UVM_OFF(Kokkos::complex<float>, int)
133 #endif
134 #ifdef HAVE_TPETRA_INST_COMPLEX_DOUBLE
135  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<double>, int, Kokkos::Cuda)
136  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER_UVM_OFF(Kokkos::complex<double>, int)
137 #endif
138 #endif // KOKKOS_ENABLE_CUDA
139 
140 #if defined(KOKKOS_ENABLE_HIP)
141 #ifdef HAVE_TPETRA_INST_FLOAT
142  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(float, int, Kokkos::Experimental::HIP)
143 #endif
144 #ifdef HAVE_TPETRA_INST_DOUBLE
145  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(double, int, Kokkos::Experimental::HIP)
146 #endif
147 #ifdef HAVE_TPETRA_INST_COMPLEX_FLOAT
148  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<float>, int, Kokkos::Experimental::HIP)
149 #endif
150 #ifdef HAVE_TPETRA_INST_COMPLEX_DOUBLE
151  AMESOS2_KOKKOS_LOCAL_INSTANT_KOKKOS_ADAPTER(Kokkos::complex<double>, int, Kokkos::Experimental::HIP)
152 #endif
153 #endif // KOKKOS_ENABLE_HIP
154 
155 #endif // AMESOS2_KOKKOS_IMPL_HPP
Amesos2::MultiVecAdapter specialization for the Kokkos::View class.