mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2015 Google Inc. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
| 7 | |
| 8 | #ifndef SkNx_sse_DEFINED |
| 9 | #define SkNx_sse_DEFINED |
| 10 | |
mtklein | e18fa44 | 2016-06-09 13:40:56 -0700 | [diff] [blame] | 11 | #include <immintrin.h> |
mtklein | 244a653 | 2016-04-19 14:21:30 -0700 | [diff] [blame] | 12 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 13 | // This file may assume <= SSE2, but must check SK_CPU_SSE_LEVEL for anything more recent. |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 14 | // If you do, make sure this is in a static inline function... anywhere else risks violating ODR. |
mtklein | aa999cb | 2015-05-22 17:18:21 -0700 | [diff] [blame] | 15 | |
fmalita | c2e0ac4 | 2015-12-03 09:15:25 -0800 | [diff] [blame] | 16 | #define SKNX_IS_FAST |
| 17 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 18 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 19 | class SkNx<2, float> { |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 20 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 21 | SkNx(const __m128& vec) : fVec(vec) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 22 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 23 | SkNx() {} |
| 24 | SkNx(float val) : fVec(_mm_set1_ps(val)) {} |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 25 | static SkNx Load(const void* ptr) { |
| 26 | return _mm_castsi128_ps(_mm_loadl_epi64((const __m128i*)ptr)); |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 27 | } |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 28 | SkNx(float a, float b) : fVec(_mm_setr_ps(a,b,0,0)) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 29 | |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 30 | void store(void* ptr) const { _mm_storel_pi((__m64*)ptr, fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 31 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 32 | SkNx operator + (const SkNx& o) const { return _mm_add_ps(fVec, o.fVec); } |
| 33 | SkNx operator - (const SkNx& o) const { return _mm_sub_ps(fVec, o.fVec); } |
| 34 | SkNx operator * (const SkNx& o) const { return _mm_mul_ps(fVec, o.fVec); } |
| 35 | SkNx operator / (const SkNx& o) const { return _mm_div_ps(fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 36 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 37 | SkNx operator == (const SkNx& o) const { return _mm_cmpeq_ps (fVec, o.fVec); } |
| 38 | SkNx operator != (const SkNx& o) const { return _mm_cmpneq_ps(fVec, o.fVec); } |
| 39 | SkNx operator < (const SkNx& o) const { return _mm_cmplt_ps (fVec, o.fVec); } |
| 40 | SkNx operator > (const SkNx& o) const { return _mm_cmpgt_ps (fVec, o.fVec); } |
| 41 | SkNx operator <= (const SkNx& o) const { return _mm_cmple_ps (fVec, o.fVec); } |
| 42 | SkNx operator >= (const SkNx& o) const { return _mm_cmpge_ps (fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 43 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 44 | static SkNx Min(const SkNx& l, const SkNx& r) { return _mm_min_ps(l.fVec, r.fVec); } |
| 45 | static SkNx Max(const SkNx& l, const SkNx& r) { return _mm_max_ps(l.fVec, r.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 46 | |
mtklein | f8f90e4 | 2016-03-21 10:04:46 -0700 | [diff] [blame] | 47 | SkNx sqrt() const { return _mm_sqrt_ps (fVec); } |
| 48 | SkNx rsqrt() const { return _mm_rsqrt_ps(fVec); } |
| 49 | SkNx invert() const { return _mm_rcp_ps(fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 50 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 51 | float operator[](int k) const { |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 52 | SkASSERT(0 <= k && k < 2); |
| 53 | union { __m128 v; float fs[4]; } pun = {fVec}; |
mtklein | a156a8f | 2015-04-03 06:16:13 -0700 | [diff] [blame] | 54 | return pun.fs[k&1]; |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 55 | } |
| 56 | |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 57 | bool allTrue() const { return 0xff == (_mm_movemask_epi8(_mm_castps_si128(fVec)) & 0xff); } |
| 58 | bool anyTrue() const { return 0x00 != (_mm_movemask_epi8(_mm_castps_si128(fVec)) & 0xff); } |
| 59 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 60 | __m128 fVec; |
| 61 | }; |
| 62 | |
| 63 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 64 | class SkNx<4, float> { |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 65 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 66 | SkNx(const __m128& vec) : fVec(vec) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 67 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 68 | SkNx() {} |
| 69 | SkNx(float val) : fVec( _mm_set1_ps(val) ) {} |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 70 | SkNx(float a, float b, float c, float d) : fVec(_mm_setr_ps(a,b,c,d)) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 71 | |
Mike Klein | 33cbfd7 | 2016-10-06 11:09:27 -0400 | [diff] [blame] | 72 | static SkNx Load(const void* ptr) { return _mm_loadu_ps((const float*)ptr); } |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 73 | void store(void* ptr) const { _mm_storeu_ps((float*)ptr, fVec); } |
mtklein | 9db43ac | 2015-12-01 07:10:21 -0800 | [diff] [blame] | 74 | |
Mike Klein | 33cbfd7 | 2016-10-06 11:09:27 -0400 | [diff] [blame] | 75 | static void Load4(const void* ptr, SkNx* r, SkNx* g, SkNx* b, SkNx* a) { |
| 76 | __m128 v0 = _mm_loadu_ps(((float*)ptr) + 0), |
| 77 | v1 = _mm_loadu_ps(((float*)ptr) + 4), |
| 78 | v2 = _mm_loadu_ps(((float*)ptr) + 8), |
| 79 | v3 = _mm_loadu_ps(((float*)ptr) + 12); |
| 80 | _MM_TRANSPOSE4_PS(v0, v1, v2, v3); |
| 81 | *r = v0; |
| 82 | *g = v1; |
| 83 | *b = v2; |
| 84 | *a = v3; |
| 85 | } |
| 86 | static void Store4(void* dst, const SkNx& r, const SkNx& g, const SkNx& b, const SkNx& a) { |
| 87 | __m128 v0 = r.fVec, |
| 88 | v1 = g.fVec, |
| 89 | v2 = b.fVec, |
| 90 | v3 = a.fVec; |
| 91 | _MM_TRANSPOSE4_PS(v0, v1, v2, v3); |
| 92 | _mm_storeu_ps(((float*) dst) + 0, v0); |
| 93 | _mm_storeu_ps(((float*) dst) + 4, v1); |
| 94 | _mm_storeu_ps(((float*) dst) + 8, v2); |
| 95 | _mm_storeu_ps(((float*) dst) + 12, v3); |
| 96 | } |
| 97 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 98 | SkNx operator + (const SkNx& o) const { return _mm_add_ps(fVec, o.fVec); } |
| 99 | SkNx operator - (const SkNx& o) const { return _mm_sub_ps(fVec, o.fVec); } |
| 100 | SkNx operator * (const SkNx& o) const { return _mm_mul_ps(fVec, o.fVec); } |
| 101 | SkNx operator / (const SkNx& o) const { return _mm_div_ps(fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 102 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 103 | SkNx operator == (const SkNx& o) const { return _mm_cmpeq_ps (fVec, o.fVec); } |
| 104 | SkNx operator != (const SkNx& o) const { return _mm_cmpneq_ps(fVec, o.fVec); } |
| 105 | SkNx operator < (const SkNx& o) const { return _mm_cmplt_ps (fVec, o.fVec); } |
| 106 | SkNx operator > (const SkNx& o) const { return _mm_cmpgt_ps (fVec, o.fVec); } |
| 107 | SkNx operator <= (const SkNx& o) const { return _mm_cmple_ps (fVec, o.fVec); } |
| 108 | SkNx operator >= (const SkNx& o) const { return _mm_cmpge_ps (fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 109 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 110 | static SkNx Min(const SkNx& l, const SkNx& r) { return _mm_min_ps(l.fVec, r.fVec); } |
| 111 | static SkNx Max(const SkNx& l, const SkNx& r) { return _mm_max_ps(l.fVec, r.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 112 | |
mtklein | c33065a | 2016-01-15 12:16:40 -0800 | [diff] [blame] | 113 | SkNx abs() const { return _mm_andnot_ps(_mm_set1_ps(-0.0f), fVec); } |
mtklein | 244a653 | 2016-04-19 14:21:30 -0700 | [diff] [blame] | 114 | SkNx floor() const { |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 115 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
| 116 | return _mm_floor_ps(fVec); |
| 117 | #else |
mtklein | 244a653 | 2016-04-19 14:21:30 -0700 | [diff] [blame] | 118 | // Emulate _mm_floor_ps() with SSE2: |
| 119 | // - roundtrip through integers via truncation |
| 120 | // - subtract 1 if that's too big (possible for negative values). |
| 121 | // This restricts the domain of our inputs to a maximum somehwere around 2^31. |
| 122 | // Seems plenty big. |
| 123 | __m128 roundtrip = _mm_cvtepi32_ps(_mm_cvttps_epi32(fVec)); |
| 124 | __m128 too_big = _mm_cmpgt_ps(roundtrip, fVec); |
| 125 | return _mm_sub_ps(roundtrip, _mm_and_ps(too_big, _mm_set1_ps(1.0f))); |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 126 | #endif |
mtklein | 244a653 | 2016-04-19 14:21:30 -0700 | [diff] [blame] | 127 | } |
mtklein | c33065a | 2016-01-15 12:16:40 -0800 | [diff] [blame] | 128 | |
mtklein | f8f90e4 | 2016-03-21 10:04:46 -0700 | [diff] [blame] | 129 | SkNx sqrt() const { return _mm_sqrt_ps (fVec); } |
| 130 | SkNx rsqrt() const { return _mm_rsqrt_ps(fVec); } |
| 131 | SkNx invert() const { return _mm_rcp_ps(fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 132 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 133 | float operator[](int k) const { |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 134 | SkASSERT(0 <= k && k < 4); |
| 135 | union { __m128 v; float fs[4]; } pun = {fVec}; |
mtklein | a156a8f | 2015-04-03 06:16:13 -0700 | [diff] [blame] | 136 | return pun.fs[k&3]; |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 137 | } |
| 138 | |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 139 | bool allTrue() const { return 0xffff == _mm_movemask_epi8(_mm_castps_si128(fVec)); } |
| 140 | bool anyTrue() const { return 0x0000 != _mm_movemask_epi8(_mm_castps_si128(fVec)); } |
| 141 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 142 | SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 143 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
msarett | d280957 | 2016-06-20 06:07:45 -0700 | [diff] [blame] | 144 | return _mm_blendv_ps(e.fVec, t.fVec, fVec); |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 145 | #else |
mtklein | 2aab22a | 2015-06-26 10:46:31 -0700 | [diff] [blame] | 146 | return _mm_or_ps(_mm_and_ps (fVec, t.fVec), |
| 147 | _mm_andnot_ps(fVec, e.fVec)); |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 148 | #endif |
mtklein | 2aab22a | 2015-06-26 10:46:31 -0700 | [diff] [blame] | 149 | } |
| 150 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 151 | __m128 fVec; |
| 152 | }; |
| 153 | |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 154 | template <> |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 155 | class SkNx<4, int32_t> { |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 156 | public: |
| 157 | SkNx(const __m128i& vec) : fVec(vec) {} |
| 158 | |
| 159 | SkNx() {} |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 160 | SkNx(int32_t val) : fVec(_mm_set1_epi32(val)) {} |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 161 | static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 162 | SkNx(int32_t a, int32_t b, int32_t c, int32_t d) : fVec(_mm_setr_epi32(a,b,c,d)) {} |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 163 | |
| 164 | void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
| 165 | |
| 166 | SkNx operator + (const SkNx& o) const { return _mm_add_epi32(fVec, o.fVec); } |
| 167 | SkNx operator - (const SkNx& o) const { return _mm_sub_epi32(fVec, o.fVec); } |
| 168 | SkNx operator * (const SkNx& o) const { |
| 169 | __m128i mul20 = _mm_mul_epu32(fVec, o.fVec), |
| 170 | mul31 = _mm_mul_epu32(_mm_srli_si128(fVec, 4), _mm_srli_si128(o.fVec, 4)); |
| 171 | return _mm_unpacklo_epi32(_mm_shuffle_epi32(mul20, _MM_SHUFFLE(0,0,2,0)), |
| 172 | _mm_shuffle_epi32(mul31, _MM_SHUFFLE(0,0,2,0))); |
| 173 | } |
| 174 | |
mtklein | 281b33f | 2016-07-12 15:01:26 -0700 | [diff] [blame] | 175 | SkNx operator & (const SkNx& o) const { return _mm_and_si128(fVec, o.fVec); } |
mtklein | 64f061a | 2016-06-17 12:09:16 -0700 | [diff] [blame] | 176 | SkNx operator | (const SkNx& o) const { return _mm_or_si128(fVec, o.fVec); } |
mtklein | 58e389b | 2016-07-15 07:00:11 -0700 | [diff] [blame] | 177 | SkNx operator ^ (const SkNx& o) const { return _mm_xor_si128(fVec, o.fVec); } |
mtklein | 64f061a | 2016-06-17 12:09:16 -0700 | [diff] [blame] | 178 | |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 179 | SkNx operator << (int bits) const { return _mm_slli_epi32(fVec, bits); } |
| 180 | SkNx operator >> (int bits) const { return _mm_srai_epi32(fVec, bits); } |
| 181 | |
mtklein | 58e389b | 2016-07-15 07:00:11 -0700 | [diff] [blame] | 182 | SkNx operator == (const SkNx& o) const { return _mm_cmpeq_epi32 (fVec, o.fVec); } |
| 183 | SkNx operator < (const SkNx& o) const { return _mm_cmplt_epi32 (fVec, o.fVec); } |
| 184 | SkNx operator > (const SkNx& o) const { return _mm_cmpgt_epi32 (fVec, o.fVec); } |
| 185 | |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 186 | int32_t operator[](int k) const { |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 187 | SkASSERT(0 <= k && k < 4); |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 188 | union { __m128i v; int32_t is[4]; } pun = {fVec}; |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 189 | return pun.is[k&3]; |
| 190 | } |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 191 | |
mtklein | 58e389b | 2016-07-15 07:00:11 -0700 | [diff] [blame] | 192 | SkNx thenElse(const SkNx& t, const SkNx& e) const { |
| 193 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
| 194 | return _mm_blendv_epi8(e.fVec, t.fVec, fVec); |
| 195 | #else |
| 196 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 197 | _mm_andnot_si128(fVec, e.fVec)); |
| 198 | #endif |
| 199 | } |
| 200 | |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 201 | __m128i fVec; |
| 202 | }; |
| 203 | |
| 204 | template <> |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 205 | class SkNx<4, uint32_t> { |
| 206 | public: |
| 207 | SkNx(const __m128i& vec) : fVec(vec) {} |
| 208 | |
| 209 | SkNx() {} |
| 210 | SkNx(uint32_t val) : fVec(_mm_set1_epi32(val)) {} |
| 211 | static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
| 212 | SkNx(uint32_t a, uint32_t b, uint32_t c, uint32_t d) : fVec(_mm_setr_epi32(a,b,c,d)) {} |
| 213 | |
| 214 | void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
| 215 | |
| 216 | SkNx operator + (const SkNx& o) const { return _mm_add_epi32(fVec, o.fVec); } |
| 217 | SkNx operator - (const SkNx& o) const { return _mm_sub_epi32(fVec, o.fVec); } |
| 218 | // Not quite sure how to best do operator * in SSE2. We probably don't use it. |
| 219 | |
| 220 | SkNx operator & (const SkNx& o) const { return _mm_and_si128(fVec, o.fVec); } |
| 221 | SkNx operator | (const SkNx& o) const { return _mm_or_si128(fVec, o.fVec); } |
| 222 | SkNx operator ^ (const SkNx& o) const { return _mm_xor_si128(fVec, o.fVec); } |
| 223 | |
| 224 | SkNx operator << (int bits) const { return _mm_slli_epi32(fVec, bits); } |
| 225 | SkNx operator >> (int bits) const { return _mm_srli_epi32(fVec, bits); } |
| 226 | |
| 227 | SkNx operator == (const SkNx& o) const { return _mm_cmpeq_epi32 (fVec, o.fVec); } |
| 228 | // operator < and > take a little extra fiddling to make work for unsigned ints. |
| 229 | |
| 230 | uint32_t operator[](int k) const { |
| 231 | SkASSERT(0 <= k && k < 4); |
| 232 | union { __m128i v; uint32_t us[4]; } pun = {fVec}; |
| 233 | return pun.us[k&3]; |
| 234 | } |
| 235 | |
| 236 | SkNx thenElse(const SkNx& t, const SkNx& e) const { |
| 237 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
| 238 | return _mm_blendv_epi8(e.fVec, t.fVec, fVec); |
| 239 | #else |
| 240 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 241 | _mm_andnot_si128(fVec, e.fVec)); |
| 242 | #endif |
| 243 | } |
| 244 | |
| 245 | __m128i fVec; |
| 246 | }; |
| 247 | |
| 248 | |
| 249 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 250 | class SkNx<4, uint16_t> { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 251 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 252 | SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 253 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 254 | SkNx() {} |
| 255 | SkNx(uint16_t val) : fVec(_mm_set1_epi16(val)) {} |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 256 | SkNx(uint16_t a, uint16_t b, uint16_t c, uint16_t d) : fVec(_mm_setr_epi16(a,b,c,d,0,0,0,0)) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 257 | |
Mike Klein | 33cbfd7 | 2016-10-06 11:09:27 -0400 | [diff] [blame] | 258 | static SkNx Load(const void* ptr) { return _mm_loadl_epi64((const __m128i*)ptr); } |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 259 | void store(void* ptr) const { _mm_storel_epi64((__m128i*)ptr, fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 260 | |
Mike Klein | 33cbfd7 | 2016-10-06 11:09:27 -0400 | [diff] [blame] | 261 | static void Load4(const void* ptr, SkNx* r, SkNx* g, SkNx* b, SkNx* a) { |
| 262 | __m128i lo = _mm_loadu_si128(((__m128i*)ptr) + 0), |
| 263 | hi = _mm_loadu_si128(((__m128i*)ptr) + 1); |
| 264 | __m128i even = _mm_unpacklo_epi16(lo, hi), // r0 r2 g0 g2 b0 b2 a0 a2 |
| 265 | odd = _mm_unpackhi_epi16(lo, hi); // r1 r3 ... |
| 266 | __m128i rg = _mm_unpacklo_epi16(even, odd), // r0 r1 r2 r3 g0 g1 g2 g3 |
| 267 | ba = _mm_unpackhi_epi16(even, odd); // b0 b1 ... a0 a1 ... |
| 268 | *r = rg; |
| 269 | *g = _mm_srli_si128(rg, 8); |
| 270 | *b = ba; |
| 271 | *a = _mm_srli_si128(ba, 8); |
| 272 | } |
| 273 | static void Store4(void* dst, const SkNx& r, const SkNx& g, const SkNx& b, const SkNx& a) { |
| 274 | __m128i rg = _mm_unpacklo_epi16(r.fVec, g.fVec); |
| 275 | __m128i ba = _mm_unpacklo_epi16(b.fVec, a.fVec); |
| 276 | __m128i lo = _mm_unpacklo_epi32(rg, ba); |
| 277 | __m128i hi = _mm_unpackhi_epi32(rg, ba); |
| 278 | _mm_storeu_si128(((__m128i*) dst) + 0, lo); |
| 279 | _mm_storeu_si128(((__m128i*) dst) + 1, hi); |
| 280 | } |
| 281 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 282 | SkNx operator + (const SkNx& o) const { return _mm_add_epi16(fVec, o.fVec); } |
| 283 | SkNx operator - (const SkNx& o) const { return _mm_sub_epi16(fVec, o.fVec); } |
| 284 | SkNx operator * (const SkNx& o) const { return _mm_mullo_epi16(fVec, o.fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 285 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 286 | SkNx operator << (int bits) const { return _mm_slli_epi16(fVec, bits); } |
| 287 | SkNx operator >> (int bits) const { return _mm_srli_epi16(fVec, bits); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 288 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 289 | uint16_t operator[](int k) const { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 290 | SkASSERT(0 <= k && k < 4); |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 291 | union { __m128i v; uint16_t us[8]; } pun = {fVec}; |
| 292 | return pun.us[k&3]; |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 293 | } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 294 | |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 295 | __m128i fVec; |
| 296 | }; |
| 297 | |
| 298 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 299 | class SkNx<8, uint16_t> { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 300 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 301 | SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 302 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 303 | SkNx() {} |
| 304 | SkNx(uint16_t val) : fVec(_mm_set1_epi16(val)) {} |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 305 | SkNx(uint16_t a, uint16_t b, uint16_t c, uint16_t d, |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 306 | uint16_t e, uint16_t f, uint16_t g, uint16_t h) : fVec(_mm_setr_epi16(a,b,c,d,e,f,g,h)) {} |
| 307 | |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 308 | static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 309 | void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 310 | |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 311 | static void Load4(const void* ptr, SkNx* r, SkNx* g, SkNx* b, SkNx* a) { |
| 312 | // TODO: AVX2 version |
| 313 | __m128i _01 = _mm_loadu_si128(((__m128i*)ptr) + 0), |
| 314 | _23 = _mm_loadu_si128(((__m128i*)ptr) + 1), |
| 315 | _45 = _mm_loadu_si128(((__m128i*)ptr) + 2), |
| 316 | _67 = _mm_loadu_si128(((__m128i*)ptr) + 3); |
| 317 | |
| 318 | __m128i _02 = _mm_unpacklo_epi16(_01, _23), // r0 r2 g0 g2 b0 b2 a0 a2 |
| 319 | _13 = _mm_unpackhi_epi16(_01, _23), // r1 r3 g1 g3 b1 b3 a1 a3 |
| 320 | _46 = _mm_unpacklo_epi16(_45, _67), |
| 321 | _57 = _mm_unpackhi_epi16(_45, _67); |
| 322 | |
| 323 | __m128i rg0123 = _mm_unpacklo_epi16(_02, _13), // r0 r1 r2 r3 g0 g1 g2 g3 |
| 324 | ba0123 = _mm_unpackhi_epi16(_02, _13), // b0 b1 b2 b3 a0 a1 a2 a3 |
| 325 | rg4567 = _mm_unpacklo_epi16(_46, _57), |
| 326 | ba4567 = _mm_unpackhi_epi16(_46, _57); |
| 327 | |
| 328 | *r = _mm_unpacklo_epi64(rg0123, rg4567); |
| 329 | *g = _mm_unpackhi_epi64(rg0123, rg4567); |
| 330 | *b = _mm_unpacklo_epi64(ba0123, ba4567); |
| 331 | *a = _mm_unpackhi_epi64(ba0123, ba4567); |
| 332 | } |
| 333 | static void Store4(void* ptr, const SkNx& r, const SkNx& g, const SkNx& b, const SkNx& a) { |
| 334 | // TODO: AVX2 version |
| 335 | __m128i rg0123 = _mm_unpacklo_epi16(r.fVec, g.fVec), // r0 g0 r1 g1 r2 g2 r3 g3 |
| 336 | rg4567 = _mm_unpackhi_epi16(r.fVec, g.fVec), // r4 g4 r5 g5 r6 g6 r7 g7 |
| 337 | ba0123 = _mm_unpacklo_epi16(b.fVec, a.fVec), |
| 338 | ba4567 = _mm_unpackhi_epi16(b.fVec, a.fVec); |
| 339 | |
| 340 | _mm_storeu_si128((__m128i*)ptr + 0, _mm_unpacklo_epi32(rg0123, ba0123)); |
| 341 | _mm_storeu_si128((__m128i*)ptr + 1, _mm_unpackhi_epi32(rg0123, ba0123)); |
| 342 | _mm_storeu_si128((__m128i*)ptr + 2, _mm_unpacklo_epi32(rg4567, ba4567)); |
| 343 | _mm_storeu_si128((__m128i*)ptr + 3, _mm_unpackhi_epi32(rg4567, ba4567)); |
| 344 | } |
| 345 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 346 | SkNx operator + (const SkNx& o) const { return _mm_add_epi16(fVec, o.fVec); } |
| 347 | SkNx operator - (const SkNx& o) const { return _mm_sub_epi16(fVec, o.fVec); } |
| 348 | SkNx operator * (const SkNx& o) const { return _mm_mullo_epi16(fVec, o.fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 349 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 350 | SkNx operator << (int bits) const { return _mm_slli_epi16(fVec, bits); } |
| 351 | SkNx operator >> (int bits) const { return _mm_srli_epi16(fVec, bits); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 352 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 353 | static SkNx Min(const SkNx& a, const SkNx& b) { |
mtklein | 27e517a | 2015-05-14 17:53:04 -0700 | [diff] [blame] | 354 | // No unsigned _mm_min_epu16, so we'll shift into a space where we can use the |
| 355 | // signed version, _mm_min_epi16, then shift back. |
| 356 | const uint16_t top = 0x8000; // Keep this separate from _mm_set1_epi16 or MSVC will whine. |
| 357 | const __m128i top_8x = _mm_set1_epi16(top); |
| 358 | return _mm_add_epi8(top_8x, _mm_min_epi16(_mm_sub_epi8(a.fVec, top_8x), |
| 359 | _mm_sub_epi8(b.fVec, top_8x))); |
| 360 | } |
| 361 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 362 | SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | 4be181e | 2015-07-14 10:54:19 -0700 | [diff] [blame] | 363 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 364 | _mm_andnot_si128(fVec, e.fVec)); |
| 365 | } |
| 366 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 367 | uint16_t operator[](int k) const { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 368 | SkASSERT(0 <= k && k < 8); |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 369 | union { __m128i v; uint16_t us[8]; } pun = {fVec}; |
| 370 | return pun.us[k&7]; |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 371 | } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 372 | |
| 373 | __m128i fVec; |
| 374 | }; |
| 375 | |
| 376 | template <> |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 377 | class SkNx<4, uint8_t> { |
| 378 | public: |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 379 | SkNx() {} |
herb | fd5a260 | 2016-03-01 07:01:23 -0800 | [diff] [blame] | 380 | SkNx(const __m128i& vec) : fVec(vec) {} |
| 381 | SkNx(uint8_t a, uint8_t b, uint8_t c, uint8_t d) |
| 382 | : fVec(_mm_setr_epi8(a,b,c,d, 0,0,0,0, 0,0,0,0, 0,0,0,0)) {} |
| 383 | |
| 384 | |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 385 | static SkNx Load(const void* ptr) { return _mm_cvtsi32_si128(*(const int*)ptr); } |
| 386 | void store(void* ptr) const { *(int*)ptr = _mm_cvtsi128_si32(fVec); } |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 387 | |
herb | fd5a260 | 2016-03-01 07:01:23 -0800 | [diff] [blame] | 388 | uint8_t operator[](int k) const { |
| 389 | SkASSERT(0 <= k && k < 4); |
| 390 | union { __m128i v; uint8_t us[16]; } pun = {fVec}; |
| 391 | return pun.us[k&3]; |
| 392 | } |
| 393 | |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 394 | // TODO as needed |
| 395 | |
| 396 | __m128i fVec; |
| 397 | }; |
| 398 | |
| 399 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 400 | class SkNx<16, uint8_t> { |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 401 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 402 | SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 403 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 404 | SkNx() {} |
| 405 | SkNx(uint8_t val) : fVec(_mm_set1_epi8(val)) {} |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 406 | static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 407 | SkNx(uint8_t a, uint8_t b, uint8_t c, uint8_t d, |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 408 | uint8_t e, uint8_t f, uint8_t g, uint8_t h, |
| 409 | uint8_t i, uint8_t j, uint8_t k, uint8_t l, |
| 410 | uint8_t m, uint8_t n, uint8_t o, uint8_t p) |
| 411 | : fVec(_mm_setr_epi8(a,b,c,d, e,f,g,h, i,j,k,l, m,n,o,p)) {} |
| 412 | |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 413 | void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 414 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 415 | SkNx saturatedAdd(const SkNx& o) const { return _mm_adds_epu8(fVec, o.fVec); } |
mtklein | 6cbf18c | 2015-05-12 15:48:09 -0700 | [diff] [blame] | 416 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 417 | SkNx operator + (const SkNx& o) const { return _mm_add_epi8(fVec, o.fVec); } |
| 418 | SkNx operator - (const SkNx& o) const { return _mm_sub_epi8(fVec, o.fVec); } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 419 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 420 | static SkNx Min(const SkNx& a, const SkNx& b) { return _mm_min_epu8(a.fVec, b.fVec); } |
| 421 | SkNx operator < (const SkNx& o) const { |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 422 | // There's no unsigned _mm_cmplt_epu8, so we flip the sign bits then use a signed compare. |
| 423 | auto flip = _mm_set1_epi8(char(0x80)); |
| 424 | return _mm_cmplt_epi8(_mm_xor_si128(flip, fVec), _mm_xor_si128(flip, o.fVec)); |
| 425 | } |
mtklein | 27e517a | 2015-05-14 17:53:04 -0700 | [diff] [blame] | 426 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 427 | uint8_t operator[](int k) const { |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 428 | SkASSERT(0 <= k && k < 16); |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 429 | union { __m128i v; uint8_t us[16]; } pun = {fVec}; |
| 430 | return pun.us[k&15]; |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 431 | } |
| 432 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 433 | SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 434 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 435 | _mm_andnot_si128(fVec, e.fVec)); |
| 436 | } |
| 437 | |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 438 | __m128i fVec; |
| 439 | }; |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 440 | |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 441 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_AVX2 |
| 442 | |
| 443 | // There are two different SkNx<8, uint8_t>, SkNx<8, int32_t>, SkNx<8, uint32_t>, SkNx<8, float>: |
| 444 | // - the default paired SkNx<4, ...> versions used without AVX2 |
| 445 | // - the native AVX2 versions. |
| 446 | // It is important that we don't call methods for one from the other. |
| 447 | // Usually these methods inline, but they don't always in Debug builds. |
| 448 | // For now, try to fix this by marking all the AVX2 versions as always-inline. |
| 449 | // We may want or need to extend this strategy to all SkNx methods. |
| 450 | #define I SK_ALWAYS_INLINE |
| 451 | |
| 452 | template <> |
| 453 | class SkNx<8, uint8_t> { |
| 454 | public: |
| 455 | I SkNx(const __m128i& vec) : fVec(vec) {} |
| 456 | |
| 457 | I SkNx() {} |
| 458 | I SkNx(uint8_t v) : fVec(_mm_set1_epi8(v)) {} |
| 459 | I SkNx(uint8_t a, uint8_t b, uint8_t c, uint8_t d, |
| 460 | uint8_t e, uint8_t f, uint8_t g, uint8_t h) |
| 461 | : fVec(_mm_setr_epi8(a,b,c,d, e,f,g,h, 0,0,0,0, 0,0,0,0)) {} |
| 462 | |
| 463 | |
| 464 | I static SkNx Load(const void* ptr) { return _mm_loadl_epi64((const __m128i*)ptr); } |
| 465 | I void store(void* ptr) const { _mm_storel_epi64((__m128i*)ptr, fVec); } |
| 466 | |
| 467 | I uint8_t operator[](int k) const { |
| 468 | SkASSERT(0 <= k && k < 8); |
| 469 | union { __m128i v; uint8_t us[16]; } pun = {fVec}; |
| 470 | return pun.us[k&7]; |
| 471 | } |
| 472 | |
| 473 | __m128i fVec; |
| 474 | }; |
| 475 | |
| 476 | template <> |
| 477 | class SkNx<8, int32_t> { |
| 478 | public: |
| 479 | I SkNx(const __m256i& vec) : fVec(vec) {} |
| 480 | |
| 481 | I SkNx() {} |
| 482 | I SkNx(int32_t v) : fVec(_mm256_set1_epi32(v)) {} |
| 483 | I SkNx(int32_t a, int32_t b, int32_t c, int32_t d, |
| 484 | int32_t e, int32_t f, int32_t g, int32_t h) |
| 485 | : fVec(_mm256_setr_epi32(a,b,c,d, e,f,g,h)) {} |
| 486 | |
| 487 | I static SkNx Load(const void* ptr) { return _mm256_loadu_si256((const __m256i*)ptr); } |
| 488 | I void store(void* ptr) const { _mm256_storeu_si256((__m256i*)ptr, fVec); } |
| 489 | |
| 490 | I SkNx operator + (const SkNx& o) const { return _mm256_add_epi32(fVec, o.fVec); } |
| 491 | I SkNx operator - (const SkNx& o) const { return _mm256_sub_epi32(fVec, o.fVec); } |
| 492 | I SkNx operator * (const SkNx& o) const { return _mm256_mullo_epi32(fVec, o.fVec); } |
| 493 | |
| 494 | I SkNx operator & (const SkNx& o) const { return _mm256_and_si256(fVec, o.fVec); } |
| 495 | I SkNx operator | (const SkNx& o) const { return _mm256_or_si256(fVec, o.fVec); } |
| 496 | I SkNx operator ^ (const SkNx& o) const { return _mm256_xor_si256(fVec, o.fVec); } |
| 497 | |
| 498 | I SkNx operator << (int bits) const { return _mm256_slli_epi32(fVec, bits); } |
| 499 | I SkNx operator >> (int bits) const { return _mm256_srai_epi32(fVec, bits); } |
| 500 | |
| 501 | I int32_t operator[](int k) const { |
| 502 | SkASSERT(0 <= k && k < 8); |
| 503 | union { __m256i v; int32_t is[8]; } pun = {fVec}; |
| 504 | return pun.is[k&7]; |
| 505 | } |
| 506 | |
| 507 | __m256i fVec; |
| 508 | }; |
| 509 | |
| 510 | template <> |
| 511 | class SkNx<8, uint32_t> { |
| 512 | public: |
| 513 | I SkNx(const __m256i& vec) : fVec(vec) {} |
| 514 | |
| 515 | I SkNx() {} |
| 516 | I SkNx(uint32_t v) : fVec(_mm256_set1_epi32(v)) {} |
| 517 | I SkNx(uint32_t a, uint32_t b, uint32_t c, uint32_t d, |
| 518 | uint32_t e, uint32_t f, uint32_t g, uint32_t h) |
| 519 | : fVec(_mm256_setr_epi32(a,b,c,d, e,f,g,h)) {} |
| 520 | |
| 521 | I static SkNx Load(const void* ptr) { return _mm256_loadu_si256((const __m256i*)ptr); } |
| 522 | I void store(void* ptr) const { _mm256_storeu_si256((__m256i*)ptr, fVec); } |
| 523 | |
| 524 | I SkNx operator + (const SkNx& o) const { return _mm256_add_epi32(fVec, o.fVec); } |
| 525 | I SkNx operator - (const SkNx& o) const { return _mm256_sub_epi32(fVec, o.fVec); } |
| 526 | I SkNx operator * (const SkNx& o) const { return _mm256_mullo_epi32(fVec, o.fVec); } |
| 527 | |
| 528 | I SkNx operator & (const SkNx& o) const { return _mm256_and_si256(fVec, o.fVec); } |
| 529 | I SkNx operator | (const SkNx& o) const { return _mm256_or_si256(fVec, o.fVec); } |
| 530 | I SkNx operator ^ (const SkNx& o) const { return _mm256_xor_si256(fVec, o.fVec); } |
| 531 | |
| 532 | I SkNx operator << (int bits) const { return _mm256_slli_epi32(fVec, bits); } |
| 533 | I SkNx operator >> (int bits) const { return _mm256_srli_epi32(fVec, bits); } |
| 534 | |
| 535 | I uint32_t operator[](int k) const { |
| 536 | SkASSERT(0 <= k && k < 8); |
| 537 | union { __m256i v; uint32_t us[8]; } pun = {fVec}; |
| 538 | return pun.us[k&7]; |
| 539 | } |
| 540 | |
| 541 | __m256i fVec; |
| 542 | }; |
| 543 | |
| 544 | template <> |
| 545 | class SkNx<8, float> { |
| 546 | public: |
| 547 | I SkNx(const __m256& vec) : fVec(vec) {} |
| 548 | |
| 549 | I SkNx() {} |
| 550 | I SkNx(float val) : fVec(_mm256_set1_ps(val)) {} |
| 551 | I SkNx(float a, float b, float c, float d, |
| 552 | float e, float f, float g, float h) : fVec(_mm256_setr_ps(a,b,c,d,e,f,g,h)) {} |
| 553 | |
| 554 | I static SkNx Load(const void* ptr) { return _mm256_loadu_ps((const float*)ptr); } |
| 555 | I void store(void* ptr) const { _mm256_storeu_ps((float*)ptr, fVec); } |
| 556 | |
| 557 | I SkNx operator+(const SkNx& o) const { return _mm256_add_ps(fVec, o.fVec); } |
| 558 | I SkNx operator-(const SkNx& o) const { return _mm256_sub_ps(fVec, o.fVec); } |
| 559 | I SkNx operator*(const SkNx& o) const { return _mm256_mul_ps(fVec, o.fVec); } |
| 560 | I SkNx operator/(const SkNx& o) const { return _mm256_div_ps(fVec, o.fVec); } |
| 561 | |
| 562 | I SkNx operator==(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_EQ_OQ); } |
| 563 | I SkNx operator!=(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_NEQ_OQ); } |
| 564 | I SkNx operator <(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_LT_OQ); } |
| 565 | I SkNx operator >(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_GT_OQ); } |
| 566 | I SkNx operator<=(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_LE_OQ); } |
| 567 | I SkNx operator>=(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_GE_OQ); } |
| 568 | |
| 569 | I static SkNx Min(const SkNx& l, const SkNx& r) { return _mm256_min_ps(l.fVec, r.fVec); } |
| 570 | I static SkNx Max(const SkNx& l, const SkNx& r) { return _mm256_max_ps(l.fVec, r.fVec); } |
| 571 | |
| 572 | I SkNx sqrt() const { return _mm256_sqrt_ps (fVec); } |
| 573 | I SkNx rsqrt() const { return _mm256_rsqrt_ps(fVec); } |
| 574 | I SkNx invert() const { return _mm256_rcp_ps (fVec); } |
| 575 | |
| 576 | I float operator[](int k) const { |
| 577 | SkASSERT(0 <= k && k < 8); |
| 578 | union { __m256 v; float fs[8]; } pun = {fVec}; |
| 579 | return pun.fs[k&7]; |
| 580 | } |
| 581 | |
| 582 | I SkNx thenElse(const SkNx& t, const SkNx& e) const { |
| 583 | return _mm256_blendv_ps(e.fVec, t.fVec, fVec); |
| 584 | } |
| 585 | |
| 586 | __m256 fVec; |
| 587 | }; |
| 588 | |
| 589 | static I void SkNx_split(const Sk8f& v, Sk4f* lo, Sk4f* hi) { |
| 590 | *lo = _mm256_extractf128_ps(v.fVec, 0); |
| 591 | *hi = _mm256_extractf128_ps(v.fVec, 1); |
| 592 | } |
| 593 | |
| 594 | static I Sk8f SkNx_join(const Sk4f& lo, const Sk4f& hi) { |
| 595 | return _mm256_insertf128_ps(_mm256_castps128_ps256(lo.fVec), hi.fVec, 1); |
| 596 | } |
| 597 | |
| 598 | static I Sk8f SkNx_fma(const Sk8f& a, const Sk8f& b, const Sk8f& c) { |
| 599 | return _mm256_fmadd_ps(a.fVec, b.fVec, c.fVec); |
| 600 | } |
| 601 | |
| 602 | template<> /*static*/ I Sk8f SkNx_cast<float>(const Sk8b& src) { |
| 603 | return _mm256_cvtepi32_ps(_mm256_cvtepu8_epi32(src.fVec)); |
| 604 | } |
| 605 | |
| 606 | template<> /*static*/ I Sk8f SkNx_cast<float>(const Sk8i& src) { |
| 607 | return _mm256_cvtepi32_ps(src.fVec); |
| 608 | } |
| 609 | |
| 610 | template<> /*static*/ I Sk8i SkNx_cast<int>(const Sk8f& src) { |
| 611 | return _mm256_cvttps_epi32(src.fVec); |
| 612 | } |
| 613 | |
| 614 | template<> /*static*/ I Sk8i SkNx_cast<int>(const Sk8h& src) { |
| 615 | return _mm256_cvtepu16_epi32(src.fVec); |
| 616 | } |
| 617 | template<> /*static*/ I Sk8h SkNx_cast<uint16_t>(const Sk8i& src) { |
| 618 | __m128i lo = _mm256_extractf128_si256(src.fVec, 0), |
| 619 | hi = _mm256_extractf128_si256(src.fVec, 1); |
| 620 | return _mm_packus_epi32(lo, hi); |
| 621 | } |
| 622 | |
| 623 | #undef I |
| 624 | |
| 625 | #endif |
| 626 | |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 627 | template<> /*static*/ inline Sk4f SkNx_cast<float, int32_t>(const Sk4i& src) { |
mtklein | 0cf795f | 2016-02-17 07:23:36 -0800 | [diff] [blame] | 628 | return _mm_cvtepi32_ps(src.fVec); |
| 629 | } |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 630 | template<> /*static*/ inline Sk4f SkNx_cast<float, uint32_t>(const Sk4u& src) { |
| 631 | return SkNx_cast<float>(Sk4i::Load(&src)); |
| 632 | } |
mtklein | 0cf795f | 2016-02-17 07:23:36 -0800 | [diff] [blame] | 633 | |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 634 | template <> /*static*/ inline Sk4i SkNx_cast<int32_t, float>(const Sk4f& src) { |
mtklein | 0cf795f | 2016-02-17 07:23:36 -0800 | [diff] [blame] | 635 | return _mm_cvttps_epi32(src.fVec); |
| 636 | } |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 637 | |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 638 | template<> /*static*/ inline Sk4h SkNx_cast<uint16_t, int32_t>(const Sk4i& src) { |
mtklein | 036e183 | 2016-07-15 07:45:53 -0700 | [diff] [blame] | 639 | #if 0 && SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
| 640 | // TODO: This seems to be causing code generation problems. Investigate? |
| 641 | return _mm_packus_epi32(src.fVec); |
| 642 | #elif SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3 |
mtklein | 629f25a | 2016-02-08 05:54:38 -0800 | [diff] [blame] | 643 | // With SSSE3, we can just shuffle the low 2 bytes from each lane right into place. |
| 644 | const int _ = ~0; |
mtklein | 036e183 | 2016-07-15 07:45:53 -0700 | [diff] [blame] | 645 | return _mm_shuffle_epi8(src.fVec, _mm_setr_epi8(0,1, 4,5, 8,9, 12,13, _,_,_,_,_,_,_,_)); |
mtklein | 629f25a | 2016-02-08 05:54:38 -0800 | [diff] [blame] | 646 | #else |
mtklein | 036e183 | 2016-07-15 07:45:53 -0700 | [diff] [blame] | 647 | // With SSE2, we have to sign extend our input, making _mm_packs_epi32 do the pack we want. |
| 648 | __m128i x = _mm_srai_epi32(_mm_slli_epi32(src.fVec, 16), 16); |
| 649 | return _mm_packs_epi32(x,x); |
mtklein | 629f25a | 2016-02-08 05:54:38 -0800 | [diff] [blame] | 650 | #endif |
mtklein | 2d340f2 | 2016-02-06 19:38:39 -0800 | [diff] [blame] | 651 | } |
| 652 | |
mtklein | 036e183 | 2016-07-15 07:45:53 -0700 | [diff] [blame] | 653 | template<> /*static*/ inline Sk4h SkNx_cast<uint16_t, float>(const Sk4f& src) { |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 654 | return SkNx_cast<uint16_t>(SkNx_cast<int32_t>(src)); |
mtklein | 036e183 | 2016-07-15 07:45:53 -0700 | [diff] [blame] | 655 | } |
| 656 | |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 657 | template<> /*static*/ inline Sk4b SkNx_cast<uint8_t, float>(const Sk4f& src) { |
| 658 | auto _32 = _mm_cvttps_epi32(src.fVec); |
| 659 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3 |
| 660 | const int _ = ~0; |
| 661 | return _mm_shuffle_epi8(_32, _mm_setr_epi8(0,4,8,12, _,_,_,_, _,_,_,_, _,_,_,_)); |
| 662 | #else |
| 663 | auto _16 = _mm_packus_epi16(_32, _32); |
| 664 | return _mm_packus_epi16(_16, _16); |
| 665 | #endif |
mtklein | c1eb311 | 2016-02-11 08:10:22 -0800 | [diff] [blame] | 666 | } |
| 667 | |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 668 | template<> /*static*/ inline Sk4f SkNx_cast<float, uint8_t>(const Sk4b& src) { |
| 669 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3 |
| 670 | const int _ = ~0; |
| 671 | auto _32 = _mm_shuffle_epi8(src.fVec, _mm_setr_epi8(0,_,_,_, 1,_,_,_, 2,_,_,_, 3,_,_,_)); |
| 672 | #else |
| 673 | auto _16 = _mm_unpacklo_epi8(src.fVec, _mm_setzero_si128()), |
| 674 | _32 = _mm_unpacklo_epi16(_16, _mm_setzero_si128()); |
| 675 | #endif |
| 676 | return _mm_cvtepi32_ps(_32); |
mtklein | c1eb311 | 2016-02-11 08:10:22 -0800 | [diff] [blame] | 677 | } |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 678 | |
| 679 | template<> /*static*/ inline Sk4f SkNx_cast<float, uint16_t>(const Sk4h& src) { |
| 680 | auto _32 = _mm_unpacklo_epi16(src.fVec, _mm_setzero_si128()); |
| 681 | return _mm_cvtepi32_ps(_32); |
mtklein | 2d340f2 | 2016-02-06 19:38:39 -0800 | [diff] [blame] | 682 | } |
| 683 | |
mtklein | f8f90e4 | 2016-03-21 10:04:46 -0700 | [diff] [blame] | 684 | template<> /*static*/ inline Sk16b SkNx_cast<uint8_t, float>(const Sk16f& src) { |
| 685 | Sk8f ab, cd; |
| 686 | SkNx_split(src, &ab, &cd); |
| 687 | |
| 688 | Sk4f a,b,c,d; |
| 689 | SkNx_split(ab, &a, &b); |
| 690 | SkNx_split(cd, &c, &d); |
| 691 | |
| 692 | return _mm_packus_epi16(_mm_packus_epi16(_mm_cvttps_epi32(a.fVec), |
| 693 | _mm_cvttps_epi32(b.fVec)), |
| 694 | _mm_packus_epi16(_mm_cvttps_epi32(c.fVec), |
| 695 | _mm_cvttps_epi32(d.fVec))); |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 696 | } |
| 697 | |
| 698 | template<> /*static*/ inline Sk4h SkNx_cast<uint16_t, uint8_t>(const Sk4b& src) { |
| 699 | return _mm_unpacklo_epi8(src.fVec, _mm_setzero_si128()); |
| 700 | } |
| 701 | |
| 702 | template<> /*static*/ inline Sk4b SkNx_cast<uint8_t, uint16_t>(const Sk4h& src) { |
| 703 | return _mm_packus_epi16(src.fVec, src.fVec); |
mtklein | 550e9b0 | 2016-01-20 11:55:51 -0800 | [diff] [blame] | 704 | } |
| 705 | |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 706 | template<> /*static*/ inline Sk4i SkNx_cast<int32_t, uint16_t>(const Sk4h& src) { |
mtklein | 58e389b | 2016-07-15 07:00:11 -0700 | [diff] [blame] | 707 | return _mm_unpacklo_epi16(src.fVec, _mm_setzero_si128()); |
| 708 | } |
| 709 | |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 710 | template<> /*static*/ inline Sk4b SkNx_cast<uint8_t, int32_t>(const Sk4i& src) { |
msarett | 7d3ff71 | 2016-07-12 14:55:45 -0700 | [diff] [blame] | 711 | return _mm_packus_epi16(_mm_packus_epi16(src.fVec, src.fVec), src.fVec); |
| 712 | } |
| 713 | |
msarett | 15ee3de | 2016-08-02 11:30:30 -0700 | [diff] [blame] | 714 | template<> /*static*/ inline Sk4i SkNx_cast<int32_t, uint32_t>(const Sk4u& src) { |
| 715 | return src.fVec; |
| 716 | } |
| 717 | |
msarett | 7d3ff71 | 2016-07-12 14:55:45 -0700 | [diff] [blame] | 718 | static inline Sk4i Sk4f_round(const Sk4f& x) { |
| 719 | return _mm_cvtps_epi32(x.fVec); |
| 720 | } |
| 721 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 722 | #endif//SkNx_sse_DEFINED |