gcc/testsuite/gcc.target/i386/avx512f-vfmaddsubXXXps-2.c

   1 /* { dg-do run } */
   2 /* { dg-options "-O2 -mavx512f" } */
   3 /* { dg-require-effective-target avx512f } */
   4
   5 #define AVX512F
   6
   7 #include "avx512f-helper.h"
   8
   9 #define SIZE (AVX512F_LEN / 32)
  10 #include "avx512f-mask-type.h"
  11
  12 static void
  13 CALC (float *s1, float *s2, float *s3, float *r)
  14 {
  15   int i;
  16
  17   for (i = 0; i < SIZE; i++)
  18     {
  19       if (i % 2)
  20         r[i] = s1[i] * s2[i] + s3[i];
  21       else
  22         r[i] = s1[i] * s2[i] - s3[i];
  23     }
  24 }
  25
  26 void
  27 TEST (void)
  28 {
  29   UNION_TYPE (AVX512F_LEN, ) s1, s2, s3, res1, res2, res3, res4;
  30   MASK_TYPE mask = MASK_VALUE;
  31   float res_ref1[SIZE];
  32   float res_ref2[SIZE];
  33   int i, sign = 1;
  34
  35   for (i = 0; i < SIZE; i++)
  36     {
  37       s1.a[i] = DEFAULT_VALUE;
  38       s2.a[i] = 56.78 * (i + 1) * sign;
  39       s3.a[i] = 90.12 * (i + 2) * sign;
  40       sign = -sign;
  41     }
  42
  43 #if AVX512F_LEN == 512
  44   res1.x = INTRINSIC (_fmaddsub_ps) (s1.x, s2.x, s3.x);
  45 #endif
  46   res2.x = INTRINSIC (_mask_fmaddsub_ps) (s1.x, mask, s2.x, s3.x);
  47   res3.x = INTRINSIC (_mask3_fmaddsub_ps) (s2.x, s3.x, s1.x, mask);
  48   res4.x = INTRINSIC (_maskz_fmaddsub_ps) (mask, s1.x, s2.x, s3.x);
  49
  50   CALC (s1.a, s2.a, s3.a, res_ref1);
  51   CALC (s2.a, s3.a, s1.a, res_ref2);
  52
  53 #if AVX512F_LEN == 512
  54   if (UNION_ROUGH_CHECK (AVX512F_LEN, ) (res1, res_ref1, 0.0001))
  55     abort ();
  56 #endif
  57
  58   MASK_MERGE () (res_ref1, mask, SIZE);
  59   if (UNION_ROUGH_CHECK (AVX512F_LEN, ) (res2, res_ref1, 0.0001))
  60     abort ();
  61
  62   MASK_MERGE () (res_ref2, mask, SIZE);
  63   if (UNION_ROUGH_CHECK (AVX512F_LEN, ) (res3, res_ref2, 0.0001))
  64     abort ();
  65
  66   MASK_ZERO () (res_ref1, mask, SIZE);
  67   if (UNION_ROUGH_CHECK (AVX512F_LEN, ) (res4, res_ref1, 0.0001))
  68     abort ();
  69 }