Event

class paddle.device.cuda. Event ( enable_timing=False, blocking=False, interprocess=False )

CUDA event 的句柄。

参数

  • enable_timing (bool,可选) - Event 是否需要统计时间。默认值为 False。

  • blocking (bool,可选) - wait()函数是否被阻塞。默认值为 False。

  • interprocess (bool,可选) - Event 是否能在进程间共享。默认值为 False。

返回

None

代码示例

# required: gpu
import paddle
event = paddle.device.cuda.Event()

方法

record(CUDAStream=None)

记录 event 到给定的 stream。

参数

  • stream (CUDAStream,可选) - CUDA stream 的句柄。如果为 None,stream 为当前的 stream。默认值为 False。

代码示例

COPY-FROM: paddle.device.cuda.Event.record

query()

查询 event 的状态。

返回

一个 boolean 变量,用于标识当前 event 获取的所有任务是否被完成。

代码示例

COPY-FROM: paddle.device.cuda.Event.query

synchronize()

等待当前 event 完成。

代码示例

COPY-FROM: paddle.device.cuda.Event.synchronize