获取当前硬件平台芯片架构Vec计算单元位宽。
1 | uint32_t GetVecRegLen(void) const |
无
无
1 2 3 4 5 6 7 8 | void GetLayerNormMaxMinTmpSize(...) { platform_ascendc::PlatformAscendC* platform = platform_ascendc::PlatformAscendCManager::GetInstance(); ... const uint32_t vecLenB32 = platform->GetVecRegLen() / LAYERNORM_SIZEOF_FLOAT; const uint32_t vecLenB16 = platform->GetVecRegLen() / LAYERNORM_SIZEOF_HALF; ... } |