pub fn fused_moe_f32( input: &[f32], weight: &[f32], routed_weight: &[f32], topk_ids: &[i32], tokens: usize, top_k: usize, columns: usize, reduction: usize, mul_routed_weight: bool, ) -> Vec<f32>