Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Add new entry in Release_Notes.
[simgrid.git] / src / instr / instr_platform.cpp
1 /* Copyright (c) 2010-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 #include <simgrid/kernel/routing/NetPoint.hpp>
7 #include <simgrid/kernel/routing/NetZoneImpl.hpp>
8 #include <simgrid/s4u/Actor.hpp>
9 #include <simgrid/s4u/Comm.hpp>
10 #include <simgrid/s4u/Engine.hpp>
11 #include <simgrid/s4u/Exec.hpp>
12 #include <simgrid/s4u/Host.hpp>
13 #include <simgrid/s4u/VirtualMachine.hpp>
14 #include <xbt/graph.h>
15
16 #include "src/instr/instr_private.hpp"
17 #include "src/kernel/activity/ExecImpl.hpp"
18 #include "src/kernel/resource/CpuImpl.hpp"
19 #include "src/kernel/resource/NetworkModel.hpp"
20
21 #include <fstream>
22
23 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(instr_routing, instr, "Tracing platform hierarchy");
24
25 std::string instr_pid(simgrid::s4u::Actor const& proc)
26 {
27   return proc.get_name() + "-" + std::to_string(proc.get_pid());
28 }
29
30 static simgrid::instr::Container* lowestCommonAncestor(const simgrid::instr::Container* a1,
31                                                        const simgrid::instr::Container* a2)
32 {
33   // this is only an optimization (since most of a1 and a2 share the same parent)
34   if (a1->get_parent() == a2->get_parent())
35     return a1->get_parent();
36
37   // create an array with all ancestors of a1
38   std::vector<simgrid::instr::Container*> ancestors_a1;
39   for (auto* p = a1->get_parent(); p != nullptr; p = p->get_parent())
40     ancestors_a1.push_back(p);
41
42   // create an array with all ancestors of a2
43   std::vector<simgrid::instr::Container*> ancestors_a2;
44   for (auto* p = a2->get_parent(); p != nullptr; p = p->get_parent())
45     ancestors_a2.push_back(p);
46
47   // find the lowest ancestor
48   simgrid::instr::Container* p = nullptr;
49   int i = static_cast<int>(ancestors_a1.size()) - 1;
50   int j = static_cast<int>(ancestors_a2.size()) - 1;
51   while (i >= 0 && j >= 0) {
52     simgrid::instr::Container* a1p       = ancestors_a1.at(i);
53     if (a1p != ancestors_a2.at(j))
54       break;
55     p = a1p;
56     i--;
57     j--;
58   }
59   return p;
60 }
61
62 static void linkContainers(simgrid::instr::Container* src, simgrid::instr::Container* dst,
63                            std::set<std::string, std::less<>>* filter)
64 {
65   // ignore loopback
66   if (src->get_name() == "__loopback__" || dst->get_name() == "__loopback__") {
67     XBT_DEBUG("  linkContainers: ignoring loopback link");
68     return;
69   }
70
71   // find common parent
72   simgrid::instr::Container* parent = lowestCommonAncestor(src, dst);
73   xbt_assert(parent, "common parent unknown, this is a tracing problem");
74
75   // check if we already register this pair (we only need one direction)
76   std::string aux1 = src->get_name() + dst->get_name();
77   std::string aux2 = dst->get_name() + src->get_name();
78   if (filter->find(aux1) != filter->end()) {
79     XBT_DEBUG("  linkContainers: already registered %s <-> %s (1)", src->get_cname(), dst->get_cname());
80     return;
81   }
82   if (filter->find(aux2) != filter->end()) {
83     XBT_DEBUG("  linkContainers: already registered %s <-> %s (2)", dst->get_cname(), src->get_cname());
84     return;
85   }
86
87   // ok, not found, register it
88   filter->insert(aux1);
89   filter->insert(aux2);
90
91   // declare type
92   std::string link_typename = parent->get_type()->get_name() + "-" + src->get_type()->get_name() +
93                               std::to_string(src->get_type()->get_id()) + "-" + dst->get_type()->get_name() +
94                               std::to_string(dst->get_type()->get_id());
95   simgrid::instr::LinkType* link =
96       parent->get_type()->by_name_or_create(link_typename, src->get_type(), dst->get_type());
97   link->set_calling_container(parent);
98
99   // create the link
100   static long long counter = 0;
101
102   std::string key = std::to_string(counter);
103   counter++;
104
105   link->start_event(src, "topology", key);
106   link->end_event(dst, "topology", key);
107
108   XBT_DEBUG("  linkContainers %s <-> %s", src->get_cname(), dst->get_cname());
109 }
110
111 static void recursiveGraphExtraction(const simgrid::s4u::NetZone* netzone, const simgrid::instr::Container* container,
112                                      std::set<std::string, std::less<>>* filter)
113 {
114   if (not TRACE_platform_topology()) {
115     XBT_DEBUG("Graph extraction disabled by user.");
116     return;
117   }
118   XBT_DEBUG("Graph extraction for NetZone = %s", netzone->get_cname());
119
120   // bottom-up recursion
121   for (auto const& nz_son : netzone->get_children()) {
122     const simgrid::instr::Container* child_container = container->get_child_by_name(nz_son->get_name());
123     recursiveGraphExtraction(nz_son, child_container, filter);
124   }
125
126   auto* graph = xbt_graph_new_graph(0, nullptr);
127   std::map<std::string, xbt_node_t, std::less<>> nodes;
128   std::map<std::string, xbt_edge_t, std::less<>> edges;
129
130   netzone->get_impl()->get_graph(graph, &nodes, &edges);
131   for (auto const& [_, edge] : edges) {
132     linkContainers(simgrid::instr::Container::by_name(static_cast<const char*>(edge->src->data)),
133                    simgrid::instr::Container::by_name(static_cast<const char*>(edge->dst->data)), filter);
134   }
135   xbt_graph_free_graph(graph, xbt_free_f, xbt_free_f, nullptr);
136 }
137
138 /*
139  * user categories support
140  */
141 static void recursiveNewVariableType(const std::string& new_typename, const std::string& color,
142                                      simgrid::instr::Type* root)
143 {
144   if (root->get_name() == "HOST" || root->get_name() == "VM")
145     root->by_name_or_create("p" + new_typename, color);
146
147   if (root->get_name() == "LINK")
148     root->by_name_or_create("b" + new_typename, color);
149
150   for (auto const& [_, child] : root->get_children()) {
151     recursiveNewVariableType(new_typename, color, child.get());
152   }
153 }
154
155 void instr_new_variable_type(const std::string& new_typename, const std::string& color)
156 {
157   recursiveNewVariableType(new_typename, color, simgrid::instr::Container::get_root()->get_type());
158 }
159
160 static void recursiveNewUserVariableType(const std::string& parent_type, const std::string& new_typename,
161                                          const std::string& color, simgrid::instr::Type* root)
162 {
163   if (root->get_name() == parent_type) {
164     root->by_name_or_create(new_typename, color);
165   }
166   for (auto const& [_, child] : root->get_children())
167     recursiveNewUserVariableType(parent_type, new_typename, color, child.get());
168 }
169
170 void instr_new_user_variable_type(const std::string& parent_type, const std::string& new_typename,
171                                   const std::string& color)
172 {
173   recursiveNewUserVariableType(parent_type, new_typename, color, simgrid::instr::Container::get_root()->get_type());
174 }
175
176 static void recursiveNewUserStateType(const std::string& parent_type, const std::string& new_typename,
177                                       simgrid::instr::Type* root)
178 {
179   if (root->get_name() == parent_type)
180     root->by_name_or_create<simgrid::instr::StateType>(new_typename);
181
182   for (auto const& [_, child] : root->get_children())
183     recursiveNewUserStateType(parent_type, new_typename, child.get());
184 }
185
186 void instr_new_user_state_type(const std::string& parent_type, const std::string& new_typename)
187 {
188   recursiveNewUserStateType(parent_type, new_typename, simgrid::instr::Container::get_root()->get_type());
189 }
190
191 static void recursiveNewValueForUserStateType(const std::string& type_name, const char* val, const std::string& color,
192                                               simgrid::instr::Type* root)
193 {
194   if (root->get_name() == type_name)
195     static_cast<simgrid::instr::StateType*>(root)->add_entity_value(val, color);
196
197   for (auto const& [_, child] : root->get_children())
198     recursiveNewValueForUserStateType(type_name, val, color, child.get());
199 }
200
201 void instr_new_value_for_user_state_type(const std::string& type_name, const char* value, const std::string& color)
202 {
203   recursiveNewValueForUserStateType(type_name, value, color, simgrid::instr::Container::get_root()->get_type());
204 }
205
206 namespace simgrid::instr {
207
208 /** @brief Creates a file with the topology of the platform file used for the simulator.
209  *
210  *  The graph topology will have the following properties: all hosts, links and routers of the platform file are mapped
211  *  to graph nodes; routes are mapped to edges. The platform's zones are not represented in the output.
212  */
213 void platform_graph_export_graphviz(const std::string& output_filename)
214 {
215   auto* g     = xbt_graph_new_graph(0, nullptr);
216   std::map<std::string, xbt_node_t, std::less<>> nodes;
217   std::map<std::string, xbt_edge_t, std::less<>> edges;
218   s4u::Engine::get_instance()->get_netzone_root()->extract_xbt_graph(g, &nodes, &edges);
219
220   std::ofstream fs;
221   fs.open(output_filename, std::ofstream::out);
222   xbt_assert(not fs.fail(), "Failed to open %s", output_filename.c_str());
223
224   if (g->directed)
225     fs << "digraph test {\n";
226   else
227     fs << "graph test {\n";
228
229   fs << "  graph [overlap=scale]\n";
230
231   fs << "  node [shape=box, style=filled]\n";
232   fs << "  node [width=.3, height=.3, style=filled, color=skyblue]\n\n";
233
234   for (auto const& [node, _] : nodes)
235     fs << "  \"" << node << "\";\n";
236
237   for (auto const& [_, edge] : edges) {
238     const char* src_s = static_cast<char*>(edge->src->data);
239     const char* dst_s = static_cast<char*>(edge->dst->data);
240     if (g->directed)
241       fs << "  \"" << src_s << "\" -> \"" << dst_s << "\";\n";
242     else
243       fs << "  \"" << src_s << "\" -- \"" << dst_s << "\";\n";
244   }
245   fs << "}\n";
246   fs.close();
247
248   xbt_graph_free_graph(g, xbt_free_f, xbt_free_f, nullptr);
249 }
250
251 void platform_graph_export_csv(const std::string& output_filename)
252 {
253   auto* g         = xbt_graph_new_graph(0, nullptr);
254   std::map<std::string, xbt_node_t, std::less<>> nodes;
255   std::map<std::string, xbt_edge_t, std::less<>> edges;
256   s4u::Engine::get_instance()->get_netzone_root()->extract_xbt_graph(g, &nodes, &edges);
257
258   std::ofstream fs;
259   fs.open(output_filename, std::ofstream::out);
260   xbt_assert(not fs.fail(), "Failed to open %s", output_filename.c_str());
261
262   fs << "src,dst" << std::endl;
263   for (auto const& [_, edge] : edges) {
264     const char* src_s = static_cast<char*>(edge->src->data);
265     const char* dst_s = static_cast<char*>(edge->dst->data);
266     fs << src_s << "," << dst_s << "\n";
267   }
268   fs.close();
269   xbt_graph_free_graph(g, xbt_free_f, xbt_free_f, nullptr);
270 }
271
272 /* Callbacks */
273 static std::vector<NetZoneContainer*> currentContainer; /* push and pop, used only in creation */
274 static void on_netzone_creation(s4u::NetZone const& netzone)
275 {
276   std::string id = netzone.get_name();
277   if (Container::get_root() == nullptr) {
278     auto* root = new NetZoneContainer(id, 0, nullptr);
279     xbt_assert(Container::get_root() == root);
280
281     if (TRACE_smpi_is_enabled()) {
282       auto* mpi = root->get_type()->by_name_or_create<ContainerType>("MPI");
283       if (not TRACE_smpi_is_grouped())
284         mpi->by_name_or_create<StateType>("MPI_STATE");
285       root->get_type()->by_name_or_create("MPI_LINK", mpi, mpi);
286       root->get_type()->by_name_or_create("MIGRATE_LINK", mpi, mpi);
287       mpi->by_name_or_create<StateType>("MIGRATE_STATE");
288     }
289
290     if (TRACE_needs_platform()) {
291       currentContainer.push_back(root);
292     }
293     return;
294   }
295
296   if (TRACE_needs_platform()) {
297     auto level      = static_cast<unsigned>(currentContainer.size());
298     auto* container = new NetZoneContainer(id, level, currentContainer.back());
299     currentContainer.push_back(container);
300   }
301 }
302
303 static void on_link_creation(s4u::Link const& link)
304 {
305   if (currentContainer.empty()) // No ongoing parsing. Are you creating the loopback?
306     return;
307
308   auto* container = new Container(link.get_name(), "LINK", currentContainer.back());
309
310   if ((TRACE_categorized() || TRACE_uncategorized() || TRACE_platform()) && (not TRACE_disable_link())) {
311     VariableType* bandwidth = container->get_type()->by_name_or_create("bandwidth", "");
312     bandwidth->set_calling_container(container);
313     bandwidth->set_event(0, link.get_bandwidth());
314     VariableType* latency = container->get_type()->by_name_or_create("latency", "");
315     latency->set_calling_container(container);
316     latency->set_event(0, link.get_latency());
317   }
318
319   if (TRACE_uncategorized()) {
320     container->get_type()->by_name_or_create("bandwidth_used", "0.5 0.5 0.5");
321   }
322 }
323
324 static void on_host_creation(s4u::Host const& host)
325 {
326   if (Container::by_name_or_null(host.get_name())) // This host already exists, do nothing
327     return;
328
329   Container* container  = new HostContainer(host, currentContainer.back());
330   const Container* root = Container::get_root();
331
332   if ((TRACE_categorized() || TRACE_uncategorized() || TRACE_platform()) && (not TRACE_disable_speed())) {
333     VariableType* speed = container->get_type()->by_name_or_create("speed", "");
334     speed->set_calling_container(container);
335     speed->set_event(0, host.get_speed());
336
337     VariableType* cores = container->get_type()->by_name_or_create("core_count", "");
338     cores->set_calling_container(container);
339     cores->set_event(0, host.get_core_count());
340   }
341
342   if (TRACE_uncategorized())
343     container->get_type()->by_name_or_create("speed_used", "0.5 0.5 0.5");
344
345   if (TRACE_smpi_is_enabled() && TRACE_smpi_is_grouped()) {
346     auto* mpi = container->get_type()->by_name_or_create<ContainerType>("MPI");
347     mpi->by_name_or_create<StateType>("MPI_STATE");
348     root->get_type()->by_name_or_create("MIGRATE_LINK", mpi, mpi);
349     mpi->by_name_or_create<StateType>("MIGRATE_STATE");
350   }
351
352    if (TRACE_actor_is_enabled()) {
353     auto* host_type = container->get_type();
354     auto* state     = host_type->by_name_or_create<StateType>("HOST_STATE");
355     state->set_calling_container(container);
356     state->add_entity_value("receive", "1 0 0");
357     state->add_entity_value("send", "0 0 1");
358     state->add_entity_value("execute", "0 1 1");
359   }
360 }
361
362 static void on_action_state_change(kernel::resource::Action const& action,
363                                    kernel::resource::Action::State /* previous */)
364 {
365   auto n = static_cast<unsigned>(action.get_variable()->get_number_of_constraint());
366
367   for (unsigned i = 0; i < n; i++) {
368     double value = action.get_rate() * action.get_variable()->get_constraint_weight(i);
369     /* Beware of composite actions: ptasks put links and cpus together. Extra pb: we cannot dynamic_cast from void* */
370     kernel::resource::Resource* resource = action.get_variable()->get_constraint(i)->get_id();
371     if (const auto* cpu = dynamic_cast<kernel::resource::CpuImpl*>(resource))
372       resource_set_utilization("HOST", "speed_used", cpu->get_cname(), action.get_category(), value,
373                                action.get_last_update(), simgrid_get_clock() - action.get_last_update());
374
375     else if (const auto* link = dynamic_cast<kernel::resource::StandardLinkImpl*>(resource))
376       resource_set_utilization("LINK", "bandwidth_used", link->get_cname(), action.get_category(), value,
377                                action.get_last_update(), simgrid_get_clock() - action.get_last_update());
378   }
379 }
380
381 static void on_activity_suspend_resume(s4u::Activity const& activity)
382 {
383   on_action_state_change(*activity.get_impl()->model_action_, /*ignored*/ kernel::resource::Action::State::STARTED);
384 }
385
386 static void on_platform_created()
387 {
388   currentContainer.clear();
389   std::set<std::string, std::less<>> filter;
390   XBT_DEBUG("Starting graph extraction.");
391   recursiveGraphExtraction(s4u::Engine::get_instance()->get_netzone_root(), Container::get_root(), &filter);
392   XBT_DEBUG("Graph extraction finished.");
393   dump_buffer(true);
394 }
395
396 static void on_actor_creation(s4u::Actor const& actor)
397 {
398   const Container* root      = Container::get_root();
399   Container* container       = Container::by_name(actor.get_host()->get_name());
400   std::string container_name = instr_pid(actor);
401
402   container->create_child(container_name, "ACTOR");
403   auto* actor_type = container->get_type()->by_name_or_create<ContainerType>("ACTOR");
404   auto* state      = actor_type->by_name_or_create<StateType>("ACTOR_STATE");
405   state->add_entity_value("suspend", "1 0 1");
406   state->add_entity_value("sleep", "1 1 0");
407   state->add_entity_value("receive", "1 0 0");
408   state->add_entity_value("send", "0 0 1");
409   state->add_entity_value("execute", "0 1 1");
410   root->get_type()->by_name_or_create("ACTOR_LINK", actor_type, actor_type);
411
412   actor.on_exit([container_name](bool failed) {
413     if (failed)
414       // kill means that this actor no longer exists, let's destroy it
415       Container::by_name(container_name)->remove_from_parent();
416   });
417 }
418
419 static void on_actor_host_change(s4u::Actor const& actor, s4u::Host const& /*previous_location*/)
420 {
421   static long long int counter = 0;
422   Container* container         = Container::by_name(instr_pid(actor));
423   LinkType* link               = Container::get_root()->get_link("ACTOR_LINK");
424
425   // start link
426   link->start_event(container, "M", std::to_string(counter));
427   // destroy existing container of this process
428   container->remove_from_parent();
429   // create new container on the new_host location
430   Container::by_name(actor.get_host()->get_name())->create_child(instr_pid(actor), "ACTOR");
431   // end link
432   link->end_event(Container::by_name(instr_pid(actor)), "M", std::to_string(counter));
433   counter++;
434 }
435
436 static void on_vm_creation(s4u::Host const& host)
437 {
438   const Container* container = new HostContainer(host, currentContainer.back());
439   const Container* root      = Container::get_root();
440   auto* vm                   = container->get_type()->by_name_or_create<ContainerType>("VM");
441   auto* state                = vm->by_name_or_create<StateType>("VM_STATE");
442   state->add_entity_value("suspend", "1 0 1");
443   state->add_entity_value("sleep", "1 1 0");
444   state->add_entity_value("receive", "1 0 0");
445   state->add_entity_value("send", "0 0 1");
446   state->add_entity_value("execute", "0 1 1");
447   root->get_type()->by_name_or_create("VM_LINK", vm, vm);
448   root->get_type()->by_name_or_create("VM_ACTOR_LINK", vm, vm);
449 }
450
451 void define_callbacks()
452 {
453   // always need the callbacks to zones (we need only the root zone), to create the rootContainer and the rootType
454   // properly
455   if (TRACE_needs_platform()) {
456     s4u::Engine::on_platform_created_cb(on_platform_created);
457     s4u::Host::on_creation_cb(on_host_creation);
458     s4u::Host::on_speed_change_cb([](s4u::Host const& host) {
459       Container::by_name(host.get_name())
460           ->get_variable("speed")
461           ->set_event(simgrid_get_clock(), host.get_core_count() * host.get_available_speed());
462     });
463     s4u::Link::on_creation_cb(on_link_creation);
464     s4u::Link::on_bandwidth_change_cb([](s4u::Link const& link) {
465       Container::by_name(link.get_name())
466           ->get_variable("bandwidth")
467           ->set_event(
468               simgrid_get_clock(),
469               static_cast<kernel::resource::NetworkModel*>(link.get_impl()->get_model())->get_bandwidth_factor() *
470                   link.get_bandwidth());
471     });
472     kernel::routing::NetPoint::on_creation.connect([](kernel::routing::NetPoint const& netpoint) {
473       if (netpoint.is_router())
474         new RouterContainer(netpoint.get_name(), currentContainer.back());
475     });
476   }
477
478   s4u::NetZone::on_creation_cb(on_netzone_creation);
479
480   s4u::Host::on_exec_state_change_cb(on_action_state_change);
481   s4u::Link::on_communication_state_change_cb(on_action_state_change);
482   s4u::Exec::on_suspend_cb(on_activity_suspend_resume);
483   s4u::Exec::on_resume_cb(on_activity_suspend_resume);
484
485   if (TRACE_actor_is_enabled()) {
486     s4u::Actor::on_creation_cb(on_actor_creation);
487     s4u::Actor::on_destruction_cb([](s4u::Actor const& actor) {
488       if (auto* container = Container::by_name_or_null(instr_pid(actor)))
489         container->remove_from_parent();
490     });
491     s4u::Actor::on_suspend_cb([](s4u::Actor const& actor) {
492       Container::by_name(instr_pid(actor))->get_state("ACTOR_STATE")->push_event("suspend");
493     });
494     s4u::Actor::on_resume_cb(
495         [](s4u::Actor const& actor) { Container::by_name(instr_pid(actor))->get_state("ACTOR_STATE")->pop_event(); });
496     s4u::Actor::on_sleep_cb([](s4u::Actor const& actor) {
497       Container::by_name(instr_pid(actor))->get_state("ACTOR_STATE")->push_event("sleep");
498     });
499     s4u::Actor::on_wake_up_cb(
500         [](s4u::Actor const& actor) { Container::by_name(instr_pid(actor))->get_state("ACTOR_STATE")->pop_event(); });
501
502     s4u::Exec::on_start_cb([](s4u::Exec const& e) {
503       std::string pid = instr_pid(*s4u::Actor::self());
504       if (pid == "-0") //Exec is launched directly by Maestro, use the host as container
505         Container::by_name(e.get_host()->get_name())->get_state("HOST_STATE")->push_event("execute");
506       else
507         Container::by_name(pid)->get_state("ACTOR_STATE")->push_event("execute");
508     });
509
510     s4u::Exec::on_completion_cb([](const s4u::Exec& e) {
511       std::string pid = instr_pid(*s4u::Actor::self());
512       if (pid == "-0") //Exec is launched directly by Maestro, use the host as container
513         Container::by_name(e.get_host()->get_name())->get_state("HOST_STATE")->pop_event();
514       else
515         Container::by_name(pid)->get_state("ACTOR_STATE")->pop_event();
516     });
517
518     s4u::Comm::on_completion_cb([](const s4u::Comm& c) {
519       if (c.get_sender()) {
520         Container::by_name(instr_pid(*c.get_sender()))->get_state("ACTOR_STATE")->pop_event();
521         Container::by_name(instr_pid(*c.get_receiver()))->get_state("ACTOR_STATE")->pop_event();
522       } else {
523         Container::by_name(c.get_source()->get_name())->get_state("HOST_STATE")->pop_event();
524         Container::by_name(c.get_destination()->get_name())->get_state("HOST_STATE")->pop_event();
525       }
526     });
527     s4u::Comm::on_start_cb([](s4u::Comm const& c) {
528       std::string pid = instr_pid(*s4u::Actor::self());
529       if (pid == "-0") { //Comm is launched directly by Maestro, use the host as container
530         Container::by_name(c.get_source()->get_name())->get_state("HOST_STATE")->push_event("start");
531         Container::by_name(c.get_destination()->get_name())->get_state("HOST_STATE")->push_event("start");
532       }
533     });
534     s4u::Comm::on_send_cb([](s4u::Comm const&) {
535       Container::by_name(instr_pid(*s4u::Actor::self()))->get_state("ACTOR_STATE")->push_event("send");
536     });
537     s4u::Comm::on_recv_cb([](s4u::Comm const&) {
538       Container::by_name(instr_pid(*s4u::Actor::self()))->get_state("ACTOR_STATE")->push_event("receive");
539     });
540     s4u::Actor::on_host_change_cb(on_actor_host_change);
541   }
542
543   if (TRACE_smpi_is_enabled() && TRACE_smpi_is_computing()) {
544     s4u::Exec::on_start_cb([](s4u::Exec const& exec) {
545       Container::by_name("rank-" + std::to_string(s4u::Actor::self()->get_pid()))
546           ->get_state("MPI_STATE")
547           ->push_event("computing", new CpuTIData("compute", exec.get_cost()));
548     });
549     s4u::Exec::on_completion_cb([](const s4u::Exec&) {
550       Container::by_name("rank-" + std::to_string(s4u::Actor::self()->get_pid()))->get_state("MPI_STATE")->pop_event();
551     });
552   }
553
554   if (TRACE_vm_is_enabled()) {
555     s4u::Host::on_creation_cb(on_vm_creation);
556     s4u::VirtualMachine::on_start_cb([](s4u::VirtualMachine const& vm) {
557       Container::by_name(vm.get_name())->get_state("VM_STATE")->push_event("start");
558     });
559     s4u::VirtualMachine::on_started_cb(
560         [](s4u::VirtualMachine const& vm) { Container::by_name(vm.get_name())->get_state("VM_STATE")->pop_event(); });
561     s4u::VirtualMachine::on_suspend_cb([](s4u::VirtualMachine const& vm) {
562       Container::by_name(vm.get_name())->get_state("VM_STATE")->push_event("suspend");
563     });
564     s4u::VirtualMachine::on_resume_cb(
565         [](s4u::VirtualMachine const& vm) { Container::by_name(vm.get_name())->get_state("VM_STATE")->pop_event(); });
566     s4u::Host::on_destruction_cb(
567         [](s4u::Host const& host) { Container::by_name(host.get_name())->remove_from_parent(); });
568   }
569 }
570 } // namespace simgrid::instr