diskann-wide 0.50.0

DiskANN is a fast approximate nearest neighbor search library for high dimensional data
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

/*
 * Copyright (c) Microsoft Corporation.
 * Licensed under the MIT license.
 */

// x86 intrinsics
use std::arch::x86_64::*;

/// A trait to get a SIMD intrinsic completely full of 1's
pub(crate) trait AllOnes {
    fn all_ones() -> Self;
}

impl AllOnes for __m128i {
    fn all_ones() -> Self {
        // SAFETY: `_mm_set1_epi32` requires SSE2, which is baseline for x86_64.
        unsafe { _mm_set1_epi32(-1) }
    }
}

impl AllOnes for __m256i {
    fn all_ones() -> Self {
        // SAFETY: `_mm256_set1_epi32` requires AVX, implied by the caller's architecture.
        unsafe { _mm256_set1_epi32(-1) }
    }
}

impl AllOnes for __m512i {
    fn all_ones() -> Self {
        // SAFETY: `_mm512_set1_epi32` requires AVX-512F, implied by the caller's architecture.
        unsafe { _mm512_set1_epi32(-1) }
    }
}