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

   1 /* Copyright (c) 2019-2022. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #include "mpi.h"
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <string.h>
  10 #include <memory.h>
  11
  12 /* Test reading and writing zero bytes (set status correctly) */
  13
  14 int main( int argc, char *argv[] )
  15 {
  16     int errs = 0;
  17     int size;
  18     int rank;
  19     int* buf;
  20     int count;
  21     MPI_File fh;
  22     MPI_Comm comm;
  23     MPI_Status status;
  24
  25     MPI_Init( &argc, &argv );
  26
  27     comm = MPI_COMM_WORLD;
  28     MPI_File_open( comm, (char*)"/scratch/testfile", MPI_MODE_RDWR | MPI_MODE_CREATE | MPI_MODE_DELETE_ON_CLOSE, MPI_INFO_NULL, &fh );
  29     MPI_Comm_size( comm, &size );
  30     MPI_Comm_rank( comm, &rank );
  31     buf = (int *)malloc( 10 * sizeof(int) );
  32     buf[0] = rank;
  33
  34     /* Write to file */
  35     MPI_File_seek(fh, sizeof(int) * rank, MPI_SEEK_SET);
  36     MPI_File_write_all( fh, buf, 1, MPI_INT, &status );
  37     MPI_Get_count( &status, MPI_INT, &count );
  38     if (count != 1) {
  39         errs++;
  40         fprintf( stderr, "Wrong count (%d) on write\n", count );fflush(stderr);
  41     }
  42     /* Write to file, overlapping */
  43     MPI_File_write_at_all( fh, sizeof(int)*rank, buf, 10, MPI_INT, &status );
  44     MPI_Get_count( &status, MPI_INT, &count );
  45     if (count != 10) {
  46         errs++;
  47         fprintf( stderr, "Wrong count (%d) on write\n", count );fflush(stderr);
  48     }
  49     /* Read nothing (check status) */
  50     memset( &status, 0xff, sizeof(MPI_Status) );
  51     MPI_File_read_at_all( fh, 0, buf, 0, MPI_INT, &status );
  52     MPI_Get_count( &status, MPI_INT, &count );
  53     if (count != 0) {
  54         errs++;
  55         fprintf( stderr, "Count not zero (%d) on read\n", count );fflush(stderr);
  56     }
  57
  58     /* Write nothing (check status) */
  59     memset( &status, 0xff, sizeof(MPI_Status) );
  60     MPI_File_write_at_all( fh, 0, buf, 0, MPI_INT, &status );
  61     if (count != 0) {
  62         errs++;
  63         fprintf( stderr, "Count not zero (%d) on write\n", count );fflush(stderr);
  64     }
  65
  66     MPI_Barrier( comm );
  67
  68     for (int i = 0; i < size; i++)
  69       buf[i] = -1;
  70     MPI_File_read_at_all( fh, sizeof(int)*rank, buf, 10, MPI_INT, &status );
  71     // if (buf[0] != rank) {
  72         // errs++;
  73         // fprintf( stderr, "%d: buf = %d\n", rank, buf[0] );fflush(stderr);
  74     // }
  75
  76     free( buf );
  77     MPI_File_close( &fh );
  78
  79     MPI_Finalize();
  80     return errs;
  81 }