Tpetra parallel linear algebra  Version of the Day
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Tpetra_Details_unpackCrsMatrixAndCombine_decl.hpp
Go to the documentation of this file.
1 // @HEADER
2 // *****************************************************************************
3 // Tpetra: Templated Linear Algebra Services Package
4 //
5 // Copyright 2008 NTESS and the Tpetra contributors.
6 // SPDX-License-Identifier: BSD-3-Clause
7 // *****************************************************************************
8 // @HEADER
9 
10 #ifndef TPETRA_DETAILS_UNPACKCRSMATRIXANDCOMBINE_DECL_HPP
11 #define TPETRA_DETAILS_UNPACKCRSMATRIXANDCOMBINE_DECL_HPP
12 
13 #include "TpetraCore_config.h"
14 #include "Tpetra_CombineMode.hpp"
15 #include "Kokkos_DualView.hpp"
16 #include "Tpetra_CrsMatrix_fwd.hpp"
17 #include "Tpetra_DistObject_decl.hpp"
18 #include "Tpetra_Details_DefaultTypes.hpp"
19 
40 
41 #ifndef DOXYGEN_SHOULD_SKIP_THIS
42 namespace Teuchos {
43 // Forward declaration of Array
44 template <class T>
45 class Array;
46 // Forward declaration of ArrayView
47 template <class T>
48 class ArrayView;
49 } // namespace Teuchos
50 #endif // DOXYGEN_SHOULD_SKIP_THIS
51 
52 namespace Tpetra {
53 
54 //
55 // Users must never rely on anything in the Details namespace.
56 //
57 namespace Details {
58 
97 template <typename ST, typename LO, typename GO, typename NT>
98 void unpackCrsMatrixAndCombine(const CrsMatrix<ST, LO, GO, NT>& sourceMatrix,
99  const Teuchos::ArrayView<const char>& imports,
100  const Teuchos::ArrayView<const size_t>& numPacketsPerLID,
101  const Teuchos::ArrayView<const LO>& importLIDs,
102  size_t constantNumPackets,
103  CombineMode combineMode);
104 
105 template <typename ST, typename LO, typename GO, typename NT>
106 void unpackCrsMatrixAndCombineNew(
107  const CrsMatrix<ST, LO, GO, NT>& sourceMatrix,
108  Kokkos::DualView<char*,
109  typename DistObject<char, LO, GO, NT>::buffer_device_type>
110  imports,
111  Kokkos::DualView<size_t*,
112  typename DistObject<char, LO, GO, NT>::buffer_device_type>
113  numPacketsPerLID,
114  const Kokkos::DualView<const LO*,
115  typename DistObject<char, LO, GO, NT>::buffer_device_type>& importLIDs,
116  const size_t constantNumPackets,
117  const CombineMode combineMode);
118 
165 //
174 template <typename Scalar, typename LocalOrdinal, typename GlobalOrdinal, typename Node>
175 size_t
176 unpackAndCombineWithOwningPIDsCount(
177  const CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& sourceMatrix,
178  const Teuchos::ArrayView<const LocalOrdinal>& importLIDs,
179  const Teuchos::ArrayView<const char>& imports,
180  const Teuchos::ArrayView<const size_t>& numPacketsPerLID,
181  size_t constantNumPackets,
182  CombineMode combineMode,
183  size_t numSameIDs,
184  const Teuchos::ArrayView<const LocalOrdinal>& permuteToLIDs,
185  const Teuchos::ArrayView<const LocalOrdinal>& permuteFromLIDs);
186 
201 
202 template <typename Scalar, typename LocalOrdinal, typename GlobalOrdinal, typename Node>
203 void unpackAndCombineIntoCrsArrays(
204  const CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& sourceMatrix,
205  const Kokkos::View<LocalOrdinal const*,
206  Kokkos::Device<typename Node::device_type::execution_space,
207  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
208 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
209  ,
210  void, void
211 #endif
212  >,
213  const Kokkos::View<const char*,
214  Kokkos::Device<typename Node::device_type::execution_space,
215  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
216 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
217  ,
218  void, void
219 #endif
220  >,
221  const Kokkos::View<const size_t*,
222  Kokkos::Device<typename Node::device_type::execution_space,
223  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
224 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
225  ,
226  void, void
227 #endif
228  >,
229  const size_t numSameIDs,
230  const Kokkos::View<LocalOrdinal const*,
231  Kokkos::Device<typename Node::device_type::execution_space,
232  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
233 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
234  ,
235  void, void
236 #endif
237  >,
238  const Kokkos::View<LocalOrdinal const*,
239  Kokkos::Device<typename Node::device_type::execution_space,
240  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
241 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
242  ,
243  void, void
244 #endif
245  >,
246  size_t TargetNumRows,
247  const int MyTargetPID,
248  Teuchos::ArrayRCP<size_t>& CRS_rowptr,
249  Teuchos::ArrayRCP<GlobalOrdinal>& CRS_colind,
250  Teuchos::ArrayRCP<Scalar>& CRS_vals,
251  const Teuchos::ArrayView<const int>& SourcePids,
252  Teuchos::Array<int>& TargetPids);
253 
254 template <typename Scalar, typename LocalOrdinal, typename GlobalOrdinal, typename Node>
255 void unpackAndCombineIntoCrsArrays(
256  const CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& sourceMatrix,
257  const Kokkos::View<LocalOrdinal const*,
258  Kokkos::Device<typename Node::device_type::execution_space,
259  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
260 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
261  ,
262  void, void
263 #endif
264  >,
265  const Kokkos::View<const char*,
266  Kokkos::Device<typename Node::device_type::execution_space,
267  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
268 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
269  ,
270  void, void
271 #endif
272  >,
273  const Kokkos::View<const size_t*,
274  Kokkos::Device<typename Node::device_type::execution_space,
275  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
276 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
277  ,
278  void, void
279 #endif
280  >,
281  const size_t numSameIDs,
282  const Kokkos::View<LocalOrdinal const*,
283  Kokkos::Device<typename Node::device_type::execution_space,
284  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
285 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
286  ,
287  void, void
288 #endif
289  >,
290  const Kokkos::View<LocalOrdinal const*,
291  Kokkos::Device<typename Node::device_type::execution_space,
292  Tpetra::Details::DefaultTypes::comm_buffer_memory_space<typename Node::device_type>>
293 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_4
294  ,
295  void, void
296 #endif
297  >,
298  size_t TargetNumRows,
299  const int MyTargetPID,
300  Kokkos::View<size_t*, typename Node::device_type>& /*crs_rowptr_d*/,
301  Kokkos::View<GlobalOrdinal*, typename Node::device_type>& /*crs_colind_d*/,
302  Kokkos::View<typename CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::impl_scalar_type*, typename Node::device_type>& /*crs_vals_d*/,
303  const Teuchos::ArrayView<const int>& SourcePids,
304  Kokkos::View<int*, typename Node::device_type>& /*TargetPids*/);
305 
306 } // namespace Details
307 } // namespace Tpetra
308 
309 #endif // TPETRA_DETAILS_UNPACKCRSMATRIXANDCOMBINE_DECL_HPP
Tpetra::Details::unpackCrsMatrixAndCombine
void unpackCrsMatrixAndCombine(const CrsMatrix< ST, LO, GO, NT > &sourceMatrix, const Teuchos::ArrayView< const char > &imports, const Teuchos::ArrayView< const size_t > &numPacketsPerLID, const Teuchos::ArrayView< const LO > &importLIDs, size_t constantNumPackets, CombineMode combineMode)
Unpack the imported column indices and values, and combine into matrix.
Tpetra_CombineMode.hpp
Declaration of Tpetra::CombineMode enum, and a function for setting a Tpetra::CombineMode parameter i...
Tpetra::Details::unpackAndCombineIntoCrsArrays
void unpackAndCombineIntoCrsArrays(const CrsGraph< LO, GO, NT > &sourceGraph, const Teuchos::ArrayView< const LO > &importLIDs, const Teuchos::ArrayView< const typename CrsGraph< LO, GO, NT >::packet_type > &imports, const Teuchos::ArrayView< const size_t > &numPacketsPerLID, const size_t constantNumPackets, const CombineMode combineMode, const size_t numSameIDs, const Teuchos::ArrayView< const LO > &permuteToLIDs, const Teuchos::ArrayView< const LO > &permuteFromLIDs, size_t TargetNumRows, size_t TargetNumNonzeros, const int MyTargetPID, const Teuchos::ArrayView< size_t > &CRS_rowptr, const Teuchos::ArrayView< GO > &CRS_colind, const Teuchos::ArrayView< const int > &SourcePids, Teuchos::Array< int > &TargetPids)
unpackAndCombineIntoCrsArrays
Tpetra::CombineMode
CombineMode
Rule for combining data in an Import or Export.
Definition: Tpetra_CombineMode.hpp:65
Tpetra_CrsMatrix_fwd.hpp
Forward declaration of Tpetra::CrsMatrix.
Tpetra_DistObject_decl.hpp
Declaration of the Tpetra::DistObject class.
Tpetra::Details::unpackAndCombineWithOwningPIDsCount
size_t unpackAndCombineWithOwningPIDsCount(const CrsGraph< LO, GO, NT > &sourceGraph, const Teuchos::ArrayView< const LO > &importLIDs, const Teuchos::ArrayView< const typename CrsGraph< LO, GO, NT >::packet_type > &imports, const Teuchos::ArrayView< const size_t > &numPacketsPerLID, size_t constantNumPackets, CombineMode combineMode, size_t numSameIDs, const Teuchos::ArrayView< const LO > &permuteToLIDs, const Teuchos::ArrayView< const LO > &permuteFromLIDs)
Special version of Tpetra::Details::unpackCrsGraphAndCombine that also unpacks owning process ranks...
Tpetra::DistObject< char, LO, GO, NT >::buffer_device_type
Kokkos::Device< typename device_type::execution_space, buffer_memory_space > buffer_device_type
Definition: Tpetra_DistObject_decl.hpp:738
Generated by   doxygen 1.8.5