1 /* Copyright (c) 2009, 2010, 2011, 2012. The SimGrid Team.
2 * All rights reserved. */
4 /* This program is free software; you can redistribute it and/or modify it
5 * under the terms of the license (GNU LGPL) which comes with this package. */
11 #include <xbt/replay.h>
13 XBT_LOG_NEW_DEFAULT_CATEGORY(smpi_replay,"Trace Replay with SMPI");
16 static double parse_double(const char *string)
20 value = strtod(string, &endptr);
22 THROWF(unknown_error, 0, "%s is not a double", string);
26 static void action_compute(const char *const *action)
28 XBT_DEBUG("Start to compute %.0f flops", parse_double(action[2]));
29 smpi_sample_flops(parse_double(action[2]));
32 static void action_send(const char *const *action)
34 int to = atoi(action[2]);
35 const char *size_str = action[3];
36 double size=parse_double(size_str);
37 char *buffer = (char*) malloc (size);
39 XBT_DEBUG("send %.0f bytes to rank%d (%s)",size, to, action[2]);
41 MPI_Send(buffer, size, MPI_BYTE, to, 0, MPI_COMM_WORLD);
46 static void action_Isend(const char *const *action)
48 int to = atoi(action[2]);
49 const char *size_str = action[3];
50 double size=parse_double(size_str);
51 char *buffer = (char*) malloc (size);
55 MPI_Isend(buffer, size, MPI_BYTE, to, 0, MPI_COMM_WORLD, &request);
60 static void action_recv(const char *const *action)
62 int from = atoi(action[2]);
63 XBT_DEBUG("receive from rank%d (%s)",from, action[2]);
64 char *buffer = (char*) malloc (1e9);
68 MPI_Recv(buffer, 1e9, MPI_BYTE, from, 0, MPI_COMM_WORLD, &status);
73 int main(int argc, char *argv[])
77 MPI_Init(&argc, &argv);
78 MPI_Comm_rank(MPI_COMM_WORLD, &i);
79 _xbt_replay_action_init();
81 // xbt_replay_action_register("init", action_init);
82 // xbt_replay_action_register("finalize", action_finalize);
83 // xbt_replay_action_register("comm_size",action_comm_size);
84 xbt_replay_action_register("send", action_send);
85 xbt_replay_action_register("Isend", action_Isend);
86 xbt_replay_action_register("recv", action_recv);
87 // xbt_replay_action_register("Irecv", action_Irecv);
88 // xbt_replay_action_register("wait", action_wait);
89 // xbt_replay_action_register("barrier", action_barrier);
90 // xbt_replay_action_register("bcast", action_bcast);
91 // xbt_replay_action_register("reduce", action_reduce);
92 // xbt_replay_action_register("allReduce",action_allReduce);
93 // xbt_replay_action_register("sleep", action_sleep);
94 xbt_replay_action_register("compute", action_compute);
96 xbt_replay_action_runner(argc, argv);
97 /* Actually do the simulation using MSG_action_trace_run */
98 smpi_action_trace_run(argv[1]); // it's ok to pass a NULL argument here
100 XBT_DEBUG("rank%d: I'm done!",i);