Debug: check if HD=64 kernel starts

2026-05-28 15:02:00 +00:00
parent 46e4d07c71
commit 2d44f8e356
1 changed files with 1 additions and 0 deletions
--- a/tests/unit/test_fmha_hd64_smem_p.cu
+++ b/tests/unit/test_fmha_hd64_smem_p.cu
@@ -32,6 +32,7 @@ __global__ void __launch_bounds__(128)
 test_fmha_hd64_smem_p(const bf16_t* q, const bf16_t* k, const bf16_t* v,
                       bf16_t* o_out, float* o_scalar, float scale)
 {
+    if (tid == 0) printf("Kernel started! smem_buf=%p\n", sbuf);
    const int tid = threadIdx.x, wid = tid / 32, lane = tid % 32;

    extern __shared__ char sbuf[];