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{};

struct GraphComposition {
    struct Attach { };

    struct Create { };

    template <typename Queryable>
    using policy_t = std::conditional_t<stdexec::__queryable_with<Queryable, get_node_t>, Attach, Create>;
 
    template <typename Exec, typename Queryable>
    struct node_helper_t;
 
    template <typename Exec, typename Queryable>
    requires(std::same_as<policy_t<Queryable>, Create>)
    struct node_helper_t<Exec, Queryable> {
        using type = typename Kokkos::Experimental::Graph<Exec>::root_t;
    };
 
    template <typename Exec, typename Queryable>
    requires(std::same_as<policy_t<Queryable>, Attach>)
    struct node_helper_t<Exec, Queryable> {
        using type = stdexec::__query_result_t<Queryable, get_node_t>;
    };

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