Skip to content

Commit

Permalink
Tpetra: Adding GPU synch example in MultiVector
Browse files Browse the repository at this point in the history
Signed-off-by: Chris Siefert <[email protected]>
  • Loading branch information
csiefer2 committed Nov 12, 2024
1 parent a97ce9f commit f26f1bc
Show file tree
Hide file tree
Showing 2 changed files with 165 additions and 72 deletions.
8 changes: 7 additions & 1 deletion packages/tpetra/core/src/Tpetra_MultiVector_decl.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -562,7 +562,7 @@ namespace Tpetra {
/// around with multiple memory spaces.
MultiVector (const Teuchos::RCP<const map_type>& map,
const typename dual_view_type::t_dev& d_view);

/// \brief Expert mode constructor, that takes a Kokkos::DualView
/// of the MultiVector's data and the "original"
/// Kokkos::DualView of the data, and returns a MultiVector that
Expand Down Expand Up @@ -841,6 +841,9 @@ namespace Tpetra {
/// that one thread might win; it's that the value might get
/// messed up.)
///
/// GPU synchronization semantics: Will sync if last access was on device.
/// Will not sync if last access was on host.
///
/// \param gblRow [in] Global row index of the entry to modify.
/// This <i>must</i> be a valid global row index on the calling
/// process with respect to the MultiVector's Map.
Expand Down Expand Up @@ -878,6 +881,9 @@ namespace Tpetra {
/// that one thread might win; it's that the value might get
/// messed up.)
///
/// GPU synchronization semantics: Will sync if last access was on device.
/// Will not sync if last access was on host.
///
/// \param gblRow [in] Global row index of the entry to modify.
/// This <i>must</i> be a valid global row index on the calling
/// process with respect to the MultiVector's Map.
Expand Down
Loading

0 comments on commit f26f1bc

Please sign in to comment.