2 /* { dg-options "-O2 -mavx512bw" } */
3 /* { dg-require-effective-target avx512bw } */
6 #include "avx512f-helper.h"
11 #define SIZE (AVX512F_LEN / 16)
12 #include "avx512f-mask-type.h"
15 CALC (short *i1
, short *i2
, short *r
)
17 unsigned char *ub1
= (unsigned char *) i1
;
18 char *sb2
= (char *) i2
;
19 short *sout
= (short *) r
;
23 for (i
= 0; i
< SIZE
; i
++)
25 t0
= ((int) ub1
[2 * i
] * (int) sb2
[2 * i
] +
26 (int) ub1
[2 * i
+ 1] * (int) sb2
[2 * i
+ 1]);
27 if (t0
> (int) 0x7fff)
29 else if (t0
< (int) 0xffff8000)
39 UNION_TYPE (AVX512F_LEN
, i_w
) s1
, s2
, res1
, res2
, res3
;
41 MASK_TYPE mask
= MASK_VALUE
;
45 for (i
= 0; i
< SIZE
; i
++)
48 s2
.a
[i
] = i
* -17 + i
* 2;
49 res2
.a
[i
] = DEFAULT_VALUE
;
52 res1
.x
= INTRINSIC (_maddubs_epi16
) (s1
.x
, s2
.x
);
53 res2
.x
= INTRINSIC (_mask_maddubs_epi16
) (res2
.x
, mask
, s1
.x
, s2
.x
);
54 res3
.x
= INTRINSIC (_maskz_maddubs_epi16
) (mask
, s1
.x
, s2
.x
);
56 CALC(s1
.a
, s2
.a
, res_ref
);
58 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res1
, res_ref
))
61 MASK_MERGE (i_w
) (res_ref
, mask
, SIZE
);
62 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res2
, res_ref
))
65 MASK_ZERO (i_w
) (res_ref
, mask
, SIZE
);
66 if (UNION_CHECK (AVX512F_LEN
, i_w
) (res3
, res_ref
))