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