Home | History | Annotate | Download | only in neon
      1 /*
      2  *  Copyright (c) 2015 The WebM project authors. All Rights Reserved.
      3  *
      4  *  Use of this source code is governed by a BSD-style license
      5  *  that can be found in the LICENSE file in the root of the source
      6  *  tree. An additional intellectual property rights grant can be found
      7  *  in the file PATENTS.  All contributing project authors may
      8  *  be found in the AUTHORS file in the root of the source tree.
      9  */
     10 
     11 #include <arm_neon.h>
     12 #include <assert.h>
     13 
     14 #include "./vp9_rtcd.h"
     15 
     16 int64_t vp9_block_error_fp_neon(const int16_t *coeff, const int16_t *dqcoeff,
     17                                 int block_size) {
     18   int64x2_t error = vdupq_n_s64(0);
     19 
     20   assert(block_size >= 8);
     21   assert((block_size % 8) == 0);
     22 
     23   do {
     24     const int16x8_t c = vld1q_s16(coeff);
     25     const int16x8_t d = vld1q_s16(dqcoeff);
     26     const int16x8_t diff = vsubq_s16(c, d);
     27     const int16x4_t diff_lo = vget_low_s16(diff);
     28     const int16x4_t diff_hi = vget_high_s16(diff);
     29     // diff is 15-bits, the squares 30, so we can store 2 in 31-bits before
     30     // accumulating them in 64-bits.
     31     const int32x4_t err0 = vmull_s16(diff_lo, diff_lo);
     32     const int32x4_t err1 = vmlal_s16(err0, diff_hi, diff_hi);
     33     const int64x2_t err2 = vaddl_s32(vget_low_s32(err1), vget_high_s32(err1));
     34     error = vaddq_s64(error, err2);
     35     coeff += 8;
     36     dqcoeff += 8;
     37     block_size -= 8;
     38   } while (block_size != 0);
     39 
     40   return vgetq_lane_s64(error, 0) + vgetq_lane_s64(error, 1);
     41 }
     42