AddPlatformInfo
功能说明
设置硬件平台信息便于用户在算子Tiling函数调测中使用。支持以下两种设置方式:
- 自动获取当前硬件平台信息:传入空指针,自动获取当前硬件信息并添加到ContextBuilder类中。
- 指定硬件平台信息:传入具体的昇腾AI处理器型号,添加对应硬件信息至ContextBuilder类中。
函数原型
1
|
ContextBuilder &AddPlatformInfo(const char* customSocVersion) |
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
customSocVersion |
输入 |
昇腾AI处理器型号。配置方式如下:
|
返回值说明
当前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