optimize recode branch performance

Author: sleepybishop

Makefile

@@ -37,6 +37,7 @@ examples/blockchain_gossip
 CPPFLAGS = -D_DEFAULT_SOURCE -D_FILE_OFFSET_BITS=64 
 CFLAGS   = -O3 -g -std=c11 -Wall -I. -Iinclude -Ideps/
 CFLAGS  += -march=native -funroll-loops -ftree-vectorize -fno-inline -fstack-protector-all -Wno-unused -Wno-sequence-point
+CFLAGS  += -DNANORQ_USE_UNIFIED_SOLVER
 
 all: test libnanorq.a libnanorq_core.a $(EXAMPLES)
 

lib/nanorq.c

@@ -958,9 +958,11 @@ bool nanorq_repair_block(nanorq *rq, struct ioctx *io, uint8_t sbn) {
     repair_sym rs = dec->repair_bin.a[rep_idx++];
 #ifdef NANORQ_USE_UNIFIED_SOLVER
     if (rs.coefs) {
-      nanorq_core_replace_symbol_explicit(&dec->core, dec->core.P.Kprime + extra, rs.coefs);
+      nanorq_core_replace_symbol_explicit(&dec->core,
+                                          dec->core.P.Kprime + extra, rs.coefs);
     } else {
-      nanorq_core_replace_symbol(&dec->core, dec->core.P.Kprime + extra, rs.esi);
+      nanorq_core_replace_symbol(&dec->core, dec->core.P.Kprime + extra,
+                                 rs.esi);
     }
 #else
     nanorq_core_replace_symbol(&dec->core, dec->core.P.Kprime + extra, rs.esi);

lib/nanorq_core.c

@@ -287,13 +287,8 @@ void nanorq_core_replace_symbol_explicit(nanorq_core *rq, uint32_t row,
     if (coefs[i] == 0)
       continue;
 
-    u32 X = i;
-    u32 cols[GENC_MAX];
-    u32_vec src_row;
-    src_row.a = cols;
-    src_row.n = 0;
-    src_row.m = GENC_MAX;
-    params_set_idxs(P, X, &src_row);
+    u32 src_idx = P->S + P->H + i;
+    u32_vec src_row = W->A[src_idx];
 
     for (u32 it = 0; it < uv_size(src_row); it++) {
       u32 col = uv_A(src_row, it);

lib/precode.c

@@ -274,7 +274,7 @@ static void precode_matrix_fill_HDPC(params *P, pc *W) {
           W->cb.on_op(W->cb.on_op_arg, phys_row, drow, beta);
           u8 *a = (u8 *)&uv_E(W->UL, uv_A(W->F.rowmap, phys_row), 0);
           u32 *b = &uv_E(W->U, drow, 0);
-          nanorq_oblas.axpyb32(a, b, beta, W->u);
+          nanorq_oblas.axpyb32(a, b, beta, W->UL.s);
         }
       }
     }