test_scheduler.cpp

Go to the documentation of this file.

#include "gtest/gtest.h"
 
#include "kokkos-execution/graph.hpp"
 
#include "tests/utils/category.hpp"
#include "tests/utils/check_scheduler.hpp"
#include "tests/utils/graph_context.hpp"
#include "tests/utils/sink_receiver.hpp"

 
namespace Tests::GraphImpl {
 
using graph_context_t = Kokkos::Execution::GraphContext<TEST_EXECUTION_SPACE>;
using graph_scheduler_t = Kokkos::Execution::GraphImpl::Scheduler<TEST_EXECUTION_SPACE>;
using graph_schedule_sender_t = typename graph_scheduler_t::Sender;
using graph_schedule_sender_attrs_t = typename graph_schedule_sender_t::Attributes;

static_assert(Tests::Utils::check_scheduler<graph_scheduler_t>());

TEST(TEST_CATEGORY(Scheduler), round_trip_property) {
    const graph_context_t ctx{TEST_EXECUTION_SPACE{}};
    const graph_scheduler_t sch = ctx.get_scheduler();
    ASSERT_EQ(stdexec::get_completion_scheduler<stdexec::set_value_t>(stdexec::get_env(stdexec::schedule(sch))), sch);
}

consteval bool test_scheduler_schedule() {
    static_assert(
        std::same_as<decltype(stdexec::schedule(std::declval<const graph_scheduler_t&>())), graph_schedule_sender_t>);
 
    return true;
}
 
static_assert(test_scheduler_schedule());

consteval bool test_schedule_sender_attrs() {
    static_assert(std::same_as<stdexec::env_of_t<graph_schedule_sender_t>, const graph_schedule_sender_attrs_t&>);
 
    return true;
}
 
static_assert(test_schedule_sender_attrs());

consteval bool test_scheduler_queries() {
    static_assert(std::same_as<
                  stdexec::__query_result_t<graph_scheduler_t, stdexec::get_completion_domain_t<stdexec::set_value_t>>,
                  Kokkos::Execution::GraphImpl::Domain
    >);
 
    static_assert(
        std::same_as<
            stdexec::__query_result_t<graph_scheduler_t, stdexec::get_completion_scheduler_t<stdexec::set_value_t>>,
            graph_scheduler_t
        >);
 
    return true;
}
 
static_assert(test_scheduler_queries());

consteval bool test_schedule_sender_attrs_queries() {
    static_assert(std::same_as<
                  stdexec::__query_result_t<
                      graph_schedule_sender_attrs_t,
                      stdexec::get_completion_scheduler_t<stdexec::set_value_t>
                  >,
                  graph_scheduler_t
    >);
 
    static_assert(std::same_as<
                  stdexec::__query_result_t<
                      graph_schedule_sender_attrs_t,
                      stdexec::get_completion_domain_t<stdexec::set_value_t>
                  >,
                  Kokkos::Execution::GraphImpl::Domain
    >);
 
    return true;
}
 
static_assert(test_schedule_sender_attrs_queries());

consteval bool test_schedule_sender_completion_signatures() {
    constexpr auto completions = stdexec::completion_signatures<stdexec::set_value_t()>{};
 
    static_assert(stdexec::get_completion_signatures<graph_schedule_sender_t>() == completions);
 
    return true;
}
 
static_assert(test_schedule_sender_completion_signatures());

consteval bool test_schedule_sender_opstate() {
    using node_t = Kokkos::Execution::GraphImpl::GraphComposition::node_t<TEST_EXECUTION_SPACE>;
 
    using prop_t = stdexec::prop<Kokkos::Execution::GraphImpl::get_node_t, const node_t&>;
    static_assert(stdexec::__queryable_with<prop_t, Kokkos::Execution::GraphImpl::get_node_t>);
    static_assert(
        std::same_as<stdexec::__query_result_t<prop_t, Kokkos::Execution::GraphImpl::get_node_t>, const node_t&>);

    using rcvr_without_prop_t = Tests::Utils::SinkReceiver;
    static_assert(!stdexec::__queryable_with<rcvr_without_prop_t, Kokkos::Execution::GraphImpl::get_node_t>);
    static_assert(!stdexec::__queryable_with<
                  stdexec::connect_result_t<graph_schedule_sender_t&&, rcvr_without_prop_t>,
                  Kokkos::Execution::GraphImpl::get_node_t
    >);

    struct ReceiverWithProp
        : Tests::Utils::SinkReceiver
        , prop_t { };
 
    using rcvr_with_prop_t = ReceiverWithProp;
    static_assert(stdexec::__queryable_with<rcvr_with_prop_t, Kokkos::Execution::GraphImpl::get_node_t>);
    static_assert(!stdexec::__queryable_with<
                  stdexec::connect_result_t<graph_schedule_sender_t&&, rcvr_with_prop_t>,
                  Kokkos::Execution::GraphImpl::get_node_t
    >);
 
    return true;
}
 
static_assert(test_schedule_sender_opstate());
 
} // namespace Tests::GraphImpl