1 /*
2 Created by Huy Vo on 12/3/18.
3 */
4 static char help[] = "Test memory scalable AO.\n\n";
5
6 #include <petsc.h>
7 #include <petscvec.h>
8 #include <petscmat.h>
9 #include <petscao.h>
10
main(int argc,char * argv[])11 int main(int argc, char *argv[])
12 {
13 PetscInt n_global = 16;
14 MPI_Comm comm;
15 PetscLayout layout;
16 PetscInt local_size;
17 PetscInt start, end;
18 PetscMPIInt rank;
19 PetscInt *app_indices, *petsc_indices, *ia, *ia0;
20 PetscInt i;
21 AO app2petsc;
22 IS app_is, petsc_is;
23 const PetscInt n_loc = 8;
24
25 PetscFunctionBeginUser;
26 PetscCall(PetscInitialize(&argc, &argv, NULL, help));
27 comm = PETSC_COMM_WORLD;
28 PetscCallMPI(MPI_Comm_rank(comm, &rank));
29
30 PetscCall(PetscLayoutCreate(comm, &layout));
31 PetscCall(PetscLayoutSetSize(layout, n_global));
32 PetscCall(PetscLayoutSetLocalSize(layout, PETSC_DECIDE));
33 PetscCall(PetscLayoutSetUp(layout));
34 PetscCall(PetscLayoutGetLocalSize(layout, &local_size));
35 PetscCall(PetscLayoutGetRange(layout, &start, &end));
36
37 PetscCall(PetscMalloc1(local_size, &app_indices));
38 PetscCall(PetscMalloc1(local_size, &petsc_indices));
39 /* Add values for local indices for usual states */
40 for (i = 0; i < local_size; ++i) {
41 app_indices[i] = start + i;
42 petsc_indices[i] = end - 1 - i;
43 }
44
45 /* Create the AO object that maps from lexicographic ordering to PETSc Vec ordering */
46 PetscCall(ISCreateGeneral(comm, local_size, &app_indices[0], PETSC_COPY_VALUES, &app_is));
47 PetscCall(ISCreateGeneral(comm, local_size, &petsc_indices[0], PETSC_COPY_VALUES, &petsc_is));
48 PetscCall(AOCreate(comm, &app2petsc));
49 PetscCall(AOSetIS(app2petsc, app_is, petsc_is));
50 PetscCall(AOSetType(app2petsc, AOMEMORYSCALABLE));
51 PetscCall(AOSetFromOptions(app2petsc));
52 PetscCall(ISDestroy(&app_is));
53 PetscCall(ISDestroy(&petsc_is));
54 PetscCall(AOView(app2petsc, PETSC_VIEWER_STDOUT_WORLD));
55
56 /* Test AOApplicationToPetsc */
57 PetscCall(PetscMalloc1(n_loc, &ia));
58 PetscCall(PetscMalloc1(n_loc, &ia0));
59 if (rank == 0) {
60 ia[0] = 0;
61 ia[1] = -1;
62 ia[2] = 1;
63 ia[3] = 2;
64 ia[4] = -1;
65 ia[5] = 4;
66 ia[6] = 5;
67 ia[7] = 6;
68 } else {
69 ia[0] = -1;
70 ia[1] = 8;
71 ia[2] = 9;
72 ia[3] = 10;
73 ia[4] = -1;
74 ia[5] = 12;
75 ia[6] = 13;
76 ia[7] = 14;
77 }
78 PetscCall(PetscArraycpy(ia0, ia, n_loc));
79
80 PetscCall(AOApplicationToPetsc(app2petsc, n_loc, ia));
81
82 for (i = 0; i < n_loc; ++i) PetscCall(PetscSynchronizedPrintf(PETSC_COMM_WORLD, "proc = %d : %" PetscInt_FMT " -> %" PetscInt_FMT " \n", rank, ia0[i], ia[i]));
83 PetscCall(PetscSynchronizedFlush(PETSC_COMM_WORLD, PETSC_STDOUT));
84 PetscCall(AODestroy(&app2petsc));
85 PetscCall(PetscLayoutDestroy(&layout));
86 PetscCall(PetscFree(app_indices));
87 PetscCall(PetscFree(petsc_indices));
88 PetscCall(PetscFree(ia));
89 PetscCall(PetscFree(ia0));
90 PetscCall(PetscFinalize());
91 return 0;
92 }
93
94 /*TEST
95
96 test:
97 nsize: 2
98
99 TEST*/
100