AddPlatformInfo

功能说明

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

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

函数原型

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 Name和NPU Name信息，实际配置值为Chip Name_NPU Name。例如Chip Name取值为Ascendxxx，NPU 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>();
    // ... ...
}
父主题： ContextBuilder