mindspore.mint.distributed.get_backend

mindspore.mint.distributed.get_backend(group=None)

在指定通信组中获取当前分布式后端的名称。

说明

• 后端类型包含 "hccl" 、 "nccl" 和 "mccl" ，当前仅支持 "hccl" 和 "mccl" 。

• 当前仅支持PyNative模式，不支持Graph模式。

参数：

• group (str，可选) - 通信组名称，通常由 mindspore.mint.distributed.new_group() 方法创建。默认值： None ，即Ascend平台表示为 "hccl_world_group" 。

返回：

str，分布式后端的名称。

异常：

• TypeError - 参数 group 不是字符串。

支持平台：

Ascend

样例：

说明

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

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

>>> import mindspore as ms
>>> from mindspore.mint.distributed import init_process_group, get_backend
>>> ms.set_device(device_target="Ascend")
>>> init_process_group()
>>> backend = get_backend()
>>> print("backend is: ", backend)
backend is: hccl