Viewing file:  avx512fp16-vrndscalesh-1b.c (1.46 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
/* { dg-do run { target avx512fp16 } } */
/* { dg-options "-O2 -mavx512fp16 -mavx512dq" } */
#define AVX512FP16
#include "avx512fp16-helper.h"
#define N_ELEMS 8
void NOINLINE
emulate_roundscale_sh(V512 * dest, V512 op1,
__mmask8 k, int zero_mask)
{
V512 v1, v2, v3, v4, v5, v6, v7, v8;
V512 t1,t2;
int i;
unpack_ph_2twops(op1, &v1, &v2);
unpack_ph_2twops(*dest, &v7, &v8);
t1.zmm = _mm512_maskz_roundscale_ps (0xffff, v1.zmm, 0x14);
t2.zmm = _mm512_maskz_roundscale_ps (0xffff, v2.zmm, 0x14);
if ((k&1) || !k)
v5.f32[0] = t1.f32[0];
else if (zero_mask)
v5.f32[0] = 0;
else
v5.f32[0] = v7.f32[0];
for (i = 1; i < 8; i++)
v5.f32[i] = v1.f32[i];
*dest = pack_twops_2ph(v5, v6);
}
void
test_512 (void)
{
V512 res;
V512 exp;
init_src();
emulate_roundscale_sh(&exp, src1, 0x1, 0);
res.xmmh[0] = _mm_roundscale_round_sh(src1.xmmh[0], src1.xmmh[0], 0x1, 0x08);
check_results(&res, &exp, N_ELEMS, "_mm_roundscale_round_sh");
init_dest(&res, &exp);
emulate_roundscale_sh(&exp, src1, 0x1, 0);
res.xmmh[0] = _mm_mask_roundscale_round_sh(res.xmmh[0],
0x1, src1.xmmh[0], src1.xmmh[0], 0x1, 0x08);
check_results(&res, &exp, N_ELEMS, "_mm_mask_roundscale_round_sh");
emulate_roundscale_sh(&exp, src1, 0x3, 1);
res.xmmh[0] = _mm_maskz_roundscale_round_sh(0x3, src1.xmmh[0], src1.xmmh[0], 0x1, 0x08);
check_results(&res, &exp, N_ELEMS, "_mm_maskz_roundscale_round_sh");
if (n_errs != 0)
abort ();
}
|
