sender_concepts.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_EXECUTION_SPACE_SENDER_CONCEPTS_HPP
#define KOKKOS_EXECUTION_EXECUTION_SPACE_SENDER_CONCEPTS_HPP
 
#include "kokkos-execution/stdexec.hpp"
 
#include "kokkos-execution/execution_space/execution_space_fwd.hpp"
 
namespace Kokkos::Execution::ExecutionSpaceImpl {

template <typename Sndr, typename... Env>
concept execution_space_completing_sender =
    stdexec::sender<Sndr>
    && stdexec::__is_instance_of<
        std::invoke_result_t<stdexec::get_completion_scheduler_t<stdexec::set_value_t>, stdexec::env_of_t<Sndr>, Env...>,
        Scheduler
    >;
 
template <typename Sndr, typename... Env>
struct ExecutionSpaceCompletingSender : public std::bool_constant<execution_space_completing_sender<Sndr, Env...>> { };
 
struct CANNOT_DISPATCH_THIS_ALGORITHM_TO_THE_EXECUTION_SPACE_SCHEDULER;
struct BECAUSE_THERE_IS_NO_EXECUTION_SPACE_SCHEDULER_IN_THE_ENVIRONMENT;

template <typename Tag, typename Sndr, typename... Env>
auto no_execution_space_scheduler_in_env() noexcept {
    return stdexec::__not_a_sender<
        stdexec::_WHAT_(CANNOT_DISPATCH_THIS_ALGORITHM_TO_THE_EXECUTION_SPACE_SCHEDULER),
        stdexec::_WHY_(BECAUSE_THERE_IS_NO_EXECUTION_SPACE_SCHEDULER_IN_THE_ENVIRONMENT),
        stdexec::_WHERE_(stdexec::_IN_ALGORITHM_, Tag),
        stdexec::_WITH_PRETTY_SENDER_<Sndr>,
        stdexec::_WITH_ENVIRONMENT_(Env...)
    >{};
}
 
} // namespace Kokkos::Execution::ExecutionSpaceImpl
 
#endif // KOKKOS_EXECUTION_EXECUTION_SPACE_SENDER_CONCEPTS_HPP