From d2d0eec33a5d4a27969dde25d786228a9b25367b Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 20:00:33 +0000 Subject: [PATCH] SMEM-P: add debug prints for coordinates --- dsv4/kernels/attention/fmha.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/dsv4/kernels/attention/fmha.py b/dsv4/kernels/attention/fmha.py index 23905d22..4039dbe5 100644 --- a/dsv4/kernels/attention/fmha.py +++ b/dsv4/kernels/attention/fmha.py @@ -372,6 +372,10 @@ class FmhaKernel: # Write actual P value (not test pattern) p_val_bf16 = tTMEM_LOADrS_frg[k, j].to(self.q_dtype) sP[pv_coord] = p_val_bf16 # Tensor indexing + + # DEBUG: Print first few coordinates to verify mapping + if self.use_smem_p and k < 2 and j < 2: + print(f"[SMEM-P DEBUG] k={k}, j={j}, qk_coord=({m},{n}), pv_coord={pv_coord}") row_sum = row_sum + tTMEM_LOADrS_frg[k, j] s_vec = tTMEM_LOADrS_frg[None, j].load()