Amesos2 - Direct Sparse Solver Interfaces Version of the Day
Amesos2_MUMPS_decl.hpp
Go to the documentation of this file.
1// @HEADER
2// *****************************************************************************
3// Amesos2: Templated Direct Sparse Solver Package
4//
5// Copyright 2011 NTESS and the Amesos2 contributors.
6// SPDX-License-Identifier: BSD-3-Clause
7// *****************************************************************************
8// @HEADER
9
18#ifndef AMESOS2_MUMPS_DECL_HPP
19#define AMESOS2_MUMPS_DECL_HPP
20
22#include "Amesos2_SolverCore.hpp"
24
25
26namespace Amesos2 {
27
28
49template <class Matrix,class Vector>
50class MUMPS : public SolverCore<Amesos2::MUMPS, Matrix, Vector>
51{
52 friend class SolverCore<Amesos2::MUMPS,Matrix,Vector>; // Give our base access
53 // to our private
54 // implementation funcs
55public:
56
58 static const char* name; // declaration. Initialization outside.
59
60 typedef MUMPS<Matrix,Vector> type;
61 typedef SolverCore<Amesos2::MUMPS,Matrix,Vector> super_type;
62
63 // Since typedef's are not inheritted, go grab them
64 typedef typename super_type::scalar_type scalar_type;
65 typedef typename super_type::local_ordinal_type local_ordinal_type;
66 typedef typename super_type::global_ordinal_type global_ordinal_type;
67 typedef typename super_type::global_size_type global_size_type;
68
69 typedef TypeMap<Amesos2::MUMPS,scalar_type> type_map;
70
71 typedef typename type_map::type mumps_type;
72 typedef typename type_map::magnitude_type magnitude_type;
73 typedef typename type_map::MUMPS_STRUC_C MUMPS_STRUC_C;
74
75 typedef Kokkos::DefaultHostExecutionSpace HostExecSpaceType;
76 typedef typename HostExecSpaceType::memory_space HostMemSpaceType;
77
78 typedef Kokkos::View<local_ordinal_type*, HostExecSpaceType> host_ordinal_type_view;
79 typedef Kokkos::View<mumps_type*, HostExecSpaceType> host_value_type_view;
80
81 MUMPS(Teuchos::RCP<const Matrix> A,
82 Teuchos::RCP<Vector> X,
83 Teuchos::RCP<const Vector> B);
84 ~MUMPS( );
85
86
87private:
88
94 int preOrdering_impl();
95
96
97 int symbolicFactorization_impl();
98
99
106
107
119 int solve_impl(const Teuchos::Ptr<MultiVecAdapter<Vector> > X,
120 const Teuchos::Ptr<const MultiVecAdapter<Vector> > B) const;
121
122
126 bool matrixShapeOK_impl() const;
127
150 const Teuchos::RCP<Teuchos::ParameterList> & parameterList );
151
152
159 Teuchos::RCP<const Teuchos::ParameterList> getValidParameters_impl() const;
160
161
170 bool loadA_impl(EPhase current_phase);
171
172
173 int ConvertToTriplet();
174
175 void MUMPS_ERROR() const;
176
177#ifdef HAVE_MPI
178 MPI_Comm MUMPSComm;
179#endif
180
181
182 bool MUMPS_MATRIX_LOAD;
183 bool MUMPS_STRUCT;
184 mutable bool MUMPS_MATRIX_LOAD_PREORDERING;
185
186 // The following Arrays are persisting storage arrays for A, X, and B
188 host_value_type_view host_nzvals_view_;
190 host_ordinal_type_view host_rows_view_;
192 host_ordinal_type_view host_col_ptr_view_;
193
195 mutable Teuchos::Array<mumps_type> xvals_; local_ordinal_type ldx_;
197 mutable Teuchos::Array<mumps_type> bvals_; local_ordinal_type ldb_;
198
199 mutable MUMPS_STRUC_C mumps_par;
200
201 bool is_contiguous_;
202
203}; // End class MUMPS
204
205
206template <class Matrix,class Vector>
207class MUMPSNS : public SolverCore<Amesos2::MUMPS, Matrix, Vector>
208{
209 friend class SolverCore<Amesos2::MUMPS,Matrix,Vector>; // Give our base access
210 // to our private
211 // implementation funcs
212public:
213
214
215 MUMPSNS(Teuchos::RCP<const Matrix> A,
216 Teuchos::RCP<Vector> X,
217 Teuchos::RCP<const Vector> B);
218 /*
219 {
220 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
221 "This solver is not support for these types.");
222
223 }
224
225 */
226 ~MUMPSNS( )
227 {}
228
229
230private:
231
232 int preOrdering_impl()
233 {
234
235 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
236 "This solver is not support for these types.");
237
238 }
239
240 int symbolicFactorization_impl()
241 {
242
243 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
244 "This solver is not support for these types.");
245
246 }
247
248 int numericFactorization_impl()
249 {
250
251 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
252 "This solver is not support for these types.");
253
254 }
255
256 int solve_impl(const Teuchos::Ptr<MultiVecAdapter<Vector> > X,
257 const Teuchos::Ptr<const MultiVecAdapter<Vector> > B) const
258 {
259
260 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
261 "This solver is not support for these types.");
262
263 }
264
265 bool matrixShapeOK_impl() const
266 {
267
268 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
269 "This solver is not support for these types.");
270
271 }
272
273 void setParameters_impl(
274 const Teuchos::RCP<Teuchos::ParameterList> & parameterList )
275 {
276
277 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
278 "This solver is not support for these types.");
279
280 }
281
282 Teuchos::RCP<const Teuchos::ParameterList> getValidParameters_impl() const
283 {
284
285 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
286 "This solver is not support for these types.");
287
288 }
289
290
291 bool loadA_impl(EPhase current_phase)
292 {
293
294 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
295 "This solver is not support for these types.");
296
297 }
298
299
300 int ConvertToTriplet()
301 {
302
303 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
304 "This solver is not support for these types.");
305
306 }
307
308
309 void MUMPS_ERROR() const
310 {
311
312 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
313 "This solver is not support for these types.");
314
315 }
316
317
318#ifdef HAVE_MPI
319 MPI_Comm MUMPSComm;
320#endif
321
322 bool is_contiguous_;
323
324};
325
326
327
328// Specialize solver_traits struct for MUMPS
329// TODO
330template <>
331struct solver_traits<MUMPS> {
332#ifdef HAVE_TEUCHOS_COMPLEX
333 typedef Meta::make_list4<float,
334 double,
335 std::complex<float>,
336 std::complex<double> > supported_scalars;
337#else
338 typedef Meta::make_list2<float, double> supported_scalars;
339#endif
340};
341
342} // end namespace Amesos2
343
344#endif // AMESOS2_MUMPS_DECL_HPP
Provides a mechanism to map function calls to the correct Solver function based on the scalar type of...
Provides access to interesting solver traits.
Amesos2 interface to the MUMPS package.
Definition Amesos2_MUMPS_decl.hpp:51
Teuchos::Array< mumps_type > xvals_
Persisting 1D store for X.
Definition Amesos2_MUMPS_decl.hpp:195
host_ordinal_type_view host_rows_view_
Stores the location in Ai_ and Aval_ that starts row j.
Definition Amesos2_MUMPS_decl.hpp:190
bool loadA_impl(EPhase current_phase)
Reads matrix data into internal structures.
Definition Amesos2_MUMPS_def.hpp:325
bool matrixShapeOK_impl() const
Determines whether the shape of the matrix is OK for this solver.
Definition Amesos2_MUMPS_def.hpp:245
int solve_impl(const Teuchos::Ptr< MultiVecAdapter< Vector > > X, const Teuchos::Ptr< const MultiVecAdapter< Vector > > B) const
MUMPS specific solve.
Definition Amesos2_MUMPS_def.hpp:188
static const char * name
Name of this solver interface.
Definition Amesos2_MUMPS_decl.hpp:58
host_value_type_view host_nzvals_view_
Stores the values of the nonzero entries for MUMPS.
Definition Amesos2_MUMPS_decl.hpp:188
host_ordinal_type_view host_col_ptr_view_
Stores the row indices of the nonzero entries.
Definition Amesos2_MUMPS_decl.hpp:192
int numericFactorization_impl()
MUMPS specific numeric factorization.
Definition Amesos2_MUMPS_def.hpp:170
int preOrdering_impl()
Performs pre-ordering on the matrix to increase efficiency.
Definition Amesos2_MUMPS_def.hpp:138
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters_impl() const
Definition Amesos2_MUMPS_def.hpp:297
Teuchos::Array< mumps_type > bvals_
Persisting 1D store for B.
Definition Amesos2_MUMPS_decl.hpp:197
void setParameters_impl(const Teuchos::RCP< Teuchos::ParameterList > &parameterList)
Definition Amesos2_MUMPS_def.hpp:254
Amesos2::SolverCore: A templated interface for interaction with third-party direct sparse solvers.
Definition Amesos2_SolverCore_decl.hpp:72
EPhase
Used to indicate a phase in the direct solution.
Definition Amesos2_TypeDecl.hpp:31
A templated MultiVector class adapter for Amesos2.
Definition Amesos2_MultiVecAdapter_decl.hpp:142
Map types to solver-specific data-types and enums.
Definition Amesos2_TypeMap.hpp:48
Provides traits about solvers.
Definition Amesos2_SolverTraits.hpp:37