Splitting of MPI communicators

有时候为了创建2D的拓扑,需要对MPI_COMM_WORLD进行划分。所使用的MPI函数是:

[cpp]  view plain copy
  1. MPI_Comm_split(MPI_Comm comm, int color, int key, MPI_Comm *newcomm)   

例子程序(出自MPI Groups, Communicators and Topologies):

[cpp]  view plain copy
  1. #include    
  2. #include   
  3.   
  4. main (int argc, char **argv)  
  5. {  
  6.     MPI_Comm row_comm, col_comm;  
  7.     int myrank, size, P=4, Q=3, p, q;  
  8.   
  9.     MPI_Init(&argc, &argv);  
  10.     MPI_Comm_rank(MPI_COMM_WORLD, &myrank);  
  11.     MPI_Comm_size(MPI_COMM_WORLD, &size);  
  12.   
  13.     p = myrank / Q;  
  14.     q = myrank % Q;  
  15.   
  16.     /* Split comm intor row and column comms */  
  17.     MPI_Comm_split(MPI_COMM_WORLD, p, q, &row_comm);  
  18.     MPI_Comm_split(MPI_COMM_WORLD, q, p, &col_comm);  
  19.   
  20.     int sum, rowsum, colsum;  
  21.     sum = myrank;  
  22.     rowsum = myrank;  
  23.     colsum = myrank;  
  24.     MPI_Allreduce(MPI_IN_PLACE, &sum, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD);  
  25.     MPI_Allreduce(MPI_IN_PLACE, &rowsum, 1, MPI_INT, MPI_SUM, row_comm);  
  26.     MPI_Allreduce(MPI_IN_PLACE, &colsum, 1, MPI_INT, MPI_SUM, col_comm);  
  27.     if (myrank == 0)  
  28.         printf("The sum of all rank is %d\n", sum);  
  29.   
  30.     printf("[%d]: My coordinates are (%d, %d), row sum is %d, col sum is %d\n",myrank,p,q,  
  31. rowsum,colsum);  
  32.     MPI_Finalize();  
  33. }  

你可能感兴趣的:(MPI)