From 9bcddb68e103cb13aa8957bdc0e2d9cdd68d1e98 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 01:12:50 +0000 Subject: [PATCH] diag: disable O rescale properly, test n=128+256 baseline --- tests/unit/test_fmha_v3_stage_c.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/unit/test_fmha_v3_stage_c.py b/tests/unit/test_fmha_v3_stage_c.py index 30552aa8..4651ead1 100644 --- a/tests/unit/test_fmha_v3_stage_c.py +++ b/tests/unit/test_fmha_v3_stage_c.py @@ -375,10 +375,8 @@ class FmhaV3StageCMulti: cute.arch.fence_view_async_tmem_store() # === Per-tile O rescale: O *= acc_scale for kt > 0 === - # TEMP: disabled for diagnosis — O rescale not applied - if kt > 0: - pass # O rescale disabled - if kt > 0: + # DIAG: O rescale DISABLED — testing baseline without rescale + if False: for i in range(n_corr_tiles): tTMEM_LOADtO_i = cute.make_tensor( tTMEM_LOADtO.iterator + i * corr_tile_size,