check_rcvr_env.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_TESTS_UTILS_CHECK_RCVR_ENV_HPP
#define KOKKOS_EXECUTION_TESTS_UTILS_CHECK_RCVR_ENV_HPP
 
#include "kokkos-execution/stdexec.hpp"
 
#include "kokkos-execution/impl/attributes.hpp"
#include "kokkos-execution/impl/completion_signatures.hpp"
 
namespace Tests::Utils {
 
template <typename ExpectedEnv, typename Sndr>
struct CheckRcvrEnvSender {
    using sender_concept = stdexec::sender_tag;
 
    Sndr sndr;
 
    KOKKOS_EXECUTION_COMPL_SIGS_KEEP(CheckRcvrEnvSender)
 
    template <stdexec::receiver Rcvr>
    constexpr auto connect(Rcvr rcvr) && noexcept(stdexec::__nothrow_connectable<Sndr&&, Rcvr&&>) {
        static_assert(std::same_as<ExpectedEnv, stdexec::env_of_t<Rcvr>>);
        return stdexec::connect(std::forward<Sndr>(sndr), std::move(rcvr));
    }
 
    template <stdexec::receiver Rcvr>
    constexpr auto connect(Rcvr rcvr) const & noexcept(stdexec::__nothrow_connectable<Sndr const &, Rcvr&&>) {
        static_assert(std::same_as<ExpectedEnv, stdexec::env_of_t<Rcvr>>);
        return stdexec::connect(sndr, std::move(rcvr));
    }
 
    KOKKOS_EXECUTION_IMPL_FORWARDING_ATTRIBUTES_GET_ENV(Sndr, sndr)
};

template <typename ExpectedEnv>
struct check_rcvr_env_t {
    [[nodiscard]]
    constexpr auto operator()() const noexcept {
        return stdexec::__closure(*this);
    }
 
    template <stdexec::sender Sndr>
    [[nodiscard]]
    constexpr auto operator()(Sndr&& sndr) const {
        return CheckRcvrEnvSender<ExpectedEnv, Sndr>{std::forward<Sndr>(sndr)};
    }
};
 
template <typename ExpectedEnv>
inline constexpr check_rcvr_env_t<ExpectedEnv> check_rcvr_env{};
 
} // namespace Tests::Utils
 
#endif // KOKKOS_EXECUTION_TESTS_UTILS_CHECK_RCVR_ENV_HPP