文件 simdlib_avx512.h
-
namespace faiss
实现具有许多变体的 k-means 聚类。
版权所有 (c) Facebook, Inc. 及其附属公司。
此源代码根据 MIT 许可证获得许可，该许可证位于此源树的 LICENSE 文件中。
IDSelector 旨在定义要处理的向量子集（用于删除或作为搜索的子集）
PQ4 SIMD 打包和累加函数
基本内核使用 bbs = nb * 2 * 16 向量累加 nq 查询向量，并为此生成一个输出矩阵。 对于 nq * nb <= 4 很有用，否则寄存器溢出变得太大。
这些函数的实现分布在 3 个 cpp 文件中，以减少并行编译时间。 模板被显式实例化。
此文件包含用于计算距离的内核的回调。
在整个库中，向量以 float * 指针的形式提供。 当在批处理中一起处理（添加/搜索）多个向量时，大多数算法都可以优化。 在这种情况下，它们作为矩阵传入。 当大小为 d 的 n 个向量作为 float * x 提供时，向量 i 的分量 j 为
x[ i * d + j ]
其中 0 <= i < n 且 0 <= j < d。 换句话说，矩阵始终是紧凑的。 指定矩阵的大小时，我们将其称为 n*d 矩阵，这意味着行优先存储。
I/O 函数可以读取/写入文件名、文件句柄或抽象介质的对象。
读取函数返回应使用 delete 释放的对象。 这些对象中的所有引用都归该对象所有。
倒排列表的定义 + 一些实现该接口的常见类。
由于 IVF（倒排文件）索引对于大规模用例非常有用，因此我们将与它们相关的一些函数分组到这个小库中。 大多数函数都适用于 IndexIVF 和嵌入在 IndexPreTransform 中的 IndexIVF。
在此文件中，实现了 L2 和内积之外的额外指标
实现了一些神经网络层，主要用于支持 QINCo
定义一些将变换应用于一组向量的对象 通常这些是预处理步骤。
函数
-
inline simd16uint16 combine4x2(simd32uint16 a, simd32uint16 b)
-
struct simd512bit
- #include <simdlib_avx512.h>
没有解释为向量的 512 位表示
AVX 512 位寄存器的简单包装器
目标是分离相同寄存器的不同解释（作为 uint8、uint16 或 uint32 的向量），提供打印函数，并为 AVX 内在函数提供更具可读性的名称。 它并不声称是详尽无遗的，函数是根据需要添加的。
被 faiss::simd32uint16, faiss::simd64uint8 继承
公共函数
-
inline simd512bit()
-
inline explicit simd512bit(__m512i i)
-
inline explicit simd512bit(__m512 f)
-
inline explicit simd512bit(const void *x)
-
inline explicit simd512bit(simd256bit lo)
-
inline explicit simd512bit(simd256bit lo, simd256bit hi)
-
inline void clear()
-
inline void storeu(void *ptr) const
-
inline void loadu(const void *ptr)
-
inline void store(void *ptr) const
-
inline void bin(char bits[513]) const
-
inline simd512bit()
-
struct simd32uint16 : public faiss::simd512bit
- #include <simdlib_avx512.h>
uint16 中的 32 个元素的向量
公共函数
-
inline simd32uint16()
-
inline explicit simd32uint16(__m512i i)
-
inline explicit simd32uint16(int x)
-
inline explicit simd32uint16(uint16_t x)
-
inline explicit simd32uint16(simd512bit x)
-
inline explicit simd32uint16(const uint16_t *x)
-
inline explicit simd32uint16(simd256bit lo)
-
inline explicit simd32uint16(simd256bit lo, simd256bit hi)
-
inline void set1(uint16_t x)
-
inline simd32uint16 operator*(const simd32uint16 &other) const
-
inline simd32uint16 operator>>(const int shift) const
-
inline simd32uint16 operator<<(const int shift) const
-
inline simd32uint16 operator+=(simd32uint16 other)
-
inline simd32uint16 operator-=(simd32uint16 other)
-
inline simd32uint16 operator+(simd32uint16 other) const
-
inline simd32uint16 operator-(simd32uint16 other) const
-
inline simd32uint16 operator&(simd512bit other) const
-
inline simd32uint16 operator|(simd512bit other) const
-
inline simd32uint16 operator^(simd512bit other) const
-
inline simd32uint16 operator~() const
-
inline simd16uint16 low() const
-
inline simd16uint16 high() const
-
inline uint16_t operator[](int i) const
-
inline void accu_min(simd32uint16 incoming)
-
inline void accu_max(simd32uint16 incoming)
-
inline void clear()
-
inline void storeu(void *ptr) const
-
inline void loadu(const void *ptr)
-
inline void store(void *ptr) const
-
inline void bin(char bits[513]) const
-
inline simd32uint16()
-
struct simd64uint8 : public faiss::simd512bit
公共函数
-
inline simd64uint8()
-
inline explicit simd64uint8(__m512i i)
-
inline explicit simd64uint8(int x)
-
inline explicit simd64uint8(uint8_t x)
-
inline explicit simd64uint8(simd256bit lo)
-
inline explicit simd64uint8(simd256bit lo, simd256bit hi)
-
inline explicit simd64uint8(simd512bit x)
-
inline explicit simd64uint8(const uint8_t *x)
-
inline void set1(uint8_t x)
-
inline simd64uint8 operator&(simd512bit other) const
-
inline simd64uint8 operator+(simd64uint8 other) const
-
inline simd64uint8 lookup_4_lanes(simd64uint8 idx) const
-
inline simd32uint16 lane0_as_uint16() const
-
inline simd32uint16 lane1_as_uint16() const
-
inline simd64uint8 operator+=(simd64uint8 other)
-
inline uint8_t operator[](int i) const
-
inline void clear()
-
inline void storeu(void *ptr) const
-
inline void loadu(const void *ptr)
-
inline void store(void *ptr) const
-
inline void bin(char bits[513]) const
-
inline simd64uint8()
-
inline simd16uint16 combine4x2(simd32uint16 a, simd32uint16 b)