2 /* { dg-options "-O2 -mavx512f" } */
3 /* { dg-require-effective-target avx512f } */
7 #include "avx512f-helper.h"
9 #define SIZE (AVX512F_LEN / 32)
10 #include "avx512f-mask-type.h"
12 CALC (int *r
, int *s1
, int *s2
)
15 for (i
= 0; i
< SIZE
; i
++)
18 s2
[i
] < 32 ? (s1
[i
] >> s2
[i
]) : (s1
[i
] > 0 ? 0 : 0xFFFFFFFF);
26 UNION_TYPE (AVX512F_LEN
, i_d
) res1
, res2
, res3
, src1
, src2
;
27 MASK_TYPE mask
= MASK_VALUE
;
31 for (i
= 0; i
< SIZE
; i
++)
33 src1
.a
[i
] = 2 + sign
* 7 * i
% 291;
34 src2
.a
[i
] = 1 + 17 * i
% 71;
38 for (i
= 0; i
< SIZE
; i
++)
39 res2
.a
[i
] = DEFAULT_VALUE
;
41 res1
.x
= INTRINSIC (_srav_epi32
) (src1
.x
, src2
.x
);
42 res2
.x
= INTRINSIC (_mask_srav_epi32
) (res2
.x
, mask
, src1
.x
, src2
.x
);
43 res3
.x
= INTRINSIC (_maskz_srav_epi32
) (mask
, src1
.x
, src2
.x
);
45 CALC (res_ref
, src1
.a
, src2
.a
);
47 if (UNION_CHECK (AVX512F_LEN
, i_d
) (res1
, res_ref
))
50 MASK_MERGE (i_d
) (res_ref
, mask
, SIZE
);
51 if (UNION_CHECK (AVX512F_LEN
, i_d
) (res2
, res_ref
))
54 MASK_ZERO (i_d
) (res_ref
, mask
, SIZE
);
55 if (UNION_CHECK (AVX512F_LEN
, i_d
) (res3
, res_ref
))