text: /*
   * Copyright (c) 2016, Alliance for Open Media. All rights reserved
   *
   * This source code is subject to the terms of the BSD 2 Clause License and
   * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
   * was not distributed with this source code in the LICENSE file, you can
   * obtain it at www.aomedia.org/license/software. If the Alliance for Open
   * Media Patent License 1.0 was not distributed with this source code in the
   * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
   */
  
  #include <cstdlib>
  #include <new>
  
  #include "third_party/googletest/src/googletest/include/gtest/gtest.h"
  
  #include "./aom_config.h"
  #include "./aom_dsp_rtcd.h"
  #include "test/acm_random.h"
  #include "test/clear_system_state.h"
  #include "test/register_state_check.h"
  #include "aom/aom_codec.h"
  #include "aom/aom_integer.h"
  #include "aom_mem/aom_mem.h"
  #include "aom_ports/mem.h"
  
  namespace {
  
  typedef unsigned int (*VarianceMxNFunc)(const uint8_t *a, int a_stride,
                                          const uint8_t *b, int b_stride,
                                          unsigned int *sse);
  typedef unsigned int (*SubpixVarMxNFunc)(const uint8_t *a, int a_stride,
                                           int xoffset, int yoffset,
                                           const uint8_t *b, int b_stride,
                                           unsigned int *sse);
  typedef unsigned int (*SubpixAvgVarMxNFunc)(const uint8_t *a, int a_stride,
                                              int xoffset, int yoffset,
                                              const uint8_t *b, int b_stride,
                                              uint32_t *sse,
                                              const uint8_t *second_pred);
  typedef unsigned int (*Get4x4SseFunc)(const uint8_t *a, int a_stride,
                                        const uint8_t *b, int b_stride);
  typedef unsigned int (*SumOfSquaresFunction)(const int16_t *src);
  typedef unsigned int (*JntSubpixAvgVarMxNFunc)(
      const uint8_t *a, int a_stride, int xoffset, int yoffset, const uint8_t *b,
      int b_stride, uint32_t *sse, const uint8_t *second_pred,
      const JNT_COMP_PARAMS *jcp_param);
  
  using libaom_test::ACMRandom;
  
  // Truncate high bit depth results by downshifting (with rounding) by:
  // 2 * (bit_depth - 8) for sse
  // (bit_depth - 8) for se
  static void RoundHighBitDepth(int bit_depth, int64_t *se, uint64_t *sse) {
    switch (bit_depth) {
      case AOM_BITS_12:
        *sse = (*sse + 128) >> 8;
        *se = (*se + 8) >> 4;
        break;
      case AOM_BITS_10:
        *sse = (*sse + 8) >> 4;
        *se = (*se + 2) >> 2;
        break;
      case AOM_BITS_8:
      default: break;
    }
  }
  
  static unsigned int mb_ss_ref(const int16_t *src) {
    unsigned int res = 0;
    for (int i = 0; i < 256; ++i) {
      res += src[i] * src[i];
    }
    return res;
  }
  
  /* Note:
   *  Our codebase calculates the "diff" value in the variance algorithm by
   *  (src - ref).
   */
  static uint32_t variance_ref(const uint8_t *src, const uint8_t *ref, int l2w,
                               int l2h, int src_stride, int ref_stride,
                               uint32_t *sse_ptr, bool use_high_bit_depth_,
                               aom_bit_depth_t bit_depth) {
    int64_t se = 0;
    uint64_t sse = 0;
    const int w = 1 << l2w;
    const int h = 1 << l2h;
    for (int y = 0; y < h; y++) {
      for (int x = 0; x < w; x++) {
        int diff;
        if (!use_high_bit_depth_) {
          diff = src[y * src_stride + x] - ref[y * ref_stride + x];
          se += diff;
          sse += diff * diff;
        } else {
          diff = CONVERT_TO_SHORTPTR(src)[y * src_stride + x] -
                 CONVERT_TO_SHORTPTR(ref)[y * ref_stride + x];
          se += diff;
          sse += diff * diff;
        }
      }
    }
    RoundHighBitDepth(bit_depth, &se, &sse);
    *sse_ptr = static_cast<uint32_t>(sse);
    return static_cast<uint32_t>(sse - ((se * se) >> (l2w + l2h)));
  }
  
  /* The subpel reference functions differ from the codec version in one aspect:
   * they calculate the bilinear factors directly instead of using a lookup table
   * and therefore upshift xoff and yoff by 1. Only every other calculated value
   * is used so the codec version shrinks the table to save space and maintain
   * compatibility with vp8.
   */
  static uint32_t subpel_variance_ref(const uint8_t *ref, const uint8_t *src,
                                      int l2w, int l2h, int xoff, int yoff,
                                      uint32_t *sse_ptr, bool use_high_bit_depth_,
                                      aom_bit_depth_t bit_depth) {
    int64_t se = 0;
    uint64_t sse = 0;
    const int w = 1 << l2w;
    const int h = 1 << l2h;
  
    xoff <<= 1;
    yoff <<= 1;
  
    for (int y = 0; y < h; y++) {
      for (int x = 0; x < w; x++) {
        // Bilinear interpolation at a 16th pel step.
        if (!use_high_bit_depth_) {
          const int a1 = ref[(w + 1) * (y + 0) + x + 0];
          const int a2 = ref[(w + 1) * (y + 0) + x + 1];
          const int b1 = ref[(w + 1) * (y + 1) + x + 0];
          const int b2 = ref[(w + 1) * (y + 1) + x + 1];
          const int a = a1 + (((a2 - a1) * xoff + 8) >> 4);
          const int b = b1 + (((b2 - b1) * xoff + 8) >> 4);
          const int r = a + (((b - a) * yoff + 8) >> 4);
          const int diff = r - src[w * y + x];
          se += diff;
          sse += diff * diff;
        } else {
          uint16_t *ref16 = CONVERT_TO_SHORTPTR(ref);
          uint16_t *src16 = CONVERT_TO_SHORTPTR(src);
          const int a1 = ref16[(w + 1) * (y + 0) + x + 0];
          const int a2 = ref16[(w + 1) * (y + 0) + x + 1];
          const int b1 = ref16[(w + 1) * (y + 1) + x + 0];
          const int b2 = ref16[(w + 1) * (y + 1) + x + 1];
          const int a = a1 + (((a2 - a1) * xoff + 8) >> 4);
          const int b = b1 + (((b2 - b1) * xoff + 8) >> 4);
          const int r = a + (((b - a) * yoff + 8) >> 4);
          const int diff = r - src16[w * y + x];
          se += diff;
          sse += diff * diff;
        }
      }
    }
    RoundHighBitDepth(bit_depth, &se, &sse);
    *sse_ptr = static_cast<uint32_t>(sse);
    return static_cast<uint32_t>(sse - ((se * se) >> (l2w + l2h)));
  }
  
  static uint32_t subpel_avg_variance_ref(const uint8_t *ref, const uint8_t *src,
                                          const uint8_t *second_pred, int l2w,
                                          int l2h, int xoff, int yoff,
                                          uint32_t *sse_ptr,
                                          bool use_high_bit_depth,
                                          aom_bit_depth_t bit_depth) {
    int64_t se = 0;
    uint64_t sse = 0;
    const int w = 1 << l2w;
    const int h = 1 << l2h;
  
    xoff <<= 1;
    yoff <<= 1;
  
    for (int y = 0; y < h; y++) {
      for (int x = 0; x < w; x++) {
        // bilinear interpolation at a 16th pel step
        if (!use_high_bit_depth) {
          const int a1 = ref[(w + 1) * (y + 0) + x + 0];
          const int a2 = ref[(w + 1) * (y + 0) + x + 1];
          const int b1 = ref[(w + 1) * (y + 1) + x + 0];
          const int b2 = ref[(w + 1) * (y + 1) + x + 1];
          const int a = a1 + (((a2 - a1) * xoff + 8) >> 4);
          const int b = b1 + (((b2 - b1) * xoff + 8) >> 4);
          const int r = a + (((b - a) * yoff + 8) >> 4);
          const int diff =
              ((r + second_pred[w * y + x] + 1) >> 1) - src[w * y + x];
          se += diff;
          sse += diff * diff;
        } else {
          const uint16_t *ref16 = CONVERT_TO_SHORTPTR(ref);
          const uint16_t *src16 = CONVERT_TO_SHORTPTR(src);
          const uint16_t *sec16 = CONVERT_TO_SHORTPTR(second_pred);
          const int a1 = ref16[(w + 1) * (y + 0) + x + 0];
          const int a2 = ref16[(w + 1) * (y + 0) + x + 1];
          const int b1 = ref16[(w + 1) * (y + 1) + x + 0];
          const int b2 = ref16[(w + 1) * (y + 1) + x + 1];
          const int a = a1 + (((a2 - a1) * xoff + 8) >> 4);
          const int b = b1 + (((b2 - b1) * xoff + 8) >> 4);
          const int r = a + (((b - a) * yoff + 8) >> 4);
          const int diff = ((r + sec16[w * y + x] + 1) >> 1) - src16[w * y + x];
          se += diff;
          sse += diff * diff;
        }
      }
    }
    RoundHighBitDepth(bit_depth, &se, &sse);
    *sse_ptr = static_cast<uint32_t>(sse);
    return static_cast<uint32_t>(sse - ((se * se) >> (l2w + l2h)));
  }
  
  static uint32_t jnt_subpel_avg_variance_ref(
      const uint8_t *ref, const uint8_t *src, const uint8_t *second_pred, int l2w,
      int l2h, int xoff, int yoff, uint32_t *sse_ptr, bool use_high_bit_depth,
      aom_bit_depth_t bit_depth, JNT_COMP_PARAMS *jcp_param) {
    int64_t se = 0;
    uint64_t sse = 0;
    const int w = 1 << l2w;
    const int h = 1 << l2h;
  
    xoff <<= 1;
    yoff <<= 1;
  
    for (int y = 0; y < h; y++) {
      for (int x = 0; x < w; x++) {
        // bilinear interpolation at a 16th pel step
        if (!use_high_bit_depth) {
          const int a1 = ref[(w + 0) * (y + 0) + x + 0];
          const int a2 = ref[(w + 0) * (y + 0) + x + 1];
          const int b1 = ref[(w + 0) * (y + 1) + x + 0];
          const int b2 = ref[(w + 0) * (y + 1) + x + 1];
          const int a = a1 + (((a2 - a1) * xoff + 8) >> 4);
          const int b = b1 + (((b2 - b1) * xoff + 8) >> 4);
          const int r = a + (((b - a) * yoff + 8) >> 4);
          const int avg = ROUND_POWER_OF_TWO(
              r * jcp_param->fwd_offset +
                  second_pred[w * y + x] * jcp_param->bck_offset,
              DIST_PRECISION_BITS);
          const int diff = avg - src[w * y + x];
  
          se += diff;
          sse += diff * diff;
        } else {
          const uint16_t *ref16 = CONVERT_TO_SHORTPTR(ref);
          const uint16_t *src16 = CONVERT_TO_SHORTPTR(src);
          const uint16_t *sec16 = CONVERT_TO_SHORTPTR(second_pred);
          const int a1 = ref16[(w + 0) * (y + 0) + x + 0];
          const int a2 = ref16[(w + 0) * (y + 0) + x + 1];
          const int b1 = ref16[(w + 0) * (y + 1) + x + 0];
          const int b2 = ref16[(w + 0) * (y + 1) + x + 1];
          const int a = a1 + (((a2 - a1) * xoff + 8) >> 4);
          const int b = b1 + (((b2 - b1) * xoff + 8) >> 4);
          const int r = a + (((b - a) * yoff + 8) >> 4);
          const int avg =
              ROUND_POWER_OF_TWO(r * jcp_param->fwd_offset +
                                     sec16[w * y + x] * jcp_param->bck_offset,
                                 DIST_PRECISION_BITS);
          const int diff = avg - src16[w * y + x];
  
          se += diff;
          sse += diff * diff;
        }
      }
    }
    RoundHighBitDepth(bit_depth, &se, &sse);
    *sse_ptr = static_cast<uint32_t>(sse);
    return static_cast<uint32_t>(sse - ((se * se) >> (l2w + l2h)));
  }
  
  ////////////////////////////////////////////////////////////////////////////////
  
  class SumOfSquaresTest : public ::testing::TestWithParam<SumOfSquaresFunction> {
   public:
    SumOfSquaresTest() : func_(GetParam()) {}
  
    virtual ~SumOfSquaresTest() { libaom_test::ClearSystemState(); }
  
   protected:
    void ConstTest();
    void RefTest();
  
    SumOfSquaresFunction func_;
    ACMRandom rnd_;
  };
  
  void SumOfSquaresTest::ConstTest() {
    int16_t mem[256];
    unsigned int res;
    for (int v = 0; v < 256; ++v) {
      for (int i = 0; i < 256; ++i) {
        mem[i] = v;
      }
      ASM_REGISTER_STATE_CHECK(res = func_(mem));
      EXPECT_EQ(256u * (v * v), res);
    }
  }
  
  void SumOfSquaresTest::RefTest() {
    int16_t mem[256];
    for (int i = 0; i < 100; ++i) {
      for (int j = 0; j < 256; ++j) {
        mem[j] = rnd_.Rand8() - rnd_.Rand8();
      }
  
      const unsigned int expected = mb_ss_ref(mem);
      unsigned int res;
      ASM_REGISTER_STATE_CHECK(res = func_(mem));
      EXPECT_EQ(expected, res);
    }
  }
  
  ////////////////////////////////////////////////////////////////////////////////
  // Encapsulating struct to store the function to test along with
  // some testing context.
  // Can be used for MSE, SSE, Variance, etc.
  
  template <typename Func>
  struct TestParams {
    TestParams(int log2w = 0, int log2h = 0, Func function = NULL,
               int bit_depth_value = 0)
        : log2width(log2w), log2height(log2h), func(function) {
      use_high_bit_depth = (bit_depth_value > 0);
      if (use_high_bit_depth) {
        bit_depth = static_cast<aom_bit_depth_t>(bit_depth_value);
      } else {
        bit_depth = AOM_BITS_8;
      }
      width = 1 << log2width;
      height = 1 << log2height;
      block_size = width * height;
      mask = (1u << bit_depth) - 1;
    }
  
    int log2width, log2height;
    int width, height;
    int block_size;
    Func func;
    aom_bit_depth_t bit_depth;
    bool use_high_bit_depth;
    uint32_t mask;
  };
  
  template <typename Func>
  std::ostream &operator<<(std::ostream &os, const TestParams<Func> &p) {
    return os << "log2width/height:" << p.log2width << "/" << p.log2height
              << " function:" << reinterpret_cast<const void *>(p.func)
              << " bit-depth:" << p.bit_depth;
  }
  
  // Main class for testing a function type
  template <typename FunctionType>
  class MainTestClass
      : public ::testing::TestWithParam<TestParams<FunctionType> > {
   public:
    virtual void SetUp() {
      params_ = this->GetParam();
  
      rnd_.Reset(ACMRandom::DeterministicSeed());
      const size_t unit =
          use_high_bit_depth() ? sizeof(uint16_t) : sizeof(uint8_t);
      src_ = reinterpret_cast<uint8_t *>(aom_memalign(16, block_size() * unit));
      ref_ = new uint8_t[block_size() * unit];
      ASSERT_TRUE(src_ != NULL);
      ASSERT_TRUE(ref_ != NULL);
      if (use_high_bit_depth()) {
        // TODO(skal): remove!
        src_ = CONVERT_TO_BYTEPTR(src_);
        ref_ = CONVERT_TO_BYTEPTR(ref_);
      }
    }
  
    virtual void TearDown() {
      if (use_high_bit_depth()) {
        // TODO(skal): remove!
        src_ = reinterpret_cast<uint8_t *>(CONVERT_TO_SHORTPTR(src_));
        ref_ = reinterpret_cast<uint8_t *>(CONVERT_TO_SHORTPTR(ref_));
      }
  
      aom_free(src_);
      delete[] ref_;
      src_ = NULL;
      ref_ = NULL;
      libaom_test::ClearSystemState();
    }
  
   protected:
    // We could sub-class MainTestClass into dedicated class for Variance
    // and MSE/SSE, but it involves a lot of 'this->xxx' dereferencing
    // to access top class fields xxx. That's cumbersome, so for now we'll just
    // implement the testing methods here:
  
    // Variance tests
    void ZeroTest();
    void RefTest();
    void RefStrideTest();
    void OneQuarterTest();
    void SpeedTest();
  
    // MSE/SSE tests
    void RefTestMse();
    void RefTestSse();
    void MaxTestMse();
    void MaxTestSse();
  
   protected:
    ACMRandom rnd_;
    uint8_t *src_;
    uint8_t *ref_;
    TestParams<FunctionType> params_;
  
    // some relay helpers
    bool use_high_bit_depth() const { return params_.use_high_bit_depth; }
    int byte_shift() const { return params_.bit_depth - 8; }
    int block_size() const { return params_.block_size; }
    int width() const { return params_.width; }
    uint32_t mask() const { return params_.mask; }
  };
  
  ////////////////////////////////////////////////////////////////////////////////
  // Tests related to variance.
  
  template <typename VarianceFunctionType>
  void MainTestClass<VarianceFunctionType>::ZeroTest() {
    for (int i = 0; i <= 255; ++i) {
      if (!use_high_bit_depth()) {
        memset(src_, i, block_size());
      } else {
        uint16_t *const src16 = CONVERT_TO_SHORTPTR(src_);
        for (int k = 0; k < block_size(); ++k) src16[k] = i << byte_shift();
      }
      for (int j = 0; j <= 255; ++j) {
        if (!use_high_bit_depth()) {
          memset(ref_, j, block_size());
        } else {
          uint16_t *const ref16 = CONVERT_TO_SHORTPTR(ref_);
          for (int k = 0; k < block_size(); ++k) ref16[k] = j << byte_shift();
        }
        unsigned int sse, var;
        ASM_REGISTER_STATE_CHECK(
            var = params_.func(src_, width(), ref_, width(), &sse));
        EXPECT_EQ(0u, var) << "src values: " << i << " ref values: " << j;
      }
    }
  }
  
  template <typename VarianceFunctionType>
  void MainTestClass<VarianceFunctionType>::RefTest() {
    for (int i = 0; i < 10; ++i) {
      for (int j = 0; j < block_size(); j++) {
        if (!use_high_bit_depth()) {
          src_[j] = rnd_.Rand8();
          ref_[j] = rnd_.Rand8();
        } else {
          CONVERT_TO_SHORTPTR(src_)[j] = rnd_.Rand16() & mask();
          CONVERT_TO_SHORTPTR(ref_)[j] = rnd_.Rand16() & mask();
        }
      }
      unsigned int sse1, sse2, var1, var2;
      const int stride = width();
      ASM_REGISTER_STATE_CHECK(
          var1 = params_.func(src_, stride, ref_, stride, &sse1));
      var2 =
          variance_ref(src_, ref_, params_.log2width, params_.log2height, stride,
                       stride, &sse2, use_high_bit_depth(), params_.bit_depth);
      EXPECT_EQ(sse1, sse2) << "Error at test index: " << i;
      EXPECT_EQ(var1, var2) << "Error at test index: " << i;
    }
  }
  
  template <typename VarianceFunctionType>
  void MainTestClass<VarianceFunctionType>::RefStrideTest() {
    for (int i = 0; i < 10; ++i) {
      const int ref_stride = (i & 1) * width();
      const int src_stride = ((i >> 1) & 1) * width();
      for (int j = 0; j < block_size(); j++) {
        const int ref_ind = (j / width()) * ref_stride + j % width();
        const int src_ind = (j / width()) * src_stride + j % width();
        if (!use_high_bit_depth()) {
          src_[src_ind] = rnd_.Rand8();
          ref_[ref_ind] = rnd_.Rand8();
        } else {
          CONVERT_TO_SHORTPTR(src_)[src_ind] = rnd_.Rand16() & mask();
          CONVERT_TO_SHORTPTR(ref_)[ref_ind] = rnd_.Rand16() & mask();
        }
      }
      unsigned int sse1, sse2;
      unsigned int var1, var2;
  
      ASM_REGISTER_STATE_CHECK(
          var1 = params_.func(src_, src_stride, ref_, ref_stride, &sse1));
      var2 = variance_ref(src_, ref_, params_.log2width, params_.log2height,
                          src_stride, ref_stride, &sse2, use_high_bit_depth(),
                          params_.bit_depth);
      EXPECT_EQ(sse1, sse2) << "Error at test index: " << i;
      EXPECT_EQ(var1, var2) << "Error at test index: " << i;
    }
  }
  
  template <typename VarianceFunctionType>
  void MainTestClass<VarianceFunctionType>::OneQuarterTest() {
    const int half = block_size() / 2;
    if (!use_high_bit_depth()) {
      memset(src_, 255, block_size());
      memset(ref_, 255, half);
      memset(ref_ + half, 0, half);
    } else {
      aom_memset16(CONVERT_TO_SHORTPTR(src_), 255 << byte_shift(), block_size());
      aom_memset16(CONVERT_TO_SHORTPTR(ref_), 255 << byte_shift(), half);
      aom_memset16(CONVERT_TO_SHORTPTR(ref_) + half, 0, half);
    }
    unsigned int sse, var, expected;
    ASM_REGISTER_STATE_CHECK(
        var = params_.func(src_, width(), ref_, width(), &sse));
    expected = block_size() * 255 * 255 / 4;
    EXPECT_EQ(expected, var);
  }
  
  template <typename VarianceFunctionType>
  void MainTestClass<VarianceFunctionType>::SpeedTest() {
    for (int j = 0; j < block_size(); j++) {
      if (!use_high_bit_depth()) {
        src_[j] = rnd_.Rand8();
        ref_[j] = rnd_.Rand8();
      } else {
        CONVERT_TO_SHORTPTR(src_)[j] = rnd_.Rand16() & mask();
        CONVERT_TO_SHORTPTR(ref_)[j] = rnd_.Rand16() & mask();
      }
    }
    unsigned int sse1, sse2, var1, var2;
    const int stride = width();
    int run_time = 1000000000 / block_size();
  
    ASM_REGISTER_STATE_CHECK(var1 =
                                 params_.func(src_, stride, ref_, stride, &sse1));
    for (int i = 0; i < run_time; ++i) {
      ASM_REGISTER_STATE_CHECK(
          var2 = params_.func(src_, stride, ref_, stride, &sse2));
    }
    EXPECT_EQ(var1, var2);
    EXPECT_EQ(sse1, sse2);
  }
  
  ////////////////////////////////////////////////////////////////////////////////
  // Tests related to MSE / SSE.
  
  template <typename FunctionType>
  void MainTestClass<FunctionType>::RefTestMse() {
    for (int i = 0; i < 10; ++i) {
      for (int j = 0; j < block_size(); ++j) {
        src_[j] = rnd_.Rand8();
        ref_[j] = rnd_.Rand8();
      }
      unsigned int sse1, sse2;
      const int stride = width();
      ASM_REGISTER_STATE_CHECK(params_.func(src_, stride, ref_, stride, &sse1));
      variance_ref(src_, ref_, params_.log2width, params_.log2height, stride,
                   stride, &sse2, false, AOM_BITS_8);
      EXPECT_EQ(sse1, sse2);
    }
  }
  
  template <typename FunctionType>
  void MainTestClass<FunctionType>::RefTestSse() {
    for (int i = 0; i < 10; ++i) {
      for (int j = 0; j < block_size(); ++j) {
        src_[j] = rnd_.Rand8();
        ref_[j] = rnd_.Rand8();
      }
      unsigned int sse2;
      unsigned int var1;
      const int stride = width();
      ASM_REGISTER_STATE_CHECK(var1 = params_.func(src_, stride, ref_, stride));
      variance_ref(src_, ref_, params_.log2width, params_.log2height, stride,
                   stride, &sse2, false, AOM_BITS_8);
      EXPECT_EQ(var1, sse2);
    }
  }
  
  template <typename FunctionType>
  void MainTestClass<FunctionType>::MaxTestMse() {
    memset(src_, 255, block_size());
    memset(ref_, 0, block_size());
    unsigned int sse;
    ASM_REGISTER_STATE_CHECK(params_.func(src_, width(), ref_, width(), &sse));
    const unsigned int expected = block_size() * 255 * 255;
    EXPECT_EQ(expected, sse);
  }
  
  template <typename FunctionType>
  void MainTestClass<FunctionType>::MaxTestSse() {
    memset(src_, 255, block_size());
    memset(ref_, 0, block_size());
    unsigned int var;
    ASM_REGISTER_STATE_CHECK(var = params_.func(src_, width(), ref_, width()));
    const unsigned int expected = block_size() * 255 * 255;
    EXPECT_EQ(expected, var);
  }
  
  ////////////////////////////////////////////////////////////////////////////////
  
  using ::testing::get;
  using ::testing::make_tuple;
  using ::testing::tuple;
  
  template <typename FunctionType>
  class SubpelVarianceTest
      : public ::testing::TestWithParam<TestParams<FunctionType> > {
   public:
    virtual void SetUp() {
      params_ = this->GetParam();
  
      rnd_.Reset(ACMRandom::DeterministicSeed());
      if (!use_high_bit_depth()) {
        src_ = reinterpret_cast<uint8_t *>(aom_memalign(32, block_size()));
        sec_ = reinterpret_cast<uint8_t *>(aom_memalign(32, block_size()));
        ref_ = reinterpret_cast<uint8_t *>(
            aom_memalign(32, block_size() + width() + height() + 1));
      } else {
        src_ = CONVERT_TO_BYTEPTR(reinterpret_cast<uint16_t *>(
            aom_memalign(32, block_size() * sizeof(uint16_t))));
        sec_ = CONVERT_TO_BYTEPTR(reinterpret_cast<uint16_t *>(
            aom_memalign(32, block_size() * sizeof(uint16_t))));
        ref_ = CONVERT_TO_BYTEPTR(aom_memalign(
            32, (block_size() + width() + height() + 1) * sizeof(uint16_t)));
      }
      ASSERT_TRUE(src_ != NULL);
      ASSERT_TRUE(sec_ != NULL);
      ASSERT_TRUE(ref_ != NULL);
    }
  
    virtual void TearDown() {
      if (!use_high_bit_depth()) {
        aom_free(src_);
        aom_free(ref_);
        aom_free(sec_);
      } else {
        aom_free(CONVERT_TO_SHORTPTR(src_));
        aom_free(CONVERT_TO_SHORTPTR(ref_));
        aom_free(CONVERT_TO_SHORTPTR(sec_));
      }
      libaom_test::ClearSystemState();
    }
  
   protected:
    void RefTest();
    void ExtremeRefTest();
  
    ACMRandom rnd_;
    uint8_t *src_;
    uint8_t *ref_;
    uint8_t *sec_;
    TestParams<FunctionType> params_;
    JNT_COMP_PARAMS jcp_param_;
  
    // some relay helpers
    bool use_high_bit_depth() const { return params_.use_high_bit_depth; }
    int byte_shift() const { return params_.bit_depth - 8; }
    int block_size() const { return params_.block_size; }
    int width() const { return params_.width; }
    int height() const { return params_.height; }
    uint32_t mask() const { return params_.mask; }
  };
  
  template <typename SubpelVarianceFunctionType>
  void SubpelVarianceTest<SubpelVarianceFunctionType>::RefTest() {
    for (int x = 0; x < 8; ++x) {
      for (int y = 0; y < 8; ++y) {
        if (!use_high_bit_depth()) {
          for (int j = 0; j < block_size(); j++) {
            src_[j] = rnd_.Rand8();
          }
          for (int j = 0; j < block_size() + width() + height() + 1; j++) {
            ref_[j] = rnd_.Rand8();
          }
        } else {
          for (int j = 0; j < block_size(); j++) {
            CONVERT_TO_SHORTPTR(src_)[j] = rnd_.Rand16() & mask();
          }
          for (int j = 0; j < block_size() + width() + height() + 1; j++) {
            CONVERT_TO_SHORTPTR(ref_)[j] = rnd_.Rand16() & mask();
          }
        }
        unsigned int sse1, sse2;
        unsigned int var1;
        ASM_REGISTER_STATE_CHECK(
            var1 = params_.func(ref_, width() + 1, x, y, src_, width(), &sse1));
        const unsigned int var2 = subpel_variance_ref(
            ref_, src_, params_.log2width, params_.log2height, x, y, &sse2,
            use_high_bit_depth(), params_.bit_depth);
        EXPECT_EQ(sse1, sse2) << "at position " << x << ", " << y;
        EXPECT_EQ(var1, var2) << "at position " << x << ", " << y;
      }
    }
  }
  
  template <typename SubpelVarianceFunctionType>
  void SubpelVarianceTest<SubpelVarianceFunctionType>::ExtremeRefTest() {
    // Compare against reference.
    // Src: Set the first half of values to 0, the second half to the maximum.
    // Ref: Set the first half of values to the maximum, the second half to 0.
    for (int x = 0; x < 8; ++x) {
      for (int y = 0; y < 8; ++y) {
        const int half = block_size() / 2;
        if (!use_high_bit_depth()) {
          memset(src_, 0, half);
          memset(src_ + half, 255, half);
          memset(ref_, 255, half);
          memset(ref_ + half, 0, half + width() + height() + 1);
        } else {
          aom_memset16(CONVERT_TO_SHORTPTR(src_), mask(), half);
          aom_memset16(CONVERT_TO_SHORTPTR(src_) + half, 0, half);
          aom_memset16(CONVERT_TO_SHORTPTR(ref_), 0, half);
          aom_memset16(CONVERT_TO_SHORTPTR(ref_) + half, mask(),
                       half + width() + height() + 1);
        }
        unsigned int sse1, sse2;
        unsigned int var1;
        ASM_REGISTER_STATE_CHECK(
            var1 = params_.func(ref_, width() + 1, x, y, src_, width(), &sse1));
        const unsigned int var2 = subpel_variance_ref(
            ref_, src_, params_.log2width, params_.log2height, x, y, &sse2,
            use_high_bit_depth(), params_.bit_depth);
        EXPECT_EQ(sse1, sse2) << "for xoffset " << x << " and yoffset " << y;
        EXPECT_EQ(var1, var2) << "for xoffset " << x << " and yoffset " << y;
      }
    }
  }
  
  template <>
  void SubpelVarianceTest<SubpixAvgVarMxNFunc>::RefTest() {
    for (int x = 0; x < 8; ++x) {
      for (int y = 0; y < 8; ++y) {
        if (!use_high_bit_depth()) {
          for (int j = 0; j < block_size(); j++) {
            src_[j] = rnd_.Rand8();
            sec_[j] = rnd_.Rand8();
          }
          for (int j = 0; j < block_size() + width() + height() + 1; j++) {
            ref_[j] = rnd_.Rand8();
          }
        } else {
          for (int j = 0; j < block_size(); j++) {
            CONVERT_TO_SHORTPTR(src_)[j] = rnd_.Rand16() & mask();
            CONVERT_TO_SHORTPTR(sec_)[j] = rnd_.Rand16() & mask();
          }
          for (int j = 0; j < block_size() + width() + height() + 1; j++) {
            CONVERT_TO_SHORTPTR(ref_)[j] = rnd_.Rand16() & mask();
          }
        }
        uint32_t sse1, sse2;
        uint32_t var1, var2;
        ASM_REGISTER_STATE_CHECK(var1 = params_.func(ref_, width() + 1, x, y,
                                                     src_, width(), &sse1, sec_));
        var2 = subpel_avg_variance_ref(ref_, src_, sec_, params_.log2width,
                                       params_.log2height, x, y, &sse2,
                                       use_high_bit_depth(), params_.bit_depth);
        EXPECT_EQ(sse1, sse2) << "at position " << x << ", " << y;
        EXPECT_EQ(var1, var2) << "at position " << x << ", " << y;
      }
    }
  }
  
  template <>
  void SubpelVarianceTest<JntSubpixAvgVarMxNFunc>::RefTest() {
    for (int x = 0; x < 8; ++x) {
      for (int y = 0; y < 8; ++y) {
        if (!use_high_bit_depth()) {
          for (int j = 0; j < block_size(); j++) {
            src_[j] = rnd_.Rand8();
            sec_[j] = rnd_.Rand8();
          }
          for (int j = 0; j < block_size() + width() + height() + 1; j++) {
            ref_[j] = rnd_.Rand8();
          }
        } else {
          for (int j = 0; j < block_size(); j++) {
            CONVERT_TO_SHORTPTR(src_)[j] = rnd_.Rand16() & mask();
            CONVERT_TO_SHORTPTR(sec_)[j] = rnd_.Rand16() & mask();
          }
          for (int j = 0; j < block_size() + width() + height() + 1; j++) {
            CONVERT_TO_SHORTPTR(ref_)[j] = rnd_.Rand16() & mask();
          }
        }
        for (int x0 = 0; x0 < 2; ++x0) {
          for (int y0 = 0; y0 < 4; ++y0) {
            uint32_t sse1, sse2;
            uint32_t var1, var2;
            jcp_param_.fwd_offset = quant_dist_lookup_table[x0][y0][0];
            jcp_param_.bck_offset = quant_dist_lookup_table[x0][y0][1];
            ASM_REGISTER_STATE_CHECK(var1 = params_.func(ref_, width() + 0, x, y,
                                                         src_, width(), &sse1,
                                                         sec_, &jcp_param_));
            var2 = jnt_subpel_avg_variance_ref(
                ref_, src_, sec_, params_.log2width, params_.log2height, x, y,
                &sse2, use_high_bit_depth(), params_.bit_depth, &jcp_param_);
            EXPECT_EQ(sse1, sse2) << "at position " << x << ", " << y;
            EXPECT_EQ(var1, var2) << "at position " << x << ", " << y;
          }
        }
      }
    }
  }
  
  typedef MainTestClass<Get4x4SseFunc> AvxSseTest;
  typedef MainTestClass<VarianceMxNFunc> AvxMseTest;
  typedef MainTestClass<VarianceMxNFunc> AvxVarianceTest;
  typedef SubpelVarianceTest<SubpixVarMxNFunc> AvxSubpelVarianceTest;
  typedef SubpelVarianceTest<SubpixAvgVarMxNFunc> AvxSubpelAvgVarianceTest;
  typedef SubpelVarianceTest<JntSubpixAvgVarMxNFunc> AvxJntSubpelAvgVarianceTest;
  
  TEST_P(AvxSseTest, RefSse) { RefTestSse(); }
  TEST_P(AvxSseTest, MaxSse) { MaxTestSse(); }
  TEST_P(AvxMseTest, RefMse) { RefTestMse(); }
  TEST_P(AvxMseTest, MaxMse) { MaxTestMse(); }
  TEST_P(AvxVarianceTest, Zero) { ZeroTest(); }
  TEST_P(AvxVarianceTest, Ref) { RefTest(); }
  TEST_P(AvxVarianceTest, RefStride) { RefStrideTest(); }
  TEST_P(AvxVarianceTest, OneQuarter) { OneQuarterTest(); }
  TEST_P(AvxVarianceTest, DISABLED_Speed) { SpeedTest(); }
  TEST_P(SumOfSquaresTest, Const) { ConstTest(); }
  TEST_P(SumOfSquaresTest, Ref) { RefTest(); }
  TEST_P(AvxSubpelVarianceTest, Ref) { RefTest(); }
  TEST_P(AvxSubpelVarianceTest, ExtremeRef) { ExtremeRefTest(); }
  TEST_P(AvxSubpelAvgVarianceTest, Ref) { RefTest(); }
  TEST_P(AvxJntSubpelAvgVarianceTest, Ref) { RefTest(); }
  
  INSTANTIATE_TEST_CASE_P(C, SumOfSquaresTest,
                          ::testing::Values(aom_get_mb_ss_c));
  
  typedef TestParams<Get4x4SseFunc> SseParams;
  INSTANTIATE_TEST_CASE_P(C, AvxSseTest,
                          ::testing::Values(SseParams(2, 2,
                                                      &aom_get4x4sse_cs_c)));
  
  typedef TestParams<VarianceMxNFunc> MseParams;
  INSTANTIATE_TEST_CASE_P(C, AvxMseTest,
                          ::testing::Values(MseParams(4, 4, &aom_mse16x16_c),
                                            MseParams(4, 3, &aom_mse16x8_c),
                                            MseParams(3, 4, &aom_mse8x16_c),
                                            MseParams(3, 3, &aom_mse8x8_c)));
  
  typedef TestParams<VarianceMxNFunc> VarianceParams;
  INSTANTIATE_TEST_CASE_P(
      C, AvxVarianceTest,
      ::testing::Values(VarianceParams(7, 7, &aom_variance128x128_c),
                        VarianceParams(7, 6, &aom_variance128x64_c),
                        VarianceParams(6, 7, &aom_variance64x128_c),
                        VarianceParams(6, 6, &aom_variance64x64_c),
                        VarianceParams(6, 5, &aom_variance64x32_c),
                        VarianceParams(5, 6, &aom_variance32x64_c),
                        VarianceParams(5, 5, &aom_variance32x32_c),
                        VarianceParams(5, 4, &aom_variance32x16_c),
                        VarianceParams(4, 5, &aom_variance16x32_c),
                        VarianceParams(4, 4, &aom_variance16x16_c),
                        VarianceParams(4, 3, &aom_variance16x8_c),
                        VarianceParams(3, 4, &aom_variance8x16_c),
                        VarianceParams(3, 3, &aom_variance8x8_c),
                        VarianceParams(3, 2, &aom_variance8x4_c),
                        VarianceParams(2, 3, &aom_variance4x8_c),
                        VarianceParams(2, 2, &aom_variance4x4_c)));
  
  typedef TestParams<SubpixVarMxNFunc> SubpelVarianceParams;
  INSTANTIATE_TEST_CASE_P(
      C, AvxSubpelVarianceTest,
      ::testing::Values(
          SubpelVarianceParams(6, 6, &aom_sub_pixel_variance64x64_c, 0),
          SubpelVarianceParams(6, 5, &aom_sub_pixel_variance64x32_c, 0),
          SubpelVarianceParams(5, 6, &aom_sub_pixel_variance32x64_c, 0),
          SubpelVarianceParams(5, 5, &aom_sub_pixel_variance32x32_c, 0),
          SubpelVarianceParams(5, 4, &aom_sub_pixel_variance32x16_c, 0),
          SubpelVarianceParams(4, 5, &aom_sub_pixel_variance16x32_c, 0),
          SubpelVarianceParams(4, 4, &aom_sub_pixel_variance16x16_c, 0),
          SubpelVarianceParams(4, 3, &aom_sub_pixel_variance16x8_c, 0),
          SubpelVarianceParams(3, 4, &aom_sub_pixel_variance8x16_c, 0),
          SubpelVarianceParams(3, 3, &aom_sub_pixel_variance8x8_c, 0),
          SubpelVarianceParams(3, 2, &aom_sub_pixel_variance8x4_c, 0),
          SubpelVarianceParams(2, 3, &aom_sub_pixel_variance4x8_c, 0),
          SubpelVarianceParams(2, 2, &aom_sub_pixel_variance4x4_c, 0)));
  
  typedef TestParams<SubpixAvgVarMxNFunc> SubpelAvgVarianceParams;
  INSTANTIATE_TEST_CASE_P(
      C, AvxSubpelAvgVarianceTest,
      ::testing::Values(
          SubpelAvgVarianceParams(6, 6, &aom_sub_pixel_avg_variance64x64_c, 0),
          SubpelAvgVarianceParams(6, 5, &aom_sub_pixel_avg_variance64x32_c, 0),
          SubpelAvgVarianceParams(5, 6, &aom_sub_pixel_avg_variance32x64_c, 0),
          SubpelAvgVarianceParams(5, 5, &aom_sub_pixel_avg_variance32x32_c, 0),
          SubpelAvgVarianceParams(5, 4, &aom_sub_pixel_avg_variance32x16_c, 0),
          SubpelAvgVarianceParams(4, 5, &aom_sub_pixel_avg_variance16x32_c, 0),
          SubpelAvgVarianceParams(4, 4, &aom_sub_pixel_avg_variance16x16_c, 0),
          SubpelAvgVarianceParams(4, 3, &aom_sub_pixel_avg_variance16x8_c, 0),
          SubpelAvgVarianceParams(3, 4, &aom_sub_pixel_avg_variance8x16_c, 0),
          SubpelAvgVarianceParams(3, 3, &aom_sub_pixel_avg_variance8x8_c, 0),
          SubpelAvgVarianceParams(3, 2, &aom_sub_pixel_avg_variance8x4_c, 0),
          SubpelAvgVarianceParams(2, 3, &aom_sub_pixel_avg_variance4x8_c, 0),
          SubpelAvgVarianceParams(2, 2, &aom_sub_pixel_avg_variance4x4_c, 0)));
  
  typedef TestParams<JntSubpixAvgVarMxNFunc> JntSubpelAvgVarianceParams;
  INSTANTIATE_TEST_CASE_P(
      C, AvxJntSubpelAvgVarianceTest,
      ::testing::Values(
          JntSubpelAvgVarianceParams(6, 6, &aom_jnt_sub_pixel_avg_variance64x64_c,
                                     0),
          JntSubpelAvgVarianceParams(6, 5, &aom_jnt_sub_pixel_avg_variance64x32_c,
                                     0),
          JntSubpelAvgVarianceParams(5, 6, &aom_jnt_sub_pixel_avg_variance32x64_c,
                                     0),
          JntSubpelAvgVarianceParams(5, 5, &aom_jnt_sub_pixel_avg_variance32x32_c,
                                     0),
          JntSubpelAvgVarianceParams(5, 4, &aom_jnt_sub_pixel_avg_variance32x16_c,
                                     0),
          JntSubpelAvgVarianceParams(4, 5, &aom_jnt_sub_pixel_avg_variance16x32_c,
                                     0),
          JntSubpelAvgVarianceParams(4, 4, &aom_jnt_sub_pixel_avg_variance16x16_c,
                                     0),
          JntSubpelAvgVarianceParams(4, 3, &aom_jnt_sub_pixel_avg_variance16x8_c,
                                     0),
          JntSubpelAvgVarianceParams(3, 4, &aom_jnt_sub_pixel_avg_variance8x16_c,
                                     0),
          JntSubpelAvgVarianceParams(3, 3, &aom_jnt_sub_pixel_avg_variance8x8_c,
                                     0),
          JntSubpelAvgVarianceParams(3, 2, &aom_jnt_sub_pixel_avg_variance8x4_c,
                                     0),
          JntSubpelAvgVarianceParams(2, 3, &aom_jnt_sub_pixel_avg_variance4x8_c,
                                     0),
          JntSubpelAvgVarianceParams(2, 2, &aom_jnt_sub_pixel_avg_variance4x4_c,
                                     0)));
  
  typedef MainTestClass<VarianceMxNFunc> AvxHBDMseTest;
  typedef MainTestClass<VarianceMxNFunc> AvxHBDVarianceTest;
  typedef SubpelVarianceTest<SubpixVarMxNFunc> AvxHBDSubpelVarianceTest;
  typedef SubpelVarianceTest<SubpixAvgVarMxNFunc> AvxHBDSubpelAvgVarianceTest;
  
  TEST_P(AvxHBDMseTest, RefMse) { RefTestMse(); }
  TEST_P(AvxHBDMseTest, MaxMse) { MaxTestMse(); }
  TEST_P(AvxHBDVarianceTest, Zero) { ZeroTest(); }
  TEST_P(AvxHBDVarianceTest, Ref) { RefTest(); }
  TEST_P(AvxHBDVarianceTest, RefStride) { RefStrideTest(); }
  TEST_P(AvxHBDVarianceTest, OneQuarter) { OneQuarterTest(); }
  TEST_P(AvxHBDSubpelVarianceTest, Ref) { RefTest(); }
  TEST_P(AvxHBDSubpelVarianceTest, ExtremeRef) { ExtremeRefTest(); }
  TEST_P(AvxHBDSubpelAvgVarianceTest, Ref) { RefTest(); }
  
  /* TODO(debargha): This test does not support the highbd version
  INSTANTIATE_TEST_CASE_P(
      C, AvxHBDMseTest,
      ::testing::Values(make_tuple(4, 4, &aom_highbd_12_mse16x16_c),
                        make_tuple(4, 4, &aom_highbd_12_mse16x8_c),
                        make_tuple(4, 4, &aom_highbd_12_mse8x16_c),
                        make_tuple(4, 4, &aom_highbd_12_mse8x8_c),
                        make_tuple(4, 4, &aom_highbd_10_mse16x16_c),
                        make_tuple(4, 4, &aom_highbd_10_mse16x8_c),
                        make_tuple(4, 4, &aom_highbd_10_mse8x16_c),
                        make_tuple(4, 4, &aom_highbd_10_mse8x8_c),
                        make_tuple(4, 4, &aom_highbd_8_mse16x16_c),
                        make_tuple(4, 4, &aom_highbd_8_mse16x8_c),
                        make_tuple(4, 4, &aom_highbd_8_mse8x16_c),
                        make_tuple(4, 4, &aom_highbd_8_mse8x8_c)));
  */
  
  const VarianceParams kArrayHBDVariance_c[] = {
  #if CONFIG_AV1
    VarianceParams(7, 7, &aom_highbd_12_variance128x128_c, 12),
    VarianceParams(7, 6, &aom_highbd_12_variance128x64_c, 12),
    VarianceParams(6, 7, &aom_highbd_12_variance64x128_c, 12),
  #endif  // CONFIG_AV1
    VarianceParams(6, 6, &aom_highbd_12_variance64x64_c, 12),
    VarianceParams(6, 5, &aom_highbd_12_variance64x32_c, 12),
    VarianceParams(5, 6, &aom_highbd_12_variance32x64_c, 12),
    VarianceParams(5, 5, &aom_highbd_12_variance32x32_c, 12),
    VarianceParams(5, 4, &aom_highbd_12_variance32x16_c, 12),
    VarianceParams(4, 5, &aom_highbd_12_variance16x32_c, 12),
    VarianceParams(4, 4, &aom_highbd_12_variance16x16_c, 12),
    VarianceParams(4, 3, &aom_highbd_12_variance16x8_c, 12),
    VarianceParams(3, 4, &aom_highbd_12_variance8x16_c, 12),
    VarianceParams(3, 3, &aom_highbd_12_variance8x8_c, 12),
    VarianceParams(3, 2, &aom_highbd_12_variance8x4_c, 12),
    VarianceParams(2, 3, &aom_highbd_12_variance4x8_c, 12),
    VarianceParams(2, 2, &aom_highbd_12_variance4x4_c, 12),
  #if CONFIG_AV1
    VarianceParams(7, 7, &aom_highbd_10_variance128x128_c, 10),
    VarianceParams(7, 6, &aom_highbd_10_variance128x64_c, 10),
    VarianceParams(6, 7, &aom_highbd_10_variance64x128_c, 10),
  #endif  // CONFIG_AV1
    VarianceParams(6, 6, &aom_highbd_10_variance64x64_c, 10),
    VarianceParams(6, 5, &aom_highbd_10_variance64x32_c, 10),
    VarianceParams(5, 6, &aom_highbd_10_variance32x64_c, 10),
    VarianceParams(5, 5, &aom_highbd_10_variance32x32_c, 10),
    VarianceParams(5, 4, &aom_highbd_10_variance32x16_c, 10),
    VarianceParams(4, 5, &aom_highbd_10_variance16x32_c, 10),
    VarianceParams(4, 4, &aom_highbd_10_variance16x16_c, 10),
    VarianceParams(4, 3, &aom_highbd_10_variance16x8_c, 10),
    VarianceParams(3, 4, &aom_highbd_10_variance8x16_c, 10),
    VarianceParams(3, 3, &aom_highbd_10_variance8x8_c, 10),
    VarianceParams(3, 2, &aom_highbd_10_variance8x4_c, 10),
    VarianceParams(2, 3, &aom_highbd_10_variance4x8_c, 10),
    VarianceParams(2, 2, &aom_highbd_10_variance4x4_c, 10),
  #if CONFIG_AV1
    VarianceParams(7, 7, &aom_highbd_8_variance128x128_c, 8),
    VarianceParams(7, 6, &aom_highbd_8_variance128x64_c, 8),
    VarianceParams(6, 7, &aom_highbd_8_variance64x128_c, 8),
  #endif  // CONFIG_AV1
    VarianceParams(6, 6, &aom_highbd_8_variance64x64_c, 8),
    VarianceParams(6, 5, &aom_highbd_8_variance64x32_c, 8),
    VarianceParams(5, 6, &aom_highbd_8_variance32x64_c, 8),
    VarianceParams(5, 5, &aom_highbd_8_variance32x32_c, 8),
    VarianceParams(5, 4, &aom_highbd_8_variance32x16_c, 8),
    VarianceParams(4, 5, &aom_highbd_8_variance16x32_c, 8),
    VarianceParams(4, 4, &aom_highbd_8_variance16x16_c, 8),
    VarianceParams(4, 3, &aom_highbd_8_variance16x8_c, 8),
    VarianceParams(3, 4, &aom_highbd_8_variance8x16_c, 8),
    VarianceParams(3, 3, &aom_highbd_8_variance8x8_c, 8),
    VarianceParams(3, 2, &aom_highbd_8_variance8x4_c, 8),
    VarianceParams(2, 3, &aom_highbd_8_variance4x8_c, 8),
    VarianceParams(2, 2, &aom_highbd_8_variance4x4_c, 8)
  };
  INSTANTIATE_TEST_CASE_P(C, AvxHBDVarianceTest,
                          ::testing::ValuesIn(kArrayHBDVariance_c));
  
  #if HAVE_SSE4_1
  INSTANTIATE_TEST_CASE_P(
      SSE4_1, AvxHBDVarianceTest,
      ::testing::Values(
          VarianceParams(2, 2, &aom_highbd_8_variance4x4_sse4_1, 8),
          VarianceParams(2, 2, &aom_highbd_10_variance4x4_sse4_1, 10),
          VarianceParams(2, 2, &aom_highbd_12_variance4x4_sse4_1, 12)));
  #endif  // HAVE_SSE4_1
  
  const SubpelVarianceParams kArrayHBDSubpelVariance_c[] = {
  #if CONFIG_AV1
    SubpelVarianceParams(7, 7, &aom_highbd_8_sub_pixel_variance128x128_c, 8),
    SubpelVarianceParams(7, 6, &aom_highbd_8_sub_pixel_variance128x64_c, 8),
    SubpelVarianceParams(6, 7, &aom_highbd_8_sub_pixel_variance64x128_c, 8),
  #endif  // CONFIG_AV1
    SubpelVarianceParams(6, 6, &aom_highbd_8_sub_pixel_variance64x64_c, 8),
    SubpelVarianceParams(6, 5, &aom_highbd_8_sub_pixel_variance64x32_c, 8),
    SubpelVarianceParams(5, 6, &aom_highbd_8_sub_pixel_variance32x64_c, 8),
    SubpelVarianceParams(5, 5, &aom_highbd_8_sub_pixel_variance32x32_c, 8),
    SubpelVarianceParams(5, 4, &aom_highbd_8_sub_pixel_variance32x16_c, 8),
    SubpelVarianceParams(4, 5, &aom_highbd_8_sub_pixel_variance16x32_c, 8),
    SubpelVarianceParams(4, 4, &aom_highbd_8_sub_pixel_variance16x16_c, 8),
    SubpelVarianceParams(4, 3, &aom_highbd_8_sub_pixel_variance16x8_c, 8),
    SubpelVarianceParams(3, 4, &aom_highbd_8_sub_pixel_variance8x16_c, 8),
    SubpelVarianceParams(3, 3, &aom_highbd_8_sub_pixel_variance8x8_c, 8),
    SubpelVarianceParams(3, 2, &aom_highbd_8_sub_pixel_variance8x4_c, 8),
    SubpelVarianceParams(2, 3, &aom_highbd_8_sub_pixel_variance4x8_c, 8),
    SubpelVarianceParams(2, 2, &aom_highbd_8_sub_pixel_variance4x4_c, 8),
  #if CONFIG_AV1
    SubpelVarianceParams(7, 7, &aom_highbd_10_sub_pixel_variance128x128_c, 10),
    SubpelVarianceParams(7, 6, &aom_highbd_10_sub_pixel_variance128x64_c, 10),
    SubpelVarianceParams(6, 7, &aom_highbd_10_sub_pixel_variance64x128_c, 10),
  #endif  // CONFIG_AV1
    SubpelVarianceParams(6, 6, &aom_highbd_10_sub_pixel_variance64x64_c, 10),
    SubpelVarianceParams(6, 5, &aom_highbd_10_sub_pixel_variance64x32_c, 10),
    SubpelVarianceParams(5, 6, &aom_highbd_10_sub_pixel_variance32x64_c, 10),
    SubpelVarianceParams(5, 5, &aom_highbd_10_sub_pixel_variance32x32_c, 10),
    SubpelVarianceParams(5, 4, &aom_highbd_10_sub_pixel_variance32x16_c, 10),
    SubpelVarianceParams(4, 5, &aom_highbd_10_sub_pixel_variance16x32_c, 10),
    SubpelVarianceParams(4, 4, &aom_highbd_10_sub_pixel_variance16x16_c, 10),
    SubpelVarianceParams(4, 3, &aom_highbd_10_sub_pixel_variance16x8_c, 10),
    SubpelVarianceParams(3, 4, &aom_highbd_10_sub_pixel_variance8x16_c, 10),
    SubpelVarianceParams(3, 3, &aom_highbd_10_sub_pixel_variance8x8_c, 10),
    SubpelVarianceParams(3, 2, &aom_highbd_10_sub_pixel_variance8x4_c, 10),
    SubpelVarianceParams(2, 3, &aom_highbd_10_sub_pixel_variance4x8_c, 10),
    SubpelVarianceParams(2, 2, &aom_highbd_10_sub_pixel_variance4x4_c, 10),
  #if CONFIG_AV1
    SubpelVarianceParams(7, 7, &aom_highbd_12_sub_pixel_variance128x128_c, 12),
    SubpelVarianceParams(7, 6, &aom_highbd_12_sub_pixel_variance128x64_c, 12),
    SubpelVarianceParams(6, 7, &aom_highbd_12_sub_pixel_variance64x128_c, 12),
  #endif  // CONFIG_AV1
    SubpelVarianceParams(6, 6, &aom_highbd_12_sub_pixel_variance64x64_c, 12),
    SubpelVarianceParams(6, 5, &aom_highbd_12_sub_pixel_variance64x32_c, 12),
    SubpelVarianceParams(5, 6, &aom_highbd_12_sub_pixel_variance32x64_c, 12),
    SubpelVarianceParams(5, 5, &aom_highbd_12_sub_pixel_variance32x32_c, 12),
    SubpelVarianceParams(5, 4, &aom_highbd_12_sub_pixel_variance32x16_c, 12),
    SubpelVarianceParams(4, 5, &aom_highbd_12_sub_pixel_variance16x32_c, 12),
    SubpelVarianceParams(4, 4, &aom_highbd_12_sub_pixel_variance16x16_c, 12),
    SubpelVarianceParams(4, 3, &aom_highbd_12_sub_pixel_variance16x8_c, 12),
    SubpelVarianceParams(3, 4, &aom_highbd_12_sub_pixel_variance8x16_c, 12),
    SubpelVarianceParams(3, 3, &aom_highbd_12_sub_pixel_variance8x8_c, 12),
    SubpelVarianceParams(3, 2, &aom_highbd_12_sub_pixel_variance8x4_c, 12),
    SubpelVarianceParams(2, 3, &aom_highbd_12_sub_pixel_variance4x8_c, 12),
    SubpelVarianceParams(2, 2, &aom_highbd_12_sub_pixel_variance4x4_c, 12),
  };
  INSTANTIATE_TEST_CASE_P(C, AvxHBDSubpelVarianceTest,
                          ::testing::ValuesIn(kArrayHBDSubpelVariance_c));
  
  const SubpelAvgVarianceParams kArrayHBDSubpelAvgVariance_c[] = {
  #if CONFIG_AV1
    SubpelAvgVarianceParams(7, 7, &aom_highbd_8_sub_pixel_avg_variance128x128_c,
                            8),
    SubpelAvgVarianceParams(7, 6, &aom_highbd_8_sub_pixel_avg_variance128x64_c,
                            8),
    SubpelAvgVarianceParams(6, 7, &aom_highbd_8_sub_pixel_avg_variance64x128_c,
                            8),
  #endif  // CONFIG_AV1
    SubpelAvgVarianceParams(6, 6, &aom_highbd_8_sub_pixel_avg_variance64x64_c, 8),
    SubpelAvgVarianceParams(6, 5, &aom_highbd_8_sub_pixel_avg_variance64x32_c, 8),
    SubpelAvgVarianceParams(5, 6, &aom_highbd_8_sub_pixel_avg_variance32x64_c, 8),
    SubpelAvgVarianceParams(5, 5, &aom_highbd_8_sub_pixel_avg_variance32x32_c, 8),
    SubpelAvgVarianceParams(5, 4, &aom_highbd_8_sub_pixel_avg_variance32x16_c, 8),
    SubpelAvgVarianceParams(4, 5, &aom_highbd_8_sub_pixel_avg_variance16x32_c, 8),
    SubpelAvgVarianceParams(4, 4, &aom_highbd_8_sub_pixel_avg_variance16x16_c, 8),
    SubpelAvgVarianceParams(4, 3, &aom_highbd_8_sub_pixel_avg_variance16x8_c, 8),
    SubpelAvgVarianceParams(3, 4, &aom_highbd_8_sub_pixel_avg_variance8x16_c, 8),
    SubpelAvgVarianceParams(3, 3, &aom_highbd_8_sub_pixel_avg_variance8x8_c, 8),
    SubpelAvgVarianceParams(3, 2, &aom_highbd_8_sub_pixel_avg_variance8x4_c, 8),
    SubpelAvgVarianceParams(2, 3, &aom_highbd_8_sub_pixel_avg_variance4x8_c, 8),
    SubpelAvgVarianceParams(2, 2, &aom_highbd_8_sub_pixel_avg_variance4x4_c, 8),
  #if CONFIG_AV1
    SubpelAvgVarianceParams(7, 7, &aom_highbd_10_sub_pixel_avg_variance128x128_c,
                            10),
    SubpelAvgVarianceParams(7, 6, &aom_highbd_10_sub_pixel_avg_variance128x64_c,
                            10),
    SubpelAvgVarianceParams(6, 7, &aom_highbd_10_sub_pixel_avg_variance64x128_c,
                            10),
  #endif  // CONFIG_AV1
    SubpelAvgVarianceParams(6, 6, &aom_highbd_10_sub_pixel_avg_variance64x64_c,
                            10),
    SubpelAvgVarianceParams(6, 5, &aom_highbd_10_sub_pixel_avg_variance64x32_c,
                            10),
    SubpelAvgVarianceParams(5, 6, &aom_highbd_10_sub_pixel_avg_variance32x64_c,
                            10),
    SubpelAvgVarianceParams(5, 5, &aom_highbd_10_sub_pixel_avg_variance32x32_c,
                            10),
    SubpelAvgVarianceParams(5, 4, &aom_highbd_10_sub_pixel_avg_variance32x16_c,
                            10),
    SubpelAvgVarianceParams(4, 5, &aom_highbd_10_sub_pixel_avg_variance16x32_c,
                            10),
    SubpelAvgVarianceParams(4, 4, &aom_highbd_10_sub_pixel_avg_variance16x16_c,
                            10),
    SubpelAvgVarianceParams(4, 3, &aom_highbd_10_sub_pixel_avg_variance16x8_c,
                            10),
    SubpelAvgVarianceParams(3, 4, &aom_highbd_10_sub_pixel_avg_variance8x16_c,
                            10),
    SubpelAvgVarianceParams(3, 3, &aom_highbd_10_sub_pixel_avg_variance8x8_c, 10),
    SubpelAvgVarianceParams(3, 2, &aom_highbd_10_sub_pixel_avg_variance8x4_c, 10),
    SubpelAvgVarianceParams(2, 3, &aom_highbd_10_sub_pixel_avg_variance4x8_c, 10),
    SubpelAvgVarianceParams(2, 2, &aom_highbd_10_sub_pixel_avg_variance4x4_c, 10),
  #if CONFIG_AV1
    SubpelAvgVarianceParams(7, 7, &aom_highbd_12_sub_pixel_avg_variance128x128_c,
                            12),
    SubpelAvgVarianceParams(7, 6, &aom_highbd_12_sub_pixel_avg_variance128x64_c,
                            12),
    SubpelAvgVarianceParams(6, 7, &aom_highbd_12_sub_pixel_avg_variance64x128_c,
                            12),
  #endif  // CONFIG_AV1
    SubpelAvgVarianceParams(6, 6, &aom_highbd_12_sub_pixel_avg_variance64x64_c,
                            12),
    SubpelAvgVarianceParams(6, 5, &aom_highbd_12_sub_pixel_avg_variance64x32_c,
                            12),
    SubpelAvgVarianceParams(5, 6, &aom_highbd_12_sub_pixel_avg_variance32x64_c,
                            12),
    SubpelAvgVarianceParams(5, 5, &aom_highbd_12_sub_pixel_avg_variance32x32_c,
                            12),
    SubpelAvgVarianceParams(5, 4, &aom_highbd_12_sub_pixel_avg_variance32x16_c,
                            12),
    SubpelAvgVarianceParams(4, 5, &aom_highbd_12_sub_pixel_avg_variance16x32_c,
                            12),
    SubpelAvgVarianceParams(4, 4, &aom_highbd_12_sub_pixel_avg_variance16x16_c,
                            12),
    SubpelAvgVarianceParams(4, 3, &aom_highbd_12_sub_pixel_avg_variance16x8_c,
                            12),
    SubpelAvgVarianceParams(3, 4, &aom_highbd_12_sub_pixel_avg_variance8x16_c,
                            12),
    SubpelAvgVarianceParams(3, 3, &aom_highbd_12_sub_pixel_avg_variance8x8_c, 12),
    SubpelAvgVarianceParams(3, 2, &aom_highbd_12_sub_pixel_avg_variance8x4_c, 12),
    SubpelAvgVarianceParams(2, 3, &aom_highbd_12_sub_pixel_avg_variance4x8_c, 12),
    SubpelAvgVarianceParams(2, 2, &aom_highbd_12_sub_pixel_avg_variance4x4_c, 12)
  };
  INSTANTIATE_TEST_CASE_P(C, AvxHBDSubpelAvgVarianceTest,
                          ::testing::ValuesIn(kArrayHBDSubpelAvgVariance_c));
  
  #if HAVE_SSE2
  INSTANTIATE_TEST_CASE_P(SSE2, SumOfSquaresTest,
                          ::testing::Values(aom_get_mb_ss_sse2));
  
  INSTANTIATE_TEST_CASE_P(SSE2, AvxMseTest,
                          ::testing::Values(MseParams(4, 4, &aom_mse16x16_sse2),
                                            MseParams(4, 3, &aom_mse16x8_sse2),
                                            MseParams(3, 4, &aom_mse8x16_sse2),
                                            MseParams(3, 3, &aom_mse8x8_sse2)));
  
  INSTANTIATE_TEST_CASE_P(
      SSE2, AvxVarianceTest,
      ::testing::Values(VarianceParams(7, 7, &aom_variance128x128_sse2),
                        VarianceParams(7, 6, &aom_variance128x64_sse2),
                        VarianceParams(6, 7, &aom_variance64x128_sse2),
                        VarianceParams(6, 6, &aom_variance64x64_sse2),
                        VarianceParams(6, 5, &aom_variance64x32_sse2),
                        VarianceParams(5, 6, &aom_variance32x64_sse2),
                        VarianceParams(5, 5, &aom_variance32x32_sse2),
                        VarianceParams(5, 4, &aom_variance32x16_sse2),
                        VarianceParams(4, 5, &aom_variance16x32_sse2),
                        VarianceParams(4, 4, &aom_variance16x16_sse2),
                        VarianceParams(4, 3, &aom_variance16x8_sse2),
                        VarianceParams(3, 4, &aom_variance8x16_sse2),
                        VarianceParams(3, 3, &aom_variance8x8_sse2),
                        VarianceParams(3, 2, &aom_variance8x4_sse2),
                        VarianceParams(2, 3, &aom_variance4x8_sse2),
                        VarianceParams(2, 2, &aom_variance4x4_sse2)));
  
  INSTANTIATE_TEST_CASE_P(
      SSE2, AvxSubpelVarianceTest,
      ::testing::Values(
          SubpelVarianceParams(6, 6, &aom_sub_pixel_variance64x64_sse2, 0),
          SubpelVarianceParams(6, 5, &aom_sub_pixel_variance64x32_sse2, 0),
          SubpelVarianceParams(5, 6, &aom_sub_pixel_variance32x64_sse2, 0),
          SubpelVarianceParams(5, 5, &aom_sub_pixel_variance32x32_sse2, 0),
          SubpelVarianceParams(5, 4, &aom_sub_pixel_variance32x16_sse2, 0),
          SubpelVarianceParams(4, 5, &aom_sub_pixel_variance16x32_sse2, 0),
          SubpelVarianceParams(4, 4, &aom_sub_pixel_variance16x16_sse2, 0),
          SubpelVarianceParams(4, 3, &aom_sub_pixel_variance16x8_sse2, 0),
          SubpelVarianceParams(3, 4, &aom_sub_pixel_variance8x16_sse2, 0),
          SubpelVarianceParams(3, 3, &aom_sub_pixel_variance8x8_sse2, 0),
          SubpelVarianceParams(3, 2, &aom_sub_pixel_variance8x4_sse2, 0),
          SubpelVarianceParams(2, 3, &aom_sub_pixel_variance4x8_sse2, 0),
          SubpelVarianceParams(2, 2, &aom_sub_pixel_variance4x4_sse2, 0)));
  
  INSTANTIATE_TEST_CASE_P(
      SSE2, AvxSubpelAvgVarianceTest,
      ::testing::Values(
          SubpelAvgVarianceParams(6, 6, &aom_sub_pixel_avg_variance64x64_sse2, 0),
          SubpelAvgVarianceParams(6, 5, &aom_sub_pixel_avg_variance64x32_sse2, 0),
          SubpelAvgVarianceParams(5, 6, &aom_sub_pixel_avg_variance32x64_sse2, 0),
          SubpelAvgVarianceParams(5, 5, &aom_sub_pixel_avg_variance32x32_sse2, 0),
          SubpelAvgVarianceParams(5, 4, &aom_sub_pixel_avg_variance32x16_sse2, 0),
          SubpelAvgVarianceParams(4, 5, &aom_sub_pixel_avg_variance16x32_sse2, 0),
          SubpelAvgVarianceParams(4, 4, &aom_sub_pixel_avg_variance16x16_sse2, 0),
          SubpelAvgVarianceParams(4, 3, &aom_sub_pixel_avg_variance16x8_sse2, 0),
          SubpelAvgVarianceParams(3, 4, &aom_sub_pixel_avg_variance8x16_sse2, 0),
          SubpelAvgVarianceParams(3, 3, &aom_sub_pixel_avg_variance8x8_sse2, 0),
          SubpelAvgVarianceParams(3, 2, &aom_sub_pixel_avg_variance8x4_sse2, 0),
          SubpelAvgVarianceParams(2, 3, &aom_sub_pixel_avg_variance4x8_sse2, 0),
          SubpelAvgVarianceParams(2, 2, &aom_sub_pixel_avg_variance4x4_sse2, 0)));
  
  #if HAVE_SSE4_1
  INSTANTIATE_TEST_CASE_P(
      SSE4_1, AvxSubpelVarianceTest,
      ::testing::Values(
          SubpelVarianceParams(2, 2, &aom_highbd_8_sub_pixel_variance4x4_sse4_1,
                               8),
          SubpelVarianceParams(2, 2, &aom_highbd_10_sub_pixel_variance4x4_sse4_1,
                               10),
          SubpelVarianceParams(2, 2, &aom_highbd_12_sub_pixel_variance4x4_sse4_1,
                               12)));
  
  INSTANTIATE_TEST_CASE_P(
      SSE4_1, AvxSubpelAvgVarianceTest,
      ::testing::Values(
          SubpelAvgVarianceParams(2, 2,
                                  &aom_highbd_8_sub_pixel_avg_variance4x4_sse4_1,
                                  8),
          SubpelAvgVarianceParams(2, 2,
                                  &aom_highbd_10_sub_pixel_avg_variance4x4_sse4_1,
                                  10),
          SubpelAvgVarianceParams(2, 2,
                                  &aom_highbd_12_sub_pixel_avg_variance4x4_sse4_1,
                                  12)));
  #endif  // HAVE_SSE4_1
  
  /* TODO(debargha): This test does not support the highbd version
  INSTANTIATE_TEST_CASE_P(
      SSE2, AvxHBDMseTest,
      ::testing::Values(MseParams(4, 4, &aom_highbd_12_mse16x16_sse2),
                        MseParams(4, 3, &aom_highbd_12_mse16x8_sse2),
                        MseParams(3, 4, &aom_highbd_12_mse8x16_sse2),
                        MseParams(3, 3, &aom_highbd_12_mse8x8_sse2),
                        MseParams(4, 4, &aom_highbd_10_mse16x16_sse2),
                        MseParams(4, 3, &aom_highbd_10_mse16x8_sse2),
                        MseParams(3, 4, &aom_highbd_10_mse8x16_sse2),
                        MseParams(3, 3, &aom_highbd_10_mse8x8_sse2),
                        MseParams(4, 4, &aom_highbd_8_mse16x16_sse2),
                        MseParams(4, 3, &aom_highbd_8_mse16x8_sse2),
                        MseParams(3, 4, &aom_highbd_8_mse8x16_sse2),
                        MseParams(3, 3, &aom_highbd_8_mse8x8_sse2)));
  */
  
  INSTANTIATE_TEST_CASE_P(
      SSE2, AvxHBDVarianceTest,
      ::testing::Values(
          VarianceParams(6, 6, &aom_highbd_12_variance64x64_sse2, 12),
          VarianceParams(6, 5, &aom_highbd_12_variance64x32_sse2, 12),
          VarianceParams(5, 6, &aom_highbd_12_variance32x64_sse2, 12),
          VarianceParams(5, 5, &aom_highbd_12_variance32x32_sse2, 12),
          VarianceParams(5, 4, &aom_highbd_12_variance32x16_sse2, 12),
          VarianceParams(4, 5, &aom_highbd_12_variance16x32_sse2, 12),
          VarianceParams(4, 4, &aom_highbd_12_variance16x16_sse2, 12),
          VarianceParams(4, 3, &aom_highbd_12_variance16x8_sse2, 12),
          VarianceParams(3, 4, &aom_highbd_12_variance8x16_sse2, 12),
          VarianceParams(3, 3, &aom_highbd_12_variance8x8_sse2, 12),
          VarianceParams(6, 6, &aom_highbd_10_variance64x64_sse2, 10),
          VarianceParams(6, 5, &aom_highbd_10_variance64x32_sse2, 10),
          VarianceParams(5, 6, &aom_highbd_10_variance32x64_sse2, 10),
          VarianceParams(5, 5, &aom_highbd_10_variance32x32_sse2, 10),
          VarianceParams(5, 4, &aom_highbd_10_variance32x16_sse2, 10),
          VarianceParams(4, 5, &aom_highbd_10_variance16x32_sse2, 10),
          VarianceParams(4, 4, &aom_highbd_10_variance16x16_sse2, 10),
          VarianceParams(4, 3, &aom_highbd_10_variance16x8_sse2, 10),
          VarianceParams(3, 4, &aom_highbd_10_variance8x16_sse2, 10),
          VarianceParams(3, 3, &aom_highbd_10_variance8x8_sse2, 10),
          VarianceParams(6, 6, &aom_highbd_8_variance64x64_sse2, 8),
          VarianceParams(6, 5, &aom_highbd_8_variance64x32_sse2, 8),
          VarianceParams(5, 6, &aom_highbd_8_variance32x64_sse2, 8),
          VarianceParams(5, 5, &aom_highbd_8_variance32x32_sse2, 8),
          VarianceParams(5, 4, &aom_highbd_8_variance32x16_sse2, 8),
          VarianceParams(4, 5, &aom_highbd_8_variance16x32_sse2, 8),
          VarianceParams(4, 4, &aom_highbd_8_variance16x16_sse2, 8),
          VarianceParams(4, 3, &aom_highbd_8_variance16x8_sse2, 8),
          VarianceParams(3, 4, &aom_highbd_8_variance8x16_sse2, 8),
          VarianceParams(3, 3, &aom_highbd_8_variance8x8_sse2, 8)));
  
  const SubpelVarianceParams kArrayHBDSubpelVariance_sse2[] = {
    SubpelVarianceParams(6, 6, &aom_highbd_12_sub_pixel_variance64x64_sse2, 12),
    SubpelVarianceParams(6, 5, &aom_highbd_12_sub_pixel_variance64x32_sse2, 12),
    SubpelVarianceParams(5, 6, &aom_highbd_12_sub_pixel_variance32x64_sse2, 12),
    SubpelVarianceParams(5, 5, &aom_highbd_12_sub_pixel_variance32x32_sse2, 12),
    SubpelVarianceParams(5, 4, &aom_highbd_12_sub_pixel_variance32x16_sse2, 12),
    SubpelVarianceParams(4, 5, &aom_highbd_12_sub_pixel_variance16x32_sse2, 12),
    SubpelVarianceParams(4, 4, &aom_highbd_12_sub_pixel_variance16x16_sse2, 12),
    SubpelVarianceParams(4, 3, &aom_highbd_12_sub_pixel_variance16x8_sse2, 12),
    SubpelVarianceParams(3, 4, &aom_highbd_12_sub_pixel_variance8x16_sse2, 12),
    SubpelVarianceParams(3, 3, &aom_highbd_12_sub_pixel_variance8x8_sse2, 12),
    SubpelVarianceParams(3, 2, &aom_highbd_12_sub_pixel_variance8x4_sse2, 12),
    SubpelVarianceParams(6, 6, &aom_highbd_10_sub_pixel_variance64x64_sse2, 10),
    SubpelVarianceParams(6, 5, &aom_highbd_10_sub_pixel_variance64x32_sse2, 10),
    SubpelVarianceParams(5, 6, &aom_highbd_10_sub_pixel_variance32x64_sse2, 10),
    SubpelVarianceParams(5, 5, &aom_highbd_10_sub_pixel_variance32x32_sse2, 10),
    SubpelVarianceParams(5, 4, &aom_highbd_10_sub_pixel_variance32x16_sse2, 10),
    SubpelVarianceParams(4, 5, &aom_highbd_10_sub_pixel_variance16x32_sse2, 10),
    SubpelVarianceParams(4, 4, &aom_highbd_10_sub_pixel_variance16x16_sse2, 10),
    SubpelVarianceParams(4, 3, &aom_highbd_10_sub_pixel_variance16x8_sse2, 10),
    SubpelVarianceParams(3, 4, &aom_highbd_10_sub_pixel_variance8x16_sse2, 10),
    SubpelVarianceParams(3, 3, &aom_highbd_10_sub_pixel_variance8x8_sse2, 10),
    SubpelVarianceParams(3, 2, &aom_highbd_10_sub_pixel_variance8x4_sse2, 10),
    SubpelVarianceParams(6, 6, &aom_highbd_8_sub_pixel_variance64x64_sse2, 8),
    SubpelVarianceParams(6, 5, &aom_highbd_8_sub_pixel_variance64x32_sse2, 8),
    SubpelVarianceParams(5, 6, &aom_highbd_8_sub_pixel_variance32x64_sse2, 8),
    SubpelVarianceParams(5, 5, &aom_highbd_8_sub_pixel_variance32x32_sse2, 8),
    SubpelVarianceParams(5, 4, &aom_highbd_8_sub_pixel_variance32x16_sse2, 8),
    SubpelVarianceParams(4, 5, &aom_highbd_8_sub_pixel_variance16x32_sse2, 8),
    SubpelVarianceParams(4, 4, &aom_highbd_8_sub_pixel_variance16x16_sse2, 8),
    SubpelVarianceParams(4, 3, &aom_highbd_8_sub_pixel_variance16x8_sse2, 8),
    SubpelVarianceParams(3, 4, &aom_highbd_8_sub_pixel_variance8x16_sse2, 8),
    SubpelVarianceParams(3, 3, &aom_highbd_8_sub_pixel_variance8x8_sse2, 8),
    SubpelVarianceParams(3, 2, &aom_highbd_8_sub_pixel_variance8x4_sse2, 8)
  };
  
  INSTANTIATE_TEST_CASE_P(SSE2, AvxHBDSubpelVarianceTest,
                          ::testing::ValuesIn(kArrayHBDSubpelVariance_sse2));
  
  const SubpelAvgVarianceParams kArrayHBDSubpelAvgVariance_sse2[] = {
    SubpelAvgVarianceParams(6, 6, &aom_highbd_12_sub_pixel_avg_variance64x64_sse2,
                            12),
    SubpelAvgVarianceParams(6, 5, &aom_highbd_12_sub_pixel_avg_variance64x32_sse2,
                            12),
    SubpelAvgVarianceParams(5, 6, &aom_highbd_12_sub_pixel_avg_variance32x64_sse2,
                            12),
    SubpelAvgVarianceParams(5, 5, &aom_highbd_12_sub_pixel_avg_variance32x32_sse2,
                            12),
    SubpelAvgVarianceParams(5, 4, &aom_highbd_12_sub_pixel_avg_variance32x16_sse2,
                            12),
    SubpelAvgVarianceParams(4, 5, &aom_highbd_12_sub_pixel_avg_variance16x32_sse2,
                            12),
    SubpelAvgVarianceParams(4, 4, &aom_highbd_12_sub_pixel_avg_variance16x16_sse2,
                            12),
    SubpelAvgVarianceParams(4, 3, &aom_highbd_12_sub_pixel_avg_variance16x8_sse2,
                            12),
    SubpelAvgVarianceParams(3, 4, &aom_highbd_12_sub_pixel_avg_variance8x16_sse2,
                            12),
    SubpelAvgVarianceParams(3, 3, &aom_highbd_12_sub_pixel_avg_variance8x8_sse2,
                            12),
    SubpelAvgVarianceParams(3, 2, &aom_highbd_12_sub_pixel_avg_variance8x4_sse2,
                            12),
    SubpelAvgVarianceParams(6, 6, &aom_highbd_10_sub_pixel_avg_variance64x64_sse2,
                            10),
    SubpelAvgVarianceParams(6, 5, &aom_highbd_10_sub_pixel_avg_variance64x32_sse2,
                            10),
    SubpelAvgVarianceParams(5, 6, &aom_highbd_10_sub_pixel_avg_variance32x64_sse2,
                            10),
    SubpelAvgVarianceParams(5, 5, &aom_highbd_10_sub_pixel_avg_variance32x32_sse2,
                            10),
    SubpelAvgVarianceParams(5, 4, &aom_highbd_10_sub_pixel_avg_variance32x16_sse2,
                            10),
    SubpelAvgVarianceParams(4, 5, &aom_highbd_10_sub_pixel_avg_variance16x32_sse2,
                            10),
    SubpelAvgVarianceParams(4, 4, &aom_highbd_10_sub_pixel_avg_variance16x16_sse2,
                            10),
    SubpelAvgVarianceParams(4, 3, &aom_highbd_10_sub_pixel_avg_variance16x8_sse2,
                            10),
    SubpelAvgVarianceParams(3, 4, &aom_highbd_10_sub_pixel_avg_variance8x16_sse2,
                            10),
    SubpelAvgVarianceParams(3, 3, &aom_highbd_10_sub_pixel_avg_variance8x8_sse2,
                            10),
    SubpelAvgVarianceParams(3, 2, &aom_highbd_10_sub_pixel_avg_variance8x4_sse2,
                            10),
    SubpelAvgVarianceParams(6, 6, &aom_highbd_8_sub_pixel_avg_variance64x64_sse2,
                            8),
    SubpelAvgVarianceParams(6, 5, &aom_highbd_8_sub_pixel_avg_variance64x32_sse2,
                            8),
    SubpelAvgVarianceParams(5, 6, &aom_highbd_8_sub_pixel_avg_variance32x64_sse2,
                            8),
    SubpelAvgVarianceParams(5, 5, &aom_highbd_8_sub_pixel_avg_variance32x32_sse2,
                            8),
    SubpelAvgVarianceParams(5, 4, &aom_highbd_8_sub_pixel_avg_variance32x16_sse2,
                            8),
    SubpelAvgVarianceParams(4, 5, &aom_highbd_8_sub_pixel_avg_variance16x32_sse2,
                            8),
    SubpelAvgVarianceParams(4, 4, &aom_highbd_8_sub_pixel_avg_variance16x16_sse2,
                            8),
    SubpelAvgVarianceParams(4, 3, &aom_highbd_8_sub_pixel_avg_variance16x8_sse2,
                            8),
    SubpelAvgVarianceParams(3, 4, &aom_highbd_8_sub_pixel_avg_variance8x16_sse2,
                            8),
    SubpelAvgVarianceParams(3, 3, &aom_highbd_8_sub_pixel_avg_variance8x8_sse2,
                            8),
    SubpelAvgVarianceParams(3, 2, &aom_highbd_8_sub_pixel_avg_variance8x4_sse2, 8)
  };
  
  INSTANTIATE_TEST_CASE_P(SSE2, AvxHBDSubpelAvgVarianceTest,
                          ::testing::ValuesIn(kArrayHBDSubpelAvgVariance_sse2));
  #endif  // HAVE_SSE2
  
  #if HAVE_SSSE3
  INSTANTIATE_TEST_CASE_P(
      SSSE3, AvxSubpelVarianceTest,
      ::testing::Values(
          SubpelVarianceParams(6, 6, &aom_sub_pixel_variance64x64_ssse3, 0),
          SubpelVarianceParams(6, 5, &aom_sub_pixel_variance64x32_ssse3, 0),
          SubpelVarianceParams(5, 6, &aom_sub_pixel_variance32x64_ssse3, 0),
          SubpelVarianceParams(5, 5, &aom_sub_pixel_variance32x32_ssse3, 0),
          SubpelVarianceParams(5, 4, &aom_sub_pixel_variance32x16_ssse3, 0),
          SubpelVarianceParams(4, 5, &aom_sub_pixel_variance16x32_ssse3, 0),
          SubpelVarianceParams(4, 4, &aom_sub_pixel_variance16x16_ssse3, 0),
          SubpelVarianceParams(4, 3, &aom_sub_pixel_variance16x8_ssse3, 0),
          SubpelVarianceParams(3, 4, &aom_sub_pixel_variance8x16_ssse3, 0),
          SubpelVarianceParams(3, 3, &aom_sub_pixel_variance8x8_ssse3, 0),
          SubpelVarianceParams(3, 2, &aom_sub_pixel_variance8x4_ssse3, 0),
          SubpelVarianceParams(2, 3, &aom_sub_pixel_variance4x8_ssse3, 0),
          SubpelVarianceParams(2, 2, &aom_sub_pixel_variance4x4_ssse3, 0)));
  
  INSTANTIATE_TEST_CASE_P(
      SSSE3, AvxSubpelAvgVarianceTest,
      ::testing::Values(
          SubpelAvgVarianceParams(6, 6, &aom_sub_pixel_avg_variance64x64_ssse3,
                                  0),
          SubpelAvgVarianceParams(6, 5, &aom_sub_pixel_avg_variance64x32_ssse3,
                                  0),
          SubpelAvgVarianceParams(5, 6, &aom_sub_pixel_avg_variance32x64_ssse3,
                                  0),
          SubpelAvgVarianceParams(5, 5, &aom_sub_pixel_avg_variance32x32_ssse3,
                                  0),
          SubpelAvgVarianceParams(5, 4, &aom_sub_pixel_avg_variance32x16_ssse3,
                                  0),
          SubpelAvgVarianceParams(4, 5, &aom_sub_pixel_avg_variance16x32_ssse3,
                                  0),
          SubpelAvgVarianceParams(4, 4, &aom_sub_pixel_avg_variance16x16_ssse3,
                                  0),
          SubpelAvgVarianceParams(4, 3, &aom_sub_pixel_avg_variance16x8_ssse3, 0),
          SubpelAvgVarianceParams(3, 4, &aom_sub_pixel_avg_variance8x16_ssse3, 0),
          SubpelAvgVarianceParams(3, 3, &aom_sub_pixel_avg_variance8x8_ssse3, 0),
          SubpelAvgVarianceParams(3, 2, &aom_sub_pixel_avg_variance8x4_ssse3, 0),
          SubpelAvgVarianceParams(2, 3, &aom_sub_pixel_avg_variance4x8_ssse3, 0),
          SubpelAvgVarianceParams(2, 2, &aom_sub_pixel_avg_variance4x4_ssse3,
                                  0)));
  
  INSTANTIATE_TEST_CASE_P(
      SSSE3, AvxJntSubpelAvgVarianceTest,
      ::testing::Values(
          JntSubpelAvgVarianceParams(6, 6,
                                     &aom_jnt_sub_pixel_avg_variance64x64_ssse3,
                                     0),
          JntSubpelAvgVarianceParams(6, 5,
                                     &aom_jnt_sub_pixel_avg_variance64x32_ssse3,
                                     0),
          JntSubpelAvgVarianceParams(5, 6,
                                     &aom_jnt_sub_pixel_avg_variance32x64_ssse3,
                                     0),
          JntSubpelAvgVarianceParams(5, 5,
                                     &aom_jnt_sub_pixel_avg_variance32x32_ssse3,
                                     0),
          JntSubpelAvgVarianceParams(5, 4,
                                     &aom_jnt_sub_pixel_avg_variance32x16_ssse3,
                                     0),
          JntSubpelAvgVarianceParams(4, 5,
                                     &aom_jnt_sub_pixel_avg_variance16x32_ssse3,
                                     0),
          JntSubpelAvgVarianceParams(4, 4,
                                     &aom_jnt_sub_pixel_avg_variance16x16_ssse3,
                                     0),
          JntSubpelAvgVarianceParams(4, 3,
                                     &aom_jnt_sub_pixel_avg_variance16x8_ssse3,
                                     0),
          JntSubpelAvgVarianceParams(3, 4,
                                     &aom_jnt_sub_pixel_avg_variance8x16_ssse3,
                                     0),
          JntSubpelAvgVarianceParams(3, 3,
                                     &aom_jnt_sub_pixel_avg_variance8x8_ssse3, 0),
          JntSubpelAvgVarianceParams(3, 2,
                                     &aom_jnt_sub_pixel_avg_variance8x4_ssse3, 0),
          JntSubpelAvgVarianceParams(2, 3,
                                     &aom_jnt_sub_pixel_avg_variance4x8_ssse3, 0),
          JntSubpelAvgVarianceParams(2, 2,
                                     &aom_jnt_sub_pixel_avg_variance4x4_ssse3,
                                     0)));
  #endif  // HAVE_SSSE3
  
  #if HAVE_AVX2
  INSTANTIATE_TEST_CASE_P(AVX2, AvxMseTest,
                          ::testing::Values(MseParams(4, 4, &aom_mse16x16_avx2)));
  
  INSTANTIATE_TEST_CASE_P(
      AVX2, AvxVarianceTest,
      ::testing::Values(VarianceParams(7, 7, &aom_variance128x128_avx2),
                        VarianceParams(7, 6, &aom_variance128x64_avx2),
                        VarianceParams(6, 7, &aom_variance64x128_avx2),
                        VarianceParams(6, 6, &aom_variance64x64_avx2),
                        VarianceParams(6, 5, &aom_variance64x32_avx2),
                        VarianceParams(5, 5, &aom_variance32x32_avx2),
                        VarianceParams(5, 4, &aom_variance32x16_avx2),
                        VarianceParams(4, 4, &aom_variance16x16_avx2)));
  
  INSTANTIATE_TEST_CASE_P(
      AVX2, AvxSubpelVarianceTest,
      ::testing::Values(
          SubpelVarianceParams(6, 6, &aom_sub_pixel_variance64x64_avx2, 0),
          SubpelVarianceParams(5, 5, &aom_sub_pixel_variance32x32_avx2, 0)));
  
  INSTANTIATE_TEST_CASE_P(
      AVX2, AvxSubpelAvgVarianceTest,
      ::testing::Values(
          SubpelAvgVarianceParams(6, 6, &aom_sub_pixel_avg_variance64x64_avx2, 0),
          SubpelAvgVarianceParams(5, 5, &aom_sub_pixel_avg_variance32x32_avx2,
                                  0)));
  #endif  // HAVE_AVX2
  
  #if HAVE_NEON
  INSTANTIATE_TEST_CASE_P(NEON, AvxSseTest,
                          ::testing::Values(SseParams(2, 2,
                                                      &aom_get4x4sse_cs_neon)));
  
  INSTANTIATE_TEST_CASE_P(NEON, AvxMseTest,
                          ::testing::Values(MseParams(4, 4, &aom_mse16x16_neon)));
  
  INSTANTIATE_TEST_CASE_P(
      NEON, AvxVarianceTest,
      ::testing::Values(VarianceParams(6, 6, &aom_variance64x64_neon),
                        VarianceParams(6, 5, &aom_variance64x32_neon),
                        VarianceParams(5, 6, &aom_variance32x64_neon),
                        VarianceParams(5, 5, &aom_variance32x32_neon),
                        VarianceParams(4, 4, &aom_variance16x16_neon),
                        VarianceParams(4, 3, &aom_variance16x8_neon),
                        VarianceParams(3, 4, &aom_variance8x16_neon),
                        VarianceParams(3, 3, &aom_variance8x8_neon)));
  
  INSTANTIATE_TEST_CASE_P(
      NEON, AvxSubpelVarianceTest,
      ::testing::Values(
          SubpelVarianceParams(6, 6, &aom_sub_pixel_variance64x64_neon, 0),
          SubpelVarianceParams(5, 5, &aom_sub_pixel_variance32x32_neon, 0),
          SubpelVarianceParams(4, 4, &aom_sub_pixel_variance16x16_neon, 0),
          SubpelVarianceParams(3, 3, &aom_sub_pixel_variance8x8_neon, 0)));
  #endif  // HAVE_NEON
  
  #if HAVE_MSA
  INSTANTIATE_TEST_CASE_P(MSA, SumOfSquaresTest,
                          ::testing::Values(aom_get_mb_ss_msa));
  
  INSTANTIATE_TEST_CASE_P(MSA, AvxSseTest,
                          ::testing::Values(SseParams(2, 2,
                                                      &aom_get4x4sse_cs_msa)));
  
  INSTANTIATE_TEST_CASE_P(MSA, AvxMseTest,
                          ::testing::Values(MseParams(4, 4, &aom_mse16x16_msa),
                                            MseParams(4, 3, &aom_mse16x8_msa),
                                            MseParams(3, 4, &aom_mse8x16_msa),
                                            MseParams(3, 3, &aom_mse8x8_msa)));
  
  INSTANTIATE_TEST_CASE_P(
      MSA, AvxVarianceTest,
      ::testing::Values(VarianceParams(6, 6, &aom_variance64x64_msa),
                        VarianceParams(6, 5, &aom_variance64x32_msa),
                        VarianceParams(5, 6, &aom_variance32x64_msa),
                        VarianceParams(5, 5, &aom_variance32x32_msa),
                        VarianceParams(5, 4, &aom_variance32x16_msa),
                        VarianceParams(4, 5, &aom_variance16x32_msa),
                        VarianceParams(4, 4, &aom_variance16x16_msa),
                        VarianceParams(4, 3, &aom_variance16x8_msa),
                        VarianceParams(3, 4, &aom_variance8x16_msa),
                        VarianceParams(3, 3, &aom_variance8x8_msa),
                        VarianceParams(3, 2, &aom_variance8x4_msa),
                        VarianceParams(2, 3, &aom_variance4x8_msa),
                        VarianceParams(2, 2, &aom_variance4x4_msa)));
  
  INSTANTIATE_TEST_CASE_P(
      MSA, AvxSubpelVarianceTest,
      ::testing::Values(
          SubpelVarianceParams(2, 2, &aom_sub_pixel_variance4x4_msa, 0),
          SubpelVarianceParams(2, 3, &aom_sub_pixel_variance4x8_msa, 0),
          SubpelVarianceParams(3, 2, &aom_sub_pixel_variance8x4_msa, 0),
          SubpelVarianceParams(3, 3, &aom_sub_pixel_variance8x8_msa, 0),
          SubpelVarianceParams(3, 4, &aom_sub_pixel_variance8x16_msa, 0),
          SubpelVarianceParams(4, 3, &aom_sub_pixel_variance16x8_msa, 0),
          SubpelVarianceParams(4, 4, &aom_sub_pixel_variance16x16_msa, 0),
          SubpelVarianceParams(4, 5, &aom_sub_pixel_variance16x32_msa, 0),
          SubpelVarianceParams(5, 4, &aom_sub_pixel_variance32x16_msa, 0),
          SubpelVarianceParams(5, 5, &aom_sub_pixel_variance32x32_msa, 0),
          SubpelVarianceParams(5, 6, &aom_sub_pixel_variance32x64_msa, 0),
          SubpelVarianceParams(6, 5, &aom_sub_pixel_variance64x32_msa, 0),
          SubpelVarianceParams(6, 6, &aom_sub_pixel_variance64x64_msa, 0)));
  
  INSTANTIATE_TEST_CASE_P(
      MSA, AvxSubpelAvgVarianceTest,
      ::testing::Values(
          SubpelAvgVarianceParams(6, 6, &aom_sub_pixel_avg_variance64x64_msa, 0),
          SubpelAvgVarianceParams(6, 5, &aom_sub_pixel_avg_variance64x32_msa, 0),
          SubpelAvgVarianceParams(5, 6, &aom_sub_pixel_avg_variance32x64_msa, 0),
          SubpelAvgVarianceParams(5, 5, &aom_sub_pixel_avg_variance32x32_msa, 0),
          SubpelAvgVarianceParams(5, 4, &aom_sub_pixel_avg_variance32x16_msa, 0),
          SubpelAvgVarianceParams(4, 5, &aom_sub_pixel_avg_variance16x32_msa, 0),
          SubpelAvgVarianceParams(4, 4, &aom_sub_pixel_avg_variance16x16_msa, 0),
          SubpelAvgVarianceParams(4, 3, &aom_sub_pixel_avg_variance16x8_msa, 0),
          SubpelAvgVarianceParams(3, 4, &aom_sub_pixel_avg_variance8x16_msa, 0),
          SubpelAvgVarianceParams(3, 3, &aom_sub_pixel_avg_variance8x8_msa, 0),
          SubpelAvgVarianceParams(3, 2, &aom_sub_pixel_avg_variance8x4_msa, 0),
          SubpelAvgVarianceParams(2, 3, &aom_sub_pixel_avg_variance4x8_msa, 0),
          SubpelAvgVarianceParams(2, 2, &aom_sub_pixel_avg_variance4x4_msa, 0)));
  #endif  // HAVE_MSA
  }  // namespace
  