gcc/testsuite/gcc.target/i386/ssse3-pshufb.c

   1 /* { dg-do run { target i?86-*-* x86_64-*-* } } */
   2 /* { dg-options "-O2 -mssse3" } */
   3 #include <tmmintrin.h>
   4 #include <stdlib.h>
   5 #include "../../gcc.dg/i386-cpuid.h"
   6 #include "ssse3-vals.h"
   7
   8 static void ssse3_test (void);
   9
  10 int
  11 main ()
  12 {
  13   unsigned long cpu_facilities;
  14
  15   cpu_facilities = i386_cpuid_ecx ();
  16
  17   /* Run SSSE3 test only if host has SSSE3 support.  */
  18   if ((cpu_facilities & bit_SSSE3))
  19     ssse3_test ();
  20
  21   exit (0);
  22 }
  23
  24 /* Test the 64-bit form */
  25 static void
  26 ssse3_test_pshufb (int *i1, int *i2, int *r)
  27 {
  28   __m64 t1 = *(__m64 *) i1;
  29   __m64 t2 = *(__m64 *) i2;
  30   *(__m64 *)r = _mm_shuffle_pi8 (t1, t2);
  31   _mm_empty ();
  32 }
  33
  34 /* Test the 128-bit form */
  35 static void
  36 ssse3_test_pshufb128 (int *i1, int *i2, int *r)
  37 {
  38   /* Assumes incoming pointers are 16-byte aligned */
  39   __m128i t1 = *(__m128i *) i1;
  40   __m128i t2 = *(__m128i *) i2;
  41   *(__m128i *)r = _mm_shuffle_epi8 (t1, t2);
  42 }
  43
  44 /* Routine to manually compute the results */
  45 static void
  46 compute_correct_result_64 (int *i1, int *i2, int *r)
  47 {
  48   char *b1 = (char *) i1;
  49   char *b2 = (char *) i2;
  50   char *bout = (char *) r;
  51   int i;
  52   char select;
  53
  54   for (i = 0; i < 16; i++)
  55     {
  56       select = b2[i];
  57       if (select & 0x80)
  58         bout[i] = 0;
  59       else if (i < 8)
  60         bout[i] = b1[select & 0x7];
  61       else
  62         bout[i] = b1[8 + (select & 0x7)];
  63     }
  64 }
  65
  66 static void
  67 compute_correct_result_128 (int *i1, int *i2, int *r)
  68 {
  69   char *b1 = (char *) i1;
  70   char *b2 = (char *) i2;
  71   char *bout = (char *) r;
  72   int i;
  73   char select;
  74
  75   for (i = 0; i < 16; i++)
  76     {
  77       select = b2[i];
  78       if (select & 0x80)
  79         bout[i] = 0;
  80       else
  81         bout[i] = b1[select & 0xf];
  82     }
  83 }
  84
  85 static void
  86 ssse3_test (void)
  87 {
  88   int i;
  89   int r [4] __attribute__ ((aligned(16)));
  90   int ck [4];
  91   int fail = 0;
  92
  93   for (i = 0; i < 256; i += 8)
  94     {
  95       /* Manually compute the result */
  96       compute_correct_result_64 (&vals[i + 0], &vals[i + 4], ck);
  97
  98       /* Run the 64-bit tests */
  99       ssse3_test_pshufb (&vals[i + 0], &vals[i + 4], &r[0]);
 100       ssse3_test_pshufb (&vals[i + 2], &vals[i + 6], &r[2]);
 101       fail += chk_128 (ck, r);
 102
 103       /* Recompute the result for 128-bits */
 104       compute_correct_result_128 (&vals[i + 0], &vals[i + 4], ck);
 105
 106       /* Run the 128-bit tests */
 107       ssse3_test_pshufb128 (&vals[i + 0], &vals[i + 4], r);
 108       fail += chk_128 (ck, r);
 109     }
 110
 111   if (fail != 0)
 112     abort ();
 113 }