#include "m512_test_util.h"
#include <stdio.h>
#include <string.h>

/*
 * Here we check for _mm_[mask|maskz]roundscale_[ss|sd] intrinsics.
 */

void __declspec(noinline) do_roundscale_sd() {
  __mmask8 k8;
  volatile __m128d v1 = _mm_set_pd(2.0, (15.0));
  volatile __m128d v2 = _mm_set_pd(4.0, (1.6));
  volatile __m128d undef = _mm_set_pd(333.0, 111.0);
  __m128d exp = _mm_set_pd(2.0, (2.0));
  __m128d exp1 = _mm_set_pd(2.0, (1.0));
  __m128d expm1 = _mm_set_pd(2.0, (2.0));
  __m128d expm0 = _mm_set_pd(2.0, 111.0);
  __m128d expzm1 = _mm_set_pd(2.0, (1.0));
  __m128d expzm0 = _mm_set_pd(2.0, 0.0);

  {
    volatile __m128d r = _mm_roundscale_sd(v1, v2, 0x8);
    check_equal_ndf(&r, &exp, 2,
                    "_mm_"
                    "roundscale_sd{imm=0x8}",
                    __LINE__);
    r = _mm_roundscale_sd(v1, v2, 0x9);
    check_equal_ndf(&r, &exp1, 2,
                    "_mm_"
                    "roundscale_sd{imm=0x9}",
                    __LINE__);
    k8 = 1;
    r = _mm_mask_roundscale_sd(undef, k8, v1, v2, 0xA);
    check_equal_ndf(&r, &expm1, 2, "_mm_mask_roundscale_sd{1}{imm=0xA}",
                    __LINE__);
    k8 = 0;
    r = _mm_mask_roundscale_sd(undef, k8, v1, v2, 0x8);
    check_equal_ndf(&r, &expm0, 2, "_mm_mask_roundscale_sd{0}", __LINE__);
    k8 = 1;
    r = _mm_maskz_roundscale_sd(k8, v1, v2, 0xB);
    check_equal_ndf(&r, &expzm1, 2, "_mm_maskz_roundscale_sd{1}{imm=0xB}",
                    __LINE__);
    k8 = 0;
    r = _mm_maskz_roundscale_sd(k8, v1, v2, 0x8);
    check_equal_ndf(&r, &expzm0, 2, "_mm_maskz_roundscale_sd{0}", __LINE__);
  }
}
void __declspec(noinline) do_roundscale_ss() {
  __mmask8 k8;
  volatile __m128 v1 = _mm_set_ps(4.0f, 3.0f, 2.0f, (17.0f));
  volatile __m128 v2 = _mm_set_ps(8.0f, 7.0f, 6.0f, (-1.6f));
  volatile __m128 undef = _mm_set_ps(777.0f, 555.0f, 333.0f, 111.0f);
  __m128 exp = _mm_set_ps(4.0f, 3.0f, 2.0f, (-2.0f));
  __m128 exp1 = _mm_set_ps(4.0f, 3.0f, 2.0f, (-2.0f));
  __m128 expm1 = _mm_set_ps(4.0f, 3.0f, 2.0f, (-1.0f));
  __m128 expm0 = _mm_set_ps(4.0f, 3.0f, 2.0f, 111.0f);
  __m128 expzm1 = _mm_set_ps(4.0f, 3.0f, 2.0f, (-1.0f));
  __m128 expzm0 = _mm_set_ps(4.0f, 3.0f, 2.0f, 0.0f);

  {
    volatile __m128 r = _mm_roundscale_ss(v1, v2, 0x8);
    check_equal_nsf(&r, &exp, 4,
                    "_mm_"
                    "roundscale_ss{imm=0x8}",
                    __LINE__);
    r = _mm_roundscale_ss(v1, v2, 0x9);
    check_equal_nsf(&r, &exp1, 4,
                    "_mm_"
                    "roundscale_ss{imm=0x9}",
                    __LINE__);
    k8 = 1;
    r = _mm_mask_roundscale_ss(undef, k8, v1, v2, 0xA);
    check_equal_nsf(&r, &expm1, 4, "_mm_mask_roundscale_ss{imm=0xA}", __LINE__);
    k8 = 0;
    r = _mm_mask_roundscale_ss(undef, k8, v1, v2, 0x8);
    check_equal_nsf(&r, &expm0, 4, "_mm_mask_roundscale_ss", __LINE__);
    k8 = 1;
    r = _mm_maskz_roundscale_ss(k8, v1, v2, 0xB);
    check_equal_nsf(&r, &expzm1, 4, "_mm_maskz_roundscale_ss{imm=0xB}",
                    __LINE__);
    k8 = 0;
    r = _mm_maskz_roundscale_ss(k8, v1, v2, 0x8);
    check_equal_nsf(&r, &expzm0, 4, "_mm_maskz_roundscale_ss", __LINE__);
  }
}

int main(int argc, char *argv[]) {
  do_roundscale_sd();
  do_roundscale_ss();

  if (n_errs != 0) {
    printf("FAILED\n");
    return 1;
  }

  printf("PASSED\n");
  return 0;
}