Max.rar_MPI 最大值_mpi二维数组

#define MPICH_SKIP_MPICXX #include "mpi.h" #include<stdio.h> #include<math.h> #include<stdlib.h> #include<iostream.h> void main(int argc, char **argv) { MPI_Init(&argc,&argv); int rank,size; int master=0; double starttime,endtime;//用于计算运行时间 MPI_Comm_rank(MPI_COMM_WORLD,&rank); MPI_Comm_size(MPI_COMM_WORLD,&size); MPI_Status status; //主进程 if (rank==0) { int i,j; //定义行数、列数 int m=0, n=0; cout<<"This is rank "<<rank<<endl; cout<<"Please input Rows:"<<endl; scanf("%d",&m); cout<<"Please input Cols:"<<endl; scanf("%d",&n); //文件命名 char filename[30]={0}; cout<<"Please input your data file name:"<<endl; scanf("%s",filename); starttime=MPI_Wtime(); //开辟数组空间 double **array; array=(double **)malloc(sizeof(double *)*m); for(i=0;i<m;i++)array[i]=(double *)malloc(sizeof(double)*n); //文件操作 FILE *fp; //以随机数写入数组 fp=fopen(filename,"w"); for(i=0;i<m;i++) { for(j=0;j<n;j++) { array[i][j]=rand()/1000.0;//以假随机数测试 fprintf(fp,"%lf ",array[i][j]); } } fclose(fp); //读取文件中的数组 fp=fopen(filename,"r"); for(i=0;i<m;i++) { for(j=0;j<n;j++) { fscanf(fp,"%lf",&array[i][j]); } } fclose(fp); MPI_Bcast(&n,1,MPI_INT,0,MPI_COMM_WORLD);//广播向每一个进程发送列值 //将数组按行依次发送给从进程 double *buffer; int z,numsent=0; buffer=(double *)malloc(sizeof(double)*n); z=((size-1)>m)?m:size-1; for(i=1;i<=z;i++) { for(j=0;j<n;j++)buffer[j]=array[i-1][j];//取出一行数据存入缓冲区 MPI_Send(buffer,n,MPI_DOUBLE,i,i,MPI_COMM_WORLD); numsent++; } if(z<size-1)// 如果行数小于进程数 则有一部分进程不用计算 但是从进程有Recv等待接收 { for(i=z+1;i<=size-1;i++) MPI_Send(buffer,0,MPI_DOUBLE,i,0,MPI_COMM_WORLD); // 向多余的 不参与计算的从进程发送tag=0的消息 即终止信号。 } //依次接收从进程计算结果 double *rr,ans=0; int sender=0,ansrow=0; rr=(double *)malloc(sizeof(double)*m); for(i=1;i<=m;i++) { MPI_Recv(&ans,1,MPI_DOUBLE,MPI_ANY_SOURCE,MPI_ANY_TAG,MPI_COMM_WORLD,&status); sender=status.MPI_SOURCE; ansrow=status.MPI_TAG; rr[ansrow]=ans;//将各从进程运算结果存入结果数组对应位置 if(numsent<m)//若还有剩余的行没有计算，则继续发送 { for(j=0;j<n;j++)buffer[j]=array[numsent][j]; MPI_Send(buffer,n,MPI_DOUBLE,sender,numsent+1,MPI_COMM_WORLD); numsent++; } else//若无剩余，则每接收一个消息向相应从进程发送一个tag=0的空消息用于终止其运行 { MPI_Send(buffer,0,MPI_DOUBLE,sender,0,MPI_COMM_WORLD); } } free(buffer); free(array); //对接收的从进程计算结果进行比较得出最终结果 double max; max=rr[0]; for(i=1;i<m;i++) { if(max<rr[i])max=rr[i]; } free(rr); cout<<"Max="<<max<<endl; cout<<"Finish!"<<endl; endtime=MPI_Wtime(); cout<<"That tooks "<<endtime-starttime<<" seconds."<<endl; } //从进程 if(rank!=0) { int row,i,nn; cout<<"This is rank "<<rank<<endl; double *buffer2; double max,iabs=0; MPI_Bcast(&nn,1,MPI_INT,0,MPI_COMM_WORLD);//接收主进程广播的矩阵列数 buffer2=(double *)malloc(sizeof(double)*nn);//开辟数组以接收主进程发送的数组行 while(1) { MPI_Recv(buffer2,nn,MPI_DOUBLE,master,MPI_ANY_TAG,MPI_COMM_WORLD,&status); if(status.MPI_TAG==0)break; //开始计算最大值 row=(status.MPI_TAG)-1; max=(double)fabs(buffer2[0]); for(i=1;i<nn;i++) { iabs=(double)fabs(buffer2[i]); if(max<iabs)max=iabs; } MPI_Send(&max,1,MPI_DOUBLE,master,row,MPI_COMM_WORLD); } free(buffer2); } MPI_Finalize(); }