/* { dg-do run { target avx512fp16 } } */
/* { dg-options "-O2 -mavx512fp16 -mavx512dq" } */


#define AVX512FP16
#include "avx512fp16-helper.h"

#define N_ELEMS 8

V512 borrow_reduce_ps(V512 v, int imm8)
{
  V512 temp;
  switch (imm8)
    {
    case 1: temp.zmm =  _mm512_mask_reduce_ps (v.zmm, 0xffff, v.zmm, 1);break;
    case 2: temp.zmm =  _mm512_mask_reduce_ps (v.zmm, 0xffff, v.zmm, 2);break;
    case 3: temp.zmm =  _mm512_mask_reduce_ps (v.zmm, 0xffff, v.zmm, 3);break;
    case 4: temp.zmm =  _mm512_mask_reduce_ps (v.zmm, 0xffff, v.zmm, 4);break;
    case 5: temp.zmm =  _mm512_mask_reduce_ps (v.zmm, 0xffff, v.zmm, 5);break;
    case 6: temp.zmm =  _mm512_mask_reduce_ps (v.zmm, 0xffff, v.zmm, 6);break;
    case 7: temp.zmm =  _mm512_mask_reduce_ps (v.zmm, 0xffff, v.zmm, 7);break;
    case 8: temp.zmm =  _mm512_mask_reduce_ps (v.zmm, 0xffff, v.zmm, 8);break;
    }
  return temp;
}

void NOINLINE
emulate_reduce_sh(V512 * dest, V512 op1,
                  __mmask32 k, int imm8, int zero_mask) 
{
  V512 v1, v2, v3, v4, v5, v6, v7, v8;
  V512 t1;
  int i;

  unpack_ph_2twops(op1, &v1, &v2);
  unpack_ph_2twops(*dest, &v7, &v8);
  t1 = borrow_reduce_ps(v1, imm8);

  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_reduce_sh(&exp, src1,  0x1, 8, 0);
  res.xmmh[0] = _mm_reduce_round_sh(src1.xmmh[0], exp.xmmh[0], 8, _ROUND_CUR);
  check_results(&res, &exp, N_ELEMS, "_mm_reduce_round_sh");

  init_dest(&res, &exp);
  emulate_reduce_sh(&exp, src1,  0x1, 7, 0);
  res.xmmh[0] = _mm_mask_reduce_round_sh(res.xmmh[0], 0x1, src1.xmmh[0], exp.xmmh[0], 7, _ROUND_CUR);
  check_results(&res, &exp, N_ELEMS, "_mm_mask_reduce_round_sh");

  emulate_reduce_sh(&exp, src1,  0x3, 6, 1);
  res.xmmh[0] = _mm_maskz_reduce_round_sh(0x3, src1.xmmh[0], exp.xmmh[0], 6, _ROUND_CUR);
  check_results(&res, &exp, N_ELEMS, "_mm_maskz_reduce_round_sh");


  if (n_errs != 0) {
      abort ();
  }
}