biondizzle 15a2fdadbc FIX: consistent GMEM/SMEM slicing for K and V TMA partitions ...

Both GMEM and SMEM sides must be sliced to the same rank for cute.copy.
K (QK MMA B-partition): slice [(None,None,0,0)] keeps modes 0,1
  - mode 1 = GMEM iteration, indexed by kvh.count
V (PV MMA B-partition): slice [(None,0,None,0)] keeps modes 0,2
  - mode 2 = GMEM iteration, indexed by kvh.count
Q: only 1 tile, (None,0,None,0) hardcode is fine.

2026-05-22 16:56:38 +00:00

..

archive

Restructure: cutedsl/ -> dsv4/ with proper layering

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

e2e

Restructure: cutedsl/ -> dsv4/ with proper layering

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

integration

Restructure: cutedsl/ -> dsv4/ with proper layering

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

unit

FIX: consistent GMEM/SMEM slicing for K and V TMA partitions

2026-05-22 16:56:38 +00:00

fmha_v3_stage_c_example1.py

restore tBgK to kh.count indexing (single-tile working), add TODO for multi-tile

2026-05-22 15:54:03 +00:00

fmha_v3_stage_c_example2.py

FMHA Stage-C multi-tile: combined K+V barrier, final_o_bar, acc_pipe producer

2026-05-22 16:23:36 +00:00

fmha_v3_stage_c_example3.py

Stage C: integrate example3 multi-tile fixes into unit test

2026-05-22 16:39:45 +00:00

native_stage_c_patch.py

more stuff

2026-05-22 08:57:38 +00:00

requirements.txt

test: add standalone layer 0 comparison test (no vLLM, no Docker)

2026-05-16 02:13:18 +00:00

run_test.sh

fix: use setup.py install for CUTLASS extension build

2026-05-16 02:21:17 +00:00