TVM 0.21.0 中文文档已对齐。本教程介绍创建 Relax 程序的多种方式:使用 TVMScript 定义含高级算子的 MLP 模型,并可在 Relax 函数中调用 TensorIR;使用类 PyTorch 的 NNModule API 定义模型并导出为 IRModule,支持插入 TE、TensorIR 及打包函数;还介绍更底层的 Block Builder API,适合编写自定义 Pass 等复杂转换场景。
TVM 0.21.0 更新,中文文档已对齐。教程演示创建 Relax 函数的三种方式:TVMScript 定义高级运算符和 TensorIR 函数;NNModule API 类似 PyTorch,支持自定义调用;Block Builder API 为底层接口,适合复杂转换。示例包括 MLP 模型构建,并说明各 API 适用场景。
完成下面两步后,将自动完成登录并继续当前操作。