标签: "MLIR"

AMD 的 FlyDSL 是 Fly dialect MLIR 编译器的 Python 前端， 把 layout 代数和 copy / MMA atom 一路 lower 到 CDNA3 / CDNA4 的 ROCDL。 examples 目录下 vectorAdd、 tiledCopy、 tiledMma、 preshuffle GEMM 四个例子构成一条严格的进阶阶梯， 按顺序读完， 你就掌握了 paged attention、 MoE GEMM、 flash attention 这些 production kernel 反复重组的所有零件。

上一篇 FlyDSL 精读把 layout 代数收在了五个词上， 这篇补的是更深一层的子概念 —— Fly_Basis 类型和 BasisAttr。 是什么、 为什么 layout 需要、 以及 mentor 给我的'完善它'任务从哪条线进入最干净。

你可以把生产级 ML 系统写好几年， 都不知道 IR、 MLIR、 LLVM、 ISA、 FFI 这些词指什么。 这一篇是补丁 —— 写给本科 CS 念过、 但是 Compiler 和 Computer Arch 没好好上过的 ML 工程师。 配一份带 6 张 SVG 图、 中英双语的 HTML 深读。