xref: /petsc/src/ksp/pc/impls/hpddm/pchpddm.cxx (revision 371d2eb7c12e2e633516fd846e18141c0cd1b5b5)
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;
74c30dab7bSPierre Jolivet   PetscValidLogicalCollectiveBool(pc, deflation, 4);
75c30dab7bSPierre Jolivet   if (is && A) {
76c30dab7bSPierre Jolivet     PetscInt m[2];
77c30dab7bSPierre Jolivet 
78c30dab7bSPierre Jolivet     PetscCall(ISGetLocalSize(is, m));
79c30dab7bSPierre Jolivet     PetscCall(MatGetLocalSize(A, m + 1, nullptr));
80c30dab7bSPierre Jolivet     PetscCheck(m[0] == m[1], PETSC_COMM_SELF, PETSC_ERR_USER_INPUT, "Inconsistent IS and Mat sizes (%" PetscInt_FMT " v. %" PetscInt_FMT ")", m[0], m[1]);
81c30dab7bSPierre Jolivet   }
82f1580f4eSBarry Smith   if (is) {
83f1580f4eSBarry Smith     PetscCall(PetscObjectReference((PetscObject)is));
84f1580f4eSBarry Smith     if (data->is) { /* new overlap definition resets the PC */
85f1580f4eSBarry Smith       PetscCall(PCReset_HPDDM(pc));
86f1580f4eSBarry Smith       pc->setfromoptionscalled = 0;
87*371d2eb7SMartin Diehl       pc->setupcalled          = PETSC_FALSE;
88cdbd50ebSPierre Jolivet       data->correction         = type;
89f1580f4eSBarry Smith     }
90f1580f4eSBarry Smith     PetscCall(ISDestroy(&data->is));
91f1580f4eSBarry Smith     data->is = is;
92f1580f4eSBarry Smith   }
93f1580f4eSBarry Smith   if (A) {
94f1580f4eSBarry Smith     PetscCall(PetscObjectReference((PetscObject)A));
95f1580f4eSBarry Smith     PetscCall(MatDestroy(&data->aux));
96f1580f4eSBarry Smith     data->aux = A;
97f1580f4eSBarry Smith   }
98f1580f4eSBarry Smith   data->deflation = deflation;
993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
100f1580f4eSBarry Smith }
101f1580f4eSBarry Smith 
102281f8ce6SPierre Jolivet static inline PetscErrorCode PCHPDDMSplittingMatNormal_Private(Mat A, IS *is, Mat *splitting[])
103281f8ce6SPierre Jolivet {
104281f8ce6SPierre Jolivet   Mat *sub;
105281f8ce6SPierre Jolivet   IS   zero;
106281f8ce6SPierre Jolivet 
107281f8ce6SPierre Jolivet   PetscFunctionBegin;
108281f8ce6SPierre Jolivet   PetscCall(MatSetOption(A, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
109281f8ce6SPierre Jolivet   PetscCall(MatCreateSubMatrices(A, 1, is + 2, is, MAT_INITIAL_MATRIX, splitting));
110281f8ce6SPierre Jolivet   PetscCall(MatCreateSubMatrices(**splitting, 1, is + 2, is + 1, MAT_INITIAL_MATRIX, &sub));
111281f8ce6SPierre Jolivet   PetscCall(MatFindZeroRows(*sub, &zero));
112281f8ce6SPierre Jolivet   PetscCall(MatDestroySubMatrices(1, &sub));
113281f8ce6SPierre Jolivet   PetscCall(MatSetOption(**splitting, MAT_KEEP_NONZERO_PATTERN, PETSC_TRUE));
114281f8ce6SPierre Jolivet   PetscCall(MatZeroRowsIS(**splitting, zero, 0.0, nullptr, nullptr));
115281f8ce6SPierre Jolivet   PetscCall(ISDestroy(&zero));
116281f8ce6SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
117281f8ce6SPierre Jolivet }
118281f8ce6SPierre Jolivet 
119281f8ce6SPierre Jolivet static inline PetscErrorCode PCHPDDMSetAuxiliaryMatNormal_Private(PC pc, Mat A, Mat N, Mat *B, const char *pcpre, Vec *diagonal = nullptr, Mat B01 = nullptr)
120d71ae5a4SJacob Faibussowitsch {
121f1580f4eSBarry Smith   PC_HPDDM *data         = (PC_HPDDM *)pc->data;
122281f8ce6SPierre Jolivet   Mat      *splitting[2] = {}, aux;
1233df4cd7bSPierre Jolivet   Vec       d;
124281f8ce6SPierre Jolivet   IS        is[3];
125f1580f4eSBarry Smith   PetscReal norm;
126f1580f4eSBarry Smith   PetscBool flg;
127f1580f4eSBarry Smith   char      type[256] = {}; /* same size as in src/ksp/pc/interface/pcset.c */
128f1580f4eSBarry Smith 
129f1580f4eSBarry Smith   PetscFunctionBegin;
130281f8ce6SPierre Jolivet   if (!B01) PetscCall(MatConvert(N, MATAIJ, MAT_INITIAL_MATRIX, B));
131281f8ce6SPierre Jolivet   else PetscCall(MatTransposeMatMult(B01, A, MAT_INITIAL_MATRIX, PETSC_DETERMINE, B));
132feebddf4SPierre Jolivet   PetscCall(MatEliminateZeros(*B, PETSC_TRUE));
133281f8ce6SPierre Jolivet   PetscCall(ISCreateStride(PETSC_COMM_SELF, A->cmap->n, A->cmap->rstart, 1, is));
134281f8ce6SPierre Jolivet   PetscCall(MatIncreaseOverlap(*B, 1, is, 1));
135281f8ce6SPierre Jolivet   PetscCall(ISCreateStride(PETSC_COMM_SELF, A->cmap->n, A->cmap->rstart, 1, is + 2));
136281f8ce6SPierre Jolivet   PetscCall(ISEmbed(is[0], is[2], PETSC_TRUE, is + 1));
137281f8ce6SPierre Jolivet   PetscCall(ISDestroy(is + 2));
138281f8ce6SPierre Jolivet   PetscCall(ISCreateStride(PETSC_COMM_SELF, A->rmap->N, 0, 1, is + 2));
139281f8ce6SPierre Jolivet   PetscCall(PCHPDDMSplittingMatNormal_Private(A, is, &splitting[0]));
140281f8ce6SPierre Jolivet   if (B01) {
141281f8ce6SPierre Jolivet     PetscCall(PCHPDDMSplittingMatNormal_Private(B01, is, &splitting[1]));
142281f8ce6SPierre Jolivet     PetscCall(MatDestroy(&B01));
143281f8ce6SPierre Jolivet   }
144281f8ce6SPierre Jolivet   PetscCall(ISDestroy(is + 2));
145281f8ce6SPierre Jolivet   PetscCall(ISDestroy(is + 1));
146db4a47b3SPierre Jolivet   PetscCall(PetscOptionsGetString(nullptr, pcpre, "-pc_hpddm_levels_1_sub_pc_type", type, sizeof(type), nullptr));
147f1580f4eSBarry Smith   PetscCall(PetscStrcmp(type, PCQR, &flg));
148f1580f4eSBarry Smith   if (!flg) {
149281f8ce6SPierre Jolivet     Mat conjugate = *splitting[splitting[1] ? 1 : 0];
150811e8887SPierre Jolivet 
151281f8ce6SPierre Jolivet     if (PetscDefined(USE_COMPLEX) && !splitting[1]) {
152281f8ce6SPierre Jolivet       PetscCall(MatDuplicate(*splitting[0], MAT_COPY_VALUES, &conjugate));
153f1580f4eSBarry Smith       PetscCall(MatConjugate(conjugate));
154f1580f4eSBarry Smith     }
155281f8ce6SPierre Jolivet     PetscCall(MatTransposeMatMult(conjugate, *splitting[0], MAT_INITIAL_MATRIX, PETSC_DETERMINE, &aux));
156281f8ce6SPierre Jolivet     if (PetscDefined(USE_COMPLEX) && !splitting[1]) PetscCall(MatDestroy(&conjugate));
157281f8ce6SPierre Jolivet     else if (splitting[1]) PetscCall(MatDestroySubMatrices(1, &splitting[1]));
158f1580f4eSBarry Smith     PetscCall(MatNorm(aux, NORM_FROBENIUS, &norm));
159f1580f4eSBarry Smith     PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE));
1603df4cd7bSPierre Jolivet     if (diagonal) {
161811e8887SPierre Jolivet       PetscReal norm;
162811e8887SPierre Jolivet 
163feebddf4SPierre Jolivet       PetscCall(VecScale(*diagonal, -1.0));
1643df4cd7bSPierre Jolivet       PetscCall(VecNorm(*diagonal, NORM_INFINITY, &norm));
1653df4cd7bSPierre Jolivet       if (norm > PETSC_SMALL) {
16601e3c840SPierre Jolivet         PetscSF  scatter;
1673df4cd7bSPierre Jolivet         PetscInt n;
168811e8887SPierre Jolivet 
169281f8ce6SPierre Jolivet         PetscCall(ISGetLocalSize(*is, &n));
1703df4cd7bSPierre Jolivet         PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)pc), n, PETSC_DECIDE, &d));
171281f8ce6SPierre Jolivet         PetscCall(VecScatterCreate(*diagonal, *is, d, nullptr, &scatter));
1723df4cd7bSPierre Jolivet         PetscCall(VecScatterBegin(scatter, *diagonal, d, INSERT_VALUES, SCATTER_FORWARD));
1733df4cd7bSPierre Jolivet         PetscCall(VecScatterEnd(scatter, *diagonal, d, INSERT_VALUES, SCATTER_FORWARD));
17401e3c840SPierre Jolivet         PetscCall(PetscSFDestroy(&scatter));
1753df4cd7bSPierre Jolivet         PetscCall(MatDiagonalSet(aux, d, ADD_VALUES));
1763df4cd7bSPierre Jolivet         PetscCall(VecDestroy(&d));
1773df4cd7bSPierre Jolivet       } else PetscCall(VecDestroy(diagonal));
1783df4cd7bSPierre Jolivet     }
1793df4cd7bSPierre Jolivet     if (!diagonal) PetscCall(MatShift(aux, PETSC_SMALL * norm));
180f1580f4eSBarry Smith   } else {
181f1580f4eSBarry Smith     PetscBool flg;
182811e8887SPierre Jolivet 
183281f8ce6SPierre Jolivet     PetscCheck(!splitting[1], PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Cannot use PCQR when A01 != A10^T");
1843df4cd7bSPierre Jolivet     if (diagonal) {
1853df4cd7bSPierre Jolivet       PetscCall(VecNorm(*diagonal, NORM_INFINITY, &norm));
1863df4cd7bSPierre Jolivet       PetscCheck(norm < PETSC_SMALL, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Nonzero diagonal A11 block");
1873df4cd7bSPierre Jolivet       PetscCall(VecDestroy(diagonal));
1883df4cd7bSPierre Jolivet     }
189f1580f4eSBarry Smith     PetscCall(PetscObjectTypeCompare((PetscObject)N, MATNORMAL, &flg));
190281f8ce6SPierre Jolivet     if (flg) PetscCall(MatCreateNormal(*splitting[0], &aux));
191281f8ce6SPierre Jolivet     else PetscCall(MatCreateNormalHermitian(*splitting[0], &aux));
192f1580f4eSBarry Smith   }
193281f8ce6SPierre Jolivet   PetscCall(MatDestroySubMatrices(1, &splitting[0]));
194281f8ce6SPierre Jolivet   PetscCall(PCHPDDMSetAuxiliaryMat(pc, *is, aux, nullptr, nullptr));
195c8ea6600SPierre Jolivet   data->Neumann = PETSC_BOOL3_TRUE;
196281f8ce6SPierre Jolivet   PetscCall(ISDestroy(is));
197f1580f4eSBarry Smith   PetscCall(MatDestroy(&aux));
1983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
199f1580f4eSBarry Smith }
200f1580f4eSBarry Smith 
201d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetAuxiliaryMat_HPDDM(PC pc, IS is, Mat A, PetscErrorCode (*setup)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void *setup_ctx)
202d71ae5a4SJacob Faibussowitsch {
203f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
204f1580f4eSBarry Smith 
205f1580f4eSBarry Smith   PetscFunctionBegin;
206f1580f4eSBarry Smith   PetscCall(PCHPDDMSetAuxiliaryMat_Private(pc, is, A, PETSC_FALSE));
207f1580f4eSBarry Smith   if (setup) {
208f1580f4eSBarry Smith     data->setup     = setup;
209f1580f4eSBarry Smith     data->setup_ctx = setup_ctx;
210f1580f4eSBarry Smith   }
2113ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
212f1580f4eSBarry Smith }
213f1580f4eSBarry Smith 
21473329a61SPierre Jolivet typedef struct {
21573329a61SPierre Jolivet   KSP      ksp;
21673329a61SPierre Jolivet   PetscInt its;
21773329a61SPierre Jolivet } PC_KSP;
21873329a61SPierre Jolivet 
21973329a61SPierre Jolivet static inline PetscErrorCode PCSetUp_KSP_Private(PC pc)
22073329a61SPierre Jolivet {
22173329a61SPierre Jolivet   PC_KSP           *data   = (PC_KSP *)pc->data;
22273329a61SPierre Jolivet   const std::string prefix = ((PetscObject)data->ksp)->prefix;
22373329a61SPierre Jolivet   std::string       sub;
22473329a61SPierre Jolivet 
22573329a61SPierre Jolivet   PetscFunctionBegin;
22673329a61SPierre 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_");
22773329a61SPierre Jolivet   sub = prefix.substr(0, prefix.size() - 9);
22873329a61SPierre Jolivet   PetscCall(PCSetType(pc, PCHPDDM));
22973329a61SPierre Jolivet   PetscCall(PCSetOptionsPrefix(pc, sub.c_str()));
23073329a61SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
23173329a61SPierre Jolivet }
23273329a61SPierre Jolivet 
23373329a61SPierre Jolivet static PetscErrorCode PCSetUp_KSP(PC pc)
23473329a61SPierre Jolivet {
23573329a61SPierre Jolivet   PetscFunctionBegin;
23673329a61SPierre Jolivet   PetscCall(PCSetUp_KSP_Private(pc));
23773329a61SPierre Jolivet   PetscCall(PCSetFromOptions(pc));
23873329a61SPierre Jolivet   PetscCall(PCSetUp(pc));
23973329a61SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
24073329a61SPierre Jolivet }
24173329a61SPierre Jolivet 
24270009435SPierre Jolivet /*@C
24304c3f3b8SBarry Smith   PCHPDDMSetAuxiliaryMat - Sets the auxiliary matrix used by `PCHPDDM` for the concurrent GenEO problems at the finest level.
244f1580f4eSBarry Smith 
245f1580f4eSBarry Smith   Input Parameters:
246f1580f4eSBarry Smith + pc    - preconditioner context
247f1580f4eSBarry Smith . is    - index set of the local auxiliary, e.g., Neumann, matrix
248f1580f4eSBarry Smith . A     - auxiliary sequential matrix
24904c3f3b8SBarry Smith . setup - function for generating the auxiliary matrix entries, may be `NULL`
25004c3f3b8SBarry Smith - ctx   - context for `setup`, may be `NULL`
25104c3f3b8SBarry Smith 
25204c3f3b8SBarry Smith   Calling sequence of `setup`:
25304c3f3b8SBarry Smith + J   - matrix whose values are to be set
25404c3f3b8SBarry Smith . t   - time
25504c3f3b8SBarry Smith . X   - linearization point
25604c3f3b8SBarry Smith . X_t - time-derivative of the linearization point
25704c3f3b8SBarry Smith . s   - step
25804c3f3b8SBarry Smith . ovl - index set of the local auxiliary, e.g., Neumann, matrix
25904c3f3b8SBarry Smith - ctx - context for `setup`, may be `NULL`
260f1580f4eSBarry Smith 
261f1580f4eSBarry Smith   Level: intermediate
262f1580f4eSBarry Smith 
26304c3f3b8SBarry Smith   Note:
26404c3f3b8SBarry Smith   As an example, in a finite element context with nonoverlapping subdomains plus (overlapping) ghost elements, this could be the unassembled (Neumann)
26504c3f3b8SBarry Smith   local overlapping operator. As opposed to the assembled (Dirichlet) local overlapping operator obtained by summing neighborhood contributions
26604c3f3b8SBarry Smith   at the interface of ghost elements.
26704c3f3b8SBarry Smith 
26870009435SPierre Jolivet   Fortran Notes:
26904c3f3b8SBarry Smith   Only `PETSC_NULL_FUNCTION` is supported for `setup` and `ctx` is never accessed
27070009435SPierre Jolivet 
271562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHPDDMSetRHSMat()`, `MATIS`
272f1580f4eSBarry Smith @*/
27304c3f3b8SBarry 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)
274d71ae5a4SJacob Faibussowitsch {
275f1580f4eSBarry Smith   PetscFunctionBegin;
276f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
277f1580f4eSBarry Smith   if (is) PetscValidHeaderSpecific(is, IS_CLASSID, 2);
278f1580f4eSBarry Smith   if (A) PetscValidHeaderSpecific(A, MAT_CLASSID, 3);
27973329a61SPierre Jolivet   if (pc->ops->setup == PCSetUp_KSP) PetscCall(PCSetUp_KSP_Private(pc));
28004c3f3b8SBarry Smith   PetscTryMethod(pc, "PCHPDDMSetAuxiliaryMat_C", (PC, IS, Mat, PetscErrorCode (*)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void *), (pc, is, A, setup, ctx));
2813ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
282f1580f4eSBarry Smith }
283f1580f4eSBarry Smith 
284d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMHasNeumannMat_HPDDM(PC pc, PetscBool has)
285d71ae5a4SJacob Faibussowitsch {
286f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
287f1580f4eSBarry Smith 
288f1580f4eSBarry Smith   PetscFunctionBegin;
289c8ea6600SPierre Jolivet   data->Neumann = PetscBoolToBool3(has);
2903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
291f1580f4eSBarry Smith }
292f1580f4eSBarry Smith 
293f1580f4eSBarry Smith /*@
294f1580f4eSBarry Smith   PCHPDDMHasNeumannMat - Informs `PCHPDDM` that the `Mat` passed to `PCHPDDMSetAuxiliaryMat()` is the local Neumann matrix.
295f1580f4eSBarry Smith 
296f1580f4eSBarry Smith   Input Parameters:
297f1580f4eSBarry Smith + pc  - preconditioner context
298f1580f4eSBarry Smith - has - Boolean value
299f1580f4eSBarry Smith 
300f1580f4eSBarry Smith   Level: intermediate
301f1580f4eSBarry Smith 
302f1580f4eSBarry Smith   Notes:
3037eb095acSPierre Jolivet   This may be used to bypass a call to `MatCreateSubMatrices()` and to `MatConvert()` for `MATSBAIJ` matrices.
304f1580f4eSBarry Smith 
305907a3e9cSStefano 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`.
306f1580f4eSBarry Smith 
307562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMSetAuxiliaryMat()`
308f1580f4eSBarry Smith @*/
309d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMHasNeumannMat(PC pc, PetscBool has)
310d71ae5a4SJacob Faibussowitsch {
311f1580f4eSBarry Smith   PetscFunctionBegin;
312f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
313f1580f4eSBarry Smith   PetscTryMethod(pc, "PCHPDDMHasNeumannMat_C", (PC, PetscBool), (pc, has));
3143ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
315f1580f4eSBarry Smith }
316f1580f4eSBarry Smith 
317d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetRHSMat_HPDDM(PC pc, Mat B)
318d71ae5a4SJacob Faibussowitsch {
319f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
320f1580f4eSBarry Smith 
321f1580f4eSBarry Smith   PetscFunctionBegin;
322f1580f4eSBarry Smith   PetscCall(PetscObjectReference((PetscObject)B));
323f1580f4eSBarry Smith   PetscCall(MatDestroy(&data->B));
324f1580f4eSBarry Smith   data->B = B;
3253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
326f1580f4eSBarry Smith }
327f1580f4eSBarry Smith 
328f1580f4eSBarry Smith /*@
32904c3f3b8SBarry Smith   PCHPDDMSetRHSMat - Sets the right-hand side matrix used by `PCHPDDM` for the concurrent GenEO problems at the finest level.
330f1580f4eSBarry Smith 
331f1580f4eSBarry Smith   Input Parameters:
332f1580f4eSBarry Smith + pc - preconditioner context
333f1580f4eSBarry Smith - B  - right-hand side sequential matrix
334f1580f4eSBarry Smith 
335f1580f4eSBarry Smith   Level: advanced
336f1580f4eSBarry Smith 
33704c3f3b8SBarry Smith   Note:
33804c3f3b8SBarry Smith   Must be used in conjunction with `PCHPDDMSetAuxiliaryMat`(N), so that Nv = lambda Bv is solved using `EPSSetOperators`(N, B).
33904c3f3b8SBarry 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.
34004c3f3b8SBarry Smith 
341562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDMSetAuxiliaryMat()`, `PCHPDDM`
342f1580f4eSBarry Smith @*/
343d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetRHSMat(PC pc, Mat B)
344d71ae5a4SJacob Faibussowitsch {
345f1580f4eSBarry Smith   PetscFunctionBegin;
346f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
347f1580f4eSBarry Smith   if (B) {
348f1580f4eSBarry Smith     PetscValidHeaderSpecific(B, MAT_CLASSID, 2);
349f1580f4eSBarry Smith     PetscTryMethod(pc, "PCHPDDMSetRHSMat_C", (PC, Mat), (pc, B));
350f1580f4eSBarry Smith   }
3513ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
352f1580f4eSBarry Smith }
353f1580f4eSBarry Smith 
354ce78bad3SBarry Smith static PetscErrorCode PCSetFromOptions_HPDDM(PC pc, PetscOptionItems PetscOptionsObject)
355d71ae5a4SJacob Faibussowitsch {
356f1580f4eSBarry Smith   PC_HPDDM                   *data   = (PC_HPDDM *)pc->data;
357f1580f4eSBarry Smith   PC_HPDDM_Level            **levels = data->levels;
358f1580f4eSBarry Smith   char                        prefix[256];
359f1580f4eSBarry Smith   int                         i = 1;
360f1580f4eSBarry Smith   PetscMPIInt                 size, previous;
3619bb5c669SPierre Jolivet   PetscInt                    n, overlap = 1;
362f1580f4eSBarry Smith   PCHPDDMCoarseCorrectionType type;
363c8ea6600SPierre Jolivet   PetscBool                   flg = PETSC_TRUE, set;
364f1580f4eSBarry Smith 
365f1580f4eSBarry Smith   PetscFunctionBegin;
366f1580f4eSBarry Smith   if (!data->levels) {
367f1580f4eSBarry Smith     PetscCall(PetscCalloc1(PETSC_PCHPDDM_MAXLEVELS, &levels));
368f1580f4eSBarry Smith     data->levels = levels;
369f1580f4eSBarry Smith   }
370f1580f4eSBarry Smith   PetscOptionsHeadBegin(PetscOptionsObject, "PCHPDDM options");
3719bb5c669SPierre Jolivet   PetscCall(PetscOptionsBoundedInt("-pc_hpddm_harmonic_overlap", "Overlap prior to computing local harmonic extensions", "PCHPDDM", overlap, &overlap, &set, 1));
3729bb5c669SPierre Jolivet   if (!set) overlap = -1;
373f1580f4eSBarry Smith   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)pc), &size));
374f1580f4eSBarry Smith   previous = size;
375f1580f4eSBarry Smith   while (i < PETSC_PCHPDDM_MAXLEVELS) {
376f1580f4eSBarry Smith     PetscInt p = 1;
377f1580f4eSBarry Smith 
3784dfa11a4SJacob Faibussowitsch     if (!data->levels[i - 1]) PetscCall(PetscNew(data->levels + i - 1));
379f1580f4eSBarry Smith     data->levels[i - 1]->parent = data;
380f1580f4eSBarry Smith     /* if the previous level has a single process, it is not possible to coarsen further */
381f1580f4eSBarry Smith     if (previous == 1 || !flg) break;
382f1580f4eSBarry Smith     data->levels[i - 1]->nu        = 0;
383f1580f4eSBarry Smith     data->levels[i - 1]->threshold = -1.0;
384f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_nev", i));
385db4a47b3SPierre 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));
386f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_threshold", i));
387db4a47b3SPierre 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));
388f1580f4eSBarry Smith     if (i == 1) {
389aa1539e9SPierre 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");
3909bb5c669SPierre Jolivet       if (overlap != -1) {
3919bb5c669SPierre Jolivet         PetscInt nsv = 0;
392811e8887SPierre Jolivet 
3939bb5c669SPierre Jolivet         PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_svd_nsv", i));
3949bb5c669SPierre Jolivet         PetscCall(PetscOptionsBoundedInt(prefix, "Local number of deflation vectors computed by SLEPc", "SVDSetDimensions", nsv, &nsv, nullptr, 0));
3959bb5c669SPierre 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");
3969bb5c669SPierre Jolivet         if (nsv) {
3979bb5c669SPierre Jolivet           data->levels[0]->nu = nsv;
3989bb5c669SPierre Jolivet           PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_svd_relative_threshold", i));
3999bb5c669SPierre Jolivet         } else PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_relative_threshold", i));
4009bb5c669SPierre Jolivet         PetscCall(PetscOptionsReal(prefix, "Local relative threshold for selecting deflation vectors returned by SLEPc", "PCHPDDM", data->levels[0]->threshold, &data->levels[0]->threshold, nullptr));
4019bb5c669SPierre Jolivet       }
402f1580f4eSBarry Smith       PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_1_st_share_sub_ksp"));
403db4a47b3SPierre Jolivet       PetscCall(PetscOptionsBool(prefix, "Shared KSP between SLEPc ST and the fine-level subdomain solver", "PCHPDDMSetSTShareSubKSP", PETSC_FALSE, &data->share, nullptr));
404f1580f4eSBarry Smith     }
405f1580f4eSBarry Smith     /* if there is no prescribed coarsening, just break out of the loop */
4060594bca0SPierre Jolivet     if (data->levels[i - 1]->threshold <= PetscReal() && data->levels[i - 1]->nu <= 0 && !(data->deflation && i == 1)) break;
407f1580f4eSBarry Smith     else {
408f1580f4eSBarry Smith       ++i;
409f1580f4eSBarry Smith       PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_nev", i));
410f1580f4eSBarry Smith       PetscCall(PetscOptionsHasName(PetscOptionsObject->options, PetscOptionsObject->prefix, prefix, &flg));
411f1580f4eSBarry Smith       if (!flg) {
412f1580f4eSBarry Smith         PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_threshold", i));
413f1580f4eSBarry Smith         PetscCall(PetscOptionsHasName(PetscOptionsObject->options, PetscOptionsObject->prefix, prefix, &flg));
414f1580f4eSBarry Smith       }
415f1580f4eSBarry Smith       if (flg) {
416f1580f4eSBarry Smith         /* if there are coarsening options for the next level, then register it  */
417f1580f4eSBarry Smith         /* otherwise, don't to avoid having both options levels_N_p and coarse_p */
418f1580f4eSBarry Smith         PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_p", i));
419f1580f4eSBarry 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)));
420f1580f4eSBarry Smith         previous = p;
421f1580f4eSBarry Smith       }
422f1580f4eSBarry Smith     }
423f1580f4eSBarry Smith   }
424f1580f4eSBarry Smith   data->N = i;
425f1580f4eSBarry Smith   n       = 1;
426f1580f4eSBarry Smith   if (i > 1) {
427f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_coarse_p"));
428db4a47b3SPierre Jolivet     PetscCall(PetscOptionsRangeInt(prefix, "Number of processes used to assemble the coarsest operator", "PCHPDDM", n, &n, nullptr, 1, PetscMax(1, previous / 2)));
42902800ff6SPierre Jolivet #if PetscDefined(HAVE_MUMPS)
430f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "pc_hpddm_coarse_"));
431db4a47b3SPierre Jolivet     PetscCall(PetscOptionsHasName(nullptr, prefix, "-mat_mumps_use_omp_threads", &flg));
432f1580f4eSBarry Smith     if (flg) {
433f1580f4eSBarry Smith       char type[64]; /* same size as in src/ksp/pc/impls/factor/factimpl.c */
434811e8887SPierre Jolivet 
435c6a7a370SJeremy L Thompson       PetscCall(PetscStrncpy(type, n > 1 && PetscDefined(HAVE_MUMPS) ? MATSOLVERMUMPS : MATSOLVERPETSC, sizeof(type))); /* default solver for a MatMPIAIJ or a MatSeqAIJ */
436db4a47b3SPierre Jolivet       PetscCall(PetscOptionsGetString(nullptr, prefix, "-pc_factor_mat_solver_type", type, sizeof(type), nullptr));
4373ce573a3SPierre Jolivet       PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg));
438f1580f4eSBarry 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);
439f1580f4eSBarry Smith       size = n;
440f1580f4eSBarry Smith       n    = -1;
441db4a47b3SPierre Jolivet       PetscCall(PetscOptionsGetInt(nullptr, prefix, "-mat_mumps_use_omp_threads", &n, nullptr));
442f1580f4eSBarry Smith       PetscCheck(n >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Need to specify a positive integer for -%smat_mumps_use_omp_threads", prefix);
443f1580f4eSBarry 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" : "");
444f1580f4eSBarry Smith     }
445f1580f4eSBarry Smith #endif
446f1580f4eSBarry Smith     PetscCall(PetscOptionsEnum("-pc_hpddm_coarse_correction", "Type of coarse correction applied each iteration", "PCHPDDMSetCoarseCorrectionType", PCHPDDMCoarseCorrectionTypes, (PetscEnum)data->correction, (PetscEnum *)&type, &flg));
447f1580f4eSBarry Smith     if (flg) PetscCall(PCHPDDMSetCoarseCorrectionType(pc, type));
448f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_has_neumann"));
449c8ea6600SPierre Jolivet     PetscCall(PetscOptionsBool(prefix, "Is the auxiliary Mat the local Neumann matrix?", "PCHPDDMHasNeumannMat", PetscBool3ToBool(data->Neumann), &flg, &set));
450c8ea6600SPierre Jolivet     if (set) data->Neumann = PetscBoolToBool3(flg);
451f1580f4eSBarry Smith     data->log_separate = PETSC_FALSE;
452f1580f4eSBarry Smith     if (PetscDefined(USE_LOG)) {
453f1580f4eSBarry Smith       PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_log_separate"));
454db4a47b3SPierre Jolivet       PetscCall(PetscOptionsBool(prefix, "Log events level by level instead of inside PCSetUp()/KSPSolve()", nullptr, data->log_separate, &data->log_separate, nullptr));
455f1580f4eSBarry Smith     }
456f1580f4eSBarry Smith   }
457f1580f4eSBarry Smith   PetscOptionsHeadEnd();
458f1580f4eSBarry Smith   while (i < PETSC_PCHPDDM_MAXLEVELS && data->levels[i]) PetscCall(PetscFree(data->levels[i++]));
4593ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
460f1580f4eSBarry Smith }
461f1580f4eSBarry Smith 
462d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_HPDDM(PC pc, Vec x, Vec y)
463d71ae5a4SJacob Faibussowitsch {
464f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
465f1580f4eSBarry Smith 
466f1580f4eSBarry Smith   PetscFunctionBegin;
467f1580f4eSBarry Smith   PetscCall(PetscCitationsRegister(HPDDMCitation, &HPDDMCite));
468f1580f4eSBarry Smith   PetscCheck(data->levels[0]->ksp, PETSC_COMM_SELF, PETSC_ERR_PLIB, "No KSP attached to PCHPDDM");
469db4a47b3SPierre 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 */
470f1580f4eSBarry Smith   PetscCall(KSPSolve(data->levels[0]->ksp, x, y));
471db4a47b3SPierre Jolivet   if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Solve[0], data->levels[0]->ksp, nullptr, nullptr, nullptr));
4723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
473f1580f4eSBarry Smith }
474f1580f4eSBarry Smith 
475d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_HPDDM(PC pc, Mat X, Mat Y)
476d71ae5a4SJacob Faibussowitsch {
477f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
478f1580f4eSBarry Smith 
479f1580f4eSBarry Smith   PetscFunctionBegin;
480f1580f4eSBarry Smith   PetscCall(PetscCitationsRegister(HPDDMCitation, &HPDDMCite));
481f1580f4eSBarry Smith   PetscCheck(data->levels[0]->ksp, PETSC_COMM_SELF, PETSC_ERR_PLIB, "No KSP attached to PCHPDDM");
482f1580f4eSBarry Smith   PetscCall(KSPMatSolve(data->levels[0]->ksp, X, Y));
4833ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
484f1580f4eSBarry Smith }
485f1580f4eSBarry Smith 
486cc4c1da9SBarry Smith /*@
487f1580f4eSBarry Smith   PCHPDDMGetComplexities - Computes the grid and operator complexities.
488f1580f4eSBarry Smith 
489c71f06a7SPierre Jolivet   Collective
490c71f06a7SPierre Jolivet 
491f1580f4eSBarry Smith   Input Parameter:
492f1580f4eSBarry Smith . pc - preconditioner context
493f1580f4eSBarry Smith 
494f1580f4eSBarry Smith   Output Parameters:
495cc4c1da9SBarry Smith + gc - grid complexity $ \sum_i m_i / m_1 $
496cc4c1da9SBarry Smith - oc - operator complexity $ \sum_i nnz_i / nnz_1 $
497f1580f4eSBarry Smith 
498f1580f4eSBarry Smith   Level: advanced
499f1580f4eSBarry Smith 
500562efe2eSBarry Smith .seealso: [](ch_ksp), `PCMGGetGridComplexity()`, `PCHPDDM`, `PCHYPRE`, `PCGAMG`
501f1580f4eSBarry Smith @*/
502cc4c1da9SBarry Smith PetscErrorCode PCHPDDMGetComplexities(PC pc, PetscReal *gc, PetscReal *oc)
503d71ae5a4SJacob Faibussowitsch {
504f1580f4eSBarry Smith   PC_HPDDM      *data = (PC_HPDDM *)pc->data;
505f1580f4eSBarry Smith   MatInfo        info;
506f1580f4eSBarry Smith   PetscLogDouble accumulate[2]{}, nnz1 = 1.0, m1 = 1.0;
507f1580f4eSBarry Smith 
508f1580f4eSBarry Smith   PetscFunctionBegin;
509715c1178SPierre Jolivet   if (gc) {
510715c1178SPierre Jolivet     PetscAssertPointer(gc, 2);
511715c1178SPierre Jolivet     *gc = 0;
512715c1178SPierre Jolivet   }
513715c1178SPierre Jolivet   if (oc) {
514715c1178SPierre Jolivet     PetscAssertPointer(oc, 3);
515715c1178SPierre Jolivet     *oc = 0;
516715c1178SPierre Jolivet   }
517715c1178SPierre Jolivet   for (PetscInt n = 0; n < data->N; ++n) {
518f1580f4eSBarry Smith     if (data->levels[n]->ksp) {
51913044ca3SPierre Jolivet       Mat       P, A = nullptr;
520715c1178SPierre Jolivet       PetscInt  m;
52113044ca3SPierre Jolivet       PetscBool flg = PETSC_FALSE;
52213044ca3SPierre Jolivet 
523db4a47b3SPierre Jolivet       PetscCall(KSPGetOperators(data->levels[n]->ksp, nullptr, &P));
524db4a47b3SPierre Jolivet       PetscCall(MatGetSize(P, &m, nullptr));
525f1580f4eSBarry Smith       accumulate[0] += m;
526f1580f4eSBarry Smith       if (n == 0) {
527f1580f4eSBarry Smith         PetscCall(PetscObjectTypeCompareAny((PetscObject)P, &flg, MATNORMAL, MATNORMALHERMITIAN, ""));
528f1580f4eSBarry Smith         if (flg) {
529f1580f4eSBarry Smith           PetscCall(MatConvert(P, MATAIJ, MAT_INITIAL_MATRIX, &A));
530f1580f4eSBarry Smith           P = A;
53113044ca3SPierre Jolivet         } else {
53213044ca3SPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg));
53313044ca3SPierre Jolivet           PetscCall(PetscObjectReference((PetscObject)P));
534f1580f4eSBarry Smith         }
53513044ca3SPierre Jolivet       }
53613044ca3SPierre Jolivet       if (!A && flg) accumulate[1] += m * m; /* assumption that a MATSCHURCOMPLEMENT is dense if stored explicitly */
53713044ca3SPierre Jolivet       else if (P->ops->getinfo) {
538f1580f4eSBarry Smith         PetscCall(MatGetInfo(P, MAT_GLOBAL_SUM, &info));
539f1580f4eSBarry Smith         accumulate[1] += info.nz_used;
540f1580f4eSBarry Smith       }
541f1580f4eSBarry Smith       if (n == 0) {
542f1580f4eSBarry Smith         m1 = m;
54313044ca3SPierre Jolivet         if (!A && flg) nnz1 = m * m;
54413044ca3SPierre Jolivet         else if (P->ops->getinfo) nnz1 = info.nz_used;
545f1580f4eSBarry Smith         PetscCall(MatDestroy(&P));
546f1580f4eSBarry Smith       }
547f1580f4eSBarry Smith     }
548f1580f4eSBarry Smith   }
549715c1178SPierre Jolivet   /* only process #0 has access to the full hierarchy by construction, so broadcast to ensure consistent outputs */
550715c1178SPierre Jolivet   PetscCallMPI(MPI_Bcast(accumulate, 2, MPIU_PETSCLOGDOUBLE, 0, PetscObjectComm((PetscObject)pc)));
551715c1178SPierre Jolivet   if (gc) *gc = static_cast<PetscReal>(accumulate[0] / m1);
552715c1178SPierre Jolivet   if (oc) *oc = static_cast<PetscReal>(accumulate[1] / nnz1);
5533ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
554f1580f4eSBarry Smith }
555f1580f4eSBarry Smith 
556d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCView_HPDDM(PC pc, PetscViewer viewer)
557d71ae5a4SJacob Faibussowitsch {
558f1580f4eSBarry Smith   PC_HPDDM         *data = (PC_HPDDM *)pc->data;
559f1580f4eSBarry Smith   PetscViewer       subviewer;
560aa21023fSPierre Jolivet   PetscViewerFormat format;
561f1580f4eSBarry Smith   PetscSubcomm      subcomm;
562f1580f4eSBarry Smith   PetscReal         oc, gc;
563811e8887SPierre Jolivet   PetscInt          tabs;
564f1580f4eSBarry Smith   PetscMPIInt       size, color, rank;
565aa21023fSPierre Jolivet   PetscBool         flg;
566aa21023fSPierre Jolivet   const char       *name;
567f1580f4eSBarry Smith 
568f1580f4eSBarry Smith   PetscFunctionBegin;
569aa21023fSPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &flg));
570aa21023fSPierre Jolivet   if (flg) {
571f1580f4eSBarry Smith     PetscCall(PetscViewerASCIIPrintf(viewer, "level%s: %" PetscInt_FMT "\n", data->N > 1 ? "s" : "", data->N));
572f1580f4eSBarry Smith     PetscCall(PCHPDDMGetComplexities(pc, &gc, &oc));
573f1580f4eSBarry Smith     if (data->N > 1) {
574f1580f4eSBarry Smith       if (!data->deflation) {
575c8ea6600SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "Neumann matrix attached? %s\n", PetscBools[PetscBool3ToBool(data->Neumann)]));
576f1580f4eSBarry Smith         PetscCall(PetscViewerASCIIPrintf(viewer, "shared subdomain KSP between SLEPc and PETSc? %s\n", PetscBools[data->share]));
577f1580f4eSBarry Smith       } else PetscCall(PetscViewerASCIIPrintf(viewer, "user-supplied deflation matrix\n"));
578f1580f4eSBarry Smith       PetscCall(PetscViewerASCIIPrintf(viewer, "coarse correction: %s\n", PCHPDDMCoarseCorrectionTypes[data->correction]));
579f1580f4eSBarry 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" : ""));
580f1580f4eSBarry Smith       PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
581f1580f4eSBarry Smith       PetscCall(PetscViewerASCIISetTab(viewer, 0));
582811e8887SPierre Jolivet       for (PetscInt i = 1; i < data->N; ++i) {
583f1580f4eSBarry Smith         PetscCall(PetscViewerASCIIPrintf(viewer, " %" PetscInt_FMT, data->levels[i - 1]->nu));
5840594bca0SPierre Jolivet         if (data->levels[i - 1]->threshold > static_cast<PetscReal>(-0.1)) PetscCall(PetscViewerASCIIPrintf(viewer, " (%g)", (double)data->levels[i - 1]->threshold));
585f1580f4eSBarry Smith       }
586f1580f4eSBarry Smith       PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
587f1580f4eSBarry Smith       PetscCall(PetscViewerASCIISetTab(viewer, tabs));
588f1580f4eSBarry Smith     }
589f1580f4eSBarry Smith     PetscCall(PetscViewerASCIIPrintf(viewer, "grid and operator complexities: %g %g\n", (double)gc, (double)oc));
5901fe44b27SPierre Jolivet     PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)pc), &size));
591f1580f4eSBarry Smith     if (data->levels[0]->ksp) {
592f1580f4eSBarry Smith       PetscCall(KSPView(data->levels[0]->ksp, viewer));
593f1580f4eSBarry Smith       if (data->levels[0]->pc) PetscCall(PCView(data->levels[0]->pc, viewer));
594eea1a2f1SRaphael Zanella       PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)pc), &rank));
595811e8887SPierre Jolivet       for (PetscInt i = 1; i < data->N; ++i) {
596f1580f4eSBarry Smith         if (data->levels[i]->ksp) color = 1;
597f1580f4eSBarry Smith         else color = 0;
598f1580f4eSBarry Smith         PetscCall(PetscSubcommCreate(PetscObjectComm((PetscObject)pc), &subcomm));
599f1580f4eSBarry Smith         PetscCall(PetscSubcommSetNumber(subcomm, PetscMin(size, 2)));
600f1580f4eSBarry Smith         PetscCall(PetscSubcommSetTypeGeneral(subcomm, color, rank));
601f1580f4eSBarry Smith         PetscCall(PetscViewerASCIIPushTab(viewer));
602f1580f4eSBarry Smith         PetscCall(PetscViewerGetSubViewer(viewer, PetscSubcommChild(subcomm), &subviewer));
603f1580f4eSBarry Smith         if (color == 1) {
604f1580f4eSBarry Smith           PetscCall(KSPView(data->levels[i]->ksp, subviewer));
605f1580f4eSBarry Smith           if (data->levels[i]->pc) PetscCall(PCView(data->levels[i]->pc, subviewer));
606f1580f4eSBarry Smith           PetscCall(PetscViewerFlush(subviewer));
607f1580f4eSBarry Smith         }
608f1580f4eSBarry Smith         PetscCall(PetscViewerRestoreSubViewer(viewer, PetscSubcommChild(subcomm), &subviewer));
609f1580f4eSBarry Smith         PetscCall(PetscViewerASCIIPopTab(viewer));
610f1580f4eSBarry Smith         PetscCall(PetscSubcommDestroy(&subcomm));
611f1580f4eSBarry Smith       }
612f1580f4eSBarry Smith     }
613aa21023fSPierre Jolivet     PetscCall(PetscViewerGetFormat(viewer, &format));
614aa21023fSPierre Jolivet     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
615aa21023fSPierre Jolivet       PetscCall(PetscViewerFileGetName(viewer, &name));
616aa21023fSPierre Jolivet       if (name) {
617eea1a2f1SRaphael Zanella         Mat             aux[2];
618aa21023fSPierre Jolivet         IS              is;
619eea1a2f1SRaphael Zanella         const PetscInt *indices;
620eea1a2f1SRaphael Zanella         PetscInt        m, n, sizes[5] = {pc->mat->rmap->n, pc->mat->cmap->n, pc->mat->rmap->N, pc->mat->cmap->N, 0};
621aa21023fSPierre Jolivet         char           *tmp;
622aa21023fSPierre Jolivet         std::string     prefix, suffix;
623aa21023fSPierre Jolivet         size_t          pos;
624aa21023fSPierre Jolivet 
625aa21023fSPierre Jolivet         PetscCall(PetscStrstr(name, ".", &tmp));
626aa21023fSPierre Jolivet         if (tmp) {
627aa21023fSPierre Jolivet           pos    = std::distance(const_cast<char *>(name), tmp);
628aa21023fSPierre Jolivet           prefix = std::string(name, pos);
629aa21023fSPierre Jolivet           suffix = std::string(name + pos + 1);
630aa21023fSPierre Jolivet         } else prefix = name;
631aa21023fSPierre Jolivet         if (data->aux) {
632eea1a2f1SRaphael Zanella           PetscCall(MatGetSize(data->aux, &m, &n));
633eea1a2f1SRaphael Zanella           PetscCall(MatCreate(PetscObjectComm((PetscObject)pc), aux));
634eea1a2f1SRaphael Zanella           PetscCall(MatSetSizes(aux[0], m, n, PETSC_DETERMINE, PETSC_DETERMINE));
635eea1a2f1SRaphael Zanella           PetscCall(PetscObjectBaseTypeCompare((PetscObject)data->aux, MATSEQAIJ, &flg));
636eea1a2f1SRaphael Zanella           if (flg) PetscCall(MatSetType(aux[0], MATMPIAIJ));
637eea1a2f1SRaphael Zanella           else {
638eea1a2f1SRaphael Zanella             PetscCall(PetscObjectBaseTypeCompare((PetscObject)data->aux, MATSEQBAIJ, &flg));
639eea1a2f1SRaphael Zanella             if (flg) PetscCall(MatSetType(aux[0], MATMPIBAIJ));
640eea1a2f1SRaphael Zanella             else {
641eea1a2f1SRaphael Zanella               PetscCall(PetscObjectBaseTypeCompare((PetscObject)data->aux, MATSEQSBAIJ, &flg));
642eea1a2f1SRaphael 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);
643eea1a2f1SRaphael Zanella               PetscCall(MatSetType(aux[0], MATMPISBAIJ));
644eea1a2f1SRaphael Zanella             }
645eea1a2f1SRaphael Zanella           }
646eea1a2f1SRaphael Zanella           PetscCall(MatSetBlockSizesFromMats(aux[0], data->aux, data->aux));
647eea1a2f1SRaphael Zanella           PetscCall(MatAssemblyBegin(aux[0], MAT_FINAL_ASSEMBLY));
648eea1a2f1SRaphael Zanella           PetscCall(MatAssemblyEnd(aux[0], MAT_FINAL_ASSEMBLY));
649eea1a2f1SRaphael Zanella           PetscCall(MatGetDiagonalBlock(aux[0], aux + 1));
650eea1a2f1SRaphael Zanella           PetscCall(MatCopy(data->aux, aux[1], DIFFERENT_NONZERO_PATTERN));
651eea1a2f1SRaphael Zanella           PetscCall(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)pc), std::string(prefix + "_aux_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer));
652eea1a2f1SRaphael Zanella           PetscCall(MatView(aux[0], subviewer));
653aa21023fSPierre Jolivet           PetscCall(PetscViewerDestroy(&subviewer));
654eea1a2f1SRaphael Zanella           PetscCall(MatDestroy(aux));
655aa21023fSPierre Jolivet         }
656aa21023fSPierre Jolivet         if (data->is) {
657eea1a2f1SRaphael Zanella           PetscCall(ISGetIndices(data->is, &indices));
658eea1a2f1SRaphael Zanella           PetscCall(ISGetSize(data->is, sizes + 4));
659eea1a2f1SRaphael Zanella           PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)pc), sizes[4], indices, PETSC_USE_POINTER, &is));
660eea1a2f1SRaphael Zanella           PetscCall(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)pc), std::string(prefix + "_is_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer));
661eea1a2f1SRaphael Zanella           PetscCall(ISView(is, subviewer));
662aa21023fSPierre Jolivet           PetscCall(PetscViewerDestroy(&subviewer));
663eea1a2f1SRaphael Zanella           PetscCall(ISDestroy(&is));
664eea1a2f1SRaphael Zanella           PetscCall(ISRestoreIndices(data->is, &indices));
665aa21023fSPierre Jolivet         }
666eea1a2f1SRaphael Zanella         PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)pc), PETSC_STATIC_ARRAY_LENGTH(sizes), sizes, PETSC_USE_POINTER, &is));
667eea1a2f1SRaphael Zanella         PetscCall(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)pc), std::string(prefix + "_sizes_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer));
668aa21023fSPierre Jolivet         PetscCall(ISView(is, subviewer));
669aa21023fSPierre Jolivet         PetscCall(PetscViewerDestroy(&subviewer));
670aa21023fSPierre Jolivet         PetscCall(ISDestroy(&is));
671aa21023fSPierre Jolivet       }
672aa21023fSPierre Jolivet     }
673f1580f4eSBarry Smith   }
6743ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
675f1580f4eSBarry Smith }
676f1580f4eSBarry Smith 
677d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCPreSolve_HPDDM(PC pc, KSP ksp, Vec, Vec)
678d71ae5a4SJacob Faibussowitsch {
679f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
680f1580f4eSBarry Smith   Mat       A;
681cdbd50ebSPierre Jolivet   PetscBool flg;
682f1580f4eSBarry Smith 
683f1580f4eSBarry Smith   PetscFunctionBegin;
684f1580f4eSBarry Smith   if (ksp) {
685f1580f4eSBarry Smith     PetscCall(PetscObjectTypeCompare((PetscObject)ksp, KSPLSQR, &flg));
686f1580f4eSBarry Smith     if (flg && !data->normal) {
687db4a47b3SPierre Jolivet       PetscCall(KSPGetOperators(ksp, &A, nullptr));
688db4a47b3SPierre Jolivet       PetscCall(MatCreateVecs(A, nullptr, &data->normal)); /* temporary Vec used in PCApply_HPDDMShell() for coarse grid corrections */
689cdbd50ebSPierre Jolivet     } else if (!flg) {
690cdbd50ebSPierre Jolivet       PetscCall(PetscObjectTypeCompareAny((PetscObject)ksp, &flg, KSPCG, KSPGROPPCG, KSPPIPECG, KSPPIPECGRR, KSPPIPELCG, KSPPIPEPRCG, KSPPIPECG2, KSPSTCG, KSPFCG, KSPPIPEFCG, KSPMINRES, KSPNASH, KSPSYMMLQ, ""));
691cdbd50ebSPierre Jolivet       if (!flg) {
692cdbd50ebSPierre Jolivet         PetscCall(PetscObjectTypeCompare((PetscObject)ksp, KSPHPDDM, &flg));
693cdbd50ebSPierre Jolivet         if (flg) {
694cdbd50ebSPierre Jolivet           KSPHPDDMType type;
695811e8887SPierre Jolivet 
696cdbd50ebSPierre Jolivet           PetscCall(KSPHPDDMGetType(ksp, &type));
697cdbd50ebSPierre Jolivet           flg = (type == KSP_HPDDM_TYPE_CG || type == KSP_HPDDM_TYPE_BCG || type == KSP_HPDDM_TYPE_BFBCG ? PETSC_TRUE : PETSC_FALSE);
698cdbd50ebSPierre Jolivet         }
699cdbd50ebSPierre Jolivet       }
700cdbd50ebSPierre Jolivet     }
701cdbd50ebSPierre Jolivet     if (flg) {
702cdbd50ebSPierre Jolivet       if (data->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED) {
703cdbd50ebSPierre Jolivet         PetscCall(PetscOptionsHasName(((PetscObject)pc)->options, ((PetscObject)pc)->prefix, "-pc_hpddm_coarse_correction", &flg));
704cdbd50ebSPierre 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",
705cdbd50ebSPierre Jolivet                    PCHPDDMCoarseCorrectionTypes[data->correction], ((PetscObject)ksp)->type_name, ((PetscObject)pc)->prefix ? ((PetscObject)pc)->prefix : "", PCHPDDMCoarseCorrectionTypes[data->correction], PCHPDDMCoarseCorrectionTypes[PC_HPDDM_COARSE_CORRECTION_BALANCED]);
706cdbd50ebSPierre Jolivet       }
707cdbd50ebSPierre Jolivet       for (PetscInt n = 0; n < data->N; ++n) {
708cdbd50ebSPierre Jolivet         if (data->levels[n]->pc) {
709cdbd50ebSPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)data->levels[n]->pc, PCASM, &flg));
710cdbd50ebSPierre Jolivet           if (flg) {
711cdbd50ebSPierre Jolivet             PCASMType type;
712811e8887SPierre Jolivet 
713cdbd50ebSPierre Jolivet             PetscCall(PCASMGetType(data->levels[n]->pc, &type));
714cdbd50ebSPierre Jolivet             if (type == PC_ASM_RESTRICT || type == PC_ASM_INTERPOLATE) {
715cdbd50ebSPierre Jolivet               PetscCall(PetscOptionsHasName(((PetscObject)data->levels[n]->pc)->options, ((PetscObject)data->levels[n]->pc)->prefix, "-pc_asm_type", &flg));
716cdbd50ebSPierre 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],
717cdbd50ebSPierre Jolivet                          ((PetscObject)ksp)->type_name, ((PetscObject)data->levels[n]->pc)->prefix, PCASMTypes[type], PCASMTypes[PC_ASM_BASIC]);
718cdbd50ebSPierre Jolivet             }
719cdbd50ebSPierre Jolivet           }
720cdbd50ebSPierre Jolivet         }
721cdbd50ebSPierre Jolivet       }
722f1580f4eSBarry Smith     }
723f1580f4eSBarry Smith   }
7243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
725f1580f4eSBarry Smith }
726f1580f4eSBarry Smith 
727d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetUp_HPDDMShell(PC pc)
728d71ae5a4SJacob Faibussowitsch {
729f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
730f1580f4eSBarry Smith   Mat             A, P;
731f1580f4eSBarry Smith   Vec             x;
732f1580f4eSBarry Smith   const char     *pcpre;
733f1580f4eSBarry Smith 
734f1580f4eSBarry Smith   PetscFunctionBegin;
735f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
736f1580f4eSBarry Smith   PetscCall(KSPGetOptionsPrefix(ctx->ksp, &pcpre));
737f1580f4eSBarry Smith   PetscCall(KSPGetOperators(ctx->ksp, &A, &P));
738f1580f4eSBarry Smith   /* smoother */
739f1580f4eSBarry Smith   PetscCall(PCSetOptionsPrefix(ctx->pc, pcpre));
740f1580f4eSBarry Smith   PetscCall(PCSetOperators(ctx->pc, A, P));
741f1580f4eSBarry Smith   if (!ctx->v[0]) {
742f1580f4eSBarry Smith     PetscCall(VecDuplicateVecs(ctx->D, 1, &ctx->v[0]));
743f1580f4eSBarry Smith     if (!std::is_same<PetscScalar, PetscReal>::value) PetscCall(VecDestroy(&ctx->D));
744db4a47b3SPierre Jolivet     PetscCall(MatCreateVecs(A, &x, nullptr));
745f1580f4eSBarry Smith     PetscCall(VecDuplicateVecs(x, 2, &ctx->v[1]));
746f1580f4eSBarry Smith     PetscCall(VecDestroy(&x));
747f1580f4eSBarry Smith   }
748f1580f4eSBarry Smith   std::fill_n(ctx->V, 3, nullptr);
7493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
750f1580f4eSBarry Smith }
751f1580f4eSBarry Smith 
752f1580f4eSBarry Smith template <class Type, typename std::enable_if<std::is_same<Type, Vec>::value>::type * = nullptr>
753d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMDeflate_Private(PC pc, Type x, Type y)
754d71ae5a4SJacob Faibussowitsch {
755f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
756f1580f4eSBarry Smith   PetscScalar    *out;
757f1580f4eSBarry Smith 
758f1580f4eSBarry Smith   PetscFunctionBegin;
759f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
760f1580f4eSBarry Smith   /* going from PETSc to HPDDM numbering */
761f1580f4eSBarry Smith   PetscCall(VecScatterBegin(ctx->scatter, x, ctx->v[0][0], INSERT_VALUES, SCATTER_FORWARD));
762f1580f4eSBarry Smith   PetscCall(VecScatterEnd(ctx->scatter, x, ctx->v[0][0], INSERT_VALUES, SCATTER_FORWARD));
763f1580f4eSBarry Smith   PetscCall(VecGetArrayWrite(ctx->v[0][0], &out));
764db4a47b3SPierre Jolivet   ctx->P->deflation<false>(nullptr, out, 1); /* y = Q x */
765f1580f4eSBarry Smith   PetscCall(VecRestoreArrayWrite(ctx->v[0][0], &out));
766f1580f4eSBarry Smith   /* going from HPDDM to PETSc numbering */
767f1580f4eSBarry Smith   PetscCall(VecScatterBegin(ctx->scatter, ctx->v[0][0], y, INSERT_VALUES, SCATTER_REVERSE));
768f1580f4eSBarry Smith   PetscCall(VecScatterEnd(ctx->scatter, ctx->v[0][0], y, INSERT_VALUES, SCATTER_REVERSE));
7693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
770f1580f4eSBarry Smith }
771f1580f4eSBarry Smith 
772f1580f4eSBarry Smith template <class Type, typename std::enable_if<std::is_same<Type, Mat>::value>::type * = nullptr>
773d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMDeflate_Private(PC pc, Type X, Type Y)
774d71ae5a4SJacob Faibussowitsch {
775f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
776f1580f4eSBarry Smith   Vec             vX, vY, vC;
777f1580f4eSBarry Smith   PetscScalar    *out;
778811e8887SPierre Jolivet   PetscInt        N;
779f1580f4eSBarry Smith 
780f1580f4eSBarry Smith   PetscFunctionBegin;
781f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
782db4a47b3SPierre Jolivet   PetscCall(MatGetSize(X, nullptr, &N));
783f1580f4eSBarry Smith   /* going from PETSc to HPDDM numbering */
784811e8887SPierre Jolivet   for (PetscInt i = 0; i < N; ++i) {
785f1580f4eSBarry Smith     PetscCall(MatDenseGetColumnVecRead(X, i, &vX));
786f1580f4eSBarry Smith     PetscCall(MatDenseGetColumnVecWrite(ctx->V[0], i, &vC));
787f1580f4eSBarry Smith     PetscCall(VecScatterBegin(ctx->scatter, vX, vC, INSERT_VALUES, SCATTER_FORWARD));
788f1580f4eSBarry Smith     PetscCall(VecScatterEnd(ctx->scatter, vX, vC, INSERT_VALUES, SCATTER_FORWARD));
789f1580f4eSBarry Smith     PetscCall(MatDenseRestoreColumnVecWrite(ctx->V[0], i, &vC));
790f1580f4eSBarry Smith     PetscCall(MatDenseRestoreColumnVecRead(X, i, &vX));
791f1580f4eSBarry Smith   }
792f1580f4eSBarry Smith   PetscCall(MatDenseGetArrayWrite(ctx->V[0], &out));
793db4a47b3SPierre Jolivet   ctx->P->deflation<false>(nullptr, out, N); /* Y = Q X */
794f1580f4eSBarry Smith   PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &out));
795f1580f4eSBarry Smith   /* going from HPDDM to PETSc numbering */
796811e8887SPierre Jolivet   for (PetscInt i = 0; i < N; ++i) {
797f1580f4eSBarry Smith     PetscCall(MatDenseGetColumnVecRead(ctx->V[0], i, &vC));
798f1580f4eSBarry Smith     PetscCall(MatDenseGetColumnVecWrite(Y, i, &vY));
799f1580f4eSBarry Smith     PetscCall(VecScatterBegin(ctx->scatter, vC, vY, INSERT_VALUES, SCATTER_REVERSE));
800f1580f4eSBarry Smith     PetscCall(VecScatterEnd(ctx->scatter, vC, vY, INSERT_VALUES, SCATTER_REVERSE));
801f1580f4eSBarry Smith     PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &vY));
802f1580f4eSBarry Smith     PetscCall(MatDenseRestoreColumnVecRead(ctx->V[0], i, &vC));
803f1580f4eSBarry Smith   }
8043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
805f1580f4eSBarry Smith }
806f1580f4eSBarry Smith 
807f1580f4eSBarry Smith /*
808aa1539e9SPierre 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.
809f1580f4eSBarry Smith 
810f1580f4eSBarry Smith .vb
811f1580f4eSBarry Smith    (1) y =                Pmat^-1              x + Q x,
812f1580f4eSBarry Smith    (2) y =                Pmat^-1 (I - Amat Q) x + Q x (default),
813aa1539e9SPierre Jolivet    (3) y = (I - Q Amat^T) Pmat^-1 (I - Amat Q) x + Q x,
814aa1539e9SPierre Jolivet    (4) y =                Pmat^-1              x      .
815f1580f4eSBarry Smith .ve
816f1580f4eSBarry Smith 
817f1580f4eSBarry Smith    Input Parameters:
818f1580f4eSBarry Smith +     pc - preconditioner context
819f1580f4eSBarry Smith -     x - input vector
820f1580f4eSBarry Smith 
821f1580f4eSBarry Smith    Output Parameter:
822f1580f4eSBarry Smith .     y - output vector
823f1580f4eSBarry Smith 
824f1580f4eSBarry Smith    Notes:
825f1580f4eSBarry 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.
826f1580f4eSBarry 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.
827f1580f4eSBarry 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.
828f1580f4eSBarry Smith 
829f1580f4eSBarry Smith    Level: advanced
830f1580f4eSBarry Smith 
831f1580f4eSBarry Smith    Developer Note:
832f1580f4eSBarry 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
833f1580f4eSBarry Smith    to trigger it. Likely the manual page is `PCHPDDM`
834f1580f4eSBarry Smith 
835562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMCoarseCorrectionType`
836f1580f4eSBarry Smith */
837d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_HPDDMShell(PC pc, Vec x, Vec y)
838d71ae5a4SJacob Faibussowitsch {
839f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
840f1580f4eSBarry Smith   Mat             A;
841f1580f4eSBarry Smith 
842f1580f4eSBarry Smith   PetscFunctionBegin;
843f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
844f1580f4eSBarry Smith   PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object");
845db4a47b3SPierre Jolivet   PetscCall(KSPGetOperators(ctx->ksp, &A, nullptr));
846aa1539e9SPierre Jolivet   if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_NONE) PetscCall(PCApply(ctx->pc, x, y)); /* y = M^-1 x */
847aa1539e9SPierre Jolivet   else {
848f1580f4eSBarry Smith     PetscCall(PCHPDDMDeflate_Private(pc, x, y)); /* y = Q x */
849f1580f4eSBarry Smith     if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
850f1580f4eSBarry Smith       if (!ctx->parent->normal || ctx != ctx->parent->levels[0]) PetscCall(MatMult(A, y, ctx->v[1][0])); /* y = A Q x     */
851f1580f4eSBarry Smith       else { /* KSPLSQR and finest level */ PetscCall(MatMult(A, y, ctx->parent->normal));               /* y = A Q x     */
852f1580f4eSBarry Smith         PetscCall(MatMultHermitianTranspose(A, ctx->parent->normal, ctx->v[1][0]));                      /* y = A^T A Q x */
853f1580f4eSBarry Smith       }
854f1580f4eSBarry Smith       PetscCall(VecWAXPY(ctx->v[1][1], -1.0, ctx->v[1][0], x)); /* y = (I - A Q) x                             */
855f1580f4eSBarry Smith       PetscCall(PCApply(ctx->pc, ctx->v[1][1], ctx->v[1][0]));  /* y = M^-1 (I - A Q) x                        */
856f1580f4eSBarry Smith       if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
857f1580f4eSBarry 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 */
858f1580f4eSBarry Smith         else {
859f1580f4eSBarry Smith           PetscCall(MatMult(A, ctx->v[1][0], ctx->parent->normal));
860f1580f4eSBarry Smith           PetscCall(MatMultHermitianTranspose(A, ctx->parent->normal, ctx->v[1][1])); /* z = A^T A y           */
861f1580f4eSBarry Smith         }
862f1580f4eSBarry Smith         PetscCall(PCHPDDMDeflate_Private(pc, ctx->v[1][1], ctx->v[1][1]));
863f1580f4eSBarry 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      */
864f1580f4eSBarry Smith       } else PetscCall(VecAXPY(y, 1.0, ctx->v[1][0]));                         /* y = Q M^-1 (I - A Q) x + Q x */
865f1580f4eSBarry Smith     } else {
866f1580f4eSBarry 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);
867f1580f4eSBarry Smith       PetscCall(PCApply(ctx->pc, x, ctx->v[1][0]));
868f1580f4eSBarry Smith       PetscCall(VecAXPY(y, 1.0, ctx->v[1][0])); /* y = M^-1 x + Q x */
869f1580f4eSBarry Smith     }
870aa1539e9SPierre Jolivet   }
8713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
872f1580f4eSBarry Smith }
873f1580f4eSBarry Smith 
874f1580f4eSBarry Smith /*
875f1580f4eSBarry Smith      PCMatApply_HPDDMShell - Variant of PCApply_HPDDMShell() for blocks of vectors.
876f1580f4eSBarry Smith 
877f1580f4eSBarry Smith    Input Parameters:
878f1580f4eSBarry Smith +     pc - preconditioner context
879f1580f4eSBarry Smith -     X - block of input vectors
880f1580f4eSBarry Smith 
881f1580f4eSBarry Smith    Output Parameter:
882f1580f4eSBarry Smith .     Y - block of output vectors
883f1580f4eSBarry Smith 
884f1580f4eSBarry Smith    Level: advanced
885f1580f4eSBarry Smith 
886562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCApply_HPDDMShell()`, `PCHPDDMCoarseCorrectionType`
887f1580f4eSBarry Smith */
888d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_HPDDMShell(PC pc, Mat X, Mat Y)
889d71ae5a4SJacob Faibussowitsch {
890f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
891f1580f4eSBarry Smith   Mat             A, *ptr;
892db4a47b3SPierre Jolivet   PetscContainer  container = nullptr;
893f1580f4eSBarry Smith   PetscScalar    *array;
894f1580f4eSBarry Smith   PetscInt        m, M, N, prev = 0;
895f1580f4eSBarry Smith   PetscBool       reset = PETSC_FALSE;
896f1580f4eSBarry Smith 
897f1580f4eSBarry Smith   PetscFunctionBegin;
898f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
899f1580f4eSBarry Smith   PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object");
900db4a47b3SPierre Jolivet   PetscCall(MatGetSize(X, nullptr, &N));
901db4a47b3SPierre Jolivet   PetscCall(KSPGetOperators(ctx->ksp, &A, nullptr));
902aa1539e9SPierre Jolivet   if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_NONE) PetscCall(PCMatApply(ctx->pc, X, Y));
903aa1539e9SPierre Jolivet   else {
904f1580f4eSBarry Smith     PetscCall(PetscObjectQuery((PetscObject)A, "_HPDDM_MatProduct", (PetscObject *)&container));
905f1580f4eSBarry Smith     if (container) { /* MatProduct container already attached */
906f1580f4eSBarry Smith       PetscCall(PetscContainerGetPointer(container, (void **)&ptr));
907f1580f4eSBarry Smith       if (ptr[1] != ctx->V[2]) /* Mat has changed or may have been set first in KSPHPDDM */
908f1580f4eSBarry Smith         for (m = 0; m < 2; ++m) {
909f1580f4eSBarry Smith           PetscCall(MatDestroy(ctx->V + m + 1));
910f1580f4eSBarry Smith           ctx->V[m + 1] = ptr[m];
911f1580f4eSBarry Smith           PetscCall(PetscObjectReference((PetscObject)ctx->V[m + 1]));
912f1580f4eSBarry Smith         }
913f1580f4eSBarry Smith     }
914db4a47b3SPierre Jolivet     if (ctx->V[1]) PetscCall(MatGetSize(ctx->V[1], nullptr, &prev));
915f1580f4eSBarry Smith     if (N != prev || !ctx->V[0]) {
916f1580f4eSBarry Smith       PetscCall(MatDestroy(ctx->V));
917f1580f4eSBarry Smith       PetscCall(VecGetLocalSize(ctx->v[0][0], &m));
918db4a47b3SPierre Jolivet       PetscCall(MatCreateDense(PetscObjectComm((PetscObject)pc), m, PETSC_DECIDE, PETSC_DECIDE, N, nullptr, ctx->V));
919f1580f4eSBarry Smith       if (N != prev) {
920f1580f4eSBarry Smith         PetscCall(MatDestroy(ctx->V + 1));
921f1580f4eSBarry Smith         PetscCall(MatDestroy(ctx->V + 2));
922db4a47b3SPierre Jolivet         PetscCall(MatGetLocalSize(X, &m, nullptr));
923db4a47b3SPierre Jolivet         PetscCall(MatGetSize(X, &M, nullptr));
924f1580f4eSBarry Smith         if (ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) PetscCall(MatDenseGetArrayWrite(ctx->V[0], &array));
925db4a47b3SPierre Jolivet         else array = nullptr;
926f1580f4eSBarry Smith         PetscCall(MatCreateDense(PetscObjectComm((PetscObject)pc), m, PETSC_DECIDE, M, N, array, ctx->V + 1));
927f1580f4eSBarry Smith         if (ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &array));
928db4a47b3SPierre Jolivet         PetscCall(MatCreateDense(PetscObjectComm((PetscObject)pc), m, PETSC_DECIDE, M, N, nullptr, ctx->V + 2));
929db4a47b3SPierre Jolivet         PetscCall(MatProductCreateWithMat(A, Y, nullptr, ctx->V[1]));
930f1580f4eSBarry Smith         PetscCall(MatProductSetType(ctx->V[1], MATPRODUCT_AB));
931f1580f4eSBarry Smith         PetscCall(MatProductSetFromOptions(ctx->V[1]));
932f1580f4eSBarry Smith         PetscCall(MatProductSymbolic(ctx->V[1]));
933715c1178SPierre 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() */
934715c1178SPierre 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 */
935f1580f4eSBarry Smith       }
936f1580f4eSBarry Smith       if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
937db4a47b3SPierre Jolivet         PetscCall(MatProductCreateWithMat(A, ctx->V[1], nullptr, ctx->V[2]));
938f1580f4eSBarry Smith         PetscCall(MatProductSetType(ctx->V[2], MATPRODUCT_AtB));
939f1580f4eSBarry Smith         PetscCall(MatProductSetFromOptions(ctx->V[2]));
940f1580f4eSBarry Smith         PetscCall(MatProductSymbolic(ctx->V[2]));
941f1580f4eSBarry Smith       }
942f1580f4eSBarry Smith       ctx->P->start(N);
943f1580f4eSBarry Smith     }
944f1580f4eSBarry Smith     if (N == prev || container) { /* when MatProduct container is attached, always need to MatProductReplaceMats() since KSPHPDDM may have replaced the Mat as well */
945db4a47b3SPierre Jolivet       PetscCall(MatProductReplaceMats(nullptr, Y, nullptr, ctx->V[1]));
946f1580f4eSBarry Smith       if (container && ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) {
947f1580f4eSBarry Smith         PetscCall(MatDenseGetArrayWrite(ctx->V[0], &array));
948f1580f4eSBarry Smith         PetscCall(MatDensePlaceArray(ctx->V[1], array));
949f1580f4eSBarry Smith         PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &array));
950f1580f4eSBarry Smith         reset = PETSC_TRUE;
951f1580f4eSBarry Smith       }
952f1580f4eSBarry Smith     }
953f1580f4eSBarry Smith     PetscCall(PCHPDDMDeflate_Private(pc, X, Y));
954f1580f4eSBarry Smith     if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
955f1580f4eSBarry Smith       PetscCall(MatProductNumeric(ctx->V[1]));
956f1580f4eSBarry Smith       PetscCall(MatCopy(ctx->V[1], ctx->V[2], SAME_NONZERO_PATTERN));
957f1580f4eSBarry Smith       PetscCall(MatAXPY(ctx->V[2], -1.0, X, SAME_NONZERO_PATTERN));
958f1580f4eSBarry Smith       PetscCall(PCMatApply(ctx->pc, ctx->V[2], ctx->V[1]));
959f1580f4eSBarry Smith       if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) {
960f1580f4eSBarry Smith         PetscCall(MatProductNumeric(ctx->V[2]));
961f1580f4eSBarry Smith         PetscCall(PCHPDDMDeflate_Private(pc, ctx->V[2], ctx->V[2]));
962f1580f4eSBarry Smith         PetscCall(MatAXPY(ctx->V[1], -1.0, ctx->V[2], SAME_NONZERO_PATTERN));
963f1580f4eSBarry Smith       }
964f1580f4eSBarry Smith       PetscCall(MatAXPY(Y, -1.0, ctx->V[1], SAME_NONZERO_PATTERN));
965f1580f4eSBarry Smith     } else {
966f1580f4eSBarry 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);
967f1580f4eSBarry Smith       PetscCall(PCMatApply(ctx->pc, X, ctx->V[1]));
968f1580f4eSBarry Smith       PetscCall(MatAXPY(Y, 1.0, ctx->V[1], SAME_NONZERO_PATTERN));
969f1580f4eSBarry Smith     }
970f1580f4eSBarry Smith     if (reset) PetscCall(MatDenseResetArray(ctx->V[1]));
971aa1539e9SPierre Jolivet   }
9723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
973f1580f4eSBarry Smith }
974f1580f4eSBarry Smith 
975d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCDestroy_HPDDMShell(PC pc)
976d71ae5a4SJacob Faibussowitsch {
977f1580f4eSBarry Smith   PC_HPDDM_Level *ctx;
978f1580f4eSBarry Smith 
979f1580f4eSBarry Smith   PetscFunctionBegin;
980f1580f4eSBarry Smith   PetscCall(PCShellGetContext(pc, &ctx));
981f1580f4eSBarry Smith   PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(ctx, PETSC_TRUE));
982f1580f4eSBarry Smith   PetscCall(VecDestroyVecs(1, &ctx->v[0]));
983f1580f4eSBarry Smith   PetscCall(VecDestroyVecs(2, &ctx->v[1]));
984f4f49eeaSPierre Jolivet   PetscCall(PetscObjectCompose((PetscObject)ctx->pc->mat, "_HPDDM_MatProduct", nullptr));
985f1580f4eSBarry Smith   PetscCall(MatDestroy(ctx->V));
986f1580f4eSBarry Smith   PetscCall(MatDestroy(ctx->V + 1));
987f1580f4eSBarry Smith   PetscCall(MatDestroy(ctx->V + 2));
988f1580f4eSBarry Smith   PetscCall(VecDestroy(&ctx->D));
98901e3c840SPierre Jolivet   PetscCall(PetscSFDestroy(&ctx->scatter));
990f1580f4eSBarry Smith   PetscCall(PCDestroy(&ctx->pc));
9913ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
992f1580f4eSBarry Smith }
993f1580f4eSBarry Smith 
9949bb5c669SPierre Jolivet template <class Type, bool T = false, typename std::enable_if<std::is_same<Type, Vec>::value>::type * = nullptr>
9959bb5c669SPierre Jolivet static inline PetscErrorCode PCApply_Schur_Private(std::tuple<KSP, IS, Vec[2]> *p, PC factor, Type x, Type y)
9969bb5c669SPierre Jolivet {
9979bb5c669SPierre Jolivet   PetscFunctionBegin;
9989bb5c669SPierre Jolivet   PetscCall(VecISCopy(std::get<2>(*p)[0], std::get<1>(*p), SCATTER_FORWARD, x));
9999bb5c669SPierre Jolivet   if (!T) PetscCall(PCApply(factor, std::get<2>(*p)[0], std::get<2>(*p)[1]));
10009bb5c669SPierre Jolivet   else PetscCall(PCApplyTranspose(factor, std::get<2>(*p)[0], std::get<2>(*p)[1]));
10019bb5c669SPierre Jolivet   PetscCall(VecISCopy(std::get<2>(*p)[1], std::get<1>(*p), SCATTER_REVERSE, y));
10029bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
10039bb5c669SPierre Jolivet }
10049bb5c669SPierre Jolivet 
10059bb5c669SPierre Jolivet template <class Type, bool = false, typename std::enable_if<std::is_same<Type, Mat>::value>::type * = nullptr>
10069bb5c669SPierre Jolivet static inline PetscErrorCode PCApply_Schur_Private(std::tuple<KSP, IS, Vec[2]> *p, PC factor, Type X, Type Y)
10079bb5c669SPierre Jolivet {
10089bb5c669SPierre Jolivet   Mat B[2];
10099bb5c669SPierre Jolivet   Vec x, y;
10109bb5c669SPierre Jolivet 
10119bb5c669SPierre Jolivet   PetscFunctionBegin;
10129bb5c669SPierre Jolivet   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, factor->mat->rmap->n, X->cmap->n, nullptr, B));
10139bb5c669SPierre Jolivet   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, factor->mat->rmap->n, X->cmap->n, nullptr, B + 1));
10149bb5c669SPierre Jolivet   for (PetscInt i = 0; i < X->cmap->n; ++i) {
10159bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecRead(X, i, &x));
10169bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecWrite(B[0], i, &y));
10179bb5c669SPierre Jolivet     PetscCall(VecISCopy(y, std::get<1>(*p), SCATTER_FORWARD, x));
10189bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecWrite(B[0], i, &y));
10199bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecRead(X, i, &x));
10209bb5c669SPierre Jolivet   }
10219bb5c669SPierre Jolivet   PetscCall(PCMatApply(factor, B[0], B[1]));
10229bb5c669SPierre Jolivet   PetscCall(MatDestroy(B));
10239bb5c669SPierre Jolivet   for (PetscInt i = 0; i < X->cmap->n; ++i) {
10249bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecRead(B[1], i, &x));
10259bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecWrite(Y, i, &y));
10269bb5c669SPierre Jolivet     PetscCall(VecISCopy(x, std::get<1>(*p), SCATTER_REVERSE, y));
10279bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &y));
10289bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecRead(B[1], i, &x));
10299bb5c669SPierre Jolivet   }
10309bb5c669SPierre Jolivet   PetscCall(MatDestroy(B + 1));
10319bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
10329bb5c669SPierre Jolivet }
10339bb5c669SPierre Jolivet 
10349bb5c669SPierre Jolivet template <class Type = Vec, bool T = false>
10359bb5c669SPierre Jolivet static PetscErrorCode PCApply_Schur(PC pc, Type x, Type y)
10369bb5c669SPierre Jolivet {
10379bb5c669SPierre Jolivet   PC                           factor;
10389bb5c669SPierre Jolivet   Mat                          A;
10399bb5c669SPierre Jolivet   MatSolverType                type;
10409bb5c669SPierre Jolivet   PetscBool                    flg;
10419bb5c669SPierre Jolivet   std::tuple<KSP, IS, Vec[2]> *p;
10429bb5c669SPierre Jolivet 
10439bb5c669SPierre Jolivet   PetscFunctionBegin;
10449bb5c669SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
10459bb5c669SPierre Jolivet   PetscCall(KSPGetPC(std::get<0>(*p), &factor));
10469bb5c669SPierre Jolivet   PetscCall(PCFactorGetMatSolverType(factor, &type));
10479bb5c669SPierre Jolivet   PetscCall(PCFactorGetMatrix(factor, &A));
10489bb5c669SPierre Jolivet   PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg));
10499bb5c669SPierre Jolivet   if (flg) {
10509bb5c669SPierre Jolivet     PetscCheck(PetscDefined(HAVE_MUMPS), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent MatSolverType");
10519bb5c669SPierre Jolivet     PetscCall(MatMumpsSetIcntl(A, 26, 0));
10529bb5c669SPierre Jolivet   } else {
10539bb5c669SPierre Jolivet     PetscCall(PetscStrcmp(type, MATSOLVERMKL_PARDISO, &flg));
10549bb5c669SPierre Jolivet     PetscCheck(flg && PetscDefined(HAVE_MKL_PARDISO), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent MatSolverType");
10559bb5c669SPierre Jolivet     flg = PETSC_FALSE;
10569bb5c669SPierre Jolivet #if PetscDefined(HAVE_MKL_PARDISO)
10579bb5c669SPierre Jolivet     PetscCall(MatMkl_PardisoSetCntl(A, 70, 1));
10589bb5c669SPierre Jolivet #endif
10599bb5c669SPierre Jolivet   }
10609bb5c669SPierre Jolivet   PetscCall(PCApply_Schur_Private<Type, T>(p, factor, x, y));
10619bb5c669SPierre Jolivet   if (flg) {
10629bb5c669SPierre Jolivet     PetscCall(MatMumpsSetIcntl(A, 26, -1));
10639bb5c669SPierre Jolivet   } else {
10649bb5c669SPierre Jolivet #if PetscDefined(HAVE_MKL_PARDISO)
10659bb5c669SPierre Jolivet     PetscCall(MatMkl_PardisoSetCntl(A, 70, 0));
10669bb5c669SPierre Jolivet #endif
10679bb5c669SPierre Jolivet   }
10689bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
10699bb5c669SPierre Jolivet }
10709bb5c669SPierre Jolivet 
10719bb5c669SPierre Jolivet static PetscErrorCode PCDestroy_Schur(PC pc)
10729bb5c669SPierre Jolivet {
10739bb5c669SPierre Jolivet   std::tuple<KSP, IS, Vec[2]> *p;
10749bb5c669SPierre Jolivet 
10759bb5c669SPierre Jolivet   PetscFunctionBegin;
10769bb5c669SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
10779bb5c669SPierre Jolivet   PetscCall(ISDestroy(&std::get<1>(*p)));
10789bb5c669SPierre Jolivet   PetscCall(VecDestroy(std::get<2>(*p)));
10799bb5c669SPierre Jolivet   PetscCall(VecDestroy(std::get<2>(*p) + 1));
10809bb5c669SPierre Jolivet   PetscCall(PetscFree(p));
10819bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
10829bb5c669SPierre Jolivet }
10839bb5c669SPierre Jolivet 
1084d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSolve_Private(const PC_HPDDM_Level *ctx, PetscScalar *rhs, const unsigned short &mu)
1085d71ae5a4SJacob Faibussowitsch {
1086f1580f4eSBarry Smith   Mat      B, X;
1087f1580f4eSBarry Smith   PetscInt n, N, j = 0;
1088f1580f4eSBarry Smith 
1089f1580f4eSBarry Smith   PetscFunctionBegin;
1090db4a47b3SPierre Jolivet   PetscCall(KSPGetOperators(ctx->ksp, &B, nullptr));
1091db4a47b3SPierre Jolivet   PetscCall(MatGetLocalSize(B, &n, nullptr));
1092db4a47b3SPierre Jolivet   PetscCall(MatGetSize(B, &N, nullptr));
1093f1580f4eSBarry Smith   if (ctx->parent->log_separate) {
1094f1580f4eSBarry Smith     j = std::distance(ctx->parent->levels, std::find(ctx->parent->levels, ctx->parent->levels + ctx->parent->N, ctx));
1095db4a47b3SPierre Jolivet     PetscCall(PetscLogEventBegin(PC_HPDDM_Solve[j], ctx->ksp, nullptr, nullptr, nullptr));
1096f1580f4eSBarry Smith   }
1097f1580f4eSBarry Smith   if (mu == 1) {
1098f1580f4eSBarry Smith     if (!ctx->ksp->vec_rhs) {
1099db4a47b3SPierre Jolivet       PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)ctx->ksp), 1, n, N, nullptr, &ctx->ksp->vec_rhs));
1100f1580f4eSBarry Smith       PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)ctx->ksp), n, N, &ctx->ksp->vec_sol));
1101f1580f4eSBarry Smith     }
1102f1580f4eSBarry Smith     PetscCall(VecPlaceArray(ctx->ksp->vec_rhs, rhs));
1103db4a47b3SPierre Jolivet     PetscCall(KSPSolve(ctx->ksp, nullptr, nullptr));
1104f1580f4eSBarry Smith     PetscCall(VecCopy(ctx->ksp->vec_sol, ctx->ksp->vec_rhs));
1105f1580f4eSBarry Smith     PetscCall(VecResetArray(ctx->ksp->vec_rhs));
1106f1580f4eSBarry Smith   } else {
1107f1580f4eSBarry Smith     PetscCall(MatCreateDense(PetscObjectComm((PetscObject)ctx->ksp), n, PETSC_DECIDE, N, mu, rhs, &B));
1108db4a47b3SPierre Jolivet     PetscCall(MatCreateDense(PetscObjectComm((PetscObject)ctx->ksp), n, PETSC_DECIDE, N, mu, nullptr, &X));
1109f1580f4eSBarry Smith     PetscCall(KSPMatSolve(ctx->ksp, B, X));
1110f1580f4eSBarry Smith     PetscCall(MatCopy(X, B, SAME_NONZERO_PATTERN));
1111f1580f4eSBarry Smith     PetscCall(MatDestroy(&X));
1112f1580f4eSBarry Smith     PetscCall(MatDestroy(&B));
1113f1580f4eSBarry Smith   }
1114db4a47b3SPierre Jolivet   if (ctx->parent->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Solve[j], ctx->ksp, nullptr, nullptr, nullptr));
11153ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1116f1580f4eSBarry Smith }
1117f1580f4eSBarry Smith 
1118d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetUpNeumannOverlap_Private(PC pc)
1119d71ae5a4SJacob Faibussowitsch {
1120f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
1121f1580f4eSBarry Smith 
1122f1580f4eSBarry Smith   PetscFunctionBegin;
1123f1580f4eSBarry Smith   if (data->setup) {
1124f1580f4eSBarry Smith     Mat       P;
1125db4a47b3SPierre Jolivet     Vec       x, xt = nullptr;
1126f1580f4eSBarry Smith     PetscReal t = 0.0, s = 0.0;
1127f1580f4eSBarry Smith 
1128db4a47b3SPierre Jolivet     PetscCall(PCGetOperators(pc, nullptr, &P));
1129f1580f4eSBarry Smith     PetscCall(PetscObjectQuery((PetscObject)P, "__SNES_latest_X", (PetscObject *)&x));
1130f1580f4eSBarry Smith     PetscCallBack("PCHPDDM Neumann callback", (*data->setup)(data->aux, t, x, xt, s, data->is, data->setup_ctx));
1131f1580f4eSBarry Smith   }
11323ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1133f1580f4eSBarry Smith }
1134f1580f4eSBarry Smith 
1135d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMCreateSubMatrices_Private(Mat mat, PetscInt n, const IS *, const IS *, MatReuse scall, Mat *submat[])
1136d71ae5a4SJacob Faibussowitsch {
1137f1580f4eSBarry Smith   Mat       A;
113813044ca3SPierre Jolivet   PetscBool flg;
1139f1580f4eSBarry Smith 
1140f1580f4eSBarry Smith   PetscFunctionBegin;
1141f1580f4eSBarry Smith   PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "MatCreateSubMatrices() called to extract %" PetscInt_FMT " submatrices, which is different than 1", n);
1142f1580f4eSBarry Smith   /* previously composed Mat */
1143f1580f4eSBarry Smith   PetscCall(PetscObjectQuery((PetscObject)mat, "_PCHPDDM_SubMatrices", (PetscObject *)&A));
1144f1580f4eSBarry Smith   PetscCheck(A, PETSC_COMM_SELF, PETSC_ERR_PLIB, "SubMatrices not found in Mat");
114513044ca3SPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSCHURCOMPLEMENT, &flg)); /* MATSCHURCOMPLEMENT has neither a MatDuplicate() nor a MatCopy() implementation */
1146f1580f4eSBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
11476f2c871aSStefano Zampini     PetscCall(PetscCalloc1(2, submat)); /* allocate an extra Mat to avoid errors in MatDestroySubMatrices_Dummy() */
114813044ca3SPierre Jolivet     if (!flg) PetscCall(MatDuplicate(A, MAT_COPY_VALUES, *submat));
114913044ca3SPierre Jolivet   } else if (!flg) PetscCall(MatCopy(A, (*submat)[0], SAME_NONZERO_PATTERN));
115013044ca3SPierre Jolivet   if (flg) {
1151bf583f0cSPierre Jolivet     PetscCall(MatDestroy(*submat)); /* previously created Mat has to be destroyed */
115213044ca3SPierre Jolivet     (*submat)[0] = A;
115313044ca3SPierre Jolivet     PetscCall(PetscObjectReference((PetscObject)A));
115413044ca3SPierre Jolivet   }
11553ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1156f1580f4eSBarry Smith }
1157f1580f4eSBarry Smith 
1158d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMCommunicationAvoidingPCASM_Private(PC pc, Mat C, PetscBool sorted)
1159d71ae5a4SJacob Faibussowitsch {
1160f1580f4eSBarry Smith   void (*op)(void);
1161f1580f4eSBarry Smith 
1162f1580f4eSBarry Smith   PetscFunctionBegin;
1163f1580f4eSBarry Smith   /* previously-composed Mat */
1164f1580f4eSBarry Smith   PetscCall(PetscObjectCompose((PetscObject)pc->pmat, "_PCHPDDM_SubMatrices", (PetscObject)C));
1165f1580f4eSBarry Smith   PetscCall(MatGetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, &op));
1166f1580f4eSBarry Smith   /* trick suggested by Barry https://lists.mcs.anl.gov/pipermail/petsc-dev/2020-January/025491.html */
1167f1580f4eSBarry Smith   PetscCall(MatSetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, (void (*)(void))PCHPDDMCreateSubMatrices_Private));
1168f1580f4eSBarry Smith   if (sorted) PetscCall(PCASMSetSortIndices(pc, PETSC_FALSE)); /* everything is already sorted */
1169f1580f4eSBarry Smith   PetscCall(PCSetFromOptions(pc));                             /* otherwise -pc_hpddm_levels_1_pc_asm_sub_mat_type is not used */
1170f1580f4eSBarry Smith   PetscCall(PCSetUp(pc));
1171f1580f4eSBarry Smith   /* reset MatCreateSubMatrices() */
1172f1580f4eSBarry Smith   PetscCall(MatSetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, op));
1173db4a47b3SPierre Jolivet   PetscCall(PetscObjectCompose((PetscObject)pc->pmat, "_PCHPDDM_SubMatrices", nullptr));
11743ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1175f1580f4eSBarry Smith }
1176f1580f4eSBarry Smith 
1177d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMPermute_Private(IS is, IS in_is, IS *out_is, Mat in_C, Mat *out_C, IS *p)
1178d71ae5a4SJacob Faibussowitsch {
1179f1580f4eSBarry Smith   IS                           perm;
1180f1580f4eSBarry Smith   const PetscInt              *ptr;
1181811e8887SPierre Jolivet   PetscInt                    *concatenate, size, bs;
1182f1580f4eSBarry Smith   std::map<PetscInt, PetscInt> order;
1183f1580f4eSBarry Smith   PetscBool                    sorted;
1184f1580f4eSBarry Smith 
1185f1580f4eSBarry Smith   PetscFunctionBegin;
1186cf67ef9dSPierre Jolivet   PetscValidHeaderSpecific(is, IS_CLASSID, 1);
1187cf67ef9dSPierre Jolivet   PetscValidHeaderSpecific(in_C, MAT_CLASSID, 4);
1188f1580f4eSBarry Smith   PetscCall(ISSorted(is, &sorted));
1189f1580f4eSBarry Smith   if (!sorted) {
1190f1580f4eSBarry Smith     PetscCall(ISGetLocalSize(is, &size));
1191f1580f4eSBarry Smith     PetscCall(ISGetIndices(is, &ptr));
1192b07dfdedSPierre Jolivet     PetscCall(ISGetBlockSize(is, &bs));
1193f1580f4eSBarry Smith     /* MatCreateSubMatrices(), called by PCASM, follows the global numbering of Pmat */
1194811e8887SPierre Jolivet     for (PetscInt n = 0; n < size; n += bs) order.insert(std::make_pair(ptr[n] / bs, n / bs));
1195f1580f4eSBarry Smith     PetscCall(ISRestoreIndices(is, &ptr));
1196b07dfdedSPierre Jolivet     size /= bs;
1197f1580f4eSBarry Smith     if (out_C) {
1198f1580f4eSBarry Smith       PetscCall(PetscMalloc1(size, &concatenate));
1199f1580f4eSBarry Smith       for (const std::pair<const PetscInt, PetscInt> &i : order) *concatenate++ = i.second;
1200f1580f4eSBarry Smith       concatenate -= size;
1201b07dfdedSPierre Jolivet       PetscCall(ISCreateBlock(PetscObjectComm((PetscObject)in_C), bs, size, concatenate, PETSC_OWN_POINTER, &perm));
1202f1580f4eSBarry Smith       PetscCall(ISSetPermutation(perm));
1203f1580f4eSBarry Smith       /* permute user-provided Mat so that it matches with MatCreateSubMatrices() numbering */
1204f1580f4eSBarry Smith       PetscCall(MatPermute(in_C, perm, perm, out_C));
1205f1580f4eSBarry Smith       if (p) *p = perm;
1206f1580f4eSBarry Smith       else PetscCall(ISDestroy(&perm)); /* no need to save the permutation */
1207f1580f4eSBarry Smith     }
1208f1580f4eSBarry Smith     if (out_is) {
1209f1580f4eSBarry Smith       PetscCall(PetscMalloc1(size, &concatenate));
1210f1580f4eSBarry Smith       for (const std::pair<const PetscInt, PetscInt> &i : order) *concatenate++ = i.first;
1211f1580f4eSBarry Smith       concatenate -= size;
1212f1580f4eSBarry Smith       /* permute user-provided IS so that it matches with MatCreateSubMatrices() numbering */
1213b07dfdedSPierre Jolivet       PetscCall(ISCreateBlock(PetscObjectComm((PetscObject)in_is), bs, size, concatenate, PETSC_OWN_POINTER, out_is));
1214f1580f4eSBarry Smith     }
1215f1580f4eSBarry Smith   } else { /* input IS is sorted, nothing to permute, simply duplicate inputs when needed */
1216f1580f4eSBarry Smith     if (out_C) PetscCall(MatDuplicate(in_C, MAT_COPY_VALUES, out_C));
1217f1580f4eSBarry Smith     if (out_is) PetscCall(ISDuplicate(in_is, out_is));
1218f1580f4eSBarry Smith   }
12193ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1220f1580f4eSBarry Smith }
1221f1580f4eSBarry Smith 
1222281f8ce6SPierre Jolivet static PetscErrorCode PCHPDDMCheckSymmetry_Private(PC pc, Mat A01, Mat A10, Mat *B01 = nullptr)
122313044ca3SPierre Jolivet {
122413044ca3SPierre Jolivet   Mat       T, U = nullptr, B = nullptr;
122513044ca3SPierre Jolivet   IS        z;
1226281f8ce6SPierre Jolivet   PetscBool flg, conjugate = PETSC_FALSE;
122713044ca3SPierre Jolivet 
122813044ca3SPierre Jolivet   PetscFunctionBegin;
1229281f8ce6SPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, &flg));
1230281f8ce6SPierre Jolivet   if (B01) *B01 = nullptr;
1231281f8ce6SPierre Jolivet   if (flg) {
1232811e8887SPierre 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));
1233811e8887SPierre Jolivet     PetscCall(MatTransposeGetMat(A10, &U));
1234811e8887SPierre Jolivet   } else {
1235281f8ce6SPierre Jolivet     PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, &flg));
1236281f8ce6SPierre Jolivet     if (flg) {
1237811e8887SPierre 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));
1238811e8887SPierre Jolivet       PetscCall(MatHermitianTransposeGetMat(A10, &U));
1239281f8ce6SPierre Jolivet       conjugate = PETSC_TRUE;
1240811e8887SPierre Jolivet     }
124113044ca3SPierre Jolivet   }
124213044ca3SPierre Jolivet   if (U) PetscCall(MatDuplicate(U, MAT_COPY_VALUES, &T));
124313044ca3SPierre Jolivet   else PetscCall(MatHermitianTranspose(A10, MAT_INITIAL_MATRIX, &T));
1244281f8ce6SPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)A01, MATTRANSPOSEVIRTUAL, &flg));
1245281f8ce6SPierre Jolivet   if (flg) {
1246811e8887SPierre 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));
124713044ca3SPierre Jolivet     PetscCall(MatTransposeGetMat(A01, &A01));
124813044ca3SPierre Jolivet     PetscCall(MatTranspose(A01, MAT_INITIAL_MATRIX, &B));
124913044ca3SPierre Jolivet     A01 = B;
125013044ca3SPierre Jolivet   } else {
1251281f8ce6SPierre Jolivet     PetscCall(PetscObjectTypeCompare((PetscObject)A01, MATHERMITIANTRANSPOSEVIRTUAL, &flg));
1252281f8ce6SPierre Jolivet     if (flg) {
1253811e8887SPierre 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));
125413044ca3SPierre Jolivet       PetscCall(MatHermitianTransposeGetMat(A01, &A01));
125513044ca3SPierre Jolivet       PetscCall(MatHermitianTranspose(A01, MAT_INITIAL_MATRIX, &B));
125613044ca3SPierre Jolivet       A01 = B;
125713044ca3SPierre Jolivet     }
125813044ca3SPierre Jolivet   }
1259281f8ce6SPierre Jolivet   PetscCall(PetscLayoutCompare(T->rmap, A01->rmap, &flg));
1260281f8ce6SPierre Jolivet   if (flg) {
1261281f8ce6SPierre Jolivet     PetscCall(PetscLayoutCompare(T->cmap, A01->cmap, &flg));
1262281f8ce6SPierre Jolivet     if (flg) {
126313044ca3SPierre Jolivet       PetscCall(MatFindZeroRows(A01, &z)); /* for essential boundary conditions, some implementations will */
126413044ca3SPierre Jolivet       if (z) {                             /*  zero rows in [P00 A01] except for the diagonal of P00       */
1265281f8ce6SPierre Jolivet         if (B01) PetscCall(MatDuplicate(T, MAT_COPY_VALUES, B01));
126613044ca3SPierre Jolivet         PetscCall(MatSetOption(T, MAT_NO_OFF_PROC_ZERO_ROWS, PETSC_TRUE));
126713044ca3SPierre Jolivet         PetscCall(MatZeroRowsIS(T, z, 0.0, nullptr, nullptr)); /* corresponding zero rows from A01 */
1268281f8ce6SPierre Jolivet       }
1269281f8ce6SPierre Jolivet       PetscCall(MatMultEqual(A01, T, 20, &flg));
1270281f8ce6SPierre Jolivet       if (!B01) PetscCheck(flg, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "A01 != A10^T");
1271281f8ce6SPierre Jolivet       else {
1272281f8ce6SPierre Jolivet         PetscCall(PetscInfo(pc, "A01 and A10^T are equal? %s\n", PetscBools[flg]));
1273281f8ce6SPierre Jolivet         if (!flg) {
1274281f8ce6SPierre Jolivet           if (z) PetscCall(MatDestroy(&T));
1275281f8ce6SPierre Jolivet           else *B01 = T;
1276281f8ce6SPierre Jolivet           flg = PETSC_TRUE;
1277281f8ce6SPierre Jolivet         } else PetscCall(MatDestroy(B01));
1278281f8ce6SPierre Jolivet       }
127913044ca3SPierre Jolivet       PetscCall(ISDestroy(&z));
128013044ca3SPierre Jolivet     }
128113044ca3SPierre Jolivet   }
1282281f8ce6SPierre Jolivet   if (!flg) PetscCall(PetscInfo(pc, "A01 and A10^T have non-congruent layouts, cannot test for equality\n"));
1283281f8ce6SPierre Jolivet   if (!B01 || !*B01) PetscCall(MatDestroy(&T));
1284281f8ce6SPierre Jolivet   else if (conjugate) PetscCall(MatConjugate(T));
128513044ca3SPierre Jolivet   PetscCall(MatDestroy(&B));
128613044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
128713044ca3SPierre Jolivet }
128813044ca3SPierre Jolivet 
1289d16c0b94SPierre Jolivet static PetscErrorCode PCHPDDMCheckInclusion_Private(PC pc, IS is, IS is_local, PetscBool check)
1290d16c0b94SPierre Jolivet {
1291d16c0b94SPierre Jolivet   IS          intersect;
1292d16c0b94SPierre Jolivet   const char *str = "IS of the auxiliary Mat does not include all local rows of A";
1293d16c0b94SPierre Jolivet   PetscBool   equal;
1294d16c0b94SPierre Jolivet 
1295d16c0b94SPierre Jolivet   PetscFunctionBegin;
1296d16c0b94SPierre Jolivet   PetscCall(ISIntersect(is, is_local, &intersect));
1297d16c0b94SPierre Jolivet   PetscCall(ISEqualUnsorted(is_local, intersect, &equal));
1298d16c0b94SPierre Jolivet   PetscCall(ISDestroy(&intersect));
1299d16c0b94SPierre Jolivet   if (check) PetscCheck(equal, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "%s", str);
1300d16c0b94SPierre Jolivet   else if (!equal) PetscCall(PetscInfo(pc, "%s\n", str));
1301d16c0b94SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
1302d16c0b94SPierre Jolivet }
1303d16c0b94SPierre Jolivet 
1304d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMDestroySubMatrices_Private(PetscBool flg, PetscBool algebraic, Mat *sub)
1305d71ae5a4SJacob Faibussowitsch {
1306f1580f4eSBarry Smith   IS is;
1307f1580f4eSBarry Smith 
1308f1580f4eSBarry Smith   PetscFunctionBegin;
1309f1580f4eSBarry Smith   if (!flg) {
1310f1580f4eSBarry Smith     if (algebraic) {
1311f1580f4eSBarry Smith       PetscCall(PetscObjectQuery((PetscObject)sub[0], "_PCHPDDM_Embed", (PetscObject *)&is));
1312f1580f4eSBarry Smith       PetscCall(ISDestroy(&is));
1313db4a47b3SPierre Jolivet       PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Embed", nullptr));
1314db4a47b3SPierre Jolivet       PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Compact", nullptr));
1315f1580f4eSBarry Smith     }
1316f1580f4eSBarry Smith     PetscCall(MatDestroySubMatrices(algebraic ? 2 : 1, &sub));
1317f1580f4eSBarry Smith   }
13183ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1319f1580f4eSBarry Smith }
1320f1580f4eSBarry Smith 
1321d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMAlgebraicAuxiliaryMat_Private(Mat P, IS *is, Mat *sub[], PetscBool block)
1322d71ae5a4SJacob Faibussowitsch {
1323f1580f4eSBarry Smith   IS         icol[3], irow[2];
1324f1580f4eSBarry Smith   Mat       *M, Q;
1325f1580f4eSBarry Smith   PetscReal *ptr;
132658b7e2c1SStefano Zampini   PetscInt  *idx, p = 0, bs = P->cmap->bs;
1327f1580f4eSBarry Smith   PetscBool  flg;
1328f1580f4eSBarry Smith 
1329f1580f4eSBarry Smith   PetscFunctionBegin;
1330f1580f4eSBarry Smith   PetscCall(ISCreateStride(PETSC_COMM_SELF, P->cmap->N, 0, 1, icol + 2));
1331f1580f4eSBarry Smith   PetscCall(ISSetBlockSize(icol[2], bs));
1332f1580f4eSBarry Smith   PetscCall(ISSetIdentity(icol[2]));
1333f1580f4eSBarry Smith   PetscCall(PetscObjectTypeCompare((PetscObject)P, MATMPISBAIJ, &flg));
1334f1580f4eSBarry Smith   if (flg) {
1335f1580f4eSBarry Smith     /* MatCreateSubMatrices() does not handle MATMPISBAIJ properly when iscol != isrow, so convert first to MATMPIBAIJ */
1336f1580f4eSBarry Smith     PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &Q));
1337f1580f4eSBarry Smith     std::swap(P, Q);
1338f1580f4eSBarry Smith   }
1339f1580f4eSBarry Smith   PetscCall(MatCreateSubMatrices(P, 1, is, icol + 2, MAT_INITIAL_MATRIX, &M));
1340f1580f4eSBarry Smith   if (flg) {
1341f1580f4eSBarry Smith     std::swap(P, Q);
1342f1580f4eSBarry Smith     PetscCall(MatDestroy(&Q));
1343f1580f4eSBarry Smith   }
1344f1580f4eSBarry Smith   PetscCall(ISDestroy(icol + 2));
1345f1580f4eSBarry Smith   PetscCall(ISCreateStride(PETSC_COMM_SELF, M[0]->rmap->N, 0, 1, irow));
1346f1580f4eSBarry Smith   PetscCall(ISSetBlockSize(irow[0], bs));
1347f1580f4eSBarry Smith   PetscCall(ISSetIdentity(irow[0]));
1348f1580f4eSBarry Smith   if (!block) {
1349b07dfdedSPierre Jolivet     PetscCall(PetscMalloc2(P->cmap->N, &ptr, P->cmap->N / bs, &idx));
1350f1580f4eSBarry Smith     PetscCall(MatGetColumnNorms(M[0], NORM_INFINITY, ptr));
1351f1580f4eSBarry Smith     /* check for nonzero columns so that M[0] may be expressed in compact form */
1352811e8887SPierre Jolivet     for (PetscInt n = 0; n < P->cmap->N; n += bs) {
1353b07dfdedSPierre Jolivet       if (std::find_if(ptr + n, ptr + n + bs, [](PetscReal v) { return v > PETSC_MACHINE_EPSILON; }) != ptr + n + bs) idx[p++] = n / bs;
1354f1580f4eSBarry Smith     }
1355b07dfdedSPierre Jolivet     PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, p, idx, PETSC_USE_POINTER, icol + 1));
1356f1580f4eSBarry Smith     PetscCall(ISSetInfo(icol[1], IS_SORTED, IS_GLOBAL, PETSC_TRUE, PETSC_TRUE));
1357f1580f4eSBarry Smith     PetscCall(ISEmbed(*is, icol[1], PETSC_FALSE, icol + 2));
1358f1580f4eSBarry Smith     irow[1] = irow[0];
1359f1580f4eSBarry 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 */
1360f1580f4eSBarry Smith     icol[0] = is[0];
1361f1580f4eSBarry Smith     PetscCall(MatCreateSubMatrices(M[0], 2, irow, icol, MAT_INITIAL_MATRIX, sub));
1362f1580f4eSBarry Smith     PetscCall(ISDestroy(icol + 1));
1363f1580f4eSBarry Smith     PetscCall(PetscFree2(ptr, idx));
1364f1580f4eSBarry Smith     /* IS used to go back and forth between the augmented and the original local linear system, see eq. (3.4) of [2022b] */
1365f1580f4eSBarry Smith     PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Embed", (PetscObject)icol[2]));
1366f1580f4eSBarry Smith     /* Mat used in eq. (3.1) of [2022b] */
1367f1580f4eSBarry Smith     PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Compact", (PetscObject)(*sub)[1]));
1368f1580f4eSBarry Smith   } else {
1369f1580f4eSBarry Smith     Mat aux;
1370811e8887SPierre Jolivet 
1371f1580f4eSBarry Smith     PetscCall(MatSetOption(M[0], MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
1372f1580f4eSBarry Smith     /* diagonal block of the overlapping rows */
1373f1580f4eSBarry Smith     PetscCall(MatCreateSubMatrices(M[0], 1, irow, is, MAT_INITIAL_MATRIX, sub));
1374f1580f4eSBarry Smith     PetscCall(MatDuplicate((*sub)[0], MAT_COPY_VALUES, &aux));
1375f1580f4eSBarry Smith     PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE));
1376f1580f4eSBarry Smith     if (bs == 1) { /* scalar case */
1377f1580f4eSBarry Smith       Vec sum[2];
1378811e8887SPierre Jolivet 
1379f1580f4eSBarry Smith       PetscCall(MatCreateVecs(aux, sum, sum + 1));
1380f1580f4eSBarry Smith       PetscCall(MatGetRowSum(M[0], sum[0]));
1381f1580f4eSBarry Smith       PetscCall(MatGetRowSum(aux, sum[1]));
1382f1580f4eSBarry Smith       /* off-diagonal block row sum (full rows - diagonal block rows) */
1383f1580f4eSBarry Smith       PetscCall(VecAXPY(sum[0], -1.0, sum[1]));
1384f1580f4eSBarry Smith       /* subdomain matrix plus off-diagonal block row sum */
1385f1580f4eSBarry Smith       PetscCall(MatDiagonalSet(aux, sum[0], ADD_VALUES));
1386f1580f4eSBarry Smith       PetscCall(VecDestroy(sum));
1387f1580f4eSBarry Smith       PetscCall(VecDestroy(sum + 1));
1388f1580f4eSBarry Smith     } else { /* vectorial case */
1389f1580f4eSBarry Smith       /* TODO: missing MatGetValuesBlocked(), so the code below is     */
1390f1580f4eSBarry Smith       /* an extension of the scalar case for when bs > 1, but it could */
1391f1580f4eSBarry Smith       /* be more efficient by avoiding all these MatMatMult()          */
1392f1580f4eSBarry Smith       Mat          sum[2], ones;
1393f1580f4eSBarry Smith       PetscScalar *ptr;
1394811e8887SPierre Jolivet 
1395f1580f4eSBarry Smith       PetscCall(PetscCalloc1(M[0]->cmap->n * bs, &ptr));
1396f1580f4eSBarry Smith       PetscCall(MatCreateDense(PETSC_COMM_SELF, M[0]->cmap->n, bs, M[0]->cmap->n, bs, ptr, &ones));
1397811e8887SPierre Jolivet       for (PetscInt n = 0; n < M[0]->cmap->n; n += bs) {
1398f1580f4eSBarry Smith         for (p = 0; p < bs; ++p) ptr[n + p * (M[0]->cmap->n + 1)] = 1.0;
1399f1580f4eSBarry Smith       }
1400fb842aefSJose E. Roman       PetscCall(MatMatMult(M[0], ones, MAT_INITIAL_MATRIX, PETSC_CURRENT, sum));
1401f1580f4eSBarry Smith       PetscCall(MatDestroy(&ones));
1402f1580f4eSBarry Smith       PetscCall(MatCreateDense(PETSC_COMM_SELF, aux->cmap->n, bs, aux->cmap->n, bs, ptr, &ones));
1403f1580f4eSBarry Smith       PetscCall(MatDenseSetLDA(ones, M[0]->cmap->n));
1404fb842aefSJose E. Roman       PetscCall(MatMatMult(aux, ones, MAT_INITIAL_MATRIX, PETSC_CURRENT, sum + 1));
1405f1580f4eSBarry Smith       PetscCall(MatDestroy(&ones));
1406f1580f4eSBarry Smith       PetscCall(PetscFree(ptr));
1407f1580f4eSBarry Smith       /* off-diagonal block row sum (full rows - diagonal block rows) */
1408f1580f4eSBarry Smith       PetscCall(MatAXPY(sum[0], -1.0, sum[1], SAME_NONZERO_PATTERN));
1409f1580f4eSBarry Smith       PetscCall(MatDestroy(sum + 1));
1410f1580f4eSBarry Smith       /* re-order values to be consistent with MatSetValuesBlocked()           */
1411f1580f4eSBarry Smith       /* equivalent to MatTranspose() which does not truly handle              */
1412f1580f4eSBarry Smith       /* MAT_INPLACE_MATRIX in the rectangular case, as it calls PetscMalloc() */
1413f1580f4eSBarry Smith       PetscCall(MatDenseGetArrayWrite(sum[0], &ptr));
1414f1580f4eSBarry Smith       HPDDM::Wrapper<PetscScalar>::imatcopy<'T'>(bs, sum[0]->rmap->n, ptr, sum[0]->rmap->n, bs);
1415f1580f4eSBarry Smith       /* subdomain matrix plus off-diagonal block row sum */
1416811e8887SPierre Jolivet       for (PetscInt n = 0; n < aux->cmap->n / bs; ++n) PetscCall(MatSetValuesBlocked(aux, 1, &n, 1, &n, ptr + n * bs * bs, ADD_VALUES));
1417f1580f4eSBarry Smith       PetscCall(MatAssemblyBegin(aux, MAT_FINAL_ASSEMBLY));
1418f1580f4eSBarry Smith       PetscCall(MatAssemblyEnd(aux, MAT_FINAL_ASSEMBLY));
1419f1580f4eSBarry Smith       PetscCall(MatDenseRestoreArrayWrite(sum[0], &ptr));
1420f1580f4eSBarry Smith       PetscCall(MatDestroy(sum));
1421f1580f4eSBarry Smith     }
1422f1580f4eSBarry Smith     PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_TRUE));
1423f1580f4eSBarry Smith     /* left-hand side of GenEO, with the same sparsity pattern as PCASM subdomain solvers  */
1424f1580f4eSBarry Smith     PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Neumann_Mat", (PetscObject)aux));
1425f1580f4eSBarry Smith   }
1426f1580f4eSBarry Smith   PetscCall(ISDestroy(irow));
1427f1580f4eSBarry Smith   PetscCall(MatDestroySubMatrices(1, &M));
14283ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1429f1580f4eSBarry Smith }
1430f1580f4eSBarry Smith 
143113044ca3SPierre Jolivet static PetscErrorCode PCApply_Nest(PC pc, Vec x, Vec y)
143213044ca3SPierre Jolivet {
143313044ca3SPierre Jolivet   Mat                    A;
143413044ca3SPierre Jolivet   MatSolverType          type;
143513044ca3SPierre Jolivet   IS                     is[2];
143613044ca3SPierre Jolivet   PetscBool              flg;
143713044ca3SPierre Jolivet   std::pair<PC, Vec[2]> *p;
143813044ca3SPierre Jolivet 
143913044ca3SPierre Jolivet   PetscFunctionBegin;
144013044ca3SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
144113044ca3SPierre Jolivet   PetscCall(PCGetOperators(p->first, &A, nullptr));
144213044ca3SPierre Jolivet   PetscCall(MatNestGetISs(A, is, nullptr));
144313044ca3SPierre Jolivet   PetscCall(PCFactorGetMatSolverType(p->first, &type));
144413044ca3SPierre Jolivet   PetscCall(PCFactorGetMatrix(p->first, &A));
144513044ca3SPierre Jolivet   PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg));
144679578405SBarry Smith   if (flg && A->schur) PetscCall(MatMumpsSetIcntl(A, 26, 1));    /* reduction/condensation phase followed by Schur complement solve */
144713044ca3SPierre Jolivet   PetscCall(VecISCopy(p->second[0], is[1], SCATTER_FORWARD, x)); /* assign the RHS associated to the Schur complement */
144813044ca3SPierre Jolivet   PetscCall(PCApply(p->first, p->second[0], p->second[1]));
144913044ca3SPierre Jolivet   PetscCall(VecISCopy(p->second[1], is[1], SCATTER_REVERSE, y)); /* retrieve the partial solution associated to the Schur complement */
145079578405SBarry Smith   if (flg) PetscCall(MatMumpsSetIcntl(A, 26, -1));               /* default ICNTL(26) value in PETSc */
145113044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
145213044ca3SPierre Jolivet }
145313044ca3SPierre Jolivet 
145413044ca3SPierre Jolivet static PetscErrorCode PCView_Nest(PC pc, PetscViewer viewer)
145513044ca3SPierre Jolivet {
145613044ca3SPierre Jolivet   std::pair<PC, Vec[2]> *p;
145713044ca3SPierre Jolivet 
145813044ca3SPierre Jolivet   PetscFunctionBegin;
145913044ca3SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
146013044ca3SPierre Jolivet   PetscCall(PCView(p->first, viewer));
146113044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
146213044ca3SPierre Jolivet }
146313044ca3SPierre Jolivet 
146413044ca3SPierre Jolivet static PetscErrorCode PCDestroy_Nest(PC pc)
146513044ca3SPierre Jolivet {
146613044ca3SPierre Jolivet   std::pair<PC, Vec[2]> *p;
146713044ca3SPierre Jolivet 
146813044ca3SPierre Jolivet   PetscFunctionBegin;
146913044ca3SPierre Jolivet   PetscCall(PCShellGetContext(pc, &p));
147013044ca3SPierre Jolivet   PetscCall(VecDestroy(p->second));
147113044ca3SPierre Jolivet   PetscCall(VecDestroy(p->second + 1));
147213044ca3SPierre Jolivet   PetscCall(PCDestroy(&p->first));
147313044ca3SPierre Jolivet   PetscCall(PetscFree(p));
147413044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
147513044ca3SPierre Jolivet }
147613044ca3SPierre Jolivet 
147713044ca3SPierre Jolivet template <bool T = false>
147813044ca3SPierre Jolivet static PetscErrorCode MatMult_Schur(Mat A, Vec x, Vec y)
147913044ca3SPierre Jolivet {
148001e3c840SPierre Jolivet   std::tuple<Mat, PetscSF, Vec[2]> *ctx;
148113044ca3SPierre Jolivet 
148213044ca3SPierre Jolivet   PetscFunctionBegin;
148313044ca3SPierre Jolivet   PetscCall(MatShellGetContext(A, &ctx));
148413044ca3SPierre Jolivet   PetscCall(VecScatterBegin(std::get<1>(*ctx), x, std::get<2>(*ctx)[0], INSERT_VALUES, SCATTER_FORWARD)); /* local Vec with overlap */
148513044ca3SPierre Jolivet   PetscCall(VecScatterEnd(std::get<1>(*ctx), x, std::get<2>(*ctx)[0], INSERT_VALUES, SCATTER_FORWARD));
148613044ca3SPierre Jolivet   if (!T) PetscCall(MatMult(std::get<0>(*ctx), std::get<2>(*ctx)[0], std::get<2>(*ctx)[1])); /* local Schur complement */
148713044ca3SPierre Jolivet   else PetscCall(MatMultTranspose(std::get<0>(*ctx), std::get<2>(*ctx)[0], std::get<2>(*ctx)[1]));
148813044ca3SPierre Jolivet   PetscCall(VecSet(y, 0.0));
148913044ca3SPierre 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 */
149013044ca3SPierre Jolivet   PetscCall(VecScatterEnd(std::get<1>(*ctx), std::get<2>(*ctx)[1], y, ADD_VALUES, SCATTER_REVERSE));
149113044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
149213044ca3SPierre Jolivet }
149313044ca3SPierre Jolivet 
149413044ca3SPierre Jolivet static PetscErrorCode MatDestroy_Schur(Mat A)
149513044ca3SPierre Jolivet {
149601e3c840SPierre Jolivet   std::tuple<Mat, PetscSF, Vec[2]> *ctx;
149713044ca3SPierre Jolivet 
149813044ca3SPierre Jolivet   PetscFunctionBegin;
149913044ca3SPierre Jolivet   PetscCall(MatShellGetContext(A, &ctx));
150013044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<2>(*ctx)));
150113044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<2>(*ctx) + 1));
150213044ca3SPierre Jolivet   PetscCall(PetscFree(ctx));
150313044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
150413044ca3SPierre Jolivet }
150513044ca3SPierre Jolivet 
150613044ca3SPierre Jolivet static PetscErrorCode MatMult_SchurCorrection(Mat A, Vec x, Vec y)
150713044ca3SPierre Jolivet {
150813044ca3SPierre Jolivet   PC                                         pc;
150913044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx;
151013044ca3SPierre Jolivet 
151113044ca3SPierre Jolivet   PetscFunctionBegin;
151213044ca3SPierre Jolivet   PetscCall(MatShellGetContext(A, &ctx));
151313044ca3SPierre Jolivet   pc = ((PC_HPDDM *)std::get<0>(*ctx)[0]->data)->levels[0]->ksp->pc;
151413044ca3SPierre 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 */
151513044ca3SPierre Jolivet     PetscCall(MatMult(std::get<1>(*ctx)[0], x, std::get<3>(*ctx)[1]));                    /*     A_01 x                 */
151613044ca3SPierre Jolivet     PetscCall(PCHPDDMDeflate_Private(pc, std::get<3>(*ctx)[1], std::get<3>(*ctx)[1]));    /*     Q_0 A_01 x             */
151713044ca3SPierre 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        */
151813044ca3SPierre 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 */
151913044ca3SPierre Jolivet   } else {
152013044ca3SPierre Jolivet     PetscCall(PCApply(std::get<0>(*ctx)[1], x, std::get<3>(*ctx)[0]));                    /*     M_S^-1 x               */
152113044ca3SPierre Jolivet     PetscCall(MatMult(std::get<1>(*ctx)[0], std::get<3>(*ctx)[0], std::get<3>(*ctx)[1])); /*     A_01 M_S^-1 x          */
152213044ca3SPierre Jolivet     PetscCall(PCHPDDMDeflate_Private(pc, std::get<3>(*ctx)[1], std::get<3>(*ctx)[1]));    /*     Q_0 A_01 M_S^-1 x      */
152313044ca3SPierre 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 */
152413044ca3SPierre Jolivet   }
152513044ca3SPierre Jolivet   PetscCall(VecAXPY(y, -1.0, x)); /* y -= x, preconditioned eq. (24) of https://hal.science/hal-02343808v6/document (with a sign flip) */
152613044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
152713044ca3SPierre Jolivet }
152813044ca3SPierre Jolivet 
152913044ca3SPierre Jolivet static PetscErrorCode MatView_SchurCorrection(Mat A, PetscViewer viewer)
153013044ca3SPierre Jolivet {
153113044ca3SPierre Jolivet   PetscBool                                  ascii;
153213044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx;
153313044ca3SPierre Jolivet 
153413044ca3SPierre Jolivet   PetscFunctionBegin;
153513044ca3SPierre Jolivet   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &ascii));
153613044ca3SPierre Jolivet   if (ascii) {
153713044ca3SPierre Jolivet     PetscCall(MatShellGetContext(A, &ctx));
153813044ca3SPierre 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)"));
153913044ca3SPierre Jolivet     PetscCall(PCView(std::get<0>(*ctx)[1], viewer)); /* no need to PCView(Q_0) since it will be done by PCFIELDSPLIT */
154013044ca3SPierre Jolivet   }
154113044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
154213044ca3SPierre Jolivet }
154313044ca3SPierre Jolivet 
154413044ca3SPierre Jolivet static PetscErrorCode MatDestroy_SchurCorrection(Mat A)
154513044ca3SPierre Jolivet {
154613044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx;
154713044ca3SPierre Jolivet 
154813044ca3SPierre Jolivet   PetscFunctionBegin;
154913044ca3SPierre Jolivet   PetscCall(MatShellGetContext(A, &ctx));
155013044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<3>(*ctx)));
155113044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<3>(*ctx) + 1));
155213044ca3SPierre Jolivet   PetscCall(VecDestroy(std::get<3>(*ctx) + 2));
155313044ca3SPierre Jolivet   PetscCall(PCDestroy(std::get<0>(*ctx) + 1));
155413044ca3SPierre Jolivet   PetscCall(PetscFree(ctx));
155513044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
155613044ca3SPierre Jolivet }
155713044ca3SPierre Jolivet 
1558feebddf4SPierre Jolivet static PetscErrorCode PCPostSolve_SchurPreLeastSquares(PC, KSP, Vec, Vec x)
1559feebddf4SPierre Jolivet {
1560feebddf4SPierre Jolivet   PetscFunctionBegin;
1561feebddf4SPierre Jolivet   PetscCall(VecScale(x, -1.0));
1562feebddf4SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
1563feebddf4SPierre Jolivet }
1564feebddf4SPierre Jolivet 
156513044ca3SPierre Jolivet static PetscErrorCode KSPPreSolve_SchurCorrection(KSP, Vec b, Vec, void *context)
156613044ca3SPierre Jolivet {
156713044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx = reinterpret_cast<std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *>(context);
156813044ca3SPierre Jolivet 
1569a6b3e571SPierre Jolivet   PetscFunctionBegin;
157013044ca3SPierre Jolivet   if (std::get<2>(*ctx) == PC_LEFT || std::get<2>(*ctx) == PC_SIDE_DEFAULT) {
157113044ca3SPierre Jolivet     PetscCall(PCApply(std::get<0>(*ctx)[1], b, std::get<3>(*ctx)[2]));
157213044ca3SPierre 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 */
157313044ca3SPierre Jolivet   }
157413044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
157513044ca3SPierre Jolivet }
157613044ca3SPierre Jolivet 
157713044ca3SPierre Jolivet static PetscErrorCode KSPPostSolve_SchurCorrection(KSP, Vec b, Vec x, void *context)
157813044ca3SPierre Jolivet {
157913044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx = reinterpret_cast<std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *>(context);
158013044ca3SPierre Jolivet 
1581a6b3e571SPierre Jolivet   PetscFunctionBegin;
158213044ca3SPierre 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 */
158313044ca3SPierre Jolivet   else {
158413044ca3SPierre Jolivet     PetscCall(PCApply(std::get<0>(*ctx)[1], x, std::get<3>(*ctx)[2]));
158513044ca3SPierre Jolivet     PetscCall(VecCopy(std::get<3>(*ctx)[2], x)); /* replace x by M^-1 x */
158613044ca3SPierre Jolivet   }
158713044ca3SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
158813044ca3SPierre Jolivet }
158913044ca3SPierre Jolivet 
15909bb5c669SPierre Jolivet static PetscErrorCode MatMult_Harmonic(Mat, Vec, Vec);
15919bb5c669SPierre Jolivet static PetscErrorCode MatMultTranspose_Harmonic(Mat, Vec, Vec);
15929bb5c669SPierre Jolivet static PetscErrorCode MatProduct_AB_Harmonic(Mat, Mat, Mat, void *);
15939bb5c669SPierre Jolivet static PetscErrorCode MatDestroy_Harmonic(Mat);
15949bb5c669SPierre Jolivet 
1595d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetUp_HPDDM(PC pc)
1596d71ae5a4SJacob Faibussowitsch {
1597f1580f4eSBarry Smith   PC_HPDDM                                  *data = (PC_HPDDM *)pc->data;
1598f1580f4eSBarry Smith   PC                                         inner;
1599f1580f4eSBarry Smith   KSP                                       *ksp;
160013044ca3SPierre Jolivet   Mat                                       *sub, A, P, N, C = nullptr, uaux = nullptr, weighted, subA[2], S;
1601f1580f4eSBarry Smith   Vec                                        xin, v;
1602f1580f4eSBarry Smith   std::vector<Vec>                           initial;
1603db4a47b3SPierre Jolivet   IS                                         is[1], loc, uis = data->is, unsorted = nullptr;
1604f1580f4eSBarry Smith   ISLocalToGlobalMapping                     l2g;
1605f1580f4eSBarry Smith   char                                       prefix[256];
1606f1580f4eSBarry Smith   const char                                *pcpre;
1607f1580f4eSBarry Smith   const PetscScalar *const                  *ev;
16089bb5c669SPierre Jolivet   PetscInt                                   n, requested = data->N, reused = 0, overlap = -1;
1609f1580f4eSBarry Smith   MatStructure                               structure  = UNKNOWN_NONZERO_PATTERN;
1610f1580f4eSBarry Smith   PetscBool                                  subdomains = PETSC_FALSE, flg = PETSC_FALSE, ismatis, swap = PETSC_FALSE, algebraic = PETSC_FALSE, block = PETSC_FALSE;
1611f1580f4eSBarry Smith   DM                                         dm;
161213044ca3SPierre Jolivet   std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx = nullptr;
1613398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG)
1614db4a47b3SPierre Jolivet   IS  dis  = nullptr;
1615db4a47b3SPierre Jolivet   Mat daux = nullptr;
1616398c7888SPierre Jolivet #endif
1617f1580f4eSBarry Smith 
1618f1580f4eSBarry Smith   PetscFunctionBegin;
1619f1580f4eSBarry Smith   PetscCheck(data->levels && data->levels[0], PETSC_COMM_SELF, PETSC_ERR_PLIB, "Not a single level allocated");
1620f1580f4eSBarry Smith   PetscCall(PCGetOptionsPrefix(pc, &pcpre));
1621f1580f4eSBarry Smith   PetscCall(PCGetOperators(pc, &A, &P));
1622f1580f4eSBarry Smith   if (!data->levels[0]->ksp) {
1623f1580f4eSBarry Smith     PetscCall(KSPCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->ksp));
16243821be0aSBarry Smith     PetscCall(KSPSetNestLevel(data->levels[0]->ksp, pc->kspnestlevel));
1625f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_%s_", pcpre ? pcpre : "", data->N > 1 ? "levels_1" : "coarse"));
1626f1580f4eSBarry Smith     PetscCall(KSPSetOptionsPrefix(data->levels[0]->ksp, prefix));
1627f1580f4eSBarry Smith     PetscCall(KSPSetType(data->levels[0]->ksp, KSPPREONLY));
1628*371d2eb7SMartin Diehl   } else if (data->levels[0]->ksp->pc && data->levels[0]->ksp->pc->setupcalled && data->levels[0]->ksp->pc->reusepreconditioner) {
1629f1580f4eSBarry Smith     /* if the fine-level PCSHELL exists, its setup has succeeded, and one wants to reuse it, */
1630f1580f4eSBarry Smith     /* then just propagate the appropriate flag to the coarser levels                        */
1631f1580f4eSBarry Smith     for (n = 0; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) {
1632f1580f4eSBarry Smith       /* the following KSP and PC may be NULL for some processes, hence the check            */
1633f1580f4eSBarry Smith       if (data->levels[n]->ksp) PetscCall(KSPSetReusePreconditioner(data->levels[n]->ksp, PETSC_TRUE));
1634f1580f4eSBarry Smith       if (data->levels[n]->pc) PetscCall(PCSetReusePreconditioner(data->levels[n]->pc, PETSC_TRUE));
1635f1580f4eSBarry Smith     }
1636f1580f4eSBarry Smith     /* early bail out because there is nothing to do */
16373ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
1638f1580f4eSBarry Smith   } else {
1639f1580f4eSBarry Smith     /* reset coarser levels */
1640f1580f4eSBarry Smith     for (n = 1; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) {
1641*371d2eb7SMartin Diehl       if (data->levels[n]->ksp && data->levels[n]->ksp->pc && data->levels[n]->ksp->pc->setupcalled && data->levels[n]->ksp->pc->reusepreconditioner && n < data->N) {
1642f1580f4eSBarry Smith         reused = data->N - n;
1643f1580f4eSBarry Smith         break;
1644f1580f4eSBarry Smith       }
1645f1580f4eSBarry Smith       PetscCall(KSPDestroy(&data->levels[n]->ksp));
1646f1580f4eSBarry Smith       PetscCall(PCDestroy(&data->levels[n]->pc));
1647f1580f4eSBarry Smith     }
1648f1580f4eSBarry Smith     /* check if some coarser levels are being reused */
1649462c564dSBarry Smith     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &reused, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)pc)));
1650f1580f4eSBarry Smith     const int *addr = data->levels[0]->P ? data->levels[0]->P->getAddrLocal() : &HPDDM::i__0;
1651f1580f4eSBarry Smith 
1652f1580f4eSBarry Smith     if (addr != &HPDDM::i__0 && reused != data->N - 1) {
1653f1580f4eSBarry Smith       /* reuse previously computed eigenvectors */
1654f1580f4eSBarry Smith       ev = data->levels[0]->P->getVectors();
1655f1580f4eSBarry Smith       if (ev) {
1656f1580f4eSBarry Smith         initial.reserve(*addr);
1657f1580f4eSBarry Smith         PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, data->levels[0]->P->getDof(), ev[0], &xin));
1658f1580f4eSBarry Smith         for (n = 0; n < *addr; ++n) {
1659f1580f4eSBarry Smith           PetscCall(VecDuplicate(xin, &v));
1660f1580f4eSBarry Smith           PetscCall(VecPlaceArray(xin, ev[n]));
1661f1580f4eSBarry Smith           PetscCall(VecCopy(xin, v));
1662f1580f4eSBarry Smith           initial.emplace_back(v);
1663f1580f4eSBarry Smith           PetscCall(VecResetArray(xin));
1664f1580f4eSBarry Smith         }
1665f1580f4eSBarry Smith         PetscCall(VecDestroy(&xin));
1666f1580f4eSBarry Smith       }
1667f1580f4eSBarry Smith     }
1668f1580f4eSBarry Smith   }
1669f1580f4eSBarry Smith   data->N -= reused;
1670f1580f4eSBarry Smith   PetscCall(KSPSetOperators(data->levels[0]->ksp, A, P));
1671f1580f4eSBarry Smith 
1672f1580f4eSBarry Smith   PetscCall(PetscObjectTypeCompare((PetscObject)P, MATIS, &ismatis));
1673f1580f4eSBarry Smith   if (!data->is && !ismatis) {
1674db4a47b3SPierre Jolivet     PetscErrorCode (*create)(DM, IS *, Mat *, PetscErrorCode (**)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void **) = nullptr;
1675db4a47b3SPierre Jolivet     PetscErrorCode (*usetup)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *)                                                = nullptr;
1676db4a47b3SPierre Jolivet     void *uctx                                                                                                               = nullptr;
1677f1580f4eSBarry Smith 
1678f1580f4eSBarry Smith     /* first see if we can get the data from the DM */
1679f1580f4eSBarry Smith     PetscCall(MatGetDM(P, &dm));
1680f1580f4eSBarry Smith     if (!dm) PetscCall(MatGetDM(A, &dm));
1681f1580f4eSBarry Smith     if (!dm) PetscCall(PCGetDM(pc, &dm));
1682907a3e9cSStefano Zampini     if (dm) { /* this is the hook for DMPLEX for which the auxiliary Mat is the local Neumann matrix */
1683f1580f4eSBarry Smith       PetscCall(PetscObjectQueryFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", &create));
1684f1580f4eSBarry Smith       if (create) {
1685f1580f4eSBarry Smith         PetscCall((*create)(dm, &uis, &uaux, &usetup, &uctx));
1686c8ea6600SPierre 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 */
1687f1580f4eSBarry Smith       }
1688f1580f4eSBarry Smith     }
1689f1580f4eSBarry Smith     if (!create) {
1690f1580f4eSBarry Smith       if (!uis) {
1691f1580f4eSBarry Smith         PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Neumann_IS", (PetscObject *)&uis));
1692f1580f4eSBarry Smith         PetscCall(PetscObjectReference((PetscObject)uis));
1693f1580f4eSBarry Smith       }
1694f1580f4eSBarry Smith       if (!uaux) {
1695f1580f4eSBarry Smith         PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Neumann_Mat", (PetscObject *)&uaux));
1696f1580f4eSBarry Smith         PetscCall(PetscObjectReference((PetscObject)uaux));
1697f1580f4eSBarry Smith       }
1698f1580f4eSBarry Smith       /* look inside the Pmat instead of the PC, needed for MatSchurComplementComputeExplicitOperator() */
1699f1580f4eSBarry Smith       if (!uis) {
1700f1580f4eSBarry Smith         PetscCall(PetscObjectQuery((PetscObject)P, "_PCHPDDM_Neumann_IS", (PetscObject *)&uis));
1701f1580f4eSBarry Smith         PetscCall(PetscObjectReference((PetscObject)uis));
1702f1580f4eSBarry Smith       }
1703f1580f4eSBarry Smith       if (!uaux) {
1704f1580f4eSBarry Smith         PetscCall(PetscObjectQuery((PetscObject)P, "_PCHPDDM_Neumann_Mat", (PetscObject *)&uaux));
1705f1580f4eSBarry Smith         PetscCall(PetscObjectReference((PetscObject)uaux));
1706f1580f4eSBarry Smith       }
1707f1580f4eSBarry Smith     }
1708f1580f4eSBarry Smith     PetscCall(PCHPDDMSetAuxiliaryMat(pc, uis, uaux, usetup, uctx));
1709f1580f4eSBarry Smith     PetscCall(MatDestroy(&uaux));
1710f1580f4eSBarry Smith     PetscCall(ISDestroy(&uis));
1711f1580f4eSBarry Smith   }
1712f1580f4eSBarry Smith 
1713f1580f4eSBarry Smith   if (!ismatis) {
1714f1580f4eSBarry Smith     PetscCall(PCHPDDMSetUpNeumannOverlap_Private(pc));
1715db4a47b3SPierre Jolivet     PetscCall(PetscOptionsGetBool(nullptr, pcpre, "-pc_hpddm_block_splitting", &block, nullptr));
17169bb5c669SPierre Jolivet     PetscCall(PetscOptionsGetInt(nullptr, pcpre, "-pc_hpddm_harmonic_overlap", &overlap, nullptr));
17175e642048SPierre Jolivet     PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg));
17185e642048SPierre Jolivet     if (data->is || (data->N > 1 && flg)) {
17199bb5c669SPierre Jolivet       if (block || overlap != -1) {
172002800ff6SPierre Jolivet         PetscCall(ISDestroy(&data->is));
172102800ff6SPierre Jolivet         PetscCall(MatDestroy(&data->aux));
17225e642048SPierre Jolivet       } else if (data->N > 1 && flg) {
172313044ca3SPierre Jolivet         PCHPDDMSchurPreType type = PC_HPDDM_SCHUR_PRE_GENEO;
172413044ca3SPierre Jolivet 
172513044ca3SPierre Jolivet         PetscCall(PetscOptionsGetEnum(nullptr, pcpre, "-pc_hpddm_schur_precondition", PCHPDDMSchurPreTypes, (PetscEnum *)&type, &flg));
172613044ca3SPierre Jolivet         if (type == PC_HPDDM_SCHUR_PRE_LEAST_SQUARES) {
172713044ca3SPierre Jolivet           PetscCall(ISDestroy(&data->is)); /* destroy any previously user-set objects since they will be set automatically */
172813044ca3SPierre Jolivet           PetscCall(MatDestroy(&data->aux));
172913044ca3SPierre Jolivet         } else if (type == PC_HPDDM_SCHUR_PRE_GENEO) {
173013044ca3SPierre Jolivet           PetscContainer container = nullptr;
173113044ca3SPierre Jolivet 
173213044ca3SPierre Jolivet           PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Schur", (PetscObject *)&container));
173313044ca3SPierre Jolivet           if (!container) { /* first call to PCSetUp() on the PC associated to the Schur complement */
173413044ca3SPierre Jolivet             PC_HPDDM *data_00;
173513044ca3SPierre Jolivet             KSP       ksp, inner_ksp;
173613044ca3SPierre Jolivet             PC        pc_00;
1737cf67ef9dSPierre Jolivet             Mat       A11 = nullptr;
1738cf67ef9dSPierre Jolivet             Vec       d   = nullptr;
173913044ca3SPierre Jolivet             char     *prefix;
174013044ca3SPierre Jolivet 
174113044ca3SPierre Jolivet             PetscCall(MatSchurComplementGetKSP(P, &ksp));
174213044ca3SPierre Jolivet             PetscCall(KSPGetPC(ksp, &pc_00));
174313044ca3SPierre Jolivet             PetscCall(PetscObjectTypeCompare((PetscObject)pc_00, PCHPDDM, &flg));
1744fd310a01SPierre 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 : "",
174513044ca3SPierre Jolivet                        ((PetscObject)pc_00)->type_name, PCHPDDM);
174613044ca3SPierre Jolivet             data_00 = (PC_HPDDM *)pc_00->data;
1747fd310a01SPierre 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],
1748fd310a01SPierre Jolivet                        data_00->N, data_00->N > 1 ? "s" : "", ((PetscObject)pc_00)->prefix);
174913044ca3SPierre Jolivet             PetscCall(PetscObjectTypeCompare((PetscObject)data_00->levels[0]->pc, PCASM, &flg));
1750fd310a01SPierre 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,
175113044ca3SPierre Jolivet                        ((PetscObject)data_00->levels[0]->pc)->type_name, PCASM);
1752d16c0b94SPierre Jolivet             PetscCall(MatSchurComplementGetSubMatrices(P, nullptr, nullptr, nullptr, nullptr, &A11));
17535e642048SPierre Jolivet             if (PetscDefined(USE_DEBUG) || !data->is) {
17545e642048SPierre Jolivet               Mat A01, A10, B = nullptr, C = nullptr, *sub;
175513044ca3SPierre Jolivet 
17565e642048SPierre Jolivet               PetscCall(MatSchurComplementGetSubMatrices(P, &A, nullptr, &A01, &A10, nullptr));
17575e642048SPierre Jolivet               PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, &flg));
17585e642048SPierre Jolivet               if (flg) {
17595e642048SPierre Jolivet                 PetscCall(MatTransposeGetMat(A10, &C));
17605e642048SPierre Jolivet                 PetscCall(MatTranspose(C, MAT_INITIAL_MATRIX, &B));
17615e642048SPierre Jolivet               } else {
17625e642048SPierre Jolivet                 PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, &flg));
17635e642048SPierre Jolivet                 if (flg) {
17645e642048SPierre Jolivet                   PetscCall(MatHermitianTransposeGetMat(A10, &C));
17655e642048SPierre Jolivet                   PetscCall(MatHermitianTranspose(C, MAT_INITIAL_MATRIX, &B));
17665e642048SPierre Jolivet                 }
17675e642048SPierre Jolivet               }
1768811e8887SPierre Jolivet               if (flg)
1769811e8887SPierre 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));
17705e642048SPierre Jolivet               if (!B) {
17715e642048SPierre Jolivet                 B = A10;
17725e642048SPierre Jolivet                 PetscCall(PetscObjectReference((PetscObject)B));
17735e642048SPierre Jolivet               } else if (!data->is) {
17745e642048SPierre Jolivet                 PetscCall(PetscObjectTypeCompareAny((PetscObject)A01, &flg, MATTRANSPOSEVIRTUAL, MATHERMITIANTRANSPOSEVIRTUAL, ""));
17755e642048SPierre Jolivet                 if (!flg) C = A01;
1776811e8887SPierre Jolivet                 else
1777811e8887SPierre 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));
17785e642048SPierre Jolivet               }
177913044ca3SPierre Jolivet               PetscCall(ISCreateStride(PETSC_COMM_SELF, B->rmap->N, 0, 1, &uis));
17805e642048SPierre Jolivet               PetscCall(ISSetIdentity(uis));
17815e642048SPierre Jolivet               if (!data->is) {
17825e642048SPierre Jolivet                 if (C) PetscCall(PetscObjectReference((PetscObject)C));
17835e642048SPierre Jolivet                 else PetscCall(MatTranspose(B, MAT_INITIAL_MATRIX, &C));
17845e642048SPierre Jolivet                 PetscCall(ISDuplicate(data_00->is, is));
17855e642048SPierre Jolivet                 PetscCall(MatIncreaseOverlap(A, 1, is, 1));
17865e642048SPierre Jolivet                 PetscCall(MatSetOption(C, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
17875e642048SPierre Jolivet                 PetscCall(MatCreateSubMatrices(C, 1, is, &uis, MAT_INITIAL_MATRIX, &sub));
17885e642048SPierre Jolivet                 PetscCall(MatDestroy(&C));
17895e642048SPierre Jolivet                 PetscCall(MatTranspose(sub[0], MAT_INITIAL_MATRIX, &C));
17905e642048SPierre Jolivet                 PetscCall(MatDestroySubMatrices(1, &sub));
17915e642048SPierre Jolivet                 PetscCall(MatFindNonzeroRows(C, &data->is));
17925e642048SPierre Jolivet                 PetscCall(MatDestroy(&C));
17935e642048SPierre Jolivet                 PetscCall(ISDestroy(is));
1794d16c0b94SPierre Jolivet                 PetscCall(ISCreateStride(PetscObjectComm((PetscObject)data->is), A11->rmap->n, A11->rmap->rstart, 1, &loc));
1795d16c0b94SPierre Jolivet                 if (PetscDefined(USE_DEBUG)) PetscCall(PCHPDDMCheckInclusion_Private(pc, data->is, loc, PETSC_FALSE));
1796d16c0b94SPierre Jolivet                 PetscCall(ISExpand(data->is, loc, is));
1797d16c0b94SPierre Jolivet                 PetscCall(ISDestroy(&loc));
1798d16c0b94SPierre Jolivet                 PetscCall(ISDestroy(&data->is));
1799d16c0b94SPierre Jolivet                 data->is = is[0];
1800d16c0b94SPierre Jolivet                 is[0]    = nullptr;
18015e642048SPierre Jolivet               }
18025e642048SPierre Jolivet               if (PetscDefined(USE_DEBUG)) {
18035e642048SPierre Jolivet                 PetscCall(PCHPDDMCheckSymmetry_Private(pc, A01, A10));
180413044ca3SPierre 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 */
180513044ca3SPierre Jolivet                 PetscCall(ISDestroy(&uis));
180613044ca3SPierre Jolivet                 PetscCall(ISDuplicate(data->is, &uis));
180713044ca3SPierre Jolivet                 PetscCall(ISSort(uis));
18085e642048SPierre Jolivet                 PetscCall(ISComplement(uis, 0, B->rmap->N, is));
180913044ca3SPierre Jolivet                 PetscCall(MatDuplicate(sub[0], MAT_COPY_VALUES, &C));
18105e642048SPierre Jolivet                 PetscCall(MatZeroRowsIS(C, is[0], 0.0, nullptr, nullptr));
18115e642048SPierre Jolivet                 PetscCall(ISDestroy(is));
181213044ca3SPierre Jolivet                 PetscCall(MatMultEqual(sub[0], C, 20, &flg));
181313044ca3SPierre 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 */
181413044ca3SPierre Jolivet                 PetscCall(MatDestroy(&C));
181513044ca3SPierre Jolivet                 PetscCall(MatDestroySubMatrices(1, &sub));
181613044ca3SPierre Jolivet               }
18175e642048SPierre Jolivet               PetscCall(ISDestroy(&uis));
18185e642048SPierre Jolivet               PetscCall(MatDestroy(&B));
18195e642048SPierre Jolivet             }
1820cf67ef9dSPierre Jolivet             flg = PETSC_FALSE;
1821cf67ef9dSPierre Jolivet             if (!data->aux) {
1822cf67ef9dSPierre Jolivet               Mat D;
1823cf67ef9dSPierre Jolivet 
1824cf67ef9dSPierre Jolivet               PetscCall(MatCreateVecs(A11, &d, nullptr));
1825cf67ef9dSPierre Jolivet               PetscCall(MatGetDiagonal(A11, d));
1826cf67ef9dSPierre Jolivet               PetscCall(PetscObjectTypeCompareAny((PetscObject)A11, &flg, MATDIAGONAL, MATCONSTANTDIAGONAL, ""));
1827cf67ef9dSPierre Jolivet               if (!flg) {
1828cf67ef9dSPierre Jolivet                 PetscCall(MatCreateDiagonal(d, &D));
1829cf67ef9dSPierre Jolivet                 PetscCall(MatMultEqual(A11, D, 20, &flg));
1830cf67ef9dSPierre Jolivet                 PetscCall(MatDestroy(&D));
1831cf67ef9dSPierre Jolivet               }
1832cf67ef9dSPierre 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"));
1833cf67ef9dSPierre Jolivet             }
1834cf67ef9dSPierre Jolivet             if (data->Neumann != PETSC_BOOL3_TRUE && !flg && A11) {
1835cf67ef9dSPierre Jolivet               PetscReal norm;
1836cf67ef9dSPierre Jolivet 
1837cf67ef9dSPierre Jolivet               PetscCall(MatNorm(A11, NORM_INFINITY, &norm));
1838cf67ef9dSPierre 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 : "");
1839cf67ef9dSPierre 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"));
1840cf67ef9dSPierre Jolivet               PetscCall(MatDestroy(&data->aux));
1841cf67ef9dSPierre Jolivet               flg = PETSC_TRUE;
1842cf67ef9dSPierre Jolivet             }
1843cf67ef9dSPierre 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 */
184401e3c840SPierre Jolivet               PetscSF            scatter;
184513044ca3SPierre Jolivet               const PetscScalar *read;
1846cf67ef9dSPierre Jolivet               PetscScalar       *write, *diagonal = nullptr;
184713044ca3SPierre Jolivet 
184813044ca3SPierre Jolivet               PetscCall(MatDestroy(&data->aux));
184913044ca3SPierre Jolivet               PetscCall(ISGetLocalSize(data->is, &n));
1850cf67ef9dSPierre Jolivet               PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)P), n, PETSC_DECIDE, &xin));
1851cf67ef9dSPierre Jolivet               PetscCall(VecDuplicate(xin, &v));
1852cf67ef9dSPierre Jolivet               PetscCall(VecScatterCreate(xin, data->is, v, nullptr, &scatter));
185313044ca3SPierre Jolivet               PetscCall(VecSet(v, 1.0));
1854cf67ef9dSPierre Jolivet               PetscCall(VecSet(xin, 1.0));
1855cf67ef9dSPierre Jolivet               PetscCall(VecScatterBegin(scatter, v, xin, ADD_VALUES, SCATTER_REVERSE));
1856cf67ef9dSPierre Jolivet               PetscCall(VecScatterEnd(scatter, v, xin, ADD_VALUES, SCATTER_REVERSE)); /* v has the multiplicity of all unknowns on the overlap */
1857cf67ef9dSPierre Jolivet               PetscCall(PetscSFDestroy(&scatter));
1858cf67ef9dSPierre Jolivet               if (d) {
1859cf67ef9dSPierre Jolivet                 PetscCall(VecScatterCreate(d, data->is, v, nullptr, &scatter));
1860cf67ef9dSPierre Jolivet                 PetscCall(VecScatterBegin(scatter, d, v, INSERT_VALUES, SCATTER_FORWARD));
1861cf67ef9dSPierre Jolivet                 PetscCall(VecScatterEnd(scatter, d, v, INSERT_VALUES, SCATTER_FORWARD));
1862cf67ef9dSPierre Jolivet                 PetscCall(PetscSFDestroy(&scatter));
1863cf67ef9dSPierre Jolivet                 PetscCall(VecDestroy(&d));
1864cf67ef9dSPierre Jolivet                 PetscCall(PetscMalloc1(n, &diagonal));
1865cf67ef9dSPierre Jolivet                 PetscCall(VecGetArrayRead(v, &read));
1866cf67ef9dSPierre Jolivet                 PetscCallCXX(std::copy_n(read, n, diagonal));
1867cf67ef9dSPierre Jolivet                 PetscCall(VecRestoreArrayRead(v, &read));
1868cf67ef9dSPierre Jolivet               }
186913044ca3SPierre Jolivet               PetscCall(VecDestroy(&v));
187013044ca3SPierre Jolivet               PetscCall(VecCreateSeq(PETSC_COMM_SELF, n, &v));
1871cf67ef9dSPierre Jolivet               PetscCall(VecGetArrayRead(xin, &read));
187213044ca3SPierre Jolivet               PetscCall(VecGetArrayWrite(v, &write));
1873cf67ef9dSPierre 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];
1874cf67ef9dSPierre Jolivet               PetscCall(PetscFree(diagonal));
1875cf67ef9dSPierre Jolivet               PetscCall(VecRestoreArrayRead(xin, &read));
187613044ca3SPierre Jolivet               PetscCall(VecRestoreArrayWrite(v, &write));
1877cf67ef9dSPierre Jolivet               PetscCall(VecDestroy(&xin));
1878c3e1b152SPierre Jolivet               PetscCall(MatCreateDiagonal(v, &data->aux));
187913044ca3SPierre Jolivet               PetscCall(VecDestroy(&v));
188013044ca3SPierre Jolivet             }
188113044ca3SPierre Jolivet             uis  = data->is;
188213044ca3SPierre Jolivet             uaux = data->aux;
188313044ca3SPierre Jolivet             PetscCall(PetscObjectReference((PetscObject)uis));
188413044ca3SPierre Jolivet             PetscCall(PetscObjectReference((PetscObject)uaux));
188513044ca3SPierre Jolivet             PetscCall(PetscStrallocpy(pcpre, &prefix));
188613044ca3SPierre Jolivet             PetscCall(PCSetOptionsPrefix(pc, nullptr));
188713044ca3SPierre Jolivet             PetscCall(PCSetType(pc, PCKSP));                                    /* replace the PC associated to the Schur complement by PCKSP */
188813044ca3SPierre Jolivet             PetscCall(KSPCreate(PetscObjectComm((PetscObject)pc), &inner_ksp)); /* new KSP that will be attached to the previously set PC */
188973329a61SPierre Jolivet             pc->ops->setup = PCSetUp_KSP;
189013044ca3SPierre Jolivet             PetscCall(PetscObjectGetTabLevel((PetscObject)pc, &n));
189113044ca3SPierre Jolivet             PetscCall(PetscObjectSetTabLevel((PetscObject)inner_ksp, n + 2));
189213044ca3SPierre Jolivet             PetscCall(KSPSetOperators(inner_ksp, pc->mat, pc->pmat));
189313044ca3SPierre Jolivet             PetscCall(KSPSetOptionsPrefix(inner_ksp, std::string(std::string(prefix) + "pc_hpddm_").c_str()));
189413044ca3SPierre Jolivet             PetscCall(KSPSetSkipPCSetFromOptions(inner_ksp, PETSC_TRUE));
189513044ca3SPierre Jolivet             PetscCall(KSPSetFromOptions(inner_ksp));
189613044ca3SPierre Jolivet             PetscCall(KSPGetPC(inner_ksp, &inner));
189713044ca3SPierre Jolivet             PetscCall(PCSetOptionsPrefix(inner, nullptr));
189813044ca3SPierre Jolivet             PetscCall(PCSetType(inner, PCNONE)); /* no preconditioner since the action of M^-1 A or A M^-1 will be computed by the Amat */
189913044ca3SPierre Jolivet             PetscCall(PCKSPSetKSP(pc, inner_ksp));
190013044ca3SPierre Jolivet             PetscCall(PetscNew(&ctx));    /* context to pass data around for the inner-most PC, which will be a proper PCHPDDM */
190113044ca3SPierre Jolivet             std::get<0>(*ctx)[0] = pc_00; /* for coarse correction on the primal (e.g., velocity) space */
190213044ca3SPierre Jolivet             PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &std::get<0>(*ctx)[1]));
190313044ca3SPierre Jolivet             PetscCall(PCSetOptionsPrefix(std::get<0>(*ctx)[1], prefix));
190413044ca3SPierre Jolivet             PetscCall(PetscFree(prefix));
190513044ca3SPierre Jolivet             PetscCall(PCSetOperators(std::get<0>(*ctx)[1], pc->mat, pc->pmat));
190613044ca3SPierre Jolivet             PetscCall(PCSetType(std::get<0>(*ctx)[1], PCHPDDM));
190713044ca3SPierre 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 */
1908cf67ef9dSPierre Jolivet             if (flg) static_cast<PC_HPDDM *>(std::get<0>(*ctx)[1]->data)->Neumann = PETSC_BOOL3_TRUE;
190913044ca3SPierre Jolivet             PetscCall(PCSetFromOptions(std::get<0>(*ctx)[1]));
191013044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)uis));
191113044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)uaux));
191213044ca3SPierre 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 */
191313044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(S, MATOP_MULT, (void (*)(void))MatMult_SchurCorrection));
191413044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(S, MATOP_VIEW, (void (*)(void))MatView_SchurCorrection));
191513044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(S, MATOP_DESTROY, (void (*)(void))MatDestroy_SchurCorrection));
191613044ca3SPierre Jolivet             PetscCall(KSPGetPCSide(inner_ksp, &(std::get<2>(*ctx))));
191713044ca3SPierre Jolivet             if (std::get<2>(*ctx) == PC_LEFT || std::get<2>(*ctx) == PC_SIDE_DEFAULT) {
191813044ca3SPierre Jolivet               PetscCall(KSPSetPreSolve(inner_ksp, KSPPreSolve_SchurCorrection, ctx));
191913044ca3SPierre Jolivet             } else { /* no support for PC_SYMMETRIC */
192013044ca3SPierre 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]);
192113044ca3SPierre Jolivet             }
192213044ca3SPierre Jolivet             PetscCall(KSPSetPostSolve(inner_ksp, KSPPostSolve_SchurCorrection, ctx));
1923715c1178SPierre Jolivet             PetscCall(PetscObjectContainerCompose((PetscObject)std::get<0>(*ctx)[1], "_PCHPDDM_Schur", ctx, nullptr));
192413044ca3SPierre Jolivet             PetscCall(PCSetUp(std::get<0>(*ctx)[1]));
192513044ca3SPierre Jolivet             PetscCall(KSPSetOperators(inner_ksp, S, S));
192613044ca3SPierre Jolivet             PetscCall(MatCreateVecs(std::get<1>(*ctx)[0], std::get<3>(*ctx), std::get<3>(*ctx) + 1));
192713044ca3SPierre Jolivet             PetscCall(VecDuplicate(std::get<3>(*ctx)[0], std::get<3>(*ctx) + 2));
192813044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)inner_ksp));
192913044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)S));
19300307214fSPierre Jolivet             for (std::vector<Vec>::iterator it = initial.begin(); it != initial.end(); ++it) PetscCall(VecDestroy(&*it));
193113044ca3SPierre Jolivet             PetscFunctionReturn(PETSC_SUCCESS);
193213044ca3SPierre Jolivet           } else { /* second call to PCSetUp() on the PC associated to the Schur complement, retrieve previously set context */
193313044ca3SPierre Jolivet             PetscCall(PetscContainerGetPointer(container, (void **)&ctx));
193413044ca3SPierre Jolivet           }
193513044ca3SPierre Jolivet         }
193613044ca3SPierre Jolivet       }
193713044ca3SPierre Jolivet     }
1938f1580f4eSBarry Smith     if (!data->is && data->N > 1) {
1939f1580f4eSBarry Smith       char type[256] = {}; /* same size as in src/ksp/pc/interface/pcset.c */
1940811e8887SPierre Jolivet 
1941f1580f4eSBarry Smith       PetscCall(PetscObjectTypeCompareAny((PetscObject)P, &flg, MATNORMAL, MATNORMALHERMITIAN, ""));
1942f1580f4eSBarry Smith       if (flg || (A->rmap->N != A->cmap->N && P->rmap->N == P->cmap->N && P->rmap->N == A->cmap->N)) {
1943f1580f4eSBarry Smith         Mat B;
1944811e8887SPierre Jolivet 
1945f1580f4eSBarry Smith         PetscCall(PCHPDDMSetAuxiliaryMatNormal_Private(pc, A, P, &B, pcpre));
1946f1580f4eSBarry Smith         if (data->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED) data->correction = PC_HPDDM_COARSE_CORRECTION_BALANCED;
1947f1580f4eSBarry Smith         PetscCall(MatDestroy(&B));
1948f1580f4eSBarry Smith       } else {
1949f1580f4eSBarry Smith         PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg));
1950f1580f4eSBarry Smith         if (flg) {
1951f1580f4eSBarry Smith           Mat                 A00, P00, A01, A10, A11, B, N;
195213044ca3SPierre Jolivet           PCHPDDMSchurPreType type = PC_HPDDM_SCHUR_PRE_LEAST_SQUARES;
195313044ca3SPierre Jolivet 
195413044ca3SPierre Jolivet           PetscCall(MatSchurComplementGetSubMatrices(P, &A00, &P00, &A01, &A10, &A11));
195513044ca3SPierre Jolivet           PetscCall(PetscOptionsGetEnum(nullptr, pcpre, "-pc_hpddm_schur_precondition", PCHPDDMSchurPreTypes, (PetscEnum *)&type, &flg));
195613044ca3SPierre Jolivet           if (type == PC_HPDDM_SCHUR_PRE_LEAST_SQUARES) {
1957281f8ce6SPierre Jolivet             Mat                        B01;
19583df4cd7bSPierre Jolivet             Vec                        diagonal = nullptr;
1959f1580f4eSBarry Smith             const PetscScalar         *array;
1960f1580f4eSBarry Smith             MatSchurComplementAinvType type;
1961f1580f4eSBarry Smith 
1962281f8ce6SPierre Jolivet             PetscCall(PCHPDDMCheckSymmetry_Private(pc, A01, A10, &B01));
1963f1580f4eSBarry Smith             if (A11) {
19643df4cd7bSPierre Jolivet               PetscCall(MatCreateVecs(A11, &diagonal, nullptr));
19653df4cd7bSPierre Jolivet               PetscCall(MatGetDiagonal(A11, diagonal));
1966f1580f4eSBarry Smith             }
1967db4a47b3SPierre Jolivet             PetscCall(MatCreateVecs(P00, &v, nullptr));
1968f1580f4eSBarry Smith             PetscCall(MatSchurComplementGetAinvType(P, &type));
1969f1580f4eSBarry 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]);
1970f1580f4eSBarry Smith             if (type == MAT_SCHUR_COMPLEMENT_AINV_LUMP) {
1971f1580f4eSBarry Smith               PetscCall(MatGetRowSum(P00, v));
1972f1580f4eSBarry Smith               if (A00 == P00) PetscCall(PetscObjectReference((PetscObject)A00));
1973f1580f4eSBarry Smith               PetscCall(MatDestroy(&P00));
1974f1580f4eSBarry Smith               PetscCall(VecGetArrayRead(v, &array));
1975db4a47b3SPierre Jolivet               PetscCall(MatCreateAIJ(PetscObjectComm((PetscObject)A00), A00->rmap->n, A00->cmap->n, A00->rmap->N, A00->cmap->N, 1, nullptr, 0, nullptr, &P00));
1976f1580f4eSBarry Smith               PetscCall(MatSetOption(P00, MAT_NO_OFF_PROC_ENTRIES, PETSC_TRUE));
1977f1580f4eSBarry Smith               for (n = A00->rmap->rstart; n < A00->rmap->rend; ++n) PetscCall(MatSetValue(P00, n, n, array[n - A00->rmap->rstart], INSERT_VALUES));
1978f1580f4eSBarry Smith               PetscCall(MatAssemblyBegin(P00, MAT_FINAL_ASSEMBLY));
1979f1580f4eSBarry Smith               PetscCall(MatAssemblyEnd(P00, MAT_FINAL_ASSEMBLY));
1980f1580f4eSBarry Smith               PetscCall(VecRestoreArrayRead(v, &array));
1981f1580f4eSBarry Smith               PetscCall(MatSchurComplementUpdateSubMatrices(P, A00, P00, A01, A10, A11)); /* replace P00 by diag(sum of each row of P00) */
1982f1580f4eSBarry Smith               PetscCall(MatDestroy(&P00));
1983f1580f4eSBarry Smith             } else PetscCall(MatGetDiagonal(P00, v));
1984f1580f4eSBarry Smith             PetscCall(VecReciprocal(v)); /* inv(diag(P00))       */
1985f1580f4eSBarry Smith             PetscCall(VecSqrtAbs(v));    /* sqrt(inv(diag(P00))) */
1986f1580f4eSBarry Smith             PetscCall(MatDuplicate(A01, MAT_COPY_VALUES, &B));
1987db4a47b3SPierre Jolivet             PetscCall(MatDiagonalScale(B, v, nullptr));
1988281f8ce6SPierre Jolivet             if (B01) PetscCall(MatDiagonalScale(B01, v, nullptr));
1989f1580f4eSBarry Smith             PetscCall(VecDestroy(&v));
1990f1580f4eSBarry Smith             PetscCall(MatCreateNormalHermitian(B, &N));
1991281f8ce6SPierre Jolivet             PetscCall(PCHPDDMSetAuxiliaryMatNormal_Private(pc, B, N, &P, pcpre, &diagonal, B01));
1992f1580f4eSBarry Smith             PetscCall(PetscObjectTypeCompare((PetscObject)data->aux, MATSEQAIJ, &flg));
1993f1580f4eSBarry Smith             if (!flg) {
1994f1580f4eSBarry Smith               PetscCall(MatDestroy(&P));
1995f1580f4eSBarry Smith               P = N;
1996f1580f4eSBarry Smith               PetscCall(PetscObjectReference((PetscObject)P));
1997feebddf4SPierre Jolivet             }
19983df4cd7bSPierre Jolivet             if (diagonal) {
19993df4cd7bSPierre Jolivet               PetscCall(MatDiagonalSet(P, diagonal, ADD_VALUES));
2000feebddf4SPierre Jolivet               PetscCall(PCSetOperators(pc, P, P)); /* replace P by A01^T inv(diag(P00)) A01 - diag(P11) */
20013df4cd7bSPierre Jolivet               PetscCall(VecDestroy(&diagonal));
2002281f8ce6SPierre Jolivet             } else PetscCall(PCSetOperators(pc, B01 ? P : N, P));  /* replace P by A01^T inv(diag(P00)) A01                         */
2003feebddf4SPierre Jolivet             pc->ops->postsolve = PCPostSolve_SchurPreLeastSquares; /*  PCFIELDSPLIT expect a KSP for (P11 - A10 inv(diag(P00)) A01) */
2004feebddf4SPierre Jolivet             PetscCall(MatDestroy(&N));                             /*  but a PC for (A10 inv(diag(P00)) A10 - P11) is setup instead */
2005feebddf4SPierre Jolivet             PetscCall(MatDestroy(&P));                             /*  so the sign of the solution must be flipped                  */
2006f1580f4eSBarry Smith             PetscCall(MatDestroy(&B));
200713044ca3SPierre Jolivet           } else
2008fd310a01SPierre 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 : "");
20090307214fSPierre Jolivet           for (std::vector<Vec>::iterator it = initial.begin(); it != initial.end(); ++it) PetscCall(VecDestroy(&*it));
20103ba16761SJacob Faibussowitsch           PetscFunctionReturn(PETSC_SUCCESS);
2011f1580f4eSBarry Smith         } else {
2012db4a47b3SPierre Jolivet           PetscCall(PetscOptionsGetString(nullptr, pcpre, "-pc_hpddm_levels_1_st_pc_type", type, sizeof(type), nullptr));
2013f1580f4eSBarry Smith           PetscCall(PetscStrcmp(type, PCMAT, &algebraic));
2014b07dfdedSPierre Jolivet           PetscCheck(!algebraic || !block, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_levels_1_st_pc_type mat and -pc_hpddm_block_splitting");
20159bb5c669SPierre Jolivet           if (overlap != -1) {
20169bb5c669SPierre 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");
20179bb5c669SPierre Jolivet             PetscCheck(overlap >= 1, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_WRONG, "-pc_hpddm_harmonic_overlap %" PetscInt_FMT " < 1", overlap);
20189bb5c669SPierre Jolivet           }
20199bb5c669SPierre Jolivet           if (block || overlap != -1) algebraic = PETSC_TRUE;
2020f1580f4eSBarry Smith           if (algebraic) {
2021f1580f4eSBarry Smith             PetscCall(ISCreateStride(PETSC_COMM_SELF, P->rmap->n, P->rmap->rstart, 1, &data->is));
2022f1580f4eSBarry Smith             PetscCall(MatIncreaseOverlap(P, 1, &data->is, 1));
2023f1580f4eSBarry Smith             PetscCall(ISSort(data->is));
20249bb5c669SPierre Jolivet           } else
20259bb5c669SPierre 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 : ""));
2026f1580f4eSBarry Smith         }
2027f1580f4eSBarry Smith       }
2028f1580f4eSBarry Smith     }
2029f1580f4eSBarry Smith   }
2030398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG)
2031398c7888SPierre Jolivet   if (data->is) PetscCall(ISDuplicate(data->is, &dis));
2032398c7888SPierre Jolivet   if (data->aux) PetscCall(MatDuplicate(data->aux, MAT_COPY_VALUES, &daux));
2033398c7888SPierre Jolivet #endif
2034f1580f4eSBarry Smith   if (data->is || (ismatis && data->N > 1)) {
2035f1580f4eSBarry Smith     if (ismatis) {
2036f1580f4eSBarry Smith       std::initializer_list<std::string> list = {MATSEQBAIJ, MATSEQSBAIJ};
2037f1580f4eSBarry Smith       PetscCall(MatISGetLocalMat(P, &N));
2038f1580f4eSBarry Smith       std::initializer_list<std::string>::const_iterator it = std::find(list.begin(), list.end(), ((PetscObject)N)->type_name);
2039f1580f4eSBarry Smith       PetscCall(MatISRestoreLocalMat(P, &N));
2040f1580f4eSBarry Smith       switch (std::distance(list.begin(), it)) {
2041d71ae5a4SJacob Faibussowitsch       case 0:
2042d71ae5a4SJacob Faibussowitsch         PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &C));
2043d71ae5a4SJacob Faibussowitsch         break;
2044f1580f4eSBarry Smith       case 1:
2045f1580f4eSBarry Smith         /* MatCreateSubMatrices() does not work with MATSBAIJ and unsorted ISes, so convert to MPIBAIJ */
2046f1580f4eSBarry Smith         PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &C));
2047f1580f4eSBarry Smith         PetscCall(MatSetOption(C, MAT_SYMMETRIC, PETSC_TRUE));
2048f1580f4eSBarry Smith         break;
2049d71ae5a4SJacob Faibussowitsch       default:
2050d71ae5a4SJacob Faibussowitsch         PetscCall(MatConvert(P, MATMPIAIJ, MAT_INITIAL_MATRIX, &C));
2051f1580f4eSBarry Smith       }
2052db4a47b3SPierre Jolivet       PetscCall(MatISGetLocalToGlobalMapping(P, &l2g, nullptr));
2053f1580f4eSBarry Smith       PetscCall(PetscObjectReference((PetscObject)P));
2054f1580f4eSBarry Smith       PetscCall(KSPSetOperators(data->levels[0]->ksp, A, C));
2055f1580f4eSBarry Smith       std::swap(C, P);
2056f1580f4eSBarry Smith       PetscCall(ISLocalToGlobalMappingGetSize(l2g, &n));
2057f1580f4eSBarry Smith       PetscCall(ISCreateStride(PETSC_COMM_SELF, n, 0, 1, &loc));
2058f1580f4eSBarry Smith       PetscCall(ISLocalToGlobalMappingApplyIS(l2g, loc, &is[0]));
2059f1580f4eSBarry Smith       PetscCall(ISDestroy(&loc));
2060f1580f4eSBarry Smith       /* the auxiliary Mat is _not_ the local Neumann matrix                                */
2061f1580f4eSBarry Smith       /* it is the local Neumann matrix augmented (with zeros) through MatIncreaseOverlap() */
2062c8ea6600SPierre Jolivet       data->Neumann = PETSC_BOOL3_FALSE;
2063f1580f4eSBarry Smith       structure     = SAME_NONZERO_PATTERN;
2064f1580f4eSBarry Smith     } else {
2065f1580f4eSBarry Smith       is[0] = data->is;
206613044ca3SPierre Jolivet       if (algebraic || ctx) subdomains = PETSC_TRUE;
2067db4a47b3SPierre Jolivet       PetscCall(PetscOptionsGetBool(nullptr, pcpre, "-pc_hpddm_define_subdomains", &subdomains, nullptr));
206813044ca3SPierre 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);
2069c8ea6600SPierre Jolivet       if (PetscBool3ToBool(data->Neumann)) {
2070b07dfdedSPierre Jolivet         PetscCheck(!block, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_block_splitting and -pc_hpddm_has_neumann");
20719bb5c669SPierre Jolivet         PetscCheck(overlap == -1, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_harmonic_overlap %" PetscInt_FMT " and -pc_hpddm_has_neumann", overlap);
2072b07dfdedSPierre Jolivet         PetscCheck(!algebraic, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_levels_1_st_pc_type mat and -pc_hpddm_has_neumann");
2073f1580f4eSBarry Smith       }
2074c8ea6600SPierre Jolivet       if (PetscBool3ToBool(data->Neumann) || block) structure = SAME_NONZERO_PATTERN;
2075f1580f4eSBarry Smith       PetscCall(ISCreateStride(PetscObjectComm((PetscObject)data->is), P->rmap->n, P->rmap->rstart, 1, &loc));
2076f1580f4eSBarry Smith     }
2077f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : ""));
2078db4a47b3SPierre Jolivet     PetscCall(PetscOptionsGetEnum(nullptr, prefix, "-st_matstructure", MatStructures, (PetscEnum *)&structure, &flg)); /* if not user-provided, force its value when possible */
2079f1580f4eSBarry Smith     if (!flg && structure == SAME_NONZERO_PATTERN) {                                                                   /* cannot call STSetMatStructure() yet, insert the appropriate option in the database, parsed by STSetFromOptions() */
2080f1580f4eSBarry Smith       PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-%spc_hpddm_levels_1_st_matstructure", pcpre ? pcpre : ""));
2081db4a47b3SPierre Jolivet       PetscCall(PetscOptionsSetValue(nullptr, prefix, MatStructures[structure]));
2082f1580f4eSBarry Smith     }
2083398c7888SPierre Jolivet     flg = PETSC_FALSE;
2084b07dfdedSPierre Jolivet     if (data->share) {
2085398c7888SPierre Jolivet       data->share = PETSC_FALSE; /* will be reset to PETSC_TRUE if none of the conditions below are true */
2086398c7888SPierre 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 : ""));
2087398c7888SPierre Jolivet       else if (data->deflation) PetscCall(PetscInfo(pc, "Nothing to share since PCHPDDMSetDeflationMat() has been called\n"));
2088398c7888SPierre Jolivet       else if (ismatis) PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc with a Pmat of type MATIS\n"));
2089b07dfdedSPierre Jolivet       else if (!algebraic && structure != SAME_NONZERO_PATTERN)
2090398c7888SPierre 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]));
2091398c7888SPierre Jolivet       else data->share = PETSC_TRUE;
2092398c7888SPierre Jolivet     }
2093398c7888SPierre Jolivet     if (!ismatis) {
2094398c7888SPierre Jolivet       if (data->share || (!PetscBool3ToBool(data->Neumann) && subdomains)) PetscCall(ISDuplicate(is[0], &unsorted));
2095398c7888SPierre Jolivet       else unsorted = is[0];
2096398c7888SPierre Jolivet     }
2097f1580f4eSBarry Smith     if (data->N > 1 && (data->aux || ismatis || algebraic)) {
2098f1580f4eSBarry Smith       PetscCheck(loadedSym, PETSC_COMM_SELF, PETSC_ERR_PLIB, "HPDDM library not loaded, cannot use more than one level");
2099f1580f4eSBarry Smith       PetscCall(MatSetOption(P, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
2100f1580f4eSBarry Smith       if (ismatis) {
2101f1580f4eSBarry Smith         /* needed by HPDDM (currently) so that the partition of unity is 0 on subdomain interfaces */
2102f1580f4eSBarry Smith         PetscCall(MatIncreaseOverlap(P, 1, is, 1));
2103f1580f4eSBarry Smith         PetscCall(ISDestroy(&data->is));
2104f1580f4eSBarry Smith         data->is = is[0];
2105f1580f4eSBarry Smith       } else {
2106d16c0b94SPierre Jolivet         if (PetscDefined(USE_DEBUG)) PetscCall(PCHPDDMCheckInclusion_Private(pc, data->is, loc, PETSC_TRUE));
21079bb5c669SPierre Jolivet         if (overlap == -1) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", PCHPDDMAlgebraicAuxiliaryMat_Private));
21089bb5c669SPierre Jolivet         if (!PetscBool3ToBool(data->Neumann) && (!algebraic || overlap != -1)) {
2109f1580f4eSBarry Smith           PetscCall(PetscObjectTypeCompare((PetscObject)P, MATMPISBAIJ, &flg));
2110f1580f4eSBarry Smith           if (flg) {
2111f1580f4eSBarry Smith             /* maybe better to ISSort(is[0]), MatCreateSubMatrices(), and then MatPermute() */
2112f1580f4eSBarry Smith             /* but there is no MatPermute_SeqSBAIJ(), so as before, just use MATMPIBAIJ     */
2113f1580f4eSBarry Smith             PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &uaux));
2114f1580f4eSBarry Smith             flg = PETSC_FALSE;
2115f1580f4eSBarry Smith           }
2116f1580f4eSBarry Smith         }
2117f1580f4eSBarry Smith       }
21189bb5c669SPierre Jolivet       if (algebraic && overlap == -1) {
2119f1580f4eSBarry Smith         PetscUseMethod(pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", (Mat, IS *, Mat *[], PetscBool), (P, is, &sub, block));
2120f1580f4eSBarry Smith         if (block) {
2121f1580f4eSBarry Smith           PetscCall(PetscObjectQuery((PetscObject)sub[0], "_PCHPDDM_Neumann_Mat", (PetscObject *)&data->aux));
2122db4a47b3SPierre Jolivet           PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Neumann_Mat", nullptr));
2123f1580f4eSBarry Smith         }
21249bb5c669SPierre Jolivet       } else if (!uaux || overlap != -1) {
212513044ca3SPierre Jolivet         if (!ctx) {
2126c8ea6600SPierre Jolivet           if (PetscBool3ToBool(data->Neumann)) sub = &data->aux;
21279bb5c669SPierre Jolivet           else {
2128df21ef68SPierre Jolivet             PetscBool flg;
21299bb5c669SPierre Jolivet             if (overlap != -1) {
21309bb5c669SPierre Jolivet               Harmonic              h;
21319bb5c669SPierre Jolivet               Mat                   A0, *a;                    /* with an SVD: [ A_00  A_01       ] */
21329bb5c669SPierre Jolivet               IS                    ov[2], rows, cols, stride; /*              [ A_10  A_11  A_12 ] */
213358b7e2c1SStefano Zampini               const PetscInt       *i[2], bs = P->cmap->bs;    /* with a GEVP: [ A_00  A_01       ] */
21349bb5c669SPierre Jolivet               PetscInt              n[2];                      /*              [ A_10  A_11  A_12 ] */
21359bb5c669SPierre Jolivet               std::vector<PetscInt> v[2];                      /*              [       A_21  A_22 ] */
21369bb5c669SPierre Jolivet 
21379bb5c669SPierre Jolivet               PetscCall(ISDuplicate(data->is, ov));
21389bb5c669SPierre Jolivet               if (overlap > 1) PetscCall(MatIncreaseOverlap(P, 1, ov, overlap - 1));
21399bb5c669SPierre Jolivet               PetscCall(ISDuplicate(ov[0], ov + 1));
21409bb5c669SPierre Jolivet               PetscCall(MatIncreaseOverlap(P, 1, ov + 1, 1));
21419bb5c669SPierre Jolivet               PetscCall(PetscNew(&h));
21429bb5c669SPierre Jolivet               h->ksp = nullptr;
21439bb5c669SPierre Jolivet               PetscCall(PetscCalloc1(2, &h->A));
21443a60673dSPierre Jolivet               PetscCall(PetscOptionsHasName(nullptr, prefix, "-svd_nsv", &flg));
21459bb5c669SPierre Jolivet               if (!flg) PetscCall(PetscOptionsHasName(nullptr, prefix, "-svd_relative_threshold", &flg));
21469bb5c669SPierre Jolivet               PetscCall(ISSort(ov[0]));
21479bb5c669SPierre Jolivet               if (!flg) PetscCall(ISSort(ov[1]));
214832603206SJames Wright               PetscCall(PetscCalloc1(5, &h->is));
21499bb5c669SPierre Jolivet               PetscCall(MatCreateSubMatrices(uaux ? uaux : P, 1, ov + !flg, ov + 1, MAT_INITIAL_MATRIX, &a)); /* submatrix from above, either square (!flg) or rectangular (flg) */
21509bb5c669SPierre Jolivet               for (PetscInt j = 0; j < 2; ++j) {
21519bb5c669SPierre Jolivet                 PetscCall(ISGetIndices(ov[j], i + j));
21529bb5c669SPierre Jolivet                 PetscCall(ISGetLocalSize(ov[j], n + j));
21539bb5c669SPierre Jolivet               }
21549bb5c669SPierre Jolivet               v[1].reserve((n[1] - n[0]) / bs);
21559bb5c669SPierre Jolivet               for (PetscInt j = 0; j < n[1]; j += bs) { /* indices of the (2,2) block */
21569bb5c669SPierre Jolivet                 PetscInt location;
21579bb5c669SPierre Jolivet                 PetscCall(ISLocate(ov[0], i[1][j], &location));
21589bb5c669SPierre Jolivet                 if (location < 0) v[1].emplace_back(j / bs);
21599bb5c669SPierre Jolivet               }
21609bb5c669SPierre Jolivet               if (!flg) {
21619bb5c669SPierre Jolivet                 h->A[1] = a[0];
21629bb5c669SPierre Jolivet                 PetscCall(PetscObjectReference((PetscObject)h->A[1]));
21639bb5c669SPierre Jolivet                 v[0].reserve((n[0] - P->rmap->n) / bs);
21649bb5c669SPierre Jolivet                 for (PetscInt j = 0; j < n[1]; j += bs) { /* row indices of the (1,2) block */
21659bb5c669SPierre Jolivet                   PetscInt location;
21669bb5c669SPierre Jolivet                   PetscCall(ISLocate(loc, i[1][j], &location));
21679bb5c669SPierre Jolivet                   if (location < 0) {
21689bb5c669SPierre Jolivet                     PetscCall(ISLocate(ov[0], i[1][j], &location));
21699bb5c669SPierre Jolivet                     if (location >= 0) v[0].emplace_back(j / bs);
21709bb5c669SPierre Jolivet                   }
21719bb5c669SPierre Jolivet                 }
21729bb5c669SPierre Jolivet                 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[0].size(), v[0].data(), PETSC_USE_POINTER, &rows));
21739bb5c669SPierre Jolivet                 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[1].size(), v[1].data(), PETSC_COPY_VALUES, h->is + 4));
21749bb5c669SPierre Jolivet                 PetscCall(MatCreateSubMatrix(a[0], rows, h->is[4], MAT_INITIAL_MATRIX, h->A)); /* A_12 submatrix from above */
21759bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&rows));
21769bb5c669SPierre 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 */
21779bb5c669SPierre Jolivet                 PetscCall(ISEmbed(ov[0], ov[1], PETSC_TRUE, &rows));
21789bb5c669SPierre Jolivet                 PetscCall(MatCreateSubMatrix(a[0], rows, cols = rows, MAT_INITIAL_MATRIX, &A0)); /* [ A_00  A_01 ; A_10  A_11 ] submatrix from above */
21799bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&rows));
21809bb5c669SPierre Jolivet                 v[0].clear();
21819bb5c669SPierre Jolivet                 PetscCall(ISEmbed(loc, ov[1], PETSC_TRUE, h->is + 3));
21829bb5c669SPierre Jolivet                 PetscCall(ISEmbed(data->is, ov[1], PETSC_TRUE, h->is + 2));
21839bb5c669SPierre Jolivet               }
21849bb5c669SPierre Jolivet               v[0].reserve((n[0] - P->rmap->n) / bs);
21859bb5c669SPierre Jolivet               for (PetscInt j = 0; j < n[0]; j += bs) {
21869bb5c669SPierre Jolivet                 PetscInt location;
21879bb5c669SPierre Jolivet                 PetscCall(ISLocate(loc, i[0][j], &location));
21889bb5c669SPierre Jolivet                 if (location < 0) v[0].emplace_back(j / bs);
21899bb5c669SPierre Jolivet               }
21909bb5c669SPierre Jolivet               PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[0].size(), v[0].data(), PETSC_USE_POINTER, &rows));
21919bb5c669SPierre Jolivet               for (PetscInt j = 0; j < 2; ++j) PetscCall(ISRestoreIndices(ov[j], i + j));
21929bb5c669SPierre Jolivet               if (flg) {
21939bb5c669SPierre Jolivet                 IS is;
21949bb5c669SPierre Jolivet                 PetscCall(ISCreateStride(PETSC_COMM_SELF, a[0]->rmap->n, 0, 1, &is));
21959bb5c669SPierre Jolivet                 PetscCall(ISEmbed(ov[0], ov[1], PETSC_TRUE, &cols));
21969bb5c669SPierre Jolivet                 PetscCall(MatCreateSubMatrix(a[0], is, cols, MAT_INITIAL_MATRIX, &A0)); /* [ A_00  A_01 ; A_10  A_11 ] submatrix from above */
21979bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&cols));
21989bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&is));
21999bb5c669SPierre 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 */
22009bb5c669SPierre Jolivet                 PetscCall(ISEmbed(loc, data->is, PETSC_TRUE, h->is + 2));
22019bb5c669SPierre Jolivet                 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[1].size(), v[1].data(), PETSC_USE_POINTER, &cols));
22029bb5c669SPierre Jolivet                 PetscCall(MatCreateSubMatrix(a[0], rows, cols, MAT_INITIAL_MATRIX, h->A)); /* A_12 submatrix from above */
22039bb5c669SPierre Jolivet                 PetscCall(ISDestroy(&cols));
22049bb5c669SPierre Jolivet               }
22059bb5c669SPierre Jolivet               PetscCall(ISCreateStride(PETSC_COMM_SELF, A0->rmap->n, 0, 1, &stride));
22069bb5c669SPierre Jolivet               PetscCall(ISEmbed(rows, stride, PETSC_TRUE, h->is));
22079bb5c669SPierre Jolivet               PetscCall(ISDestroy(&stride));
22089bb5c669SPierre Jolivet               PetscCall(ISDestroy(&rows));
22099bb5c669SPierre Jolivet               PetscCall(ISEmbed(loc, ov[0], PETSC_TRUE, h->is + 1));
22109bb5c669SPierre Jolivet               if (subdomains) {
22119bb5c669SPierre Jolivet                 if (!data->levels[0]->pc) {
22129bb5c669SPierre Jolivet                   PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->pc));
22139bb5c669SPierre Jolivet                   PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : ""));
22149bb5c669SPierre Jolivet                   PetscCall(PCSetOptionsPrefix(data->levels[0]->pc, prefix));
22159bb5c669SPierre Jolivet                   PetscCall(PCSetOperators(data->levels[0]->pc, A, P));
22169bb5c669SPierre Jolivet                 }
22179bb5c669SPierre Jolivet                 PetscCall(PCSetType(data->levels[0]->pc, PCASM));
22189bb5c669SPierre Jolivet                 if (!data->levels[0]->pc->setupcalled) PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, ov + !flg, &loc));
22199bb5c669SPierre Jolivet                 PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(data->levels[0]->pc, flg ? A0 : a[0], PETSC_TRUE));
22209bb5c669SPierre Jolivet                 if (!flg) ++overlap;
22219bb5c669SPierre Jolivet                 if (data->share) {
22229bb5c669SPierre Jolivet                   PetscInt n = -1;
22239bb5c669SPierre Jolivet                   PetscTryMethod(data->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data->levels[0]->pc, &n, nullptr, &ksp));
22249bb5c669SPierre Jolivet                   PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", n);
22259bb5c669SPierre Jolivet                   if (flg) {
22269bb5c669SPierre Jolivet                     h->ksp = ksp[0];
22279bb5c669SPierre Jolivet                     PetscCall(PetscObjectReference((PetscObject)h->ksp));
22289bb5c669SPierre Jolivet                   }
22299bb5c669SPierre Jolivet                 }
22309bb5c669SPierre Jolivet               }
22319bb5c669SPierre Jolivet               if (!h->ksp) {
22329bb5c669SPierre Jolivet                 PetscBool share = data->share;
22339bb5c669SPierre Jolivet                 PetscCall(KSPCreate(PETSC_COMM_SELF, &h->ksp));
22349bb5c669SPierre Jolivet                 PetscCall(KSPSetType(h->ksp, KSPPREONLY));
22359bb5c669SPierre Jolivet                 PetscCall(KSPSetOperators(h->ksp, A0, A0));
22369bb5c669SPierre Jolivet                 do {
22379bb5c669SPierre Jolivet                   if (!data->share) {
22389bb5c669SPierre Jolivet                     share = PETSC_FALSE;
22399bb5c669SPierre Jolivet                     PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_%s", pcpre ? pcpre : "", flg ? "svd_" : "eps_"));
22409bb5c669SPierre Jolivet                     PetscCall(KSPSetOptionsPrefix(h->ksp, prefix));
22419bb5c669SPierre Jolivet                     PetscCall(KSPSetFromOptions(h->ksp));
22429bb5c669SPierre Jolivet                   } else {
22439bb5c669SPierre Jolivet                     MatSolverType type;
22449bb5c669SPierre Jolivet                     PetscCall(KSPGetPC(ksp[0], &pc));
22459bb5c669SPierre Jolivet                     PetscCall(PetscObjectTypeCompareAny((PetscObject)pc, &data->share, PCLU, PCCHOLESKY, ""));
22469bb5c669SPierre Jolivet                     if (data->share) {
22479bb5c669SPierre Jolivet                       PetscCall(PCFactorGetMatSolverType(pc, &type));
22489bb5c669SPierre Jolivet                       if (!type) {
22499bb5c669SPierre Jolivet                         if (PetscDefined(HAVE_MUMPS)) PetscCall(PCFactorSetMatSolverType(pc, MATSOLVERMUMPS));
22509bb5c669SPierre Jolivet                         else if (PetscDefined(HAVE_MKL_PARDISO)) PetscCall(PCFactorSetMatSolverType(pc, MATSOLVERMKL_PARDISO));
22519bb5c669SPierre Jolivet                         else data->share = PETSC_FALSE;
22529bb5c669SPierre Jolivet                         if (data->share) PetscCall(PCSetFromOptions(pc));
22539bb5c669SPierre Jolivet                       } else {
22549bb5c669SPierre Jolivet                         PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &data->share));
22559bb5c669SPierre Jolivet                         if (!data->share) PetscCall(PetscStrcmp(type, MATSOLVERMKL_PARDISO, &data->share));
22569bb5c669SPierre Jolivet                       }
22579bb5c669SPierre Jolivet                       if (data->share) {
22589bb5c669SPierre Jolivet                         std::tuple<KSP, IS, Vec[2]> *p;
22599bb5c669SPierre Jolivet                         PetscCall(PCFactorGetMatrix(pc, &A));
22609bb5c669SPierre Jolivet                         PetscCall(MatFactorSetSchurIS(A, h->is[4]));
22619bb5c669SPierre Jolivet                         PetscCall(KSPSetUp(ksp[0]));
22629bb5c669SPierre Jolivet                         PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_eps_shell_", pcpre ? pcpre : ""));
22639bb5c669SPierre Jolivet                         PetscCall(KSPSetOptionsPrefix(h->ksp, prefix));
22649bb5c669SPierre Jolivet                         PetscCall(KSPSetFromOptions(h->ksp));
22659bb5c669SPierre Jolivet                         PetscCall(KSPGetPC(h->ksp, &pc));
22669bb5c669SPierre Jolivet                         PetscCall(PCSetType(pc, PCSHELL));
22679bb5c669SPierre Jolivet                         PetscCall(PetscNew(&p));
22689bb5c669SPierre Jolivet                         std::get<0>(*p) = ksp[0];
22699bb5c669SPierre Jolivet                         PetscCall(ISEmbed(ov[0], ov[1], PETSC_TRUE, &std::get<1>(*p)));
22709bb5c669SPierre Jolivet                         PetscCall(MatCreateVecs(A, std::get<2>(*p), std::get<2>(*p) + 1));
22719bb5c669SPierre Jolivet                         PetscCall(PCShellSetContext(pc, p));
22729bb5c669SPierre Jolivet                         PetscCall(PCShellSetApply(pc, PCApply_Schur));
22739bb5c669SPierre Jolivet                         PetscCall(PCShellSetApplyTranspose(pc, PCApply_Schur<Vec, true>));
22749bb5c669SPierre Jolivet                         PetscCall(PCShellSetMatApply(pc, PCApply_Schur<Mat>));
22759bb5c669SPierre Jolivet                         PetscCall(PCShellSetDestroy(pc, PCDestroy_Schur));
22769bb5c669SPierre Jolivet                       }
22779bb5c669SPierre Jolivet                     }
22789bb5c669SPierre Jolivet                     if (!data->share) PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc since neither MUMPS nor MKL PARDISO is used\n"));
22799bb5c669SPierre Jolivet                   }
22809bb5c669SPierre 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 */
22819bb5c669SPierre Jolivet               }
22829bb5c669SPierre Jolivet               PetscCall(ISDestroy(ov));
22839bb5c669SPierre Jolivet               PetscCall(ISDestroy(ov + 1));
22849bb5c669SPierre Jolivet               if (overlap == 1 && subdomains && flg) {
22859bb5c669SPierre Jolivet                 *subA = A0;
22869bb5c669SPierre Jolivet                 sub   = subA;
22879bb5c669SPierre Jolivet                 if (uaux) PetscCall(MatDestroy(&uaux));
22889bb5c669SPierre Jolivet               } else PetscCall(MatDestroy(&A0));
22899bb5c669SPierre Jolivet               PetscCall(MatCreateShell(PETSC_COMM_SELF, P->rmap->n, n[1] - n[0], P->rmap->n, n[1] - n[0], h, &data->aux));
2290bdcd51b8SPierre Jolivet               PetscCall(KSPSetErrorIfNotConverged(h->ksp, PETSC_TRUE)); /* bail out as early as possible to avoid (apparently) unrelated error messages */
22919bb5c669SPierre Jolivet               PetscCall(MatCreateVecs(h->ksp->pc->pmat, &h->v, nullptr));
22929bb5c669SPierre Jolivet               PetscCall(MatShellSetOperation(data->aux, MATOP_MULT, (void (*)(void))MatMult_Harmonic));
22939bb5c669SPierre Jolivet               PetscCall(MatShellSetOperation(data->aux, MATOP_MULT_TRANSPOSE, (void (*)(void))MatMultTranspose_Harmonic));
22949bb5c669SPierre Jolivet               PetscCall(MatShellSetMatProductOperation(data->aux, MATPRODUCT_AB, nullptr, MatProduct_AB_Harmonic, nullptr, MATDENSE, MATDENSE));
22959bb5c669SPierre Jolivet               PetscCall(MatShellSetOperation(data->aux, MATOP_DESTROY, (void (*)(void))MatDestroy_Harmonic));
22969bb5c669SPierre Jolivet               PetscCall(MatDestroySubMatrices(1, &a));
22979bb5c669SPierre Jolivet             }
2298df21ef68SPierre Jolivet             if (overlap != 1 || !subdomains) {
2299df21ef68SPierre Jolivet               PetscCall(MatCreateSubMatrices(uaux ? uaux : P, 1, is, is, MAT_INITIAL_MATRIX, &sub));
2300df21ef68SPierre Jolivet               if (ismatis) {
2301df21ef68SPierre Jolivet                 PetscCall(MatISGetLocalMat(C, &N));
2302df21ef68SPierre Jolivet                 PetscCall(PetscObjectTypeCompare((PetscObject)N, MATSEQSBAIJ, &flg));
2303df21ef68SPierre Jolivet                 if (flg) PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub));
2304df21ef68SPierre Jolivet                 PetscCall(MatISRestoreLocalMat(C, &N));
2305df21ef68SPierre Jolivet               }
2306df21ef68SPierre Jolivet             }
23079bb5c669SPierre Jolivet             if (uaux) {
23089bb5c669SPierre Jolivet               PetscCall(MatDestroy(&uaux));
23099bb5c669SPierre Jolivet               PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub));
23109bb5c669SPierre Jolivet             }
23119bb5c669SPierre Jolivet           }
231213044ca3SPierre Jolivet         }
2313f1580f4eSBarry Smith       } else {
2314f1580f4eSBarry Smith         PetscCall(MatCreateSubMatrices(uaux, 1, is, is, MAT_INITIAL_MATRIX, &sub));
2315f1580f4eSBarry Smith         PetscCall(MatDestroy(&uaux));
2316f1580f4eSBarry Smith         PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub));
2317f1580f4eSBarry Smith       }
2318f1580f4eSBarry Smith       /* Vec holding the partition of unity */
2319f1580f4eSBarry Smith       if (!data->levels[0]->D) {
2320f1580f4eSBarry Smith         PetscCall(ISGetLocalSize(data->is, &n));
2321f1580f4eSBarry Smith         PetscCall(VecCreateMPI(PETSC_COMM_SELF, n, PETSC_DETERMINE, &data->levels[0]->D));
2322f1580f4eSBarry Smith       }
23239bb5c669SPierre Jolivet       if (data->share && overlap == -1) {
2324f1580f4eSBarry Smith         Mat      D;
2325db4a47b3SPierre Jolivet         IS       perm = nullptr;
2326f1580f4eSBarry Smith         PetscInt size = -1;
2327811e8887SPierre Jolivet 
2328398c7888SPierre Jolivet         if (!data->levels[0]->pc) {
2329398c7888SPierre Jolivet           PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : ""));
2330398c7888SPierre Jolivet           PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->pc));
2331398c7888SPierre Jolivet           PetscCall(PCSetOptionsPrefix(data->levels[0]->pc, prefix));
2332398c7888SPierre Jolivet           PetscCall(PCSetOperators(data->levels[0]->pc, A, P));
2333398c7888SPierre Jolivet         }
2334398c7888SPierre Jolivet         PetscCall(PCSetType(data->levels[0]->pc, PCASM));
233513044ca3SPierre Jolivet         if (!ctx) {
2336398c7888SPierre Jolivet           if (!data->levels[0]->pc->setupcalled) {
2337398c7888SPierre Jolivet             IS sorted; /* PCASM will sort the input IS, duplicate it to return an unmodified (PCHPDDM) input IS */
2338398c7888SPierre Jolivet             PetscCall(ISDuplicate(is[0], &sorted));
2339398c7888SPierre Jolivet             PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, &sorted, &loc));
2340398c7888SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)sorted));
2341398c7888SPierre Jolivet           }
2342398c7888SPierre Jolivet           PetscCall(PCSetFromOptions(data->levels[0]->pc));
2343398c7888SPierre Jolivet           if (block) {
2344398c7888SPierre Jolivet             PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, sub[0], &C, &perm));
2345398c7888SPierre Jolivet             PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(data->levels[0]->pc, C, algebraic));
2346398c7888SPierre Jolivet           } else PetscCall(PCSetUp(data->levels[0]->pc));
2347db4a47b3SPierre Jolivet           PetscTryMethod(data->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data->levels[0]->pc, &size, nullptr, &ksp));
2348398c7888SPierre Jolivet           if (size != 1) {
2349398c7888SPierre Jolivet             data->share = PETSC_FALSE;
2350398c7888SPierre Jolivet             PetscCheck(size == -1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", size);
2351398c7888SPierre Jolivet             PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc since PCASMGetSubKSP() not found in fine-level PC\n"));
2352398c7888SPierre Jolivet             PetscCall(ISDestroy(&unsorted));
2353398c7888SPierre Jolivet             unsorted = is[0];
2354398c7888SPierre Jolivet           } else {
2355811e8887SPierre Jolivet             const char *matpre;
2356811e8887SPierre Jolivet             PetscBool   cmp[4];
2357811e8887SPierre Jolivet 
235813044ca3SPierre Jolivet             if (!block && !ctx) PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, PetscBool3ToBool(data->Neumann) ? sub[0] : data->aux, &C, &perm));
2359c8ea6600SPierre Jolivet             if (!PetscBool3ToBool(data->Neumann) && !block) {
2360f1580f4eSBarry Smith               PetscCall(MatPermute(sub[0], perm, perm, &D)); /* permute since PCASM will call ISSort() */
2361f1580f4eSBarry Smith               PetscCall(MatHeaderReplace(sub[0], &D));
2362f1580f4eSBarry Smith             }
2363f1580f4eSBarry Smith             if (data->B) { /* see PCHPDDMSetRHSMat() */
2364f1580f4eSBarry Smith               PetscCall(MatPermute(data->B, perm, perm, &D));
2365f1580f4eSBarry Smith               PetscCall(MatHeaderReplace(data->B, &D));
2366f1580f4eSBarry Smith             }
2367f1580f4eSBarry Smith             PetscCall(ISDestroy(&perm));
2368f1580f4eSBarry Smith             PetscCall(KSPGetOperators(ksp[0], subA, subA + 1));
23690307214fSPierre Jolivet             PetscCall(PetscObjectReference((PetscObject)subA[0]));
2370f1580f4eSBarry Smith             PetscCall(MatDuplicate(subA[1], MAT_SHARE_NONZERO_PATTERN, &D));
2371f1580f4eSBarry Smith             PetscCall(MatGetOptionsPrefix(subA[1], &matpre));
2372f1580f4eSBarry Smith             PetscCall(MatSetOptionsPrefix(D, matpre));
2373f1580f4eSBarry Smith             PetscCall(PetscObjectTypeCompare((PetscObject)D, MATNORMAL, cmp));
2374f1580f4eSBarry Smith             PetscCall(PetscObjectTypeCompare((PetscObject)C, MATNORMAL, cmp + 1));
2375f1580f4eSBarry Smith             if (!cmp[0]) PetscCall(PetscObjectTypeCompare((PetscObject)D, MATNORMALHERMITIAN, cmp + 2));
2376f1580f4eSBarry Smith             else cmp[2] = PETSC_FALSE;
2377f1580f4eSBarry Smith             if (!cmp[1]) PetscCall(PetscObjectTypeCompare((PetscObject)C, MATNORMALHERMITIAN, cmp + 3));
2378f1580f4eSBarry Smith             else cmp[3] = PETSC_FALSE;
2379f1580f4eSBarry 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);
2380f1580f4eSBarry Smith             if (!cmp[0] && !cmp[2]) {
2381f1580f4eSBarry Smith               if (!block) PetscCall(MatAXPY(D, 1.0, C, SUBSET_NONZERO_PATTERN));
23825c7345deSPierre Jolivet               else {
2383db4a47b3SPierre Jolivet                 PetscCall(MatMissingDiagonal(D, cmp, nullptr));
23845c7345deSPierre Jolivet                 if (cmp[0]) structure = DIFFERENT_NONZERO_PATTERN; /* data->aux has no missing diagonal entry */
23855c7345deSPierre Jolivet                 PetscCall(MatAXPY(D, 1.0, data->aux, structure));
23865c7345deSPierre Jolivet               }
2387f1580f4eSBarry Smith             } else {
2388f1580f4eSBarry Smith               Mat mat[2];
2389811e8887SPierre Jolivet 
2390f1580f4eSBarry Smith               if (cmp[0]) {
2391f1580f4eSBarry Smith                 PetscCall(MatNormalGetMat(D, mat));
2392f1580f4eSBarry Smith                 PetscCall(MatNormalGetMat(C, mat + 1));
2393f1580f4eSBarry Smith               } else {
2394f1580f4eSBarry Smith                 PetscCall(MatNormalHermitianGetMat(D, mat));
2395f1580f4eSBarry Smith                 PetscCall(MatNormalHermitianGetMat(C, mat + 1));
2396f1580f4eSBarry Smith               }
2397f1580f4eSBarry Smith               PetscCall(MatAXPY(mat[0], 1.0, mat[1], SUBSET_NONZERO_PATTERN));
2398f1580f4eSBarry Smith             }
2399f1580f4eSBarry Smith             PetscCall(MatPropagateSymmetryOptions(C, D));
2400f1580f4eSBarry Smith             PetscCall(MatDestroy(&C));
2401f1580f4eSBarry Smith             C = D;
2402f1580f4eSBarry Smith             /* swap pointers so that variables stay consistent throughout PCSetUp() */
2403f1580f4eSBarry Smith             std::swap(C, data->aux);
2404f1580f4eSBarry Smith             std::swap(uis, data->is);
2405f1580f4eSBarry Smith             swap = PETSC_TRUE;
2406f1580f4eSBarry Smith           }
2407f1580f4eSBarry Smith         }
240813044ca3SPierre Jolivet       }
240913044ca3SPierre Jolivet       if (ctx) {
241013044ca3SPierre Jolivet         PC_HPDDM              *data_00 = (PC_HPDDM *)std::get<0>(*ctx)[0]->data;
241113044ca3SPierre Jolivet         PC                     s;
241200b491fbSPierre Jolivet         Mat                    A00, P00, A01 = nullptr, A10, A11, N, b[4];
241307d8d47fSPierre Jolivet         IS                     sorted, is[2], *is_00;
241413044ca3SPierre Jolivet         MatSolverType          type;
241513044ca3SPierre Jolivet         std::pair<PC, Vec[2]> *p;
241613044ca3SPierre Jolivet 
241707d8d47fSPierre Jolivet         n = -1;
241807d8d47fSPierre Jolivet         PetscTryMethod(data_00->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data_00->levels[0]->pc, &n, nullptr, &ksp));
241907d8d47fSPierre Jolivet         PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", n);
242007d8d47fSPierre Jolivet         PetscCall(KSPGetOperators(ksp[0], subA, subA + 1));
242107d8d47fSPierre Jolivet         PetscCall(ISGetLocalSize(data_00->is, &n));
242207d8d47fSPierre Jolivet         if (n != subA[0]->rmap->n || n != subA[0]->cmap->n) {
242307d8d47fSPierre Jolivet           PetscCall(PCASMGetLocalSubdomains(data_00->levels[0]->pc, &n, &is_00, nullptr));
242407d8d47fSPierre Jolivet           PetscCall(ISGetLocalSize(*is_00, &n));
242507d8d47fSPierre 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);
242607d8d47fSPierre Jolivet         } else is_00 = &data_00->is;
242713044ca3SPierre Jolivet         PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, data->aux, &C, nullptr)); /* permute since PCASM works with a sorted IS */
242813044ca3SPierre Jolivet         std::swap(C, data->aux);
242913044ca3SPierre Jolivet         std::swap(uis, data->is);
243013044ca3SPierre Jolivet         swap = PETSC_TRUE;
243113044ca3SPierre Jolivet         PetscCall(PCASMSetType(data->levels[0]->pc, PC_ASM_NONE)); /* "Neumann--Neumann" preconditioning with overlap and a Boolean partition of unity */
243213044ca3SPierre Jolivet         PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, &data->is, &loc));
243313044ca3SPierre Jolivet         PetscCall(PCSetFromOptions(data->levels[0]->pc)); /* action of eq. (15) of https://hal.science/hal-02343808v6/document (with a sign flip) */
243413044ca3SPierre Jolivet         PetscCall(MatSchurComplementGetSubMatrices(P, &A00, &P00, std::get<1>(*ctx), &A10, &A11));
243513044ca3SPierre Jolivet         std::get<1>(*ctx)[1] = A10;
243613044ca3SPierre Jolivet         PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, &flg));
243713044ca3SPierre Jolivet         if (flg) PetscCall(MatTransposeGetMat(A10, &A01));
243813044ca3SPierre Jolivet         else {
243913044ca3SPierre Jolivet           PetscBool flg;
244013044ca3SPierre Jolivet 
244113044ca3SPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, &flg));
244213044ca3SPierre Jolivet           if (flg) PetscCall(MatHermitianTransposeGetMat(A10, &A01));
244313044ca3SPierre Jolivet         }
244407d8d47fSPierre 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 */
244513044ca3SPierre Jolivet         PetscCall(ISSort(sorted));               /* this is to avoid changing users inputs, but it requires a new call to ISSort() here                                                                                               */
244600b491fbSPierre Jolivet         if (!A01) {
244700b491fbSPierre Jolivet           PetscCall(MatSetOption(A10, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
244800b491fbSPierre Jolivet           PetscCall(MatCreateSubMatrices(A10, 1, &data->is, &sorted, MAT_INITIAL_MATRIX, &sub));
244900b491fbSPierre Jolivet           b[2] = sub[0];
245000b491fbSPierre Jolivet           PetscCall(PetscObjectReference((PetscObject)sub[0]));
245100b491fbSPierre Jolivet           PetscCall(MatDestroySubMatrices(1, &sub));
245200b491fbSPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)std::get<1>(*ctx)[0], MATTRANSPOSEVIRTUAL, &flg));
245300b491fbSPierre Jolivet           A10 = nullptr;
245400b491fbSPierre Jolivet           if (flg) PetscCall(MatTransposeGetMat(std::get<1>(*ctx)[0], &A10));
245513044ca3SPierre Jolivet           else {
245600b491fbSPierre Jolivet             PetscBool flg;
245700b491fbSPierre Jolivet 
2458c71f06a7SPierre Jolivet             PetscCall(PetscObjectTypeCompare((PetscObject)std::get<1>(*ctx)[0], MATHERMITIANTRANSPOSEVIRTUAL, &flg));
245900b491fbSPierre Jolivet             if (flg) PetscCall(MatHermitianTransposeGetMat(std::get<1>(*ctx)[0], &A10));
246013044ca3SPierre Jolivet           }
2461811e8887SPierre Jolivet           if (!A10) PetscCall(MatCreateSubMatrices(std::get<1>(*ctx)[0], 1, &sorted, &data->is, MAT_INITIAL_MATRIX, &sub));
2462811e8887SPierre Jolivet           else {
2463811e8887SPierre Jolivet             if (flg) PetscCall(MatCreateTranspose(b[2], b + 1));
2464811e8887SPierre Jolivet             else PetscCall(MatCreateHermitianTranspose(b[2], b + 1));
246500b491fbSPierre Jolivet           }
246600b491fbSPierre Jolivet         } else {
246700b491fbSPierre Jolivet           PetscCall(MatSetOption(A01, MAT_SUBMAT_SINGLEIS, PETSC_TRUE));
246800b491fbSPierre Jolivet           PetscCall(MatCreateSubMatrices(A01, 1, &sorted, &data->is, MAT_INITIAL_MATRIX, &sub));
2469811e8887SPierre Jolivet           if (flg) PetscCall(MatCreateTranspose(*sub, b + 2));
2470811e8887SPierre Jolivet           else PetscCall(MatCreateHermitianTranspose(*sub, b + 2));
2471811e8887SPierre Jolivet         }
2472811e8887SPierre Jolivet         if (A01 || !A10) {
2473811e8887SPierre Jolivet           b[1] = sub[0];
2474811e8887SPierre Jolivet           PetscCall(PetscObjectReference((PetscObject)sub[0]));
247500b491fbSPierre Jolivet         }
247600b491fbSPierre Jolivet         PetscCall(MatDestroySubMatrices(1, &sub));
247713044ca3SPierre Jolivet         PetscCall(ISDestroy(&sorted));
247800b491fbSPierre Jolivet         PetscCall(MatCreateSchurComplement(subA[0], subA[1], b[1], b[2], data->aux, &S));
247913044ca3SPierre Jolivet         PetscCall(MatSchurComplementSetKSP(S, ksp[0]));
248013044ca3SPierre 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 */
248113044ca3SPierre Jolivet         PetscTryMethod(data->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data->levels[0]->pc, &n, nullptr, &ksp));
248213044ca3SPierre Jolivet         PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", n);
248313044ca3SPierre Jolivet         PetscCall(KSPGetPC(ksp[0], &inner));
248413044ca3SPierre Jolivet         PetscCall(PCSetType(inner, PCSHELL)); /* compute the action of the inverse of the local Schur complement with a PCSHELL */
248513044ca3SPierre Jolivet         b[0] = subA[0];
248613044ca3SPierre Jolivet         b[3] = data->aux;
248713044ca3SPierre 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 */
248800b491fbSPierre Jolivet         PetscCall(PetscObjectDereference((PetscObject)b[1]));
248900b491fbSPierre Jolivet         PetscCall(PetscObjectDereference((PetscObject)b[2]));
249013044ca3SPierre Jolivet         PetscCall(PCCreate(PETSC_COMM_SELF, &s));
249113044ca3SPierre Jolivet         PetscCall(PCSetOptionsPrefix(s, ((PetscObject)inner)->prefix));
249213044ca3SPierre Jolivet         PetscCall(PCSetOptionsPrefix(inner, nullptr));
249313044ca3SPierre Jolivet         PetscCall(KSPSetSkipPCSetFromOptions(ksp[0], PETSC_TRUE));
249413044ca3SPierre Jolivet         PetscCall(PCSetType(s, PCLU));
249513044ca3SPierre Jolivet         if (PetscDefined(HAVE_MUMPS)) { /* only MATSOLVERMUMPS handles MATNEST, so for the others, e.g., MATSOLVERPETSC or MATSOLVERMKL_PARDISO, convert to plain MATAIJ */
249613044ca3SPierre Jolivet           PetscCall(PCFactorSetMatSolverType(s, MATSOLVERMUMPS));
249713044ca3SPierre Jolivet         }
249813044ca3SPierre Jolivet         PetscCall(PCSetFromOptions(s));
249913044ca3SPierre Jolivet         PetscCall(PCFactorGetMatSolverType(s, &type));
250013044ca3SPierre Jolivet         PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg));
250113044ca3SPierre Jolivet         if (flg) {
250213044ca3SPierre Jolivet           PetscCall(PCSetOperators(s, N, N));
250300b491fbSPierre Jolivet           PetscCall(PCFactorGetMatrix(s, b));
250400b491fbSPierre Jolivet           PetscCall(MatSetOptionsPrefix(*b, ((PetscObject)s)->prefix));
250524ddd604SPierre Jolivet           n = -1;
250624ddd604SPierre Jolivet           PetscCall(PetscOptionsGetInt(nullptr, ((PetscObject)s)->prefix, "-mat_mumps_icntl_26", &n, nullptr));
2507f45b553cSPierre Jolivet           if (n == 1) {                                /* allocates a square MatDense of size is[1]->map->n, so one */
2508f45b553cSPierre Jolivet             PetscCall(MatNestGetISs(N, is, nullptr));  /*  needs to be able to deactivate this path when dealing    */
2509f45b553cSPierre Jolivet             PetscCall(MatFactorSetSchurIS(*b, is[1])); /*  with a large constraint space in order to avoid OOM      */
2510f45b553cSPierre Jolivet           }
251113044ca3SPierre Jolivet         } else {
251200b491fbSPierre Jolivet           PetscCall(MatConvert(N, MATAIJ, MAT_INITIAL_MATRIX, b));
251300b491fbSPierre Jolivet           PetscCall(PCSetOperators(s, N, *b));
251400b491fbSPierre Jolivet           PetscCall(PetscObjectDereference((PetscObject)*b));
251500b491fbSPierre 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 */
251613044ca3SPierre Jolivet         }
251713044ca3SPierre Jolivet         PetscCall(PetscNew(&p));
251813044ca3SPierre Jolivet         p->first = s;
251900b491fbSPierre Jolivet         PetscCall(MatCreateVecs(*b, p->second, p->second + 1));
252013044ca3SPierre Jolivet         PetscCall(PCShellSetContext(inner, p));
252113044ca3SPierre Jolivet         PetscCall(PCShellSetApply(inner, PCApply_Nest));
252213044ca3SPierre Jolivet         PetscCall(PCShellSetView(inner, PCView_Nest));
252313044ca3SPierre Jolivet         PetscCall(PCShellSetDestroy(inner, PCDestroy_Nest));
252413044ca3SPierre Jolivet         PetscCall(PetscObjectDereference((PetscObject)N));
252513044ca3SPierre Jolivet       }
2526f1580f4eSBarry Smith       if (!data->levels[0]->scatter) {
2527db4a47b3SPierre Jolivet         PetscCall(MatCreateVecs(P, &xin, nullptr));
2528f1580f4eSBarry Smith         if (ismatis) PetscCall(MatDestroy(&P));
2529db4a47b3SPierre Jolivet         PetscCall(VecScatterCreate(xin, data->is, data->levels[0]->D, nullptr, &data->levels[0]->scatter));
2530f1580f4eSBarry Smith         PetscCall(VecDestroy(&xin));
2531f1580f4eSBarry Smith       }
2532f1580f4eSBarry Smith       if (data->levels[0]->P) {
2533f1580f4eSBarry Smith         /* if the pattern is the same and PCSetUp() has previously succeeded, reuse HPDDM buffers and connectivity */
2534*371d2eb7SMartin Diehl         PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(data->levels[0], !pc->setupcalled || pc->flag == DIFFERENT_NONZERO_PATTERN ? PETSC_TRUE : PETSC_FALSE));
2535f1580f4eSBarry Smith       }
2536f1580f4eSBarry Smith       if (!data->levels[0]->P) data->levels[0]->P = new HPDDM::Schwarz<PetscScalar>();
2537db4a47b3SPierre Jolivet       if (data->log_separate) PetscCall(PetscLogEventBegin(PC_HPDDM_SetUp[0], data->levels[0]->ksp, nullptr, nullptr, nullptr));
2538db4a47b3SPierre Jolivet       else PetscCall(PetscLogEventBegin(PC_HPDDM_Strc, data->levels[0]->ksp, nullptr, nullptr, nullptr));
2539f1580f4eSBarry Smith       /* HPDDM internal data structure */
254013044ca3SPierre Jolivet       PetscCall(data->levels[0]->P->structure(loc, data->is, !ctx ? sub[0] : nullptr, ismatis ? C : data->aux, data->levels));
2541db4a47b3SPierre Jolivet       if (!data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Strc, data->levels[0]->ksp, nullptr, nullptr, nullptr));
2542f1580f4eSBarry Smith       /* matrix pencil of the generalized eigenvalue problem on the overlap (GenEO) */
254313044ca3SPierre Jolivet       if (!ctx) {
25449bb5c669SPierre Jolivet         if (data->deflation || overlap != -1) weighted = data->aux;
2545f1580f4eSBarry Smith         else if (!data->B) {
2546e2b46ddfSPierre Jolivet           PetscBool cmp;
2547811e8887SPierre Jolivet 
2548f1580f4eSBarry Smith           PetscCall(MatDuplicate(sub[0], MAT_COPY_VALUES, &weighted));
2549e2b46ddfSPierre Jolivet           PetscCall(PetscObjectTypeCompareAny((PetscObject)weighted, &cmp, MATNORMAL, MATNORMALHERMITIAN, ""));
2550e2b46ddfSPierre Jolivet           if (cmp) flg = PETSC_FALSE;
2551e2b46ddfSPierre Jolivet           PetscCall(MatDiagonalScale(weighted, data->levels[0]->D, data->levels[0]->D));
25525b0bb9f2SPierre Jolivet           /* neither MatDuplicate() nor MatDiagonalScale() handles the symmetry options, so propagate the options explicitly */
2553f1580f4eSBarry Smith           /* only useful for -mat_type baij -pc_hpddm_levels_1_st_pc_type cholesky (no problem with MATAIJ or MATSBAIJ)      */
2554f1580f4eSBarry Smith           PetscCall(MatPropagateSymmetryOptions(sub[0], weighted));
2555f1580f4eSBarry Smith         } else weighted = data->B;
255613044ca3SPierre Jolivet       } else weighted = nullptr;
2557f1580f4eSBarry Smith       /* SLEPc is used inside the loaded symbol */
25589bb5c669SPierre 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));
25599bb5c669SPierre Jolivet       if (!ctx && data->share && overlap == -1) {
2560f1580f4eSBarry Smith         Mat st[2];
2561f1580f4eSBarry Smith         PetscCall(KSPGetOperators(ksp[0], st, st + 1));
25625c7345deSPierre Jolivet         PetscCall(MatCopy(subA[0], st[0], structure));
2563f1580f4eSBarry Smith         if (subA[1] != subA[0] || st[1] != st[0]) PetscCall(MatCopy(subA[1], st[1], SAME_NONZERO_PATTERN));
25640307214fSPierre Jolivet         PetscCall(PetscObjectDereference((PetscObject)subA[0]));
2565f1580f4eSBarry Smith       }
2566db4a47b3SPierre Jolivet       if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_SetUp[0], data->levels[0]->ksp, nullptr, nullptr, nullptr));
2567f1580f4eSBarry Smith       if (ismatis) PetscCall(MatISGetLocalMat(C, &N));
2568f1580f4eSBarry Smith       else N = data->aux;
256913044ca3SPierre Jolivet       if (!ctx) P = sub[0];
257013044ca3SPierre Jolivet       else P = S;
2571f1580f4eSBarry Smith       /* going through the grid hierarchy */
2572f1580f4eSBarry Smith       for (n = 1; n < data->N; ++n) {
2573db4a47b3SPierre Jolivet         if (data->log_separate) PetscCall(PetscLogEventBegin(PC_HPDDM_SetUp[n], data->levels[n]->ksp, nullptr, nullptr, nullptr));
2574f1580f4eSBarry Smith         /* method composed in the loaded symbol since there, SLEPc is used as well */
2575f1580f4eSBarry 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));
2576db4a47b3SPierre Jolivet         if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_SetUp[n], data->levels[n]->ksp, nullptr, nullptr, nullptr));
2577f1580f4eSBarry Smith       }
2578f1580f4eSBarry Smith       /* reset to NULL to avoid any faulty use */
2579db4a47b3SPierre Jolivet       PetscCall(PetscObjectComposeFunction((PetscObject)data->levels[0]->ksp, "PCHPDDMSetUp_Private_C", nullptr));
2580db4a47b3SPierre Jolivet       if (!ismatis) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_C", nullptr));
2581f1580f4eSBarry Smith       else PetscCall(PetscObjectDereference((PetscObject)C)); /* matching PetscObjectReference() above */
2582f1580f4eSBarry Smith       for (n = 0; n < data->N - 1; ++n)
2583f1580f4eSBarry Smith         if (data->levels[n]->P) {
2584f1580f4eSBarry Smith           /* HPDDM internal work buffers */
2585f1580f4eSBarry Smith           data->levels[n]->P->setBuffer();
2586f1580f4eSBarry Smith           data->levels[n]->P->super::start();
2587f1580f4eSBarry Smith         }
25889bb5c669SPierre Jolivet       if (ismatis || !subdomains) PetscCall(PCHPDDMDestroySubMatrices_Private(PetscBool3ToBool(data->Neumann), PetscBool(algebraic && !block && overlap == -1), sub));
2589db4a47b3SPierre Jolivet       if (ismatis) data->is = nullptr;
2590f1580f4eSBarry Smith       for (n = 0; n < data->N - 1 + (reused > 0); ++n) {
2591f1580f4eSBarry Smith         if (data->levels[n]->P) {
2592f1580f4eSBarry Smith           PC spc;
2593f1580f4eSBarry Smith 
2594f1580f4eSBarry Smith           /* force the PC to be PCSHELL to do the coarse grid corrections */
2595f1580f4eSBarry Smith           PetscCall(KSPSetSkipPCSetFromOptions(data->levels[n]->ksp, PETSC_TRUE));
2596f1580f4eSBarry Smith           PetscCall(KSPGetPC(data->levels[n]->ksp, &spc));
2597f1580f4eSBarry Smith           PetscCall(PCSetType(spc, PCSHELL));
2598f1580f4eSBarry Smith           PetscCall(PCShellSetContext(spc, data->levels[n]));
2599f1580f4eSBarry Smith           PetscCall(PCShellSetSetUp(spc, PCSetUp_HPDDMShell));
2600f1580f4eSBarry Smith           PetscCall(PCShellSetApply(spc, PCApply_HPDDMShell));
2601f1580f4eSBarry Smith           PetscCall(PCShellSetMatApply(spc, PCMatApply_HPDDMShell));
260213044ca3SPierre Jolivet           if (ctx && n == 0) {
260313044ca3SPierre Jolivet             Mat                               Amat, Pmat;
260413044ca3SPierre Jolivet             PetscInt                          m, M;
260501e3c840SPierre Jolivet             std::tuple<Mat, PetscSF, Vec[2]> *ctx;
260613044ca3SPierre Jolivet 
260713044ca3SPierre Jolivet             PetscCall(KSPGetOperators(data->levels[n]->ksp, nullptr, &Pmat));
260813044ca3SPierre Jolivet             PetscCall(MatGetLocalSize(Pmat, &m, nullptr));
260913044ca3SPierre Jolivet             PetscCall(MatGetSize(Pmat, &M, nullptr));
261013044ca3SPierre Jolivet             PetscCall(PetscNew(&ctx));
261113044ca3SPierre Jolivet             std::get<0>(*ctx) = S;
261213044ca3SPierre Jolivet             std::get<1>(*ctx) = data->levels[n]->scatter;
261313044ca3SPierre Jolivet             PetscCall(MatCreateShell(PetscObjectComm((PetscObject)data->levels[n]->ksp), m, m, M, M, ctx, &Amat));
261413044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(Amat, MATOP_MULT, (void (*)(void))MatMult_Schur<false>));
261513044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(Amat, MATOP_MULT_TRANSPOSE, (void (*)(void))MatMult_Schur<true>));
261613044ca3SPierre Jolivet             PetscCall(MatShellSetOperation(Amat, MATOP_DESTROY, (void (*)(void))MatDestroy_Schur));
261713044ca3SPierre Jolivet             PetscCall(MatCreateVecs(S, std::get<2>(*ctx), std::get<2>(*ctx) + 1));
261813044ca3SPierre Jolivet             PetscCall(KSPSetOperators(data->levels[n]->ksp, Amat, Pmat));
261913044ca3SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)Amat));
262013044ca3SPierre Jolivet           }
2621f1580f4eSBarry Smith           PetscCall(PCShellSetDestroy(spc, PCDestroy_HPDDMShell));
2622f1580f4eSBarry Smith           if (!data->levels[n]->pc) PetscCall(PCCreate(PetscObjectComm((PetscObject)data->levels[n]->ksp), &data->levels[n]->pc));
2623f1580f4eSBarry Smith           if (n < reused) {
2624f1580f4eSBarry Smith             PetscCall(PCSetReusePreconditioner(spc, PETSC_TRUE));
2625f1580f4eSBarry Smith             PetscCall(PCSetReusePreconditioner(data->levels[n]->pc, PETSC_TRUE));
2626f1580f4eSBarry Smith           }
2627f1580f4eSBarry Smith           PetscCall(PCSetUp(spc));
2628f1580f4eSBarry Smith         }
2629f1580f4eSBarry Smith       }
263013044ca3SPierre Jolivet       if (ctx) PetscCall(MatDestroy(&S));
26319bb5c669SPierre Jolivet       if (overlap == -1) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", nullptr));
2632f1580f4eSBarry Smith     } else flg = reused ? PETSC_FALSE : PETSC_TRUE;
2633f1580f4eSBarry Smith     if (!ismatis && subdomains) {
2634f1580f4eSBarry Smith       if (flg) PetscCall(KSPGetPC(data->levels[0]->ksp, &inner));
2635f1580f4eSBarry Smith       else inner = data->levels[0]->pc;
2636f1580f4eSBarry Smith       if (inner) {
2637398c7888SPierre Jolivet         if (!inner->setupcalled) PetscCall(PCSetType(inner, PCASM));
2638398c7888SPierre Jolivet         PetscCall(PCSetFromOptions(inner));
2639398c7888SPierre Jolivet         PetscCall(PetscStrcmp(((PetscObject)inner)->type_name, PCASM, &flg));
2640398c7888SPierre Jolivet         if (flg) {
2641f1580f4eSBarry Smith           if (!inner->setupcalled) { /* evaluates to PETSC_FALSE when -pc_hpddm_block_splitting */
2642398c7888SPierre Jolivet             IS sorted;               /* PCASM will sort the input IS, duplicate it to return an unmodified (PCHPDDM) input IS */
2643811e8887SPierre Jolivet 
2644398c7888SPierre Jolivet             PetscCall(ISDuplicate(is[0], &sorted));
2645398c7888SPierre Jolivet             PetscCall(PCASMSetLocalSubdomains(inner, 1, &sorted, &loc));
2646398c7888SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)sorted));
2647398c7888SPierre Jolivet           }
2648c8ea6600SPierre Jolivet           if (!PetscBool3ToBool(data->Neumann) && data->N > 1) { /* subdomain matrices are already created for the eigenproblem, reuse them for the fine-level PC */
2649db4a47b3SPierre Jolivet             PetscCall(PCHPDDMPermute_Private(*is, nullptr, nullptr, sub[0], &P, nullptr));
2650398c7888SPierre Jolivet             PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(inner, P, algebraic));
2651398c7888SPierre Jolivet             PetscCall(PetscObjectDereference((PetscObject)P));
2652f1580f4eSBarry Smith           }
2653f1580f4eSBarry Smith         }
2654f1580f4eSBarry Smith       }
26559bb5c669SPierre Jolivet       if (data->N > 1) {
26569bb5c669SPierre Jolivet         if (overlap != 1) PetscCall(PCHPDDMDestroySubMatrices_Private(PetscBool3ToBool(data->Neumann), PetscBool(algebraic && !block && overlap == -1), sub));
26579bb5c669SPierre Jolivet         if (overlap == 1) PetscCall(MatDestroy(subA));
26589bb5c669SPierre Jolivet       }
2659f1580f4eSBarry Smith     }
2660f1580f4eSBarry Smith     PetscCall(ISDestroy(&loc));
2661f1580f4eSBarry Smith   } else data->N = 1 + reused; /* enforce this value to 1 + reused if there is no way to build another level */
2662f1580f4eSBarry Smith   if (requested != data->N + reused) {
2663f1580f4eSBarry 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,
2664f1580f4eSBarry Smith                         data->N, pcpre ? pcpre : ""));
2665f1580f4eSBarry 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));
2666f1580f4eSBarry Smith     /* cannot use PCDestroy_HPDDMShell() because PCSHELL not set for unassembled levels */
2667f1580f4eSBarry Smith     for (n = data->N - 1; n < requested - 1; ++n) {
2668f1580f4eSBarry Smith       if (data->levels[n]->P) {
2669f1580f4eSBarry Smith         PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(data->levels[n], PETSC_TRUE));
2670f1580f4eSBarry Smith         PetscCall(VecDestroyVecs(1, &data->levels[n]->v[0]));
2671f1580f4eSBarry Smith         PetscCall(VecDestroyVecs(2, &data->levels[n]->v[1]));
2672f1580f4eSBarry Smith         PetscCall(MatDestroy(data->levels[n]->V));
2673f1580f4eSBarry Smith         PetscCall(MatDestroy(data->levels[n]->V + 1));
2674f1580f4eSBarry Smith         PetscCall(MatDestroy(data->levels[n]->V + 2));
2675f1580f4eSBarry Smith         PetscCall(VecDestroy(&data->levels[n]->D));
267601e3c840SPierre Jolivet         PetscCall(PetscSFDestroy(&data->levels[n]->scatter));
2677f1580f4eSBarry Smith       }
2678f1580f4eSBarry Smith     }
2679f1580f4eSBarry Smith     if (reused) {
2680f1580f4eSBarry Smith       for (n = reused; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) {
2681f1580f4eSBarry Smith         PetscCall(KSPDestroy(&data->levels[n]->ksp));
2682f1580f4eSBarry Smith         PetscCall(PCDestroy(&data->levels[n]->pc));
2683f1580f4eSBarry Smith       }
2684f1580f4eSBarry Smith     }
2685f1580f4eSBarry 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,
2686f1580f4eSBarry Smith                data->N, reused, data->N, pcpre ? pcpre : "", pcpre ? pcpre : "", data->N);
2687f1580f4eSBarry Smith   }
2688f1580f4eSBarry Smith   /* these solvers are created after PCSetFromOptions() is called */
2689f1580f4eSBarry Smith   if (pc->setfromoptionscalled) {
2690f1580f4eSBarry Smith     for (n = 0; n < data->N; ++n) {
2691f1580f4eSBarry Smith       if (data->levels[n]->ksp) PetscCall(KSPSetFromOptions(data->levels[n]->ksp));
2692f1580f4eSBarry Smith       if (data->levels[n]->pc) PetscCall(PCSetFromOptions(data->levels[n]->pc));
2693f1580f4eSBarry Smith     }
2694f1580f4eSBarry Smith     pc->setfromoptionscalled = 0;
2695f1580f4eSBarry Smith   }
2696f1580f4eSBarry Smith   data->N += reused;
2697f1580f4eSBarry Smith   if (data->share && swap) {
2698f1580f4eSBarry Smith     /* swap back pointers so that variables follow the user-provided numbering */
2699f1580f4eSBarry Smith     std::swap(C, data->aux);
2700f1580f4eSBarry Smith     std::swap(uis, data->is);
2701f1580f4eSBarry Smith     PetscCall(MatDestroy(&C));
2702f1580f4eSBarry Smith     PetscCall(ISDestroy(&uis));
2703f1580f4eSBarry Smith   }
2704398c7888SPierre Jolivet   if (algebraic) PetscCall(MatDestroy(&data->aux));
270538fb34a1SPierre Jolivet   if (unsorted && unsorted != is[0]) {
2706398c7888SPierre Jolivet     PetscCall(ISCopy(unsorted, data->is));
2707398c7888SPierre Jolivet     PetscCall(ISDestroy(&unsorted));
2708398c7888SPierre Jolivet   }
2709398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG)
2710398c7888SPierre 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);
2711398c7888SPierre Jolivet   if (data->is) {
2712398c7888SPierre Jolivet     PetscCall(ISEqualUnsorted(data->is, dis, &flg));
2713398c7888SPierre Jolivet     PetscCall(ISDestroy(&dis));
2714398c7888SPierre Jolivet     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Input IS and output IS are not equal");
2715398c7888SPierre Jolivet   }
2716398c7888SPierre 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);
2717398c7888SPierre Jolivet   if (data->aux) {
2718398c7888SPierre Jolivet     PetscCall(MatMultEqual(data->aux, daux, 20, &flg));
2719398c7888SPierre Jolivet     PetscCall(MatDestroy(&daux));
2720398c7888SPierre Jolivet     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Input Mat and output Mat are not equal");
2721398c7888SPierre Jolivet   }
2722398c7888SPierre Jolivet #endif
27233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2724f1580f4eSBarry Smith }
2725f1580f4eSBarry Smith 
2726f1580f4eSBarry Smith /*@
2727f1580f4eSBarry Smith   PCHPDDMSetCoarseCorrectionType - Sets the coarse correction type.
2728f1580f4eSBarry Smith 
2729c3339decSBarry Smith   Collective
2730f1580f4eSBarry Smith 
2731f1580f4eSBarry Smith   Input Parameters:
2732f1580f4eSBarry Smith + pc   - preconditioner context
2733aa1539e9SPierre Jolivet - type - `PC_HPDDM_COARSE_CORRECTION_DEFLATED`, `PC_HPDDM_COARSE_CORRECTION_ADDITIVE`, `PC_HPDDM_COARSE_CORRECTION_BALANCED`, or `PC_HPDDM_COARSE_CORRECTION_NONE`
2734f1580f4eSBarry Smith 
2735f1580f4eSBarry Smith   Options Database Key:
2736aa1539e9SPierre Jolivet . -pc_hpddm_coarse_correction <deflated, additive, balanced, none> - type of coarse correction to apply
2737f1580f4eSBarry Smith 
2738f1580f4eSBarry Smith   Level: intermediate
2739f1580f4eSBarry Smith 
2740562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDMGetCoarseCorrectionType()`, `PCHPDDM`, `PCHPDDMCoarseCorrectionType`
2741f1580f4eSBarry Smith @*/
2742d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetCoarseCorrectionType(PC pc, PCHPDDMCoarseCorrectionType type)
2743d71ae5a4SJacob Faibussowitsch {
2744f1580f4eSBarry Smith   PetscFunctionBegin;
2745f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2746f1580f4eSBarry Smith   PetscValidLogicalCollectiveEnum(pc, type, 2);
2747f1580f4eSBarry Smith   PetscTryMethod(pc, "PCHPDDMSetCoarseCorrectionType_C", (PC, PCHPDDMCoarseCorrectionType), (pc, type));
27483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2749f1580f4eSBarry Smith }
2750f1580f4eSBarry Smith 
2751f1580f4eSBarry Smith /*@
2752f1580f4eSBarry Smith   PCHPDDMGetCoarseCorrectionType - Gets the coarse correction type.
2753f1580f4eSBarry Smith 
2754f1580f4eSBarry Smith   Input Parameter:
2755f1580f4eSBarry Smith . pc - preconditioner context
2756f1580f4eSBarry Smith 
2757f1580f4eSBarry Smith   Output Parameter:
2758aa1539e9SPierre Jolivet . type - `PC_HPDDM_COARSE_CORRECTION_DEFLATED`, `PC_HPDDM_COARSE_CORRECTION_ADDITIVE`, `PC_HPDDM_COARSE_CORRECTION_BALANCED`, or `PC_HPDDM_COARSE_CORRECTION_NONE`
2759f1580f4eSBarry Smith 
2760f1580f4eSBarry Smith   Level: intermediate
2761f1580f4eSBarry Smith 
2762562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDMSetCoarseCorrectionType()`, `PCHPDDM`, `PCHPDDMCoarseCorrectionType`
2763f1580f4eSBarry Smith @*/
2764d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMGetCoarseCorrectionType(PC pc, PCHPDDMCoarseCorrectionType *type)
2765d71ae5a4SJacob Faibussowitsch {
2766f1580f4eSBarry Smith   PetscFunctionBegin;
2767f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2768f1580f4eSBarry Smith   if (type) {
27694f572ea9SToby Isaac     PetscAssertPointer(type, 2);
2770f1580f4eSBarry Smith     PetscUseMethod(pc, "PCHPDDMGetCoarseCorrectionType_C", (PC, PCHPDDMCoarseCorrectionType *), (pc, type));
2771f1580f4eSBarry Smith   }
27723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2773f1580f4eSBarry Smith }
2774f1580f4eSBarry Smith 
2775d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetCoarseCorrectionType_HPDDM(PC pc, PCHPDDMCoarseCorrectionType type)
2776d71ae5a4SJacob Faibussowitsch {
2777f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
2778f1580f4eSBarry Smith 
2779f1580f4eSBarry Smith   PetscFunctionBegin;
2780f1580f4eSBarry Smith   data->correction = type;
27813ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2782f1580f4eSBarry Smith }
2783f1580f4eSBarry Smith 
2784d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMGetCoarseCorrectionType_HPDDM(PC pc, PCHPDDMCoarseCorrectionType *type)
2785d71ae5a4SJacob Faibussowitsch {
2786f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
2787f1580f4eSBarry Smith 
2788f1580f4eSBarry Smith   PetscFunctionBegin;
2789f1580f4eSBarry Smith   *type = data->correction;
27903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2791f1580f4eSBarry Smith }
2792f1580f4eSBarry Smith 
2793f1580f4eSBarry Smith /*@
2794e31fc274SPierre Jolivet   PCHPDDMSetSTShareSubKSP - Sets whether the `KSP` in SLEPc `ST` and the fine-level subdomain solver should be shared.
2795e31fc274SPierre Jolivet 
2796e31fc274SPierre Jolivet   Input Parameters:
2797e31fc274SPierre Jolivet + pc    - preconditioner context
2798e31fc274SPierre Jolivet - share - whether the `KSP` should be shared or not
2799e31fc274SPierre Jolivet 
2800e31fc274SPierre Jolivet   Note:
2801e31fc274SPierre Jolivet   This is not the same as `PCSetReusePreconditioner()`. Given certain conditions (visible using -info), a symbolic factorization can be skipped
2802e31fc274SPierre Jolivet   when using a subdomain `PCType` such as `PCLU` or `PCCHOLESKY`.
2803e31fc274SPierre Jolivet 
2804e31fc274SPierre Jolivet   Level: advanced
2805e31fc274SPierre Jolivet 
2806562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMGetSTShareSubKSP()`
2807e31fc274SPierre Jolivet @*/
2808e31fc274SPierre Jolivet PetscErrorCode PCHPDDMSetSTShareSubKSP(PC pc, PetscBool share)
2809e31fc274SPierre Jolivet {
2810e31fc274SPierre Jolivet   PetscFunctionBegin;
2811e31fc274SPierre Jolivet   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2812e31fc274SPierre Jolivet   PetscTryMethod(pc, "PCHPDDMSetSTShareSubKSP_C", (PC, PetscBool), (pc, share));
28133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2814e31fc274SPierre Jolivet }
2815e31fc274SPierre Jolivet 
2816e31fc274SPierre Jolivet /*@
2817f1580f4eSBarry Smith   PCHPDDMGetSTShareSubKSP - Gets whether the `KSP` in SLEPc `ST` and the fine-level subdomain solver is shared.
2818f1580f4eSBarry Smith 
2819f1580f4eSBarry Smith   Input Parameter:
2820f1580f4eSBarry Smith . pc - preconditioner context
2821f1580f4eSBarry Smith 
2822f1580f4eSBarry Smith   Output Parameter:
2823f1580f4eSBarry Smith . share - whether the `KSP` is shared or not
2824f1580f4eSBarry Smith 
2825f1580f4eSBarry Smith   Note:
2826f1580f4eSBarry 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
2827f1580f4eSBarry Smith   when using a subdomain `PCType` such as `PCLU` or `PCCHOLESKY`.
2828f1580f4eSBarry Smith 
2829f1580f4eSBarry Smith   Level: advanced
2830f1580f4eSBarry Smith 
2831562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMSetSTShareSubKSP()`
2832f1580f4eSBarry Smith @*/
2833d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMGetSTShareSubKSP(PC pc, PetscBool *share)
2834d71ae5a4SJacob Faibussowitsch {
2835f1580f4eSBarry Smith   PetscFunctionBegin;
2836f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2837f1580f4eSBarry Smith   if (share) {
28384f572ea9SToby Isaac     PetscAssertPointer(share, 2);
2839f1580f4eSBarry Smith     PetscUseMethod(pc, "PCHPDDMGetSTShareSubKSP_C", (PC, PetscBool *), (pc, share));
2840f1580f4eSBarry Smith   }
28413ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2842f1580f4eSBarry Smith }
2843f1580f4eSBarry Smith 
2844e31fc274SPierre Jolivet static PetscErrorCode PCHPDDMSetSTShareSubKSP_HPDDM(PC pc, PetscBool share)
2845e31fc274SPierre Jolivet {
2846e31fc274SPierre Jolivet   PC_HPDDM *data = (PC_HPDDM *)pc->data;
2847e31fc274SPierre Jolivet 
2848e31fc274SPierre Jolivet   PetscFunctionBegin;
2849e31fc274SPierre Jolivet   data->share = share;
28503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2851e31fc274SPierre Jolivet }
2852e31fc274SPierre Jolivet 
2853d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMGetSTShareSubKSP_HPDDM(PC pc, PetscBool *share)
2854d71ae5a4SJacob Faibussowitsch {
2855f1580f4eSBarry Smith   PC_HPDDM *data = (PC_HPDDM *)pc->data;
2856f1580f4eSBarry Smith 
2857f1580f4eSBarry Smith   PetscFunctionBegin;
2858f1580f4eSBarry Smith   *share = data->share;
28593ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2860f1580f4eSBarry Smith }
2861f1580f4eSBarry Smith 
2862f1580f4eSBarry Smith /*@
2863f1580f4eSBarry Smith   PCHPDDMSetDeflationMat - Sets the deflation space used to assemble a coarser operator.
2864f1580f4eSBarry Smith 
2865f1580f4eSBarry Smith   Input Parameters:
2866f1580f4eSBarry Smith + pc - preconditioner context
2867f1580f4eSBarry Smith . is - index set of the local deflation matrix
2868f1580f4eSBarry Smith - U  - deflation sequential matrix stored as a `MATSEQDENSE`
2869f1580f4eSBarry Smith 
2870f1580f4eSBarry Smith   Level: advanced
2871f1580f4eSBarry Smith 
2872562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCDeflationSetSpace()`, `PCMGSetRestriction()`
2873f1580f4eSBarry Smith @*/
2874d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetDeflationMat(PC pc, IS is, Mat U)
2875d71ae5a4SJacob Faibussowitsch {
2876f1580f4eSBarry Smith   PetscFunctionBegin;
2877f1580f4eSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
2878f1580f4eSBarry Smith   PetscValidHeaderSpecific(is, IS_CLASSID, 2);
2879f1580f4eSBarry Smith   PetscValidHeaderSpecific(U, MAT_CLASSID, 3);
2880e31fc274SPierre Jolivet   PetscTryMethod(pc, "PCHPDDMSetDeflationMat_C", (PC, IS, Mat), (pc, is, U));
28813ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2882f1580f4eSBarry Smith }
2883f1580f4eSBarry Smith 
2884d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetDeflationMat_HPDDM(PC pc, IS is, Mat U)
2885d71ae5a4SJacob Faibussowitsch {
2886f1580f4eSBarry Smith   PetscFunctionBegin;
2887f1580f4eSBarry Smith   PetscCall(PCHPDDMSetAuxiliaryMat_Private(pc, is, U, PETSC_TRUE));
28883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2889f1580f4eSBarry Smith }
2890f1580f4eSBarry Smith 
2891d71ae5a4SJacob Faibussowitsch PetscErrorCode HPDDMLoadDL_Private(PetscBool *found)
2892d71ae5a4SJacob Faibussowitsch {
2893605ad303SPierre Jolivet   PetscBool flg;
2894f1580f4eSBarry Smith   char      lib[PETSC_MAX_PATH_LEN], dlib[PETSC_MAX_PATH_LEN], dir[PETSC_MAX_PATH_LEN];
2895f1580f4eSBarry Smith 
2896f1580f4eSBarry Smith   PetscFunctionBegin;
28974f572ea9SToby Isaac   PetscAssertPointer(found, 1);
2898c6a7a370SJeremy L Thompson   PetscCall(PetscStrncpy(dir, "${PETSC_LIB_DIR}", sizeof(dir)));
2899db4a47b3SPierre Jolivet   PetscCall(PetscOptionsGetString(nullptr, nullptr, "-hpddm_dir", dir, sizeof(dir), nullptr));
2900f1580f4eSBarry Smith   PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/libhpddm_petsc", dir));
2901f1580f4eSBarry Smith   PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found));
2902605ad303SPierre Jolivet #if defined(SLEPC_LIB_DIR) /* this variable is passed during SLEPc ./configure when PETSc has not been configured   */
2903605ad303SPierre Jolivet   if (!*found) {           /* with --download-hpddm since slepcconf.h is not yet built (and thus can't be included) */
2904c6a7a370SJeremy L Thompson     PetscCall(PetscStrncpy(dir, HPDDM_STR(SLEPC_LIB_DIR), sizeof(dir)));
2905f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/libhpddm_petsc", dir));
2906f1580f4eSBarry Smith     PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found));
2907f1580f4eSBarry Smith   }
2908f1580f4eSBarry Smith #endif
2909605ad303SPierre 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 */
2910605ad303SPierre Jolivet     PetscCall(PetscOptionsGetenv(PETSC_COMM_SELF, "SLEPC_DIR", dir, sizeof(dir), &flg));
2911605ad303SPierre 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 */
2912605ad303SPierre Jolivet       PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/lib/libslepc", dir));
2913605ad303SPierre Jolivet       PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found));
2914605ad303SPierre Jolivet       PetscCheck(*found, PETSC_COMM_SELF, PETSC_ERR_PLIB, "%s not found but SLEPC_DIR=%s", lib, dir);
2915605ad303SPierre Jolivet       PetscCall(PetscDLLibraryAppend(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, dlib));
2916605ad303SPierre Jolivet       PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/lib/libhpddm_petsc", dir)); /* libhpddm_petsc is always in the same directory as libslepc */
2917605ad303SPierre Jolivet       PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found));
2918605ad303SPierre Jolivet     }
2919605ad303SPierre Jolivet   }
2920f1580f4eSBarry Smith   PetscCheck(*found, PETSC_COMM_SELF, PETSC_ERR_PLIB, "%s not found", lib);
2921f1580f4eSBarry Smith   PetscCall(PetscDLLibraryAppend(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, dlib));
29223ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2923f1580f4eSBarry Smith }
2924f1580f4eSBarry Smith 
2925f1580f4eSBarry Smith /*MC
2926f1580f4eSBarry Smith    PCHPDDM - Interface with the HPDDM library.
2927f1580f4eSBarry Smith 
29281d27aa22SBarry Smith    This `PC` may be used to build multilevel spectral domain decomposition methods based on the GenEO framework {cite}`spillane2011robust` {cite}`al2021multilevel`.
29291d27aa22SBarry Smith    It may be viewed as an alternative to spectral
29301d27aa22SBarry Smith    AMGe or `PCBDDC` with adaptive selection of constraints. The interface is explained in details in {cite}`jolivetromanzampini2020`
2931f1580f4eSBarry Smith 
2932e7593814SPierre 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`).
2933f1580f4eSBarry Smith 
2934f1580f4eSBarry Smith    For multilevel preconditioning, when using an assembled or hierarchical Pmat, one must provide an auxiliary local `Mat` (unassembled local operator for GenEO) using
2935f1580f4eSBarry Smith    `PCHPDDMSetAuxiliaryMat()`. Calling this routine is not needed when using a `MATIS` Pmat, assembly is done internally using `MatConvert()`.
2936f1580f4eSBarry Smith 
2937f1580f4eSBarry Smith    Options Database Keys:
2938f1580f4eSBarry Smith +   -pc_hpddm_define_subdomains <true, default=false>    - on the finest level, calls `PCASMSetLocalSubdomains()` with the `IS` supplied in `PCHPDDMSetAuxiliaryMat()`
2939f1580f4eSBarry Smith                                                          (not relevant with an unassembled Pmat)
2940f1580f4eSBarry Smith .   -pc_hpddm_has_neumann <true, default=false>          - on the finest level, informs the `PC` that the local Neumann matrix is supplied in `PCHPDDMSetAuxiliaryMat()`
2941f1580f4eSBarry Smith -   -pc_hpddm_coarse_correction <type, default=deflated> - determines the `PCHPDDMCoarseCorrectionType` when calling `PCApply()`
2942f1580f4eSBarry Smith 
294338bf2a8cSPierre Jolivet    Options for subdomain solvers, subdomain eigensolvers (for computing deflation vectors), and the coarse solver can be set using the following options database prefixes.
2944f1580f4eSBarry Smith .vb
2945f1580f4eSBarry Smith       -pc_hpddm_levels_%d_pc_
2946f1580f4eSBarry Smith       -pc_hpddm_levels_%d_ksp_
2947f1580f4eSBarry Smith       -pc_hpddm_levels_%d_eps_
2948f1580f4eSBarry Smith       -pc_hpddm_levels_%d_p
29494ec2a359SPierre Jolivet       -pc_hpddm_levels_%d_mat_type
2950f1580f4eSBarry Smith       -pc_hpddm_coarse_
2951f1580f4eSBarry Smith       -pc_hpddm_coarse_p
29524ec2a359SPierre Jolivet       -pc_hpddm_coarse_mat_type
29532ce66baaSPierre Jolivet       -pc_hpddm_coarse_mat_filter
2954f1580f4eSBarry Smith .ve
2955f1580f4eSBarry Smith 
295638bf2a8cSPierre 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
2957f1580f4eSBarry Smith     -pc_hpddm_coarse_p 2 -pc_hpddm_coarse_mat_type baij will use 10 deflation vectors per subdomain on the fine "level 1",
2958f1580f4eSBarry Smith     aggregate the fine subdomains into 4 "level 2" subdomains, then use 10 deflation vectors per subdomain on "level 2",
29597eb095acSPierre Jolivet     and assemble the coarse matrix (of dimension 4 x 10 = 40) on two processes as a `MATBAIJ` (default is `MATSBAIJ`).
2960f1580f4eSBarry Smith 
2961f1580f4eSBarry 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.
2962f1580f4eSBarry Smith 
29631d27aa22SBarry Smith    Level: intermediate
29641d27aa22SBarry Smith 
29651d27aa22SBarry Smith    Notes:
29661d27aa22SBarry Smith    This preconditioner requires that PETSc is built with SLEPc (``--download-slepc``).
29671d27aa22SBarry Smith 
29681d27aa22SBarry Smith    By default, the underlying concurrent eigenproblems
29691d27aa22SBarry Smith    are solved using SLEPc shift-and-invert spectral transformation. This is usually what gives the best performance for GenEO, cf.
29701d27aa22SBarry Smith    {cite}`spillane2011robust` {cite}`jolivet2013scalabledd`. As
297138bf2a8cSPierre 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
297238bf2a8cSPierre 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
297338bf2a8cSPierre Jolivet    SLEPc documentation since they are specific to `PCHPDDM`.
297438bf2a8cSPierre Jolivet .vb
297538bf2a8cSPierre Jolivet       -pc_hpddm_levels_1_st_share_sub_ksp
297638bf2a8cSPierre Jolivet       -pc_hpddm_levels_%d_eps_threshold
297738bf2a8cSPierre Jolivet       -pc_hpddm_levels_1_eps_use_inertia
297838bf2a8cSPierre Jolivet .ve
297938bf2a8cSPierre Jolivet 
298038bf2a8cSPierre Jolivet    The first option from the list only applies to the fine-level eigensolver, see `PCHPDDMSetSTShareSubKSP()`. The second option from the list is
298138bf2a8cSPierre Jolivet    used to filter eigenmodes retrieved after convergence of `EPSSolve()` at "level N" such that eigenvectors used to define a "level N+1" coarse
298238bf2a8cSPierre 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
298338bf2a8cSPierre 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
298438bf2a8cSPierre 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
298538bf2a8cSPierre Jolivet    to supply -pc_hpddm_levels_1_eps_nev. This last option also only applies to the fine-level (N = 1) eigensolver.
2986f1580f4eSBarry Smith 
29871d27aa22SBarry Smith    See also {cite}`dolean2015introduction`, {cite}`al2022robust`, {cite}`al2022robustpd`, and {cite}`nataf2022recent`
2988f1580f4eSBarry Smith 
2989562efe2eSBarry Smith .seealso: [](ch_ksp), `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHPDDMSetAuxiliaryMat()`, `MATIS`, `PCBDDC`, `PCDEFLATION`, `PCTELESCOPE`, `PCASM`,
2990e31fc274SPierre Jolivet           `PCHPDDMSetCoarseCorrectionType()`, `PCHPDDMHasNeumannMat()`, `PCHPDDMSetRHSMat()`, `PCHPDDMSetDeflationMat()`, `PCHPDDMSetSTShareSubKSP()`,
2991e31fc274SPierre Jolivet           `PCHPDDMGetSTShareSubKSP()`, `PCHPDDMGetCoarseCorrectionType()`, `PCHPDDMGetComplexities()`
2992f1580f4eSBarry Smith M*/
2993d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode PCCreate_HPDDM(PC pc)
2994d71ae5a4SJacob Faibussowitsch {
2995f1580f4eSBarry Smith   PC_HPDDM *data;
2996f1580f4eSBarry Smith   PetscBool found;
2997f1580f4eSBarry Smith 
2998f1580f4eSBarry Smith   PetscFunctionBegin;
2999f1580f4eSBarry Smith   if (!loadedSym) {
3000f1580f4eSBarry Smith     PetscCall(HPDDMLoadDL_Private(&found));
3001db4a47b3SPierre Jolivet     if (found) PetscCall(PetscDLLibrarySym(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, nullptr, "PCHPDDM_Internal", (void **)&loadedSym));
3002f1580f4eSBarry Smith   }
3003f1580f4eSBarry Smith   PetscCheck(loadedSym, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCHPDDM_Internal symbol not found in loaded libhpddm_petsc");
30044dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&data));
3005f1580f4eSBarry Smith   pc->data                = data;
3006c8ea6600SPierre Jolivet   data->Neumann           = PETSC_BOOL3_UNKNOWN;
3007f1580f4eSBarry Smith   pc->ops->reset          = PCReset_HPDDM;
3008f1580f4eSBarry Smith   pc->ops->destroy        = PCDestroy_HPDDM;
3009f1580f4eSBarry Smith   pc->ops->setfromoptions = PCSetFromOptions_HPDDM;
3010f1580f4eSBarry Smith   pc->ops->setup          = PCSetUp_HPDDM;
3011f1580f4eSBarry Smith   pc->ops->apply          = PCApply_HPDDM;
3012f1580f4eSBarry Smith   pc->ops->matapply       = PCMatApply_HPDDM;
3013f1580f4eSBarry Smith   pc->ops->view           = PCView_HPDDM;
3014f1580f4eSBarry Smith   pc->ops->presolve       = PCPreSolve_HPDDM;
3015f1580f4eSBarry Smith 
3016f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetAuxiliaryMat_C", PCHPDDMSetAuxiliaryMat_HPDDM));
3017f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMHasNeumannMat_C", PCHPDDMHasNeumannMat_HPDDM));
3018f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetRHSMat_C", PCHPDDMSetRHSMat_HPDDM));
3019f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetCoarseCorrectionType_C", PCHPDDMSetCoarseCorrectionType_HPDDM));
3020f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetCoarseCorrectionType_C", PCHPDDMGetCoarseCorrectionType_HPDDM));
3021e31fc274SPierre Jolivet   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetSTShareSubKSP_C", PCHPDDMSetSTShareSubKSP_HPDDM));
3022f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetSTShareSubKSP_C", PCHPDDMGetSTShareSubKSP_HPDDM));
3023f1580f4eSBarry Smith   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetDeflationMat_C", PCHPDDMSetDeflationMat_HPDDM));
30243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3025f1580f4eSBarry Smith }
3026f1580f4eSBarry Smith 
3027f1580f4eSBarry Smith /*@C
3028f1580f4eSBarry Smith   PCHPDDMInitializePackage - This function initializes everything in the `PCHPDDM` package. It is called from `PCInitializePackage()`.
3029f1580f4eSBarry Smith 
3030f1580f4eSBarry Smith   Level: developer
3031f1580f4eSBarry Smith 
3032562efe2eSBarry Smith .seealso: [](ch_ksp), `PetscInitialize()`
3033f1580f4eSBarry Smith @*/
3034d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMInitializePackage(void)
3035d71ae5a4SJacob Faibussowitsch {
3036f1580f4eSBarry Smith   char ename[32];
3037f1580f4eSBarry Smith 
3038f1580f4eSBarry Smith   PetscFunctionBegin;
30393ba16761SJacob Faibussowitsch   if (PCHPDDMPackageInitialized) PetscFunctionReturn(PETSC_SUCCESS);
3040f1580f4eSBarry Smith   PCHPDDMPackageInitialized = PETSC_TRUE;
3041f1580f4eSBarry Smith   PetscCall(PetscRegisterFinalize(PCHPDDMFinalizePackage));
3042f1580f4eSBarry Smith   /* general events registered once during package initialization */
3043f1580f4eSBarry Smith   /* some of these events are not triggered in libpetsc,          */
3044f1580f4eSBarry Smith   /* but rather directly in libhpddm_petsc,                       */
3045f1580f4eSBarry Smith   /* which is in charge of performing the following operations    */
3046f1580f4eSBarry Smith 
3047f1580f4eSBarry Smith   /* domain decomposition structure from Pmat sparsity pattern    */
3048f1580f4eSBarry Smith   PetscCall(PetscLogEventRegister("PCHPDDMStrc", PC_CLASSID, &PC_HPDDM_Strc));
3049f1580f4eSBarry Smith   /* Galerkin product, redistribution, and setup (not triggered in libpetsc)                */
3050f1580f4eSBarry Smith   PetscCall(PetscLogEventRegister("PCHPDDMPtAP", PC_CLASSID, &PC_HPDDM_PtAP));
3051f1580f4eSBarry Smith   /* Galerkin product with summation, redistribution, and setup (not triggered in libpetsc) */
3052f1580f4eSBarry Smith   PetscCall(PetscLogEventRegister("PCHPDDMPtBP", PC_CLASSID, &PC_HPDDM_PtBP));
3053f1580f4eSBarry Smith   /* next level construction using PtAP and PtBP (not triggered in libpetsc)                */
3054f1580f4eSBarry Smith   PetscCall(PetscLogEventRegister("PCHPDDMNext", PC_CLASSID, &PC_HPDDM_Next));
3055f1580f4eSBarry Smith   static_assert(PETSC_PCHPDDM_MAXLEVELS <= 9, "PETSC_PCHPDDM_MAXLEVELS value is too high");
3056811e8887SPierre Jolivet   for (PetscInt i = 1; i < PETSC_PCHPDDM_MAXLEVELS; ++i) {
3057f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(ename, sizeof(ename), "PCHPDDMSetUp L%1" PetscInt_FMT, i));
3058f1580f4eSBarry Smith     /* events during a PCSetUp() at level #i _except_ the assembly */
3059f1580f4eSBarry Smith     /* of the Galerkin operator of the coarser level #(i + 1)      */
3060f1580f4eSBarry Smith     PetscCall(PetscLogEventRegister(ename, PC_CLASSID, &PC_HPDDM_SetUp[i - 1]));
3061f1580f4eSBarry Smith     PetscCall(PetscSNPrintf(ename, sizeof(ename), "PCHPDDMSolve L%1" PetscInt_FMT, i));
3062f1580f4eSBarry Smith     /* events during a PCApply() at level #i _except_              */
3063f1580f4eSBarry Smith     /* the KSPSolve() of the coarser level #(i + 1)                */
3064f1580f4eSBarry Smith     PetscCall(PetscLogEventRegister(ename, PC_CLASSID, &PC_HPDDM_Solve[i - 1]));
3065f1580f4eSBarry Smith   }
30663ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3067f1580f4eSBarry Smith }
3068f1580f4eSBarry Smith 
3069f1580f4eSBarry Smith /*@C
3070f1580f4eSBarry Smith   PCHPDDMFinalizePackage - This function frees everything from the `PCHPDDM` package. It is called from `PetscFinalize()`.
3071f1580f4eSBarry Smith 
3072f1580f4eSBarry Smith   Level: developer
3073f1580f4eSBarry Smith 
3074562efe2eSBarry Smith .seealso: [](ch_ksp), `PetscFinalize()`
3075f1580f4eSBarry Smith @*/
3076d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMFinalizePackage(void)
3077d71ae5a4SJacob Faibussowitsch {
3078f1580f4eSBarry Smith   PetscFunctionBegin;
3079f1580f4eSBarry Smith   PCHPDDMPackageInitialized = PETSC_FALSE;
30803ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3081f1580f4eSBarry Smith }
30829bb5c669SPierre Jolivet 
30839bb5c669SPierre Jolivet static PetscErrorCode MatMult_Harmonic(Mat A, Vec x, Vec y)
30849bb5c669SPierre Jolivet {
30859bb5c669SPierre Jolivet   Harmonic h; /* [ A_00  A_01       ], furthermore, A_00 = [ A_loc,loc  A_loc,ovl ], thus, A_01 = [         ] */
30869bb5c669SPierre Jolivet               /* [ A_10  A_11  A_12 ]                      [ A_ovl,loc  A_ovl,ovl ]               [ A_ovl,1 ] */
30879bb5c669SPierre Jolivet   Vec sub;    /*  y = A x = R_loc R_0 [ A_00  A_01 ]^-1                                   R_loc = [  I_loc  ] */
30889bb5c669SPierre Jolivet               /*                      [ A_10  A_11 ]    R_1^T A_12 x                              [         ] */
30899bb5c669SPierre Jolivet   PetscFunctionBegin;
30909bb5c669SPierre Jolivet   PetscCall(MatShellGetContext(A, &h));
30919bb5c669SPierre Jolivet   PetscCall(VecSet(h->v, 0.0));
30929bb5c669SPierre Jolivet   PetscCall(VecGetSubVector(h->v, h->is[0], &sub));
30939bb5c669SPierre Jolivet   PetscCall(MatMult(h->A[0], x, sub));
30949bb5c669SPierre Jolivet   PetscCall(VecRestoreSubVector(h->v, h->is[0], &sub));
30959bb5c669SPierre Jolivet   PetscCall(KSPSolve(h->ksp, h->v, h->v));
30969bb5c669SPierre Jolivet   PetscCall(VecISCopy(h->v, h->is[1], SCATTER_REVERSE, y));
30979bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
30989bb5c669SPierre Jolivet }
30999bb5c669SPierre Jolivet 
31009bb5c669SPierre Jolivet static PetscErrorCode MatMultTranspose_Harmonic(Mat A, Vec y, Vec x)
31019bb5c669SPierre Jolivet {
31029bb5c669SPierre Jolivet   Harmonic h;   /* x = A^T y =            [ A_00  A_01 ]^-T R_0^T R_loc^T y */
31039bb5c669SPierre Jolivet   Vec      sub; /*             A_12^T R_1 [ A_10  A_11 ]                    */
31049bb5c669SPierre Jolivet 
31059bb5c669SPierre Jolivet   PetscFunctionBegin;
31069bb5c669SPierre Jolivet   PetscCall(MatShellGetContext(A, &h));
31079bb5c669SPierre Jolivet   PetscCall(VecSet(h->v, 0.0));
31089bb5c669SPierre Jolivet   PetscCall(VecISCopy(h->v, h->is[1], SCATTER_FORWARD, y));
31099bb5c669SPierre Jolivet   PetscCall(KSPSolveTranspose(h->ksp, h->v, h->v));
31109bb5c669SPierre Jolivet   PetscCall(VecGetSubVector(h->v, h->is[0], &sub));
31119bb5c669SPierre Jolivet   PetscCall(MatMultTranspose(h->A[0], sub, x));
31129bb5c669SPierre Jolivet   PetscCall(VecRestoreSubVector(h->v, h->is[0], &sub));
31139bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
31149bb5c669SPierre Jolivet }
31159bb5c669SPierre Jolivet 
31169bb5c669SPierre Jolivet static PetscErrorCode MatProduct_AB_Harmonic(Mat S, Mat X, Mat Y, void *)
31179bb5c669SPierre Jolivet {
31189bb5c669SPierre Jolivet   Harmonic h;
31199bb5c669SPierre Jolivet   Mat      A, B;
31209bb5c669SPierre Jolivet   Vec      a, b;
31219bb5c669SPierre Jolivet 
31229bb5c669SPierre Jolivet   PetscFunctionBegin;
31239bb5c669SPierre Jolivet   PetscCall(MatShellGetContext(S, &h));
3124fb842aefSJose E. Roman   PetscCall(MatMatMult(h->A[0], X, MAT_INITIAL_MATRIX, PETSC_CURRENT, &A));
31259bb5c669SPierre Jolivet   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, h->ksp->pc->mat->rmap->n, A->cmap->n, nullptr, &B));
31269bb5c669SPierre Jolivet   for (PetscInt i = 0; i < A->cmap->n; ++i) {
31279bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecRead(A, i, &a));
31289bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecWrite(B, i, &b));
31299bb5c669SPierre Jolivet     PetscCall(VecISCopy(b, h->is[0], SCATTER_FORWARD, a));
31309bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecWrite(B, i, &b));
31319bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecRead(A, i, &a));
31329bb5c669SPierre Jolivet   }
31339bb5c669SPierre Jolivet   PetscCall(MatDestroy(&A));
31349bb5c669SPierre Jolivet   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, h->ksp->pc->mat->rmap->n, B->cmap->n, nullptr, &A));
31359bb5c669SPierre Jolivet   PetscCall(KSPMatSolve(h->ksp, B, A));
31369bb5c669SPierre Jolivet   PetscCall(MatDestroy(&B));
31379bb5c669SPierre Jolivet   for (PetscInt i = 0; i < A->cmap->n; ++i) {
31389bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecRead(A, i, &a));
31399bb5c669SPierre Jolivet     PetscCall(MatDenseGetColumnVecWrite(Y, i, &b));
31409bb5c669SPierre Jolivet     PetscCall(VecISCopy(a, h->is[1], SCATTER_REVERSE, b));
31419bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &b));
31429bb5c669SPierre Jolivet     PetscCall(MatDenseRestoreColumnVecRead(A, i, &a));
31439bb5c669SPierre Jolivet   }
31449bb5c669SPierre Jolivet   PetscCall(MatDestroy(&A));
31459bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
31469bb5c669SPierre Jolivet }
31479bb5c669SPierre Jolivet 
31489bb5c669SPierre Jolivet static PetscErrorCode MatDestroy_Harmonic(Mat A)
31499bb5c669SPierre Jolivet {
31509bb5c669SPierre Jolivet   Harmonic h;
31519bb5c669SPierre Jolivet 
31529bb5c669SPierre Jolivet   PetscFunctionBegin;
31539bb5c669SPierre Jolivet   PetscCall(MatShellGetContext(A, &h));
315432603206SJames Wright   for (PetscInt i = 0; i < 5; ++i) PetscCall(ISDestroy(h->is + i));
31559bb5c669SPierre Jolivet   PetscCall(PetscFree(h->is));
31569bb5c669SPierre Jolivet   PetscCall(VecDestroy(&h->v));
31579bb5c669SPierre Jolivet   for (PetscInt i = 0; i < 2; ++i) PetscCall(MatDestroy(h->A + i));
31589bb5c669SPierre Jolivet   PetscCall(PetscFree(h->A));
31599bb5c669SPierre Jolivet   PetscCall(KSPDestroy(&h->ksp));
31609bb5c669SPierre Jolivet   PetscCall(PetscFree(h));
31619bb5c669SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
31629bb5c669SPierre Jolivet }
3163