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
|
# 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 Engine, Actor, Comm, Mailbox, 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 sender(mailbox: Mailbox):
this_actor.info("Send at full bandwidth")
# First send a 2.5e8 Bytes payload at full bandwidth (1.25e8 Bps)
payload = Engine.clock
mailbox.put(payload, int(2.5e8))
this_actor.info("Throttle the bandwidth at the Comm level")
# ... then send it again but throttle the Comm
payload = Engine.clock
# get a handler on the comm first
comm: Comm = mailbox.put_init(payload, int(2.5e8))
# let throttle the communication. It amounts to set the rate of the comm to half the nominal bandwidth of the link,
# i.e., 1.25e8 / 2. This second communication will thus take approximately twice as long as the first one
comm.set_rate(int(1.25e8 / 2)).wait()
def receiver(mailbox: Mailbox):
# Receive the first payload sent at full bandwidth
sender_time = mailbox.get()
communication_time = Engine.clock - sender_time
this_actor.info(f"Payload received (full bandwidth) in {communication_time} seconds")
# ... Then receive the second payload sent with a throttled Comm
sender_time = mailbox.get()
communication_time = Engine.clock - sender_time
this_actor.info(f"Payload received (throttled) in {communication_time} seconds")
def main():
settings = create_parser().parse_known_args()[0]
e = Engine(sys.argv)
e.load_platform(settings.platform)
mailbox = e.mailbox_by_name_or_create("Mailbox")
e.host_by_name("node-0.simgrid.org").add_actor("sender", sender, mailbox)
e.host_by_name("node-1.simgrid.org").add_actor("receiver", receiver, mailbox)
e.run()
if __name__ == "__main__":
main()
|
