*/
#include <stdio.h>
#include "mpi.h"
-#include "mpitest.h"
+#include "mpitest.h"
#define BUFSIZE 2000
-int main( int argc, char *argv[] )
+int main(int argc, char *argv[])
{
MPI_Status status;
- MPI_Comm comm,scomm;
+ MPI_Comm comm, scomm;
int a[10], b[10];
- int buf[BUFSIZE], *bptr, bl, i, j, rank, size, color, errs=0;
+ int buf[BUFSIZE], *bptr, bl, i, j, rank, size, color, errs = 0;
- MTest_Init( 0, 0 );
- MPI_Comm_rank( MPI_COMM_WORLD, &rank );
+ MTest_Init(0, 0);
+ MPI_Comm_rank(MPI_COMM_WORLD, &rank);
color = rank % 2;
- MPI_Comm_split( MPI_COMM_WORLD, color, rank, &scomm );
- MPI_Intercomm_create( scomm, 0, MPI_COMM_WORLD, 1-color, 52, &comm);
- MPI_Comm_rank( comm, &rank );
- MPI_Comm_remote_size( comm, &size );
- MPI_Buffer_attach( buf, BUFSIZE );
-
- for (j=0; j<10; j++) {
- for (i=0; i<10; i++) {
- a[i] = (rank + 10 * j) * size + i;
- }
- MPI_Bsend( a, 10, MPI_INT, 0, 27+j, comm );
+ MPI_Comm_split(MPI_COMM_WORLD, color, rank, &scomm);
+ MPI_Intercomm_create(scomm, 0, MPI_COMM_WORLD, 1 - color, 52, &comm);
+ MPI_Comm_rank(comm, &rank);
+ MPI_Comm_remote_size(comm, &size);
+ MPI_Buffer_attach(buf, BUFSIZE);
+
+ for (j = 0; j < 10; j++) {
+ for (i = 0; i < 10; i++) {
+ a[i] = (rank + 10 * j) * size + i;
+ }
+ MPI_Bsend(a, 10, MPI_INT, 0, 27 + j, comm);
}
if (rank == 0) {
- for (i=0; i<size; i++) {
- for (j=0; j<10; j++) {
- int k;
- status.MPI_TAG = -10;
- status.MPI_SOURCE = -20;
- MPI_Recv( b, 10, MPI_INT, i, 27+j, comm, &status );
-
- if (status.MPI_TAG != 27+j) {
- errs++;
- printf( "Wrong tag = %d\n", status.MPI_TAG );
- }
- if (status.MPI_SOURCE != i) {
- errs++;
- printf( "Wrong source = %d\n", status.MPI_SOURCE );
- }
- for (k=0; k<10; k++) {
- if (b[k] != (i + 10 * j) * size + k) {
- errs++;
- printf( "received b[%d] = %d from %d tag %d\n",
- k, b[k], i, 27+j );
- }
- }
- }
- }
+ for (i = 0; i < size; i++) {
+ for (j = 0; j < 10; j++) {
+ int k;
+ status.MPI_TAG = -10;
+ status.MPI_SOURCE = -20;
+ MPI_Recv(b, 10, MPI_INT, i, 27 + j, comm, &status);
+
+ if (status.MPI_TAG != 27 + j) {
+ errs++;
+ printf("Wrong tag = %d\n", status.MPI_TAG);
+ }
+ if (status.MPI_SOURCE != i) {
+ errs++;
+ printf("Wrong source = %d\n", status.MPI_SOURCE);
+ }
+ for (k = 0; k < 10; k++) {
+ if (b[k] != (i + 10 * j) * size + k) {
+ errs++;
+ printf("received b[%d] = %d from %d tag %d\n", k, b[k], i, 27 + j);
+ }
+ }
+ }
+ }
}
- MPI_Buffer_detach( &bptr, &bl );
+ MPI_Buffer_detach(&bptr, &bl);
MPI_Comm_free(&scomm);
MPI_Comm_free(&comm);
- MTest_Finalize( errs );
-
+ MTest_Finalize(errs);
+
MPI_Finalize();
return 0;
}