examples/python/task-switch-host/task-switch-host.py

   1 # Copyright (c) 2006-2023. 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 """
   7 /* This example demonstrates how to dynamically modify a graph of tasks.
   8  *
   9  * Assuming we have two instances of a service placed on different hosts,
  10  * we want to send data alternatively to thoses instances.
  11  *
  12  * We consider the following graph:
  13
  14            comm1
  15      ┌────────────────────────┐
  16      │                        │
  17      │               Fafard   │
  18      │              ┌───────┐ │
  19      │      ┌──────►│ exec1 ├─┘
  20      ▼      │       └───────┘
  21  Tremblay ──┤comm0
  22      ▲      │        Jupiter
  23      │      │       ┌───────┐
  24      │      └──────►│ exec2 ├─┐
  25      │              └───────┘ │
  26      │                        │
  27      └────────────────────────┘
  28            comm2
  29  */
  30  """
  31
  32 from argparse import ArgumentParser
  33 import sys
  34 from simgrid import Engine, Task, CommTask, ExecTask
  35
  36 def parse():
  37     parser = ArgumentParser()
  38     parser.add_argument(
  39         '--platform',
  40         type=str,
  41         required=True,
  42         help='path to the platform description'
  43     )
  44     return parser.parse_args()
  45
  46 def callback(t):
  47     print(f'[{Engine.clock}] {t} finished ({t.count})')
  48
  49 def switch(t, hosts, execs):
  50     comm0.destination = hosts[t.count % 2]
  51     comm0.remove_successor(execs[t.count % 2 - 1])
  52     comm0.add_successor(execs[t.count % 2])
  53
  54 if __name__ == '__main__':
  55     args = parse()
  56     e = Engine(sys.argv)
  57     e.load_platform(args.platform)
  58     Task.init()
  59
  60     # Retrieve hosts
  61     tremblay = e.host_by_name('Tremblay')
  62     jupiter = e.host_by_name('Jupiter')
  63     fafard = e.host_by_name('Fafard')
  64
  65     # Create tasks
  66     comm0 = CommTask.init("comm0")
  67     comm0.bytes = 1e7
  68     comm0.source = tremblay
  69     exec1 = ExecTask.init("exec1", 1e9, jupiter)
  70     exec2 = ExecTask.init("exec2", 1e9, fafard)
  71     comm1 = CommTask.init("comm1", 1e7, jupiter, tremblay)
  72     comm2 = CommTask.init("comm2", 1e7, fafard, tremblay)
  73
  74     # Create the initial graph by defining dependencies between tasks
  75     exec1.add_successor(comm1)
  76     exec2.add_successor(comm2)
  77
  78     # Add a function to be called when tasks end for log purpose
  79     Task.on_end_cb(callback)
  80
  81     # Add a function to be called before each executions of comm0
  82     # This function modifies the graph of tasks by adding or removing
  83     # successors to comm0
  84     comm0.on_this_start_cb(lambda t: switch(t, [jupiter, fafard], [exec1,exec2]))
  85
  86     # Enqueue two executions for task exec1
  87     comm0.enqueue_execs(4)
  88
  89     # runs the simulation
  90     e.run()