From c73cc58306750e572ebdc6f62695c3d5a03ccb58 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 22 May 2026 23:34:03 +0000 Subject: [PATCH] auto: pre-test commit --- tests/unit/test_fmha_v3.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/unit/test_fmha_v3.py b/tests/unit/test_fmha_v3.py index 69a7ab33..e65cc2aa 100644 --- a/tests/unit/test_fmha_v3.py +++ b/tests/unit/test_fmha_v3.py @@ -134,7 +134,16 @@ class FmhaV3: b_lay = cute.make_layout(cute.slice_(cl_vmnk,(0,None,0,0)).shape) tBsK,tBgK = cpasync.tma_partition(tma_k,0,b_lay,cute.group_modes(sK,0,3),cute.group_modes(tCgK,0,3)) tVsV,tVgV = cpasync.tma_partition(tma_v,0,b_lay,cute.group_modes(sV,0,3),cute.group_modes(tCgV,0,3)) + print(f"DIAG tAgQ: shape={cute.shape(tAgQ)} stride={tAgQ.layout.stride}") + print(f"DIAG tBgK: shape={cute.shape(tBgK)} stride={tBgK.layout.stride}") + print(f"DIAG tVgV: shape={cute.shape(tVgV)} stride={tVgV.layout.stride}") + print(f"DIAG tAsQ: shape={cute.shape(tAsQ)} stride={tAsQ.layout.stride}") + print(f"DIAG tBsK: shape={cute.shape(tBsK)} stride={tBsK.layout.stride}") + print(f"DIAG tVsV: shape={cute.shape(tVsV)} stride={tVsV.layout.stride}") tAgQ = tAgQ[(None,0,None,0)]; tBgK = tBgK[(None,0,None,0)]; tVgV = tVgV[(None,0,None,0)] + print(f"DIAG tAgQ sliced (None,0,None,0): shape={cute.shape(tAgQ)} stride={tAgQ.layout.stride}") + print(f"DIAG tBgK sliced (None,0,None,0): shape={cute.shape(tBgK)} stride={tBgK.layout.stride}") + print(f"DIAG tVgV sliced (None,0,None,0): shape={cute.shape(tVgV)} stride={tVgV.layout.stride}") tCrQ = qk_mma.make_fragment_A(sQ); tCrK = qk_mma.make_fragment_B(sK) tCrV = pv_mma.make_fragment_B(sV)