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
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
/* This is a simple test program to see how well balanced the output from
MPI_Dims_create is. This test can be run with a single process
and does not need mpiexec.
There is no standard result. Instead, the output should be examined to
see if the results are acceptable. For example, make sure that Dims_create
of 2*2*2 in three dimensions returns 2,2,2 (and not 1,1,8).
*/
#include "mpi.h"
#include <stdio.h>
static int verbose = 1;
int main(int argc, char *argv[])
{
int i, j, dims[3], size;
MPI_Init(&argc, &argv);
for (i = 2; i < 12; i++) {
for (j = 0; j < 2; j++)
dims[j] = 0;
MPI_Dims_create(i * i, 2, dims);
if (verbose)
printf("Dims_create(%d,2) = (%d,%d)\n", i * i, dims[0], dims[1]);
for (j = 0; j < 3; j++)
dims[j] = 0;
MPI_Dims_create(i * i * i, 3, dims);
if (verbose)
printf("Dims_create(%d,3) = (%d,%d,%d)\n", i * i * i, dims[0], dims[1], dims[2]);
}
size = 2 * 5 * 7 * 11;
for (j = 0; j < 3; j++)
dims[j] = 0;
MPI_Dims_create(size, 2, dims);
if (verbose)
printf("Dims_create(%d,2) = (%d,%d)\n", size, dims[0], dims[1]);
for (j = 0; j < 3; j++)
dims[j] = 0;
MPI_Dims_create(size, 3, dims);
if (verbose)
printf("Dims_create(%d,3) = (%d,%d,%d)\n", size, dims[0], dims[1], dims[2]);
size = 5 * 5 * 2 * 7 * 7 * 7;
for (j = 0; j < 3; j++)
dims[j] = 0;
MPI_Dims_create(size, 2, dims);
if (verbose)
printf("Dims_create(%d,2) = (%d,%d)\n", size, dims[0], dims[1]);
for (j = 0; j < 3; j++)
dims[j] = 0;
MPI_Dims_create(size, 3, dims);
if (verbose)
printf("Dims_create(%d,3) = (%d,%d,%d)\n", size, dims[0], dims[1], dims[2]);
size = 12;
for (j = 0; j < 3; j++)
dims[j] = 0;
MPI_Dims_create(size, 2, dims);
if (verbose)
printf("Dims_create(%d,2) = (%d,%d)\n", size, dims[0], dims[1]);
MPI_Finalize();
return 0;
}
|
