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

HcclCommConfigInit

产品支持情况

产品

是否支持

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

Atlas A2 训练系列产品

Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

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

功能说明

初始化通信域配置项,并将其中的可配置参数设为默认值。

函数原型

1
inline void HcclCommConfigInit(HcclCommConfig *config)

参数说明

参数名

输入/输出

描述

config

输出

需要初始化的通信域配置项。

HcclCommConfig类型的定义可参见HcclCommConfig

返回值

约束说明

调用示例

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
uint32_t rankSize = 8;
uint32_t deviceId = 0;
// 生成 root 节点的 rank 标识信息
HcclRootInfo rootInfo;
HcclGetRootInfo(&rootInfo);

// 创建并初始化通信域配置项
HcclCommConfig config;
HcclCommConfigInit(&config);
// 按需修改通信域配置
config.hcclBufferSize = 1024;  // 共享数据的缓存区大小,单位为:MB,取值需 >= 1,默认值为:200
config.hcclDeterministic = 1;  // 开启归约类通信算子的确定性计算,默认值为:0,表示关闭确定性计算功能
std::strcpy(config.hcclCommName, "comm_1");
// 初始化集合通信域
HcclComm hcclComm;
HCCLCHECK(HcclCommInitRootInfoConfig(rankSize, &rootInfo, deviceId, &config, &hcclComm));

// 销毁通信域
HcclCommDestroy(hcclComm);