X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/6b514200338a9ba7dbb0f0b8f0d244bd74daa07a..0f3d85e5d91158b55c5dcd30d0400059574e4425:/teshsuite/smpi/mpich3-test/perf/non_zero_root.c

diff --git a/teshsuite/smpi/mpich3-test/perf/non_zero_root.c b/teshsuite/smpi/mpich3-test/perf/non_zero_root.c
new file mode 100644
index 0000000000..9384d6bac7
--- /dev/null
+++ b/teshsuite/smpi/mpich3-test/perf/non_zero_root.c
@@ -0,0 +1,77 @@
+/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
+/*
+ *  (C) 2008 by Argonne National Laboratory.
+ *      See COPYRIGHT in top-level directory.
+ */
+#include "mpi.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/time.h>
+
+#define SIZE 100000
+#define ITER 1000
+
+#define ERROR_MARGIN 0.5
+
+static int verbose = 0;
+
+int main(int argc, char* argv[])
+{
+	char *sbuf, *rbuf;
+	int i, j;
+	double t1, t2, t, ts;
+	int rank, size;
+	MPI_Status status;
+
+	MPI_Init(&argc,&argv);
+	MPI_Comm_rank(MPI_COMM_WORLD,&rank);
+	MPI_Comm_size(MPI_COMM_WORLD, &size);
+
+	if (getenv("MPITEST_VERBOSE")) verbose = 1;
+
+	/* Allocate memory regions to communicate */
+	sbuf = (char*) malloc(SIZE);
+	rbuf = (char*) malloc(size * SIZE);
+
+	/* Touch the buffers to make sure they are allocated */
+	for (i = 0; i < SIZE; i++) sbuf[i] = '0';
+	for (i = 0; i < SIZE * size; i++) rbuf[i] = '0';
+
+	/* Time when rank 0 gathers the data */
+	MPI_Barrier(MPI_COMM_WORLD);
+	t1 = MPI_Wtime();
+	for (i = 0; i < ITER; i++) {
+		MPI_Gather(sbuf, SIZE, MPI_BYTE, rbuf, SIZE, MPI_BYTE, 0, MPI_COMM_WORLD);
+		MPI_Barrier(MPI_COMM_WORLD);
+	}
+	t2 = MPI_Wtime();
+	t = (t2-t1)/ITER;
+
+	/* Time when rank 1 gathers the data */
+	MPI_Barrier(MPI_COMM_WORLD);
+	t1 = MPI_Wtime();
+	for (j = 0; j < ITER; j++) {
+		MPI_Gather(sbuf, SIZE, MPI_BYTE, rbuf, SIZE, MPI_BYTE, 1, MPI_COMM_WORLD);
+		MPI_Barrier(MPI_COMM_WORLD);
+	}
+	t2 = MPI_Wtime();
+	ts = (t2-t1)/ITER;
+
+	if (rank == 1)
+		MPI_Send(&ts, 1, MPI_DOUBLE, 0, 0, MPI_COMM_WORLD);
+	if (rank == 0)
+		MPI_Recv(&ts, 1, MPI_DOUBLE, 1, 0, MPI_COMM_WORLD, &status);
+
+	/* Print out the results */
+	if (!rank) {
+		if ((ts / t) > (1 + ERROR_MARGIN)) { /* If the difference is more than 10%, it's an error */
+			printf("%.3f\t%.3f\n", 1000000.0 * ts, 1000000.0 * t);
+			printf("Too much difference in performance\n");
+		}
+		else printf(" No Errors\n");
+	}
+	
+	MPI_Finalize();
+
+	return 0;
+}