【TVM教程】TensorIR 创建
💡
原文中文，约7800字，阅读约需19分钟。
📝
内容提要
TVM 更新至 0.21.0 版本，中文文档已同步。TVM 是一个支持多种硬件的深度学习编译框架。使用 TVMScript 创建 TensorIR 函数可以简化代码，支持动态形状和与 Python 的交互。Tensor Expression 提供了更简洁的计算描述方式。
🎯
关键要点
- TVM 更新至 0.21.0 版本，中文文档已同步。
- Apache TVM 是一个支持多种硬件的深度学习编译框架。
- 使用 TVMScript 创建 TensorIR 函数可以简化代码，支持动态形状和与 Python 的交互。
- Tensor Expression 提供了更简洁的计算描述方式。
- TVMScript 是用于表示 TVM 中 TensorIR 的 Python 语言。
- 可以使用 T.grid 和 T.axis.remap 来简化代码。
- TVMScript 与 Python 变量可以进行一定程度的交互。
- 支持动态形状的 TensorIR 函数可以通过 Python 变量定义。
- Tensor Expression 是一种领域特定语言，用于描述计算过程。
- 可以使用 Tensor Expression 创建静态和动态形状的函数。
➡️
