+/**
+ * MESSAGE PASSING INTERFACE TEST CASE SUITE
+ *
+ * Copyright IBM Corp. 1995
+ *
+ * IBM Corp. hereby grants a non-exclusive license to use, copy, modify, and
+ *distribute this software for any purpose and without fee provided that the
+ *above copyright notice and the following paragraphs appear in all copies.
+
+ * IBM Corp. makes no representation that the test cases comprising this
+ * suite are correct or are an accurate representation of any standard.
+
+ * In no event shall IBM be liable to any party for direct, indirect, special
+ * incidental, or consequential damage arising out of the use of this software
+ * even if IBM Corp. has been advised of the possibility of such damage.
+
+ * IBM CORP. SPECIFICALLY DISCLAIMS ANY WARRANTIES INCLUDING, BUT NOT LIMITED
+ * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE. THE SOFTWARE PROVIDED HEREUNDER IS ON AN "AS IS" BASIS AND IBM
+ * CORP. HAS NO OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT, UPDATES,
+ * ENHANCEMENTS, OR MODIFICATIONS.
+ * ***************************************************************************
+ **/
+#include <stdio.h>
+#include <mpi.h>
+
+int ibm_test(int rank, int size) {
+
+#define MAXLEN 10000
+
+ int success = 1;
+ int root,i,j,k;
+ int out[MAXLEN*64];
+ int in[MAXLEN];
+
+
+ for(j=1,root=0;j<=MAXLEN;j*=10,root=(root+1)%size) {
+ if(rank == root)
+ for(i=0;i<j*size;i++) out[i] = i;
+
+ MPI_Scatter(out,j,MPI_INT,in,j,MPI_INT,root,MPI_COMM_WORLD);
+
+ for(k=0;k<j;k++) {
+ if(in[k] != k+rank*j) {
+ fprintf(stderr,"task %d bad answer (%d) at index %d k of %d (should be %d)", rank,in[k],k,j,(k+rank*j));
+ return( 0 );
+ }
+ }
+ }
+ return( success );
+ MPI_Barrier( MPI_COMM_WORLD );
+}
+
+/**
+ * small test: the root sends a single distinct double to other processes
+ **/
+int small_test( int rank, int size ) {
+ int success=1;
+ int sendcount=size;
+ int recvcount=1;
+ int i;
+ double *sndbuf =NULL;
+ double rcvd;
+ int root=0; // arbitrary choice
+
+ // on root, initialize sendbuf
+ if (root == rank ) {
+ sndbuf = malloc( sendcount*sizeof(double));
+ for (i=0; i<sendcount;i++) {
+ sndbuf[i] = (double) rank;
+ }
+ }
+
+ MPI_Scatter(sndbuf, sendcount, MPI_DOUBLE, &rcvd,recvcount,MPI_DOUBLE,root,MPI_COMM_WORLD);
+
+ return(success);
+}
+
+
+
+int main(int argc, char **argv)
+{
+ int size, rank;
+
+
+ MPI_Init(&argc, &argv);
+ MPI_Comm_size(MPI_COMM_WORLD, &size);
+ MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+
+
+
+ if (0 == rank)
+ printf("** Small Test Result: ... \n");
+ small_test( rank, size );
+
+
+ if (0 == rank)
+ printf("** IBM Test Result: ... \n");
+/* if (!ibm_test(rank, size))
+ printf("\t[%d] failed.\n", rank);
+ else
+ printf("\t[%d] ok.\n", rank);
+*/
+ MPI_Finalize();
+ return 0;
+}