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

static void vec_op_test (void);
#define DO_TEST vec_op_test
#define AVX512FP16
#define AVX512VL
#include "avx512f-check.h"
#include "avx512fp16-64-32-vecop-1.c"

_Float16 a[4], b[4], fexp[4], fref[4];

#define EMULATE_VEC_OP_VV(size, op, name) \
void \
__attribute__ ((noinline, noclone)) \
scalar_vecop_v##size##hf##name ( \
  _Float16 * restrict dst, _Float16 * restrict src1,  \
  _Float16 * restrict src2)  \
{ \
  int i;  \
  for (i = 0; i < size; i++)  \
    dst[i] = src1[i] op src2[i];  \
}

EMULATE_VEC_OP_VV (4, +, add)
EMULATE_VEC_OP_VV (2, +, add)
EMULATE_VEC_OP_VV (4, -, sub)
EMULATE_VEC_OP_VV (2, -, sub)
EMULATE_VEC_OP_VV (4, *, mul)
EMULATE_VEC_OP_VV (2, *, mul)
EMULATE_VEC_OP_VV (4, /, div)
EMULATE_VEC_OP_VV (2, /, div)

void init()
{
  int i;
  for (i = 0; i < 4; i++)
    {
      a[i] = i + 0.5; 
      b[i] = i * 1.5;
      fexp[i] = fref[i] = 2.75 * i;
    }
}

int check_cond(void *a, void *b, int size)
{
  int i;
  unsigned short *pa = (unsigned short *)a,
		 *pb = (unsigned short *)b;
  for (i = 0; i < size; i++)
    if (pa[i] != pb[i])
      return 0;
  return 1;
}

#define TEST_VEC_OP_VV(size, name)	\
{ \
  init ();  \
  scalar_vecop_v##size##hf##name (a, b, fexp);  \
  vecop_v##size##hf##name (a, b, fref);  \
  if (!check_cond ((void *)fexp, (void *)fref, size)) \
    abort();  \
}

static void vec_op_test()
{
  TEST_VEC_OP_VV (4, add)
  TEST_VEC_OP_VV (2, add)
  TEST_VEC_OP_VV (4, sub)
  TEST_VEC_OP_VV (2, sub)
  TEST_VEC_OP_VV (4, mul)
  TEST_VEC_OP_VV (2, mul)
  TEST_VEC_OP_VV (4, div)
  TEST_VEC_OP_VV (2, div)
}