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"

 
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());
 
} // namespace Tests::GraphImpl