hwloc: null check for cpuset
[mpich-dev.git] / examples / spawn_merge_parent.c
1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
2 /*
3  *  (C) 2001 by Argonne National Laboratory.
4  *      See COPYRIGHT in top-level directory.
5  */
6
7 #include <stdio.h>
8 #include "mpi.h"
9
10 int main( int argc, char *argv[] )
11 {
12     char str[10];
13     int err=0, errcodes[256], rank, rank1, nprocs;
14     MPI_Comm intercomm1, intercomm2, intracomm;
15
16     MPI_Init(&argc, &argv);
17
18     MPI_Comm_size(MPI_COMM_WORLD,&nprocs); 
19     MPI_Comm_rank(MPI_COMM_WORLD,&rank1); 
20
21     if (nprocs != 2) {
22         printf("Run this program with 2 processes\n");
23         MPI_Abort(MPI_COMM_WORLD,1);
24     }
25
26     if (rank1 == 0) {
27         printf("Parents spawning 2 children...\n");
28         fflush(stdout);
29     }
30
31     err = MPI_Comm_spawn("spawn_merge_child1", MPI_ARGV_NULL, 2,
32                          MPI_INFO_NULL, 1, MPI_COMM_WORLD,
33                          &intercomm1, errcodes);  
34     if (err) printf("Error in MPI_Comm_spawn\n");
35
36     if (rank1 == 0) {
37         printf("Parents and children merging to form new intracommunicator...\n");
38         fflush(stdout);
39     }
40
41     MPI_Intercomm_merge(intercomm1, 0, &intracomm);
42
43     if (rank1 == 0) {
44         printf("Merged parents spawning 2 more children and communicating with them...\n");
45         fflush(stdout);
46     }
47
48     err = MPI_Comm_spawn("spawn_merge_child2", MPI_ARGV_NULL, 2,
49                          MPI_INFO_NULL, 2, intracomm,
50                          &intercomm2, errcodes);  
51     if (err) printf("Error in MPI_Comm_spawn\n");
52
53     MPI_Comm_rank(intercomm2, &rank);
54
55     if (rank == 2) {
56         err = MPI_Recv(str, 3, MPI_CHAR, 1, 0, intercomm2, MPI_STATUS_IGNORE);
57         printf("Parent received from child: %s\n", str);
58         fflush(stdout);
59         
60         err = MPI_Send("bye", 4, MPI_CHAR, 1, 0, intercomm2); 
61     }
62
63     MPI_Finalize();
64
65     return 0;
66 }
67