Tpetra: Implementation of CrsGraph::transferAndFillComplete

packages/tpetra/core/src/CMakeLists.txt

@@ -386,5 +386,3 @@ SET_PROPERTY(
 # subdirectory.  That ensures that running "make" will also rerun
 # CMake in order to regenerate Makefiles.
 #
-# Here is another such change.
-#

packages/tpetra/core/src/Tpetra_Details_packCrsGraph.cpp

@@ -0,0 +1,67 @@
+/*
+// @HEADER
+// ***********************************************************************
+//
+//          Tpetra: Templated Linear Algebra Services Package
+//                 Copyright (2008) Sandia Corporation
+//
+// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
+// the U.S. Government retains certain rights in this software.
+//
+// Redistribution and use in source and binary forms, with or without
+// modification, are permitted provided that the following conditions are
+// met:
+//
+// 1. Redistributions of source code must retain the above copyright
+// notice, this list of conditions and the following disclaimer.
+//
+// 2. Redistributions in binary form must reproduce the above copyright
+// notice, this list of conditions and the following disclaimer in the
+// documentation and/or other materials provided with the distribution.
+//
+// 3. Neither the name of the Corporation nor the names of the
+// contributors may be used to endorse or promote products derived from
+// this software without specific prior written permission.
+//
+// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
+// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
+// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+//
+// Questions? Contact Michael A. Heroux ([email protected])
+//
+// ************************************************************************
+// @HEADER
+*/
+
+#include "TpetraCore_config.h"
+
+#if defined(HAVE_TPETRA_EXPLICIT_INSTANTIATION)
+
+// We protect the contents of this file with macros, to assist
+// applications that circumvent Trilinos' build system.  (We do NOT
+// recommend this.)  That way, they can still build this file, but as
+// long as the macros have correct definitions, they won't build
+// anything that's not enabled.
+
+#include "KokkosCompat_ClassicNodeAPI_Wrapper.hpp"
+#include "Tpetra_Details_packCrsGraph_decl.hpp"
+#include "Tpetra_Details_packCrsGraph_def.hpp"
+#include "TpetraCore_ETIHelperMacros.h"
+
+namespace Tpetra {
+
+  TPETRA_ETI_MANGLING_TYPEDEFS()
+
+  TPETRA_INSTANTIATE_LGN( TPETRA_DETAILS_PACKCRSGRAPH_INSTANT )
+
+} // namespace Tpetra
+
+#endif // Whether we should build this specialization

packages/tpetra/core/src/Tpetra_Details_packCrsGraph_decl.hpp

@@ -0,0 +1,220 @@
+// @HEADER
+// ***********************************************************************
+//
+//          Tpetra: Templated Linear Algebra Services Package
+//                 Copyright (2008) Sandia Corporation
+//
+// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
+// the U.S. Government retains certain rights in this software.
+//
+// Redistribution and use in source and binary forms, with or without
+// modification, are permitted provided that the following conditions are
+// met:
+//
+// 1. Redistributions of source code must retain the above copyright
+// notice, this list of conditions and the following disclaimer.
+//
+// 2. Redistributions in binary form must reproduce the above copyright
+// notice, this list of conditions and the following disclaimer in the
+// documentation and/or other materials provided with the distribution.
+//
+// 3. Neither the name of the Corporation nor the names of the
+// contributors may be used to endorse or promote products derived from
+// this software without specific prior written permission.
+//
+// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
+// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
+// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+//
+// Questions? Contact Michael A. Heroux ([email protected])
+//
+// ************************************************************************
+// @HEADER
+
+#ifndef TPETRA_DETAILS_PACKCRSGRAPH_DECL_HPP
+#define TPETRA_DETAILS_PACKCRSGRAPH_DECL_HPP
+
+#include "TpetraCore_config.h"
+#include "Kokkos_DualView.hpp"
+#include "Tpetra_DistObject_decl.hpp"
+
+/// \file Tpetra_Details_packCrsGraph.hpp
+/// \brief Functions for packing the entries of a Tpetra::CrsGraph
+///   for communication, in the case where it is valid to go to the
+///   KokkosSparse::CrsGraph (local sparse graph data structure)
+///   directly.
+/// \warning This file, and its contents, are implementation details
+///   of Tpetra.  The file itself or its contents may disappear or
+///   change at any time.
+///
+/// Data (bytes) describing the row of the CRS graph are "packed"
+/// (concatenated) in to a (view of) packet_type* object in the following order:
+///
+///   1. number of entries (LocalOrdinal)
+///   2. global column indices (GlobalOrdinal)
+///   3. proces IDs (optional, int)
+///
+/// The functions in this file are companions to
+/// Tpetra_Details_unpackCrsGraph.hpp, i.e., Tpetra_Details_unpackCrsGraph.hpp
+/// implements the reverse of the packing order described above to ensure proper
+/// unpacking.
+
+#ifndef DOXYGEN_SHOULD_SKIP_THIS
+namespace Teuchos {
+// Forward declaration of Array
+template<class T> class Array;
+// Forward declaration of ArrayView
+template<class T> class ArrayView;
+} // namespace Teuchos
+#endif // DOXYGEN_SHOULD_SKIP_THIS
+
+namespace Tpetra {
+
+#ifndef DOXYGEN_SHOULD_SKIP_THIS
+// Forward declaration of Distributor
+class Distributor;
+// Forward declaration of CrsGraph
+template<class LO, class GO, class NT>
+class CrsGraph;
+#endif // DOXYGEN_SHOULD_SKIP_THIS
+
+//
+// Users must never rely on anything in the Details namespace.
+//
+namespace Details {
+
+/// \brief Pack specified entries of the given local sparse graph for
+///   communication.
+///
+/// \tparam LO The type of local indices.  See the
+///   documentation of Map for requirements.
+/// \tparam GO The type of global indices.  See the
+///   documentation of Map for requirements.
+/// \tparam NT The Kokkos Node type.  See the documentation of Map
+///   for requirements.
+///
+/// \param sourceGraph [in] the CrsGraph source
+///
+/// \param exports [in/out] Output pack buffer; resized if needed.
+///
+/// \param numPacketsPerLID [out] Entry k gives the number of bytes
+///   packed for row exportLIDs[k] of the local graph.
+///
+/// \param exportLIDs [in] Local indices of the rows to pack.
+///
+/// \param constantNumPackets [out] Setting this to zero tells the caller
+///   to expect a possibly /// different ("nonconstant") number of packets per local index
+///   (i.e., a possibly different number of entries per row).
+///
+/// \param distor [in] The distributor (not used)
+///
+/// This is the public interface to the pack machinery
+/// converts passed Teuchos::ArrayView objects to Kokkos::View objects (and
+/// copies back in to the Teuchos::ArrayView objects, if needed).  When
+/// CrsGraph migrates fully to adopting Kokkos::DualView objects for its storage
+/// of data, this procedure could be bypassed.
+template<typename LO, typename GO, typename NT>
+void
+packCrsGraph (const CrsGraph<LO, GO, NT>& sourceGraph,
+               Teuchos::Array<typename CrsGraph<LO,GO,NT>::packet_type>& exports,
+               const Teuchos::ArrayView<size_t>& numPacketsPerLID,
+               const Teuchos::ArrayView<const LO>& exportLIDs,
+               size_t& constantNumPackets,
+               Distributor& distor);
+
+/// \brief Pack specified entries of the given local sparse graph for
+///   communication, for "new" DistObject interface.
+///
+/// \tparam LO The type of local indices.  This must be the same as
+///   the LocalOrdinal template parameter of Tpetra::CrsGraph.
+/// \tparam GO The type of global indices.  This must be the same as
+///   the GlobalOrdinal template parameter of Tpetra::CrsGraph.
+/// \tparam NT The Node type.  This must be the same as the Node
+///   template parameter of Tpetra::CrsGraph.
+///
+/// \param sourceGraph [in] The "source" graph to pack.
+///
+/// \param exports [in/out] Output pack buffer; resized if needed.
+///
+/// \param numPacketsPerLID [out] On output,
+///   numPacketsPerLID.d_view[k] is the number of bytes packed for row
+///   exportLIDs.d_view[k] of the local graph.
+///
+/// \param exportLIDs [in] Local indices of the rows to pack.
+///
+/// \param constantNumPackets [out] Same as the constantNumPackets
+///   output argument of Tpetra::DistObject::packAndPrepareNew (which
+///   see).
+///
+/// \param distor [in] (Not used.)
+///
+/// This method implements CrsGraph::packNew, and thus
+/// CrsGraph::packAndPrepareNew, for the case where the graph to
+/// pack has a valid KokkosSparse::CrsGraph.
+template<typename LO, typename GO, typename NT>
+void
+packCrsGraphNew (const CrsGraph<LO, GO, NT>& sourceGraph,
+                 Kokkos::DualView<typename CrsGraph<LO,GO,NT>::packet_type*,
+                                  typename CrsGraph<LO,GO,NT>::buffer_device_type>&
+                                  exports,
+                 const Kokkos::DualView<size_t*,
+                                        typename CrsGraph<LO,GO,NT>::buffer_device_type>&
+                                        numPacketsPerLID,
+                 const Kokkos::DualView<const LO*, typename NT::device_type>& exportLIDs,
+                 size_t& constantNumPackets,
+                 Distributor& distor);
+
+/// \brief Pack specified entries of the given local sparse graph for
+///   communication.
+///
+/// \tparam LO The type of local indices.  See the
+///   documentation of Map for requirements.
+/// \tparam GO The type of global indices.  See the
+///   documentation of Map for requirements.
+/// \tparam NT The Kokkos Node type.  See the documentation of Map
+///   for requirements.
+///
+/// \param sourceGraph [in] the CrsGraph source
+///
+/// \param exports [in/out] Output pack buffer; resized if needed.
+///
+/// \param numPacketsPerLID [out] Entry k gives the number of bytes
+///   packed for row exportLIDs[k] of the local graph.
+///
+/// \param exportLIDs [in] Local indices of the rows to pack.
+///
+/// \param constantNumPackets [out] Setting this to zero tells the caller
+///   to expect a possibly /// different ("nonconstant") number of packets per local index
+///   (i.e., a possibly different number of entries per row).
+///
+/// \param distor [in] The distributor (not used)
+///
+/// This is the public interface to the pack machinery
+/// converts passed Teuchos::ArrayView objects to Kokkos::View objects (and
+/// copies back in to the Teuchos::ArrayView objects, if needed).  When
+/// CrsGraph migrates fully to adopting Kokkos::DualView objects for its storage
+/// of data, this procedure could be bypassed.
+template<typename LO, typename GO, typename NT>
+void
+packCrsGraphWithOwningPIDs (const CrsGraph<LO,GO,NT>& sourceGraph,
+                            Kokkos::DualView<typename CrsGraph<LO,GO,NT>::packet_type*,
+                                             typename CrsGraph<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,
+                            Distributor& distor);
+
+} // namespace Details
+} // namespace Tpetra
+
+#endif // TPETRA_DETAILS_PACKCRSGRAPH_DECL_HPP