| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. 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 |
| * aomedia.org/license/patent-license/. |
| */ |
| #include <math.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <tuple> |
| |
| #include "third_party/googletest/src/googletest/include/gtest/gtest.h" |
| #include "test/acm_random.h" |
| #include "test/clear_system_state.h" |
| #include "test/register_state_check.h" |
| #include "test/util.h" |
| |
| #include "config/aom_config.h" |
| #include "config/aom_dsp_rtcd.h" |
| |
| #include "aom/aom_integer.h" |
| |
| using libaom_test::ACMRandom; |
| |
| namespace { |
| const int number_of_iterations = 200; |
| |
| typedef unsigned int (*HighbdMaskedSADFunc)(const uint16_t *src, int src_stride, |
| const uint16_t *ref, int ref_stride, |
| const uint16_t *second_pred, |
| const uint8_t *msk, int msk_stride, |
| int invert_mask); |
| typedef std::tuple<HighbdMaskedSADFunc, HighbdMaskedSADFunc> |
| HighbdMaskedSADParam; |
| |
| class HighbdMaskedSADTest |
| : public ::testing::TestWithParam<HighbdMaskedSADParam> { |
| public: |
| virtual ~HighbdMaskedSADTest() {} |
| virtual void SetUp() { |
| maskedSAD_op_ = GET_PARAM(0); |
| ref_maskedSAD_op_ = GET_PARAM(1); |
| } |
| |
| virtual void TearDown() { libaom_test::ClearSystemState(); } |
| void runHighbdMaskedSADTest(int run_times); |
| |
| protected: |
| HighbdMaskedSADFunc maskedSAD_op_; |
| HighbdMaskedSADFunc ref_maskedSAD_op_; |
| }; |
| GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(HighbdMaskedSADTest); |
| |
| void HighbdMaskedSADTest::runHighbdMaskedSADTest(int run_times) { |
| unsigned int ref_ret = 0, ret = 1; |
| ACMRandom rnd(ACMRandom::DeterministicSeed()); |
| DECLARE_ALIGNED(16, uint16_t, src_ptr[MAX_SB_SIZE * MAX_SB_SIZE]); |
| DECLARE_ALIGNED(16, uint16_t, ref_ptr[MAX_SB_SIZE * MAX_SB_SIZE]); |
| DECLARE_ALIGNED(16, uint16_t, second_pred_ptr[MAX_SB_SIZE * MAX_SB_SIZE]); |
| DECLARE_ALIGNED(16, uint8_t, msk_ptr[MAX_SB_SIZE * MAX_SB_SIZE]); |
| int err_count = 0; |
| int first_failure = -1; |
| int src_stride = MAX_SB_SIZE; |
| int ref_stride = MAX_SB_SIZE; |
| int msk_stride = MAX_SB_SIZE; |
| const int iters = run_times == 1 ? number_of_iterations : 1; |
| for (int i = 0; i < iters; ++i) { |
| for (int j = 0; j < MAX_SB_SIZE * MAX_SB_SIZE; j++) { |
| src_ptr[j] = rnd.Rand16() & 0xfff; |
| ref_ptr[j] = rnd.Rand16() & 0xfff; |
| second_pred_ptr[j] = rnd.Rand16() & 0xfff; |
| msk_ptr[j] = ((rnd.Rand8() & 0x7f) > 64) ? rnd.Rand8() & 0x3f : 64; |
| } |
| |
| for (int invert_mask = 0; invert_mask < 2; ++invert_mask) { |
| aom_usec_timer timer; |
| aom_usec_timer_start(&timer); |
| for (int repeat = 0; repeat < run_times; ++repeat) { |
| ref_ret = ref_maskedSAD_op_(src_ptr, src_stride, ref_ptr, ref_stride, |
| second_pred_ptr, msk_ptr, msk_stride, |
| invert_mask); |
| } |
| aom_usec_timer_mark(&timer); |
| const double time1 = static_cast<double>(aom_usec_timer_elapsed(&timer)); |
| aom_usec_timer_start(&timer); |
| if (run_times == 1) { |
| ASM_REGISTER_STATE_CHECK(ret = maskedSAD_op_(src_ptr, src_stride, |
| ref_ptr, ref_stride, |
| second_pred_ptr, msk_ptr, |
| msk_stride, invert_mask)); |
| } else { |
| for (int repeat = 0; repeat < run_times; ++repeat) { |
| ret = |
| maskedSAD_op_(src_ptr, src_stride, ref_ptr, ref_stride, |
| second_pred_ptr, msk_ptr, msk_stride, invert_mask); |
| } |
| } |
| aom_usec_timer_mark(&timer); |
| const double time2 = static_cast<double>(aom_usec_timer_elapsed(&timer)); |
| if (run_times > 10) { |
| printf("%7.2f/%7.2fns", time1, time2); |
| printf("(%3.2f)\n", time1 / time2); |
| } |
| if (ret != ref_ret) { |
| err_count++; |
| if (first_failure == -1) first_failure = i; |
| } |
| } |
| } |
| EXPECT_EQ(0, err_count) |
| << "Error: High BD Masked SAD Test, output doesn't match. " |
| << "First failed at test case " << first_failure; |
| } |
| |
| TEST_P(HighbdMaskedSADTest, OperationCheck) { runHighbdMaskedSADTest(1); } |
| |
| TEST_P(HighbdMaskedSADTest, DISABLED_Speed) { runHighbdMaskedSADTest(1000000); } |
| |
| using std::make_tuple; |
| |
| #if HAVE_SSSE3 |
| const HighbdMaskedSADParam hbd_msad_test[] = { |
| make_tuple(&aom_highbd_masked_sad4x4_ssse3, &aom_highbd_masked_sad4x4_c), |
| make_tuple(&aom_highbd_masked_sad4x8_ssse3, &aom_highbd_masked_sad4x8_c), |
| make_tuple(&aom_highbd_masked_sad8x4_ssse3, &aom_highbd_masked_sad8x4_c), |
| make_tuple(&aom_highbd_masked_sad8x8_ssse3, &aom_highbd_masked_sad8x8_c), |
| make_tuple(&aom_highbd_masked_sad8x16_ssse3, &aom_highbd_masked_sad8x16_c), |
| make_tuple(&aom_highbd_masked_sad16x8_ssse3, &aom_highbd_masked_sad16x8_c), |
| make_tuple(&aom_highbd_masked_sad16x16_ssse3, &aom_highbd_masked_sad16x16_c), |
| make_tuple(&aom_highbd_masked_sad16x32_ssse3, &aom_highbd_masked_sad16x32_c), |
| make_tuple(&aom_highbd_masked_sad32x16_ssse3, &aom_highbd_masked_sad32x16_c), |
| make_tuple(&aom_highbd_masked_sad32x32_ssse3, &aom_highbd_masked_sad32x32_c), |
| make_tuple(&aom_highbd_masked_sad32x64_ssse3, &aom_highbd_masked_sad32x64_c), |
| make_tuple(&aom_highbd_masked_sad64x32_ssse3, &aom_highbd_masked_sad64x32_c), |
| make_tuple(&aom_highbd_masked_sad64x64_ssse3, &aom_highbd_masked_sad64x64_c), |
| make_tuple(&aom_highbd_masked_sad64x128_ssse3, |
| &aom_highbd_masked_sad64x128_c), |
| make_tuple(&aom_highbd_masked_sad128x64_ssse3, |
| &aom_highbd_masked_sad128x64_c), |
| make_tuple(&aom_highbd_masked_sad128x128_ssse3, |
| &aom_highbd_masked_sad128x128_c), |
| make_tuple(&aom_highbd_masked_sad4x16_ssse3, &aom_highbd_masked_sad4x16_c), |
| make_tuple(&aom_highbd_masked_sad16x4_ssse3, &aom_highbd_masked_sad16x4_c), |
| make_tuple(&aom_highbd_masked_sad8x32_ssse3, &aom_highbd_masked_sad8x32_c), |
| make_tuple(&aom_highbd_masked_sad32x8_ssse3, &aom_highbd_masked_sad32x8_c), |
| make_tuple(&aom_highbd_masked_sad16x64_ssse3, &aom_highbd_masked_sad16x64_c), |
| make_tuple(&aom_highbd_masked_sad64x16_ssse3, &aom_highbd_masked_sad64x16_c), |
| make_tuple(&aom_highbd_masked_sad4x32_ssse3, &aom_highbd_masked_sad4x32_c), |
| make_tuple(&aom_highbd_masked_sad32x4_ssse3, &aom_highbd_masked_sad32x4_c), |
| make_tuple(&aom_highbd_masked_sad8x64_ssse3, &aom_highbd_masked_sad8x64_c), |
| make_tuple(&aom_highbd_masked_sad64x8_ssse3, &aom_highbd_masked_sad64x8_c), |
| make_tuple(&aom_highbd_masked_sad4x64_ssse3, &aom_highbd_masked_sad4x64_c), |
| make_tuple(&aom_highbd_masked_sad64x4_ssse3, &aom_highbd_masked_sad64x4_c), |
| }; |
| |
| INSTANTIATE_TEST_SUITE_P(SSSE3, HighbdMaskedSADTest, |
| ::testing::ValuesIn(hbd_msad_test)); |
| #endif // HAVE_SSSE3 |
| |
| #if HAVE_AVX2 |
| const HighbdMaskedSADParam hbd_msad_avx2_test[] = { |
| make_tuple(&aom_highbd_masked_sad4x4_avx2, &aom_highbd_masked_sad4x4_ssse3), |
| make_tuple(&aom_highbd_masked_sad4x8_avx2, &aom_highbd_masked_sad4x8_ssse3), |
| make_tuple(&aom_highbd_masked_sad8x4_avx2, &aom_highbd_masked_sad8x4_ssse3), |
| make_tuple(&aom_highbd_masked_sad8x8_avx2, &aom_highbd_masked_sad8x8_ssse3), |
| make_tuple(&aom_highbd_masked_sad8x16_avx2, &aom_highbd_masked_sad8x16_ssse3), |
| make_tuple(&aom_highbd_masked_sad16x8_avx2, &aom_highbd_masked_sad16x8_ssse3), |
| make_tuple(&aom_highbd_masked_sad16x16_avx2, |
| &aom_highbd_masked_sad16x16_ssse3), |
| make_tuple(&aom_highbd_masked_sad16x32_avx2, |
| &aom_highbd_masked_sad16x32_ssse3), |
| make_tuple(&aom_highbd_masked_sad32x16_avx2, |
| &aom_highbd_masked_sad32x16_ssse3), |
| make_tuple(&aom_highbd_masked_sad32x32_avx2, |
| &aom_highbd_masked_sad32x32_ssse3), |
| make_tuple(&aom_highbd_masked_sad32x64_avx2, |
| &aom_highbd_masked_sad32x64_ssse3), |
| make_tuple(&aom_highbd_masked_sad64x32_avx2, |
| &aom_highbd_masked_sad64x32_ssse3), |
| make_tuple(&aom_highbd_masked_sad64x64_avx2, |
| &aom_highbd_masked_sad64x64_ssse3), |
| make_tuple(&aom_highbd_masked_sad64x128_avx2, |
| &aom_highbd_masked_sad64x128_ssse3), |
| make_tuple(&aom_highbd_masked_sad128x64_avx2, |
| &aom_highbd_masked_sad128x64_ssse3), |
| make_tuple(&aom_highbd_masked_sad128x128_avx2, |
| &aom_highbd_masked_sad128x128_ssse3), |
| |
| make_tuple(&aom_highbd_masked_sad128x256_avx2, |
| &aom_highbd_masked_sad128x256_c), |
| make_tuple(&aom_highbd_masked_sad256x128_avx2, |
| &aom_highbd_masked_sad256x128_c), |
| make_tuple(&aom_highbd_masked_sad256x256_avx2, |
| &aom_highbd_masked_sad256x256_c), |
| |
| make_tuple(&aom_highbd_masked_sad4x16_avx2, &aom_highbd_masked_sad4x16_ssse3), |
| make_tuple(&aom_highbd_masked_sad16x4_avx2, &aom_highbd_masked_sad16x4_ssse3), |
| make_tuple(&aom_highbd_masked_sad8x32_avx2, &aom_highbd_masked_sad8x32_ssse3), |
| make_tuple(&aom_highbd_masked_sad32x8_avx2, &aom_highbd_masked_sad32x8_ssse3), |
| make_tuple(&aom_highbd_masked_sad16x64_avx2, |
| &aom_highbd_masked_sad16x64_ssse3), |
| make_tuple(&aom_highbd_masked_sad64x16_avx2, |
| &aom_highbd_masked_sad64x16_ssse3), |
| make_tuple(&aom_highbd_masked_sad4x32_avx2, &aom_highbd_masked_sad4x32_ssse3), |
| make_tuple(&aom_highbd_masked_sad32x4_avx2, &aom_highbd_masked_sad32x4_ssse3), |
| make_tuple(&aom_highbd_masked_sad8x64_avx2, &aom_highbd_masked_sad8x64_ssse3), |
| make_tuple(&aom_highbd_masked_sad64x8_avx2, &aom_highbd_masked_sad64x8_ssse3), |
| make_tuple(&aom_highbd_masked_sad4x64_avx2, &aom_highbd_masked_sad4x64_ssse3), |
| make_tuple(&aom_highbd_masked_sad64x4_avx2, &aom_highbd_masked_sad64x4_ssse3), |
| }; |
| |
| INSTANTIATE_TEST_SUITE_P(AVX2, HighbdMaskedSADTest, |
| ::testing::ValuesIn(hbd_msad_avx2_test)); |
| #endif // HAVE_AVX2 |
| |
| } // namespace |
