Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
[SAMPI] Added AmpiMigrateTIData
[simgrid.git] / src / instr / instr_private.hpp
1 /* Copyright (c) 2010-2018. 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 #ifndef INSTR_PRIVATE_HPP
7 #define INSTR_PRIVATE_HPP
8
9 #include <xbt/base.h>
10
11 #include "simgrid/instr.h"
12 #include "simgrid/s4u/Actor.hpp"
13 #include "src/instr/instr_paje_containers.hpp"
14 #include "src/instr/instr_paje_events.hpp"
15 #include "src/instr/instr_paje_types.hpp"
16 #include "src/instr/instr_paje_values.hpp"
17 #include "src/internal_config.h"
18 #include "xbt/graph.h"
19
20 #include <fstream>
21 #include <iomanip> /** std::setprecision **/
22 #include <iostream>
23 #include <map>
24 #include <memory>
25 #include <set>
26 #include <sstream>
27 #include <string>
28 #include <sys/stat.h>
29 #ifdef WIN32
30 #include <direct.h> // _mkdir
31 /* Need to define function drand48 for Windows */
32 #define drand48() (rand() / (RAND_MAX + 1.0))
33 #endif
34
35 typedef simgrid::instr::Container* container_t;
36
37 namespace simgrid {
38 namespace instr {
39
40 /* Format of TRACING output.
41  *   - paje is the regular format, that we all know
42  *   - TI is a trick to reuse the tracing functions to generate a time independent trace during the execution. Such
43  *     trace can easily be replayed with smpi_replay afterward. This trick should be removed and replaced by some code
44  *     using the signal that we will create to cleanup the TRACING
45  */
46 enum class TraceFormat { Paje, /*TimeIndependent*/ Ti };
47 extern TraceFormat trace_format;
48
49 class TIData {
50   std::string name_;
51   double amount_ = 0;
52
53 public:
54   int endpoint                 = 0;
55   int send_size                = 0;
56   std::shared_ptr<std::vector<int>> sendcounts = nullptr;
57   int recv_size                = 0;
58   std::shared_ptr<std::vector<int>> recvcounts = nullptr;
59   std::string send_type        = "";
60   std::string recv_type        = "";
61
62   // NoOpTI: init, finalize, test, wait, barrier
63   explicit TIData(std::string name) : name_(name){};
64   // CPuTI: compute, sleep (+ waitAny and waitall out of laziness)
65   explicit TIData(std::string name, double amount) : name_(name), amount_(amount){};
66   // Pt2PtTI: send, isend, sssend, issend, recv, irecv
67   explicit TIData(std::string name, int endpoint, int size, std::string datatype)
68       : name_(name), endpoint(endpoint), send_size(size), send_type(datatype){};
69   // CollTI: bcast, reduce, allreduce, gather, scatter, allgather, alltoall
70   explicit TIData(std::string name, int root, double amount, int send_size, int recv_size, std::string send_type,
71                   std::string recv_type)
72       : name_(name)
73       , amount_(amount)
74       , endpoint(root)
75       , send_size(send_size)
76       , recv_size(recv_size)
77       , send_type(send_type)
78       , recv_type(recv_type){};
79   // VarCollTI: gatherv, scatterv, allgatherv, alltoallv (+ reducescatter out of laziness)
80   explicit TIData(std::string name, int root, int send_size, std::vector<int>* sendcounts, int recv_size,
81                   std::vector<int>* recvcounts, std::string send_type, std::string recv_type)
82       : TIData(name, root, send_size, std::shared_ptr<std::vector<int>>(sendcounts), recv_size,
83                std::shared_ptr<std::vector<int>>(recvcounts), send_type, recv_type){};
84
85   explicit TIData(std::string name, int root, int send_size, std::shared_ptr<std::vector<int>> sendcounts,
86                   int recv_size, std::shared_ptr<std::vector<int>> recvcounts, std::string send_type,
87                   std::string recv_type)
88       : name_(name)
89       , endpoint(root)
90       , send_size(send_size)
91       , sendcounts(sendcounts)
92       , recv_size(recv_size)
93       , recvcounts(recvcounts)
94       , send_type(send_type)
95       , recv_type(recv_type){};
96
97   virtual ~TIData() {}
98
99   std::string getName() { return name_; }
100   double getAmount() { return amount_; }
101   virtual std::string print()        = 0;
102   virtual std::string display_size() = 0;
103 };
104
105 class NoOpTIData : public TIData {
106 public:
107   explicit NoOpTIData(std::string name) : TIData(name){};
108   std::string print() override { return getName(); }
109   std::string display_size() override { return ""; }
110 };
111
112 class CpuTIData : public TIData {
113 public:
114   explicit CpuTIData(std::string name, double amount) : TIData(name, amount){};
115   std::string print() override
116   {
117     std::stringstream stream;
118     stream << getName() << " " << getAmount();
119     return stream.str();
120   }
121   std::string display_size() override { return std::to_string(getAmount()); }
122 };
123
124 class Pt2PtTIData : public TIData {
125   int tag;
126 public:
127   explicit Pt2PtTIData(std::string name, int endpoint, int size, int tag, std::string datatype)
128       : TIData(name, endpoint, size, datatype), tag(tag) {};
129
130   explicit Pt2PtTIData(std::string name, int endpoint, int size, std::string datatype)
131       : TIData(name, endpoint, size, datatype), tag(0) {};
132   std::string print() override
133   {
134     std::stringstream stream;
135     stream << getName() << " ";
136     if (endpoint >= 0)
137       stream << endpoint << " ";
138     stream << tag << " " << send_size << " " << send_type;
139     return stream.str();
140   }
141   std::string display_size() override { return std::to_string(send_size); }
142 };
143
144 class CollTIData : public TIData {
145 public:
146   explicit CollTIData(std::string name, int root, double amount, int send_size, int recv_size, std::string send_type,
147                       std::string recv_type)
148       : TIData(name, root, amount, send_size, recv_size, send_type, recv_type){};
149   std::string print() override
150   {
151     std::stringstream stream;
152     stream << getName() << " " << send_size << " ";
153     if (recv_size >= 0)
154       stream << recv_size << " ";
155     if (getAmount() >= 0.0)
156       stream << getAmount() << " ";
157     if (endpoint > 0 || (endpoint == 0 && not send_type.empty()))
158       stream << endpoint << " ";
159     stream << send_type << " " << recv_type;
160
161     return stream.str();
162   }
163   std::string display_size() override { return std::to_string(send_size); }
164 };
165
166 class VarCollTIData : public TIData {
167 public:
168   explicit VarCollTIData(std::string name, int root, int send_size, std::vector<int>* sendcounts, int recv_size,
169                          std::vector<int>* recvcounts, std::string send_type, std::string recv_type)
170       : TIData(name, root, send_size, sendcounts, recv_size, recvcounts, send_type, recv_type){};
171
172   explicit VarCollTIData(std::string name, int root, int send_size, std::shared_ptr<std::vector<int>> sendcounts,
173                          int recv_size, std::shared_ptr<std::vector<int>> recvcounts, std::string send_type,
174                          std::string recv_type)
175       : TIData(name, root, send_size, sendcounts, recv_size, recvcounts, send_type, recv_type){};
176
177   std::string print() override
178   {
179     std::stringstream stream;
180     stream << getName() << " ";
181     if (send_size >= 0)
182       stream << send_size << " ";
183     if (sendcounts != nullptr)
184       for (auto count : *sendcounts)
185         stream << count << " ";
186     if (recv_size >= 0)
187       stream << recv_size << " ";
188     if (recvcounts != nullptr)
189       for (auto count : *recvcounts)
190         stream << count << " ";
191     if (endpoint > 0 || (endpoint == 0 && not send_type.empty()))
192       stream << endpoint << " ";
193     stream << send_type << " " << recv_type;
194
195     return stream.str();
196   }
197   std::string display_size() override { return std::to_string(send_size > 0 ? send_size : recv_size); }
198 };
199
200 /**
201  * If we want to wait for a request of asynchronous communication, we need to be able
202  * to identify this request. We do this by searching for a request identified by (src, dest, tag).
203  */
204 class WaitTIData : public TIData {
205   int src;
206   int dest;
207   int tag;
208
209 public:
210   explicit WaitTIData(int src, int dest, int tag) : TIData("wait"), src(src), dest(dest), tag(tag){};
211
212   std::string print() override
213   {
214     std::stringstream stream;
215     stream << getName() << " " << src << " " << dest << " " << tag;
216
217     return stream.str();
218   }
219
220   std::string display_size() override { return ""; }
221 };
222
223 class AmpiMigrateTIData : public TIData {
224   size_t memory_consumption;
225 public:
226   explicit AmpiMigrateTIData(size_t memory_conso) : TIData("migrate"), memory_consumption(memory_conso) { };
227
228   std::string print() override
229   {
230     std::stringstream stream;
231     stream << getName() << " " << memory_consumption;
232
233     return stream.str();
234   }
235
236   std::string display_size() override { return ""; }
237 };
238 }
239 }
240
241 XBT_PRIVATE std::string instr_pid(s4u_Actor* proc);
242
243 extern XBT_PRIVATE std::set<std::string> created_categories;
244 extern XBT_PRIVATE std::set<std::string> declared_marks;
245 extern XBT_PRIVATE std::set<std::string> user_host_variables;
246 extern XBT_PRIVATE std::set<std::string> user_vm_variables;
247 extern XBT_PRIVATE std::set<std::string> user_link_variables;
248 extern XBT_PRIVATE double TRACE_last_timestamp_to_dump;
249
250 /* instr_paje_header.c */
251 XBT_PRIVATE void TRACE_header(bool basic, bool size);
252
253 /* from instr_config.c */
254 XBT_PRIVATE bool TRACE_needs_platform();
255 XBT_PRIVATE bool TRACE_is_enabled();
256 XBT_PRIVATE bool TRACE_platform();
257 XBT_PRIVATE bool TRACE_platform_topology();
258 XBT_PRIVATE bool TRACE_categorized();
259 XBT_PRIVATE bool TRACE_uncategorized();
260 XBT_PRIVATE bool TRACE_actor_is_enabled();
261 XBT_PRIVATE bool TRACE_vm_is_enabled();
262 XBT_PRIVATE bool TRACE_disable_link();
263 XBT_PRIVATE bool TRACE_disable_speed();
264 XBT_PRIVATE bool TRACE_disable_destroy();
265 XBT_PRIVATE bool TRACE_basic();
266 XBT_PRIVATE bool TRACE_display_sizes();
267 XBT_PRIVATE int TRACE_precision();
268
269 /* Public functions used in SMPI */
270 XBT_PUBLIC bool TRACE_smpi_is_enabled();
271 XBT_PUBLIC bool TRACE_smpi_is_grouped();
272 XBT_PUBLIC bool TRACE_smpi_is_computing();
273 XBT_PUBLIC bool TRACE_smpi_is_sleeping();
274 XBT_PUBLIC bool TRACE_smpi_view_internals();
275
276 XBT_PRIVATE void TRACE_surf_resource_set_utilization(const char* type, const char* name, const char* resource,
277                                                      std::string category, double value, double now, double delta);
278
279 /* instr_paje.c */
280 extern XBT_PRIVATE std::set<std::string> trivaNodeTypes;
281 extern XBT_PRIVATE std::set<std::string> trivaEdgeTypes;
282 XBT_PRIVATE long long int instr_new_paje_id();
283 XBT_PRIVATE void instr_define_callbacks();
284 void instr_new_variable_type(std::string new_typename, std::string color);
285 void instr_new_user_variable_type(std::string father_type, std::string new_typename, std::string color);
286 void instr_new_user_state_type(std::string father_type, std::string new_typename);
287 void instr_new_value_for_user_state_type(std::string new_typename, const char* value, std::string color);
288
289 XBT_PRIVATE void TRACE_global_init();
290 XBT_PRIVATE void TRACE_help();
291
292 XBT_PRIVATE void TRACE_paje_dump_buffer(bool force);
293 XBT_PRIVATE void dump_comment_file(std::string filename);
294 XBT_PRIVATE void dump_comment(std::string comment);
295
296 XBT_PRIVATE std::string TRACE_get_filename();
297
298 /* instr_platform */
299 xbt_graph_t instr_routing_platform_graph();
300 void instr_routing_platform_graph_export_graphviz(xbt_graph_t g, const char* filename);
301
302 #endif