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的场景示意图
![](https://raw.gitcode.com/cann/asc-devkit/raw/8ddd8427a6d4c53bc58ad284e1cb4f23effabeae/docs/api/figures/参数传入-1的场景示意图.png "参数传入-1的场景示意图"?utm_source=gitcode_repo_files)

函数原型

int32_t SetShape(int32_t m, int32_t n, int32_t k)

参数说明

表1参数说明

返回值说明

-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