Merge pull request #13714 from trilinos/tpetra/copy-and-permute-3

Tpetra: copy and permute improvements
trilinos · Feb 15, 2025 · fcf1459 · fcf1459
2 parents 984b943 + 6b87bc4
commit fcf1459
Show file tree

Hide file tree

Showing 7 changed files with 790 additions and 330 deletions.
diff --git a/packages/tpetra/core/src/Tpetra_CrsGraph_decl.hpp b/packages/tpetra/core/src/Tpetra_CrsGraph_decl.hpp
@@ -1194,6 +1194,26 @@ namespace Tpetra {
        buffer_device_type>& permuteFromLIDs,
      const CombineMode CM) override;
 
+
+    void
+    insertGlobalIndicesDevice
+    (const CrsGraph<LocalOrdinal,GlobalOrdinal,Node>& srcCrsGraph,
+     CrsGraph<LocalOrdinal,GlobalOrdinal,Node>& tgtCrsGraph,
+     const Kokkos::DualView<const local_ordinal_type*, buffer_device_type>& permuteToLIDs,
+     const Kokkos::DualView<const local_ordinal_type*, buffer_device_type>& permuteFromLIDs,
+     LocalOrdinal loopEnd);
+
+    void
+    copyAndPermuteImpl
+    (const row_graph_type& source,
+     row_graph_type& target,
+     const size_t numSameIDs,
+     const Kokkos::DualView<const local_ordinal_type*,
+       buffer_device_type>& permuteToLIDs,
+     const Kokkos::DualView<const local_ordinal_type*,
+       buffer_device_type>& permuteFromLIDs,
+     const CombineMode CM);
+
     using padding_type = Details::CrsPadding<
       local_ordinal_type, global_ordinal_type>;