HcclThreadAcquire

产品支持情况

针对Atlas A2 训练系列产品/Atlas A2 推理系列产品，仅支持Atlas 800T A2 训练服务器、Atlas 900 A2 PoD 集群基础单元、Atlas 200T A2 Box16 异构子框。

功能说明

基于通信域获取通信线程。

函数原型

1

HcclResult HcclThreadAcquire(HcclComm comm, CommEngine engine, uint32_t threadNum, uint32_t notifyNumPerThread, ThreadHandle *threads)

参数说明

1

typedef void *HcclComm;

返回值

HcclResult：接口成功返回HCCL_SUCCESS，其他失败。

约束说明

返回的通信线程与同步资源由库内管理，调用者严禁释放。

调用示例

创建线程资源示例如下：

1
2
3
4
5

HcclComm comm;
CommEngine engine = COMM_ENGINE_AICPU_TS;
ThreadHandle threads[5];
// 申请5条流，每条流2个notify
HcclThreadAcquire(comm, engine, 5, 2, threads);

同步Host线程资源示例如下：

// 申请1条host流
aclrtStream stream;
aclrtCreateStream(&stream);
// 根据申请的流创建CPU_TS类型thread
ThreadHandle cpuThread;
HcclThreadAcquireWithStream(comm, COMM_ENGINE_CPU_TS, streams, 1, &cpuThread);
// 任务编排
// ...
// 流同步
aclrtSynchronizeStream(stream);

同步AICPU线程资源示例如下：

// --host侧调用流程--
// 申请1条host流
aclrtStream stream;
aclrtCreateStream(&stream);
// 根据申请的流创建CPU_TS类型thread
ThreadHandle cpuThread;
HcclThreadAcquireWithStream(comm, COMM_ENGINE_CPU_TS, streams, 1, &cpuThread);

// 创建一个AICPU_TS类型的thread
ThreadHandle aicpuThread;
HcclThreadAcquire(comm, COMM_ENGINE_AICPU_TS, 1, 1, &aicpuThread);

// 把创建的AICPU_TS类型的thread导出为CPU上可用的thread
ThreadHandle exportedCpuThread;
HcclThreadExportToCommEngine(comm, 1, &aicpuThread, COMM_ENGINE_CPU_TS, &exportedCpuThread);
// 把创建的CPU类型的thread导出为AICPU上可用的thread
ThreadHandle exportedAicpuThread;
HcclThreadExportToCommEngine(comm, 1, &cpuThread, COMM_ENGINE_AICPU_TS, &exportedAicpuThread);

// 发送同步信号
HcommThreadNotifyRecordOnThread(cpuThread, exportedCpuThread, 0);
// 下发kernel，把exportedAicpuThread，aicpuThread带到aicpu侧
// ...
uint32_t timeout = 1;
// 等待同步信号
HcommThreadNotifyWaitOnThread(cpuThread, 0, timeout);

// --aicpu侧调用流程--
// 等待同步信号
uint32_t timeout = 1;
HcommThreadNotifyWaitOnThread(aicpuThread, 0, timeout);
// 任务编排，将任务下发到aicpuThread
// ...
// 发送同步信号
HcommThreadNotifyRecordOnThread(aicpuThread, exportedAicpuThread, 0);

// 流同步
aclrtSynchronizeStream(stream);