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
|
/* { dg-do assemble } */
/* { dg-options "-O2 -ftree-vectorize -ffast-math -march=armv8-a+sve --save-temps" } */
#include <stdint.h>
#ifndef INDEX32
#define INDEX32 int32_t
#define INDEX64 int64_t
#endif
#define TEST_LOOP(DATA_TYPE, CMP_TYPE, BITS) \
void \
f_##DATA_TYPE##_##CMP_TYPE \
(DATA_TYPE *restrict dest, DATA_TYPE *restrict src, \
CMP_TYPE *cmp1, CMP_TYPE *cmp2, INDEX##BITS *indices, int n) \
{ \
for (int i = 0; i < n; ++i) \
if (cmp1[i] == cmp2[i]) \
dest[i] += src[indices[i]]; \
}
#define TEST32(T, DATA_TYPE) \
T (DATA_TYPE, int32_t, 32) \
T (DATA_TYPE, uint32_t, 32) \
T (DATA_TYPE, float, 32)
#define TEST64(T, DATA_TYPE) \
T (DATA_TYPE, int64_t, 64) \
T (DATA_TYPE, uint64_t, 64) \
T (DATA_TYPE, double, 64)
#define TEST_ALL(T) \
TEST32 (T, int32_t) \
TEST32 (T, uint32_t) \
TEST32 (T, float) \
TEST64 (T, int64_t) \
TEST64 (T, uint64_t) \
TEST64 (T, double)
TEST_ALL (TEST_LOOP)
/* { dg-final { scan-assembler-times {\tld1w\tz[0-9]+\.s, p[0-7]/z, \[x[0-9]+, x[0-9]+, lsl 2\]\n} 36 } } */
/* { dg-final { scan-assembler-times {\tcmpeq\tp[0-7]\.s, p[0-7]/z, z[0-9]+\.s, z[0-9]+\.s\n} 6 } } */
/* { dg-final { scan-assembler-times {\tfcmeq\tp[0-7]\.s, p[0-7]/z, z[0-9]+\.s, z[0-9]+\.s\n} 3 } } */
/* { dg-final { scan-assembler-times {\tld1w\tz[0-9]+\.s, p[0-7]/z, \[x[0-9]+, z[0-9]+\.s, sxtw 2\]\n} 9 } } */
/* { dg-final { scan-assembler-times {\tst1w\tz[0-9]+\.s, p[0-7], \[x[0-9]+, x[0-9]+, lsl 2\]\n} 9 } } */
/* { dg-final { scan-assembler-times {\tld1d\tz[0-9]+\.d, p[0-7]/z, \[x[0-9]+, x[0-9]+, lsl 3\]\n} 36 } } */
/* { dg-final { scan-assembler-times {\tcmpeq\tp[0-7]\.d, p[0-7]/z, z[0-9]+\.d, z[0-9]+\.d\n} 6 } } */
/* { dg-final { scan-assembler-times {\tfcmeq\tp[0-7]\.d, p[0-7]/z, z[0-9]+\.d, z[0-9]+\.d\n} 3 } } */
/* { dg-final { scan-assembler-times {\tld1d\tz[0-9]+\.d, p[0-7]/z, \[x[0-9]+, z[0-9]+\.d, lsl 3\]\n} 9 } } */
/* { dg-final { scan-assembler-times {\tst1d\tz[0-9]+\.d, p[0-7], \[x[0-9]+, x[0-9]+, lsl 3\]\n} 9 } } */
|
