SMEM-P: fix coordinate access - qk_coord is (m,n) not ((m,n),0,0)

2026-05-23 19:38:11 +00:00
parent 096a48b5cb
commit ca1fb7a1ca
1 changed files with 3 additions and 3 deletions
--- a/dsv4/kernels/attention/fmha.py
+++ b/dsv4/kernels/attention/fmha.py
@@ -349,9 +349,9 @@ class FmhaKernel:
                        if self.use_smem_p:
                            # Get QK coordinate for this position
                            qk_coord = tTMEM_LOADcS_frg[k, j]
-                            mn = qk_coord[0]
-                            m = mn[0]
-                            n = mn[1]
+                            # qk_coord is (m, n) coordinate
+                            m = qk_coord[0]
+                            n = qk_coord[1]
                            
                            # Map to PV SMEM coordinate
                            n0 = n % 16