X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/ab6013dda2f80b21aff83d529738b048e508828f..9a5d187c1f279fb658e4b86dbcf84e7cb73c722c:/examples/smpi/sendrecv.c

diff --git a/examples/smpi/sendrecv.c b/examples/smpi/sendrecv.c
index 7b11ce0bac..ae9e297194 100644
--- a/examples/smpi/sendrecv.c
+++ b/examples/smpi/sendrecv.c
@@ -1,26 +1,70 @@
+/* Copyright (c) 2009, 2010. The SimGrid Team.
+ * All rights reserved.                                                     */
+
+/* This program is free software; you can redistribute it and/or modify it
+ * under the terms of the license (GNU LGPL) which comes with this package. */
+
 #include "mpi.h"
 #include <stdio.h>
- 
-int main(int argc, char *argv[])
+
+static int test(int myid, int numprocs)
 {
-#define TAG_RCV 998
+// The tags should match on the sender and receiver side.
+// The distinction between sendtag and recvtag is mainly
+// useful to make some other Recv or Send calls match the sendrecv. 
+#define TAG_RCV 999
 #define TAG_SND 999
-    int myid, numprocs, left, right;
-    int buffer[10], buffer2[10];
-    MPI_Request request;
-    MPI_Status status;
- 
-    MPI_Init(&argc,&argv);
-    MPI_Comm_size(MPI_COMM_WORLD, &numprocs);
-    MPI_Comm_rank(MPI_COMM_WORLD, &myid);
- 
-    right = (myid + 1) % numprocs;
-    left = myid - 1;
-    if (left < 0)
-        left = numprocs - 1;
- 
-    MPI_Sendrecv(buffer, 10, MPI_INT, left, TAG_SND, buffer2, 10, MPI_INT, right, TAG_RCV, MPI_COMM_WORLD, &status);
- 
-    MPI_Finalize();
-    return 0;
+
+
+#define BUFLEN 10
+  int left, right;
+  int buffer[BUFLEN], buffer2[BUFLEN];
+  int i;
+  MPI_Status status;
+
+  for (i = 0; i < BUFLEN; i++) {
+    buffer[i] = myid;
+  }
+
+  right = (myid + 1) % numprocs;
+  left = myid - 1;
+  if (left < 0)
+    left = numprocs - 1;
+
+  /* performs a right-to-left shift of vectors */
+  MPI_Sendrecv(buffer, 10, MPI_INT, left, TAG_SND, buffer2, 10, MPI_INT,
+               right, TAG_RCV, MPI_COMM_WORLD, &status);
+
+  for (i = 0; i < BUFLEN; i++) {
+    if (buffer2[i] != right) {
+      fprintf(stderr, "[%d] error: should have values %d, has %d\n", myid,
+              right, buffer2[i]);
+      return (0);
+    }
+  }
+  return (1);
+}
+
+int main(int argc, char *argv[])
+{
+
+  int myid, numprocs;
+
+  MPI_Init(&argc, &argv);
+  MPI_Comm_size(MPI_COMM_WORLD, &numprocs);
+  MPI_Comm_rank(MPI_COMM_WORLD, &myid);
+
+
+  if (0 == myid)
+    printf("\n    *** MPI_Sendrecv test ***\n\n");
+
+  if (test(myid, numprocs)) {
+    fprintf(stderr, "[%d] ok.\n", myid);
+  } else {
+    fprintf(stderr, "[%d] failed.\n", myid);
+  }
+
+
+  MPI_Finalize();
+  return 0;
 }