GetCoreMemBw

函数功能

获取硬件平台存储空间的带宽大小,仅支持L2、HBM。硬件存储空间类型定义如下:

enum class CoreMemType {
L0_A = 0,
L0_B = 1,
L0_C = 2,
L1 = 3,
L2 = 4,
UB = 5,
HBM = 6,
RESERVED
};

函数原型

void GetCoreMemBw(const CoreMemType &memType, uint64_t &bwSize) const;

参数说明

参数

输入/输出

说明

mem_type

输入

硬件存储空间类型

bw_size

输出

对应硬件的存储空间的带宽大小

返回值说明

约束说明

mem_type输入仅支持L2、HBM。

调用示例

ge::graphStatus TilingXXX(gert::TilingContext* context) {
    auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
    uint64_t l2_bw;
    ascendcPlatform.GetCoreMemBw(platform_ascendc::CoreMemType::L2, l2_bw);
    // ...
    return ret;
}