使用说明

Ascend C提供一组Matmul Tiling API,方便用户获取MatMul kernel计算时所需的Tiling参数。您只需要传入A/B/C矩阵等信息,调用API接口,即可获取到Init中TCubeTiling结构体中的相关参数。

Matmul Tiling API分为单核Tiling接口多核Tiling接口。获取Tiling参数的流程如下:

  1. 创建一个单核Tiling对象或者多核Tiling对象
  2. 设置A、B、C、Bias的参数类型信息;M、N、K形状信息等。
  3. 调用GetTiling接口,获取Tiling信息。

使用单核Tiling接口和多核Tiling接口获取Tiling参数的样例如下: