Tpetra parallel linear algebra  Version of the Day
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Tpetra_Details_packCrsMatrix_decl.hpp
Go to the documentation of this file.
1 // @HEADER
2 // ***********************************************************************
3 //
4 // Tpetra: Templated Linear Algebra Services Package
5 // Copyright (2008) Sandia Corporation
6 //
7 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
8 // the U.S. Government retains certain rights in this software.
9 //
10 // Redistribution and use in source and binary forms, with or without
11 // modification, are permitted provided that the following conditions are
12 // met:
13 //
14 // 1. Redistributions of source code must retain the above copyright
15 // notice, this list of conditions and the following disclaimer.
16 //
17 // 2. Redistributions in binary form must reproduce the above copyright
18 // notice, this list of conditions and the following disclaimer in the
19 // documentation and/or other materials provided with the distribution.
20 //
21 // 3. Neither the name of the Corporation nor the names of the
22 // contributors may be used to endorse or promote products derived from
23 // this software without specific prior written permission.
24 //
25 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36 //
37 // Questions? Contact Michael A. Heroux (maherou@sandia.gov)
38 //
39 // ************************************************************************
40 // @HEADER
41 
42 #ifndef TPETRA_DETAILS_PACKCRSMATRIX_DECL_HPP
43 #define TPETRA_DETAILS_PACKCRSMATRIX_DECL_HPP
44 
45 #include "TpetraCore_config.h"
46 #include "Kokkos_DualView.hpp"
48 #include "Tpetra_CrsMatrix_fwd.hpp"
49 
71 
72 #ifndef DOXYGEN_SHOULD_SKIP_THIS
73 namespace Teuchos {
74 // Forward declaration of Array
75 template<class T> class Array;
76 // Forward declaration of ArrayView
77 template<class T> class ArrayView;
78 } // namespace Teuchos
79 #endif // DOXYGEN_SHOULD_SKIP_THIS
80 
81 namespace Tpetra {
82 
83 //
84 // Users must never rely on anything in the Details namespace.
85 //
86 namespace Details {
87 
119 template<typename ST, typename LO, typename GO, typename NT>
120 void
121 packCrsMatrix (const CrsMatrix<ST, LO, GO, NT>& sourceMatrix,
122  Teuchos::Array<char>& exports,
123  const Teuchos::ArrayView<size_t>& numPacketsPerLID,
124  const Teuchos::ArrayView<const LO>& exportLIDs,
125  size_t& constantNumPackets);
126 
156 template<typename ST, typename LO, typename GO, typename NT>
157 void
158 packCrsMatrixNew (const CrsMatrix<ST, LO, GO, NT>& sourceMatrix,
159  Kokkos::DualView<char*,
161  const Kokkos::DualView<size_t*,
162  typename DistObject<char, LO, GO, NT>::buffer_device_type>& numPacketsPerLID,
163  const Kokkos::DualView<const LO*,
165  size_t& constantNumPackets);
166 
198 template<typename ST, typename LO, typename GO, typename NT>
199 void
200 packCrsMatrixWithOwningPIDs (const CrsMatrix<ST, LO, GO, NT>& sourceMatrix,
201  Kokkos::DualView<char*, typename DistObject<char, LO, GO, NT>::buffer_device_type>& exports_dv,
202  const Teuchos::ArrayView<size_t>& numPacketsPerLID,
203  const Teuchos::ArrayView<const LO>& exportLIDs,
204  const Teuchos::ArrayView<const int>& sourcePIDs,
205  size_t& constantNumPackets);
206 
207 } // namespace Details
208 } // namespace Tpetra
209 
210 #endif // TPETRA_DETAILS_PACKCRSMATRIX_DECL_HPP
void packCrsMatrixNew(const CrsMatrix< ST, LO, GO, NT > &sourceMatrix, Kokkos::DualView< char *, typename DistObject< char, LO, GO, NT >::buffer_device_type > &exports, const Kokkos::DualView< size_t *, typename DistObject< char, LO, GO, NT >::buffer_device_type > &numPacketsPerLID, const Kokkos::DualView< const LO *, typename DistObject< char, LO, GO, NT >::buffer_device_type > &exportLIDs, size_t &constantNumPackets)
Pack specified entries of the given local sparse matrix for communication, for &quot;new&quot; DistObject inter...
Forward declaration of Tpetra::CrsMatrix.
void packCrsMatrix(const CrsMatrix< ST, LO, GO, NT > &sourceMatrix, Teuchos::Array< char > &exports, const Teuchos::ArrayView< size_t > &numPacketsPerLID, const Teuchos::ArrayView< const LO > &exportLIDs, size_t &constantNumPackets)
Pack specified entries of the given local sparse matrix for communication.
Declaration of the Tpetra::DistObject class.
Kokkos::Device< typename device_type::execution_space, buffer_memory_space > buffer_device_type
void packCrsMatrixWithOwningPIDs(const CrsMatrix< ST, LO, GO, NT > &sourceMatrix, Kokkos::DualView< char *, typename DistObject< char, LO, GO, NT >::buffer_device_type > &exports_dv, const Teuchos::ArrayView< size_t > &numPacketsPerLID, const Teuchos::ArrayView< const LO > &exportLIDs, const Teuchos::ArrayView< const int > &sourcePIDs, size_t &constantNumPackets)
Pack specified entries of the given local sparse matrix for communication.