CANN/asc-devkit：SetShape矩阵形状设置

SetShape【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明设置Matmul计算的形状m、n、k该形状可以为原始完整矩阵或其局部矩阵单位为元素。该形状的矩阵乘可以由单核或多核计算完成。使用本接口时有两种参数传入方式传入Matmul计算的形状m、n、k调用GetTiling接口时按照m、n、k计算并返回Tiling参数。m、n、k中任意一个或多个参数位置传入-1调用GetTiling接口时该位置取SetOrgShape接口中设置的原始形状M、N、K或Ka/Kb然后由接口内部计算最优Tiling参数。如下图所示原始A矩阵的K方向最后一列为不参与计算的脏数据在SetOrgShape接口中设置包含该列数据的原始形状在本接口中设置Matmul计算的K方向大小同时参数m、n设置为-1表示按照原始形状M、N计算Tiling。图1参数传入-1的场景示意图函数原型int32_t SetShape(int32_t m, int32_t n, int32_t k)参数说明表1参数说明参数名输入/输出描述m输入设置Matmul计算的M方向大小单位为元素。n输入设置Matmul计算的N方向大小单位为元素。k输入设置Matmul计算的K方向大小单位为元素。返回值说明-1表示设置失败0表示设置成功。约束说明无调用示例auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); matmul_tiling::MatmulApiTiling tiling(ascendcPlatform); tiling.SetShape(1024, 1024, 1024); // 设置Matmul计算的形状 tiling.SetOrgShape(1024, 1024, 1024);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考