文档
注册
评分
提单
论坛
小AI

PlatformAscendCManager类

功能说明

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

  • 使用该功能需要包含"tiling/platform/platform_ascendc.h"头文件。样例如下:
    #include "tiling/platform/platform_ascendc.h"
  • 当前该类仅支持如下型号:
    • Atlas 推理系列产品(Ascend 310P处理器)
    • Atlas 训练系列产品
    • Atlas A2训练系列产品/Atlas 800I A2推理产品

函数原型

class PlatformAscendCManager {
public:
    static PlatformAscendC* GetInstance();
    // 在仅有CPU环境、无对应的NPU硬件环境时,需要传入customSocVersion来指定对应的AI处理器型号。注意:因为GetInstance实现属于单例模式,仅在第一次调用时传入的customSocVersion生效。
    static PlatformAscendC* GetInstance(const char *customSocVersion);
private:
...
}

参数说明

参数

输入/输出

说明

customSocVersion

输入

AI处理器型号。

如果无法确定具体的AI处理器型号,则在安装昇腾AI处理器的服务器执行npu-smi info命令进行查询,在查询到的“Name”前增加Ascend信息,例如“Name”对应取值为xxxyy,实际配置的<soc_version>值为Ascendxxxyy

返回值说明

约束说明

调用示例

GetInfoFun() {
    ...
    auto coreNum = platform_ascendc::PlatformAscendCManager::GetInstance()->GetCoreNum();
    ...
    return;
}
搜索结果
找到“0”个结果

当前产品无相关内容

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