Skip to content

Commit

Permalink
add TODO for using shmem for indices in multirz
Browse files Browse the repository at this point in the history
  • Loading branch information
josephleekl committed Nov 13, 2024
1 parent bd8b70a commit cf1d66e
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,7 @@ template <class PrecisionT, class FuncT> class applyNCNFunctor {
Kokkos::parallel_for(Kokkos::TeamPolicy(two2N, Kokkos::AUTO, dim),
*this);
}
// TODO: Runtime selection for copying indices to scratch level 0/shmem
KOKKOS_FUNCTION void operator()(const MemberType &teamMember) const {
const std::size_t k = teamMember.league_rank();
const std::size_t offset = Util::parity_2_offset(parity, k);
Expand Down

0 comments on commit cf1d66e

Please sign in to comment.