基于MPI求解pi

#include
#include
#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;
}

程序运行图:
基于MPI求解pi_第1张图片

你可能感兴趣的:(并行计算)