teshsuite/smpi/io-all/io-all.c

   1 #include "mpi.h"\r
   2 #include <stdio.h>\r
   3 #include <stdlib.h>\r
   4 #include <string.h>\r
   5 #include <memory.h>\r
   6 \r
   7 /* Test reading and writing zero bytes (set status correctly) */\r
   8 \r
   9 int main( int argc, char *argv[] )\r
  10 {\r
  11     int errs = 0;\r
  12     int size, rank, i, *buf, count;\r
  13     MPI_File fh;\r
  14     MPI_Comm comm;\r
  15     MPI_Status status;\r
  16  \r
  17     MPI_Init( &argc, &argv );\r
  18  \r
  19     comm = MPI_COMM_WORLD;\r
  20     MPI_File_open( comm, (char*)"/scratch/lib/libsimgrid.so.3.6.2", MPI_MODE_RDWR | MPI_MODE_CREATE | MPI_MODE_DELETE_ON_CLOSE, MPI_INFO_NULL, &fh );\r
  21     MPI_Comm_size( comm, &size );\r
  22     MPI_Comm_rank( comm, &rank );\r
  23     buf = (int *)malloc( 10+ size * sizeof(int) );\r
  24     buf[0] = rank;\r
  25  \r
  26     /* Write to file */\r
  27     MPI_File_seek( fh, sizeof(int)*rank, MPI_SEEK_SET ); \r
  28     MPI_File_write_all( fh, buf, 1, MPI_INT, &status );\r
  29     MPI_Get_count( &status, MPI_INT, &count );\r
  30     if (count != 1) {\r
  31         errs++;\r
  32         fprintf( stderr, "Wrong count (%d) on write\n", count );fflush(stderr);\r
  33     }\r
  34     /* Write to file, overlapping */\r
  35     MPI_File_seek( fh, sizeof(int)*rank, MPI_SEEK_SET ); \r
  36     MPI_File_write_all( fh, buf, 10, MPI_INT, &status );\r
  37     MPI_Get_count( &status, MPI_INT, &count );\r
  38     if (count != 10) {\r
  39         errs++;\r
  40         fprintf( stderr, "Wrong count (%d) on write\n", count );fflush(stderr);\r
  41     }\r
  42     /* Read nothing (check status) */\r
  43     memset( &status, 0xff, sizeof(MPI_Status) );\r
  44     MPI_File_read_all( fh, buf, 0, MPI_INT, &status );\r
  45     MPI_Get_count( &status, MPI_INT, &count );\r
  46     if (count != 0) {\r
  47         errs++;\r
  48         fprintf( stderr, "Count not zero (%d) on read\n", count );fflush(stderr);\r
  49     }\r
  50 \r
  51     /* Write nothing (check status) */\r
  52     memset( &status, 0xff, sizeof(MPI_Status) );\r
  53     MPI_File_write_all( fh, buf, 0, MPI_INT, &status );\r
  54     if (count != 0) {\r
  55         errs++;\r
  56         fprintf( stderr, "Count not zero (%d) on write\n", count );fflush(stderr);\r
  57     }\r
  58 \r
  59     MPI_Barrier( comm );\r
  60 \r
  61     MPI_File_seek( fh, sizeof(int)*rank, MPI_SEEK_SET );\r
  62     for (i=0; i<size; i++) buf[i] = -1;\r
  63     MPI_File_read_all( fh, buf, 10, MPI_INT, &status );\r
  64     // if (buf[0] != rank) {\r
  65         // errs++;\r
  66         // fprintf( stderr, "%d: buf = %d\n", rank, buf[0] );fflush(stderr);\r
  67     // }\r
  68  \r
  69     free( buf );\r
  70     MPI_File_close( &fh );\r
  71  \r
  72     MPI_Finalize();\r
  73     return errs;\r
  74 }