42#ifndef AMESOS2_SOLVER_MP_VECTOR_HPP
43#define AMESOS2_SOLVER_MP_VECTOR_HPP
45#include "Amesos2_Solver.hpp"
46#include "Amesos2_Factory.hpp"
52 template <
class S,
class LO,
class GO,
class N>
59 return S::static_size;
62 template <
class S,
class LO,
class GO,
class D>
64 const Teuchos::RCP<
const Tpetra::CrsMatrix<
Sacado::MP::Vector<S>, LO, GO, Kokkos::Compat::KokkosDeviceWrapperNode<D> > >& A = Teuchos::null,
65 const Teuchos::RCP<Tpetra::MultiVector<
Sacado::MP::Vector<S>, LO, GO, Kokkos::Compat::KokkosDeviceWrapperNode<D> > >& X = Teuchos::null,
66 const Teuchos::RCP<
const Tpetra::MultiVector<
Sacado::MP::Vector<S>, LO, GO, Kokkos::Compat::KokkosDeviceWrapperNode<D> > >& B = Teuchos::null)
68 if (A != Teuchos::null) {
71 else if (X != Teuchos::null) {
74 else if (B != Teuchos::null) {
87 template<
class,
class>
class ConcreteSolver>
89 public Solver< Tpetra::CrsMatrix<Sacado::MP::Vector<Storage>,
93 Tpetra::MultiVector<Sacado::MP::Vector<Storage>,
102 typedef Tpetra::CrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node>
Matrix;
103 typedef Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>
Vector;
106 typedef Tpetra::Map<LocalOrdinal,GlobalOrdinal,Node>
Map;
107 typedef Tpetra::CrsGraph<LocalOrdinal,GlobalOrdinal,Node>
FlatGraph;
108 typedef Tpetra::CrsMatrix<BaseScalar,LocalOrdinal,GlobalOrdinal,Node>
FlatMatrix;
109 typedef Tpetra::MultiVector<BaseScalar,LocalOrdinal,GlobalOrdinal,Node>
FlatVector;
114 typedef typename solver_type::type
type;
118 const Teuchos::RCP<const Matrix>& A_,
119 const Teuchos::RCP<Vector>& X_,
120 const Teuchos::RCP<const Vector>& B_) :
121 A(A_),
X(X_),
B(B_) {
127 if (
A != Teuchos::null)
129 if (
X != Teuchos::null)
131 if (
B != Teuchos::null)
134 create_solver_with_supported_type<ConcreteSolver,FlatMatrix,FlatVector>::apply(
flat_A,
flat_X,
flat_B);
216 virtual void solve(
const Teuchos::Ptr<Vector> XX,
217 const Teuchos::Ptr<const Vector> BB)
const {
267 const Teuchos::RCP<Teuchos::ParameterList> & parameterList ) {
277 virtual Teuchos::RCP<const Teuchos::ParameterList>
308 virtual void setA(
const Teuchos::RCP<const Matrix> a,
309 EPhase keep_phase = CLEAN ) {
314 if (keep_phase <= CLEAN) {
323 if (keep_phase <= SYMBFACT)
348 virtual void setA(
const Matrix* a, EPhase keep_phase = CLEAN ) {
349 this->
setA(Teuchos::rcp(a,
false), keep_phase);
360 virtual void setX(
const Teuchos::RCP<Vector> x ) {
362 if (x != Teuchos::null)
373 X = Teuchos::rcp(x,
false);
385 virtual const Teuchos::RCP<Vector>
getX(
void ) {
397 virtual void setB(
const Teuchos::RCP<const Vector> b ) {
399 if (b != Teuchos::null)
410 B = Teuchos::rcp(b,
false);
422 virtual const Teuchos::RCP<const Vector>
getB(
void ) {
434 virtual Teuchos::RCP<const Teuchos::Comm<int> >
getComm(
void )
const {
446 virtual std::string
name(
void )
const {
466 const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default )
const {
479 const Teuchos::EVerbosityLevel verbLevel = Teuchos::Describable::verbLevel_default )
const{
492 virtual void getTiming( Teuchos::ParameterList& timingParameterList )
const{
500 Teuchos::RCP<const Matrix>
A;
501 Teuchos::RCP<Vector>
X;
502 Teuchos::RCP<const Vector>
B;
512 template <
template <
class,
class>
class ConcreteSolver,
513 class ST,
class LO,
class GO,
class NO >
516 typedef Tpetra::CrsMatrix<SC,LO,GO,NO>
Matrix;
517 typedef Tpetra::MultiVector<SC,LO,GO,NO>
Vector;
518 static Teuchos::RCP<Solver<Matrix,Vector> >
520 Teuchos::RCP<Vector> X,
521 Teuchos::RCP<const Vector> B ) {
524 typename MatrixTraits<Matrix>::scalar_t,
525 typename MultiVecAdapter<Vector>::scalar_t
527 > same_scalar_assertion;
528 (void)same_scalar_assertion;
531 return Teuchos::rcp(
new MPVectorSolverAdapter<ST,LO,GO,NO,ConcreteSolver>(A, X, B) );
538 template <
template <
class,
class>
class ConcreteSolver,
539 class ST,
class LO,
class GO,
class NO >
540 struct create_solver_with_supported_type<
542 Tpetra::CrsMatrix<Sacado::MP::Vector<ST>,LO,GO,NO>,
543 Tpetra::MultiVector<Sacado::MP::Vector<ST>,LO,GO,NO> > :
549 template <
template <
class,
class>
class ConcreteSolver,
551 struct solver_supports_scalar<ConcreteSolver,
Sacado::MP::Vector<Storage> > {
555 static const bool value =
556 Meta::if_then_else<Meta::is_same<supported_scalars, Meta::nil_t>::value,
Amesos2 solver adapter for MP::Vector scalar type.
virtual Status & getStatus() const
Returns a reference to this solver's internal status object.
virtual bool matrixShapeOK(void)
Returns true if the solver can handle the matrix shape.
virtual void setX(const Teuchos::RCP< Vector > x)
Sets the LHS vector X.
virtual void setB(const Vector *b)
Sets the RHS vector B using a raw pointer.
Teuchos::RCP< const Vector > B
Tpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > Matrix
Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > FlatGraph
virtual const Teuchos::RCP< const Vector > getB(void)
Returns the vector that is the RHS of the linear system.
virtual void getTiming(Teuchos::ParameterList &timingParameterList) const
Extracts timing information from the current solver.
Teuchos::RCP< FlatVector > flat_X
Teuchos::RCP< const FlatVector > flat_B
Solver< FlatMatrix, FlatVector > FlatSolver
Teuchos::RCP< const Matrix > A
virtual Teuchos::RCP< const Teuchos::ParameterList > getValidParameters(void) const
Return a const parameter list of all of the valid parameters that this->setParameterList(....
virtual void setA(const Matrix *a, EPhase keep_phase=CLEAN)
Sets the matrix A of this solver.
virtual void setA(const Teuchos::RCP< const Matrix > a, EPhase keep_phase=CLEAN)
Sets the matrix A of this solver.
MPVectorSolverAdapter(const Teuchos::RCP< const Matrix > &A_, const Teuchos::RCP< Vector > &X_, const Teuchos::RCP< const Vector > &B_)
Constructor.
Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > Map
virtual void solve(Vector *XX, const Vector *BB) const
Solve using the given XX and BB (multi)vectors.
Scalar::value_type BaseScalar
Solver< Matrix, Vector > solver_type
virtual type & preOrdering(void)
Pre-orders the matrix.
virtual void setB(const Teuchos::RCP< const Vector > b)
Sets the RHS vector B.
virtual const Teuchos::RCP< Vector > getX(void)
Returns the vector that is the LHS of the linear system.
Tpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > Vector
virtual type & numericFactorization(void)
Performs numeric factorization on the matrix.
Teuchos::RCP< const FlatMatrix > flat_A
Teuchos::RCP< const Map > flat_X_map
virtual std::string name(void) const
Return the name of this solver.
virtual void printTiming(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const
Prints timing information about the current solver.
Teuchos::RCP< const Map > flat_B_map
virtual void setX(Vector *x)
Sets the LHS vector X using a raw pointer.
virtual const Vector * getBRaw(void)
Returns a raw pointer to the RHS of the linear system.
virtual void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const
Tpetra::MultiVector< BaseScalar, LocalOrdinal, GlobalOrdinal, Node > FlatVector
Teuchos::RCP< const FlatGraph > flat_graph
virtual std::string description(void) const
Returns a short description of this Solver.
virtual type & setParameters(const Teuchos::RCP< Teuchos::ParameterList > ¶meterList)
Set/update internal variables and solver options.
virtual Teuchos::RCP< const Teuchos::Comm< int > > getComm(void) const
Returns a pointer to the Teuchos::Comm communicator with this matrix.
virtual Vector * getXRaw(void)
Returns a raw pointer to the LHS of the linear system.
virtual void solve(const Teuchos::Ptr< Vector > XX, const Teuchos::Ptr< const Vector > BB) const
Solve using the given XX and BB (multi)vectors.
virtual type & symbolicFactorization(void)
Performs symbolic factorization on the matrix.
Tpetra::CrsMatrix< BaseScalar, LocalOrdinal, GlobalOrdinal, Node > FlatMatrix
ConcreteSolver< FlatMatrix, FlatVector > FlatConcreteSolver
virtual void solve(void)
Solves (or )
Sacado::MP::Vector< Storage > Scalar
Teuchos::RCP< FlatSolver > flat_solver
KokkosClassic::DefaultNode::DefaultNodeType Node
Stokhos::StandardStorage< int, double > Storage
LO get_mp_vector_size(const Teuchos::RCP< const Tpetra::CrsMatrix< Sacado::MP::Vector< S >, LO, GO, N > > &A=Teuchos::null, const Teuchos::RCP< Tpetra::MultiVector< Sacado::MP::Vector< S >, LO, GO, N > > &X=Teuchos::null, const Teuchos::RCP< const Tpetra::MultiVector< Sacado::MP::Vector< S >, LO, GO, N > > &B=Teuchos::null)
KOKKOS_INLINE_FUNCTION constexpr std::enable_if< is_view_uq_pce< View< T, P... > >::value, unsigned >::type dimension_scalar(const View< T, P... > &view)
Teuchos::RCP< const Tpetra::MultiVector< typename Storage::value_type, LocalOrdinal, GlobalOrdinal, Kokkos::Compat::KokkosDeviceWrapperNode< Device > > > create_flat_vector_view(const Tpetra::MultiVector< Sacado::UQ::PCE< Storage >, LocalOrdinal, GlobalOrdinal, Kokkos::Compat::KokkosDeviceWrapperNode< Device > > &vec, const Teuchos::RCP< const Tpetra::Map< LocalOrdinal, GlobalOrdinal, Kokkos::Compat::KokkosDeviceWrapperNode< Device > > > &flat_map)
Teuchos::RCP< Tpetra::CrsMatrix< typename Storage::value_type, LocalOrdinal, GlobalOrdinal, Kokkos::Compat::KokkosDeviceWrapperNode< Device > > > create_flat_matrix(const Tpetra::CrsMatrix< Sacado::UQ::PCE< Storage >, LocalOrdinal, GlobalOrdinal, Kokkos::Compat::KokkosDeviceWrapperNode< Device > > &mat, const Teuchos::RCP< const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Kokkos::Compat::KokkosDeviceWrapperNode< Device > > > &flat_graph, const Teuchos::RCP< const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Kokkos::Compat::KokkosDeviceWrapperNode< Device > > > &cijk_graph, const CijkType &cijk_dev)
Teuchos::RCP< Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > create_flat_mp_graph(const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > &graph, Teuchos::RCP< const Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > &flat_domain_map, Teuchos::RCP< const Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > &flat_range_map, const LocalOrdinal block_size)
static Teuchos::RCP< Solver< Matrix, Vector > > apply(Teuchos::RCP< const Matrix > A, Teuchos::RCP< Vector > X, Teuchos::RCP< const Vector > B)
Sacado::MP::Vector< ST > SC
Tpetra::CrsMatrix< SC, LO, GO, NO > Matrix
Tpetra::MultiVector< SC, LO, GO, NO > Vector
solver_traits< ConcreteSolver >::supported_scalars supported_scalars
Scalar::value_type BaseScalar
Sacado::MP::Vector< Storage > Scalar