biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: remove unused v_fmha_layout from probe

Browse Source
This commit is contained in:
biondizzle
2026-05-23 06:38:08 +00:00
parent de277f08ac
commit 25a08f5de2
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/unit/test_tmem_budget.py
View File

@@ -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()