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 (short *s1
, short *s2
, short *r
)
15 for (i
= 0; i
< SIZE
; i
++)
16 r
[i
] = ((unsigned short) s1
[i
] +
17 (unsigned short) s2
[i
] + 1) >> 1;
23 UNION_TYPE (AVX512F_LEN
, i_w
) s1
, s2
, res1
, res2
,res3
;
24 MASK_TYPE mask
= MASK_VALUE
;
28 for (i
= 0; i
< SIZE
; i
++)
32 res2
.a
[i
] = DEFAULT_VALUE
;
35 res1
.x
= INTRINSIC (_avg_epu16
) (s1
.x
, s2
.x
);
36 res2
.x
= INTRINSIC (_mask_avg_epu16
) (res2
.x
, mask
, s1
.x
, s2
.x
);
37 res3
.x
= INTRINSIC (_maskz_avg_epu16
) (mask
, s1
.x
, s2
.x
);
39 CALC (s1
.a
, s2
.a
, res_ref
);
41 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res1
, res_ref
))
44 MASK_MERGE (i_w
) (res_ref
, mask
, SIZE
);
45 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res2
, res_ref
))
48 MASK_ZERO (i_w
) (res_ref
, mask
, SIZE
);
49 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res3
, res_ref
))