fix: missing newline after self.s_k = s_k

2026-05-22 10:20:35 +00:00
parent 1982cc4d39
commit 0bcb5aba2b
1 changed files with 2 additions and 1 deletions
--- 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