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

   1 /* { dg-do run } */
   2 /* { dg-options "-O2 -mavx512f" } */
   3 /* { dg-require-effective-target avx512f } */
   4
   5 #include "avx512f-check.h"
   6
   7 #include "avx512f-helper.h"
   8
   9 #define SIZE (128 / 32)
  10 #include "avx512f-mask-type.h"
  11
  12 static void
  13 calc_add (float *s1, float *s2, float *s3, float* r)
  14 {
  15   r[0] = s1[0] * s2[0] - s3[0];
  16
  17   int i;
  18   for (i = 1; i < SIZE; i++)
  19     r[i] = s1[i];
  20 }
  21
  22 static void
  23 calc_add_3 (float *s1, float *s2, float *s3, float* r)
  24 {
  25   r[0] = s2[0] * s3[0] - s1[0];
  26
  27   int i;
  28   for (i = 1; i < SIZE; i++)
  29     r[i] = s1[i];
  30 }
  31
  32 void
  33 avx512f_test (void)
  34 {
  35   int i, sign;
  36   union128 res1, res2, res3, res4, res5, res6, res7, src1, src2, src3;
  37   MASK_TYPE mask = MASK_VALUE;
  38   float res_ref1[SIZE], res_ref2[SIZE];
  39
  40   sign = -1;
  41   for (i = 0; i < SIZE; i++)
  42     {
  43       src1.a[i] = DEFAULT_VALUE;
  44       src2.a[i] = 56.78 * (i + 1) * sign;
  45       src3.a[i] = 90.12 * (i + 2) * sign;
  46       sign = sign * -1;
  47     }
  48   for (i = 0; i < SIZE; i++)
  49     {
  50       res1.a[i] = DEFAULT_VALUE;
  51       res2.a[i] = DEFAULT_VALUE;
  52       res5.a[i] = DEFAULT_VALUE;
  53       res6.a[i] = DEFAULT_VALUE;
  54     }
  55
  56   calc_add (src1.a, src2.a, src3.a, res_ref1);
  57   calc_add_3(src1.a, src2.a, src3.a, res_ref2);
  58
  59   res1.x = _mm_mask_fmsub_ss (src1.x, mask, src2.x, src3.x);
  60   res2.x = _mm_mask3_fmsub_ss (src2.x, src3.x, src1.x, mask);
  61   res3.x = _mm_maskz_fmsub_ss (mask, src1.x, src2.x, src3.x);
  62   res4.x = _mm_fmsub_round_ss (src1.x, src2.x, src3.x, _MM_FROUND_NO_EXC);
  63   res5.x = _mm_mask_fmsub_round_ss (src1.x, mask, src2.x, src3.x, _MM_FROUND_NO_EXC);
  64   res6.x = _mm_mask3_fmsub_round_ss (src2.x, src3.x, src1.x, mask, _MM_FROUND_NO_EXC);
  65   res7.x = _mm_maskz_fmsub_round_ss (mask, src1.x, src2.x, src3.x, _MM_FROUND_NO_EXC);
  66
  67   if (check_union128 (res4, res_ref1))
  68     abort();
  69
  70   MASK_ZERO () (res_ref1, mask, 1);
  71   if (check_union128 (res3, res_ref1))
  72     abort ();
  73
  74   MASK_ZERO () (res_ref1, mask, 1);
  75   if (check_union128 (res7, res_ref1))
  76     abort ();
  77
  78   MASK_MERGE () (res_ref2, mask, 1);
  79   if (check_union128 (res2, res_ref2))
  80     abort ();
  81
  82   MASK_MERGE () (res_ref2, mask, 1);
  83   if (check_union128 (res6, res_ref2))
  84     abort ();
  85
  86   MASK_MERGE () (res_ref1, mask, 1);
  87   if (check_union128 (res1, res_ref1))
  88     abort ();
  89
  90   MASK_MERGE () (res_ref1, mask, 1);
  91   if (check_union128 (res5, res_ref1))
  92     abort ();
  93 }
  94