BatchMatmulGetTmpBufSize

功能说明

Batch Matmul切分后， 根据tiling结构体信息获取L1/UB/L0C的使用大小。

函数原型

1

int32_t BatchMatmulGetTmpBufSize(optiling::TCubeTiling & tiling，matmul_tiling::SysTilingTempBufSize &bufSize)

参数说明

struct SysTilingTempBufSize {
    int32_t ubSize = 0;
    int32_t l1Size = 0;
    int32_t l0cSize = 0;
};

返回值

-1表示获取失败； 0表示获取成功。

注意事项

无

调用示例

optiling::TCubeTiling tilingData; 
...初始化tilingData，详见Matmul Tiling类

SysTilingTempBufSize bufSize;
BatchMatmulGetTmpBufSize(tilingData, bufSize);