Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
plug leaks
[simgrid.git] / src / smpi / smpi_replay_main.cpp
1 #include "simgrid/s4u.hpp"
2 #include "smpi/smpi.h"
3 #include "xbt/asserts.h"
4 #include "xbt/replay.hpp"
5 #include "xbt/str.h"
6
7 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(smpi_replay);
8
9 int main(int argc, char* argv[])
10 {
11   if (simgrid::s4u::Actor::self() == nullptr) {
12     XBT_ERROR("smpireplaymain should not be called directly. Please use smpirun -replay instead.");
13     return 1;
14   }
15
16   auto properties = simgrid::s4u::Actor::self()->get_properties();
17   if (properties->find("smpi_replay") == properties->end()) {
18     XBT_ERROR("invalid smpireplaymain execution. Please use smpirun -replay instead.");
19     return 1;
20   }
21
22   const char* instance_id    = properties->at("instance_id").c_str();
23   const int rank             = xbt_str_parse_int(properties->at("rank").c_str(), "Cannot parse rank");
24   const char* trace_filename = argv[1];
25   double start_delay_flops   = 0;
26
27   if (argc > 2) {
28     start_delay_flops = xbt_str_parse_double(argv[2], "Cannot parse start_delay_flops");
29   }
30
31   /* Setup things and register default actions */
32   smpi_replay_init(instance_id, rank, start_delay_flops);
33
34   /* A small check, just to make sure SMPI ranks are consistent with input arguments */
35   int new_rank;
36   MPI_Comm_rank(MPI_COMM_WORLD, &new_rank);
37   xbt_assert(new_rank == rank, "Rank inconsistency. Got %d, expected %d", new_rank, rank);
38
39   /* The regular run of the replayer */
40   smpi_replay_main(rank, trace_filename);
41   return 0;
42 }