类 faiss::gpu::KernelTimer

class KernelTimer

用于计时内核执行的实用工具类。

公共函数

KernelTimer(cudaStream_t stream = nullptr)

构造函数启动计时器,并将事件添加到当前设备流中

~KernelTimer()

析构函数释放事件资源。

float elapsedMilliseconds()

添加停止事件,然后同步停止事件,以获取在当前流中启动的任何内核的实际 GPU 端内核计时。返回经过的毫秒数。只能调用一次。

私有成员

cudaEvent_t startEvent_
cudaEvent_t stopEvent_
cudaStream_t stream_
bool valid_