| 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 | |
|---|
| 12 | #include "ep_global.hpp" |
|---|
| 13 | |
|---|
| 14 | namespace 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 | 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 |
|---|
| 40 | |
|---|
| 41 | int MPI_Send(void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm); |
|---|
| 42 | int MPI_Ssend(void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm); |
|---|
| 43 | int MPI_Isend(const void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request *request); |
|---|
| 44 | int MPI_Issend(const void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request *request); |
|---|
| 45 | |
|---|
| 46 | int tag_combine(int real_tag, int src, int dest); |
|---|
| 47 | int get_ep_rank(MPI_Comm comm, int ep_rank_loc, int mpi_rank); |
|---|
| 48 | int get_ep_rank_intercomm(MPI_Comm comm, int ep_rank_loc, int mpi_rank); |
|---|
| 49 | |
|---|
| 50 | int Message_Check(MPI_Comm comm); |
|---|
| 51 | int Request_Check(); |
|---|
| 52 | |
|---|
| 53 | int MPI_Recv (void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Status *status); |
|---|
| 54 | int MPI_Irecv (void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Request *request); |
|---|
| 55 | int MPI_Mrecv (void *buf, int count, MPI_Datatype datatype, MPI_Message *message, MPI_Status *status); |
|---|
| 56 | int MPI_Imrecv(void *buf, int count, MPI_Datatype datatype, MPI_Message *message, MPI_Request *request); |
|---|
| 57 | |
|---|
| 58 | |
|---|
| 59 | int MPI_Test(MPI_Request *request, int *flag, MPI_Status *status); |
|---|
| 60 | int MPI_Testall(int count, MPI_Request *array_of_requests, int *flag, MPI_Status *array_of_statuses); |
|---|
| 61 | |
|---|
| 62 | int MPI_Wait(MPI_Request *request, MPI_Status *status); |
|---|
| 63 | int MPI_Waitall(int count, MPI_Request *array_of_requests, MPI_Status *array_of_statuses); |
|---|
| 64 | |
|---|
| 65 | |
|---|
| 66 | int MPI_Iprobe(int source, int tag, MPI_Comm comm, int *flag, MPI_Status *status); |
|---|
| 67 | int MPI_Improbe(int source, int tag, MPI_Comm comm, int *flag, MPI_Message *message, MPI_Status *status); |
|---|
| 68 | |
|---|
| 69 | int MPI_Alloc_mem(MPI_Aint size, MPI_Info info, void *baseptr); |
|---|
| 70 | |
|---|
| 71 | //int MPI_Alloc_mem(unsigned long size, MPI_Info info, void *baseptr); |
|---|
| 72 | |
|---|
| 73 | |
|---|
| 74 | void check_sum_send(const void *buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, int type); |
|---|
| 75 | void check_sum_recv(void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, int type); |
|---|
| 76 | |
|---|
| 77 | bool valid_type(MPI_Datatype type); |
|---|
| 78 | bool valid_op(MPI_Op op); |
|---|
| 79 | |
|---|
| 80 | } |
|---|
| 81 | |
|---|
| 82 | |
|---|
| 83 | #endif // EP_LIB_HPP_INCLUDED |
|---|
