[mpich-commits] r10617 - mpich2/trunk/test/mpi/coll

[gropp at mcs.anl.gov]

Author: gropp
Date: 2012-11-19 10:28:07 -0600 (Mon, 19 Nov 2012)
New Revision: 10617

Added:
   mpich2/trunk/test/mpi/coll/uoplong.c
Modified:
   mpich2/trunk/test/mpi/coll/Makefile.am
   mpich2/trunk/test/mpi/coll/testlist
Log:
Added test for user-defined operations with large count

Modified: mpich2/trunk/test/mpi/coll/Makefile.am
===================================================================
--- mpich2/trunk/test/mpi/coll/Makefile.am	2012-11-19 16:22:13 UTC (rev 10616)
+++ mpich2/trunk/test/mpi/coll/Makefile.am	2012-11-19 16:28:07 UTC (rev 10617)
@@ -99,6 +99,7 @@
     scatter3           \
     scattern           \
     scatterv           \
+    uoplong            \
     nballtoall1        \
     nbredscat          \
     nbredscat3	       \

Modified: mpich2/trunk/test/mpi/coll/testlist
===================================================================
--- mpich2/trunk/test/mpi/coll/testlist	2012-11-19 16:22:13 UTC (rev 10616)
+++ mpich2/trunk/test/mpi/coll/testlist	2012-11-19 16:28:07 UTC (rev 10617)
@@ -125,6 +125,9 @@
 opminloc 4
 opmax 5
 opmaxloc 5
+uoplong 4
+uoplong 11
+uoplong 16
 nonblocking 4 mpiversion=3.0
 nonblocking 5 mpiversion=3.0
 nonblocking 10 mpiversion=3.0

Added: mpich2/trunk/test/mpi/coll/uoplong.c
===================================================================
--- mpich2/trunk/test/mpi/coll/uoplong.c	                        (rev 0)
+++ mpich2/trunk/test/mpi/coll/uoplong.c	2012-11-19 16:28:07 UTC (rev 10617)
@@ -0,0 +1,109 @@
+/* -*- Mode: C; c-basic-offset:4 ; -*- */
+/*
+ *  (C) 2012 by Argonne National Laboratory.
+ *      See COPYRIGHT in top-level directory.
+ */
+
+#include "mpi.h"
+#include "mpitest.h"
+#include <stdio.h>
+#include <stdlib.h>
+
+/* 
+ * Test user-defined operations with a large number of elements.  
+ * Added because a talk at EuroMPI'12 claimed that these failed with
+ * more than 64k elements
+ */
+
+#define MAX_ERRS 10
+#define MAX_COUNT 1200000
+
+void myop( void *cinPtr, void *coutPtr, int *count, MPI_Datatype *dtype );
+
+/* 
+ * myop takes a datatype that is a triple of doubles, and computes
+ * the sum, max, min of the respective elements of the triple.
+ */
+void myop( void *cinPtr, void *coutPtr, int *count, MPI_Datatype *dtype )
+{
+    int i, n = *count;
+    double const *cin = (double *)cinPtr;
+    double *cout = (double *)coutPtr;
+    
+    for (i=0; i<n; i++) {
+	cout[0] += cin[0];
+	cout[1] = (cout[1] > cin[1]) ? cout[1] : cin[1];
+	cout[2] = (cout[2] < cin[2]) ? cout[2] : cin[2];
+	cin  += 3;
+	cout += 3;
+    }
+}
+
+int main( int argc, char *argv[] )
+{
+    int errs = 0;
+    int wsize, wrank, i, count;
+    MPI_Datatype tripleType;
+    double *inVal, *outVal;
+    double maxval, sumval;
+    MPI_Op op;
+
+    MTest_Init( &argc, &argv );
+    MPI_Op_create( myop, 0, &op );
+    MPI_Type_contiguous( 3, MPI_DOUBLE, &tripleType );
+    MPI_Type_commit( &tripleType );
+
+    MPI_Comm_size( MPI_COMM_WORLD, &wsize );
+    MPI_Comm_rank( MPI_COMM_WORLD, &wrank );
+
+    for (count=1; count<MAX_COUNT; count += count) {
+	if (wrank == 0) 
+	    MTestPrintfMsg( 1, "Count = %d\n", count );
+	inVal = (double *)malloc( 3 * count * sizeof(double) );
+	outVal = (double *)malloc( 3 * count * sizeof(double) );
+	if (!inVal || !outVal) {
+	    fprintf( stderr, "Unable to allocated %d words for data\n", 
+		     3 * count );
+	    MPI_Abort( MPI_COMM_WORLD, 1 );
+	}
+	for (i=0; i<count*3; i++) {
+	    outVal[i] = -1;
+	    inVal[i]  = 1 + (i & 0x3);
+	}
+	MPI_Reduce( inVal, outVal, count, tripleType, op, 0, MPI_COMM_WORLD );
+	/* Check Result values */
+	if (wrank == 0) {
+	    for (i=0; i<3*count; i+=3) {
+		sumval = wsize * (1 + (i & 0x3));
+		maxval = 1 + ((i+1) & 0x3);
+		if (outVal[i] != sumval) {
+		    if (errs < MAX_ERRS) 
+			fprintf( stderr, "%d: outval[%d] = %f, expected %f (sum)\n", 
+				 count, i, outVal[i], sumval );
+		    errs++;
+		}
+		if (outVal[i+1] != maxval) {
+		    if (errs < MAX_ERRS) 
+			fprintf( stderr, "%d: outval[%d] = %f, expected %f (max)\n", 
+				 count, i+1, outVal[i+1], maxval );
+		    errs++;
+		}
+		if (outVal[i+2] != 1 + ((i+2)&0x3)) {
+		    if (errs < MAX_ERRS) 
+			fprintf( stderr, "%d: outval[%d] = %f, expected %f (min)\n", 
+				 count, i+2, outVal[i+2], (double)(1 + ((i+2)^0x3)) );
+		    errs++;
+		}
+	    }
+	}
+
+	free( inVal );
+	free( outVal );
+    }
+    
+    MPI_Op_free( &op );
+    MPI_Type_free( &tripleType );
+    MTest_Finalize( errs );
+    MPI_Finalize( );
+    return 0;
+}