teshsuite/smpi/io-simple-at/io-simple-at.c

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