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

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