iree-org/wave

View on GitHub

Wave: Python Domain-Specific Language for High Performance Machine Learning

Stars: 59
Forks: 32
Open beginner issues: 0
Indexed issues: 93

Dominant language: Python
License: Apache-2.0
Last GitHub push: Apr 27, 2026
Latest indexed: Sep 17, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: good first issue help wanted
PR merge metrics: No merged PRs in 30d

Open All

93 open issues indexed

Core kernels to implement for Wave Open

iree-org/wave#63 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Investigate tensor leaks Open

iree-org/wave#67 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Slow compile times for large kernels Open

iree-org/wave#92 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
GPU error encountered while running sglang/test_wave_attention_backend.py Open

iree-org/wave#97 · 7 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
SGLang integration/maintenance tasks Open

tracker

iree-org/wave#106 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 17, 2026
FAv3 Open

tracker

iree-org/wave#142 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Paged decode failure cases on MI350X Open

iree-org/wave#143 · 4 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Allow tkw.iterate to emit scf.for without reduction Open

iree-org/wave#171 · 0 comments · 1 reaction · 0 assignees · Indexed Sep 17, 2026
tkw.broadcast after tkw.iterate loop does not inherit correct IndexSequence / vector shape Open

iree-org/wave#172 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Implement codegen for optimized RMSNorm kernel in LLaMA4 Open

iree-org/wave#173 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
MoE Tracker Open

tracker

iree-org/wave#189 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Language Feature Request : Hoisting out of wave.iterate Open

iree-org/wave#199 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
`tkw.debug_log` should check that `register_` is `tkl.Register` Open

enhancement

iree-org/wave#208 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 17, 2026
Reductions only work on `workgroup_dim` 0 Open

iree-org/wave#211 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Unnecessary amdgpu.lds_barrier in GEMM kernel Open

iree-org/wave#230 · 0 comments · 1 reaction · 0 assignees · Indexed Sep 17, 2026
Capture file/line/col information by default Open

iree-org/wave#304 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Test failure due to (presumed) race condition Open

iree-org/wave#314 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Fix the semantics of Sympy `/` operator Open

iree-org/wave#337 · 11 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Better support for unaligned shapes Open

iree-org/wave#458 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
[SGLang] Wave runtime crashes when cache is turned off Open

iree-org/wave#459 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
[SGLang] Llama3.1 model crashes with cuda graphs Open

iree-org/wave#466 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 17, 2026
[SGLang] Wave backend fails accuracy threshold for Llama3.1 model on `mmlu` eval Open

iree-org/wave#477 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
igemm_conv e2e tests are flaky Open

iree-org/wave#519 · 1 comment · 0 reactions · 1 assignee · Indexed Sep 17, 2026
[water] remove custom gpu-to-binary pass Open

iree-org/wave#543 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 17, 2026
E2E execution using water pipeline Open

iree-org/wave#581 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 17, 2026
Water pipeline e2e execution using IREE runtime Open

iree-org/wave#582 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 17, 2026
Water pipeline e2e execution using ExecutionEngine Open

iree-org/wave#583 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 17, 2026
Contain the dependency on IREE Open

iree-org/wave#611 · 1 comment · 0 reactions · 1 assignee · Indexed Sep 17, 2026
Bring the CI time to <10 minutes Open

iree-org/wave#612 · 7 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Clean up driver dynamic library logic Open

iree-org/wave#635 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Investigate race-y tests Open

iree-org/wave#651 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Port passes preceding index sequence analysis to MLIR Open

tracker

iree-org/wave#681 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
[Wave-OSS] Wave OSS features 2026Q1 Open

good first issue help wanted

iree-org/wave#739 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
PyTorch 2.10 Support Open

iree-org/wave#766 · 3 comments · 0 reactions · 1 assignee · Indexed Sep 17, 2026
Pin Rust Toolchain Open

iree-org/wave#769 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Minor cleanup: introduce a trait for equal element types Open

iree-org/wave#805 · 0 comments · 1 reaction · 2 assignees · Indexed Sep 17, 2026
document the difference between permute and reshape Open

iree-org/wave#817 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
investigate waitcnt increase in 8c9c0124501bb26282b0ac03810cb974ffe0ef0f Open

iree-org/wave#819 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 17, 2026
iree dependencies in pyproject.toml are broken Open

iree-org/wave#821 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
investigate long lit test times on CI Open

iree-org/wave#838 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
bounds not generated/exported for loop induction variable Open

iree-org/wave#841 · 1 comment · 0 reactions · 1 assignee · Indexed Sep 17, 2026
Bind operation-specific methods on operation classes instead of dialect module Open

iree-org/wave#855 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Add MLIR verifier to check HardwareConstraint.vector_shapes consistency with tile sizes Open

iree-org/wave#864 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Model `start` and `condition` in MLIR `IterateOp` Open

iree-org/wave#870 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
Unify attribute conversion registry between FX → MLIR and MLIR → FX directions Open

iree-org/wave#872 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
in-thread transpose creates incorrect reshape operations Open

iree-org/wave#873 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 17, 2026
[water] make index attribute verifier check that it provides index mappings for all symbols used in the operation Open

iree-org/wave#878 · 0 comments · 1 reaction · 2 assignees · Indexed Sep 17, 2026
attempt to solve the Type/TypeConstraint conundrum locally Open

iree-org/wave#889 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
consider having a WaveExprAttr that is not a list Open

iree-org/wave#898 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026
`apply_expr` typing is broken for comparisons Open

iree-org/wave#901 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 17, 2026

iree-org/wave

Get new issues in your inbox