vx-lang / vx-lang/Vx

MLIR `linalg` Dialect Lowering for Matrix Multiplication

Open
#37 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

codegen enhancement
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 on regions.
  • Ensure the emitted IR uses linalg.matmul with the proper memref indices and memory spaces.
  • Validate that the NPU plugin can successfully ingest and lower this linalg block to native NPU commands.

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.