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
126 void describe_impl(Teuchos::FancyOStream &out,
127 const Teuchos::EVerbosityLevel verbLevel) const;
128
132 bool matrixShapeOK_impl() const;
133
157 const Teuchos::RCP<Teuchos::ParameterList> & parameterList );
158
159
166 Teuchos::RCP<const Teuchos::ParameterList> getValidParameters_impl() const;
167
168
177 bool loadA_impl(EPhase current_phase);
178
179
180 int ConvertToTriplet();
181
182 void MUMPS_ERROR() const;
183
184#ifdef HAVE_MPI
185 MPI_Comm MUMPSComm;
186#endif
187
188
189 bool MUMPS_MATRIX_LOAD;
190 bool MUMPS_STRUCT;
191 mutable bool MUMPS_MATRIX_LOAD_PREORDERING;
192
193 // The following Arrays are persisting storage arrays for A, X, and B
195 host_value_type_view host_nzvals_view_;
197 host_ordinal_type_view host_rows_view_;
199 host_ordinal_type_view host_col_ptr_view_;
200
202 mutable Teuchos::Array<mumps_type> xvals_; local_ordinal_type ldx_;
204 mutable Teuchos::Array<mumps_type> bvals_; local_ordinal_type ldb_;
205
206 mutable MUMPS_STRUC_C mumps_par;
207
208 bool is_contiguous_;
209
210}; // End class MUMPS
211
212
213template <class Matrix,class Vector>
214class MUMPSNS : public SolverCore<Amesos2::MUMPS, Matrix, Vector>
215{
216 friend class SolverCore<Amesos2::MUMPS,Matrix,Vector>; // Give our base access
217 // to our private
218 // implementation funcs
219public:
220
221
222 MUMPSNS(Teuchos::RCP<const Matrix> A,
223 Teuchos::RCP<Vector> X,
224 Teuchos::RCP<const Vector> B);
225 /*
226 {
227 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
228 "This solver is not support for these types.");
229
230 }
231
232 */
233 ~MUMPSNS( )
234 {}
235
236
237private:
238
239 int preOrdering_impl()
240 {
241
242 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
243 "This solver is not support for these types.");
244
245 }
246
247 int symbolicFactorization_impl()
248 {
249
250 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
251 "This solver is not support for these types.");
252
253 }
254
255 int numericFactorization_impl()
256 {
257
258 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
259 "This solver is not support for these types.");
260
261 }
262
263 int solve_impl(const Teuchos::Ptr<MultiVecAdapter<Vector> > X,
264 const Teuchos::Ptr<const MultiVecAdapter<Vector> > B) const
265 {
266
267 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
268 "This solver is not support for these types.");
269
270 }
271
272 bool matrixShapeOK_impl() const
273 {
274
275 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
276 "This solver is not support for these types.");
277
278 }
279
280 void setParameters_impl(
281 const Teuchos::RCP<Teuchos::ParameterList> & parameterList )
282 {
283
284 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
285 "This solver is not support for these types.");
286
287 }
288
289 Teuchos::RCP<const Teuchos::ParameterList> getValidParameters_impl() const
290 {
291
292 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
293 "This solver is not support for these types.");
294
295 }
296
297
298 bool loadA_impl(EPhase current_phase)
299 {
300
301 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
302 "This solver is not support for these types.");
303
304 }
305
306
307 int ConvertToTriplet()
308 {
309
310 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
311 "This solver is not support for these types.");
312
313 }
314
315
316 void MUMPS_ERROR() const
317 {
318
319 TEUCHOS_TEST_FOR_EXCEPTION(true, std::logic_error,
320 "This solver is not support for these types.");
321
322 }
323
324
325#ifdef HAVE_MPI
326 MPI_Comm MUMPSComm;
327#endif
328
329 bool is_contiguous_;
330
331};
332
333
334
335// Specialize solver_traits struct for MUMPS
336// TODO
337template <>
338struct solver_traits<MUMPS> {
339#ifdef HAVE_TEUCHOS_COMPLEX
340 typedef Meta::make_list4<float,
341 double,
342 std::complex<float>,
343 std::complex<double> > supported_scalars;
344#else
345 typedef Meta::make_list2<float, double> supported_scalars;
346#endif
347};
348
349} // end namespace Amesos2
350
351#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:202
host_ordinal_type_view host_rows_view_
Stores the location in Ai_ and Aval_ that starts row j.
Definition Amesos2_MUMPS_decl.hpp:197
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:195
host_ordinal_type_view host_col_ptr_view_
Stores the row indices of the nonzero entries.
Definition Amesos2_MUMPS_decl.hpp:199
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:204
void setParameters_impl(const Teuchos::RCP< Teuchos::ParameterList > &parameterList)
Definition Amesos2_MUMPS_def.hpp:254
void describe_impl(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel) const
Prints the status information about the current solver with some level of verbosity.
Definition Amesos2_MUMPS_def.hpp:380
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