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

test: only 1 tmem_store to verify single column works

Browse Source
This commit is contained in:
biondizzle
2026-05-28 09:49:21 +00:00
parent 5795589abc
commit abfe9dbaa1
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_tmem_cols.cu
View File

@@ -45,9 +45,9 @@ __global__ void test_tmem_loop(float* out) {
memcpy(&u2, &v2, 4); memcpy(&u3, &v3, 4);
tmem_store(tb + 0, u0, u1, u2, u3);
tmem_store(tb + 1, u0, u1, u2, u3);
tmem_store(tb + 2, u0, u1, u2, u3);
tmem_store(tb + 3, u0, u1, u2, u3);
// tmem_store(tb + 1, u0, u1, u2, u3); // Skip for now
// tmem_store(tb + 2, u0, u1, u2, u3); // Skip for now
// tmem_store(tb + 3, u0, u1, u2, u3); // Skip for now
}
__syncthreads();