HCOMM获取拓扑层级rank数量

HcclRankGraphGetRankSizeByLayer

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

产品支持情况

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

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

功能说明

给定通信域和拓扑层级编号，返回该层级下本rank所在拓扑实例的rank数量。

以上述拓扑模型为例：

• Layer 0中包含两个拓扑实例，为方便理解，定义拓扑实例ID分别为0和1。
• Layer1中包含1个拓扑实例。

假设在rank 0上调用此接口，如果指定layer层级为0，则此接口返回的rank数量为3；如果指定layer层级为1，则此接口返回的rank数量为6。

函数原型

HcclResult HcclRankGraphGetRankSizeByLayer(HcclComm comm, uint32_t netLayer, uint32_t *rankNum)

参数说明

参数名	输入/输出	描述
comm	输入	通信域。 HcclComm类型的定义如下： typedef void *HcclComm;
netLayer	输入	拓扑层级编号。
rankNum	输出	rank数量。

返回值

HcclResult：接口成功返回HCCL_SUCCESS，其他失败。

约束说明

无

调用示例

以功能说明的拓扑模型为例：

HcclComm comm; uint32_t rankNum; HcclRankGraphGetRankSizeByLayer(comm, 0, &rankNum); // rankNum=3 HcclRankGraphGetRankSizeByLayer(comm, 1, &rankNum); // rankNum=6

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