53#ifndef AMESOS2_SUPERLUDIST_DECL_HPP
54#define AMESOS2_SUPERLUDIST_DECL_HPP
57#include "Amesos2_SolverCore.hpp"
88template <
class Matrix,
103 typedef Matrix matrix_type;
104 typedef Vector vector_type;
107 typedef typename super_type::scalar_type scalar_type;
108 typedef typename super_type::local_ordinal_type local_ordinal_type;
109 typedef typename super_type::global_ordinal_type global_ordinal_type;
110 typedef typename super_type::global_size_type global_size_type;
111 typedef typename super_type::node_type node_type;
115 typedef typename type_map::type slu_type;
116 typedef typename type_map::magnitude_type magnitude_type;
120 typedef Kokkos::DefaultHostExecutionSpace HostExecSpaceType;
121 typedef Kokkos::View<SLUD::int_t*, HostExecSpaceType> host_size_type_array;
122 typedef Kokkos::View<SLUD::int_t*, HostExecSpaceType> host_ordinal_type_array;
123 typedef Kokkos::View<slu_type*, HostExecSpaceType> host_value_type_array;
135 Teuchos::RCP<Vector> X,
136 Teuchos::RCP<const Vector> B);
188 int
solve_impl(const Teuchos::Ptr<MultiVecAdapter<Vector> > X,
189 const Teuchos::Ptr<const MultiVecAdapter<Vector> > B) const;
231 const Teuchos::RCP<Teuchos::ParameterList> & parameterList );
276 mutable struct SLUData {
278 SLUD::SuperMatrix AC;
279 typename type_map::LUstruct_t LU;
280 SLUD::Glu_freeable_t glu_freeable;
287 SLUD::int_t *sizes, *fstVtxSep;
288 SLUD::Pslu_freeable_t pslu_freeable;
290 SLUD::amesos2_superlu_dist_options_t options;
291 SLUD::amesos2_superlu_dist_mem_usage_t mem_usage;
292 SLUD::gridinfo_t grid;
294 typename type_map::LUstruct_t lu;
295 SLUD::SuperLUStat_t stat;
296 typename type_map::SOLVEstruct_t solve_struct;
298 Teuchos::Array<magnitude_type> berr;
299 Teuchos::Array<magnitude_type> ferr;
302 typename type_map::ScalePermstruct_t scale_perm;
304 Teuchos::Array<magnitude_type> R, C;
305 Teuchos::Array<magnitude_type> R1, C1;
306 Teuchos::Array<SLUD::int_t> perm_r, perm_c;
308 SLUD::DiagScale_t equed;
310 magnitude_type rowcnd, colcnd, amax;
328 mutable bool same_solve_struct_;
331 Teuchos::RCP<
const Tpetra::Map<local_ordinal_type,
343#if defined(HAVE_TEUCHOS_COMPLEX) && !defined(__clang__)
344 typedef Meta::make_list3<double, std::complex<double>, SLUD::Z::doublecomplex> supported_scalars;
346 typedef Meta::make_list1<double> supported_scalars;
Provides access to interesting solver traits.
Provides a mechanism to map function calls to the correct Solver function based on the scalar type of...
Amesos2::SolverCore: A templated interface for interaction with third-party direct sparse solvers.
Definition: Amesos2_SolverCore_decl.hpp:106
Amesos2 interface to the distributed memory version of SuperLU.
Definition: Amesos2_Superludist_decl.hpp:91
Teuchos::RCP< const Tpetra::Map< local_ordinal_type, global_ordinal_type, node_type > > superlu_rowmap_
Maps rows of the matrix to processors in the SuperLU_DIST processor grid.
Definition: Amesos2_Superludist_decl.hpp:333
bool in_grid_
true if this processor is in SuperLU_DISTS's 2D process grid
Definition: Amesos2_Superludist_decl.hpp:326
bool matrixShapeOK_impl() const
Determines whether the shape of the matrix is OK for this solver.
Definition: Amesos2_Superludist_def.hpp:690
void setParameters_impl(const Teuchos::RCP< Teuchos::ParameterList > ¶meterList)
Definition: Amesos2_Superludist_def.hpp:699
static const char * name
Name of this solver interface.
Definition: Amesos2_Superludist_decl.hpp:98
int numericFactorization_impl()
SuperLU_DIST specific numeric factorization.
Definition: Amesos2_Superludist_def.hpp:426
host_ordinal_type_array colind_view_
Stores the row indices of the nonzero entries.
Definition: Amesos2_Superludist_decl.hpp:317
bool loadA_impl(EPhase current_phase)
Reads matrix data into internal solver structures.
Definition: Amesos2_Superludist_def.hpp:867
host_value_type_array nzvals_view_
Stores the values of the nonzero entries for SuperLU_DIST.
Definition: Amesos2_Superludist_decl.hpp:315
int preOrdering_impl()
Performs pre-ordering on the matrix to increase efficiency.
Definition: Amesos2_Superludist_def.hpp:328
Teuchos::Array< slu_type > bvals_
1D store for B values
Definition: Amesos2_Superludist_decl.hpp:321
host_size_type_array rowptr_view_
Stores the location in Ai_ and Aval_ that starts row j.
Definition: Amesos2_Superludist_decl.hpp:319
int solve_impl(const Teuchos::Ptr< MultiVecAdapter< Vector > > X, const Teuchos::Ptr< const MultiVecAdapter< Vector > > B) const
SuperLU_DIST specific solve.
Definition: Amesos2_Superludist_def.hpp:526
void get_default_grid_size(int nprocs, SLUD::int_t &nprow, SLUD::int_t &npcol) const
Definition: Amesos2_Superludist_def.hpp:842
int symbolicFactorization_impl()
Perform symbolic factorization of the matrix using SuperLU_DIST.
Definition: Amesos2_Superludist_def.hpp:383
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters_impl() const
Definition: Amesos2_Superludist_def.hpp:772
Teuchos::Array< slu_type > xvals_
1D store for X values
Definition: Amesos2_Superludist_decl.hpp:323
Passes functions to TPL functions based on type.
Definition: Amesos2_FunctionMap.hpp:77
Map types to solver-specific data-types and enums.
Definition: Amesos2_TypeMap.hpp:82
Provides traits about solvers.
Definition: Amesos2_SolverTraits.hpp:71