chromium/third_party/libaom/source/libaom/aom_dsp/x86/sum_squares_avx2.c

/*
 * Copyright (c) 2018, Alliance for Open Media. All rights reserved.
 *
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. 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 www.aomedia.org/license/patent.
 */

#include <immintrin.h>
#include <smmintrin.h>

#include "aom_dsp/x86/synonyms.h"
#include "aom_dsp/x86/synonyms_avx2.h"
#include "aom_dsp/x86/sum_squares_sse2.h"
#include "config/aom_dsp_rtcd.h"

static uint64_t aom_sum_squares_2d_i16_nxn_avx2(const int16_t *src, int stride,
                                                int width, int height) { … }

uint64_t aom_sum_squares_2d_i16_avx2(const int16_t *src, int stride, int width,
                                     int height) { … }

static uint64_t aom_sum_sse_2d_i16_nxn_avx2(const int16_t *src, int stride,
                                            int width, int height, int *sum) { … }

uint64_t aom_sum_sse_2d_i16_avx2(const int16_t *src, int src_stride, int width,
                                 int height, int *sum) { … }

// Accumulate sum of 16-bit elements in the vector
static inline int32_t mm256_accumulate_epi16(__m256i vec_a) { … }

// Accumulate sum of 32-bit elements in the vector
static inline int32_t mm256_accumulate_epi32(__m256i vec_a) { … }

uint64_t aom_var_2d_u8_avx2(uint8_t *src, int src_stride, int width,
                            int height) { … }

uint64_t aom_var_2d_u16_avx2(uint8_t *src, int src_stride, int width,
                             int height) { … }