mirror of https://github.com/PentHertz/srsLTE.git
212 lines
9.0 KiB
C
212 lines
9.0 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_MAT_H
|
|
#define SRSRAN_MAT_H
|
|
|
|
#include "srsran/config.h"
|
|
#include "srsran/phy/utils/simd.h"
|
|
#include <inttypes.h>
|
|
|
|
/* Generic implementation for complex reciprocal */
|
|
SRSRAN_API cf_t srsran_mat_cf_recip_gen(cf_t a);
|
|
|
|
/* Generic implementation for 2x2 determinant */
|
|
SRSRAN_API cf_t srsran_mat_2x2_det_gen(cf_t a00, cf_t a01, cf_t a10, cf_t a11);
|
|
|
|
/* Generic implementation for 2x2 Matrix Inversion */
|
|
SRSRAN_API void
|
|
srsran_mat_2x2_inv_gen(cf_t a00, cf_t a01, cf_t a10, cf_t a11, cf_t* r00, cf_t* r01, cf_t* r10, cf_t* r11);
|
|
|
|
/* Generic implementation for Zero Forcing (ZF) solver */
|
|
SRSRAN_API void
|
|
srsran_mat_2x2_zf_gen(cf_t y0, cf_t y1, cf_t h00, cf_t h01, cf_t h10, cf_t h11, cf_t* x0, cf_t* x1, float norm);
|
|
|
|
/* Generic implementation for Minimum Mean Squared Error (MMSE) solver */
|
|
SRSRAN_API void srsran_mat_2x2_mmse_gen(cf_t y0,
|
|
cf_t y1,
|
|
cf_t h00,
|
|
cf_t h01,
|
|
cf_t h10,
|
|
cf_t h11,
|
|
cf_t* x0,
|
|
cf_t* x1,
|
|
float noise_estimate,
|
|
float norm);
|
|
|
|
SRSRAN_API void srsran_mat_2x2_mmse_csi_gen(cf_t y0,
|
|
cf_t y1,
|
|
cf_t h00,
|
|
cf_t h01,
|
|
cf_t h10,
|
|
cf_t h11,
|
|
cf_t* x0,
|
|
cf_t* x1,
|
|
float* csi0,
|
|
float* csi1,
|
|
float noise_estimate,
|
|
float norm);
|
|
|
|
SRSRAN_API int srsran_mat_2x2_cn(cf_t h00, cf_t h01, cf_t h10, cf_t h11, float* cn);
|
|
|
|
#ifdef LV_HAVE_SSE
|
|
|
|
/* SSE implementation for complex reciprocal */
|
|
SRSRAN_API __m128 srsran_mat_cf_recip_sse(__m128 a);
|
|
|
|
/* SSE implementation for 2x2 determinant */
|
|
SRSRAN_API __m128 srsran_mat_2x2_det_sse(__m128 a00, __m128 a01, __m128 a10, __m128 a11);
|
|
|
|
#endif /* LV_HAVE_SSE */
|
|
|
|
#ifdef LV_HAVE_AVX
|
|
|
|
/* AVX implementation for complex reciprocal */
|
|
SRSRAN_API __m256 srsran_mat_cf_recip_avx(__m256 a);
|
|
|
|
/* AVX implementation for 2x2 determinant */
|
|
SRSRAN_API __m256 srsran_mat_2x2_det_avx(__m256 a00, __m256 a01, __m256 a10, __m256 a11);
|
|
|
|
#endif /* LV_HAVE_AVX */
|
|
|
|
#if SRSRAN_SIMD_CF_SIZE != 0
|
|
|
|
/* Generic SIMD implementation for 2x2 determinant */
|
|
static inline simd_cf_t srsran_mat_2x2_det_simd(simd_cf_t a00, simd_cf_t a01, simd_cf_t a10, simd_cf_t a11)
|
|
{
|
|
return srsran_simd_cf_sub(srsran_simd_cf_prod(a00, a11), srsran_simd_cf_prod(a01, a10));
|
|
}
|
|
|
|
/* Generic SIMD implementation for Zero Forcing (ZF) solver */
|
|
static inline void srsran_mat_2x2_zf_csi_simd(simd_cf_t y0,
|
|
simd_cf_t y1,
|
|
simd_cf_t h00,
|
|
simd_cf_t h01,
|
|
simd_cf_t h10,
|
|
simd_cf_t h11,
|
|
simd_cf_t* x0,
|
|
simd_cf_t* x1,
|
|
simd_f_t* csi0,
|
|
simd_f_t* csi1,
|
|
float norm)
|
|
{
|
|
simd_cf_t det = srsran_mat_2x2_det_simd(h00, h01, h10, h11);
|
|
simd_cf_t detrec = srsran_simd_cf_mul(srsran_simd_cf_rcp(det), srsran_simd_f_set1(norm));
|
|
|
|
*x0 = srsran_simd_cf_prod(srsran_simd_cf_sub(srsran_simd_cf_prod(h11, y0), srsran_simd_cf_prod(h01, y1)), detrec);
|
|
*x1 = srsran_simd_cf_prod(srsran_simd_cf_sub(srsran_simd_cf_prod(h00, y1), srsran_simd_cf_prod(h10, y0)), detrec);
|
|
|
|
*csi0 = srsran_simd_f_set1(1.0f);
|
|
*csi1 = srsran_simd_f_set1(1.0f);
|
|
}
|
|
|
|
static inline void srsran_mat_2x2_zf_simd(simd_cf_t y0,
|
|
simd_cf_t y1,
|
|
simd_cf_t h00,
|
|
simd_cf_t h01,
|
|
simd_cf_t h10,
|
|
simd_cf_t h11,
|
|
simd_cf_t* x0,
|
|
simd_cf_t* x1,
|
|
float norm)
|
|
{
|
|
simd_f_t csi1, csi2;
|
|
srsran_mat_2x2_zf_csi_simd(y0, y1, h00, h01, h10, h11, x0, x1, &csi1, &csi2, norm);
|
|
}
|
|
|
|
/* Generic SIMD implementation for Minimum Mean Squared Error (MMSE) solver */
|
|
static inline void srsran_mat_2x2_mmse_csi_simd(simd_cf_t y0,
|
|
simd_cf_t y1,
|
|
simd_cf_t h00,
|
|
simd_cf_t h01,
|
|
simd_cf_t h10,
|
|
simd_cf_t h11,
|
|
simd_cf_t* x0,
|
|
simd_cf_t* x1,
|
|
simd_f_t* csi0,
|
|
simd_f_t* csi1,
|
|
float noise_estimate,
|
|
float norm)
|
|
{
|
|
simd_cf_t _noise_estimate;
|
|
simd_f_t _norm = srsran_simd_f_set1(norm);
|
|
|
|
#if HAVE_NEON
|
|
_noise_estimate.val[0] = srsran_simd_f_set1(noise_estimate);
|
|
_noise_estimate.val[1] = srsran_simd_f_zero();
|
|
#else /* HAVE_NEON */
|
|
_noise_estimate.re = srsran_simd_f_set1(noise_estimate);
|
|
_noise_estimate.im = srsran_simd_f_zero();
|
|
#endif /* HAVE_NEON */
|
|
|
|
/* 1. A = H' x H + No*/
|
|
simd_cf_t a00 = srsran_simd_cf_add(
|
|
srsran_simd_cf_add(srsran_simd_cf_conjprod(h00, h00), srsran_simd_cf_conjprod(h10, h10)), _noise_estimate);
|
|
simd_cf_t a01 = srsran_simd_cf_add(srsran_simd_cf_conjprod(h01, h00), srsran_simd_cf_conjprod(h11, h10));
|
|
simd_cf_t a10 = srsran_simd_cf_add(srsran_simd_cf_conjprod(h00, h01), srsran_simd_cf_conjprod(h10, h11));
|
|
simd_cf_t a11 = srsran_simd_cf_add(
|
|
srsran_simd_cf_add(srsran_simd_cf_conjprod(h01, h01), srsran_simd_cf_conjprod(h11, h11)), _noise_estimate);
|
|
simd_cf_t a_det_rcp = srsran_simd_cf_rcp(srsran_mat_2x2_det_simd(a00, a01, a10, a11));
|
|
|
|
/* 2. B = inv(H' x H + No) = inv(A) */
|
|
simd_cf_t _norm2 = srsran_simd_cf_mul(a_det_rcp, _norm);
|
|
simd_cf_t b00 = srsran_simd_cf_prod(a11, _norm2);
|
|
simd_cf_t b01 = srsran_simd_cf_prod(srsran_simd_cf_neg(a01), _norm2);
|
|
simd_cf_t b10 = srsran_simd_cf_prod(srsran_simd_cf_neg(a10), _norm2);
|
|
simd_cf_t b11 = srsran_simd_cf_prod(a00, _norm2);
|
|
|
|
/* 3. W = inv(H' x H + No) x H' = B x H' */
|
|
simd_cf_t w00 = srsran_simd_cf_add(srsran_simd_cf_conjprod(b00, h00), srsran_simd_cf_conjprod(b01, h01));
|
|
simd_cf_t w01 = srsran_simd_cf_add(srsran_simd_cf_conjprod(b00, h10), srsran_simd_cf_conjprod(b01, h11));
|
|
simd_cf_t w10 = srsran_simd_cf_add(srsran_simd_cf_conjprod(b10, h00), srsran_simd_cf_conjprod(b11, h01));
|
|
simd_cf_t w11 = srsran_simd_cf_add(srsran_simd_cf_conjprod(b10, h10), srsran_simd_cf_conjprod(b11, h11));
|
|
|
|
/* 4. X = W x Y */
|
|
*x0 = srsran_simd_cf_add(srsran_simd_cf_prod(y0, w00), srsran_simd_cf_prod(y1, w01));
|
|
*x1 = srsran_simd_cf_add(srsran_simd_cf_prod(y0, w10), srsran_simd_cf_prod(y1, w11));
|
|
|
|
/* 5. Extract CSI */
|
|
*csi0 = srsran_simd_f_rcp(srsran_simd_cf_re(b00));
|
|
*csi1 = srsran_simd_f_rcp(srsran_simd_cf_re(b11));
|
|
}
|
|
|
|
static inline void srsran_mat_2x2_mmse_simd(simd_cf_t y0,
|
|
simd_cf_t y1,
|
|
simd_cf_t h00,
|
|
simd_cf_t h01,
|
|
simd_cf_t h10,
|
|
simd_cf_t h11,
|
|
simd_cf_t* x0,
|
|
simd_cf_t* x1,
|
|
float noise_estimate,
|
|
float norm)
|
|
{
|
|
simd_f_t csi0, csi1;
|
|
srsran_mat_2x2_mmse_csi_simd(y0, y1, h00, h01, h10, h11, x0, x1, &csi0, &csi1, noise_estimate, norm);
|
|
}
|
|
|
|
#endif /* SRSRAN_SIMD_CF_SIZE != 0 */
|
|
|
|
typedef struct {
|
|
uint32_t N;
|
|
cf_t* row_buffer;
|
|
cf_t* matrix;
|
|
} srsran_matrix_NxN_inv_t;
|
|
|
|
SRSRAN_API int srsran_matrix_NxN_inv_init(srsran_matrix_NxN_inv_t* q, uint32_t N);
|
|
|
|
SRSRAN_API void srsran_matrix_NxN_inv_run(srsran_matrix_NxN_inv_t* q, cf_t* in, cf_t* out);
|
|
|
|
SRSRAN_API void srsran_matrix_NxN_inv_free(srsran_matrix_NxN_inv_t* q);
|
|
|
|
#endif /* SRSRAN_MAT_H */
|