UB算子融合推荐分析样例
背景介绍
算子是组成模型的基本单元。模型转换时,会进行算子融合来达到网络性能提升的目的。当模型中存在算子融合规则未覆盖且可融合的算子时,模型转换完成则无法达到最优性能。本节以YOLOV5模型为例,介绍通过专家系统UB算子融合推荐功能自动发现并输出模型中可融合的Cube和Vector算子,帮助用户快速定位可融合算子,提升模型性能。
- 本文仅介绍专家系统UB算子融合推荐功能的操作及分析过程,对于OM模型中算子融合的具体操作此处不做阐述。
- 有关UB算子融合的详细介绍请参见《图融合和UB融合规则参考》。
- 有关模型转换的详细介绍请参见模型转换。
专家系统操作
问题解决
具体融合操作需开发者自行修改算子代码,完成融合后的模型结构示例如图5所示。
图中以YOLOV5网络模型的Conv2D、Mul和AscendQuart算子融合为Conv2D算子为例,左侧部分为融合前的YOLOV5网络模型结构,右侧部分为优化后的YOLOV5网络模型结构,可以看到优化后模型结构得到了简化。
结论
通过专家系统工具对OM模型的UB算子融合分析可知专家系统工具可以快速准确定位到模型中的可融合算子,减少用户的定位时间,提升效率。
父主题: 专家系统分析样例参考