2 /* { dg-options "-O2 -mavx512f" } */
3 /* { dg-require-effective-target avx512f } */
7 #include "avx512f-helper.h"
9 #define SIZE (AVX512F_LEN / 64)
10 #include "avx512f-mask-type.h"
13 CALC (unsigned long long *r
, unsigned long long *s1
,
14 unsigned long long count
)
17 for (i
= 0; i
< SIZE
; i
++)
19 r
[i
] = count
< 64 ? (s1
[i
] >> count
) : 0;
27 UNION_TYPE (AVX512F_LEN
, i_uq
) res1
, res2
, res3
, src1
;
28 MASK_TYPE mask
= MASK_VALUE
;
29 unsigned long long res_ref
[SIZE
];
31 for (i
= 0; i
< SIZE
; i
++)
33 src1
.a
[i
] = 2 + 7 * i
% 291;
36 for (i
= 0; i
< SIZE
; i
++)
37 res2
.a
[i
] = DEFAULT_VALUE
;
39 res1
.x
= INTRINSIC (_srli_epi64
) (src1
.x
, 3);
41 INTRINSIC (_mask_srli_epi64
) (res2
.x
, mask
, src1
.x
, 3);
42 res3
.x
= INTRINSIC (_maskz_srli_epi64
) (mask
, src1
.x
, 3);
44 CALC (res_ref
, src1
.a
, 3);
46 if (UNION_CHECK (AVX512F_LEN
, i_uq
) (res1
, res_ref
))
49 MASK_MERGE (i_uq
) (res_ref
, mask
, SIZE
);
50 if (UNION_CHECK (AVX512F_LEN
, i_uq
) (res2
, res_ref
))
53 MASK_ZERO (i_uq
) (res_ref
, mask
, SIZE
);
54 if (UNION_CHECK (AVX512F_LEN
, i_uq
) (res3
, res_ref
))
58 for (i
= 0; i
< SIZE
; i
++)
59 res2
.a
[i
] = DEFAULT_VALUE
;
61 res1
.x
= INTRINSIC (_srli_epi64
) (src1
.x
, 65);
63 INTRINSIC (_mask_srli_epi64
) (res2
.x
, mask
, src1
.x
, 65);
64 res3
.x
= INTRINSIC (_maskz_srli_epi64
) (mask
, src1
.x
, 65);
66 CALC (res_ref
, src1
.a
, 65);
68 if (UNION_CHECK (AVX512F_LEN
, i_uq
) (res1
, res_ref
))
71 MASK_MERGE (i_uq
) (res_ref
, mask
, SIZE
);
72 if (UNION_CHECK (AVX512F_LEN
, i_uq
) (res2
, res_ref
))
75 MASK_ZERO (i_uq
) (res_ref
, mask
, SIZE
);
76 if (UNION_CHECK (AVX512F_LEN
, i_uq
) (res3
, res_ref
))