CANN/asc-devkit SetFixSplit矩阵分块设置

SetFixSplit【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明设置固定的baseM、baseN、baseK单位为元素个数。函数原型int32_t SetFixSplit(int32_t baseMIn -1, int32_t baseNIn -1, int32_t baseKIn -1)参数说明表1参数说明参数名输入/输出描述baseMIn输入设置固定的baseM默认值为-1表示不设置固定baseM由tiling函数进行计算。baseNIn输入设置固定的baseN默认值为-1表示不设置固定baseN由tiling函数进行计算。baseKIn输入当前仅支持取值为-1暂不支持设置其它值。返回值说明-1表示设置失败0表示设置成功。约束说明baseM*baseN个输出元素所占的存储空间大小不能超过L0C Buffer大小即baseM * baseN * sizeof(C_TYPE) L0CSize。baseM需要小于等于singleM按16个元素向上对齐后的值如ceil(singleM/16)*16baseN需要小于等于singleN以C0_size个元素向上对齐的值其中singleM为单核内M轴长度singleN为单核内N轴长度half/bfloat16_t数据类型的C0_size为16float数据类型的C0_size为8int8_t数据类型的C0_size为32int4b_t数据类型的C0_size为64。例如singleM12则baseM需要小于等于16同时baseM需要满足分形对齐的要求所以baseM只能取16如果baseM取其他超过16的值获取Tiling将失败。调用示例auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); matmul_tiling::MatmulApiTiling tiling(ascendcPlatform); tiling.SetFixSplit(16, 16, -1); // 设置固定的baseM、baseN【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考