xref: /libCEED/examples/petsc/bpsraw.c (revision debcf9199d3725add920e6dd742a46fc0f662bb0)
1cb32e2e7SValeria Barra // Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at
2cb32e2e7SValeria Barra // the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights
3cb32e2e7SValeria Barra // reserved. See files LICENSE and NOTICE for details.
4cb32e2e7SValeria Barra //
5cb32e2e7SValeria Barra // This file is part of CEED, a collection of benchmarks, miniapps, software
6cb32e2e7SValeria Barra // libraries and APIs for efficient high-order finite element and spectral
7cb32e2e7SValeria Barra // element discretizations for exascale applications. For more information and
8cb32e2e7SValeria Barra // source code availability see http://github.com/ceed.
9cb32e2e7SValeria Barra //
10cb32e2e7SValeria Barra // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC,
11cb32e2e7SValeria Barra // a collaborative effort of two U.S. Department of Energy organizations (Office
12cb32e2e7SValeria Barra // of Science and the National Nuclear Security Administration) responsible for
13cb32e2e7SValeria Barra // the planning and preparation of a capable exascale ecosystem, including
14cb32e2e7SValeria Barra // software, applications, hardware, advanced system engineering and early
15cb32e2e7SValeria Barra // testbed platforms, in support of the nation's exascale computing imperative.
16cb32e2e7SValeria Barra 
17cb32e2e7SValeria Barra //                        libCEED + PETSc Example: CEED BPs
18cb32e2e7SValeria Barra //
19cb32e2e7SValeria Barra // This example demonstrates a simple usage of libCEED with PETSc to solve the
20cb32e2e7SValeria Barra // CEED BP benchmark problems, see http://ceed.exascaleproject.org/bps.
21cb32e2e7SValeria Barra //
22cb32e2e7SValeria Barra // The code is intentionally "raw", using only low-level communication
23cb32e2e7SValeria Barra // primitives.
24cb32e2e7SValeria Barra //
25cb32e2e7SValeria Barra // Build with:
26cb32e2e7SValeria Barra //
27cb32e2e7SValeria Barra //     make bpsraw [PETSC_DIR=</path/to/petsc>] [CEED_DIR=</path/to/libceed>]
28cb32e2e7SValeria Barra //
29cb32e2e7SValeria Barra // Sample runs:
30cb32e2e7SValeria Barra //
3132d2ee49SValeria Barra //     ./bpsraw -problem bp1
3232d2ee49SValeria Barra //     ./bpsraw -problem bp2 -ceed /cpu/self
3332d2ee49SValeria Barra //     ./bpsraw -problem bp3 -ceed /gpu/occa
3432d2ee49SValeria Barra //     ./bpsraw -problem bp4 -ceed /cpu/occa
3532d2ee49SValeria Barra //     ./bpsraw -problem bp5 -ceed /omp/occa
3632d2ee49SValeria Barra //     ./bpsraw -problem bp6 -ceed /ocl/occa
37cb32e2e7SValeria Barra //
38cb32e2e7SValeria Barra //TESTARGS -ceed {ceed_resource} -test -problem bp2 -degree 5 -qextra 5
39cb32e2e7SValeria Barra 
40cb32e2e7SValeria Barra /// @file
41cb32e2e7SValeria Barra /// CEED BPs example using PETSc
42cb32e2e7SValeria Barra /// See bps.c for an implementation using DMPlex unstructured grids.
43cb32e2e7SValeria Barra const char help[] = "Solve CEED BPs using PETSc\n";
44cb32e2e7SValeria Barra 
45cb32e2e7SValeria Barra #include <stdbool.h>
46cb32e2e7SValeria Barra #include <string.h>
47cb32e2e7SValeria Barra #include <petscksp.h>
48cb32e2e7SValeria Barra #include <ceed.h>
49cb32e2e7SValeria Barra #include "qfunctions/bps/common.h"
50cb32e2e7SValeria Barra #include "qfunctions/bps/bp1.h"
51cb32e2e7SValeria Barra #include "qfunctions/bps/bp2.h"
52cb32e2e7SValeria Barra #include "qfunctions/bps/bp3.h"
53cb32e2e7SValeria Barra #include "qfunctions/bps/bp4.h"
54cb32e2e7SValeria Barra 
559396343dSjeremylt #include <petscsys.h>
569396343dSjeremylt #if PETSC_VERSION_LT(3,12,0)
579396343dSjeremylt #ifdef PETSC_HAVE_CUDA
589396343dSjeremylt #include <petsccuda.h>
599396343dSjeremylt // Note: With PETSc prior to version 3.12.0, providing the source path to
609396343dSjeremylt //       include 'cublas_v2.h' will be needed to use 'petsccuda.h'.
619396343dSjeremylt #endif
629396343dSjeremylt #endif
639396343dSjeremylt 
64cb32e2e7SValeria Barra static void Split3(PetscInt size, PetscInt m[3], bool reverse) {
65cb32e2e7SValeria Barra   for (PetscInt d=0,sizeleft=size; d<3; d++) {
66cb32e2e7SValeria Barra     PetscInt try = (PetscInt)PetscCeilReal(PetscPowReal(sizeleft, 1./(3 - d)));
67cb32e2e7SValeria Barra     while (try * (sizeleft / try) != sizeleft) try++;
68cb32e2e7SValeria Barra     m[reverse ? 2-d : d] = try;
69cb32e2e7SValeria Barra     sizeleft /= try;
70cb32e2e7SValeria Barra   }
71cb32e2e7SValeria Barra }
72cb32e2e7SValeria Barra 
73cb32e2e7SValeria Barra static PetscInt Max3(const PetscInt a[3]) {
74cb32e2e7SValeria Barra   return PetscMax(a[0], PetscMax(a[1], a[2]));
75cb32e2e7SValeria Barra }
76cb32e2e7SValeria Barra static PetscInt Min3(const PetscInt a[3]) {
77cb32e2e7SValeria Barra   return PetscMin(a[0], PetscMin(a[1], a[2]));
78cb32e2e7SValeria Barra }
79cb32e2e7SValeria Barra static void GlobalNodes(const PetscInt p[3], const PetscInt irank[3],
80cb32e2e7SValeria Barra                         PetscInt degree, const PetscInt melem[3],
81cb32e2e7SValeria Barra                         PetscInt mnodes[3]) {
82cb32e2e7SValeria Barra   for (int d=0; d<3; d++)
83cb32e2e7SValeria Barra     mnodes[d] = degree*melem[d] + (irank[d] == p[d]-1);
84cb32e2e7SValeria Barra }
85cb32e2e7SValeria Barra static PetscInt GlobalStart(const PetscInt p[3], const PetscInt irank[3],
86cb32e2e7SValeria Barra                             PetscInt degree, const PetscInt melem[3]) {
87cb32e2e7SValeria Barra   PetscInt start = 0;
88cb32e2e7SValeria Barra   // Dumb brute-force is easier to read
89cb32e2e7SValeria Barra   for (PetscInt i=0; i<p[0]; i++) {
90cb32e2e7SValeria Barra     for (PetscInt j=0; j<p[1]; j++) {
91cb32e2e7SValeria Barra       for (PetscInt k=0; k<p[2]; k++) {
92cb32e2e7SValeria Barra         PetscInt mnodes[3], ijkrank[] = {i,j,k};
93cb32e2e7SValeria Barra         if (i == irank[0] && j == irank[1] && k == irank[2]) return start;
94cb32e2e7SValeria Barra         GlobalNodes(p, ijkrank, degree, melem, mnodes);
95cb32e2e7SValeria Barra         start += mnodes[0] * mnodes[1] * mnodes[2];
96cb32e2e7SValeria Barra       }
97cb32e2e7SValeria Barra     }
98cb32e2e7SValeria Barra   }
99cb32e2e7SValeria Barra   return -1;
100cb32e2e7SValeria Barra }
10161dbc9d2Sjeremylt static int CreateRestriction(Ceed ceed, CeedInterlaceMode imode,
102a8d32208Sjeremylt                              const CeedInt melem[3], CeedInt P, CeedInt ncomp,
103cb32e2e7SValeria Barra                              CeedElemRestriction *Erestrict) {
104cb32e2e7SValeria Barra   const PetscInt nelem = melem[0]*melem[1]*melem[2];
105cb32e2e7SValeria Barra   PetscInt mnodes[3], *idx, *idxp;
106cb32e2e7SValeria Barra 
107cb32e2e7SValeria Barra   // Get indicies
108cb32e2e7SValeria Barra   for (int d=0; d<3; d++) mnodes[d] = melem[d]*(P-1) + 1;
109cb32e2e7SValeria Barra   idxp = idx = malloc(nelem*P*P*P*sizeof idx[0]);
110cb32e2e7SValeria Barra   for (CeedInt i=0; i<melem[0]; i++)
111cb32e2e7SValeria Barra     for (CeedInt j=0; j<melem[1]; j++)
112cb32e2e7SValeria Barra       for (CeedInt k=0; k<melem[2]; k++,idxp += P*P*P)
113cb32e2e7SValeria Barra         for (CeedInt ii=0; ii<P; ii++)
114cb32e2e7SValeria Barra           for (CeedInt jj=0; jj<P; jj++)
115cb32e2e7SValeria Barra             for (CeedInt kk=0; kk<P; kk++) {
116cb32e2e7SValeria Barra               if (0) { // This is the C-style (i,j,k) ordering that I prefer
117cb32e2e7SValeria Barra                 idxp[(ii*P+jj)*P+kk] = (((i*(P-1)+ii)*mnodes[1]
118cb32e2e7SValeria Barra                                          + (j*(P-1)+jj))*mnodes[2]
119cb32e2e7SValeria Barra                                         + (k*(P-1)+kk));
120cb32e2e7SValeria Barra               } else { // (k,j,i) ordering for consistency with MFEM example
121cb32e2e7SValeria Barra                 idxp[ii+P*(jj+P*kk)] = (((i*(P-1)+ii)*mnodes[1]
122cb32e2e7SValeria Barra                                          + (j*(P-1)+jj))*mnodes[2]
123cb32e2e7SValeria Barra                                         + (k*(P-1)+kk));
124cb32e2e7SValeria Barra               }
125cb32e2e7SValeria Barra             }
126cb32e2e7SValeria Barra 
127cb32e2e7SValeria Barra   // Setup CEED restriction
12861dbc9d2Sjeremylt   CeedElemRestrictionCreate(ceed, imode, nelem, P*P*P,
129a8d32208Sjeremylt                             mnodes[0]*mnodes[1]*mnodes[2], ncomp,
130a8d32208Sjeremylt                             CEED_MEM_HOST, CEED_OWN_POINTER, idx, Erestrict);
131cb32e2e7SValeria Barra 
132cb32e2e7SValeria Barra   PetscFunctionReturn(0);
133cb32e2e7SValeria Barra }
134cb32e2e7SValeria Barra 
135cb32e2e7SValeria Barra // Data for PETSc
136cb32e2e7SValeria Barra typedef struct User_ *User;
137cb32e2e7SValeria Barra struct User_ {
138cb32e2e7SValeria Barra   MPI_Comm comm;
139cb32e2e7SValeria Barra   VecScatter ltog;              // Scatter for all entries
140cb32e2e7SValeria Barra   VecScatter ltog0;             // Skip Dirichlet values
141cb32e2e7SValeria Barra   VecScatter gtogD;             // global-to-global; only Dirichlet values
142cb32e2e7SValeria Barra   Vec Xloc, Yloc;
143cb32e2e7SValeria Barra   CeedVector xceed, yceed;
144cb32e2e7SValeria Barra   CeedOperator op;
145cb32e2e7SValeria Barra   CeedVector qdata;
146cb32e2e7SValeria Barra   Ceed ceed;
1479396343dSjeremylt   CeedMemType memtype;
1489396343dSjeremylt   int (*VecGetArray)(Vec, PetscScalar **);
1499396343dSjeremylt   int (*VecGetArrayRead)(Vec, const PetscScalar **);
1509396343dSjeremylt   int (*VecRestoreArray)(Vec, PetscScalar **);
1519396343dSjeremylt   int (*VecRestoreArrayRead)(Vec, const PetscScalar **);
1529396343dSjeremylt };
1539396343dSjeremylt 
1549396343dSjeremylt // MemType Options
1559396343dSjeremylt static const char *const memTypes[] = {"host","device","memType",
1569396343dSjeremylt                                        "CEED_MEM_",0
157cb32e2e7SValeria Barra                                       };
158cb32e2e7SValeria Barra 
159cb32e2e7SValeria Barra // BP Options
160cb32e2e7SValeria Barra typedef enum {
161cb32e2e7SValeria Barra   CEED_BP1 = 0, CEED_BP2 = 1, CEED_BP3 = 2,
162cb32e2e7SValeria Barra   CEED_BP4 = 3, CEED_BP5 = 4, CEED_BP6 = 5
163cb32e2e7SValeria Barra } bpType;
164cb32e2e7SValeria Barra static const char *const bpTypes[] = {"bp1","bp2","bp3","bp4","bp5","bp6",
165cb32e2e7SValeria Barra                                       "bpType","CEED_BP",0
166cb32e2e7SValeria Barra                                      };
167cb32e2e7SValeria Barra 
168cb32e2e7SValeria Barra // BP specific data
169cb32e2e7SValeria Barra typedef struct {
170cb32e2e7SValeria Barra   CeedInt ncompu, qdatasize, qextra;
171cb32e2e7SValeria Barra   CeedQFunctionUser setupgeo, setuprhs, apply, error;
172cb32e2e7SValeria Barra   const char *setupgeofname, *setuprhsfname, *applyfname, *errorfname;
173cb32e2e7SValeria Barra   CeedEvalMode inmode, outmode;
174cb32e2e7SValeria Barra   CeedQuadMode qmode;
175cb32e2e7SValeria Barra } bpData;
176cb32e2e7SValeria Barra 
177cb32e2e7SValeria Barra bpData bpOptions[6] = {
178cb32e2e7SValeria Barra   [CEED_BP1] = {
179cb32e2e7SValeria Barra     .ncompu = 1,
180cb32e2e7SValeria Barra     .qdatasize = 1,
181cb32e2e7SValeria Barra     .qextra = 1,
182cb32e2e7SValeria Barra     .setupgeo = SetupMassGeo,
183cb32e2e7SValeria Barra     .setuprhs = SetupMassRhs,
184cb32e2e7SValeria Barra     .apply = Mass,
185cb32e2e7SValeria Barra     .error = Error,
186cb32e2e7SValeria Barra     .setupgeofname = SetupMassGeo_loc,
187cb32e2e7SValeria Barra     .setuprhsfname = SetupMassRhs_loc,
188cb32e2e7SValeria Barra     .applyfname = Mass_loc,
189cb32e2e7SValeria Barra     .errorfname = Error_loc,
190cb32e2e7SValeria Barra     .inmode = CEED_EVAL_INTERP,
191cb32e2e7SValeria Barra     .outmode = CEED_EVAL_INTERP,
192cb32e2e7SValeria Barra     .qmode = CEED_GAUSS
193cb32e2e7SValeria Barra   },
194cb32e2e7SValeria Barra   [CEED_BP2] = {
195cb32e2e7SValeria Barra     .ncompu = 3,
196cb32e2e7SValeria Barra     .qdatasize = 1,
197cb32e2e7SValeria Barra     .qextra = 1,
198cb32e2e7SValeria Barra     .setupgeo = SetupMassGeo,
199cb32e2e7SValeria Barra     .setuprhs = SetupMassRhs3,
200cb32e2e7SValeria Barra     .apply = Mass3,
201cb32e2e7SValeria Barra     .error = Error3,
202cb32e2e7SValeria Barra     .setupgeofname = SetupMassGeo_loc,
203cb32e2e7SValeria Barra     .setuprhsfname = SetupMassRhs3_loc,
204cb32e2e7SValeria Barra     .applyfname = Mass3_loc,
205cb32e2e7SValeria Barra     .errorfname = Error3_loc,
206cb32e2e7SValeria Barra     .inmode = CEED_EVAL_INTERP,
207cb32e2e7SValeria Barra     .outmode = CEED_EVAL_INTERP,
208cb32e2e7SValeria Barra     .qmode = CEED_GAUSS
209cb32e2e7SValeria Barra   },
210cb32e2e7SValeria Barra   [CEED_BP3] = {
211cb32e2e7SValeria Barra     .ncompu = 1,
212cb32e2e7SValeria Barra     .qdatasize = 6,
213cb32e2e7SValeria Barra     .qextra = 1,
214cb32e2e7SValeria Barra     .setupgeo = SetupDiffGeo,
215cb32e2e7SValeria Barra     .setuprhs = SetupDiffRhs,
216cb32e2e7SValeria Barra     .apply = Diff,
217cb32e2e7SValeria Barra     .error = Error,
218cb32e2e7SValeria Barra     .setupgeofname = SetupDiffGeo_loc,
219cb32e2e7SValeria Barra     .setuprhsfname = SetupDiffRhs_loc,
220cb32e2e7SValeria Barra     .applyfname = Diff_loc,
221cb32e2e7SValeria Barra     .errorfname = Error_loc,
222cb32e2e7SValeria Barra     .inmode = CEED_EVAL_GRAD,
223cb32e2e7SValeria Barra     .outmode = CEED_EVAL_GRAD,
224cb32e2e7SValeria Barra     .qmode = CEED_GAUSS
225cb32e2e7SValeria Barra   },
226cb32e2e7SValeria Barra   [CEED_BP4] = {
227cb32e2e7SValeria Barra     .ncompu = 3,
228cb32e2e7SValeria Barra     .qdatasize = 6,
229cb32e2e7SValeria Barra     .qextra = 1,
230cb32e2e7SValeria Barra     .setupgeo = SetupDiffGeo,
231cb32e2e7SValeria Barra     .setuprhs = SetupDiffRhs3,
232cb32e2e7SValeria Barra     .apply = Diff3,
233cb32e2e7SValeria Barra     .error = Error3,
234cb32e2e7SValeria Barra     .setupgeofname = SetupDiffGeo_loc,
235cb32e2e7SValeria Barra     .setuprhsfname = SetupDiffRhs3_loc,
236cb32e2e7SValeria Barra     .applyfname = Diff_loc,
237cb32e2e7SValeria Barra     .errorfname = Error3_loc,
238cb32e2e7SValeria Barra     .inmode = CEED_EVAL_GRAD,
239cb32e2e7SValeria Barra     .outmode = CEED_EVAL_GRAD,
240cb32e2e7SValeria Barra     .qmode = CEED_GAUSS
241cb32e2e7SValeria Barra   },
242cb32e2e7SValeria Barra   [CEED_BP5] = {
243cb32e2e7SValeria Barra     .ncompu = 1,
244cb32e2e7SValeria Barra     .qdatasize = 6,
245cb32e2e7SValeria Barra     .qextra = 0,
246cb32e2e7SValeria Barra     .setupgeo = SetupDiffGeo,
247cb32e2e7SValeria Barra     .setuprhs = SetupDiffRhs,
248cb32e2e7SValeria Barra     .apply = Diff,
249cb32e2e7SValeria Barra     .error = Error,
250cb32e2e7SValeria Barra     .setupgeofname = SetupDiffGeo_loc,
251cb32e2e7SValeria Barra     .setuprhsfname = SetupDiffRhs_loc,
252cb32e2e7SValeria Barra     .applyfname = Diff_loc,
253cb32e2e7SValeria Barra     .errorfname = Error_loc,
254cb32e2e7SValeria Barra     .inmode = CEED_EVAL_GRAD,
255cb32e2e7SValeria Barra     .outmode = CEED_EVAL_GRAD,
256cb32e2e7SValeria Barra     .qmode = CEED_GAUSS_LOBATTO
257cb32e2e7SValeria Barra   },
258cb32e2e7SValeria Barra   [CEED_BP6] = {
259cb32e2e7SValeria Barra     .ncompu = 3,
260cb32e2e7SValeria Barra     .qdatasize = 6,
261cb32e2e7SValeria Barra     .qextra = 0,
262cb32e2e7SValeria Barra     .setupgeo = SetupDiffGeo,
263cb32e2e7SValeria Barra     .setuprhs = SetupDiffRhs3,
264cb32e2e7SValeria Barra     .apply = Diff3,
265cb32e2e7SValeria Barra     .error = Error3,
266cb32e2e7SValeria Barra     .setupgeofname = SetupDiffGeo_loc,
267cb32e2e7SValeria Barra     .setuprhsfname = SetupDiffRhs3_loc,
268cb32e2e7SValeria Barra     .applyfname = Diff_loc,
269cb32e2e7SValeria Barra     .errorfname = Error3_loc,
270cb32e2e7SValeria Barra     .inmode = CEED_EVAL_GRAD,
271cb32e2e7SValeria Barra     .outmode = CEED_EVAL_GRAD,
272cb32e2e7SValeria Barra     .qmode = CEED_GAUSS_LOBATTO
273cb32e2e7SValeria Barra   }
274cb32e2e7SValeria Barra };
275cb32e2e7SValeria Barra 
276cb32e2e7SValeria Barra // This function uses libCEED to compute the action of the mass matrix
277cb32e2e7SValeria Barra static PetscErrorCode MatMult_Mass(Mat A, Vec X, Vec Y) {
278cb32e2e7SValeria Barra   PetscErrorCode ierr;
279cb32e2e7SValeria Barra   User user;
280cb32e2e7SValeria Barra   PetscScalar *x, *y;
281cb32e2e7SValeria Barra 
282cb32e2e7SValeria Barra   PetscFunctionBeginUser;
2839396343dSjeremylt 
284cb32e2e7SValeria Barra   ierr = MatShellGetContext(A, &user); CHKERRQ(ierr);
2859396343dSjeremylt 
2869396343dSjeremylt   // Global-to-local
287cb32e2e7SValeria Barra   ierr = VecScatterBegin(user->ltog, X, user->Xloc, INSERT_VALUES,
288cb32e2e7SValeria Barra                          SCATTER_REVERSE); CHKERRQ(ierr);
2899396343dSjeremylt   ierr = VecScatterEnd(user->ltog, X, user->Xloc, INSERT_VALUES,
2909396343dSjeremylt                        SCATTER_REVERSE); CHKERRQ(ierr);
291cb32e2e7SValeria Barra   ierr = VecZeroEntries(user->Yloc); CHKERRQ(ierr);
292cb32e2e7SValeria Barra 
2939396343dSjeremylt   // Setup libCEED vectors
2949396343dSjeremylt   ierr = user->VecGetArrayRead(user->Xloc, (const PetscScalar **)&x);
2959396343dSjeremylt   CHKERRQ(ierr);
2969396343dSjeremylt   ierr = user->VecGetArray(user->Yloc, &y); CHKERRQ(ierr);
2979396343dSjeremylt   CeedVectorSetArray(user->xceed, user->memtype, CEED_USE_POINTER, x);
2989396343dSjeremylt   CeedVectorSetArray(user->yceed, user->memtype, CEED_USE_POINTER, y);
299cb32e2e7SValeria Barra 
3009396343dSjeremylt   // Apply libCEED operator
301cb32e2e7SValeria Barra   CeedOperatorApply(user->op, user->xceed, user->yceed,
302cb32e2e7SValeria Barra                     CEED_REQUEST_IMMEDIATE);
3039396343dSjeremylt   CeedVectorSyncArray(user->yceed, user->memtype);
304cb32e2e7SValeria Barra 
3059396343dSjeremylt   // Restore PETSc vectors
3069396343dSjeremylt   ierr = user->VecRestoreArrayRead(user->Xloc, (const PetscScalar **)&x);
3079396343dSjeremylt   CHKERRQ(ierr);
3089396343dSjeremylt   ierr = user->VecRestoreArray(user->Yloc, &y); CHKERRQ(ierr);
309cb32e2e7SValeria Barra 
3109396343dSjeremylt   // Local-to-global
311cb32e2e7SValeria Barra   if (Y) {
312cb32e2e7SValeria Barra     ierr = VecZeroEntries(Y); CHKERRQ(ierr);
3139396343dSjeremylt     ierr = VecScatterBegin(user->ltog, user->Yloc, Y, ADD_VALUES,
3149396343dSjeremylt                            SCATTER_FORWARD); CHKERRQ(ierr);
3159396343dSjeremylt     ierr = VecScatterEnd(user->ltog, user->Yloc, Y, ADD_VALUES,
3169396343dSjeremylt                          SCATTER_FORWARD); CHKERRQ(ierr);
317cb32e2e7SValeria Barra   }
318cb32e2e7SValeria Barra   PetscFunctionReturn(0);
319cb32e2e7SValeria Barra }
320cb32e2e7SValeria Barra 
321cb32e2e7SValeria Barra // This function uses libCEED to compute the action of the Laplacian with
322cb32e2e7SValeria Barra // Dirichlet boundary conditions
323cb32e2e7SValeria Barra static PetscErrorCode MatMult_Diff(Mat A, Vec X, Vec Y) {
324cb32e2e7SValeria Barra   PetscErrorCode ierr;
325cb32e2e7SValeria Barra   User user;
326cb32e2e7SValeria Barra   PetscScalar *x, *y;
327cb32e2e7SValeria Barra 
328cb32e2e7SValeria Barra   PetscFunctionBeginUser;
3299396343dSjeremylt 
330cb32e2e7SValeria Barra   ierr = MatShellGetContext(A, &user); CHKERRQ(ierr);
331cb32e2e7SValeria Barra 
332cb32e2e7SValeria Barra   // Global-to-local
333cb32e2e7SValeria Barra   ierr = VecScatterBegin(user->ltog0, X, user->Xloc, INSERT_VALUES,
334cb32e2e7SValeria Barra                          SCATTER_REVERSE); CHKERRQ(ierr);
335cb32e2e7SValeria Barra   ierr = VecScatterEnd(user->ltog0, X, user->Xloc, INSERT_VALUES,
336cb32e2e7SValeria Barra                        SCATTER_REVERSE);
337cb32e2e7SValeria Barra   CHKERRQ(ierr);
338cb32e2e7SValeria Barra   ierr = VecZeroEntries(user->Yloc); CHKERRQ(ierr);
339cb32e2e7SValeria Barra 
3409396343dSjeremylt   // Setup libCEED vectors
3419396343dSjeremylt   ierr = user->VecGetArrayRead(user->Xloc, (const PetscScalar **)&x);
3429396343dSjeremylt   CHKERRQ(ierr);
3439396343dSjeremylt   ierr = user->VecGetArray(user->Yloc, &y); CHKERRQ(ierr);
3449396343dSjeremylt   CeedVectorSetArray(user->xceed, user->memtype, CEED_USE_POINTER, x);
3459396343dSjeremylt   CeedVectorSetArray(user->yceed, user->memtype, CEED_USE_POINTER, y);
346cb32e2e7SValeria Barra 
3479396343dSjeremylt   // Apply libCEED operator
348cb32e2e7SValeria Barra   CeedOperatorApply(user->op, user->xceed, user->yceed,
349cb32e2e7SValeria Barra                     CEED_REQUEST_IMMEDIATE);
3509396343dSjeremylt   CeedVectorSyncArray(user->yceed, user->memtype);
351cb32e2e7SValeria Barra 
352cb32e2e7SValeria Barra   // Restore PETSc vectors
3539396343dSjeremylt   ierr = user->VecRestoreArrayRead(user->Xloc, (const PetscScalar **)&x);
3549396343dSjeremylt   CHKERRQ(ierr);
3559396343dSjeremylt   ierr = user->VecRestoreArray(user->Yloc, &y); CHKERRQ(ierr);
356cb32e2e7SValeria Barra 
357cb32e2e7SValeria Barra   // Local-to-global
358cb32e2e7SValeria Barra   ierr = VecZeroEntries(Y); CHKERRQ(ierr);
359cb32e2e7SValeria Barra   ierr = VecScatterBegin(user->gtogD, X, Y, INSERT_VALUES, SCATTER_FORWARD);
360cb32e2e7SValeria Barra   CHKERRQ(ierr);
361cb32e2e7SValeria Barra   ierr = VecScatterEnd(user->gtogD, X, Y, INSERT_VALUES, SCATTER_FORWARD);
362cb32e2e7SValeria Barra   CHKERRQ(ierr);
3639396343dSjeremylt   ierr = VecScatterBegin(user->ltog0, user->Yloc, Y, ADD_VALUES,
3649396343dSjeremylt                          SCATTER_FORWARD); CHKERRQ(ierr);
365cb32e2e7SValeria Barra   ierr = VecScatterEnd(user->ltog0, user->Yloc, Y, ADD_VALUES, SCATTER_FORWARD);
366cb32e2e7SValeria Barra   CHKERRQ(ierr);
367cb32e2e7SValeria Barra 
368cb32e2e7SValeria Barra   PetscFunctionReturn(0);
369cb32e2e7SValeria Barra }
370cb32e2e7SValeria Barra 
371cb32e2e7SValeria Barra // This function calculates the error in the final solution
372c70bd2a0Sjeremylt static PetscErrorCode ComputeErrorMax(User user, CeedOperator operror, Vec X,
373cb32e2e7SValeria Barra                                       CeedVector target, PetscReal *maxerror) {
374cb32e2e7SValeria Barra   PetscErrorCode ierr;
375cb32e2e7SValeria Barra   PetscScalar *x;
376cb32e2e7SValeria Barra   CeedVector collocated_error;
377cb32e2e7SValeria Barra   CeedInt length;
378cb32e2e7SValeria Barra 
379cb32e2e7SValeria Barra   PetscFunctionBeginUser;
3809396343dSjeremylt 
381cb32e2e7SValeria Barra   CeedVectorGetLength(target, &length);
382cb32e2e7SValeria Barra   CeedVectorCreate(user->ceed, length, &collocated_error);
383cb32e2e7SValeria Barra 
384cb32e2e7SValeria Barra   // Global-to-local
385cb32e2e7SValeria Barra   ierr = VecScatterBegin(user->ltog, X, user->Xloc, INSERT_VALUES,
386cb32e2e7SValeria Barra                          SCATTER_REVERSE); CHKERRQ(ierr);
3879396343dSjeremylt   ierr = VecScatterEnd(user->ltog, X, user->Xloc, INSERT_VALUES,
3889396343dSjeremylt                        SCATTER_REVERSE); CHKERRQ(ierr);
3899396343dSjeremylt 
3909396343dSjeremylt   // Setup libCEED vector
3919396343dSjeremylt   ierr = user->VecGetArrayRead(user->Xloc, (const PetscScalar **)&x);
392cb32e2e7SValeria Barra   CHKERRQ(ierr);
3939396343dSjeremylt   CeedVectorSetArray(user->xceed, user->memtype, CEED_USE_POINTER, x);
394cb32e2e7SValeria Barra 
3959396343dSjeremylt   // Apply libCEED operator
396c70bd2a0Sjeremylt   CeedOperatorApply(operror, user->xceed, collocated_error,
397cb32e2e7SValeria Barra                     CEED_REQUEST_IMMEDIATE);
398cb32e2e7SValeria Barra 
399cb32e2e7SValeria Barra   // Restore PETSc vector
4009396343dSjeremylt   ierr = user->VecRestoreArrayRead(user->Xloc, (const PetscScalar **)&x);
4019396343dSjeremylt   CHKERRQ(ierr);
402cb32e2e7SValeria Barra 
403cb32e2e7SValeria Barra   // Reduce max error
404cb32e2e7SValeria Barra   *maxerror = 0;
405cb32e2e7SValeria Barra   const CeedScalar *e;
406cb32e2e7SValeria Barra   CeedVectorGetArrayRead(collocated_error, CEED_MEM_HOST, &e);
407cb32e2e7SValeria Barra   for (CeedInt i=0; i<length; i++) {
408cb32e2e7SValeria Barra     *maxerror = PetscMax(*maxerror, PetscAbsScalar(e[i]));
409cb32e2e7SValeria Barra   }
410cb32e2e7SValeria Barra   CeedVectorRestoreArrayRead(collocated_error, &e);
4119396343dSjeremylt   ierr = MPI_Allreduce(MPI_IN_PLACE, maxerror, 1, MPIU_REAL, MPIU_MAX,
4129396343dSjeremylt                        user->comm); CHKERRQ(ierr);
413cb32e2e7SValeria Barra 
414cb32e2e7SValeria Barra   // Cleanup
415cb32e2e7SValeria Barra   CeedVectorDestroy(&collocated_error);
416cb32e2e7SValeria Barra 
417cb32e2e7SValeria Barra   PetscFunctionReturn(0);
418cb32e2e7SValeria Barra }
419cb32e2e7SValeria Barra 
420cb32e2e7SValeria Barra int main(int argc, char **argv) {
421cb32e2e7SValeria Barra   PetscInt ierr;
422cb32e2e7SValeria Barra   MPI_Comm comm;
423cb32e2e7SValeria Barra   char ceedresource[PETSC_MAX_PATH_LEN] = "/cpu/self";
424cb32e2e7SValeria Barra   double my_rt_start, my_rt, rt_min, rt_max;
425cb32e2e7SValeria Barra   PetscInt degree, qextra, localnodes, localelem, melem[3], mnodes[3], p[3],
426cb32e2e7SValeria Barra            irank[3], lnodes[3], lsize, ncompu = 1;
427cb32e2e7SValeria Barra   PetscScalar *r;
428cb32e2e7SValeria Barra   PetscBool test_mode, benchmark_mode, write_solution;
429cb32e2e7SValeria Barra   PetscMPIInt size, rank;
43009a940d7Sjeremylt   PetscLogStage solvestage;
431cb32e2e7SValeria Barra   Vec X, Xloc, rhs, rhsloc;
432cb32e2e7SValeria Barra   Mat mat;
433cb32e2e7SValeria Barra   KSP ksp;
434cb32e2e7SValeria Barra   VecScatter ltog, ltog0, gtogD;
435cb32e2e7SValeria Barra   User user;
436cb32e2e7SValeria Barra   Ceed ceed;
437cb32e2e7SValeria Barra   CeedBasis basisx, basisu;
43815910d16Sjeremylt   CeedElemRestriction Erestrictx, Erestrictu, Erestrictui, Erestrictqdi;
439c70bd2a0Sjeremylt   CeedQFunction qfsetupgeo, qfsetuprhs, qfapply, qferror;
440c70bd2a0Sjeremylt   CeedOperator opsetupgeo, opsetuprhs, opapply, operror;
441cb32e2e7SValeria Barra   CeedVector xcoord, qdata, rhsceed, target;
4429396343dSjeremylt   CeedMemType memtyperequested;
443cb32e2e7SValeria Barra   CeedInt P, Q;
444cb32e2e7SValeria Barra   const CeedInt dim = 3, ncompx = 3;
445199551f5Sjeremylt   bpType bpchoice;
446cb32e2e7SValeria Barra 
4479396343dSjeremylt   // Check for PETSc CUDA availability
4489396343dSjeremylt   PetscBool petschavecuda, setmemtyperequest = PETSC_FALSE;
4499396343dSjeremylt   // *INDENT-OFF*
4509396343dSjeremylt   #ifdef PETSC_HAVE_CUDA
4519396343dSjeremylt   petschavecuda = PETSC_TRUE;
4529396343dSjeremylt   #else
4539396343dSjeremylt   petschavecuda = PETSC_FALSE;
4549396343dSjeremylt   #endif
4559396343dSjeremylt   // *INDENT-ON*
4569396343dSjeremylt 
457cb32e2e7SValeria Barra   ierr = PetscInitialize(&argc, &argv, NULL, help);
458cb32e2e7SValeria Barra   if (ierr) return ierr;
459cb32e2e7SValeria Barra   comm = PETSC_COMM_WORLD;
46032d2ee49SValeria Barra 
46132d2ee49SValeria Barra   // Read command line options
462cb32e2e7SValeria Barra   ierr = PetscOptionsBegin(comm, NULL, "CEED BPs in PETSc", NULL); CHKERRQ(ierr);
463199551f5Sjeremylt   bpchoice = CEED_BP1;
464cb32e2e7SValeria Barra   ierr = PetscOptionsEnum("-problem",
465cb32e2e7SValeria Barra                           "CEED benchmark problem to solve", NULL,
466199551f5Sjeremylt                           bpTypes, (PetscEnum)bpchoice, (PetscEnum *)&bpchoice,
467cb32e2e7SValeria Barra                           NULL); CHKERRQ(ierr);
468199551f5Sjeremylt   ncompu = bpOptions[bpchoice].ncompu;
469cb32e2e7SValeria Barra   test_mode = PETSC_FALSE;
470cb32e2e7SValeria Barra   ierr = PetscOptionsBool("-test",
471cb32e2e7SValeria Barra                           "Testing mode (do not print unless error is large)",
472cb32e2e7SValeria Barra                           NULL, test_mode, &test_mode, NULL); CHKERRQ(ierr);
473cb32e2e7SValeria Barra   benchmark_mode = PETSC_FALSE;
474cb32e2e7SValeria Barra   ierr = PetscOptionsBool("-benchmark",
475cb32e2e7SValeria Barra                           "Benchmarking mode (prints benchmark statistics)",
476cb32e2e7SValeria Barra                           NULL, benchmark_mode, &benchmark_mode, NULL);
477cb32e2e7SValeria Barra   CHKERRQ(ierr);
478cb32e2e7SValeria Barra   write_solution = PETSC_FALSE;
479cb32e2e7SValeria Barra   ierr = PetscOptionsBool("-write_solution",
480cb32e2e7SValeria Barra                           "Write solution for visualization",
481cb32e2e7SValeria Barra                           NULL, write_solution, &write_solution, NULL);
482cb32e2e7SValeria Barra   CHKERRQ(ierr);
483cb32e2e7SValeria Barra   degree = test_mode ? 3 : 1;
484cb32e2e7SValeria Barra   ierr = PetscOptionsInt("-degree", "Polynomial degree of tensor product basis",
485cb32e2e7SValeria Barra                          NULL, degree, &degree, NULL); CHKERRQ(ierr);
486199551f5Sjeremylt   qextra = bpOptions[bpchoice].qextra;
487cb32e2e7SValeria Barra   ierr = PetscOptionsInt("-qextra", "Number of extra quadrature points",
488cb32e2e7SValeria Barra                          NULL, qextra, &qextra, NULL); CHKERRQ(ierr);
489cb32e2e7SValeria Barra   ierr = PetscOptionsString("-ceed", "CEED resource specifier",
490cb32e2e7SValeria Barra                             NULL, ceedresource, ceedresource,
491cb32e2e7SValeria Barra                             sizeof(ceedresource), NULL); CHKERRQ(ierr);
492cb32e2e7SValeria Barra   localnodes = 1000;
493cb32e2e7SValeria Barra   ierr = PetscOptionsInt("-local",
494cb32e2e7SValeria Barra                          "Target number of locally owned nodes per process",
495cb32e2e7SValeria Barra                          NULL, localnodes, &localnodes, NULL); CHKERRQ(ierr);
4969396343dSjeremylt   memtyperequested = petschavecuda ? CEED_MEM_DEVICE : CEED_MEM_HOST;
4979396343dSjeremylt   ierr = PetscOptionsEnum("-memtype",
4989396343dSjeremylt                           "CEED MemType requested", NULL,
4999396343dSjeremylt                           memTypes, (PetscEnum)memtyperequested,
5009396343dSjeremylt                           (PetscEnum *)&memtyperequested, &setmemtyperequest);
5019396343dSjeremylt   CHKERRQ(ierr);
502cb32e2e7SValeria Barra   ierr = PetscOptionsEnd(); CHKERRQ(ierr);
503cb32e2e7SValeria Barra   P = degree + 1;
504cb32e2e7SValeria Barra   Q = P + qextra;
505cb32e2e7SValeria Barra 
5069396343dSjeremylt   // Set up libCEED
5079396343dSjeremylt   CeedInit(ceedresource, &ceed);
5089396343dSjeremylt   CeedMemType memtypebackend;
5099396343dSjeremylt   CeedGetPreferredMemType(ceed, &memtypebackend);
5109396343dSjeremylt 
5119396343dSjeremylt   // Check memtype compatibility
5129396343dSjeremylt   if (!setmemtyperequest)
5139396343dSjeremylt     memtyperequested = memtypebackend;
5149396343dSjeremylt   else if (!petschavecuda && memtyperequested == CEED_MEM_DEVICE)
5159396343dSjeremylt     SETERRQ1(PETSC_COMM_WORLD, PETSC_ERR_SUP_SYS,
5169396343dSjeremylt              "PETSc was not built with CUDA. "
5179396343dSjeremylt              "Requested MemType CEED_MEM_DEVICE is not supported.", NULL);
5189396343dSjeremylt 
519cb32e2e7SValeria Barra   // Determine size of process grid
520cb32e2e7SValeria Barra   ierr = MPI_Comm_size(comm, &size); CHKERRQ(ierr);
521cb32e2e7SValeria Barra   Split3(size, p, false);
522cb32e2e7SValeria Barra 
523cb32e2e7SValeria Barra   // Find a nicely composite number of elements no less than localnodes
524cb32e2e7SValeria Barra   for (localelem = PetscMax(1, localnodes / (degree*degree*degree)); ;
525cb32e2e7SValeria Barra        localelem++) {
526cb32e2e7SValeria Barra     Split3(localelem, melem, true);
527cb32e2e7SValeria Barra     if (Max3(melem) / Min3(melem) <= 2) break;
528cb32e2e7SValeria Barra   }
529cb32e2e7SValeria Barra 
530cb32e2e7SValeria Barra   // Find my location in the process grid
531cb32e2e7SValeria Barra   ierr = MPI_Comm_rank(comm, &rank); CHKERRQ(ierr);
532cb32e2e7SValeria Barra   for (int d=0,rankleft=rank; d<dim; d++) {
533cb32e2e7SValeria Barra     const int pstride[3] = {p[1] *p[2], p[2], 1};
534cb32e2e7SValeria Barra     irank[d] = rankleft / pstride[d];
535cb32e2e7SValeria Barra     rankleft -= irank[d] * pstride[d];
536cb32e2e7SValeria Barra   }
537cb32e2e7SValeria Barra 
538cb32e2e7SValeria Barra   GlobalNodes(p, irank, degree, melem, mnodes);
539cb32e2e7SValeria Barra 
540cb32e2e7SValeria Barra   // Setup global vector
5414a5da23aSJed Brown   ierr = VecCreate(comm, &X); CHKERRQ(ierr);
5429396343dSjeremylt   if (memtyperequested == CEED_MEM_DEVICE) {
5439396343dSjeremylt     ierr = VecSetType(X, VECCUDA); CHKERRQ(ierr);
5449396343dSjeremylt   }
545cb32e2e7SValeria Barra   ierr = VecSetSizes(X, mnodes[0]*mnodes[1]*mnodes[2]*ncompu, PETSC_DECIDE);
546cb32e2e7SValeria Barra   CHKERRQ(ierr);
547cb32e2e7SValeria Barra   ierr = VecSetUp(X); CHKERRQ(ierr);
548cb32e2e7SValeria Barra 
549cb32e2e7SValeria Barra   // Set up libCEED
550cb32e2e7SValeria Barra   CeedInit(ceedresource, &ceed);
551cb32e2e7SValeria Barra 
552cb32e2e7SValeria Barra   // Print summary
553cb32e2e7SValeria Barra   CeedInt gsize;
554cb32e2e7SValeria Barra   ierr = VecGetSize(X, &gsize); CHKERRQ(ierr);
555dc7d240cSValeria Barra   if (!test_mode) {
556cb32e2e7SValeria Barra     const char *usedresource;
557cb32e2e7SValeria Barra     CeedGetResource(ceed, &usedresource);
5589396343dSjeremylt 
5599396343dSjeremylt     VecType vectype;
5609396343dSjeremylt     ierr = VecGetType(X, &vectype); CHKERRQ(ierr);
5619396343dSjeremylt 
562cb32e2e7SValeria Barra     ierr = PetscPrintf(comm,
563cb32e2e7SValeria Barra                        "\n-- CEED Benchmark Problem %d -- libCEED + PETSc --\n"
5649396343dSjeremylt                        "  PETSc:\n"
5659396343dSjeremylt                        "    PETSc Vec Type                     : %s\n"
566cb32e2e7SValeria Barra                        "  libCEED:\n"
567cb32e2e7SValeria Barra                        "    libCEED Backend                    : %s\n"
5689396343dSjeremylt                        "    libCEED Backend MemType            : %s\n"
5699396343dSjeremylt                        "    libCEED User Requested MemType     : %s\n"
570cb32e2e7SValeria Barra                        "  Mesh:\n"
571cb32e2e7SValeria Barra                        "    Number of 1D Basis Nodes (p)       : %d\n"
572cb32e2e7SValeria Barra                        "    Number of 1D Quadrature Points (q) : %d\n"
573cb32e2e7SValeria Barra                        "    Global nodes                       : %D\n"
574cb32e2e7SValeria Barra                        "    Process Decomposition              : %D %D %D\n"
575cb32e2e7SValeria Barra                        "    Local Elements                     : %D = %D %D %D\n"
576db419314Sjeremylt                        "    Owned nodes                        : %D = %D %D %D\n"
577db419314Sjeremylt                        "    DoF per node                       : %D\n",
5789396343dSjeremylt                        bpchoice+1, vectype, usedresource,
5799396343dSjeremylt                        CeedMemTypes[memtypebackend],
5809396343dSjeremylt                        (setmemtyperequest) ?
5819396343dSjeremylt                        CeedMemTypes[memtyperequested] : "none",
5829396343dSjeremylt                        P, Q,  gsize/ncompu, p[0], p[1], p[2], localelem,
5839396343dSjeremylt                        melem[0], melem[1], melem[2],
584cb32e2e7SValeria Barra                        mnodes[0]*mnodes[1]*mnodes[2], mnodes[0], mnodes[1],
585db419314Sjeremylt                        mnodes[2], ncompu); CHKERRQ(ierr);
586cb32e2e7SValeria Barra   }
587cb32e2e7SValeria Barra 
588cb32e2e7SValeria Barra   {
589cb32e2e7SValeria Barra     lsize = 1;
590cb32e2e7SValeria Barra     for (int d=0; d<dim; d++) {
591cb32e2e7SValeria Barra       lnodes[d] = melem[d]*degree + 1;
592cb32e2e7SValeria Barra       lsize *= lnodes[d];
593cb32e2e7SValeria Barra     }
5944a5da23aSJed Brown     ierr = VecCreate(PETSC_COMM_SELF, &Xloc); CHKERRQ(ierr);
5959396343dSjeremylt     if (memtyperequested == CEED_MEM_DEVICE) {
5969396343dSjeremylt       ierr = VecSetType(Xloc, VECCUDA); CHKERRQ(ierr);
5979396343dSjeremylt     }
598cb32e2e7SValeria Barra     ierr = VecSetSizes(Xloc, lsize*ncompu, PETSC_DECIDE); CHKERRQ(ierr);
599cb32e2e7SValeria Barra     ierr = VecSetUp(Xloc); CHKERRQ(ierr);
600cb32e2e7SValeria Barra 
601cb32e2e7SValeria Barra     // Create local-to-global scatter
602cb32e2e7SValeria Barra     PetscInt *ltogind, *ltogind0, *locind, l0count;
603cb32e2e7SValeria Barra     IS ltogis, ltogis0, locis;
604cb32e2e7SValeria Barra     PetscInt gstart[2][2][2], gmnodes[2][2][2][dim];
605cb32e2e7SValeria Barra 
606cb32e2e7SValeria Barra     for (int i=0; i<2; i++) {
607cb32e2e7SValeria Barra       for (int j=0; j<2; j++) {
608cb32e2e7SValeria Barra         for (int k=0; k<2; k++) {
609cb32e2e7SValeria Barra           PetscInt ijkrank[3] = {irank[0]+i, irank[1]+j, irank[2]+k};
610cb32e2e7SValeria Barra           gstart[i][j][k] = GlobalStart(p, ijkrank, degree, melem);
611cb32e2e7SValeria Barra           GlobalNodes(p, ijkrank, degree, melem, gmnodes[i][j][k]);
612cb32e2e7SValeria Barra         }
613cb32e2e7SValeria Barra       }
614cb32e2e7SValeria Barra     }
615cb32e2e7SValeria Barra 
616cb32e2e7SValeria Barra     ierr = PetscMalloc1(lsize, &ltogind); CHKERRQ(ierr);
617cb32e2e7SValeria Barra     ierr = PetscMalloc1(lsize, &ltogind0); CHKERRQ(ierr);
618cb32e2e7SValeria Barra     ierr = PetscMalloc1(lsize, &locind); CHKERRQ(ierr);
619cb32e2e7SValeria Barra     l0count = 0;
620cb32e2e7SValeria Barra     for (PetscInt i=0,ir,ii; ir=i>=mnodes[0], ii=i-ir*mnodes[0], i<lnodes[0]; i++)
621cb32e2e7SValeria Barra       for (PetscInt j=0,jr,jj; jr=j>=mnodes[1], jj=j-jr*mnodes[1], j<lnodes[1]; j++)
622cb32e2e7SValeria Barra         for (PetscInt k=0,kr,kk; kr=k>=mnodes[2], kk=k-kr*mnodes[2], k<lnodes[2]; k++) {
623cb32e2e7SValeria Barra           PetscInt here = (i*lnodes[1]+j)*lnodes[2]+k;
624cb32e2e7SValeria Barra           ltogind[here] =
625cb32e2e7SValeria Barra             gstart[ir][jr][kr] + (ii*gmnodes[ir][jr][kr][1]+jj)*gmnodes[ir][jr][kr][2]+kk;
626cb32e2e7SValeria Barra           if ((irank[0] == 0 && i == 0)
627cb32e2e7SValeria Barra               || (irank[1] == 0 && j == 0)
628cb32e2e7SValeria Barra               || (irank[2] == 0 && k == 0)
629cb32e2e7SValeria Barra               || (irank[0]+1 == p[0] && i+1 == lnodes[0])
630cb32e2e7SValeria Barra               || (irank[1]+1 == p[1] && j+1 == lnodes[1])
631cb32e2e7SValeria Barra               || (irank[2]+1 == p[2] && k+1 == lnodes[2]))
632cb32e2e7SValeria Barra             continue;
633cb32e2e7SValeria Barra           ltogind0[l0count] = ltogind[here];
634cb32e2e7SValeria Barra           locind[l0count++] = here;
635cb32e2e7SValeria Barra         }
636cb32e2e7SValeria Barra     ierr = ISCreateBlock(comm, ncompu, lsize, ltogind, PETSC_OWN_POINTER,
637cb32e2e7SValeria Barra                          &ltogis); CHKERRQ(ierr);
638cb32e2e7SValeria Barra     ierr = VecScatterCreate(Xloc, NULL, X, ltogis, &ltog); CHKERRQ(ierr);
639cb32e2e7SValeria Barra     CHKERRQ(ierr);
640cb32e2e7SValeria Barra     ierr = ISCreateBlock(comm, ncompu, l0count, ltogind0, PETSC_OWN_POINTER,
641cb32e2e7SValeria Barra                          &ltogis0); CHKERRQ(ierr);
642cb32e2e7SValeria Barra     ierr = ISCreateBlock(comm, ncompu, l0count, locind, PETSC_OWN_POINTER,
643cb32e2e7SValeria Barra                          &locis); CHKERRQ(ierr);
644cb32e2e7SValeria Barra     ierr = VecScatterCreate(Xloc, locis, X, ltogis0, &ltog0); CHKERRQ(ierr);
645cb32e2e7SValeria Barra     {
646cb32e2e7SValeria Barra       // Create global-to-global scatter for Dirichlet values (everything not in
647cb32e2e7SValeria Barra       // ltogis0, which is the range of ltog0)
648cb32e2e7SValeria Barra       PetscInt xstart, xend, *indD, countD = 0;
649cb32e2e7SValeria Barra       IS isD;
650cb32e2e7SValeria Barra       const PetscScalar *x;
651cb32e2e7SValeria Barra       ierr = VecZeroEntries(Xloc); CHKERRQ(ierr);
652cb32e2e7SValeria Barra       ierr = VecSet(X, 1.0); CHKERRQ(ierr);
653cb32e2e7SValeria Barra       ierr = VecScatterBegin(ltog0, Xloc, X, INSERT_VALUES, SCATTER_FORWARD);
654cb32e2e7SValeria Barra       CHKERRQ(ierr);
655cb32e2e7SValeria Barra       ierr = VecScatterEnd(ltog0, Xloc, X, INSERT_VALUES, SCATTER_FORWARD);
656cb32e2e7SValeria Barra       CHKERRQ(ierr);
657cb32e2e7SValeria Barra       ierr = VecGetOwnershipRange(X, &xstart, &xend); CHKERRQ(ierr);
658cb32e2e7SValeria Barra       ierr = PetscMalloc1(xend-xstart, &indD); CHKERRQ(ierr);
659cb32e2e7SValeria Barra       ierr = VecGetArrayRead(X, &x); CHKERRQ(ierr);
660cb32e2e7SValeria Barra       for (PetscInt i=0; i<xend-xstart; i++) {
661cb32e2e7SValeria Barra         if (x[i] == 1.) indD[countD++] = xstart + i;
662cb32e2e7SValeria Barra       }
663cb32e2e7SValeria Barra       ierr = VecRestoreArrayRead(X, &x); CHKERRQ(ierr);
664cb32e2e7SValeria Barra       ierr = ISCreateGeneral(comm, countD, indD, PETSC_COPY_VALUES, &isD);
665cb32e2e7SValeria Barra       CHKERRQ(ierr);
666cb32e2e7SValeria Barra       ierr = PetscFree(indD); CHKERRQ(ierr);
667cb32e2e7SValeria Barra       ierr = VecScatterCreate(X, isD, X, isD, &gtogD); CHKERRQ(ierr);
668cb32e2e7SValeria Barra       ierr = ISDestroy(&isD); CHKERRQ(ierr);
669cb32e2e7SValeria Barra     }
670cb32e2e7SValeria Barra     ierr = ISDestroy(&ltogis); CHKERRQ(ierr);
671cb32e2e7SValeria Barra     ierr = ISDestroy(&ltogis0); CHKERRQ(ierr);
672cb32e2e7SValeria Barra     ierr = ISDestroy(&locis); CHKERRQ(ierr);
673cb32e2e7SValeria Barra   }
674cb32e2e7SValeria Barra 
675cb32e2e7SValeria Barra   // CEED bases
676cb32e2e7SValeria Barra   CeedBasisCreateTensorH1Lagrange(ceed, dim, ncompu, P, Q,
677199551f5Sjeremylt                                   bpOptions[bpchoice].qmode, &basisu);
678cb32e2e7SValeria Barra   CeedBasisCreateTensorH1Lagrange(ceed, dim, ncompx, 2, Q,
679199551f5Sjeremylt                                   bpOptions[bpchoice].qmode, &basisx);
680cb32e2e7SValeria Barra 
681cb32e2e7SValeria Barra   // CEED restrictions
68274312f09Sjeremylt   CreateRestriction(ceed, CEED_INTERLACED, melem, P, ncompu, &Erestrictu);
68374312f09Sjeremylt   CreateRestriction(ceed, CEED_NONINTERLACED, melem, 2, dim, &Erestrictx);
684cb32e2e7SValeria Barra   CeedInt nelem = melem[0]*melem[1]*melem[2];
6857509a596Sjeremylt   CeedElemRestrictionCreateStrided(ceed, nelem, Q*Q*Q, nelem*Q*Q*Q, ncompu,
686523b8ea0Sjeremylt                                    CEED_STRIDES_BACKEND, &Erestrictui);
6877509a596Sjeremylt   CeedElemRestrictionCreateStrided(ceed, nelem, Q*Q*Q, nelem*Q*Q*Q,
688199551f5Sjeremylt                                    bpOptions[bpchoice].qdatasize,
689523b8ea0Sjeremylt                                    CEED_STRIDES_BACKEND, &Erestrictqdi);
690cb32e2e7SValeria Barra   {
691cb32e2e7SValeria Barra     CeedScalar *xloc;
692cb32e2e7SValeria Barra     CeedInt shape[3] = {melem[0]+1, melem[1]+1, melem[2]+1}, len =
693cb32e2e7SValeria Barra                          shape[0]*shape[1]*shape[2];
694cb32e2e7SValeria Barra     xloc = malloc(len*ncompx*sizeof xloc[0]);
695cb32e2e7SValeria Barra     for (CeedInt i=0; i<shape[0]; i++) {
696cb32e2e7SValeria Barra       for (CeedInt j=0; j<shape[1]; j++) {
697cb32e2e7SValeria Barra         for (CeedInt k=0; k<shape[2]; k++) {
698cb32e2e7SValeria Barra           xloc[((i*shape[1]+j)*shape[2]+k) + 0*len] = 1.*(irank[0]*melem[0]+i) /
699cb32e2e7SValeria Barra               (p[0]*melem[0]);
700cb32e2e7SValeria Barra           xloc[((i*shape[1]+j)*shape[2]+k) + 1*len] = 1.*(irank[1]*melem[1]+j) /
701cb32e2e7SValeria Barra               (p[1]*melem[1]);
702cb32e2e7SValeria Barra           xloc[((i*shape[1]+j)*shape[2]+k) + 2*len] = 1.*(irank[2]*melem[2]+k) /
703cb32e2e7SValeria Barra               (p[2]*melem[2]);
704cb32e2e7SValeria Barra         }
705cb32e2e7SValeria Barra       }
706cb32e2e7SValeria Barra     }
707cb32e2e7SValeria Barra     CeedVectorCreate(ceed, len*ncompx, &xcoord);
708cb32e2e7SValeria Barra     CeedVectorSetArray(xcoord, CEED_MEM_HOST, CEED_OWN_POINTER, xloc);
709cb32e2e7SValeria Barra   }
710cb32e2e7SValeria Barra 
711cb32e2e7SValeria Barra   // Create the Qfunction that builds the operator quadrature data
712199551f5Sjeremylt   CeedQFunctionCreateInterior(ceed, 1, bpOptions[bpchoice].setupgeo,
713199551f5Sjeremylt                               bpOptions[bpchoice].setupgeofname, &qfsetupgeo);
714c70bd2a0Sjeremylt   CeedQFunctionAddInput(qfsetupgeo, "dx", ncompx*dim, CEED_EVAL_GRAD);
715c70bd2a0Sjeremylt   CeedQFunctionAddInput(qfsetupgeo, "weight", 1, CEED_EVAL_WEIGHT);
716199551f5Sjeremylt   CeedQFunctionAddOutput(qfsetupgeo, "qdata", bpOptions[bpchoice].qdatasize,
717cb32e2e7SValeria Barra                          CEED_EVAL_NONE);
718cb32e2e7SValeria Barra 
719cb32e2e7SValeria Barra   // Create the Qfunction that sets up the RHS and true solution
720199551f5Sjeremylt   CeedQFunctionCreateInterior(ceed, 1, bpOptions[bpchoice].setuprhs,
721199551f5Sjeremylt                               bpOptions[bpchoice].setuprhsfname, &qfsetuprhs);
722c70bd2a0Sjeremylt   CeedQFunctionAddInput(qfsetuprhs, "x", ncompx, CEED_EVAL_INTERP);
723c70bd2a0Sjeremylt   CeedQFunctionAddInput(qfsetuprhs, "dx", ncompx*dim, CEED_EVAL_GRAD);
724c70bd2a0Sjeremylt   CeedQFunctionAddInput(qfsetuprhs, "weight", 1, CEED_EVAL_WEIGHT);
725c70bd2a0Sjeremylt   CeedQFunctionAddOutput(qfsetuprhs, "true_soln", ncompu, CEED_EVAL_NONE);
726c70bd2a0Sjeremylt   CeedQFunctionAddOutput(qfsetuprhs, "rhs", ncompu, CEED_EVAL_INTERP);
727cb32e2e7SValeria Barra 
728cb32e2e7SValeria Barra   // Set up PDE operator
729199551f5Sjeremylt   CeedQFunctionCreateInterior(ceed, 1, bpOptions[bpchoice].apply,
730199551f5Sjeremylt                               bpOptions[bpchoice].applyfname, &qfapply);
731cb32e2e7SValeria Barra   // Add inputs and outputs
732199551f5Sjeremylt   CeedInt inscale = bpOptions[bpchoice].inmode==CEED_EVAL_GRAD ? 3 : 1;
733199551f5Sjeremylt   CeedInt outscale = bpOptions[bpchoice].outmode==CEED_EVAL_GRAD ? 3 : 1;
734c70bd2a0Sjeremylt   CeedQFunctionAddInput(qfapply, "u", ncompu*inscale,
735199551f5Sjeremylt                         bpOptions[bpchoice].inmode);
736199551f5Sjeremylt   CeedQFunctionAddInput(qfapply, "qdata", bpOptions[bpchoice].qdatasize,
737cb32e2e7SValeria Barra                         CEED_EVAL_NONE);
738c70bd2a0Sjeremylt   CeedQFunctionAddOutput(qfapply, "v", ncompu*outscale,
739199551f5Sjeremylt                          bpOptions[bpchoice].outmode);
740cb32e2e7SValeria Barra 
741cb32e2e7SValeria Barra   // Create the error qfunction
742199551f5Sjeremylt   CeedQFunctionCreateInterior(ceed, 1, bpOptions[bpchoice].error,
743199551f5Sjeremylt                               bpOptions[bpchoice].errorfname, &qferror);
744c70bd2a0Sjeremylt   CeedQFunctionAddInput(qferror, "u", ncompu, CEED_EVAL_INTERP);
745c70bd2a0Sjeremylt   CeedQFunctionAddInput(qferror, "true_soln", ncompu, CEED_EVAL_NONE);
746c70bd2a0Sjeremylt   CeedQFunctionAddOutput(qferror, "error", ncompu, CEED_EVAL_NONE);
747cb32e2e7SValeria Barra 
748cb32e2e7SValeria Barra   // Create the persistent vectors that will be needed in setup
749cb32e2e7SValeria Barra   CeedInt nqpts;
750cb32e2e7SValeria Barra   CeedBasisGetNumQuadraturePoints(basisu, &nqpts);
751199551f5Sjeremylt   CeedVectorCreate(ceed, bpOptions[bpchoice].qdatasize*nelem*nqpts, &qdata);
752cb32e2e7SValeria Barra   CeedVectorCreate(ceed, nelem*nqpts*ncompu, &target);
753cb32e2e7SValeria Barra   CeedVectorCreate(ceed, lsize*ncompu, &rhsceed);
754cb32e2e7SValeria Barra 
755cb32e2e7SValeria Barra   // Create the operator that builds the quadrature data for the ceed operator
756c70bd2a0Sjeremylt   CeedOperatorCreate(ceed, qfsetupgeo, CEED_QFUNCTION_NONE,
757c70bd2a0Sjeremylt                      CEED_QFUNCTION_NONE, &opsetupgeo);
758c70bd2a0Sjeremylt   CeedOperatorSetField(opsetupgeo, "dx", Erestrictx, basisx,
759a8d32208Sjeremylt                        CEED_VECTOR_ACTIVE);
760c70bd2a0Sjeremylt   CeedOperatorSetField(opsetupgeo, "weight", CEED_ELEMRESTRICTION_NONE, basisx,
761a8d32208Sjeremylt                        CEED_VECTOR_NONE);
762c70bd2a0Sjeremylt   CeedOperatorSetField(opsetupgeo, "qdata", Erestrictqdi,
763cb32e2e7SValeria Barra                        CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
764cb32e2e7SValeria Barra 
765cb32e2e7SValeria Barra   // Create the operator that builds the RHS and true solution
766c70bd2a0Sjeremylt   CeedOperatorCreate(ceed, qfsetuprhs, CEED_QFUNCTION_NONE,
767c70bd2a0Sjeremylt                      CEED_QFUNCTION_NONE, &opsetuprhs);
768c70bd2a0Sjeremylt   CeedOperatorSetField(opsetuprhs, "x", Erestrictx, basisx,
769a8d32208Sjeremylt                        CEED_VECTOR_ACTIVE);
770c70bd2a0Sjeremylt   CeedOperatorSetField(opsetuprhs, "dx", Erestrictx, basisx,
771a8d32208Sjeremylt                        CEED_VECTOR_ACTIVE);
772c70bd2a0Sjeremylt   CeedOperatorSetField(opsetuprhs, "weight", CEED_ELEMRESTRICTION_NONE, basisx,
773a8d32208Sjeremylt                        CEED_VECTOR_NONE);
774c70bd2a0Sjeremylt   CeedOperatorSetField(opsetuprhs, "true_soln", Erestrictui,
775cb32e2e7SValeria Barra                        CEED_BASIS_COLLOCATED, target);
776c70bd2a0Sjeremylt   CeedOperatorSetField(opsetuprhs, "rhs", Erestrictu, basisu,
777a8d32208Sjeremylt                        CEED_VECTOR_ACTIVE);
778cb32e2e7SValeria Barra 
779cb32e2e7SValeria Barra   // Create the mass or diff operator
780c70bd2a0Sjeremylt   CeedOperatorCreate(ceed, qfapply, CEED_QFUNCTION_NONE, CEED_QFUNCTION_NONE,
781c70bd2a0Sjeremylt                      &opapply);
782c70bd2a0Sjeremylt   CeedOperatorSetField(opapply, "u", Erestrictu, basisu, CEED_VECTOR_ACTIVE);
783c70bd2a0Sjeremylt   CeedOperatorSetField(opapply, "qdata", Erestrictqdi, CEED_BASIS_COLLOCATED,
784a8d32208Sjeremylt                        qdata);
785c70bd2a0Sjeremylt   CeedOperatorSetField(opapply, "v", Erestrictu, basisu, CEED_VECTOR_ACTIVE);
786cb32e2e7SValeria Barra 
787cb32e2e7SValeria Barra   // Create the error operator
788c70bd2a0Sjeremylt   CeedOperatorCreate(ceed, qferror, CEED_QFUNCTION_NONE, CEED_QFUNCTION_NONE,
789c70bd2a0Sjeremylt                      &operror);
790c70bd2a0Sjeremylt   CeedOperatorSetField(operror, "u", Erestrictu, basisu, CEED_VECTOR_ACTIVE);
791c70bd2a0Sjeremylt   CeedOperatorSetField(operror, "true_soln", Erestrictui,
792cb32e2e7SValeria Barra                        CEED_BASIS_COLLOCATED, target);
793c70bd2a0Sjeremylt   CeedOperatorSetField(operror, "error", Erestrictui, CEED_BASIS_COLLOCATED,
794a8d32208Sjeremylt                        CEED_VECTOR_ACTIVE);
795cb32e2e7SValeria Barra 
796cb32e2e7SValeria Barra   // Set up Mat
797cb32e2e7SValeria Barra   ierr = PetscMalloc1(1, &user); CHKERRQ(ierr);
798cb32e2e7SValeria Barra   user->comm = comm;
799cb32e2e7SValeria Barra   user->ltog = ltog;
800199551f5Sjeremylt   if (bpchoice != CEED_BP1 && bpchoice != CEED_BP2) {
801cb32e2e7SValeria Barra     user->ltog0 = ltog0;
802cb32e2e7SValeria Barra     user->gtogD = gtogD;
803cb32e2e7SValeria Barra   }
804cb32e2e7SValeria Barra   user->Xloc = Xloc;
805cb32e2e7SValeria Barra   ierr = VecDuplicate(Xloc, &user->Yloc); CHKERRQ(ierr);
806cb32e2e7SValeria Barra   CeedVectorCreate(ceed, lsize*ncompu, &user->xceed);
807cb32e2e7SValeria Barra   CeedVectorCreate(ceed, lsize*ncompu, &user->yceed);
808c70bd2a0Sjeremylt   user->op = opapply;
809cb32e2e7SValeria Barra   user->qdata = qdata;
810cb32e2e7SValeria Barra   user->ceed = ceed;
8119396343dSjeremylt   user->memtype = memtyperequested;
8129396343dSjeremylt   if (memtyperequested == CEED_MEM_HOST) {
8139396343dSjeremylt     user->VecGetArray = VecGetArray;
8149396343dSjeremylt     user->VecGetArrayRead = VecGetArrayRead;
8159396343dSjeremylt     user->VecRestoreArray = VecRestoreArray;
8169396343dSjeremylt     user->VecRestoreArrayRead = VecRestoreArrayRead;
8179396343dSjeremylt   } else {
8189396343dSjeremylt     user->VecGetArray = VecCUDAGetArray;
8199396343dSjeremylt     user->VecGetArrayRead = VecCUDAGetArrayRead;
8209396343dSjeremylt     user->VecRestoreArray = VecCUDARestoreArray;
8219396343dSjeremylt     user->VecRestoreArrayRead = VecCUDARestoreArrayRead;
8229396343dSjeremylt   }
823cb32e2e7SValeria Barra 
824cb32e2e7SValeria Barra   ierr = MatCreateShell(comm, mnodes[0]*mnodes[1]*mnodes[2]*ncompu,
825cb32e2e7SValeria Barra                         mnodes[0]*mnodes[1]*mnodes[2]*ncompu,
826cb32e2e7SValeria Barra                         PETSC_DECIDE, PETSC_DECIDE, user, &mat); CHKERRQ(ierr);
827199551f5Sjeremylt   if (bpchoice == CEED_BP1 || bpchoice == CEED_BP2) {
828cb32e2e7SValeria Barra     ierr = MatShellSetOperation(mat, MATOP_MULT, (void(*)(void))MatMult_Mass);
829cb32e2e7SValeria Barra     CHKERRQ(ierr);
830cb32e2e7SValeria Barra   } else {
831cb32e2e7SValeria Barra     ierr = MatShellSetOperation(mat, MATOP_MULT, (void(*)(void))MatMult_Diff);
832cb32e2e7SValeria Barra     CHKERRQ(ierr);
833cb32e2e7SValeria Barra   }
8349396343dSjeremylt   if (user->memtype == CEED_MEM_DEVICE) {
8359396343dSjeremylt     ierr = MatShellSetVecType(mat, VECCUDA); CHKERRQ(ierr);
8369396343dSjeremylt   }
837cb32e2e7SValeria Barra 
838cb32e2e7SValeria Barra   // Get RHS vector
8399396343dSjeremylt   ierr = VecDuplicate(X, &rhs); CHKERRQ(ierr);
840cb32e2e7SValeria Barra   ierr = VecDuplicate(Xloc, &rhsloc); CHKERRQ(ierr);
841cb32e2e7SValeria Barra   ierr = VecZeroEntries(rhsloc); CHKERRQ(ierr);
8429396343dSjeremylt   ierr = user->VecGetArray(rhsloc, &r); CHKERRQ(ierr);
8439396343dSjeremylt   CeedVectorSetArray(rhsceed, user->memtype, CEED_USE_POINTER, r);
844cb32e2e7SValeria Barra 
845cb32e2e7SValeria Barra   // Setup qdata, rhs, and target
846c70bd2a0Sjeremylt   CeedOperatorApply(opsetupgeo, xcoord, qdata, CEED_REQUEST_IMMEDIATE);
847c70bd2a0Sjeremylt   CeedOperatorApply(opsetuprhs, xcoord, rhsceed, CEED_REQUEST_IMMEDIATE);
8489396343dSjeremylt   ierr = CeedVectorSyncArray(rhsceed, user->memtype); CHKERRQ(ierr);
849cb32e2e7SValeria Barra   CeedVectorDestroy(&xcoord);
850cb32e2e7SValeria Barra 
851cb32e2e7SValeria Barra   // Gather RHS
8529396343dSjeremylt   ierr = user->VecRestoreArray(rhsloc, &r); CHKERRQ(ierr);
853cb32e2e7SValeria Barra   ierr = VecZeroEntries(rhs); CHKERRQ(ierr);
854cb32e2e7SValeria Barra   ierr = VecScatterBegin(ltog, rhsloc, rhs, ADD_VALUES, SCATTER_FORWARD);
855cb32e2e7SValeria Barra   CHKERRQ(ierr);
856cb32e2e7SValeria Barra   ierr = VecScatterEnd(ltog, rhsloc, rhs, ADD_VALUES, SCATTER_FORWARD);
857cb32e2e7SValeria Barra   CHKERRQ(ierr);
858cb32e2e7SValeria Barra   CeedVectorDestroy(&rhsceed);
859cb32e2e7SValeria Barra 
860cb32e2e7SValeria Barra   ierr = KSPCreate(comm, &ksp); CHKERRQ(ierr);
861cb32e2e7SValeria Barra   {
862cb32e2e7SValeria Barra     PC pc;
863cb32e2e7SValeria Barra     ierr = KSPGetPC(ksp, &pc); CHKERRQ(ierr);
864199551f5Sjeremylt     if (bpchoice == CEED_BP1 || bpchoice == CEED_BP2) {
865cb32e2e7SValeria Barra       ierr = PCSetType(pc, PCJACOBI); CHKERRQ(ierr);
866cb32e2e7SValeria Barra       ierr = PCJacobiSetType(pc, PC_JACOBI_ROWSUM); CHKERRQ(ierr);
867cb32e2e7SValeria Barra     } else {
868cb32e2e7SValeria Barra       ierr = PCSetType(pc, PCNONE); CHKERRQ(ierr);
869cb32e2e7SValeria Barra     }
870cb32e2e7SValeria Barra     ierr = KSPSetType(ksp, KSPCG); CHKERRQ(ierr);
871cb32e2e7SValeria Barra     ierr = KSPSetNormType(ksp, KSP_NORM_NATURAL); CHKERRQ(ierr);
872cb32e2e7SValeria Barra     ierr = KSPSetTolerances(ksp, 1e-10, PETSC_DEFAULT, PETSC_DEFAULT,
873cb32e2e7SValeria Barra                             PETSC_DEFAULT); CHKERRQ(ierr);
874cb32e2e7SValeria Barra   }
875cb32e2e7SValeria Barra   ierr = KSPSetOperators(ksp, mat, mat); CHKERRQ(ierr);
876cb32e2e7SValeria Barra   // First run, if benchmarking
877cb32e2e7SValeria Barra   if (benchmark_mode) {
878cb32e2e7SValeria Barra     ierr = KSPSetTolerances(ksp, 1e-10, PETSC_DEFAULT, PETSC_DEFAULT, 1);
879cb32e2e7SValeria Barra     CHKERRQ(ierr);
880cb32e2e7SValeria Barra     my_rt_start = MPI_Wtime();
881cb32e2e7SValeria Barra     ierr = KSPSolve(ksp, rhs, X); CHKERRQ(ierr);
882cb32e2e7SValeria Barra     my_rt = MPI_Wtime() - my_rt_start;
883cb32e2e7SValeria Barra     ierr = MPI_Allreduce(MPI_IN_PLACE, &my_rt, 1, MPI_DOUBLE, MPI_MIN, comm);
884cb32e2e7SValeria Barra     CHKERRQ(ierr);
885cb32e2e7SValeria Barra     // Set maxits based on first iteration timing
886cb32e2e7SValeria Barra     if (my_rt > 0.02) {
887cb32e2e7SValeria Barra       ierr = KSPSetTolerances(ksp, 1e-10, PETSC_DEFAULT, PETSC_DEFAULT, 5);
888cb32e2e7SValeria Barra       CHKERRQ(ierr);
889cb32e2e7SValeria Barra     } else {
890cb32e2e7SValeria Barra       ierr = KSPSetTolerances(ksp, 1e-10, PETSC_DEFAULT, PETSC_DEFAULT, 20);
891cb32e2e7SValeria Barra       CHKERRQ(ierr);
892cb32e2e7SValeria Barra     }
893cb32e2e7SValeria Barra   }
894*debcf919SJed Brown   ierr = KSPSetFromOptions(ksp); CHKERRQ(ierr);
89509a940d7Sjeremylt 
896cb32e2e7SValeria Barra   // Timed solve
89709a940d7Sjeremylt   ierr = VecZeroEntries(X); CHKERRQ(ierr);
898cb32e2e7SValeria Barra   ierr = PetscBarrier((PetscObject)ksp); CHKERRQ(ierr);
89909a940d7Sjeremylt 
90009a940d7Sjeremylt   // -- Performance logging
90109a940d7Sjeremylt   ierr = PetscLogStageRegister("Solve Stage", &solvestage); CHKERRQ(ierr);
90209a940d7Sjeremylt   ierr = PetscLogStagePush(solvestage); CHKERRQ(ierr);
90309a940d7Sjeremylt 
90409a940d7Sjeremylt   // -- Solve
905cb32e2e7SValeria Barra   my_rt_start = MPI_Wtime();
906cb32e2e7SValeria Barra   ierr = KSPSolve(ksp, rhs, X); CHKERRQ(ierr);
907cb32e2e7SValeria Barra   my_rt = MPI_Wtime() - my_rt_start;
90809a940d7Sjeremylt 
90909a940d7Sjeremylt   // -- Performance logging
91009a940d7Sjeremylt   ierr = PetscLogStagePop();
91109a940d7Sjeremylt 
9128e87e98bSJed Brown   // Output results
913cb32e2e7SValeria Barra   {
914cb32e2e7SValeria Barra     KSPType ksptype;
915cb32e2e7SValeria Barra     KSPConvergedReason reason;
916cb32e2e7SValeria Barra     PetscReal rnorm;
917cb32e2e7SValeria Barra     PetscInt its;
918cb32e2e7SValeria Barra     ierr = KSPGetType(ksp, &ksptype); CHKERRQ(ierr);
919cb32e2e7SValeria Barra     ierr = KSPGetConvergedReason(ksp, &reason); CHKERRQ(ierr);
920cb32e2e7SValeria Barra     ierr = KSPGetIterationNumber(ksp, &its); CHKERRQ(ierr);
921cb32e2e7SValeria Barra     ierr = KSPGetResidualNorm(ksp, &rnorm); CHKERRQ(ierr);
922cb32e2e7SValeria Barra     if (!test_mode || reason < 0 || rnorm > 1e-8) {
923cb32e2e7SValeria Barra       ierr = PetscPrintf(comm,
924cb32e2e7SValeria Barra                          "  KSP:\n"
925cb32e2e7SValeria Barra                          "    KSP Type                           : %s\n"
926cb32e2e7SValeria Barra                          "    KSP Convergence                    : %s\n"
927cb32e2e7SValeria Barra                          "    Total KSP Iterations               : %D\n"
928cb32e2e7SValeria Barra                          "    Final rnorm                        : %e\n",
929cb32e2e7SValeria Barra                          ksptype, KSPConvergedReasons[reason], its,
930cb32e2e7SValeria Barra                          (double)rnorm); CHKERRQ(ierr);
931cb32e2e7SValeria Barra     }
9328e87e98bSJed Brown     if (!test_mode) {
9338e87e98bSJed Brown       ierr = PetscPrintf(comm,"  Performance:\n"); CHKERRQ(ierr);
9348e87e98bSJed Brown     }
9358e87e98bSJed Brown     {
9368e87e98bSJed Brown       PetscReal maxerror;
9378e87e98bSJed Brown       ierr = ComputeErrorMax(user, operror, X, target, &maxerror);
9388e87e98bSJed Brown       CHKERRQ(ierr);
9398e87e98bSJed Brown       PetscReal tol = 5e-2;
9408e87e98bSJed Brown       if (!test_mode || maxerror > tol) {
941cb32e2e7SValeria Barra         ierr = MPI_Allreduce(&my_rt, &rt_min, 1, MPI_DOUBLE, MPI_MIN, comm);
942cb32e2e7SValeria Barra         CHKERRQ(ierr);
943cb32e2e7SValeria Barra         ierr = MPI_Allreduce(&my_rt, &rt_max, 1, MPI_DOUBLE, MPI_MAX, comm);
944cb32e2e7SValeria Barra         CHKERRQ(ierr);
945cb32e2e7SValeria Barra         ierr = PetscPrintf(comm,
9468e87e98bSJed Brown                            "    Pointwise Error (max)              : %e\n"
9478e87e98bSJed Brown                            "    CG Solve Time                      : %g (%g) sec\n",
9488e87e98bSJed Brown                            (double)maxerror, rt_max, rt_min); CHKERRQ(ierr);
949cb32e2e7SValeria Barra       }
950cb32e2e7SValeria Barra     }
9518e87e98bSJed Brown     if (benchmark_mode && (!test_mode)) {
952cb32e2e7SValeria Barra       ierr = PetscPrintf(comm,
9538e87e98bSJed Brown                          "    DoFs/Sec in CG                     : %g (%g) million\n",
9548e87e98bSJed Brown                          1e-6*gsize*its/rt_max,
9558e87e98bSJed Brown                          1e-6*gsize*its/rt_min); CHKERRQ(ierr);
956cb32e2e7SValeria Barra     }
957cb32e2e7SValeria Barra   }
958cb32e2e7SValeria Barra 
959cb32e2e7SValeria Barra   if (write_solution) {
960cb32e2e7SValeria Barra     PetscViewer vtkviewersoln;
961cb32e2e7SValeria Barra 
962cb32e2e7SValeria Barra     ierr = PetscViewerCreate(comm, &vtkviewersoln); CHKERRQ(ierr);
963cb32e2e7SValeria Barra     ierr = PetscViewerSetType(vtkviewersoln, PETSCVIEWERVTK); CHKERRQ(ierr);
964cb32e2e7SValeria Barra     ierr = PetscViewerFileSetName(vtkviewersoln, "solution.vtk"); CHKERRQ(ierr);
965cb32e2e7SValeria Barra     ierr = VecView(X, vtkviewersoln); CHKERRQ(ierr);
966cb32e2e7SValeria Barra     ierr = PetscViewerDestroy(&vtkviewersoln); CHKERRQ(ierr);
967cb32e2e7SValeria Barra   }
968cb32e2e7SValeria Barra 
969cb32e2e7SValeria Barra   ierr = VecDestroy(&rhs); CHKERRQ(ierr);
970cb32e2e7SValeria Barra   ierr = VecDestroy(&rhsloc); CHKERRQ(ierr);
971cb32e2e7SValeria Barra   ierr = VecDestroy(&X); CHKERRQ(ierr);
972cb32e2e7SValeria Barra   ierr = VecDestroy(&user->Xloc); CHKERRQ(ierr);
973cb32e2e7SValeria Barra   ierr = VecDestroy(&user->Yloc); CHKERRQ(ierr);
974cb32e2e7SValeria Barra   ierr = VecScatterDestroy(&ltog); CHKERRQ(ierr);
975cb32e2e7SValeria Barra   ierr = VecScatterDestroy(&ltog0); CHKERRQ(ierr);
976cb32e2e7SValeria Barra   ierr = VecScatterDestroy(&gtogD); CHKERRQ(ierr);
977cb32e2e7SValeria Barra   ierr = MatDestroy(&mat); CHKERRQ(ierr);
978cb32e2e7SValeria Barra   ierr = KSPDestroy(&ksp); CHKERRQ(ierr);
979cb32e2e7SValeria Barra 
980cb32e2e7SValeria Barra   CeedVectorDestroy(&user->xceed);
981cb32e2e7SValeria Barra   CeedVectorDestroy(&user->yceed);
982cb32e2e7SValeria Barra   CeedVectorDestroy(&user->qdata);
983cb32e2e7SValeria Barra   CeedVectorDestroy(&target);
984c70bd2a0Sjeremylt   CeedOperatorDestroy(&opsetupgeo);
985c70bd2a0Sjeremylt   CeedOperatorDestroy(&opsetuprhs);
986c70bd2a0Sjeremylt   CeedOperatorDestroy(&opapply);
987c70bd2a0Sjeremylt   CeedOperatorDestroy(&operror);
988cb32e2e7SValeria Barra   CeedElemRestrictionDestroy(&Erestrictu);
989cb32e2e7SValeria Barra   CeedElemRestrictionDestroy(&Erestrictx);
990cb32e2e7SValeria Barra   CeedElemRestrictionDestroy(&Erestrictui);
991cb32e2e7SValeria Barra   CeedElemRestrictionDestroy(&Erestrictqdi);
992c70bd2a0Sjeremylt   CeedQFunctionDestroy(&qfsetupgeo);
993c70bd2a0Sjeremylt   CeedQFunctionDestroy(&qfsetuprhs);
994c70bd2a0Sjeremylt   CeedQFunctionDestroy(&qfapply);
995c70bd2a0Sjeremylt   CeedQFunctionDestroy(&qferror);
996cb32e2e7SValeria Barra   CeedBasisDestroy(&basisu);
997cb32e2e7SValeria Barra   CeedBasisDestroy(&basisx);
998cb32e2e7SValeria Barra   CeedDestroy(&ceed);
999cb32e2e7SValeria Barra   ierr = PetscFree(user); CHKERRQ(ierr);
1000cb32e2e7SValeria Barra   return PetscFinalize();
1001cb32e2e7SValeria Barra }
1002