1 /* { dg-do run } */
2 /* { dg-options "-O2 -mavx512bw" } */
3 /* { dg-require-effective-target avx512bw } */
4
5 #define AVX512BW
6 #include "avx512f-helper.h"
7
8 #include <math.h>
9 #define SIZE (AVX512F_LEN / 16)
10 #include "avx512f-mask-type.h"
11
12 #if AVX512F_LEN == 512
13 #undef CMP
14 #define CMP(imm, rel) \
15 dst_ref = 0; \
16 for (i = 0; i < 32; i++) \
17 { \
18 dst_ref = ((rel) << i) | dst_ref; \
19 } \
20 source1.x = _mm512_loadu_si512 (s1); \
21 source2.x = _mm512_loadu_si512 (s2); \
22 dst1 = _mm512_cmp_epi16_mask (source1.x, source2.x, imm);\
23 dst2 = _mm512_mask_cmp_epi16_mask (mask, source1.x, source2.x, imm);\
24 if (dst_ref != dst1) abort(); \
25 if ((mask & dst_ref) != dst2) abort();
26 #endif
27
28 #if AVX512F_LEN == 256
29 #undef CMP
30 #define CMP(imm, rel) \
31 dst_ref = 0; \
32 for (i = 0; i < 16; i++) \
33 { \
34 dst_ref = ((rel) << i) | dst_ref; \
35 } \
36 source1.x = _mm256_loadu_si256 ((__m256i*)s1); \
37 source2.x = _mm256_loadu_si256 ((__m256i*)s2); \
38 dst1 = _mm256_cmp_epi16_mask (source1.x, source2.x, imm);\
39 dst2 = _mm256_mask_cmp_epi16_mask (mask, source1.x, source2.x, imm);\
40 if (dst_ref != dst1) abort(); \
41 if ((mask & dst_ref) != dst2) abort();
42 #endif
43
44 #if AVX512F_LEN == 128
45 #undef CMP
46 #define CMP(imm, rel) \
47 dst_ref = 0; \
48 for (i = 0; i < 8; i++) \
49 { \
50 dst_ref = ((rel) << i) | dst_ref; \
51 } \
52 source1.x = _mm_loadu_si128 ((__m128i*)s1); \
53 source2.x = _mm_loadu_si128 ((__m128i*)s2); \
54 dst1 = _mm_cmp_epi16_mask (source1.x, source2.x, imm);\
55 dst2 = _mm_mask_cmp_epi16_mask (mask, source1.x, source2.x, imm);\
56 if (dst_ref != dst1) abort(); \
57 if ((mask & dst_ref) != dst2) abort();
58 #endif
59
60 void
TEST()61 TEST ()
62 {
63 short s1[32] = {2134, 6678, 453, 54646,
64 231, 5674, 111, 23241,
65 12314, 145, 671, 7575,
66 23455, 166, 5321, 5673,
67 2134, 6678, 453, 54646,
68 231, 5674, 111, 23241,
69 12314, 145, 671, 7575,
70 23455, 166, 5321, 5673};
71 short s2[32] = {41124, 6678, 8653, 856,
72 231, 4646, 111, 124,
73 2745, 4567, 3676, 123,
74 714, 3589, 5683, 5673,
75 41124, 6678, 8653, 856,
76 231, 4646, 111, 124,
77 2745, 4567, 3676, 123,
78 714, 3589, 5683, 5673};
79 UNION_TYPE (AVX512F_LEN, i_w) source1, source2;
80 MASK_TYPE dst1, dst2, dst_ref;
81 MASK_TYPE mask = MASK_VALUE;
82 int i;
83
84 CMP(0x00, s1[i] == s2[i]);
85 CMP(0x01, s1[i] < s2[i]);
86 CMP(0x02, s1[i] <= s2[i]);
87 CMP(0x03, 0);
88 CMP(0x04, s1[i] != s2[i]);
89 CMP(0x05, s1[i] >= s2[i]);
90 CMP(0x06, s1[i] > s2[i]);
91 CMP(0x07, 1);
92 }
93