1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
/* { dg-do compile } */
/* { dg-options "-O2 -ftree-vectorize -march=armv8-a+sve" } */
#include <stdint.h>
#define DEF_SEL_IMM(TYPE, SUFFIX, IMM) \
void \
sel_##TYPE##_##SUFFIX (TYPE *restrict a, TYPE *restrict b, int n) \
{ \
for (int i = 0; i < n; i++) \
a[i] = b[i] != 0 ? IMM : 0; \
}
#define DEF_SEL_VAR(TYPE) \
void \
sel_##TYPE##_var (TYPE *restrict a, TYPE *restrict b, TYPE val, int n) \
{ \
for (int i = 0; i < n; i++) \
a[i] = b[i] != 0 ? val : 0; \
}
#define TEST_TYPE8(TYPE) \
DEF_SEL_VAR (TYPE) \
DEF_SEL_IMM (TYPE, m128, -128) \
DEF_SEL_IMM (TYPE, m127, -127) \
DEF_SEL_IMM (TYPE, 2, 2) \
DEF_SEL_IMM (TYPE, 127, 127)
#define TEST_TYPE16(TYPE) \
TEST_TYPE8 (TYPE) \
DEF_SEL_IMM (TYPE, m32768, -32768) \
DEF_SEL_IMM (TYPE, m32767, -32767) \
DEF_SEL_IMM (TYPE, m32512, -32512) \
DEF_SEL_IMM (TYPE, m32511, -32511) \
DEF_SEL_IMM (TYPE, m256, -256) \
DEF_SEL_IMM (TYPE, m255, -255) \
DEF_SEL_IMM (TYPE, m129, -129) \
DEF_SEL_IMM (TYPE, 128, 128) \
DEF_SEL_IMM (TYPE, 256, 256) \
DEF_SEL_IMM (TYPE, 32511, 32511) \
DEF_SEL_IMM (TYPE, 32512, 32512) \
DEF_SEL_IMM (TYPE, 32767, 32767)
#define TEST_TYPE32(TYPE) \
TEST_TYPE16 (TYPE) \
DEF_SEL_IMM (TYPE, m65536, -65536) \
DEF_SEL_IMM (TYPE, m32769, -32769) \
DEF_SEL_IMM (TYPE, 32768, 32768)
TEST_TYPE8 (int8_t)
TEST_TYPE16 (int16_t)
TEST_TYPE32 (int32_t)
TEST_TYPE32 (int64_t)
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.b, p[0-7]/z, #-128\n} 1 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.b, p[0-7]/z, #-127\n} 1 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.b, p[0-7]/z, #2\n} 1 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.b, p[0-7]/z, #127\n} 1 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.[hsd], p[0-7]/z, #-32768\n} 3 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.[hsd], p[0-7]/z, #-32512\n} 3 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.[hsd], p[0-7]/z, #-256\n} 3 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.[hsd], p[0-7]/z, #-128\n} 3 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.[hsd], p[0-7]/z, #-127\n} 3 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.[hsd], p[0-7]/z, #2\n} 3 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.[hsd], p[0-7]/z, #127\n} 3 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.[hsd], p[0-7]/z, #256\n} 3 } } */
/* { dg-final { scan-assembler-times {\tmov\tz[0-9]+\.[hsd], p[0-7]/z, #32512\n} 3 } } */
|
