Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
91d84892bb51b74e05d9570f026d9b32509f634e
[simgrid.git] / src / smpi / private.h
1 /* Copyright (c) 2007, 2009, 2010. 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 #ifndef SMPI_PRIVATE_H
8 #define SMPI_PRIVATE_H
9
10 #include "xbt.h"
11 #include "xbt/xbt_os_time.h"
12 #include "simix/simix.h"
13 #include "smpi/smpi.h"
14 #include "instr/private.h"
15
16 struct s_smpi_process_data;
17 typedef struct s_smpi_process_data* smpi_process_data_t;
18
19 #define PERSISTENT     0x0
20 #define NON_PERSISTENT 0x1
21 #define SEND           0x0
22 #define RECV           0x2
23
24 typedef struct s_smpi_mpi_request {
25   void* buf;
26   size_t size;
27   int src;
28   int dst;
29   int tag;
30   MPI_Comm comm;
31   smx_rdv_t rdv;
32   smx_comm_t pair;
33   int complete;
34   MPI_Request match;
35   unsigned flags;
36 #ifdef HAVE_TRACING
37   int send;
38   int recv;
39 #endif
40 } s_smpi_mpi_request_t;
41
42 void smpi_process_init(int* argc, char*** argv);
43 void smpi_process_destroy(void);
44
45 smpi_process_data_t smpi_process_data(void);
46 smpi_process_data_t smpi_process_remote_data(int index);
47 int smpi_process_count(void);
48 int smpi_process_index(void);
49 xbt_os_timer_t smpi_process_timer(void);
50 void print_request(const char* message, MPI_Request request);
51 void smpi_process_post_send(MPI_Comm comm, MPI_Request request);
52 void smpi_process_post_recv(MPI_Request request);
53
54 void smpi_global_init(void);
55 void smpi_global_destroy(void);
56
57 size_t smpi_datatype_size(MPI_Datatype datatype);
58 MPI_Aint smpi_datatype_lb(MPI_Datatype datatype);
59 MPI_Aint smpi_datatype_ub(MPI_Datatype datatype);
60 int smpi_datatype_extent(MPI_Datatype datatype, MPI_Aint* lb, MPI_Aint * extent);
61 int smpi_datatype_copy(void* sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int recvcount, MPI_Datatype recvtype);
62
63 MPI_Op smpi_op_new(MPI_User_function* function, int commute);
64 void smpi_op_destroy(MPI_Op op);
65 void smpi_op_apply(MPI_Op op, void* invec, void* inoutvec, int* len, MPI_Datatype* datatype);
66
67 MPI_Group smpi_group_new(int size);
68 void smpi_group_destroy(MPI_Group group);
69 void smpi_group_set_mapping(MPI_Group group, int index, int rank);
70 int smpi_group_index(MPI_Group group, int rank);
71 int smpi_group_rank(MPI_Group group, int index);
72 int smpi_group_use(MPI_Group group);
73 int smpi_group_unuse(MPI_Group group);
74 int smpi_group_size(MPI_Group group);
75 int smpi_group_compare(MPI_Group group1, MPI_Group group2);
76
77 MPI_Comm smpi_comm_new(MPI_Group group);
78 void smpi_comm_destroy(MPI_Comm comm);
79 MPI_Group smpi_comm_group(MPI_Comm comm);
80 int smpi_comm_size(MPI_Comm comm);
81 int smpi_comm_rank(MPI_Comm comm);
82
83 MPI_Request smpi_mpi_send_init(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
84 MPI_Request smpi_mpi_recv_init(void* buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm);
85 void smpi_mpi_start(MPI_Request request);
86 void smpi_mpi_startall(int count, MPI_Request* requests);
87 void smpi_mpi_request_free(MPI_Request* request);
88 MPI_Request smpi_isend_init(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
89 MPI_Request smpi_mpi_isend(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
90 MPI_Request smpi_irecv_init(void* buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm);
91 MPI_Request smpi_mpi_irecv(void* buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm);
92 void smpi_mpi_recv(void* buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Status* status);
93 void smpi_mpi_send(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
94 void smpi_mpi_sendrecv(void* sendbuf, int sendcount, MPI_Datatype sendtype, int dst, int sendtag, void* recvbuf, int recvcount, MPI_Datatype recvtype, int src, int recvtag, MPI_Comm comm, MPI_Status* status);
95 int smpi_mpi_test(MPI_Request* request, MPI_Status* status);
96 int smpi_mpi_testany(int count, MPI_Request requests[], int* index, MPI_Status* status);
97 int smpi_mpi_get_count(MPI_Status* status, MPI_Datatype datatype);
98 void smpi_mpi_wait(MPI_Request* request, MPI_Status* status);
99 int smpi_mpi_waitany(int count, MPI_Request requests[], MPI_Status* status);
100 void smpi_mpi_waitall(int count, MPI_Request requests[],  MPI_Status status[]);
101 int smpi_mpi_waitsome(int incount, MPI_Request requests[], int* indices, MPI_Status status[]);
102 void smpi_mpi_bcast(void* buf, int count, MPI_Datatype datatype, int root, MPI_Comm comm);
103 void smpi_mpi_barrier(MPI_Comm comm);
104 void smpi_mpi_gather(void* sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int recvcount, MPI_Datatype recvtype, int root, MPI_Comm comm);
105 void smpi_mpi_gatherv(void* sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int* recvcounts, int* displs, MPI_Datatype recvtype, int root, MPI_Comm comm);
106 void smpi_mpi_allgather(void* sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int recvcount, MPI_Datatype recvtype, MPI_Comm comm);
107 void smpi_mpi_allgatherv(void* sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int* recvcounts, int* displs, MPI_Datatype recvtype, MPI_Comm comm);
108 void smpi_mpi_scatter(void* sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int recvcount, MPI_Datatype recvtype, int root, MPI_Comm comm);
109 void smpi_mpi_scatterv(void* sendbuf, int* sendcounts, int* displs, MPI_Datatype sendtype, void* recvbuf, int recvcount, MPI_Datatype recvtype, int root, MPI_Comm comm);
110 void smpi_mpi_reduce(void* sendbuf, void* recvbuf, int count, MPI_Datatype datatype, MPI_Op op, int root, MPI_Comm comm);
111 void smpi_mpi_allreduce(void* sendbuf, void* recvbuf, int count, MPI_Datatype datatype, MPI_Op op, MPI_Comm comm);
112 void smpi_mpi_scan(void* sendbuf, void* recvbuf, int count, MPI_Datatype datatype, MPI_Op op, MPI_Comm comm);
113
114 void nary_tree_bcast(void* buf, int count, MPI_Datatype datatype, int root, MPI_Comm comm, int arity);
115 void nary_tree_barrier(MPI_Comm comm, int arity);
116
117 int smpi_coll_tuned_alltoall_bruck(void* sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int recvcount, MPI_Datatype recvtype, MPI_Comm comm);
118 int smpi_coll_tuned_alltoall_basic_linear(void *sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int recvcount, MPI_Datatype recvtype, MPI_Comm comm);
119 int smpi_coll_tuned_alltoall_pairwise(void* sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int recvcount, MPI_Datatype recvtype, MPI_Comm comm);
120 int smpi_coll_basic_alltoallv(void* sendbuf, int* sendcounts, int* senddisps, MPI_Datatype sendtype, void* recvbuf, int *recvcounts, int* recvdisps, MPI_Datatype recvtype, MPI_Comm comm);
121
122 // utilities
123 void smpi_bench_destroy(void);
124 void smpi_bench_begin(int rank, const char* mpi_call);
125 void smpi_bench_end(int rank, const char* mpi_call);
126
127 #endif