From b61df947065c7084baba0c09e1f4c92c4bf4fc99 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 00:05:07 +0000 Subject: [PATCH] auto: pre-test commit --- tests/fmha_v3_stage_c_example9.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/fmha_v3_stage_c_example9.py b/tests/fmha_v3_stage_c_example9.py index ba1e6fa2..e025adfc 100644 --- a/tests/fmha_v3_stage_c_example9.py +++ b/tests/fmha_v3_stage_c_example9.py @@ -243,7 +243,7 @@ class FmhaV3StageCMulti: kvc.reset() acc_st = pipeline.make_pipeline_state(pipeline.PipelineUserType.Producer, self.num_acc_stage) acc_pipe.producer_acquire(acc_st) - for kt in range(n_kv_tiles): + for kt in range(self.n_kv_tiles): kvh = kvc.wait_and_advance() sh = s_prod.acquire_and_advance() qk_mma.set(tcgen05.Field.ACCUMULATE, False) @@ -321,7 +321,9 @@ class FmhaV3StageCMulti: scale_log2 = Float32(self.scale_softmax_log2) # Per-tile softmax loop with online rescale. - for kt in range(n_kv_tiles): + # Use self.n_kv_tiles (Python int) not the CuTeDSL symbolic n_kv_tiles + # — Python range() needs a Python int. + for kt in range(self.n_kv_tiles): si_handle = s_cons.wait_and_advance() cute.printf("SOFTMAX kt=%d row_max_before=%f row_sum_before=%f\n", kt, row_max, row_sum)