stash

starky/src/fft_p.rs

@@ -3,6 +3,7 @@ use crate::constant::{get_max_workers, MAX_OPS_PER_THREAD, MIN_OPS_PER_THREAD, S
 use crate::fft_worker::{fft_block, interpolate_prepare_block};
 use crate::helper::log2_any;
 use crate::traits::FieldExtension;
+use crate::utils::parallells::parallelize;
 use core::cmp::min;
 use lazy_static::lazy_static;
 use rayon::prelude::*;
@@ -95,11 +96,13 @@ pub fn bit_reverse<F: FieldExtension>(
     let n = 1 << nbits;
     let ris = BRs(0, n, nbits); // move it outside the loop. obtain it from cache.
 
-    // todo remove the double loop, and parallel
-    for i in 0..n {
-        for k in 0..n_pols {
-            buffdst[i * n_pols + k] = buffsrc[ris[i] * n_pols + k];
-        }
+    // todo parallel. Does the buffdst
+    let len = n * n_pols;
+    assert_eq!(len, buffdst.len()); // is ok?
+    for j in 0..len {
+        let i = j / n_pols;
+        let k = j % n_pols;
+        buffdst[j] = buffsrc[ris[i] * n_pols + k];
     }
 }