faiss::gpu::CudaEvent 类

class CudaEvent

公共函数

explicit CudaEvent(cudaStream_t stream, bool timer = false)

创建事件并在流中记录它。

CudaEvent(const CudaEvent &event) = delete

CudaEvent(CudaEvent &&event) noexcept

~CudaEvent()

inline cudaEvent_t get()

void streamWaitOnEvent(cudaStream_t stream)

在此流中等待此事件。

void cpuWaitOnEvent()

让 CPU 等待此事件完成。

CudaEvent &operator=(CudaEvent &&event) noexcept

CudaEvent &operator=(CudaEvent &event) = delete

私有成员

cudaEvent_t event_