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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136

use lamellar::array::prelude::*;

macro_rules! into_test_identity {
    ($array1:ident) => {{
        let world = lamellar::LamellarWorldBuilder::new().build();
        let _num_pes = world.num_pes();
        let _my_pe = world.my_pe();

        let array = $array1::<u32>::new(world.clone(), 1000, Distribution::Block).block();
        let _array2: $array1<u32> = array;
    }};
}

macro_rules! into_test_conv {
    ($array1:ident, $array2:ident, $method:ident) => {{
        let world = lamellar::LamellarWorldBuilder::new().build();
        let _num_pes = world.num_pes();
        let _my_pe = world.my_pe();

        let array = $array1::<u32>::new(world.clone(), 1000, Distribution::Block).block();
        let _array2: $array2<u32> = array.$method().block();
    }};
}

macro_rules! into_test {
    (UnsafeArray, UnsafeArray) => {
        into_test_identity!(UnsafeArray)
    };
    (ReadOnlyArray, ReadOnlyArray) => {
        into_test_identity!(ReadOnlyArray)
    };
    (AtomicArray, AtomicArray) => {
        into_test_identity!(AtomicArray)
    };
    (LocalLockArray, LocalLockArray) => {
        into_test_identity!(LocalLockArray)
    };
    (GlobalLockArray, GlobalLockArray) => {
        into_test_identity!(GlobalLockArray)
    };

    (UnsafeArray, ReadOnlyArray) => {
        into_test_conv!(UnsafeArray, ReadOnlyArray, into_read_only)
    };
    (UnsafeArray, AtomicArray) => {
        into_test_conv!(UnsafeArray, AtomicArray, into_atomic)
    };
    (UnsafeArray, LocalLockArray) => {
        into_test_conv!(UnsafeArray, LocalLockArray, into_local_lock)
    };
    (UnsafeArray, GlobalLockArray) => {
        into_test_conv!(UnsafeArray, GlobalLockArray, into_global_lock)
    };

    (ReadOnlyArray, UnsafeArray) => {
        into_test_conv!(ReadOnlyArray, UnsafeArray, into_unsafe)
    };
    (ReadOnlyArray, AtomicArray) => {
        into_test_conv!(ReadOnlyArray, AtomicArray, into_atomic)
    };
    (ReadOnlyArray, LocalLockArray) => {
        into_test_conv!(ReadOnlyArray, LocalLockArray, into_local_lock)
    };
    (ReadOnlyArray, GlobalLockArray) => {
        into_test_conv!(ReadOnlyArray, GlobalLockArray, into_global_lock)
    };

    (AtomicArray, UnsafeArray) => {
        into_test_conv!(AtomicArray, UnsafeArray, into_unsafe)
    };
    (AtomicArray, ReadOnlyArray) => {
        into_test_conv!(AtomicArray, ReadOnlyArray, into_read_only)
    };
    (AtomicArray, LocalLockArray) => {
        into_test_conv!(AtomicArray, LocalLockArray, into_local_lock)
    };
    (AtomicArray, GlobalLockArray) => {
        into_test_conv!(AtomicArray, GlobalLockArray, into_global_lock)
    };

    (LocalLockArray, UnsafeArray) => {
        into_test_conv!(LocalLockArray, UnsafeArray, into_unsafe)
    };
    (LocalLockArray, ReadOnlyArray) => {
        into_test_conv!(LocalLockArray, ReadOnlyArray, into_read_only)
    };
    (LocalLockArray, AtomicArray) => {
        into_test_conv!(LocalLockArray, AtomicArray, into_atomic)
    };
    (LocalLockArray, GlobalLockArray) => {
        into_test_conv!(LocalLockArray, GlobalLockArray, into_global_lock)
    };

    (GlobalLockArray, UnsafeArray) => {
        into_test_conv!(GlobalLockArray, UnsafeArray, into_unsafe)
    };
    (GlobalLockArray, ReadOnlyArray) => {
        into_test_conv!(GlobalLockArray, ReadOnlyArray, into_read_only)
    };
    (GlobalLockArray, AtomicArray) => {
        into_test_conv!(GlobalLockArray, AtomicArray, into_atomic)
    };
    (GlobalLockArray, LocalLockArray) => {
        into_test_conv!(GlobalLockArray, LocalLockArray, into_local_lock)
    };
}

macro_rules! match_array2 {
    ($array_ty:ident, $array2_str:tt) => {
        match $array2_str.as_str() {
            "UnsafeArray" => into_test!($array_ty, UnsafeArray),
            // "LocalOnlyArray" => into_test!($array_ty, LocalOnlyArray),
            "ReadOnlyArray" => into_test!($array_ty, ReadOnlyArray),
            "AtomicArray" => into_test!($array_ty, AtomicArray),
            "LocalLockArray" => into_test!($array_ty, LocalLockArray),
            "GlobalLockArray" => into_test!($array_ty, GlobalLockArray),
            _ => panic!("Unknown array type: {}", $array2_str),
        }
    };
}

fn main() {
    let args: Vec<String> = std::env::args().collect();
    let array1 = args[1].clone();
    let array2 = args[2].clone();

    match array1.as_str() {
        "UnsafeArray" => match_array2!(UnsafeArray, array2),
        // "LocalOnlyArray" => match_array2!(LocalOnlyArray, array2),
        "ReadOnlyArray" => match_array2!(ReadOnlyArray, array2),
        "AtomicArray" => match_array2!(AtomicArray, array2),
        "LocalLockArray" => match_array2!(LocalLockArray, array2),
        "GlobalLockArray" => match_array2!(GlobalLockArray, array2),
        _ => panic!("Unknown array type: {}", array1),
    }
}