From eb5c538c9b30213f0800988ae7af247e135fb74a Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 09:02:26 +0000 Subject: [PATCH] Complete multi-PV-tile fixes: pv_n_tile, v_fmha layout, MMA construction, n_corr_tiles --- dsv4/kernels/attention/fmha.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/dsv4/kernels/attention/fmha.py b/dsv4/kernels/attention/fmha.py index c7f3a3bc..41c7b7c2 100644 --- a/dsv4/kernels/attention/fmha.py +++ b/dsv4/kernels/attention/fmha.py @@ -19,6 +19,8 @@ class FmhaKernel: self.head_dim = head_dim self.s_k = s_k self.n_kv_tiles = s_k // 128 + self.pv_n_tile = min(head_dim, 256) # tcgen05 MMA max N=256 + self.n_pv_tiles = head_dim // self.pv_n_tile self.pv_n_tile = min(head_dim, 256) # tcgen05 MMA max N=256 self.n_pv_tiles = head_dim // self.pv_n_tile