1 2 static char help[] = "Tests application ordering.\n\n"; 3 4 #include <petscsys.h> 5 #include <petscao.h> 6 #include <petscviewer.h> 7 8 int main(int argc,char **argv) 9 { 10 PetscErrorCode ierr; 11 PetscMPIInt rank,size; 12 PetscInt n,*ispetsc,*isapp,start,N,i; 13 AO ao; 14 15 ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 16 ierr = PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL);CHKERRQ(ierr); 17 ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr);n = rank + 2; 18 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr); 19 20 /* create the orderings */ 21 ierr = PetscMalloc2(n,&ispetsc,n,&isapp);CHKERRQ(ierr); 22 23 ierr = MPI_Scan(&n,&start,1,MPIU_INT,MPI_SUM,PETSC_COMM_WORLD);CHKERRMPI(ierr); 24 ierr = MPI_Allreduce(&n,&N,1,MPIU_INT,MPI_SUM,PETSC_COMM_WORLD);CHKERRMPI(ierr); 25 start -= n; 26 27 for (i=0; i<n; i++) { 28 ispetsc[i] = start + i; 29 isapp[i] = N - start - i - 1; 30 } 31 32 /* create the application ordering */ 33 ierr = AOCreateBasic(PETSC_COMM_WORLD,n,isapp,ispetsc,&ao);CHKERRQ(ierr); 34 ierr = AOView(ao,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 35 36 /* check the mapping */ 37 ierr = AOPetscToApplication(ao,n,ispetsc);CHKERRQ(ierr); 38 for (i=0; i<n; i++) { 39 if (ispetsc[i] != isapp[i]) { 40 ierr = PetscPrintf(PETSC_COMM_WORLD,"[%d] Problem with mapping %D to %D\n",rank,i,ispetsc[i]);CHKERRQ(ierr); 41 } 42 } 43 ierr = PetscFree2(ispetsc,isapp);CHKERRQ(ierr); 44 45 ierr = AODestroy(&ao);CHKERRQ(ierr); 46 ierr = PetscFinalize(); 47 return ierr; 48 } 49 50 /*TEST 51 52 test: 53 54 test: 55 suffix: 2 56 nsize: 2 57 58 test: 59 suffix: 3 60 nsize: 3 61 62 TEST*/ 63