source: XIOS/dev/branch_openmp/extern/src_ep_dev/ep_lib.hpp @ 1368

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

unify type : MPI_Op

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