下载
中文
注册

概述

本章节包含适配后的分布式接口,提供并行计算能力。

表1 Distributed API

API名称

说明

(beta)torch.distributed.is_hccl_available

判断HCCL通信后端是否可用,与torch.distributed.is_nccl_available 类似。

torch.distributed.distributed_c10d._world.default_pg._get_backend(torch.device("npu")).get_hccl_comm_name

从初始化完成的集合通信域中获取集合通信域名字。

(beta)torch.distributed.ProcessGroupHCCL

创建一个ProcessGroupHCCL对象并返回。

(beta)torch.distributed.ProcessGroupHCCL

重新构建processgroup集合通信域。