Skip to content

Add DSA native matmul dump support - #20

Merged
kateyijian merged 1 commit into
CommonIRfrom
poc-native-matmul-dsa
Jul 10, 2026
Merged

Add DSA native matmul dump support #20
kateyijian merged 1 commit into
CommonIRfrom
poc-native-matmul-dsa

Conversation

@macqueen09

Copy link
Copy Markdown
Collaborator

新增 test/CommonIR/native_matmul_dsa.py,作为 native_matmul 的 DSA 对照版本。

主要改动:

  1. 使用 tle.dsa.alloc / tle.dsa.copy / tle.dsa.to_tensor 替换原 tile buffer 路径
  2. 支持 --dump-ttir、--dump-tileir、--dump-linalg
  3. --dump-tileir 可导出包含 tile.alloc / tile.copy / tile.to_tensor 的前端 TileIR
  4. 本地验证 dump-ttir、dump-tileir、dump-linalg 均通过,linalg lowering 各阶段 verify=True

验证:
python test/CommonIR/native_matmul_dsa.py --dump-tileir /tmp/native_matmul_dsa_tileir.mlir --M 256 --N 256 --K 128 --num-cores 1
python test/CommonIR/native_matmul_dsa.py --dump-linalg /tmp/native_matmul_dsa_linalg.mlir --M 256 --N 256 --K 128 --num-cores 1
python -m pytest python/test/tle/test_bind_buffer.py -q

@kateyijian
kateyijian merged commit 245dc28 into CommonIR Jul 10, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants