CANN/hcomm通信域初始化
2026/5/10 0:59:38 网站建设 项目流程

HcclCommInitRootInfo

【免费下载链接】hcommHCOMM(Huawei Communication)是HCCL的通信基础库,提供通信域以及通信资源的管理能力。项目地址: https://gitcode.com/cann/hcomm

产品支持情况

  • Ascend 950PR/Ascend 950DT:支持
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
  • Atlas 推理系列产品:支持
  • Atlas 训练系列产品:支持

[!NOTE]说明 针对Atlas A2 训练系列产品/Atlas A2 推理系列产品,仅支持Atlas 800T A2 训练服务器、Atlas 900 A2 PoD 集群基础单元、Atlas 200T A2 Box16 异构子框。 针对Atlas 推理系列产品,仅支持Atlas 300I Duo 推理卡。

功能说明

根据rootInfo初始化HCCL,创建HCCL通信域。

该接口在同一进程内支持多线程并发调用,但仅支持单卡单线程的场景,若是单卡多线程,不支持并发调用。

如下图所示,不支持step0与step1并发调用,需要step0执行结束后,再串行执行step1。

函数原型

HcclResult HcclCommInitRootInfo(uint32_t nRanks, const HcclRootInfo *rootInfo, uint32_t rank, HcclComm *comm)

参数说明

参数名输入/输出描述
nRanks输入集群中的rank数量。
rootInfo输入root rank信息,主要包含root rank的ip、id等信息,由HcclGetRootInfo接口生成。
rank输入本rank的rank id。
comm输出初始化后的通信域指针。
HcclComm类型的定义可参见HcclComm。

返回值

HcclResult:接口成功返回HCCL_SUCCESS,其他失败。

约束说明

同一通信域中所有rank的nRanks、rootInfo均应相同。

调用示例

uint32_t rankSize = 8; uint32_t deviceId = 0; // 生成 root 节点的 rank 标识信息 HcclRootInfo rootInfo; HcclGetRootInfo(&rootInfo); // 初始化通信域 HcclComm hcclComm; HcclCommInitRootInfo(rankSize, &rootInfo, deviceId, &hcclComm); // 销毁通信域 HcclCommDestroy(hcclComm);

【免费下载链接】hcommHCOMM(Huawei Communication)是HCCL的通信基础库,提供通信域以及通信资源的管理能力。项目地址: https://gitcode.com/cann/hcomm

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询