bulk.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_IMPL_BULK_HPP
#define KOKKOS_EXECUTION_IMPL_BULK_HPP
 
#include "kokkos-execution/stdexec.hpp"
 
#include "kokkos-execution/impl/make_policy.hpp"
 
namespace Kokkos::Execution::Impl {

template <typename Data>
concept has_parallel_policy = requires(const Data& data) {
    { data.__pol_ } -> std::same_as<const stdexec::__bulk::__policy_wrapper<stdexec::parallel_policy>&>;
};

template <typename...>
struct BulkTraits;
 
template <typename Policy, typename Shape, typename Functor>
struct BulkTraits<stdexec::__bulk::__data<Policy, Shape, Functor>> {
    using policy_t = Policy;
    using shape_t = Shape;
    using functor_t = Functor;
};
 
template <typename Data>
using bulk_traits = BulkTraits<std::remove_cvref_t<Data>>;

template <Kokkos::ExecutionSpace Exec, typename Data>
using make_policy_for_bulk_t = Impl::make_policy_t<Exec, Kokkos::IndexType<typename Impl::bulk_traits<Data>::shape_t>>;
 
template <typename Data>
using functor_for_bulk_t = typename Impl::bulk_traits<Data>::functor_t;
 
template <typename Data>
using iter_space_for_bulk_t = Kokkos::dims<1, typename Impl::bulk_traits<Data>::shape_t>;
 
} // namespace Kokkos::Execution::Impl
 
#endif // KOKKOS_EXECUTION_IMPL_BULK_HPP