概览

概览#

本书的主线内容位于第一至第四部分。附录收录了阅读过程中可能需要查询的补充内容:

需要查询的内容

对应页面

TIRx 语言特性的准确写法和语义

TIRx 语言参考

可复现地测量、比较和分析 GPU kernel 性能

GPU Kernel 性能测量与分析

编译器内部机制与 lowering 流程

编译器内部机制

排查异步 GEMM 或 Flash Attention kernel 的卡死、崩溃、错误结果和性能下降

调试 Warp-Specialized Kernels

完整的 tvm.tirx Python API 请参阅 TVM 官方文档

第二部分介绍了 TIRx 编程模型(TIRx 入门)和 tensor layout 模型 (TIRx Layout API)。