1  /* { dg-do run } */
       2  /* { dg-options "-mavx512f -O2" } */
       3  /* { dg-require-effective-target avx512f } */
       4  
       5  #include <math.h>
       6  #include "avx512f-check.h"
       7  
       8  #define SIZE (128 / 32)
       9  #include "avx512f-mask-type.h"
      10  
      11  static void
      12  compute_sqrtss (float *s1, float *s2, float *r)
      13  {
      14    r[0] = sqrt(s2[0]);
      15    int i;
      16    for (i = 1; i < SIZE; i++)
      17      {
      18        r[i] = s1[i];
      19      }
      20  }
      21  
      22  static void
      23  avx512f_test (void)
      24  {
      25    union128 res1, res2, res3, res4, res5;
      26    union128 s1, s2;
      27    float res_ref[SIZE];
      28    MASK_TYPE mask = MASK_VALUE;
      29    int i;
      30  
      31    for (i = 0; i < SIZE; i++)
      32      {
      33        s1.a[i] = 11.5 * (i + 1);
      34        s2.a[i] = 10.5 * (i + 1);
      35        res_ref[i] = 9.5 * (i + 1);
      36        res1.a[i] = DEFAULT_VALUE;
      37        res2.a[i] = DEFAULT_VALUE;
      38        res3.a[i] = DEFAULT_VALUE;
      39        res4.a[i] = DEFAULT_VALUE;
      40        res5.a[i] = DEFAULT_VALUE;
      41      }
      42  
      43    res1.x = _mm_sqrt_round_ss (s1.x, s2.x,
      44                  _MM_FROUND_TO_NEAREST_INT | _MM_FROUND_NO_EXC);
      45    res2.x = _mm_mask_sqrt_round_ss (s1.x, mask, s1.x, s2.x,
      46                  _MM_FROUND_TO_NEAREST_INT | _MM_FROUND_NO_EXC);
      47    res3.x = _mm_maskz_sqrt_round_ss (mask, s1.x, s2.x,
      48                  _MM_FROUND_TO_NEAREST_INT | _MM_FROUND_NO_EXC);
      49    res4.x = _mm_mask_sqrt_ss (s1.x, mask, s1.x, s2.x);
      50    res5.x = _mm_maskz_sqrt_ss (mask, s1.x, s2.x);
      51  
      52    compute_sqrtss (s1.a, s2.a, res_ref);
      53  
      54    if (check_union128 (res1, res_ref))
      55      abort ();
      56  
      57    MASK_MERGE () (res_ref, mask, 1);
      58  
      59    if (check_union128 (res2, res_ref))
      60      abort ();
      61  
      62    if (check_union128 (res4, res_ref))
      63      abort ();
      64  
      65    MASK_ZERO () (res_ref, mask, 1);
      66  
      67    if (check_union128 (res3, res_ref))
      68      abort ();
      69  
      70    if (check_union128 (res5, res_ref))
      71      abort ();
      72  }
      73