synchronize

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

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

参数

  • device (paddle.CUDAPlace|paddle.CustomPlace|str) - 希望等待的设备或者设备类型。如果为 None,则为当前期望的 place。默认值为 None。

代码示例

>>> import paddle

>>> paddle.set_device('custom_cpu')
>>> paddle.device.synchronize()
>>> paddle.device.synchronize("custom_cpu:0")
>>> place = paddle.CustomPlace('custom_cpu', 0)
>>> paddle.device.synchronize(place)