Update mlir-air and mlir-aie to latest wheels (LLVM 278dba37)#30
Merged
Conversation
Bump mlir-air to 5798472 (2026032704) and mlir-aie to 3a1aba2 (2026032704) to pick up LLVM 278dba37 update. Two compatibility fixes for the new LLVM: 1. Replace `transform.apply_patterns.vector.lower_multi_reduction` with the three ops it was split into: `reorder_multi_reduction_dims`, `multi_reduction_flattening`, and `multi_reduction_unrolling`. 2. Add explicit `--air-runtime-loop-tiling-sizes=4` to the aircc command in driver.py since mlir-air #1470 changed the default from [4,4] to [] (no tiling). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Contributor
There was a problem hiding this comment.
Pull request overview
Updates the repo’s pinned mlir-air / mlir-aie wheels to pick up an LLVM revision bump, and adjusts the transform scripts + backend driver to remain compatible with upstream API/default changes.
Changes:
- Bump pinned wheel commit hashes/timestamps for
mlir-airandmlir-aie. - Replace removed
transform.apply_patterns.vector.lower_multi_reductionwith its successor patterns across the transform library and example scripts. - Add explicit
--air-runtime-loop-tiling-sizesarguments to theairccinvocation to match the prior default behavior.
Reviewed changes
Copilot reviewed 9 out of 9 changed files in this pull request and generated 3 comments.
Show a summary per file
| File | Description |
|---|---|
| utils/mlir-air-hash.txt | Update pinned mlir-air commit/timestamp. |
| utils/mlir-aie-hash.txt | Update pinned mlir-aie commit/timestamp. |
| examples/test_layernorm/transform_aie2p.mlir | Swap deprecated multi-reduction lowering pattern for successor patterns. |
| examples/test_layernorm/transform_aie2.mlir | Swap deprecated multi-reduction lowering pattern for successor patterns. |
| examples/rms_norm/transform_aie2p.mlir | Swap deprecated multi-reduction lowering pattern for successor patterns. |
| examples/average_pool/transform_aie2p.mlir | Swap deprecated multi-reduction lowering pattern for successor patterns. |
| examples/average_pool/transform_aie2.mlir | Swap deprecated multi-reduction lowering pattern for successor patterns. |
| amd_triton_npu/backend/transform_library/vectorization.mlir | Update reduction cleanup sequence to use successor multi-reduction lowering patterns. |
| amd_triton_npu/backend/driver.py | Add explicit runtime loop tiling sizing flags to aircc invocation. |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
- Fix driver.py comment to say "to [4,4]" matching the actual CLI args - Put pattern + lowering_strategy on single lines in vectorization.mlir for consistency with the rest of the codebase - Update comment to match actual pattern ordering (cast_away first, then multi_reduction lowering) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
5798472and mlir-aie to3a1aba2(both timestamp2026032704) to pick up the LLVM 278dba37 updatetransform.apply_patterns.vector.lower_multi_reductionwith its three successor ops (reorder_multi_reduction_dims,multi_reduction_flattening,multi_reduction_unrolling) across the transform library and 6 example transform scripts--air-runtime-loop-tiling-sizes=4 4to aircc invocation in driver.py to match the new default (changed from[4,4]to[]in mlir-air #1470)Test plan
🤖 Generated with Claude Code