参数

• device (int | str | CUDAPlace | CustomPlace | None, optional) – 指定需要同步的设备。

  • None：同步当前设备上的计算流。

  • int：设备索引，例如 0 表示 cuda:0。

  • str：设备字符串，例如 'cuda:0' 或 'gpu:0'。

  • CUDAPlace：Paddle 的 CUDAPlace 对象。

  • CustomPlace：Paddle 的自定义设备 Place 对象。

返回

无。

代码示例

>>> import paddle

# synchronize the current device
>>> paddle.cuda.synchronize()