synchronize¶

paddle.device.cuda. synchronize ( device=None ) [source]

Wait for the compute on the given CUDA device to finish.

Parameters

• device (paddle.CUDAPlace()|int, optional) – The device or the ID of the device.

• None (If device is) – None.

• Default (the device is the current device.) – None.

Examples

# required: gpu
import paddle

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