test_operation_state.cpp

Go to the documentation of this file.

#include "kokkos-execution/execution_space.hpp"
 
#include "tests/utils/execution_space_context.hpp"
#include "tests/utils/functors/labeled.hpp"
#include "tests/utils/functors/sum_indices.hpp"
#include "tests/utils/sink_receiver.hpp"
#include "tests/utils/stdexec.hpp"

 
namespace Tests::ExecutionSpaceImpl {
 
class OpStateTest : public Tests::Utils::ExecutionSpaceContextTest<TEST_EXECUTION_SPACE> { };

consteval bool test_op_state_traits() {
    using schd_sndr_t = typename OpStateTest::schedule_sender_t;

    using functor_t = Tests::Utils::Functors::SumIndices<typename OpStateTest::view_s_t>;
    using policy_t = Kokkos::RangePolicy<TEST_EXECUTION_SPACE>;
    using clsr_t = Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure<
        TEST_EXECUTION_SPACE,
        Kokkos::Execution::Impl::ParallelForData<functor_t, policy_t, Kokkos::Execution::Impl::Label<std::string_view>>
    >;

    using rcvr_t = Tests::Utils::SinkReceiver;

    using op_state_t = Kokkos::Execution::ExecutionSpaceImpl::OpState<schd_sndr_t, rcvr_t, clsr_t>;

    static_assert(Tests::Utils::operation_state<op_state_t>);

    static_assert(std::derived_from<op_state_t, Kokkos::Execution::Impl::Immovable>);
    static_assert(!std::move_constructible<op_state_t>);
    static_assert(!std::is_move_assignable_v<op_state_t>);
 
    static_assert(!std::copy_constructible<op_state_t>);
    static_assert(!std::is_copy_assignable_v<op_state_t>);
 
    return true;
}
 
static_assert(test_op_state_traits());

constexpr bool test_op_state_passed_by_const_ref() {
    using sndr_t = decltype(Kokkos::Execution::parallel_for(
        stdexec::schedule(std::declval<typename OpStateTest::context_t>().get_scheduler()),
        Kokkos::Execution::label("hello from pfor"),
        Kokkos::RangePolicy<TEST_EXECUTION_SPACE>(0, 10),
        Tests::Utils::Functors::Labeled<'a'>{}));
 
    static_assert(!std::is_const_v<sndr_t>);

    using op_state_from_sndr_const_ref_t = stdexec::connect_result_t<const sndr_t&, Tests::Utils::SinkReceiver>;
 
    static_assert(std::same_as<
                  op_state_from_sndr_const_ref_t,
                  Kokkos::Execution::ExecutionSpaceImpl::OpState<
                      const typename OpStateTest::schedule_sender_t&,
                      Tests::Utils::SinkReceiver,
                      Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure<
                          TEST_EXECUTION_SPACE,
                          Kokkos::Execution::Impl::ParallelForData<
                              Tests::Utils::Functors::Labeled<'a'>,
                              Kokkos::RangePolicy<TEST_EXECUTION_SPACE>,
                              Kokkos::Execution::Impl::Label<std::string_view>
                          >
                      >
                  >
    >);
 
    return true;
}
 
static_assert(test_op_state_passed_by_const_ref());

TEST_F(OpStateTest, construct_query_and_start) {
    constexpr size_t size = 10;
 
    const view_s_t witness(Kokkos::view_alloc(exec, "witness - shared space"));
 
    const context_t esc{exec};
 
    using functor_t = Tests::Utils::Functors::SumIndices<view_s_t>;
 
    Kokkos::Execution::Impl::ParallelForData<
        functor_t,
        Kokkos::RangePolicy<TEST_EXECUTION_SPACE>,
        Kokkos::Execution::Impl::Label<std::string>
    >
        pfor_data{{"hello from pfor"}, functor_t{.data = witness}, Kokkos::RangePolicy(exec, 2, size)};
    Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure clsr{.state = &esc.m_state, .data = std::move(pfor_data)};
 
    auto op_state = Kokkos::Execution::ExecutionSpaceImpl::OpState{
        stdexec::schedule(esc.get_scheduler()), Tests::Utils::SinkReceiver{}, std::move(clsr)};
 
    ASSERT_EQ(Kokkos::Execution::Impl::get_exec(op_state).get(), exec);
 
    op_state.start();
    exec.fence();
 
    ASSERT_EQ(witness(), size / 2 * (size - 1) - 1);
}

consteval bool test_op_state_flattened_from_two() {
    using sndr_t = decltype(Kokkos::Execution::parallel_for(
        Kokkos::Execution::parallel_for(
            stdexec::schedule(std::declval<typename OpStateTest::context_t>().get_scheduler()),
            "hello from pfor",
            Kokkos::RangePolicy<TEST_EXECUTION_SPACE>(0, 10),
            Tests::Utils::Functors::Labeled<'a'>{}),
        "hello again from pfor",
        Kokkos::RangePolicy<TEST_EXECUTION_SPACE>(0, 10),
        Tests::Utils::Functors::Labeled<'b'>{}));
 
    using op_state_t = stdexec::connect_result_t<sndr_t&&, Tests::Utils::SinkReceiver>;
 
    using clsr_0_t = Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure<
        TEST_EXECUTION_SPACE,
        Kokkos::Execution::Impl::ParallelForData<
            Tests::Utils::Functors::Labeled<'a'>,
            Kokkos::RangePolicy<TEST_EXECUTION_SPACE>,
            Kokkos::Execution::Impl::Label<std::string>
        >
    >;
    using clsr_1_t = Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure<
        TEST_EXECUTION_SPACE,
        Kokkos::Execution::Impl::ParallelForData<
            Tests::Utils::Functors::Labeled<'b'>,
            Kokkos::RangePolicy<TEST_EXECUTION_SPACE>,
            Kokkos::Execution::Impl::Label<std::string>
        >
    >;
 
    static_assert(std::same_as<
                  op_state_t,
                  Kokkos::Execution::ExecutionSpaceImpl::OpState<
                      typename OpStateTest::schedule_sender_t,
                      Tests::Utils::SinkReceiver,
                      clsr_0_t,
                      clsr_1_t
                  >
    >);
 
    static_assert(!std::is_nothrow_constructible_v<op_state_t, sndr_t, Tests::Utils::SinkReceiver, clsr_0_t, clsr_1_t>);
 
    static_assert(stdexec::__tuple_size_v<typename op_state_t::closures_t> == 2);
 
    return true;
}
 
static_assert(test_op_state_flattened_from_two());

consteval bool test_op_state_flattened_from_three() {
    using sndr_t = decltype(Kokkos::Execution::parallel_for(
        Kokkos::Execution::parallel_for(
            Kokkos::Execution::parallel_for(
                stdexec::schedule(std::declval<typename OpStateTest::context_t>().get_scheduler()),
                "hello from pfor",
                Kokkos::RangePolicy<TEST_EXECUTION_SPACE>(0, 10),
                Tests::Utils::Functors::Labeled<'a'>{}),
            "hello again from pfor",
            Kokkos::RangePolicy<TEST_EXECUTION_SPACE>(0, 10),
            Tests::Utils::Functors::Labeled<'b'>{}),
        "hello one more time from pfor",
        Kokkos::RangePolicy<TEST_EXECUTION_SPACE>(0, 10),
        Tests::Utils::Functors::Labeled<'c'>{}));
 
    using op_state_t = stdexec::connect_result_t<sndr_t&&, Tests::Utils::SinkReceiver>;
 
    using clsr_0_t = Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure<
        TEST_EXECUTION_SPACE,
        Kokkos::Execution::Impl::ParallelForData<
            Tests::Utils::Functors::Labeled<'a'>,
            Kokkos::RangePolicy<TEST_EXECUTION_SPACE>,
            Kokkos::Execution::Impl::Label<std::string>
        >
    >;
    using clsr_1_t = Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure<
        TEST_EXECUTION_SPACE,
        Kokkos::Execution::Impl::ParallelForData<
            Tests::Utils::Functors::Labeled<'b'>,
            Kokkos::RangePolicy<TEST_EXECUTION_SPACE>,
            Kokkos::Execution::Impl::Label<std::string>
        >
    >;
    using clsr_2_t = Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure<
        TEST_EXECUTION_SPACE,
        Kokkos::Execution::Impl::ParallelForData<
            Tests::Utils::Functors::Labeled<'c'>,
            Kokkos::RangePolicy<TEST_EXECUTION_SPACE>,
            Kokkos::Execution::Impl::Label<std::string>
        >
    >;
 
    static_assert(std::same_as<
                  op_state_t,
                  Kokkos::Execution::ExecutionSpaceImpl::OpState<
                      typename OpStateTest::schedule_sender_t,
                      Tests::Utils::SinkReceiver,
                      clsr_0_t,
                      clsr_1_t,
                      clsr_2_t
                  >
    >);
 
    static_assert(
        !std::is_nothrow_constructible_v<op_state_t, sndr_t, Tests::Utils::SinkReceiver, clsr_0_t, clsr_1_t, clsr_2_t>);
 
    static_assert(stdexec::__tuple_size_v<typename op_state_t::closures_t> == 3);
 
    return true;
}
 
static_assert(test_op_state_flattened_from_three());

consteval bool test_op_state_flattened_from_three_mixed_tags() {
    using sndr_t = decltype(stdexec::then(
        stdexec::bulk(
            Kokkos::Execution::parallel_for(
                stdexec::schedule(std::declval<typename OpStateTest::context_t>().get_scheduler()),
                Kokkos::RangePolicy<TEST_EXECUTION_SPACE, Kokkos::IndexType<size_t>>(0, 10),
                Tests::Utils::Functors::Labeled<'a'>{}),
            stdexec::par,
            char(10),
            Tests::Utils::Functors::Labeled<'b'>{}),
        Tests::Utils::Functors::Labeled<'c'>{}));
 
    using op_state_t = stdexec::connect_result_t<sndr_t&&, Tests::Utils::SinkReceiver>;
 
    using clsr_0_t = Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure<
        TEST_EXECUTION_SPACE,
        Kokkos::Execution::Impl::ParallelForData<
            Tests::Utils::Functors::Labeled<'a'>,
            Kokkos::RangePolicy<TEST_EXECUTION_SPACE, Kokkos::IndexType<size_t>>
        >
    >;
    using clsr_1_t = Kokkos::Execution::ExecutionSpaceImpl::ParallelForClosure<
        TEST_EXECUTION_SPACE,
        Kokkos::Execution::Impl::ParallelForData<
            Tests::Utils::Functors::Labeled<'b'>,
            Kokkos::RangePolicy<TEST_EXECUTION_SPACE, Kokkos::IndexType<char>>,
            Kokkos::Execution::Impl::Label<std::string_view>
        >
    >;
    using clsr_2_t = Kokkos::Execution::ExecutionSpaceImpl::ThenClosure<
        TEST_EXECUTION_SPACE,
        std::string_view,
        Tests::Utils::Functors::Labeled<'c'>
    >;
 
    static_assert(std::same_as<
                  op_state_t,
                  Kokkos::Execution::ExecutionSpaceImpl::OpState<
                      typename OpStateTest::schedule_sender_t,
                      Tests::Utils::SinkReceiver,
                      clsr_0_t,
                      clsr_1_t,
                      clsr_2_t
                  >
    >);
 
    static_assert(
        !std::is_nothrow_constructible_v<op_state_t, sndr_t, Tests::Utils::SinkReceiver, clsr_0_t, clsr_1_t, clsr_2_t>);
 
    static_assert(stdexec::__tuple_size_v<typename op_state_t::closures_t> == 3);
 
    return true;
}
 
static_assert(test_op_state_flattened_from_three_mixed_tags());
 
} // namespace Tests::ExecutionSpaceImpl