17765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org/* 27765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org * Copyright (c) 2014 The WebM project authors. All Rights Reserved. 37765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org * 47765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org * Usee of this source code is governed by a BSD-style license 57765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org * that can be found in the LICENSE file in the root of the source 67765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org * tree. An additional intellectual property rights grant can be found 77765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org * in the file PATENTS. All contributing project authors may 87765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org * be found in the AUTHORS file in the root of the source tree. 97765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org */ 107765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 117765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org#include <immintrin.h> // AVX2 127765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org#include "vpx/vpx_integer.h" 137765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 147765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 157765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.orgint64_t vp9_block_error_avx2(const int16_t *coeff, 167765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org const int16_t *dqcoeff, 177765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org intptr_t block_size, 187765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org int64_t *ssz) { 197765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org __m256i sse_reg, ssz_reg, coeff_reg, dqcoeff_reg; 207765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org __m256i exp_dqcoeff_lo, exp_dqcoeff_hi, exp_coeff_lo, exp_coeff_hi; 217765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org __m256i sse_reg_64hi, ssz_reg_64hi; 227765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org __m128i sse_reg128, ssz_reg128; 237765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org int64_t sse; 247765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org int i; 257765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org const __m256i zero_reg = _mm256_set1_epi16(0); 267765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 277765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // init sse and ssz registerd to zero 287765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org sse_reg = _mm256_set1_epi16(0); 297765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org ssz_reg = _mm256_set1_epi16(0); 307765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 317765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org for (i = 0 ; i < block_size ; i+= 16) { 327765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // load 32 bytes from coeff and dqcoeff 337765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org coeff_reg = _mm256_loadu_si256((const __m256i *)(coeff + i)); 347765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org dqcoeff_reg = _mm256_loadu_si256((const __m256i *)(dqcoeff + i)); 357765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // dqcoeff - coeff 367765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org dqcoeff_reg = _mm256_sub_epi16(dqcoeff_reg, coeff_reg); 377765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // madd (dqcoeff - coeff) 387765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org dqcoeff_reg = _mm256_madd_epi16(dqcoeff_reg, dqcoeff_reg); 397765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // madd coeff 407765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org coeff_reg = _mm256_madd_epi16(coeff_reg, coeff_reg); 417765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // expand each double word of madd (dqcoeff - coeff) to quad word 427765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org exp_dqcoeff_lo = _mm256_unpacklo_epi32(dqcoeff_reg, zero_reg); 437765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org exp_dqcoeff_hi = _mm256_unpackhi_epi32(dqcoeff_reg, zero_reg); 447765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // expand each double word of madd (coeff) to quad word 457765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org exp_coeff_lo = _mm256_unpacklo_epi32(coeff_reg, zero_reg); 467765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org exp_coeff_hi = _mm256_unpackhi_epi32(coeff_reg, zero_reg); 477765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // add each quad word of madd (dqcoeff - coeff) and madd (coeff) 487765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org sse_reg = _mm256_add_epi64(sse_reg, exp_dqcoeff_lo); 497765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org ssz_reg = _mm256_add_epi64(ssz_reg, exp_coeff_lo); 507765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org sse_reg = _mm256_add_epi64(sse_reg, exp_dqcoeff_hi); 517765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org ssz_reg = _mm256_add_epi64(ssz_reg, exp_coeff_hi); 527765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org } 537765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // save the higher 64 bit of each 128 bit lane 547765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org sse_reg_64hi = _mm256_srli_si256(sse_reg, 8); 557765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org ssz_reg_64hi = _mm256_srli_si256(ssz_reg, 8); 567765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // add the higher 64 bit to the low 64 bit 577765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org sse_reg = _mm256_add_epi64(sse_reg, sse_reg_64hi); 587765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org ssz_reg = _mm256_add_epi64(ssz_reg, ssz_reg_64hi); 597765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 607765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // add each 64 bit from each of the 128 bit lane of the 256 bit 617765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org sse_reg128 = _mm_add_epi64(_mm256_castsi256_si128(sse_reg), 627765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org _mm256_extractf128_si256(sse_reg, 1)); 637765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 647765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org ssz_reg128 = _mm_add_epi64(_mm256_castsi256_si128(ssz_reg), 657765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org _mm256_extractf128_si256(ssz_reg, 1)); 667765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 677765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org // store the results 687765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org _mm_storel_epi64((__m128i*)(&sse), sse_reg128); 697765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 707765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org _mm_storel_epi64((__m128i*)(ssz), ssz_reg128); 717765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org return sse; 727765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org} 73