我们现在使用MPI 来对一个SIN 函数进行定积分求面积的计算。
简单说下定积分,定积分根据将X划分为一个一个的小梯形,将面积球和之后近似于目标面积。
如上图所示,我们需要的是使用MPI 将整个图形分解成N 个梯形,其中每一个梯形的面积开一个进程来计算,然后使用MPI 的规约求和函数来计算最终的结果。
#include<stdio.h>
#include<stdlib.h>
#include<mpi.h>
#include<math.h>
const double a = 0.0;
const double b = 3.14;
int n = 100;
/*计算梯形面积的函数*/
double sum_quer(double a,double b,int n,double h){
double *x = (double *)malloc(sizeof(double) * (n + 1));
double *f = (double *)malloc(sizeof(double) * (n + 1));
double inte = (sin(a) + sin(b)) /2;
for(int i = 0;i < (n+1);i++){
x[i] = x[i-1] + h;
f[i] = sin(x[i]);
inte += f[i];
}
inte = inte*h;
return 0;
}
int main(int argc,char **argv){
int myid,nprocs;
int local_n;
double local_a;
double local_b;
double total_inte;
MPI_Init(&argc,&argv); /*初始化MPI 系统*/
MPI_Comm_rank(MPI_COMM_WORLD,&myid); /*设置当前通信子*/
MPI_Comm_size(MPI_COMM_WORLD,&nprocs);/*设置工作进程个数*/
local_n = n / nprocs;
int h = (b - a)/n;
local_a = a + myid *local_n*h;
local_b = local_a + local_n * h;
double local_inte = sum_quer(local_a,local_b,local_n,h);
MPI_Reduce(&local_inte,&total_inte,1,MPI_DOUBLE,MPI_SUM,0,MPI_COMM_WORLD); /*规约各个计算结果*/
if(myid == 0){
printf("integral out put is %d",total_inte);
}
MPI_Finalize();
return 0;
}
这里还有几个小点需要注意。
关于IO处理。
我们的程序是SPMD 类型的,程序只需要编写一次然后多进程执行罢了,但是如果我们需要输入数据呢?比如使用scanf 来输入数据?
解决方法:给0号进程编写scanf 函数,其他的进程使用 MPI_Send 来接受变量。
#include<stdio.h>
#include<mpi.h>
void get_data(){
int my_rank;
int comm_zz;
double *a;
double *b;
int *n;
int dest;
if(my_rank == 0){ /*如果是0号进程就接受并发送数据*/
scanf("%lf %lf %d",a,b,n);
for(dest = 0;dest < comm_zz;dest++){
MPI_Send(a,1,MPI_DOUBLE,dest,0,MPI_COMM_WORLD);
MPI_Send(b,1,MPI_DOUBLE,dest,0,MPI_COMM_WORLD);
MPI_Send(n,1,MPI_INT,dest,0,MPI_COMM_WORLD);
}
}else{ /*其他进程接受参数*/
MPI_Recv(a,1,MPI_DOUBLE,0,0,MPI_COMM_WORLD,MPI_STATUS_IGNORE);
MPI_Recv(b,1,MPI_DOUBLE,0,0,MPI_COMM_WORLD,MPI_STATUS_IGNORE);
MPI_Recv(n,1,MPI_INT,0,0,MPI_COMM_WORLD,MPI_STATUS_IGNORE);
}
}
MPI_Reduce
全局规约函数,很简单,最核心的是第五个参数操作符参数,根据不同的参数我们可以实现不同的功能。
MPI 中定义的规约操作符
MPI_MAX 最大值
MPI_MIN 最小值
MPI_SUM 累加和
MPI_PROD 累乘积
MPI_LAND 逻辑与
MPI_BAND 按位于
MPI_LOR 逻辑或
MPI_BOR 按位或
MPI_LXOR 逻辑异或
MPI_BXOR 按位异或
MPI_MAXLOC 求最大值和位置
MPI_MINLOC 求最小值和位置
集和通信与点对点通信
1.在通信子中的所有进程都必须调用相同的集和通信函数
2.每个进程传递给MPI集合通信函数的参数必须是相融的
3.参数output_data_p 只用在dest_process 上。
4.点对点通信函数是通过标签和通信子来匹配的。集合通信函数不使用标签,只通过通信子和调用顺序来进行匹配。
查看原文:http://zmrlinux.com/2017/01/24/%e4%bd%bf%e7%94%a8mpi-%e8%ae%a1%e7%ae%97%e7%a7%af%e5%88%86/