1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "Tests application ordering.\n\n"; 3c4762a1bSJed Brown 4c4762a1bSJed Brown #include <petscsys.h> 5c4762a1bSJed Brown #include <petscao.h> 6c4762a1bSJed Brown #include <petscviewer.h> 7c4762a1bSJed Brown 8*d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv) 9*d71ae5a4SJacob Faibussowitsch { 10c4762a1bSJed Brown PetscMPIInt rank, size; 11c4762a1bSJed Brown PetscInt n, *ispetsc, *isapp, start, N, i; 12c4762a1bSJed Brown AO ao; 13c4762a1bSJed Brown 14327415f7SBarry Smith PetscFunctionBeginUser; 159566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, (char *)0, help)); 169566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-n", &n, NULL)); 179371c9d4SSatish Balay PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank)); 189371c9d4SSatish Balay n = rank + 2; 199566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size)); 20c4762a1bSJed Brown 21c4762a1bSJed Brown /* create the orderings */ 229566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(n, &ispetsc, n, &isapp)); 23c4762a1bSJed Brown 249566063dSJacob Faibussowitsch PetscCallMPI(MPI_Scan(&n, &start, 1, MPIU_INT, MPI_SUM, PETSC_COMM_WORLD)); 259566063dSJacob Faibussowitsch PetscCallMPI(MPI_Allreduce(&n, &N, 1, MPIU_INT, MPI_SUM, PETSC_COMM_WORLD)); 26c4762a1bSJed Brown start -= n; 27c4762a1bSJed Brown 28c4762a1bSJed Brown for (i = 0; i < n; i++) { 29c4762a1bSJed Brown ispetsc[i] = start + i; 30c4762a1bSJed Brown isapp[i] = N - start - i - 1; 31c4762a1bSJed Brown } 32c4762a1bSJed Brown 33c4762a1bSJed Brown /* create the application ordering */ 349566063dSJacob Faibussowitsch PetscCall(AOCreateBasic(PETSC_COMM_WORLD, n, isapp, ispetsc, &ao)); 359566063dSJacob Faibussowitsch PetscCall(AOView(ao, PETSC_VIEWER_STDOUT_WORLD)); 36c4762a1bSJed Brown 37c4762a1bSJed Brown /* check the mapping */ 389566063dSJacob Faibussowitsch PetscCall(AOPetscToApplication(ao, n, ispetsc)); 39c4762a1bSJed Brown for (i = 0; i < n; i++) { 4048a46eb9SPierre 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])); 41c4762a1bSJed Brown } 429566063dSJacob Faibussowitsch PetscCall(PetscFree2(ispetsc, isapp)); 43c4762a1bSJed Brown 449566063dSJacob Faibussowitsch PetscCall(AODestroy(&ao)); 459566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 46b122ec5aSJacob Faibussowitsch return 0; 47c4762a1bSJed Brown } 48c4762a1bSJed Brown 49c4762a1bSJed Brown /*TEST 50c4762a1bSJed Brown 51c4762a1bSJed Brown test: 52c4762a1bSJed Brown 53c4762a1bSJed Brown test: 54c4762a1bSJed Brown suffix: 2 55c4762a1bSJed Brown nsize: 2 56c4762a1bSJed Brown 57c4762a1bSJed Brown test: 58c4762a1bSJed Brown suffix: 3 59c4762a1bSJed Brown nsize: 3 60c4762a1bSJed Brown 61c4762a1bSJed Brown TEST*/ 62