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

test: try M=64 in descriptor + idesc to debug 4x factor

Browse Source
This commit is contained in:
biondizzle
2026-05-28 09:41:50 +00:00
parent 4cb656e583
commit 764ed01d6f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/unit/test_umma_qk.cu
View File

@@ -54,9 +54,9 @@ test_umma_qk_hd16(
// Descriptors
uint32_t sQ_smem = __cvta_generic_to_shared(sQ);
uint32_t sK_smem = __cvta_generic_to_shared(sK);
uint64_t desc_q = make_umma_desc_kmajor_none(sQ_smem, 128);
uint64_t desc_k = make_umma_desc_kmajor_none(sK_smem, 128);
uint32_t idesc = 0; // Try idesc=0 (same as gau-nernst)
uint64_t desc_q = make_umma_desc_kmajor_none(sQ_smem, 64); // Try M=64
uint64_t desc_k = make_umma_desc_kmajor_none(sK_smem, 64); // Try M=64
uint32_t idesc = make_idesc(64, 128); // M=64, N=128
// Verify SMEM Q and K by reading back row 0
if (tid == 0) {