CANN HCOMM流式线程获取
HcclThreadAcquireWithStream
【免费下载链接】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 异构子框。
功能说明
基于已有runtime stream获取指定notifyNum的通信线程资源。
当前适用于通信引擎为HOST CPU+TS、CCU的场景。
函数原型
HcclResult HcclThreadAcquireWithStream(HcclComm comm, CommEngine engine, aclrtStream stream, uint32_t notifyNum, ThreadHandle *thread)参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| comm | 输入 | 通信域句柄。 HcclComm类型的定义如下： typedef void *HcclComm; |
| engine | 输入 | 通信引擎类型。 CommEngine类型的定义可参见CommEngine。 |
| stream | 输入 | stream句柄。 |
| notifyNum | 输入 | 同步信号数量。 |
| thread | 输出 | 线程句柄。 ThreadHandle类型的定义可参见ThreadHandle。 |
返回值
HcclResult：接口成功返回HCCL_SUCCESS，其他失败。
约束说明
无
调用示例
HcclComm comm; CommEngine engine = COMM_ENGINE_CPU_TS ; aclrtStream stream; aclrtCreateStream(&stream); ThreadHandle thread; HcclResult result = HcclThreadAcquireWithStream(comm, engine, stream, 2, &thread); // 数据面操作 // ... // 流同步 aclrtSynchronizeStream(stream);【免费下载链接】hcommHCOMM（Huawei Communication）是HCCL的通信基础库，提供通信域以及通信资源的管理能力。项目地址: https://gitcode.com/cann/hcomm
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考