
MLIR的增量编译与缓存机制一个让我熬夜到凌晨三点的bug去年冬天,我在调试一个基于MLIR的AI编译器。模型编译一次要40分钟,每次改一行pass代码就得全量重编。那天我加了个简单的constant folding优化,编译跑了三遍都报同一个段错误——但诡异的是,每次报错的行号都不一样。第四遍,我盯着终端输出,发现MLIR在重复解析同一个.mlir文件,每次解析完都重新做一遍dialect注册。那一刻我意识到:MLIR的编译流水线里,根本没有缓存这回事。后来翻了LLVM的邮件列表,发现这不是我一个人踩的坑。MLIR的增量编译机制,在2022年之前几乎是个“黑箱”——官方文档里只有一句话“MLIR supports incremental compilation through pass pipelining”,但怎么支持、缓存粒度多大、什么场景会失效,全得靠读源码。MLIR的编译单元:从Module到Operation的“脏”标记MLIR的增量编译核心,在于它把IR拆成了可独立编译的“单元”。这个单元不是文件,也不是函数,而是Operation——更准确地说,是顶层Operation(通常是builtin.module)。每个Operation内部维护了一个isDirty标志位。当你修改某个pass时,MLIR不会傻乎乎地重跑整个pipeline,而是从根Operation开始,逐层检查哪些子Operat