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

   1 /* { dg-do run { target aarch64_sve_hw } } */
   2 /* { dg-options "-O2 -ftree-vectorize" } */
   3
   4 #include "slp_12.c"
   5
   6 #define N2 (31 * 2)
   7
   8 #define HARNESS(TYPE)                                           \
   9   {                                                             \
  10     TYPE a[N2], b[N2 * 4];                                      \
  11     for (unsigned int i = 0; i < N2; ++i)                       \
  12       {                                                         \
  13         a[i] = i * 2 + i % 5;                                   \
  14         b[i * 4] = i * 3 + i % 7;                               \
  15         b[i * 4 + 1] = i * 5 + i % 9;                           \
  16         b[i * 4 + 2] = i * 7 + i % 11;                          \
  17         b[i * 4 + 3] = i * 9 + i % 13;                          \
  18       }                                                         \
  19     vec_slp_##TYPE (a, b);                                      \
  20     for (unsigned int i = 0; i < N2; ++i)                       \
  21       {                                                         \
  22         TYPE orig_a = i * 2 + i % 5;                            \
  23         TYPE orig_b1 = i * 3 + i % 7;                           \
  24         TYPE orig_b2 = i * 5 + i % 9;                           \
  25         TYPE orig_b3 = i * 7 + i % 11;                          \
  26         TYPE orig_b4 = i * 9 + i % 13;                          \
  27         TYPE expected_a = orig_a;                               \
  28         TYPE expected_b1 = orig_b1;                             \
  29         TYPE expected_b2 = orig_b2;                             \
  30         TYPE expected_b3 = orig_b3;                             \
  31         TYPE expected_b4 = orig_b4;                             \
  32         if (i < N1)                                             \
  33           {                                                     \
  34             expected_a += 1;                                    \
  35             expected_b1 += 2;                                   \
  36             expected_b2 += 3;                                   \
  37             expected_b3 += 4;                                   \
  38             expected_b4 += 5;                                   \
  39           }                                                     \
  40         if (a[i] != expected_a                                  \
  41             || b[i * 4] != expected_b1                          \
  42             || b[i * 4 + 1] != expected_b2                      \
  43             || b[i * 4 + 2] != expected_b3                      \
  44             || b[i * 4 + 3] != expected_b4)                     \
  45           __builtin_abort ();                                   \
  46       }                                                         \
  47   }
  48
  49 int __attribute__ ((optimize (1)))
  50 main (void)
  51 {
  52   TEST_ALL (HARNESS)
  53 }