raphtory-api 0.17.0

Raphtory common interface and APIs
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use rayon::prelude::*;
/// Compute cumulative sum in parallel over `num_chunks` chunks
pub fn par_cum_sum(values: &mut [usize]) {
    let num_chunks = rayon::current_num_threads();
    let chunk_size = values.len().div_ceil(num_chunks);
    let mut chunk_sums = Vec::with_capacity(num_chunks);
    values
        .par_chunks_mut(chunk_size)
        .map(|chunk| {
            let mut cum_sum = 0;
            for v in chunk {
                *v += cum_sum;
                cum_sum = *v;
            }
            cum_sum
        })
        .collect_into_vec(&mut chunk_sums);

    let mut cum_sum = 0;
    for (partial_sum, next_chunk) in chunk_sums
        .into_iter()
        .zip(values.chunks_mut(chunk_size).skip(1))
    {
        cum_sum += partial_sum;
        next_chunk.par_iter_mut().for_each(|v| *v += cum_sum);
    }
}

#[cfg(test)]
mod test {
    use super::par_cum_sum;

    #[test]
    fn test_cum_sum() {
        let mut values: Vec<_> = (0..100).collect();
        par_cum_sum(&mut values);
        let mut cum_sum = 0;
        for (index, v) in values.into_iter().enumerate() {
            cum_sum += index;
            assert_eq!(v, cum_sum)
        }
    }
}