xref: /petsc/src/ksp/pc/impls/hpddm/pchpddm.cxx (revision eea1a2f1f05c74439d09816deccded0bd3d94cfc)
101e3c840SPierre Jolivet #include <petscsf.h>
213044ca3SPierre Jolivet #include <petsc/private/vecimpl.h>
3f1580f4eSBarry Smith #include <petsc/private/matimpl.h>
4f1580f4eSBarry Smith #include <petsc/private/petschpddm.h> /*I "petscpc.h" I*/
5e905f78bSJacob Faibussowitsch #include <petsc/private/pcimpl.h>
6e905f78bSJacob Faibussowitsch #include <petsc/private/dmimpl.h> /* this must be included after petschpddm.h so that DM_MAX_WORK_VECTORS is not defined  */
7f1580f4eSBarry Smith                                   /* otherwise, it is assumed that one is compiling libhpddm_petsc => circular dependency */
8f1580f4eSBarry Smith 
9db4a47b3SPierre Jolivet static PetscErrorCode (*loadedSym)(HPDDM::Schwarz<PetscScalar> *const, IS, Mat, Mat, Mat, std::vector<Vec>, PC_HPDDM_Level **const) = nullptr;
10f1580f4eSBarry Smith 
11f1580f4eSBarry Smith static PetscBool PCHPDDMPackageInitialized = PETSC_FALSE;
12f1580f4eSBarry Smith 
13f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_Strc;
14f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_PtAP;
15f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_PtBP;
16f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_Next;
17f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_SetUp[PETSC_PCHPDDM_MAXLEVELS];
18f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_Solve[PETSC_PCHPDDM_MAXLEVELS];
19f1580f4eSBarry Smith 
20aa1539e9SPierre Jolivet const char *const PCHPDDMCoarseCorrectionTypes[] = {"DEFLATED", "ADDITIVE", "BALANCED", "NONE", "PCHPDDMCoarseCorrectionType", "PC_HPDDM_COARSE_CORRECTION_", nullptr};
2113044ca3SPierre Jolivet const char *const PCHPDDMSchurPreTypes[]         = {"LEAST_SQUARES", "GENEO", "PCHPDDMSchurPreType", "PC_HPDDM_SCHUR_PRE", nullptr};
22f1580f4eSBarry Smith 
23d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCReset_HPDDM(PC pc)
24d71ae5a4SJacob Faibussowitsch {
25f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
26f1580f4eSBarry Smith 
27f1580f4eSBarry Smith   PetscFunctionBegin;
28f1580f4eSBarry Smith   if (data->levels) {
29811e8887SPierre Jolivet     for (PetscInt i = 0; i < PETSC_PCHPDDM_MAXLEVELS && data->levels[i]; ++i) {
30f1580f4eSBarry Smith       PetscCall(KSPDestroy(&data->levels[i]->ksp));
31f1580f4eSBarry Smith       PetscCall(PCDestroy(&data->levels[i]->pc));
32f1580f4eSBarry Smith       PetscCall(PetscFree(data->levels[i]));
33f1580f4eSBarry Smith     }
34f1580f4eSBarry Smith     PetscCall(PetscFree(data->levels));
35f1580f4eSBarry Smith   }
36f1580f4eSBarry Smith   PetscCall(ISDestroy(&data->is));
37f1580f4eSBarry Smith   PetscCall(MatDestroy(&data->aux));
38f1580f4eSBarry Smith   PetscCall(MatDestroy(&data->B));
39f1580f4eSBarry Smith   PetscCall(VecDestroy(&data->normal));
40f1580f4eSBarry Smith   data->correction = PC_HPDDM_COARSE_CORRECTION_DEFLATED;
41c8ea6600SPierre Jolivet   data->Neumann    = PETSC_BOOL3_UNKNOWN;
42f1580f4eSBarry Smith   data->deflation  = PETSC_FALSE;
43db4a47b3SPierre Jolivet   data->setup      = nullptr;
44db4a47b3SPierre Jolivet   data->setup_ctx  = nullptr;
453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
46f1580f4eSBarry Smith }
47f1580f4eSBarry Smith 
48d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCDestroy_HPDDM(PC pc)
49d71ae5a4SJacob Faibussowitsch {
50f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
51f1580f4eSBarry Smith 
52f1580f4eSBarry Smith   PetscFunctionBegin;
53f1580f4eSBarry Smith   PetscCall(PCReset_HPDDM(pc));
54f1580f4eSBarry Smith   PetscCall(PetscFree(data));
55db4a47b3SPierre Jolivet   PetscCall(PetscObjectChangeTypeName((PetscObject)pc, nullptr));
56db4a47b3SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetAuxiliaryMat_C", nullptr));
57db4a47b3SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMHasNeumannMat_C", nullptr));
58db4a47b3SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetRHSMat_C", nullptr));
59db4a47b3SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetCoarseCorrectionType_C", nullptr));
60db4a47b3SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetCoarseCorrectionType_C", nullptr));
61db4a47b3SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetSTShareSubKSP_C", nullptr));
62db4a47b3SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetSTShareSubKSP_C", nullptr));
63db4a47b3SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetDeflationMat_C", nullptr));
6413044ca3SPierre Jolivet   PetscCall(PetscObjectCompose((PetscObject)pc, "_PCHPDDM_Schur", nullptr));
653ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
66f1580f4eSBarry Smith }
67f1580f4eSBarry Smith 
68d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMSetAuxiliaryMat_Private(PC pc, IS is, Mat A, PetscBool deflation)
69d71ae5a4SJacob Faibussowitsch {
70f1580f4eSBarry Smith   PC_HPDDM                   *data = (PC_HPDDM *)pc->data;
71cdbd50ebSPierre Jolivet   PCHPDDMCoarseCorrectionType type = data->correction;
72f1580f4eSBarry Smith 
73f1580f4eSBarry Smith   PetscFunctionBegin;
74f1580f4eSBarry Smith   if (is) {
75f1580f4eSBarry Smith     PetscCall(PetscObjectReference((PetscObject)is));
76f1580f4eSBarry Smith     if (data->is) { /* new overlap definition resets the PC */
77f1580f4eSBarry Smith       PetscCall(PCReset_HPDDM(pc));
78f1580f4eSBarry Smith       pc->setfromoptionscalled = 0;
79b853e353SPierre Jolivet       pc->setupcalled          = 0;
80cdbd50ebSPierre Jolivet       data->correction         = type;
81f1580f4eSBarry Smith     }
82f1580f4eSBarry Smith     PetscCall(ISDestroy(&data->is));
83f1580f4eSBarry Smith     data->is = is;
84f1580f4eSBarry Smith   }
85f1580f4eSBarry Smith   if (A) {
86f1580f4eSBarry Smith     PetscCall(PetscObjectReference((PetscObject)A));
87f1580f4eSBarry Smith     PetscCall(MatDestroy(&data->aux));
88f1580f4eSBarry Smith     data->aux = A;
89f1580f4eSBarry Smith   }
90f1580f4eSBarry Smith   data->deflation = deflation;
913ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
92f1580f4eSBarry Smith }
93f1580f4eSBarry Smith 
943df4cd7bSPierre Jolivet static inline PetscErrorCode PCHPDDMSetAuxiliaryMatNormal_Private(PC pc, Mat A, Mat N, Mat *B, const char *pcpre, Vec *diagonal = nullptr)
95d71ae5a4SJacob Faibussowitsch {
96f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
97f1580f4eSBarry Smith   Mat      *splitting, *sub, aux;
983df4cd7bSPierre Jolivet   Vec       d;
99f1580f4eSBarry Smith   IS        is, cols[2], rows;
100f1580f4eSBarry Smith   PetscReal norm;
101f1580f4eSBarry Smith   PetscBool flg;
102f1580f4eSBarry Smith   char      type[256] = {}; /* same size as in src/ksp/pc/interface/pcset.c */
103f1580f4eSBarry Smith 
104f1580f4eSBarry Smith   PetscFunctionBegin;
105f1580f4eSBarry Smith   PetscCall(MatConvert(N, MATAIJ, MAT_INITIAL_MATRIX, B));
106f1580f4eSBarry Smith   PetscCall(ISCreateStride(PETSC_COMM_SELF, A->cmap->n, A->cmap->rstart, 1, cols));
107f1580f4eSBarry Smith   PetscCall(ISCreateStride(PETSC_COMM_SELF, A->rmap->N, 0, 1, &rows));
108f1580f4eSBarry Smith   PetscCall(MatSetOption(A, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
109f1580f4eSBarry Smith   PetscCall(MatIncreaseOverlap(*B, 1, cols, 1));
110f1580f4eSBarry Smith   PetscCall(MatCreateSubMatrices(A, 1, &rows, cols, MAT_INITIAL_MATRIX, &splitting));
111f1580f4eSBarry Smith   PetscCall(ISCreateStride(PETSC_COMM_SELF, A->cmap->n, A->cmap->rstart, 1, &is));
112f1580f4eSBarry Smith   PetscCall(ISEmbed(*cols, is, PETSC_TRUE, cols + 1));
113f1580f4eSBarry Smith   PetscCall(ISDestroy(&is));
114f1580f4eSBarry Smith   PetscCall(MatCreateSubMatrices(*splitting, 1, &rows, cols + 1, MAT_INITIAL_MATRIX, &sub));
115f1580f4eSBarry Smith   PetscCall(ISDestroy(cols + 1));
116f1580f4eSBarry Smith   PetscCall(MatFindZeroRows(*sub, &is));
117f1580f4eSBarry Smith   PetscCall(MatDestroySubMatrices(1, &sub));
118f1580f4eSBarry Smith   PetscCall(ISDestroy(&rows));
119f1580f4eSBarry Smith   PetscCall(MatSetOption(*splitting, MAT_KEEP_NONZERO_PATTERN, PETSC_TRUE));
120db4a47b3SPierre Jolivet   PetscCall(MatZeroRowsIS(*splitting, is, 0.0, nullptr, nullptr));
121f1580f4eSBarry Smith   PetscCall(ISDestroy(&is));
122db4a47b3SPierre Jolivet   PetscCall(PetscOptionsGetString(nullptr, pcpre, "-pc_hpddm_levels_1_sub_pc_type", type, sizeof(type), nullptr));
123f1580f4eSBarry Smith   PetscCall(PetscStrcmp(type, PCQR, &flg));
124f1580f4eSBarry Smith   if (!flg) {
125f1580f4eSBarry Smith     Mat conjugate = *splitting;
126811e8887SPierre Jolivet 
127f1580f4eSBarry Smith     if (PetscDefined(USE_COMPLEX)) {
128f1580f4eSBarry Smith       PetscCall(MatDuplicate(*splitting, MAT_COPY_VALUES, &conjugate));
129f1580f4eSBarry Smith       PetscCall(MatConjugate(conjugate));
130f1580f4eSBarry Smith     }
131fb842aefSJose E. Roman     PetscCall(MatTransposeMatMult(conjugate, *splitting, MAT_INITIAL_MATRIX, PETSC_DETERMINE, &aux));
132f1580f4eSBarry Smith     if (PetscDefined(USE_COMPLEX)) PetscCall(MatDestroy(&conjugate));
133f1580f4eSBarry Smith     PetscCall(MatNorm(aux, NORM_FROBENIUS, &norm));
134f1580f4eSBarry Smith     PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE));
1353df4cd7bSPierre Jolivet     if (diagonal) {
136811e8887SPierre Jolivet       PetscReal norm;
137811e8887SPierre Jolivet 
1383df4cd7bSPierre Jolivet       PetscCall(VecNorm(*diagonal, NORM_INFINITY, &norm));
1393df4cd7bSPierre Jolivet       if (norm > PETSC_SMALL) {
14001e3c840SPierre Jolivet         PetscSF  scatter;
1413df4cd7bSPierre Jolivet         PetscInt n;
142811e8887SPierre Jolivet 
1433df4cd7bSPierre Jolivet         PetscCall(ISGetLocalSize(*cols, &n));
1443df4cd7bSPierre Jolivet         PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)pc), n, PETSC_DECIDE, &d));
1453df4cd7bSPierre Jolivet         PetscCall(VecScatterCreate(*diagonal, *cols, d, nullptr, &scatter));
1463df4cd7bSPierre Jolivet         PetscCall(VecScatterBegin(scatter, *diagonal, d, INSERT_VALUES, SCATTER_FORWARD));
1473df4cd7bSPierre Jolivet         PetscCall(VecScatterEnd(scatter, *diagonal, d, INSERT_VALUES, SCATTER_FORWARD));
14801e3c840SPierre Jolivet         PetscCall(PetscSFDestroy(&scatter));
1493df4cd7bSPierre Jolivet         PetscCall(MatScale(aux, -1.0));
1503df4cd7bSPierre Jolivet         PetscCall(MatDiagonalSet(aux, d, ADD_VALUES));
1513df4cd7bSPierre Jolivet         PetscCall(VecDestroy(&d));
1523df4cd7bSPierre Jolivet       } else PetscCall(VecDestroy(diagonal));
1533df4cd7bSPierre Jolivet     }
1543df4cd7bSPierre Jolivet     if (!diagonal) PetscCall(MatShift(aux, PETSC_SMALL * norm));
155f1580f4eSBarry Smith   } else {
156f1580f4eSBarry Smith     PetscBool flg;
157811e8887SPierre Jolivet 
1583df4cd7bSPierre Jolivet     if (diagonal) {
1593df4cd7bSPierre Jolivet       PetscCall(VecNorm(*diagonal, NORM_INFINITY, &norm));
1603df4cd7bSPierre Jolivet       PetscCheck(norm < PETSC_SMALL, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Nonzero diagonal A11 block");
1613df4cd7bSPierre Jolivet       PetscCall(VecDestroy(diagonal));
1623df4cd7bSPierre Jolivet     }
163f1580f4eSBarry Smith     PetscCall(PetscObjectTypeCompare((PetscObject)N, MATNORMAL, &flg));
164f1580f4eSBarry Smith     if (flg) PetscCall(MatCreateNormal(*splitting, &aux));
165f1580f4eSBarry Smith     else PetscCall(MatCreateNormalHermitian(*splitting, &aux));
166f1580f4eSBarry Smith   }
167f1580f4eSBarry Smith   PetscCall(MatDestroySubMatrices(1, &splitting));
168db4a47b3SPierre Jolivet   PetscCall(PCHPDDMSetAuxiliaryMat(pc, *cols, aux, nullptr, nullptr));
169c8ea6600SPierre Jolivet   data->Neumann = PETSC_BOOL3_TRUE;
170f1580f4eSBarry Smith   PetscCall(ISDestroy(cols));
171f1580f4eSBarry Smith   PetscCall(MatDestroy(&aux));
1723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
173f1580f4eSBarry Smith }
174f1580f4eSBarry Smith 
175d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetAuxiliaryMat_HPDDM(PC pc, IS is, Mat A, PetscErrorCode (*setup)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void *setup_ctx)
176d71ae5a4SJacob Faibussowitsch {
177f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
178f1580f4eSBarry Smith 
179f1580f4eSBarry Smith   PetscFunctionBegin;
180f1580f4eSBarry Smith   PetscCall(PCHPDDMSetAuxiliaryMat_Private(pc, is, A, PETSC_FALSE));
181f1580f4eSBarry Smith   if (setup) {
182f1580f4eSBarry Smith     data->setup     = setup;
183f1580f4eSBarry Smith     data->setup_ctx = setup_ctx;
184f1580f4eSBarry Smith   }
1853ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
186f1580f4eSBarry Smith }
187f1580f4eSBarry Smith 
18873329a61SPierre Jolivet typedef struct {
18973329a61SPierre Jolivet   KSP      ksp;
19073329a61SPierre Jolivet   PetscInt its;
19173329a61SPierre Jolivet } PC_KSP;
19273329a61SPierre Jolivet 
19373329a61SPierre Jolivet static inline PetscErrorCode PCSetUp_KSP_Private(PC pc)
19473329a61SPierre Jolivet {
19573329a61SPierre Jolivet   PC_KSP           *data   = (PC_KSP *)pc->data;
19673329a61SPierre Jolivet   const std::string prefix = ((PetscObject)data->ksp)->prefix;
19773329a61SPierre Jolivet   std::string       sub;
19873329a61SPierre Jolivet 
19973329a61SPierre Jolivet   PetscFunctionBegin;
20073329a61SPierre Jolivet   PetscCheck(prefix.size() >= 9, PETSC_COMM_SELF, PETSC_ERR_PLIB, "The prefix of this PCKSP should be of length at least 9 to hold the suffix pc_hpddm_");
20173329a61SPierre Jolivet   sub = prefix.substr(0, prefix.size() - 9);
20273329a61SPierre Jolivet   PetscCall(PCSetType(pc, PCHPDDM));
20373329a61SPierre Jolivet   PetscCall(PCSetOptionsPrefix(pc, sub.c_str()));
20473329a61SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
20573329a61SPierre Jolivet }
20673329a61SPierre Jolivet 
20773329a61SPierre Jolivet static PetscErrorCode PCSetUp_KSP(PC pc)
20873329a61SPierre Jolivet {
20973329a61SPierre Jolivet   PetscFunctionBegin;
21073329a61SPierre Jolivet   PetscCall(PCSetUp_KSP_Private(pc));
21173329a61SPierre Jolivet   PetscCall(PCSetFromOptions(pc));
21273329a61SPierre Jolivet   PetscCall(PCSetUp(pc));
21373329a61SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
21473329a61SPierre Jolivet }
21573329a61SPierre Jolivet 
21670009435SPierre Jolivet /*@C
21704c3f3b8SBarry Smith   PCHPDDMSetAuxiliaryMat - Sets the auxiliary matrix used by `PCHPDDM` for the concurrent GenEO problems at the finest level.
218f1580f4eSBarry Smith 
219f1580f4eSBarry Smith   Input Parameters:
220f1580f4eSBarry Smith + pc    - preconditioner context
221f1580f4eSBarry Smith . is    - index set of the local auxiliary, e.g., Neumann, matrix
222f1580f4eSBarry Smith . A     - auxiliary sequential matrix
22304c3f3b8SBarry Smith . setup - function for generating the auxiliary matrix entries, may be `NULL`
22404c3f3b8SBarry Smith - ctx   - context for `setup`, may be `NULL`
22504c3f3b8SBarry Smith 
22604c3f3b8SBarry Smith   Calling sequence of `setup`:
22704c3f3b8SBarry Smith + J   - matrix whose values are to be set
22804c3f3b8SBarry Smith . t   - time
22904c3f3b8SBarry Smith . X   - linearization point
23004c3f3b8SBarry Smith . X_t - time-derivative of the linearization point
23104c3f3b8SBarry Smith . s   - step
23204c3f3b8SBarry Smith . ovl - index set of the local auxiliary, e.g., Neumann, matrix
23304c3f3b8SBarry Smith - ctx - context for `setup`, may be `NULL`
234f1580f4eSBarry Smith 
235f1580f4eSBarry Smith   Level: intermediate
236f1580f4eSBarry Smith 
23704c3f3b8SBarry Smith   Note:
23804c3f3b8SBarry Smith   As an example, in a finite element context with nonoverlapping subdomains plus (overlapping) ghost elements, this could be the unassembled (Neumann)
23904c3f3b8SBarry Smith   local overlapping operator. As opposed to the assembled (Dirichlet) local overlapping operator obtained by summing neighborhood contributions
24004c3f3b8SBarry Smith   at the interface of ghost elements.
24104c3f3b8SBarry Smith 
24270009435SPierre Jolivet   Fortran Notes:
24304c3f3b8SBarry Smith   Only `PETSC_NULL_FUNCTION` is supported for `setup` and `ctx` is never accessed
24470009435SPierre Jolivet 
245562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHPDDMSetRHSMat()`, `MATIS`
246f1580f4eSBarry Smith @*/
24704c3f3b8SBarry Smith PetscErrorCode PCHPDDMSetAuxiliaryMat(PC pc, IS is, Mat A, PetscErrorCode (*setup)(Mat J, PetscReal t, Vec X, Vec X_t, PetscReal s, IS ovl, void *ctx), void *ctx)
248d71ae5a4SJacob Faibussowitsch {
249f1580f4eSBarry Smith   PetscFunctionBegin;
250f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
251f1580f4eSBarry Smith   if (is) PetscValidHeaderSpecific(is, IS_CLASSID, 2);
252f1580f4eSBarry Smith   if (A) PetscValidHeaderSpecific(A, MAT_CLASSID, 3);
25373329a61SPierre Jolivet   if (pc->ops->setup == PCSetUp_KSP) PetscCall(PCSetUp_KSP_Private(pc));
25404c3f3b8SBarry Smith   PetscTryMethod(pc, "PCHPDDMSetAuxiliaryMat_C", (PC, IS, Mat, PetscErrorCode (*)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void *), (pc, is, A, setup, ctx));
2553ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
256f1580f4eSBarry Smith }
257f1580f4eSBarry Smith 
258d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMHasNeumannMat_HPDDM(PC pc, PetscBool has)
259d71ae5a4SJacob Faibussowitsch {
260f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
261f1580f4eSBarry Smith 
262f1580f4eSBarry Smith   PetscFunctionBegin;
263c8ea6600SPierre Jolivet   data->Neumann = PetscBoolToBool3(has);
2643ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
265f1580f4eSBarry Smith }
266f1580f4eSBarry Smith 
267f1580f4eSBarry Smith /*@
268f1580f4eSBarry Smith   PCHPDDMHasNeumannMat - Informs `PCHPDDM` that the `Mat` passed to `PCHPDDMSetAuxiliaryMat()` is the local Neumann matrix.
269f1580f4eSBarry Smith 
270f1580f4eSBarry Smith   Input Parameters:
271f1580f4eSBarry Smith + pc  - preconditioner context
272f1580f4eSBarry Smith - has - Boolean value
273f1580f4eSBarry Smith 
274f1580f4eSBarry Smith   Level: intermediate
275f1580f4eSBarry Smith 
276f1580f4eSBarry Smith   Notes:
2777eb095acSPierre Jolivet   This may be used to bypass a call to `MatCreateSubMatrices()` and to `MatConvert()` for `MATSBAIJ` matrices.
278f1580f4eSBarry Smith 
279907a3e9cSStefano Zampini   If a function is composed with DMCreateNeumannOverlap_C implementation is available in the `DM` attached to the Pmat, or the Amat, or the `PC`, the flag is internally set to `PETSC_TRUE`. Its default value is otherwise `PETSC_FALSE`.
280f1580f4eSBarry Smith 
281562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMSetAuxiliaryMat()`
282f1580f4eSBarry Smith @*/
283d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMHasNeumannMat(PC pc, PetscBool has)
284d71ae5a4SJacob Faibussowitsch {
285f1580f4eSBarry Smith   PetscFunctionBegin;
286f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
287f1580f4eSBarry Smith   PetscTryMethod(pc, "PCHPDDMHasNeumannMat_C", (PC, PetscBool), (pc, has));
2883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
289f1580f4eSBarry Smith }
290f1580f4eSBarry Smith 
291d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetRHSMat_HPDDM(PC pc, Mat B)
292d71ae5a4SJacob Faibussowitsch {
293f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
294f1580f4eSBarry Smith 
295f1580f4eSBarry Smith   PetscFunctionBegin;
296f1580f4eSBarry Smith   PetscCall(PetscObjectReference((PetscObject)B));
297f1580f4eSBarry Smith   PetscCall(MatDestroy(&data->B));
298f1580f4eSBarry Smith   data->B = B;
2993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
300f1580f4eSBarry Smith }
301f1580f4eSBarry Smith 
302f1580f4eSBarry Smith /*@
30304c3f3b8SBarry Smith   PCHPDDMSetRHSMat - Sets the right-hand side matrix used by `PCHPDDM` for the concurrent GenEO problems at the finest level.
304f1580f4eSBarry Smith 
305f1580f4eSBarry Smith   Input Parameters:
306f1580f4eSBarry Smith + pc - preconditioner context
307f1580f4eSBarry Smith - B  - right-hand side sequential matrix
308f1580f4eSBarry Smith 
309f1580f4eSBarry Smith   Level: advanced
310f1580f4eSBarry Smith 
31104c3f3b8SBarry Smith   Note:
31204c3f3b8SBarry Smith   Must be used in conjunction with `PCHPDDMSetAuxiliaryMat`(N), so that Nv = lambda Bv is solved using `EPSSetOperators`(N, B).
31304c3f3b8SBarry Smith   It is assumed that N and `B` are provided using the same numbering. This provides a means to try more advanced methods such as GenEO-II or H-GenEO.
31404c3f3b8SBarry Smith 
315562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDMSetAuxiliaryMat()`, `PCHPDDM`
316f1580f4eSBarry Smith @*/
317d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetRHSMat(PC pc, Mat B)
318d71ae5a4SJacob Faibussowitsch {
319f1580f4eSBarry Smith   PetscFunctionBegin;
320f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
321f1580f4eSBarry Smith   if (B) {
322f1580f4eSBarry Smith     PetscValidHeaderSpecific(B, MAT_CLASSID, 2);
323f1580f4eSBarry Smith     PetscTryMethod(pc, "PCHPDDMSetRHSMat_C", (PC, Mat), (pc, B));
324f1580f4eSBarry Smith   }
3253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
326f1580f4eSBarry Smith }
327f1580f4eSBarry Smith 
328d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetFromOptions_HPDDM(PC pc, PetscOptionItems *PetscOptionsObject)
329d71ae5a4SJacob Faibussowitsch {
330f1580f4eSBarry Smith   PC_HPDDM                   *data   = (PC_HPDDM *)pc->data;
331f1580f4eSBarry Smith   PC_HPDDM_Level            **levels = data->levels;
332f1580f4eSBarry Smith   char                        prefix[256];
333f1580f4eSBarry Smith   int                         i = 1;
334f1580f4eSBarry Smith   PetscMPIInt                 size, previous;
3359bb5c669SPierre Jolivet   PetscInt                    n, overlap = 1;
336f1580f4eSBarry Smith   PCHPDDMCoarseCorrectionType type;
337c8ea6600SPierre Jolivet   PetscBool                   flg = PETSC_TRUE, set;
338f1580f4eSBarry Smith 
339f1580f4eSBarry Smith   PetscFunctionBegin;
340f1580f4eSBarry Smith   if (!data->levels) {
341f1580f4eSBarry Smith     PetscCall(PetscCalloc1(PETSC_PCHPDDM_MAXLEVELS, &levels));
342f1580f4eSBarry Smith     data->levels = levels;
343f1580f4eSBarry Smith   }
344f1580f4eSBarry Smith   PetscOptionsHeadBegin(PetscOptionsObject, "PCHPDDM options");
3459bb5c669SPierre Jolivet   PetscCall(PetscOptionsBoundedInt("-pc_hpddm_harmonic_overlap", "Overlap prior to computing local harmonic extensions", "PCHPDDM", overlap, &overlap, &set, 1));
3469bb5c669SPierre Jolivet   if (!set) overlap = -1;
347f1580f4eSBarry Smith   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)pc), &size));
348f1580f4eSBarry Smith   previous = size;
349f1580f4eSBarry Smith   while (i < PETSC_PCHPDDM_MAXLEVELS) {
350f1580f4eSBarry Smith     PetscInt p = 1;
351f1580f4eSBarry Smith 
3524dfa11a4SJacob Faibussowitsch     if (!data->levels[i - 1]) PetscCall(PetscNew(data->levels + i - 1));
353f1580f4eSBarry Smith     data->levels[i - 1]->parent = data;
354f1580f4eSBarry Smith     /* if the previous level has a single process, it is not possible to coarsen further */
355f1580f4eSBarry Smith     if (previous == 1 || !flg) break;
356f1580f4eSBarry Smith     data->levels[i - 1]->nu        = 0;
357f1580f4eSBarry Smith     data->levels[i - 1]->threshold = -1.0;
358f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_nev", i));
359db4a47b3SPierre Jolivet     PetscCall(PetscOptionsBoundedInt(prefix, "Local number of deflation vectors computed by SLEPc", "EPSSetDimensions", data->levels[i - 1]->nu, &data->levels[i - 1]->nu, nullptr, 0));
360f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_threshold", i));
361db4a47b3SPierre Jolivet     PetscCall(PetscOptionsReal(prefix, "Local threshold for selecting deflation vectors returned by SLEPc", "PCHPDDM", data->levels[i - 1]->threshold, &data->levels[i - 1]->threshold, nullptr));
362f1580f4eSBarry Smith     if (i == 1) {
363aa1539e9SPierre Jolivet       PetscCheck(overlap == -1 || PetscAbsReal(data->levels[i - 1]->threshold + static_cast<PetscReal>(1.0)) < PETSC_MACHINE_EPSILON, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot supply both -pc_hpddm_levels_1_eps_threshold and -pc_hpddm_harmonic_overlap");
3649bb5c669SPierre Jolivet       if (overlap != -1) {
3659bb5c669SPierre Jolivet         PetscInt nsv = 0;
366811e8887SPierre Jolivet 
3679bb5c669SPierre Jolivet         PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_svd_nsv", i));
3689bb5c669SPierre Jolivet         PetscCall(PetscOptionsBoundedInt(prefix, "Local number of deflation vectors computed by SLEPc", "SVDSetDimensions", nsv, &nsv, nullptr, 0));
3699bb5c669SPierre Jolivet         PetscCheck(bool(data->levels[0]->nu) != bool(nsv), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot supply %s -pc_hpddm_levels_1_eps_nev %s -pc_hpddm_levels_1_svd_nsv", nsv ? "both" : "neither", nsv ? "and" : "nor");
3709bb5c669SPierre Jolivet         if (nsv) {
3719bb5c669SPierre Jolivet           data->levels[0]->nu = nsv;
3729bb5c669SPierre Jolivet           PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_svd_relative_threshold", i));
3739bb5c669SPierre Jolivet         } else PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_relative_threshold", i));
3749bb5c669SPierre Jolivet         PetscCall(PetscOptionsReal(prefix, "Local relative threshold for selecting deflation vectors returned by SLEPc", "PCHPDDM", data->levels[0]->threshold, &data->levels[0]->threshold, nullptr));
3759bb5c669SPierre Jolivet       }
376f1580f4eSBarry Smith       PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_1_st_share_sub_ksp"));
377db4a47b3SPierre Jolivet       PetscCall(PetscOptionsBool(prefix, "Shared KSP between SLEPc ST and the fine-level subdomain solver", "PCHPDDMSetSTShareSubKSP", PETSC_FALSE, &data->share, nullptr));
378f1580f4eSBarry Smith     }
379f1580f4eSBarry Smith     /* if there is no prescribed coarsening, just break out of the loop */
3800594bca0SPierre Jolivet     if (data->levels[i - 1]->threshold <= PetscReal() && data->levels[i - 1]->nu <= 0 && !(data->deflation && i == 1)) break;
381f1580f4eSBarry Smith     else {
382f1580f4eSBarry Smith       ++i;
383f1580f4eSBarry Smith       PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_nev", i));
384f1580f4eSBarry Smith       PetscCall(PetscOptionsHasName(PetscOptionsObject->options, PetscOptionsObject->prefix, prefix, &flg));
385f1580f4eSBarry Smith       if (!flg) {
386f1580f4eSBarry Smith         PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_threshold", i));
387f1580f4eSBarry Smith         PetscCall(PetscOptionsHasName(PetscOptionsObject->options, PetscOptionsObject->prefix, prefix, &flg));
388f1580f4eSBarry Smith       }
389f1580f4eSBarry Smith       if (flg) {
390f1580f4eSBarry Smith         /* if there are coarsening options for the next level, then register it  */
391f1580f4eSBarry Smith         /* otherwise, don't to avoid having both options levels_N_p and coarse_p */
392f1580f4eSBarry Smith         PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_p", i));
393f1580f4eSBarry Smith         PetscCall(PetscOptionsRangeInt(prefix, "Number of processes used to assemble the coarse operator at this level", "PCHPDDM", p, &p, &flg, 1, PetscMax(1, previous / 2)));
394f1580f4eSBarry Smith         previous = p;
395f1580f4eSBarry Smith       }
396f1580f4eSBarry Smith     }
397f1580f4eSBarry Smith   }
398f1580f4eSBarry Smith   data->N = i;
399f1580f4eSBarry Smith   n       = 1;
400f1580f4eSBarry Smith   if (i > 1) {
401f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_coarse_p"));
402db4a47b3SPierre Jolivet     PetscCall(PetscOptionsRangeInt(prefix, "Number of processes used to assemble the coarsest operator", "PCHPDDM", n, &n, nullptr, 1, PetscMax(1, previous / 2)));
40302800ff6SPierre Jolivet #if PetscDefined(HAVE_MUMPS)
404f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "pc_hpddm_coarse_"));
405db4a47b3SPierre Jolivet     PetscCall(PetscOptionsHasName(nullptr, prefix, "-mat_mumps_use_omp_threads", &flg));
406f1580f4eSBarry Smith     if (flg) {
407f1580f4eSBarry Smith       char type[64]; /* same size as in src/ksp/pc/impls/factor/factimpl.c */
408811e8887SPierre Jolivet 
409c6a7a370SJeremy L Thompson       PetscCall(PetscStrncpy(type, n > 1 && PetscDefined(HAVE_MUMPS) ? MATSOLVERMUMPS : MATSOLVERPETSC, sizeof(type))); /* default solver for a MatMPIAIJ or a MatSeqAIJ */
410db4a47b3SPierre Jolivet       PetscCall(PetscOptionsGetString(nullptr, prefix, "-pc_factor_mat_solver_type", type, sizeof(type), nullptr));
4113ce573a3SPierre Jolivet       PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg));
412f1580f4eSBarry Smith       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "-%smat_mumps_use_omp_threads and -%spc_factor_mat_solver_type != %s", prefix, prefix, MATSOLVERMUMPS);
413f1580f4eSBarry Smith       size = n;
414f1580f4eSBarry Smith       n    = -1;
415db4a47b3SPierre Jolivet       PetscCall(PetscOptionsGetInt(nullptr, prefix, "-mat_mumps_use_omp_threads", &n, nullptr));
416f1580f4eSBarry Smith       PetscCheck(n >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Need to specify a positive integer for -%smat_mumps_use_omp_threads", prefix);
417f1580f4eSBarry Smith       PetscCheck(n * size <= previous, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "%d MPI process%s x %d OpenMP thread%s greater than %d available MPI process%s for the coarsest operator", (int)size, size > 1 ? "es" : "", (int)n, n > 1 ? "s" : "", (int)previous, previous > 1 ? "es" : "");
418f1580f4eSBarry Smith     }
419f1580f4eSBarry Smith #endif
420f1580f4eSBarry Smith     PetscCall(PetscOptionsEnum("-pc_hpddm_coarse_correction", "Type of coarse correction applied each iteration", "PCHPDDMSetCoarseCorrectionType", PCHPDDMCoarseCorrectionTypes, (PetscEnum)data->correction, (PetscEnum *)&type, &flg));
421f1580f4eSBarry Smith     if (flg) PetscCall(PCHPDDMSetCoarseCorrectionType(pc, type));
422f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_has_neumann"));
423c8ea6600SPierre Jolivet     PetscCall(PetscOptionsBool(prefix, "Is the auxiliary Mat the local Neumann matrix?", "PCHPDDMHasNeumannMat", PetscBool3ToBool(data->Neumann), &flg, &set));
424c8ea6600SPierre Jolivet     if (set) data->Neumann = PetscBoolToBool3(flg);
425f1580f4eSBarry Smith     data->log_separate = PETSC_FALSE;
426f1580f4eSBarry Smith     if (PetscDefined(USE_LOG)) {
427f1580f4eSBarry Smith       PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_log_separate"));
428db4a47b3SPierre Jolivet       PetscCall(PetscOptionsBool(prefix, "Log events level by level instead of inside PCSetUp()/KSPSolve()", nullptr, data->log_separate, &data->log_separate, nullptr));
429f1580f4eSBarry Smith     }
430f1580f4eSBarry Smith   }
431f1580f4eSBarry Smith   PetscOptionsHeadEnd();
432f1580f4eSBarry Smith   while (i < PETSC_PCHPDDM_MAXLEVELS && data->levels[i]) PetscCall(PetscFree(data->levels[i++]));
4333ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
434f1580f4eSBarry Smith }
435f1580f4eSBarry Smith 
436d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_HPDDM(PC pc, Vec x, Vec y)
437d71ae5a4SJacob Faibussowitsch {
438f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
439f1580f4eSBarry Smith 
440f1580f4eSBarry Smith   PetscFunctionBegin;
441f1580f4eSBarry Smith   PetscCall(PetscCitationsRegister(HPDDMCitation, &HPDDMCite));
442f1580f4eSBarry Smith   PetscCheck(data->levels[0]->ksp, PETSC_COMM_SELF, PETSC_ERR_PLIB, "No KSP attached to PCHPDDM");
443db4a47b3SPierre Jolivet   if (data->log_separate) PetscCall(PetscLogEventBegin(PC_HPDDM_Solve[0], data->levels[0]->ksp, nullptr, nullptr, nullptr)); /* coarser-level events are directly triggered in HPDDM */
444f1580f4eSBarry Smith   PetscCall(KSPSolve(data->levels[0]->ksp, x, y));
445db4a47b3SPierre Jolivet   if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Solve[0], data->levels[0]->ksp, nullptr, nullptr, nullptr));
4463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
447f1580f4eSBarry Smith }
448f1580f4eSBarry Smith 
449d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_HPDDM(PC pc, Mat X, Mat Y)
450d71ae5a4SJacob Faibussowitsch {
451f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
452f1580f4eSBarry Smith 
453f1580f4eSBarry Smith   PetscFunctionBegin;
454f1580f4eSBarry Smith   PetscCall(PetscCitationsRegister(HPDDMCitation, &HPDDMCite));
455f1580f4eSBarry Smith   PetscCheck(data->levels[0]->ksp, PETSC_COMM_SELF, PETSC_ERR_PLIB, "No KSP attached to PCHPDDM");
456f1580f4eSBarry Smith   PetscCall(KSPMatSolve(data->levels[0]->ksp, X, Y));
4573ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
458f1580f4eSBarry Smith }
459f1580f4eSBarry Smith 
460cc4c1da9SBarry Smith /*@
461f1580f4eSBarry Smith   PCHPDDMGetComplexities - Computes the grid and operator complexities.
462f1580f4eSBarry Smith 
463c71f06a7SPierre Jolivet   Collective
464c71f06a7SPierre Jolivet 
465f1580f4eSBarry Smith   Input Parameter:
466f1580f4eSBarry Smith . pc - preconditioner context
467f1580f4eSBarry Smith 
468f1580f4eSBarry Smith   Output Parameters:
469cc4c1da9SBarry Smith + gc - grid complexity $ \sum_i m_i / m_1 $
470cc4c1da9SBarry Smith - oc - operator complexity $ \sum_i nnz_i / nnz_1 $
471f1580f4eSBarry Smith 
472f1580f4eSBarry Smith   Level: advanced
473f1580f4eSBarry Smith 
474562efe2eSBarry Smith .seealso: [](ch_ksp), `PCMGGetGridComplexity()`, `PCHPDDM`, `PCHYPRE`, `PCGAMG`
475f1580f4eSBarry Smith @*/
476cc4c1da9SBarry Smith PetscErrorCode PCHPDDMGetComplexities(PC pc, PetscReal *gc, PetscReal *oc)
477d71ae5a4SJacob Faibussowitsch {
478f1580f4eSBarry Smith   PC_HPDDM      *data = (PC_HPDDM *)pc->data;
479f1580f4eSBarry Smith   MatInfo        info;
480f1580f4eSBarry Smith   PetscLogDouble accumulate[2]{}, nnz1 = 1.0, m1 = 1.0;
481f1580f4eSBarry Smith 
482f1580f4eSBarry Smith   PetscFunctionBegin;
483715c1178SPierre Jolivet   if (gc) {
484715c1178SPierre Jolivet     PetscAssertPointer(gc, 2);
485715c1178SPierre Jolivet     *gc = 0;
486715c1178SPierre Jolivet   }
487715c1178SPierre Jolivet   if (oc) {
488715c1178SPierre Jolivet     PetscAssertPointer(oc, 3);
489715c1178SPierre Jolivet     *oc = 0;
490715c1178SPierre Jolivet   }
491715c1178SPierre Jolivet   for (PetscInt n = 0; n < data->N; ++n) {
492f1580f4eSBarry Smith     if (data->levels[n]->ksp) {
49313044ca3SPierre Jolivet       Mat       P, A = nullptr;
494715c1178SPierre Jolivet       PetscInt  m;
49513044ca3SPierre Jolivet       PetscBool flg = PETSC_FALSE;
49613044ca3SPierre Jolivet 
497db4a47b3SPierre Jolivet       PetscCall(KSPGetOperators(data->levels[n]->ksp, nullptr, &P));
498db4a47b3SPierre Jolivet       PetscCall(MatGetSize(P, &m, nullptr));
499f1580f4eSBarry Smith       accumulate[0] += m;
500f1580f4eSBarry Smith       if (n == 0) {
501f1580f4eSBarry Smith         PetscCall(PetscObjectTypeCompareAny((PetscObject)P, &flg, MATNORMAL, MATNORMALHERMITIAN, ""));
502f1580f4eSBarry Smith         if (flg) {
503f1580f4eSBarry Smith           PetscCall(MatConvert(P, MATAIJ, MAT_INITIAL_MATRIX, &A));
504f1580f4eSBarry Smith           P = A;
50513044ca3SPierre Jolivet         } else {
50613044ca3SPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg));
50713044ca3SPierre Jolivet           PetscCall(PetscObjectReference((PetscObject)P));
508f1580f4eSBarry Smith         }
50913044ca3SPierre Jolivet       }
51013044ca3SPierre Jolivet       if (!A && flg) accumulate[1] += m * m; /* assumption that a MATSCHURCOMPLEMENT is dense if stored explicitly */
51113044ca3SPierre Jolivet       else if (P->ops->getinfo) {
512f1580f4eSBarry Smith         PetscCall(MatGetInfo(P, MAT_GLOBAL_SUM, &info));
513f1580f4eSBarry Smith         accumulate[1] += info.nz_used;
514f1580f4eSBarry Smith       }
515f1580f4eSBarry Smith       if (n == 0) {
516f1580f4eSBarry Smith         m1 = m;
51713044ca3SPierre Jolivet         if (!A && flg) nnz1 = m * m;
51813044ca3SPierre Jolivet         else if (P->ops->getinfo) nnz1 = info.nz_used;
519f1580f4eSBarry Smith         PetscCall(MatDestroy(&P));
520f1580f4eSBarry Smith       }
521f1580f4eSBarry Smith     }
522f1580f4eSBarry Smith   }
523715c1178SPierre Jolivet   /* only process #0 has access to the full hierarchy by construction, so broadcast to ensure consistent outputs */
524715c1178SPierre Jolivet   PetscCallMPI(MPI_Bcast(accumulate, 2, MPIU_PETSCLOGDOUBLE, 0, PetscObjectComm((PetscObject)pc)));
525715c1178SPierre Jolivet   if (gc) *gc = static_cast<PetscReal>(accumulate[0] / m1);
526715c1178SPierre Jolivet   if (oc) *oc = static_cast<PetscReal>(accumulate[1] / nnz1);
5273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
528f1580f4eSBarry Smith }
529f1580f4eSBarry Smith 
530d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCView_HPDDM(PC pc, PetscViewer viewer)
531d71ae5a4SJacob Faibussowitsch {
532f1580f4eSBarry Smith   PC_HPDDM         *data = (PC_HPDDM *)pc->data;
533f1580f4eSBarry Smith   PetscViewer       subviewer;
534aa21023fSPierre Jolivet   PetscViewerFormat format;
535f1580f4eSBarry Smith   PetscSubcomm      subcomm;
536f1580f4eSBarry Smith   PetscReal         oc, gc;
537811e8887SPierre Jolivet   PetscInt          tabs;
538f1580f4eSBarry Smith   PetscMPIInt       size, color, rank;
539aa21023fSPierre Jolivet   PetscBool         flg;
540aa21023fSPierre Jolivet   const char       *name;
541f1580f4eSBarry Smith 
542f1580f4eSBarry Smith   PetscFunctionBegin;
543aa21023fSPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &flg));
544aa21023fSPierre Jolivet   if (flg) {
545f1580f4eSBarry Smith     PetscCall(PetscViewerASCIIPrintf(viewer, "level%s: %" PetscInt_FMT "\n", data->N > 1 ? "s" : "", data->N));
546f1580f4eSBarry Smith     PetscCall(PCHPDDMGetComplexities(pc, &gc, &oc));
547f1580f4eSBarry Smith     if (data->N > 1) {
548f1580f4eSBarry Smith       if (!data->deflation) {
549c8ea6600SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "Neumann matrix attached? %s\n", PetscBools[PetscBool3ToBool(data->Neumann)]));
550f1580f4eSBarry Smith         PetscCall(PetscViewerASCIIPrintf(viewer, "shared subdomain KSP between SLEPc and PETSc? %s\n", PetscBools[data->share]));
551f1580f4eSBarry Smith       } else PetscCall(PetscViewerASCIIPrintf(viewer, "user-supplied deflation matrix\n"));
552f1580f4eSBarry Smith       PetscCall(PetscViewerASCIIPrintf(viewer, "coarse correction: %s\n", PCHPDDMCoarseCorrectionTypes[data->correction]));
553f1580f4eSBarry Smith       PetscCall(PetscViewerASCIIPrintf(viewer, "on process #0, value%s (+ threshold%s if available) for selecting deflation vectors:", data->N > 2 ? "s" : "", data->N > 2 ? "s" : ""));
554f1580f4eSBarry Smith       PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
555f1580f4eSBarry Smith       PetscCall(PetscViewerASCIISetTab(viewer, 0));
556811e8887SPierre Jolivet       for (PetscInt i = 1; i < data->N; ++i) {
557f1580f4eSBarry Smith         PetscCall(PetscViewerASCIIPrintf(viewer, " %" PetscInt_FMT, data->levels[i - 1]->nu));
5580594bca0SPierre Jolivet         if (data->levels[i - 1]->threshold > static_cast<PetscReal>(-0.1)) PetscCall(PetscViewerASCIIPrintf(viewer, " (%g)", (double)data->levels[i - 1]->threshold));
559f1580f4eSBarry Smith       }
560f1580f4eSBarry Smith       PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
561f1580f4eSBarry Smith       PetscCall(PetscViewerASCIISetTab(viewer, tabs));
562f1580f4eSBarry Smith     }
563f1580f4eSBarry Smith     PetscCall(PetscViewerASCIIPrintf(viewer, "grid and operator complexities: %g %g\n", (double)gc, (double)oc));
5641fe44b27SPierre Jolivet     PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)pc), &size));
565f1580f4eSBarry Smith     if (data->levels[0]->ksp) {
566f1580f4eSBarry Smith       PetscCall(KSPView(data->levels[0]->ksp, viewer));
567f1580f4eSBarry Smith       if (data->levels[0]->pc) PetscCall(PCView(data->levels[0]->pc, viewer));
568*eea1a2f1SRaphael Zanella       PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)pc), &rank));
569811e8887SPierre Jolivet       for (PetscInt i = 1; i < data->N; ++i) {
570f1580f4eSBarry Smith         if (data->levels[i]->ksp) color = 1;
571f1580f4eSBarry Smith         else color = 0;
572f1580f4eSBarry Smith         PetscCall(PetscSubcommCreate(PetscObjectComm((PetscObject)pc), &subcomm));
573f1580f4eSBarry Smith         PetscCall(PetscSubcommSetNumber(subcomm, PetscMin(size, 2)));
574f1580f4eSBarry Smith         PetscCall(PetscSubcommSetTypeGeneral(subcomm, color, rank));
575f1580f4eSBarry Smith         PetscCall(PetscViewerASCIIPushTab(viewer));
576f1580f4eSBarry Smith         PetscCall(PetscViewerGetSubViewer(viewer, PetscSubcommChild(subcomm), &subviewer));
577f1580f4eSBarry Smith         if (color == 1) {
578f1580f4eSBarry Smith           PetscCall(KSPView(data->levels[i]->ksp, subviewer));
579f1580f4eSBarry Smith           if (data->levels[i]->pc) PetscCall(PCView(data->levels[i]->pc, subviewer));
580f1580f4eSBarry Smith           PetscCall(PetscViewerFlush(subviewer));
581f1580f4eSBarry Smith         }
582f1580f4eSBarry Smith         PetscCall(PetscViewerRestoreSubViewer(viewer, PetscSubcommChild(subcomm), &subviewer));
583f1580f4eSBarry Smith         PetscCall(PetscViewerASCIIPopTab(viewer));
584f1580f4eSBarry Smith         PetscCall(PetscSubcommDestroy(&subcomm));
585f1580f4eSBarry Smith       }
586f1580f4eSBarry Smith     }
587aa21023fSPierre Jolivet     PetscCall(PetscViewerGetFormat(viewer, &format));
588aa21023fSPierre Jolivet     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
589aa21023fSPierre Jolivet       PetscCall(PetscViewerFileGetName(viewer, &name));
590aa21023fSPierre Jolivet       if (name) {
591*eea1a2f1SRaphael Zanella         Mat             aux[2];
592aa21023fSPierre Jolivet         IS              is;
593*eea1a2f1SRaphael Zanella         const PetscInt *indices;
594*eea1a2f1SRaphael Zanella         PetscInt        m, n, sizes[5] = {pc->mat->rmap->n, pc->mat->cmap->n, pc->mat->rmap->N, pc->mat->cmap->N, 0};
595aa21023fSPierre Jolivet         char           *tmp;
596aa21023fSPierre Jolivet         std::string     prefix, suffix;
597aa21023fSPierre Jolivet         size_t          pos;
598aa21023fSPierre Jolivet 
599aa21023fSPierre Jolivet         PetscCall(PetscStrstr(name, ".", &tmp));
600aa21023fSPierre Jolivet         if (tmp) {
601aa21023fSPierre Jolivet           pos    = std::distance(const_cast<char *>(name), tmp);
602aa21023fSPierre Jolivet           prefix = std::string(name, pos);
603aa21023fSPierre Jolivet           suffix = std::string(name + pos + 1);
604aa21023fSPierre Jolivet         } else prefix = name;
605aa21023fSPierre Jolivet         if (data->aux) {
606*eea1a2f1SRaphael Zanella           PetscCall(MatGetSize(data->aux, &m, &n));
607*eea1a2f1SRaphael Zanella           PetscCall(MatCreate(PetscObjectComm((PetscObject)pc), aux));
608*eea1a2f1SRaphael Zanella           PetscCall(MatSetSizes(aux[0], m, n, PETSC_DETERMINE, PETSC_DETERMINE));
609*eea1a2f1SRaphael Zanella           PetscCall(PetscObjectBaseTypeCompare((PetscObject)data->aux, MATSEQAIJ, &flg));
610*eea1a2f1SRaphael Zanella           if (flg) PetscCall(MatSetType(aux[0], MATMPIAIJ));
611*eea1a2f1SRaphael Zanella           else {
612*eea1a2f1SRaphael Zanella             PetscCall(PetscObjectBaseTypeCompare((PetscObject)data->aux, MATSEQBAIJ, &flg));
613*eea1a2f1SRaphael Zanella             if (flg) PetscCall(MatSetType(aux[0], MATMPIBAIJ));
614*eea1a2f1SRaphael Zanella             else {
615*eea1a2f1SRaphael Zanella               PetscCall(PetscObjectBaseTypeCompare((PetscObject)data->aux, MATSEQSBAIJ, &flg));
616*eea1a2f1SRaphael Zanella               PetscCheck(flg, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "MatType of auxiliary Mat (%s) is not any of the following: MATSEQAIJ, MATSEQBAIJ, or MATSEQSBAIJ", ((PetscObject)data->aux)->type_name);
617*eea1a2f1SRaphael Zanella               PetscCall(MatSetType(aux[0], MATMPISBAIJ));
618*eea1a2f1SRaphael Zanella             }
619*eea1a2f1SRaphael Zanella           }
620*eea1a2f1SRaphael Zanella           PetscCall(MatSetBlockSizesFromMats(aux[0], data->aux, data->aux));
621*eea1a2f1SRaphael Zanella           PetscCall(MatAssemblyBegin(aux[0], MAT_FINAL_ASSEMBLY));
622*eea1a2f1SRaphael Zanella           PetscCall(MatAssemblyEnd(aux[0], MAT_FINAL_ASSEMBLY));
623*eea1a2f1SRaphael Zanella           PetscCall(MatGetDiagonalBlock(aux[0], aux + 1));
624*eea1a2f1SRaphael Zanella           PetscCall(MatCopy(data->aux, aux[1], DIFFERENT_NONZERO_PATTERN));
625*eea1a2f1SRaphael Zanella           PetscCall(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)pc), std::string(prefix + "_aux_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer));
626*eea1a2f1SRaphael Zanella           PetscCall(MatView(aux[0], subviewer));
627aa21023fSPierre Jolivet           PetscCall(PetscViewerDestroy(&subviewer));
628*eea1a2f1SRaphael Zanella           PetscCall(MatDestroy(aux));
629aa21023fSPierre Jolivet         }
630aa21023fSPierre Jolivet         if (data->is) {
631*eea1a2f1SRaphael Zanella           PetscCall(ISGetIndices(data->is, &indices));
632*eea1a2f1SRaphael Zanella           PetscCall(ISGetSize(data->is, sizes + 4));
633*eea1a2f1SRaphael Zanella           PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)pc), sizes[4], indices, PETSC_USE_POINTER, &is));
634*eea1a2f1SRaphael Zanella           PetscCall(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)pc), std::string(prefix + "_is_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer));
635*eea1a2f1SRaphael Zanella           PetscCall(ISView(is, subviewer));
636aa21023fSPierre Jolivet           PetscCall(PetscViewerDestroy(&subviewer));
637*eea1a2f1SRaphael Zanella           PetscCall(ISDestroy(&is));
638*eea1a2f1SRaphael Zanella           PetscCall(ISRestoreIndices(data->is, &indices));
639aa21023fSPierre Jolivet         }
640*eea1a2f1SRaphael Zanella         PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)pc), PETSC_STATIC_ARRAY_LENGTH(sizes), sizes, PETSC_USE_POINTER, &is));
641*eea1a2f1SRaphael Zanella         PetscCall(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)pc), std::string(prefix + "_sizes_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer));
642aa21023fSPierre Jolivet         PetscCall(ISView(is, subviewer));
643aa21023fSPierre Jolivet         PetscCall(PetscViewerDestroy(&subviewer));
644aa21023fSPierre Jolivet         PetscCall(ISDestroy(&is));
645aa21023fSPierre Jolivet       }
646aa21023fSPierre Jolivet     }
647f1580f4eSBarry Smith   }
6483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
649f1580f4eSBarry Smith }
650f1580f4eSBarry Smith 
651d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCPreSolve_HPDDM(PC pc, KSP ksp, Vec, Vec)
652d71ae5a4SJacob Faibussowitsch {
653f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
654f1580f4eSBarry Smith   Mat       A;
655cdbd50ebSPierre Jolivet   PetscBool flg;
656f1580f4eSBarry Smith 
657f1580f4eSBarry Smith   PetscFunctionBegin;
658f1580f4eSBarry Smith   if (ksp) {
659f1580f4eSBarry Smith     PetscCall(PetscObjectTypeCompare((PetscObject)ksp, KSPLSQR, &flg));
660f1580f4eSBarry Smith     if (flg && !data->normal) {
661db4a47b3SPierre Jolivet       PetscCall(KSPGetOperators(ksp, &A, nullptr));
662db4a47b3SPierre Jolivet       PetscCall(MatCreateVecs(A, nullptr, &data->normal)); /* temporary Vec used in PCApply_HPDDMShell() for coarse grid corrections */
663cdbd50ebSPierre Jolivet     } else if (!flg) {
664cdbd50ebSPierre Jolivet       PetscCall(PetscObjectTypeCompareAny((PetscObject)ksp, &flg, KSPCG, KSPGROPPCG, KSPPIPECG, KSPPIPECGRR, KSPPIPELCG, KSPPIPEPRCG, KSPPIPECG2, KSPSTCG, KSPFCG, KSPPIPEFCG, KSPMINRES, KSPNASH, KSPSYMMLQ, ""));
665cdbd50ebSPierre Jolivet       if (!flg) {
666cdbd50ebSPierre Jolivet         PetscCall(PetscObjectTypeCompare((PetscObject)ksp, KSPHPDDM, &flg));
667cdbd50ebSPierre Jolivet         if (flg) {
668cdbd50ebSPierre Jolivet           KSPHPDDMType type;
669811e8887SPierre Jolivet 
670cdbd50ebSPierre Jolivet           PetscCall(KSPHPDDMGetType(ksp, &type));
671cdbd50ebSPierre Jolivet           flg = (type == KSP_HPDDM_TYPE_CG || type == KSP_HPDDM_TYPE_BCG || type == KSP_HPDDM_TYPE_BFBCG ? PETSC_TRUE : PETSC_FALSE);
672cdbd50ebSPierre Jolivet         }
673cdbd50ebSPierre Jolivet       }
674cdbd50ebSPierre Jolivet     }
675cdbd50ebSPierre Jolivet     if (flg) {
676cdbd50ebSPierre Jolivet       if (data->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED) {
677cdbd50ebSPierre Jolivet         PetscCall(PetscOptionsHasName(((PetscObject)pc)->options, ((PetscObject)pc)->prefix, "-pc_hpddm_coarse_correction", &flg));
678cdbd50ebSPierre Jolivet         PetscCheck(flg, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_INCOMP, "PCHPDDMCoarseCorrectionType %s is known to be not symmetric, but KSPType %s requires a symmetric PC, if you insist on using this configuration, use the additional option -%spc_hpddm_coarse_correction %s, or alternatively, switch to a symmetric PCHPDDMCoarseCorrectionType such as %s",
679cdbd50ebSPierre Jolivet                    PCHPDDMCoarseCorrectionTypes[data->correction], ((PetscObject)ksp)->type_name, ((PetscObject)pc)->prefix ? ((PetscObject)pc)->prefix : "", PCHPDDMCoarseCorrectionTypes[data->correction], PCHPDDMCoarseCorrectionTypes[PC_HPDDM_COARSE_CORRECTION_BALANCED]);
680cdbd50ebSPierre Jolivet       }
681cdbd50ebSPierre Jolivet       for (PetscInt n = 0; n < data->N; ++n) {
682cdbd50ebSPierre Jolivet         if (data->levels[n]->pc) {
683cdbd50ebSPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)data->levels[n]->pc, PCASM, &flg));
684cdbd50ebSPierre Jolivet           if (flg) {
685cdbd50ebSPierre Jolivet             PCASMType type;
686811e8887SPierre Jolivet 
687cdbd50ebSPierre Jolivet             PetscCall(PCASMGetType(data->levels[n]->pc, &type));
688cdbd50ebSPierre Jolivet             if (type == PC_ASM_RESTRICT || type == PC_ASM_INTERPOLATE) {
689cdbd50ebSPierre Jolivet               PetscCall(PetscOptionsHasName(((PetscObject)data->levels[n]->pc)->options, ((PetscObject)data->levels[n]->pc)->prefix, "-pc_asm_type", &flg));
690cdbd50ebSPierre Jolivet               PetscCheck(flg, PetscObjectComm((PetscObject)data->levels[n]->pc), PETSC_ERR_ARG_INCOMP, "PCASMType %s is known to be not symmetric, but KSPType %s requires a symmetric PC, if you insist on using this configuration, use the additional option -%spc_asm_type %s, or alternatively, switch to a symmetric PCASMType such as %s", PCASMTypes[type],
691cdbd50ebSPierre Jolivet                          ((PetscObject)ksp)->type_name, ((PetscObject)data->levels[n]->pc)->prefix, PCASMTypes[type], PCASMTypes[PC_ASM_BASIC]);
692cdbd50ebSPierre Jolivet             }
693cdbd50ebSPierre Jolivet           }
694cdbd50ebSPierre Jolivet         }
695cdbd50ebSPierre Jolivet       }
696f1580f4eSBarry Smith     }
697f1580f4eSBarry Smith   }
6983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
699f1580f4eSBarry Smith }
700f1580f4eSBarry Smith 
701d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetUp_HPDDMShell(PC pc)
702d71ae5a4SJacob Faibussowitsch {
703f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
704f1580f4eSBarry Smith   Mat             A, P;
705f1580f4eSBarry Smith   Vec             x;
706f1580f4eSBarry Smith   const char     *pcpre;
707f1580f4eSBarry Smith 
708f1580f4eSBarry Smith   PetscFunctionBegin;
709f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
710f1580f4eSBarry Smith   PetscCall(KSPGetOptionsPrefix(ctx->ksp, &pcpre));
711f1580f4eSBarry Smith   PetscCall(KSPGetOperators(ctx->ksp, &A, &P));
712f1580f4eSBarry Smith   /* smoother */
713f1580f4eSBarry Smith   PetscCall(PCSetOptionsPrefix(ctx->pc, pcpre));
714f1580f4eSBarry Smith   PetscCall(PCSetOperators(ctx->pc, A, P));
715f1580f4eSBarry Smith   if (!ctx->v[0]) {
716f1580f4eSBarry Smith     PetscCall(VecDuplicateVecs(ctx->D, 1, &ctx->v[0]));
717f1580f4eSBarry Smith     if (!std::is_same<PetscScalar, PetscReal>::value) PetscCall(VecDestroy(&ctx->D));
718db4a47b3SPierre Jolivet     PetscCall(MatCreateVecs(A, &x, nullptr));
719f1580f4eSBarry Smith     PetscCall(VecDuplicateVecs(x, 2, &ctx->v[1]));
720f1580f4eSBarry Smith     PetscCall(VecDestroy(&x));
721f1580f4eSBarry Smith   }
722f1580f4eSBarry Smith   std::fill_n(ctx->V, 3, nullptr);
7233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
724f1580f4eSBarry Smith }
725f1580f4eSBarry Smith 
726f1580f4eSBarry Smith template <class Type, typename std::enable_if<std::is_same<Type, Vec>::value>::type * = nullptr>
727d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMDeflate_Private(PC pc, Type x, Type y)
728d71ae5a4SJacob Faibussowitsch {
729f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
730f1580f4eSBarry Smith   PetscScalar    *out;
731f1580f4eSBarry Smith 
732f1580f4eSBarry Smith   PetscFunctionBegin;
733f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
734f1580f4eSBarry Smith   /* going from PETSc to HPDDM numbering */
735f1580f4eSBarry Smith   PetscCall(VecScatterBegin(ctx->scatter, x, ctx->v[0][0], INSERT_VALUES, SCATTER_FORWARD));
736f1580f4eSBarry Smith   PetscCall(VecScatterEnd(ctx->scatter, x, ctx->v[0][0], INSERT_VALUES, SCATTER_FORWARD));
737f1580f4eSBarry Smith   PetscCall(VecGetArrayWrite(ctx->v[0][0], &out));
738db4a47b3SPierre Jolivet   ctx->P->deflation<false>(nullptr, out, 1); /* y = Q x */
739f1580f4eSBarry Smith   PetscCall(VecRestoreArrayWrite(ctx->v[0][0], &out));
740f1580f4eSBarry Smith   /* going from HPDDM to PETSc numbering */
741f1580f4eSBarry Smith   PetscCall(VecScatterBegin(ctx->scatter, ctx->v[0][0], y, INSERT_VALUES, SCATTER_REVERSE));
742f1580f4eSBarry Smith   PetscCall(VecScatterEnd(ctx->scatter, ctx->v[0][0], y, INSERT_VALUES, SCATTER_REVERSE));
7433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
744f1580f4eSBarry Smith }
745f1580f4eSBarry Smith 
746f1580f4eSBarry Smith template <class Type, typename std::enable_if<std::is_same<Type, Mat>::value>::type * = nullptr>
747d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMDeflate_Private(PC pc, Type X, Type Y)
748d71ae5a4SJacob Faibussowitsch {
749f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
750f1580f4eSBarry Smith   Vec             vX, vY, vC;
751f1580f4eSBarry Smith   PetscScalar    *out;
752811e8887SPierre Jolivet   PetscInt        N;
753f1580f4eSBarry Smith 
754f1580f4eSBarry Smith   PetscFunctionBegin;
755f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
756db4a47b3SPierre Jolivet   PetscCall(MatGetSize(X, nullptr, &N));
757f1580f4eSBarry Smith   /* going from PETSc to HPDDM numbering */
758811e8887SPierre Jolivet   for (PetscInt i = 0; i < N; ++i) {
759f1580f4eSBarry Smith     PetscCall(MatDenseGetColumnVecRead(X, i, &vX));
760f1580f4eSBarry Smith     PetscCall(MatDenseGetColumnVecWrite(ctx->V[0], i, &vC));
761f1580f4eSBarry Smith     PetscCall(VecScatterBegin(ctx->scatter, vX, vC, INSERT_VALUES, SCATTER_FORWARD));
762f1580f4eSBarry Smith     PetscCall(VecScatterEnd(ctx->scatter, vX, vC, INSERT_VALUES, SCATTER_FORWARD));
763f1580f4eSBarry Smith     PetscCall(MatDenseRestoreColumnVecWrite(ctx->V[0], i, &vC));
764f1580f4eSBarry Smith     PetscCall(MatDenseRestoreColumnVecRead(X, i, &vX));
765f1580f4eSBarry Smith   }
766f1580f4eSBarry Smith   PetscCall(MatDenseGetArrayWrite(ctx->V[0], &out));
767db4a47b3SPierre Jolivet   ctx->P->deflation<false>(nullptr, out, N); /* Y = Q X */
768f1580f4eSBarry Smith   PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &out));
769f1580f4eSBarry Smith   /* going from HPDDM to PETSc numbering */
770811e8887SPierre Jolivet   for (PetscInt i = 0; i < N; ++i) {
771f1580f4eSBarry Smith     PetscCall(MatDenseGetColumnVecRead(ctx->V[0], i, &vC));
772f1580f4eSBarry Smith     PetscCall(MatDenseGetColumnVecWrite(Y, i, &vY));
773f1580f4eSBarry Smith     PetscCall(VecScatterBegin(ctx->scatter, vC, vY, INSERT_VALUES, SCATTER_REVERSE));
774f1580f4eSBarry Smith     PetscCall(VecScatterEnd(ctx->scatter, vC, vY, INSERT_VALUES, SCATTER_REVERSE));
775f1580f4eSBarry Smith     PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &vY));
776f1580f4eSBarry Smith     PetscCall(MatDenseRestoreColumnVecRead(ctx->V[0], i, &vC));
777f1580f4eSBarry Smith   }
7783ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
779f1580f4eSBarry Smith }
780f1580f4eSBarry Smith 
781f1580f4eSBarry Smith /*
782aa1539e9SPierre Jolivet      PCApply_HPDDMShell - Applies a (2) deflated, (1) additive, (3) balanced, or (4) no coarse correction. In what follows, E = Z Pmat Z^T and Q = Z^T E^-1 Z.
783f1580f4eSBarry Smith 
784f1580f4eSBarry Smith .vb
785f1580f4eSBarry Smith    (1) y =                Pmat^-1              x + Q x,
786f1580f4eSBarry Smith    (2) y =                Pmat^-1 (I - Amat Q) x + Q x (default),
787aa1539e9SPierre Jolivet    (3) y = (I - Q Amat^T) Pmat^-1 (I - Amat Q) x + Q x,
788aa1539e9SPierre Jolivet    (4) y =                Pmat^-1              x      .
789f1580f4eSBarry Smith .ve
790f1580f4eSBarry Smith 
791f1580f4eSBarry Smith    Input Parameters:
792f1580f4eSBarry Smith +     pc - preconditioner context
793f1580f4eSBarry Smith -     x - input vector
794f1580f4eSBarry Smith 
795f1580f4eSBarry Smith    Output Parameter:
796f1580f4eSBarry Smith .     y - output vector
797f1580f4eSBarry Smith 
798f1580f4eSBarry Smith    Notes:
799f1580f4eSBarry Smith      The options of Pmat^1 = pc(Pmat) are prefixed by -pc_hpddm_levels_1_pc_. Z is a tall-and-skiny matrix assembled by HPDDM. The number of processes on which (Z Pmat Z^T) is aggregated is set via -pc_hpddm_coarse_p.
800f1580f4eSBarry Smith      The options of (Z Pmat Z^T)^-1 = ksp(Z Pmat Z^T) are prefixed by -pc_hpddm_coarse_ (`KSPPREONLY` and `PCCHOLESKY` by default), unless a multilevel correction is turned on, in which case, this function is called recursively at each level except the coarsest one.
801f1580f4eSBarry Smith      (1) and (2) visit the "next" level (in terms of coarsening) once per application, while (3) visits it twice, so it is asymptotically twice costlier. (2) is not symmetric even if both Amat and Pmat are symmetric.
802f1580f4eSBarry Smith 
803f1580f4eSBarry Smith    Level: advanced
804f1580f4eSBarry Smith 
805f1580f4eSBarry Smith    Developer Note:
806f1580f4eSBarry Smith    Since this is not an actual manual page the material below should be moved to an appropriate manual page with the appropriate context, i.e. explaining when it is used and how
807f1580f4eSBarry Smith    to trigger it. Likely the manual page is `PCHPDDM`
808f1580f4eSBarry Smith 
809562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMCoarseCorrectionType`
810f1580f4eSBarry Smith */
811d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_HPDDMShell(PC pc, Vec x, Vec y)
812d71ae5a4SJacob Faibussowitsch {
813f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
814f1580f4eSBarry Smith   Mat             A;
815f1580f4eSBarry Smith 
816f1580f4eSBarry Smith   PetscFunctionBegin;
817f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
818f1580f4eSBarry Smith   PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object");
819db4a47b3SPierre Jolivet   PetscCall(KSPGetOperators(ctx->ksp, &A, nullptr));
820aa1539e9SPierre Jolivet   if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_NONE) PetscCall(PCApply(ctx->pc, x, y)); /* y = M^-1 x */
821aa1539e9SPierre Jolivet   else {
822f1580f4eSBarry Smith     PetscCall(PCHPDDMDeflate_Private(pc, x, y)); /* y = Q x */
823f1580f4eSBarry Smith     if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
824f1580f4eSBarry Smith       if (!ctx->parent->normal || ctx != ctx->parent->levels[0]) PetscCall(MatMult(A, y, ctx->v[1][0])); /* y = A Q x     */
825f1580f4eSBarry Smith       else { /* KSPLSQR and finest level */ PetscCall(MatMult(A, y, ctx->parent->normal));               /* y = A Q x     */
826f1580f4eSBarry Smith         PetscCall(MatMultHermitianTranspose(A, ctx->parent->normal, ctx->v[1][0]));                      /* y = A^T A Q x */
827f1580f4eSBarry Smith       }
828f1580f4eSBarry Smith       PetscCall(VecWAXPY(ctx->v[1][1], -1.0, ctx->v[1][0], x)); /* y = (I - A Q) x                             */
829f1580f4eSBarry Smith       PetscCall(PCApply(ctx->pc, ctx->v[1][1], ctx->v[1][0]));  /* y = M^-1 (I - A Q) x                        */
830f1580f4eSBarry Smith       if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
831f1580f4eSBarry Smith         if (!ctx->parent->normal || ctx != ctx->parent->levels[0]) PetscCall(MatMultHermitianTranspose(A, ctx->v[1][0], ctx->v[1][1])); /* z = A^T y */
832f1580f4eSBarry Smith         else {
833f1580f4eSBarry Smith           PetscCall(MatMult(A, ctx->v[1][0], ctx->parent->normal));
834f1580f4eSBarry Smith           PetscCall(MatMultHermitianTranspose(A, ctx->parent->normal, ctx->v[1][1])); /* z = A^T A y           */
835f1580f4eSBarry Smith         }
836f1580f4eSBarry Smith         PetscCall(PCHPDDMDeflate_Private(pc, ctx->v[1][1], ctx->v[1][1]));
837f1580f4eSBarry Smith         PetscCall(VecAXPBYPCZ(y, -1.0, 1.0, 1.0, ctx->v[1][1], ctx->v[1][0])); /* y = (I - Q A^T) y + Q x      */
838f1580f4eSBarry Smith       } else PetscCall(VecAXPY(y, 1.0, ctx->v[1][0]));                         /* y = Q M^-1 (I - A Q) x + Q x */
839f1580f4eSBarry Smith     } else {
840f1580f4eSBarry Smith       PetscCheck(ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_ADDITIVE, PetscObjectComm((PetscObject)pc), PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with an unknown PCHPDDMCoarseCorrectionType %d", ctx->parent->correction);
841f1580f4eSBarry Smith       PetscCall(PCApply(ctx->pc, x, ctx->v[1][0]));
842f1580f4eSBarry Smith       PetscCall(VecAXPY(y, 1.0, ctx->v[1][0])); /* y = M^-1 x + Q x */
843f1580f4eSBarry Smith     }
844aa1539e9SPierre Jolivet   }
8453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
846f1580f4eSBarry Smith }
847f1580f4eSBarry Smith 
848f1580f4eSBarry Smith /*
849f1580f4eSBarry Smith      PCMatApply_HPDDMShell - Variant of PCApply_HPDDMShell() for blocks of vectors.
850f1580f4eSBarry Smith 
851f1580f4eSBarry Smith    Input Parameters:
852f1580f4eSBarry Smith +     pc - preconditioner context
853f1580f4eSBarry Smith -     X - block of input vectors
854f1580f4eSBarry Smith 
855f1580f4eSBarry Smith    Output Parameter:
856f1580f4eSBarry Smith .     Y - block of output vectors
857f1580f4eSBarry Smith 
858f1580f4eSBarry Smith    Level: advanced
859f1580f4eSBarry Smith 
860562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCApply_HPDDMShell()`, `PCHPDDMCoarseCorrectionType`
861f1580f4eSBarry Smith */
862d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_HPDDMShell(PC pc, Mat X, Mat Y)
863d71ae5a4SJacob Faibussowitsch {
864f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
865f1580f4eSBarry Smith   Mat             A, *ptr;
866db4a47b3SPierre Jolivet   PetscContainer  container = nullptr;
867f1580f4eSBarry Smith   PetscScalar    *array;
868f1580f4eSBarry Smith   PetscInt        m, M, N, prev = 0;
869f1580f4eSBarry Smith   PetscBool       reset = PETSC_FALSE;
870f1580f4eSBarry Smith 
871f1580f4eSBarry Smith   PetscFunctionBegin;
872f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
873f1580f4eSBarry Smith   PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object");
874db4a47b3SPierre Jolivet   PetscCall(MatGetSize(X, nullptr, &N));
875db4a47b3SPierre Jolivet   PetscCall(KSPGetOperators(ctx->ksp, &A, nullptr));
876aa1539e9SPierre Jolivet   if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_NONE) PetscCall(PCMatApply(ctx->pc, X, Y));
877aa1539e9SPierre Jolivet   else {
878f1580f4eSBarry Smith     PetscCall(PetscObjectQuery((PetscObject)A, "_HPDDM_MatProduct", (PetscObject *)&container));
879f1580f4eSBarry Smith     if (container) { /* MatProduct container already attached */
880f1580f4eSBarry Smith       PetscCall(PetscContainerGetPointer(container, (void **)&ptr));
881f1580f4eSBarry Smith       if (ptr[1] != ctx->V[2]) /* Mat has changed or may have been set first in KSPHPDDM */
882f1580f4eSBarry Smith         for (m = 0; m < 2; ++m) {
883f1580f4eSBarry Smith           PetscCall(MatDestroy(ctx->V + m + 1));
884f1580f4eSBarry Smith           ctx->V[m + 1] = ptr[m];
885f1580f4eSBarry Smith           PetscCall(PetscObjectReference((PetscObject)ctx->V[m + 1]));
886f1580f4eSBarry Smith         }
887f1580f4eSBarry Smith     }
888db4a47b3SPierre Jolivet     if (ctx->V[1]) PetscCall(MatGetSize(ctx->V[1], nullptr, &prev));
889f1580f4eSBarry Smith     if (N != prev || !ctx->V[0]) {
890f1580f4eSBarry Smith       PetscCall(MatDestroy(ctx->V));
891f1580f4eSBarry Smith       PetscCall(VecGetLocalSize(ctx->v[0][0], &m));
892db4a47b3SPierre Jolivet       PetscCall(MatCreateDense(PetscObjectComm((PetscObject)pc), m, PETSC_DECIDE, PETSC_DECIDE, N, nullptr, ctx->V));
893f1580f4eSBarry Smith       if (N != prev) {
894f1580f4eSBarry Smith         PetscCall(MatDestroy(ctx->V + 1));
895f1580f4eSBarry Smith         PetscCall(MatDestroy(ctx->V + 2));
896db4a47b3SPierre Jolivet         PetscCall(MatGetLocalSize(X, &m, nullptr));
897db4a47b3SPierre Jolivet         PetscCall(MatGetSize(X, &M, nullptr));
898f1580f4eSBarry Smith         if (ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) PetscCall(MatDenseGetArrayWrite(ctx->V[0], &array));
899db4a47b3SPierre Jolivet         else array = nullptr;
900f1580f4eSBarry Smith         PetscCall(MatCreateDense(PetscObjectComm((PetscObject)pc), m, PETSC_DECIDE, M, N, array, ctx->V + 1));
901f1580f4eSBarry Smith         if (ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &array));
902db4a47b3SPierre Jolivet         PetscCall(MatCreateDense(PetscObjectComm((PetscObject)pc), m, PETSC_DECIDE, M, N, nullptr, ctx->V + 2));
903db4a47b3SPierre Jolivet         PetscCall(MatProductCreateWithMat(A, Y, nullptr, ctx->V[1]));
904f1580f4eSBarry Smith         PetscCall(MatProductSetType(ctx->V[1], MATPRODUCT_AB));
905f1580f4eSBarry Smith         PetscCall(MatProductSetFromOptions(ctx->V[1]));
906f1580f4eSBarry Smith         PetscCall(MatProductSymbolic(ctx->V[1]));
907715c1178SPierre Jolivet         if (!container) PetscCall(PetscObjectContainerCompose((PetscObject)A, "_HPDDM_MatProduct", ctx->V + 1, nullptr)); /* no MatProduct container attached, create one to be queried in KSPHPDDM or at the next call to PCMatApply() */
908715c1178SPierre Jolivet         else PetscCall(PetscContainerSetPointer(container, ctx->V + 1));                                                  /* need to compose B and D from MatProductCreateWithMat(A, B, NULL, D), which are stored in the contiguous array ctx->V */
909f1580f4eSBarry Smith       }
910f1580f4eSBarry Smith       if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
911db4a47b3SPierre Jolivet         PetscCall(MatProductCreateWithMat(A, ctx->V[1], nullptr, ctx->V[2]));
912f1580f4eSBarry Smith         PetscCall(MatProductSetType(ctx->V[2], MATPRODUCT_AtB));
913f1580f4eSBarry Smith         PetscCall(MatProductSetFromOptions(ctx->V[2]));
914f1580f4eSBarry Smith         PetscCall(MatProductSymbolic(ctx->V[2]));
915f1580f4eSBarry Smith       }
916f1580f4eSBarry Smith       ctx->P->start(N);
917f1580f4eSBarry Smith     }
918f1580f4eSBarry Smith     if (N == prev || container) { /* when MatProduct container is attached, always need to MatProductReplaceMats() since KSPHPDDM may have replaced the Mat as well */
919db4a47b3SPierre Jolivet       PetscCall(MatProductReplaceMats(nullptr, Y, nullptr, ctx->V[1]));
920f1580f4eSBarry Smith       if (container && ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) {
921f1580f4eSBarry Smith         PetscCall(MatDenseGetArrayWrite(ctx->V[0], &array));
922f1580f4eSBarry Smith         PetscCall(MatDensePlaceArray(ctx->V[1], array));
923f1580f4eSBarry Smith         PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &array));
924f1580f4eSBarry Smith         reset = PETSC_TRUE;
925f1580f4eSBarry Smith       }
926f1580f4eSBarry Smith     }
927f1580f4eSBarry Smith     PetscCall(PCHPDDMDeflate_Private(pc, X, Y));
928f1580f4eSBarry Smith     if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
929f1580f4eSBarry Smith       PetscCall(MatProductNumeric(ctx->V[1]));
930f1580f4eSBarry Smith       PetscCall(MatCopy(ctx->V[1], ctx->V[2], SAME_NONZERO_PATTERN));
931f1580f4eSBarry Smith       PetscCall(MatAXPY(ctx->V[2], -1.0, X, SAME_NONZERO_PATTERN));
932f1580f4eSBarry Smith       PetscCall(PCMatApply(ctx->pc, ctx->V[2], ctx->V[1]));
933f1580f4eSBarry Smith       if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
934f1580f4eSBarry Smith         PetscCall(MatProductNumeric(ctx->V[2]));
935f1580f4eSBarry Smith         PetscCall(PCHPDDMDeflate_Private(pc, ctx->V[2], ctx->V[2]));
936f1580f4eSBarry Smith         PetscCall(MatAXPY(ctx->V[1], -1.0, ctx->V[2], SAME_NONZERO_PATTERN));
937f1580f4eSBarry Smith       }
938f1580f4eSBarry Smith       PetscCall(MatAXPY(Y, -1.0, ctx->V[1], SAME_NONZERO_PATTERN));
939f1580f4eSBarry Smith     } else {
940f1580f4eSBarry Smith       PetscCheck(ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_ADDITIVE, PetscObjectComm((PetscObject)pc), PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with an unknown PCHPDDMCoarseCorrectionType %d", ctx->parent->correction);
941f1580f4eSBarry Smith       PetscCall(PCMatApply(ctx->pc, X, ctx->V[1]));
942f1580f4eSBarry Smith       PetscCall(MatAXPY(Y, 1.0, ctx->V[1], SAME_NONZERO_PATTERN));
943f1580f4eSBarry Smith     }
944f1580f4eSBarry Smith     if (reset) PetscCall(MatDenseResetArray(ctx->V[1]));
945aa1539e9SPierre Jolivet   }
9463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
947f1580f4eSBarry Smith }
948f1580f4eSBarry Smith 
949d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCDestroy_HPDDMShell(PC pc)
950d71ae5a4SJacob Faibussowitsch {
951f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
952f1580f4eSBarry Smith 
953f1580f4eSBarry Smith   PetscFunctionBegin;
954f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
955f1580f4eSBarry Smith   PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(ctx, PETSC_TRUE));
956f1580f4eSBarry Smith   PetscCall(VecDestroyVecs(1, &ctx->v[0]));
957f1580f4eSBarry Smith   PetscCall(VecDestroyVecs(2, &ctx->v[1]));
958f4f49eeaSPierre Jolivet   PetscCall(PetscObjectCompose((PetscObject)ctx->pc->mat, "_HPDDM_MatProduct", nullptr));
959f1580f4eSBarry Smith   PetscCall(MatDestroy(ctx->V));
960f1580f4eSBarry Smith   PetscCall(MatDestroy(ctx->V + 1));
961f1580f4eSBarry Smith   PetscCall(MatDestroy(ctx->V + 2));
962f1580f4eSBarry Smith   PetscCall(VecDestroy(&ctx->D));
96301e3c840SPierre Jolivet   PetscCall(PetscSFDestroy(&ctx->scatter));
964f1580f4eSBarry Smith   PetscCall(PCDestroy(&ctx->pc));
9653ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
966f1580f4eSBarry Smith }
967f1580f4eSBarry Smith 
9689bb5c669SPierre Jolivet template <class Type, bool T = false, typename std::enable_if<std::is_same<Type, Vec>::value>::type * = nullptr>
9699bb5c669SPierre Jolivet static inline PetscErrorCode PCApply_Schur_Private(std::tuple<KSP, IS, Vec[2]> *p, PC factor, Type x, Type y)
9709bb5c669SPierre Jolivet {
9719bb5c669SPierre Jolivet   PetscFunctionBegin;
9729bb5c669SPierre Jolivet   PetscCall(VecISCopy(std::get<2>(*p)[0], std::get<1>(*p), SCATTER_FORWARD, x));
9739bb5c669SPierre Jolivet   if (!T) PetscCall(PCApply(factor, std::get<2>(*p)[0], std::get<2>(*p)[1]));
9749bb5c669SPierre Jolivet   else PetscCall(PCApplyTranspose(factor, std::get<2>(*p)[0], std::get<2>(*p)[1]));
9759bb5c669SPierre Jolivet   PetscCall(VecISCopy(std::get<2>(*p)[1], std::get<1>(*p), SCATTER_REVERSE, y));
9769bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
9779bb5c669SPierre Jolivet }
9789bb5c669SPierre Jolivet 
9799bb5c669SPierre Jolivet template <class Type, bool = false, typename std::enable_if<std::is_same<Type, Mat>::value>::type * = nullptr>
9809bb5c669SPierre Jolivet static inline PetscErrorCode PCApply_Schur_Private(std::tuple<KSP, IS, Vec[2]> *p, PC factor, Type X, Type Y)
9819bb5c669SPierre Jolivet {
9829bb5c669SPierre Jolivet   Mat B[2];
9839bb5c669SPierre Jolivet   Vec x, y;
9849bb5c669SPierre Jolivet 
9859bb5c669SPierre Jolivet   PetscFunctionBegin;
9869bb5c669SPierre Jolivet   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, factor->mat->rmap->n, X->cmap->n, nullptr, B));
9879bb5c669SPierre Jolivet   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, factor->mat->rmap->n, X->cmap->n, nullptr, B + 1));
9889bb5c669SPierre Jolivet   for (PetscInt i = 0; i < X->cmap->n; ++i) {
9899bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecRead(X, i, &x));
9909bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecWrite(B[0], i, &y));
9919bb5c669SPierre Jolivet     PetscCall(VecISCopy(y, std::get<1>(*p), SCATTER_FORWARD, x));
9929bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecWrite(B[0], i, &y));
9939bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecRead(X, i, &x));
9949bb5c669SPierre Jolivet   }
9959bb5c669SPierre Jolivet   PetscCall(PCMatApply(factor, B[0], B[1]));
9969bb5c669SPierre Jolivet   PetscCall(MatDestroy(B));
9979bb5c669SPierre Jolivet   for (PetscInt i = 0; i < X->cmap->n; ++i) {
9989bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecRead(B[1], i, &x));
9999bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecWrite(Y, i, &y));
10009bb5c669SPierre Jolivet     PetscCall(VecISCopy(x, std::get<1>(*p), SCATTER_REVERSE, y));
10019bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &y));
10029bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecRead(B[1], i, &x));
10039bb5c669SPierre Jolivet   }
10049bb5c669SPierre Jolivet   PetscCall(MatDestroy(B + 1));
10059bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
10069bb5c669SPierre Jolivet }
10079bb5c669SPierre Jolivet 
10089bb5c669SPierre Jolivet template <class Type = Vec, bool T = false>
10099bb5c669SPierre Jolivet static PetscErrorCode PCApply_Schur(PC pc, Type x, Type y)
10109bb5c669SPierre Jolivet {
10119bb5c669SPierre Jolivet   PC                           factor;
10129bb5c669SPierre Jolivet   Mat                          A;
10139bb5c669SPierre Jolivet   MatSolverType                type;
10149bb5c669SPierre Jolivet   PetscBool                    flg;
10159bb5c669SPierre Jolivet   std::tuple<KSP, IS, Vec[2]> *p;
10169bb5c669SPierre Jolivet 
10179bb5c669SPierre Jolivet   PetscFunctionBegin;
10189bb5c669SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
10199bb5c669SPierre Jolivet   PetscCall(KSPGetPC(std::get<0>(*p), &factor));
10209bb5c669SPierre Jolivet   PetscCall(PCFactorGetMatSolverType(factor, &type));
10219bb5c669SPierre Jolivet   PetscCall(PCFactorGetMatrix(factor, &A));
10229bb5c669SPierre Jolivet   PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg));
10239bb5c669SPierre Jolivet   if (flg) {
10249bb5c669SPierre Jolivet     PetscCheck(PetscDefined(HAVE_MUMPS), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent MatSolverType");
10259bb5c669SPierre Jolivet #if PetscDefined(HAVE_MUMPS)
10269bb5c669SPierre Jolivet     PetscCall(MatMumpsSetIcntl(A, 26, 0));
10279bb5c669SPierre Jolivet #endif
10289bb5c669SPierre Jolivet   } else {
10299bb5c669SPierre Jolivet     PetscCall(PetscStrcmp(type, MATSOLVERMKL_PARDISO, &flg));
10309bb5c669SPierre Jolivet     PetscCheck(flg && PetscDefined(HAVE_MKL_PARDISO), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent MatSolverType");
10319bb5c669SPierre Jolivet     flg = PETSC_FALSE;
10329bb5c669SPierre Jolivet #if PetscDefined(HAVE_MKL_PARDISO)
10339bb5c669SPierre Jolivet     PetscCall(MatMkl_PardisoSetCntl(A, 70, 1));
10349bb5c669SPierre Jolivet #endif
10359bb5c669SPierre Jolivet   }
10369bb5c669SPierre Jolivet   PetscCall(PCApply_Schur_Private<Type, T>(p, factor, x, y));
10379bb5c669SPierre Jolivet   if (flg) {
10389bb5c669SPierre Jolivet #if PetscDefined(HAVE_MUMPS)
10399bb5c669SPierre Jolivet     PetscCall(MatMumpsSetIcntl(A, 26, -1));
10409bb5c669SPierre Jolivet #endif
10419bb5c669SPierre Jolivet   } else {
10429bb5c669SPierre Jolivet #if PetscDefined(HAVE_MKL_PARDISO)
10439bb5c669SPierre Jolivet     PetscCall(MatMkl_PardisoSetCntl(A, 70, 0));
10449bb5c669SPierre Jolivet #endif
10459bb5c669SPierre Jolivet   }
10469bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
10479bb5c669SPierre Jolivet }
10489bb5c669SPierre Jolivet 
10499bb5c669SPierre Jolivet static PetscErrorCode PCDestroy_Schur(PC pc)
10509bb5c669SPierre Jolivet {
10519bb5c669SPierre Jolivet   std::tuple<KSP, IS, Vec[2]> *p;
10529bb5c669SPierre Jolivet 
10539bb5c669SPierre Jolivet   PetscFunctionBegin;
10549bb5c669SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
10559bb5c669SPierre Jolivet   PetscCall(ISDestroy(&std::get<1>(*p)));
10569bb5c669SPierre Jolivet   PetscCall(VecDestroy(std::get<2>(*p)));
10579bb5c669SPierre Jolivet   PetscCall(VecDestroy(std::get<2>(*p) + 1));
10589bb5c669SPierre Jolivet   PetscCall(PetscFree(p));
10599bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
10609bb5c669SPierre Jolivet }
10619bb5c669SPierre Jolivet 
1062d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSolve_Private(const PC_HPDDM_Level *ctx, PetscScalar *rhs, const unsigned short &mu)
1063d71ae5a4SJacob Faibussowitsch {
1064f1580f4eSBarry Smith   Mat      B, X;
1065f1580f4eSBarry Smith   PetscInt n, N, j = 0;
1066f1580f4eSBarry Smith 
1067f1580f4eSBarry Smith   PetscFunctionBegin;
1068db4a47b3SPierre Jolivet   PetscCall(KSPGetOperators(ctx->ksp, &B, nullptr));
1069db4a47b3SPierre Jolivet   PetscCall(MatGetLocalSize(B, &n, nullptr));
1070db4a47b3SPierre Jolivet   PetscCall(MatGetSize(B, &N, nullptr));
1071f1580f4eSBarry Smith   if (ctx->parent->log_separate) {
1072f1580f4eSBarry Smith     j = std::distance(ctx->parent->levels, std::find(ctx->parent->levels, ctx->parent->levels + ctx->parent->N, ctx));
1073db4a47b3SPierre Jolivet     PetscCall(PetscLogEventBegin(PC_HPDDM_Solve[j], ctx->ksp, nullptr, nullptr, nullptr));
1074f1580f4eSBarry Smith   }
1075f1580f4eSBarry Smith   if (mu == 1) {
1076f1580f4eSBarry Smith     if (!ctx->ksp->vec_rhs) {
1077db4a47b3SPierre Jolivet       PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)ctx->ksp), 1, n, N, nullptr, &ctx->ksp->vec_rhs));
1078f1580f4eSBarry Smith       PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)ctx->ksp), n, N, &ctx->ksp->vec_sol));
1079f1580f4eSBarry Smith     }
1080f1580f4eSBarry Smith     PetscCall(VecPlaceArray(ctx->ksp->vec_rhs, rhs));
1081db4a47b3SPierre Jolivet     PetscCall(KSPSolve(ctx->ksp, nullptr, nullptr));
1082f1580f4eSBarry Smith     PetscCall(VecCopy(ctx->ksp->vec_sol, ctx->ksp->vec_rhs));
1083f1580f4eSBarry Smith     PetscCall(VecResetArray(ctx->ksp->vec_rhs));
1084f1580f4eSBarry Smith   } else {
1085f1580f4eSBarry Smith     PetscCall(MatCreateDense(PetscObjectComm((PetscObject)ctx->ksp), n, PETSC_DECIDE, N, mu, rhs, &B));
1086db4a47b3SPierre Jolivet     PetscCall(MatCreateDense(PetscObjectComm((PetscObject)ctx->ksp), n, PETSC_DECIDE, N, mu, nullptr, &X));
1087f1580f4eSBarry Smith     PetscCall(KSPMatSolve(ctx->ksp, B, X));
1088f1580f4eSBarry Smith     PetscCall(MatCopy(X, B, SAME_NONZERO_PATTERN));
1089f1580f4eSBarry Smith     PetscCall(MatDestroy(&X));
1090f1580f4eSBarry Smith     PetscCall(MatDestroy(&B));
1091f1580f4eSBarry Smith   }
1092db4a47b3SPierre Jolivet   if (ctx->parent->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Solve[j], ctx->ksp, nullptr, nullptr, nullptr));
10933ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1094f1580f4eSBarry Smith }
1095f1580f4eSBarry Smith 
1096d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetUpNeumannOverlap_Private(PC pc)
1097d71ae5a4SJacob Faibussowitsch {
1098f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
1099f1580f4eSBarry Smith 
1100f1580f4eSBarry Smith   PetscFunctionBegin;
1101f1580f4eSBarry Smith   if (data->setup) {
1102f1580f4eSBarry Smith     Mat       P;
1103db4a47b3SPierre Jolivet     Vec       x, xt = nullptr;
1104f1580f4eSBarry Smith     PetscReal t = 0.0, s = 0.0;
1105f1580f4eSBarry Smith 
1106db4a47b3SPierre Jolivet     PetscCall(PCGetOperators(pc, nullptr, &P));
1107f1580f4eSBarry Smith     PetscCall(PetscObjectQuery((PetscObject)P, "__SNES_latest_X", (PetscObject *)&x));
1108f1580f4eSBarry Smith     PetscCallBack("PCHPDDM Neumann callback", (*data->setup)(data->aux, t, x, xt, s, data->is, data->setup_ctx));
1109f1580f4eSBarry Smith   }
11103ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1111f1580f4eSBarry Smith }
1112f1580f4eSBarry Smith 
1113d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMCreateSubMatrices_Private(Mat mat, PetscInt n, const IS *, const IS *, MatReuse scall, Mat *submat[])
1114d71ae5a4SJacob Faibussowitsch {
1115f1580f4eSBarry Smith   Mat       A;
111613044ca3SPierre Jolivet   PetscBool flg;
1117f1580f4eSBarry Smith 
1118f1580f4eSBarry Smith   PetscFunctionBegin;
1119f1580f4eSBarry Smith   PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "MatCreateSubMatrices() called to extract %" PetscInt_FMT " submatrices, which is different than 1", n);
1120f1580f4eSBarry Smith   /* previously composed Mat */
1121f1580f4eSBarry Smith   PetscCall(PetscObjectQuery((PetscObject)mat, "_PCHPDDM_SubMatrices", (PetscObject *)&A));
1122f1580f4eSBarry Smith   PetscCheck(A, PETSC_COMM_SELF, PETSC_ERR_PLIB, "SubMatrices not found in Mat");
112313044ca3SPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSCHURCOMPLEMENT, &flg)); /* MATSCHURCOMPLEMENT has neither a MatDuplicate() nor a MatCopy() implementation */
1124f1580f4eSBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
11256f2c871aSStefano Zampini     PetscCall(PetscCalloc1(2, submat)); /* allocate an extra Mat to avoid errors in MatDestroySubMatrices_Dummy() */
112613044ca3SPierre Jolivet     if (!flg) PetscCall(MatDuplicate(A, MAT_COPY_VALUES, *submat));
112713044ca3SPierre Jolivet   } else if (!flg) PetscCall(MatCopy(A, (*submat)[0], SAME_NONZERO_PATTERN));
112813044ca3SPierre Jolivet   if (flg) {
1129bf583f0cSPierre Jolivet     PetscCall(MatDestroy(*submat)); /* previously created Mat has to be destroyed */
113013044ca3SPierre Jolivet     (*submat)[0] = A;
113113044ca3SPierre Jolivet     PetscCall(PetscObjectReference((PetscObject)A));
113213044ca3SPierre Jolivet   }
11333ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1134f1580f4eSBarry Smith }
1135f1580f4eSBarry Smith 
1136d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMCommunicationAvoidingPCASM_Private(PC pc, Mat C, PetscBool sorted)
1137d71ae5a4SJacob Faibussowitsch {
1138f1580f4eSBarry Smith   void (*op)(void);
1139f1580f4eSBarry Smith 
1140f1580f4eSBarry Smith   PetscFunctionBegin;
1141f1580f4eSBarry Smith   /* previously-composed Mat */
1142f1580f4eSBarry Smith   PetscCall(PetscObjectCompose((PetscObject)pc->pmat, "_PCHPDDM_SubMatrices", (PetscObject)C));
1143f1580f4eSBarry Smith   PetscCall(MatGetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, &op));
1144f1580f4eSBarry Smith   /* trick suggested by Barry https://lists.mcs.anl.gov/pipermail/petsc-dev/2020-January/025491.html */
1145f1580f4eSBarry Smith   PetscCall(MatSetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, (void (*)(void))PCHPDDMCreateSubMatrices_Private));
1146f1580f4eSBarry Smith   if (sorted) PetscCall(PCASMSetSortIndices(pc, PETSC_FALSE)); /* everything is already sorted */
1147f1580f4eSBarry Smith   PetscCall(PCSetFromOptions(pc));                             /* otherwise -pc_hpddm_levels_1_pc_asm_sub_mat_type is not used */
1148f1580f4eSBarry Smith   PetscCall(PCSetUp(pc));
1149f1580f4eSBarry Smith   /* reset MatCreateSubMatrices() */
1150f1580f4eSBarry Smith   PetscCall(MatSetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, op));
1151db4a47b3SPierre Jolivet   PetscCall(PetscObjectCompose((PetscObject)pc->pmat, "_PCHPDDM_SubMatrices", nullptr));
11523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1153f1580f4eSBarry Smith }
1154f1580f4eSBarry Smith 
1155d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMPermute_Private(IS is, IS in_is, IS *out_is, Mat in_C, Mat *out_C, IS *p)
1156d71ae5a4SJacob Faibussowitsch {
1157f1580f4eSBarry Smith   IS                           perm;
1158f1580f4eSBarry Smith   const PetscInt              *ptr;
1159811e8887SPierre Jolivet   PetscInt                    *concatenate, size, bs;
1160f1580f4eSBarry Smith   std::map<PetscInt, PetscInt> order;
1161f1580f4eSBarry Smith   PetscBool                    sorted;
1162f1580f4eSBarry Smith 
1163f1580f4eSBarry Smith   PetscFunctionBegin;
1164cf67ef9dSPierre Jolivet   PetscValidHeaderSpecific(is, IS_CLASSID, 1);
1165cf67ef9dSPierre Jolivet   PetscValidHeaderSpecific(in_C, MAT_CLASSID, 4);
1166f1580f4eSBarry Smith   PetscCall(ISSorted(is, &sorted));
1167f1580f4eSBarry Smith   if (!sorted) {
1168f1580f4eSBarry Smith     PetscCall(ISGetLocalSize(is, &size));
1169f1580f4eSBarry Smith     PetscCall(ISGetIndices(is, &ptr));
1170b07dfdedSPierre Jolivet     PetscCall(ISGetBlockSize(is, &bs));
1171f1580f4eSBarry Smith     /* MatCreateSubMatrices(), called by PCASM, follows the global numbering of Pmat */
1172811e8887SPierre Jolivet     for (PetscInt n = 0; n < size; n += bs) order.insert(std::make_pair(ptr[n] / bs, n / bs));
1173f1580f4eSBarry Smith     PetscCall(ISRestoreIndices(is, &ptr));
1174b07dfdedSPierre Jolivet     size /= bs;
1175f1580f4eSBarry Smith     if (out_C) {
1176f1580f4eSBarry Smith       PetscCall(PetscMalloc1(size, &concatenate));
1177f1580f4eSBarry Smith       for (const std::pair<const PetscInt, PetscInt> &i : order) *concatenate++ = i.second;
1178f1580f4eSBarry Smith       concatenate -= size;
1179b07dfdedSPierre Jolivet       PetscCall(ISCreateBlock(PetscObjectComm((PetscObject)in_C), bs, size, concatenate, PETSC_OWN_POINTER, &perm));
1180f1580f4eSBarry Smith       PetscCall(ISSetPermutation(perm));
1181f1580f4eSBarry Smith       /* permute user-provided Mat so that it matches with MatCreateSubMatrices() numbering */
1182f1580f4eSBarry Smith       PetscCall(MatPermute(in_C, perm, perm, out_C));
1183f1580f4eSBarry Smith       if (p) *p = perm;
1184f1580f4eSBarry Smith       else PetscCall(ISDestroy(&perm)); /* no need to save the permutation */
1185f1580f4eSBarry Smith     }
1186f1580f4eSBarry Smith     if (out_is) {
1187f1580f4eSBarry Smith       PetscCall(PetscMalloc1(size, &concatenate));
1188f1580f4eSBarry Smith       for (const std::pair<const PetscInt, PetscInt> &i : order) *concatenate++ = i.first;
1189f1580f4eSBarry Smith       concatenate -= size;
1190f1580f4eSBarry Smith       /* permute user-provided IS so that it matches with MatCreateSubMatrices() numbering */
1191b07dfdedSPierre Jolivet       PetscCall(ISCreateBlock(PetscObjectComm((PetscObject)in_is), bs, size, concatenate, PETSC_OWN_POINTER, out_is));
1192f1580f4eSBarry Smith     }
1193f1580f4eSBarry Smith   } else { /* input IS is sorted, nothing to permute, simply duplicate inputs when needed */
1194f1580f4eSBarry Smith     if (out_C) PetscCall(MatDuplicate(in_C, MAT_COPY_VALUES, out_C));
1195f1580f4eSBarry Smith     if (out_is) PetscCall(ISDuplicate(in_is, out_is));
1196f1580f4eSBarry Smith   }
11973ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1198f1580f4eSBarry Smith }
1199f1580f4eSBarry Smith 
12005e642048SPierre Jolivet static PetscErrorCode PCHPDDMCheckSymmetry_Private(PC pc, Mat A01, Mat A10)
120113044ca3SPierre Jolivet {
120213044ca3SPierre Jolivet   Mat       T, U = nullptr, B = nullptr;
120313044ca3SPierre Jolivet   IS        z;
120413044ca3SPierre Jolivet   PetscBool flg[2];
120513044ca3SPierre Jolivet 
120613044ca3SPierre Jolivet   PetscFunctionBegin;
120713044ca3SPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, flg));
1208811e8887SPierre Jolivet   if (flg[0]) {
1209811e8887SPierre Jolivet     PetscCall(MatShellGetScalingShifts(A10, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Mat *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED));
1210811e8887SPierre Jolivet     PetscCall(MatTransposeGetMat(A10, &U));
1211811e8887SPierre Jolivet   } else {
121213044ca3SPierre Jolivet     PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, flg + 1));
1213811e8887SPierre Jolivet     if (flg[1]) {
1214811e8887SPierre Jolivet       PetscCall(MatShellGetScalingShifts(A10, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Mat *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED));
1215811e8887SPierre Jolivet       PetscCall(MatHermitianTransposeGetMat(A10, &U));
1216811e8887SPierre Jolivet     }
121713044ca3SPierre Jolivet   }
121813044ca3SPierre Jolivet   if (U) PetscCall(MatDuplicate(U, MAT_COPY_VALUES, &T));
121913044ca3SPierre Jolivet   else PetscCall(MatHermitianTranspose(A10, MAT_INITIAL_MATRIX, &T));
122013044ca3SPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)A01, MATTRANSPOSEVIRTUAL, flg));
122113044ca3SPierre Jolivet   if (flg[0]) {
1222811e8887SPierre Jolivet     PetscCall(MatShellGetScalingShifts(A01, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Mat *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED));
122313044ca3SPierre Jolivet     PetscCall(MatTransposeGetMat(A01, &A01));
122413044ca3SPierre Jolivet     PetscCall(MatTranspose(A01, MAT_INITIAL_MATRIX, &B));
122513044ca3SPierre Jolivet     A01 = B;
122613044ca3SPierre Jolivet   } else {
122713044ca3SPierre Jolivet     PetscCall(PetscObjectTypeCompare((PetscObject)A01, MATHERMITIANTRANSPOSEVIRTUAL, flg));
122813044ca3SPierre Jolivet     if (flg[0]) {
1229811e8887SPierre Jolivet       PetscCall(MatShellGetScalingShifts(A01, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Mat *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED));
123013044ca3SPierre Jolivet       PetscCall(MatHermitianTransposeGetMat(A01, &A01));
123113044ca3SPierre Jolivet       PetscCall(MatHermitianTranspose(A01, MAT_INITIAL_MATRIX, &B));
123213044ca3SPierre Jolivet       A01 = B;
123313044ca3SPierre Jolivet     }
123413044ca3SPierre Jolivet   }
123513044ca3SPierre Jolivet   PetscCall(PetscLayoutCompare(T->rmap, A01->rmap, flg));
123613044ca3SPierre Jolivet   if (flg[0]) {
123713044ca3SPierre Jolivet     PetscCall(PetscLayoutCompare(T->cmap, A01->cmap, flg));
123813044ca3SPierre Jolivet     if (flg[0]) {
123913044ca3SPierre Jolivet       PetscCall(MatFindZeroRows(A01, &z)); /* for essential boundary conditions, some implementations will */
124013044ca3SPierre Jolivet       if (z) {                             /*  zero rows in [P00 A01] except for the diagonal of P00       */
124113044ca3SPierre Jolivet         PetscCall(MatSetOption(T, MAT_NO_OFF_PROC_ZERO_ROWS, PETSC_TRUE));
124213044ca3SPierre Jolivet         PetscCall(MatZeroRowsIS(T, z, 0.0, nullptr, nullptr)); /* corresponding zero rows from A01 */
124313044ca3SPierre Jolivet         PetscCall(ISDestroy(&z));
124413044ca3SPierre Jolivet       }
124513044ca3SPierre Jolivet       PetscCall(MatMultEqual(A01, T, 20, flg));
124613044ca3SPierre Jolivet       PetscCheck(flg[0], PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "A01 != A10^T");
124713044ca3SPierre Jolivet     } else PetscCall(PetscInfo(pc, "A01 and A10^T have non-congruent column layouts, cannot test for equality\n"));
124813044ca3SPierre Jolivet   }
124913044ca3SPierre Jolivet   PetscCall(MatDestroy(&B));
125013044ca3SPierre Jolivet   PetscCall(MatDestroy(&T));
125113044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
125213044ca3SPierre Jolivet }
125313044ca3SPierre Jolivet 
1254d16c0b94SPierre Jolivet static PetscErrorCode PCHPDDMCheckInclusion_Private(PC pc, IS is, IS is_local, PetscBool check)
1255d16c0b94SPierre Jolivet {
1256d16c0b94SPierre Jolivet   IS          intersect;
1257d16c0b94SPierre Jolivet   const char *str = "IS of the auxiliary Mat does not include all local rows of A";
1258d16c0b94SPierre Jolivet   PetscBool   equal;
1259d16c0b94SPierre Jolivet 
1260d16c0b94SPierre Jolivet   PetscFunctionBegin;
1261d16c0b94SPierre Jolivet   PetscCall(ISIntersect(is, is_local, &intersect));
1262d16c0b94SPierre Jolivet   PetscCall(ISEqualUnsorted(is_local, intersect, &equal));
1263d16c0b94SPierre Jolivet   PetscCall(ISDestroy(&intersect));
1264d16c0b94SPierre Jolivet   if (check) PetscCheck(equal, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "%s", str);
1265d16c0b94SPierre Jolivet   else if (!equal) PetscCall(PetscInfo(pc, "%s\n", str));
1266d16c0b94SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
1267d16c0b94SPierre Jolivet }
1268d16c0b94SPierre Jolivet 
1269d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMDestroySubMatrices_Private(PetscBool flg, PetscBool algebraic, Mat *sub)
1270d71ae5a4SJacob Faibussowitsch {
1271f1580f4eSBarry Smith   IS is;
1272f1580f4eSBarry Smith 
1273f1580f4eSBarry Smith   PetscFunctionBegin;
1274f1580f4eSBarry Smith   if (!flg) {
1275f1580f4eSBarry Smith     if (algebraic) {
1276f1580f4eSBarry Smith       PetscCall(PetscObjectQuery((PetscObject)sub[0], "_PCHPDDM_Embed", (PetscObject *)&is));
1277f1580f4eSBarry Smith       PetscCall(ISDestroy(&is));
1278db4a47b3SPierre Jolivet       PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Embed", nullptr));
1279db4a47b3SPierre Jolivet       PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Compact", nullptr));
1280f1580f4eSBarry Smith     }
1281f1580f4eSBarry Smith     PetscCall(MatDestroySubMatrices(algebraic ? 2 : 1, &sub));
1282f1580f4eSBarry Smith   }
12833ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1284f1580f4eSBarry Smith }
1285f1580f4eSBarry Smith 
1286d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMAlgebraicAuxiliaryMat_Private(Mat P, IS *is, Mat *sub[], PetscBool block)
1287d71ae5a4SJacob Faibussowitsch {
1288f1580f4eSBarry Smith   IS         icol[3], irow[2];
1289f1580f4eSBarry Smith   Mat       *M, Q;
1290f1580f4eSBarry Smith   PetscReal *ptr;
1291811e8887SPierre Jolivet   PetscInt  *idx, p = 0, bs = PetscAbs(P->cmap->bs);
1292f1580f4eSBarry Smith   PetscBool  flg;
1293f1580f4eSBarry Smith 
1294f1580f4eSBarry Smith   PetscFunctionBegin;
1295f1580f4eSBarry Smith   PetscCall(ISCreateStride(PETSC_COMM_SELF, P->cmap->N, 0, 1, icol + 2));
1296f1580f4eSBarry Smith   PetscCall(ISSetBlockSize(icol[2], bs));
1297f1580f4eSBarry Smith   PetscCall(ISSetIdentity(icol[2]));
1298f1580f4eSBarry Smith   PetscCall(PetscObjectTypeCompare((PetscObject)P, MATMPISBAIJ, &flg));
1299f1580f4eSBarry Smith   if (flg) {
1300f1580f4eSBarry Smith     /* MatCreateSubMatrices() does not handle MATMPISBAIJ properly when iscol != isrow, so convert first to MATMPIBAIJ */
1301f1580f4eSBarry Smith     PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &Q));
1302f1580f4eSBarry Smith     std::swap(P, Q);
1303f1580f4eSBarry Smith   }
1304f1580f4eSBarry Smith   PetscCall(MatCreateSubMatrices(P, 1, is, icol + 2, MAT_INITIAL_MATRIX, &M));
1305f1580f4eSBarry Smith   if (flg) {
1306f1580f4eSBarry Smith     std::swap(P, Q);
1307f1580f4eSBarry Smith     PetscCall(MatDestroy(&Q));
1308f1580f4eSBarry Smith   }
1309f1580f4eSBarry Smith   PetscCall(ISDestroy(icol + 2));
1310f1580f4eSBarry Smith   PetscCall(ISCreateStride(PETSC_COMM_SELF, M[0]->rmap->N, 0, 1, irow));
1311f1580f4eSBarry Smith   PetscCall(ISSetBlockSize(irow[0], bs));
1312f1580f4eSBarry Smith   PetscCall(ISSetIdentity(irow[0]));
1313f1580f4eSBarry Smith   if (!block) {
1314b07dfdedSPierre Jolivet     PetscCall(PetscMalloc2(P->cmap->N, &ptr, P->cmap->N / bs, &idx));
1315f1580f4eSBarry Smith     PetscCall(MatGetColumnNorms(M[0], NORM_INFINITY, ptr));
1316f1580f4eSBarry Smith     /* check for nonzero columns so that M[0] may be expressed in compact form */
1317811e8887SPierre Jolivet     for (PetscInt n = 0; n < P->cmap->N; n += bs) {
1318b07dfdedSPierre Jolivet       if (std::find_if(ptr + n, ptr + n + bs, [](PetscReal v) { return v > PETSC_MACHINE_EPSILON; }) != ptr + n + bs) idx[p++] = n / bs;
1319f1580f4eSBarry Smith     }
1320b07dfdedSPierre Jolivet     PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, p, idx, PETSC_USE_POINTER, icol + 1));
1321f1580f4eSBarry Smith     PetscCall(ISSetInfo(icol[1], IS_SORTED, IS_GLOBAL, PETSC_TRUE, PETSC_TRUE));
1322f1580f4eSBarry Smith     PetscCall(ISEmbed(*is, icol[1], PETSC_FALSE, icol + 2));
1323f1580f4eSBarry Smith     irow[1] = irow[0];
1324f1580f4eSBarry Smith     /* first Mat will be used in PCASM (if it is used as a PC on this level) and as the left-hand side of GenEO */
1325f1580f4eSBarry Smith     icol[0] = is[0];
1326f1580f4eSBarry Smith     PetscCall(MatCreateSubMatrices(M[0], 2, irow, icol, MAT_INITIAL_MATRIX, sub));
1327f1580f4eSBarry Smith     PetscCall(ISDestroy(icol + 1));
1328f1580f4eSBarry Smith     PetscCall(PetscFree2(ptr, idx));
1329f1580f4eSBarry Smith     /* IS used to go back and forth between the augmented and the original local linear system, see eq. (3.4) of [2022b] */
1330f1580f4eSBarry Smith     PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Embed", (PetscObject)icol[2]));
1331f1580f4eSBarry Smith     /* Mat used in eq. (3.1) of [2022b] */
1332f1580f4eSBarry Smith     PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Compact", (PetscObject)(*sub)[1]));
1333f1580f4eSBarry Smith   } else {
1334f1580f4eSBarry Smith     Mat aux;
1335811e8887SPierre Jolivet 
1336f1580f4eSBarry Smith     PetscCall(MatSetOption(M[0], MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
1337f1580f4eSBarry Smith     /* diagonal block of the overlapping rows */
1338f1580f4eSBarry Smith     PetscCall(MatCreateSubMatrices(M[0], 1, irow, is, MAT_INITIAL_MATRIX, sub));
1339f1580f4eSBarry Smith     PetscCall(MatDuplicate((*sub)[0], MAT_COPY_VALUES, &aux));
1340f1580f4eSBarry Smith     PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE));
1341f1580f4eSBarry Smith     if (bs == 1) { /* scalar case */
1342f1580f4eSBarry Smith       Vec sum[2];
1343811e8887SPierre Jolivet 
1344f1580f4eSBarry Smith       PetscCall(MatCreateVecs(aux, sum, sum + 1));
1345f1580f4eSBarry Smith       PetscCall(MatGetRowSum(M[0], sum[0]));
1346f1580f4eSBarry Smith       PetscCall(MatGetRowSum(aux, sum[1]));
1347f1580f4eSBarry Smith       /* off-diagonal block row sum (full rows - diagonal block rows) */
1348f1580f4eSBarry Smith       PetscCall(VecAXPY(sum[0], -1.0, sum[1]));
1349f1580f4eSBarry Smith       /* subdomain matrix plus off-diagonal block row sum */
1350f1580f4eSBarry Smith       PetscCall(MatDiagonalSet(aux, sum[0], ADD_VALUES));
1351f1580f4eSBarry Smith       PetscCall(VecDestroy(sum));
1352f1580f4eSBarry Smith       PetscCall(VecDestroy(sum + 1));
1353f1580f4eSBarry Smith     } else { /* vectorial case */
1354f1580f4eSBarry Smith       /* TODO: missing MatGetValuesBlocked(), so the code below is     */
1355f1580f4eSBarry Smith       /* an extension of the scalar case for when bs > 1, but it could */
1356f1580f4eSBarry Smith       /* be more efficient by avoiding all these MatMatMult()          */
1357f1580f4eSBarry Smith       Mat          sum[2], ones;
1358f1580f4eSBarry Smith       PetscScalar *ptr;
1359811e8887SPierre Jolivet 
1360f1580f4eSBarry Smith       PetscCall(PetscCalloc1(M[0]->cmap->n * bs, &ptr));
1361f1580f4eSBarry Smith       PetscCall(MatCreateDense(PETSC_COMM_SELF, M[0]->cmap->n, bs, M[0]->cmap->n, bs, ptr, &ones));
1362811e8887SPierre Jolivet       for (PetscInt n = 0; n < M[0]->cmap->n; n += bs) {
1363f1580f4eSBarry Smith         for (p = 0; p < bs; ++p) ptr[n + p * (M[0]->cmap->n + 1)] = 1.0;
1364f1580f4eSBarry Smith       }
1365fb842aefSJose E. Roman       PetscCall(MatMatMult(M[0], ones, MAT_INITIAL_MATRIX, PETSC_CURRENT, sum));
1366f1580f4eSBarry Smith       PetscCall(MatDestroy(&ones));
1367f1580f4eSBarry Smith       PetscCall(MatCreateDense(PETSC_COMM_SELF, aux->cmap->n, bs, aux->cmap->n, bs, ptr, &ones));
1368f1580f4eSBarry Smith       PetscCall(MatDenseSetLDA(ones, M[0]->cmap->n));
1369fb842aefSJose E. Roman       PetscCall(MatMatMult(aux, ones, MAT_INITIAL_MATRIX, PETSC_CURRENT, sum + 1));
1370f1580f4eSBarry Smith       PetscCall(MatDestroy(&ones));
1371f1580f4eSBarry Smith       PetscCall(PetscFree(ptr));
1372f1580f4eSBarry Smith       /* off-diagonal block row sum (full rows - diagonal block rows) */
1373f1580f4eSBarry Smith       PetscCall(MatAXPY(sum[0], -1.0, sum[1], SAME_NONZERO_PATTERN));
1374f1580f4eSBarry Smith       PetscCall(MatDestroy(sum + 1));
1375f1580f4eSBarry Smith       /* re-order values to be consistent with MatSetValuesBlocked()           */
1376f1580f4eSBarry Smith       /* equivalent to MatTranspose() which does not truly handle              */
1377f1580f4eSBarry Smith       /* MAT_INPLACE_MATRIX in the rectangular case, as it calls PetscMalloc() */
1378f1580f4eSBarry Smith       PetscCall(MatDenseGetArrayWrite(sum[0], &ptr));
1379f1580f4eSBarry Smith       HPDDM::Wrapper<PetscScalar>::imatcopy<'T'>(bs, sum[0]->rmap->n, ptr, sum[0]->rmap->n, bs);
1380f1580f4eSBarry Smith       /* subdomain matrix plus off-diagonal block row sum */
1381811e8887SPierre Jolivet       for (PetscInt n = 0; n < aux->cmap->n / bs; ++n) PetscCall(MatSetValuesBlocked(aux, 1, &n, 1, &n, ptr + n * bs * bs, ADD_VALUES));
1382f1580f4eSBarry Smith       PetscCall(MatAssemblyBegin(aux, MAT_FINAL_ASSEMBLY));
1383f1580f4eSBarry Smith       PetscCall(MatAssemblyEnd(aux, MAT_FINAL_ASSEMBLY));
1384f1580f4eSBarry Smith       PetscCall(MatDenseRestoreArrayWrite(sum[0], &ptr));
1385f1580f4eSBarry Smith       PetscCall(MatDestroy(sum));
1386f1580f4eSBarry Smith     }
1387f1580f4eSBarry Smith     PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_TRUE));
1388f1580f4eSBarry Smith     /* left-hand side of GenEO, with the same sparsity pattern as PCASM subdomain solvers  */
1389f1580f4eSBarry Smith     PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Neumann_Mat", (PetscObject)aux));
1390f1580f4eSBarry Smith   }
1391f1580f4eSBarry Smith   PetscCall(ISDestroy(irow));
1392f1580f4eSBarry Smith   PetscCall(MatDestroySubMatrices(1, &M));
13933ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1394f1580f4eSBarry Smith }
1395f1580f4eSBarry Smith 
139613044ca3SPierre Jolivet static PetscErrorCode PCApply_Nest(PC pc, Vec x, Vec y)
139713044ca3SPierre Jolivet {
139813044ca3SPierre Jolivet   Mat                    A;
139913044ca3SPierre Jolivet   MatSolverType          type;
140013044ca3SPierre Jolivet   IS                     is[2];
140113044ca3SPierre Jolivet   PetscBool              flg;
140213044ca3SPierre Jolivet   std::pair<PC, Vec[2]> *p;
140313044ca3SPierre Jolivet 
140413044ca3SPierre Jolivet   PetscFunctionBegin;
140513044ca3SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
140613044ca3SPierre Jolivet   PetscCall(PCGetOperators(p->first, &A, nullptr));
140713044ca3SPierre Jolivet   PetscCall(MatNestGetISs(A, is, nullptr));
140813044ca3SPierre Jolivet   PetscCall(PCFactorGetMatSolverType(p->first, &type));
140913044ca3SPierre Jolivet   PetscCall(PCFactorGetMatrix(p->first, &A));
141013044ca3SPierre Jolivet   PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg));
141124ddd604SPierre Jolivet   if (flg && A->schur) {
141213044ca3SPierre Jolivet #if PetscDefined(HAVE_MUMPS)
141313044ca3SPierre Jolivet     PetscCall(MatMumpsSetIcntl(A, 26, 1)); /* reduction/condensation phase followed by Schur complement solve */
141413044ca3SPierre Jolivet #endif
141513044ca3SPierre Jolivet   }
141613044ca3SPierre Jolivet   PetscCall(VecISCopy(p->second[0], is[1], SCATTER_FORWARD, x)); /* assign the RHS associated to the Schur complement */
141713044ca3SPierre Jolivet   PetscCall(PCApply(p->first, p->second[0], p->second[1]));
141813044ca3SPierre Jolivet   PetscCall(VecISCopy(p->second[1], is[1], SCATTER_REVERSE, y)); /* retrieve the partial solution associated to the Schur complement */
141913044ca3SPierre Jolivet   if (flg) {
142013044ca3SPierre Jolivet #if PetscDefined(HAVE_MUMPS)
142113044ca3SPierre Jolivet     PetscCall(MatMumpsSetIcntl(A, 26, -1)); /* default ICNTL(26) value in PETSc */
142213044ca3SPierre Jolivet #endif
142313044ca3SPierre Jolivet   }
142413044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
142513044ca3SPierre Jolivet }
142613044ca3SPierre Jolivet 
142713044ca3SPierre Jolivet static PetscErrorCode PCView_Nest(PC pc, PetscViewer viewer)
142813044ca3SPierre Jolivet {
142913044ca3SPierre Jolivet   std::pair<PC, Vec[2]> *p;
143013044ca3SPierre Jolivet 
143113044ca3SPierre Jolivet   PetscFunctionBegin;
143213044ca3SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
143313044ca3SPierre Jolivet   PetscCall(PCView(p->first, viewer));
143413044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
143513044ca3SPierre Jolivet }
143613044ca3SPierre Jolivet 
143713044ca3SPierre Jolivet static PetscErrorCode PCDestroy_Nest(PC pc)
143813044ca3SPierre Jolivet {
143913044ca3SPierre Jolivet   std::pair<PC, Vec[2]> *p;
144013044ca3SPierre Jolivet 
144113044ca3SPierre Jolivet   PetscFunctionBegin;
144213044ca3SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
144313044ca3SPierre Jolivet   PetscCall(VecDestroy(p->second));
144413044ca3SPierre Jolivet   PetscCall(VecDestroy(p->second + 1));
144513044ca3SPierre Jolivet   PetscCall(PCDestroy(&p->first));
144613044ca3SPierre Jolivet   PetscCall(PetscFree(p));
144713044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
144813044ca3SPierre Jolivet }
144913044ca3SPierre Jolivet 
145013044ca3SPierre Jolivet template <bool T = false>
145113044ca3SPierre Jolivet static PetscErrorCode MatMult_Schur(Mat A, Vec x, Vec y)
145213044ca3SPierre Jolivet {
145301e3c840SPierre Jolivet   std::tuple<Mat, PetscSF, Vec[2]> *ctx;
145413044ca3SPierre Jolivet 
145513044ca3SPierre Jolivet   PetscFunctionBegin;
145613044ca3SPierre Jolivet   PetscCall(MatShellGetContext(A, &ctx));
145713044ca3SPierre Jolivet   PetscCall(VecScatterBegin(std::get<1>(*ctx), x, std::get<2>(*ctx)[0], INSERT_VALUES, SCATTER_FORWARD)); /* local Vec with overlap */
145813044ca3SPierre Jolivet   PetscCall(VecScatterEnd(std::get<1>(*ctx), x, std::get<2>(*ctx)[0], INSERT_VALUES, SCATTER_FORWARD));
145913044ca3SPierre Jolivet   if (!T) PetscCall(MatMult(std::get<0>(*ctx), std::get<2>(*ctx)[0], std::get<2>(*ctx)[1])); /* local Schur complement */
146013044ca3SPierre Jolivet   else PetscCall(MatMultTranspose(std::get<0>(*ctx), std::get<2>(*ctx)[0], std::get<2>(*ctx)[1]));
146113044ca3SPierre Jolivet   PetscCall(VecSet(y, 0.0));
146213044ca3SPierre Jolivet   PetscCall(VecScatterBegin(std::get<1>(*ctx), std::get<2>(*ctx)[1], y, ADD_VALUES, SCATTER_REVERSE)); /* global Vec with summed up contributions on the overlap */
146313044ca3SPierre Jolivet   PetscCall(VecScatterEnd(std::get<1>(*ctx), std::get<2>(*ctx)[1], y, ADD_VALUES, SCATTER_REVERSE));
146413044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
146513044ca3SPierre Jolivet }
146613044ca3SPierre Jolivet 
146713044ca3SPierre Jolivet static PetscErrorCode MatDestroy_Schur(Mat A)
146813044ca3SPierre Jolivet {
146901e3c840SPierre Jolivet   std::tuple<Mat, PetscSF, Vec[2]> *ctx;
147013044ca3SPierre Jolivet 
147113044ca3SPierre Jolivet   PetscFunctionBegin;
147213044ca3SPierre Jolivet   PetscCall(MatShellGetContext(A, &ctx));
147313044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<2>(*ctx)));
147413044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<2>(*ctx) + 1));
147513044ca3SPierre Jolivet   PetscCall(PetscFree(ctx));
147613044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
147713044ca3SPierre Jolivet }
147813044ca3SPierre Jolivet 
147913044ca3SPierre Jolivet static PetscErrorCode MatMult_SchurCorrection(Mat A, Vec x, Vec y)
148013044ca3SPierre Jolivet {
148113044ca3SPierre Jolivet   PC                                         pc;
148213044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx;
148313044ca3SPierre Jolivet 
148413044ca3SPierre Jolivet   PetscFunctionBegin;
148513044ca3SPierre Jolivet   PetscCall(MatShellGetContext(A, &ctx));
148613044ca3SPierre Jolivet   pc = ((PC_HPDDM *)std::get<0>(*ctx)[0]->data)->levels[0]->ksp->pc;
148713044ca3SPierre Jolivet   if (std::get<2>(*ctx) == PC_LEFT || std::get<2>(*ctx) == PC_SIDE_DEFAULT) {             /* Q_0 is the coarse correction associated to the A00 block from PCFIELDSPLIT */
148813044ca3SPierre Jolivet     PetscCall(MatMult(std::get<1>(*ctx)[0], x, std::get<3>(*ctx)[1]));                    /*     A_01 x                 */
148913044ca3SPierre Jolivet     PetscCall(PCHPDDMDeflate_Private(pc, std::get<3>(*ctx)[1], std::get<3>(*ctx)[1]));    /*     Q_0 A_01 x             */
149013044ca3SPierre Jolivet     PetscCall(MatMult(std::get<1>(*ctx)[1], std::get<3>(*ctx)[1], std::get<3>(*ctx)[0])); /*     A_10 Q_0 A_01 x        */
149113044ca3SPierre Jolivet     PetscCall(PCApply(std::get<0>(*ctx)[1], std::get<3>(*ctx)[0], y));                    /* y = M_S^-1 A_10 Q_0 A_01 x */
149213044ca3SPierre Jolivet   } else {
149313044ca3SPierre Jolivet     PetscCall(PCApply(std::get<0>(*ctx)[1], x, std::get<3>(*ctx)[0]));                    /*     M_S^-1 x               */
149413044ca3SPierre Jolivet     PetscCall(MatMult(std::get<1>(*ctx)[0], std::get<3>(*ctx)[0], std::get<3>(*ctx)[1])); /*     A_01 M_S^-1 x          */
149513044ca3SPierre Jolivet     PetscCall(PCHPDDMDeflate_Private(pc, std::get<3>(*ctx)[1], std::get<3>(*ctx)[1]));    /*     Q_0 A_01 M_S^-1 x      */
149613044ca3SPierre Jolivet     PetscCall(MatMult(std::get<1>(*ctx)[1], std::get<3>(*ctx)[1], y));                    /* y = A_10 Q_0 A_01 M_S^-1 x */
149713044ca3SPierre Jolivet   }
149813044ca3SPierre Jolivet   PetscCall(VecAXPY(y, -1.0, x)); /* y -= x, preconditioned eq. (24) of https://hal.science/hal-02343808v6/document (with a sign flip) */
149913044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
150013044ca3SPierre Jolivet }
150113044ca3SPierre Jolivet 
150213044ca3SPierre Jolivet static PetscErrorCode MatView_SchurCorrection(Mat A, PetscViewer viewer)
150313044ca3SPierre Jolivet {
150413044ca3SPierre Jolivet   PetscBool                                  ascii;
150513044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx;
150613044ca3SPierre Jolivet 
150713044ca3SPierre Jolivet   PetscFunctionBegin;
150813044ca3SPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &ascii));
150913044ca3SPierre Jolivet   if (ascii) {
151013044ca3SPierre Jolivet     PetscCall(MatShellGetContext(A, &ctx));
151113044ca3SPierre Jolivet     PetscCall(PetscViewerASCIIPrintf(viewer, "action of %s\n", std::get<2>(*ctx) == PC_LEFT || std::get<2>(*ctx) == PC_SIDE_DEFAULT ? "(I - M_S^-1 A_10 Q_0 A_01)" : "(I - A_10 Q_0 A_01 M_S^-1)"));
151213044ca3SPierre Jolivet     PetscCall(PCView(std::get<0>(*ctx)[1], viewer)); /* no need to PCView(Q_0) since it will be done by PCFIELDSPLIT */
151313044ca3SPierre Jolivet   }
151413044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
151513044ca3SPierre Jolivet }
151613044ca3SPierre Jolivet 
151713044ca3SPierre Jolivet static PetscErrorCode MatDestroy_SchurCorrection(Mat A)
151813044ca3SPierre Jolivet {
151913044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx;
152013044ca3SPierre Jolivet 
152113044ca3SPierre Jolivet   PetscFunctionBegin;
152213044ca3SPierre Jolivet   PetscCall(MatShellGetContext(A, &ctx));
152313044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<3>(*ctx)));
152413044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<3>(*ctx) + 1));
152513044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<3>(*ctx) + 2));
152613044ca3SPierre Jolivet   PetscCall(PCDestroy(std::get<0>(*ctx) + 1));
152713044ca3SPierre Jolivet   PetscCall(PetscFree(ctx));
152813044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
152913044ca3SPierre Jolivet }
153013044ca3SPierre Jolivet 
153113044ca3SPierre Jolivet static PetscErrorCode KSPPreSolve_SchurCorrection(KSP, Vec b, Vec, void *context)
153213044ca3SPierre Jolivet {
153313044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx = reinterpret_cast<std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *>(context);
153413044ca3SPierre Jolivet 
1535a6b3e571SPierre Jolivet   PetscFunctionBegin;
153613044ca3SPierre Jolivet   if (std::get<2>(*ctx) == PC_LEFT || std::get<2>(*ctx) == PC_SIDE_DEFAULT) {
153713044ca3SPierre Jolivet     PetscCall(PCApply(std::get<0>(*ctx)[1], b, std::get<3>(*ctx)[2]));
153813044ca3SPierre Jolivet     std::swap(*b, *std::get<3>(*ctx)[2]); /* replace b by M^-1 b, but need to keep a copy of the original RHS, so swap it with the work Vec */
153913044ca3SPierre Jolivet   }
154013044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
154113044ca3SPierre Jolivet }
154213044ca3SPierre Jolivet 
154313044ca3SPierre Jolivet static PetscErrorCode KSPPostSolve_SchurCorrection(KSP, Vec b, Vec x, void *context)
154413044ca3SPierre Jolivet {
154513044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx = reinterpret_cast<std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *>(context);
154613044ca3SPierre Jolivet 
1547a6b3e571SPierre Jolivet   PetscFunctionBegin;
154813044ca3SPierre Jolivet   if (std::get<2>(*ctx) == PC_LEFT || std::get<2>(*ctx) == PC_SIDE_DEFAULT) std::swap(*b, *std::get<3>(*ctx)[2]); /* put back the original RHS where it belongs */
154913044ca3SPierre Jolivet   else {
155013044ca3SPierre Jolivet     PetscCall(PCApply(std::get<0>(*ctx)[1], x, std::get<3>(*ctx)[2]));
155113044ca3SPierre Jolivet     PetscCall(VecCopy(std::get<3>(*ctx)[2], x)); /* replace x by M^-1 x */
155213044ca3SPierre Jolivet   }
155313044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
155413044ca3SPierre Jolivet }
155513044ca3SPierre Jolivet 
15569bb5c669SPierre Jolivet static PetscErrorCode MatMult_Harmonic(Mat, Vec, Vec);
15579bb5c669SPierre Jolivet static PetscErrorCode MatMultTranspose_Harmonic(Mat, Vec, Vec);
15589bb5c669SPierre Jolivet static PetscErrorCode MatProduct_AB_Harmonic(Mat, Mat, Mat, void *);
15599bb5c669SPierre Jolivet static PetscErrorCode MatDestroy_Harmonic(Mat);
15609bb5c669SPierre Jolivet 
1561d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetUp_HPDDM(PC pc)
1562d71ae5a4SJacob Faibussowitsch {
1563f1580f4eSBarry Smith   PC_HPDDM                                  *data = (PC_HPDDM *)pc->data;
1564f1580f4eSBarry Smith   PC                                         inner;
1565f1580f4eSBarry Smith   KSP                                       *ksp;
156613044ca3SPierre Jolivet   Mat                                       *sub, A, P, N, C = nullptr, uaux = nullptr, weighted, subA[2], S;
1567f1580f4eSBarry Smith   Vec                                        xin, v;
1568f1580f4eSBarry Smith   std::vector<Vec>                           initial;
1569db4a47b3SPierre Jolivet   IS                                         is[1], loc, uis = data->is, unsorted = nullptr;
1570f1580f4eSBarry Smith   ISLocalToGlobalMapping                     l2g;
1571f1580f4eSBarry Smith   char                                       prefix[256];
1572f1580f4eSBarry Smith   const char                                *pcpre;
1573f1580f4eSBarry Smith   const PetscScalar *const                  *ev;
15749bb5c669SPierre Jolivet   PetscInt                                   n, requested = data->N, reused = 0, overlap = -1;
1575f1580f4eSBarry Smith   MatStructure                               structure  = UNKNOWN_NONZERO_PATTERN;
1576f1580f4eSBarry Smith   PetscBool                                  subdomains = PETSC_FALSE, flg = PETSC_FALSE, ismatis, swap = PETSC_FALSE, algebraic = PETSC_FALSE, block = PETSC_FALSE;
1577f1580f4eSBarry Smith   DM                                         dm;
157813044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx = nullptr;
1579398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG)
1580db4a47b3SPierre Jolivet   IS  dis  = nullptr;
1581db4a47b3SPierre Jolivet   Mat daux = nullptr;
1582398c7888SPierre Jolivet #endif
1583f1580f4eSBarry Smith 
1584f1580f4eSBarry Smith   PetscFunctionBegin;
1585f1580f4eSBarry Smith   PetscCheck(data->levels && data->levels[0], PETSC_COMM_SELF, PETSC_ERR_PLIB, "Not a single level allocated");
1586f1580f4eSBarry Smith   PetscCall(PCGetOptionsPrefix(pc, &pcpre));
1587f1580f4eSBarry Smith   PetscCall(PCGetOperators(pc, &A, &P));
1588f1580f4eSBarry Smith   if (!data->levels[0]->ksp) {
1589f1580f4eSBarry Smith     PetscCall(KSPCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->ksp));
15903821be0aSBarry Smith     PetscCall(KSPSetNestLevel(data->levels[0]->ksp, pc->kspnestlevel));
1591f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_%s_", pcpre ? pcpre : "", data->N > 1 ? "levels_1" : "coarse"));
1592f1580f4eSBarry Smith     PetscCall(KSPSetOptionsPrefix(data->levels[0]->ksp, prefix));
1593f1580f4eSBarry Smith     PetscCall(KSPSetType(data->levels[0]->ksp, KSPPREONLY));
1594f1580f4eSBarry Smith   } else if (data->levels[0]->ksp->pc && data->levels[0]->ksp->pc->setupcalled == 1 && data->levels[0]->ksp->pc->reusepreconditioner) {
1595f1580f4eSBarry Smith     /* if the fine-level PCSHELL exists, its setup has succeeded, and one wants to reuse it, */
1596f1580f4eSBarry Smith     /* then just propagate the appropriate flag to the coarser levels                        */
1597f1580f4eSBarry Smith     for (n = 0; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) {
1598f1580f4eSBarry Smith       /* the following KSP and PC may be NULL for some processes, hence the check            */
1599f1580f4eSBarry Smith       if (data->levels[n]->ksp) PetscCall(KSPSetReusePreconditioner(data->levels[n]->ksp, PETSC_TRUE));
1600f1580f4eSBarry Smith       if (data->levels[n]->pc) PetscCall(PCSetReusePreconditioner(data->levels[n]->pc, PETSC_TRUE));
1601f1580f4eSBarry Smith     }
1602f1580f4eSBarry Smith     /* early bail out because there is nothing to do */
16033ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
1604f1580f4eSBarry Smith   } else {
1605f1580f4eSBarry Smith     /* reset coarser levels */
1606f1580f4eSBarry Smith     for (n = 1; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) {
1607f1580f4eSBarry Smith       if (data->levels[n]->ksp && data->levels[n]->ksp->pc && data->levels[n]->ksp->pc->setupcalled == 1 && data->levels[n]->ksp->pc->reusepreconditioner && n < data->N) {
1608f1580f4eSBarry Smith         reused = data->N - n;
1609f1580f4eSBarry Smith         break;
1610f1580f4eSBarry Smith       }
1611f1580f4eSBarry Smith       PetscCall(KSPDestroy(&data->levels[n]->ksp));
1612f1580f4eSBarry Smith       PetscCall(PCDestroy(&data->levels[n]->pc));
1613f1580f4eSBarry Smith     }
1614f1580f4eSBarry Smith     /* check if some coarser levels are being reused */
1615462c564dSBarry Smith     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &reused, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)pc)));
1616f1580f4eSBarry Smith     const int *addr = data->levels[0]->P ? data->levels[0]->P->getAddrLocal() : &HPDDM::i__0;
1617f1580f4eSBarry Smith 
1618f1580f4eSBarry Smith     if (addr != &HPDDM::i__0 && reused != data->N - 1) {
1619f1580f4eSBarry Smith       /* reuse previously computed eigenvectors */
1620f1580f4eSBarry Smith       ev = data->levels[0]->P->getVectors();
1621f1580f4eSBarry Smith       if (ev) {
1622f1580f4eSBarry Smith         initial.reserve(*addr);
1623f1580f4eSBarry Smith         PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, data->levels[0]->P->getDof(), ev[0], &xin));
1624f1580f4eSBarry Smith         for (n = 0; n < *addr; ++n) {
1625f1580f4eSBarry Smith           PetscCall(VecDuplicate(xin, &v));
1626f1580f4eSBarry Smith           PetscCall(VecPlaceArray(xin, ev[n]));
1627f1580f4eSBarry Smith           PetscCall(VecCopy(xin, v));
1628f1580f4eSBarry Smith           initial.emplace_back(v);
1629f1580f4eSBarry Smith           PetscCall(VecResetArray(xin));
1630f1580f4eSBarry Smith         }
1631f1580f4eSBarry Smith         PetscCall(VecDestroy(&xin));
1632f1580f4eSBarry Smith       }
1633f1580f4eSBarry Smith     }
1634f1580f4eSBarry Smith   }
1635f1580f4eSBarry Smith   data->N -= reused;
1636f1580f4eSBarry Smith   PetscCall(KSPSetOperators(data->levels[0]->ksp, A, P));
1637f1580f4eSBarry Smith 
1638f1580f4eSBarry Smith   PetscCall(PetscObjectTypeCompare((PetscObject)P, MATIS, &ismatis));
1639f1580f4eSBarry Smith   if (!data->is && !ismatis) {
1640db4a47b3SPierre Jolivet     PetscErrorCode (*create)(DM, IS *, Mat *, PetscErrorCode (**)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void **) = nullptr;
1641db4a47b3SPierre Jolivet     PetscErrorCode (*usetup)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *)                                                = nullptr;
1642db4a47b3SPierre Jolivet     void *uctx                                                                                                               = nullptr;
1643f1580f4eSBarry Smith 
1644f1580f4eSBarry Smith     /* first see if we can get the data from the DM */
1645f1580f4eSBarry Smith     PetscCall(MatGetDM(P, &dm));
1646f1580f4eSBarry Smith     if (!dm) PetscCall(MatGetDM(A, &dm));
1647f1580f4eSBarry Smith     if (!dm) PetscCall(PCGetDM(pc, &dm));
1648907a3e9cSStefano Zampini     if (dm) { /* this is the hook for DMPLEX for which the auxiliary Mat is the local Neumann matrix */
1649f1580f4eSBarry Smith       PetscCall(PetscObjectQueryFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", &create));
1650f1580f4eSBarry Smith       if (create) {
1651f1580f4eSBarry Smith         PetscCall((*create)(dm, &uis, &uaux, &usetup, &uctx));
1652c8ea6600SPierre Jolivet         if (data->Neumann == PETSC_BOOL3_UNKNOWN) data->Neumann = PETSC_BOOL3_TRUE; /* set the value only if it was not already provided by the user */
1653f1580f4eSBarry Smith       }
1654f1580f4eSBarry Smith     }
1655f1580f4eSBarry Smith     if (!create) {
1656f1580f4eSBarry Smith       if (!uis) {
1657f1580f4eSBarry Smith         PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Neumann_IS", (PetscObject *)&uis));
1658f1580f4eSBarry Smith         PetscCall(PetscObjectReference((PetscObject)uis));
1659f1580f4eSBarry Smith       }
1660f1580f4eSBarry Smith       if (!uaux) {
1661f1580f4eSBarry Smith         PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Neumann_Mat", (PetscObject *)&uaux));
1662f1580f4eSBarry Smith         PetscCall(PetscObjectReference((PetscObject)uaux));
1663f1580f4eSBarry Smith       }
1664f1580f4eSBarry Smith       /* look inside the Pmat instead of the PC, needed for MatSchurComplementComputeExplicitOperator() */
1665f1580f4eSBarry Smith       if (!uis) {
1666f1580f4eSBarry Smith         PetscCall(PetscObjectQuery((PetscObject)P, "_PCHPDDM_Neumann_IS", (PetscObject *)&uis));
1667f1580f4eSBarry Smith         PetscCall(PetscObjectReference((PetscObject)uis));
1668f1580f4eSBarry Smith       }
1669f1580f4eSBarry Smith       if (!uaux) {
1670f1580f4eSBarry Smith         PetscCall(PetscObjectQuery((PetscObject)P, "_PCHPDDM_Neumann_Mat", (PetscObject *)&uaux));
1671f1580f4eSBarry Smith         PetscCall(PetscObjectReference((PetscObject)uaux));
1672f1580f4eSBarry Smith       }
1673f1580f4eSBarry Smith     }
1674f1580f4eSBarry Smith     PetscCall(PCHPDDMSetAuxiliaryMat(pc, uis, uaux, usetup, uctx));
1675f1580f4eSBarry Smith     PetscCall(MatDestroy(&uaux));
1676f1580f4eSBarry Smith     PetscCall(ISDestroy(&uis));
1677f1580f4eSBarry Smith   }
1678f1580f4eSBarry Smith 
1679f1580f4eSBarry Smith   if (!ismatis) {
1680f1580f4eSBarry Smith     PetscCall(PCHPDDMSetUpNeumannOverlap_Private(pc));
1681db4a47b3SPierre Jolivet     PetscCall(PetscOptionsGetBool(nullptr, pcpre, "-pc_hpddm_block_splitting", &block, nullptr));
16829bb5c669SPierre Jolivet     PetscCall(PetscOptionsGetInt(nullptr, pcpre, "-pc_hpddm_harmonic_overlap", &overlap, nullptr));
16835e642048SPierre Jolivet     PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg));
16845e642048SPierre Jolivet     if (data->is || (data->N > 1 && flg)) {
16859bb5c669SPierre Jolivet       if (block || overlap != -1) {
168602800ff6SPierre Jolivet         PetscCall(ISDestroy(&data->is));
168702800ff6SPierre Jolivet         PetscCall(MatDestroy(&data->aux));
16885e642048SPierre Jolivet       } else if (data->N > 1 && flg) {
168913044ca3SPierre Jolivet         PCHPDDMSchurPreType type = PC_HPDDM_SCHUR_PRE_GENEO;
169013044ca3SPierre Jolivet 
169113044ca3SPierre Jolivet         PetscCall(PetscOptionsGetEnum(nullptr, pcpre, "-pc_hpddm_schur_precondition", PCHPDDMSchurPreTypes, (PetscEnum *)&type, &flg));
169213044ca3SPierre Jolivet         if (type == PC_HPDDM_SCHUR_PRE_LEAST_SQUARES) {
169313044ca3SPierre Jolivet           PetscCall(ISDestroy(&data->is)); /* destroy any previously user-set objects since they will be set automatically */
169413044ca3SPierre Jolivet           PetscCall(MatDestroy(&data->aux));
169513044ca3SPierre Jolivet         } else if (type == PC_HPDDM_SCHUR_PRE_GENEO) {
169613044ca3SPierre Jolivet           PetscContainer container = nullptr;
169713044ca3SPierre Jolivet 
169813044ca3SPierre Jolivet           PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Schur", (PetscObject *)&container));
169913044ca3SPierre Jolivet           if (!container) { /* first call to PCSetUp() on the PC associated to the Schur complement */
170013044ca3SPierre Jolivet             PC_HPDDM *data_00;
170113044ca3SPierre Jolivet             KSP       ksp, inner_ksp;
170213044ca3SPierre Jolivet             PC        pc_00;
1703cf67ef9dSPierre Jolivet             Mat       A11 = nullptr;
1704cf67ef9dSPierre Jolivet             Vec       d   = nullptr;
170513044ca3SPierre Jolivet             char     *prefix;
170613044ca3SPierre Jolivet 
170713044ca3SPierre Jolivet             PetscCall(MatSchurComplementGetKSP(P, &ksp));
170813044ca3SPierre Jolivet             PetscCall(KSPGetPC(ksp, &pc_00));
170913044ca3SPierre Jolivet             PetscCall(PetscObjectTypeCompare((PetscObject)pc_00, PCHPDDM, &flg));
1710fd310a01SPierre Jolivet             PetscCheck(flg, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-%spc_hpddm_schur_precondition %s and -%spc_type %s (!= %s)", pcpre ? pcpre : "", PCHPDDMSchurPreTypes[type], ((PetscObject)pc_00)->prefix ? ((PetscObject)pc_00)->prefix : "",
171113044ca3SPierre Jolivet                        ((PetscObject)pc_00)->type_name, PCHPDDM);
171213044ca3SPierre Jolivet             data_00 = (PC_HPDDM *)pc_00->data;
1713fd310a01SPierre Jolivet             PetscCheck(data_00->N == 2, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-%spc_hpddm_schur_precondition %s and %" PetscInt_FMT " level%s instead of 2 for the A00 block -%s", pcpre ? pcpre : "", PCHPDDMSchurPreTypes[type],
1714fd310a01SPierre Jolivet                        data_00->N, data_00->N > 1 ? "s" : "", ((PetscObject)pc_00)->prefix);
171513044ca3SPierre Jolivet             PetscCall(PetscObjectTypeCompare((PetscObject)data_00->levels[0]->pc, PCASM, &flg));
1716fd310a01SPierre Jolivet             PetscCheck(flg, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-%spc_hpddm_schur_precondition %s and -%spc_type %s (!= %s)", pcpre ? pcpre : "", PCHPDDMSchurPreTypes[type], ((PetscObject)data_00->levels[0]->pc)->prefix,
171713044ca3SPierre Jolivet                        ((PetscObject)data_00->levels[0]->pc)->type_name, PCASM);
1718d16c0b94SPierre Jolivet             PetscCall(MatSchurComplementGetSubMatrices(P, nullptr, nullptr, nullptr, nullptr, &A11));
17195e642048SPierre Jolivet             if (PetscDefined(USE_DEBUG) || !data->is) {
17205e642048SPierre Jolivet               Mat A01, A10, B = nullptr, C = nullptr, *sub;
172113044ca3SPierre Jolivet 
17225e642048SPierre Jolivet               PetscCall(MatSchurComplementGetSubMatrices(P, &A, nullptr, &A01, &A10, nullptr));
17235e642048SPierre Jolivet               PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, &flg));
17245e642048SPierre Jolivet               if (flg) {
17255e642048SPierre Jolivet                 PetscCall(MatTransposeGetMat(A10, &C));
17265e642048SPierre Jolivet                 PetscCall(MatTranspose(C, MAT_INITIAL_MATRIX, &B));
17275e642048SPierre Jolivet               } else {
17285e642048SPierre Jolivet                 PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, &flg));
17295e642048SPierre Jolivet                 if (flg) {
17305e642048SPierre Jolivet                   PetscCall(MatHermitianTransposeGetMat(A10, &C));
17315e642048SPierre Jolivet                   PetscCall(MatHermitianTranspose(C, MAT_INITIAL_MATRIX, &B));
17325e642048SPierre Jolivet                 }
17335e642048SPierre Jolivet               }
1734811e8887SPierre Jolivet               if (flg)
1735811e8887SPierre Jolivet                 PetscCall(MatShellGetScalingShifts(A10, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Mat *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED));
17365e642048SPierre Jolivet               if (!B) {
17375e642048SPierre Jolivet                 B = A10;
17385e642048SPierre Jolivet                 PetscCall(PetscObjectReference((PetscObject)B));
17395e642048SPierre Jolivet               } else if (!data->is) {
17405e642048SPierre Jolivet                 PetscCall(PetscObjectTypeCompareAny((PetscObject)A01, &flg, MATTRANSPOSEVIRTUAL, MATHERMITIANTRANSPOSEVIRTUAL, ""));
17415e642048SPierre Jolivet                 if (!flg) C = A01;
1742811e8887SPierre Jolivet                 else
1743811e8887SPierre Jolivet                   PetscCall(MatShellGetScalingShifts(A01, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Mat *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED));
17445e642048SPierre Jolivet               }
174513044ca3SPierre Jolivet               PetscCall(ISCreateStride(PETSC_COMM_SELF, B->rmap->N, 0, 1, &uis));
17465e642048SPierre Jolivet               PetscCall(ISSetIdentity(uis));
17475e642048SPierre Jolivet               if (!data->is) {
17485e642048SPierre Jolivet                 if (C) PetscCall(PetscObjectReference((PetscObject)C));
17495e642048SPierre Jolivet                 else PetscCall(MatTranspose(B, MAT_INITIAL_MATRIX, &C));
17505e642048SPierre Jolivet                 PetscCall(ISDuplicate(data_00->is, is));
17515e642048SPierre Jolivet                 PetscCall(MatIncreaseOverlap(A, 1, is, 1));
17525e642048SPierre Jolivet                 PetscCall(MatSetOption(C, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
17535e642048SPierre Jolivet                 PetscCall(MatCreateSubMatrices(C, 1, is, &uis, MAT_INITIAL_MATRIX, &sub));
17545e642048SPierre Jolivet                 PetscCall(MatDestroy(&C));
17555e642048SPierre Jolivet                 PetscCall(MatTranspose(sub[0], MAT_INITIAL_MATRIX, &C));
17565e642048SPierre Jolivet                 PetscCall(MatDestroySubMatrices(1, &sub));
17575e642048SPierre Jolivet                 PetscCall(MatFindNonzeroRows(C, &data->is));
17585e642048SPierre Jolivet                 PetscCall(MatDestroy(&C));
17595e642048SPierre Jolivet                 PetscCall(ISDestroy(is));
1760d16c0b94SPierre Jolivet                 PetscCall(ISCreateStride(PetscObjectComm((PetscObject)data->is), A11->rmap->n, A11->rmap->rstart, 1, &loc));
1761d16c0b94SPierre Jolivet                 if (PetscDefined(USE_DEBUG)) PetscCall(PCHPDDMCheckInclusion_Private(pc, data->is, loc, PETSC_FALSE));
1762d16c0b94SPierre Jolivet                 PetscCall(ISExpand(data->is, loc, is));
1763d16c0b94SPierre Jolivet                 PetscCall(ISDestroy(&loc));
1764d16c0b94SPierre Jolivet                 PetscCall(ISDestroy(&data->is));
1765d16c0b94SPierre Jolivet                 data->is = is[0];
1766d16c0b94SPierre Jolivet                 is[0]    = nullptr;
17675e642048SPierre Jolivet               }
17685e642048SPierre Jolivet               if (PetscDefined(USE_DEBUG)) {
17695e642048SPierre Jolivet                 PetscCall(PCHPDDMCheckSymmetry_Private(pc, A01, A10));
177013044ca3SPierre Jolivet                 PetscCall(MatCreateSubMatrices(B, 1, &uis, &data_00->is, MAT_INITIAL_MATRIX, &sub)); /* expensive check since all processes fetch all rows (but only some columns) of the constraint matrix */
177113044ca3SPierre Jolivet                 PetscCall(ISDestroy(&uis));
177213044ca3SPierre Jolivet                 PetscCall(ISDuplicate(data->is, &uis));
177313044ca3SPierre Jolivet                 PetscCall(ISSort(uis));
17745e642048SPierre Jolivet                 PetscCall(ISComplement(uis, 0, B->rmap->N, is));
177513044ca3SPierre Jolivet                 PetscCall(MatDuplicate(sub[0], MAT_COPY_VALUES, &C));
17765e642048SPierre Jolivet                 PetscCall(MatZeroRowsIS(C, is[0], 0.0, nullptr, nullptr));
17775e642048SPierre Jolivet                 PetscCall(ISDestroy(is));
177813044ca3SPierre Jolivet                 PetscCall(MatMultEqual(sub[0], C, 20, &flg));
177913044ca3SPierre Jolivet                 PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "The image of A_10 (R_i^p)^T from the local primal (e.g., velocity) space to the full dual (e.g., pressure) space is not restricted to the local dual space: A_10 (R_i^p)^T != R_i^d (R_i^d)^T A_10 (R_i^p)^T"); /* cf. eq. (9) of https://hal.science/hal-02343808v6/document */
178013044ca3SPierre Jolivet                 PetscCall(MatDestroy(&C));
178113044ca3SPierre Jolivet                 PetscCall(MatDestroySubMatrices(1, &sub));
178213044ca3SPierre Jolivet               }
17835e642048SPierre Jolivet               PetscCall(ISDestroy(&uis));
17845e642048SPierre Jolivet               PetscCall(MatDestroy(&B));
17855e642048SPierre Jolivet             }
1786cf67ef9dSPierre Jolivet             flg = PETSC_FALSE;
1787cf67ef9dSPierre Jolivet             if (!data->aux) {
1788cf67ef9dSPierre Jolivet               Mat D;
1789cf67ef9dSPierre Jolivet 
1790cf67ef9dSPierre Jolivet               PetscCall(MatCreateVecs(A11, &d, nullptr));
1791cf67ef9dSPierre Jolivet               PetscCall(MatGetDiagonal(A11, d));
1792cf67ef9dSPierre Jolivet               PetscCall(PetscObjectTypeCompareAny((PetscObject)A11, &flg, MATDIAGONAL, MATCONSTANTDIAGONAL, ""));
1793cf67ef9dSPierre Jolivet               if (!flg) {
1794cf67ef9dSPierre Jolivet                 PetscCall(MatCreateDiagonal(d, &D));
1795cf67ef9dSPierre Jolivet                 PetscCall(MatMultEqual(A11, D, 20, &flg));
1796cf67ef9dSPierre Jolivet                 PetscCall(MatDestroy(&D));
1797cf67ef9dSPierre Jolivet               }
1798cf67ef9dSPierre Jolivet               if (flg) PetscCall(PetscInfo(pc, "A11 block is likely diagonal so the PC will build an auxiliary Mat (which was not initially provided by the user)\n"));
1799cf67ef9dSPierre Jolivet             }
1800cf67ef9dSPierre Jolivet             if (data->Neumann != PETSC_BOOL3_TRUE && !flg && A11) {
1801cf67ef9dSPierre Jolivet               PetscReal norm;
1802cf67ef9dSPierre Jolivet 
1803cf67ef9dSPierre Jolivet               PetscCall(MatNorm(A11, NORM_INFINITY, &norm));
1804cf67ef9dSPierre Jolivet               PetscCheck(norm < PETSC_MACHINE_EPSILON * static_cast<PetscReal>(10.0), PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-%spc_hpddm_schur_precondition geneo and -%spc_hpddm_has_neumann != true with a nonzero or non-diagonal A11 block", pcpre ? pcpre : "", pcpre ? pcpre : "");
1805cf67ef9dSPierre Jolivet               PetscCall(PetscInfo(pc, "A11 block is likely zero so the PC will build an auxiliary Mat (which was%s initially provided by the user)\n", data->aux ? "" : " not"));
1806cf67ef9dSPierre Jolivet               PetscCall(MatDestroy(&data->aux));
1807cf67ef9dSPierre Jolivet               flg = PETSC_TRUE;
1808cf67ef9dSPierre Jolivet             }
1809cf67ef9dSPierre Jolivet             if (!data->aux) { /* if A11 is near zero, e.g., Stokes equation, or diagonal, build an auxiliary (Neumann) Mat which is a (possibly slightly shifted) diagonal weighted by the inverse of the multiplicity */
181001e3c840SPierre Jolivet               PetscSF            scatter;
181113044ca3SPierre Jolivet               const PetscScalar *read;
1812cf67ef9dSPierre Jolivet               PetscScalar       *write, *diagonal = nullptr;
181313044ca3SPierre Jolivet 
181413044ca3SPierre Jolivet               PetscCall(MatDestroy(&data->aux));
181513044ca3SPierre Jolivet               PetscCall(ISGetLocalSize(data->is, &n));
1816cf67ef9dSPierre Jolivet               PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)P), n, PETSC_DECIDE, &xin));
1817cf67ef9dSPierre Jolivet               PetscCall(VecDuplicate(xin, &v));
1818cf67ef9dSPierre Jolivet               PetscCall(VecScatterCreate(xin, data->is, v, nullptr, &scatter));
181913044ca3SPierre Jolivet               PetscCall(VecSet(v, 1.0));
1820cf67ef9dSPierre Jolivet               PetscCall(VecSet(xin, 1.0));
1821cf67ef9dSPierre Jolivet               PetscCall(VecScatterBegin(scatter, v, xin, ADD_VALUES, SCATTER_REVERSE));
1822cf67ef9dSPierre Jolivet               PetscCall(VecScatterEnd(scatter, v, xin, ADD_VALUES, SCATTER_REVERSE)); /* v has the multiplicity of all unknowns on the overlap */
1823cf67ef9dSPierre Jolivet               PetscCall(PetscSFDestroy(&scatter));
1824cf67ef9dSPierre Jolivet               if (d) {
1825cf67ef9dSPierre Jolivet                 PetscCall(VecScatterCreate(d, data->is, v, nullptr, &scatter));
1826cf67ef9dSPierre Jolivet                 PetscCall(VecScatterBegin(scatter, d, v, INSERT_VALUES, SCATTER_FORWARD));
1827cf67ef9dSPierre Jolivet                 PetscCall(VecScatterEnd(scatter, d, v, INSERT_VALUES, SCATTER_FORWARD));
1828cf67ef9dSPierre Jolivet                 PetscCall(PetscSFDestroy(&scatter));
1829cf67ef9dSPierre Jolivet                 PetscCall(VecDestroy(&d));
1830cf67ef9dSPierre Jolivet                 PetscCall(PetscMalloc1(n, &diagonal));
1831cf67ef9dSPierre Jolivet                 PetscCall(VecGetArrayRead(v, &read));
1832cf67ef9dSPierre Jolivet                 PetscCallCXX(std::copy_n(read, n, diagonal));
1833cf67ef9dSPierre Jolivet                 PetscCall(VecRestoreArrayRead(v, &read));
1834cf67ef9dSPierre Jolivet               }
183513044ca3SPierre Jolivet               PetscCall(VecDestroy(&v));
183613044ca3SPierre Jolivet               PetscCall(VecCreateSeq(PETSC_COMM_SELF, n, &v));
1837cf67ef9dSPierre Jolivet               PetscCall(VecGetArrayRead(xin, &read));
183813044ca3SPierre Jolivet               PetscCall(VecGetArrayWrite(v, &write));
1839cf67ef9dSPierre Jolivet               for (PetscInt i = 0; i < n; ++i) write[i] = (!diagonal || std::abs(diagonal[i]) < PETSC_MACHINE_EPSILON) ? PETSC_SMALL / (static_cast<PetscReal>(1000.0) * read[i]) : diagonal[i] / read[i];
1840cf67ef9dSPierre Jolivet               PetscCall(PetscFree(diagonal));
1841cf67ef9dSPierre Jolivet               PetscCall(VecRestoreArrayRead(xin, &read));
184213044ca3SPierre Jolivet               PetscCall(VecRestoreArrayWrite(v, &write));
1843cf67ef9dSPierre Jolivet               PetscCall(VecDestroy(&xin));
1844c3e1b152SPierre Jolivet               PetscCall(MatCreateDiagonal(v, &data->aux));
184513044ca3SPierre Jolivet               PetscCall(VecDestroy(&v));
184613044ca3SPierre Jolivet             }
184713044ca3SPierre Jolivet             uis  = data->is;
184813044ca3SPierre Jolivet             uaux = data->aux;
184913044ca3SPierre Jolivet             PetscCall(PetscObjectReference((PetscObject)uis));
185013044ca3SPierre Jolivet             PetscCall(PetscObjectReference((PetscObject)uaux));
185113044ca3SPierre Jolivet             PetscCall(PetscStrallocpy(pcpre, &prefix));
185213044ca3SPierre Jolivet             PetscCall(PCSetOptionsPrefix(pc, nullptr));
185313044ca3SPierre Jolivet             PetscCall(PCSetType(pc, PCKSP));                                    /* replace the PC associated to the Schur complement by PCKSP */
185413044ca3SPierre Jolivet             PetscCall(KSPCreate(PetscObjectComm((PetscObject)pc), &inner_ksp)); /* new KSP that will be attached to the previously set PC */
185573329a61SPierre Jolivet             pc->ops->setup = PCSetUp_KSP;
185613044ca3SPierre Jolivet             PetscCall(PetscObjectGetTabLevel((PetscObject)pc, &n));
185713044ca3SPierre Jolivet             PetscCall(PetscObjectSetTabLevel((PetscObject)inner_ksp, n + 2));
185813044ca3SPierre Jolivet             PetscCall(KSPSetOperators(inner_ksp, pc->mat, pc->pmat));
185913044ca3SPierre Jolivet             PetscCall(KSPSetOptionsPrefix(inner_ksp, std::string(std::string(prefix) + "pc_hpddm_").c_str()));
186013044ca3SPierre Jolivet             PetscCall(KSPSetSkipPCSetFromOptions(inner_ksp, PETSC_TRUE));
186113044ca3SPierre Jolivet             PetscCall(KSPSetFromOptions(inner_ksp));
186213044ca3SPierre Jolivet             PetscCall(KSPGetPC(inner_ksp, &inner));
186313044ca3SPierre Jolivet             PetscCall(PCSetOptionsPrefix(inner, nullptr));
186413044ca3SPierre Jolivet             PetscCall(PCSetType(inner, PCNONE)); /* no preconditioner since the action of M^-1 A or A M^-1 will be computed by the Amat */
186513044ca3SPierre Jolivet             PetscCall(PCKSPSetKSP(pc, inner_ksp));
186613044ca3SPierre Jolivet             PetscCall(PetscNew(&ctx));    /* context to pass data around for the inner-most PC, which will be a proper PCHPDDM */
186713044ca3SPierre Jolivet             std::get<0>(*ctx)[0] = pc_00; /* for coarse correction on the primal (e.g., velocity) space */
186813044ca3SPierre Jolivet             PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &std::get<0>(*ctx)[1]));
186913044ca3SPierre Jolivet             PetscCall(PCSetOptionsPrefix(std::get<0>(*ctx)[1], prefix));
187013044ca3SPierre Jolivet             PetscCall(PetscFree(prefix));
187113044ca3SPierre Jolivet             PetscCall(PCSetOperators(std::get<0>(*ctx)[1], pc->mat, pc->pmat));
187213044ca3SPierre Jolivet             PetscCall(PCSetType(std::get<0>(*ctx)[1], PCHPDDM));
187313044ca3SPierre Jolivet             PetscCall(PCHPDDMSetAuxiliaryMat(std::get<0>(*ctx)[1], uis, uaux, nullptr, nullptr)); /* transfer ownership of the auxiliary inputs from the inner (PCKSP) to the inner-most (PCHPDDM) PC */
1874cf67ef9dSPierre Jolivet             if (flg) static_cast<PC_HPDDM *>(std::get<0>(*ctx)[1]->data)->Neumann = PETSC_BOOL3_TRUE;
187513044ca3SPierre Jolivet             PetscCall(PCSetFromOptions(std::get<0>(*ctx)[1]));
187613044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)uis));
187713044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)uaux));
187813044ca3SPierre Jolivet             PetscCall(MatCreateShell(PetscObjectComm((PetscObject)pc), inner->mat->rmap->n, inner->mat->cmap->n, inner->mat->rmap->N, inner->mat->cmap->N, ctx, &S)); /* MatShell computing the action of M^-1 A or A M^-1 */
187913044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(S, MATOP_MULT, (void (*)(void))MatMult_SchurCorrection));
188013044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(S, MATOP_VIEW, (void (*)(void))MatView_SchurCorrection));
188113044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(S, MATOP_DESTROY, (void (*)(void))MatDestroy_SchurCorrection));
188213044ca3SPierre Jolivet             PetscCall(KSPGetPCSide(inner_ksp, &(std::get<2>(*ctx))));
188313044ca3SPierre Jolivet             if (std::get<2>(*ctx) == PC_LEFT || std::get<2>(*ctx) == PC_SIDE_DEFAULT) {
188413044ca3SPierre Jolivet               PetscCall(KSPSetPreSolve(inner_ksp, KSPPreSolve_SchurCorrection, ctx));
188513044ca3SPierre Jolivet             } else { /* no support for PC_SYMMETRIC */
188613044ca3SPierre Jolivet               PetscCheck(std::get<2>(*ctx) == PC_RIGHT, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "PCSide %s (!= %s or %s or %s)", PCSides[std::get<2>(*ctx)], PCSides[PC_SIDE_DEFAULT], PCSides[PC_LEFT], PCSides[PC_RIGHT]);
188713044ca3SPierre Jolivet             }
188813044ca3SPierre Jolivet             PetscCall(KSPSetPostSolve(inner_ksp, KSPPostSolve_SchurCorrection, ctx));
1889715c1178SPierre Jolivet             PetscCall(PetscObjectContainerCompose((PetscObject)std::get<0>(*ctx)[1], "_PCHPDDM_Schur", ctx, nullptr));
189013044ca3SPierre Jolivet             PetscCall(PCSetUp(std::get<0>(*ctx)[1]));
189113044ca3SPierre Jolivet             PetscCall(KSPSetOperators(inner_ksp, S, S));
189213044ca3SPierre Jolivet             PetscCall(MatCreateVecs(std::get<1>(*ctx)[0], std::get<3>(*ctx), std::get<3>(*ctx) + 1));
189313044ca3SPierre Jolivet             PetscCall(VecDuplicate(std::get<3>(*ctx)[0], std::get<3>(*ctx) + 2));
189413044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)inner_ksp));
189513044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)S));
18960307214fSPierre Jolivet             for (std::vector<Vec>::iterator it = initial.begin(); it != initial.end(); ++it) PetscCall(VecDestroy(&*it));
189713044ca3SPierre Jolivet             PetscFunctionReturn(PETSC_SUCCESS);
189813044ca3SPierre Jolivet           } else { /* second call to PCSetUp() on the PC associated to the Schur complement, retrieve previously set context */
189913044ca3SPierre Jolivet             PetscCall(PetscContainerGetPointer(container, (void **)&ctx));
190013044ca3SPierre Jolivet           }
190113044ca3SPierre Jolivet         }
190213044ca3SPierre Jolivet       }
190313044ca3SPierre Jolivet     }
1904f1580f4eSBarry Smith     if (!data->is && data->N > 1) {
1905f1580f4eSBarry Smith       char type[256] = {}; /* same size as in src/ksp/pc/interface/pcset.c */
1906811e8887SPierre Jolivet 
1907f1580f4eSBarry Smith       PetscCall(PetscObjectTypeCompareAny((PetscObject)P, &flg, MATNORMAL, MATNORMALHERMITIAN, ""));
1908f1580f4eSBarry Smith       if (flg || (A->rmap->N != A->cmap->N && P->rmap->N == P->cmap->N && P->rmap->N == A->cmap->N)) {
1909f1580f4eSBarry Smith         Mat B;
1910811e8887SPierre Jolivet 
1911f1580f4eSBarry Smith         PetscCall(PCHPDDMSetAuxiliaryMatNormal_Private(pc, A, P, &B, pcpre));
1912f1580f4eSBarry Smith         if (data->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED) data->correction = PC_HPDDM_COARSE_CORRECTION_BALANCED;
1913f1580f4eSBarry Smith         PetscCall(MatDestroy(&B));
1914f1580f4eSBarry Smith       } else {
1915f1580f4eSBarry Smith         PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg));
1916f1580f4eSBarry Smith         if (flg) {
1917f1580f4eSBarry Smith           Mat                 A00, P00, A01, A10, A11, B, N;
191813044ca3SPierre Jolivet           PCHPDDMSchurPreType type = PC_HPDDM_SCHUR_PRE_LEAST_SQUARES;
191913044ca3SPierre Jolivet 
192013044ca3SPierre Jolivet           PetscCall(MatSchurComplementGetSubMatrices(P, &A00, &P00, &A01, &A10, &A11));
192113044ca3SPierre Jolivet           if (PetscDefined(USE_DEBUG)) PetscCall(PCHPDDMCheckSymmetry_Private(pc, A01, A10));
192213044ca3SPierre Jolivet           PetscCall(PetscOptionsGetEnum(nullptr, pcpre, "-pc_hpddm_schur_precondition", PCHPDDMSchurPreTypes, (PetscEnum *)&type, &flg));
192313044ca3SPierre Jolivet           if (type == PC_HPDDM_SCHUR_PRE_LEAST_SQUARES) {
19243df4cd7bSPierre Jolivet             Vec                        diagonal = nullptr;
1925f1580f4eSBarry Smith             const PetscScalar         *array;
1926f1580f4eSBarry Smith             MatSchurComplementAinvType type;
1927f1580f4eSBarry Smith 
1928f1580f4eSBarry Smith             if (A11) {
19293df4cd7bSPierre Jolivet               PetscCall(MatCreateVecs(A11, &diagonal, nullptr));
19303df4cd7bSPierre Jolivet               PetscCall(MatGetDiagonal(A11, diagonal));
1931f1580f4eSBarry Smith             }
1932db4a47b3SPierre Jolivet             PetscCall(MatCreateVecs(P00, &v, nullptr));
1933f1580f4eSBarry Smith             PetscCall(MatSchurComplementGetAinvType(P, &type));
1934f1580f4eSBarry Smith             PetscCheck(type == MAT_SCHUR_COMPLEMENT_AINV_DIAG || type == MAT_SCHUR_COMPLEMENT_AINV_LUMP, PetscObjectComm((PetscObject)P), PETSC_ERR_SUP, "-%smat_schur_complement_ainv_type %s", ((PetscObject)P)->prefix ? ((PetscObject)P)->prefix : "", MatSchurComplementAinvTypes[type]);
1935f1580f4eSBarry Smith             if (type == MAT_SCHUR_COMPLEMENT_AINV_LUMP) {
1936f1580f4eSBarry Smith               PetscCall(MatGetRowSum(P00, v));
1937f1580f4eSBarry Smith               if (A00 == P00) PetscCall(PetscObjectReference((PetscObject)A00));
1938f1580f4eSBarry Smith               PetscCall(MatDestroy(&P00));
1939f1580f4eSBarry Smith               PetscCall(VecGetArrayRead(v, &array));
1940db4a47b3SPierre Jolivet               PetscCall(MatCreateAIJ(PetscObjectComm((PetscObject)A00), A00->rmap->n, A00->cmap->n, A00->rmap->N, A00->cmap->N, 1, nullptr, 0, nullptr, &P00));
1941f1580f4eSBarry Smith               PetscCall(MatSetOption(P00, MAT_NO_OFF_PROC_ENTRIES, PETSC_TRUE));
1942f1580f4eSBarry Smith               for (n = A00->rmap->rstart; n < A00->rmap->rend; ++n) PetscCall(MatSetValue(P00, n, n, array[n - A00->rmap->rstart], INSERT_VALUES));
1943f1580f4eSBarry Smith               PetscCall(MatAssemblyBegin(P00, MAT_FINAL_ASSEMBLY));
1944f1580f4eSBarry Smith               PetscCall(MatAssemblyEnd(P00, MAT_FINAL_ASSEMBLY));
1945f1580f4eSBarry Smith               PetscCall(VecRestoreArrayRead(v, &array));
1946f1580f4eSBarry Smith               PetscCall(MatSchurComplementUpdateSubMatrices(P, A00, P00, A01, A10, A11)); /* replace P00 by diag(sum of each row of P00) */
1947f1580f4eSBarry Smith               PetscCall(MatDestroy(&P00));
1948f1580f4eSBarry Smith             } else PetscCall(MatGetDiagonal(P00, v));
1949f1580f4eSBarry Smith             PetscCall(VecReciprocal(v)); /* inv(diag(P00))       */
1950f1580f4eSBarry Smith             PetscCall(VecSqrtAbs(v));    /* sqrt(inv(diag(P00))) */
1951f1580f4eSBarry Smith             PetscCall(MatDuplicate(A01, MAT_COPY_VALUES, &B));
1952db4a47b3SPierre Jolivet             PetscCall(MatDiagonalScale(B, v, nullptr));
1953f1580f4eSBarry Smith             PetscCall(VecDestroy(&v));
1954f1580f4eSBarry Smith             PetscCall(MatCreateNormalHermitian(B, &N));
19553df4cd7bSPierre Jolivet             PetscCall(PCHPDDMSetAuxiliaryMatNormal_Private(pc, B, N, &P, pcpre, &diagonal));
1956f1580f4eSBarry Smith             PetscCall(PetscObjectTypeCompare((PetscObject)data->aux, MATSEQAIJ, &flg));
1957f1580f4eSBarry Smith             if (!flg) {
1958f1580f4eSBarry Smith               PetscCall(MatDestroy(&P));
1959f1580f4eSBarry Smith               P = N;
1960f1580f4eSBarry Smith               PetscCall(PetscObjectReference((PetscObject)P));
1961f1580f4eSBarry Smith             } else PetscCall(MatScale(P, -1.0));
19623df4cd7bSPierre Jolivet             if (diagonal) {
19633df4cd7bSPierre Jolivet               PetscCall(MatDiagonalSet(P, diagonal, ADD_VALUES));
1964aa21023fSPierre Jolivet               PetscCall(PCSetOperators(pc, P, P)); /* replace P by diag(P11) - A01^T inv(diag(P00)) A01 */
19653df4cd7bSPierre Jolivet               PetscCall(VecDestroy(&diagonal));
19663df4cd7bSPierre Jolivet             } else {
1967f1580f4eSBarry Smith               PetscCall(MatScale(N, -1.0));
1968aa21023fSPierre Jolivet               PetscCall(PCSetOperators(pc, N, P)); /* replace P by - A01^T inv(diag(P00)) A01 */
19693df4cd7bSPierre Jolivet             }
1970f1580f4eSBarry Smith             PetscCall(MatDestroy(&N));
1971f1580f4eSBarry Smith             PetscCall(MatDestroy(&P));
1972f1580f4eSBarry Smith             PetscCall(MatDestroy(&B));
197313044ca3SPierre Jolivet           } else
1974fd310a01SPierre Jolivet             PetscCheck(type != PC_HPDDM_SCHUR_PRE_GENEO, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-%spc_hpddm_schur_precondition %s without a prior call to PCHPDDMSetAuxiliaryMat() on the A11 block%s%s", pcpre ? pcpre : "", PCHPDDMSchurPreTypes[type], pcpre ? " -" : "", pcpre ? pcpre : "");
19750307214fSPierre Jolivet           for (std::vector<Vec>::iterator it = initial.begin(); it != initial.end(); ++it) PetscCall(VecDestroy(&*it));
19763ba16761SJacob Faibussowitsch           PetscFunctionReturn(PETSC_SUCCESS);
1977f1580f4eSBarry Smith         } else {
1978db4a47b3SPierre Jolivet           PetscCall(PetscOptionsGetString(nullptr, pcpre, "-pc_hpddm_levels_1_st_pc_type", type, sizeof(type), nullptr));
1979f1580f4eSBarry Smith           PetscCall(PetscStrcmp(type, PCMAT, &algebraic));
1980b07dfdedSPierre Jolivet           PetscCheck(!algebraic || !block, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_levels_1_st_pc_type mat and -pc_hpddm_block_splitting");
19819bb5c669SPierre Jolivet           if (overlap != -1) {
19829bb5c669SPierre Jolivet             PetscCheck(!block && !algebraic, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_%s and -pc_hpddm_harmonic_overlap", block ? "block_splitting" : "levels_1_st_pc_type mat");
19839bb5c669SPierre Jolivet             PetscCheck(overlap >= 1, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_WRONG, "-pc_hpddm_harmonic_overlap %" PetscInt_FMT " < 1", overlap);
19849bb5c669SPierre Jolivet           }
19859bb5c669SPierre Jolivet           if (block || overlap != -1) algebraic = PETSC_TRUE;
1986f1580f4eSBarry Smith           if (algebraic) {
1987f1580f4eSBarry Smith             PetscCall(ISCreateStride(PETSC_COMM_SELF, P->rmap->n, P->rmap->rstart, 1, &data->is));
1988f1580f4eSBarry Smith             PetscCall(MatIncreaseOverlap(P, 1, &data->is, 1));
1989f1580f4eSBarry Smith             PetscCall(ISSort(data->is));
19909bb5c669SPierre Jolivet           } else
19919bb5c669SPierre Jolivet             PetscCall(PetscInfo(pc, "Cannot assemble a fully-algebraic coarse operator with an assembled Pmat and -%spc_hpddm_levels_1_st_pc_type != mat and -%spc_hpddm_block_splitting != true and -%spc_hpddm_harmonic_overlap < 1\n", pcpre ? pcpre : "", pcpre ? pcpre : "", pcpre ? pcpre : ""));
1992f1580f4eSBarry Smith         }
1993f1580f4eSBarry Smith       }
1994f1580f4eSBarry Smith     }
1995f1580f4eSBarry Smith   }
1996398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG)
1997398c7888SPierre Jolivet   if (data->is) PetscCall(ISDuplicate(data->is, &dis));
1998398c7888SPierre Jolivet   if (data->aux) PetscCall(MatDuplicate(data->aux, MAT_COPY_VALUES, &daux));
1999398c7888SPierre Jolivet #endif
2000f1580f4eSBarry Smith   if (data->is || (ismatis && data->N > 1)) {
2001f1580f4eSBarry Smith     if (ismatis) {
2002f1580f4eSBarry Smith       std::initializer_list<std::string> list = {MATSEQBAIJ, MATSEQSBAIJ};
2003f1580f4eSBarry Smith       PetscCall(MatISGetLocalMat(P, &N));
2004f1580f4eSBarry Smith       std::initializer_list<std::string>::const_iterator it = std::find(list.begin(), list.end(), ((PetscObject)N)->type_name);
2005f1580f4eSBarry Smith       PetscCall(MatISRestoreLocalMat(P, &N));
2006f1580f4eSBarry Smith       switch (std::distance(list.begin(), it)) {
2007d71ae5a4SJacob Faibussowitsch       case 0:
2008d71ae5a4SJacob Faibussowitsch         PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &C));
2009d71ae5a4SJacob Faibussowitsch         break;
2010f1580f4eSBarry Smith       case 1:
2011f1580f4eSBarry Smith         /* MatCreateSubMatrices() does not work with MATSBAIJ and unsorted ISes, so convert to MPIBAIJ */
2012f1580f4eSBarry Smith         PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &C));
2013f1580f4eSBarry Smith         PetscCall(MatSetOption(C, MAT_SYMMETRIC, PETSC_TRUE));
2014f1580f4eSBarry Smith         break;
2015d71ae5a4SJacob Faibussowitsch       default:
2016d71ae5a4SJacob Faibussowitsch         PetscCall(MatConvert(P, MATMPIAIJ, MAT_INITIAL_MATRIX, &C));
2017f1580f4eSBarry Smith       }
2018db4a47b3SPierre Jolivet       PetscCall(MatISGetLocalToGlobalMapping(P, &l2g, nullptr));
2019f1580f4eSBarry Smith       PetscCall(PetscObjectReference((PetscObject)P));
2020f1580f4eSBarry Smith       PetscCall(KSPSetOperators(data->levels[0]->ksp, A, C));
2021f1580f4eSBarry Smith       std::swap(C, P);
2022f1580f4eSBarry Smith       PetscCall(ISLocalToGlobalMappingGetSize(l2g, &n));
2023f1580f4eSBarry Smith       PetscCall(ISCreateStride(PETSC_COMM_SELF, n, 0, 1, &loc));
2024f1580f4eSBarry Smith       PetscCall(ISLocalToGlobalMappingApplyIS(l2g, loc, &is[0]));
2025f1580f4eSBarry Smith       PetscCall(ISDestroy(&loc));
2026f1580f4eSBarry Smith       /* the auxiliary Mat is _not_ the local Neumann matrix                                */
2027f1580f4eSBarry Smith       /* it is the local Neumann matrix augmented (with zeros) through MatIncreaseOverlap() */
2028c8ea6600SPierre Jolivet       data->Neumann = PETSC_BOOL3_FALSE;
2029f1580f4eSBarry Smith       structure     = SAME_NONZERO_PATTERN;
2030f1580f4eSBarry Smith     } else {
2031f1580f4eSBarry Smith       is[0] = data->is;
203213044ca3SPierre Jolivet       if (algebraic || ctx) subdomains = PETSC_TRUE;
2033db4a47b3SPierre Jolivet       PetscCall(PetscOptionsGetBool(nullptr, pcpre, "-pc_hpddm_define_subdomains", &subdomains, nullptr));
203413044ca3SPierre Jolivet       if (ctx) PetscCheck(subdomains, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-%spc_hpddm_schur_precondition geneo and -%spc_hpddm_define_subdomains false", pcpre, pcpre);
2035c8ea6600SPierre Jolivet       if (PetscBool3ToBool(data->Neumann)) {
2036b07dfdedSPierre Jolivet         PetscCheck(!block, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_block_splitting and -pc_hpddm_has_neumann");
20379bb5c669SPierre Jolivet         PetscCheck(overlap == -1, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_harmonic_overlap %" PetscInt_FMT " and -pc_hpddm_has_neumann", overlap);
2038b07dfdedSPierre Jolivet         PetscCheck(!algebraic, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_levels_1_st_pc_type mat and -pc_hpddm_has_neumann");
2039f1580f4eSBarry Smith       }
2040c8ea6600SPierre Jolivet       if (PetscBool3ToBool(data->Neumann) || block) structure = SAME_NONZERO_PATTERN;
2041f1580f4eSBarry Smith       PetscCall(ISCreateStride(PetscObjectComm((PetscObject)data->is), P->rmap->n, P->rmap->rstart, 1, &loc));
2042f1580f4eSBarry Smith     }
2043f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : ""));
2044db4a47b3SPierre Jolivet     PetscCall(PetscOptionsGetEnum(nullptr, prefix, "-st_matstructure", MatStructures, (PetscEnum *)&structure, &flg)); /* if not user-provided, force its value when possible */
2045f1580f4eSBarry Smith     if (!flg && structure == SAME_NONZERO_PATTERN) {                                                                   /* cannot call STSetMatStructure() yet, insert the appropriate option in the database, parsed by STSetFromOptions() */
2046f1580f4eSBarry Smith       PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-%spc_hpddm_levels_1_st_matstructure", pcpre ? pcpre : ""));
2047db4a47b3SPierre Jolivet       PetscCall(PetscOptionsSetValue(nullptr, prefix, MatStructures[structure]));
2048f1580f4eSBarry Smith     }
2049398c7888SPierre Jolivet     flg = PETSC_FALSE;
2050b07dfdedSPierre Jolivet     if (data->share) {
2051398c7888SPierre Jolivet       data->share = PETSC_FALSE; /* will be reset to PETSC_TRUE if none of the conditions below are true */
2052398c7888SPierre Jolivet       if (!subdomains) PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc since -%spc_hpddm_define_subdomains is not true\n", pcpre ? pcpre : ""));
2053398c7888SPierre Jolivet       else if (data->deflation) PetscCall(PetscInfo(pc, "Nothing to share since PCHPDDMSetDeflationMat() has been called\n"));
2054398c7888SPierre Jolivet       else if (ismatis) PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc with a Pmat of type MATIS\n"));
2055b07dfdedSPierre Jolivet       else if (!algebraic && structure != SAME_NONZERO_PATTERN)
2056398c7888SPierre Jolivet         PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc since -%spc_hpddm_levels_1_st_matstructure %s (!= %s)\n", pcpre ? pcpre : "", MatStructures[structure], MatStructures[SAME_NONZERO_PATTERN]));
2057398c7888SPierre Jolivet       else data->share = PETSC_TRUE;
2058398c7888SPierre Jolivet     }
2059398c7888SPierre Jolivet     if (!ismatis) {
2060398c7888SPierre Jolivet       if (data->share || (!PetscBool3ToBool(data->Neumann) && subdomains)) PetscCall(ISDuplicate(is[0], &unsorted));
2061398c7888SPierre Jolivet       else unsorted = is[0];
2062398c7888SPierre Jolivet     }
2063f1580f4eSBarry Smith     if (data->N > 1 && (data->aux || ismatis || algebraic)) {
2064f1580f4eSBarry Smith       PetscCheck(loadedSym, PETSC_COMM_SELF, PETSC_ERR_PLIB, "HPDDM library not loaded, cannot use more than one level");
2065f1580f4eSBarry Smith       PetscCall(MatSetOption(P, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
2066f1580f4eSBarry Smith       if (ismatis) {
2067f1580f4eSBarry Smith         /* needed by HPDDM (currently) so that the partition of unity is 0 on subdomain interfaces */
2068f1580f4eSBarry Smith         PetscCall(MatIncreaseOverlap(P, 1, is, 1));
2069f1580f4eSBarry Smith         PetscCall(ISDestroy(&data->is));
2070f1580f4eSBarry Smith         data->is = is[0];
2071f1580f4eSBarry Smith       } else {
2072d16c0b94SPierre Jolivet         if (PetscDefined(USE_DEBUG)) PetscCall(PCHPDDMCheckInclusion_Private(pc, data->is, loc, PETSC_TRUE));
20739bb5c669SPierre Jolivet         if (overlap == -1) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", PCHPDDMAlgebraicAuxiliaryMat_Private));
20749bb5c669SPierre Jolivet         if (!PetscBool3ToBool(data->Neumann) && (!algebraic || overlap != -1)) {
2075f1580f4eSBarry Smith           PetscCall(PetscObjectTypeCompare((PetscObject)P, MATMPISBAIJ, &flg));
2076f1580f4eSBarry Smith           if (flg) {
2077f1580f4eSBarry Smith             /* maybe better to ISSort(is[0]), MatCreateSubMatrices(), and then MatPermute() */
2078f1580f4eSBarry Smith             /* but there is no MatPermute_SeqSBAIJ(), so as before, just use MATMPIBAIJ     */
2079f1580f4eSBarry Smith             PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &uaux));
2080f1580f4eSBarry Smith             flg = PETSC_FALSE;
2081f1580f4eSBarry Smith           }
2082f1580f4eSBarry Smith         }
2083f1580f4eSBarry Smith       }
20849bb5c669SPierre Jolivet       if (algebraic && overlap == -1) {
2085f1580f4eSBarry Smith         PetscUseMethod(pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", (Mat, IS *, Mat *[], PetscBool), (P, is, &sub, block));
2086f1580f4eSBarry Smith         if (block) {
2087f1580f4eSBarry Smith           PetscCall(PetscObjectQuery((PetscObject)sub[0], "_PCHPDDM_Neumann_Mat", (PetscObject *)&data->aux));
2088db4a47b3SPierre Jolivet           PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Neumann_Mat", nullptr));
2089f1580f4eSBarry Smith         }
20909bb5c669SPierre Jolivet       } else if (!uaux || overlap != -1) {
209113044ca3SPierre Jolivet         if (!ctx) {
2092c8ea6600SPierre Jolivet           if (PetscBool3ToBool(data->Neumann)) sub = &data->aux;
20939bb5c669SPierre Jolivet           else {
2094df21ef68SPierre Jolivet             PetscBool flg;
20959bb5c669SPierre Jolivet             if (overlap != -1) {
20969bb5c669SPierre Jolivet               Harmonic              h;
20979bb5c669SPierre Jolivet               Mat                   A0, *a;                           /* with an SVD: [ A_00  A_01       ] */
20989bb5c669SPierre Jolivet               IS                    ov[2], rows, cols, stride;        /*              [ A_10  A_11  A_12 ] */
20999bb5c669SPierre Jolivet               const PetscInt       *i[2], bs = PetscAbs(P->cmap->bs); /* with a GEVP: [ A_00  A_01       ] */
21009bb5c669SPierre Jolivet               PetscInt              n[2];                             /*              [ A_10  A_11  A_12 ] */
21019bb5c669SPierre Jolivet               std::vector<PetscInt> v[2];                             /*              [       A_21  A_22 ] */
21029bb5c669SPierre Jolivet 
21039bb5c669SPierre Jolivet               PetscCall(ISDuplicate(data->is, ov));
21049bb5c669SPierre Jolivet               if (overlap > 1) PetscCall(MatIncreaseOverlap(P, 1, ov, overlap - 1));
21059bb5c669SPierre Jolivet               PetscCall(ISDuplicate(ov[0], ov + 1));
21069bb5c669SPierre Jolivet               PetscCall(MatIncreaseOverlap(P, 1, ov + 1, 1));
21079bb5c669SPierre Jolivet               PetscCall(PetscNew(&h));
21089bb5c669SPierre Jolivet               h->ksp = nullptr;
21099bb5c669SPierre Jolivet               PetscCall(PetscCalloc1(2, &h->A));
21103a60673dSPierre Jolivet               PetscCall(PetscOptionsHasName(nullptr, prefix, "-svd_nsv", &flg));
21119bb5c669SPierre Jolivet               if (!flg) PetscCall(PetscOptionsHasName(nullptr, prefix, "-svd_relative_threshold", &flg));
21129bb5c669SPierre Jolivet               PetscCall(ISSort(ov[0]));
21139bb5c669SPierre Jolivet               if (!flg) PetscCall(ISSort(ov[1]));
211432603206SJames Wright               PetscCall(PetscCalloc1(5, &h->is));
21159bb5c669SPierre Jolivet               PetscCall(MatCreateSubMatrices(uaux ? uaux : P, 1, ov + !flg, ov + 1, MAT_INITIAL_MATRIX, &a)); /* submatrix from above, either square (!flg) or rectangular (flg) */
21169bb5c669SPierre Jolivet               for (PetscInt j = 0; j < 2; ++j) {
21179bb5c669SPierre Jolivet                 PetscCall(ISGetIndices(ov[j], i + j));
21189bb5c669SPierre Jolivet                 PetscCall(ISGetLocalSize(ov[j], n + j));
21199bb5c669SPierre Jolivet               }
21209bb5c669SPierre Jolivet               v[1].reserve((n[1] - n[0]) / bs);
21219bb5c669SPierre Jolivet               for (PetscInt j = 0; j < n[1]; j += bs) { /* indices of the (2,2) block */
21229bb5c669SPierre Jolivet                 PetscInt location;
21239bb5c669SPierre Jolivet                 PetscCall(ISLocate(ov[0], i[1][j], &location));
21249bb5c669SPierre Jolivet                 if (location < 0) v[1].emplace_back(j / bs);
21259bb5c669SPierre Jolivet               }
21269bb5c669SPierre Jolivet               if (!flg) {
21279bb5c669SPierre Jolivet                 h->A[1] = a[0];
21289bb5c669SPierre Jolivet                 PetscCall(PetscObjectReference((PetscObject)h->A[1]));
21299bb5c669SPierre Jolivet                 v[0].reserve((n[0] - P->rmap->n) / bs);
21309bb5c669SPierre Jolivet                 for (PetscInt j = 0; j < n[1]; j += bs) { /* row indices of the (1,2) block */
21319bb5c669SPierre Jolivet                   PetscInt location;
21329bb5c669SPierre Jolivet                   PetscCall(ISLocate(loc, i[1][j], &location));
21339bb5c669SPierre Jolivet                   if (location < 0) {
21349bb5c669SPierre Jolivet                     PetscCall(ISLocate(ov[0], i[1][j], &location));
21359bb5c669SPierre Jolivet                     if (location >= 0) v[0].emplace_back(j / bs);
21369bb5c669SPierre Jolivet                   }
21379bb5c669SPierre Jolivet                 }
21389bb5c669SPierre Jolivet                 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[0].size(), v[0].data(), PETSC_USE_POINTER, &rows));
21399bb5c669SPierre Jolivet                 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[1].size(), v[1].data(), PETSC_COPY_VALUES, h->is + 4));
21409bb5c669SPierre Jolivet                 PetscCall(MatCreateSubMatrix(a[0], rows, h->is[4], MAT_INITIAL_MATRIX, h->A)); /* A_12 submatrix from above */
21419bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&rows));
21429bb5c669SPierre Jolivet                 if (uaux) PetscCall(MatConvert(a[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, a)); /* initial Pmat was MATSBAIJ, convert back to the same format since the rectangular A_12 submatrix has been created */
21439bb5c669SPierre Jolivet                 PetscCall(ISEmbed(ov[0], ov[1], PETSC_TRUE, &rows));
21449bb5c669SPierre Jolivet                 PetscCall(MatCreateSubMatrix(a[0], rows, cols = rows, MAT_INITIAL_MATRIX, &A0)); /* [ A_00  A_01 ; A_10  A_11 ] submatrix from above */
21459bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&rows));
21469bb5c669SPierre Jolivet                 v[0].clear();
21479bb5c669SPierre Jolivet                 PetscCall(ISEmbed(loc, ov[1], PETSC_TRUE, h->is + 3));
21489bb5c669SPierre Jolivet                 PetscCall(ISEmbed(data->is, ov[1], PETSC_TRUE, h->is + 2));
21499bb5c669SPierre Jolivet               }
21509bb5c669SPierre Jolivet               v[0].reserve((n[0] - P->rmap->n) / bs);
21519bb5c669SPierre Jolivet               for (PetscInt j = 0; j < n[0]; j += bs) {
21529bb5c669SPierre Jolivet                 PetscInt location;
21539bb5c669SPierre Jolivet                 PetscCall(ISLocate(loc, i[0][j], &location));
21549bb5c669SPierre Jolivet                 if (location < 0) v[0].emplace_back(j / bs);
21559bb5c669SPierre Jolivet               }
21569bb5c669SPierre Jolivet               PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[0].size(), v[0].data(), PETSC_USE_POINTER, &rows));
21579bb5c669SPierre Jolivet               for (PetscInt j = 0; j < 2; ++j) PetscCall(ISRestoreIndices(ov[j], i + j));
21589bb5c669SPierre Jolivet               if (flg) {
21599bb5c669SPierre Jolivet                 IS is;
21609bb5c669SPierre Jolivet                 PetscCall(ISCreateStride(PETSC_COMM_SELF, a[0]->rmap->n, 0, 1, &is));
21619bb5c669SPierre Jolivet                 PetscCall(ISEmbed(ov[0], ov[1], PETSC_TRUE, &cols));
21629bb5c669SPierre Jolivet                 PetscCall(MatCreateSubMatrix(a[0], is, cols, MAT_INITIAL_MATRIX, &A0)); /* [ A_00  A_01 ; A_10  A_11 ] submatrix from above */
21639bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&cols));
21649bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&is));
21659bb5c669SPierre Jolivet                 if (uaux) PetscCall(MatConvert(A0, MATSEQSBAIJ, MAT_INPLACE_MATRIX, &A0)); /* initial Pmat was MATSBAIJ, convert back to the same format since this submatrix is square */
21669bb5c669SPierre Jolivet                 PetscCall(ISEmbed(loc, data->is, PETSC_TRUE, h->is + 2));
21679bb5c669SPierre Jolivet                 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[1].size(), v[1].data(), PETSC_USE_POINTER, &cols));
21689bb5c669SPierre Jolivet                 PetscCall(MatCreateSubMatrix(a[0], rows, cols, MAT_INITIAL_MATRIX, h->A)); /* A_12 submatrix from above */
21699bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&cols));
21709bb5c669SPierre Jolivet               }
21719bb5c669SPierre Jolivet               PetscCall(ISCreateStride(PETSC_COMM_SELF, A0->rmap->n, 0, 1, &stride));
21729bb5c669SPierre Jolivet               PetscCall(ISEmbed(rows, stride, PETSC_TRUE, h->is));
21739bb5c669SPierre Jolivet               PetscCall(ISDestroy(&stride));
21749bb5c669SPierre Jolivet               PetscCall(ISDestroy(&rows));
21759bb5c669SPierre Jolivet               PetscCall(ISEmbed(loc, ov[0], PETSC_TRUE, h->is + 1));
21769bb5c669SPierre Jolivet               if (subdomains) {
21779bb5c669SPierre Jolivet                 if (!data->levels[0]->pc) {
21789bb5c669SPierre Jolivet                   PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->pc));
21799bb5c669SPierre Jolivet                   PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : ""));
21809bb5c669SPierre Jolivet                   PetscCall(PCSetOptionsPrefix(data->levels[0]->pc, prefix));
21819bb5c669SPierre Jolivet                   PetscCall(PCSetOperators(data->levels[0]->pc, A, P));
21829bb5c669SPierre Jolivet                 }
21839bb5c669SPierre Jolivet                 PetscCall(PCSetType(data->levels[0]->pc, PCASM));
21849bb5c669SPierre Jolivet                 if (!data->levels[0]->pc->setupcalled) PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, ov + !flg, &loc));
21859bb5c669SPierre Jolivet                 PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(data->levels[0]->pc, flg ? A0 : a[0], PETSC_TRUE));
21869bb5c669SPierre Jolivet                 if (!flg) ++overlap;
21879bb5c669SPierre Jolivet                 if (data->share) {
21889bb5c669SPierre Jolivet                   PetscInt n = -1;
21899bb5c669SPierre Jolivet                   PetscTryMethod(data->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data->levels[0]->pc, &n, nullptr, &ksp));
21909bb5c669SPierre Jolivet                   PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", n);
21919bb5c669SPierre Jolivet                   if (flg) {
21929bb5c669SPierre Jolivet                     h->ksp = ksp[0];
21939bb5c669SPierre Jolivet                     PetscCall(PetscObjectReference((PetscObject)h->ksp));
21949bb5c669SPierre Jolivet                   }
21959bb5c669SPierre Jolivet                 }
21969bb5c669SPierre Jolivet               }
21979bb5c669SPierre Jolivet               if (!h->ksp) {
21989bb5c669SPierre Jolivet                 PetscBool share = data->share;
21999bb5c669SPierre Jolivet                 PetscCall(KSPCreate(PETSC_COMM_SELF, &h->ksp));
22009bb5c669SPierre Jolivet                 PetscCall(KSPSetType(h->ksp, KSPPREONLY));
22019bb5c669SPierre Jolivet                 PetscCall(KSPSetOperators(h->ksp, A0, A0));
22029bb5c669SPierre Jolivet                 do {
22039bb5c669SPierre Jolivet                   if (!data->share) {
22049bb5c669SPierre Jolivet                     share = PETSC_FALSE;
22059bb5c669SPierre Jolivet                     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_%s", pcpre ? pcpre : "", flg ? "svd_" : "eps_"));
22069bb5c669SPierre Jolivet                     PetscCall(KSPSetOptionsPrefix(h->ksp, prefix));
22079bb5c669SPierre Jolivet                     PetscCall(KSPSetFromOptions(h->ksp));
22089bb5c669SPierre Jolivet                   } else {
22099bb5c669SPierre Jolivet                     MatSolverType type;
22109bb5c669SPierre Jolivet                     PetscCall(KSPGetPC(ksp[0], &pc));
22119bb5c669SPierre Jolivet                     PetscCall(PetscObjectTypeCompareAny((PetscObject)pc, &data->share, PCLU, PCCHOLESKY, ""));
22129bb5c669SPierre Jolivet                     if (data->share) {
22139bb5c669SPierre Jolivet                       PetscCall(PCFactorGetMatSolverType(pc, &type));
22149bb5c669SPierre Jolivet                       if (!type) {
22159bb5c669SPierre Jolivet                         if (PetscDefined(HAVE_MUMPS)) PetscCall(PCFactorSetMatSolverType(pc, MATSOLVERMUMPS));
22169bb5c669SPierre Jolivet                         else if (PetscDefined(HAVE_MKL_PARDISO)) PetscCall(PCFactorSetMatSolverType(pc, MATSOLVERMKL_PARDISO));
22179bb5c669SPierre Jolivet                         else data->share = PETSC_FALSE;
22189bb5c669SPierre Jolivet                         if (data->share) PetscCall(PCSetFromOptions(pc));
22199bb5c669SPierre Jolivet                       } else {
22209bb5c669SPierre Jolivet                         PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &data->share));
22219bb5c669SPierre Jolivet                         if (!data->share) PetscCall(PetscStrcmp(type, MATSOLVERMKL_PARDISO, &data->share));
22229bb5c669SPierre Jolivet                       }
22239bb5c669SPierre Jolivet                       if (data->share) {
22249bb5c669SPierre Jolivet                         std::tuple<KSP, IS, Vec[2]> *p;
22259bb5c669SPierre Jolivet                         PetscCall(PCFactorGetMatrix(pc, &A));
22269bb5c669SPierre Jolivet                         PetscCall(MatFactorSetSchurIS(A, h->is[4]));
22279bb5c669SPierre Jolivet                         PetscCall(KSPSetUp(ksp[0]));
22289bb5c669SPierre Jolivet                         PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_eps_shell_", pcpre ? pcpre : ""));
22299bb5c669SPierre Jolivet                         PetscCall(KSPSetOptionsPrefix(h->ksp, prefix));
22309bb5c669SPierre Jolivet                         PetscCall(KSPSetFromOptions(h->ksp));
22319bb5c669SPierre Jolivet                         PetscCall(KSPGetPC(h->ksp, &pc));
22329bb5c669SPierre Jolivet                         PetscCall(PCSetType(pc, PCSHELL));
22339bb5c669SPierre Jolivet                         PetscCall(PetscNew(&p));
22349bb5c669SPierre Jolivet                         std::get<0>(*p) = ksp[0];
22359bb5c669SPierre Jolivet                         PetscCall(ISEmbed(ov[0], ov[1], PETSC_TRUE, &std::get<1>(*p)));
22369bb5c669SPierre Jolivet                         PetscCall(MatCreateVecs(A, std::get<2>(*p), std::get<2>(*p) + 1));
22379bb5c669SPierre Jolivet                         PetscCall(PCShellSetContext(pc, p));
22389bb5c669SPierre Jolivet                         PetscCall(PCShellSetApply(pc, PCApply_Schur));
22399bb5c669SPierre Jolivet                         PetscCall(PCShellSetApplyTranspose(pc, PCApply_Schur<Vec, true>));
22409bb5c669SPierre Jolivet                         PetscCall(PCShellSetMatApply(pc, PCApply_Schur<Mat>));
22419bb5c669SPierre Jolivet                         PetscCall(PCShellSetDestroy(pc, PCDestroy_Schur));
22429bb5c669SPierre Jolivet                       }
22439bb5c669SPierre Jolivet                     }
22449bb5c669SPierre Jolivet                     if (!data->share) PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc since neither MUMPS nor MKL PARDISO is used\n"));
22459bb5c669SPierre Jolivet                   }
22469bb5c669SPierre Jolivet                 } while (!share != !data->share); /* if data->share is initially PETSC_TRUE, but then reset to PETSC_FALSE, then go back to the beginning of the do loop */
22479bb5c669SPierre Jolivet               }
22489bb5c669SPierre Jolivet               PetscCall(ISDestroy(ov));
22499bb5c669SPierre Jolivet               PetscCall(ISDestroy(ov + 1));
22509bb5c669SPierre Jolivet               if (overlap == 1 && subdomains && flg) {
22519bb5c669SPierre Jolivet                 *subA = A0;
22529bb5c669SPierre Jolivet                 sub   = subA;
22539bb5c669SPierre Jolivet                 if (uaux) PetscCall(MatDestroy(&uaux));
22549bb5c669SPierre Jolivet               } else PetscCall(MatDestroy(&A0));
22559bb5c669SPierre Jolivet               PetscCall(MatCreateShell(PETSC_COMM_SELF, P->rmap->n, n[1] - n[0], P->rmap->n, n[1] - n[0], h, &data->aux));
2256bdcd51b8SPierre Jolivet               PetscCall(KSPSetErrorIfNotConverged(h->ksp, PETSC_TRUE)); /* bail out as early as possible to avoid (apparently) unrelated error messages */
22579bb5c669SPierre Jolivet               PetscCall(MatCreateVecs(h->ksp->pc->pmat, &h->v, nullptr));
22589bb5c669SPierre Jolivet               PetscCall(MatShellSetOperation(data->aux, MATOP_MULT, (void (*)(void))MatMult_Harmonic));
22599bb5c669SPierre Jolivet               PetscCall(MatShellSetOperation(data->aux, MATOP_MULT_TRANSPOSE, (void (*)(void))MatMultTranspose_Harmonic));
22609bb5c669SPierre Jolivet               PetscCall(MatShellSetMatProductOperation(data->aux, MATPRODUCT_AB, nullptr, MatProduct_AB_Harmonic, nullptr, MATDENSE, MATDENSE));
22619bb5c669SPierre Jolivet               PetscCall(MatShellSetOperation(data->aux, MATOP_DESTROY, (void (*)(void))MatDestroy_Harmonic));
22629bb5c669SPierre Jolivet               PetscCall(MatDestroySubMatrices(1, &a));
22639bb5c669SPierre Jolivet             }
2264df21ef68SPierre Jolivet             if (overlap != 1 || !subdomains) {
2265df21ef68SPierre Jolivet               PetscCall(MatCreateSubMatrices(uaux ? uaux : P, 1, is, is, MAT_INITIAL_MATRIX, &sub));
2266df21ef68SPierre Jolivet               if (ismatis) {
2267df21ef68SPierre Jolivet                 PetscCall(MatISGetLocalMat(C, &N));
2268df21ef68SPierre Jolivet                 PetscCall(PetscObjectTypeCompare((PetscObject)N, MATSEQSBAIJ, &flg));
2269df21ef68SPierre Jolivet                 if (flg) PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub));
2270df21ef68SPierre Jolivet                 PetscCall(MatISRestoreLocalMat(C, &N));
2271df21ef68SPierre Jolivet               }
2272df21ef68SPierre Jolivet             }
22739bb5c669SPierre Jolivet             if (uaux) {
22749bb5c669SPierre Jolivet               PetscCall(MatDestroy(&uaux));
22759bb5c669SPierre Jolivet               PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub));
22769bb5c669SPierre Jolivet             }
22779bb5c669SPierre Jolivet           }
227813044ca3SPierre Jolivet         }
2279f1580f4eSBarry Smith       } else {
2280f1580f4eSBarry Smith         PetscCall(MatCreateSubMatrices(uaux, 1, is, is, MAT_INITIAL_MATRIX, &sub));
2281f1580f4eSBarry Smith         PetscCall(MatDestroy(&uaux));
2282f1580f4eSBarry Smith         PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub));
2283f1580f4eSBarry Smith       }
2284f1580f4eSBarry Smith       /* Vec holding the partition of unity */
2285f1580f4eSBarry Smith       if (!data->levels[0]->D) {
2286f1580f4eSBarry Smith         PetscCall(ISGetLocalSize(data->is, &n));
2287f1580f4eSBarry Smith         PetscCall(VecCreateMPI(PETSC_COMM_SELF, n, PETSC_DETERMINE, &data->levels[0]->D));
2288f1580f4eSBarry Smith       }
22899bb5c669SPierre Jolivet       if (data->share && overlap == -1) {
2290f1580f4eSBarry Smith         Mat      D;
2291db4a47b3SPierre Jolivet         IS       perm = nullptr;
2292f1580f4eSBarry Smith         PetscInt size = -1;
2293811e8887SPierre Jolivet 
2294398c7888SPierre Jolivet         if (!data->levels[0]->pc) {
2295398c7888SPierre Jolivet           PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : ""));
2296398c7888SPierre Jolivet           PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->pc));
2297398c7888SPierre Jolivet           PetscCall(PCSetOptionsPrefix(data->levels[0]->pc, prefix));
2298398c7888SPierre Jolivet           PetscCall(PCSetOperators(data->levels[0]->pc, A, P));
2299398c7888SPierre Jolivet         }
2300398c7888SPierre Jolivet         PetscCall(PCSetType(data->levels[0]->pc, PCASM));
230113044ca3SPierre Jolivet         if (!ctx) {
2302398c7888SPierre Jolivet           if (!data->levels[0]->pc->setupcalled) {
2303398c7888SPierre Jolivet             IS sorted; /* PCASM will sort the input IS, duplicate it to return an unmodified (PCHPDDM) input IS */
2304398c7888SPierre Jolivet             PetscCall(ISDuplicate(is[0], &sorted));
2305398c7888SPierre Jolivet             PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, &sorted, &loc));
2306398c7888SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)sorted));
2307398c7888SPierre Jolivet           }
2308398c7888SPierre Jolivet           PetscCall(PCSetFromOptions(data->levels[0]->pc));
2309398c7888SPierre Jolivet           if (block) {
2310398c7888SPierre Jolivet             PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, sub[0], &C, &perm));
2311398c7888SPierre Jolivet             PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(data->levels[0]->pc, C, algebraic));
2312398c7888SPierre Jolivet           } else PetscCall(PCSetUp(data->levels[0]->pc));
2313db4a47b3SPierre Jolivet           PetscTryMethod(data->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data->levels[0]->pc, &size, nullptr, &ksp));
2314398c7888SPierre Jolivet           if (size != 1) {
2315398c7888SPierre Jolivet             data->share = PETSC_FALSE;
2316398c7888SPierre Jolivet             PetscCheck(size == -1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", size);
2317398c7888SPierre Jolivet             PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc since PCASMGetSubKSP() not found in fine-level PC\n"));
2318398c7888SPierre Jolivet             PetscCall(ISDestroy(&unsorted));
2319398c7888SPierre Jolivet             unsorted = is[0];
2320398c7888SPierre Jolivet           } else {
2321811e8887SPierre Jolivet             const char *matpre;
2322811e8887SPierre Jolivet             PetscBool   cmp[4];
2323811e8887SPierre Jolivet 
232413044ca3SPierre Jolivet             if (!block && !ctx) PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, PetscBool3ToBool(data->Neumann) ? sub[0] : data->aux, &C, &perm));
2325c8ea6600SPierre Jolivet             if (!PetscBool3ToBool(data->Neumann) && !block) {
2326f1580f4eSBarry Smith               PetscCall(MatPermute(sub[0], perm, perm, &D)); /* permute since PCASM will call ISSort() */
2327f1580f4eSBarry Smith               PetscCall(MatHeaderReplace(sub[0], &D));
2328f1580f4eSBarry Smith             }
2329f1580f4eSBarry Smith             if (data->B) { /* see PCHPDDMSetRHSMat() */
2330f1580f4eSBarry Smith               PetscCall(MatPermute(data->B, perm, perm, &D));
2331f1580f4eSBarry Smith               PetscCall(MatHeaderReplace(data->B, &D));
2332f1580f4eSBarry Smith             }
2333f1580f4eSBarry Smith             PetscCall(ISDestroy(&perm));
2334f1580f4eSBarry Smith             PetscCall(KSPGetOperators(ksp[0], subA, subA + 1));
23350307214fSPierre Jolivet             PetscCall(PetscObjectReference((PetscObject)subA[0]));
2336f1580f4eSBarry Smith             PetscCall(MatDuplicate(subA[1], MAT_SHARE_NONZERO_PATTERN, &D));
2337f1580f4eSBarry Smith             PetscCall(MatGetOptionsPrefix(subA[1], &matpre));
2338f1580f4eSBarry Smith             PetscCall(MatSetOptionsPrefix(D, matpre));
2339f1580f4eSBarry Smith             PetscCall(PetscObjectTypeCompare((PetscObject)D, MATNORMAL, cmp));
2340f1580f4eSBarry Smith             PetscCall(PetscObjectTypeCompare((PetscObject)C, MATNORMAL, cmp + 1));
2341f1580f4eSBarry Smith             if (!cmp[0]) PetscCall(PetscObjectTypeCompare((PetscObject)D, MATNORMALHERMITIAN, cmp + 2));
2342f1580f4eSBarry Smith             else cmp[2] = PETSC_FALSE;
2343f1580f4eSBarry Smith             if (!cmp[1]) PetscCall(PetscObjectTypeCompare((PetscObject)C, MATNORMALHERMITIAN, cmp + 3));
2344f1580f4eSBarry Smith             else cmp[3] = PETSC_FALSE;
2345f1580f4eSBarry Smith             PetscCheck(cmp[0] == cmp[1] && cmp[2] == cmp[3], PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "-%spc_hpddm_levels_1_pc_asm_sub_mat_type %s and auxiliary Mat of type %s", pcpre ? pcpre : "", ((PetscObject)D)->type_name, ((PetscObject)C)->type_name);
2346f1580f4eSBarry Smith             if (!cmp[0] && !cmp[2]) {
2347f1580f4eSBarry Smith               if (!block) PetscCall(MatAXPY(D, 1.0, C, SUBSET_NONZERO_PATTERN));
23485c7345deSPierre Jolivet               else {
2349db4a47b3SPierre Jolivet                 PetscCall(MatMissingDiagonal(D, cmp, nullptr));
23505c7345deSPierre Jolivet                 if (cmp[0]) structure = DIFFERENT_NONZERO_PATTERN; /* data->aux has no missing diagonal entry */
23515c7345deSPierre Jolivet                 PetscCall(MatAXPY(D, 1.0, data->aux, structure));
23525c7345deSPierre Jolivet               }
2353f1580f4eSBarry Smith             } else {
2354f1580f4eSBarry Smith               Mat mat[2];
2355811e8887SPierre Jolivet 
2356f1580f4eSBarry Smith               if (cmp[0]) {
2357f1580f4eSBarry Smith                 PetscCall(MatNormalGetMat(D, mat));
2358f1580f4eSBarry Smith                 PetscCall(MatNormalGetMat(C, mat + 1));
2359f1580f4eSBarry Smith               } else {
2360f1580f4eSBarry Smith                 PetscCall(MatNormalHermitianGetMat(D, mat));
2361f1580f4eSBarry Smith                 PetscCall(MatNormalHermitianGetMat(C, mat + 1));
2362f1580f4eSBarry Smith               }
2363f1580f4eSBarry Smith               PetscCall(MatAXPY(mat[0], 1.0, mat[1], SUBSET_NONZERO_PATTERN));
2364f1580f4eSBarry Smith             }
2365f1580f4eSBarry Smith             PetscCall(MatPropagateSymmetryOptions(C, D));
2366f1580f4eSBarry Smith             PetscCall(MatDestroy(&C));
2367f1580f4eSBarry Smith             C = D;
2368f1580f4eSBarry Smith             /* swap pointers so that variables stay consistent throughout PCSetUp() */
2369f1580f4eSBarry Smith             std::swap(C, data->aux);
2370f1580f4eSBarry Smith             std::swap(uis, data->is);
2371f1580f4eSBarry Smith             swap = PETSC_TRUE;
2372f1580f4eSBarry Smith           }
2373f1580f4eSBarry Smith         }
237413044ca3SPierre Jolivet       }
237513044ca3SPierre Jolivet       if (ctx) {
237613044ca3SPierre Jolivet         PC_HPDDM              *data_00 = (PC_HPDDM *)std::get<0>(*ctx)[0]->data;
237713044ca3SPierre Jolivet         PC                     s;
237800b491fbSPierre Jolivet         Mat                    A00, P00, A01 = nullptr, A10, A11, N, b[4];
237907d8d47fSPierre Jolivet         IS                     sorted, is[2], *is_00;
238013044ca3SPierre Jolivet         MatSolverType          type;
238113044ca3SPierre Jolivet         std::pair<PC, Vec[2]> *p;
238213044ca3SPierre Jolivet 
238307d8d47fSPierre Jolivet         n = -1;
238407d8d47fSPierre Jolivet         PetscTryMethod(data_00->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data_00->levels[0]->pc, &n, nullptr, &ksp));
238507d8d47fSPierre Jolivet         PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", n);
238607d8d47fSPierre Jolivet         PetscCall(KSPGetOperators(ksp[0], subA, subA + 1));
238707d8d47fSPierre Jolivet         PetscCall(ISGetLocalSize(data_00->is, &n));
238807d8d47fSPierre Jolivet         if (n != subA[0]->rmap->n || n != subA[0]->cmap->n) {
238907d8d47fSPierre Jolivet           PetscCall(PCASMGetLocalSubdomains(data_00->levels[0]->pc, &n, &is_00, nullptr));
239007d8d47fSPierre Jolivet           PetscCall(ISGetLocalSize(*is_00, &n));
239107d8d47fSPierre Jolivet           PetscCheck(n == subA[0]->rmap->n && n == subA[0]->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "-%spc_hpddm_schur_precondition geneo and -%spc_hpddm_define_subdomains false", pcpre ? pcpre : "", ((PetscObject)pc)->prefix);
239207d8d47fSPierre Jolivet         } else is_00 = &data_00->is;
239313044ca3SPierre Jolivet         PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, data->aux, &C, nullptr)); /* permute since PCASM works with a sorted IS */
239413044ca3SPierre Jolivet         std::swap(C, data->aux);
239513044ca3SPierre Jolivet         std::swap(uis, data->is);
239613044ca3SPierre Jolivet         swap = PETSC_TRUE;
239713044ca3SPierre Jolivet         PetscCall(PCASMSetType(data->levels[0]->pc, PC_ASM_NONE)); /* "Neumann--Neumann" preconditioning with overlap and a Boolean partition of unity */
239813044ca3SPierre Jolivet         PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, &data->is, &loc));
239913044ca3SPierre Jolivet         PetscCall(PCSetFromOptions(data->levels[0]->pc)); /* action of eq. (15) of https://hal.science/hal-02343808v6/document (with a sign flip) */
240013044ca3SPierre Jolivet         PetscCall(MatSchurComplementGetSubMatrices(P, &A00, &P00, std::get<1>(*ctx), &A10, &A11));
240113044ca3SPierre Jolivet         std::get<1>(*ctx)[1] = A10;
240213044ca3SPierre Jolivet         PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, &flg));
240313044ca3SPierre Jolivet         if (flg) PetscCall(MatTransposeGetMat(A10, &A01));
240413044ca3SPierre Jolivet         else {
240513044ca3SPierre Jolivet           PetscBool flg;
240613044ca3SPierre Jolivet 
240713044ca3SPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, &flg));
240813044ca3SPierre Jolivet           if (flg) PetscCall(MatHermitianTransposeGetMat(A10, &A01));
240913044ca3SPierre Jolivet         }
241007d8d47fSPierre Jolivet         PetscCall(ISDuplicate(*is_00, &sorted)); /* during setup of the PC associated to the A00 block, this IS has already been sorted, but it's put back to its original state at the end of PCSetUp_HPDDM(), which may be unsorted */
241113044ca3SPierre Jolivet         PetscCall(ISSort(sorted));               /* this is to avoid changing users inputs, but it requires a new call to ISSort() here                                                                                               */
241200b491fbSPierre Jolivet         if (!A01) {
241300b491fbSPierre Jolivet           PetscCall(MatSetOption(A10, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
241400b491fbSPierre Jolivet           PetscCall(MatCreateSubMatrices(A10, 1, &data->is, &sorted, MAT_INITIAL_MATRIX, &sub));
241500b491fbSPierre Jolivet           b[2] = sub[0];
241600b491fbSPierre Jolivet           PetscCall(PetscObjectReference((PetscObject)sub[0]));
241700b491fbSPierre Jolivet           PetscCall(MatDestroySubMatrices(1, &sub));
241800b491fbSPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)std::get<1>(*ctx)[0], MATTRANSPOSEVIRTUAL, &flg));
241900b491fbSPierre Jolivet           A10 = nullptr;
242000b491fbSPierre Jolivet           if (flg) PetscCall(MatTransposeGetMat(std::get<1>(*ctx)[0], &A10));
242113044ca3SPierre Jolivet           else {
242200b491fbSPierre Jolivet             PetscBool flg;
242300b491fbSPierre Jolivet 
2424c71f06a7SPierre Jolivet             PetscCall(PetscObjectTypeCompare((PetscObject)std::get<1>(*ctx)[0], MATHERMITIANTRANSPOSEVIRTUAL, &flg));
242500b491fbSPierre Jolivet             if (flg) PetscCall(MatHermitianTransposeGetMat(std::get<1>(*ctx)[0], &A10));
242613044ca3SPierre Jolivet           }
2427811e8887SPierre Jolivet           if (!A10) PetscCall(MatCreateSubMatrices(std::get<1>(*ctx)[0], 1, &sorted, &data->is, MAT_INITIAL_MATRIX, &sub));
2428811e8887SPierre Jolivet           else {
2429811e8887SPierre Jolivet             if (flg) PetscCall(MatCreateTranspose(b[2], b + 1));
2430811e8887SPierre Jolivet             else PetscCall(MatCreateHermitianTranspose(b[2], b + 1));
243100b491fbSPierre Jolivet           }
243200b491fbSPierre Jolivet         } else {
243300b491fbSPierre Jolivet           PetscCall(MatSetOption(A01, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
243400b491fbSPierre Jolivet           PetscCall(MatCreateSubMatrices(A01, 1, &sorted, &data->is, MAT_INITIAL_MATRIX, &sub));
2435811e8887SPierre Jolivet           if (flg) PetscCall(MatCreateTranspose(*sub, b + 2));
2436811e8887SPierre Jolivet           else PetscCall(MatCreateHermitianTranspose(*sub, b + 2));
2437811e8887SPierre Jolivet         }
2438811e8887SPierre Jolivet         if (A01 || !A10) {
2439811e8887SPierre Jolivet           b[1] = sub[0];
2440811e8887SPierre Jolivet           PetscCall(PetscObjectReference((PetscObject)sub[0]));
244100b491fbSPierre Jolivet         }
244200b491fbSPierre Jolivet         PetscCall(MatDestroySubMatrices(1, &sub));
244313044ca3SPierre Jolivet         PetscCall(ISDestroy(&sorted));
244400b491fbSPierre Jolivet         PetscCall(MatCreateSchurComplement(subA[0], subA[1], b[1], b[2], data->aux, &S));
244513044ca3SPierre Jolivet         PetscCall(MatSchurComplementSetKSP(S, ksp[0]));
244613044ca3SPierre Jolivet         PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(data->levels[0]->pc, S, PETSC_TRUE)); /* the subdomain Mat is already known and the input IS of PCASMSetLocalSubdomains() is already sorted */
244713044ca3SPierre Jolivet         PetscTryMethod(data->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data->levels[0]->pc, &n, nullptr, &ksp));
244813044ca3SPierre Jolivet         PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", n);
244913044ca3SPierre Jolivet         PetscCall(KSPGetPC(ksp[0], &inner));
245013044ca3SPierre Jolivet         PetscCall(PCSetType(inner, PCSHELL)); /* compute the action of the inverse of the local Schur complement with a PCSHELL */
245113044ca3SPierre Jolivet         b[0] = subA[0];
245213044ca3SPierre Jolivet         b[3] = data->aux;
245313044ca3SPierre Jolivet         PetscCall(MatCreateNest(PETSC_COMM_SELF, 2, nullptr, 2, nullptr, b, &N)); /* instead of computing inv(A11 - A10 inv(A00) A01), compute inv([A00, A01; A10, A11]) followed by a partial solution associated to the A11 block */
245400b491fbSPierre Jolivet         PetscCall(PetscObjectDereference((PetscObject)b[1]));
245500b491fbSPierre Jolivet         PetscCall(PetscObjectDereference((PetscObject)b[2]));
245613044ca3SPierre Jolivet         PetscCall(PCCreate(PETSC_COMM_SELF, &s));
245713044ca3SPierre Jolivet         PetscCall(PCSetOptionsPrefix(s, ((PetscObject)inner)->prefix));
245813044ca3SPierre Jolivet         PetscCall(PCSetOptionsPrefix(inner, nullptr));
245913044ca3SPierre Jolivet         PetscCall(KSPSetSkipPCSetFromOptions(ksp[0], PETSC_TRUE));
246013044ca3SPierre Jolivet         PetscCall(PCSetType(s, PCLU));
246113044ca3SPierre Jolivet         if (PetscDefined(HAVE_MUMPS)) { /* only MATSOLVERMUMPS handles MATNEST, so for the others, e.g., MATSOLVERPETSC or MATSOLVERMKL_PARDISO, convert to plain MATAIJ */
246213044ca3SPierre Jolivet           PetscCall(PCFactorSetMatSolverType(s, MATSOLVERMUMPS));
246313044ca3SPierre Jolivet         }
246413044ca3SPierre Jolivet         PetscCall(PCSetFromOptions(s));
246513044ca3SPierre Jolivet         PetscCall(PCFactorGetMatSolverType(s, &type));
246613044ca3SPierre Jolivet         PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg));
246713044ca3SPierre Jolivet         if (flg) {
246813044ca3SPierre Jolivet           PetscCall(PCSetOperators(s, N, N));
246900b491fbSPierre Jolivet           PetscCall(PCFactorGetMatrix(s, b));
247000b491fbSPierre Jolivet           PetscCall(MatSetOptionsPrefix(*b, ((PetscObject)s)->prefix));
247124ddd604SPierre Jolivet           n = -1;
247224ddd604SPierre Jolivet           PetscCall(PetscOptionsGetInt(nullptr, ((PetscObject)s)->prefix, "-mat_mumps_icntl_26", &n, nullptr));
2473f45b553cSPierre Jolivet           if (n == 1) {                                /* allocates a square MatDense of size is[1]->map->n, so one */
2474f45b553cSPierre Jolivet             PetscCall(MatNestGetISs(N, is, nullptr));  /*  needs to be able to deactivate this path when dealing    */
2475f45b553cSPierre Jolivet             PetscCall(MatFactorSetSchurIS(*b, is[1])); /*  with a large constraint space in order to avoid OOM      */
2476f45b553cSPierre Jolivet           }
247713044ca3SPierre Jolivet         } else {
247800b491fbSPierre Jolivet           PetscCall(MatConvert(N, MATAIJ, MAT_INITIAL_MATRIX, b));
247900b491fbSPierre Jolivet           PetscCall(PCSetOperators(s, N, *b));
248000b491fbSPierre Jolivet           PetscCall(PetscObjectDereference((PetscObject)*b));
248100b491fbSPierre Jolivet           PetscCall(PCFactorGetMatrix(s, b)); /* MATSOLVERMKL_PARDISO cannot compute in PETSc (yet) a partial solution associated to the A11 block, only partial solution associated to the A00 block or full solution */
248213044ca3SPierre Jolivet         }
248313044ca3SPierre Jolivet         PetscCall(PetscNew(&p));
248413044ca3SPierre Jolivet         p->first = s;
248500b491fbSPierre Jolivet         PetscCall(MatCreateVecs(*b, p->second, p->second + 1));
248613044ca3SPierre Jolivet         PetscCall(PCShellSetContext(inner, p));
248713044ca3SPierre Jolivet         PetscCall(PCShellSetApply(inner, PCApply_Nest));
248813044ca3SPierre Jolivet         PetscCall(PCShellSetView(inner, PCView_Nest));
248913044ca3SPierre Jolivet         PetscCall(PCShellSetDestroy(inner, PCDestroy_Nest));
249013044ca3SPierre Jolivet         PetscCall(PetscObjectDereference((PetscObject)N));
249113044ca3SPierre Jolivet       }
2492f1580f4eSBarry Smith       if (!data->levels[0]->scatter) {
2493db4a47b3SPierre Jolivet         PetscCall(MatCreateVecs(P, &xin, nullptr));
2494f1580f4eSBarry Smith         if (ismatis) PetscCall(MatDestroy(&P));
2495db4a47b3SPierre Jolivet         PetscCall(VecScatterCreate(xin, data->is, data->levels[0]->D, nullptr, &data->levels[0]->scatter));
2496f1580f4eSBarry Smith         PetscCall(VecDestroy(&xin));
2497f1580f4eSBarry Smith       }
2498f1580f4eSBarry Smith       if (data->levels[0]->P) {
2499f1580f4eSBarry Smith         /* if the pattern is the same and PCSetUp() has previously succeeded, reuse HPDDM buffers and connectivity */
2500f1580f4eSBarry Smith         PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(data->levels[0], pc->setupcalled < 1 || pc->flag == DIFFERENT_NONZERO_PATTERN ? PETSC_TRUE : PETSC_FALSE));
2501f1580f4eSBarry Smith       }
2502f1580f4eSBarry Smith       if (!data->levels[0]->P) data->levels[0]->P = new HPDDM::Schwarz<PetscScalar>();
2503db4a47b3SPierre Jolivet       if (data->log_separate) PetscCall(PetscLogEventBegin(PC_HPDDM_SetUp[0], data->levels[0]->ksp, nullptr, nullptr, nullptr));
2504db4a47b3SPierre Jolivet       else PetscCall(PetscLogEventBegin(PC_HPDDM_Strc, data->levels[0]->ksp, nullptr, nullptr, nullptr));
2505f1580f4eSBarry Smith       /* HPDDM internal data structure */
250613044ca3SPierre Jolivet       PetscCall(data->levels[0]->P->structure(loc, data->is, !ctx ? sub[0] : nullptr, ismatis ? C : data->aux, data->levels));
2507db4a47b3SPierre Jolivet       if (!data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Strc, data->levels[0]->ksp, nullptr, nullptr, nullptr));
2508f1580f4eSBarry Smith       /* matrix pencil of the generalized eigenvalue problem on the overlap (GenEO) */
250913044ca3SPierre Jolivet       if (!ctx) {
25109bb5c669SPierre Jolivet         if (data->deflation || overlap != -1) weighted = data->aux;
2511f1580f4eSBarry Smith         else if (!data->B) {
2512e2b46ddfSPierre Jolivet           PetscBool cmp;
2513811e8887SPierre Jolivet 
2514f1580f4eSBarry Smith           PetscCall(MatDuplicate(sub[0], MAT_COPY_VALUES, &weighted));
2515e2b46ddfSPierre Jolivet           PetscCall(PetscObjectTypeCompareAny((PetscObject)weighted, &cmp, MATNORMAL, MATNORMALHERMITIAN, ""));
2516e2b46ddfSPierre Jolivet           if (cmp) flg = PETSC_FALSE;
2517e2b46ddfSPierre Jolivet           PetscCall(MatDiagonalScale(weighted, data->levels[0]->D, data->levels[0]->D));
25185b0bb9f2SPierre Jolivet           /* neither MatDuplicate() nor MatDiagonalScale() handles the symmetry options, so propagate the options explicitly */
2519f1580f4eSBarry Smith           /* only useful for -mat_type baij -pc_hpddm_levels_1_st_pc_type cholesky (no problem with MATAIJ or MATSBAIJ)      */
2520f1580f4eSBarry Smith           PetscCall(MatPropagateSymmetryOptions(sub[0], weighted));
2521f1580f4eSBarry Smith         } else weighted = data->B;
252213044ca3SPierre Jolivet       } else weighted = nullptr;
2523f1580f4eSBarry Smith       /* SLEPc is used inside the loaded symbol */
25249bb5c669SPierre Jolivet       PetscCall((*loadedSym)(data->levels[0]->P, data->is, ismatis ? C : (algebraic && !block && overlap == -1 ? sub[0] : (!ctx ? data->aux : S)), weighted, data->B, initial, data->levels));
25259bb5c669SPierre Jolivet       if (!ctx && data->share && overlap == -1) {
2526f1580f4eSBarry Smith         Mat st[2];
2527f1580f4eSBarry Smith         PetscCall(KSPGetOperators(ksp[0], st, st + 1));
25285c7345deSPierre Jolivet         PetscCall(MatCopy(subA[0], st[0], structure));
2529f1580f4eSBarry Smith         if (subA[1] != subA[0] || st[1] != st[0]) PetscCall(MatCopy(subA[1], st[1], SAME_NONZERO_PATTERN));
25300307214fSPierre Jolivet         PetscCall(PetscObjectDereference((PetscObject)subA[0]));
2531f1580f4eSBarry Smith       }
2532db4a47b3SPierre Jolivet       if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_SetUp[0], data->levels[0]->ksp, nullptr, nullptr, nullptr));
2533f1580f4eSBarry Smith       if (ismatis) PetscCall(MatISGetLocalMat(C, &N));
2534f1580f4eSBarry Smith       else N = data->aux;
253513044ca3SPierre Jolivet       if (!ctx) P = sub[0];
253613044ca3SPierre Jolivet       else P = S;
2537f1580f4eSBarry Smith       /* going through the grid hierarchy */
2538f1580f4eSBarry Smith       for (n = 1; n < data->N; ++n) {
2539db4a47b3SPierre Jolivet         if (data->log_separate) PetscCall(PetscLogEventBegin(PC_HPDDM_SetUp[n], data->levels[n]->ksp, nullptr, nullptr, nullptr));
2540f1580f4eSBarry Smith         /* method composed in the loaded symbol since there, SLEPc is used as well */
2541f1580f4eSBarry Smith         PetscTryMethod(data->levels[0]->ksp, "PCHPDDMSetUp_Private_C", (Mat *, Mat *, PetscInt, PetscInt *const, PC_HPDDM_Level **const), (&P, &N, n, &data->N, data->levels));
2542db4a47b3SPierre Jolivet         if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_SetUp[n], data->levels[n]->ksp, nullptr, nullptr, nullptr));
2543f1580f4eSBarry Smith       }
2544f1580f4eSBarry Smith       /* reset to NULL to avoid any faulty use */
2545db4a47b3SPierre Jolivet       PetscCall(PetscObjectComposeFunction((PetscObject)data->levels[0]->ksp, "PCHPDDMSetUp_Private_C", nullptr));
2546db4a47b3SPierre Jolivet       if (!ismatis) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_C", nullptr));
2547f1580f4eSBarry Smith       else PetscCall(PetscObjectDereference((PetscObject)C)); /* matching PetscObjectReference() above */
2548f1580f4eSBarry Smith       for (n = 0; n < data->N - 1; ++n)
2549f1580f4eSBarry Smith         if (data->levels[n]->P) {
2550f1580f4eSBarry Smith           /* HPDDM internal work buffers */
2551f1580f4eSBarry Smith           data->levels[n]->P->setBuffer();
2552f1580f4eSBarry Smith           data->levels[n]->P->super::start();
2553f1580f4eSBarry Smith         }
25549bb5c669SPierre Jolivet       if (ismatis || !subdomains) PetscCall(PCHPDDMDestroySubMatrices_Private(PetscBool3ToBool(data->Neumann), PetscBool(algebraic && !block && overlap == -1), sub));
2555db4a47b3SPierre Jolivet       if (ismatis) data->is = nullptr;
2556f1580f4eSBarry Smith       for (n = 0; n < data->N - 1 + (reused > 0); ++n) {
2557f1580f4eSBarry Smith         if (data->levels[n]->P) {
2558f1580f4eSBarry Smith           PC spc;
2559f1580f4eSBarry Smith 
2560f1580f4eSBarry Smith           /* force the PC to be PCSHELL to do the coarse grid corrections */
2561f1580f4eSBarry Smith           PetscCall(KSPSetSkipPCSetFromOptions(data->levels[n]->ksp, PETSC_TRUE));
2562f1580f4eSBarry Smith           PetscCall(KSPGetPC(data->levels[n]->ksp, &spc));
2563f1580f4eSBarry Smith           PetscCall(PCSetType(spc, PCSHELL));
2564f1580f4eSBarry Smith           PetscCall(PCShellSetContext(spc, data->levels[n]));
2565f1580f4eSBarry Smith           PetscCall(PCShellSetSetUp(spc, PCSetUp_HPDDMShell));
2566f1580f4eSBarry Smith           PetscCall(PCShellSetApply(spc, PCApply_HPDDMShell));
2567f1580f4eSBarry Smith           PetscCall(PCShellSetMatApply(spc, PCMatApply_HPDDMShell));
256813044ca3SPierre Jolivet           if (ctx && n == 0) {
256913044ca3SPierre Jolivet             Mat                               Amat, Pmat;
257013044ca3SPierre Jolivet             PetscInt                          m, M;
257101e3c840SPierre Jolivet             std::tuple<Mat, PetscSF, Vec[2]> *ctx;
257213044ca3SPierre Jolivet 
257313044ca3SPierre Jolivet             PetscCall(KSPGetOperators(data->levels[n]->ksp, nullptr, &Pmat));
257413044ca3SPierre Jolivet             PetscCall(MatGetLocalSize(Pmat, &m, nullptr));
257513044ca3SPierre Jolivet             PetscCall(MatGetSize(Pmat, &M, nullptr));
257613044ca3SPierre Jolivet             PetscCall(PetscNew(&ctx));
257713044ca3SPierre Jolivet             std::get<0>(*ctx) = S;
257813044ca3SPierre Jolivet             std::get<1>(*ctx) = data->levels[n]->scatter;
257913044ca3SPierre Jolivet             PetscCall(MatCreateShell(PetscObjectComm((PetscObject)data->levels[n]->ksp), m, m, M, M, ctx, &Amat));
258013044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(Amat, MATOP_MULT, (void (*)(void))MatMult_Schur<false>));
258113044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(Amat, MATOP_MULT_TRANSPOSE, (void (*)(void))MatMult_Schur<true>));
258213044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(Amat, MATOP_DESTROY, (void (*)(void))MatDestroy_Schur));
258313044ca3SPierre Jolivet             PetscCall(MatCreateVecs(S, std::get<2>(*ctx), std::get<2>(*ctx) + 1));
258413044ca3SPierre Jolivet             PetscCall(KSPSetOperators(data->levels[n]->ksp, Amat, Pmat));
258513044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)Amat));
258613044ca3SPierre Jolivet           }
2587f1580f4eSBarry Smith           PetscCall(PCShellSetDestroy(spc, PCDestroy_HPDDMShell));
2588f1580f4eSBarry Smith           if (!data->levels[n]->pc) PetscCall(PCCreate(PetscObjectComm((PetscObject)data->levels[n]->ksp), &data->levels[n]->pc));
2589f1580f4eSBarry Smith           if (n < reused) {
2590f1580f4eSBarry Smith             PetscCall(PCSetReusePreconditioner(spc, PETSC_TRUE));
2591f1580f4eSBarry Smith             PetscCall(PCSetReusePreconditioner(data->levels[n]->pc, PETSC_TRUE));
2592f1580f4eSBarry Smith           }
2593f1580f4eSBarry Smith           PetscCall(PCSetUp(spc));
2594f1580f4eSBarry Smith         }
2595f1580f4eSBarry Smith       }
259613044ca3SPierre Jolivet       if (ctx) PetscCall(MatDestroy(&S));
25979bb5c669SPierre Jolivet       if (overlap == -1) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", nullptr));
2598f1580f4eSBarry Smith     } else flg = reused ? PETSC_FALSE : PETSC_TRUE;
2599f1580f4eSBarry Smith     if (!ismatis && subdomains) {
2600f1580f4eSBarry Smith       if (flg) PetscCall(KSPGetPC(data->levels[0]->ksp, &inner));
2601f1580f4eSBarry Smith       else inner = data->levels[0]->pc;
2602f1580f4eSBarry Smith       if (inner) {
2603398c7888SPierre Jolivet         if (!inner->setupcalled) PetscCall(PCSetType(inner, PCASM));
2604398c7888SPierre Jolivet         PetscCall(PCSetFromOptions(inner));
2605398c7888SPierre Jolivet         PetscCall(PetscStrcmp(((PetscObject)inner)->type_name, PCASM, &flg));
2606398c7888SPierre Jolivet         if (flg) {
2607f1580f4eSBarry Smith           if (!inner->setupcalled) { /* evaluates to PETSC_FALSE when -pc_hpddm_block_splitting */
2608398c7888SPierre Jolivet             IS sorted;               /* PCASM will sort the input IS, duplicate it to return an unmodified (PCHPDDM) input IS */
2609811e8887SPierre Jolivet 
2610398c7888SPierre Jolivet             PetscCall(ISDuplicate(is[0], &sorted));
2611398c7888SPierre Jolivet             PetscCall(PCASMSetLocalSubdomains(inner, 1, &sorted, &loc));
2612398c7888SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)sorted));
2613398c7888SPierre Jolivet           }
2614c8ea6600SPierre Jolivet           if (!PetscBool3ToBool(data->Neumann) && data->N > 1) { /* subdomain matrices are already created for the eigenproblem, reuse them for the fine-level PC */
2615db4a47b3SPierre Jolivet             PetscCall(PCHPDDMPermute_Private(*is, nullptr, nullptr, sub[0], &P, nullptr));
2616398c7888SPierre Jolivet             PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(inner, P, algebraic));
2617398c7888SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)P));
2618f1580f4eSBarry Smith           }
2619f1580f4eSBarry Smith         }
2620f1580f4eSBarry Smith       }
26219bb5c669SPierre Jolivet       if (data->N > 1) {
26229bb5c669SPierre Jolivet         if (overlap != 1) PetscCall(PCHPDDMDestroySubMatrices_Private(PetscBool3ToBool(data->Neumann), PetscBool(algebraic && !block && overlap == -1), sub));
26239bb5c669SPierre Jolivet         if (overlap == 1) PetscCall(MatDestroy(subA));
26249bb5c669SPierre Jolivet       }
2625f1580f4eSBarry Smith     }
2626f1580f4eSBarry Smith     PetscCall(ISDestroy(&loc));
2627f1580f4eSBarry Smith   } else data->N = 1 + reused; /* enforce this value to 1 + reused if there is no way to build another level */
2628f1580f4eSBarry Smith   if (requested != data->N + reused) {
2629f1580f4eSBarry Smith     PetscCall(PetscInfo(pc, "%" PetscInt_FMT " levels requested, only %" PetscInt_FMT " built + %" PetscInt_FMT " reused. Options for level(s) > %" PetscInt_FMT ", including -%spc_hpddm_coarse_ will not be taken into account\n", requested, data->N, reused,
2630f1580f4eSBarry Smith                         data->N, pcpre ? pcpre : ""));
2631f1580f4eSBarry Smith     PetscCall(PetscInfo(pc, "It is best to tune parameters, e.g., a higher value for -%spc_hpddm_levels_%" PetscInt_FMT "_eps_threshold so that at least one local deflation vector will be selected\n", pcpre ? pcpre : "", data->N));
2632f1580f4eSBarry Smith     /* cannot use PCDestroy_HPDDMShell() because PCSHELL not set for unassembled levels */
2633f1580f4eSBarry Smith     for (n = data->N - 1; n < requested - 1; ++n) {
2634f1580f4eSBarry Smith       if (data->levels[n]->P) {
2635f1580f4eSBarry Smith         PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(data->levels[n], PETSC_TRUE));
2636f1580f4eSBarry Smith         PetscCall(VecDestroyVecs(1, &data->levels[n]->v[0]));
2637f1580f4eSBarry Smith         PetscCall(VecDestroyVecs(2, &data->levels[n]->v[1]));
2638f1580f4eSBarry Smith         PetscCall(MatDestroy(data->levels[n]->V));
2639f1580f4eSBarry Smith         PetscCall(MatDestroy(data->levels[n]->V + 1));
2640f1580f4eSBarry Smith         PetscCall(MatDestroy(data->levels[n]->V + 2));
2641f1580f4eSBarry Smith         PetscCall(VecDestroy(&data->levels[n]->D));
264201e3c840SPierre Jolivet         PetscCall(PetscSFDestroy(&data->levels[n]->scatter));
2643f1580f4eSBarry Smith       }
2644f1580f4eSBarry Smith     }
2645f1580f4eSBarry Smith     if (reused) {
2646f1580f4eSBarry Smith       for (n = reused; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) {
2647f1580f4eSBarry Smith         PetscCall(KSPDestroy(&data->levels[n]->ksp));
2648f1580f4eSBarry Smith         PetscCall(PCDestroy(&data->levels[n]->pc));
2649f1580f4eSBarry Smith       }
2650f1580f4eSBarry Smith     }
2651f1580f4eSBarry Smith     PetscCheck(!PetscDefined(USE_DEBUG), PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONG, "%" PetscInt_FMT " levels requested, only %" PetscInt_FMT " built + %" PetscInt_FMT " reused. Options for level(s) > %" PetscInt_FMT ", including -%spc_hpddm_coarse_ will not be taken into account. It is best to tune parameters, e.g., a higher value for -%spc_hpddm_levels_%" PetscInt_FMT "_eps_threshold so that at least one local deflation vector will be selected. If you don't want this to error out, compile --with-debugging=0", requested,
2652f1580f4eSBarry Smith                data->N, reused, data->N, pcpre ? pcpre : "", pcpre ? pcpre : "", data->N);
2653f1580f4eSBarry Smith   }
2654f1580f4eSBarry Smith   /* these solvers are created after PCSetFromOptions() is called */
2655f1580f4eSBarry Smith   if (pc->setfromoptionscalled) {
2656f1580f4eSBarry Smith     for (n = 0; n < data->N; ++n) {
2657f1580f4eSBarry Smith       if (data->levels[n]->ksp) PetscCall(KSPSetFromOptions(data->levels[n]->ksp));
2658f1580f4eSBarry Smith       if (data->levels[n]->pc) PetscCall(PCSetFromOptions(data->levels[n]->pc));
2659f1580f4eSBarry Smith     }
2660f1580f4eSBarry Smith     pc->setfromoptionscalled = 0;
2661f1580f4eSBarry Smith   }
2662f1580f4eSBarry Smith   data->N += reused;
2663f1580f4eSBarry Smith   if (data->share && swap) {
2664f1580f4eSBarry Smith     /* swap back pointers so that variables follow the user-provided numbering */
2665f1580f4eSBarry Smith     std::swap(C, data->aux);
2666f1580f4eSBarry Smith     std::swap(uis, data->is);
2667f1580f4eSBarry Smith     PetscCall(MatDestroy(&C));
2668f1580f4eSBarry Smith     PetscCall(ISDestroy(&uis));
2669f1580f4eSBarry Smith   }
2670398c7888SPierre Jolivet   if (algebraic) PetscCall(MatDestroy(&data->aux));
267138fb34a1SPierre Jolivet   if (unsorted && unsorted != is[0]) {
2672398c7888SPierre Jolivet     PetscCall(ISCopy(unsorted, data->is));
2673398c7888SPierre Jolivet     PetscCall(ISDestroy(&unsorted));
2674398c7888SPierre Jolivet   }
2675398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG)
2676398c7888SPierre Jolivet   PetscCheck((data->is && dis) || (!data->is && !dis), PETSC_COMM_SELF, PETSC_ERR_PLIB, "An IS pointer is NULL but not the other: input IS pointer (%p) v. output IS pointer (%p)", (void *)dis, (void *)data->is);
2677398c7888SPierre Jolivet   if (data->is) {
2678398c7888SPierre Jolivet     PetscCall(ISEqualUnsorted(data->is, dis, &flg));
2679398c7888SPierre Jolivet     PetscCall(ISDestroy(&dis));
2680398c7888SPierre Jolivet     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Input IS and output IS are not equal");
2681398c7888SPierre Jolivet   }
2682398c7888SPierre Jolivet   PetscCheck((data->aux && daux) || (!data->aux && !daux), PETSC_COMM_SELF, PETSC_ERR_PLIB, "A Mat pointer is NULL but not the other: input Mat pointer (%p) v. output Mat pointer (%p)", (void *)daux, (void *)data->aux);
2683398c7888SPierre Jolivet   if (data->aux) {
2684398c7888SPierre Jolivet     PetscCall(MatMultEqual(data->aux, daux, 20, &flg));
2685398c7888SPierre Jolivet     PetscCall(MatDestroy(&daux));
2686398c7888SPierre Jolivet     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Input Mat and output Mat are not equal");
2687398c7888SPierre Jolivet   }
2688398c7888SPierre Jolivet #endif
26893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2690f1580f4eSBarry Smith }
2691f1580f4eSBarry Smith 
2692f1580f4eSBarry Smith /*@
2693f1580f4eSBarry Smith   PCHPDDMSetCoarseCorrectionType - Sets the coarse correction type.
2694f1580f4eSBarry Smith 
2695c3339decSBarry Smith   Collective
2696f1580f4eSBarry Smith 
2697f1580f4eSBarry Smith   Input Parameters:
2698f1580f4eSBarry Smith + pc   - preconditioner context
2699aa1539e9SPierre Jolivet - type - `PC_HPDDM_COARSE_CORRECTION_DEFLATED`, `PC_HPDDM_COARSE_CORRECTION_ADDITIVE`, `PC_HPDDM_COARSE_CORRECTION_BALANCED`, or `PC_HPDDM_COARSE_CORRECTION_NONE`
2700f1580f4eSBarry Smith 
2701f1580f4eSBarry Smith   Options Database Key:
2702aa1539e9SPierre Jolivet . -pc_hpddm_coarse_correction <deflated, additive, balanced, none> - type of coarse correction to apply
2703f1580f4eSBarry Smith 
2704f1580f4eSBarry Smith   Level: intermediate
2705f1580f4eSBarry Smith 
2706562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDMGetCoarseCorrectionType()`, `PCHPDDM`, `PCHPDDMCoarseCorrectionType`
2707f1580f4eSBarry Smith @*/
2708d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetCoarseCorrectionType(PC pc, PCHPDDMCoarseCorrectionType type)
2709d71ae5a4SJacob Faibussowitsch {
2710f1580f4eSBarry Smith   PetscFunctionBegin;
2711f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2712f1580f4eSBarry Smith   PetscValidLogicalCollectiveEnum(pc, type, 2);
2713f1580f4eSBarry Smith   PetscTryMethod(pc, "PCHPDDMSetCoarseCorrectionType_C", (PC, PCHPDDMCoarseCorrectionType), (pc, type));
27143ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2715f1580f4eSBarry Smith }
2716f1580f4eSBarry Smith 
2717f1580f4eSBarry Smith /*@
2718f1580f4eSBarry Smith   PCHPDDMGetCoarseCorrectionType - Gets the coarse correction type.
2719f1580f4eSBarry Smith 
2720f1580f4eSBarry Smith   Input Parameter:
2721f1580f4eSBarry Smith . pc - preconditioner context
2722f1580f4eSBarry Smith 
2723f1580f4eSBarry Smith   Output Parameter:
2724aa1539e9SPierre Jolivet . type - `PC_HPDDM_COARSE_CORRECTION_DEFLATED`, `PC_HPDDM_COARSE_CORRECTION_ADDITIVE`, `PC_HPDDM_COARSE_CORRECTION_BALANCED`, or `PC_HPDDM_COARSE_CORRECTION_NONE`
2725f1580f4eSBarry Smith 
2726f1580f4eSBarry Smith   Level: intermediate
2727f1580f4eSBarry Smith 
2728562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDMSetCoarseCorrectionType()`, `PCHPDDM`, `PCHPDDMCoarseCorrectionType`
2729f1580f4eSBarry Smith @*/
2730d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMGetCoarseCorrectionType(PC pc, PCHPDDMCoarseCorrectionType *type)
2731d71ae5a4SJacob Faibussowitsch {
2732f1580f4eSBarry Smith   PetscFunctionBegin;
2733f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2734f1580f4eSBarry Smith   if (type) {
27354f572ea9SToby Isaac     PetscAssertPointer(type, 2);
2736f1580f4eSBarry Smith     PetscUseMethod(pc, "PCHPDDMGetCoarseCorrectionType_C", (PC, PCHPDDMCoarseCorrectionType *), (pc, type));
2737f1580f4eSBarry Smith   }
27383ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2739f1580f4eSBarry Smith }
2740f1580f4eSBarry Smith 
2741d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetCoarseCorrectionType_HPDDM(PC pc, PCHPDDMCoarseCorrectionType type)
2742d71ae5a4SJacob Faibussowitsch {
2743f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
2744f1580f4eSBarry Smith 
2745f1580f4eSBarry Smith   PetscFunctionBegin;
2746f1580f4eSBarry Smith   data->correction = type;
27473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2748f1580f4eSBarry Smith }
2749f1580f4eSBarry Smith 
2750d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMGetCoarseCorrectionType_HPDDM(PC pc, PCHPDDMCoarseCorrectionType *type)
2751d71ae5a4SJacob Faibussowitsch {
2752f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
2753f1580f4eSBarry Smith 
2754f1580f4eSBarry Smith   PetscFunctionBegin;
2755f1580f4eSBarry Smith   *type = data->correction;
27563ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2757f1580f4eSBarry Smith }
2758f1580f4eSBarry Smith 
2759f1580f4eSBarry Smith /*@
2760e31fc274SPierre Jolivet   PCHPDDMSetSTShareSubKSP - Sets whether the `KSP` in SLEPc `ST` and the fine-level subdomain solver should be shared.
2761e31fc274SPierre Jolivet 
2762e31fc274SPierre Jolivet   Input Parameters:
2763e31fc274SPierre Jolivet + pc    - preconditioner context
2764e31fc274SPierre Jolivet - share - whether the `KSP` should be shared or not
2765e31fc274SPierre Jolivet 
2766e31fc274SPierre Jolivet   Note:
2767e31fc274SPierre Jolivet   This is not the same as `PCSetReusePreconditioner()`. Given certain conditions (visible using -info), a symbolic factorization can be skipped
2768e31fc274SPierre Jolivet   when using a subdomain `PCType` such as `PCLU` or `PCCHOLESKY`.
2769e31fc274SPierre Jolivet 
2770e31fc274SPierre Jolivet   Level: advanced
2771e31fc274SPierre Jolivet 
2772562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMGetSTShareSubKSP()`
2773e31fc274SPierre Jolivet @*/
2774e31fc274SPierre Jolivet PetscErrorCode PCHPDDMSetSTShareSubKSP(PC pc, PetscBool share)
2775e31fc274SPierre Jolivet {
2776e31fc274SPierre Jolivet   PetscFunctionBegin;
2777e31fc274SPierre Jolivet   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2778e31fc274SPierre Jolivet   PetscTryMethod(pc, "PCHPDDMSetSTShareSubKSP_C", (PC, PetscBool), (pc, share));
27793ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2780e31fc274SPierre Jolivet }
2781e31fc274SPierre Jolivet 
2782e31fc274SPierre Jolivet /*@
2783f1580f4eSBarry Smith   PCHPDDMGetSTShareSubKSP - Gets whether the `KSP` in SLEPc `ST` and the fine-level subdomain solver is shared.
2784f1580f4eSBarry Smith 
2785f1580f4eSBarry Smith   Input Parameter:
2786f1580f4eSBarry Smith . pc - preconditioner context
2787f1580f4eSBarry Smith 
2788f1580f4eSBarry Smith   Output Parameter:
2789f1580f4eSBarry Smith . share - whether the `KSP` is shared or not
2790f1580f4eSBarry Smith 
2791f1580f4eSBarry Smith   Note:
2792f1580f4eSBarry Smith   This is not the same as `PCGetReusePreconditioner()`. The return value is unlikely to be true, but when it is, a symbolic factorization can be skipped
2793f1580f4eSBarry Smith   when using a subdomain `PCType` such as `PCLU` or `PCCHOLESKY`.
2794f1580f4eSBarry Smith 
2795f1580f4eSBarry Smith   Level: advanced
2796f1580f4eSBarry Smith 
2797562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMSetSTShareSubKSP()`
2798f1580f4eSBarry Smith @*/
2799d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMGetSTShareSubKSP(PC pc, PetscBool *share)
2800d71ae5a4SJacob Faibussowitsch {
2801f1580f4eSBarry Smith   PetscFunctionBegin;
2802f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2803f1580f4eSBarry Smith   if (share) {
28044f572ea9SToby Isaac     PetscAssertPointer(share, 2);
2805f1580f4eSBarry Smith     PetscUseMethod(pc, "PCHPDDMGetSTShareSubKSP_C", (PC, PetscBool *), (pc, share));
2806f1580f4eSBarry Smith   }
28073ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2808f1580f4eSBarry Smith }
2809f1580f4eSBarry Smith 
2810e31fc274SPierre Jolivet static PetscErrorCode PCHPDDMSetSTShareSubKSP_HPDDM(PC pc, PetscBool share)
2811e31fc274SPierre Jolivet {
2812e31fc274SPierre Jolivet   PC_HPDDM *data = (PC_HPDDM *)pc->data;
2813e31fc274SPierre Jolivet 
2814e31fc274SPierre Jolivet   PetscFunctionBegin;
2815e31fc274SPierre Jolivet   data->share = share;
28163ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2817e31fc274SPierre Jolivet }
2818e31fc274SPierre Jolivet 
2819d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMGetSTShareSubKSP_HPDDM(PC pc, PetscBool *share)
2820d71ae5a4SJacob Faibussowitsch {
2821f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
2822f1580f4eSBarry Smith 
2823f1580f4eSBarry Smith   PetscFunctionBegin;
2824f1580f4eSBarry Smith   *share = data->share;
28253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2826f1580f4eSBarry Smith }
2827f1580f4eSBarry Smith 
2828f1580f4eSBarry Smith /*@
2829f1580f4eSBarry Smith   PCHPDDMSetDeflationMat - Sets the deflation space used to assemble a coarser operator.
2830f1580f4eSBarry Smith 
2831f1580f4eSBarry Smith   Input Parameters:
2832f1580f4eSBarry Smith + pc - preconditioner context
2833f1580f4eSBarry Smith . is - index set of the local deflation matrix
2834f1580f4eSBarry Smith - U  - deflation sequential matrix stored as a `MATSEQDENSE`
2835f1580f4eSBarry Smith 
2836f1580f4eSBarry Smith   Level: advanced
2837f1580f4eSBarry Smith 
2838562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCDeflationSetSpace()`, `PCMGSetRestriction()`
2839f1580f4eSBarry Smith @*/
2840d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetDeflationMat(PC pc, IS is, Mat U)
2841d71ae5a4SJacob Faibussowitsch {
2842f1580f4eSBarry Smith   PetscFunctionBegin;
2843f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2844f1580f4eSBarry Smith   PetscValidHeaderSpecific(is, IS_CLASSID, 2);
2845f1580f4eSBarry Smith   PetscValidHeaderSpecific(U, MAT_CLASSID, 3);
2846e31fc274SPierre Jolivet   PetscTryMethod(pc, "PCHPDDMSetDeflationMat_C", (PC, IS, Mat), (pc, is, U));
28473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2848f1580f4eSBarry Smith }
2849f1580f4eSBarry Smith 
2850d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetDeflationMat_HPDDM(PC pc, IS is, Mat U)
2851d71ae5a4SJacob Faibussowitsch {
2852f1580f4eSBarry Smith   PetscFunctionBegin;
2853f1580f4eSBarry Smith   PetscCall(PCHPDDMSetAuxiliaryMat_Private(pc, is, U, PETSC_TRUE));
28543ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2855f1580f4eSBarry Smith }
2856f1580f4eSBarry Smith 
2857d71ae5a4SJacob Faibussowitsch PetscErrorCode HPDDMLoadDL_Private(PetscBool *found)
2858d71ae5a4SJacob Faibussowitsch {
2859605ad303SPierre Jolivet   PetscBool flg;
2860f1580f4eSBarry Smith   char      lib[PETSC_MAX_PATH_LEN], dlib[PETSC_MAX_PATH_LEN], dir[PETSC_MAX_PATH_LEN];
2861f1580f4eSBarry Smith 
2862f1580f4eSBarry Smith   PetscFunctionBegin;
28634f572ea9SToby Isaac   PetscAssertPointer(found, 1);
2864c6a7a370SJeremy L Thompson   PetscCall(PetscStrncpy(dir, "${PETSC_LIB_DIR}", sizeof(dir)));
2865db4a47b3SPierre Jolivet   PetscCall(PetscOptionsGetString(nullptr, nullptr, "-hpddm_dir", dir, sizeof(dir), nullptr));
2866f1580f4eSBarry Smith   PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/libhpddm_petsc", dir));
2867f1580f4eSBarry Smith   PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found));
2868605ad303SPierre Jolivet #if defined(SLEPC_LIB_DIR) /* this variable is passed during SLEPc ./configure when PETSc has not been configured   */
2869605ad303SPierre Jolivet   if (!*found) {           /* with --download-hpddm since slepcconf.h is not yet built (and thus can't be included) */
2870c6a7a370SJeremy L Thompson     PetscCall(PetscStrncpy(dir, HPDDM_STR(SLEPC_LIB_DIR), sizeof(dir)));
2871f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/libhpddm_petsc", dir));
2872f1580f4eSBarry Smith     PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found));
2873f1580f4eSBarry Smith   }
2874f1580f4eSBarry Smith #endif
2875605ad303SPierre Jolivet   if (!*found) { /* probable options for this to evaluate to PETSC_TRUE: system inconsistency (libhpddm_petsc moved by user?) or PETSc configured without --download-slepc */
2876605ad303SPierre Jolivet     PetscCall(PetscOptionsGetenv(PETSC_COMM_SELF, "SLEPC_DIR", dir, sizeof(dir), &flg));
2877605ad303SPierre Jolivet     if (flg) { /* if both PETSc and SLEPc are configured with --download-hpddm but PETSc has been configured without --download-slepc, one must ensure that libslepc is loaded before libhpddm_petsc */
2878605ad303SPierre Jolivet       PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/lib/libslepc", dir));
2879605ad303SPierre Jolivet       PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found));
2880605ad303SPierre Jolivet       PetscCheck(*found, PETSC_COMM_SELF, PETSC_ERR_PLIB, "%s not found but SLEPC_DIR=%s", lib, dir);
2881605ad303SPierre Jolivet       PetscCall(PetscDLLibraryAppend(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, dlib));
2882605ad303SPierre Jolivet       PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/lib/libhpddm_petsc", dir)); /* libhpddm_petsc is always in the same directory as libslepc */
2883605ad303SPierre Jolivet       PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found));
2884605ad303SPierre Jolivet     }
2885605ad303SPierre Jolivet   }
2886f1580f4eSBarry Smith   PetscCheck(*found, PETSC_COMM_SELF, PETSC_ERR_PLIB, "%s not found", lib);
2887f1580f4eSBarry Smith   PetscCall(PetscDLLibraryAppend(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, dlib));
28883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2889f1580f4eSBarry Smith }
2890f1580f4eSBarry Smith 
2891f1580f4eSBarry Smith /*MC
2892f1580f4eSBarry Smith    PCHPDDM - Interface with the HPDDM library.
2893f1580f4eSBarry Smith 
28941d27aa22SBarry Smith    This `PC` may be used to build multilevel spectral domain decomposition methods based on the GenEO framework {cite}`spillane2011robust` {cite}`al2021multilevel`.
28951d27aa22SBarry Smith    It may be viewed as an alternative to spectral
28961d27aa22SBarry Smith    AMGe or `PCBDDC` with adaptive selection of constraints. The interface is explained in details in {cite}`jolivetromanzampini2020`
2897f1580f4eSBarry Smith 
2898e7593814SPierre Jolivet    The matrix used for building the preconditioner (Pmat) may be unassembled (`MATIS`), assembled (`MATAIJ`, `MATBAIJ`, or `MATSBAIJ`), hierarchical (`MATHTOOL`), `MATNORMAL`, `MATNORMALHERMITIAN`, or `MATSCHURCOMPLEMENT` (when `PCHPDDM` is used as part of an outer `PCFIELDSPLIT`).
2899f1580f4eSBarry Smith 
2900f1580f4eSBarry Smith    For multilevel preconditioning, when using an assembled or hierarchical Pmat, one must provide an auxiliary local `Mat` (unassembled local operator for GenEO) using
2901f1580f4eSBarry Smith    `PCHPDDMSetAuxiliaryMat()`. Calling this routine is not needed when using a `MATIS` Pmat, assembly is done internally using `MatConvert()`.
2902f1580f4eSBarry Smith 
2903f1580f4eSBarry Smith    Options Database Keys:
2904f1580f4eSBarry Smith +   -pc_hpddm_define_subdomains <true, default=false>    - on the finest level, calls `PCASMSetLocalSubdomains()` with the `IS` supplied in `PCHPDDMSetAuxiliaryMat()`
2905f1580f4eSBarry Smith                                                          (not relevant with an unassembled Pmat)
2906f1580f4eSBarry Smith .   -pc_hpddm_has_neumann <true, default=false>          - on the finest level, informs the `PC` that the local Neumann matrix is supplied in `PCHPDDMSetAuxiliaryMat()`
2907f1580f4eSBarry Smith -   -pc_hpddm_coarse_correction <type, default=deflated> - determines the `PCHPDDMCoarseCorrectionType` when calling `PCApply()`
2908f1580f4eSBarry Smith 
290938bf2a8cSPierre Jolivet    Options for subdomain solvers, subdomain eigensolvers (for computing deflation vectors), and the coarse solver can be set using the following options database prefixes.
2910f1580f4eSBarry Smith .vb
2911f1580f4eSBarry Smith       -pc_hpddm_levels_%d_pc_
2912f1580f4eSBarry Smith       -pc_hpddm_levels_%d_ksp_
2913f1580f4eSBarry Smith       -pc_hpddm_levels_%d_eps_
2914f1580f4eSBarry Smith       -pc_hpddm_levels_%d_p
29154ec2a359SPierre Jolivet       -pc_hpddm_levels_%d_mat_type
2916f1580f4eSBarry Smith       -pc_hpddm_coarse_
2917f1580f4eSBarry Smith       -pc_hpddm_coarse_p
29184ec2a359SPierre Jolivet       -pc_hpddm_coarse_mat_type
29192ce66baaSPierre Jolivet       -pc_hpddm_coarse_mat_filter
2920f1580f4eSBarry Smith .ve
2921f1580f4eSBarry Smith 
292238bf2a8cSPierre Jolivet    E.g., -pc_hpddm_levels_1_sub_pc_type lu -pc_hpddm_levels_1_eps_nev 10 -pc_hpddm_levels_2_p 4 -pc_hpddm_levels_2_sub_pc_type lu -pc_hpddm_levels_2_eps_nev 10
2923f1580f4eSBarry Smith     -pc_hpddm_coarse_p 2 -pc_hpddm_coarse_mat_type baij will use 10 deflation vectors per subdomain on the fine "level 1",
2924f1580f4eSBarry Smith     aggregate the fine subdomains into 4 "level 2" subdomains, then use 10 deflation vectors per subdomain on "level 2",
29257eb095acSPierre Jolivet     and assemble the coarse matrix (of dimension 4 x 10 = 40) on two processes as a `MATBAIJ` (default is `MATSBAIJ`).
2926f1580f4eSBarry Smith 
2927f1580f4eSBarry Smith    In order to activate a "level N+1" coarse correction, it is mandatory to call -pc_hpddm_levels_N_eps_nev <nu> or -pc_hpddm_levels_N_eps_threshold <val>. The default -pc_hpddm_coarse_p value is 1, meaning that the coarse operator is aggregated on a single process.
2928f1580f4eSBarry Smith 
29291d27aa22SBarry Smith    Level: intermediate
29301d27aa22SBarry Smith 
29311d27aa22SBarry Smith    Notes:
29321d27aa22SBarry Smith    This preconditioner requires that PETSc is built with SLEPc (``--download-slepc``).
29331d27aa22SBarry Smith 
29341d27aa22SBarry Smith    By default, the underlying concurrent eigenproblems
29351d27aa22SBarry Smith    are solved using SLEPc shift-and-invert spectral transformation. This is usually what gives the best performance for GenEO, cf.
29361d27aa22SBarry Smith    {cite}`spillane2011robust` {cite}`jolivet2013scalabledd`. As
293738bf2a8cSPierre Jolivet    stated above, SLEPc options are available through -pc_hpddm_levels_%d_, e.g., -pc_hpddm_levels_1_eps_type arpack -pc_hpddm_levels_1_eps_nev 10
293838bf2a8cSPierre Jolivet    -pc_hpddm_levels_1_st_type sinvert. There are furthermore three options related to the (subdomain-wise local) eigensolver that are not described in
293938bf2a8cSPierre Jolivet    SLEPc documentation since they are specific to `PCHPDDM`.
294038bf2a8cSPierre Jolivet .vb
294138bf2a8cSPierre Jolivet       -pc_hpddm_levels_1_st_share_sub_ksp
294238bf2a8cSPierre Jolivet       -pc_hpddm_levels_%d_eps_threshold
294338bf2a8cSPierre Jolivet       -pc_hpddm_levels_1_eps_use_inertia
294438bf2a8cSPierre Jolivet .ve
294538bf2a8cSPierre Jolivet 
294638bf2a8cSPierre Jolivet    The first option from the list only applies to the fine-level eigensolver, see `PCHPDDMSetSTShareSubKSP()`. The second option from the list is
294738bf2a8cSPierre Jolivet    used to filter eigenmodes retrieved after convergence of `EPSSolve()` at "level N" such that eigenvectors used to define a "level N+1" coarse
294838bf2a8cSPierre Jolivet    correction are associated to eigenvalues whose magnitude are lower or equal than -pc_hpddm_levels_N_eps_threshold. When using an `EPS` which cannot
294938bf2a8cSPierre Jolivet    determine a priori the proper -pc_hpddm_levels_N_eps_nev such that all wanted eigenmodes are retrieved, it is possible to get an estimation of the
295038bf2a8cSPierre Jolivet    correct value using the third option from the list, -pc_hpddm_levels_1_eps_use_inertia, see `MatGetInertia()`. In that case, there is no need
295138bf2a8cSPierre Jolivet    to supply -pc_hpddm_levels_1_eps_nev. This last option also only applies to the fine-level (N = 1) eigensolver.
2952f1580f4eSBarry Smith 
29531d27aa22SBarry Smith    See also {cite}`dolean2015introduction`, {cite}`al2022robust`, {cite}`al2022robustpd`, and {cite}`nataf2022recent`
2954f1580f4eSBarry Smith 
2955562efe2eSBarry Smith .seealso: [](ch_ksp), `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHPDDMSetAuxiliaryMat()`, `MATIS`, `PCBDDC`, `PCDEFLATION`, `PCTELESCOPE`, `PCASM`,
2956e31fc274SPierre Jolivet           `PCHPDDMSetCoarseCorrectionType()`, `PCHPDDMHasNeumannMat()`, `PCHPDDMSetRHSMat()`, `PCHPDDMSetDeflationMat()`, `PCHPDDMSetSTShareSubKSP()`,
2957e31fc274SPierre Jolivet           `PCHPDDMGetSTShareSubKSP()`, `PCHPDDMGetCoarseCorrectionType()`, `PCHPDDMGetComplexities()`
2958f1580f4eSBarry Smith M*/
2959d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode PCCreate_HPDDM(PC pc)
2960d71ae5a4SJacob Faibussowitsch {
2961f1580f4eSBarry Smith   PC_HPDDM *data;
2962f1580f4eSBarry Smith   PetscBool found;
2963f1580f4eSBarry Smith 
2964f1580f4eSBarry Smith   PetscFunctionBegin;
2965f1580f4eSBarry Smith   if (!loadedSym) {
2966f1580f4eSBarry Smith     PetscCall(HPDDMLoadDL_Private(&found));
2967db4a47b3SPierre Jolivet     if (found) PetscCall(PetscDLLibrarySym(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, nullptr, "PCHPDDM_Internal", (void **)&loadedSym));
2968f1580f4eSBarry Smith   }
2969f1580f4eSBarry Smith   PetscCheck(loadedSym, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCHPDDM_Internal symbol not found in loaded libhpddm_petsc");
29704dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&data));
2971f1580f4eSBarry Smith   pc->data                = data;
2972c8ea6600SPierre Jolivet   data->Neumann           = PETSC_BOOL3_UNKNOWN;
2973f1580f4eSBarry Smith   pc->ops->reset          = PCReset_HPDDM;
2974f1580f4eSBarry Smith   pc->ops->destroy        = PCDestroy_HPDDM;
2975f1580f4eSBarry Smith   pc->ops->setfromoptions = PCSetFromOptions_HPDDM;
2976f1580f4eSBarry Smith   pc->ops->setup          = PCSetUp_HPDDM;
2977f1580f4eSBarry Smith   pc->ops->apply          = PCApply_HPDDM;
2978f1580f4eSBarry Smith   pc->ops->matapply       = PCMatApply_HPDDM;
2979f1580f4eSBarry Smith   pc->ops->view           = PCView_HPDDM;
2980f1580f4eSBarry Smith   pc->ops->presolve       = PCPreSolve_HPDDM;
2981f1580f4eSBarry Smith 
2982f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetAuxiliaryMat_C", PCHPDDMSetAuxiliaryMat_HPDDM));
2983f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMHasNeumannMat_C", PCHPDDMHasNeumannMat_HPDDM));
2984f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetRHSMat_C", PCHPDDMSetRHSMat_HPDDM));
2985f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetCoarseCorrectionType_C", PCHPDDMSetCoarseCorrectionType_HPDDM));
2986f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetCoarseCorrectionType_C", PCHPDDMGetCoarseCorrectionType_HPDDM));
2987e31fc274SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetSTShareSubKSP_C", PCHPDDMSetSTShareSubKSP_HPDDM));
2988f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetSTShareSubKSP_C", PCHPDDMGetSTShareSubKSP_HPDDM));
2989f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetDeflationMat_C", PCHPDDMSetDeflationMat_HPDDM));
29903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2991f1580f4eSBarry Smith }
2992f1580f4eSBarry Smith 
2993f1580f4eSBarry Smith /*@C
2994f1580f4eSBarry Smith   PCHPDDMInitializePackage - This function initializes everything in the `PCHPDDM` package. It is called from `PCInitializePackage()`.
2995f1580f4eSBarry Smith 
2996f1580f4eSBarry Smith   Level: developer
2997f1580f4eSBarry Smith 
2998562efe2eSBarry Smith .seealso: [](ch_ksp), `PetscInitialize()`
2999f1580f4eSBarry Smith @*/
3000d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMInitializePackage(void)
3001d71ae5a4SJacob Faibussowitsch {
3002f1580f4eSBarry Smith   char ename[32];
3003f1580f4eSBarry Smith 
3004f1580f4eSBarry Smith   PetscFunctionBegin;
30053ba16761SJacob Faibussowitsch   if (PCHPDDMPackageInitialized) PetscFunctionReturn(PETSC_SUCCESS);
3006f1580f4eSBarry Smith   PCHPDDMPackageInitialized = PETSC_TRUE;
3007f1580f4eSBarry Smith   PetscCall(PetscRegisterFinalize(PCHPDDMFinalizePackage));
3008f1580f4eSBarry Smith   /* general events registered once during package initialization */
3009f1580f4eSBarry Smith   /* some of these events are not triggered in libpetsc,          */
3010f1580f4eSBarry Smith   /* but rather directly in libhpddm_petsc,                       */
3011f1580f4eSBarry Smith   /* which is in charge of performing the following operations    */
3012f1580f4eSBarry Smith 
3013f1580f4eSBarry Smith   /* domain decomposition structure from Pmat sparsity pattern    */
3014f1580f4eSBarry Smith   PetscCall(PetscLogEventRegister("PCHPDDMStrc", PC_CLASSID, &PC_HPDDM_Strc));
3015f1580f4eSBarry Smith   /* Galerkin product, redistribution, and setup (not triggered in libpetsc)                */
3016f1580f4eSBarry Smith   PetscCall(PetscLogEventRegister("PCHPDDMPtAP", PC_CLASSID, &PC_HPDDM_PtAP));
3017f1580f4eSBarry Smith   /* Galerkin product with summation, redistribution, and setup (not triggered in libpetsc) */
3018f1580f4eSBarry Smith   PetscCall(PetscLogEventRegister("PCHPDDMPtBP", PC_CLASSID, &PC_HPDDM_PtBP));
3019f1580f4eSBarry Smith   /* next level construction using PtAP and PtBP (not triggered in libpetsc)                */
3020f1580f4eSBarry Smith   PetscCall(PetscLogEventRegister("PCHPDDMNext", PC_CLASSID, &PC_HPDDM_Next));
3021f1580f4eSBarry Smith   static_assert(PETSC_PCHPDDM_MAXLEVELS <= 9, "PETSC_PCHPDDM_MAXLEVELS value is too high");
3022811e8887SPierre Jolivet   for (PetscInt i = 1; i < PETSC_PCHPDDM_MAXLEVELS; ++i) {
3023f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(ename, sizeof(ename), "PCHPDDMSetUp L%1" PetscInt_FMT, i));
3024f1580f4eSBarry Smith     /* events during a PCSetUp() at level #i _except_ the assembly */
3025f1580f4eSBarry Smith     /* of the Galerkin operator of the coarser level #(i + 1)      */
3026f1580f4eSBarry Smith     PetscCall(PetscLogEventRegister(ename, PC_CLASSID, &PC_HPDDM_SetUp[i - 1]));
3027f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(ename, sizeof(ename), "PCHPDDMSolve L%1" PetscInt_FMT, i));
3028f1580f4eSBarry Smith     /* events during a PCApply() at level #i _except_              */
3029f1580f4eSBarry Smith     /* the KSPSolve() of the coarser level #(i + 1)                */
3030f1580f4eSBarry Smith     PetscCall(PetscLogEventRegister(ename, PC_CLASSID, &PC_HPDDM_Solve[i - 1]));
3031f1580f4eSBarry Smith   }
30323ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3033f1580f4eSBarry Smith }
3034f1580f4eSBarry Smith 
3035f1580f4eSBarry Smith /*@C
3036f1580f4eSBarry Smith   PCHPDDMFinalizePackage - This function frees everything from the `PCHPDDM` package. It is called from `PetscFinalize()`.
3037f1580f4eSBarry Smith 
3038f1580f4eSBarry Smith   Level: developer
3039f1580f4eSBarry Smith 
3040562efe2eSBarry Smith .seealso: [](ch_ksp), `PetscFinalize()`
3041f1580f4eSBarry Smith @*/
3042d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMFinalizePackage(void)
3043d71ae5a4SJacob Faibussowitsch {
3044f1580f4eSBarry Smith   PetscFunctionBegin;
3045f1580f4eSBarry Smith   PCHPDDMPackageInitialized = PETSC_FALSE;
30463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3047f1580f4eSBarry Smith }
30489bb5c669SPierre Jolivet 
30499bb5c669SPierre Jolivet static PetscErrorCode MatMult_Harmonic(Mat A, Vec x, Vec y)
30509bb5c669SPierre Jolivet {
30519bb5c669SPierre Jolivet   Harmonic h; /* [ A_00  A_01       ], furthermore, A_00 = [ A_loc,loc  A_loc,ovl ], thus, A_01 = [         ] */
30529bb5c669SPierre Jolivet               /* [ A_10  A_11  A_12 ]                      [ A_ovl,loc  A_ovl,ovl ]               [ A_ovl,1 ] */
30539bb5c669SPierre Jolivet   Vec sub;    /*  y = A x = R_loc R_0 [ A_00  A_01 ]^-1                                   R_loc = [  I_loc  ] */
30549bb5c669SPierre Jolivet               /*                      [ A_10  A_11 ]    R_1^T A_12 x                              [         ] */
30559bb5c669SPierre Jolivet   PetscFunctionBegin;
30569bb5c669SPierre Jolivet   PetscCall(MatShellGetContext(A, &h));
30579bb5c669SPierre Jolivet   PetscCall(VecSet(h->v, 0.0));
30589bb5c669SPierre Jolivet   PetscCall(VecGetSubVector(h->v, h->is[0], &sub));
30599bb5c669SPierre Jolivet   PetscCall(MatMult(h->A[0], x, sub));
30609bb5c669SPierre Jolivet   PetscCall(VecRestoreSubVector(h->v, h->is[0], &sub));
30619bb5c669SPierre Jolivet   PetscCall(KSPSolve(h->ksp, h->v, h->v));
30629bb5c669SPierre Jolivet   PetscCall(VecISCopy(h->v, h->is[1], SCATTER_REVERSE, y));
30639bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
30649bb5c669SPierre Jolivet }
30659bb5c669SPierre Jolivet 
30669bb5c669SPierre Jolivet static PetscErrorCode MatMultTranspose_Harmonic(Mat A, Vec y, Vec x)
30679bb5c669SPierre Jolivet {
30689bb5c669SPierre Jolivet   Harmonic h;   /* x = A^T y =            [ A_00  A_01 ]^-T R_0^T R_loc^T y */
30699bb5c669SPierre Jolivet   Vec      sub; /*             A_12^T R_1 [ A_10  A_11 ]                    */
30709bb5c669SPierre Jolivet 
30719bb5c669SPierre Jolivet   PetscFunctionBegin;
30729bb5c669SPierre Jolivet   PetscCall(MatShellGetContext(A, &h));
30739bb5c669SPierre Jolivet   PetscCall(VecSet(h->v, 0.0));
30749bb5c669SPierre Jolivet   PetscCall(VecISCopy(h->v, h->is[1], SCATTER_FORWARD, y));
30759bb5c669SPierre Jolivet   PetscCall(KSPSolveTranspose(h->ksp, h->v, h->v));
30769bb5c669SPierre Jolivet   PetscCall(VecGetSubVector(h->v, h->is[0], &sub));
30779bb5c669SPierre Jolivet   PetscCall(MatMultTranspose(h->A[0], sub, x));
30789bb5c669SPierre Jolivet   PetscCall(VecRestoreSubVector(h->v, h->is[0], &sub));
30799bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
30809bb5c669SPierre Jolivet }
30819bb5c669SPierre Jolivet 
30829bb5c669SPierre Jolivet static PetscErrorCode MatProduct_AB_Harmonic(Mat S, Mat X, Mat Y, void *)
30839bb5c669SPierre Jolivet {
30849bb5c669SPierre Jolivet   Harmonic h;
30859bb5c669SPierre Jolivet   Mat      A, B;
30869bb5c669SPierre Jolivet   Vec      a, b;
30879bb5c669SPierre Jolivet 
30889bb5c669SPierre Jolivet   PetscFunctionBegin;
30899bb5c669SPierre Jolivet   PetscCall(MatShellGetContext(S, &h));
3090fb842aefSJose E. Roman   PetscCall(MatMatMult(h->A[0], X, MAT_INITIAL_MATRIX, PETSC_CURRENT, &A));
30919bb5c669SPierre Jolivet   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, h->ksp->pc->mat->rmap->n, A->cmap->n, nullptr, &B));
30929bb5c669SPierre Jolivet   for (PetscInt i = 0; i < A->cmap->n; ++i) {
30939bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecRead(A, i, &a));
30949bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecWrite(B, i, &b));
30959bb5c669SPierre Jolivet     PetscCall(VecISCopy(b, h->is[0], SCATTER_FORWARD, a));
30969bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecWrite(B, i, &b));
30979bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecRead(A, i, &a));
30989bb5c669SPierre Jolivet   }
30999bb5c669SPierre Jolivet   PetscCall(MatDestroy(&A));
31009bb5c669SPierre Jolivet   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, h->ksp->pc->mat->rmap->n, B->cmap->n, nullptr, &A));
31019bb5c669SPierre Jolivet   PetscCall(KSPMatSolve(h->ksp, B, A));
31029bb5c669SPierre Jolivet   PetscCall(MatDestroy(&B));
31039bb5c669SPierre Jolivet   for (PetscInt i = 0; i < A->cmap->n; ++i) {
31049bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecRead(A, i, &a));
31059bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecWrite(Y, i, &b));
31069bb5c669SPierre Jolivet     PetscCall(VecISCopy(a, h->is[1], SCATTER_REVERSE, b));
31079bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &b));
31089bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecRead(A, i, &a));
31099bb5c669SPierre Jolivet   }
31109bb5c669SPierre Jolivet   PetscCall(MatDestroy(&A));
31119bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
31129bb5c669SPierre Jolivet }
31139bb5c669SPierre Jolivet 
31149bb5c669SPierre Jolivet static PetscErrorCode MatDestroy_Harmonic(Mat A)
31159bb5c669SPierre Jolivet {
31169bb5c669SPierre Jolivet   Harmonic h;
31179bb5c669SPierre Jolivet 
31189bb5c669SPierre Jolivet   PetscFunctionBegin;
31199bb5c669SPierre Jolivet   PetscCall(MatShellGetContext(A, &h));
312032603206SJames Wright   for (PetscInt i = 0; i < 5; ++i) PetscCall(ISDestroy(h->is + i));
31219bb5c669SPierre Jolivet   PetscCall(PetscFree(h->is));
31229bb5c669SPierre Jolivet   PetscCall(VecDestroy(&h->v));
31239bb5c669SPierre Jolivet   for (PetscInt i = 0; i < 2; ++i) PetscCall(MatDestroy(h->A + i));
31249bb5c669SPierre Jolivet   PetscCall(PetscFree(h->A));
31259bb5c669SPierre Jolivet   PetscCall(KSPDestroy(&h->ksp));
31269bb5c669SPierre Jolivet   PetscCall(PetscFree(h));
31279bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
31289bb5c669SPierre Jolivet }
3129