(a: &[f32], b: &[f32])
| 5 | #![cfg(target_arch = "x86_64")] |
| 6 | |
| 7 | pub fn l2_squared(a: &[f32], b: &[f32]) -> f32 { |
| 8 | assert_eq!(a.len(), b.len(), "avx512 l2: length mismatch"); |
| 9 | unsafe { l2_impl(a, b) } |
| 10 | } |
| 11 | |
| 12 | #[target_feature(enable = "avx512f")] |
| 13 | unsafe fn l2_impl(a: &[f32], b: &[f32]) -> f32 { |