SetBatchInfoForNormal
功能说明
Layout类型为NORMAL的场景，调用IterateBatch或者IterateNBatch接口之前，需要在host侧Tiling实现中通过本接口设置A/B矩阵的M/N/K轴信息，以及他们各自的Batch数。
函数原型
int32_t SetBatchInfoForNormal(int32_t batchA, int32_t batchB, int32_t M, int32_t N, int32_t K)
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
batchA |
输入 |
A矩阵的batch数 |
batchB |
输入 |
B矩阵的batch数 |
M |
输入 |
A矩阵的M轴信息 |
N |
输入 |
A矩阵的N轴信息 |
K |
输入 |
A/B矩阵的K轴信息 |
返回值
-1表示设置失败； 0表示设置成功。
注意事项
无
父主题： Matmul Tiling类