简介
算子信息库作为算子开发的交付件之一,主要体现算子在昇腾AI处理器上的具体实现规格。针对TBE算子,算子信息库包括算子支持的输入输出dtype、format以及输入shape等信息;针对AI CPU算子,算子信息库包括算子输入输出的name,支持的dtype、format等信息。网络运行时,图编译器会根据算子信息库中的算子信息做基本校验,并进行算子匹配。
算子匹配的规则如下:
图编译器优先查找TBE算子信息库,若能匹配到TBE算子信息库中的算子,则选择TBE算子进行图编译;若无法匹配到TBE算子信息库中的算子,则查找AI CPU算子信息库,选择AI CPU算子进行图编译。
父主题: 算子信息库定义