From 5a11f7c09ac6edbe7ef4874c7836181972d22b34 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 20:14:21 +0000 Subject: [PATCH] =?UTF-8?q?SMEM-P:=20test=20permutation=201=20(swap=20m?= =?UTF-8?q?=E2=86=94n0)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dsv4/kernels/attention/fmha.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dsv4/kernels/attention/fmha.py b/dsv4/kernels/attention/fmha.py index a2d0a43f..699f6d47 100644 --- a/dsv4/kernels/attention/fmha.py +++ b/dsv4/kernels/attention/fmha.py @@ -24,7 +24,7 @@ class FmhaKernel: self.use_smem_p = use_smem_p if use_smem_p is not None else (head_dim > 64) self.debug_p_one = False # DEBUG: write constant P=1.0 to verify mapping self.debug_swap_mn = False # DEBUG: try swapping m and n0 in coordinate mapping - self.debug_permute = 0 # DEBUG: try different coordinate permutations + self.debug_permute = 1 # DEBUG: try different coordinate permutations (0=original,1=swap m↔n0) self.acc_dtype = Float32; self.qk_acc_dtype = Float32 self.q_dtype = BFloat16; self.o_dtype = BFloat16; self.c_dtype = BFloat16 self.use_2cta_instrs = False; self.epilog_sync_bar_id = 1