1f1580f4eSBarry Smith #include <petsc/private/matimpl.h> 2f1580f4eSBarry Smith #include <petsc/private/petschpddm.h> /*I "petscpc.h" I*/ 3e905f78bSJacob Faibussowitsch #include <petsc/private/pcimpl.h> 4e905f78bSJacob Faibussowitsch #include <petsc/private/dmimpl.h> /* this must be included after petschpddm.h so that DM_MAX_WORK_VECTORS is not defined */ 5f1580f4eSBarry Smith /* otherwise, it is assumed that one is compiling libhpddm_petsc => circular dependency */ 6fbf9dbe5SBarry Smith #if PetscDefined(USE_FORTRAN_BINDINGS) 7f1580f4eSBarry Smith #include <petsc/private/fortranimpl.h> 8f1580f4eSBarry Smith #endif 9f1580f4eSBarry Smith 10db4a47b3SPierre Jolivet static PetscErrorCode (*loadedSym)(HPDDM::Schwarz<PetscScalar> *const, IS, Mat, Mat, Mat, std::vector<Vec>, PC_HPDDM_Level **const) = nullptr; 11f1580f4eSBarry Smith 12f1580f4eSBarry Smith static PetscBool PCHPDDMPackageInitialized = PETSC_FALSE; 13f1580f4eSBarry Smith 14f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_Strc; 15f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_PtAP; 16f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_PtBP; 17f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_Next; 18f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_SetUp[PETSC_PCHPDDM_MAXLEVELS]; 19f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_Solve[PETSC_PCHPDDM_MAXLEVELS]; 20f1580f4eSBarry Smith 21db4a47b3SPierre Jolivet const char *const PCHPDDMCoarseCorrectionTypes[] = {"DEFLATED", "ADDITIVE", "BALANCED", "PCHPDDMCoarseCorrectionType", "PC_HPDDM_COARSE_CORRECTION_", nullptr}; 22f1580f4eSBarry Smith 23d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCReset_HPDDM(PC pc) 24d71ae5a4SJacob Faibussowitsch { 25f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 26f1580f4eSBarry Smith PetscInt i; 27f1580f4eSBarry Smith 28f1580f4eSBarry Smith PetscFunctionBegin; 29f1580f4eSBarry Smith if (data->levels) { 30f1580f4eSBarry Smith for (i = 0; i < PETSC_PCHPDDM_MAXLEVELS && data->levels[i]; ++i) { 31f1580f4eSBarry Smith PetscCall(KSPDestroy(&data->levels[i]->ksp)); 32f1580f4eSBarry Smith PetscCall(PCDestroy(&data->levels[i]->pc)); 33f1580f4eSBarry Smith PetscCall(PetscFree(data->levels[i])); 34f1580f4eSBarry Smith } 35f1580f4eSBarry Smith PetscCall(PetscFree(data->levels)); 36f1580f4eSBarry Smith } 37f1580f4eSBarry Smith 38f1580f4eSBarry Smith PetscCall(ISDestroy(&data->is)); 39f1580f4eSBarry Smith PetscCall(MatDestroy(&data->aux)); 40f1580f4eSBarry Smith PetscCall(MatDestroy(&data->B)); 41f1580f4eSBarry Smith PetscCall(VecDestroy(&data->normal)); 42f1580f4eSBarry Smith data->correction = PC_HPDDM_COARSE_CORRECTION_DEFLATED; 43c8ea6600SPierre Jolivet data->Neumann = PETSC_BOOL3_UNKNOWN; 44f1580f4eSBarry Smith data->deflation = PETSC_FALSE; 45db4a47b3SPierre Jolivet data->setup = nullptr; 46db4a47b3SPierre Jolivet data->setup_ctx = nullptr; 473ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 48f1580f4eSBarry Smith } 49f1580f4eSBarry Smith 50d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCDestroy_HPDDM(PC pc) 51d71ae5a4SJacob Faibussowitsch { 52f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 53f1580f4eSBarry Smith 54f1580f4eSBarry Smith PetscFunctionBegin; 55f1580f4eSBarry Smith PetscCall(PCReset_HPDDM(pc)); 56f1580f4eSBarry Smith PetscCall(PetscFree(data)); 57db4a47b3SPierre Jolivet PetscCall(PetscObjectChangeTypeName((PetscObject)pc, nullptr)); 58db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetAuxiliaryMat_C", nullptr)); 59db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMHasNeumannMat_C", nullptr)); 60db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetRHSMat_C", nullptr)); 61db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetCoarseCorrectionType_C", nullptr)); 62db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetCoarseCorrectionType_C", nullptr)); 63db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetSTShareSubKSP_C", nullptr)); 64db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetSTShareSubKSP_C", nullptr)); 65db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetDeflationMat_C", nullptr)); 663ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 67f1580f4eSBarry Smith } 68f1580f4eSBarry Smith 69d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMSetAuxiliaryMat_Private(PC pc, IS is, Mat A, PetscBool deflation) 70d71ae5a4SJacob Faibussowitsch { 71f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 72f1580f4eSBarry Smith 73f1580f4eSBarry Smith PetscFunctionBegin; 74f1580f4eSBarry Smith if (is) { 75f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)is)); 76f1580f4eSBarry Smith if (data->is) { /* new overlap definition resets the PC */ 77f1580f4eSBarry Smith PetscCall(PCReset_HPDDM(pc)); 78f1580f4eSBarry Smith pc->setfromoptionscalled = 0; 79b853e353SPierre Jolivet pc->setupcalled = 0; 80f1580f4eSBarry Smith } 81f1580f4eSBarry Smith PetscCall(ISDestroy(&data->is)); 82f1580f4eSBarry Smith data->is = is; 83f1580f4eSBarry Smith } 84f1580f4eSBarry Smith if (A) { 85f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)A)); 86f1580f4eSBarry Smith PetscCall(MatDestroy(&data->aux)); 87f1580f4eSBarry Smith data->aux = A; 88f1580f4eSBarry Smith } 89f1580f4eSBarry Smith data->deflation = deflation; 903ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 91f1580f4eSBarry Smith } 92f1580f4eSBarry Smith 933df4cd7bSPierre Jolivet static inline PetscErrorCode PCHPDDMSetAuxiliaryMatNormal_Private(PC pc, Mat A, Mat N, Mat *B, const char *pcpre, Vec *diagonal = nullptr) 94d71ae5a4SJacob Faibussowitsch { 95f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 96f1580f4eSBarry Smith Mat *splitting, *sub, aux; 973df4cd7bSPierre Jolivet Vec d; 98f1580f4eSBarry Smith IS is, cols[2], rows; 99f1580f4eSBarry Smith PetscReal norm; 100f1580f4eSBarry Smith PetscBool flg; 101f1580f4eSBarry Smith char type[256] = {}; /* same size as in src/ksp/pc/interface/pcset.c */ 102f1580f4eSBarry Smith 103f1580f4eSBarry Smith PetscFunctionBegin; 104f1580f4eSBarry Smith PetscCall(MatConvert(N, MATAIJ, MAT_INITIAL_MATRIX, B)); 105f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, A->cmap->n, A->cmap->rstart, 1, cols)); 106f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, A->rmap->N, 0, 1, &rows)); 107f1580f4eSBarry Smith PetscCall(MatSetOption(A, MAT_SUBMAT_SINGLEIS, PETSC_TRUE)); 108f1580f4eSBarry Smith PetscCall(MatIncreaseOverlap(*B, 1, cols, 1)); 109f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(A, 1, &rows, cols, MAT_INITIAL_MATRIX, &splitting)); 110f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, A->cmap->n, A->cmap->rstart, 1, &is)); 111f1580f4eSBarry Smith PetscCall(ISEmbed(*cols, is, PETSC_TRUE, cols + 1)); 112f1580f4eSBarry Smith PetscCall(ISDestroy(&is)); 113f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(*splitting, 1, &rows, cols + 1, MAT_INITIAL_MATRIX, &sub)); 114f1580f4eSBarry Smith PetscCall(ISDestroy(cols + 1)); 115f1580f4eSBarry Smith PetscCall(MatFindZeroRows(*sub, &is)); 116f1580f4eSBarry Smith PetscCall(MatDestroySubMatrices(1, &sub)); 117f1580f4eSBarry Smith PetscCall(ISDestroy(&rows)); 118f1580f4eSBarry Smith PetscCall(MatSetOption(*splitting, MAT_KEEP_NONZERO_PATTERN, PETSC_TRUE)); 119db4a47b3SPierre Jolivet PetscCall(MatZeroRowsIS(*splitting, is, 0.0, nullptr, nullptr)); 120f1580f4eSBarry Smith PetscCall(ISDestroy(&is)); 121db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetString(nullptr, pcpre, "-pc_hpddm_levels_1_sub_pc_type", type, sizeof(type), nullptr)); 122f1580f4eSBarry Smith PetscCall(PetscStrcmp(type, PCQR, &flg)); 123f1580f4eSBarry Smith if (!flg) { 124f1580f4eSBarry Smith Mat conjugate = *splitting; 125f1580f4eSBarry Smith if (PetscDefined(USE_COMPLEX)) { 126f1580f4eSBarry Smith PetscCall(MatDuplicate(*splitting, MAT_COPY_VALUES, &conjugate)); 127f1580f4eSBarry Smith PetscCall(MatConjugate(conjugate)); 128f1580f4eSBarry Smith } 129f1580f4eSBarry Smith PetscCall(MatTransposeMatMult(conjugate, *splitting, MAT_INITIAL_MATRIX, PETSC_DEFAULT, &aux)); 130f1580f4eSBarry Smith if (PetscDefined(USE_COMPLEX)) PetscCall(MatDestroy(&conjugate)); 131f1580f4eSBarry Smith PetscCall(MatNorm(aux, NORM_FROBENIUS, &norm)); 132f1580f4eSBarry Smith PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE)); 1333df4cd7bSPierre Jolivet if (diagonal) { 1343df4cd7bSPierre Jolivet PetscCall(VecNorm(*diagonal, NORM_INFINITY, &norm)); 1353df4cd7bSPierre Jolivet if (norm > PETSC_SMALL) { 1363df4cd7bSPierre Jolivet VecScatter scatter; 1373df4cd7bSPierre Jolivet PetscInt n; 1383df4cd7bSPierre Jolivet PetscCall(ISGetLocalSize(*cols, &n)); 1393df4cd7bSPierre Jolivet PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)pc), n, PETSC_DECIDE, &d)); 1403df4cd7bSPierre Jolivet PetscCall(VecScatterCreate(*diagonal, *cols, d, nullptr, &scatter)); 1413df4cd7bSPierre Jolivet PetscCall(VecScatterBegin(scatter, *diagonal, d, INSERT_VALUES, SCATTER_FORWARD)); 1423df4cd7bSPierre Jolivet PetscCall(VecScatterEnd(scatter, *diagonal, d, INSERT_VALUES, SCATTER_FORWARD)); 1433df4cd7bSPierre Jolivet PetscCall(VecScatterDestroy(&scatter)); 1443df4cd7bSPierre Jolivet PetscCall(MatScale(aux, -1.0)); 1453df4cd7bSPierre Jolivet PetscCall(MatDiagonalSet(aux, d, ADD_VALUES)); 1463df4cd7bSPierre Jolivet PetscCall(VecDestroy(&d)); 1473df4cd7bSPierre Jolivet } else PetscCall(VecDestroy(diagonal)); 1483df4cd7bSPierre Jolivet } 1493df4cd7bSPierre Jolivet if (!diagonal) PetscCall(MatShift(aux, PETSC_SMALL * norm)); 150f1580f4eSBarry Smith } else { 151f1580f4eSBarry Smith PetscBool flg; 1523df4cd7bSPierre Jolivet if (diagonal) { 1533df4cd7bSPierre Jolivet PetscCall(VecNorm(*diagonal, NORM_INFINITY, &norm)); 1543df4cd7bSPierre Jolivet PetscCheck(norm < PETSC_SMALL, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Nonzero diagonal A11 block"); 1553df4cd7bSPierre Jolivet PetscCall(VecDestroy(diagonal)); 1563df4cd7bSPierre Jolivet } 157f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)N, MATNORMAL, &flg)); 158f1580f4eSBarry Smith if (flg) PetscCall(MatCreateNormal(*splitting, &aux)); 159f1580f4eSBarry Smith else PetscCall(MatCreateNormalHermitian(*splitting, &aux)); 160f1580f4eSBarry Smith } 161f1580f4eSBarry Smith PetscCall(MatDestroySubMatrices(1, &splitting)); 162db4a47b3SPierre Jolivet PetscCall(PCHPDDMSetAuxiliaryMat(pc, *cols, aux, nullptr, nullptr)); 163c8ea6600SPierre Jolivet data->Neumann = PETSC_BOOL3_TRUE; 164f1580f4eSBarry Smith PetscCall(ISDestroy(cols)); 165f1580f4eSBarry Smith PetscCall(MatDestroy(&aux)); 1663ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 167f1580f4eSBarry Smith } 168f1580f4eSBarry Smith 169d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetAuxiliaryMat_HPDDM(PC pc, IS is, Mat A, PetscErrorCode (*setup)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void *setup_ctx) 170d71ae5a4SJacob Faibussowitsch { 171f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 172f1580f4eSBarry Smith 173f1580f4eSBarry Smith PetscFunctionBegin; 174f1580f4eSBarry Smith PetscCall(PCHPDDMSetAuxiliaryMat_Private(pc, is, A, PETSC_FALSE)); 175f1580f4eSBarry Smith if (setup) { 176f1580f4eSBarry Smith data->setup = setup; 177f1580f4eSBarry Smith data->setup_ctx = setup_ctx; 178f1580f4eSBarry Smith } 1793ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 180f1580f4eSBarry Smith } 181f1580f4eSBarry Smith 18270009435SPierre Jolivet /*@C 18304c3f3b8SBarry Smith PCHPDDMSetAuxiliaryMat - Sets the auxiliary matrix used by `PCHPDDM` for the concurrent GenEO problems at the finest level. 184f1580f4eSBarry Smith 185f1580f4eSBarry Smith Input Parameters: 186f1580f4eSBarry Smith + pc - preconditioner context 187f1580f4eSBarry Smith . is - index set of the local auxiliary, e.g., Neumann, matrix 188f1580f4eSBarry Smith . A - auxiliary sequential matrix 18904c3f3b8SBarry Smith . setup - function for generating the auxiliary matrix entries, may be `NULL` 19004c3f3b8SBarry Smith - ctx - context for `setup`, may be `NULL` 19104c3f3b8SBarry Smith 19204c3f3b8SBarry Smith Calling sequence of `setup`: 19304c3f3b8SBarry Smith + J - matrix whose values are to be set 19404c3f3b8SBarry Smith . t - time 19504c3f3b8SBarry Smith . X - linearization point 19604c3f3b8SBarry Smith . X_t - time-derivative of the linearization point 19704c3f3b8SBarry Smith . s - step 19804c3f3b8SBarry Smith . ovl - index set of the local auxiliary, e.g., Neumann, matrix 19904c3f3b8SBarry Smith - ctx - context for `setup`, may be `NULL` 200f1580f4eSBarry Smith 201f1580f4eSBarry Smith Level: intermediate 202f1580f4eSBarry Smith 20304c3f3b8SBarry Smith Note: 20404c3f3b8SBarry Smith As an example, in a finite element context with nonoverlapping subdomains plus (overlapping) ghost elements, this could be the unassembled (Neumann) 20504c3f3b8SBarry Smith local overlapping operator. As opposed to the assembled (Dirichlet) local overlapping operator obtained by summing neighborhood contributions 20604c3f3b8SBarry Smith at the interface of ghost elements. 20704c3f3b8SBarry Smith 20870009435SPierre Jolivet Fortran Notes: 20904c3f3b8SBarry Smith Only `PETSC_NULL_FUNCTION` is supported for `setup` and `ctx` is never accessed 21070009435SPierre Jolivet 211f1580f4eSBarry Smith .seealso: `PCHPDDM`, `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHPDDMSetRHSMat()`, `MATIS` 212f1580f4eSBarry Smith @*/ 21304c3f3b8SBarry 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) 214d71ae5a4SJacob Faibussowitsch { 215f1580f4eSBarry Smith PetscFunctionBegin; 216f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 217f1580f4eSBarry Smith if (is) PetscValidHeaderSpecific(is, IS_CLASSID, 2); 218f1580f4eSBarry Smith if (A) PetscValidHeaderSpecific(A, MAT_CLASSID, 3); 21904c3f3b8SBarry Smith PetscTryMethod(pc, "PCHPDDMSetAuxiliaryMat_C", (PC, IS, Mat, PetscErrorCode(*)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void *), (pc, is, A, setup, ctx)); 2203ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 221f1580f4eSBarry Smith } 222f1580f4eSBarry Smith 223d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMHasNeumannMat_HPDDM(PC pc, PetscBool has) 224d71ae5a4SJacob Faibussowitsch { 225f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 226f1580f4eSBarry Smith 227f1580f4eSBarry Smith PetscFunctionBegin; 228c8ea6600SPierre Jolivet data->Neumann = PetscBoolToBool3(has); 2293ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 230f1580f4eSBarry Smith } 231f1580f4eSBarry Smith 232f1580f4eSBarry Smith /*@ 233f1580f4eSBarry Smith PCHPDDMHasNeumannMat - Informs `PCHPDDM` that the `Mat` passed to `PCHPDDMSetAuxiliaryMat()` is the local Neumann matrix. 234f1580f4eSBarry Smith 235f1580f4eSBarry Smith Input Parameters: 236f1580f4eSBarry Smith + pc - preconditioner context 237f1580f4eSBarry Smith - has - Boolean value 238f1580f4eSBarry Smith 239f1580f4eSBarry Smith Level: intermediate 240f1580f4eSBarry Smith 241f1580f4eSBarry Smith Notes: 2427eb095acSPierre Jolivet This may be used to bypass a call to `MatCreateSubMatrices()` and to `MatConvert()` for `MATSBAIJ` matrices. 243f1580f4eSBarry Smith 244f1580f4eSBarry Smith If a `DMCreateNeumannOverlap()` 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`. 245f1580f4eSBarry Smith 246f1580f4eSBarry Smith .seealso: `PCHPDDM`, `PCHPDDMSetAuxiliaryMat()` 247f1580f4eSBarry Smith @*/ 248d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMHasNeumannMat(PC pc, PetscBool has) 249d71ae5a4SJacob Faibussowitsch { 250f1580f4eSBarry Smith PetscFunctionBegin; 251f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 252f1580f4eSBarry Smith PetscTryMethod(pc, "PCHPDDMHasNeumannMat_C", (PC, PetscBool), (pc, has)); 2533ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 254f1580f4eSBarry Smith } 255f1580f4eSBarry Smith 256d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetRHSMat_HPDDM(PC pc, Mat B) 257d71ae5a4SJacob Faibussowitsch { 258f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 259f1580f4eSBarry Smith 260f1580f4eSBarry Smith PetscFunctionBegin; 261f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)B)); 262f1580f4eSBarry Smith PetscCall(MatDestroy(&data->B)); 263f1580f4eSBarry Smith data->B = B; 2643ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 265f1580f4eSBarry Smith } 266f1580f4eSBarry Smith 267f1580f4eSBarry Smith /*@ 26804c3f3b8SBarry Smith PCHPDDMSetRHSMat - Sets the right-hand side matrix used by `PCHPDDM` for the concurrent GenEO problems at the finest level. 269f1580f4eSBarry Smith 270f1580f4eSBarry Smith Input Parameters: 271f1580f4eSBarry Smith + pc - preconditioner context 272f1580f4eSBarry Smith - B - right-hand side sequential matrix 273f1580f4eSBarry Smith 274f1580f4eSBarry Smith Level: advanced 275f1580f4eSBarry Smith 27604c3f3b8SBarry Smith Note: 27704c3f3b8SBarry Smith Must be used in conjunction with `PCHPDDMSetAuxiliaryMat`(N), so that Nv = lambda Bv is solved using `EPSSetOperators`(N, B). 27804c3f3b8SBarry 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. 27904c3f3b8SBarry Smith 280f1580f4eSBarry Smith .seealso: `PCHPDDMSetAuxiliaryMat()`, `PCHPDDM` 281f1580f4eSBarry Smith @*/ 282d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetRHSMat(PC pc, Mat B) 283d71ae5a4SJacob Faibussowitsch { 284f1580f4eSBarry Smith PetscFunctionBegin; 285f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 286f1580f4eSBarry Smith if (B) { 287f1580f4eSBarry Smith PetscValidHeaderSpecific(B, MAT_CLASSID, 2); 288f1580f4eSBarry Smith PetscTryMethod(pc, "PCHPDDMSetRHSMat_C", (PC, Mat), (pc, B)); 289f1580f4eSBarry Smith } 2903ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 291f1580f4eSBarry Smith } 292f1580f4eSBarry Smith 293d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetFromOptions_HPDDM(PC pc, PetscOptionItems *PetscOptionsObject) 294d71ae5a4SJacob Faibussowitsch { 295f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 296f1580f4eSBarry Smith PC_HPDDM_Level **levels = data->levels; 297f1580f4eSBarry Smith char prefix[256]; 298f1580f4eSBarry Smith int i = 1; 299f1580f4eSBarry Smith PetscMPIInt size, previous; 300f1580f4eSBarry Smith PetscInt n; 301f1580f4eSBarry Smith PCHPDDMCoarseCorrectionType type; 302c8ea6600SPierre Jolivet PetscBool flg = PETSC_TRUE, set; 303f1580f4eSBarry Smith 304f1580f4eSBarry Smith PetscFunctionBegin; 305f1580f4eSBarry Smith if (!data->levels) { 306f1580f4eSBarry Smith PetscCall(PetscCalloc1(PETSC_PCHPDDM_MAXLEVELS, &levels)); 307f1580f4eSBarry Smith data->levels = levels; 308f1580f4eSBarry Smith } 309f1580f4eSBarry Smith PetscOptionsHeadBegin(PetscOptionsObject, "PCHPDDM options"); 310f1580f4eSBarry Smith PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)pc), &size)); 311f1580f4eSBarry Smith previous = size; 312f1580f4eSBarry Smith while (i < PETSC_PCHPDDM_MAXLEVELS) { 313f1580f4eSBarry Smith PetscInt p = 1; 314f1580f4eSBarry Smith 3154dfa11a4SJacob Faibussowitsch if (!data->levels[i - 1]) PetscCall(PetscNew(data->levels + i - 1)); 316f1580f4eSBarry Smith data->levels[i - 1]->parent = data; 317f1580f4eSBarry Smith /* if the previous level has a single process, it is not possible to coarsen further */ 318f1580f4eSBarry Smith if (previous == 1 || !flg) break; 319f1580f4eSBarry Smith data->levels[i - 1]->nu = 0; 320f1580f4eSBarry Smith data->levels[i - 1]->threshold = -1.0; 321f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_nev", i)); 322db4a47b3SPierre 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)); 323f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_threshold", i)); 324db4a47b3SPierre 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)); 325f1580f4eSBarry Smith if (i == 1) { 326f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_1_st_share_sub_ksp")); 327db4a47b3SPierre Jolivet PetscCall(PetscOptionsBool(prefix, "Shared KSP between SLEPc ST and the fine-level subdomain solver", "PCHPDDMSetSTShareSubKSP", PETSC_FALSE, &data->share, nullptr)); 328f1580f4eSBarry Smith } 329f1580f4eSBarry Smith /* if there is no prescribed coarsening, just break out of the loop */ 330f1580f4eSBarry Smith if (data->levels[i - 1]->threshold <= 0.0 && data->levels[i - 1]->nu <= 0 && !(data->deflation && i == 1)) break; 331f1580f4eSBarry Smith else { 332f1580f4eSBarry Smith ++i; 333f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_nev", i)); 334f1580f4eSBarry Smith PetscCall(PetscOptionsHasName(PetscOptionsObject->options, PetscOptionsObject->prefix, prefix, &flg)); 335f1580f4eSBarry Smith if (!flg) { 336f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_threshold", i)); 337f1580f4eSBarry Smith PetscCall(PetscOptionsHasName(PetscOptionsObject->options, PetscOptionsObject->prefix, prefix, &flg)); 338f1580f4eSBarry Smith } 339f1580f4eSBarry Smith if (flg) { 340f1580f4eSBarry Smith /* if there are coarsening options for the next level, then register it */ 341f1580f4eSBarry Smith /* otherwise, don't to avoid having both options levels_N_p and coarse_p */ 342f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_p", i)); 343f1580f4eSBarry 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))); 344f1580f4eSBarry Smith previous = p; 345f1580f4eSBarry Smith } 346f1580f4eSBarry Smith } 347f1580f4eSBarry Smith } 348f1580f4eSBarry Smith data->N = i; 349f1580f4eSBarry Smith n = 1; 350f1580f4eSBarry Smith if (i > 1) { 351f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_coarse_p")); 352db4a47b3SPierre Jolivet PetscCall(PetscOptionsRangeInt(prefix, "Number of processes used to assemble the coarsest operator", "PCHPDDM", n, &n, nullptr, 1, PetscMax(1, previous / 2))); 35302800ff6SPierre Jolivet #if PetscDefined(HAVE_MUMPS) 354f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "pc_hpddm_coarse_")); 355db4a47b3SPierre Jolivet PetscCall(PetscOptionsHasName(nullptr, prefix, "-mat_mumps_use_omp_threads", &flg)); 356f1580f4eSBarry Smith if (flg) { 357f1580f4eSBarry Smith char type[64]; /* same size as in src/ksp/pc/impls/factor/factimpl.c */ 358c6a7a370SJeremy L Thompson PetscCall(PetscStrncpy(type, n > 1 && PetscDefined(HAVE_MUMPS) ? MATSOLVERMUMPS : MATSOLVERPETSC, sizeof(type))); /* default solver for a MatMPIAIJ or a MatSeqAIJ */ 359db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetString(nullptr, prefix, "-pc_factor_mat_solver_type", type, sizeof(type), nullptr)); 3603ce573a3SPierre Jolivet PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg)); 361f1580f4eSBarry 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); 362f1580f4eSBarry Smith size = n; 363f1580f4eSBarry Smith n = -1; 364db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetInt(nullptr, prefix, "-mat_mumps_use_omp_threads", &n, nullptr)); 365f1580f4eSBarry Smith PetscCheck(n >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Need to specify a positive integer for -%smat_mumps_use_omp_threads", prefix); 366f1580f4eSBarry 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" : ""); 367f1580f4eSBarry Smith } 368f1580f4eSBarry Smith #endif 369f1580f4eSBarry Smith PetscCall(PetscOptionsEnum("-pc_hpddm_coarse_correction", "Type of coarse correction applied each iteration", "PCHPDDMSetCoarseCorrectionType", PCHPDDMCoarseCorrectionTypes, (PetscEnum)data->correction, (PetscEnum *)&type, &flg)); 370f1580f4eSBarry Smith if (flg) PetscCall(PCHPDDMSetCoarseCorrectionType(pc, type)); 371f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_has_neumann")); 372c8ea6600SPierre Jolivet PetscCall(PetscOptionsBool(prefix, "Is the auxiliary Mat the local Neumann matrix?", "PCHPDDMHasNeumannMat", PetscBool3ToBool(data->Neumann), &flg, &set)); 373c8ea6600SPierre Jolivet if (set) data->Neumann = PetscBoolToBool3(flg); 374f1580f4eSBarry Smith data->log_separate = PETSC_FALSE; 375f1580f4eSBarry Smith if (PetscDefined(USE_LOG)) { 376f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_log_separate")); 377db4a47b3SPierre Jolivet PetscCall(PetscOptionsBool(prefix, "Log events level by level instead of inside PCSetUp()/KSPSolve()", nullptr, data->log_separate, &data->log_separate, nullptr)); 378f1580f4eSBarry Smith } 379f1580f4eSBarry Smith } 380f1580f4eSBarry Smith PetscOptionsHeadEnd(); 381f1580f4eSBarry Smith while (i < PETSC_PCHPDDM_MAXLEVELS && data->levels[i]) PetscCall(PetscFree(data->levels[i++])); 3823ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 383f1580f4eSBarry Smith } 384f1580f4eSBarry Smith 385d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_HPDDM(PC pc, Vec x, Vec y) 386d71ae5a4SJacob Faibussowitsch { 387f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 388f1580f4eSBarry Smith 389f1580f4eSBarry Smith PetscFunctionBegin; 390f1580f4eSBarry Smith PetscCall(PetscCitationsRegister(HPDDMCitation, &HPDDMCite)); 391f1580f4eSBarry Smith PetscCheck(data->levels[0]->ksp, PETSC_COMM_SELF, PETSC_ERR_PLIB, "No KSP attached to PCHPDDM"); 392db4a47b3SPierre 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 */ 393f1580f4eSBarry Smith PetscCall(KSPSolve(data->levels[0]->ksp, x, y)); 394db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Solve[0], data->levels[0]->ksp, nullptr, nullptr, nullptr)); 3953ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 396f1580f4eSBarry Smith } 397f1580f4eSBarry Smith 398d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_HPDDM(PC pc, Mat X, Mat Y) 399d71ae5a4SJacob Faibussowitsch { 400f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 401f1580f4eSBarry Smith 402f1580f4eSBarry Smith PetscFunctionBegin; 403f1580f4eSBarry Smith PetscCall(PetscCitationsRegister(HPDDMCitation, &HPDDMCite)); 404f1580f4eSBarry Smith PetscCheck(data->levels[0]->ksp, PETSC_COMM_SELF, PETSC_ERR_PLIB, "No KSP attached to PCHPDDM"); 405f1580f4eSBarry Smith PetscCall(KSPMatSolve(data->levels[0]->ksp, X, Y)); 4063ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 407f1580f4eSBarry Smith } 408f1580f4eSBarry Smith 40904c3f3b8SBarry Smith // PetscClangLinter pragma disable: -fdoc-internal-linkage 410f1580f4eSBarry Smith /*@C 411f1580f4eSBarry Smith PCHPDDMGetComplexities - Computes the grid and operator complexities. 412f1580f4eSBarry Smith 413f1580f4eSBarry Smith Input Parameter: 414f1580f4eSBarry Smith . pc - preconditioner context 415f1580f4eSBarry Smith 416f1580f4eSBarry Smith Output Parameters: 417f1580f4eSBarry Smith + gc - grid complexity = sum_i(m_i) / m_1 418f1580f4eSBarry Smith - oc - operator complexity = sum_i(nnz_i) / nnz_1 419f1580f4eSBarry Smith 420f1580f4eSBarry Smith Level: advanced 421f1580f4eSBarry Smith 422f1580f4eSBarry Smith .seealso: `PCMGGetGridComplexity()`, `PCHPDDM`, `PCHYPRE`, `PCGAMG` 423f1580f4eSBarry Smith @*/ 424d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMGetComplexities(PC pc, PetscReal *gc, PetscReal *oc) 425d71ae5a4SJacob Faibussowitsch { 426f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 427f1580f4eSBarry Smith MatInfo info; 428f1580f4eSBarry Smith PetscInt n, m; 429f1580f4eSBarry Smith PetscLogDouble accumulate[2]{}, nnz1 = 1.0, m1 = 1.0; 430f1580f4eSBarry Smith 431f1580f4eSBarry Smith PetscFunctionBegin; 432f1580f4eSBarry Smith for (n = 0, *gc = 0, *oc = 0; n < data->N; ++n) { 433f1580f4eSBarry Smith if (data->levels[n]->ksp) { 434f1580f4eSBarry Smith Mat P, A; 435db4a47b3SPierre Jolivet PetscCall(KSPGetOperators(data->levels[n]->ksp, nullptr, &P)); 436db4a47b3SPierre Jolivet PetscCall(MatGetSize(P, &m, nullptr)); 437f1580f4eSBarry Smith accumulate[0] += m; 438f1580f4eSBarry Smith if (n == 0) { 439f1580f4eSBarry Smith PetscBool flg; 440f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompareAny((PetscObject)P, &flg, MATNORMAL, MATNORMALHERMITIAN, "")); 441f1580f4eSBarry Smith if (flg) { 442f1580f4eSBarry Smith PetscCall(MatConvert(P, MATAIJ, MAT_INITIAL_MATRIX, &A)); 443f1580f4eSBarry Smith P = A; 444f1580f4eSBarry Smith } else PetscCall(PetscObjectReference((PetscObject)P)); 445f1580f4eSBarry Smith } 446f1580f4eSBarry Smith if (P->ops->getinfo) { 447f1580f4eSBarry Smith PetscCall(MatGetInfo(P, MAT_GLOBAL_SUM, &info)); 448f1580f4eSBarry Smith accumulate[1] += info.nz_used; 449f1580f4eSBarry Smith } 450f1580f4eSBarry Smith if (n == 0) { 451f1580f4eSBarry Smith m1 = m; 452f1580f4eSBarry Smith if (P->ops->getinfo) nnz1 = info.nz_used; 453f1580f4eSBarry Smith PetscCall(MatDestroy(&P)); 454f1580f4eSBarry Smith } 455f1580f4eSBarry Smith } 456f1580f4eSBarry Smith } 457f1580f4eSBarry Smith *gc = static_cast<PetscReal>(accumulate[0] / m1); 458f1580f4eSBarry Smith *oc = static_cast<PetscReal>(accumulate[1] / nnz1); 4593ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 460f1580f4eSBarry Smith } 461f1580f4eSBarry Smith 462d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCView_HPDDM(PC pc, PetscViewer viewer) 463d71ae5a4SJacob Faibussowitsch { 464f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 465f1580f4eSBarry Smith PetscViewer subviewer; 466*aa21023fSPierre Jolivet PetscViewerFormat format; 467f1580f4eSBarry Smith PetscSubcomm subcomm; 468f1580f4eSBarry Smith PetscReal oc, gc; 469f1580f4eSBarry Smith PetscInt i, tabs; 470f1580f4eSBarry Smith PetscMPIInt size, color, rank; 471*aa21023fSPierre Jolivet PetscBool flg; 472*aa21023fSPierre Jolivet const char *name; 473f1580f4eSBarry Smith 474f1580f4eSBarry Smith PetscFunctionBegin; 475*aa21023fSPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &flg)); 476*aa21023fSPierre Jolivet if (flg) { 477f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "level%s: %" PetscInt_FMT "\n", data->N > 1 ? "s" : "", data->N)); 478f1580f4eSBarry Smith PetscCall(PCHPDDMGetComplexities(pc, &gc, &oc)); 479f1580f4eSBarry Smith if (data->N > 1) { 480f1580f4eSBarry Smith if (!data->deflation) { 481c8ea6600SPierre Jolivet PetscCall(PetscViewerASCIIPrintf(viewer, "Neumann matrix attached? %s\n", PetscBools[PetscBool3ToBool(data->Neumann)])); 482f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "shared subdomain KSP between SLEPc and PETSc? %s\n", PetscBools[data->share])); 483f1580f4eSBarry Smith } else PetscCall(PetscViewerASCIIPrintf(viewer, "user-supplied deflation matrix\n")); 484f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "coarse correction: %s\n", PCHPDDMCoarseCorrectionTypes[data->correction])); 485f1580f4eSBarry 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" : "")); 486f1580f4eSBarry Smith PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 487f1580f4eSBarry Smith PetscCall(PetscViewerASCIISetTab(viewer, 0)); 488f1580f4eSBarry Smith for (i = 1; i < data->N; ++i) { 489f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, " %" PetscInt_FMT, data->levels[i - 1]->nu)); 490f1580f4eSBarry Smith if (data->levels[i - 1]->threshold > -0.1) PetscCall(PetscViewerASCIIPrintf(viewer, " (%g)", (double)data->levels[i - 1]->threshold)); 491f1580f4eSBarry Smith } 492f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "\n")); 493f1580f4eSBarry Smith PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 494f1580f4eSBarry Smith } 495f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "grid and operator complexities: %g %g\n", (double)gc, (double)oc)); 496f1580f4eSBarry Smith if (data->levels[0]->ksp) { 497f1580f4eSBarry Smith PetscCall(KSPView(data->levels[0]->ksp, viewer)); 498f1580f4eSBarry Smith if (data->levels[0]->pc) PetscCall(PCView(data->levels[0]->pc, viewer)); 499f1580f4eSBarry Smith for (i = 1; i < data->N; ++i) { 500f1580f4eSBarry Smith if (data->levels[i]->ksp) color = 1; 501f1580f4eSBarry Smith else color = 0; 502f1580f4eSBarry Smith PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)pc), &size)); 503f1580f4eSBarry Smith PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)pc), &rank)); 504f1580f4eSBarry Smith PetscCall(PetscSubcommCreate(PetscObjectComm((PetscObject)pc), &subcomm)); 505f1580f4eSBarry Smith PetscCall(PetscSubcommSetNumber(subcomm, PetscMin(size, 2))); 506f1580f4eSBarry Smith PetscCall(PetscSubcommSetTypeGeneral(subcomm, color, rank)); 507f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPushTab(viewer)); 508f1580f4eSBarry Smith PetscCall(PetscViewerGetSubViewer(viewer, PetscSubcommChild(subcomm), &subviewer)); 509f1580f4eSBarry Smith if (color == 1) { 510f1580f4eSBarry Smith PetscCall(KSPView(data->levels[i]->ksp, subviewer)); 511f1580f4eSBarry Smith if (data->levels[i]->pc) PetscCall(PCView(data->levels[i]->pc, subviewer)); 512f1580f4eSBarry Smith PetscCall(PetscViewerFlush(subviewer)); 513f1580f4eSBarry Smith } 514f1580f4eSBarry Smith PetscCall(PetscViewerRestoreSubViewer(viewer, PetscSubcommChild(subcomm), &subviewer)); 515f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPopTab(viewer)); 516f1580f4eSBarry Smith PetscCall(PetscSubcommDestroy(&subcomm)); 517f1580f4eSBarry Smith PetscCall(PetscViewerFlush(viewer)); 518f1580f4eSBarry Smith } 519f1580f4eSBarry Smith } 520*aa21023fSPierre Jolivet PetscCall(PetscViewerGetFormat(viewer, &format)); 521*aa21023fSPierre Jolivet if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 522*aa21023fSPierre Jolivet PetscCall(PetscViewerFileGetName(viewer, &name)); 523*aa21023fSPierre Jolivet if (name) { 524*aa21023fSPierre Jolivet IS is; 525*aa21023fSPierre Jolivet PetscInt sizes[4] = {pc->mat->rmap->n, pc->mat->cmap->n, pc->mat->rmap->N, pc->mat->cmap->N}; 526*aa21023fSPierre Jolivet char *tmp; 527*aa21023fSPierre Jolivet std::string prefix, suffix; 528*aa21023fSPierre Jolivet size_t pos; 529*aa21023fSPierre Jolivet 530*aa21023fSPierre Jolivet PetscCall(PetscStrstr(name, ".", &tmp)); 531*aa21023fSPierre Jolivet if (tmp) { 532*aa21023fSPierre Jolivet pos = std::distance(const_cast<char *>(name), tmp); 533*aa21023fSPierre Jolivet prefix = std::string(name, pos); 534*aa21023fSPierre Jolivet suffix = std::string(name + pos + 1); 535*aa21023fSPierre Jolivet } else prefix = name; 536*aa21023fSPierre Jolivet if (data->aux) { 537*aa21023fSPierre Jolivet PetscCall(PetscViewerBinaryOpen(PETSC_COMM_SELF, std::string(prefix + "_aux_" + std::to_string(rank) + "_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer)); 538*aa21023fSPierre Jolivet PetscCall(MatView(data->aux, subviewer)); 539*aa21023fSPierre Jolivet PetscCall(PetscViewerDestroy(&subviewer)); 540*aa21023fSPierre Jolivet } 541*aa21023fSPierre Jolivet if (data->is) { 542*aa21023fSPierre Jolivet PetscCall(PetscViewerBinaryOpen(PETSC_COMM_SELF, std::string(prefix + "_is_" + std::to_string(rank) + "_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer)); 543*aa21023fSPierre Jolivet PetscCall(ISView(data->is, subviewer)); 544*aa21023fSPierre Jolivet PetscCall(PetscViewerDestroy(&subviewer)); 545*aa21023fSPierre Jolivet } 546*aa21023fSPierre Jolivet PetscCall(ISCreateGeneral(PETSC_COMM_SELF, PETSC_STATIC_ARRAY_LENGTH(sizes), sizes, PETSC_USE_POINTER, &is)); 547*aa21023fSPierre Jolivet PetscCall(PetscViewerBinaryOpen(PETSC_COMM_SELF, std::string(prefix + "_sizes_" + std::to_string(rank) + "_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer)); 548*aa21023fSPierre Jolivet PetscCall(ISView(is, subviewer)); 549*aa21023fSPierre Jolivet PetscCall(PetscViewerDestroy(&subviewer)); 550*aa21023fSPierre Jolivet PetscCall(ISDestroy(&is)); 551*aa21023fSPierre Jolivet } 552*aa21023fSPierre Jolivet } 553f1580f4eSBarry Smith } 5543ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 555f1580f4eSBarry Smith } 556f1580f4eSBarry Smith 557d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCPreSolve_HPDDM(PC pc, KSP ksp, Vec, Vec) 558d71ae5a4SJacob Faibussowitsch { 559f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 560f1580f4eSBarry Smith PetscBool flg; 561f1580f4eSBarry Smith Mat A; 562f1580f4eSBarry Smith 563f1580f4eSBarry Smith PetscFunctionBegin; 564f1580f4eSBarry Smith if (ksp) { 565f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)ksp, KSPLSQR, &flg)); 566f1580f4eSBarry Smith if (flg && !data->normal) { 567db4a47b3SPierre Jolivet PetscCall(KSPGetOperators(ksp, &A, nullptr)); 568db4a47b3SPierre Jolivet PetscCall(MatCreateVecs(A, nullptr, &data->normal)); /* temporary Vec used in PCApply_HPDDMShell() for coarse grid corrections */ 569f1580f4eSBarry Smith } 570f1580f4eSBarry Smith } 5713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 572f1580f4eSBarry Smith } 573f1580f4eSBarry Smith 574d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetUp_HPDDMShell(PC pc) 575d71ae5a4SJacob Faibussowitsch { 576f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 577f1580f4eSBarry Smith Mat A, P; 578f1580f4eSBarry Smith Vec x; 579f1580f4eSBarry Smith const char *pcpre; 580f1580f4eSBarry Smith 581f1580f4eSBarry Smith PetscFunctionBegin; 582f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 583f1580f4eSBarry Smith PetscCall(KSPGetOptionsPrefix(ctx->ksp, &pcpre)); 584f1580f4eSBarry Smith PetscCall(KSPGetOperators(ctx->ksp, &A, &P)); 585f1580f4eSBarry Smith /* smoother */ 586f1580f4eSBarry Smith PetscCall(PCSetOptionsPrefix(ctx->pc, pcpre)); 587f1580f4eSBarry Smith PetscCall(PCSetOperators(ctx->pc, A, P)); 588f1580f4eSBarry Smith if (!ctx->v[0]) { 589f1580f4eSBarry Smith PetscCall(VecDuplicateVecs(ctx->D, 1, &ctx->v[0])); 590f1580f4eSBarry Smith if (!std::is_same<PetscScalar, PetscReal>::value) PetscCall(VecDestroy(&ctx->D)); 591db4a47b3SPierre Jolivet PetscCall(MatCreateVecs(A, &x, nullptr)); 592f1580f4eSBarry Smith PetscCall(VecDuplicateVecs(x, 2, &ctx->v[1])); 593f1580f4eSBarry Smith PetscCall(VecDestroy(&x)); 594f1580f4eSBarry Smith } 595f1580f4eSBarry Smith std::fill_n(ctx->V, 3, nullptr); 5963ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 597f1580f4eSBarry Smith } 598f1580f4eSBarry Smith 599f1580f4eSBarry Smith template <class Type, typename std::enable_if<std::is_same<Type, Vec>::value>::type * = nullptr> 600d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMDeflate_Private(PC pc, Type x, Type y) 601d71ae5a4SJacob Faibussowitsch { 602f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 603f1580f4eSBarry Smith PetscScalar *out; 604f1580f4eSBarry Smith 605f1580f4eSBarry Smith PetscFunctionBegin; 606f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 607f1580f4eSBarry Smith /* going from PETSc to HPDDM numbering */ 608f1580f4eSBarry Smith PetscCall(VecScatterBegin(ctx->scatter, x, ctx->v[0][0], INSERT_VALUES, SCATTER_FORWARD)); 609f1580f4eSBarry Smith PetscCall(VecScatterEnd(ctx->scatter, x, ctx->v[0][0], INSERT_VALUES, SCATTER_FORWARD)); 610f1580f4eSBarry Smith PetscCall(VecGetArrayWrite(ctx->v[0][0], &out)); 611db4a47b3SPierre Jolivet ctx->P->deflation<false>(nullptr, out, 1); /* y = Q x */ 612f1580f4eSBarry Smith PetscCall(VecRestoreArrayWrite(ctx->v[0][0], &out)); 613f1580f4eSBarry Smith /* going from HPDDM to PETSc numbering */ 614f1580f4eSBarry Smith PetscCall(VecScatterBegin(ctx->scatter, ctx->v[0][0], y, INSERT_VALUES, SCATTER_REVERSE)); 615f1580f4eSBarry Smith PetscCall(VecScatterEnd(ctx->scatter, ctx->v[0][0], y, INSERT_VALUES, SCATTER_REVERSE)); 6163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 617f1580f4eSBarry Smith } 618f1580f4eSBarry Smith 619f1580f4eSBarry Smith template <class Type, typename std::enable_if<std::is_same<Type, Mat>::value>::type * = nullptr> 620d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMDeflate_Private(PC pc, Type X, Type Y) 621d71ae5a4SJacob Faibussowitsch { 622f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 623f1580f4eSBarry Smith Vec vX, vY, vC; 624f1580f4eSBarry Smith PetscScalar *out; 625f1580f4eSBarry Smith PetscInt i, N; 626f1580f4eSBarry Smith 627f1580f4eSBarry Smith PetscFunctionBegin; 628f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 629db4a47b3SPierre Jolivet PetscCall(MatGetSize(X, nullptr, &N)); 630f1580f4eSBarry Smith /* going from PETSc to HPDDM numbering */ 631f1580f4eSBarry Smith for (i = 0; i < N; ++i) { 632f1580f4eSBarry Smith PetscCall(MatDenseGetColumnVecRead(X, i, &vX)); 633f1580f4eSBarry Smith PetscCall(MatDenseGetColumnVecWrite(ctx->V[0], i, &vC)); 634f1580f4eSBarry Smith PetscCall(VecScatterBegin(ctx->scatter, vX, vC, INSERT_VALUES, SCATTER_FORWARD)); 635f1580f4eSBarry Smith PetscCall(VecScatterEnd(ctx->scatter, vX, vC, INSERT_VALUES, SCATTER_FORWARD)); 636f1580f4eSBarry Smith PetscCall(MatDenseRestoreColumnVecWrite(ctx->V[0], i, &vC)); 637f1580f4eSBarry Smith PetscCall(MatDenseRestoreColumnVecRead(X, i, &vX)); 638f1580f4eSBarry Smith } 639f1580f4eSBarry Smith PetscCall(MatDenseGetArrayWrite(ctx->V[0], &out)); 640db4a47b3SPierre Jolivet ctx->P->deflation<false>(nullptr, out, N); /* Y = Q X */ 641f1580f4eSBarry Smith PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &out)); 642f1580f4eSBarry Smith /* going from HPDDM to PETSc numbering */ 643f1580f4eSBarry Smith for (i = 0; i < N; ++i) { 644f1580f4eSBarry Smith PetscCall(MatDenseGetColumnVecRead(ctx->V[0], i, &vC)); 645f1580f4eSBarry Smith PetscCall(MatDenseGetColumnVecWrite(Y, i, &vY)); 646f1580f4eSBarry Smith PetscCall(VecScatterBegin(ctx->scatter, vC, vY, INSERT_VALUES, SCATTER_REVERSE)); 647f1580f4eSBarry Smith PetscCall(VecScatterEnd(ctx->scatter, vC, vY, INSERT_VALUES, SCATTER_REVERSE)); 648f1580f4eSBarry Smith PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &vY)); 649f1580f4eSBarry Smith PetscCall(MatDenseRestoreColumnVecRead(ctx->V[0], i, &vC)); 650f1580f4eSBarry Smith } 6513ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 652f1580f4eSBarry Smith } 653f1580f4eSBarry Smith 654f1580f4eSBarry Smith /* 655f1580f4eSBarry Smith PCApply_HPDDMShell - Applies a (2) deflated, (1) additive, or (3) balanced coarse correction. In what follows, E = Z Pmat Z^T and Q = Z^T E^-1 Z. 656f1580f4eSBarry Smith 657f1580f4eSBarry Smith .vb 658f1580f4eSBarry Smith (1) y = Pmat^-1 x + Q x, 659f1580f4eSBarry Smith (2) y = Pmat^-1 (I - Amat Q) x + Q x (default), 660f1580f4eSBarry Smith (3) y = (I - Q Amat^T) Pmat^-1 (I - Amat Q) x + Q x. 661f1580f4eSBarry Smith .ve 662f1580f4eSBarry Smith 663f1580f4eSBarry Smith Input Parameters: 664f1580f4eSBarry Smith + pc - preconditioner context 665f1580f4eSBarry Smith - x - input vector 666f1580f4eSBarry Smith 667f1580f4eSBarry Smith Output Parameter: 668f1580f4eSBarry Smith . y - output vector 669f1580f4eSBarry Smith 670f1580f4eSBarry Smith Notes: 671f1580f4eSBarry 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. 672f1580f4eSBarry 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. 673f1580f4eSBarry 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. 674f1580f4eSBarry Smith 675f1580f4eSBarry Smith Level: advanced 676f1580f4eSBarry Smith 677f1580f4eSBarry Smith Developer Note: 678f1580f4eSBarry 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 679f1580f4eSBarry Smith to trigger it. Likely the manual page is `PCHPDDM` 680f1580f4eSBarry Smith 681f1580f4eSBarry Smith .seealso: `PCHPDDM`, `PCHPDDMCoarseCorrectionType` 682f1580f4eSBarry Smith */ 683d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_HPDDMShell(PC pc, Vec x, Vec y) 684d71ae5a4SJacob Faibussowitsch { 685f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 686f1580f4eSBarry Smith Mat A; 687f1580f4eSBarry Smith 688f1580f4eSBarry Smith PetscFunctionBegin; 689f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 690f1580f4eSBarry Smith PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object"); 691db4a47b3SPierre Jolivet PetscCall(KSPGetOperators(ctx->ksp, &A, nullptr)); 692f1580f4eSBarry Smith PetscCall(PCHPDDMDeflate_Private(pc, x, y)); /* y = Q x */ 693f1580f4eSBarry Smith if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 694f1580f4eSBarry Smith if (!ctx->parent->normal || ctx != ctx->parent->levels[0]) PetscCall(MatMult(A, y, ctx->v[1][0])); /* y = A Q x */ 695f1580f4eSBarry Smith else { /* KSPLSQR and finest level */ PetscCall(MatMult(A, y, ctx->parent->normal)); /* y = A Q x */ 696f1580f4eSBarry Smith PetscCall(MatMultHermitianTranspose(A, ctx->parent->normal, ctx->v[1][0])); /* y = A^T A Q x */ 697f1580f4eSBarry Smith } 698f1580f4eSBarry Smith PetscCall(VecWAXPY(ctx->v[1][1], -1.0, ctx->v[1][0], x)); /* y = (I - A Q) x */ 699f1580f4eSBarry Smith PetscCall(PCApply(ctx->pc, ctx->v[1][1], ctx->v[1][0])); /* y = M^-1 (I - A Q) x */ 700f1580f4eSBarry Smith if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 701f1580f4eSBarry 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 */ 702f1580f4eSBarry Smith else { 703f1580f4eSBarry Smith PetscCall(MatMult(A, ctx->v[1][0], ctx->parent->normal)); 704f1580f4eSBarry Smith PetscCall(MatMultHermitianTranspose(A, ctx->parent->normal, ctx->v[1][1])); /* z = A^T A y */ 705f1580f4eSBarry Smith } 706f1580f4eSBarry Smith PetscCall(PCHPDDMDeflate_Private(pc, ctx->v[1][1], ctx->v[1][1])); 707f1580f4eSBarry 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 */ 708f1580f4eSBarry Smith } else PetscCall(VecAXPY(y, 1.0, ctx->v[1][0])); /* y = Q M^-1 (I - A Q) x + Q x */ 709f1580f4eSBarry Smith } else { 710f1580f4eSBarry 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); 711f1580f4eSBarry Smith PetscCall(PCApply(ctx->pc, x, ctx->v[1][0])); 712f1580f4eSBarry Smith PetscCall(VecAXPY(y, 1.0, ctx->v[1][0])); /* y = M^-1 x + Q x */ 713f1580f4eSBarry Smith } 7143ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 715f1580f4eSBarry Smith } 716f1580f4eSBarry Smith 717f1580f4eSBarry Smith /* 718f1580f4eSBarry Smith PCMatApply_HPDDMShell - Variant of PCApply_HPDDMShell() for blocks of vectors. 719f1580f4eSBarry Smith 720f1580f4eSBarry Smith Input Parameters: 721f1580f4eSBarry Smith + pc - preconditioner context 722f1580f4eSBarry Smith - X - block of input vectors 723f1580f4eSBarry Smith 724f1580f4eSBarry Smith Output Parameter: 725f1580f4eSBarry Smith . Y - block of output vectors 726f1580f4eSBarry Smith 727f1580f4eSBarry Smith Level: advanced 728f1580f4eSBarry Smith 729f1580f4eSBarry Smith .seealso: `PCHPDDM`, `PCApply_HPDDMShell()`, `PCHPDDMCoarseCorrectionType` 730f1580f4eSBarry Smith */ 731d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_HPDDMShell(PC pc, Mat X, Mat Y) 732d71ae5a4SJacob Faibussowitsch { 733f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 734f1580f4eSBarry Smith Mat A, *ptr; 735db4a47b3SPierre Jolivet PetscContainer container = nullptr; 736f1580f4eSBarry Smith PetscScalar *array; 737f1580f4eSBarry Smith PetscInt m, M, N, prev = 0; 738f1580f4eSBarry Smith PetscBool reset = PETSC_FALSE; 739f1580f4eSBarry Smith 740f1580f4eSBarry Smith PetscFunctionBegin; 741f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 742f1580f4eSBarry Smith PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object"); 743db4a47b3SPierre Jolivet PetscCall(MatGetSize(X, nullptr, &N)); 744db4a47b3SPierre Jolivet PetscCall(KSPGetOperators(ctx->ksp, &A, nullptr)); 745f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)A, "_HPDDM_MatProduct", (PetscObject *)&container)); 746f1580f4eSBarry Smith if (container) { /* MatProduct container already attached */ 747f1580f4eSBarry Smith PetscCall(PetscContainerGetPointer(container, (void **)&ptr)); 748f1580f4eSBarry Smith if (ptr[1] != ctx->V[2]) /* Mat has changed or may have been set first in KSPHPDDM */ 749f1580f4eSBarry Smith for (m = 0; m < 2; ++m) { 750f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V + m + 1)); 751f1580f4eSBarry Smith ctx->V[m + 1] = ptr[m]; 752f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)ctx->V[m + 1])); 753f1580f4eSBarry Smith } 754f1580f4eSBarry Smith } 755db4a47b3SPierre Jolivet if (ctx->V[1]) PetscCall(MatGetSize(ctx->V[1], nullptr, &prev)); 756f1580f4eSBarry Smith if (N != prev || !ctx->V[0]) { 757f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V)); 758f1580f4eSBarry Smith PetscCall(VecGetLocalSize(ctx->v[0][0], &m)); 759db4a47b3SPierre Jolivet PetscCall(MatCreateDense(PetscObjectComm((PetscObject)pc), m, PETSC_DECIDE, PETSC_DECIDE, N, nullptr, ctx->V)); 760f1580f4eSBarry Smith if (N != prev) { 761f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V + 1)); 762f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V + 2)); 763db4a47b3SPierre Jolivet PetscCall(MatGetLocalSize(X, &m, nullptr)); 764db4a47b3SPierre Jolivet PetscCall(MatGetSize(X, &M, nullptr)); 765f1580f4eSBarry Smith if (ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) PetscCall(MatDenseGetArrayWrite(ctx->V[0], &array)); 766db4a47b3SPierre Jolivet else array = nullptr; 767f1580f4eSBarry Smith PetscCall(MatCreateDense(PetscObjectComm((PetscObject)pc), m, PETSC_DECIDE, M, N, array, ctx->V + 1)); 768f1580f4eSBarry Smith if (ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &array)); 769db4a47b3SPierre Jolivet PetscCall(MatCreateDense(PetscObjectComm((PetscObject)pc), m, PETSC_DECIDE, M, N, nullptr, ctx->V + 2)); 770db4a47b3SPierre Jolivet PetscCall(MatProductCreateWithMat(A, Y, nullptr, ctx->V[1])); 771f1580f4eSBarry Smith PetscCall(MatProductSetType(ctx->V[1], MATPRODUCT_AB)); 772f1580f4eSBarry Smith PetscCall(MatProductSetFromOptions(ctx->V[1])); 773f1580f4eSBarry Smith PetscCall(MatProductSymbolic(ctx->V[1])); 774f1580f4eSBarry Smith if (!container) { /* no MatProduct container attached, create one to be queried in KSPHPDDM or at the next call to PCMatApply() */ 775f1580f4eSBarry Smith PetscCall(PetscContainerCreate(PetscObjectComm((PetscObject)A), &container)); 776f1580f4eSBarry Smith PetscCall(PetscObjectCompose((PetscObject)A, "_HPDDM_MatProduct", (PetscObject)container)); 777f1580f4eSBarry Smith } 778f1580f4eSBarry Smith PetscCall(PetscContainerSetPointer(container, ctx->V + 1)); /* need to compose B and D from MatProductCreateWithMath(A, B, NULL, D), which are stored in the contiguous array ctx->V */ 779f1580f4eSBarry Smith } 780f1580f4eSBarry Smith if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 781db4a47b3SPierre Jolivet PetscCall(MatProductCreateWithMat(A, ctx->V[1], nullptr, ctx->V[2])); 782f1580f4eSBarry Smith PetscCall(MatProductSetType(ctx->V[2], MATPRODUCT_AtB)); 783f1580f4eSBarry Smith PetscCall(MatProductSetFromOptions(ctx->V[2])); 784f1580f4eSBarry Smith PetscCall(MatProductSymbolic(ctx->V[2])); 785f1580f4eSBarry Smith } 786f1580f4eSBarry Smith ctx->P->start(N); 787f1580f4eSBarry Smith } 788f1580f4eSBarry Smith if (N == prev || container) { /* when MatProduct container is attached, always need to MatProductReplaceMats() since KSPHPDDM may have replaced the Mat as well */ 789db4a47b3SPierre Jolivet PetscCall(MatProductReplaceMats(nullptr, Y, nullptr, ctx->V[1])); 790f1580f4eSBarry Smith if (container && ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) { 791f1580f4eSBarry Smith PetscCall(MatDenseGetArrayWrite(ctx->V[0], &array)); 792f1580f4eSBarry Smith PetscCall(MatDensePlaceArray(ctx->V[1], array)); 793f1580f4eSBarry Smith PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &array)); 794f1580f4eSBarry Smith reset = PETSC_TRUE; 795f1580f4eSBarry Smith } 796f1580f4eSBarry Smith } 797f1580f4eSBarry Smith PetscCall(PCHPDDMDeflate_Private(pc, X, Y)); 798f1580f4eSBarry Smith if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 799f1580f4eSBarry Smith PetscCall(MatProductNumeric(ctx->V[1])); 800f1580f4eSBarry Smith PetscCall(MatCopy(ctx->V[1], ctx->V[2], SAME_NONZERO_PATTERN)); 801f1580f4eSBarry Smith PetscCall(MatAXPY(ctx->V[2], -1.0, X, SAME_NONZERO_PATTERN)); 802f1580f4eSBarry Smith PetscCall(PCMatApply(ctx->pc, ctx->V[2], ctx->V[1])); 803f1580f4eSBarry Smith if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 804f1580f4eSBarry Smith PetscCall(MatProductNumeric(ctx->V[2])); 805f1580f4eSBarry Smith PetscCall(PCHPDDMDeflate_Private(pc, ctx->V[2], ctx->V[2])); 806f1580f4eSBarry Smith PetscCall(MatAXPY(ctx->V[1], -1.0, ctx->V[2], SAME_NONZERO_PATTERN)); 807f1580f4eSBarry Smith } 808f1580f4eSBarry Smith PetscCall(MatAXPY(Y, -1.0, ctx->V[1], SAME_NONZERO_PATTERN)); 809f1580f4eSBarry Smith } else { 810f1580f4eSBarry 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); 811f1580f4eSBarry Smith PetscCall(PCMatApply(ctx->pc, X, ctx->V[1])); 812f1580f4eSBarry Smith PetscCall(MatAXPY(Y, 1.0, ctx->V[1], SAME_NONZERO_PATTERN)); 813f1580f4eSBarry Smith } 814f1580f4eSBarry Smith if (reset) PetscCall(MatDenseResetArray(ctx->V[1])); 8153ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 816f1580f4eSBarry Smith } 817f1580f4eSBarry Smith 818d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCDestroy_HPDDMShell(PC pc) 819d71ae5a4SJacob Faibussowitsch { 820f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 821f1580f4eSBarry Smith PetscContainer container; 822f1580f4eSBarry Smith 823f1580f4eSBarry Smith PetscFunctionBegin; 824f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 825f1580f4eSBarry Smith PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(ctx, PETSC_TRUE)); 826f1580f4eSBarry Smith PetscCall(VecDestroyVecs(1, &ctx->v[0])); 827f1580f4eSBarry Smith PetscCall(VecDestroyVecs(2, &ctx->v[1])); 828f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)(ctx->pc)->mat, "_HPDDM_MatProduct", (PetscObject *)&container)); 829f1580f4eSBarry Smith PetscCall(PetscContainerDestroy(&container)); 830db4a47b3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)(ctx->pc)->mat, "_HPDDM_MatProduct", nullptr)); 831f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V)); 832f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V + 1)); 833f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V + 2)); 834f1580f4eSBarry Smith PetscCall(VecDestroy(&ctx->D)); 835f1580f4eSBarry Smith PetscCall(VecScatterDestroy(&ctx->scatter)); 836f1580f4eSBarry Smith PetscCall(PCDestroy(&ctx->pc)); 8373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 838f1580f4eSBarry Smith } 839f1580f4eSBarry Smith 840d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSolve_Private(const PC_HPDDM_Level *ctx, PetscScalar *rhs, const unsigned short &mu) 841d71ae5a4SJacob Faibussowitsch { 842f1580f4eSBarry Smith Mat B, X; 843f1580f4eSBarry Smith PetscInt n, N, j = 0; 844f1580f4eSBarry Smith 845f1580f4eSBarry Smith PetscFunctionBegin; 846db4a47b3SPierre Jolivet PetscCall(KSPGetOperators(ctx->ksp, &B, nullptr)); 847db4a47b3SPierre Jolivet PetscCall(MatGetLocalSize(B, &n, nullptr)); 848db4a47b3SPierre Jolivet PetscCall(MatGetSize(B, &N, nullptr)); 849f1580f4eSBarry Smith if (ctx->parent->log_separate) { 850f1580f4eSBarry Smith j = std::distance(ctx->parent->levels, std::find(ctx->parent->levels, ctx->parent->levels + ctx->parent->N, ctx)); 851db4a47b3SPierre Jolivet PetscCall(PetscLogEventBegin(PC_HPDDM_Solve[j], ctx->ksp, nullptr, nullptr, nullptr)); 852f1580f4eSBarry Smith } 853f1580f4eSBarry Smith if (mu == 1) { 854f1580f4eSBarry Smith if (!ctx->ksp->vec_rhs) { 855db4a47b3SPierre Jolivet PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)ctx->ksp), 1, n, N, nullptr, &ctx->ksp->vec_rhs)); 856f1580f4eSBarry Smith PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)ctx->ksp), n, N, &ctx->ksp->vec_sol)); 857f1580f4eSBarry Smith } 858f1580f4eSBarry Smith PetscCall(VecPlaceArray(ctx->ksp->vec_rhs, rhs)); 859db4a47b3SPierre Jolivet PetscCall(KSPSolve(ctx->ksp, nullptr, nullptr)); 860f1580f4eSBarry Smith PetscCall(VecCopy(ctx->ksp->vec_sol, ctx->ksp->vec_rhs)); 861f1580f4eSBarry Smith PetscCall(VecResetArray(ctx->ksp->vec_rhs)); 862f1580f4eSBarry Smith } else { 863f1580f4eSBarry Smith PetscCall(MatCreateDense(PetscObjectComm((PetscObject)ctx->ksp), n, PETSC_DECIDE, N, mu, rhs, &B)); 864db4a47b3SPierre Jolivet PetscCall(MatCreateDense(PetscObjectComm((PetscObject)ctx->ksp), n, PETSC_DECIDE, N, mu, nullptr, &X)); 865f1580f4eSBarry Smith PetscCall(KSPMatSolve(ctx->ksp, B, X)); 866f1580f4eSBarry Smith PetscCall(MatCopy(X, B, SAME_NONZERO_PATTERN)); 867f1580f4eSBarry Smith PetscCall(MatDestroy(&X)); 868f1580f4eSBarry Smith PetscCall(MatDestroy(&B)); 869f1580f4eSBarry Smith } 870db4a47b3SPierre Jolivet if (ctx->parent->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Solve[j], ctx->ksp, nullptr, nullptr, nullptr)); 8713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 872f1580f4eSBarry Smith } 873f1580f4eSBarry Smith 874d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetUpNeumannOverlap_Private(PC pc) 875d71ae5a4SJacob Faibussowitsch { 876f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 877f1580f4eSBarry Smith 878f1580f4eSBarry Smith PetscFunctionBegin; 879f1580f4eSBarry Smith if (data->setup) { 880f1580f4eSBarry Smith Mat P; 881db4a47b3SPierre Jolivet Vec x, xt = nullptr; 882f1580f4eSBarry Smith PetscReal t = 0.0, s = 0.0; 883f1580f4eSBarry Smith 884db4a47b3SPierre Jolivet PetscCall(PCGetOperators(pc, nullptr, &P)); 885f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)P, "__SNES_latest_X", (PetscObject *)&x)); 886f1580f4eSBarry Smith PetscCallBack("PCHPDDM Neumann callback", (*data->setup)(data->aux, t, x, xt, s, data->is, data->setup_ctx)); 887f1580f4eSBarry Smith } 8883ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 889f1580f4eSBarry Smith } 890f1580f4eSBarry Smith 891d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMCreateSubMatrices_Private(Mat mat, PetscInt n, const IS *, const IS *, MatReuse scall, Mat *submat[]) 892d71ae5a4SJacob Faibussowitsch { 893f1580f4eSBarry Smith Mat A; 894f1580f4eSBarry Smith 895f1580f4eSBarry Smith PetscFunctionBegin; 896f1580f4eSBarry Smith PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "MatCreateSubMatrices() called to extract %" PetscInt_FMT " submatrices, which is different than 1", n); 897f1580f4eSBarry Smith /* previously composed Mat */ 898f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)mat, "_PCHPDDM_SubMatrices", (PetscObject *)&A)); 899f1580f4eSBarry Smith PetscCheck(A, PETSC_COMM_SELF, PETSC_ERR_PLIB, "SubMatrices not found in Mat"); 900f1580f4eSBarry Smith if (scall == MAT_INITIAL_MATRIX) { 9016f2c871aSStefano Zampini PetscCall(PetscCalloc1(2, submat)); /* allocate an extra Mat to avoid errors in MatDestroySubMatrices_Dummy() */ 902f1580f4eSBarry Smith PetscCall(MatDuplicate(A, MAT_COPY_VALUES, *submat)); 903f1580f4eSBarry Smith } else PetscCall(MatCopy(A, (*submat)[0], SAME_NONZERO_PATTERN)); 9043ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 905f1580f4eSBarry Smith } 906f1580f4eSBarry Smith 907d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMCommunicationAvoidingPCASM_Private(PC pc, Mat C, PetscBool sorted) 908d71ae5a4SJacob Faibussowitsch { 909f1580f4eSBarry Smith void (*op)(void); 910f1580f4eSBarry Smith 911f1580f4eSBarry Smith PetscFunctionBegin; 912f1580f4eSBarry Smith /* previously-composed Mat */ 913f1580f4eSBarry Smith PetscCall(PetscObjectCompose((PetscObject)pc->pmat, "_PCHPDDM_SubMatrices", (PetscObject)C)); 914f1580f4eSBarry Smith PetscCall(MatGetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, &op)); 915f1580f4eSBarry Smith /* trick suggested by Barry https://lists.mcs.anl.gov/pipermail/petsc-dev/2020-January/025491.html */ 916f1580f4eSBarry Smith PetscCall(MatSetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, (void (*)(void))PCHPDDMCreateSubMatrices_Private)); 917f1580f4eSBarry Smith if (sorted) PetscCall(PCASMSetSortIndices(pc, PETSC_FALSE)); /* everything is already sorted */ 918f1580f4eSBarry Smith PetscCall(PCSetFromOptions(pc)); /* otherwise -pc_hpddm_levels_1_pc_asm_sub_mat_type is not used */ 919f1580f4eSBarry Smith PetscCall(PCSetUp(pc)); 920f1580f4eSBarry Smith /* reset MatCreateSubMatrices() */ 921f1580f4eSBarry Smith PetscCall(MatSetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, op)); 922db4a47b3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)pc->pmat, "_PCHPDDM_SubMatrices", nullptr)); 9233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 924f1580f4eSBarry Smith } 925f1580f4eSBarry Smith 926d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMPermute_Private(IS is, IS in_is, IS *out_is, Mat in_C, Mat *out_C, IS *p) 927d71ae5a4SJacob Faibussowitsch { 928f1580f4eSBarry Smith IS perm; 929f1580f4eSBarry Smith const PetscInt *ptr; 930b07dfdedSPierre Jolivet PetscInt *concatenate, size, n, bs; 931f1580f4eSBarry Smith std::map<PetscInt, PetscInt> order; 932f1580f4eSBarry Smith PetscBool sorted; 933f1580f4eSBarry Smith 934f1580f4eSBarry Smith PetscFunctionBegin; 935f1580f4eSBarry Smith PetscCall(ISSorted(is, &sorted)); 936f1580f4eSBarry Smith if (!sorted) { 937f1580f4eSBarry Smith PetscCall(ISGetLocalSize(is, &size)); 938f1580f4eSBarry Smith PetscCall(ISGetIndices(is, &ptr)); 939b07dfdedSPierre Jolivet PetscCall(ISGetBlockSize(is, &bs)); 940f1580f4eSBarry Smith /* MatCreateSubMatrices(), called by PCASM, follows the global numbering of Pmat */ 941b07dfdedSPierre Jolivet for (n = 0; n < size; n += bs) order.insert(std::make_pair(ptr[n] / bs, n / bs)); 942f1580f4eSBarry Smith PetscCall(ISRestoreIndices(is, &ptr)); 943b07dfdedSPierre Jolivet size /= bs; 944f1580f4eSBarry Smith if (out_C) { 945f1580f4eSBarry Smith PetscCall(PetscMalloc1(size, &concatenate)); 946f1580f4eSBarry Smith for (const std::pair<const PetscInt, PetscInt> &i : order) *concatenate++ = i.second; 947f1580f4eSBarry Smith concatenate -= size; 948b07dfdedSPierre Jolivet PetscCall(ISCreateBlock(PetscObjectComm((PetscObject)in_C), bs, size, concatenate, PETSC_OWN_POINTER, &perm)); 949f1580f4eSBarry Smith PetscCall(ISSetPermutation(perm)); 950f1580f4eSBarry Smith /* permute user-provided Mat so that it matches with MatCreateSubMatrices() numbering */ 951f1580f4eSBarry Smith PetscCall(MatPermute(in_C, perm, perm, out_C)); 952f1580f4eSBarry Smith if (p) *p = perm; 953f1580f4eSBarry Smith else PetscCall(ISDestroy(&perm)); /* no need to save the permutation */ 954f1580f4eSBarry Smith } 955f1580f4eSBarry Smith if (out_is) { 956f1580f4eSBarry Smith PetscCall(PetscMalloc1(size, &concatenate)); 957f1580f4eSBarry Smith for (const std::pair<const PetscInt, PetscInt> &i : order) *concatenate++ = i.first; 958f1580f4eSBarry Smith concatenate -= size; 959f1580f4eSBarry Smith /* permute user-provided IS so that it matches with MatCreateSubMatrices() numbering */ 960b07dfdedSPierre Jolivet PetscCall(ISCreateBlock(PetscObjectComm((PetscObject)in_is), bs, size, concatenate, PETSC_OWN_POINTER, out_is)); 961f1580f4eSBarry Smith } 962f1580f4eSBarry Smith } else { /* input IS is sorted, nothing to permute, simply duplicate inputs when needed */ 963f1580f4eSBarry Smith if (out_C) PetscCall(MatDuplicate(in_C, MAT_COPY_VALUES, out_C)); 964f1580f4eSBarry Smith if (out_is) PetscCall(ISDuplicate(in_is, out_is)); 965f1580f4eSBarry Smith } 9663ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 967f1580f4eSBarry Smith } 968f1580f4eSBarry Smith 969d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMDestroySubMatrices_Private(PetscBool flg, PetscBool algebraic, Mat *sub) 970d71ae5a4SJacob Faibussowitsch { 971f1580f4eSBarry Smith IS is; 972f1580f4eSBarry Smith 973f1580f4eSBarry Smith PetscFunctionBegin; 974f1580f4eSBarry Smith if (!flg) { 975f1580f4eSBarry Smith if (algebraic) { 976f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)sub[0], "_PCHPDDM_Embed", (PetscObject *)&is)); 977f1580f4eSBarry Smith PetscCall(ISDestroy(&is)); 978db4a47b3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Embed", nullptr)); 979db4a47b3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Compact", nullptr)); 980f1580f4eSBarry Smith } 981f1580f4eSBarry Smith PetscCall(MatDestroySubMatrices(algebraic ? 2 : 1, &sub)); 982f1580f4eSBarry Smith } 9833ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 984f1580f4eSBarry Smith } 985f1580f4eSBarry Smith 986d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMAlgebraicAuxiliaryMat_Private(Mat P, IS *is, Mat *sub[], PetscBool block) 987d71ae5a4SJacob Faibussowitsch { 988f1580f4eSBarry Smith IS icol[3], irow[2]; 989f1580f4eSBarry Smith Mat *M, Q; 990f1580f4eSBarry Smith PetscReal *ptr; 991f1580f4eSBarry Smith PetscInt *idx, p = 0, n, bs = PetscAbs(P->cmap->bs); 992f1580f4eSBarry Smith PetscBool flg; 993f1580f4eSBarry Smith 994f1580f4eSBarry Smith PetscFunctionBegin; 995f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, P->cmap->N, 0, 1, icol + 2)); 996f1580f4eSBarry Smith PetscCall(ISSetBlockSize(icol[2], bs)); 997f1580f4eSBarry Smith PetscCall(ISSetIdentity(icol[2])); 998f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)P, MATMPISBAIJ, &flg)); 999f1580f4eSBarry Smith if (flg) { 1000f1580f4eSBarry Smith /* MatCreateSubMatrices() does not handle MATMPISBAIJ properly when iscol != isrow, so convert first to MATMPIBAIJ */ 1001f1580f4eSBarry Smith PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &Q)); 1002f1580f4eSBarry Smith std::swap(P, Q); 1003f1580f4eSBarry Smith } 1004f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(P, 1, is, icol + 2, MAT_INITIAL_MATRIX, &M)); 1005f1580f4eSBarry Smith if (flg) { 1006f1580f4eSBarry Smith std::swap(P, Q); 1007f1580f4eSBarry Smith PetscCall(MatDestroy(&Q)); 1008f1580f4eSBarry Smith } 1009f1580f4eSBarry Smith PetscCall(ISDestroy(icol + 2)); 1010f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, M[0]->rmap->N, 0, 1, irow)); 1011f1580f4eSBarry Smith PetscCall(ISSetBlockSize(irow[0], bs)); 1012f1580f4eSBarry Smith PetscCall(ISSetIdentity(irow[0])); 1013f1580f4eSBarry Smith if (!block) { 1014b07dfdedSPierre Jolivet PetscCall(PetscMalloc2(P->cmap->N, &ptr, P->cmap->N / bs, &idx)); 1015f1580f4eSBarry Smith PetscCall(MatGetColumnNorms(M[0], NORM_INFINITY, ptr)); 1016f1580f4eSBarry Smith /* check for nonzero columns so that M[0] may be expressed in compact form */ 1017b07dfdedSPierre Jolivet for (n = 0; n < P->cmap->N; n += bs) { 1018b07dfdedSPierre Jolivet if (std::find_if(ptr + n, ptr + n + bs, [](PetscReal v) { return v > PETSC_MACHINE_EPSILON; }) != ptr + n + bs) idx[p++] = n / bs; 1019f1580f4eSBarry Smith } 1020b07dfdedSPierre Jolivet PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, p, idx, PETSC_USE_POINTER, icol + 1)); 1021f1580f4eSBarry Smith PetscCall(ISSetInfo(icol[1], IS_SORTED, IS_GLOBAL, PETSC_TRUE, PETSC_TRUE)); 1022f1580f4eSBarry Smith PetscCall(ISEmbed(*is, icol[1], PETSC_FALSE, icol + 2)); 1023f1580f4eSBarry Smith irow[1] = irow[0]; 1024f1580f4eSBarry 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 */ 1025f1580f4eSBarry Smith icol[0] = is[0]; 1026f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(M[0], 2, irow, icol, MAT_INITIAL_MATRIX, sub)); 1027f1580f4eSBarry Smith PetscCall(ISDestroy(icol + 1)); 1028f1580f4eSBarry Smith PetscCall(PetscFree2(ptr, idx)); 1029f1580f4eSBarry Smith /* IS used to go back and forth between the augmented and the original local linear system, see eq. (3.4) of [2022b] */ 1030f1580f4eSBarry Smith PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Embed", (PetscObject)icol[2])); 1031f1580f4eSBarry Smith /* Mat used in eq. (3.1) of [2022b] */ 1032f1580f4eSBarry Smith PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Compact", (PetscObject)(*sub)[1])); 1033f1580f4eSBarry Smith } else { 1034f1580f4eSBarry Smith Mat aux; 1035f1580f4eSBarry Smith PetscCall(MatSetOption(M[0], MAT_SUBMAT_SINGLEIS, PETSC_TRUE)); 1036f1580f4eSBarry Smith /* diagonal block of the overlapping rows */ 1037f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(M[0], 1, irow, is, MAT_INITIAL_MATRIX, sub)); 1038f1580f4eSBarry Smith PetscCall(MatDuplicate((*sub)[0], MAT_COPY_VALUES, &aux)); 1039f1580f4eSBarry Smith PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE)); 1040f1580f4eSBarry Smith if (bs == 1) { /* scalar case */ 1041f1580f4eSBarry Smith Vec sum[2]; 1042f1580f4eSBarry Smith PetscCall(MatCreateVecs(aux, sum, sum + 1)); 1043f1580f4eSBarry Smith PetscCall(MatGetRowSum(M[0], sum[0])); 1044f1580f4eSBarry Smith PetscCall(MatGetRowSum(aux, sum[1])); 1045f1580f4eSBarry Smith /* off-diagonal block row sum (full rows - diagonal block rows) */ 1046f1580f4eSBarry Smith PetscCall(VecAXPY(sum[0], -1.0, sum[1])); 1047f1580f4eSBarry Smith /* subdomain matrix plus off-diagonal block row sum */ 1048f1580f4eSBarry Smith PetscCall(MatDiagonalSet(aux, sum[0], ADD_VALUES)); 1049f1580f4eSBarry Smith PetscCall(VecDestroy(sum)); 1050f1580f4eSBarry Smith PetscCall(VecDestroy(sum + 1)); 1051f1580f4eSBarry Smith } else { /* vectorial case */ 1052f1580f4eSBarry Smith /* TODO: missing MatGetValuesBlocked(), so the code below is */ 1053f1580f4eSBarry Smith /* an extension of the scalar case for when bs > 1, but it could */ 1054f1580f4eSBarry Smith /* be more efficient by avoiding all these MatMatMult() */ 1055f1580f4eSBarry Smith Mat sum[2], ones; 1056f1580f4eSBarry Smith PetscScalar *ptr; 1057f1580f4eSBarry Smith PetscCall(PetscCalloc1(M[0]->cmap->n * bs, &ptr)); 1058f1580f4eSBarry Smith PetscCall(MatCreateDense(PETSC_COMM_SELF, M[0]->cmap->n, bs, M[0]->cmap->n, bs, ptr, &ones)); 1059f1580f4eSBarry Smith for (n = 0; n < M[0]->cmap->n; n += bs) { 1060f1580f4eSBarry Smith for (p = 0; p < bs; ++p) ptr[n + p * (M[0]->cmap->n + 1)] = 1.0; 1061f1580f4eSBarry Smith } 1062f1580f4eSBarry Smith PetscCall(MatMatMult(M[0], ones, MAT_INITIAL_MATRIX, PETSC_DEFAULT, sum)); 1063f1580f4eSBarry Smith PetscCall(MatDestroy(&ones)); 1064f1580f4eSBarry Smith PetscCall(MatCreateDense(PETSC_COMM_SELF, aux->cmap->n, bs, aux->cmap->n, bs, ptr, &ones)); 1065f1580f4eSBarry Smith PetscCall(MatDenseSetLDA(ones, M[0]->cmap->n)); 1066f1580f4eSBarry Smith PetscCall(MatMatMult(aux, ones, MAT_INITIAL_MATRIX, PETSC_DEFAULT, sum + 1)); 1067f1580f4eSBarry Smith PetscCall(MatDestroy(&ones)); 1068f1580f4eSBarry Smith PetscCall(PetscFree(ptr)); 1069f1580f4eSBarry Smith /* off-diagonal block row sum (full rows - diagonal block rows) */ 1070f1580f4eSBarry Smith PetscCall(MatAXPY(sum[0], -1.0, sum[1], SAME_NONZERO_PATTERN)); 1071f1580f4eSBarry Smith PetscCall(MatDestroy(sum + 1)); 1072f1580f4eSBarry Smith /* re-order values to be consistent with MatSetValuesBlocked() */ 1073f1580f4eSBarry Smith /* equivalent to MatTranspose() which does not truly handle */ 1074f1580f4eSBarry Smith /* MAT_INPLACE_MATRIX in the rectangular case, as it calls PetscMalloc() */ 1075f1580f4eSBarry Smith PetscCall(MatDenseGetArrayWrite(sum[0], &ptr)); 1076f1580f4eSBarry Smith HPDDM::Wrapper<PetscScalar>::imatcopy<'T'>(bs, sum[0]->rmap->n, ptr, sum[0]->rmap->n, bs); 1077f1580f4eSBarry Smith /* subdomain matrix plus off-diagonal block row sum */ 1078f1580f4eSBarry Smith for (n = 0; n < aux->cmap->n / bs; ++n) PetscCall(MatSetValuesBlocked(aux, 1, &n, 1, &n, ptr + n * bs * bs, ADD_VALUES)); 1079f1580f4eSBarry Smith PetscCall(MatAssemblyBegin(aux, MAT_FINAL_ASSEMBLY)); 1080f1580f4eSBarry Smith PetscCall(MatAssemblyEnd(aux, MAT_FINAL_ASSEMBLY)); 1081f1580f4eSBarry Smith PetscCall(MatDenseRestoreArrayWrite(sum[0], &ptr)); 1082f1580f4eSBarry Smith PetscCall(MatDestroy(sum)); 1083f1580f4eSBarry Smith } 1084f1580f4eSBarry Smith PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_TRUE)); 1085f1580f4eSBarry Smith /* left-hand side of GenEO, with the same sparsity pattern as PCASM subdomain solvers */ 1086f1580f4eSBarry Smith PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Neumann_Mat", (PetscObject)aux)); 1087f1580f4eSBarry Smith } 1088f1580f4eSBarry Smith PetscCall(ISDestroy(irow)); 1089f1580f4eSBarry Smith PetscCall(MatDestroySubMatrices(1, &M)); 10903ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1091f1580f4eSBarry Smith } 1092f1580f4eSBarry Smith 1093d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetUp_HPDDM(PC pc) 1094d71ae5a4SJacob Faibussowitsch { 1095f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 1096f1580f4eSBarry Smith PC inner; 1097f1580f4eSBarry Smith KSP *ksp; 1098db4a47b3SPierre Jolivet Mat *sub, A, P, N, C = nullptr, uaux = nullptr, weighted, subA[2]; 1099f1580f4eSBarry Smith Vec xin, v; 1100f1580f4eSBarry Smith std::vector<Vec> initial; 1101db4a47b3SPierre Jolivet IS is[1], loc, uis = data->is, unsorted = nullptr; 1102f1580f4eSBarry Smith ISLocalToGlobalMapping l2g; 1103f1580f4eSBarry Smith char prefix[256]; 1104f1580f4eSBarry Smith const char *pcpre; 1105f1580f4eSBarry Smith const PetscScalar *const *ev; 1106f1580f4eSBarry Smith PetscInt n, requested = data->N, reused = 0; 1107f1580f4eSBarry Smith MatStructure structure = UNKNOWN_NONZERO_PATTERN; 1108f1580f4eSBarry Smith PetscBool subdomains = PETSC_FALSE, flg = PETSC_FALSE, ismatis, swap = PETSC_FALSE, algebraic = PETSC_FALSE, block = PETSC_FALSE; 1109f1580f4eSBarry Smith DM dm; 1110398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG) 1111db4a47b3SPierre Jolivet IS dis = nullptr; 1112db4a47b3SPierre Jolivet Mat daux = nullptr; 1113398c7888SPierre Jolivet #endif 1114f1580f4eSBarry Smith 1115f1580f4eSBarry Smith PetscFunctionBegin; 1116f1580f4eSBarry Smith PetscCheck(data->levels && data->levels[0], PETSC_COMM_SELF, PETSC_ERR_PLIB, "Not a single level allocated"); 1117f1580f4eSBarry Smith PetscCall(PCGetOptionsPrefix(pc, &pcpre)); 1118f1580f4eSBarry Smith PetscCall(PCGetOperators(pc, &A, &P)); 1119f1580f4eSBarry Smith if (!data->levels[0]->ksp) { 1120f1580f4eSBarry Smith PetscCall(KSPCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->ksp)); 11213821be0aSBarry Smith PetscCall(KSPSetNestLevel(data->levels[0]->ksp, pc->kspnestlevel)); 1122f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_%s_", pcpre ? pcpre : "", data->N > 1 ? "levels_1" : "coarse")); 1123f1580f4eSBarry Smith PetscCall(KSPSetOptionsPrefix(data->levels[0]->ksp, prefix)); 1124f1580f4eSBarry Smith PetscCall(KSPSetType(data->levels[0]->ksp, KSPPREONLY)); 1125f1580f4eSBarry Smith } else if (data->levels[0]->ksp->pc && data->levels[0]->ksp->pc->setupcalled == 1 && data->levels[0]->ksp->pc->reusepreconditioner) { 1126f1580f4eSBarry Smith /* if the fine-level PCSHELL exists, its setup has succeeded, and one wants to reuse it, */ 1127f1580f4eSBarry Smith /* then just propagate the appropriate flag to the coarser levels */ 1128f1580f4eSBarry Smith for (n = 0; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) { 1129f1580f4eSBarry Smith /* the following KSP and PC may be NULL for some processes, hence the check */ 1130f1580f4eSBarry Smith if (data->levels[n]->ksp) PetscCall(KSPSetReusePreconditioner(data->levels[n]->ksp, PETSC_TRUE)); 1131f1580f4eSBarry Smith if (data->levels[n]->pc) PetscCall(PCSetReusePreconditioner(data->levels[n]->pc, PETSC_TRUE)); 1132f1580f4eSBarry Smith } 1133f1580f4eSBarry Smith /* early bail out because there is nothing to do */ 11343ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1135f1580f4eSBarry Smith } else { 1136f1580f4eSBarry Smith /* reset coarser levels */ 1137f1580f4eSBarry Smith for (n = 1; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) { 1138f1580f4eSBarry Smith if (data->levels[n]->ksp && data->levels[n]->ksp->pc && data->levels[n]->ksp->pc->setupcalled == 1 && data->levels[n]->ksp->pc->reusepreconditioner && n < data->N) { 1139f1580f4eSBarry Smith reused = data->N - n; 1140f1580f4eSBarry Smith break; 1141f1580f4eSBarry Smith } 1142f1580f4eSBarry Smith PetscCall(KSPDestroy(&data->levels[n]->ksp)); 1143f1580f4eSBarry Smith PetscCall(PCDestroy(&data->levels[n]->pc)); 1144f1580f4eSBarry Smith } 1145f1580f4eSBarry Smith /* check if some coarser levels are being reused */ 1146f1580f4eSBarry Smith PetscCall(MPIU_Allreduce(MPI_IN_PLACE, &reused, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)pc))); 1147f1580f4eSBarry Smith const int *addr = data->levels[0]->P ? data->levels[0]->P->getAddrLocal() : &HPDDM::i__0; 1148f1580f4eSBarry Smith 1149f1580f4eSBarry Smith if (addr != &HPDDM::i__0 && reused != data->N - 1) { 1150f1580f4eSBarry Smith /* reuse previously computed eigenvectors */ 1151f1580f4eSBarry Smith ev = data->levels[0]->P->getVectors(); 1152f1580f4eSBarry Smith if (ev) { 1153f1580f4eSBarry Smith initial.reserve(*addr); 1154f1580f4eSBarry Smith PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, data->levels[0]->P->getDof(), ev[0], &xin)); 1155f1580f4eSBarry Smith for (n = 0; n < *addr; ++n) { 1156f1580f4eSBarry Smith PetscCall(VecDuplicate(xin, &v)); 1157f1580f4eSBarry Smith PetscCall(VecPlaceArray(xin, ev[n])); 1158f1580f4eSBarry Smith PetscCall(VecCopy(xin, v)); 1159f1580f4eSBarry Smith initial.emplace_back(v); 1160f1580f4eSBarry Smith PetscCall(VecResetArray(xin)); 1161f1580f4eSBarry Smith } 1162f1580f4eSBarry Smith PetscCall(VecDestroy(&xin)); 1163f1580f4eSBarry Smith } 1164f1580f4eSBarry Smith } 1165f1580f4eSBarry Smith } 1166f1580f4eSBarry Smith data->N -= reused; 1167f1580f4eSBarry Smith PetscCall(KSPSetOperators(data->levels[0]->ksp, A, P)); 1168f1580f4eSBarry Smith 1169f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)P, MATIS, &ismatis)); 1170f1580f4eSBarry Smith if (!data->is && !ismatis) { 1171db4a47b3SPierre Jolivet PetscErrorCode (*create)(DM, IS *, Mat *, PetscErrorCode (**)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void **) = nullptr; 1172db4a47b3SPierre Jolivet PetscErrorCode (*usetup)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *) = nullptr; 1173db4a47b3SPierre Jolivet void *uctx = nullptr; 1174f1580f4eSBarry Smith 1175f1580f4eSBarry Smith /* first see if we can get the data from the DM */ 1176f1580f4eSBarry Smith PetscCall(MatGetDM(P, &dm)); 1177f1580f4eSBarry Smith if (!dm) PetscCall(MatGetDM(A, &dm)); 1178f1580f4eSBarry Smith if (!dm) PetscCall(PCGetDM(pc, &dm)); 1179f1580f4eSBarry Smith if (dm) { /* this is the hook for DMPLEX and DMDA for which the auxiliary Mat is the local Neumann matrix */ 1180f1580f4eSBarry Smith PetscCall(PetscObjectQueryFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", &create)); 1181f1580f4eSBarry Smith if (create) { 1182f1580f4eSBarry Smith PetscCall((*create)(dm, &uis, &uaux, &usetup, &uctx)); 1183c8ea6600SPierre 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 */ 1184f1580f4eSBarry Smith } 1185f1580f4eSBarry Smith } 1186f1580f4eSBarry Smith if (!create) { 1187f1580f4eSBarry Smith if (!uis) { 1188f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Neumann_IS", (PetscObject *)&uis)); 1189f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)uis)); 1190f1580f4eSBarry Smith } 1191f1580f4eSBarry Smith if (!uaux) { 1192f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Neumann_Mat", (PetscObject *)&uaux)); 1193f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)uaux)); 1194f1580f4eSBarry Smith } 1195f1580f4eSBarry Smith /* look inside the Pmat instead of the PC, needed for MatSchurComplementComputeExplicitOperator() */ 1196f1580f4eSBarry Smith if (!uis) { 1197f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)P, "_PCHPDDM_Neumann_IS", (PetscObject *)&uis)); 1198f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)uis)); 1199f1580f4eSBarry Smith } 1200f1580f4eSBarry Smith if (!uaux) { 1201f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)P, "_PCHPDDM_Neumann_Mat", (PetscObject *)&uaux)); 1202f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)uaux)); 1203f1580f4eSBarry Smith } 1204f1580f4eSBarry Smith } 1205f1580f4eSBarry Smith PetscCall(PCHPDDMSetAuxiliaryMat(pc, uis, uaux, usetup, uctx)); 1206f1580f4eSBarry Smith PetscCall(MatDestroy(&uaux)); 1207f1580f4eSBarry Smith PetscCall(ISDestroy(&uis)); 1208f1580f4eSBarry Smith } 1209f1580f4eSBarry Smith 1210f1580f4eSBarry Smith if (!ismatis) { 1211f1580f4eSBarry Smith PetscCall(PCHPDDMSetUpNeumannOverlap_Private(pc)); 1212db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetBool(nullptr, pcpre, "-pc_hpddm_block_splitting", &block, nullptr)); 121302800ff6SPierre Jolivet if (data->is && block) { 121402800ff6SPierre Jolivet PetscCall(ISDestroy(&data->is)); 121502800ff6SPierre Jolivet PetscCall(MatDestroy(&data->aux)); 121602800ff6SPierre Jolivet } 1217f1580f4eSBarry Smith if (!data->is && data->N > 1) { 1218f1580f4eSBarry Smith char type[256] = {}; /* same size as in src/ksp/pc/interface/pcset.c */ 1219f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompareAny((PetscObject)P, &flg, MATNORMAL, MATNORMALHERMITIAN, "")); 1220f1580f4eSBarry Smith if (flg || (A->rmap->N != A->cmap->N && P->rmap->N == P->cmap->N && P->rmap->N == A->cmap->N)) { 1221f1580f4eSBarry Smith Mat B; 1222f1580f4eSBarry Smith PetscCall(PCHPDDMSetAuxiliaryMatNormal_Private(pc, A, P, &B, pcpre)); 1223f1580f4eSBarry Smith if (data->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED) data->correction = PC_HPDDM_COARSE_CORRECTION_BALANCED; 1224f1580f4eSBarry Smith PetscCall(MatDestroy(&B)); 1225f1580f4eSBarry Smith } else { 1226f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg)); 1227f1580f4eSBarry Smith if (flg) { 1228f1580f4eSBarry Smith Mat A00, P00, A01, A10, A11, B, N; 12293df4cd7bSPierre Jolivet Vec diagonal = nullptr; 1230f1580f4eSBarry Smith const PetscScalar *array; 1231f1580f4eSBarry Smith MatSchurComplementAinvType type; 1232f1580f4eSBarry Smith 1233f1580f4eSBarry Smith PetscCall(MatSchurComplementGetSubMatrices(P, &A00, &P00, &A01, &A10, &A11)); 1234f1580f4eSBarry Smith if (A11) { 12353df4cd7bSPierre Jolivet PetscCall(MatCreateVecs(A11, &diagonal, nullptr)); 12363df4cd7bSPierre Jolivet PetscCall(MatGetDiagonal(A11, diagonal)); 1237f1580f4eSBarry Smith } 1238f1580f4eSBarry Smith if (PetscDefined(USE_DEBUG)) { 1239db4a47b3SPierre Jolivet Mat T, U = nullptr; 1240f1580f4eSBarry Smith IS z; 1241f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, &flg)); 1242f1580f4eSBarry Smith if (flg) PetscCall(MatTransposeGetMat(A10, &U)); 1243f1580f4eSBarry Smith else { 1244f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, &flg)); 1245f1580f4eSBarry Smith if (flg) PetscCall(MatHermitianTransposeGetMat(A10, &U)); 1246f1580f4eSBarry Smith } 1247f1580f4eSBarry Smith if (U) PetscCall(MatDuplicate(U, MAT_COPY_VALUES, &T)); 1248f1580f4eSBarry Smith else PetscCall(MatHermitianTranspose(A10, MAT_INITIAL_MATRIX, &T)); 1249f1580f4eSBarry Smith PetscCall(PetscLayoutCompare(T->rmap, A01->rmap, &flg)); 1250f1580f4eSBarry Smith if (flg) { 1251f1580f4eSBarry Smith PetscCall(PetscLayoutCompare(T->cmap, A01->cmap, &flg)); 1252f1580f4eSBarry Smith if (flg) { 1253f1580f4eSBarry Smith PetscCall(MatFindZeroRows(A01, &z)); /* for essential boundary conditions, some implementations will */ 1254f1580f4eSBarry Smith if (z) { /* zero rows in [P00 A01] except for the diagonal of P00 */ 1255f1580f4eSBarry Smith PetscCall(MatSetOption(T, MAT_NO_OFF_PROC_ZERO_ROWS, PETSC_TRUE)); 1256db4a47b3SPierre Jolivet PetscCall(MatZeroRowsIS(T, z, 0.0, nullptr, nullptr)); /* corresponding zero rows from A01 */ 1257f1580f4eSBarry Smith PetscCall(ISDestroy(&z)); 1258f1580f4eSBarry Smith } 1259f1580f4eSBarry Smith PetscCall(MatMultEqual(A01, T, 10, &flg)); 1260f1580f4eSBarry Smith PetscCheck(flg, PetscObjectComm((PetscObject)P), PETSC_ERR_SUP, "A01 != A10^T"); 1261f1580f4eSBarry Smith } else PetscCall(PetscInfo(pc, "A01 and A10^T have non-congruent column layouts, cannot test for equality\n")); 1262f1580f4eSBarry Smith } 1263f1580f4eSBarry Smith PetscCall(MatDestroy(&T)); 1264f1580f4eSBarry Smith } 1265db4a47b3SPierre Jolivet PetscCall(MatCreateVecs(P00, &v, nullptr)); 1266f1580f4eSBarry Smith PetscCall(MatSchurComplementGetAinvType(P, &type)); 1267f1580f4eSBarry 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]); 1268f1580f4eSBarry Smith if (type == MAT_SCHUR_COMPLEMENT_AINV_LUMP) { 1269f1580f4eSBarry Smith PetscCall(MatGetRowSum(P00, v)); 1270f1580f4eSBarry Smith if (A00 == P00) PetscCall(PetscObjectReference((PetscObject)A00)); 1271f1580f4eSBarry Smith PetscCall(MatDestroy(&P00)); 1272f1580f4eSBarry Smith PetscCall(VecGetArrayRead(v, &array)); 1273db4a47b3SPierre Jolivet PetscCall(MatCreateAIJ(PetscObjectComm((PetscObject)A00), A00->rmap->n, A00->cmap->n, A00->rmap->N, A00->cmap->N, 1, nullptr, 0, nullptr, &P00)); 1274f1580f4eSBarry Smith PetscCall(MatSetOption(P00, MAT_NO_OFF_PROC_ENTRIES, PETSC_TRUE)); 1275f1580f4eSBarry Smith for (n = A00->rmap->rstart; n < A00->rmap->rend; ++n) PetscCall(MatSetValue(P00, n, n, array[n - A00->rmap->rstart], INSERT_VALUES)); 1276f1580f4eSBarry Smith PetscCall(MatAssemblyBegin(P00, MAT_FINAL_ASSEMBLY)); 1277f1580f4eSBarry Smith PetscCall(MatAssemblyEnd(P00, MAT_FINAL_ASSEMBLY)); 1278f1580f4eSBarry Smith PetscCall(VecRestoreArrayRead(v, &array)); 1279f1580f4eSBarry Smith PetscCall(MatSchurComplementUpdateSubMatrices(P, A00, P00, A01, A10, A11)); /* replace P00 by diag(sum of each row of P00) */ 1280f1580f4eSBarry Smith PetscCall(MatDestroy(&P00)); 1281f1580f4eSBarry Smith } else PetscCall(MatGetDiagonal(P00, v)); 1282f1580f4eSBarry Smith PetscCall(VecReciprocal(v)); /* inv(diag(P00)) */ 1283f1580f4eSBarry Smith PetscCall(VecSqrtAbs(v)); /* sqrt(inv(diag(P00))) */ 1284f1580f4eSBarry Smith PetscCall(MatDuplicate(A01, MAT_COPY_VALUES, &B)); 1285db4a47b3SPierre Jolivet PetscCall(MatDiagonalScale(B, v, nullptr)); 1286f1580f4eSBarry Smith PetscCall(VecDestroy(&v)); 1287f1580f4eSBarry Smith PetscCall(MatCreateNormalHermitian(B, &N)); 12883df4cd7bSPierre Jolivet PetscCall(PCHPDDMSetAuxiliaryMatNormal_Private(pc, B, N, &P, pcpre, &diagonal)); 1289f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)data->aux, MATSEQAIJ, &flg)); 1290f1580f4eSBarry Smith if (!flg) { 1291f1580f4eSBarry Smith PetscCall(MatDestroy(&P)); 1292f1580f4eSBarry Smith P = N; 1293f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)P)); 1294f1580f4eSBarry Smith } else PetscCall(MatScale(P, -1.0)); 12953df4cd7bSPierre Jolivet if (diagonal) { 12963df4cd7bSPierre Jolivet PetscCall(MatDiagonalSet(P, diagonal, ADD_VALUES)); 1297*aa21023fSPierre Jolivet PetscCall(PCSetOperators(pc, P, P)); /* replace P by diag(P11) - A01^T inv(diag(P00)) A01 */ 12983df4cd7bSPierre Jolivet PetscCall(VecDestroy(&diagonal)); 12993df4cd7bSPierre Jolivet } else { 1300f1580f4eSBarry Smith PetscCall(MatScale(N, -1.0)); 1301*aa21023fSPierre Jolivet PetscCall(PCSetOperators(pc, N, P)); /* replace P by - A01^T inv(diag(P00)) A01 */ 13023df4cd7bSPierre Jolivet } 1303f1580f4eSBarry Smith PetscCall(MatDestroy(&N)); 1304f1580f4eSBarry Smith PetscCall(MatDestroy(&P)); 1305f1580f4eSBarry Smith PetscCall(MatDestroy(&B)); 13063ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1307f1580f4eSBarry Smith } else { 1308db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetString(nullptr, pcpre, "-pc_hpddm_levels_1_st_pc_type", type, sizeof(type), nullptr)); 1309f1580f4eSBarry Smith PetscCall(PetscStrcmp(type, PCMAT, &algebraic)); 1310db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetBool(nullptr, pcpre, "-pc_hpddm_block_splitting", &block, nullptr)); 1311b07dfdedSPierre Jolivet PetscCheck(!algebraic || !block, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_levels_1_st_pc_type mat and -pc_hpddm_block_splitting"); 1312f1580f4eSBarry Smith if (block) algebraic = PETSC_TRUE; 1313f1580f4eSBarry Smith if (algebraic) { 1314f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, P->rmap->n, P->rmap->rstart, 1, &data->is)); 1315f1580f4eSBarry Smith PetscCall(MatIncreaseOverlap(P, 1, &data->is, 1)); 1316f1580f4eSBarry Smith PetscCall(ISSort(data->is)); 1317f1580f4eSBarry Smith } else 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\n", pcpre ? pcpre : "", pcpre ? pcpre : "")); 1318f1580f4eSBarry Smith } 1319f1580f4eSBarry Smith } 1320f1580f4eSBarry Smith } 1321f1580f4eSBarry Smith } 1322398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG) 1323398c7888SPierre Jolivet if (data->is) PetscCall(ISDuplicate(data->is, &dis)); 1324398c7888SPierre Jolivet if (data->aux) PetscCall(MatDuplicate(data->aux, MAT_COPY_VALUES, &daux)); 1325398c7888SPierre Jolivet #endif 1326f1580f4eSBarry Smith if (data->is || (ismatis && data->N > 1)) { 1327f1580f4eSBarry Smith if (ismatis) { 1328f1580f4eSBarry Smith std::initializer_list<std::string> list = {MATSEQBAIJ, MATSEQSBAIJ}; 1329f1580f4eSBarry Smith PetscCall(MatISGetLocalMat(P, &N)); 1330f1580f4eSBarry Smith std::initializer_list<std::string>::const_iterator it = std::find(list.begin(), list.end(), ((PetscObject)N)->type_name); 1331f1580f4eSBarry Smith PetscCall(MatISRestoreLocalMat(P, &N)); 1332f1580f4eSBarry Smith switch (std::distance(list.begin(), it)) { 1333d71ae5a4SJacob Faibussowitsch case 0: 1334d71ae5a4SJacob Faibussowitsch PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &C)); 1335d71ae5a4SJacob Faibussowitsch break; 1336f1580f4eSBarry Smith case 1: 1337f1580f4eSBarry Smith /* MatCreateSubMatrices() does not work with MATSBAIJ and unsorted ISes, so convert to MPIBAIJ */ 1338f1580f4eSBarry Smith PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &C)); 1339f1580f4eSBarry Smith PetscCall(MatSetOption(C, MAT_SYMMETRIC, PETSC_TRUE)); 1340f1580f4eSBarry Smith break; 1341d71ae5a4SJacob Faibussowitsch default: 1342d71ae5a4SJacob Faibussowitsch PetscCall(MatConvert(P, MATMPIAIJ, MAT_INITIAL_MATRIX, &C)); 1343f1580f4eSBarry Smith } 1344db4a47b3SPierre Jolivet PetscCall(MatISGetLocalToGlobalMapping(P, &l2g, nullptr)); 1345f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)P)); 1346f1580f4eSBarry Smith PetscCall(KSPSetOperators(data->levels[0]->ksp, A, C)); 1347f1580f4eSBarry Smith std::swap(C, P); 1348f1580f4eSBarry Smith PetscCall(ISLocalToGlobalMappingGetSize(l2g, &n)); 1349f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, n, 0, 1, &loc)); 1350f1580f4eSBarry Smith PetscCall(ISLocalToGlobalMappingApplyIS(l2g, loc, &is[0])); 1351f1580f4eSBarry Smith PetscCall(ISDestroy(&loc)); 1352f1580f4eSBarry Smith /* the auxiliary Mat is _not_ the local Neumann matrix */ 1353f1580f4eSBarry Smith /* it is the local Neumann matrix augmented (with zeros) through MatIncreaseOverlap() */ 1354c8ea6600SPierre Jolivet data->Neumann = PETSC_BOOL3_FALSE; 1355f1580f4eSBarry Smith structure = SAME_NONZERO_PATTERN; 1356f1580f4eSBarry Smith } else { 1357f1580f4eSBarry Smith is[0] = data->is; 1358f1580f4eSBarry Smith if (algebraic) subdomains = PETSC_TRUE; 1359db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetBool(nullptr, pcpre, "-pc_hpddm_define_subdomains", &subdomains, nullptr)); 1360c8ea6600SPierre Jolivet if (PetscBool3ToBool(data->Neumann)) { 1361b07dfdedSPierre Jolivet PetscCheck(!block, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_block_splitting and -pc_hpddm_has_neumann"); 1362b07dfdedSPierre Jolivet PetscCheck(!algebraic, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_levels_1_st_pc_type mat and -pc_hpddm_has_neumann"); 1363f1580f4eSBarry Smith } 1364c8ea6600SPierre Jolivet if (PetscBool3ToBool(data->Neumann) || block) structure = SAME_NONZERO_PATTERN; 1365f1580f4eSBarry Smith PetscCall(ISCreateStride(PetscObjectComm((PetscObject)data->is), P->rmap->n, P->rmap->rstart, 1, &loc)); 1366f1580f4eSBarry Smith } 1367f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : "")); 1368db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetEnum(nullptr, prefix, "-st_matstructure", MatStructures, (PetscEnum *)&structure, &flg)); /* if not user-provided, force its value when possible */ 1369f1580f4eSBarry Smith if (!flg && structure == SAME_NONZERO_PATTERN) { /* cannot call STSetMatStructure() yet, insert the appropriate option in the database, parsed by STSetFromOptions() */ 1370f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-%spc_hpddm_levels_1_st_matstructure", pcpre ? pcpre : "")); 1371db4a47b3SPierre Jolivet PetscCall(PetscOptionsSetValue(nullptr, prefix, MatStructures[structure])); 1372f1580f4eSBarry Smith } 1373398c7888SPierre Jolivet flg = PETSC_FALSE; 1374b07dfdedSPierre Jolivet if (data->share) { 1375398c7888SPierre Jolivet data->share = PETSC_FALSE; /* will be reset to PETSC_TRUE if none of the conditions below are true */ 1376398c7888SPierre 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 : "")); 1377398c7888SPierre Jolivet else if (data->deflation) PetscCall(PetscInfo(pc, "Nothing to share since PCHPDDMSetDeflationMat() has been called\n")); 1378398c7888SPierre Jolivet else if (ismatis) PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc with a Pmat of type MATIS\n")); 1379b07dfdedSPierre Jolivet else if (!algebraic && structure != SAME_NONZERO_PATTERN) 1380398c7888SPierre 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])); 1381398c7888SPierre Jolivet else data->share = PETSC_TRUE; 1382398c7888SPierre Jolivet } 1383398c7888SPierre Jolivet if (!ismatis) { 1384398c7888SPierre Jolivet if (data->share || (!PetscBool3ToBool(data->Neumann) && subdomains)) PetscCall(ISDuplicate(is[0], &unsorted)); 1385398c7888SPierre Jolivet else unsorted = is[0]; 1386398c7888SPierre Jolivet } 1387f1580f4eSBarry Smith if (data->N > 1 && (data->aux || ismatis || algebraic)) { 1388f1580f4eSBarry Smith PetscCheck(loadedSym, PETSC_COMM_SELF, PETSC_ERR_PLIB, "HPDDM library not loaded, cannot use more than one level"); 1389f1580f4eSBarry Smith PetscCall(MatSetOption(P, MAT_SUBMAT_SINGLEIS, PETSC_TRUE)); 1390f1580f4eSBarry Smith if (ismatis) { 1391f1580f4eSBarry Smith /* needed by HPDDM (currently) so that the partition of unity is 0 on subdomain interfaces */ 1392f1580f4eSBarry Smith PetscCall(MatIncreaseOverlap(P, 1, is, 1)); 1393f1580f4eSBarry Smith PetscCall(ISDestroy(&data->is)); 1394f1580f4eSBarry Smith data->is = is[0]; 1395f1580f4eSBarry Smith } else { 1396f1580f4eSBarry Smith if (PetscDefined(USE_DEBUG)) { 1397f1580f4eSBarry Smith PetscBool equal; 1398f1580f4eSBarry Smith IS intersect; 1399f1580f4eSBarry Smith 1400f1580f4eSBarry Smith PetscCall(ISIntersect(data->is, loc, &intersect)); 1401f1580f4eSBarry Smith PetscCall(ISEqualUnsorted(loc, intersect, &equal)); 1402f1580f4eSBarry Smith PetscCall(ISDestroy(&intersect)); 1403f1580f4eSBarry Smith PetscCheck(equal, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "IS of the auxiliary Mat does not include all local rows of A"); 1404f1580f4eSBarry Smith } 1405f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", PCHPDDMAlgebraicAuxiliaryMat_Private)); 1406c8ea6600SPierre Jolivet if (!PetscBool3ToBool(data->Neumann) && !algebraic) { 1407f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)P, MATMPISBAIJ, &flg)); 1408f1580f4eSBarry Smith if (flg) { 1409f1580f4eSBarry Smith /* maybe better to ISSort(is[0]), MatCreateSubMatrices(), and then MatPermute() */ 1410f1580f4eSBarry Smith /* but there is no MatPermute_SeqSBAIJ(), so as before, just use MATMPIBAIJ */ 1411f1580f4eSBarry Smith PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &uaux)); 1412f1580f4eSBarry Smith flg = PETSC_FALSE; 1413f1580f4eSBarry Smith } 1414f1580f4eSBarry Smith } 1415f1580f4eSBarry Smith } 1416f1580f4eSBarry Smith if (algebraic) { 1417f1580f4eSBarry Smith PetscUseMethod(pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", (Mat, IS *, Mat *[], PetscBool), (P, is, &sub, block)); 1418f1580f4eSBarry Smith if (block) { 1419f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)sub[0], "_PCHPDDM_Neumann_Mat", (PetscObject *)&data->aux)); 1420db4a47b3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Neumann_Mat", nullptr)); 1421f1580f4eSBarry Smith } 1422f1580f4eSBarry Smith } else if (!uaux) { 1423c8ea6600SPierre Jolivet if (PetscBool3ToBool(data->Neumann)) sub = &data->aux; 1424f1580f4eSBarry Smith else PetscCall(MatCreateSubMatrices(P, 1, is, is, MAT_INITIAL_MATRIX, &sub)); 1425f1580f4eSBarry Smith } else { 1426f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(uaux, 1, is, is, MAT_INITIAL_MATRIX, &sub)); 1427f1580f4eSBarry Smith PetscCall(MatDestroy(&uaux)); 1428f1580f4eSBarry Smith PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub)); 1429f1580f4eSBarry Smith } 1430f1580f4eSBarry Smith /* Vec holding the partition of unity */ 1431f1580f4eSBarry Smith if (!data->levels[0]->D) { 1432f1580f4eSBarry Smith PetscCall(ISGetLocalSize(data->is, &n)); 1433f1580f4eSBarry Smith PetscCall(VecCreateMPI(PETSC_COMM_SELF, n, PETSC_DETERMINE, &data->levels[0]->D)); 1434f1580f4eSBarry Smith } 1435398c7888SPierre Jolivet if (data->share) { 1436f1580f4eSBarry Smith Mat D; 1437db4a47b3SPierre Jolivet IS perm = nullptr; 1438f1580f4eSBarry Smith PetscInt size = -1; 1439398c7888SPierre Jolivet if (!data->levels[0]->pc) { 1440398c7888SPierre Jolivet PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : "")); 1441398c7888SPierre Jolivet PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->pc)); 1442398c7888SPierre Jolivet PetscCall(PCSetOptionsPrefix(data->levels[0]->pc, prefix)); 1443398c7888SPierre Jolivet PetscCall(PCSetOperators(data->levels[0]->pc, A, P)); 1444398c7888SPierre Jolivet } 1445398c7888SPierre Jolivet PetscCall(PCSetType(data->levels[0]->pc, PCASM)); 1446398c7888SPierre Jolivet if (!data->levels[0]->pc->setupcalled) { 1447398c7888SPierre Jolivet IS sorted; /* PCASM will sort the input IS, duplicate it to return an unmodified (PCHPDDM) input IS */ 1448398c7888SPierre Jolivet PetscCall(ISDuplicate(is[0], &sorted)); 1449398c7888SPierre Jolivet PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, &sorted, &loc)); 1450398c7888SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)sorted)); 1451398c7888SPierre Jolivet } 1452398c7888SPierre Jolivet PetscCall(PCSetFromOptions(data->levels[0]->pc)); 1453398c7888SPierre Jolivet if (block) { 1454398c7888SPierre Jolivet PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, sub[0], &C, &perm)); 1455398c7888SPierre Jolivet PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(data->levels[0]->pc, C, algebraic)); 1456398c7888SPierre Jolivet } else PetscCall(PCSetUp(data->levels[0]->pc)); 1457db4a47b3SPierre Jolivet PetscTryMethod(data->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data->levels[0]->pc, &size, nullptr, &ksp)); 1458398c7888SPierre Jolivet if (size != 1) { 1459398c7888SPierre Jolivet data->share = PETSC_FALSE; 1460398c7888SPierre Jolivet PetscCheck(size == -1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", size); 1461398c7888SPierre Jolivet PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc since PCASMGetSubKSP() not found in fine-level PC\n")); 1462398c7888SPierre Jolivet PetscCall(ISDestroy(&unsorted)); 1463398c7888SPierre Jolivet unsorted = is[0]; 1464398c7888SPierre Jolivet } else { 1465398c7888SPierre Jolivet if (!block) PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, PetscBool3ToBool(data->Neumann) ? sub[0] : data->aux, &C, &perm)); 1466c8ea6600SPierre Jolivet if (!PetscBool3ToBool(data->Neumann) && !block) { 1467f1580f4eSBarry Smith PetscCall(MatPermute(sub[0], perm, perm, &D)); /* permute since PCASM will call ISSort() */ 1468f1580f4eSBarry Smith PetscCall(MatHeaderReplace(sub[0], &D)); 1469f1580f4eSBarry Smith } 1470f1580f4eSBarry Smith if (data->B) { /* see PCHPDDMSetRHSMat() */ 1471f1580f4eSBarry Smith PetscCall(MatPermute(data->B, perm, perm, &D)); 1472f1580f4eSBarry Smith PetscCall(MatHeaderReplace(data->B, &D)); 1473f1580f4eSBarry Smith } 1474f1580f4eSBarry Smith PetscCall(ISDestroy(&perm)); 1475f1580f4eSBarry Smith const char *matpre; 1476f1580f4eSBarry Smith PetscBool cmp[4]; 1477f1580f4eSBarry Smith PetscCall(KSPGetOperators(ksp[0], subA, subA + 1)); 1478f1580f4eSBarry Smith PetscCall(MatDuplicate(subA[1], MAT_SHARE_NONZERO_PATTERN, &D)); 1479f1580f4eSBarry Smith PetscCall(MatGetOptionsPrefix(subA[1], &matpre)); 1480f1580f4eSBarry Smith PetscCall(MatSetOptionsPrefix(D, matpre)); 1481f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)D, MATNORMAL, cmp)); 1482f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)C, MATNORMAL, cmp + 1)); 1483f1580f4eSBarry Smith if (!cmp[0]) PetscCall(PetscObjectTypeCompare((PetscObject)D, MATNORMALHERMITIAN, cmp + 2)); 1484f1580f4eSBarry Smith else cmp[2] = PETSC_FALSE; 1485f1580f4eSBarry Smith if (!cmp[1]) PetscCall(PetscObjectTypeCompare((PetscObject)C, MATNORMALHERMITIAN, cmp + 3)); 1486f1580f4eSBarry Smith else cmp[3] = PETSC_FALSE; 1487f1580f4eSBarry 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); 1488f1580f4eSBarry Smith if (!cmp[0] && !cmp[2]) { 1489f1580f4eSBarry Smith if (!block) PetscCall(MatAXPY(D, 1.0, C, SUBSET_NONZERO_PATTERN)); 14905c7345deSPierre Jolivet else { 1491db4a47b3SPierre Jolivet PetscCall(MatMissingDiagonal(D, cmp, nullptr)); 14925c7345deSPierre Jolivet if (cmp[0]) structure = DIFFERENT_NONZERO_PATTERN; /* data->aux has no missing diagonal entry */ 14935c7345deSPierre Jolivet PetscCall(MatAXPY(D, 1.0, data->aux, structure)); 14945c7345deSPierre Jolivet } 1495f1580f4eSBarry Smith } else { 1496f1580f4eSBarry Smith Mat mat[2]; 1497f1580f4eSBarry Smith if (cmp[0]) { 1498f1580f4eSBarry Smith PetscCall(MatNormalGetMat(D, mat)); 1499f1580f4eSBarry Smith PetscCall(MatNormalGetMat(C, mat + 1)); 1500f1580f4eSBarry Smith } else { 1501f1580f4eSBarry Smith PetscCall(MatNormalHermitianGetMat(D, mat)); 1502f1580f4eSBarry Smith PetscCall(MatNormalHermitianGetMat(C, mat + 1)); 1503f1580f4eSBarry Smith } 1504f1580f4eSBarry Smith PetscCall(MatAXPY(mat[0], 1.0, mat[1], SUBSET_NONZERO_PATTERN)); 1505f1580f4eSBarry Smith } 1506f1580f4eSBarry Smith PetscCall(MatPropagateSymmetryOptions(C, D)); 1507f1580f4eSBarry Smith PetscCall(MatDestroy(&C)); 1508f1580f4eSBarry Smith C = D; 1509f1580f4eSBarry Smith /* swap pointers so that variables stay consistent throughout PCSetUp() */ 1510f1580f4eSBarry Smith std::swap(C, data->aux); 1511f1580f4eSBarry Smith std::swap(uis, data->is); 1512f1580f4eSBarry Smith swap = PETSC_TRUE; 1513f1580f4eSBarry Smith } 1514f1580f4eSBarry Smith } 1515f1580f4eSBarry Smith if (!data->levels[0]->scatter) { 1516db4a47b3SPierre Jolivet PetscCall(MatCreateVecs(P, &xin, nullptr)); 1517f1580f4eSBarry Smith if (ismatis) PetscCall(MatDestroy(&P)); 1518db4a47b3SPierre Jolivet PetscCall(VecScatterCreate(xin, data->is, data->levels[0]->D, nullptr, &data->levels[0]->scatter)); 1519f1580f4eSBarry Smith PetscCall(VecDestroy(&xin)); 1520f1580f4eSBarry Smith } 1521f1580f4eSBarry Smith if (data->levels[0]->P) { 1522f1580f4eSBarry Smith /* if the pattern is the same and PCSetUp() has previously succeeded, reuse HPDDM buffers and connectivity */ 1523f1580f4eSBarry Smith PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(data->levels[0], pc->setupcalled < 1 || pc->flag == DIFFERENT_NONZERO_PATTERN ? PETSC_TRUE : PETSC_FALSE)); 1524f1580f4eSBarry Smith } 1525f1580f4eSBarry Smith if (!data->levels[0]->P) data->levels[0]->P = new HPDDM::Schwarz<PetscScalar>(); 1526db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventBegin(PC_HPDDM_SetUp[0], data->levels[0]->ksp, nullptr, nullptr, nullptr)); 1527db4a47b3SPierre Jolivet else PetscCall(PetscLogEventBegin(PC_HPDDM_Strc, data->levels[0]->ksp, nullptr, nullptr, nullptr)); 1528f1580f4eSBarry Smith /* HPDDM internal data structure */ 1529f1580f4eSBarry Smith PetscCall(data->levels[0]->P->structure(loc, data->is, sub[0], ismatis ? C : data->aux, data->levels)); 1530db4a47b3SPierre Jolivet if (!data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Strc, data->levels[0]->ksp, nullptr, nullptr, nullptr)); 1531f1580f4eSBarry Smith /* matrix pencil of the generalized eigenvalue problem on the overlap (GenEO) */ 1532f1580f4eSBarry Smith if (data->deflation) weighted = data->aux; 1533f1580f4eSBarry Smith else if (!data->B) { 1534f1580f4eSBarry Smith PetscBool cmp[2]; 1535f1580f4eSBarry Smith PetscCall(MatDuplicate(sub[0], MAT_COPY_VALUES, &weighted)); 1536f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)weighted, MATNORMAL, cmp)); 1537f1580f4eSBarry Smith if (!cmp[0]) PetscCall(PetscObjectTypeCompare((PetscObject)weighted, MATNORMALHERMITIAN, cmp + 1)); 1538f1580f4eSBarry Smith else cmp[1] = PETSC_FALSE; 1539f1580f4eSBarry Smith if (!cmp[0] && !cmp[1]) PetscCall(MatDiagonalScale(weighted, data->levels[0]->D, data->levels[0]->D)); 1540f1580f4eSBarry Smith else { /* MATNORMAL applies MatDiagonalScale() in a matrix-free fashion, not what is needed since this won't be passed to SLEPc during the eigensolve */ 1541f1580f4eSBarry Smith if (cmp[0]) PetscCall(MatNormalGetMat(weighted, &data->B)); 1542f1580f4eSBarry Smith else PetscCall(MatNormalHermitianGetMat(weighted, &data->B)); 1543db4a47b3SPierre Jolivet PetscCall(MatDiagonalScale(data->B, nullptr, data->levels[0]->D)); 1544db4a47b3SPierre Jolivet data->B = nullptr; 1545f1580f4eSBarry Smith flg = PETSC_FALSE; 1546f1580f4eSBarry Smith } 1547f1580f4eSBarry Smith /* neither MatDuplicate() nor MatDiagonaleScale() handles the symmetry options, so propagate the options explicitly */ 1548f1580f4eSBarry Smith /* only useful for -mat_type baij -pc_hpddm_levels_1_st_pc_type cholesky (no problem with MATAIJ or MATSBAIJ) */ 1549f1580f4eSBarry Smith PetscCall(MatPropagateSymmetryOptions(sub[0], weighted)); 1550f1580f4eSBarry Smith } else weighted = data->B; 1551f1580f4eSBarry Smith /* SLEPc is used inside the loaded symbol */ 1552f1580f4eSBarry Smith PetscCall((*loadedSym)(data->levels[0]->P, data->is, ismatis ? C : (algebraic && !block ? sub[0] : data->aux), weighted, data->B, initial, data->levels)); 1553f1580f4eSBarry Smith if (data->share) { 1554f1580f4eSBarry Smith Mat st[2]; 1555f1580f4eSBarry Smith PetscCall(KSPGetOperators(ksp[0], st, st + 1)); 15565c7345deSPierre Jolivet PetscCall(MatCopy(subA[0], st[0], structure)); 1557f1580f4eSBarry Smith if (subA[1] != subA[0] || st[1] != st[0]) PetscCall(MatCopy(subA[1], st[1], SAME_NONZERO_PATTERN)); 1558f1580f4eSBarry Smith } 1559db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_SetUp[0], data->levels[0]->ksp, nullptr, nullptr, nullptr)); 1560f1580f4eSBarry Smith if (ismatis) PetscCall(MatISGetLocalMat(C, &N)); 1561f1580f4eSBarry Smith else N = data->aux; 1562f1580f4eSBarry Smith P = sub[0]; 1563f1580f4eSBarry Smith /* going through the grid hierarchy */ 1564f1580f4eSBarry Smith for (n = 1; n < data->N; ++n) { 1565db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventBegin(PC_HPDDM_SetUp[n], data->levels[n]->ksp, nullptr, nullptr, nullptr)); 1566f1580f4eSBarry Smith /* method composed in the loaded symbol since there, SLEPc is used as well */ 1567f1580f4eSBarry 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)); 1568db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_SetUp[n], data->levels[n]->ksp, nullptr, nullptr, nullptr)); 1569f1580f4eSBarry Smith } 1570f1580f4eSBarry Smith /* reset to NULL to avoid any faulty use */ 1571db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)data->levels[0]->ksp, "PCHPDDMSetUp_Private_C", nullptr)); 1572db4a47b3SPierre Jolivet if (!ismatis) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_C", nullptr)); 1573f1580f4eSBarry Smith else PetscCall(PetscObjectDereference((PetscObject)C)); /* matching PetscObjectReference() above */ 1574f1580f4eSBarry Smith for (n = 0; n < data->N - 1; ++n) 1575f1580f4eSBarry Smith if (data->levels[n]->P) { 1576f1580f4eSBarry Smith /* HPDDM internal work buffers */ 1577f1580f4eSBarry Smith data->levels[n]->P->setBuffer(); 1578f1580f4eSBarry Smith data->levels[n]->P->super::start(); 1579f1580f4eSBarry Smith } 1580c8ea6600SPierre Jolivet if (ismatis || !subdomains) PetscCall(PCHPDDMDestroySubMatrices_Private(PetscBool3ToBool(data->Neumann), PetscBool(algebraic && !block), sub)); 1581db4a47b3SPierre Jolivet if (ismatis) data->is = nullptr; 1582f1580f4eSBarry Smith for (n = 0; n < data->N - 1 + (reused > 0); ++n) { 1583f1580f4eSBarry Smith if (data->levels[n]->P) { 1584f1580f4eSBarry Smith PC spc; 1585f1580f4eSBarry Smith 1586f1580f4eSBarry Smith /* force the PC to be PCSHELL to do the coarse grid corrections */ 1587f1580f4eSBarry Smith PetscCall(KSPSetSkipPCSetFromOptions(data->levels[n]->ksp, PETSC_TRUE)); 1588f1580f4eSBarry Smith PetscCall(KSPGetPC(data->levels[n]->ksp, &spc)); 1589f1580f4eSBarry Smith PetscCall(PCSetType(spc, PCSHELL)); 1590f1580f4eSBarry Smith PetscCall(PCShellSetContext(spc, data->levels[n])); 1591f1580f4eSBarry Smith PetscCall(PCShellSetSetUp(spc, PCSetUp_HPDDMShell)); 1592f1580f4eSBarry Smith PetscCall(PCShellSetApply(spc, PCApply_HPDDMShell)); 1593f1580f4eSBarry Smith PetscCall(PCShellSetMatApply(spc, PCMatApply_HPDDMShell)); 1594f1580f4eSBarry Smith PetscCall(PCShellSetDestroy(spc, PCDestroy_HPDDMShell)); 1595f1580f4eSBarry Smith if (!data->levels[n]->pc) PetscCall(PCCreate(PetscObjectComm((PetscObject)data->levels[n]->ksp), &data->levels[n]->pc)); 1596f1580f4eSBarry Smith if (n < reused) { 1597f1580f4eSBarry Smith PetscCall(PCSetReusePreconditioner(spc, PETSC_TRUE)); 1598f1580f4eSBarry Smith PetscCall(PCSetReusePreconditioner(data->levels[n]->pc, PETSC_TRUE)); 1599f1580f4eSBarry Smith } 1600f1580f4eSBarry Smith PetscCall(PCSetUp(spc)); 1601f1580f4eSBarry Smith } 1602f1580f4eSBarry Smith } 1603db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", nullptr)); 1604f1580f4eSBarry Smith } else flg = reused ? PETSC_FALSE : PETSC_TRUE; 1605f1580f4eSBarry Smith if (!ismatis && subdomains) { 1606f1580f4eSBarry Smith if (flg) PetscCall(KSPGetPC(data->levels[0]->ksp, &inner)); 1607f1580f4eSBarry Smith else inner = data->levels[0]->pc; 1608f1580f4eSBarry Smith if (inner) { 1609398c7888SPierre Jolivet if (!inner->setupcalled) PetscCall(PCSetType(inner, PCASM)); 1610398c7888SPierre Jolivet PetscCall(PCSetFromOptions(inner)); 1611398c7888SPierre Jolivet PetscCall(PetscStrcmp(((PetscObject)inner)->type_name, PCASM, &flg)); 1612398c7888SPierre Jolivet if (flg) { 1613f1580f4eSBarry Smith if (!inner->setupcalled) { /* evaluates to PETSC_FALSE when -pc_hpddm_block_splitting */ 1614398c7888SPierre Jolivet IS sorted; /* PCASM will sort the input IS, duplicate it to return an unmodified (PCHPDDM) input IS */ 1615398c7888SPierre Jolivet PetscCall(ISDuplicate(is[0], &sorted)); 1616398c7888SPierre Jolivet PetscCall(PCASMSetLocalSubdomains(inner, 1, &sorted, &loc)); 1617398c7888SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)sorted)); 1618398c7888SPierre Jolivet } 1619c8ea6600SPierre Jolivet if (!PetscBool3ToBool(data->Neumann) && data->N > 1) { /* subdomain matrices are already created for the eigenproblem, reuse them for the fine-level PC */ 1620db4a47b3SPierre Jolivet PetscCall(PCHPDDMPermute_Private(*is, nullptr, nullptr, sub[0], &P, nullptr)); 1621398c7888SPierre Jolivet PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(inner, P, algebraic)); 1622398c7888SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)P)); 1623f1580f4eSBarry Smith } 1624f1580f4eSBarry Smith } 1625f1580f4eSBarry Smith } 1626c8ea6600SPierre Jolivet if (data->N > 1) PetscCall(PCHPDDMDestroySubMatrices_Private(PetscBool3ToBool(data->Neumann), PetscBool(algebraic && !block), sub)); 1627f1580f4eSBarry Smith } 1628f1580f4eSBarry Smith PetscCall(ISDestroy(&loc)); 1629f1580f4eSBarry Smith } else data->N = 1 + reused; /* enforce this value to 1 + reused if there is no way to build another level */ 1630f1580f4eSBarry Smith if (requested != data->N + reused) { 1631f1580f4eSBarry 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, 1632f1580f4eSBarry Smith data->N, pcpre ? pcpre : "")); 1633f1580f4eSBarry 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)); 1634f1580f4eSBarry Smith /* cannot use PCDestroy_HPDDMShell() because PCSHELL not set for unassembled levels */ 1635f1580f4eSBarry Smith for (n = data->N - 1; n < requested - 1; ++n) { 1636f1580f4eSBarry Smith if (data->levels[n]->P) { 1637f1580f4eSBarry Smith PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(data->levels[n], PETSC_TRUE)); 1638f1580f4eSBarry Smith PetscCall(VecDestroyVecs(1, &data->levels[n]->v[0])); 1639f1580f4eSBarry Smith PetscCall(VecDestroyVecs(2, &data->levels[n]->v[1])); 1640f1580f4eSBarry Smith PetscCall(MatDestroy(data->levels[n]->V)); 1641f1580f4eSBarry Smith PetscCall(MatDestroy(data->levels[n]->V + 1)); 1642f1580f4eSBarry Smith PetscCall(MatDestroy(data->levels[n]->V + 2)); 1643f1580f4eSBarry Smith PetscCall(VecDestroy(&data->levels[n]->D)); 1644f1580f4eSBarry Smith PetscCall(VecScatterDestroy(&data->levels[n]->scatter)); 1645f1580f4eSBarry Smith } 1646f1580f4eSBarry Smith } 1647f1580f4eSBarry Smith if (reused) { 1648f1580f4eSBarry Smith for (n = reused; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) { 1649f1580f4eSBarry Smith PetscCall(KSPDestroy(&data->levels[n]->ksp)); 1650f1580f4eSBarry Smith PetscCall(PCDestroy(&data->levels[n]->pc)); 1651f1580f4eSBarry Smith } 1652f1580f4eSBarry Smith } 1653f1580f4eSBarry 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, 1654f1580f4eSBarry Smith data->N, reused, data->N, pcpre ? pcpre : "", pcpre ? pcpre : "", data->N); 1655f1580f4eSBarry Smith } 1656f1580f4eSBarry Smith /* these solvers are created after PCSetFromOptions() is called */ 1657f1580f4eSBarry Smith if (pc->setfromoptionscalled) { 1658f1580f4eSBarry Smith for (n = 0; n < data->N; ++n) { 1659f1580f4eSBarry Smith if (data->levels[n]->ksp) PetscCall(KSPSetFromOptions(data->levels[n]->ksp)); 1660f1580f4eSBarry Smith if (data->levels[n]->pc) PetscCall(PCSetFromOptions(data->levels[n]->pc)); 1661f1580f4eSBarry Smith } 1662f1580f4eSBarry Smith pc->setfromoptionscalled = 0; 1663f1580f4eSBarry Smith } 1664f1580f4eSBarry Smith data->N += reused; 1665f1580f4eSBarry Smith if (data->share && swap) { 1666f1580f4eSBarry Smith /* swap back pointers so that variables follow the user-provided numbering */ 1667f1580f4eSBarry Smith std::swap(C, data->aux); 1668f1580f4eSBarry Smith std::swap(uis, data->is); 1669f1580f4eSBarry Smith PetscCall(MatDestroy(&C)); 1670f1580f4eSBarry Smith PetscCall(ISDestroy(&uis)); 1671f1580f4eSBarry Smith } 1672398c7888SPierre Jolivet if (algebraic) PetscCall(MatDestroy(&data->aux)); 167338fb34a1SPierre Jolivet if (unsorted && unsorted != is[0]) { 1674398c7888SPierre Jolivet PetscCall(ISCopy(unsorted, data->is)); 1675398c7888SPierre Jolivet PetscCall(ISDestroy(&unsorted)); 1676398c7888SPierre Jolivet } 1677398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG) 1678398c7888SPierre 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); 1679398c7888SPierre Jolivet if (data->is) { 1680398c7888SPierre Jolivet PetscCall(ISEqualUnsorted(data->is, dis, &flg)); 1681398c7888SPierre Jolivet PetscCall(ISDestroy(&dis)); 1682398c7888SPierre Jolivet PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Input IS and output IS are not equal"); 1683398c7888SPierre Jolivet } 1684398c7888SPierre 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); 1685398c7888SPierre Jolivet if (data->aux) { 1686398c7888SPierre Jolivet PetscCall(MatMultEqual(data->aux, daux, 20, &flg)); 1687398c7888SPierre Jolivet PetscCall(MatDestroy(&daux)); 1688398c7888SPierre Jolivet PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Input Mat and output Mat are not equal"); 1689398c7888SPierre Jolivet } 1690398c7888SPierre Jolivet #endif 16913ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1692f1580f4eSBarry Smith } 1693f1580f4eSBarry Smith 1694f1580f4eSBarry Smith /*@ 1695f1580f4eSBarry Smith PCHPDDMSetCoarseCorrectionType - Sets the coarse correction type. 1696f1580f4eSBarry Smith 1697c3339decSBarry Smith Collective 1698f1580f4eSBarry Smith 1699f1580f4eSBarry Smith Input Parameters: 1700f1580f4eSBarry Smith + pc - preconditioner context 1701f1580f4eSBarry Smith - type - `PC_HPDDM_COARSE_CORRECTION_DEFLATED`, `PC_HPDDM_COARSE_CORRECTION_ADDITIVE`, or `PC_HPDDM_COARSE_CORRECTION_BALANCED` 1702f1580f4eSBarry Smith 1703f1580f4eSBarry Smith Options Database Key: 1704f1580f4eSBarry Smith . -pc_hpddm_coarse_correction <deflated, additive, balanced> - type of coarse correction to apply 1705f1580f4eSBarry Smith 1706f1580f4eSBarry Smith Level: intermediate 1707f1580f4eSBarry Smith 1708f1580f4eSBarry Smith .seealso: `PCHPDDMGetCoarseCorrectionType()`, `PCHPDDM`, `PCHPDDMCoarseCorrectionType` 1709f1580f4eSBarry Smith @*/ 1710d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetCoarseCorrectionType(PC pc, PCHPDDMCoarseCorrectionType type) 1711d71ae5a4SJacob Faibussowitsch { 1712f1580f4eSBarry Smith PetscFunctionBegin; 1713f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1714f1580f4eSBarry Smith PetscValidLogicalCollectiveEnum(pc, type, 2); 1715f1580f4eSBarry Smith PetscTryMethod(pc, "PCHPDDMSetCoarseCorrectionType_C", (PC, PCHPDDMCoarseCorrectionType), (pc, type)); 17163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1717f1580f4eSBarry Smith } 1718f1580f4eSBarry Smith 1719f1580f4eSBarry Smith /*@ 1720f1580f4eSBarry Smith PCHPDDMGetCoarseCorrectionType - Gets the coarse correction type. 1721f1580f4eSBarry Smith 1722f1580f4eSBarry Smith Input Parameter: 1723f1580f4eSBarry Smith . pc - preconditioner context 1724f1580f4eSBarry Smith 1725f1580f4eSBarry Smith Output Parameter: 1726f1580f4eSBarry Smith . type - `PC_HPDDM_COARSE_CORRECTION_DEFLATED`, `PC_HPDDM_COARSE_CORRECTION_ADDITIVE`, or `PC_HPDDM_COARSE_CORRECTION_BALANCED` 1727f1580f4eSBarry Smith 1728f1580f4eSBarry Smith Level: intermediate 1729f1580f4eSBarry Smith 1730f1580f4eSBarry Smith .seealso: `PCHPDDMSetCoarseCorrectionType()`, `PCHPDDM`, `PCHPDDMCoarseCorrectionType` 1731f1580f4eSBarry Smith @*/ 1732d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMGetCoarseCorrectionType(PC pc, PCHPDDMCoarseCorrectionType *type) 1733d71ae5a4SJacob Faibussowitsch { 1734f1580f4eSBarry Smith PetscFunctionBegin; 1735f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1736f1580f4eSBarry Smith if (type) { 17374f572ea9SToby Isaac PetscAssertPointer(type, 2); 1738f1580f4eSBarry Smith PetscUseMethod(pc, "PCHPDDMGetCoarseCorrectionType_C", (PC, PCHPDDMCoarseCorrectionType *), (pc, type)); 1739f1580f4eSBarry Smith } 17403ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1741f1580f4eSBarry Smith } 1742f1580f4eSBarry Smith 1743d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetCoarseCorrectionType_HPDDM(PC pc, PCHPDDMCoarseCorrectionType type) 1744d71ae5a4SJacob Faibussowitsch { 1745f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 1746f1580f4eSBarry Smith 1747f1580f4eSBarry Smith PetscFunctionBegin; 1748f1580f4eSBarry Smith PetscCheck(type >= 0 && type <= 2, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unknown PCHPDDMCoarseCorrectionType %d", type); 1749f1580f4eSBarry Smith data->correction = type; 17503ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1751f1580f4eSBarry Smith } 1752f1580f4eSBarry Smith 1753d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMGetCoarseCorrectionType_HPDDM(PC pc, PCHPDDMCoarseCorrectionType *type) 1754d71ae5a4SJacob Faibussowitsch { 1755f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 1756f1580f4eSBarry Smith 1757f1580f4eSBarry Smith PetscFunctionBegin; 1758f1580f4eSBarry Smith *type = data->correction; 17593ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1760f1580f4eSBarry Smith } 1761f1580f4eSBarry Smith 1762f1580f4eSBarry Smith /*@ 1763e31fc274SPierre Jolivet PCHPDDMSetSTShareSubKSP - Sets whether the `KSP` in SLEPc `ST` and the fine-level subdomain solver should be shared. 1764e31fc274SPierre Jolivet 1765e31fc274SPierre Jolivet Input Parameters: 1766e31fc274SPierre Jolivet + pc - preconditioner context 1767e31fc274SPierre Jolivet - share - whether the `KSP` should be shared or not 1768e31fc274SPierre Jolivet 1769e31fc274SPierre Jolivet Note: 1770e31fc274SPierre Jolivet This is not the same as `PCSetReusePreconditioner()`. Given certain conditions (visible using -info), a symbolic factorization can be skipped 1771e31fc274SPierre Jolivet when using a subdomain `PCType` such as `PCLU` or `PCCHOLESKY`. 1772e31fc274SPierre Jolivet 1773e31fc274SPierre Jolivet Level: advanced 1774e31fc274SPierre Jolivet 1775e31fc274SPierre Jolivet .seealso: `PCHPDDM`, `PCHPDDMGetSTShareSubKSP()` 1776e31fc274SPierre Jolivet @*/ 1777e31fc274SPierre Jolivet PetscErrorCode PCHPDDMSetSTShareSubKSP(PC pc, PetscBool share) 1778e31fc274SPierre Jolivet { 1779e31fc274SPierre Jolivet PetscFunctionBegin; 1780e31fc274SPierre Jolivet PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1781e31fc274SPierre Jolivet PetscTryMethod(pc, "PCHPDDMSetSTShareSubKSP_C", (PC, PetscBool), (pc, share)); 17823ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1783e31fc274SPierre Jolivet } 1784e31fc274SPierre Jolivet 1785e31fc274SPierre Jolivet /*@ 1786f1580f4eSBarry Smith PCHPDDMGetSTShareSubKSP - Gets whether the `KSP` in SLEPc `ST` and the fine-level subdomain solver is shared. 1787f1580f4eSBarry Smith 1788f1580f4eSBarry Smith Input Parameter: 1789f1580f4eSBarry Smith . pc - preconditioner context 1790f1580f4eSBarry Smith 1791f1580f4eSBarry Smith Output Parameter: 1792f1580f4eSBarry Smith . share - whether the `KSP` is shared or not 1793f1580f4eSBarry Smith 1794f1580f4eSBarry Smith Note: 1795f1580f4eSBarry 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 1796f1580f4eSBarry Smith when using a subdomain `PCType` such as `PCLU` or `PCCHOLESKY`. 1797f1580f4eSBarry Smith 1798f1580f4eSBarry Smith Level: advanced 1799f1580f4eSBarry Smith 1800e31fc274SPierre Jolivet .seealso: `PCHPDDM`, `PCHPDDMSetSTShareSubKSP()` 1801f1580f4eSBarry Smith @*/ 1802d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMGetSTShareSubKSP(PC pc, PetscBool *share) 1803d71ae5a4SJacob Faibussowitsch { 1804f1580f4eSBarry Smith PetscFunctionBegin; 1805f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1806f1580f4eSBarry Smith if (share) { 18074f572ea9SToby Isaac PetscAssertPointer(share, 2); 1808f1580f4eSBarry Smith PetscUseMethod(pc, "PCHPDDMGetSTShareSubKSP_C", (PC, PetscBool *), (pc, share)); 1809f1580f4eSBarry Smith } 18103ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1811f1580f4eSBarry Smith } 1812f1580f4eSBarry Smith 1813e31fc274SPierre Jolivet static PetscErrorCode PCHPDDMSetSTShareSubKSP_HPDDM(PC pc, PetscBool share) 1814e31fc274SPierre Jolivet { 1815e31fc274SPierre Jolivet PC_HPDDM *data = (PC_HPDDM *)pc->data; 1816e31fc274SPierre Jolivet 1817e31fc274SPierre Jolivet PetscFunctionBegin; 1818e31fc274SPierre Jolivet data->share = share; 18193ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1820e31fc274SPierre Jolivet } 1821e31fc274SPierre Jolivet 1822d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMGetSTShareSubKSP_HPDDM(PC pc, PetscBool *share) 1823d71ae5a4SJacob Faibussowitsch { 1824f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 1825f1580f4eSBarry Smith 1826f1580f4eSBarry Smith PetscFunctionBegin; 1827f1580f4eSBarry Smith *share = data->share; 18283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1829f1580f4eSBarry Smith } 1830f1580f4eSBarry Smith 1831f1580f4eSBarry Smith /*@ 1832f1580f4eSBarry Smith PCHPDDMSetDeflationMat - Sets the deflation space used to assemble a coarser operator. 1833f1580f4eSBarry Smith 1834f1580f4eSBarry Smith Input Parameters: 1835f1580f4eSBarry Smith + pc - preconditioner context 1836f1580f4eSBarry Smith . is - index set of the local deflation matrix 1837f1580f4eSBarry Smith - U - deflation sequential matrix stored as a `MATSEQDENSE` 1838f1580f4eSBarry Smith 1839f1580f4eSBarry Smith Level: advanced 1840f1580f4eSBarry Smith 1841f1580f4eSBarry Smith .seealso: `PCHPDDM`, `PCDeflationSetSpace()`, `PCMGSetRestriction()` 1842f1580f4eSBarry Smith @*/ 1843d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetDeflationMat(PC pc, IS is, Mat U) 1844d71ae5a4SJacob Faibussowitsch { 1845f1580f4eSBarry Smith PetscFunctionBegin; 1846f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1847f1580f4eSBarry Smith PetscValidHeaderSpecific(is, IS_CLASSID, 2); 1848f1580f4eSBarry Smith PetscValidHeaderSpecific(U, MAT_CLASSID, 3); 1849e31fc274SPierre Jolivet PetscTryMethod(pc, "PCHPDDMSetDeflationMat_C", (PC, IS, Mat), (pc, is, U)); 18503ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1851f1580f4eSBarry Smith } 1852f1580f4eSBarry Smith 1853d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetDeflationMat_HPDDM(PC pc, IS is, Mat U) 1854d71ae5a4SJacob Faibussowitsch { 1855f1580f4eSBarry Smith PetscFunctionBegin; 1856f1580f4eSBarry Smith PetscCall(PCHPDDMSetAuxiliaryMat_Private(pc, is, U, PETSC_TRUE)); 18573ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1858f1580f4eSBarry Smith } 1859f1580f4eSBarry Smith 1860d71ae5a4SJacob Faibussowitsch PetscErrorCode HPDDMLoadDL_Private(PetscBool *found) 1861d71ae5a4SJacob Faibussowitsch { 1862605ad303SPierre Jolivet PetscBool flg; 1863f1580f4eSBarry Smith char lib[PETSC_MAX_PATH_LEN], dlib[PETSC_MAX_PATH_LEN], dir[PETSC_MAX_PATH_LEN]; 1864f1580f4eSBarry Smith 1865f1580f4eSBarry Smith PetscFunctionBegin; 18664f572ea9SToby Isaac PetscAssertPointer(found, 1); 1867c6a7a370SJeremy L Thompson PetscCall(PetscStrncpy(dir, "${PETSC_LIB_DIR}", sizeof(dir))); 1868db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetString(nullptr, nullptr, "-hpddm_dir", dir, sizeof(dir), nullptr)); 1869f1580f4eSBarry Smith PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/libhpddm_petsc", dir)); 1870f1580f4eSBarry Smith PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found)); 1871605ad303SPierre Jolivet #if defined(SLEPC_LIB_DIR) /* this variable is passed during SLEPc ./configure when PETSc has not been configured */ 1872605ad303SPierre Jolivet if (!*found) { /* with --download-hpddm since slepcconf.h is not yet built (and thus can't be included) */ 1873c6a7a370SJeremy L Thompson PetscCall(PetscStrncpy(dir, HPDDM_STR(SLEPC_LIB_DIR), sizeof(dir))); 1874f1580f4eSBarry Smith PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/libhpddm_petsc", dir)); 1875f1580f4eSBarry Smith PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found)); 1876f1580f4eSBarry Smith } 1877f1580f4eSBarry Smith #endif 1878605ad303SPierre 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 */ 1879605ad303SPierre Jolivet PetscCall(PetscOptionsGetenv(PETSC_COMM_SELF, "SLEPC_DIR", dir, sizeof(dir), &flg)); 1880605ad303SPierre 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 */ 1881605ad303SPierre Jolivet PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/lib/libslepc", dir)); 1882605ad303SPierre Jolivet PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found)); 1883605ad303SPierre Jolivet PetscCheck(*found, PETSC_COMM_SELF, PETSC_ERR_PLIB, "%s not found but SLEPC_DIR=%s", lib, dir); 1884605ad303SPierre Jolivet PetscCall(PetscDLLibraryAppend(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, dlib)); 1885605ad303SPierre Jolivet PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/lib/libhpddm_petsc", dir)); /* libhpddm_petsc is always in the same directory as libslepc */ 1886605ad303SPierre Jolivet PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found)); 1887605ad303SPierre Jolivet } 1888605ad303SPierre Jolivet } 1889f1580f4eSBarry Smith PetscCheck(*found, PETSC_COMM_SELF, PETSC_ERR_PLIB, "%s not found", lib); 1890f1580f4eSBarry Smith PetscCall(PetscDLLibraryAppend(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, dlib)); 18913ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1892f1580f4eSBarry Smith } 1893f1580f4eSBarry Smith 1894f1580f4eSBarry Smith /*MC 1895f1580f4eSBarry Smith PCHPDDM - Interface with the HPDDM library. 1896f1580f4eSBarry Smith 1897f1580f4eSBarry Smith This `PC` may be used to build multilevel spectral domain decomposition methods based on the GenEO framework [2011, 2019]. It may be viewed as an alternative to spectral 1898f1580f4eSBarry Smith AMGe or `PCBDDC` with adaptive selection of constraints. The interface is explained in details in [2021] (see references below) 1899f1580f4eSBarry Smith 19007eb095acSPierre Jolivet The matrix to be preconditioned (Pmat) may be unassembled (`MATIS`), assembled (`MATAIJ`, `MATBAIJ`, or `MATSBAIJ`), hierarchical (`MATHTOOL`), or `MATNORMAL`. 1901f1580f4eSBarry Smith 1902f1580f4eSBarry Smith For multilevel preconditioning, when using an assembled or hierarchical Pmat, one must provide an auxiliary local `Mat` (unassembled local operator for GenEO) using 1903f1580f4eSBarry Smith `PCHPDDMSetAuxiliaryMat()`. Calling this routine is not needed when using a `MATIS` Pmat, assembly is done internally using `MatConvert()`. 1904f1580f4eSBarry Smith 1905f1580f4eSBarry Smith Options Database Keys: 1906f1580f4eSBarry Smith + -pc_hpddm_define_subdomains <true, default=false> - on the finest level, calls `PCASMSetLocalSubdomains()` with the `IS` supplied in `PCHPDDMSetAuxiliaryMat()` 1907f1580f4eSBarry Smith (not relevant with an unassembled Pmat) 1908f1580f4eSBarry Smith . -pc_hpddm_has_neumann <true, default=false> - on the finest level, informs the `PC` that the local Neumann matrix is supplied in `PCHPDDMSetAuxiliaryMat()` 1909f1580f4eSBarry Smith - -pc_hpddm_coarse_correction <type, default=deflated> - determines the `PCHPDDMCoarseCorrectionType` when calling `PCApply()` 1910f1580f4eSBarry Smith 191138bf2a8cSPierre Jolivet Options for subdomain solvers, subdomain eigensolvers (for computing deflation vectors), and the coarse solver can be set using the following options database prefixes. 1912f1580f4eSBarry Smith .vb 1913f1580f4eSBarry Smith -pc_hpddm_levels_%d_pc_ 1914f1580f4eSBarry Smith -pc_hpddm_levels_%d_ksp_ 1915f1580f4eSBarry Smith -pc_hpddm_levels_%d_eps_ 1916f1580f4eSBarry Smith -pc_hpddm_levels_%d_p 19174ec2a359SPierre Jolivet -pc_hpddm_levels_%d_mat_type 1918f1580f4eSBarry Smith -pc_hpddm_coarse_ 1919f1580f4eSBarry Smith -pc_hpddm_coarse_p 19204ec2a359SPierre Jolivet -pc_hpddm_coarse_mat_type 19212ce66baaSPierre Jolivet -pc_hpddm_coarse_mat_filter 1922f1580f4eSBarry Smith .ve 1923f1580f4eSBarry Smith 192438bf2a8cSPierre 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 1925f1580f4eSBarry Smith -pc_hpddm_coarse_p 2 -pc_hpddm_coarse_mat_type baij will use 10 deflation vectors per subdomain on the fine "level 1", 1926f1580f4eSBarry Smith aggregate the fine subdomains into 4 "level 2" subdomains, then use 10 deflation vectors per subdomain on "level 2", 19277eb095acSPierre Jolivet and assemble the coarse matrix (of dimension 4 x 10 = 40) on two processes as a `MATBAIJ` (default is `MATSBAIJ`). 1928f1580f4eSBarry Smith 1929f1580f4eSBarry 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. 1930f1580f4eSBarry Smith 193138bf2a8cSPierre Jolivet This preconditioner requires that you build PETSc with SLEPc (``--download-slepc``). By default, the underlying concurrent eigenproblems 193238bf2a8cSPierre Jolivet are solved using SLEPc shift-and-invert spectral transformation. This is usually what gives the best performance for GenEO, cf. [2011, 2013]. As 193338bf2a8cSPierre 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 193438bf2a8cSPierre 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 193538bf2a8cSPierre Jolivet SLEPc documentation since they are specific to `PCHPDDM`. 193638bf2a8cSPierre Jolivet .vb 193738bf2a8cSPierre Jolivet -pc_hpddm_levels_1_st_share_sub_ksp 193838bf2a8cSPierre Jolivet -pc_hpddm_levels_%d_eps_threshold 193938bf2a8cSPierre Jolivet -pc_hpddm_levels_1_eps_use_inertia 194038bf2a8cSPierre Jolivet .ve 194138bf2a8cSPierre Jolivet 194238bf2a8cSPierre Jolivet The first option from the list only applies to the fine-level eigensolver, see `PCHPDDMSetSTShareSubKSP()`. The second option from the list is 194338bf2a8cSPierre Jolivet used to filter eigenmodes retrieved after convergence of `EPSSolve()` at "level N" such that eigenvectors used to define a "level N+1" coarse 194438bf2a8cSPierre 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 194538bf2a8cSPierre 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 194638bf2a8cSPierre 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 194738bf2a8cSPierre Jolivet to supply -pc_hpddm_levels_1_eps_nev. This last option also only applies to the fine-level (N = 1) eigensolver. 1948f1580f4eSBarry Smith 1949f1580f4eSBarry Smith References: 1950f1580f4eSBarry Smith + 2011 - A robust two-level domain decomposition preconditioner for systems of PDEs. Spillane, Dolean, Hauret, Nataf, Pechstein, and Scheichl. Comptes Rendus Mathematique. 1951f1580f4eSBarry Smith . 2013 - Scalable domain decomposition preconditioners for heterogeneous elliptic problems. Jolivet, Hecht, Nataf, and Prud'homme. SC13. 1952f1580f4eSBarry Smith . 2015 - An introduction to domain decomposition methods: algorithms, theory, and parallel implementation. Dolean, Jolivet, and Nataf. SIAM. 1953f1580f4eSBarry Smith . 2019 - A multilevel Schwarz preconditioner based on a hierarchy of robust coarse spaces. Al Daas, Grigori, Jolivet, and Tournier. SIAM Journal on Scientific Computing. 1954f1580f4eSBarry Smith . 2021 - KSPHPDDM and PCHPDDM: extending PETSc with advanced Krylov methods and robust multilevel overlapping Schwarz preconditioners. Jolivet, Roman, and Zampini. Computer & Mathematics with Applications. 1955f1580f4eSBarry Smith . 2022a - A robust algebraic domain decomposition preconditioner for sparse normal equations. Al Daas, Jolivet, and Scott. SIAM Journal on Scientific Computing. 1956f1580f4eSBarry Smith - 2022b - A robust algebraic multilevel domain decomposition preconditioner for sparse symmetric positive definite matrices. Al Daas and Jolivet. 1957f1580f4eSBarry Smith 1958f1580f4eSBarry Smith Level: intermediate 1959f1580f4eSBarry Smith 1960f1580f4eSBarry Smith .seealso: `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHPDDMSetAuxiliaryMat()`, `MATIS`, `PCBDDC`, `PCDEFLATION`, `PCTELESCOPE`, `PCASM`, 1961e31fc274SPierre Jolivet `PCHPDDMSetCoarseCorrectionType()`, `PCHPDDMHasNeumannMat()`, `PCHPDDMSetRHSMat()`, `PCHPDDMSetDeflationMat()`, `PCHPDDMSetSTShareSubKSP()`, 1962e31fc274SPierre Jolivet `PCHPDDMGetSTShareSubKSP()`, `PCHPDDMGetCoarseCorrectionType()`, `PCHPDDMGetComplexities()` 1963f1580f4eSBarry Smith M*/ 1964d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode PCCreate_HPDDM(PC pc) 1965d71ae5a4SJacob Faibussowitsch { 1966f1580f4eSBarry Smith PC_HPDDM *data; 1967f1580f4eSBarry Smith PetscBool found; 1968f1580f4eSBarry Smith 1969f1580f4eSBarry Smith PetscFunctionBegin; 1970f1580f4eSBarry Smith if (!loadedSym) { 1971f1580f4eSBarry Smith PetscCall(HPDDMLoadDL_Private(&found)); 1972db4a47b3SPierre Jolivet if (found) PetscCall(PetscDLLibrarySym(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, nullptr, "PCHPDDM_Internal", (void **)&loadedSym)); 1973f1580f4eSBarry Smith } 1974f1580f4eSBarry Smith PetscCheck(loadedSym, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCHPDDM_Internal symbol not found in loaded libhpddm_petsc"); 19754dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&data)); 1976f1580f4eSBarry Smith pc->data = data; 1977c8ea6600SPierre Jolivet data->Neumann = PETSC_BOOL3_UNKNOWN; 1978f1580f4eSBarry Smith pc->ops->reset = PCReset_HPDDM; 1979f1580f4eSBarry Smith pc->ops->destroy = PCDestroy_HPDDM; 1980f1580f4eSBarry Smith pc->ops->setfromoptions = PCSetFromOptions_HPDDM; 1981f1580f4eSBarry Smith pc->ops->setup = PCSetUp_HPDDM; 1982f1580f4eSBarry Smith pc->ops->apply = PCApply_HPDDM; 1983f1580f4eSBarry Smith pc->ops->matapply = PCMatApply_HPDDM; 1984f1580f4eSBarry Smith pc->ops->view = PCView_HPDDM; 1985f1580f4eSBarry Smith pc->ops->presolve = PCPreSolve_HPDDM; 1986f1580f4eSBarry Smith 1987f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetAuxiliaryMat_C", PCHPDDMSetAuxiliaryMat_HPDDM)); 1988f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMHasNeumannMat_C", PCHPDDMHasNeumannMat_HPDDM)); 1989f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetRHSMat_C", PCHPDDMSetRHSMat_HPDDM)); 1990f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetCoarseCorrectionType_C", PCHPDDMSetCoarseCorrectionType_HPDDM)); 1991f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetCoarseCorrectionType_C", PCHPDDMGetCoarseCorrectionType_HPDDM)); 1992e31fc274SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetSTShareSubKSP_C", PCHPDDMSetSTShareSubKSP_HPDDM)); 1993f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetSTShareSubKSP_C", PCHPDDMGetSTShareSubKSP_HPDDM)); 1994f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetDeflationMat_C", PCHPDDMSetDeflationMat_HPDDM)); 19953ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1996f1580f4eSBarry Smith } 1997f1580f4eSBarry Smith 1998f1580f4eSBarry Smith /*@C 1999f1580f4eSBarry Smith PCHPDDMInitializePackage - This function initializes everything in the `PCHPDDM` package. It is called from `PCInitializePackage()`. 2000f1580f4eSBarry Smith 2001f1580f4eSBarry Smith Level: developer 2002f1580f4eSBarry Smith 2003f1580f4eSBarry Smith .seealso: `PetscInitialize()` 2004f1580f4eSBarry Smith @*/ 2005d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMInitializePackage(void) 2006d71ae5a4SJacob Faibussowitsch { 2007f1580f4eSBarry Smith char ename[32]; 2008f1580f4eSBarry Smith PetscInt i; 2009f1580f4eSBarry Smith 2010f1580f4eSBarry Smith PetscFunctionBegin; 20113ba16761SJacob Faibussowitsch if (PCHPDDMPackageInitialized) PetscFunctionReturn(PETSC_SUCCESS); 2012f1580f4eSBarry Smith PCHPDDMPackageInitialized = PETSC_TRUE; 2013f1580f4eSBarry Smith PetscCall(PetscRegisterFinalize(PCHPDDMFinalizePackage)); 2014f1580f4eSBarry Smith /* general events registered once during package initialization */ 2015f1580f4eSBarry Smith /* some of these events are not triggered in libpetsc, */ 2016f1580f4eSBarry Smith /* but rather directly in libhpddm_petsc, */ 2017f1580f4eSBarry Smith /* which is in charge of performing the following operations */ 2018f1580f4eSBarry Smith 2019f1580f4eSBarry Smith /* domain decomposition structure from Pmat sparsity pattern */ 2020f1580f4eSBarry Smith PetscCall(PetscLogEventRegister("PCHPDDMStrc", PC_CLASSID, &PC_HPDDM_Strc)); 2021f1580f4eSBarry Smith /* Galerkin product, redistribution, and setup (not triggered in libpetsc) */ 2022f1580f4eSBarry Smith PetscCall(PetscLogEventRegister("PCHPDDMPtAP", PC_CLASSID, &PC_HPDDM_PtAP)); 2023f1580f4eSBarry Smith /* Galerkin product with summation, redistribution, and setup (not triggered in libpetsc) */ 2024f1580f4eSBarry Smith PetscCall(PetscLogEventRegister("PCHPDDMPtBP", PC_CLASSID, &PC_HPDDM_PtBP)); 2025f1580f4eSBarry Smith /* next level construction using PtAP and PtBP (not triggered in libpetsc) */ 2026f1580f4eSBarry Smith PetscCall(PetscLogEventRegister("PCHPDDMNext", PC_CLASSID, &PC_HPDDM_Next)); 2027f1580f4eSBarry Smith static_assert(PETSC_PCHPDDM_MAXLEVELS <= 9, "PETSC_PCHPDDM_MAXLEVELS value is too high"); 2028f1580f4eSBarry Smith for (i = 1; i < PETSC_PCHPDDM_MAXLEVELS; ++i) { 2029f1580f4eSBarry Smith PetscCall(PetscSNPrintf(ename, sizeof(ename), "PCHPDDMSetUp L%1" PetscInt_FMT, i)); 2030f1580f4eSBarry Smith /* events during a PCSetUp() at level #i _except_ the assembly */ 2031f1580f4eSBarry Smith /* of the Galerkin operator of the coarser level #(i + 1) */ 2032f1580f4eSBarry Smith PetscCall(PetscLogEventRegister(ename, PC_CLASSID, &PC_HPDDM_SetUp[i - 1])); 2033f1580f4eSBarry Smith PetscCall(PetscSNPrintf(ename, sizeof(ename), "PCHPDDMSolve L%1" PetscInt_FMT, i)); 2034f1580f4eSBarry Smith /* events during a PCApply() at level #i _except_ */ 2035f1580f4eSBarry Smith /* the KSPSolve() of the coarser level #(i + 1) */ 2036f1580f4eSBarry Smith PetscCall(PetscLogEventRegister(ename, PC_CLASSID, &PC_HPDDM_Solve[i - 1])); 2037f1580f4eSBarry Smith } 20383ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2039f1580f4eSBarry Smith } 2040f1580f4eSBarry Smith 2041f1580f4eSBarry Smith /*@C 2042f1580f4eSBarry Smith PCHPDDMFinalizePackage - This function frees everything from the `PCHPDDM` package. It is called from `PetscFinalize()`. 2043f1580f4eSBarry Smith 2044f1580f4eSBarry Smith Level: developer 2045f1580f4eSBarry Smith 2046f1580f4eSBarry Smith .seealso: `PetscFinalize()` 2047f1580f4eSBarry Smith @*/ 2048d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMFinalizePackage(void) 2049d71ae5a4SJacob Faibussowitsch { 2050f1580f4eSBarry Smith PetscFunctionBegin; 2051f1580f4eSBarry Smith PCHPDDMPackageInitialized = PETSC_FALSE; 20523ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2053f1580f4eSBarry Smith } 2054