avx2intrin.h 39 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168
  1. /*===---- avx2intrin.h - AVX2 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. #ifndef __IMMINTRIN_H
  24. #error "Never use <avx2intrin.h> directly; include <immintrin.h> instead."
  25. #endif
  26. #ifndef __AVX2INTRIN_H
  27. #define __AVX2INTRIN_H
  28. /* Define the default attributes for the functions in this file. */
  29. #define __DEFAULT_FN_ATTRS256 __attribute__((__always_inline__, __nodebug__, __target__("avx2"), __min_vector_width__(256)))
  30. #define __DEFAULT_FN_ATTRS128 __attribute__((__always_inline__, __nodebug__, __target__("avx2"), __min_vector_width__(128)))
  31. /* SSE4 Multiple Packed Sums of Absolute Difference. */
  32. #define _mm256_mpsadbw_epu8(X, Y, M) \
  33. (__m256i)__builtin_ia32_mpsadbw256((__v32qi)(__m256i)(X), \
  34. (__v32qi)(__m256i)(Y), (int)(M))
  35. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  36. _mm256_abs_epi8(__m256i __a)
  37. {
  38. return (__m256i)__builtin_ia32_pabsb256((__v32qi)__a);
  39. }
  40. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  41. _mm256_abs_epi16(__m256i __a)
  42. {
  43. return (__m256i)__builtin_ia32_pabsw256((__v16hi)__a);
  44. }
  45. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  46. _mm256_abs_epi32(__m256i __a)
  47. {
  48. return (__m256i)__builtin_ia32_pabsd256((__v8si)__a);
  49. }
  50. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  51. _mm256_packs_epi16(__m256i __a, __m256i __b)
  52. {
  53. return (__m256i)__builtin_ia32_packsswb256((__v16hi)__a, (__v16hi)__b);
  54. }
  55. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  56. _mm256_packs_epi32(__m256i __a, __m256i __b)
  57. {
  58. return (__m256i)__builtin_ia32_packssdw256((__v8si)__a, (__v8si)__b);
  59. }
  60. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  61. _mm256_packus_epi16(__m256i __a, __m256i __b)
  62. {
  63. return (__m256i)__builtin_ia32_packuswb256((__v16hi)__a, (__v16hi)__b);
  64. }
  65. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  66. _mm256_packus_epi32(__m256i __V1, __m256i __V2)
  67. {
  68. return (__m256i) __builtin_ia32_packusdw256((__v8si)__V1, (__v8si)__V2);
  69. }
  70. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  71. _mm256_add_epi8(__m256i __a, __m256i __b)
  72. {
  73. return (__m256i)((__v32qu)__a + (__v32qu)__b);
  74. }
  75. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  76. _mm256_add_epi16(__m256i __a, __m256i __b)
  77. {
  78. return (__m256i)((__v16hu)__a + (__v16hu)__b);
  79. }
  80. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  81. _mm256_add_epi32(__m256i __a, __m256i __b)
  82. {
  83. return (__m256i)((__v8su)__a + (__v8su)__b);
  84. }
  85. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  86. _mm256_add_epi64(__m256i __a, __m256i __b)
  87. {
  88. return (__m256i)((__v4du)__a + (__v4du)__b);
  89. }
  90. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  91. _mm256_adds_epi8(__m256i __a, __m256i __b)
  92. {
  93. return (__m256i)__builtin_ia32_paddsb256((__v32qi)__a, (__v32qi)__b);
  94. }
  95. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  96. _mm256_adds_epi16(__m256i __a, __m256i __b)
  97. {
  98. return (__m256i)__builtin_ia32_paddsw256((__v16hi)__a, (__v16hi)__b);
  99. }
  100. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  101. _mm256_adds_epu8(__m256i __a, __m256i __b)
  102. {
  103. return (__m256i)__builtin_ia32_paddusb256((__v32qi)__a, (__v32qi)__b);
  104. }
  105. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  106. _mm256_adds_epu16(__m256i __a, __m256i __b)
  107. {
  108. return (__m256i)__builtin_ia32_paddusw256((__v16hi)__a, (__v16hi)__b);
  109. }
  110. #define _mm256_alignr_epi8(a, b, n) \
  111. (__m256i)__builtin_ia32_palignr256((__v32qi)(__m256i)(a), \
  112. (__v32qi)(__m256i)(b), (n))
  113. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  114. _mm256_and_si256(__m256i __a, __m256i __b)
  115. {
  116. return (__m256i)((__v4du)__a & (__v4du)__b);
  117. }
  118. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  119. _mm256_andnot_si256(__m256i __a, __m256i __b)
  120. {
  121. return (__m256i)(~(__v4du)__a & (__v4du)__b);
  122. }
  123. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  124. _mm256_avg_epu8(__m256i __a, __m256i __b)
  125. {
  126. typedef unsigned short __v32hu __attribute__((__vector_size__(64)));
  127. return (__m256i)__builtin_convertvector(
  128. ((__builtin_convertvector((__v32qu)__a, __v32hu) +
  129. __builtin_convertvector((__v32qu)__b, __v32hu)) + 1)
  130. >> 1, __v32qu);
  131. }
  132. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  133. _mm256_avg_epu16(__m256i __a, __m256i __b)
  134. {
  135. typedef unsigned int __v16su __attribute__((__vector_size__(64)));
  136. return (__m256i)__builtin_convertvector(
  137. ((__builtin_convertvector((__v16hu)__a, __v16su) +
  138. __builtin_convertvector((__v16hu)__b, __v16su)) + 1)
  139. >> 1, __v16hu);
  140. }
  141. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  142. _mm256_blendv_epi8(__m256i __V1, __m256i __V2, __m256i __M)
  143. {
  144. return (__m256i)__builtin_ia32_pblendvb256((__v32qi)__V1, (__v32qi)__V2,
  145. (__v32qi)__M);
  146. }
  147. #define _mm256_blend_epi16(V1, V2, M) \
  148. (__m256i)__builtin_ia32_pblendw256((__v16hi)(__m256i)(V1), \
  149. (__v16hi)(__m256i)(V2), (int)(M))
  150. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  151. _mm256_cmpeq_epi8(__m256i __a, __m256i __b)
  152. {
  153. return (__m256i)((__v32qi)__a == (__v32qi)__b);
  154. }
  155. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  156. _mm256_cmpeq_epi16(__m256i __a, __m256i __b)
  157. {
  158. return (__m256i)((__v16hi)__a == (__v16hi)__b);
  159. }
  160. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  161. _mm256_cmpeq_epi32(__m256i __a, __m256i __b)
  162. {
  163. return (__m256i)((__v8si)__a == (__v8si)__b);
  164. }
  165. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  166. _mm256_cmpeq_epi64(__m256i __a, __m256i __b)
  167. {
  168. return (__m256i)((__v4di)__a == (__v4di)__b);
  169. }
  170. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  171. _mm256_cmpgt_epi8(__m256i __a, __m256i __b)
  172. {
  173. /* This function always performs a signed comparison, but __v32qi is a char
  174. which may be signed or unsigned, so use __v32qs. */
  175. return (__m256i)((__v32qs)__a > (__v32qs)__b);
  176. }
  177. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  178. _mm256_cmpgt_epi16(__m256i __a, __m256i __b)
  179. {
  180. return (__m256i)((__v16hi)__a > (__v16hi)__b);
  181. }
  182. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  183. _mm256_cmpgt_epi32(__m256i __a, __m256i __b)
  184. {
  185. return (__m256i)((__v8si)__a > (__v8si)__b);
  186. }
  187. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  188. _mm256_cmpgt_epi64(__m256i __a, __m256i __b)
  189. {
  190. return (__m256i)((__v4di)__a > (__v4di)__b);
  191. }
  192. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  193. _mm256_hadd_epi16(__m256i __a, __m256i __b)
  194. {
  195. return (__m256i)__builtin_ia32_phaddw256((__v16hi)__a, (__v16hi)__b);
  196. }
  197. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  198. _mm256_hadd_epi32(__m256i __a, __m256i __b)
  199. {
  200. return (__m256i)__builtin_ia32_phaddd256((__v8si)__a, (__v8si)__b);
  201. }
  202. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  203. _mm256_hadds_epi16(__m256i __a, __m256i __b)
  204. {
  205. return (__m256i)__builtin_ia32_phaddsw256((__v16hi)__a, (__v16hi)__b);
  206. }
  207. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  208. _mm256_hsub_epi16(__m256i __a, __m256i __b)
  209. {
  210. return (__m256i)__builtin_ia32_phsubw256((__v16hi)__a, (__v16hi)__b);
  211. }
  212. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  213. _mm256_hsub_epi32(__m256i __a, __m256i __b)
  214. {
  215. return (__m256i)__builtin_ia32_phsubd256((__v8si)__a, (__v8si)__b);
  216. }
  217. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  218. _mm256_hsubs_epi16(__m256i __a, __m256i __b)
  219. {
  220. return (__m256i)__builtin_ia32_phsubsw256((__v16hi)__a, (__v16hi)__b);
  221. }
  222. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  223. _mm256_maddubs_epi16(__m256i __a, __m256i __b)
  224. {
  225. return (__m256i)__builtin_ia32_pmaddubsw256((__v32qi)__a, (__v32qi)__b);
  226. }
  227. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  228. _mm256_madd_epi16(__m256i __a, __m256i __b)
  229. {
  230. return (__m256i)__builtin_ia32_pmaddwd256((__v16hi)__a, (__v16hi)__b);
  231. }
  232. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  233. _mm256_max_epi8(__m256i __a, __m256i __b)
  234. {
  235. return (__m256i)__builtin_ia32_pmaxsb256((__v32qi)__a, (__v32qi)__b);
  236. }
  237. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  238. _mm256_max_epi16(__m256i __a, __m256i __b)
  239. {
  240. return (__m256i)__builtin_ia32_pmaxsw256((__v16hi)__a, (__v16hi)__b);
  241. }
  242. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  243. _mm256_max_epi32(__m256i __a, __m256i __b)
  244. {
  245. return (__m256i)__builtin_ia32_pmaxsd256((__v8si)__a, (__v8si)__b);
  246. }
  247. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  248. _mm256_max_epu8(__m256i __a, __m256i __b)
  249. {
  250. return (__m256i)__builtin_ia32_pmaxub256((__v32qi)__a, (__v32qi)__b);
  251. }
  252. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  253. _mm256_max_epu16(__m256i __a, __m256i __b)
  254. {
  255. return (__m256i)__builtin_ia32_pmaxuw256((__v16hi)__a, (__v16hi)__b);
  256. }
  257. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  258. _mm256_max_epu32(__m256i __a, __m256i __b)
  259. {
  260. return (__m256i)__builtin_ia32_pmaxud256((__v8si)__a, (__v8si)__b);
  261. }
  262. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  263. _mm256_min_epi8(__m256i __a, __m256i __b)
  264. {
  265. return (__m256i)__builtin_ia32_pminsb256((__v32qi)__a, (__v32qi)__b);
  266. }
  267. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  268. _mm256_min_epi16(__m256i __a, __m256i __b)
  269. {
  270. return (__m256i)__builtin_ia32_pminsw256((__v16hi)__a, (__v16hi)__b);
  271. }
  272. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  273. _mm256_min_epi32(__m256i __a, __m256i __b)
  274. {
  275. return (__m256i)__builtin_ia32_pminsd256((__v8si)__a, (__v8si)__b);
  276. }
  277. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  278. _mm256_min_epu8(__m256i __a, __m256i __b)
  279. {
  280. return (__m256i)__builtin_ia32_pminub256((__v32qi)__a, (__v32qi)__b);
  281. }
  282. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  283. _mm256_min_epu16(__m256i __a, __m256i __b)
  284. {
  285. return (__m256i)__builtin_ia32_pminuw256 ((__v16hi)__a, (__v16hi)__b);
  286. }
  287. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  288. _mm256_min_epu32(__m256i __a, __m256i __b)
  289. {
  290. return (__m256i)__builtin_ia32_pminud256((__v8si)__a, (__v8si)__b);
  291. }
  292. static __inline__ int __DEFAULT_FN_ATTRS256
  293. _mm256_movemask_epi8(__m256i __a)
  294. {
  295. return __builtin_ia32_pmovmskb256((__v32qi)__a);
  296. }
  297. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  298. _mm256_cvtepi8_epi16(__m128i __V)
  299. {
  300. /* This function always performs a signed extension, but __v16qi is a char
  301. which may be signed or unsigned, so use __v16qs. */
  302. return (__m256i)__builtin_convertvector((__v16qs)__V, __v16hi);
  303. }
  304. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  305. _mm256_cvtepi8_epi32(__m128i __V)
  306. {
  307. /* This function always performs a signed extension, but __v16qi is a char
  308. which may be signed or unsigned, so use __v16qs. */
  309. return (__m256i)__builtin_convertvector(__builtin_shufflevector((__v16qs)__V, (__v16qs)__V, 0, 1, 2, 3, 4, 5, 6, 7), __v8si);
  310. }
  311. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  312. _mm256_cvtepi8_epi64(__m128i __V)
  313. {
  314. /* This function always performs a signed extension, but __v16qi is a char
  315. which may be signed or unsigned, so use __v16qs. */
  316. return (__m256i)__builtin_convertvector(__builtin_shufflevector((__v16qs)__V, (__v16qs)__V, 0, 1, 2, 3), __v4di);
  317. }
  318. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  319. _mm256_cvtepi16_epi32(__m128i __V)
  320. {
  321. return (__m256i)__builtin_convertvector((__v8hi)__V, __v8si);
  322. }
  323. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  324. _mm256_cvtepi16_epi64(__m128i __V)
  325. {
  326. return (__m256i)__builtin_convertvector(__builtin_shufflevector((__v8hi)__V, (__v8hi)__V, 0, 1, 2, 3), __v4di);
  327. }
  328. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  329. _mm256_cvtepi32_epi64(__m128i __V)
  330. {
  331. return (__m256i)__builtin_convertvector((__v4si)__V, __v4di);
  332. }
  333. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  334. _mm256_cvtepu8_epi16(__m128i __V)
  335. {
  336. return (__m256i)__builtin_convertvector((__v16qu)__V, __v16hi);
  337. }
  338. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  339. _mm256_cvtepu8_epi32(__m128i __V)
  340. {
  341. return (__m256i)__builtin_convertvector(__builtin_shufflevector((__v16qu)__V, (__v16qu)__V, 0, 1, 2, 3, 4, 5, 6, 7), __v8si);
  342. }
  343. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  344. _mm256_cvtepu8_epi64(__m128i __V)
  345. {
  346. return (__m256i)__builtin_convertvector(__builtin_shufflevector((__v16qu)__V, (__v16qu)__V, 0, 1, 2, 3), __v4di);
  347. }
  348. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  349. _mm256_cvtepu16_epi32(__m128i __V)
  350. {
  351. return (__m256i)__builtin_convertvector((__v8hu)__V, __v8si);
  352. }
  353. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  354. _mm256_cvtepu16_epi64(__m128i __V)
  355. {
  356. return (__m256i)__builtin_convertvector(__builtin_shufflevector((__v8hu)__V, (__v8hu)__V, 0, 1, 2, 3), __v4di);
  357. }
  358. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  359. _mm256_cvtepu32_epi64(__m128i __V)
  360. {
  361. return (__m256i)__builtin_convertvector((__v4su)__V, __v4di);
  362. }
  363. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  364. _mm256_mul_epi32(__m256i __a, __m256i __b)
  365. {
  366. return (__m256i)__builtin_ia32_pmuldq256((__v8si)__a, (__v8si)__b);
  367. }
  368. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  369. _mm256_mulhrs_epi16(__m256i __a, __m256i __b)
  370. {
  371. return (__m256i)__builtin_ia32_pmulhrsw256((__v16hi)__a, (__v16hi)__b);
  372. }
  373. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  374. _mm256_mulhi_epu16(__m256i __a, __m256i __b)
  375. {
  376. return (__m256i)__builtin_ia32_pmulhuw256((__v16hi)__a, (__v16hi)__b);
  377. }
  378. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  379. _mm256_mulhi_epi16(__m256i __a, __m256i __b)
  380. {
  381. return (__m256i)__builtin_ia32_pmulhw256((__v16hi)__a, (__v16hi)__b);
  382. }
  383. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  384. _mm256_mullo_epi16(__m256i __a, __m256i __b)
  385. {
  386. return (__m256i)((__v16hu)__a * (__v16hu)__b);
  387. }
  388. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  389. _mm256_mullo_epi32 (__m256i __a, __m256i __b)
  390. {
  391. return (__m256i)((__v8su)__a * (__v8su)__b);
  392. }
  393. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  394. _mm256_mul_epu32(__m256i __a, __m256i __b)
  395. {
  396. return __builtin_ia32_pmuludq256((__v8si)__a, (__v8si)__b);
  397. }
  398. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  399. _mm256_or_si256(__m256i __a, __m256i __b)
  400. {
  401. return (__m256i)((__v4du)__a | (__v4du)__b);
  402. }
  403. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  404. _mm256_sad_epu8(__m256i __a, __m256i __b)
  405. {
  406. return __builtin_ia32_psadbw256((__v32qi)__a, (__v32qi)__b);
  407. }
  408. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  409. _mm256_shuffle_epi8(__m256i __a, __m256i __b)
  410. {
  411. return (__m256i)__builtin_ia32_pshufb256((__v32qi)__a, (__v32qi)__b);
  412. }
  413. #define _mm256_shuffle_epi32(a, imm) \
  414. (__m256i)__builtin_ia32_pshufd256((__v8si)(__m256i)(a), (int)(imm))
  415. #define _mm256_shufflehi_epi16(a, imm) \
  416. (__m256i)__builtin_ia32_pshufhw256((__v16hi)(__m256i)(a), (int)(imm))
  417. #define _mm256_shufflelo_epi16(a, imm) \
  418. (__m256i)__builtin_ia32_pshuflw256((__v16hi)(__m256i)(a), (int)(imm))
  419. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  420. _mm256_sign_epi8(__m256i __a, __m256i __b)
  421. {
  422. return (__m256i)__builtin_ia32_psignb256((__v32qi)__a, (__v32qi)__b);
  423. }
  424. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  425. _mm256_sign_epi16(__m256i __a, __m256i __b)
  426. {
  427. return (__m256i)__builtin_ia32_psignw256((__v16hi)__a, (__v16hi)__b);
  428. }
  429. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  430. _mm256_sign_epi32(__m256i __a, __m256i __b)
  431. {
  432. return (__m256i)__builtin_ia32_psignd256((__v8si)__a, (__v8si)__b);
  433. }
  434. #define _mm256_slli_si256(a, imm) \
  435. (__m256i)__builtin_ia32_pslldqi256_byteshift((__v4di)(__m256i)(a), (int)(imm))
  436. #define _mm256_bslli_epi128(a, imm) \
  437. (__m256i)__builtin_ia32_pslldqi256_byteshift((__v4di)(__m256i)(a), (int)(imm))
  438. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  439. _mm256_slli_epi16(__m256i __a, int __count)
  440. {
  441. return (__m256i)__builtin_ia32_psllwi256((__v16hi)__a, __count);
  442. }
  443. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  444. _mm256_sll_epi16(__m256i __a, __m128i __count)
  445. {
  446. return (__m256i)__builtin_ia32_psllw256((__v16hi)__a, (__v8hi)__count);
  447. }
  448. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  449. _mm256_slli_epi32(__m256i __a, int __count)
  450. {
  451. return (__m256i)__builtin_ia32_pslldi256((__v8si)__a, __count);
  452. }
  453. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  454. _mm256_sll_epi32(__m256i __a, __m128i __count)
  455. {
  456. return (__m256i)__builtin_ia32_pslld256((__v8si)__a, (__v4si)__count);
  457. }
  458. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  459. _mm256_slli_epi64(__m256i __a, int __count)
  460. {
  461. return __builtin_ia32_psllqi256((__v4di)__a, __count);
  462. }
  463. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  464. _mm256_sll_epi64(__m256i __a, __m128i __count)
  465. {
  466. return __builtin_ia32_psllq256((__v4di)__a, __count);
  467. }
  468. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  469. _mm256_srai_epi16(__m256i __a, int __count)
  470. {
  471. return (__m256i)__builtin_ia32_psrawi256((__v16hi)__a, __count);
  472. }
  473. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  474. _mm256_sra_epi16(__m256i __a, __m128i __count)
  475. {
  476. return (__m256i)__builtin_ia32_psraw256((__v16hi)__a, (__v8hi)__count);
  477. }
  478. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  479. _mm256_srai_epi32(__m256i __a, int __count)
  480. {
  481. return (__m256i)__builtin_ia32_psradi256((__v8si)__a, __count);
  482. }
  483. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  484. _mm256_sra_epi32(__m256i __a, __m128i __count)
  485. {
  486. return (__m256i)__builtin_ia32_psrad256((__v8si)__a, (__v4si)__count);
  487. }
  488. #define _mm256_srli_si256(a, imm) \
  489. (__m256i)__builtin_ia32_psrldqi256_byteshift((__m256i)(a), (int)(imm))
  490. #define _mm256_bsrli_epi128(a, imm) \
  491. (__m256i)__builtin_ia32_psrldqi256_byteshift((__m256i)(a), (int)(imm))
  492. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  493. _mm256_srli_epi16(__m256i __a, int __count)
  494. {
  495. return (__m256i)__builtin_ia32_psrlwi256((__v16hi)__a, __count);
  496. }
  497. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  498. _mm256_srl_epi16(__m256i __a, __m128i __count)
  499. {
  500. return (__m256i)__builtin_ia32_psrlw256((__v16hi)__a, (__v8hi)__count);
  501. }
  502. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  503. _mm256_srli_epi32(__m256i __a, int __count)
  504. {
  505. return (__m256i)__builtin_ia32_psrldi256((__v8si)__a, __count);
  506. }
  507. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  508. _mm256_srl_epi32(__m256i __a, __m128i __count)
  509. {
  510. return (__m256i)__builtin_ia32_psrld256((__v8si)__a, (__v4si)__count);
  511. }
  512. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  513. _mm256_srli_epi64(__m256i __a, int __count)
  514. {
  515. return __builtin_ia32_psrlqi256((__v4di)__a, __count);
  516. }
  517. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  518. _mm256_srl_epi64(__m256i __a, __m128i __count)
  519. {
  520. return __builtin_ia32_psrlq256((__v4di)__a, __count);
  521. }
  522. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  523. _mm256_sub_epi8(__m256i __a, __m256i __b)
  524. {
  525. return (__m256i)((__v32qu)__a - (__v32qu)__b);
  526. }
  527. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  528. _mm256_sub_epi16(__m256i __a, __m256i __b)
  529. {
  530. return (__m256i)((__v16hu)__a - (__v16hu)__b);
  531. }
  532. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  533. _mm256_sub_epi32(__m256i __a, __m256i __b)
  534. {
  535. return (__m256i)((__v8su)__a - (__v8su)__b);
  536. }
  537. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  538. _mm256_sub_epi64(__m256i __a, __m256i __b)
  539. {
  540. return (__m256i)((__v4du)__a - (__v4du)__b);
  541. }
  542. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  543. _mm256_subs_epi8(__m256i __a, __m256i __b)
  544. {
  545. return (__m256i)__builtin_ia32_psubsb256((__v32qi)__a, (__v32qi)__b);
  546. }
  547. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  548. _mm256_subs_epi16(__m256i __a, __m256i __b)
  549. {
  550. return (__m256i)__builtin_ia32_psubsw256((__v16hi)__a, (__v16hi)__b);
  551. }
  552. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  553. _mm256_subs_epu8(__m256i __a, __m256i __b)
  554. {
  555. return (__m256i)__builtin_ia32_psubusb256((__v32qi)__a, (__v32qi)__b);
  556. }
  557. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  558. _mm256_subs_epu16(__m256i __a, __m256i __b)
  559. {
  560. return (__m256i)__builtin_ia32_psubusw256((__v16hi)__a, (__v16hi)__b);
  561. }
  562. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  563. _mm256_unpackhi_epi8(__m256i __a, __m256i __b)
  564. {
  565. return (__m256i)__builtin_shufflevector((__v32qi)__a, (__v32qi)__b, 8, 32+8, 9, 32+9, 10, 32+10, 11, 32+11, 12, 32+12, 13, 32+13, 14, 32+14, 15, 32+15, 24, 32+24, 25, 32+25, 26, 32+26, 27, 32+27, 28, 32+28, 29, 32+29, 30, 32+30, 31, 32+31);
  566. }
  567. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  568. _mm256_unpackhi_epi16(__m256i __a, __m256i __b)
  569. {
  570. return (__m256i)__builtin_shufflevector((__v16hi)__a, (__v16hi)__b, 4, 16+4, 5, 16+5, 6, 16+6, 7, 16+7, 12, 16+12, 13, 16+13, 14, 16+14, 15, 16+15);
  571. }
  572. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  573. _mm256_unpackhi_epi32(__m256i __a, __m256i __b)
  574. {
  575. return (__m256i)__builtin_shufflevector((__v8si)__a, (__v8si)__b, 2, 8+2, 3, 8+3, 6, 8+6, 7, 8+7);
  576. }
  577. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  578. _mm256_unpackhi_epi64(__m256i __a, __m256i __b)
  579. {
  580. return (__m256i)__builtin_shufflevector((__v4di)__a, (__v4di)__b, 1, 4+1, 3, 4+3);
  581. }
  582. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  583. _mm256_unpacklo_epi8(__m256i __a, __m256i __b)
  584. {
  585. return (__m256i)__builtin_shufflevector((__v32qi)__a, (__v32qi)__b, 0, 32+0, 1, 32+1, 2, 32+2, 3, 32+3, 4, 32+4, 5, 32+5, 6, 32+6, 7, 32+7, 16, 32+16, 17, 32+17, 18, 32+18, 19, 32+19, 20, 32+20, 21, 32+21, 22, 32+22, 23, 32+23);
  586. }
  587. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  588. _mm256_unpacklo_epi16(__m256i __a, __m256i __b)
  589. {
  590. return (__m256i)__builtin_shufflevector((__v16hi)__a, (__v16hi)__b, 0, 16+0, 1, 16+1, 2, 16+2, 3, 16+3, 8, 16+8, 9, 16+9, 10, 16+10, 11, 16+11);
  591. }
  592. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  593. _mm256_unpacklo_epi32(__m256i __a, __m256i __b)
  594. {
  595. return (__m256i)__builtin_shufflevector((__v8si)__a, (__v8si)__b, 0, 8+0, 1, 8+1, 4, 8+4, 5, 8+5);
  596. }
  597. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  598. _mm256_unpacklo_epi64(__m256i __a, __m256i __b)
  599. {
  600. return (__m256i)__builtin_shufflevector((__v4di)__a, (__v4di)__b, 0, 4+0, 2, 4+2);
  601. }
  602. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  603. _mm256_xor_si256(__m256i __a, __m256i __b)
  604. {
  605. return (__m256i)((__v4du)__a ^ (__v4du)__b);
  606. }
  607. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  608. _mm256_stream_load_si256(__m256i const *__V)
  609. {
  610. typedef __v4di __v4di_aligned __attribute__((aligned(32)));
  611. return (__m256i)__builtin_nontemporal_load((const __v4di_aligned *)__V);
  612. }
  613. static __inline__ __m128 __DEFAULT_FN_ATTRS128
  614. _mm_broadcastss_ps(__m128 __X)
  615. {
  616. return (__m128)__builtin_shufflevector((__v4sf)__X, (__v4sf)__X, 0, 0, 0, 0);
  617. }
  618. static __inline__ __m128d __DEFAULT_FN_ATTRS128
  619. _mm_broadcastsd_pd(__m128d __a)
  620. {
  621. return __builtin_shufflevector((__v2df)__a, (__v2df)__a, 0, 0);
  622. }
  623. static __inline__ __m256 __DEFAULT_FN_ATTRS256
  624. _mm256_broadcastss_ps(__m128 __X)
  625. {
  626. return (__m256)__builtin_shufflevector((__v4sf)__X, (__v4sf)__X, 0, 0, 0, 0, 0, 0, 0, 0);
  627. }
  628. static __inline__ __m256d __DEFAULT_FN_ATTRS256
  629. _mm256_broadcastsd_pd(__m128d __X)
  630. {
  631. return (__m256d)__builtin_shufflevector((__v2df)__X, (__v2df)__X, 0, 0, 0, 0);
  632. }
  633. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  634. _mm256_broadcastsi128_si256(__m128i __X)
  635. {
  636. return (__m256i)__builtin_shufflevector((__v2di)__X, (__v2di)__X, 0, 1, 0, 1);
  637. }
  638. #define _mm_blend_epi32(V1, V2, M) \
  639. (__m128i)__builtin_ia32_pblendd128((__v4si)(__m128i)(V1), \
  640. (__v4si)(__m128i)(V2), (int)(M))
  641. #define _mm256_blend_epi32(V1, V2, M) \
  642. (__m256i)__builtin_ia32_pblendd256((__v8si)(__m256i)(V1), \
  643. (__v8si)(__m256i)(V2), (int)(M))
  644. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  645. _mm256_broadcastb_epi8(__m128i __X)
  646. {
  647. return (__m256i)__builtin_shufflevector((__v16qi)__X, (__v16qi)__X, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  648. }
  649. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  650. _mm256_broadcastw_epi16(__m128i __X)
  651. {
  652. return (__m256i)__builtin_shufflevector((__v8hi)__X, (__v8hi)__X, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  653. }
  654. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  655. _mm256_broadcastd_epi32(__m128i __X)
  656. {
  657. return (__m256i)__builtin_shufflevector((__v4si)__X, (__v4si)__X, 0, 0, 0, 0, 0, 0, 0, 0);
  658. }
  659. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  660. _mm256_broadcastq_epi64(__m128i __X)
  661. {
  662. return (__m256i)__builtin_shufflevector((__v2di)__X, (__v2di)__X, 0, 0, 0, 0);
  663. }
  664. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  665. _mm_broadcastb_epi8(__m128i __X)
  666. {
  667. return (__m128i)__builtin_shufflevector((__v16qi)__X, (__v16qi)__X, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  668. }
  669. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  670. _mm_broadcastw_epi16(__m128i __X)
  671. {
  672. return (__m128i)__builtin_shufflevector((__v8hi)__X, (__v8hi)__X, 0, 0, 0, 0, 0, 0, 0, 0);
  673. }
  674. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  675. _mm_broadcastd_epi32(__m128i __X)
  676. {
  677. return (__m128i)__builtin_shufflevector((__v4si)__X, (__v4si)__X, 0, 0, 0, 0);
  678. }
  679. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  680. _mm_broadcastq_epi64(__m128i __X)
  681. {
  682. return (__m128i)__builtin_shufflevector((__v2di)__X, (__v2di)__X, 0, 0);
  683. }
  684. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  685. _mm256_permutevar8x32_epi32(__m256i __a, __m256i __b)
  686. {
  687. return (__m256i)__builtin_ia32_permvarsi256((__v8si)__a, (__v8si)__b);
  688. }
  689. #define _mm256_permute4x64_pd(V, M) \
  690. (__m256d)__builtin_ia32_permdf256((__v4df)(__m256d)(V), (int)(M))
  691. static __inline__ __m256 __DEFAULT_FN_ATTRS256
  692. _mm256_permutevar8x32_ps(__m256 __a, __m256i __b)
  693. {
  694. return (__m256)__builtin_ia32_permvarsf256((__v8sf)__a, (__v8si)__b);
  695. }
  696. #define _mm256_permute4x64_epi64(V, M) \
  697. (__m256i)__builtin_ia32_permdi256((__v4di)(__m256i)(V), (int)(M))
  698. #define _mm256_permute2x128_si256(V1, V2, M) \
  699. (__m256i)__builtin_ia32_permti256((__m256i)(V1), (__m256i)(V2), (int)(M))
  700. #define _mm256_extracti128_si256(V, M) \
  701. (__m128i)__builtin_ia32_extract128i256((__v4di)(__m256i)(V), (int)(M))
  702. #define _mm256_inserti128_si256(V1, V2, M) \
  703. (__m256i)__builtin_ia32_insert128i256((__v4di)(__m256i)(V1), \
  704. (__v2di)(__m128i)(V2), (int)(M))
  705. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  706. _mm256_maskload_epi32(int const *__X, __m256i __M)
  707. {
  708. return (__m256i)__builtin_ia32_maskloadd256((const __v8si *)__X, (__v8si)__M);
  709. }
  710. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  711. _mm256_maskload_epi64(long long const *__X, __m256i __M)
  712. {
  713. return (__m256i)__builtin_ia32_maskloadq256((const __v4di *)__X, (__v4di)__M);
  714. }
  715. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  716. _mm_maskload_epi32(int const *__X, __m128i __M)
  717. {
  718. return (__m128i)__builtin_ia32_maskloadd((const __v4si *)__X, (__v4si)__M);
  719. }
  720. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  721. _mm_maskload_epi64(long long const *__X, __m128i __M)
  722. {
  723. return (__m128i)__builtin_ia32_maskloadq((const __v2di *)__X, (__v2di)__M);
  724. }
  725. static __inline__ void __DEFAULT_FN_ATTRS256
  726. _mm256_maskstore_epi32(int *__X, __m256i __M, __m256i __Y)
  727. {
  728. __builtin_ia32_maskstored256((__v8si *)__X, (__v8si)__M, (__v8si)__Y);
  729. }
  730. static __inline__ void __DEFAULT_FN_ATTRS256
  731. _mm256_maskstore_epi64(long long *__X, __m256i __M, __m256i __Y)
  732. {
  733. __builtin_ia32_maskstoreq256((__v4di *)__X, (__v4di)__M, (__v4di)__Y);
  734. }
  735. static __inline__ void __DEFAULT_FN_ATTRS128
  736. _mm_maskstore_epi32(int *__X, __m128i __M, __m128i __Y)
  737. {
  738. __builtin_ia32_maskstored((__v4si *)__X, (__v4si)__M, (__v4si)__Y);
  739. }
  740. static __inline__ void __DEFAULT_FN_ATTRS128
  741. _mm_maskstore_epi64(long long *__X, __m128i __M, __m128i __Y)
  742. {
  743. __builtin_ia32_maskstoreq(( __v2di *)__X, (__v2di)__M, (__v2di)__Y);
  744. }
  745. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  746. _mm256_sllv_epi32(__m256i __X, __m256i __Y)
  747. {
  748. return (__m256i)__builtin_ia32_psllv8si((__v8si)__X, (__v8si)__Y);
  749. }
  750. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  751. _mm_sllv_epi32(__m128i __X, __m128i __Y)
  752. {
  753. return (__m128i)__builtin_ia32_psllv4si((__v4si)__X, (__v4si)__Y);
  754. }
  755. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  756. _mm256_sllv_epi64(__m256i __X, __m256i __Y)
  757. {
  758. return (__m256i)__builtin_ia32_psllv4di((__v4di)__X, (__v4di)__Y);
  759. }
  760. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  761. _mm_sllv_epi64(__m128i __X, __m128i __Y)
  762. {
  763. return (__m128i)__builtin_ia32_psllv2di((__v2di)__X, (__v2di)__Y);
  764. }
  765. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  766. _mm256_srav_epi32(__m256i __X, __m256i __Y)
  767. {
  768. return (__m256i)__builtin_ia32_psrav8si((__v8si)__X, (__v8si)__Y);
  769. }
  770. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  771. _mm_srav_epi32(__m128i __X, __m128i __Y)
  772. {
  773. return (__m128i)__builtin_ia32_psrav4si((__v4si)__X, (__v4si)__Y);
  774. }
  775. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  776. _mm256_srlv_epi32(__m256i __X, __m256i __Y)
  777. {
  778. return (__m256i)__builtin_ia32_psrlv8si((__v8si)__X, (__v8si)__Y);
  779. }
  780. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  781. _mm_srlv_epi32(__m128i __X, __m128i __Y)
  782. {
  783. return (__m128i)__builtin_ia32_psrlv4si((__v4si)__X, (__v4si)__Y);
  784. }
  785. static __inline__ __m256i __DEFAULT_FN_ATTRS256
  786. _mm256_srlv_epi64(__m256i __X, __m256i __Y)
  787. {
  788. return (__m256i)__builtin_ia32_psrlv4di((__v4di)__X, (__v4di)__Y);
  789. }
  790. static __inline__ __m128i __DEFAULT_FN_ATTRS128
  791. _mm_srlv_epi64(__m128i __X, __m128i __Y)
  792. {
  793. return (__m128i)__builtin_ia32_psrlv2di((__v2di)__X, (__v2di)__Y);
  794. }
  795. #define _mm_mask_i32gather_pd(a, m, i, mask, s) \
  796. (__m128d)__builtin_ia32_gatherd_pd((__v2df)(__m128i)(a), \
  797. (double const *)(m), \
  798. (__v4si)(__m128i)(i), \
  799. (__v2df)(__m128d)(mask), (s))
  800. #define _mm256_mask_i32gather_pd(a, m, i, mask, s) \
  801. (__m256d)__builtin_ia32_gatherd_pd256((__v4df)(__m256d)(a), \
  802. (double const *)(m), \
  803. (__v4si)(__m128i)(i), \
  804. (__v4df)(__m256d)(mask), (s))
  805. #define _mm_mask_i64gather_pd(a, m, i, mask, s) \
  806. (__m128d)__builtin_ia32_gatherq_pd((__v2df)(__m128d)(a), \
  807. (double const *)(m), \
  808. (__v2di)(__m128i)(i), \
  809. (__v2df)(__m128d)(mask), (s))
  810. #define _mm256_mask_i64gather_pd(a, m, i, mask, s) \
  811. (__m256d)__builtin_ia32_gatherq_pd256((__v4df)(__m256d)(a), \
  812. (double const *)(m), \
  813. (__v4di)(__m256i)(i), \
  814. (__v4df)(__m256d)(mask), (s))
  815. #define _mm_mask_i32gather_ps(a, m, i, mask, s) \
  816. (__m128)__builtin_ia32_gatherd_ps((__v4sf)(__m128)(a), \
  817. (float const *)(m), \
  818. (__v4si)(__m128i)(i), \
  819. (__v4sf)(__m128)(mask), (s))
  820. #define _mm256_mask_i32gather_ps(a, m, i, mask, s) \
  821. (__m256)__builtin_ia32_gatherd_ps256((__v8sf)(__m256)(a), \
  822. (float const *)(m), \
  823. (__v8si)(__m256i)(i), \
  824. (__v8sf)(__m256)(mask), (s))
  825. #define _mm_mask_i64gather_ps(a, m, i, mask, s) \
  826. (__m128)__builtin_ia32_gatherq_ps((__v4sf)(__m128)(a), \
  827. (float const *)(m), \
  828. (__v2di)(__m128i)(i), \
  829. (__v4sf)(__m128)(mask), (s))
  830. #define _mm256_mask_i64gather_ps(a, m, i, mask, s) \
  831. (__m128)__builtin_ia32_gatherq_ps256((__v4sf)(__m128)(a), \
  832. (float const *)(m), \
  833. (__v4di)(__m256i)(i), \
  834. (__v4sf)(__m128)(mask), (s))
  835. #define _mm_mask_i32gather_epi32(a, m, i, mask, s) \
  836. (__m128i)__builtin_ia32_gatherd_d((__v4si)(__m128i)(a), \
  837. (int const *)(m), \
  838. (__v4si)(__m128i)(i), \
  839. (__v4si)(__m128i)(mask), (s))
  840. #define _mm256_mask_i32gather_epi32(a, m, i, mask, s) \
  841. (__m256i)__builtin_ia32_gatherd_d256((__v8si)(__m256i)(a), \
  842. (int const *)(m), \
  843. (__v8si)(__m256i)(i), \
  844. (__v8si)(__m256i)(mask), (s))
  845. #define _mm_mask_i64gather_epi32(a, m, i, mask, s) \
  846. (__m128i)__builtin_ia32_gatherq_d((__v4si)(__m128i)(a), \
  847. (int const *)(m), \
  848. (__v2di)(__m128i)(i), \
  849. (__v4si)(__m128i)(mask), (s))
  850. #define _mm256_mask_i64gather_epi32(a, m, i, mask, s) \
  851. (__m128i)__builtin_ia32_gatherq_d256((__v4si)(__m128i)(a), \
  852. (int const *)(m), \
  853. (__v4di)(__m256i)(i), \
  854. (__v4si)(__m128i)(mask), (s))
  855. #define _mm_mask_i32gather_epi64(a, m, i, mask, s) \
  856. (__m128i)__builtin_ia32_gatherd_q((__v2di)(__m128i)(a), \
  857. (long long const *)(m), \
  858. (__v4si)(__m128i)(i), \
  859. (__v2di)(__m128i)(mask), (s))
  860. #define _mm256_mask_i32gather_epi64(a, m, i, mask, s) \
  861. (__m256i)__builtin_ia32_gatherd_q256((__v4di)(__m256i)(a), \
  862. (long long const *)(m), \
  863. (__v4si)(__m128i)(i), \
  864. (__v4di)(__m256i)(mask), (s))
  865. #define _mm_mask_i64gather_epi64(a, m, i, mask, s) \
  866. (__m128i)__builtin_ia32_gatherq_q((__v2di)(__m128i)(a), \
  867. (long long const *)(m), \
  868. (__v2di)(__m128i)(i), \
  869. (__v2di)(__m128i)(mask), (s))
  870. #define _mm256_mask_i64gather_epi64(a, m, i, mask, s) \
  871. (__m256i)__builtin_ia32_gatherq_q256((__v4di)(__m256i)(a), \
  872. (long long const *)(m), \
  873. (__v4di)(__m256i)(i), \
  874. (__v4di)(__m256i)(mask), (s))
  875. #define _mm_i32gather_pd(m, i, s) \
  876. (__m128d)__builtin_ia32_gatherd_pd((__v2df)_mm_undefined_pd(), \
  877. (double const *)(m), \
  878. (__v4si)(__m128i)(i), \
  879. (__v2df)_mm_cmpeq_pd(_mm_setzero_pd(), \
  880. _mm_setzero_pd()), \
  881. (s))
  882. #define _mm256_i32gather_pd(m, i, s) \
  883. (__m256d)__builtin_ia32_gatherd_pd256((__v4df)_mm256_undefined_pd(), \
  884. (double const *)(m), \
  885. (__v4si)(__m128i)(i), \
  886. (__v4df)_mm256_cmp_pd(_mm256_setzero_pd(), \
  887. _mm256_setzero_pd(), \
  888. _CMP_EQ_OQ), \
  889. (s))
  890. #define _mm_i64gather_pd(m, i, s) \
  891. (__m128d)__builtin_ia32_gatherq_pd((__v2df)_mm_undefined_pd(), \
  892. (double const *)(m), \
  893. (__v2di)(__m128i)(i), \
  894. (__v2df)_mm_cmpeq_pd(_mm_setzero_pd(), \
  895. _mm_setzero_pd()), \
  896. (s))
  897. #define _mm256_i64gather_pd(m, i, s) \
  898. (__m256d)__builtin_ia32_gatherq_pd256((__v4df)_mm256_undefined_pd(), \
  899. (double const *)(m), \
  900. (__v4di)(__m256i)(i), \
  901. (__v4df)_mm256_cmp_pd(_mm256_setzero_pd(), \
  902. _mm256_setzero_pd(), \
  903. _CMP_EQ_OQ), \
  904. (s))
  905. #define _mm_i32gather_ps(m, i, s) \
  906. (__m128)__builtin_ia32_gatherd_ps((__v4sf)_mm_undefined_ps(), \
  907. (float const *)(m), \
  908. (__v4si)(__m128i)(i), \
  909. (__v4sf)_mm_cmpeq_ps(_mm_setzero_ps(), \
  910. _mm_setzero_ps()), \
  911. (s))
  912. #define _mm256_i32gather_ps(m, i, s) \
  913. (__m256)__builtin_ia32_gatherd_ps256((__v8sf)_mm256_undefined_ps(), \
  914. (float const *)(m), \
  915. (__v8si)(__m256i)(i), \
  916. (__v8sf)_mm256_cmp_ps(_mm256_setzero_ps(), \
  917. _mm256_setzero_ps(), \
  918. _CMP_EQ_OQ), \
  919. (s))
  920. #define _mm_i64gather_ps(m, i, s) \
  921. (__m128)__builtin_ia32_gatherq_ps((__v4sf)_mm_undefined_ps(), \
  922. (float const *)(m), \
  923. (__v2di)(__m128i)(i), \
  924. (__v4sf)_mm_cmpeq_ps(_mm_setzero_ps(), \
  925. _mm_setzero_ps()), \
  926. (s))
  927. #define _mm256_i64gather_ps(m, i, s) \
  928. (__m128)__builtin_ia32_gatherq_ps256((__v4sf)_mm_undefined_ps(), \
  929. (float const *)(m), \
  930. (__v4di)(__m256i)(i), \
  931. (__v4sf)_mm_cmpeq_ps(_mm_setzero_ps(), \
  932. _mm_setzero_ps()), \
  933. (s))
  934. #define _mm_i32gather_epi32(m, i, s) \
  935. (__m128i)__builtin_ia32_gatherd_d((__v4si)_mm_undefined_si128(), \
  936. (int const *)(m), (__v4si)(__m128i)(i), \
  937. (__v4si)_mm_set1_epi32(-1), (s))
  938. #define _mm256_i32gather_epi32(m, i, s) \
  939. (__m256i)__builtin_ia32_gatherd_d256((__v8si)_mm256_undefined_si256(), \
  940. (int const *)(m), (__v8si)(__m256i)(i), \
  941. (__v8si)_mm256_set1_epi32(-1), (s))
  942. #define _mm_i64gather_epi32(m, i, s) \
  943. (__m128i)__builtin_ia32_gatherq_d((__v4si)_mm_undefined_si128(), \
  944. (int const *)(m), (__v2di)(__m128i)(i), \
  945. (__v4si)_mm_set1_epi32(-1), (s))
  946. #define _mm256_i64gather_epi32(m, i, s) \
  947. (__m128i)__builtin_ia32_gatherq_d256((__v4si)_mm_undefined_si128(), \
  948. (int const *)(m), (__v4di)(__m256i)(i), \
  949. (__v4si)_mm_set1_epi32(-1), (s))
  950. #define _mm_i32gather_epi64(m, i, s) \
  951. (__m128i)__builtin_ia32_gatherd_q((__v2di)_mm_undefined_si128(), \
  952. (long long const *)(m), \
  953. (__v4si)(__m128i)(i), \
  954. (__v2di)_mm_set1_epi64x(-1), (s))
  955. #define _mm256_i32gather_epi64(m, i, s) \
  956. (__m256i)__builtin_ia32_gatherd_q256((__v4di)_mm256_undefined_si256(), \
  957. (long long const *)(m), \
  958. (__v4si)(__m128i)(i), \
  959. (__v4di)_mm256_set1_epi64x(-1), (s))
  960. #define _mm_i64gather_epi64(m, i, s) \
  961. (__m128i)__builtin_ia32_gatherq_q((__v2di)_mm_undefined_si128(), \
  962. (long long const *)(m), \
  963. (__v2di)(__m128i)(i), \
  964. (__v2di)_mm_set1_epi64x(-1), (s))
  965. #define _mm256_i64gather_epi64(m, i, s) \
  966. (__m256i)__builtin_ia32_gatherq_q256((__v4di)_mm256_undefined_si256(), \
  967. (long long const *)(m), \
  968. (__v4di)(__m256i)(i), \
  969. (__v4di)_mm256_set1_epi64x(-1), (s))
  970. #undef __DEFAULT_FN_ATTRS256
  971. #undef __DEFAULT_FN_ATTRS128
  972. #endif /* __AVX2INTRIN_H */