test_make_policy.cpp

Go to the documentation of this file.

#include "gtest/gtest.h"
 
#include "kokkos-utils/tests/scoped/ExecutionSpace.hpp"
 
#include "kokkos-execution/impl/make_policy.hpp"

 
namespace Tests::Impl {
 
struct MakePolicyTest
    : public virtual testing::Test
    , public Kokkos::utils::tests::scoped::ExecutionSpace<TEST_EXECUTION_SPACE> { };
 
consteval bool test_make_policy_result_t() {
    static_assert(
        std::same_as<
            Kokkos::Execution::Impl::make_policy_result_t<Kokkos::Execution::Impl::make_policy_t<TEST_EXECUTION_SPACE>>,
            Kokkos::SinglePolicy<TEST_EXECUTION_SPACE>
        >);
 
    static_assert(std::same_as<
                  Kokkos::Execution::Impl::make_policy_result_t<
                      Kokkos::Execution::Impl::make_policy_t<TEST_EXECUTION_SPACE>,
                      Kokkos::dims<1>
                  >,
                  Kokkos::RangePolicy<TEST_EXECUTION_SPACE>
    >);
 
    static_assert(std::same_as<
                  Kokkos::Execution::Impl::make_policy_result_t<
                      Kokkos::Execution::Impl::make_policy_t<TEST_EXECUTION_SPACE, Kokkos::IndexType<char>>,
                      Kokkos::dims<1>
                  >,
                  Kokkos::RangePolicy<TEST_EXECUTION_SPACE, Kokkos::IndexType<char>>
    >);
 
    static_assert(std::same_as<
                  Kokkos::Execution::Impl::make_policy_result_t<
                      Kokkos::Execution::Impl::make_policy_t<TEST_EXECUTION_SPACE>,
                      Kokkos::range_slice<size_t, size_t>
                  >,
                  Kokkos::RangePolicy<TEST_EXECUTION_SPACE>
    >);
 
    static_assert(std::same_as<
                  Kokkos::Execution::Impl::make_policy_result_t<
                      Kokkos::Execution::Impl::make_policy_t<TEST_EXECUTION_SPACE>,
                      Kokkos::RangePolicy<TEST_EXECUTION_SPACE>
                  >,
                  Kokkos::RangePolicy<TEST_EXECUTION_SPACE>
    >);
 
    return true;
}
 
static_assert(test_make_policy_result_t());
 
consteval bool test_make_policy_noexceptness() {
    static_assert(noexcept(Kokkos::Execution::Impl::make_policy_t<TEST_EXECUTION_SPACE>{}()));
 
    return true;
}
 
static_assert(test_make_policy_noexceptness());

TEST_F(MakePolicyTest, execution_space_instance) {
    const auto single_policy = Kokkos::Execution::Impl::make_policy_t<TEST_EXECUTION_SPACE>{}(this->exec);
    static_assert(
        std::same_as<std::remove_const_t<decltype(single_policy)>, Kokkos::SinglePolicy<TEST_EXECUTION_SPACE>>);
 
    const auto policy_dims1 =
        Kokkos::Execution::Impl::make_policy_t<TEST_EXECUTION_SPACE>{}(this->exec, Kokkos::dims<1>{10});
    static_assert(std::same_as<std::remove_const_t<decltype(policy_dims1)>, Kokkos::RangePolicy<TEST_EXECUTION_SPACE>>);
    ASSERT_EQ(policy_dims1.space(), this->exec);
    ASSERT_EQ(policy_dims1.begin(), 0);
    ASSERT_EQ(policy_dims1.end(), 10);
 
    const auto policy_update = Kokkos::Execution::Impl::make_policy_t<TEST_EXECUTION_SPACE>{}(
        this->exec, Kokkos::RangePolicy<TEST_EXECUTION_SPACE>(this->exec, 0, 10));
    ASSERT_EQ(policy_update.space(), this->exec);
}
 
} // namespace Tests::Impl