Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
e699015ed113a0a4c653f7aa712ab97fa12d0e20
[simgrid.git] / examples / s4u / energy-ptask / s4u-energy-ptask.cpp
1 /* Copyright (c) 2017. 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 #include "simgrid/plugins/energy.h"
7 #include <simgrid/s4u.hpp>
8 #include <xbt/ex.hpp>
9 #include <xbt/log.h>
10
11 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_energyptask, "Messages specific for this s4u example");
12
13 static void runner()
14 {
15   /* Retrieve the list of all hosts as an array of hosts */
16   int hosts_count            = sg_host_count();
17   simgrid::s4u::Host** hosts = sg_host_list();
18
19   XBT_INFO("First, build a classical parallel task, with 1 Gflop to execute on each node, "
20            "and 10MB to exchange between each pair");
21   double* computation_amounts   = new double[hosts_count]();
22   double* communication_amounts = new double[hosts_count * hosts_count]();
23
24   for (int i               = 0; i < hosts_count; i++)
25     computation_amounts[i] = 1e9; // 1 Gflop
26
27   for (int i = 0; i < hosts_count; i++)
28     for (int j = i + 1; j < hosts_count; j++)
29       communication_amounts[i * hosts_count + j] = 1e7; // 10 MB
30
31   simgrid::s4u::this_actor::parallel_execute(hosts_count, hosts, computation_amounts, communication_amounts);
32
33   XBT_INFO("We can do the same with a timeout of one second enabled.");
34   computation_amounts   = new double[hosts_count]();
35   communication_amounts = new double[hosts_count * hosts_count]();
36
37   for (int i               = 0; i < hosts_count; i++)
38     computation_amounts[i] = 1e9; // 1 Gflop
39
40   for (int i = 0; i < hosts_count; i++)
41     for (int j = i + 1; j < hosts_count; j++)
42       communication_amounts[i * hosts_count + j] = 1e7; // 10 MB
43
44   try {
45     simgrid::s4u::this_actor::parallel_execute(hosts_count, hosts, computation_amounts, communication_amounts,
46                                                1.0 /* timeout (in seconds)*/);
47     XBT_WARN("Woops, this did not timeout as expected... Please report that bug.");
48   } catch (xbt_ex& e) {
49     /* Do nothing this exception on timeout was expected */
50   }
51
52   XBT_INFO("Then, build a parallel task involving only computations and no communication (1 Gflop per node)");
53   computation_amounts = new double[hosts_count]();
54   for (int i               = 0; i < hosts_count; i++)
55     computation_amounts[i] = 1e9; // 1 Gflop
56   simgrid::s4u::this_actor::parallel_execute(hosts_count, hosts, computation_amounts, nullptr /* no comm */);
57
58   XBT_INFO("Then, build a parallel task with no computation nor communication (synchro only)");
59   computation_amounts   = new double[hosts_count]();
60   communication_amounts = new double[hosts_count * hosts_count]();
61   simgrid::s4u::this_actor::parallel_execute(hosts_count, hosts, computation_amounts, communication_amounts);
62
63   XBT_INFO("Finally, trick the ptask to do a 'remote execution', on host %s", hosts[1]->getCname());
64   computation_amounts = new double[1]{1e9};
65
66   simgrid::s4u::Host* remote[] = {hosts[1]};
67   simgrid::s4u::this_actor::parallel_execute(1, remote, computation_amounts, nullptr);
68
69   XBT_INFO("Goodbye now!");
70   std::free(hosts);
71 }
72
73 int main(int argc, char* argv[])
74 {
75   simgrid::s4u::Engine e(&argc, argv);
76
77   xbt_assert(argc <= 3, "1Usage: %s <platform file> [--energy]", argv[0]);
78   xbt_assert(argc >= 2, "2Usage: %s <platform file> [--energy]", argv[0]);
79
80   if (argc == 3 && argv[2][2] == 'e')
81     sg_host_energy_plugin_init();
82
83   e.loadPlatform(argv[1]);
84
85   /* Pick a process, no matter which, from the platform file */
86   simgrid::s4u::Actor::createActor("test", simgrid::s4u::Host::by_name("MyHost1"), runner);
87
88   e.run();
89   XBT_INFO("Simulation done.");
90   return 0;
91 }