From 0bcb5aba2b10decc08056c9fa3e2a66df16fc188 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 22 May 2026 10:20:35 +0000 Subject: [PATCH] fix: missing newline after self.s_k = s_k --- tests/unit/test_fmha_v3_stage_c_full.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_fmha_v3_stage_c_full.py b/tests/unit/test_fmha_v3_stage_c_full.py index 93706a73..921420dd 100644 --- a/tests/unit/test_fmha_v3_stage_c_full.py +++ b/tests/unit/test_fmha_v3_stage_c_full.py @@ -17,7 +17,8 @@ HEAD_DIM = 64 class FmhaV3StageC: def __init__(self, s_k=128, scale_softmax=None): - self.s_k = s_k self.acc_dtype = Float32; self.qk_acc_dtype = Float32 + self.s_k = s_k + self.acc_dtype = Float32; self.qk_acc_dtype = Float32 self.q_dtype = BFloat16; self.o_dtype = BFloat16; self.c_dtype = BFloat16 self.use_2cta_instrs = False; self.epilog_sync_bar_id = 1 self.cluster_shape_mn = (1, 1); self.cta_group = tcgen05.CtaGroup.ONE