From da4eebc9a49a92aa027a76f2bc6caa16d49f950f Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 22 May 2026 20:33:03 +0000 Subject: [PATCH] DEBUG: add cute.printf for kv_coord runtime value --- tests/unit/test_fmha_v3_stage_c.py | 1 + 1 file changed, 1 insertion(+) 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)