PlatformAscendCManager类
功能说明
基于Kernel Launch算子工程,通过基础调用(Kernel Launch)方式调用算子的场景下,可能需要获取硬件平台相关信息,比如获取硬件平台的核数。PlatformAscendCManager类提供获取平台信息的功能:通过该类的GetInstance方法可以获取一个PlatformAscendC类的指针,再通过该指针获取硬件平台相关信息,支持获取的信息可参考PlatformAscendC类。
- 使用该功能需要包含"tiling/platform/platform_ascendc.h"头文件。样例如下:
#include "tiling/platform/platform_ascendc.h"
- 当前该类仅支持如下型号:
- Atlas 推理系列产品
- 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; }
父主题: 平台信息获取