TVM 变换#
- 变换概述
AnnotateTIROpPattern- TVM
FuseOps变换- 简单算子融合测试
- 测试卷积算子融合的情况
- 涉及连接算子和元组节点的融合测试
- 测试当
Tuple节点是其组中的根节点时的算子融合情况 - 元组获取与元素级算子的融合测试
- 元组获取作为根节点的融合测试
- 元组作为中间节点的融合测试
- 连续元组算子的融合测试
- 测试类似 Inception 网络结构的算子融合情况
- 测试并行的单射算子融合
- 测试softmax算子与后续算子的融合情况
- 测试多个 Relax 函数的算子融合
- 测试包含
call_dps_packed调用的算子融合 - 测试
call_dps_packed调用边缘的算子融合 - 测试 LayerNorm 和 SiLU 激活函数的融合
- 测试多路径网络结构的算子融合
- 测试包含无效组的算子融合
- 测试符号形状感知的算子融合
- 测试形状表达式参数的算子融合
- 测试跳过
match_cast的算子融合 - 测试包含原始值参数的算子融合
- 测试部分使用元组参数的算子融合。
- 测试包含 in-place 算子的算子融合
- 测试包含打包参数的算子融合
FoldConstantFoldBatchnormToConv2D- 融合算子
- 融合 Transpose 和 Matmul
- 合并复合函数
- AttachExternModules
- RunCodegen