From 59746b46fc2e80d21fb1411ab5ee24329280f1bc Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 22 May 2026 17:30:06 +0000 Subject: [PATCH] DEBUG: try K slice (None,0,None,0) keeping mode 2 free --- tests/unit/test_fmha_v3_stage_c_full.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_fmha_v3_stage_c_full.py b/tests/unit/test_fmha_v3_stage_c_full.py index 7976bc5b..04526b9a 100644 --- a/tests/unit/test_fmha_v3_stage_c_full.py +++ b/tests/unit/test_fmha_v3_stage_c_full.py @@ -174,7 +174,7 @@ class FmhaV3StageC: print(f"DEBUG tBgK shape before slice: {tBgK.shape}") print(f"DEBUG tVgV shape before slice: {tVgV.shape}") tAgQ = tAgQ[(None,0,None,0)] # Q: 1 tile, hardcode is fine - tBgK = tBgK[(None,None,0,0)] # K: keep mode 1 (GMEM iter) free + tBgK = tBgK[(None,0,None,0)] # K: try keeping mode 2 free (like V) tVgV = tVgV[(None,0,None,0)] # V: keep mode 2 (GMEM iter) free print(f"DEBUG tBgK shape after slice: {tBgK.shape}") print(f"DEBUG tVgV shape after slice: {tVgV.shape}")