pub fn kda_chunked(
q: &[f32],
k: &[f32],
v: &[f32],
gate: &[f32],
beta: &[f32],
dims: KdaDims,
chunk: usize,
state: &mut [f32],
) -> Vec<f32>Expand description
2026-09-25: Prefill formulation: the chunked (WY-style) delta rule with a per-channel decay
mask, checked against the golden’s chunk_kimi_delta_attention output. tokens % chunk != 0
is handled by zero-padding.