1 /* { dg-do run } */
2 /* { dg-options "-O2 -mavx512f" } */
3 /* { dg-require-effective-target avx512f } */
4
5 #define AVX512F
6
7 #include "avx512f-helper.h"
8
9 #include <math.h>
10 #define SIZE (AVX512F_LEN / 32)
11 #include "avx512f-mask-type.h"
12
13 __mmask8 dst_ref;
14
15 #if AVX512F_LEN == 512
16 #undef CMP
17 #define CMP(imm, rel) \
18 dst_ref = 0; \
19 for (i = 0; i < 8; i++) \
20 { \
21 dst_ref = ((rel) << i) | dst_ref; \
22 } \
23 source1.x = _mm512_loadu_si512 (s1); \
24 source2.x = _mm512_loadu_si512 (s2); \
25 dst1 = _mm512_cmp_epi64_mask (source1.x, source2.x, imm);\
26 dst2 = _mm512_mask_cmp_epi64_mask (mask, source1.x, source2.x, imm);\
27 if (dst_ref != dst1) abort(); \
28 if ((mask & dst_ref) != dst2) abort();
29 #endif
30
31 #if AVX512F_LEN == 256
32 #undef CMP
33 #define CMP(imm, rel) \
34 dst_ref = 0; \
35 for (i = 0; i < 4; i++) \
36 { \
37 dst_ref = ((rel) << i) | dst_ref; \
38 } \
39 source1.x = _mm256_loadu_si256 ((__m256i*)s1); \
40 source2.x = _mm256_loadu_si256 ((__m256i*)s2); \
41 dst1 = _mm256_cmp_epi64_mask (source1.x, source2.x, imm);\
42 dst2 = _mm256_mask_cmp_epi64_mask (mask, source1.x, source2.x, imm);\
43 if (dst_ref != dst1) abort(); \
44 if ((mask & dst_ref) != dst2) abort();
45 #endif
46
47 #if AVX512F_LEN == 128
48 #undef CMP
49 #define CMP(imm, rel) \
50 dst_ref = 0; \
51 for (i = 0; i < 2; i++) \
52 { \
53 dst_ref = ((rel) << i) | dst_ref; \
54 } \
55 source1.x = _mm_loadu_si128 ((__m128i*)s1); \
56 source2.x = _mm_loadu_si128 ((__m128i*)s2); \
57 dst1 = _mm_cmp_epi64_mask (source1.x, source2.x, imm);\
58 dst2 = _mm_mask_cmp_epi64_mask (mask, source1.x, source2.x, imm);\
59 if (dst_ref != dst1) abort(); \
60 if ((mask & dst_ref) != dst2) abort();
61 #endif
62
63 void
TEST()64 TEST ()
65 {
66 UNION_TYPE (AVX512F_LEN, i_d) source1, source2;
67 MASK_TYPE dst1, dst2, dst_ref;
68 MASK_TYPE mask = MASK_VALUE;
69 long long s1[8] = {2134, 6678, 453, 54646,
70 231, 5674, 111, 23241};
71 long long s2[8] = {41124, 6678, 8653, 856,
72 231, 4646, 111, 124};
73 int i;
74
75 CMP(0x00, s1[i] == s2[i]);
76 CMP(0x01, s1[i] < s2[i]);
77 CMP(0x02, s1[i] <= s2[i]);
78 CMP(0x03, 0);
79 CMP(0x04, s1[i] != s2[i]);
80 CMP(0x05, s1[i] >= s2[i]);
81 CMP(0x06, s1[i] > s2[i]);
82 CMP(0x07, 1);
83 }
84