通知服务端后续无通信任务,执行结束后退出。客户端检测并等待最后一个通信任务执行结束。
1 | __aicore__ inline void Finalize() |
无
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(); // 后续无其他通信任务,通知服务端执行上述ReduceScatter任务之后即可以退出 } |