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

#include <string.h>

static void do_test (void);

#define DO_TEST do_test
#define AVX512FP16
#include "avx512-check.h"

void
__attribute__ ((noinline, noclone))
foo128 (_Float16 *p, __m128h x)
{
  *p = ((__v8hf)x)[4];
}

void
__attribute__ ((noinline, noclone))
foo256 (_Float16 *p, __m256h x)
{
  *p = ((__v16hf)x)[10];
}

void
__attribute__ ((noinline, noclone))
foo512 (_Float16 *p, __m512h x)
{
  *p = ((__v32hf)x)[30];
}

static void
do_test (void)
{
  _Float16 x = 25.3;
  union128h u128 = { 0.0f, x, 0.0f, 0.0f, x, 0.0f, 0.0f, x };
  union256h u256 = { x, 0.0f, 0.0f, 0.0f, x, 0.0f, 0.0f, 0.0f,
		     0.0f, 0.0f, x, 0.0f, 0.0f, x, 0.0f, 0.0f };
  union512h u512 = { x, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f,
		     0.0f, x, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f,
		     0.0f, 0.0f, x, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f,
		     0.0f, 0.0f, 0.0f, x, 0.0f, 0.0f, x, 0.0f };
  __m128h v128 = _mm_setr_ph (0.0f, x, 0.0f, 0.0f,
			      x, 0.0f, 0.0f, x);
  __m256h v256 = _mm256_setr_ph (x, 0.0f, 0.0f, 0.0f,
				 x, 0.0f, 0.0f, 0.0f,
				 0.0f, 0.0f, x, 0.0f,
				 0.0f, x, 0.0f, 0.0f);
  __m512h v512 = _mm512_setr_ph (x, 0.0f, 0.0f, 0.0f,
				 0.0f, 0.0f, 0.0f, 0.0f,
				 0.0f, x, 0.0f, 0.0f,
				 0.0f, 0.0f, 0.0f, 0.0f,
				 0.0f, 0.0f, x, 0.0f,
				 0.0f, 0.0f, 0.0f, 0.0f,
				 0.0f, 0.0f, 0.0f, x,
				 0.0f, 0.0f, x, 0.0f);
  union128h a128;
  union256h a256;
  union512h a512;
  _Float16 y;

  a128.x = v128;
  if (check_union128h (a128, u128.a))
    abort ();

  a256.x = v256;
  if (check_union256h (a256, u256.a))
    abort ();

  a512.x = v512;
  if (check_union512h (a512, u512.a))
    abort ();

  foo128 (&y, u128.x);
  if (x != y)
    abort ();

  foo256 (&y, u256.x);
  if (x != y)
    abort ();

  foo512 (&y, u512.x);
  if (x != y)
    abort ();
}