source: XIOS/dev/branch_yushan/extern/src_ep_dev/ep_lib.hpp @ 1037

Last change on this file since 1037 was 1037, checked in by yushan, 7 years ago

initialize the branch

File size: 3.3 KB
Line 
1#ifndef EP_LIB_HPP_INCLUDED
2#define EP_LIB_HPP_INCLUDED
3
4#include "ep_type.hpp"
5#include "ep_lib_intercomm.hpp"
6#include "ep_lib_local.hpp"
7#include "ep_lib_collective.hpp"
8#include "ep_tag.hpp"
9#include "ep_lib_fortran.hpp"
10
11
12namespace ep_lib
13{
14#ifdef _intelmpi
15  typedef int MPI_Datatype;
16  typedef int MPI_Op;
17#elif _openmpi
18  typedef void* MPI_Datatype;
19  typedef void* MPI_Op;
20#endif
21
22  #define MPI_ANY_SOURCE -2 
23  #define MPI_ANY_TAG -1 
24
25  int MPI_Init_thread(int* argc, char*** argv, int required, int*provided);
26
27  int MPI_Init(int *argc, char ***argv);
28
29  int MPI_Initialized(int *flag);
30
31  int MPI_Comm_rank(MPI_Comm comm, int* rank);
32
33  int MPI_Comm_size(MPI_Comm comm, int* rank);
34
35  int MPI_Comm_free(MPI_Comm* comm);
36
37  int MPI_Finalize();
38 
39  double MPI_Wtime();
40
41  int MPI_Abort(MPI_Comm comm, int errorcode);
42
43  int MPI_Get_count(const MPI_Status *status, MPI_Datatype datatype, int *count);
44
45  #ifdef _openmpi
46  int MPI_Comm_create_endpoints(void* mpi_comm, int num_ep, MPI_Info info, MPI_Comm *& out_comm_hdls); // from MPI to create endpoints
47  #elif _intelmpi
48  int MPI_Comm_create_endpoints(int mpi_comm, int num_ep, MPI_Info info, MPI_Comm *& out_comm_hdls); // from MPI to create endpoints
49  #endif
50 
51  int MPI_Comm_create_endpoints(MPI_Comm base_comm, int num_ep, MPI_Info info, MPI_Comm *& out_comm_hdls); // from EP to create endpoints
52
53  int MPI_Send(void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm);
54  int MPI_Ssend(void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm);
55  int MPI_Isend(const void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request *request);
56  int MPI_Issend(const void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request *request);
57
58  int tag_combine(int real_tag, int src, int dest);
59  int get_ep_rank(MPI_Comm comm, int ep_rank_loc, int mpi_rank);
60  int get_ep_rank_intercomm(MPI_Comm comm, int ep_rank_loc, int mpi_rank);
61
62  int Message_Check(MPI_Comm comm);
63
64  int MPI_Recv  (void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Status *status);
65  int MPI_Irecv (void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Request *request);
66  int MPI_Mrecv (void *buf, int count, MPI_Datatype datatype, MPI_Message *message, MPI_Status *status);
67  int MPI_Imrecv(void *buf, int count, MPI_Datatype datatype, MPI_Message *message, MPI_Request *request);
68
69
70  int MPI_Test(MPI_Request *request, int *flag, MPI_Status *status);
71  int MPI_Testall(int count, MPI_Request *array_of_requests, int *flag, MPI_Status *array_of_statuses);
72
73  int MPI_Wait(MPI_Request *request, MPI_Status *status);
74  int MPI_Waitall(int count, MPI_Request *array_of_requests, MPI_Status *array_of_statuses);
75
76
77  int MPI_Iprobe(int source, int tag, MPI_Comm comm, int *flag, MPI_Status *status);
78  int MPI_Improbe(int source, int tag, MPI_Comm comm, int *flag, MPI_Message *message, MPI_Status *status);
79 
80  int MPI_Alloc_mem(MPI_Aint size, MPI_Info info, void *baseptr);
81 
82  int MPI_Alloc_mem(unsigned long size, MPI_Info info, void *baseptr);
83
84
85  void check_sum_send(const void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, int type);
86  void check_sum_recv(void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, int type);
87}
88
89
90
91#endif // EP_LIB_HPP_INCLUDED
Note: See TracBrowser for help on using the repository browser.