1 /* { dg-do compile } */ 2 /* { dg-options "-O2 -ftree-vectorize" } */ 3 4 #include <stdint.h> 5 6 #define DEF_LOOP(TYPE, NAME, OP, CONST) \ 7 void __attribute__ ((noipa)) \ 8 test_##TYPE##_##NAME##_##CONST (TYPE *__restrict r, \ 9 TYPE *__restrict a, \ 10 TYPE *__restrict b, \ 11 TYPE *__restrict pred, int n) \ 12 { \ 13 for (int i = 0; i < n; ++i) \ 14 r[i] = pred[i] != 1 ? a[i] OP b[i] * CONST : a[i]; \ 15 } 16 17 #define TEST_COUNT(T, TYPE, CONST) \ 18 T (TYPE, add, +, CONST) \ 19 T (TYPE, sub, -, CONST) 20 21 #define TEST_TYPE(T, TYPE, CONST) \ 22 TEST_COUNT (T, TYPE, 2) \ 23 TEST_COUNT (T, TYPE, 4) \ 24 TEST_COUNT (T, TYPE, CONST) 25 26 #define TEST_ALL(T) \ 27 TEST_TYPE (T, uint8_t, 0x80) \ 28 TEST_TYPE (T, uint16_t, 0x8000) \ 29 TEST_TYPE (T, uint32_t, 0x80000000) \ 30 TEST_TYPE (T, uint64_t, 0x8000000000000000ULL) 31 32 TEST_ALL (DEF_LOOP) 33 34 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.b, z[0-9]+\.b, #1\n} 2 } } */ 35 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.b, z[0-9]+\.b, #2\n} 2 } } */ 36 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.b, z[0-9]+\.b, #7\n} 2 } } */ 37 38 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.h, z[0-9]+\.h, #1\n} 2 } } */ 39 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.h, z[0-9]+\.h, #2\n} 2 } } */ 40 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.h, z[0-9]+\.h, #15\n} 2 } } */ 41 42 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.s, z[0-9]+\.s, #1\n} 2 } } */ 43 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.s, z[0-9]+\.s, #2\n} 2 } } */ 44 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.s, z[0-9]+\.s, #31\n} 2 } } */ 45 46 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.d, z[0-9]+\.d, #1\n} 2 } } */ 47 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.d, z[0-9]+\.d, #2\n} 2 } } */ 48 /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.d, z[0-9]+\.d, #63\n} 2 } } */ 49 50 /* { dg-final { scan-assembler-times {\tadd\tz[0-9]+\.b, p[0-7]/m, z[0-9]+\.b, z[0-9]+\.b\n} 4 } } */ 51 /* { dg-final { scan-assembler-times {\tadd\tz[0-9]+\.h, p[0-7]/m, z[0-9]+\.h, z[0-9]+\.h\n} 4 } } */ 52 /* { dg-final { scan-assembler-times {\tadd\tz[0-9]+\.s, p[0-7]/m, z[0-9]+\.s, z[0-9]+\.s\n} 4 } } */ 53 /* { dg-final { scan-assembler-times {\tadd\tz[0-9]+\.d, p[0-7]/m, z[0-9]+\.d, z[0-9]+\.d\n} 4 } } */ 54 55 /* { dg-final { scan-assembler-times {\tsub\tz[0-9]+\.b, p[0-7]/m, z[0-9]+\.b, z[0-9]+\.b\n} 2 } } */ 56 /* { dg-final { scan-assembler-times {\tsub\tz[0-9]+\.h, p[0-7]/m, z[0-9]+\.h, z[0-9]+\.h\n} 2 } } */ 57 /* { dg-final { scan-assembler-times {\tsub\tz[0-9]+\.s, p[0-7]/m, z[0-9]+\.s, z[0-9]+\.s\n} 2 } } */ 58 /* { dg-final { scan-assembler-times {\tsub\tz[0-9]+\.d, p[0-7]/m, z[0-9]+\.d, z[0-9]+\.d\n} 2 } } */ 59 60 /* { dg-final { scan-assembler-not {\tmov\tz[^,]*z} } } */ 61 /* { dg-final { scan-assembler-not {\tmovprfx\t} } } */ 62 /* { dg-final { scan-assembler-not {\tsel\t} } } */ 63