mirror of https://github.com/PentHertz/srsLTE.git
142 lines
5.7 KiB
C
142 lines
5.7 KiB
C
/**
|
|
*
|
|
* \section COPYRIGHT
|
|
*
|
|
* Copyright 2013-2021 Software Radio Systems Limited
|
|
*
|
|
* By using this file, you agree to the terms and conditions set
|
|
* forth in the LICENSE file which can be found at the top level of
|
|
* the distribution.
|
|
*
|
|
*/
|
|
|
|
#ifndef SRSRAN_VECTOR_SIMD_H
|
|
#define SRSRAN_VECTOR_SIMD_H
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#include "srsran/config.h"
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
|
|
/*SIMD Logical operations*/
|
|
SRSRAN_API void srsran_vec_xor_bbb_simd(const uint8_t* x, const uint8_t* y, uint8_t* z, int len);
|
|
|
|
/* SIMD Basic vector math */
|
|
SRSRAN_API void srsran_vec_sum_sss_simd(const int16_t* x, const int16_t* y, int16_t* z, int len);
|
|
|
|
SRSRAN_API void srsran_vec_sub_sss_simd(const int16_t* x, const int16_t* y, int16_t* z, int len);
|
|
|
|
SRSRAN_API void srsran_vec_sub_bbb_simd(const int8_t* x, const int8_t* y, int8_t* z, int len);
|
|
|
|
SRSRAN_API float srsran_vec_acc_ff_simd(const float* x, int len);
|
|
|
|
SRSRAN_API cf_t srsran_vec_acc_cc_simd(const cf_t* x, int len);
|
|
|
|
SRSRAN_API void srsran_vec_add_fff_simd(const float* x, const float* y, float* z, int len);
|
|
|
|
SRSRAN_API void srsran_vec_sub_fff_simd(const float* x, const float* y, float* z, int len);
|
|
|
|
/* SIMD Vector Scalar Product */
|
|
SRSRAN_API void srsran_vec_sc_prod_cfc_simd(const cf_t* x, const float h, cf_t* y, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_sc_prod_fcc_simd(const float* x, const cf_t h, cf_t* y, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_sc_prod_fff_simd(const float* x, const float h, float* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_sc_prod_ccc_simd(const cf_t* x, const cf_t h, cf_t* z, const int len);
|
|
|
|
SRSRAN_API int srsran_vec_sc_prod_ccc_simd2(const cf_t* x, const cf_t h, cf_t* z, const int len);
|
|
|
|
/* SIMD Vector Product */
|
|
SRSRAN_API void srsran_vec_prod_ccc_split_simd(const float* a_re,
|
|
const float* a_im,
|
|
const float* b_re,
|
|
const float* b_im,
|
|
float* r_re,
|
|
float* r_im,
|
|
const int len);
|
|
|
|
SRSRAN_API void srsran_vec_prod_ccc_c16_simd(const int16_t* a_re,
|
|
const int16_t* a_im,
|
|
const int16_t* b_re,
|
|
const int16_t* b_im,
|
|
int16_t* r_re,
|
|
int16_t* r_im,
|
|
const int len);
|
|
|
|
SRSRAN_API void srsran_vec_prod_sss_simd(const int16_t* x, const int16_t* y, int16_t* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_neg_sss_simd(const int16_t* x, const int16_t* y, int16_t* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_neg_bbb_simd(const int8_t* x, const int8_t* y, int8_t* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_prod_cfc_simd(const cf_t* x, const float* y, cf_t* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_prod_fff_simd(const float* x, const float* y, float* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_prod_ccc_simd(const cf_t* x, const cf_t* y, cf_t* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_prod_conj_ccc_simd(const cf_t* x, const cf_t* y, cf_t* z, const int len);
|
|
|
|
/* SIMD Division */
|
|
SRSRAN_API void srsran_vec_div_ccc_simd(const cf_t* x, const cf_t* y, cf_t* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_div_cfc_simd(const cf_t* x, const float* y, cf_t* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_div_fff_simd(const float* x, const float* y, float* z, const int len);
|
|
|
|
/* SIMD Dot product */
|
|
SRSRAN_API cf_t srsran_vec_dot_prod_conj_ccc_simd(const cf_t* x, const cf_t* y, const int len);
|
|
|
|
SRSRAN_API cf_t srsran_vec_dot_prod_ccc_simd(const cf_t* x, const cf_t* y, const int len);
|
|
|
|
#ifdef ENABLE_C16
|
|
SRSRAN_API c16_t srsran_vec_dot_prod_ccc_c16i_simd(const c16_t* x, const c16_t* y, const int len);
|
|
#endif /* ENABLE_C16 */
|
|
|
|
SRSRAN_API int srsran_vec_dot_prod_sss_simd(const int16_t* x, const int16_t* y, const int len);
|
|
|
|
/* SIMD Modulus functions */
|
|
SRSRAN_API void srsran_vec_abs_cf_simd(const cf_t* x, float* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_abs_square_cf_simd(const cf_t* x, float* z, const int len);
|
|
|
|
/* Other Functions */
|
|
SRSRAN_API void srsran_vec_lut_sss_simd(const short* x, const unsigned short* lut, short* y, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_lut_bbb_simd(const int8_t* x, const unsigned short* lut, int8_t* y, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_convert_if_simd(const int16_t* x, float* z, const float scale, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_convert_fi_simd(const float* x, int16_t* z, const float scale, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_convert_conj_cs_simd(const cf_t* x, int16_t* z, const float scale, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_convert_fb_simd(const float* x, int8_t* z, const float scale, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_interleave_simd(const cf_t* x, const cf_t* y, cf_t* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_interleave_add_simd(const cf_t* x, const cf_t* y, cf_t* z, const int len);
|
|
|
|
SRSRAN_API void srsran_vec_gen_sine_simd(cf_t amplitude, float freq, cf_t* z, int len);
|
|
|
|
SRSRAN_API void srsran_vec_apply_cfo_simd(const cf_t* x, float cfo, cf_t* z, int len);
|
|
|
|
SRSRAN_API float srsran_vec_estimate_frequency_simd(const cf_t* x, int len);
|
|
|
|
/* SIMD Find Max functions */
|
|
SRSRAN_API uint32_t srsran_vec_max_fi_simd(const float* x, const int len);
|
|
|
|
SRSRAN_API uint32_t srsran_vec_max_abs_fi_simd(const float* x, const int len);
|
|
|
|
SRSRAN_API uint32_t srsran_vec_max_ci_simd(const cf_t* x, const int len);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // SRSRAN_VECTOR_SIMD_H
|