CANN/asc-devkit HCCL Tiling使用说明

发布时间:2026/7/17 9:46:21
CANN/asc-devkit HCCL Tiling使用说明 HCCL Tiling使用说明【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit[!NOTE]说明 本章节详述通信算子必需的TilingData结构体。本章节提供了三个版本的TilingData建议使用TilingData结构体不推荐使用v1版本TilingData废弃和v2版本TilingData废弃。 对于TilingData结构体在定义通算融合算子的Tiling结构体时Mc2InitTiling必须定义为算子Tiling结构体的第一个参数Mc2CcTiling对于在算子Tiling结构体中被定义的位置没有要求。Ascend C提供一组HCCL Tiling API方便用户获取HCCL Kernel计算时所需的Tiling参数。您只需要传入通信的相关信息调用API接口即可获取通信相关的Tiling参数。HCCL Tiling API获取Tiling参数的流程如下创建一个Mc2CcTilingConfig类对象。const char* groupName testGroup; uint32_t opType HCCL_CMD_REDUCE_SCATTER; std::string algConfig ReduceScatterlevel0:fullmesh; uint32_t reduceType HCCL_REDUCE_SUM; AscendC::Mc2CcTilingConfig mc2CcTilingConfig(groupName, opType, algConfig, reduceType);通过配置接口设置通信信息可选。mc2CcTilingConfig.SetSkipLocalRankCopy(0); mc2CcTilingConfig.SetSkipBufferWindowCopy(1);可调用的配置接口列于下表。表1Mc2CcTilingConfig类对象的配置接口列表接口功能SetOpType设置通信任务类型。SetGroupName设置通信任务所在的通信域。SetAlgConfig设置通信算法。SetReduceType设置Reduce操作类型。SetStepSize设置细粒度通信时通信算法的步长。SetSkipLocalRankCopy设置本卡的通信算法的计算结果是否输出到recvBuf。SetSkipBufferWindowCopy设置通信算法获取输入数据的位置。SetDebugMode设置调测模式。调用GetTiling接口获取Tiling信息。mc2CcTilingConfig.GetTiling(tiling-mc2InitTiling); // tiling为算子组装的TilingData结构体 mc2CcTilingConfig.GetTiling(tiling-reduceScatterTiling);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考