4 #include "xbt/mallocator.h"
5 #include "xbt/xbt_os_time.h"
7 #include "simix/simix.h"
11 #define SMPI_DEFAULT_SPEED 100
12 #define SMPI_REQUEST_MALLOCATOR_SIZE 100
13 #define SMPI_MESSAGE_MALLOCATOR_SIZE 100
15 // smpi mpi communicator
16 typedef struct smpi_mpi_communicator_t {
19 smx_mutex_t barrier_mutex;
20 smx_cond_t barrier_cond;
22 int *rank_to_index_map;
23 int *index_to_rank_map;
25 } s_smpi_mpi_communicator_t;
28 typedef struct smpi_mpi_datatype_t {
32 uint16_t flags; /* flags: has it been committed, etc ...*/
33 uint16_t id; /* unused so far : data id, normally the index in the data array. */
34 } s_smpi_mpi_datatype_t;
37 typedef struct smpi_mpi_request_t {
38 smpi_mpi_communicator_t comm;
45 smpi_mpi_datatype_t datatype;
47 short int completed:1;
48 short int consumed:1; /* for waitany */
56 } s_smpi_mpi_request_t;
59 typedef struct smpi_mpi_op_t {
60 void (*func) (void *a, void *b, int *length, MPI_Datatype * datatype);
63 // smpi received message
64 typedef struct smpi_received_message_t {
65 smpi_mpi_communicator_t comm;
74 } s_smpi_received_message_t;
75 typedef struct smpi_received_message_t *smpi_received_message_t;
77 typedef struct smpi_do_once_duration_node_t {
81 struct smpi_do_once_duration_node_t *next;
82 } s_smpi_do_once_duration_node_t;
83 typedef struct smpi_do_once_duration_node_t *smpi_do_once_duration_node_t;
85 typedef struct smpi_global_t {
88 double reference_speed;
92 xbt_mallocator_t request_mallocator;
93 xbt_mallocator_t message_mallocator;
95 smx_process_t *main_processes;
98 smx_cond_t timer_cond;
100 // keeps track of previous times
101 smpi_do_once_duration_node_t do_once_duration_nodes;
102 smx_mutex_t do_once_mutex;
103 double *do_once_duration;
106 typedef struct smpi_global_t *smpi_global_t;
107 extern smpi_global_t smpi_global;
109 typedef struct smpi_host_data_t {
115 smx_process_t sender;
116 smx_process_t receiver;
118 int finalize; /* so that main process stops its sender&receiver */
120 xbt_fifo_t pending_recv_request_queue;
121 xbt_fifo_t pending_send_request_queue;
122 xbt_fifo_t received_message_queue;
123 } s_smpi_process_data_t;
124 typedef struct smpi_host_data_t *smpi_process_data_t;
126 // function prototypes
127 void smpi_process_init(int *argc, char ***argv);
128 void smpi_process_finalize(void);
129 int smpi_mpi_comm_rank(smpi_mpi_communicator_t comm);
130 int smpi_mpi_type_get_extent(MPI_Datatype datatype, MPI_Aint *lb, MPI_Aint *extent);
132 int smpi_mpi_bcast(void *buf, int count, MPI_Datatype datatype, int root, MPI_Comm comm);
133 int smpi_mpi_barrier(smpi_mpi_communicator_t comm);
135 int smpi_mpi_isend(smpi_mpi_request_t request);
136 int smpi_mpi_irecv(smpi_mpi_request_t request);
137 int smpi_mpi_reduce(void *sendbuf, void *recvbuf, int count, MPI_Datatype datatype, MPI_Op op, int root, MPI_Comm comm);
138 int smpi_mpi_sendrecv(void *sendbuf, int sendcount, MPI_Datatype sendtype, int dest, int sendtag,
139 void *recvbuf, int recvcount, MPI_Datatype recvtype, int source, int recvtag,
140 MPI_Comm comm, MPI_Status *status);
141 int smpi_mpi_wait(smpi_mpi_request_t request, smpi_mpi_status_t * status);
142 int smpi_mpi_waitall(int count, smpi_mpi_request_t requests[], smpi_mpi_status_t status[]);
143 int smpi_mpi_waitany(int count, smpi_mpi_request_t requests[], int *index, smpi_mpi_status_t status[]);
147 void smpi_execute(double duration);
148 void smpi_start_timer(void);
149 double smpi_stop_timer(void);
150 void smpi_bench_begin(void);
151 void smpi_bench_end(void);
152 void smpi_bench_skip(void);
154 void smpi_global_init(void);
155 void smpi_global_destroy(void);
156 int smpi_process_index(void);
157 smx_mutex_t smpi_process_mutex(void);
158 smx_cond_t smpi_process_cond(void);
159 int smpi_run_simulation(int *argc, char **argv);
160 int smpi_create_request(void *buf, int count, smpi_mpi_datatype_t datatype,
161 int src, int dst, int tag,
162 smpi_mpi_communicator_t comm,
163 smpi_mpi_request_t * request);
165 int smpi_sender(int argc, char *argv[]);
166 int smpi_receiver(int argc, char *argv[]);