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。
代码示例:
# required: gpu
import paddle
event = paddle.device.cuda.Event()
event.record()
- query ( ) ¶
查询event的状态。
返回: 一个boolean 变量,用于标识当前event 获取的所有任务是否被完成。
代码示例:
# required: gpu
import paddle
event = paddle.device.cuda.Event()
is_done = event.query()
- synchronize ( ) ¶
等待当前event 完成。
代码示例:
# required: gpu
import paddle
event = paddle.device.cuda.Event()
event.synchronize()