1dba47a55SKris Buschelman 24b9ad928SBarry Smith /* 34b9ad928SBarry Smith Defines the multigrid preconditioner interface. 44b9ad928SBarry Smith */ 5af0996ceSBarry Smith #include <petsc/private/pcmgimpl.h> /*I "petscksp.h" I*/ 641b6fd38SMatthew G. Knepley #include <petsc/private/kspimpl.h> 71e25c274SJed Brown #include <petscdm.h> 8391689abSStefano Zampini PETSC_INTERN PetscErrorCode PCPreSolveChangeRHS(PC, PetscBool *); 94b9ad928SBarry Smith 10f3b08a26SMatthew G. Knepley /* 11f3b08a26SMatthew G. Knepley Contains the list of registered coarse space construction routines 12f3b08a26SMatthew G. Knepley */ 13f3b08a26SMatthew G. Knepley PetscFunctionList PCMGCoarseList = NULL; 14f3b08a26SMatthew G. Knepley 15d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGMCycle_Private(PC pc, PC_MG_Levels **mglevelsin, PetscBool transpose, PetscBool matapp, PCRichardsonConvergedReason *reason) 16d71ae5a4SJacob Faibussowitsch { 1731567311SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 1831567311SBarry Smith PC_MG_Levels *mgc, *mglevels = *mglevelsin; 1931567311SBarry Smith PetscInt cycles = (mglevels->level == 1) ? 1 : (PetscInt)mglevels->cycles; 204b9ad928SBarry Smith 214b9ad928SBarry Smith PetscFunctionBegin; 229566063dSJacob Faibussowitsch if (mglevels->eventsmoothsolve) PetscCall(PetscLogEventBegin(mglevels->eventsmoothsolve, 0, 0, 0, 0)); 23fcb023d4SJed Brown if (!transpose) { 2430b0564aSStefano Zampini if (matapp) { 259566063dSJacob Faibussowitsch PetscCall(KSPMatSolve(mglevels->smoothd, mglevels->B, mglevels->X)); /* pre-smooth */ 269566063dSJacob Faibussowitsch PetscCall(KSPCheckSolve(mglevels->smoothd, pc, NULL)); 2730b0564aSStefano Zampini } else { 289566063dSJacob Faibussowitsch PetscCall(KSPSolve(mglevels->smoothd, mglevels->b, mglevels->x)); /* pre-smooth */ 299566063dSJacob Faibussowitsch PetscCall(KSPCheckSolve(mglevels->smoothd, pc, mglevels->x)); 3030b0564aSStefano Zampini } 31fcb023d4SJed Brown } else { 3228b400f6SJacob Faibussowitsch PetscCheck(!matapp, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Not supported"); 339566063dSJacob Faibussowitsch PetscCall(KSPSolveTranspose(mglevels->smoothu, mglevels->b, mglevels->x)); /* transpose of post-smooth */ 349566063dSJacob Faibussowitsch PetscCall(KSPCheckSolve(mglevels->smoothu, pc, mglevels->x)); 35fcb023d4SJed Brown } 369566063dSJacob Faibussowitsch if (mglevels->eventsmoothsolve) PetscCall(PetscLogEventEnd(mglevels->eventsmoothsolve, 0, 0, 0, 0)); 3731567311SBarry Smith if (mglevels->level) { /* not the coarsest grid */ 389566063dSJacob Faibussowitsch if (mglevels->eventresidual) PetscCall(PetscLogEventBegin(mglevels->eventresidual, 0, 0, 0, 0)); 3948a46eb9SPierre Jolivet if (matapp && !mglevels->R) PetscCall(MatDuplicate(mglevels->B, MAT_DO_NOT_COPY_VALUES, &mglevels->R)); 40fcb023d4SJed Brown if (!transpose) { 419566063dSJacob Faibussowitsch if (matapp) PetscCall((*mglevels->matresidual)(mglevels->A, mglevels->B, mglevels->X, mglevels->R)); 429566063dSJacob Faibussowitsch else PetscCall((*mglevels->residual)(mglevels->A, mglevels->b, mglevels->x, mglevels->r)); 43fcb023d4SJed Brown } else { 449566063dSJacob Faibussowitsch if (matapp) PetscCall((*mglevels->matresidualtranspose)(mglevels->A, mglevels->B, mglevels->X, mglevels->R)); 459566063dSJacob Faibussowitsch else PetscCall((*mglevels->residualtranspose)(mglevels->A, mglevels->b, mglevels->x, mglevels->r)); 46fcb023d4SJed Brown } 479566063dSJacob Faibussowitsch if (mglevels->eventresidual) PetscCall(PetscLogEventEnd(mglevels->eventresidual, 0, 0, 0, 0)); 484b9ad928SBarry Smith 494b9ad928SBarry Smith /* if on finest level and have convergence criteria set */ 5031567311SBarry Smith if (mglevels->level == mglevels->levels - 1 && mg->ttol && reason) { 514b9ad928SBarry Smith PetscReal rnorm; 529566063dSJacob Faibussowitsch PetscCall(VecNorm(mglevels->r, NORM_2, &rnorm)); 534b9ad928SBarry Smith if (rnorm <= mg->ttol) { 5470441072SBarry Smith if (rnorm < mg->abstol) { 554d0a8057SBarry Smith *reason = PCRICHARDSON_CONVERGED_ATOL; 569566063dSJacob Faibussowitsch PetscCall(PetscInfo(pc, "Linear solver has converged. Residual norm %g is less than absolute tolerance %g\n", (double)rnorm, (double)mg->abstol)); 574b9ad928SBarry Smith } else { 584d0a8057SBarry Smith *reason = PCRICHARDSON_CONVERGED_RTOL; 599566063dSJacob Faibussowitsch PetscCall(PetscInfo(pc, "Linear solver has converged. Residual norm %g is less than relative tolerance times initial residual norm %g\n", (double)rnorm, (double)mg->ttol)); 604b9ad928SBarry Smith } 613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 624b9ad928SBarry Smith } 634b9ad928SBarry Smith } 644b9ad928SBarry Smith 6531567311SBarry Smith mgc = *(mglevelsin - 1); 669566063dSJacob Faibussowitsch if (mglevels->eventinterprestrict) PetscCall(PetscLogEventBegin(mglevels->eventinterprestrict, 0, 0, 0, 0)); 67fcb023d4SJed Brown if (!transpose) { 689566063dSJacob Faibussowitsch if (matapp) PetscCall(MatMatRestrict(mglevels->restrct, mglevels->R, &mgc->B)); 699566063dSJacob Faibussowitsch else PetscCall(MatRestrict(mglevels->restrct, mglevels->r, mgc->b)); 70fcb023d4SJed Brown } else { 719566063dSJacob Faibussowitsch if (matapp) PetscCall(MatMatRestrict(mglevels->interpolate, mglevels->R, &mgc->B)); 729566063dSJacob Faibussowitsch else PetscCall(MatRestrict(mglevels->interpolate, mglevels->r, mgc->b)); 73fcb023d4SJed Brown } 749566063dSJacob Faibussowitsch if (mglevels->eventinterprestrict) PetscCall(PetscLogEventEnd(mglevels->eventinterprestrict, 0, 0, 0, 0)); 7530b0564aSStefano Zampini if (matapp) { 7630b0564aSStefano Zampini if (!mgc->X) { 779566063dSJacob Faibussowitsch PetscCall(MatDuplicate(mgc->B, MAT_DO_NOT_COPY_VALUES, &mgc->X)); 7830b0564aSStefano Zampini } else { 799566063dSJacob Faibussowitsch PetscCall(MatZeroEntries(mgc->X)); 8030b0564aSStefano Zampini } 8130b0564aSStefano Zampini } else { 829566063dSJacob Faibussowitsch PetscCall(VecZeroEntries(mgc->x)); 8330b0564aSStefano Zampini } 8448a46eb9SPierre Jolivet while (cycles--) PetscCall(PCMGMCycle_Private(pc, mglevelsin - 1, transpose, matapp, reason)); 859566063dSJacob Faibussowitsch if (mglevels->eventinterprestrict) PetscCall(PetscLogEventBegin(mglevels->eventinterprestrict, 0, 0, 0, 0)); 86fcb023d4SJed Brown if (!transpose) { 879566063dSJacob Faibussowitsch if (matapp) PetscCall(MatMatInterpolateAdd(mglevels->interpolate, mgc->X, mglevels->X, &mglevels->X)); 889566063dSJacob Faibussowitsch else PetscCall(MatInterpolateAdd(mglevels->interpolate, mgc->x, mglevels->x, mglevels->x)); 89fcb023d4SJed Brown } else { 909566063dSJacob Faibussowitsch PetscCall(MatInterpolateAdd(mglevels->restrct, mgc->x, mglevels->x, mglevels->x)); 91fcb023d4SJed Brown } 929566063dSJacob Faibussowitsch if (mglevels->eventinterprestrict) PetscCall(PetscLogEventEnd(mglevels->eventinterprestrict, 0, 0, 0, 0)); 939566063dSJacob Faibussowitsch if (mglevels->eventsmoothsolve) PetscCall(PetscLogEventBegin(mglevels->eventsmoothsolve, 0, 0, 0, 0)); 94fcb023d4SJed Brown if (!transpose) { 9530b0564aSStefano Zampini if (matapp) { 969566063dSJacob Faibussowitsch PetscCall(KSPMatSolve(mglevels->smoothu, mglevels->B, mglevels->X)); /* post smooth */ 979566063dSJacob Faibussowitsch PetscCall(KSPCheckSolve(mglevels->smoothu, pc, NULL)); 9830b0564aSStefano Zampini } else { 999566063dSJacob Faibussowitsch PetscCall(KSPSolve(mglevels->smoothu, mglevels->b, mglevels->x)); /* post smooth */ 1009566063dSJacob Faibussowitsch PetscCall(KSPCheckSolve(mglevels->smoothu, pc, mglevels->x)); 10130b0564aSStefano Zampini } 102fcb023d4SJed Brown } else { 10328b400f6SJacob Faibussowitsch PetscCheck(!matapp, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Not supported"); 1049566063dSJacob Faibussowitsch PetscCall(KSPSolveTranspose(mglevels->smoothd, mglevels->b, mglevels->x)); /* post smooth */ 1059566063dSJacob Faibussowitsch PetscCall(KSPCheckSolve(mglevels->smoothd, pc, mglevels->x)); 106fcb023d4SJed Brown } 10741b6fd38SMatthew G. Knepley if (mglevels->cr) { 10828b400f6SJacob Faibussowitsch PetscCheck(!matapp, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Not supported"); 10941b6fd38SMatthew G. Knepley /* TODO Turn on copy and turn off noisy if we have an exact solution 1109566063dSJacob Faibussowitsch PetscCall(VecCopy(mglevels->x, mglevels->crx)); 1119566063dSJacob Faibussowitsch PetscCall(VecCopy(mglevels->b, mglevels->crb)); */ 1129566063dSJacob Faibussowitsch PetscCall(KSPSetNoisy_Private(mglevels->crx)); 1139566063dSJacob Faibussowitsch PetscCall(KSPSolve(mglevels->cr, mglevels->crb, mglevels->crx)); /* compatible relaxation */ 1149566063dSJacob Faibussowitsch PetscCall(KSPCheckSolve(mglevels->cr, pc, mglevels->crx)); 11541b6fd38SMatthew G. Knepley } 1169566063dSJacob Faibussowitsch if (mglevels->eventsmoothsolve) PetscCall(PetscLogEventEnd(mglevels->eventsmoothsolve, 0, 0, 0, 0)); 1174b9ad928SBarry Smith } 1183ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1194b9ad928SBarry Smith } 1204b9ad928SBarry Smith 121d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApplyRichardson_MG(PC pc, Vec b, Vec x, Vec w, PetscReal rtol, PetscReal abstol, PetscReal dtol, PetscInt its, PetscBool zeroguess, PetscInt *outits, PCRichardsonConvergedReason *reason) 122d71ae5a4SJacob Faibussowitsch { 123f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 124f3fbd535SBarry Smith PC_MG_Levels **mglevels = mg->levels; 125391689abSStefano Zampini PC tpc; 126391689abSStefano Zampini PetscBool changeu, changed; 127f3fbd535SBarry Smith PetscInt levels = mglevels[0]->levels, i; 1284b9ad928SBarry Smith 1294b9ad928SBarry Smith PetscFunctionBegin; 130a762d673SBarry Smith /* When the DM is supplying the matrix then it will not exist until here */ 131a762d673SBarry Smith for (i = 0; i < levels; i++) { 132a762d673SBarry Smith if (!mglevels[i]->A) { 1339566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[i]->smoothu, &mglevels[i]->A, NULL)); 1349566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)mglevels[i]->A)); 135a762d673SBarry Smith } 136a762d673SBarry Smith } 137391689abSStefano Zampini 1389566063dSJacob Faibussowitsch PetscCall(KSPGetPC(mglevels[levels - 1]->smoothd, &tpc)); 1399566063dSJacob Faibussowitsch PetscCall(PCPreSolveChangeRHS(tpc, &changed)); 1409566063dSJacob Faibussowitsch PetscCall(KSPGetPC(mglevels[levels - 1]->smoothu, &tpc)); 1419566063dSJacob Faibussowitsch PetscCall(PCPreSolveChangeRHS(tpc, &changeu)); 142391689abSStefano Zampini if (!changed && !changeu) { 1439566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[levels - 1]->b)); 144f3fbd535SBarry Smith mglevels[levels - 1]->b = b; 145391689abSStefano Zampini } else { /* if the smoother changes the rhs during PreSolve, we cannot use the input vector */ 146391689abSStefano Zampini if (!mglevels[levels - 1]->b) { 147391689abSStefano Zampini Vec *vec; 148391689abSStefano Zampini 1499566063dSJacob Faibussowitsch PetscCall(KSPCreateVecs(mglevels[levels - 1]->smoothd, 1, &vec, 0, NULL)); 150391689abSStefano Zampini mglevels[levels - 1]->b = *vec; 1519566063dSJacob Faibussowitsch PetscCall(PetscFree(vec)); 152391689abSStefano Zampini } 1539566063dSJacob Faibussowitsch PetscCall(VecCopy(b, mglevels[levels - 1]->b)); 154391689abSStefano Zampini } 155f3fbd535SBarry Smith mglevels[levels - 1]->x = x; 1564b9ad928SBarry Smith 15731567311SBarry Smith mg->rtol = rtol; 15831567311SBarry Smith mg->abstol = abstol; 15931567311SBarry Smith mg->dtol = dtol; 1604b9ad928SBarry Smith if (rtol) { 1614b9ad928SBarry Smith /* compute initial residual norm for relative convergence test */ 1624b9ad928SBarry Smith PetscReal rnorm; 1637319c654SBarry Smith if (zeroguess) { 1649566063dSJacob Faibussowitsch PetscCall(VecNorm(b, NORM_2, &rnorm)); 1657319c654SBarry Smith } else { 1669566063dSJacob Faibussowitsch PetscCall((*mglevels[levels - 1]->residual)(mglevels[levels - 1]->A, b, x, w)); 1679566063dSJacob Faibussowitsch PetscCall(VecNorm(w, NORM_2, &rnorm)); 1687319c654SBarry Smith } 16931567311SBarry Smith mg->ttol = PetscMax(rtol * rnorm, abstol); 1702fa5cd67SKarl Rupp } else if (abstol) mg->ttol = abstol; 1712fa5cd67SKarl Rupp else mg->ttol = 0.0; 1724b9ad928SBarry Smith 1734d0a8057SBarry Smith /* since smoother is applied to full system, not just residual we need to make sure that smoothers don't 174336babb1SJed Brown stop prematurely due to small residual */ 1754d0a8057SBarry Smith for (i = 1; i < levels; i++) { 1769566063dSJacob Faibussowitsch PetscCall(KSPSetTolerances(mglevels[i]->smoothu, 0, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT)); 177f3fbd535SBarry Smith if (mglevels[i]->smoothu != mglevels[i]->smoothd) { 17823067569SBarry Smith /* For Richardson the initial guess is nonzero since it is solving in each cycle the original system not just applying as a preconditioner */ 1799566063dSJacob Faibussowitsch PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->smoothd, PETSC_TRUE)); 1809566063dSJacob Faibussowitsch PetscCall(KSPSetTolerances(mglevels[i]->smoothd, 0, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT)); 1814b9ad928SBarry Smith } 1824d0a8057SBarry Smith } 1834d0a8057SBarry Smith 1844d0a8057SBarry Smith *reason = (PCRichardsonConvergedReason)0; 1854d0a8057SBarry Smith for (i = 0; i < its; i++) { 1869566063dSJacob Faibussowitsch PetscCall(PCMGMCycle_Private(pc, mglevels + levels - 1, PETSC_FALSE, PETSC_FALSE, reason)); 1874d0a8057SBarry Smith if (*reason) break; 1884d0a8057SBarry Smith } 1894d0a8057SBarry Smith if (!*reason) *reason = PCRICHARDSON_CONVERGED_ITS; 1904d0a8057SBarry Smith *outits = i; 191391689abSStefano Zampini if (!changed && !changeu) mglevels[levels - 1]->b = NULL; 1923ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1934b9ad928SBarry Smith } 1944b9ad928SBarry Smith 195d71ae5a4SJacob Faibussowitsch PetscErrorCode PCReset_MG(PC pc) 196d71ae5a4SJacob Faibussowitsch { 1973aeaf226SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 1983aeaf226SBarry Smith PC_MG_Levels **mglevels = mg->levels; 1992b3cbbdaSStefano Zampini PetscInt i, n; 2003aeaf226SBarry Smith 2013aeaf226SBarry Smith PetscFunctionBegin; 2023aeaf226SBarry Smith if (mglevels) { 2033aeaf226SBarry Smith n = mglevels[0]->levels; 2043aeaf226SBarry Smith for (i = 0; i < n - 1; i++) { 2059566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[i + 1]->r)); 2069566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[i]->b)); 2079566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[i]->x)); 2089566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i + 1]->R)); 2099566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i]->B)); 2109566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i]->X)); 2119566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[i]->crx)); 2129566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[i]->crb)); 2139566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i + 1]->restrct)); 2149566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i + 1]->interpolate)); 2159566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i + 1]->inject)); 2169566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[i + 1]->rscale)); 2173aeaf226SBarry Smith } 2189566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[n - 1]->crx)); 2199566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[n - 1]->crb)); 220391689abSStefano Zampini /* this is not null only if the smoother on the finest level 221391689abSStefano Zampini changes the rhs during PreSolve */ 2229566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[n - 1]->b)); 2239566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[n - 1]->B)); 2243aeaf226SBarry Smith 2253aeaf226SBarry Smith for (i = 0; i < n; i++) { 2262b3cbbdaSStefano Zampini PetscCall(MatDestroy(&mglevels[i]->coarseSpace)); 2279566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i]->A)); 22848a46eb9SPierre Jolivet if (mglevels[i]->smoothd != mglevels[i]->smoothu) PetscCall(KSPReset(mglevels[i]->smoothd)); 2299566063dSJacob Faibussowitsch PetscCall(KSPReset(mglevels[i]->smoothu)); 2309566063dSJacob Faibussowitsch if (mglevels[i]->cr) PetscCall(KSPReset(mglevels[i]->cr)); 2313aeaf226SBarry Smith } 232f3b08a26SMatthew G. Knepley mg->Nc = 0; 2333aeaf226SBarry Smith } 2343ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2353aeaf226SBarry Smith } 2363aeaf226SBarry Smith 23741b6fd38SMatthew G. Knepley /* Implementing CR 23841b6fd38SMatthew G. Knepley 23941b6fd38SMatthew G. Knepley We only want to make corrections that ``do not change'' the coarse solution. What we mean by not changing is that if I prolong my coarse solution to the fine grid and then inject that fine solution back to the coarse grid, I get the same answer. Injection is what Brannick calls R. We want the complementary projector to Inj, which we will call S, after Brannick, so that Inj S = 0. Now the orthogonal projector onto the range of Inj^T is 24041b6fd38SMatthew G. Knepley 24141b6fd38SMatthew G. Knepley Inj^T (Inj Inj^T)^{-1} Inj 24241b6fd38SMatthew G. Knepley 24341b6fd38SMatthew G. Knepley and if Inj is a VecScatter, as it is now in PETSc, we have 24441b6fd38SMatthew G. Knepley 24541b6fd38SMatthew G. Knepley Inj^T Inj 24641b6fd38SMatthew G. Knepley 24741b6fd38SMatthew G. Knepley and 24841b6fd38SMatthew G. Knepley 24941b6fd38SMatthew G. Knepley S = I - Inj^T Inj 25041b6fd38SMatthew G. Knepley 25141b6fd38SMatthew G. Knepley since 25241b6fd38SMatthew G. Knepley 25341b6fd38SMatthew G. Knepley Inj S = Inj - (Inj Inj^T) Inj = 0. 25441b6fd38SMatthew G. Knepley 25541b6fd38SMatthew G. Knepley Brannick suggests 25641b6fd38SMatthew G. Knepley 25741b6fd38SMatthew G. Knepley A \to S^T A S \qquad\mathrm{and}\qquad M \to S^T M S 25841b6fd38SMatthew G. Knepley 25941b6fd38SMatthew G. Knepley but I do not think his :math:`S^T S = I` is correct. Our S is an orthogonal projector, so :math:`S^T S = S^2 = S`. We will use 26041b6fd38SMatthew G. Knepley 26141b6fd38SMatthew G. Knepley M^{-1} A \to S M^{-1} A S 26241b6fd38SMatthew G. Knepley 26341b6fd38SMatthew G. Knepley In fact, since it is somewhat hard in PETSc to do the symmetric application, we will just apply S on the left. 26441b6fd38SMatthew G. Knepley 26541b6fd38SMatthew G. Knepley Check: || Inj P - I ||_F < tol 26641b6fd38SMatthew G. Knepley Check: In general, Inj Inj^T = I 26741b6fd38SMatthew G. Knepley */ 26841b6fd38SMatthew G. Knepley 26941b6fd38SMatthew G. Knepley typedef struct { 27041b6fd38SMatthew G. Knepley PC mg; /* The PCMG object */ 27141b6fd38SMatthew G. Knepley PetscInt l; /* The multigrid level for this solver */ 27241b6fd38SMatthew G. Knepley Mat Inj; /* The injection matrix */ 27341b6fd38SMatthew G. Knepley Mat S; /* I - Inj^T Inj */ 27441b6fd38SMatthew G. Knepley } CRContext; 27541b6fd38SMatthew G. Knepley 276d71ae5a4SJacob Faibussowitsch static PetscErrorCode CRSetup_Private(PC pc) 277d71ae5a4SJacob Faibussowitsch { 27841b6fd38SMatthew G. Knepley CRContext *ctx; 27941b6fd38SMatthew G. Knepley Mat It; 28041b6fd38SMatthew G. Knepley 28141b6fd38SMatthew G. Knepley PetscFunctionBeginUser; 2829566063dSJacob Faibussowitsch PetscCall(PCShellGetContext(pc, &ctx)); 2839566063dSJacob Faibussowitsch PetscCall(PCMGGetInjection(ctx->mg, ctx->l, &It)); 28428b400f6SJacob Faibussowitsch PetscCheck(It, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "CR requires that injection be defined for this PCMG"); 2859566063dSJacob Faibussowitsch PetscCall(MatCreateTranspose(It, &ctx->Inj)); 2869566063dSJacob Faibussowitsch PetscCall(MatCreateNormal(ctx->Inj, &ctx->S)); 2879566063dSJacob Faibussowitsch PetscCall(MatScale(ctx->S, -1.0)); 2889566063dSJacob Faibussowitsch PetscCall(MatShift(ctx->S, 1.0)); 2893ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 29041b6fd38SMatthew G. Knepley } 29141b6fd38SMatthew G. Knepley 292d71ae5a4SJacob Faibussowitsch static PetscErrorCode CRApply_Private(PC pc, Vec x, Vec y) 293d71ae5a4SJacob Faibussowitsch { 29441b6fd38SMatthew G. Knepley CRContext *ctx; 29541b6fd38SMatthew G. Knepley 29641b6fd38SMatthew G. Knepley PetscFunctionBeginUser; 2979566063dSJacob Faibussowitsch PetscCall(PCShellGetContext(pc, &ctx)); 2989566063dSJacob Faibussowitsch PetscCall(MatMult(ctx->S, x, y)); 2993ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 30041b6fd38SMatthew G. Knepley } 30141b6fd38SMatthew G. Knepley 302d71ae5a4SJacob Faibussowitsch static PetscErrorCode CRDestroy_Private(PC pc) 303d71ae5a4SJacob Faibussowitsch { 30441b6fd38SMatthew G. Knepley CRContext *ctx; 30541b6fd38SMatthew G. Knepley 30641b6fd38SMatthew G. Knepley PetscFunctionBeginUser; 3079566063dSJacob Faibussowitsch PetscCall(PCShellGetContext(pc, &ctx)); 3089566063dSJacob Faibussowitsch PetscCall(MatDestroy(&ctx->Inj)); 3099566063dSJacob Faibussowitsch PetscCall(MatDestroy(&ctx->S)); 3109566063dSJacob Faibussowitsch PetscCall(PetscFree(ctx)); 3119566063dSJacob Faibussowitsch PetscCall(PCShellSetContext(pc, NULL)); 3123ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 31341b6fd38SMatthew G. Knepley } 31441b6fd38SMatthew G. Knepley 315d71ae5a4SJacob Faibussowitsch static PetscErrorCode CreateCR_Private(PC pc, PetscInt l, PC *cr) 316d71ae5a4SJacob Faibussowitsch { 31741b6fd38SMatthew G. Knepley CRContext *ctx; 31841b6fd38SMatthew G. Knepley 31941b6fd38SMatthew G. Knepley PetscFunctionBeginUser; 3209566063dSJacob Faibussowitsch PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), cr)); 3219566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)*cr, "S (complementary projector to injection)")); 3229566063dSJacob Faibussowitsch PetscCall(PetscCalloc1(1, &ctx)); 32341b6fd38SMatthew G. Knepley ctx->mg = pc; 32441b6fd38SMatthew G. Knepley ctx->l = l; 3259566063dSJacob Faibussowitsch PetscCall(PCSetType(*cr, PCSHELL)); 3269566063dSJacob Faibussowitsch PetscCall(PCShellSetContext(*cr, ctx)); 3279566063dSJacob Faibussowitsch PetscCall(PCShellSetApply(*cr, CRApply_Private)); 3289566063dSJacob Faibussowitsch PetscCall(PCShellSetSetUp(*cr, CRSetup_Private)); 3299566063dSJacob Faibussowitsch PetscCall(PCShellSetDestroy(*cr, CRDestroy_Private)); 3303ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 33141b6fd38SMatthew G. Knepley } 33241b6fd38SMatthew G. Knepley 333d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetLevels_MG(PC pc, PetscInt levels, MPI_Comm *comms) 334d71ae5a4SJacob Faibussowitsch { 335f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 336ce94432eSBarry Smith MPI_Comm comm; 3373aeaf226SBarry Smith PC_MG_Levels **mglevels = mg->levels; 33810eca3edSLisandro Dalcin PCMGType mgtype = mg->am; 33910167fecSBarry Smith PetscInt mgctype = (PetscInt)PC_MG_CYCLE_V; 340f3fbd535SBarry Smith PetscInt i; 341f3fbd535SBarry Smith PetscMPIInt size; 342f3fbd535SBarry Smith const char *prefix; 343f3fbd535SBarry Smith PC ipc; 3443aeaf226SBarry Smith PetscInt n; 3454b9ad928SBarry Smith 3464b9ad928SBarry Smith PetscFunctionBegin; 3470700a824SBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 348548767e0SJed Brown PetscValidLogicalCollectiveInt(pc, levels, 2); 3493ba16761SJacob Faibussowitsch if (mg->nlevels == levels) PetscFunctionReturn(PETSC_SUCCESS); 3509566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)pc, &comm)); 3513aeaf226SBarry Smith if (mglevels) { 35210eca3edSLisandro Dalcin mgctype = mglevels[0]->cycles; 3533aeaf226SBarry Smith /* changing the number of levels so free up the previous stuff */ 3549566063dSJacob Faibussowitsch PetscCall(PCReset_MG(pc)); 3553aeaf226SBarry Smith n = mglevels[0]->levels; 3563aeaf226SBarry Smith for (i = 0; i < n; i++) { 35748a46eb9SPierre Jolivet if (mglevels[i]->smoothd != mglevels[i]->smoothu) PetscCall(KSPDestroy(&mglevels[i]->smoothd)); 3589566063dSJacob Faibussowitsch PetscCall(KSPDestroy(&mglevels[i]->smoothu)); 3599566063dSJacob Faibussowitsch PetscCall(KSPDestroy(&mglevels[i]->cr)); 3609566063dSJacob Faibussowitsch PetscCall(PetscFree(mglevels[i])); 3613aeaf226SBarry Smith } 3629566063dSJacob Faibussowitsch PetscCall(PetscFree(mg->levels)); 3633aeaf226SBarry Smith } 364f3fbd535SBarry Smith 365f3fbd535SBarry Smith mg->nlevels = levels; 366f3fbd535SBarry Smith 3679566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(levels, &mglevels)); 368f3fbd535SBarry Smith 3699566063dSJacob Faibussowitsch PetscCall(PCGetOptionsPrefix(pc, &prefix)); 370f3fbd535SBarry Smith 371a9db87a2SMatthew G Knepley mg->stageApply = 0; 372f3fbd535SBarry Smith for (i = 0; i < levels; i++) { 3734dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&mglevels[i])); 3742fa5cd67SKarl Rupp 375f3fbd535SBarry Smith mglevels[i]->level = i; 376f3fbd535SBarry Smith mglevels[i]->levels = levels; 37710eca3edSLisandro Dalcin mglevels[i]->cycles = mgctype; 378336babb1SJed Brown mg->default_smoothu = 2; 379336babb1SJed Brown mg->default_smoothd = 2; 38063e6d426SJed Brown mglevels[i]->eventsmoothsetup = 0; 38163e6d426SJed Brown mglevels[i]->eventsmoothsolve = 0; 38263e6d426SJed Brown mglevels[i]->eventresidual = 0; 38363e6d426SJed Brown mglevels[i]->eventinterprestrict = 0; 384f3fbd535SBarry Smith 385f3fbd535SBarry Smith if (comms) comm = comms[i]; 386d5a8f7e6SBarry Smith if (comm != MPI_COMM_NULL) { 3879566063dSJacob Faibussowitsch PetscCall(KSPCreate(comm, &mglevels[i]->smoothd)); 388*3821be0aSBarry Smith PetscCall(KSPSetNestLevel(mglevels[i]->smoothd, pc->kspnestlevel)); 3899566063dSJacob Faibussowitsch PetscCall(KSPSetErrorIfNotConverged(mglevels[i]->smoothd, pc->erroriffailure)); 3909566063dSJacob Faibussowitsch PetscCall(PetscObjectIncrementTabLevel((PetscObject)mglevels[i]->smoothd, (PetscObject)pc, levels - i)); 3919566063dSJacob Faibussowitsch PetscCall(KSPSetOptionsPrefix(mglevels[i]->smoothd, prefix)); 3929566063dSJacob Faibussowitsch PetscCall(PetscObjectComposedDataSetInt((PetscObject)mglevels[i]->smoothd, PetscMGLevelId, mglevels[i]->level)); 3930ee9a668SBarry Smith if (i || levels == 1) { 3940ee9a668SBarry Smith char tprefix[128]; 3950ee9a668SBarry Smith 3969566063dSJacob Faibussowitsch PetscCall(KSPSetType(mglevels[i]->smoothd, KSPCHEBYSHEV)); 3979566063dSJacob Faibussowitsch PetscCall(KSPSetConvergenceTest(mglevels[i]->smoothd, KSPConvergedSkip, NULL, NULL)); 3989566063dSJacob Faibussowitsch PetscCall(KSPSetNormType(mglevels[i]->smoothd, KSP_NORM_NONE)); 3999566063dSJacob Faibussowitsch PetscCall(KSPGetPC(mglevels[i]->smoothd, &ipc)); 4009566063dSJacob Faibussowitsch PetscCall(PCSetType(ipc, PCSOR)); 4019566063dSJacob Faibussowitsch PetscCall(KSPSetTolerances(mglevels[i]->smoothd, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, mg->default_smoothd)); 402f3fbd535SBarry Smith 4039566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(tprefix, 128, "mg_levels_%d_", (int)i)); 4049566063dSJacob Faibussowitsch PetscCall(KSPAppendOptionsPrefix(mglevels[i]->smoothd, tprefix)); 4050ee9a668SBarry Smith } else { 4069566063dSJacob Faibussowitsch PetscCall(KSPAppendOptionsPrefix(mglevels[0]->smoothd, "mg_coarse_")); 407f3fbd535SBarry Smith 4089dbfc187SHong Zhang /* coarse solve is (redundant) LU by default; set shifttype NONZERO to avoid annoying zero-pivot in LU preconditioner */ 4099566063dSJacob Faibussowitsch PetscCall(KSPSetType(mglevels[0]->smoothd, KSPPREONLY)); 4109566063dSJacob Faibussowitsch PetscCall(KSPGetPC(mglevels[0]->smoothd, &ipc)); 4119566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(comm, &size)); 412f3fbd535SBarry Smith if (size > 1) { 4139566063dSJacob Faibussowitsch PetscCall(PCSetType(ipc, PCREDUNDANT)); 414f3fbd535SBarry Smith } else { 4159566063dSJacob Faibussowitsch PetscCall(PCSetType(ipc, PCLU)); 416f3fbd535SBarry Smith } 4179566063dSJacob Faibussowitsch PetscCall(PCFactorSetShiftType(ipc, MAT_SHIFT_INBLOCKS)); 418f3fbd535SBarry Smith } 419d5a8f7e6SBarry Smith } 420f3fbd535SBarry Smith mglevels[i]->smoothu = mglevels[i]->smoothd; 42131567311SBarry Smith mg->rtol = 0.0; 42231567311SBarry Smith mg->abstol = 0.0; 42331567311SBarry Smith mg->dtol = 0.0; 42431567311SBarry Smith mg->ttol = 0.0; 42531567311SBarry Smith mg->cyclesperpcapply = 1; 426f3fbd535SBarry Smith } 427f3fbd535SBarry Smith mg->levels = mglevels; 4289566063dSJacob Faibussowitsch PetscCall(PCMGSetType(pc, mgtype)); 4293ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 4304b9ad928SBarry Smith } 4314b9ad928SBarry Smith 43297d33e41SMatthew G. Knepley /*@C 433f1580f4eSBarry Smith PCMGSetLevels - Sets the number of levels to use with `PCMG`. 434f1580f4eSBarry Smith Must be called before any other `PCMG` routine. 43597d33e41SMatthew G. Knepley 436c3339decSBarry Smith Logically Collective 43797d33e41SMatthew G. Knepley 43897d33e41SMatthew G. Knepley Input Parameters: 43997d33e41SMatthew G. Knepley + pc - the preconditioner context 44097d33e41SMatthew G. Knepley . levels - the number of levels 44197d33e41SMatthew G. Knepley - comms - optional communicators for each level; this is to allow solving the coarser problems 442d5a8f7e6SBarry Smith on smaller sets of processes. For processes that are not included in the computation 44320f4b53cSBarry Smith you must pass `MPI_COMM_NULL`. Use comms = `NULL` to specify that all processes 44405552d4bSJunchao Zhang should participate in each level of problem. 44597d33e41SMatthew G. Knepley 44697d33e41SMatthew G. Knepley Level: intermediate 44797d33e41SMatthew G. Knepley 44897d33e41SMatthew G. Knepley Notes: 44920f4b53cSBarry Smith If the number of levels is one then the multigrid uses the `-mg_levels` prefix 45020f4b53cSBarry Smith for setting the level options rather than the `-mg_coarse` prefix. 45197d33e41SMatthew G. Knepley 452d5a8f7e6SBarry Smith You can free the information in comms after this routine is called. 453d5a8f7e6SBarry Smith 454f1580f4eSBarry Smith The array of MPI communicators must contain `MPI_COMM_NULL` for those ranks that at each level 455d5a8f7e6SBarry Smith are not participating in the coarser solve. For example, with 2 levels and 1 and 2 ranks on 456d5a8f7e6SBarry Smith the two levels, rank 0 in the original communicator will pass in an array of 2 communicators 457d5a8f7e6SBarry Smith of size 2 and 1, while rank 1 in the original communicator will pass in array of 2 communicators 458f1580f4eSBarry Smith the first of size 2 and the second of value `MPI_COMM_NULL` since the rank 1 does not participate 459d5a8f7e6SBarry Smith in the coarse grid solve. 460d5a8f7e6SBarry Smith 461f1580f4eSBarry Smith Since each coarser level may have a new `MPI_Comm` with fewer ranks than the previous, one 462d5a8f7e6SBarry Smith must take special care in providing the restriction and interpolation operation. We recommend 463d5a8f7e6SBarry Smith providing these as two step operations; first perform a standard restriction or interpolation on 464d5a8f7e6SBarry Smith the full number of ranks for that level and then use an MPI call to copy the resulting vector 46505552d4bSJunchao Zhang array entries (after calls to VecGetArray()) to the smaller or larger number of ranks, note in both 466d5a8f7e6SBarry Smith cases the MPI calls must be made on the larger of the two communicators. Traditional MPI send and 46720f4b53cSBarry Smith receives or `MPI_AlltoAllv()` could be used to do the reshuffling of the vector entries. 468d5a8f7e6SBarry Smith 469feefa0e1SJacob Faibussowitsch Fortran Notes: 47020f4b53cSBarry Smith Use comms = `PETSC_NULL_MPI_COMM` as the equivalent of `NULL` in the C interface. Note `PETSC_NULL_MPI_COMM` 471f1580f4eSBarry Smith is not `MPI_COMM_NULL`. It is more like `PETSC_NULL_INTEGER`, `PETSC_NULL_REAL` etc. 472d5a8f7e6SBarry Smith 473db781477SPatrick Sanan .seealso: `PCMGSetType()`, `PCMGGetLevels()` 47497d33e41SMatthew G. Knepley @*/ 475d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetLevels(PC pc, PetscInt levels, MPI_Comm *comms) 476d71ae5a4SJacob Faibussowitsch { 47797d33e41SMatthew G. Knepley PetscFunctionBegin; 47897d33e41SMatthew G. Knepley PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 4794f572ea9SToby Isaac if (comms) PetscAssertPointer(comms, 3); 480cac4c232SBarry Smith PetscTryMethod(pc, "PCMGSetLevels_C", (PC, PetscInt, MPI_Comm *), (pc, levels, comms)); 4813ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 48297d33e41SMatthew G. Knepley } 48397d33e41SMatthew G. Knepley 484d71ae5a4SJacob Faibussowitsch PetscErrorCode PCDestroy_MG(PC pc) 485d71ae5a4SJacob Faibussowitsch { 486a06653b4SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 487a06653b4SBarry Smith PC_MG_Levels **mglevels = mg->levels; 488a06653b4SBarry Smith PetscInt i, n; 489c07bf074SBarry Smith 490c07bf074SBarry Smith PetscFunctionBegin; 4919566063dSJacob Faibussowitsch PetscCall(PCReset_MG(pc)); 492a06653b4SBarry Smith if (mglevels) { 493a06653b4SBarry Smith n = mglevels[0]->levels; 494a06653b4SBarry Smith for (i = 0; i < n; i++) { 49548a46eb9SPierre Jolivet if (mglevels[i]->smoothd != mglevels[i]->smoothu) PetscCall(KSPDestroy(&mglevels[i]->smoothd)); 4969566063dSJacob Faibussowitsch PetscCall(KSPDestroy(&mglevels[i]->smoothu)); 4979566063dSJacob Faibussowitsch PetscCall(KSPDestroy(&mglevels[i]->cr)); 4989566063dSJacob Faibussowitsch PetscCall(PetscFree(mglevels[i])); 499a06653b4SBarry Smith } 5009566063dSJacob Faibussowitsch PetscCall(PetscFree(mg->levels)); 501a06653b4SBarry Smith } 5029566063dSJacob Faibussowitsch PetscCall(PetscFree(pc->data)); 5039566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetInterpolations_C", NULL)); 5049566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetCoarseOperators_C", NULL)); 5052b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetGalerkin_C", NULL)); 5062b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetLevels_C", NULL)); 5072b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetLevels_C", NULL)); 5082b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetInterpolations_C", NULL)); 5092b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetCoarseOperators_C", NULL)); 5102b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptInterpolation_C", NULL)); 5112b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptInterpolation_C", NULL)); 5122b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptCR_C", NULL)); 5132b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptCR_C", NULL)); 5142b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptCoarseSpaceType_C", NULL)); 5152b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptCoarseSpaceType_C", NULL)); 5163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 517f3fbd535SBarry Smith } 518f3fbd535SBarry Smith 519f3fbd535SBarry Smith /* 520f3fbd535SBarry Smith PCApply_MG - Runs either an additive, multiplicative, Kaskadic 521f3fbd535SBarry Smith or full cycle of multigrid. 522f3fbd535SBarry Smith 523f3fbd535SBarry Smith Note: 524f3fbd535SBarry Smith A simple wrapper which calls PCMGMCycle(),PCMGACycle(), or PCMGFCycle(). 525f3fbd535SBarry Smith */ 526d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_MG_Internal(PC pc, Vec b, Vec x, Mat B, Mat X, PetscBool transpose) 527d71ae5a4SJacob Faibussowitsch { 528f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 529f3fbd535SBarry Smith PC_MG_Levels **mglevels = mg->levels; 530391689abSStefano Zampini PC tpc; 531f3fbd535SBarry Smith PetscInt levels = mglevels[0]->levels, i; 53230b0564aSStefano Zampini PetscBool changeu, changed, matapp; 533f3fbd535SBarry Smith 534f3fbd535SBarry Smith PetscFunctionBegin; 53530b0564aSStefano Zampini matapp = (PetscBool)(B && X); 5369566063dSJacob Faibussowitsch if (mg->stageApply) PetscCall(PetscLogStagePush(mg->stageApply)); 537e1d8e5deSBarry Smith /* When the DM is supplying the matrix then it will not exist until here */ 538298cc208SBarry Smith for (i = 0; i < levels; i++) { 539e1d8e5deSBarry Smith if (!mglevels[i]->A) { 5409566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[i]->smoothu, &mglevels[i]->A, NULL)); 5419566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)mglevels[i]->A)); 542e1d8e5deSBarry Smith } 543e1d8e5deSBarry Smith } 544e1d8e5deSBarry Smith 5459566063dSJacob Faibussowitsch PetscCall(KSPGetPC(mglevels[levels - 1]->smoothd, &tpc)); 5469566063dSJacob Faibussowitsch PetscCall(PCPreSolveChangeRHS(tpc, &changed)); 5479566063dSJacob Faibussowitsch PetscCall(KSPGetPC(mglevels[levels - 1]->smoothu, &tpc)); 5489566063dSJacob Faibussowitsch PetscCall(PCPreSolveChangeRHS(tpc, &changeu)); 549391689abSStefano Zampini if (!changeu && !changed) { 55030b0564aSStefano Zampini if (matapp) { 5519566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[levels - 1]->B)); 55230b0564aSStefano Zampini mglevels[levels - 1]->B = B; 55330b0564aSStefano Zampini } else { 5549566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mglevels[levels - 1]->b)); 555f3fbd535SBarry Smith mglevels[levels - 1]->b = b; 55630b0564aSStefano Zampini } 557391689abSStefano Zampini } else { /* if the smoother changes the rhs during PreSolve, we cannot use the input vector */ 55830b0564aSStefano Zampini if (matapp) { 55930b0564aSStefano Zampini if (mglevels[levels - 1]->B) { 56030b0564aSStefano Zampini PetscInt N1, N2; 56130b0564aSStefano Zampini PetscBool flg; 56230b0564aSStefano Zampini 5639566063dSJacob Faibussowitsch PetscCall(MatGetSize(mglevels[levels - 1]->B, NULL, &N1)); 5649566063dSJacob Faibussowitsch PetscCall(MatGetSize(B, NULL, &N2)); 5659566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)mglevels[levels - 1]->B, ((PetscObject)B)->type_name, &flg)); 56648a46eb9SPierre Jolivet if (N1 != N2 || !flg) PetscCall(MatDestroy(&mglevels[levels - 1]->B)); 56730b0564aSStefano Zampini } 56830b0564aSStefano Zampini if (!mglevels[levels - 1]->B) { 5699566063dSJacob Faibussowitsch PetscCall(MatDuplicate(B, MAT_COPY_VALUES, &mglevels[levels - 1]->B)); 57030b0564aSStefano Zampini } else { 5719566063dSJacob Faibussowitsch PetscCall(MatCopy(B, mglevels[levels - 1]->B, SAME_NONZERO_PATTERN)); 57230b0564aSStefano Zampini } 57330b0564aSStefano Zampini } else { 574391689abSStefano Zampini if (!mglevels[levels - 1]->b) { 575391689abSStefano Zampini Vec *vec; 576391689abSStefano Zampini 5779566063dSJacob Faibussowitsch PetscCall(KSPCreateVecs(mglevels[levels - 1]->smoothd, 1, &vec, 0, NULL)); 578391689abSStefano Zampini mglevels[levels - 1]->b = *vec; 5799566063dSJacob Faibussowitsch PetscCall(PetscFree(vec)); 580391689abSStefano Zampini } 5819566063dSJacob Faibussowitsch PetscCall(VecCopy(b, mglevels[levels - 1]->b)); 582391689abSStefano Zampini } 58330b0564aSStefano Zampini } 5849371c9d4SSatish Balay if (matapp) { 5859371c9d4SSatish Balay mglevels[levels - 1]->X = X; 5869371c9d4SSatish Balay } else { 5879371c9d4SSatish Balay mglevels[levels - 1]->x = x; 5889371c9d4SSatish Balay } 58930b0564aSStefano Zampini 59030b0564aSStefano Zampini /* If coarser Xs are present, it means we have already block applied the PC at least once 59130b0564aSStefano Zampini Reset operators if sizes/type do no match */ 59230b0564aSStefano Zampini if (matapp && levels > 1 && mglevels[levels - 2]->X) { 59330b0564aSStefano Zampini PetscInt Xc, Bc; 59430b0564aSStefano Zampini PetscBool flg; 59530b0564aSStefano Zampini 5969566063dSJacob Faibussowitsch PetscCall(MatGetSize(mglevels[levels - 2]->X, NULL, &Xc)); 5979566063dSJacob Faibussowitsch PetscCall(MatGetSize(mglevels[levels - 1]->B, NULL, &Bc)); 5989566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)mglevels[levels - 2]->X, ((PetscObject)mglevels[levels - 1]->X)->type_name, &flg)); 59930b0564aSStefano Zampini if (Xc != Bc || !flg) { 6009566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[levels - 1]->R)); 60130b0564aSStefano Zampini for (i = 0; i < levels - 1; i++) { 6029566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i]->R)); 6039566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i]->B)); 6049566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mglevels[i]->X)); 60530b0564aSStefano Zampini } 60630b0564aSStefano Zampini } 60730b0564aSStefano Zampini } 608391689abSStefano Zampini 60931567311SBarry Smith if (mg->am == PC_MG_MULTIPLICATIVE) { 6109566063dSJacob Faibussowitsch if (matapp) PetscCall(MatZeroEntries(X)); 6119566063dSJacob Faibussowitsch else PetscCall(VecZeroEntries(x)); 61248a46eb9SPierre Jolivet for (i = 0; i < mg->cyclesperpcapply; i++) PetscCall(PCMGMCycle_Private(pc, mglevels + levels - 1, transpose, matapp, NULL)); 6132fa5cd67SKarl Rupp } else if (mg->am == PC_MG_ADDITIVE) { 6149566063dSJacob Faibussowitsch PetscCall(PCMGACycle_Private(pc, mglevels, transpose, matapp)); 6152fa5cd67SKarl Rupp } else if (mg->am == PC_MG_KASKADE) { 6169566063dSJacob Faibussowitsch PetscCall(PCMGKCycle_Private(pc, mglevels, transpose, matapp)); 6172fa5cd67SKarl Rupp } else { 6189566063dSJacob Faibussowitsch PetscCall(PCMGFCycle_Private(pc, mglevels, transpose, matapp)); 619f3fbd535SBarry Smith } 6209566063dSJacob Faibussowitsch if (mg->stageApply) PetscCall(PetscLogStagePop()); 62130b0564aSStefano Zampini if (!changeu && !changed) { 6229371c9d4SSatish Balay if (matapp) { 6239371c9d4SSatish Balay mglevels[levels - 1]->B = NULL; 6249371c9d4SSatish Balay } else { 6259371c9d4SSatish Balay mglevels[levels - 1]->b = NULL; 6269371c9d4SSatish Balay } 62730b0564aSStefano Zampini } 6283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 629f3fbd535SBarry Smith } 630f3fbd535SBarry Smith 631d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_MG(PC pc, Vec b, Vec x) 632d71ae5a4SJacob Faibussowitsch { 633fcb023d4SJed Brown PetscFunctionBegin; 6349566063dSJacob Faibussowitsch PetscCall(PCApply_MG_Internal(pc, b, x, NULL, NULL, PETSC_FALSE)); 6353ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 636fcb023d4SJed Brown } 637fcb023d4SJed Brown 638d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApplyTranspose_MG(PC pc, Vec b, Vec x) 639d71ae5a4SJacob Faibussowitsch { 640fcb023d4SJed Brown PetscFunctionBegin; 6419566063dSJacob Faibussowitsch PetscCall(PCApply_MG_Internal(pc, b, x, NULL, NULL, PETSC_TRUE)); 6423ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 64330b0564aSStefano Zampini } 64430b0564aSStefano Zampini 645d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_MG(PC pc, Mat b, Mat x) 646d71ae5a4SJacob Faibussowitsch { 64730b0564aSStefano Zampini PetscFunctionBegin; 6489566063dSJacob Faibussowitsch PetscCall(PCApply_MG_Internal(pc, NULL, NULL, b, x, PETSC_FALSE)); 6493ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 650fcb023d4SJed Brown } 651f3fbd535SBarry Smith 652d71ae5a4SJacob Faibussowitsch PetscErrorCode PCSetFromOptions_MG(PC pc, PetscOptionItems *PetscOptionsObject) 653d71ae5a4SJacob Faibussowitsch { 654710315b6SLawrence Mitchell PetscInt levels, cycles; 655f3b08a26SMatthew G. Knepley PetscBool flg, flg2; 656f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 6573d3eaba7SBarry Smith PC_MG_Levels **mglevels; 658f3fbd535SBarry Smith PCMGType mgtype; 659f3fbd535SBarry Smith PCMGCycleType mgctype; 6602134b1e4SBarry Smith PCMGGalerkinType gtype; 6612b3cbbdaSStefano Zampini PCMGCoarseSpaceType coarseSpaceType; 662f3fbd535SBarry Smith 663f3fbd535SBarry Smith PetscFunctionBegin; 6641d743356SStefano Zampini levels = PetscMax(mg->nlevels, 1); 665d0609cedSBarry Smith PetscOptionsHeadBegin(PetscOptionsObject, "Multigrid options"); 6669566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-pc_mg_levels", "Number of Levels", "PCMGSetLevels", levels, &levels, &flg)); 6671a97d7b6SStefano Zampini if (!flg && !mg->levels && pc->dm) { 6689566063dSJacob Faibussowitsch PetscCall(DMGetRefineLevel(pc->dm, &levels)); 669eb3f98d2SBarry Smith levels++; 6703aeaf226SBarry Smith mg->usedmfornumberoflevels = PETSC_TRUE; 671eb3f98d2SBarry Smith } 6729566063dSJacob Faibussowitsch PetscCall(PCMGSetLevels(pc, levels, NULL)); 6733aeaf226SBarry Smith mglevels = mg->levels; 6743aeaf226SBarry Smith 675f3fbd535SBarry Smith mgctype = (PCMGCycleType)mglevels[0]->cycles; 6769566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-pc_mg_cycle_type", "V cycle or for W-cycle", "PCMGSetCycleType", PCMGCycleTypes, (PetscEnum)mgctype, (PetscEnum *)&mgctype, &flg)); 6771baa6e33SBarry Smith if (flg) PetscCall(PCMGSetCycleType(pc, mgctype)); 6782134b1e4SBarry Smith gtype = mg->galerkin; 6799566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-pc_mg_galerkin", "Use Galerkin process to compute coarser operators", "PCMGSetGalerkin", PCMGGalerkinTypes, (PetscEnum)gtype, (PetscEnum *)>ype, &flg)); 6801baa6e33SBarry Smith if (flg) PetscCall(PCMGSetGalerkin(pc, gtype)); 6812b3cbbdaSStefano Zampini coarseSpaceType = mg->coarseSpaceType; 6822b3cbbdaSStefano Zampini PetscCall(PetscOptionsEnum("-pc_mg_adapt_interp_coarse_space", "Type of adaptive coarse space: none, polynomial, harmonic, eigenvector, generalized_eigenvector, gdsw", "PCMGSetAdaptCoarseSpaceType", PCMGCoarseSpaceTypes, (PetscEnum)coarseSpaceType, (PetscEnum *)&coarseSpaceType, &flg)); 6832b3cbbdaSStefano Zampini if (flg) PetscCall(PCMGSetAdaptCoarseSpaceType(pc, coarseSpaceType)); 6849566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-pc_mg_adapt_interp_n", "Size of the coarse space for adaptive interpolation", "PCMGSetCoarseSpace", mg->Nc, &mg->Nc, &flg)); 6859566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-pc_mg_mesp_monitor", "Monitor the multilevel eigensolver", "PCMGSetAdaptInterpolation", PETSC_FALSE, &mg->mespMonitor, &flg)); 68641b6fd38SMatthew G. Knepley flg2 = PETSC_FALSE; 6879566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-pc_mg_adapt_cr", "Monitor coarse space quality using Compatible Relaxation (CR)", "PCMGSetAdaptCR", PETSC_FALSE, &flg2, &flg)); 6889566063dSJacob Faibussowitsch if (flg) PetscCall(PCMGSetAdaptCR(pc, flg2)); 689f56b1265SBarry Smith flg = PETSC_FALSE; 6909566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-pc_mg_distinct_smoothup", "Create separate smoothup KSP and append the prefix _up", "PCMGSetDistinctSmoothUp", PETSC_FALSE, &flg, NULL)); 6911baa6e33SBarry Smith if (flg) PetscCall(PCMGSetDistinctSmoothUp(pc)); 69231567311SBarry Smith mgtype = mg->am; 6939566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-pc_mg_type", "Multigrid type", "PCMGSetType", PCMGTypes, (PetscEnum)mgtype, (PetscEnum *)&mgtype, &flg)); 6941baa6e33SBarry Smith if (flg) PetscCall(PCMGSetType(pc, mgtype)); 69531567311SBarry Smith if (mg->am == PC_MG_MULTIPLICATIVE) { 6969566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-pc_mg_multiplicative_cycles", "Number of cycles for each preconditioner step", "PCMGMultiplicativeSetCycles", mg->cyclesperpcapply, &cycles, &flg)); 6971baa6e33SBarry Smith if (flg) PetscCall(PCMGMultiplicativeSetCycles(pc, cycles)); 698f3fbd535SBarry Smith } 699f3fbd535SBarry Smith flg = PETSC_FALSE; 7009566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-pc_mg_log", "Log times for each multigrid level", "None", flg, &flg, NULL)); 701f3fbd535SBarry Smith if (flg) { 702f3fbd535SBarry Smith PetscInt i; 703f3fbd535SBarry Smith char eventname[128]; 7041a97d7b6SStefano Zampini 705f3fbd535SBarry Smith levels = mglevels[0]->levels; 706f3fbd535SBarry Smith for (i = 0; i < levels; i++) { 707a364092eSJacob Faibussowitsch PetscCall(PetscSNPrintf(eventname, PETSC_STATIC_ARRAY_LENGTH(eventname), "MGSetup Level %d", (int)i)); 7089566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister(eventname, ((PetscObject)pc)->classid, &mglevels[i]->eventsmoothsetup)); 709a364092eSJacob Faibussowitsch PetscCall(PetscSNPrintf(eventname, PETSC_STATIC_ARRAY_LENGTH(eventname), "MGSmooth Level %d", (int)i)); 7109566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister(eventname, ((PetscObject)pc)->classid, &mglevels[i]->eventsmoothsolve)); 711f3fbd535SBarry Smith if (i) { 712a364092eSJacob Faibussowitsch PetscCall(PetscSNPrintf(eventname, PETSC_STATIC_ARRAY_LENGTH(eventname), "MGResid Level %d", (int)i)); 7139566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister(eventname, ((PetscObject)pc)->classid, &mglevels[i]->eventresidual)); 714a364092eSJacob Faibussowitsch PetscCall(PetscSNPrintf(eventname, PETSC_STATIC_ARRAY_LENGTH(eventname), "MGInterp Level %d", (int)i)); 7159566063dSJacob Faibussowitsch PetscCall(PetscLogEventRegister(eventname, ((PetscObject)pc)->classid, &mglevels[i]->eventinterprestrict)); 716f3fbd535SBarry Smith } 717f3fbd535SBarry Smith } 718eec35531SMatthew G Knepley 719ec60ca73SSatish Balay #if defined(PETSC_USE_LOG) 720eec35531SMatthew G Knepley { 721eec35531SMatthew G Knepley const char *sname = "MG Apply"; 722eec35531SMatthew G Knepley PetscStageLog stageLog; 723eec35531SMatthew G Knepley PetscInt st; 724eec35531SMatthew G Knepley 7259566063dSJacob Faibussowitsch PetscCall(PetscLogGetStageLog(&stageLog)); 726eec35531SMatthew G Knepley for (st = 0; st < stageLog->numStages; ++st) { 727eec35531SMatthew G Knepley PetscBool same; 728eec35531SMatthew G Knepley 7299566063dSJacob Faibussowitsch PetscCall(PetscStrcmp(stageLog->stageInfo[st].name, sname, &same)); 7302fa5cd67SKarl Rupp if (same) mg->stageApply = st; 731eec35531SMatthew G Knepley } 73248a46eb9SPierre Jolivet if (!mg->stageApply) PetscCall(PetscLogStageRegister(sname, &mg->stageApply)); 733eec35531SMatthew G Knepley } 734ec60ca73SSatish Balay #endif 735f3fbd535SBarry Smith } 736d0609cedSBarry Smith PetscOptionsHeadEnd(); 737f3b08a26SMatthew G. Knepley /* Check option consistency */ 7389566063dSJacob Faibussowitsch PetscCall(PCMGGetGalerkin(pc, >ype)); 7399566063dSJacob Faibussowitsch PetscCall(PCMGGetAdaptInterpolation(pc, &flg)); 74008401ef6SPierre Jolivet PetscCheck(!flg || !(gtype >= PC_MG_GALERKIN_NONE), PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_INCOMP, "Must use Galerkin coarse operators when adapting the interpolator"); 7413ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 742f3fbd535SBarry Smith } 743f3fbd535SBarry Smith 7440a545947SLisandro Dalcin const char *const PCMGTypes[] = {"MULTIPLICATIVE", "ADDITIVE", "FULL", "KASKADE", "PCMGType", "PC_MG", NULL}; 7450a545947SLisandro Dalcin const char *const PCMGCycleTypes[] = {"invalid", "v", "w", "PCMGCycleType", "PC_MG_CYCLE", NULL}; 7460a545947SLisandro Dalcin const char *const PCMGGalerkinTypes[] = {"both", "pmat", "mat", "none", "external", "PCMGGalerkinType", "PC_MG_GALERKIN", NULL}; 7472b3cbbdaSStefano Zampini const char *const PCMGCoarseSpaceTypes[] = {"none", "polynomial", "harmonic", "eigenvector", "generalized_eigenvector", "gdsw", "PCMGCoarseSpaceType", "PCMG_ADAPT_NONE", NULL}; 748f3fbd535SBarry Smith 7499804daf3SBarry Smith #include <petscdraw.h> 750d71ae5a4SJacob Faibussowitsch PetscErrorCode PCView_MG(PC pc, PetscViewer viewer) 751d71ae5a4SJacob Faibussowitsch { 752f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 753f3fbd535SBarry Smith PC_MG_Levels **mglevels = mg->levels; 754e3deeeafSJed Brown PetscInt levels = mglevels ? mglevels[0]->levels : 0, i; 755219b1045SBarry Smith PetscBool iascii, isbinary, isdraw; 756f3fbd535SBarry Smith 757f3fbd535SBarry Smith PetscFunctionBegin; 7589566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii)); 7599566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERBINARY, &isbinary)); 7609566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERDRAW, &isdraw)); 761f3fbd535SBarry Smith if (iascii) { 762e3deeeafSJed Brown const char *cyclename = levels ? (mglevels[0]->cycles == PC_MG_CYCLE_V ? "v" : "w") : "unknown"; 76363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " type is %s, levels=%" PetscInt_FMT " cycles=%s\n", PCMGTypes[mg->am], levels, cyclename)); 76448a46eb9SPierre Jolivet if (mg->am == PC_MG_MULTIPLICATIVE) PetscCall(PetscViewerASCIIPrintf(viewer, " Cycles per PCApply=%" PetscInt_FMT "\n", mg->cyclesperpcapply)); 7652134b1e4SBarry Smith if (mg->galerkin == PC_MG_GALERKIN_BOTH) { 7669566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Using Galerkin computed coarse grid matrices\n")); 7672134b1e4SBarry Smith } else if (mg->galerkin == PC_MG_GALERKIN_PMAT) { 7689566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Using Galerkin computed coarse grid matrices for pmat\n")); 7692134b1e4SBarry Smith } else if (mg->galerkin == PC_MG_GALERKIN_MAT) { 7709566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Using Galerkin computed coarse grid matrices for mat\n")); 7712134b1e4SBarry Smith } else if (mg->galerkin == PC_MG_GALERKIN_EXTERNAL) { 7729566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Using externally compute Galerkin coarse grid matrices\n")); 7734f66f45eSBarry Smith } else { 7749566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Not using Galerkin computed coarse grid matrices\n")); 775f3fbd535SBarry Smith } 7761baa6e33SBarry Smith if (mg->view) PetscCall((*mg->view)(pc, viewer)); 777f3fbd535SBarry Smith for (i = 0; i < levels; i++) { 77863a3b9bcSJacob Faibussowitsch if (i) { 77963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Down solver (pre-smoother) on level %" PetscInt_FMT " -------------------------------\n", i)); 780f3fbd535SBarry Smith } else { 78163a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Coarse grid solver -- level %" PetscInt_FMT " -------------------------------\n", i)); 782f3fbd535SBarry Smith } 7839566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 7849566063dSJacob Faibussowitsch PetscCall(KSPView(mglevels[i]->smoothd, viewer)); 7859566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 786f3fbd535SBarry Smith if (i && mglevels[i]->smoothd == mglevels[i]->smoothu) { 7879566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Up solver (post-smoother) same as down solver (pre-smoother)\n")); 788f3fbd535SBarry Smith } else if (i) { 78963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Up solver (post-smoother) on level %" PetscInt_FMT " -------------------------------\n", i)); 7909566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 7919566063dSJacob Faibussowitsch PetscCall(KSPView(mglevels[i]->smoothu, viewer)); 7929566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 793f3fbd535SBarry Smith } 79441b6fd38SMatthew G. Knepley if (i && mglevels[i]->cr) { 79563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "CR solver on level %" PetscInt_FMT " -------------------------------\n", i)); 7969566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 7979566063dSJacob Faibussowitsch PetscCall(KSPView(mglevels[i]->cr, viewer)); 7989566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 79941b6fd38SMatthew G. Knepley } 800f3fbd535SBarry Smith } 8015b0b0462SBarry Smith } else if (isbinary) { 8025b0b0462SBarry Smith for (i = levels - 1; i >= 0; i--) { 8039566063dSJacob Faibussowitsch PetscCall(KSPView(mglevels[i]->smoothd, viewer)); 80448a46eb9SPierre Jolivet if (i && mglevels[i]->smoothd != mglevels[i]->smoothu) PetscCall(KSPView(mglevels[i]->smoothu, viewer)); 8055b0b0462SBarry Smith } 806219b1045SBarry Smith } else if (isdraw) { 807219b1045SBarry Smith PetscDraw draw; 808b4375e8dSPeter Brune PetscReal x, w, y, bottom, th; 8099566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDraw(viewer, 0, &draw)); 8109566063dSJacob Faibussowitsch PetscCall(PetscDrawGetCurrentPoint(draw, &x, &y)); 8119566063dSJacob Faibussowitsch PetscCall(PetscDrawStringGetSize(draw, NULL, &th)); 812219b1045SBarry Smith bottom = y - th; 813219b1045SBarry Smith for (i = levels - 1; i >= 0; i--) { 814b4375e8dSPeter Brune if (!mglevels[i]->smoothu || (mglevels[i]->smoothu == mglevels[i]->smoothd)) { 8159566063dSJacob Faibussowitsch PetscCall(PetscDrawPushCurrentPoint(draw, x, bottom)); 8169566063dSJacob Faibussowitsch PetscCall(KSPView(mglevels[i]->smoothd, viewer)); 8179566063dSJacob Faibussowitsch PetscCall(PetscDrawPopCurrentPoint(draw)); 818b4375e8dSPeter Brune } else { 819b4375e8dSPeter Brune w = 0.5 * PetscMin(1.0 - x, x); 8209566063dSJacob Faibussowitsch PetscCall(PetscDrawPushCurrentPoint(draw, x + w, bottom)); 8219566063dSJacob Faibussowitsch PetscCall(KSPView(mglevels[i]->smoothd, viewer)); 8229566063dSJacob Faibussowitsch PetscCall(PetscDrawPopCurrentPoint(draw)); 8239566063dSJacob Faibussowitsch PetscCall(PetscDrawPushCurrentPoint(draw, x - w, bottom)); 8249566063dSJacob Faibussowitsch PetscCall(KSPView(mglevels[i]->smoothu, viewer)); 8259566063dSJacob Faibussowitsch PetscCall(PetscDrawPopCurrentPoint(draw)); 826b4375e8dSPeter Brune } 8279566063dSJacob Faibussowitsch PetscCall(PetscDrawGetBoundingBox(draw, NULL, &bottom, NULL, NULL)); 8281cd381d6SBarry Smith bottom -= th; 829219b1045SBarry Smith } 8305b0b0462SBarry Smith } 8313ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 832f3fbd535SBarry Smith } 833f3fbd535SBarry Smith 834af0996ceSBarry Smith #include <petsc/private/kspimpl.h> 835cab2e9ccSBarry Smith 836f3fbd535SBarry Smith /* 837f3fbd535SBarry Smith Calls setup for the KSP on each level 838f3fbd535SBarry Smith */ 839d71ae5a4SJacob Faibussowitsch PetscErrorCode PCSetUp_MG(PC pc) 840d71ae5a4SJacob Faibussowitsch { 841f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 842f3fbd535SBarry Smith PC_MG_Levels **mglevels = mg->levels; 8431a97d7b6SStefano Zampini PetscInt i, n; 84498e04984SBarry Smith PC cpc; 8453db492dfSBarry Smith PetscBool dump = PETSC_FALSE, opsset, use_amat, missinginterpolate = PETSC_FALSE; 846f3fbd535SBarry Smith Mat dA, dB; 847f3fbd535SBarry Smith Vec tvec; 848218a07d4SBarry Smith DM *dms; 8490a545947SLisandro Dalcin PetscViewer viewer = NULL; 85041b6fd38SMatthew G. Knepley PetscBool dAeqdB = PETSC_FALSE, needRestricts = PETSC_FALSE, doCR = PETSC_FALSE; 8512b3cbbdaSStefano Zampini PetscBool adaptInterpolation = mg->adaptInterpolation; 852f3fbd535SBarry Smith 853f3fbd535SBarry Smith PetscFunctionBegin; 85428b400f6SJacob Faibussowitsch PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "Must set MG levels with PCMGSetLevels() before setting up"); 8551a97d7b6SStefano Zampini n = mglevels[0]->levels; 85601bc414fSDmitry Karpeev /* FIX: Move this to PCSetFromOptions_MG? */ 8573aeaf226SBarry Smith if (mg->usedmfornumberoflevels) { 8583aeaf226SBarry Smith PetscInt levels; 8599566063dSJacob Faibussowitsch PetscCall(DMGetRefineLevel(pc->dm, &levels)); 8603aeaf226SBarry Smith levels++; 8613aeaf226SBarry Smith if (levels > n) { /* the problem is now being solved on a finer grid */ 8629566063dSJacob Faibussowitsch PetscCall(PCMGSetLevels(pc, levels, NULL)); 8633aeaf226SBarry Smith n = levels; 8649566063dSJacob Faibussowitsch PetscCall(PCSetFromOptions(pc)); /* it is bad to call this here, but otherwise will never be called for the new hierarchy */ 8653aeaf226SBarry Smith mglevels = mg->levels; 8663aeaf226SBarry Smith } 8673aeaf226SBarry Smith } 8689566063dSJacob Faibussowitsch PetscCall(KSPGetPC(mglevels[0]->smoothd, &cpc)); 8693aeaf226SBarry Smith 870f3fbd535SBarry Smith /* If user did not provide fine grid operators OR operator was not updated since last global KSPSetOperators() */ 871f3fbd535SBarry Smith /* so use those from global PC */ 872f3fbd535SBarry Smith /* Is this what we always want? What if user wants to keep old one? */ 8739566063dSJacob Faibussowitsch PetscCall(KSPGetOperatorsSet(mglevels[n - 1]->smoothd, NULL, &opsset)); 87498e04984SBarry Smith if (opsset) { 87598e04984SBarry Smith Mat mmat; 8769566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[n - 1]->smoothd, NULL, &mmat)); 87798e04984SBarry Smith if (mmat == pc->pmat) opsset = PETSC_FALSE; 87898e04984SBarry Smith } 8794a5f07a5SMark F. Adams 88041b6fd38SMatthew G. Knepley /* Create CR solvers */ 8819566063dSJacob Faibussowitsch PetscCall(PCMGGetAdaptCR(pc, &doCR)); 88241b6fd38SMatthew G. Knepley if (doCR) { 88341b6fd38SMatthew G. Knepley const char *prefix; 88441b6fd38SMatthew G. Knepley 8859566063dSJacob Faibussowitsch PetscCall(PCGetOptionsPrefix(pc, &prefix)); 88641b6fd38SMatthew G. Knepley for (i = 1; i < n; ++i) { 88741b6fd38SMatthew G. Knepley PC ipc, cr; 88841b6fd38SMatthew G. Knepley char crprefix[128]; 88941b6fd38SMatthew G. Knepley 8909566063dSJacob Faibussowitsch PetscCall(KSPCreate(PetscObjectComm((PetscObject)pc), &mglevels[i]->cr)); 891*3821be0aSBarry Smith PetscCall(KSPSetNestLevel(mglevels[i]->cr, pc->kspnestlevel)); 8929566063dSJacob Faibussowitsch PetscCall(KSPSetErrorIfNotConverged(mglevels[i]->cr, PETSC_FALSE)); 8939566063dSJacob Faibussowitsch PetscCall(PetscObjectIncrementTabLevel((PetscObject)mglevels[i]->cr, (PetscObject)pc, n - i)); 8949566063dSJacob Faibussowitsch PetscCall(KSPSetOptionsPrefix(mglevels[i]->cr, prefix)); 8959566063dSJacob Faibussowitsch PetscCall(PetscObjectComposedDataSetInt((PetscObject)mglevels[i]->cr, PetscMGLevelId, mglevels[i]->level)); 8969566063dSJacob Faibussowitsch PetscCall(KSPSetType(mglevels[i]->cr, KSPCHEBYSHEV)); 8979566063dSJacob Faibussowitsch PetscCall(KSPSetConvergenceTest(mglevels[i]->cr, KSPConvergedSkip, NULL, NULL)); 8989566063dSJacob Faibussowitsch PetscCall(KSPSetNormType(mglevels[i]->cr, KSP_NORM_PRECONDITIONED)); 8999566063dSJacob Faibussowitsch PetscCall(KSPGetPC(mglevels[i]->cr, &ipc)); 90041b6fd38SMatthew G. Knepley 9019566063dSJacob Faibussowitsch PetscCall(PCSetType(ipc, PCCOMPOSITE)); 9029566063dSJacob Faibussowitsch PetscCall(PCCompositeSetType(ipc, PC_COMPOSITE_MULTIPLICATIVE)); 9039566063dSJacob Faibussowitsch PetscCall(PCCompositeAddPCType(ipc, PCSOR)); 9049566063dSJacob Faibussowitsch PetscCall(CreateCR_Private(pc, i, &cr)); 9059566063dSJacob Faibussowitsch PetscCall(PCCompositeAddPC(ipc, cr)); 9069566063dSJacob Faibussowitsch PetscCall(PCDestroy(&cr)); 90741b6fd38SMatthew G. Knepley 9089566063dSJacob Faibussowitsch PetscCall(KSPSetTolerances(mglevels[i]->cr, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, mg->default_smoothd)); 9099566063dSJacob Faibussowitsch PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->cr, PETSC_TRUE)); 9109566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(crprefix, 128, "mg_levels_%d_cr_", (int)i)); 9119566063dSJacob Faibussowitsch PetscCall(KSPAppendOptionsPrefix(mglevels[i]->cr, crprefix)); 91241b6fd38SMatthew G. Knepley } 91341b6fd38SMatthew G. Knepley } 91441b6fd38SMatthew G. Knepley 9154a5f07a5SMark F. Adams if (!opsset) { 9169566063dSJacob Faibussowitsch PetscCall(PCGetUseAmat(pc, &use_amat)); 9174a5f07a5SMark F. Adams if (use_amat) { 9189566063dSJacob Faibussowitsch PetscCall(PetscInfo(pc, "Using outer operators to define finest grid operator \n because PCMGGetSmoother(pc,nlevels-1,&ksp);KSPSetOperators(ksp,...); was not called.\n")); 9199566063dSJacob Faibussowitsch PetscCall(KSPSetOperators(mglevels[n - 1]->smoothd, pc->mat, pc->pmat)); 92069858f1bSStefano Zampini } else { 9219566063dSJacob Faibussowitsch PetscCall(PetscInfo(pc, "Using matrix (pmat) operators to define finest grid operator \n because PCMGGetSmoother(pc,nlevels-1,&ksp);KSPSetOperators(ksp,...); was not called.\n")); 9229566063dSJacob Faibussowitsch PetscCall(KSPSetOperators(mglevels[n - 1]->smoothd, pc->pmat, pc->pmat)); 9234a5f07a5SMark F. Adams } 9244a5f07a5SMark F. Adams } 925f3fbd535SBarry Smith 9269c7ffb39SBarry Smith for (i = n - 1; i > 0; i--) { 9279c7ffb39SBarry Smith if (!(mglevels[i]->interpolate || mglevels[i]->restrct)) { 9289c7ffb39SBarry Smith missinginterpolate = PETSC_TRUE; 9292b3cbbdaSStefano Zampini break; 9309c7ffb39SBarry Smith } 9319c7ffb39SBarry Smith } 9322134b1e4SBarry Smith 9339566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[n - 1]->smoothd, &dA, &dB)); 9342134b1e4SBarry Smith if (dA == dB) dAeqdB = PETSC_TRUE; 9352b3cbbdaSStefano Zampini if (mg->galerkin == PC_MG_GALERKIN_NONE || ((mg->galerkin == PC_MG_GALERKIN_PMAT || mg->galerkin == PC_MG_GALERKIN_MAT) && !dAeqdB)) { 9362134b1e4SBarry Smith needRestricts = PETSC_TRUE; /* user must compute either mat, pmat, or both so must restrict x to coarser levels */ 9372134b1e4SBarry Smith } 9382134b1e4SBarry Smith 9392b3cbbdaSStefano Zampini if (pc->dm && !pc->setupcalled) { 9402b3cbbdaSStefano Zampini /* finest smoother also gets DM but it is not active, independent of whether galerkin==PC_MG_GALERKIN_EXTERNAL */ 9412b3cbbdaSStefano Zampini PetscCall(KSPSetDM(mglevels[n - 1]->smoothd, pc->dm)); 9422b3cbbdaSStefano Zampini PetscCall(KSPSetDMActive(mglevels[n - 1]->smoothd, PETSC_FALSE)); 9432b3cbbdaSStefano Zampini if (mglevels[n - 1]->smoothd != mglevels[n - 1]->smoothu) { 9442b3cbbdaSStefano Zampini PetscCall(KSPSetDM(mglevels[n - 1]->smoothu, pc->dm)); 9452b3cbbdaSStefano Zampini PetscCall(KSPSetDMActive(mglevels[n - 1]->smoothu, PETSC_FALSE)); 9462b3cbbdaSStefano Zampini } 9472b3cbbdaSStefano Zampini if (mglevels[n - 1]->cr) { 9482b3cbbdaSStefano Zampini PetscCall(KSPSetDM(mglevels[n - 1]->cr, pc->dm)); 9492b3cbbdaSStefano Zampini PetscCall(KSPSetDMActive(mglevels[n - 1]->cr, PETSC_FALSE)); 9502b3cbbdaSStefano Zampini } 9512b3cbbdaSStefano Zampini } 9522b3cbbdaSStefano Zampini 9539c7ffb39SBarry Smith /* 9549c7ffb39SBarry Smith Skipping if user has provided all interpolation/restriction needed (since DM might not be able to produce them (when coming from SNES/TS) 9552b3cbbdaSStefano Zampini Skipping for externally managed hierarchy (such as ML and GAMG). Cleaner logic here would be great. Wrap ML/GAMG as DMs? 9569c7ffb39SBarry Smith */ 95732fe681dSStefano Zampini if (missinginterpolate && mg->galerkin != PC_MG_GALERKIN_EXTERNAL && !pc->setupcalled) { 95832fe681dSStefano Zampini /* first see if we can compute a coarse space */ 95932fe681dSStefano Zampini if (mg->coarseSpaceType == PCMG_ADAPT_GDSW) { 96032fe681dSStefano Zampini for (i = n - 2; i > -1; i--) { 96132fe681dSStefano Zampini if (!mglevels[i + 1]->restrct && !mglevels[i + 1]->interpolate) { 96232fe681dSStefano Zampini PetscCall(PCMGComputeCoarseSpace_Internal(pc, i + 1, mg->coarseSpaceType, mg->Nc, NULL, &mglevels[i + 1]->coarseSpace)); 96332fe681dSStefano Zampini PetscCall(PCMGSetInterpolation(pc, i + 1, mglevels[i + 1]->coarseSpace)); 96432fe681dSStefano Zampini } 96532fe681dSStefano Zampini } 96632fe681dSStefano Zampini } else { /* construct the interpolation from the DMs */ 9672e499ae9SBarry Smith Mat p; 96873250ac0SBarry Smith Vec rscale; 9699566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n, &dms)); 970218a07d4SBarry Smith dms[n - 1] = pc->dm; 971ef1267afSMatthew G. Knepley /* Separately create them so we do not get DMKSP interference between levels */ 9729566063dSJacob Faibussowitsch for (i = n - 2; i > -1; i--) PetscCall(DMCoarsen(dms[i + 1], MPI_COMM_NULL, &dms[i])); 973218a07d4SBarry Smith for (i = n - 2; i > -1; i--) { 974942e3340SBarry Smith DMKSP kdm; 975eab52d2bSLawrence Mitchell PetscBool dmhasrestrict, dmhasinject; 9769566063dSJacob Faibussowitsch PetscCall(KSPSetDM(mglevels[i]->smoothd, dms[i])); 9779566063dSJacob Faibussowitsch if (!needRestricts) PetscCall(KSPSetDMActive(mglevels[i]->smoothd, PETSC_FALSE)); 978c27ee7a3SPatrick Farrell if (mglevels[i]->smoothd != mglevels[i]->smoothu) { 9799566063dSJacob Faibussowitsch PetscCall(KSPSetDM(mglevels[i]->smoothu, dms[i])); 9809566063dSJacob Faibussowitsch if (!needRestricts) PetscCall(KSPSetDMActive(mglevels[i]->smoothu, PETSC_FALSE)); 981c27ee7a3SPatrick Farrell } 98241b6fd38SMatthew G. Knepley if (mglevels[i]->cr) { 9839566063dSJacob Faibussowitsch PetscCall(KSPSetDM(mglevels[i]->cr, dms[i])); 9849566063dSJacob Faibussowitsch if (!needRestricts) PetscCall(KSPSetDMActive(mglevels[i]->cr, PETSC_FALSE)); 98541b6fd38SMatthew G. Knepley } 9869566063dSJacob Faibussowitsch PetscCall(DMGetDMKSPWrite(dms[i], &kdm)); 987d1a3e677SJed Brown /* Ugly hack so that the next KSPSetUp() will use the RHS that we set. A better fix is to change dmActive to take 988d1a3e677SJed Brown * a bitwise OR of computing the matrix, RHS, and initial iterate. */ 9890298fd71SBarry Smith kdm->ops->computerhs = NULL; 9900298fd71SBarry Smith kdm->rhsctx = NULL; 99124c3aa18SBarry Smith if (!mglevels[i + 1]->interpolate) { 9929566063dSJacob Faibussowitsch PetscCall(DMCreateInterpolation(dms[i], dms[i + 1], &p, &rscale)); 9939566063dSJacob Faibussowitsch PetscCall(PCMGSetInterpolation(pc, i + 1, p)); 9949566063dSJacob Faibussowitsch if (rscale) PetscCall(PCMGSetRScale(pc, i + 1, rscale)); 9959566063dSJacob Faibussowitsch PetscCall(VecDestroy(&rscale)); 9969566063dSJacob Faibussowitsch PetscCall(MatDestroy(&p)); 997218a07d4SBarry Smith } 9989566063dSJacob Faibussowitsch PetscCall(DMHasCreateRestriction(dms[i], &dmhasrestrict)); 9993ad4599aSBarry Smith if (dmhasrestrict && !mglevels[i + 1]->restrct) { 10009566063dSJacob Faibussowitsch PetscCall(DMCreateRestriction(dms[i], dms[i + 1], &p)); 10019566063dSJacob Faibussowitsch PetscCall(PCMGSetRestriction(pc, i + 1, p)); 10029566063dSJacob Faibussowitsch PetscCall(MatDestroy(&p)); 10033ad4599aSBarry Smith } 10049566063dSJacob Faibussowitsch PetscCall(DMHasCreateInjection(dms[i], &dmhasinject)); 1005eab52d2bSLawrence Mitchell if (dmhasinject && !mglevels[i + 1]->inject) { 10069566063dSJacob Faibussowitsch PetscCall(DMCreateInjection(dms[i], dms[i + 1], &p)); 10079566063dSJacob Faibussowitsch PetscCall(PCMGSetInjection(pc, i + 1, p)); 10089566063dSJacob Faibussowitsch PetscCall(MatDestroy(&p)); 1009eab52d2bSLawrence Mitchell } 101024c3aa18SBarry Smith } 10112d2b81a6SBarry Smith 10129566063dSJacob Faibussowitsch for (i = n - 2; i > -1; i--) PetscCall(DMDestroy(&dms[i])); 10139566063dSJacob Faibussowitsch PetscCall(PetscFree(dms)); 1014ce4cda84SJed Brown } 101532fe681dSStefano Zampini } 1016cab2e9ccSBarry Smith 10172134b1e4SBarry Smith if (mg->galerkin < PC_MG_GALERKIN_NONE) { 10182134b1e4SBarry Smith Mat A, B; 10192134b1e4SBarry Smith PetscBool doA = PETSC_FALSE, doB = PETSC_FALSE; 10202134b1e4SBarry Smith MatReuse reuse = MAT_INITIAL_MATRIX; 10212134b1e4SBarry Smith 10222b3cbbdaSStefano Zampini if (mg->galerkin == PC_MG_GALERKIN_PMAT || mg->galerkin == PC_MG_GALERKIN_BOTH) doB = PETSC_TRUE; 10232b3cbbdaSStefano Zampini if (mg->galerkin == PC_MG_GALERKIN_MAT || (mg->galerkin == PC_MG_GALERKIN_BOTH && dA != dB)) doA = PETSC_TRUE; 10242134b1e4SBarry Smith if (pc->setupcalled) reuse = MAT_REUSE_MATRIX; 1025f3fbd535SBarry Smith for (i = n - 2; i > -1; i--) { 10267827d75bSBarry Smith PetscCheck(mglevels[i + 1]->restrct || mglevels[i + 1]->interpolate, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "Must provide interpolation or restriction for each MG level except level 0"); 102748a46eb9SPierre Jolivet if (!mglevels[i + 1]->interpolate) PetscCall(PCMGSetInterpolation(pc, i + 1, mglevels[i + 1]->restrct)); 102848a46eb9SPierre Jolivet if (!mglevels[i + 1]->restrct) PetscCall(PCMGSetRestriction(pc, i + 1, mglevels[i + 1]->interpolate)); 102948a46eb9SPierre Jolivet if (reuse == MAT_REUSE_MATRIX) PetscCall(KSPGetOperators(mglevels[i]->smoothd, &A, &B)); 103048a46eb9SPierre Jolivet if (doA) PetscCall(MatGalerkin(mglevels[i + 1]->restrct, dA, mglevels[i + 1]->interpolate, reuse, 1.0, &A)); 103148a46eb9SPierre Jolivet if (doB) PetscCall(MatGalerkin(mglevels[i + 1]->restrct, dB, mglevels[i + 1]->interpolate, reuse, 1.0, &B)); 10322134b1e4SBarry Smith /* the management of the PetscObjectReference() and PetscObjecDereference() below is rather delicate */ 10332134b1e4SBarry Smith if (!doA && dAeqdB) { 10349566063dSJacob Faibussowitsch if (reuse == MAT_INITIAL_MATRIX) PetscCall(PetscObjectReference((PetscObject)B)); 10352134b1e4SBarry Smith A = B; 10362134b1e4SBarry Smith } else if (!doA && reuse == MAT_INITIAL_MATRIX) { 10379566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[i]->smoothd, &A, NULL)); 10389566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)A)); 1039b9367914SBarry Smith } 10402134b1e4SBarry Smith if (!doB && dAeqdB) { 10419566063dSJacob Faibussowitsch if (reuse == MAT_INITIAL_MATRIX) PetscCall(PetscObjectReference((PetscObject)A)); 10422134b1e4SBarry Smith B = A; 10432134b1e4SBarry Smith } else if (!doB && reuse == MAT_INITIAL_MATRIX) { 10449566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[i]->smoothd, NULL, &B)); 10459566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)B)); 10467d537d92SJed Brown } 10472134b1e4SBarry Smith if (reuse == MAT_INITIAL_MATRIX) { 10489566063dSJacob Faibussowitsch PetscCall(KSPSetOperators(mglevels[i]->smoothd, A, B)); 10499566063dSJacob Faibussowitsch PetscCall(PetscObjectDereference((PetscObject)A)); 10509566063dSJacob Faibussowitsch PetscCall(PetscObjectDereference((PetscObject)B)); 10512134b1e4SBarry Smith } 10522134b1e4SBarry Smith dA = A; 1053f3fbd535SBarry Smith dB = B; 1054f3fbd535SBarry Smith } 1055f3fbd535SBarry Smith } 1056f3b08a26SMatthew G. Knepley 1057f3b08a26SMatthew G. Knepley /* Adapt interpolation matrices */ 10582b3cbbdaSStefano Zampini if (adaptInterpolation) { 1059f3b08a26SMatthew G. Knepley for (i = 0; i < n; ++i) { 106048a46eb9SPierre Jolivet if (!mglevels[i]->coarseSpace) PetscCall(PCMGComputeCoarseSpace_Internal(pc, i, mg->coarseSpaceType, mg->Nc, !i ? NULL : mglevels[i - 1]->coarseSpace, &mglevels[i]->coarseSpace)); 10612b3cbbdaSStefano Zampini if (i) PetscCall(PCMGAdaptInterpolator_Internal(pc, i, mglevels[i - 1]->smoothu, mglevels[i]->smoothu, mglevels[i - 1]->coarseSpace, mglevels[i]->coarseSpace)); 1062f3b08a26SMatthew G. Knepley } 106348a46eb9SPierre Jolivet for (i = n - 2; i > -1; --i) PetscCall(PCMGRecomputeLevelOperators_Internal(pc, i)); 1064f3b08a26SMatthew G. Knepley } 1065f3b08a26SMatthew G. Knepley 10662134b1e4SBarry Smith if (needRestricts && pc->dm) { 1067caa4e7f2SJed Brown for (i = n - 2; i >= 0; i--) { 1068ccceb50cSJed Brown DM dmfine, dmcoarse; 1069caa4e7f2SJed Brown Mat Restrict, Inject; 1070caa4e7f2SJed Brown Vec rscale; 10719566063dSJacob Faibussowitsch PetscCall(KSPGetDM(mglevels[i + 1]->smoothd, &dmfine)); 10729566063dSJacob Faibussowitsch PetscCall(KSPGetDM(mglevels[i]->smoothd, &dmcoarse)); 10739566063dSJacob Faibussowitsch PetscCall(PCMGGetRestriction(pc, i + 1, &Restrict)); 10749566063dSJacob Faibussowitsch PetscCall(PCMGGetRScale(pc, i + 1, &rscale)); 10759566063dSJacob Faibussowitsch PetscCall(PCMGGetInjection(pc, i + 1, &Inject)); 10769566063dSJacob Faibussowitsch PetscCall(DMRestrict(dmfine, Restrict, rscale, Inject, dmcoarse)); 1077caa4e7f2SJed Brown } 1078caa4e7f2SJed Brown } 1079f3fbd535SBarry Smith 1080f3fbd535SBarry Smith if (!pc->setupcalled) { 108148a46eb9SPierre Jolivet for (i = 0; i < n; i++) PetscCall(KSPSetFromOptions(mglevels[i]->smoothd)); 1082f3fbd535SBarry Smith for (i = 1; i < n; i++) { 108348a46eb9SPierre Jolivet if (mglevels[i]->smoothu && (mglevels[i]->smoothu != mglevels[i]->smoothd)) PetscCall(KSPSetFromOptions(mglevels[i]->smoothu)); 108448a46eb9SPierre Jolivet if (mglevels[i]->cr) PetscCall(KSPSetFromOptions(mglevels[i]->cr)); 1085f3fbd535SBarry Smith } 10863ad4599aSBarry Smith /* insure that if either interpolation or restriction is set the other other one is set */ 1087f3fbd535SBarry Smith for (i = 1; i < n; i++) { 10889566063dSJacob Faibussowitsch PetscCall(PCMGGetInterpolation(pc, i, NULL)); 10899566063dSJacob Faibussowitsch PetscCall(PCMGGetRestriction(pc, i, NULL)); 1090f3fbd535SBarry Smith } 1091f3fbd535SBarry Smith for (i = 0; i < n - 1; i++) { 1092f3fbd535SBarry Smith if (!mglevels[i]->b) { 1093f3fbd535SBarry Smith Vec *vec; 10949566063dSJacob Faibussowitsch PetscCall(KSPCreateVecs(mglevels[i]->smoothd, 1, &vec, 0, NULL)); 10959566063dSJacob Faibussowitsch PetscCall(PCMGSetRhs(pc, i, *vec)); 10969566063dSJacob Faibussowitsch PetscCall(VecDestroy(vec)); 10979566063dSJacob Faibussowitsch PetscCall(PetscFree(vec)); 1098f3fbd535SBarry Smith } 1099f3fbd535SBarry Smith if (!mglevels[i]->r && i) { 11009566063dSJacob Faibussowitsch PetscCall(VecDuplicate(mglevels[i]->b, &tvec)); 11019566063dSJacob Faibussowitsch PetscCall(PCMGSetR(pc, i, tvec)); 11029566063dSJacob Faibussowitsch PetscCall(VecDestroy(&tvec)); 1103f3fbd535SBarry Smith } 1104f3fbd535SBarry Smith if (!mglevels[i]->x) { 11059566063dSJacob Faibussowitsch PetscCall(VecDuplicate(mglevels[i]->b, &tvec)); 11069566063dSJacob Faibussowitsch PetscCall(PCMGSetX(pc, i, tvec)); 11079566063dSJacob Faibussowitsch PetscCall(VecDestroy(&tvec)); 1108f3fbd535SBarry Smith } 110941b6fd38SMatthew G. Knepley if (doCR) { 11109566063dSJacob Faibussowitsch PetscCall(VecDuplicate(mglevels[i]->b, &mglevels[i]->crx)); 11119566063dSJacob Faibussowitsch PetscCall(VecDuplicate(mglevels[i]->b, &mglevels[i]->crb)); 11129566063dSJacob Faibussowitsch PetscCall(VecZeroEntries(mglevels[i]->crb)); 111341b6fd38SMatthew G. Knepley } 1114f3fbd535SBarry Smith } 1115f3fbd535SBarry Smith if (n != 1 && !mglevels[n - 1]->r) { 1116f3fbd535SBarry Smith /* PCMGSetR() on the finest level if user did not supply it */ 1117f3fbd535SBarry Smith Vec *vec; 11189566063dSJacob Faibussowitsch PetscCall(KSPCreateVecs(mglevels[n - 1]->smoothd, 1, &vec, 0, NULL)); 11199566063dSJacob Faibussowitsch PetscCall(PCMGSetR(pc, n - 1, *vec)); 11209566063dSJacob Faibussowitsch PetscCall(VecDestroy(vec)); 11219566063dSJacob Faibussowitsch PetscCall(PetscFree(vec)); 1122f3fbd535SBarry Smith } 112341b6fd38SMatthew G. Knepley if (doCR) { 11249566063dSJacob Faibussowitsch PetscCall(VecDuplicate(mglevels[n - 1]->r, &mglevels[n - 1]->crx)); 11259566063dSJacob Faibussowitsch PetscCall(VecDuplicate(mglevels[n - 1]->r, &mglevels[n - 1]->crb)); 11269566063dSJacob Faibussowitsch PetscCall(VecZeroEntries(mglevels[n - 1]->crb)); 112741b6fd38SMatthew G. Knepley } 1128f3fbd535SBarry Smith } 1129f3fbd535SBarry Smith 113098e04984SBarry Smith if (pc->dm) { 113198e04984SBarry Smith /* need to tell all the coarser levels to rebuild the matrix using the DM for that level */ 113298e04984SBarry Smith for (i = 0; i < n - 1; i++) { 113398e04984SBarry Smith if (mglevels[i]->smoothd->setupstage != KSP_SETUP_NEW) mglevels[i]->smoothd->setupstage = KSP_SETUP_NEWMATRIX; 113498e04984SBarry Smith } 113598e04984SBarry Smith } 113608549ed4SJed Brown // We got here (PCSetUp_MG) because the matrix has changed, which means the smoother needs to be set up again (e.g., 113708549ed4SJed Brown // new diagonal for Jacobi). Setting it here allows it to be logged under PCSetUp rather than deep inside a PCApply. 113808549ed4SJed Brown if (mglevels[n - 1]->smoothd->setupstage != KSP_SETUP_NEW) mglevels[n - 1]->smoothd->setupstage = KSP_SETUP_NEWMATRIX; 1139f3fbd535SBarry Smith 1140f3fbd535SBarry Smith for (i = 1; i < n; i++) { 11412a21e185SBarry Smith if (mglevels[i]->smoothu == mglevels[i]->smoothd || mg->am == PC_MG_FULL || mg->am == PC_MG_KASKADE || mg->cyclesperpcapply > 1) { 1142f3fbd535SBarry Smith /* if doing only down then initial guess is zero */ 11439566063dSJacob Faibussowitsch PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->smoothd, PETSC_TRUE)); 1144f3fbd535SBarry Smith } 11459566063dSJacob Faibussowitsch if (mglevels[i]->cr) PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->cr, PETSC_TRUE)); 11469566063dSJacob Faibussowitsch if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventBegin(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0)); 11479566063dSJacob Faibussowitsch PetscCall(KSPSetUp(mglevels[i]->smoothd)); 1148d4645a75SStefano Zampini if (mglevels[i]->smoothd->reason) pc->failedreason = PC_SUBPC_ERROR; 11499566063dSJacob Faibussowitsch if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventEnd(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0)); 1150d42688cbSBarry Smith if (!mglevels[i]->residual) { 1151d42688cbSBarry Smith Mat mat; 11529566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[i]->smoothd, &mat, NULL)); 11539566063dSJacob Faibussowitsch PetscCall(PCMGSetResidual(pc, i, PCMGResidualDefault, mat)); 1154d42688cbSBarry Smith } 1155fcb023d4SJed Brown if (!mglevels[i]->residualtranspose) { 1156fcb023d4SJed Brown Mat mat; 11579566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[i]->smoothd, &mat, NULL)); 11589566063dSJacob Faibussowitsch PetscCall(PCMGSetResidualTranspose(pc, i, PCMGResidualTransposeDefault, mat)); 1159fcb023d4SJed Brown } 1160f3fbd535SBarry Smith } 1161f3fbd535SBarry Smith for (i = 1; i < n; i++) { 1162f3fbd535SBarry Smith if (mglevels[i]->smoothu && mglevels[i]->smoothu != mglevels[i]->smoothd) { 1163f3fbd535SBarry Smith Mat downmat, downpmat; 1164f3fbd535SBarry Smith 1165f3fbd535SBarry Smith /* check if operators have been set for up, if not use down operators to set them */ 11669566063dSJacob Faibussowitsch PetscCall(KSPGetOperatorsSet(mglevels[i]->smoothu, &opsset, NULL)); 1167f3fbd535SBarry Smith if (!opsset) { 11689566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[i]->smoothd, &downmat, &downpmat)); 11699566063dSJacob Faibussowitsch PetscCall(KSPSetOperators(mglevels[i]->smoothu, downmat, downpmat)); 1170f3fbd535SBarry Smith } 1171f3fbd535SBarry Smith 11729566063dSJacob Faibussowitsch PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->smoothu, PETSC_TRUE)); 11739566063dSJacob Faibussowitsch if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventBegin(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0)); 11749566063dSJacob Faibussowitsch PetscCall(KSPSetUp(mglevels[i]->smoothu)); 1175d4645a75SStefano Zampini if (mglevels[i]->smoothu->reason) pc->failedreason = PC_SUBPC_ERROR; 11769566063dSJacob Faibussowitsch if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventEnd(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0)); 1177f3fbd535SBarry Smith } 117841b6fd38SMatthew G. Knepley if (mglevels[i]->cr) { 117941b6fd38SMatthew G. Knepley Mat downmat, downpmat; 118041b6fd38SMatthew G. Knepley 118141b6fd38SMatthew G. Knepley /* check if operators have been set for up, if not use down operators to set them */ 11829566063dSJacob Faibussowitsch PetscCall(KSPGetOperatorsSet(mglevels[i]->cr, &opsset, NULL)); 118341b6fd38SMatthew G. Knepley if (!opsset) { 11849566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[i]->smoothd, &downmat, &downpmat)); 11859566063dSJacob Faibussowitsch PetscCall(KSPSetOperators(mglevels[i]->cr, downmat, downpmat)); 118641b6fd38SMatthew G. Knepley } 118741b6fd38SMatthew G. Knepley 11889566063dSJacob Faibussowitsch PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->cr, PETSC_TRUE)); 11899566063dSJacob Faibussowitsch if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventBegin(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0)); 11909566063dSJacob Faibussowitsch PetscCall(KSPSetUp(mglevels[i]->cr)); 1191d4645a75SStefano Zampini if (mglevels[i]->cr->reason) pc->failedreason = PC_SUBPC_ERROR; 11929566063dSJacob Faibussowitsch if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventEnd(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0)); 119341b6fd38SMatthew G. Knepley } 1194f3fbd535SBarry Smith } 1195f3fbd535SBarry Smith 11969566063dSJacob Faibussowitsch if (mglevels[0]->eventsmoothsetup) PetscCall(PetscLogEventBegin(mglevels[0]->eventsmoothsetup, 0, 0, 0, 0)); 11979566063dSJacob Faibussowitsch PetscCall(KSPSetUp(mglevels[0]->smoothd)); 1198d4645a75SStefano Zampini if (mglevels[0]->smoothd->reason) pc->failedreason = PC_SUBPC_ERROR; 11999566063dSJacob Faibussowitsch if (mglevels[0]->eventsmoothsetup) PetscCall(PetscLogEventEnd(mglevels[0]->eventsmoothsetup, 0, 0, 0, 0)); 1200f3fbd535SBarry Smith 1201f3fbd535SBarry Smith /* 1202f3fbd535SBarry Smith Dump the interpolation/restriction matrices plus the 1203e3c5b3baSBarry Smith Jacobian/stiffness on each level. This allows MATLAB users to 1204f3fbd535SBarry Smith easily check if the Galerkin condition A_c = R A_f R^T is satisfied. 1205f3fbd535SBarry Smith 1206f3fbd535SBarry Smith Only support one or the other at the same time. 1207f3fbd535SBarry Smith */ 1208f3fbd535SBarry Smith #if defined(PETSC_USE_SOCKET_VIEWER) 12099566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(((PetscObject)pc)->options, ((PetscObject)pc)->prefix, "-pc_mg_dump_matlab", &dump, NULL)); 1210ce94432eSBarry Smith if (dump) viewer = PETSC_VIEWER_SOCKET_(PetscObjectComm((PetscObject)pc)); 1211f3fbd535SBarry Smith dump = PETSC_FALSE; 1212f3fbd535SBarry Smith #endif 12139566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(((PetscObject)pc)->options, ((PetscObject)pc)->prefix, "-pc_mg_dump_binary", &dump, NULL)); 1214ce94432eSBarry Smith if (dump) viewer = PETSC_VIEWER_BINARY_(PetscObjectComm((PetscObject)pc)); 1215f3fbd535SBarry Smith 1216f3fbd535SBarry Smith if (viewer) { 121748a46eb9SPierre Jolivet for (i = 1; i < n; i++) PetscCall(MatView(mglevels[i]->restrct, viewer)); 1218f3fbd535SBarry Smith for (i = 0; i < n; i++) { 12199566063dSJacob Faibussowitsch PetscCall(KSPGetPC(mglevels[i]->smoothd, &pc)); 12209566063dSJacob Faibussowitsch PetscCall(MatView(pc->mat, viewer)); 1221f3fbd535SBarry Smith } 1222f3fbd535SBarry Smith } 12233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1224f3fbd535SBarry Smith } 1225f3fbd535SBarry Smith 1226f3fbd535SBarry Smith /* -------------------------------------------------------------------------------------*/ 1227f3fbd535SBarry Smith 1228d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetLevels_MG(PC pc, PetscInt *levels) 1229d71ae5a4SJacob Faibussowitsch { 123097d33e41SMatthew G. Knepley PC_MG *mg = (PC_MG *)pc->data; 123197d33e41SMatthew G. Knepley 123297d33e41SMatthew G. Knepley PetscFunctionBegin; 123397d33e41SMatthew G. Knepley *levels = mg->nlevels; 12343ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 123597d33e41SMatthew G. Knepley } 123697d33e41SMatthew G. Knepley 12374b9ad928SBarry Smith /*@ 1238f1580f4eSBarry Smith PCMGGetLevels - Gets the number of levels to use with `PCMG`. 12394b9ad928SBarry Smith 12404b9ad928SBarry Smith Not Collective 12414b9ad928SBarry Smith 12424b9ad928SBarry Smith Input Parameter: 12434b9ad928SBarry Smith . pc - the preconditioner context 12444b9ad928SBarry Smith 1245feefa0e1SJacob Faibussowitsch Output Parameter: 12464b9ad928SBarry Smith . levels - the number of levels 12474b9ad928SBarry Smith 12484b9ad928SBarry Smith Level: advanced 12494b9ad928SBarry Smith 1250f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetLevels()` 12514b9ad928SBarry Smith @*/ 1252d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetLevels(PC pc, PetscInt *levels) 1253d71ae5a4SJacob Faibussowitsch { 12544b9ad928SBarry Smith PetscFunctionBegin; 12550700a824SBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 12564f572ea9SToby Isaac PetscAssertPointer(levels, 2); 125797d33e41SMatthew G. Knepley *levels = 0; 1258cac4c232SBarry Smith PetscTryMethod(pc, "PCMGGetLevels_C", (PC, PetscInt *), (pc, levels)); 12593ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 12604b9ad928SBarry Smith } 12614b9ad928SBarry Smith 12624b9ad928SBarry Smith /*@ 1263f1580f4eSBarry Smith PCMGGetGridComplexity - compute operator and grid complexity of the `PCMG` hierarchy 1264e7d4b4cbSMark Adams 1265e7d4b4cbSMark Adams Input Parameter: 1266e7d4b4cbSMark Adams . pc - the preconditioner context 1267e7d4b4cbSMark Adams 126890db8557SMark Adams Output Parameters: 126990db8557SMark Adams + gc - grid complexity = sum_i(n_i) / n_0 127090db8557SMark Adams - oc - operator complexity = sum_i(nnz_i) / nnz_0 1271e7d4b4cbSMark Adams 1272e7d4b4cbSMark Adams Level: advanced 1273e7d4b4cbSMark Adams 1274f1580f4eSBarry Smith Note: 1275f1580f4eSBarry Smith This is often call the operator complexity in multigrid literature 1276f1580f4eSBarry Smith 1277f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetLevels()`, `PCMGSetLevels()` 1278e7d4b4cbSMark Adams @*/ 1279d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetGridComplexity(PC pc, PetscReal *gc, PetscReal *oc) 1280d71ae5a4SJacob Faibussowitsch { 1281e7d4b4cbSMark Adams PC_MG *mg = (PC_MG *)pc->data; 1282e7d4b4cbSMark Adams PC_MG_Levels **mglevels = mg->levels; 1283e7d4b4cbSMark Adams PetscInt lev, N; 1284e7d4b4cbSMark Adams PetscLogDouble nnz0 = 0, sgc = 0, soc = 0, n0 = 0; 1285e7d4b4cbSMark Adams MatInfo info; 1286e7d4b4cbSMark Adams 1287e7d4b4cbSMark Adams PetscFunctionBegin; 128890db8557SMark Adams PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 12894f572ea9SToby Isaac if (gc) PetscAssertPointer(gc, 2); 12904f572ea9SToby Isaac if (oc) PetscAssertPointer(oc, 3); 1291e7d4b4cbSMark Adams if (!pc->setupcalled) { 129290db8557SMark Adams if (gc) *gc = 0; 129390db8557SMark Adams if (oc) *oc = 0; 12943ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1295e7d4b4cbSMark Adams } 1296e7d4b4cbSMark Adams PetscCheck(mg->nlevels > 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "MG has no levels"); 1297e7d4b4cbSMark Adams for (lev = 0; lev < mg->nlevels; lev++) { 1298e7d4b4cbSMark Adams Mat dB; 12999566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[lev]->smoothd, NULL, &dB)); 13009566063dSJacob Faibussowitsch PetscCall(MatGetInfo(dB, MAT_GLOBAL_SUM, &info)); /* global reduction */ 13019566063dSJacob Faibussowitsch PetscCall(MatGetSize(dB, &N, NULL)); 1302e7d4b4cbSMark Adams sgc += N; 1303e7d4b4cbSMark Adams soc += info.nz_used; 13049371c9d4SSatish Balay if (lev == mg->nlevels - 1) { 13059371c9d4SSatish Balay nnz0 = info.nz_used; 13069371c9d4SSatish Balay n0 = N; 13079371c9d4SSatish Balay } 1308e7d4b4cbSMark Adams } 13090fdf79fbSJacob Faibussowitsch PetscCheck(n0 > 0 && gc, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number for grid points on finest level is not available"); 13100fdf79fbSJacob Faibussowitsch *gc = (PetscReal)(sgc / n0); 131190db8557SMark Adams if (nnz0 > 0 && oc) *oc = (PetscReal)(soc / nnz0); 13123ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1313e7d4b4cbSMark Adams } 1314e7d4b4cbSMark Adams 1315e7d4b4cbSMark Adams /*@ 131697177400SBarry Smith PCMGSetType - Determines the form of multigrid to use: 13174b9ad928SBarry Smith multiplicative, additive, full, or the Kaskade algorithm. 13184b9ad928SBarry Smith 1319c3339decSBarry Smith Logically Collective 13204b9ad928SBarry Smith 13214b9ad928SBarry Smith Input Parameters: 13224b9ad928SBarry Smith + pc - the preconditioner context 1323f1580f4eSBarry Smith - form - multigrid form, one of `PC_MG_MULTIPLICATIVE`, `PC_MG_ADDITIVE`, `PC_MG_FULL`, `PC_MG_KASKADE` 13244b9ad928SBarry Smith 13254b9ad928SBarry Smith Options Database Key: 132620f4b53cSBarry Smith . -pc_mg_type <form> - Sets <form>, one of multiplicative, additive, full, kaskade 13274b9ad928SBarry Smith 13284b9ad928SBarry Smith Level: advanced 13294b9ad928SBarry Smith 1330f1580f4eSBarry Smith .seealso: `PCMGType`, `PCMG`, `PCMGGetLevels()`, `PCMGSetLevels()`, `PCMGGetType()`, `PCMGCycleType` 13314b9ad928SBarry Smith @*/ 1332d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetType(PC pc, PCMGType form) 1333d71ae5a4SJacob Faibussowitsch { 1334f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 13354b9ad928SBarry Smith 13364b9ad928SBarry Smith PetscFunctionBegin; 13370700a824SBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1338c5eb9154SBarry Smith PetscValidLogicalCollectiveEnum(pc, form, 2); 133931567311SBarry Smith mg->am = form; 13409dcbbd2bSBarry Smith if (form == PC_MG_MULTIPLICATIVE) pc->ops->applyrichardson = PCApplyRichardson_MG; 1341c60c7ad4SBarry Smith else pc->ops->applyrichardson = NULL; 13423ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1343c60c7ad4SBarry Smith } 1344c60c7ad4SBarry Smith 1345c60c7ad4SBarry Smith /*@ 1346f1580f4eSBarry Smith PCMGGetType - Finds the form of multigrid the `PCMG` is using multiplicative, additive, full, or the Kaskade algorithm. 1347c60c7ad4SBarry Smith 1348c3339decSBarry Smith Logically Collective 1349c60c7ad4SBarry Smith 1350c60c7ad4SBarry Smith Input Parameter: 1351c60c7ad4SBarry Smith . pc - the preconditioner context 1352c60c7ad4SBarry Smith 1353c60c7ad4SBarry Smith Output Parameter: 1354f1580f4eSBarry Smith . type - one of `PC_MG_MULTIPLICATIVE`, `PC_MG_ADDITIVE`, `PC_MG_FULL`, `PC_MG_KASKADE`, `PCMGCycleType` 1355c60c7ad4SBarry Smith 1356c60c7ad4SBarry Smith Level: advanced 1357c60c7ad4SBarry Smith 1358f1580f4eSBarry Smith .seealso: `PCMGType`, `PCMG`, `PCMGGetLevels()`, `PCMGSetLevels()`, `PCMGSetType()` 1359c60c7ad4SBarry Smith @*/ 1360d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetType(PC pc, PCMGType *type) 1361d71ae5a4SJacob Faibussowitsch { 1362c60c7ad4SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 1363c60c7ad4SBarry Smith 1364c60c7ad4SBarry Smith PetscFunctionBegin; 1365c60c7ad4SBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1366c60c7ad4SBarry Smith *type = mg->am; 13673ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 13684b9ad928SBarry Smith } 13694b9ad928SBarry Smith 13704b9ad928SBarry Smith /*@ 1371f1580f4eSBarry Smith PCMGSetCycleType - Sets the type cycles to use. Use `PCMGSetCycleTypeOnLevel()` for more 13724b9ad928SBarry Smith complicated cycling. 13734b9ad928SBarry Smith 1374c3339decSBarry Smith Logically Collective 13754b9ad928SBarry Smith 13764b9ad928SBarry Smith Input Parameters: 1377c2be2410SBarry Smith + pc - the multigrid context 1378f1580f4eSBarry Smith - n - either `PC_MG_CYCLE_V` or `PC_MG_CYCLE_W` 13794b9ad928SBarry Smith 13804b9ad928SBarry Smith Options Database Key: 1381c1cbb1deSBarry Smith . -pc_mg_cycle_type <v,w> - provide the cycle desired 13824b9ad928SBarry Smith 13834b9ad928SBarry Smith Level: advanced 13844b9ad928SBarry Smith 1385f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetCycleTypeOnLevel()`, `PCMGType`, `PCMGCycleType` 13864b9ad928SBarry Smith @*/ 1387d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetCycleType(PC pc, PCMGCycleType n) 1388d71ae5a4SJacob Faibussowitsch { 1389f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 1390f3fbd535SBarry Smith PC_MG_Levels **mglevels = mg->levels; 139179416396SBarry Smith PetscInt i, levels; 13924b9ad928SBarry Smith 13934b9ad928SBarry Smith PetscFunctionBegin; 13940700a824SBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 139569fbec6eSBarry Smith PetscValidLogicalCollectiveEnum(pc, n, 2); 139628b400f6SJacob Faibussowitsch PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ORDER, "Must set MG levels with PCMGSetLevels() before calling"); 1397f3fbd535SBarry Smith levels = mglevels[0]->levels; 13982fa5cd67SKarl Rupp for (i = 0; i < levels; i++) mglevels[i]->cycles = n; 13993ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 14004b9ad928SBarry Smith } 14014b9ad928SBarry Smith 14028cc2d5dfSBarry Smith /*@ 14038cc2d5dfSBarry Smith PCMGMultiplicativeSetCycles - Sets the number of cycles to use for each preconditioner step 1404f1580f4eSBarry Smith of multigrid when `PCMGType` is `PC_MG_MULTIPLICATIVE` 14058cc2d5dfSBarry Smith 1406c3339decSBarry Smith Logically Collective 14078cc2d5dfSBarry Smith 14088cc2d5dfSBarry Smith Input Parameters: 14098cc2d5dfSBarry Smith + pc - the multigrid context 14108cc2d5dfSBarry Smith - n - number of cycles (default is 1) 14118cc2d5dfSBarry Smith 14128cc2d5dfSBarry Smith Options Database Key: 1413147403d9SBarry Smith . -pc_mg_multiplicative_cycles n - set the number of cycles 14148cc2d5dfSBarry Smith 14158cc2d5dfSBarry Smith Level: advanced 14168cc2d5dfSBarry Smith 1417f1580f4eSBarry Smith Note: 1418f1580f4eSBarry Smith This is not associated with setting a v or w cycle, that is set with `PCMGSetCycleType()` 14198cc2d5dfSBarry Smith 1420f1580f4eSBarry Smith .seealso: `PCMGSetCycleTypeOnLevel()`, `PCMGSetCycleType()`, `PCMGCycleType`, `PCMGType` 14218cc2d5dfSBarry Smith @*/ 1422d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGMultiplicativeSetCycles(PC pc, PetscInt n) 1423d71ae5a4SJacob Faibussowitsch { 1424f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 14258cc2d5dfSBarry Smith 14268cc2d5dfSBarry Smith PetscFunctionBegin; 14270700a824SBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1428c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(pc, n, 2); 14292a21e185SBarry Smith mg->cyclesperpcapply = n; 14303ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 14318cc2d5dfSBarry Smith } 14328cc2d5dfSBarry Smith 1433d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetGalerkin_MG(PC pc, PCMGGalerkinType use) 1434d71ae5a4SJacob Faibussowitsch { 1435fb15c04eSBarry Smith PC_MG *mg = (PC_MG *)pc->data; 1436fb15c04eSBarry Smith 1437fb15c04eSBarry Smith PetscFunctionBegin; 14382134b1e4SBarry Smith mg->galerkin = use; 14393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1440fb15c04eSBarry Smith } 1441fb15c04eSBarry Smith 1442c2be2410SBarry Smith /*@ 144397177400SBarry Smith PCMGSetGalerkin - Causes the coarser grid matrices to be computed from the 144482b87d87SMatthew G. Knepley finest grid via the Galerkin process: A_i-1 = r_i * A_i * p_i 1445c2be2410SBarry Smith 1446c3339decSBarry Smith Logically Collective 1447c2be2410SBarry Smith 1448c2be2410SBarry Smith Input Parameters: 1449c91913d3SJed Brown + pc - the multigrid context 1450f1580f4eSBarry Smith - use - one of `PC_MG_GALERKIN_BOTH`, `PC_MG_GALERKIN_PMAT`, `PC_MG_GALERKIN_MAT`, or `PC_MG_GALERKIN_NONE` 1451c2be2410SBarry Smith 1452c2be2410SBarry Smith Options Database Key: 1453147403d9SBarry Smith . -pc_mg_galerkin <both,pmat,mat,none> - set the matrices to form via the Galerkin process 1454c2be2410SBarry Smith 1455c2be2410SBarry Smith Level: intermediate 1456c2be2410SBarry Smith 1457f1580f4eSBarry Smith Note: 1458f1580f4eSBarry Smith Some codes that use `PCMG` such as `PCGAMG` use Galerkin internally while constructing the hierarchy and thus do not 1459f1580f4eSBarry Smith use the `PCMG` construction of the coarser grids. 14601c1aac46SBarry Smith 1461f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetGalerkin()`, `PCMGGalerkinType` 1462c2be2410SBarry Smith @*/ 1463d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetGalerkin(PC pc, PCMGGalerkinType use) 1464d71ae5a4SJacob Faibussowitsch { 1465c2be2410SBarry Smith PetscFunctionBegin; 14660700a824SBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1467cac4c232SBarry Smith PetscTryMethod(pc, "PCMGSetGalerkin_C", (PC, PCMGGalerkinType), (pc, use)); 14683ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1469c2be2410SBarry Smith } 1470c2be2410SBarry Smith 14713fc8bf9cSBarry Smith /*@ 1472f1580f4eSBarry Smith PCMGGetGalerkin - Checks if Galerkin multigrid is being used, i.e. A_i-1 = r_i * A_i * p_i 14733fc8bf9cSBarry Smith 14743fc8bf9cSBarry Smith Not Collective 14753fc8bf9cSBarry Smith 14763fc8bf9cSBarry Smith Input Parameter: 14773fc8bf9cSBarry Smith . pc - the multigrid context 14783fc8bf9cSBarry Smith 14793fc8bf9cSBarry Smith Output Parameter: 1480f1580f4eSBarry Smith . galerkin - one of `PC_MG_GALERKIN_BOTH`,`PC_MG_GALERKIN_PMAT`,`PC_MG_GALERKIN_MAT`, `PC_MG_GALERKIN_NONE`, or `PC_MG_GALERKIN_EXTERNAL` 14813fc8bf9cSBarry Smith 14823fc8bf9cSBarry Smith Level: intermediate 14833fc8bf9cSBarry Smith 1484f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetGalerkin()`, `PCMGGalerkinType` 14853fc8bf9cSBarry Smith @*/ 1486d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetGalerkin(PC pc, PCMGGalerkinType *galerkin) 1487d71ae5a4SJacob Faibussowitsch { 1488f3fbd535SBarry Smith PC_MG *mg = (PC_MG *)pc->data; 14893fc8bf9cSBarry Smith 14903fc8bf9cSBarry Smith PetscFunctionBegin; 14910700a824SBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 14922134b1e4SBarry Smith *galerkin = mg->galerkin; 14933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 14943fc8bf9cSBarry Smith } 14953fc8bf9cSBarry Smith 1496d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptInterpolation_MG(PC pc, PetscBool adapt) 1497d71ae5a4SJacob Faibussowitsch { 1498f3b08a26SMatthew G. Knepley PC_MG *mg = (PC_MG *)pc->data; 1499f3b08a26SMatthew G. Knepley 1500f3b08a26SMatthew G. Knepley PetscFunctionBegin; 1501f3b08a26SMatthew G. Knepley mg->adaptInterpolation = adapt; 15023ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1503f3b08a26SMatthew G. Knepley } 1504f3b08a26SMatthew G. Knepley 1505d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptInterpolation_MG(PC pc, PetscBool *adapt) 1506d71ae5a4SJacob Faibussowitsch { 1507f3b08a26SMatthew G. Knepley PC_MG *mg = (PC_MG *)pc->data; 1508f3b08a26SMatthew G. Knepley 1509f3b08a26SMatthew G. Knepley PetscFunctionBegin; 1510f3b08a26SMatthew G. Knepley *adapt = mg->adaptInterpolation; 15113ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1512f3b08a26SMatthew G. Knepley } 1513f3b08a26SMatthew G. Knepley 1514d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptCoarseSpaceType_MG(PC pc, PCMGCoarseSpaceType ctype) 1515d71ae5a4SJacob Faibussowitsch { 15162b3cbbdaSStefano Zampini PC_MG *mg = (PC_MG *)pc->data; 15172b3cbbdaSStefano Zampini 15182b3cbbdaSStefano Zampini PetscFunctionBegin; 15192b3cbbdaSStefano Zampini mg->adaptInterpolation = ctype != PCMG_ADAPT_NONE ? PETSC_TRUE : PETSC_FALSE; 15202b3cbbdaSStefano Zampini mg->coarseSpaceType = ctype; 15213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 15222b3cbbdaSStefano Zampini } 15232b3cbbdaSStefano Zampini 1524d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptCoarseSpaceType_MG(PC pc, PCMGCoarseSpaceType *ctype) 1525d71ae5a4SJacob Faibussowitsch { 15262b3cbbdaSStefano Zampini PC_MG *mg = (PC_MG *)pc->data; 15272b3cbbdaSStefano Zampini 15282b3cbbdaSStefano Zampini PetscFunctionBegin; 15292b3cbbdaSStefano Zampini *ctype = mg->coarseSpaceType; 15303ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 15312b3cbbdaSStefano Zampini } 15322b3cbbdaSStefano Zampini 1533d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptCR_MG(PC pc, PetscBool cr) 1534d71ae5a4SJacob Faibussowitsch { 153541b6fd38SMatthew G. Knepley PC_MG *mg = (PC_MG *)pc->data; 153641b6fd38SMatthew G. Knepley 153741b6fd38SMatthew G. Knepley PetscFunctionBegin; 153841b6fd38SMatthew G. Knepley mg->compatibleRelaxation = cr; 15393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 154041b6fd38SMatthew G. Knepley } 154141b6fd38SMatthew G. Knepley 1542d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptCR_MG(PC pc, PetscBool *cr) 1543d71ae5a4SJacob Faibussowitsch { 154441b6fd38SMatthew G. Knepley PC_MG *mg = (PC_MG *)pc->data; 154541b6fd38SMatthew G. Knepley 154641b6fd38SMatthew G. Knepley PetscFunctionBegin; 154741b6fd38SMatthew G. Knepley *cr = mg->compatibleRelaxation; 15483ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 154941b6fd38SMatthew G. Knepley } 155041b6fd38SMatthew G. Knepley 15512b3cbbdaSStefano Zampini /*@C 15522b3cbbdaSStefano Zampini PCMGSetAdaptCoarseSpaceType - Set the type of adaptive coarse space. 15532b3cbbdaSStefano Zampini 15542b3cbbdaSStefano Zampini Adapts or creates the interpolator based upon a vector space which should be accurately captured by the next coarser mesh, and thus accurately interpolated. 15552b3cbbdaSStefano Zampini 1556c3339decSBarry Smith Logically Collective 15572b3cbbdaSStefano Zampini 15582b3cbbdaSStefano Zampini Input Parameters: 15592b3cbbdaSStefano Zampini + pc - the multigrid context 15602b3cbbdaSStefano Zampini - ctype - the type of coarse space 15612b3cbbdaSStefano Zampini 15622b3cbbdaSStefano Zampini Options Database Keys: 15632b3cbbdaSStefano Zampini + -pc_mg_adapt_interp_n <int> - The number of modes to use 15642b3cbbdaSStefano Zampini - -pc_mg_adapt_interp_coarse_space <type> - The type of coarse space: none, polynomial, harmonic, eigenvector, generalized_eigenvector, gdsw 15652b3cbbdaSStefano Zampini 15662b3cbbdaSStefano Zampini Level: intermediate 15672b3cbbdaSStefano Zampini 1568f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGCoarseSpaceType`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetGalerkin()`, `PCMGSetAdaptInterpolation()` 15692b3cbbdaSStefano Zampini @*/ 1570d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptCoarseSpaceType(PC pc, PCMGCoarseSpaceType ctype) 1571d71ae5a4SJacob Faibussowitsch { 15722b3cbbdaSStefano Zampini PetscFunctionBegin; 15732b3cbbdaSStefano Zampini PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 15742b3cbbdaSStefano Zampini PetscValidLogicalCollectiveEnum(pc, ctype, 2); 15752b3cbbdaSStefano Zampini PetscTryMethod(pc, "PCMGSetAdaptCoarseSpaceType_C", (PC, PCMGCoarseSpaceType), (pc, ctype)); 15763ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 15772b3cbbdaSStefano Zampini } 15782b3cbbdaSStefano Zampini 15792b3cbbdaSStefano Zampini /*@C 15802b3cbbdaSStefano Zampini PCMGGetAdaptCoarseSpaceType - Get the type of adaptive coarse space. 15812b3cbbdaSStefano Zampini 15822b3cbbdaSStefano Zampini Not Collective 15832b3cbbdaSStefano Zampini 15842b3cbbdaSStefano Zampini Input Parameter: 15852b3cbbdaSStefano Zampini . pc - the multigrid context 15862b3cbbdaSStefano Zampini 15872b3cbbdaSStefano Zampini Output Parameter: 15882b3cbbdaSStefano Zampini . ctype - the type of coarse space 15892b3cbbdaSStefano Zampini 15902b3cbbdaSStefano Zampini Level: intermediate 15912b3cbbdaSStefano Zampini 1592f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGCoarseSpaceType`, `PCMGSetAdaptCoarseSpaceType()`, `PCMGSetGalerkin()`, `PCMGSetAdaptInterpolation()` 15932b3cbbdaSStefano Zampini @*/ 1594d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptCoarseSpaceType(PC pc, PCMGCoarseSpaceType *ctype) 1595d71ae5a4SJacob Faibussowitsch { 15962b3cbbdaSStefano Zampini PetscFunctionBegin; 15972b3cbbdaSStefano Zampini PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 15984f572ea9SToby Isaac PetscAssertPointer(ctype, 2); 15992b3cbbdaSStefano Zampini PetscUseMethod(pc, "PCMGGetAdaptCoarseSpaceType_C", (PC, PCMGCoarseSpaceType *), (pc, ctype)); 16003ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 16012b3cbbdaSStefano Zampini } 16022b3cbbdaSStefano Zampini 16032b3cbbdaSStefano Zampini /* MATT: REMOVE? */ 1604f3b08a26SMatthew G. Knepley /*@ 1605f3b08a26SMatthew G. Knepley PCMGSetAdaptInterpolation - Adapt the interpolator based upon a vector space which should be accurately captured by the next coarser mesh, and thus accurately interpolated. 1606f3b08a26SMatthew G. Knepley 1607c3339decSBarry Smith Logically Collective 1608f3b08a26SMatthew G. Knepley 1609f3b08a26SMatthew G. Knepley Input Parameters: 1610f3b08a26SMatthew G. Knepley + pc - the multigrid context 1611f3b08a26SMatthew G. Knepley - adapt - flag for adaptation of the interpolator 1612f3b08a26SMatthew G. Knepley 1613f3b08a26SMatthew G. Knepley Options Database Keys: 1614f3b08a26SMatthew G. Knepley + -pc_mg_adapt_interp - Turn on adaptation 1615f3b08a26SMatthew G. Knepley . -pc_mg_adapt_interp_n <int> - The number of modes to use, should be divisible by dimension 1616f3b08a26SMatthew G. Knepley - -pc_mg_adapt_interp_coarse_space <type> - The type of coarse space: polynomial, harmonic, eigenvector, generalized_eigenvector 1617f3b08a26SMatthew G. Knepley 1618f3b08a26SMatthew G. Knepley Level: intermediate 1619f3b08a26SMatthew G. Knepley 1620f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()` 1621f3b08a26SMatthew G. Knepley @*/ 1622d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptInterpolation(PC pc, PetscBool adapt) 1623d71ae5a4SJacob Faibussowitsch { 1624f3b08a26SMatthew G. Knepley PetscFunctionBegin; 1625f3b08a26SMatthew G. Knepley PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1626cac4c232SBarry Smith PetscTryMethod(pc, "PCMGSetAdaptInterpolation_C", (PC, PetscBool), (pc, adapt)); 16273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1628f3b08a26SMatthew G. Knepley } 1629f3b08a26SMatthew G. Knepley 1630f3b08a26SMatthew G. Knepley /*@ 1631f1580f4eSBarry Smith PCMGGetAdaptInterpolation - Get the flag to adapt the interpolator based upon a vector space which should be accurately captured by the next coarser mesh, 1632f1580f4eSBarry Smith and thus accurately interpolated. 1633f3b08a26SMatthew G. Knepley 163420f4b53cSBarry Smith Not Collective 1635f3b08a26SMatthew G. Knepley 1636f3b08a26SMatthew G. Knepley Input Parameter: 1637f3b08a26SMatthew G. Knepley . pc - the multigrid context 1638f3b08a26SMatthew G. Knepley 1639f3b08a26SMatthew G. Knepley Output Parameter: 1640f3b08a26SMatthew G. Knepley . adapt - flag for adaptation of the interpolator 1641f3b08a26SMatthew G. Knepley 1642f3b08a26SMatthew G. Knepley Level: intermediate 1643f3b08a26SMatthew G. Knepley 1644f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()` 1645f3b08a26SMatthew G. Knepley @*/ 1646d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptInterpolation(PC pc, PetscBool *adapt) 1647d71ae5a4SJacob Faibussowitsch { 1648f3b08a26SMatthew G. Knepley PetscFunctionBegin; 1649f3b08a26SMatthew G. Knepley PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 16504f572ea9SToby Isaac PetscAssertPointer(adapt, 2); 1651cac4c232SBarry Smith PetscUseMethod(pc, "PCMGGetAdaptInterpolation_C", (PC, PetscBool *), (pc, adapt)); 16523ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1653f3b08a26SMatthew G. Knepley } 1654f3b08a26SMatthew G. Knepley 16554b9ad928SBarry Smith /*@ 165641b6fd38SMatthew G. Knepley PCMGSetAdaptCR - Monitor the coarse space quality using an auxiliary solve with compatible relaxation. 165741b6fd38SMatthew G. Knepley 1658c3339decSBarry Smith Logically Collective 165941b6fd38SMatthew G. Knepley 166041b6fd38SMatthew G. Knepley Input Parameters: 166141b6fd38SMatthew G. Knepley + pc - the multigrid context 166241b6fd38SMatthew G. Knepley - cr - flag for compatible relaxation 166341b6fd38SMatthew G. Knepley 1664f1580f4eSBarry Smith Options Database Key: 166541b6fd38SMatthew G. Knepley . -pc_mg_adapt_cr - Turn on compatible relaxation 166641b6fd38SMatthew G. Knepley 166741b6fd38SMatthew G. Knepley Level: intermediate 166841b6fd38SMatthew G. Knepley 1669f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetAdaptCR()`, `PCMGSetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()` 167041b6fd38SMatthew G. Knepley @*/ 1671d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptCR(PC pc, PetscBool cr) 1672d71ae5a4SJacob Faibussowitsch { 167341b6fd38SMatthew G. Knepley PetscFunctionBegin; 167441b6fd38SMatthew G. Knepley PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 1675cac4c232SBarry Smith PetscTryMethod(pc, "PCMGSetAdaptCR_C", (PC, PetscBool), (pc, cr)); 16763ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 167741b6fd38SMatthew G. Knepley } 167841b6fd38SMatthew G. Knepley 167941b6fd38SMatthew G. Knepley /*@ 168041b6fd38SMatthew G. Knepley PCMGGetAdaptCR - Get the flag to monitor coarse space quality using an auxiliary solve with compatible relaxation. 168141b6fd38SMatthew G. Knepley 168220f4b53cSBarry Smith Not Collective 168341b6fd38SMatthew G. Knepley 168441b6fd38SMatthew G. Knepley Input Parameter: 168541b6fd38SMatthew G. Knepley . pc - the multigrid context 168641b6fd38SMatthew G. Knepley 168741b6fd38SMatthew G. Knepley Output Parameter: 168841b6fd38SMatthew G. Knepley . cr - flag for compatible relaxaion 168941b6fd38SMatthew G. Knepley 169041b6fd38SMatthew G. Knepley Level: intermediate 169141b6fd38SMatthew G. Knepley 1692f1580f4eSBarry Smith .seealso: `PCMGSetAdaptCR()`, `PCMGGetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()` 169341b6fd38SMatthew G. Knepley @*/ 1694d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptCR(PC pc, PetscBool *cr) 1695d71ae5a4SJacob Faibussowitsch { 169641b6fd38SMatthew G. Knepley PetscFunctionBegin; 169741b6fd38SMatthew G. Knepley PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 16984f572ea9SToby Isaac PetscAssertPointer(cr, 2); 1699cac4c232SBarry Smith PetscUseMethod(pc, "PCMGGetAdaptCR_C", (PC, PetscBool *), (pc, cr)); 17003ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 170141b6fd38SMatthew G. Knepley } 170241b6fd38SMatthew G. Knepley 170341b6fd38SMatthew G. Knepley /*@ 170406792cafSBarry Smith PCMGSetNumberSmooth - Sets the number of pre and post-smoothing steps to use 1705f1580f4eSBarry Smith on all levels. Use `PCMGDistinctSmoothUp()` to create separate up and down smoothers if you want different numbers of 1706710315b6SLawrence Mitchell pre- and post-smoothing steps. 170706792cafSBarry Smith 1708c3339decSBarry Smith Logically Collective 170906792cafSBarry Smith 171006792cafSBarry Smith Input Parameters: 1711feefa0e1SJacob Faibussowitsch + pc - the multigrid context 171206792cafSBarry Smith - n - the number of smoothing steps 171306792cafSBarry Smith 171406792cafSBarry Smith Options Database Key: 1715a2b725a8SWilliam Gropp . -mg_levels_ksp_max_it <n> - Sets number of pre and post-smoothing steps 171606792cafSBarry Smith 171706792cafSBarry Smith Level: advanced 171806792cafSBarry Smith 1719f1580f4eSBarry Smith Note: 1720f1580f4eSBarry Smith This does not set a value on the coarsest grid, since we assume that there is no separate smooth up on the coarsest grid. 172106792cafSBarry Smith 1722f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetDistinctSmoothUp()` 172306792cafSBarry Smith @*/ 1724d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetNumberSmooth(PC pc, PetscInt n) 1725d71ae5a4SJacob Faibussowitsch { 172606792cafSBarry Smith PC_MG *mg = (PC_MG *)pc->data; 172706792cafSBarry Smith PC_MG_Levels **mglevels = mg->levels; 172806792cafSBarry Smith PetscInt i, levels; 172906792cafSBarry Smith 173006792cafSBarry Smith PetscFunctionBegin; 173106792cafSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 173206792cafSBarry Smith PetscValidLogicalCollectiveInt(pc, n, 2); 173328b400f6SJacob Faibussowitsch PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ORDER, "Must set MG levels with PCMGSetLevels() before calling"); 173406792cafSBarry Smith levels = mglevels[0]->levels; 173506792cafSBarry Smith 173606792cafSBarry Smith for (i = 1; i < levels; i++) { 17379566063dSJacob Faibussowitsch PetscCall(KSPSetTolerances(mglevels[i]->smoothu, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, n)); 17389566063dSJacob Faibussowitsch PetscCall(KSPSetTolerances(mglevels[i]->smoothd, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, n)); 173906792cafSBarry Smith mg->default_smoothu = n; 174006792cafSBarry Smith mg->default_smoothd = n; 174106792cafSBarry Smith } 17423ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 174306792cafSBarry Smith } 174406792cafSBarry Smith 1745f442ab6aSBarry Smith /*@ 1746f1580f4eSBarry Smith PCMGSetDistinctSmoothUp - sets the up (post) smoother to be a separate `KSP` from the down (pre) smoother on all levels 1747710315b6SLawrence Mitchell and adds the suffix _up to the options name 1748f442ab6aSBarry Smith 1749c3339decSBarry Smith Logically Collective 1750f442ab6aSBarry Smith 1751f1580f4eSBarry Smith Input Parameter: 1752f442ab6aSBarry Smith . pc - the preconditioner context 1753f442ab6aSBarry Smith 1754f442ab6aSBarry Smith Options Database Key: 1755147403d9SBarry Smith . -pc_mg_distinct_smoothup <bool> - use distinct smoothing objects 1756f442ab6aSBarry Smith 1757f442ab6aSBarry Smith Level: advanced 1758f442ab6aSBarry Smith 1759f1580f4eSBarry Smith Note: 1760f1580f4eSBarry Smith This does not set a value on the coarsest grid, since we assume that there is no separate smooth up on the coarsest grid. 1761f442ab6aSBarry Smith 1762f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetNumberSmooth()` 1763f442ab6aSBarry Smith @*/ 1764d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetDistinctSmoothUp(PC pc) 1765d71ae5a4SJacob Faibussowitsch { 1766f442ab6aSBarry Smith PC_MG *mg = (PC_MG *)pc->data; 1767f442ab6aSBarry Smith PC_MG_Levels **mglevels = mg->levels; 1768f442ab6aSBarry Smith PetscInt i, levels; 1769f442ab6aSBarry Smith KSP subksp; 1770f442ab6aSBarry Smith 1771f442ab6aSBarry Smith PetscFunctionBegin; 1772f442ab6aSBarry Smith PetscValidHeaderSpecific(pc, PC_CLASSID, 1); 177328b400f6SJacob Faibussowitsch PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ORDER, "Must set MG levels with PCMGSetLevels() before calling"); 1774f442ab6aSBarry Smith levels = mglevels[0]->levels; 1775f442ab6aSBarry Smith 1776f442ab6aSBarry Smith for (i = 1; i < levels; i++) { 1777710315b6SLawrence Mitchell const char *prefix = NULL; 1778f442ab6aSBarry Smith /* make sure smoother up and down are different */ 17799566063dSJacob Faibussowitsch PetscCall(PCMGGetSmootherUp(pc, i, &subksp)); 17809566063dSJacob Faibussowitsch PetscCall(KSPGetOptionsPrefix(mglevels[i]->smoothd, &prefix)); 17819566063dSJacob Faibussowitsch PetscCall(KSPSetOptionsPrefix(subksp, prefix)); 17829566063dSJacob Faibussowitsch PetscCall(KSPAppendOptionsPrefix(subksp, "up_")); 1783f442ab6aSBarry Smith } 17843ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1785f442ab6aSBarry Smith } 1786f442ab6aSBarry Smith 178707a4832bSFande Kong /* No new matrices are created, and the coarse operator matrices are the references to the original ones */ 1788d71ae5a4SJacob Faibussowitsch PetscErrorCode PCGetInterpolations_MG(PC pc, PetscInt *num_levels, Mat *interpolations[]) 1789d71ae5a4SJacob Faibussowitsch { 179007a4832bSFande Kong PC_MG *mg = (PC_MG *)pc->data; 179107a4832bSFande Kong PC_MG_Levels **mglevels = mg->levels; 179207a4832bSFande Kong Mat *mat; 179307a4832bSFande Kong PetscInt l; 179407a4832bSFande Kong 179507a4832bSFande Kong PetscFunctionBegin; 179628b400f6SJacob Faibussowitsch PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "Must set MG levels before calling"); 17979566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(mg->nlevels, &mat)); 179807a4832bSFande Kong for (l = 1; l < mg->nlevels; l++) { 179907a4832bSFande Kong mat[l - 1] = mglevels[l]->interpolate; 18009566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)mat[l - 1])); 180107a4832bSFande Kong } 180207a4832bSFande Kong *num_levels = mg->nlevels; 180307a4832bSFande Kong *interpolations = mat; 18043ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 180507a4832bSFande Kong } 180607a4832bSFande Kong 180707a4832bSFande Kong /* No new matrices are created, and the coarse operator matrices are the references to the original ones */ 1808d71ae5a4SJacob Faibussowitsch PetscErrorCode PCGetCoarseOperators_MG(PC pc, PetscInt *num_levels, Mat *coarseOperators[]) 1809d71ae5a4SJacob Faibussowitsch { 181007a4832bSFande Kong PC_MG *mg = (PC_MG *)pc->data; 181107a4832bSFande Kong PC_MG_Levels **mglevels = mg->levels; 181207a4832bSFande Kong PetscInt l; 181307a4832bSFande Kong Mat *mat; 181407a4832bSFande Kong 181507a4832bSFande Kong PetscFunctionBegin; 181628b400f6SJacob Faibussowitsch PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "Must set MG levels before calling"); 18179566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(mg->nlevels, &mat)); 181807a4832bSFande Kong for (l = 0; l < mg->nlevels - 1; l++) { 18199566063dSJacob Faibussowitsch PetscCall(KSPGetOperators(mglevels[l]->smoothd, NULL, &(mat[l]))); 18209566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)mat[l])); 182107a4832bSFande Kong } 182207a4832bSFande Kong *num_levels = mg->nlevels; 182307a4832bSFande Kong *coarseOperators = mat; 18243ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 182507a4832bSFande Kong } 182607a4832bSFande Kong 1827f3b08a26SMatthew G. Knepley /*@C 1828f1580f4eSBarry Smith PCMGRegisterCoarseSpaceConstructor - Adds a method to the `PCMG` package for coarse space construction. 1829f3b08a26SMatthew G. Knepley 183020f4b53cSBarry Smith Not Collective 1831f3b08a26SMatthew G. Knepley 1832f3b08a26SMatthew G. Knepley Input Parameters: 1833f3b08a26SMatthew G. Knepley + name - name of the constructor 1834f3b08a26SMatthew G. Knepley - function - constructor routine 1835f3b08a26SMatthew G. Knepley 183620f4b53cSBarry Smith Calling sequence of `function`: 183720f4b53cSBarry Smith $ PetscErrorCode my_csp(PC pc, PetscInt l, DM dm, KSP smooth, PetscInt Nc, Mat initGuess, Mat *coarseSp) 183820f4b53cSBarry Smith + pc - The `PC` object 1839f1580f4eSBarry Smith . l - The multigrid level, 0 is the coarse level 184020f4b53cSBarry Smith . dm - The `DM` for this level 1841f1580f4eSBarry Smith . smooth - The level smoother 1842f1580f4eSBarry Smith . Nc - The size of the coarse space 1843f1580f4eSBarry Smith . initGuess - Basis for an initial guess for the space 1844f1580f4eSBarry Smith - coarseSp - A basis for the computed coarse space 1845f3b08a26SMatthew G. Knepley 1846f3b08a26SMatthew G. Knepley Level: advanced 1847f3b08a26SMatthew G. Knepley 1848feefa0e1SJacob Faibussowitsch Developer Notes: 1849f1580f4eSBarry Smith How come this is not used by `PCGAMG`? 1850f1580f4eSBarry Smith 1851f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetCoarseSpaceConstructor()`, `PCRegister()` 1852f3b08a26SMatthew G. Knepley @*/ 1853d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGRegisterCoarseSpaceConstructor(const char name[], PetscErrorCode (*function)(PC, PetscInt, DM, KSP, PetscInt, Mat, Mat *)) 1854d71ae5a4SJacob Faibussowitsch { 1855f3b08a26SMatthew G. Knepley PetscFunctionBegin; 18569566063dSJacob Faibussowitsch PetscCall(PCInitializePackage()); 18579566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&PCMGCoarseList, name, function)); 18583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1859f3b08a26SMatthew G. Knepley } 1860f3b08a26SMatthew G. Knepley 1861f3b08a26SMatthew G. Knepley /*@C 1862f3b08a26SMatthew G. Knepley PCMGGetCoarseSpaceConstructor - Returns the given coarse space construction method. 1863f3b08a26SMatthew G. Knepley 186420f4b53cSBarry Smith Not Collective 1865f3b08a26SMatthew G. Knepley 1866f3b08a26SMatthew G. Knepley Input Parameter: 1867f3b08a26SMatthew G. Knepley . name - name of the constructor 1868f3b08a26SMatthew G. Knepley 1869f3b08a26SMatthew G. Knepley Output Parameter: 1870f3b08a26SMatthew G. Knepley . function - constructor routine 1871f3b08a26SMatthew G. Knepley 1872f3b08a26SMatthew G. Knepley Level: advanced 1873f3b08a26SMatthew G. Knepley 1874f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGRegisterCoarseSpaceConstructor()`, `PCRegister()` 1875f3b08a26SMatthew G. Knepley @*/ 1876d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetCoarseSpaceConstructor(const char name[], PetscErrorCode (**function)(PC, PetscInt, DM, KSP, PetscInt, Mat, Mat *)) 1877d71ae5a4SJacob Faibussowitsch { 1878f3b08a26SMatthew G. Knepley PetscFunctionBegin; 18799566063dSJacob Faibussowitsch PetscCall(PetscFunctionListFind(PCMGCoarseList, name, function)); 18803ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1881f3b08a26SMatthew G. Knepley } 1882f3b08a26SMatthew G. Knepley 18833b09bd56SBarry Smith /*MC 1884ccb205f8SBarry Smith PCMG - Use multigrid preconditioning. This preconditioner requires you provide additional 18853b09bd56SBarry Smith information about the coarser grid matrices and restriction/interpolation operators. 18863b09bd56SBarry Smith 18873b09bd56SBarry Smith Options Database Keys: 18883b09bd56SBarry Smith + -pc_mg_levels <nlevels> - number of levels including finest 1889391689abSStefano Zampini . -pc_mg_cycle_type <v,w> - provide the cycle desired 18908c1c2452SJed Brown . -pc_mg_type <additive,multiplicative,full,kaskade> - multiplicative is the default 18913b09bd56SBarry Smith . -pc_mg_log - log information about time spent on each level of the solver 1892710315b6SLawrence Mitchell . -pc_mg_distinct_smoothup - configure up (after interpolation) and down (before restriction) smoothers separately (with different options prefixes) 18932134b1e4SBarry Smith . -pc_mg_galerkin <both,pmat,mat,none> - use Galerkin process to compute coarser operators, i.e. Acoarse = R A R' 18948cf6037eSBarry Smith . -pc_mg_multiplicative_cycles - number of cycles to use as the preconditioner (defaults to 1) 18958cf6037eSBarry Smith . -pc_mg_dump_matlab - dumps the matrices for each level and the restriction/interpolation matrices 1896e3c5b3baSBarry Smith to the Socket viewer for reading from MATLAB. 18978cf6037eSBarry Smith - -pc_mg_dump_binary - dumps the matrices for each level and the restriction/interpolation matrices 18988cf6037eSBarry Smith to the binary output file called binaryoutput 18993b09bd56SBarry Smith 190020f4b53cSBarry Smith Level: intermediate 190120f4b53cSBarry Smith 190295452b02SPatrick Sanan Notes: 1903f1580f4eSBarry Smith If one uses a Krylov method such `KSPGMRES` or `KSPCG` as the smoother then one must use `KSPFGMRES`, `KSPGCR`, or `KSPRICHARDSON` as the outer Krylov method 19043b09bd56SBarry Smith 19058cf6037eSBarry Smith When run with a single level the smoother options are used on that level NOT the coarse grid solver options 19068cf6037eSBarry Smith 1907f1580f4eSBarry Smith When run with `KSPRICHARDSON` the convergence test changes slightly if monitor is turned on. The iteration count may change slightly. This 190823067569SBarry Smith is because without monitoring the residual norm is computed WITHIN each multigrid cycle on the finest level after the pre-smoothing 190923067569SBarry Smith (because the residual has just been computed for the multigrid algorithm and is hence available for free) while with monitoring the 191023067569SBarry Smith residual is computed at the end of each cycle. 191123067569SBarry Smith 1912db781477SPatrick Sanan .seealso: `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCMGType`, `PCEXOTIC`, `PCGAMG`, `PCML`, `PCHYPRE` 1913db781477SPatrick Sanan `PCMGSetLevels()`, `PCMGGetLevels()`, `PCMGSetType()`, `PCMGSetCycleType()`, 1914db781477SPatrick Sanan `PCMGSetDistinctSmoothUp()`, `PCMGGetCoarseSolve()`, `PCMGSetResidual()`, `PCMGSetInterpolation()`, 1915db781477SPatrick Sanan `PCMGSetRestriction()`, `PCMGGetSmoother()`, `PCMGGetSmootherUp()`, `PCMGGetSmootherDown()`, 1916f1580f4eSBarry Smith `PCMGSetCycleTypeOnLevel()`, `PCMGSetRhs()`, `PCMGSetX()`, `PCMGSetR()`, 1917f1580f4eSBarry Smith `PCMGSetAdaptCR()`, `PCMGGetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()` 19183b09bd56SBarry Smith M*/ 19193b09bd56SBarry Smith 1920d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode PCCreate_MG(PC pc) 1921d71ae5a4SJacob Faibussowitsch { 1922f3fbd535SBarry Smith PC_MG *mg; 1923f3fbd535SBarry Smith 19244b9ad928SBarry Smith PetscFunctionBegin; 19254dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&mg)); 19263ec1f749SStefano Zampini pc->data = mg; 1927f3fbd535SBarry Smith mg->nlevels = -1; 192810eca3edSLisandro Dalcin mg->am = PC_MG_MULTIPLICATIVE; 19292134b1e4SBarry Smith mg->galerkin = PC_MG_GALERKIN_NONE; 1930f3b08a26SMatthew G. Knepley mg->adaptInterpolation = PETSC_FALSE; 1931f3b08a26SMatthew G. Knepley mg->Nc = -1; 1932f3b08a26SMatthew G. Knepley mg->eigenvalue = -1; 1933f3fbd535SBarry Smith 193437a44384SMark Adams pc->useAmat = PETSC_TRUE; 193537a44384SMark Adams 19364b9ad928SBarry Smith pc->ops->apply = PCApply_MG; 1937fcb023d4SJed Brown pc->ops->applytranspose = PCApplyTranspose_MG; 193830b0564aSStefano Zampini pc->ops->matapply = PCMatApply_MG; 19394b9ad928SBarry Smith pc->ops->setup = PCSetUp_MG; 1940a06653b4SBarry Smith pc->ops->reset = PCReset_MG; 19414b9ad928SBarry Smith pc->ops->destroy = PCDestroy_MG; 19424b9ad928SBarry Smith pc->ops->setfromoptions = PCSetFromOptions_MG; 19434b9ad928SBarry Smith pc->ops->view = PCView_MG; 1944fb15c04eSBarry Smith 19459566063dSJacob Faibussowitsch PetscCall(PetscObjectComposedDataRegister(&mg->eigenvalue)); 19469566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetGalerkin_C", PCMGSetGalerkin_MG)); 19479566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetLevels_C", PCMGGetLevels_MG)); 19489566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetLevels_C", PCMGSetLevels_MG)); 19499566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetInterpolations_C", PCGetInterpolations_MG)); 19509566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetCoarseOperators_C", PCGetCoarseOperators_MG)); 19519566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptInterpolation_C", PCMGSetAdaptInterpolation_MG)); 19529566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptInterpolation_C", PCMGGetAdaptInterpolation_MG)); 19539566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptCR_C", PCMGSetAdaptCR_MG)); 19549566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptCR_C", PCMGGetAdaptCR_MG)); 19552b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptCoarseSpaceType_C", PCMGSetAdaptCoarseSpaceType_MG)); 19562b3cbbdaSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptCoarseSpaceType_C", PCMGGetAdaptCoarseSpaceType_MG)); 19573ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 19584b9ad928SBarry Smith } 1959