barrier

paddle.distributed. barrier ( group=0 ) [源代码]

同步进程组内的所有进程。

参数

  • group (int,可选) - 工作的进程组编号,默认为 0。

返回

代码示例

import paddle
from paddle.distributed import init_parallel_env

paddle.set_device('gpu:%d'%paddle.distributed.ParallelEnv().dev_id)
init_parallel_env()
paddle.distributed.barrier()