From 4712a0f90de80ea897be7306094c09dd140f1cdc Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sat, 23 May 2026 01:11:25 +0000
Subject: [PATCH] fix: indentation error in diag disable

---
 tests/unit/test_fmha_v3_stage_c.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/unit/test_fmha_v3_stage_c.py b/tests/unit/test_fmha_v3_stage_c.py
index bf4fe9ed..51360a9e 100644
--- 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(