1 /* Copyright (c) 2007-2022. The SimGrid Team. All rights reserved. */
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. */
6 #include "simgrid/s4u.hpp"
7 #include "simgrid/plugins/live_migration.h"
8 #include "simgrid/s4u/VirtualMachine.hpp"
10 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_test, "Messages specific for this s4u example");
12 static void worker(double computation_amount, bool use_bound, double bound)
14 double clock_start = simgrid::s4u::Engine::get_clock();
16 simgrid::s4u::ExecPtr exec = simgrid::s4u::this_actor::exec_init(computation_amount);
19 if (bound < 1e-12) /* close enough to 0 without any floating precision surprise */
20 XBT_INFO("bound == 0 means no capping (i.e., unlimited).");
21 exec->set_bound(bound);
25 double clock_end = simgrid::s4u::Engine::get_clock();
26 double duration = clock_end - clock_start;
27 double flops_per_sec = computation_amount / duration;
30 XBT_INFO("bound to %f => duration %f (%f flops/s)", bound, duration, flops_per_sec);
32 XBT_INFO("not bound => duration %f (%f flops/s)", duration, flops_per_sec);
35 static void worker_busy_loop(const char* name, double speed)
37 double exec_remain_prev = 1e11;
38 simgrid::s4u::ExecPtr exec = simgrid::s4u::this_actor::exec_async(exec_remain_prev);
39 for (int i = 0; i < 10; i++) {
41 double new_bound = (speed / 10) * i;
42 XBT_INFO("set bound of VM1 to %f", new_bound);
43 static_cast<simgrid::s4u::VirtualMachine*>(simgrid::s4u::this_actor::get_host())->set_bound(new_bound);
45 simgrid::s4u::this_actor::sleep_for(100);
46 double exec_remain_now = exec->get_remaining();
47 double flops_per_sec = exec_remain_prev - exec_remain_now;
48 XBT_INFO("%s@%s: %.0f flops/s", name, simgrid::s4u::this_actor::get_host()->get_cname(), flops_per_sec / 100);
49 exec_remain_prev = exec_remain_now;
50 simgrid::s4u::this_actor::sleep_for(1);
55 static void test_dynamic_change()
57 simgrid::s4u::Host* pm0 = simgrid::s4u::Host::by_name("Fafard");
59 auto* vm0 = pm0->create_vm("VM0", 1);
60 auto* vm1 = pm0->create_vm("VM1", 1);
64 simgrid::s4u::Actor::create("worker0", vm0, worker_busy_loop, "Task0", -1);
65 simgrid::s4u::Actor::create("worker1", vm1, worker_busy_loop, "Task1", pm0->get_speed());
67 simgrid::s4u::this_actor::sleep_for(3000); // let the activities end
72 static void test_one_activity(simgrid::s4u::Host* host)
74 const double cpu_speed = host->get_speed();
75 const double computation_amount = cpu_speed * 10;
77 XBT_INFO("### Test: with/without activity set_bound");
79 XBT_INFO("### Test: no bound for Task1@%s", host->get_cname());
80 simgrid::s4u::Actor::create("worker0", host, worker, computation_amount, false, 0);
82 simgrid::s4u::this_actor::sleep_for(1000);
84 XBT_INFO("### Test: 50%% for Task1@%s", host->get_cname());
85 simgrid::s4u::Actor::create("worker0", host, worker, computation_amount, true, cpu_speed / 2);
87 simgrid::s4u::this_actor::sleep_for(1000);
89 XBT_INFO("### Test: 33%% for Task1@%s", host->get_cname());
90 simgrid::s4u::Actor::create("worker0", host, worker, computation_amount, true, cpu_speed / 3);
92 simgrid::s4u::this_actor::sleep_for(1000);
94 XBT_INFO("### Test: zero for Task1@%s (i.e., unlimited)", host->get_cname());
95 simgrid::s4u::Actor::create("worker0", host, worker, computation_amount, true, 0);
97 simgrid::s4u::this_actor::sleep_for(1000);
99 XBT_INFO("### Test: 200%% for Task1@%s (i.e., meaningless)", host->get_cname());
100 simgrid::s4u::Actor::create("worker0", host, worker, computation_amount, true, cpu_speed * 2);
102 simgrid::s4u::this_actor::sleep_for(1000);
105 static void test_two_activities(simgrid::s4u::Host* hostA, simgrid::s4u::Host* hostB)
107 const double cpu_speed = hostA->get_speed();
108 xbt_assert(cpu_speed == hostB->get_speed());
109 const double computation_amount = cpu_speed * 10;
110 const char* hostA_name = hostA->get_cname();
111 const char* hostB_name = hostB->get_cname();
113 XBT_INFO("### Test: no bound for Task1@%s, no bound for Task2@%s", hostA_name, hostB_name);
114 simgrid::s4u::Actor::create("worker0", hostA, worker, computation_amount, false, 0);
115 simgrid::s4u::Actor::create("worker1", hostB, worker, computation_amount, false, 0);
117 simgrid::s4u::this_actor::sleep_for(1000);
119 XBT_INFO("### Test: 0 for Task1@%s, 0 for Task2@%s (i.e., unlimited)", hostA_name, hostB_name);
120 simgrid::s4u::Actor::create("worker0", hostA, worker, computation_amount, true, 0);
121 simgrid::s4u::Actor::create("worker1", hostB, worker, computation_amount, true, 0);
123 simgrid::s4u::this_actor::sleep_for(1000);
125 XBT_INFO("### Test: 50%% for Task1@%s, 50%% for Task2@%s", hostA_name, hostB_name);
126 simgrid::s4u::Actor::create("worker0", hostA, worker, computation_amount, true, cpu_speed / 2);
127 simgrid::s4u::Actor::create("worker1", hostB, worker, computation_amount, true, cpu_speed / 2);
129 simgrid::s4u::this_actor::sleep_for(1000);
131 XBT_INFO("### Test: 25%% for Task1@%s, 25%% for Task2@%s", hostA_name, hostB_name);
132 simgrid::s4u::Actor::create("worker0", hostA, worker, computation_amount, true, cpu_speed / 4);
133 simgrid::s4u::Actor::create("worker1", hostB, worker, computation_amount, true, cpu_speed / 4);
135 simgrid::s4u::this_actor::sleep_for(1000);
137 XBT_INFO("### Test: 75%% for Task1@%s, 100%% for Task2@%s", hostA_name, hostB_name);
138 simgrid::s4u::Actor::create("worker0", hostA, worker, computation_amount, true, cpu_speed * 0.75);
139 simgrid::s4u::Actor::create("worker1", hostB, worker, computation_amount, true, cpu_speed);
141 simgrid::s4u::this_actor::sleep_for(1000);
143 XBT_INFO("### Test: no bound for Task1@%s, 25%% for Task2@%s", hostA_name, hostB_name);
144 simgrid::s4u::Actor::create("worker0", hostA, worker, computation_amount, false, 0);
145 simgrid::s4u::Actor::create("worker1", hostB, worker, computation_amount, true, cpu_speed / 4);
147 simgrid::s4u::this_actor::sleep_for(1000);
149 XBT_INFO("### Test: 75%% for Task1@%s, 25%% for Task2@%s", hostA_name, hostB_name);
150 simgrid::s4u::Actor::create("worker0", hostA, worker, computation_amount, true, cpu_speed * 0.75);
151 simgrid::s4u::Actor::create("worker1", hostB, worker, computation_amount, true, cpu_speed / 4);
153 simgrid::s4u::this_actor::sleep_for(1000);
156 static void master_main()
158 simgrid::s4u::Host* pm0 = simgrid::s4u::Host::by_name("Fafard");
160 XBT_INFO("# 1. Put a single activity on a PM.");
161 test_one_activity(pm0);
164 XBT_INFO("# 2. Put two activities on a PM.");
165 test_two_activities(pm0, pm0);
168 auto* vm0 = pm0->create_vm("VM0", 1);
171 XBT_INFO("# 3. Put a single activity on a VM.");
172 test_one_activity(vm0);
175 XBT_INFO("# 4. Put two activities on a VM.");
176 test_two_activities(vm0, vm0);
181 vm0 = pm0->create_vm("VM0", 1);
184 XBT_INFO("# 6. Put an activity on a PM and an activity on a VM.");
185 test_two_activities(pm0, vm0);
190 vm0 = pm0->create_vm("VM0", 1);
191 vm0->set_bound(pm0->get_speed() / 10);
194 XBT_INFO("# 7. Put a single activity on the VM capped by 10%%.");
195 test_one_activity(vm0);
198 XBT_INFO("# 8. Put two activities on the VM capped by 10%%.");
199 test_two_activities(vm0, vm0);
202 XBT_INFO("# 9. Put an activity on a PM and an activity on the VM capped by 10%%.");
203 test_two_activities(pm0, vm0);
208 vm0 = pm0->create_vm("VM0", 1);
209 vm0->set_ramsize(1e9); // 1GB
212 double cpu_speed = pm0->get_speed();
214 XBT_INFO("# 10. Test migration");
215 const double computation_amount = cpu_speed * 10;
217 XBT_INFO("# 10. (a) Put an activity on a VM without any bound.");
218 simgrid::s4u::Actor::create("worker0", vm0, worker, computation_amount, false, 0);
219 simgrid::s4u::this_actor::sleep_for(1000);
222 XBT_INFO("# 10. (b) set 10%% bound to the VM, and then put an activity on the VM.");
223 vm0->set_bound(cpu_speed / 10);
224 simgrid::s4u::Actor::create("worker0", vm0, worker, computation_amount, false, 0);
225 simgrid::s4u::this_actor::sleep_for(1000);
228 XBT_INFO("# 10. (c) migrate");
229 simgrid::s4u::Host* pm1 = simgrid::s4u::Host::by_name("Fafard");
230 sg_vm_migrate(vm0, pm1);
233 XBT_INFO("# 10. (d) Put an activity again on the VM.");
234 simgrid::s4u::Actor::create("worker0", vm0, worker, computation_amount, false, 0);
235 simgrid::s4u::this_actor::sleep_for(1000);
240 XBT_INFO("# 11. Change a bound dynamically.");
241 test_dynamic_change();
244 int main(int argc, char* argv[])
246 simgrid::s4u::Engine e(&argc, argv);
247 sg_vm_live_migration_plugin_init();
248 /* load the platform file */
249 xbt_assert(argc == 2, "Usage: %s platform_file\n\tExample: %s ../platforms/small_platform.xml\n", argv[0], argv[0]);
251 e.load_platform(argv[1]);
253 simgrid::s4u::Actor::create("master_", e.host_by_name("Fafard"), master_main);
256 XBT_INFO("Bye (simulation time %g)", simgrid::s4u::Engine::get_clock());