usearch 2.9.2 → 2.10.1

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