synchronize

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

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

参数

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

返回

None

代码示例

>>> import paddle
>>> paddle.device.set_device('xpu')
>>> paddle.device.xpu.synchronize()
>>> paddle.device.xpu.synchronize(0)
>>> paddle.device.xpu.synchronize(paddle.XPUPlace(0))