auto: pre-test commit

2026-05-23 00:29:49 +00:00
parent 17f80de485
commit fc5b6811d2
1 changed files with 2 additions and 0 deletions
--- 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)