xref: /petsc/src/vec/is/ao/tests/ex4.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254)
1c4762a1bSJed Brown 
2c4762a1bSJed Brown static char help[] = "Test AO with on IS with 0 entries - contributed by Ethan Coon <ecoon@lanl.gov>, Apr 2011.\n\n";
3c4762a1bSJed Brown 
4c4762a1bSJed Brown #include <petscsys.h>
5c4762a1bSJed Brown #include <petscao.h>
6c4762a1bSJed Brown 
7c4762a1bSJed Brown int main(int argc,char **argv)
8c4762a1bSJed Brown {
9c4762a1bSJed Brown   AO             ao;
10c4762a1bSJed Brown   PetscInt       *localvert=NULL, nlocal;
11c4762a1bSJed Brown   PetscMPIInt    rank;
12c4762a1bSJed Brown 
13*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscInitialize(&argc,&argv,(char*)0,help));
145f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank));
155f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(4,&localvert));
16c4762a1bSJed Brown 
17dd400576SPatrick Sanan   if (rank == 0) {
18c4762a1bSJed Brown     nlocal       = 4;
19c4762a1bSJed Brown     localvert[0] = 0;
20c4762a1bSJed Brown     localvert[1] = 1;
21c4762a1bSJed Brown     localvert[2] = 2;
22c4762a1bSJed Brown     localvert[3] = 3;
23c4762a1bSJed Brown   } else {
24c4762a1bSJed Brown     nlocal = 0;
25c4762a1bSJed Brown   }
26c4762a1bSJed Brown 
27c4762a1bSJed Brown   /* Test AOCreateBasic() */
285f80ce2aSJacob Faibussowitsch   CHKERRQ(AOCreateBasic(PETSC_COMM_WORLD, nlocal, localvert, NULL, &ao));
295f80ce2aSJacob Faibussowitsch   CHKERRQ(AODestroy(&ao));
30c4762a1bSJed Brown 
31c4762a1bSJed Brown   /* Test AOCreateMemoryScalable() */
325f80ce2aSJacob Faibussowitsch   CHKERRQ(AOCreateMemoryScalable(PETSC_COMM_WORLD, nlocal, localvert, NULL, &ao));
335f80ce2aSJacob Faibussowitsch   CHKERRQ(AODestroy(&ao));
34c4762a1bSJed Brown 
355f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(localvert));
36*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscFinalize());
37*b122ec5aSJacob Faibussowitsch   return 0;
38c4762a1bSJed Brown }
39c4762a1bSJed Brown 
40c4762a1bSJed Brown /*TEST
41c4762a1bSJed Brown 
42c4762a1bSJed Brown    test:
43c4762a1bSJed Brown 
44c4762a1bSJed Brown    test:
45c4762a1bSJed Brown       suffix: 2
46c4762a1bSJed Brown       nsize: 2
47c4762a1bSJed Brown       output_file: output/ex4_1.out
48c4762a1bSJed Brown 
49c4762a1bSJed Brown TEST*/
50