pub fn attention_sink_decode_f32(
query: &[f32],
key: &[f32],
value: &[f32],
sinks: &[f32],
batch: usize,
key_len: usize,
heads: usize,
kv_heads: usize,
head_dim: usize,
start_q: usize,
window: usize,
scale: f32,
) -> Vec<f32>