Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
MPI_Allreduce()
[simgrid.git] / include / smpi / smpi.h
1 #ifndef SMPI_H
2 #define SMPI_H
3
4 #include <stddef.h>
5 #include <sys/time.h>
6 #include <xbt/misc.h>
7 #include <xbt/function_types.h>
8
9 SG_BEGIN_DECL()
10 #define SMPI_RAND_SEED 5
11 #define MPI_ANY_SOURCE -1
12 #define MPI_ANY_TAG -1
13 #define MPI_UNDEFINED -1
14 // errorcodes
15 #define MPI_SUCCESS     0
16 #define MPI_ERR_COMM    1
17 #define MPI_ERR_ARG     2
18 #define MPI_ERR_TYPE    3
19 #define MPI_ERR_REQUEST 4
20 #define MPI_ERR_INTERN  5
21 #define MPI_ERR_COUNT   6
22 #define MPI_ERR_RANK    7
23 #define MPI_ERR_TAG     8
24 // MPI_Comm
25      typedef struct smpi_mpi_communicator_t *smpi_mpi_communicator_t;
26      typedef smpi_mpi_communicator_t MPI_Comm;
27
28 // MPI_Datatype
29      typedef struct smpi_mpi_datatype_t *smpi_mpi_datatype_t;
30      typedef smpi_mpi_datatype_t MPI_Datatype;
31
32 // MPI_Request
33      typedef struct smpi_mpi_request_t *smpi_mpi_request_t;
34      typedef smpi_mpi_request_t MPI_Request;
35
36 // MPI_Op
37      typedef struct smpi_mpi_op_t *smpi_mpi_op_t;
38      typedef smpi_mpi_op_t MPI_Op;
39
40 // MPI_Status
41      struct smpi_mpi_status_t {
42        int MPI_SOURCE;
43        int MPI_TAG;
44        int MPI_ERROR;
45      };
46      typedef struct smpi_mpi_status_t smpi_mpi_status_t;
47      typedef smpi_mpi_status_t MPI_Status;
48
49 // global SMPI data structure
50      typedef struct smpi_mpi_global_t {
51
52        smpi_mpi_communicator_t mpi_comm_world;
53
54        smpi_mpi_datatype_t mpi_byte;
55        smpi_mpi_datatype_t mpi_int;
56        smpi_mpi_datatype_t mpi_float;
57        smpi_mpi_datatype_t mpi_double;
58
59        smpi_mpi_op_t mpi_land;
60        smpi_mpi_op_t mpi_sum;
61        smpi_mpi_op_t mpi_min;
62        smpi_mpi_op_t mpi_max;
63
64      } s_smpi_mpi_global_t;
65      typedef struct smpi_mpi_global_t *smpi_mpi_global_t;
66      extern smpi_mpi_global_t smpi_mpi_global;
67
68 #define MPI_COMM_WORLD    (smpi_mpi_global->mpi_comm_world)
69 #define MPI_COMM_NULL     NULL
70
71 #define MPI_STATUS_IGNORE NULL
72
73 #define MPI_BYTE          (smpi_mpi_global->mpi_byte)
74 #define MPI_INT           (smpi_mpi_global->mpi_int)
75 #define MPI_FLOAT         (smpi_mpi_global->mpi_float)
76 #define MPI_DOUBLE        (smpi_mpi_global->mpi_double)
77
78 #define MPI_LAND          (smpi_mpi_global->mpi_land)
79 #define MPI_SUM           (smpi_mpi_global->mpi_sum)
80 #define MPI_MIN           (smpi_mpi_global->mpi_min)
81 #define MPI_MAX           (smpi_mpi_global->mpi_max)
82
83 // MPI macros
84 #define MPI_Init(a, b) SMPI_MPI_Init(a, b)
85 #define MPI_Finalize() SMPI_MPI_Finalize()
86 #define MPI_Abort(a, b) SMPI_MPI_Abort(a, b)
87 #define MPI_Comm_size(a, b) SMPI_MPI_Comm_size(a, b)
88 #define MPI_Comm_rank(a, b) SMPI_MPI_Comm_rank(a, b)
89 #define MPI_Type_size(a, b) SMPI_MPI_Type_size(a, b)
90 #define MPI_Barrier(a) SMPI_MPI_Barrier(a)
91 #define MPI_Irecv(a, b, c, d, e, f, g) SMPI_MPI_Irecv(a, b, c, d, e, f, g)
92 #define MPI_Recv(a, b, c, d, e, f, g) SMPI_MPI_Recv(a, b, c, d, e, f, g)
93 #define MPI_Isend(a, b, c, d, e, f, g) SMPI_MPI_Isend(a, b, c, d, e, f, g)
94 #define MPI_Send(a, b, c, d, e, f) SMPI_MPI_Send(a, b, c, d, e, f)
95 #define MPI_Bcast(a, b, c, d, e) SMPI_MPI_Bcast(a, b, c, d, e)
96 #define MPI_Wait(a, b) SMPI_MPI_Wait(a, b)
97 #define MPI_Waitall(a, b, c) SMPI_MPI_Waitall(a, b, c)
98 #define MPI_Waitany(a, b, c, d) SMPI_MPI_Waitany(a, b, c, d)
99 #define MPI_Comm_split(a, b, c, d) SMPI_MPI_Comm_split(a, b, c, d)
100 #define MPI_Wtime() SMPI_MPI_Wtime()
101 #define MPI_Reduce( a, b, c, d, e, f, g) SMPI_MPI_Reduce( a, b, c, d, e, f, g)
102 #define MPI_Allreduce( a, b, c, d, e, f) SMPI_MPI_Allreduce( a, b, c, d, e, f)
103
104 // SMPI Functions
105 XBT_PUBLIC(int) SMPI_MPI_Init(int *argc, char ***argv);
106 XBT_PUBLIC(int) SMPI_MPI_Finalize(void);
107 XBT_PUBLIC(int) SMPI_MPI_Abort(MPI_Comm comm, int errorcode);
108 XBT_PUBLIC(int) SMPI_MPI_Comm_size(MPI_Comm comm, int *size);
109 XBT_PUBLIC(int) SMPI_MPI_Comm_rank(MPI_Comm comm, int *rank);
110 XBT_PUBLIC(int) SMPI_MPI_Type_size(MPI_Datatype datatype, size_t * size);
111 XBT_PUBLIC(int) SMPI_MPI_Barrier(MPI_Comm comm);
112 XBT_PUBLIC(int) SMPI_MPI_Irecv(void *buf, int count, MPI_Datatype datatype,
113                                int src, int tag, MPI_Comm comm,
114                                MPI_Request * request);
115 XBT_PUBLIC(int) SMPI_MPI_Recv(void *buf, int count, MPI_Datatype datatype,
116                               int src, int tag, MPI_Comm comm,
117                               MPI_Status * status);
118 XBT_PUBLIC(int) SMPI_MPI_Isend(void *buf, int count, MPI_Datatype datatype,
119                                int dst, int tag, MPI_Comm comm,
120                                MPI_Request * request);
121 XBT_PUBLIC(int) SMPI_MPI_Send(void *buf, int count, MPI_Datatype datatype,
122                               int dst, int tag, MPI_Comm comm);
123 XBT_PUBLIC(int) SMPI_MPI_Bcast(void *buf, int count, MPI_Datatype datatype,
124                                int root, MPI_Comm comm);
125 XBT_PUBLIC(int) SMPI_MPI_Wait(MPI_Request * request, MPI_Status * status);
126 XBT_PUBLIC(int) SMPI_MPI_Waitall(int count, MPI_Request requests[],
127                                  MPI_Status status[]);
128 XBT_PUBLIC(int) SMPI_MPI_Waitany(int count, MPI_Request requests[],
129                                  int *index, MPI_Status status[]);
130 XBT_PUBLIC(int) SMPI_MPI_Comm_split(MPI_Comm comm, int color, int key,
131                                     MPI_Comm * comm_out);
132 XBT_PUBLIC(double) SMPI_MPI_Wtime(void);
133
134 XBT_PUBLIC(int) SMPI_MPI_Reduce(void *sendbuf, void *recvbuf, int count,
135                                 MPI_Datatype datatype, MPI_Op op, int root,
136                                 MPI_Comm comm);
137 XBT_PUBLIC(int) SMPI_MPI_Allreduce(void *sendbuf, void *recvbuf, int count,
138                                     MPI_Datatype datatype, MPI_Op op, MPI_Comm comm);
139
140
141 // smpi functions
142 XBT_IMPORT_NO_EXPORT(int) smpi_simulated_main(int argc, char **argv);
143 XBT_PUBLIC(unsigned int) smpi_sleep(unsigned int);
144 XBT_PUBLIC(void) smpi_exit(int);
145 XBT_PUBLIC(int) smpi_gettimeofday(struct timeval *tv, struct timezone *tz);
146
147 XBT_PUBLIC(void) smpi_do_once_1(const char *file, int line);
148 XBT_PUBLIC(int) smpi_do_once_2(void);
149 XBT_PUBLIC(void) smpi_do_once_3(void);
150
151 #define SMPI_DO_ONCE for (smpi_do_once_1(__FILE__, __LINE__); smpi_do_once_2(); smpi_do_once_3())
152
153 SG_END_DECL()
154 #endif