bulk.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_IMPL_BULK_HPP
#define KOKKOS_EXECUTION_IMPL_BULK_HPP
 
#include "stdexec/execution.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>>;
 
} // namespace Kokkos::Execution::Impl
 
#endif // KOKKOS_EXECUTION_IMPL_BULK_HPP