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
|
# 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.
"""
# ##################################################################################
# Take this tutorial online: https://simgrid.org/doc/latest/Tutorial_Algorithms.html
# ##################################################################################
"""
import sys
from simgrid import Actor, Engine, Mailbox, this_actor
# master-begin
def master(*args):
if len(args) == 2:
raise AssertionError(f"Actor master requires 4 parameters, but only {len(args)}")
tasks_count = int(args[0])
compute_cost = int(args[1])
communicate_cost = int(args[2])
this_actor.info(f"Got {tasks_count} tasks to process")
hosts = Engine.instance.all_hosts
for h in hosts:
Actor.create(f'Worker-{h.name}', h, worker)
for i in range(tasks_count): # For each task to be executed:
# - Select a worker in a round-robin way
mailbox = Mailbox.by_name(f'Worker-{hosts[i%len(hosts)].name}')
# - Send the computation amount to the worker
if (tasks_count < 10000 or (tasks_count < 100000 and i % 10000 == 0) or i % 100000 == 0):
this_actor.info(f"Sending task {i} of {tasks_count} to mailbox '{mailbox.name}'")
mailbox.put(compute_cost, communicate_cost)
this_actor.info("All tasks have been dispatched. Request all workers to stop.")
for h in hosts:
# The workers stop when receiving a negative compute_cost
mailbox = Mailbox.by_name(f'Worker-{h.name}')
mailbox.put(-1, 0)
# master-end
# worker-begin
def worker(*args):
assert not args, "The worker expects no argument"
mailbox = Mailbox.by_name(f'Worker-{this_actor.get_host().name}')
done = False
while not done:
compute_cost = mailbox.get()
if compute_cost > 0: # If compute_cost is valid, execute a computation of that cost
this_actor.execute(compute_cost)
else: # Stop when receiving an invalid compute_cost
done = True
this_actor.info("Exiting now.")
# worker-end
# main-begin
if __name__ == '__main__':
assert len(sys.argv) > 2, f"Usage: python app-masterworkers.py platform_file deployment_file"
e = Engine(sys.argv)
# Register the classes representing the actors
e.register_actor("master", master)
e.register_actor("worker", worker)
# Load the platform description and then deploy the application
e.load_platform(sys.argv[1])
e.load_deployment(sys.argv[2])
# Run the simulation
e.run()
this_actor.info("Simulation is over")
# main-end
|
