Restructure teshsuite smpi

[simgrid.git] / teshsuite / smpi / vector / vector_test.c

diff --git a/teshsuite/smpi/vector/vector_test.c b/teshsuite/smpi/vector/vector_test.c
new file mode 100644 (file)
index 0000000..1025915
--- /dev/null
+++ b/teshsuite/smpi/vector/vector_test.c
@@ -0,0 +1,48 @@
+/* Copyright (c) 2012-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 <stdio.h>
+#include "mpi.h"
+#define SIZE 4
+
+int main(int argc, char **argv) {
+
+  int rank, i, j;
+  double a[SIZE][SIZE] = {{0}};
+
+  MPI_Datatype columntype;
+
+  MPI_Init(&argc,&argv);
+  MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+
+  MPI_Type_vector(SIZE, 1, SIZE, MPI_DOUBLE, &columntype);
+  MPI_Type_commit(&columntype);
+
+
+    if (rank == 0) {
+      for(i=0; i <SIZE;i++)
+        for(j=0; j <SIZE;j++)
+          a[i][j] = i*SIZE+j;
+    }
+
+    /* only one column is send
+     * this is an exemple for non-contignous data*/
+    MPI_Bcast(a, 1, columntype, 0, MPI_COMM_WORLD);
+
+    for(i=0; i<SIZE; i++){
+      for (j=0; j < SIZE; j++) {
+        printf("rank= %d, a[%d][%d]=%f\n",
+               rank, i, j, a[i][j]);
+      }
+      printf("\n");
+    }
+
+
+  MPI_Type_free(&columntype);
+  MPI_Finalize();
+  return 0;
+}
+