Skip to content

AddPlatformInfo

功能说明

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

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

若设置失败,会打印报错信息。关于日志配置和查看,请参考《环境变量参考》中日志相关环境变量。

函数原型

Text
ContextBuilder &AddPlatformInfo(const char* customSocVersion)

参数说明

参数名输入/输出描述
customSocVersion输入AI处理器型号。

返回值说明

当前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 分支自动编译生成,属于持续开发版本,可能存在缺陷,仅供预览与参考。如需稳定及商用资料,请查阅官方 昇腾社区