pub fn build_axial_rope_tables(
ids: &[(u32, u32)],
inv_freq: &[f32],
head_dim: usize,
) -> (Vec<f32>, Vec<f32>)Expand description
2026-09-25: (cos, sin) tables of [num_tokens, head_dim] f32, in the layout
glm_vit_qknorm_rope_deint reads.
With inv_freq of head_dim/4 entries, the per-token angle row is
cat([h*inv_freq, w*inv_freq]) (length head_dim/2), written twice to
fill head_dim. The kernel rotates every dimension of the head.