gcc/testsuite/gcc.target/aarch64/sve/mask_struct_store_1.c

   1 /* { dg-do compile } */
   2 /* { dg-options "-O2 -ftree-vectorize -ffast-math" } */
   3
   4 #define TEST_LOOP(NAME, OUTTYPE, INTYPE, MASKTYPE)              \
   5   void __attribute__ ((noinline, noclone))                      \
   6   NAME##_2 (OUTTYPE *__restrict dest, INTYPE *__restrict src,   \
   7             MASKTYPE *__restrict cond, INTYPE bias, int n)      \
   8   {                                                             \
   9     for (int i = 0; i < n; ++i)                                 \
  10       {                                                         \
  11         INTYPE value = src[i] + bias;                           \
  12         if (cond[i])                                            \
  13           {                                                     \
  14             dest[i * 2] = value;                                \
  15             dest[i * 2 + 1] = value;                            \
  16           }                                                     \
  17       }                                                         \
  18   }
  19
  20 #define TEST2(NAME, OUTTYPE, INTYPE) \
  21   TEST_LOOP (NAME##_i8, OUTTYPE, INTYPE, signed char) \
  22   TEST_LOOP (NAME##_i16, OUTTYPE, INTYPE, unsigned short) \
  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, signed char) \
  28   TEST2 (NAME##_i16, OUTTYPE, unsigned short) \
  29   TEST2 (NAME##_i32, OUTTYPE, int) \
  30   TEST2 (NAME##_i64, OUTTYPE, unsigned long)
  31
  32 #define TEST(NAME) \
  33   TEST1 (NAME##_i8, signed char) \
  34   TEST1 (NAME##_i16, unsigned short) \
  35   TEST1 (NAME##_i32, int) \
  36   TEST1 (NAME##_i64, unsigned long) \
  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 /*    Mask |  8 16 32 64
  44     -------+------------
  45     In   8 |  1  1  1  1
  46         16 |  1  1  1  1
  47         32 |  1  1  1  1
  48         64 |  1  1  1  1.  */
  49 /* { dg-final { scan-assembler-times {\tst2b\t.z[0-9]} 16 } } */
  50
  51 /*    Mask |  8 16 32 64
  52     -------+------------
  53     In   8 |  2  2  2  2
  54         16 |  2  1  1  1 x2 (for _Float16)
  55         32 |  2  1  1  1
  56         64 |  2  1  1  1.  */
  57 /* { dg-final { scan-assembler-times {\tst2h\t.z[0-9]} 28 } } */
  58
  59 /*    Mask |  8 16 32 64
  60     -------+------------
  61     In   8 |  4  4  4  4
  62         16 |  4  2  2  2
  63         32 |  4  2  1  1 x2 (for float)
  64         64 |  4  2  1  1.  */
  65 /* { dg-final { scan-assembler-times {\tst2w\t.z[0-9]} 50 } } */
  66
  67 /*    Mask |  8 16 32 64
  68     -------+------------
  69     In   8 |  8  8  8  8
  70         16 |  8  4  4  4
  71         32 |  8  4  2  2
  72         64 |  8  4  2  1 x2 (for double).  */
  73 /* { dg-final { scan-assembler-times {\tst2d\t.z[0-9]} 98 } } */