teshsuite/smpi/mpich3-test/rma/allocmem.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *
   4  *  (C) 2003 by Argonne National Laboratory.
   5  *      See COPYRIGHT in top-level directory.
   6  */
   7 #include "mpi.h"
   8 #include <stdio.h>
   9 #include "mpitest.h"
  10
  11 /*
  12 static char MTEST_Descrip[] = "Simple test that alloc_mem and free_mem work together";
  13 */
  14
  15 int main(int argc, char *argv[])
  16 {
  17     int errs = 0, err;
  18     int j, count;
  19     char *ap;
  20
  21     MTest_Init(&argc, &argv);
  22
  23     MPI_Errhandler_set(MPI_COMM_WORLD, MPI_ERRORS_RETURN);
  24     for (count = 1; count < 128000; count *= 2) {
  25
  26         err = MPI_Alloc_mem(count, MPI_INFO_NULL, &ap);
  27         if (err) {
  28             int errclass;
  29             /* An error of  MPI_ERR_NO_MEM is allowed */
  30             MPI_Error_class(err, &errclass);
  31             if (errclass != MPI_ERR_NO_MEM) {
  32                 errs++;
  33                 MTestPrintError(err);
  34             }
  35
  36         }
  37         else {
  38             /* Access all of this memory */
  39             for (j = 0; j < count; j++) {
  40                 ap[j] = (char) (j & 0x7f);
  41             }
  42             MPI_Free_mem(ap);
  43         }
  44     }
  45
  46     MTest_Finalize(errs);
  47     MPI_Finalize();
  48     return 0;
  49 }