PlatformAscendC类简介
函数功能
在实现Host侧的Tiling函数时,可能需要获取一些硬件平台的信息,来支撑Tiling的计算,比如获取硬件平台的核数等信息。PlatformAscendC类提供获取这些平台信息的功能。
使用该功能需要包含"tiling/platform/platform_ascendc.h"头文件。样例如下:
#include "tiling/platform/platform_ascendc.h"
函数原型
1 2 3 |
PlatformAscendC() = delete ~PlatformAscendC() = default explicit PlatformAscendC(fe::PlatFormInfos *platformInfo): platformInfo_(platformInfo) {} |
返回值说明
无
约束说明
无
调用示例
ge::graphStatus TilingXXX(gert::TilingContext* context) { auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); uint64_t ub_size, l1_size; ascendcPlatform.GetCoreMemSize(platform_ascendc::CoreMemType::UB, ub_size); ascendcPlatform.GetCoreMemSize(platform_ascendc::CoreMemType::L1, l1_size); auto aicNum = ascendcPlatform.GetCoreNumAic(); auto aivNum = ascendcPlatform.GetCoreNumAiv(); // ... 按照aivNum切分 context->SetBlockDim(ascendcPlatform.CalcTschBlockDim(aivNum, aicNum, aivNum)); return ret; }
父主题: PlatformAscendC