kiddo 6.0.0

A high-performance, flexible, ergonomic k-d tree library. Ideal for geo- and astro- nearest-neighbour and k-nearest-neighbor queries
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
35
36
37
38
39
40
41
42
43
44
45

#[cfg(all(feature = "simd", target_arch = "x86_64", target_feature = "avx2"))]
pub(crate) mod avx2 {
    use std::arch::x86_64::*;

    #[inline(always)]
    pub(crate) unsafe fn custom_mm256_abs_pd(x: __m256d) -> __m256d {
        let sign = _mm256_set1_pd(-0.0);
        _mm256_andnot_pd(sign, x)
    }

    #[inline(always)]
    pub(crate) unsafe fn custom_mm_abs_pd(x: __m128d) -> __m128d {
        let sign = _mm_set1_pd(-0.0);
        _mm_andnot_pd(sign, x)
    }

    #[inline(always)]
    pub(crate) unsafe fn custom_mm256_abs_ps(x: __m256) -> __m256 {
        let sign = _mm256_set1_ps(-0.0);
        _mm256_andnot_ps(sign, x)
    }

    #[inline(always)]
    pub(crate) unsafe fn custom_mm_abs_ps(x: __m128) -> __m128 {
        let sign = _mm_set1_ps(-0.0);
        _mm_andnot_ps(sign, x)
    }
}

#[cfg(all(feature = "simd", target_feature = "avx512f"))]
pub(crate) mod avx512 {
    use std::arch::x86_64::*;

    #[inline(always)]
    pub(crate) unsafe fn custom_mm512_abs_pd(x: __m512d) -> __m512d {
        let sign = _mm512_set1_pd(-0.0);
        _mm512_andnot_pd(sign, x)
    }

    #[inline(always)]
    pub(crate) unsafe fn custom_mm512_abs_ps(x: __m512) -> __m512 {
        let sign = _mm512_set1_ps(-0.0);
        _mm512_andnot_ps(sign, x)
    }
}