101e3c840SPierre Jolivet #include <petscsf.h> 213044ca3SPierre Jolivet #include <petsc/private/vecimpl.h> 3f1580f4eSBarry Smith #include <petsc/private/matimpl.h> 4f1580f4eSBarry Smith #include <petsc/private/petschpddm.h> /*I "petscpc.h" I*/ 5e905f78bSJacob Faibussowitsch #include <petsc/private/pcimpl.h> 6e905f78bSJacob Faibussowitsch #include <petsc/private/dmimpl.h> /* this must be included after petschpddm.h so that DM_MAX_WORK_VECTORS is not defined */ 7f1580f4eSBarry Smith /* otherwise, it is assumed that one is compiling libhpddm_petsc => circular dependency */ 8f1580f4eSBarry Smith 9db4a47b3SPierre Jolivet static PetscErrorCode (*loadedSym)(HPDDM::Schwarz<PetscScalar> *const, IS, Mat, Mat, Mat, std::vector<Vec>, PC_HPDDM_Level **const) = nullptr; 10f1580f4eSBarry Smith 11f1580f4eSBarry Smith static PetscBool PCHPDDMPackageInitialized = PETSC_FALSE; 12f1580f4eSBarry Smith 13f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_Strc; 14f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_PtAP; 15f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_PtBP; 16f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_Next; 17f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_SetUp[PETSC_PCHPDDM_MAXLEVELS]; 18f1580f4eSBarry Smith PetscLogEvent PC_HPDDM_Solve[PETSC_PCHPDDM_MAXLEVELS]; 19f1580f4eSBarry Smith 20aa1539e9SPierre Jolivet const char *const PCHPDDMCoarseCorrectionTypes[] = {"DEFLATED", "ADDITIVE", "BALANCED", "NONE", "PCHPDDMCoarseCorrectionType", "PC_HPDDM_COARSE_CORRECTION_", nullptr}; 2113044ca3SPierre Jolivet const char *const PCHPDDMSchurPreTypes[] = {"LEAST_SQUARES", "GENEO", "PCHPDDMSchurPreType", "PC_HPDDM_SCHUR_PRE", nullptr}; 22f1580f4eSBarry Smith 23d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCReset_HPDDM(PC pc) 24d71ae5a4SJacob Faibussowitsch { 25f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 26f1580f4eSBarry Smith 27f1580f4eSBarry Smith PetscFunctionBegin; 28f1580f4eSBarry Smith if (data->levels) { 29811e8887SPierre Jolivet for (PetscInt i = 0; i < PETSC_PCHPDDM_MAXLEVELS && data->levels[i]; ++i) { 30f1580f4eSBarry Smith PetscCall(KSPDestroy(&data->levels[i]->ksp)); 31f1580f4eSBarry Smith PetscCall(PCDestroy(&data->levels[i]->pc)); 32f1580f4eSBarry Smith PetscCall(PetscFree(data->levels[i])); 33f1580f4eSBarry Smith } 34f1580f4eSBarry Smith PetscCall(PetscFree(data->levels)); 35f1580f4eSBarry Smith } 36f1580f4eSBarry Smith PetscCall(ISDestroy(&data->is)); 37f1580f4eSBarry Smith PetscCall(MatDestroy(&data->aux)); 38f1580f4eSBarry Smith PetscCall(MatDestroy(&data->B)); 39f1580f4eSBarry Smith PetscCall(VecDestroy(&data->normal)); 40f1580f4eSBarry Smith data->correction = PC_HPDDM_COARSE_CORRECTION_DEFLATED; 41c8ea6600SPierre Jolivet data->Neumann = PETSC_BOOL3_UNKNOWN; 42f1580f4eSBarry Smith data->deflation = PETSC_FALSE; 43db4a47b3SPierre Jolivet data->setup = nullptr; 44db4a47b3SPierre Jolivet data->setup_ctx = nullptr; 453ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 46f1580f4eSBarry Smith } 47f1580f4eSBarry Smith 48d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCDestroy_HPDDM(PC pc) 49d71ae5a4SJacob Faibussowitsch { 50f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 51f1580f4eSBarry Smith 52f1580f4eSBarry Smith PetscFunctionBegin; 53f1580f4eSBarry Smith PetscCall(PCReset_HPDDM(pc)); 54f1580f4eSBarry Smith PetscCall(PetscFree(data)); 55db4a47b3SPierre Jolivet PetscCall(PetscObjectChangeTypeName((PetscObject)pc, nullptr)); 56db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetAuxiliaryMat_C", nullptr)); 57db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMHasNeumannMat_C", nullptr)); 58db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetRHSMat_C", nullptr)); 59db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetCoarseCorrectionType_C", nullptr)); 60db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetCoarseCorrectionType_C", nullptr)); 61db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetSTShareSubKSP_C", nullptr)); 62db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetSTShareSubKSP_C", nullptr)); 63db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetDeflationMat_C", nullptr)); 6413044ca3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)pc, "_PCHPDDM_Schur", nullptr)); 653ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 66f1580f4eSBarry Smith } 67f1580f4eSBarry Smith 68d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMSetAuxiliaryMat_Private(PC pc, IS is, Mat A, PetscBool deflation) 69d71ae5a4SJacob Faibussowitsch { 70f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 71cdbd50ebSPierre Jolivet PCHPDDMCoarseCorrectionType type = data->correction; 72f1580f4eSBarry Smith 73f1580f4eSBarry Smith PetscFunctionBegin; 74c30dab7bSPierre Jolivet PetscValidLogicalCollectiveBool(pc, deflation, 4); 75c30dab7bSPierre Jolivet if (is && A) { 76c30dab7bSPierre Jolivet PetscInt m[2]; 77c30dab7bSPierre Jolivet 78c30dab7bSPierre Jolivet PetscCall(ISGetLocalSize(is, m)); 79c30dab7bSPierre Jolivet PetscCall(MatGetLocalSize(A, m + 1, nullptr)); 80c30dab7bSPierre Jolivet PetscCheck(m[0] == m[1], PETSC_COMM_SELF, PETSC_ERR_USER_INPUT, "Inconsistent IS and Mat sizes (%" PetscInt_FMT " v. %" PetscInt_FMT ")", m[0], m[1]); 81c30dab7bSPierre Jolivet } 82f1580f4eSBarry Smith if (is) { 83f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)is)); 84f1580f4eSBarry Smith if (data->is) { /* new overlap definition resets the PC */ 85f1580f4eSBarry Smith PetscCall(PCReset_HPDDM(pc)); 86f1580f4eSBarry Smith pc->setfromoptionscalled = 0; 87371d2eb7SMartin Diehl pc->setupcalled = PETSC_FALSE; 88cdbd50ebSPierre Jolivet data->correction = type; 89f1580f4eSBarry Smith } 90f1580f4eSBarry Smith PetscCall(ISDestroy(&data->is)); 91f1580f4eSBarry Smith data->is = is; 92f1580f4eSBarry Smith } 93f1580f4eSBarry Smith if (A) { 94f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)A)); 95f1580f4eSBarry Smith PetscCall(MatDestroy(&data->aux)); 96f1580f4eSBarry Smith data->aux = A; 97f1580f4eSBarry Smith } 98f1580f4eSBarry Smith data->deflation = deflation; 993ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 100f1580f4eSBarry Smith } 101f1580f4eSBarry Smith 102281f8ce6SPierre Jolivet static inline PetscErrorCode PCHPDDMSplittingMatNormal_Private(Mat A, IS *is, Mat *splitting[]) 103281f8ce6SPierre Jolivet { 104281f8ce6SPierre Jolivet Mat *sub; 105281f8ce6SPierre Jolivet IS zero; 106281f8ce6SPierre Jolivet 107281f8ce6SPierre Jolivet PetscFunctionBegin; 108281f8ce6SPierre Jolivet PetscCall(MatSetOption(A, MAT_SUBMAT_SINGLEIS, PETSC_TRUE)); 109281f8ce6SPierre Jolivet PetscCall(MatCreateSubMatrices(A, 1, is + 2, is, MAT_INITIAL_MATRIX, splitting)); 110281f8ce6SPierre Jolivet PetscCall(MatCreateSubMatrices(**splitting, 1, is + 2, is + 1, MAT_INITIAL_MATRIX, &sub)); 111281f8ce6SPierre Jolivet PetscCall(MatFindZeroRows(*sub, &zero)); 112281f8ce6SPierre Jolivet PetscCall(MatDestroySubMatrices(1, &sub)); 113281f8ce6SPierre Jolivet PetscCall(MatSetOption(**splitting, MAT_KEEP_NONZERO_PATTERN, PETSC_TRUE)); 114281f8ce6SPierre Jolivet PetscCall(MatZeroRowsIS(**splitting, zero, 0.0, nullptr, nullptr)); 115281f8ce6SPierre Jolivet PetscCall(ISDestroy(&zero)); 116281f8ce6SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 117281f8ce6SPierre Jolivet } 118281f8ce6SPierre Jolivet 119281f8ce6SPierre Jolivet static inline PetscErrorCode PCHPDDMSetAuxiliaryMatNormal_Private(PC pc, Mat A, Mat N, Mat *B, const char *pcpre, Vec *diagonal = nullptr, Mat B01 = nullptr) 120d71ae5a4SJacob Faibussowitsch { 121f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 122281f8ce6SPierre Jolivet Mat *splitting[2] = {}, aux; 1233df4cd7bSPierre Jolivet Vec d; 124281f8ce6SPierre Jolivet IS is[3]; 125f1580f4eSBarry Smith PetscReal norm; 126f1580f4eSBarry Smith PetscBool flg; 127f1580f4eSBarry Smith char type[256] = {}; /* same size as in src/ksp/pc/interface/pcset.c */ 128f1580f4eSBarry Smith 129f1580f4eSBarry Smith PetscFunctionBegin; 130281f8ce6SPierre Jolivet if (!B01) PetscCall(MatConvert(N, MATAIJ, MAT_INITIAL_MATRIX, B)); 131281f8ce6SPierre Jolivet else PetscCall(MatTransposeMatMult(B01, A, MAT_INITIAL_MATRIX, PETSC_DETERMINE, B)); 132feebddf4SPierre Jolivet PetscCall(MatEliminateZeros(*B, PETSC_TRUE)); 133281f8ce6SPierre Jolivet PetscCall(ISCreateStride(PETSC_COMM_SELF, A->cmap->n, A->cmap->rstart, 1, is)); 134281f8ce6SPierre Jolivet PetscCall(MatIncreaseOverlap(*B, 1, is, 1)); 135281f8ce6SPierre Jolivet PetscCall(ISCreateStride(PETSC_COMM_SELF, A->cmap->n, A->cmap->rstart, 1, is + 2)); 136281f8ce6SPierre Jolivet PetscCall(ISEmbed(is[0], is[2], PETSC_TRUE, is + 1)); 137281f8ce6SPierre Jolivet PetscCall(ISDestroy(is + 2)); 138281f8ce6SPierre Jolivet PetscCall(ISCreateStride(PETSC_COMM_SELF, A->rmap->N, 0, 1, is + 2)); 139281f8ce6SPierre Jolivet PetscCall(PCHPDDMSplittingMatNormal_Private(A, is, &splitting[0])); 140281f8ce6SPierre Jolivet if (B01) { 141281f8ce6SPierre Jolivet PetscCall(PCHPDDMSplittingMatNormal_Private(B01, is, &splitting[1])); 142281f8ce6SPierre Jolivet PetscCall(MatDestroy(&B01)); 143281f8ce6SPierre Jolivet } 144281f8ce6SPierre Jolivet PetscCall(ISDestroy(is + 2)); 145281f8ce6SPierre Jolivet PetscCall(ISDestroy(is + 1)); 146db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetString(nullptr, pcpre, "-pc_hpddm_levels_1_sub_pc_type", type, sizeof(type), nullptr)); 147f1580f4eSBarry Smith PetscCall(PetscStrcmp(type, PCQR, &flg)); 148f1580f4eSBarry Smith if (!flg) { 149281f8ce6SPierre Jolivet Mat conjugate = *splitting[splitting[1] ? 1 : 0]; 150811e8887SPierre Jolivet 151281f8ce6SPierre Jolivet if (PetscDefined(USE_COMPLEX) && !splitting[1]) { 152281f8ce6SPierre Jolivet PetscCall(MatDuplicate(*splitting[0], MAT_COPY_VALUES, &conjugate)); 153f1580f4eSBarry Smith PetscCall(MatConjugate(conjugate)); 154f1580f4eSBarry Smith } 155281f8ce6SPierre Jolivet PetscCall(MatTransposeMatMult(conjugate, *splitting[0], MAT_INITIAL_MATRIX, PETSC_DETERMINE, &aux)); 156281f8ce6SPierre Jolivet if (PetscDefined(USE_COMPLEX) && !splitting[1]) PetscCall(MatDestroy(&conjugate)); 157281f8ce6SPierre Jolivet else if (splitting[1]) PetscCall(MatDestroySubMatrices(1, &splitting[1])); 158f1580f4eSBarry Smith PetscCall(MatNorm(aux, NORM_FROBENIUS, &norm)); 159f1580f4eSBarry Smith PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE)); 1603df4cd7bSPierre Jolivet if (diagonal) { 161811e8887SPierre Jolivet PetscReal norm; 162811e8887SPierre Jolivet 163feebddf4SPierre Jolivet PetscCall(VecScale(*diagonal, -1.0)); 1643df4cd7bSPierre Jolivet PetscCall(VecNorm(*diagonal, NORM_INFINITY, &norm)); 1653df4cd7bSPierre Jolivet if (norm > PETSC_SMALL) { 16601e3c840SPierre Jolivet PetscSF scatter; 1673df4cd7bSPierre Jolivet PetscInt n; 168811e8887SPierre Jolivet 169281f8ce6SPierre Jolivet PetscCall(ISGetLocalSize(*is, &n)); 1703df4cd7bSPierre Jolivet PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)pc), n, PETSC_DECIDE, &d)); 171281f8ce6SPierre Jolivet PetscCall(VecScatterCreate(*diagonal, *is, d, nullptr, &scatter)); 1723df4cd7bSPierre Jolivet PetscCall(VecScatterBegin(scatter, *diagonal, d, INSERT_VALUES, SCATTER_FORWARD)); 1733df4cd7bSPierre Jolivet PetscCall(VecScatterEnd(scatter, *diagonal, d, INSERT_VALUES, SCATTER_FORWARD)); 17401e3c840SPierre Jolivet PetscCall(PetscSFDestroy(&scatter)); 1753df4cd7bSPierre Jolivet PetscCall(MatDiagonalSet(aux, d, ADD_VALUES)); 1763df4cd7bSPierre Jolivet PetscCall(VecDestroy(&d)); 1773df4cd7bSPierre Jolivet } else PetscCall(VecDestroy(diagonal)); 1783df4cd7bSPierre Jolivet } 1793df4cd7bSPierre Jolivet if (!diagonal) PetscCall(MatShift(aux, PETSC_SMALL * norm)); 180f1580f4eSBarry Smith } else { 181f1580f4eSBarry Smith PetscBool flg; 182811e8887SPierre Jolivet 183281f8ce6SPierre Jolivet PetscCheck(!splitting[1], PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Cannot use PCQR when A01 != A10^T"); 1843df4cd7bSPierre Jolivet if (diagonal) { 1853df4cd7bSPierre Jolivet PetscCall(VecNorm(*diagonal, NORM_INFINITY, &norm)); 1863df4cd7bSPierre Jolivet PetscCheck(norm < PETSC_SMALL, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Nonzero diagonal A11 block"); 1873df4cd7bSPierre Jolivet PetscCall(VecDestroy(diagonal)); 1883df4cd7bSPierre Jolivet } 189f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)N, MATNORMAL, &flg)); 190281f8ce6SPierre Jolivet if (flg) PetscCall(MatCreateNormal(*splitting[0], &aux)); 191281f8ce6SPierre Jolivet else PetscCall(MatCreateNormalHermitian(*splitting[0], &aux)); 192f1580f4eSBarry Smith } 193281f8ce6SPierre Jolivet PetscCall(MatDestroySubMatrices(1, &splitting[0])); 194281f8ce6SPierre Jolivet PetscCall(PCHPDDMSetAuxiliaryMat(pc, *is, aux, nullptr, nullptr)); 195c8ea6600SPierre Jolivet data->Neumann = PETSC_BOOL3_TRUE; 196281f8ce6SPierre Jolivet PetscCall(ISDestroy(is)); 197f1580f4eSBarry Smith PetscCall(MatDestroy(&aux)); 1983ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 199f1580f4eSBarry Smith } 200f1580f4eSBarry Smith 201d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetAuxiliaryMat_HPDDM(PC pc, IS is, Mat A, PetscErrorCode (*setup)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void *setup_ctx) 202d71ae5a4SJacob Faibussowitsch { 203f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 204f1580f4eSBarry Smith 205f1580f4eSBarry Smith PetscFunctionBegin; 206f1580f4eSBarry Smith PetscCall(PCHPDDMSetAuxiliaryMat_Private(pc, is, A, PETSC_FALSE)); 207f1580f4eSBarry Smith if (setup) { 208f1580f4eSBarry Smith data->setup = setup; 209f1580f4eSBarry Smith data->setup_ctx = setup_ctx; 210f1580f4eSBarry Smith } 2113ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 212f1580f4eSBarry Smith } 213f1580f4eSBarry Smith 21470009435SPierre Jolivet /*@C 21504c3f3b8SBarry Smith PCHPDDMSetAuxiliaryMat - Sets the auxiliary matrix used by `PCHPDDM` for the concurrent GenEO problems at the finest level. 216f1580f4eSBarry Smith 217f1580f4eSBarry Smith Input Parameters: 218f1580f4eSBarry Smith + pc - preconditioner context 219f1580f4eSBarry Smith . is - index set of the local auxiliary, e.g., Neumann, matrix 220f1580f4eSBarry Smith . A - auxiliary sequential matrix 22104c3f3b8SBarry Smith . setup - function for generating the auxiliary matrix entries, may be `NULL` 22204c3f3b8SBarry Smith - ctx - context for `setup`, may be `NULL` 22304c3f3b8SBarry Smith 22404c3f3b8SBarry Smith Calling sequence of `setup`: 22504c3f3b8SBarry Smith + J - matrix whose values are to be set 22604c3f3b8SBarry Smith . t - time 22704c3f3b8SBarry Smith . X - linearization point 22804c3f3b8SBarry Smith . X_t - time-derivative of the linearization point 22904c3f3b8SBarry Smith . s - step 23004c3f3b8SBarry Smith . ovl - index set of the local auxiliary, e.g., Neumann, matrix 23104c3f3b8SBarry Smith - ctx - context for `setup`, may be `NULL` 232f1580f4eSBarry Smith 233f1580f4eSBarry Smith Level: intermediate 234f1580f4eSBarry Smith 23504c3f3b8SBarry Smith Note: 23604c3f3b8SBarry Smith As an example, in a finite element context with nonoverlapping subdomains plus (overlapping) ghost elements, this could be the unassembled (Neumann) 23704c3f3b8SBarry Smith local overlapping operator. As opposed to the assembled (Dirichlet) local overlapping operator obtained by summing neighborhood contributions 23804c3f3b8SBarry Smith at the interface of ghost elements. 23904c3f3b8SBarry Smith 24070009435SPierre Jolivet Fortran Notes: 24104c3f3b8SBarry Smith Only `PETSC_NULL_FUNCTION` is supported for `setup` and `ctx` is never accessed 24270009435SPierre Jolivet 243562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHPDDMSetRHSMat()`, `MATIS` 244f1580f4eSBarry Smith @*/ 24504c3f3b8SBarry 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) 246d71ae5a4SJacob Faibussowitsch { 247f1580f4eSBarry Smith PetscFunctionBegin; 248f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 249f1580f4eSBarry Smith if (is) PetscValidHeaderSpecific(is, IS_CLASSID, 2); 250f1580f4eSBarry Smith if (A) PetscValidHeaderSpecific(A, MAT_CLASSID, 3); 25104c3f3b8SBarry Smith PetscTryMethod(pc, "PCHPDDMSetAuxiliaryMat_C", (PC, IS, Mat, PetscErrorCode (*)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void *), (pc, is, A, setup, ctx)); 2523ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 253f1580f4eSBarry Smith } 254f1580f4eSBarry Smith 255d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMHasNeumannMat_HPDDM(PC pc, PetscBool has) 256d71ae5a4SJacob Faibussowitsch { 257f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 258f1580f4eSBarry Smith 259f1580f4eSBarry Smith PetscFunctionBegin; 260c8ea6600SPierre Jolivet data->Neumann = PetscBoolToBool3(has); 2613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 262f1580f4eSBarry Smith } 263f1580f4eSBarry Smith 264f1580f4eSBarry Smith /*@ 265f1580f4eSBarry Smith PCHPDDMHasNeumannMat - Informs `PCHPDDM` that the `Mat` passed to `PCHPDDMSetAuxiliaryMat()` is the local Neumann matrix. 266f1580f4eSBarry Smith 267f1580f4eSBarry Smith Input Parameters: 268f1580f4eSBarry Smith + pc - preconditioner context 269f1580f4eSBarry Smith - has - Boolean value 270f1580f4eSBarry Smith 271f1580f4eSBarry Smith Level: intermediate 272f1580f4eSBarry Smith 273f1580f4eSBarry Smith Notes: 2747eb095acSPierre Jolivet This may be used to bypass a call to `MatCreateSubMatrices()` and to `MatConvert()` for `MATSBAIJ` matrices. 275f1580f4eSBarry Smith 276907a3e9cSStefano 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`. 277f1580f4eSBarry Smith 278562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMSetAuxiliaryMat()` 279f1580f4eSBarry Smith @*/ 280d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMHasNeumannMat(PC pc, PetscBool has) 281d71ae5a4SJacob Faibussowitsch { 282f1580f4eSBarry Smith PetscFunctionBegin; 283f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 284f1580f4eSBarry Smith PetscTryMethod(pc, "PCHPDDMHasNeumannMat_C", (PC, PetscBool), (pc, has)); 2853ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 286f1580f4eSBarry Smith } 287f1580f4eSBarry Smith 288d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetRHSMat_HPDDM(PC pc, Mat B) 289d71ae5a4SJacob Faibussowitsch { 290f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 291f1580f4eSBarry Smith 292f1580f4eSBarry Smith PetscFunctionBegin; 293f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)B)); 294f1580f4eSBarry Smith PetscCall(MatDestroy(&data->B)); 295f1580f4eSBarry Smith data->B = B; 2963ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 297f1580f4eSBarry Smith } 298f1580f4eSBarry Smith 299f1580f4eSBarry Smith /*@ 30004c3f3b8SBarry Smith PCHPDDMSetRHSMat - Sets the right-hand side matrix used by `PCHPDDM` for the concurrent GenEO problems at the finest level. 301f1580f4eSBarry Smith 302f1580f4eSBarry Smith Input Parameters: 303f1580f4eSBarry Smith + pc - preconditioner context 304f1580f4eSBarry Smith - B - right-hand side sequential matrix 305f1580f4eSBarry Smith 306f1580f4eSBarry Smith Level: advanced 307f1580f4eSBarry Smith 30804c3f3b8SBarry Smith Note: 30904c3f3b8SBarry Smith Must be used in conjunction with `PCHPDDMSetAuxiliaryMat`(N), so that Nv = lambda Bv is solved using `EPSSetOperators`(N, B). 31004c3f3b8SBarry 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. 31104c3f3b8SBarry Smith 312562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDMSetAuxiliaryMat()`, `PCHPDDM` 313f1580f4eSBarry Smith @*/ 314d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetRHSMat(PC pc, Mat B) 315d71ae5a4SJacob Faibussowitsch { 316f1580f4eSBarry Smith PetscFunctionBegin; 317f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 318f1580f4eSBarry Smith if (B) { 319f1580f4eSBarry Smith PetscValidHeaderSpecific(B, MAT_CLASSID, 2); 320f1580f4eSBarry Smith PetscTryMethod(pc, "PCHPDDMSetRHSMat_C", (PC, Mat), (pc, B)); 321f1580f4eSBarry Smith } 3223ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 323f1580f4eSBarry Smith } 324f1580f4eSBarry Smith 325ce78bad3SBarry Smith static PetscErrorCode PCSetFromOptions_HPDDM(PC pc, PetscOptionItems PetscOptionsObject) 326d71ae5a4SJacob Faibussowitsch { 327f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 328f1580f4eSBarry Smith PC_HPDDM_Level **levels = data->levels; 329b5a302b3SPierre Jolivet char prefix[256], deprecated[256]; 330f1580f4eSBarry Smith int i = 1; 331f1580f4eSBarry Smith PetscMPIInt size, previous; 3329bb5c669SPierre Jolivet PetscInt n, overlap = 1; 333f1580f4eSBarry Smith PCHPDDMCoarseCorrectionType type; 334c8ea6600SPierre Jolivet PetscBool flg = PETSC_TRUE, set; 335f1580f4eSBarry Smith 336f1580f4eSBarry Smith PetscFunctionBegin; 337f1580f4eSBarry Smith if (!data->levels) { 338f1580f4eSBarry Smith PetscCall(PetscCalloc1(PETSC_PCHPDDM_MAXLEVELS, &levels)); 339f1580f4eSBarry Smith data->levels = levels; 340f1580f4eSBarry Smith } 341f1580f4eSBarry Smith PetscOptionsHeadBegin(PetscOptionsObject, "PCHPDDM options"); 3429bb5c669SPierre Jolivet PetscCall(PetscOptionsBoundedInt("-pc_hpddm_harmonic_overlap", "Overlap prior to computing local harmonic extensions", "PCHPDDM", overlap, &overlap, &set, 1)); 3439bb5c669SPierre Jolivet if (!set) overlap = -1; 344f1580f4eSBarry Smith PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)pc), &size)); 345f1580f4eSBarry Smith previous = size; 346f1580f4eSBarry Smith while (i < PETSC_PCHPDDM_MAXLEVELS) { 347f1580f4eSBarry Smith PetscInt p = 1; 348f1580f4eSBarry Smith 3494dfa11a4SJacob Faibussowitsch if (!data->levels[i - 1]) PetscCall(PetscNew(data->levels + i - 1)); 350f1580f4eSBarry Smith data->levels[i - 1]->parent = data; 351f1580f4eSBarry Smith /* if the previous level has a single process, it is not possible to coarsen further */ 352f1580f4eSBarry Smith if (previous == 1 || !flg) break; 353f1580f4eSBarry Smith data->levels[i - 1]->nu = 0; 354f1580f4eSBarry Smith data->levels[i - 1]->threshold = -1.0; 355f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_nev", i)); 356db4a47b3SPierre 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)); 357b5a302b3SPierre Jolivet PetscCall(PetscSNPrintf(deprecated, sizeof(deprecated), "-pc_hpddm_levels_%d_eps_threshold", i)); 358b5a302b3SPierre Jolivet PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_threshold_absolute", i)); 3598cb7430dSRaphael Zanella PetscCall(PetscOptionsDeprecated(deprecated, prefix, "3.24", nullptr)); 360b5a302b3SPierre Jolivet PetscCall(PetscOptionsReal(prefix, "Local absolute threshold for selecting deflation vectors returned by SLEPc", "PCHPDDM", data->levels[i - 1]->threshold, &data->levels[i - 1]->threshold, nullptr)); 361f1580f4eSBarry Smith if (i == 1) { 362b5a302b3SPierre 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_absolute and -pc_hpddm_harmonic_overlap"); 3639bb5c669SPierre Jolivet if (overlap != -1) { 3649bb5c669SPierre Jolivet PetscInt nsv = 0; 365b5a302b3SPierre Jolivet PetscBool set[2] = {PETSC_FALSE, PETSC_FALSE}; 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)); 369b5a302b3SPierre Jolivet PetscCheck(data->levels[0]->nu == 0 || nsv == 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot supply both -pc_hpddm_levels_1_eps_nev and -pc_hpddm_levels_1_svd_nsv"); 370b5a302b3SPierre Jolivet if (data->levels[0]->nu == 0) { /* -eps_nev has not been used, so nu is 0 */ 371b5a302b3SPierre Jolivet data->levels[0]->nu = nsv; /* nu may still be 0 if -svd_nsv has not been used */ 372b5a302b3SPierre Jolivet PetscCall(PetscSNPrintf(deprecated, sizeof(deprecated), "-pc_hpddm_levels_%d_svd_relative_threshold", i)); 373b5a302b3SPierre Jolivet PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_svd_threshold_relative", i)); 3748cb7430dSRaphael Zanella PetscCall(PetscOptionsDeprecated(deprecated, prefix, "3.24", nullptr)); 375b5a302b3SPierre Jolivet PetscCall(PetscOptionsReal(prefix, "Local relative threshold for selecting deflation vectors returned by SLEPc", "PCHPDDM", data->levels[0]->threshold, &data->levels[0]->threshold, set)); /* cache whether this option has been used or not to error out in case of exclusive options being used simultaneously later on */ 376b5a302b3SPierre Jolivet } 377b5a302b3SPierre Jolivet if (data->levels[0]->nu == 0 || nsv == 0) { /* if neither -eps_nev nor -svd_nsv has been used */ 378b5a302b3SPierre Jolivet PetscCall(PetscSNPrintf(deprecated, sizeof(deprecated), "-pc_hpddm_levels_%d_eps_relative_threshold", i)); 379b5a302b3SPierre Jolivet PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_threshold_relative", i)); 3808cb7430dSRaphael Zanella PetscCall(PetscOptionsDeprecated(deprecated, prefix, "3.24", nullptr)); 381b5a302b3SPierre Jolivet PetscCall(PetscOptionsReal(prefix, "Local relative threshold for selecting deflation vectors returned by SLEPc", "PCHPDDM", data->levels[0]->threshold, &data->levels[0]->threshold, set + 1)); 382b5a302b3SPierre Jolivet PetscCheck(!set[0] || !set[1], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot supply both -pc_hpddm_levels_1_eps_threshold_relative and -pc_hpddm_levels_1_svd_threshold_relative"); 383b5a302b3SPierre Jolivet } 384b5a302b3SPierre Jolivet PetscCheck(data->levels[0]->nu || PetscAbsReal(data->levels[i - 1]->threshold + static_cast<PetscReal>(1.0)) > PETSC_MACHINE_EPSILON, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Need to supply at least one of 1) -pc_hpddm_levels_1_eps_nev, 2) -pc_hpddm_levels_1_svd_nsv, 3) -pc_hpddm_levels_1_eps_threshold_relative, 4) -pc_hpddm_levels_1_svd_threshold_relative (for nonsymmetric matrices, only option 2 and option 4 are appropriate)"); 3859bb5c669SPierre Jolivet } 386f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_1_st_share_sub_ksp")); 387db4a47b3SPierre Jolivet PetscCall(PetscOptionsBool(prefix, "Shared KSP between SLEPc ST and the fine-level subdomain solver", "PCHPDDMSetSTShareSubKSP", PETSC_FALSE, &data->share, nullptr)); 388f1580f4eSBarry Smith } 389f1580f4eSBarry Smith /* if there is no prescribed coarsening, just break out of the loop */ 3900594bca0SPierre Jolivet if (data->levels[i - 1]->threshold <= PetscReal() && data->levels[i - 1]->nu <= 0 && !(data->deflation && i == 1)) break; 391f1580f4eSBarry Smith else { 392f1580f4eSBarry Smith ++i; 393f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_nev", i)); 394f1580f4eSBarry Smith PetscCall(PetscOptionsHasName(PetscOptionsObject->options, PetscOptionsObject->prefix, prefix, &flg)); 395f1580f4eSBarry Smith if (!flg) { 396b5a302b3SPierre Jolivet PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_eps_threshold_absolute", i)); 397f1580f4eSBarry Smith PetscCall(PetscOptionsHasName(PetscOptionsObject->options, PetscOptionsObject->prefix, prefix, &flg)); 398f1580f4eSBarry Smith } 399f1580f4eSBarry Smith if (flg) { 400f1580f4eSBarry Smith /* if there are coarsening options for the next level, then register it */ 401f1580f4eSBarry Smith /* otherwise, don't to avoid having both options levels_N_p and coarse_p */ 402f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_levels_%d_p", i)); 403f1580f4eSBarry 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))); 404f1580f4eSBarry Smith previous = p; 405f1580f4eSBarry Smith } 406f1580f4eSBarry Smith } 407f1580f4eSBarry Smith } 408f1580f4eSBarry Smith data->N = i; 409f1580f4eSBarry Smith n = 1; 410f1580f4eSBarry Smith if (i > 1) { 411f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_coarse_p")); 412db4a47b3SPierre Jolivet PetscCall(PetscOptionsRangeInt(prefix, "Number of processes used to assemble the coarsest operator", "PCHPDDM", n, &n, nullptr, 1, PetscMax(1, previous / 2))); 41302800ff6SPierre Jolivet #if PetscDefined(HAVE_MUMPS) 414f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "pc_hpddm_coarse_")); 415db4a47b3SPierre Jolivet PetscCall(PetscOptionsHasName(nullptr, prefix, "-mat_mumps_use_omp_threads", &flg)); 416f1580f4eSBarry Smith if (flg) { 417f1580f4eSBarry Smith char type[64]; /* same size as in src/ksp/pc/impls/factor/factimpl.c */ 418811e8887SPierre Jolivet 419c6a7a370SJeremy L Thompson PetscCall(PetscStrncpy(type, n > 1 && PetscDefined(HAVE_MUMPS) ? MATSOLVERMUMPS : MATSOLVERPETSC, sizeof(type))); /* default solver for a MatMPIAIJ or a MatSeqAIJ */ 420db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetString(nullptr, prefix, "-pc_factor_mat_solver_type", type, sizeof(type), nullptr)); 4213ce573a3SPierre Jolivet PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg)); 422f1580f4eSBarry 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); 423f1580f4eSBarry Smith size = n; 424f1580f4eSBarry Smith n = -1; 425db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetInt(nullptr, prefix, "-mat_mumps_use_omp_threads", &n, nullptr)); 426f1580f4eSBarry Smith PetscCheck(n >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Need to specify a positive integer for -%smat_mumps_use_omp_threads", prefix); 427f1580f4eSBarry 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" : ""); 428f1580f4eSBarry Smith } 429f1580f4eSBarry Smith #endif 430f1580f4eSBarry Smith PetscCall(PetscOptionsEnum("-pc_hpddm_coarse_correction", "Type of coarse correction applied each iteration", "PCHPDDMSetCoarseCorrectionType", PCHPDDMCoarseCorrectionTypes, (PetscEnum)data->correction, (PetscEnum *)&type, &flg)); 431f1580f4eSBarry Smith if (flg) PetscCall(PCHPDDMSetCoarseCorrectionType(pc, type)); 432f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_has_neumann")); 433c8ea6600SPierre Jolivet PetscCall(PetscOptionsBool(prefix, "Is the auxiliary Mat the local Neumann matrix?", "PCHPDDMHasNeumannMat", PetscBool3ToBool(data->Neumann), &flg, &set)); 434c8ea6600SPierre Jolivet if (set) data->Neumann = PetscBoolToBool3(flg); 435f1580f4eSBarry Smith data->log_separate = PETSC_FALSE; 436f1580f4eSBarry Smith if (PetscDefined(USE_LOG)) { 437f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-pc_hpddm_log_separate")); 438db4a47b3SPierre Jolivet PetscCall(PetscOptionsBool(prefix, "Log events level by level instead of inside PCSetUp()/KSPSolve()", nullptr, data->log_separate, &data->log_separate, nullptr)); 439f1580f4eSBarry Smith } 440f1580f4eSBarry Smith } 441f1580f4eSBarry Smith PetscOptionsHeadEnd(); 442f1580f4eSBarry Smith while (i < PETSC_PCHPDDM_MAXLEVELS && data->levels[i]) PetscCall(PetscFree(data->levels[i++])); 4433ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 444f1580f4eSBarry Smith } 445f1580f4eSBarry Smith 446d4f06b61SRaphael Zanella template <bool transpose> 447d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_HPDDM(PC pc, Vec x, Vec y) 448d71ae5a4SJacob Faibussowitsch { 449f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 450f1580f4eSBarry Smith 451f1580f4eSBarry Smith PetscFunctionBegin; 452f1580f4eSBarry Smith PetscCall(PetscCitationsRegister(HPDDMCitation, &HPDDMCite)); 453f1580f4eSBarry Smith PetscCheck(data->levels[0]->ksp, PETSC_COMM_SELF, PETSC_ERR_PLIB, "No KSP attached to PCHPDDM"); 454db4a47b3SPierre 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 */ 455d4f06b61SRaphael Zanella if (!transpose) PetscCall(KSPSolve(data->levels[0]->ksp, x, y)); 456d4f06b61SRaphael Zanella else PetscCall(KSPSolveTranspose(data->levels[0]->ksp, x, y)); 457db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Solve[0], data->levels[0]->ksp, nullptr, nullptr, nullptr)); 4583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 459f1580f4eSBarry Smith } 460f1580f4eSBarry Smith 4618cb7430dSRaphael Zanella template <bool transpose> 462d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_HPDDM(PC pc, Mat X, Mat Y) 463d71ae5a4SJacob Faibussowitsch { 464f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 465f1580f4eSBarry Smith 466f1580f4eSBarry Smith PetscFunctionBegin; 467f1580f4eSBarry Smith PetscCall(PetscCitationsRegister(HPDDMCitation, &HPDDMCite)); 468f1580f4eSBarry Smith PetscCheck(data->levels[0]->ksp, PETSC_COMM_SELF, PETSC_ERR_PLIB, "No KSP attached to PCHPDDM"); 4698cb7430dSRaphael Zanella if (!transpose) PetscCall(KSPMatSolve(data->levels[0]->ksp, X, Y)); 4708cb7430dSRaphael Zanella else PetscCall(KSPMatSolveTranspose(data->levels[0]->ksp, X, Y)); 4713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 472f1580f4eSBarry Smith } 473f1580f4eSBarry Smith 474cc4c1da9SBarry Smith /*@ 475f1580f4eSBarry Smith PCHPDDMGetComplexities - Computes the grid and operator complexities. 476f1580f4eSBarry Smith 477c71f06a7SPierre Jolivet Collective 478c71f06a7SPierre Jolivet 479f1580f4eSBarry Smith Input Parameter: 480f1580f4eSBarry Smith . pc - preconditioner context 481f1580f4eSBarry Smith 482f1580f4eSBarry Smith Output Parameters: 483cc4c1da9SBarry Smith + gc - grid complexity $ \sum_i m_i / m_1 $ 484cc4c1da9SBarry Smith - oc - operator complexity $ \sum_i nnz_i / nnz_1 $ 485f1580f4eSBarry Smith 486f1580f4eSBarry Smith Level: advanced 487f1580f4eSBarry Smith 488562efe2eSBarry Smith .seealso: [](ch_ksp), `PCMGGetGridComplexity()`, `PCHPDDM`, `PCHYPRE`, `PCGAMG` 489f1580f4eSBarry Smith @*/ 490cc4c1da9SBarry Smith PetscErrorCode PCHPDDMGetComplexities(PC pc, PetscReal *gc, PetscReal *oc) 491d71ae5a4SJacob Faibussowitsch { 492f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 493f1580f4eSBarry Smith MatInfo info; 494f1580f4eSBarry Smith PetscLogDouble accumulate[2]{}, nnz1 = 1.0, m1 = 1.0; 495f1580f4eSBarry Smith 496f1580f4eSBarry Smith PetscFunctionBegin; 497715c1178SPierre Jolivet if (gc) { 498715c1178SPierre Jolivet PetscAssertPointer(gc, 2); 499715c1178SPierre Jolivet *gc = 0; 500715c1178SPierre Jolivet } 501715c1178SPierre Jolivet if (oc) { 502715c1178SPierre Jolivet PetscAssertPointer(oc, 3); 503715c1178SPierre Jolivet *oc = 0; 504715c1178SPierre Jolivet } 505715c1178SPierre Jolivet for (PetscInt n = 0; n < data->N; ++n) { 506f1580f4eSBarry Smith if (data->levels[n]->ksp) { 50713044ca3SPierre Jolivet Mat P, A = nullptr; 508715c1178SPierre Jolivet PetscInt m; 50913044ca3SPierre Jolivet PetscBool flg = PETSC_FALSE; 51013044ca3SPierre Jolivet 511db4a47b3SPierre Jolivet PetscCall(KSPGetOperators(data->levels[n]->ksp, nullptr, &P)); 512db4a47b3SPierre Jolivet PetscCall(MatGetSize(P, &m, nullptr)); 513f1580f4eSBarry Smith accumulate[0] += m; 514f1580f4eSBarry Smith if (n == 0) { 515f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompareAny((PetscObject)P, &flg, MATNORMAL, MATNORMALHERMITIAN, "")); 516f1580f4eSBarry Smith if (flg) { 517f1580f4eSBarry Smith PetscCall(MatConvert(P, MATAIJ, MAT_INITIAL_MATRIX, &A)); 518f1580f4eSBarry Smith P = A; 51913044ca3SPierre Jolivet } else { 52013044ca3SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg)); 52113044ca3SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)P)); 522f1580f4eSBarry Smith } 52313044ca3SPierre Jolivet } 52413044ca3SPierre Jolivet if (!A && flg) accumulate[1] += m * m; /* assumption that a MATSCHURCOMPLEMENT is dense if stored explicitly */ 52513044ca3SPierre Jolivet else if (P->ops->getinfo) { 526f1580f4eSBarry Smith PetscCall(MatGetInfo(P, MAT_GLOBAL_SUM, &info)); 527f1580f4eSBarry Smith accumulate[1] += info.nz_used; 528f1580f4eSBarry Smith } 529f1580f4eSBarry Smith if (n == 0) { 530f1580f4eSBarry Smith m1 = m; 53113044ca3SPierre Jolivet if (!A && flg) nnz1 = m * m; 53213044ca3SPierre Jolivet else if (P->ops->getinfo) nnz1 = info.nz_used; 533f1580f4eSBarry Smith PetscCall(MatDestroy(&P)); 534f1580f4eSBarry Smith } 535f1580f4eSBarry Smith } 536f1580f4eSBarry Smith } 537715c1178SPierre Jolivet /* only process #0 has access to the full hierarchy by construction, so broadcast to ensure consistent outputs */ 538715c1178SPierre Jolivet PetscCallMPI(MPI_Bcast(accumulate, 2, MPIU_PETSCLOGDOUBLE, 0, PetscObjectComm((PetscObject)pc))); 539715c1178SPierre Jolivet if (gc) *gc = static_cast<PetscReal>(accumulate[0] / m1); 540715c1178SPierre Jolivet if (oc) *oc = static_cast<PetscReal>(accumulate[1] / nnz1); 5413ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 542f1580f4eSBarry Smith } 543f1580f4eSBarry Smith 544d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCView_HPDDM(PC pc, PetscViewer viewer) 545d71ae5a4SJacob Faibussowitsch { 546f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 547f1580f4eSBarry Smith PetscViewer subviewer; 548aa21023fSPierre Jolivet PetscViewerFormat format; 549f1580f4eSBarry Smith PetscSubcomm subcomm; 550f1580f4eSBarry Smith PetscReal oc, gc; 551811e8887SPierre Jolivet PetscInt tabs; 552f1580f4eSBarry Smith PetscMPIInt size, color, rank; 553aa21023fSPierre Jolivet PetscBool flg; 554aa21023fSPierre Jolivet const char *name; 555f1580f4eSBarry Smith 556f1580f4eSBarry Smith PetscFunctionBegin; 557aa21023fSPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &flg)); 558aa21023fSPierre Jolivet if (flg) { 559f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "level%s: %" PetscInt_FMT "\n", data->N > 1 ? "s" : "", data->N)); 560f1580f4eSBarry Smith PetscCall(PCHPDDMGetComplexities(pc, &gc, &oc)); 561f1580f4eSBarry Smith if (data->N > 1) { 562f1580f4eSBarry Smith if (!data->deflation) { 563c8ea6600SPierre Jolivet PetscCall(PetscViewerASCIIPrintf(viewer, "Neumann matrix attached? %s\n", PetscBools[PetscBool3ToBool(data->Neumann)])); 564f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "shared subdomain KSP between SLEPc and PETSc? %s\n", PetscBools[data->share])); 565f1580f4eSBarry Smith } else PetscCall(PetscViewerASCIIPrintf(viewer, "user-supplied deflation matrix\n")); 566f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "coarse correction: %s\n", PCHPDDMCoarseCorrectionTypes[data->correction])); 567f1580f4eSBarry 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" : "")); 568f1580f4eSBarry Smith PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 569f1580f4eSBarry Smith PetscCall(PetscViewerASCIISetTab(viewer, 0)); 570811e8887SPierre Jolivet for (PetscInt i = 1; i < data->N; ++i) { 571f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, " %" PetscInt_FMT, data->levels[i - 1]->nu)); 5720594bca0SPierre Jolivet if (data->levels[i - 1]->threshold > static_cast<PetscReal>(-0.1)) PetscCall(PetscViewerASCIIPrintf(viewer, " (%g)", (double)data->levels[i - 1]->threshold)); 573f1580f4eSBarry Smith } 574f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "\n")); 575f1580f4eSBarry Smith PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 576f1580f4eSBarry Smith } 577f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "grid and operator complexities: %g %g\n", (double)gc, (double)oc)); 5781fe44b27SPierre Jolivet PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)pc), &size)); 579f1580f4eSBarry Smith if (data->levels[0]->ksp) { 580f1580f4eSBarry Smith PetscCall(KSPView(data->levels[0]->ksp, viewer)); 581f1580f4eSBarry Smith if (data->levels[0]->pc) PetscCall(PCView(data->levels[0]->pc, viewer)); 582eea1a2f1SRaphael Zanella PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)pc), &rank)); 583811e8887SPierre Jolivet for (PetscInt i = 1; i < data->N; ++i) { 584f1580f4eSBarry Smith if (data->levels[i]->ksp) color = 1; 585f1580f4eSBarry Smith else color = 0; 586f1580f4eSBarry Smith PetscCall(PetscSubcommCreate(PetscObjectComm((PetscObject)pc), &subcomm)); 587f1580f4eSBarry Smith PetscCall(PetscSubcommSetNumber(subcomm, PetscMin(size, 2))); 588f1580f4eSBarry Smith PetscCall(PetscSubcommSetTypeGeneral(subcomm, color, rank)); 589f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPushTab(viewer)); 590f1580f4eSBarry Smith PetscCall(PetscViewerGetSubViewer(viewer, PetscSubcommChild(subcomm), &subviewer)); 591f1580f4eSBarry Smith if (color == 1) { 592f1580f4eSBarry Smith PetscCall(KSPView(data->levels[i]->ksp, subviewer)); 593f1580f4eSBarry Smith if (data->levels[i]->pc) PetscCall(PCView(data->levels[i]->pc, subviewer)); 594f1580f4eSBarry Smith PetscCall(PetscViewerFlush(subviewer)); 595f1580f4eSBarry Smith } 596f1580f4eSBarry Smith PetscCall(PetscViewerRestoreSubViewer(viewer, PetscSubcommChild(subcomm), &subviewer)); 597f1580f4eSBarry Smith PetscCall(PetscViewerASCIIPopTab(viewer)); 598f1580f4eSBarry Smith PetscCall(PetscSubcommDestroy(&subcomm)); 599f1580f4eSBarry Smith } 600f1580f4eSBarry Smith } 601aa21023fSPierre Jolivet PetscCall(PetscViewerGetFormat(viewer, &format)); 602aa21023fSPierre Jolivet if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 603aa21023fSPierre Jolivet PetscCall(PetscViewerFileGetName(viewer, &name)); 604aa21023fSPierre Jolivet if (name) { 605eea1a2f1SRaphael Zanella Mat aux[2]; 606aa21023fSPierre Jolivet IS is; 607eea1a2f1SRaphael Zanella const PetscInt *indices; 608eea1a2f1SRaphael Zanella PetscInt m, n, sizes[5] = {pc->mat->rmap->n, pc->mat->cmap->n, pc->mat->rmap->N, pc->mat->cmap->N, 0}; 609aa21023fSPierre Jolivet char *tmp; 610aa21023fSPierre Jolivet std::string prefix, suffix; 611aa21023fSPierre Jolivet size_t pos; 612aa21023fSPierre Jolivet 613aa21023fSPierre Jolivet PetscCall(PetscStrstr(name, ".", &tmp)); 614aa21023fSPierre Jolivet if (tmp) { 615aa21023fSPierre Jolivet pos = std::distance(const_cast<char *>(name), tmp); 616aa21023fSPierre Jolivet prefix = std::string(name, pos); 617aa21023fSPierre Jolivet suffix = std::string(name + pos + 1); 618aa21023fSPierre Jolivet } else prefix = name; 619aa21023fSPierre Jolivet if (data->aux) { 620eea1a2f1SRaphael Zanella PetscCall(MatGetSize(data->aux, &m, &n)); 621eea1a2f1SRaphael Zanella PetscCall(MatCreate(PetscObjectComm((PetscObject)pc), aux)); 622eea1a2f1SRaphael Zanella PetscCall(MatSetSizes(aux[0], m, n, PETSC_DETERMINE, PETSC_DETERMINE)); 623eea1a2f1SRaphael Zanella PetscCall(PetscObjectBaseTypeCompare((PetscObject)data->aux, MATSEQAIJ, &flg)); 624eea1a2f1SRaphael Zanella if (flg) PetscCall(MatSetType(aux[0], MATMPIAIJ)); 625eea1a2f1SRaphael Zanella else { 626eea1a2f1SRaphael Zanella PetscCall(PetscObjectBaseTypeCompare((PetscObject)data->aux, MATSEQBAIJ, &flg)); 627eea1a2f1SRaphael Zanella if (flg) PetscCall(MatSetType(aux[0], MATMPIBAIJ)); 628eea1a2f1SRaphael Zanella else { 629eea1a2f1SRaphael Zanella PetscCall(PetscObjectBaseTypeCompare((PetscObject)data->aux, MATSEQSBAIJ, &flg)); 630eea1a2f1SRaphael 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); 631eea1a2f1SRaphael Zanella PetscCall(MatSetType(aux[0], MATMPISBAIJ)); 632eea1a2f1SRaphael Zanella } 633eea1a2f1SRaphael Zanella } 634eea1a2f1SRaphael Zanella PetscCall(MatSetBlockSizesFromMats(aux[0], data->aux, data->aux)); 635eea1a2f1SRaphael Zanella PetscCall(MatAssemblyBegin(aux[0], MAT_FINAL_ASSEMBLY)); 636eea1a2f1SRaphael Zanella PetscCall(MatAssemblyEnd(aux[0], MAT_FINAL_ASSEMBLY)); 637eea1a2f1SRaphael Zanella PetscCall(MatGetDiagonalBlock(aux[0], aux + 1)); 638eea1a2f1SRaphael Zanella PetscCall(MatCopy(data->aux, aux[1], DIFFERENT_NONZERO_PATTERN)); 639eea1a2f1SRaphael Zanella PetscCall(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)pc), std::string(prefix + "_aux_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer)); 640eea1a2f1SRaphael Zanella PetscCall(MatView(aux[0], subviewer)); 641aa21023fSPierre Jolivet PetscCall(PetscViewerDestroy(&subviewer)); 642eea1a2f1SRaphael Zanella PetscCall(MatDestroy(aux)); 643aa21023fSPierre Jolivet } 644aa21023fSPierre Jolivet if (data->is) { 645eea1a2f1SRaphael Zanella PetscCall(ISGetIndices(data->is, &indices)); 646eea1a2f1SRaphael Zanella PetscCall(ISGetSize(data->is, sizes + 4)); 647eea1a2f1SRaphael Zanella PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)pc), sizes[4], indices, PETSC_USE_POINTER, &is)); 648eea1a2f1SRaphael Zanella PetscCall(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)pc), std::string(prefix + "_is_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer)); 649eea1a2f1SRaphael Zanella PetscCall(ISView(is, subviewer)); 650aa21023fSPierre Jolivet PetscCall(PetscViewerDestroy(&subviewer)); 651eea1a2f1SRaphael Zanella PetscCall(ISDestroy(&is)); 652eea1a2f1SRaphael Zanella PetscCall(ISRestoreIndices(data->is, &indices)); 653aa21023fSPierre Jolivet } 654eea1a2f1SRaphael Zanella PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)pc), PETSC_STATIC_ARRAY_LENGTH(sizes), sizes, PETSC_USE_POINTER, &is)); 655eea1a2f1SRaphael Zanella PetscCall(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)pc), std::string(prefix + "_sizes_" + std::to_string(size) + (tmp ? ("." + suffix) : "")).c_str(), FILE_MODE_WRITE, &subviewer)); 656aa21023fSPierre Jolivet PetscCall(ISView(is, subviewer)); 657aa21023fSPierre Jolivet PetscCall(PetscViewerDestroy(&subviewer)); 658aa21023fSPierre Jolivet PetscCall(ISDestroy(&is)); 659aa21023fSPierre Jolivet } 660aa21023fSPierre Jolivet } 661f1580f4eSBarry Smith } 6623ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 663f1580f4eSBarry Smith } 664f1580f4eSBarry Smith 665d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCPreSolve_HPDDM(PC pc, KSP ksp, Vec, Vec) 666d71ae5a4SJacob Faibussowitsch { 667f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 668f1580f4eSBarry Smith Mat A; 669cdbd50ebSPierre Jolivet PetscBool flg; 670f1580f4eSBarry Smith 671f1580f4eSBarry Smith PetscFunctionBegin; 672f1580f4eSBarry Smith if (ksp) { 673f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)ksp, KSPLSQR, &flg)); 674f1580f4eSBarry Smith if (flg && !data->normal) { 675db4a47b3SPierre Jolivet PetscCall(KSPGetOperators(ksp, &A, nullptr)); 676db4a47b3SPierre Jolivet PetscCall(MatCreateVecs(A, nullptr, &data->normal)); /* temporary Vec used in PCApply_HPDDMShell() for coarse grid corrections */ 677cdbd50ebSPierre Jolivet } else if (!flg) { 678cdbd50ebSPierre Jolivet PetscCall(PetscObjectTypeCompareAny((PetscObject)ksp, &flg, KSPCG, KSPGROPPCG, KSPPIPECG, KSPPIPECGRR, KSPPIPELCG, KSPPIPEPRCG, KSPPIPECG2, KSPSTCG, KSPFCG, KSPPIPEFCG, KSPMINRES, KSPNASH, KSPSYMMLQ, "")); 679cdbd50ebSPierre Jolivet if (!flg) { 680cdbd50ebSPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)ksp, KSPHPDDM, &flg)); 681cdbd50ebSPierre Jolivet if (flg) { 682cdbd50ebSPierre Jolivet KSPHPDDMType type; 683811e8887SPierre Jolivet 684cdbd50ebSPierre Jolivet PetscCall(KSPHPDDMGetType(ksp, &type)); 685cdbd50ebSPierre Jolivet flg = (type == KSP_HPDDM_TYPE_CG || type == KSP_HPDDM_TYPE_BCG || type == KSP_HPDDM_TYPE_BFBCG ? PETSC_TRUE : PETSC_FALSE); 686cdbd50ebSPierre Jolivet } 687cdbd50ebSPierre Jolivet } 688cdbd50ebSPierre Jolivet } 689cdbd50ebSPierre Jolivet if (flg) { 690cdbd50ebSPierre Jolivet if (data->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED) { 691cdbd50ebSPierre Jolivet PetscCall(PetscOptionsHasName(((PetscObject)pc)->options, ((PetscObject)pc)->prefix, "-pc_hpddm_coarse_correction", &flg)); 692cdbd50ebSPierre 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", 693cdbd50ebSPierre Jolivet PCHPDDMCoarseCorrectionTypes[data->correction], ((PetscObject)ksp)->type_name, ((PetscObject)pc)->prefix ? ((PetscObject)pc)->prefix : "", PCHPDDMCoarseCorrectionTypes[data->correction], PCHPDDMCoarseCorrectionTypes[PC_HPDDM_COARSE_CORRECTION_BALANCED]); 694cdbd50ebSPierre Jolivet } 695cdbd50ebSPierre Jolivet for (PetscInt n = 0; n < data->N; ++n) { 696cdbd50ebSPierre Jolivet if (data->levels[n]->pc) { 697cdbd50ebSPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)data->levels[n]->pc, PCASM, &flg)); 698cdbd50ebSPierre Jolivet if (flg) { 699cdbd50ebSPierre Jolivet PCASMType type; 700811e8887SPierre Jolivet 701cdbd50ebSPierre Jolivet PetscCall(PCASMGetType(data->levels[n]->pc, &type)); 702cdbd50ebSPierre Jolivet if (type == PC_ASM_RESTRICT || type == PC_ASM_INTERPOLATE) { 703cdbd50ebSPierre Jolivet PetscCall(PetscOptionsHasName(((PetscObject)data->levels[n]->pc)->options, ((PetscObject)data->levels[n]->pc)->prefix, "-pc_asm_type", &flg)); 704cdbd50ebSPierre 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], 705cdbd50ebSPierre Jolivet ((PetscObject)ksp)->type_name, ((PetscObject)data->levels[n]->pc)->prefix, PCASMTypes[type], PCASMTypes[PC_ASM_BASIC]); 706cdbd50ebSPierre Jolivet } 707cdbd50ebSPierre Jolivet } 708cdbd50ebSPierre Jolivet } 709cdbd50ebSPierre Jolivet } 710f1580f4eSBarry Smith } 711f1580f4eSBarry Smith } 7123ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 713f1580f4eSBarry Smith } 714f1580f4eSBarry Smith 715d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetUp_HPDDMShell(PC pc) 716d71ae5a4SJacob Faibussowitsch { 717f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 718f1580f4eSBarry Smith Mat A, P; 719f1580f4eSBarry Smith Vec x; 720f1580f4eSBarry Smith const char *pcpre; 721f1580f4eSBarry Smith 722f1580f4eSBarry Smith PetscFunctionBegin; 723f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 724f1580f4eSBarry Smith PetscCall(KSPGetOptionsPrefix(ctx->ksp, &pcpre)); 725f1580f4eSBarry Smith PetscCall(KSPGetOperators(ctx->ksp, &A, &P)); 726f1580f4eSBarry Smith /* smoother */ 727f1580f4eSBarry Smith PetscCall(PCSetOptionsPrefix(ctx->pc, pcpre)); 728f1580f4eSBarry Smith PetscCall(PCSetOperators(ctx->pc, A, P)); 729f1580f4eSBarry Smith if (!ctx->v[0]) { 730f1580f4eSBarry Smith PetscCall(VecDuplicateVecs(ctx->D, 1, &ctx->v[0])); 731f1580f4eSBarry Smith if (!std::is_same<PetscScalar, PetscReal>::value) PetscCall(VecDestroy(&ctx->D)); 732db4a47b3SPierre Jolivet PetscCall(MatCreateVecs(A, &x, nullptr)); 733f1580f4eSBarry Smith PetscCall(VecDuplicateVecs(x, 2, &ctx->v[1])); 734f1580f4eSBarry Smith PetscCall(VecDestroy(&x)); 735f1580f4eSBarry Smith } 736f1580f4eSBarry Smith std::fill_n(ctx->V, 3, nullptr); 7373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 738f1580f4eSBarry Smith } 739f1580f4eSBarry Smith 740d4f06b61SRaphael Zanella template <bool transpose = false, class Type = Vec, typename std::enable_if<std::is_same<Type, Vec>::value>::type * = nullptr> 741d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMDeflate_Private(PC pc, Type x, Type y) 742d71ae5a4SJacob Faibussowitsch { 743f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 744f1580f4eSBarry Smith PetscScalar *out; 745f1580f4eSBarry Smith 746f1580f4eSBarry Smith PetscFunctionBegin; 747f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 748f1580f4eSBarry Smith /* going from PETSc to HPDDM numbering */ 749f1580f4eSBarry Smith PetscCall(VecScatterBegin(ctx->scatter, x, ctx->v[0][0], INSERT_VALUES, SCATTER_FORWARD)); 750f1580f4eSBarry Smith PetscCall(VecScatterEnd(ctx->scatter, x, ctx->v[0][0], INSERT_VALUES, SCATTER_FORWARD)); 751f1580f4eSBarry Smith PetscCall(VecGetArrayWrite(ctx->v[0][0], &out)); 752d4f06b61SRaphael Zanella ctx->P->deflation<false, transpose>(nullptr, out, 1); /* y = Q x */ 753f1580f4eSBarry Smith PetscCall(VecRestoreArrayWrite(ctx->v[0][0], &out)); 754f1580f4eSBarry Smith /* going from HPDDM to PETSc numbering */ 755f1580f4eSBarry Smith PetscCall(VecScatterBegin(ctx->scatter, ctx->v[0][0], y, INSERT_VALUES, SCATTER_REVERSE)); 756f1580f4eSBarry Smith PetscCall(VecScatterEnd(ctx->scatter, ctx->v[0][0], y, INSERT_VALUES, SCATTER_REVERSE)); 7573ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 758f1580f4eSBarry Smith } 759f1580f4eSBarry Smith 7608cb7430dSRaphael Zanella template <bool transpose = false, class Type = Mat, typename std::enable_if<std::is_same<Type, Mat>::value>::type * = nullptr> 761d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PCHPDDMDeflate_Private(PC pc, Type X, Type Y) 762d71ae5a4SJacob Faibussowitsch { 763f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 764f1580f4eSBarry Smith Vec vX, vY, vC; 765f1580f4eSBarry Smith PetscScalar *out; 766811e8887SPierre Jolivet PetscInt N; 767f1580f4eSBarry Smith 768f1580f4eSBarry Smith PetscFunctionBegin; 769f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 770db4a47b3SPierre Jolivet PetscCall(MatGetSize(X, nullptr, &N)); 771f1580f4eSBarry Smith /* going from PETSc to HPDDM numbering */ 772811e8887SPierre Jolivet for (PetscInt i = 0; i < N; ++i) { 773f1580f4eSBarry Smith PetscCall(MatDenseGetColumnVecRead(X, i, &vX)); 774f1580f4eSBarry Smith PetscCall(MatDenseGetColumnVecWrite(ctx->V[0], i, &vC)); 775f1580f4eSBarry Smith PetscCall(VecScatterBegin(ctx->scatter, vX, vC, INSERT_VALUES, SCATTER_FORWARD)); 776f1580f4eSBarry Smith PetscCall(VecScatterEnd(ctx->scatter, vX, vC, INSERT_VALUES, SCATTER_FORWARD)); 777f1580f4eSBarry Smith PetscCall(MatDenseRestoreColumnVecWrite(ctx->V[0], i, &vC)); 778f1580f4eSBarry Smith PetscCall(MatDenseRestoreColumnVecRead(X, i, &vX)); 779f1580f4eSBarry Smith } 780f1580f4eSBarry Smith PetscCall(MatDenseGetArrayWrite(ctx->V[0], &out)); 7818cb7430dSRaphael Zanella ctx->P->deflation<false, transpose>(nullptr, out, N); /* Y = Q X */ 782f1580f4eSBarry Smith PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &out)); 783f1580f4eSBarry Smith /* going from HPDDM to PETSc numbering */ 784811e8887SPierre Jolivet for (PetscInt i = 0; i < N; ++i) { 785f1580f4eSBarry Smith PetscCall(MatDenseGetColumnVecRead(ctx->V[0], i, &vC)); 786f1580f4eSBarry Smith PetscCall(MatDenseGetColumnVecWrite(Y, i, &vY)); 787f1580f4eSBarry Smith PetscCall(VecScatterBegin(ctx->scatter, vC, vY, INSERT_VALUES, SCATTER_REVERSE)); 788f1580f4eSBarry Smith PetscCall(VecScatterEnd(ctx->scatter, vC, vY, INSERT_VALUES, SCATTER_REVERSE)); 789f1580f4eSBarry Smith PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &vY)); 790f1580f4eSBarry Smith PetscCall(MatDenseRestoreColumnVecRead(ctx->V[0], i, &vC)); 791f1580f4eSBarry Smith } 7923ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 793f1580f4eSBarry Smith } 794f1580f4eSBarry Smith 795f1580f4eSBarry Smith /* 796aa1539e9SPierre 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. 797f1580f4eSBarry Smith 798f1580f4eSBarry Smith .vb 799f1580f4eSBarry Smith (1) y = Pmat^-1 x + Q x, 800f1580f4eSBarry Smith (2) y = Pmat^-1 (I - Amat Q) x + Q x (default), 801aa1539e9SPierre Jolivet (3) y = (I - Q Amat^T) Pmat^-1 (I - Amat Q) x + Q x, 802aa1539e9SPierre Jolivet (4) y = Pmat^-1 x . 803f1580f4eSBarry Smith .ve 804f1580f4eSBarry Smith 805f1580f4eSBarry Smith Input Parameters: 806f1580f4eSBarry Smith + pc - preconditioner context 807f1580f4eSBarry Smith - x - input vector 808f1580f4eSBarry Smith 809f1580f4eSBarry Smith Output Parameter: 810f1580f4eSBarry Smith . y - output vector 811f1580f4eSBarry Smith 812f1580f4eSBarry Smith Notes: 813f1580f4eSBarry 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. 814f1580f4eSBarry 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. 815f1580f4eSBarry 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. 816f1580f4eSBarry Smith 817f1580f4eSBarry Smith Level: advanced 818f1580f4eSBarry Smith 819f1580f4eSBarry Smith Developer Note: 820f1580f4eSBarry 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 821f1580f4eSBarry Smith to trigger it. Likely the manual page is `PCHPDDM` 822f1580f4eSBarry Smith 823562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMCoarseCorrectionType` 824f1580f4eSBarry Smith */ 825d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_HPDDMShell(PC pc, Vec x, Vec y) 826d71ae5a4SJacob Faibussowitsch { 827f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 828f1580f4eSBarry Smith Mat A; 829f1580f4eSBarry Smith 830f1580f4eSBarry Smith PetscFunctionBegin; 831f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 832f1580f4eSBarry Smith PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object"); 833db4a47b3SPierre Jolivet PetscCall(KSPGetOperators(ctx->ksp, &A, nullptr)); 834aa1539e9SPierre Jolivet if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_NONE) PetscCall(PCApply(ctx->pc, x, y)); /* y = M^-1 x */ 835aa1539e9SPierre Jolivet else { 836f1580f4eSBarry Smith PetscCall(PCHPDDMDeflate_Private(pc, x, y)); /* y = Q x */ 837f1580f4eSBarry Smith if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 838f1580f4eSBarry Smith if (!ctx->parent->normal || ctx != ctx->parent->levels[0]) PetscCall(MatMult(A, y, ctx->v[1][0])); /* y = A Q x */ 839*8a8e6071SPierre Jolivet else { 840*8a8e6071SPierre Jolivet /* KSPLSQR and finest level */ 841*8a8e6071SPierre Jolivet PetscCall(MatMult(A, y, ctx->parent->normal)); /* y = A Q x */ 842f1580f4eSBarry Smith PetscCall(MatMultHermitianTranspose(A, ctx->parent->normal, ctx->v[1][0])); /* y = A^T A Q x */ 843f1580f4eSBarry Smith } 844f1580f4eSBarry Smith PetscCall(VecWAXPY(ctx->v[1][1], -1.0, ctx->v[1][0], x)); /* y = (I - A Q) x */ 845f1580f4eSBarry Smith PetscCall(PCApply(ctx->pc, ctx->v[1][1], ctx->v[1][0])); /* y = M^-1 (I - A Q) x */ 846f1580f4eSBarry Smith if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 847f1580f4eSBarry 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 */ 848f1580f4eSBarry Smith else { 849f1580f4eSBarry Smith PetscCall(MatMult(A, ctx->v[1][0], ctx->parent->normal)); 850f1580f4eSBarry Smith PetscCall(MatMultHermitianTranspose(A, ctx->parent->normal, ctx->v[1][1])); /* z = A^T A y */ 851f1580f4eSBarry Smith } 852d4f06b61SRaphael Zanella PetscCall(PCHPDDMDeflate_Private(pc, ctx->v[1][1], ctx->v[1][1])); /* z = Q z */ 853f1580f4eSBarry 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 */ 854f1580f4eSBarry Smith } else PetscCall(VecAXPY(y, 1.0, ctx->v[1][0])); /* y = Q M^-1 (I - A Q) x + Q x */ 855f1580f4eSBarry Smith } else { 856f1580f4eSBarry 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); 857f1580f4eSBarry Smith PetscCall(PCApply(ctx->pc, x, ctx->v[1][0])); 858f1580f4eSBarry Smith PetscCall(VecAXPY(y, 1.0, ctx->v[1][0])); /* y = M^-1 x + Q x */ 859f1580f4eSBarry Smith } 860aa1539e9SPierre Jolivet } 8613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 862f1580f4eSBarry Smith } 863f1580f4eSBarry Smith 8648cb7430dSRaphael Zanella template <bool transpose> 8658cb7430dSRaphael Zanella static PetscErrorCode PCHPDDMMatApply_Private(PC_HPDDM_Level *ctx, Mat Y, PetscBool *reset) 8668cb7430dSRaphael Zanella { 8678cb7430dSRaphael Zanella Mat A, *ptr; 8688cb7430dSRaphael Zanella PetscScalar *array; 8698cb7430dSRaphael Zanella PetscInt m, M, N, prev = 0; 8708cb7430dSRaphael Zanella PetscContainer container = nullptr; 8718cb7430dSRaphael Zanella 8728cb7430dSRaphael Zanella PetscFunctionBegin; 8738cb7430dSRaphael Zanella PetscCall(KSPGetOperators(ctx->ksp, &A, nullptr)); 8748cb7430dSRaphael Zanella PetscCall(MatGetSize(Y, nullptr, &N)); 8758cb7430dSRaphael Zanella PetscCall(PetscObjectQuery((PetscObject)A, "_HPDDM_MatProduct", (PetscObject *)&container)); 8768cb7430dSRaphael Zanella if (container) { /* MatProduct container already attached */ 8778cb7430dSRaphael Zanella PetscCall(PetscContainerGetPointer(container, (void **)&ptr)); 8788cb7430dSRaphael Zanella if (ptr[1] != ctx->V[2]) /* Mat has changed or may have been set first in KSPHPDDM */ 8798cb7430dSRaphael Zanella for (m = 0; m < 2; ++m) { 8808cb7430dSRaphael Zanella PetscCall(MatDestroy(ctx->V + m + 1)); 8818cb7430dSRaphael Zanella ctx->V[m + 1] = ptr[m]; 8828cb7430dSRaphael Zanella PetscCall(PetscObjectReference((PetscObject)ctx->V[m + 1])); 8838cb7430dSRaphael Zanella } 8848cb7430dSRaphael Zanella } 8858cb7430dSRaphael Zanella if (ctx->V[1]) PetscCall(MatGetSize(ctx->V[1], nullptr, &prev)); 8868cb7430dSRaphael Zanella if (N != prev || !ctx->V[0]) { 8878cb7430dSRaphael Zanella PetscCall(MatDestroy(ctx->V)); 8888cb7430dSRaphael Zanella PetscCall(VecGetLocalSize(ctx->v[0][0], &m)); 8898cb7430dSRaphael Zanella PetscCall(MatCreateDense(PetscObjectComm((PetscObject)Y), m, PETSC_DECIDE, PETSC_DECIDE, N, nullptr, ctx->V)); 8908cb7430dSRaphael Zanella if (N != prev) { 8918cb7430dSRaphael Zanella PetscCall(MatDestroy(ctx->V + 1)); 8928cb7430dSRaphael Zanella PetscCall(MatDestroy(ctx->V + 2)); 8938cb7430dSRaphael Zanella PetscCall(MatGetLocalSize(Y, &m, nullptr)); 8948cb7430dSRaphael Zanella PetscCall(MatGetSize(Y, &M, nullptr)); 8958cb7430dSRaphael Zanella if (ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) PetscCall(MatDenseGetArrayWrite(ctx->V[0], &array)); 8968cb7430dSRaphael Zanella else array = nullptr; 8978cb7430dSRaphael Zanella PetscCall(MatCreateDense(PetscObjectComm((PetscObject)Y), m, PETSC_DECIDE, M, N, array, ctx->V + 1)); 8988cb7430dSRaphael Zanella if (ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &array)); 8998cb7430dSRaphael Zanella PetscCall(MatCreateDense(PetscObjectComm((PetscObject)Y), m, PETSC_DECIDE, M, N, nullptr, ctx->V + 2)); 9008cb7430dSRaphael Zanella PetscCall(MatProductCreateWithMat(A, !transpose ? Y : ctx->V[2], nullptr, ctx->V[1])); 9018cb7430dSRaphael Zanella PetscCall(MatProductSetType(ctx->V[1], !transpose ? MATPRODUCT_AB : MATPRODUCT_AtB)); 9028cb7430dSRaphael Zanella PetscCall(MatProductSetFromOptions(ctx->V[1])); 9038cb7430dSRaphael Zanella PetscCall(MatProductSymbolic(ctx->V[1])); 9048cb7430dSRaphael Zanella 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() */ 9058cb7430dSRaphael Zanella 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 */ 9068cb7430dSRaphael Zanella } 9078cb7430dSRaphael Zanella if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 9088cb7430dSRaphael Zanella PetscCall(MatProductCreateWithMat(A, !transpose ? ctx->V[1] : Y, nullptr, ctx->V[2])); 9098cb7430dSRaphael Zanella PetscCall(MatProductSetType(ctx->V[2], !transpose ? MATPRODUCT_AtB : MATPRODUCT_AB)); 9108cb7430dSRaphael Zanella PetscCall(MatProductSetFromOptions(ctx->V[2])); 9118cb7430dSRaphael Zanella PetscCall(MatProductSymbolic(ctx->V[2])); 9128cb7430dSRaphael Zanella } 9138cb7430dSRaphael Zanella ctx->P->start(N); 9148cb7430dSRaphael Zanella } 9158cb7430dSRaphael Zanella if (N == prev || container) { /* when MatProduct container is attached, always need to MatProductReplaceMats() since KSPHPDDM may have replaced the Mat as well */ 9168cb7430dSRaphael Zanella PetscCall(MatProductReplaceMats(nullptr, !transpose ? Y : ctx->V[2], nullptr, ctx->V[1])); 9178cb7430dSRaphael Zanella if (container && ctx->parent->correction != PC_HPDDM_COARSE_CORRECTION_BALANCED) { 9188cb7430dSRaphael Zanella PetscCall(MatDenseGetArrayWrite(ctx->V[0], &array)); 9198cb7430dSRaphael Zanella PetscCall(MatDensePlaceArray(ctx->V[1], array)); 9208cb7430dSRaphael Zanella PetscCall(MatDenseRestoreArrayWrite(ctx->V[0], &array)); 9218cb7430dSRaphael Zanella *reset = PETSC_TRUE; 9228cb7430dSRaphael Zanella } 9238cb7430dSRaphael Zanella } 9248cb7430dSRaphael Zanella PetscFunctionReturn(PETSC_SUCCESS); 9258cb7430dSRaphael Zanella } 9268cb7430dSRaphael Zanella 927f1580f4eSBarry Smith /* 928f1580f4eSBarry Smith PCMatApply_HPDDMShell - Variant of PCApply_HPDDMShell() for blocks of vectors. 929f1580f4eSBarry Smith 930f1580f4eSBarry Smith Input Parameters: 931f1580f4eSBarry Smith + pc - preconditioner context 932f1580f4eSBarry Smith - X - block of input vectors 933f1580f4eSBarry Smith 934f1580f4eSBarry Smith Output Parameter: 935f1580f4eSBarry Smith . Y - block of output vectors 936f1580f4eSBarry Smith 937f1580f4eSBarry Smith Level: advanced 938f1580f4eSBarry Smith 939562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCApply_HPDDMShell()`, `PCHPDDMCoarseCorrectionType` 940f1580f4eSBarry Smith */ 941d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_HPDDMShell(PC pc, Mat X, Mat Y) 942d71ae5a4SJacob Faibussowitsch { 943f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 944f1580f4eSBarry Smith PetscBool reset = PETSC_FALSE; 945f1580f4eSBarry Smith 946f1580f4eSBarry Smith PetscFunctionBegin; 947f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 948f1580f4eSBarry Smith PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object"); 949aa1539e9SPierre Jolivet if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_NONE) PetscCall(PCMatApply(ctx->pc, X, Y)); 950aa1539e9SPierre Jolivet else { 9518cb7430dSRaphael Zanella PetscCall(PCHPDDMMatApply_Private<false>(ctx, Y, &reset)); 952f1580f4eSBarry Smith PetscCall(PCHPDDMDeflate_Private(pc, X, Y)); 953f1580f4eSBarry Smith if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 954f1580f4eSBarry Smith PetscCall(MatProductNumeric(ctx->V[1])); 955f1580f4eSBarry Smith PetscCall(MatCopy(ctx->V[1], ctx->V[2], SAME_NONZERO_PATTERN)); 956f1580f4eSBarry Smith PetscCall(MatAXPY(ctx->V[2], -1.0, X, SAME_NONZERO_PATTERN)); 957f1580f4eSBarry Smith PetscCall(PCMatApply(ctx->pc, ctx->V[2], ctx->V[1])); 958f1580f4eSBarry Smith if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 959f1580f4eSBarry Smith PetscCall(MatProductNumeric(ctx->V[2])); 960f1580f4eSBarry Smith PetscCall(PCHPDDMDeflate_Private(pc, ctx->V[2], ctx->V[2])); 961f1580f4eSBarry Smith PetscCall(MatAXPY(ctx->V[1], -1.0, ctx->V[2], SAME_NONZERO_PATTERN)); 962f1580f4eSBarry Smith } 963f1580f4eSBarry Smith PetscCall(MatAXPY(Y, -1.0, ctx->V[1], SAME_NONZERO_PATTERN)); 964f1580f4eSBarry Smith } else { 965f1580f4eSBarry 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); 966f1580f4eSBarry Smith PetscCall(PCMatApply(ctx->pc, X, ctx->V[1])); 967f1580f4eSBarry Smith PetscCall(MatAXPY(Y, 1.0, ctx->V[1], SAME_NONZERO_PATTERN)); 968f1580f4eSBarry Smith } 969f1580f4eSBarry Smith if (reset) PetscCall(MatDenseResetArray(ctx->V[1])); 970aa1539e9SPierre Jolivet } 9713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 972f1580f4eSBarry Smith } 973f1580f4eSBarry Smith 974d4f06b61SRaphael Zanella /* 975d4f06b61SRaphael Zanella PCApplyTranspose_HPDDMShell - Applies the transpose of 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. 976d4f06b61SRaphael Zanella 977d4f06b61SRaphael Zanella .vb 978d4f06b61SRaphael Zanella (1) y = Pmat^-T x + Q^T x, 979d4f06b61SRaphael Zanella (2) y = (I - Q^T Amat^T) Pmat^-T x + Q^T x (default), 980d4f06b61SRaphael Zanella (3) y = (I - Q^T Amat^T) Pmat^-T (I - Amat Q^T) x + Q^T x, 981d4f06b61SRaphael Zanella (4) y = Pmat^-T x . 982d4f06b61SRaphael Zanella .ve 983d4f06b61SRaphael Zanella 984d4f06b61SRaphael Zanella Input Parameters: 985d4f06b61SRaphael Zanella + pc - preconditioner context 986d4f06b61SRaphael Zanella - x - input vector 987d4f06b61SRaphael Zanella 988d4f06b61SRaphael Zanella Output Parameter: 989d4f06b61SRaphael Zanella . y - output vector 990d4f06b61SRaphael Zanella 991d4f06b61SRaphael Zanella Level: advanced 992d4f06b61SRaphael Zanella 993d4f06b61SRaphael Zanella Developer Note: 994d4f06b61SRaphael Zanella 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 995d4f06b61SRaphael Zanella to trigger it. Likely the manual page is `PCHPDDM` 996d4f06b61SRaphael Zanella 997d4f06b61SRaphael Zanella .seealso: [](ch_ksp), `PCHPDDM`, `PCApply_HPDDMShell()`, `PCHPDDMCoarseCorrectionType` 998d4f06b61SRaphael Zanella */ 999d4f06b61SRaphael Zanella static PetscErrorCode PCApplyTranspose_HPDDMShell(PC pc, Vec x, Vec y) 1000d4f06b61SRaphael Zanella { 1001d4f06b61SRaphael Zanella PC_HPDDM_Level *ctx; 1002d4f06b61SRaphael Zanella Mat A; 1003d4f06b61SRaphael Zanella 1004d4f06b61SRaphael Zanella PetscFunctionBegin; 1005d4f06b61SRaphael Zanella PetscCall(PCShellGetContext(pc, &ctx)); 1006d4f06b61SRaphael Zanella PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object"); 10078cb7430dSRaphael Zanella PetscCheck(!ctx->parent->normal, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Not implemented for the normal equations"); 1008d4f06b61SRaphael Zanella PetscCall(KSPGetOperators(ctx->ksp, &A, nullptr)); 1009d4f06b61SRaphael Zanella if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_NONE) PetscCall(PCApplyTranspose(ctx->pc, x, y)); /* y = M^-T x */ 1010d4f06b61SRaphael Zanella else { 1011d4f06b61SRaphael Zanella PetscCall(PCHPDDMDeflate_Private<true>(pc, x, y)); /* y = Q^T x */ 1012d4f06b61SRaphael Zanella if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 1013d4f06b61SRaphael Zanella if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 10148cb7430dSRaphael Zanella PetscCall(MatMult(A, y, ctx->v[1][0])); /* y = A Q^T x */ 1015d4f06b61SRaphael Zanella PetscCall(VecWAXPY(ctx->v[1][1], -1.0, ctx->v[1][0], x)); /* y = (I - A Q^T) x */ 1016d4f06b61SRaphael Zanella PetscCall(PCApplyTranspose(ctx->pc, ctx->v[1][1], ctx->v[1][0])); /* y = M^-T (I - A Q^T) x */ 1017d4f06b61SRaphael Zanella } else PetscCall(PCApplyTranspose(ctx->pc, x, ctx->v[1][0])); /* y = M^-T x */ 10188cb7430dSRaphael Zanella PetscCall(MatMultHermitianTranspose(A, ctx->v[1][0], ctx->v[1][1])); /* z = A^T y */ 1019d4f06b61SRaphael Zanella PetscCall(PCHPDDMDeflate_Private<true>(pc, ctx->v[1][1], ctx->v[1][1])); /* z = Q^T z */ 1020d4f06b61SRaphael Zanella PetscCall(VecAXPBYPCZ(y, -1.0, 1.0, 1.0, ctx->v[1][1], ctx->v[1][0])); /* y = (I - Q^T A^T) y + Q^T x */ 1021d4f06b61SRaphael Zanella } else { 1022d4f06b61SRaphael Zanella 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); 1023d4f06b61SRaphael Zanella PetscCall(PCApplyTranspose(ctx->pc, x, ctx->v[1][0])); 1024d4f06b61SRaphael Zanella PetscCall(VecAXPY(y, 1.0, ctx->v[1][0])); /* y = M^-T x + Q^T x */ 1025d4f06b61SRaphael Zanella } 1026d4f06b61SRaphael Zanella } 1027d4f06b61SRaphael Zanella PetscFunctionReturn(PETSC_SUCCESS); 1028d4f06b61SRaphael Zanella } 1029d4f06b61SRaphael Zanella 10308cb7430dSRaphael Zanella /* 10318cb7430dSRaphael Zanella PCMatApplyTranspose_HPDDMShell - Variant of PCApplyTranspose_HPDDMShell() for blocks of vectors. 10328cb7430dSRaphael Zanella 10338cb7430dSRaphael Zanella Input Parameters: 10348cb7430dSRaphael Zanella + pc - preconditioner context 10358cb7430dSRaphael Zanella - X - block of input vectors 10368cb7430dSRaphael Zanella 10378cb7430dSRaphael Zanella Output Parameter: 10388cb7430dSRaphael Zanella . Y - block of output vectors 10398cb7430dSRaphael Zanella 10408cb7430dSRaphael Zanella Level: advanced 10418cb7430dSRaphael Zanella 10428cb7430dSRaphael Zanella .seealso: [](ch_ksp), `PCHPDDM`, `PCApplyTranspose_HPDDMShell()`, `PCHPDDMCoarseCorrectionType` 10438cb7430dSRaphael Zanella */ 10448cb7430dSRaphael Zanella static PetscErrorCode PCMatApplyTranspose_HPDDMShell(PC pc, Mat X, Mat Y) 10458cb7430dSRaphael Zanella { 10468cb7430dSRaphael Zanella PC_HPDDM_Level *ctx; 10478cb7430dSRaphael Zanella PetscScalar *array; 10488cb7430dSRaphael Zanella PetscBool reset = PETSC_FALSE; 10498cb7430dSRaphael Zanella 10508cb7430dSRaphael Zanella PetscFunctionBegin; 10518cb7430dSRaphael Zanella PetscCall(PCShellGetContext(pc, &ctx)); 10528cb7430dSRaphael Zanella PetscCheck(ctx->P, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCSHELL from PCHPDDM called with no HPDDM object"); 10538cb7430dSRaphael Zanella if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_NONE) PetscCall(PCMatApplyTranspose(ctx->pc, X, Y)); 10548cb7430dSRaphael Zanella else { 10558cb7430dSRaphael Zanella PetscCall(PCHPDDMMatApply_Private<true>(ctx, Y, &reset)); 10568cb7430dSRaphael Zanella PetscCall(PCHPDDMDeflate_Private<true>(pc, X, Y)); 10578cb7430dSRaphael Zanella if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED || ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 10588cb7430dSRaphael Zanella if (ctx->parent->correction == PC_HPDDM_COARSE_CORRECTION_BALANCED) { 10598cb7430dSRaphael Zanella PetscCall(MatProductNumeric(ctx->V[2])); 10608cb7430dSRaphael Zanella PetscCall(MatCopy(ctx->V[2], ctx->V[1], SAME_NONZERO_PATTERN)); 10618cb7430dSRaphael Zanella PetscCall(MatAYPX(ctx->V[1], -1.0, X, SAME_NONZERO_PATTERN)); 10628cb7430dSRaphael Zanella PetscCall(PCMatApplyTranspose(ctx->pc, ctx->V[1], ctx->V[2])); 10638cb7430dSRaphael Zanella } else PetscCall(PCMatApplyTranspose(ctx->pc, X, ctx->V[2])); 10648cb7430dSRaphael Zanella PetscCall(MatAXPY(Y, 1.0, ctx->V[2], SAME_NONZERO_PATTERN)); 10658cb7430dSRaphael Zanella PetscCall(MatProductNumeric(ctx->V[1])); 1066847b6ef2SPierre Jolivet /* ctx->V[0] and ctx->V[1] memory regions overlap, so need to copy to ctx->V[2] and switch array */ 10678cb7430dSRaphael Zanella PetscCall(MatCopy(ctx->V[1], ctx->V[2], SAME_NONZERO_PATTERN)); 1068847b6ef2SPierre Jolivet if (reset) PetscCall(MatDenseResetArray(ctx->V[1])); 10698cb7430dSRaphael Zanella PetscCall(MatDenseGetArrayWrite(ctx->V[2], &array)); 10708cb7430dSRaphael Zanella PetscCall(MatDensePlaceArray(ctx->V[1], array)); 10718cb7430dSRaphael Zanella PetscCall(MatDenseRestoreArrayWrite(ctx->V[2], &array)); 1072847b6ef2SPierre Jolivet reset = PETSC_TRUE; 10738cb7430dSRaphael Zanella PetscCall(PCHPDDMDeflate_Private<true>(pc, ctx->V[1], ctx->V[1])); 10748cb7430dSRaphael Zanella PetscCall(MatAXPY(Y, -1.0, ctx->V[1], SAME_NONZERO_PATTERN)); 10758cb7430dSRaphael Zanella } else { 10768cb7430dSRaphael Zanella 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); 10778cb7430dSRaphael Zanella PetscCall(PCMatApplyTranspose(ctx->pc, X, ctx->V[1])); 10788cb7430dSRaphael Zanella PetscCall(MatAXPY(Y, 1.0, ctx->V[1], SAME_NONZERO_PATTERN)); 10798cb7430dSRaphael Zanella } 10808cb7430dSRaphael Zanella if (reset) PetscCall(MatDenseResetArray(ctx->V[1])); 10818cb7430dSRaphael Zanella } 10828cb7430dSRaphael Zanella PetscFunctionReturn(PETSC_SUCCESS); 10838cb7430dSRaphael Zanella } 10848cb7430dSRaphael Zanella 1085d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCDestroy_HPDDMShell(PC pc) 1086d71ae5a4SJacob Faibussowitsch { 1087f1580f4eSBarry Smith PC_HPDDM_Level *ctx; 1088f1580f4eSBarry Smith 1089f1580f4eSBarry Smith PetscFunctionBegin; 1090f1580f4eSBarry Smith PetscCall(PCShellGetContext(pc, &ctx)); 1091f1580f4eSBarry Smith PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(ctx, PETSC_TRUE)); 1092f1580f4eSBarry Smith PetscCall(VecDestroyVecs(1, &ctx->v[0])); 1093f1580f4eSBarry Smith PetscCall(VecDestroyVecs(2, &ctx->v[1])); 1094f4f49eeaSPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)ctx->pc->mat, "_HPDDM_MatProduct", nullptr)); 1095f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V)); 1096f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V + 1)); 1097f1580f4eSBarry Smith PetscCall(MatDestroy(ctx->V + 2)); 1098f1580f4eSBarry Smith PetscCall(VecDestroy(&ctx->D)); 109901e3c840SPierre Jolivet PetscCall(PetscSFDestroy(&ctx->scatter)); 1100f1580f4eSBarry Smith PetscCall(PCDestroy(&ctx->pc)); 11013ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1102f1580f4eSBarry Smith } 1103f1580f4eSBarry Smith 11049bb5c669SPierre Jolivet template <class Type, bool T = false, typename std::enable_if<std::is_same<Type, Vec>::value>::type * = nullptr> 11059bb5c669SPierre Jolivet static inline PetscErrorCode PCApply_Schur_Private(std::tuple<KSP, IS, Vec[2]> *p, PC factor, Type x, Type y) 11069bb5c669SPierre Jolivet { 11079bb5c669SPierre Jolivet PetscFunctionBegin; 11089bb5c669SPierre Jolivet PetscCall(VecISCopy(std::get<2>(*p)[0], std::get<1>(*p), SCATTER_FORWARD, x)); 11099bb5c669SPierre Jolivet if (!T) PetscCall(PCApply(factor, std::get<2>(*p)[0], std::get<2>(*p)[1])); 11109bb5c669SPierre Jolivet else PetscCall(PCApplyTranspose(factor, std::get<2>(*p)[0], std::get<2>(*p)[1])); 11119bb5c669SPierre Jolivet PetscCall(VecISCopy(std::get<2>(*p)[1], std::get<1>(*p), SCATTER_REVERSE, y)); 11129bb5c669SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 11139bb5c669SPierre Jolivet } 11149bb5c669SPierre Jolivet 11159bb5c669SPierre Jolivet template <class Type, bool = false, typename std::enable_if<std::is_same<Type, Mat>::value>::type * = nullptr> 11169bb5c669SPierre Jolivet static inline PetscErrorCode PCApply_Schur_Private(std::tuple<KSP, IS, Vec[2]> *p, PC factor, Type X, Type Y) 11179bb5c669SPierre Jolivet { 11189bb5c669SPierre Jolivet Mat B[2]; 11199bb5c669SPierre Jolivet Vec x, y; 11209bb5c669SPierre Jolivet 11219bb5c669SPierre Jolivet PetscFunctionBegin; 11229bb5c669SPierre Jolivet PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, factor->mat->rmap->n, X->cmap->n, nullptr, B)); 11239bb5c669SPierre Jolivet PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, factor->mat->rmap->n, X->cmap->n, nullptr, B + 1)); 11249bb5c669SPierre Jolivet for (PetscInt i = 0; i < X->cmap->n; ++i) { 11259bb5c669SPierre Jolivet PetscCall(MatDenseGetColumnVecRead(X, i, &x)); 11269bb5c669SPierre Jolivet PetscCall(MatDenseGetColumnVecWrite(B[0], i, &y)); 11279bb5c669SPierre Jolivet PetscCall(VecISCopy(y, std::get<1>(*p), SCATTER_FORWARD, x)); 11289bb5c669SPierre Jolivet PetscCall(MatDenseRestoreColumnVecWrite(B[0], i, &y)); 11299bb5c669SPierre Jolivet PetscCall(MatDenseRestoreColumnVecRead(X, i, &x)); 11309bb5c669SPierre Jolivet } 11319bb5c669SPierre Jolivet PetscCall(PCMatApply(factor, B[0], B[1])); 11329bb5c669SPierre Jolivet PetscCall(MatDestroy(B)); 11339bb5c669SPierre Jolivet for (PetscInt i = 0; i < X->cmap->n; ++i) { 11349bb5c669SPierre Jolivet PetscCall(MatDenseGetColumnVecRead(B[1], i, &x)); 11359bb5c669SPierre Jolivet PetscCall(MatDenseGetColumnVecWrite(Y, i, &y)); 11369bb5c669SPierre Jolivet PetscCall(VecISCopy(x, std::get<1>(*p), SCATTER_REVERSE, y)); 11379bb5c669SPierre Jolivet PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &y)); 11389bb5c669SPierre Jolivet PetscCall(MatDenseRestoreColumnVecRead(B[1], i, &x)); 11399bb5c669SPierre Jolivet } 11409bb5c669SPierre Jolivet PetscCall(MatDestroy(B + 1)); 11419bb5c669SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 11429bb5c669SPierre Jolivet } 11439bb5c669SPierre Jolivet 11449bb5c669SPierre Jolivet template <class Type = Vec, bool T = false> 11459bb5c669SPierre Jolivet static PetscErrorCode PCApply_Schur(PC pc, Type x, Type y) 11469bb5c669SPierre Jolivet { 11479bb5c669SPierre Jolivet PC factor; 11489bb5c669SPierre Jolivet Mat A; 11499bb5c669SPierre Jolivet MatSolverType type; 11509bb5c669SPierre Jolivet PetscBool flg; 11519bb5c669SPierre Jolivet std::tuple<KSP, IS, Vec[2]> *p; 11529bb5c669SPierre Jolivet 11539bb5c669SPierre Jolivet PetscFunctionBegin; 11549bb5c669SPierre Jolivet PetscCall(PCShellGetContext(pc, &p)); 11559bb5c669SPierre Jolivet PetscCall(KSPGetPC(std::get<0>(*p), &factor)); 11569bb5c669SPierre Jolivet PetscCall(PCFactorGetMatSolverType(factor, &type)); 11579bb5c669SPierre Jolivet PetscCall(PCFactorGetMatrix(factor, &A)); 11589bb5c669SPierre Jolivet PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg)); 11599bb5c669SPierre Jolivet if (flg) { 11609bb5c669SPierre Jolivet PetscCheck(PetscDefined(HAVE_MUMPS), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent MatSolverType"); 11619bb5c669SPierre Jolivet PetscCall(MatMumpsSetIcntl(A, 26, 0)); 11629bb5c669SPierre Jolivet } else { 11639bb5c669SPierre Jolivet PetscCall(PetscStrcmp(type, MATSOLVERMKL_PARDISO, &flg)); 11649bb5c669SPierre Jolivet PetscCheck(flg && PetscDefined(HAVE_MKL_PARDISO), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent MatSolverType"); 11659bb5c669SPierre Jolivet flg = PETSC_FALSE; 11669bb5c669SPierre Jolivet #if PetscDefined(HAVE_MKL_PARDISO) 11679bb5c669SPierre Jolivet PetscCall(MatMkl_PardisoSetCntl(A, 70, 1)); 11689bb5c669SPierre Jolivet #endif 11699bb5c669SPierre Jolivet } 11709bb5c669SPierre Jolivet PetscCall(PCApply_Schur_Private<Type, T>(p, factor, x, y)); 11719bb5c669SPierre Jolivet if (flg) { 11729bb5c669SPierre Jolivet PetscCall(MatMumpsSetIcntl(A, 26, -1)); 11739bb5c669SPierre Jolivet } else { 11749bb5c669SPierre Jolivet #if PetscDefined(HAVE_MKL_PARDISO) 11759bb5c669SPierre Jolivet PetscCall(MatMkl_PardisoSetCntl(A, 70, 0)); 11769bb5c669SPierre Jolivet #endif 11779bb5c669SPierre Jolivet } 11789bb5c669SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 11799bb5c669SPierre Jolivet } 11809bb5c669SPierre Jolivet 11819bb5c669SPierre Jolivet static PetscErrorCode PCDestroy_Schur(PC pc) 11829bb5c669SPierre Jolivet { 11839bb5c669SPierre Jolivet std::tuple<KSP, IS, Vec[2]> *p; 11849bb5c669SPierre Jolivet 11859bb5c669SPierre Jolivet PetscFunctionBegin; 11869bb5c669SPierre Jolivet PetscCall(PCShellGetContext(pc, &p)); 11879bb5c669SPierre Jolivet PetscCall(ISDestroy(&std::get<1>(*p))); 11889bb5c669SPierre Jolivet PetscCall(VecDestroy(std::get<2>(*p))); 11899bb5c669SPierre Jolivet PetscCall(VecDestroy(std::get<2>(*p) + 1)); 11909bb5c669SPierre Jolivet PetscCall(PetscFree(p)); 11919bb5c669SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 11929bb5c669SPierre Jolivet } 11939bb5c669SPierre Jolivet 1194d4f06b61SRaphael Zanella template <bool transpose> 1195d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSolve_Private(const PC_HPDDM_Level *ctx, PetscScalar *rhs, const unsigned short &mu) 1196d71ae5a4SJacob Faibussowitsch { 1197f1580f4eSBarry Smith Mat B, X; 1198f1580f4eSBarry Smith PetscInt n, N, j = 0; 1199f1580f4eSBarry Smith 1200f1580f4eSBarry Smith PetscFunctionBegin; 1201db4a47b3SPierre Jolivet PetscCall(KSPGetOperators(ctx->ksp, &B, nullptr)); 1202db4a47b3SPierre Jolivet PetscCall(MatGetLocalSize(B, &n, nullptr)); 1203db4a47b3SPierre Jolivet PetscCall(MatGetSize(B, &N, nullptr)); 1204f1580f4eSBarry Smith if (ctx->parent->log_separate) { 1205f1580f4eSBarry Smith j = std::distance(ctx->parent->levels, std::find(ctx->parent->levels, ctx->parent->levels + ctx->parent->N, ctx)); 1206db4a47b3SPierre Jolivet PetscCall(PetscLogEventBegin(PC_HPDDM_Solve[j], ctx->ksp, nullptr, nullptr, nullptr)); 1207f1580f4eSBarry Smith } 1208f1580f4eSBarry Smith if (mu == 1) { 1209f1580f4eSBarry Smith if (!ctx->ksp->vec_rhs) { 1210db4a47b3SPierre Jolivet PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)ctx->ksp), 1, n, N, nullptr, &ctx->ksp->vec_rhs)); 1211f1580f4eSBarry Smith PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)ctx->ksp), n, N, &ctx->ksp->vec_sol)); 1212f1580f4eSBarry Smith } 1213f1580f4eSBarry Smith PetscCall(VecPlaceArray(ctx->ksp->vec_rhs, rhs)); 1214d4f06b61SRaphael Zanella if (!transpose) PetscCall(KSPSolve(ctx->ksp, nullptr, nullptr)); 1215d4f06b61SRaphael Zanella else PetscCall(KSPSolveTranspose(ctx->ksp, nullptr, nullptr)); 1216f1580f4eSBarry Smith PetscCall(VecCopy(ctx->ksp->vec_sol, ctx->ksp->vec_rhs)); 1217f1580f4eSBarry Smith PetscCall(VecResetArray(ctx->ksp->vec_rhs)); 1218f1580f4eSBarry Smith } else { 1219f1580f4eSBarry Smith PetscCall(MatCreateDense(PetscObjectComm((PetscObject)ctx->ksp), n, PETSC_DECIDE, N, mu, rhs, &B)); 1220db4a47b3SPierre Jolivet PetscCall(MatCreateDense(PetscObjectComm((PetscObject)ctx->ksp), n, PETSC_DECIDE, N, mu, nullptr, &X)); 1221f1580f4eSBarry Smith PetscCall(KSPMatSolve(ctx->ksp, B, X)); 1222f1580f4eSBarry Smith PetscCall(MatCopy(X, B, SAME_NONZERO_PATTERN)); 1223f1580f4eSBarry Smith PetscCall(MatDestroy(&X)); 1224f1580f4eSBarry Smith PetscCall(MatDestroy(&B)); 1225f1580f4eSBarry Smith } 1226db4a47b3SPierre Jolivet if (ctx->parent->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Solve[j], ctx->ksp, nullptr, nullptr, nullptr)); 12273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1228f1580f4eSBarry Smith } 1229f1580f4eSBarry Smith 1230d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetUpNeumannOverlap_Private(PC pc) 1231d71ae5a4SJacob Faibussowitsch { 1232f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 1233f1580f4eSBarry Smith 1234f1580f4eSBarry Smith PetscFunctionBegin; 1235f1580f4eSBarry Smith if (data->setup) { 1236f1580f4eSBarry Smith Mat P; 1237db4a47b3SPierre Jolivet Vec x, xt = nullptr; 1238f1580f4eSBarry Smith PetscReal t = 0.0, s = 0.0; 1239f1580f4eSBarry Smith 1240db4a47b3SPierre Jolivet PetscCall(PCGetOperators(pc, nullptr, &P)); 1241f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)P, "__SNES_latest_X", (PetscObject *)&x)); 1242f1580f4eSBarry Smith PetscCallBack("PCHPDDM Neumann callback", (*data->setup)(data->aux, t, x, xt, s, data->is, data->setup_ctx)); 1243f1580f4eSBarry Smith } 12443ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1245f1580f4eSBarry Smith } 1246f1580f4eSBarry Smith 1247d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMCreateSubMatrices_Private(Mat mat, PetscInt n, const IS *, const IS *, MatReuse scall, Mat *submat[]) 1248d71ae5a4SJacob Faibussowitsch { 1249f1580f4eSBarry Smith Mat A; 125013044ca3SPierre Jolivet PetscBool flg; 1251f1580f4eSBarry Smith 1252f1580f4eSBarry Smith PetscFunctionBegin; 1253f1580f4eSBarry Smith PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "MatCreateSubMatrices() called to extract %" PetscInt_FMT " submatrices, which is different than 1", n); 1254f1580f4eSBarry Smith /* previously composed Mat */ 1255f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)mat, "_PCHPDDM_SubMatrices", (PetscObject *)&A)); 1256f1580f4eSBarry Smith PetscCheck(A, PETSC_COMM_SELF, PETSC_ERR_PLIB, "SubMatrices not found in Mat"); 125713044ca3SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSCHURCOMPLEMENT, &flg)); /* MATSCHURCOMPLEMENT has neither a MatDuplicate() nor a MatCopy() implementation */ 1258f1580f4eSBarry Smith if (scall == MAT_INITIAL_MATRIX) { 12596f2c871aSStefano Zampini PetscCall(PetscCalloc1(2, submat)); /* allocate an extra Mat to avoid errors in MatDestroySubMatrices_Dummy() */ 126013044ca3SPierre Jolivet if (!flg) PetscCall(MatDuplicate(A, MAT_COPY_VALUES, *submat)); 126113044ca3SPierre Jolivet } else if (!flg) PetscCall(MatCopy(A, (*submat)[0], SAME_NONZERO_PATTERN)); 126213044ca3SPierre Jolivet if (flg) { 1263bf583f0cSPierre Jolivet PetscCall(MatDestroy(*submat)); /* previously created Mat has to be destroyed */ 126413044ca3SPierre Jolivet (*submat)[0] = A; 126513044ca3SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)A)); 126613044ca3SPierre Jolivet } 12673ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1268f1580f4eSBarry Smith } 1269f1580f4eSBarry Smith 1270d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMCommunicationAvoidingPCASM_Private(PC pc, Mat C, PetscBool sorted) 1271d71ae5a4SJacob Faibussowitsch { 127257d50842SBarry Smith PetscErrorCodeFn *op; 1273f1580f4eSBarry Smith 1274f1580f4eSBarry Smith PetscFunctionBegin; 1275f1580f4eSBarry Smith /* previously-composed Mat */ 1276f1580f4eSBarry Smith PetscCall(PetscObjectCompose((PetscObject)pc->pmat, "_PCHPDDM_SubMatrices", (PetscObject)C)); 1277f1580f4eSBarry Smith PetscCall(MatGetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, &op)); 1278f1580f4eSBarry Smith /* trick suggested by Barry https://lists.mcs.anl.gov/pipermail/petsc-dev/2020-January/025491.html */ 127957d50842SBarry Smith PetscCall(MatSetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, (PetscErrorCodeFn *)PCHPDDMCreateSubMatrices_Private)); 1280f1580f4eSBarry Smith if (sorted) PetscCall(PCASMSetSortIndices(pc, PETSC_FALSE)); /* everything is already sorted */ 1281f1580f4eSBarry Smith PetscCall(PCSetFromOptions(pc)); /* otherwise -pc_hpddm_levels_1_pc_asm_sub_mat_type is not used */ 1282f1580f4eSBarry Smith PetscCall(PCSetUp(pc)); 1283f1580f4eSBarry Smith /* reset MatCreateSubMatrices() */ 1284f1580f4eSBarry Smith PetscCall(MatSetOperation(pc->pmat, MATOP_CREATE_SUBMATRICES, op)); 1285db4a47b3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)pc->pmat, "_PCHPDDM_SubMatrices", nullptr)); 12863ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1287f1580f4eSBarry Smith } 1288f1580f4eSBarry Smith 1289d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMPermute_Private(IS is, IS in_is, IS *out_is, Mat in_C, Mat *out_C, IS *p) 1290d71ae5a4SJacob Faibussowitsch { 1291f1580f4eSBarry Smith IS perm; 1292f1580f4eSBarry Smith const PetscInt *ptr; 1293811e8887SPierre Jolivet PetscInt *concatenate, size, bs; 1294f1580f4eSBarry Smith std::map<PetscInt, PetscInt> order; 1295f1580f4eSBarry Smith PetscBool sorted; 1296f1580f4eSBarry Smith 1297f1580f4eSBarry Smith PetscFunctionBegin; 1298cf67ef9dSPierre Jolivet PetscValidHeaderSpecific(is, IS_CLASSID, 1); 1299cf67ef9dSPierre Jolivet PetscValidHeaderSpecific(in_C, MAT_CLASSID, 4); 1300f1580f4eSBarry Smith PetscCall(ISSorted(is, &sorted)); 1301f1580f4eSBarry Smith if (!sorted) { 1302f1580f4eSBarry Smith PetscCall(ISGetLocalSize(is, &size)); 1303f1580f4eSBarry Smith PetscCall(ISGetIndices(is, &ptr)); 1304b07dfdedSPierre Jolivet PetscCall(ISGetBlockSize(is, &bs)); 1305f1580f4eSBarry Smith /* MatCreateSubMatrices(), called by PCASM, follows the global numbering of Pmat */ 1306811e8887SPierre Jolivet for (PetscInt n = 0; n < size; n += bs) order.insert(std::make_pair(ptr[n] / bs, n / bs)); 1307f1580f4eSBarry Smith PetscCall(ISRestoreIndices(is, &ptr)); 1308b07dfdedSPierre Jolivet size /= bs; 1309f1580f4eSBarry Smith if (out_C) { 1310f1580f4eSBarry Smith PetscCall(PetscMalloc1(size, &concatenate)); 1311f1580f4eSBarry Smith for (const std::pair<const PetscInt, PetscInt> &i : order) *concatenate++ = i.second; 1312f1580f4eSBarry Smith concatenate -= size; 1313b07dfdedSPierre Jolivet PetscCall(ISCreateBlock(PetscObjectComm((PetscObject)in_C), bs, size, concatenate, PETSC_OWN_POINTER, &perm)); 1314f1580f4eSBarry Smith PetscCall(ISSetPermutation(perm)); 1315f1580f4eSBarry Smith /* permute user-provided Mat so that it matches with MatCreateSubMatrices() numbering */ 1316f1580f4eSBarry Smith PetscCall(MatPermute(in_C, perm, perm, out_C)); 1317f1580f4eSBarry Smith if (p) *p = perm; 1318f1580f4eSBarry Smith else PetscCall(ISDestroy(&perm)); /* no need to save the permutation */ 1319f1580f4eSBarry Smith } 1320f1580f4eSBarry Smith if (out_is) { 1321f1580f4eSBarry Smith PetscCall(PetscMalloc1(size, &concatenate)); 1322f1580f4eSBarry Smith for (const std::pair<const PetscInt, PetscInt> &i : order) *concatenate++ = i.first; 1323f1580f4eSBarry Smith concatenate -= size; 1324f1580f4eSBarry Smith /* permute user-provided IS so that it matches with MatCreateSubMatrices() numbering */ 1325b07dfdedSPierre Jolivet PetscCall(ISCreateBlock(PetscObjectComm((PetscObject)in_is), bs, size, concatenate, PETSC_OWN_POINTER, out_is)); 1326f1580f4eSBarry Smith } 1327f1580f4eSBarry Smith } else { /* input IS is sorted, nothing to permute, simply duplicate inputs when needed */ 1328f1580f4eSBarry Smith if (out_C) PetscCall(MatDuplicate(in_C, MAT_COPY_VALUES, out_C)); 1329f1580f4eSBarry Smith if (out_is) PetscCall(ISDuplicate(in_is, out_is)); 1330f1580f4eSBarry Smith } 13313ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1332f1580f4eSBarry Smith } 1333f1580f4eSBarry Smith 1334281f8ce6SPierre Jolivet static PetscErrorCode PCHPDDMCheckSymmetry_Private(PC pc, Mat A01, Mat A10, Mat *B01 = nullptr) 133513044ca3SPierre Jolivet { 133613044ca3SPierre Jolivet Mat T, U = nullptr, B = nullptr; 133713044ca3SPierre Jolivet IS z; 1338281f8ce6SPierre Jolivet PetscBool flg, conjugate = PETSC_FALSE; 133913044ca3SPierre Jolivet 134013044ca3SPierre Jolivet PetscFunctionBegin; 1341281f8ce6SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, &flg)); 1342281f8ce6SPierre Jolivet if (B01) *B01 = nullptr; 1343281f8ce6SPierre Jolivet if (flg) { 1344811e8887SPierre 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)); 1345811e8887SPierre Jolivet PetscCall(MatTransposeGetMat(A10, &U)); 1346811e8887SPierre Jolivet } else { 1347281f8ce6SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, &flg)); 1348281f8ce6SPierre Jolivet if (flg) { 1349811e8887SPierre 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)); 1350811e8887SPierre Jolivet PetscCall(MatHermitianTransposeGetMat(A10, &U)); 1351281f8ce6SPierre Jolivet conjugate = PETSC_TRUE; 1352811e8887SPierre Jolivet } 135313044ca3SPierre Jolivet } 135413044ca3SPierre Jolivet if (U) PetscCall(MatDuplicate(U, MAT_COPY_VALUES, &T)); 135513044ca3SPierre Jolivet else PetscCall(MatHermitianTranspose(A10, MAT_INITIAL_MATRIX, &T)); 1356281f8ce6SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)A01, MATTRANSPOSEVIRTUAL, &flg)); 1357281f8ce6SPierre Jolivet if (flg) { 1358811e8887SPierre 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)); 135913044ca3SPierre Jolivet PetscCall(MatTransposeGetMat(A01, &A01)); 136013044ca3SPierre Jolivet PetscCall(MatTranspose(A01, MAT_INITIAL_MATRIX, &B)); 136113044ca3SPierre Jolivet A01 = B; 136213044ca3SPierre Jolivet } else { 1363281f8ce6SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)A01, MATHERMITIANTRANSPOSEVIRTUAL, &flg)); 1364281f8ce6SPierre Jolivet if (flg) { 1365811e8887SPierre 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)); 136613044ca3SPierre Jolivet PetscCall(MatHermitianTransposeGetMat(A01, &A01)); 136713044ca3SPierre Jolivet PetscCall(MatHermitianTranspose(A01, MAT_INITIAL_MATRIX, &B)); 136813044ca3SPierre Jolivet A01 = B; 136913044ca3SPierre Jolivet } 137013044ca3SPierre Jolivet } 1371281f8ce6SPierre Jolivet PetscCall(PetscLayoutCompare(T->rmap, A01->rmap, &flg)); 1372281f8ce6SPierre Jolivet if (flg) { 1373281f8ce6SPierre Jolivet PetscCall(PetscLayoutCompare(T->cmap, A01->cmap, &flg)); 1374281f8ce6SPierre Jolivet if (flg) { 137513044ca3SPierre Jolivet PetscCall(MatFindZeroRows(A01, &z)); /* for essential boundary conditions, some implementations will */ 137613044ca3SPierre Jolivet if (z) { /* zero rows in [P00 A01] except for the diagonal of P00 */ 1377281f8ce6SPierre Jolivet if (B01) PetscCall(MatDuplicate(T, MAT_COPY_VALUES, B01)); 137813044ca3SPierre Jolivet PetscCall(MatSetOption(T, MAT_NO_OFF_PROC_ZERO_ROWS, PETSC_TRUE)); 137913044ca3SPierre Jolivet PetscCall(MatZeroRowsIS(T, z, 0.0, nullptr, nullptr)); /* corresponding zero rows from A01 */ 1380281f8ce6SPierre Jolivet } 1381281f8ce6SPierre Jolivet PetscCall(MatMultEqual(A01, T, 20, &flg)); 1382281f8ce6SPierre Jolivet if (!B01) PetscCheck(flg, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "A01 != A10^T"); 1383281f8ce6SPierre Jolivet else { 1384281f8ce6SPierre Jolivet PetscCall(PetscInfo(pc, "A01 and A10^T are equal? %s\n", PetscBools[flg])); 1385281f8ce6SPierre Jolivet if (!flg) { 1386281f8ce6SPierre Jolivet if (z) PetscCall(MatDestroy(&T)); 1387281f8ce6SPierre Jolivet else *B01 = T; 1388281f8ce6SPierre Jolivet flg = PETSC_TRUE; 1389281f8ce6SPierre Jolivet } else PetscCall(MatDestroy(B01)); 1390281f8ce6SPierre Jolivet } 139113044ca3SPierre Jolivet PetscCall(ISDestroy(&z)); 139213044ca3SPierre Jolivet } 139313044ca3SPierre Jolivet } 1394281f8ce6SPierre Jolivet if (!flg) PetscCall(PetscInfo(pc, "A01 and A10^T have non-congruent layouts, cannot test for equality\n")); 1395281f8ce6SPierre Jolivet if (!B01 || !*B01) PetscCall(MatDestroy(&T)); 1396281f8ce6SPierre Jolivet else if (conjugate) PetscCall(MatConjugate(T)); 139713044ca3SPierre Jolivet PetscCall(MatDestroy(&B)); 139813044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 139913044ca3SPierre Jolivet } 140013044ca3SPierre Jolivet 1401d16c0b94SPierre Jolivet static PetscErrorCode PCHPDDMCheckInclusion_Private(PC pc, IS is, IS is_local, PetscBool check) 1402d16c0b94SPierre Jolivet { 1403d16c0b94SPierre Jolivet IS intersect; 1404d16c0b94SPierre Jolivet const char *str = "IS of the auxiliary Mat does not include all local rows of A"; 1405d16c0b94SPierre Jolivet PetscBool equal; 1406d16c0b94SPierre Jolivet 1407d16c0b94SPierre Jolivet PetscFunctionBegin; 1408d16c0b94SPierre Jolivet PetscCall(ISIntersect(is, is_local, &intersect)); 1409d16c0b94SPierre Jolivet PetscCall(ISEqualUnsorted(is_local, intersect, &equal)); 1410d16c0b94SPierre Jolivet PetscCall(ISDestroy(&intersect)); 1411d16c0b94SPierre Jolivet if (check) PetscCheck(equal, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "%s", str); 1412d16c0b94SPierre Jolivet else if (!equal) PetscCall(PetscInfo(pc, "%s\n", str)); 1413d16c0b94SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 1414d16c0b94SPierre Jolivet } 1415d16c0b94SPierre Jolivet 1416*8a8e6071SPierre Jolivet static PetscErrorCode PCHPDDMCheckMatStructure_Private(PC pc, Mat A, Mat B) 1417*8a8e6071SPierre Jolivet { 1418*8a8e6071SPierre Jolivet Mat X, Y; 1419*8a8e6071SPierre Jolivet const PetscInt *i[2], *j[2]; 1420*8a8e6071SPierre Jolivet PetscBool flg = PETSC_TRUE; 1421*8a8e6071SPierre Jolivet 1422*8a8e6071SPierre Jolivet PetscFunctionBegin; 1423*8a8e6071SPierre Jolivet PetscCall(MatConvert(A, MATAIJ, MAT_INITIAL_MATRIX, &X)); /* no common way to compare sparsity pattern, so just convert to MATSEQAIJ */ 1424*8a8e6071SPierre Jolivet PetscCall(MatConvert(B, MATAIJ, MAT_INITIAL_MATRIX, &Y)); /* the second Mat (B = Neumann) should have a SUBSET_NONZERO_PATTERN MatStructure of the first one (A = Dirichlet) */ 1425*8a8e6071SPierre Jolivet PetscCall(MatSeqAIJGetCSRAndMemType(X, &i[0], &j[0], nullptr, nullptr)); 1426*8a8e6071SPierre Jolivet PetscCall(MatSeqAIJGetCSRAndMemType(Y, &i[1], &j[1], nullptr, nullptr)); 1427*8a8e6071SPierre Jolivet for (PetscInt row = 0; (row < X->rmap->n) && flg; ++row) { 1428*8a8e6071SPierre Jolivet const PetscInt n = i[0][row + 1] - i[0][row]; 1429*8a8e6071SPierre Jolivet 1430*8a8e6071SPierre Jolivet for (PetscInt k = i[1][row]; k < i[1][row + 1]; ++k) { 1431*8a8e6071SPierre Jolivet PetscInt loc; 1432*8a8e6071SPierre Jolivet 1433*8a8e6071SPierre Jolivet PetscCall(PetscFindInt(j[1][k], n, j[0] + i[0][row], &loc)); 1434*8a8e6071SPierre Jolivet if (loc < 0) { 1435*8a8e6071SPierre Jolivet flg = PETSC_FALSE; 1436*8a8e6071SPierre Jolivet break; 1437*8a8e6071SPierre Jolivet } 1438*8a8e6071SPierre Jolivet } 1439*8a8e6071SPierre Jolivet } 1440*8a8e6071SPierre Jolivet PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &flg, 1, MPI_C_BOOL, MPI_LAND, PetscObjectComm((PetscObject)pc))); 1441*8a8e6071SPierre Jolivet PetscCheck(flg, PetscObjectComm((PetscObject)pc), PETSC_ERR_USER_INPUT, "Auxiliary Mat is supposedly the local Neumann matrix but it has a sparsity pattern which is not a subset of the one of the local assembled matrix"); 1442*8a8e6071SPierre Jolivet PetscCall(MatDestroy(&Y)); 1443*8a8e6071SPierre Jolivet PetscCall(MatDestroy(&X)); 1444*8a8e6071SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 1445*8a8e6071SPierre Jolivet } 1446*8a8e6071SPierre Jolivet 1447d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMDestroySubMatrices_Private(PetscBool flg, PetscBool algebraic, Mat *sub) 1448d71ae5a4SJacob Faibussowitsch { 1449f1580f4eSBarry Smith IS is; 1450f1580f4eSBarry Smith 1451f1580f4eSBarry Smith PetscFunctionBegin; 1452f1580f4eSBarry Smith if (!flg) { 1453f1580f4eSBarry Smith if (algebraic) { 1454f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)sub[0], "_PCHPDDM_Embed", (PetscObject *)&is)); 1455f1580f4eSBarry Smith PetscCall(ISDestroy(&is)); 1456db4a47b3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Embed", nullptr)); 1457db4a47b3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Compact", nullptr)); 1458f1580f4eSBarry Smith } 1459f1580f4eSBarry Smith PetscCall(MatDestroySubMatrices(algebraic ? 2 : 1, &sub)); 1460f1580f4eSBarry Smith } 14613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1462f1580f4eSBarry Smith } 1463f1580f4eSBarry Smith 1464d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMAlgebraicAuxiliaryMat_Private(Mat P, IS *is, Mat *sub[], PetscBool block) 1465d71ae5a4SJacob Faibussowitsch { 1466f1580f4eSBarry Smith IS icol[3], irow[2]; 1467f1580f4eSBarry Smith Mat *M, Q; 1468f1580f4eSBarry Smith PetscReal *ptr; 146958b7e2c1SStefano Zampini PetscInt *idx, p = 0, bs = P->cmap->bs; 1470f1580f4eSBarry Smith PetscBool flg; 1471f1580f4eSBarry Smith 1472f1580f4eSBarry Smith PetscFunctionBegin; 1473f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, P->cmap->N, 0, 1, icol + 2)); 1474f1580f4eSBarry Smith PetscCall(ISSetBlockSize(icol[2], bs)); 1475f1580f4eSBarry Smith PetscCall(ISSetIdentity(icol[2])); 1476f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)P, MATMPISBAIJ, &flg)); 1477f1580f4eSBarry Smith if (flg) { 1478f1580f4eSBarry Smith /* MatCreateSubMatrices() does not handle MATMPISBAIJ properly when iscol != isrow, so convert first to MATMPIBAIJ */ 1479f1580f4eSBarry Smith PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &Q)); 1480f1580f4eSBarry Smith std::swap(P, Q); 1481f1580f4eSBarry Smith } 1482f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(P, 1, is, icol + 2, MAT_INITIAL_MATRIX, &M)); 1483f1580f4eSBarry Smith if (flg) { 1484f1580f4eSBarry Smith std::swap(P, Q); 1485f1580f4eSBarry Smith PetscCall(MatDestroy(&Q)); 1486f1580f4eSBarry Smith } 1487f1580f4eSBarry Smith PetscCall(ISDestroy(icol + 2)); 1488f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, M[0]->rmap->N, 0, 1, irow)); 1489f1580f4eSBarry Smith PetscCall(ISSetBlockSize(irow[0], bs)); 1490f1580f4eSBarry Smith PetscCall(ISSetIdentity(irow[0])); 1491f1580f4eSBarry Smith if (!block) { 1492b07dfdedSPierre Jolivet PetscCall(PetscMalloc2(P->cmap->N, &ptr, P->cmap->N / bs, &idx)); 1493f1580f4eSBarry Smith PetscCall(MatGetColumnNorms(M[0], NORM_INFINITY, ptr)); 1494f1580f4eSBarry Smith /* check for nonzero columns so that M[0] may be expressed in compact form */ 1495811e8887SPierre Jolivet for (PetscInt n = 0; n < P->cmap->N; n += bs) { 1496b07dfdedSPierre Jolivet if (std::find_if(ptr + n, ptr + n + bs, [](PetscReal v) { return v > PETSC_MACHINE_EPSILON; }) != ptr + n + bs) idx[p++] = n / bs; 1497f1580f4eSBarry Smith } 1498b07dfdedSPierre Jolivet PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, p, idx, PETSC_USE_POINTER, icol + 1)); 1499f1580f4eSBarry Smith PetscCall(ISSetInfo(icol[1], IS_SORTED, IS_GLOBAL, PETSC_TRUE, PETSC_TRUE)); 1500f1580f4eSBarry Smith PetscCall(ISEmbed(*is, icol[1], PETSC_FALSE, icol + 2)); 1501f1580f4eSBarry Smith irow[1] = irow[0]; 1502f1580f4eSBarry 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 */ 1503f1580f4eSBarry Smith icol[0] = is[0]; 1504f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(M[0], 2, irow, icol, MAT_INITIAL_MATRIX, sub)); 1505f1580f4eSBarry Smith PetscCall(ISDestroy(icol + 1)); 1506f1580f4eSBarry Smith PetscCall(PetscFree2(ptr, idx)); 1507f1580f4eSBarry Smith /* IS used to go back and forth between the augmented and the original local linear system, see eq. (3.4) of [2022b] */ 1508f1580f4eSBarry Smith PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Embed", (PetscObject)icol[2])); 1509f1580f4eSBarry Smith /* Mat used in eq. (3.1) of [2022b] */ 1510f1580f4eSBarry Smith PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Compact", (PetscObject)(*sub)[1])); 1511f1580f4eSBarry Smith } else { 1512f1580f4eSBarry Smith Mat aux; 1513811e8887SPierre Jolivet 1514f1580f4eSBarry Smith PetscCall(MatSetOption(M[0], MAT_SUBMAT_SINGLEIS, PETSC_TRUE)); 1515f1580f4eSBarry Smith /* diagonal block of the overlapping rows */ 1516f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(M[0], 1, irow, is, MAT_INITIAL_MATRIX, sub)); 1517f1580f4eSBarry Smith PetscCall(MatDuplicate((*sub)[0], MAT_COPY_VALUES, &aux)); 1518f1580f4eSBarry Smith PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE)); 1519f1580f4eSBarry Smith if (bs == 1) { /* scalar case */ 1520f1580f4eSBarry Smith Vec sum[2]; 1521811e8887SPierre Jolivet 1522f1580f4eSBarry Smith PetscCall(MatCreateVecs(aux, sum, sum + 1)); 1523f1580f4eSBarry Smith PetscCall(MatGetRowSum(M[0], sum[0])); 1524f1580f4eSBarry Smith PetscCall(MatGetRowSum(aux, sum[1])); 1525f1580f4eSBarry Smith /* off-diagonal block row sum (full rows - diagonal block rows) */ 1526f1580f4eSBarry Smith PetscCall(VecAXPY(sum[0], -1.0, sum[1])); 1527f1580f4eSBarry Smith /* subdomain matrix plus off-diagonal block row sum */ 1528f1580f4eSBarry Smith PetscCall(MatDiagonalSet(aux, sum[0], ADD_VALUES)); 1529f1580f4eSBarry Smith PetscCall(VecDestroy(sum)); 1530f1580f4eSBarry Smith PetscCall(VecDestroy(sum + 1)); 1531f1580f4eSBarry Smith } else { /* vectorial case */ 1532f1580f4eSBarry Smith /* TODO: missing MatGetValuesBlocked(), so the code below is */ 1533f1580f4eSBarry Smith /* an extension of the scalar case for when bs > 1, but it could */ 1534f1580f4eSBarry Smith /* be more efficient by avoiding all these MatMatMult() */ 1535f1580f4eSBarry Smith Mat sum[2], ones; 1536f1580f4eSBarry Smith PetscScalar *ptr; 1537811e8887SPierre Jolivet 1538f1580f4eSBarry Smith PetscCall(PetscCalloc1(M[0]->cmap->n * bs, &ptr)); 1539f1580f4eSBarry Smith PetscCall(MatCreateDense(PETSC_COMM_SELF, M[0]->cmap->n, bs, M[0]->cmap->n, bs, ptr, &ones)); 1540811e8887SPierre Jolivet for (PetscInt n = 0; n < M[0]->cmap->n; n += bs) { 1541f1580f4eSBarry Smith for (p = 0; p < bs; ++p) ptr[n + p * (M[0]->cmap->n + 1)] = 1.0; 1542f1580f4eSBarry Smith } 1543fb842aefSJose E. Roman PetscCall(MatMatMult(M[0], ones, MAT_INITIAL_MATRIX, PETSC_CURRENT, sum)); 1544f1580f4eSBarry Smith PetscCall(MatDestroy(&ones)); 1545f1580f4eSBarry Smith PetscCall(MatCreateDense(PETSC_COMM_SELF, aux->cmap->n, bs, aux->cmap->n, bs, ptr, &ones)); 1546f1580f4eSBarry Smith PetscCall(MatDenseSetLDA(ones, M[0]->cmap->n)); 1547fb842aefSJose E. Roman PetscCall(MatMatMult(aux, ones, MAT_INITIAL_MATRIX, PETSC_CURRENT, sum + 1)); 1548f1580f4eSBarry Smith PetscCall(MatDestroy(&ones)); 1549f1580f4eSBarry Smith PetscCall(PetscFree(ptr)); 1550f1580f4eSBarry Smith /* off-diagonal block row sum (full rows - diagonal block rows) */ 1551f1580f4eSBarry Smith PetscCall(MatAXPY(sum[0], -1.0, sum[1], SAME_NONZERO_PATTERN)); 1552f1580f4eSBarry Smith PetscCall(MatDestroy(sum + 1)); 1553f1580f4eSBarry Smith /* re-order values to be consistent with MatSetValuesBlocked() */ 1554f1580f4eSBarry Smith /* equivalent to MatTranspose() which does not truly handle */ 1555f1580f4eSBarry Smith /* MAT_INPLACE_MATRIX in the rectangular case, as it calls PetscMalloc() */ 1556f1580f4eSBarry Smith PetscCall(MatDenseGetArrayWrite(sum[0], &ptr)); 1557f1580f4eSBarry Smith HPDDM::Wrapper<PetscScalar>::imatcopy<'T'>(bs, sum[0]->rmap->n, ptr, sum[0]->rmap->n, bs); 1558f1580f4eSBarry Smith /* subdomain matrix plus off-diagonal block row sum */ 1559811e8887SPierre Jolivet for (PetscInt n = 0; n < aux->cmap->n / bs; ++n) PetscCall(MatSetValuesBlocked(aux, 1, &n, 1, &n, ptr + n * bs * bs, ADD_VALUES)); 1560f1580f4eSBarry Smith PetscCall(MatAssemblyBegin(aux, MAT_FINAL_ASSEMBLY)); 1561f1580f4eSBarry Smith PetscCall(MatAssemblyEnd(aux, MAT_FINAL_ASSEMBLY)); 1562f1580f4eSBarry Smith PetscCall(MatDenseRestoreArrayWrite(sum[0], &ptr)); 1563f1580f4eSBarry Smith PetscCall(MatDestroy(sum)); 1564f1580f4eSBarry Smith } 1565f1580f4eSBarry Smith PetscCall(MatSetOption(aux, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_TRUE)); 1566f1580f4eSBarry Smith /* left-hand side of GenEO, with the same sparsity pattern as PCASM subdomain solvers */ 1567f1580f4eSBarry Smith PetscCall(PetscObjectCompose((PetscObject)(*sub)[0], "_PCHPDDM_Neumann_Mat", (PetscObject)aux)); 1568f1580f4eSBarry Smith } 1569f1580f4eSBarry Smith PetscCall(ISDestroy(irow)); 1570f1580f4eSBarry Smith PetscCall(MatDestroySubMatrices(1, &M)); 15713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1572f1580f4eSBarry Smith } 1573f1580f4eSBarry Smith 157413044ca3SPierre Jolivet static PetscErrorCode PCApply_Nest(PC pc, Vec x, Vec y) 157513044ca3SPierre Jolivet { 157613044ca3SPierre Jolivet Mat A; 157713044ca3SPierre Jolivet MatSolverType type; 157813044ca3SPierre Jolivet IS is[2]; 157913044ca3SPierre Jolivet PetscBool flg; 158013044ca3SPierre Jolivet std::pair<PC, Vec[2]> *p; 158113044ca3SPierre Jolivet 158213044ca3SPierre Jolivet PetscFunctionBegin; 158313044ca3SPierre Jolivet PetscCall(PCShellGetContext(pc, &p)); 158451ea4bc8SPierre Jolivet if (p->second[0]) { /* in case of a centralized Schur complement, some processes may have no local operator */ 158513044ca3SPierre Jolivet PetscCall(PCGetOperators(p->first, &A, nullptr)); 158613044ca3SPierre Jolivet PetscCall(MatNestGetISs(A, is, nullptr)); 158751ea4bc8SPierre Jolivet PetscCall(PetscObjectTypeCompareAny((PetscObject)p->first, &flg, PCLU, PCCHOLESKY, "")); 158851ea4bc8SPierre Jolivet if (flg) { /* partial solve currently only makes sense with exact factorizations */ 158913044ca3SPierre Jolivet PetscCall(PCFactorGetMatSolverType(p->first, &type)); 159013044ca3SPierre Jolivet PetscCall(PCFactorGetMatrix(p->first, &A)); 159151ea4bc8SPierre Jolivet if (A->schur) { 159213044ca3SPierre Jolivet PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg)); 159351ea4bc8SPierre Jolivet if (flg) PetscCall(MatMumpsSetIcntl(A, 26, 1)); /* reduction/condensation phase followed by Schur complement solve */ 159451ea4bc8SPierre Jolivet } else flg = PETSC_FALSE; 159513044ca3SPierre Jolivet } 159613044ca3SPierre Jolivet PetscCall(VecISCopy(p->second[0], is[1], SCATTER_FORWARD, x)); /* assign the RHS associated to the Schur complement */ 159713044ca3SPierre Jolivet PetscCall(PCApply(p->first, p->second[0], p->second[1])); 159813044ca3SPierre Jolivet PetscCall(VecISCopy(p->second[1], is[1], SCATTER_REVERSE, y)); /* retrieve the partial solution associated to the Schur complement */ 159979578405SBarry Smith if (flg) PetscCall(MatMumpsSetIcntl(A, 26, -1)); /* default ICNTL(26) value in PETSc */ 160013044ca3SPierre Jolivet } 160113044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 160213044ca3SPierre Jolivet } 160313044ca3SPierre Jolivet 160413044ca3SPierre Jolivet static PetscErrorCode PCView_Nest(PC pc, PetscViewer viewer) 160513044ca3SPierre Jolivet { 160613044ca3SPierre Jolivet std::pair<PC, Vec[2]> *p; 160713044ca3SPierre Jolivet 160813044ca3SPierre Jolivet PetscFunctionBegin; 160913044ca3SPierre Jolivet PetscCall(PCShellGetContext(pc, &p)); 161013044ca3SPierre Jolivet PetscCall(PCView(p->first, viewer)); 161113044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 161213044ca3SPierre Jolivet } 161313044ca3SPierre Jolivet 161413044ca3SPierre Jolivet static PetscErrorCode PCDestroy_Nest(PC pc) 161513044ca3SPierre Jolivet { 161613044ca3SPierre Jolivet std::pair<PC, Vec[2]> *p; 161713044ca3SPierre Jolivet 161813044ca3SPierre Jolivet PetscFunctionBegin; 161913044ca3SPierre Jolivet PetscCall(PCShellGetContext(pc, &p)); 162013044ca3SPierre Jolivet PetscCall(VecDestroy(p->second)); 162113044ca3SPierre Jolivet PetscCall(VecDestroy(p->second + 1)); 162213044ca3SPierre Jolivet PetscCall(PCDestroy(&p->first)); 162313044ca3SPierre Jolivet PetscCall(PetscFree(p)); 162413044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 162513044ca3SPierre Jolivet } 162613044ca3SPierre Jolivet 162713044ca3SPierre Jolivet template <bool T = false> 162813044ca3SPierre Jolivet static PetscErrorCode MatMult_Schur(Mat A, Vec x, Vec y) 162913044ca3SPierre Jolivet { 163001e3c840SPierre Jolivet std::tuple<Mat, PetscSF, Vec[2]> *ctx; 163113044ca3SPierre Jolivet 163213044ca3SPierre Jolivet PetscFunctionBegin; 163313044ca3SPierre Jolivet PetscCall(MatShellGetContext(A, &ctx)); 163413044ca3SPierre Jolivet PetscCall(VecScatterBegin(std::get<1>(*ctx), x, std::get<2>(*ctx)[0], INSERT_VALUES, SCATTER_FORWARD)); /* local Vec with overlap */ 163513044ca3SPierre Jolivet PetscCall(VecScatterEnd(std::get<1>(*ctx), x, std::get<2>(*ctx)[0], INSERT_VALUES, SCATTER_FORWARD)); 163613044ca3SPierre Jolivet if (!T) PetscCall(MatMult(std::get<0>(*ctx), std::get<2>(*ctx)[0], std::get<2>(*ctx)[1])); /* local Schur complement */ 163713044ca3SPierre Jolivet else PetscCall(MatMultTranspose(std::get<0>(*ctx), std::get<2>(*ctx)[0], std::get<2>(*ctx)[1])); 163813044ca3SPierre Jolivet PetscCall(VecSet(y, 0.0)); 163913044ca3SPierre 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 */ 164013044ca3SPierre Jolivet PetscCall(VecScatterEnd(std::get<1>(*ctx), std::get<2>(*ctx)[1], y, ADD_VALUES, SCATTER_REVERSE)); 164113044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 164213044ca3SPierre Jolivet } 164313044ca3SPierre Jolivet 164413044ca3SPierre Jolivet static PetscErrorCode MatDestroy_Schur(Mat A) 164513044ca3SPierre Jolivet { 164601e3c840SPierre Jolivet std::tuple<Mat, PetscSF, Vec[2]> *ctx; 164713044ca3SPierre Jolivet 164813044ca3SPierre Jolivet PetscFunctionBegin; 164913044ca3SPierre Jolivet PetscCall(MatShellGetContext(A, &ctx)); 165013044ca3SPierre Jolivet PetscCall(VecDestroy(std::get<2>(*ctx))); 165113044ca3SPierre Jolivet PetscCall(VecDestroy(std::get<2>(*ctx) + 1)); 165213044ca3SPierre Jolivet PetscCall(PetscFree(ctx)); 165313044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 165413044ca3SPierre Jolivet } 165513044ca3SPierre Jolivet 165613044ca3SPierre Jolivet static PetscErrorCode MatMult_SchurCorrection(Mat A, Vec x, Vec y) 165713044ca3SPierre Jolivet { 165813044ca3SPierre Jolivet PC pc; 165913044ca3SPierre Jolivet std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx; 166013044ca3SPierre Jolivet 166113044ca3SPierre Jolivet PetscFunctionBegin; 166213044ca3SPierre Jolivet PetscCall(MatShellGetContext(A, &ctx)); 166313044ca3SPierre Jolivet pc = ((PC_HPDDM *)std::get<0>(*ctx)[0]->data)->levels[0]->ksp->pc; 166413044ca3SPierre 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 */ 166513044ca3SPierre Jolivet PetscCall(MatMult(std::get<1>(*ctx)[0], x, std::get<3>(*ctx)[1])); /* A_01 x */ 166613044ca3SPierre Jolivet PetscCall(PCHPDDMDeflate_Private(pc, std::get<3>(*ctx)[1], std::get<3>(*ctx)[1])); /* Q_0 A_01 x */ 166713044ca3SPierre 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 */ 166813044ca3SPierre 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 */ 166913044ca3SPierre Jolivet } else { 167013044ca3SPierre Jolivet PetscCall(PCApply(std::get<0>(*ctx)[1], x, std::get<3>(*ctx)[0])); /* M_S^-1 x */ 167113044ca3SPierre Jolivet PetscCall(MatMult(std::get<1>(*ctx)[0], std::get<3>(*ctx)[0], std::get<3>(*ctx)[1])); /* A_01 M_S^-1 x */ 167213044ca3SPierre Jolivet PetscCall(PCHPDDMDeflate_Private(pc, std::get<3>(*ctx)[1], std::get<3>(*ctx)[1])); /* Q_0 A_01 M_S^-1 x */ 167313044ca3SPierre 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 */ 167413044ca3SPierre Jolivet } 167513044ca3SPierre Jolivet PetscCall(VecAXPY(y, -1.0, x)); /* y -= x, preconditioned eq. (24) of https://hal.science/hal-02343808v6/document (with a sign flip) */ 167613044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 167713044ca3SPierre Jolivet } 167813044ca3SPierre Jolivet 167913044ca3SPierre Jolivet static PetscErrorCode MatView_SchurCorrection(Mat A, PetscViewer viewer) 168013044ca3SPierre Jolivet { 168113044ca3SPierre Jolivet PetscBool ascii; 168213044ca3SPierre Jolivet std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx; 168313044ca3SPierre Jolivet 168413044ca3SPierre Jolivet PetscFunctionBegin; 168513044ca3SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &ascii)); 168613044ca3SPierre Jolivet if (ascii) { 168713044ca3SPierre Jolivet PetscCall(MatShellGetContext(A, &ctx)); 168813044ca3SPierre 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)")); 168913044ca3SPierre Jolivet PetscCall(PCView(std::get<0>(*ctx)[1], viewer)); /* no need to PCView(Q_0) since it will be done by PCFIELDSPLIT */ 169013044ca3SPierre Jolivet } 169113044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 169213044ca3SPierre Jolivet } 169313044ca3SPierre Jolivet 169413044ca3SPierre Jolivet static PetscErrorCode MatDestroy_SchurCorrection(Mat A) 169513044ca3SPierre Jolivet { 169613044ca3SPierre Jolivet std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx; 169713044ca3SPierre Jolivet 169813044ca3SPierre Jolivet PetscFunctionBegin; 169913044ca3SPierre Jolivet PetscCall(MatShellGetContext(A, &ctx)); 170013044ca3SPierre Jolivet PetscCall(VecDestroy(std::get<3>(*ctx))); 170113044ca3SPierre Jolivet PetscCall(VecDestroy(std::get<3>(*ctx) + 1)); 170213044ca3SPierre Jolivet PetscCall(VecDestroy(std::get<3>(*ctx) + 2)); 170313044ca3SPierre Jolivet PetscCall(PCDestroy(std::get<0>(*ctx) + 1)); 170413044ca3SPierre Jolivet PetscCall(PetscFree(ctx)); 170513044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 170613044ca3SPierre Jolivet } 170713044ca3SPierre Jolivet 1708feebddf4SPierre Jolivet static PetscErrorCode PCPostSolve_SchurPreLeastSquares(PC, KSP, Vec, Vec x) 1709feebddf4SPierre Jolivet { 1710feebddf4SPierre Jolivet PetscFunctionBegin; 1711feebddf4SPierre Jolivet PetscCall(VecScale(x, -1.0)); 1712feebddf4SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 1713feebddf4SPierre Jolivet } 1714feebddf4SPierre Jolivet 171513044ca3SPierre Jolivet static PetscErrorCode KSPPreSolve_SchurCorrection(KSP, Vec b, Vec, void *context) 171613044ca3SPierre Jolivet { 171713044ca3SPierre Jolivet std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx = reinterpret_cast<std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *>(context); 171813044ca3SPierre Jolivet 1719a6b3e571SPierre Jolivet PetscFunctionBegin; 172013044ca3SPierre Jolivet if (std::get<2>(*ctx) == PC_LEFT || std::get<2>(*ctx) == PC_SIDE_DEFAULT) { 172113044ca3SPierre Jolivet PetscCall(PCApply(std::get<0>(*ctx)[1], b, std::get<3>(*ctx)[2])); 172213044ca3SPierre 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 */ 172313044ca3SPierre Jolivet } 172413044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 172513044ca3SPierre Jolivet } 172613044ca3SPierre Jolivet 172713044ca3SPierre Jolivet static PetscErrorCode KSPPostSolve_SchurCorrection(KSP, Vec b, Vec x, void *context) 172813044ca3SPierre Jolivet { 172913044ca3SPierre Jolivet std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx = reinterpret_cast<std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *>(context); 173013044ca3SPierre Jolivet 1731a6b3e571SPierre Jolivet PetscFunctionBegin; 173213044ca3SPierre 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 */ 173313044ca3SPierre Jolivet else { 173413044ca3SPierre Jolivet PetscCall(PCApply(std::get<0>(*ctx)[1], x, std::get<3>(*ctx)[2])); 173513044ca3SPierre Jolivet PetscCall(VecCopy(std::get<3>(*ctx)[2], x)); /* replace x by M^-1 x */ 173613044ca3SPierre Jolivet } 173713044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 173813044ca3SPierre Jolivet } 173913044ca3SPierre Jolivet 17409bb5c669SPierre Jolivet static PetscErrorCode MatMult_Harmonic(Mat, Vec, Vec); 17419bb5c669SPierre Jolivet static PetscErrorCode MatMultTranspose_Harmonic(Mat, Vec, Vec); 17429bb5c669SPierre Jolivet static PetscErrorCode MatProduct_AB_Harmonic(Mat, Mat, Mat, void *); 1743f21e3f8aSPierre Jolivet static PetscErrorCode MatProduct_AtB_Harmonic(Mat, Mat, Mat, void *); 17449bb5c669SPierre Jolivet static PetscErrorCode MatDestroy_Harmonic(Mat); 17459bb5c669SPierre Jolivet 1746d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCSetUp_HPDDM(PC pc) 1747d71ae5a4SJacob Faibussowitsch { 1748f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 1749f1580f4eSBarry Smith PC inner; 1750f1580f4eSBarry Smith KSP *ksp; 175113044ca3SPierre Jolivet Mat *sub, A, P, N, C = nullptr, uaux = nullptr, weighted, subA[2], S; 1752f1580f4eSBarry Smith Vec xin, v; 1753f1580f4eSBarry Smith std::vector<Vec> initial; 1754db4a47b3SPierre Jolivet IS is[1], loc, uis = data->is, unsorted = nullptr; 1755f1580f4eSBarry Smith ISLocalToGlobalMapping l2g; 1756f1580f4eSBarry Smith char prefix[256]; 1757f1580f4eSBarry Smith const char *pcpre; 1758f1580f4eSBarry Smith const PetscScalar *const *ev; 17599bb5c669SPierre Jolivet PetscInt n, requested = data->N, reused = 0, overlap = -1; 1760f1580f4eSBarry Smith MatStructure structure = UNKNOWN_NONZERO_PATTERN; 1761f1580f4eSBarry Smith PetscBool subdomains = PETSC_FALSE, flg = PETSC_FALSE, ismatis, swap = PETSC_FALSE, algebraic = PETSC_FALSE, block = PETSC_FALSE; 1762f1580f4eSBarry Smith DM dm; 176313044ca3SPierre Jolivet std::tuple<PC[2], Mat[2], PCSide, Vec[3]> *ctx = nullptr; 1764398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG) 1765db4a47b3SPierre Jolivet IS dis = nullptr; 1766db4a47b3SPierre Jolivet Mat daux = nullptr; 1767398c7888SPierre Jolivet #endif 1768f1580f4eSBarry Smith 1769f1580f4eSBarry Smith PetscFunctionBegin; 1770f1580f4eSBarry Smith PetscCheck(data->levels && data->levels[0], PETSC_COMM_SELF, PETSC_ERR_PLIB, "Not a single level allocated"); 1771f1580f4eSBarry Smith PetscCall(PCGetOptionsPrefix(pc, &pcpre)); 1772f1580f4eSBarry Smith PetscCall(PCGetOperators(pc, &A, &P)); 1773f1580f4eSBarry Smith if (!data->levels[0]->ksp) { 1774f1580f4eSBarry Smith PetscCall(KSPCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->ksp)); 17753821be0aSBarry Smith PetscCall(KSPSetNestLevel(data->levels[0]->ksp, pc->kspnestlevel)); 1776f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_%s_", pcpre ? pcpre : "", data->N > 1 ? "levels_1" : "coarse")); 1777f1580f4eSBarry Smith PetscCall(KSPSetOptionsPrefix(data->levels[0]->ksp, prefix)); 1778f1580f4eSBarry Smith PetscCall(KSPSetType(data->levels[0]->ksp, KSPPREONLY)); 1779371d2eb7SMartin Diehl } else if (data->levels[0]->ksp->pc && data->levels[0]->ksp->pc->setupcalled && data->levels[0]->ksp->pc->reusepreconditioner) { 1780f1580f4eSBarry Smith /* if the fine-level PCSHELL exists, its setup has succeeded, and one wants to reuse it, */ 1781f1580f4eSBarry Smith /* then just propagate the appropriate flag to the coarser levels */ 1782f1580f4eSBarry Smith for (n = 0; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) { 1783f1580f4eSBarry Smith /* the following KSP and PC may be NULL for some processes, hence the check */ 1784f1580f4eSBarry Smith if (data->levels[n]->ksp) PetscCall(KSPSetReusePreconditioner(data->levels[n]->ksp, PETSC_TRUE)); 1785f1580f4eSBarry Smith if (data->levels[n]->pc) PetscCall(PCSetReusePreconditioner(data->levels[n]->pc, PETSC_TRUE)); 1786f1580f4eSBarry Smith } 1787f1580f4eSBarry Smith /* early bail out because there is nothing to do */ 17883ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1789f1580f4eSBarry Smith } else { 1790f1580f4eSBarry Smith /* reset coarser levels */ 1791f1580f4eSBarry Smith for (n = 1; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) { 1792371d2eb7SMartin Diehl if (data->levels[n]->ksp && data->levels[n]->ksp->pc && data->levels[n]->ksp->pc->setupcalled && data->levels[n]->ksp->pc->reusepreconditioner && n < data->N) { 1793f1580f4eSBarry Smith reused = data->N - n; 1794f1580f4eSBarry Smith break; 1795f1580f4eSBarry Smith } 1796f1580f4eSBarry Smith PetscCall(KSPDestroy(&data->levels[n]->ksp)); 1797f1580f4eSBarry Smith PetscCall(PCDestroy(&data->levels[n]->pc)); 1798f1580f4eSBarry Smith } 1799f1580f4eSBarry Smith /* check if some coarser levels are being reused */ 1800462c564dSBarry Smith PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &reused, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)pc))); 1801f1580f4eSBarry Smith const int *addr = data->levels[0]->P ? data->levels[0]->P->getAddrLocal() : &HPDDM::i__0; 1802f1580f4eSBarry Smith 1803f1580f4eSBarry Smith if (addr != &HPDDM::i__0 && reused != data->N - 1) { 1804f1580f4eSBarry Smith /* reuse previously computed eigenvectors */ 1805f1580f4eSBarry Smith ev = data->levels[0]->P->getVectors(); 1806f1580f4eSBarry Smith if (ev) { 1807f1580f4eSBarry Smith initial.reserve(*addr); 1808f1580f4eSBarry Smith PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, data->levels[0]->P->getDof(), ev[0], &xin)); 1809f1580f4eSBarry Smith for (n = 0; n < *addr; ++n) { 1810f1580f4eSBarry Smith PetscCall(VecDuplicate(xin, &v)); 1811f1580f4eSBarry Smith PetscCall(VecPlaceArray(xin, ev[n])); 1812f1580f4eSBarry Smith PetscCall(VecCopy(xin, v)); 1813f1580f4eSBarry Smith initial.emplace_back(v); 1814f1580f4eSBarry Smith PetscCall(VecResetArray(xin)); 1815f1580f4eSBarry Smith } 1816f1580f4eSBarry Smith PetscCall(VecDestroy(&xin)); 1817f1580f4eSBarry Smith } 1818f1580f4eSBarry Smith } 1819f1580f4eSBarry Smith } 1820f1580f4eSBarry Smith data->N -= reused; 1821f1580f4eSBarry Smith PetscCall(KSPSetOperators(data->levels[0]->ksp, A, P)); 1822f1580f4eSBarry Smith 1823f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)P, MATIS, &ismatis)); 1824f1580f4eSBarry Smith if (!data->is && !ismatis) { 1825db4a47b3SPierre Jolivet PetscErrorCode (*create)(DM, IS *, Mat *, PetscErrorCode (**)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void **) = nullptr; 1826db4a47b3SPierre Jolivet PetscErrorCode (*usetup)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *) = nullptr; 1827db4a47b3SPierre Jolivet void *uctx = nullptr; 1828f1580f4eSBarry Smith 1829f1580f4eSBarry Smith /* first see if we can get the data from the DM */ 1830f1580f4eSBarry Smith PetscCall(MatGetDM(P, &dm)); 1831f1580f4eSBarry Smith if (!dm) PetscCall(MatGetDM(A, &dm)); 1832f1580f4eSBarry Smith if (!dm) PetscCall(PCGetDM(pc, &dm)); 1833907a3e9cSStefano Zampini if (dm) { /* this is the hook for DMPLEX for which the auxiliary Mat is the local Neumann matrix */ 1834f1580f4eSBarry Smith PetscCall(PetscObjectQueryFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", &create)); 1835f1580f4eSBarry Smith if (create) { 1836f1580f4eSBarry Smith PetscCall((*create)(dm, &uis, &uaux, &usetup, &uctx)); 1837c8ea6600SPierre 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 */ 1838f1580f4eSBarry Smith } 1839f1580f4eSBarry Smith } 1840f1580f4eSBarry Smith if (!create) { 1841f1580f4eSBarry Smith if (!uis) { 1842f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Neumann_IS", (PetscObject *)&uis)); 1843f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)uis)); 1844f1580f4eSBarry Smith } 1845f1580f4eSBarry Smith if (!uaux) { 1846f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Neumann_Mat", (PetscObject *)&uaux)); 1847f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)uaux)); 1848f1580f4eSBarry Smith } 1849f1580f4eSBarry Smith /* look inside the Pmat instead of the PC, needed for MatSchurComplementComputeExplicitOperator() */ 1850f1580f4eSBarry Smith if (!uis) { 1851f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)P, "_PCHPDDM_Neumann_IS", (PetscObject *)&uis)); 1852f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)uis)); 1853f1580f4eSBarry Smith } 1854f1580f4eSBarry Smith if (!uaux) { 1855f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)P, "_PCHPDDM_Neumann_Mat", (PetscObject *)&uaux)); 1856f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)uaux)); 1857f1580f4eSBarry Smith } 1858f1580f4eSBarry Smith } 1859f1580f4eSBarry Smith PetscCall(PCHPDDMSetAuxiliaryMat(pc, uis, uaux, usetup, uctx)); 1860f1580f4eSBarry Smith PetscCall(MatDestroy(&uaux)); 1861f1580f4eSBarry Smith PetscCall(ISDestroy(&uis)); 1862f1580f4eSBarry Smith } 1863f1580f4eSBarry Smith 1864f1580f4eSBarry Smith if (!ismatis) { 1865f1580f4eSBarry Smith PetscCall(PCHPDDMSetUpNeumannOverlap_Private(pc)); 1866db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetBool(nullptr, pcpre, "-pc_hpddm_block_splitting", &block, nullptr)); 18679bb5c669SPierre Jolivet PetscCall(PetscOptionsGetInt(nullptr, pcpre, "-pc_hpddm_harmonic_overlap", &overlap, nullptr)); 18685e642048SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg)); 186951ea4bc8SPierre Jolivet if (data->is || flg) { 18709bb5c669SPierre Jolivet if (block || overlap != -1) { 187102800ff6SPierre Jolivet PetscCall(ISDestroy(&data->is)); 187202800ff6SPierre Jolivet PetscCall(MatDestroy(&data->aux)); 187351ea4bc8SPierre Jolivet } else if (flg) { 187413044ca3SPierre Jolivet PCHPDDMSchurPreType type = PC_HPDDM_SCHUR_PRE_GENEO; 187513044ca3SPierre Jolivet 187613044ca3SPierre Jolivet PetscCall(PetscOptionsGetEnum(nullptr, pcpre, "-pc_hpddm_schur_precondition", PCHPDDMSchurPreTypes, (PetscEnum *)&type, &flg)); 187713044ca3SPierre Jolivet if (type == PC_HPDDM_SCHUR_PRE_LEAST_SQUARES) { 187813044ca3SPierre Jolivet PetscCall(ISDestroy(&data->is)); /* destroy any previously user-set objects since they will be set automatically */ 187913044ca3SPierre Jolivet PetscCall(MatDestroy(&data->aux)); 188013044ca3SPierre Jolivet } else if (type == PC_HPDDM_SCHUR_PRE_GENEO) { 188113044ca3SPierre Jolivet PetscContainer container = nullptr; 188213044ca3SPierre Jolivet 188313044ca3SPierre Jolivet PetscCall(PetscObjectQuery((PetscObject)pc, "_PCHPDDM_Schur", (PetscObject *)&container)); 188413044ca3SPierre Jolivet if (!container) { /* first call to PCSetUp() on the PC associated to the Schur complement */ 188513044ca3SPierre Jolivet PC_HPDDM *data_00; 188613044ca3SPierre Jolivet KSP ksp, inner_ksp; 188713044ca3SPierre Jolivet PC pc_00; 1888cf67ef9dSPierre Jolivet Mat A11 = nullptr; 1889cf67ef9dSPierre Jolivet Vec d = nullptr; 1890*8a8e6071SPierre Jolivet PetscReal norm; 189151ea4bc8SPierre Jolivet const PetscInt *ranges; 189251ea4bc8SPierre Jolivet PetscMPIInt size; 189313044ca3SPierre Jolivet char *prefix; 189413044ca3SPierre Jolivet 189513044ca3SPierre Jolivet PetscCall(MatSchurComplementGetKSP(P, &ksp)); 189613044ca3SPierre Jolivet PetscCall(KSPGetPC(ksp, &pc_00)); 189713044ca3SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)pc_00, PCHPDDM, &flg)); 1898fd310a01SPierre 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 : "", 189913044ca3SPierre Jolivet ((PetscObject)pc_00)->type_name, PCHPDDM); 190013044ca3SPierre Jolivet data_00 = (PC_HPDDM *)pc_00->data; 1901fd310a01SPierre 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], 1902fd310a01SPierre Jolivet data_00->N, data_00->N > 1 ? "s" : "", ((PetscObject)pc_00)->prefix); 190351ea4bc8SPierre Jolivet PetscCheck(data_00->levels[0]->pc, PetscObjectComm((PetscObject)P), PETSC_ERR_ORDER, "PC of the first block%s not setup yet", ((PetscObject)pc_00)->prefix ? std::string(std::string(" (") + ((PetscObject)pc_00)->prefix + std::string(")")).c_str() : ""); 190413044ca3SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)data_00->levels[0]->pc, PCASM, &flg)); 1905fd310a01SPierre 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, 190613044ca3SPierre Jolivet ((PetscObject)data_00->levels[0]->pc)->type_name, PCASM); 190751ea4bc8SPierre Jolivet PetscCall(PetscNew(&ctx)); /* context to pass data around for the inner-most PC, which will be a proper PCHPDDM (or a dummy variable if the Schur complement is centralized on a single process) */ 1908d16c0b94SPierre Jolivet PetscCall(MatSchurComplementGetSubMatrices(P, nullptr, nullptr, nullptr, nullptr, &A11)); 190951ea4bc8SPierre Jolivet PetscCall(MatGetOwnershipRanges(A11, &ranges)); 191051ea4bc8SPierre Jolivet PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A11), &size)); 191151ea4bc8SPierre Jolivet flg = PetscBool(std::find_if(ranges, ranges + size + 1, [&](PetscInt v) { return v != ranges[0] && v != ranges[size]; }) == ranges + size + 1); /* are all local matrices but one of dimension 0 (centralized Schur complement)? */ 191251ea4bc8SPierre Jolivet if (!flg) { 19135e642048SPierre Jolivet if (PetscDefined(USE_DEBUG) || !data->is) { 19145e642048SPierre Jolivet Mat A01, A10, B = nullptr, C = nullptr, *sub; 191513044ca3SPierre Jolivet 19165e642048SPierre Jolivet PetscCall(MatSchurComplementGetSubMatrices(P, &A, nullptr, &A01, &A10, nullptr)); 19175e642048SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, &flg)); 19185e642048SPierre Jolivet if (flg) { 19195e642048SPierre Jolivet PetscCall(MatTransposeGetMat(A10, &C)); 19205e642048SPierre Jolivet PetscCall(MatTranspose(C, MAT_INITIAL_MATRIX, &B)); 19215e642048SPierre Jolivet } else { 19225e642048SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, &flg)); 19235e642048SPierre Jolivet if (flg) { 19245e642048SPierre Jolivet PetscCall(MatHermitianTransposeGetMat(A10, &C)); 19255e642048SPierre Jolivet PetscCall(MatHermitianTranspose(C, MAT_INITIAL_MATRIX, &B)); 19265e642048SPierre Jolivet } 19275e642048SPierre Jolivet } 1928811e8887SPierre Jolivet if (flg) 1929811e8887SPierre 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)); 19305e642048SPierre Jolivet if (!B) { 19315e642048SPierre Jolivet B = A10; 19325e642048SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)B)); 19335e642048SPierre Jolivet } else if (!data->is) { 19345e642048SPierre Jolivet PetscCall(PetscObjectTypeCompareAny((PetscObject)A01, &flg, MATTRANSPOSEVIRTUAL, MATHERMITIANTRANSPOSEVIRTUAL, "")); 19355e642048SPierre Jolivet if (!flg) C = A01; 1936811e8887SPierre Jolivet else 1937811e8887SPierre 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)); 19385e642048SPierre Jolivet } 193913044ca3SPierre Jolivet PetscCall(ISCreateStride(PETSC_COMM_SELF, B->rmap->N, 0, 1, &uis)); 19405e642048SPierre Jolivet PetscCall(ISSetIdentity(uis)); 19415e642048SPierre Jolivet if (!data->is) { 19425e642048SPierre Jolivet if (C) PetscCall(PetscObjectReference((PetscObject)C)); 19435e642048SPierre Jolivet else PetscCall(MatTranspose(B, MAT_INITIAL_MATRIX, &C)); 19445e642048SPierre Jolivet PetscCall(ISDuplicate(data_00->is, is)); 19455e642048SPierre Jolivet PetscCall(MatIncreaseOverlap(A, 1, is, 1)); 19465e642048SPierre Jolivet PetscCall(MatSetOption(C, MAT_SUBMAT_SINGLEIS, PETSC_TRUE)); 19475e642048SPierre Jolivet PetscCall(MatCreateSubMatrices(C, 1, is, &uis, MAT_INITIAL_MATRIX, &sub)); 19485e642048SPierre Jolivet PetscCall(MatDestroy(&C)); 19495e642048SPierre Jolivet PetscCall(MatTranspose(sub[0], MAT_INITIAL_MATRIX, &C)); 19505e642048SPierre Jolivet PetscCall(MatDestroySubMatrices(1, &sub)); 19515e642048SPierre Jolivet PetscCall(MatFindNonzeroRows(C, &data->is)); 19525e642048SPierre Jolivet PetscCall(MatDestroy(&C)); 19535e642048SPierre Jolivet PetscCall(ISDestroy(is)); 1954d16c0b94SPierre Jolivet PetscCall(ISCreateStride(PetscObjectComm((PetscObject)data->is), A11->rmap->n, A11->rmap->rstart, 1, &loc)); 1955d16c0b94SPierre Jolivet if (PetscDefined(USE_DEBUG)) PetscCall(PCHPDDMCheckInclusion_Private(pc, data->is, loc, PETSC_FALSE)); 1956d16c0b94SPierre Jolivet PetscCall(ISExpand(data->is, loc, is)); 1957d16c0b94SPierre Jolivet PetscCall(ISDestroy(&loc)); 1958d16c0b94SPierre Jolivet PetscCall(ISDestroy(&data->is)); 1959d16c0b94SPierre Jolivet data->is = is[0]; 1960d16c0b94SPierre Jolivet is[0] = nullptr; 19615e642048SPierre Jolivet } 19625e642048SPierre Jolivet if (PetscDefined(USE_DEBUG)) { 19635e642048SPierre Jolivet PetscCall(PCHPDDMCheckSymmetry_Private(pc, A01, A10)); 196413044ca3SPierre 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 */ 196513044ca3SPierre Jolivet PetscCall(ISDestroy(&uis)); 196613044ca3SPierre Jolivet PetscCall(ISDuplicate(data->is, &uis)); 196713044ca3SPierre Jolivet PetscCall(ISSort(uis)); 19685e642048SPierre Jolivet PetscCall(ISComplement(uis, 0, B->rmap->N, is)); 196913044ca3SPierre Jolivet PetscCall(MatDuplicate(sub[0], MAT_COPY_VALUES, &C)); 19705e642048SPierre Jolivet PetscCall(MatZeroRowsIS(C, is[0], 0.0, nullptr, nullptr)); 19715e642048SPierre Jolivet PetscCall(ISDestroy(is)); 197213044ca3SPierre Jolivet PetscCall(MatMultEqual(sub[0], C, 20, &flg)); 197313044ca3SPierre 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 */ 197413044ca3SPierre Jolivet PetscCall(MatDestroy(&C)); 197513044ca3SPierre Jolivet PetscCall(MatDestroySubMatrices(1, &sub)); 197613044ca3SPierre Jolivet } 19775e642048SPierre Jolivet PetscCall(ISDestroy(&uis)); 19785e642048SPierre Jolivet PetscCall(MatDestroy(&B)); 19795e642048SPierre Jolivet } 1980cf67ef9dSPierre Jolivet flg = PETSC_FALSE; 1981cf67ef9dSPierre Jolivet if (!data->aux) { 1982cf67ef9dSPierre Jolivet Mat D; 1983cf67ef9dSPierre Jolivet 1984cf67ef9dSPierre Jolivet PetscCall(MatCreateVecs(A11, &d, nullptr)); 1985cf67ef9dSPierre Jolivet PetscCall(MatGetDiagonal(A11, d)); 1986cf67ef9dSPierre Jolivet PetscCall(PetscObjectTypeCompareAny((PetscObject)A11, &flg, MATDIAGONAL, MATCONSTANTDIAGONAL, "")); 1987cf67ef9dSPierre Jolivet if (!flg) { 1988cf67ef9dSPierre Jolivet PetscCall(MatCreateDiagonal(d, &D)); 1989cf67ef9dSPierre Jolivet PetscCall(MatMultEqual(A11, D, 20, &flg)); 1990cf67ef9dSPierre Jolivet PetscCall(MatDestroy(&D)); 1991cf67ef9dSPierre Jolivet } 1992cf67ef9dSPierre 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")); 1993cf67ef9dSPierre Jolivet } 1994*8a8e6071SPierre Jolivet if ((PetscDefined(USE_DEBUG) || (data->Neumann != PETSC_BOOL3_TRUE && !flg)) && A11) { 1995cf67ef9dSPierre Jolivet PetscCall(MatNorm(A11, NORM_INFINITY, &norm)); 1996*8a8e6071SPierre Jolivet if (data->Neumann != PETSC_BOOL3_TRUE && !flg) { 1997cf67ef9dSPierre 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 : ""); 1998cf67ef9dSPierre 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")); 1999cf67ef9dSPierre Jolivet PetscCall(MatDestroy(&data->aux)); 2000cf67ef9dSPierre Jolivet flg = PETSC_TRUE; 2001cf67ef9dSPierre Jolivet } 2002*8a8e6071SPierre Jolivet } 2003cf67ef9dSPierre 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 */ 200401e3c840SPierre Jolivet PetscSF scatter; 200513044ca3SPierre Jolivet const PetscScalar *read; 2006cf67ef9dSPierre Jolivet PetscScalar *write, *diagonal = nullptr; 200713044ca3SPierre Jolivet 200813044ca3SPierre Jolivet PetscCall(MatDestroy(&data->aux)); 200913044ca3SPierre Jolivet PetscCall(ISGetLocalSize(data->is, &n)); 2010cf67ef9dSPierre Jolivet PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)P), n, PETSC_DECIDE, &xin)); 2011cf67ef9dSPierre Jolivet PetscCall(VecDuplicate(xin, &v)); 2012cf67ef9dSPierre Jolivet PetscCall(VecScatterCreate(xin, data->is, v, nullptr, &scatter)); 201313044ca3SPierre Jolivet PetscCall(VecSet(v, 1.0)); 2014cf67ef9dSPierre Jolivet PetscCall(VecSet(xin, 1.0)); 2015cf67ef9dSPierre Jolivet PetscCall(VecScatterBegin(scatter, v, xin, ADD_VALUES, SCATTER_REVERSE)); 2016cf67ef9dSPierre Jolivet PetscCall(VecScatterEnd(scatter, v, xin, ADD_VALUES, SCATTER_REVERSE)); /* v has the multiplicity of all unknowns on the overlap */ 2017cf67ef9dSPierre Jolivet PetscCall(PetscSFDestroy(&scatter)); 2018cf67ef9dSPierre Jolivet if (d) { 2019cf67ef9dSPierre Jolivet PetscCall(VecScatterCreate(d, data->is, v, nullptr, &scatter)); 2020cf67ef9dSPierre Jolivet PetscCall(VecScatterBegin(scatter, d, v, INSERT_VALUES, SCATTER_FORWARD)); 2021cf67ef9dSPierre Jolivet PetscCall(VecScatterEnd(scatter, d, v, INSERT_VALUES, SCATTER_FORWARD)); 2022cf67ef9dSPierre Jolivet PetscCall(PetscSFDestroy(&scatter)); 2023cf67ef9dSPierre Jolivet PetscCall(VecDestroy(&d)); 2024cf67ef9dSPierre Jolivet PetscCall(PetscMalloc1(n, &diagonal)); 2025cf67ef9dSPierre Jolivet PetscCall(VecGetArrayRead(v, &read)); 2026cf67ef9dSPierre Jolivet PetscCallCXX(std::copy_n(read, n, diagonal)); 2027cf67ef9dSPierre Jolivet PetscCall(VecRestoreArrayRead(v, &read)); 2028cf67ef9dSPierre Jolivet } 202913044ca3SPierre Jolivet PetscCall(VecDestroy(&v)); 203013044ca3SPierre Jolivet PetscCall(VecCreateSeq(PETSC_COMM_SELF, n, &v)); 2031cf67ef9dSPierre Jolivet PetscCall(VecGetArrayRead(xin, &read)); 203213044ca3SPierre Jolivet PetscCall(VecGetArrayWrite(v, &write)); 2033cf67ef9dSPierre 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]; 2034cf67ef9dSPierre Jolivet PetscCall(PetscFree(diagonal)); 2035cf67ef9dSPierre Jolivet PetscCall(VecRestoreArrayRead(xin, &read)); 203613044ca3SPierre Jolivet PetscCall(VecRestoreArrayWrite(v, &write)); 2037cf67ef9dSPierre Jolivet PetscCall(VecDestroy(&xin)); 2038c3e1b152SPierre Jolivet PetscCall(MatCreateDiagonal(v, &data->aux)); 203913044ca3SPierre Jolivet PetscCall(VecDestroy(&v)); 204013044ca3SPierre Jolivet } 204113044ca3SPierre Jolivet uis = data->is; 204213044ca3SPierre Jolivet uaux = data->aux; 204313044ca3SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)uis)); 204413044ca3SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)uaux)); 204513044ca3SPierre Jolivet PetscCall(PetscStrallocpy(pcpre, &prefix)); 204613044ca3SPierre Jolivet PetscCall(PCSetOptionsPrefix(pc, nullptr)); 204713044ca3SPierre Jolivet PetscCall(PCSetType(pc, PCKSP)); /* replace the PC associated to the Schur complement by PCKSP */ 204813044ca3SPierre Jolivet PetscCall(KSPCreate(PetscObjectComm((PetscObject)pc), &inner_ksp)); /* new KSP that will be attached to the previously set PC */ 204913044ca3SPierre Jolivet PetscCall(PetscObjectGetTabLevel((PetscObject)pc, &n)); 205013044ca3SPierre Jolivet PetscCall(PetscObjectSetTabLevel((PetscObject)inner_ksp, n + 2)); 205113044ca3SPierre Jolivet PetscCall(KSPSetOperators(inner_ksp, pc->mat, pc->pmat)); 205213044ca3SPierre Jolivet PetscCall(KSPSetOptionsPrefix(inner_ksp, std::string(std::string(prefix) + "pc_hpddm_").c_str())); 205313044ca3SPierre Jolivet PetscCall(KSPSetSkipPCSetFromOptions(inner_ksp, PETSC_TRUE)); 205413044ca3SPierre Jolivet PetscCall(KSPSetFromOptions(inner_ksp)); 205513044ca3SPierre Jolivet PetscCall(KSPGetPC(inner_ksp, &inner)); 205613044ca3SPierre Jolivet PetscCall(PCSetOptionsPrefix(inner, nullptr)); 205713044ca3SPierre Jolivet PetscCall(PCSetType(inner, PCNONE)); /* no preconditioner since the action of M^-1 A or A M^-1 will be computed by the Amat */ 205813044ca3SPierre Jolivet PetscCall(PCKSPSetKSP(pc, inner_ksp)); 205913044ca3SPierre Jolivet std::get<0>(*ctx)[0] = pc_00; /* for coarse correction on the primal (e.g., velocity) space */ 206013044ca3SPierre Jolivet PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &std::get<0>(*ctx)[1])); 206151ea4bc8SPierre Jolivet PetscCall(PCSetOptionsPrefix(pc, prefix)); /* both PC share the same prefix so that the outer PC can be reset with PCSetFromOptions() */ 206213044ca3SPierre Jolivet PetscCall(PCSetOptionsPrefix(std::get<0>(*ctx)[1], prefix)); 206313044ca3SPierre Jolivet PetscCall(PetscFree(prefix)); 206413044ca3SPierre Jolivet PetscCall(PCSetOperators(std::get<0>(*ctx)[1], pc->mat, pc->pmat)); 206513044ca3SPierre Jolivet PetscCall(PCSetType(std::get<0>(*ctx)[1], PCHPDDM)); 206613044ca3SPierre 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 */ 2067cf67ef9dSPierre Jolivet if (flg) static_cast<PC_HPDDM *>(std::get<0>(*ctx)[1]->data)->Neumann = PETSC_BOOL3_TRUE; 2068*8a8e6071SPierre Jolivet else if (PetscDefined(USE_DEBUG) && norm > PETSC_MACHINE_EPSILON * static_cast<PetscReal>(10.0)) { 2069*8a8e6071SPierre Jolivet /* no check when A11 is near zero */ 2070*8a8e6071SPierre Jolivet PetscCall(MatCreateSubMatrices(A11, 1, &uis, &uis, MAT_INITIAL_MATRIX, &sub)); 2071*8a8e6071SPierre Jolivet PetscCall(PCHPDDMCheckMatStructure_Private(pc, sub[0], uaux)); 2072*8a8e6071SPierre Jolivet PetscCall(MatDestroySubMatrices(1, &sub)); 2073*8a8e6071SPierre Jolivet } 207413044ca3SPierre Jolivet PetscCall(PCSetFromOptions(std::get<0>(*ctx)[1])); 207513044ca3SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)uis)); 207613044ca3SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)uaux)); 207713044ca3SPierre 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 */ 207857d50842SBarry Smith PetscCall(MatShellSetOperation(S, MATOP_MULT, (PetscErrorCodeFn *)MatMult_SchurCorrection)); 207957d50842SBarry Smith PetscCall(MatShellSetOperation(S, MATOP_VIEW, (PetscErrorCodeFn *)MatView_SchurCorrection)); 208057d50842SBarry Smith PetscCall(MatShellSetOperation(S, MATOP_DESTROY, (PetscErrorCodeFn *)MatDestroy_SchurCorrection)); 208113044ca3SPierre Jolivet PetscCall(KSPGetPCSide(inner_ksp, &(std::get<2>(*ctx)))); 208213044ca3SPierre Jolivet if (std::get<2>(*ctx) == PC_LEFT || std::get<2>(*ctx) == PC_SIDE_DEFAULT) { 208313044ca3SPierre Jolivet PetscCall(KSPSetPreSolve(inner_ksp, KSPPreSolve_SchurCorrection, ctx)); 208413044ca3SPierre Jolivet } else { /* no support for PC_SYMMETRIC */ 208513044ca3SPierre 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]); 208613044ca3SPierre Jolivet } 208713044ca3SPierre Jolivet PetscCall(KSPSetPostSolve(inner_ksp, KSPPostSolve_SchurCorrection, ctx)); 2088715c1178SPierre Jolivet PetscCall(PetscObjectContainerCompose((PetscObject)std::get<0>(*ctx)[1], "_PCHPDDM_Schur", ctx, nullptr)); 208913044ca3SPierre Jolivet PetscCall(PCSetUp(std::get<0>(*ctx)[1])); 209013044ca3SPierre Jolivet PetscCall(KSPSetOperators(inner_ksp, S, S)); 209113044ca3SPierre Jolivet PetscCall(MatCreateVecs(std::get<1>(*ctx)[0], std::get<3>(*ctx), std::get<3>(*ctx) + 1)); 209213044ca3SPierre Jolivet PetscCall(VecDuplicate(std::get<3>(*ctx)[0], std::get<3>(*ctx) + 2)); 209313044ca3SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)inner_ksp)); 209413044ca3SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)S)); 209551ea4bc8SPierre Jolivet } else { 209651ea4bc8SPierre Jolivet std::get<0>(*ctx)[0] = pc_00; 209751ea4bc8SPierre Jolivet PetscCall(PetscObjectContainerCompose((PetscObject)pc, "_PCHPDDM_Schur", ctx, nullptr)); 209851ea4bc8SPierre Jolivet PetscCall(ISCreateStride(PetscObjectComm((PetscObject)data_00->is), A11->rmap->n, A11->rmap->rstart, 1, &data->is)); /* dummy variables in the case of a centralized Schur complement */ 209951ea4bc8SPierre Jolivet PetscCall(MatGetDiagonalBlock(A11, &data->aux)); 210051ea4bc8SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)data->aux)); 210151ea4bc8SPierre Jolivet PetscCall(PCSetUp(pc)); 210251ea4bc8SPierre Jolivet } 21030307214fSPierre Jolivet for (std::vector<Vec>::iterator it = initial.begin(); it != initial.end(); ++it) PetscCall(VecDestroy(&*it)); 210413044ca3SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 210513044ca3SPierre Jolivet } else { /* second call to PCSetUp() on the PC associated to the Schur complement, retrieve previously set context */ 210613044ca3SPierre Jolivet PetscCall(PetscContainerGetPointer(container, (void **)&ctx)); 210713044ca3SPierre Jolivet } 210813044ca3SPierre Jolivet } 210913044ca3SPierre Jolivet } 211013044ca3SPierre Jolivet } 2111f1580f4eSBarry Smith if (!data->is && data->N > 1) { 2112f1580f4eSBarry Smith char type[256] = {}; /* same size as in src/ksp/pc/interface/pcset.c */ 2113811e8887SPierre Jolivet 2114f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompareAny((PetscObject)P, &flg, MATNORMAL, MATNORMALHERMITIAN, "")); 2115f1580f4eSBarry Smith if (flg || (A->rmap->N != A->cmap->N && P->rmap->N == P->cmap->N && P->rmap->N == A->cmap->N)) { 2116f1580f4eSBarry Smith Mat B; 2117811e8887SPierre Jolivet 2118f1580f4eSBarry Smith PetscCall(PCHPDDMSetAuxiliaryMatNormal_Private(pc, A, P, &B, pcpre)); 2119f1580f4eSBarry Smith if (data->correction == PC_HPDDM_COARSE_CORRECTION_DEFLATED) data->correction = PC_HPDDM_COARSE_CORRECTION_BALANCED; 2120f1580f4eSBarry Smith PetscCall(MatDestroy(&B)); 2121f1580f4eSBarry Smith } else { 2122f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)P, MATSCHURCOMPLEMENT, &flg)); 2123f1580f4eSBarry Smith if (flg) { 2124f1580f4eSBarry Smith Mat A00, P00, A01, A10, A11, B, N; 212513044ca3SPierre Jolivet PCHPDDMSchurPreType type = PC_HPDDM_SCHUR_PRE_LEAST_SQUARES; 212613044ca3SPierre Jolivet 212713044ca3SPierre Jolivet PetscCall(MatSchurComplementGetSubMatrices(P, &A00, &P00, &A01, &A10, &A11)); 212813044ca3SPierre Jolivet PetscCall(PetscOptionsGetEnum(nullptr, pcpre, "-pc_hpddm_schur_precondition", PCHPDDMSchurPreTypes, (PetscEnum *)&type, &flg)); 212913044ca3SPierre Jolivet if (type == PC_HPDDM_SCHUR_PRE_LEAST_SQUARES) { 2130281f8ce6SPierre Jolivet Mat B01; 21313df4cd7bSPierre Jolivet Vec diagonal = nullptr; 2132f1580f4eSBarry Smith const PetscScalar *array; 2133f1580f4eSBarry Smith MatSchurComplementAinvType type; 2134f1580f4eSBarry Smith 2135281f8ce6SPierre Jolivet PetscCall(PCHPDDMCheckSymmetry_Private(pc, A01, A10, &B01)); 2136f1580f4eSBarry Smith if (A11) { 21373df4cd7bSPierre Jolivet PetscCall(MatCreateVecs(A11, &diagonal, nullptr)); 21383df4cd7bSPierre Jolivet PetscCall(MatGetDiagonal(A11, diagonal)); 2139f1580f4eSBarry Smith } 2140db4a47b3SPierre Jolivet PetscCall(MatCreateVecs(P00, &v, nullptr)); 2141f1580f4eSBarry Smith PetscCall(MatSchurComplementGetAinvType(P, &type)); 2142f1580f4eSBarry 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]); 2143f1580f4eSBarry Smith if (type == MAT_SCHUR_COMPLEMENT_AINV_LUMP) { 2144f1580f4eSBarry Smith PetscCall(MatGetRowSum(P00, v)); 2145f1580f4eSBarry Smith if (A00 == P00) PetscCall(PetscObjectReference((PetscObject)A00)); 2146f1580f4eSBarry Smith PetscCall(MatDestroy(&P00)); 2147f1580f4eSBarry Smith PetscCall(VecGetArrayRead(v, &array)); 2148db4a47b3SPierre Jolivet PetscCall(MatCreateAIJ(PetscObjectComm((PetscObject)A00), A00->rmap->n, A00->cmap->n, A00->rmap->N, A00->cmap->N, 1, nullptr, 0, nullptr, &P00)); 2149f1580f4eSBarry Smith PetscCall(MatSetOption(P00, MAT_NO_OFF_PROC_ENTRIES, PETSC_TRUE)); 2150f1580f4eSBarry Smith for (n = A00->rmap->rstart; n < A00->rmap->rend; ++n) PetscCall(MatSetValue(P00, n, n, array[n - A00->rmap->rstart], INSERT_VALUES)); 2151f1580f4eSBarry Smith PetscCall(MatAssemblyBegin(P00, MAT_FINAL_ASSEMBLY)); 2152f1580f4eSBarry Smith PetscCall(MatAssemblyEnd(P00, MAT_FINAL_ASSEMBLY)); 2153f1580f4eSBarry Smith PetscCall(VecRestoreArrayRead(v, &array)); 2154f1580f4eSBarry Smith PetscCall(MatSchurComplementUpdateSubMatrices(P, A00, P00, A01, A10, A11)); /* replace P00 by diag(sum of each row of P00) */ 2155f1580f4eSBarry Smith PetscCall(MatDestroy(&P00)); 2156f1580f4eSBarry Smith } else PetscCall(MatGetDiagonal(P00, v)); 2157f1580f4eSBarry Smith PetscCall(VecReciprocal(v)); /* inv(diag(P00)) */ 2158f1580f4eSBarry Smith PetscCall(VecSqrtAbs(v)); /* sqrt(inv(diag(P00))) */ 2159f1580f4eSBarry Smith PetscCall(MatDuplicate(A01, MAT_COPY_VALUES, &B)); 2160db4a47b3SPierre Jolivet PetscCall(MatDiagonalScale(B, v, nullptr)); 2161281f8ce6SPierre Jolivet if (B01) PetscCall(MatDiagonalScale(B01, v, nullptr)); 2162f1580f4eSBarry Smith PetscCall(VecDestroy(&v)); 2163f1580f4eSBarry Smith PetscCall(MatCreateNormalHermitian(B, &N)); 2164281f8ce6SPierre Jolivet PetscCall(PCHPDDMSetAuxiliaryMatNormal_Private(pc, B, N, &P, pcpre, &diagonal, B01)); 2165f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)data->aux, MATSEQAIJ, &flg)); 2166f1580f4eSBarry Smith if (!flg) { 2167f1580f4eSBarry Smith PetscCall(MatDestroy(&P)); 2168f1580f4eSBarry Smith P = N; 2169f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)P)); 2170feebddf4SPierre Jolivet } 21713df4cd7bSPierre Jolivet if (diagonal) { 21723df4cd7bSPierre Jolivet PetscCall(MatDiagonalSet(P, diagonal, ADD_VALUES)); 2173feebddf4SPierre Jolivet PetscCall(PCSetOperators(pc, P, P)); /* replace P by A01^T inv(diag(P00)) A01 - diag(P11) */ 21743df4cd7bSPierre Jolivet PetscCall(VecDestroy(&diagonal)); 2175281f8ce6SPierre Jolivet } else PetscCall(PCSetOperators(pc, B01 ? P : N, P)); /* replace P by A01^T inv(diag(P00)) A01 */ 2176feebddf4SPierre Jolivet pc->ops->postsolve = PCPostSolve_SchurPreLeastSquares; /* PCFIELDSPLIT expect a KSP for (P11 - A10 inv(diag(P00)) A01) */ 2177feebddf4SPierre Jolivet PetscCall(MatDestroy(&N)); /* but a PC for (A10 inv(diag(P00)) A10 - P11) is setup instead */ 2178feebddf4SPierre Jolivet PetscCall(MatDestroy(&P)); /* so the sign of the solution must be flipped */ 2179f1580f4eSBarry Smith PetscCall(MatDestroy(&B)); 218013044ca3SPierre Jolivet } else 2181fd310a01SPierre 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 : ""); 21820307214fSPierre Jolivet for (std::vector<Vec>::iterator it = initial.begin(); it != initial.end(); ++it) PetscCall(VecDestroy(&*it)); 21833ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2184f1580f4eSBarry Smith } else { 2185db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetString(nullptr, pcpre, "-pc_hpddm_levels_1_st_pc_type", type, sizeof(type), nullptr)); 2186f1580f4eSBarry Smith PetscCall(PetscStrcmp(type, PCMAT, &algebraic)); 2187b07dfdedSPierre Jolivet PetscCheck(!algebraic || !block, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_levels_1_st_pc_type mat and -pc_hpddm_block_splitting"); 21889bb5c669SPierre Jolivet if (overlap != -1) { 21899bb5c669SPierre 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"); 21909bb5c669SPierre Jolivet PetscCheck(overlap >= 1, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_WRONG, "-pc_hpddm_harmonic_overlap %" PetscInt_FMT " < 1", overlap); 21919bb5c669SPierre Jolivet } 21929bb5c669SPierre Jolivet if (block || overlap != -1) algebraic = PETSC_TRUE; 2193f1580f4eSBarry Smith if (algebraic) { 2194f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, P->rmap->n, P->rmap->rstart, 1, &data->is)); 2195f1580f4eSBarry Smith PetscCall(MatIncreaseOverlap(P, 1, &data->is, 1)); 2196f1580f4eSBarry Smith PetscCall(ISSort(data->is)); 21979bb5c669SPierre Jolivet } else 21989bb5c669SPierre 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 : "")); 2199f1580f4eSBarry Smith } 2200f1580f4eSBarry Smith } 2201f1580f4eSBarry Smith } 2202f1580f4eSBarry Smith } 2203398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG) 2204398c7888SPierre Jolivet if (data->is) PetscCall(ISDuplicate(data->is, &dis)); 2205398c7888SPierre Jolivet if (data->aux) PetscCall(MatDuplicate(data->aux, MAT_COPY_VALUES, &daux)); 2206398c7888SPierre Jolivet #endif 2207f1580f4eSBarry Smith if (data->is || (ismatis && data->N > 1)) { 2208f1580f4eSBarry Smith if (ismatis) { 2209f1580f4eSBarry Smith std::initializer_list<std::string> list = {MATSEQBAIJ, MATSEQSBAIJ}; 2210f1580f4eSBarry Smith PetscCall(MatISGetLocalMat(P, &N)); 2211f1580f4eSBarry Smith std::initializer_list<std::string>::const_iterator it = std::find(list.begin(), list.end(), ((PetscObject)N)->type_name); 2212f1580f4eSBarry Smith PetscCall(MatISRestoreLocalMat(P, &N)); 2213f1580f4eSBarry Smith switch (std::distance(list.begin(), it)) { 2214d71ae5a4SJacob Faibussowitsch case 0: 2215d71ae5a4SJacob Faibussowitsch PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &C)); 2216d71ae5a4SJacob Faibussowitsch break; 2217f1580f4eSBarry Smith case 1: 2218f1580f4eSBarry Smith /* MatCreateSubMatrices() does not work with MATSBAIJ and unsorted ISes, so convert to MPIBAIJ */ 2219f1580f4eSBarry Smith PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &C)); 2220f1580f4eSBarry Smith PetscCall(MatSetOption(C, MAT_SYMMETRIC, PETSC_TRUE)); 2221f1580f4eSBarry Smith break; 2222d71ae5a4SJacob Faibussowitsch default: 2223d71ae5a4SJacob Faibussowitsch PetscCall(MatConvert(P, MATMPIAIJ, MAT_INITIAL_MATRIX, &C)); 2224f1580f4eSBarry Smith } 2225db4a47b3SPierre Jolivet PetscCall(MatISGetLocalToGlobalMapping(P, &l2g, nullptr)); 2226f1580f4eSBarry Smith PetscCall(PetscObjectReference((PetscObject)P)); 2227f1580f4eSBarry Smith PetscCall(KSPSetOperators(data->levels[0]->ksp, A, C)); 2228f1580f4eSBarry Smith std::swap(C, P); 2229f1580f4eSBarry Smith PetscCall(ISLocalToGlobalMappingGetSize(l2g, &n)); 2230f1580f4eSBarry Smith PetscCall(ISCreateStride(PETSC_COMM_SELF, n, 0, 1, &loc)); 2231f1580f4eSBarry Smith PetscCall(ISLocalToGlobalMappingApplyIS(l2g, loc, &is[0])); 2232f1580f4eSBarry Smith PetscCall(ISDestroy(&loc)); 2233f1580f4eSBarry Smith /* the auxiliary Mat is _not_ the local Neumann matrix */ 2234f1580f4eSBarry Smith /* it is the local Neumann matrix augmented (with zeros) through MatIncreaseOverlap() */ 2235c8ea6600SPierre Jolivet data->Neumann = PETSC_BOOL3_FALSE; 2236f1580f4eSBarry Smith structure = SAME_NONZERO_PATTERN; 2237f1580f4eSBarry Smith } else { 2238f1580f4eSBarry Smith is[0] = data->is; 223913044ca3SPierre Jolivet if (algebraic || ctx) subdomains = PETSC_TRUE; 2240db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetBool(nullptr, pcpre, "-pc_hpddm_define_subdomains", &subdomains, nullptr)); 224113044ca3SPierre 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); 2242c8ea6600SPierre Jolivet if (PetscBool3ToBool(data->Neumann)) { 2243b07dfdedSPierre Jolivet PetscCheck(!block, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_block_splitting and -pc_hpddm_has_neumann"); 22449bb5c669SPierre Jolivet PetscCheck(overlap == -1, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_harmonic_overlap %" PetscInt_FMT " and -pc_hpddm_has_neumann", overlap); 2245b07dfdedSPierre Jolivet PetscCheck(!algebraic, PetscObjectComm((PetscObject)P), PETSC_ERR_ARG_INCOMP, "-pc_hpddm_levels_1_st_pc_type mat and -pc_hpddm_has_neumann"); 2246f1580f4eSBarry Smith } 2247c8ea6600SPierre Jolivet if (PetscBool3ToBool(data->Neumann) || block) structure = SAME_NONZERO_PATTERN; 2248f1580f4eSBarry Smith PetscCall(ISCreateStride(PetscObjectComm((PetscObject)data->is), P->rmap->n, P->rmap->rstart, 1, &loc)); 2249f1580f4eSBarry Smith } 2250f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : "")); 2251db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetEnum(nullptr, prefix, "-st_matstructure", MatStructures, (PetscEnum *)&structure, &flg)); /* if not user-provided, force its value when possible */ 2252f1580f4eSBarry Smith if (!flg && structure == SAME_NONZERO_PATTERN) { /* cannot call STSetMatStructure() yet, insert the appropriate option in the database, parsed by STSetFromOptions() */ 2253f1580f4eSBarry Smith PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "-%spc_hpddm_levels_1_st_matstructure", pcpre ? pcpre : "")); 2254db4a47b3SPierre Jolivet PetscCall(PetscOptionsSetValue(nullptr, prefix, MatStructures[structure])); 2255f1580f4eSBarry Smith } 2256398c7888SPierre Jolivet flg = PETSC_FALSE; 2257b07dfdedSPierre Jolivet if (data->share) { 2258398c7888SPierre Jolivet data->share = PETSC_FALSE; /* will be reset to PETSC_TRUE if none of the conditions below are true */ 2259398c7888SPierre 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 : "")); 2260398c7888SPierre Jolivet else if (data->deflation) PetscCall(PetscInfo(pc, "Nothing to share since PCHPDDMSetDeflationMat() has been called\n")); 2261398c7888SPierre Jolivet else if (ismatis) PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc with a Pmat of type MATIS\n")); 2262b07dfdedSPierre Jolivet else if (!algebraic && structure != SAME_NONZERO_PATTERN) 2263398c7888SPierre 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])); 2264398c7888SPierre Jolivet else data->share = PETSC_TRUE; 2265398c7888SPierre Jolivet } 2266398c7888SPierre Jolivet if (!ismatis) { 2267398c7888SPierre Jolivet if (data->share || (!PetscBool3ToBool(data->Neumann) && subdomains)) PetscCall(ISDuplicate(is[0], &unsorted)); 2268398c7888SPierre Jolivet else unsorted = is[0]; 2269398c7888SPierre Jolivet } 227051ea4bc8SPierre Jolivet if ((ctx || data->N > 1) && (data->aux || ismatis || algebraic)) { 2271f1580f4eSBarry Smith PetscCheck(loadedSym, PETSC_COMM_SELF, PETSC_ERR_PLIB, "HPDDM library not loaded, cannot use more than one level"); 2272f1580f4eSBarry Smith PetscCall(MatSetOption(P, MAT_SUBMAT_SINGLEIS, PETSC_TRUE)); 2273f1580f4eSBarry Smith if (ismatis) { 2274f1580f4eSBarry Smith /* needed by HPDDM (currently) so that the partition of unity is 0 on subdomain interfaces */ 2275f1580f4eSBarry Smith PetscCall(MatIncreaseOverlap(P, 1, is, 1)); 2276f1580f4eSBarry Smith PetscCall(ISDestroy(&data->is)); 2277f1580f4eSBarry Smith data->is = is[0]; 2278f1580f4eSBarry Smith } else { 2279d16c0b94SPierre Jolivet if (PetscDefined(USE_DEBUG)) PetscCall(PCHPDDMCheckInclusion_Private(pc, data->is, loc, PETSC_TRUE)); 228051ea4bc8SPierre Jolivet if (!ctx && overlap == -1) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", PCHPDDMAlgebraicAuxiliaryMat_Private)); 22819bb5c669SPierre Jolivet if (!PetscBool3ToBool(data->Neumann) && (!algebraic || overlap != -1)) { 2282f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)P, MATMPISBAIJ, &flg)); 2283f1580f4eSBarry Smith if (flg) { 2284f1580f4eSBarry Smith /* maybe better to ISSort(is[0]), MatCreateSubMatrices(), and then MatPermute() */ 2285f1580f4eSBarry Smith /* but there is no MatPermute_SeqSBAIJ(), so as before, just use MATMPIBAIJ */ 2286f1580f4eSBarry Smith PetscCall(MatConvert(P, MATMPIBAIJ, MAT_INITIAL_MATRIX, &uaux)); 2287f1580f4eSBarry Smith flg = PETSC_FALSE; 2288f1580f4eSBarry Smith } 2289f1580f4eSBarry Smith } 2290f1580f4eSBarry Smith } 22919bb5c669SPierre Jolivet if (algebraic && overlap == -1) { 2292f1580f4eSBarry Smith PetscUseMethod(pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", (Mat, IS *, Mat *[], PetscBool), (P, is, &sub, block)); 2293f1580f4eSBarry Smith if (block) { 2294f1580f4eSBarry Smith PetscCall(PetscObjectQuery((PetscObject)sub[0], "_PCHPDDM_Neumann_Mat", (PetscObject *)&data->aux)); 2295db4a47b3SPierre Jolivet PetscCall(PetscObjectCompose((PetscObject)sub[0], "_PCHPDDM_Neumann_Mat", nullptr)); 2296f1580f4eSBarry Smith } 22979bb5c669SPierre Jolivet } else if (!uaux || overlap != -1) { 229813044ca3SPierre Jolivet if (!ctx) { 2299c8ea6600SPierre Jolivet if (PetscBool3ToBool(data->Neumann)) sub = &data->aux; 23009bb5c669SPierre Jolivet else { 2301df21ef68SPierre Jolivet PetscBool flg; 23029bb5c669SPierre Jolivet if (overlap != -1) { 23039bb5c669SPierre Jolivet Harmonic h; 23049bb5c669SPierre Jolivet Mat A0, *a; /* with an SVD: [ A_00 A_01 ] */ 23059bb5c669SPierre Jolivet IS ov[2], rows, cols, stride; /* [ A_10 A_11 A_12 ] */ 230658b7e2c1SStefano Zampini const PetscInt *i[2], bs = P->cmap->bs; /* with a GEVP: [ A_00 A_01 ] */ 23079bb5c669SPierre Jolivet PetscInt n[2]; /* [ A_10 A_11 A_12 ] */ 23089bb5c669SPierre Jolivet std::vector<PetscInt> v[2]; /* [ A_21 A_22 ] */ 23099bb5c669SPierre Jolivet 23102589ceddSPierre Jolivet do { 23119bb5c669SPierre Jolivet PetscCall(ISDuplicate(data->is, ov)); 23129bb5c669SPierre Jolivet if (overlap > 1) PetscCall(MatIncreaseOverlap(P, 1, ov, overlap - 1)); 23139bb5c669SPierre Jolivet PetscCall(ISDuplicate(ov[0], ov + 1)); 23149bb5c669SPierre Jolivet PetscCall(MatIncreaseOverlap(P, 1, ov + 1, 1)); 23152589ceddSPierre Jolivet PetscCall(ISGetLocalSize(ov[0], n)); 23162589ceddSPierre Jolivet PetscCall(ISGetLocalSize(ov[1], n + 1)); 23178fad524dSPierre Jolivet flg = PetscBool(n[0] == n[1] && n[0] != P->rmap->n); 23185440e5dcSBarry Smith PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &flg, 1, MPI_C_BOOL, MPI_LOR, PetscObjectComm((PetscObject)pc))); 23192589ceddSPierre Jolivet if (flg) { 23202589ceddSPierre Jolivet PetscCall(ISDestroy(ov)); 23212589ceddSPierre Jolivet PetscCall(ISDestroy(ov + 1)); 23222589ceddSPierre Jolivet PetscCheck(--overlap, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "No oversampling possible"); 23232589ceddSPierre Jolivet PetscCall(PetscInfo(pc, "Supplied -%spc_hpddm_harmonic_overlap parameter is too large, it has been decreased to %" PetscInt_FMT "\n", pcpre ? pcpre : "", overlap)); 23242589ceddSPierre Jolivet } else break; 23252589ceddSPierre Jolivet } while (1); 23269bb5c669SPierre Jolivet PetscCall(PetscNew(&h)); 23279bb5c669SPierre Jolivet h->ksp = nullptr; 23289bb5c669SPierre Jolivet PetscCall(PetscCalloc1(2, &h->A)); 2329b5a302b3SPierre Jolivet PetscCall(PetscOptionsHasName(nullptr, prefix, "-eps_nev", &flg)); 2330b5a302b3SPierre Jolivet if (!flg) { 23313a60673dSPierre Jolivet PetscCall(PetscOptionsHasName(nullptr, prefix, "-svd_nsv", &flg)); 2332b5a302b3SPierre Jolivet if (!flg) PetscCall(PetscOptionsHasName(nullptr, prefix, "-svd_threshold_relative", &flg)); 2333b5a302b3SPierre Jolivet } else flg = PETSC_FALSE; 23349bb5c669SPierre Jolivet PetscCall(ISSort(ov[0])); 23359bb5c669SPierre Jolivet if (!flg) PetscCall(ISSort(ov[1])); 233632603206SJames Wright PetscCall(PetscCalloc1(5, &h->is)); 23379bb5c669SPierre Jolivet PetscCall(MatCreateSubMatrices(uaux ? uaux : P, 1, ov + !flg, ov + 1, MAT_INITIAL_MATRIX, &a)); /* submatrix from above, either square (!flg) or rectangular (flg) */ 23382589ceddSPierre Jolivet for (PetscInt j = 0; j < 2; ++j) PetscCall(ISGetIndices(ov[j], i + j)); 23399bb5c669SPierre Jolivet v[1].reserve((n[1] - n[0]) / bs); 23409bb5c669SPierre Jolivet for (PetscInt j = 0; j < n[1]; j += bs) { /* indices of the (2,2) block */ 23419bb5c669SPierre Jolivet PetscInt location; 23429bb5c669SPierre Jolivet PetscCall(ISLocate(ov[0], i[1][j], &location)); 23439bb5c669SPierre Jolivet if (location < 0) v[1].emplace_back(j / bs); 23449bb5c669SPierre Jolivet } 23459bb5c669SPierre Jolivet if (!flg) { 23469bb5c669SPierre Jolivet h->A[1] = a[0]; 23479bb5c669SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)h->A[1])); 23489bb5c669SPierre Jolivet v[0].reserve((n[0] - P->rmap->n) / bs); 23499bb5c669SPierre Jolivet for (PetscInt j = 0; j < n[1]; j += bs) { /* row indices of the (1,2) block */ 23509bb5c669SPierre Jolivet PetscInt location; 23519bb5c669SPierre Jolivet PetscCall(ISLocate(loc, i[1][j], &location)); 23529bb5c669SPierre Jolivet if (location < 0) { 23539bb5c669SPierre Jolivet PetscCall(ISLocate(ov[0], i[1][j], &location)); 23549bb5c669SPierre Jolivet if (location >= 0) v[0].emplace_back(j / bs); 23559bb5c669SPierre Jolivet } 23569bb5c669SPierre Jolivet } 23579bb5c669SPierre Jolivet PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[0].size(), v[0].data(), PETSC_USE_POINTER, &rows)); 23589bb5c669SPierre Jolivet PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[1].size(), v[1].data(), PETSC_COPY_VALUES, h->is + 4)); 23599bb5c669SPierre Jolivet PetscCall(MatCreateSubMatrix(a[0], rows, h->is[4], MAT_INITIAL_MATRIX, h->A)); /* A_12 submatrix from above */ 23609bb5c669SPierre Jolivet PetscCall(ISDestroy(&rows)); 23619bb5c669SPierre 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 */ 23629bb5c669SPierre Jolivet PetscCall(ISEmbed(ov[0], ov[1], PETSC_TRUE, &rows)); 23639bb5c669SPierre Jolivet PetscCall(MatCreateSubMatrix(a[0], rows, cols = rows, MAT_INITIAL_MATRIX, &A0)); /* [ A_00 A_01 ; A_10 A_11 ] submatrix from above */ 23649bb5c669SPierre Jolivet PetscCall(ISDestroy(&rows)); 23659bb5c669SPierre Jolivet v[0].clear(); 23669bb5c669SPierre Jolivet PetscCall(ISEmbed(loc, ov[1], PETSC_TRUE, h->is + 3)); 23679bb5c669SPierre Jolivet PetscCall(ISEmbed(data->is, ov[1], PETSC_TRUE, h->is + 2)); 23689bb5c669SPierre Jolivet } 23699bb5c669SPierre Jolivet v[0].reserve((n[0] - P->rmap->n) / bs); 23709bb5c669SPierre Jolivet for (PetscInt j = 0; j < n[0]; j += bs) { 23719bb5c669SPierre Jolivet PetscInt location; 23729bb5c669SPierre Jolivet PetscCall(ISLocate(loc, i[0][j], &location)); 23739bb5c669SPierre Jolivet if (location < 0) v[0].emplace_back(j / bs); 23749bb5c669SPierre Jolivet } 23759bb5c669SPierre Jolivet PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[0].size(), v[0].data(), PETSC_USE_POINTER, &rows)); 23769bb5c669SPierre Jolivet for (PetscInt j = 0; j < 2; ++j) PetscCall(ISRestoreIndices(ov[j], i + j)); 23779bb5c669SPierre Jolivet if (flg) { 23789bb5c669SPierre Jolivet IS is; 23799bb5c669SPierre Jolivet PetscCall(ISCreateStride(PETSC_COMM_SELF, a[0]->rmap->n, 0, 1, &is)); 23809bb5c669SPierre Jolivet PetscCall(ISEmbed(ov[0], ov[1], PETSC_TRUE, &cols)); 23819bb5c669SPierre Jolivet PetscCall(MatCreateSubMatrix(a[0], is, cols, MAT_INITIAL_MATRIX, &A0)); /* [ A_00 A_01 ; A_10 A_11 ] submatrix from above */ 23829bb5c669SPierre Jolivet PetscCall(ISDestroy(&cols)); 23839bb5c669SPierre Jolivet PetscCall(ISDestroy(&is)); 23849bb5c669SPierre 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 */ 23859bb5c669SPierre Jolivet PetscCall(ISEmbed(loc, data->is, PETSC_TRUE, h->is + 2)); 23869bb5c669SPierre Jolivet PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, v[1].size(), v[1].data(), PETSC_USE_POINTER, &cols)); 23879bb5c669SPierre Jolivet PetscCall(MatCreateSubMatrix(a[0], rows, cols, MAT_INITIAL_MATRIX, h->A)); /* A_12 submatrix from above */ 23889bb5c669SPierre Jolivet PetscCall(ISDestroy(&cols)); 23899bb5c669SPierre Jolivet } 23909bb5c669SPierre Jolivet PetscCall(ISCreateStride(PETSC_COMM_SELF, A0->rmap->n, 0, 1, &stride)); 23919bb5c669SPierre Jolivet PetscCall(ISEmbed(rows, stride, PETSC_TRUE, h->is)); 23929bb5c669SPierre Jolivet PetscCall(ISDestroy(&stride)); 23939bb5c669SPierre Jolivet PetscCall(ISDestroy(&rows)); 23949bb5c669SPierre Jolivet PetscCall(ISEmbed(loc, ov[0], PETSC_TRUE, h->is + 1)); 23959bb5c669SPierre Jolivet if (subdomains) { 23969bb5c669SPierre Jolivet if (!data->levels[0]->pc) { 23979bb5c669SPierre Jolivet PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->pc)); 23989bb5c669SPierre Jolivet PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : "")); 23999bb5c669SPierre Jolivet PetscCall(PCSetOptionsPrefix(data->levels[0]->pc, prefix)); 24009bb5c669SPierre Jolivet PetscCall(PCSetOperators(data->levels[0]->pc, A, P)); 24019bb5c669SPierre Jolivet } 24029bb5c669SPierre Jolivet PetscCall(PCSetType(data->levels[0]->pc, PCASM)); 24039bb5c669SPierre Jolivet if (!data->levels[0]->pc->setupcalled) PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, ov + !flg, &loc)); 24049bb5c669SPierre Jolivet PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(data->levels[0]->pc, flg ? A0 : a[0], PETSC_TRUE)); 24059bb5c669SPierre Jolivet if (!flg) ++overlap; 24069bb5c669SPierre Jolivet if (data->share) { 24079bb5c669SPierre Jolivet PetscInt n = -1; 24089bb5c669SPierre Jolivet PetscTryMethod(data->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data->levels[0]->pc, &n, nullptr, &ksp)); 24099bb5c669SPierre Jolivet PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", n); 24109bb5c669SPierre Jolivet if (flg) { 24119bb5c669SPierre Jolivet h->ksp = ksp[0]; 24129bb5c669SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)h->ksp)); 24139bb5c669SPierre Jolivet } 24149bb5c669SPierre Jolivet } 24159bb5c669SPierre Jolivet } 24169bb5c669SPierre Jolivet if (!h->ksp) { 24179bb5c669SPierre Jolivet PetscBool share = data->share; 24189bb5c669SPierre Jolivet PetscCall(KSPCreate(PETSC_COMM_SELF, &h->ksp)); 24199bb5c669SPierre Jolivet PetscCall(KSPSetType(h->ksp, KSPPREONLY)); 24209bb5c669SPierre Jolivet PetscCall(KSPSetOperators(h->ksp, A0, A0)); 24219bb5c669SPierre Jolivet do { 24229bb5c669SPierre Jolivet if (!data->share) { 24239bb5c669SPierre Jolivet share = PETSC_FALSE; 24249bb5c669SPierre Jolivet PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_%s", pcpre ? pcpre : "", flg ? "svd_" : "eps_")); 24259bb5c669SPierre Jolivet PetscCall(KSPSetOptionsPrefix(h->ksp, prefix)); 24269bb5c669SPierre Jolivet PetscCall(KSPSetFromOptions(h->ksp)); 24279bb5c669SPierre Jolivet } else { 24289bb5c669SPierre Jolivet MatSolverType type; 24299bb5c669SPierre Jolivet PetscCall(KSPGetPC(ksp[0], &pc)); 24309bb5c669SPierre Jolivet PetscCall(PetscObjectTypeCompareAny((PetscObject)pc, &data->share, PCLU, PCCHOLESKY, "")); 24319bb5c669SPierre Jolivet if (data->share) { 24329bb5c669SPierre Jolivet PetscCall(PCFactorGetMatSolverType(pc, &type)); 24339bb5c669SPierre Jolivet if (!type) { 24349bb5c669SPierre Jolivet if (PetscDefined(HAVE_MUMPS)) PetscCall(PCFactorSetMatSolverType(pc, MATSOLVERMUMPS)); 24359bb5c669SPierre Jolivet else if (PetscDefined(HAVE_MKL_PARDISO)) PetscCall(PCFactorSetMatSolverType(pc, MATSOLVERMKL_PARDISO)); 24369bb5c669SPierre Jolivet else data->share = PETSC_FALSE; 24379bb5c669SPierre Jolivet if (data->share) PetscCall(PCSetFromOptions(pc)); 24389bb5c669SPierre Jolivet } else { 24399bb5c669SPierre Jolivet PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &data->share)); 24409bb5c669SPierre Jolivet if (!data->share) PetscCall(PetscStrcmp(type, MATSOLVERMKL_PARDISO, &data->share)); 24419bb5c669SPierre Jolivet } 24429bb5c669SPierre Jolivet if (data->share) { 24439bb5c669SPierre Jolivet std::tuple<KSP, IS, Vec[2]> *p; 24449bb5c669SPierre Jolivet PetscCall(PCFactorGetMatrix(pc, &A)); 24459bb5c669SPierre Jolivet PetscCall(MatFactorSetSchurIS(A, h->is[4])); 24469bb5c669SPierre Jolivet PetscCall(KSPSetUp(ksp[0])); 24479bb5c669SPierre Jolivet PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_eps_shell_", pcpre ? pcpre : "")); 24489bb5c669SPierre Jolivet PetscCall(KSPSetOptionsPrefix(h->ksp, prefix)); 24499bb5c669SPierre Jolivet PetscCall(KSPSetFromOptions(h->ksp)); 24509bb5c669SPierre Jolivet PetscCall(KSPGetPC(h->ksp, &pc)); 24519bb5c669SPierre Jolivet PetscCall(PCSetType(pc, PCSHELL)); 24529bb5c669SPierre Jolivet PetscCall(PetscNew(&p)); 24539bb5c669SPierre Jolivet std::get<0>(*p) = ksp[0]; 24549bb5c669SPierre Jolivet PetscCall(ISEmbed(ov[0], ov[1], PETSC_TRUE, &std::get<1>(*p))); 24559bb5c669SPierre Jolivet PetscCall(MatCreateVecs(A, std::get<2>(*p), std::get<2>(*p) + 1)); 24569bb5c669SPierre Jolivet PetscCall(PCShellSetContext(pc, p)); 24579bb5c669SPierre Jolivet PetscCall(PCShellSetApply(pc, PCApply_Schur)); 24589bb5c669SPierre Jolivet PetscCall(PCShellSetApplyTranspose(pc, PCApply_Schur<Vec, true>)); 24599bb5c669SPierre Jolivet PetscCall(PCShellSetMatApply(pc, PCApply_Schur<Mat>)); 24609bb5c669SPierre Jolivet PetscCall(PCShellSetDestroy(pc, PCDestroy_Schur)); 24619bb5c669SPierre Jolivet } 24629bb5c669SPierre Jolivet } 24639bb5c669SPierre Jolivet if (!data->share) PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc since neither MUMPS nor MKL PARDISO is used\n")); 24649bb5c669SPierre Jolivet } 24659bb5c669SPierre 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 */ 24669bb5c669SPierre Jolivet } 24679bb5c669SPierre Jolivet PetscCall(ISDestroy(ov)); 24689bb5c669SPierre Jolivet PetscCall(ISDestroy(ov + 1)); 24699bb5c669SPierre Jolivet if (overlap == 1 && subdomains && flg) { 24709bb5c669SPierre Jolivet *subA = A0; 24719bb5c669SPierre Jolivet sub = subA; 24729bb5c669SPierre Jolivet if (uaux) PetscCall(MatDestroy(&uaux)); 24739bb5c669SPierre Jolivet } else PetscCall(MatDestroy(&A0)); 24749bb5c669SPierre Jolivet PetscCall(MatCreateShell(PETSC_COMM_SELF, P->rmap->n, n[1] - n[0], P->rmap->n, n[1] - n[0], h, &data->aux)); 2475bdcd51b8SPierre Jolivet PetscCall(KSPSetErrorIfNotConverged(h->ksp, PETSC_TRUE)); /* bail out as early as possible to avoid (apparently) unrelated error messages */ 24769bb5c669SPierre Jolivet PetscCall(MatCreateVecs(h->ksp->pc->pmat, &h->v, nullptr)); 247757d50842SBarry Smith PetscCall(MatShellSetOperation(data->aux, MATOP_MULT, (PetscErrorCodeFn *)MatMult_Harmonic)); 247857d50842SBarry Smith PetscCall(MatShellSetOperation(data->aux, MATOP_MULT_TRANSPOSE, (PetscErrorCodeFn *)MatMultTranspose_Harmonic)); 24799bb5c669SPierre Jolivet PetscCall(MatShellSetMatProductOperation(data->aux, MATPRODUCT_AB, nullptr, MatProduct_AB_Harmonic, nullptr, MATDENSE, MATDENSE)); 2480f21e3f8aSPierre Jolivet PetscCall(MatShellSetMatProductOperation(data->aux, MATPRODUCT_AtB, nullptr, MatProduct_AtB_Harmonic, nullptr, MATDENSE, MATDENSE)); 248157d50842SBarry Smith PetscCall(MatShellSetOperation(data->aux, MATOP_DESTROY, (PetscErrorCodeFn *)MatDestroy_Harmonic)); 24829bb5c669SPierre Jolivet PetscCall(MatDestroySubMatrices(1, &a)); 24839bb5c669SPierre Jolivet } 2484df21ef68SPierre Jolivet if (overlap != 1 || !subdomains) { 2485df21ef68SPierre Jolivet PetscCall(MatCreateSubMatrices(uaux ? uaux : P, 1, is, is, MAT_INITIAL_MATRIX, &sub)); 2486df21ef68SPierre Jolivet if (ismatis) { 2487df21ef68SPierre Jolivet PetscCall(MatISGetLocalMat(C, &N)); 2488df21ef68SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)N, MATSEQSBAIJ, &flg)); 2489df21ef68SPierre Jolivet if (flg) PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub)); 2490df21ef68SPierre Jolivet PetscCall(MatISRestoreLocalMat(C, &N)); 2491df21ef68SPierre Jolivet } 2492df21ef68SPierre Jolivet } 24939bb5c669SPierre Jolivet if (uaux) { 24949bb5c669SPierre Jolivet PetscCall(MatDestroy(&uaux)); 24959bb5c669SPierre Jolivet PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub)); 24969bb5c669SPierre Jolivet } 24979bb5c669SPierre Jolivet } 249813044ca3SPierre Jolivet } 249951ea4bc8SPierre Jolivet } else if (!ctx) { 2500f1580f4eSBarry Smith PetscCall(MatCreateSubMatrices(uaux, 1, is, is, MAT_INITIAL_MATRIX, &sub)); 2501f1580f4eSBarry Smith PetscCall(MatDestroy(&uaux)); 2502f1580f4eSBarry Smith PetscCall(MatConvert(sub[0], MATSEQSBAIJ, MAT_INPLACE_MATRIX, sub)); 2503f1580f4eSBarry Smith } 250451ea4bc8SPierre Jolivet if (data->N > 1) { 2505f1580f4eSBarry Smith /* Vec holding the partition of unity */ 2506f1580f4eSBarry Smith if (!data->levels[0]->D) { 2507f1580f4eSBarry Smith PetscCall(ISGetLocalSize(data->is, &n)); 2508f1580f4eSBarry Smith PetscCall(VecCreateMPI(PETSC_COMM_SELF, n, PETSC_DETERMINE, &data->levels[0]->D)); 2509f1580f4eSBarry Smith } 25109bb5c669SPierre Jolivet if (data->share && overlap == -1) { 2511f1580f4eSBarry Smith Mat D; 2512db4a47b3SPierre Jolivet IS perm = nullptr; 2513f1580f4eSBarry Smith PetscInt size = -1; 2514811e8887SPierre Jolivet 2515398c7888SPierre Jolivet if (!data->levels[0]->pc) { 2516398c7888SPierre Jolivet PetscCall(PetscSNPrintf(prefix, sizeof(prefix), "%spc_hpddm_levels_1_", pcpre ? pcpre : "")); 2517398c7888SPierre Jolivet PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), &data->levels[0]->pc)); 2518398c7888SPierre Jolivet PetscCall(PCSetOptionsPrefix(data->levels[0]->pc, prefix)); 2519398c7888SPierre Jolivet PetscCall(PCSetOperators(data->levels[0]->pc, A, P)); 2520398c7888SPierre Jolivet } 2521398c7888SPierre Jolivet PetscCall(PCSetType(data->levels[0]->pc, PCASM)); 252213044ca3SPierre Jolivet if (!ctx) { 2523398c7888SPierre Jolivet if (!data->levels[0]->pc->setupcalled) { 2524398c7888SPierre Jolivet IS sorted; /* PCASM will sort the input IS, duplicate it to return an unmodified (PCHPDDM) input IS */ 2525398c7888SPierre Jolivet PetscCall(ISDuplicate(is[0], &sorted)); 2526398c7888SPierre Jolivet PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, &sorted, &loc)); 2527398c7888SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)sorted)); 2528398c7888SPierre Jolivet } 2529398c7888SPierre Jolivet PetscCall(PCSetFromOptions(data->levels[0]->pc)); 2530398c7888SPierre Jolivet if (block) { 2531398c7888SPierre Jolivet PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, sub[0], &C, &perm)); 2532398c7888SPierre Jolivet PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(data->levels[0]->pc, C, algebraic)); 2533398c7888SPierre Jolivet } else PetscCall(PCSetUp(data->levels[0]->pc)); 2534db4a47b3SPierre Jolivet PetscTryMethod(data->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data->levels[0]->pc, &size, nullptr, &ksp)); 2535398c7888SPierre Jolivet if (size != 1) { 2536398c7888SPierre Jolivet data->share = PETSC_FALSE; 2537398c7888SPierre Jolivet PetscCheck(size == -1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", size); 2538398c7888SPierre Jolivet PetscCall(PetscInfo(pc, "Cannot share subdomain KSP between SLEPc and PETSc since PCASMGetSubKSP() not found in fine-level PC\n")); 2539398c7888SPierre Jolivet PetscCall(ISDestroy(&unsorted)); 2540398c7888SPierre Jolivet unsorted = is[0]; 2541398c7888SPierre Jolivet } else { 2542811e8887SPierre Jolivet const char *matpre; 2543811e8887SPierre Jolivet PetscBool cmp[4]; 2544811e8887SPierre Jolivet 254513044ca3SPierre Jolivet if (!block && !ctx) PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, PetscBool3ToBool(data->Neumann) ? sub[0] : data->aux, &C, &perm)); 254651ea4bc8SPierre Jolivet if (perm) { /* unsorted input IS */ 2547c8ea6600SPierre Jolivet if (!PetscBool3ToBool(data->Neumann) && !block) { 2548f1580f4eSBarry Smith PetscCall(MatPermute(sub[0], perm, perm, &D)); /* permute since PCASM will call ISSort() */ 2549f1580f4eSBarry Smith PetscCall(MatHeaderReplace(sub[0], &D)); 2550f1580f4eSBarry Smith } 2551f1580f4eSBarry Smith if (data->B) { /* see PCHPDDMSetRHSMat() */ 2552f1580f4eSBarry Smith PetscCall(MatPermute(data->B, perm, perm, &D)); 2553f1580f4eSBarry Smith PetscCall(MatHeaderReplace(data->B, &D)); 2554f1580f4eSBarry Smith } 2555f1580f4eSBarry Smith PetscCall(ISDestroy(&perm)); 255651ea4bc8SPierre Jolivet } 2557f1580f4eSBarry Smith PetscCall(KSPGetOperators(ksp[0], subA, subA + 1)); 25580307214fSPierre Jolivet PetscCall(PetscObjectReference((PetscObject)subA[0])); 2559f1580f4eSBarry Smith PetscCall(MatDuplicate(subA[1], MAT_SHARE_NONZERO_PATTERN, &D)); 2560f1580f4eSBarry Smith PetscCall(MatGetOptionsPrefix(subA[1], &matpre)); 2561f1580f4eSBarry Smith PetscCall(MatSetOptionsPrefix(D, matpre)); 2562f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)D, MATNORMAL, cmp)); 2563f1580f4eSBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)C, MATNORMAL, cmp + 1)); 2564f1580f4eSBarry Smith if (!cmp[0]) PetscCall(PetscObjectTypeCompare((PetscObject)D, MATNORMALHERMITIAN, cmp + 2)); 2565f1580f4eSBarry Smith else cmp[2] = PETSC_FALSE; 2566f1580f4eSBarry Smith if (!cmp[1]) PetscCall(PetscObjectTypeCompare((PetscObject)C, MATNORMALHERMITIAN, cmp + 3)); 2567f1580f4eSBarry Smith else cmp[3] = PETSC_FALSE; 2568f1580f4eSBarry 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); 2569f1580f4eSBarry Smith if (!cmp[0] && !cmp[2]) { 2570*8a8e6071SPierre Jolivet if (!block) { 2571*8a8e6071SPierre Jolivet if (PetscDefined(USE_DEBUG)) PetscCall(PCHPDDMCheckMatStructure_Private(pc, D, C)); 2572*8a8e6071SPierre Jolivet PetscCall(MatAXPY(D, 1.0, C, SUBSET_NONZERO_PATTERN)); 2573*8a8e6071SPierre Jolivet } else { 2574421480d9SBarry Smith structure = DIFFERENT_NONZERO_PATTERN; 25755c7345deSPierre Jolivet PetscCall(MatAXPY(D, 1.0, data->aux, structure)); 25765c7345deSPierre Jolivet } 2577f1580f4eSBarry Smith } else { 2578f1580f4eSBarry Smith Mat mat[2]; 2579811e8887SPierre Jolivet 2580f1580f4eSBarry Smith if (cmp[0]) { 2581f1580f4eSBarry Smith PetscCall(MatNormalGetMat(D, mat)); 2582f1580f4eSBarry Smith PetscCall(MatNormalGetMat(C, mat + 1)); 2583f1580f4eSBarry Smith } else { 2584f1580f4eSBarry Smith PetscCall(MatNormalHermitianGetMat(D, mat)); 2585f1580f4eSBarry Smith PetscCall(MatNormalHermitianGetMat(C, mat + 1)); 2586f1580f4eSBarry Smith } 2587f1580f4eSBarry Smith PetscCall(MatAXPY(mat[0], 1.0, mat[1], SUBSET_NONZERO_PATTERN)); 2588f1580f4eSBarry Smith } 2589f1580f4eSBarry Smith PetscCall(MatPropagateSymmetryOptions(C, D)); 2590f1580f4eSBarry Smith PetscCall(MatDestroy(&C)); 2591f1580f4eSBarry Smith C = D; 2592f1580f4eSBarry Smith /* swap pointers so that variables stay consistent throughout PCSetUp() */ 2593f1580f4eSBarry Smith std::swap(C, data->aux); 2594f1580f4eSBarry Smith std::swap(uis, data->is); 2595f1580f4eSBarry Smith swap = PETSC_TRUE; 2596f1580f4eSBarry Smith } 2597f1580f4eSBarry Smith } 259813044ca3SPierre Jolivet } 259951ea4bc8SPierre Jolivet } 260013044ca3SPierre Jolivet if (ctx) { 260113044ca3SPierre Jolivet PC_HPDDM *data_00 = (PC_HPDDM *)std::get<0>(*ctx)[0]->data; 260213044ca3SPierre Jolivet PC s; 260300b491fbSPierre Jolivet Mat A00, P00, A01 = nullptr, A10, A11, N, b[4]; 260407d8d47fSPierre Jolivet IS sorted, is[2], *is_00; 260513044ca3SPierre Jolivet MatSolverType type; 260613044ca3SPierre Jolivet std::pair<PC, Vec[2]> *p; 260713044ca3SPierre Jolivet 260807d8d47fSPierre Jolivet n = -1; 260907d8d47fSPierre Jolivet PetscTryMethod(data_00->levels[0]->pc, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (data_00->levels[0]->pc, &n, nullptr, &ksp)); 261007d8d47fSPierre Jolivet PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", n); 261107d8d47fSPierre Jolivet PetscCall(KSPGetOperators(ksp[0], subA, subA + 1)); 261207d8d47fSPierre Jolivet PetscCall(ISGetLocalSize(data_00->is, &n)); 261307d8d47fSPierre Jolivet if (n != subA[0]->rmap->n || n != subA[0]->cmap->n) { 261407d8d47fSPierre Jolivet PetscCall(PCASMGetLocalSubdomains(data_00->levels[0]->pc, &n, &is_00, nullptr)); 261507d8d47fSPierre Jolivet PetscCall(ISGetLocalSize(*is_00, &n)); 261607d8d47fSPierre 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); 261707d8d47fSPierre Jolivet } else is_00 = &data_00->is; 261813044ca3SPierre Jolivet PetscCall(PCHPDDMPermute_Private(unsorted, data->is, &uis, data->aux, &C, nullptr)); /* permute since PCASM works with a sorted IS */ 261913044ca3SPierre Jolivet std::swap(C, data->aux); 262013044ca3SPierre Jolivet std::swap(uis, data->is); 262113044ca3SPierre Jolivet swap = PETSC_TRUE; 262213044ca3SPierre Jolivet PetscCall(MatSchurComplementGetSubMatrices(P, &A00, &P00, std::get<1>(*ctx), &A10, &A11)); 262313044ca3SPierre Jolivet std::get<1>(*ctx)[1] = A10; 262413044ca3SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATTRANSPOSEVIRTUAL, &flg)); 262513044ca3SPierre Jolivet if (flg) PetscCall(MatTransposeGetMat(A10, &A01)); 262613044ca3SPierre Jolivet else { 262713044ca3SPierre Jolivet PetscBool flg; 262813044ca3SPierre Jolivet 262913044ca3SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)A10, MATHERMITIANTRANSPOSEVIRTUAL, &flg)); 263013044ca3SPierre Jolivet if (flg) PetscCall(MatHermitianTransposeGetMat(A10, &A01)); 263113044ca3SPierre Jolivet } 263207d8d47fSPierre 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 */ 263313044ca3SPierre Jolivet PetscCall(ISSort(sorted)); /* this is to avoid changing users inputs, but it requires a new call to ISSort() here */ 263400b491fbSPierre Jolivet if (!A01) { 263500b491fbSPierre Jolivet PetscCall(MatSetOption(A10, MAT_SUBMAT_SINGLEIS, PETSC_TRUE)); 263600b491fbSPierre Jolivet PetscCall(MatCreateSubMatrices(A10, 1, &data->is, &sorted, MAT_INITIAL_MATRIX, &sub)); 263700b491fbSPierre Jolivet b[2] = sub[0]; 263800b491fbSPierre Jolivet PetscCall(PetscObjectReference((PetscObject)sub[0])); 263900b491fbSPierre Jolivet PetscCall(MatDestroySubMatrices(1, &sub)); 264000b491fbSPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)std::get<1>(*ctx)[0], MATTRANSPOSEVIRTUAL, &flg)); 264100b491fbSPierre Jolivet A10 = nullptr; 264200b491fbSPierre Jolivet if (flg) PetscCall(MatTransposeGetMat(std::get<1>(*ctx)[0], &A10)); 264313044ca3SPierre Jolivet else { 264400b491fbSPierre Jolivet PetscBool flg; 264500b491fbSPierre Jolivet 2646c71f06a7SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)std::get<1>(*ctx)[0], MATHERMITIANTRANSPOSEVIRTUAL, &flg)); 264700b491fbSPierre Jolivet if (flg) PetscCall(MatHermitianTransposeGetMat(std::get<1>(*ctx)[0], &A10)); 264813044ca3SPierre Jolivet } 2649811e8887SPierre Jolivet if (!A10) PetscCall(MatCreateSubMatrices(std::get<1>(*ctx)[0], 1, &sorted, &data->is, MAT_INITIAL_MATRIX, &sub)); 2650811e8887SPierre Jolivet else { 2651811e8887SPierre Jolivet if (flg) PetscCall(MatCreateTranspose(b[2], b + 1)); 2652811e8887SPierre Jolivet else PetscCall(MatCreateHermitianTranspose(b[2], b + 1)); 265300b491fbSPierre Jolivet } 265400b491fbSPierre Jolivet } else { 265500b491fbSPierre Jolivet PetscCall(MatSetOption(A01, MAT_SUBMAT_SINGLEIS, PETSC_TRUE)); 265600b491fbSPierre Jolivet PetscCall(MatCreateSubMatrices(A01, 1, &sorted, &data->is, MAT_INITIAL_MATRIX, &sub)); 2657811e8887SPierre Jolivet if (flg) PetscCall(MatCreateTranspose(*sub, b + 2)); 2658811e8887SPierre Jolivet else PetscCall(MatCreateHermitianTranspose(*sub, b + 2)); 2659811e8887SPierre Jolivet } 2660811e8887SPierre Jolivet if (A01 || !A10) { 2661811e8887SPierre Jolivet b[1] = sub[0]; 2662811e8887SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)sub[0])); 266300b491fbSPierre Jolivet } 266400b491fbSPierre Jolivet PetscCall(MatDestroySubMatrices(1, &sub)); 266513044ca3SPierre Jolivet PetscCall(ISDestroy(&sorted)); 266651ea4bc8SPierre Jolivet b[3] = data->aux; 266751ea4bc8SPierre Jolivet PetscCall(MatCreateSchurComplement(subA[0], subA[1], b[1], b[2], b[3], &S)); 266813044ca3SPierre Jolivet PetscCall(MatSchurComplementSetKSP(S, ksp[0])); 266951ea4bc8SPierre Jolivet if (data->N != 1) { 267051ea4bc8SPierre Jolivet PetscCall(PCASMSetType(data->levels[0]->pc, PC_ASM_NONE)); /* "Neumann--Neumann" preconditioning with overlap and a Boolean partition of unity */ 267151ea4bc8SPierre Jolivet PetscCall(PCASMSetLocalSubdomains(data->levels[0]->pc, 1, &data->is, &loc)); 267251ea4bc8SPierre Jolivet PetscCall(PCSetFromOptions(data->levels[0]->pc)); /* action of eq. (15) of https://hal.science/hal-02343808v6/document (with a sign flip) */ 267351ea4bc8SPierre Jolivet s = data->levels[0]->pc; 267451ea4bc8SPierre Jolivet } else { 267551ea4bc8SPierre Jolivet is[0] = data->is; 267651ea4bc8SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)is[0])); 267751ea4bc8SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)b[3])); 267851ea4bc8SPierre Jolivet PetscCall(PCSetType(pc, PCASM)); /* change the type of the current PC */ 267951ea4bc8SPierre Jolivet data = nullptr; /* destroyed in the previous PCSetType(), so reset to NULL to avoid any faulty use */ 268051ea4bc8SPierre Jolivet PetscCall(PCAppendOptionsPrefix(pc, "pc_hpddm_coarse_")); /* same prefix as when using PCHPDDM with a single level */ 268151ea4bc8SPierre Jolivet PetscCall(PCASMSetLocalSubdomains(pc, 1, is, &loc)); 268251ea4bc8SPierre Jolivet PetscCall(ISDestroy(is)); 268351ea4bc8SPierre Jolivet PetscCall(ISDestroy(&loc)); 268451ea4bc8SPierre Jolivet s = pc; 268551ea4bc8SPierre Jolivet } 268651ea4bc8SPierre Jolivet PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(s, S, PETSC_TRUE)); /* the subdomain Mat is already known and the input IS of PCASMSetLocalSubdomains() is already sorted */ 268751ea4bc8SPierre Jolivet PetscTryMethod(s, "PCASMGetSubKSP_C", (PC, PetscInt *, PetscInt *, KSP **), (s, &n, nullptr, &ksp)); 268813044ca3SPierre Jolivet PetscCheck(n == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of subdomain solver %" PetscInt_FMT " != 1", n); 268913044ca3SPierre Jolivet PetscCall(KSPGetPC(ksp[0], &inner)); 269013044ca3SPierre Jolivet PetscCall(PCSetType(inner, PCSHELL)); /* compute the action of the inverse of the local Schur complement with a PCSHELL */ 269113044ca3SPierre Jolivet b[0] = subA[0]; 269213044ca3SPierre 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 */ 269351ea4bc8SPierre Jolivet if (!data) PetscCall(PetscObjectDereference((PetscObject)b[3])); 269400b491fbSPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)b[1])); 269500b491fbSPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)b[2])); 269613044ca3SPierre Jolivet PetscCall(PCCreate(PETSC_COMM_SELF, &s)); 269713044ca3SPierre Jolivet PetscCall(PCSetOptionsPrefix(s, ((PetscObject)inner)->prefix)); 269813044ca3SPierre Jolivet PetscCall(PCSetOptionsPrefix(inner, nullptr)); 269913044ca3SPierre Jolivet PetscCall(KSPSetSkipPCSetFromOptions(ksp[0], PETSC_TRUE)); 270013044ca3SPierre Jolivet PetscCall(PCSetType(s, PCLU)); 270151ea4bc8SPierre Jolivet if (PetscDefined(HAVE_MUMPS)) PetscCall(PCFactorSetMatSolverType(s, MATSOLVERMUMPS)); /* only MATSOLVERMUMPS handles MATNEST, so for the others, e.g., MATSOLVERPETSC or MATSOLVERMKL_PARDISO, convert to plain MATAIJ */ 270213044ca3SPierre Jolivet PetscCall(PCSetFromOptions(s)); 270313044ca3SPierre Jolivet PetscCall(PCFactorGetMatSolverType(s, &type)); 270413044ca3SPierre Jolivet PetscCall(PetscStrcmp(type, MATSOLVERMUMPS, &flg)); 270551ea4bc8SPierre Jolivet PetscCall(MatGetLocalSize(A11, &n, nullptr)); 270651ea4bc8SPierre Jolivet if (flg || n == 0) { 270713044ca3SPierre Jolivet PetscCall(PCSetOperators(s, N, N)); 270851ea4bc8SPierre Jolivet if (n) { 270900b491fbSPierre Jolivet PetscCall(PCFactorGetMatrix(s, b)); 271000b491fbSPierre Jolivet PetscCall(MatSetOptionsPrefix(*b, ((PetscObject)s)->prefix)); 271124ddd604SPierre Jolivet n = -1; 271224ddd604SPierre Jolivet PetscCall(PetscOptionsGetInt(nullptr, ((PetscObject)s)->prefix, "-mat_mumps_icntl_26", &n, nullptr)); 2713f45b553cSPierre Jolivet if (n == 1) { /* allocates a square MatDense of size is[1]->map->n, so one */ 2714f45b553cSPierre Jolivet PetscCall(MatNestGetISs(N, is, nullptr)); /* needs to be able to deactivate this path when dealing */ 2715f45b553cSPierre Jolivet PetscCall(MatFactorSetSchurIS(*b, is[1])); /* with a large constraint space in order to avoid OOM */ 2716f45b553cSPierre Jolivet } 271751ea4bc8SPierre Jolivet } else PetscCall(PCSetType(s, PCNONE)); /* empty local Schur complement (e.g., centralized on another process) */ 271813044ca3SPierre Jolivet } else { 271900b491fbSPierre Jolivet PetscCall(MatConvert(N, MATAIJ, MAT_INITIAL_MATRIX, b)); 272000b491fbSPierre Jolivet PetscCall(PCSetOperators(s, N, *b)); 272100b491fbSPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)*b)); 272251ea4bc8SPierre Jolivet PetscCall(PetscObjectTypeCompareAny((PetscObject)s, &flg, PCLU, PCCHOLESKY, PCILU, PCICC, PCQR, "")); 272351ea4bc8SPierre Jolivet if (flg) 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 */ 272413044ca3SPierre Jolivet } 272513044ca3SPierre Jolivet PetscCall(PetscNew(&p)); 272613044ca3SPierre Jolivet p->first = s; 272751ea4bc8SPierre Jolivet if (n != 0) PetscCall(MatCreateVecs(*b, p->second, p->second + 1)); 272851ea4bc8SPierre Jolivet else p->second[0] = p->second[1] = nullptr; 272913044ca3SPierre Jolivet PetscCall(PCShellSetContext(inner, p)); 273013044ca3SPierre Jolivet PetscCall(PCShellSetApply(inner, PCApply_Nest)); 273113044ca3SPierre Jolivet PetscCall(PCShellSetView(inner, PCView_Nest)); 273213044ca3SPierre Jolivet PetscCall(PCShellSetDestroy(inner, PCDestroy_Nest)); 273313044ca3SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)N)); 273451ea4bc8SPierre Jolivet if (!data) { 273551ea4bc8SPierre Jolivet PetscCall(MatDestroy(&S)); 273651ea4bc8SPierre Jolivet PetscCall(ISDestroy(&unsorted)); 273751ea4bc8SPierre Jolivet PetscCall(MatDestroy(&C)); 273851ea4bc8SPierre Jolivet PetscCall(ISDestroy(&uis)); 273951ea4bc8SPierre Jolivet PetscCall(PetscFree(ctx)); 274051ea4bc8SPierre Jolivet #if PetscDefined(USE_DEBUG) 274151ea4bc8SPierre Jolivet PetscCall(ISDestroy(&dis)); 274251ea4bc8SPierre Jolivet PetscCall(MatDestroy(&daux)); 274351ea4bc8SPierre Jolivet #endif 274451ea4bc8SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 274551ea4bc8SPierre Jolivet } 274613044ca3SPierre Jolivet } 2747f1580f4eSBarry Smith if (!data->levels[0]->scatter) { 2748db4a47b3SPierre Jolivet PetscCall(MatCreateVecs(P, &xin, nullptr)); 2749f1580f4eSBarry Smith if (ismatis) PetscCall(MatDestroy(&P)); 2750db4a47b3SPierre Jolivet PetscCall(VecScatterCreate(xin, data->is, data->levels[0]->D, nullptr, &data->levels[0]->scatter)); 2751f1580f4eSBarry Smith PetscCall(VecDestroy(&xin)); 2752f1580f4eSBarry Smith } 2753f1580f4eSBarry Smith if (data->levels[0]->P) { 2754f1580f4eSBarry Smith /* if the pattern is the same and PCSetUp() has previously succeeded, reuse HPDDM buffers and connectivity */ 2755371d2eb7SMartin Diehl PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(data->levels[0], !pc->setupcalled || pc->flag == DIFFERENT_NONZERO_PATTERN ? PETSC_TRUE : PETSC_FALSE)); 2756f1580f4eSBarry Smith } 2757f1580f4eSBarry Smith if (!data->levels[0]->P) data->levels[0]->P = new HPDDM::Schwarz<PetscScalar>(); 2758db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventBegin(PC_HPDDM_SetUp[0], data->levels[0]->ksp, nullptr, nullptr, nullptr)); 2759db4a47b3SPierre Jolivet else PetscCall(PetscLogEventBegin(PC_HPDDM_Strc, data->levels[0]->ksp, nullptr, nullptr, nullptr)); 2760f1580f4eSBarry Smith /* HPDDM internal data structure */ 276113044ca3SPierre Jolivet PetscCall(data->levels[0]->P->structure(loc, data->is, !ctx ? sub[0] : nullptr, ismatis ? C : data->aux, data->levels)); 2762db4a47b3SPierre Jolivet if (!data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_Strc, data->levels[0]->ksp, nullptr, nullptr, nullptr)); 2763f1580f4eSBarry Smith /* matrix pencil of the generalized eigenvalue problem on the overlap (GenEO) */ 276413044ca3SPierre Jolivet if (!ctx) { 27659bb5c669SPierre Jolivet if (data->deflation || overlap != -1) weighted = data->aux; 2766f1580f4eSBarry Smith else if (!data->B) { 2767e2b46ddfSPierre Jolivet PetscBool cmp; 2768811e8887SPierre Jolivet 2769f1580f4eSBarry Smith PetscCall(MatDuplicate(sub[0], MAT_COPY_VALUES, &weighted)); 2770e2b46ddfSPierre Jolivet PetscCall(PetscObjectTypeCompareAny((PetscObject)weighted, &cmp, MATNORMAL, MATNORMALHERMITIAN, "")); 2771e2b46ddfSPierre Jolivet if (cmp) flg = PETSC_FALSE; 2772e2b46ddfSPierre Jolivet PetscCall(MatDiagonalScale(weighted, data->levels[0]->D, data->levels[0]->D)); 27735b0bb9f2SPierre Jolivet /* neither MatDuplicate() nor MatDiagonalScale() handles the symmetry options, so propagate the options explicitly */ 2774f1580f4eSBarry Smith /* only useful for -mat_type baij -pc_hpddm_levels_1_st_pc_type cholesky (no problem with MATAIJ or MATSBAIJ) */ 2775f1580f4eSBarry Smith PetscCall(MatPropagateSymmetryOptions(sub[0], weighted)); 27766d61d5b2SPierre Jolivet if (PetscDefined(USE_DEBUG) && PetscBool3ToBool(data->Neumann)) { 27776d61d5b2SPierre Jolivet Mat *sub, A[3]; 27786d61d5b2SPierre Jolivet PetscReal norm[2]; 27796d61d5b2SPierre Jolivet PetscBool flg; 27806d61d5b2SPierre Jolivet 27816d61d5b2SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)P, MATMPISBAIJ, &flg)); /* MatCreateSubMatrices() does not work with MATSBAIJ and unsorted ISes, so convert to MPIAIJ */ 27826d61d5b2SPierre Jolivet if (flg) PetscCall(MatConvert(P, MATMPIAIJ, MAT_INITIAL_MATRIX, A)); 27836d61d5b2SPierre Jolivet else { 27846d61d5b2SPierre Jolivet A[0] = P; 27856d61d5b2SPierre Jolivet PetscCall(PetscObjectReference((PetscObject)P)); 27866d61d5b2SPierre Jolivet } 27876d61d5b2SPierre Jolivet PetscCall(MatCreateSubMatrices(A[0], 1, &data->is, &data->is, MAT_INITIAL_MATRIX, &sub)); 27886d61d5b2SPierre Jolivet PetscCall(MatDiagonalScale(sub[0], data->levels[0]->D, data->levels[0]->D)); 27896d61d5b2SPierre Jolivet PetscCall(MatConvert(sub[0], MATSEQAIJ, MAT_INITIAL_MATRIX, A + 1)); /* too many corner cases to handle (MATNORMAL, MATNORMALHERMITIAN, MATBAIJ with different block sizes...), so just MatConvert() to MATSEQAIJ since this is just for debugging */ 27906d61d5b2SPierre Jolivet PetscCall(MatConvert(weighted, MATSEQAIJ, MAT_INITIAL_MATRIX, A + 2)); 27916d61d5b2SPierre Jolivet PetscCall(MatAXPY(A[1], -1.0, A[2], UNKNOWN_NONZERO_PATTERN)); 27926d61d5b2SPierre Jolivet PetscCall(MatNorm(A[1], NORM_FROBENIUS, norm)); 27936d61d5b2SPierre Jolivet if (norm[0]) { 27946d61d5b2SPierre Jolivet PetscCall(MatNorm(A[2], NORM_FROBENIUS, norm + 1)); 27956d61d5b2SPierre Jolivet PetscCheck(PetscAbsReal(norm[0] / norm[1]) < PetscSqrtReal(PETSC_SMALL), PETSC_COMM_SELF, PETSC_ERR_USER_INPUT, "Auxiliary Mat is different from the (assembled) subdomain Mat for the interior unknowns, so it cannot be the Neumann matrix, remove -%spc_hpddm_has_neumann", pcpre ? pcpre : ""); 27966d61d5b2SPierre Jolivet } 27976d61d5b2SPierre Jolivet PetscCall(MatDestroySubMatrices(1, &sub)); 27986d61d5b2SPierre Jolivet for (PetscInt i = 0; i < 3; ++i) PetscCall(MatDestroy(A + i)); 27996d61d5b2SPierre Jolivet } 2800f1580f4eSBarry Smith } else weighted = data->B; 280113044ca3SPierre Jolivet } else weighted = nullptr; 2802f1580f4eSBarry Smith /* SLEPc is used inside the loaded symbol */ 28039bb5c669SPierre 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)); 28049bb5c669SPierre Jolivet if (!ctx && data->share && overlap == -1) { 2805f1580f4eSBarry Smith Mat st[2]; 28066d61d5b2SPierre Jolivet 2807f1580f4eSBarry Smith PetscCall(KSPGetOperators(ksp[0], st, st + 1)); 28085c7345deSPierre Jolivet PetscCall(MatCopy(subA[0], st[0], structure)); 2809f1580f4eSBarry Smith if (subA[1] != subA[0] || st[1] != st[0]) PetscCall(MatCopy(subA[1], st[1], SAME_NONZERO_PATTERN)); 28100307214fSPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)subA[0])); 2811f1580f4eSBarry Smith } 2812db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_SetUp[0], data->levels[0]->ksp, nullptr, nullptr, nullptr)); 2813f1580f4eSBarry Smith if (ismatis) PetscCall(MatISGetLocalMat(C, &N)); 2814f1580f4eSBarry Smith else N = data->aux; 281513044ca3SPierre Jolivet if (!ctx) P = sub[0]; 281613044ca3SPierre Jolivet else P = S; 2817f1580f4eSBarry Smith /* going through the grid hierarchy */ 2818f1580f4eSBarry Smith for (n = 1; n < data->N; ++n) { 2819db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventBegin(PC_HPDDM_SetUp[n], data->levels[n]->ksp, nullptr, nullptr, nullptr)); 2820f1580f4eSBarry Smith /* method composed in the loaded symbol since there, SLEPc is used as well */ 2821f1580f4eSBarry 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)); 2822db4a47b3SPierre Jolivet if (data->log_separate) PetscCall(PetscLogEventEnd(PC_HPDDM_SetUp[n], data->levels[n]->ksp, nullptr, nullptr, nullptr)); 2823f1580f4eSBarry Smith } 2824f1580f4eSBarry Smith /* reset to NULL to avoid any faulty use */ 2825db4a47b3SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)data->levels[0]->ksp, "PCHPDDMSetUp_Private_C", nullptr)); 2826db4a47b3SPierre Jolivet if (!ismatis) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_C", nullptr)); 2827f1580f4eSBarry Smith else PetscCall(PetscObjectDereference((PetscObject)C)); /* matching PetscObjectReference() above */ 2828f1580f4eSBarry Smith for (n = 0; n < data->N - 1; ++n) 2829f1580f4eSBarry Smith if (data->levels[n]->P) { 2830f1580f4eSBarry Smith /* HPDDM internal work buffers */ 2831f1580f4eSBarry Smith data->levels[n]->P->setBuffer(); 2832f1580f4eSBarry Smith data->levels[n]->P->super::start(); 2833f1580f4eSBarry Smith } 28349bb5c669SPierre Jolivet if (ismatis || !subdomains) PetscCall(PCHPDDMDestroySubMatrices_Private(PetscBool3ToBool(data->Neumann), PetscBool(algebraic && !block && overlap == -1), sub)); 2835db4a47b3SPierre Jolivet if (ismatis) data->is = nullptr; 2836f1580f4eSBarry Smith for (n = 0; n < data->N - 1 + (reused > 0); ++n) { 2837f1580f4eSBarry Smith if (data->levels[n]->P) { 2838f1580f4eSBarry Smith PC spc; 2839f1580f4eSBarry Smith 2840f1580f4eSBarry Smith /* force the PC to be PCSHELL to do the coarse grid corrections */ 2841f1580f4eSBarry Smith PetscCall(KSPSetSkipPCSetFromOptions(data->levels[n]->ksp, PETSC_TRUE)); 2842f1580f4eSBarry Smith PetscCall(KSPGetPC(data->levels[n]->ksp, &spc)); 2843f1580f4eSBarry Smith PetscCall(PCSetType(spc, PCSHELL)); 2844f1580f4eSBarry Smith PetscCall(PCShellSetContext(spc, data->levels[n])); 2845f1580f4eSBarry Smith PetscCall(PCShellSetSetUp(spc, PCSetUp_HPDDMShell)); 2846f1580f4eSBarry Smith PetscCall(PCShellSetApply(spc, PCApply_HPDDMShell)); 2847f1580f4eSBarry Smith PetscCall(PCShellSetMatApply(spc, PCMatApply_HPDDMShell)); 2848d4f06b61SRaphael Zanella PetscCall(PCShellSetApplyTranspose(spc, PCApplyTranspose_HPDDMShell)); 28498cb7430dSRaphael Zanella PetscCall(PCShellSetMatApplyTranspose(spc, PCMatApplyTranspose_HPDDMShell)); 285013044ca3SPierre Jolivet if (ctx && n == 0) { 285113044ca3SPierre Jolivet Mat Amat, Pmat; 285213044ca3SPierre Jolivet PetscInt m, M; 285301e3c840SPierre Jolivet std::tuple<Mat, PetscSF, Vec[2]> *ctx; 285413044ca3SPierre Jolivet 285513044ca3SPierre Jolivet PetscCall(KSPGetOperators(data->levels[n]->ksp, nullptr, &Pmat)); 285613044ca3SPierre Jolivet PetscCall(MatGetLocalSize(Pmat, &m, nullptr)); 285713044ca3SPierre Jolivet PetscCall(MatGetSize(Pmat, &M, nullptr)); 285813044ca3SPierre Jolivet PetscCall(PetscNew(&ctx)); 285913044ca3SPierre Jolivet std::get<0>(*ctx) = S; 286013044ca3SPierre Jolivet std::get<1>(*ctx) = data->levels[n]->scatter; 286113044ca3SPierre Jolivet PetscCall(MatCreateShell(PetscObjectComm((PetscObject)data->levels[n]->ksp), m, m, M, M, ctx, &Amat)); 286257d50842SBarry Smith PetscCall(MatShellSetOperation(Amat, MATOP_MULT, (PetscErrorCodeFn *)MatMult_Schur<false>)); 286357d50842SBarry Smith PetscCall(MatShellSetOperation(Amat, MATOP_MULT_TRANSPOSE, (PetscErrorCodeFn *)MatMult_Schur<true>)); 286457d50842SBarry Smith PetscCall(MatShellSetOperation(Amat, MATOP_DESTROY, (PetscErrorCodeFn *)MatDestroy_Schur)); 286513044ca3SPierre Jolivet PetscCall(MatCreateVecs(S, std::get<2>(*ctx), std::get<2>(*ctx) + 1)); 286613044ca3SPierre Jolivet PetscCall(KSPSetOperators(data->levels[n]->ksp, Amat, Pmat)); 286713044ca3SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)Amat)); 286813044ca3SPierre Jolivet } 2869f1580f4eSBarry Smith PetscCall(PCShellSetDestroy(spc, PCDestroy_HPDDMShell)); 2870f1580f4eSBarry Smith if (!data->levels[n]->pc) PetscCall(PCCreate(PetscObjectComm((PetscObject)data->levels[n]->ksp), &data->levels[n]->pc)); 2871f1580f4eSBarry Smith if (n < reused) { 2872f1580f4eSBarry Smith PetscCall(PCSetReusePreconditioner(spc, PETSC_TRUE)); 2873f1580f4eSBarry Smith PetscCall(PCSetReusePreconditioner(data->levels[n]->pc, PETSC_TRUE)); 2874f1580f4eSBarry Smith } 2875f1580f4eSBarry Smith PetscCall(PCSetUp(spc)); 2876f1580f4eSBarry Smith } 2877f1580f4eSBarry Smith } 287813044ca3SPierre Jolivet if (ctx) PetscCall(MatDestroy(&S)); 28799bb5c669SPierre Jolivet if (overlap == -1) PetscCall(PetscObjectComposeFunction((PetscObject)pc->pmat, "PCHPDDMAlgebraicAuxiliaryMat_Private_C", nullptr)); 2880f1580f4eSBarry Smith } else flg = reused ? PETSC_FALSE : PETSC_TRUE; 2881f1580f4eSBarry Smith if (!ismatis && subdomains) { 2882f1580f4eSBarry Smith if (flg) PetscCall(KSPGetPC(data->levels[0]->ksp, &inner)); 2883f1580f4eSBarry Smith else inner = data->levels[0]->pc; 2884f1580f4eSBarry Smith if (inner) { 2885398c7888SPierre Jolivet if (!inner->setupcalled) PetscCall(PCSetType(inner, PCASM)); 2886398c7888SPierre Jolivet PetscCall(PCSetFromOptions(inner)); 2887398c7888SPierre Jolivet PetscCall(PetscStrcmp(((PetscObject)inner)->type_name, PCASM, &flg)); 2888398c7888SPierre Jolivet if (flg) { 2889f1580f4eSBarry Smith if (!inner->setupcalled) { /* evaluates to PETSC_FALSE when -pc_hpddm_block_splitting */ 2890398c7888SPierre Jolivet IS sorted; /* PCASM will sort the input IS, duplicate it to return an unmodified (PCHPDDM) input IS */ 2891811e8887SPierre Jolivet 2892398c7888SPierre Jolivet PetscCall(ISDuplicate(is[0], &sorted)); 2893398c7888SPierre Jolivet PetscCall(PCASMSetLocalSubdomains(inner, 1, &sorted, &loc)); 2894398c7888SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)sorted)); 2895398c7888SPierre Jolivet } 2896c8ea6600SPierre Jolivet if (!PetscBool3ToBool(data->Neumann) && data->N > 1) { /* subdomain matrices are already created for the eigenproblem, reuse them for the fine-level PC */ 2897db4a47b3SPierre Jolivet PetscCall(PCHPDDMPermute_Private(*is, nullptr, nullptr, sub[0], &P, nullptr)); 2898398c7888SPierre Jolivet PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(inner, P, algebraic)); 2899398c7888SPierre Jolivet PetscCall(PetscObjectDereference((PetscObject)P)); 2900f1580f4eSBarry Smith } 2901f1580f4eSBarry Smith } 2902f1580f4eSBarry Smith } 29039bb5c669SPierre Jolivet if (data->N > 1) { 29049bb5c669SPierre Jolivet if (overlap != 1) PetscCall(PCHPDDMDestroySubMatrices_Private(PetscBool3ToBool(data->Neumann), PetscBool(algebraic && !block && overlap == -1), sub)); 29059bb5c669SPierre Jolivet if (overlap == 1) PetscCall(MatDestroy(subA)); 29069bb5c669SPierre Jolivet } 2907f1580f4eSBarry Smith } 2908f1580f4eSBarry Smith PetscCall(ISDestroy(&loc)); 2909f1580f4eSBarry Smith } else data->N = 1 + reused; /* enforce this value to 1 + reused if there is no way to build another level */ 2910f1580f4eSBarry Smith if (requested != data->N + reused) { 2911f1580f4eSBarry 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, 2912f1580f4eSBarry Smith data->N, pcpre ? pcpre : "")); 2913b5a302b3SPierre Jolivet PetscCall(PetscInfo(pc, "It is best to tune parameters, e.g., a higher value for -%spc_hpddm_levels_%" PetscInt_FMT "_eps_threshold_absolute or a lower value for -%spc_hpddm_levels_%" PetscInt_FMT "_svd_threshold_relative, so that at least one local deflation vector will be selected\n", pcpre ? pcpre : "", 2914b5a302b3SPierre Jolivet data->N, pcpre ? pcpre : "", data->N)); 2915f1580f4eSBarry Smith /* cannot use PCDestroy_HPDDMShell() because PCSHELL not set for unassembled levels */ 2916f1580f4eSBarry Smith for (n = data->N - 1; n < requested - 1; ++n) { 2917f1580f4eSBarry Smith if (data->levels[n]->P) { 2918f1580f4eSBarry Smith PetscCall(HPDDM::Schwarz<PetscScalar>::destroy(data->levels[n], PETSC_TRUE)); 2919f1580f4eSBarry Smith PetscCall(VecDestroyVecs(1, &data->levels[n]->v[0])); 2920f1580f4eSBarry Smith PetscCall(VecDestroyVecs(2, &data->levels[n]->v[1])); 2921f1580f4eSBarry Smith PetscCall(MatDestroy(data->levels[n]->V)); 2922f1580f4eSBarry Smith PetscCall(MatDestroy(data->levels[n]->V + 1)); 2923f1580f4eSBarry Smith PetscCall(MatDestroy(data->levels[n]->V + 2)); 2924f1580f4eSBarry Smith PetscCall(VecDestroy(&data->levels[n]->D)); 292501e3c840SPierre Jolivet PetscCall(PetscSFDestroy(&data->levels[n]->scatter)); 2926f1580f4eSBarry Smith } 2927f1580f4eSBarry Smith } 2928f1580f4eSBarry Smith if (reused) { 2929f1580f4eSBarry Smith for (n = reused; n < PETSC_PCHPDDM_MAXLEVELS && data->levels[n]; ++n) { 2930f1580f4eSBarry Smith PetscCall(KSPDestroy(&data->levels[n]->ksp)); 2931f1580f4eSBarry Smith PetscCall(PCDestroy(&data->levels[n]->pc)); 2932f1580f4eSBarry Smith } 2933f1580f4eSBarry Smith } 2934b5a302b3SPierre Jolivet 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 or a lower value for -%spc_hpddm_levels_%" PetscInt_FMT "_svd_threshold_relative, 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, 2935b5a302b3SPierre Jolivet data->N, reused, data->N, pcpre ? pcpre : "", pcpre ? pcpre : "", data->N, pcpre ? pcpre : "", data->N); 2936f1580f4eSBarry Smith } 2937f1580f4eSBarry Smith /* these solvers are created after PCSetFromOptions() is called */ 2938f1580f4eSBarry Smith if (pc->setfromoptionscalled) { 2939f1580f4eSBarry Smith for (n = 0; n < data->N; ++n) { 2940f1580f4eSBarry Smith if (data->levels[n]->ksp) PetscCall(KSPSetFromOptions(data->levels[n]->ksp)); 2941f1580f4eSBarry Smith if (data->levels[n]->pc) PetscCall(PCSetFromOptions(data->levels[n]->pc)); 2942f1580f4eSBarry Smith } 2943f1580f4eSBarry Smith pc->setfromoptionscalled = 0; 2944f1580f4eSBarry Smith } 2945f1580f4eSBarry Smith data->N += reused; 2946f1580f4eSBarry Smith if (data->share && swap) { 2947f1580f4eSBarry Smith /* swap back pointers so that variables follow the user-provided numbering */ 2948f1580f4eSBarry Smith std::swap(C, data->aux); 2949f1580f4eSBarry Smith std::swap(uis, data->is); 2950f1580f4eSBarry Smith PetscCall(MatDestroy(&C)); 2951f1580f4eSBarry Smith PetscCall(ISDestroy(&uis)); 2952f1580f4eSBarry Smith } 2953398c7888SPierre Jolivet if (algebraic) PetscCall(MatDestroy(&data->aux)); 295438fb34a1SPierre Jolivet if (unsorted && unsorted != is[0]) { 2955398c7888SPierre Jolivet PetscCall(ISCopy(unsorted, data->is)); 2956398c7888SPierre Jolivet PetscCall(ISDestroy(&unsorted)); 2957398c7888SPierre Jolivet } 2958398c7888SPierre Jolivet #if PetscDefined(USE_DEBUG) 2959398c7888SPierre 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); 2960398c7888SPierre Jolivet if (data->is) { 2961398c7888SPierre Jolivet PetscCall(ISEqualUnsorted(data->is, dis, &flg)); 2962398c7888SPierre Jolivet PetscCall(ISDestroy(&dis)); 2963398c7888SPierre Jolivet PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Input IS and output IS are not equal"); 2964398c7888SPierre Jolivet } 2965398c7888SPierre 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); 2966398c7888SPierre Jolivet if (data->aux) { 2967398c7888SPierre Jolivet PetscCall(MatMultEqual(data->aux, daux, 20, &flg)); 2968398c7888SPierre Jolivet PetscCall(MatDestroy(&daux)); 2969398c7888SPierre Jolivet PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Input Mat and output Mat are not equal"); 2970398c7888SPierre Jolivet } 2971398c7888SPierre Jolivet #endif 29723ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2973f1580f4eSBarry Smith } 2974f1580f4eSBarry Smith 2975f1580f4eSBarry Smith /*@ 2976f1580f4eSBarry Smith PCHPDDMSetCoarseCorrectionType - Sets the coarse correction type. 2977f1580f4eSBarry Smith 2978c3339decSBarry Smith Collective 2979f1580f4eSBarry Smith 2980f1580f4eSBarry Smith Input Parameters: 2981f1580f4eSBarry Smith + pc - preconditioner context 2982aa1539e9SPierre Jolivet - type - `PC_HPDDM_COARSE_CORRECTION_DEFLATED`, `PC_HPDDM_COARSE_CORRECTION_ADDITIVE`, `PC_HPDDM_COARSE_CORRECTION_BALANCED`, or `PC_HPDDM_COARSE_CORRECTION_NONE` 2983f1580f4eSBarry Smith 2984f1580f4eSBarry Smith Options Database Key: 2985aa1539e9SPierre Jolivet . -pc_hpddm_coarse_correction <deflated, additive, balanced, none> - type of coarse correction to apply 2986f1580f4eSBarry Smith 2987f1580f4eSBarry Smith Level: intermediate 2988f1580f4eSBarry Smith 2989562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDMGetCoarseCorrectionType()`, `PCHPDDM`, `PCHPDDMCoarseCorrectionType` 2990f1580f4eSBarry Smith @*/ 2991d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetCoarseCorrectionType(PC pc, PCHPDDMCoarseCorrectionType type) 2992d71ae5a4SJacob Faibussowitsch { 2993f1580f4eSBarry Smith PetscFunctionBegin; 2994f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 2995f1580f4eSBarry Smith PetscValidLogicalCollectiveEnum(pc, type, 2); 2996f1580f4eSBarry Smith PetscTryMethod(pc, "PCHPDDMSetCoarseCorrectionType_C", (PC, PCHPDDMCoarseCorrectionType), (pc, type)); 29973ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2998f1580f4eSBarry Smith } 2999f1580f4eSBarry Smith 3000f1580f4eSBarry Smith /*@ 3001f1580f4eSBarry Smith PCHPDDMGetCoarseCorrectionType - Gets the coarse correction type. 3002f1580f4eSBarry Smith 3003f1580f4eSBarry Smith Input Parameter: 3004f1580f4eSBarry Smith . pc - preconditioner context 3005f1580f4eSBarry Smith 3006f1580f4eSBarry Smith Output Parameter: 3007aa1539e9SPierre Jolivet . type - `PC_HPDDM_COARSE_CORRECTION_DEFLATED`, `PC_HPDDM_COARSE_CORRECTION_ADDITIVE`, `PC_HPDDM_COARSE_CORRECTION_BALANCED`, or `PC_HPDDM_COARSE_CORRECTION_NONE` 3008f1580f4eSBarry Smith 3009f1580f4eSBarry Smith Level: intermediate 3010f1580f4eSBarry Smith 3011562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDMSetCoarseCorrectionType()`, `PCHPDDM`, `PCHPDDMCoarseCorrectionType` 3012f1580f4eSBarry Smith @*/ 3013d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMGetCoarseCorrectionType(PC pc, PCHPDDMCoarseCorrectionType *type) 3014d71ae5a4SJacob Faibussowitsch { 3015f1580f4eSBarry Smith PetscFunctionBegin; 3016f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 3017f1580f4eSBarry Smith if (type) { 30184f572ea9SToby Isaac PetscAssertPointer(type, 2); 3019f1580f4eSBarry Smith PetscUseMethod(pc, "PCHPDDMGetCoarseCorrectionType_C", (PC, PCHPDDMCoarseCorrectionType *), (pc, type)); 3020f1580f4eSBarry Smith } 30213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3022f1580f4eSBarry Smith } 3023f1580f4eSBarry Smith 3024d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetCoarseCorrectionType_HPDDM(PC pc, PCHPDDMCoarseCorrectionType type) 3025d71ae5a4SJacob Faibussowitsch { 3026f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 3027f1580f4eSBarry Smith 3028f1580f4eSBarry Smith PetscFunctionBegin; 3029f1580f4eSBarry Smith data->correction = type; 30303ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3031f1580f4eSBarry Smith } 3032f1580f4eSBarry Smith 3033d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMGetCoarseCorrectionType_HPDDM(PC pc, PCHPDDMCoarseCorrectionType *type) 3034d71ae5a4SJacob Faibussowitsch { 3035f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 3036f1580f4eSBarry Smith 3037f1580f4eSBarry Smith PetscFunctionBegin; 3038f1580f4eSBarry Smith *type = data->correction; 30393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3040f1580f4eSBarry Smith } 3041f1580f4eSBarry Smith 3042f1580f4eSBarry Smith /*@ 3043e31fc274SPierre Jolivet PCHPDDMSetSTShareSubKSP - Sets whether the `KSP` in SLEPc `ST` and the fine-level subdomain solver should be shared. 3044e31fc274SPierre Jolivet 3045e31fc274SPierre Jolivet Input Parameters: 3046e31fc274SPierre Jolivet + pc - preconditioner context 3047e31fc274SPierre Jolivet - share - whether the `KSP` should be shared or not 3048e31fc274SPierre Jolivet 3049e31fc274SPierre Jolivet Note: 3050e31fc274SPierre Jolivet This is not the same as `PCSetReusePreconditioner()`. Given certain conditions (visible using -info), a symbolic factorization can be skipped 3051e31fc274SPierre Jolivet when using a subdomain `PCType` such as `PCLU` or `PCCHOLESKY`. 3052e31fc274SPierre Jolivet 3053e31fc274SPierre Jolivet Level: advanced 3054e31fc274SPierre Jolivet 3055562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMGetSTShareSubKSP()` 3056e31fc274SPierre Jolivet @*/ 3057e31fc274SPierre Jolivet PetscErrorCode PCHPDDMSetSTShareSubKSP(PC pc, PetscBool share) 3058e31fc274SPierre Jolivet { 3059e31fc274SPierre Jolivet PetscFunctionBegin; 3060e31fc274SPierre Jolivet PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 3061e31fc274SPierre Jolivet PetscTryMethod(pc, "PCHPDDMSetSTShareSubKSP_C", (PC, PetscBool), (pc, share)); 30623ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3063e31fc274SPierre Jolivet } 3064e31fc274SPierre Jolivet 3065e31fc274SPierre Jolivet /*@ 3066f1580f4eSBarry Smith PCHPDDMGetSTShareSubKSP - Gets whether the `KSP` in SLEPc `ST` and the fine-level subdomain solver is shared. 3067f1580f4eSBarry Smith 3068f1580f4eSBarry Smith Input Parameter: 3069f1580f4eSBarry Smith . pc - preconditioner context 3070f1580f4eSBarry Smith 3071f1580f4eSBarry Smith Output Parameter: 3072f1580f4eSBarry Smith . share - whether the `KSP` is shared or not 3073f1580f4eSBarry Smith 3074f1580f4eSBarry Smith Note: 3075f1580f4eSBarry 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 3076f1580f4eSBarry Smith when using a subdomain `PCType` such as `PCLU` or `PCCHOLESKY`. 3077f1580f4eSBarry Smith 3078f1580f4eSBarry Smith Level: advanced 3079f1580f4eSBarry Smith 3080562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCHPDDMSetSTShareSubKSP()` 3081f1580f4eSBarry Smith @*/ 3082d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMGetSTShareSubKSP(PC pc, PetscBool *share) 3083d71ae5a4SJacob Faibussowitsch { 3084f1580f4eSBarry Smith PetscFunctionBegin; 3085f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 3086f1580f4eSBarry Smith if (share) { 30874f572ea9SToby Isaac PetscAssertPointer(share, 2); 3088f1580f4eSBarry Smith PetscUseMethod(pc, "PCHPDDMGetSTShareSubKSP_C", (PC, PetscBool *), (pc, share)); 3089f1580f4eSBarry Smith } 30903ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3091f1580f4eSBarry Smith } 3092f1580f4eSBarry Smith 3093e31fc274SPierre Jolivet static PetscErrorCode PCHPDDMSetSTShareSubKSP_HPDDM(PC pc, PetscBool share) 3094e31fc274SPierre Jolivet { 3095e31fc274SPierre Jolivet PC_HPDDM *data = (PC_HPDDM *)pc->data; 3096e31fc274SPierre Jolivet 3097e31fc274SPierre Jolivet PetscFunctionBegin; 3098e31fc274SPierre Jolivet data->share = share; 30993ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3100e31fc274SPierre Jolivet } 3101e31fc274SPierre Jolivet 3102d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMGetSTShareSubKSP_HPDDM(PC pc, PetscBool *share) 3103d71ae5a4SJacob Faibussowitsch { 3104f1580f4eSBarry Smith PC_HPDDM *data = (PC_HPDDM *)pc->data; 3105f1580f4eSBarry Smith 3106f1580f4eSBarry Smith PetscFunctionBegin; 3107f1580f4eSBarry Smith *share = data->share; 31083ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3109f1580f4eSBarry Smith } 3110f1580f4eSBarry Smith 3111f1580f4eSBarry Smith /*@ 3112f1580f4eSBarry Smith PCHPDDMSetDeflationMat - Sets the deflation space used to assemble a coarser operator. 3113f1580f4eSBarry Smith 3114f1580f4eSBarry Smith Input Parameters: 3115f1580f4eSBarry Smith + pc - preconditioner context 3116f1580f4eSBarry Smith . is - index set of the local deflation matrix 3117f1580f4eSBarry Smith - U - deflation sequential matrix stored as a `MATSEQDENSE` 3118f1580f4eSBarry Smith 3119f1580f4eSBarry Smith Level: advanced 3120f1580f4eSBarry Smith 3121562efe2eSBarry Smith .seealso: [](ch_ksp), `PCHPDDM`, `PCDeflationSetSpace()`, `PCMGSetRestriction()` 3122f1580f4eSBarry Smith @*/ 3123d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMSetDeflationMat(PC pc, IS is, Mat U) 3124d71ae5a4SJacob Faibussowitsch { 3125f1580f4eSBarry Smith PetscFunctionBegin; 3126f1580f4eSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 3127f1580f4eSBarry Smith PetscValidHeaderSpecific(is, IS_CLASSID, 2); 3128f1580f4eSBarry Smith PetscValidHeaderSpecific(U, MAT_CLASSID, 3); 3129e31fc274SPierre Jolivet PetscTryMethod(pc, "PCHPDDMSetDeflationMat_C", (PC, IS, Mat), (pc, is, U)); 31303ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3131f1580f4eSBarry Smith } 3132f1580f4eSBarry Smith 3133d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCHPDDMSetDeflationMat_HPDDM(PC pc, IS is, Mat U) 3134d71ae5a4SJacob Faibussowitsch { 3135f1580f4eSBarry Smith PetscFunctionBegin; 3136f1580f4eSBarry Smith PetscCall(PCHPDDMSetAuxiliaryMat_Private(pc, is, U, PETSC_TRUE)); 31373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3138f1580f4eSBarry Smith } 3139f1580f4eSBarry Smith 3140d71ae5a4SJacob Faibussowitsch PetscErrorCode HPDDMLoadDL_Private(PetscBool *found) 3141d71ae5a4SJacob Faibussowitsch { 3142605ad303SPierre Jolivet PetscBool flg; 3143f1580f4eSBarry Smith char lib[PETSC_MAX_PATH_LEN], dlib[PETSC_MAX_PATH_LEN], dir[PETSC_MAX_PATH_LEN]; 3144f1580f4eSBarry Smith 3145f1580f4eSBarry Smith PetscFunctionBegin; 31464f572ea9SToby Isaac PetscAssertPointer(found, 1); 3147c6a7a370SJeremy L Thompson PetscCall(PetscStrncpy(dir, "${PETSC_LIB_DIR}", sizeof(dir))); 3148db4a47b3SPierre Jolivet PetscCall(PetscOptionsGetString(nullptr, nullptr, "-hpddm_dir", dir, sizeof(dir), nullptr)); 3149f1580f4eSBarry Smith PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/libhpddm_petsc", dir)); 3150f1580f4eSBarry Smith PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found)); 3151605ad303SPierre Jolivet #if defined(SLEPC_LIB_DIR) /* this variable is passed during SLEPc ./configure when PETSc has not been configured */ 3152605ad303SPierre Jolivet if (!*found) { /* with --download-hpddm since slepcconf.h is not yet built (and thus can't be included) */ 3153c6a7a370SJeremy L Thompson PetscCall(PetscStrncpy(dir, HPDDM_STR(SLEPC_LIB_DIR), sizeof(dir))); 3154f1580f4eSBarry Smith PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/libhpddm_petsc", dir)); 3155f1580f4eSBarry Smith PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found)); 3156f1580f4eSBarry Smith } 3157f1580f4eSBarry Smith #endif 3158605ad303SPierre 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 */ 3159605ad303SPierre Jolivet PetscCall(PetscOptionsGetenv(PETSC_COMM_SELF, "SLEPC_DIR", dir, sizeof(dir), &flg)); 3160605ad303SPierre 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 */ 3161605ad303SPierre Jolivet PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/lib/libslepc", dir)); 3162605ad303SPierre Jolivet PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found)); 3163605ad303SPierre Jolivet PetscCheck(*found, PETSC_COMM_SELF, PETSC_ERR_PLIB, "%s not found but SLEPC_DIR=%s", lib, dir); 3164605ad303SPierre Jolivet PetscCall(PetscDLLibraryAppend(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, dlib)); 3165605ad303SPierre Jolivet PetscCall(PetscSNPrintf(lib, sizeof(lib), "%s/lib/libhpddm_petsc", dir)); /* libhpddm_petsc is always in the same directory as libslepc */ 3166605ad303SPierre Jolivet PetscCall(PetscDLLibraryRetrieve(PETSC_COMM_SELF, lib, dlib, 1024, found)); 3167605ad303SPierre Jolivet } 3168605ad303SPierre Jolivet } 3169f1580f4eSBarry Smith PetscCheck(*found, PETSC_COMM_SELF, PETSC_ERR_PLIB, "%s not found", lib); 3170f1580f4eSBarry Smith PetscCall(PetscDLLibraryAppend(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, dlib)); 31713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3172f1580f4eSBarry Smith } 3173f1580f4eSBarry Smith 3174f1580f4eSBarry Smith /*MC 3175f1580f4eSBarry Smith PCHPDDM - Interface with the HPDDM library. 3176f1580f4eSBarry Smith 31771d27aa22SBarry Smith This `PC` may be used to build multilevel spectral domain decomposition methods based on the GenEO framework {cite}`spillane2011robust` {cite}`al2021multilevel`. 31781d27aa22SBarry Smith It may be viewed as an alternative to spectral 31791d27aa22SBarry Smith AMGe or `PCBDDC` with adaptive selection of constraints. The interface is explained in details in {cite}`jolivetromanzampini2020` 3180f1580f4eSBarry Smith 3181e7593814SPierre 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`). 3182f1580f4eSBarry Smith 3183f1580f4eSBarry Smith For multilevel preconditioning, when using an assembled or hierarchical Pmat, one must provide an auxiliary local `Mat` (unassembled local operator for GenEO) using 3184f1580f4eSBarry Smith `PCHPDDMSetAuxiliaryMat()`. Calling this routine is not needed when using a `MATIS` Pmat, assembly is done internally using `MatConvert()`. 3185f1580f4eSBarry Smith 3186f1580f4eSBarry Smith Options Database Keys: 3187f1580f4eSBarry Smith + -pc_hpddm_define_subdomains <true, default=false> - on the finest level, calls `PCASMSetLocalSubdomains()` with the `IS` supplied in `PCHPDDMSetAuxiliaryMat()` 3188f1580f4eSBarry Smith (not relevant with an unassembled Pmat) 3189f1580f4eSBarry Smith . -pc_hpddm_has_neumann <true, default=false> - on the finest level, informs the `PC` that the local Neumann matrix is supplied in `PCHPDDMSetAuxiliaryMat()` 3190f1580f4eSBarry Smith - -pc_hpddm_coarse_correction <type, default=deflated> - determines the `PCHPDDMCoarseCorrectionType` when calling `PCApply()` 3191f1580f4eSBarry Smith 319238bf2a8cSPierre Jolivet Options for subdomain solvers, subdomain eigensolvers (for computing deflation vectors), and the coarse solver can be set using the following options database prefixes. 3193f1580f4eSBarry Smith .vb 3194f1580f4eSBarry Smith -pc_hpddm_levels_%d_pc_ 3195f1580f4eSBarry Smith -pc_hpddm_levels_%d_ksp_ 3196f1580f4eSBarry Smith -pc_hpddm_levels_%d_eps_ 3197f1580f4eSBarry Smith -pc_hpddm_levels_%d_p 31984ec2a359SPierre Jolivet -pc_hpddm_levels_%d_mat_type 3199f1580f4eSBarry Smith -pc_hpddm_coarse_ 3200f1580f4eSBarry Smith -pc_hpddm_coarse_p 32014ec2a359SPierre Jolivet -pc_hpddm_coarse_mat_type 32022ce66baaSPierre Jolivet -pc_hpddm_coarse_mat_filter 3203f1580f4eSBarry Smith .ve 3204f1580f4eSBarry Smith 320538bf2a8cSPierre 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 3206f1580f4eSBarry Smith -pc_hpddm_coarse_p 2 -pc_hpddm_coarse_mat_type baij will use 10 deflation vectors per subdomain on the fine "level 1", 3207f1580f4eSBarry Smith aggregate the fine subdomains into 4 "level 2" subdomains, then use 10 deflation vectors per subdomain on "level 2", 32087eb095acSPierre Jolivet and assemble the coarse matrix (of dimension 4 x 10 = 40) on two processes as a `MATBAIJ` (default is `MATSBAIJ`). 3209f1580f4eSBarry Smith 3210b5a302b3SPierre Jolivet 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_absolute <val>. The default -pc_hpddm_coarse_p value is 1, meaning that the coarse operator is aggregated on a single process. 3211f1580f4eSBarry Smith 32121d27aa22SBarry Smith Level: intermediate 32131d27aa22SBarry Smith 32141d27aa22SBarry Smith Notes: 32151d27aa22SBarry Smith This preconditioner requires that PETSc is built with SLEPc (``--download-slepc``). 32161d27aa22SBarry Smith 32171d27aa22SBarry Smith By default, the underlying concurrent eigenproblems 32181d27aa22SBarry Smith are solved using SLEPc shift-and-invert spectral transformation. This is usually what gives the best performance for GenEO, cf. 32191d27aa22SBarry Smith {cite}`spillane2011robust` {cite}`jolivet2013scalabledd`. As 322038bf2a8cSPierre 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 322138bf2a8cSPierre 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 322238bf2a8cSPierre Jolivet SLEPc documentation since they are specific to `PCHPDDM`. 322338bf2a8cSPierre Jolivet .vb 322438bf2a8cSPierre Jolivet -pc_hpddm_levels_1_st_share_sub_ksp 3225b5a302b3SPierre Jolivet -pc_hpddm_levels_%d_eps_threshold_absolute 322638bf2a8cSPierre Jolivet -pc_hpddm_levels_1_eps_use_inertia 322738bf2a8cSPierre Jolivet .ve 322838bf2a8cSPierre Jolivet 322938bf2a8cSPierre Jolivet The first option from the list only applies to the fine-level eigensolver, see `PCHPDDMSetSTShareSubKSP()`. The second option from the list is 323038bf2a8cSPierre Jolivet used to filter eigenmodes retrieved after convergence of `EPSSolve()` at "level N" such that eigenvectors used to define a "level N+1" coarse 3231b5a302b3SPierre Jolivet correction are associated to eigenvalues whose magnitude are lower or equal than -pc_hpddm_levels_N_eps_threshold_absolute. When using an `EPS` which cannot 323238bf2a8cSPierre 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 323338bf2a8cSPierre 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 323438bf2a8cSPierre Jolivet to supply -pc_hpddm_levels_1_eps_nev. This last option also only applies to the fine-level (N = 1) eigensolver. 3235f1580f4eSBarry Smith 32361d27aa22SBarry Smith See also {cite}`dolean2015introduction`, {cite}`al2022robust`, {cite}`al2022robustpd`, and {cite}`nataf2022recent` 3237f1580f4eSBarry Smith 3238562efe2eSBarry Smith .seealso: [](ch_ksp), `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCHPDDMSetAuxiliaryMat()`, `MATIS`, `PCBDDC`, `PCDEFLATION`, `PCTELESCOPE`, `PCASM`, 3239e31fc274SPierre Jolivet `PCHPDDMSetCoarseCorrectionType()`, `PCHPDDMHasNeumannMat()`, `PCHPDDMSetRHSMat()`, `PCHPDDMSetDeflationMat()`, `PCHPDDMSetSTShareSubKSP()`, 3240e31fc274SPierre Jolivet `PCHPDDMGetSTShareSubKSP()`, `PCHPDDMGetCoarseCorrectionType()`, `PCHPDDMGetComplexities()` 3241f1580f4eSBarry Smith M*/ 3242d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode PCCreate_HPDDM(PC pc) 3243d71ae5a4SJacob Faibussowitsch { 3244f1580f4eSBarry Smith PC_HPDDM *data; 3245f1580f4eSBarry Smith PetscBool found; 3246f1580f4eSBarry Smith 3247f1580f4eSBarry Smith PetscFunctionBegin; 3248f1580f4eSBarry Smith if (!loadedSym) { 3249f1580f4eSBarry Smith PetscCall(HPDDMLoadDL_Private(&found)); 3250db4a47b3SPierre Jolivet if (found) PetscCall(PetscDLLibrarySym(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, nullptr, "PCHPDDM_Internal", (void **)&loadedSym)); 3251f1580f4eSBarry Smith } 3252f1580f4eSBarry Smith PetscCheck(loadedSym, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCHPDDM_Internal symbol not found in loaded libhpddm_petsc"); 32534dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&data)); 3254f1580f4eSBarry Smith pc->data = data; 3255c8ea6600SPierre Jolivet data->Neumann = PETSC_BOOL3_UNKNOWN; 3256f1580f4eSBarry Smith pc->ops->reset = PCReset_HPDDM; 3257f1580f4eSBarry Smith pc->ops->destroy = PCDestroy_HPDDM; 3258f1580f4eSBarry Smith pc->ops->setfromoptions = PCSetFromOptions_HPDDM; 3259f1580f4eSBarry Smith pc->ops->setup = PCSetUp_HPDDM; 3260d4f06b61SRaphael Zanella pc->ops->apply = PCApply_HPDDM<false>; 32618cb7430dSRaphael Zanella pc->ops->matapply = PCMatApply_HPDDM<false>; 3262d4f06b61SRaphael Zanella pc->ops->applytranspose = PCApply_HPDDM<true>; 32638cb7430dSRaphael Zanella pc->ops->matapplytranspose = PCMatApply_HPDDM<true>; 3264f1580f4eSBarry Smith pc->ops->view = PCView_HPDDM; 3265f1580f4eSBarry Smith pc->ops->presolve = PCPreSolve_HPDDM; 3266f1580f4eSBarry Smith 3267f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetAuxiliaryMat_C", PCHPDDMSetAuxiliaryMat_HPDDM)); 3268f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMHasNeumannMat_C", PCHPDDMHasNeumannMat_HPDDM)); 3269f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetRHSMat_C", PCHPDDMSetRHSMat_HPDDM)); 3270f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetCoarseCorrectionType_C", PCHPDDMSetCoarseCorrectionType_HPDDM)); 3271f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetCoarseCorrectionType_C", PCHPDDMGetCoarseCorrectionType_HPDDM)); 3272e31fc274SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetSTShareSubKSP_C", PCHPDDMSetSTShareSubKSP_HPDDM)); 3273f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMGetSTShareSubKSP_C", PCHPDDMGetSTShareSubKSP_HPDDM)); 3274f1580f4eSBarry Smith PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCHPDDMSetDeflationMat_C", PCHPDDMSetDeflationMat_HPDDM)); 32753ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3276f1580f4eSBarry Smith } 3277f1580f4eSBarry Smith 3278f1580f4eSBarry Smith /*@C 3279f1580f4eSBarry Smith PCHPDDMInitializePackage - This function initializes everything in the `PCHPDDM` package. It is called from `PCInitializePackage()`. 3280f1580f4eSBarry Smith 3281f1580f4eSBarry Smith Level: developer 3282f1580f4eSBarry Smith 3283562efe2eSBarry Smith .seealso: [](ch_ksp), `PetscInitialize()` 3284f1580f4eSBarry Smith @*/ 3285d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMInitializePackage(void) 3286d71ae5a4SJacob Faibussowitsch { 3287f1580f4eSBarry Smith char ename[32]; 3288f1580f4eSBarry Smith 3289f1580f4eSBarry Smith PetscFunctionBegin; 32903ba16761SJacob Faibussowitsch if (PCHPDDMPackageInitialized) PetscFunctionReturn(PETSC_SUCCESS); 3291f1580f4eSBarry Smith PCHPDDMPackageInitialized = PETSC_TRUE; 3292f1580f4eSBarry Smith PetscCall(PetscRegisterFinalize(PCHPDDMFinalizePackage)); 3293f1580f4eSBarry Smith /* general events registered once during package initialization */ 3294f1580f4eSBarry Smith /* some of these events are not triggered in libpetsc, */ 3295f1580f4eSBarry Smith /* but rather directly in libhpddm_petsc, */ 3296f1580f4eSBarry Smith /* which is in charge of performing the following operations */ 3297f1580f4eSBarry Smith 3298f1580f4eSBarry Smith /* domain decomposition structure from Pmat sparsity pattern */ 3299f1580f4eSBarry Smith PetscCall(PetscLogEventRegister("PCHPDDMStrc", PC_CLASSID, &PC_HPDDM_Strc)); 3300f1580f4eSBarry Smith /* Galerkin product, redistribution, and setup (not triggered in libpetsc) */ 3301f1580f4eSBarry Smith PetscCall(PetscLogEventRegister("PCHPDDMPtAP", PC_CLASSID, &PC_HPDDM_PtAP)); 3302f1580f4eSBarry Smith /* Galerkin product with summation, redistribution, and setup (not triggered in libpetsc) */ 3303f1580f4eSBarry Smith PetscCall(PetscLogEventRegister("PCHPDDMPtBP", PC_CLASSID, &PC_HPDDM_PtBP)); 3304f1580f4eSBarry Smith /* next level construction using PtAP and PtBP (not triggered in libpetsc) */ 3305f1580f4eSBarry Smith PetscCall(PetscLogEventRegister("PCHPDDMNext", PC_CLASSID, &PC_HPDDM_Next)); 3306f1580f4eSBarry Smith static_assert(PETSC_PCHPDDM_MAXLEVELS <= 9, "PETSC_PCHPDDM_MAXLEVELS value is too high"); 3307811e8887SPierre Jolivet for (PetscInt i = 1; i < PETSC_PCHPDDM_MAXLEVELS; ++i) { 3308f1580f4eSBarry Smith PetscCall(PetscSNPrintf(ename, sizeof(ename), "PCHPDDMSetUp L%1" PetscInt_FMT, i)); 3309f1580f4eSBarry Smith /* events during a PCSetUp() at level #i _except_ the assembly */ 3310f1580f4eSBarry Smith /* of the Galerkin operator of the coarser level #(i + 1) */ 3311f1580f4eSBarry Smith PetscCall(PetscLogEventRegister(ename, PC_CLASSID, &PC_HPDDM_SetUp[i - 1])); 3312f1580f4eSBarry Smith PetscCall(PetscSNPrintf(ename, sizeof(ename), "PCHPDDMSolve L%1" PetscInt_FMT, i)); 3313f1580f4eSBarry Smith /* events during a PCApply() at level #i _except_ */ 3314f1580f4eSBarry Smith /* the KSPSolve() of the coarser level #(i + 1) */ 3315f1580f4eSBarry Smith PetscCall(PetscLogEventRegister(ename, PC_CLASSID, &PC_HPDDM_Solve[i - 1])); 3316f1580f4eSBarry Smith } 33173ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3318f1580f4eSBarry Smith } 3319f1580f4eSBarry Smith 3320f1580f4eSBarry Smith /*@C 3321f1580f4eSBarry Smith PCHPDDMFinalizePackage - This function frees everything from the `PCHPDDM` package. It is called from `PetscFinalize()`. 3322f1580f4eSBarry Smith 3323f1580f4eSBarry Smith Level: developer 3324f1580f4eSBarry Smith 3325562efe2eSBarry Smith .seealso: [](ch_ksp), `PetscFinalize()` 3326f1580f4eSBarry Smith @*/ 3327d71ae5a4SJacob Faibussowitsch PetscErrorCode PCHPDDMFinalizePackage(void) 3328d71ae5a4SJacob Faibussowitsch { 3329f1580f4eSBarry Smith PetscFunctionBegin; 3330f1580f4eSBarry Smith PCHPDDMPackageInitialized = PETSC_FALSE; 33313ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3332f1580f4eSBarry Smith } 33339bb5c669SPierre Jolivet 33349bb5c669SPierre Jolivet static PetscErrorCode MatMult_Harmonic(Mat A, Vec x, Vec y) 33359bb5c669SPierre Jolivet { 33369bb5c669SPierre Jolivet Harmonic h; /* [ A_00 A_01 ], furthermore, A_00 = [ A_loc,loc A_loc,ovl ], thus, A_01 = [ ] */ 33379bb5c669SPierre Jolivet /* [ A_10 A_11 A_12 ] [ A_ovl,loc A_ovl,ovl ] [ A_ovl,1 ] */ 33389bb5c669SPierre Jolivet Vec sub; /* y = A x = R_loc R_0 [ A_00 A_01 ]^-1 R_loc = [ I_loc ] */ 33399bb5c669SPierre Jolivet /* [ A_10 A_11 ] R_1^T A_12 x [ ] */ 33409bb5c669SPierre Jolivet PetscFunctionBegin; 33419bb5c669SPierre Jolivet PetscCall(MatShellGetContext(A, &h)); 33429bb5c669SPierre Jolivet PetscCall(VecSet(h->v, 0.0)); 33439bb5c669SPierre Jolivet PetscCall(VecGetSubVector(h->v, h->is[0], &sub)); 33449bb5c669SPierre Jolivet PetscCall(MatMult(h->A[0], x, sub)); 33459bb5c669SPierre Jolivet PetscCall(VecRestoreSubVector(h->v, h->is[0], &sub)); 33469bb5c669SPierre Jolivet PetscCall(KSPSolve(h->ksp, h->v, h->v)); 33479bb5c669SPierre Jolivet PetscCall(VecISCopy(h->v, h->is[1], SCATTER_REVERSE, y)); 33489bb5c669SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 33499bb5c669SPierre Jolivet } 33509bb5c669SPierre Jolivet 33519bb5c669SPierre Jolivet static PetscErrorCode MatMultTranspose_Harmonic(Mat A, Vec y, Vec x) 33529bb5c669SPierre Jolivet { 33539bb5c669SPierre Jolivet Harmonic h; /* x = A^T y = [ A_00 A_01 ]^-T R_0^T R_loc^T y */ 33549bb5c669SPierre Jolivet Vec sub; /* A_12^T R_1 [ A_10 A_11 ] */ 33559bb5c669SPierre Jolivet 33569bb5c669SPierre Jolivet PetscFunctionBegin; 33579bb5c669SPierre Jolivet PetscCall(MatShellGetContext(A, &h)); 33589bb5c669SPierre Jolivet PetscCall(VecSet(h->v, 0.0)); 33599bb5c669SPierre Jolivet PetscCall(VecISCopy(h->v, h->is[1], SCATTER_FORWARD, y)); 33609bb5c669SPierre Jolivet PetscCall(KSPSolveTranspose(h->ksp, h->v, h->v)); 33619bb5c669SPierre Jolivet PetscCall(VecGetSubVector(h->v, h->is[0], &sub)); 33629bb5c669SPierre Jolivet PetscCall(MatMultTranspose(h->A[0], sub, x)); 33639bb5c669SPierre Jolivet PetscCall(VecRestoreSubVector(h->v, h->is[0], &sub)); 33649bb5c669SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 33659bb5c669SPierre Jolivet } 33669bb5c669SPierre Jolivet 33679bb5c669SPierre Jolivet static PetscErrorCode MatProduct_AB_Harmonic(Mat S, Mat X, Mat Y, void *) 33689bb5c669SPierre Jolivet { 33699bb5c669SPierre Jolivet Harmonic h; 33709bb5c669SPierre Jolivet Mat A, B; 33719bb5c669SPierre Jolivet Vec a, b; 33729bb5c669SPierre Jolivet 33739bb5c669SPierre Jolivet PetscFunctionBegin; 33749bb5c669SPierre Jolivet PetscCall(MatShellGetContext(S, &h)); 3375fb842aefSJose E. Roman PetscCall(MatMatMult(h->A[0], X, MAT_INITIAL_MATRIX, PETSC_CURRENT, &A)); 33769bb5c669SPierre Jolivet PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, h->ksp->pc->mat->rmap->n, A->cmap->n, nullptr, &B)); 33779bb5c669SPierre Jolivet for (PetscInt i = 0; i < A->cmap->n; ++i) { 33789bb5c669SPierre Jolivet PetscCall(MatDenseGetColumnVecRead(A, i, &a)); 33799bb5c669SPierre Jolivet PetscCall(MatDenseGetColumnVecWrite(B, i, &b)); 33809bb5c669SPierre Jolivet PetscCall(VecISCopy(b, h->is[0], SCATTER_FORWARD, a)); 33819bb5c669SPierre Jolivet PetscCall(MatDenseRestoreColumnVecWrite(B, i, &b)); 33829bb5c669SPierre Jolivet PetscCall(MatDenseRestoreColumnVecRead(A, i, &a)); 33839bb5c669SPierre Jolivet } 33849bb5c669SPierre Jolivet PetscCall(MatDestroy(&A)); 33859bb5c669SPierre Jolivet PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, h->ksp->pc->mat->rmap->n, B->cmap->n, nullptr, &A)); 33869bb5c669SPierre Jolivet PetscCall(KSPMatSolve(h->ksp, B, A)); 33879bb5c669SPierre Jolivet PetscCall(MatDestroy(&B)); 33889bb5c669SPierre Jolivet for (PetscInt i = 0; i < A->cmap->n; ++i) { 33899bb5c669SPierre Jolivet PetscCall(MatDenseGetColumnVecRead(A, i, &a)); 33909bb5c669SPierre Jolivet PetscCall(MatDenseGetColumnVecWrite(Y, i, &b)); 33919bb5c669SPierre Jolivet PetscCall(VecISCopy(a, h->is[1], SCATTER_REVERSE, b)); 33929bb5c669SPierre Jolivet PetscCall(MatDenseRestoreColumnVecWrite(Y, i, &b)); 33939bb5c669SPierre Jolivet PetscCall(MatDenseRestoreColumnVecRead(A, i, &a)); 33949bb5c669SPierre Jolivet } 33959bb5c669SPierre Jolivet PetscCall(MatDestroy(&A)); 33969bb5c669SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 33979bb5c669SPierre Jolivet } 33989bb5c669SPierre Jolivet 3399f21e3f8aSPierre Jolivet static PetscErrorCode MatProduct_AtB_Harmonic(Mat S, Mat Y, Mat X, void *) 3400f21e3f8aSPierre Jolivet { 3401f21e3f8aSPierre Jolivet Harmonic h; 3402f21e3f8aSPierre Jolivet Mat A, B; 3403f21e3f8aSPierre Jolivet Vec a, b; 3404f21e3f8aSPierre Jolivet 3405f21e3f8aSPierre Jolivet PetscFunctionBegin; 3406f21e3f8aSPierre Jolivet PetscCall(MatShellGetContext(S, &h)); 3407f21e3f8aSPierre Jolivet PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, h->ksp->pc->mat->rmap->n, Y->cmap->n, nullptr, &A)); 3408f21e3f8aSPierre Jolivet for (PetscInt i = 0; i < A->cmap->n; ++i) { 3409f21e3f8aSPierre Jolivet PetscCall(MatDenseGetColumnVecRead(Y, i, &b)); 3410f21e3f8aSPierre Jolivet PetscCall(MatDenseGetColumnVecWrite(A, i, &a)); 3411f21e3f8aSPierre Jolivet PetscCall(VecISCopy(a, h->is[1], SCATTER_FORWARD, b)); 3412f21e3f8aSPierre Jolivet PetscCall(MatDenseRestoreColumnVecWrite(A, i, &a)); 3413f21e3f8aSPierre Jolivet PetscCall(MatDenseRestoreColumnVecRead(Y, i, &b)); 3414f21e3f8aSPierre Jolivet } 3415f21e3f8aSPierre Jolivet PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, h->ksp->pc->mat->rmap->n, A->cmap->n, nullptr, &B)); 3416f21e3f8aSPierre Jolivet PetscCall(KSPMatSolveTranspose(h->ksp, A, B)); 3417f21e3f8aSPierre Jolivet PetscCall(MatDestroy(&A)); 3418f21e3f8aSPierre Jolivet PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, h->A[0]->rmap->n, B->cmap->n, nullptr, &A)); 3419f21e3f8aSPierre Jolivet for (PetscInt i = 0; i < A->cmap->n; ++i) { 3420f21e3f8aSPierre Jolivet PetscCall(MatDenseGetColumnVecRead(B, i, &b)); 3421f21e3f8aSPierre Jolivet PetscCall(MatDenseGetColumnVecWrite(A, i, &a)); 3422f21e3f8aSPierre Jolivet PetscCall(VecISCopy(b, h->is[0], SCATTER_REVERSE, a)); 3423f21e3f8aSPierre Jolivet PetscCall(MatDenseRestoreColumnVecWrite(A, i, &a)); 3424f21e3f8aSPierre Jolivet PetscCall(MatDenseRestoreColumnVecRead(B, i, &b)); 3425f21e3f8aSPierre Jolivet } 3426f21e3f8aSPierre Jolivet PetscCall(MatDestroy(&B)); 3427f21e3f8aSPierre Jolivet PetscCall(MatTransposeMatMult(h->A[0], A, MAT_REUSE_MATRIX, PETSC_CURRENT, &X)); 3428f21e3f8aSPierre Jolivet PetscCall(MatDestroy(&A)); 3429f21e3f8aSPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 3430f21e3f8aSPierre Jolivet } 3431f21e3f8aSPierre Jolivet 34329bb5c669SPierre Jolivet static PetscErrorCode MatDestroy_Harmonic(Mat A) 34339bb5c669SPierre Jolivet { 34349bb5c669SPierre Jolivet Harmonic h; 34359bb5c669SPierre Jolivet 34369bb5c669SPierre Jolivet PetscFunctionBegin; 34379bb5c669SPierre Jolivet PetscCall(MatShellGetContext(A, &h)); 343832603206SJames Wright for (PetscInt i = 0; i < 5; ++i) PetscCall(ISDestroy(h->is + i)); 34399bb5c669SPierre Jolivet PetscCall(PetscFree(h->is)); 34409bb5c669SPierre Jolivet PetscCall(VecDestroy(&h->v)); 34419bb5c669SPierre Jolivet for (PetscInt i = 0; i < 2; ++i) PetscCall(MatDestroy(h->A + i)); 34429bb5c669SPierre Jolivet PetscCall(PetscFree(h->A)); 34439bb5c669SPierre Jolivet PetscCall(KSPDestroy(&h->ksp)); 34449bb5c669SPierre Jolivet PetscCall(PetscFree(h)); 34459bb5c669SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 34469bb5c669SPierre Jolivet } 3447