CANN/hcomm HCCL通信管理器API

HcclGetRootInfo【免费下载链接】hcommHCOMMHuawei 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推理卡。功能说明此接口需要在HCCL初始化接口HcclCommInitRootInfo或HcclCommInitRootInfoConfig前调用仅需在root节点调用用于生成root节点的rank标识信息HcclRootInfo。该接口需要和初始化接口HcclCommInitRootInfo或HcclCommInitRootInfoConfig接口配对使用不能单独使用。该接口支持单线程循环调用即开发者可在一个for循环中通过“指定不同的Device 调用此接口”从而实现在一个线程中获取不同设备的rootInfo信息。假设一个AI Server中有8张卡8张卡分成4个通信域每个通信域中的两张卡之间通信如下图所示。获取rootInfo信息并进行集合通信初始化的流程如下图所示。首先在一个线程中通过切换Device创建4个rootInfo信息并存入一个长度为4的数组中。rootInfo信息获取完成后起4个线程分别调用HcclCommInitRootInfo或者HcclCommInitRootInfoConfig接口上图中以HcclCommInitRootInfo接口示意根据不同的rootInfo信息进行通信域初始化。多机集合通信场景调用HcclGetRootInfo前可以进行如下操作非必选配置环境变量HCCL_IF_IP或HCCL_SOCKET_IFNAME指定HCCL的初始化root网卡IP环境变量HCCL_IF_IP的优先级高于HCCL_SOCKET_IFNAME若二者都不配置默认使用网卡名称的字典序升序选择root网卡。配置环境变量HCCL_WHITELIST_DISABLE开启白名单校验并通过HCCL_WHITELIST_FILE指定通信白名单配置文件若不配置默认关闭通信白名单校验。函数原型HcclResult HcclGetRootInfo(HcclRootInfo *rootInfo)参数说明参数名输入/输出描述rootInfo输出本rank的标识信息主要包含device ip、device id等信息。此信息需广播至集群内所有rank用来进行HCCL初始化。HcclRootInfo类型的定义可参见HcclRootInfo。返回值HcclResult接口成功返回HCCL_SUCCESS其他失败。约束说明无调用示例uint32_t rankSize 8; uint32_t deviceId 0; // 生成 root 节点的 rank 标识信息 HcclRootInfo rootInfo; HcclGetRootInfo(rootInfo); // 初始化通信域 HcclComm hcclComm; HcclCommInitRootInfo(rankSize, rootInfo, deviceId, hcclComm); // 销毁通信域 HcclCommDestroy(hcclComm);【免费下载链接】hcommHCOMMHuawei Communication是HCCL的通信基础库提供通信域以及通信资源的管理能力。项目地址: https://gitcode.com/cann/hcomm创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考