blob: eb0d86aee73821fecefe5998199a5c58684cf075 [file] [log] [blame]
Marat Dukhan403b7d42019-12-05 12:49:11 -08001// Auto-generated file. Do not edit!
2// Template: src/f32-vbinary/vopc-sse.c.in
3// Generator: tools/xngen
4//
5// Copyright 2019 Google LLC
6//
7// This source code is licensed under the BSD-style license found in the
8// LICENSE file in the root directory of this source tree.
9
10#include <assert.h>
11
12#include <xmmintrin.h>
13
14#include <xnnpack/common.h>
Marat Dukhan8aaf1862020-04-28 10:26:13 -070015#include <xnnpack/intrinsics-polyfill.h>
Marat Dukhan403b7d42019-12-05 12:49:11 -080016#include <xnnpack/vbinary.h>
17
18
19void xnn_f32_vminc_ukernel__sse_x4(
20 size_t n,
21 const float* a,
22 const float* b,
23 float* y,
Marat Dukhanb2217dd2020-05-28 17:30:28 -070024 const union xnn_f32_default_params params[restrict XNN_MIN_ELEMENTS(1)]) XNN_DISABLE_TSAN
Marat Dukhan403b7d42019-12-05 12:49:11 -080025{
26 assert(n != 0);
27 assert(n % sizeof(float) == 0);
Frank Barchard0822dde2020-07-04 12:47:24 -070028 assert(a != NULL);
29 assert(b != NULL);
30 assert(y != NULL);
Marat Dukhan403b7d42019-12-05 12:49:11 -080031
Marat Dukhan403b7d42019-12-05 12:49:11 -080032
33 const __m128 vb = _mm_load1_ps(b);
34 for (; n >= 4 * sizeof(float); n -= 4 * sizeof(float)) {
35 const __m128 va0123 = _mm_loadu_ps(a);
36 a += 4;
37
38 __m128 vy0123 = _mm_min_ps(va0123, vb);
39
Marat Dukhan403b7d42019-12-05 12:49:11 -080040
Marat Dukhan13bafb02020-06-05 00:43:11 -070041
Marat Dukhan403b7d42019-12-05 12:49:11 -080042 _mm_storeu_ps(y, vy0123);
43 y += 4;
44 }
Marat Dukhan403b7d42019-12-05 12:49:11 -080045 if XNN_UNLIKELY(n != 0) {
Marat Dukhanb2217dd2020-05-28 17:30:28 -070046 const __m128 va0123 = _mm_loadu_ps(a);
Marat Dukhan403b7d42019-12-05 12:49:11 -080047
48 __m128 vy0123 = _mm_min_ps(va0123, vb);
Marat Dukhan403b7d42019-12-05 12:49:11 -080049 if (n & (2 * sizeof(float))) {
50 _mm_storel_pi((__m64*) y, vy0123);
51 vy0123 = _mm_movehl_ps(vy0123, vy0123);
52 y += 2;
53 }
54 if (n & (1 * sizeof(float))) {
55 _mm_store_ss(y, vy0123);
56 }
57 }
58}