cuda_event.hpp

 
#pragma once
 
#include <mutex>
 
#include <cuda_runtime.h>
 
#include <rmm/cuda_stream_view.hpp>
 
#include <rapidsmpf/error.hpp>
 
namespace rapidsmpf {
 
 
class CudaEvent {
  public:
    CudaEvent(unsigned flags = cudaEventDisableTiming);
 
    static std::shared_ptr<CudaEvent> make_shared_record(
        rmm::cuda_stream_view stream, unsigned flags = cudaEventDisableTiming
    );
 
    ~CudaEvent() noexcept;
 
    CudaEvent(CudaEvent const&) = delete;  
    CudaEvent& operator=(CudaEvent const&) = delete;  
 
    CudaEvent(CudaEvent&& other) noexcept;
 
    CudaEvent& operator=(CudaEvent&& other);
 
    void record(rmm::cuda_stream_view stream);
 
    [[nodiscard]] bool is_ready() const;
 
    void host_wait() const;
 
    void stream_wait(rmm::cuda_stream_view stream) const;
 
    [[nodiscard]] cudaEvent_t const& value() const noexcept;
 
    operator cudaEvent_t const&() const noexcept {
        return value();
    }
 
  private:
    cudaEvent_t event_{};
};
 
 
}  // namespace rapidsmpf