CANN/hcomm通信域初始化
HcclCommInitRootInfo【免费下载链接】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 推理卡。功能说明根据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);【免费下载链接】hcommHCOMMHuawei Communication是HCCL的通信基础库提供通信域以及通信资源的管理能力。项目地址: https://gitcode.com/cann/hcomm创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2599118.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!