get_graph.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_GRAPH_GET_GRAPH_HPP
#define KOKKOS_EXECUTION_GRAPH_GET_GRAPH_HPP
 
#include "kokkos-execution/stdexec.hpp"
 
#include "kokkos-execution/graph/get_node.hpp"
 
namespace Kokkos::Execution::GraphImpl {

struct get_graph_t : public stdexec::__query<get_graph_t> { };
 
inline constexpr get_graph_t get_graph{};
 
#if defined(KOKKOS_COMPILER_GNU)                                                                                       \
    && (KOKKOS_COMPILER_GNU == 1420 || KOKKOS_COMPILER_GNU == 1520 || KOKKOS_COMPILER_GNU == 1601)
template <typename... Booleans>
struct gcc_mor_helper {
    static constexpr bool value = (Booleans::value || ...);
};
 
template <typename... Booleans>
static constexpr bool gcc_mor_v = gcc_mor_helper<Booleans...>::value;
#endif

struct GraphComposition {
    struct Attach { };

    struct Create { };

    template <typename... Queryables>
    using policy_t = std::conditional_t<
#if defined(KOKKOS_COMPILER_GNU)                                                                                       \
    && (KOKKOS_COMPILER_GNU == 1420 || KOKKOS_COMPILER_GNU == 1520 || KOKKOS_COMPILER_GNU == 1601)
        gcc_mor_v<stdexec::__mbool<stdexec::__queryable_with<Queryables, get_node_t>>...>,
#else
        (stdexec::__queryable_with<Queryables, get_node_t> || ...),
#endif
        Attach,
        Create
    >;
 
    template <typename Exec, typename... Queryables>
    struct node_helper_t;
 
    template <typename Exec, typename... Queryables>
    requires std::same_as<policy_t<Queryables...>, Create>
    struct node_helper_t<Exec, Queryables...> {
        using type = typename Kokkos::Experimental::Graph<Exec>::root_t;
    };
 
    template <typename Exec, typename FirstQueryable, typename... RestOfQueryables>
    requires(
        std::same_as<policy_t<FirstQueryable, RestOfQueryables...>, Attach>
        && stdexec::__queryable_with<FirstQueryable, get_node_t>)
    struct node_helper_t<Exec, FirstQueryable, RestOfQueryables...> {
        using type = stdexec::__query_result_t<FirstQueryable, get_node_t>;
    };
 
    template <typename Exec, typename FirstQueryable, typename... RestOfQueryables>
    requires(std::same_as<policy_t<FirstQueryable, RestOfQueryables...>, Attach>)
    struct node_helper_t<Exec, FirstQueryable, RestOfQueryables...> {
        using type = typename node_helper_t<Exec, RestOfQueryables...>::type;
    };

    template <typename Exec, typename... Queryables>
    using node_t = typename node_helper_t<Exec, Queryables...>::type;
};
 
} // namespace Kokkos::Execution::GraphImpl
 
#endif // KOKKOS_EXECUTION_GRAPH_GET_GRAPH_HPP