如何使用MPI Allreduce添加armadillo矢量

How can I add armadillo vectors using MPI Allreduce?

本文关键字：armadillo 矢量 添加 Allreduce 何使用 MPI      更新时间：2023-10-16

我想对Armadllio向量执行MPI_Allreduce operation，但我得到了0输出。本质上，我每个处理器都有arma向量v。使用MPI_Allreduce，我想添加w中的所有向量。我不确定是否必须添加一些同步。下面是我的代码。

#include <iostream>
#include <stdlib.h>
#include <mpi.h>
#include <math.h>
#include <assert.h>
#include<armadillo>  
using namespace std;
using namespace arma;
int main(int argc, char** argv) {

MPI_Init(NULL, NULL);
int world_rank;
MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
int world_size;
MPI_Comm_size(MPI_COMM_WORLD, &world_size);
vec v(2,fill::randu);
vec w(2,fill::zeros);
MPI_Allreduce(&w, &v, 1, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
if(world_rank==0)
cout<<"w: "<<w<<endl;
MPI_Finalize();
}

我的输出：

 w:     0
      0

您的MPI_Allreduce调用有两个问题。首先，您的vec中有2个double，所以计数应该是2。其次，您已经切换了发送和接收缓冲区。

int MPI_Allreduce（void*sendbuf，void*recvbuf，int count，MPI_Datatype数据类型，MPI_Op Op，MPI_Comm Comm）

所以你的命令应该是：

MPI_Allreduce(&v, &w, 2, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);

假设armadillo向量中的数据从类/结构本身的第一个字节开始连续存储，这应该可以正常工作。

编辑：我仔细看了一下armadillo文档。有一个函数memptr()，它返回一个指向数据本身的指针。这是您应该用于MPI_Allreduce()而不是&vector的内容。因此，您的实际函数调用应该是：

MPI_Allreduce(v.memptr(), w.memptr(), 2, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
相关文章：