本章节包含适配后的分布式接口,提供并行计算能力。
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_npu.distributed.reinit_process_group
重新构建processgroup集合通信域。