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 );
41 for (testnum=0; msgsizes[testnum] > 0; testnum++) {
42 if (rank == source || rank == dest) {
43 int nmsg = nmsgs[testnum];
44 int msgSize = msgsizes[testnum];
45 MPI_Request r[MAX_NMSGS];
48 for (i=0; i<nmsg; i++) {
49 buf[i] = (int *)malloc( msgSize * sizeof(int) );
51 fprintf( stderr, "Unable to allocate %d bytes\n",
53 MPI_Abort( MPI_COMM_WORLD, 1 );
57 partner = (rank + 1) % size;
59 MPI_Sendrecv( MPI_BOTTOM, 0, MPI_INT, partner, 10,
60 MPI_BOTTOM, 0, MPI_INT, partner, 10, comm,
62 /* Try to fill up the outgoing message buffers */
63 for (i=0; i<nmsg; i++) {
64 MPI_Isend( buf[i], msgSize, MPI_INT, partner, testnum, comm,
67 for (i=0; i<nmsg; i++) {
68 MPI_Recv( buf[i], msgSize, MPI_INT, partner, testnum, comm,
71 MPI_Waitall( nmsg, r, MPI_STATUSES_IGNORE );
73 /* Repeat the test, but make one of the processes sleep */
74 MPI_Sendrecv( MPI_BOTTOM, 0, MPI_INT, partner, 10,
75 MPI_BOTTOM, 0, MPI_INT, partner, 10, comm,
77 if (rank == dest) MTestSleep( 1 );
78 /* Try to fill up the outgoing message buffers */
80 for (i=0; i<nmsg; i++) {
81 MPI_Isend( buf[i], msgSize, MPI_INT, partner, testnum, comm,
84 tsend = MPI_Wtime() - tsend;
85 for (i=0; i<nmsg; i++) {
86 MPI_Recv( buf[i], msgSize, MPI_INT, partner, testnum, comm,
89 MPI_Waitall( nmsg, r, MPI_STATUSES_IGNORE );
92 printf( "Isends for %d messages of size %d took too long (%f seconds)\n", nmsg, msgSize, tsend );
95 MTestPrintfMsg( 1, "%d Isends for size = %d took %f seconds\n",
96 nmsg, msgSize, tsend );
98 for (i=0; i<nmsg; i++) {
104 MTest_Finalize( errs );