From ff98194e39e0b0069ee20d203cbc7db14306bf84 Mon Sep 17 00:00:00 2001 From: Philip Fackler Date: Fri, 6 Sep 2024 14:53:34 -0500 Subject: [PATCH] Update Kokkos::Array interface --- include/plsm/detail/KokkosExtension.h | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/include/plsm/detail/KokkosExtension.h b/include/plsm/detail/KokkosExtension.h index b041436..70ecb56 100644 --- a/include/plsm/detail/KokkosExtension.h +++ b/include/plsm/detail/KokkosExtension.h @@ -12,10 +12,10 @@ namespace Kokkos * @brief Provide begin/end pairs so that Kokkos::Array can be used in a * range-based for loop */ -template +template KOKKOS_INLINE_FUNCTION auto -begin(Array& a) noexcept +begin(Array& a) noexcept { return a.data(); } @@ -24,10 +24,10 @@ begin(Array& a) noexcept * @brief Provide begin/end pairs so that Kokkos::Array can be used in a * range-based for loop */ -template +template KOKKOS_INLINE_FUNCTION auto -end(Array& a) noexcept +end(Array& a) noexcept { return a.data() + static_cast(N); } @@ -36,10 +36,10 @@ end(Array& a) noexcept * @brief Provide begin/end pairs so that Kokkos::Array can be used in a * range-based for loop */ -template +template KOKKOS_INLINE_FUNCTION auto -begin(const Array& a) noexcept +begin(const Array& a) noexcept { return a.data(); } @@ -48,10 +48,10 @@ begin(const Array& a) noexcept * @brief Provide begin/end pairs so that Kokkos::Array can be used in a * range-based for loop */ -template +template KOKKOS_INLINE_FUNCTION auto -end(const Array& a) noexcept +end(const Array& a) noexcept { return a.data() + static_cast(N); }