simd_ssse3.c 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. #include<simdconfig.h>
  2. #include<simdfuncs.h>
  3. #include<emmintrin.h>
  4. #include<tmmintrin.h>
  5. #ifdef _MSC_VER
  6. #include<intrin.h>
  7. int ssse3_available() {
  8. return 1;
  9. }
  10. #else
  11. #include<cpuid.h>
  12. #include<stdint.h>
  13. int ssse3_available() {
  14. #ifdef __APPLE__
  15. return 1;
  16. #elif defined(__clang__)
  17. /* https://github.com/numpy/numpy/issues/8130 */
  18. return __builtin_cpu_supports("sse4.1");
  19. #else
  20. return __builtin_cpu_supports("ssse3");
  21. #endif
  22. }
  23. #endif
  24. void increment_ssse3(float arr[4]) {
  25. double darr[4];
  26. __m128d val1 = _mm_set_pd(arr[0], arr[1]);
  27. __m128d val2 = _mm_set_pd(arr[2], arr[3]);
  28. __m128d one = _mm_set_pd(1.0, 1.0);
  29. __m128d result = _mm_add_pd(val1, one);
  30. __m128i tmp1, tmp2;
  31. tmp1 = tmp2 = _mm_set1_epi16(0);
  32. _mm_store_pd(darr, result);
  33. result = _mm_add_pd(val2, one);
  34. _mm_store_pd(&darr[2], result);
  35. tmp1 = _mm_hadd_epi32(tmp1, tmp2); /* This does nothing. Only here so we use an SSSE3 instruction. */
  36. arr[0] = (float)darr[1];
  37. arr[1] = (float)darr[0];
  38. arr[2] = (float)darr[3];
  39. arr[3] = (float)darr[2];
  40. }