Don't use Vecs in modmul, use fixed-size buffers

This commit is contained in:
2018-06-09 17:32:15 -07:00
parent fee68cca18
commit 26eb05ceeb

View File

@@ -85,16 +85,12 @@ macro_rules! generate_multipliers
impl ModMul for $name { impl ModMul for $name {
fn modmul(&mut self, x: &$name, m: &$name) { fn modmul(&mut self, x: &$name, m: &$name) {
let mut mulres = Vec::with_capacity(2 * self.values.len()); let mut mulres = [0; $size/32];
mulres.resize(2 * self.values.len(), 0);
raw_multiplication(&self.values, &x.values, &mut mulres); raw_multiplication(&self.values, &x.values, &mut mulres);
let mut widerm = Vec::with_capacity(mulres.len()); let mut widerm = [0; $size/32];
widerm.extend_from_slice(&m.values); for (idx,val) in m.values.iter().enumerate() { widerm[idx] = *val; }
widerm.resize(mulres.len(), 0); let mut dead = [0; $size/32];
let mut dead = Vec::with_capacity(widerm.len()); let mut answer = [0; $size/32];
dead.resize(widerm.len(), 0);
let mut answer = Vec::with_capacity(widerm.len());
answer.resize(widerm.len(), 0);
divmod(&mulres, &widerm, &mut dead, &mut answer); divmod(&mulres, &widerm, &mut dead, &mut answer);
for i in 0..answer.len() { for i in 0..answer.len() {
if i < self.values.len() { if i < self.values.len() {