Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
2a8b43acc6fe640dab4dee7f3d3a2557e7111218
[simgrid.git] / examples / smpi / replay.c
1 /* Copyright (c) 2009, 2010, 2011, 2012. The SimGrid Team.
2  * All rights reserved.                                                     */
3
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. */
6
7
8 #include <stdio.h>
9 #include <mpi.h>
10 #include <xbt.h>
11 #include <xbt/replay.h>
12
13 XBT_LOG_NEW_DEFAULT_CATEGORY(smpi_replay,"Trace Replay with SMPI");
14
15 /* Helper function */
16 static double parse_double(const char *string)
17 {
18   double value;
19   char *endptr;
20   value = strtod(string, &endptr);
21   if (*endptr != '\0')
22     THROWF(unknown_error, 0, "%s is not a double", string);
23   return value;
24 }
25
26 static void action_compute(const char *const *action)
27 {
28   XBT_DEBUG("Start to compute %.0f flops", parse_double(action[2]));
29   smpi_sample_flops(parse_double(action[2]));
30 }
31
32 static void action_send(const char *const *action)
33 {
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);
38
39   XBT_DEBUG("send %.0f bytes to rank%d (%s)",size, to, action[2]);
40
41   MPI_Send(buffer, size, MPI_BYTE, to, 0, MPI_COMM_WORLD);
42
43   free(buffer);
44 }
45
46 static void action_Isend(const char *const *action)
47 {
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);
52
53   MPI_Request request;
54
55   MPI_Isend(buffer, size, MPI_BYTE, to, 0, MPI_COMM_WORLD, &request);
56
57   free(buffer);
58 }
59
60 static void action_recv(const char *const *action)
61 {
62   int from = atoi(action[2]);
63   XBT_DEBUG("receive from rank%d (%s)",from, action[2]);
64   char *buffer = (char*) malloc (1e9);
65
66   MPI_Status status;
67
68   MPI_Recv(buffer, 1e9, MPI_BYTE, from, 0, MPI_COMM_WORLD, &status);
69
70   free(buffer);
71 }
72
73 int main(int argc, char *argv[])
74 {
75   int i;
76
77   MPI_Init(&argc, &argv);
78   MPI_Comm_rank(MPI_COMM_WORLD, &i);
79   _xbt_replay_action_init();
80
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);
95
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
99
100   XBT_DEBUG("rank%d: I'm done!",i);
101   MPI_Finalize();
102   return 0;
103 }