test_dispatch_label.cpp

Go to the documentation of this file.

#include "gtest/gtest.h"
 
#include "kokkos-execution/impl/dispatch_label.hpp"
#include "kokkos-execution/impl/type_info.hpp"

 
namespace Tests::Impl {
 
constexpr char suffix[] = " with the given suffix";
 
constexpr std::string_view label_of_exec = Kokkos::Execution::Impl::display_string_of<TEST_EXECUTION_SPACE>();
 
constexpr std::string_view label_as_view = KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL(TEST_EXECUTION_SPACE, suffix);
 
#if !defined(__cpp_impl_reflection)
consteval bool test_dispatch_label_v() {
    constexpr auto label_as_array = Kokkos::Execution::Impl::dispatch_label_v<TEST_EXECUTION_SPACE, suffix>;
 
    static_assert(
        std::same_as<decltype(label_as_array), const std::array<char, label_of_exec.size() + sizeof(suffix) - 1>>);
 
    return true;
}
 
static_assert(test_dispatch_label_v());
#endif

consteval bool test_dispatch_label() {
    static_assert(
        std::same_as<decltype(KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL(TEST_EXECUTION_SPACE, suffix)), std::string_view>);
    static_assert(std::same_as<decltype(label_as_view), const std::string_view>);
    static_assert(label_as_view.starts_with(label_of_exec));
    static_assert(label_as_view.ends_with(suffix));
    static_assert(label_as_view.size() == label_of_exec.size() + sizeof(suffix) - 1);
 
    return true;
}
 
static_assert(test_dispatch_label());

consteval bool test_dispatch_label_noexcept() {
    static_assert(noexcept(KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL(TEST_EXECUTION_SPACE, suffix)));
 
    return true;
}
 
static_assert(test_dispatch_label_noexcept());
 
constexpr std::string_view view_a = KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL(TEST_EXECUTION_SPACE, suffix);
constexpr std::string_view view_b = KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL(TEST_EXECUTION_SPACE, suffix);

TEST(DispatchLabelTest, deduplicate) {
    static_assert(static_cast<const void*>(view_a.data()) == static_cast<const void*>(view_b.data()));
 
    ASSERT_EQ(static_cast<const void*>(view_a.data()), static_cast<const void*>(view_b.data()));
}
 
} // namespace Tests::Impl