test_scheduler.cpp

Go to the documentation of this file.

#include "gtest/gtest.h"
 
#include "kokkos-execution/execution_space.hpp"
 
#include "tests/utils/category.hpp"
#include "tests/utils/check_scheduler.hpp"
#include "tests/utils/execution_space_context.hpp"

 
namespace Tests::ExecutionSpaceImpl {
 
using execution_space_context_t = Kokkos::Execution::ExecutionSpaceContext<TEST_EXECUTION_SPACE>;
using execution_space_scheduler_t = Kokkos::Execution::ExecutionSpaceImpl::Scheduler<TEST_EXECUTION_SPACE>;
using execution_space_schedule_sender_t = typename execution_space_scheduler_t::Sender;
using execution_space_schedule_sender_attrs_t = typename execution_space_schedule_sender_t::Attributes;

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

TEST(TEST_CATEGORY(Scheduler), round_trip_property) {
    const execution_space_context_t ctx{TEST_EXECUTION_SPACE{}};
    const execution_space_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 execution_space_scheduler_t&>())),
                  execution_space_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<execution_space_schedule_sender_t>,
                  const execution_space_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<
                      execution_space_scheduler_t,
                      stdexec::get_completion_domain_t<stdexec::set_value_t>
                  >,
                  Kokkos::Execution::ExecutionSpaceImpl::Domain
    >);
 
    static_assert(std::same_as<
                  stdexec::__query_result_t<
                      execution_space_scheduler_t,
                      stdexec::get_completion_scheduler_t<stdexec::set_value_t>
                  >,
                  execution_space_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<
                      execution_space_schedule_sender_attrs_t,
                      stdexec::get_completion_scheduler_t<stdexec::set_value_t>
                  >,
                  execution_space_scheduler_t
    >);
 
    static_assert(std::same_as<
                  stdexec::__query_result_t<
                      execution_space_schedule_sender_attrs_t,
                      stdexec::get_completion_domain_t<stdexec::set_value_t>
                  >,
                  Kokkos::Execution::ExecutionSpaceImpl::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<execution_space_schedule_sender_t>() == completions);
 
    return true;
}
 
static_assert(test_schedule_sender_completion_signatures());
 
} // namespace Tests::ExecutionSpaceImpl