graph_fwd.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_GRAPH_GRAPH_FWD_HPP
#define KOKKOS_EXECUTION_GRAPH_GRAPH_FWD_HPP
 
#include "kokkos-execution/stdexec.hpp"
 
#include "Kokkos_Macros.hpp"
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#else
#    include "Kokkos_Concepts.hpp"
#endif
 
#include "kokkos-execution/graph/diagnostics.hpp"
#include "kokkos-execution/impl/diagnostics.hpp"
 
namespace Kokkos::Execution {
 
template <Kokkos::ExecutionSpace Exec>
struct GraphContext;
 
} // namespace Kokkos::Execution
 
namespace Kokkos::Execution::GraphImpl {
 
template <typename Tag>
struct ApplySenderFor;
 
template <typename Tag>
struct TransformSenderFor;
 
struct Domain;
 
template <Kokkos::ExecutionSpace Exec>
struct Scheduler;
 
template <stdexec::operation_state OpState, Kokkos::ExecutionSpace Exec>
struct GraphOperationStateFor : public std::false_type { };
 
template <typename OpState, typename Exec>
concept graph_operation_state_for = stdexec::operation_state<OpState> && Kokkos::ExecutionSpace<Exec>
                                 && GraphOperationStateFor<OpState, Exec>::value;
 
template <stdexec::operation_state OpState, Kokkos::ExecutionSpace Exec>
struct RemainsOnGraphFor : public std::false_type {
    static constexpr auto diagnose() noexcept {
        return stdexec::__mexception<
            stdexec::_WHAT_(OPERATION_STATE_DOES_NOT_REMAIN_ON_GRAPH),
            Impl::WITH_OPSTATE<OpState>,
            Impl::WITH_EXECUTION_SPACE<Exec>
        >{};
    }
};
 
template <typename OpState, typename Exec>
concept remains_on_graph_for = stdexec::operation_state<OpState> && Kokkos::ExecutionSpace<Exec>
                            && RemainsOnGraphFor<OpState, Exec>::value;

template <typename Exec, typename Sndr, typename Rcvr>
concept connect_result_remains_on_graph_for = Kokkos::ExecutionSpace<Exec> && stdexec::sender<Sndr>
                                           && stdexec::receiver<Rcvr>
                                           && remains_on_graph_for<stdexec::connect_result_t<Sndr, Rcvr>, Exec>;
 
} // namespace Kokkos::Execution::GraphImpl
 
#endif // KOKKOS_EXECUTION_GRAPH_GRAPH_FWD_HPP