使用C ++ OpenMPI在主/从进程之间发送和接收列表

时间:2016-06-12 17:15:06

标签: c++ list openmpi

我正在尝试在主人和奴隶之间来回发送std::list<int>。但我不确定如何创建MPI数据类型并将列表传递给从属。我想在不使用boost之类的其他库的情况下这样做。我正试图解决这个post,但我无法使用list。以下是我正在使用的代码。

#include <iostream>
#include <list>
#include "mpi.h"
#include <stdio.h>
#include <stdlib.h>

using namespace std;

int main(int argc, char **argv)
{
int rank, size, tag, rc, i;
MPI_Status status;
char message[20];
char new_message[20];
MPI_Init(&argc, &argv);
MPI_Comm_size(MPI_COMM_WORLD, &size);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
tag=7;

if (rank==0) {
    strcpy(message, "Hello, world");
    for (int i=1; i<size; ++i){
        list<int> rand_list;
        list<int>::iterator it = rand_list.end();
        list<int> *lt_ptr = &rand_list;
        for(int j = 0; j < 5; ++j)
        {
             int r;
             r = rand();
             rand_list.push_front(r);
             it++;
        }
        //Instead of sending the string I'd like to send the list of random ints
        MPI_Send(message, 13, MPI_CHAR, i, tag, MPI_COMM_WORLD);
        MPI_Recv(new_message, 13, MPI_CHAR, i, tag, MPI_COMM_WORLD, &status);
        cout << "master: " << new_message << endl;
        }
}
else{
   // slaves processes
   rc = MPI_Recv(message, 13, MPI_CHAR, 0, tag, MPI_COMM_WORLD, &status);
   cout << "node " << rank << ": " << message << endl;
   strcpy(new_message, "Goodbye, world");
   // code to manipulate the lists and send back to the master process
   rc = MPI_Send(new_message, 13, MPI_CHAR, 0, tag, MPI_COMM_WORLD);
   }
MPI_Finalize();
}

以下是我用来编译和运行的命令。

mpic++ mpi_prog.cpp -o mpi_prog.o
mpirun -np 5 mpi_prog.o

2 个答案:

答案 0 :(得分:0)

发送列表的最简单方法可能是发送列表的长度(您有5个),然后循环遍历发送它们的元素。这会有相当差的性能,但由于许多原因,使用列表可能是一个有问题的设计选择。

由于您还要在主服务器中逐个循环处理这些流程,向每个流程发送不同的数字,因此您可能会更好地使用不同的设计。对于您的特定用例,如何在实际使用它们的进程上并行生成随机数。有目的设计的并行随机数生成算法,以及可推进的随机数生成器,您可以告诉每个从站在流中向前跳过多远,这样它们就不会重叠。

答案 1 :(得分:0)

为什么要在这里使用std::vectorstd::rand似乎更合适,数据是连续存储的,它应该直接与MPI一起工作。您过于复杂,并且更喜欢使用std::uniform_int_distribution而不是std::vector<int> v(5); std::random_device rd; std::mt19937 gen(rd()); // mersenne twister engine std::uniform_int_distribution<> dis; std::generate(v.begin(), v.end(), [&](){ return dis(gen); }); // fill the vector with random values

填充随机向量的主代码应该类似于:

MPI_Send(&v[0], v.size(), MPI_INT, i, tag, MPI_COMM_WORLD);

您的主节点可以直接在MPI中调用:

MPI_Recv(&v[0], v.size(), MPI_INT, 0, tag, MPI_COMM_WORLD, &status);

从节点代码:

size

我不确定这是你想要做的。您的主人将使用随机值初始化{{1}}个不同的向量,并将每个向量发送到一个从属节点。从属接收的每个向量将是不同的。这是对的吗?

相关问题