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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
/* { dg-do run } */
/* { dg-options "-O2 -mavx512bw" } */
/* { dg-require-effective-target avx512bw } */
#define AVX512BW
#include "avx512f-helper.h"
#include <math.h>
#define SIZE (AVX512F_LEN / 16)
#include "avx512f-mask-type.h"
#if AVX512F_LEN == 512
#undef CMP
#define CMP(imm, rel) \
dst_ref = 0; \
for (i = 0; i < 32; i++) \
{ \
dst_ref = ((rel) << i) | dst_ref; \
} \
source1.x = _mm512_loadu_si512 (s1); \
source2.x = _mm512_loadu_si512 (s2); \
dst1 = _mm512_cmp_epi16_mask (source1.x, source2.x, imm);\
dst2 = _mm512_mask_cmp_epi16_mask (mask, source1.x, source2.x, imm);\
if (dst_ref != dst1) abort(); \
if ((mask & dst_ref) != dst2) abort();
#endif
#if AVX512F_LEN == 256
#undef CMP
#define CMP(imm, rel) \
dst_ref = 0; \
for (i = 0; i < 16; i++) \
{ \
dst_ref = ((rel) << i) | dst_ref; \
} \
source1.x = _mm256_loadu_si256 ((__m256i*)s1); \
source2.x = _mm256_loadu_si256 ((__m256i*)s2); \
dst1 = _mm256_cmp_epi16_mask (source1.x, source2.x, imm);\
dst2 = _mm256_mask_cmp_epi16_mask (mask, source1.x, source2.x, imm);\
if (dst_ref != dst1) abort(); \
if ((mask & dst_ref) != dst2) abort();
#endif
#if AVX512F_LEN == 128
#undef CMP
#define CMP(imm, rel) \
dst_ref = 0; \
for (i = 0; i < 8; i++) \
{ \
dst_ref = ((rel) << i) | dst_ref; \
} \
source1.x = _mm_loadu_si128 ((__m128i*)s1); \
source2.x = _mm_loadu_si128 ((__m128i*)s2); \
dst1 = _mm_cmp_epi16_mask (source1.x, source2.x, imm);\
dst2 = _mm_mask_cmp_epi16_mask (mask, source1.x, source2.x, imm);\
if (dst_ref != dst1) abort(); \
if ((mask & dst_ref) != dst2) abort();
#endif
void
TEST ()
{
short s1[32] = {2134, 6678, 453, 54646,
231, 5674, 111, 23241,
12314, 145, 671, 7575,
23455, 166, 5321, 5673,
2134, 6678, 453, 54646,
231, 5674, 111, 23241,
12314, 145, 671, 7575,
23455, 166, 5321, 5673};
short s2[32] = {41124, 6678, 8653, 856,
231, 4646, 111, 124,
2745, 4567, 3676, 123,
714, 3589, 5683, 5673,
41124, 6678, 8653, 856,
231, 4646, 111, 124,
2745, 4567, 3676, 123,
714, 3589, 5683, 5673};
UNION_TYPE (AVX512F_LEN, i_w) source1, source2;
MASK_TYPE dst1, dst2, dst_ref;
MASK_TYPE mask = MASK_VALUE;
int i;
CMP(0x00, s1[i] == s2[i]);
CMP(0x01, s1[i] < s2[i]);
CMP(0x02, s1[i] <= s2[i]);
CMP(0x03, 0);
CMP(0x04, s1[i] != s2[i]);
CMP(0x05, s1[i] >= s2[i]);
CMP(0x06, s1[i] > s2[i]);
CMP(0x07, 1);
}
|
