src/smpi/smpi_replay_main.cpp

   1 /* Copyright (c) 2018-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/s4u.hpp"
   7 #include "smpi/smpi.h"
   8 #include "xbt/asserts.h"
   9 #include "xbt/replay.hpp"
  10 #include "xbt/str.h"
  11
  12 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(smpi_replay);
  13
  14 int main(int argc, char* argv[])
  15 {
  16   if (simgrid::s4u::Actor::self() == nullptr) {
  17     XBT_ERROR("smpireplaymain should not be called directly. Please use smpirun -replay instead.");
  18     return 1;
  19   }
  20
  21   auto properties = simgrid::s4u::Actor::self()->get_properties();
  22   if (properties->find("smpi_replay") == properties->end()) {
  23     XBT_ERROR("invalid smpireplaymain execution. Please use smpirun -replay instead.");
  24     return 1;
  25   }
  26
  27   const char* instance_id = properties->at("instance_id").c_str();
  28   const int rank          = xbt_str_parse_int(properties->at("rank").c_str(), "Cannot parse rank");
  29   const char* shared_trace =
  30       simgrid::s4u::Actor::self()->get_property("tracefile"); // Cannot use properties because this can be nullptr
  31   const char* private_trace = argv[1];
  32   double start_delay_flops  = 0;
  33
  34   if (argc > 2) {
  35     start_delay_flops = xbt_str_parse_double(argv[2], "Cannot parse start_delay_flops");
  36   }
  37
  38   /* Setup things and register default actions */
  39   smpi_replay_init(instance_id, rank, start_delay_flops);
  40
  41   /* A small check, just to make sure SMPI ranks are consistent with input arguments */
  42   int new_rank;
  43   MPI_Comm_rank(MPI_COMM_WORLD, &new_rank);
  44   xbt_assert(new_rank == rank, "Rank inconsistency. Got %d, expected %d", new_rank, rank);
  45
  46   /* The regular run of the replayer */
  47   if (shared_trace)
  48     xbt_replay_set_tracefile(shared_trace);
  49   smpi_replay_main(rank, private_trace);
  50   return 0;
  51 }