文档
注册
评分
提单
论坛
小AI

MatmulGetTmpBufSize

功能说明

单核Matmul切分后, 根据Tiling结构体信息获取L1/UB/L0C的使用大小。

函数原型

1
int32_t MatmulGetTmpBufSize(optiling::TCubeTiling & tilingmatmul_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;
MatmulGetTmpBufSize(tilingData, bufSize);
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词