broadcast_object_list

paddle.distributed. broadcast_object_list ( object_list, src, group=None ) [源代码]

将一个 picklable 对象发送到每个进程。 过程与 broadcast 类似，但可以传入自定义的 python 对象。

注解

该 API 只支持动态图模式。

参数

  • object_list (List[Any]) - 在目标进程上为待广播的 object 列表，在其他进程上为用于接收广播结果的 object 列表。

  • src (int) - 目标进程的 rank，该进程传入的 object 列表将被发送到其他进程上。

  • group (Group，可选) - 执行该操作的进程组实例（通过 new_group 创建）。默认为 None，即使用全局默认进程组。

返回

无返回值。

代码示例

>>> import paddle.distributed as dist

>>> dist.init_parallel_env()
>>> if dist.get_rank() == 0:
...     object_list = [{"foo": [1, 2, 3]}]
>>> else:
...     object_list = [{"bar": [4, 5, 6]}]
>>> dist.broadcast_object_list(object_list, src=1)
>>> print(object_list)
>>> # [{"bar": [4, 5, 6]}] (2 GPUs)