Finalize
产品支持情况
产品 |
是否支持 |
|---|---|
√ |
|
√ |
|
x |
|
x |
|
x |
|
x |
|
x |
功能说明
通知服务端后续无通信任务，执行结束后退出。
函数原型
1 2 | template <bool sync = true> __aicore__ inline void Finalize() |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
sync |
输入 |
是否需要等待服务端的通信完成。bool类型，参数取值如下：
|
返回值说明
无
约束说明
- 调用本接口前确保已调用过Init和SetCcTiling接口。
- 本接口在AI Cube核或者AI Vector核上调用必须与对应的Prepare接口的调用核保持一致。
调用示例
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | REGISTER_TILING_DEFAULT(ReduceScatterCustomTilingData); //ReduceScatterCustomTilingData为对应算子头文件定义的结构体 auto tiling = (__gm__ ReduceScatterCustomTilingData*)tilingGM; Hccl hccl; GM_ADDR contextGM = AscendC::GetHcclContext<0>(); // AscendC自定义算子kernel中，通过此方式获取Hccl context __gm__ void *mc2InitTiling = (__gm__ void *)(&tiling->mc2InitTiling); __gm__ void *mc2CcTiling = (__gm__ void *)(&(tiling->mc2CcTiling)); hccl.Init(contextGM, mc2InitTiling); if (AscendC::g_coreType == AIC) { HcclHandle handleId = hccl.ReduceScatter(sendBuf, recvBuf, 100, HcclDataType::HCCL_DATA_TYPE_INT8, HcclReduceOp::HCCL_REDUCE_SUM, 10); hccl.Commit(handleId ); // 通知服务端可以执行上述的ReduceScatter任务 for (uint8_t i=0; i<10; i++) { hccl.Wait(handleId); // 阻塞接口，需等待上述ReduceScatter任务执行完毕 } hccl.Finalize<true>(); // 后续无其他通信任务，通知服务端执行上述ReduceScatter任务之后即可以退出 } |
父主题： Hccl