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
|
# 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.
from argparse import ArgumentParser
import sys
from simgrid import Actor, Barrier, Engine, Host, this_actor
def create_parser() -> ArgumentParser:
parser = ArgumentParser()
parser.add_argument(
'--platform',
type=str,
required=True,
help='path to the platform description'
)
parser.add_argument(
"--actors",
type=int,
required=True,
help="Number of actors to start"
)
return parser
def worker(barrier: Barrier):
""" Wait on the barrier and exits.
:param barrier: Barrier to be awaited
"""
this_actor.info("Waiting on the barrier")
barrier.wait()
this_actor.info("Bye")
def master(actor_count: int):
""" Create barrier with `actor_count` expected actors, spawns `actor_count - 1` workers, then wait on the barrier
and exits.
:param actor_count: Spawn actor_count-1 workers and do a barrier with them
"""
barrier = Barrier(actor_count)
workers_count = actor_count - 1
this_actor.info(f"Spawning {workers_count} workers")
for i in range(workers_count):
Host.by_name("Jupiter").add_actor(f"worker-{i}", worker, barrier)
this_actor.info("Waiting on the barrier")
barrier.wait()
this_actor.info("Bye")
def main():
settings = create_parser().parse_known_args()[0]
if settings.actors < 1:
raise ValueError("--actors must be greater than 0")
e = Engine(sys.argv)
e.load_platform(settings.platform)
e.host_by_name("Tremblay").add_actor("master", master, settings.actors)
e.run()
if __name__ == "__main__":
main()
|
