diff options
Diffstat (limited to 'third_party/aom/aom_dsp/arm/sum_neon.h')
-rw-r--r-- | third_party/aom/aom_dsp/arm/sum_neon.h | 311 |
1 files changed, 311 insertions, 0 deletions
diff --git a/third_party/aom/aom_dsp/arm/sum_neon.h b/third_party/aom/aom_dsp/arm/sum_neon.h new file mode 100644 index 0000000000..30a108e70a --- /dev/null +++ b/third_party/aom/aom_dsp/arm/sum_neon.h @@ -0,0 +1,311 @@ +/* + * 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. + */ + +#ifndef AOM_AOM_DSP_ARM_SUM_NEON_H_ +#define AOM_AOM_DSP_ARM_SUM_NEON_H_ + +#include "config/aom_dsp_rtcd.h" +#include "config/aom_config.h" + +#include "aom/aom_integer.h" +#include "aom_ports/mem.h" + +static INLINE int horizontal_add_u8x8(const uint8x8_t a) { +#if AOM_ARCH_AARCH64 + return vaddlv_u8(a); +#else + uint16x4_t b = vpaddl_u8(a); + uint32x2_t c = vpaddl_u16(b); + return vget_lane_u32(c, 0) + vget_lane_u32(c, 1); +#endif +} + +static INLINE int horizontal_add_s16x8(const int16x8_t a) { +#if AOM_ARCH_AARCH64 + return vaddlvq_s16(a); +#else + const int32x4_t b = vpaddlq_s16(a); + const int64x2_t c = vpaddlq_s32(b); + const int32x2_t d = vadd_s32(vreinterpret_s32_s64(vget_low_s64(c)), + vreinterpret_s32_s64(vget_high_s64(c))); + return vget_lane_s32(d, 0); +#endif +} + +static INLINE int horizontal_add_s32x4(const int32x4_t a) { +#if AOM_ARCH_AARCH64 + return vaddvq_s32(a); +#else + const int64x2_t b = vpaddlq_s32(a); + const int32x2_t c = vadd_s32(vreinterpret_s32_s64(vget_low_s64(b)), + vreinterpret_s32_s64(vget_high_s64(b))); + return vget_lane_s32(c, 0); +#endif +} + +static INLINE int64_t horizontal_add_s64x2(const int64x2_t a) { +#if AOM_ARCH_AARCH64 + return vaddvq_s64(a); +#else + return vgetq_lane_s64(a, 0) + vgetq_lane_s64(a, 1); +#endif +} + +static INLINE uint64_t horizontal_add_u64x2(const uint64x2_t a) { +#if AOM_ARCH_AARCH64 + return vaddvq_u64(a); +#else + return vgetq_lane_u64(a, 0) + vgetq_lane_u64(a, 1); +#endif +} + +static INLINE uint64_t horizontal_long_add_u32x4(const uint32x4_t a) { +#if AOM_ARCH_AARCH64 + return vaddlvq_u32(a); +#else + const uint64x2_t b = vpaddlq_u32(a); + return vgetq_lane_u64(b, 0) + vgetq_lane_u64(b, 1); +#endif +} + +static INLINE int64_t horizontal_long_add_s32x4(const int32x4_t a) { +#if AOM_ARCH_AARCH64 + return vaddlvq_s32(a); +#else + const int64x2_t b = vpaddlq_s32(a); + return vgetq_lane_s64(b, 0) + vgetq_lane_s64(b, 1); +#endif +} + +static INLINE uint32_t horizontal_add_u32x4(const uint32x4_t a) { +#if AOM_ARCH_AARCH64 + return vaddvq_u32(a); +#else + const uint64x2_t b = vpaddlq_u32(a); + const uint32x2_t c = vadd_u32(vreinterpret_u32_u64(vget_low_u64(b)), + vreinterpret_u32_u64(vget_high_u64(b))); + return vget_lane_u32(c, 0); +#endif +} + +static INLINE uint32x4_t horizontal_add_4d_u32x4(const uint32x4_t sum[4]) { +#if AOM_ARCH_AARCH64 + uint32x4_t res01 = vpaddq_u32(sum[0], sum[1]); + uint32x4_t res23 = vpaddq_u32(sum[2], sum[3]); + return vpaddq_u32(res01, res23); +#else + uint32x4_t res = vdupq_n_u32(0); + res = vsetq_lane_u32(horizontal_add_u32x4(sum[0]), res, 0); + res = vsetq_lane_u32(horizontal_add_u32x4(sum[1]), res, 1); + res = vsetq_lane_u32(horizontal_add_u32x4(sum[2]), res, 2); + res = vsetq_lane_u32(horizontal_add_u32x4(sum[3]), res, 3); + return res; +#endif +} + +static INLINE int32x4_t horizontal_add_4d_s32x4(const int32x4_t sum[4]) { +#if AOM_ARCH_AARCH64 + int32x4_t res01 = vpaddq_s32(sum[0], sum[1]); + int32x4_t res23 = vpaddq_s32(sum[2], sum[3]); + return vpaddq_s32(res01, res23); +#else + int32x4_t res = vdupq_n_s32(0); + res = vsetq_lane_s32(horizontal_add_s32x4(sum[0]), res, 0); + res = vsetq_lane_s32(horizontal_add_s32x4(sum[1]), res, 1); + res = vsetq_lane_s32(horizontal_add_s32x4(sum[2]), res, 2); + res = vsetq_lane_s32(horizontal_add_s32x4(sum[3]), res, 3); + return res; +#endif +} + +static INLINE uint32_t horizontal_long_add_u16x8(const uint16x8_t vec_lo, + const uint16x8_t vec_hi) { +#if AOM_ARCH_AARCH64 + return vaddlvq_u16(vec_lo) + vaddlvq_u16(vec_hi); +#else + const uint32x4_t vec_l_lo = + vaddl_u16(vget_low_u16(vec_lo), vget_high_u16(vec_lo)); + const uint32x4_t vec_l_hi = + vaddl_u16(vget_low_u16(vec_hi), vget_high_u16(vec_hi)); + const uint32x4_t a = vaddq_u32(vec_l_lo, vec_l_hi); + const uint64x2_t b = vpaddlq_u32(a); + const uint32x2_t c = vadd_u32(vreinterpret_u32_u64(vget_low_u64(b)), + vreinterpret_u32_u64(vget_high_u64(b))); + return vget_lane_u32(c, 0); +#endif +} + +static INLINE uint32x4_t horizontal_long_add_4d_u16x8( + const uint16x8_t sum_lo[4], const uint16x8_t sum_hi[4]) { + const uint32x4_t a0 = vpaddlq_u16(sum_lo[0]); + const uint32x4_t a1 = vpaddlq_u16(sum_lo[1]); + const uint32x4_t a2 = vpaddlq_u16(sum_lo[2]); + const uint32x4_t a3 = vpaddlq_u16(sum_lo[3]); + const uint32x4_t b0 = vpadalq_u16(a0, sum_hi[0]); + const uint32x4_t b1 = vpadalq_u16(a1, sum_hi[1]); + const uint32x4_t b2 = vpadalq_u16(a2, sum_hi[2]); + const uint32x4_t b3 = vpadalq_u16(a3, sum_hi[3]); +#if AOM_ARCH_AARCH64 + const uint32x4_t c0 = vpaddq_u32(b0, b1); + const uint32x4_t c1 = vpaddq_u32(b2, b3); + return vpaddq_u32(c0, c1); +#else + const uint32x2_t c0 = vadd_u32(vget_low_u32(b0), vget_high_u32(b0)); + const uint32x2_t c1 = vadd_u32(vget_low_u32(b1), vget_high_u32(b1)); + const uint32x2_t c2 = vadd_u32(vget_low_u32(b2), vget_high_u32(b2)); + const uint32x2_t c3 = vadd_u32(vget_low_u32(b3), vget_high_u32(b3)); + const uint32x2_t d0 = vpadd_u32(c0, c1); + const uint32x2_t d1 = vpadd_u32(c2, c3); + return vcombine_u32(d0, d1); +#endif +} + +static INLINE uint32_t horizontal_add_u16x8(const uint16x8_t a) { +#if AOM_ARCH_AARCH64 + return vaddlvq_u16(a); +#else + const uint32x4_t b = vpaddlq_u16(a); + const uint64x2_t c = vpaddlq_u32(b); + const uint32x2_t d = vadd_u32(vreinterpret_u32_u64(vget_low_u64(c)), + vreinterpret_u32_u64(vget_high_u64(c))); + return vget_lane_u32(d, 0); +#endif +} + +static INLINE uint32x4_t horizontal_add_4d_u16x8(const uint16x8_t sum[4]) { +#if AOM_ARCH_AARCH64 + const uint16x8_t a0 = vpaddq_u16(sum[0], sum[1]); + const uint16x8_t a1 = vpaddq_u16(sum[2], sum[3]); + const uint16x8_t b0 = vpaddq_u16(a0, a1); + return vpaddlq_u16(b0); +#else + const uint16x4_t a0 = vadd_u16(vget_low_u16(sum[0]), vget_high_u16(sum[0])); + const uint16x4_t a1 = vadd_u16(vget_low_u16(sum[1]), vget_high_u16(sum[1])); + const uint16x4_t a2 = vadd_u16(vget_low_u16(sum[2]), vget_high_u16(sum[2])); + const uint16x4_t a3 = vadd_u16(vget_low_u16(sum[3]), vget_high_u16(sum[3])); + const uint16x4_t b0 = vpadd_u16(a0, a1); + const uint16x4_t b1 = vpadd_u16(a2, a3); + return vpaddlq_u16(vcombine_u16(b0, b1)); +#endif +} + +static INLINE int32x4_t horizontal_add_4d_s16x8(const int16x8_t sum[4]) { +#if AOM_ARCH_AARCH64 + const int16x8_t a0 = vpaddq_s16(sum[0], sum[1]); + const int16x8_t a1 = vpaddq_s16(sum[2], sum[3]); + const int16x8_t b0 = vpaddq_s16(a0, a1); + return vpaddlq_s16(b0); +#else + const int16x4_t a0 = vadd_s16(vget_low_s16(sum[0]), vget_high_s16(sum[0])); + const int16x4_t a1 = vadd_s16(vget_low_s16(sum[1]), vget_high_s16(sum[1])); + const int16x4_t a2 = vadd_s16(vget_low_s16(sum[2]), vget_high_s16(sum[2])); + const int16x4_t a3 = vadd_s16(vget_low_s16(sum[3]), vget_high_s16(sum[3])); + const int16x4_t b0 = vpadd_s16(a0, a1); + const int16x4_t b1 = vpadd_s16(a2, a3); + return vpaddlq_s16(vcombine_s16(b0, b1)); +#endif +} + +static INLINE uint32_t horizontal_add_u32x2(const uint32x2_t a) { +#if AOM_ARCH_AARCH64 + return vaddv_u32(a); +#else + const uint64x1_t b = vpaddl_u32(a); + return vget_lane_u32(vreinterpret_u32_u64(b), 0); +#endif +} + +static INLINE uint64_t horizontal_long_add_u32x2(const uint32x2_t a) { +#if AOM_ARCH_AARCH64 + return vaddlv_u32(a); +#else + const uint64x1_t b = vpaddl_u32(a); + return vget_lane_u64(b, 0); +#endif +} + +static INLINE uint32_t horizontal_add_u16x4(const uint16x4_t a) { +#if AOM_ARCH_AARCH64 + return vaddlv_u16(a); +#else + const uint32x2_t b = vpaddl_u16(a); + const uint64x1_t c = vpaddl_u32(b); + return vget_lane_u32(vreinterpret_u32_u64(c), 0); +#endif +} + +static INLINE int32x4_t horizontal_add_2d_s32(int32x4_t a, int32x4_t b) { +#if AOM_ARCH_AARCH64 + return vpaddq_s32(a, b); +#else + const int32x2_t a0 = vpadd_s32(vget_low_s32(a), vget_high_s32(a)); + const int32x2_t b0 = vpadd_s32(vget_low_s32(b), vget_high_s32(b)); + return vcombine_s32(a0, b0); +#endif +} + +static INLINE int32x2_t add_pairwise_s32x4(int32x4_t a) { +#if AOM_ARCH_AARCH64 + return vget_low_s32(vpaddq_s32(a, a)); +#else + return vpadd_s32(vget_low_s32(a), vget_high_s32(a)); +#endif +} + +static INLINE uint64_t horizontal_long_add_u32x4_x2(const uint32x4_t a[2]) { + return horizontal_long_add_u32x4(a[0]) + horizontal_long_add_u32x4(a[1]); +} + +static INLINE uint64_t horizontal_long_add_u32x4_x4(const uint32x4_t a[4]) { + uint64x2_t sum = vpaddlq_u32(a[0]); + sum = vpadalq_u32(sum, a[1]); + sum = vpadalq_u32(sum, a[2]); + sum = vpadalq_u32(sum, a[3]); + + return horizontal_add_u64x2(sum); +} + +static INLINE uint64_t horizontal_long_add_u32x4_x8(const uint32x4_t a[8]) { + uint64x2_t sum[2]; + sum[0] = vpaddlq_u32(a[0]); + sum[1] = vpaddlq_u32(a[1]); + sum[0] = vpadalq_u32(sum[0], a[2]); + sum[1] = vpadalq_u32(sum[1], a[3]); + sum[0] = vpadalq_u32(sum[0], a[4]); + sum[1] = vpadalq_u32(sum[1], a[5]); + sum[0] = vpadalq_u32(sum[0], a[6]); + sum[1] = vpadalq_u32(sum[1], a[7]); + + return horizontal_add_u64x2(vaddq_u64(sum[0], sum[1])); +} + +static INLINE uint64_t horizontal_long_add_u32x4_x16(const uint32x4_t a[16]) { + uint64x2_t sum[2]; + sum[0] = vpaddlq_u32(a[0]); + sum[1] = vpaddlq_u32(a[1]); + sum[0] = vpadalq_u32(sum[0], a[2]); + sum[1] = vpadalq_u32(sum[1], a[3]); + sum[0] = vpadalq_u32(sum[0], a[4]); + sum[1] = vpadalq_u32(sum[1], a[5]); + sum[0] = vpadalq_u32(sum[0], a[6]); + sum[1] = vpadalq_u32(sum[1], a[7]); + sum[0] = vpadalq_u32(sum[0], a[8]); + sum[1] = vpadalq_u32(sum[1], a[9]); + sum[0] = vpadalq_u32(sum[0], a[10]); + sum[1] = vpadalq_u32(sum[1], a[11]); + sum[0] = vpadalq_u32(sum[0], a[12]); + sum[1] = vpadalq_u32(sum[1], a[13]); + sum[0] = vpadalq_u32(sum[0], a[14]); + sum[1] = vpadalq_u32(sum[1], a[15]); + + return horizontal_add_u64x2(vaddq_u64(sum[0], sum[1])); +} + +#endif // AOM_AOM_DSP_ARM_SUM_NEON_H_ |