我是 MPI 初学者。我有gmat
一大堆数字(双精度类型,尺寸 1x14000000),它们是预先计算并存储在二进制文件中的。它将使用大约 100 MB 的内存(14000000 x8 字节 /1024 /1024)。我想编写一个 MPI 代码,它将对该数组进行一些计算(例如,将 的所有元素乘以gmat
进程的秩数)。该数组gmat
本身在运行时保持不变。代码应该是这样的
#include <iostream>
#include "mpi.h"
double* gmat;
long int imax;
int main(int argc, char* argv[])
{
void performcomputation(int rank); // this function performs the computation and will be called by all processes
imax=atoi(argv[1]); // user inputs the length of gmat
MPI::Init();
rank = MPI::COMM_WORLD.Get_rank();
size = MPI::COMM_WORLD.Get_size(); //i will use -np 16 = 4 processors x 4 cores
if rank==0 // read the gmat array using one of the processes
{
gmat = new double[imax];
// read values of gmat from a file
// next line is supposed to broadcast values of gmat to all processes which will use it
MPI::COMM_WORLD.Bcast(&gmat,imax,MPI::DOUBLE,1);
}
MPI::COMM_WORLD.Barrier();
performcomputation(rank);
MPI::Finalize();
return 0;
}
void performcomputation(int rank)
{
int i;
for (i=0;i <imax; i++)
cout << "the new value is" << gmat[i]*rank << endl;
}
我的问题是,当我使用 16 个进程(-np 16)运行此代码时,它们的 gmat 是否相同?我的意思是,代码会在内存中使用 16 x 100 MB 来存储每个进程的 gmat 还是只使用 100 MB,因为我已将 gmat 定义为 global ?而且我不希望不同的进程从文件中单独读取 gmat,因为读取这么多数字需要时间。有什么更好的方法来做到这一点?谢谢。