| /* { dg-do run { target aarch64_sve_hw } } */ |
| /* { dg-options "-O2 -ftree-vectorize" } */ |
| |
| #include "vcond_2.c" |
| |
| #define N 97 |
| |
| #define TEST_VCOND_VAR(DATA_TYPE, CMP_TYPE, COND, SUFFIX) \ |
| { \ |
| DATA_TYPE x[N], y[N], r[N]; \ |
| CMP_TYPE a[N], b[N]; \ |
| for (int i = 0; i < N; ++i) \ |
| { \ |
| x[i] = i; \ |
| y[i] = (i & 1) + 5; \ |
| a[i] = i - N / 3; \ |
| b[i] = N - N / 3 - i; \ |
| asm volatile ("" ::: "memory"); \ |
| } \ |
| vcond_var_##CMP_TYPE##_##SUFFIX (r, x, y, a, b, N); \ |
| for (int i = 0; i < N; ++i) \ |
| if (r[i] != (a[i] COND b[i] ? x[i] : y[i])) \ |
| __builtin_abort (); \ |
| } |
| |
| #define TEST_VCOND_IMM(DATA_TYPE, CMP_TYPE, COND, IMM, SUFFIX) \ |
| { \ |
| DATA_TYPE x[N], y[N], r[N]; \ |
| CMP_TYPE a[N]; \ |
| for (int i = 0; i < N; ++i) \ |
| { \ |
| x[i] = i; \ |
| y[i] = (i & 1) + 5; \ |
| a[i] = IMM - N / 3 + i; \ |
| asm volatile ("" ::: "memory"); \ |
| } \ |
| vcond_imm_##CMP_TYPE##_##SUFFIX (r, x, y, a, N); \ |
| for (int i = 0; i < N; ++i) \ |
| if (r[i] != (a[i] COND (CMP_TYPE) IMM ? x[i] : y[i])) \ |
| __builtin_abort (); \ |
| } |
| |
| int __attribute__ ((optimize (1))) |
| main (int argc, char **argv) |
| { |
| TEST_VAR_ALL (TEST_VCOND_VAR) |
| TEST_IMM_ALL (TEST_VCOND_IMM) |
| return 0; |
| } |