mpi

使用MPI，計算cos x 函數的積分值，積分區間爲（0,2PI）
這裏寫圖片描述
基本思路：

 

 

把積分區間，分爲相等若干塊（此處起多少個線程，分多少塊），每一個線程分得一塊積分區域，每塊在分若干小塊（此處定義分10塊）。每一個線程計算每一個小塊的面積之和，返回給0號線程。
所得，即爲cos(x )在積分域上的值。

代碼實現：

#include<stdio.h>
#include<stdlib.h>
#include<malloc.h>
#include<math.h>
#include"mpi.h"
#define PI 3.1415
int main(int argc,char** argv)
{
        int rank,p,i;     //
    double* sbuf;     //
    double rbuf[10];
    double add;       //

        double  h;            //
    int a;            //
    double sum ;      //
    double* ssum;
    double asum    ;
    int n=10;         
        //初始化
        MPI_Init(&argc,&argv);
        //從命令行獲取開啓的線程數
        MPI_Comm_size(MPI_COMM_WORLD,&p);
        //獲取本線程的線程號
        MPI_Comm_rank(MPI_COMM_WORLD,&rank);
    add= (2*PI)/p;
        if(rank==0)
        {
        ssum=(double*)malloc(sizeof(double)*p);
            sbuf=(double*)malloc(sizeof(double)*p);

       printf("Process's begin x is ");
      for(i=0;i<p;i++)
      {
      //計算各塊起始x
       sbuf[i]=add*(i);
      printf(" %lf ",sbuf[i]);
         }
          printf("\n");
         }
        //向各個線程分發起始x值
        MPI_Scatter(sbuf,1,MPI_DOUBLE,rbuf,1,MPI_DOUBLE,0,MPI_COMM_WORLD);

    h=(add)/n;
    for(i=0;i<n;i++)    
    {
    double x=rbuf[0]+h*i+h/2;
    sum=sum+cos(x)*h;
     }
    printf("Process %d 's sum is %.2lf\n",rank,sum);
    //收集各線程計算結果
    MPI_Gather(&sum,1,MPI_DOUBLE,ssum,1,MPI_DOUBLE,0,MPI_COMM_WORLD);
    if(rank==0)
    {
      for(i=0;i<p;i++)
      {

       asum=asum+ssum[i];
       }
      printf("cos x 0~2PI de ji fen shi %.2lf\n",asum);
      }        
    MPI_Finalize();
    return 0;
}

 

 
程序主要是經過計算出各個積分域塊的起始x值，向各個線程分發，這樣各個線程就能夠負責一個單獨的積分域。
代碼很簡單，很容易看懂。

結果：

---------------------
做者：熱血小碼哥
來源：CSDN
原文：https://blog.csdn.net/mmayanshuo/article/details/72853324
版權聲明：本文爲博主原創文章，轉載請附上博文連接！