1 /*
2 Provides an interface to the LLNL package hypre
3 */
4
5 #include <petscpkg_version.h>
6 #include <petsc/private/pcimpl.h> /*I "petscpc.h" I*/
7 /* this include is needed ONLY to allow access to the private data inside the Mat object specific to hypre */
8 #include <petsc/private/matimpl.h>
9 #include <petsc/private/vecimpl.h>
10 #include <../src/vec/vec/impls/hypre/vhyp.h>
11 #include <../src/mat/impls/hypre/mhypre.h>
12 #include <../src/dm/impls/da/hypre/mhyp.h>
13 #include <_hypre_parcsr_ls.h>
14 #include <petscmathypre.h>
15
16 #if defined(PETSC_HAVE_HYPRE_DEVICE)
17 #include <petsc/private/deviceimpl.h>
18 #endif
19
20 static PetscBool cite = PETSC_FALSE;
21 static const char hypreCitation[] = "@manual{hypre-web-page,\n title = {{\\sl hypre}: High Performance Preconditioners},\n organization = {Lawrence Livermore National Laboratory},\n note = "
22 "{\\url{https://www.llnl.gov/casc/hypre}}\n}\n";
23
24 /*
25 Private context (data structure) for the preconditioner.
26 */
27 typedef struct {
28 HYPRE_Solver hsolver;
29 Mat hpmat; /* MatHYPRE */
30
31 HYPRE_Int (*destroy)(HYPRE_Solver);
32 HYPRE_Int (*solve)(HYPRE_Solver, HYPRE_ParCSRMatrix, HYPRE_ParVector, HYPRE_ParVector);
33 HYPRE_Int (*setup)(HYPRE_Solver, HYPRE_ParCSRMatrix, HYPRE_ParVector, HYPRE_ParVector);
34
35 MPI_Comm comm_hypre;
36 char *hypre_type;
37
38 /* options for Pilut and BoomerAMG*/
39 PetscInt maxiter;
40 PetscReal tol;
41
42 /* options for Pilut */
43 PetscInt factorrowsize;
44
45 /* options for ParaSails */
46 PetscInt nlevels;
47 PetscReal threshold;
48 PetscReal filter;
49 PetscReal loadbal;
50 PetscInt logging;
51 PetscInt ruse;
52 PetscInt symt;
53
54 /* options for BoomerAMG */
55 PetscBool printstatistics;
56
57 /* options for BoomerAMG */
58 PetscInt cycletype;
59 PetscInt maxlevels;
60 PetscReal strongthreshold;
61 PetscReal maxrowsum;
62 PetscInt gridsweeps[3];
63 PetscObjectParameterDeclare(PetscInt, coarsentype);
64 PetscInt measuretype;
65 PetscInt smoothtype;
66 PetscInt smoothsweeps;
67 PetscInt smoothnumlevels;
68 PetscInt eu_level; /* Number of levels for ILU(k) in Euclid */
69 PetscReal eu_droptolerance; /* Drop tolerance for ILU(k) in Euclid */
70 PetscInt eu_bj; /* Defines use of Block Jacobi ILU in Euclid */
71 PetscObjectParameterDeclare(PetscInt, relaxtype[3]);
72 PetscReal relaxweight;
73 PetscReal outerrelaxweight;
74 PetscObjectParameterDeclare(PetscInt, relaxorder);
75 PetscReal truncfactor;
76 PetscBool applyrichardson;
77 PetscInt pmax;
78 PetscObjectParameterDeclare(PetscInt, interptype);
79 PetscInt maxc;
80 PetscInt minc;
81 #if PETSC_PKG_HYPRE_VERSION_GE(2, 23, 0)
82 PetscObjectParameterDeclarePtr(const char, spgemm_type); // this is a global hypre parameter but is closely associated with BoomerAMG
83 #endif
84 /* GPU */
85 PetscObjectParameterDeclare(PetscBool3, keeptranspose);
86 PetscInt rap2;
87 PetscObjectParameterDeclare(PetscInt, mod_rap2);
88
89 /* AIR */
90 PetscInt Rtype;
91 PetscReal Rstrongthreshold;
92 PetscReal Rfilterthreshold;
93 PetscInt Adroptype;
94 PetscReal Adroptol;
95
96 PetscInt agg_nl;
97 PetscObjectParameterDeclare(PetscInt, agg_interptype);
98 PetscInt agg_num_paths;
99 PetscBool nodal_relax;
100 PetscInt nodal_relax_levels;
101
102 PetscInt nodal_coarsening;
103 PetscInt nodal_coarsening_diag;
104 PetscInt vec_interp_variant;
105 PetscInt vec_interp_qmax;
106 PetscBool vec_interp_smooth;
107 PetscInt interp_refine;
108
109 /* NearNullSpace support */
110 VecHYPRE_IJVector *hmnull;
111 HYPRE_ParVector *phmnull;
112 PetscInt n_hmnull;
113 Vec hmnull_constant;
114
115 /* options for AS (Auxiliary Space preconditioners) */
116 PetscInt as_print;
117 PetscInt as_max_iter;
118 PetscReal as_tol;
119 PetscInt as_relax_type;
120 PetscInt as_relax_times;
121 PetscReal as_relax_weight;
122 PetscReal as_omega;
123 PetscInt as_amg_alpha_opts[5]; /* AMG coarsen type, agg_levels, relax_type, interp_type, Pmax for vector Poisson (AMS) or Curl problem (ADS) */
124 PetscReal as_amg_alpha_theta; /* AMG strength for vector Poisson (AMS) or Curl problem (ADS) */
125 PetscInt as_amg_beta_opts[5]; /* AMG coarsen type, agg_levels, relax_type, interp_type, Pmax for scalar Poisson (AMS) or vector Poisson (ADS) */
126 PetscReal as_amg_beta_theta; /* AMG strength for scalar Poisson (AMS) or vector Poisson (ADS) */
127 PetscInt ams_cycle_type;
128 PetscInt ads_cycle_type;
129
130 /* additional data */
131 Mat G; /* MatHYPRE */
132 Mat C; /* MatHYPRE */
133 Mat alpha_Poisson; /* MatHYPRE */
134 Mat beta_Poisson; /* MatHYPRE */
135
136 /* extra information for AMS */
137 PetscInt dim; /* geometrical dimension */
138 VecHYPRE_IJVector coords[3];
139 VecHYPRE_IJVector constants[3];
140 VecHYPRE_IJVector interior;
141 Mat RT_PiFull, RT_Pi[3];
142 Mat ND_PiFull, ND_Pi[3];
143 PetscBool ams_beta_is_zero;
144 PetscBool ams_beta_is_zero_part;
145 PetscInt ams_proj_freq;
146 } PC_HYPRE;
147
148 /*
149 Matrices with AIJ format are created IN PLACE with using (I,J,data) from BoomerAMG. Since the data format in hypre_ParCSRMatrix
150 is different from that used in PETSc, the original hypre_ParCSRMatrix can not be used any more after call this routine.
151 It is used in PCHMG. Other users should avoid using this function.
152 */
PCGetCoarseOperators_BoomerAMG(PC pc,PetscInt * nlevels,Mat * operators[])153 static PetscErrorCode PCGetCoarseOperators_BoomerAMG(PC pc, PetscInt *nlevels, Mat *operators[])
154 {
155 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
156 PetscBool same;
157 PetscInt num_levels, l;
158 Mat *mattmp;
159 hypre_ParCSRMatrix **A_array;
160
161 PetscFunctionBegin;
162 PetscCall(PetscStrcmp(jac->hypre_type, "boomeramg", &same));
163 PetscCheck(same, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_NOTSAMETYPE, "Hypre type is not BoomerAMG");
164 num_levels = hypre_ParAMGDataNumLevels((hypre_ParAMGData *)jac->hsolver);
165 PetscCall(PetscMalloc1(num_levels, &mattmp));
166 A_array = hypre_ParAMGDataAArray((hypre_ParAMGData *)jac->hsolver);
167 for (l = 1; l < num_levels; l++) {
168 PetscCall(MatCreateFromParCSR(A_array[l], MATAIJ, PETSC_OWN_POINTER, &mattmp[num_levels - 1 - l]));
169 /* We want to own the data, and HYPRE can not touch this matrix any more */
170 A_array[l] = NULL;
171 }
172 *nlevels = num_levels;
173 *operators = mattmp;
174 PetscFunctionReturn(PETSC_SUCCESS);
175 }
176
177 /*
178 Matrices with AIJ format are created IN PLACE with using (I,J,data) from BoomerAMG. Since the data format in hypre_ParCSRMatrix
179 is different from that used in PETSc, the original hypre_ParCSRMatrix can not be used any more after call this routine.
180 It is used in PCHMG. Other users should avoid using this function.
181 */
PCGetInterpolations_BoomerAMG(PC pc,PetscInt * nlevels,Mat * interpolations[])182 static PetscErrorCode PCGetInterpolations_BoomerAMG(PC pc, PetscInt *nlevels, Mat *interpolations[])
183 {
184 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
185 PetscBool same;
186 PetscInt num_levels, l;
187 Mat *mattmp;
188 hypre_ParCSRMatrix **P_array;
189
190 PetscFunctionBegin;
191 PetscCall(PetscStrcmp(jac->hypre_type, "boomeramg", &same));
192 PetscCheck(same, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_NOTSAMETYPE, "Hypre type is not BoomerAMG");
193 num_levels = hypre_ParAMGDataNumLevels((hypre_ParAMGData *)jac->hsolver);
194 PetscCall(PetscMalloc1(num_levels, &mattmp));
195 P_array = hypre_ParAMGDataPArray((hypre_ParAMGData *)jac->hsolver);
196 for (l = 1; l < num_levels; l++) {
197 PetscCall(MatCreateFromParCSR(P_array[num_levels - 1 - l], MATAIJ, PETSC_OWN_POINTER, &mattmp[l - 1]));
198 /* We want to own the data, and HYPRE can not touch this matrix any more */
199 P_array[num_levels - 1 - l] = NULL;
200 }
201 *nlevels = num_levels;
202 *interpolations = mattmp;
203 PetscFunctionReturn(PETSC_SUCCESS);
204 }
205
206 /*
207 Boolean Vecs are created IN PLACE with using data from BoomerAMG.
208 */
PCHYPREGetCFMarkers_BoomerAMG(PC pc,PetscInt * n_per_level[],PetscBT * CFMarkers[])209 static PetscErrorCode PCHYPREGetCFMarkers_BoomerAMG(PC pc, PetscInt *n_per_level[], PetscBT *CFMarkers[])
210 {
211 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
212 PetscBool same;
213 PetscInt num_levels, fine_nodes = 0, coarse_nodes;
214 PetscInt *n_per_temp;
215 PetscBT *markertmp;
216 hypre_IntArray **CF_marker_array;
217
218 PetscFunctionBegin;
219 PetscCall(PetscStrcmp(jac->hypre_type, "boomeramg", &same));
220 PetscCheck(same, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_NOTSAMETYPE, "Hypre type is not BoomerAMG");
221 num_levels = hypre_ParAMGDataNumLevels((hypre_ParAMGData *)jac->hsolver);
222 PetscCall(PetscMalloc1(num_levels, &n_per_temp));
223 PetscCall(PetscMalloc1(num_levels - 1, &markertmp));
224 CF_marker_array = hypre_ParAMGDataCFMarkerArray((hypre_ParAMGData *)jac->hsolver);
225 for (PetscInt l = 0, CFMaxIndex = num_levels - 2; CFMaxIndex >= 0; l++, CFMaxIndex--) {
226 fine_nodes = hypre_IntArraySize(CF_marker_array[CFMaxIndex]);
227 coarse_nodes = 0;
228 PetscCall(PetscBTCreate(fine_nodes, &markertmp[l]));
229 for (PetscInt k = 0; k < fine_nodes; k++) {
230 if (hypre_IntArrayDataI(CF_marker_array[CFMaxIndex], k) > 0) {
231 PetscCall(PetscBTSet(markertmp[l], k));
232 coarse_nodes++;
233 }
234 }
235 n_per_temp[l] = coarse_nodes;
236 }
237 n_per_temp[num_levels - 1] = fine_nodes;
238 *n_per_level = n_per_temp;
239 *CFMarkers = markertmp;
240 PetscFunctionReturn(PETSC_SUCCESS);
241 }
242
243 /* Resets (frees) Hypre's representation of the near null space */
PCHYPREResetNearNullSpace_Private(PC pc)244 static PetscErrorCode PCHYPREResetNearNullSpace_Private(PC pc)
245 {
246 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
247 PetscInt i;
248
249 PetscFunctionBegin;
250 for (i = 0; i < jac->n_hmnull; i++) PetscCall(VecHYPRE_IJVectorDestroy(&jac->hmnull[i]));
251 PetscCall(PetscFree(jac->hmnull));
252 PetscCall(PetscFree(jac->phmnull));
253 PetscCall(VecDestroy(&jac->hmnull_constant));
254 jac->n_hmnull = 0;
255 PetscFunctionReturn(PETSC_SUCCESS);
256 }
257
258 static const char *HYPRESpgemmTypes[] = {"cusparse", "hypre"};
PCMGGalerkinSetMatProductAlgorithm_HYPRE_BoomerAMG(PC pc,const char name[])259 static PetscErrorCode PCMGGalerkinSetMatProductAlgorithm_HYPRE_BoomerAMG(PC pc, const char name[])
260 {
261 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
262
263 #if PETSC_PKG_HYPRE_VERSION_GE(2, 23, 0)
264 PetscFunctionBegin;
265 jac->spgemm_type = name;
266 PetscFunctionReturn(PETSC_SUCCESS);
267 #endif
268 }
269
PCSetUp_HYPRE(PC pc)270 static PetscErrorCode PCSetUp_HYPRE(PC pc)
271 {
272 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
273 Mat_HYPRE *hjac;
274 HYPRE_ParCSRMatrix hmat;
275 HYPRE_ParVector bv, xv;
276 PetscBool ishypre;
277
278 PetscFunctionBegin;
279 /* default type is boomerAMG */
280 if (!jac->hypre_type) PetscCall(PCHYPRESetType(pc, "boomeramg"));
281
282 /* get hypre matrix */
283 if (pc->flag == DIFFERENT_NONZERO_PATTERN) PetscCall(MatDestroy(&jac->hpmat));
284 PetscCall(PetscObjectTypeCompare((PetscObject)pc->pmat, MATHYPRE, &ishypre));
285 if (!ishypre) {
286 #if defined(PETSC_HAVE_HYPRE_DEVICE) && PETSC_PKG_HYPRE_VERSION_LE(2, 30, 0)
287 /* Temporary fix since we do not support MAT_REUSE_MATRIX with HYPRE device */
288 PetscBool iscuda, iship, iskokkos;
289
290 PetscCall(PetscObjectTypeCompareAny((PetscObject)pc->pmat, &iscuda, MATSEQAIJCUSPARSE, MATMPIAIJCUSPARSE, ""));
291 PetscCall(PetscObjectTypeCompareAny((PetscObject)pc->pmat, &iship, MATSEQAIJHIPSPARSE, MATMPIAIJHIPSPARSE, ""));
292 PetscCall(PetscObjectTypeCompareAny((PetscObject)pc->pmat, &iskokkos, MATSEQAIJKOKKOS, MATMPIAIJKOKKOS, ""));
293 if (iscuda || iship || iskokkos) PetscCall(MatDestroy(&jac->hpmat));
294 #endif
295 PetscCall(MatConvert(pc->pmat, MATHYPRE, jac->hpmat ? MAT_REUSE_MATRIX : MAT_INITIAL_MATRIX, &jac->hpmat));
296 } else {
297 PetscCall(PetscObjectReference((PetscObject)pc->pmat));
298 PetscCall(MatDestroy(&jac->hpmat));
299 jac->hpmat = pc->pmat;
300 }
301
302 /* allow debug */
303 PetscCall(MatViewFromOptions(jac->hpmat, NULL, "-pc_hypre_mat_view"));
304 hjac = (Mat_HYPRE *)jac->hpmat->data;
305
306 /* special case for BoomerAMG */
307 if (jac->setup == HYPRE_BoomerAMGSetup) {
308 MatNullSpace mnull;
309 PetscBool has_const;
310 PetscInt bs, nvec, i;
311 PetscMemType memtype;
312 const Vec *vecs;
313
314 PetscCall(MatGetCurrentMemType(jac->hpmat, &memtype));
315 if (PetscMemTypeDevice(memtype)) {
316 /* GPU defaults
317 From https://hypre.readthedocs.io/en/latest/solvers-boomeramg.html#gpu-supported-options
318 and /src/parcsr_ls/par_amg.c
319 First handle options which users have interfaces for changing */
320 PetscObjectParameterSetDefault(jac, coarsentype, 8);
321 PetscObjectParameterSetDefault(jac, relaxorder, 0);
322 PetscObjectParameterSetDefault(jac, interptype, 6);
323 PetscObjectParameterSetDefault(jac, relaxtype[0], 18);
324 PetscObjectParameterSetDefault(jac, relaxtype[1], 18);
325 #if PETSC_PKG_HYPRE_VERSION_GE(2, 23, 0)
326 PetscObjectParameterSetDefault(jac, spgemm_type, HYPRESpgemmTypes[0]);
327 #endif
328 #if PETSC_PKG_HYPRE_VERSION_GE(2, 18, 0)
329 PetscObjectParameterSetDefault(jac, keeptranspose, PETSC_BOOL3_TRUE);
330 PetscObjectParameterSetDefault(jac, mod_rap2, 1);
331 #endif
332 PetscObjectParameterSetDefault(jac, agg_interptype, 7);
333 } else {
334 PetscObjectParameterSetDefault(jac, coarsentype, 6);
335 PetscObjectParameterSetDefault(jac, relaxorder, 1);
336 PetscObjectParameterSetDefault(jac, interptype, 0);
337 PetscObjectParameterSetDefault(jac, relaxtype[0], 6);
338 PetscObjectParameterSetDefault(jac, relaxtype[1], 6); /* Defaults to SYMMETRIC since in PETSc we are using a PC - most likely with CG */
339 #if PETSC_PKG_HYPRE_VERSION_GE(2, 23, 0)
340 PetscObjectParameterSetDefault(jac, spgemm_type, "hypre");
341 #endif
342 #if PETSC_PKG_HYPRE_VERSION_GE(2, 18, 0)
343 PetscObjectParameterSetDefault(jac, keeptranspose, PETSC_BOOL3_FALSE);
344 PetscObjectParameterSetDefault(jac, mod_rap2, 0);
345 #endif
346 PetscObjectParameterSetDefault(jac, agg_interptype, 4);
347 }
348 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleType(jac->hsolver, (HYPRE_Int)jac->cycletype));
349 PetscCallHYPRE(HYPRE_BoomerAMGSetMaxLevels(jac->hsolver, (HYPRE_Int)jac->maxlevels));
350 PetscCallHYPRE(HYPRE_BoomerAMGSetMaxIter(jac->hsolver, (HYPRE_Int)jac->maxiter));
351 PetscCallHYPRE(HYPRE_BoomerAMGSetTol(jac->hsolver, jac->tol));
352 PetscCallHYPRE(HYPRE_BoomerAMGSetTruncFactor(jac->hsolver, jac->truncfactor));
353 PetscCallHYPRE(HYPRE_BoomerAMGSetStrongThreshold(jac->hsolver, jac->strongthreshold));
354 PetscCallHYPRE(HYPRE_BoomerAMGSetMaxRowSum(jac->hsolver, jac->maxrowsum));
355 PetscCallHYPRE(HYPRE_BoomerAMGSetMeasureType(jac->hsolver, (HYPRE_Int)jac->measuretype));
356 PetscCallHYPRE(HYPRE_BoomerAMGSetAggNumLevels(jac->hsolver, (HYPRE_Int)jac->agg_nl));
357 PetscCallHYPRE(HYPRE_BoomerAMGSetPMaxElmts(jac->hsolver, (HYPRE_Int)jac->pmax));
358 PetscCallHYPRE(HYPRE_BoomerAMGSetNumPaths(jac->hsolver, (HYPRE_Int)jac->agg_num_paths));
359 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleNumSweeps(jac->hsolver, (HYPRE_Int)jac->gridsweeps[0], 1));
360 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleNumSweeps(jac->hsolver, (HYPRE_Int)jac->gridsweeps[1], 2));
361 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleNumSweeps(jac->hsolver, (HYPRE_Int)jac->gridsweeps[2], 3));
362 PetscCallHYPRE(HYPRE_BoomerAMGSetMaxCoarseSize(jac->hsolver, (HYPRE_Int)jac->maxc));
363 PetscCallHYPRE(HYPRE_BoomerAMGSetMinCoarseSize(jac->hsolver, (HYPRE_Int)jac->minc));
364 PetscCallHYPRE(HYPRE_BoomerAMGSetCoarsenType(jac->hsolver, (HYPRE_Int)jac->coarsentype));
365 PetscCallHYPRE(HYPRE_BoomerAMGSetRelaxOrder(jac->hsolver, (HYPRE_Int)jac->relaxorder));
366 PetscCallHYPRE(HYPRE_BoomerAMGSetInterpType(jac->hsolver, (HYPRE_Int)jac->interptype));
367 PetscCallHYPRE(HYPRE_BoomerAMGSetRelaxType(jac->hsolver, (HYPRE_Int)jac->relaxtype[0]));
368 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleRelaxType(jac->hsolver, (HYPRE_Int)jac->relaxtype[0], 1));
369 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleRelaxType(jac->hsolver, (HYPRE_Int)jac->relaxtype[1], 2));
370 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleRelaxType(jac->hsolver, (HYPRE_Int)jac->relaxtype[2], 3));
371 /* GPU */
372 #if PETSC_PKG_HYPRE_VERSION_GE(2, 23, 0)
373 {
374 PetscBool flg_cusparse, flg_hypre;
375
376 PetscCall(PetscStrcmp("cusparse", jac->spgemm_type, &flg_cusparse));
377 PetscCall(PetscStrcmp("hypre", jac->spgemm_type, &flg_hypre));
378 if (flg_cusparse) PetscCallHYPRE(HYPRE_SetSpGemmUseCusparse(1));
379 else if (flg_hypre) PetscCallHYPRE(HYPRE_SetSpGemmUseCusparse(0));
380 else SETERRQ(PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unknown HYPRE SpGEMM type %s; Choices are cusparse, hypre", jac->spgemm_type);
381 }
382 #endif
383 #if PETSC_PKG_HYPRE_VERSION_GE(2, 18, 0)
384 PetscCallHYPRE(HYPRE_BoomerAMGSetKeepTranspose(jac->hsolver, jac->keeptranspose == PETSC_BOOL3_TRUE ? 1 : 0));
385 PetscCallHYPRE(HYPRE_BoomerAMGSetRAP2(jac->hsolver, (HYPRE_Int)jac->rap2));
386 PetscCallHYPRE(HYPRE_BoomerAMGSetModuleRAP2(jac->hsolver, (HYPRE_Int)jac->mod_rap2));
387 #endif
388 PetscCallHYPRE(HYPRE_BoomerAMGSetAggInterpType(jac->hsolver, (HYPRE_Int)jac->agg_interptype));
389
390 /* AIR */
391 #if PETSC_PKG_HYPRE_VERSION_GE(2, 18, 0)
392 PetscCallHYPRE(HYPRE_BoomerAMGSetRestriction(jac->hsolver, (HYPRE_Int)jac->Rtype));
393 PetscCallHYPRE(HYPRE_BoomerAMGSetStrongThresholdR(jac->hsolver, jac->Rstrongthreshold));
394 PetscCallHYPRE(HYPRE_BoomerAMGSetFilterThresholdR(jac->hsolver, jac->Rfilterthreshold));
395 PetscCallHYPRE(HYPRE_BoomerAMGSetADropTol(jac->hsolver, jac->Adroptol));
396 PetscCallHYPRE(HYPRE_BoomerAMGSetADropType(jac->hsolver, (HYPRE_Int)jac->Adroptype));
397 #endif
398
399 PetscCall(MatGetBlockSize(pc->pmat, &bs));
400 if (bs > 1) PetscCallHYPRE(HYPRE_BoomerAMGSetNumFunctions(jac->hsolver, (HYPRE_Int)bs));
401 PetscCall(MatGetNearNullSpace(pc->mat, &mnull));
402 if (mnull) {
403 PetscCall(PCHYPREResetNearNullSpace_Private(pc));
404 PetscCall(MatNullSpaceGetVecs(mnull, &has_const, &nvec, &vecs));
405 PetscCall(PetscMalloc1(nvec + 1, &jac->hmnull));
406 PetscCall(PetscMalloc1(nvec + 1, &jac->phmnull));
407 for (i = 0; i < nvec; i++) {
408 PetscCall(VecHYPRE_IJVectorCreate(vecs[i]->map, &jac->hmnull[i]));
409 PetscCall(VecHYPRE_IJVectorCopy(vecs[i], jac->hmnull[i]));
410 PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->hmnull[i]->ij, (void **)&jac->phmnull[i]));
411 }
412 if (has_const) {
413 PetscCall(MatCreateVecs(pc->pmat, &jac->hmnull_constant, NULL));
414 PetscCall(VecSet(jac->hmnull_constant, 1));
415 PetscCall(VecNormalize(jac->hmnull_constant, NULL));
416 PetscCall(VecHYPRE_IJVectorCreate(jac->hmnull_constant->map, &jac->hmnull[nvec]));
417 PetscCall(VecHYPRE_IJVectorCopy(jac->hmnull_constant, jac->hmnull[nvec]));
418 PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->hmnull[nvec]->ij, (void **)&jac->phmnull[nvec]));
419 nvec++;
420 }
421 PetscCallHYPRE(HYPRE_BoomerAMGSetInterpVectors(jac->hsolver, (HYPRE_Int)nvec, jac->phmnull));
422 jac->n_hmnull = nvec;
423 }
424 }
425
426 /* special case for AMS */
427 if (jac->setup == HYPRE_AMSSetup) {
428 Mat_HYPRE *hm;
429 HYPRE_ParCSRMatrix parcsr;
430 PetscCheck(jac->coords[0] || jac->constants[0] || jac->ND_PiFull || (jac->ND_Pi[0] && jac->ND_Pi[1]), PetscObjectComm((PetscObject)pc), PETSC_ERR_USER, "HYPRE AMS preconditioner needs either the coordinate vectors via PCSetCoordinates() or the edge constant vectors via PCHYPRESetEdgeConstantVectors() or the interpolation matrix via PCHYPRESetInterpolations()");
431 if (jac->dim) PetscCallHYPRE(HYPRE_AMSSetDimension(jac->hsolver, (HYPRE_Int)jac->dim));
432 if (jac->constants[0]) {
433 HYPRE_ParVector ozz, zoz, zzo = NULL;
434 PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->constants[0]->ij, (void **)(&ozz)));
435 PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->constants[1]->ij, (void **)(&zoz)));
436 if (jac->constants[2]) PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->constants[2]->ij, (void **)(&zzo)));
437 PetscCallHYPRE(HYPRE_AMSSetEdgeConstantVectors(jac->hsolver, ozz, zoz, zzo));
438 }
439 if (jac->coords[0]) {
440 HYPRE_ParVector coords[3];
441 coords[0] = NULL;
442 coords[1] = NULL;
443 coords[2] = NULL;
444 if (jac->coords[0]) PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->coords[0]->ij, (void **)(&coords[0])));
445 if (jac->coords[1]) PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->coords[1]->ij, (void **)(&coords[1])));
446 if (jac->coords[2]) PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->coords[2]->ij, (void **)(&coords[2])));
447 PetscCallHYPRE(HYPRE_AMSSetCoordinateVectors(jac->hsolver, coords[0], coords[1], coords[2]));
448 }
449 PetscCheck(jac->G, PetscObjectComm((PetscObject)pc), PETSC_ERR_USER, "HYPRE AMS preconditioner needs the discrete gradient operator via PCHYPRESetDiscreteGradient");
450 hm = (Mat_HYPRE *)jac->G->data;
451 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&parcsr)));
452 PetscCallHYPRE(HYPRE_AMSSetDiscreteGradient(jac->hsolver, parcsr));
453 if (jac->alpha_Poisson) {
454 hm = (Mat_HYPRE *)jac->alpha_Poisson->data;
455 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&parcsr)));
456 PetscCallHYPRE(HYPRE_AMSSetAlphaPoissonMatrix(jac->hsolver, parcsr));
457 }
458 if (jac->ams_beta_is_zero) {
459 PetscCallHYPRE(HYPRE_AMSSetBetaPoissonMatrix(jac->hsolver, NULL));
460 } else if (jac->beta_Poisson) {
461 hm = (Mat_HYPRE *)jac->beta_Poisson->data;
462 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&parcsr)));
463 PetscCallHYPRE(HYPRE_AMSSetBetaPoissonMatrix(jac->hsolver, parcsr));
464 } else if (jac->ams_beta_is_zero_part) {
465 if (jac->interior) {
466 HYPRE_ParVector interior = NULL;
467 PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->interior->ij, (void **)(&interior)));
468 PetscCallHYPRE(HYPRE_AMSSetInteriorNodes(jac->hsolver, interior));
469 } else {
470 jac->ams_beta_is_zero_part = PETSC_FALSE;
471 }
472 }
473 if (jac->ND_PiFull || (jac->ND_Pi[0] && jac->ND_Pi[1])) {
474 PetscInt i;
475 HYPRE_ParCSRMatrix nd_parcsrfull, nd_parcsr[3];
476 if (jac->ND_PiFull) {
477 hm = (Mat_HYPRE *)jac->ND_PiFull->data;
478 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&nd_parcsrfull)));
479 } else {
480 nd_parcsrfull = NULL;
481 }
482 for (i = 0; i < 3; ++i) {
483 if (jac->ND_Pi[i]) {
484 hm = (Mat_HYPRE *)jac->ND_Pi[i]->data;
485 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&nd_parcsr[i])));
486 } else {
487 nd_parcsr[i] = NULL;
488 }
489 }
490 PetscCallHYPRE(HYPRE_AMSSetInterpolations(jac->hsolver, nd_parcsrfull, nd_parcsr[0], nd_parcsr[1], nd_parcsr[2]));
491 }
492 }
493 /* special case for ADS */
494 if (jac->setup == HYPRE_ADSSetup) {
495 Mat_HYPRE *hm;
496 HYPRE_ParCSRMatrix parcsr;
497 if (!jac->coords[0] && !((jac->RT_PiFull || (jac->RT_Pi[0] && jac->RT_Pi[1])) && (jac->ND_PiFull || (jac->ND_Pi[0] && jac->ND_Pi[1])))) {
498 SETERRQ(PetscObjectComm((PetscObject)pc), PETSC_ERR_USER, "HYPRE ADS preconditioner needs either the coordinate vectors via PCSetCoordinates() or the interpolation matrices via PCHYPRESetInterpolations");
499 } else PetscCheck(jac->coords[1] && jac->coords[2], PetscObjectComm((PetscObject)pc), PETSC_ERR_USER, "HYPRE ADS preconditioner has been designed for three dimensional problems! For two dimensional problems, use HYPRE AMS instead");
500 PetscCheck(jac->G, PetscObjectComm((PetscObject)pc), PETSC_ERR_USER, "HYPRE ADS preconditioner needs the discrete gradient operator via PCHYPRESetDiscreteGradient");
501 PetscCheck(jac->C, PetscObjectComm((PetscObject)pc), PETSC_ERR_USER, "HYPRE ADS preconditioner needs the discrete curl operator via PCHYPRESetDiscreteGradient");
502 if (jac->coords[0]) {
503 HYPRE_ParVector coords[3];
504 coords[0] = NULL;
505 coords[1] = NULL;
506 coords[2] = NULL;
507 if (jac->coords[0]) PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->coords[0]->ij, (void **)(&coords[0])));
508 if (jac->coords[1]) PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->coords[1]->ij, (void **)(&coords[1])));
509 if (jac->coords[2]) PetscCallHYPRE(HYPRE_IJVectorGetObject(jac->coords[2]->ij, (void **)(&coords[2])));
510 PetscCallHYPRE(HYPRE_ADSSetCoordinateVectors(jac->hsolver, coords[0], coords[1], coords[2]));
511 }
512 hm = (Mat_HYPRE *)jac->G->data;
513 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&parcsr)));
514 PetscCallHYPRE(HYPRE_ADSSetDiscreteGradient(jac->hsolver, parcsr));
515 hm = (Mat_HYPRE *)jac->C->data;
516 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&parcsr)));
517 PetscCallHYPRE(HYPRE_ADSSetDiscreteCurl(jac->hsolver, parcsr));
518 if ((jac->RT_PiFull || (jac->RT_Pi[0] && jac->RT_Pi[1])) && (jac->ND_PiFull || (jac->ND_Pi[0] && jac->ND_Pi[1]))) {
519 PetscInt i;
520 HYPRE_ParCSRMatrix rt_parcsrfull, rt_parcsr[3];
521 HYPRE_ParCSRMatrix nd_parcsrfull, nd_parcsr[3];
522 if (jac->RT_PiFull) {
523 hm = (Mat_HYPRE *)jac->RT_PiFull->data;
524 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&rt_parcsrfull)));
525 } else {
526 rt_parcsrfull = NULL;
527 }
528 for (i = 0; i < 3; ++i) {
529 if (jac->RT_Pi[i]) {
530 hm = (Mat_HYPRE *)jac->RT_Pi[i]->data;
531 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&rt_parcsr[i])));
532 } else {
533 rt_parcsr[i] = NULL;
534 }
535 }
536 if (jac->ND_PiFull) {
537 hm = (Mat_HYPRE *)jac->ND_PiFull->data;
538 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&nd_parcsrfull)));
539 } else {
540 nd_parcsrfull = NULL;
541 }
542 for (i = 0; i < 3; ++i) {
543 if (jac->ND_Pi[i]) {
544 hm = (Mat_HYPRE *)jac->ND_Pi[i]->data;
545 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hm->ij, (void **)(&nd_parcsr[i])));
546 } else {
547 nd_parcsr[i] = NULL;
548 }
549 }
550 PetscCallHYPRE(HYPRE_ADSSetInterpolations(jac->hsolver, rt_parcsrfull, rt_parcsr[0], rt_parcsr[1], rt_parcsr[2], nd_parcsrfull, nd_parcsr[0], nd_parcsr[1], nd_parcsr[2]));
551 }
552 }
553 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hjac->ij, (void **)&hmat));
554 PetscCallHYPRE(HYPRE_IJVectorGetObject(hjac->b->ij, (void **)&bv));
555 PetscCallHYPRE(HYPRE_IJVectorGetObject(hjac->x->ij, (void **)&xv));
556 PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF));
557 PetscCallHYPRE((*jac->setup)(jac->hsolver, hmat, bv, xv));
558 PetscCall(PetscFPTrapPop());
559 PetscFunctionReturn(PETSC_SUCCESS);
560 }
561
PCApply_HYPRE(PC pc,Vec b,Vec x)562 static PetscErrorCode PCApply_HYPRE(PC pc, Vec b, Vec x)
563 {
564 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
565 Mat_HYPRE *hjac = (Mat_HYPRE *)jac->hpmat->data;
566 HYPRE_ParCSRMatrix hmat;
567 HYPRE_ParVector jbv, jxv;
568
569 PetscFunctionBegin;
570 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
571 if (!jac->applyrichardson) PetscCall(VecSet(x, 0.0));
572 PetscCall(VecHYPRE_IJVectorPushVecRead(hjac->b, b));
573 if (jac->applyrichardson) PetscCall(VecHYPRE_IJVectorPushVec(hjac->x, x));
574 else PetscCall(VecHYPRE_IJVectorPushVecWrite(hjac->x, x));
575 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hjac->ij, (void **)&hmat));
576 PetscCallHYPRE(HYPRE_IJVectorGetObject(hjac->b->ij, (void **)&jbv));
577 PetscCallHYPRE(HYPRE_IJVectorGetObject(hjac->x->ij, (void **)&jxv));
578 PetscStackCallExternalVoid(
579 "Hypre solve", do {
580 HYPRE_Int hierr = (*jac->solve)(jac->hsolver, hmat, jbv, jxv);
581 if (hierr) {
582 PetscCheck(hierr == HYPRE_ERROR_CONV, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error in HYPRE solver, error code %d", (int)hierr);
583 HYPRE_ClearAllErrors();
584 }
585 } while (0));
586
587 if (jac->setup == HYPRE_AMSSetup && jac->ams_beta_is_zero_part) PetscCallHYPRE(HYPRE_AMSProjectOutGradients(jac->hsolver, jxv));
588 PetscCall(VecHYPRE_IJVectorPopVec(hjac->x));
589 PetscCall(VecHYPRE_IJVectorPopVec(hjac->b));
590 PetscFunctionReturn(PETSC_SUCCESS);
591 }
592
PCMatApply_HYPRE_BoomerAMG(PC pc,Mat B,Mat X)593 static PetscErrorCode PCMatApply_HYPRE_BoomerAMG(PC pc, Mat B, Mat X)
594 {
595 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
596 Mat_HYPRE *hjac = (Mat_HYPRE *)jac->hpmat->data;
597 hypre_ParCSRMatrix *par_matrix;
598 HYPRE_ParVector hb, hx;
599 const PetscScalar *b;
600 PetscScalar *x;
601 PetscInt m, N, lda;
602 hypre_Vector *x_local;
603 PetscMemType type;
604
605 PetscFunctionBegin;
606 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
607 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hjac->ij, (void **)&par_matrix));
608 PetscCall(MatGetLocalSize(B, &m, NULL));
609 PetscCall(MatGetSize(B, NULL, &N));
610 PetscCallHYPRE(HYPRE_ParMultiVectorCreate(hypre_ParCSRMatrixComm(par_matrix), hypre_ParCSRMatrixGlobalNumRows(par_matrix), hypre_ParCSRMatrixRowStarts(par_matrix), (HYPRE_Int)N, &hb));
611 PetscCallHYPRE(HYPRE_ParMultiVectorCreate(hypre_ParCSRMatrixComm(par_matrix), hypre_ParCSRMatrixGlobalNumRows(par_matrix), hypre_ParCSRMatrixRowStarts(par_matrix), (HYPRE_Int)N, &hx));
612 PetscCall(MatZeroEntries(X));
613 PetscCall(MatDenseGetArrayReadAndMemType(B, &b, &type));
614 PetscCall(MatDenseGetLDA(B, &lda));
615 PetscCheck(lda == m, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Cannot use a LDA different than the number of local rows: % " PetscInt_FMT " != % " PetscInt_FMT, lda, m);
616 PetscCall(MatDenseGetLDA(X, &lda));
617 PetscCheck(lda == m, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Cannot use a LDA different than the number of local rows: % " PetscInt_FMT " != % " PetscInt_FMT, lda, m);
618 x_local = hypre_ParVectorLocalVector(hb);
619 PetscCallHYPRE(hypre_SeqVectorSetDataOwner(x_local, 0));
620 hypre_VectorData(x_local) = (HYPRE_Complex *)b;
621 PetscCall(MatDenseGetArrayWriteAndMemType(X, &x, NULL));
622 x_local = hypre_ParVectorLocalVector(hx);
623 PetscCallHYPRE(hypre_SeqVectorSetDataOwner(x_local, 0));
624 hypre_VectorData(x_local) = (HYPRE_Complex *)x;
625 PetscCallHYPRE(hypre_ParVectorInitialize_v2(hb, type == PETSC_MEMTYPE_HOST ? HYPRE_MEMORY_HOST : HYPRE_MEMORY_DEVICE));
626 PetscCallHYPRE(hypre_ParVectorInitialize_v2(hx, type == PETSC_MEMTYPE_HOST ? HYPRE_MEMORY_HOST : HYPRE_MEMORY_DEVICE));
627 PetscStackCallExternalVoid(
628 "Hypre solve", do {
629 HYPRE_Int hierr = (*jac->solve)(jac->hsolver, par_matrix, hb, hx);
630 if (hierr) {
631 PetscCheck(hierr == HYPRE_ERROR_CONV, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error in HYPRE solver, error code %d", (int)hierr);
632 HYPRE_ClearAllErrors();
633 }
634 } while (0));
635 PetscCallHYPRE(HYPRE_ParVectorDestroy(hb));
636 PetscCallHYPRE(HYPRE_ParVectorDestroy(hx));
637 PetscCall(MatDenseRestoreArrayReadAndMemType(B, &b));
638 PetscCall(MatDenseRestoreArrayWriteAndMemType(X, &x));
639 PetscFunctionReturn(PETSC_SUCCESS);
640 }
641
PCReset_HYPRE(PC pc)642 static PetscErrorCode PCReset_HYPRE(PC pc)
643 {
644 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
645
646 PetscFunctionBegin;
647 PetscCall(MatDestroy(&jac->hpmat));
648 PetscCall(MatDestroy(&jac->G));
649 PetscCall(MatDestroy(&jac->C));
650 PetscCall(MatDestroy(&jac->alpha_Poisson));
651 PetscCall(MatDestroy(&jac->beta_Poisson));
652 PetscCall(MatDestroy(&jac->RT_PiFull));
653 PetscCall(MatDestroy(&jac->RT_Pi[0]));
654 PetscCall(MatDestroy(&jac->RT_Pi[1]));
655 PetscCall(MatDestroy(&jac->RT_Pi[2]));
656 PetscCall(MatDestroy(&jac->ND_PiFull));
657 PetscCall(MatDestroy(&jac->ND_Pi[0]));
658 PetscCall(MatDestroy(&jac->ND_Pi[1]));
659 PetscCall(MatDestroy(&jac->ND_Pi[2]));
660 PetscCall(VecHYPRE_IJVectorDestroy(&jac->coords[0]));
661 PetscCall(VecHYPRE_IJVectorDestroy(&jac->coords[1]));
662 PetscCall(VecHYPRE_IJVectorDestroy(&jac->coords[2]));
663 PetscCall(VecHYPRE_IJVectorDestroy(&jac->constants[0]));
664 PetscCall(VecHYPRE_IJVectorDestroy(&jac->constants[1]));
665 PetscCall(VecHYPRE_IJVectorDestroy(&jac->constants[2]));
666 PetscCall(VecHYPRE_IJVectorDestroy(&jac->interior));
667 PetscCall(PCHYPREResetNearNullSpace_Private(pc));
668 jac->ams_beta_is_zero = PETSC_FALSE;
669 jac->ams_beta_is_zero_part = PETSC_FALSE;
670 jac->dim = 0;
671 PetscFunctionReturn(PETSC_SUCCESS);
672 }
673
PCDestroy_HYPRE(PC pc)674 static PetscErrorCode PCDestroy_HYPRE(PC pc)
675 {
676 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
677
678 PetscFunctionBegin;
679 PetscCall(PCReset_HYPRE(pc));
680 if (jac->destroy) PetscCallHYPRE((*jac->destroy)(jac->hsolver));
681 PetscCall(PetscFree(jac->hypre_type));
682 if (jac->comm_hypre != MPI_COMM_NULL) PetscCall(PetscCommRestoreComm(PetscObjectComm((PetscObject)pc), &jac->comm_hypre));
683 PetscCall(PetscFree(pc->data));
684
685 PetscCall(PetscObjectChangeTypeName((PetscObject)pc, 0));
686 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetType_C", NULL));
687 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPREGetType_C", NULL));
688 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetDiscreteGradient_C", NULL));
689 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetDiscreteCurl_C", NULL));
690 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetInterpolations_C", NULL));
691 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetConstantEdgeVectors_C", NULL));
692 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetPoissonMatrix_C", NULL));
693 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetEdgeConstantVectors_C", NULL));
694 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPREAMSSetInteriorNodes_C", NULL));
695 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetInterpolations_C", NULL));
696 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetCoarseOperators_C", NULL));
697 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPREGetCFMarkers_C", NULL));
698 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGalerkinSetMatProductAlgorithm_C", NULL));
699 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGalerkinGetMatProductAlgorithm_C", NULL));
700 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCSetCoordinates_C", NULL));
701 PetscFunctionReturn(PETSC_SUCCESS);
702 }
703
PCSetFromOptions_HYPRE_Pilut(PC pc,PetscOptionItems PetscOptionsObject)704 static PetscErrorCode PCSetFromOptions_HYPRE_Pilut(PC pc, PetscOptionItems PetscOptionsObject)
705 {
706 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
707 PetscBool flag;
708
709 PetscFunctionBegin;
710 PetscOptionsHeadBegin(PetscOptionsObject, "HYPRE Pilut Options");
711 PetscCall(PetscOptionsInt("-pc_hypre_pilut_maxiter", "Number of iterations", "None", jac->maxiter, &jac->maxiter, &flag));
712 if (flag) PetscCallHYPRE(HYPRE_ParCSRPilutSetMaxIter(jac->hsolver, (HYPRE_Int)jac->maxiter));
713 PetscCall(PetscOptionsReal("-pc_hypre_pilut_tol", "Drop tolerance", "None", jac->tol, &jac->tol, &flag));
714 if (flag) PetscCallHYPRE(HYPRE_ParCSRPilutSetDropTolerance(jac->hsolver, jac->tol));
715 PetscCall(PetscOptionsInt("-pc_hypre_pilut_factorrowsize", "FactorRowSize", "None", jac->factorrowsize, &jac->factorrowsize, &flag));
716 if (flag) PetscCallHYPRE(HYPRE_ParCSRPilutSetFactorRowSize(jac->hsolver, (HYPRE_Int)jac->factorrowsize));
717 PetscOptionsHeadEnd();
718 PetscFunctionReturn(PETSC_SUCCESS);
719 }
720
PCView_HYPRE_Pilut(PC pc,PetscViewer viewer)721 static PetscErrorCode PCView_HYPRE_Pilut(PC pc, PetscViewer viewer)
722 {
723 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
724 PetscBool isascii;
725
726 PetscFunctionBegin;
727 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
728 if (isascii) {
729 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE Pilut preconditioning\n"));
730 if (jac->maxiter != PETSC_DEFAULT) {
731 PetscCall(PetscViewerASCIIPrintf(viewer, " maximum number of iterations %" PetscInt_FMT "\n", jac->maxiter));
732 } else {
733 PetscCall(PetscViewerASCIIPrintf(viewer, " default maximum number of iterations \n"));
734 }
735 if (jac->tol != PETSC_DEFAULT) {
736 PetscCall(PetscViewerASCIIPrintf(viewer, " drop tolerance %g\n", (double)jac->tol));
737 } else {
738 PetscCall(PetscViewerASCIIPrintf(viewer, " default drop tolerance \n"));
739 }
740 if (jac->factorrowsize != PETSC_DEFAULT) {
741 PetscCall(PetscViewerASCIIPrintf(viewer, " factor row size %" PetscInt_FMT "\n", jac->factorrowsize));
742 } else {
743 PetscCall(PetscViewerASCIIPrintf(viewer, " default factor row size \n"));
744 }
745 }
746 PetscFunctionReturn(PETSC_SUCCESS);
747 }
748
749 static const char *HYPREILUType[] = {
750 "Block-Jacobi-ILUk", "Block-Jacobi-ILUT", "", "", "", "", "", "", "", "", /* 0-9 */
751 "GMRES-ILUk", "GMRES-ILUT", "", "", "", "", "", "", "", "", /* 10-19 */
752 "NSH-ILUk", "NSH-ILUT", "", "", "", "", "", "", "", "", /* 20-29 */
753 "RAS-ILUk", "RAS-ILUT", "", "", "", "", "", "", "", "", /* 30-39 */
754 "ddPQ-GMRES-ILUk", "ddPQ-GMRES-ILUT", "", "", "", "", "", "", "", "", /* 40-49 */
755 "GMRES-ILU0" /* 50 */
756 };
757
758 static const char *HYPREILUIterSetup[] = {"default", "async-in-place", "async-explicit", "sync-explicit", "semisync-explicit"};
759
PCSetFromOptions_HYPRE_ILU(PC pc,PetscOptionItems PetscOptionsObject)760 static PetscErrorCode PCSetFromOptions_HYPRE_ILU(PC pc, PetscOptionItems PetscOptionsObject)
761 {
762 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
763 PetscBool flg;
764 PetscInt indx;
765 PetscReal tmpdbl;
766 PetscBool tmp_truth;
767
768 PetscFunctionBegin;
769 PetscOptionsHeadBegin(PetscOptionsObject, "HYPRE ILU Options");
770
771 /* ILU: ILU Type */
772 PetscCall(PetscOptionsEList("-pc_hypre_ilu_type", "Choose ILU Type", "None", HYPREILUType, PETSC_STATIC_ARRAY_LENGTH(HYPREILUType), HYPREILUType[0], &indx, &flg));
773 if (flg) PetscCallHYPRE(HYPRE_ILUSetType(jac->hsolver, (HYPRE_Int)indx));
774
775 /* ILU: ILU iterative setup type*/
776 PetscCall(PetscOptionsEList("-pc_hypre_ilu_iterative_setup_type", "Set ILU iterative setup type", "None", HYPREILUIterSetup, PETSC_STATIC_ARRAY_LENGTH(HYPREILUIterSetup), HYPREILUIterSetup[0], &indx, &flg));
777 if (flg) PetscCallHYPRE(HYPRE_ILUSetIterativeSetupType(jac->hsolver, (HYPRE_Int)indx));
778
779 /* ILU: ILU iterative setup option*/
780 PetscCall(PetscOptionsInt("-pc_hypre_ilu_iterative_setup_option", "Set ILU iterative setup option", "None", 0, &indx, &flg));
781 if (flg) PetscCallHYPRE(HYPRE_ILUSetIterativeSetupOption(jac->hsolver, (HYPRE_Int)indx));
782
783 /* ILU: ILU iterative setup maxiter */
784 PetscCall(PetscOptionsInt("-pc_hypre_ilu_iterative_setup_maxiter", "Set ILU iterative setup maximum iteration count", "None", 0, &indx, &flg));
785 if (flg) PetscCallHYPRE(HYPRE_ILUSetIterativeSetupMaxIter(jac->hsolver, (HYPRE_Int)indx));
786
787 /* ILU: ILU iterative setup tolerance */
788 PetscCall(PetscOptionsReal("-pc_hypre_ilu_iterative_setup_tolerance", "Set ILU iterative setup tolerance", "None", 0, &tmpdbl, &flg));
789 if (flg) PetscCallHYPRE(HYPRE_ILUSetIterativeSetupTolerance(jac->hsolver, tmpdbl));
790
791 /* ILU: ILU Print Level */
792 PetscCall(PetscOptionsInt("-pc_hypre_ilu_print_level", "Set ILU print level", "None", 0, &indx, &flg));
793 if (flg) PetscCallHYPRE(HYPRE_ILUSetPrintLevel(jac->hsolver, (HYPRE_Int)indx));
794
795 /* ILU: Logging */
796 PetscCall(PetscOptionsInt("-pc_hypre_ilu_logging", "Set ILU logging level", "None", 0, &indx, &flg));
797 if (flg) PetscCallHYPRE(HYPRE_ILUSetLogging(jac->hsolver, (HYPRE_Int)indx));
798
799 /* ILU: ILU Level */
800 PetscCall(PetscOptionsInt("-pc_hypre_ilu_level", "Set ILU level", "None", 0, &indx, &flg));
801 if (flg) PetscCallHYPRE(HYPRE_ILUSetLevelOfFill(jac->hsolver, (HYPRE_Int)indx));
802
803 /* ILU: ILU Max NNZ per row */
804 PetscCall(PetscOptionsInt("-pc_hypre_ilu_max_nnz_per_row", "Set maximum NNZ per row", "None", 0, &indx, &flg));
805 if (flg) PetscCallHYPRE(HYPRE_ILUSetMaxNnzPerRow(jac->hsolver, (HYPRE_Int)indx));
806
807 /* ILU: tolerance */
808 PetscCall(PetscOptionsReal("-pc_hypre_ilu_tol", "Tolerance for ILU", "None", 0, &tmpdbl, &flg));
809 if (flg) PetscCallHYPRE(HYPRE_ILUSetTol(jac->hsolver, tmpdbl));
810
811 /* ILU: maximum iteration count */
812 PetscCall(PetscOptionsInt("-pc_hypre_ilu_maxiter", "Set ILU max iterations", "None", 0, &indx, &flg));
813 if (flg) PetscCallHYPRE(HYPRE_ILUSetMaxIter(jac->hsolver, (HYPRE_Int)indx));
814
815 /* ILU: drop threshold */
816 PetscCall(PetscOptionsReal("-pc_hypre_ilu_drop_threshold", "Drop threshold for ILU", "None", 0, &tmpdbl, &flg));
817 if (flg) PetscCallHYPRE(HYPRE_ILUSetDropThreshold(jac->hsolver, tmpdbl));
818
819 /* ILU: Triangular Solve */
820 PetscCall(PetscOptionsBool("-pc_hypre_ilu_tri_solve", "Enable triangular solve", "None", PETSC_FALSE, &tmp_truth, &flg));
821 if (flg) PetscCallHYPRE(HYPRE_ILUSetTriSolve(jac->hsolver, tmp_truth));
822
823 /* ILU: Lower Jacobi iteration */
824 PetscCall(PetscOptionsInt("-pc_hypre_ilu_lower_jacobi_iters", "Set lower Jacobi iteration count", "None", 0, &indx, &flg));
825 if (flg) PetscCallHYPRE(HYPRE_ILUSetLowerJacobiIters(jac->hsolver, (HYPRE_Int)indx));
826
827 /* ILU: Upper Jacobi iteration */
828 PetscCall(PetscOptionsInt("-pc_hypre_ilu_upper_jacobi_iters", "Set upper Jacobi iteration count", "None", 0, &indx, &flg));
829 if (flg) PetscCallHYPRE(HYPRE_ILUSetUpperJacobiIters(jac->hsolver, (HYPRE_Int)indx));
830
831 /* ILU: local reordering */
832 PetscCall(PetscOptionsBool("-pc_hypre_ilu_local_reordering", "Enable local reordering", "None", PETSC_FALSE, &tmp_truth, &flg));
833 if (flg) PetscCallHYPRE(HYPRE_ILUSetLocalReordering(jac->hsolver, tmp_truth));
834
835 PetscOptionsHeadEnd();
836 PetscFunctionReturn(PETSC_SUCCESS);
837 }
838
PCView_HYPRE_ILU(PC pc,PetscViewer viewer)839 static PetscErrorCode PCView_HYPRE_ILU(PC pc, PetscViewer viewer)
840 {
841 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
842 hypre_ParILUData *ilu_data = (hypre_ParILUData *)jac->hsolver;
843 PetscBool isascii;
844 PetscInt indx;
845 PetscReal tmpdbl;
846 PetscReal *tmpdbl3;
847
848 PetscFunctionBegin;
849 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
850 if (isascii) {
851 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE ILU preconditioning\n"));
852 PetscStackCallExternalVoid("hypre_ParILUDataIluType", indx = hypre_ParILUDataIluType(ilu_data));
853 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU type %s (%" PetscInt_FMT ")\n", HYPREILUType[indx], indx));
854 PetscStackCallExternalVoid("hypre_ParILUDataLfil", indx = hypre_ParILUDataLfil(ilu_data));
855 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU level %" PetscInt_FMT "\n", indx));
856 PetscStackCallExternalVoid("hypre_ParILUDataMaxIter", indx = hypre_ParILUDataMaxIter(ilu_data));
857 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU max iterations %" PetscInt_FMT "\n", indx));
858 PetscStackCallExternalVoid("hypre_ParILUDataMaxRowNnz", indx = hypre_ParILUDataMaxRowNnz(ilu_data));
859 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU max NNZ per row %" PetscInt_FMT "\n", indx));
860 PetscStackCallExternalVoid("hypre_ParILUDataTriSolve", indx = hypre_ParILUDataTriSolve(ilu_data));
861 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU triangular solve %" PetscInt_FMT "\n", indx));
862 PetscStackCallExternalVoid("hypre_ParILUDataTol", tmpdbl = hypre_ParILUDataTol(ilu_data));
863 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU tolerance %e\n", tmpdbl));
864 PetscStackCallExternalVoid("hypre_ParILUDataDroptol", tmpdbl3 = hypre_ParILUDataDroptol(ilu_data));
865 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU drop tolerance %e / %e / %e\n", tmpdbl3[0], tmpdbl3[1], tmpdbl3[2]));
866 PetscStackCallExternalVoid("hypre_ParILUDataReorderingType", indx = hypre_ParILUDataReorderingType(ilu_data));
867 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU local reordering %" PetscInt_FMT "\n", indx));
868 PetscStackCallExternalVoid("hypre_ParILUDataLowerJacobiIters", indx = hypre_ParILUDataLowerJacobiIters(ilu_data));
869 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU lower Jacobi iterations %" PetscInt_FMT "\n", indx));
870 PetscStackCallExternalVoid("hypre_ParILUDataUpperJacobiIters", indx = hypre_ParILUDataUpperJacobiIters(ilu_data));
871 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU upper Jacobi iterations %" PetscInt_FMT "\n", indx));
872 PetscStackCallExternalVoid("hypre_ParILUDataPrintLevel", indx = hypre_ParILUDataPrintLevel(ilu_data));
873 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU print level %" PetscInt_FMT "\n", indx));
874 PetscStackCallExternalVoid("hypre_ParILUDataLogging", indx = hypre_ParILUDataLogging(ilu_data));
875 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU logging level %" PetscInt_FMT "\n", indx));
876 PetscStackCallExternalVoid("hypre_ParILUDataIterativeSetupType", indx = hypre_ParILUDataIterativeSetupType(ilu_data));
877 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU iterative setup type %s (%" PetscInt_FMT ")\n", HYPREILUIterSetup[indx], indx));
878 PetscStackCallExternalVoid("hypre_ParILUDataIterativeSetupOption", indx = hypre_ParILUDataIterativeSetupOption(ilu_data));
879 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU iterative setup option %" PetscInt_FMT "\n", indx));
880 PetscStackCallExternalVoid("hypre_ParILUDataIterativeSetupMaxIter", indx = hypre_ParILUDataIterativeSetupMaxIter(ilu_data));
881 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU iterative setup max iterations %" PetscInt_FMT "\n", indx));
882 PetscStackCallExternalVoid("hypre_ParILUDataIterativeSetupTolerance", tmpdbl = hypre_ParILUDataIterativeSetupTolerance(ilu_data));
883 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU iterative setup tolerance %e\n", tmpdbl));
884 }
885 PetscFunctionReturn(PETSC_SUCCESS);
886 }
887
PCSetFromOptions_HYPRE_Euclid(PC pc,PetscOptionItems PetscOptionsObject)888 static PetscErrorCode PCSetFromOptions_HYPRE_Euclid(PC pc, PetscOptionItems PetscOptionsObject)
889 {
890 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
891 PetscBool flag, eu_bj = jac->eu_bj ? PETSC_TRUE : PETSC_FALSE;
892
893 PetscFunctionBegin;
894 PetscOptionsHeadBegin(PetscOptionsObject, "HYPRE Euclid Options");
895 PetscCall(PetscOptionsInt("-pc_hypre_euclid_level", "Factorization levels", "None", jac->eu_level, &jac->eu_level, &flag));
896 if (flag) PetscCallHYPRE(HYPRE_EuclidSetLevel(jac->hsolver, (HYPRE_Int)jac->eu_level));
897
898 PetscCall(PetscOptionsReal("-pc_hypre_euclid_droptolerance", "Drop tolerance for ILU(k) in Euclid", "None", jac->eu_droptolerance, &jac->eu_droptolerance, &flag));
899 if (flag) {
900 PetscMPIInt size;
901
902 PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)pc), &size));
903 PetscCheck(size == 1, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "hypre's Euclid does not support a parallel drop tolerance");
904 PetscCallHYPRE(HYPRE_EuclidSetILUT(jac->hsolver, jac->eu_droptolerance));
905 }
906
907 PetscCall(PetscOptionsBool("-pc_hypre_euclid_bj", "Use Block Jacobi for ILU in Euclid", "None", eu_bj, &eu_bj, &flag));
908 if (flag) {
909 jac->eu_bj = eu_bj ? 1 : 0;
910 PetscCallHYPRE(HYPRE_EuclidSetBJ(jac->hsolver, (HYPRE_Int)jac->eu_bj));
911 }
912 PetscOptionsHeadEnd();
913 PetscFunctionReturn(PETSC_SUCCESS);
914 }
915
PCView_HYPRE_Euclid(PC pc,PetscViewer viewer)916 static PetscErrorCode PCView_HYPRE_Euclid(PC pc, PetscViewer viewer)
917 {
918 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
919 PetscBool isascii;
920
921 PetscFunctionBegin;
922 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
923 if (isascii) {
924 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE Euclid preconditioning\n"));
925 if (jac->eu_level != PETSC_DEFAULT) {
926 PetscCall(PetscViewerASCIIPrintf(viewer, " factorization levels %" PetscInt_FMT "\n", jac->eu_level));
927 } else {
928 PetscCall(PetscViewerASCIIPrintf(viewer, " default factorization levels \n"));
929 }
930 PetscCall(PetscViewerASCIIPrintf(viewer, " drop tolerance %g\n", (double)jac->eu_droptolerance));
931 PetscCall(PetscViewerASCIIPrintf(viewer, " use Block-Jacobi? %" PetscInt_FMT "\n", jac->eu_bj));
932 }
933 PetscFunctionReturn(PETSC_SUCCESS);
934 }
935
PCApplyTranspose_HYPRE_BoomerAMG(PC pc,Vec b,Vec x)936 static PetscErrorCode PCApplyTranspose_HYPRE_BoomerAMG(PC pc, Vec b, Vec x)
937 {
938 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
939 Mat_HYPRE *hjac = (Mat_HYPRE *)jac->hpmat->data;
940 HYPRE_ParCSRMatrix hmat;
941 HYPRE_ParVector jbv, jxv;
942
943 PetscFunctionBegin;
944 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
945 PetscCall(VecSet(x, 0.0));
946 PetscCall(VecHYPRE_IJVectorPushVecRead(hjac->b, b));
947 PetscCall(VecHYPRE_IJVectorPushVecWrite(hjac->x, x));
948
949 PetscCallHYPRE(HYPRE_IJMatrixGetObject(hjac->ij, (void **)&hmat));
950 PetscCallHYPRE(HYPRE_IJVectorGetObject(hjac->b->ij, (void **)&jbv));
951 PetscCallHYPRE(HYPRE_IJVectorGetObject(hjac->x->ij, (void **)&jxv));
952
953 PetscStackCallExternalVoid(
954 "Hypre Transpose solve", do {
955 HYPRE_Int hierr = HYPRE_BoomerAMGSolveT(jac->hsolver, hmat, jbv, jxv);
956 if (hierr) {
957 /* error code of 1 in BoomerAMG merely means convergence not achieved */
958 PetscCheck(hierr == 1, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error in HYPRE solver, error code %d", (int)hierr);
959 HYPRE_ClearAllErrors();
960 }
961 } while (0));
962
963 PetscCall(VecHYPRE_IJVectorPopVec(hjac->x));
964 PetscCall(VecHYPRE_IJVectorPopVec(hjac->b));
965 PetscFunctionReturn(PETSC_SUCCESS);
966 }
967
PCMGGalerkinGetMatProductAlgorithm_HYPRE_BoomerAMG(PC pc,const char * spgemm[])968 static PetscErrorCode PCMGGalerkinGetMatProductAlgorithm_HYPRE_BoomerAMG(PC pc, const char *spgemm[])
969 {
970 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
971
972 PetscFunctionBegin;
973 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
974 #if PETSC_PKG_HYPRE_VERSION_GE(2, 23, 0)
975 *spgemm = jac->spgemm_type;
976 #endif
977 PetscFunctionReturn(PETSC_SUCCESS);
978 }
979
980 static const char *HYPREBoomerAMGCycleType[] = {"", "V", "W"};
981 static const char *HYPREBoomerAMGCoarsenType[] = {"CLJP", "Ruge-Stueben", "", "modifiedRuge-Stueben", "", "", "Falgout", "", "PMIS", "", "HMIS"};
982 static const char *HYPREBoomerAMGMeasureType[] = {"local", "global"};
983 /* The following corresponds to HYPRE_BoomerAMGSetRelaxType which has many missing numbers in the enum */
984 static const char *HYPREBoomerAMGSmoothType[] = {"ILU", "Schwarz-smoothers", "Pilut", "ParaSails", "Euclid"};
985 static const char *HYPREBoomerAMGRelaxType[] = {"Jacobi", "sequential-Gauss-Seidel", "seqboundary-Gauss-Seidel", "SOR/Jacobi", "backward-SOR/Jacobi", "" /* [5] hybrid chaotic Gauss-Seidel (works only with OpenMP) */, "symmetric-SOR/Jacobi", "" /* 7 */, "l1scaled-SOR/Jacobi", "Gaussian-elimination", "" /* 10 */, "" /* 11 */, "" /* 12 */, "l1-Gauss-Seidel" /* nonsymmetric */, "backward-l1-Gauss-Seidel" /* nonsymmetric */, "CG" /* non-stationary */, "Chebyshev", "FCF-Jacobi", "l1scaled-Jacobi"};
986 static const char *HYPREBoomerAMGInterpType[] = {"classical", "", "", "direct", "multipass", "multipass-wts", "ext+i", "ext+i-cc", "standard", "standard-wts", "block", "block-wtd", "FF", "FF1", "ext", "ad-wts", "ext-mm", "ext+i-mm", "ext+e-mm"};
987
PCSetFromOptions_HYPRE_BoomerAMG(PC pc,PetscOptionItems PetscOptionsObject)988 static PetscErrorCode PCSetFromOptions_HYPRE_BoomerAMG(PC pc, PetscOptionItems PetscOptionsObject)
989 {
990 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
991 PetscInt bs, n, indx, level;
992 PetscBool flg, tmp_truth;
993 PetscReal tmpdbl, twodbl[2];
994 const char *symtlist[] = {"nonsymmetric", "SPD", "nonsymmetric,SPD"};
995
996 PetscFunctionBegin;
997 PetscOptionsHeadBegin(PetscOptionsObject, "HYPRE BoomerAMG Options");
998 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_cycle_type", "Cycle type", "None", HYPREBoomerAMGCycleType + 1, 2, HYPREBoomerAMGCycleType[jac->cycletype], &indx, &flg));
999 if (flg) {
1000 jac->cycletype = indx + 1;
1001 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleType(jac->hsolver, (HYPRE_Int)jac->cycletype));
1002 }
1003 PetscCall(PetscOptionsBoundedInt("-pc_hypre_boomeramg_max_levels", "Number of levels (of grids) allowed", "None", jac->maxlevels, &jac->maxlevels, &flg, 2));
1004 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetMaxLevels(jac->hsolver, (HYPRE_Int)jac->maxlevels));
1005 PetscCall(PetscOptionsBoundedInt("-pc_hypre_boomeramg_max_iter", "Maximum iterations used PER hypre call", "None", jac->maxiter, &jac->maxiter, &flg, 1));
1006 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetMaxIter(jac->hsolver, (HYPRE_Int)jac->maxiter));
1007 PetscCall(PetscOptionsBoundedReal("-pc_hypre_boomeramg_tol", "Convergence tolerance PER hypre call (0.0 = use a fixed number of iterations)", "None", jac->tol, &jac->tol, &flg, 0.0));
1008 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetTol(jac->hsolver, jac->tol));
1009 bs = 1;
1010 if (pc->pmat) PetscCall(MatGetBlockSize(pc->pmat, &bs));
1011 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_numfunctions", "Number of functions", "HYPRE_BoomerAMGSetNumFunctions", bs, &bs, &flg));
1012 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetNumFunctions(jac->hsolver, (HYPRE_Int)bs));
1013
1014 PetscCall(PetscOptionsBoundedReal("-pc_hypre_boomeramg_truncfactor", "Truncation factor for interpolation (0=no truncation)", "None", jac->truncfactor, &jac->truncfactor, &flg, 0.0));
1015 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetTruncFactor(jac->hsolver, jac->truncfactor));
1016
1017 PetscCall(PetscOptionsBoundedInt("-pc_hypre_boomeramg_P_max", "Max elements per row for interpolation operator (0=unlimited)", "None", jac->pmax, &jac->pmax, &flg, 0));
1018 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetPMaxElmts(jac->hsolver, (HYPRE_Int)jac->pmax));
1019
1020 PetscCall(PetscOptionsRangeInt("-pc_hypre_boomeramg_agg_nl", "Number of levels of aggressive coarsening", "None", jac->agg_nl, &jac->agg_nl, &flg, 0, jac->maxlevels));
1021 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetAggNumLevels(jac->hsolver, (HYPRE_Int)jac->agg_nl));
1022
1023 PetscCall(PetscOptionsBoundedInt("-pc_hypre_boomeramg_agg_num_paths", "Number of paths for aggressive coarsening", "None", jac->agg_num_paths, &jac->agg_num_paths, &flg, 1));
1024 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetNumPaths(jac->hsolver, (HYPRE_Int)jac->agg_num_paths));
1025
1026 PetscCall(PetscOptionsBoundedReal("-pc_hypre_boomeramg_strong_threshold", "Threshold for being strongly connected", "None", jac->strongthreshold, &jac->strongthreshold, &flg, 0.0));
1027 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetStrongThreshold(jac->hsolver, jac->strongthreshold));
1028 PetscCall(PetscOptionsRangeReal("-pc_hypre_boomeramg_max_row_sum", "Maximum row sum", "None", jac->maxrowsum, &jac->maxrowsum, &flg, 0.0, 1.0));
1029 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetMaxRowSum(jac->hsolver, jac->maxrowsum));
1030
1031 /* Grid sweeps */
1032 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_grid_sweeps_all", "Number of sweeps for the up and down grid levels", "None", jac->gridsweeps[0], &indx, &flg));
1033 if (flg) {
1034 /* modify the jac structure so we can view the updated options with PC_View */
1035 jac->gridsweeps[0] = indx;
1036 jac->gridsweeps[1] = indx;
1037 /*defaults coarse to 1 */
1038 jac->gridsweeps[2] = 1;
1039 }
1040 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_nodal_coarsen", "Use a nodal based coarsening 1-6", "HYPRE_BoomerAMGSetNodal", jac->nodal_coarsening, &jac->nodal_coarsening, &flg));
1041 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetNodal(jac->hsolver, (HYPRE_Int)jac->nodal_coarsening));
1042 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_nodal_coarsen_diag", "Diagonal in strength matrix for nodal based coarsening 0-2", "HYPRE_BoomerAMGSetNodalDiag", jac->nodal_coarsening_diag, &jac->nodal_coarsening_diag, &flg));
1043 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetNodalDiag(jac->hsolver, (HYPRE_Int)jac->nodal_coarsening_diag));
1044 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_vec_interp_variant", "Variant of algorithm 1-3", "HYPRE_BoomerAMGSetInterpVecVariant", jac->vec_interp_variant, &jac->vec_interp_variant, &flg));
1045 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetInterpVecVariant(jac->hsolver, (HYPRE_Int)jac->vec_interp_variant));
1046 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_vec_interp_qmax", "Max elements per row for each Q", "HYPRE_BoomerAMGSetInterpVecQMax", jac->vec_interp_qmax, &jac->vec_interp_qmax, &flg));
1047 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetInterpVecQMax(jac->hsolver, (HYPRE_Int)jac->vec_interp_qmax));
1048 PetscCall(PetscOptionsBool("-pc_hypre_boomeramg_vec_interp_smooth", "Whether to smooth the interpolation vectors", "HYPRE_BoomerAMGSetSmoothInterpVectors", jac->vec_interp_smooth, &jac->vec_interp_smooth, &flg));
1049 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetSmoothInterpVectors(jac->hsolver, jac->vec_interp_smooth));
1050 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_interp_refine", "Preprocess the interpolation matrix through iterative weight refinement", "HYPRE_BoomerAMGSetInterpRefine", jac->interp_refine, &jac->interp_refine, &flg));
1051 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetInterpRefine(jac->hsolver, (HYPRE_Int)jac->interp_refine));
1052 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_grid_sweeps_down", "Number of sweeps for the down cycles", "None", jac->gridsweeps[0], &indx, &flg));
1053 if (flg) {
1054 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleNumSweeps(jac->hsolver, (HYPRE_Int)indx, 1));
1055 jac->gridsweeps[0] = indx;
1056 }
1057 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_grid_sweeps_up", "Number of sweeps for the up cycles", "None", jac->gridsweeps[1], &indx, &flg));
1058 if (flg) {
1059 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleNumSweeps(jac->hsolver, (HYPRE_Int)indx, 2));
1060 jac->gridsweeps[1] = indx;
1061 }
1062 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_grid_sweeps_coarse", "Number of sweeps for the coarse level", "None", jac->gridsweeps[2], &indx, &flg));
1063 if (flg) {
1064 PetscCallHYPRE(HYPRE_BoomerAMGSetCycleNumSweeps(jac->hsolver, (HYPRE_Int)indx, 3));
1065 jac->gridsweeps[2] = indx;
1066 }
1067
1068 /* Smooth type */
1069 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_smooth_type", "Enable more complex smoothers", "None", HYPREBoomerAMGSmoothType, PETSC_STATIC_ARRAY_LENGTH(HYPREBoomerAMGSmoothType), HYPREBoomerAMGSmoothType[0], &indx, &flg));
1070 if (flg) {
1071 jac->smoothtype = indx;
1072 PetscCallHYPRE(HYPRE_BoomerAMGSetSmoothType(jac->hsolver, (HYPRE_Int)indx + 5));
1073 jac->smoothnumlevels = 25;
1074 PetscCallHYPRE(HYPRE_BoomerAMGSetSmoothNumLevels(jac->hsolver, 25));
1075 }
1076
1077 /* Number of smoothing levels */
1078 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_smooth_num_levels", "Number of levels on which more complex smoothers are used", "None", 25, &indx, &flg));
1079 if (flg && (jac->smoothtype != -1)) {
1080 jac->smoothnumlevels = indx;
1081 PetscCallHYPRE(HYPRE_BoomerAMGSetSmoothNumLevels(jac->hsolver, (HYPRE_Int)indx));
1082 }
1083
1084 /* Smooth num sweeps */
1085 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_smooth_num_sweeps", "Set number of smoother sweeps", "None", 1, &indx, &flg));
1086 if (flg && indx > 0) {
1087 jac->smoothsweeps = indx;
1088 PetscCallHYPRE(HYPRE_BoomerAMGSetSmoothNumSweeps(jac->hsolver, (HYPRE_Int)indx));
1089 }
1090
1091 /* ILU: ILU Type */
1092 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_ilu_type", "Choose ILU Type", "None", HYPREILUType, PETSC_STATIC_ARRAY_LENGTH(HYPREILUType), HYPREILUType[0], &indx, &flg));
1093 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILUType(jac->hsolver, (HYPRE_Int)indx));
1094
1095 /* ILU: ILU iterative setup type*/
1096 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_ilu_iterative_setup_type", "Set ILU iterative setup type", "None", HYPREILUIterSetup, PETSC_STATIC_ARRAY_LENGTH(HYPREILUIterSetup), HYPREILUIterSetup[0], &indx, &flg));
1097 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILUIterSetupType(jac->hsolver, (HYPRE_Int)indx));
1098
1099 /* ILU: ILU iterative setup option*/
1100 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_ilu_iterative_setup_option", "Set ILU iterative setup option", "None", 0, &indx, &flg));
1101 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILUIterSetupOption(jac->hsolver, (HYPRE_Int)indx));
1102
1103 /* ILU: ILU iterative setup maxiter */
1104 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_ilu_iterative_setup_maxiter", "Set ILU iterative setup maximum iteration count", "None", 0, &indx, &flg));
1105 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILUIterSetupMaxIter(jac->hsolver, (HYPRE_Int)indx));
1106
1107 /* ILU: ILU iterative setup tolerance */
1108 PetscCall(PetscOptionsReal("-pc_hypre_boomeramg_ilu_iterative_setup_tolerance", "Set ILU iterative setup tolerance", "None", 0, &tmpdbl, &flg));
1109 if (flg) PetscCallHYPRE(hypre_BoomerAMGSetILUIterSetupTolerance(jac->hsolver, tmpdbl));
1110
1111 /* ILU: ILU Print Level */
1112 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_ilu_print_level", "Set ILU print level", "None", 0, &indx, &flg));
1113 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetPrintLevel(jac->hsolver, (HYPRE_Int)indx));
1114
1115 /* ILU: Logging */
1116 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_ilu_logging", "Set ILU logging level", "None", 0, &indx, &flg));
1117 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetLogging(jac->hsolver, (HYPRE_Int)indx));
1118
1119 /* ILU: ILU Level */
1120 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_ilu_level", "Set ILU level", "None", 0, &indx, &flg));
1121 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILULevel(jac->hsolver, (HYPRE_Int)indx));
1122
1123 /* ILU: ILU Max NNZ per row */
1124 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_ilu_max_nnz_per_row", "Set maximum NNZ per row", "None", 0, &indx, &flg));
1125 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILUMaxRowNnz(jac->hsolver, (HYPRE_Int)indx));
1126
1127 /* ILU: maximum iteration count */
1128 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_ilu_maxiter", "Set ILU max iterations", "None", 0, &indx, &flg));
1129 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILUMaxIter(jac->hsolver, (HYPRE_Int)indx));
1130
1131 /* ILU: drop threshold */
1132 PetscCall(PetscOptionsReal("-pc_hypre_boomeramg_ilu_drop_tol", "Drop tolerance for ILU", "None", 0, &tmpdbl, &flg));
1133 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILUDroptol(jac->hsolver, tmpdbl));
1134
1135 /* ILU: Triangular Solve */
1136 PetscCall(PetscOptionsBool("-pc_hypre_boomeramg_ilu_tri_solve", "Enable triangular solve", "None", PETSC_FALSE, &tmp_truth, &flg));
1137 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILUTriSolve(jac->hsolver, tmp_truth));
1138
1139 /* ILU: Lower Jacobi iteration */
1140 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_ilu_lower_jacobi_iters", "Set lower Jacobi iteration count", "None", 0, &indx, &flg));
1141 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILULowerJacobiIters(jac->hsolver, (HYPRE_Int)indx));
1142
1143 /* ILU: Upper Jacobi iteration */
1144 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_ilu_upper_jacobi_iters", "Set upper Jacobi iteration count", "None", 0, &indx, &flg));
1145 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILUUpperJacobiIters(jac->hsolver, (HYPRE_Int)indx));
1146
1147 /* ILU: local reordering */
1148 PetscCall(PetscOptionsBool("-pc_hypre_boomeramg_ilu_local_reordering", "Enable local reordering", "None", PETSC_FALSE, &tmp_truth, &flg));
1149 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetILULocalReordering(jac->hsolver, tmp_truth));
1150
1151 /* Number of levels for ILU(k) for Euclid */
1152 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_eu_level", "Number of levels for ILU(k) in Euclid smoother", "None", 0, &indx, &flg));
1153 if (flg && (jac->smoothtype == 4)) {
1154 jac->eu_level = indx;
1155 PetscCallHYPRE(HYPRE_BoomerAMGSetEuLevel(jac->hsolver, (HYPRE_Int)indx));
1156 }
1157
1158 /* Filter for ILU(k) for Euclid */
1159 PetscReal droptolerance;
1160 PetscCall(PetscOptionsReal("-pc_hypre_boomeramg_eu_droptolerance", "Drop tolerance for ILU(k) in Euclid smoother", "None", 0, &droptolerance, &flg));
1161 if (flg && (jac->smoothtype == 4)) {
1162 jac->eu_droptolerance = droptolerance;
1163 PetscCallHYPRE(HYPRE_BoomerAMGSetEuLevel(jac->hsolver, droptolerance));
1164 }
1165
1166 /* Use Block Jacobi ILUT for Euclid */
1167 PetscCall(PetscOptionsBool("-pc_hypre_boomeramg_eu_bj", "Use Block Jacobi for ILU in Euclid smoother?", "None", PETSC_FALSE, &tmp_truth, &flg));
1168 if (flg && (jac->smoothtype == 4)) {
1169 jac->eu_bj = tmp_truth;
1170 PetscCallHYPRE(HYPRE_BoomerAMGSetEuBJ(jac->hsolver, (HYPRE_Int)jac->eu_bj));
1171 }
1172
1173 /* Relax type */
1174 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_relax_type_all", "Relax type for the up and down cycles", "None", HYPREBoomerAMGRelaxType, PETSC_STATIC_ARRAY_LENGTH(HYPREBoomerAMGRelaxType),
1175 jac->relaxtype[0] < 0 ? "not yet set" : HYPREBoomerAMGRelaxType[jac->relaxtype[0]], &indx, &flg));
1176 if (flg) jac->relaxtype[0] = jac->relaxtype[1] = indx;
1177 PetscCall(
1178 PetscOptionsEList("-pc_hypre_boomeramg_relax_type_down", "Relax type for the down cycles", "None", HYPREBoomerAMGRelaxType, PETSC_STATIC_ARRAY_LENGTH(HYPREBoomerAMGRelaxType), jac->relaxtype[0] < 0 ? "not yet set" : HYPREBoomerAMGRelaxType[jac->relaxtype[0]], &indx, &flg));
1179 if (flg) jac->relaxtype[0] = indx;
1180 PetscCall(
1181 PetscOptionsEList("-pc_hypre_boomeramg_relax_type_up", "Relax type for the up cycles", "None", HYPREBoomerAMGRelaxType, PETSC_STATIC_ARRAY_LENGTH(HYPREBoomerAMGRelaxType), jac->relaxtype[1] < 0 ? "not yet set" : HYPREBoomerAMGRelaxType[jac->relaxtype[1]], &indx, &flg));
1182 if (flg) jac->relaxtype[1] = indx;
1183 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_relax_type_coarse", "Relax type on coarse grid", "None", HYPREBoomerAMGRelaxType, PETSC_STATIC_ARRAY_LENGTH(HYPREBoomerAMGRelaxType), HYPREBoomerAMGRelaxType[jac->relaxtype[2]], &indx, &flg));
1184 if (flg) jac->relaxtype[2] = indx;
1185
1186 /* Relaxation Weight */
1187 PetscCall(PetscOptionsReal("-pc_hypre_boomeramg_relax_weight_all", "Relaxation weight for all levels (0 = hypre estimates, -k = determined with k CG steps)", "None", jac->relaxweight, &tmpdbl, &flg));
1188 if (flg) {
1189 PetscCallHYPRE(HYPRE_BoomerAMGSetRelaxWt(jac->hsolver, tmpdbl));
1190 jac->relaxweight = tmpdbl;
1191 }
1192
1193 n = 2;
1194 twodbl[0] = twodbl[1] = 1.0;
1195 PetscCall(PetscOptionsRealArray("-pc_hypre_boomeramg_relax_weight_level", "Set the relaxation weight for a particular level (weight,level)", "None", twodbl, &n, &flg));
1196 if (flg) {
1197 PetscCheck(n == 2, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_OUTOFRANGE, "Relax weight level: you must provide 2 values separated by a comma (and no space), you provided %" PetscInt_FMT, n);
1198 indx = (int)PetscAbsReal(twodbl[1]);
1199 PetscCallHYPRE(HYPRE_BoomerAMGSetLevelRelaxWt(jac->hsolver, twodbl[0], (HYPRE_Int)indx));
1200 }
1201
1202 /* Outer relaxation Weight */
1203 PetscCall(PetscOptionsReal("-pc_hypre_boomeramg_outer_relax_weight_all", "Outer relaxation weight for all levels (-k = determined with k CG steps)", "None", jac->outerrelaxweight, &tmpdbl, &flg));
1204 if (flg) {
1205 PetscCallHYPRE(HYPRE_BoomerAMGSetOuterWt(jac->hsolver, tmpdbl));
1206 jac->outerrelaxweight = tmpdbl;
1207 }
1208
1209 n = 2;
1210 twodbl[0] = twodbl[1] = 1.0;
1211 PetscCall(PetscOptionsRealArray("-pc_hypre_boomeramg_outer_relax_weight_level", "Set the outer relaxation weight for a particular level (weight,level)", "None", twodbl, &n, &flg));
1212 if (flg) {
1213 PetscCheck(n == 2, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_OUTOFRANGE, "Relax weight outer level: You must provide 2 values separated by a comma (and no space), you provided %" PetscInt_FMT, n);
1214 indx = (int)PetscAbsReal(twodbl[1]);
1215 PetscCallHYPRE(HYPRE_BoomerAMGSetLevelOuterWt(jac->hsolver, twodbl[0], (HYPRE_Int)indx));
1216 }
1217
1218 /* the Relax Order */
1219 PetscCall(PetscOptionsBool("-pc_hypre_boomeramg_no_CF", "Do not use CF-relaxation", "None", PETSC_FALSE, &tmp_truth, &flg));
1220 if (flg) jac->relaxorder = !tmp_truth;
1221 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_measure_type", "Measure type", "None", HYPREBoomerAMGMeasureType, PETSC_STATIC_ARRAY_LENGTH(HYPREBoomerAMGMeasureType), HYPREBoomerAMGMeasureType[0], &indx, &flg));
1222 if (flg) {
1223 jac->measuretype = indx;
1224 PetscCallHYPRE(HYPRE_BoomerAMGSetMeasureType(jac->hsolver, (HYPRE_Int)jac->measuretype));
1225 }
1226 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_coarsen_type", "Coarsen type", "None", HYPREBoomerAMGCoarsenType, PETSC_STATIC_ARRAY_LENGTH(HYPREBoomerAMGCoarsenType), jac->coarsentype < 0 ? "unknown" : HYPREBoomerAMGCoarsenType[jac->coarsentype], &indx, &flg));
1227 if (flg) jac->coarsentype = indx;
1228
1229 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_max_coarse_size", "Maximum size of coarsest grid", "None", jac->maxc, &jac->maxc, &flg));
1230 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetMaxCoarseSize(jac->hsolver, (HYPRE_Int)jac->maxc));
1231 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_min_coarse_size", "Minimum size of coarsest grid", "None", jac->minc, &jac->minc, &flg));
1232 if (flg) PetscCallHYPRE(HYPRE_BoomerAMGSetMinCoarseSize(jac->hsolver, (HYPRE_Int)jac->minc));
1233 #if PETSC_PKG_HYPRE_VERSION_GE(2, 23, 0)
1234 // global parameter but is closely associated with BoomerAMG
1235 PetscCall(PetscOptionsEList("-pc_mg_galerkin_mat_product_algorithm", "Type of SpGEMM to use in hypre (only for now)", "PCMGGalerkinSetMatProductAlgorithm", HYPRESpgemmTypes, PETSC_STATIC_ARRAY_LENGTH(HYPRESpgemmTypes), jac->spgemm_type, &indx, &flg));
1236 if (flg) PetscCall(PCMGGalerkinSetMatProductAlgorithm_HYPRE_BoomerAMG(pc, HYPRESpgemmTypes[indx]));
1237 #endif
1238 /* AIR */
1239 #if PETSC_PKG_HYPRE_VERSION_GE(2, 18, 0)
1240 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_restriction_type", "Type of AIR method (distance 1 or 2, 0 means no AIR)", "None", jac->Rtype, &jac->Rtype, NULL));
1241 PetscCallHYPRE(HYPRE_BoomerAMGSetRestriction(jac->hsolver, (HYPRE_Int)jac->Rtype));
1242 if (jac->Rtype) {
1243 HYPRE_Int **grid_relax_points = hypre_TAlloc(HYPRE_Int *, 4, HYPRE_MEMORY_HOST);
1244 char *prerelax[256];
1245 char *postrelax[256];
1246 char stringF[2] = "F", stringC[2] = "C", stringA[2] = "A";
1247 PetscInt ns_down = 256, ns_up = 256;
1248 PetscBool matchF, matchC, matchA;
1249
1250 jac->interptype = 100; /* no way we can pass this with strings... Set it as default as in MFEM, then users can still customize it back to a different one */
1251
1252 PetscCall(PetscOptionsReal("-pc_hypre_boomeramg_strongthresholdR", "Threshold for R", "None", jac->Rstrongthreshold, &jac->Rstrongthreshold, NULL));
1253 PetscCallHYPRE(HYPRE_BoomerAMGSetStrongThresholdR(jac->hsolver, jac->Rstrongthreshold));
1254
1255 PetscCall(PetscOptionsReal("-pc_hypre_boomeramg_filterthresholdR", "Filter threshold for R", "None", jac->Rfilterthreshold, &jac->Rfilterthreshold, NULL));
1256 PetscCallHYPRE(HYPRE_BoomerAMGSetFilterThresholdR(jac->hsolver, jac->Rfilterthreshold));
1257
1258 PetscCall(PetscOptionsReal("-pc_hypre_boomeramg_Adroptol", "Defines the drop tolerance for the A-matrices from the 2nd level of AMG", "None", jac->Adroptol, &jac->Adroptol, NULL));
1259 PetscCallHYPRE(HYPRE_BoomerAMGSetADropTol(jac->hsolver, (HYPRE_Int)jac->Adroptol));
1260
1261 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_Adroptype", "Drops the entries that are not on the diagonal and smaller than its row norm: type 1: 1-norm, 2: 2-norm, -1: infinity norm", "None", jac->Adroptype, &jac->Adroptype, NULL));
1262 PetscCallHYPRE(HYPRE_BoomerAMGSetADropType(jac->hsolver, (HYPRE_Int)jac->Adroptype));
1263 PetscCall(PetscOptionsStringArray("-pc_hypre_boomeramg_prerelax", "Defines prerelax scheme", "None", prerelax, &ns_down, NULL));
1264 PetscCall(PetscOptionsStringArray("-pc_hypre_boomeramg_postrelax", "Defines postrelax scheme", "None", postrelax, &ns_up, NULL));
1265 PetscCheck(ns_down == jac->gridsweeps[0], PetscObjectComm((PetscObject)jac), PETSC_ERR_ARG_SIZ, "The number of arguments passed to -pc_hypre_boomeramg_prerelax must match the number passed to -pc_hypre_bomeramg_grid_sweeps_down");
1266 PetscCheck(ns_up == jac->gridsweeps[1], PetscObjectComm((PetscObject)jac), PETSC_ERR_ARG_SIZ, "The number of arguments passed to -pc_hypre_boomeramg_postrelax must match the number passed to -pc_hypre_bomeramg_grid_sweeps_up");
1267
1268 grid_relax_points[0] = NULL;
1269 grid_relax_points[1] = hypre_TAlloc(HYPRE_Int, ns_down, HYPRE_MEMORY_HOST);
1270 grid_relax_points[2] = hypre_TAlloc(HYPRE_Int, ns_up, HYPRE_MEMORY_HOST);
1271 grid_relax_points[3] = hypre_TAlloc(HYPRE_Int, jac->gridsweeps[2], HYPRE_MEMORY_HOST);
1272 grid_relax_points[3][0] = 0;
1273
1274 // set down relax scheme
1275 for (PetscInt i = 0; i < ns_down; i++) {
1276 PetscCall(PetscStrcasecmp(prerelax[i], stringF, &matchF));
1277 PetscCall(PetscStrcasecmp(prerelax[i], stringC, &matchC));
1278 PetscCall(PetscStrcasecmp(prerelax[i], stringA, &matchA));
1279 PetscCheck(matchF || matchC || matchA, PetscObjectComm((PetscObject)jac), PETSC_ERR_ARG_WRONG, "Valid argument options for -pc_hypre_boomeramg_prerelax are C, F, and A");
1280 if (matchF) grid_relax_points[1][i] = -1;
1281 else if (matchC) grid_relax_points[1][i] = 1;
1282 else if (matchA) grid_relax_points[1][i] = 0;
1283 }
1284
1285 // set up relax scheme
1286 for (PetscInt i = 0; i < ns_up; i++) {
1287 PetscCall(PetscStrcasecmp(postrelax[i], stringF, &matchF));
1288 PetscCall(PetscStrcasecmp(postrelax[i], stringC, &matchC));
1289 PetscCall(PetscStrcasecmp(postrelax[i], stringA, &matchA));
1290 PetscCheck(matchF || matchC || matchA, PetscObjectComm((PetscObject)jac), PETSC_ERR_ARG_WRONG, "Valid argument options for -pc_hypre_boomeramg_postrelax are C, F, and A");
1291 if (matchF) grid_relax_points[2][i] = -1;
1292 else if (matchC) grid_relax_points[2][i] = 1;
1293 else if (matchA) grid_relax_points[2][i] = 0;
1294 }
1295
1296 // set coarse relax scheme
1297 for (PetscInt i = 0; i < jac->gridsweeps[2]; i++) grid_relax_points[3][i] = 0;
1298
1299 // Pass relax schemes to hypre
1300 PetscCallHYPRE(HYPRE_BoomerAMGSetGridRelaxPoints(jac->hsolver, grid_relax_points));
1301
1302 // cleanup memory
1303 for (PetscInt i = 0; i < ns_down; i++) PetscCall(PetscFree(prerelax[i]));
1304 for (PetscInt i = 0; i < ns_up; i++) PetscCall(PetscFree(postrelax[i]));
1305 }
1306 #endif
1307
1308 #if PETSC_PKG_HYPRE_VERSION_LE(9, 9, 9)
1309 PetscCheck(!jac->Rtype || !jac->agg_nl, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_INCOMP, "-pc_hypre_boomeramg_restriction_type (%" PetscInt_FMT ") and -pc_hypre_boomeramg_agg_nl (%" PetscInt_FMT ")", jac->Rtype, jac->agg_nl);
1310 #endif
1311
1312 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_interp_type", "Interpolation type", "None", HYPREBoomerAMGInterpType, PETSC_STATIC_ARRAY_LENGTH(HYPREBoomerAMGInterpType), jac->interptype < 0 ? "unknown" : HYPREBoomerAMGInterpType[jac->interptype], &indx, &flg));
1313 if (flg) jac->interptype = indx;
1314
1315 PetscCall(PetscOptionsName("-pc_hypre_boomeramg_print_statistics", "Print statistics", "None", &flg));
1316 if (flg) {
1317 level = 3;
1318 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_print_statistics", "Print statistics", "None", level, &level, NULL));
1319
1320 jac->printstatistics = PETSC_TRUE;
1321 PetscCallHYPRE(HYPRE_BoomerAMGSetPrintLevel(jac->hsolver, (HYPRE_Int)level));
1322 }
1323
1324 PetscCall(PetscOptionsName("-pc_hypre_boomeramg_print_debug", "Print debug information", "None", &flg));
1325 if (flg) {
1326 level = 3;
1327 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_print_debug", "Print debug information", "None", level, &level, NULL));
1328
1329 jac->printstatistics = PETSC_TRUE;
1330 PetscCallHYPRE(HYPRE_BoomerAMGSetDebugFlag(jac->hsolver, (HYPRE_Int)level));
1331 }
1332
1333 PetscCall(PetscOptionsBool("-pc_hypre_boomeramg_nodal_relaxation", "Nodal relaxation via Schwarz", "None", PETSC_FALSE, &tmp_truth, &flg));
1334 if (flg && tmp_truth) {
1335 PetscInt tmp_int;
1336 PetscCall(PetscOptionsInt("-pc_hypre_boomeramg_nodal_relaxation", "Nodal relaxation via Schwarz", "None", (HYPRE_Int)jac->nodal_relax_levels, &tmp_int, &flg));
1337 if (flg) jac->nodal_relax_levels = tmp_int;
1338 PetscCallHYPRE(HYPRE_BoomerAMGSetSmoothType(jac->hsolver, 6));
1339 PetscCallHYPRE(HYPRE_BoomerAMGSetDomainType(jac->hsolver, 1));
1340 PetscCallHYPRE(HYPRE_BoomerAMGSetOverlap(jac->hsolver, 0));
1341 PetscCallHYPRE(HYPRE_BoomerAMGSetSmoothNumLevels(jac->hsolver, (HYPRE_Int)jac->nodal_relax_levels));
1342 }
1343
1344 PetscCall(PetscOptionsBool3("-pc_hypre_boomeramg_keeptranspose", "Avoid transpose matvecs in preconditioner application", "None", jac->keeptranspose, &jac->keeptranspose, NULL));
1345
1346 /* options for ParaSails solvers */
1347 PetscCall(PetscOptionsEList("-pc_hypre_boomeramg_parasails_sym", "Symmetry of matrix and preconditioner", "None", symtlist, PETSC_STATIC_ARRAY_LENGTH(symtlist), symtlist[0], &indx, &flg));
1348 if (flg) {
1349 jac->symt = indx;
1350 PetscCallHYPRE(HYPRE_BoomerAMGSetSym(jac->hsolver, (HYPRE_Int)jac->symt));
1351 }
1352
1353 PetscOptionsHeadEnd();
1354 PetscFunctionReturn(PETSC_SUCCESS);
1355 }
1356
PCApplyRichardson_HYPRE_BoomerAMG(PC pc,Vec b,Vec y,Vec w,PetscReal rtol,PetscReal abstol,PetscReal dtol,PetscInt its,PetscBool guesszero,PetscInt * outits,PCRichardsonConvergedReason * reason)1357 static PetscErrorCode PCApplyRichardson_HYPRE_BoomerAMG(PC pc, Vec b, Vec y, Vec w, PetscReal rtol, PetscReal abstol, PetscReal dtol, PetscInt its, PetscBool guesszero, PetscInt *outits, PCRichardsonConvergedReason *reason)
1358 {
1359 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1360 HYPRE_Int oits;
1361
1362 PetscFunctionBegin;
1363 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
1364 PetscCallHYPRE(HYPRE_BoomerAMGSetMaxIter(jac->hsolver, (HYPRE_Int)(its * jac->maxiter)));
1365 PetscCallHYPRE(HYPRE_BoomerAMGSetTol(jac->hsolver, rtol));
1366 jac->applyrichardson = PETSC_TRUE;
1367 PetscCall(PCApply_HYPRE(pc, b, y));
1368 jac->applyrichardson = PETSC_FALSE;
1369 PetscCallHYPRE(HYPRE_BoomerAMGGetNumIterations(jac->hsolver, &oits));
1370 *outits = oits;
1371 if (oits == its) *reason = PCRICHARDSON_CONVERGED_ITS;
1372 else *reason = PCRICHARDSON_CONVERGED_RTOL;
1373 PetscCallHYPRE(HYPRE_BoomerAMGSetTol(jac->hsolver, jac->tol));
1374 PetscCallHYPRE(HYPRE_BoomerAMGSetMaxIter(jac->hsolver, (HYPRE_Int)jac->maxiter));
1375 PetscFunctionReturn(PETSC_SUCCESS);
1376 }
1377
PCView_HYPRE_BoomerAMG(PC pc,PetscViewer viewer)1378 static PetscErrorCode PCView_HYPRE_BoomerAMG(PC pc, PetscViewer viewer)
1379 {
1380 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1381 hypre_ParAMGData *amg_data = (hypre_ParAMGData *)jac->hsolver;
1382 PetscBool isascii;
1383 PetscInt indx;
1384 PetscReal val;
1385
1386 PetscFunctionBegin;
1387 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
1388 if (isascii) {
1389 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE BoomerAMG preconditioning\n"));
1390 PetscCall(PetscViewerASCIIPrintf(viewer, " Cycle type %s\n", HYPREBoomerAMGCycleType[jac->cycletype]));
1391 PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum number of levels %" PetscInt_FMT "\n", jac->maxlevels));
1392 PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum number of iterations PER hypre call %" PetscInt_FMT "\n", jac->maxiter));
1393 PetscCall(PetscViewerASCIIPrintf(viewer, " Convergence tolerance PER hypre call %g\n", (double)jac->tol));
1394 PetscCall(PetscViewerASCIIPrintf(viewer, " Threshold for strong coupling %g\n", (double)jac->strongthreshold));
1395 PetscCall(PetscViewerASCIIPrintf(viewer, " Interpolation truncation factor %g\n", (double)jac->truncfactor));
1396 PetscCall(PetscViewerASCIIPrintf(viewer, " Interpolation: max elements per row %" PetscInt_FMT "\n", jac->pmax));
1397 if (jac->interp_refine) PetscCall(PetscViewerASCIIPrintf(viewer, " Interpolation: number of steps of weighted refinement %" PetscInt_FMT "\n", jac->interp_refine));
1398 PetscCall(PetscViewerASCIIPrintf(viewer, " Number of levels of aggressive coarsening %" PetscInt_FMT "\n", jac->agg_nl));
1399 PetscCall(PetscViewerASCIIPrintf(viewer, " Number of paths for aggressive coarsening %" PetscInt_FMT "\n", jac->agg_num_paths));
1400
1401 PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum row sums %g\n", (double)jac->maxrowsum));
1402
1403 PetscCall(PetscViewerASCIIPrintf(viewer, " Sweeps down %" PetscInt_FMT "\n", jac->gridsweeps[0]));
1404 PetscCall(PetscViewerASCIIPrintf(viewer, " Sweeps up %" PetscInt_FMT "\n", jac->gridsweeps[1]));
1405 PetscCall(PetscViewerASCIIPrintf(viewer, " Sweeps on coarse %" PetscInt_FMT "\n", jac->gridsweeps[2]));
1406
1407 PetscCall(PetscViewerASCIIPrintf(viewer, " Relax down %s\n", jac->relaxtype[0] < 0 ? "not yet set" : HYPREBoomerAMGRelaxType[jac->relaxtype[0]]));
1408 PetscCall(PetscViewerASCIIPrintf(viewer, " Relax up %s\n", jac->relaxtype[1] < 0 ? "not yet set" : HYPREBoomerAMGRelaxType[jac->relaxtype[1]]));
1409 PetscCall(PetscViewerASCIIPrintf(viewer, " Relax on coarse %s\n", HYPREBoomerAMGRelaxType[jac->relaxtype[2]]));
1410
1411 PetscCall(PetscViewerASCIIPrintf(viewer, " Relax weight (all) %g\n", (double)jac->relaxweight));
1412 PetscCall(PetscViewerASCIIPrintf(viewer, " Outer relax weight (all) %g\n", (double)jac->outerrelaxweight));
1413
1414 PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum size of coarsest grid %" PetscInt_FMT "\n", jac->maxc));
1415 PetscCall(PetscViewerASCIIPrintf(viewer, " Minimum size of coarsest grid %" PetscInt_FMT "\n", jac->minc));
1416
1417 if (jac->relaxorder == PETSC_DECIDE) {
1418 PetscCall(PetscViewerASCIIPrintf(viewer, " CF-relaxation option not yet determined\n"));
1419 } else if (jac->relaxorder) {
1420 PetscCall(PetscViewerASCIIPrintf(viewer, " Using CF-relaxation\n"));
1421 } else {
1422 PetscCall(PetscViewerASCIIPrintf(viewer, " Not using CF-relaxation\n"));
1423 }
1424 if (jac->smoothtype != -1) {
1425 PetscCall(PetscViewerASCIIPrintf(viewer, " Smooth type %s\n", HYPREBoomerAMGSmoothType[jac->smoothtype]));
1426 PetscCall(PetscViewerASCIIPrintf(viewer, " Smooth num levels %" PetscInt_FMT "\n", jac->smoothnumlevels));
1427 PetscCall(PetscViewerASCIIPrintf(viewer, " Smooth num sweeps %" PetscInt_FMT "\n", jac->smoothsweeps));
1428 if (jac->smoothtype == 0) {
1429 PetscStackCallExternalVoid("hypre_ParAMGDataILUType", indx = hypre_ParAMGDataILUType(amg_data));
1430 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU type %s (%" PetscInt_FMT ")\n", HYPREILUType[indx], indx));
1431 PetscStackCallExternalVoid("hypre_ParAMGDataILULevel", indx = hypre_ParAMGDataILULevel(amg_data));
1432 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU level %" PetscInt_FMT "\n", indx));
1433 PetscStackCallExternalVoid("hypre_ParAMGDataILUMaxIter", indx = hypre_ParAMGDataILUMaxIter(amg_data));
1434 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU max iterations %" PetscInt_FMT "\n", indx));
1435 PetscStackCallExternalVoid("hypre_ParAMGDataILUMaxRowNnz", indx = hypre_ParAMGDataILUMaxRowNnz(amg_data));
1436 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU max NNZ per row %" PetscInt_FMT "\n", indx));
1437 PetscStackCallExternalVoid("hypre_ParAMGDataILUTriSolve", indx = hypre_ParAMGDataILUTriSolve(amg_data));
1438 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU triangular solve %" PetscInt_FMT "\n", indx));
1439 PetscStackCallExternalVoid("hypre_ParAMGDataTol", val = hypre_ParAMGDataTol(amg_data));
1440 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU tolerance %e\n", val));
1441 PetscStackCallExternalVoid("hypre_ParAMGDataILUDroptol", val = hypre_ParAMGDataILUDroptol(amg_data));
1442 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU drop tolerance %e\n", val));
1443 PetscStackCallExternalVoid("hypre_ParAMGDataILULocalReordering", indx = hypre_ParAMGDataILULocalReordering(amg_data));
1444 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU local reordering %" PetscInt_FMT "\n", indx));
1445 PetscStackCallExternalVoid("hypre_ParAMGDataILULowerJacobiIters", indx = hypre_ParAMGDataILULowerJacobiIters(amg_data));
1446 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU lower Jacobi iterations %" PetscInt_FMT "\n", indx));
1447 PetscStackCallExternalVoid("hypre_ParAMGDataILUUpperJacobiIters", indx = hypre_ParAMGDataILUUpperJacobiIters(amg_data));
1448 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU upper Jacobi iterations %" PetscInt_FMT "\n", indx));
1449 PetscStackCallExternalVoid("hypre_ParAMGDataPrintLevel", indx = hypre_ParAMGDataPrintLevel(amg_data));
1450 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU print level %" PetscInt_FMT "\n", indx));
1451 PetscStackCallExternalVoid("hypre_ParAMGDataLogging", indx = hypre_ParAMGDataLogging(amg_data));
1452 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU logging level %" PetscInt_FMT "\n", indx));
1453 PetscStackCallExternalVoid("hypre_ParAMGDataILUIterSetupType", indx = hypre_ParAMGDataILUIterSetupType(amg_data));
1454 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU iterative setup type %s (%" PetscInt_FMT ")\n", HYPREILUIterSetup[indx], indx));
1455 PetscStackCallExternalVoid("hypre_ParAMGDataILUIterSetupOption", indx = hypre_ParAMGDataILUIterSetupOption(amg_data));
1456 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU iterative setup option %" PetscInt_FMT "\n", indx));
1457 PetscStackCallExternalVoid("hypre_ParAMGDataILUIterSetupMaxIter", indx = hypre_ParAMGDataILUIterSetupMaxIter(amg_data));
1458 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU iterative setup max iterations %" PetscInt_FMT "\n", indx));
1459 PetscStackCallExternalVoid("hypre_ParAMGDataILUIterSetupTolerance", val = hypre_ParAMGDataILUIterSetupTolerance(amg_data));
1460 PetscCall(PetscViewerASCIIPrintf(viewer, " ILU iterative setup tolerance %e\n", val));
1461 }
1462 } else {
1463 PetscCall(PetscViewerASCIIPrintf(viewer, " Not using more complex smoothers.\n"));
1464 }
1465 if (jac->smoothtype == 3) {
1466 PetscCall(PetscViewerASCIIPrintf(viewer, " Euclid ILU(k) levels %" PetscInt_FMT "\n", jac->eu_level));
1467 PetscCall(PetscViewerASCIIPrintf(viewer, " Euclid ILU(k) drop tolerance %g\n", (double)jac->eu_droptolerance));
1468 PetscCall(PetscViewerASCIIPrintf(viewer, " Euclid ILU use Block-Jacobi? %" PetscInt_FMT "\n", jac->eu_bj));
1469 }
1470 PetscCall(PetscViewerASCIIPrintf(viewer, " Measure type %s\n", HYPREBoomerAMGMeasureType[jac->measuretype]));
1471 PetscCall(PetscViewerASCIIPrintf(viewer, " Coarsen type %s\n", jac->coarsentype < 0 ? "not yet set" : HYPREBoomerAMGCoarsenType[jac->coarsentype]));
1472 PetscCall(PetscViewerASCIIPrintf(viewer, " Interpolation type %s\n", jac->interptype != 100 ? (jac->interptype < 0 ? "not yet set" : HYPREBoomerAMGInterpType[jac->interptype]) : "1pt"));
1473 if (jac->nodal_coarsening) PetscCall(PetscViewerASCIIPrintf(viewer, " Using nodal coarsening with HYPRE_BOOMERAMGSetNodal() %" PetscInt_FMT "\n", jac->nodal_coarsening));
1474 if (jac->vec_interp_variant) {
1475 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE_BoomerAMGSetInterpVecVariant() %" PetscInt_FMT "\n", jac->vec_interp_variant));
1476 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE_BoomerAMGSetInterpVecQMax() %" PetscInt_FMT "\n", jac->vec_interp_qmax));
1477 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE_BoomerAMGSetSmoothInterpVectors() %d\n", jac->vec_interp_smooth));
1478 }
1479 if (jac->nodal_relax) PetscCall(PetscViewerASCIIPrintf(viewer, " Using nodal relaxation via Schwarz smoothing on levels %" PetscInt_FMT "\n", jac->nodal_relax_levels));
1480 #if PETSC_PKG_HYPRE_VERSION_GE(2, 23, 0)
1481 PetscCall(PetscViewerASCIIPrintf(viewer, " SpGEMM type %s\n", jac->spgemm_type));
1482 #else
1483 PetscCall(PetscViewerASCIIPrintf(viewer, " SpGEMM type %s\n", "hypre"));
1484 #endif
1485 /* AIR */
1486 if (jac->Rtype) {
1487 PetscCall(PetscViewerASCIIPrintf(viewer, " Using approximate ideal restriction type %" PetscInt_FMT "\n", jac->Rtype));
1488 PetscCall(PetscViewerASCIIPrintf(viewer, " Threshold for R %g\n", (double)jac->Rstrongthreshold));
1489 PetscCall(PetscViewerASCIIPrintf(viewer, " Filter for R %g\n", (double)jac->Rfilterthreshold));
1490 PetscCall(PetscViewerASCIIPrintf(viewer, " A drop tolerance %g\n", (double)jac->Adroptol));
1491 PetscCall(PetscViewerASCIIPrintf(viewer, " A drop type %" PetscInt_FMT "\n", jac->Adroptype));
1492 }
1493 }
1494 PetscFunctionReturn(PETSC_SUCCESS);
1495 }
1496
PCSetFromOptions_HYPRE_ParaSails(PC pc,PetscOptionItems PetscOptionsObject)1497 static PetscErrorCode PCSetFromOptions_HYPRE_ParaSails(PC pc, PetscOptionItems PetscOptionsObject)
1498 {
1499 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1500 PetscInt indx;
1501 PetscBool flag;
1502 const char *symtlist[] = {"nonsymmetric", "SPD", "nonsymmetric,SPD"};
1503
1504 PetscFunctionBegin;
1505 PetscOptionsHeadBegin(PetscOptionsObject, "HYPRE ParaSails Options");
1506 PetscCall(PetscOptionsInt("-pc_hypre_parasails_nlevels", "Number of number of levels", "None", jac->nlevels, &jac->nlevels, 0));
1507 PetscCall(PetscOptionsReal("-pc_hypre_parasails_thresh", "Threshold", "None", jac->threshold, &jac->threshold, &flag));
1508 if (flag) PetscCallHYPRE(HYPRE_ParaSailsSetParams(jac->hsolver, jac->threshold, (HYPRE_Int)jac->nlevels));
1509
1510 PetscCall(PetscOptionsReal("-pc_hypre_parasails_filter", "filter", "None", jac->filter, &jac->filter, &flag));
1511 if (flag) PetscCallHYPRE(HYPRE_ParaSailsSetFilter(jac->hsolver, jac->filter));
1512
1513 PetscCall(PetscOptionsReal("-pc_hypre_parasails_loadbal", "Load balance", "None", jac->loadbal, &jac->loadbal, &flag));
1514 if (flag) PetscCallHYPRE(HYPRE_ParaSailsSetLoadbal(jac->hsolver, (HYPRE_Int)jac->loadbal));
1515
1516 PetscCall(PetscOptionsBool("-pc_hypre_parasails_logging", "Print info to screen", "None", (PetscBool)jac->logging, (PetscBool *)&jac->logging, &flag));
1517 if (flag) PetscCallHYPRE(HYPRE_ParaSailsSetLogging(jac->hsolver, (HYPRE_Int)jac->logging));
1518
1519 PetscCall(PetscOptionsBool("-pc_hypre_parasails_reuse", "Reuse nonzero pattern in preconditioner", "None", (PetscBool)jac->ruse, (PetscBool *)&jac->ruse, &flag));
1520 if (flag) PetscCallHYPRE(HYPRE_ParaSailsSetReuse(jac->hsolver, (HYPRE_Int)jac->ruse));
1521
1522 PetscCall(PetscOptionsEList("-pc_hypre_parasails_sym", "Symmetry of matrix and preconditioner", "None", symtlist, PETSC_STATIC_ARRAY_LENGTH(symtlist), symtlist[0], &indx, &flag));
1523 if (flag) {
1524 jac->symt = indx;
1525 PetscCallHYPRE(HYPRE_ParaSailsSetSym(jac->hsolver, (HYPRE_Int)jac->symt));
1526 }
1527
1528 PetscOptionsHeadEnd();
1529 PetscFunctionReturn(PETSC_SUCCESS);
1530 }
1531
PCView_HYPRE_ParaSails(PC pc,PetscViewer viewer)1532 static PetscErrorCode PCView_HYPRE_ParaSails(PC pc, PetscViewer viewer)
1533 {
1534 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1535 PetscBool isascii;
1536 const char *symt = 0;
1537
1538 PetscFunctionBegin;
1539 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
1540 if (isascii) {
1541 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE ParaSails preconditioning\n"));
1542 PetscCall(PetscViewerASCIIPrintf(viewer, " nlevels %" PetscInt_FMT "\n", jac->nlevels));
1543 PetscCall(PetscViewerASCIIPrintf(viewer, " threshold %g\n", (double)jac->threshold));
1544 PetscCall(PetscViewerASCIIPrintf(viewer, " filter %g\n", (double)jac->filter));
1545 PetscCall(PetscViewerASCIIPrintf(viewer, " load balance %g\n", (double)jac->loadbal));
1546 PetscCall(PetscViewerASCIIPrintf(viewer, " reuse nonzero structure %s\n", PetscBools[jac->ruse]));
1547 PetscCall(PetscViewerASCIIPrintf(viewer, " print info to screen %s\n", PetscBools[jac->logging]));
1548 if (!jac->symt) symt = "nonsymmetric matrix and preconditioner";
1549 else if (jac->symt == 1) symt = "SPD matrix and preconditioner";
1550 else if (jac->symt == 2) symt = "nonsymmetric matrix but SPD preconditioner";
1551 else SETERRQ(PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONG, "Unknown HYPRE ParaSails symmetric option %" PetscInt_FMT, jac->symt);
1552 PetscCall(PetscViewerASCIIPrintf(viewer, " %s\n", symt));
1553 }
1554 PetscFunctionReturn(PETSC_SUCCESS);
1555 }
1556
PCSetFromOptions_HYPRE_AMS(PC pc,PetscOptionItems PetscOptionsObject)1557 static PetscErrorCode PCSetFromOptions_HYPRE_AMS(PC pc, PetscOptionItems PetscOptionsObject)
1558 {
1559 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1560 PetscInt n;
1561 PetscBool flag, flag2, flag3, flag4;
1562
1563 PetscFunctionBegin;
1564 PetscOptionsHeadBegin(PetscOptionsObject, "HYPRE AMS Options");
1565 PetscCall(PetscOptionsInt("-pc_hypre_ams_print_level", "Debugging output level for AMS", "None", jac->as_print, &jac->as_print, &flag));
1566 if (flag) PetscCallHYPRE(HYPRE_AMSSetPrintLevel(jac->hsolver, (HYPRE_Int)jac->as_print));
1567 PetscCall(PetscOptionsInt("-pc_hypre_ams_max_iter", "Maximum number of AMS multigrid iterations within PCApply", "None", jac->as_max_iter, &jac->as_max_iter, &flag));
1568 if (flag) PetscCallHYPRE(HYPRE_AMSSetMaxIter(jac->hsolver, (HYPRE_Int)jac->as_max_iter));
1569 PetscCall(PetscOptionsInt("-pc_hypre_ams_cycle_type", "Cycle type for AMS multigrid", "None", jac->ams_cycle_type, &jac->ams_cycle_type, &flag));
1570 if (flag) PetscCallHYPRE(HYPRE_AMSSetCycleType(jac->hsolver, (HYPRE_Int)jac->ams_cycle_type));
1571 PetscCall(PetscOptionsReal("-pc_hypre_ams_tol", "Error tolerance for AMS multigrid", "None", jac->as_tol, &jac->as_tol, &flag));
1572 if (flag) PetscCallHYPRE(HYPRE_AMSSetTol(jac->hsolver, jac->as_tol));
1573 PetscCall(PetscOptionsInt("-pc_hypre_ams_relax_type", "Relaxation type for AMS smoother", "None", jac->as_relax_type, &jac->as_relax_type, &flag));
1574 PetscCall(PetscOptionsInt("-pc_hypre_ams_relax_times", "Number of relaxation steps for AMS smoother", "None", jac->as_relax_times, &jac->as_relax_times, &flag2));
1575 PetscCall(PetscOptionsReal("-pc_hypre_ams_relax_weight", "Relaxation weight for AMS smoother", "None", jac->as_relax_weight, &jac->as_relax_weight, &flag3));
1576 PetscCall(PetscOptionsReal("-pc_hypre_ams_omega", "SSOR coefficient for AMS smoother", "None", jac->as_omega, &jac->as_omega, &flag4));
1577 if (flag || flag2 || flag3 || flag4) PetscCallHYPRE(HYPRE_AMSSetSmoothingOptions(jac->hsolver, (HYPRE_Int)jac->as_relax_type, (HYPRE_Int)jac->as_relax_times, jac->as_relax_weight, jac->as_omega));
1578 PetscCall(PetscOptionsReal("-pc_hypre_ams_amg_alpha_theta", "Threshold for strong coupling of vector Poisson AMG solver", "None", jac->as_amg_alpha_theta, &jac->as_amg_alpha_theta, &flag));
1579 n = 5;
1580 PetscCall(PetscOptionsIntArray("-pc_hypre_ams_amg_alpha_options", "AMG options for vector Poisson", "None", jac->as_amg_alpha_opts, &n, &flag2));
1581 if (flag || flag2) {
1582 PetscCallHYPRE(HYPRE_AMSSetAlphaAMGOptions(jac->hsolver, (HYPRE_Int)jac->as_amg_alpha_opts[0], /* AMG coarsen type */
1583 (HYPRE_Int)jac->as_amg_alpha_opts[1], /* AMG agg_levels */
1584 (HYPRE_Int)jac->as_amg_alpha_opts[2], /* AMG relax_type */
1585 jac->as_amg_alpha_theta, (HYPRE_Int)jac->as_amg_alpha_opts[3], /* AMG interp_type */
1586 (HYPRE_Int)jac->as_amg_alpha_opts[4])); /* AMG Pmax */
1587 }
1588 PetscCall(PetscOptionsReal("-pc_hypre_ams_amg_beta_theta", "Threshold for strong coupling of scalar Poisson AMG solver", "None", jac->as_amg_beta_theta, &jac->as_amg_beta_theta, &flag));
1589 n = 5;
1590 PetscCall(PetscOptionsIntArray("-pc_hypre_ams_amg_beta_options", "AMG options for scalar Poisson solver", "None", jac->as_amg_beta_opts, &n, &flag2));
1591 if (flag || flag2) {
1592 PetscCallHYPRE(HYPRE_AMSSetBetaAMGOptions(jac->hsolver, (HYPRE_Int)jac->as_amg_beta_opts[0], /* AMG coarsen type */
1593 (HYPRE_Int)jac->as_amg_beta_opts[1], /* AMG agg_levels */
1594 (HYPRE_Int)jac->as_amg_beta_opts[2], /* AMG relax_type */
1595 jac->as_amg_beta_theta, (HYPRE_Int)jac->as_amg_beta_opts[3], /* AMG interp_type */
1596 (HYPRE_Int)jac->as_amg_beta_opts[4])); /* AMG Pmax */
1597 }
1598 PetscCall(PetscOptionsInt("-pc_hypre_ams_projection_frequency", "Frequency at which a projection onto the compatible subspace for problems with zero conductivity regions is performed", "None", jac->ams_proj_freq, &jac->ams_proj_freq, &flag));
1599 if (flag) { /* override HYPRE's default only if the options is used */
1600 PetscCallHYPRE(HYPRE_AMSSetProjectionFrequency(jac->hsolver, (HYPRE_Int)jac->ams_proj_freq));
1601 }
1602 PetscOptionsHeadEnd();
1603 PetscFunctionReturn(PETSC_SUCCESS);
1604 }
1605
PCView_HYPRE_AMS(PC pc,PetscViewer viewer)1606 static PetscErrorCode PCView_HYPRE_AMS(PC pc, PetscViewer viewer)
1607 {
1608 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1609 PetscBool isascii;
1610
1611 PetscFunctionBegin;
1612 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
1613 if (isascii) {
1614 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE AMS preconditioning\n"));
1615 PetscCall(PetscViewerASCIIPrintf(viewer, " subspace iterations per application %" PetscInt_FMT "\n", jac->as_max_iter));
1616 PetscCall(PetscViewerASCIIPrintf(viewer, " subspace cycle type %" PetscInt_FMT "\n", jac->ams_cycle_type));
1617 PetscCall(PetscViewerASCIIPrintf(viewer, " subspace iteration tolerance %g\n", (double)jac->as_tol));
1618 PetscCall(PetscViewerASCIIPrintf(viewer, " smoother type %" PetscInt_FMT "\n", jac->as_relax_type));
1619 PetscCall(PetscViewerASCIIPrintf(viewer, " number of smoothing steps %" PetscInt_FMT "\n", jac->as_relax_times));
1620 PetscCall(PetscViewerASCIIPrintf(viewer, " smoother weight %g\n", (double)jac->as_relax_weight));
1621 PetscCall(PetscViewerASCIIPrintf(viewer, " smoother omega %g\n", (double)jac->as_omega));
1622 if (jac->alpha_Poisson) {
1623 PetscCall(PetscViewerASCIIPrintf(viewer, " vector Poisson solver (passed in by user)\n"));
1624 } else {
1625 PetscCall(PetscViewerASCIIPrintf(viewer, " vector Poisson solver (computed) \n"));
1626 }
1627 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG coarsening type %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[0]));
1628 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG levels of aggressive coarsening %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[1]));
1629 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG relaxation type %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[2]));
1630 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG interpolation type %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[3]));
1631 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG max nonzero elements in interpolation rows %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[4]));
1632 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG strength threshold %g\n", (double)jac->as_amg_alpha_theta));
1633 if (!jac->ams_beta_is_zero) {
1634 if (jac->beta_Poisson) {
1635 PetscCall(PetscViewerASCIIPrintf(viewer, " scalar Poisson solver (passed in by user)\n"));
1636 } else {
1637 PetscCall(PetscViewerASCIIPrintf(viewer, " scalar Poisson solver (computed) \n"));
1638 }
1639 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG coarsening type %" PetscInt_FMT "\n", jac->as_amg_beta_opts[0]));
1640 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG levels of aggressive coarsening %" PetscInt_FMT "\n", jac->as_amg_beta_opts[1]));
1641 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG relaxation type %" PetscInt_FMT "\n", jac->as_amg_beta_opts[2]));
1642 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG interpolation type %" PetscInt_FMT "\n", jac->as_amg_beta_opts[3]));
1643 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG max nonzero elements in interpolation rows %" PetscInt_FMT "\n", jac->as_amg_beta_opts[4]));
1644 PetscCall(PetscViewerASCIIPrintf(viewer, " boomerAMG strength threshold %g\n", (double)jac->as_amg_beta_theta));
1645 if (jac->ams_beta_is_zero_part) PetscCall(PetscViewerASCIIPrintf(viewer, " compatible subspace projection frequency %" PetscInt_FMT " (-1 HYPRE uses default)\n", jac->ams_proj_freq));
1646 } else {
1647 PetscCall(PetscViewerASCIIPrintf(viewer, " scalar Poisson solver not used (zero-conductivity everywhere) \n"));
1648 }
1649 }
1650 PetscFunctionReturn(PETSC_SUCCESS);
1651 }
1652
PCSetFromOptions_HYPRE_ADS(PC pc,PetscOptionItems PetscOptionsObject)1653 static PetscErrorCode PCSetFromOptions_HYPRE_ADS(PC pc, PetscOptionItems PetscOptionsObject)
1654 {
1655 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1656 PetscInt n;
1657 PetscBool flag, flag2, flag3, flag4;
1658
1659 PetscFunctionBegin;
1660 PetscOptionsHeadBegin(PetscOptionsObject, "HYPRE ADS Options");
1661 PetscCall(PetscOptionsInt("-pc_hypre_ads_print_level", "Debugging output level for ADS", "None", jac->as_print, &jac->as_print, &flag));
1662 if (flag) PetscCallHYPRE(HYPRE_ADSSetPrintLevel(jac->hsolver, (HYPRE_Int)jac->as_print));
1663 PetscCall(PetscOptionsInt("-pc_hypre_ads_max_iter", "Maximum number of ADS multigrid iterations within PCApply", "None", jac->as_max_iter, &jac->as_max_iter, &flag));
1664 if (flag) PetscCallHYPRE(HYPRE_ADSSetMaxIter(jac->hsolver, (HYPRE_Int)jac->as_max_iter));
1665 PetscCall(PetscOptionsInt("-pc_hypre_ads_cycle_type", "Cycle type for ADS multigrid", "None", jac->ads_cycle_type, &jac->ads_cycle_type, &flag));
1666 if (flag) PetscCallHYPRE(HYPRE_ADSSetCycleType(jac->hsolver, (HYPRE_Int)jac->ads_cycle_type));
1667 PetscCall(PetscOptionsReal("-pc_hypre_ads_tol", "Error tolerance for ADS multigrid", "None", jac->as_tol, &jac->as_tol, &flag));
1668 if (flag) PetscCallHYPRE(HYPRE_ADSSetTol(jac->hsolver, jac->as_tol));
1669 PetscCall(PetscOptionsInt("-pc_hypre_ads_relax_type", "Relaxation type for ADS smoother", "None", jac->as_relax_type, &jac->as_relax_type, &flag));
1670 PetscCall(PetscOptionsInt("-pc_hypre_ads_relax_times", "Number of relaxation steps for ADS smoother", "None", jac->as_relax_times, &jac->as_relax_times, &flag2));
1671 PetscCall(PetscOptionsReal("-pc_hypre_ads_relax_weight", "Relaxation weight for ADS smoother", "None", jac->as_relax_weight, &jac->as_relax_weight, &flag3));
1672 PetscCall(PetscOptionsReal("-pc_hypre_ads_omega", "SSOR coefficient for ADS smoother", "None", jac->as_omega, &jac->as_omega, &flag4));
1673 if (flag || flag2 || flag3 || flag4) PetscCallHYPRE(HYPRE_ADSSetSmoothingOptions(jac->hsolver, (HYPRE_Int)jac->as_relax_type, (HYPRE_Int)jac->as_relax_times, jac->as_relax_weight, jac->as_omega));
1674 PetscCall(PetscOptionsReal("-pc_hypre_ads_ams_theta", "Threshold for strong coupling of AMS solver inside ADS", "None", jac->as_amg_alpha_theta, &jac->as_amg_alpha_theta, &flag));
1675 n = 5;
1676 PetscCall(PetscOptionsIntArray("-pc_hypre_ads_ams_options", "AMG options for AMS solver inside ADS", "None", jac->as_amg_alpha_opts, &n, &flag2));
1677 PetscCall(PetscOptionsInt("-pc_hypre_ads_ams_cycle_type", "Cycle type for AMS solver inside ADS", "None", jac->ams_cycle_type, &jac->ams_cycle_type, &flag3));
1678 if (flag || flag2 || flag3) {
1679 PetscCallHYPRE(HYPRE_ADSSetAMSOptions(jac->hsolver, (HYPRE_Int)jac->ams_cycle_type, /* AMS cycle type */
1680 (HYPRE_Int)jac->as_amg_alpha_opts[0], /* AMG coarsen type */
1681 (HYPRE_Int)jac->as_amg_alpha_opts[1], /* AMG agg_levels */
1682 (HYPRE_Int)jac->as_amg_alpha_opts[2], /* AMG relax_type */
1683 jac->as_amg_alpha_theta, (HYPRE_Int)jac->as_amg_alpha_opts[3], /* AMG interp_type */
1684 (HYPRE_Int)jac->as_amg_alpha_opts[4])); /* AMG Pmax */
1685 }
1686 PetscCall(PetscOptionsReal("-pc_hypre_ads_amg_theta", "Threshold for strong coupling of vector AMG solver inside ADS", "None", jac->as_amg_beta_theta, &jac->as_amg_beta_theta, &flag));
1687 n = 5;
1688 PetscCall(PetscOptionsIntArray("-pc_hypre_ads_amg_options", "AMG options for vector AMG solver inside ADS", "None", jac->as_amg_beta_opts, &n, &flag2));
1689 if (flag || flag2) {
1690 PetscCallHYPRE(HYPRE_ADSSetAMGOptions(jac->hsolver, (HYPRE_Int)jac->as_amg_beta_opts[0], /* AMG coarsen type */
1691 (HYPRE_Int)jac->as_amg_beta_opts[1], /* AMG agg_levels */
1692 (HYPRE_Int)jac->as_amg_beta_opts[2], /* AMG relax_type */
1693 jac->as_amg_beta_theta, (HYPRE_Int)jac->as_amg_beta_opts[3], /* AMG interp_type */
1694 (HYPRE_Int)jac->as_amg_beta_opts[4])); /* AMG Pmax */
1695 }
1696 PetscOptionsHeadEnd();
1697 PetscFunctionReturn(PETSC_SUCCESS);
1698 }
1699
PCView_HYPRE_ADS(PC pc,PetscViewer viewer)1700 static PetscErrorCode PCView_HYPRE_ADS(PC pc, PetscViewer viewer)
1701 {
1702 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1703 PetscBool isascii;
1704
1705 PetscFunctionBegin;
1706 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
1707 if (isascii) {
1708 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE ADS preconditioning\n"));
1709 PetscCall(PetscViewerASCIIPrintf(viewer, " subspace iterations per application %" PetscInt_FMT "\n", jac->as_max_iter));
1710 PetscCall(PetscViewerASCIIPrintf(viewer, " subspace cycle type %" PetscInt_FMT "\n", jac->ads_cycle_type));
1711 PetscCall(PetscViewerASCIIPrintf(viewer, " subspace iteration tolerance %g\n", (double)jac->as_tol));
1712 PetscCall(PetscViewerASCIIPrintf(viewer, " smoother type %" PetscInt_FMT "\n", jac->as_relax_type));
1713 PetscCall(PetscViewerASCIIPrintf(viewer, " number of smoothing steps %" PetscInt_FMT "\n", jac->as_relax_times));
1714 PetscCall(PetscViewerASCIIPrintf(viewer, " smoother weight %g\n", (double)jac->as_relax_weight));
1715 PetscCall(PetscViewerASCIIPrintf(viewer, " smoother omega %g\n", (double)jac->as_omega));
1716 PetscCall(PetscViewerASCIIPrintf(viewer, " AMS solver using boomerAMG\n"));
1717 PetscCall(PetscViewerASCIIPrintf(viewer, " subspace cycle type %" PetscInt_FMT "\n", jac->ams_cycle_type));
1718 PetscCall(PetscViewerASCIIPrintf(viewer, " coarsening type %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[0]));
1719 PetscCall(PetscViewerASCIIPrintf(viewer, " levels of aggressive coarsening %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[1]));
1720 PetscCall(PetscViewerASCIIPrintf(viewer, " relaxation type %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[2]));
1721 PetscCall(PetscViewerASCIIPrintf(viewer, " interpolation type %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[3]));
1722 PetscCall(PetscViewerASCIIPrintf(viewer, " max nonzero elements in interpolation rows %" PetscInt_FMT "\n", jac->as_amg_alpha_opts[4]));
1723 PetscCall(PetscViewerASCIIPrintf(viewer, " strength threshold %g\n", (double)jac->as_amg_alpha_theta));
1724 PetscCall(PetscViewerASCIIPrintf(viewer, " vector Poisson solver using boomerAMG\n"));
1725 PetscCall(PetscViewerASCIIPrintf(viewer, " coarsening type %" PetscInt_FMT "\n", jac->as_amg_beta_opts[0]));
1726 PetscCall(PetscViewerASCIIPrintf(viewer, " levels of aggressive coarsening %" PetscInt_FMT "\n", jac->as_amg_beta_opts[1]));
1727 PetscCall(PetscViewerASCIIPrintf(viewer, " relaxation type %" PetscInt_FMT "\n", jac->as_amg_beta_opts[2]));
1728 PetscCall(PetscViewerASCIIPrintf(viewer, " interpolation type %" PetscInt_FMT "\n", jac->as_amg_beta_opts[3]));
1729 PetscCall(PetscViewerASCIIPrintf(viewer, " max nonzero elements in interpolation rows %" PetscInt_FMT "\n", jac->as_amg_beta_opts[4]));
1730 PetscCall(PetscViewerASCIIPrintf(viewer, " strength threshold %g\n", (double)jac->as_amg_beta_theta));
1731 }
1732 PetscFunctionReturn(PETSC_SUCCESS);
1733 }
1734
PCHYPRESetDiscreteGradient_HYPRE(PC pc,Mat G)1735 static PetscErrorCode PCHYPRESetDiscreteGradient_HYPRE(PC pc, Mat G)
1736 {
1737 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1738 PetscBool ishypre;
1739
1740 PetscFunctionBegin;
1741 PetscCall(PetscObjectTypeCompare((PetscObject)G, MATHYPRE, &ishypre));
1742 if (ishypre) {
1743 PetscCall(PetscObjectReference((PetscObject)G));
1744 PetscCall(MatDestroy(&jac->G));
1745 jac->G = G;
1746 } else {
1747 PetscCall(MatDestroy(&jac->G));
1748 PetscCall(MatConvert(G, MATHYPRE, MAT_INITIAL_MATRIX, &jac->G));
1749 }
1750 PetscFunctionReturn(PETSC_SUCCESS);
1751 }
1752
1753 /*@
1754 PCHYPRESetDiscreteGradient - Set the discrete gradient matrix for `PCHYPRE` type of AMS or ADS
1755
1756 Collective
1757
1758 Input Parameters:
1759 + pc - the preconditioning context
1760 - G - the discrete gradient
1761
1762 Level: intermediate
1763
1764 Notes:
1765 `G` should have as many rows as the number of edges and as many columns as the number of vertices in the mesh
1766
1767 Each row of `G` has 2 nonzeros, with column indexes being the global indexes of edge's endpoints: matrix entries are +1 and -1 depending on edge orientation
1768
1769 Developer Note:
1770 This automatically converts the matrix to `MATHYPRE` if it is not already of that type
1771
1772 .seealso: [](ch_ksp), `PCHYPRE`, `PCHYPRESetDiscreteCurl()`
1773 @*/
PCHYPRESetDiscreteGradient(PC pc,Mat G)1774 PetscErrorCode PCHYPRESetDiscreteGradient(PC pc, Mat G)
1775 {
1776 PetscFunctionBegin;
1777 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
1778 PetscValidHeaderSpecific(G, MAT_CLASSID, 2);
1779 PetscCheckSameComm(pc, 1, G, 2);
1780 PetscTryMethod(pc, "PCHYPRESetDiscreteGradient_C", (PC, Mat), (pc, G));
1781 PetscFunctionReturn(PETSC_SUCCESS);
1782 }
1783
PCHYPRESetDiscreteCurl_HYPRE(PC pc,Mat C)1784 static PetscErrorCode PCHYPRESetDiscreteCurl_HYPRE(PC pc, Mat C)
1785 {
1786 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1787 PetscBool ishypre;
1788
1789 PetscFunctionBegin;
1790 PetscCall(PetscObjectTypeCompare((PetscObject)C, MATHYPRE, &ishypre));
1791 if (ishypre) {
1792 PetscCall(PetscObjectReference((PetscObject)C));
1793 PetscCall(MatDestroy(&jac->C));
1794 jac->C = C;
1795 } else {
1796 PetscCall(MatDestroy(&jac->C));
1797 PetscCall(MatConvert(C, MATHYPRE, MAT_INITIAL_MATRIX, &jac->C));
1798 }
1799 PetscFunctionReturn(PETSC_SUCCESS);
1800 }
1801
1802 /*@
1803 PCHYPRESetDiscreteCurl - Set the discrete curl matrix for `PCHYPRE` type of ADS
1804
1805 Collective
1806
1807 Input Parameters:
1808 + pc - the preconditioning context
1809 - C - the discrete curl
1810
1811 Level: intermediate
1812
1813 Notes:
1814 `C` should have as many rows as the number of faces and as many columns as the number of edges in the mesh
1815
1816 Each row of `C` has as many nonzeros as the number of edges of a face, with column indexes being the global indexes of the corresponding edge.
1817 Matrix entries are +1 and -1 depending on edge orientation with respect to the face orientation
1818
1819 Developer Notes:
1820 This automatically converts the matrix to `MATHYPRE` if it is not already of that type
1821
1822 If this is only for `PCHYPRE` type of ADS it should be called `PCHYPREADSSetDiscreteCurl()`
1823
1824 .seealso: [](ch_ksp), `PCHYPRE`, `PCHYPRESetDiscreteGradient()`
1825 @*/
PCHYPRESetDiscreteCurl(PC pc,Mat C)1826 PetscErrorCode PCHYPRESetDiscreteCurl(PC pc, Mat C)
1827 {
1828 PetscFunctionBegin;
1829 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
1830 PetscValidHeaderSpecific(C, MAT_CLASSID, 2);
1831 PetscCheckSameComm(pc, 1, C, 2);
1832 PetscTryMethod(pc, "PCHYPRESetDiscreteCurl_C", (PC, Mat), (pc, C));
1833 PetscFunctionReturn(PETSC_SUCCESS);
1834 }
1835
PCHYPRESetInterpolations_HYPRE(PC pc,PetscInt dim,Mat RT_PiFull,Mat RT_Pi[],Mat ND_PiFull,Mat ND_Pi[])1836 static PetscErrorCode PCHYPRESetInterpolations_HYPRE(PC pc, PetscInt dim, Mat RT_PiFull, Mat RT_Pi[], Mat ND_PiFull, Mat ND_Pi[])
1837 {
1838 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1839 PetscBool ishypre;
1840 PetscInt i;
1841
1842 PetscFunctionBegin;
1843 PetscCall(MatDestroy(&jac->RT_PiFull));
1844 PetscCall(MatDestroy(&jac->ND_PiFull));
1845 for (i = 0; i < 3; ++i) {
1846 PetscCall(MatDestroy(&jac->RT_Pi[i]));
1847 PetscCall(MatDestroy(&jac->ND_Pi[i]));
1848 }
1849
1850 jac->dim = dim;
1851 if (RT_PiFull) {
1852 PetscCall(PetscObjectTypeCompare((PetscObject)RT_PiFull, MATHYPRE, &ishypre));
1853 if (ishypre) {
1854 PetscCall(PetscObjectReference((PetscObject)RT_PiFull));
1855 jac->RT_PiFull = RT_PiFull;
1856 } else {
1857 PetscCall(MatConvert(RT_PiFull, MATHYPRE, MAT_INITIAL_MATRIX, &jac->RT_PiFull));
1858 }
1859 }
1860 if (RT_Pi) {
1861 for (i = 0; i < dim; ++i) {
1862 if (RT_Pi[i]) {
1863 PetscCall(PetscObjectTypeCompare((PetscObject)RT_Pi[i], MATHYPRE, &ishypre));
1864 if (ishypre) {
1865 PetscCall(PetscObjectReference((PetscObject)RT_Pi[i]));
1866 jac->RT_Pi[i] = RT_Pi[i];
1867 } else {
1868 PetscCall(MatConvert(RT_Pi[i], MATHYPRE, MAT_INITIAL_MATRIX, &jac->RT_Pi[i]));
1869 }
1870 }
1871 }
1872 }
1873 if (ND_PiFull) {
1874 PetscCall(PetscObjectTypeCompare((PetscObject)ND_PiFull, MATHYPRE, &ishypre));
1875 if (ishypre) {
1876 PetscCall(PetscObjectReference((PetscObject)ND_PiFull));
1877 jac->ND_PiFull = ND_PiFull;
1878 } else {
1879 PetscCall(MatConvert(ND_PiFull, MATHYPRE, MAT_INITIAL_MATRIX, &jac->ND_PiFull));
1880 }
1881 }
1882 if (ND_Pi) {
1883 for (i = 0; i < dim; ++i) {
1884 if (ND_Pi[i]) {
1885 PetscCall(PetscObjectTypeCompare((PetscObject)ND_Pi[i], MATHYPRE, &ishypre));
1886 if (ishypre) {
1887 PetscCall(PetscObjectReference((PetscObject)ND_Pi[i]));
1888 jac->ND_Pi[i] = ND_Pi[i];
1889 } else {
1890 PetscCall(MatConvert(ND_Pi[i], MATHYPRE, MAT_INITIAL_MATRIX, &jac->ND_Pi[i]));
1891 }
1892 }
1893 }
1894 }
1895 PetscFunctionReturn(PETSC_SUCCESS);
1896 }
1897
1898 /*@
1899 PCHYPRESetInterpolations - Set the interpolation matrices for `PCHYPRE` type of AMS or ADS
1900
1901 Collective
1902
1903 Input Parameters:
1904 + pc - the preconditioning context
1905 . dim - the dimension of the problem, only used in AMS
1906 . RT_PiFull - Raviart-Thomas interpolation matrix
1907 . RT_Pi - x/y/z component of Raviart-Thomas interpolation matrix
1908 . ND_PiFull - Nedelec interpolation matrix
1909 - ND_Pi - x/y/z component of Nedelec interpolation matrix
1910
1911 Level: intermediate
1912
1913 Notes:
1914 For AMS, only Nedelec interpolation matrices are needed, the Raviart-Thomas interpolation matrices can be set to `NULL`.
1915
1916 For ADS, both type of interpolation matrices are needed.
1917
1918 Developer Note:
1919 This automatically converts the matrix to `MATHYPRE` if it is not already of that type
1920
1921 .seealso: [](ch_ksp), `PCHYPRE`
1922 @*/
PCHYPRESetInterpolations(PC pc,PetscInt dim,Mat RT_PiFull,Mat RT_Pi[],Mat ND_PiFull,Mat ND_Pi[])1923 PetscErrorCode PCHYPRESetInterpolations(PC pc, PetscInt dim, Mat RT_PiFull, Mat RT_Pi[], Mat ND_PiFull, Mat ND_Pi[])
1924 {
1925 PetscInt i;
1926
1927 PetscFunctionBegin;
1928 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
1929 if (RT_PiFull) {
1930 PetscValidHeaderSpecific(RT_PiFull, MAT_CLASSID, 3);
1931 PetscCheckSameComm(pc, 1, RT_PiFull, 3);
1932 }
1933 if (RT_Pi) {
1934 PetscAssertPointer(RT_Pi, 4);
1935 for (i = 0; i < dim; ++i) {
1936 if (RT_Pi[i]) {
1937 PetscValidHeaderSpecific(RT_Pi[i], MAT_CLASSID, 4);
1938 PetscCheckSameComm(pc, 1, RT_Pi[i], 4);
1939 }
1940 }
1941 }
1942 if (ND_PiFull) {
1943 PetscValidHeaderSpecific(ND_PiFull, MAT_CLASSID, 5);
1944 PetscCheckSameComm(pc, 1, ND_PiFull, 5);
1945 }
1946 if (ND_Pi) {
1947 PetscAssertPointer(ND_Pi, 6);
1948 for (i = 0; i < dim; ++i) {
1949 if (ND_Pi[i]) {
1950 PetscValidHeaderSpecific(ND_Pi[i], MAT_CLASSID, 6);
1951 PetscCheckSameComm(pc, 1, ND_Pi[i], 6);
1952 }
1953 }
1954 }
1955 PetscTryMethod(pc, "PCHYPRESetInterpolations_C", (PC, PetscInt, Mat, Mat[], Mat, Mat[]), (pc, dim, RT_PiFull, RT_Pi, ND_PiFull, ND_Pi));
1956 PetscFunctionReturn(PETSC_SUCCESS);
1957 }
1958
PCHYPRESetPoissonMatrix_HYPRE(PC pc,Mat A,PetscBool isalpha)1959 static PetscErrorCode PCHYPRESetPoissonMatrix_HYPRE(PC pc, Mat A, PetscBool isalpha)
1960 {
1961 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
1962 PetscBool ishypre;
1963
1964 PetscFunctionBegin;
1965 PetscCall(PetscObjectTypeCompare((PetscObject)A, MATHYPRE, &ishypre));
1966 if (ishypre) {
1967 if (isalpha) {
1968 PetscCall(PetscObjectReference((PetscObject)A));
1969 PetscCall(MatDestroy(&jac->alpha_Poisson));
1970 jac->alpha_Poisson = A;
1971 } else {
1972 if (A) {
1973 PetscCall(PetscObjectReference((PetscObject)A));
1974 } else {
1975 jac->ams_beta_is_zero = PETSC_TRUE;
1976 }
1977 PetscCall(MatDestroy(&jac->beta_Poisson));
1978 jac->beta_Poisson = A;
1979 }
1980 } else {
1981 if (isalpha) {
1982 PetscCall(MatDestroy(&jac->alpha_Poisson));
1983 PetscCall(MatConvert(A, MATHYPRE, MAT_INITIAL_MATRIX, &jac->alpha_Poisson));
1984 } else {
1985 if (A) {
1986 PetscCall(MatDestroy(&jac->beta_Poisson));
1987 PetscCall(MatConvert(A, MATHYPRE, MAT_INITIAL_MATRIX, &jac->beta_Poisson));
1988 } else {
1989 PetscCall(MatDestroy(&jac->beta_Poisson));
1990 jac->ams_beta_is_zero = PETSC_TRUE;
1991 }
1992 }
1993 }
1994 PetscFunctionReturn(PETSC_SUCCESS);
1995 }
1996
1997 /*@
1998 PCHYPRESetAlphaPoissonMatrix - Set the vector Poisson matrix for `PCHYPRE` of type AMS
1999
2000 Collective
2001
2002 Input Parameters:
2003 + pc - the preconditioning context
2004 - A - the matrix
2005
2006 Level: intermediate
2007
2008 Note:
2009 `A` should be obtained by discretizing the vector valued Poisson problem with linear finite elements
2010
2011 Developer Notes:
2012 This automatically converts the matrix to `MATHYPRE` if it is not already of that type
2013
2014 If this is only for `PCHYPRE` type of AMS it should be called `PCHYPREAMSSetAlphaPoissonMatrix()`
2015
2016 .seealso: [](ch_ksp), `PCHYPRE`, `PCHYPRESetDiscreteGradient()`, `PCHYPRESetDiscreteCurl()`, `PCHYPRESetBetaPoissonMatrix()`
2017 @*/
PCHYPRESetAlphaPoissonMatrix(PC pc,Mat A)2018 PetscErrorCode PCHYPRESetAlphaPoissonMatrix(PC pc, Mat A)
2019 {
2020 PetscFunctionBegin;
2021 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2022 PetscValidHeaderSpecific(A, MAT_CLASSID, 2);
2023 PetscCheckSameComm(pc, 1, A, 2);
2024 PetscTryMethod(pc, "PCHYPRESetPoissonMatrix_C", (PC, Mat, PetscBool), (pc, A, PETSC_TRUE));
2025 PetscFunctionReturn(PETSC_SUCCESS);
2026 }
2027
2028 /*@
2029 PCHYPRESetBetaPoissonMatrix - Set the Poisson matrix for `PCHYPRE` of type AMS
2030
2031 Collective
2032
2033 Input Parameters:
2034 + pc - the preconditioning context
2035 - A - the matrix, or `NULL` to turn it off
2036
2037 Level: intermediate
2038
2039 Note:
2040 `A` should be obtained by discretizing the Poisson problem with linear finite elements.
2041
2042 Developer Notes:
2043 This automatically converts the matrix to `MATHYPRE` if it is not already of that type
2044
2045 If this is only for `PCHYPRE` type of AMS it should be called `PCHYPREAMSPCHYPRESetBetaPoissonMatrix()`
2046
2047 .seealso: [](ch_ksp), `PCHYPRE`, `PCHYPRESetDiscreteGradient()`, `PCHYPRESetDiscreteCurl()`, `PCHYPRESetAlphaPoissonMatrix()`
2048 @*/
PCHYPRESetBetaPoissonMatrix(PC pc,Mat A)2049 PetscErrorCode PCHYPRESetBetaPoissonMatrix(PC pc, Mat A)
2050 {
2051 PetscFunctionBegin;
2052 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2053 if (A) {
2054 PetscValidHeaderSpecific(A, MAT_CLASSID, 2);
2055 PetscCheckSameComm(pc, 1, A, 2);
2056 }
2057 PetscTryMethod(pc, "PCHYPRESetPoissonMatrix_C", (PC, Mat, PetscBool), (pc, A, PETSC_FALSE));
2058 PetscFunctionReturn(PETSC_SUCCESS);
2059 }
2060
PCHYPRESetEdgeConstantVectors_HYPRE(PC pc,Vec ozz,Vec zoz,Vec zzo)2061 static PetscErrorCode PCHYPRESetEdgeConstantVectors_HYPRE(PC pc, Vec ozz, Vec zoz, Vec zzo)
2062 {
2063 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
2064
2065 PetscFunctionBegin;
2066 /* throw away any vector if already set */
2067 PetscCall(VecHYPRE_IJVectorDestroy(&jac->constants[0]));
2068 PetscCall(VecHYPRE_IJVectorDestroy(&jac->constants[1]));
2069 PetscCall(VecHYPRE_IJVectorDestroy(&jac->constants[2]));
2070 PetscCall(VecHYPRE_IJVectorCreate(ozz->map, &jac->constants[0]));
2071 PetscCall(VecHYPRE_IJVectorCopy(ozz, jac->constants[0]));
2072 PetscCall(VecHYPRE_IJVectorCreate(zoz->map, &jac->constants[1]));
2073 PetscCall(VecHYPRE_IJVectorCopy(zoz, jac->constants[1]));
2074 jac->dim = 2;
2075 if (zzo) {
2076 PetscCall(VecHYPRE_IJVectorCreate(zzo->map, &jac->constants[2]));
2077 PetscCall(VecHYPRE_IJVectorCopy(zzo, jac->constants[2]));
2078 jac->dim++;
2079 }
2080 PetscFunctionReturn(PETSC_SUCCESS);
2081 }
2082
2083 /*@
2084 PCHYPRESetEdgeConstantVectors - Set the representation of the constant vector fields in the edge element basis for `PCHYPRE` of type AMS
2085
2086 Collective
2087
2088 Input Parameters:
2089 + pc - the preconditioning context
2090 . ozz - vector representing (1,0,0) (or (1,0) in 2D)
2091 . zoz - vector representing (0,1,0) (or (0,1) in 2D)
2092 - zzo - vector representing (0,0,1) (use NULL in 2D)
2093
2094 Level: intermediate
2095
2096 Developer Note:
2097 If this is only for `PCHYPRE` type of AMS it should be called `PCHYPREAMSSetEdgeConstantVectors()`
2098
2099 .seealso: [](ch_ksp), `PCHYPRE`, `PCHYPRESetDiscreteGradient()`, `PCHYPRESetDiscreteCurl()`, `PCHYPRESetAlphaPoissonMatrix()`
2100 @*/
PCHYPRESetEdgeConstantVectors(PC pc,Vec ozz,Vec zoz,Vec zzo)2101 PetscErrorCode PCHYPRESetEdgeConstantVectors(PC pc, Vec ozz, Vec zoz, Vec zzo)
2102 {
2103 PetscFunctionBegin;
2104 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2105 PetscValidHeaderSpecific(ozz, VEC_CLASSID, 2);
2106 PetscValidHeaderSpecific(zoz, VEC_CLASSID, 3);
2107 if (zzo) PetscValidHeaderSpecific(zzo, VEC_CLASSID, 4);
2108 PetscCheckSameComm(pc, 1, ozz, 2);
2109 PetscCheckSameComm(pc, 1, zoz, 3);
2110 if (zzo) PetscCheckSameComm(pc, 1, zzo, 4);
2111 PetscTryMethod(pc, "PCHYPRESetEdgeConstantVectors_C", (PC, Vec, Vec, Vec), (pc, ozz, zoz, zzo));
2112 PetscFunctionReturn(PETSC_SUCCESS);
2113 }
2114
PCHYPREAMSSetInteriorNodes_HYPRE(PC pc,Vec interior)2115 static PetscErrorCode PCHYPREAMSSetInteriorNodes_HYPRE(PC pc, Vec interior)
2116 {
2117 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
2118
2119 PetscFunctionBegin;
2120 PetscCall(VecHYPRE_IJVectorDestroy(&jac->interior));
2121 PetscCall(VecHYPRE_IJVectorCreate(interior->map, &jac->interior));
2122 PetscCall(VecHYPRE_IJVectorCopy(interior, jac->interior));
2123 jac->ams_beta_is_zero_part = PETSC_TRUE;
2124 PetscFunctionReturn(PETSC_SUCCESS);
2125 }
2126
2127 /*@
2128 PCHYPREAMSSetInteriorNodes - Set the list of interior nodes to a zero-conductivity region for `PCHYPRE` of type AMS
2129
2130 Collective
2131
2132 Input Parameters:
2133 + pc - the preconditioning context
2134 - interior - vector. node is interior if its entry in the array is 1.0.
2135
2136 Level: intermediate
2137
2138 Note:
2139 This calls `HYPRE_AMSSetInteriorNodes()`
2140
2141 .seealso: [](ch_ksp), `PCHYPRE`, `PCHYPRESetDiscreteGradient()`, `PCHYPRESetDiscreteCurl()`, `PCHYPRESetAlphaPoissonMatrix()`
2142 @*/
PCHYPREAMSSetInteriorNodes(PC pc,Vec interior)2143 PetscErrorCode PCHYPREAMSSetInteriorNodes(PC pc, Vec interior)
2144 {
2145 PetscFunctionBegin;
2146 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2147 PetscValidHeaderSpecific(interior, VEC_CLASSID, 2);
2148 PetscCheckSameComm(pc, 1, interior, 2);
2149 PetscTryMethod(pc, "PCHYPREAMSSetInteriorNodes_C", (PC, Vec), (pc, interior));
2150 PetscFunctionReturn(PETSC_SUCCESS);
2151 }
2152
PCSetCoordinates_HYPRE(PC pc,PetscInt dim,PetscInt nloc,PetscReal * coords)2153 static PetscErrorCode PCSetCoordinates_HYPRE(PC pc, PetscInt dim, PetscInt nloc, PetscReal *coords)
2154 {
2155 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
2156 Vec tv;
2157 PetscInt i;
2158
2159 PetscFunctionBegin;
2160 /* throw away any coordinate vector if already set */
2161 PetscCall(VecHYPRE_IJVectorDestroy(&jac->coords[0]));
2162 PetscCall(VecHYPRE_IJVectorDestroy(&jac->coords[1]));
2163 PetscCall(VecHYPRE_IJVectorDestroy(&jac->coords[2]));
2164 jac->dim = dim;
2165
2166 /* compute IJ vector for coordinates */
2167 PetscCall(VecCreate(PetscObjectComm((PetscObject)pc), &tv));
2168 PetscCall(VecSetType(tv, VECSTANDARD));
2169 PetscCall(VecSetSizes(tv, nloc, PETSC_DECIDE));
2170 for (i = 0; i < dim; i++) {
2171 PetscScalar *array;
2172 PetscInt j;
2173
2174 PetscCall(VecHYPRE_IJVectorCreate(tv->map, &jac->coords[i]));
2175 PetscCall(VecGetArrayWrite(tv, &array));
2176 for (j = 0; j < nloc; j++) array[j] = coords[j * dim + i];
2177 PetscCall(VecRestoreArrayWrite(tv, &array));
2178 PetscCall(VecHYPRE_IJVectorCopy(tv, jac->coords[i]));
2179 }
2180 PetscCall(VecDestroy(&tv));
2181 PetscFunctionReturn(PETSC_SUCCESS);
2182 }
2183
PCHYPREGetType_HYPRE(PC pc,const char * name[])2184 static PetscErrorCode PCHYPREGetType_HYPRE(PC pc, const char *name[])
2185 {
2186 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
2187
2188 PetscFunctionBegin;
2189 *name = jac->hypre_type;
2190 PetscFunctionReturn(PETSC_SUCCESS);
2191 }
2192
PCHYPRESetType_HYPRE(PC pc,const char name[])2193 static PetscErrorCode PCHYPRESetType_HYPRE(PC pc, const char name[])
2194 {
2195 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
2196 PetscBool flag;
2197
2198 PetscFunctionBegin;
2199 if (jac->hypre_type) {
2200 PetscCall(PetscStrcmp(jac->hypre_type, name, &flag));
2201 if (flag) PetscFunctionReturn(PETSC_SUCCESS);
2202 }
2203
2204 PetscCall(PCReset_HYPRE(pc));
2205 PetscCall(PetscFree(jac->hypre_type));
2206 PetscCall(PetscStrallocpy(name, &jac->hypre_type));
2207
2208 jac->maxiter = PETSC_DEFAULT;
2209 jac->tol = PETSC_DEFAULT;
2210 jac->printstatistics = PetscLogPrintInfo;
2211
2212 PetscCall(PetscStrcmp("ilu", jac->hypre_type, &flag));
2213 if (flag) {
2214 PetscCall(PetscCommGetComm(PetscObjectComm((PetscObject)pc), &jac->comm_hypre));
2215 PetscCallHYPRE(HYPRE_ILUCreate(&jac->hsolver));
2216 pc->ops->setfromoptions = PCSetFromOptions_HYPRE_ILU;
2217 pc->ops->view = PCView_HYPRE_ILU;
2218 jac->destroy = HYPRE_ILUDestroy;
2219 jac->setup = HYPRE_ILUSetup;
2220 jac->solve = HYPRE_ILUSolve;
2221 jac->factorrowsize = PETSC_DEFAULT;
2222 PetscFunctionReturn(PETSC_SUCCESS);
2223 }
2224
2225 PetscCall(PetscStrcmp("pilut", jac->hypre_type, &flag));
2226 if (flag) {
2227 PetscCall(PetscCommGetComm(PetscObjectComm((PetscObject)pc), &jac->comm_hypre));
2228 PetscCallHYPRE(HYPRE_ParCSRPilutCreate(jac->comm_hypre, &jac->hsolver));
2229 pc->ops->setfromoptions = PCSetFromOptions_HYPRE_Pilut;
2230 pc->ops->view = PCView_HYPRE_Pilut;
2231 jac->destroy = HYPRE_ParCSRPilutDestroy;
2232 jac->setup = HYPRE_ParCSRPilutSetup;
2233 jac->solve = HYPRE_ParCSRPilutSolve;
2234 jac->factorrowsize = PETSC_DEFAULT;
2235 PetscFunctionReturn(PETSC_SUCCESS);
2236 }
2237 PetscCall(PetscStrcmp("euclid", jac->hypre_type, &flag));
2238 if (flag) {
2239 #if defined(PETSC_USE_64BIT_INDICES)
2240 SETERRQ(PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Hypre Euclid does not support 64-bit indices");
2241 #endif
2242 PetscCall(PetscCommGetComm(PetscObjectComm((PetscObject)pc), &jac->comm_hypre));
2243 PetscCallHYPRE(HYPRE_EuclidCreate(jac->comm_hypre, &jac->hsolver));
2244 pc->ops->setfromoptions = PCSetFromOptions_HYPRE_Euclid;
2245 pc->ops->view = PCView_HYPRE_Euclid;
2246 jac->destroy = HYPRE_EuclidDestroy;
2247 jac->setup = HYPRE_EuclidSetup;
2248 jac->solve = HYPRE_EuclidSolve;
2249 jac->factorrowsize = PETSC_DEFAULT;
2250 jac->eu_level = PETSC_DEFAULT; /* default */
2251 PetscFunctionReturn(PETSC_SUCCESS);
2252 }
2253 PetscCall(PetscStrcmp("parasails", jac->hypre_type, &flag));
2254 if (flag) {
2255 PetscCall(PetscCommGetComm(PetscObjectComm((PetscObject)pc), &jac->comm_hypre));
2256 PetscCallHYPRE(HYPRE_ParaSailsCreate(jac->comm_hypre, &jac->hsolver));
2257 pc->ops->setfromoptions = PCSetFromOptions_HYPRE_ParaSails;
2258 pc->ops->view = PCView_HYPRE_ParaSails;
2259 jac->destroy = HYPRE_ParaSailsDestroy;
2260 jac->setup = HYPRE_ParaSailsSetup;
2261 jac->solve = HYPRE_ParaSailsSolve;
2262 /* initialize */
2263 jac->nlevels = 1;
2264 jac->threshold = .1;
2265 jac->filter = .1;
2266 jac->loadbal = 0;
2267 if (PetscLogPrintInfo) jac->logging = (int)PETSC_TRUE;
2268 else jac->logging = (int)PETSC_FALSE;
2269
2270 jac->ruse = (int)PETSC_FALSE;
2271 jac->symt = 0;
2272 PetscCallHYPRE(HYPRE_ParaSailsSetParams(jac->hsolver, jac->threshold, (HYPRE_Int)jac->nlevels));
2273 PetscCallHYPRE(HYPRE_ParaSailsSetFilter(jac->hsolver, jac->filter));
2274 PetscCallHYPRE(HYPRE_ParaSailsSetLoadbal(jac->hsolver, (HYPRE_Int)jac->loadbal));
2275 PetscCallHYPRE(HYPRE_ParaSailsSetLogging(jac->hsolver, (HYPRE_Int)jac->logging));
2276 PetscCallHYPRE(HYPRE_ParaSailsSetReuse(jac->hsolver, (HYPRE_Int)jac->ruse));
2277 PetscCallHYPRE(HYPRE_ParaSailsSetSym(jac->hsolver, (HYPRE_Int)jac->symt));
2278 PetscFunctionReturn(PETSC_SUCCESS);
2279 }
2280 PetscCall(PetscStrcmp("boomeramg", jac->hypre_type, &flag));
2281 if (flag) {
2282 PetscCallHYPRE(HYPRE_BoomerAMGCreate(&jac->hsolver));
2283 pc->ops->setfromoptions = PCSetFromOptions_HYPRE_BoomerAMG;
2284 pc->ops->view = PCView_HYPRE_BoomerAMG;
2285 pc->ops->applytranspose = PCApplyTranspose_HYPRE_BoomerAMG;
2286 pc->ops->applyrichardson = PCApplyRichardson_HYPRE_BoomerAMG;
2287 pc->ops->matapply = PCMatApply_HYPRE_BoomerAMG;
2288 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetInterpolations_C", PCGetInterpolations_BoomerAMG));
2289 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetCoarseOperators_C", PCGetCoarseOperators_BoomerAMG));
2290 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPREGetCFMarkers_C", PCHYPREGetCFMarkers_BoomerAMG));
2291 jac->destroy = HYPRE_BoomerAMGDestroy;
2292 jac->setup = HYPRE_BoomerAMGSetup;
2293 jac->solve = HYPRE_BoomerAMGSolve;
2294 jac->applyrichardson = PETSC_FALSE;
2295 /* these defaults match the hypre defaults */
2296 jac->cycletype = 1;
2297 jac->maxlevels = 25;
2298 jac->maxiter = 1;
2299 jac->tol = 0.0; /* tolerance of zero indicates use as preconditioner (suppresses convergence errors) */
2300 jac->truncfactor = 0.0;
2301 jac->strongthreshold = .25;
2302 jac->maxrowsum = .9;
2303 jac->measuretype = 0;
2304 jac->gridsweeps[0] = jac->gridsweeps[1] = jac->gridsweeps[2] = 1;
2305 jac->smoothtype = -1; /* Not set by default */
2306 jac->smoothnumlevels = 25;
2307 jac->eu_level = 0;
2308 jac->eu_droptolerance = 0;
2309 jac->eu_bj = 0;
2310 jac->relaxweight = 1.0;
2311 jac->outerrelaxweight = 1.0;
2312 jac->Rtype = 0;
2313 jac->Rstrongthreshold = 0.25;
2314 jac->Rfilterthreshold = 0.0;
2315 jac->Adroptype = -1;
2316 jac->Adroptol = 0.0;
2317 jac->agg_nl = 0;
2318 jac->pmax = 0;
2319 jac->truncfactor = 0.0;
2320 jac->agg_num_paths = 1;
2321 jac->maxc = 9;
2322 jac->minc = 1;
2323 jac->nodal_coarsening = 0;
2324 jac->nodal_coarsening_diag = 0;
2325 jac->vec_interp_variant = 0;
2326 jac->vec_interp_qmax = 0;
2327 jac->vec_interp_smooth = PETSC_FALSE;
2328 jac->interp_refine = 0;
2329 jac->nodal_relax = PETSC_FALSE;
2330 jac->nodal_relax_levels = 1;
2331 jac->rap2 = 0;
2332 PetscObjectParameterSetDefault(jac, relaxtype[2], 9); /* G.E. */
2333
2334 /*
2335 Initialize the following parameters with invalid value so we can recognize user input that sets the parameter.
2336 If there is no user input they are overwritten in PCSetUp_HYPRE() depending on if the matrix is on the CPU or the GPU
2337 */
2338 PetscObjectParameterSetDefault(jac, relaxorder, PETSC_DECIDE);
2339 PetscObjectParameterSetDefault(jac, coarsentype, PETSC_DECIDE);
2340 PetscObjectParameterSetDefault(jac, interptype, PETSC_DECIDE);
2341 PetscObjectParameterSetDefault(jac, relaxtype[0], PETSC_DECIDE);
2342 PetscObjectParameterSetDefault(jac, relaxtype[1], PETSC_DECIDE);
2343 #if PETSC_PKG_HYPRE_VERSION_GE(2, 23, 0)
2344 PetscObjectParameterSetDefault(jac, spgemm_type, "not yet set");
2345 #endif
2346 #if PETSC_PKG_HYPRE_VERSION_GE(2, 18, 0)
2347 PetscObjectParameterSetDefault(jac, keeptranspose, PETSC_BOOL3_UNKNOWN);
2348 PetscObjectParameterSetDefault(jac, mod_rap2, PETSC_DECIDE);
2349 #endif
2350 PetscObjectParameterSetDefault(jac, agg_interptype, PETSC_DECIDE);
2351 PetscFunctionReturn(PETSC_SUCCESS);
2352 }
2353 PetscCall(PetscStrcmp("ams", jac->hypre_type, &flag));
2354 if (flag) {
2355 PetscCallHYPRE(HYPRE_AMSCreate(&jac->hsolver));
2356 pc->ops->setfromoptions = PCSetFromOptions_HYPRE_AMS;
2357 pc->ops->view = PCView_HYPRE_AMS;
2358 jac->destroy = HYPRE_AMSDestroy;
2359 jac->setup = HYPRE_AMSSetup;
2360 jac->solve = HYPRE_AMSSolve;
2361 jac->coords[0] = NULL;
2362 jac->coords[1] = NULL;
2363 jac->coords[2] = NULL;
2364 jac->interior = NULL;
2365 /* solver parameters: these are borrowed from mfem package, and they are not the default values from HYPRE AMS */
2366 jac->as_print = 0;
2367 jac->as_max_iter = 1; /* used as a preconditioner */
2368 jac->as_tol = 0.; /* used as a preconditioner */
2369 jac->ams_cycle_type = 13;
2370 /* Smoothing options */
2371 jac->as_relax_type = 2;
2372 jac->as_relax_times = 1;
2373 jac->as_relax_weight = 1.0;
2374 jac->as_omega = 1.0;
2375 /* Vector valued Poisson AMG solver parameters: coarsen type, agg_levels, relax_type, interp_type, Pmax */
2376 jac->as_amg_alpha_opts[0] = 10;
2377 jac->as_amg_alpha_opts[1] = 1;
2378 jac->as_amg_alpha_opts[2] = 6;
2379 jac->as_amg_alpha_opts[3] = 6;
2380 jac->as_amg_alpha_opts[4] = 4;
2381 jac->as_amg_alpha_theta = 0.25;
2382 /* Scalar Poisson AMG solver parameters: coarsen type, agg_levels, relax_type, interp_type, Pmax */
2383 jac->as_amg_beta_opts[0] = 10;
2384 jac->as_amg_beta_opts[1] = 1;
2385 jac->as_amg_beta_opts[2] = 6;
2386 jac->as_amg_beta_opts[3] = 6;
2387 jac->as_amg_beta_opts[4] = 4;
2388 jac->as_amg_beta_theta = 0.25;
2389 PetscCallHYPRE(HYPRE_AMSSetPrintLevel(jac->hsolver, (HYPRE_Int)jac->as_print));
2390 PetscCallHYPRE(HYPRE_AMSSetMaxIter(jac->hsolver, (HYPRE_Int)jac->as_max_iter));
2391 PetscCallHYPRE(HYPRE_AMSSetCycleType(jac->hsolver, (HYPRE_Int)jac->ams_cycle_type));
2392 PetscCallHYPRE(HYPRE_AMSSetTol(jac->hsolver, jac->as_tol));
2393 PetscCallHYPRE(HYPRE_AMSSetSmoothingOptions(jac->hsolver, (HYPRE_Int)jac->as_relax_type, (HYPRE_Int)jac->as_relax_times, jac->as_relax_weight, jac->as_omega));
2394 PetscCallHYPRE(HYPRE_AMSSetAlphaAMGOptions(jac->hsolver, (HYPRE_Int)jac->as_amg_alpha_opts[0], /* AMG coarsen type */
2395 (HYPRE_Int)jac->as_amg_alpha_opts[1], /* AMG agg_levels */
2396 (HYPRE_Int)jac->as_amg_alpha_opts[2], /* AMG relax_type */
2397 jac->as_amg_alpha_theta, (HYPRE_Int)jac->as_amg_alpha_opts[3], /* AMG interp_type */
2398 (HYPRE_Int)jac->as_amg_alpha_opts[4])); /* AMG Pmax */
2399 PetscCallHYPRE(HYPRE_AMSSetBetaAMGOptions(jac->hsolver, (HYPRE_Int)jac->as_amg_beta_opts[0], /* AMG coarsen type */
2400 (HYPRE_Int)jac->as_amg_beta_opts[1], /* AMG agg_levels */
2401 (HYPRE_Int)jac->as_amg_beta_opts[2], /* AMG relax_type */
2402 jac->as_amg_beta_theta, (HYPRE_Int)jac->as_amg_beta_opts[3], /* AMG interp_type */
2403 (HYPRE_Int)jac->as_amg_beta_opts[4])); /* AMG Pmax */
2404 /* Zero conductivity */
2405 jac->ams_beta_is_zero = PETSC_FALSE;
2406 jac->ams_beta_is_zero_part = PETSC_FALSE;
2407 PetscFunctionReturn(PETSC_SUCCESS);
2408 }
2409 PetscCall(PetscStrcmp("ads", jac->hypre_type, &flag));
2410 if (flag) {
2411 PetscCallHYPRE(HYPRE_ADSCreate(&jac->hsolver));
2412 pc->ops->setfromoptions = PCSetFromOptions_HYPRE_ADS;
2413 pc->ops->view = PCView_HYPRE_ADS;
2414 jac->destroy = HYPRE_ADSDestroy;
2415 jac->setup = HYPRE_ADSSetup;
2416 jac->solve = HYPRE_ADSSolve;
2417 jac->coords[0] = NULL;
2418 jac->coords[1] = NULL;
2419 jac->coords[2] = NULL;
2420 /* solver parameters: these are borrowed from mfem package, and they are not the default values from HYPRE ADS */
2421 jac->as_print = 0;
2422 jac->as_max_iter = 1; /* used as a preconditioner */
2423 jac->as_tol = 0.; /* used as a preconditioner */
2424 jac->ads_cycle_type = 13;
2425 /* Smoothing options */
2426 jac->as_relax_type = 2;
2427 jac->as_relax_times = 1;
2428 jac->as_relax_weight = 1.0;
2429 jac->as_omega = 1.0;
2430 /* AMS solver parameters: cycle_type, coarsen type, agg_levels, relax_type, interp_type, Pmax */
2431 jac->ams_cycle_type = 14;
2432 jac->as_amg_alpha_opts[0] = 10;
2433 jac->as_amg_alpha_opts[1] = 1;
2434 jac->as_amg_alpha_opts[2] = 6;
2435 jac->as_amg_alpha_opts[3] = 6;
2436 jac->as_amg_alpha_opts[4] = 4;
2437 jac->as_amg_alpha_theta = 0.25;
2438 /* Vector Poisson AMG solver parameters: coarsen type, agg_levels, relax_type, interp_type, Pmax */
2439 jac->as_amg_beta_opts[0] = 10;
2440 jac->as_amg_beta_opts[1] = 1;
2441 jac->as_amg_beta_opts[2] = 6;
2442 jac->as_amg_beta_opts[3] = 6;
2443 jac->as_amg_beta_opts[4] = 4;
2444 jac->as_amg_beta_theta = 0.25;
2445 PetscCallHYPRE(HYPRE_ADSSetPrintLevel(jac->hsolver, (HYPRE_Int)jac->as_print));
2446 PetscCallHYPRE(HYPRE_ADSSetMaxIter(jac->hsolver, (HYPRE_Int)jac->as_max_iter));
2447 PetscCallHYPRE(HYPRE_ADSSetCycleType(jac->hsolver, (HYPRE_Int)jac->ams_cycle_type));
2448 PetscCallHYPRE(HYPRE_ADSSetTol(jac->hsolver, jac->as_tol));
2449 PetscCallHYPRE(HYPRE_ADSSetSmoothingOptions(jac->hsolver, (HYPRE_Int)jac->as_relax_type, (HYPRE_Int)jac->as_relax_times, jac->as_relax_weight, jac->as_omega));
2450 PetscCallHYPRE(HYPRE_ADSSetAMSOptions(jac->hsolver, (HYPRE_Int)jac->ams_cycle_type, /* AMG coarsen type */
2451 (HYPRE_Int)jac->as_amg_alpha_opts[0], /* AMG coarsen type */
2452 (HYPRE_Int)jac->as_amg_alpha_opts[1], /* AMG agg_levels */
2453 (HYPRE_Int)jac->as_amg_alpha_opts[2], /* AMG relax_type */
2454 jac->as_amg_alpha_theta, (HYPRE_Int)jac->as_amg_alpha_opts[3], /* AMG interp_type */
2455 (HYPRE_Int)jac->as_amg_alpha_opts[4])); /* AMG Pmax */
2456 PetscCallHYPRE(HYPRE_ADSSetAMGOptions(jac->hsolver, (HYPRE_Int)jac->as_amg_beta_opts[0], /* AMG coarsen type */
2457 (HYPRE_Int)jac->as_amg_beta_opts[1], /* AMG agg_levels */
2458 (HYPRE_Int)jac->as_amg_beta_opts[2], /* AMG relax_type */
2459 jac->as_amg_beta_theta, (HYPRE_Int)jac->as_amg_beta_opts[3], /* AMG interp_type */
2460 (HYPRE_Int)jac->as_amg_beta_opts[4])); /* AMG Pmax */
2461 PetscFunctionReturn(PETSC_SUCCESS);
2462 }
2463 PetscCall(PetscFree(jac->hypre_type));
2464
2465 jac->hypre_type = NULL;
2466 SETERRQ(PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unknown HYPRE preconditioner %s; Choices are euclid, ilu, pilut, parasails, boomeramg, ams, ads", name);
2467 }
2468
2469 /*
2470 It only gets here if the HYPRE type has not been set before the call to
2471 ...SetFromOptions() which actually is most of the time
2472 */
PCSetFromOptions_HYPRE(PC pc,PetscOptionItems PetscOptionsObject)2473 static PetscErrorCode PCSetFromOptions_HYPRE(PC pc, PetscOptionItems PetscOptionsObject)
2474 {
2475 PetscInt indx;
2476 const char *type[] = {"ilu", "euclid", "pilut", "parasails", "boomeramg", "ams", "ads"};
2477 PetscBool flg;
2478 PC_HYPRE *jac = (PC_HYPRE *)pc->data;
2479
2480 PetscFunctionBegin;
2481 PetscOptionsHeadBegin(PetscOptionsObject, "HYPRE preconditioner options");
2482 PetscCall(PetscOptionsEList("-pc_hypre_type", "HYPRE preconditioner type", "PCHYPRESetType", type, PETSC_STATIC_ARRAY_LENGTH(type), "boomeramg", &indx, &flg));
2483 if (flg) PetscCall(PCHYPRESetType_HYPRE(pc, type[indx]));
2484 /*
2485 Set the type if it was never set.
2486 */
2487 if (!jac->hypre_type) PetscCall(PCHYPRESetType_HYPRE(pc, "boomeramg"));
2488 PetscTryTypeMethod(pc, setfromoptions, PetscOptionsObject);
2489 PetscOptionsHeadEnd();
2490 PetscFunctionReturn(PETSC_SUCCESS);
2491 }
2492
2493 /*@
2494 PCHYPRESetType - Sets which hypre preconditioner you wish to use
2495
2496 Input Parameters:
2497 + pc - the preconditioner context
2498 - name - either euclid, ilu, pilut, parasails, boomeramg, ams, or ads
2499
2500 Options Database Key:
2501 . pc_hypre_type - One of euclid, ilu, pilut, parasails, boomeramg, ams, or ads
2502
2503 Level: intermediate
2504
2505 .seealso: [](ch_ksp), `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHYPRE`
2506 @*/
PCHYPRESetType(PC pc,const char name[])2507 PetscErrorCode PCHYPRESetType(PC pc, const char name[])
2508 {
2509 PetscFunctionBegin;
2510 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2511 PetscAssertPointer(name, 2);
2512 PetscTryMethod(pc, "PCHYPRESetType_C", (PC, const char[]), (pc, name));
2513 PetscFunctionReturn(PETSC_SUCCESS);
2514 }
2515
2516 /*@C
2517 PCHYPREGetCFMarkers - Gets CF marker arrays for all levels (except the finest level)
2518
2519 Logically Collective
2520
2521 Input Parameter:
2522 . pc - the preconditioner context
2523
2524 Output Parameters:
2525 + n_per_level - the number of nodes per level (size of `num_levels`)
2526 - CFMarkers - the Coarse/Fine Boolean arrays (size of `num_levels` - 1)
2527
2528 Level: advanced
2529
2530 Note:
2531 Caller is responsible for memory management of `n_per_level` and `CFMarkers` pointers. That is they should free them with `PetscFree()` when no longer needed.
2532
2533 .seealso: [](ch_ksp), `PC`, `PCMG`, `PCMGGetRestriction()`, `PCMGSetInterpolation()`, `PCMGGetRScale()`, `PCMGGetInterpolation()`, `PCGetInterpolations()`
2534 @*/
PCHYPREGetCFMarkers(PC pc,PetscInt * n_per_level[],PetscBT * CFMarkers[])2535 PetscErrorCode PCHYPREGetCFMarkers(PC pc, PetscInt *n_per_level[], PetscBT *CFMarkers[])
2536 {
2537 PetscFunctionBegin;
2538 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2539 PetscAssertPointer(n_per_level, 2);
2540 PetscAssertPointer(CFMarkers, 3);
2541 PetscUseMethod(pc, "PCHYPREGetCFMarkers_C", (PC, PetscInt *[], PetscBT *[]), (pc, n_per_level, CFMarkers));
2542 PetscFunctionReturn(PETSC_SUCCESS);
2543 }
2544
2545 /*@
2546 PCHYPREGetType - Gets which hypre preconditioner you are using
2547
2548 Input Parameter:
2549 . pc - the preconditioner context
2550
2551 Output Parameter:
2552 . name - either euclid, ilu, pilut, parasails, boomeramg, ams, or ads
2553
2554 Level: intermediate
2555
2556 .seealso: [](ch_ksp), `PCCreate()`, `PCHYPRESetType()`, `PCType`, `PC`, `PCHYPRE`
2557 @*/
PCHYPREGetType(PC pc,const char * name[])2558 PetscErrorCode PCHYPREGetType(PC pc, const char *name[])
2559 {
2560 PetscFunctionBegin;
2561 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2562 PetscAssertPointer(name, 2);
2563 PetscTryMethod(pc, "PCHYPREGetType_C", (PC, const char *[]), (pc, name));
2564 PetscFunctionReturn(PETSC_SUCCESS);
2565 }
2566
2567 /*@
2568 PCMGGalerkinSetMatProductAlgorithm - Set type of sparse matrix-matrix product for hypre's BoomerAMG to use on GPUs
2569
2570 Logically Collective
2571
2572 Input Parameters:
2573 + pc - the hypre context
2574 - name - one of 'cusparse', 'hypre'
2575
2576 Options Database Key:
2577 . -pc_mg_galerkin_mat_product_algorithm <cusparse,hypre> - Type of sparse matrix-matrix product to use in hypre
2578
2579 Level: intermediate
2580
2581 Developer Note:
2582 How the name starts with `PCMG`, should it not be `PCHYPREBoomerAMG`?
2583
2584 .seealso: [](ch_ksp), `PCHYPRE`, `PCMGGalerkinGetMatProductAlgorithm()`
2585 @*/
PCMGGalerkinSetMatProductAlgorithm(PC pc,const char name[])2586 PetscErrorCode PCMGGalerkinSetMatProductAlgorithm(PC pc, const char name[])
2587 {
2588 PetscFunctionBegin;
2589 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2590 PetscTryMethod(pc, "PCMGGalerkinSetMatProductAlgorithm_C", (PC, const char[]), (pc, name));
2591 PetscFunctionReturn(PETSC_SUCCESS);
2592 }
2593
2594 /*@
2595 PCMGGalerkinGetMatProductAlgorithm - Get type of sparse matrix-matrix product for hypre's BoomerAMG to use on GPUs
2596
2597 Not Collective
2598
2599 Input Parameter:
2600 . pc - the multigrid context
2601
2602 Output Parameter:
2603 . name - one of 'cusparse', 'hypre'
2604
2605 Level: intermediate
2606
2607 .seealso: [](ch_ksp), `PCHYPRE`, `PCMGGalerkinSetMatProductAlgorithm()`
2608 @*/
PCMGGalerkinGetMatProductAlgorithm(PC pc,const char * name[])2609 PetscErrorCode PCMGGalerkinGetMatProductAlgorithm(PC pc, const char *name[])
2610 {
2611 PetscFunctionBegin;
2612 PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2613 PetscTryMethod(pc, "PCMGGalerkinGetMatProductAlgorithm_C", (PC, const char *[]), (pc, name));
2614 PetscFunctionReturn(PETSC_SUCCESS);
2615 }
2616
2617 /*MC
2618 PCHYPRE - Allows you to use the matrix element based preconditioners in the LLNL package hypre as PETSc `PC`
2619
2620 Options Database Keys:
2621 + -pc_hypre_type - One of `euclid`, `ilu`, `pilut`, `parasails`, `boomeramg`, `ams`, or `ads`
2622 . -pc_hypre_boomeramg_nodal_coarsen <n> - where `n` is from 1 to 6 (see `HYPRE_BoomerAMGSetNodal()`)
2623 . -pc_hypre_boomeramg_vec_interp_variant <v> - where `v` is from 1 to 3 (see `HYPRE_BoomerAMGSetInterpVecVariant()`)
2624 - Many others - run with `-pc_type hypre` `-pc_hypre_type XXX` `-help` to see options for the XXX preconditioner
2625
2626 Level: intermediate
2627
2628 Notes:
2629 Apart from `-pc_hypre_type` (for which there is `PCHYPRESetType()`),
2630 the many hypre options can ONLY be set via the options database (e.g. the command line
2631 or with `PetscOptionsSetValue()`, there are no functions to set them)
2632
2633 The options `-pc_hypre_boomeramg_max_iter` and `-pc_hypre_boomeramg_tol` refer to the number of iterations
2634 (V-cycles) and tolerance that boomerAMG does EACH time it is called. So for example, if
2635 `-pc_hypre_boomeramg_max_iter` is set to 2 then 2-V-cycles are being used to define the preconditioner
2636 (`-pc_hypre_boomeramg_tol` should be set to 0.0 - the default - to strictly use a fixed number of
2637 iterations per hypre call). `-ksp_max_it` and `-ksp_rtol` STILL determine the total number of iterations
2638 and tolerance for the Krylov solver. For example, if `-pc_hypre_boomeramg_max_iter` is 2 and `-ksp_max_it` is 10
2639 then AT MOST twenty V-cycles of BoomerAMG will be used.
2640
2641 Note that the option `-pc_hypre_boomeramg_relax_type_all` defaults to symmetric relaxation
2642 (symmetric-SOR/Jacobi), which is required for Krylov solvers like CG that expect symmetry.
2643 Otherwise, you may want to use `-pc_hypre_boomeramg_relax_type_all SOR/Jacobi`.
2644
2645 If you provide a near null space to your matrix with `MatSetNearNullSpace()` it is ignored by hypre's BoomerAMG UNLESS you also use
2646 the following two options: `-pc_hypre_boomeramg_nodal_coarsen <n> -pc_hypre_boomeramg_vec_interp_variant <v>`
2647
2648 See `PCPFMG`, `PCSMG`, and `PCSYSPFMG` for access to hypre's other (nonalgebraic) multigrid solvers
2649
2650 For `PCHYPRE` type of `ams` or `ads` auxiliary data must be provided to the preconditioner with `PCHYPRESetDiscreteGradient()`,
2651 `PCHYPRESetDiscreteCurl()`, `PCHYPRESetInterpolations()`, `PCHYPRESetAlphaPoissonMatrix()`, `PCHYPRESetBetaPoissonMatrix()`, `PCHYPRESetEdgeConstantVectors()`,
2652 `PCHYPREAMSSetInteriorNodes()`
2653
2654 Sometimes people want to try algebraic multigrid as a "standalone" solver, that is not accelerating it with a Krylov method. Though we generally do not recommend this
2655 since it is usually slower, one should use a `KSPType` of `KSPRICHARDSON`
2656 (or equivalently `-ksp_type richardson`) to achieve this. Using `KSPPREONLY` will not work since it only applies a single cycle of multigrid.
2657
2658 PETSc provides its own geometric and algebraic multigrid solvers `PCMG` and `PCGAMG`, also see `PCHMG` which is useful for certain multicomponent problems.
2659
2660 hypre supports performance logging via the `Caliper` library. With `--download-hypre --download-caliper`, hypre will be automatically configured with the support.
2661
2662 Enabling Caliper logging requires setting the `CALI_CONFIG` environment variable before running your hypre code. For example,
2663
2664 .vb
2665 export CALI_CONFIG=runtime-report,max_column_width=200,calc.inclusive,mpi-report,output=stdout
2666 .ve
2667
2668 Then run a hypre code, and you will see profiling results on stdout. See https://software.llnl.gov/Caliper/#guides for more options.
2669
2670 GPU Notes:
2671 To configure hypre BoomerAMG so that it can utilize NVIDIA GPUs run ./configure --download-hypre --with-cuda
2672 Then pass `VECCUDA` vectors and `MATAIJCUSPARSE` matrices to the solvers and PETSc will automatically utilize hypre's GPU solvers.
2673
2674 To configure hypre BoomerAMG so that it can utilize AMD GPUs run ./configure --download-hypre --with-hip
2675 Then pass `VECHIP` vectors to the solvers and PETSc will automatically utilize hypre's GPU solvers.
2676
2677 .seealso: [](ch_ksp), `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHYPRESetType()`, `PCPFMG`, `PCGAMG`, `PCSYSPFMG`, `PCSMG`, `PCHYPRESetDiscreteGradient()`,
2678 `PCHYPRESetDiscreteCurl()`, `PCHYPRESetInterpolations()`, `PCHYPRESetAlphaPoissonMatrix()`, `PCHYPRESetBetaPoissonMatrix()`, `PCHYPRESetEdgeConstantVectors()`,
2679 PCHYPREAMSSetInteriorNodes()
2680 M*/
2681
PCCreate_HYPRE(PC pc)2682 PETSC_EXTERN PetscErrorCode PCCreate_HYPRE(PC pc)
2683 {
2684 PC_HYPRE *jac;
2685
2686 PetscFunctionBegin;
2687 PetscCall(PetscNew(&jac));
2688
2689 pc->data = jac;
2690 pc->ops->reset = PCReset_HYPRE;
2691 pc->ops->destroy = PCDestroy_HYPRE;
2692 pc->ops->setfromoptions = PCSetFromOptions_HYPRE;
2693 pc->ops->setup = PCSetUp_HYPRE;
2694 pc->ops->apply = PCApply_HYPRE;
2695 jac->hypre_type = NULL;
2696 jac->comm_hypre = MPI_COMM_NULL;
2697 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetType_C", PCHYPRESetType_HYPRE));
2698 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPREGetType_C", PCHYPREGetType_HYPRE));
2699 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCSetCoordinates_C", PCSetCoordinates_HYPRE));
2700 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetDiscreteGradient_C", PCHYPRESetDiscreteGradient_HYPRE));
2701 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetDiscreteCurl_C", PCHYPRESetDiscreteCurl_HYPRE));
2702 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetInterpolations_C", PCHYPRESetInterpolations_HYPRE));
2703 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetEdgeConstantVectors_C", PCHYPRESetEdgeConstantVectors_HYPRE));
2704 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPREAMSSetInteriorNodes_C", PCHYPREAMSSetInteriorNodes_HYPRE));
2705 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHYPRESetPoissonMatrix_C", PCHYPRESetPoissonMatrix_HYPRE));
2706 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGalerkinSetMatProductAlgorithm_C", PCMGGalerkinSetMatProductAlgorithm_HYPRE_BoomerAMG));
2707 PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGalerkinGetMatProductAlgorithm_C", PCMGGalerkinGetMatProductAlgorithm_HYPRE_BoomerAMG));
2708 #if defined(PETSC_HAVE_HYPRE_DEVICE)
2709 #if defined(HYPRE_USING_HIP)
2710 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_HIP));
2711 #endif
2712 #if defined(HYPRE_USING_CUDA)
2713 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA));
2714 #endif
2715 #endif
2716 PetscCall(PetscHYPREInitialize());
2717 PetscFunctionReturn(PETSC_SUCCESS);
2718 }
2719
2720 typedef struct {
2721 MPI_Comm hcomm; /* does not share comm with HYPRE_StructMatrix because need to create solver before getting matrix */
2722 HYPRE_StructSolver hsolver;
2723
2724 /* keep copy of PFMG options used so may view them */
2725 PetscInt its;
2726 PetscReal tol;
2727 PetscInt relax_type;
2728 PetscInt rap_type;
2729 PetscInt num_pre_relax, num_post_relax;
2730 PetscInt max_levels;
2731 PetscInt skip_relax;
2732 PetscBool print_statistics;
2733 } PC_PFMG;
2734
PCDestroy_PFMG(PC pc)2735 static PetscErrorCode PCDestroy_PFMG(PC pc)
2736 {
2737 PC_PFMG *ex = (PC_PFMG *)pc->data;
2738
2739 PetscFunctionBegin;
2740 if (ex->hsolver) PetscCallHYPRE(HYPRE_StructPFMGDestroy(ex->hsolver));
2741 PetscCall(PetscCommRestoreComm(PetscObjectComm((PetscObject)pc), &ex->hcomm));
2742 PetscCall(PetscFree(pc->data));
2743 PetscFunctionReturn(PETSC_SUCCESS);
2744 }
2745
2746 static const char *PFMGRelaxType[] = {"Jacobi", "Weighted-Jacobi", "symmetric-Red/Black-Gauss-Seidel", "Red/Black-Gauss-Seidel"};
2747 static const char *PFMGRAPType[] = {"Galerkin", "non-Galerkin"};
2748
PCView_PFMG(PC pc,PetscViewer viewer)2749 static PetscErrorCode PCView_PFMG(PC pc, PetscViewer viewer)
2750 {
2751 PetscBool isascii;
2752 PC_PFMG *ex = (PC_PFMG *)pc->data;
2753
2754 PetscFunctionBegin;
2755 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
2756 if (isascii) {
2757 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE PFMG preconditioning\n"));
2758 PetscCall(PetscViewerASCIIPrintf(viewer, " max iterations %" PetscInt_FMT "\n", ex->its));
2759 PetscCall(PetscViewerASCIIPrintf(viewer, " tolerance %g\n", ex->tol));
2760 PetscCall(PetscViewerASCIIPrintf(viewer, " relax type %s\n", PFMGRelaxType[ex->relax_type]));
2761 PetscCall(PetscViewerASCIIPrintf(viewer, " RAP type %s\n", PFMGRAPType[ex->rap_type]));
2762 PetscCall(PetscViewerASCIIPrintf(viewer, " number pre-relax %" PetscInt_FMT " post-relax %" PetscInt_FMT "\n", ex->num_pre_relax, ex->num_post_relax));
2763 PetscCall(PetscViewerASCIIPrintf(viewer, " max levels %" PetscInt_FMT "\n", ex->max_levels));
2764 PetscCall(PetscViewerASCIIPrintf(viewer, " skip relax %" PetscInt_FMT "\n", ex->skip_relax));
2765 }
2766 PetscFunctionReturn(PETSC_SUCCESS);
2767 }
2768
PCSetFromOptions_PFMG(PC pc,PetscOptionItems PetscOptionsObject)2769 static PetscErrorCode PCSetFromOptions_PFMG(PC pc, PetscOptionItems PetscOptionsObject)
2770 {
2771 PC_PFMG *ex = (PC_PFMG *)pc->data;
2772
2773 PetscFunctionBegin;
2774 PetscOptionsHeadBegin(PetscOptionsObject, "PFMG options");
2775 PetscCall(PetscOptionsBool("-pc_pfmg_print_statistics", "Print statistics", "HYPRE_StructPFMGSetPrintLevel", ex->print_statistics, &ex->print_statistics, NULL));
2776 PetscCall(PetscOptionsInt("-pc_pfmg_its", "Number of iterations of PFMG to use as preconditioner", "HYPRE_StructPFMGSetMaxIter", ex->its, &ex->its, NULL));
2777 PetscCallHYPRE(HYPRE_StructPFMGSetMaxIter(ex->hsolver, (HYPRE_Int)ex->its));
2778 PetscCall(PetscOptionsInt("-pc_pfmg_num_pre_relax", "Number of smoothing steps before coarse grid", "HYPRE_StructPFMGSetNumPreRelax", ex->num_pre_relax, &ex->num_pre_relax, NULL));
2779 PetscCallHYPRE(HYPRE_StructPFMGSetNumPreRelax(ex->hsolver, (HYPRE_Int)ex->num_pre_relax));
2780 PetscCall(PetscOptionsInt("-pc_pfmg_num_post_relax", "Number of smoothing steps after coarse grid", "HYPRE_StructPFMGSetNumPostRelax", ex->num_post_relax, &ex->num_post_relax, NULL));
2781 PetscCallHYPRE(HYPRE_StructPFMGSetNumPostRelax(ex->hsolver, (HYPRE_Int)ex->num_post_relax));
2782
2783 PetscCall(PetscOptionsInt("-pc_pfmg_max_levels", "Max Levels for MG hierarchy", "HYPRE_StructPFMGSetMaxLevels", ex->max_levels, &ex->max_levels, NULL));
2784 PetscCallHYPRE(HYPRE_StructPFMGSetMaxLevels(ex->hsolver, (HYPRE_Int)ex->max_levels));
2785
2786 PetscCall(PetscOptionsReal("-pc_pfmg_tol", "Tolerance of PFMG", "HYPRE_StructPFMGSetTol", ex->tol, &ex->tol, NULL));
2787 PetscCallHYPRE(HYPRE_StructPFMGSetTol(ex->hsolver, ex->tol));
2788 PetscCall(PetscOptionsEList("-pc_pfmg_relax_type", "Relax type for the up and down cycles", "HYPRE_StructPFMGSetRelaxType", PFMGRelaxType, PETSC_STATIC_ARRAY_LENGTH(PFMGRelaxType), PFMGRelaxType[ex->relax_type], &ex->relax_type, NULL));
2789 PetscCallHYPRE(HYPRE_StructPFMGSetRelaxType(ex->hsolver, (HYPRE_Int)ex->relax_type));
2790 PetscCall(PetscOptionsEList("-pc_pfmg_rap_type", "RAP type", "HYPRE_StructPFMGSetRAPType", PFMGRAPType, PETSC_STATIC_ARRAY_LENGTH(PFMGRAPType), PFMGRAPType[ex->rap_type], &ex->rap_type, NULL));
2791 PetscCallHYPRE(HYPRE_StructPFMGSetRAPType(ex->hsolver, (HYPRE_Int)ex->rap_type));
2792 PetscCall(PetscOptionsInt("-pc_pfmg_skip_relax", "Skip relaxation on certain grids for isotropic problems. This can greatly improve efficiency by eliminating unnecessary relaxations when the underlying problem is isotropic", "HYPRE_StructPFMGSetSkipRelax", ex->skip_relax, &ex->skip_relax, NULL));
2793 PetscCallHYPRE(HYPRE_StructPFMGSetSkipRelax(ex->hsolver, (HYPRE_Int)ex->skip_relax));
2794 PetscOptionsHeadEnd();
2795 PetscFunctionReturn(PETSC_SUCCESS);
2796 }
2797
PCApply_PFMG(PC pc,Vec x,Vec y)2798 static PetscErrorCode PCApply_PFMG(PC pc, Vec x, Vec y)
2799 {
2800 PC_PFMG *ex = (PC_PFMG *)pc->data;
2801 PetscScalar *yy;
2802 const PetscScalar *xx;
2803 PetscInt ilower[3], iupper[3];
2804 HYPRE_Int hlower[3], hupper[3];
2805 Mat_HYPREStruct *mx = (Mat_HYPREStruct *)pc->pmat->data;
2806
2807 PetscFunctionBegin;
2808 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
2809 PetscCall(DMDAGetCorners(mx->da, &ilower[0], &ilower[1], &ilower[2], &iupper[0], &iupper[1], &iupper[2]));
2810 /* when HYPRE_MIXEDINT is defined, sizeof(HYPRE_Int) == 32 */
2811 iupper[0] += ilower[0] - 1;
2812 iupper[1] += ilower[1] - 1;
2813 iupper[2] += ilower[2] - 1;
2814 hlower[0] = (HYPRE_Int)ilower[0];
2815 hlower[1] = (HYPRE_Int)ilower[1];
2816 hlower[2] = (HYPRE_Int)ilower[2];
2817 hupper[0] = (HYPRE_Int)iupper[0];
2818 hupper[1] = (HYPRE_Int)iupper[1];
2819 hupper[2] = (HYPRE_Int)iupper[2];
2820
2821 /* copy x values over to hypre */
2822 PetscCallHYPRE(HYPRE_StructVectorSetConstantValues(mx->hb, 0.0));
2823 PetscCall(VecGetArrayRead(x, &xx));
2824 PetscCallHYPRE(HYPRE_StructVectorSetBoxValues(mx->hb, hlower, hupper, (HYPRE_Complex *)xx));
2825 PetscCall(VecRestoreArrayRead(x, &xx));
2826 PetscCallHYPRE(HYPRE_StructVectorAssemble(mx->hb));
2827 PetscCallHYPRE(HYPRE_StructPFMGSolve(ex->hsolver, mx->hmat, mx->hb, mx->hx));
2828
2829 /* copy solution values back to PETSc */
2830 PetscCall(VecGetArray(y, &yy));
2831 PetscCallHYPRE(HYPRE_StructVectorGetBoxValues(mx->hx, hlower, hupper, (HYPRE_Complex *)yy));
2832 PetscCall(VecRestoreArray(y, &yy));
2833 PetscFunctionReturn(PETSC_SUCCESS);
2834 }
2835
PCApplyRichardson_PFMG(PC pc,Vec b,Vec y,Vec w,PetscReal rtol,PetscReal abstol,PetscReal dtol,PetscInt its,PetscBool guesszero,PetscInt * outits,PCRichardsonConvergedReason * reason)2836 static PetscErrorCode PCApplyRichardson_PFMG(PC pc, Vec b, Vec y, Vec w, PetscReal rtol, PetscReal abstol, PetscReal dtol, PetscInt its, PetscBool guesszero, PetscInt *outits, PCRichardsonConvergedReason *reason)
2837 {
2838 PC_PFMG *jac = (PC_PFMG *)pc->data;
2839 HYPRE_Int oits;
2840
2841 PetscFunctionBegin;
2842 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
2843 PetscCallHYPRE(HYPRE_StructPFMGSetMaxIter(jac->hsolver, (HYPRE_Int)(its * jac->its)));
2844 PetscCallHYPRE(HYPRE_StructPFMGSetTol(jac->hsolver, rtol));
2845
2846 PetscCall(PCApply_PFMG(pc, b, y));
2847 PetscCallHYPRE(HYPRE_StructPFMGGetNumIterations(jac->hsolver, &oits));
2848 *outits = oits;
2849 if (oits == its) *reason = PCRICHARDSON_CONVERGED_ITS;
2850 else *reason = PCRICHARDSON_CONVERGED_RTOL;
2851 PetscCallHYPRE(HYPRE_StructPFMGSetTol(jac->hsolver, jac->tol));
2852 PetscCallHYPRE(HYPRE_StructPFMGSetMaxIter(jac->hsolver, (HYPRE_Int)jac->its));
2853 PetscFunctionReturn(PETSC_SUCCESS);
2854 }
2855
PCSetUp_PFMG(PC pc)2856 static PetscErrorCode PCSetUp_PFMG(PC pc)
2857 {
2858 PC_PFMG *ex = (PC_PFMG *)pc->data;
2859 Mat_HYPREStruct *mx = (Mat_HYPREStruct *)pc->pmat->data;
2860 PetscBool flg;
2861
2862 PetscFunctionBegin;
2863 PetscCall(PetscObjectTypeCompare((PetscObject)pc->pmat, MATHYPRESTRUCT, &flg));
2864 PetscCheck(flg, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_INCOMP, "Must use MATHYPRESTRUCT with this preconditioner");
2865
2866 /* create the hypre solver object and set its information */
2867 if (ex->hsolver) PetscCallHYPRE(HYPRE_StructPFMGDestroy(ex->hsolver));
2868 PetscCallHYPRE(HYPRE_StructPFMGCreate(ex->hcomm, &ex->hsolver));
2869
2870 // Print Hypre statistics about the solve process
2871 if (ex->print_statistics) PetscCallHYPRE(HYPRE_StructPFMGSetPrintLevel(ex->hsolver, 3));
2872
2873 // The hypre options must be repeated here because the StructPFMG was destroyed and recreated
2874 PetscCallHYPRE(HYPRE_StructPFMGSetMaxIter(ex->hsolver, (HYPRE_Int)ex->its));
2875 PetscCallHYPRE(HYPRE_StructPFMGSetNumPreRelax(ex->hsolver, (HYPRE_Int)ex->num_pre_relax));
2876 PetscCallHYPRE(HYPRE_StructPFMGSetNumPostRelax(ex->hsolver, (HYPRE_Int)ex->num_post_relax));
2877 PetscCallHYPRE(HYPRE_StructPFMGSetMaxLevels(ex->hsolver, (HYPRE_Int)ex->max_levels));
2878 PetscCallHYPRE(HYPRE_StructPFMGSetTol(ex->hsolver, ex->tol));
2879 PetscCallHYPRE(HYPRE_StructPFMGSetRelaxType(ex->hsolver, (HYPRE_Int)ex->relax_type));
2880 PetscCallHYPRE(HYPRE_StructPFMGSetRAPType(ex->hsolver, (HYPRE_Int)ex->rap_type));
2881
2882 PetscCallHYPRE(HYPRE_StructPFMGSetup(ex->hsolver, mx->hmat, mx->hb, mx->hx));
2883 PetscCallHYPRE(HYPRE_StructPFMGSetZeroGuess(ex->hsolver));
2884 PetscFunctionReturn(PETSC_SUCCESS);
2885 }
2886
2887 /*MC
2888 PCPFMG - the hypre PFMG multigrid solver
2889
2890 Options Database Keys:
2891 + -pc_pfmg_its <its> - number of iterations of PFMG to use as preconditioner
2892 . -pc_pfmg_num_pre_relax <steps> - number of smoothing steps before coarse grid solve
2893 . -pc_pfmg_num_post_relax <steps> - number of smoothing steps after coarse grid solve
2894 . -pc_pfmg_tol <tol> - tolerance of PFMG
2895 . -pc_pfmg_relax_type - relaxation type for the up and down cycles, one of Jacobi,Weighted-Jacobi,symmetric-Red/Black-Gauss-Seidel,Red/Black-Gauss-Seidel
2896 . -pc_pfmg_rap_type - type of coarse matrix generation, one of Galerkin,non-Galerkin
2897 - -pc_pfmg_skip_relax - skip relaxation on certain grids for isotropic problems. This can greatly improve efficiency by eliminating unnecessary relaxations
2898 when the underlying problem is isotropic, one of 0,1
2899
2900 Level: advanced
2901
2902 Notes:
2903 This is for CELL-centered descretizations
2904
2905 See `PCSYSPFMG` for a version suitable for systems of PDEs, and `PCSMG`
2906
2907 See `PCHYPRE` for hypre's BoomerAMG algebraic multigrid solver
2908
2909 This must be used with the `MATHYPRESTRUCT` matrix type.
2910
2911 This provides only some of the functionality of PFMG, it supports only one block per process defined by a PETSc `DMDA`.
2912
2913 .seealso: [](ch_ksp), `PCMG`, `MATHYPRESTRUCT`, `PCHYPRE`, `PCGAMG`, `PCSYSPFMG`, `PCSMG`
2914 M*/
2915
PCCreate_PFMG(PC pc)2916 PETSC_EXTERN PetscErrorCode PCCreate_PFMG(PC pc)
2917 {
2918 PC_PFMG *ex;
2919
2920 PetscFunctionBegin;
2921 PetscCall(PetscNew(&ex));
2922 pc->data = ex;
2923
2924 ex->its = 1;
2925 ex->tol = 1.e-8;
2926 ex->relax_type = 1;
2927 ex->rap_type = 0;
2928 ex->num_pre_relax = 1;
2929 ex->num_post_relax = 1;
2930 ex->max_levels = 0;
2931 ex->skip_relax = 0;
2932 ex->print_statistics = PETSC_FALSE;
2933
2934 pc->ops->setfromoptions = PCSetFromOptions_PFMG;
2935 pc->ops->view = PCView_PFMG;
2936 pc->ops->destroy = PCDestroy_PFMG;
2937 pc->ops->apply = PCApply_PFMG;
2938 pc->ops->applyrichardson = PCApplyRichardson_PFMG;
2939 pc->ops->setup = PCSetUp_PFMG;
2940
2941 PetscCall(PetscCommGetComm(PetscObjectComm((PetscObject)pc), &ex->hcomm));
2942 PetscCall(PetscHYPREInitialize());
2943 PetscCallHYPRE(HYPRE_StructPFMGCreate(ex->hcomm, &ex->hsolver));
2944 PetscFunctionReturn(PETSC_SUCCESS);
2945 }
2946
2947 /* we know we are working with a HYPRE_SStructMatrix */
2948 typedef struct {
2949 MPI_Comm hcomm; /* does not share comm with HYPRE_SStructMatrix because need to create solver before getting matrix */
2950 HYPRE_SStructSolver ss_solver;
2951
2952 /* keep copy of SYSPFMG options used so may view them */
2953 PetscInt its;
2954 PetscReal tol;
2955 PetscInt relax_type;
2956 PetscInt num_pre_relax, num_post_relax;
2957 } PC_SysPFMG;
2958
PCDestroy_SysPFMG(PC pc)2959 static PetscErrorCode PCDestroy_SysPFMG(PC pc)
2960 {
2961 PC_SysPFMG *ex = (PC_SysPFMG *)pc->data;
2962
2963 PetscFunctionBegin;
2964 if (ex->ss_solver) PetscCallHYPRE(HYPRE_SStructSysPFMGDestroy(ex->ss_solver));
2965 PetscCall(PetscCommRestoreComm(PetscObjectComm((PetscObject)pc), &ex->hcomm));
2966 PetscCall(PetscFree(pc->data));
2967 PetscFunctionReturn(PETSC_SUCCESS);
2968 }
2969
2970 static const char *SysPFMGRelaxType[] = {"Weighted-Jacobi", "Red/Black-Gauss-Seidel"};
2971
PCView_SysPFMG(PC pc,PetscViewer viewer)2972 static PetscErrorCode PCView_SysPFMG(PC pc, PetscViewer viewer)
2973 {
2974 PetscBool isascii;
2975 PC_SysPFMG *ex = (PC_SysPFMG *)pc->data;
2976
2977 PetscFunctionBegin;
2978 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
2979 if (isascii) {
2980 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE SysPFMG preconditioning\n"));
2981 PetscCall(PetscViewerASCIIPrintf(viewer, " max iterations %" PetscInt_FMT "\n", ex->its));
2982 PetscCall(PetscViewerASCIIPrintf(viewer, " tolerance %g\n", ex->tol));
2983 PetscCall(PetscViewerASCIIPrintf(viewer, " relax type %s\n", PFMGRelaxType[ex->relax_type]));
2984 PetscCall(PetscViewerASCIIPrintf(viewer, " number pre-relax %" PetscInt_FMT " post-relax %" PetscInt_FMT "\n", ex->num_pre_relax, ex->num_post_relax));
2985 }
2986 PetscFunctionReturn(PETSC_SUCCESS);
2987 }
2988
PCSetFromOptions_SysPFMG(PC pc,PetscOptionItems PetscOptionsObject)2989 static PetscErrorCode PCSetFromOptions_SysPFMG(PC pc, PetscOptionItems PetscOptionsObject)
2990 {
2991 PC_SysPFMG *ex = (PC_SysPFMG *)pc->data;
2992 PetscBool flg = PETSC_FALSE;
2993
2994 PetscFunctionBegin;
2995 PetscOptionsHeadBegin(PetscOptionsObject, "SysPFMG options");
2996 PetscCall(PetscOptionsBool("-pc_syspfmg_print_statistics", "Print statistics", "HYPRE_SStructSysPFMGSetPrintLevel", flg, &flg, NULL));
2997 if (flg) PetscCallHYPRE(HYPRE_SStructSysPFMGSetPrintLevel(ex->ss_solver, 3));
2998 PetscCall(PetscOptionsInt("-pc_syspfmg_its", "Number of iterations of SysPFMG to use as preconditioner", "HYPRE_SStructSysPFMGSetMaxIter", ex->its, &ex->its, NULL));
2999 PetscCallHYPRE(HYPRE_SStructSysPFMGSetMaxIter(ex->ss_solver, (HYPRE_Int)ex->its));
3000 PetscCall(PetscOptionsInt("-pc_syspfmg_num_pre_relax", "Number of smoothing steps before coarse grid", "HYPRE_SStructSysPFMGSetNumPreRelax", ex->num_pre_relax, &ex->num_pre_relax, NULL));
3001 PetscCallHYPRE(HYPRE_SStructSysPFMGSetNumPreRelax(ex->ss_solver, (HYPRE_Int)ex->num_pre_relax));
3002 PetscCall(PetscOptionsInt("-pc_syspfmg_num_post_relax", "Number of smoothing steps after coarse grid", "HYPRE_SStructSysPFMGSetNumPostRelax", ex->num_post_relax, &ex->num_post_relax, NULL));
3003 PetscCallHYPRE(HYPRE_SStructSysPFMGSetNumPostRelax(ex->ss_solver, (HYPRE_Int)ex->num_post_relax));
3004
3005 PetscCall(PetscOptionsReal("-pc_syspfmg_tol", "Tolerance of SysPFMG", "HYPRE_SStructSysPFMGSetTol", ex->tol, &ex->tol, NULL));
3006 PetscCallHYPRE(HYPRE_SStructSysPFMGSetTol(ex->ss_solver, ex->tol));
3007 PetscCall(PetscOptionsEList("-pc_syspfmg_relax_type", "Relax type for the up and down cycles", "HYPRE_SStructSysPFMGSetRelaxType", SysPFMGRelaxType, PETSC_STATIC_ARRAY_LENGTH(SysPFMGRelaxType), SysPFMGRelaxType[ex->relax_type], &ex->relax_type, NULL));
3008 PetscCallHYPRE(HYPRE_SStructSysPFMGSetRelaxType(ex->ss_solver, (HYPRE_Int)ex->relax_type));
3009 PetscOptionsHeadEnd();
3010 PetscFunctionReturn(PETSC_SUCCESS);
3011 }
3012
PCApply_SysPFMG(PC pc,Vec x,Vec y)3013 static PetscErrorCode PCApply_SysPFMG(PC pc, Vec x, Vec y)
3014 {
3015 PC_SysPFMG *ex = (PC_SysPFMG *)pc->data;
3016 PetscScalar *yy;
3017 const PetscScalar *xx;
3018 PetscInt ilower[3], iupper[3];
3019 HYPRE_Int hlower[3], hupper[3];
3020 Mat_HYPRESStruct *mx = (Mat_HYPRESStruct *)pc->pmat->data;
3021 PetscInt ordering = mx->dofs_order;
3022 PetscInt nvars = mx->nvars;
3023 HYPRE_Int part = 0;
3024 PetscInt size;
3025 PetscInt i;
3026
3027 PetscFunctionBegin;
3028 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
3029 PetscCall(DMDAGetCorners(mx->da, &ilower[0], &ilower[1], &ilower[2], &iupper[0], &iupper[1], &iupper[2]));
3030 /* when HYPRE_MIXEDINT is defined, sizeof(HYPRE_Int) == 32 */
3031 iupper[0] += ilower[0] - 1;
3032 iupper[1] += ilower[1] - 1;
3033 iupper[2] += ilower[2] - 1;
3034 hlower[0] = (HYPRE_Int)ilower[0];
3035 hlower[1] = (HYPRE_Int)ilower[1];
3036 hlower[2] = (HYPRE_Int)ilower[2];
3037 hupper[0] = (HYPRE_Int)iupper[0];
3038 hupper[1] = (HYPRE_Int)iupper[1];
3039 hupper[2] = (HYPRE_Int)iupper[2];
3040
3041 size = 1;
3042 for (i = 0; i < 3; i++) size *= (iupper[i] - ilower[i] + 1);
3043
3044 /* copy x values over to hypre for variable ordering */
3045 if (ordering) {
3046 PetscCallHYPRE(HYPRE_SStructVectorSetConstantValues(mx->ss_b, 0.0));
3047 PetscCall(VecGetArrayRead(x, &xx));
3048 for (i = 0; i < nvars; i++) PetscCallHYPRE(HYPRE_SStructVectorSetBoxValues(mx->ss_b, part, hlower, hupper, (HYPRE_Int)i, (HYPRE_Complex *)(xx + (size * i))));
3049 PetscCall(VecRestoreArrayRead(x, &xx));
3050 PetscCallHYPRE(HYPRE_SStructVectorAssemble(mx->ss_b));
3051 PetscCallHYPRE(HYPRE_SStructMatrixMatvec(1.0, mx->ss_mat, mx->ss_b, 0.0, mx->ss_x));
3052 PetscCallHYPRE(HYPRE_SStructSysPFMGSolve(ex->ss_solver, mx->ss_mat, mx->ss_b, mx->ss_x));
3053
3054 /* copy solution values back to PETSc */
3055 PetscCall(VecGetArray(y, &yy));
3056 for (i = 0; i < nvars; i++) PetscCallHYPRE(HYPRE_SStructVectorGetBoxValues(mx->ss_x, part, hlower, hupper, (HYPRE_Int)i, (HYPRE_Complex *)(yy + (size * i))));
3057 PetscCall(VecRestoreArray(y, &yy));
3058 } else { /* nodal ordering must be mapped to variable ordering for sys_pfmg */
3059 PetscScalar *z;
3060 PetscInt j, k;
3061
3062 PetscCall(PetscMalloc1(nvars * size, &z));
3063 PetscCallHYPRE(HYPRE_SStructVectorSetConstantValues(mx->ss_b, 0.0));
3064 PetscCall(VecGetArrayRead(x, &xx));
3065
3066 /* transform nodal to hypre's variable ordering for sys_pfmg */
3067 for (i = 0; i < size; i++) {
3068 k = i * nvars;
3069 for (j = 0; j < nvars; j++) z[j * size + i] = xx[k + j];
3070 }
3071 for (i = 0; i < nvars; i++) PetscCallHYPRE(HYPRE_SStructVectorSetBoxValues(mx->ss_b, part, hlower, hupper, (HYPRE_Int)i, (HYPRE_Complex *)(z + (size * i))));
3072 PetscCall(VecRestoreArrayRead(x, &xx));
3073 PetscCallHYPRE(HYPRE_SStructVectorAssemble(mx->ss_b));
3074 PetscCallHYPRE(HYPRE_SStructSysPFMGSolve(ex->ss_solver, mx->ss_mat, mx->ss_b, mx->ss_x));
3075
3076 /* copy solution values back to PETSc */
3077 PetscCall(VecGetArray(y, &yy));
3078 for (i = 0; i < nvars; i++) PetscCallHYPRE(HYPRE_SStructVectorGetBoxValues(mx->ss_x, part, hlower, hupper, (HYPRE_Int)i, (HYPRE_Complex *)(z + (size * i))));
3079 /* transform hypre's variable ordering for sys_pfmg to nodal ordering */
3080 for (i = 0; i < size; i++) {
3081 k = i * nvars;
3082 for (j = 0; j < nvars; j++) yy[k + j] = z[j * size + i];
3083 }
3084 PetscCall(VecRestoreArray(y, &yy));
3085 PetscCall(PetscFree(z));
3086 }
3087 PetscFunctionReturn(PETSC_SUCCESS);
3088 }
3089
PCApplyRichardson_SysPFMG(PC pc,Vec b,Vec y,Vec w,PetscReal rtol,PetscReal abstol,PetscReal dtol,PetscInt its,PetscBool guesszero,PetscInt * outits,PCRichardsonConvergedReason * reason)3090 static PetscErrorCode PCApplyRichardson_SysPFMG(PC pc, Vec b, Vec y, Vec w, PetscReal rtol, PetscReal abstol, PetscReal dtol, PetscInt its, PetscBool guesszero, PetscInt *outits, PCRichardsonConvergedReason *reason)
3091 {
3092 PC_SysPFMG *jac = (PC_SysPFMG *)pc->data;
3093 HYPRE_Int oits;
3094
3095 PetscFunctionBegin;
3096 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
3097 PetscCallHYPRE(HYPRE_SStructSysPFMGSetMaxIter(jac->ss_solver, (HYPRE_Int)(its * jac->its)));
3098 PetscCallHYPRE(HYPRE_SStructSysPFMGSetTol(jac->ss_solver, rtol));
3099 PetscCall(PCApply_SysPFMG(pc, b, y));
3100 PetscCallHYPRE(HYPRE_SStructSysPFMGGetNumIterations(jac->ss_solver, &oits));
3101 *outits = oits;
3102 if (oits == its) *reason = PCRICHARDSON_CONVERGED_ITS;
3103 else *reason = PCRICHARDSON_CONVERGED_RTOL;
3104 PetscCallHYPRE(HYPRE_SStructSysPFMGSetTol(jac->ss_solver, jac->tol));
3105 PetscCallHYPRE(HYPRE_SStructSysPFMGSetMaxIter(jac->ss_solver, (HYPRE_Int)jac->its));
3106 PetscFunctionReturn(PETSC_SUCCESS);
3107 }
3108
PCSetUp_SysPFMG(PC pc)3109 static PetscErrorCode PCSetUp_SysPFMG(PC pc)
3110 {
3111 PC_SysPFMG *ex = (PC_SysPFMG *)pc->data;
3112 Mat_HYPRESStruct *mx = (Mat_HYPRESStruct *)pc->pmat->data;
3113 PetscBool flg;
3114
3115 PetscFunctionBegin;
3116 PetscCall(PetscObjectTypeCompare((PetscObject)pc->pmat, MATHYPRESSTRUCT, &flg));
3117 PetscCheck(flg, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_INCOMP, "Must use MATHYPRESSTRUCT with this preconditioner");
3118
3119 /* create the hypre sstruct solver object and set its information */
3120 if (ex->ss_solver) PetscCallHYPRE(HYPRE_SStructSysPFMGDestroy(ex->ss_solver));
3121 PetscCallHYPRE(HYPRE_SStructSysPFMGCreate(ex->hcomm, &ex->ss_solver));
3122 PetscCallHYPRE(HYPRE_SStructSysPFMGSetZeroGuess(ex->ss_solver));
3123 PetscCallHYPRE(HYPRE_SStructSysPFMGSetup(ex->ss_solver, mx->ss_mat, mx->ss_b, mx->ss_x));
3124 PetscFunctionReturn(PETSC_SUCCESS);
3125 }
3126
3127 /*MC
3128 PCSYSPFMG - the hypre SysPFMG multigrid solver
3129
3130 Level: advanced
3131
3132 Options Database Keys:
3133 + -pc_syspfmg_its <its> - number of iterations of SysPFMG to use as preconditioner
3134 . -pc_syspfmg_num_pre_relax <steps> - number of smoothing steps before coarse grid
3135 . -pc_syspfmg_num_post_relax <steps> - number of smoothing steps after coarse grid
3136 . -pc_syspfmg_tol <tol> - tolerance of SysPFMG
3137 - -pc_syspfmg_relax_type <Weighted-Jacobi,Red/Black-Gauss-Seidel> - relaxation type for the up and down cycles
3138
3139 Notes:
3140 See `PCPFMG` for hypre's PFMG that works for a scalar PDE and `PCSMG`
3141
3142 See `PCHYPRE` for hypre's BoomerAMG algebraic multigrid solver
3143
3144 This is for CELL-centered descretizations
3145
3146 This must be used with the `MATHYPRESSTRUCT` matrix type.
3147
3148 This does not give access to all the functionality of hypres SysPFMG, it supports only one part, and one block per process defined by a PETSc `DMDA`.
3149
3150 .seealso: [](ch_ksp), `PCMG`, `MATHYPRESSTRUCT`, `PCPFMG`, `PCHYPRE`, `PCGAMG`, `PCSMG`
3151 M*/
3152
PCCreate_SysPFMG(PC pc)3153 PETSC_EXTERN PetscErrorCode PCCreate_SysPFMG(PC pc)
3154 {
3155 PC_SysPFMG *ex;
3156
3157 PetscFunctionBegin;
3158 PetscCall(PetscNew(&ex));
3159 pc->data = ex;
3160
3161 ex->its = 1;
3162 ex->tol = 1.e-8;
3163 ex->relax_type = 1;
3164 ex->num_pre_relax = 1;
3165 ex->num_post_relax = 1;
3166
3167 pc->ops->setfromoptions = PCSetFromOptions_SysPFMG;
3168 pc->ops->view = PCView_SysPFMG;
3169 pc->ops->destroy = PCDestroy_SysPFMG;
3170 pc->ops->apply = PCApply_SysPFMG;
3171 pc->ops->applyrichardson = PCApplyRichardson_SysPFMG;
3172 pc->ops->setup = PCSetUp_SysPFMG;
3173
3174 PetscCall(PetscCommGetComm(PetscObjectComm((PetscObject)pc), &ex->hcomm));
3175 PetscCall(PetscHYPREInitialize());
3176 PetscCallHYPRE(HYPRE_SStructSysPFMGCreate(ex->hcomm, &ex->ss_solver));
3177 PetscFunctionReturn(PETSC_SUCCESS);
3178 }
3179
3180 /* PC SMG */
3181 typedef struct {
3182 MPI_Comm hcomm; /* does not share comm with HYPRE_StructMatrix because need to create solver before getting matrix */
3183 HYPRE_StructSolver hsolver;
3184 PetscInt its; /* keep copy of SMG options used so may view them */
3185 PetscReal tol;
3186 PetscBool print_statistics;
3187 PetscInt num_pre_relax, num_post_relax;
3188 } PC_SMG;
3189
PCDestroy_SMG(PC pc)3190 static PetscErrorCode PCDestroy_SMG(PC pc)
3191 {
3192 PC_SMG *ex = (PC_SMG *)pc->data;
3193
3194 PetscFunctionBegin;
3195 if (ex->hsolver) PetscCallHYPRE(HYPRE_StructSMGDestroy(ex->hsolver));
3196 PetscCall(PetscCommRestoreComm(PetscObjectComm((PetscObject)pc), &ex->hcomm));
3197 PetscCall(PetscFree(pc->data));
3198 PetscFunctionReturn(PETSC_SUCCESS);
3199 }
3200
PCView_SMG(PC pc,PetscViewer viewer)3201 static PetscErrorCode PCView_SMG(PC pc, PetscViewer viewer)
3202 {
3203 PetscBool isascii;
3204 PC_SMG *ex = (PC_SMG *)pc->data;
3205
3206 PetscFunctionBegin;
3207 PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
3208 if (isascii) {
3209 PetscCall(PetscViewerASCIIPrintf(viewer, " HYPRE SMG preconditioning\n"));
3210 PetscCall(PetscViewerASCIIPrintf(viewer, " max iterations %" PetscInt_FMT "\n", ex->its));
3211 PetscCall(PetscViewerASCIIPrintf(viewer, " tolerance %g\n", ex->tol));
3212 PetscCall(PetscViewerASCIIPrintf(viewer, " number pre-relax %" PetscInt_FMT " post-relax %" PetscInt_FMT "\n", ex->num_pre_relax, ex->num_post_relax));
3213 }
3214 PetscFunctionReturn(PETSC_SUCCESS);
3215 }
3216
PCSetFromOptions_SMG(PC pc,PetscOptionItems PetscOptionsObject)3217 static PetscErrorCode PCSetFromOptions_SMG(PC pc, PetscOptionItems PetscOptionsObject)
3218 {
3219 PC_SMG *ex = (PC_SMG *)pc->data;
3220
3221 PetscFunctionBegin;
3222 PetscOptionsHeadBegin(PetscOptionsObject, "SMG options");
3223
3224 PetscCall(PetscOptionsInt("-pc_smg_its", "Number of iterations of SMG to use as preconditioner", "HYPRE_StructSMGSetMaxIter", ex->its, &ex->its, NULL));
3225 PetscCall(PetscOptionsInt("-pc_smg_num_pre_relax", "Number of smoothing steps before coarse grid", "HYPRE_StructSMGSetNumPreRelax", ex->num_pre_relax, &ex->num_pre_relax, NULL));
3226 PetscCall(PetscOptionsInt("-pc_smg_num_post_relax", "Number of smoothing steps after coarse grid", "HYPRE_StructSMGSetNumPostRelax", ex->num_post_relax, &ex->num_post_relax, NULL));
3227 PetscCall(PetscOptionsReal("-pc_smg_tol", "Tolerance of SMG", "HYPRE_StructSMGSetTol", ex->tol, &ex->tol, NULL));
3228
3229 PetscOptionsHeadEnd();
3230 PetscFunctionReturn(PETSC_SUCCESS);
3231 }
3232
PCApply_SMG(PC pc,Vec x,Vec y)3233 static PetscErrorCode PCApply_SMG(PC pc, Vec x, Vec y)
3234 {
3235 PC_SMG *ex = (PC_SMG *)pc->data;
3236 PetscScalar *yy;
3237 const PetscScalar *xx;
3238 PetscInt ilower[3], iupper[3];
3239 HYPRE_Int hlower[3], hupper[3];
3240 Mat_HYPREStruct *mx = (Mat_HYPREStruct *)pc->pmat->data;
3241
3242 PetscFunctionBegin;
3243 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
3244 PetscCall(DMDAGetCorners(mx->da, &ilower[0], &ilower[1], &ilower[2], &iupper[0], &iupper[1], &iupper[2]));
3245 /* when HYPRE_MIXEDINT is defined, sizeof(HYPRE_Int) == 32 */
3246 iupper[0] += ilower[0] - 1;
3247 iupper[1] += ilower[1] - 1;
3248 iupper[2] += ilower[2] - 1;
3249 hlower[0] = (HYPRE_Int)ilower[0];
3250 hlower[1] = (HYPRE_Int)ilower[1];
3251 hlower[2] = (HYPRE_Int)ilower[2];
3252 hupper[0] = (HYPRE_Int)iupper[0];
3253 hupper[1] = (HYPRE_Int)iupper[1];
3254 hupper[2] = (HYPRE_Int)iupper[2];
3255
3256 /* copy x values over to hypre */
3257 PetscCallHYPRE(HYPRE_StructVectorSetConstantValues(mx->hb, 0.0));
3258 PetscCall(VecGetArrayRead(x, &xx));
3259 PetscCallHYPRE(HYPRE_StructVectorSetBoxValues(mx->hb, hlower, hupper, (HYPRE_Complex *)xx));
3260 PetscCall(VecRestoreArrayRead(x, &xx));
3261 PetscCallHYPRE(HYPRE_StructVectorAssemble(mx->hb));
3262 PetscCallHYPRE(HYPRE_StructSMGSolve(ex->hsolver, mx->hmat, mx->hb, mx->hx));
3263
3264 /* copy solution values back to PETSc */
3265 PetscCall(VecGetArray(y, &yy));
3266 PetscCallHYPRE(HYPRE_StructVectorGetBoxValues(mx->hx, hlower, hupper, (HYPRE_Complex *)yy));
3267 PetscCall(VecRestoreArray(y, &yy));
3268 PetscFunctionReturn(PETSC_SUCCESS);
3269 }
3270
PCApplyRichardson_SMG(PC pc,Vec b,Vec y,Vec w,PetscReal rtol,PetscReal abstol,PetscReal dtol,PetscInt its,PetscBool guesszero,PetscInt * outits,PCRichardsonConvergedReason * reason)3271 static PetscErrorCode PCApplyRichardson_SMG(PC pc, Vec b, Vec y, Vec w, PetscReal rtol, PetscReal abstol, PetscReal dtol, PetscInt its, PetscBool guesszero, PetscInt *outits, PCRichardsonConvergedReason *reason)
3272 {
3273 PC_SMG *jac = (PC_SMG *)pc->data;
3274 HYPRE_Int oits;
3275
3276 PetscFunctionBegin;
3277 PetscCall(PetscCitationsRegister(hypreCitation, &cite));
3278 PetscCallHYPRE(HYPRE_StructSMGSetMaxIter(jac->hsolver, (HYPRE_Int)(its * jac->its)));
3279 PetscCallHYPRE(HYPRE_StructSMGSetTol(jac->hsolver, rtol));
3280
3281 PetscCall(PCApply_SMG(pc, b, y));
3282 PetscCallHYPRE(HYPRE_StructSMGGetNumIterations(jac->hsolver, &oits));
3283 *outits = oits;
3284 if (oits == its) *reason = PCRICHARDSON_CONVERGED_ITS;
3285 else *reason = PCRICHARDSON_CONVERGED_RTOL;
3286 PetscCallHYPRE(HYPRE_StructSMGSetTol(jac->hsolver, jac->tol));
3287 PetscCallHYPRE(HYPRE_StructSMGSetMaxIter(jac->hsolver, (HYPRE_Int)jac->its));
3288 PetscFunctionReturn(PETSC_SUCCESS);
3289 }
3290
PCSetUp_SMG(PC pc)3291 static PetscErrorCode PCSetUp_SMG(PC pc)
3292 {
3293 PetscInt i, dim;
3294 PC_SMG *ex = (PC_SMG *)pc->data;
3295 Mat_HYPREStruct *mx = (Mat_HYPREStruct *)pc->pmat->data;
3296 PetscBool flg;
3297 DMBoundaryType p[3];
3298 PetscInt M[3];
3299
3300 PetscFunctionBegin;
3301 PetscCall(PetscObjectTypeCompare((PetscObject)pc->pmat, MATHYPRESTRUCT, &flg));
3302 PetscCheck(flg, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_INCOMP, "Must use MATHYPRESTRUCT with this preconditioner");
3303
3304 PetscCall(DMDAGetInfo(mx->da, &dim, &M[0], &M[1], &M[2], 0, 0, 0, 0, 0, &p[0], &p[1], &p[2], 0));
3305 // Check if power of 2 in periodic directions
3306 for (i = 0; i < dim; i++) {
3307 PetscCheck((M[i] & (M[i] - 1)) == 0 || p[i] != DM_BOUNDARY_PERIODIC, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_INCOMP, "With SMG, the number of points in a periodic direction must be a power of 2, but is here %" PetscInt_FMT ".", M[i]);
3308 }
3309
3310 /* create the hypre solver object and set its information */
3311 if (ex->hsolver) PetscCallHYPRE(HYPRE_StructSMGDestroy(ex->hsolver));
3312 PetscCallHYPRE(HYPRE_StructSMGCreate(ex->hcomm, &ex->hsolver));
3313 // The hypre options must be set here and not in SetFromOptions because it is created here!
3314 PetscCallHYPRE(HYPRE_StructSMGSetMaxIter(ex->hsolver, (HYPRE_Int)ex->its));
3315 PetscCallHYPRE(HYPRE_StructSMGSetNumPreRelax(ex->hsolver, (HYPRE_Int)ex->num_pre_relax));
3316 PetscCallHYPRE(HYPRE_StructSMGSetNumPostRelax(ex->hsolver, (HYPRE_Int)ex->num_post_relax));
3317 PetscCallHYPRE(HYPRE_StructSMGSetTol(ex->hsolver, ex->tol));
3318
3319 PetscCallHYPRE(HYPRE_StructSMGSetup(ex->hsolver, mx->hmat, mx->hb, mx->hx));
3320 PetscCallHYPRE(HYPRE_StructSMGSetZeroGuess(ex->hsolver));
3321 PetscFunctionReturn(PETSC_SUCCESS);
3322 }
3323
3324 /*MC
3325 PCSMG - the hypre (structured grid) SMG multigrid solver
3326
3327 Level: advanced
3328
3329 Options Database Keys:
3330 + -pc_smg_its <its> - number of iterations of SMG to use as preconditioner
3331 . -pc_smg_num_pre_relax <steps> - number of smoothing steps before coarse grid
3332 . -pc_smg_num_post_relax <steps> - number of smoothing steps after coarse grid
3333 - -pc_smg_tol <tol> - tolerance of SMG
3334
3335 Notes:
3336 This is for CELL-centered descretizations
3337
3338 This must be used with the `MATHYPRESTRUCT` `MatType`.
3339
3340 This does not provide all the functionality of hypre's SMG solver, it supports only one block per process defined by a PETSc `DMDA`.
3341
3342 See `PCSYSPFMG`, `PCSMG`, `PCPFMG`, and `PCHYPRE` for access to hypre's other preconditioners
3343
3344 .seealso: `PCMG`, `MATHYPRESTRUCT`, `PCPFMG`, `PCSYSPFMG`, `PCHYPRE`, `PCGAMG`
3345 M*/
3346
PCCreate_SMG(PC pc)3347 PETSC_EXTERN PetscErrorCode PCCreate_SMG(PC pc)
3348 {
3349 PC_SMG *ex;
3350
3351 PetscFunctionBegin;
3352 PetscCall(PetscNew(&ex));
3353 pc->data = ex;
3354
3355 ex->its = 1;
3356 ex->tol = 1.e-8;
3357 ex->num_pre_relax = 1;
3358 ex->num_post_relax = 1;
3359
3360 pc->ops->setfromoptions = PCSetFromOptions_SMG;
3361 pc->ops->view = PCView_SMG;
3362 pc->ops->destroy = PCDestroy_SMG;
3363 pc->ops->apply = PCApply_SMG;
3364 pc->ops->applyrichardson = PCApplyRichardson_SMG;
3365 pc->ops->setup = PCSetUp_SMG;
3366
3367 PetscCall(PetscCommGetComm(PetscObjectComm((PetscObject)pc), &ex->hcomm));
3368 PetscCall(PetscHYPREInitialize());
3369 PetscCallHYPRE(HYPRE_StructSMGCreate(ex->hcomm, &ex->hsolver));
3370 PetscFunctionReturn(PETSC_SUCCESS);
3371 }
3372