gcc/testsuite/gcc.target/i386/avx512bw-vdbpsadbw-2.c

   1 /* { dg-do run } */
   2 /* { dg-options "-O2 -mavx512bw -DAVX512BW" } */
   3 /* { dg-require-effective-target avx512bw } */
   4
   5 #include "avx512f-helper.h"
   6
   7 #define SIZE (AVX512F_LEN / 16)
   8 #include "avx512f-mask-type.h"
   9
  10 void
  11 CALC (unsigned short *dst, unsigned char *src1, unsigned char *src2,
  12       int imm)
  13 {
  14   int i, j, k, part, power;
  15   unsigned char tmp[2 * SIZE];;
  16
  17   for (i = 0; i < 2 * SIZE; i += 16)
  18     {
  19       for (j = 0; j < 4; j++)
  20         {
  21           power = 1;
  22           for (k = 0; k < j; k++)
  23             power *= 4;
  24           part = (imm & (3 * power)) >> (2 * j);
  25           for (k = 0; k < 4; k++)
  26             tmp[i + 4 * j + k] = src2[i + 4 * part + k];
  27         }
  28     }
  29
  30   for (i = 0; i < SIZE; i += 4)
  31     {
  32       dst[i] = dst[i + 1] = dst[i + 2] = dst[i + 3] = 0;
  33       for (j = 0; j < 4; j++)
  34         {
  35           dst[i] += abs (src1[2 * i + j] - tmp[2 * i + j]);
  36           dst[i + 1] += abs (src1[2 * i + j] - tmp[2 * i + j + 1]);
  37           dst[i + 2] += abs (src1[2 * i + j + 4] - tmp[2 * i + j + 2]);
  38           dst[i + 3] += abs (src1[2 * i + j + 4] - tmp[2 * i + j + 3]);
  39         }
  40     }
  41 }
  42
  43 void
  44 TEST (void)
  45 {
  46   int i, sign;
  47   UNION_TYPE (AVX512F_LEN, i_w) res1, res2, res3;
  48   UNION_TYPE (AVX512F_LEN, i_b) src1, src2;
  49   MASK_TYPE mask = MASK_VALUE;
  50   unsigned short res_ref[SIZE];
  51   int imm = 0x22;
  52
  53   sign = -1;
  54   for (i = 0; i < 2*SIZE; i++)
  55     {
  56       src1.a[i] = 1 + 34 * i * sign;
  57       src1.a[i] = 179 - i;
  58       sign = sign * -1;
  59     }
  60
  61   for (i = 0; i < SIZE; i++)
  62       res2.a[i] = DEFAULT_VALUE;
  63
  64   res1.x = INTRINSIC (_dbsad_epu8) (src1.x, src2.x, imm);
  65   res2.x = INTRINSIC (_mask_dbsad_epu8) (res2.x, mask, src1.x, src2.x, imm);
  66   res3.x = INTRINSIC (_maskz_dbsad_epu8) (mask, src1.x, src2.x, imm);
  67
  68   CALC (res_ref, src1.a, src2.a, imm);
  69
  70   if (UNION_CHECK (AVX512F_LEN, i_w) (res1, res_ref))
  71     abort ();
  72
  73   MASK_MERGE (i_w) (res_ref, mask, SIZE);
  74   if (UNION_CHECK (AVX512F_LEN, i_w) (res2, res_ref))
  75     abort ();
  76
  77   MASK_ZERO (i_w) (res_ref, mask, SIZE);
  78   if (UNION_CHECK (AVX512F_LEN, i_w) (res3, res_ref))
  79     abort ();
  80 }