get_world_size

paddle.distributed. get_world_size ( group=None )

返回指定通信组中的进程数,如果没有指定通信组,则默认使用全局通信组。

参数

  • group (Group,可选) - 指定想在得到哪个通信组下的进程数,如果没有指定,默认使用全局通信组。

返回

(int) 返回指定通信组中的进程数,如果当前进程不在该通信组中,则返回-1。

警告

参数 group 仅在动态图模式下支持

代码示例

COPY-FROM: paddle.distributed.get_world_size