diff --git a/tests/unit/test_tmem_cols.cu b/tests/unit/test_tmem_cols.cu index c8c21c23..bf957b4b 100644 --- a/tests/unit/test_tmem_cols.cu +++ b/tests/unit/test_tmem_cols.cu @@ -49,16 +49,16 @@ __global__ void test_tmem_loop(float* out) { asm volatile("tcgen05.wait::st.sync.aligned;" ::: "memory"); __syncthreads(); - // Read back - if (threadIdx.x < 32) { - for (int c = 0; c < 4; c++) { - uint32_t u0, u1, u2, u3; - tmem_load(tb + c, u0, u1, u2, u3); - float v0; memcpy(&v0, &u0, 4); - if (lane == 0) out[c] = v0; - } - } - __syncthreads(); + // Read back — skip for now + // if (threadIdx.x < 32) { + // for (int c = 0; c < 4; c++) { + // uint32_t u0, u1, u2, u3; + // tmem_load(tb + c, u0, u1, u2, u3); + // float v0; memcpy(&v0, &u0, 4); + // if (lane == 0) out[c] = v0; + // } + // } + // __syncthreads(); if (threadIdx.x < 32) tmem_dealloc(tb, 32); }