Fix rho computation

openair1/PHY/NR_TRANSPORT/nr_ulsch_demodulation.c

@@ -1100,8 +1100,10 @@ void nr_ulsch_detection_mrc(NR_DL_FRAME_PARMS *frame_parms,
 
   if (n_rx > 1) {
     #if defined(__x86_64__) || defined(__i386__)
-    for (int aatx=0; aatx<nrOfLayers; aatx++) {
-      int nb_re = nb_rb*12;
+
+    int nb_re = nb_rb * 12;
+
+    for (int aatx = 0; aatx < nrOfLayers; aatx++) {
 
       rxdataF_comp128[0]   = (__m128i *)&rxdataF_comp[aatx*frame_parms->nb_antennas_rx][(symbol*(nb_re + off))];
       ul_ch_mag128[0]      = (__m128i *)&ul_ch_mag[aatx*frame_parms->nb_antennas_rx][(symbol*(nb_re + off))];
@@ -1123,6 +1125,20 @@ void nr_ulsch_detection_mrc(NR_DL_FRAME_PARMS *frame_parms,
             //rxdataF_comp128[0][i] = _mm_add_epi16(rxdataF_comp128_0[i],(*(__m128i *)&jitterc[0]));
         }
       }
+
+      if (rho) {
+        __m128i *rho128[2];
+        for (int aatx2 = 0; aatx2 < nrOfLayers; aatx2++) {
+          rho128[0] = (__m128i *) &rho[0][aatx * nrOfLayers + aatx2][(symbol * (nb_re + off))];
+          for (int aa = 1; aa < n_rx; aa++) {
+            rho128[1] = (__m128i *) &rho[aa][aatx * nrOfLayers + aatx2][(symbol * (nb_re + off))];
+            for (i = 0; i < nb_rb_0 * 3; i++) {
+              rho128[0][i] = _mm_adds_epi16(rho128[0][i], rho128[1][i]);
+            }
+          }
+        }
+      }
+
     }
     #elif defined(__arm__) || defined(__aarch64__)
     rxdataF_comp128_0   = (int16x8_t *)&rxdataF_comp[0][symbol*frame_parms->N_RB_DL*12];