xref: /petsc/src/vec/is/ao/tests/ex4.c (revision 5f80ce2ab25dff0f4601e710601cbbcecf323266)
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   PetscErrorCode ierr;
10c4762a1bSJed Brown   AO             ao;
11c4762a1bSJed Brown   PetscInt       *localvert=NULL, nlocal;
12c4762a1bSJed Brown   PetscMPIInt    rank;
13c4762a1bSJed Brown 
14c4762a1bSJed Brown   ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr;
15*5f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank));
16*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(4,&localvert));
17c4762a1bSJed Brown 
18dd400576SPatrick Sanan   if (rank == 0) {
19c4762a1bSJed Brown     nlocal       = 4;
20c4762a1bSJed Brown     localvert[0] = 0;
21c4762a1bSJed Brown     localvert[1] = 1;
22c4762a1bSJed Brown     localvert[2] = 2;
23c4762a1bSJed Brown     localvert[3] = 3;
24c4762a1bSJed Brown   } else {
25c4762a1bSJed Brown     nlocal = 0;
26c4762a1bSJed Brown   }
27c4762a1bSJed Brown 
28c4762a1bSJed Brown   /* Test AOCreateBasic() */
29*5f80ce2aSJacob Faibussowitsch   CHKERRQ(AOCreateBasic(PETSC_COMM_WORLD, nlocal, localvert, NULL, &ao));
30*5f80ce2aSJacob Faibussowitsch   CHKERRQ(AODestroy(&ao));
31c4762a1bSJed Brown 
32c4762a1bSJed Brown   /* Test AOCreateMemoryScalable() */
33*5f80ce2aSJacob Faibussowitsch   CHKERRQ(AOCreateMemoryScalable(PETSC_COMM_WORLD, nlocal, localvert, NULL, &ao));
34*5f80ce2aSJacob Faibussowitsch   CHKERRQ(AODestroy(&ao));
35c4762a1bSJed Brown 
36*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(localvert));
37c4762a1bSJed Brown   ierr = PetscFinalize();
38c4762a1bSJed Brown   return ierr;
39c4762a1bSJed Brown }
40c4762a1bSJed Brown 
41c4762a1bSJed Brown /*TEST
42c4762a1bSJed Brown 
43c4762a1bSJed Brown    test:
44c4762a1bSJed Brown 
45c4762a1bSJed Brown    test:
46c4762a1bSJed Brown       suffix: 2
47c4762a1bSJed Brown       nsize: 2
48c4762a1bSJed Brown       output_file: output/ex4_1.out
49c4762a1bSJed Brown 
50c4762a1bSJed Brown TEST*/
51