Not that long, 150 lines.
Here it is:
#include <mpi.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#define BUFLEN 25000
#define LOOPS 10
#define BUFFERS 4
#define GROUP_SIZE 4
int main(int argc, char *argv[])
{
int myid, numprocs, next, namelen;
int color, key, newid;
char buffer[BUFLEN], processor_name[MPI_MAX_PROCESSOR_NAME];
MPI_Comm world_comm, comp_comm, server_comm;
MPI_Init(&argc,&argv);
MPI_Comm_size(MPI_COMM_WORLD,&numprocs);
MPI_Comm_rank(MPI_COMM_WORLD,&myid);
MPI_Get_processor_name(processor_name,&namelen);
MPI_Comm_dup( MPI_COMM_WORLD, &world_comm );
if ( myid == 0 )
color = MPI_UNDEFINED;
else
{
color = (myid - 1) / GROUP_SIZE;
key = (myid - 1) % GROUP_SIZE;
}
MPI_Comm_split( MPI_COMM_WORLD, color, key, &comp_comm );
if ( myid == 0 || (myid - 1) % GROUP_SIZE == 0 )
color = MPI_UNDEFINED;
else
{
int r = myid - 2;
color = 1;
key = r - r / GROUP_SIZE;
}
MPI_Comm_split( MPI_COMM_WORLD, color, key, &server_comm );
if ( myid == 0 )
{
fprintf(stderr,"Master [id = %d of %d] is running on %s\n", myid,
numprocs, processor_name);
}
else
{
int s_id;
MPI_Comm_rank( comp_comm, &newid );
if ( (myid - 1) % GROUP_SIZE == 0 )
fprintf(stderr,"Compositor [id = %d, %d of %d] is running on %s\n",
myid, newid, numprocs, processor_name);
else
{
MPI_Comm_rank( server_comm, &s_id );
fprintf(stderr,"Server [id = %d, %d, %d of %d] is running on %s\n",
myid, newid, s_id, numprocs, processor_name);
}
}
if ( myid == 0 )
next = 10;
MPI_Bcast( &next, 1, MPI_INT, 0, world_comm );
fprintf(stderr,"[%d] next = %d\n", myid, next );
if ( myid > 0 )
{
int i, j;
int rank, size;
MPI_Status status;
MPI_Comm_size( comp_comm, &size );
MPI_Comm_rank( comp_comm, &rank );
if ( rank == 0 )
{
char buffer[BUFLEN];
for (i = 0; i < LOOPS * ( size - 1 ); i++)
{
int which_source, which_tag;
MPI_Probe( MPI_ANY_SOURCE, MPI_ANY_TAG, comp_comm, &status );
which_source = status.MPI_SOURCE;
which_tag = status.MPI_TAG;
printf( "Receiving buffer from %d, buffer = ", which_source );
MPI_Recv( buffer, BUFLEN, MPI_CHAR, which_source, which_tag,
comp_comm, &status );
printf( "%s\n", buffer );
}
}
else
{
MPI_Request* request[BUFFERS];
int sent[ BUFFERS ];
int index = 0;
char* buffer[BUFFERS];
for (i = 0; i < BUFFERS; i++)
{
MPI_Request* requester = (MPI_Request *) malloc( sizeof( MPI_Request
) );
char* c = (char *) malloc(BUFLEN * sizeof( MPI_Request ) );
/* Should really check for failure, but not for this test */
request[ i ] = requester;
sent[ i ] = 0;
buffer[ i ] = c;
}
for (i = 0; i < LOOPS; i++)
{
printf( "Sending buffer %d from %d\n", i, rank );
sprintf( buffer[ index ], "hello from %d for the %d time", rank, i
);
if ( sent[ index ] )
{
sent[ index ] = 0;
MPI_Wait( request[ index ], &status );
}
MPI_Isend( buffer[ index ], BUFLEN, MPI_CHAR, 0, 99, comp_comm,
request[ index ] );
sent[ index ] = 1;
index = ( index + 1 ) % BUFFERS;
/* Randomly sleep to fake a computation loop*/
usleep( (unsigned long)(500000 * drand48()) );
}
/* Clean up */
for (i = 0; i < BUFFERS; i++)
{
if ( sent[ i ] )
{
sent[ i ] = 0;
MPI_Wait( request[ i ], &status );
}
free( request[ i ] );
free( buffer[ i ] );
}
}
}
MPI_Barrier( world_comm );
MPI_Finalize();
return (0);
}
On Fri, Jun 19, 2009 at 10:50 AM, Eugene Loh <eugene....@sun.com> wrote:
> Mark Bolstad wrote:
>
> I'll post the test code if requested (this email is already long)
>>
>
> Yipes, how long is the test code? Short enough to send, yes? Please send.
> _______________________________________________
> users mailing list
> us...@open-mpi.org
> http://www.open-mpi.org/mailman/listinfo.cgi/users
>
