c++ - 在 MPI 中发送和接收数组
问题描述
我是 MPI 的新手,我正在编写一个简单的 MPI 程序来获取矩阵和向量的点积,即 A*b=c。但是,我的代码不起作用。下面列出了源代码。
如果我将 A、b、c 和缓冲区的声明替换为
double A[16], b[4], c[4], buffer[8];
并注释与分配和释放操作相关的那些行,我的代码有效并且结果是正确的。在这种情况下,我想知道问题应该与指针有关,但我不知道如何解决问题。
还有一件事,在我的代码中,缓冲区只有 4 个元素,但缓冲区大小必须大于 8,否则它不起作用。
#include<mpi.h>
#include<iostream>
#include<stdlib.h>
using namespace std;
int nx = 4, ny = 4, nxny;
int ix, iy;
double *A = nullptr, *b = nullptr, *c = nullptr, *buffer = nullptr;
double ans;
// info MPI
int myGlobalID, root = 0, numProc;
int numSent;
MPI_Status status;
// functions
void get_ixiy(int);
int main(){
MPI_Init(NULL, NULL);
MPI_Comm_size(MPI_COMM_WORLD, &numProc);
MPI_Comm_rank(MPI_COMM_WORLD, &myGlobalID);
nxny = nx * ny;
A = new double(nxny);
b = new double(ny);
c = new double(nx);
buffer = new double(ny);
if(myGlobalID == root){
// init A, b
for(int k = 0; k < nxny; ++k){
get_ixiy(k);
b[iy] = 1;
A[k] = k;
}
numSent = 0;
// send b to each worker processor
MPI_Bcast(&b, ny, MPI_DOUBLE, root, MPI_COMM_WORLD);
// send a row of A to each worker processor, tag with row number
for(ix = 0; ix < min(numProc - 1, nx); ++ix){
for(iy = 0; iy < ny; ++iy){
buffer[iy] = A[iy + ix * ny];
}
MPI_Send(&buffer, ny, MPI_DOUBLE, ix+1, ix+1, MPI_COMM_WORLD);
numSent += 1;
}
for(ix = 0; ix < nx; ++ix){
MPI_Recv(&ans, 1, MPI_DOUBLE, MPI_ANY_SOURCE, MPI_ANY_TAG, MPI_COMM_WORLD, &status);
int sender = status.MPI_SOURCE;
int ansType = status.MPI_TAG;
c[ansType] = ans;
// send another row to worker process
if(numSent < nx){
for(iy = 0; iy < ny; ++iy){
buffer[iy] = A[iy + numSent * ny];
}
MPI_Send(&buffer, ny, MPI_DOUBLE, sender, numSent+1,
MPI_COMM_WORLD);
numSent += 1;
}
else
MPI_Send(MPI_BOTTOM, 0, MPI_DOUBLE, sender, 0, MPI_COMM_WORLD);
}
for(ix = 0; ix < nx; ++ix){
std::cout << c[ix] << " ";
}
std::cout << std::endl;
delete [] A;
delete [] b;
delete [] c;
delete [] buffer;
}
else{
MPI_Bcast(&b, ny, MPI_DOUBLE, root, MPI_COMM_WORLD);
if(myGlobalID <= nx){
while(1){
MPI_Recv(&buffer, ny, MPI_DOUBLE, root, MPI_ANY_TAG, MPI_COMM_WORLD, &status);
if(status.MPI_TAG == 0) break;
int row = status.MPI_TAG - 1;
ans = 0.0;
for(iy = 0; iy < ny; ++iy) ans += buffer[iy] * b[iy];
MPI_Send(&ans, 1, MPI_DOUBLE, root, row, MPI_COMM_WORLD);
}
}
}
MPI_Finalize();
return 0;
} // main
void get_ixiy(int k){
ix = k / ny;
iy = k % ny;
}
下面列出了错误信息。
= BAD TERMINATION OF ONE OF YOUR APPLICATION PROCESSES
= PID 7455 RUNNING AT ***
= EXIT CODE: 11
= CLEANING UP REMAINING PROCESSES
= YOU CAN IGNORE THE BELOW CLEANUP MESSAGES
YOUR APPLICATION TERMINATED WITH THE EXIT STRING: Segmentation fault:
11 (signal 11)
This typically refers to a problem with your application.
Please see the FAQ page for debugging suggestions
解决方案
您的代码中有几个问题,您必须先解决这些问题。
首先,您要b[]
在此 for 循环中访问不存在的元素:
for(int k = 0; k < nxny; ++k){
get_ixiy(k);
b[k] = 1; // WARNING: this is an error
A[k] = k;
}
其次,您只为根进程删除分配的内存。这会导致内存泄漏:
if(myGlobalID == root){
// ...
delete [] A;
delete [] b;
delete [] c;
delete [] buffer;
}
您必须删除为所有进程分配的内存。
第三,你有一个void get_ixiy(int);
改变全局变量 ix, iy 的无用函数。这是无用的,因为在调用此函数后,您永远不会使用 ix, iy 直到您手动更改它们。看这里:
for(ix = 0; ix < min(numProc - 1, nx); ++ix){
for(iy = 0; iy < ny; ++iy){
// ...
}
}
第四,您正在以完全错误的方式使用MPI_Send()
和MPI_Recv()
。你很幸运,你没有得到更多的错误。
推荐阅读
- node.js - Homebrew 安装 Node 和 NPM 时出现问题
- javascript - 我想从字符串中删除特殊字符,但我需要 javascript 中的字符串中的点
- python - 使用 pandas to_csv 时删除标题中的前导逗号
- algorithm - 选择数组元素的方法有多少?
- ios - 计算 iPhone 行进的短垂直距离
- bash - Bash - 检查给定参数是否存在
- javascript - 非单词是否有任何javascript单词边界替代方案?
- java - 由于 IllegalAccessException,Clojure 无法在 java 11 上调用接口默认方法
- c++ - 使用基于范围的循环迭代向量时出现问题
- python - 不明白为什么 UDS 长度对我来说似乎是错误的