BatchMatmulGetTmpBufSize
功能说明
Batch Matmul切分后, 根据tiling结构体信息获取L1/UB/L0C的使用大小。
函数原型
1
|
int32_t BatchMatmulGetTmpBufSize(optiling::TCubeTiling & tiling,matmul_tiling::SysTilingTempBufSize &bufSize) |
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
tiling |
输入 |
Batch Matmul Tiling结构体 |
bufSize |
输出 |
根据Tiling结构体信息获取L1/UB/L0C的使用大小。 SysTilingTempBufSize结构定义如下: 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);
父主题: 获取MatMul计算所需空间