test_inline_scheduler.cpp

Go to the documentation of this file.

#include "gtest/gtest.h"
 
#include <thread>
 
#include "kokkos-execution/utils/ignore_warnings.hpp"
PRAGMA_DIAGNOSTIC_PUSH
KOKKOS_EXECUTION_STDEXEC_PRAGMA_DIAGNOSTIC_IGNORED
#include "exec/completion_behavior.hpp"
#include "exec/static_thread_pool.hpp"
PRAGMA_DIAGNOSTIC_POP
 
#include "kokkos-execution/stdexec.hpp"
 
#include "tests/utils/functors/store_thread_id.hpp"
#include "tests/utils/return_thread_id.hpp"

 
namespace Tests {

consteval bool test_completion_behavior() {
    static_assert(
        exec::get_completion_behavior<stdexec::set_value_t, stdexec::schedule_result_t<stdexec::inline_scheduler>>()
        == exec::completion_behavior::inline_completion);
 
    return true;
}
 
static_assert(test_completion_behavior());

TEST(inline_scheduler, main_thread) {
    std::thread::id tid;
 
    stdexec::sync_wait(stdexec::schedule(stdexec::inline_scheduler{}) | THEN_STORE_THREAD_ID(&tid));
 
    ASSERT_EQ(tid, std::this_thread::get_id());
}

TEST(inline_scheduler, pool_thread) {
    experimental::execution::static_thread_pool pool{1};
 
    const auto [tid_pool] = stdexec::sync_wait(stdexec::schedule(pool.get_scheduler()) | THEN_RETURN_THREAD_ID).value();
 
    ASSERT_NE(tid_pool, std::this_thread::get_id());
 
    std::thread::id tid;
 
    stdexec::sync_wait(
        stdexec::schedule(pool.get_scheduler()) | stdexec::continues_on(stdexec::inline_scheduler{})
        | THEN_STORE_THREAD_ID(&tid));
 
    ASSERT_EQ(tid, tid_pool);
}
 
} // namespace Tests