From c815560d646ffee699a5300e692cc6ad6fc50024 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sun, 24 May 2026 01:54:20 +0000 Subject: [PATCH] fix: hardcode v_major for diag test --- tests/unit/test_cotiled_diag.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/tests/unit/test_cotiled_diag.py b/tests/unit/test_cotiled_diag.py index e58c106a..199a274d 100644 --- a/tests/unit/test_cotiled_diag.py +++ b/tests/unit/test_cotiled_diag.py @@ -42,15 +42,9 @@ def main(): a_major = LayoutEnum.from_tensor(mQ).mma_major_mode() b_major = LayoutEnum.from_tensor(mK).mma_major_mode() - # V FMHA layout (same as FmhaKernel.__call__) - v_fmha = cute.make_tensor( - mV.iterator, - cute.make_layout( - (pv_n_tile, s_k, 1), - stride=(1, pv_n_tile, pv_n_tile * s_k), - ), - ) - v_major = LayoutEnum.from_tensor(v_fmha).mma_major_mode() + # V FMHA layout: use explicit constants + # pv_n_tile=256, s_k=128 + v_major = LayoutEnum.COLUMN_MAJOR # layout (256, 128, 1) stride (1, 256, 32768) = col-major c_layout = LayoutEnum.from_tensor(mC)