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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include "mpi.h"
#include <stdio.h>
#include <assert.h>
#include "mpitest.h"
/* This is example 10.8 from MPI Standard 4.0
* Simple example illustrating creation of an MPI communicator using the Session
* Model. The pre-defined "mpi://WORLD" process set can be used to first create
* a local MPI group and then subsequently to create an MPI communicator from
* this group.
*/
int errs = 0;
int library_foo_test(void);
void library_foo_init(void);
void library_foo_finalize(void);
int main(int argc, char *argv[])
{
#ifdef MULT_INIT
int rank, size, provided;
int num_repeat = 1;
if (argc > 1) {
num_repeat = atoi(argv[1]);
/* basic sanity check */
assert(num_repeat > 0 && num_repeat < 100);
}
MPI_Init_thread(&argc, &argv, MPI_THREAD_MULTIPLE, &provided);
MPI_Comm_size(MPI_COMM_WORLD, &size);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
for (int i = 0; i < num_repeat; i++) {
library_foo_test();
}
MPI_Finalize();
#else
int rank = library_foo_test();
#ifdef RE_INIT
library_foo_test();
#endif
#endif
if (rank == 0 && errs == 0) {
printf("No Errors\n");
}
return MTestReturnValue(errs);
}
static MPI_Session lib_shandle = MPI_SESSION_NULL;
static MPI_Comm lib_comm = MPI_COMM_NULL;
int check_thread_level(char *value)
{
int ret = MPI_SUCCESS;
if (strcmp(value, "MPI_THREAD_MULTIPLE") &&
strcmp(value, "MPI_THREAD_SERIALIZED") &&
strcmp(value, "MPI_THREAD_SINGLE") && strcmp(value, "MPI_THREAD_FUNNELED")) {
ret = MPI_ERR_OTHER;
}
return ret;
}
int library_foo_test(void)
{
int rank, size;
library_foo_init();
MPI_Comm_size(lib_comm, &size);
MPI_Comm_rank(lib_comm, &rank);
int sum;
MPI_Reduce(&rank, &sum, 1, MPI_INT, MPI_SUM, 0, lib_comm);
if (rank == 0) {
if (sum != (size - 1) * size / 2) {
printf("MPI_Reduce: expect %d, got %d\n", (size - 1) * size / 2, sum);
errs++;
}
}
library_foo_finalize();
return rank;
}
void library_foo_init(void)
{
int rc, flag;
int ret = 0;
const char pset_name[] = "mpi://WORLD";
const char mt_key[] = "thread_level";
const char mt_value[] = "MPI_THREAD_MULTIPLE";
char out_value[100]; /* large enough */
MPI_Group wgroup = MPI_GROUP_NULL;
MPI_Info sinfo = MPI_INFO_NULL;
MPI_Info tinfo = MPI_INFO_NULL;
MPI_Info_create(&sinfo);
MPI_Info_set(sinfo, mt_key, mt_value);
rc = MPI_Session_init(sinfo, MPI_ERRORS_RETURN, &lib_shandle);
if (rc != MPI_SUCCESS) {
errs++;
goto fn_exit;
}
/* check we got thread support level foo library needs */
rc = MPI_Session_get_info(lib_shandle, &tinfo);
if (rc != MPI_SUCCESS) {
errs++;
goto fn_exit;
}
MPI_Info_get(tinfo, mt_key, sizeof(out_value), out_value, &flag);
if (!flag) {
printf("Could not find key %s\n", mt_key);
errs++;
goto fn_exit;
}
if (check_thread_level(out_value)) {
printf("Did not get valid thread level, got %s\n", out_value);
errs++;
goto fn_exit;
}
/* create a group from the WORLD process set */
rc = MPI_Group_from_session_pset(lib_shandle, pset_name, &wgroup);
if (rc != MPI_SUCCESS) {
errs++;
goto fn_exit;
}
/* get a communicator */
rc = MPI_Comm_create_from_group(wgroup, "org.mpi-forum.mpi-v4_0.example-ex10_8",
MPI_INFO_NULL, MPI_ERRORS_RETURN, &lib_comm);
if (rc != MPI_SUCCESS) {
errs++;
goto fn_exit;
}
/* free group, library doesn’t need it. */
fn_exit:
if (wgroup != MPI_GROUP_NULL) {
MPI_Group_free(&wgroup);
}
if (sinfo != MPI_INFO_NULL) {
MPI_Info_free(&sinfo);
}
if (tinfo != MPI_INFO_NULL) {
MPI_Info_free(&tinfo);
}
if (ret != MPI_SUCCESS) {
MPI_Session_finalize(&lib_shandle);
}
}
void library_foo_finalize(void)
{
int rc;
rc = MPI_Comm_free(&lib_comm);
if (rc != MPI_SUCCESS) {
printf("MPI_Comm_free returned %d\n", rc);
errs++;
return;
}
rc = MPI_Session_finalize(&lib_shandle);
if (rc != MPI_SUCCESS) {
printf("MPI_Session_finalize returned %d\n", rc);
errs++;
return;
}
}
|
