dependency.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_IMPL_DEPENDENCY_HPP
#define KOKKOS_EXECUTION_IMPL_DEPENDENCY_HPP
 
#include "Kokkos_Core.hpp"
 
#include "kokkos-execution/impl/dispatch_label.hpp"
#include "kokkos-execution/impl/event.hpp"
 
namespace Kokkos::Execution::Impl {
 
template <Kokkos::ExecutionSpace Exec>
struct HasExecWaitEvent : std::false_type { };

template <typename Exec>
concept has_exec_wait_event = HasExecWaitEvent<Exec>::value;

template <Kokkos::ExecutionSpace ExecTo, Kokkos::ExecutionSpace ExecFrom>
struct Dependency {
    Dependency(const ExecTo&, const ExecFrom& exec_from) {
        exec_from.fence(std::string(Impl::dispatch_label<ExecFrom, ": dependency">()));
    }
};
 
template <Kokkos::ExecutionSpace Exec>
requires(!has_exec_wait_event<Exec>)
struct Dependency<Exec, Exec> {
    Dependency(const Exec& exec_to, const Exec& exec_from) {
        if (exec_from != exec_to) {
            exec_from.fence(std::string(Impl::dispatch_label<Exec, ": dependency">()));
        }
    }
};
 
template <Kokkos::ExecutionSpace Exec>
requires has_exec_wait_event<Exec>
struct [[nodiscard]]
Dependency<Exec, Exec> {
    Event<Exec> event{};

    Dependency(const Exec& exec_to, const Exec& exec_from) {
        if (exec_from != exec_to) {
            record(event, exec_from);
            wait(exec_to, event);
        }
    }
};
 
} // namespace Kokkos::Execution::Impl
 
#if defined(KOKKOS_ENABLE_CUDA)
#    include "kokkos-execution/impl/Cuda/dependency.hpp"
#endif
#if defined(KOKKOS_ENABLE_HIP)
#    include "kokkos-execution/impl/HIP/dependency.hpp"
#endif
#if defined(KOKKOS_ENABLE_HPX)
#    include "kokkos-execution/impl/HPX/dependency.hpp"
#endif
#if defined(KOKKOS_ENABLE_SYCL)
#    include "kokkos-execution/impl/SYCL/dependency.hpp"
#endif
 
#endif // KOKKOS_EXECUTION_IMPL_DEPENDENCY_HPP