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 <stdio.h>   
2. #include <mpi.h>  
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. }