mindspore.communication

集合通信接口。

注意,集合通信接口需要先配置好通信环境变量。

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

mindspore.communication.GlobalComm

GlobalComm 是一个存储通信信息的全局类。

mindspore.communication.init

初始化通信服务需要的分布式后端,例如 HCCLNCCLMCCL 服务。

mindspore.communication.release

释放分布式资源,例如 HCCLNCCLMCCL 服务。

mindspore.communication.create_group

创建用户自定义的通信组实例。

mindspore.communication.destroy_group

注销用户通信组。

mindspore.communication.get_comm_name

获取指定通讯组的通讯器名称。

mindspore.communication.get_group_size

获取指定通信组实例的rank_size。

mindspore.communication.get_group_rank_from_world_rank

由通信集群中的全局设备序号获取指定用户通信组中的rank ID。

mindspore.communication.get_local_rank

获取指定通信组中当前设备的本地设备序号。

mindspore.communication.get_local_rank_size

获取指定通信组的本地设备总数。

mindspore.communication.get_process_group_ranks

获取指定通信组中的进程编号,并以列表方式返回。

mindspore.communication.get_rank

在指定通信组中获取当前的设备序号。

mindspore.communication.get_world_rank_from_group_rank

由指定通信组中的设备序号获取通信集群中的全局设备序号。

mindspore.communication.HCCL_WORLD_COMM_GROUP

"hccl_world_group" 字符串,指的是由HCCL创建的默认通信组。

mindspore.communication.NCCL_WORLD_COMM_GROUP

"nccl_world_group" 字符串,指的是由NCCL创建的默认通信组。

mindspore.communication.MCCL_WORLD_COMM_GROUP

"mccl_world_group" 字符串,指的是由MCCL创建的默认通信组。

mindspore.communication.comm_func

集合通信函数式接口。

mindspore.communication.comm_func.all_gather_into_tensor

mindspore.communication.comm_func.all_gather_into_tensor 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.all_reduce

mindspore.communication.comm_func.all_reduce 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.all_to_all_single_with_output_shape

mindspore.communication.comm_func.all_to_all_single_with_output_shape 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.all_to_all_v_c

mindspore.communication.comm_func.all_to_all_v_c 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.all_to_all_with_output_shape

mindspore.communication.comm_func.all_to_all_with_output_shape 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.barrier

mindspore.communication.comm_func.barrier 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.batch_isend_irecv

mindspore.communication.comm_func.batch_isend_irecv 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.broadcast

mindspore.communication.comm_func.broadcast 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.gather_into_tensor

mindspore.communication.comm_func.gather_into_tensor 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.irecv

mindspore.communication.comm_func.irecv 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.isend

mindspore.communication.comm_func.isend 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.recv

mindspore.communication.comm_func.recv 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.send

mindspore.communication.comm_func.send 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.P2POp

mindspore.communication.comm_func.P2POp 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.reduce

mindspore.communication.comm_func.reduce 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.reduce_scatter_tensor

mindspore.communication.comm_func.reduce_scatter_tensor 从2.9.0版本开始已被弃用,并将在未来版本中被移除。

mindspore.communication.comm_func.scatter_tensor

mindspore.communication.comm_func.scatter_tensor 从2.9.0版本开始已被弃用,并将在未来版本中被移除。