mindspore.mint.distributed.get_backend

查看源文件
mindspore.mint.distributed.get_backend(group=None)[源代码]

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

说明

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

参数：

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

返回：

str，分布式后端的名称。

异常：

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

支持平台：

Ascend

样例：

说明

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

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

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