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

   1 /* { dg-do run { target aarch64_sve_hw } } */
   2 /* { dg-options "-O2 -ftree-vectorize -ffast-math" } */
   3
   4 #include "mask_struct_store_1.c"
   5
   6 #define N 100
   7
   8 #undef TEST_LOOP
   9 #define TEST_LOOP(NAME, OUTTYPE, INTYPE, MASKTYPE)              \
  10   {                                                             \
  11     OUTTYPE out[N * 2];                                         \
  12     INTYPE in[N];                                               \
  13     MASKTYPE mask[N];                                           \
  14     for (int i = 0; i < N; ++i)                                 \
  15       {                                                         \
  16         in[i] = i * 7 / 2;                                      \
  17         mask[i] = i % 5 <= i % 3;                               \
  18         asm volatile ("" ::: "memory");                         \
  19       }                                                         \
  20     for (int i = 0; i < N * 2; ++i)                             \
  21       out[i] = i * 9 / 2;                                       \
  22     NAME##_2 (out, in, mask, 17, N);                            \
  23     for (int i = 0; i < N * 2; ++i)                             \
  24       {                                                         \
  25         OUTTYPE if_true = (INTYPE) (in[i / 2] + 17);            \
  26         OUTTYPE if_false = i * 9 / 2;                           \
  27         if (out[i] != (mask[i / 2] ? if_true : if_false))       \
  28           __builtin_abort ();                                   \
  29         asm volatile ("" ::: "memory");                         \
  30       }                                                         \
  31   }
  32
  33 int __attribute__ ((optimize (1)))
  34 main (void)
  35 {
  36   TEST (test);
  37   return 0;
  38 }