event.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_IMPL_SYCL_EVENT_HPP
#define KOKKOS_EXECUTION_IMPL_SYCL_EVENT_HPP
 
#include "Kokkos_Macros.hpp"
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#    include "Kokkos_Assert.hpp"
#else
#    include "Kokkos_Core.hpp"
#endif

 
namespace Kokkos::Execution::Impl {
 
template <>
struct HasNonBlockingDispatch<Kokkos::SYCL> : std::true_type { };
 
template <>
struct Event<Kokkos::SYCL> {
    mutable std::optional<sycl::event> m_event = std::nullopt;
    uint64_t m_event_id = invalid_event_id;
 
    Event() = default;
    Event(const Event&) = delete;
    Event& operator=(const Event&) = delete;
    Event(Event&& other) noexcept = delete;
    Event& operator=(Event&& other) noexcept = delete;

    void record(const Kokkos::SYCL& exec) {
        m_event = exec.sycl_queue().ext_oneapi_submit_barrier();
    }
 
    void wait() const {
        if (m_event.has_value()) {
            m_event->wait_and_throw();
            m_event = std::nullopt;
        }
    }
 
    [[nodiscard]]
    constexpr const sycl::event& sycl_event() const noexcept {
        KOKKOS_EXPECTS(m_event.has_value());
        return *m_event;
    }
};
 
template <Kokkos::ExecutionSpace... ExecFrom>
requires(std::same_as<ExecFrom, Kokkos::SYCL> && ...)
void impl_wait(const Kokkos::SYCL& exec, const Event<ExecFrom>&... events) {
    exec.sycl_queue().submit([&](sycl::handler& cgh) { (cgh.depends_on(events.sycl_event()), ...); });
}
 
} // namespace Kokkos::Execution::Impl
 
#endif // KOKKOS_EXECUTION_IMPL_SYCL_EVENT_HPP