MultiCoreMatmulGetTmpBufSize

功能说明

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

函数原型

int32_t MultiCoreMatmulGetTmpBufSize(optiling::TCubeTiling & tiling, matmul_tiling::SysTilingTempBufSize &bufSize)

参数说明

表1 参数说明

参数名

输入/输出

描述

tiling

输入

多核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;
MultiCoreMatmulGetTmpBufSize(tilingData, bufSize);

父主题： 获取MatMul计算所需空间