source: XIOS/dev/branch_openmp/extern/ep_dev/ep_lib.hpp @ 1503

Last change on this file since 1503 was 1503, checked in by yushan, 3 years ago

rank_map is passed from vector to map, in order to have more flexibility in comm_split

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