Sendrecv Usage

 有时候，一个进程需要给另一个进程发送数据，同时接受第三个进程的数据。如果是Send和Recv来实现，需要开启两次通讯域。MPI提供了Sendrecv来以开启一次通讯域的成本来达到上述目的。该函数的具体代码如下：

 

MPI::COMM_WORLD.Sendrecv( &dSecondData, sizeof(dSecondData)/sizeof(double), MPI_DOUBLE, nRecvProcessId, nMsgTag+2, &testData, sizeof(testData)/sizeof(CTestSendRecv), newType, mpi.GetCurrentProcessId()-1, nMsgTag+1 );

 共10个输入参数，前五个与Send相同，后五个与Recv相同。发送缓冲区（变量）和接收缓冲区（变量）必须分开, 但可以是不同长度和类型。

 完整示例代码如下：

#include "Class_mpi.h" #include <cmath> class CTestSendRecv { public: CTestSendRecv(){Init();} void Init() { m_d =11.1; m_f =22.2f; m_n =33; m_arr[0]= 1.1; m_arr[1]= 2.2; m_arr[2]= 3.3;m_arr[3]= 4.4; } static MPI::Datatype GenerateMPINewType() { // 与成员变量对应 int blockLength[] = {1,1,4,1}; MPI::Datatype oldTypes[] = {MPI_INT,MPI_DOUBLE,MPI_DOUBLE,MPI_FLOAT}; //直接计算偏移量,每一个数据以double对齐 MPI::Aint addressOffsets[]={0,sizeof(double),2*sizeof(double),6*sizeof(double)}; return MPI::Datatype::Create_struct( sizeof(blockLength)/sizeof(int), /**/ blockLength, addressOffsets, oldTypes);; } static MPI::Aint GetLastAddressOffset(){ return 7*sizeof(double); } int m_n; double m_d; double m_arr[4]; float m_f; }; int main(int argc, char* argv[]) { Cmpi mpi; //mpi.PrintMachineInfo(); mpi.PrintMachineInfoWithStdIO(); MPI::Datatype newType = CTestSendRecv::GenerateMPINewType();//CTestSendRecv::GenerateMPINewType(); newType.Commit(); const int nMsgTag =999; /*消息标识号 */ int nSendProcessId(MPI_PROC_NULL); /*发送数据的进程的编号*/ int nRecvProcessId = MPI_PROC_NULL;/*目的进程的标识号*/ const int minProcessorCount =3; if (mpi.GetTotalProcessCount() < minProcessorCount) { std::cout<<"Not enough processor"<<std::endl; return 0; } double dSendRecvData = -9.9; // 所有进程均要使用的变量 CTestSendRecv testData; switch (mpi.GetCurrentProcessId()) { case 0: /* 第一个进程不接受数据，向第2个进程发送CTestSendRecv数据 */ nRecvProcessId = mpi.GetCurrentProcessId()+1; testData.m_d = dSendRecvData-0.1; MPI::COMM_WORLD.Sendrecv( &testData, sizeof(testData)/sizeof(CTestSendRecv), newType, nRecvProcessId, nMsgTag+1, &testData, sizeof(testData)/sizeof(CTestSendRecv), newType, MPI_PROC_NULL, nMsgTag ); break; case 1: /* 第二个进程接收第一个进程的CTestSendRecv数据，同时向第三个进程发送double数据 */ std::cout<<"The 2nd process before receiving data:"<<testData.m_d<<std::endl; nRecvProcessId = mpi.GetCurrentProcessId()+1; double dSecondData = dSendRecvData * 9; MPI::COMM_WORLD.Sendrecv( &dSecondData, sizeof(dSecondData)/sizeof(double), MPI_DOUBLE, nRecvProcessId, nMsgTag+2, &testData, sizeof(testData)/sizeof(CTestSendRecv), newType, mpi.GetCurrentProcessId()-1, nMsgTag+1 ); std::cout<<"The 2nd process after receiving data:"<<testData.m_d<<std::endl; break; case 2: /* 第3个进程接收第2个进程的数据，不发送数据 */ std::cout<<"The 3rd process before receiving data:"<<dSendRecvData<<std::endl; nRecvProcessId = mpi.GetCurrentProcessId()+1; MPI::COMM_WORLD.Sendrecv( &dSendRecvData, sizeof(dSendRecvData)/sizeof(double), MPI_DOUBLE, MPI_PROC_NULL, nMsgTag, &dSendRecvData, sizeof(dSendRecvData)/sizeof(double), MPI_DOUBLE, mpi.GetCurrentProcessId()-1, nMsgTag+2 ); std::cout<<"The 3rd process after receiving data:"<<dSendRecvData<<std::endl; break; } newType.Free(); return 0; }

 

在三核的HP台式机上的运行结果如下：

 

Machine Name:                   Justin
Current Process ID:             1
Total Number of Processes:      3
The 2nd process before receiving data:11.1
Machine Name:                   Justin
Current Process ID:             2
Total Number of Processes:      3
The 3rd process before receiving data:-9.9
The 3rd process after receiving data:-89.1
The 2nd process after receiving data:-10
Machine Name:                   Justin
Current Process ID:             0
Total Number of Processes:      3