lamellar 0.8.0

Lamellar is an asynchronous tasking runtime for HPC systems developed in RUST.
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use crate::array::iterator::{distributed_iterator::*, IterLockFuture};

/// `Monotonic` is an iterator that produces elements in a monotonically increasing order based on the index of the elements in the input iterator.
#[derive(Clone, Debug)]
pub struct Monotonic<I> {
    iter: I,
    cur_index: usize,
}

impl<I: InnerIter> InnerIter for Monotonic<I> {
    fn lock_if_needed(&self, _s: Sealed) -> Option<IterLockFuture> {
        self.iter.lock_if_needed(_s)
    }
    fn iter_clone(&self, _s: Sealed) -> Self {
        Monotonic {
            iter: self.iter.iter_clone(Sealed),
            cur_index: self.cur_index,
        }
    }
}

impl<I> Monotonic<I>
where
    I: DistributedIterator,
{
    pub(crate) fn new(iter: I, cur_index: usize) -> Monotonic<I> {
        // println!("new Monotonic {:?} ",count);
        Monotonic { iter, cur_index }
    }
}

impl<I> DistributedIterator for Monotonic<I>
where
    I: DistributedIterator,
{
    type Item = (usize, <I as DistributedIterator>::Item);
    type Array = <I as DistributedIterator>::Array;
    fn init(&self, start_i: usize, cnt: usize, _s: Sealed) -> Monotonic<I> {
        let val = Monotonic::new(self.iter.init(start_i, cnt, _s), start_i);
        // println!("{:?} Monotonic init {start_i} {cnt} {start_i}",std::thread::current().id());
        val
    }
    fn array(&self) -> Self::Array {
        self.iter.array()
    }
    fn next(&mut self) -> Option<Self::Item> {
        if let Some(a) = self.iter.next() {
            let i = self.cur_index;
            // println!("{:?} Monotonic next {:?} i: {:?}",std::thread::current().id(),self.cur_index,i);
            self.cur_index += 1;
            Some((i, a))
        } else {
            // println!("{:?} Monotonic done",std::thread::current().id());
            None
        }
    }

    fn elems(&self, in_elems: usize) -> usize {
        let in_elems = self.iter.elems(in_elems);
        in_elems
    }

    fn advance_index(&mut self, count: usize) {
        self.iter.advance_index(count);
        self.cur_index += 1;
    }
}