44 #ifndef TPETRA_DETAILS_GETGRAPHOFFRANKOFFSETS_DECL_HPP
45 #define TPETRA_DETAILS_GETGRAPHOFFRANKOFFSETS_DECL_HPP
52 #include "TpetraCore_config.h"
53 #include "Kokkos_Core.hpp"
54 #include "Kokkos_StaticCrsGraph.hpp"
56 #include <type_traits>
78 class OffsetType =
size_t>
81 typedef typename DeviceType::device_type device_type;
82 typedef OffsetType offset_type;
83 typedef ::Kokkos::View<offset_type*,
85 ::Kokkos::MemoryUnmanaged> offsets_type;
86 typedef ::Kokkos::StaticCrsGraph<LO,
89 void,
size_t> local_graph_type;
90 typedef ::Tpetra::Details::LocalMap<LO, GO, device_type>
local_map_type;
91 typedef ::Kokkos::View<
const typename local_graph_type::size_type*,
94 ::Kokkos::MemoryUnmanaged> row_offsets_type;
97 typedef ::Kokkos::View<
const LO*,
100 ::Kokkos::MemoryUnmanaged> lcl_col_inds_type;
106 const row_offsets_type& ptr,
107 const lcl_col_inds_type& ind);
110 KOKKOS_FUNCTION
void operator() (
const LO& lclRowInd)
const;
113 offsets_type OffRankOffsets_;
116 row_offsets_type ptr_;
117 lcl_col_inds_type ind_;
123 template<
class OffsetsType,
127 getGraphOffRankOffsets (
const OffsetsType& OffRankOffsets,
128 const LclMapType& lclColMap,
129 const LclMapType& lclDomMap,
130 const LclGraphType& lclGraph)
132 typedef typename OffsetsType::non_const_value_type offset_type;
133 typedef typename LclMapType::local_ordinal_type LO;
134 typedef typename LclMapType::global_ordinal_type GO;
135 typedef typename LclMapType::device_type DT;
140 impl_type impl (OffRankOffsets, lclColMap, lclDomMap, lclGraph.row_map, lclGraph.entries);
146 #endif // TPETRA_DETAILS_GETGRAPHOFFRANKOFFSETS_DECL_HPP
Implementation detail of Tpetra::Details::getGraphOffRankOffsets, which in turn is an implementation ...
Declaration and definition of the Tpetra::Map class, an implementation detail of Tpetra::Map.
GetGraphOffRankOffsets(const offsets_type &OffRankOffsets, const local_map_type &lclColMap, const local_map_type &lclDomMap, const row_offsets_type &ptr, const lcl_col_inds_type &ind)
Constructor; also runs the functor.
KOKKOS_FUNCTION void operator()(const LO &lclRowInd) const
Kokkos::parallel_for loop body.