44#ifndef TPETRA_DETAILS_GETDIAGCOPYWITHOUTOFFSETS_DECL_HPP
45#define TPETRA_DETAILS_GETDIAGCOPYWITHOUTOFFSETS_DECL_HPP
56#include "TpetraCore_config.h"
57#include "Kokkos_Core.hpp"
58#include "Kokkos_ArithTraits.hpp"
60#include "Tpetra_RowMatrix_decl.hpp"
79template<
class DiagType,
83 typedef typename LocalMapType::local_ordinal_type LO;
84 typedef typename LocalMapType::global_ordinal_type GO;
85 typedef typename CrsMatrixType::device_type device_type;
86 typedef typename CrsMatrixType::value_type scalar_type;
87 typedef typename CrsMatrixType::size_type offset_type;
103 D_ (
D), rowMap_ (rowMap), colMap_ (colMap), A_ (
A)
112 const LO
INV = Tpetra::Details::OrdinalTraits<LO>::invalid ();
113 const scalar_type
ZERO =
114 Kokkos::Details::ArithTraits<scalar_type>::zero ();
181static typename LocalMapType::local_ordinal_type
187 static_assert (Kokkos::is_view<DiagType>::value,
188 "DiagType must be a Kokkos::View.");
189 static_assert (
static_cast<int> (DiagType::rank) == 1,
190 "DiagType must be a 1-D Kokkos::View.");
191 static_assert (std::is_same<typename DiagType::value_type, typename DiagType::non_const_value_type>::value,
192 "DiagType must be a nonconst Kokkos::View.");
193 typedef typename LocalMapType::local_ordinal_type LO;
194 typedef typename CrsMatrixType::device_type device_type;
195 typedef typename device_type::execution_space execution_space;
196 typedef Kokkos::RangePolicy<execution_space, LO>
policy_type;
198 typedef Kokkos::View<
typename DiagType::non_const_value_type*,
199 typename DiagType::array_layout,
200 typename DiagType::device_type,
201 Kokkos::MemoryUnmanaged> diag_type;
205 const LO
numRows =
static_cast<LO
> (
D.extent (0));
247template<
class SC,
class LO,
class GO,
class NT>
250 const ::Tpetra::RowMatrix<SC, LO, GO, NT>&
A,
Import KokkosSparse::OrdinalTraits, a traits class for "invalid" (flag) values of integer types,...
Declaration of the Tpetra::Vector class.
Struct that holds views of the contents of a CrsMatrix.
Implementation details of Tpetra.
LO getLocalDiagCopyWithoutOffsetsNotFillComplete(::Tpetra::Vector< SC, LO, GO, NT > &diag, const ::Tpetra::RowMatrix< SC, LO, GO, NT > &A, const bool debug=false)
Given a locally indexed, global sparse matrix, extract the matrix's diagonal entries into a Tpetra::V...
static LocalMapType::local_ordinal_type getDiagCopyWithoutOffsets(const DiagType &D, const LocalMapType &rowMap, const LocalMapType &colMap, const CrsMatrixType &A)
Given a locally indexed, local sparse matrix, and corresponding local row and column Maps,...
Namespace Tpetra contains the class and methods constituting the Tpetra library.
@ ZERO
Replace old values with zero.
Functor that implements much of the one-argument overload of Tpetra::CrsMatrix::getLocalDiagCopy,...
CrsMatrixGetDiagCopyFunctor(const DiagType &D, const LocalMapType &rowMap, const LocalMapType &colMap, const CrsMatrixType &A)
Constructor.
LO value_type
The result of the reduction; number of errors.
KOKKOS_FUNCTION void operator()(const LO &lclRowInd, value_type &errCount) const
Operator for Kokkos::parallel_for.