test/f32-vscaleextexp.cc

// Copyright 2019 Google LLC
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.

#include <gtest/gtest.h>

#include <xnnpack/common.h>
#include <xnnpack/isa-checks.h>

#include <xnnpack/vscaleextexp.h>
#include "vscaleextexp-microkernel-tester.h"


#if CPUINFO_ARCH_X86 || CPUINFO_ARCH_X86_64
  TEST(F32_VSCALEEXTEXP__AVX2_P5_UNROLL64, n_lt_64) {
    TEST_REQUIRES_X86_AVX2;
    for (size_t n = 1; n < 64; n++) {
      VScaleExtExpMicrokernelTester()
        .n(n)
        .test(xnn_f32_vscaleextexp_ukernel__avx2_p5_unroll64);
    }
  }

  TEST(F32_VSCALEEXTEXP__AVX2_P5_UNROLL64, n_eq_64) {
    TEST_REQUIRES_X86_AVX2;
    VScaleExtExpMicrokernelTester()
      .n(64)
      .test(xnn_f32_vscaleextexp_ukernel__avx2_p5_unroll64);
  }

  TEST(F32_VSCALEEXTEXP__AVX2_P5_UNROLL64, n_div_64) {
    TEST_REQUIRES_X86_AVX2;
    for (size_t n = 128; n < 384; n += 64) {
      VScaleExtExpMicrokernelTester()
        .n(n)
        .test(xnn_f32_vscaleextexp_ukernel__avx2_p5_unroll64);
    }
  }

  TEST(F32_VSCALEEXTEXP__AVX2_P5_UNROLL64, n_gt_64) {
    TEST_REQUIRES_X86_AVX2;
    for (size_t n = 64; n < 128; n++) {
      VScaleExtExpMicrokernelTester()
        .n(n)
        .test(xnn_f32_vscaleextexp_ukernel__avx2_p5_unroll64);
    }
  }
#endif  // CPUINFO_ARCH_X86 || CPUINFO_ARCH_X86_64

#if CPUINFO_ARCH_X86 || CPUINFO_ARCH_X86_64
  TEST(F32_VSCALEEXTEXP__AVX512F_P5_SCALEF_UNROLL128, n_lt_128) {
    TEST_REQUIRES_X86_AVX512F;
    for (size_t n = 1; n < 128; n++) {
      VScaleExtExpMicrokernelTester()
        .n(n)
        .test(xnn_f32_vscaleextexp_ukernel__avx512f_p5_scalef_unroll128);
    }
  }

  TEST(F32_VSCALEEXTEXP__AVX512F_P5_SCALEF_UNROLL128, n_eq_128) {
    TEST_REQUIRES_X86_AVX512F;
    VScaleExtExpMicrokernelTester()
      .n(128)
      .test(xnn_f32_vscaleextexp_ukernel__avx512f_p5_scalef_unroll128);
  }

  TEST(F32_VSCALEEXTEXP__AVX512F_P5_SCALEF_UNROLL128, n_div_128) {
    TEST_REQUIRES_X86_AVX512F;
    for (size_t n = 256; n < 768; n += 128) {
      VScaleExtExpMicrokernelTester()
        .n(n)
        .test(xnn_f32_vscaleextexp_ukernel__avx512f_p5_scalef_unroll128);
    }
  }

  TEST(F32_VSCALEEXTEXP__AVX512F_P5_SCALEF_UNROLL128, n_gt_128) {
    TEST_REQUIRES_X86_AVX512F;
    for (size_t n = 128; n < 256; n++) {
      VScaleExtExpMicrokernelTester()
        .n(n)
        .test(xnn_f32_vscaleextexp_ukernel__avx512f_p5_scalef_unroll128);
    }
  }
#endif  // CPUINFO_ARCH_X86 || CPUINFO_ARCH_X86_64