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 static char MTEST_Descrip[] = "Head to head send-recv to test backoff in device when large messages are being transferred";
17 int main(int argc, char *argv[])
20 int rank, size, source, dest, partner;
23 static int msgsizes[] = { 100, 1000, 10000, 100000, -1 };
24 static int nmsgs[] = { 100, 10, 10, 4 };
27 MTest_Init(&argc, &argv);
29 comm = MPI_COMM_WORLD;
31 MPI_Comm_rank(comm, &rank);
32 MPI_Comm_size(comm, &size);
36 printf("This test requires at least 2 processes\n");
37 MPI_Abort(MPI_COMM_WORLD, 1);
40 for (testnum = 0; msgsizes[testnum] > 0; testnum++) {
41 if (rank == source || rank == dest) {
42 int nmsg = nmsgs[testnum];
43 int msgSize = msgsizes[testnum];
44 MPI_Request r[MAX_NMSGS];
47 for (i = 0; i < nmsg; i++) {
48 buf[i] = (int *) malloc(msgSize * sizeof(int));
50 fprintf(stderr, "Unable to allocate %d bytes\n", msgSize);
51 MPI_Abort(MPI_COMM_WORLD, 1);
53 MTEST_VG_MEM_INIT(buf[i], msgSize * sizeof(int));
55 partner = (rank + 1) % size;
57 MPI_Sendrecv(MPI_BOTTOM, 0, MPI_INT, partner, 10,
58 MPI_BOTTOM, 0, MPI_INT, partner, 10, comm, MPI_STATUS_IGNORE);
59 /* Try to fill up the outgoing message buffers */
60 for (i = 0; i < nmsg; i++) {
61 MPI_Isend(buf[i], msgSize, MPI_INT, partner, testnum, comm, &r[i]);
63 for (i = 0; i < nmsg; i++) {
64 MPI_Recv(buf[i], msgSize, MPI_INT, partner, testnum, comm, MPI_STATUS_IGNORE);
66 MPI_Waitall(nmsg, r, MPI_STATUSES_IGNORE);
68 /* Repeat the test, but make one of the processes sleep */
69 MPI_Sendrecv(MPI_BOTTOM, 0, MPI_INT, partner, 10,
70 MPI_BOTTOM, 0, MPI_INT, partner, 10, comm, MPI_STATUS_IGNORE);
73 /* Try to fill up the outgoing message buffers */
75 for (i = 0; i < nmsg; i++) {
76 MPI_Isend(buf[i], msgSize, MPI_INT, partner, testnum, comm, &r[i]);
78 tsend = MPI_Wtime() - tsend;
79 for (i = 0; i < nmsg; i++) {
80 MPI_Recv(buf[i], msgSize, MPI_INT, partner, testnum, comm, MPI_STATUS_IGNORE);
82 MPI_Waitall(nmsg, r, MPI_STATUSES_IGNORE);
85 printf("Isends for %d messages of size %d took too long (%f seconds)\n", nmsg,
89 MTestPrintfMsg(1, "%d Isends for size = %d took %f seconds\n", nmsg, msgSize, tsend);
91 for (i = 0; i < nmsg; i++) {