smmintrin.h revision 8484375b0fc442c704b3ec06e2e795d88591606f
1/*===---- smmintrin.h - SSE4 intrinsics ------------------------------------=== 2 * 3 * Permission is hereby granted, free of charge, to any person obtaining a copy 4 * of this software and associated documentation files (the "Software"), to deal 5 * in the Software without restriction, including without limitation the rights 6 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7 * copies of the Software, and to permit persons to whom the Software is 8 * furnished to do so, subject to the following conditions: 9 * 10 * The above copyright notice and this permission notice shall be included in 11 * all copies or substantial portions of the Software. 12 * 13 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19 * THE SOFTWARE. 20 * 21 *===-----------------------------------------------------------------------=== 22 */ 23 24#ifndef _SMMINTRIN_H 25#define _SMMINTRIN_H 26 27#ifndef __SSE4_1__ 28#error "SSE4.1 instruction set not enabled" 29#else 30 31#include <tmmintrin.h> 32 33/* SSE4 Rounding macros. */ 34#define _MM_FROUND_TO_NEAREST_INT 0x00 35#define _MM_FROUND_TO_NEG_INF 0x01 36#define _MM_FROUND_TO_POS_INF 0x02 37#define _MM_FROUND_TO_ZERO 0x03 38#define _MM_FROUND_CUR_DIRECTION 0x04 39 40#define _MM_FROUND_RAISE_EXC 0x00 41#define _MM_FROUND_NO_EXC 0x08 42 43#define _MM_FROUND_NINT (_MM_FROUND_RAISE_EXC | _MM_FROUND_TO_NEAREST_INT) 44#define _MM_FROUND_FLOOR (_MM_FROUND_RAISE_EXC | _MM_FROUND_TO_NEG_INF) 45#define _MM_FROUND_CEIL (_MM_FROUND_RAISE_EXC | _MM_FROUND_TO_POS_INF) 46#define _MM_FROUND_TRUNC (_MM_FROUND_RAISE_EXC | _MM_FROUND_TO_ZERO) 47#define _MM_FROUND_RINT (_MM_FROUND_RAISE_EXC | _MM_FROUND_CUR_DIRECTION) 48#define _MM_FROUND_NEARBYINT (_MM_FROUND_NO_EXC | _MM_FROUND_CUR_DIRECTION) 49 50#define _mm_ceil_ps(X) _mm_round_ps((X), _MM_FROUND_CEIL) 51#define _mm_ceil_pd(X) _mm_round_pd((X), _MM_FROUND_CEIL) 52#define _mm_ceil_ss(X, Y) _mm_round_ss((X), (Y), _MM_FROUND_CEIL) 53#define _mm_ceil_sd(X, Y) _mm_round_sd((X), (Y), _MM_FROUND_CEIL) 54 55#define _mm_floor_ps(X) _mm_round_ps((X), _MM_FROUND_FLOOR) 56#define _mm_floor_pd(X) _mm_round_pd((X), _MM_FROUND_FLOOR) 57#define _mm_floor_ss(X, Y) _mm_round_ss((X), (Y), _MM_FROUND_FLOOR) 58#define _mm_floor_sd(X, Y) _mm_round_sd((X), (Y), _MM_FROUND_FLOOR) 59 60#define _mm_round_ps(X, M) __extension__ ({ \ 61 __m128 __X = (X); \ 62 (__m128) __builtin_ia32_roundps((__v4sf)__X, (M)); }) 63 64#define _mm_round_ss(X, Y, M) __extension__ ({ \ 65 __m128 __X = (X); \ 66 __m128 __Y = (Y); \ 67 (__m128) __builtin_ia32_roundss((__v4sf)__X, (__v4sf)__Y, (M)); }) 68 69#define _mm_round_pd(X, M) __extension__ ({ \ 70 __m128d __X = (X); \ 71 (__m128d) __builtin_ia32_roundpd((__v2df)__X, (M)); }) 72 73#define _mm_round_sd(X, Y, M) __extension__ ({ \ 74 __m128d __X = (X); \ 75 __m128d __Y = (Y); \ 76 (__m128d) __builtin_ia32_roundsd((__v2df)__X, (__v2df)__Y, (M)); }) 77 78/* SSE4 Packed Blending Intrinsics. */ 79#define _mm_blend_pd(V1, V2, M) __extension__ ({ \ 80 __m128d __V1 = (V1); \ 81 __m128d __V2 = (V2); \ 82 (__m128d) __builtin_ia32_blendpd ((__v2df)__V1, (__v2df)__V2, (M)); }) 83 84#define _mm_blend_ps(V1, V2, M) __extension__ ({ \ 85 __m128 __V1 = (V1); \ 86 __m128 __V2 = (V2); \ 87 (__m128) __builtin_ia32_blendps ((__v4sf)__V1, (__v4sf)__V2, (M)); }) 88 89static __inline__ __m128d __attribute__((__always_inline__, __nodebug__)) 90_mm_blendv_pd (__m128d __V1, __m128d __V2, __m128d __M) 91{ 92 return (__m128d) __builtin_ia32_blendvpd ((__v2df)__V1, (__v2df)__V2, 93 (__v2df)__M); 94} 95 96static __inline__ __m128 __attribute__((__always_inline__, __nodebug__)) 97_mm_blendv_ps (__m128 __V1, __m128 __V2, __m128 __M) 98{ 99 return (__m128) __builtin_ia32_blendvps ((__v4sf)__V1, (__v4sf)__V2, 100 (__v4sf)__M); 101} 102 103static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 104_mm_blendv_epi8 (__m128i __V1, __m128i __V2, __m128i __M) 105{ 106 return (__m128i) __builtin_ia32_pblendvb128 ((__v16qi)__V1, (__v16qi)__V2, 107 (__v16qi)__M); 108} 109 110#define _mm_blend_epi16(V1, V2, M) __extension__ ({ \ 111 __m128i __V1 = (V1); \ 112 __m128i __V2 = (V2); \ 113 (__m128i) __builtin_ia32_pblendw128 ((__v8hi)__V1, (__v8hi)__V2, (M)); }) 114 115/* SSE4 Dword Multiply Instructions. */ 116static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 117_mm_mullo_epi32 (__m128i __V1, __m128i __V2) 118{ 119 return (__m128i) ((__v4si)__V1 * (__v4si)__V2); 120} 121 122static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 123_mm_mul_epi32 (__m128i __V1, __m128i __V2) 124{ 125 return (__m128i) __builtin_ia32_pmuldq128 ((__v4si)__V1, (__v4si)__V2); 126} 127 128/* SSE4 Floating Point Dot Product Instructions. */ 129#define _mm_dp_ps(X, Y, M) __extension__ ({ \ 130 __m128 __X = (X); \ 131 __m128 __Y = (Y); \ 132 (__m128) __builtin_ia32_dpps((__v4sf)__X, (__v4sf)__Y, (M)); }) 133 134#define _mm_dp_pd(X, Y, M) __extension__ ({\ 135 __m128d __X = (X); \ 136 __m128d __Y = (Y); \ 137 (__m128d) __builtin_ia32_dppd((__v2df)__X, (__v2df)__Y, (M)); }) 138 139/* SSE4 Streaming Load Hint Instruction. */ 140static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 141_mm_stream_load_si128 (__m128i *__V) 142{ 143 return (__m128i) __builtin_ia32_movntdqa ((__v2di *) __V); 144} 145 146/* SSE4 Packed Integer Min/Max Instructions. */ 147static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 148_mm_min_epi8 (__m128i __V1, __m128i __V2) 149{ 150 return (__m128i) __builtin_ia32_pminsb128 ((__v16qi) __V1, (__v16qi) __V2); 151} 152 153static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 154_mm_max_epi8 (__m128i __V1, __m128i __V2) 155{ 156 return (__m128i) __builtin_ia32_pmaxsb128 ((__v16qi) __V1, (__v16qi) __V2); 157} 158 159static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 160_mm_min_epu16 (__m128i __V1, __m128i __V2) 161{ 162 return (__m128i) __builtin_ia32_pminuw128 ((__v8hi) __V1, (__v8hi) __V2); 163} 164 165static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 166_mm_max_epu16 (__m128i __V1, __m128i __V2) 167{ 168 return (__m128i) __builtin_ia32_pmaxuw128 ((__v8hi) __V1, (__v8hi) __V2); 169} 170 171static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 172_mm_min_epi32 (__m128i __V1, __m128i __V2) 173{ 174 return (__m128i) __builtin_ia32_pminsd128 ((__v4si) __V1, (__v4si) __V2); 175} 176 177static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 178_mm_max_epi32 (__m128i __V1, __m128i __V2) 179{ 180 return (__m128i) __builtin_ia32_pmaxsd128 ((__v4si) __V1, (__v4si) __V2); 181} 182 183static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 184_mm_min_epu32 (__m128i __V1, __m128i __V2) 185{ 186 return (__m128i) __builtin_ia32_pminud128((__v4si) __V1, (__v4si) __V2); 187} 188 189static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 190_mm_max_epu32 (__m128i __V1, __m128i __V2) 191{ 192 return (__m128i) __builtin_ia32_pmaxud128((__v4si) __V1, (__v4si) __V2); 193} 194 195/* SSE4 Insertion and Extraction from XMM Register Instructions. */ 196#define _mm_insert_ps(X, Y, N) __builtin_ia32_insertps128((X), (Y), (N)) 197#define _mm_extract_ps(X, N) (__extension__ \ 198 ({ union { int __i; float __f; } __t; \ 199 __v4sf __a = (__v4sf)(X); \ 200 __t.__f = __a[(N) & 3]; \ 201 __t.__i;})) 202 203/* Miscellaneous insert and extract macros. */ 204/* Extract a single-precision float from X at index N into D. */ 205#define _MM_EXTRACT_FLOAT(D, X, N) (__extension__ ({ __v4sf __a = (__v4sf)(X); \ 206 (D) = __a[N]; })) 207 208/* Or together 2 sets of indexes (X and Y) with the zeroing bits (Z) to create 209 an index suitable for _mm_insert_ps. */ 210#define _MM_MK_INSERTPS_NDX(X, Y, Z) (((X) << 6) | ((Y) << 4) | (Z)) 211 212/* Extract a float from X at index N into the first index of the return. */ 213#define _MM_PICK_OUT_PS(X, N) _mm_insert_ps (_mm_setzero_ps(), (X), \ 214 _MM_MK_INSERTPS_NDX((N), 0, 0x0e)) 215 216/* Insert int into packed integer array at index. */ 217#define _mm_insert_epi8(X, I, N) (__extension__ ({ __v16qi __a = (__v16qi)(X); \ 218 __a[(N) & 15] = (I); \ 219 __a;})) 220#define _mm_insert_epi32(X, I, N) (__extension__ ({ __v4si __a = (__v4si)(X); \ 221 __a[(N) & 3] = (I); \ 222 __a;})) 223#ifdef __x86_64__ 224#define _mm_insert_epi64(X, I, N) (__extension__ ({ __v2di __a = (__v2di)(X); \ 225 __a[(N) & 1] = (I); \ 226 __a;})) 227#endif /* __x86_64__ */ 228 229/* Extract int from packed integer array at index. This returns the element 230 * as a zero extended value, so it is unsigned. 231 */ 232#define _mm_extract_epi8(X, N) (__extension__ ({ __v16qi __a = (__v16qi)(X); \ 233 (int)(unsigned char) \ 234 __a[(N) & 15];})) 235#define _mm_extract_epi32(X, N) (__extension__ ({ __v4si __a = (__v4si)(X); \ 236 __a[(N) & 3];})) 237#ifdef __x86_64__ 238#define _mm_extract_epi64(X, N) (__extension__ ({ __v2di __a = (__v2di)(X); \ 239 __a[(N) & 1];})) 240#endif /* __x86_64 */ 241 242/* SSE4 128-bit Packed Integer Comparisons. */ 243static __inline__ int __attribute__((__always_inline__, __nodebug__)) 244_mm_testz_si128(__m128i __M, __m128i __V) 245{ 246 return __builtin_ia32_ptestz128((__v2di)__M, (__v2di)__V); 247} 248 249static __inline__ int __attribute__((__always_inline__, __nodebug__)) 250_mm_testc_si128(__m128i __M, __m128i __V) 251{ 252 return __builtin_ia32_ptestc128((__v2di)__M, (__v2di)__V); 253} 254 255static __inline__ int __attribute__((__always_inline__, __nodebug__)) 256_mm_testnzc_si128(__m128i __M, __m128i __V) 257{ 258 return __builtin_ia32_ptestnzc128((__v2di)__M, (__v2di)__V); 259} 260 261#define _mm_test_all_ones(V) _mm_testc_si128((V), _mm_cmpeq_epi32((V), (V))) 262#define _mm_test_mix_ones_zeros(M, V) _mm_testnzc_si128((M), (V)) 263#define _mm_test_all_zeros(M, V) _mm_testz_si128 ((M), (V)) 264 265/* SSE4 64-bit Packed Integer Comparisons. */ 266static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 267_mm_cmpeq_epi64(__m128i __V1, __m128i __V2) 268{ 269 return (__m128i)((__v2di)__V1 == (__v2di)__V2); 270} 271 272/* SSE4 Packed Integer Sign-Extension. */ 273static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 274_mm_cvtepi8_epi16(__m128i __V) 275{ 276 return (__m128i) __builtin_ia32_pmovsxbw128((__v16qi) __V); 277} 278 279static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 280_mm_cvtepi8_epi32(__m128i __V) 281{ 282 return (__m128i) __builtin_ia32_pmovsxbd128((__v16qi) __V); 283} 284 285static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 286_mm_cvtepi8_epi64(__m128i __V) 287{ 288 return (__m128i) __builtin_ia32_pmovsxbq128((__v16qi) __V); 289} 290 291static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 292_mm_cvtepi16_epi32(__m128i __V) 293{ 294 return (__m128i) __builtin_ia32_pmovsxwd128((__v8hi) __V); 295} 296 297static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 298_mm_cvtepi16_epi64(__m128i __V) 299{ 300 return (__m128i) __builtin_ia32_pmovsxwq128((__v8hi)__V); 301} 302 303static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 304_mm_cvtepi32_epi64(__m128i __V) 305{ 306 return (__m128i) __builtin_ia32_pmovsxdq128((__v4si)__V); 307} 308 309/* SSE4 Packed Integer Zero-Extension. */ 310static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 311_mm_cvtepu8_epi16(__m128i __V) 312{ 313 return (__m128i) __builtin_ia32_pmovzxbw128((__v16qi) __V); 314} 315 316static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 317_mm_cvtepu8_epi32(__m128i __V) 318{ 319 return (__m128i) __builtin_ia32_pmovzxbd128((__v16qi)__V); 320} 321 322static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 323_mm_cvtepu8_epi64(__m128i __V) 324{ 325 return (__m128i) __builtin_ia32_pmovzxbq128((__v16qi)__V); 326} 327 328static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 329_mm_cvtepu16_epi32(__m128i __V) 330{ 331 return (__m128i) __builtin_ia32_pmovzxwd128((__v8hi)__V); 332} 333 334static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 335_mm_cvtepu16_epi64(__m128i __V) 336{ 337 return (__m128i) __builtin_ia32_pmovzxwq128((__v8hi)__V); 338} 339 340static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 341_mm_cvtepu32_epi64(__m128i __V) 342{ 343 return (__m128i) __builtin_ia32_pmovzxdq128((__v4si)__V); 344} 345 346/* SSE4 Pack with Unsigned Saturation. */ 347static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 348_mm_packus_epi32(__m128i __V1, __m128i __V2) 349{ 350 return (__m128i) __builtin_ia32_packusdw128((__v4si)__V1, (__v4si)__V2); 351} 352 353/* SSE4 Multiple Packed Sums of Absolute Difference. */ 354#define _mm_mpsadbw_epu8(X, Y, M) __extension__ ({ \ 355 __m128i __X = (X); \ 356 __m128i __Y = (Y); \ 357 (__m128i) __builtin_ia32_mpsadbw128((__v16qi)__X, (__v16qi)__Y, (M)); }) 358 359static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 360_mm_minpos_epu16(__m128i __V) 361{ 362 return (__m128i) __builtin_ia32_phminposuw128((__v8hi)__V); 363} 364 365/* These definitions are normally in nmmintrin.h, but gcc puts them in here 366 so we'll do the same. */ 367#ifdef __SSE4_2__ 368 369/* These specify the type of data that we're comparing. */ 370#define _SIDD_UBYTE_OPS 0x00 371#define _SIDD_UWORD_OPS 0x01 372#define _SIDD_SBYTE_OPS 0x02 373#define _SIDD_SWORD_OPS 0x03 374 375/* These specify the type of comparison operation. */ 376#define _SIDD_CMP_EQUAL_ANY 0x00 377#define _SIDD_CMP_RANGES 0x04 378#define _SIDD_CMP_EQUAL_EACH 0x08 379#define _SIDD_CMP_EQUAL_ORDERED 0x0c 380 381/* These macros specify the polarity of the operation. */ 382#define _SIDD_POSITIVE_POLARITY 0x00 383#define _SIDD_NEGATIVE_POLARITY 0x10 384#define _SIDD_MASKED_POSITIVE_POLARITY 0x20 385#define _SIDD_MASKED_NEGATIVE_POLARITY 0x30 386 387/* These macros are used in _mm_cmpXstri() to specify the return. */ 388#define _SIDD_LEAST_SIGNIFICANT 0x00 389#define _SIDD_MOST_SIGNIFICANT 0x40 390 391/* These macros are used in _mm_cmpXstri() to specify the return. */ 392#define _SIDD_BIT_MASK 0x00 393#define _SIDD_UNIT_MASK 0x40 394 395/* SSE4.2 Packed Comparison Intrinsics. */ 396#define _mm_cmpistrm(A, B, M) __builtin_ia32_pcmpistrm128((A), (B), (M)) 397#define _mm_cmpistri(A, B, M) __builtin_ia32_pcmpistri128((A), (B), (M)) 398 399#define _mm_cmpestrm(A, LA, B, LB, M) \ 400 __builtin_ia32_pcmpestrm128((A), (LA), (B), (LB), (M)) 401#define _mm_cmpestri(A, LA, B, LB, M) \ 402 __builtin_ia32_pcmpestri128((A), (LA), (B), (LB), (M)) 403 404/* SSE4.2 Packed Comparison Intrinsics and EFlag Reading. */ 405#define _mm_cmpistra(A, B, M) \ 406 __builtin_ia32_pcmpistria128((A), (B), (M)) 407#define _mm_cmpistrc(A, B, M) \ 408 __builtin_ia32_pcmpistric128((A), (B), (M)) 409#define _mm_cmpistro(A, B, M) \ 410 __builtin_ia32_pcmpistrio128((A), (B), (M)) 411#define _mm_cmpistrs(A, B, M) \ 412 __builtin_ia32_pcmpistris128((A), (B), (M)) 413#define _mm_cmpistrz(A, B, M) \ 414 __builtin_ia32_pcmpistriz128((A), (B), (M)) 415 416#define _mm_cmpestra(A, LA, B, LB, M) \ 417 __builtin_ia32_pcmpestria128((A), (LA), (B), (LB), (M)) 418#define _mm_cmpestrc(A, LA, B, LB, M) \ 419 __builtin_ia32_pcmpestric128((A), (LA), (B), (LB), (M)) 420#define _mm_cmpestro(A, LA, B, LB, M) \ 421 __builtin_ia32_pcmpestrio128((A), (LA), (B), (LB), (M)) 422#define _mm_cmpestrs(A, LA, B, LB, M) \ 423 __builtin_ia32_pcmpestris128((A), (LA), (B), (LB), (M)) 424#define _mm_cmpestrz(A, LA, B, LB, M) \ 425 __builtin_ia32_pcmpestriz128((A), (LA), (B), (LB), (M)) 426 427/* SSE4.2 Compare Packed Data -- Greater Than. */ 428static __inline__ __m128i __attribute__((__always_inline__, __nodebug__)) 429_mm_cmpgt_epi64(__m128i __V1, __m128i __V2) 430{ 431 return (__m128i)((__v2di)__V1 > (__v2di)__V2); 432} 433 434/* SSE4.2 Accumulate CRC32. */ 435static __inline__ unsigned int __attribute__((__always_inline__, __nodebug__)) 436_mm_crc32_u8(unsigned int __C, unsigned char __D) 437{ 438 return __builtin_ia32_crc32qi(__C, __D); 439} 440 441static __inline__ unsigned int __attribute__((__always_inline__, __nodebug__)) 442_mm_crc32_u16(unsigned int __C, unsigned short __D) 443{ 444 return __builtin_ia32_crc32hi(__C, __D); 445} 446 447static __inline__ unsigned int __attribute__((__always_inline__, __nodebug__)) 448_mm_crc32_u32(unsigned int __C, unsigned int __D) 449{ 450 return __builtin_ia32_crc32si(__C, __D); 451} 452 453#ifdef __x86_64__ 454static __inline__ unsigned long long __attribute__((__always_inline__, __nodebug__)) 455_mm_crc32_u64(unsigned long long __C, unsigned long long __D) 456{ 457 return __builtin_ia32_crc32di(__C, __D); 458} 459#endif /* __x86_64__ */ 460 461#ifdef __POPCNT__ 462#include <popcntintrin.h> 463#endif 464 465#endif /* __SSE4_2__ */ 466#endif /* __SSE4_1__ */ 467 468#endif /* _SMMINTRIN_H */ 469