1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include <mpi.h>
#include <stdio.h>
#include <stdlib.h>
/*
* This test ensures that MPI_Comm_revoke works when a process failure has not
* occurred yet.
*/
int main(int argc, char **argv)
{
int rank, size;
int rc, ec;
char error[MPI_MAX_ERROR_STRING];
MPI_Comm world_dup;
MPI_Init(&argc, &argv);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_size(MPI_COMM_WORLD, &size);
if (size < 2) {
fprintf(stderr, "Must run with at least 2 processes\n");
MPI_Abort(MPI_COMM_WORLD, 1);
}
MPI_Comm_set_errhandler(MPI_COMM_WORLD, MPI_ERRORS_RETURN);
MPI_Comm_dup(MPI_COMM_WORLD, &world_dup);
if (rank == 1) {
MPIX_Comm_revoke(world_dup);
}
rc = MPI_Barrier(world_dup);
MPI_Error_class(rc, &ec);
if (ec != MPIX_ERR_REVOKED) {
MPI_Error_string(ec, error, &size);
fprintf(stderr,
"[%d] MPI_Barrier should have returned MPIX_ERR_REVOKED (%d), but it actually returned: %d\n%s\n",
rank, MPIX_ERR_REVOKED, ec, error);
MPI_Abort(MPI_COMM_WORLD, 1);
}
MPI_Comm_free(&world_dup);
if (rank == 0)
fprintf(stdout, " No errors\n");
MPI_Finalize();
return 0;
}
|