synchronize

paddle.device.cuda. synchronize ( device=None )

等待给定的 CUDA 设备上的计算完成。

参数

  • device (paddle.CUDAPlace()|int，可选) - 设备或者设备 ID。如果为 None，则为当前的设备。默认值为 None。

返回

None

代码示例

COPY-FROM: paddle.device.cuda.synchronize