Tpetra parallel linear algebra
Version of the Day
|
Abstract interface for local operators (e.g., matrices and preconditioners). More...
#include <Tpetra_LocalCrsMatrixOperator_decl.hpp>
Public Member Functions | |
void | applyImbalancedRows (Kokkos::View< const mv_scalar_type **, array_layout, device_type, Kokkos::MemoryTraits< Kokkos::Unmanaged > > X, Kokkos::View< mv_scalar_type **, array_layout, device_type, Kokkos::MemoryTraits< Kokkos::Unmanaged > > Y, const Teuchos::ETransp mode, const mv_scalar_type alpha, const mv_scalar_type beta) const |
Same behavior as apply() above, except give KokkosKernels a hint to use an SPMV algorithm that can efficiently handle matrices with imbalanced rows. More... | |
Abstract interface for local operators (e.g., matrices and preconditioners).
MultiVectorScalar | The type of the entries of the input and output (multi)vectors. |
MatrixScalar | The type of the entries of the sparse matrix. |
Device | The Kokkos Device type; must be a specialization of Kokkos::Device. |
Definition at line 60 of file Tpetra_LocalCrsMatrixOperator_decl.hpp.
void Tpetra::LocalCrsMatrixOperator< MultiVectorScalar, MatrixScalar, Device >::applyImbalancedRows | ( | Kokkos::View< const mv_scalar_type **, array_layout, device_type, Kokkos::MemoryTraits< Kokkos::Unmanaged > > | X, |
Kokkos::View< mv_scalar_type **, array_layout, device_type, Kokkos::MemoryTraits< Kokkos::Unmanaged > > | Y, | ||
const Teuchos::ETransp | mode, | ||
const mv_scalar_type | alpha, | ||
const mv_scalar_type | beta | ||
) | const |
Same behavior as apply()
above, except give KokkosKernels a hint to use an SPMV algorithm that can efficiently handle matrices with imbalanced rows.
Definition at line 108 of file Tpetra_LocalCrsMatrixOperator_def.hpp.