mindspore.mint.distributed.recv_object_list

查看源文件
mindspore.mint.distributed.recv_object_list(object_list, src=0, group=None, device=None)[源代码]

同步接收源的进程的Python对象列表。

说明

警告

本接口会隐式使用pickle模块,该模块是不安全的。核心原因是反序列化时会执行任意代码,攻击者可通过构造恶意对象触发系统命令执行。因此,调用者须自行保障接口使用数据的安全性。

参数:
  • object_list (list[Any]) - 要接收对象的列表。必须提供与发送列表大小相等的列表。

  • src (int,可选) - 表示发送源的全局进程编号。默认值: 0

  • group (str,可选) - 通信组名称。如果为 None ,Ascend平台表示为 "hccl_world_group" 。默认值: None

  • device (str,可选) - 目前这是一个保留参数。默认值: None

返回:

int。若成功,返回 0

异常:
  • TypeError - 如果 src 不是int类型、 group 不是str类型,或者 object_list 不是list类型。

  • ValueError - 如果 src 线程和当前线程号相同。

支持平台:

Ascend

样例:

说明

运行以下样例之前,需要配置好通信环境变量。

针对Ascend设备,推荐使用msrun启动方式,无第三方以及配置文件依赖。详见 msrun启动

>>> from mindspore.mint.distributed import init_process_group, send_object_list, recv_object_list, get_rank
>>> init_process_group()
>>> this_rank = get_rank()
>>> if this_rank == 0:
...     objects = ["foo", 12, {1: 2}]  # any picklable object
...     send_object_list(objects, dst=1)
>>> else:
...     objects = [None, None, None]
...     recv_object_list(objects, src=0)
...     print(objects)
['foo', 12, {1: 2}]