00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029 #ifdef __GNUC__
00030 #pragma interface
00031 #endif
00032
00033 #ifndef _util_group_memmpi2_h
00034 #define _util_group_memmpi2_h
00035
00036 #include <util/group/memamsg.h>
00037
00038 #include <mpi.h>
00039
00040 class MPI2MemoryGrp: public MsgMemoryGrp {
00041 private:
00042 MPI_Win rma_win_;
00043 void *data_;
00044
00045 void retrieve_data(void *, int node, int offset, int size);
00046 void replace_data(void *, int node, int offset, int size);
00047 void sum_data(double *data, int node, int doffset, int dsize);
00048
00049 public:
00050 MPI2MemoryGrp(const Ref<MessageGrp>& msg);
00051 MPI2MemoryGrp(const Ref<KeyVal>&);
00052 ~MPI2MemoryGrp();
00053
00054 void activate();
00055 void deactivate();
00056
00057 void *localdata();
00058 void set_localsize(int);
00059
00060 void *obtain_writeonly(distsize_t offset, int size);
00061 void *obtain_readwrite(distsize_t offset, int size);
00062 void *obtain_readonly(distsize_t offset, int size);
00063 void release_readonly(void *data, distsize_t offset, int size);
00064 void release_writeonly(void *data, distsize_t offset, int size);
00065 void release_readwrite(void *data, distsize_t offset, int size);
00066 void sum_reduction(double *data, distsize_t doffset, int dsize);
00067 void sum_reduction_on_node(double *data, int doffset, int dsize,
00068 int node = -1);
00069
00070 void sync();
00071 };
00072
00073 #endif
00074
00075
00076
00077
00078