1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
4 * (C) 2003 by Argonne National Laboratory.
5 * See COPYRIGHT in top-level directory.
13 /* Gather data from a vector to contiguous. Use IN_PLACE */
15 int main(int argc, char **argv)
18 double *vecin, *vecout;
20 int count, minsize = 2;
21 int root, i, n, stride, errs = 0;
24 MTest_Init(&argc, &argv);
26 while (MTestGetIntracommGeneral(&comm, minsize, 1)) {
27 if (comm == MPI_COMM_NULL)
29 /* Determine the sender and receiver */
30 MPI_Comm_rank(comm, &rank);
31 MPI_Comm_size(comm, &size);
33 for (root = 0; root < size; root++) {
34 for (count = 1; count < 65000; count = count * 2) {
37 vecin = (double *) malloc(n * stride * size * sizeof(double));
38 vecout = (double *) malloc(size * n * sizeof(double));
40 MPI_Type_vector(n, 1, stride, MPI_DOUBLE, &vec);
41 MPI_Type_commit(&vec);
43 for (i = 0; i < n * stride; i++)
45 for (i = 0; i < n; i++)
46 vecin[i * stride] = rank * n + i;
49 for (i = 0; i < n; i++) {
50 vecout[rank * n + i] = rank * n + i;
52 MPI_Gather(MPI_IN_PLACE, -1, MPI_DATATYPE_NULL,
53 vecout, n, MPI_DOUBLE, root, comm);
56 MPI_Gather(vecin, 1, vec, NULL, -1, MPI_DATATYPE_NULL, root, comm);
59 for (i = 0; i < n * size; i++) {
63 fprintf(stderr, "vecout[%d]=%d\n", i, (int) vecout[i]);
76 /* do a zero length gather */
77 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
79 MPI_Gather(MPI_IN_PLACE, -1, MPI_DATATYPE_NULL, NULL, 0, MPI_BYTE, 0, MPI_COMM_WORLD);
82 MPI_Gather(NULL, 0, MPI_BYTE, NULL, 0, MPI_BYTE, 0, MPI_COMM_WORLD);