simplify alltoall

[simgrid.git] / teshsuite / smpi / coll-alltoall / coll-alltoall.c

diff --git a/teshsuite/smpi/coll-alltoall/coll-alltoall.c b/teshsuite/smpi/coll-alltoall/coll-alltoall.c
new file mode 100644 (file)
index 0000000..8a9bd73
--- /dev/null
+++ b/teshsuite/smpi/coll-alltoall/coll-alltoall.c
@@ -0,0 +1,61 @@
+/* Copyright (c) 2009-2010, 2013-2014. 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 <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <errno.h>
+#include "mpi.h"
+
+#ifndef EXIT_SUCCESS
+#define EXIT_SUCCESS 0
+#define EXIT_FAILURE 1
+#endif
+
+int main(int argc, char *argv[])
+{
+  int rank, size;
+  int i;
+  int *sb;
+  int *rb;
+  int status;
+
+  MPI_Init(&argc, &argv);
+  MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+  MPI_Comm_size(MPI_COMM_WORLD, &size);
+
+  sb = (int *) xbt_malloc(size * sizeof(int) * 2);
+  rb = (int *) xbt_malloc(size * sizeof(int) * 2);
+  
+  for (i = 0; i < size; ++i) {
+    sb[i] = rank*size + i;
+    rb[i] = 0;
+  }
+
+  printf("[%d] sndbuf=[", rank);
+  for (i = 0; i < size; i++)
+    printf("%d ", sb[i]);
+  printf("]\n");
+
+  status = MPI_Alltoall(sb, 1, MPI_INT, rb, 1, MPI_INT, MPI_COMM_WORLD);
+
+  printf("[%d] rcvbuf=[", rank);
+  for (i = 0; i < size; i++)
+    printf("%d ", rb[i]);
+  printf("]\n");
+
+
+  if (rank == 0) {
+    if (status != MPI_SUCCESS) {
+      printf("all_to_all returned %d\n", status);
+      fflush(stdout);
+    }
+  }
+  free(sb);
+  free(rb);
+  MPI_Finalize();
+  return (EXIT_SUCCESS);
+}