Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Merge branch 'add_missing_comm_python_bindings' into 'master'
[simgrid.git] / src / instr / instr_config.cpp
1 /* Copyright (c) 2010-2022. 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/Exception.hpp>
7 #include <simgrid/s4u/Engine.hpp>
8
9 #include "src/instr/instr_private.hpp"
10 #include "xbt/config.hpp"
11 #include "xbt/xbt_os_time.h"
12
13 #include <sys/stat.h>
14 #ifdef WIN32
15 #include <direct.h> // _mkdir
16 #endif
17
18 #include <fstream>
19 #include <string>
20 #include <vector>
21
22 XBT_LOG_NEW_CATEGORY(instr, "Logging the behavior of the tracing system (used for Visualization/Analysis of simulations)");
23 XBT_LOG_NEW_DEFAULT_SUBCATEGORY (instr_config, instr, "Configuration");
24
25 std::ofstream tracing_file;
26 std::map<const simgrid::instr::Container*, std::ofstream*> tracing_files; // TI specific
27
28 constexpr char OPT_TRACING_BASIC[]             = "tracing/basic";
29 constexpr char OPT_TRACING_COMMENT_FILE[]      = "tracing/comment-file";
30 constexpr char OPT_TRACING_DISABLE_DESTROY[]   = "tracing/disable-destroy";
31 constexpr char OPT_TRACING_FORMAT_TI_ONEFILE[] = "tracing/smpi/format/ti-one-file";
32 constexpr char OPT_TRACING_SMPI[]              = "tracing/smpi";
33 constexpr char OPT_TRACING_TOPOLOGY[]          = "tracing/platform/topology";
34
35 static simgrid::config::Flag<bool> trace_enabled{
36     "tracing", "Enable the tracing system. You have to enable this option to use other tracing options.", false};
37
38 static simgrid::config::Flag<bool> trace_actor_enabled{
39     "tracing/actor",
40     "Trace the behavior of all categorized actors, grouping them by host. "
41     "Can be used to track actor location if the simulator does actor migration.",
42     false};
43
44 static simgrid::config::Flag<bool> trace_vm_enabled{"tracing/vm", "Trace the behavior of all virtual machines.", false};
45
46 static simgrid::config::Flag<bool> trace_platform{"tracing/platform",
47                                                   "Register the platform in the trace as a hierarchy.", false};
48
49 static simgrid::config::Flag<bool> trace_platform_topology{
50     OPT_TRACING_TOPOLOGY, "Register the platform topology in the trace as a graph.", true};
51 static simgrid::config::Flag<bool> trace_smpi_enabled{OPT_TRACING_SMPI, "Tracing of the SMPI interface.", false};
52 static simgrid::config::Flag<bool> trace_smpi_grouped{"tracing/smpi/group", "Group MPI processes by host.", false};
53
54 static simgrid::config::Flag<bool> trace_smpi_computing{
55     "tracing/smpi/computing", "Generate 'Computing' states to trace the out-of-SMPI parts of the application", false};
56
57 static simgrid::config::Flag<bool> trace_smpi_sleeping{
58     "tracing/smpi/sleeping", "Generate 'Sleeping' states for the sleeps in the application that do not pertain to SMPI",
59     false};
60
61 static simgrid::config::Flag<bool> trace_view_internals{
62     "tracing/smpi/internals",
63     "Generate tracing events corresponding to point-to-point messages sent by SMPI collective communications", false};
64
65 static simgrid::config::Flag<bool> trace_categorized{
66     "tracing/categorized", "Trace categorized resource utilization of hosts and links.", false};
67
68 static simgrid::config::Flag<bool> trace_uncategorized{
69     "tracing/uncategorized",
70     "Trace uncategorized resource utilization of hosts and links. "
71     "To use if the simulator does not use tracing categories but resource utilization have to be traced.",
72     false};
73
74 static simgrid::config::Flag<bool> trace_disable_destroy{OPT_TRACING_DISABLE_DESTROY,
75                                                          "Disable platform containers destruction.", false};
76 static simgrid::config::Flag<bool> trace_basic{OPT_TRACING_BASIC, "Avoid extended events (impoverished trace file).",
77                                                false};
78
79 static simgrid::config::Flag<bool> trace_display_sizes{
80     "tracing/smpi/display-sizes",
81     "Add message size information (in bytes) to the to links and states (SMPI only). "
82     "For collectives, it usually corresponds to the total number of bytes sent by a process.",
83     false};
84
85 static simgrid::config::Flag<bool> trace_disable_link{"tracing/disable_link",
86                                                       "Do not trace link bandwidth and latency.", false};
87 static simgrid::config::Flag<bool> trace_disable_power{"tracing/disable_power", "Do not trace host power.", false};
88
89 bool TRACE_needs_platform ()
90 {
91   return TRACE_actor_is_enabled() || TRACE_vm_is_enabled() || TRACE_categorized() || TRACE_uncategorized() ||
92          TRACE_platform() || (TRACE_smpi_is_enabled() && TRACE_smpi_is_grouped());
93 }
94
95 bool TRACE_is_enabled()
96 {
97   return trace_enabled;
98 }
99
100 bool TRACE_platform()
101 {
102   return trace_platform;
103 }
104
105 bool TRACE_platform_topology()
106 {
107   return trace_platform_topology;
108 }
109
110 bool TRACE_smpi_is_enabled()
111 {
112   return (trace_smpi_enabled || TRACE_smpi_is_grouped()) && TRACE_is_enabled();
113 }
114
115 bool TRACE_smpi_is_grouped()
116 {
117   return trace_smpi_grouped;
118 }
119
120 bool TRACE_smpi_is_computing()
121 {
122   return trace_smpi_computing;
123 }
124
125 bool TRACE_smpi_is_sleeping()
126 {
127   return trace_smpi_sleeping;
128 }
129
130 bool TRACE_smpi_view_internals()
131 {
132   return trace_view_internals;
133 }
134
135 bool TRACE_categorized ()
136 {
137   return trace_categorized;
138 }
139
140 bool TRACE_uncategorized ()
141 {
142   return trace_uncategorized;
143 }
144
145 bool TRACE_actor_is_enabled()
146 {
147   return trace_actor_enabled && trace_enabled;
148 }
149
150 bool TRACE_vm_is_enabled()
151 {
152   return trace_vm_enabled && trace_enabled;
153 }
154
155 bool TRACE_disable_link()
156 {
157   return trace_disable_link && trace_enabled;
158 }
159
160 bool TRACE_disable_speed()
161 {
162   return trace_disable_power && trace_enabled;
163 }
164
165 bool TRACE_display_sizes ()
166 {
167   return trace_display_sizes && trace_smpi_enabled && trace_enabled;
168 }
169
170 static void print_line(const char* option, const char* desc, const char* longdesc)
171 {
172   std::string str = std::string("--cfg=") + option + " ";
173
174   int len = static_cast<int>(str.size());
175   XBT_HELP("%s%*.*s %s", str.c_str(), 30 - len, 30 - len, "", desc);
176   if (longdesc != nullptr) {
177     XBT_HELP("%s\n", longdesc);
178   }
179 }
180
181 void TRACE_help()
182 {
183   XBT_HELP("Description of the tracing options accepted by this simulator:\n");
184   print_line(OPT_TRACING_SMPI, "Trace the MPI Interface (SMPI)",
185              "  This option only has effect if this simulator is SMPI-based. Traces the MPI\n"
186              "  interface and generates a trace that can be analyzed using Gantt-like\n"
187              "  visualizations. Every MPI function (implemented by SMPI) is transformed in a\n"
188              "  state, and point-to-point communications can be analyzed with arrows.");
189   print_line(OPT_TRACING_DISABLE_DESTROY, "Disable platform containers destruction",
190              "  Disable the destruction of containers at the end of simulation. This can be\n"
191              "  used with simulators that have a different notion of time (different from\n"
192              "  the simulated time).");
193   print_line(OPT_TRACING_BASIC, "Avoid extended events (impoverished trace file).",
194              "  Some visualization tools are not able to parse correctly the Paje file format.\n"
195              "  Use this option if you are using one of these tools to visualize the simulation\n"
196              "  trace. Keep in mind that the trace might be incomplete, without all the\n"
197              "  information that would be registered otherwise.");
198   print_line(OPT_TRACING_FORMAT_TI_ONEFILE, "Only works for SMPI now, and TI output format",
199              "  By default, each process outputs to a separate file, inside a filename_files folder\n"
200              "  By setting this option to yes, all processes will output to only one file\n"
201              "  This is meant to avoid opening thousands of files with large simulations");
202   print_line(OPT_TRACING_TOPOLOGY, "Register the platform topology as a graph",
203              "  This option (enabled by default) can be used to disable the tracing of\n"
204              "  the platform topology in the trace file. Sometimes, such task is really\n"
205              "  time consuming, since it must get the route from each host to other hosts\n"
206              "  within the same Autonomous System (AS).");
207 }
208
209 namespace simgrid {
210 namespace instr {
211 static bool trace_active = false;
212 TraceFormat trace_format = TraceFormat::Paje;
213 int trace_precision;
214
215 /*************
216  * Callbacks *
217  *************/
218 xbt::signal<void(Container const&)> Container::on_creation;
219 xbt::signal<void(Container const&)> Container::on_destruction;
220 xbt::signal<void(Type const&, PajeEventType)> Type::on_creation;
221 xbt::signal<void(LinkType const&, Type const&, Type const&)> LinkType::on_creation;
222 xbt::signal<void(PajeEvent&)> PajeEvent::on_creation;
223 xbt::signal<void(PajeEvent const&)> PajeEvent::on_destruction;
224 xbt::signal<void(StateEvent const&)> StateEvent::on_destruction;
225 xbt::signal<void(EntityValue const&)> EntityValue::on_creation;
226
227 static void on_container_creation_paje(const Container& c)
228 {
229   double timestamp = simgrid_get_clock();
230   std::stringstream stream;
231
232   XBT_DEBUG("%s: event_type=%u, timestamp=%f", __func__, static_cast<unsigned>(PajeEventType::CreateContainer),
233             timestamp);
234
235   stream << std::fixed << std::setprecision(trace_precision) << PajeEventType::CreateContainer << " ";
236   stream << timestamp << " " << c.get_id() << " " << c.get_type()->get_id() << " " << c.get_parent()->get_id() << " \"";
237   if (c.get_name().find("rank-") != 0)
238     stream << c.get_name() << "\"";
239   else
240     /* Subtract -1 because this is the process id and we transform it to the rank id */
241     stream << "rank-" << stoi(c.get_name().substr(5)) - 1 << "\"";
242
243   XBT_DEBUG("Dump %s", stream.str().c_str());
244   tracing_file << stream.str() << std::endl;
245 }
246
247 static void on_container_destruction_paje(const Container& c)
248 {
249   // trace my destruction, but not if user requests so or if the container is root
250   if (not trace_disable_destroy && &c != Container::get_root()) {
251     std::stringstream stream;
252     double timestamp = simgrid_get_clock();
253
254     XBT_DEBUG("%s: event_type=%u, timestamp=%f", __func__, static_cast<unsigned>(PajeEventType::DestroyContainer),
255               timestamp);
256
257     stream << std::fixed << std::setprecision(trace_precision) << PajeEventType::DestroyContainer << " ";
258     stream << timestamp << " " << c.get_type()->get_id() << " " << c.get_id();
259     XBT_DEBUG("Dump %s", stream.str().c_str());
260     tracing_file << stream.str() << std::endl;
261   }
262 }
263
264 static void on_container_creation_ti(const Container& c)
265 {
266   XBT_DEBUG("%s: event_type=%u, timestamp=%f", __func__, static_cast<unsigned>(PajeEventType::CreateContainer),
267             simgrid_get_clock());
268   // if we are in the mode with only one file
269   static std::ofstream* ti_unique_file = nullptr;
270   static double prefix                 = 0.0;
271
272   if (tracing_files.empty()) {
273     // generate unique run id with time
274     prefix = xbt_os_time();
275   }
276
277   if (not simgrid::config::get_value<bool>("tracing/smpi/format/ti-one-file") || ti_unique_file == nullptr) {
278     std::string folder_name = simgrid::config::get_value<std::string>("tracing/filename") + "_files";
279     std::string filename    = folder_name + "/" + std::to_string(prefix) + "_" + c.get_name() + ".txt";
280 #ifdef WIN32
281     _mkdir(folder_name.c_str());
282 #else
283     mkdir(folder_name.c_str(), S_IRWXU | S_IRWXG | S_IRWXO);
284 #endif
285     ti_unique_file = new std::ofstream(filename.c_str(), std::ofstream::out);
286     xbt_assert(not ti_unique_file->fail(), "Tracefile %s could not be opened for writing", filename.c_str());
287     tracing_file << filename << std::endl;
288   }
289   tracing_files.insert({&c, ti_unique_file});
290 }
291
292 static void on_container_destruction_ti(const Container& c)
293 {
294   if (not trace_disable_destroy && &c != Container::get_root()) {
295     if (not simgrid::config::get_value<bool>("tracing/smpi/format/ti-one-file") || tracing_files.size() == 1) {
296       tracing_files.at(&c)->close();
297       delete tracing_files.at(&c);
298     }
299     tracing_files.erase(&c);
300   }
301 }
302
303 static void on_entity_value_creation(const EntityValue& value)
304 {
305   std::stringstream stream;
306   XBT_DEBUG("%s: event_type=%u", __func__, static_cast<unsigned>(PajeEventType::DefineEntityValue));
307   stream << std::fixed << std::setprecision(trace_precision) << PajeEventType::DefineEntityValue;
308   stream << " " << value.get_id() << " " << value.get_parent()->get_id() << " " << value.get_name();
309   if (not value.get_color().empty())
310     stream << " \"" << value.get_color() << "\"";
311   XBT_DEBUG("Dump %s", stream.str().c_str());
312   tracing_file << stream.str() << std::endl;
313 }
314
315 static void on_event_creation(PajeEvent& event)
316 {
317   XBT_DEBUG("%s: event_type=%u, timestamp=%.*f", __func__, static_cast<unsigned>(event.eventType_), trace_precision,
318             event.timestamp_);
319   event.stream_ << std::fixed << std::setprecision(trace_precision);
320   event.stream_ << event.eventType_ << " " << event.timestamp_ << " ";
321   event.stream_ << event.get_type()->get_id() << " " << event.get_container()->get_id();
322 }
323
324 static void on_event_destruction(const PajeEvent& event)
325 {
326   XBT_DEBUG("Dump %s", event.stream_.str().c_str());
327   tracing_file << event.stream_.str() << std::endl;
328 }
329
330 static void on_state_event_destruction(const StateEvent& event)
331 {
332   if (event.has_extra())
333     *tracing_files.at(event.get_container()) << event.stream_.str() << std::endl;
334 }
335
336 static void on_type_creation(const Type& type, PajeEventType event_type)
337 {
338   if (event_type == PajeEventType::DefineLinkType)
339     return; // this kind of type has to be handled differently
340
341   std::stringstream stream;
342   stream << std::fixed << std::setprecision(trace_precision);
343   XBT_DEBUG("%s: event_type=%u, timestamp=%.*f", __func__, static_cast<unsigned>(event_type), trace_precision, 0.);
344   stream << event_type << " " << type.get_id() << " " << type.get_parent()->get_id() << " " << type.get_name();
345   if (type.is_colored())
346     stream << " \"" << type.get_color() << "\"";
347   XBT_DEBUG("Dump %s", stream.str().c_str());
348   tracing_file << stream.str() << std::endl;
349 }
350
351 static void on_link_type_creation(const Type& type, const Type& source, const Type& dest)
352 {
353   std::stringstream stream;
354   XBT_DEBUG("%s: event_type=%u, timestamp=%.*f", __func__, static_cast<unsigned>(PajeEventType::DefineLinkType),
355             trace_precision, 0.);
356   stream << PajeEventType::DefineLinkType << " " << type.get_id() << " " << type.get_parent()->get_id();
357   stream << " " << source.get_id() << " " << dest.get_id() << " " << type.get_name();
358   XBT_DEBUG("Dump %s", stream.str().c_str());
359   tracing_file << stream.str() << std::endl;
360 }
361
362 static void on_simulation_start()
363 {
364   if (trace_active || not TRACE_is_enabled())
365     return;
366
367   define_callbacks();
368
369   XBT_DEBUG("Tracing starts");
370   trace_precision = config::get_value<int>("tracing/precision");
371
372   /* init the tracing module to generate the right output */
373   std::string format = config::get_value<std::string>("tracing/smpi/format");
374   XBT_DEBUG("Tracing format %s", format.c_str());
375
376   /* open the trace file(s) */
377   std::string filename = simgrid::config::get_value<std::string>("tracing/filename");
378   tracing_file.open(filename.c_str(), std::ofstream::out);
379   if (tracing_file.fail()) {
380     throw TracingError(XBT_THROW_POINT,
381                        xbt::string_printf("Tracefile %s could not be opened for writing.", filename.c_str()));
382   }
383
384   XBT_DEBUG("Filename %s is open for writing", filename.c_str());
385
386   if (format == "Paje") {
387     Container::on_creation_cb(on_container_creation_paje);
388     Container::on_destruction_cb(on_container_destruction_paje);
389     EntityValue::on_creation_cb(on_entity_value_creation);
390     Type::on_creation_cb(on_type_creation);
391     LinkType::on_creation_cb(on_link_type_creation);
392     PajeEvent::on_creation_cb(on_event_creation);
393     PajeEvent::on_destruction_cb(on_event_destruction);
394
395     paje::dump_generator_version();
396
397     /* output one line comment */
398     std::string comment = simgrid::config::get_value<std::string>("tracing/comment");
399     if (not comment.empty())
400       tracing_file << "# " << comment << std::endl;
401
402     /* output comment file */
403     paje::dump_comment_file(config::get_value<std::string>(OPT_TRACING_COMMENT_FILE));
404     paje::dump_header(trace_basic, TRACE_display_sizes());
405   } else {
406     trace_format = TraceFormat::Ti;
407     Container::on_creation_cb(on_container_creation_ti);
408     Container::on_destruction_cb(on_container_destruction_ti);
409     StateEvent::on_destruction_cb(on_state_event_destruction);
410   }
411
412   trace_active = true;
413   XBT_DEBUG("Tracing is on");
414 }
415
416 static void on_simulation_end()
417 {
418   if (not trace_active)
419     return;
420
421   /* dump trace buffer */
422   last_timestamp_to_dump = simgrid_get_clock();
423   dump_buffer(true);
424
425   const Type* root_type = Container::get_root()->get_type();
426   /* destroy all data structures of tracing (and free) */
427   delete Container::get_root();
428   delete root_type;
429
430   /* close the trace files */
431   tracing_file.close();
432   XBT_DEBUG("Filename %s is closed", config::get_value<std::string>("tracing/filename").c_str());
433
434   /* de-activate trace */
435   trace_active = false;
436   XBT_DEBUG("Tracing is off");
437   XBT_DEBUG("Tracing system is shutdown");
438 }
439
440 void init()
441 {
442   static bool is_initialized = false;
443   if (is_initialized)
444     return;
445
446   is_initialized = true;
447
448   /* name of the tracefile */
449   config::declare_flag<std::string>("tracing/filename", "Trace file created by the instrumented SimGrid.",
450                                     "simgrid.trace");
451   config::declare_flag<std::string>("tracing/smpi/format",
452                                     "Select trace output format used by SMPI. The default is the 'Paje' format. "
453                                     "The 'TI' (Time-Independent) format allows for trace replay.",
454                                     "Paje");
455
456   config::declare_flag<bool>(OPT_TRACING_FORMAT_TI_ONEFILE,
457                              "(smpi only) For replay format only : output to one file only", false);
458   config::declare_flag<std::string>("tracing/comment", "Add a comment line to the top of the trace file.", "");
459   config::declare_flag<std::string>(OPT_TRACING_COMMENT_FILE,
460                                     "Add the contents of a file as comments to the top of the trace.", "");
461   config::declare_flag<int>("tracing/precision",
462                             "Numerical precision used when timestamping events "
463                             "(expressed in number of digits after decimal point)",
464                             6);
465
466   /* Connect Engine callbacks */
467   s4u::Engine::on_platform_creation_cb(on_simulation_start);
468   s4u::Engine::on_time_advance_cb([](double /*time_delta*/) { dump_buffer(false); });
469   s4u::Engine::on_deadlock_cb(on_simulation_end);
470   s4u::Engine::on_simulation_end_cb(on_simulation_end);
471 }
472 } // namespace instr
473 } // namespace simgrid