CANN/hcomm通信通道创建API

HcommChannelCreate

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

产品支持情况

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

功能说明

该接口为创建通信通道的资源管理接口，基于已创建的网络端点（Endpoint），根据给定的通道描述信息批量创建通信通道，为点对点通信或集合通信提供数据传输的基础设施。

函数原型

HcommResult HcommChannelCreate(EndpointHandle endpointHandle, CommEngine engine, HcommChannelDesc *channelDescs, uint32_t channelNum, ChannelHandle *channels);

参数说明

返回值

HcommResult：接口成功返回0，其他失败。

约束说明

• channelDescs数组长度必须与channelNum参数一致。
• HcommChannelDesc中的remoteEndpoint必须正确填充远端端点信息。
• 当HcommChannelDesc中exchangeAllMems为false时，必须配置memHandles和memHandleNum。
• 当前CommEngine配置为CCU时，仅支持交换1份memHandle。
• 当前CommEngine配置为CCU时，不支持外部配置NotifyNum，默认为8个CCU Notify。
• 支持的通信协议包括：RoCE、UBC_TP、UBC_CTP、UBoE。

调用示例

// 1. 调用 HcclRankGraphGetLinks 获取链路信息 CommLink *linkList = nullptr; uint32_t listSize; CHK_RET(HcclRankGraphGetLinks(comm, netLayer, myRank, rank, &linkList, &listSize)); // 2. 遍历每个 CommLink，填充 HcommChannelDesc uint32_t channelNum = listSize; std::vector<HcommChannelDesc> hcommDescVec(channelNum); HcommChannelDescInit(hcommDescVec.data(), hcommDescVec.size()); for (uint32_t idx = 0; idx < listSize; idx++) { HcommChannelDesc channelDesc; CommLink link = linkList[idx]; // 核心映射：从 CommLink 提取 Endpoint 信息 channelDesc.localEndpoint.protocol = link.srcEndpointDesc.protocol; channelDesc.localEndpoint.commAddr = link.srcEndpointDesc.commAddr; channelDesc.localEndpoint.loc = link.srcEndpointDesc.loc; channelDesc.remoteEndpoint.protocol = link.dstEndpointDesc.protocol; channelDesc.remoteEndpoint.commAddr = link.dstEndpointDesc.commAddr; channelDesc.remoteEndpoint.loc = link.dstEndpointDesc.loc; channelDesc.channelProtocol = link.linkAttr.linkProtocol; channelDesc.notifyNum = NORMAL_NOTIFY_NUM; hcommDescVec[idx] = channelDesc; // socket配置为空 HcommSocket hcommSocket = nullptr; hcommDescVec[idx].socket = hcommSocket; } // 参考 MyRank 下 BatchCreateChannels 获取 EndpointHandle EndpointHandle epHandle = nullptr; ... // 3. 批量创建 Channel CommEngine engine = CommEngine::COMM_ENGINE_CPU_TS; std::vector<ChannelHandle> channels(channelNum); HcommChannelCreate(epHandle, engine, hcommDescVec.data(), channelNum, channels.data());

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