使用MPI实现直方图

使用MPI实现直方图:

#include <stdio.h>
#include <stdlib.h>

//mpi
#include <mpi.h>
//

#define MAXLINE 10000001

int main()
{
    double *ptd;
    char *ptc;
    FILE *fp;
    int i = 0;   
    double start, finish;
    int count[10]={0};
 int total_count[10]={0};
    int num=0,source=0;

ptd = (double *)malloc(MAXLINE * sizeof(double));
    ptc = (char *)malloc(10 * sizeof(char));


 //mpi
 int comm_sz;
 int my_rank;
 int local_num=0,local_a=0,local_b=0;
 MPI_Init(NULL,NULL);
 MPI_Comm_size(MPI_COMM_WORLD, &comm_sz);
 MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
 //
    if((fp = fopen("data.txt","r")) == NULL)
    {
      printf("Can't open \n");
      exit(1);
    }
    i=0;
    while(fgets(ptc, 10, fp) != NULL && ptc[0] != '\n' && i < MAXLINE)
    {       
        ptd[i] = atof(ptc);
        i++;
    }

num = i;
    local_b = num-1;

free(ptc);
    if(fclose(fp)!=0)
    {
        fprintf(stderr,"Error closing file\n");
    }
 
    start = MPI_Wtime();
 //mpi
 local_num = num / comm_sz;
 local_a = 0 + my_rank*local_num;
 local_b = local_a + local_num-1;
 for(i=local_a; i<local_b; i++)
 {
    count[(int)ptd[i]]++;
 }

if(my_rank != 0)
 {
    MPI_Send(count,10,MPI_INT,0,1,MPI_COMM_WORLD);
 }
 else
 {
    for(i=0;i<10;i++)
  {
      total_count[i] = count[i];
  }
  for(source=1;source<comm_sz;source++)
  {
      MPI_Recv(count,10,MPI_INT,source,1,MPI_COMM_WORLD,MPI_STATUS_IGNORE);
   for(i=0;i<10;i++)
   {
      total_count[i] += count[i];
   }
  }
 }

finish = MPI_Wtime();
 if(my_rank==0)
 {
        for(i=0;i<10;i++)
        {
            printf("%d<=x<%d: %d条\n",i,i+1,total_count[i]);
        }
  printf("It cost : %f ms\n",(finish-start)*1000.0);
 
 }
/*
 for(i=0;i<10;i++)
 {
        printf("%d<=x<%d: %d条\n",i,i+1,count[i]);
 } 
        printf("Proc %d  cost : %f ms\n",my_rank,(finish-start)*1000.0);
*/
 MPI_Finalize();
 //
    free(ptd);
    return 0;
}

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:http://www.heiqu.com/2651f63f60cde3acde117471453517ca.html