cupy.cuda.Event#

class cupy.cuda.Event(block=False, disable_timing=False, interprocess=False)[source]#

CUDA 事件,CUDA 流的同步点。

此类以 RAII(资源获取即初始化)方式管理 CUDA 事件句柄,即当 Event 实例被 GC(垃圾回收器)销毁时,其句柄也会被销毁。

参数
  • block (bool) – 如果为 True,则事件在 synchronize() 方法处阻塞。

  • disable_timing (bool) – 如果为 True,则事件不准备计时数据。

  • interprocess (bool) – 如果为 True,则事件可以传递给其他进程。

变量

~Event.ptr (intptr_t) – 原始事件句柄。

方法

record(self, stream=None)#

将事件记录到流中。

参数

stream (cupy.cuda.Stream) – 用于记录事件的 CUDA 流。默认为空流。

synchronize(self)#

将所有设备工作同步到事件。

如果事件创建为阻塞事件,它还会阻塞 CPU 线程直到事件完成。

__eq__(value, /)#

返回 self==value。

__ne__(value, /)#

返回 self!=value。

__lt__(value, /)#

返回 self<value。

__le__(value, /)#

返回 self<=value。

__gt__(value, /)#

返回 self>value。

__ge__(value, /)#

返回 self>=value。

属性

done#

如果事件完成,则为 True。