gcc/testsuite/gcc.target/i386/sse2-mmx.c

   1 /* { dg-do run } */
   2 /* { dg-options "-O2 -msse2" } */
   3
   4 #include "sse2-check.h"
   5
   6 #include <mmintrin.h>
   7
   8 #define N 4
   9
  10 unsigned long long a[N], b[N], result[N];
  11
  12 unsigned long long check[N] =
  13   { 0x101010101010100full,
  14     0x1010101010101010ull,
  15     0x1010101010101010ull,
  16     0x1010101010101010ull };
  17
  18 __m64
  19 unsigned_add3 (const __m64 * a, const __m64 * b,
  20                __m64 * result, unsigned int count)
  21 {
  22   __m64 _a, _b, one, sum, carry, onesCarry;
  23
  24   unsigned int i;
  25
  26   one = _mm_cmpeq_pi8 (_a, _a);
  27   one = _mm_sub_si64 (_mm_xor_si64 (one, one), one);
  28
  29   carry = _mm_xor_si64 (one, one);
  30
  31   for (i = 0; i < count; i++)
  32     {
  33       _a = a[i];
  34       _b = b[i];
  35
  36       sum = _mm_add_si64 (_a, _b);
  37       sum = _mm_add_si64 (sum, carry);
  38
  39       result[i] = sum;
  40
  41       onesCarry = _mm_and_si64 (_mm_xor_si64 (_a, _b), carry);
  42       onesCarry = _mm_or_si64 (_mm_and_si64 (_a, _b), onesCarry);
  43       onesCarry = _mm_and_si64 (onesCarry, one);
  44
  45       _a = _mm_srli_si64 (_a, 1);
  46       _b = _mm_srli_si64 (_b, 1);
  47
  48       carry = _mm_add_si64 (_mm_add_si64 (_a, _b), onesCarry);
  49       carry = _mm_srli_si64 (carry, 63);
  50     }
  51
  52   _mm_empty ();
  53   return carry;
  54 }
  55
  56 void __attribute__((noinline))
  57 sse2_test (void)
  58 {
  59   unsigned long long carry;
  60   int i;
  61
  62   /* Really long numbers.  */
  63   a[3] = a[2] = a[1] = a[0] = 0xd3d3d3d3d3d3d3d3ull;
  64   b[3] = b[2] = b[1] = b[0] = 0x3c3c3c3c3c3c3c3cull;
  65
  66   carry = (unsigned long long) unsigned_add3
  67     ((__m64 *)a, (__m64 *)b, (__m64 *)result, N);
  68
  69   if (carry != 1)
  70     abort ();
  71
  72   for (i = 0; i < N; i++)
  73     if (result [i] != check[i])
  74       abort ();
  75 }