iteration_space.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_IMPL_ITERATION_SPACE_HPP
#define KOKKOS_EXECUTION_IMPL_ITERATION_SPACE_HPP
 
#include <type_traits>
 
#include "Kokkos_Macros.hpp"
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
import kokkos.execution.mdspan;
#else
#    include "Kokkos_Core.hpp"
#endif
 
namespace Kokkos::Execution::Impl {
 
template <typename T>
concept extents = Kokkos::detail::impl_is_extents_v<T>;
 
template <typename T>
concept range_slice = Kokkos::detail::is_range_slice<T>::value;
 
template <typename T>
concept range_slice_with_default_stride = range_slice<T>
                                       && Kokkos::detail::is_constant_wrapper<decltype(std::declval<T>().stride)>
                                       && requires(const T& rgsl) { rgsl.stride == 1; };

template <typename T>
concept iteration_space = (extents<T> && requires { T::rank() == 1; }) || range_slice_with_default_stride<T>;
 
template <typename T>
concept execution_policy_or_iteration_space = Kokkos::ExecutionPolicy<T> || iteration_space<T>;

template <std::integral T>
constexpr auto make_iter_space(const T work_count) noexcept -> Kokkos::dims<1, T> {
    return Kokkos::dims<1, T>{work_count};
}
 
} // namespace Kokkos::Execution::Impl
 
#endif // KOKKOS_EXECUTION_IMPL_ITERATION_SPACE_HPP