DEBUG: add cute.printf for kv_coord runtime value

2026-05-22 20:33:03 +00:00
parent b8d4ac0ecb
commit da4eebc9a4
1 changed files with 1 additions and 0 deletions
--- a/tests/unit/test_fmha_v3_stage_c.py
+++ b/tests/unit/test_fmha_v3_stage_c.py
@@ -226,6 +226,7 @@ class FmhaV3StageCMulti:
            kvp.reset()
            kv_coord = n_kv_tiles - n_kv_tiles  # SSA runtime zero
            for kt in range(n_kv_tiles):
+                cute.printf("TMA kt=%d kv_coord=%d\n", kt, kv_coord)
                kvh = kvp.acquire_and_advance()
                cute.copy(tma_k, tBgK[(None, kv_coord)], tBsK[(None, kvh.index)], tma_bar_ptr=kvh.barrier)
                cute.copy(tma_v, tVgV[(None, kv_coord)], tVsV[(None, kvh.index)], tma_bar_ptr=kvh.barrier)