Fix missig update of private var

parthenon-hpc-lab · Nov 7, 2024 · eb44166 · eb44166
1 parent 120ceef
commit eb44166
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/interface/sparse_pack_base.hpp b/src/interface/sparse_pack_base.hpp
@@ -63,7 +63,7 @@ class SparsePackBase {
   using bounds_h_t = typename bounds_t::HostMirror;
   using block_props_t = ParArray2D<int>;
   using block_props_h_t = typename block_props_t::HostMirror;
-  using coords_t = ParArray1D<ParArray0D<Coordinates_t>>;
+  using coords_t = Kokkos::View<ParArray0D<Coordinates_t> *, LayoutWrapper, DevMemSpace>;
 
   // Returns a SparsePackBase object that is either newly created or taken
   // from the cache in pmd. The cache itself handles the all of this logic

diff --git a/src/mesh/meshblock_pack.hpp b/src/mesh/meshblock_pack.hpp
@@ -21,6 +21,7 @@
 #include <utility>
 #include <vector>
 
+#include "Kokkos_Core_fwd.hpp"
 #include "coordinates/coordinates.hpp"
 #include "interface/variable_pack.hpp"
 #include "kokkos_abstraction.hpp"
@@ -38,7 +39,8 @@ class MeshBlockPack {
   using pack_type = T;
 
   MeshBlockPack() = default;
-  MeshBlockPack(const ParArray1D<T> view, const std::array<int, 5> dims)
+  MeshBlockPack(const Kokkos::View<T *, LayoutWrapper, DevMemSpace> view,
+                const std::array<int, 5> dims)
       : v_(view), dims_(dims), ndim_((dims[2] > 1 ? 3 : (dims[1] > 1 ? 2 : 1))) {}
 
   KOKKOS_FORCEINLINE_FUNCTION
@@ -85,7 +87,7 @@ class MeshBlockPack {
   const Coordinates_t &GetCoords(const int i) const { return v_(i).GetCoords(); }
 
  private:
-  ParArray1D<T> v_;
+  Kokkos::View<T *, LayoutWrapper, DevMemSpace> v_;
   std::array<int, 5> dims_;
   int ndim_;
 };