Actual source code: ex4.c
1: static char help[] = "Introductory example that illustrates running PETSc on a subset of processes.\n\n";
3: #include <petscsys.h>
5: int main(int argc, char *argv[])
6: {
7: PetscMPIInt rank, size;
9: /* We must call MPI_Init() first, making us, not PETSc, responsible for MPI */
10: PetscCallMPI(MPI_Init(&argc, &argv));
11: #if defined(PETSC_HAVE_ELEMENTAL)
12: PetscCall(PetscElementalInitializePackage());
13: #endif
14: /* We can now change the communicator universe for PETSc */
15: PetscCallMPI(MPI_Comm_rank(MPI_COMM_WORLD, &rank));
16: PetscCallMPI(MPI_Comm_split(MPI_COMM_WORLD, rank % 2, 0, &PETSC_COMM_WORLD));
18: /*
19: Every PETSc routine should begin with the PetscInitialize() routine.
20: argc, argv - These command line arguments are taken to extract the options
21: supplied to PETSc and options supplied to MPI.
22: help - When PETSc executable is invoked with the option -help,
23: it prints the various options that can be applied at
24: runtime. The user can use the "help" variable place
25: additional help messages in this printout.
26: */
27: PetscFunctionBeginUser;
28: PetscCall(PetscInitialize(&argc, &argv, (char *)0, help));
30: /*
31: The following MPI calls return the number of processes
32: being used and the rank of this process in the group.
33: */
34: PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
35: PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
37: /*
38: Here we would like to print only one message that represents
39: all the processes in the group. We use PetscPrintf() with the
40: communicator PETSC_COMM_WORLD. Thus, only one message is
41: printed representng PETSC_COMM_WORLD, i.e., all the processors.
42: */
43: PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Number of processors = %d, rank = %d\n", size, rank));
45: /*
46: Always call PetscFinalize() before exiting a program. This routine
47: - finalizes the PETSc libraries as well as MPI
48: - provides summary and diagnostic information if certain runtime
49: options are chosen (e.g., -log_view). See PetscFinalize()
50: manpage for more information.
51: */
52: PetscCall(PetscFinalize());
53: PetscCallMPI(MPI_Comm_free(&PETSC_COMM_WORLD));
54: #if defined(PETSC_HAVE_ELEMENTAL)
55: PetscCall(PetscElementalFinalizePackage());
56: #endif
57: /* Since we initialized MPI, we must call MPI_Finalize() */
58: PetscCallMPI(MPI_Finalize());
59: return 0;
60: }
62: /*TEST
64: test:
65: nsize: 5
66: args: -options_left no
67: filter: sort -b | grep -v saws_port_auto_selectcd
68: filter_output: sort -b
70: TEST*/