CANN/AscendC HCCL Wait接口文档
Wait【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品支持Atlas A2 训练系列产品/Atlas A2 推理系列产品支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品AI Core不支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明阻塞AI Core非细粒度通信时等待handleId对应的通信任务完成细粒度通信时等待handleId对应的步长长度的子通信任务完成。handleId调用Wait接口的顺序须和Prepare接口一致。该接口默认在所有核上工作用户也可以在调用前通过GetBlockIdx指定其在某一个核上运行。函数原型__aicore__ inline int32_t Wait(HcclHandle handleId)参数说明表1接口参数说明参数名输入/输出描述handleId输入对应通信任务的标识ID只能使用Prepare原语接口的返回值。using HcclHandle int8_t;返回值说明0表示成功-1表示失败约束说明调用本接口前确保已调用过InitV2和SetCcTilingV2接口。入参handleId只能使用Prepare原语对应接口的返回值。本接口在AIC核或者AIV核上调用必须与对应的Prepare接口的调用核保持一致。非细粒度通信时本接口的调用次数应该与Prepare的repeat次数一致细粒度通信时本接口的调用次数应该与通信任务的总步骤数/步长*Prepare的repeat次数一致。handleId调用Wait接口的顺序必须和Prepare接口一致。调用示例REGISTER_TILING_DEFAULT(ReduceScatterCustomTilingData); // ReduceScatterCustomTilingData为对应算子头文件定义的结构体 GET_TILING_DATA_WITH_STRUCT(ReduceScatterCustomTilingData, tilingData, tilingGM); Hccl hccl; GM_ADDR contextGM AscendC::GetHcclContext0(); // AscendC自定义算子kernel中通过此方式获取HCCL context __gm__ void* mc2InitTiling (__gm__ void*)(tiling-mc2InitTiling); __gm__ void* mc2CcTiling (__gm__ void*)((tiling-mc2CcTiling)); hccl.InitV2(contextGM, tilingData); auto ret hccl.SetCcTilingV2(offsetof(ReduceScatterCustomTilingData, mc2CcTiling)); if (ret ! HCCL_SUCCESS) { return; } if (AscendC::g_coreType AIC) { HcclHandle handleId hccl.ReduceScatter(sendBuf, recvBuf, 100, HcclDataType::HCCL_DATA_TYPE_INT8, HcclReduceOp::HCCL_REDUCE_SUM, 10); for (uint8_t i 0; i 10; i) { hccl.Commit(handleId); // 通知服务端可以执行上述的ReduceScatter任务 } for (uint8_t i 0; i 10; i) { hccl.Wait(handleId); // 阻塞接口需等待上述ReduceScatter任务执行完毕 } hccl.Finalize(); // 后续无其他通信任务通知服务端执行上述ReduceScatter任务之后即可退出 }【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
