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
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include "mpi.h"
#include <stdio.h>
#include "mpitest.h"
/*
static char MTEST_Descrip[] = "Test the routines to control error handlers on windows";
*/
static int calls = 0;
static int errs = 0;
static MPI_Win mywin;
static int expected_err_class = MPI_ERR_OTHER;
static int w1Called = 0;
static int w2Called = 0;
void weh1(MPI_Win * win, int *err, ...);
void weh1(MPI_Win * win, int *err, ...)
{
int errclass;
w1Called++;
MPI_Error_class(*err, &errclass);
if (errclass != expected_err_class) {
errs++;
printf("Unexpected error code (class = %d)\n", errclass);
}
if (*win != mywin) {
errs++;
printf("Unexpected window\n");
}
calls++;
return;
}
void weh2(MPI_Win * win, int *err, ...);
void weh2(MPI_Win * win, int *err, ...)
{
int errclass;
w2Called++;
MPI_Error_class(*err, &errclass);
if (errclass != expected_err_class) {
errs++;
printf("Unexpected error code (class = %d)\n", errclass);
}
if (*win != mywin) {
errs++;
printf("Unexpected window\n");
}
calls++;
return;
}
int main(int argc, char *argv[])
{
int buf[2];
MPI_Win win;
MPI_Comm comm;
MPI_Errhandler newerr1, newerr2, olderr;
MTEST_VG_MEM_INIT(buf, 2 * sizeof(int));
MTest_Init(&argc, &argv);
comm = MPI_COMM_WORLD;
MPI_Win_create_errhandler(weh1, &newerr1);
MPI_Win_create_errhandler(weh2, &newerr2);
MPI_Win_create(buf, 2 * sizeof(int), sizeof(int), MPI_INFO_NULL, comm, &win);
mywin = win;
MPI_Win_get_errhandler(win, &olderr);
if (olderr != MPI_ERRORS_ARE_FATAL) {
errs++;
printf("Expected errors are fatal\n");
}
MPI_Win_set_errhandler(win, newerr1);
/* We should be able to free the error handler now since the window is
* using it */
MPI_Errhandler_free(&newerr1);
expected_err_class = MPI_ERR_RANK;
MPI_Put(buf, 1, MPI_INT, -5, 0, 1, MPI_INT, win);
if (w1Called != 1) {
errs++;
printf("newerr1 not called\n");
w1Called = 1;
}
expected_err_class = MPI_ERR_OTHER;
MPI_Win_call_errhandler(win, MPI_ERR_OTHER);
if (w1Called != 2) {
errs++;
printf("newerr1 not called (2)\n");
}
if (w1Called != 2 || w2Called != 0) {
errs++;
printf("Error handler weh1 not called the expected number of times\n");
}
/* Try another error handler. This should allow the MPI implementation to
* free the first error handler */
MPI_Win_set_errhandler(win, newerr2);
MPI_Errhandler_free(&newerr2);
expected_err_class = MPI_ERR_RANK;
MPI_Put(buf, 1, MPI_INT, -5, 0, 1, MPI_INT, win);
if (w2Called != 1) {
errs++;
printf("newerr2 not called\n");
calls = 1;
}
expected_err_class = MPI_ERR_OTHER;
MPI_Win_call_errhandler(win, MPI_ERR_OTHER);
if (w2Called != 2) {
errs++;
printf("newerr2 not called (2)\n");
}
if (w1Called != 2 || w2Called != 2) {
errs++;
printf("Error handler weh1 not called the expected number of times\n");
}
MPI_Win_free(&win);
MTest_Finalize(errs);
return MTestReturnValue(errs);
}
|
