1c4762a1bSJed Brown static char help[] = "Tests application ordering.\n\n";
2c4762a1bSJed Brown
3c4762a1bSJed Brown #include <petscsys.h>
4c4762a1bSJed Brown #include <petscao.h>
5c4762a1bSJed Brown #include <petscviewer.h>
6c4762a1bSJed Brown
main(int argc,char ** argv)7d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv)
8d71ae5a4SJacob Faibussowitsch {
9c4762a1bSJed Brown PetscMPIInt rank, size;
10c4762a1bSJed Brown PetscInt n, *ispetsc, *isapp, start, N, i;
11c4762a1bSJed Brown AO ao;
12c4762a1bSJed Brown
13327415f7SBarry Smith PetscFunctionBeginUser;
14*c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &argv, NULL, help));
159566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-n", &n, NULL));
169371c9d4SSatish Balay PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
179371c9d4SSatish Balay n = rank + 2;
189566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
19c4762a1bSJed Brown
20c4762a1bSJed Brown /* create the orderings */
219566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(n, &ispetsc, n, &isapp));
22c4762a1bSJed Brown
239566063dSJacob Faibussowitsch PetscCallMPI(MPI_Scan(&n, &start, 1, MPIU_INT, MPI_SUM, PETSC_COMM_WORLD));
24462c564dSBarry Smith PetscCallMPI(MPIU_Allreduce(&n, &N, 1, MPIU_INT, MPI_SUM, PETSC_COMM_WORLD));
25c4762a1bSJed Brown start -= n;
26c4762a1bSJed Brown
27c4762a1bSJed Brown for (i = 0; i < n; i++) {
28c4762a1bSJed Brown ispetsc[i] = start + i;
29c4762a1bSJed Brown isapp[i] = N - start - i - 1;
30c4762a1bSJed Brown }
31c4762a1bSJed Brown
32c4762a1bSJed Brown /* create the application ordering */
339566063dSJacob Faibussowitsch PetscCall(AOCreateBasic(PETSC_COMM_WORLD, n, isapp, ispetsc, &ao));
349566063dSJacob Faibussowitsch PetscCall(AOView(ao, PETSC_VIEWER_STDOUT_WORLD));
35c4762a1bSJed Brown
36c4762a1bSJed Brown /* check the mapping */
379566063dSJacob Faibussowitsch PetscCall(AOPetscToApplication(ao, n, ispetsc));
38c4762a1bSJed Brown for (i = 0; i < n; i++) {
3948a46eb9SPierre Jolivet if (ispetsc[i] != isapp[i]) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "[%d] Problem with mapping %" PetscInt_FMT " to %" PetscInt_FMT "\n", rank, i, ispetsc[i]));
40c4762a1bSJed Brown }
419566063dSJacob Faibussowitsch PetscCall(PetscFree2(ispetsc, isapp));
42c4762a1bSJed Brown
439566063dSJacob Faibussowitsch PetscCall(AODestroy(&ao));
449566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
45b122ec5aSJacob Faibussowitsch return 0;
46c4762a1bSJed Brown }
47c4762a1bSJed Brown
48c4762a1bSJed Brown /*TEST
49c4762a1bSJed Brown
50c4762a1bSJed Brown test:
51c4762a1bSJed Brown
52c4762a1bSJed Brown test:
53c4762a1bSJed Brown suffix: 2
54c4762a1bSJed Brown nsize: 2
55c4762a1bSJed Brown
56c4762a1bSJed Brown test:
57c4762a1bSJed Brown suffix: 3
58c4762a1bSJed Brown nsize: 3
59c4762a1bSJed Brown
60c4762a1bSJed Brown TEST*/
61