CANN/HCOMM线程Notify等待API

HcommAclrtNotifyWaitOnThread

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

产品支持情况

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

功能说明

基于acl接口创建的Notify等待同步信号，须与HcommAclrtNotifyRecordOnThread配对使用。

函数原型

int32_t HcommAclrtNotifyWaitOnThread(ThreadHandle thread, uint64_t notifyId, uint32_t timeOut)

参数说明

返回值

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

约束说明

无

调用示例

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]); aclrtNotify notify; uint32_t notifyId; aclrtCreateNotify(&(notify), ACL_NOTIFY_DEFAULT); aclrtGetNotifyId(notify, &(notifyId)); // 发送同步信号 HcommAclrtNotifyRecordOnThread(threads[0], notifyId); // 等待同步信号 uint32_t timeout = 1; HcommAclrtNotifyWaitOnThread(threads[1], notifyId, timeout);

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