Query

功能说明

查询handleId对应的通信任务已经完成的轮次，最多返回repeat轮。该接口默认在所有核上工作，用户也可以在调用前通过GetBlockIdx指定其在某一个核上运行。

函数原型

1

__aicore__ inline int32_t Query(HcclHandle handleId)

参数说明

using HcclHandle = int8_t;

返回值

• 返回handleId对应的通信任务已执行的次数，最大值为repeat。
• 当执行异常时，返回-1。

支持的型号

Atlas A2训练系列产品/Atlas 800I A2推理产品

注意事项

• 调用本接口前确保已调用过Init接口
• 入参handleId只能使用Prepare原语对应接口的返回值
• 该接口只能在AI Cube核或者AI Vector核两者之一上调用

调用示例

// 假设已构造好HcclCombineOpParam对象hcclCombineOpParam，以及recvBuf和sendBuf
Hccl hccl;
hccl.Init(reinterpret_cast<GM_ADDR>(&hcclCombineOpParam));
if (g_coreType == AIC) {
    auto repeat = 10;
    HcclHandle handleId = hccl.ReduceScatter(sendBuf, recvBuf, 100, HcclDataType::HCCL_DATA_TYPE_INT8, HcclReduceOp::HCCL_REDUCE_SUM, repeat);
    hccl.Commit(handleId ); // 通知服务端可以执行上述的ReduceScatter任务
    int32_t finishedCount = hccl.Query(handleId);
    while (hccl.Query(handleId) < repeat) {} // 等待查询到handleId对应的通信任务执行repeat次
    hccl.Finalize(); // 后续无其他通信任务，通知服务端执行上述ReduceScatter任务之后即可以退出
}