synchronize¶

paddle.device.cuda. synchronize ( device=None ) [源代码] ¶

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

参数¶

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

返回¶

None

代码示例¶

          # required: gpu
import paddle

paddle.device.cuda.synchronize()
paddle.device.cuda.synchronize(0)
paddle.device.cuda.synchronize(paddle.CUDAPlace(0))

         