CANN/HCOMM线程通知等待函数

HcommThreadNotifyWaitOnThread

【免费下载链接】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被record完成。

函数原型

int32_t HcommThreadNotifyWaitOnThread(ThreadHandle thread, uint32_t notifyIdx, uint32_t timeout)

参数说明

返回值

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

约束说明

该接口需要配合HcommThreadNotifyRecordOnThread使用。

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

调用示例

HcclComm comm; CommEngine engine = COMM_ENGINE_CPU_TS; aclrtStream streams[2]; ThreadHandle threads[2]; // 申请2条流，每条流2个Notify aclrtCreateStream(&streams[0]); aclrtCreateStream(&streams[1]); HcclResult result = HcclThreadAcquireWithStream(comm, engine, streams[0], 2, &threads[0]); result = HcclThreadAcquireWithStream(comm, engine, streams[1], 2, &threads[1]); uint32_t notifyIdx = 0; // 发送同步信号 HcommThreadNotifyRecordOnThread(threads[0], threads[1], notifyIdx); uint32_t timeout = 1; // 等待同步信号 HcommThreadNotifyWaitOnThread(threads[1], notifyIdx, timeout);

在 Ascend 950PR/Ascend 950DT 上，该函数需要编译到 Device 侧使用：

HcclComm comm; CommEngine engine = COMM_ENGINE_AICPU_TS; ThreadHandle threads[2]; uint32_t notifyNumPerThread = 2; HcclThreadAcquire(comm, engine, 1, notifyNumPerThread, &threads[0]); HcclThreadAcquire(comm, engine, 1, notifyNumPerThread, &threads[1]); // 其余资源申请 // 拷贝参数并 Launch Kernel // Device 侧算法编排 uint32_t notifyIdx = 0; // 发送同步信号 HcommThreadNotifyRecordOnThread(threads[0], threads[1], notifyIdx); uint32_t timeout = 1; // 等待同步信号 HcommThreadNotifyWaitOnThread(threads[1], notifyIdx, timeout);

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