Splitting of MPI communicators

谁借莪１个温暖的怀抱￠ 2022-05-15 11:39 128阅读 0赞

有时候为了创建2D的拓扑，需要对MPI\_COMM\_WORLD进行划分。所使用的MPI函数是：

**\[cpp\]**  [view plain][] [copy][view plain]

1.  MPI\_Comm\_split(MPI\_Comm comm, **int** color, **int** key, MPI\_Comm \*newcomm)

例子程序（出自[MPI Groups, Communicators and Topologies][MPI Groups_ Communicators and Topologies]）：

**\[cpp\]**  [view plain][] [copy][view plain]

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. \}  
34.

[view plain]: http://blog.csdn.net/lvhuiwei/article/details/7868070#
[MPI Groups_ Communicators and Topologies]: http://www.cs.utexas.edu/~pingali/CS378/2011sp/lectures/MPIGroups.ppt