Function rope_half_partial 

pub fn rope_half_partial(
    x: &mut [f32],
    head_dim: usize,
    rotary_dim: usize,
    pos: usize,
    theta: f32,
) -> Result<(), EngineError>

Rotate only the first rotary_dim dims of each head (partial_rotary_factor).