https://github.com/hanhanW approved this pull request. Ugh, I've hit similar issues before. This is a bit annoying, thanks for fixing it. Will we have CI running with `MLIR_ENABLE_EXPENSIVE_PATTERN_API_CHECKS`? https://github.com/llvm/llvm-project/pull/78146