test: add 2nd tmem_store to column 1

2026-05-28 09:52:05 +00:00
parent 4b129c146e
commit 7a21fa4bd8
1 changed files with 1 additions and 0 deletions
--- a/tests/unit/test_tmem_cols.cu
+++ b/tests/unit/test_tmem_cols.cu
@@ -45,6 +45,7 @@ __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);
    }
    asm volatile("tcgen05.wait::st.sync.aligned;" ::: "memory");
    __syncthreads();