WIP: adding missing function to bGrid::Partition

Autodesk · Dec 6, 2023 · 13c51e2 · 13c51e2
1 parent 4760cf5
commit 13c51e2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/libNeonDomain/include/Neon/domain/details/bGrid/bPartition.h b/libNeonDomain/include/Neon/domain/details/bGrid/bPartition.h
@@ -136,6 +136,8 @@ class bPartition
     getDomainSize()
         const -> Neon::index_3d;
 
+    auto mem() const -> T const *;
+
     /**
      * Gets the Idx for in the block view space.
      */

diff --git a/libNeonDomain/include/Neon/domain/details/bGrid/bPartition_imp.h b/libNeonDomain/include/Neon/domain/details/bGrid/bPartition_imp.h
@@ -95,6 +95,12 @@ operator()(const Idx& cell,
     return mMem[helpGetPitch(cell, card)];
 }
 
+    template <typename T, int C, typename SBlock>
+    NEON_CUDA_HOST_DEVICE inline auto bPartition<T, C, SBlock>::
+    mem() const -> T const *{
+        return mMem;
+    }
+
 template <typename T, int C, typename SBlock>
 inline NEON_CUDA_HOST_DEVICE auto bPartition<T, C, SBlock>::
     helpGetPitch(const Idx& idx, int card)