fix: indentation error in diag disable

2026-05-23 01:11:25 +00:00
parent 26fb90bbba
commit 4712a0f90d
1 changed files with 3 additions and 2 deletions
--- a/tests/unit/test_fmha_v3_stage_c.py
+++ b/tests/unit/test_fmha_v3_stage_c.py
@@ -375,8 +375,9 @@ class FmhaV3StageCMulti:
                cute.arch.fence_view_async_tmem_store()

                # === Per-tile O rescale: O *= acc_scale for kt > 0 ===
-                # TEMP: disabled for diagnosis
-                if False and kt > 0:
+                # TEMP: disabled for diagnosis — O rescale not applied
+                if kt > 0:
+                    pass  # O rescale disabled
                if kt > 0:
                    for i in range(n_corr_tiles):
                        tTMEM_LOADtO_i = cute.make_tensor(