调用IterateBatch接口之前，需要在host侧Tiling实现中通过本接口设置A矩阵的Layout轴信息，包括B、S、N、G、D轴。
int32_t SetALayout(int32_t B, int32_t S, int32_t N, int32_t G, int32_t D)
参数名 |
输入/输出 |
描述 |
|---|---|---|
B |
输入 |
A矩阵Layout的B轴信息 |
S |
输入 |
A矩阵Layout的S轴信息 |
N |
输入 |
A矩阵Layout的N轴信息 |
G |
输入 |
A矩阵Layout的G轴信息 |
D |
输入 |
A矩阵Layout的D轴信息 |
-1表示设置失败； 0表示设置成功。
无