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 );
50 fprintf( stderr, "Unable to allocate %d bytes\n",
52 MPI_Abort( MPI_COMM_WORLD, 1 );
55 partner = (rank + 1) % size;
57 MPI_Sendrecv( MPI_BOTTOM, 0, MPI_INT, partner, 10,
58 MPI_BOTTOM, 0, MPI_INT, partner, 10, comm,
60 /* Try to fill up the outgoing message buffers */
61 for (i=0; i<nmsg; i++) {
62 MPI_Isend( buf[i], msgSize, MPI_CHAR, partner, testnum, comm,
65 for (i=0; i<nmsg; i++) {
66 MPI_Recv( buf[i], msgSize, MPI_CHAR, partner, testnum, comm,
69 MPI_Waitall( nmsg, r, MPI_STATUSES_IGNORE );
71 /* Repeat the test, but make one of the processes sleep */
72 MPI_Sendrecv( MPI_BOTTOM, 0, MPI_INT, partner, 10,
73 MPI_BOTTOM, 0, MPI_INT, partner, 10, comm,
75 if (rank == dest) MTestSleep( 1 );
76 /* Try to fill up the outgoing message buffers */
78 for (i=0; i<nmsg; i++) {
79 MPI_Isend( buf[i], msgSize, MPI_CHAR, partner, testnum, comm,
82 tsend = MPI_Wtime() - tsend;
83 for (i=0; i<nmsg; i++) {
84 MPI_Recv( buf[i], msgSize, MPI_CHAR, partner, testnum, comm,
87 MPI_Waitall( nmsg, r, MPI_STATUSES_IGNORE );
90 printf( "Isends for %d messages of size %d took too long (%f seconds)\n", nmsg, msgSize, tsend );
93 MTestPrintfMsg( 1, "%d Isends for size = %d took %f seconds\n",
94 nmsg, msgSize, tsend );
96 for (i=0; i<nmsg; i++) {
102 MTest_Finalize( errs );