Fix tb scope

2026-05-28 14:40:55 +00:00
parent a6f7ef7c45
commit 342193e0b4
1 changed files with 2 additions and 1 deletions
--- a/tests/unit/test_minimal_pv.cu
+++ b/tests/unit/test_minimal_pv.cu
@@ -81,10 +81,11 @@ test_minimal_pv()
    }
    __syncthreads();

+    uint32_t tb = 0;
    // TMEM alloc
    if (wid == 1) tmem_alloc(__cvta_generic_to_shared(sTmemBase), 128);
    __syncthreads();
-    uint32_t tb = *sTmemBase;
+    tb = *sTmemBase;

    // QK GEMM (SS) — uses sQ0 and sK0
    {