From fc5b6811d22ab333f05600932b5e7b7a6b4cd91c Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 00:29:49 +0000 Subject: [PATCH] auto: pre-test commit --- tests/unit/test_fmha_v3_stage_c.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/unit/test_fmha_v3_stage_c.py b/tests/unit/test_fmha_v3_stage_c.py index c96d8b7d..ff347d50 100644 --- a/tests/unit/test_fmha_v3_stage_c.py +++ b/tests/unit/test_fmha_v3_stage_c.py @@ -323,6 +323,8 @@ class FmhaV3StageCMulti: # the missing rescale shows as accuracy drift. for kt in range(n_kv_tiles): si_handle = s_cons.wait_and_advance() + if kt == 0: + cute.printf("SOFTMAX n_kv_tiles=%d\n", Int32(n_kv_tiles)) # Load S[kt] tTMEM_LOADrS = cute.make_rmem_tensor(tTMEM_LOADcS.shape, self.qk_acc_dtype)