昇腾社区首页
中文
注册

PlatformAscendCManager类

函数功能

基于Kernel Launch算子工程,通过基础调用(Kernel Launch)方式调用算子的场景下,可能需要获取硬件平台相关信息,比如获取硬件平台的核数。PlatformAscendCManager类提供获取平台信息的功能:通过该类的GetInstance方法可以获取一个PlatformAscendC类的实例,再通过该实例获取硬件平台相关信息,支持获取的信息可参考PlatformAscendC类

使用该功能需要包含"tiling/platform/platform_ascendc.h"头文件。样例如下:

#include "tiling/platform/platform_ascendc.h"

参数说明

参数

输入/输出

说明

-

-

-

返回值说明

约束说明

调用示例

ge::graphStatus TilingXXX() {
    auto coreNum = platform_ascendc::PlatformAscendCManager::GetInstance()->GetCoreNum();
    // ... 根据核数自行设计Tiling策略
    context->SetBlockDim(coreNum);
    return ret;
}