teshsuite/smpi/mpich3-test/io/i_noncontig_coll.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *  (C) 2014 by Argonne National Laboratory.
   4  *      See COPYRIGHT in top-level directory.
   5  */
   6
   7 #include "mpi.h"
   8 #include <stdio.h>
   9 #include <string.h>
  10 #include <stdlib.h>
  11
  12 /* tests noncontiguous reads/writes using nonblocking collective I/O */
  13
  14 #define SIZE 5000
  15
  16 #define VERBOSE 0
  17 int main(int argc, char **argv)
  18 {
  19     int *buf, i, mynod, nprocs, len, b[3];
  20     int errs = 0, toterrs;
  21     MPI_Aint d[3];
  22     MPI_File fh;
  23     MPI_Request request;
  24     MPI_Status status;
  25     char *filename;
  26     MPI_Datatype typevec, newtype, t[3];
  27
  28     MPI_Init(&argc, &argv);
  29     MPI_Comm_size(MPI_COMM_WORLD, &nprocs);
  30     MPI_Comm_rank(MPI_COMM_WORLD, &mynod);
  31
  32     if (nprocs != 2) {
  33         fprintf(stderr, "Run this program on two processes\n");
  34         MPI_Abort(MPI_COMM_WORLD, 1);
  35     }
  36
  37     /* process 0 broadcasts the file name to other processes */
  38     if (!mynod) {
  39         filename = (char*)"testfile";
  40         len = strlen(filename);
  41         MPI_Bcast(&len, 1, MPI_INT, 0, MPI_COMM_WORLD);
  42         MPI_Bcast(filename, len + 1, MPI_CHAR, 0, MPI_COMM_WORLD);
  43     }
  44     else {
  45         MPI_Bcast(&len, 1, MPI_INT, 0, MPI_COMM_WORLD);
  46         filename = (char *) malloc(len + 1);
  47         MPI_Bcast(filename, len + 1, MPI_CHAR, 0, MPI_COMM_WORLD);
  48     }
  49
  50     buf = (int *) malloc(SIZE * sizeof(int));
  51
  52     MPI_Type_vector(SIZE / 2, 1, 2, MPI_INT, &typevec);
  53
  54     b[0] = b[1] = b[2] = 1;
  55     d[0] = 0;
  56     d[1] = mynod * sizeof(int);
  57     d[2] = SIZE * sizeof(int);
  58     t[0] = MPI_LB;
  59     t[1] = typevec;
  60     t[2] = MPI_UB;
  61
  62     MPI_Type_struct(3, b, d, t, &newtype);
  63     MPI_Type_commit(&newtype);
  64     MPI_Type_free(&typevec);
  65
  66     if (!mynod) {
  67 #if VERBOSE
  68         fprintf(stderr, "\ntesting noncontiguous in memory, noncontiguous in "
  69                 "file using collective I/O\n");
  70 #endif
  71         MPI_File_delete(filename, MPI_INFO_NULL);
  72     }
  73     MPI_Barrier(MPI_COMM_WORLD);
  74
  75     MPI_File_open(MPI_COMM_WORLD, filename, MPI_MODE_CREATE | MPI_MODE_RDWR, MPI_INFO_NULL, &fh);
  76
  77     MPI_File_set_view(fh, 0, MPI_INT, newtype, "native", MPI_INFO_NULL);
  78
  79     for (i = 0; i < SIZE; i++)
  80         buf[i] = i + mynod * SIZE;
  81     MPI_File_iwrite_all(fh, buf, 1, newtype, &request);
  82
  83     MPI_Barrier(MPI_COMM_WORLD);
  84     MPI_Wait(&request, &status);
  85
  86     for (i = 0; i < SIZE; i++)
  87         buf[i] = -1;
  88
  89     MPI_File_iread_at_all(fh, 0, buf, 1, newtype, &request);
  90     MPI_Wait(&request, &status);
  91
  92     for (i = 0; i < SIZE; i++) {
  93         if (!mynod) {
  94             if ((i % 2) && (buf[i] != -1)) {
  95                 errs++;
  96                 fprintf(stderr, "Process %d: buf %d is %d, should be -1\n", mynod, i, buf[i]);
  97             }
  98             if (!(i % 2) && (buf[i] != i)) {
  99                 errs++;
 100                 fprintf(stderr, "Process %d: buf %d is %d, should be %d\n", mynod, i, buf[i], i);
 101             }
 102         }
 103         else {
 104             if ((i % 2) && (buf[i] != i + mynod * SIZE)) {
 105                 errs++;
 106                 fprintf(stderr, "Process %d: buf %d is %d, should be %d\n",
 107                         mynod, i, buf[i], i + mynod * SIZE);
 108             }
 109             if (!(i % 2) && (buf[i] != -1)) {
 110                 errs++;
 111                 fprintf(stderr, "Process %d: buf %d is %d, should be -1\n", mynod, i, buf[i]);
 112             }
 113         }
 114     }
 115
 116     MPI_File_close(&fh);
 117
 118     MPI_Barrier(MPI_COMM_WORLD);
 119
 120     if (!mynod) {
 121 #if VERBOSE
 122         fprintf(stderr, "\ntesting noncontiguous in memory, contiguous in file "
 123                 "using collective I/O\n");
 124 #endif
 125         MPI_File_delete(filename, MPI_INFO_NULL);
 126     }
 127     MPI_Barrier(MPI_COMM_WORLD);
 128
 129     MPI_File_open(MPI_COMM_WORLD, filename, MPI_MODE_CREATE | MPI_MODE_RDWR, MPI_INFO_NULL, &fh);
 130
 131     for (i = 0; i < SIZE; i++)
 132         buf[i] = i + mynod * SIZE;
 133     MPI_File_iwrite_at_all(fh, mynod * (SIZE / 2) * sizeof(int), buf, 1, newtype, &request);
 134
 135     MPI_Barrier(MPI_COMM_WORLD);
 136     MPI_Wait(&request, &status);
 137
 138     for (i = 0; i < SIZE; i++)
 139         buf[i] = -1;
 140
 141     MPI_File_iread_at_all(fh, mynod * (SIZE / 2) * sizeof(int), buf, 1, newtype, &request);
 142     MPI_Wait(&request, &status);
 143
 144     for (i = 0; i < SIZE; i++) {
 145         if (!mynod) {
 146             if ((i % 2) && (buf[i] != -1)) {
 147                 errs++;
 148                 fprintf(stderr, "Process %d: buf %d is %d, should be -1\n", mynod, i, buf[i]);
 149             }
 150             if (!(i % 2) && (buf[i] != i)) {
 151                 errs++;
 152                 fprintf(stderr, "Process %d: buf %d is %d, should be %d\n", mynod, i, buf[i], i);
 153             }
 154         }
 155         else {
 156             if ((i % 2) && (buf[i] != i + mynod * SIZE)) {
 157                 errs++;
 158                 fprintf(stderr, "Process %d: buf %d is %d, should be %d\n",
 159                         mynod, i, buf[i], i + mynod * SIZE);
 160             }
 161             if (!(i % 2) && (buf[i] != -1)) {
 162                 errs++;
 163                 fprintf(stderr, "Process %d: buf %d is %d, should be -1\n", mynod, i, buf[i]);
 164             }
 165         }
 166     }
 167
 168     MPI_File_close(&fh);
 169
 170     MPI_Barrier(MPI_COMM_WORLD);
 171
 172     if (!mynod) {
 173 #if VERBOSE
 174         fprintf(stderr, "\ntesting contiguous in memory, noncontiguous in file "
 175                 "using collective I/O\n");
 176 #endif
 177         MPI_File_delete(filename, MPI_INFO_NULL);
 178     }
 179     MPI_Barrier(MPI_COMM_WORLD);
 180
 181     MPI_File_open(MPI_COMM_WORLD, filename, MPI_MODE_CREATE | MPI_MODE_RDWR, MPI_INFO_NULL, &fh);
 182
 183     MPI_File_set_view(fh, 0, MPI_INT, newtype, "native", MPI_INFO_NULL);
 184
 185     for (i = 0; i < SIZE; i++)
 186         buf[i] = i + mynod * SIZE;
 187     MPI_File_iwrite_all(fh, buf, SIZE, MPI_INT, &request);
 188
 189     MPI_Barrier(MPI_COMM_WORLD);
 190     MPI_Wait(&request, &status);
 191
 192     for (i = 0; i < SIZE; i++)
 193         buf[i] = -1;
 194
 195     MPI_File_iread_at_all(fh, 0, buf, SIZE, MPI_INT, &request);
 196     MPI_Wait(&request, &status);
 197
 198     for (i = 0; i < SIZE; i++) {
 199         if (!mynod) {
 200             if (buf[i] != i) {
 201                 errs++;
 202                 fprintf(stderr, "Process %d: buf %d is %d, should be %d\n", mynod, i, buf[i], i);
 203             }
 204         }
 205         else {
 206             if (buf[i] != i + mynod * SIZE) {
 207                 errs++;
 208                 fprintf(stderr, "Process %d: buf %d is %d, should be %d\n",
 209                         mynod, i, buf[i], i + mynod * SIZE);
 210             }
 211         }
 212     }
 213
 214     MPI_File_close(&fh);
 215
 216     MPI_Allreduce(&errs, &toterrs, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
 217     if (mynod == 0) {
 218         if (toterrs > 0) {
 219             fprintf(stderr, "Found %d errors\n", toterrs);
 220         }
 221         else {
 222             fprintf(stdout, " No Errors\n");
 223         }
 224     }
 225
 226     MPI_Type_free(&newtype);
 227     free(buf);
 228     if (mynod)
 229         free(filename);
 230     MPI_Finalize();
 231     return 0;
 232 }