biondizzle 6c73069cb9 D5b: Per-row LSE output + Python KV merge test ...

- Fix LSE output: all 128 rows now write (mLSE[sfw_idx, 0, 0])
  instead of only row 0 (mLSE[0])
- Each softmax thread (sfw_idx 0..127) independently writes its LSE
- This enables accurate Python-side KV merge for multi-KV-tile
- New test: test_d5b_perrow_lse.py with LSE verification + KV merge

2026-05-26 10:57:54 +00:00

..

__init__.py

Restructure: cutedsl/ -> dsv4/ with proper layering

2026-05-21 17:30:44 +00:00

fmha_backup_pre_epilog.py

D1.5: Replace TMEM round-trip normalize with correction epilog (one-way: TMEM→reg→SMEM→GMEM)

2026-05-24 00:24:24 +00:00

fmha.py

D5b: Per-row LSE output + Python KV merge test

2026-05-26 10:57:54 +00:00

fmha.py.backup

auto: pre-test commit

2026-05-23 20:08:31 +00:00