diff --git a/tests/unit/test_fmha_v3_stage_c.py b/tests/unit/test_fmha_v3_stage_c.py index fc629e7b..6253651f 100644 --- 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)