gcc/testsuite/gcc.target/aarch64/vect-add-sub-cond.c

   1 /* Make sure that vector comaprison results are not unnecessarily ANDed
   2    with vectors of 1.  */
   3 /* { dg-do compile } */
   4 /* { dg-options "-O2 -ftree-vectorize" } */
   5
   6 #pragma GCC target "+nosve"
   7
   8 #define COUNT1(X) if (X) count += 1
   9 #define COUNT2(X) if (X) count -= 1
  10 #define COUNT3(X) count += (X)
  11 #define COUNT4(X) count -= (X)
  12
  13 #define COND1(X) (X)
  14 #define COND2(X) ((X) ? 1 : 0)
  15 #define COND3(X) ((X) ? -1 : 0)
  16 #define COND4(X) ((X) ? 0 : 1)
  17 #define COND5(X) ((X) ? 0 : -1)
  18
  19 #define TEST_LT(X, Y) ((X) < (Y))
  20 #define TEST_LE(X, Y) ((X) <= (Y))
  21 #define TEST_GT(X, Y) ((X) > (Y))
  22 #define TEST_GE(X, Y) ((X) >= (Y))
  23 #define TEST_EQ(X, Y) ((X) == (Y))
  24 #define TEST_NE(X, Y) ((X) != (Y))
  25
  26 #define COUNT_LOOP(ID, TYPE, CMP_ARRAY, TEST, COUNT) \
  27   TYPE \
  28   reduc_##ID (__typeof__ (CMP_ARRAY[0]) x) \
  29   { \
  30     TYPE count = 0; \
  31     for (unsigned int i = 0; i < 1024; ++i) \
  32       COUNT (TEST (CMP_ARRAY[i], x)); \
  33     return count; \
  34   }
  35
  36 #define COND_LOOP(ID, ARRAY, CMP_ARRAY, TEST, COND) \
  37   void \
  38   plus_##ID (__typeof__ (CMP_ARRAY[0]) x) \
  39   { \
  40     for (unsigned int i = 0; i < 1024; ++i) \
  41       ARRAY[i] += COND (TEST (CMP_ARRAY[i], x)); \
  42   } \
  43   void \
  44   plusc_##ID (void) \
  45   { \
  46     for (unsigned int i = 0; i < 1024; ++i) \
  47       ARRAY[i] += COND (TEST (CMP_ARRAY[i], 10)); \
  48   } \
  49   void \
  50   minus_##ID (__typeof__ (CMP_ARRAY[0]) x) \
  51   { \
  52     for (unsigned int i = 0; i < 1024; ++i) \
  53       ARRAY[i] -= COND (TEST (CMP_ARRAY[i], x)); \
  54   } \
  55   void \
  56   minusc_##ID (void) \
  57   { \
  58     for (unsigned int i = 0; i < 1024; ++i) \
  59       ARRAY[i] += COND (TEST (CMP_ARRAY[i], 1)); \
  60   }
  61
  62 #define ALL_LOOPS(ID, ARRAY, CMP_ARRAY, TEST) \
  63   typedef __typeof__(ARRAY[0]) ID##_type; \
  64   COUNT_LOOP (ID##_1, ID##_type, CMP_ARRAY, TEST, COUNT1) \
  65   COUNT_LOOP (ID##_2, ID##_type, CMP_ARRAY, TEST, COUNT2) \
  66   COUNT_LOOP (ID##_3, ID##_type, CMP_ARRAY, TEST, COUNT3) \
  67   COUNT_LOOP (ID##_4, ID##_type, CMP_ARRAY, TEST, COUNT4) \
  68   COND_LOOP (ID##_1, ARRAY, CMP_ARRAY, TEST, COND1) \
  69   COND_LOOP (ID##_2, ARRAY, CMP_ARRAY, TEST, COND2) \
  70   COND_LOOP (ID##_3, ARRAY, CMP_ARRAY, TEST, COND3) \
  71   COND_LOOP (ID##_4, ARRAY, CMP_ARRAY, TEST, COND4) \
  72   COND_LOOP (ID##_5, ARRAY, CMP_ARRAY, TEST, COND5)
  73
  74 signed int asi[1024] __attribute__ ((aligned (16)));
  75 unsigned int aui[1024] __attribute__ ((aligned (16)));
  76 signed long long asl[1024] __attribute__ ((aligned (16)));
  77 unsigned long long aul[1024] __attribute__ ((aligned (16)));
  78 float af[1024] __attribute__ ((aligned (16)));
  79 double ad[1024] __attribute__ ((aligned (16)));
  80
  81 ALL_LOOPS (si_si, aui, asi, TEST_LT)
  82 ALL_LOOPS (ui_si, aui, asi, TEST_LE)
  83 ALL_LOOPS (si_ui, aui, asi, TEST_GT)
  84 ALL_LOOPS (ui_ui, aui, asi, TEST_GE)
  85 ALL_LOOPS (sl_sl, asl, asl, TEST_NE)
  86 ALL_LOOPS (ul_ul, aul, aul, TEST_EQ)
  87 ALL_LOOPS (si_f, asi, af, TEST_LE)
  88 ALL_LOOPS (ui_f, aui, af, TEST_GT)
  89 ALL_LOOPS (sl_d, asl, ad, TEST_GE)
  90 ALL_LOOPS (ul_d, aul, ad, TEST_GT)
  91
  92 /* { dg-final { scan-assembler-not "\tand\t" } } */
  93 /* { dg-final { scan-assembler-not "\tld\[^\t\]*\t\[wx\]" } } */
  94 /* { dg-final { scan-assembler-not "\tst\[^\t\]*\t\[wx\]" } } */
  95 /* { dg-final { scan-assembler "\tldr\tq" } } */
  96 /* { dg-final { scan-assembler "\tstr\tq" } } */