neo_frizbee 0.7.1

Fast fuzzy matching via SIMD smith waterman, similar algorithm to FZF/FZY
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

use std::simd::{LaneCount, Simd, SupportedLaneCount};

#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
pub mod avx2;
#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
pub mod avx512;
pub mod generic;

#[inline(never)]
pub fn interleave<const W: usize, const L: usize>(strs: [&str; L]) -> [Simd<u16, L>; W]
where
    LaneCount<L>: SupportedLaneCount,
{
    match L {
        #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
        32 if is_x86_feature_detected!("avx512f")
            && is_x86_feature_detected!("avx512bw")
            && is_x86_feature_detected!("avx2") =>
        unsafe {
            let strs_ptr = &strs as *const [&str; L] as *const [&str; 32];
            let result = avx512::interleave::<W>(*strs_ptr);
            let result_ptr = &result as *const [Simd<u16, 32>; W] as *const [Simd<u16, L>; W];
            std::ptr::read(result_ptr)
        },
        #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
        16 if is_x86_feature_detected!("avx2") => unsafe {
            let strs_ptr = &strs as *const [&str; L] as *const [&str; 16];
            let result = avx2::interleave::<W>(*strs_ptr);
            let result_ptr = &result as *const [Simd<u16, 16>; W] as *const [Simd<u16, L>; W];
            std::ptr::read(result_ptr)
        },
        _ => generic::interleave::<W, L>(strs),
    }
}