AddPlatformInfo
功能说明
设置硬件平台信息便于用户在算子Tiling函数调测中使用。支持以下两种设置方式:
- 自动获取当前硬件平台信息:传入空指针,自动获取当前硬件信息并添加到ContextBuilder类中。
- 指定硬件平台信息:传入具体的昇腾AI处理器型号,添加对应硬件信息至ContextBuilder类中。
若设置失败,会打印报错信息。关于日志配置和查看,请参考。
函数原型
ContextBuilder &AddPlatformInfo(const char* customSocVersion)
参数说明
返回值说明
当前ContextBuilder对象。
约束说明
AddPlatformInfo调用后需要通过BuildTilingContext来构建Tiling的上下文,并传递给Tiling函数来使用。
调用示例
void AddPlatformInfoDemo(......)
{
auto holder = context_ascendc::ContextBuilder()
// ... ... // 增加算子输入输出接口的调用
.AddPlatformInfo("Ascendxxxyy")
.BuildTilingContext();
auto tilingContext = holder->GetContext<gert::TilingContext>();
// ... ...
}