MLIR `linalg` Dialect Lowering for Matrix Multiplication
Nobody has claimed this yet.
- Dominant language
- Rust
- Stars
- 14
- Forks
- 2
- Avg merge
- 12h 42m
- Merged PRs (30d)
- 61
Description
Right now, benchmark hardware dispatch testing (e.g. tests/backend/pass/plugin_npe.vx) uses a dummy flat C-ABI dispatch test via spawn on(Topology::ANE).
Objective:
To officially wire the standard tensor::matmul operations inside spawn on blocks to lower directly to proper linalg.matmul MLIR operations, allowing proper pass handoff to hardware vendor plugins.
Tasks:
- Update the MLIR code generator to recognize matrix multiplication inside
spawn onregions. - Ensure the emitted IR uses
linalg.matmulwith the proper memref indices and memory spaces. - Validate that the NPU plugin can successfully ingest and lower this
linalgblock to native NPU commands.
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with tests/backend/pass/plugin_npe.vx and trace how tensor::matmul inside spawn on(Topology::ANE) reaches the MLIR code generator. Compare the emitted IR with the expected linalg.matmul form, including memref indices and memory spaces, then validate that the NPU plugin ingests and lowers the resulting block to native NPU commands.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- rust
- Domain
- compilers
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Quiet
- Clarity
- Mostly clear
- Newbie friendliness
- 42/100