Amesos2: fix host/device access issue in vector gather/scatter

Signed-off-by: iyamazaki <[email protected]>
trilinos · Feb 26, 2025 · 6823b2a · 6823b2a
1 parent b1b5160
commit 6823b2a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/amesos2/src/Amesos2_TpetraMultiVecAdapter_def.hpp b/packages/amesos2/src/Amesos2_TpetraMultiVecAdapter_def.hpp
@@ -607,7 +607,8 @@ namespace Amesos2 {
     }
     {
       auto nRows_l = this->mv_->getLocalLength();
-      auto lclMV = this->mv_->getLocalViewHost(Tpetra::Access::ReadOnly);
+      auto lclMV_d = this->mv_->getLocalViewDevice(Tpetra::Access::ReadOnly);
+      auto lclMV = Kokkos::create_mirror_view(lclMV_d);
       for (size_t j=0; j<nCols; j++) {
         // lclMV with ReadOnly = const sendbuf
         const Scalar * sendbuf = reinterpret_cast<const Scalar*> (nRows_l > 0 ? &lclMV(0,j) : lclMV.data());
@@ -642,7 +643,8 @@ namespace Amesos2 {
     auto comm = this->mv_->getMap()->getComm();
     auto myRank = comm->getRank();
     {
-      auto lclMV = this->mv_->getLocalViewHost(Tpetra::Access::OverwriteAll);
+      auto lclMV_d = this->mv_->getLocalViewDevice(Tpetra::Access::OverwriteAll);
+      auto lclMV = Kokkos::create_mirror_view(lclMV_d);
       for (size_t j=0; j<nCols; j++) {
         if (myRank == 0 && this->buf_.extent(0) > 0) {
           for (global_size_t i=0; i<nRows; i++) this->buf_(i, 0) = kokkos_new_view(perm_g2l(i),j);