43 #ifndef IFPACK2_CONTAINER_DECL_HPP
44 #define IFPACK2_CONTAINER_DECL_HPP
49 #include "Ifpack2_ConfigDefs.hpp"
50 #include "Tpetra_RowMatrix.hpp"
51 #include "Teuchos_Describable.hpp"
52 #include <Tpetra_Map.hpp>
53 #include <Tpetra_BlockCrsMatrix.hpp>
57 #ifndef DOXYGEN_SHOULD_SKIP_THIS
62 #endif // DOXYGEN_SHOULD_SKIP_THIS
111 template<
class MatrixType>
115 using scalar_type =
typename MatrixType::scalar_type;
116 using local_ordinal_type =
typename MatrixType::local_ordinal_type;
117 using global_ordinal_type =
typename MatrixType::global_ordinal_type;
118 using node_type =
typename MatrixType::node_type;
119 using SC = scalar_type;
120 using LO = local_ordinal_type;
121 using GO = global_ordinal_type;
122 using NO = node_type;
123 using import_type = Tpetra::Import<LO, GO, NO>;
124 using mv_type = Tpetra::MultiVector<SC, LO, GO, NO>;
125 using vector_type = Tpetra::Vector<SC, LO, GO, NO>;
126 using map_type = Tpetra::Map<LO, GO, NO>;
127 using crs_matrix_type = Tpetra::CrsMatrix<SC, LO, GO, NO>;
128 using block_crs_matrix_type = Tpetra::BlockCrsMatrix<SC, LO, GO, NO>;
129 using row_matrix_type = Tpetra::RowMatrix<SC, LO, GO, NO>;
131 static_assert(std::is_same<MatrixType, row_matrix_type>::value,
132 "Ifpack2::Container: Please use MatrixType = Tpetra::RowMatrix.");
135 using ISC =
typename Kokkos::ArithTraits<SC>::val_type;
139 using HostView =
typename mv_type::dual_view_type::t_host;
140 using ConstHostView =
typename HostView::const_type;
190 void getMatDiag()
const;
208 void DoJacobi(ConstHostView X,
HostView Y, SC dampingFactor)
const;
209 void DoOverlappingJacobi(ConstHostView X,
HostView Y, ConstHostView W, SC dampingFactor,
bool nonsymScaling)
const;
210 void DoGaussSeidel(ConstHostView X,
HostView Y,
HostView Y2, SC dampingFactor)
const;
211 void DoSGS(ConstHostView X,
HostView Y,
HostView Y2, SC dampingFactor)
const;
229 apply(ConstHostView X,
260 virtual void applyMV (
const mv_type& X, mv_type& Y)
const;
265 vector_type& W)
const;
267 virtual void clearBlocks();
270 virtual std::ostream&
print (std::ostream& os)
const = 0;
280 SC dampingFactor, LO i)
const;
329 template<
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
342 template<
class MatrixType,
class LocalScalarType>
348 using local_scalar_type = LocalScalarType;
349 using SC =
typename Container<MatrixType>::scalar_type;
350 using LO =
typename Container<MatrixType>::local_ordinal_type;
351 using GO =
typename Container<MatrixType>::global_ordinal_type;
352 using NO =
typename Container<MatrixType>::node_type;
354 using typename Container<MatrixType>::import_type;
355 using typename Container<MatrixType>::row_matrix_type;
356 using typename Container<MatrixType>::crs_matrix_type;
357 using typename Container<MatrixType>::block_crs_matrix_type;
358 using typename Container<MatrixType>::mv_type;
359 using typename Container<MatrixType>::vector_type;
360 using typename Container<MatrixType>::map_type;
363 using LSC = LocalScalarType;
364 using LISC =
typename Kokkos::ArithTraits<LSC>::val_type;
366 using local_mv_type = Tpetra::MultiVector<LSC, LO, GO, NO>;
369 using typename Container<MatrixType>::ConstHostView;
370 using HostViewLocal =
typename local_mv_type::dual_view_type::t_host;
371 using HostSubviewLocal = Kokkos::View<LISC**, Kokkos::LayoutStride, typename HostViewLocal::memory_space>;
372 using ConstHostSubviewLocal = Kokkos::View<const LISC**, Kokkos::LayoutStride, typename HostViewLocal::memory_space>;
374 static_assert(std::is_same<MatrixType, row_matrix_type>::value,
375 "Ifpack2::Container: Please use MatrixType = Tpetra::RowMatrix.");
441 apply(ConstHostView X,
472 void applyMV (
const mv_type& X, mv_type& Y)
const;
477 vector_type& W)
const;
479 virtual void clearBlocks();
482 virtual std::ostream&
print (std::ostream& os)
const = 0;
491 SC dampingFactor, LO i)
const;
504 const LSC beta)
const;
508 mutable HostViewLocal Y_local_;
535 template<
typename Scalar,
typename LocalOrdinal,
typename GlobalOrdinal,
typename Node>
536 struct StridedRowView
539 using LO = LocalOrdinal;
541 using block_crs_matrix_type = Tpetra::BlockCrsMatrix<SC, LO, GlobalOrdinal, Node>;
543 using h_inds_type =
typename block_crs_matrix_type::local_inds_host_view_type;
544 using h_vals_type =
typename block_crs_matrix_type::values_host_view_type;
546 StridedRowView(h_vals_type vals_, h_inds_type inds_,
int blockSize_,
size_t nnz_);
554 SC val(
size_t i)
const;
555 LO ind(
size_t i)
const;
573 template <
class MatrixType>
574 std::ostream& operator<<(std::ostream& os, const Ifpack2::Container<MatrixType>& obj);
582 template<
class MatrixType>
586 static std::string name () {
587 return std::string (
"Ifpack2::Container<") +
591 static std::string concreteName (const ::Ifpack2::Container<MatrixType>&) {
598 #endif // IFPACK2_CONTAINER_HPP
virtual void applyInverseJacobi(const mv_type &, mv_type &, SC dampingFactor, bool, int) const =0
Compute Y := (1 - a) Y + a D^{-1} (X - R*Y).
virtual void apply(ConstHostView X, HostView Y, int blockIndex, Teuchos::ETransp mode=Teuchos::NO_TRANS, SC alpha=Teuchos::ScalarTraits< SC >::one(), SC beta=Teuchos::ScalarTraits< SC >::zero()) const
Compute Y := alpha * M^{-1} X + beta*Y.
Definition: Ifpack2_Container_def.hpp:628
virtual void initialize()=0
Do all set-up operations that only require matrix structure.
void weightedApplyMV(const mv_type &X, mv_type &Y, vector_type &W) const
Wrapper for weightedApply with MVs, used in unit tests (never called by BlockRelaxation) ...
Definition: Ifpack2_Container_def.hpp:553
bool isComputed() const
Whether the container has been successfully computed.
Definition: Ifpack2_Container_def.hpp:170
void DoGSBlock(ConstHostView X, HostView Y, HostView Y2, HostView Resid, SC dampingFactor, LO i) const
Do one step of Gauss-Seidel on block i (used by DoGaussSeidel and DoSGS)
Definition: Ifpack2_Container_def.hpp:286
std::vector< HostSubviewLocal > X_localBlocks_
Views for holding pieces of X corresponding to each block.
Definition: Ifpack2_Container_decl.hpp:523
Teuchos::Array< LO > blockRows_
Local indices of the rows of the input matrix that belong to this block.
Definition: Ifpack2_Container_decl.hpp:294
void setBlockSizes(const Teuchos::Array< Teuchos::Array< LO > > &partitions)
Initialize arrays with information about block sizes.
Definition: Ifpack2_Container_def.hpp:125
bool IsInitialized_
If true, the container has been successfully initialized.
Definition: Ifpack2_Container_decl.hpp:320
Teuchos::Array< LO > blockSizes_
Number of rows in each block.
Definition: Ifpack2_Container_decl.hpp:296
typename Kokkos::ArithTraits< SC >::val_type ISC
Internal representation of Scalar in Kokkos::View.
Definition: Ifpack2_Container_decl.hpp:135
StridedRowView(h_vals_type vals_, h_inds_type inds_, int blockSize_, size_t nnz_)
Constructor for row views (preferred)
Definition: Ifpack2_Container_def.hpp:959
std::vector< HostSubviewLocal > Y_localBlocks_
Views for holding pieces of Y corresponding to each block.
Definition: Ifpack2_Container_decl.hpp:526
virtual void applyMV(const mv_type &X, mv_type &Y) const
Wrapper for apply with MultiVector.
Definition: Ifpack2_Container_def.hpp:176
virtual void initialize()=0
Do all set-up operations that only require matrix structure.
bool pointIndexed_
(If hasBlockCrs_) Whether the blocks are described using sub-block row indices instead of full block ...
Definition: Ifpack2_Container_decl.hpp:314
int numBlocks_
The number of blocks (partitions) in the container.
Definition: Ifpack2_Container_decl.hpp:292
The implementation of the numerical features of Container (Jacobi, Gauss-Seidel, SGS). This class allows a custom scalar type (LocalScalarType) to be used for storing blocks and solving the block systems. Hiding this template parameter from the Container interface simplifies the BlockRelaxation and ContainerFactory classes.
Definition: Ifpack2_Container_decl.hpp:343
virtual void setParameters(const Teuchos::ParameterList &List)
Set parameters, if any.
Definition: Ifpack2_Container_def.hpp:530
bool isInitialized() const
Whether the container has been successfully initialized.
Definition: Ifpack2_Container_def.hpp:165
LO scalarsPerRow_
Definition: Ifpack2_Container_decl.hpp:317
virtual ~Container()
Destructor.
Definition: Ifpack2_Container_def.hpp:114
bool hasBlockCrs_
Whether the input matrix is a BlockCRS matrix.
Definition: Ifpack2_Container_decl.hpp:310
Details::StridedRowView< SC, LO, GO, NO > getInputRowView(LO row) const
View a row of the input matrix.
Definition: Ifpack2_Container_def.hpp:898
HostViewLocal weightedApplyScratch_
Definition: Ifpack2_Container_decl.hpp:520
virtual void compute()=0
Extract the local diagonal blocks and prepare the solver.
InverseType::scalar_type LSC
The internal representation of LocalScalarType in Kokkos::View.
Definition: Ifpack2_Container_decl.hpp:363
Container(const Teuchos::RCP< const row_matrix_type > &matrix, const Teuchos::Array< Teuchos::Array< LO > > &partitions, bool pointIndexed)
Constructor.
Definition: Ifpack2_Container_def.hpp:54
int bcrsBlockSize_
If hasBlockCrs_, the number of DOFs per vertex. Otherwise 1.
Definition: Ifpack2_Container_decl.hpp:312
virtual void apply(ConstHostView X, HostView Y, int blockIndex, Teuchos::ETransp mode=Teuchos::NO_TRANS, SC alpha=Teuchos::ScalarTraits< SC >::one(), SC beta=Teuchos::ScalarTraits< SC >::zero()) const =0
Compute Y := alpha * M^{-1} X + beta*Y.
LO translateRowToCol(LO row)
Definition: Ifpack2_Container_def.hpp:585
HostViewLocal X_local_
Scratch vectors used in apply().
Definition: Ifpack2_Container_decl.hpp:507
Teuchos::RCP< const block_crs_matrix_type > inputBlockMatrix_
The input matrix, dynamic cast to BlockCrsMatrix. May be null.
Definition: Ifpack2_Container_decl.hpp:289
Teuchos::RCP< const row_matrix_type > inputMatrix_
The input matrix to the constructor.
Definition: Ifpack2_Container_decl.hpp:283
virtual void weightedApplyMV(const mv_type &X, mv_type &Y, vector_type &W) const
Wrapper for weightedApply with MultiVector.
Definition: Ifpack2_Container_def.hpp:183
void applyMV(const mv_type &X, mv_type &Y) const
Wrapper for apply with MVs, used in unit tests (never called by BlockRelaxation)
Definition: Ifpack2_Container_def.hpp:544
virtual void applyInverseJacobi(const mv_type &, mv_type &, SC dampingFactor, bool, int) const
Compute Y := (1 - a) Y + a D^{-1} (X - R*Y).
Definition: Ifpack2_Container_def.hpp:534
typename mv_type::dual_view_type::t_host HostView
Definition: Ifpack2_Container_decl.hpp:139
virtual void weightedApply(ConstHostView X, HostView Y, ConstHostView D, int blockIndex, Teuchos::ETransp mode=Teuchos::NO_TRANS, SC alpha=Teuchos::ScalarTraits< SC >::one(), SC beta=Teuchos::ScalarTraits< SC >::zero()) const
Compute Y := alpha * diag(D) * M^{-1} (diag(D) * X) + beta*Y.
Definition: Ifpack2_Container_def.hpp:739
Teuchos::RCP< vector_type > Diag_
Diagonal elements.
Definition: Ifpack2_Container_decl.hpp:300
virtual ~ContainerImpl()
Destructor.
Definition: Ifpack2_Container_def.hpp:526
bool IsParallel_
Whether the problem is distributed across multiple MPI processes.
Definition: Ifpack2_Container_decl.hpp:302
static std::string getName()
Definition: Ifpack2_Container_def.hpp:565
bool IsComputed_
If true, the container has been successfully computed.
Definition: Ifpack2_Container_decl.hpp:322
virtual void weightedApply(ConstHostView X, HostView Y, ConstHostView D, int blockIndex, Teuchos::ETransp mode=Teuchos::NO_TRANS, SC alpha=Teuchos::ScalarTraits< SC >::one(), SC beta=Teuchos::ScalarTraits< SC >::zero()) const =0
Compute Y := alpha * diag(D) * M^{-1} (diag(D) * X) + beta*Y.
virtual void compute()=0
Extract the local diagonal blocks and prepare the solver.
Interface for creating and solving a set of local linear problems.
Definition: Ifpack2_Container_decl.hpp:112
Teuchos::ArrayView< const LO > getBlockRows(int blockIndex) const
Local indices of the rows of the input matrix that belong to this block.
Definition: Ifpack2_Container_def.hpp:118
virtual void extract()=0
Extract the submatrices identified by the local indices set by the constructor. BlockMatrix may be an...
virtual void setParameters(const Teuchos::ParameterList &List)=0
Set parameters, if any.
Structure for read-only views of general matrix rows.
Definition: Ifpack2_Container_decl.hpp:330
virtual std::ostream & print(std::ostream &os) const =0
Print basic information about the container to os.
static std::string getName()
Definition: Ifpack2_Container_def.hpp:190
GO NumGlobalRows_
Number of global rows in input matrix.
Definition: Ifpack2_Container_decl.hpp:306
GO NumGlobalNonzeros_
Number of nonzeros in input matrix.
Definition: Ifpack2_Container_decl.hpp:308
LO NumLocalRows_
Number of local rows in input matrix.
Definition: Ifpack2_Container_decl.hpp:304
Teuchos::Array< LO > blockOffsets_
Starting index in blockRows_ of local row indices for each block.
Definition: Ifpack2_Container_decl.hpp:298
virtual std::ostream & print(std::ostream &os) const =0
Print basic information about the container to os.
static std::string name()
virtual void solveBlock(ConstHostSubviewLocal X, HostSubviewLocal Y, int blockIndex, Teuchos::ETransp mode, const LSC alpha, const LSC beta) const
Definition: Ifpack2_Container_def.hpp:572
Teuchos::RCP< const crs_matrix_type > inputCrsMatrix_
The input matrix, dynamic cast to CrsMatrix. May be null.
Definition: Ifpack2_Container_decl.hpp:286
virtual void DoGSBlock(ConstHostView X, HostView Y, HostView Y2, HostView Resid, SC dampingFactor, LO i) const
Do one step of Gauss-Seidel on block i (used by DoGaussSeidel and DoSGS)
Definition: Ifpack2_Container_def.hpp:196