Line | Branch | Exec | Source |
---|---|---|---|
1 | /* -*- c++ -*- */ | ||
2 | /* | ||
3 | * Copyright 2023 Daniel Estevez <daniel@destevez.net> | ||
4 | * Copyright 2012, 2014 Free Software Foundation, Inc. | ||
5 | * | ||
6 | * This file is part of VOLK | ||
7 | * | ||
8 | * SPDX-License-Identifier: LGPL-3.0-or-later | ||
9 | */ | ||
10 | |||
11 | /*! | ||
12 | * \page volk_32f_s32f_x2_convert_8u | ||
13 | * | ||
14 | * \b Overview | ||
15 | * | ||
16 | * Converts a floating point number to an 8-bit unsigned int after applying a | ||
17 | * multiplicative scaling factor and an additive bias. | ||
18 | * | ||
19 | * <b>Dispatcher Prototype</b> | ||
20 | * \code | ||
21 | * void volk_32f_s32f_x2_convert_8u(uint8_t* outputVector, const float* inputVector, | ||
22 | const float scale, const float bias, unsigned int num_points) | ||
23 | * \endcode | ||
24 | * | ||
25 | * \b Inputs | ||
26 | * \li inputVector: the input vector of floats. | ||
27 | * \li scale: The value multiplied against each point in the input buffer. | ||
28 | * \li bias: The value added to each multiplication by the scale. | ||
29 | * \li num_points: The number of data points. | ||
30 | * | ||
31 | * \b Outputs | ||
32 | * \li outputVector: The output vector. | ||
33 | * | ||
34 | * \b Example | ||
35 | * Convert floats from [-1,1] to 8-bit unsigend integers with a scale of 128 and a bias of | ||
36 | 128 | ||
37 | * int N = 10; | ||
38 | * unsigned int alignment = volk_get_alignment(); | ||
39 | * float* increasing = (float*)volk_malloc(sizeof(float)*N, alignment); | ||
40 | * uint8_t* out = (uint8_t*)volk_malloc(sizeof(uint8_t)*N, alignment); | ||
41 | * | ||
42 | * for(unsigned int ii = 0; ii < N; ++ii){ | ||
43 | * increasing[ii] = 2.f * ((float)ii / (float)N) - 1.f; | ||
44 | * } | ||
45 | * | ||
46 | * float scale = 128.0f; | ||
47 | * float bias = 128.0f; | ||
48 | * | ||
49 | * volk_32f_s32f_x2_convert_8u(out, increasing, scale, bias, N); | ||
50 | * | ||
51 | * for(unsigned int ii = 0; ii < N; ++ii){ | ||
52 | * printf("out[%u] = %i\n", ii, out[ii]); | ||
53 | * } | ||
54 | * | ||
55 | * volk_free(increasing); | ||
56 | * volk_free(out); | ||
57 | * \endcode | ||
58 | */ | ||
59 | |||
60 | #ifndef INCLUDED_volk_32f_s32f_x2_convert_8u_u_H | ||
61 | #define INCLUDED_volk_32f_s32f_x2_convert_8u_u_H | ||
62 | |||
63 | #include <inttypes.h> | ||
64 | |||
65 | 262462 | static inline void volk_32f_s32f_x2_convert_8u_single(uint8_t* out, const float in) | |
66 | { | ||
67 | 262462 | const float min_val = 0.0f; | |
68 | 262462 | const float max_val = UINT8_MAX; | |
69 |
2/2✓ Branch 0 taken 80080 times.
✓ Branch 1 taken 182382 times.
|
262462 | if (in > max_val) { |
70 | 80080 | *out = (uint8_t)(max_val); | |
71 |
2/2✓ Branch 0 taken 79991 times.
✓ Branch 1 taken 102391 times.
|
182382 | } else if (in < min_val) { |
72 | 79991 | *out = (uint8_t)(min_val); | |
73 | } else { | ||
74 | 102391 | *out = (uint8_t)(rintf(in)); | |
75 | } | ||
76 | 262462 | } | |
77 | |||
78 | |||
79 | #ifdef LV_HAVE_GENERIC | ||
80 | |||
81 | 2 | static inline void volk_32f_s32f_x2_convert_8u_generic(uint8_t* outputVector, | |
82 | const float* inputVector, | ||
83 | const float scale, | ||
84 | const float bias, | ||
85 | unsigned int num_points) | ||
86 | { | ||
87 | 2 | const float* inputVectorPtr = inputVector; | |
88 | |||
89 |
2/2✓ Branch 0 taken 262142 times.
✓ Branch 1 taken 2 times.
|
262144 | for (unsigned int number = 0; number < num_points; number++) { |
90 | 262142 | const float r = *inputVectorPtr++ * scale + bias; | |
91 | 262142 | volk_32f_s32f_x2_convert_8u_single(&outputVector[number], r); | |
92 | } | ||
93 | 2 | } | |
94 | |||
95 | #endif /* LV_HAVE_GENERIC */ | ||
96 | |||
97 | |||
98 | #if LV_HAVE_AVX2 && LV_HAVE_FMA | ||
99 | #include <immintrin.h> | ||
100 | |||
101 | 2 | static inline void volk_32f_s32f_x2_convert_8u_u_avx2_fma(uint8_t* outputVector, | |
102 | const float* inputVector, | ||
103 | const float scale, | ||
104 | const float bias, | ||
105 | unsigned int num_points) | ||
106 | { | ||
107 | 2 | const unsigned int thirtysecondPoints = num_points / 32; | |
108 | |||
109 | 2 | const float* inputVectorPtr = (const float*)inputVector; | |
110 | 2 | uint8_t* outputVectorPtr = outputVector; | |
111 | |||
112 | 2 | const float min_val = 0.0f; | |
113 | 2 | const float max_val = UINT8_MAX; | |
114 | 2 | const __m256 vmin_val = _mm256_set1_ps(min_val); | |
115 | 2 | const __m256 vmax_val = _mm256_set1_ps(max_val); | |
116 | |||
117 | 2 | const __m256 vScale = _mm256_set1_ps(scale); | |
118 | 2 | const __m256 vBias = _mm256_set1_ps(bias); | |
119 | |||
120 |
2/2✓ Branch 0 taken 8190 times.
✓ Branch 1 taken 2 times.
|
8192 | for (unsigned int number = 0; number < thirtysecondPoints; number++) { |
121 | 8190 | __m256 inputVal1 = _mm256_loadu_ps(inputVectorPtr); | |
122 | 8190 | inputVectorPtr += 8; | |
123 | 8190 | __m256 inputVal2 = _mm256_loadu_ps(inputVectorPtr); | |
124 | 8190 | inputVectorPtr += 8; | |
125 | 8190 | __m256 inputVal3 = _mm256_loadu_ps(inputVectorPtr); | |
126 | 8190 | inputVectorPtr += 8; | |
127 | 8190 | __m256 inputVal4 = _mm256_loadu_ps(inputVectorPtr); | |
128 | 8190 | inputVectorPtr += 8; | |
129 | |||
130 | 24570 | inputVal1 = _mm256_max_ps( | |
131 | _mm256_min_ps(_mm256_fmadd_ps(inputVal1, vScale, vBias), vmax_val), vmin_val); | ||
132 | 24570 | inputVal2 = _mm256_max_ps( | |
133 | _mm256_min_ps(_mm256_fmadd_ps(inputVal2, vScale, vBias), vmax_val), vmin_val); | ||
134 | 24570 | inputVal3 = _mm256_max_ps( | |
135 | _mm256_min_ps(_mm256_fmadd_ps(inputVal3, vScale, vBias), vmax_val), vmin_val); | ||
136 | 24570 | inputVal4 = _mm256_max_ps( | |
137 | _mm256_min_ps(_mm256_fmadd_ps(inputVal4, vScale, vBias), vmax_val), vmin_val); | ||
138 | |||
139 | 8190 | __m256i intInputVal1 = _mm256_cvtps_epi32(inputVal1); | |
140 | 8190 | __m256i intInputVal2 = _mm256_cvtps_epi32(inputVal2); | |
141 | 8190 | __m256i intInputVal3 = _mm256_cvtps_epi32(inputVal3); | |
142 | 8190 | __m256i intInputVal4 = _mm256_cvtps_epi32(inputVal4); | |
143 | |||
144 | 8190 | intInputVal1 = _mm256_packs_epi32(intInputVal1, intInputVal2); | |
145 | 8190 | intInputVal1 = _mm256_permute4x64_epi64(intInputVal1, 0b11011000); | |
146 | 8190 | intInputVal3 = _mm256_packs_epi32(intInputVal3, intInputVal4); | |
147 | 8190 | intInputVal3 = _mm256_permute4x64_epi64(intInputVal3, 0b11011000); | |
148 | |||
149 | 8190 | intInputVal1 = _mm256_packus_epi16(intInputVal1, intInputVal3); | |
150 | 8190 | const __m256i intInputVal = _mm256_permute4x64_epi64(intInputVal1, 0b11011000); | |
151 | |||
152 | _mm256_storeu_si256((__m256i*)outputVectorPtr, intInputVal); | ||
153 | 8190 | outputVectorPtr += 32; | |
154 | } | ||
155 | |||
156 |
2/2✓ Branch 0 taken 62 times.
✓ Branch 1 taken 2 times.
|
64 | for (unsigned int number = thirtysecondPoints * 32; number < num_points; number++) { |
157 | 62 | const float r = inputVector[number] * scale + bias; | |
158 | 62 | volk_32f_s32f_x2_convert_8u_single(&outputVector[number], r); | |
159 | } | ||
160 | 2 | } | |
161 | |||
162 | #endif /* LV_HAVE_AVX2 && LV_HAVE_FMA */ | ||
163 | |||
164 | |||
165 | #ifdef LV_HAVE_AVX2 | ||
166 | #include <immintrin.h> | ||
167 | |||
168 | 2 | static inline void volk_32f_s32f_x2_convert_8u_u_avx2(uint8_t* outputVector, | |
169 | const float* inputVector, | ||
170 | const float scale, | ||
171 | const float bias, | ||
172 | unsigned int num_points) | ||
173 | { | ||
174 | 2 | const unsigned int thirtysecondPoints = num_points / 32; | |
175 | |||
176 | 2 | const float* inputVectorPtr = (const float*)inputVector; | |
177 | 2 | uint8_t* outputVectorPtr = outputVector; | |
178 | |||
179 | 2 | const float min_val = 0.0f; | |
180 | 2 | const float max_val = UINT8_MAX; | |
181 | 2 | const __m256 vmin_val = _mm256_set1_ps(min_val); | |
182 | 2 | const __m256 vmax_val = _mm256_set1_ps(max_val); | |
183 | |||
184 | 2 | const __m256 vScale = _mm256_set1_ps(scale); | |
185 | 2 | const __m256 vBias = _mm256_set1_ps(bias); | |
186 | |||
187 |
2/2✓ Branch 0 taken 8190 times.
✓ Branch 1 taken 2 times.
|
8192 | for (unsigned int number = 0; number < thirtysecondPoints; number++) { |
188 | 8190 | __m256 inputVal1 = _mm256_loadu_ps(inputVectorPtr); | |
189 | 8190 | inputVectorPtr += 8; | |
190 | 8190 | __m256 inputVal2 = _mm256_loadu_ps(inputVectorPtr); | |
191 | 8190 | inputVectorPtr += 8; | |
192 | 8190 | __m256 inputVal3 = _mm256_loadu_ps(inputVectorPtr); | |
193 | 8190 | inputVectorPtr += 8; | |
194 | 8190 | __m256 inputVal4 = _mm256_loadu_ps(inputVectorPtr); | |
195 | 8190 | inputVectorPtr += 8; | |
196 | |||
197 | 32760 | inputVal1 = _mm256_max_ps( | |
198 | _mm256_min_ps(_mm256_add_ps(_mm256_mul_ps(inputVal1, vScale), vBias), | ||
199 | vmax_val), | ||
200 | vmin_val); | ||
201 | 32760 | inputVal2 = _mm256_max_ps( | |
202 | _mm256_min_ps(_mm256_add_ps(_mm256_mul_ps(inputVal2, vScale), vBias), | ||
203 | vmax_val), | ||
204 | vmin_val); | ||
205 | 32760 | inputVal3 = _mm256_max_ps( | |
206 | _mm256_min_ps(_mm256_add_ps(_mm256_mul_ps(inputVal3, vScale), vBias), | ||
207 | vmax_val), | ||
208 | vmin_val); | ||
209 | 32760 | inputVal4 = _mm256_max_ps( | |
210 | _mm256_min_ps(_mm256_add_ps(_mm256_mul_ps(inputVal4, vScale), vBias), | ||
211 | vmax_val), | ||
212 | vmin_val); | ||
213 | |||
214 | 8190 | __m256i intInputVal1 = _mm256_cvtps_epi32(inputVal1); | |
215 | 8190 | __m256i intInputVal2 = _mm256_cvtps_epi32(inputVal2); | |
216 | 8190 | __m256i intInputVal3 = _mm256_cvtps_epi32(inputVal3); | |
217 | 8190 | __m256i intInputVal4 = _mm256_cvtps_epi32(inputVal4); | |
218 | |||
219 | 8190 | intInputVal1 = _mm256_packs_epi32(intInputVal1, intInputVal2); | |
220 | 8190 | intInputVal1 = _mm256_permute4x64_epi64(intInputVal1, 0b11011000); | |
221 | 8190 | intInputVal3 = _mm256_packs_epi32(intInputVal3, intInputVal4); | |
222 | 8190 | intInputVal3 = _mm256_permute4x64_epi64(intInputVal3, 0b11011000); | |
223 | |||
224 | 8190 | intInputVal1 = _mm256_packus_epi16(intInputVal1, intInputVal3); | |
225 | 8190 | const __m256i intInputVal = _mm256_permute4x64_epi64(intInputVal1, 0b11011000); | |
226 | |||
227 | _mm256_storeu_si256((__m256i*)outputVectorPtr, intInputVal); | ||
228 | 8190 | outputVectorPtr += 32; | |
229 | } | ||
230 | |||
231 |
2/2✓ Branch 0 taken 62 times.
✓ Branch 1 taken 2 times.
|
64 | for (unsigned int number = thirtysecondPoints * 32; number < num_points; number++) { |
232 | 62 | float r = inputVector[number] * scale + bias; | |
233 | 62 | volk_32f_s32f_x2_convert_8u_single(&outputVector[number], r); | |
234 | } | ||
235 | 2 | } | |
236 | |||
237 | #endif /* LV_HAVE_AVX2 */ | ||
238 | |||
239 | |||
240 | #ifdef LV_HAVE_SSE2 | ||
241 | #include <emmintrin.h> | ||
242 | |||
243 | 2 | static inline void volk_32f_s32f_x2_convert_8u_u_sse2(uint8_t* outputVector, | |
244 | const float* inputVector, | ||
245 | const float scale, | ||
246 | const float bias, | ||
247 | unsigned int num_points) | ||
248 | { | ||
249 | 2 | const unsigned int sixteenthPoints = num_points / 16; | |
250 | |||
251 | 2 | const float* inputVectorPtr = (const float*)inputVector; | |
252 | 2 | uint8_t* outputVectorPtr = outputVector; | |
253 | |||
254 | 2 | const float min_val = 0.0f; | |
255 | 2 | const float max_val = UINT8_MAX; | |
256 | 2 | const __m128 vmin_val = _mm_set_ps1(min_val); | |
257 | 2 | const __m128 vmax_val = _mm_set_ps1(max_val); | |
258 | |||
259 | 2 | const __m128 vScale = _mm_set_ps1(scale); | |
260 | 2 | const __m128 vBias = _mm_set_ps1(bias); | |
261 | |||
262 |
2/2✓ Branch 0 taken 16382 times.
✓ Branch 1 taken 2 times.
|
16384 | for (unsigned int number = 0; number < sixteenthPoints; number++) { |
263 | 16382 | __m128 inputVal1 = _mm_loadu_ps(inputVectorPtr); | |
264 | 16382 | inputVectorPtr += 4; | |
265 | 16382 | __m128 inputVal2 = _mm_loadu_ps(inputVectorPtr); | |
266 | 16382 | inputVectorPtr += 4; | |
267 | 16382 | __m128 inputVal3 = _mm_loadu_ps(inputVectorPtr); | |
268 | 16382 | inputVectorPtr += 4; | |
269 | 16382 | __m128 inputVal4 = _mm_loadu_ps(inputVectorPtr); | |
270 | 16382 | inputVectorPtr += 4; | |
271 | |||
272 | 65528 | inputVal1 = _mm_max_ps( | |
273 | _mm_min_ps(_mm_add_ps(_mm_mul_ps(inputVal1, vScale), vBias), vmax_val), | ||
274 | vmin_val); | ||
275 | 65528 | inputVal2 = _mm_max_ps( | |
276 | _mm_min_ps(_mm_add_ps(_mm_mul_ps(inputVal2, vScale), vBias), vmax_val), | ||
277 | vmin_val); | ||
278 | 65528 | inputVal3 = _mm_max_ps( | |
279 | _mm_min_ps(_mm_add_ps(_mm_mul_ps(inputVal3, vScale), vBias), vmax_val), | ||
280 | vmin_val); | ||
281 | 65528 | inputVal4 = _mm_max_ps( | |
282 | _mm_min_ps(_mm_add_ps(_mm_mul_ps(inputVal4, vScale), vBias), vmax_val), | ||
283 | vmin_val); | ||
284 | |||
285 | 16382 | __m128i intInputVal1 = _mm_cvtps_epi32(inputVal1); | |
286 | 16382 | __m128i intInputVal2 = _mm_cvtps_epi32(inputVal2); | |
287 | 16382 | __m128i intInputVal3 = _mm_cvtps_epi32(inputVal3); | |
288 | 16382 | __m128i intInputVal4 = _mm_cvtps_epi32(inputVal4); | |
289 | |||
290 | 16382 | intInputVal1 = _mm_packs_epi32(intInputVal1, intInputVal2); | |
291 | 16382 | intInputVal3 = _mm_packs_epi32(intInputVal3, intInputVal4); | |
292 | |||
293 | 16382 | intInputVal1 = _mm_packus_epi16(intInputVal1, intInputVal3); | |
294 | |||
295 | _mm_storeu_si128((__m128i*)outputVectorPtr, intInputVal1); | ||
296 | 16382 | outputVectorPtr += 16; | |
297 | } | ||
298 | |||
299 |
2/2✓ Branch 0 taken 30 times.
✓ Branch 1 taken 2 times.
|
32 | for (unsigned int number = sixteenthPoints * 16; number < num_points; number++) { |
300 | 30 | const float r = inputVector[number] * scale + bias; | |
301 | 30 | volk_32f_s32f_x2_convert_8u_single(&outputVector[number], r); | |
302 | } | ||
303 | 2 | } | |
304 | |||
305 | #endif /* LV_HAVE_SSE2 */ | ||
306 | |||
307 | |||
308 | #ifdef LV_HAVE_SSE | ||
309 | #include <xmmintrin.h> | ||
310 | |||
311 | 2 | static inline void volk_32f_s32f_x2_convert_8u_u_sse(uint8_t* outputVector, | |
312 | const float* inputVector, | ||
313 | const float scale, | ||
314 | const float bias, | ||
315 | unsigned int num_points) | ||
316 | { | ||
317 | 2 | const unsigned int quarterPoints = num_points / 4; | |
318 | |||
319 | 2 | const float* inputVectorPtr = (const float*)inputVector; | |
320 | 2 | uint8_t* outputVectorPtr = outputVector; | |
321 | |||
322 | 2 | const float min_val = 0.0f; | |
323 | 2 | const float max_val = UINT8_MAX; | |
324 | 2 | const __m128 vmin_val = _mm_set_ps1(min_val); | |
325 | 2 | const __m128 vmax_val = _mm_set_ps1(max_val); | |
326 | |||
327 | 2 | const __m128 vScale = _mm_set_ps1(scale); | |
328 | 2 | const __m128 vBias = _mm_set_ps1(bias); | |
329 | |||
330 | __VOLK_ATTR_ALIGNED(16) float outputFloatBuffer[4]; | ||
331 | |||
332 |
2/2✓ Branch 0 taken 65534 times.
✓ Branch 1 taken 2 times.
|
65536 | for (unsigned int number = 0; number < quarterPoints; number++) { |
333 | 65534 | __m128 ret = _mm_loadu_ps(inputVectorPtr); | |
334 | 65534 | inputVectorPtr += 4; | |
335 | |||
336 | 262136 | ret = _mm_max_ps(_mm_min_ps(_mm_add_ps(_mm_mul_ps(ret, vScale), vBias), vmax_val), | |
337 | vmin_val); | ||
338 | |||
339 | _mm_store_ps(outputFloatBuffer, ret); | ||
340 |
2/2✓ Branch 0 taken 262136 times.
✓ Branch 1 taken 65534 times.
|
327670 | for (size_t inner_loop = 0; inner_loop < 4; inner_loop++) { |
341 | 262136 | *outputVectorPtr++ = (uint8_t)(rintf(outputFloatBuffer[inner_loop])); | |
342 | } | ||
343 | } | ||
344 | |||
345 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 2 times.
|
8 | for (unsigned int number = quarterPoints * 4; number < num_points; number++) { |
346 | 6 | const float r = inputVector[number] * scale + bias; | |
347 | 6 | volk_32f_s32f_x2_convert_8u_single(&outputVector[number], r); | |
348 | } | ||
349 | 2 | } | |
350 | |||
351 | #endif /* LV_HAVE_SSE */ | ||
352 | |||
353 | |||
354 | #endif /* INCLUDED_volk_32f_s32f_x2_convert_8u_u_H */ | ||
355 | #ifndef INCLUDED_volk_32f_s32f_x2_convert_8u_a_H | ||
356 | #define INCLUDED_volk_32f_s32f_x2_convert_8u_a_H | ||
357 | |||
358 | #include <inttypes.h> | ||
359 | #include <volk/volk_common.h> | ||
360 | |||
361 | #if LV_HAVE_AVX2 && LV_HAVE_FMA | ||
362 | #include <immintrin.h> | ||
363 | |||
364 | 2 | static inline void volk_32f_s32f_x2_convert_8u_a_avx2_fma(uint8_t* outputVector, | |
365 | const float* inputVector, | ||
366 | const float scale, | ||
367 | const float bias, | ||
368 | unsigned int num_points) | ||
369 | { | ||
370 | 2 | const unsigned int thirtysecondPoints = num_points / 32; | |
371 | |||
372 | 2 | const float* inputVectorPtr = (const float*)inputVector; | |
373 | 2 | uint8_t* outputVectorPtr = outputVector; | |
374 | |||
375 | 2 | const float min_val = 0.0f; | |
376 | 2 | const float max_val = UINT8_MAX; | |
377 | 2 | const __m256 vmin_val = _mm256_set1_ps(min_val); | |
378 | 2 | const __m256 vmax_val = _mm256_set1_ps(max_val); | |
379 | |||
380 | 2 | const __m256 vScale = _mm256_set1_ps(scale); | |
381 | 2 | const __m256 vBias = _mm256_set1_ps(bias); | |
382 | |||
383 |
2/2✓ Branch 0 taken 8190 times.
✓ Branch 1 taken 2 times.
|
8192 | for (unsigned int number = 0; number < thirtysecondPoints; number++) { |
384 | 8190 | __m256 inputVal1 = _mm256_load_ps(inputVectorPtr); | |
385 | 8190 | inputVectorPtr += 8; | |
386 | 8190 | __m256 inputVal2 = _mm256_load_ps(inputVectorPtr); | |
387 | 8190 | inputVectorPtr += 8; | |
388 | 8190 | __m256 inputVal3 = _mm256_load_ps(inputVectorPtr); | |
389 | 8190 | inputVectorPtr += 8; | |
390 | 8190 | __m256 inputVal4 = _mm256_load_ps(inputVectorPtr); | |
391 | 8190 | inputVectorPtr += 8; | |
392 | |||
393 | 24570 | inputVal1 = _mm256_max_ps( | |
394 | _mm256_min_ps(_mm256_fmadd_ps(inputVal1, vScale, vBias), vmax_val), vmin_val); | ||
395 | 24570 | inputVal2 = _mm256_max_ps( | |
396 | _mm256_min_ps(_mm256_fmadd_ps(inputVal2, vScale, vBias), vmax_val), vmin_val); | ||
397 | 24570 | inputVal3 = _mm256_max_ps( | |
398 | _mm256_min_ps(_mm256_fmadd_ps(inputVal3, vScale, vBias), vmax_val), vmin_val); | ||
399 | 24570 | inputVal4 = _mm256_max_ps( | |
400 | _mm256_min_ps(_mm256_fmadd_ps(inputVal4, vScale, vBias), vmax_val), vmin_val); | ||
401 | |||
402 | 8190 | __m256i intInputVal1 = _mm256_cvtps_epi32(inputVal1); | |
403 | 8190 | __m256i intInputVal2 = _mm256_cvtps_epi32(inputVal2); | |
404 | 8190 | __m256i intInputVal3 = _mm256_cvtps_epi32(inputVal3); | |
405 | 8190 | __m256i intInputVal4 = _mm256_cvtps_epi32(inputVal4); | |
406 | |||
407 | 8190 | intInputVal1 = _mm256_packs_epi32(intInputVal1, intInputVal2); | |
408 | 8190 | intInputVal1 = _mm256_permute4x64_epi64(intInputVal1, 0b11011000); | |
409 | 8190 | intInputVal3 = _mm256_packs_epi32(intInputVal3, intInputVal4); | |
410 | 8190 | intInputVal3 = _mm256_permute4x64_epi64(intInputVal3, 0b11011000); | |
411 | |||
412 | 8190 | intInputVal1 = _mm256_packus_epi16(intInputVal1, intInputVal3); | |
413 | 8190 | const __m256i intInputVal = _mm256_permute4x64_epi64(intInputVal1, 0b11011000); | |
414 | |||
415 | _mm256_store_si256((__m256i*)outputVectorPtr, intInputVal); | ||
416 | 8190 | outputVectorPtr += 32; | |
417 | } | ||
418 | |||
419 |
2/2✓ Branch 0 taken 62 times.
✓ Branch 1 taken 2 times.
|
64 | for (unsigned int number = thirtysecondPoints * 32; number < num_points; number++) { |
420 | 62 | const float r = inputVector[number] * scale + bias; | |
421 | 62 | volk_32f_s32f_x2_convert_8u_single(&outputVector[number], r); | |
422 | } | ||
423 | 2 | } | |
424 | |||
425 | #endif /* LV_HAVE_AVX2 && LV_HAVE_FMA */ | ||
426 | |||
427 | |||
428 | #ifdef LV_HAVE_AVX2 | ||
429 | #include <immintrin.h> | ||
430 | |||
431 | 2 | static inline void volk_32f_s32f_x2_convert_8u_a_avx2(uint8_t* outputVector, | |
432 | const float* inputVector, | ||
433 | const float scale, | ||
434 | const float bias, | ||
435 | unsigned int num_points) | ||
436 | { | ||
437 | 2 | const unsigned int thirtysecondPoints = num_points / 32; | |
438 | |||
439 | 2 | const float* inputVectorPtr = (const float*)inputVector; | |
440 | 2 | uint8_t* outputVectorPtr = outputVector; | |
441 | |||
442 | 2 | const float min_val = 0.0f; | |
443 | 2 | const float max_val = UINT8_MAX; | |
444 | 2 | const __m256 vmin_val = _mm256_set1_ps(min_val); | |
445 | 2 | const __m256 vmax_val = _mm256_set1_ps(max_val); | |
446 | |||
447 | 2 | const __m256 vScale = _mm256_set1_ps(scale); | |
448 | 2 | const __m256 vBias = _mm256_set1_ps(bias); | |
449 | |||
450 |
2/2✓ Branch 0 taken 8190 times.
✓ Branch 1 taken 2 times.
|
8192 | for (unsigned int number = 0; number < thirtysecondPoints; number++) { |
451 | 8190 | __m256 inputVal1 = _mm256_load_ps(inputVectorPtr); | |
452 | 8190 | inputVectorPtr += 8; | |
453 | 8190 | __m256 inputVal2 = _mm256_load_ps(inputVectorPtr); | |
454 | 8190 | inputVectorPtr += 8; | |
455 | 8190 | __m256 inputVal3 = _mm256_load_ps(inputVectorPtr); | |
456 | 8190 | inputVectorPtr += 8; | |
457 | 8190 | __m256 inputVal4 = _mm256_load_ps(inputVectorPtr); | |
458 | 8190 | inputVectorPtr += 8; | |
459 | |||
460 | 32760 | inputVal1 = _mm256_max_ps( | |
461 | _mm256_min_ps(_mm256_add_ps(_mm256_mul_ps(inputVal1, vScale), vBias), | ||
462 | vmax_val), | ||
463 | vmin_val); | ||
464 | 32760 | inputVal2 = _mm256_max_ps( | |
465 | _mm256_min_ps(_mm256_add_ps(_mm256_mul_ps(inputVal2, vScale), vBias), | ||
466 | vmax_val), | ||
467 | vmin_val); | ||
468 | 32760 | inputVal3 = _mm256_max_ps( | |
469 | _mm256_min_ps(_mm256_add_ps(_mm256_mul_ps(inputVal3, vScale), vBias), | ||
470 | vmax_val), | ||
471 | vmin_val); | ||
472 | 32760 | inputVal4 = _mm256_max_ps( | |
473 | _mm256_min_ps(_mm256_add_ps(_mm256_mul_ps(inputVal4, vScale), vBias), | ||
474 | vmax_val), | ||
475 | vmin_val); | ||
476 | |||
477 | 8190 | __m256i intInputVal1 = _mm256_cvtps_epi32(inputVal1); | |
478 | 8190 | __m256i intInputVal2 = _mm256_cvtps_epi32(inputVal2); | |
479 | 8190 | __m256i intInputVal3 = _mm256_cvtps_epi32(inputVal3); | |
480 | 8190 | __m256i intInputVal4 = _mm256_cvtps_epi32(inputVal4); | |
481 | |||
482 | 8190 | intInputVal1 = _mm256_packs_epi32(intInputVal1, intInputVal2); | |
483 | 8190 | intInputVal1 = _mm256_permute4x64_epi64(intInputVal1, 0b11011000); | |
484 | 8190 | intInputVal3 = _mm256_packs_epi32(intInputVal3, intInputVal4); | |
485 | 8190 | intInputVal3 = _mm256_permute4x64_epi64(intInputVal3, 0b11011000); | |
486 | |||
487 | 8190 | intInputVal1 = _mm256_packus_epi16(intInputVal1, intInputVal3); | |
488 | 8190 | const __m256i intInputVal = _mm256_permute4x64_epi64(intInputVal1, 0b11011000); | |
489 | |||
490 | _mm256_store_si256((__m256i*)outputVectorPtr, intInputVal); | ||
491 | 8190 | outputVectorPtr += 32; | |
492 | } | ||
493 | |||
494 |
2/2✓ Branch 0 taken 62 times.
✓ Branch 1 taken 2 times.
|
64 | for (unsigned int number = thirtysecondPoints * 32; number < num_points; number++) { |
495 | 62 | const float r = inputVector[number] * scale + bias; | |
496 | 62 | volk_32f_s32f_x2_convert_8u_single(&outputVector[number], r); | |
497 | } | ||
498 | 2 | } | |
499 | |||
500 | #endif /* LV_HAVE_AVX2 */ | ||
501 | |||
502 | |||
503 | #ifdef LV_HAVE_SSE2 | ||
504 | #include <emmintrin.h> | ||
505 | |||
506 | 2 | static inline void volk_32f_s32f_x2_convert_8u_a_sse2(uint8_t* outputVector, | |
507 | const float* inputVector, | ||
508 | const float scale, | ||
509 | const float bias, | ||
510 | unsigned int num_points) | ||
511 | { | ||
512 | 2 | const unsigned int sixteenthPoints = num_points / 16; | |
513 | |||
514 | 2 | const float* inputVectorPtr = (const float*)inputVector; | |
515 | 2 | uint8_t* outputVectorPtr = outputVector; | |
516 | |||
517 | 2 | const float min_val = 0.0f; | |
518 | 2 | const float max_val = UINT8_MAX; | |
519 | 2 | const __m128 vmin_val = _mm_set_ps1(min_val); | |
520 | 2 | const __m128 vmax_val = _mm_set_ps1(max_val); | |
521 | |||
522 | 2 | const __m128 vScale = _mm_set_ps1(scale); | |
523 | 2 | const __m128 vBias = _mm_set_ps1(bias); | |
524 | |||
525 |
2/2✓ Branch 0 taken 16382 times.
✓ Branch 1 taken 2 times.
|
16384 | for (unsigned int number = 0; number < sixteenthPoints; number++) { |
526 | 16382 | __m128 inputVal1 = _mm_load_ps(inputVectorPtr); | |
527 | 16382 | inputVectorPtr += 4; | |
528 | 16382 | __m128 inputVal2 = _mm_load_ps(inputVectorPtr); | |
529 | 16382 | inputVectorPtr += 4; | |
530 | 16382 | __m128 inputVal3 = _mm_load_ps(inputVectorPtr); | |
531 | 16382 | inputVectorPtr += 4; | |
532 | 16382 | __m128 inputVal4 = _mm_load_ps(inputVectorPtr); | |
533 | 16382 | inputVectorPtr += 4; | |
534 | |||
535 | 65528 | inputVal1 = _mm_max_ps( | |
536 | _mm_min_ps(_mm_add_ps(_mm_mul_ps(inputVal1, vScale), vBias), vmax_val), | ||
537 | vmin_val); | ||
538 | 65528 | inputVal2 = _mm_max_ps( | |
539 | _mm_min_ps(_mm_add_ps(_mm_mul_ps(inputVal2, vScale), vBias), vmax_val), | ||
540 | vmin_val); | ||
541 | 65528 | inputVal3 = _mm_max_ps( | |
542 | _mm_min_ps(_mm_add_ps(_mm_mul_ps(inputVal3, vScale), vBias), vmax_val), | ||
543 | vmin_val); | ||
544 | 65528 | inputVal4 = _mm_max_ps( | |
545 | _mm_min_ps(_mm_add_ps(_mm_mul_ps(inputVal4, vScale), vBias), vmax_val), | ||
546 | vmin_val); | ||
547 | |||
548 | 16382 | __m128i intInputVal1 = _mm_cvtps_epi32(inputVal1); | |
549 | 16382 | __m128i intInputVal2 = _mm_cvtps_epi32(inputVal2); | |
550 | 16382 | __m128i intInputVal3 = _mm_cvtps_epi32(inputVal3); | |
551 | 16382 | __m128i intInputVal4 = _mm_cvtps_epi32(inputVal4); | |
552 | |||
553 | 16382 | intInputVal1 = _mm_packs_epi32(intInputVal1, intInputVal2); | |
554 | 16382 | intInputVal3 = _mm_packs_epi32(intInputVal3, intInputVal4); | |
555 | |||
556 | 16382 | intInputVal1 = _mm_packus_epi16(intInputVal1, intInputVal3); | |
557 | |||
558 | _mm_store_si128((__m128i*)outputVectorPtr, intInputVal1); | ||
559 | 16382 | outputVectorPtr += 16; | |
560 | } | ||
561 | |||
562 |
2/2✓ Branch 0 taken 30 times.
✓ Branch 1 taken 2 times.
|
32 | for (unsigned int number = sixteenthPoints * 16; number < num_points; number++) { |
563 | 30 | const float r = inputVector[number] * scale + bias; | |
564 | 30 | volk_32f_s32f_x2_convert_8u_single(&outputVector[number], r); | |
565 | } | ||
566 | 2 | } | |
567 | #endif /* LV_HAVE_SSE2 */ | ||
568 | |||
569 | |||
570 | #ifdef LV_HAVE_SSE | ||
571 | #include <xmmintrin.h> | ||
572 | |||
573 | 2 | static inline void volk_32f_s32f_x2_convert_8u_a_sse(uint8_t* outputVector, | |
574 | const float* inputVector, | ||
575 | const float scale, | ||
576 | const float bias, | ||
577 | unsigned int num_points) | ||
578 | { | ||
579 | 2 | const unsigned int quarterPoints = num_points / 4; | |
580 | |||
581 | 2 | const float* inputVectorPtr = (const float*)inputVector; | |
582 | 2 | uint8_t* outputVectorPtr = outputVector; | |
583 | |||
584 | 2 | const float min_val = 0.0f; | |
585 | 2 | const float max_val = UINT8_MAX; | |
586 | 2 | const __m128 vmin_val = _mm_set_ps1(min_val); | |
587 | 2 | const __m128 vmax_val = _mm_set_ps1(max_val); | |
588 | |||
589 | 2 | const __m128 vScalar = _mm_set_ps1(scale); | |
590 | 2 | const __m128 vBias = _mm_set_ps1(bias); | |
591 | |||
592 | __VOLK_ATTR_ALIGNED(16) float outputFloatBuffer[4]; | ||
593 | |||
594 |
2/2✓ Branch 0 taken 65534 times.
✓ Branch 1 taken 2 times.
|
65536 | for (unsigned int number = 0; number < quarterPoints; number++) { |
595 | 65534 | __m128 ret = _mm_load_ps(inputVectorPtr); | |
596 | 65534 | inputVectorPtr += 4; | |
597 | |||
598 | 262136 | ret = _mm_max_ps( | |
599 | _mm_min_ps(_mm_add_ps(_mm_mul_ps(ret, vScalar), vBias), vmax_val), vmin_val); | ||
600 | |||
601 | _mm_store_ps(outputFloatBuffer, ret); | ||
602 |
2/2✓ Branch 0 taken 262136 times.
✓ Branch 1 taken 65534 times.
|
327670 | for (size_t inner_loop = 0; inner_loop < 4; inner_loop++) { |
603 | 262136 | *outputVectorPtr++ = (uint8_t)(rintf(outputFloatBuffer[inner_loop])); | |
604 | } | ||
605 | } | ||
606 | |||
607 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 2 times.
|
8 | for (unsigned int number = quarterPoints * 4; number < num_points; number++) { |
608 | 6 | const float r = inputVector[number] * scale + bias; | |
609 | 6 | volk_32f_s32f_x2_convert_8u_single(&outputVector[number], r); | |
610 | } | ||
611 | 2 | } | |
612 | |||
613 | #endif /* LV_HAVE_SSE */ | ||
614 | |||
615 | |||
616 | #endif /* INCLUDED_volk_32f_s32f_x2_convert_8u_a_H */ | ||
617 |