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

test: try idesc=0 (same as gau-nernst)

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

2
tests/unit/test_umma_qk.cu
View File

@@ -56,7 +56,7 @@ test_umma_qk_hd16(
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 = make_idesc(128, 128); // Try N=128 (full extent)
uint32_t idesc = 0; // Try idesc=0 (same as gau-nernst)
// Verify SMEM Q and K by reading back row 0
if (tid == 0) {