昇腾社区首页
中文
注册

AddPlatformInfo

功能说明

设置硬件平台信息便于用户在算子Tiling函数调测中使用。支持以下两种设置方式:

  • 自动获取当前硬件平台信息:传入空指针,自动获取当前硬件信息并添加到ContextBuilder类中。
  • 指定硬件平台信息:传入具体的昇腾AI处理器型号,添加对应硬件信息至ContextBuilder类中。

若设置失败,会打印报错信息。关于日志配置和查看,请参考日志

函数原型

1
ContextBuilder &AddPlatformInfo(const char* customSocVersion)

参数说明

参数名

输入/输出

描述

customSocVersion

输入

昇腾AI处理器型号。配置方式如下:

  • Atlas A3 训练系列产品 / Atlas A3 推理系列产品 :在安装昇腾AI处理器的服务器执行npu-smi info命令进行查询,获取Name信息。实际配置值为AscendName,例如Name取值为xxxyy,实际配置值为Ascendxxxyy
  • Atlas A3 训练系列产品 / Atlas A3 推理系列产品 :在安装昇腾AI处理器的服务器执行npu-smi info -t board -i id -c chip_id命令进行查询,获取Chip NameNPU Name信息,实际配置值为Chip Name_NPU Name。例如Chip Name取值为AscendxxxNPU Name取值为1234,实际配置值为Ascendxxx_1234。

    其中:

    • id:设备id,通过npu-smi info -l命令查出的NPU ID即为设备id。
    • chip_id:芯片id,通过npu-smi info -m命令查出的Chip ID即为芯片id。

返回值说明

当前ContextBuilder对象。

约束说明

AddPlatformInfo调用后需要通过BuildTilingContext来构建Tiling的上下文,并传递给Tiling函数来使用。

调用示例

1
2
3
4
5
6
7
8
9
void AddPlatformInfoDemo(......)
{
    auto holder = context_ascendc::ContextBuilder()
	// ... ... // 增加算子输入输出接口的调用
	.AddPlatformInfo("Ascendxxxyy")
	.BuildTilingContext();
    auto tilingContext = holder->GetContext<gert::TilingContext>();
    // ... ...
}