Exact Chunks Parallel Iterator #1192

Activity

1. jturner314 commented on Aug 1, 2022

   

   jturner314

   on Aug 1, 2022

   Last edited by jturner314

   Member

   More actions

   We should add a ParallelIterator implementation for Parallel<ExactChunks/Mut>.

   In the meantime, a workaround is to Zip's parallel methods:

   use ndarray::prelude::*;
   use ndarray::Zip;
   
   fn main() {
       let mut a = Array2::<u8>::zeros([6, 6]);
       Zip::from(a.exact_chunks_mut([3, 3])).par_for_each(|mut chunk| chunk.diag_mut().fill(1));
       assert_eq!(
           a,
           array![
               [1, 0, 0, 1, 0, 0],
               [0, 1, 0, 0, 1, 0],
               [0, 0, 1, 0, 0, 1],
               [1, 0, 0, 1, 0, 0],
               [0, 1, 0, 0, 1, 0],
               [0, 0, 1, 0, 0, 1],
           ],
       );
   }

   The par_azip! macro may be a bit more convenient.

   Regardless, make sure to enable to rayon feature.

   👍React with 👍1Reacted by Benjamin Wilson
2. jturner314

   added

   enhancement

   help wanted

   on Aug 1, 2022