TVM 快速上手

• PackedFunc
  • 全局函数
  • Python 调用 C++ 接口
  • convert 函数
  • 返回节点的回调
  • 返回函数
  • 设备函数
  • 右值引用
  • numpy scalar 和 args
  • dict 函数值类型
  • Hook Python 函数作为 Extern
  • PyTorch 调用 TVM 接口
• 解读 tvm.tir.trace()
  • trace expr 赋值
  • trace expr 加法
  • trace 常量表达式
• 自定义 relay 算子(python)
  • 构建 reshape4d_softmax_reshape2d ONNX 算子
  • 变换 reshape4d_softmax_reshape2d 为 softmax_transpose_reshape2d
  • 声明 softmax_transpose_reshape2d 算子类型关系
  • 注册 softmax_transpose_reshape2d 算子
  • 测试 softmax_transpose_reshape2d 算子类型推断
  • 融合 mod 部分算子为 softmax_transpose_reshape2d
  • 定义 softmax_transpose_reshape2d 计算与调度
  • 验证 softmax_transpose_reshape2d 数值一致性
  • 打印带有 softmax_transpose_reshape2d 算子的量化问题
• 重写 relay 量化算子(python)
  • 未成功量化 softmax_transpose_reshape2d 结构
  • 为 nn.softmax 添加分区规则
  • 为 nn.softmax 添加注解规则
  • 校验量化 softmax 结果
• 自定义 VTA 算子(python)
  • 重写预处理算子
  • graph pack 预处理算子