split/
split.rs

1#![deny(warnings)]
2
3use mpi::topology::{Color, GroupRelation, SystemGroup};
4use mpi::traits::*;
5
6fn main() {
7    let universe = mpi::initialize().unwrap();
8    let world = universe.world();
9
10    let odd = (0..world.size()).filter(|x| x % 2 != 0).collect::<Vec<_>>();
11    let odd_group = world.group().include(&odd[..]);
12    let even_group = world.group().difference(&odd_group);
13    assert!(
14        (world.rank() % 2 == 0 && even_group.rank().is_some() && odd_group.rank().is_none())
15            || (even_group.rank().is_none() && odd_group.rank().is_some())
16    );
17    let my_group = if odd_group.rank().is_some() {
18        &odd_group
19    } else {
20        &even_group
21    };
22    let empty_group = SystemGroup::empty();
23
24    let oddness_comm = world.split_by_subgroup_collective(my_group);
25    assert!(oddness_comm.is_some());
26    let oddness_comm = oddness_comm.unwrap();
27    assert_eq!(
28        GroupRelation::Identical,
29        oddness_comm.group().compare(my_group)
30    );
31
32    let odd_comm = if odd_group.rank().is_some() {
33        world.split_by_subgroup_collective(&odd_group)
34    } else {
35        world.split_by_subgroup_collective(&empty_group)
36    };
37    if odd_group.rank().is_some() {
38        assert!(odd_comm.is_some());
39        let odd_comm = odd_comm.unwrap();
40        assert_eq!(
41            GroupRelation::Identical,
42            odd_comm.group().compare(&odd_group)
43        );
44    } else {
45        assert!(odd_comm.is_none());
46    }
47
48    #[cfg(not(msmpi))]
49    {
50        if even_group.rank().is_some() {
51            let even_comm = world.split_by_subgroup(&even_group);
52            assert!(even_comm.is_some());
53            let even_comm = even_comm.unwrap();
54            assert_eq!(
55                GroupRelation::Identical,
56                even_comm.group().compare(&even_group)
57            );
58
59            let no_comm = world.split_by_subgroup(&odd_group);
60            assert!(no_comm.is_none());
61        }
62    }
63
64    let oddness_comm = world.split_by_color(Color::with_value(world.rank() % 2));
65    assert!(oddness_comm.is_some());
66    let oddness_comm = oddness_comm.unwrap();
67    assert_eq!(
68        GroupRelation::Identical,
69        oddness_comm.group().compare(my_group)
70    );
71
72    let odd_comm = world.split_by_color(if world.rank() % 2 != 0 {
73        Color::with_value(0)
74    } else {
75        Color::undefined()
76    });
77    if world.rank() % 2 != 0 {
78        assert!(odd_comm.is_some());
79        let odd_comm = odd_comm.unwrap();
80        assert_eq!(
81            GroupRelation::Identical,
82            odd_comm.group().compare(&odd_group)
83        );
84    } else {
85        assert!(odd_comm.is_none());
86    }
87}
split/split.rs

split/
split.rs
