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_memmtmpi_h
00034 #define _util_group_memmtmpi_h
00035
00036 #include <fstream>
00037 #include <mpi.h>
00038
00039 #include <util/group/message.h>
00040 #include <util/group/memamsg.h>
00041 #include <util/group/thread.h>
00042
00043 class MTMPIThread;
00044
00049 class MTMPIMemoryGrp: public ActiveMsgMemoryGrp {
00050 private:
00051 Ref<ThreadGrp> th_;
00052
00053 Ref<ThreadLock> serial_lock_;
00054 int serial_;
00055 int serial(int node);
00056
00057 MPI_Comm comm_;
00058
00059 int req_tag_;
00060
00061 int active_;
00062
00063 unsigned int *nreq_sent_;
00064 unsigned int *nreq_sent_buf_;
00065
00066 MTMPIThread **thread_;
00067 Ref<ThreadLock> print_lock_;
00068 std::ofstream hout;
00069 std::ofstream mout;
00070
00071 void init_mtmpimg(int nthreads);
00072
00073
00074 void retrieve_data(void *, int node, int offset, int size, int lock);
00075 void replace_data(void *, int node, int offset, int size, int unlock);
00076 void sum_data(double *data, int node, int doffset, int dsize);
00077
00078 friend class MTMPIThread;
00079 public:
00080 MTMPIMemoryGrp(const Ref<MessageGrp>& msg, const Ref<ThreadGrp> &th);
00081 MTMPIMemoryGrp(const Ref<KeyVal> &);
00082 ~MTMPIMemoryGrp();
00083
00084 void activate();
00085 void deactivate();
00086
00087 void sync();
00088 };
00089
00090 #endif
00091
00092
00093
00094
00095