Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
[SMPI] Replay: Make start_time a local variable
[simgrid.git] / src / smpi / internals / smpi_replay.cpp
index 8067fd4..beb0191 100644 (file)
@@ -39,10 +39,11 @@ static MPI_Datatype MPI_DEFAULT_TYPE;
              static_cast<unsigned long>(optional));                                                                    \
   }
 
-static void log_timed_action (simgrid::xbt::ReplayAction& action, double clock){
+static void log_timed_action(simgrid::xbt::ReplayAction& action, double clock)
+{
   if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
     std::string s = boost::algorithm::join(action, " ");
-    XBT_VERB("%s %f", s.c_str(), smpi_process()->simulated_elapsed()-clock);
+    XBT_VERB("%s %f", s.c_str(), smpi_process()->simulated_elapsed() - clock);
   }
 }
 
@@ -71,26 +72,51 @@ public:
   virtual void parse(simgrid::xbt::ReplayAction& action){};
 };
 
-template<class T> class ReplayAction {
-  protected:
+class SendRecvParser : public ActionArgParser {
+public:
+  /* communication partner; if we send, this is the receiver and vice versa */
+  int partner;
+  double size;
+  MPI_Datatype datatype1 = MPI_DEFAULT_TYPE;
+
+  void parse(simgrid::xbt::ReplayAction& action) override
+  {
+    CHECK_ACTION_PARAMS(action, 2, 1)
+    partner = std::stoi(action[2]);
+    size    = parse_double(action[3]);
+    if (action.size() > 4)
+      datatype1 = simgrid::smpi::Datatype::decode(action[4]);
+  }
+};
+
+class ComputeParser : public ActionArgParser {
+public:
+  /* communication partner; if we send, this is the receiver and vice versa */
+  double flops;
+
+  void parse(simgrid::xbt::ReplayAction& action) override
+  {
+    CHECK_ACTION_PARAMS(action, 1, 0)
+    flops = parse_double(action[2]);
+  }
+};
+
+template <class T> class ReplayAction {
+protected:
   const std::string name;
   T args;
 
-  /*
-   * Used to compute the duration of this action.
-   */
-  double start_time;
-
   int my_proc_id;
 
 public:
-  explicit ReplayAction(std::string name)
-      : name(name), start_time(smpi_process()->simulated_elapsed()), my_proc_id(simgrid::s4u::Actor::self()->getPid())
+  explicit ReplayAction(std::string name) : name(name), my_proc_id(simgrid::s4u::Actor::self()->getPid())
   {
   }
 
   virtual void execute(simgrid::xbt::ReplayAction& action)
   {
+    // Needs to be re-initialized for every action, hence here
+    double start_time = smpi_process()->simulated_elapsed();
     args.parse(action);
     kernel(action);
     log_timed_action(action, start_time);
@@ -135,6 +161,78 @@ public:
   }
 };
 
+class SendAction : public ReplayAction<SendRecvParser> {
+public:
+  SendAction() = delete;
+  SendAction(std::string name) : ReplayAction(name) {}
+  void kernel(simgrid::xbt::ReplayAction& action) override
+  {
+    int dst_traced = MPI_COMM_WORLD->group()->actor(args.partner)->getPid();
+
+    TRACE_smpi_comm_in(my_proc_id, __FUNCTION__, new simgrid::instr::Pt2PtTIData(name, args.partner, args.size,
+                                                                                 Datatype::encode(args.datatype1)));
+    if (not TRACE_smpi_view_internals())
+      TRACE_smpi_send(my_proc_id, my_proc_id, dst_traced, 0, args.size * args.datatype1->size());
+
+    if (name == "send") {
+      Request::send(nullptr, args.size, args.datatype1, args.partner, 0, MPI_COMM_WORLD);
+    } else if (name == "Isend") {
+      MPI_Request request = Request::isend(nullptr, args.size, args.datatype1, args.partner, 0, MPI_COMM_WORLD);
+      get_reqq_self()->push_back(request);
+    } else {
+      xbt_die("Don't know this action, %s", name.c_str());
+    }
+
+    TRACE_smpi_comm_out(my_proc_id);
+  }
+};
+
+class RecvAction : public ReplayAction<SendRecvParser> {
+public:
+  RecvAction() = delete;
+  explicit RecvAction(std::string name) : ReplayAction(name) {}
+  void kernel(simgrid::xbt::ReplayAction& action) override
+  {
+    int src_traced = MPI_COMM_WORLD->group()->actor(args.partner)->getPid();
+
+    TRACE_smpi_comm_in(my_proc_id, __FUNCTION__, new simgrid::instr::Pt2PtTIData(name, args.partner, args.size,
+                                                                                 Datatype::encode(args.datatype1)));
+
+    MPI_Status status;
+    // unknown size from the receiver point of view
+    if (args.size <= 0.0) {
+      Request::probe(args.partner, 0, MPI_COMM_WORLD, &status);
+      args.size = status.count;
+    }
+
+    if (name == "recv") {
+      Request::recv(nullptr, args.size, args.datatype1, args.partner, 0, MPI_COMM_WORLD, &status);
+    } else if (name == "Irecv") {
+      MPI_Request request = Request::irecv(nullptr, args.size, args.datatype1, args.partner, 0, MPI_COMM_WORLD);
+      get_reqq_self()->push_back(request);
+    }
+
+    TRACE_smpi_comm_out(my_proc_id);
+    // TODO: Check why this was only activated in the "recv" case and not in the "Irecv" case
+    if (name == "recv" && not TRACE_smpi_view_internals()) {
+      TRACE_smpi_recv(src_traced, my_proc_id, 0);
+    }
+  }
+};
+
+class ComputeAction : public ReplayAction<ComputeParser> {
+public:
+  ComputeAction() : ReplayAction("compute") {}
+  void kernel(simgrid::xbt::ReplayAction& action) override
+  {
+    TRACE_smpi_computing_in(my_proc_id, args.flops);
+    smpi_execute_flops(args.flops);
+    TRACE_smpi_computing_out(my_proc_id);
+  }
+};
+
+} // Replay Namespace
+
 static void action_init(simgrid::xbt::ReplayAction& action)
 {
   XBT_DEBUG("Initialize the counters");
@@ -175,124 +273,7 @@ static void action_comm_dup(simgrid::xbt::ReplayAction& action)
 
 static void action_compute(simgrid::xbt::ReplayAction& action)
 {
-  CHECK_ACTION_PARAMS(action, 1, 0)
-  double clock = smpi_process()->simulated_elapsed();
-  double flops= parse_double(action[2]);
-  int my_proc_id = Actor::self()->getPid();
-
-  TRACE_smpi_computing_in(my_proc_id, flops);
-  smpi_execute_flops(flops);
-  TRACE_smpi_computing_out(my_proc_id);
-
-  log_timed_action (action, clock);
-}
-
-static void action_send(simgrid::xbt::ReplayAction& action)
-{
-  CHECK_ACTION_PARAMS(action, 2, 1)
-  int to       = std::stoi(action[2]);
-  double size=parse_double(action[3]);
-  double clock = smpi_process()->simulated_elapsed();
-
-  MPI_Datatype MPI_CURRENT_TYPE = (action.size() > 4) ? simgrid::smpi::Datatype::decode(action[4]) : MPI_DEFAULT_TYPE;
-
-  int my_proc_id = Actor::self()->getPid();
-  int dst_traced = MPI_COMM_WORLD->group()->actor(to)->getPid();
-
-  TRACE_smpi_comm_in(my_proc_id, __FUNCTION__,
-                     new simgrid::instr::Pt2PtTIData("send", to, size, Datatype::encode(MPI_CURRENT_TYPE)));
-  if (not TRACE_smpi_view_internals())
-    TRACE_smpi_send(my_proc_id, my_proc_id, dst_traced, 0, size * MPI_CURRENT_TYPE->size());
-
-  Request::send(nullptr, size, MPI_CURRENT_TYPE, to , 0, MPI_COMM_WORLD);
-
-  TRACE_smpi_comm_out(my_proc_id);
-
-  log_timed_action(action, clock);
-}
-
-static void action_Isend(simgrid::xbt::ReplayAction& action)
-{
-  CHECK_ACTION_PARAMS(action, 2, 1)
-  int to       = std::stoi(action[2]);
-  double size=parse_double(action[3]);
-  double clock = smpi_process()->simulated_elapsed();
-
-  MPI_Datatype MPI_CURRENT_TYPE = (action.size() > 4) ? simgrid::smpi::Datatype::decode(action[4]) : MPI_DEFAULT_TYPE;
-
-  int my_proc_id = Actor::self()->getPid();
-  int dst_traced = MPI_COMM_WORLD->group()->actor(to)->getPid();
-  TRACE_smpi_comm_in(my_proc_id, __FUNCTION__,
-                     new simgrid::instr::Pt2PtTIData("Isend", to, size, Datatype::encode(MPI_CURRENT_TYPE)));
-  if (not TRACE_smpi_view_internals())
-    TRACE_smpi_send(my_proc_id, my_proc_id, dst_traced, 0, size * MPI_CURRENT_TYPE->size());
-
-  MPI_Request request = Request::isend(nullptr, size, MPI_CURRENT_TYPE, to, 0, MPI_COMM_WORLD);
-
-  TRACE_smpi_comm_out(my_proc_id);
-
-  get_reqq_self()->push_back(request);
-
-  log_timed_action (action, clock);
-}
-
-static void action_recv(simgrid::xbt::ReplayAction& action)
-{
-  CHECK_ACTION_PARAMS(action, 2, 1)
-  int from     = std::stoi(action[2]);
-  double size=parse_double(action[3]);
-  double clock = smpi_process()->simulated_elapsed();
-  MPI_Status status;
-
-  MPI_Datatype MPI_CURRENT_TYPE = (action.size() > 4) ? simgrid::smpi::Datatype::decode(action[4]) : MPI_DEFAULT_TYPE;
-
-  int my_proc_id = Actor::self()->getPid();
-  int src_traced = MPI_COMM_WORLD->group()->actor(from)->getPid();
-
-  TRACE_smpi_comm_in(my_proc_id, __FUNCTION__,
-                     new simgrid::instr::Pt2PtTIData("recv", from, size, Datatype::encode(MPI_CURRENT_TYPE)));
-
-  //unknown size from the receiver point of view
-  if (size <= 0.0) {
-    Request::probe(from, 0, MPI_COMM_WORLD, &status);
-    size=status.count;
-  }
-
-  Request::recv(nullptr, size, MPI_CURRENT_TYPE, from, 0, MPI_COMM_WORLD, &status);
-
-  TRACE_smpi_comm_out(my_proc_id);
-  if (not TRACE_smpi_view_internals()) {
-    TRACE_smpi_recv(src_traced, my_proc_id, 0);
-  }
-
-  log_timed_action (action, clock);
-}
-
-static void action_Irecv(simgrid::xbt::ReplayAction& action)
-{
-  CHECK_ACTION_PARAMS(action, 2, 1)
-  int from     = std::stoi(action[2]);
-  double size=parse_double(action[3]);
-  double clock = smpi_process()->simulated_elapsed();
-
-  MPI_Datatype MPI_CURRENT_TYPE = (action.size() > 4) ? simgrid::smpi::Datatype::decode(action[4]) : MPI_DEFAULT_TYPE;
-
-  int my_proc_id = Actor::self()->getPid();
-  TRACE_smpi_comm_in(my_proc_id, __FUNCTION__,
-                     new simgrid::instr::Pt2PtTIData("Irecv", from, size, Datatype::encode(MPI_CURRENT_TYPE)));
-  MPI_Status status;
-  //unknow size from the receiver pov
-  if (size <= 0.0) {
-    Request::probe(from, 0, MPI_COMM_WORLD, &status);
-    size = status.count;
-  }
-
-  MPI_Request request = Request::irecv(nullptr, size, MPI_CURRENT_TYPE, from, 0, MPI_COMM_WORLD);
-
-  TRACE_smpi_comm_out(my_proc_id);
-  get_reqq_self()->push_back(request);
-
-  log_timed_action (action, clock);
+  Replay::ComputeAction().execute(action);
 }
 
 static void action_test(simgrid::xbt::ReplayAction& action)
@@ -321,11 +302,6 @@ static void action_test(simgrid::xbt::ReplayAction& action)
   log_timed_action (action, clock);
 }
 
-static void action_wait(simgrid::xbt::ReplayAction& action)
-{
-  Replay::WaitAction().execute(action);
-}
-
 static void action_waitall(simgrid::xbt::ReplayAction& action)
 {
   CHECK_ACTION_PARAMS(action, 0, 0)
@@ -836,12 +812,24 @@ void smpi_replay_init(int* argc, char*** argv)
   xbt_replay_action_register("comm_size",  simgrid::smpi::action_comm_size);
   xbt_replay_action_register("comm_split", simgrid::smpi::action_comm_split);
   xbt_replay_action_register("comm_dup",   simgrid::smpi::action_comm_dup);
-  xbt_replay_action_register("send",       simgrid::smpi::action_send);
-  xbt_replay_action_register("Isend",      simgrid::smpi::action_Isend);
-  xbt_replay_action_register("recv",       simgrid::smpi::action_recv);
-  xbt_replay_action_register("Irecv",      simgrid::smpi::action_Irecv);
-  xbt_replay_action_register("test",       simgrid::smpi::action_test);
-  xbt_replay_action_register("wait",       simgrid::smpi::action_wait);
+
+  std::shared_ptr<simgrid::smpi::Replay::SendAction> isend(new simgrid::smpi::Replay::SendAction("Isend"));
+  std::shared_ptr<simgrid::smpi::Replay::SendAction> send(new simgrid::smpi::Replay::SendAction("send"));
+  std::shared_ptr<simgrid::smpi::Replay::RecvAction> irecv(new simgrid::smpi::Replay::RecvAction("Irecv"));
+  std::shared_ptr<simgrid::smpi::Replay::RecvAction> recv(new simgrid::smpi::Replay::RecvAction("recv"));
+  std::shared_ptr<simgrid::smpi::Replay::WaitAction> wait(new simgrid::smpi::Replay::WaitAction());
+
+  xbt_replay_action_register("send",
+                             std::bind(&simgrid::smpi::Replay::SendAction::execute, send, std::placeholders::_1));
+  xbt_replay_action_register("Isend",
+                             std::bind(&simgrid::smpi::Replay::SendAction::execute, isend, std::placeholders::_1));
+  xbt_replay_action_register("recv",
+                             std::bind(&simgrid::smpi::Replay::RecvAction::execute, recv, std::placeholders::_1));
+  xbt_replay_action_register("Irecv",
+                             std::bind(&simgrid::smpi::Replay::RecvAction::execute, irecv, std::placeholders::_1));
+  xbt_replay_action_register("test", simgrid::smpi::action_test);
+  xbt_replay_action_register("wait",
+                             std::bind(&simgrid::smpi::Replay::WaitAction::execute, wait, std::placeholders::_1));
   xbt_replay_action_register("waitAll",    simgrid::smpi::action_waitall);
   xbt_replay_action_register("barrier",    simgrid::smpi::action_barrier);
   xbt_replay_action_register("bcast",      simgrid::smpi::action_bcast);
@@ -856,7 +844,7 @@ void smpi_replay_init(int* argc, char*** argv)
   xbt_replay_action_register("allGather",  simgrid::smpi::action_allgather);
   xbt_replay_action_register("allGatherV", simgrid::smpi::action_allgatherv);
   xbt_replay_action_register("reduceScatter",  simgrid::smpi::action_reducescatter);
-  xbt_replay_action_register("compute",    simgrid::smpi::action_compute);
+  xbt_replay_action_register("compute", simgrid::smpi::action_compute);
 
   //if we have a delayed start, sleep here.
   if(*argc>2){