void error_handler(MPI_Comm *communicator, int *error_code, ...) { MPI_Comm *new_comm = malloc(sizeof(MPI_Comm)); MPIX_Comm_revoke(comm_all); MPIX_Comm_shrink(comm_all, new_comm); MPI_Comm_free(&comm_all); comm_all = *new_comm; }
int main(int argc, char **argv) { int rank, size, rc, ec, errs = 0; int flag = 1; MPI_Comm dup, shrunk; MPI_Init(&argc, &argv); MPI_Comm_dup(MPI_COMM_WORLD, &dup); MPI_Comm_rank(dup, &rank); MPI_Comm_size(dup, &size); MPI_Comm_set_errhandler(dup, MPI_ERRORS_RETURN); if (size < 4) { fprintf(stderr, "Must run with at least 4 processes\n"); MPI_Abort(dup, 1); } if (2 == rank) exit(EXIT_FAILURE); if (MPI_SUCCESS == (rc = MPIX_Comm_agree(dup, &flag))) { MPI_Error_class(rc, &ec); fprintf(stderr, "[%d] Expected MPIX_ERR_PROC_FAILED after agree. Received: %d\n", rank, ec); errs++; MPI_Abort(dup, 1); } if (MPI_SUCCESS != (rc = MPIX_Comm_shrink(dup, &shrunk))) { MPI_Error_class(rc, &ec); fprintf(stderr, "[%d] Expected MPI_SUCCESS after shrink. Received: %d\n", rank, ec); errs++; MPI_Abort(dup, 1); } if (MPI_SUCCESS != (rc = MPIX_Comm_agree(shrunk, &flag))) { MPI_Error_class(rc, &ec); fprintf(stderr, "[%d] Expected MPI_SUCCESS after agree. Received: %d\n", rank, ec); errs++; MPI_Abort(dup, 1); } MPI_Comm_free(&shrunk); MPI_Comm_free(&dup); if (0 == rank) { if (errs) fprintf(stdout, " Found %d errors\n", errs); else fprintf(stdout, " No errors\n"); } MPI_Finalize(); }
/* * This test ensures that shrink works correctly */ int main(int argc, char **argv) { int rank, size, newsize, rc, errclass, errs = 0; MPI_Comm newcomm; MPI_Init(&argc, &argv); MPI_Comm_rank(MPI_COMM_WORLD, &rank); MPI_Comm_size(MPI_COMM_WORLD, &size); MPI_Comm_set_errhandler(MPI_COMM_WORLD, MPI_ERRORS_RETURN); if (size < 4) { fprintf(stderr, "Must run with at least 4 processes\n"); MPI_Abort(MPI_COMM_WORLD, 1); } if (2 == rank) exit(EXIT_FAILURE); rc = MPIX_Comm_shrink(MPI_COMM_WORLD, &newcomm); if (rc) { MPI_Error_class(rc, &errclass); fprintf(stderr, "Expected MPI_SUCCESS from MPIX_Comm_shrink. Received: %d\n", errclass); errs++; MPI_Abort(MPI_COMM_WORLD, 1); } MPI_Comm_size(newcomm, &newsize); if (newsize != size - 1) errs++; rc = MPI_Barrier(newcomm); if (rc) { MPI_Error_class(rc, &errclass); fprintf(stderr, "Expected MPI_SUCCESS from MPI_BARRIER. Received: %d\n", errclass); errs++; MPI_Abort(MPI_COMM_WORLD, 1); } MPI_Comm_free(&newcomm); if (0 == rank) fprintf(stdout, " No Errors\n"); MPI_Finalize(); return 0; }
FORT_DLL_SPEC void FORT_CALL mpix_comm_shrink_ ( MPI_Fint *v1, MPI_Fint *v2, MPI_Fint *ierr ){ *ierr = MPIX_Comm_shrink( (MPI_Comm)(*v1), (MPI_Comm *)(v2) ); }