Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
48b4b4931f2ee504158a06410adb10a765ea983f
[simgrid.git] / examples / python / actor-migrate / actor-migrate.py
1 # Copyright (c) 2017-2019. The SimGrid Team. All rights reserved.
2 #
3 # This program is free software; you can redistribute it and/or modify it
4 # under the terms of the license (GNU LGPL) which comes with this package.
5
6 # This example demonstrate the actor migrations.
7 #
8 # The worker actor first move by itself, and then start an execution.
9 # During that execution, the monitor migrates the worker, that wakes up on another host.
10 # The execution was of the right amount of flops to take exactly 5 seconds on the first host
11 # and 5 other seconds on the second one, so it stops after 10 seconds.
12 #
13 # Then another migration is done by the monitor while the worker is suspended.
14 #
15 # Note that worker() takes an uncommon set of parameters,
16 # and that this is perfectly accepted by create().
17
18 from simgrid import *
19 import sys
20
21 def worker(first_host, second_host):
22     flop_amount = first_host.speed * 5 + second_host.speed * 5
23
24     this_actor.info("Let's move to {:s} to execute {:.2f} Mflops (5sec on {:s} and 5sec on {:s})".format(first_host.name, flop_amount / 1e6, first_host.name, second_host.name))
25
26     this_actor.migrate(first_host)
27     this_actor.execute(flop_amount)
28
29     this_actor.info("I wake up on {:s}. Let's suspend a bit".format(this_actor.get_host().name))
30
31     this_actor.suspend()
32
33     this_actor.info("I wake up on {:s}".format(this_actor.get_host().name))
34     this_actor.info("Done")
35
36 def monitor():
37   boivin    = Host.by_name("Boivin")
38   jacquelin = Host.by_name("Jacquelin")
39   fafard    = Host.by_name("Fafard")
40
41   actor = Actor.create("worker", fafard, worker, boivin, jacquelin)
42
43   this_actor.sleep_for(5)
44
45   this_actor.info("After 5 seconds, move the process to {:s}".format(jacquelin.name))
46   actor.migrate(jacquelin)
47
48   this_actor.sleep_until(15)
49   this_actor.info("At t=15, move the process to {:s} and resume it.".format(fafard.name))
50   actor.migrate(fafard)
51   actor.resume()
52
53 if __name__ == '__main__':
54     e = Engine(sys.argv)
55     if len(sys.argv) < 2: raise AssertionError("Usage: actor-migration.py platform_file [other parameters]")
56     e.load_platform(sys.argv[1])
57
58     Actor.create("monitor", Host.by_name("Boivin"), monitor)
59     e.run()
60