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(