1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
/*
*
* (C) 2007 by Argonne National Laboratory.
* See COPYRIGHT in top-level directory.
*/
#include "mpi.h"
#include <stdio.h>
#include <stdlib.h>
#include "mpitest.h"
/*
* This program tests that MPI_Comm_create applies to intercommunicators;
* this is an extension added in MPI-2
*/
int main(int argc, char *argv[])
{
int errs = 0;
int size, isLeft, wrank;
MPI_Comm intercomm, newcomm;
MPI_Group oldgroup, newgroup;
MTest_Init(&argc, &argv);
MPI_Comm_size(MPI_COMM_WORLD, &size);
if (size < 4) {
printf("This test requires at least 4 processes\n");
MPI_Abort(MPI_COMM_WORLD, 1);
}
MPI_Comm_rank(MPI_COMM_WORLD, &wrank);
while (MTestGetIntercomm(&intercomm, &isLeft, 2)) {
int ranks[10], nranks, result;
if (intercomm == MPI_COMM_NULL)
continue;
MPI_Comm_group(intercomm, &oldgroup);
ranks[0] = 0;
nranks = 1;
MTestPrintfMsg(1, "Creating a new intercomm 0-0\n");
MPI_Group_incl(oldgroup, nranks, ranks, &newgroup);
MPI_Comm_create(intercomm, newgroup, &newcomm);
/* Make sure that the new communicator has the appropriate pieces */
if (newcomm != MPI_COMM_NULL) {
int new_rsize, new_size, flag, commok = 1;
MPI_Comm_set_name(newcomm, (char *) "Single rank in each group");
MPI_Comm_test_inter(intercomm, &flag);
if (!flag) {
errs++;
printf("[%d] Output communicator is not an intercomm\n", wrank);
commok = 0;
}
MPI_Comm_remote_size(newcomm, &new_rsize);
MPI_Comm_size(newcomm, &new_size);
/* The new communicator has 1 process in each group */
if (new_rsize != 1) {
errs++;
printf("[%d] Remote size is %d, should be one\n", wrank, new_rsize);
commok = 0;
}
if (new_size != 1) {
errs++;
printf("[%d] Local size is %d, should be one\n", wrank, new_size);
commok = 0;
}
/* ... more to do */
if (commok) {
errs += MTestTestComm(newcomm);
}
}
MPI_Group_free(&newgroup);
if (newcomm != MPI_COMM_NULL) {
MPI_Comm_free(&newcomm);
}
/* Now, do a sort of dup, using the original group */
MTestPrintfMsg(1, "Creating a new intercomm (manual dup)\n");
MPI_Comm_create(intercomm, oldgroup, &newcomm);
MPI_Comm_set_name(newcomm, (char *) "Dup of original");
MTestPrintfMsg(1, "Creating a new intercomm (manual dup (done))\n");
MPI_Comm_compare(intercomm, newcomm, &result);
MTestPrintfMsg(1, "Result of comm/intercomm compare is %d\n", result);
if (result != MPI_CONGRUENT) {
const char *rname = 0;
errs++;
switch (result) {
case MPI_IDENT:
rname = "IDENT";
break;
case MPI_CONGRUENT:
rname = "CONGRUENT";
break;
case MPI_SIMILAR:
rname = "SIMILAR";
break;
case MPI_UNEQUAL:
rname = "UNEQUAL";
break;
printf("[%d] Expected MPI_CONGRUENT but saw %d (%s)", wrank, result, rname);
fflush(stdout);
}
} else {
/* Try to communication between each member of intercomm */
errs += MTestTestComm(newcomm);
}
if (newcomm != MPI_COMM_NULL) {
MPI_Comm_free(&newcomm);
}
/* test that an empty group in either side of the intercomm results in
* MPI_COMM_NULL for all members of the comm */
if (isLeft) {
/* left side reuses oldgroup, our local group in intercomm */
MPI_Comm_create(intercomm, oldgroup, &newcomm);
} else {
/* right side passes MPI_GROUP_EMPTY */
MPI_Comm_create(intercomm, MPI_GROUP_EMPTY, &newcomm);
}
if (newcomm != MPI_COMM_NULL) {
printf("[%d] expected MPI_COMM_NULL, but got a different communicator\n", wrank);
fflush(stdout);
errs++;
}
if (newcomm != MPI_COMM_NULL) {
MPI_Comm_free(&newcomm);
}
MPI_Group_free(&oldgroup);
MPI_Comm_free(&intercomm);
}
MTest_Finalize(errs);
return MTestReturnValue(errs);
}
|
