usearch 2.9.2 → 2.10.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +39 -38
- package/javascript/{types → dist/cjs}/usearch.d.ts +90 -115
- package/javascript/dist/cjs/usearch.js +531 -0
- package/javascript/dist/esm/usearch.d.ts +212 -0
- package/javascript/{usearch.js → dist/esm/usearch.js} +199 -153
- package/package.json +25 -23
- package/prebuilds/darwin-arm64+x64/usearch.node +0 -0
- package/prebuilds/linux-arm64/usearch.node +0 -0
- package/prebuilds/linux-x64/usearch.node +0 -0
- package/prebuilds/win32-ia32/usearch.node +0 -0
- package/prebuilds/win32-x64/usearch.node +0 -0
- package/.clang-format +0 -10
- package/.clang-tidy +0 -60
- package/.cmake-format.py +0 -19
- package/.dockerignore +0 -93
- package/.gitattributes +0 -9
- package/.github/ISSUE_TEMPLATE/bug_report.yml +0 -90
- package/.github/ISSUE_TEMPLATE/config.yml +0 -5
- package/.github/ISSUE_TEMPLATE/feature_request.yml +0 -64
- package/.github/workflows/Dockerfile.libs +0 -36
- package/.github/workflows/prerelease.yml +0 -331
- package/.github/workflows/release.yml +0 -722
- package/.github/workflows/update_version.sh +0 -15
- package/.gitmodules +0 -9
- package/.releaserc +0 -90
- package/CITATION.cff +0 -11
- package/CMakeLists.txt +0 -324
- package/CODE_OF_CONDUCT.md +0 -128
- package/CONTRIBUTING.md +0 -361
- package/Cargo.toml +0 -19
- package/Dockerfile +0 -16
- package/VERSION +0 -1
- package/assets/unum.png +0 -0
- package/assets/usearch-approaches-transparent.png +0 -0
- package/assets/usearch-approaches-white.png +0 -0
- package/assets/usearch-neighbor-types.png +0 -0
- package/binding.gyp +0 -87
- package/build.gradle +0 -204
- package/build.rs +0 -18
- package/c/CMakeLists.txt +0 -25
- package/c/README.md +0 -34
- package/c/lib.cpp +0 -358
- package/c/test.c +0 -382
- package/c/usearch.h +0 -311
- package/cmake/clipp.cmake +0 -7
- package/cmake/config.cmake.in +0 -15
- package/cmake/emscripten.cmake +0 -8
- package/cmake/jemalloc.cmake +0 -62
- package/cmake/pkg-config.pc.in +0 -4
- package/cmake/usearchConfigVersion.cmake.in +0 -20
- package/conanfile.py +0 -26
- package/cpp/CMakeLists.txt +0 -14
- package/cpp/README.md +0 -161
- package/cpp/bench.cpp +0 -618
- package/cpp/test.cpp +0 -323
- package/fp16/CMakeLists.txt +0 -176
- package/fp16/LICENSE +0 -11
- package/fp16/README.md +0 -20
- package/fp16/bench/alt-element.cc +0 -89
- package/fp16/bench/from-alt-array.cc +0 -175
- package/fp16/bench/from-ieee-array.cc +0 -399
- package/fp16/bench/ieee-element.cc +0 -272
- package/fp16/bench/to-alt-array.cc +0 -91
- package/fp16/bench/to-ieee-array.cc +0 -303
- package/fp16/cmake/DownloadGoogleBenchmark.cmake +0 -15
- package/fp16/cmake/DownloadGoogleTest.cmake +0 -15
- package/fp16/cmake/DownloadPSimd.cmake +0 -15
- package/fp16/configure.py +0 -65
- package/fp16/confu.yaml +0 -8
- package/fp16/include/fp16/__init__.py +0 -0
- package/fp16/include/fp16/avx.py +0 -50
- package/fp16/include/fp16/avx2.py +0 -53
- package/fp16/include/fp16/bitcasts.h +0 -92
- package/fp16/include/fp16/fp16.h +0 -451
- package/fp16/include/fp16/psimd.h +0 -131
- package/fp16/include/fp16.h +0 -11
- package/fp16/jni/Android.mk +0 -6
- package/fp16/jni/Application.mk +0 -4
- package/fp16/test/alt-from-fp32-value.cc +0 -498
- package/fp16/test/alt-to-fp32-bits.cc +0 -263
- package/fp16/test/alt-to-fp32-psimd.cc +0 -145
- package/fp16/test/alt-to-fp32-value.cc +0 -348
- package/fp16/test/alt-to-fp32x2-psimd.cc +0 -245
- package/fp16/test/bitcasts.cc +0 -56
- package/fp16/test/ieee-from-fp32-value.cc +0 -523
- package/fp16/test/ieee-to-fp32-bits.cc +0 -333
- package/fp16/test/ieee-to-fp32-psimd.cc +0 -312
- package/fp16/test/ieee-to-fp32-value.cc +0 -420
- package/fp16/test/ieee-to-fp32x2-psimd.cc +0 -541
- package/fp16/test/peachpy/alt-xmm-to-fp32-xmm-avx.cc +0 -96
- package/fp16/test/peachpy/alt-xmm-to-fp32-ymm-avx2.cc +0 -96
- package/fp16/test/peachpy/stubs.py +0 -38
- package/fp16/test/tables.cc +0 -530
- package/fp16/test/tables.h +0 -13
- package/fp16/third-party/THHalf.h +0 -92
- package/fp16/third-party/eigen-half.h +0 -170
- package/fp16/third-party/float16-compressor.h +0 -80
- package/fp16/third-party/half.hpp +0 -2916
- package/fp16/third-party/npy-halffloat.h +0 -328
- package/include/usearch/index.hpp +0 -3836
- package/include/usearch/index_dense.hpp +0 -2004
- package/include/usearch/index_plugins.hpp +0 -2213
- package/javascript/lib.cpp +0 -341
- package/javascript/types/usearch.d.ts.map +0 -1
- package/javascript/usearch.test.js +0 -74
- package/package-ci.json +0 -10
- package/pyproject.toml +0 -51
- package/setup.py +0 -201
- package/simsimd/.releaserc +0 -86
- package/simsimd/CONTRIBUTING.md +0 -88
- package/simsimd/Cargo.lock +0 -709
- package/simsimd/Cargo.toml +0 -44
- package/simsimd/LICENSE +0 -201
- package/simsimd/README.md +0 -317
- package/simsimd/binding.gyp +0 -36
- package/simsimd/build.rs +0 -20
- package/simsimd/include/simsimd/binary.h +0 -245
- package/simsimd/include/simsimd/complex.h +0 -174
- package/simsimd/include/simsimd/probability.h +0 -548
- package/simsimd/include/simsimd/simsimd.h +0 -465
- package/simsimd/include/simsimd/spatial.h +0 -1267
- package/simsimd/include/simsimd/types.h +0 -208
- package/simsimd/javascript/bench.js +0 -127
- package/simsimd/javascript/fallback.ts +0 -193
- package/simsimd/javascript/lib.c +0 -102
- package/simsimd/javascript/node-gyp-build.d.ts +0 -1
- package/simsimd/javascript/simsimd.ts +0 -124
- package/simsimd/javascript/test.mjs +0 -153
- package/simsimd/javascript/tsconfig-base.json +0 -25
- package/simsimd/javascript/tsconfig-cjs.json +0 -8
- package/simsimd/javascript/tsconfig-esm.json +0 -8
- package/simsimd/package-lock.json +0 -1688
- package/simsimd/package.json +0 -61
- package/simsimd/rust/benches/cosine.rs +0 -27
- package/simsimd/rust/benches/native.rs +0 -30
- package/simsimd/rust/benches/sqeuclidean.rs +0 -27
- package/simsimd/rust/lib.c +0 -45
- package/simsimd/rust/lib.rs +0 -550
- package/sqlite/README.md +0 -178
- package/sqlite/lib.cpp +0 -325
- package/stringzilla/CONTRIBUTING.md +0 -446
- package/stringzilla/Cargo.lock +0 -25
- package/stringzilla/Cargo.toml +0 -26
- package/stringzilla/LICENSE +0 -201
- package/stringzilla/README.md +0 -1422
- package/stringzilla/VERSION +0 -1
- package/stringzilla/assets/cover-strinzilla.jpeg +0 -0
- package/stringzilla/assets/meme-stringzilla-v2.jpeg +0 -0
- package/stringzilla/assets/meme-stringzilla-v3.jpeg +0 -0
- package/stringzilla/binding.gyp +0 -16
- package/stringzilla/build.rs +0 -87
- package/stringzilla/c/lib.c +0 -337
- package/stringzilla/cli/README.md +0 -65
- package/stringzilla/cli/__init__.py +0 -0
- package/stringzilla/cli/split.py +0 -105
- package/stringzilla/cli/wc.py +0 -117
- package/stringzilla/include/stringzilla/experimental.h +0 -845
- package/stringzilla/include/stringzilla/module.modulemap +0 -9
- package/stringzilla/include/stringzilla/stringzilla.h +0 -5370
- package/stringzilla/include/stringzilla/stringzilla.hpp +0 -3838
- package/stringzilla/javascript/lib.c +0 -125
- package/stringzilla/javascript/stringzilla.js +0 -22
- package/stringzilla/package-ci.json +0 -10
- package/stringzilla/package.json +0 -32
- package/stringzilla/rust/lib.rs +0 -873
- package/stringzilla/swift/StringProtocol+StringZilla.swift +0 -280
- package/stringzilla/swift/Test.swift +0 -60
- package/tsconfig.json +0 -21
- package/wasmer.toml +0 -12
- package/winlibinstaller.bat +0 -42
- /package/{simsimd/javascript/dist-package-cjs.json → javascript/dist/cjs/package.json} +0 -0
- /package/{simsimd/javascript/dist-package-esm.json → javascript/dist/esm/package.json} +0 -0
|
@@ -1,1267 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @brief SIMD-accelerated Spatial Similarity Measures.
|
|
3
|
-
* @author Ash Vardanian
|
|
4
|
-
* @date March 14, 2023
|
|
5
|
-
*
|
|
6
|
-
* Contains:
|
|
7
|
-
* - L2 (Euclidean) squared distance
|
|
8
|
-
* - Inner product distance
|
|
9
|
-
* - Cosine similarity
|
|
10
|
-
*
|
|
11
|
-
* For datatypes:
|
|
12
|
-
* - 32-bit floating point numbers
|
|
13
|
-
* - 16-bit floating point numbers
|
|
14
|
-
* - 8-bit signed integral numbers
|
|
15
|
-
*
|
|
16
|
-
* For hardware architectures:
|
|
17
|
-
* - Arm (NEON, SVE)
|
|
18
|
-
* - x86 (AVX2, AVX512)
|
|
19
|
-
*
|
|
20
|
-
* x86 intrinsics: https://www.intel.com/content/www/us/en/docs/intrinsics-guide/
|
|
21
|
-
* Arm intrinsics: https://developer.arm.com/architectures/instruction-sets/intrinsics/
|
|
22
|
-
*/
|
|
23
|
-
#ifndef SIMSIMD_SPATIAL_H
|
|
24
|
-
#define SIMSIMD_SPATIAL_H
|
|
25
|
-
|
|
26
|
-
#include "types.h"
|
|
27
|
-
|
|
28
|
-
#define SIMSIMD_MAKE_L2SQ(name, input_type, accumulator_type, converter) \
|
|
29
|
-
inline static simsimd_f32_t simsimd_##name##_##input_type##_l2sq( \
|
|
30
|
-
simsimd_##input_type##_t const* a, simsimd_##input_type##_t const* b, simsimd_size_t n) { \
|
|
31
|
-
simsimd_##accumulator_type##_t d2 = 0; \
|
|
32
|
-
for (simsimd_size_t i = 0; i != n; ++i) { \
|
|
33
|
-
simsimd_##accumulator_type##_t ai = converter(a[i]); \
|
|
34
|
-
simsimd_##accumulator_type##_t bi = converter(b[i]); \
|
|
35
|
-
d2 += (ai - bi) * (ai - bi); \
|
|
36
|
-
} \
|
|
37
|
-
return d2; \
|
|
38
|
-
}
|
|
39
|
-
|
|
40
|
-
#define SIMSIMD_MAKE_IP(name, input_type, accumulator_type, converter) \
|
|
41
|
-
inline static simsimd_f32_t simsimd_##name##_##input_type##_ip( \
|
|
42
|
-
simsimd_##input_type##_t const* a, simsimd_##input_type##_t const* b, simsimd_size_t n) { \
|
|
43
|
-
simsimd_##accumulator_type##_t ab = 0; \
|
|
44
|
-
for (simsimd_size_t i = 0; i != n; ++i) { \
|
|
45
|
-
simsimd_##accumulator_type##_t ai = converter(a[i]); \
|
|
46
|
-
simsimd_##accumulator_type##_t bi = converter(b[i]); \
|
|
47
|
-
ab += ai * bi; \
|
|
48
|
-
} \
|
|
49
|
-
return 1 - ab; \
|
|
50
|
-
}
|
|
51
|
-
|
|
52
|
-
#define SIMSIMD_MAKE_COS(name, input_type, accumulator_type, converter) \
|
|
53
|
-
inline static simsimd_f32_t simsimd_##name##_##input_type##_cos( \
|
|
54
|
-
simsimd_##input_type##_t const* a, simsimd_##input_type##_t const* b, simsimd_size_t n) { \
|
|
55
|
-
simsimd_##accumulator_type##_t ab = 0, a2 = 0, b2 = 0; \
|
|
56
|
-
for (simsimd_size_t i = 0; i != n; ++i) { \
|
|
57
|
-
simsimd_##accumulator_type##_t ai = converter(a[i]); \
|
|
58
|
-
simsimd_##accumulator_type##_t bi = converter(b[i]); \
|
|
59
|
-
ab += ai * bi; \
|
|
60
|
-
a2 += ai * ai; \
|
|
61
|
-
b2 += bi * bi; \
|
|
62
|
-
} \
|
|
63
|
-
return ab != 0 ? (1 - ab * SIMSIMD_RSQRT(a2) * SIMSIMD_RSQRT(b2)) : 1; \
|
|
64
|
-
}
|
|
65
|
-
|
|
66
|
-
#ifdef __cplusplus
|
|
67
|
-
extern "C" {
|
|
68
|
-
#endif
|
|
69
|
-
|
|
70
|
-
SIMSIMD_MAKE_L2SQ(serial, f64, f64, SIMSIMD_IDENTIFY) // simsimd_serial_f64_l2sq
|
|
71
|
-
SIMSIMD_MAKE_IP(serial, f64, f64, SIMSIMD_IDENTIFY) // simsimd_serial_f64_ip
|
|
72
|
-
SIMSIMD_MAKE_COS(serial, f64, f64, SIMSIMD_IDENTIFY) // simsimd_serial_f64_cos
|
|
73
|
-
|
|
74
|
-
SIMSIMD_MAKE_L2SQ(serial, f32, f32, SIMSIMD_IDENTIFY) // simsimd_serial_f32_l2sq
|
|
75
|
-
SIMSIMD_MAKE_IP(serial, f32, f32, SIMSIMD_IDENTIFY) // simsimd_serial_f32_ip
|
|
76
|
-
SIMSIMD_MAKE_COS(serial, f32, f32, SIMSIMD_IDENTIFY) // simsimd_serial_f32_cos
|
|
77
|
-
|
|
78
|
-
SIMSIMD_MAKE_L2SQ(serial, f16, f32, SIMSIMD_UNCOMPRESS_F16) // simsimd_serial_f16_l2sq
|
|
79
|
-
SIMSIMD_MAKE_IP(serial, f16, f32, SIMSIMD_UNCOMPRESS_F16) // simsimd_serial_f16_ip
|
|
80
|
-
SIMSIMD_MAKE_COS(serial, f16, f32, SIMSIMD_UNCOMPRESS_F16) // simsimd_serial_f16_cos
|
|
81
|
-
|
|
82
|
-
SIMSIMD_MAKE_L2SQ(serial, i8, i32, SIMSIMD_IDENTIFY) // simsimd_serial_i8_l2sq
|
|
83
|
-
SIMSIMD_MAKE_COS(serial, i8, i32, SIMSIMD_IDENTIFY) // simsimd_serial_i8_cos
|
|
84
|
-
|
|
85
|
-
inline static simsimd_f32_t simsimd_serial_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
86
|
-
return simsimd_serial_i8_cos(a, b, n);
|
|
87
|
-
}
|
|
88
|
-
|
|
89
|
-
SIMSIMD_MAKE_L2SQ(accurate, f32, f64, SIMSIMD_IDENTIFY) // simsimd_accurate_f32_l2sq
|
|
90
|
-
SIMSIMD_MAKE_IP(accurate, f32, f64, SIMSIMD_IDENTIFY) // simsimd_accurate_f32_ip
|
|
91
|
-
SIMSIMD_MAKE_COS(accurate, f32, f64, SIMSIMD_IDENTIFY) // simsimd_accurate_f32_cos
|
|
92
|
-
|
|
93
|
-
SIMSIMD_MAKE_L2SQ(accurate, f16, f64, SIMSIMD_UNCOMPRESS_F16) // simsimd_accurate_f16_l2sq
|
|
94
|
-
SIMSIMD_MAKE_IP(accurate, f16, f64, SIMSIMD_UNCOMPRESS_F16) // simsimd_accurate_f16_ip
|
|
95
|
-
SIMSIMD_MAKE_COS(accurate, f16, f64, SIMSIMD_UNCOMPRESS_F16) // simsimd_accurate_f16_cos
|
|
96
|
-
|
|
97
|
-
SIMSIMD_MAKE_L2SQ(accurate, i8, i32, SIMSIMD_IDENTIFY) // simsimd_accurate_i8_l2sq
|
|
98
|
-
SIMSIMD_MAKE_COS(accurate, i8, i32, SIMSIMD_IDENTIFY) // simsimd_accurate_i8_cos
|
|
99
|
-
|
|
100
|
-
inline static simsimd_f32_t simsimd_accurate_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
101
|
-
return simsimd_accurate_i8_cos(a, b, n);
|
|
102
|
-
}
|
|
103
|
-
|
|
104
|
-
#if SIMSIMD_TARGET_ARM
|
|
105
|
-
#if SIMSIMD_TARGET_ARM_NEON
|
|
106
|
-
|
|
107
|
-
/*
|
|
108
|
-
* @file arm_neon_f32.h
|
|
109
|
-
* @brief Arm NEON implementation of the most common similarity metrics for 32-bit floating point numbers.
|
|
110
|
-
* @author Ash Vardanian
|
|
111
|
-
*
|
|
112
|
-
* - Implements: L2 squared, inner product, cosine similarity.
|
|
113
|
-
* - Uses `f32` for storage and `f32` for accumulation.
|
|
114
|
-
* - Requires compiler capabilities: +simd.
|
|
115
|
-
*/
|
|
116
|
-
|
|
117
|
-
__attribute__((target("+simd"))) //
|
|
118
|
-
inline static simsimd_f32_t
|
|
119
|
-
simsimd_neon_f32_l2sq(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
|
|
120
|
-
float32x4_t sum_vec = vdupq_n_f32(0);
|
|
121
|
-
simsimd_size_t i = 0;
|
|
122
|
-
for (; i + 4 <= n; i += 4) {
|
|
123
|
-
float32x4_t a_vec = vld1q_f32(a + i);
|
|
124
|
-
float32x4_t b_vec = vld1q_f32(b + i);
|
|
125
|
-
float32x4_t diff_vec = vsubq_f32(a_vec, b_vec);
|
|
126
|
-
sum_vec = vfmaq_f32(sum_vec, diff_vec, diff_vec);
|
|
127
|
-
}
|
|
128
|
-
simsimd_f32_t sum = vaddvq_f32(sum_vec);
|
|
129
|
-
for (; i < n; ++i) {
|
|
130
|
-
simsimd_f32_t diff = a[i] - b[i];
|
|
131
|
-
sum += diff * diff;
|
|
132
|
-
}
|
|
133
|
-
return sum;
|
|
134
|
-
}
|
|
135
|
-
|
|
136
|
-
__attribute__((target("+simd"))) //
|
|
137
|
-
inline static simsimd_f32_t
|
|
138
|
-
simsimd_neon_f32_ip(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
|
|
139
|
-
float32x4_t ab_vec = vdupq_n_f32(0);
|
|
140
|
-
simsimd_size_t i = 0;
|
|
141
|
-
for (; i + 4 <= n; i += 4) {
|
|
142
|
-
float32x4_t a_vec = vld1q_f32(a + i);
|
|
143
|
-
float32x4_t b_vec = vld1q_f32(b + i);
|
|
144
|
-
ab_vec = vfmaq_f32(ab_vec, a_vec, b_vec);
|
|
145
|
-
}
|
|
146
|
-
simsimd_f32_t ab = vaddvq_f32(ab_vec);
|
|
147
|
-
for (; i < n; ++i)
|
|
148
|
-
ab += a[i] * b[i];
|
|
149
|
-
return 1 - ab;
|
|
150
|
-
}
|
|
151
|
-
|
|
152
|
-
__attribute__((target("+simd"))) //
|
|
153
|
-
inline static simsimd_f32_t
|
|
154
|
-
simsimd_neon_f32_cos(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
|
|
155
|
-
float32x4_t ab_vec = vdupq_n_f32(0), a2_vec = vdupq_n_f32(0), b2_vec = vdupq_n_f32(0);
|
|
156
|
-
simsimd_size_t i = 0;
|
|
157
|
-
for (; i + 4 <= n; i += 4) {
|
|
158
|
-
float32x4_t a_vec = vld1q_f32(a + i);
|
|
159
|
-
float32x4_t b_vec = vld1q_f32(b + i);
|
|
160
|
-
ab_vec = vfmaq_f32(ab_vec, a_vec, b_vec);
|
|
161
|
-
a2_vec = vfmaq_f32(a2_vec, a_vec, a_vec);
|
|
162
|
-
b2_vec = vfmaq_f32(b2_vec, b_vec, b_vec);
|
|
163
|
-
}
|
|
164
|
-
simsimd_f32_t ab = vaddvq_f32(ab_vec), a2 = vaddvq_f32(a2_vec), b2 = vaddvq_f32(b2_vec);
|
|
165
|
-
for (; i < n; ++i) {
|
|
166
|
-
simsimd_f32_t ai = a[i], bi = b[i];
|
|
167
|
-
ab += ai * bi, a2 += ai * ai, b2 += bi * bi;
|
|
168
|
-
}
|
|
169
|
-
|
|
170
|
-
// Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
|
|
171
|
-
simsimd_f32_t a2_b2_arr[2] = {a2, b2};
|
|
172
|
-
vst1_f32(a2_b2_arr, vrsqrte_f32(vld1_f32(a2_b2_arr)));
|
|
173
|
-
return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
|
|
174
|
-
}
|
|
175
|
-
|
|
176
|
-
/*
|
|
177
|
-
* @file arm_neon_f16.h
|
|
178
|
-
* @brief Arm NEON implementation of the most common similarity metrics for 16-bit floating point numbers.
|
|
179
|
-
* @author Ash Vardanian
|
|
180
|
-
*
|
|
181
|
-
* - Implements: L2 squared, inner product, cosine similarity.
|
|
182
|
-
* - Uses `f16` for storage and `f32` for accumulation, as the 16-bit FMA may not always be available.
|
|
183
|
-
* - Requires compiler capabilities: +simd+fp16.
|
|
184
|
-
*/
|
|
185
|
-
|
|
186
|
-
__attribute__((target("+simd+fp16"))) //
|
|
187
|
-
inline static simsimd_f32_t
|
|
188
|
-
simsimd_neon_f16_l2sq(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
|
|
189
|
-
float32x4_t sum_vec = vdupq_n_f32(0);
|
|
190
|
-
simsimd_size_t i = 0;
|
|
191
|
-
for (; i + 4 <= n; i += 4) {
|
|
192
|
-
float32x4_t a_vec = vcvt_f32_f16(vld1_f16((float16_t const*)a + i));
|
|
193
|
-
float32x4_t b_vec = vcvt_f32_f16(vld1_f16((float16_t const*)b + i));
|
|
194
|
-
float32x4_t diff_vec = vsubq_f32(a_vec, b_vec);
|
|
195
|
-
sum_vec = vfmaq_f32(sum_vec, diff_vec, diff_vec);
|
|
196
|
-
}
|
|
197
|
-
|
|
198
|
-
// In case the software emulation for `f16` scalars is enabled, the `simsimd_uncompress_f16`
|
|
199
|
-
// function will run. It is extremely slow, so even for the tail, let's combine serial
|
|
200
|
-
// loads and stores with vectorized math.
|
|
201
|
-
if (i < n) {
|
|
202
|
-
union {
|
|
203
|
-
float16x4_t f16_vec;
|
|
204
|
-
simsimd_f16_t f16[4];
|
|
205
|
-
} a_padded_tail, b_padded_tail;
|
|
206
|
-
simsimd_size_t j = 0;
|
|
207
|
-
for (; i < n; ++i, ++j)
|
|
208
|
-
a_padded_tail.f16[j] = a[i], b_padded_tail.f16[j] = b[i];
|
|
209
|
-
for (; j < 4; ++j)
|
|
210
|
-
a_padded_tail.f16[j] = 0, b_padded_tail.f16[j] = 0;
|
|
211
|
-
float32x4_t diff_vec = vsubq_f32(vcvt_f32_f16(a_padded_tail.f16_vec), vcvt_f32_f16(b_padded_tail.f16_vec));
|
|
212
|
-
sum_vec = vfmaq_f32(sum_vec, diff_vec, diff_vec);
|
|
213
|
-
}
|
|
214
|
-
|
|
215
|
-
simsimd_f32_t sum = vaddvq_f32(sum_vec);
|
|
216
|
-
return sum;
|
|
217
|
-
}
|
|
218
|
-
|
|
219
|
-
__attribute__((target("+simd+fp16"))) //
|
|
220
|
-
inline static simsimd_f32_t
|
|
221
|
-
simsimd_neon_f16_ip(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
|
|
222
|
-
float32x4_t ab_vec = vdupq_n_f32(0);
|
|
223
|
-
simsimd_size_t i = 0;
|
|
224
|
-
for (; i + 4 <= n; i += 4) {
|
|
225
|
-
float32x4_t a_vec = vcvt_f32_f16(vld1_f16((float16_t const*)a + i));
|
|
226
|
-
float32x4_t b_vec = vcvt_f32_f16(vld1_f16((float16_t const*)b + i));
|
|
227
|
-
ab_vec = vfmaq_f32(ab_vec, a_vec, b_vec);
|
|
228
|
-
}
|
|
229
|
-
|
|
230
|
-
// In case the software emulation for `f16` scalars is enabled, the `simsimd_uncompress_f16`
|
|
231
|
-
// function will run. It is extremely slow, so even for the tail, let's combine serial
|
|
232
|
-
// loads and stores with vectorized math.
|
|
233
|
-
if (i < n) {
|
|
234
|
-
union {
|
|
235
|
-
float16x4_t f16_vec;
|
|
236
|
-
simsimd_f16_t f16[4];
|
|
237
|
-
} a_padded_tail, b_padded_tail;
|
|
238
|
-
simsimd_size_t j = 0;
|
|
239
|
-
for (; i < n; ++i, ++j)
|
|
240
|
-
a_padded_tail.f16[j] = a[i], b_padded_tail.f16[j] = b[i];
|
|
241
|
-
for (; j < 4; ++j)
|
|
242
|
-
a_padded_tail.f16[j] = 0, b_padded_tail.f16[j] = 0;
|
|
243
|
-
ab_vec = vfmaq_f32(ab_vec, vcvt_f32_f16(a_padded_tail.f16_vec), vcvt_f32_f16(b_padded_tail.f16_vec));
|
|
244
|
-
}
|
|
245
|
-
simsimd_f32_t ab = vaddvq_f32(ab_vec);
|
|
246
|
-
return 1 - ab;
|
|
247
|
-
}
|
|
248
|
-
|
|
249
|
-
__attribute__((target("+simd+fp16"))) //
|
|
250
|
-
inline static simsimd_f32_t
|
|
251
|
-
simsimd_neon_f16_cos(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
|
|
252
|
-
float32x4_t ab_vec = vdupq_n_f32(0), a2_vec = vdupq_n_f32(0), b2_vec = vdupq_n_f32(0);
|
|
253
|
-
simsimd_size_t i = 0;
|
|
254
|
-
for (; i + 4 <= n; i += 4) {
|
|
255
|
-
float32x4_t a_vec = vcvt_f32_f16(vld1_f16((float16_t const*)a + i));
|
|
256
|
-
float32x4_t b_vec = vcvt_f32_f16(vld1_f16((float16_t const*)b + i));
|
|
257
|
-
ab_vec = vfmaq_f32(ab_vec, a_vec, b_vec);
|
|
258
|
-
a2_vec = vfmaq_f32(a2_vec, a_vec, a_vec);
|
|
259
|
-
b2_vec = vfmaq_f32(b2_vec, b_vec, b_vec);
|
|
260
|
-
}
|
|
261
|
-
|
|
262
|
-
// In case the software emulation for `f16` scalars is enabled, the `simsimd_uncompress_f16`
|
|
263
|
-
// function will run. It is extremely slow, so even for the tail, let's combine serial
|
|
264
|
-
// loads and stores with vectorized math.
|
|
265
|
-
if (i < n) {
|
|
266
|
-
union {
|
|
267
|
-
float16x4_t f16_vec;
|
|
268
|
-
simsimd_f16_t f16[4];
|
|
269
|
-
} a_padded_tail, b_padded_tail;
|
|
270
|
-
simsimd_size_t j = 0;
|
|
271
|
-
for (; i < n; ++i, ++j)
|
|
272
|
-
a_padded_tail.f16[j] = a[i], b_padded_tail.f16[j] = b[i];
|
|
273
|
-
for (; j < 4; ++j)
|
|
274
|
-
a_padded_tail.f16[j] = 0, b_padded_tail.f16[j] = 0;
|
|
275
|
-
float32x4_t a_vec = vcvt_f32_f16(a_padded_tail.f16_vec);
|
|
276
|
-
float32x4_t b_vec = vcvt_f32_f16(b_padded_tail.f16_vec);
|
|
277
|
-
ab_vec = vfmaq_f32(ab_vec, a_vec, b_vec);
|
|
278
|
-
a2_vec = vfmaq_f32(a2_vec, a_vec, a_vec);
|
|
279
|
-
b2_vec = vfmaq_f32(b2_vec, b_vec, b_vec);
|
|
280
|
-
}
|
|
281
|
-
|
|
282
|
-
// Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
|
|
283
|
-
simsimd_f32_t ab = vaddvq_f32(ab_vec), a2 = vaddvq_f32(a2_vec), b2 = vaddvq_f32(b2_vec);
|
|
284
|
-
simsimd_f32_t a2_b2_arr[2] = {a2, b2};
|
|
285
|
-
float32x2_t a2_b2 = vld1_f32(a2_b2_arr);
|
|
286
|
-
a2_b2 = vrsqrte_f32(a2_b2);
|
|
287
|
-
vst1_f32(a2_b2_arr, a2_b2);
|
|
288
|
-
return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
|
|
289
|
-
}
|
|
290
|
-
|
|
291
|
-
/*
|
|
292
|
-
* @file arm_neon_i8.h
|
|
293
|
-
* @brief Arm NEON implementation of the most common similarity metrics for 8-bit signed integral numbers.
|
|
294
|
-
* @author Ash Vardanian
|
|
295
|
-
*
|
|
296
|
-
* - Implements: L2 squared, cosine similarity, inner product (same as cosine).
|
|
297
|
-
* - Uses `i8` for storage, `i16` for multiplication, and `i32` for accumulation, if no better option is available.
|
|
298
|
-
* - Requires compiler capabilities: +simd+dotprod.
|
|
299
|
-
*/
|
|
300
|
-
|
|
301
|
-
__attribute__((target("+simd"))) //
|
|
302
|
-
inline static simsimd_f32_t
|
|
303
|
-
simsimd_neon_i8_l2sq(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
304
|
-
int32x4_t d2_vec = vdupq_n_s32(0);
|
|
305
|
-
simsimd_size_t i = 0;
|
|
306
|
-
for (; i + 7 < n; i += 8) {
|
|
307
|
-
int8x8_t a_vec = vld1_s8(a + i);
|
|
308
|
-
int8x8_t b_vec = vld1_s8(b + i);
|
|
309
|
-
int16x8_t a_vec16 = vmovl_s8(a_vec);
|
|
310
|
-
int16x8_t b_vec16 = vmovl_s8(b_vec);
|
|
311
|
-
int16x8_t d_vec = vsubq_s16(a_vec16, b_vec16);
|
|
312
|
-
int32x4_t d_low = vmull_s16(vget_low_s16(d_vec), vget_low_s16(d_vec));
|
|
313
|
-
int32x4_t d_high = vmull_s16(vget_high_s16(d_vec), vget_high_s16(d_vec));
|
|
314
|
-
d2_vec = vaddq_s32(d2_vec, vaddq_s32(d_low, d_high));
|
|
315
|
-
}
|
|
316
|
-
int32_t d2 = vaddvq_s32(d2_vec);
|
|
317
|
-
for (; i < n; ++i) {
|
|
318
|
-
int32_t n = a[i] - b[i];
|
|
319
|
-
d2 += n * n;
|
|
320
|
-
}
|
|
321
|
-
return d2;
|
|
322
|
-
}
|
|
323
|
-
|
|
324
|
-
__attribute__((target("arch=armv8.2-a+dotprod"))) //
|
|
325
|
-
inline static simsimd_f32_t
|
|
326
|
-
simsimd_neon_i8_cos(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
327
|
-
|
|
328
|
-
int32x4_t ab_vec = vdupq_n_s32(0);
|
|
329
|
-
int32x4_t a2_vec = vdupq_n_s32(0);
|
|
330
|
-
int32x4_t b2_vec = vdupq_n_s32(0);
|
|
331
|
-
simsimd_size_t i = 0;
|
|
332
|
-
|
|
333
|
-
// If the 128-bit `vdot_s32` intrinsic is unavailable, we can use the 64-bit `vdot_s32`.
|
|
334
|
-
// for (simsimd_size_t i = 0; i != n; i += 8) {
|
|
335
|
-
// int16x8_t a_vec = vmovl_s8(vld1_s8(a + i));
|
|
336
|
-
// int16x8_t b_vec = vmovl_s8(vld1_s8(b + i));
|
|
337
|
-
// int16x8_t ab_part_vec = vmulq_s16(a_vec, b_vec);
|
|
338
|
-
// int16x8_t a2_part_vec = vmulq_s16(a_vec, a_vec);
|
|
339
|
-
// int16x8_t b2_part_vec = vmulq_s16(b_vec, b_vec);
|
|
340
|
-
// ab_vec = vaddq_s32(ab_vec, vaddq_s32(vmovl_s16(vget_high_s16(ab_part_vec)), //
|
|
341
|
-
// vmovl_s16(vget_low_s16(ab_part_vec))));
|
|
342
|
-
// a2_vec = vaddq_s32(a2_vec, vaddq_s32(vmovl_s16(vget_high_s16(a2_part_vec)), //
|
|
343
|
-
// vmovl_s16(vget_low_s16(a2_part_vec))));
|
|
344
|
-
// b2_vec = vaddq_s32(b2_vec, vaddq_s32(vmovl_s16(vget_high_s16(b2_part_vec)), //
|
|
345
|
-
// vmovl_s16(vget_low_s16(b2_part_vec))));
|
|
346
|
-
// }
|
|
347
|
-
for (; i + 15 < n; i += 16) {
|
|
348
|
-
int8x16_t a_vec = vld1q_s8(a + i);
|
|
349
|
-
int8x16_t b_vec = vld1q_s8(b + i);
|
|
350
|
-
ab_vec = vdotq_s32(ab_vec, a_vec, b_vec);
|
|
351
|
-
a2_vec = vdotq_s32(a2_vec, a_vec, a_vec);
|
|
352
|
-
b2_vec = vdotq_s32(b2_vec, b_vec, b_vec);
|
|
353
|
-
}
|
|
354
|
-
|
|
355
|
-
int32_t ab = vaddvq_s32(ab_vec);
|
|
356
|
-
int32_t a2 = vaddvq_s32(a2_vec);
|
|
357
|
-
int32_t b2 = vaddvq_s32(b2_vec);
|
|
358
|
-
|
|
359
|
-
// Take care of the tail:
|
|
360
|
-
for (; i < n; ++i) {
|
|
361
|
-
int32_t ai = a[i], bi = b[i];
|
|
362
|
-
ab += ai * bi, a2 += ai * ai, b2 += bi * bi;
|
|
363
|
-
}
|
|
364
|
-
|
|
365
|
-
// Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
|
|
366
|
-
simsimd_f32_t a2_b2_arr[2] = {(simsimd_f32_t)a2, (simsimd_f32_t)b2};
|
|
367
|
-
float32x2_t a2_b2 = vld1_f32(a2_b2_arr);
|
|
368
|
-
a2_b2 = vrsqrte_f32(a2_b2);
|
|
369
|
-
vst1_f32(a2_b2_arr, a2_b2);
|
|
370
|
-
return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
|
|
371
|
-
}
|
|
372
|
-
|
|
373
|
-
__attribute__((target("arch=armv8.2-a+dotprod"))) //
|
|
374
|
-
inline static simsimd_f32_t
|
|
375
|
-
simsimd_neon_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
376
|
-
return simsimd_neon_i8_cos(a, b, n);
|
|
377
|
-
}
|
|
378
|
-
|
|
379
|
-
#endif // SIMSIMD_TARGET_ARM_NEON
|
|
380
|
-
|
|
381
|
-
#if SIMSIMD_TARGET_ARM_SVE
|
|
382
|
-
|
|
383
|
-
/*
|
|
384
|
-
* @file arm_sve_f32.h
|
|
385
|
-
* @brief Arm SVE implementation of the most common similarity metrics for 32-bit floating point numbers.
|
|
386
|
-
* @author Ash Vardanian
|
|
387
|
-
*
|
|
388
|
-
* - Implements: L2 squared, inner product, cosine similarity.
|
|
389
|
-
* - Uses `f32` for both storage and accumulation.
|
|
390
|
-
* - Requires compiler capabilities: +sve.
|
|
391
|
-
*/
|
|
392
|
-
|
|
393
|
-
__attribute__((target("+sve"))) //
|
|
394
|
-
inline static simsimd_f32_t
|
|
395
|
-
simsimd_sve_f32_l2sq(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
|
|
396
|
-
simsimd_size_t i = 0;
|
|
397
|
-
svfloat32_t d2_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
|
|
398
|
-
do {
|
|
399
|
-
svbool_t pg_vec = svwhilelt_b32((unsigned int)i, (unsigned int)n);
|
|
400
|
-
svfloat32_t a_vec = svld1_f32(pg_vec, a + i);
|
|
401
|
-
svfloat32_t b_vec = svld1_f32(pg_vec, b + i);
|
|
402
|
-
svfloat32_t a_minus_b_vec = svsub_f32_x(pg_vec, a_vec, b_vec);
|
|
403
|
-
d2_vec = svmla_f32_x(pg_vec, d2_vec, a_minus_b_vec, a_minus_b_vec);
|
|
404
|
-
i += svcntw();
|
|
405
|
-
} while (i < n);
|
|
406
|
-
simsimd_f32_t d2 = svaddv_f32(svptrue_b32(), d2_vec);
|
|
407
|
-
return d2;
|
|
408
|
-
}
|
|
409
|
-
__attribute__((target("+sve"))) //
|
|
410
|
-
inline static simsimd_f32_t
|
|
411
|
-
simsimd_sve_f32_ip(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
|
|
412
|
-
simsimd_size_t i = 0;
|
|
413
|
-
svfloat32_t ab_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
|
|
414
|
-
do {
|
|
415
|
-
svbool_t pg_vec = svwhilelt_b32((unsigned int)i, (unsigned int)n);
|
|
416
|
-
svfloat32_t a_vec = svld1_f32(pg_vec, a + i);
|
|
417
|
-
svfloat32_t b_vec = svld1_f32(pg_vec, b + i);
|
|
418
|
-
ab_vec = svmla_f32_x(pg_vec, ab_vec, a_vec, b_vec);
|
|
419
|
-
i += svcntw();
|
|
420
|
-
} while (i < n);
|
|
421
|
-
return 1 - svaddv_f32(svptrue_b32(), ab_vec);
|
|
422
|
-
}
|
|
423
|
-
|
|
424
|
-
__attribute__((target("+sve"))) //
|
|
425
|
-
inline static simsimd_f32_t
|
|
426
|
-
simsimd_sve_f32_cos(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
|
|
427
|
-
simsimd_size_t i = 0;
|
|
428
|
-
svfloat32_t ab_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
|
|
429
|
-
svfloat32_t a2_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
|
|
430
|
-
svfloat32_t b2_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
|
|
431
|
-
do {
|
|
432
|
-
svbool_t pg_vec = svwhilelt_b32((unsigned int)i, (unsigned int)n);
|
|
433
|
-
svfloat32_t a_vec = svld1_f32(pg_vec, a + i);
|
|
434
|
-
svfloat32_t b_vec = svld1_f32(pg_vec, b + i);
|
|
435
|
-
ab_vec = svmla_f32_x(pg_vec, ab_vec, a_vec, b_vec);
|
|
436
|
-
a2_vec = svmla_f32_x(pg_vec, a2_vec, a_vec, a_vec);
|
|
437
|
-
b2_vec = svmla_f32_x(pg_vec, b2_vec, b_vec, b_vec);
|
|
438
|
-
i += svcntw();
|
|
439
|
-
} while (i < n);
|
|
440
|
-
|
|
441
|
-
simsimd_f32_t ab = svaddv_f32(svptrue_b32(), ab_vec);
|
|
442
|
-
simsimd_f32_t a2 = svaddv_f32(svptrue_b32(), a2_vec);
|
|
443
|
-
simsimd_f32_t b2 = svaddv_f32(svptrue_b32(), b2_vec);
|
|
444
|
-
|
|
445
|
-
// Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
|
|
446
|
-
simsimd_f32_t a2_b2_arr[2] = {a2, b2};
|
|
447
|
-
float32x2_t a2_b2 = vld1_f32(a2_b2_arr);
|
|
448
|
-
a2_b2 = vrsqrte_f32(a2_b2);
|
|
449
|
-
vst1_f32(a2_b2_arr, a2_b2);
|
|
450
|
-
return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
|
|
451
|
-
}
|
|
452
|
-
|
|
453
|
-
/*
|
|
454
|
-
* @file arm_sve_f16.h
|
|
455
|
-
* @brief Arm SVE implementation of the most common similarity metrics for 16-bit floating point numbers.
|
|
456
|
-
* @author Ash Vardanian
|
|
457
|
-
*
|
|
458
|
-
* - Implements: L2 squared, inner product, cosine similarity.
|
|
459
|
-
* - Uses `f16` for storage and `f32` for accumulation.
|
|
460
|
-
* - Requires compiler capabilities: +sve+fp16.
|
|
461
|
-
*/
|
|
462
|
-
|
|
463
|
-
__attribute__((target("+sve+fp16"))) //
|
|
464
|
-
inline static simsimd_f32_t
|
|
465
|
-
simsimd_sve_f16_l2sq(simsimd_f16_t const* a_enum, simsimd_f16_t const* b_enum, simsimd_size_t n) {
|
|
466
|
-
simsimd_size_t i = 0;
|
|
467
|
-
svfloat16_t d2_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
|
|
468
|
-
simsimd_f16_t const* a = (simsimd_f16_t const*)(a_enum);
|
|
469
|
-
simsimd_f16_t const* b = (simsimd_f16_t const*)(b_enum);
|
|
470
|
-
do {
|
|
471
|
-
svbool_t pg_vec = svwhilelt_b16((unsigned int)i, (unsigned int)n);
|
|
472
|
-
svfloat16_t a_vec = svld1_f16(pg_vec, (float16_t const*)a + i);
|
|
473
|
-
svfloat16_t b_vec = svld1_f16(pg_vec, (float16_t const*)b + i);
|
|
474
|
-
svfloat16_t a_minus_b_vec = svsub_f16_x(pg_vec, a_vec, b_vec);
|
|
475
|
-
d2_vec = svmla_f16_x(pg_vec, d2_vec, a_minus_b_vec, a_minus_b_vec);
|
|
476
|
-
i += svcnth();
|
|
477
|
-
} while (i < n);
|
|
478
|
-
float16_t d2_f16 = svaddv_f16(svptrue_b16(), d2_vec);
|
|
479
|
-
return d2_f16;
|
|
480
|
-
}
|
|
481
|
-
__attribute__((target("+sve+fp16"))) //
|
|
482
|
-
inline static simsimd_f32_t
|
|
483
|
-
simsimd_sve_f16_ip(simsimd_f16_t const* a_enum, simsimd_f16_t const* b_enum, simsimd_size_t n) {
|
|
484
|
-
simsimd_size_t i = 0;
|
|
485
|
-
svfloat16_t ab_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
|
|
486
|
-
simsimd_f16_t const* a = (simsimd_f16_t const*)(a_enum);
|
|
487
|
-
simsimd_f16_t const* b = (simsimd_f16_t const*)(b_enum);
|
|
488
|
-
do {
|
|
489
|
-
svbool_t pg_vec = svwhilelt_b16((unsigned int)i, (unsigned int)n);
|
|
490
|
-
svfloat16_t a_vec = svld1_f16(pg_vec, (float16_t const*)a + i);
|
|
491
|
-
svfloat16_t b_vec = svld1_f16(pg_vec, (float16_t const*)b + i);
|
|
492
|
-
ab_vec = svmla_f16_x(pg_vec, ab_vec, a_vec, b_vec);
|
|
493
|
-
i += svcnth();
|
|
494
|
-
} while (i < n);
|
|
495
|
-
simsimd_f16_t ab = svaddv_f16(svptrue_b16(), ab_vec);
|
|
496
|
-
return 1 - ab;
|
|
497
|
-
}
|
|
498
|
-
|
|
499
|
-
__attribute__((target("+sve+fp16"))) //
|
|
500
|
-
inline static simsimd_f32_t
|
|
501
|
-
simsimd_sve_f16_cos(simsimd_f16_t const* a_enum, simsimd_f16_t const* b_enum, simsimd_size_t n) {
|
|
502
|
-
simsimd_size_t i = 0;
|
|
503
|
-
svfloat16_t ab_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
|
|
504
|
-
svfloat16_t a2_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
|
|
505
|
-
svfloat16_t b2_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
|
|
506
|
-
simsimd_f16_t const* a = (simsimd_f16_t const*)(a_enum);
|
|
507
|
-
simsimd_f16_t const* b = (simsimd_f16_t const*)(b_enum);
|
|
508
|
-
do {
|
|
509
|
-
svbool_t pg_vec = svwhilelt_b16((unsigned int)i, (unsigned int)n);
|
|
510
|
-
svfloat16_t a_vec = svld1_f16(pg_vec, (float16_t const*)a + i);
|
|
511
|
-
svfloat16_t b_vec = svld1_f16(pg_vec, (float16_t const*)b + i);
|
|
512
|
-
ab_vec = svmla_f16_x(pg_vec, ab_vec, a_vec, b_vec);
|
|
513
|
-
a2_vec = svmla_f16_x(pg_vec, a2_vec, a_vec, a_vec);
|
|
514
|
-
b2_vec = svmla_f16_x(pg_vec, b2_vec, b_vec, b_vec);
|
|
515
|
-
i += svcnth();
|
|
516
|
-
} while (i < n);
|
|
517
|
-
|
|
518
|
-
simsimd_f16_t ab = svaddv_f16(svptrue_b16(), ab_vec);
|
|
519
|
-
simsimd_f16_t a2 = svaddv_f16(svptrue_b16(), a2_vec);
|
|
520
|
-
simsimd_f16_t b2 = svaddv_f16(svptrue_b16(), b2_vec);
|
|
521
|
-
|
|
522
|
-
// Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
|
|
523
|
-
simsimd_f32_t a2_b2_arr[2] = {a2, b2};
|
|
524
|
-
float32x2_t a2_b2 = vld1_f32(a2_b2_arr);
|
|
525
|
-
a2_b2 = vrsqrte_f32(a2_b2);
|
|
526
|
-
vst1_f32(a2_b2_arr, a2_b2);
|
|
527
|
-
return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
|
|
528
|
-
}
|
|
529
|
-
|
|
530
|
-
/*
|
|
531
|
-
* @file arm_sve_f64.h
|
|
532
|
-
* @brief Arm SVE implementation of the most common similarity metrics for 64-bit floating point numbers.
|
|
533
|
-
* @author Ash Vardanian
|
|
534
|
-
*
|
|
535
|
-
* - Implements: L2 squared, inner product, cosine similarity.
|
|
536
|
-
* - Uses `f64` for both storage and for accumulation.
|
|
537
|
-
* - Requires compiler capabilities: +sve.
|
|
538
|
-
*/
|
|
539
|
-
|
|
540
|
-
__attribute__((target("+sve"))) //
|
|
541
|
-
inline static simsimd_f32_t
|
|
542
|
-
simsimd_sve_f64_l2sq(simsimd_f64_t const* a, simsimd_f64_t const* b, simsimd_size_t n) {
|
|
543
|
-
simsimd_size_t i = 0;
|
|
544
|
-
svfloat64_t d2_vec = svdupq_n_f64(0.0, 0.0);
|
|
545
|
-
do {
|
|
546
|
-
svbool_t pg_vec = svwhilelt_b64((unsigned int)i, (unsigned int)n);
|
|
547
|
-
svfloat64_t a_vec = svld1_f64(pg_vec, a + i);
|
|
548
|
-
svfloat64_t b_vec = svld1_f64(pg_vec, b + i);
|
|
549
|
-
svfloat64_t a_minus_b_vec = svsub_f64_x(pg_vec, a_vec, b_vec);
|
|
550
|
-
d2_vec = svmla_f64_x(pg_vec, d2_vec, a_minus_b_vec, a_minus_b_vec);
|
|
551
|
-
i += svcntd();
|
|
552
|
-
} while (i < n);
|
|
553
|
-
simsimd_f64_t d2 = svaddv_f64(svptrue_b32(), d2_vec);
|
|
554
|
-
return d2;
|
|
555
|
-
}
|
|
556
|
-
__attribute__((target("+sve"))) //
|
|
557
|
-
inline static simsimd_f32_t
|
|
558
|
-
simsimd_sve_f64_ip(simsimd_f64_t const* a, simsimd_f64_t const* b, simsimd_size_t n) {
|
|
559
|
-
simsimd_size_t i = 0;
|
|
560
|
-
svfloat64_t ab_vec = svdupq_n_f64(0.0, 0.0);
|
|
561
|
-
do {
|
|
562
|
-
svbool_t pg_vec = svwhilelt_b64((unsigned int)i, (unsigned int)n);
|
|
563
|
-
svfloat64_t a_vec = svld1_f64(pg_vec, a + i);
|
|
564
|
-
svfloat64_t b_vec = svld1_f64(pg_vec, b + i);
|
|
565
|
-
ab_vec = svmla_f64_x(pg_vec, ab_vec, a_vec, b_vec);
|
|
566
|
-
i += svcntd();
|
|
567
|
-
} while (i < n);
|
|
568
|
-
return 1 - svaddv_f64(svptrue_b32(), ab_vec);
|
|
569
|
-
}
|
|
570
|
-
|
|
571
|
-
__attribute__((target("+sve"))) //
|
|
572
|
-
inline static simsimd_f32_t
|
|
573
|
-
simsimd_sve_f64_cos(simsimd_f64_t const* a, simsimd_f64_t const* b, simsimd_size_t n) {
|
|
574
|
-
simsimd_size_t i = 0;
|
|
575
|
-
svfloat64_t ab_vec = svdupq_n_f64(0.0, 0.0);
|
|
576
|
-
svfloat64_t a2_vec = svdupq_n_f64(0.0, 0.0);
|
|
577
|
-
svfloat64_t b2_vec = svdupq_n_f64(0.0, 0.0);
|
|
578
|
-
do {
|
|
579
|
-
svbool_t pg_vec = svwhilelt_b64((unsigned int)i, (unsigned int)n);
|
|
580
|
-
svfloat64_t a_vec = svld1_f64(pg_vec, a + i);
|
|
581
|
-
svfloat64_t b_vec = svld1_f64(pg_vec, b + i);
|
|
582
|
-
ab_vec = svmla_f64_x(pg_vec, ab_vec, a_vec, b_vec);
|
|
583
|
-
a2_vec = svmla_f64_x(pg_vec, a2_vec, a_vec, a_vec);
|
|
584
|
-
b2_vec = svmla_f64_x(pg_vec, b2_vec, b_vec, b_vec);
|
|
585
|
-
i += svcntd();
|
|
586
|
-
} while (i < n);
|
|
587
|
-
|
|
588
|
-
simsimd_f64_t ab = svaddv_f64(svptrue_b32(), ab_vec);
|
|
589
|
-
simsimd_f64_t a2 = svaddv_f64(svptrue_b32(), a2_vec);
|
|
590
|
-
simsimd_f64_t b2 = svaddv_f64(svptrue_b32(), b2_vec);
|
|
591
|
-
|
|
592
|
-
// Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
|
|
593
|
-
simsimd_f64_t a2_b2_arr[2] = {a2, b2};
|
|
594
|
-
float64x2_t a2_b2 = vld1q_f64(a2_b2_arr);
|
|
595
|
-
a2_b2 = vrsqrteq_f64(a2_b2);
|
|
596
|
-
vst1q_f64(a2_b2_arr, a2_b2);
|
|
597
|
-
return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
|
|
598
|
-
}
|
|
599
|
-
|
|
600
|
-
#endif // SIMSIMD_TARGET_ARM_SVE
|
|
601
|
-
#endif // SIMSIMD_TARGET_ARM
|
|
602
|
-
|
|
603
|
-
#if SIMSIMD_TARGET_X86
|
|
604
|
-
#if SIMSIMD_TARGET_X86_AVX2
|
|
605
|
-
|
|
606
|
-
/*
|
|
607
|
-
* @file x86_avx2_f16.h
|
|
608
|
-
* @brief x86 AVX2 implementation of the most common similarity metrics for 16-bit floating point numbers.
|
|
609
|
-
* @author Ash Vardanian
|
|
610
|
-
*
|
|
611
|
-
* - Implements: L2 squared, inner product, cosine similarity.
|
|
612
|
-
* - As AVX2 doesn't support masked loads of 16-bit words, implementations have a separate `for`-loop for tails.
|
|
613
|
-
* - Uses `f16` for both storage and `f32` for accumulation.
|
|
614
|
-
* - Requires compiler capabilities: avx2, f16c, fma.
|
|
615
|
-
*/
|
|
616
|
-
|
|
617
|
-
__attribute__((target("avx2,f16c,fma"))) //
|
|
618
|
-
inline static simsimd_f32_t
|
|
619
|
-
simsimd_avx2_f16_l2sq(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
|
|
620
|
-
__m256 d2_vec = _mm256_setzero_ps();
|
|
621
|
-
simsimd_size_t i = 0;
|
|
622
|
-
for (; i + 8 <= n; i += 8) {
|
|
623
|
-
__m256 a_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(a + i)));
|
|
624
|
-
__m256 b_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(b + i)));
|
|
625
|
-
__m256 d_vec = _mm256_sub_ps(a_vec, b_vec);
|
|
626
|
-
d2_vec = _mm256_fmadd_ps(d_vec, d_vec, d2_vec);
|
|
627
|
-
}
|
|
628
|
-
|
|
629
|
-
// In case the software emulation for `f16` scalars is enabled, the `simsimd_uncompress_f16`
|
|
630
|
-
// function will run. It is extremely slow, so even for the tail, let's combine serial
|
|
631
|
-
// loads and stores with vectorized math.
|
|
632
|
-
if (i < n) {
|
|
633
|
-
union {
|
|
634
|
-
__m128i f16_vec;
|
|
635
|
-
simsimd_f16_t f16[8];
|
|
636
|
-
} a_padded_tail, b_padded_tail;
|
|
637
|
-
simsimd_size_t j = 0;
|
|
638
|
-
for (; i < n; ++i, ++j)
|
|
639
|
-
a_padded_tail.f16[j] = a[i], b_padded_tail.f16[j] = b[i];
|
|
640
|
-
for (; j < 8; ++j)
|
|
641
|
-
a_padded_tail.f16[j] = 0, b_padded_tail.f16[j] = 0;
|
|
642
|
-
__m256 a_vec = _mm256_cvtph_ps(a_padded_tail.f16_vec);
|
|
643
|
-
__m256 b_vec = _mm256_cvtph_ps(b_padded_tail.f16_vec);
|
|
644
|
-
__m256 d_vec = _mm256_sub_ps(a_vec, b_vec);
|
|
645
|
-
d2_vec = _mm256_fmadd_ps(d_vec, d_vec, d2_vec);
|
|
646
|
-
}
|
|
647
|
-
|
|
648
|
-
d2_vec = _mm256_add_ps(_mm256_permute2f128_ps(d2_vec, d2_vec, 1), d2_vec);
|
|
649
|
-
d2_vec = _mm256_hadd_ps(d2_vec, d2_vec);
|
|
650
|
-
d2_vec = _mm256_hadd_ps(d2_vec, d2_vec);
|
|
651
|
-
|
|
652
|
-
simsimd_f32_t result;
|
|
653
|
-
_mm_store_ss(&result, _mm256_castps256_ps128(d2_vec));
|
|
654
|
-
return result;
|
|
655
|
-
}
|
|
656
|
-
|
|
657
|
-
__attribute__((target("avx2,f16c,fma"))) //
|
|
658
|
-
inline static simsimd_f32_t
|
|
659
|
-
simsimd_avx2_f16_ip(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
|
|
660
|
-
__m256 ab_vec = _mm256_setzero_ps();
|
|
661
|
-
simsimd_size_t i = 0;
|
|
662
|
-
for (; i + 8 <= n; i += 8) {
|
|
663
|
-
__m256 a_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(a + i)));
|
|
664
|
-
__m256 b_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(b + i)));
|
|
665
|
-
ab_vec = _mm256_fmadd_ps(a_vec, b_vec, ab_vec);
|
|
666
|
-
}
|
|
667
|
-
|
|
668
|
-
// In case the software emulation for `f16` scalars is enabled, the `simsimd_uncompress_f16`
|
|
669
|
-
// function will run. It is extremely slow, so even for the tail, let's combine serial
|
|
670
|
-
// loads and stores with vectorized math.
|
|
671
|
-
if (i < n) {
|
|
672
|
-
union {
|
|
673
|
-
__m128i f16_vec;
|
|
674
|
-
simsimd_f16_t f16[8];
|
|
675
|
-
} a_padded_tail, b_padded_tail;
|
|
676
|
-
simsimd_size_t j = 0;
|
|
677
|
-
for (; i < n; ++i, ++j)
|
|
678
|
-
a_padded_tail.f16[j] = a[i], b_padded_tail.f16[j] = b[i];
|
|
679
|
-
for (; j < 8; ++j)
|
|
680
|
-
a_padded_tail.f16[j] = 0, b_padded_tail.f16[j] = 0;
|
|
681
|
-
__m256 a_vec = _mm256_cvtph_ps(a_padded_tail.f16_vec);
|
|
682
|
-
__m256 b_vec = _mm256_cvtph_ps(b_padded_tail.f16_vec);
|
|
683
|
-
ab_vec = _mm256_fmadd_ps(a_vec, b_vec, ab_vec);
|
|
684
|
-
}
|
|
685
|
-
|
|
686
|
-
ab_vec = _mm256_add_ps(_mm256_permute2f128_ps(ab_vec, ab_vec, 1), ab_vec);
|
|
687
|
-
ab_vec = _mm256_hadd_ps(ab_vec, ab_vec);
|
|
688
|
-
ab_vec = _mm256_hadd_ps(ab_vec, ab_vec);
|
|
689
|
-
|
|
690
|
-
simsimd_f32_t result;
|
|
691
|
-
_mm_store_ss(&result, _mm256_castps256_ps128(ab_vec));
|
|
692
|
-
return 1 - result;
|
|
693
|
-
}
|
|
694
|
-
|
|
695
|
-
__attribute__((target("avx2,f16c,fma"))) //
|
|
696
|
-
inline static simsimd_f32_t
|
|
697
|
-
simsimd_avx2_f16_cos(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
|
|
698
|
-
|
|
699
|
-
__m256 ab_vec = _mm256_setzero_ps(), a2_vec = _mm256_setzero_ps(), b2_vec = _mm256_setzero_ps();
|
|
700
|
-
simsimd_size_t i = 0;
|
|
701
|
-
for (; i + 8 <= n; i += 8) {
|
|
702
|
-
__m256 a_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(a + i)));
|
|
703
|
-
__m256 b_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(b + i)));
|
|
704
|
-
ab_vec = _mm256_fmadd_ps(a_vec, b_vec, ab_vec);
|
|
705
|
-
a2_vec = _mm256_fmadd_ps(a_vec, a_vec, a2_vec);
|
|
706
|
-
b2_vec = _mm256_fmadd_ps(b_vec, b_vec, b2_vec);
|
|
707
|
-
}
|
|
708
|
-
|
|
709
|
-
// In case the software emulation for `f16` scalars is enabled, the `simsimd_uncompress_f16`
|
|
710
|
-
// function will run. It is extremely slow, so even for the tail, let's combine serial
|
|
711
|
-
// loads and stores with vectorized math.
|
|
712
|
-
if (i < n) {
|
|
713
|
-
union {
|
|
714
|
-
__m128i f16_vec;
|
|
715
|
-
simsimd_f16_t f16[8];
|
|
716
|
-
} a_padded_tail, b_padded_tail;
|
|
717
|
-
simsimd_size_t j = 0;
|
|
718
|
-
for (; i < n; ++i, ++j)
|
|
719
|
-
a_padded_tail.f16[j] = a[i], b_padded_tail.f16[j] = b[i];
|
|
720
|
-
for (; j < 8; ++j)
|
|
721
|
-
a_padded_tail.f16[j] = 0, b_padded_tail.f16[j] = 0;
|
|
722
|
-
__m256 a_vec = _mm256_cvtph_ps(a_padded_tail.f16_vec);
|
|
723
|
-
__m256 b_vec = _mm256_cvtph_ps(b_padded_tail.f16_vec);
|
|
724
|
-
ab_vec = _mm256_fmadd_ps(a_vec, b_vec, ab_vec);
|
|
725
|
-
a2_vec = _mm256_fmadd_ps(a_vec, a_vec, a2_vec);
|
|
726
|
-
b2_vec = _mm256_fmadd_ps(b_vec, b_vec, b2_vec);
|
|
727
|
-
}
|
|
728
|
-
|
|
729
|
-
// Horizontal reductions:
|
|
730
|
-
ab_vec = _mm256_add_ps(_mm256_permute2f128_ps(ab_vec, ab_vec, 1), ab_vec);
|
|
731
|
-
ab_vec = _mm256_hadd_ps(ab_vec, ab_vec);
|
|
732
|
-
ab_vec = _mm256_hadd_ps(ab_vec, ab_vec);
|
|
733
|
-
|
|
734
|
-
a2_vec = _mm256_add_ps(_mm256_permute2f128_ps(a2_vec, a2_vec, 1), a2_vec);
|
|
735
|
-
a2_vec = _mm256_hadd_ps(a2_vec, a2_vec);
|
|
736
|
-
a2_vec = _mm256_hadd_ps(a2_vec, a2_vec);
|
|
737
|
-
|
|
738
|
-
b2_vec = _mm256_add_ps(_mm256_permute2f128_ps(b2_vec, b2_vec, 1), b2_vec);
|
|
739
|
-
b2_vec = _mm256_hadd_ps(b2_vec, b2_vec);
|
|
740
|
-
b2_vec = _mm256_hadd_ps(b2_vec, b2_vec);
|
|
741
|
-
|
|
742
|
-
simsimd_f32_t ab, a2, b2;
|
|
743
|
-
_mm_store_ss(&ab, _mm256_castps256_ps128(ab_vec));
|
|
744
|
-
_mm_store_ss(&a2, _mm256_castps256_ps128(a2_vec));
|
|
745
|
-
_mm_store_ss(&b2, _mm256_castps256_ps128(b2_vec));
|
|
746
|
-
|
|
747
|
-
// Replace simsimd_approximate_inverse_square_root with `rsqrtss`
|
|
748
|
-
__m128 a2_sqrt_recip = _mm_rsqrt_ss(_mm_set_ss((float)a2));
|
|
749
|
-
__m128 b2_sqrt_recip = _mm_rsqrt_ss(_mm_set_ss((float)b2));
|
|
750
|
-
__m128 result = _mm_mul_ss(a2_sqrt_recip, b2_sqrt_recip); // Multiply the reciprocal square roots
|
|
751
|
-
result = _mm_mul_ss(result, _mm_set_ss((float)ab)); // Multiply by ab
|
|
752
|
-
result = _mm_sub_ss(_mm_set_ss(1.0f), result); // Subtract from 1
|
|
753
|
-
return ab != 0 ? _mm_cvtss_f32(result) : 1; // Extract the final result
|
|
754
|
-
}
|
|
755
|
-
|
|
756
|
-
/*
|
|
757
|
-
* @file x86_avx2_i8.h
|
|
758
|
-
* @brief x86 AVX2 implementation of the most common similarity metrics for 8-bit signed integral numbers.
|
|
759
|
-
* @author Ash Vardanian
|
|
760
|
-
*
|
|
761
|
-
* - Implements: L2 squared, cosine similarity, inner product (same as cosine).
|
|
762
|
-
* - As AVX2 doesn't support masked loads of 16-bit words, implementations have a separate `for`-loop for tails.
|
|
763
|
-
* - Uses `i8` for storage, `i16` for multiplication, and `i32` for accumulation, if no better option is available.
|
|
764
|
-
* - Requires compiler capabilities: avx2.
|
|
765
|
-
*/
|
|
766
|
-
|
|
767
|
-
__attribute__((target("avx2"))) //
|
|
768
|
-
inline static simsimd_f32_t
|
|
769
|
-
simsimd_avx2_i8_l2sq(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
770
|
-
|
|
771
|
-
__m256i d2_high_vec = _mm256_setzero_si256();
|
|
772
|
-
__m256i d2_low_vec = _mm256_setzero_si256();
|
|
773
|
-
|
|
774
|
-
simsimd_size_t i = 0;
|
|
775
|
-
for (; i + 32 <= n; i += 32) {
|
|
776
|
-
__m256i a_vec = _mm256_loadu_si256((__m256i const*)(a + i));
|
|
777
|
-
__m256i b_vec = _mm256_loadu_si256((__m256i const*)(b + i));
|
|
778
|
-
|
|
779
|
-
// Sign extend int8 to int16
|
|
780
|
-
__m256i a_low = _mm256_cvtepi8_epi16(_mm256_castsi256_si128(a_vec));
|
|
781
|
-
__m256i a_high = _mm256_cvtepi8_epi16(_mm256_extracti128_si256(a_vec, 1));
|
|
782
|
-
__m256i b_low = _mm256_cvtepi8_epi16(_mm256_castsi256_si128(b_vec));
|
|
783
|
-
__m256i b_high = _mm256_cvtepi8_epi16(_mm256_extracti128_si256(b_vec, 1));
|
|
784
|
-
|
|
785
|
-
// Subtract and multiply
|
|
786
|
-
__m256i d_low = _mm256_sub_epi16(a_low, b_low);
|
|
787
|
-
__m256i d_high = _mm256_sub_epi16(a_high, b_high);
|
|
788
|
-
__m256i d2_low_part = _mm256_madd_epi16(d_low, d_low);
|
|
789
|
-
__m256i d2_high_part = _mm256_madd_epi16(d_high, d_high);
|
|
790
|
-
|
|
791
|
-
// Accumulate into int32 vectors
|
|
792
|
-
d2_low_vec = _mm256_add_epi32(d2_low_vec, d2_low_part);
|
|
793
|
-
d2_high_vec = _mm256_add_epi32(d2_high_vec, d2_high_part);
|
|
794
|
-
}
|
|
795
|
-
|
|
796
|
-
// Accumulate the 32-bit integers from `d2_high_vec` and `d2_low_vec`
|
|
797
|
-
__m256i d2_vec = _mm256_add_epi32(d2_low_vec, d2_high_vec);
|
|
798
|
-
__m128i d2_sum = _mm_add_epi32(_mm256_extracti128_si256(d2_vec, 0), _mm256_extracti128_si256(d2_vec, 1));
|
|
799
|
-
d2_sum = _mm_hadd_epi32(d2_sum, d2_sum);
|
|
800
|
-
d2_sum = _mm_hadd_epi32(d2_sum, d2_sum);
|
|
801
|
-
int d2 = _mm_extract_epi32(d2_sum, 0);
|
|
802
|
-
|
|
803
|
-
// Take care of the tail:
|
|
804
|
-
for (; i < n; ++i) {
|
|
805
|
-
int n = a[i] - b[i];
|
|
806
|
-
d2 += n * n;
|
|
807
|
-
}
|
|
808
|
-
|
|
809
|
-
return (simsimd_f32_t)d2;
|
|
810
|
-
}
|
|
811
|
-
__attribute__((target("avx2"))) //
|
|
812
|
-
inline static simsimd_f32_t
|
|
813
|
-
simsimd_avx2_i8_cos(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
814
|
-
|
|
815
|
-
__m256i ab_low_vec = _mm256_setzero_si256();
|
|
816
|
-
__m256i ab_high_vec = _mm256_setzero_si256();
|
|
817
|
-
__m256i a2_low_vec = _mm256_setzero_si256();
|
|
818
|
-
__m256i a2_high_vec = _mm256_setzero_si256();
|
|
819
|
-
__m256i b2_low_vec = _mm256_setzero_si256();
|
|
820
|
-
__m256i b2_high_vec = _mm256_setzero_si256();
|
|
821
|
-
|
|
822
|
-
simsimd_size_t i = 0;
|
|
823
|
-
for (; i + 32 <= n; i += 32) {
|
|
824
|
-
__m256i a_vec = _mm256_loadu_si256((__m256i const*)(a + i));
|
|
825
|
-
__m256i b_vec = _mm256_loadu_si256((__m256i const*)(b + i));
|
|
826
|
-
|
|
827
|
-
// Unpack int8 to int16
|
|
828
|
-
__m256i a_low_16 = _mm256_cvtepi8_epi16(_mm256_extracti128_si256(a_vec, 0));
|
|
829
|
-
__m256i a_high_16 = _mm256_cvtepi8_epi16(_mm256_extracti128_si256(a_vec, 1));
|
|
830
|
-
__m256i b_low_16 = _mm256_cvtepi8_epi16(_mm256_extracti128_si256(b_vec, 0));
|
|
831
|
-
__m256i b_high_16 = _mm256_cvtepi8_epi16(_mm256_extracti128_si256(b_vec, 1));
|
|
832
|
-
|
|
833
|
-
// Multiply and accumulate at int16 level, accumulate at int32 level
|
|
834
|
-
ab_low_vec = _mm256_add_epi32(ab_low_vec, _mm256_madd_epi16(a_low_16, b_low_16));
|
|
835
|
-
ab_high_vec = _mm256_add_epi32(ab_high_vec, _mm256_madd_epi16(a_high_16, b_high_16));
|
|
836
|
-
a2_low_vec = _mm256_add_epi32(a2_low_vec, _mm256_madd_epi16(a_low_16, a_low_16));
|
|
837
|
-
a2_high_vec = _mm256_add_epi32(a2_high_vec, _mm256_madd_epi16(a_high_16, a_high_16));
|
|
838
|
-
b2_low_vec = _mm256_add_epi32(b2_low_vec, _mm256_madd_epi16(b_low_16, b_low_16));
|
|
839
|
-
b2_high_vec = _mm256_add_epi32(b2_high_vec, _mm256_madd_epi16(b_high_16, b_high_16));
|
|
840
|
-
}
|
|
841
|
-
|
|
842
|
-
// Horizontal sum across the 256-bit register
|
|
843
|
-
__m256i ab_vec = _mm256_add_epi32(ab_low_vec, ab_high_vec);
|
|
844
|
-
__m128i ab_sum = _mm_add_epi32(_mm256_extracti128_si256(ab_vec, 0), _mm256_extracti128_si256(ab_vec, 1));
|
|
845
|
-
ab_sum = _mm_hadd_epi32(ab_sum, ab_sum);
|
|
846
|
-
ab_sum = _mm_hadd_epi32(ab_sum, ab_sum);
|
|
847
|
-
|
|
848
|
-
__m256i a2_vec = _mm256_add_epi32(a2_low_vec, a2_high_vec);
|
|
849
|
-
__m128i a2_sum = _mm_add_epi32(_mm256_extracti128_si256(a2_vec, 0), _mm256_extracti128_si256(a2_vec, 1));
|
|
850
|
-
a2_sum = _mm_hadd_epi32(a2_sum, a2_sum);
|
|
851
|
-
a2_sum = _mm_hadd_epi32(a2_sum, a2_sum);
|
|
852
|
-
|
|
853
|
-
__m256i b2_vec = _mm256_add_epi32(b2_low_vec, b2_high_vec);
|
|
854
|
-
__m128i b2_sum = _mm_add_epi32(_mm256_extracti128_si256(b2_vec, 0), _mm256_extracti128_si256(b2_vec, 1));
|
|
855
|
-
b2_sum = _mm_hadd_epi32(b2_sum, b2_sum);
|
|
856
|
-
b2_sum = _mm_hadd_epi32(b2_sum, b2_sum);
|
|
857
|
-
|
|
858
|
-
// Further reduce to a single sum for each vector
|
|
859
|
-
int ab = _mm_extract_epi32(ab_sum, 0);
|
|
860
|
-
int a2 = _mm_extract_epi32(a2_sum, 0);
|
|
861
|
-
int b2 = _mm_extract_epi32(b2_sum, 0);
|
|
862
|
-
|
|
863
|
-
// Take care of the tail:
|
|
864
|
-
for (; i < n; ++i) {
|
|
865
|
-
int ai = a[i], bi = b[i];
|
|
866
|
-
ab += ai * bi, a2 += ai * ai, b2 += bi * bi;
|
|
867
|
-
}
|
|
868
|
-
|
|
869
|
-
// Compute the reciprocal of the square roots
|
|
870
|
-
__m128 a2_sqrt_recip = _mm_rsqrt_ss(_mm_set_ss((float)a2));
|
|
871
|
-
__m128 b2_sqrt_recip = _mm_rsqrt_ss(_mm_set_ss((float)b2));
|
|
872
|
-
|
|
873
|
-
// Compute cosine similarity: ab / sqrt(a2 * b2)
|
|
874
|
-
__m128 denom = _mm_mul_ss(a2_sqrt_recip, b2_sqrt_recip); // Reciprocal of sqrt(a2 * b2)
|
|
875
|
-
__m128 result = _mm_mul_ss(_mm_set_ss((float)ab), denom); // ab * reciprocal of sqrt(a2 * b2)
|
|
876
|
-
|
|
877
|
-
return ab != 0 ? 1 - _mm_cvtss_f32(result) : 0; // Extract the final result
|
|
878
|
-
}
|
|
879
|
-
|
|
880
|
-
__attribute__((target("avx2"))) //
|
|
881
|
-
inline static simsimd_f32_t
|
|
882
|
-
simsimd_avx2_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
883
|
-
return simsimd_avx2_i8_cos(a, b, n);
|
|
884
|
-
}
|
|
885
|
-
|
|
886
|
-
#endif // SIMSIMD_TARGET_X86_AVX2
|
|
887
|
-
|
|
888
|
-
#if SIMSIMD_TARGET_X86_AVX512
|
|
889
|
-
|
|
890
|
-
/*
|
|
891
|
-
* @file x86_avx512_f32.h
|
|
892
|
-
* @brief x86 AVX-512 implementation of the most common similarity metrics for 32-bit floating point numbers.
|
|
893
|
-
* @author Ash Vardanian
|
|
894
|
-
*
|
|
895
|
-
* - Implements: L2 squared, inner product, cosine similarity.
|
|
896
|
-
* - Uses `f32` for storage and `f32` for accumulation.
|
|
897
|
-
* - Requires compiler capabilities: avx512f, avx512vl, bmi2.
|
|
898
|
-
*/
|
|
899
|
-
|
|
900
|
-
__attribute__((target("avx512f,avx512vl,bmi2"))) //
|
|
901
|
-
inline static simsimd_f32_t
|
|
902
|
-
simsimd_avx512_f32_l2sq(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
|
|
903
|
-
__m512 d2_vec = _mm512_setzero();
|
|
904
|
-
__m512 a_vec, b_vec;
|
|
905
|
-
|
|
906
|
-
simsimd_avx512_f32_l2sq_cycle:
|
|
907
|
-
if (n < 16) {
|
|
908
|
-
__mmask16 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
909
|
-
a_vec = _mm512_maskz_loadu_ps(mask, a);
|
|
910
|
-
b_vec = _mm512_maskz_loadu_ps(mask, b);
|
|
911
|
-
n = 0;
|
|
912
|
-
} else {
|
|
913
|
-
a_vec = _mm512_loadu_ps(a);
|
|
914
|
-
b_vec = _mm512_loadu_ps(b);
|
|
915
|
-
a += 16, b += 16, n -= 16;
|
|
916
|
-
}
|
|
917
|
-
__m512 d_vec = _mm512_sub_ps(a_vec, b_vec);
|
|
918
|
-
d2_vec = _mm512_fmadd_ps(d_vec, d_vec, d2_vec);
|
|
919
|
-
if (n)
|
|
920
|
-
goto simsimd_avx512_f32_l2sq_cycle;
|
|
921
|
-
|
|
922
|
-
return _mm512_reduce_add_ps(d2_vec);
|
|
923
|
-
}
|
|
924
|
-
|
|
925
|
-
__attribute__((target("avx512f,avx512vl,bmi2"))) //
|
|
926
|
-
inline static simsimd_f32_t
|
|
927
|
-
simsimd_avx512_f32_ip(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
|
|
928
|
-
__m512 ab_vec = _mm512_setzero();
|
|
929
|
-
__m512 a_vec, b_vec;
|
|
930
|
-
|
|
931
|
-
simsimd_avx512_f32_ip_cycle:
|
|
932
|
-
if (n < 16) {
|
|
933
|
-
__mmask16 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
934
|
-
a_vec = _mm512_maskz_loadu_ps(mask, a);
|
|
935
|
-
b_vec = _mm512_maskz_loadu_ps(mask, b);
|
|
936
|
-
n = 0;
|
|
937
|
-
} else {
|
|
938
|
-
a_vec = _mm512_loadu_ps(a);
|
|
939
|
-
b_vec = _mm512_loadu_ps(b);
|
|
940
|
-
a += 16, b += 16, n -= 16;
|
|
941
|
-
}
|
|
942
|
-
ab_vec = _mm512_fmadd_ps(a_vec, b_vec, ab_vec);
|
|
943
|
-
if (n)
|
|
944
|
-
goto simsimd_avx512_f32_ip_cycle;
|
|
945
|
-
|
|
946
|
-
return 1 - _mm512_reduce_add_ps(ab_vec);
|
|
947
|
-
}
|
|
948
|
-
|
|
949
|
-
__attribute__((target("avx512f,avx512vl,bmi2"))) //
|
|
950
|
-
inline static simsimd_f32_t
|
|
951
|
-
simsimd_avx512_f32_cos(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
|
|
952
|
-
__m512 ab_vec = _mm512_setzero();
|
|
953
|
-
__m512 a2_vec = _mm512_setzero();
|
|
954
|
-
__m512 b2_vec = _mm512_setzero();
|
|
955
|
-
__m512 a_vec, b_vec;
|
|
956
|
-
|
|
957
|
-
simsimd_avx512_f32_cos_cycle:
|
|
958
|
-
if (n < 16) {
|
|
959
|
-
__mmask16 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
960
|
-
a_vec = _mm512_maskz_loadu_ps(mask, a);
|
|
961
|
-
b_vec = _mm512_maskz_loadu_ps(mask, b);
|
|
962
|
-
n = 0;
|
|
963
|
-
} else {
|
|
964
|
-
a_vec = _mm512_loadu_ps(a);
|
|
965
|
-
b_vec = _mm512_loadu_ps(b);
|
|
966
|
-
a += 16, b += 16, n -= 16;
|
|
967
|
-
}
|
|
968
|
-
ab_vec = _mm512_fmadd_ps(a_vec, b_vec, ab_vec);
|
|
969
|
-
a2_vec = _mm512_fmadd_ps(a_vec, a_vec, a2_vec);
|
|
970
|
-
b2_vec = _mm512_fmadd_ps(b_vec, b_vec, b2_vec);
|
|
971
|
-
if (n)
|
|
972
|
-
goto simsimd_avx512_f32_cos_cycle;
|
|
973
|
-
|
|
974
|
-
simsimd_f32_t ab = _mm512_reduce_add_ps(ab_vec);
|
|
975
|
-
simsimd_f32_t a2 = _mm512_reduce_add_ps(a2_vec);
|
|
976
|
-
simsimd_f32_t b2 = _mm512_reduce_add_ps(b2_vec);
|
|
977
|
-
|
|
978
|
-
// Compute the reciprocal square roots of a2 and b2
|
|
979
|
-
__m128 rsqrts = _mm_rsqrt14_ps(_mm_set_ps(0.f, 0.f, a2 + 1.e-9f, b2 + 1.e-9f));
|
|
980
|
-
simsimd_f32_t rsqrt_a2 = _mm_cvtss_f32(rsqrts);
|
|
981
|
-
simsimd_f32_t rsqrt_b2 = _mm_cvtss_f32(_mm_shuffle_ps(rsqrts, rsqrts, _MM_SHUFFLE(0, 0, 0, 1)));
|
|
982
|
-
return 1 - ab * rsqrt_a2 * rsqrt_b2;
|
|
983
|
-
}
|
|
984
|
-
|
|
985
|
-
/*
|
|
986
|
-
* @file x86_avx512_f16.h
|
|
987
|
-
* @brief x86 AVX-512 implementation of the most common similarity metrics for 16-bit floating point numbers.
|
|
988
|
-
* @author Ash Vardanian
|
|
989
|
-
*
|
|
990
|
-
* - Implements: L2 squared, inner product, cosine similarity.
|
|
991
|
-
* - Uses `_mm512_maskz_loadu_epi16` intrinsics to perform masked unaligned loads.
|
|
992
|
-
* - Uses `f16` for both storage and accumulation, assuming it's resolution is enough for average case.
|
|
993
|
-
* - Requires compiler capabilities: avx512fp16, avx512f, avx512vl, bmi2.
|
|
994
|
-
*/
|
|
995
|
-
|
|
996
|
-
__attribute__((target("avx512fp16,avx512vl,avx512f,bmi2"))) //
|
|
997
|
-
inline static simsimd_f32_t
|
|
998
|
-
simsimd_avx512_f16_l2sq(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
|
|
999
|
-
__m512h d2_vec = _mm512_setzero_ph();
|
|
1000
|
-
__m512i a_i16_vec, b_i16_vec;
|
|
1001
|
-
|
|
1002
|
-
simsimd_avx512_f16_l2sq_cycle:
|
|
1003
|
-
if (n < 32) {
|
|
1004
|
-
__mmask32 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
1005
|
-
a_i16_vec = _mm512_maskz_loadu_epi16(mask, a);
|
|
1006
|
-
b_i16_vec = _mm512_maskz_loadu_epi16(mask, b);
|
|
1007
|
-
n = 0;
|
|
1008
|
-
} else {
|
|
1009
|
-
a_i16_vec = _mm512_loadu_epi16(a);
|
|
1010
|
-
b_i16_vec = _mm512_loadu_epi16(b);
|
|
1011
|
-
a += 32, b += 32, n -= 32;
|
|
1012
|
-
}
|
|
1013
|
-
__m512h d_vec = _mm512_sub_ph(_mm512_castsi512_ph(a_i16_vec), _mm512_castsi512_ph(b_i16_vec));
|
|
1014
|
-
d2_vec = _mm512_fmadd_ph(d_vec, d_vec, d2_vec);
|
|
1015
|
-
if (n)
|
|
1016
|
-
goto simsimd_avx512_f16_l2sq_cycle;
|
|
1017
|
-
|
|
1018
|
-
return _mm512_reduce_add_ph(d2_vec);
|
|
1019
|
-
}
|
|
1020
|
-
|
|
1021
|
-
__attribute__((target("avx512fp16,avx512vl,avx512f,bmi2"))) //
|
|
1022
|
-
inline static simsimd_f32_t
|
|
1023
|
-
simsimd_avx512_f16_ip(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
|
|
1024
|
-
__m512h ab_vec = _mm512_setzero_ph();
|
|
1025
|
-
__m512i a_i16_vec, b_i16_vec;
|
|
1026
|
-
|
|
1027
|
-
simsimd_avx512_f16_ip_cycle:
|
|
1028
|
-
if (n < 32) {
|
|
1029
|
-
__mmask32 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
1030
|
-
a_i16_vec = _mm512_maskz_loadu_epi16(mask, a);
|
|
1031
|
-
b_i16_vec = _mm512_maskz_loadu_epi16(mask, b);
|
|
1032
|
-
n = 0;
|
|
1033
|
-
} else {
|
|
1034
|
-
a_i16_vec = _mm512_loadu_epi16(a);
|
|
1035
|
-
b_i16_vec = _mm512_loadu_epi16(b);
|
|
1036
|
-
a += 32, b += 32, n -= 32;
|
|
1037
|
-
}
|
|
1038
|
-
ab_vec = _mm512_fmadd_ph(_mm512_castsi512_ph(a_i16_vec), _mm512_castsi512_ph(b_i16_vec), ab_vec);
|
|
1039
|
-
if (n)
|
|
1040
|
-
goto simsimd_avx512_f16_ip_cycle;
|
|
1041
|
-
|
|
1042
|
-
return 1 - _mm512_reduce_add_ph(ab_vec);
|
|
1043
|
-
}
|
|
1044
|
-
|
|
1045
|
-
__attribute__((target("avx512fp16,avx512vl,avx512f,bmi2"))) //
|
|
1046
|
-
inline static simsimd_f32_t
|
|
1047
|
-
simsimd_avx512_f16_cos(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
|
|
1048
|
-
__m512h ab_vec = _mm512_setzero_ph();
|
|
1049
|
-
__m512h a2_vec = _mm512_setzero_ph();
|
|
1050
|
-
__m512h b2_vec = _mm512_setzero_ph();
|
|
1051
|
-
__m512i a_i16_vec, b_i16_vec;
|
|
1052
|
-
|
|
1053
|
-
simsimd_avx512_f16_cos_cycle:
|
|
1054
|
-
if (n < 32) {
|
|
1055
|
-
__mmask32 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
1056
|
-
a_i16_vec = _mm512_maskz_loadu_epi16(mask, a);
|
|
1057
|
-
b_i16_vec = _mm512_maskz_loadu_epi16(mask, b);
|
|
1058
|
-
n = 0;
|
|
1059
|
-
} else {
|
|
1060
|
-
a_i16_vec = _mm512_loadu_epi16(a);
|
|
1061
|
-
b_i16_vec = _mm512_loadu_epi16(b);
|
|
1062
|
-
a += 32, b += 32, n -= 32;
|
|
1063
|
-
}
|
|
1064
|
-
ab_vec = _mm512_fmadd_ph(_mm512_castsi512_ph(a_i16_vec), _mm512_castsi512_ph(b_i16_vec), ab_vec);
|
|
1065
|
-
a2_vec = _mm512_fmadd_ph(_mm512_castsi512_ph(a_i16_vec), _mm512_castsi512_ph(a_i16_vec), a2_vec);
|
|
1066
|
-
b2_vec = _mm512_fmadd_ph(_mm512_castsi512_ph(b_i16_vec), _mm512_castsi512_ph(b_i16_vec), b2_vec);
|
|
1067
|
-
if (n)
|
|
1068
|
-
goto simsimd_avx512_f16_cos_cycle;
|
|
1069
|
-
|
|
1070
|
-
simsimd_f32_t ab = _mm512_reduce_add_ph(ab_vec);
|
|
1071
|
-
simsimd_f32_t a2 = _mm512_reduce_add_ph(a2_vec);
|
|
1072
|
-
simsimd_f32_t b2 = _mm512_reduce_add_ph(b2_vec);
|
|
1073
|
-
|
|
1074
|
-
// Compute the reciprocal square roots of a2 and b2
|
|
1075
|
-
__m128 rsqrts = _mm_rsqrt14_ps(_mm_set_ps(0.f, 0.f, a2 + 1.e-9f, b2 + 1.e-9f));
|
|
1076
|
-
simsimd_f32_t rsqrt_a2 = _mm_cvtss_f32(rsqrts);
|
|
1077
|
-
simsimd_f32_t rsqrt_b2 = _mm_cvtss_f32(_mm_shuffle_ps(rsqrts, rsqrts, _MM_SHUFFLE(0, 0, 0, 1)));
|
|
1078
|
-
return 1 - ab * rsqrt_a2 * rsqrt_b2;
|
|
1079
|
-
}
|
|
1080
|
-
|
|
1081
|
-
/*
|
|
1082
|
-
* @file x86_avx512_i8.h
|
|
1083
|
-
* @brief x86 AVX-512 implementation of the most common similarity metrics for 8-bit integers.
|
|
1084
|
-
* @author Ash Vardanian
|
|
1085
|
-
*
|
|
1086
|
-
* - Implements: L2 squared, cosine similarity, inner product (same as cosine).
|
|
1087
|
-
* - Uses `_mm512_maskz_loadu_epi16` intrinsics to perform masked unaligned loads.
|
|
1088
|
-
* - Uses `i8` for storage, `i16` for multiplication, and `i32` for accumulation, if no better option is available.
|
|
1089
|
-
* - Uses BMI2 `bzhi` instructions to build masks without branches or conditional moves.
|
|
1090
|
-
* - Requires compiler capabilities: avx512f, avx512vl, avx512bw, avx512vnni, bmi2.
|
|
1091
|
-
*/
|
|
1092
|
-
|
|
1093
|
-
__attribute__((target("avx512vl,avx512f,avx512bw,avx512vnni,bmi2"))) //
|
|
1094
|
-
inline static simsimd_f32_t
|
|
1095
|
-
simsimd_avx512_i8_l2sq(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
1096
|
-
__m512i d2_i32s_vec = _mm512_setzero_si512();
|
|
1097
|
-
__m512i a_vec, b_vec, d_i16s_vec;
|
|
1098
|
-
|
|
1099
|
-
simsimd_avx512_i8_l2sq_cycle:
|
|
1100
|
-
if (n < 32) {
|
|
1101
|
-
__mmask32 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
1102
|
-
a_vec = _mm512_cvtepi8_epi16(_mm256_maskz_loadu_epi8(mask, a));
|
|
1103
|
-
b_vec = _mm512_cvtepi8_epi16(_mm256_maskz_loadu_epi8(mask, b));
|
|
1104
|
-
n = 0;
|
|
1105
|
-
} else {
|
|
1106
|
-
a_vec = _mm512_cvtepi8_epi16(_mm256_loadu_epi8(a));
|
|
1107
|
-
b_vec = _mm512_cvtepi8_epi16(_mm256_loadu_epi8(b));
|
|
1108
|
-
a += 32, b += 32, n -= 32;
|
|
1109
|
-
}
|
|
1110
|
-
d_i16s_vec = _mm512_sub_epi16(a_vec, b_vec);
|
|
1111
|
-
d2_i32s_vec = _mm512_dpwssd_epi32(d2_i32s_vec, d_i16s_vec, d_i16s_vec);
|
|
1112
|
-
if (n)
|
|
1113
|
-
goto simsimd_avx512_i8_l2sq_cycle;
|
|
1114
|
-
|
|
1115
|
-
return _mm512_reduce_add_epi32(d2_i32s_vec);
|
|
1116
|
-
}
|
|
1117
|
-
|
|
1118
|
-
__attribute__((target("avx512vl,avx512f,avx512bw,avx512vnni,bmi2"))) //
|
|
1119
|
-
inline static simsimd_f32_t
|
|
1120
|
-
simsimd_avx512_i8_cos(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
1121
|
-
__m512i ab_i32s_vec = _mm512_setzero_si512();
|
|
1122
|
-
__m512i a2_i32s_vec = _mm512_setzero_si512();
|
|
1123
|
-
__m512i b2_i32s_vec = _mm512_setzero_si512();
|
|
1124
|
-
__m512i a_vec, b_vec;
|
|
1125
|
-
|
|
1126
|
-
simsimd_avx512_i8_cos_cycle:
|
|
1127
|
-
if (n < 32) {
|
|
1128
|
-
__mmask32 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
1129
|
-
a_vec = _mm512_cvtepi8_epi16(_mm256_maskz_loadu_epi8(mask, a));
|
|
1130
|
-
b_vec = _mm512_cvtepi8_epi16(_mm256_maskz_loadu_epi8(mask, b));
|
|
1131
|
-
n = 0;
|
|
1132
|
-
} else {
|
|
1133
|
-
a_vec = _mm512_cvtepi8_epi16(_mm256_loadu_epi8(a));
|
|
1134
|
-
b_vec = _mm512_cvtepi8_epi16(_mm256_loadu_epi8(b));
|
|
1135
|
-
a += 32, b += 32, n -= 32;
|
|
1136
|
-
}
|
|
1137
|
-
// Unfortunately we can't use the `_mm512_dpbusd_epi32` intrinsics here either,
|
|
1138
|
-
// as it's assymetric with respect to the sign of the input arguments:
|
|
1139
|
-
// Signed(ZeroExtend16(a.byte[4*j]) * SignExtend16(b.byte[4*j]))
|
|
1140
|
-
// So we have to use the `_mm512_dpwssd_epi32` intrinsics instead, upcasting
|
|
1141
|
-
// to 16-bit beforehand.
|
|
1142
|
-
ab_i32s_vec = _mm512_dpwssd_epi32(ab_i32s_vec, a_vec, b_vec);
|
|
1143
|
-
a2_i32s_vec = _mm512_dpwssd_epi32(a2_i32s_vec, a_vec, a_vec);
|
|
1144
|
-
b2_i32s_vec = _mm512_dpwssd_epi32(b2_i32s_vec, b_vec, b_vec);
|
|
1145
|
-
if (n)
|
|
1146
|
-
goto simsimd_avx512_i8_cos_cycle;
|
|
1147
|
-
|
|
1148
|
-
simsimd_f32_t ab = _mm512_reduce_add_epi32(ab_i32s_vec);
|
|
1149
|
-
simsimd_f32_t a2 = _mm512_reduce_add_epi32(a2_i32s_vec);
|
|
1150
|
-
simsimd_f32_t b2 = _mm512_reduce_add_epi32(b2_i32s_vec);
|
|
1151
|
-
|
|
1152
|
-
// Compute the reciprocal square roots of a2 and b2
|
|
1153
|
-
__m128 rsqrts = _mm_rsqrt14_ps(_mm_set_ps(0.f, 0.f, a2 + 1.e-9f, b2 + 1.e-9f));
|
|
1154
|
-
simsimd_f32_t rsqrt_a2 = _mm_cvtss_f32(rsqrts);
|
|
1155
|
-
simsimd_f32_t rsqrt_b2 = _mm_cvtss_f32(_mm_shuffle_ps(rsqrts, rsqrts, _MM_SHUFFLE(0, 0, 0, 1)));
|
|
1156
|
-
return 1 - ab * rsqrt_a2 * rsqrt_b2;
|
|
1157
|
-
}
|
|
1158
|
-
|
|
1159
|
-
__attribute__((target("avx512vl,avx512f,avx512bw,avx512vnni,bmi2"))) //
|
|
1160
|
-
inline static simsimd_f32_t
|
|
1161
|
-
simsimd_avx512_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
|
|
1162
|
-
return simsimd_avx512_i8_cos(a, b, n);
|
|
1163
|
-
}
|
|
1164
|
-
|
|
1165
|
-
/*
|
|
1166
|
-
* @file x86_avx512_f64.h
|
|
1167
|
-
* @brief x86 AVX-512 implementation of the most common similarity metrics for 64-bit floating point numbers.
|
|
1168
|
-
* @author Ash Vardanian
|
|
1169
|
-
*
|
|
1170
|
-
* - Implements: L2 squared, inner product, cosine similarity.
|
|
1171
|
-
* - Uses `f64` for both storage and accumulation.
|
|
1172
|
-
* - Requires compiler capabilities: avx512f, avx512vl, bmi2.
|
|
1173
|
-
*/
|
|
1174
|
-
|
|
1175
|
-
__attribute__((target("avx512f,avx512vl,bmi2"))) //
|
|
1176
|
-
inline static simsimd_f32_t
|
|
1177
|
-
simsimd_avx512_f64_l2sq(simsimd_f64_t const* a, simsimd_f64_t const* b, simsimd_size_t n) {
|
|
1178
|
-
__m512d d2_vec = _mm512_setzero_pd();
|
|
1179
|
-
__m512d a_vec, b_vec;
|
|
1180
|
-
|
|
1181
|
-
simsimd_avx512_f64_l2sq_cycle:
|
|
1182
|
-
if (n < 8) {
|
|
1183
|
-
__mmask8 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
1184
|
-
a_vec = _mm512_maskz_loadu_pd(mask, a);
|
|
1185
|
-
b_vec = _mm512_maskz_loadu_pd(mask, b);
|
|
1186
|
-
n = 0;
|
|
1187
|
-
} else {
|
|
1188
|
-
a_vec = _mm512_loadu_pd(a);
|
|
1189
|
-
b_vec = _mm512_loadu_pd(b);
|
|
1190
|
-
a += 8, b += 8, n -= 8;
|
|
1191
|
-
}
|
|
1192
|
-
__m512d d_vec = _mm512_sub_pd(a_vec, b_vec);
|
|
1193
|
-
d2_vec = _mm512_fmadd_pd(d_vec, d_vec, d2_vec);
|
|
1194
|
-
if (n)
|
|
1195
|
-
goto simsimd_avx512_f64_l2sq_cycle;
|
|
1196
|
-
|
|
1197
|
-
return (simsimd_f32_t)_mm512_reduce_add_pd(d2_vec);
|
|
1198
|
-
}
|
|
1199
|
-
|
|
1200
|
-
__attribute__((target("avx512f,avx512vl,bmi2"))) //
|
|
1201
|
-
inline static simsimd_f32_t
|
|
1202
|
-
simsimd_avx512_f64_ip(simsimd_f64_t const* a, simsimd_f64_t const* b, simsimd_size_t n) {
|
|
1203
|
-
__m512d ab_vec = _mm512_setzero_pd();
|
|
1204
|
-
__m512d a_vec, b_vec;
|
|
1205
|
-
|
|
1206
|
-
simsimd_avx512_f64_ip_cycle:
|
|
1207
|
-
if (n < 8) {
|
|
1208
|
-
__mmask8 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
1209
|
-
a_vec = _mm512_maskz_loadu_pd(mask, a);
|
|
1210
|
-
b_vec = _mm512_maskz_loadu_pd(mask, b);
|
|
1211
|
-
n = 0;
|
|
1212
|
-
} else {
|
|
1213
|
-
a_vec = _mm512_loadu_pd(a);
|
|
1214
|
-
b_vec = _mm512_loadu_pd(b);
|
|
1215
|
-
a += 8, b += 8, n -= 8;
|
|
1216
|
-
}
|
|
1217
|
-
ab_vec = _mm512_fmadd_pd(a_vec, b_vec, ab_vec);
|
|
1218
|
-
if (n)
|
|
1219
|
-
goto simsimd_avx512_f64_ip_cycle;
|
|
1220
|
-
|
|
1221
|
-
return 1 - (simsimd_f32_t)_mm512_reduce_add_pd(ab_vec);
|
|
1222
|
-
}
|
|
1223
|
-
|
|
1224
|
-
__attribute__((target("avx512f,avx512vl,bmi2"))) //
|
|
1225
|
-
inline static simsimd_f32_t
|
|
1226
|
-
simsimd_avx512_f64_cos(simsimd_f64_t const* a, simsimd_f64_t const* b, simsimd_size_t n) {
|
|
1227
|
-
__m512d ab_vec = _mm512_setzero_pd();
|
|
1228
|
-
__m512d a2_vec = _mm512_setzero_pd();
|
|
1229
|
-
__m512d b2_vec = _mm512_setzero_pd();
|
|
1230
|
-
__m512d a_vec, b_vec;
|
|
1231
|
-
|
|
1232
|
-
simsimd_avx512_f64_cos_cycle:
|
|
1233
|
-
if (n < 8) {
|
|
1234
|
-
__mmask8 mask = _bzhi_u32(0xFFFFFFFF, n);
|
|
1235
|
-
a_vec = _mm512_maskz_loadu_pd(mask, a);
|
|
1236
|
-
b_vec = _mm512_maskz_loadu_pd(mask, b);
|
|
1237
|
-
n = 0;
|
|
1238
|
-
} else {
|
|
1239
|
-
a_vec = _mm512_loadu_pd(a);
|
|
1240
|
-
b_vec = _mm512_loadu_pd(b);
|
|
1241
|
-
a += 8, b += 8, n -= 8;
|
|
1242
|
-
}
|
|
1243
|
-
ab_vec = _mm512_fmadd_pd(a_vec, b_vec, ab_vec);
|
|
1244
|
-
a2_vec = _mm512_fmadd_pd(a_vec, a_vec, a2_vec);
|
|
1245
|
-
b2_vec = _mm512_fmadd_pd(b_vec, b_vec, b2_vec);
|
|
1246
|
-
if (n)
|
|
1247
|
-
goto simsimd_avx512_f64_cos_cycle;
|
|
1248
|
-
|
|
1249
|
-
simsimd_f32_t ab = (simsimd_f32_t)_mm512_reduce_add_pd(ab_vec);
|
|
1250
|
-
simsimd_f32_t a2 = (simsimd_f32_t)_mm512_reduce_add_pd(a2_vec);
|
|
1251
|
-
simsimd_f32_t b2 = (simsimd_f32_t)_mm512_reduce_add_pd(b2_vec);
|
|
1252
|
-
|
|
1253
|
-
// Compute the reciprocal square roots of a2 and b2
|
|
1254
|
-
__m128 rsqrts = _mm_rsqrt14_ps(_mm_set_ps(0.f, 0.f, a2 + 1.e-9f, b2 + 1.e-9f));
|
|
1255
|
-
simsimd_f32_t rsqrt_a2 = _mm_cvtss_f32(rsqrts);
|
|
1256
|
-
simsimd_f32_t rsqrt_b2 = _mm_cvtss_f32(_mm_shuffle_ps(rsqrts, rsqrts, _MM_SHUFFLE(0, 0, 0, 1)));
|
|
1257
|
-
return 1 - ab * rsqrt_a2 * rsqrt_b2;
|
|
1258
|
-
}
|
|
1259
|
-
|
|
1260
|
-
#endif // SIMSIMD_TARGET_X86_AVX512
|
|
1261
|
-
#endif // SIMSIMD_TARGET_X86
|
|
1262
|
-
|
|
1263
|
-
#ifdef __cplusplus
|
|
1264
|
-
}
|
|
1265
|
-
#endif
|
|
1266
|
-
|
|
1267
|
-
#endif
|