diff --git a/tests/unit/test_tmem_budget.py b/tests/unit/test_tmem_budget.py index c9dc694f..a1eb3a96 100644 --- a/tests/unit/test_tmem_budget.py +++ b/tests/unit/test_tmem_budget.py @@ -25,10 +25,6 @@ def probe_hd(hd): mQ = ct.from_dlpack(q).mark_layout_dynamic(leading_dim=ct.get_leading_dim(q)) mK = ct.from_dlpack(k).mark_layout_dynamic(leading_dim=ct.get_leading_dim(k)) - # V with FMHA layout: (head_dim, s_k, 1) - v_fmha_layout = cute.make_layout( - (hd, n, 1), stride=(1, hd, hd * n), - ) # Get major modes from the actual tensors a_major = LayoutEnum.from_tensor(mQ).mma_major_mode() b_major = LayoutEnum.from_tensor(mK).mma_major_mode()