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
|
# Copyright (c) 2010-2025. The SimGrid Team. All rights reserved.
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the license (GNU LGPL) which comes with this package.
import sys
from argparse import ArgumentParser
import numpy as np
import simgrid
from simgrid import Engine, this_actor
def create_parser() -> ArgumentParser:
parser = ArgumentParser()
parser.add_argument(
'--platform',
type=str,
required=True,
help='path to the platform description'
)
return parser
def pinger():
simgrid.smpi.MPI.Init()
MPI_COMM_WORLD = simgrid.smpi.MPI.Comm.WORLD
size = int(0)
size = simgrid.smpi.MPI.Comm.size(MPI_COMM_WORLD)
this_actor.info(f"ping {size}")
rank = 0
rank = simgrid.smpi.MPI.Comm.rank(MPI_COMM_WORLD)
this_actor.info(f"ping {rank}")
input_size = 10
half = input_size // 2
root = 0
np.random.seed(5)
np_in = np.random.randint(100, size=input_size, dtype=np.int32)
np_out = np.zeros(half, dtype=np.int32)
this_actor.info(f"input buffer {np_in}")
op = simgrid.smpi.MPI.Op.SUM
simgrid.smpi.MPI.Reduce_scatter(np_in, np_out, [half, half], simgrid.smpi.MPI.Datatype.INT, op, MPI_COMM_WORLD)
this_actor.info(f"result buffer {np_out}")
np_out = np.zeros(half, dtype=np.int32)
simgrid.smpi.MPI.Reduce_scatter_block(np_in, np_out, half, simgrid.smpi.MPI.Datatype.INT, op, MPI_COMM_WORLD)
this_actor.info(f"result buffer {np_out}")
simgrid.smpi.MPI.Finalize()
def main():
settings = create_parser().parse_known_args()[0]
e = Engine(sys.argv)
simgrid.smpi.init()
e.load_platform(settings.platform)
simgrid.smpi.app_instance_start("pinger", pinger, [e.host_by_name("Jupiter"), e.host_by_name("Tremblay")])
e.run()
this_actor.info(f"Total simulation time: {e.clock:.3f}")
if __name__ == "__main__":
main()
|
