event.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_IMPL_HPX_EVENT_HPP
#define KOKKOS_EXECUTION_IMPL_HPX_EVENT_HPP
 
#include "Kokkos_Core.hpp"

 
#if !defined(KOKKOS_ENABLE_IMPL_HPX_ASYNC_DISPATCH)
#    error "This is not supported."
#endif
 
namespace Kokkos::Execution::Impl {
 
template <>
struct HasNonBlockingDispatch<Kokkos::Experimental::HPX> : std::true_type { };
 
template <>
struct Event<Kokkos::Experimental::HPX> {
    mutable std::optional<Kokkos::Experimental::HPX> m_exec = std::nullopt;
    uint64_t m_event_id = invalid_event_id;
 
    Event() = default;
    Event(const Event&) = delete;
    Event& operator=(const Event&) = delete;
    Event(Event&&) noexcept = delete;
    Event& operator=(Event&&) noexcept = delete;
    ~Event() = default;
 
    void record(const Kokkos::Experimental::HPX& exec) {
        m_exec = exec;
    }

    void wait() const {
        if (m_exec.has_value()) {
            auto& instance_data = m_exec->impl_get_instance_data();
 
            {
                const std::lock_guard<hpx::spinlock> lock(instance_data.m_sender_mutex);
 
                auto& sndr = instance_data.m_sender;
                hpx::this_thread::experimental::sync_wait(std::move(sndr));
                sndr = hpx::execution::experimental::unique_any_sender<>(hpx::execution::experimental::just());
            }
 
            m_exec = std::nullopt;
        }
    }
};
 
} // namespace Kokkos::Execution::Impl
 
#endif // KOKKOS_EXECUTION_IMPL_HPX_EVENT_HPP