欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

基于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;
}

程序运行图:
基于MPI求解pi

相关标签: 并行计算