基于MPI求解pi
程序员文章站
2022-03-15 19:18:44
...
#include<stdio.h>
#include<math.h>
#include "mpi.h"
//给定函数
//求解横坐标为xi时的值
double func(double xi)
{
return (4.0 / (1.0 + xi*xi));
}
int main(int argc,char* argv[])
{
//n:需要求解的点;myid:当前的进程id;numprocs:进程的数量
int n=10000,myid,numprocs;
//pi:需要求解的结果
//h:间距
//xi:变量
//res:每个进程获得的结果
//startTime:开始运行时间
//endTime:计算结束的时间
double pi,h,xi,res,startTime,endTime;
MPI_Init(&argc,&argv);
MPI_Comm_rank(MPI_COMM_WORLD,&myid);
MPI_Comm_size(MPI_COMM_WORLD,&numprocs);
MPI_Bcast(&n,1,MPI_INT,0,MPI_COMM_WORLD);
//开始计算时计时
if(myid==0)
{
startTime=MPI_Wtime();
}
//计算间距
h = 1.0/(double)n;
//当前进程计算的结果
res = 0.0;
//每个进程负责计算自己那一部分
//比如有一万个点分配给十个进程每个进程需要完成一千个点
//每个进程从自己的进程id开始进行累加
for(int i=myid;i<n;i+=numprocs)
{
xi = h * ((double)i + 0.5);
res += func(xi);
}
res = h * res;
//归约计算将每个进程获得的结果接收并相加到根结点
//MPI_SUM即求和方式
MPI_Reduce(&res,&pi,1,MPI_DOUBLE,MPI_SUM,0,MPI_COMM_WORLD);
if(myid==0)
{
endTime = MPI_Wtime();
//记录运行时间+最终的pi值
printf("\nPI is %f\nTime is : %f\n",pi,endTime - startTime);
}
MPI_Finalize();
return 0;
}
程序运行图:
上一篇: 苹果不放弃:或将收购Intel基带部门