AddConfig
功能说明
注册算子支持的AI处理器型号以及OpAICoreConfig信息。
函数原型
1 2 | void AddConfig(const char *soc) void AddConfig(const char *soc, OpAICoreConfig &aicore_config) |
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
soc |
输入 |
支持的AI处理器型号。 |
aicore_config |
输入 |
AI Core配置信息请参考OpAICoreConfig定义。 |
返回值说明
无
约束说明
不传入aicore_config参数时,对OpAICoreConfig结构中的部分参数会配置成默认值,具体的参数和默认值如下表所示:
配置参数 |
说明 |
默认值 |
---|---|---|
用于标识该算子实现是否支持入图时的静态Shape编译。 |
true |
|
标识是否根据SetOpSelectFormat设置的函数自动推导算子输入输出支持的dtype和format。 |
true |
|
标识算子是否支持dynamicRanK(动态维度)。 |
true |
|
用于标识该算子是否支持入图时的动态Shape场景。 |
true |
|
标识是否在算子融合阶段调用算子参数校验函数进行data type与Shape的校验。 |
false |
|
此字段用于进行ATC模型转换或者进行网络调测时,控制算子的精度模式。 |
true |
父主题: OpAICoreDef