我打算用 MPI 学习并行编程。我有一些错误
#include "mpi.h"
#include <stdio.h>
#include <stdlib.h>
int main(int argc, char** argv)
{
int procNum, procRank;
int m,n;
int sumProc = 0, sumAll = 0;
int** arr;
MPI_Status status;
MPI_Init ( &argc, &argv );
MPI_Comm_size ( MPI_COMM_WORLD, &procNum );
MPI_Comm_rank ( MPI_COMM_WORLD, &procRank );
if (procRank == 0)
{
printf("Type the array size \n");
scanf("%i %i", &m, &n);
}
MPI_Bcast(&m, 1, MPI_INT, 0, MPI_COMM_WORLD);
MPI_Bcast(&n, 1, MPI_INT, 0, MPI_COMM_WORLD);
arr = new int*[m];
for (int i = 0; i < m; i++)
arr[i] = new int[n];
if (procRank == 0)
{
for (int i = 0; i < m; i++)
{
for (int j = 0; j < n; j++)
{
arr[i][j] = rand() % 30;
printf("%i ", arr[i][j]);
}
printf("\n");
}
}
MPI_Bcast(&arr[0][0], m*n, MPI_INT, 0, MPI_COMM_WORLD);
for (int i = procRank; i < n; i += procNum)
for (int j = 0; j < m; j++)
sumProc += arr[j][i];
MPI_Reduce(&sumProc,&sumAll,1,MPI_INT,MPI_SUM,0,MPI_COMM_WORLD);
if (procRank == 0)
{
printf("sumAll = %i", sumAll);
}
delete *arr;
MPI_Finalize();
return 0;
}
我正在尝试将二维数组传递给其他进程,但是当我检查它时,我得到了错误的数组。像这样的东西:
Original array
11 17 4
10 29 4
18 18 22
Array which camed
11 17 4
26 0 0
28 0 0
这是什么问题?也许问题出在 MPI_Bcast
PS我加了
for (int i = 0; i < m; i++)
MPI_Bcast(arr[i], n, MPI_INT, 0, MPI_COMM_WORLD);
代替
MPI_Bcast(&arr[0][0], m*n, MPI_INT, 0, MPI_COMM_WORLD);
它解决了我的问题