1  /* { dg-do compile } */
       2  /* { dg-options "-O2 -ftree-vectorize" } */
       3  
       4  #include <stdint.h>
       5  
       6  #define TEST(TYPE, NAME, OP)						\
       7    void __attribute__ ((noinline, noclone))				\
       8    test_##TYPE##_##NAME (TYPE *__restrict x,				\
       9  			TYPE *__restrict y,				\
      10  			TYPE z1, TYPE z2,				\
      11  			TYPE *__restrict pred, int n)			\
      12    {									\
      13      for (int i = 0; i < n; i += 2)					\
      14        {									\
      15  	x[i] = (pred[i] != 1 ? y[i] OP z1 : y[i]);			\
      16  	x[i + 1] = (pred[i + 1] != 1 ? y[i + 1] OP z2 : y[i + 1]);	\
      17        }									\
      18    }
      19  
      20  #define TEST_INT_TYPE(TYPE) \
      21    TEST (TYPE, div, /)
      22  
      23  #define TEST_FP_TYPE(TYPE) \
      24    TEST (TYPE, add, +) \
      25    TEST (TYPE, sub, -) \
      26    TEST (TYPE, mul, *) \
      27    TEST (TYPE, div, /)
      28  
      29  #define TEST_ALL \
      30    TEST_INT_TYPE (int32_t) \
      31    TEST_INT_TYPE (uint32_t) \
      32    TEST_INT_TYPE (int64_t) \
      33    TEST_INT_TYPE (uint64_t) \
      34    TEST_FP_TYPE (float) \
      35    TEST_FP_TYPE (double)
      36  
      37  TEST_ALL
      38  
      39  /* { dg-final { scan-assembler-times {\tsdiv\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      40  /* { dg-final { scan-assembler-times {\tudiv\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      41  /* { dg-final { scan-assembler-times {\tsdiv\tz[0-9]+\.d, p[0-7]/m,} 1 } } */
      42  /* { dg-final { scan-assembler-times {\tudiv\tz[0-9]+\.d, p[0-7]/m,} 1 } } */
      43  
      44  /* { dg-final { scan-assembler-times {\tfadd\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      45  /* { dg-final { scan-assembler-times {\tfadd\tz[0-9]+\.d, p[0-7]/m,} 1 } } */
      46  
      47  /* { dg-final { scan-assembler-times {\tfsub\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      48  /* { dg-final { scan-assembler-times {\tfsub\tz[0-9]+\.d, p[0-7]/m,} 1 } } */
      49  
      50  /* { dg-final { scan-assembler-times {\tfmul\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      51  /* { dg-final { scan-assembler-times {\tfmul\tz[0-9]+\.d, p[0-7]/m,} 1 } } */
      52  
      53  /* { dg-final { scan-assembler-times {\tfdiv\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      54  /* { dg-final { scan-assembler-times {\tfdiv\tz[0-9]+\.d, p[0-7]/m,} 1 } } */
      55  
      56  /* { dg-final { scan-assembler-times {\tld1w\tz[0-9]+\.s, p[0-7]/z,} 12 } } */
      57  /* { dg-final { scan-assembler-times {\tst1w\tz[0-9]+\.s, p[0-7],} 6 } } */
      58  
      59  /* { dg-final { scan-assembler-times {\tld1d\tz[0-9]+\.d, p[0-7]/z,} 12 } } */
      60  /* { dg-final { scan-assembler-times {\tst1d\tz[0-9]+\.d, p[0-7],} 6 } } */
      61  
      62  /* { dg-final { scan-assembler-not {\tsel\t} } } */