From 3aa597460e99f78149691f01709b57d21e4e4f0e Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 22 May 2026 09:30:32 +0000 Subject: [PATCH] fix: add missing old_row_max = row_max before softmax max computation --- tests/unit/test_fmha_v3_stage_c_full.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/unit/test_fmha_v3_stage_c_full.py b/tests/unit/test_fmha_v3_stage_c_full.py index 69108df8..6026fb9f 100644 --- a/tests/unit/test_fmha_v3_stage_c_full.py +++ b/tests/unit/test_fmha_v3_stage_c_full.py @@ -248,6 +248,7 @@ class FmhaV3StageC: cute.arch.fence_view_async_tmem_load() # Compute row_max from S (element-wise, before exp2) + old_row_max = row_max frg_cnt = 4 frg_tile = cute.size(tTMEM_LOADrS) // frg_cnt tTMEM_LOADrS_frg = cute.logical_divide(tTMEM_LOADrS, cute.make_layout(frg_tile))