42 #ifndef IFPACK2_BANDEDCONTAINER_DECL_HPP 43 #define IFPACK2_BANDEDCONTAINER_DECL_HPP 48 #include "Ifpack2_Container.hpp" 49 #include "Tpetra_MultiVector.hpp" 50 #include "Tpetra_Map.hpp" 51 #include "Tpetra_RowMatrix.hpp" 52 #include "Teuchos_SerialBandDenseMatrix.hpp" 101 template<
class MatrixType,
class LocalScalarType>
114 using matrix_type = MatrixType;
116 using LSC = LocalScalarType;
120 using typename Container<MatrixType>::SC;
122 using typename Container<MatrixType>::LO;
124 using typename Container<MatrixType>::GO;
126 using typename Container<MatrixType>::NO;
129 using typename ContainerImpl<MatrixType, LSC>::LISC;
131 using typename Container<MatrixType>::mv_type;
132 using typename Container<MatrixType>::map_type;
133 using local_mv_type = Tpetra::MultiVector<LSC, LO, GO, NO>;
134 using typename Container<MatrixType>::vector_type;
135 using typename Container<MatrixType>::import_type;
138 using typename ContainerImpl<MatrixType, LSC>::HostSubviewLocal;
139 using typename ContainerImpl<MatrixType, LSC>::ConstHostSubviewLocal;
140 using typename ContainerImpl<MatrixType,LSC>::block_crs_matrix_type;
141 using HostViewLocal =
typename local_mv_type::dual_view_type::t_host;
143 static_assert(std::is_same<MatrixType,
144 Tpetra::RowMatrix<SC, LO, GO, NO> >::value,
145 "Ifpack2::BandedContainer: Please use MatrixType = Tpetra::RowMatrix.");
155 using typename Container<MatrixType>::row_matrix_type;
173 const Teuchos::Array<Teuchos::Array<LO> >& partitions,
174 const Teuchos::RCP<const import_type>&,
187 virtual void setParameters (
const Teuchos::ParameterList& List)
override;
197 virtual void compute ()
override;
203 void clearBlocks()
override;
212 virtual std::ostream&
print (std::ostream& os)
const override;
223 describe (Teuchos::FancyOStream &out,
224 const Teuchos::EVerbosityLevel verbLevel =
225 Teuchos::Describable::verbLevel_default)
const override;
235 void extract()
override;
251 solveBlock(ConstHostSubviewLocal X,
254 Teuchos::ETransp mode,
256 const LSC beta)
const override;
259 std::vector<Teuchos::SerialBandDenseMatrix<int, LSC> > diagBlocks_;
262 Teuchos::Array<int> ipiv_;
264 Teuchos::Array<LO> kl_;
265 Teuchos::Array<LO> ku_;
268 Teuchos::Array<LSC> scalars_;
271 Teuchos::Array<LO> scalarOffsets_;
276 #endif // IFPACK2_BANDEDCONTAINER_DECL_HPP virtual void initialize() override
Do all set-up operations that only require matrix structure.
Definition: Ifpack2_BandedContainer_def.hpp:208
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 compute() override
Initialize and compute each block.
Definition: Ifpack2_BandedContainer_def.hpp:242
typename Kokkos::Details::ArithTraits< SC >::val_type ISC
Internal representation of Scalar in Kokkos::View.
Definition: Ifpack2_Container_decl.hpp:135
typename mv_type::dual_view_type::t_host HostView
Definition: Ifpack2_Container_decl.hpp:139
static std::string getName()
Get the name of this container type for Details::constructContainer()
Definition: Ifpack2_BandedContainer_def.hpp:582
virtual std::ostream & print(std::ostream &os) const override
Print information about this object to the given output stream.
Definition: Ifpack2_BandedContainer_def.hpp:530
virtual ~BandedContainer()
Destructor (declared virtual for memory safety of derived classes).
Definition: Ifpack2_BandedContainer_def.hpp:79
void computeBandwidth()
Definition: Ifpack2_BandedContainer_def.hpp:101
BandedContainer(const Teuchos::RCP< const row_matrix_type > &matrix, const Teuchos::Array< Teuchos::Array< LO > > &partitions, const Teuchos::RCP< const import_type > &, bool pointIndexed)
Constructor.
Definition: Ifpack2_BandedContainer_def.hpp:62
virtual void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const override
Print the object with some verbosity level to the given FancyOStream.
Definition: Ifpack2_BandedContainer_def.hpp:563
Preconditioners and smoothers for Tpetra sparse matrices.
Definition: Ifpack2_AdditiveSchwarz_decl.hpp:73
virtual void setParameters(const Teuchos::ParameterList &List) override
Set all necessary parameters (super- and sub-diagonal bandwidth)
Definition: Ifpack2_BandedContainer_def.hpp:83
Store and solve a local Banded linear problem.
Definition: Ifpack2_BandedContainer_decl.hpp:102
virtual std::string description() const override
A one-line description of this object.
Definition: Ifpack2_BandedContainer_def.hpp:541