usearch 2.7.2 → 2.7.6

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.
Files changed (56) hide show
  1. package/.github/workflows/Dockerfile.libs +4 -6
  2. package/.github/workflows/prerelease.yml +4 -1
  3. package/.github/workflows/release.yml +8 -28
  4. package/.github/workflows/update_version.sh +1 -0
  5. package/CITATION.cff +2 -2
  6. package/CMakeLists.txt +1 -1
  7. package/Cargo.toml +1 -1
  8. package/README.md +4 -4
  9. package/binding.gyp +1 -1
  10. package/build.gradle +20 -5
  11. package/conanfile.py +1 -1
  12. package/csharp/nuget/nuget-package.props +1 -1
  13. package/docs/compilation.md +0 -2
  14. package/golang/README.md +0 -1
  15. package/include/usearch/index.hpp +1 -1
  16. package/include/usearch/index_plugins.hpp +119 -243
  17. package/javascript/lib.cpp +4 -3
  18. package/package.json +1 -1
  19. package/simsimd/.github/workflows/prerelease.yml +33 -58
  20. package/simsimd/.github/workflows/release.yml +22 -9
  21. package/simsimd/.github/workflows/update_version.sh +3 -1
  22. package/simsimd/.releaserc +9 -1
  23. package/simsimd/.vscode/launch.json +3 -0
  24. package/simsimd/.vscode/settings.json +15 -2
  25. package/simsimd/.vscode/tasks.json +2 -2
  26. package/simsimd/CMakeLists.txt +36 -55
  27. package/simsimd/README.md +163 -109
  28. package/simsimd/binding.gyp +18 -0
  29. package/simsimd/cpp/bench.cxx +187 -0
  30. package/simsimd/go.mod +3 -0
  31. package/simsimd/golang/bench_test.go +40 -0
  32. package/simsimd/golang/simsimd.go +47 -0
  33. package/simsimd/golang/unit_test.go +40 -0
  34. package/simsimd/include/simsimd/binary.h +186 -0
  35. package/simsimd/include/simsimd/probability.h +645 -0
  36. package/simsimd/include/simsimd/simsimd.h +335 -575
  37. package/simsimd/include/simsimd/spatial.h +924 -0
  38. package/simsimd/include/simsimd/types.h +196 -0
  39. package/simsimd/javascript/bench.js +117 -0
  40. package/simsimd/javascript/lib.c +102 -0
  41. package/simsimd/javascript/simsimd.js +45 -0
  42. package/simsimd/javascript/test.js +29 -0
  43. package/simsimd/package.json +31 -2
  44. package/simsimd/pyproject.toml +5 -6
  45. package/simsimd/python/bench.ipynb +187 -0
  46. package/simsimd/python/bench.py +268 -0
  47. package/simsimd/python/lib.c +453 -72
  48. package/simsimd/python/test.py +167 -35
  49. package/simsimd/setup.py +51 -8
  50. package/wasmer.toml +1 -1
  51. package/simsimd/bench.cxx +0 -137
  52. package/simsimd/include/simsimd/simsimd_bio.h +0 -54
  53. package/simsimd/include/simsimd/simsimd_chem.h +0 -167
  54. package/simsimd/include/simsimd/simsimd_text.h +0 -92
  55. package/simsimd/python/run.sh +0 -1
  56. package/simsimd/python/test.c +0 -109
@@ -0,0 +1,924 @@
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
+
24
+ #pragma once
25
+ #include "types.h"
26
+
27
+ #define SIMSIMD_MAKE_L2SQ(name, input_type, accumulator_type, converter) \
28
+ inline static simsimd_f32_t simsimd_##name##_##input_type##_l2sq( \
29
+ simsimd_##input_type##_t const* a, simsimd_##input_type##_t const* b, simsimd_size_t n) { \
30
+ simsimd_##accumulator_type##_t d2 = 0; \
31
+ for (simsimd_size_t i = 0; i != n; ++i) { \
32
+ simsimd_##accumulator_type##_t ai = converter(a[i]); \
33
+ simsimd_##accumulator_type##_t bi = converter(b[i]); \
34
+ d2 += (ai - bi) * (ai - bi); \
35
+ } \
36
+ return d2; \
37
+ }
38
+
39
+ #define SIMSIMD_MAKE_IP(name, input_type, accumulator_type, converter) \
40
+ inline static simsimd_f32_t simsimd_##name##_##input_type##_ip( \
41
+ simsimd_##input_type##_t const* a, simsimd_##input_type##_t const* b, simsimd_size_t n) { \
42
+ simsimd_##accumulator_type##_t ab = 0; \
43
+ for (simsimd_size_t i = 0; i != n; ++i) { \
44
+ simsimd_##accumulator_type##_t ai = converter(a[i]); \
45
+ simsimd_##accumulator_type##_t bi = converter(b[i]); \
46
+ ab += ai * bi; \
47
+ } \
48
+ return 1 - ab; \
49
+ }
50
+
51
+ #define SIMSIMD_MAKE_COS(name, input_type, accumulator_type, converter) \
52
+ inline static simsimd_f32_t simsimd_##name##_##input_type##_cos( \
53
+ simsimd_##input_type##_t const* a, simsimd_##input_type##_t const* b, simsimd_size_t n) { \
54
+ simsimd_##accumulator_type##_t ab = 0, a2 = 0, b2 = 0; \
55
+ for (simsimd_size_t i = 0; i != n; ++i) { \
56
+ simsimd_##accumulator_type##_t ai = converter(a[i]); \
57
+ simsimd_##accumulator_type##_t bi = converter(b[i]); \
58
+ ab += ai * bi; \
59
+ a2 += ai * ai; \
60
+ b2 += bi * bi; \
61
+ } \
62
+ return ab != 0 ? 1 - ab * SIMSIMD_RSQRT(a2) * SIMSIMD_RSQRT(b2) : 1; \
63
+ }
64
+
65
+ #ifdef __cplusplus
66
+ extern "C" {
67
+ #endif
68
+
69
+ SIMSIMD_MAKE_L2SQ(serial, f32, f32, SIMSIMD_IDENTIFY) // simsimd_serial_f32_l2sq
70
+ SIMSIMD_MAKE_IP(serial, f32, f32, SIMSIMD_IDENTIFY) // simsimd_serial_f32_ip
71
+ SIMSIMD_MAKE_COS(serial, f32, f32, SIMSIMD_IDENTIFY) // simsimd_serial_f32_cos
72
+
73
+ SIMSIMD_MAKE_L2SQ(serial, f16, f32, SIMSIMD_UNCOMPRESS_F16) // simsimd_serial_f16_l2sq
74
+ SIMSIMD_MAKE_IP(serial, f16, f32, SIMSIMD_UNCOMPRESS_F16) // simsimd_serial_f16_ip
75
+ SIMSIMD_MAKE_COS(serial, f16, f32, SIMSIMD_UNCOMPRESS_F16) // simsimd_serial_f16_cos
76
+
77
+ SIMSIMD_MAKE_L2SQ(serial, i8, i32, SIMSIMD_IDENTIFY) // simsimd_serial_i8_l2sq
78
+ SIMSIMD_MAKE_COS(serial, i8, i32, SIMSIMD_IDENTIFY) // simsimd_serial_i8_cos
79
+
80
+ inline static simsimd_f32_t simsimd_serial_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
81
+ return simsimd_serial_i8_cos(a, b, n);
82
+ }
83
+
84
+ SIMSIMD_MAKE_L2SQ(accurate, f32, f64, SIMSIMD_IDENTIFY) // simsimd_accurate_f32_l2sq
85
+ SIMSIMD_MAKE_IP(accurate, f32, f64, SIMSIMD_IDENTIFY) // simsimd_accurate_f32_ip
86
+ SIMSIMD_MAKE_COS(accurate, f32, f64, SIMSIMD_IDENTIFY) // simsimd_accurate_f32_cos
87
+
88
+ SIMSIMD_MAKE_L2SQ(accurate, f16, f64, SIMSIMD_UNCOMPRESS_F16) // simsimd_accurate_f16_l2sq
89
+ SIMSIMD_MAKE_IP(accurate, f16, f64, SIMSIMD_UNCOMPRESS_F16) // simsimd_accurate_f16_ip
90
+ SIMSIMD_MAKE_COS(accurate, f16, f64, SIMSIMD_UNCOMPRESS_F16) // simsimd_accurate_f16_cos
91
+
92
+ SIMSIMD_MAKE_L2SQ(accurate, i8, i32, SIMSIMD_IDENTIFY) // simsimd_accurate_i8_l2sq
93
+ SIMSIMD_MAKE_COS(accurate, i8, i32, SIMSIMD_IDENTIFY) // simsimd_accurate_i8_cos
94
+
95
+ inline static simsimd_f32_t simsimd_accurate_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
96
+ return simsimd_accurate_i8_cos(a, b, n);
97
+ }
98
+
99
+ #if SIMSIMD_TARGET_ARM
100
+ #if SIMSIMD_TARGET_ARM_NEON
101
+
102
+ /*
103
+ * @file arm_neon_f32.h
104
+ * @brief Arm NEON implementation of the most common similarity metrics for 32-bit floating point numbers.
105
+ * @author Ash Vardanian
106
+ *
107
+ * - Implements: L2 squared, inner product, cosine similarity.
108
+ * - Uses `f32` for storage and `f32` for accumulation.
109
+ * - Requires compiler capabilities: +simd.
110
+ */
111
+
112
+ __attribute__((target("+simd"))) //
113
+ inline static simsimd_f32_t
114
+ simsimd_neon_f32_l2sq(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
115
+ float32x4_t sum_vec = vdupq_n_f32(0);
116
+ simsimd_size_t i = 0;
117
+ for (; i + 4 <= n; i += 4) {
118
+ float32x4_t a_vec = vld1q_f32(a + i);
119
+ float32x4_t b_vec = vld1q_f32(b + i);
120
+ float32x4_t diff_vec = vsubq_f32(a_vec, b_vec);
121
+ sum_vec = vfmaq_f32(sum_vec, diff_vec, diff_vec);
122
+ }
123
+ simsimd_f32_t sum = vaddvq_f32(sum_vec);
124
+ for (; i < n; ++i) {
125
+ simsimd_f32_t diff = a[i] - b[i];
126
+ sum += diff * diff;
127
+ }
128
+ return sum;
129
+ }
130
+
131
+ __attribute__((target("+simd"))) //
132
+ inline static simsimd_f32_t
133
+ simsimd_neon_f32_ip(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
134
+ float32x4_t ab_vec = vdupq_n_f32(0);
135
+ simsimd_size_t i = 0;
136
+ for (; i + 4 <= n; i += 4) {
137
+ float32x4_t a_vec = vld1q_f32(a + i);
138
+ float32x4_t b_vec = vld1q_f32(b + i);
139
+ ab_vec = vfmaq_f32(ab_vec, a_vec, b_vec);
140
+ }
141
+ simsimd_f32_t ab = vaddvq_f32(ab_vec);
142
+ for (; i < n; ++i)
143
+ ab += a[i] * b[i];
144
+ return 1 - ab;
145
+ }
146
+
147
+ __attribute__((target("+simd"))) //
148
+ inline static simsimd_f32_t
149
+ simsimd_neon_f32_cos(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
150
+ float32x4_t ab_vec = vdupq_n_f32(0), a2_vec = vdupq_n_f32(0), b2_vec = vdupq_n_f32(0);
151
+ simsimd_size_t i = 0;
152
+ for (; i + 4 <= n; i += 4) {
153
+ float32x4_t a_vec = vld1q_f32(a + i);
154
+ float32x4_t b_vec = vld1q_f32(b + i);
155
+ ab_vec = vfmaq_f32(ab_vec, a_vec, b_vec);
156
+ a2_vec = vfmaq_f32(a2_vec, a_vec, a_vec);
157
+ b2_vec = vfmaq_f32(b2_vec, b_vec, b_vec);
158
+ }
159
+ simsimd_f32_t ab = vaddvq_f32(ab_vec), a2 = vaddvq_f32(a2_vec), b2 = vaddvq_f32(b2_vec);
160
+ for (; i < n; ++i) {
161
+ simsimd_f32_t ai = a[i], bi = b[i];
162
+ ab += ai * bi, a2 += ai * ai, b2 += bi * bi;
163
+ }
164
+
165
+ // Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
166
+ simsimd_f32_t a2_b2_arr[2] = {a2, b2};
167
+ vst1_f32(a2_b2_arr, vrsqrte_f32(vld1_f32(a2_b2_arr)));
168
+ return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
169
+ }
170
+
171
+ /*
172
+ * @file arm_neon_f16.h
173
+ * @brief Arm NEON implementation of the most common similarity metrics for 16-bit floating point numbers.
174
+ * @author Ash Vardanian
175
+ *
176
+ * - Implements: L2 squared, inner product, cosine similarity.
177
+ * - Uses `f16` for storage and `f32` for accumulation, as the 16-bit FMA may not always be available.
178
+ * - Requires compiler capabilities: +simd+fp16.
179
+ */
180
+
181
+ __attribute__((target("+simd+fp16"))) //
182
+ inline static simsimd_f32_t
183
+ simsimd_neon_f16_l2sq(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
184
+ float32x4_t sum_vec = vdupq_n_f32(0);
185
+ simsimd_size_t i = 0;
186
+ for (; i + 4 <= n; i += 4) {
187
+ float32x4_t a_vec = vcvt_f32_f16(vld1_f16((float16_t const*)a + i));
188
+ float32x4_t b_vec = vcvt_f32_f16(vld1_f16((float16_t const*)b + i));
189
+ float32x4_t diff_vec = vsubq_f32(a_vec, b_vec);
190
+ sum_vec = vfmaq_f32(sum_vec, diff_vec, diff_vec);
191
+ }
192
+ simsimd_f32_t sum = vaddvq_f32(sum_vec);
193
+ for (; i < n; ++i) {
194
+ simsimd_f32_t diff = SIMSIMD_UNCOMPRESS_F16(a[i]) - SIMSIMD_UNCOMPRESS_F16(b[i]);
195
+ sum += diff * diff;
196
+ }
197
+ return sum;
198
+ }
199
+
200
+ __attribute__((target("+simd+fp16"))) //
201
+ inline static simsimd_f32_t
202
+ simsimd_neon_f16_ip(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
203
+ float32x4_t ab_vec = vdupq_n_f32(0);
204
+ simsimd_size_t i = 0;
205
+ for (; i + 4 <= n; i += 4) {
206
+ float32x4_t a_vec = vcvt_f32_f16(vld1_f16((float16_t const*)a + i));
207
+ float32x4_t b_vec = vcvt_f32_f16(vld1_f16((float16_t const*)b + i));
208
+ ab_vec = vfmaq_f32(ab_vec, a_vec, b_vec);
209
+ }
210
+ simsimd_f32_t ab = vaddvq_f32(ab_vec);
211
+ for (; i < n; ++i)
212
+ ab += SIMSIMD_UNCOMPRESS_F16(a[i]) * SIMSIMD_UNCOMPRESS_F16(b[i]);
213
+ return 1 - ab;
214
+ }
215
+
216
+ __attribute__((target("+simd+fp16"))) //
217
+ inline static simsimd_f32_t
218
+ simsimd_neon_f16_cos(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
219
+ float32x4_t ab_vec = vdupq_n_f32(0), a2_vec = vdupq_n_f32(0), b2_vec = vdupq_n_f32(0);
220
+ simsimd_size_t i = 0;
221
+ for (; i + 4 <= n; i += 4) {
222
+ float32x4_t a_vec = vcvt_f32_f16(vld1_f16((float16_t const*)a + i));
223
+ float32x4_t b_vec = vcvt_f32_f16(vld1_f16((float16_t const*)b + i));
224
+ ab_vec = vfmaq_f32(ab_vec, a_vec, b_vec);
225
+ a2_vec = vfmaq_f32(a2_vec, a_vec, a_vec);
226
+ b2_vec = vfmaq_f32(b2_vec, b_vec, b_vec);
227
+ }
228
+ simsimd_f32_t ab = vaddvq_f32(ab_vec), a2 = vaddvq_f32(a2_vec), b2 = vaddvq_f32(b2_vec);
229
+ for (; i < n; ++i) {
230
+ simsimd_f32_t ai = SIMSIMD_UNCOMPRESS_F16(a[i]), bi = SIMSIMD_UNCOMPRESS_F16(b[i]);
231
+ ab += ai * bi, a2 += ai * ai, b2 += bi * bi;
232
+ }
233
+
234
+ // Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
235
+ simsimd_f32_t a2_b2_arr[2] = {a2, b2};
236
+ float32x2_t a2_b2 = vld1_f32(a2_b2_arr);
237
+ a2_b2 = vrsqrte_f32(a2_b2);
238
+ vst1_f32(a2_b2_arr, a2_b2);
239
+ return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
240
+ }
241
+
242
+ /*
243
+ * @file arm_neon_i8.h
244
+ * @brief Arm NEON implementation of the most common similarity metrics for 8-bit signed integral numbers.
245
+ * @author Ash Vardanian
246
+ *
247
+ * - Implements: L2 squared, cosine similarity, inner product (same as cosine).
248
+ * - Uses `i8` for storage, `i16` for multiplication, and `i32` for accumulation, if no better option is available.
249
+ * - Requires compiler capabilities: +simd+dotprod.
250
+ */
251
+
252
+ __attribute__((target("+simd"))) //
253
+ inline static simsimd_f32_t
254
+ simsimd_neon_i8_l2sq(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
255
+ int32x4_t d2_vec = vdupq_n_s32(0);
256
+ simsimd_size_t i = 0;
257
+ for (; i + 7 < n; i += 8) {
258
+ int8x8_t a_vec = vld1_s8(a + i);
259
+ int8x8_t b_vec = vld1_s8(b + i);
260
+ int16x8_t a_vec16 = vmovl_s8(a_vec);
261
+ int16x8_t b_vec16 = vmovl_s8(b_vec);
262
+ int16x8_t d_vec = vsubq_s16(a_vec16, b_vec16);
263
+ int32x4_t d_low = vmull_s16(vget_low_s16(d_vec), vget_low_s16(d_vec));
264
+ int32x4_t d_high = vmull_s16(vget_high_s16(d_vec), vget_high_s16(d_vec));
265
+ d2_vec = vaddq_s32(d2_vec, vaddq_s32(d_low, d_high));
266
+ }
267
+ int32_t d2 = vaddvq_s32(d2_vec);
268
+ for (; i < n; ++i) {
269
+ int32_t n = a[i] - b[i];
270
+ d2 += n * n;
271
+ }
272
+ return d2;
273
+ }
274
+
275
+ __attribute__((target("arch=armv8.2-a+dotprod"))) //
276
+ inline static simsimd_f32_t
277
+ simsimd_neon_i8_cos(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
278
+
279
+ int32x4_t ab_vec = vdupq_n_s32(0);
280
+ int32x4_t a2_vec = vdupq_n_s32(0);
281
+ int32x4_t b2_vec = vdupq_n_s32(0);
282
+ simsimd_size_t i = 0;
283
+
284
+ // If the 128-bit `vdot_s32` intrinsic is unavailable, we can use the 64-bit `vdot_s32`.
285
+ // for (simsimd_size_t i = 0; i != n; i += 8) {
286
+ // int16x8_t a_vec = vmovl_s8(vld1_s8(a + i));
287
+ // int16x8_t b_vec = vmovl_s8(vld1_s8(b + i));
288
+ // int16x8_t ab_part_vec = vmulq_s16(a_vec, b_vec);
289
+ // int16x8_t a2_part_vec = vmulq_s16(a_vec, a_vec);
290
+ // int16x8_t b2_part_vec = vmulq_s16(b_vec, b_vec);
291
+ // ab_vec = vaddq_s32(ab_vec, vaddq_s32(vmovl_s16(vget_high_s16(ab_part_vec)), //
292
+ // vmovl_s16(vget_low_s16(ab_part_vec))));
293
+ // a2_vec = vaddq_s32(a2_vec, vaddq_s32(vmovl_s16(vget_high_s16(a2_part_vec)), //
294
+ // vmovl_s16(vget_low_s16(a2_part_vec))));
295
+ // b2_vec = vaddq_s32(b2_vec, vaddq_s32(vmovl_s16(vget_high_s16(b2_part_vec)), //
296
+ // vmovl_s16(vget_low_s16(b2_part_vec))));
297
+ // }
298
+ for (; i + 15 < n; i += 16) {
299
+ int8x16_t a_vec = vld1q_s8(a + i);
300
+ int8x16_t b_vec = vld1q_s8(b + i);
301
+ ab_vec = vdotq_s32(ab_vec, a_vec, b_vec);
302
+ a2_vec = vdotq_s32(a2_vec, a_vec, a_vec);
303
+ b2_vec = vdotq_s32(b2_vec, b_vec, b_vec);
304
+ }
305
+
306
+ int32_t ab = vaddvq_s32(ab_vec);
307
+ int32_t a2 = vaddvq_s32(a2_vec);
308
+ int32_t b2 = vaddvq_s32(b2_vec);
309
+
310
+ // Take care of the tail:
311
+ for (; i < n; ++i) {
312
+ int32_t ai = a[i], bi = b[i];
313
+ ab += ai * bi, a2 += ai * ai, b2 += bi * bi;
314
+ }
315
+
316
+ // Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
317
+ simsimd_f32_t a2_b2_arr[2] = {(simsimd_f32_t)a2, (simsimd_f32_t)b2};
318
+ float32x2_t a2_b2 = vld1_f32(a2_b2_arr);
319
+ a2_b2 = vrsqrte_f32(a2_b2);
320
+ vst1_f32(a2_b2_arr, a2_b2);
321
+ return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
322
+ }
323
+
324
+ __attribute__((target("arch=armv8.2-a+dotprod"))) //
325
+ inline static simsimd_f32_t
326
+ simsimd_neon_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
327
+ return simsimd_neon_i8_cos(a, b, n);
328
+ }
329
+
330
+ #endif // SIMSIMD_TARGET_ARM_NEON
331
+
332
+ #if SIMSIMD_TARGET_ARM_SVE
333
+
334
+ /*
335
+ * @file arm_sve_f32.h
336
+ * @brief Arm SVE implementation of the most common similarity metrics for 32-bit floating point numbers.
337
+ * @author Ash Vardanian
338
+ *
339
+ * - Implements: L2 squared, inner product, cosine similarity.
340
+ * - Uses `f16` for both storage and `f32` for accumulation.
341
+ * - Requires compiler capabilities: +sve.
342
+ */
343
+
344
+ __attribute__((target("+sve"))) //
345
+ inline static simsimd_f32_t
346
+ simsimd_sve_f32_l2sq(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
347
+ simsimd_size_t i = 0;
348
+ svfloat32_t d2_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
349
+ do {
350
+ svbool_t pg_vec = svwhilelt_b32((unsigned int)i, (unsigned int)n);
351
+ svfloat32_t a_vec = svld1_f32(pg_vec, a + i);
352
+ svfloat32_t b_vec = svld1_f32(pg_vec, b + i);
353
+ svfloat32_t a_minus_b_vec = svsub_f32_x(pg_vec, a_vec, b_vec);
354
+ d2_vec = svmla_f32_x(pg_vec, d2_vec, a_minus_b_vec, a_minus_b_vec);
355
+ i += svcntw();
356
+ } while (i < n);
357
+ simsimd_f32_t d2 = svaddv_f32(svptrue_b32(), d2_vec);
358
+ return d2;
359
+ }
360
+ __attribute__((target("+sve"))) //
361
+ inline static simsimd_f32_t
362
+ simsimd_sve_f32_ip(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
363
+ simsimd_size_t i = 0;
364
+ svfloat32_t ab_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
365
+ do {
366
+ svbool_t pg_vec = svwhilelt_b32((unsigned int)i, (unsigned int)n);
367
+ svfloat32_t a_vec = svld1_f32(pg_vec, a + i);
368
+ svfloat32_t b_vec = svld1_f32(pg_vec, b + i);
369
+ ab_vec = svmla_f32_x(pg_vec, ab_vec, a_vec, b_vec);
370
+ i += svcntw();
371
+ } while (i < n);
372
+ return 1 - svaddv_f32(svptrue_b32(), ab_vec);
373
+ }
374
+
375
+ __attribute__((target("+sve"))) //
376
+ inline static simsimd_f32_t
377
+ simsimd_sve_f32_cos(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
378
+ simsimd_size_t i = 0;
379
+ svfloat32_t ab_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
380
+ svfloat32_t a2_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
381
+ svfloat32_t b2_vec = svdupq_n_f32(0.f, 0.f, 0.f, 0.f);
382
+ do {
383
+ svbool_t pg_vec = svwhilelt_b32((unsigned int)i, (unsigned int)n);
384
+ svfloat32_t a_vec = svld1_f32(pg_vec, a + i);
385
+ svfloat32_t b_vec = svld1_f32(pg_vec, b + i);
386
+ ab_vec = svmla_f32_x(pg_vec, ab_vec, a_vec, b_vec);
387
+ a2_vec = svmla_f32_x(pg_vec, a2_vec, a_vec, a_vec);
388
+ b2_vec = svmla_f32_x(pg_vec, b2_vec, b_vec, b_vec);
389
+ i += svcntw();
390
+ } while (i < n);
391
+
392
+ simsimd_f32_t ab = svaddv_f32(svptrue_b32(), ab_vec);
393
+ simsimd_f32_t a2 = svaddv_f32(svptrue_b32(), a2_vec);
394
+ simsimd_f32_t b2 = svaddv_f32(svptrue_b32(), b2_vec);
395
+
396
+ // Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
397
+ simsimd_f32_t a2_b2_arr[2] = {a2, b2};
398
+ float32x2_t a2_b2 = vld1_f32(a2_b2_arr);
399
+ a2_b2 = vrsqrte_f32(a2_b2);
400
+ vst1_f32(a2_b2_arr, a2_b2);
401
+ return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
402
+ }
403
+
404
+ /*
405
+ * @file arm_sve_f16.h
406
+ * @brief Arm SVE implementation of the most common similarity metrics for 16-bit floating point numbers.
407
+ * @author Ash Vardanian
408
+ *
409
+ * - Implements: L2 squared, inner product, cosine similarity.
410
+ * - Uses `f16` for both storage and `f32` for accumulation.
411
+ * - Requires compiler capabilities: +sve+fp16.
412
+ */
413
+
414
+ __attribute__((target("+sve+fp16"))) //
415
+ inline static simsimd_f32_t
416
+ simsimd_sve_f16_l2sq(simsimd_f16_t const* a_enum, simsimd_f16_t const* b_enum, simsimd_size_t n) {
417
+ simsimd_size_t i = 0;
418
+ svfloat16_t d2_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
419
+ simsimd_f16_t const* a = (simsimd_f16_t const*)(a_enum);
420
+ simsimd_f16_t const* b = (simsimd_f16_t const*)(b_enum);
421
+ do {
422
+ svbool_t pg_vec = svwhilelt_b16((unsigned int)i, (unsigned int)n);
423
+ svfloat16_t a_vec = svld1_f16(pg_vec, (float16_t const*)a + i);
424
+ svfloat16_t b_vec = svld1_f16(pg_vec, (float16_t const*)b + i);
425
+ svfloat16_t a_minus_b_vec = svsub_f16_x(pg_vec, a_vec, b_vec);
426
+ d2_vec = svmla_f16_x(pg_vec, d2_vec, a_minus_b_vec, a_minus_b_vec);
427
+ i += svcnth();
428
+ } while (i < n);
429
+ float16_t d2_f16 = svaddv_f16(svptrue_b16(), d2_vec);
430
+ return d2_f16;
431
+ }
432
+ __attribute__((target("+sve+fp16"))) //
433
+ inline static simsimd_f32_t
434
+ simsimd_sve_f16_ip(simsimd_f16_t const* a_enum, simsimd_f16_t const* b_enum, simsimd_size_t n) {
435
+ simsimd_size_t i = 0;
436
+ svfloat16_t ab_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
437
+ simsimd_f16_t const* a = (simsimd_f16_t const*)(a_enum);
438
+ simsimd_f16_t const* b = (simsimd_f16_t const*)(b_enum);
439
+ do {
440
+ svbool_t pg_vec = svwhilelt_b16((unsigned int)i, (unsigned int)n);
441
+ svfloat16_t a_vec = svld1_f16(pg_vec, (float16_t const*)a + i);
442
+ svfloat16_t b_vec = svld1_f16(pg_vec, (float16_t const*)b + i);
443
+ ab_vec = svmla_f16_x(pg_vec, ab_vec, a_vec, b_vec);
444
+ i += svcnth();
445
+ } while (i < n);
446
+ simsimd_f16_t ab = svaddv_f16(svptrue_b16(), ab_vec);
447
+ return 1 - ab;
448
+ }
449
+
450
+ __attribute__((target("+sve+fp16"))) //
451
+ inline static simsimd_f32_t
452
+ simsimd_sve_f16_cos(simsimd_f16_t const* a_enum, simsimd_f16_t const* b_enum, simsimd_size_t n) {
453
+ simsimd_size_t i = 0;
454
+ svfloat16_t ab_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
455
+ svfloat16_t a2_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
456
+ svfloat16_t b2_vec = svdupq_n_f16(0, 0, 0, 0, 0, 0, 0, 0);
457
+ simsimd_f16_t const* a = (simsimd_f16_t const*)(a_enum);
458
+ simsimd_f16_t const* b = (simsimd_f16_t const*)(b_enum);
459
+ do {
460
+ svbool_t pg_vec = svwhilelt_b16((unsigned int)i, (unsigned int)n);
461
+ svfloat16_t a_vec = svld1_f16(pg_vec, (float16_t const*)a + i);
462
+ svfloat16_t b_vec = svld1_f16(pg_vec, (float16_t const*)b + i);
463
+ ab_vec = svmla_f16_x(pg_vec, ab_vec, a_vec, b_vec);
464
+ a2_vec = svmla_f16_x(pg_vec, a2_vec, a_vec, a_vec);
465
+ b2_vec = svmla_f16_x(pg_vec, b2_vec, b_vec, b_vec);
466
+ i += svcnth();
467
+ } while (i < n);
468
+
469
+ simsimd_f16_t ab = svaddv_f16(svptrue_b16(), ab_vec);
470
+ simsimd_f16_t a2 = svaddv_f16(svptrue_b16(), a2_vec);
471
+ simsimd_f16_t b2 = svaddv_f16(svptrue_b16(), b2_vec);
472
+
473
+ // Avoid `simsimd_approximate_inverse_square_root` on Arm NEON
474
+ simsimd_f32_t a2_b2_arr[2] = {a2, b2};
475
+ float32x2_t a2_b2 = vld1_f32(a2_b2_arr);
476
+ a2_b2 = vrsqrte_f32(a2_b2);
477
+ vst1_f32(a2_b2_arr, a2_b2);
478
+ return ab != 0 ? 1 - ab * a2_b2_arr[0] * a2_b2_arr[1] : 1;
479
+ }
480
+
481
+ #endif // SIMSIMD_TARGET_ARM_SVE
482
+ #endif // SIMSIMD_TARGET_ARM
483
+
484
+ #if SIMSIMD_TARGET_X86
485
+ #if SIMSIMD_TARGET_X86_AVX2
486
+
487
+ /*
488
+ * @file x86_avx2_f16.h
489
+ * @brief x86 AVX2 implementation of the most common similarity metrics for 16-bit floating point numbers.
490
+ * @author Ash Vardanian
491
+ *
492
+ * - Implements: L2 squared, inner product, cosine similarity.
493
+ * - As AVX2 doesn't support masked loads of 16-bit words, implementations have a separate `for`-loop for tails.
494
+ * - Uses `f16` for both storage and `f32` for accumulation.
495
+ * - Requires compiler capabilities: avx2, f16c, fma.
496
+ */
497
+
498
+ __attribute__((target("avx2,f16c,fma"))) //
499
+ inline static simsimd_f32_t
500
+ simsimd_avx2_f16_l2sq(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
501
+ __m256 d2_vec = _mm256_set1_ps(0);
502
+ simsimd_size_t i = 0;
503
+ for (; i + 8 <= n; i += 8) {
504
+ __m256 a_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(a + i)));
505
+ __m256 b_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(b + i)));
506
+ __m256 d_vec = _mm256_sub_ps(a_vec, b_vec);
507
+ d2_vec = _mm256_fmadd_ps(d_vec, d_vec, d2_vec);
508
+ }
509
+
510
+ d2_vec = _mm256_add_ps(_mm256_permute2f128_ps(d2_vec, d2_vec, 1), d2_vec);
511
+ d2_vec = _mm256_hadd_ps(d2_vec, d2_vec);
512
+ d2_vec = _mm256_hadd_ps(d2_vec, d2_vec);
513
+
514
+ simsimd_f32_t result;
515
+ _mm_store_ss(&result, _mm256_castps256_ps128(d2_vec));
516
+
517
+ // Accumulate the tail:
518
+ for (; i < n; ++i) {
519
+ simsimd_f32_t n = SIMSIMD_UNCOMPRESS_F16(a[i]) - SIMSIMD_UNCOMPRESS_F16(b[i]);
520
+ result += n * n;
521
+ }
522
+ return result;
523
+ }
524
+
525
+ __attribute__((target("avx2,f16c,fma"))) //
526
+ inline static simsimd_f32_t
527
+ simsimd_avx2_f16_ip(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
528
+ __m256 ab_vec = _mm256_set1_ps(0);
529
+ simsimd_size_t i = 0;
530
+ for (; i + 8 <= n; i += 8) {
531
+ __m256 a_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(a + i)));
532
+ __m256 b_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(b + i)));
533
+ ab_vec = _mm256_fmadd_ps(a_vec, b_vec, ab_vec);
534
+ }
535
+
536
+ ab_vec = _mm256_add_ps(_mm256_permute2f128_ps(ab_vec, ab_vec, 1), ab_vec);
537
+ ab_vec = _mm256_hadd_ps(ab_vec, ab_vec);
538
+ ab_vec = _mm256_hadd_ps(ab_vec, ab_vec);
539
+
540
+ simsimd_f32_t result;
541
+ _mm_store_ss(&result, _mm256_castps256_ps128(ab_vec));
542
+
543
+ // Accumulate the tail:
544
+ for (; i < n; ++i)
545
+ result += SIMSIMD_UNCOMPRESS_F16(a[i]) * SIMSIMD_UNCOMPRESS_F16(b[i]);
546
+ return 1 - result;
547
+ }
548
+ __attribute__((target("avx2,f16c,fma"))) //
549
+ inline static simsimd_f32_t
550
+ simsimd_avx2_f16_cos(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
551
+
552
+ __m256 ab_vec = _mm256_set1_ps(0), a2_vec = _mm256_set1_ps(0), b2_vec = _mm256_set1_ps(0);
553
+ simsimd_size_t i = 0;
554
+ for (; i + 8 <= n; i += 8) {
555
+ __m256 a_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(a + i)));
556
+ __m256 b_vec = _mm256_cvtph_ps(_mm_loadu_si128((__m128i const*)(b + i)));
557
+ ab_vec = _mm256_fmadd_ps(a_vec, b_vec, ab_vec);
558
+ a2_vec = _mm256_fmadd_ps(a_vec, a_vec, a2_vec);
559
+ b2_vec = _mm256_fmadd_ps(b_vec, b_vec, b2_vec);
560
+ }
561
+
562
+ ab_vec = _mm256_add_ps(_mm256_permute2f128_ps(ab_vec, ab_vec, 1), ab_vec);
563
+ ab_vec = _mm256_hadd_ps(ab_vec, ab_vec);
564
+ ab_vec = _mm256_hadd_ps(ab_vec, ab_vec);
565
+
566
+ a2_vec = _mm256_add_ps(_mm256_permute2f128_ps(a2_vec, a2_vec, 1), a2_vec);
567
+ a2_vec = _mm256_hadd_ps(a2_vec, a2_vec);
568
+ a2_vec = _mm256_hadd_ps(a2_vec, a2_vec);
569
+
570
+ b2_vec = _mm256_add_ps(_mm256_permute2f128_ps(b2_vec, b2_vec, 1), b2_vec);
571
+ b2_vec = _mm256_hadd_ps(b2_vec, b2_vec);
572
+ b2_vec = _mm256_hadd_ps(b2_vec, b2_vec);
573
+
574
+ simsimd_f32_t ab, a2, b2;
575
+ _mm_store_ss(&ab, _mm256_castps256_ps128(ab_vec));
576
+ _mm_store_ss(&a2, _mm256_castps256_ps128(a2_vec));
577
+ _mm_store_ss(&b2, _mm256_castps256_ps128(b2_vec));
578
+
579
+ // Accumulate the tail:
580
+ for (; i < n; ++i) {
581
+ simsimd_f32_t ai = SIMSIMD_UNCOMPRESS_F16(a[i]), bi = SIMSIMD_UNCOMPRESS_F16(b[i]);
582
+ ab += ai * bi, a2 += ai * ai, b2 += bi * bi;
583
+ }
584
+
585
+ // Replace simsimd_approximate_inverse_square_root with `rsqrtss`
586
+ __m128 a2_sqrt_recip = _mm_rsqrt_ss(_mm_set_ss((float)a2));
587
+ __m128 b2_sqrt_recip = _mm_rsqrt_ss(_mm_set_ss((float)b2));
588
+ __m128 result = _mm_mul_ss(a2_sqrt_recip, b2_sqrt_recip); // Multiply the reciprocal square roots
589
+ result = _mm_mul_ss(result, _mm_set_ss((float)ab)); // Multiply by ab
590
+ result = _mm_sub_ss(_mm_set_ss(1.0f), result); // Subtract from 1
591
+ return ab != 0 ? _mm_cvtss_f32(result) : 1; // Extract the final result
592
+ }
593
+
594
+ /*
595
+ * @file x86_avx2_i8.h
596
+ * @brief x86 AVX2 implementation of the most common similarity metrics for 8-bit signed integral numbers.
597
+ * @author Ash Vardanian
598
+ *
599
+ * - Implements: L2 squared, cosine similarity, inner product (same as cosine).
600
+ * - As AVX2 doesn't support masked loads of 16-bit words, implementations have a separate `for`-loop for tails.
601
+ * - Uses `i8` for storage, `i16` for multiplication, and `i32` for accumulation, if no better option is available.
602
+ * - Requires compiler capabilities: avx2.
603
+ */
604
+
605
+ __attribute__((target("avx2"))) //
606
+ inline static simsimd_f32_t
607
+ simsimd_avx2_i8_l2sq(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
608
+
609
+ __m256i d2_high_vec = _mm256_setzero_si256();
610
+ __m256i d2_low_vec = _mm256_setzero_si256();
611
+
612
+ simsimd_size_t i = 0;
613
+ for (; i + 32 <= n; i += 32) {
614
+ __m256i a_vec = _mm256_loadu_si256((__m256i const*)(a + i));
615
+ __m256i b_vec = _mm256_loadu_si256((__m256i const*)(b + i));
616
+
617
+ // Sign extend int8 to int16
618
+ __m256i a_low = _mm256_cvtepi8_epi16(_mm256_castsi256_si128(a_vec));
619
+ __m256i a_high = _mm256_cvtepi8_epi16(_mm256_extracti128_si256(a_vec, 1));
620
+ __m256i b_low = _mm256_cvtepi8_epi16(_mm256_castsi256_si128(b_vec));
621
+ __m256i b_high = _mm256_cvtepi8_epi16(_mm256_extracti128_si256(b_vec, 1));
622
+
623
+ // Subtract and multiply
624
+ __m256i d_low = _mm256_sub_epi16(a_low, b_low);
625
+ __m256i d_high = _mm256_sub_epi16(a_high, b_high);
626
+ __m256i d2_low_part = _mm256_madd_epi16(d_low, d_low);
627
+ __m256i d2_high_part = _mm256_madd_epi16(d_high, d_high);
628
+
629
+ // Accumulate into int32 vectors
630
+ d2_low_vec = _mm256_add_epi32(d2_low_vec, d2_low_part);
631
+ d2_high_vec = _mm256_add_epi32(d2_high_vec, d2_high_part);
632
+ }
633
+
634
+ // Accumulate the 32-bit integers from `d2_high_vec` and `d2_low_vec`
635
+ __m256i d2_vec = _mm256_add_epi32(d2_low_vec, d2_high_vec);
636
+ __m128i d2_sum = _mm_add_epi32(_mm256_extracti128_si256(d2_vec, 0), _mm256_extracti128_si256(d2_vec, 1));
637
+ d2_sum = _mm_hadd_epi32(d2_sum, d2_sum);
638
+ d2_sum = _mm_hadd_epi32(d2_sum, d2_sum);
639
+ int d2 = _mm_extract_epi32(d2_sum, 0);
640
+
641
+ // Take care of the tail:
642
+ for (; i < n; ++i) {
643
+ int n = a[i] - b[i];
644
+ d2 += n * n;
645
+ }
646
+
647
+ return (simsimd_f32_t)d2;
648
+ }
649
+ __attribute__((target("avx2"))) //
650
+ inline static simsimd_f32_t
651
+ simsimd_avx2_i8_cos(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
652
+
653
+ __m256i ab_high_vec = _mm256_setzero_si256();
654
+ __m256i ab_low_vec = _mm256_setzero_si256();
655
+ __m256i a2_high_vec = _mm256_setzero_si256();
656
+ __m256i a2_low_vec = _mm256_setzero_si256();
657
+ __m256i b2_high_vec = _mm256_setzero_si256();
658
+ __m256i b2_low_vec = _mm256_setzero_si256();
659
+
660
+ simsimd_size_t i = 0;
661
+ for (; i + 32 <= n; i += 32) {
662
+ __m256i a_vec = _mm256_loadu_si256((__m256i const*)(a + i));
663
+ __m256i b_vec = _mm256_loadu_si256((__m256i const*)(b + i));
664
+
665
+ // Unpack int8 to int32
666
+ __m256i a_low = _mm256_cvtepi8_epi32(_mm256_castsi256_si128(a_vec));
667
+ __m256i a_high = _mm256_cvtepi8_epi32(_mm256_extracti128_si256(a_vec, 1));
668
+ __m256i b_low = _mm256_cvtepi8_epi32(_mm256_castsi256_si128(b_vec));
669
+ __m256i b_high = _mm256_cvtepi8_epi32(_mm256_extracti128_si256(b_vec, 1));
670
+
671
+ // Multiply and accumulate
672
+ ab_low_vec = _mm256_add_epi32(ab_low_vec, _mm256_mullo_epi32(a_low, b_low));
673
+ ab_high_vec = _mm256_add_epi32(ab_high_vec, _mm256_mullo_epi32(a_high, b_high));
674
+ a2_low_vec = _mm256_add_epi32(a2_low_vec, _mm256_mullo_epi32(a_low, a_low));
675
+ a2_high_vec = _mm256_add_epi32(a2_high_vec, _mm256_mullo_epi32(a_high, a_high));
676
+ b2_low_vec = _mm256_add_epi32(b2_low_vec, _mm256_mullo_epi32(b_low, b_low));
677
+ b2_high_vec = _mm256_add_epi32(b2_high_vec, _mm256_mullo_epi32(b_high, b_high));
678
+ }
679
+
680
+ // Horizontal sum across the 256-bit register
681
+ __m256i ab_vec = _mm256_add_epi32(ab_low_vec, ab_high_vec);
682
+ __m128i ab_sum = _mm_add_epi32(_mm256_extracti128_si256(ab_vec, 0), _mm256_extracti128_si256(ab_vec, 1));
683
+ ab_sum = _mm_hadd_epi32(ab_sum, ab_sum);
684
+ ab_sum = _mm_hadd_epi32(ab_sum, ab_sum);
685
+
686
+ __m256i a2_vec = _mm256_add_epi32(a2_low_vec, a2_high_vec);
687
+ __m128i a2_sum = _mm_add_epi32(_mm256_extracti128_si256(a2_vec, 0), _mm256_extracti128_si256(a2_vec, 1));
688
+ a2_sum = _mm_hadd_epi32(a2_sum, a2_sum);
689
+ a2_sum = _mm_hadd_epi32(a2_sum, a2_sum);
690
+
691
+ __m256i b2_vec = _mm256_add_epi32(b2_low_vec, b2_high_vec);
692
+ __m128i b2_sum = _mm_add_epi32(_mm256_extracti128_si256(b2_vec, 0), _mm256_extracti128_si256(b2_vec, 1));
693
+ b2_sum = _mm_hadd_epi32(b2_sum, b2_sum);
694
+ b2_sum = _mm_hadd_epi32(b2_sum, b2_sum);
695
+
696
+ // Further reduce to a single sum for each vector
697
+ int ab = _mm_extract_epi32(ab_sum, 0);
698
+ int a2 = _mm_extract_epi32(a2_sum, 0);
699
+ int b2 = _mm_extract_epi32(b2_sum, 0);
700
+
701
+ // Take care of the tail:
702
+ for (; i < n; ++i) {
703
+ int ai = a[i], bi = b[i];
704
+ ab += ai * bi, a2 += ai * ai, b2 += bi * bi;
705
+ }
706
+
707
+ // Replace simsimd_approximate_inverse_square_root with `rsqrtss`
708
+ __m128 a2_sqrt_recip = _mm_rsqrt_ss(_mm_set_ss((float)a2));
709
+ __m128 b2_sqrt_recip = _mm_rsqrt_ss(_mm_set_ss((float)b2));
710
+ __m128 result = _mm_mul_ss(a2_sqrt_recip, b2_sqrt_recip); // Multiply the reciprocal square roots
711
+ result = _mm_mul_ss(result, _mm_set_ss((float)ab)); // Multiply by ab
712
+ result = _mm_sub_ss(_mm_set_ss(1.0f), result); // Subtract from 1
713
+ return ab != 0 ? _mm_cvtss_f32(result) : 1; // Extract the final result
714
+ }
715
+
716
+ __attribute__((target("avx2"))) //
717
+ inline static simsimd_f32_t
718
+ simsimd_avx2_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
719
+ return simsimd_avx2_i8_cos(a, b, n);
720
+ }
721
+
722
+ #endif // SIMSIMD_TARGET_X86_AVX2
723
+
724
+ #if SIMSIMD_TARGET_X86_AVX512
725
+
726
+ /*
727
+ * @file x86_avx512_f32.h
728
+ * @brief x86 AVX-512 implementation of the most common similarity metrics for 32-bit floating point numbers.
729
+ * @author Ash Vardanian
730
+ *
731
+ * - Implements: L2 squared, inner product, cosine similarity.
732
+ * - Uses `f32` for storage and `f32` for accumulation.
733
+ * - Requires compiler capabilities: avx512f, avx512vl.
734
+ */
735
+
736
+ __attribute__((target("avx512f,avx512vl"))) //
737
+ inline static simsimd_f32_t
738
+ simsimd_avx512_f32_l2sq(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
739
+ __m512 d2_vec = _mm512_set1_ps(0);
740
+ for (simsimd_size_t i = 0; i < n; i += 16) {
741
+ __mmask16 mask = n - i >= 16 ? 0xFFFF : ((1u << (n - i)) - 1u);
742
+ __m512 a_vec = _mm512_maskz_loadu_ps(mask, a + i);
743
+ __m512 b_vec = _mm512_maskz_loadu_ps(mask, b + i);
744
+ __m512 d_vec = _mm512_sub_ps(a_vec, b_vec);
745
+ d2_vec = _mm512_fmadd_ps(d_vec, d_vec, d2_vec);
746
+ }
747
+ return _mm512_reduce_add_ps(d2_vec);
748
+ }
749
+
750
+ __attribute__((target("avx512f,avx512vl"))) //
751
+ inline static simsimd_f32_t
752
+ simsimd_avx512_f32_ip(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
753
+ __m512 ab_vec = _mm512_set1_ps(0);
754
+ for (simsimd_size_t i = 0; i < n; i += 16) {
755
+ __mmask16 mask = n - i >= 16 ? 0xFFFF : ((1u << (n - i)) - 1u);
756
+ __m512 a_vec = _mm512_maskz_loadu_ps(mask, a + i);
757
+ __m512 b_vec = _mm512_maskz_loadu_ps(mask, b + i);
758
+ ab_vec = _mm512_fmadd_ps(a_vec, b_vec, ab_vec);
759
+ }
760
+ return 1 - _mm512_reduce_add_ps(ab_vec);
761
+ }
762
+
763
+ __attribute__((target("avx512f,avx512vl"))) //
764
+ inline static simsimd_f32_t
765
+ simsimd_avx512_f32_cos(simsimd_f32_t const* a, simsimd_f32_t const* b, simsimd_size_t n) {
766
+ __m512 ab_vec = _mm512_set1_ps(0);
767
+ __m512 a2_vec = _mm512_set1_ps(0);
768
+ __m512 b2_vec = _mm512_set1_ps(0);
769
+
770
+ for (simsimd_size_t i = 0; i < n; i += 16) {
771
+ __mmask16 mask = n - i >= 16 ? 0xFFFF : ((1u << (n - i)) - 1u);
772
+ __m512 a_vec = _mm512_maskz_loadu_ps(mask, a + i);
773
+ __m512 b_vec = _mm512_maskz_loadu_ps(mask, b + i);
774
+ ab_vec = _mm512_fmadd_ps(a_vec, b_vec, ab_vec);
775
+ a2_vec = _mm512_fmadd_ps(a_vec, a_vec, a2_vec);
776
+ b2_vec = _mm512_fmadd_ps(b_vec, b_vec, b2_vec);
777
+ }
778
+
779
+ simsimd_f32_t ab = _mm512_reduce_add_ps(ab_vec);
780
+ simsimd_f32_t a2 = _mm512_reduce_add_ps(a2_vec);
781
+ simsimd_f32_t b2 = _mm512_reduce_add_ps(b2_vec);
782
+
783
+ __m128d a2_b2 = _mm_set_pd((double)a2, (double)b2);
784
+ __m128d rsqrts = _mm_mask_rsqrt14_pd(_mm_setzero_pd(), 0xFF, a2_b2);
785
+ double rsqrts_array[2];
786
+ _mm_storeu_pd(rsqrts_array, rsqrts);
787
+ return ab != 0 ? 1 - ab * rsqrts_array[0] * rsqrts_array[1] : 1;
788
+ }
789
+
790
+ /*
791
+ * @file x86_avx512_f16.h
792
+ * @brief x86 AVX-512 implementation of the most common similarity metrics for 16-bit floating point numbers.
793
+ * @author Ash Vardanian
794
+ *
795
+ * - Implements: L2 squared, inner product, cosine similarity.
796
+ * - Uses `_mm512_maskz_loadu_epi16` intrinsics to perform masked unaligned loads.
797
+ * - Uses `f16` for both storage and accumulation, assuming it's resolution is enough for average case.
798
+ * - Requires compiler capabilities: avx512fp16, avx512f, avx512vl.
799
+ */
800
+
801
+ __attribute__((target("avx512fp16,avx512vl,avx512f"))) //
802
+ inline static simsimd_f32_t
803
+ simsimd_avx512_f16_l2sq(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
804
+ __m512h d2_vec = _mm512_set1_ph(0);
805
+ for (simsimd_size_t i = 0; i < n; i += 32) {
806
+ __mmask32 mask = n - i >= 32 ? 0xFFFFFFFF : ((1u << (n - i)) - 1u);
807
+ __m512i a_vec = _mm512_maskz_loadu_epi16(mask, a + i);
808
+ __m512i b_vec = _mm512_maskz_loadu_epi16(mask, b + i);
809
+ __m512h d_vec = _mm512_sub_ph(_mm512_castsi512_ph(a_vec), _mm512_castsi512_ph(b_vec));
810
+ d2_vec = _mm512_fmadd_ph(d_vec, d_vec, d2_vec);
811
+ }
812
+ return _mm512_reduce_add_ph(d2_vec);
813
+ }
814
+
815
+ __attribute__((target("avx512fp16,avx512vl,avx512f"))) //
816
+ inline static simsimd_f32_t
817
+ simsimd_avx512_f16_ip(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
818
+ __m512h ab_vec = _mm512_set1_ph(0);
819
+ simsimd_size_t i = 0;
820
+
821
+ for (simsimd_size_t i = 0; i < n; i += 32) {
822
+ __mmask32 mask = n - i >= 32 ? 0xFFFFFFFF : ((1u << (n - i)) - 1u);
823
+ __m512i a_vec = _mm512_maskz_loadu_epi16(mask, a + i);
824
+ __m512i b_vec = _mm512_maskz_loadu_epi16(mask, b + i);
825
+ ab_vec = _mm512_fmadd_ph(_mm512_castsi512_ph(a_vec), _mm512_castsi512_ph(b_vec), ab_vec);
826
+ }
827
+ return 1 - _mm512_reduce_add_ph(ab_vec);
828
+ }
829
+
830
+ __attribute__((target("avx512fp16,avx512vl,avx512f"))) //
831
+ inline static simsimd_f32_t
832
+ simsimd_avx512_f16_cos(simsimd_f16_t const* a, simsimd_f16_t const* b, simsimd_size_t n) {
833
+ __m512h ab_vec = _mm512_set1_ph(0);
834
+ __m512h a2_vec = _mm512_set1_ph(0);
835
+ __m512h b2_vec = _mm512_set1_ph(0);
836
+
837
+ for (simsimd_size_t i = 0; i < n; i += 32) {
838
+ __mmask32 mask = n - i >= 32 ? 0xFFFFFFFF : ((1u << (n - i)) - 1u);
839
+ __m512i a_vec = _mm512_maskz_loadu_epi16(mask, a + i);
840
+ __m512i b_vec = _mm512_maskz_loadu_epi16(mask, b + i);
841
+ ab_vec = _mm512_fmadd_ph(_mm512_castsi512_ph(a_vec), _mm512_castsi512_ph(b_vec), ab_vec);
842
+ a2_vec = _mm512_fmadd_ph(_mm512_castsi512_ph(a_vec), _mm512_castsi512_ph(a_vec), a2_vec);
843
+ b2_vec = _mm512_fmadd_ph(_mm512_castsi512_ph(b_vec), _mm512_castsi512_ph(b_vec), b2_vec);
844
+ }
845
+
846
+ simsimd_f32_t ab = _mm512_reduce_add_ph(ab_vec);
847
+ simsimd_f32_t a2 = _mm512_reduce_add_ph(a2_vec);
848
+ simsimd_f32_t b2 = _mm512_reduce_add_ph(b2_vec);
849
+
850
+ __m128d a2_b2 = _mm_set_pd((double)a2, (double)b2);
851
+ __m128d rsqrts = _mm_mask_rsqrt14_pd(_mm_setzero_pd(), 0xFF, a2_b2);
852
+ double rsqrts_array[2];
853
+ _mm_storeu_pd(rsqrts_array, rsqrts);
854
+ return ab != 0 ? 1 - ab * rsqrts_array[0] * rsqrts_array[1] : 1;
855
+ }
856
+
857
+ /*
858
+ * @file x86_avx512_i8.h
859
+ * @brief x86 AVX-512 implementation of the most common similarity metrics for 8-bit integers.
860
+ * @author Ash Vardanian
861
+ *
862
+ * - Implements: L2 squared, cosine similarity, inner product (same as cosine).
863
+ * - Uses `_mm512_maskz_loadu_epi16` intrinsics to perform masked unaligned loads.
864
+ * - Uses `i8` for storage, `i16` for multiplication, and `i32` for accumulation, if no better option is available.
865
+ * - Requires compiler capabilities: avx512f, avx512vl, avx512bw.
866
+ */
867
+
868
+ __attribute__((target("avx512vl,avx512f,avx512bw"))) //
869
+ inline static simsimd_f32_t
870
+ simsimd_avx512_i8_l2sq(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
871
+ __m512i d2_i32s_vec = _mm512_setzero_si512();
872
+
873
+ for (simsimd_size_t i = 0; i < n; i += 32) {
874
+ __mmask32 mask = n - i >= 32 ? 0xFFFFFFFF : ((1u << (n - i)) - 1u);
875
+ __m512i a_vec = _mm512_cvtepi8_epi16(_mm256_maskz_loadu_epi8(mask, a + i)); // Load 8-bit integers
876
+ __m512i b_vec = _mm512_cvtepi8_epi16(_mm256_maskz_loadu_epi8(mask, b + i)); // Load 8-bit integers
877
+ __m512i d_i16s_vec = _mm512_sub_epi16(a_vec, b_vec);
878
+ d2_i32s_vec = _mm512_add_epi32(d2_i32s_vec, _mm512_madd_epi16(d_i16s_vec, d_i16s_vec));
879
+ }
880
+
881
+ return _mm512_reduce_add_epi32(d2_i32s_vec);
882
+ }
883
+
884
+ __attribute__((target("avx512vl,avx512f,avx512bw"))) //
885
+ inline static simsimd_f32_t
886
+ simsimd_avx512_i8_cos(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
887
+ __m512i ab_i32s_vec = _mm512_setzero_si512();
888
+ __m512i a2_i32s_vec = _mm512_setzero_si512();
889
+ __m512i b2_i32s_vec = _mm512_setzero_si512();
890
+ simsimd_size_t i = 0;
891
+
892
+ for (simsimd_size_t i = 0; i < n; i += 32) {
893
+ __mmask32 mask = n - i >= 32 ? 0xFFFFFFFF : ((1u << (n - i)) - 1u);
894
+ __m512i a_vec = _mm512_cvtepi8_epi16(_mm256_maskz_loadu_epi8(mask, a + i)); // Load 8-bit integers
895
+ __m512i b_vec = _mm512_cvtepi8_epi16(_mm256_maskz_loadu_epi8(mask, b + i)); // Load 8-bit integers
896
+
897
+ ab_i32s_vec = _mm512_add_epi32(ab_i32s_vec, _mm512_madd_epi16(a_vec, b_vec));
898
+ a2_i32s_vec = _mm512_add_epi32(a2_i32s_vec, _mm512_madd_epi16(a_vec, a_vec));
899
+ b2_i32s_vec = _mm512_add_epi32(b2_i32s_vec, _mm512_madd_epi16(b_vec, b_vec));
900
+ }
901
+
902
+ int ab = _mm512_reduce_add_epi32(ab_i32s_vec);
903
+ int a2 = _mm512_reduce_add_epi32(a2_i32s_vec);
904
+ int b2 = _mm512_reduce_add_epi32(b2_i32s_vec);
905
+
906
+ __m128d a2_b2 = _mm_set_pd((double)a2, (double)b2);
907
+ __m128d rsqrts = _mm_mask_rsqrt14_pd(_mm_setzero_pd(), 0xFF, a2_b2);
908
+ double rsqrts_array[2];
909
+ _mm_storeu_pd(rsqrts_array, rsqrts);
910
+ return ab != 0 ? 1 - ab * rsqrts_array[0] * rsqrts_array[1] : 1;
911
+ }
912
+
913
+ __attribute__((target("avx512vl,avx512f,avx512bw"))) //
914
+ inline static simsimd_f32_t
915
+ simsimd_avx512_i8_ip(simsimd_i8_t const* a, simsimd_i8_t const* b, simsimd_size_t n) {
916
+ return simsimd_avx512_i8_cos(a, b, n);
917
+ }
918
+
919
+ #endif // SIMSIMD_TARGET_X86_AVX512
920
+ #endif // SIMSIMD_TARGET_X86
921
+
922
+ #ifdef __cplusplus
923
+ }
924
+ #endif