Lines Matching refs:PetscCall
21 PetscCall(PetscInitialize(&argc, &argv, NULL, help)); in main()
25 PetscCall(PetscOptionsGetString(NULL, NULL, "-op", opname, sizeof(opname), &flag)); in main()
26 PetscCall(PetscStrcmp(opname, "replace", &isreplace)); in main()
27 PetscCall(PetscStrcmp(opname, "sum", &issum)); in main()
37 PetscCall(VecCreate(PETSC_COMM_WORLD, &x)); in main()
38 PetscCall(VecSetFromOptions(x)); in main()
39 PetscCall(VecSetSizes(x, PETSC_DECIDE, N)); in main()
46 PetscCall(VecGetOwnershipRange(x, &low, &high)); in main()
47 for (i = low; i < high; i++) PetscCall(VecSetValue(x, i, (PetscScalar)i + 1.0, INSERT_VALUES)); in main()
48 PetscCall(VecAssemblyBegin(x)); in main()
49 PetscCall(VecAssemblyEnd(x)); in main()
52 …PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\nTesting PetscSFFetchAndOp on a PETSCSF_PATTERN_GATHER g… in main()
53 PetscCall(VecGetLayout(x, &layout)); in main()
54 PetscCall(PetscSFCreate(PETSC_COMM_WORLD, &gathersf)); in main()
55 PetscCall(PetscSFSetGraphWithPattern(gathersf, layout, PETSCSF_PATTERN_GATHER)); in main()
58 PetscCall(PetscSFGetGraph(gathersf, NULL, &nleaves, NULL, NULL)); in main()
59 PetscCall(VecCreateSeq(PETSC_COMM_SELF, nleaves, &y)); in main()
60 PetscCall(VecDuplicate(y, &y2)); in main()
62 PetscCall(VecGetArray(x, &rootdata)); in main()
63 PetscCall(VecGetArray(y, &leafdata)); in main()
64 PetscCall(VecGetArray(y2, &leafupdate)); in main()
67 PetscCall(PetscSFBcastBegin(gathersf, MPIU_SCALAR, rootdata, leafdata, MPI_REPLACE)); in main()
68 PetscCall(PetscSFBcastEnd(gathersf, MPIU_SCALAR, rootdata, leafdata, MPI_REPLACE)); in main()
69 PetscCall(VecRestoreArray(y, &leafdata)); in main()
70 PetscCall(VecScale(y, 2)); in main()
71 PetscCall(VecGetArray(y, &leafdata)); in main()
74 PetscCall(PetscSFFetchAndOpBegin(gathersf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
75 PetscCall(PetscSFFetchAndOpEnd(gathersf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
78 PetscCall(VecCreateMPIWithArray(PETSC_COMM_WORLD, 1, nleaves, PETSC_DECIDE, leafupdate, &gy2)); in main()
79 PetscCall(PetscObjectSetName((PetscObject)x, "rootdata")); in main()
80 PetscCall(PetscObjectSetName((PetscObject)gy2, "leafupdate")); in main()
82 PetscCall(VecView(x, PETSC_VIEWER_STDOUT_WORLD)); in main()
83 PetscCall(VecView(gy2, PETSC_VIEWER_STDOUT_WORLD)); in main()
84 PetscCall(VecDestroy(&gy2)); in main()
86 PetscCall(VecRestoreArray(y2, &leafupdate)); in main()
87 PetscCall(VecDestroy(&y2)); in main()
89 PetscCall(VecRestoreArray(y, &leafdata)); in main()
90 PetscCall(VecDestroy(&y)); in main()
92 PetscCall(VecRestoreArray(x, &rootdata)); in main()
95 PetscCall(PetscSFDestroy(&gathersf)); in main()
102 for (i = low; i < high; i++) PetscCall(VecSetValue(x, i, (PetscScalar)i + 1.0, INSERT_VALUES)); in main()
103 PetscCall(VecAssemblyBegin(x)); in main()
104 PetscCall(VecAssemblyEnd(x)); in main()
107 …PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\nTesting PetscSFFetchAndOp on a PETSCSF_PATTERN_ALLGATHE… in main()
108 PetscCall(VecGetLayout(x, &layout)); in main()
109 PetscCall(PetscSFCreate(PETSC_COMM_WORLD, &allgathersf)); in main()
110 PetscCall(PetscSFSetGraphWithPattern(allgathersf, layout, PETSCSF_PATTERN_ALLGATHER)); in main()
113 PetscCall(PetscSFGetGraph(allgathersf, NULL, &nleaves, NULL, NULL)); in main()
114 PetscCall(VecCreateSeq(PETSC_COMM_SELF, nleaves, &y)); in main()
115 PetscCall(VecDuplicate(y, &y2)); in main()
117 PetscCall(VecGetArray(x, &rootdata)); in main()
118 PetscCall(VecGetArray(y, &leafdata)); in main()
119 PetscCall(VecGetArray(y2, &leafupdate)); in main()
122 PetscCall(PetscSFBcastBegin(allgathersf, MPIU_SCALAR, rootdata, leafdata, MPI_REPLACE)); in main()
123 PetscCall(PetscSFBcastEnd(allgathersf, MPIU_SCALAR, rootdata, leafdata, MPI_REPLACE)); in main()
124 PetscCall(VecRestoreArray(y, &leafdata)); in main()
125 PetscCall(VecScale(y, 2)); in main()
126 PetscCall(VecGetArray(y, &leafdata)); in main()
129 PetscCall(PetscSFFetchAndOpBegin(allgathersf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
130 PetscCall(PetscSFFetchAndOpEnd(allgathersf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
138 PetscCall(VecCreateMPIWithArray(PETSC_COMM_WORLD, 1, nleaves, PETSC_DECIDE, leafupdate, &gy2)); in main()
139 PetscCall(PetscObjectSetName((PetscObject)x, "rootdata")); in main()
140 PetscCall(PetscObjectSetName((PetscObject)gy2, "leafupdate")); in main()
142 PetscCall(VecView(x, PETSC_VIEWER_STDOUT_WORLD)); in main()
143 PetscCall(VecView(gy2, PETSC_VIEWER_STDOUT_WORLD)); in main()
144 PetscCall(VecDestroy(&gy2)); in main()
146 PetscCall(VecRestoreArray(y2, &leafupdate)); in main()
147 PetscCall(VecDestroy(&y2)); in main()
149 PetscCall(VecRestoreArray(y, &leafdata)); in main()
150 PetscCall(VecDestroy(&y)); in main()
152 PetscCall(VecRestoreArray(x, &rootdata)); in main()
153 PetscCall(VecDestroy(&x)); /* We won't reuse x in ALLGATHER, so destroy it */ in main()
155 PetscCall(PetscSFDestroy(&allgathersf)); in main()
161 PetscCall(VecCreate(PETSC_COMM_WORLD, &x)); in main()
162 PetscCall(VecSetFromOptions(x)); in main()
163 PetscCall(VecSetSizes(x, size, PETSC_DECIDE)); in main()
166 PetscCall(VecGetOwnershipRange(x, &low, &high)); in main()
167 for (i = low; i < high; i++) PetscCall(VecSetValue(x, i, (PetscScalar)i + 1.0, INSERT_VALUES)); in main()
168 PetscCall(VecAssemblyBegin(x)); in main()
169 PetscCall(VecAssemblyEnd(x)); in main()
172 …PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\nTesting PetscSFFetchAndOp on a PETSCSF_PATTERN_ALLTOALL… in main()
173 PetscCall(PetscSFCreate(PETSC_COMM_WORLD, &alltoallsf)); in main()
174 …PetscCall(PetscSFSetGraphWithPattern(alltoallsf, NULL /*insignificant*/, PETSCSF_PATTERN_ALLTOALL)… in main()
177 PetscCall(PetscSFGetGraph(alltoallsf, NULL, &nleaves, NULL, NULL)); in main()
178 PetscCall(VecCreateSeq(PETSC_COMM_SELF, nleaves, &y)); in main()
179 PetscCall(VecDuplicate(y, &y2)); in main()
181 PetscCall(VecGetArray(x, &rootdata)); in main()
182 PetscCall(VecGetArray(y, &leafdata)); in main()
183 PetscCall(VecGetArray(y2, &leafupdate)); in main()
186 PetscCall(PetscSFBcastBegin(alltoallsf, MPIU_SCALAR, rootdata, leafdata, MPI_REPLACE)); in main()
187 PetscCall(PetscSFBcastEnd(alltoallsf, MPIU_SCALAR, rootdata, leafdata, MPI_REPLACE)); in main()
190 PetscCall(PetscSFFetchAndOpBegin(alltoallsf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
191 PetscCall(PetscSFFetchAndOpEnd(alltoallsf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
196 PetscCall(VecCreateMPIWithArray(PETSC_COMM_WORLD, 1, nleaves, PETSC_DECIDE, leafupdate, &gy2)); in main()
197 PetscCall(PetscObjectSetName((PetscObject)x, "rootdata")); in main()
198 PetscCall(PetscObjectSetName((PetscObject)gy2, "leafupdate")); in main()
200 PetscCall(VecView(x, PETSC_VIEWER_STDOUT_WORLD)); in main()
201 PetscCall(VecView(gy2, PETSC_VIEWER_STDOUT_WORLD)); in main()
202 PetscCall(VecDestroy(&gy2)); in main()
204 PetscCall(VecRestoreArray(y2, &leafupdate)); in main()
205 PetscCall(VecDestroy(&y2)); in main()
207 PetscCall(VecRestoreArray(y, &leafdata)); in main()
208 PetscCall(VecDestroy(&y)); in main()
210 PetscCall(VecRestoreArray(x, &rootdata)); in main()
211 PetscCall(VecDestroy(&x)); in main()
213 PetscCall(PetscSFDestroy(&alltoallsf)); in main()
215 PetscCall(PetscFinalize()); in main()