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 8c4762a1bSJed Brown int main(int argc,char **argv) 9c4762a1bSJed Brown { 10c4762a1bSJed Brown PetscErrorCode ierr; 11c4762a1bSJed Brown PetscMPIInt rank,size; 12c4762a1bSJed Brown PetscInt n,*ispetsc,*isapp,start,N,i; 13c4762a1bSJed Brown AO ao; 14c4762a1bSJed Brown 15c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 16*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL)); 17*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank));n = rank + 2; 18*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size)); 19c4762a1bSJed Brown 20c4762a1bSJed Brown /* create the orderings */ 21*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc2(n,&ispetsc,n,&isapp)); 22c4762a1bSJed Brown 23*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Scan(&n,&start,1,MPIU_INT,MPI_SUM,PETSC_COMM_WORLD)); 24*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_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 */ 33*5f80ce2aSJacob Faibussowitsch CHKERRQ(AOCreateBasic(PETSC_COMM_WORLD,n,isapp,ispetsc,&ao)); 34*5f80ce2aSJacob Faibussowitsch CHKERRQ(AOView(ao,PETSC_VIEWER_STDOUT_WORLD)); 35c4762a1bSJed Brown 36c4762a1bSJed Brown /* check the mapping */ 37*5f80ce2aSJacob Faibussowitsch CHKERRQ(AOPetscToApplication(ao,n,ispetsc)); 38c4762a1bSJed Brown for (i=0; i<n; i++) { 39c4762a1bSJed Brown if (ispetsc[i] != isapp[i]) { 40*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"[%d] Problem with mapping %" PetscInt_FMT " to %" PetscInt_FMT "\n",rank,i,ispetsc[i])); 41c4762a1bSJed Brown } 42c4762a1bSJed Brown } 43*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree2(ispetsc,isapp)); 44c4762a1bSJed Brown 45*5f80ce2aSJacob Faibussowitsch CHKERRQ(AODestroy(&ao)); 46c4762a1bSJed Brown ierr = PetscFinalize(); 47c4762a1bSJed Brown return ierr; 48c4762a1bSJed Brown } 49c4762a1bSJed Brown 50c4762a1bSJed Brown /*TEST 51c4762a1bSJed Brown 52c4762a1bSJed Brown test: 53c4762a1bSJed Brown 54c4762a1bSJed Brown test: 55c4762a1bSJed Brown suffix: 2 56c4762a1bSJed Brown nsize: 2 57c4762a1bSJed Brown 58c4762a1bSJed Brown test: 59c4762a1bSJed Brown suffix: 3 60c4762a1bSJed Brown nsize: 3 61c4762a1bSJed Brown 62c4762a1bSJed Brown TEST*/ 63