CANN/HCOMM线程通知记录API

HcommChannelNotifyRecordOnThread

【免费下载链接】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 异构子框。

功能说明

发送同步信号，在Thread上记录一个Notify。该接口为异步接口，主要用于Channel两端同步等待场景。

函数原型

int32_t HcommChannelNotifyRecordOnThread(ThreadHandle thread, ChannelHandle channel, uint32_t remoteNotifyIdx)

参数说明

返回值

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

约束说明

该接口需要配合HcommChannelNotifyWaitOnThread使用。

在 Ascend 950PR/Ascend 950DT 上，仅支持 AICPU_TS 模式下、在 Device 侧调用该接口。

调用示例

// 申请通信线程资源 CommEngine engine = CommEngine::COMM_ENGINE_CPU_TS; CommEngine engine = CommEngine::COMM_ENGINE_AICPU_TS; // Ascend 950PR/Ascend 950DT时配置 uint32_t threadNum = 1; uint32_t notifyNumPerThread = 1; ThreadHandle thread; HcclThreadAcquire(engine, threadNum, notifyNumPerThread, &thread); // 申请通信通道资源 uint32_t channelNum = 1; HcclChannelDesc channelDesc; HcclChannelDescInit(&channelDesc, channelNum); HcclComm comm; ChannelHandle channel; HcclChannelAcquire(comm, engine, &channelDesc, channelNum, &channel); // Ascend 950PR/Ascend 950DT时，需要在 Device 侧调用以下接口 // 通知对端 HcommChannelNotifyRecordOnThread(thread, channel, 0); // 数据面操作 // ... // 等待对端通知本端 uint32_t notifyTimeout = 1800; HcommChannelNotifyWaitOnThread(thread, channel, 0, notifyTimeout);

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