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
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include "mpitest.h"
#include <assert.h>
#ifdef MULTI_TESTS
#define run coll_allred5
int run(const char *arg);
#endif
/*
static char MTEST_Descrip[] = "Test MPI_Allreduce with count greater than the number of processes";
*/
/* We make the error count global so that we can easily control the output
of error information (in particular, limiting it after the first 10
errors */
static int errs = 0;
int run(const char *arg)
{
MPI_Comm comm;
MPI_Datatype dtype;
int count, *bufin, *bufout, size, i, minsize = 1;
while (MTestGetIntracommGeneral(&comm, minsize, 1)) {
if (comm == MPI_COMM_NULL) {
continue;
}
MPI_Comm_size(comm, &size);
count = size * 2;
bufin = (int *) malloc(count * sizeof(int));
bufout = (int *) malloc(count * sizeof(int));
if (!bufin || !bufout) {
fprintf(stderr, "Unable to allocated space for buffers (%d)\n", count);
MPI_Abort(MPI_COMM_WORLD, 1);
}
for (i = 0; i < count; i++) {
bufin[i] = i;
bufout[i] = -1;
}
dtype = MPI_INT;
MPI_Allreduce(bufin, bufout, count, dtype, MPI_SUM, comm);
/* Check output */
for (i = 0; i < count; i++) {
if (bufout[i] != i * size) {
fprintf(stderr, "Expected bufout[%d] = %d but found %d\n", i, i * size, bufout[i]);
errs++;
}
}
free(bufin);
free(bufout);
MTestFreeComm(&comm);
}
return errs;
}
|
