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

Test K-tiles 0-1 accumulated

Browse Source
This commit is contained in:
biondizzle
2026-05-28 14:33:31 +00:00
parent bad31d9476
commit dacbf53081
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_pv_ss_128.cu
View File

@@ -72,7 +72,7 @@ test_pv_ss_128()
{
uint64_t dv = make_umma_desc_kmajor_none(__cvta_generic_to_shared(sV), 16);
uint32_t idesc = make_idesc(BLOCK_MN, HD);
for (int kt = 1; kt < 2; kt++) {
for (int kt = 0; kt < 2; kt++) {
bf16_t* sp = sP + kt * 2048;
uint64_t dp = make_umma_desc_kmajor_none(__cvta_generic_to_shared(sp), BLOCK_MN);
if (tid == 0) umma_ss_f16(tb, dp, dv, idesc, true); // ALL accumulate