From c989dc78d91b7cfca89eda8539d6f9c5e0fe01e1 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Thu, 28 May 2026 07:47:37 +0000 Subject: [PATCH] debug: print sPvBuf[32] value --- dsv4/kernels/attention/fmha_epilogue_sm100.cuh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dsv4/kernels/attention/fmha_epilogue_sm100.cuh b/dsv4/kernels/attention/fmha_epilogue_sm100.cuh index 988bcfba..fd44c954 100644 --- a/dsv4/kernels/attention/fmha_epilogue_sm100.cuh +++ b/dsv4/kernels/attention/fmha_epilogue_sm100.cuh @@ -207,7 +207,7 @@ fmha_decode_tmem( sRowSums[0] = row_sum; } __syncthreads(); - if (tid == 0) printf("[tmem] attention computed, row_sum=%f, max=%f\n", sRowSums[0], sPvBuf[0]); + if (tid == 0) printf("[tmem] attention computed, row_sum=%f, sPvBuf[0]=%f, sPvBuf[32]=%f\n", sRowSums[0], sPvBuf[0], HD>32?sPvBuf[32]:0.0f); // ================================================================ // One-way Correction Epilogue: SMEM → TMEM → regs → normalize → GMEM