Blame test/mpi/errors/comm/subcomm_abort.c

Packit Service c5cf8c
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
Packit Service c5cf8c
/* vim: set ft=c.mpich : */
Packit Service c5cf8c
/*
Packit Service c5cf8c
 *  (C) 2018 by Argonne National Laboratory.
Packit Service c5cf8c
 *      See COPYRIGHT in top-level directory.
Packit Service c5cf8c
 */
Packit Service c5cf8c
Packit Service c5cf8c
#include <stdio.h>
Packit Service c5cf8c
#include <assert.h>
Packit Service c5cf8c
#include "mpi.h"
Packit Service c5cf8c
Packit Service c5cf8c
Packit Service c5cf8c
int main(int argc, char *argv[])
Packit Service c5cf8c
{
Packit Service c5cf8c
    int size, rank;
Packit Service c5cf8c
    int local_rank, local_size;
Packit Service c5cf8c
    MPI_Comm split_comm;
Packit Service c5cf8c
    int buffer = 0;
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Init(&argc, &argv);
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Comm_size(MPI_COMM_WORLD, &size);
Packit Service c5cf8c
    MPI_Comm_rank(MPI_COMM_WORLD, &rank;;
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Comm_split(MPI_COMM_WORLD, rank % 2, rank, &split_comm);
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Barrier(split_comm);
Packit Service c5cf8c
    MPI_Comm_size(split_comm, &local_size);
Packit Service c5cf8c
    MPI_Comm_rank(split_comm, &local_rank);
Packit Service c5cf8c
Packit Service c5cf8c
    if (rank == 1) {
Packit Service c5cf8c
        MPI_Abort(split_comm, 8);
Packit Service c5cf8c
    }
Packit Service c5cf8c
Packit Service c5cf8c
    if (rank == 0) {
Packit Service c5cf8c
        buffer = 4;
Packit Service c5cf8c
        MPI_Send(&buffer, 1, MPI_INT, 1, 0, split_comm);
Packit Service c5cf8c
    }
Packit Service c5cf8c
Packit Service c5cf8c
    if (rank == 2) {
Packit Service c5cf8c
        MPI_Recv(&buffer, 1, MPI_INT, 0, 0, split_comm, MPI_STATUS_IGNORE);
Packit Service c5cf8c
        assert(buffer == 4);
Packit Service c5cf8c
        printf("No Errors\n");
Packit Service c5cf8c
        fflush(stdout);
Packit Service c5cf8c
    }
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Comm_free(&split_comm);
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Finalize();
Packit Service c5cf8c
Packit Service c5cf8c
    return 0;
Packit Service c5cf8c
}