1、main.cpp
cpp
//main.cpp
#include<iostream>
#include<mpi.h>
#include <QtCore>
#include <QThread>
enum MyTag
{
TAG_TEST = 1,
TAG_TEST_RSP,
TAG_FORECAST,
TAG_COLWARN
};
void doMaster()
{
MPI_Status status;
QList<int> dataList;
for (int i = 0; i < 20000; ++i)
{
dataList << i;
}
char message[100] = {0};
int nCoreSize;
MPI_Comm_size(MPI_COMM_WORLD, &nCoreSize);
for (int i = 1; i < nCoreSize; ++i)
{
sprintf(message, "Message-%05d", i);
MPI_Send(message, strlen(message), MPI_CHAR, i, TAG_TEST, MPI_COMM_WORLD);
}
std::cout << "Master Sent Over\n";
int nCoreId;
int nRspCout = 0;
while (nRspCout < nCoreSize-1)
{
//使用MPI_ANY_SOURCE 接收每轮优先到达的消息
int ret = MPI_Recv((void*)(&nCoreId), 1, MPI_INT, MPI_ANY_SOURCE, TAG_TEST_RSP, MPI_COMM_WORLD, &status);
if (ret != 0) std::cout << "Master recv ret: " << ret << "," << status.MPI_ERROR << std::endl;
if (status.MPI_ERROR == 0)
{
nRspCout++;
printf("Master recv response from #%d, %d/%d\n", nCoreId, nRspCout, nCoreSize-1);
}
}
std::cout << "Master END" << std::endl;
}
void doSlave(int nCoreId)
{
MPI_Status status;
char message[100] = {0};
while (true)
{
QThread::msleep(100 + qrand() % 100); //使其随机化
MPI_Recv((void*)&message, 100, MPI_CHAR, 0, TAG_TEST, MPI_COMM_WORLD, &status);
if (status.MPI_ERROR != 0)
{
QThread::msleep(10);
continue;
}
std::cout << "Slave revec message (" << message << ") by core " << nCoreId << std::endl;
MPI_Send((void*)(&nCoreId), 1, MPI_INT, 0, TAG_TEST_RSP, MPI_COMM_WORLD);
break;
};
}
int main(int argc,char* argv[])
{
int nCoreId;
MPI_Init(&argc,&argv);
MPI_Comm_rank(MPI_COMM_WORLD,&nCoreId);
if (nCoreId == 0)
{
doMaster();
}
else
{
doSlave(nCoreId);
}
MPI_Finalize();
return 0;
}
2、TestCpp.pro
#pro file
TEMPLATE = app
CONFIG += console c++11
CONFIG -= app_bundle
#CONFIG -= qt
QT -= gui
SOURCES += \
main.cpp
INCLUDEPATH += /usr/mpi/gcc/openmpi-4.0.4rc3/include/
LIBS += -L/usr/mpi/gcc/openmpi-4.0.4rc3/lib64/ -lmpi
3、运行结果示例
$ mpirun -np 8 TestCpp
Master Sent Over
Slave revec message (Message-00006) by core 6
Master recv response from #6, 1/7
Slave revec message (Message-00001) by core 1
Master recv response from #1, 2/7
Slave revec message (Message-00007) by core 7
Master recv response from #7, 3/7
Slave revec message (Message-00002) by core 2
Master recv response from #2, 4/7
Slave revec message (Message-00005) by core 5
Slave revec message (Message-00004) by core 4
Master recv response from #4, 5/7
Master recv response from #5, 6/7
Slave revec message (Message-00003) by core 3
Master recv response from #3, 7/7
Master END