概览#
本书的主线内容位于第一至第四部分。附录收录了阅读过程中可能需要查询的补充内容:
需要查询的内容 |
对应页面 |
|---|---|
TIRx 语言特性的准确写法和语义 |
|
可复现地测量、比较和分析 GPU kernel 性能 |
|
编译器内部机制与 lowering 流程 |
|
排查异步 GEMM 或 Flash Attention kernel 的卡死、崩溃、错误结果和性能下降 |
完整的 tvm.tirx Python API 请参阅
TVM 官方文档。
第二部分介绍了 TIRx 编程模型(TIRx 入门)和 tensor layout 模型 (TIRx Layout API)。