autofuse 样例使用指导

功能描述

使用 autofuse 完成各种类型的算子融合

目录结构

├── af_pointwise                  # pointwise 类型算子融合的样例
│  └── af_add_ge.py               # 通过 autofuse 完成 add 和 ge 两个 pointwise 类型算子的融合
├── af_reduce                     # reduce 类型算子融合的样例
│  └── af_mul_reducesum.py        # 通过 autofuse 完成 mul 和 reducesum 两个 pointwise 类型算子的融合

前置说明

请务必参考《Autofuse 简介与快速上手》完成前置环境准备。

用例演示

用例1 用例2

参考

请参考inductor-npu-ext使用手册的相关内容。