Skip to content

Commit

Permalink
Merge pull request #1085 from parthenon-hpc-lab/jmm/sparsepack-wishlist
Browse files Browse the repository at this point in the history
[trivial] add get_max_nij to sparse pack
  • Loading branch information
Yurlungur authored May 22, 2024
2 parents f9bb863 + eaa458a commit fbff9e2
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion src/utils/index_split.hpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
//========================================================================================
// (C) (or copyright) 2023. Triad National Security, LLC. All rights reserved.
// (C) (or copyright) 2023-2024. Triad National Security, LLC. All rights reserved.
//
// This program was produced under U.S. Government contract 89233218CNA000001 for Los
// Alamos National Laboratory (LANL), which is operated by Triad National Security, LLC
Expand Down Expand Up @@ -91,8 +91,12 @@ class IndexSplit {
KOKKOS_INLINE_FUNCTION
int get_max_ni() const { return ibe_entire_ + 1; }
// TODO(@jdolence) these overestimate max size...should probably fix
KOKKOS_INLINE_FUNCTION
int get_max_nj() const { return (jbe_entire_ + 1) / njp_ + 1; }
KOKKOS_INLINE_FUNCTION
int get_max_nk() const { return (kbe_entire_ + 1) / nkp_ + 1; }
KOKKOS_INLINE_FUNCTION
int get_max_nij() const { return get_max_ni() * get_max_nj(); }
// inner_size could be used to find the bounds for a loop that is collapsed over
// 1, 2, or 3 dimensions by providing the right starting and stopping indices
template <typename V>
Expand Down

0 comments on commit fbff9e2

Please sign in to comment.