ssse3.h 36 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058
  1. /* SPDX-License-Identifier: MIT
  2. *
  3. * Permission is hereby granted, free of charge, to any person
  4. * obtaining a copy of this software and associated documentation
  5. * files (the "Software"), to deal in the Software without
  6. * restriction, including without limitation the rights to use, copy,
  7. * modify, merge, publish, distribute, sublicense, and/or sell copies
  8. * of the Software, and to permit persons to whom the Software is
  9. * furnished to do so, subject to the following conditions:
  10. *
  11. * The above copyright notice and this permission notice shall be
  12. * included in all copies or substantial portions of the Software.
  13. *
  14. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  15. * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  16. * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  17. * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  18. * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  19. * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  20. * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21. * SOFTWARE.
  22. *
  23. * Copyright:
  24. * 2017-2020 Evan Nemerson <evan@nemerson.com>
  25. */
  26. #if !defined(SIMDE_X86_SSSE3_H)
  27. #define SIMDE_X86_SSSE3_H
  28. #include "sse3.h"
  29. HEDLEY_DIAGNOSTIC_PUSH
  30. SIMDE_DISABLE_UNWANTED_DIAGNOSTICS
  31. SIMDE_BEGIN_DECLS_
  32. SIMDE_FUNCTION_ATTRIBUTES
  33. simde__m128i
  34. simde_mm_abs_epi8 (simde__m128i a) {
  35. #if defined(SIMDE_X86_SSSE3_NATIVE)
  36. return _mm_abs_epi8(a);
  37. #elif defined(SIMDE_X86_SSE2_NATIVE)
  38. return _mm_min_epu8(a, _mm_sub_epi8(_mm_setzero_si128(), a));
  39. #else
  40. simde__m128i_private
  41. r_,
  42. a_ = simde__m128i_to_private(a);
  43. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  44. r_.neon_i8 = vabsq_s8(a_.neon_i8);
  45. #elif defined(SIMDE_POWER_ALTIVEC_P6_NATIVE) || defined(SIMDE_ZARCH_ZVECTOR_13_NATIVE)
  46. r_.altivec_i8 = vec_abs(a_.altivec_i8);
  47. #elif defined(SIMDE_WASM_SIMD128_NATIVE)
  48. r_.wasm_v128 = wasm_i8x16_abs(a_.wasm_v128);
  49. #else
  50. SIMDE_VECTORIZE
  51. for (size_t i = 0 ; i < (sizeof(r_.i8) / sizeof(r_.i8[0])) ; i++) {
  52. r_.u8[i] = HEDLEY_STATIC_CAST(uint8_t, (a_.i8[i] < 0) ? (- a_.i8[i]) : a_.i8[i]);
  53. }
  54. #endif
  55. return simde__m128i_from_private(r_);
  56. #endif
  57. }
  58. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  59. # define _mm_abs_epi8(a) simde_mm_abs_epi8(a)
  60. #endif
  61. SIMDE_FUNCTION_ATTRIBUTES
  62. simde__m128i
  63. simde_mm_abs_epi16 (simde__m128i a) {
  64. #if defined(SIMDE_X86_SSSE3_NATIVE)
  65. return _mm_abs_epi16(a);
  66. #elif defined(SIMDE_X86_SSE2_NATIVE)
  67. return _mm_max_epi16(a, _mm_sub_epi16(_mm_setzero_si128(), a));
  68. #else
  69. simde__m128i_private
  70. r_,
  71. a_ = simde__m128i_to_private(a);
  72. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  73. r_.neon_i16 = vabsq_s16(a_.neon_i16);
  74. #elif defined(SIMDE_POWER_ALTIVEC_P6_NATIVE) || defined(SIMDE_ZARCH_ZVECTOR_13_NATIVE)
  75. r_.altivec_i16 = vec_abs(a_.altivec_i16);
  76. #elif defined(SIMDE_WASM_SIMD128_NATIVE)
  77. r_.wasm_v128 = wasm_i16x8_abs(a_.wasm_v128);
  78. #else
  79. SIMDE_VECTORIZE
  80. for (size_t i = 0 ; i < (sizeof(r_.i16) / sizeof(r_.i16[0])) ; i++) {
  81. r_.u16[i] = HEDLEY_STATIC_CAST(uint16_t, (a_.i16[i] < 0) ? (- a_.i16[i]) : a_.i16[i]);
  82. }
  83. #endif
  84. return simde__m128i_from_private(r_);
  85. #endif
  86. }
  87. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  88. # define _mm_abs_epi16(a) simde_mm_abs_epi16(a)
  89. #endif
  90. SIMDE_FUNCTION_ATTRIBUTES
  91. simde__m128i
  92. simde_mm_abs_epi32 (simde__m128i a) {
  93. #if defined(SIMDE_X86_SSSE3_NATIVE)
  94. return _mm_abs_epi32(a);
  95. #elif defined(SIMDE_X86_SSE2_NATIVE)
  96. const __m128i m = _mm_cmpgt_epi32(_mm_setzero_si128(), a);
  97. return _mm_sub_epi32(_mm_xor_si128(a, m), m);
  98. #else
  99. simde__m128i_private
  100. r_,
  101. a_ = simde__m128i_to_private(a);
  102. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  103. r_.neon_i32 = vabsq_s32(a_.neon_i32);
  104. #elif defined(SIMDE_POWER_ALTIVEC_P6_NATIVE) || defined(SIMDE_ZARCH_ZVECTOR_13_NATIVE)
  105. r_.altivec_i32 = vec_abs(a_.altivec_i32);
  106. #elif defined(SIMDE_WASM_SIMD128_NATIVE)
  107. r_.wasm_v128 = wasm_i32x4_abs(a_.wasm_v128);
  108. #else
  109. SIMDE_VECTORIZE
  110. for (size_t i = 0 ; i < (sizeof(r_.i32) / sizeof(r_.i32[0])) ; i++) {
  111. #if defined(_MSC_VER)
  112. HEDLEY_DIAGNOSTIC_PUSH
  113. #pragma warning(disable:4146)
  114. #endif
  115. r_.u32[i] = (a_.i32[i] < 0) ? (- HEDLEY_STATIC_CAST(uint32_t, a_.i32[i])) : HEDLEY_STATIC_CAST(uint32_t, a_.i32[i]);
  116. #if defined(_MSC_VER)
  117. HEDLEY_DIAGNOSTIC_POP
  118. #endif
  119. }
  120. #endif
  121. return simde__m128i_from_private(r_);
  122. #endif
  123. }
  124. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  125. # define _mm_abs_epi32(a) simde_mm_abs_epi32(a)
  126. #endif
  127. SIMDE_FUNCTION_ATTRIBUTES
  128. simde__m64
  129. simde_mm_abs_pi8 (simde__m64 a) {
  130. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  131. return _mm_abs_pi8(a);
  132. #else
  133. simde__m64_private
  134. r_,
  135. a_ = simde__m64_to_private(a);
  136. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  137. r_.neon_i8 = vabs_s8(a_.neon_i8);
  138. #else
  139. SIMDE_VECTORIZE
  140. for (size_t i = 0 ; i < (sizeof(r_.i8) / sizeof(r_.i8[0])) ; i++) {
  141. r_.u8[i] = HEDLEY_STATIC_CAST(uint8_t, (a_.i8[i] < 0) ? (- a_.i8[i]) : a_.i8[i]);
  142. }
  143. #endif
  144. return simde__m64_from_private(r_);
  145. #endif
  146. }
  147. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  148. # define _mm_abs_pi8(a) simde_mm_abs_pi8(a)
  149. #endif
  150. SIMDE_FUNCTION_ATTRIBUTES
  151. simde__m64
  152. simde_mm_abs_pi16 (simde__m64 a) {
  153. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  154. return _mm_abs_pi16(a);
  155. #else
  156. simde__m64_private
  157. r_,
  158. a_ = simde__m64_to_private(a);
  159. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  160. r_.neon_i16 = vabs_s16(a_.neon_i16);
  161. #else
  162. SIMDE_VECTORIZE
  163. for (size_t i = 0 ; i < (sizeof(r_.i16) / sizeof(r_.i16[0])) ; i++) {
  164. r_.u16[i] = HEDLEY_STATIC_CAST(uint16_t, (a_.i16[i] < 0) ? (- a_.i16[i]) : a_.i16[i]);
  165. }
  166. #endif
  167. return simde__m64_from_private(r_);
  168. #endif
  169. }
  170. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  171. # define _mm_abs_pi16(a) simde_mm_abs_pi16(a)
  172. #endif
  173. SIMDE_FUNCTION_ATTRIBUTES
  174. simde__m64
  175. simde_mm_abs_pi32 (simde__m64 a) {
  176. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  177. return _mm_abs_pi32(a);
  178. #else
  179. simde__m64_private
  180. r_,
  181. a_ = simde__m64_to_private(a);
  182. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  183. r_.neon_i32 = vabs_s32(a_.neon_i32);
  184. #else
  185. SIMDE_VECTORIZE
  186. for (size_t i = 0 ; i < (sizeof(r_.i32) / sizeof(r_.i32[0])) ; i++) {
  187. r_.u32[i] = HEDLEY_STATIC_CAST(uint32_t, (a_.i32[i] < 0) ? (- a_.i32[i]) : a_.i32[i]);
  188. }
  189. #endif
  190. return simde__m64_from_private(r_);
  191. #endif
  192. }
  193. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  194. # define _mm_abs_pi32(a) simde_mm_abs_pi32(a)
  195. #endif
  196. SIMDE_FUNCTION_ATTRIBUTES
  197. simde__m128i
  198. simde_mm_alignr_epi8 (simde__m128i a, simde__m128i b, int count)
  199. SIMDE_REQUIRE_CONSTANT_RANGE(count, 0, 255) {
  200. simde__m128i_private
  201. r_,
  202. a_ = simde__m128i_to_private(a),
  203. b_ = simde__m128i_to_private(b);
  204. if (HEDLEY_UNLIKELY(count > 31))
  205. return simde_mm_setzero_si128();
  206. for (size_t i = 0 ; i < (sizeof(r_.i8) / sizeof(r_.i8[0])) ; i++) {
  207. const int srcpos = count + HEDLEY_STATIC_CAST(int, i);
  208. if (srcpos > 31) {
  209. r_.i8[i] = 0;
  210. } else if (srcpos > 15) {
  211. r_.i8[i] = a_.i8[(srcpos) & 15];
  212. } else {
  213. r_.i8[i] = b_.i8[srcpos];
  214. }
  215. }
  216. return simde__m128i_from_private(r_);
  217. }
  218. #if defined(SIMDE_X86_SSSE3_NATIVE)
  219. #define simde_mm_alignr_epi8(a, b, count) _mm_alignr_epi8(a, b, count)
  220. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  221. #define simde_mm_alignr_epi8(a, b, count) \
  222. ( \
  223. ((count) > 31) \
  224. ? simde__m128i_from_neon_i8(vdupq_n_s8(0)) \
  225. : ( \
  226. ((count) > 15) \
  227. ? (simde__m128i_from_neon_i8(vextq_s8(simde__m128i_to_neon_i8(a), vdupq_n_s8(0), (count) & 15))) \
  228. : (simde__m128i_from_neon_i8(vextq_s8(simde__m128i_to_neon_i8(b), simde__m128i_to_neon_i8(a), ((count) & 15))))))
  229. #endif
  230. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  231. #define _mm_alignr_epi8(a, b, count) simde_mm_alignr_epi8(a, b, count)
  232. #endif
  233. SIMDE_FUNCTION_ATTRIBUTES
  234. simde__m64
  235. simde_mm_alignr_pi8 (simde__m64 a, simde__m64 b, const int count)
  236. SIMDE_REQUIRE_CONSTANT(count) {
  237. simde__m64_private
  238. r_,
  239. a_ = simde__m64_to_private(a),
  240. b_ = simde__m64_to_private(b);
  241. if (HEDLEY_UNLIKELY(count > 15))
  242. return simde_mm_setzero_si64();
  243. for (size_t i = 0 ; i < (sizeof(r_.i8) / sizeof(r_.i8[0])) ; i++) {
  244. const int srcpos = count + HEDLEY_STATIC_CAST(int, i);
  245. if (srcpos > 15) {
  246. r_.i8[i] = 0;
  247. } else if (srcpos > 7) {
  248. r_.i8[i] = a_.i8[(srcpos) & 7];
  249. } else {
  250. r_.i8[i] = b_.i8[srcpos];
  251. }
  252. }
  253. return simde__m64_from_private(r_);
  254. }
  255. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  256. # define simde_mm_alignr_pi8(a, b, count) _mm_alignr_pi8(a, b, count)
  257. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  258. #define simde_mm_alignr_pi8(a, b, count) \
  259. ( \
  260. ((count) > 15) \
  261. ? simde__m64_from_neon_i8(vdup_n_s8(0)) \
  262. : ( \
  263. ((count) > 7) \
  264. ? (simde__m64_from_neon_i8(vext_s8(simde__m64_to_neon_i8(a), vdup_n_s8(0), (count) & 7))) \
  265. : (simde__m64_from_neon_i8(vext_s8(simde__m64_to_neon_i8(b), simde__m64_to_neon_i8(a), ((count) & 7))))))
  266. #endif
  267. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  268. # define _mm_alignr_pi8(a, b, count) simde_mm_alignr_pi8(a, b, count)
  269. #endif
  270. SIMDE_FUNCTION_ATTRIBUTES
  271. simde__m128i
  272. simde_mm_shuffle_epi8 (simde__m128i a, simde__m128i b) {
  273. #if defined(SIMDE_X86_SSSE3_NATIVE)
  274. return _mm_shuffle_epi8(a, b);
  275. #else
  276. simde__m128i_private
  277. r_,
  278. a_ = simde__m128i_to_private(a),
  279. b_ = simde__m128i_to_private(b);
  280. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  281. r_.neon_i8 = vqtbl1q_s8(a_.neon_i8, vandq_u8(b_.neon_u8, vdupq_n_u8(0x8F)));
  282. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  283. /* Mask out the bits we're not interested in. vtbl will result in 0
  284. * for any values outside of [0, 15], so if the high bit is set it
  285. * will return 0, just like in SSSE3. */
  286. b_.neon_i8 = vandq_s8(b_.neon_i8, vdupq_n_s8(HEDLEY_STATIC_CAST(int8_t, (1 << 7) | 15)));
  287. /* Convert a from an int8x16_t to an int8x8x2_t */
  288. int8x8x2_t i;
  289. i.val[0] = vget_low_s8(a_.neon_i8);
  290. i.val[1] = vget_high_s8(a_.neon_i8);
  291. /* Table lookups */
  292. int8x8_t l = vtbl2_s8(i, vget_low_s8(b_.neon_i8));
  293. int8x8_t h = vtbl2_s8(i, vget_high_s8(b_.neon_i8));
  294. r_.neon_i8 = vcombine_s8(l, h);
  295. #elif defined(SIMDE_POWER_ALTIVEC_P6_NATIVE)
  296. /* This is a bit ugly because of the casts and the awful type
  297. * macros (SIMDE_POWER_ALTIVEC_VECTOR), but it's really just
  298. * vec_sel(vec_perm(a, a, b), 0, vec_cmplt(b, 0)) */
  299. SIMDE_POWER_ALTIVEC_VECTOR(signed char) z = { 0, };
  300. SIMDE_POWER_ALTIVEC_VECTOR(signed char) msb_mask = HEDLEY_REINTERPRET_CAST(SIMDE_POWER_ALTIVEC_VECTOR(signed char), vec_cmplt(b_.altivec_i8, z));
  301. SIMDE_POWER_ALTIVEC_VECTOR(signed char) c = vec_perm(a_.altivec_i8, a_.altivec_i8, HEDLEY_REINTERPRET_CAST(SIMDE_POWER_ALTIVEC_VECTOR(unsigned char), b_.altivec_i8));
  302. r_.altivec_i8 = vec_sel(c, z, HEDLEY_REINTERPRET_CAST(SIMDE_POWER_ALTIVEC_VECTOR(unsigned char), msb_mask));
  303. #elif defined(SIMDE_WASM_SIMD128_NATIVE)
  304. r_.wasm_v128 = wasm_i8x16_swizzle(
  305. a_.wasm_v128, wasm_v128_and(b_.wasm_v128, wasm_i8x16_splat(0x8F)));
  306. #else
  307. for (size_t i = 0 ; i < (sizeof(r_.i8) / sizeof(r_.i8[0])) ; i++) {
  308. r_.i8[i] = a_.i8[b_.i8[i] & 15] & (~(b_.i8[i]) >> 7);
  309. }
  310. #endif
  311. return simde__m128i_from_private(r_);
  312. #endif
  313. }
  314. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  315. # define _mm_shuffle_epi8(a, b) simde_mm_shuffle_epi8(a, b)
  316. #endif
  317. SIMDE_FUNCTION_ATTRIBUTES
  318. simde__m64
  319. simde_mm_shuffle_pi8 (simde__m64 a, simde__m64 b) {
  320. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  321. return _mm_shuffle_pi8(a, b);
  322. #else
  323. simde__m64_private
  324. r_,
  325. a_ = simde__m64_to_private(a),
  326. b_ = simde__m64_to_private(b);
  327. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  328. b_.neon_i8 = vand_s8(b_.neon_i8, vdup_n_s8(HEDLEY_STATIC_CAST(int8_t, (1 << 7) | 7)));
  329. r_.neon_i8 = vtbl1_s8(a_.neon_i8, b_.neon_i8);
  330. #else
  331. for (size_t i = 0 ; i < (sizeof(r_.u8) / sizeof(r_.u8[0])) ; i++) {
  332. r_.i8[i] = a_.i8[b_.i8[i] & 7] & (~(b_.i8[i]) >> 7);
  333. }
  334. #endif
  335. return simde__m64_from_private(r_);
  336. #endif
  337. }
  338. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  339. # define _mm_shuffle_pi8(a, b) simde_mm_shuffle_pi8(a, b)
  340. #endif
  341. SIMDE_FUNCTION_ATTRIBUTES
  342. simde__m128i
  343. simde_mm_hadd_epi16 (simde__m128i a, simde__m128i b) {
  344. #if defined(SIMDE_X86_SSSE3_NATIVE)
  345. return _mm_hadd_epi16(a, b);
  346. #elif defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  347. return simde__m128i_from_neon_i16(vpaddq_s16(simde__m128i_to_neon_i16(a), simde__m128i_to_neon_i16(b)));
  348. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  349. int16x8x2_t t = vuzpq_s16(simde__m128i_to_neon_i16(a), simde__m128i_to_neon_i16(b));
  350. return simde__m128i_from_neon_i16(vaddq_s16(t.val[0], t.val[1]));
  351. #else
  352. return simde_mm_add_epi16(simde_x_mm_deinterleaveeven_epi16(a, b), simde_x_mm_deinterleaveodd_epi16(a, b));
  353. #endif
  354. }
  355. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  356. # define _mm_hadd_epi16(a, b) simde_mm_hadd_epi16(a, b)
  357. #endif
  358. SIMDE_FUNCTION_ATTRIBUTES
  359. simde__m128i
  360. simde_mm_hadd_epi32 (simde__m128i a, simde__m128i b) {
  361. #if defined(SIMDE_X86_SSSE3_NATIVE)
  362. return _mm_hadd_epi32(a, b);
  363. #elif defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  364. return simde__m128i_from_neon_i32(vpaddq_s32(simde__m128i_to_neon_i32(a), simde__m128i_to_neon_i32(b)));
  365. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  366. int32x4x2_t t = vuzpq_s32(simde__m128i_to_neon_i32(a), simde__m128i_to_neon_i32(b));
  367. return simde__m128i_from_neon_i32(vaddq_s32(t.val[0], t.val[1]));
  368. #else
  369. return simde_mm_add_epi32(simde_x_mm_deinterleaveeven_epi32(a, b), simde_x_mm_deinterleaveodd_epi32(a, b));
  370. #endif
  371. }
  372. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  373. # define _mm_hadd_epi32(a, b) simde_mm_hadd_epi32(a, b)
  374. #endif
  375. SIMDE_FUNCTION_ATTRIBUTES
  376. simde__m64
  377. simde_mm_hadd_pi16 (simde__m64 a, simde__m64 b) {
  378. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  379. return _mm_hadd_pi16(a, b);
  380. #else
  381. simde__m64_private
  382. r_,
  383. a_ = simde__m64_to_private(a),
  384. b_ = simde__m64_to_private(b);
  385. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  386. r_.neon_i16 = vpadd_s16(a_.neon_i16, b_.neon_i16);
  387. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  388. int16x4x2_t t = vuzp_s16(a_.neon_i16, b_.neon_i16);
  389. r_.neon_i16 = vadd_s16(t.val[0], t.val[1]);
  390. #elif (SIMDE_NATURAL_VECTOR_SIZE > 0) && defined(SIMDE_SHUFFLE_VECTOR_)
  391. r_.i16 =
  392. SIMDE_SHUFFLE_VECTOR_(16, 8, a_.i16, b_.i16, 0, 2, 4, 6) +
  393. SIMDE_SHUFFLE_VECTOR_(16, 8, a_.i16, b_.i16, 1, 3, 5, 7);
  394. #else
  395. r_.i16[0] = a_.i16[0] + a_.i16[1];
  396. r_.i16[1] = a_.i16[2] + a_.i16[3];
  397. r_.i16[2] = b_.i16[0] + b_.i16[1];
  398. r_.i16[3] = b_.i16[2] + b_.i16[3];
  399. #endif
  400. return simde__m64_from_private(r_);
  401. #endif
  402. }
  403. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  404. # define _mm_hadd_pi16(a, b) simde_mm_hadd_pi16(a, b)
  405. #endif
  406. SIMDE_FUNCTION_ATTRIBUTES
  407. simde__m64
  408. simde_mm_hadd_pi32 (simde__m64 a, simde__m64 b) {
  409. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  410. return _mm_hadd_pi32(a, b);
  411. #else
  412. simde__m64_private
  413. r_,
  414. a_ = simde__m64_to_private(a),
  415. b_ = simde__m64_to_private(b);
  416. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  417. r_.neon_i32 = vpadd_s32(a_.neon_i32, b_.neon_i32);
  418. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  419. int32x2x2_t t = vuzp_s32(a_.neon_i32, b_.neon_i32);
  420. r_.neon_i32 = vadd_s32(t.val[0], t.val[1]);
  421. #elif (SIMDE_NATURAL_VECTOR_SIZE > 0) && defined(SIMDE_SHUFFLE_VECTOR_)
  422. r_.i32 =
  423. SIMDE_SHUFFLE_VECTOR_(32, 8, a_.i32, b_.i32, 0, 2) +
  424. SIMDE_SHUFFLE_VECTOR_(32, 8, a_.i32, b_.i32, 1, 3);
  425. #else
  426. r_.i32[0] = a_.i32[0] + a_.i32[1];
  427. r_.i32[1] = b_.i32[0] + b_.i32[1];
  428. #endif
  429. return simde__m64_from_private(r_);
  430. #endif
  431. }
  432. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  433. # define _mm_hadd_pi32(a, b) simde_mm_hadd_pi32(a, b)
  434. #endif
  435. SIMDE_FUNCTION_ATTRIBUTES
  436. simde__m128i
  437. simde_mm_hadds_epi16 (simde__m128i a, simde__m128i b) {
  438. #if defined(SIMDE_X86_SSSE3_NATIVE)
  439. return _mm_hadds_epi16(a, b);
  440. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  441. int16x8x2_t t = vuzpq_s16(simde__m128i_to_neon_i16(a), simde__m128i_to_neon_i16(b));
  442. return simde__m128i_from_neon_i16(vqaddq_s16(t.val[0], t.val[1]));
  443. #else
  444. return simde_mm_adds_epi16(simde_x_mm_deinterleaveeven_epi16(a, b), simde_x_mm_deinterleaveodd_epi16(a, b));
  445. #endif
  446. }
  447. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  448. # define _mm_hadds_epi16(a, b) simde_mm_hadds_epi16(a, b)
  449. #endif
  450. SIMDE_FUNCTION_ATTRIBUTES
  451. simde__m64
  452. simde_mm_hadds_pi16 (simde__m64 a, simde__m64 b) {
  453. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  454. return _mm_hadds_pi16(a, b);
  455. #else
  456. simde__m64_private
  457. r_,
  458. a_ = simde__m64_to_private(a),
  459. b_ = simde__m64_to_private(b);
  460. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  461. int16x4x2_t t = vuzp_s16(a_.neon_i16, b_.neon_i16);
  462. r_.neon_i16 = vqadd_s16(t.val[0], t.val[1]);
  463. #else
  464. for (size_t i = 0 ; i < ((sizeof(r_.i16) / sizeof(r_.i16[0])) / 2) ; i++) {
  465. int32_t ta = HEDLEY_STATIC_CAST(int32_t, a_.i16[i * 2]) + HEDLEY_STATIC_CAST(int32_t, a_.i16[(i * 2) + 1]);
  466. r_.i16[ i ] = HEDLEY_LIKELY(ta > INT16_MIN) ? (HEDLEY_LIKELY(ta < INT16_MAX) ? HEDLEY_STATIC_CAST(int16_t, ta) : INT16_MAX) : INT16_MIN;
  467. int32_t tb = HEDLEY_STATIC_CAST(int32_t, b_.i16[i * 2]) + HEDLEY_STATIC_CAST(int32_t, b_.i16[(i * 2) + 1]);
  468. r_.i16[i + 2] = HEDLEY_LIKELY(tb > INT16_MIN) ? (HEDLEY_LIKELY(tb < INT16_MAX) ? HEDLEY_STATIC_CAST(int16_t, tb) : INT16_MAX) : INT16_MIN;
  469. }
  470. #endif
  471. return simde__m64_from_private(r_);
  472. #endif
  473. }
  474. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  475. # define _mm_hadds_pi16(a, b) simde_mm_hadds_pi16(a, b)
  476. #endif
  477. SIMDE_FUNCTION_ATTRIBUTES
  478. simde__m128i
  479. simde_mm_hsub_epi16 (simde__m128i a, simde__m128i b) {
  480. #if defined(SIMDE_X86_SSSE3_NATIVE)
  481. return _mm_hsub_epi16(a, b);
  482. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  483. int16x8x2_t t = vuzpq_s16(simde__m128i_to_neon_i16(a), simde__m128i_to_neon_i16(b));
  484. return simde__m128i_from_neon_i16(vsubq_s16(t.val[0], t.val[1]));
  485. #else
  486. return simde_mm_sub_epi16(simde_x_mm_deinterleaveeven_epi16(a, b), simde_x_mm_deinterleaveodd_epi16(a, b));
  487. #endif
  488. }
  489. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  490. # define _mm_hsub_epi16(a, b) simde_mm_hsub_epi16(a, b)
  491. #endif
  492. SIMDE_FUNCTION_ATTRIBUTES
  493. simde__m128i
  494. simde_mm_hsub_epi32 (simde__m128i a, simde__m128i b) {
  495. #if defined(SIMDE_X86_SSSE3_NATIVE)
  496. return _mm_hsub_epi32(a, b);
  497. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  498. int32x4x2_t t = vuzpq_s32(simde__m128i_to_neon_i32(a), simde__m128i_to_neon_i32(b));
  499. return simde__m128i_from_neon_i32(vsubq_s32(t.val[0], t.val[1]));
  500. #else
  501. return simde_mm_sub_epi32(simde_x_mm_deinterleaveeven_epi32(a, b), simde_x_mm_deinterleaveodd_epi32(a, b));
  502. #endif
  503. }
  504. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  505. # define _mm_hsub_epi32(a, b) simde_mm_hsub_epi32(a, b)
  506. #endif
  507. SIMDE_FUNCTION_ATTRIBUTES
  508. simde__m64
  509. simde_mm_hsub_pi16 (simde__m64 a, simde__m64 b) {
  510. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  511. return _mm_hsub_pi16(a, b);
  512. #else
  513. simde__m64_private
  514. r_,
  515. a_ = simde__m64_to_private(a),
  516. b_ = simde__m64_to_private(b);
  517. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  518. int16x4x2_t t = vuzp_s16(a_.neon_i16, b_.neon_i16);
  519. r_.neon_i16 = vsub_s16(t.val[0], t.val[1]);
  520. #elif (SIMDE_NATURAL_VECTOR_SIZE > 0) && defined(SIMDE_SHUFFLE_VECTOR_)
  521. r_.i16 =
  522. SIMDE_SHUFFLE_VECTOR_(16, 8, a_.i16, b_.i16, 0, 2, 4, 6) -
  523. SIMDE_SHUFFLE_VECTOR_(16, 8, a_.i16, b_.i16, 1, 3, 5, 7);
  524. #else
  525. r_.i16[0] = a_.i16[0] - a_.i16[1];
  526. r_.i16[1] = a_.i16[2] - a_.i16[3];
  527. r_.i16[2] = b_.i16[0] - b_.i16[1];
  528. r_.i16[3] = b_.i16[2] - b_.i16[3];
  529. #endif
  530. return simde__m64_from_private(r_);
  531. #endif
  532. }
  533. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  534. # define _mm_hsub_pi16(a, b) simde_mm_hsub_pi16(a, b)
  535. #endif
  536. SIMDE_FUNCTION_ATTRIBUTES
  537. simde__m64
  538. simde_mm_hsub_pi32 (simde__m64 a, simde__m64 b) {
  539. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  540. return _mm_hsub_pi32(a, b);
  541. #else
  542. simde__m64_private
  543. r_,
  544. a_ = simde__m64_to_private(a),
  545. b_ = simde__m64_to_private(b);
  546. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  547. int32x2x2_t t = vuzp_s32(a_.neon_i32, b_.neon_i32);
  548. r_.neon_i32 = vsub_s32(t.val[0], t.val[1]);
  549. #elif (SIMDE_NATURAL_VECTOR_SIZE > 0) && defined(SIMDE_SHUFFLE_VECTOR_)
  550. r_.i32 =
  551. SIMDE_SHUFFLE_VECTOR_(32, 8, a_.i32, b_.i32, 0, 2) -
  552. SIMDE_SHUFFLE_VECTOR_(32, 8, a_.i32, b_.i32, 1, 3);
  553. #else
  554. r_.i32[0] = a_.i32[0] - a_.i32[1];
  555. r_.i32[1] = b_.i32[0] - b_.i32[1];
  556. #endif
  557. return simde__m64_from_private(r_);
  558. #endif
  559. }
  560. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  561. # define _mm_hsub_pi32(a, b) simde_mm_hsub_pi32(a, b)
  562. #endif
  563. SIMDE_FUNCTION_ATTRIBUTES
  564. simde__m128i
  565. simde_mm_hsubs_epi16 (simde__m128i a, simde__m128i b) {
  566. #if defined(SIMDE_X86_SSSE3_NATIVE)
  567. return _mm_hsubs_epi16(a, b);
  568. #elif defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  569. int16x8x2_t t = vuzpq_s16(simde__m128i_to_neon_i16(a), simde__m128i_to_neon_i16(b));
  570. return simde__m128i_from_neon_i16(vqsubq_s16(t.val[0], t.val[1]));
  571. #else
  572. return simde_mm_subs_epi16(simde_x_mm_deinterleaveeven_epi16(a, b), simde_x_mm_deinterleaveodd_epi16(a, b));
  573. #endif
  574. }
  575. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  576. # define _mm_hsubs_epi16(a, b) simde_mm_hsubs_epi16(a, b)
  577. #endif
  578. SIMDE_FUNCTION_ATTRIBUTES
  579. simde__m64
  580. simde_mm_hsubs_pi16 (simde__m64 a, simde__m64 b) {
  581. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  582. return _mm_hsubs_pi16(a, b);
  583. #else
  584. simde__m64_private
  585. r_,
  586. a_ = simde__m64_to_private(a),
  587. b_ = simde__m64_to_private(b);
  588. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  589. int16x4x2_t t = vuzp_s16(a_.neon_i16, b_.neon_i16);
  590. r_.neon_i16 = vqsub_s16(t.val[0], t.val[1]);
  591. #else
  592. for (size_t i = 0 ; i < ((sizeof(r_.i16) / sizeof(r_.i16[0])) / 2) ; i++) {
  593. r_.i16[ i ] = simde_math_subs_i16(a_.i16[i * 2], a_.i16[(i * 2) + 1]);
  594. r_.i16[i + 2] = simde_math_subs_i16(b_.i16[i * 2], b_.i16[(i * 2) + 1]);
  595. }
  596. #endif
  597. return simde__m64_from_private(r_);
  598. #endif
  599. }
  600. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  601. # define _mm_hsubs_pi16(a, b) simde_mm_hsubs_pi16(a, b)
  602. #endif
  603. SIMDE_FUNCTION_ATTRIBUTES
  604. simde__m128i
  605. simde_mm_maddubs_epi16 (simde__m128i a, simde__m128i b) {
  606. #if defined(SIMDE_X86_SSSE3_NATIVE)
  607. return _mm_maddubs_epi16(a, b);
  608. #else
  609. simde__m128i_private
  610. r_,
  611. a_ = simde__m128i_to_private(a),
  612. b_ = simde__m128i_to_private(b);
  613. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  614. /* Zero extend a */
  615. int16x8_t a_odd = vreinterpretq_s16_u16(vshrq_n_u16(a_.neon_u16, 8));
  616. int16x8_t a_even = vreinterpretq_s16_u16(vbicq_u16(a_.neon_u16, vdupq_n_u16(0xff00)));
  617. /* Sign extend by shifting left then shifting right. */
  618. int16x8_t b_even = vshrq_n_s16(vshlq_n_s16(b_.neon_i16, 8), 8);
  619. int16x8_t b_odd = vshrq_n_s16(b_.neon_i16, 8);
  620. /* multiply */
  621. int16x8_t prod1 = vmulq_s16(a_even, b_even);
  622. int16x8_t prod2 = vmulq_s16(a_odd, b_odd);
  623. /* saturated add */
  624. r_.neon_i16 = vqaddq_s16(prod1, prod2);
  625. #else
  626. for (size_t i = 0 ; i < (sizeof(r_.i16) / sizeof(r_.i16[0])) ; i++) {
  627. const int idx = HEDLEY_STATIC_CAST(int, i) << 1;
  628. int32_t ts =
  629. (HEDLEY_STATIC_CAST(int16_t, a_.u8[ idx ]) * HEDLEY_STATIC_CAST(int16_t, b_.i8[ idx ])) +
  630. (HEDLEY_STATIC_CAST(int16_t, a_.u8[idx + 1]) * HEDLEY_STATIC_CAST(int16_t, b_.i8[idx + 1]));
  631. r_.i16[i] = (ts > INT16_MIN) ? ((ts < INT16_MAX) ? HEDLEY_STATIC_CAST(int16_t, ts) : INT16_MAX) : INT16_MIN;
  632. }
  633. #endif
  634. return simde__m128i_from_private(r_);
  635. #endif
  636. }
  637. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  638. # define _mm_maddubs_epi16(a, b) simde_mm_maddubs_epi16(a, b)
  639. #endif
  640. SIMDE_FUNCTION_ATTRIBUTES
  641. simde__m64
  642. simde_mm_maddubs_pi16 (simde__m64 a, simde__m64 b) {
  643. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  644. return _mm_maddubs_pi16(a, b);
  645. #else
  646. simde__m64_private
  647. r_,
  648. a_ = simde__m64_to_private(a),
  649. b_ = simde__m64_to_private(b);
  650. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  651. int16x8_t ai = vreinterpretq_s16_u16(vmovl_u8(a_.neon_u8));
  652. int16x8_t bi = vmovl_s8(b_.neon_i8);
  653. int16x8_t p = vmulq_s16(ai, bi);
  654. int16x4_t l = vget_low_s16(p);
  655. int16x4_t h = vget_high_s16(p);
  656. r_.neon_i16 = vqadd_s16(vuzp1_s16(l, h), vuzp2_s16(l, h));
  657. #else
  658. for (size_t i = 0 ; i < (sizeof(r_.i16) / sizeof(r_.i16[0])) ; i++) {
  659. const int idx = HEDLEY_STATIC_CAST(int, i) << 1;
  660. int32_t ts =
  661. (HEDLEY_STATIC_CAST(int16_t, a_.u8[ idx ]) * HEDLEY_STATIC_CAST(int16_t, b_.i8[ idx ])) +
  662. (HEDLEY_STATIC_CAST(int16_t, a_.u8[idx + 1]) * HEDLEY_STATIC_CAST(int16_t, b_.i8[idx + 1]));
  663. r_.i16[i] = (ts > INT16_MIN) ? ((ts < INT16_MAX) ? HEDLEY_STATIC_CAST(int16_t, ts) : INT16_MAX) : INT16_MIN;
  664. }
  665. #endif
  666. return simde__m64_from_private(r_);
  667. #endif
  668. }
  669. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  670. # define _mm_maddubs_pi16(a, b) simde_mm_maddubs_pi16(a, b)
  671. #endif
  672. SIMDE_FUNCTION_ATTRIBUTES
  673. simde__m128i
  674. simde_mm_mulhrs_epi16 (simde__m128i a, simde__m128i b) {
  675. #if defined(SIMDE_X86_SSSE3_NATIVE)
  676. return _mm_mulhrs_epi16(a, b);
  677. #else
  678. simde__m128i_private
  679. r_,
  680. a_ = simde__m128i_to_private(a),
  681. b_ = simde__m128i_to_private(b);
  682. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  683. /* Multiply */
  684. int32x4_t mul_lo = vmull_s16(vget_low_s16(a_.neon_i16),
  685. vget_low_s16(b_.neon_i16));
  686. int32x4_t mul_hi = vmull_s16(vget_high_s16(a_.neon_i16),
  687. vget_high_s16(b_.neon_i16));
  688. /* Rounding narrowing shift right
  689. * narrow = (int16_t)((mul + 16384) >> 15); */
  690. int16x4_t narrow_lo = vrshrn_n_s32(mul_lo, 15);
  691. int16x4_t narrow_hi = vrshrn_n_s32(mul_hi, 15);
  692. /* Join together */
  693. r_.neon_i16 = vcombine_s16(narrow_lo, narrow_hi);
  694. #elif defined(SIMDE_WASM_SIMD128_NATIVE)
  695. v128_t __lo = wasm_i32x4_mul(wasm_i32x4_extend_low_i16x8(a_.wasm_v128), wasm_i32x4_extend_low_i16x8(b_.wasm_v128));
  696. v128_t __hi = wasm_i32x4_mul(wasm_i32x4_extend_high_i16x8(a_.wasm_v128), wasm_i32x4_extend_high_i16x8(b_.wasm_v128));
  697. const v128_t __inc = wasm_i32x4_splat(0x4000);
  698. __lo = wasm_i32x4_add(__lo, __inc);
  699. __hi = wasm_i32x4_add(__hi, __inc);
  700. __lo = wasm_i32x4_add(__lo, __lo);
  701. __hi = wasm_i32x4_add(__hi, __hi);
  702. r_.wasm_v128 = wasm_i16x8_shuffle(__lo, __hi, 1, 3, 5, 7, 9, 11, 13, 15);
  703. #else
  704. SIMDE_VECTORIZE
  705. for (size_t i = 0 ; i < (sizeof(r_.i16) / sizeof(r_.i16[0])) ; i++) {
  706. r_.i16[i] = HEDLEY_STATIC_CAST(int16_t, (((HEDLEY_STATIC_CAST(int32_t, a_.i16[i]) * HEDLEY_STATIC_CAST(int32_t, b_.i16[i])) + 0x4000) >> 15));
  707. }
  708. #endif
  709. return simde__m128i_from_private(r_);
  710. #endif
  711. }
  712. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  713. # define _mm_mulhrs_epi16(a, b) simde_mm_mulhrs_epi16(a, b)
  714. #endif
  715. SIMDE_FUNCTION_ATTRIBUTES
  716. simde__m64
  717. simde_mm_mulhrs_pi16 (simde__m64 a, simde__m64 b) {
  718. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  719. return _mm_mulhrs_pi16(a, b);
  720. #else
  721. simde__m64_private
  722. r_,
  723. a_ = simde__m64_to_private(a),
  724. b_ = simde__m64_to_private(b);
  725. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  726. /* Multiply */
  727. int32x4_t mul = vmull_s16(a_.neon_i16, b_.neon_i16);
  728. /* Rounding narrowing shift right
  729. * narrow = (int16_t)((mul + 16384) >> 15); */
  730. int16x4_t narrow = vrshrn_n_s32(mul, 15);
  731. /* Join together */
  732. r_.neon_i16 = narrow;
  733. #else
  734. SIMDE_VECTORIZE
  735. for (size_t i = 0 ; i < (sizeof(r_.i16) / sizeof(r_.i16[0])) ; i++) {
  736. r_.i16[i] = HEDLEY_STATIC_CAST(int16_t, (((HEDLEY_STATIC_CAST(int32_t, a_.i16[i]) * HEDLEY_STATIC_CAST(int32_t, b_.i16[i])) + 0x4000) >> 15));
  737. }
  738. #endif
  739. return simde__m64_from_private(r_);
  740. #endif
  741. }
  742. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  743. # define _mm_mulhrs_pi16(a, b) simde_mm_mulhrs_pi16(a, b)
  744. #endif
  745. SIMDE_FUNCTION_ATTRIBUTES
  746. simde__m128i
  747. simde_mm_sign_epi8 (simde__m128i a, simde__m128i b) {
  748. #if defined(SIMDE_X86_SSSE3_NATIVE)
  749. return _mm_sign_epi8(a, b);
  750. #else
  751. simde__m128i_private
  752. r_,
  753. a_ = simde__m128i_to_private(a),
  754. b_ = simde__m128i_to_private(b);
  755. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  756. uint8x16_t aneg_mask = vreinterpretq_u8_s8(vshrq_n_s8(b_.neon_i8, 7));
  757. uint8x16_t bnz_mask;
  758. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  759. bnz_mask = vceqzq_s8(b_.neon_i8);
  760. #else
  761. bnz_mask = vceqq_s8(b_.neon_i8, vdupq_n_s8(0));
  762. #endif
  763. bnz_mask = vmvnq_u8(bnz_mask);
  764. r_.neon_i8 = vbslq_s8(aneg_mask, vnegq_s8(a_.neon_i8), vandq_s8(a_.neon_i8, vreinterpretq_s8_u8(bnz_mask)));
  765. #elif defined(SIMDE_WASM_SIMD128_NATIVE)
  766. simde__m128i mask = wasm_i8x16_shr(b_.wasm_v128, 7);
  767. simde__m128i zeromask = simde_mm_cmpeq_epi8(b_.wasm_v128, simde_mm_setzero_si128());
  768. r_.wasm_v128 = simde_mm_andnot_si128(zeromask, simde_mm_xor_si128(simde_mm_add_epi8(a_.wasm_v128, mask), mask));
  769. #else
  770. SIMDE_VECTORIZE
  771. for (size_t i = 0 ; i < (sizeof(r_.i8) / sizeof(r_.i8[0])) ; i++) {
  772. r_.i8[i] = (b_.i8[i] < 0) ? (- a_.i8[i]) : ((b_.i8[i] != 0) ? (a_.i8[i]) : INT8_C(0));
  773. }
  774. #endif
  775. return simde__m128i_from_private(r_);
  776. #endif
  777. }
  778. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  779. # define _mm_sign_epi8(a, b) simde_mm_sign_epi8(a, b)
  780. #endif
  781. SIMDE_FUNCTION_ATTRIBUTES
  782. simde__m128i
  783. simde_mm_sign_epi16 (simde__m128i a, simde__m128i b) {
  784. #if defined(SIMDE_X86_SSSE3_NATIVE)
  785. return _mm_sign_epi16(a, b);
  786. #else
  787. simde__m128i_private
  788. r_,
  789. a_ = simde__m128i_to_private(a),
  790. b_ = simde__m128i_to_private(b);
  791. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  792. uint16x8_t aneg_mask = vreinterpretq_u16_s16(vshrq_n_s16(b_.neon_i16, 15));
  793. uint16x8_t bnz_mask;
  794. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  795. bnz_mask = vceqzq_s16(b_.neon_i16);
  796. #else
  797. bnz_mask = vceqq_s16(b_.neon_i16, vdupq_n_s16(0));
  798. #endif
  799. bnz_mask = vmvnq_u16(bnz_mask);
  800. r_.neon_i16 = vbslq_s16(aneg_mask, vnegq_s16(a_.neon_i16), vandq_s16(a_.neon_i16, vreinterpretq_s16_u16(bnz_mask)));
  801. #elif defined(SIMDE_WASM_SIMD128_NATIVE)
  802. simde__m128i mask = simde_mm_srai_epi16(b_.wasm_v128, 15);
  803. simde__m128i zeromask = simde_mm_cmpeq_epi16(b_.wasm_v128, simde_mm_setzero_si128());
  804. r_.wasm_v128 = simde_mm_andnot_si128(zeromask, simde_mm_xor_si128(simde_mm_add_epi16(a_.wasm_v128, mask), mask));
  805. #else
  806. SIMDE_VECTORIZE
  807. for (size_t i = 0 ; i < (sizeof(r_.i16) / sizeof(r_.i16[0])) ; i++) {
  808. r_.i16[i] = (b_.i16[i] < 0) ? (- a_.i16[i]) : ((b_.i16[i] != 0) ? (a_.i16[i]) : INT16_C(0));
  809. }
  810. #endif
  811. return simde__m128i_from_private(r_);
  812. #endif
  813. }
  814. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  815. # define _mm_sign_epi16(a, b) simde_mm_sign_epi16(a, b)
  816. #endif
  817. SIMDE_FUNCTION_ATTRIBUTES
  818. simde__m128i
  819. simde_mm_sign_epi32 (simde__m128i a, simde__m128i b) {
  820. #if defined(SIMDE_X86_SSSE3_NATIVE)
  821. return _mm_sign_epi32(a, b);
  822. #else
  823. simde__m128i_private
  824. r_,
  825. a_ = simde__m128i_to_private(a),
  826. b_ = simde__m128i_to_private(b);
  827. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  828. uint32x4_t aneg_mask = vreinterpretq_u32_s32(vshrq_n_s32(b_.neon_i32, 31));
  829. uint32x4_t bnz_mask;
  830. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  831. bnz_mask = vceqzq_s32(b_.neon_i32);
  832. #else
  833. bnz_mask = vceqq_s32(b_.neon_i32, vdupq_n_s32(0));
  834. #endif
  835. bnz_mask = vmvnq_u32(bnz_mask);
  836. r_.neon_i32 = vbslq_s32(aneg_mask, vnegq_s32(a_.neon_i32), vandq_s32(a_.neon_i32, vreinterpretq_s32_u32(bnz_mask)));
  837. #elif defined(SIMDE_WASM_SIMD128_NATIVE)
  838. simde__m128i mask = simde_mm_srai_epi32(b_.wasm_v128, 31);
  839. simde__m128i zeromask = simde_mm_cmpeq_epi32(b_.wasm_v128, simde_mm_setzero_si128());
  840. r_.wasm_v128 = simde_mm_andnot_si128(zeromask, simde_mm_xor_si128(simde_mm_add_epi32(a_.wasm_v128, mask), mask));
  841. #else
  842. for (size_t i = 0 ; i < (sizeof(r_.i32) / sizeof(r_.i32[0])) ; i++) {
  843. r_.i32[i] = (b_.i32[i] < 0) ? (- a_.i32[i]) : ((b_.i32[i] != 0) ? (a_.i32[i]) : INT32_C(0));
  844. }
  845. #endif
  846. return simde__m128i_from_private(r_);
  847. #endif
  848. }
  849. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  850. # define _mm_sign_epi32(a, b) simde_mm_sign_epi32(a, b)
  851. #endif
  852. SIMDE_FUNCTION_ATTRIBUTES
  853. simde__m64
  854. simde_mm_sign_pi8 (simde__m64 a, simde__m64 b) {
  855. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  856. return _mm_sign_pi8(a, b);
  857. #else
  858. simde__m64_private
  859. r_,
  860. a_ = simde__m64_to_private(a),
  861. b_ = simde__m64_to_private(b);
  862. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  863. uint8x8_t aneg_mask = vreinterpret_u8_s8(vshr_n_s8(b_.neon_i8, 7));
  864. uint8x8_t bnz_mask;
  865. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  866. bnz_mask = vceqz_s8(b_.neon_i8);
  867. #else
  868. bnz_mask = vceq_s8(b_.neon_i8, vdup_n_s8(0));
  869. #endif
  870. bnz_mask = vmvn_u8(bnz_mask);
  871. r_.neon_i8 = vbsl_s8(aneg_mask, vneg_s8(a_.neon_i8), vand_s8(a_.neon_i8, vreinterpret_s8_u8(bnz_mask)));
  872. #else
  873. SIMDE_VECTORIZE
  874. for (size_t i = 0 ; i < (sizeof(r_.i8) / sizeof(r_.i8[0])) ; i++) {
  875. r_.i8[i] = (b_.i8[i] < 0) ? (- a_.i8[i]) : ((b_.i8[i] != 0) ? (a_.i8[i]) : INT8_C(0));
  876. }
  877. #endif
  878. return simde__m64_from_private(r_);
  879. #endif
  880. }
  881. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  882. # define _mm_sign_pi8(a, b) simde_mm_sign_pi8(a, b)
  883. #endif
  884. SIMDE_FUNCTION_ATTRIBUTES
  885. simde__m64
  886. simde_mm_sign_pi16 (simde__m64 a, simde__m64 b) {
  887. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  888. return _mm_sign_pi16(a, b);
  889. #else
  890. simde__m64_private
  891. r_,
  892. a_ = simde__m64_to_private(a),
  893. b_ = simde__m64_to_private(b);
  894. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  895. uint16x4_t aneg_mask = vreinterpret_u16_s16(vshr_n_s16(b_.neon_i16, 15));
  896. uint16x4_t bnz_mask;
  897. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  898. bnz_mask = vceqz_s16(b_.neon_i16);
  899. #else
  900. bnz_mask = vceq_s16(b_.neon_i16, vdup_n_s16(0));
  901. #endif
  902. bnz_mask = vmvn_u16(bnz_mask);
  903. r_.neon_i16 = vbsl_s16(aneg_mask, vneg_s16(a_.neon_i16), vand_s16(a_.neon_i16, vreinterpret_s16_u16(bnz_mask)));
  904. #else
  905. SIMDE_VECTORIZE
  906. for (size_t i = 0 ; i < (sizeof(r_.i16) / sizeof(r_.i16[0])) ; i++) {
  907. r_.i16[i] = (b_.i16[i] < 0) ? (- a_.i16[i]) : ((b_.i16[i] > 0) ? (a_.i16[i]) : INT16_C(0));
  908. }
  909. #endif
  910. return simde__m64_from_private(r_);
  911. #endif
  912. }
  913. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  914. # define _mm_sign_pi16(a, b) simde_mm_sign_pi16(a, b)
  915. #endif
  916. SIMDE_FUNCTION_ATTRIBUTES
  917. simde__m64
  918. simde_mm_sign_pi32 (simde__m64 a, simde__m64 b) {
  919. #if defined(SIMDE_X86_SSSE3_NATIVE) && defined(SIMDE_X86_MMX_NATIVE)
  920. return _mm_sign_pi32(a, b);
  921. #else
  922. simde__m64_private
  923. r_,
  924. a_ = simde__m64_to_private(a),
  925. b_ = simde__m64_to_private(b);
  926. #if defined(SIMDE_ARM_NEON_A32V7_NATIVE)
  927. uint32x2_t aneg_mask = vreinterpret_u32_s32(vshr_n_s32(b_.neon_i32, 31));
  928. uint32x2_t bnz_mask;
  929. #if defined(SIMDE_ARM_NEON_A64V8_NATIVE)
  930. bnz_mask = vceqz_s32(b_.neon_i32);
  931. #else
  932. bnz_mask = vceq_s32(b_.neon_i32, vdup_n_s32(0));
  933. #endif
  934. bnz_mask = vmvn_u32(bnz_mask);
  935. r_.neon_i32 = vbsl_s32(aneg_mask, vneg_s32(a_.neon_i32), vand_s32(a_.neon_i32, vreinterpret_s32_u32(bnz_mask)));
  936. #else
  937. for (size_t i = 0 ; i < (sizeof(r_.i32) / sizeof(r_.i32[0])) ; i++) {
  938. r_.i32[i] = (b_.i32[i] < 0) ? (- a_.i32[i]) : ((b_.i32[i] > 0) ? (a_.i32[i]) : INT32_C(0));
  939. }
  940. #endif
  941. return simde__m64_from_private(r_);
  942. #endif
  943. }
  944. #if defined(SIMDE_X86_SSSE3_ENABLE_NATIVE_ALIASES)
  945. # define _mm_sign_pi32(a, b) simde_mm_sign_pi32(a, b)
  946. #endif
  947. SIMDE_END_DECLS_
  948. HEDLEY_DIAGNOSTIC_POP
  949. #endif /* !defined(SIMDE_X86_SSE2_H) */