dispatch_label.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL_HPP
#define KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL_HPP
 
#include <algorithm>
#include <array>
#if defined(__cpp_impl_reflection)
#    include <meta>
#endif
#include <ranges>
#include <string_view>
 
#include "stdexec/__detail/__config.hpp"
 
#include "Kokkos_Macros.hpp"
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#    if !defined(__cpp_impl_reflection)
#        include "Kokkos_TypeInfo.hpp"
#    endif
#else
#    include "Kokkos_Core.hpp"
#endif
 
#include "kokkos-execution/impl/fixed_string.hpp"
 
namespace Kokkos::Execution::Impl {
 
#if !defined(__cpp_impl_reflection)
namespace detail {
 
template <Kokkos::ExecutionSpace Exec, FixedString Suffix>
consteval auto dispatch_label() {
    constexpr std::string_view prefix = Kokkos::Impl::TypeInfo<Exec>::name();
    std::array<char, prefix.size() + Suffix.size()> buf{};
    std::ranges::copy(Suffix, std::ranges::copy(prefix, buf.begin()).out);
    return buf;
}
 
} // namespace detail

template <Kokkos::ExecutionSpace Exec, FixedString Suffix>
constexpr auto dispatch_label_v = detail::dispatch_label<Exec, Suffix>();

template <Kokkos::ExecutionSpace Exec, FixedString Suffix>
consteval auto dispatch_label()
    STDEXEC_AUTO_RETURN(std::string_view{dispatch_label_v<Exec, Suffix>.data(), dispatch_label_v<Exec, Suffix>.size()})
 
// NOLINTNEXTLINE(cppcoreguidelines-macro-usage)
#    define KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL(_exec_, _suffix_)                                                     \
        Kokkos::Execution::Impl::dispatch_label<KOKKOS_IMPL_STRIP_PARENS(_exec_), _suffix_>()
#else
template <Kokkos::ExecutionSpace Exec>
consteval std::string_view dispatch_label(std::string_view suffix) noexcept {
    return std::define_static_string(std::views::concat(std::meta::display_string_of(^^Exec), suffix));
}
 
#    define KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL(_exec_, _suffix_)                                                     \
        Kokkos::Execution::Impl::dispatch_label<KOKKOS_IMPL_STRIP_PARENS(_exec_)>(_suffix_)
#endif
} // namespace Kokkos::Execution::Impl
 
#endif // KOKKOS_EXECUTION_IMPL_DISPATCH_LABEL_HPP