Skip to content

Commit

Permalink
Merge branch 'develop' into lroberts36/add-fine-variables
Browse files Browse the repository at this point in the history
  • Loading branch information
lroberts36 authored May 22, 2024
2 parents 46cbb5e + fbff9e2 commit 22bfe57
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 22bfe57

Please sign in to comment.