Issues

• Introduce GatherToLDS MLIR operation

  iree-org/wave#1290 · Filter by author martin-luecke opened on Apr 10, 2026

• [water] check dynamic dimension support for mxfp4

  iree-org/wave#1268 · Filter by author ftynse opened on Apr 7, 2026

• [water] scheduling pass

  iree-org/wave#1267 · Filter by author ftynse opened on Apr 7, 2026
  1

• [wave] NSA: SGLang / inference backend integration

  enhancementnsa
  iree-org/wave#1263 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: water dialect lowering for NSA composite ops

  enhancementnsa
  iree-org/wave#1262 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: performance benchmarks on MI350

  enhancementnsa
  iree-org/wave#1261 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: gradient correctness tests (backward pass)

  enhancementnsa
  iree-org/wave#1260 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: numerical correctness tests vs dense attention reference

  enhancementnsa
  iree-org/wave#1259 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: GQA-aware memory layout & tiling for MI350 cache hierarchy

  enhancementnsa
  iree-org/wave#1258 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: LDS optimization for block-gather in selection attention

  enhancementnsa
  iree-org/wave#1257 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: CDNA4 wavefront scheduling for sparse gather patterns

  enhancementnsa
  iree-org/wave#1256 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: end-to-end training integration with autograd

  enhancementnsa
  iree-org/wave#1255 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: gating parameter gradients & top-k STE backward

  enhancementnsa
  iree-org/wave#1254 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: compressed attention backward kernel

  enhancementnsa
  iree-org/wave#1253 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: selection attention backward kernel (dQ, dK, dV)

  enhancementnsa
  iree-org/wave#1252 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: end-to-end inference pipeline

  enhancementnsa
  iree-org/wave#1251 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: gated output combination kernel

  enhancementnsa
  iree-org/wave#1250 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: sliding window attention forward kernel

  enhancementnsa
  iree-org/wave#1249 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: selection attention forward kernel

  enhancementnsa
  iree-org/wave#1248 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: top-k block selection kernel

  enhancementnsa
  iree-org/wave#1247 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: compressed attention forward kernel

  enhancementnsa
  iree-org/wave#1246 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: token compression kernel (mean pooling KV blocks)

  enhancementnsa
  iree-org/wave#1245 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] NSA: architecture mapping & design doc for MI350

  enhancementnsa
  iree-org/wave#1244 · Filter by author harsh-nod opened on Apr 4, 2026

• [wave] DeepSeek Native Sparse Attention (NSA) kernels for MI350

  nsatracker
  iree-org/wave#1243 · Filter by author harsh-nod opened on Apr 4, 2026

• [water] Use InferIndexExprsOpInterface descriptions in ScaledMmaOp EPT propagation

  iree-org/wave#1189 · Filter by author tgymnich opened on Mar 25, 2026