Skip to content
版 本

AddPlatformInfo

功 能 说 明

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

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

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

函 数 原 型

Text
ContextBuilder &AddPlatformInfo(const char* customSocVersion)

参 数 说 明

参 数 名

输 入/输 出

描 述

customSocVersion

输 入

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

  • 针 对 如 下 产 品:在 安 装AI处 理 器的 服 务 器 执 行npu-smi info命 令 进 行 查 询,获 取Name信 息。实 际 配 置 值 为AscendName,例 如Name取 值 为xxxyy,实 际 配 置 值 为Ascendxxxyy

    Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品

    Atlas 200I/500 A2 推 理 产 品

    Atlas 推 理 系 列 产 品

    Atlas 训 练 系 列 产 品

  • 针 对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。
  • 针 对Ascend 950PR/Ascend 950DT,在 安 装AI处 理 器的 服 务 器 执 行npu-smi info -t board -i id命 令 进 行 查 询,获 取Chip NameNPU Name信 息,实 际 配 置 值 为Chip Name_NPU Name。例 如Chip Name取 值 为AscendxxxNPU Name取 值 为1234,实 际 配 置 值 为Ascendxxx_1234。

    其 中,id为 设 备id,通 过npu-smi info -l命 令 查 出 的NPU ID即 为 设 备id。

返 回 值 说 明

当 前ContextBuilder对 象。

约 束 说 明

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

调 用 示 例

Text
void AddPlatformInfoDemo(......)
{
    auto holder = context_ascendc::ContextBuilder()
	// ... ... // 增 加 算 子 输 入 输 出 接 口 的 调 用
	.AddPlatformInfo("Ascendxxxyy")
	.BuildTilingContext();
    auto tilingContext = holder->GetContext<gert::TilingContext>();
    // ... ...
}

免 责 声 明:本 站 内 容 由 asc-devkit 仓 master 分 支 自 动 编 译 生 成,属 于 持 续 开 发 版 本,可 能 存 在 缺 陷,仅 供 预 览 与 参 考。如 需 稳 定 及 商 用 资 料,请 查 阅 官 方 昇 腾 社 区