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