XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_replay,smpi,"Trace Replay with SMPI");
+int communicator_size = 0;
+
+typedef struct {
+ xbt_dynar_t isends; /* of MPI_Request */
+ xbt_dynar_t irecvs; /* of MPI_Request */
+} s_smpi_replay_globals_t, *smpi_replay_globals_t;
+
/* Helper function */
static double parse_double(const char *string)
{
return value;
}
+static void action_init(const char *const *action)
+{
+ XBT_DEBUG("Initialize the counters");
+ smpi_replay_globals_t globals = xbt_new(s_smpi_replay_globals_t, 1);
+ globals->isends = xbt_dynar_new(sizeof(MPI_Request),NULL);
+ globals->irecvs = xbt_dynar_new(sizeof(MPI_Request),NULL);
+
+ smpi_process_set_user_data((void*) globals);
+
+ /* start a simulated timer */
+ smpi_process_simulated_start();
+}
+
+static void action_finalize(const char *const *action)
+{
+ double sim_time= 1.;
+ smpi_replay_globals_t globals =
+ (smpi_replay_globals_t) smpi_process_get_user_data();
+
+ if (globals){
+ XBT_DEBUG("There are %lu isends and %lu irecvs in the dynars",
+ xbt_dynar_length(globals->isends),xbt_dynar_length(globals->irecvs));
+ xbt_dynar_free_container(&(globals->isends));
+ xbt_dynar_free_container(&(globals->irecvs));
+ }
+ free(globals);
+ /* end the simulated timer */
+ sim_time = smpi_process_simulated_elapsed();
+ if (!smpi_process_index())
+ XBT_INFO("Simulation time %g", sim_time);
+ smpi_process_finalize();
+ smpi_process_destroy();
+}
+
+static void action_comm_size(const char *const *action)
+{
+ double clock = smpi_process_simulated_elapsed();
+
+ communicator_size = parse_double(action[2]);
+
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
+}
+
+
static void action_compute(const char *const *action)
{
- XBT_DEBUG("Start to compute %.0f flops", parse_double(action[2]));
+ double clock = smpi_process_simulated_elapsed();
smpi_execute_flops(parse_double(action[2]));
+
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
}
static void action_send(const char *const *action)
{
int to = atoi(action[2]);
double size=parse_double(action[3]);
-
- XBT_DEBUG("send %.0f bytes to rank%d (%s)",size, to, action[2]);
+ double clock = smpi_process_simulated_elapsed();
+#ifdef HAVE_TRACING
+ int rank = smpi_comm_rank(MPI_COMM_WORLD);
+ TRACE_smpi_computing_out(rank);
+ int dst_traced = smpi_group_rank(smpi_comm_group(MPI_COMM_WORLD), to);
+ TRACE_smpi_ptp_in(rank, rank, dst_traced, __FUNCTION__);
+ TRACE_smpi_send(rank, rank, dst_traced);
+#endif
smpi_mpi_send(NULL, size, MPI_BYTE, to , 0, MPI_COMM_WORLD);
+
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
+
+ #ifdef HAVE_TRACING
+ TRACE_smpi_ptp_out(rank, rank, dst_traced, __FUNCTION__);
+ TRACE_smpi_computing_in(rank);
+#endif
+
}
static void action_Isend(const char *const *action)
{
int to = atoi(action[2]);
double size=parse_double(action[3]);
+ double clock = smpi_process_simulated_elapsed();
+ smpi_replay_globals_t globals =
+ (smpi_replay_globals_t) smpi_process_get_user_data();
+#ifdef HAVE_TRACING
+ int rank = smpi_comm_rank(MPI_COMM_WORLD);
+ TRACE_smpi_computing_out(rank);
+ int dst_traced = smpi_group_rank(smpi_comm_group(MPI_COMM_WORLD), to);
+ TRACE_smpi_ptp_in(rank, rank, dst_traced, __FUNCTION__);
+ TRACE_smpi_send(rank, rank, dst_traced);
+#endif
MPI_Request request = smpi_mpi_isend(NULL, size, MPI_BYTE, to, 0,
MPI_COMM_WORLD);
+#ifdef HAVE_TRACING
+ TRACE_smpi_ptp_out(rank, rank, dst_traced, __FUNCTION__);
+ request->send = 1;
+ TRACE_smpi_computing_in(rank);
+#endif
- //TODO do something with request
- request = NULL;
+ xbt_dynar_push(globals->isends,&request);
+
+ //TODO do the asynchronous cleanup
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
}
static void action_recv(const char *const *action) {
int from = atoi(action[2]);
- XBT_DEBUG("receive from rank%d (%s)",from, action[2]);
-
+ double size=parse_double(action[3]);
+ double clock = smpi_process_simulated_elapsed();
MPI_Status status;
- //TODO find a way to get the message size
- smpi_mpi_recv(NULL, 1e9, MPI_BYTE, from, 0, MPI_COMM_WORLD, &status);
+#ifdef HAVE_TRACING
+ int rank = smpi_comm_rank(MPI_COMM_WORLD);
+ int src_traced = smpi_group_rank(smpi_comm_group(MPI_COMM_WORLD), from);
+ TRACE_smpi_computing_out(rank);
+
+ TRACE_smpi_ptp_in(rank, src_traced, rank, __FUNCTION__);
+#endif
+
+ smpi_mpi_recv(NULL, size, MPI_BYTE, from, 0, MPI_COMM_WORLD, &status);
+
+#ifdef HAVE_TRACING
+ TRACE_smpi_ptp_out(rank, src_traced, rank, __FUNCTION__);
+ TRACE_smpi_recv(rank, src_traced, rank);
+ TRACE_smpi_computing_in(rank);
+#endif
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
}
static void action_Irecv(const char *const *action)
{
int from = atoi(action[2]);
+ double size=parse_double(action[3]);
+ double clock = smpi_process_simulated_elapsed();
MPI_Request request;
+ smpi_replay_globals_t globals =
+ (smpi_replay_globals_t) smpi_process_get_user_data();
- XBT_DEBUG("Asynchronous receive from rank%d (%s)",from, action[2]);
+#ifdef HAVE_TRACING
+ int rank = smpi_comm_rank(MPI_COMM_WORLD);
+ int src_traced = smpi_group_rank(smpi_comm_group(MPI_COMM_WORLD), from);
+ TRACE_smpi_ptp_in(rank, src_traced, rank, __FUNCTION__);
+#endif
- //TODO find a way to get the message size
- request = smpi_mpi_irecv(NULL, 1e9, MPI_BYTE, from, 0, MPI_COMM_WORLD);
+ request = smpi_mpi_irecv(NULL, size, MPI_BYTE, from, 0, MPI_COMM_WORLD);
+#ifdef HAVE_TRACING
+ TRACE_smpi_ptp_out(rank, src_traced, rank, __FUNCTION__);
+ request->recv = 1;
+#endif
+ xbt_dynar_push(globals->irecvs,&request);
- //TODO do something with request
- request = NULL;
+ //TODO do the asynchronous cleanup
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
}
static void action_wait(const char *const *action){
+ double clock = smpi_process_simulated_elapsed();
MPI_Request request;
MPI_Status status;
+ smpi_replay_globals_t globals =
+ (smpi_replay_globals_t) smpi_process_get_user_data();
+
+ xbt_assert(xbt_dynar_length(globals->irecvs),
+ "action wait not preceded by any irecv: %s",
+ xbt_str_join_array(action," "));
+ request = xbt_dynar_pop_as(globals->irecvs,MPI_Request);
+#ifdef HAVE_TRACING
+ int rank = request && request->comm != MPI_COMM_NULL
+ ? smpi_comm_rank(request->comm)
+ : -1;
+ TRACE_smpi_computing_out(rank);
+ MPI_Group group = smpi_comm_group(request->comm);
+ int src_traced = smpi_group_rank(group, request->src);
+ int dst_traced = smpi_group_rank(group, request->dst);
+ int is_wait_for_receive = request->recv;
+ TRACE_smpi_ptp_in(rank, src_traced, dst_traced, __FUNCTION__);
+#endif
smpi_mpi_wait(&request, &status);
+#ifdef HAVE_TRACING
+ TRACE_smpi_ptp_out(rank, src_traced, dst_traced, __FUNCTION__);
+ if (is_wait_for_receive) {
+ TRACE_smpi_recv(rank, src_traced, dst_traced);
+ }
+ TRACE_smpi_computing_in(rank);
+#endif
+
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
}
static void action_barrier(const char *const *action){
+ double clock = smpi_process_simulated_elapsed();
+#ifdef HAVE_TRACING
+ int rank = smpi_comm_rank(MPI_COMM_WORLD);
+ TRACE_smpi_computing_out(rank);
+ TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
+#endif
smpi_mpi_barrier(MPI_COMM_WORLD);
+#ifdef HAVE_TRACING
+ TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
+ TRACE_smpi_computing_in(rank);
+#endif
+
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
+}
+
+static void action_bcast(const char *const *action)
+{
+ double size = parse_double(action[2]);
+ double clock = smpi_process_simulated_elapsed();
+#ifdef HAVE_TRACING
+ int rank = smpi_comm_rank(MPI_COMM_WORLD);
+ TRACE_smpi_computing_out(rank);
+ int root_traced = smpi_group_rank(smpi_comm_group(MPI_COMM_WORLD), 0);
+ TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
+#endif
+
+ smpi_mpi_bcast(NULL, size, MPI_BYTE, 0, MPI_COMM_WORLD);
+#ifdef HAVE_TRACING
+ TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
+ TRACE_smpi_computing_in(rank);
+#endif
+
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
+}
+
+static void action_reduce(const char *const *action)
+{
+ double size = parse_double(action[2]);
+ double clock = smpi_process_simulated_elapsed();
+#ifdef HAVE_TRACING
+ int rank = smpi_comm_rank(MPI_COMM_WORLD);
+ TRACE_smpi_computing_out(rank);
+ int root_traced = smpi_group_rank(smpi_comm_group(MPI_COMM_WORLD), 0);
+ TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
+#endif
+ smpi_mpi_reduce(NULL, NULL, size, MPI_BYTE, MPI_OP_NULL, 0, MPI_COMM_WORLD);
+#ifdef HAVE_TRACING
+ TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
+ TRACE_smpi_computing_in(rank);
+#endif
+
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
+}
+
+static void action_allReduce(const char *const *action) {
+ double comm_size = parse_double(action[2]);
+ double comp_size = parse_double(action[3]);
+ double clock = smpi_process_simulated_elapsed();
+#ifdef HAVE_TRACING
+ int rank = smpi_comm_rank(MPI_COMM_WORLD);
+ TRACE_smpi_computing_out(rank);
+ TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
+#endif
+ smpi_mpi_reduce(NULL, NULL, comm_size, MPI_BYTE, MPI_OP_NULL, 0, MPI_COMM_WORLD);
+ smpi_execute_flops(comp_size);
+ smpi_mpi_bcast(NULL, comm_size, MPI_BYTE, 0, MPI_COMM_WORLD);
+#ifdef HAVE_TRACING
+ TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
+ TRACE_smpi_computing_in(rank);
+#endif
+
+ if (XBT_LOG_ISENABLED(smpi_replay, xbt_log_priority_verbose)){
+ char *name = xbt_str_join_array(action, " ");
+ XBT_VERB("%s %f", name, smpi_process_simulated_elapsed()-clock);
+ free(name);
+ }
}
void smpi_replay_init(int *argc, char***argv){
PMPI_Init(argc, argv);
_xbt_replay_action_init();
-// xbt_replay_action_register("init", action_init);
-// xbt_replay_action_register("finalize", action_finalize);
-// xbt_replay_action_register("comm_size",action_comm_size);
+ xbt_replay_action_register("init", action_init);
+ xbt_replay_action_register("finalize", action_finalize);
+ xbt_replay_action_register("comm_size",action_comm_size);
xbt_replay_action_register("send", action_send);
xbt_replay_action_register("Isend", action_Isend);
xbt_replay_action_register("recv", action_recv);
xbt_replay_action_register("Irecv", action_Irecv);
xbt_replay_action_register("wait", action_wait);
xbt_replay_action_register("barrier", action_barrier);
-// xbt_replay_action_register("bcast", action_bcast);
-// xbt_replay_action_register("reduce", action_reduce);
-// xbt_replay_action_register("allReduce",action_allReduce);
-// xbt_replay_action_register("sleep", action_sleep);
+ xbt_replay_action_register("bcast", action_bcast);
+ xbt_replay_action_register("reduce", action_reduce);
+ xbt_replay_action_register("allReduce",action_allReduce);
xbt_replay_action_register("compute", action_compute);
xbt_replay_action_runner(*argc, *argv);
