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. If device is None, the device is the current device. Default: None.

Examples

>>> 
>>> import paddle

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