extents.hpp

Go to the documentation of this file.

#ifndef KOKKOS_UTILS_VIEW_EXTENTS_HPP
#define KOKKOS_UTILS_VIEW_EXTENTS_HPP
 
#include "kokkos-utils/concepts/View.hpp"

 
namespace Kokkos::utils::view
{
 
namespace impl
{

template <concepts::View ViewType, size_t... Ints>
KOKKOS_FUNCTION
constexpr auto extents(const ViewType& view, std::index_sequence<Ints...>)
{
    Kokkos::Array<size_t, sizeof...(Ints)> extents {};
    ((extents[Ints] = view.extent(Ints)), ...);
    return extents;
}
 
} // namespace impl

template <concepts::View ViewType>
KOKKOS_FUNCTION
constexpr auto extents(const ViewType& view)
{
    constexpr auto rank = ViewType::rank();
    constexpr auto dims = std::make_index_sequence<rank>{};
    return impl::extents(view, dims);
}
 
} // namespace Kokkos::utils::view
 
#endif // KOKKOS_UTILS_VIEW_EXTENTS_HPP