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