| /* |
| * Copyright (c) 2019, Alliance for Open Media. All Rights Reserved. |
| * |
| * Use of this source code is governed by a BSD-style license |
| * that can be found in the LICENSE file in the root of the source |
| * tree. An additional intellectual property rights grant can be found |
| * in the file PATENTS. All contributing project authors may |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| |
| #include <arm_neon.h> |
| #include <assert.h> |
| |
| #include "av1/common/arm/mem_neon.h" |
| #include "aom_dsp/aom_dsp_common.h" |
| |
| int64_t av1_block_error_neon(const tran_low_t *coeff, const tran_low_t *dqcoeff, |
| intptr_t block_size, int64_t *ssz) { |
| int64x2_t error = vdupq_n_s64(0); |
| int64x2_t sqcoeff = vdupq_n_s64(0); |
| |
| assert(block_size >= 8); |
| assert((block_size % 8) == 0); |
| |
| do { |
| const int16x8_t c = load_tran_low_to_s16q(coeff); |
| const int16x8_t d = load_tran_low_to_s16q(dqcoeff); |
| const int16x8_t diff = vsubq_s16(c, d); |
| const int16x4_t diff_lo = vget_low_s16(diff); |
| const int16x4_t diff_hi = vget_high_s16(diff); |
| // diff is 15-bits, the squares 30, so we can store 2 in 31-bits before |
| // accumulating them in 64-bits. |
| const int32x4_t err0 = vmull_s16(diff_lo, diff_lo); |
| const int32x4_t err1 = vmlal_s16(err0, diff_hi, diff_hi); |
| const int64x2_t err2 = vaddl_s32(vget_low_s32(err1), vget_high_s32(err1)); |
| error = vaddq_s64(error, err2); |
| |
| const int16x4_t coeff_lo = vget_low_s16(c); |
| const int16x4_t coeff_hi = vget_high_s16(c); |
| const int32x4_t sqcoeff0 = vmull_s16(coeff_lo, coeff_lo); |
| const int32x4_t sqcoeff1 = vmlal_s16(sqcoeff0, coeff_hi, coeff_hi); |
| const int64x2_t sqcoeff2 = |
| vaddl_s32(vget_low_s32(sqcoeff1), vget_high_s32(sqcoeff1)); |
| sqcoeff = vaddq_s64(sqcoeff, sqcoeff2); |
| |
| coeff += 8; |
| dqcoeff += 8; |
| block_size -= 8; |
| } while (block_size != 0); |
| |
| #if defined(__aarch64__) |
| *ssz = vaddvq_s64(sqcoeff); |
| return vaddvq_s64(error); |
| #else |
| *ssz = vgetq_lane_s64(sqcoeff, 0) + vgetq_lane_s64(sqcoeff, 1); |
| return vgetq_lane_s64(error, 0) + vgetq_lane_s64(error, 1); |
| #endif |
| } |