zec
/
mpc
mirror of https://github.com/zcash/mpc.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Multithread the last step of the FFT.

This commit is contained in:
Sean Bowe 2016-09-17 12:17:41 -06:00
parent 8a32e77949
commit 00a7c93ea0
No known key found for this signature in database
GPG Key ID: 95684257D8F8B031
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/protocol/qap.rs
View File

@ -94,11 +94,26 @@ fn fft<G: Group>(v: &[G], omega: Fr, threads: usize) -> Vec<G>
(rx_evens.recv().unwrap(), rx_odds.recv().unwrap())
};
let mut acc = omega;
let mut res = Vec::with_capacity(v.len());
for i in 0..v.len() {
res.push(evens[i%d2] + odds[i%d2] * acc);
acc = acc * omega;
if threads < 2 {
let mut acc = omega;
for i in 0..v.len() {
res.push(evens[i%d2] + odds[i%d2] * acc);
acc = acc * omega;
}
} else {
res.extend_from_slice(&evens);
res.extend_from_slice(&evens);
parallel(&mut res, |mut i, v| {
let mut acc = omega.pow(Fr::from_str(&format!("{}", i+1)).unwrap());
for a in v {
*a = *a + odds[i%d2] * acc;
acc = acc * omega;
i += 1;
}
}, threads);
}
res