test_check_scheduler_type.cpp

Go to the documentation of this file.

#include "gtest/gtest.h"
 
#include "kokkos-execution/utils/ignore_warnings.hpp"
PRAGMA_DIAGNOSTIC_PUSH
KOKKOS_EXECUTION_STDEXEC_PRAGMA_DIAGNOSTIC_IGNORED
#include "exec/split.hpp"
#include "exec/static_thread_pool.hpp"
PRAGMA_DIAGNOSTIC_POP
 
#include "kokkos-execution/stdexec.hpp"
 
#include "tests/utils/check_scheduler_type.hpp"
#include "tests/utils/functors/show_thread_id.hpp"

 
namespace Tests {
 
using run_loop_scheduler_t = stdexec::run_loop::scheduler;
using static_thread_pool_scheduler_t = experimental::execution::_pool_::_static_thread_pool::scheduler;

TEST(check_scheduler, default) {
    stdexec::sync_wait(
        stdexec::just() | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>()
        | THEN_SHOW_THREAD_ID);
}

TEST(check_scheduler, static_thread_pool) {
    experimental::execution::static_thread_pool pool{1};
 
    auto chain = stdexec::schedule(pool.get_scheduler())
               | Tests::Utils::check_scheduler_type<stdexec::set_value_t, static_thread_pool_scheduler_t>()
               | THEN_SHOW_THREAD_ID;
 
    stdexec::sync_wait(std::move(chain)); // NOLINT(performance-move-const-arg)
}

TEST(check_scheduler, split_when_all_no_forward) {
    experimental::execution::static_thread_pool pool{1};
 
    auto fork = stdexec::schedule(pool.get_scheduler())
              | Tests::Utils::check_scheduler_type<stdexec::set_value_t, static_thread_pool_scheduler_t>()
              | THEN_SHOW_THREAD_ID | experimental::execution::split();
 
    auto chain = stdexec::when_all(
        fork | THEN_SHOW_THREAD_ID | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>(),
        fork | THEN_SHOW_THREAD_ID | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>());
 
    stdexec::sync_wait(
        std::move(chain) | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>());
}

TEST(check_scheduler, split_transfer_when_all_no_forward) {
    ::exec::static_thread_pool pool{1};
 
    auto fork = stdexec::schedule(pool.get_scheduler())
              | Tests::Utils::check_scheduler_type<stdexec::set_value_t, static_thread_pool_scheduler_t>()
              | THEN_SHOW_THREAD_ID | experimental::execution::split();
 
    auto chain = stdexec::transfer_when_all(
        pool.get_scheduler(),
        fork | THEN_SHOW_THREAD_ID | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>(),
        fork | THEN_SHOW_THREAD_ID | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>());
 
    stdexec::sync_wait(
        std::move(chain) | Tests::Utils::check_scheduler_type<stdexec::set_value_t, static_thread_pool_scheduler_t>());
}

TEST(check_scheduler, multiple_splits) {
    ::exec::static_thread_pool pool{1};
 
    auto fork_A = stdexec::schedule(pool.get_scheduler())
                | Tests::Utils::check_scheduler_type<stdexec::set_value_t, static_thread_pool_scheduler_t>()
                | THEN_SHOW_THREAD_ID | experimental::execution::split();
 
    auto chain_A_branch_a = fork_A | THEN_SHOW_THREAD_ID;
    auto chain_A_branch_b = std::move(fork_A) | THEN_SHOW_THREAD_ID;
 
    auto chain_A = stdexec::when_all(std::move(chain_A_branch_a), std::move(chain_A_branch_b)) | THEN_SHOW_THREAD_ID;
 
    auto fork_B = std::move(chain_A) | stdexec::continues_on(pool.get_scheduler())
                | Tests::Utils::check_scheduler_type<stdexec::set_value_t, static_thread_pool_scheduler_t>()
                | experimental::execution::split();
 
    auto chain_B_branch_a = fork_B | THEN_SHOW_THREAD_ID
                          | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>();
    auto chain_B_branch_b = std::move(fork_B) | THEN_SHOW_THREAD_ID
                          | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>();
 
    auto chain_B = stdexec::when_all(std::move(chain_B_branch_a), std::move(chain_B_branch_b)) | THEN_SHOW_THREAD_ID
                 | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>();
 
    stdexec::sync_wait(
        std::move(chain_B) | Tests::Utils::check_scheduler_type<stdexec::set_value_t, run_loop_scheduler_t>());
}
 
} // namespace Tests