Blame examples/spawn_merge_parent.c

Packit Service c5cf8c
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
Packit Service c5cf8c
/*
Packit Service c5cf8c
 *  (C) 2001 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 "mpi.h"
Packit Service c5cf8c
Packit Service c5cf8c
int main(int argc, char *argv[])
Packit Service c5cf8c
{
Packit Service c5cf8c
    char str[10];
Packit Service c5cf8c
    int err = 0, errcodes[256], rank, rank1, nprocs;
Packit Service c5cf8c
    MPI_Comm intercomm1, intercomm2, intracomm;
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Init(&argc, &argv);
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Comm_size(MPI_COMM_WORLD, &nprocs);
Packit Service c5cf8c
    MPI_Comm_rank(MPI_COMM_WORLD, &rank1);
Packit Service c5cf8c
Packit Service c5cf8c
    if (nprocs != 2) {
Packit Service c5cf8c
        printf("Run this program with 2 processes\n");
Packit Service c5cf8c
        MPI_Abort(MPI_COMM_WORLD, 1);
Packit Service c5cf8c
    }
Packit Service c5cf8c
Packit Service c5cf8c
    if (rank1 == 0) {
Packit Service c5cf8c
        printf("Parents spawning 2 children...\n");
Packit Service c5cf8c
        fflush(stdout);
Packit Service c5cf8c
    }
Packit Service c5cf8c
Packit Service c5cf8c
    err = MPI_Comm_spawn("spawn_merge_child1", MPI_ARGV_NULL, 2,
Packit Service c5cf8c
                         MPI_INFO_NULL, 1, MPI_COMM_WORLD, &intercomm1, errcodes);
Packit Service c5cf8c
    if (err)
Packit Service c5cf8c
        printf("Error in MPI_Comm_spawn\n");
Packit Service c5cf8c
Packit Service c5cf8c
    if (rank1 == 0) {
Packit Service c5cf8c
        printf("Parents and children merging to form new intracommunicator...\n");
Packit Service c5cf8c
        fflush(stdout);
Packit Service c5cf8c
    }
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Intercomm_merge(intercomm1, 0, &intracomm);
Packit Service c5cf8c
Packit Service c5cf8c
    if (rank1 == 0) {
Packit Service c5cf8c
        printf("Merged parents spawning 2 more children and communicating with them...\n");
Packit Service c5cf8c
        fflush(stdout);
Packit Service c5cf8c
    }
Packit Service c5cf8c
Packit Service c5cf8c
    err = MPI_Comm_spawn("spawn_merge_child2", MPI_ARGV_NULL, 2,
Packit Service c5cf8c
                         MPI_INFO_NULL, 2, intracomm, &intercomm2, errcodes);
Packit Service c5cf8c
    if (err)
Packit Service c5cf8c
        printf("Error in MPI_Comm_spawn\n");
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Comm_rank(intercomm2, &rank;;
Packit Service c5cf8c
Packit Service c5cf8c
    if (rank == 2) {
Packit Service c5cf8c
        err = MPI_Recv(str, 3, MPI_CHAR, 1, 0, intercomm2, MPI_STATUS_IGNORE);
Packit Service c5cf8c
        printf("Parent received from child: %s\n", str);
Packit Service c5cf8c
        fflush(stdout);
Packit Service c5cf8c
Packit Service c5cf8c
        err = MPI_Send("bye", 4, MPI_CHAR, 1, 0, intercomm2);
Packit Service c5cf8c
    }
Packit Service c5cf8c
Packit Service c5cf8c
    MPI_Finalize();
Packit Service c5cf8c
Packit Service c5cf8c
    return 0;
Packit Service c5cf8c
}