teshsuite/smpi/mpich3-test/attr/attrend.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *
   4  *  (C) 2008 by Argonne National Laboratory.
   5  *      See COPYRIGHT in top-level directory.
   6  */
   7 /*
   8       The MPI-2 specification makes it clear that delect attributes are
   9       called on MPI_COMM_WORLD and MPI_COMM_SELF at the very beginning of
  10       MPI_Finalize.  This is useful for tools that want to perform the MPI
  11       equivalent of an "at_exit" action.
  12  */
  13 #include <stdio.h>
  14 #include "mpi.h"
  15 #include "mpitest.h"
  16
  17 int exit_key = MPI_KEYVAL_INVALID;
  18 int wasCalled = 0;
  19 int foundError = 0;
  20 /* #define DEBUG */
  21 int delete_fn(MPI_Comm, int, void *, void *);
  22 #ifdef DEBUG
  23 #define FFLUSH fflush(stdout);
  24 #else
  25 #define FFLUSH
  26 #endif
  27
  28 int main(int argc, char **argv)
  29 {
  30     int errs = 0, wrank;
  31
  32     MTest_Init(&argc, &argv);
  33
  34     MPI_Comm_rank(MPI_COMM_WORLD, &wrank);
  35
  36     /* create the keyval for the exit handler */
  37     MPI_Keyval_create(MPI_NULL_COPY_FN, delete_fn, &exit_key, (void *) 0);
  38
  39     /* Attach to comm_self */
  40     MPI_Attr_put(MPI_COMM_SELF, exit_key, (void *) 0);
  41     /* We can free the key now */
  42     MPI_Keyval_free(&exit_key);
  43
  44     /* Now, exit MPI */
  45     /* MTest_Finalize(errs); */
  46     MPI_Finalize();
  47
  48     /* Check that the exit handler was called, and without error */
  49     if (wrank == 0) {
  50         /* In case more than one process exits MPI_Finalize */
  51         if (wasCalled != 1) {
  52             errs++;
  53             printf("Attribute delete function on MPI_COMM_SELF was not called\n");
  54         }
  55         if (foundError != 0) {
  56             errs++;
  57             printf("Found %d errors while executing delete function in MPI_COMM_SELF\n",
  58                    foundError);
  59         }
  60         if (errs == 0) {
  61             printf(" No Errors\n");
  62         }
  63         else {
  64             printf(" Found %d errors\n", errs);
  65         }
  66         fflush(stdout);
  67     }
  68
  69     return 0;
  70 }
  71
  72 int delete_fn(MPI_Comm comm, int keyval, void *attribute_val, void *extra_state)
  73 {
  74     int flag;
  75     wasCalled++;
  76     MPI_Finalized(&flag);
  77     if (flag) {
  78         foundError++;
  79     }
  80     return MPI_SUCCESS;
  81 }