下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

MultiCoreMatmulGetTmpBufSize

功能说明

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

函数原型

1
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);
搜索结果
找到“0”个结果

当前产品无相关内容

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