1  /* { dg-do compile } */
       2  /* { dg-options "-O2 -ftree-vectorize -ffast-math" } */
       3  
       4  #include <stdint.h>
       5  
       6  #define TEST_LOOP(NAME, OUTTYPE, INTYPE, MASKTYPE)		\
       7    void __attribute__ ((noinline, noclone))			\
       8    NAME##_2 (OUTTYPE *__restrict dest, INTYPE *__restrict src,	\
       9  	    MASKTYPE *__restrict cond, intptr_t n)		\
      10    {								\
      11      for (intptr_t i = 0; i < n; ++i)				\
      12        {								\
      13  	if (cond[i] < 8)					\
      14  	  dest[i * 2] = src[i];					\
      15  	if (cond[i] > 2)					\
      16  	  dest[i * 2 + 1] = src[i];				\
      17        }								\
      18    }
      19  
      20  #define TEST2(NAME, OUTTYPE, INTYPE) \
      21    TEST_LOOP (NAME##_i8, OUTTYPE, INTYPE, int8_t) \
      22    TEST_LOOP (NAME##_i16, OUTTYPE, INTYPE, uint16_t) \
      23    TEST_LOOP (NAME##_f32, OUTTYPE, INTYPE, float) \
      24    TEST_LOOP (NAME##_f64, OUTTYPE, INTYPE, double)
      25  
      26  #define TEST1(NAME, OUTTYPE) \
      27    TEST2 (NAME##_i8, OUTTYPE, int8_t) \
      28    TEST2 (NAME##_i16, OUTTYPE, uint16_t) \
      29    TEST2 (NAME##_i32, OUTTYPE, int32_t) \
      30    TEST2 (NAME##_i64, OUTTYPE, uint64_t)
      31  
      32  #define TEST(NAME) \
      33    TEST1 (NAME##_i8, int8_t) \
      34    TEST1 (NAME##_i16, uint16_t) \
      35    TEST1 (NAME##_i32, int32_t) \
      36    TEST1 (NAME##_i64, uint64_t) \
      37    TEST2 (NAME##_f16_f16, _Float16, _Float16) \
      38    TEST2 (NAME##_f32_f32, float, float) \
      39    TEST2 (NAME##_f64_f64, double, double)
      40  
      41  TEST (test)
      42  
      43  /* { dg-final { scan-assembler-not {\tst2b\t.z[0-9]} } } */
      44  /* { dg-final { scan-assembler-not {\tst2h\t.z[0-9]} } } */
      45  /* { dg-final { scan-assembler-not {\tst2w\t.z[0-9]} } } */
      46  /* { dg-final { scan-assembler-not {\tst2d\t.z[0-9]} } } */