昇腾社区首页
中文
注册
开发者
下载

HcclCommInitClusterInfoConfig

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品

Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

针对Atlas 推理系列产品,仅支持此系列产品中的Atlas 300I Duo 推理卡

功能说明

基于rank table初始化HCCL,创建具有特定配置的HCCL通信域。

函数原型

1
HcclResult HcclCommInitClusterInfoConfig(const char *clusterInfo, uint32_t rank, HcclCommConfig *config, HcclComm *comm)

参数说明

参数名

输入/输出

描述

clusterInfo

输入

rank table文件路径(含文件名),作为字符串最大长度为4096字节,含结束符。

rank

输入

本rank的id。

需要注意,此参数取值需要与rank table中对应的“rank_id”字段取值一致。

config

输入

通信域配置项,包括buffer大小、确定性计算开关、通信域名称、通信算法编排展开位置等信息,配置参数需确保在合法值域内,关于HcclCommConfig中的详细参数含义及优先级可参见HcclCommConfig的定义。

需要注意:传入的config必须先调用HcclCommConfigInit对其进行初始化。

comm

输出

将初始化后的通信域以指针的信息回传给调用者。

HcclComm类型的定义可参见HcclComm

返回值

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

约束说明

同一通信域不支持重复初始化。

调用示例

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
// 设备资源初始化
aclInit(NULL);
// rank table配置文件路径
const char *rankTableFile = "/path/to/rank_table.json";
// 指定集合通信操作使用的设备
uint32_t rankSize = 8;
uint32_t devId = 0;
aclrtSetDevice(devId);
// 创建并初始化通信域配置项
HcclCommConfig config;
HcclCommConfigInit(&config);
// 按需修改通信域配置
config.hcclBufferSize = 50;  // 共享数据的缓存区大小,单位为:MB,取值需 >= 1,默认值为:200
std::strcpy(config.hcclCommName, "comm_1");
// 初始化通信域
HcclComm hcclComm;
HcclCommInitClusterInfoConfig(rankTableFile, devId, &config, &hcclComm);
// 销毁通信域
HcclCommDestroy(hcclComm);
// 设备资源去初始化
aclFinalize();