gcc/testsuite/gcc.target/x86_64/abi/avx512fp16/m512h/test_varargs-m512.c

/* Test variable number of 512-bit vector arguments passed to functions.  */

#include <stdio.h>
#include "avx512fp16-zmm-check.h"
#include "args.h"

struct IntegerRegisters iregs;
struct FloatRegisters fregs;

/* This struct holds values for argument checking.  */
struct 
{
  ZMM_T i0, i1, i2, i3, i4, i5, i6, i7, i8, i9;
} values;

char *pass;
int failed = 0;

#undef assert
#define assert(c) do { \
  if (!(c)) {failed++; printf ("failed %s\n", pass); } \
} while (0)

#define compare(X1,X2,T) do { \
  assert (memcmp (&X1, &X2, sizeof (T)) == 0); \
} while (0)

void
fun_check_passing_m512_varargs (__m512 i0, __m512 i1, __m512 i2,
				__m512 i3, ...)
{
  /* Check argument values.  */
  void **fp = __builtin_frame_address (0);
  void *ra = __builtin_return_address (0);
  __m512 *argp;

  compare (values.i0, i0, __m512);
  compare (values.i1, i1, __m512);
  compare (values.i2, i2, __m512);
  compare (values.i3, i3, __m512);

  /* Get the pointer to the return address on stack.  */
  while (*fp != ra)
    fp++;

  /* Skip the return address stack slot.  */
  argp = (__m512 *)(((char *) fp) + 8);

  /* Check __m512 arguments passed on stack.  */
  compare (values.i4, argp[0], __m512);
  compare (values.i5, argp[1], __m512);
  compare (values.i6, argp[2], __m512);
  compare (values.i7, argp[3], __m512);
  compare (values.i8, argp[4], __m512);
  compare (values.i9, argp[5], __m512);

  /* Check register contents.  */
  compare (fregs.zmm0, zmm_regs[0], __m512);
  compare (fregs.zmm1, zmm_regs[1], __m512);
  compare (fregs.zmm2, zmm_regs[2], __m512);
  compare (fregs.zmm3, zmm_regs[3], __m512);
}

void
fun_check_passing_m512h_varargs (__m512h i0, __m512h i1, __m512h i2,
				 __m512h i3, ...)
{
  /* Check argument values.  */
  void **fp = __builtin_frame_address (0);
  void *ra = __builtin_return_address (0);
  __m512h *argp;

  compare (values.i0, i0, __m512h);
  compare (values.i1, i1, __m512h);
  compare (values.i2, i2, __m512h);
  compare (values.i3, i3, __m512h);

  /* Get the pointer to the return address on stack.  */
  while (*fp != ra)
    fp++;

  /* Skip the return address stack slot.  */
  argp = (__m512h *)(((char *) fp) + 8);

  /* Check __m512h arguments passed on stack.  */
  compare (values.i4, argp[0], __m512h);
  compare (values.i5, argp[1], __m512h);
  compare (values.i6, argp[2], __m512h);
  compare (values.i7, argp[3], __m512h);
  compare (values.i8, argp[4], __m512h);
  compare (values.i9, argp[5], __m512h);

  /* Check register contents.  */
  compare (fregs.zmm0, zmm_regs[0], __m512h);
  compare (fregs.zmm1, zmm_regs[1], __m512h);
  compare (fregs.zmm2, zmm_regs[2], __m512h);
  compare (fregs.zmm3, zmm_regs[3], __m512h);
}

#define def_check_int_passing_varargs(_i0, _i1, _i2, _i3, _i4, _i5, \
				      _i6, _i7, _i8, _i9, \
				      _func, TYPE) \
  values.i0.TYPE[0] = _i0; \
  values.i1.TYPE[0] = _i1; \
  values.i2.TYPE[0] = _i2; \
  values.i3.TYPE[0] = _i3; \
  values.i4.TYPE[0] = _i4; \
  values.i5.TYPE[0] = _i5; \
  values.i6.TYPE[0] = _i6; \
  values.i7.TYPE[0] = _i7; \
  values.i8.TYPE[0] = _i8; \
  values.i9.TYPE[0] = _i9; \
  clear_struct_registers; \
  fregs.F0.TYPE[0] = _i0; \
  fregs.F1.TYPE[0] = _i1; \
  fregs.F2.TYPE[0] = _i2; \
  fregs.F3.TYPE[0] = _i3; \
  WRAP_CALL(_func) (_i0, _i1, _i2, _i3, _i4, _i5, _i6, _i7, _i8, _i9);

void
test_m512_varargs (void)
{
  __m512 x[10];
  int i;
  for (i = 0; i < 10; i++)
    x[i] = (__m512){32+i, 0, 0, 0, 0, 0, 0, 0};
  pass = "m512-varargs";
  def_check_int_passing_varargs (x[0], x[1], x[2], x[3], x[4], x[5],
				 x[6], x[7], x[8], x[9],
				 fun_check_passing_m512_varargs,
				 _m512);
}

void
test_m512h_varargs (void)
{
  __m512h x[10];
  int i;
  for (i = 0; i < 10; i++)
    x[i] = (__m512h) {
        1.1f16 + i, 2.2f16 + i, 3.3f16 + i, 4.4f16 + i,
	5.5f16 + i, 6.6f16 + i, 7.7f16 + i, 8.8f16 + i,
	9.9f16 + i, 10.10f16 + i, 11.11f16 + i, 12.12f16 + i,
	13.13f16 + i, 14.14f16 + i, 15.15f16 + i, 16.16f16 + i,
	17.17f16 + i, 18.18f16 + i, 19.19f16 + i, 20.20f16 + i,
	21.21f16 + i, 22.22f16 + i, 23.23f16 + i, 24.24f16 + i,
	25.25f16 + i, 26.26f16 + i, 27.27f16 + i, 28.28f16 + i,
	29.29f16 + i, 30.30f16 + i, 31.31f16 + i, 32.32f16 + i
    };
  pass = "m512h-varargs";
  def_check_int_passing_varargs (x[0], x[1], x[2], x[3], x[4], x[5],
				 x[6], x[7], x[8], x[9],
				 fun_check_passing_m512h_varargs,
				 _m512h);
}

void
do_test (void)
{
  test_m512_varargs ();
  test_m512h_varargs ();
  if (failed)
    abort ();
}