sad/raptorq
1
0
Fork 0
mirror of https://github.com/cberner/raptorq.git synced 2024-06-29 18:21:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tracking of Symbol add ops when decoding

Browse Source
This commit is contained in:
Christopher Berner 2019-02-03 19:35:20 -08:00
parent 4cb675c618
commit f0f68e64a1
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/base.rs
View File

@ -101,7 +101,8 @@ pub struct IntermediateSymbolDecoder {
u: usize,
L: usize,
num_source_symbols: u32,
debug_symbol_mul_ops: u32
debug_symbol_mul_ops: u32,
debug_symbol_add_ops: u32
}
impl IntermediateSymbolDecoder {
@ -126,7 +127,8 @@ impl IntermediateSymbolDecoder {
u: num_pi_symbols(num_source_symbols) as usize,
L: num_intermediate_symbols(num_source_symbols) as usize,
num_source_symbols,
debug_symbol_mul_ops: 0
debug_symbol_mul_ops: 0,
debug_symbol_add_ops: 0
}
}
@ -444,11 +446,13 @@ impl IntermediateSymbolDecoder {
for col in 0..row {
if self.X[row][col] == Octet::one() {
self.debug_symbol_add_ops += 1;
let temp = self.D[self.d[col]].clone();
self.D[self.d[row]] += &temp;
}
else {
self.debug_symbol_mul_ops += 1;
self.debug_symbol_add_ops += 1;
let temp = self.D[self.d[col]].clone();
self.D[self.d[row]].fused_addassign_mul_scalar(&temp, &self.X[row][col]);
}
@ -621,6 +625,10 @@ impl IntermediateSymbolDecoder {
self.debug_symbol_mul_ops
}
pub fn get_symbol_add_ops(&self) -> u32 {
self.debug_symbol_add_ops
}
// Helper operations to apply operations to A, also to D
fn mul_row(&mut self, i: usize, beta: Octet) {
self.debug_symbol_mul_ops += 1;
@ -632,10 +640,12 @@ impl IntermediateSymbolDecoder {
fn fma_rows(&mut self, i: usize, iprime: usize, beta: Octet) {
if beta == Octet::one() {
self.debug_symbol_add_ops += 1;
let temp = self.D[self.d[i]].clone();
self.D[self.d[iprime]] += &temp;
}
else {
self.debug_symbol_add_ops += 1;
self.debug_symbol_mul_ops += 1;
let temp = self.D[self.d[i]].clone();
self.D[self.d[iprime]].fused_addassign_mul_scalar(&temp, &beta);

2
src/bin_matrix_sparsity.rs
View File

@ -50,6 +50,6 @@ fn main() {
let symbols = vec![Symbol::zero(1); a.width()];
let mut decoder = IntermediateSymbolDecoder::new(&a, &symbols, num_symbols);
decoder.execute();
println!("Optimized decoder mul ops: {}", decoder.get_symbol_mul_ops());
println!("Optimized decoder mul ops: {}, add ops: {}", decoder.get_symbol_mul_ops(), decoder.get_symbol_add_ops());
}
}