2 /* { dg-options "-O2 -mavx512bw -DAVX512BW" } */
3 /* { dg-require-effective-target avx512bw } */
5 #include "avx512f-helper.h"
7 #define SIZE (AVX512F_LEN / 16)
8 #include "avx512f-mask-type.h"
11 CALC (unsigned short *r
, unsigned short *s1
, unsigned short count
)
14 for (i
= 0; i
< SIZE
; i
++)
16 r
[i
] = count
< 16 ? (s1
[i
] >> count
) : 0;
24 UNION_TYPE (AVX512F_LEN
, i_w
) res1
, res2
, res3
, src1
;
25 MASK_TYPE mask
= MASK_VALUE
;
26 unsigned short res_ref
[SIZE
];
28 for (i
= 0; i
< SIZE
; i
++)
30 src1
.a
[i
] = 2 + 7 * i
% 291;
33 for (i
= 0; i
< SIZE
; i
++)
34 res2
.a
[i
] = DEFAULT_VALUE
;
36 res1
.x
= INTRINSIC (_srli_epi16
) (src1
.x
, 5);
37 res2
.x
= INTRINSIC (_mask_srli_epi16
) (res2
.x
, mask
, src1
.x
, 5);
38 res3
.x
= INTRINSIC (_maskz_srli_epi16
) (mask
, src1
.x
, 5);
40 CALC (res_ref
, src1
.a
, 5);
42 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res1
, res_ref
))
45 MASK_MERGE (i_w
) (res_ref
, mask
, SIZE
);
46 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res2
, res_ref
))
49 MASK_ZERO (i_w
) (res_ref
, mask
, SIZE
);
50 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res3
, res_ref
))
53 for (i
= 0; i
< SIZE
; i
++)
54 res2
.a
[i
] = DEFAULT_VALUE
;
56 res1
.x
= INTRINSIC (_srli_epi16
) (src1
.x
, 17);
57 res2
.x
= INTRINSIC (_mask_srli_epi16
) (res2
.x
, mask
, src1
.x
, 17);
58 res3
.x
= INTRINSIC (_maskz_srli_epi16
) (mask
, src1
.x
, 17);
60 CALC (res_ref
, src1
.a
, 17);
62 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res1
, res_ref
))
65 MASK_MERGE (i_w
) (res_ref
, mask
, SIZE
);
66 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res2
, res_ref
))
69 MASK_ZERO (i_w
) (res_ref
, mask
, SIZE
);
70 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res3
, res_ref
))