1af0996ceSBarry Smith #include <petsc/private/pcimpl.h> /*I "petscpc.h" I*/ 2a80b646eSBarry Smith #include <petsc/private/kspimpl.h> /* This is needed to provide the appropriate PETSC_EXTERN for KSP_Solve_FS ....*/ 31e25c274SJed Brown #include <petscdm.h> 40971522cSBarry Smith 50a545947SLisandro Dalcin const char *const PCFieldSplitSchurPreTypes[] = {"SELF","SELFP","A11","USER","FULL","PCFieldSplitSchurPreType","PC_FIELDSPLIT_SCHUR_PRE_",NULL}; 60a545947SLisandro Dalcin const char *const PCFieldSplitSchurFactTypes[] = {"DIAG","LOWER","UPPER","FULL","PCFieldSplitSchurFactType","PC_FIELDSPLIT_SCHUR_FACT_",NULL}; 7c5d2311dSJed Brown 89df09d43SBarry Smith PetscLogEvent KSP_Solve_FS_0,KSP_Solve_FS_1,KSP_Solve_FS_S,KSP_Solve_FS_U,KSP_Solve_FS_L,KSP_Solve_FS_2,KSP_Solve_FS_3,KSP_Solve_FS_4; 99df09d43SBarry Smith 100971522cSBarry Smith typedef struct _PC_FieldSplitLink *PC_FieldSplitLink; 110971522cSBarry Smith struct _PC_FieldSplitLink { 1269a612a9SBarry Smith KSP ksp; 13443836d0SMatthew G Knepley Vec x,y,z; 14db4c96c1SJed Brown char *splitname; 150971522cSBarry Smith PetscInt nfields; 165d4c12cdSJungho Lee PetscInt *fields,*fields_col; 171b9fc7fcSBarry Smith VecScatter sctx; 18f5f0d762SBarry Smith IS is,is_col; 1951f519a2SBarry Smith PC_FieldSplitLink next,previous; 209df09d43SBarry Smith PetscLogEvent event; 215ddf11f8SNicolas Barnafi 225ddf11f8SNicolas Barnafi /* Used only when setting coordinates with PCSetCoordinates */ 235ddf11f8SNicolas Barnafi PetscInt dim; 245ddf11f8SNicolas Barnafi PetscInt ndofs; 255ddf11f8SNicolas Barnafi PetscReal *coords; 260971522cSBarry Smith }; 270971522cSBarry Smith 280971522cSBarry Smith typedef struct { 2968dd23aaSBarry Smith PCCompositeType type; 30ace3abfcSBarry Smith PetscBool defaultsplit; /* Flag for a system with a set of 'k' scalar fields with the same layout (and bs = k) */ 31ace3abfcSBarry Smith PetscBool splitdefined; /* Flag is set after the splits have been defined, to prevent more splits from being added */ 3230ad9308SMatthew Knepley PetscInt bs; /* Block size for IS and Mat structures */ 3330ad9308SMatthew Knepley PetscInt nsplits; /* Number of field divisions defined */ 3479416396SBarry Smith Vec *x,*y,w1,w2; 35519d70e2SJed Brown Mat *mat; /* The diagonal block for each split */ 36519d70e2SJed Brown Mat *pmat; /* The preconditioning diagonal block for each split */ 3730ad9308SMatthew Knepley Mat *Afield; /* The rows of the matrix associated with each split */ 38ace3abfcSBarry Smith PetscBool issetup; 392fa5cd67SKarl Rupp 4030ad9308SMatthew Knepley /* Only used when Schur complement preconditioning is used */ 4130ad9308SMatthew Knepley Mat B; /* The (0,1) block */ 4230ad9308SMatthew Knepley Mat C; /* The (1,0) block */ 43443836d0SMatthew G Knepley Mat schur; /* The Schur complement S = A11 - A10 A00^{-1} A01, the KSP here, kspinner, is H_1 in [El08] */ 44a7476a74SDmitry Karpeev Mat schurp; /* Assembled approximation to S built by MatSchurComplement to be used as a preconditioning matrix when solving with S */ 45084e4875SJed Brown Mat schur_user; /* User-provided preconditioning matrix for the Schur complement */ 46084e4875SJed Brown PCFieldSplitSchurPreType schurpre; /* Determines which preconditioning matrix is used for the Schur complement */ 47c9c6ffaaSJed Brown PCFieldSplitSchurFactType schurfactorization; 4830ad9308SMatthew Knepley KSP kspschur; /* The solver for S */ 49443836d0SMatthew G Knepley KSP kspupper; /* The solver for A in the upper diagonal part of the factorization (H_2 in [El08]) */ 50c096484dSStefano Zampini PetscScalar schurscale; /* Scaling factor for the Schur complement solution with DIAG factorization */ 51c096484dSStefano Zampini 52a51937d4SCarola Kruse /* Only used when Golub-Kahan bidiagonalization preconditioning is used */ 53a51937d4SCarola Kruse Mat H; /* The modified matrix H = A00 + nu*A01*A01' */ 54a51937d4SCarola Kruse PetscReal gkbtol; /* Stopping tolerance for lower bound estimate */ 55a51937d4SCarola Kruse PetscInt gkbdelay; /* The delay window for the stopping criterion */ 56a51937d4SCarola Kruse PetscReal gkbnu; /* Parameter for augmented Lagrangian H = A + nu*A01*A01' */ 57a51937d4SCarola Kruse PetscInt gkbmaxit; /* Maximum number of iterations for outer loop */ 58de482cd7SCarola Kruse PetscBool gkbmonitor; /* Monitor for gkb iterations and the lower bound error */ 59de482cd7SCarola Kruse PetscViewer gkbviewer; /* Viewer context for gkbmonitor */ 60e071a0a4SCarola Kruse Vec u,v,d,Hu; /* Work vectors for the GKB algorithm */ 61e071a0a4SCarola Kruse PetscScalar *vecz; /* Contains intermediate values, eg for lower bound */ 62a51937d4SCarola Kruse 6397bbdb24SBarry Smith PC_FieldSplitLink head; 646dbb499eSCian Wilson PetscBool isrestrict; /* indicates PCFieldSplitRestrictIS() has been last called on this object, hack */ 65c1570756SJed Brown PetscBool suboptionsset; /* Indicates that the KSPSetFromOptions() has been called on the sub-KSPs */ 664ab8060aSDmitry Karpeev PetscBool dm_splits; /* Whether to use DMCreateFieldDecomposition() whenever possible */ 67c84da90fSDmitry Karpeev PetscBool diag_use_amat; /* Whether to extract diagonal matrix blocks from Amat, rather than Pmat (weaker than -pc_use_amat) */ 68c84da90fSDmitry Karpeev PetscBool offdiag_use_amat; /* Whether to extract off-diagonal matrix blocks from Amat, rather than Pmat (weaker than -pc_use_amat) */ 697b752e3dSPatrick Sanan PetscBool detect; /* Whether to form 2-way split by finding zero diagonal entries */ 705ddf11f8SNicolas Barnafi PetscBool coordinates_set; /* Whether PCSetCoordinates has been called */ 710971522cSBarry Smith } PC_FieldSplit; 720971522cSBarry Smith 7316913363SBarry Smith /* 7495452b02SPatrick Sanan Notes: 7595452b02SPatrick Sanan there is no particular reason that pmat, x, and y are stored as arrays in PC_FieldSplit instead of 7616913363SBarry Smith inside PC_FieldSplitLink, just historical. If you want to be able to add new fields after already using the 7716913363SBarry Smith PC you could change this. 7816913363SBarry Smith */ 79084e4875SJed Brown 80e6cab6aaSJed Brown /* This helper is so that setting a user-provided preconditioning matrix is orthogonal to choosing to use it. This way the 81084e4875SJed Brown * application-provided FormJacobian can provide this matrix without interfering with the user's (command-line) choices. */ 82084e4875SJed Brown static Mat FieldSplitSchurPre(PC_FieldSplit *jac) 83084e4875SJed Brown { 84084e4875SJed Brown switch (jac->schurpre) { 85084e4875SJed Brown case PC_FIELDSPLIT_SCHUR_PRE_SELF: return jac->schur; 86a7476a74SDmitry Karpeev case PC_FIELDSPLIT_SCHUR_PRE_SELFP: return jac->schurp; 87e87fab1bSBarry Smith case PC_FIELDSPLIT_SCHUR_PRE_A11: return jac->pmat[1]; 88e74569cdSMatthew G. Knepley case PC_FIELDSPLIT_SCHUR_PRE_FULL: /* We calculate this and store it in schur_user */ 89084e4875SJed Brown case PC_FIELDSPLIT_SCHUR_PRE_USER: /* Use a user-provided matrix if it is given, otherwise diagonal block */ 90084e4875SJed Brown default: 91084e4875SJed Brown return jac->schur_user ? jac->schur_user : jac->pmat[1]; 92084e4875SJed Brown } 93084e4875SJed Brown } 94084e4875SJed Brown 959804daf3SBarry Smith #include <petscdraw.h> 960971522cSBarry Smith static PetscErrorCode PCView_FieldSplit(PC pc,PetscViewer viewer) 970971522cSBarry Smith { 980971522cSBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 99d9884438SBarry Smith PetscBool iascii,isdraw; 1000971522cSBarry Smith PetscInt i,j; 1015a9f2f41SSatish Balay PC_FieldSplitLink ilink = jac->head; 1020971522cSBarry Smith 1030971522cSBarry Smith PetscFunctionBegin; 1045f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii)); 1055f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw)); 1060971522cSBarry Smith if (iascii) { 1072c9966d7SBarry Smith if (jac->bs > 0) { 1085f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," FieldSplit with %s composition: total splits = %D, blocksize = %D\n",PCCompositeTypes[jac->type],jac->nsplits,jac->bs)); 1092c9966d7SBarry Smith } else { 1105f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," FieldSplit with %s composition: total splits = %D\n",PCCompositeTypes[jac->type],jac->nsplits)); 1112c9966d7SBarry Smith } 112f5236f50SJed Brown if (pc->useAmat) { 1135f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," using Amat (not Pmat) as operator for blocks\n")); 114a3df900dSMatthew G Knepley } 115c84da90fSDmitry Karpeev if (jac->diag_use_amat) { 1165f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," using Amat (not Pmat) as operator for diagonal blocks\n")); 117c84da90fSDmitry Karpeev } 118c84da90fSDmitry Karpeev if (jac->offdiag_use_amat) { 1195f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," using Amat (not Pmat) as operator for off-diagonal blocks\n")); 120c84da90fSDmitry Karpeev } 1215f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Solver info for each split is in the following KSP objects:\n")); 1220971522cSBarry Smith for (i=0; i<jac->nsplits; i++) { 1231ab39975SBarry Smith if (ilink->fields) { 1245f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"Split number %D Fields ",i)); 1255f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE)); 1265a9f2f41SSatish Balay for (j=0; j<ilink->nfields; j++) { 12779416396SBarry Smith if (j > 0) { 1285f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,",")); 12979416396SBarry Smith } 1305f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," %D",ilink->fields[j])); 1310971522cSBarry Smith } 1325f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"\n")); 1335f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE)); 1341ab39975SBarry Smith } else { 1355f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"Split number %D Defined by IS\n",i)); 1361ab39975SBarry Smith } 1375f80ce2aSJacob Faibussowitsch CHKERRQ(KSPView(ilink->ksp,viewer)); 1385a9f2f41SSatish Balay ilink = ilink->next; 1390971522cSBarry Smith } 1402fa5cd67SKarl Rupp } 1412fa5cd67SKarl Rupp 1422fa5cd67SKarl Rupp if (isdraw) { 143d9884438SBarry Smith PetscDraw draw; 144d9884438SBarry Smith PetscReal x,y,w,wd; 145d9884438SBarry Smith 1465f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerDrawGetDraw(viewer,0,&draw)); 1475f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawGetCurrentPoint(draw,&x,&y)); 148d9884438SBarry Smith w = 2*PetscMin(1.0 - x,x); 149d9884438SBarry Smith wd = w/(jac->nsplits + 1); 150d9884438SBarry Smith x = x - wd*(jac->nsplits-1)/2.0; 151d9884438SBarry Smith for (i=0; i<jac->nsplits; i++) { 1525f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPushCurrentPoint(draw,x,y)); 1535f80ce2aSJacob Faibussowitsch CHKERRQ(KSPView(ilink->ksp,viewer)); 1545f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPopCurrentPoint(draw)); 155d9884438SBarry Smith x += wd; 156d9884438SBarry Smith ilink = ilink->next; 157d9884438SBarry Smith } 1580971522cSBarry Smith } 1590971522cSBarry Smith PetscFunctionReturn(0); 1600971522cSBarry Smith } 1610971522cSBarry Smith 1623b224e63SBarry Smith static PetscErrorCode PCView_FieldSplit_Schur(PC pc,PetscViewer viewer) 1633b224e63SBarry Smith { 1643b224e63SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 1654996c5bdSBarry Smith PetscBool iascii,isdraw; 1663b224e63SBarry Smith PetscInt i,j; 1673b224e63SBarry Smith PC_FieldSplitLink ilink = jac->head; 168a9908d51SBarry Smith MatSchurComplementAinvType atype; 1693b224e63SBarry Smith 1703b224e63SBarry Smith PetscFunctionBegin; 1715f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii)); 1725f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw)); 1733b224e63SBarry Smith if (iascii) { 1742c9966d7SBarry Smith if (jac->bs > 0) { 1755f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," FieldSplit with Schur preconditioner, blocksize = %D, factorization %s\n",jac->bs,PCFieldSplitSchurFactTypes[jac->schurfactorization])); 1762c9966d7SBarry Smith } else { 1775f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," FieldSplit with Schur preconditioner, factorization %s\n",PCFieldSplitSchurFactTypes[jac->schurfactorization])); 1782c9966d7SBarry Smith } 179f5236f50SJed Brown if (pc->useAmat) { 1805f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," using Amat (not Pmat) as operator for blocks\n")); 181a3df900dSMatthew G Knepley } 1823e8b8b31SMatthew G Knepley switch (jac->schurpre) { 1833e8b8b31SMatthew G Knepley case PC_FIELDSPLIT_SCHUR_PRE_SELF: 1845f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Preconditioner for the Schur complement formed from S itself\n")); 185a9908d51SBarry Smith break; 186a7476a74SDmitry Karpeev case PC_FIELDSPLIT_SCHUR_PRE_SELFP: 1875f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementGetAinvType(jac->schur,&atype)); 1885f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Preconditioner for the Schur complement formed from Sp, an assembled approximation to S, which uses A00's %sdiagonal's inverse\n",atype == MAT_SCHUR_COMPLEMENT_AINV_DIAG ? "" : (atype == MAT_SCHUR_COMPLEMENT_AINV_BLOCK_DIAG ? "block " : "lumped ")));break; 189e87fab1bSBarry Smith case PC_FIELDSPLIT_SCHUR_PRE_A11: 1905f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Preconditioner for the Schur complement formed from A11\n")); 191a9908d51SBarry Smith break; 192e74569cdSMatthew G. Knepley case PC_FIELDSPLIT_SCHUR_PRE_FULL: 1935f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Preconditioner for the Schur complement formed from the exact Schur complement\n")); 194a9908d51SBarry Smith break; 1953e8b8b31SMatthew G Knepley case PC_FIELDSPLIT_SCHUR_PRE_USER: 1963e8b8b31SMatthew G Knepley if (jac->schur_user) { 1975f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Preconditioner for the Schur complement formed from user provided matrix\n")); 1983e8b8b31SMatthew G Knepley } else { 1995f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Preconditioner for the Schur complement formed from A11\n")); 2003e8b8b31SMatthew G Knepley } 2013e8b8b31SMatthew G Knepley break; 2023e8b8b31SMatthew G Knepley default: 20398921bdaSJacob Faibussowitsch SETERRQ(PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_OUTOFRANGE, "Invalid Schur preconditioning type: %d", jac->schurpre); 2043e8b8b31SMatthew G Knepley } 2055f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Split info:\n")); 2065f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPushTab(viewer)); 2073b224e63SBarry Smith for (i=0; i<jac->nsplits; i++) { 2083b224e63SBarry Smith if (ilink->fields) { 2095f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"Split number %D Fields ",i)); 2105f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE)); 2113b224e63SBarry Smith for (j=0; j<ilink->nfields; j++) { 2123b224e63SBarry Smith if (j > 0) { 2135f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,",")); 2143b224e63SBarry Smith } 2155f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," %D",ilink->fields[j])); 2163b224e63SBarry Smith } 2175f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"\n")); 2185f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE)); 2193b224e63SBarry Smith } else { 2205f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"Split number %D Defined by IS\n",i)); 2213b224e63SBarry Smith } 2223b224e63SBarry Smith ilink = ilink->next; 2233b224e63SBarry Smith } 2245f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"KSP solver for A00 block\n")); 2255f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPushTab(viewer)); 22606de4afeSJed Brown if (jac->head) { 2275f80ce2aSJacob Faibussowitsch CHKERRQ(KSPView(jac->head->ksp,viewer)); 2285f80ce2aSJacob Faibussowitsch } else CHKERRQ(PetscViewerASCIIPrintf(viewer," not yet available\n")); 2295f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPopTab(viewer)); 23006de4afeSJed Brown if (jac->head && jac->kspupper != jac->head->ksp) { 2315f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"KSP solver for upper A00 in upper triangular factor \n")); 2325f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPushTab(viewer)); 2335f80ce2aSJacob Faibussowitsch if (jac->kspupper) CHKERRQ(KSPView(jac->kspupper,viewer)); 2345f80ce2aSJacob Faibussowitsch else CHKERRQ(PetscViewerASCIIPrintf(viewer," not yet available\n")); 2355f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPopTab(viewer)); 236443836d0SMatthew G Knepley } 2375f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"KSP solver for S = A11 - A10 inv(A00) A01 \n")); 2385f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPushTab(viewer)); 23912cae6f2SJed Brown if (jac->kspschur) { 2405f80ce2aSJacob Faibussowitsch CHKERRQ(KSPView(jac->kspschur,viewer)); 24112cae6f2SJed Brown } else { 2425f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," not yet available\n")); 24312cae6f2SJed Brown } 2445f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPopTab(viewer)); 2455f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPopTab(viewer)); 24606de4afeSJed Brown } else if (isdraw && jac->head) { 2474996c5bdSBarry Smith PetscDraw draw; 2484996c5bdSBarry Smith PetscReal x,y,w,wd,h; 2494996c5bdSBarry Smith PetscInt cnt = 2; 2504996c5bdSBarry Smith char str[32]; 2514996c5bdSBarry Smith 2525f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerDrawGetDraw(viewer,0,&draw)); 2535f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawGetCurrentPoint(draw,&x,&y)); 254c74581afSBarry Smith if (jac->kspupper != jac->head->ksp) cnt++; 255c74581afSBarry Smith w = 2*PetscMin(1.0 - x,x); 256c74581afSBarry Smith wd = w/(cnt + 1); 257c74581afSBarry Smith 2585f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(str,32,"Schur fact. %s",PCFieldSplitSchurFactTypes[jac->schurfactorization])); 2595f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawStringBoxed(draw,x,y,PETSC_DRAW_RED,PETSC_DRAW_BLACK,str,NULL,&h)); 2604996c5bdSBarry Smith y -= h; 2614996c5bdSBarry Smith if (jac->schurpre == PC_FIELDSPLIT_SCHUR_PRE_USER && !jac->schur_user) { 2625f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(str,32,"Prec. for Schur from %s",PCFieldSplitSchurPreTypes[PC_FIELDSPLIT_SCHUR_PRE_A11])); 2633b224e63SBarry Smith } else { 2645f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(str,32,"Prec. for Schur from %s",PCFieldSplitSchurPreTypes[jac->schurpre])); 2654996c5bdSBarry Smith } 2665f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawStringBoxed(draw,x+wd*(cnt-1)/2.0,y,PETSC_DRAW_RED,PETSC_DRAW_BLACK,str,NULL,&h)); 2674996c5bdSBarry Smith y -= h; 2684996c5bdSBarry Smith x = x - wd*(cnt-1)/2.0; 2694996c5bdSBarry Smith 2705f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPushCurrentPoint(draw,x,y)); 2715f80ce2aSJacob Faibussowitsch CHKERRQ(KSPView(jac->head->ksp,viewer)); 2725f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPopCurrentPoint(draw)); 2734996c5bdSBarry Smith if (jac->kspupper != jac->head->ksp) { 2744996c5bdSBarry Smith x += wd; 2755f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPushCurrentPoint(draw,x,y)); 2765f80ce2aSJacob Faibussowitsch CHKERRQ(KSPView(jac->kspupper,viewer)); 2775f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPopCurrentPoint(draw)); 2784996c5bdSBarry Smith } 2794996c5bdSBarry Smith x += wd; 2805f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPushCurrentPoint(draw,x,y)); 2815f80ce2aSJacob Faibussowitsch CHKERRQ(KSPView(jac->kspschur,viewer)); 2825f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPopCurrentPoint(draw)); 2833b224e63SBarry Smith } 2843b224e63SBarry Smith PetscFunctionReturn(0); 2853b224e63SBarry Smith } 2863b224e63SBarry Smith 287a51937d4SCarola Kruse static PetscErrorCode PCView_FieldSplit_GKB(PC pc,PetscViewer viewer) 288a51937d4SCarola Kruse { 289a51937d4SCarola Kruse PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 290a51937d4SCarola Kruse PetscBool iascii,isdraw; 291a51937d4SCarola Kruse PetscInt i,j; 292a51937d4SCarola Kruse PC_FieldSplitLink ilink = jac->head; 293a51937d4SCarola Kruse 294a51937d4SCarola Kruse PetscFunctionBegin; 2955f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii)); 2965f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw)); 297a51937d4SCarola Kruse if (iascii) { 298a51937d4SCarola Kruse if (jac->bs > 0) { 2995f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," FieldSplit with %s composition: total splits = %D, blocksize = %D\n",PCCompositeTypes[jac->type],jac->nsplits,jac->bs)); 300a51937d4SCarola Kruse } else { 3015f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," FieldSplit with %s composition: total splits = %D\n",PCCompositeTypes[jac->type],jac->nsplits)); 302a51937d4SCarola Kruse } 303a51937d4SCarola Kruse if (pc->useAmat) { 3045f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," using Amat (not Pmat) as operator for blocks\n")); 305a51937d4SCarola Kruse } 306a51937d4SCarola Kruse if (jac->diag_use_amat) { 3075f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," using Amat (not Pmat) as operator for diagonal blocks\n")); 308a51937d4SCarola Kruse } 309a51937d4SCarola Kruse if (jac->offdiag_use_amat) { 3105f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," using Amat (not Pmat) as operator for off-diagonal blocks\n")); 311a51937d4SCarola Kruse } 312a51937d4SCarola Kruse 3135f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Stopping tolerance=%.1e, delay in error estimate=%D, maximum iterations=%D\n",jac->gkbtol,jac->gkbdelay,jac->gkbmaxit)); 3145f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," Solver info for H = A00 + nu*A01*A01' matrix:\n")); 3155f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPushTab(viewer)); 316a51937d4SCarola Kruse 317a51937d4SCarola Kruse if (ilink->fields) { 3185f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"Split number %D Fields ",0)); 3195f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE)); 320a51937d4SCarola Kruse for (j=0; j<ilink->nfields; j++) { 321a51937d4SCarola Kruse if (j > 0) { 3225f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,",")); 323a51937d4SCarola Kruse } 3245f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer," %D",ilink->fields[j])); 325a51937d4SCarola Kruse } 3265f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"\n")); 3275f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE)); 328a51937d4SCarola Kruse } else { 3295f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(viewer,"Split number %D Defined by IS\n",0)); 330a51937d4SCarola Kruse } 3315f80ce2aSJacob Faibussowitsch CHKERRQ(KSPView(ilink->ksp,viewer)); 332a51937d4SCarola Kruse 3335f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPopTab(viewer)); 334a51937d4SCarola Kruse } 335a51937d4SCarola Kruse 336a51937d4SCarola Kruse if (isdraw) { 337a51937d4SCarola Kruse PetscDraw draw; 338a51937d4SCarola Kruse PetscReal x,y,w,wd; 339a51937d4SCarola Kruse 3405f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerDrawGetDraw(viewer,0,&draw)); 3415f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawGetCurrentPoint(draw,&x,&y)); 342a51937d4SCarola Kruse w = 2*PetscMin(1.0 - x,x); 343a51937d4SCarola Kruse wd = w/(jac->nsplits + 1); 344a51937d4SCarola Kruse x = x - wd*(jac->nsplits-1)/2.0; 345a51937d4SCarola Kruse for (i=0; i<jac->nsplits; i++) { 3465f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPushCurrentPoint(draw,x,y)); 3475f80ce2aSJacob Faibussowitsch CHKERRQ(KSPView(ilink->ksp,viewer)); 3485f80ce2aSJacob Faibussowitsch CHKERRQ(PetscDrawPopCurrentPoint(draw)); 349a51937d4SCarola Kruse x += wd; 350a51937d4SCarola Kruse ilink = ilink->next; 351a51937d4SCarola Kruse } 352a51937d4SCarola Kruse } 353a51937d4SCarola Kruse PetscFunctionReturn(0); 354a51937d4SCarola Kruse } 355a51937d4SCarola Kruse 3566c924f48SJed Brown /* Precondition: jac->bs is set to a meaningful value */ 3576c924f48SJed Brown static PetscErrorCode PCFieldSplitSetRuntimeSplits_Private(PC pc) 3586c924f48SJed Brown { 3596c924f48SJed Brown PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 3605d4c12cdSJungho Lee PetscInt i,nfields,*ifields,nfields_col,*ifields_col; 3615d4c12cdSJungho Lee PetscBool flg,flg_col; 3625d4c12cdSJungho Lee char optionname[128],splitname[8],optionname_col[128]; 3636c924f48SJed Brown 3646c924f48SJed Brown PetscFunctionBegin; 3655f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(jac->bs,&ifields)); 3665f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(jac->bs,&ifields_col)); 3676c924f48SJed Brown for (i=0,flg=PETSC_TRUE;; i++) { 3685f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(splitname,sizeof(splitname),"%D",i)); 3695f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(optionname,sizeof(optionname),"-pc_fieldsplit_%D_fields",i)); 3705f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(optionname_col,sizeof(optionname_col),"-pc_fieldsplit_%D_fields_col",i)); 3716c924f48SJed Brown nfields = jac->bs; 37229499fbbSJungho Lee nfields_col = jac->bs; 3735f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsGetIntArray(((PetscObject)pc)->options,((PetscObject)pc)->prefix,optionname,ifields,&nfields,&flg)); 3745f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsGetIntArray(((PetscObject)pc)->options,((PetscObject)pc)->prefix,optionname_col,ifields_col,&nfields_col,&flg_col)); 3756c924f48SJed Brown if (!flg) break; 3765d4c12cdSJungho Lee else if (flg && !flg_col) { 377*28b400f6SJacob Faibussowitsch PetscCheck(nfields,PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot list zero fields"); 3785f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetFields(pc,splitname,nfields,ifields,ifields)); 3792fa5cd67SKarl Rupp } else { 3802c71b3e2SJacob Faibussowitsch PetscCheckFalse(!nfields || !nfields_col,PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot list zero fields"); 3812c71b3e2SJacob Faibussowitsch PetscCheckFalse(nfields != nfields_col,PETSC_COMM_SELF,PETSC_ERR_USER,"Number of row and column fields must match"); 3825f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetFields(pc,splitname,nfields,ifields,ifields_col)); 3835d4c12cdSJungho Lee } 3846c924f48SJed Brown } 3856c924f48SJed Brown if (i > 0) { 3866c924f48SJed Brown /* Makes command-line setting of splits take precedence over setting them in code. 3876c924f48SJed Brown Otherwise subsequent calls to PCFieldSplitSetIS() or PCFieldSplitSetFields() would 3886c924f48SJed Brown create new splits, which would probably not be what the user wanted. */ 3896c924f48SJed Brown jac->splitdefined = PETSC_TRUE; 3906c924f48SJed Brown } 3915f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(ifields)); 3925f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(ifields_col)); 3936c924f48SJed Brown PetscFunctionReturn(0); 3946c924f48SJed Brown } 3956c924f48SJed Brown 39669a612a9SBarry Smith static PetscErrorCode PCFieldSplitSetDefaults(PC pc) 3970971522cSBarry Smith { 3980971522cSBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 3995a9f2f41SSatish Balay PC_FieldSplitLink ilink = jac->head; 4007b752e3dSPatrick Sanan PetscBool fieldsplit_default = PETSC_FALSE,coupling = PETSC_FALSE; 4016c924f48SJed Brown PetscInt i; 4020971522cSBarry Smith 4030971522cSBarry Smith PetscFunctionBegin; 4047287d2a3SDmitry Karpeev /* 405f5f0d762SBarry Smith Kinda messy, but at least this now uses DMCreateFieldDecomposition(). 4067287d2a3SDmitry Karpeev Should probably be rewritten. 4077287d2a3SDmitry Karpeev */ 408f5f0d762SBarry Smith if (!ilink) { 4095f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsGetBool(((PetscObject)pc)->options,((PetscObject)pc)->prefix,"-pc_fieldsplit_detect_coupling",&coupling,NULL)); 4107b752e3dSPatrick Sanan if (pc->dm && jac->dm_splits && !jac->detect && !coupling) { 411bafc1b83SMatthew G Knepley PetscInt numFields, f, i, j; 4120784a22cSJed Brown char **fieldNames; 4137b62db95SJungho Lee IS *fields; 414e7c4fc90SDmitry Karpeev DM *dms; 415bafc1b83SMatthew G Knepley DM subdm[128]; 416bafc1b83SMatthew G Knepley PetscBool flg; 417bafc1b83SMatthew G Knepley 4185f80ce2aSJacob Faibussowitsch CHKERRQ(DMCreateFieldDecomposition(pc->dm, &numFields, &fieldNames, &fields, &dms)); 419bafc1b83SMatthew G Knepley /* Allow the user to prescribe the splits */ 420bafc1b83SMatthew G Knepley for (i = 0, flg = PETSC_TRUE;; i++) { 421bafc1b83SMatthew G Knepley PetscInt ifields[128]; 422bafc1b83SMatthew G Knepley IS compField; 423bafc1b83SMatthew G Knepley char optionname[128], splitname[8]; 424bafc1b83SMatthew G Knepley PetscInt nfields = numFields; 425bafc1b83SMatthew G Knepley 4265f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(optionname, sizeof(optionname), "-pc_fieldsplit_%D_fields", i)); 4275f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsGetIntArray(((PetscObject)pc)->options,((PetscObject)pc)->prefix, optionname, ifields, &nfields, &flg)); 428bafc1b83SMatthew G Knepley if (!flg) break; 4292c71b3e2SJacob Faibussowitsch PetscCheckFalse(numFields > 128,PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot currently support %d > 128 fields", numFields); 4305f80ce2aSJacob Faibussowitsch CHKERRQ(DMCreateSubDM(pc->dm, nfields, ifields, &compField, &subdm[i])); 431bafc1b83SMatthew G Knepley if (nfields == 1) { 4325f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetIS(pc, fieldNames[ifields[0]], compField)); 433bafc1b83SMatthew G Knepley } else { 4345f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(splitname, sizeof(splitname), "%D", i)); 4355f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetIS(pc, splitname, compField)); 4367287d2a3SDmitry Karpeev } 4375f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&compField)); 438bafc1b83SMatthew G Knepley for (j = 0; j < nfields; ++j) { 439bafc1b83SMatthew G Knepley f = ifields[j]; 4405f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(fieldNames[f])); 4415f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&fields[f])); 4427b62db95SJungho Lee } 443bafc1b83SMatthew G Knepley } 444bafc1b83SMatthew G Knepley if (i == 0) { 445bafc1b83SMatthew G Knepley for (f = 0; f < numFields; ++f) { 4465f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetIS(pc, fieldNames[f], fields[f])); 4475f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(fieldNames[f])); 4485f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&fields[f])); 449bafc1b83SMatthew G Knepley } 450bafc1b83SMatthew G Knepley } else { 451d724dfffSBarry Smith for (j=0; j<numFields; j++) { 4525f80ce2aSJacob Faibussowitsch CHKERRQ(DMDestroy(dms+j)); 453d724dfffSBarry Smith } 4545f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(dms)); 4555f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(i, &dms)); 4562fa5cd67SKarl Rupp for (j = 0; j < i; ++j) dms[j] = subdm[j]; 457bafc1b83SMatthew G Knepley } 4585f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(fieldNames)); 4595f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(fields)); 460e7c4fc90SDmitry Karpeev if (dms) { 4615f80ce2aSJacob Faibussowitsch CHKERRQ(PetscInfo(pc, "Setting up physics based fieldsplit preconditioner using the embedded DM\n")); 462bafc1b83SMatthew G Knepley for (ilink = jac->head, i = 0; ilink; ilink = ilink->next, ++i) { 4637287d2a3SDmitry Karpeev const char *prefix; 4645f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectGetOptionsPrefix((PetscObject)(ilink->ksp),&prefix)); 4655f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectSetOptionsPrefix((PetscObject)(dms[i]), prefix)); 4665f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetDM(ilink->ksp, dms[i])); 4675f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetDMActive(ilink->ksp, PETSC_FALSE)); 46895dbaa6fSMatthew G. Knepley { 46995dbaa6fSMatthew G. Knepley PetscErrorCode (*func)(KSP,Mat,Mat,void*); 47095dbaa6fSMatthew G. Knepley void *ctx; 47195dbaa6fSMatthew G. Knepley 4725f80ce2aSJacob Faibussowitsch CHKERRQ(DMKSPGetComputeOperators(pc->dm, &func, &ctx)); 4735f80ce2aSJacob Faibussowitsch CHKERRQ(DMKSPSetComputeOperators(dms[i], func, ctx)); 47495dbaa6fSMatthew G. Knepley } 4755f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectIncrementTabLevel((PetscObject)dms[i],(PetscObject)ilink->ksp,0)); 4765f80ce2aSJacob Faibussowitsch CHKERRQ(DMDestroy(&dms[i])); 4772fa5ba8aSJed Brown } 4785f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(dms)); 4798b8307b2SJed Brown } 48066ffff09SJed Brown } else { 481521d7252SBarry Smith if (jac->bs <= 0) { 482704ba839SBarry Smith if (pc->pmat) { 4835f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetBlockSize(pc->pmat,&jac->bs)); 4842fa5cd67SKarl Rupp } else jac->bs = 1; 485521d7252SBarry Smith } 486d32f9abdSBarry Smith 4877b752e3dSPatrick Sanan if (jac->detect) { 4886ce1633cSBarry Smith IS zerodiags,rest; 4896ce1633cSBarry Smith PetscInt nmin,nmax; 4906ce1633cSBarry Smith 4915f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOwnershipRange(pc->mat,&nmin,&nmax)); 4927199da05SBarry Smith if (jac->diag_use_amat) { 4935f80ce2aSJacob Faibussowitsch CHKERRQ(MatFindZeroDiagonals(pc->mat,&zerodiags)); 4947199da05SBarry Smith } else { 4955f80ce2aSJacob Faibussowitsch CHKERRQ(MatFindZeroDiagonals(pc->pmat,&zerodiags)); 4967199da05SBarry Smith } 4975f80ce2aSJacob Faibussowitsch CHKERRQ(ISComplement(zerodiags,nmin,nmax,&rest)); 4985f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetIS(pc,"0",rest)); 4995f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetIS(pc,"1",zerodiags)); 5005f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&zerodiags)); 5015f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&rest)); 5023a062f41SBarry Smith } else if (coupling) { 5033a062f41SBarry Smith IS coupling,rest; 5043a062f41SBarry Smith PetscInt nmin,nmax; 5053a062f41SBarry Smith 5065f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOwnershipRange(pc->mat,&nmin,&nmax)); 5077199da05SBarry Smith if (jac->offdiag_use_amat) { 5085f80ce2aSJacob Faibussowitsch CHKERRQ(MatFindOffBlockDiagonalEntries(pc->mat,&coupling)); 5097199da05SBarry Smith } else { 5105f80ce2aSJacob Faibussowitsch CHKERRQ(MatFindOffBlockDiagonalEntries(pc->pmat,&coupling)); 5117199da05SBarry Smith } 5125f80ce2aSJacob Faibussowitsch CHKERRQ(ISCreateStride(PetscObjectComm((PetscObject)pc->mat),nmax-nmin,nmin,1,&rest)); 5135f80ce2aSJacob Faibussowitsch CHKERRQ(ISSetIdentity(rest)); 5145f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetIS(pc,"0",rest)); 5155f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetIS(pc,"1",coupling)); 5165f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&coupling)); 5175f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&rest)); 5186ce1633cSBarry Smith } else { 5195f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsGetBool(((PetscObject)pc)->options,((PetscObject)pc)->prefix,"-pc_fieldsplit_default",&fieldsplit_default,NULL)); 5207287d2a3SDmitry Karpeev if (!fieldsplit_default) { 521d32f9abdSBarry Smith /* Allow user to set fields from command line, if bs was known at the time of PCSetFromOptions_FieldSplit() 522d32f9abdSBarry Smith then it is set there. This is not ideal because we should only have options set in XXSetFromOptions(). */ 5235f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetRuntimeSplits_Private(pc)); 5245f80ce2aSJacob Faibussowitsch if (jac->splitdefined) CHKERRQ(PetscInfo(pc,"Splits defined using the options database\n")); 525d32f9abdSBarry Smith } 5266dbb499eSCian Wilson if ((fieldsplit_default || !jac->splitdefined) && !jac->isrestrict) { 5279f001fe8SStefano Zampini Mat M = pc->pmat; 528f3b928b9SStefano Zampini PetscBool isnest; 529f3b928b9SStefano Zampini 5305f80ce2aSJacob Faibussowitsch CHKERRQ(PetscInfo(pc,"Using default splitting of fields\n")); 5315f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)pc->pmat,MATNEST,&isnest)); 532f3b928b9SStefano Zampini if (!isnest) { 5339f001fe8SStefano Zampini M = pc->mat; 5345f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)pc->mat,MATNEST,&isnest)); 535f3b928b9SStefano Zampini } 536f3b928b9SStefano Zampini if (isnest) { 537f3b928b9SStefano Zampini IS *fields; 538f3b928b9SStefano Zampini PetscInt nf; 539f3b928b9SStefano Zampini 5405f80ce2aSJacob Faibussowitsch CHKERRQ(MatNestGetSize(M,&nf,NULL)); 5415f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(nf,&fields)); 5425f80ce2aSJacob Faibussowitsch CHKERRQ(MatNestGetISs(M,fields,NULL)); 543f3b928b9SStefano Zampini for (i=0;i<nf;i++) { 5445f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetIS(pc,NULL,fields[i])); 545f3b928b9SStefano Zampini } 5465f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(fields)); 547f3b928b9SStefano Zampini } else { 548db4c96c1SJed Brown for (i=0; i<jac->bs; i++) { 5496c924f48SJed Brown char splitname[8]; 5505f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(splitname,sizeof(splitname),"%D",i)); 5515f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetFields(pc,splitname,1,&i,&i)); 55279416396SBarry Smith } 5535d4c12cdSJungho Lee jac->defaultsplit = PETSC_TRUE; 554521d7252SBarry Smith } 55566ffff09SJed Brown } 5566ce1633cSBarry Smith } 557f3b928b9SStefano Zampini } 558edf189efSBarry Smith } else if (jac->nsplits == 1) { 559edf189efSBarry Smith if (ilink->is) { 560edf189efSBarry Smith IS is2; 561edf189efSBarry Smith PetscInt nmin,nmax; 562edf189efSBarry Smith 5635f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOwnershipRange(pc->mat,&nmin,&nmax)); 5645f80ce2aSJacob Faibussowitsch CHKERRQ(ISComplement(ilink->is,nmin,nmax,&is2)); 5655f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetIS(pc,"1",is2)); 5665f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&is2)); 56782f516ccSBarry Smith } else SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Must provide at least two sets of fields to PCFieldSplit()"); 568edf189efSBarry Smith } 569d0af7cd3SBarry Smith 5702c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->nsplits < 2,PetscObjectComm((PetscObject)pc),PETSC_ERR_PLIB,"Unhandled case, must have at least two fields, not %d", jac->nsplits); 57169a612a9SBarry Smith PetscFunctionReturn(0); 57269a612a9SBarry Smith } 57369a612a9SBarry Smith 574a51937d4SCarola Kruse static PetscErrorCode MatGolubKahanComputeExplicitOperator(Mat A,Mat B,Mat C,Mat *H,PetscReal gkbnu) 575a51937d4SCarola Kruse { 576a51937d4SCarola Kruse Mat BT,T; 577de482cd7SCarola Kruse PetscReal nrmT,nrmB; 578a51937d4SCarola Kruse 579a51937d4SCarola Kruse PetscFunctionBegin; 5805f80ce2aSJacob Faibussowitsch CHKERRQ(MatHermitianTranspose(C,MAT_INITIAL_MATRIX,&T)); /* Test if augmented matrix is symmetric */ 5815f80ce2aSJacob Faibussowitsch CHKERRQ(MatAXPY(T,-1.0,B,DIFFERENT_NONZERO_PATTERN)); 5825f80ce2aSJacob Faibussowitsch CHKERRQ(MatNorm(T,NORM_1,&nrmT)); 5835f80ce2aSJacob Faibussowitsch CHKERRQ(MatNorm(B,NORM_1,&nrmB)); 584de482cd7SCarola Kruse if (nrmB > 0) { 585de482cd7SCarola Kruse if (nrmT/nrmB >= PETSC_SMALL) { 586de482cd7SCarola Kruse SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Matrix is not symmetric/hermitian, GKB is not applicable."); 587de482cd7SCarola Kruse } 588a51937d4SCarola Kruse } 589a51937d4SCarola Kruse /* Compute augmented Lagrangian matrix H = A00 + nu*A01*A01'. This corresponds to */ 590a51937d4SCarola Kruse /* setting N := 1/nu*I in [Ar13]. */ 5915f80ce2aSJacob Faibussowitsch CHKERRQ(MatHermitianTranspose(B,MAT_INITIAL_MATRIX,&BT)); 5925f80ce2aSJacob Faibussowitsch CHKERRQ(MatMatMult(B,BT,MAT_INITIAL_MATRIX,PETSC_DEFAULT,H)); /* H = A01*A01' */ 5935f80ce2aSJacob Faibussowitsch CHKERRQ(MatAYPX(*H,gkbnu,A,DIFFERENT_NONZERO_PATTERN)); /* H = A00 + nu*A01*A01' */ 594a51937d4SCarola Kruse 5955f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&BT)); 5965f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&T)); 597a51937d4SCarola Kruse PetscFunctionReturn(0); 598a51937d4SCarola Kruse } 599a51937d4SCarola Kruse 6002d747510SLisandro Dalcin PETSC_EXTERN PetscErrorCode PetscOptionsFindPairPrefix_Private(PetscOptions,const char pre[], const char name[],const char *value[],PetscBool *flg); 601514bf10dSMatthew G Knepley 60269a612a9SBarry Smith static PetscErrorCode PCSetUp_FieldSplit(PC pc) 60369a612a9SBarry Smith { 60469a612a9SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 6055a9f2f41SSatish Balay PC_FieldSplitLink ilink; 6062c9966d7SBarry Smith PetscInt i,nsplit; 6075f4ab4e1SJungho Lee PetscBool sorted, sorted_col; 60869a612a9SBarry Smith 60969a612a9SBarry Smith PetscFunctionBegin; 6105da88fe4STristan Konolige pc->failedreason = PC_NOERROR; 6115f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetDefaults(pc)); 61297bbdb24SBarry Smith nsplit = jac->nsplits; 6135a9f2f41SSatish Balay ilink = jac->head; 61497bbdb24SBarry Smith 61597bbdb24SBarry Smith /* get the matrices for each split */ 616704ba839SBarry Smith if (!jac->issetup) { 6171b9fc7fcSBarry Smith PetscInt rstart,rend,nslots,bs; 61897bbdb24SBarry Smith 619704ba839SBarry Smith jac->issetup = PETSC_TRUE; 620704ba839SBarry Smith 6215d4c12cdSJungho Lee /* This is done here instead of in PCFieldSplitSetFields() because may not have matrix at that point */ 6222c9966d7SBarry Smith if (jac->defaultsplit || !ilink->is) { 6232c9966d7SBarry Smith if (jac->bs <= 0) jac->bs = nsplit; 6242c9966d7SBarry Smith } 62551f519a2SBarry Smith bs = jac->bs; 6265f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOwnershipRange(pc->pmat,&rstart,&rend)); 6271b9fc7fcSBarry Smith nslots = (rend - rstart)/bs; 6281b9fc7fcSBarry Smith for (i=0; i<nsplit; i++) { 6291b9fc7fcSBarry Smith if (jac->defaultsplit) { 6305f80ce2aSJacob Faibussowitsch CHKERRQ(ISCreateStride(PetscObjectComm((PetscObject)pc),nslots,rstart+i,nsplit,&ilink->is)); 6315f80ce2aSJacob Faibussowitsch CHKERRQ(ISDuplicate(ilink->is,&ilink->is_col)); 632704ba839SBarry Smith } else if (!ilink->is) { 633ccb205f8SBarry Smith if (ilink->nfields > 1) { 6345f4ab4e1SJungho Lee PetscInt *ii,*jj,j,k,nfields = ilink->nfields,*fields = ilink->fields,*fields_col = ilink->fields_col; 6355f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(ilink->nfields*nslots,&ii)); 6365f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(ilink->nfields*nslots,&jj)); 6371b9fc7fcSBarry Smith for (j=0; j<nslots; j++) { 6381b9fc7fcSBarry Smith for (k=0; k<nfields; k++) { 6391b9fc7fcSBarry Smith ii[nfields*j + k] = rstart + bs*j + fields[k]; 6405f4ab4e1SJungho Lee jj[nfields*j + k] = rstart + bs*j + fields_col[k]; 64197bbdb24SBarry Smith } 64297bbdb24SBarry Smith } 6435f80ce2aSJacob Faibussowitsch CHKERRQ(ISCreateGeneral(PetscObjectComm((PetscObject)pc),nslots*nfields,ii,PETSC_OWN_POINTER,&ilink->is)); 6445f80ce2aSJacob Faibussowitsch CHKERRQ(ISCreateGeneral(PetscObjectComm((PetscObject)pc),nslots*nfields,jj,PETSC_OWN_POINTER,&ilink->is_col)); 6455f80ce2aSJacob Faibussowitsch CHKERRQ(ISSetBlockSize(ilink->is, nfields)); 6465f80ce2aSJacob Faibussowitsch CHKERRQ(ISSetBlockSize(ilink->is_col, nfields)); 647ccb205f8SBarry Smith } else { 6485f80ce2aSJacob Faibussowitsch CHKERRQ(ISCreateStride(PetscObjectComm((PetscObject)pc),nslots,rstart+ilink->fields[0],bs,&ilink->is)); 6495f80ce2aSJacob Faibussowitsch CHKERRQ(ISCreateStride(PetscObjectComm((PetscObject)pc),nslots,rstart+ilink->fields_col[0],bs,&ilink->is_col)); 650ccb205f8SBarry Smith } 6513e197d65SBarry Smith } 6525f80ce2aSJacob Faibussowitsch CHKERRQ(ISSorted(ilink->is,&sorted)); 6535f80ce2aSJacob Faibussowitsch if (ilink->is_col) CHKERRQ(ISSorted(ilink->is_col,&sorted_col)); 6542c71b3e2SJacob Faibussowitsch PetscCheckFalse(!sorted || !sorted_col,PETSC_COMM_SELF,PETSC_ERR_USER,"Fields must be sorted when creating split"); 655704ba839SBarry Smith ilink = ilink->next; 6561b9fc7fcSBarry Smith } 6571b9fc7fcSBarry Smith } 6581b9fc7fcSBarry Smith 659704ba839SBarry Smith ilink = jac->head; 66097bbdb24SBarry Smith if (!jac->pmat) { 661bdddcaaaSMatthew G Knepley Vec xtmp; 662bdddcaaaSMatthew G Knepley 6635f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateVecs(pc->pmat,&xtmp,NULL)); 6645f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(nsplit,&jac->pmat)); 6655f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc2(nsplit,&jac->x,nsplit,&jac->y)); 666cf502942SBarry Smith for (i=0; i<nsplit; i++) { 667bdddcaaaSMatthew G Knepley MatNullSpace sp; 668bdddcaaaSMatthew G Knepley 669a3df900dSMatthew G Knepley /* Check for preconditioning matrix attached to IS */ 6705f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectQuery((PetscObject) ilink->is, "pmat", (PetscObject*) &jac->pmat[i])); 671a3df900dSMatthew G Knepley if (jac->pmat[i]) { 6725f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectReference((PetscObject) jac->pmat[i])); 673a3df900dSMatthew G Knepley if (jac->type == PC_COMPOSITE_SCHUR) { 674a3df900dSMatthew G Knepley jac->schur_user = jac->pmat[i]; 6752fa5cd67SKarl Rupp 6765f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectReference((PetscObject) jac->schur_user)); 677a3df900dSMatthew G Knepley } 678a3df900dSMatthew G Knepley } else { 6793a062f41SBarry Smith const char *prefix; 6805f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,ilink->is_col,MAT_INITIAL_MATRIX,&jac->pmat[i])); 6815f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetOptionsPrefix(ilink->ksp,&prefix)); 6825f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetOptionsPrefix(jac->pmat[i],prefix)); 6835f80ce2aSJacob Faibussowitsch CHKERRQ(MatViewFromOptions(jac->pmat[i],NULL,"-mat_view")); 684a3df900dSMatthew G Knepley } 685bdddcaaaSMatthew G Knepley /* create work vectors for each split */ 6865f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateVecs(jac->pmat[i],&jac->x[i],&jac->y[i])); 6870298fd71SBarry Smith ilink->x = jac->x[i]; ilink->y = jac->y[i]; ilink->z = NULL; 688bdddcaaaSMatthew G Knepley /* compute scatter contexts needed by multiplicative versions and non-default splits */ 6895f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterCreate(xtmp,ilink->is,jac->x[i],NULL,&ilink->sctx)); 6905f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectQuery((PetscObject) ilink->is, "nearnullspace", (PetscObject*) &sp)); 691ed1f0337SMatthew G Knepley if (sp) { 6925f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetNearNullSpace(jac->pmat[i], sp)); 693ed1f0337SMatthew G Knepley } 694704ba839SBarry Smith ilink = ilink->next; 695cf502942SBarry Smith } 6965f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&xtmp)); 69797bbdb24SBarry Smith } else { 698ef7efd37SHong Zhang MatReuse scall; 699ef7efd37SHong Zhang if (pc->flag == DIFFERENT_NONZERO_PATTERN) { 700ef7efd37SHong Zhang for (i=0; i<nsplit; i++) { 7015f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->pmat[i])); 702ef7efd37SHong Zhang } 703ef7efd37SHong Zhang scall = MAT_INITIAL_MATRIX; 704ef7efd37SHong Zhang } else scall = MAT_REUSE_MATRIX; 705ef7efd37SHong Zhang 706cf502942SBarry Smith for (i=0; i<nsplit; i++) { 707a3df900dSMatthew G Knepley Mat pmat; 708a3df900dSMatthew G Knepley 709a3df900dSMatthew G Knepley /* Check for preconditioning matrix attached to IS */ 7105f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectQuery((PetscObject) ilink->is, "pmat", (PetscObject*) &pmat)); 711a3df900dSMatthew G Knepley if (!pmat) { 7125f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,ilink->is_col,scall,&jac->pmat[i])); 713a3df900dSMatthew G Knepley } 714704ba839SBarry Smith ilink = ilink->next; 715cf502942SBarry Smith } 71697bbdb24SBarry Smith } 7174e39094bSDmitry Karpeev if (jac->diag_use_amat) { 718519d70e2SJed Brown ilink = jac->head; 719519d70e2SJed Brown if (!jac->mat) { 7205f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(nsplit,&jac->mat)); 721519d70e2SJed Brown for (i=0; i<nsplit; i++) { 7225f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,ilink->is_col,MAT_INITIAL_MATRIX,&jac->mat[i])); 723519d70e2SJed Brown ilink = ilink->next; 724519d70e2SJed Brown } 725519d70e2SJed Brown } else { 726ef7efd37SHong Zhang MatReuse scall; 727ef7efd37SHong Zhang if (pc->flag == DIFFERENT_NONZERO_PATTERN) { 728519d70e2SJed Brown for (i=0; i<nsplit; i++) { 7295f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->mat[i])); 730ef7efd37SHong Zhang } 731ef7efd37SHong Zhang scall = MAT_INITIAL_MATRIX; 732ef7efd37SHong Zhang } else scall = MAT_REUSE_MATRIX; 733ef7efd37SHong Zhang 734ef7efd37SHong Zhang for (i=0; i<nsplit; i++) { 7355f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,ilink->is_col,scall,&jac->mat[i])); 736519d70e2SJed Brown ilink = ilink->next; 737519d70e2SJed Brown } 738519d70e2SJed Brown } 739519d70e2SJed Brown } else { 740519d70e2SJed Brown jac->mat = jac->pmat; 741519d70e2SJed Brown } 74297bbdb24SBarry Smith 74353935eafSBarry Smith /* Check for null space attached to IS */ 74453935eafSBarry Smith ilink = jac->head; 74553935eafSBarry Smith for (i=0; i<nsplit; i++) { 74653935eafSBarry Smith MatNullSpace sp; 74753935eafSBarry Smith 7485f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectQuery((PetscObject) ilink->is, "nullspace", (PetscObject*) &sp)); 74953935eafSBarry Smith if (sp) { 7505f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetNullSpace(jac->mat[i], sp)); 75153935eafSBarry Smith } 75253935eafSBarry Smith ilink = ilink->next; 75353935eafSBarry Smith } 75453935eafSBarry Smith 755a51937d4SCarola Kruse if (jac->type != PC_COMPOSITE_ADDITIVE && jac->type != PC_COMPOSITE_SCHUR && jac->type != PC_COMPOSITE_GKB) { 75668dd23aaSBarry Smith /* extract the rows of the matrix associated with each field: used for efficient computation of residual inside algorithm */ 7574e39094bSDmitry Karpeev /* FIXME: Can/should we reuse jac->mat whenever (jac->diag_use_amat) is true? */ 75868dd23aaSBarry Smith ilink = jac->head; 759e52d2c62SBarry Smith if (nsplit == 2 && jac->type == PC_COMPOSITE_MULTIPLICATIVE) { 760e52d2c62SBarry Smith /* special case need where Afield[0] is not needed and only certain columns of Afield[1] are needed since update is only on those rows of the solution */ 761e52d2c62SBarry Smith if (!jac->Afield) { 7625f80ce2aSJacob Faibussowitsch CHKERRQ(PetscCalloc1(nsplit,&jac->Afield)); 76380c96bb1SFande Kong if (jac->offdiag_use_amat) { 7645f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->next->is,ilink->is,MAT_INITIAL_MATRIX,&jac->Afield[1])); 765e52d2c62SBarry Smith } else { 7665f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->next->is,ilink->is,MAT_INITIAL_MATRIX,&jac->Afield[1])); 76780c96bb1SFande Kong } 76880c96bb1SFande Kong } else { 769ef7efd37SHong Zhang MatReuse scall; 770e9422dd5SStefano Zampini 771ef7efd37SHong Zhang if (pc->flag == DIFFERENT_NONZERO_PATTERN) { 7725f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->Afield[1])); 773ef7efd37SHong Zhang scall = MAT_INITIAL_MATRIX; 774ef7efd37SHong Zhang } else scall = MAT_REUSE_MATRIX; 775ef7efd37SHong Zhang 77680c96bb1SFande Kong if (jac->offdiag_use_amat) { 7775f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->next->is,ilink->is,scall,&jac->Afield[1])); 77880c96bb1SFande Kong } else { 7795f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->next->is,ilink->is,scall,&jac->Afield[1])); 78080c96bb1SFande Kong } 781e52d2c62SBarry Smith } 782e52d2c62SBarry Smith } else { 78368dd23aaSBarry Smith if (!jac->Afield) { 7845f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(nsplit,&jac->Afield)); 78568dd23aaSBarry Smith for (i=0; i<nsplit; i++) { 78680c96bb1SFande Kong if (jac->offdiag_use_amat) { 7875f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,NULL,MAT_INITIAL_MATRIX,&jac->Afield[i])); 78880c96bb1SFande Kong } else { 7895f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,NULL,MAT_INITIAL_MATRIX,&jac->Afield[i])); 79080c96bb1SFande Kong } 79168dd23aaSBarry Smith ilink = ilink->next; 79268dd23aaSBarry Smith } 79368dd23aaSBarry Smith } else { 794ef7efd37SHong Zhang MatReuse scall; 795ef7efd37SHong Zhang if (pc->flag == DIFFERENT_NONZERO_PATTERN) { 796ef7efd37SHong Zhang for (i=0; i<nsplit; i++) { 7975f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->Afield[i])); 798ef7efd37SHong Zhang } 799ef7efd37SHong Zhang scall = MAT_INITIAL_MATRIX; 800ef7efd37SHong Zhang } else scall = MAT_REUSE_MATRIX; 801ef7efd37SHong Zhang 80268dd23aaSBarry Smith for (i=0; i<nsplit; i++) { 80380c96bb1SFande Kong if (jac->offdiag_use_amat) { 8045f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,NULL,scall,&jac->Afield[i])); 80580c96bb1SFande Kong } else { 8065f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,NULL,scall,&jac->Afield[i])); 80780c96bb1SFande Kong } 80868dd23aaSBarry Smith ilink = ilink->next; 80968dd23aaSBarry Smith } 81068dd23aaSBarry Smith } 81168dd23aaSBarry Smith } 812e52d2c62SBarry Smith } 81368dd23aaSBarry Smith 8143b224e63SBarry Smith if (jac->type == PC_COMPOSITE_SCHUR) { 8153b224e63SBarry Smith IS ccis; 816c096484dSStefano Zampini PetscBool isspd; 8174aa3045dSJed Brown PetscInt rstart,rend; 818093c86ffSJed Brown char lscname[256]; 819093c86ffSJed Brown PetscObject LSC_L; 820ce94432eSBarry Smith 8212c71b3e2SJacob Faibussowitsch PetscCheckFalse(nsplit != 2,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_INCOMP,"To use Schur complement preconditioner you must have exactly 2 fields"); 82268dd23aaSBarry Smith 823c096484dSStefano Zampini /* If pc->mat is SPD, don't scale by -1 the Schur complement */ 824c096484dSStefano Zampini if (jac->schurscale == (PetscScalar)-1.0) { 8255f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOption(pc->pmat,MAT_SPD,&isspd)); 826c096484dSStefano Zampini jac->schurscale = (isspd == PETSC_TRUE) ? 1.0 : -1.0; 827c096484dSStefano Zampini } 828c096484dSStefano Zampini 829e6cab6aaSJed Brown /* When extracting off-diagonal submatrices, we take complements from this range */ 8305f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOwnershipRangeColumn(pc->mat,&rstart,&rend)); 831e6cab6aaSJed Brown 8323b224e63SBarry Smith if (jac->schur) { 8330298fd71SBarry Smith KSP kspA = jac->head->ksp, kspInner = NULL, kspUpper = jac->kspupper; 834e9422dd5SStefano Zampini MatReuse scall; 835e9422dd5SStefano Zampini 836e9422dd5SStefano Zampini if (pc->flag == DIFFERENT_NONZERO_PATTERN) { 837e9422dd5SStefano Zampini scall = MAT_INITIAL_MATRIX; 8385f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->B)); 8395f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->C)); 840e9422dd5SStefano Zampini } else scall = MAT_REUSE_MATRIX; 841443836d0SMatthew G Knepley 8425f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementGetKSP(jac->schur, &kspInner)); 8433b224e63SBarry Smith ilink = jac->head; 8445f80ce2aSJacob Faibussowitsch CHKERRQ(ISComplement(ilink->is_col,rstart,rend,&ccis)); 8454e39094bSDmitry Karpeev if (jac->offdiag_use_amat) { 8465f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,ccis,scall,&jac->B)); 847c84da90fSDmitry Karpeev } else { 8485f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,ccis,scall,&jac->B)); 849c84da90fSDmitry Karpeev } 8505f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ccis)); 8513b224e63SBarry Smith ilink = ilink->next; 8525f80ce2aSJacob Faibussowitsch CHKERRQ(ISComplement(ilink->is_col,rstart,rend,&ccis)); 8534e39094bSDmitry Karpeev if (jac->offdiag_use_amat) { 8545f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,ccis,scall,&jac->C)); 855c84da90fSDmitry Karpeev } else { 8565f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,ccis,scall,&jac->C)); 857c84da90fSDmitry Karpeev } 8585f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ccis)); 8595f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementUpdateSubMatrices(jac->schur,jac->mat[0],jac->pmat[0],jac->B,jac->C,jac->mat[1])); 860a7476a74SDmitry Karpeev if (jac->schurpre == PC_FIELDSPLIT_SCHUR_PRE_SELFP) { 8615f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->schurp)); 8625f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementGetPmat(jac->schur,MAT_INITIAL_MATRIX,&jac->schurp)); 863a7476a74SDmitry Karpeev } 864443836d0SMatthew G Knepley if (kspA != kspInner) { 8655f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOperators(kspA,jac->mat[0],jac->pmat[0])); 866443836d0SMatthew G Knepley } 867443836d0SMatthew G Knepley if (kspUpper != kspA) { 8685f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOperators(kspUpper,jac->mat[0],jac->pmat[0])); 869443836d0SMatthew G Knepley } 8705f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOperators(jac->kspschur,jac->schur,FieldSplitSchurPre(jac))); 8713b224e63SBarry Smith } else { 872bafc1b83SMatthew G Knepley const char *Dprefix; 873470b340bSDmitry Karpeev char schurprefix[256], schurmatprefix[256]; 874514bf10dSMatthew G Knepley char schurtestoption[256]; 875bdddcaaaSMatthew G Knepley MatNullSpace sp; 876514bf10dSMatthew G Knepley PetscBool flg; 877686bed4dSStefano Zampini KSP kspt; 8783b224e63SBarry Smith 879a04f6461SBarry Smith /* extract the A01 and A10 matrices */ 8803b224e63SBarry Smith ilink = jac->head; 8815f80ce2aSJacob Faibussowitsch CHKERRQ(ISComplement(ilink->is_col,rstart,rend,&ccis)); 8824e39094bSDmitry Karpeev if (jac->offdiag_use_amat) { 8835f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,ccis,MAT_INITIAL_MATRIX,&jac->B)); 884c84da90fSDmitry Karpeev } else { 8855f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,ccis,MAT_INITIAL_MATRIX,&jac->B)); 886c84da90fSDmitry Karpeev } 8875f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ccis)); 8883b224e63SBarry Smith ilink = ilink->next; 8895f80ce2aSJacob Faibussowitsch CHKERRQ(ISComplement(ilink->is_col,rstart,rend,&ccis)); 8904e39094bSDmitry Karpeev if (jac->offdiag_use_amat) { 8915f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,ccis,MAT_INITIAL_MATRIX,&jac->C)); 892c84da90fSDmitry Karpeev } else { 8935f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,ccis,MAT_INITIAL_MATRIX,&jac->C)); 894c84da90fSDmitry Karpeev } 8955f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ccis)); 89620252d06SBarry Smith 897f5236f50SJed Brown /* Use mat[0] (diagonal block of Amat) preconditioned by pmat[0] to define Schur complement */ 8985f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreate(((PetscObject)jac->mat[0])->comm,&jac->schur)); 8995f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetType(jac->schur,MATSCHURCOMPLEMENT)); 9005f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementSetSubMatrices(jac->schur,jac->mat[0],jac->pmat[0],jac->B,jac->C,jac->mat[1])); 9015f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(schurmatprefix, sizeof(schurmatprefix), "%sfieldsplit_%s_", ((PetscObject)pc)->prefix ? ((PetscObject)pc)->prefix : "", ilink->splitname)); 9025f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetOptionsPrefix(jac->schur,schurmatprefix)); 9035f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementGetKSP(jac->schur,&kspt)); 9045f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOptionsPrefix(kspt,schurmatprefix)); 905686bed4dSStefano Zampini 906686bed4dSStefano Zampini /* Note: this is not true in general */ 9075f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetNullSpace(jac->mat[1], &sp)); 90820252d06SBarry Smith if (sp) { 9095f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetNullSpace(jac->schur, sp)); 91020252d06SBarry Smith } 91120252d06SBarry Smith 9125f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(schurtestoption, sizeof(schurtestoption), "-fieldsplit_%s_inner_", ilink->splitname)); 9135f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsFindPairPrefix_Private(((PetscObject)pc)->options,((PetscObject)pc)->prefix, schurtestoption, NULL, &flg)); 914514bf10dSMatthew G Knepley if (flg) { 915514bf10dSMatthew G Knepley DM dmInner; 91621635b76SJed Brown KSP kspInner; 917686bed4dSStefano Zampini PC pcInner; 91821635b76SJed Brown 9195f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementGetKSP(jac->schur, &kspInner)); 9205f80ce2aSJacob Faibussowitsch CHKERRQ(KSPReset(kspInner)); 9215f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOperators(kspInner,jac->mat[0],jac->pmat[0])); 9225f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(schurprefix, sizeof(schurprefix), "%sfieldsplit_%s_inner_", ((PetscObject)pc)->prefix ? ((PetscObject)pc)->prefix : "", ilink->splitname)); 92321635b76SJed Brown /* Indent this deeper to emphasize the "inner" nature of this solver. */ 9245f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectIncrementTabLevel((PetscObject)kspInner, (PetscObject) pc, 2)); 9255f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectIncrementTabLevel((PetscObject)kspInner->pc, (PetscObject) pc, 2)); 9265f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOptionsPrefix(kspInner, schurprefix)); 927514bf10dSMatthew G Knepley 928514bf10dSMatthew G Knepley /* Set DM for new solver */ 9295f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetDM(jac->head->ksp, &dmInner)); 9305f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetDM(kspInner, dmInner)); 9315f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetDMActive(kspInner, PETSC_FALSE)); 932686bed4dSStefano Zampini 933686bed4dSStefano Zampini /* Defaults to PCKSP as preconditioner */ 9345f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetPC(kspInner, &pcInner)); 9355f80ce2aSJacob Faibussowitsch CHKERRQ(PCSetType(pcInner, PCKSP)); 9365f80ce2aSJacob Faibussowitsch CHKERRQ(PCKSPSetKSP(pcInner, jac->head->ksp)); 937514bf10dSMatthew G Knepley } else { 93821635b76SJed Brown /* Use the outer solver for the inner solve, but revert the KSPPREONLY from PCFieldSplitSetFields_FieldSplit or 93921635b76SJed Brown * PCFieldSplitSetIS_FieldSplit. We don't want KSPPREONLY because it makes the Schur complement inexact, 94021635b76SJed Brown * preventing Schur complement reduction to be an accurate solve. Usually when an iterative solver is used for 94121635b76SJed Brown * S = D - C A_inner^{-1} B, we expect S to be defined using an accurate definition of A_inner^{-1}, so we make 94221635b76SJed Brown * GMRES the default. Note that it is also common to use PREONLY for S, in which case S may not be used 94321635b76SJed Brown * directly, and the user is responsible for setting an inexact method for fieldsplit's A^{-1}. */ 9445f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetType(jac->head->ksp,KSPGMRES)); 9455f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementSetKSP(jac->schur,jac->head->ksp)); 946bafc1b83SMatthew G Knepley } 9475f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOperators(jac->head->ksp,jac->mat[0],jac->pmat[0])); 9485f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetFromOptions(jac->head->ksp)); 9495f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetFromOptions(jac->schur)); 9503b224e63SBarry Smith 9515f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)jac->schur, MATSCHURCOMPLEMENT, &flg)); 952686bed4dSStefano Zampini if (flg) { /* Need to do this otherwise PCSetUp_KSP will overwrite the amat of jac->head->ksp */ 953686bed4dSStefano Zampini KSP kspInner; 954686bed4dSStefano Zampini PC pcInner; 955686bed4dSStefano Zampini 9565f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementGetKSP(jac->schur, &kspInner)); 9575f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetPC(kspInner, &pcInner)); 9585f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)pcInner, PCKSP, &flg)); 959686bed4dSStefano Zampini if (flg) { 960686bed4dSStefano Zampini KSP ksp; 961686bed4dSStefano Zampini 9625f80ce2aSJacob Faibussowitsch CHKERRQ(PCKSPGetKSP(pcInner, &ksp)); 963686bed4dSStefano Zampini if (ksp == jac->head->ksp) { 9645f80ce2aSJacob Faibussowitsch CHKERRQ(PCSetUseAmat(pcInner, PETSC_TRUE)); 965686bed4dSStefano Zampini } 966686bed4dSStefano Zampini } 967686bed4dSStefano Zampini } 9685f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(schurtestoption, sizeof(schurtestoption), "-fieldsplit_%s_upper_", ilink->splitname)); 9695f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsFindPairPrefix_Private(((PetscObject)pc)->options,((PetscObject)pc)->prefix, schurtestoption, NULL, &flg)); 970443836d0SMatthew G Knepley if (flg) { 971443836d0SMatthew G Knepley DM dmInner; 972443836d0SMatthew G Knepley 9735f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(schurprefix, sizeof(schurprefix), "%sfieldsplit_%s_upper_", ((PetscObject)pc)->prefix ? ((PetscObject)pc)->prefix : "", ilink->splitname)); 9745f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCreate(PetscObjectComm((PetscObject)pc), &jac->kspupper)); 9755f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetErrorIfNotConverged(jac->kspupper,pc->erroriffailure)); 9765f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOptionsPrefix(jac->kspupper, schurprefix)); 9775f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectIncrementTabLevel((PetscObject)jac->kspupper, (PetscObject) pc, 1)); 9785f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectIncrementTabLevel((PetscObject)jac->kspupper->pc, (PetscObject) pc, 1)); 9795f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetDM(jac->head->ksp, &dmInner)); 9805f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetDM(jac->kspupper, dmInner)); 9815f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetDMActive(jac->kspupper, PETSC_FALSE)); 9825f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetFromOptions(jac->kspupper)); 9835f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOperators(jac->kspupper,jac->mat[0],jac->pmat[0])); 9845f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(jac->head->x, &jac->head->z)); 985443836d0SMatthew G Knepley } else { 986443836d0SMatthew G Knepley jac->kspupper = jac->head->ksp; 9875f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectReference((PetscObject) jac->head->ksp)); 988443836d0SMatthew G Knepley } 989443836d0SMatthew G Knepley 990a7476a74SDmitry Karpeev if (jac->schurpre == PC_FIELDSPLIT_SCHUR_PRE_SELFP) { 9915f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementGetPmat(jac->schur,MAT_INITIAL_MATRIX,&jac->schurp)); 992a7476a74SDmitry Karpeev } 9935f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCreate(PetscObjectComm((PetscObject)pc),&jac->kspschur)); 9945f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetErrorIfNotConverged(jac->kspschur,pc->erroriffailure)); 9955f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogObjectParent((PetscObject)pc,(PetscObject)jac->kspschur)); 9965f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectIncrementTabLevel((PetscObject)jac->kspschur,(PetscObject)pc,1)); 997084e4875SJed Brown if (jac->schurpre == PC_FIELDSPLIT_SCHUR_PRE_SELF) { 9987233a360SDmitry Karpeev PC pcschur; 9995f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetPC(jac->kspschur,&pcschur)); 10005f80ce2aSJacob Faibussowitsch CHKERRQ(PCSetType(pcschur,PCNONE)); 1001084e4875SJed Brown /* Note: This is bad if there exist preconditioners for MATSCHURCOMPLEMENT */ 1002e74569cdSMatthew G. Knepley } else if (jac->schurpre == PC_FIELDSPLIT_SCHUR_PRE_FULL) { 10035f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementComputeExplicitOperator(jac->schur, &jac->schur_user)); 1004e69d4d44SBarry Smith } 10055f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOperators(jac->kspschur,jac->schur,FieldSplitSchurPre(jac))); 10065f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetOptionsPrefix(jac->head->next->ksp, &Dprefix)); 10075f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOptionsPrefix(jac->kspschur, Dprefix)); 1008c096484dSStefano Zampini /* propagate DM */ 1009b20b4189SMatthew G. Knepley { 1010b20b4189SMatthew G. Knepley DM sdm; 10115f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetDM(jac->head->next->ksp, &sdm)); 1012b20b4189SMatthew G. Knepley if (sdm) { 10135f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetDM(jac->kspschur, sdm)); 10145f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetDMActive(jac->kspschur, PETSC_FALSE)); 1015b20b4189SMatthew G. Knepley } 1016b20b4189SMatthew G. Knepley } 10173b224e63SBarry Smith /* really want setfromoptions called in PCSetFromOptions_FieldSplit(), but it is not ready yet */ 101820b26d62SBarry Smith /* need to call this every time, since the jac->kspschur is freshly created, otherwise its options never get set */ 10195f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetFromOptions(jac->kspschur)); 10203b224e63SBarry Smith } 10215f80ce2aSJacob Faibussowitsch CHKERRQ(MatAssemblyBegin(jac->schur,MAT_FINAL_ASSEMBLY)); 10225f80ce2aSJacob Faibussowitsch CHKERRQ(MatAssemblyEnd(jac->schur,MAT_FINAL_ASSEMBLY)); 1023093c86ffSJed Brown 1024093c86ffSJed Brown /* HACK: special support to forward L and Lp matrices that might be used by PCLSC */ 10255f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(lscname,sizeof(lscname),"%s_LSC_L",ilink->splitname)); 10265f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectQuery((PetscObject)pc->mat,lscname,(PetscObject*)&LSC_L)); 10275f80ce2aSJacob Faibussowitsch if (!LSC_L) CHKERRQ(PetscObjectQuery((PetscObject)pc->pmat,lscname,(PetscObject*)&LSC_L)); 10285f80ce2aSJacob Faibussowitsch if (LSC_L) CHKERRQ(PetscObjectCompose((PetscObject)jac->schur,"LSC_L",(PetscObject)LSC_L)); 10295f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(lscname,sizeof(lscname),"%s_LSC_Lp",ilink->splitname)); 10305f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectQuery((PetscObject)pc->pmat,lscname,(PetscObject*)&LSC_L)); 10315f80ce2aSJacob Faibussowitsch if (!LSC_L) CHKERRQ(PetscObjectQuery((PetscObject)pc->mat,lscname,(PetscObject*)&LSC_L)); 10325f80ce2aSJacob Faibussowitsch if (LSC_L) CHKERRQ(PetscObjectCompose((PetscObject)jac->schur,"LSC_Lp",(PetscObject)LSC_L)); 1033a51937d4SCarola Kruse } else if (jac->type == PC_COMPOSITE_GKB) { 1034a51937d4SCarola Kruse IS ccis; 1035a51937d4SCarola Kruse PetscInt rstart,rend; 1036a51937d4SCarola Kruse 10372c71b3e2SJacob Faibussowitsch PetscCheckFalse(nsplit != 2,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_INCOMP,"To use GKB preconditioner you must have exactly 2 fields"); 1038a51937d4SCarola Kruse 1039a51937d4SCarola Kruse ilink = jac->head; 1040a51937d4SCarola Kruse 1041a51937d4SCarola Kruse /* When extracting off-diagonal submatrices, we take complements from this range */ 10425f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOwnershipRangeColumn(pc->mat,&rstart,&rend)); 1043a51937d4SCarola Kruse 10445f80ce2aSJacob Faibussowitsch CHKERRQ(ISComplement(ilink->is_col,rstart,rend,&ccis)); 1045a51937d4SCarola Kruse if (jac->offdiag_use_amat) { 10465f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,ccis,MAT_INITIAL_MATRIX,&jac->B)); 1047a51937d4SCarola Kruse } else { 10485f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,ccis,MAT_INITIAL_MATRIX,&jac->B)); 1049a51937d4SCarola Kruse } 10505f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ccis)); 1051e071a0a4SCarola Kruse /* Create work vectors for GKB algorithm */ 10525f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(ilink->x,&jac->u)); 10535f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(ilink->x,&jac->Hu)); 10545f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(ilink->x,&jac->w2)); 1055a51937d4SCarola Kruse ilink = ilink->next; 10565f80ce2aSJacob Faibussowitsch CHKERRQ(ISComplement(ilink->is_col,rstart,rend,&ccis)); 1057a51937d4SCarola Kruse if (jac->offdiag_use_amat) { 10585f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->mat,ilink->is,ccis,MAT_INITIAL_MATRIX,&jac->C)); 1059a51937d4SCarola Kruse } else { 10605f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(pc->pmat,ilink->is,ccis,MAT_INITIAL_MATRIX,&jac->C)); 1061a51937d4SCarola Kruse } 10625f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ccis)); 1063e071a0a4SCarola Kruse /* Create work vectors for GKB algorithm */ 10645f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(ilink->x,&jac->v)); 10655f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(ilink->x,&jac->d)); 10665f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(ilink->x,&jac->w1)); 10675f80ce2aSJacob Faibussowitsch CHKERRQ(MatGolubKahanComputeExplicitOperator(jac->mat[0],jac->B,jac->C,&jac->H,jac->gkbnu)); 10685f80ce2aSJacob Faibussowitsch CHKERRQ(PetscCalloc1(jac->gkbdelay,&jac->vecz)); 1069e071a0a4SCarola Kruse 1070a51937d4SCarola Kruse ilink = jac->head; 10715f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOperators(ilink->ksp,jac->H,jac->H)); 10725f80ce2aSJacob Faibussowitsch if (!jac->suboptionsset) CHKERRQ(KSPSetFromOptions(ilink->ksp)); 1073e071a0a4SCarola Kruse /* Create gkb_monitor context */ 1074de482cd7SCarola Kruse if (jac->gkbmonitor) { 1075de482cd7SCarola Kruse PetscInt tablevel; 10765f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerCreate(PETSC_COMM_WORLD,&jac->gkbviewer)); 10775f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerSetType(jac->gkbviewer,PETSCVIEWERASCII)); 10785f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectGetTabLevel((PetscObject)ilink->ksp,&tablevel)); 10795f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIISetTab(jac->gkbviewer,tablevel)); 10805f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectIncrementTabLevel((PetscObject)ilink->ksp,(PetscObject)ilink->ksp,1)); 1081de482cd7SCarola Kruse } 10823b224e63SBarry Smith } else { 108368bd789dSDmitry Karpeev /* set up the individual splits' PCs */ 108497bbdb24SBarry Smith i = 0; 10855a9f2f41SSatish Balay ilink = jac->head; 10865a9f2f41SSatish Balay while (ilink) { 10875f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOperators(ilink->ksp,jac->mat[i],jac->pmat[i])); 10883b224e63SBarry Smith /* really want setfromoptions called in PCSetFromOptions_FieldSplit(), but it is not ready yet */ 10895f80ce2aSJacob Faibussowitsch if (!jac->suboptionsset) CHKERRQ(KSPSetFromOptions(ilink->ksp)); 109097bbdb24SBarry Smith i++; 10915a9f2f41SSatish Balay ilink = ilink->next; 10920971522cSBarry Smith } 10933b224e63SBarry Smith } 10943b224e63SBarry Smith 10955ddf11f8SNicolas Barnafi /* Set coordinates to the sub PC objects whenever these are set */ 10965ddf11f8SNicolas Barnafi if (jac->coordinates_set) { 10975ddf11f8SNicolas Barnafi PC pc_coords; 10985ddf11f8SNicolas Barnafi if (jac->type == PC_COMPOSITE_SCHUR) { 10995ddf11f8SNicolas Barnafi // Head is first block. 11005f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetPC(jac->head->ksp, &pc_coords)); 11015f80ce2aSJacob Faibussowitsch CHKERRQ(PCSetCoordinates(pc_coords, jac->head->dim, jac->head->ndofs, jac->head->coords)); 11025ddf11f8SNicolas Barnafi // Second one is Schur block, but its KSP object is in kspschur. 11035f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetPC(jac->kspschur, &pc_coords)); 11045f80ce2aSJacob Faibussowitsch CHKERRQ(PCSetCoordinates(pc_coords, jac->head->next->dim, jac->head->next->ndofs, jac->head->next->coords)); 11055ddf11f8SNicolas Barnafi } else if (jac->type == PC_COMPOSITE_GKB) { 11065f80ce2aSJacob Faibussowitsch CHKERRQ(PetscInfo(pc, "Warning: Setting coordinates does nothing for the GKB Fieldpslit preconditioner")); 11075ddf11f8SNicolas Barnafi } else { 11085ddf11f8SNicolas Barnafi ilink = jac->head; 11095ddf11f8SNicolas Barnafi while (ilink) { 11105f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetPC(ilink->ksp, &pc_coords)); 11115f80ce2aSJacob Faibussowitsch CHKERRQ(PCSetCoordinates(pc_coords, ilink->dim, ilink->ndofs, ilink->coords)); 11125ddf11f8SNicolas Barnafi ilink = ilink->next; 11135ddf11f8SNicolas Barnafi } 11145ddf11f8SNicolas Barnafi } 11155ddf11f8SNicolas Barnafi } 11165ddf11f8SNicolas Barnafi 1117c1570756SJed Brown jac->suboptionsset = PETSC_TRUE; 11180971522cSBarry Smith PetscFunctionReturn(0); 11190971522cSBarry Smith } 11200971522cSBarry Smith 11215a9f2f41SSatish Balay #define FieldSplitSplitSolveAdd(ilink,xx,yy) \ 1122ca9f406cSSatish Balay (VecScatterBegin(ilink->sctx,xx,ilink->x,INSERT_VALUES,SCATTER_FORWARD) || \ 1123ca9f406cSSatish Balay VecScatterEnd(ilink->sctx,xx,ilink->x,INSERT_VALUES,SCATTER_FORWARD) || \ 11249df09d43SBarry Smith PetscLogEventBegin(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL) ||\ 11255a9f2f41SSatish Balay KSPSolve(ilink->ksp,ilink->x,ilink->y) || \ 1126c0decd05SBarry Smith KSPCheckSolve(ilink->ksp,pc,ilink->y) || \ 11279df09d43SBarry Smith PetscLogEventEnd(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL) ||\ 1128ca9f406cSSatish Balay VecScatterBegin(ilink->sctx,ilink->y,yy,ADD_VALUES,SCATTER_REVERSE) || \ 1129ca9f406cSSatish Balay VecScatterEnd(ilink->sctx,ilink->y,yy,ADD_VALUES,SCATTER_REVERSE)) 113079416396SBarry Smith 11313b224e63SBarry Smith static PetscErrorCode PCApply_FieldSplit_Schur(PC pc,Vec x,Vec y) 11323b224e63SBarry Smith { 11333b224e63SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 11343b224e63SBarry Smith PC_FieldSplitLink ilinkA = jac->head, ilinkD = ilinkA->next; 1135443836d0SMatthew G Knepley KSP kspA = ilinkA->ksp, kspLower = kspA, kspUpper = jac->kspupper; 11363b224e63SBarry Smith 11373b224e63SBarry Smith PetscFunctionBegin; 1138c5d2311dSJed Brown switch (jac->schurfactorization) { 1139c9c6ffaaSJed Brown case PC_FIELDSPLIT_SCHUR_FACT_DIAG: 1140a04f6461SBarry Smith /* [A00 0; 0 -S], positive definite, suitable for MINRES */ 11415f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,x,ilinkA->x,INSERT_VALUES,SCATTER_FORWARD)); 11425f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,x,ilinkD->x,INSERT_VALUES,SCATTER_FORWARD)); 11435f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,x,ilinkA->x,INSERT_VALUES,SCATTER_FORWARD)); 11445f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilinkA->event,kspA,ilinkA->x,ilinkA->y,NULL)); 11455f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(kspA,ilinkA->x,ilinkA->y)); 11465f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(kspA,pc,ilinkA->y)); 11475f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilinkA->event,kspA,ilinkA->x,ilinkA->y,NULL)); 11485f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 11495f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,x,ilinkD->x,INSERT_VALUES,SCATTER_FORWARD)); 11505f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(KSP_Solve_FS_S,jac->kspschur,ilinkD->x,ilinkD->y,NULL)); 11515f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(jac->kspschur,ilinkD->x,ilinkD->y)); 11525f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(jac->kspschur,pc,ilinkD->y)); 11535f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(KSP_Solve_FS_S,jac->kspschur,ilinkD->x,ilinkD->y,NULL)); 11545f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(ilinkD->y,jac->schurscale)); 11555f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 11565f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 11575f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 1158c5d2311dSJed Brown break; 1159c9c6ffaaSJed Brown case PC_FIELDSPLIT_SCHUR_FACT_LOWER: 1160a04f6461SBarry Smith /* [A00 0; A10 S], suitable for left preconditioning */ 11615f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,x,ilinkA->x,INSERT_VALUES,SCATTER_FORWARD)); 11625f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,x,ilinkA->x,INSERT_VALUES,SCATTER_FORWARD)); 11635f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilinkA->event,kspA,ilinkA->x,ilinkA->y,NULL)); 11645f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(kspA,ilinkA->x,ilinkA->y)); 11655f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(kspA,pc,ilinkA->y)); 11665f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilinkA->event,kspA,ilinkA->x,ilinkA->y,NULL)); 11675f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->C,ilinkA->y,ilinkD->x)); 11685f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(ilinkD->x,-1.)); 11695f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,x,ilinkD->x,ADD_VALUES,SCATTER_FORWARD)); 11705f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 11715f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,x,ilinkD->x,ADD_VALUES,SCATTER_FORWARD)); 11725f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(KSP_Solve_FS_S,jac->kspschur,ilinkD->x,ilinkD->y,NULL)); 11735f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(jac->kspschur,ilinkD->x,ilinkD->y)); 11745f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(jac->kspschur,pc,ilinkD->y)); 11755f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(KSP_Solve_FS_S,jac->kspschur,ilinkD->x,ilinkD->y,NULL)); 11765f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 11775f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 11785f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 1179c5d2311dSJed Brown break; 1180c9c6ffaaSJed Brown case PC_FIELDSPLIT_SCHUR_FACT_UPPER: 1181a04f6461SBarry Smith /* [A00 A01; 0 S], suitable for right preconditioning */ 11825f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,x,ilinkD->x,INSERT_VALUES,SCATTER_FORWARD)); 11835f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,x,ilinkD->x,INSERT_VALUES,SCATTER_FORWARD)); 11845f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(KSP_Solve_FS_S,jac->kspschur,ilinkD->x,ilinkD->y,NULL)); 11855f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(jac->kspschur,ilinkD->x,ilinkD->y)); 11865f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(jac->kspschur,pc,ilinkD->y)); 11875f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(KSP_Solve_FS_S,jac->kspschur,ilinkD->x,ilinkD->y,NULL)); CHKERRQ(MatMult(jac->B,ilinkD->y,ilinkA->x)); 11885f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(ilinkA->x,-1.)); 11895f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,x,ilinkA->x,ADD_VALUES,SCATTER_FORWARD)); 11905f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 11915f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,x,ilinkA->x,ADD_VALUES,SCATTER_FORWARD)); 11925f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilinkA->event,kspA,ilinkA->x,ilinkA->y,NULL)); 11935f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(kspA,ilinkA->x,ilinkA->y)); 11945f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(kspA,pc,ilinkA->y)); 11955f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilinkA->event,kspA,ilinkA->x,ilinkA->y,NULL)); 11965f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 11975f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 11985f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 1199c5d2311dSJed Brown break; 1200c9c6ffaaSJed Brown case PC_FIELDSPLIT_SCHUR_FACT_FULL: 1201c238f8cdSStefano Zampini /* [1 0; A10 A00^{-1} 1] [A00 0; 0 S] [1 A00^{-1}A01; 0 1] */ 12025f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,x,ilinkA->x,INSERT_VALUES,SCATTER_FORWARD)); 12035f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,x,ilinkA->x,INSERT_VALUES,SCATTER_FORWARD)); 12045f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(KSP_Solve_FS_L,kspLower,ilinkA->x,ilinkA->y,NULL)); 12055f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(kspLower,ilinkA->x,ilinkA->y)); 12065f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(kspLower,pc,ilinkA->y)); 12075f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(KSP_Solve_FS_L,kspLower,ilinkA->x,ilinkA->y,NULL)); 12085f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->C,ilinkA->y,ilinkD->x)); 12095f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(ilinkD->x,-1.0)); 12105f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,x,ilinkD->x,ADD_VALUES,SCATTER_FORWARD)); 12115f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,x,ilinkD->x,ADD_VALUES,SCATTER_FORWARD)); 12123b224e63SBarry Smith 12135f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(KSP_Solve_FS_S,jac->kspschur,ilinkD->x,ilinkD->y,NULL)); 12145f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(jac->kspschur,ilinkD->x,ilinkD->y)); 12155f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(jac->kspschur,pc,ilinkD->y)); 12165f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(KSP_Solve_FS_S,jac->kspschur,ilinkD->x,ilinkD->y,NULL)); 12175f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 12183b224e63SBarry Smith 1219443836d0SMatthew G Knepley if (kspUpper == kspA) { 12205f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->B,ilinkD->y,ilinkA->y)); 12215f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPY(ilinkA->x,-1.0,ilinkA->y)); 12225f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilinkA->event,kspA,ilinkA->x,ilinkA->y,NULL)); 12235f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(kspA,ilinkA->x,ilinkA->y)); 12245f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(kspA,pc,ilinkA->y)); 12255f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilinkA->event,kspA,ilinkA->x,ilinkA->y,NULL)); 1226443836d0SMatthew G Knepley } else { 12275f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilinkA->event,kspA,ilinkA->x,ilinkA->y,NULL)); 12285f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(kspA,ilinkA->x,ilinkA->y)); 12295f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(kspA,pc,ilinkA->y)); 12305f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->B,ilinkD->y,ilinkA->x)); 12315f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(KSP_Solve_FS_U,kspUpper,ilinkA->x,ilinkA->z,NULL)); 12325f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(kspUpper,ilinkA->x,ilinkA->z)); 12335f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(kspUpper,pc,ilinkA->z)); 12345f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(KSP_Solve_FS_U,kspUpper,ilinkA->x,ilinkA->z,NULL)); 12355f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPY(ilinkA->y,-1.0,ilinkA->z)); 1236443836d0SMatthew G Knepley } 12375f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 12385f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 12395f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 1240c5d2311dSJed Brown } 12413b224e63SBarry Smith PetscFunctionReturn(0); 12423b224e63SBarry Smith } 12433b224e63SBarry Smith 12440971522cSBarry Smith static PetscErrorCode PCApply_FieldSplit(PC pc,Vec x,Vec y) 12450971522cSBarry Smith { 12460971522cSBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 12475a9f2f41SSatish Balay PC_FieldSplitLink ilink = jac->head; 1248939b8a20SBarry Smith PetscInt cnt,bs; 12490971522cSBarry Smith 12500971522cSBarry Smith PetscFunctionBegin; 125179416396SBarry Smith if (jac->type == PC_COMPOSITE_ADDITIVE) { 12521b9fc7fcSBarry Smith if (jac->defaultsplit) { 12535f80ce2aSJacob Faibussowitsch CHKERRQ(VecGetBlockSize(x,&bs)); 12542c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->bs > 0 && bs != jac->bs,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONGSTATE,"Blocksize of x vector %D does not match fieldsplit blocksize %D",bs,jac->bs); 12555f80ce2aSJacob Faibussowitsch CHKERRQ(VecGetBlockSize(y,&bs)); 12562c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->bs > 0 && bs != jac->bs,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONGSTATE,"Blocksize of y vector %D does not match fieldsplit blocksize %D",bs,jac->bs); 12575f80ce2aSJacob Faibussowitsch CHKERRQ(VecStrideGatherAll(x,jac->x,INSERT_VALUES)); 12585a9f2f41SSatish Balay while (ilink) { 12595f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 12605f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(ilink->ksp,ilink->x,ilink->y)); 12615f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(ilink->ksp,pc,ilink->y)); 12625f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 12635a9f2f41SSatish Balay ilink = ilink->next; 12640971522cSBarry Smith } 12655f80ce2aSJacob Faibussowitsch CHKERRQ(VecStrideScatterAll(jac->y,y,INSERT_VALUES)); 12661b9fc7fcSBarry Smith } else { 12675f80ce2aSJacob Faibussowitsch CHKERRQ(VecSet(y,0.0)); 12685a9f2f41SSatish Balay while (ilink) { 12695f80ce2aSJacob Faibussowitsch CHKERRQ(FieldSplitSplitSolveAdd(ilink,x,y)); 12705a9f2f41SSatish Balay ilink = ilink->next; 12711b9fc7fcSBarry Smith } 12721b9fc7fcSBarry Smith } 1273e52d2c62SBarry Smith } else if (jac->type == PC_COMPOSITE_MULTIPLICATIVE && jac->nsplits == 2) { 12745f80ce2aSJacob Faibussowitsch CHKERRQ(VecSet(y,0.0)); 1275e52d2c62SBarry Smith /* solve on first block for first block variables */ 12765f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilink->sctx,x,ilink->x,INSERT_VALUES,SCATTER_FORWARD)); 12775f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilink->sctx,x,ilink->x,INSERT_VALUES,SCATTER_FORWARD)); 12785f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 12795f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(ilink->ksp,ilink->x,ilink->y)); 12805f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(ilink->ksp,pc,ilink->y)); 12815f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 12825f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilink->sctx,ilink->y,y,ADD_VALUES,SCATTER_REVERSE)); 12835f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilink->sctx,ilink->y,y,ADD_VALUES,SCATTER_REVERSE)); 1284e52d2c62SBarry Smith 1285e52d2c62SBarry Smith /* compute the residual only onto second block variables using first block variables */ 12865f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->Afield[1],ilink->y,ilink->next->x)); 1287e52d2c62SBarry Smith ilink = ilink->next; 12885f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(ilink->x,-1.0)); 12895f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilink->sctx,x,ilink->x,ADD_VALUES,SCATTER_FORWARD)); 12905f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilink->sctx,x,ilink->x,ADD_VALUES,SCATTER_FORWARD)); 1291e52d2c62SBarry Smith 1292e52d2c62SBarry Smith /* solve on second block variables */ 12935f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 12945f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(ilink->ksp,ilink->x,ilink->y)); 12955f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(ilink->ksp,pc,ilink->y)); 12965f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 12975f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilink->sctx,ilink->y,y,ADD_VALUES,SCATTER_REVERSE)); 12985f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilink->sctx,ilink->y,y,ADD_VALUES,SCATTER_REVERSE)); 129916913363SBarry Smith } else if (jac->type == PC_COMPOSITE_MULTIPLICATIVE || jac->type == PC_COMPOSITE_SYMMETRIC_MULTIPLICATIVE) { 130079416396SBarry Smith if (!jac->w1) { 13015f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(x,&jac->w1)); 13025f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(x,&jac->w2)); 130379416396SBarry Smith } 13045f80ce2aSJacob Faibussowitsch CHKERRQ(VecSet(y,0.0)); 13055f80ce2aSJacob Faibussowitsch CHKERRQ(FieldSplitSplitSolveAdd(ilink,x,y)); 13063e197d65SBarry Smith cnt = 1; 13075a9f2f41SSatish Balay while (ilink->next) { 13085a9f2f41SSatish Balay ilink = ilink->next; 13093e197d65SBarry Smith /* compute the residual only over the part of the vector needed */ 13105f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->Afield[cnt++],y,ilink->x)); 13115f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(ilink->x,-1.0)); 13125f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilink->sctx,x,ilink->x,ADD_VALUES,SCATTER_FORWARD)); 13135f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilink->sctx,x,ilink->x,ADD_VALUES,SCATTER_FORWARD)); 13145f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 13155f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(ilink->ksp,ilink->x,ilink->y)); 13165f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(ilink->ksp,pc,ilink->y)); 13175f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 13185f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilink->sctx,ilink->y,y,ADD_VALUES,SCATTER_REVERSE)); 13195f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilink->sctx,ilink->y,y,ADD_VALUES,SCATTER_REVERSE)); 13203e197d65SBarry Smith } 132151f519a2SBarry Smith if (jac->type == PC_COMPOSITE_SYMMETRIC_MULTIPLICATIVE) { 132211755939SBarry Smith cnt -= 2; 132351f519a2SBarry Smith while (ilink->previous) { 132451f519a2SBarry Smith ilink = ilink->previous; 132511755939SBarry Smith /* compute the residual only over the part of the vector needed */ 13265f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->Afield[cnt--],y,ilink->x)); 13275f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(ilink->x,-1.0)); 13285f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilink->sctx,x,ilink->x,ADD_VALUES,SCATTER_FORWARD)); 13295f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilink->sctx,x,ilink->x,ADD_VALUES,SCATTER_FORWARD)); 13305f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 13315f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(ilink->ksp,ilink->x,ilink->y)); 13325f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(ilink->ksp,pc,ilink->y)); 13335f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 13345f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilink->sctx,ilink->y,y,ADD_VALUES,SCATTER_REVERSE)); 13355f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilink->sctx,ilink->y,y,ADD_VALUES,SCATTER_REVERSE)); 133651f519a2SBarry Smith } 133711755939SBarry Smith } 133898921bdaSJacob Faibussowitsch } else SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Unsupported or unknown composition",(int) jac->type); 13390971522cSBarry Smith PetscFunctionReturn(0); 13400971522cSBarry Smith } 13410971522cSBarry Smith 1342a51937d4SCarola Kruse static PetscErrorCode PCApply_FieldSplit_GKB(PC pc,Vec x,Vec y) 1343a51937d4SCarola Kruse { 1344a51937d4SCarola Kruse PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 1345a51937d4SCarola Kruse PC_FieldSplitLink ilinkA = jac->head,ilinkD = ilinkA->next; 1346a51937d4SCarola Kruse KSP ksp = ilinkA->ksp; 1347de482cd7SCarola Kruse Vec u,v,Hu,d,work1,work2; 1348e071a0a4SCarola Kruse PetscScalar alpha,z,nrmz2,*vecz; 1349e071a0a4SCarola Kruse PetscReal lowbnd,nu,beta; 1350a51937d4SCarola Kruse PetscInt j,iterGKB; 1351a51937d4SCarola Kruse 1352a51937d4SCarola Kruse PetscFunctionBegin; 13535f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,x,ilinkA->x,INSERT_VALUES,SCATTER_FORWARD)); 13545f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,x,ilinkD->x,INSERT_VALUES,SCATTER_FORWARD)); 13555f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,x,ilinkA->x,INSERT_VALUES,SCATTER_FORWARD)); 13565f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,x,ilinkD->x,INSERT_VALUES,SCATTER_FORWARD)); 1357e071a0a4SCarola Kruse 1358e071a0a4SCarola Kruse u = jac->u; 1359e071a0a4SCarola Kruse v = jac->v; 1360e071a0a4SCarola Kruse Hu = jac->Hu; 1361e071a0a4SCarola Kruse d = jac->d; 1362e071a0a4SCarola Kruse work1 = jac->w1; 1363e071a0a4SCarola Kruse work2 = jac->w2; 1364e071a0a4SCarola Kruse vecz = jac->vecz; 1365a51937d4SCarola Kruse 1366a51937d4SCarola Kruse /* Change RHS to comply with matrix regularization H = A + nu*B*B' */ 1367a51937d4SCarola Kruse /* Add q = q + nu*B*b */ 1368a51937d4SCarola Kruse if (jac->gkbnu) { 1369a51937d4SCarola Kruse nu = jac->gkbnu; 13705f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(ilinkD->x,jac->gkbnu)); 13715f80ce2aSJacob Faibussowitsch CHKERRQ(MatMultAdd(jac->B,ilinkD->x,ilinkA->x,ilinkA->x)); /* q = q + nu*B*b */ 1372a51937d4SCarola Kruse } else { 1373a51937d4SCarola Kruse /* Situation when no augmented Lagrangian is used. Then we set inner */ 1374a51937d4SCarola Kruse /* matrix N = I in [Ar13], and thus nu = 1. */ 1375a51937d4SCarola Kruse nu = 1; 1376a51937d4SCarola Kruse } 1377a51937d4SCarola Kruse 1378a51937d4SCarola Kruse /* Transform rhs from [q,tilde{b}] to [0,b] */ 13795f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilinkA->event,ksp,ilinkA->x,ilinkA->y,NULL)); 13805f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(ksp,ilinkA->x,ilinkA->y)); 13815f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(ksp,pc,ilinkA->y)); 13825f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilinkA->event,ksp,ilinkA->x,ilinkA->y,NULL)); 13835f80ce2aSJacob Faibussowitsch CHKERRQ(MatMultHermitianTranspose(jac->B,ilinkA->y,work1)); 13845f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPBY(work1,1.0/nu,-1.0,ilinkD->x)); /* c = b - B'*x */ 1385a51937d4SCarola Kruse 1386a51937d4SCarola Kruse /* First step of algorithm */ 13875f80ce2aSJacob Faibussowitsch CHKERRQ(VecNorm(work1,NORM_2,&beta)); /* beta = sqrt(nu*c'*c)*/ 1388e071a0a4SCarola Kruse KSPCheckDot(ksp,beta); 1389addd1e01SJunchao Zhang beta = PetscSqrtReal(nu)*beta; 13905f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPBY(v,nu/beta,0.0,work1)); /* v = nu/beta *c */ 13915f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->B,v,work2)); /* u = H^{-1}*B*v */ 13925f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilinkA->event,ksp,work2,u,NULL)); 13935f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(ksp,work2,u)); 13945f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(ksp,pc,u)); 13955f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilinkA->event,ksp,work2,u,NULL)); 13965f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->H,u,Hu)); /* alpha = u'*H*u */ 13975f80ce2aSJacob Faibussowitsch CHKERRQ(VecDot(Hu,u,&alpha)); 1398e071a0a4SCarola Kruse KSPCheckDot(ksp,alpha); 13992c71b3e2SJacob Faibussowitsch PetscCheckFalse(PetscRealPart(alpha) <= 0.0,PETSC_COMM_SELF,PETSC_ERR_NOT_CONVERGED,"GKB preconditioner diverged, H is not positive definite"); 1400addd1e01SJunchao Zhang alpha = PetscSqrtReal(PetscAbsScalar(alpha)); 14015f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(u,1.0/alpha)); 14025f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPBY(d,1.0/alpha,0.0,v)); /* v = nu/beta *c */ 1403de482cd7SCarola Kruse 1404a51937d4SCarola Kruse z = beta/alpha; 1405a51937d4SCarola Kruse vecz[1] = z; 1406a51937d4SCarola Kruse 1407de482cd7SCarola Kruse /* Computation of first iterate x(1) and p(1) */ 14085f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPY(ilinkA->y,z,u)); 14095f80ce2aSJacob Faibussowitsch CHKERRQ(VecCopy(d,ilinkD->y)); 14105f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(ilinkD->y,-z)); 1411a51937d4SCarola Kruse 1412a51937d4SCarola Kruse iterGKB = 1; lowbnd = 2*jac->gkbtol; 1413de482cd7SCarola Kruse if (jac->gkbmonitor) { 14145f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(jac->gkbviewer,"%3D GKB Lower bound estimate %14.12e\n",iterGKB,lowbnd)); 1415de482cd7SCarola Kruse } 1416de482cd7SCarola Kruse 1417a51937d4SCarola Kruse while (iterGKB < jac->gkbmaxit && lowbnd > jac->gkbtol) { 1418a51937d4SCarola Kruse iterGKB += 1; 14195f80ce2aSJacob Faibussowitsch CHKERRQ(MatMultHermitianTranspose(jac->B,u,work1)); /* v <- nu*(B'*u-alpha/nu*v) */ 14205f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPBY(v,nu,-alpha,work1)); 14215f80ce2aSJacob Faibussowitsch CHKERRQ(VecNorm(v,NORM_2,&beta)); /* beta = sqrt(nu)*v'*v */ 1422addd1e01SJunchao Zhang beta = beta/PetscSqrtReal(nu); 14235f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(v,1.0/beta)); 14245f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->B,v,work2)); /* u <- H^{-1}*(B*v-beta*H*u) */ 14255f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->H,u,Hu)); 14265f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPY(work2,-beta,Hu)); 14275f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilinkA->event,ksp,work2,u,NULL)); 14285f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolve(ksp,work2,u)); 14295f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(ksp,pc,u)); 14305f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilinkA->event,ksp,work2,u,NULL)); 14315f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->H,u,Hu)); /* alpha = u'*H*u */ 14325f80ce2aSJacob Faibussowitsch CHKERRQ(VecDot(Hu,u,&alpha)); 1433e071a0a4SCarola Kruse KSPCheckDot(ksp,alpha); 14342c71b3e2SJacob Faibussowitsch PetscCheckFalse(PetscRealPart(alpha) <= 0.0,PETSC_COMM_SELF,PETSC_ERR_NOT_CONVERGED,"GKB preconditioner diverged, H is not positive definite"); 1435addd1e01SJunchao Zhang alpha = PetscSqrtReal(PetscAbsScalar(alpha)); 14365f80ce2aSJacob Faibussowitsch CHKERRQ(VecScale(u,1.0/alpha)); 1437a51937d4SCarola Kruse 1438e071a0a4SCarola Kruse z = -beta/alpha*z; /* z <- beta/alpha*z */ 1439a51937d4SCarola Kruse vecz[0] = z; 1440a51937d4SCarola Kruse 1441a51937d4SCarola Kruse /* Computation of new iterate x(i+1) and p(i+1) */ 14425f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPBY(d,1.0/alpha,-beta/alpha,v)); /* d = (v-beta*d)/alpha */ 14435f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPY(ilinkA->y,z,u)); /* r = r + z*u */ 14445f80ce2aSJacob Faibussowitsch CHKERRQ(VecAXPY(ilinkD->y,-z,d)); /* p = p - z*d */ 14455f80ce2aSJacob Faibussowitsch CHKERRQ(MatMult(jac->H,ilinkA->y,Hu)); /* ||u||_H = u'*H*u */ 14465f80ce2aSJacob Faibussowitsch CHKERRQ(VecDot(Hu,ilinkA->y,&nrmz2)); 1447a51937d4SCarola Kruse 1448a51937d4SCarola Kruse /* Compute Lower Bound estimate */ 1449a51937d4SCarola Kruse if (iterGKB > jac->gkbdelay) { 1450a51937d4SCarola Kruse lowbnd = 0.0; 1451a51937d4SCarola Kruse for (j=0; j<jac->gkbdelay; j++) { 1452a51937d4SCarola Kruse lowbnd += PetscAbsScalar(vecz[j]*vecz[j]); 1453a51937d4SCarola Kruse } 1454addd1e01SJunchao Zhang lowbnd = PetscSqrtReal(lowbnd/PetscAbsScalar(nrmz2)); 1455a51937d4SCarola Kruse } 1456a51937d4SCarola Kruse 1457a51937d4SCarola Kruse for (j=0; j<jac->gkbdelay-1; j++) { 1458a51937d4SCarola Kruse vecz[jac->gkbdelay-j-1] = vecz[jac->gkbdelay-j-2]; 1459a51937d4SCarola Kruse } 1460de482cd7SCarola Kruse if (jac->gkbmonitor) { 14615f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(jac->gkbviewer,"%3D GKB Lower bound estimate %14.12e\n",iterGKB,lowbnd)); 1462de482cd7SCarola Kruse } 1463a51937d4SCarola Kruse } 1464a51937d4SCarola Kruse 14655f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 14665f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkA->sctx,ilinkA->y,y,INSERT_VALUES,SCATTER_REVERSE)); 14675f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterBegin(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 14685f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterEnd(ilinkD->sctx,ilinkD->y,y,INSERT_VALUES,SCATTER_REVERSE)); 1469a51937d4SCarola Kruse 1470a51937d4SCarola Kruse PetscFunctionReturn(0); 1471a51937d4SCarola Kruse } 1472a51937d4SCarola Kruse 1473421e10b8SBarry Smith #define FieldSplitSplitSolveAddTranspose(ilink,xx,yy) \ 1474ca9f406cSSatish Balay (VecScatterBegin(ilink->sctx,xx,ilink->y,INSERT_VALUES,SCATTER_FORWARD) || \ 1475ca9f406cSSatish Balay VecScatterEnd(ilink->sctx,xx,ilink->y,INSERT_VALUES,SCATTER_FORWARD) || \ 14769df09d43SBarry Smith PetscLogEventBegin(ilink->event,ilink->ksp,ilink->y,ilink->x,NULL) || \ 1477421e10b8SBarry Smith KSPSolveTranspose(ilink->ksp,ilink->y,ilink->x) || \ 1478c0decd05SBarry Smith KSPCheckSolve(ilink->ksp,pc,ilink->x) || \ 1479c0decd05SBarry Smith PetscLogEventEnd(ilink->event,ilink->ksp,ilink->y,ilink->x,NULL) || \ 1480ca9f406cSSatish Balay VecScatterBegin(ilink->sctx,ilink->x,yy,ADD_VALUES,SCATTER_REVERSE) || \ 1481ca9f406cSSatish Balay VecScatterEnd(ilink->sctx,ilink->x,yy,ADD_VALUES,SCATTER_REVERSE)) 1482421e10b8SBarry Smith 1483421e10b8SBarry Smith static PetscErrorCode PCApplyTranspose_FieldSplit(PC pc,Vec x,Vec y) 1484421e10b8SBarry Smith { 1485421e10b8SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 1486421e10b8SBarry Smith PC_FieldSplitLink ilink = jac->head; 1487939b8a20SBarry Smith PetscInt bs; 1488421e10b8SBarry Smith 1489421e10b8SBarry Smith PetscFunctionBegin; 1490421e10b8SBarry Smith if (jac->type == PC_COMPOSITE_ADDITIVE) { 1491421e10b8SBarry Smith if (jac->defaultsplit) { 14925f80ce2aSJacob Faibussowitsch CHKERRQ(VecGetBlockSize(x,&bs)); 14932c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->bs > 0 && bs != jac->bs,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONGSTATE,"Blocksize of x vector %D does not match fieldsplit blocksize %D",bs,jac->bs); 14945f80ce2aSJacob Faibussowitsch CHKERRQ(VecGetBlockSize(y,&bs)); 14952c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->bs > 0 && bs != jac->bs,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONGSTATE,"Blocksize of y vector %D does not match fieldsplit blocksize %D",bs,jac->bs); 14965f80ce2aSJacob Faibussowitsch CHKERRQ(VecStrideGatherAll(x,jac->x,INSERT_VALUES)); 1497421e10b8SBarry Smith while (ilink) { 14985f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 14995f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSolveTranspose(ilink->ksp,ilink->x,ilink->y)); 15005f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCheckSolve(ilink->ksp,pc,ilink->y)); 15015f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(ilink->event,ilink->ksp,ilink->x,ilink->y,NULL)); 1502421e10b8SBarry Smith ilink = ilink->next; 1503421e10b8SBarry Smith } 15045f80ce2aSJacob Faibussowitsch CHKERRQ(VecStrideScatterAll(jac->y,y,INSERT_VALUES)); 1505421e10b8SBarry Smith } else { 15065f80ce2aSJacob Faibussowitsch CHKERRQ(VecSet(y,0.0)); 1507421e10b8SBarry Smith while (ilink) { 15085f80ce2aSJacob Faibussowitsch CHKERRQ(FieldSplitSplitSolveAddTranspose(ilink,x,y)); 1509421e10b8SBarry Smith ilink = ilink->next; 1510421e10b8SBarry Smith } 1511421e10b8SBarry Smith } 1512421e10b8SBarry Smith } else { 1513421e10b8SBarry Smith if (!jac->w1) { 15145f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(x,&jac->w1)); 15155f80ce2aSJacob Faibussowitsch CHKERRQ(VecDuplicate(x,&jac->w2)); 1516421e10b8SBarry Smith } 15175f80ce2aSJacob Faibussowitsch CHKERRQ(VecSet(y,0.0)); 1518421e10b8SBarry Smith if (jac->type == PC_COMPOSITE_SYMMETRIC_MULTIPLICATIVE) { 15195f80ce2aSJacob Faibussowitsch CHKERRQ(FieldSplitSplitSolveAddTranspose(ilink,x,y)); 1520421e10b8SBarry Smith while (ilink->next) { 1521421e10b8SBarry Smith ilink = ilink->next; 15225f80ce2aSJacob Faibussowitsch CHKERRQ(MatMultTranspose(pc->mat,y,jac->w1)); 15235f80ce2aSJacob Faibussowitsch CHKERRQ(VecWAXPY(jac->w2,-1.0,jac->w1,x)); 15245f80ce2aSJacob Faibussowitsch CHKERRQ(FieldSplitSplitSolveAddTranspose(ilink,jac->w2,y)); 1525421e10b8SBarry Smith } 1526421e10b8SBarry Smith while (ilink->previous) { 1527421e10b8SBarry Smith ilink = ilink->previous; 15285f80ce2aSJacob Faibussowitsch CHKERRQ(MatMultTranspose(pc->mat,y,jac->w1)); 15295f80ce2aSJacob Faibussowitsch CHKERRQ(VecWAXPY(jac->w2,-1.0,jac->w1,x)); 15305f80ce2aSJacob Faibussowitsch CHKERRQ(FieldSplitSplitSolveAddTranspose(ilink,jac->w2,y)); 1531421e10b8SBarry Smith } 1532421e10b8SBarry Smith } else { 1533421e10b8SBarry Smith while (ilink->next) { /* get to last entry in linked list */ 1534421e10b8SBarry Smith ilink = ilink->next; 1535421e10b8SBarry Smith } 15365f80ce2aSJacob Faibussowitsch CHKERRQ(FieldSplitSplitSolveAddTranspose(ilink,x,y)); 1537421e10b8SBarry Smith while (ilink->previous) { 1538421e10b8SBarry Smith ilink = ilink->previous; 15395f80ce2aSJacob Faibussowitsch CHKERRQ(MatMultTranspose(pc->mat,y,jac->w1)); 15405f80ce2aSJacob Faibussowitsch CHKERRQ(VecWAXPY(jac->w2,-1.0,jac->w1,x)); 15415f80ce2aSJacob Faibussowitsch CHKERRQ(FieldSplitSplitSolveAddTranspose(ilink,jac->w2,y)); 1542421e10b8SBarry Smith } 1543421e10b8SBarry Smith } 1544421e10b8SBarry Smith } 1545421e10b8SBarry Smith PetscFunctionReturn(0); 1546421e10b8SBarry Smith } 1547421e10b8SBarry Smith 1548574deadeSBarry Smith static PetscErrorCode PCReset_FieldSplit(PC pc) 15490971522cSBarry Smith { 15500971522cSBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 15515a9f2f41SSatish Balay PC_FieldSplitLink ilink = jac->head,next; 15520971522cSBarry Smith 15530971522cSBarry Smith PetscFunctionBegin; 15545a9f2f41SSatish Balay while (ilink) { 15555f80ce2aSJacob Faibussowitsch CHKERRQ(KSPDestroy(&ilink->ksp)); 15565f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&ilink->x)); 15575f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&ilink->y)); 15585f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&ilink->z)); 15595f80ce2aSJacob Faibussowitsch CHKERRQ(VecScatterDestroy(&ilink->sctx)); 15605f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ilink->is)); 15615f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ilink->is_col)); 15625f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(ilink->splitname)); 15635f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(ilink->fields)); 15645f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(ilink->fields_col)); 15655a9f2f41SSatish Balay next = ilink->next; 15665f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(ilink)); 15675a9f2f41SSatish Balay ilink = next; 15680971522cSBarry Smith } 1569f5f0d762SBarry Smith jac->head = NULL; 15705f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree2(jac->x,jac->y)); 1571574deadeSBarry Smith if (jac->mat && jac->mat != jac->pmat) { 15725f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroyMatrices(jac->nsplits,&jac->mat)); 1573574deadeSBarry Smith } else if (jac->mat) { 15740298fd71SBarry Smith jac->mat = NULL; 1575574deadeSBarry Smith } 15765f80ce2aSJacob Faibussowitsch if (jac->pmat) CHKERRQ(MatDestroyMatrices(jac->nsplits,&jac->pmat)); 15775f80ce2aSJacob Faibussowitsch if (jac->Afield) CHKERRQ(MatDestroyMatrices(jac->nsplits,&jac->Afield)); 1578f5f0d762SBarry Smith jac->nsplits = 0; 15795f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&jac->w1)); 15805f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&jac->w2)); 15815f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->schur)); 15825f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->schurp)); 15835f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->schur_user)); 15845f80ce2aSJacob Faibussowitsch CHKERRQ(KSPDestroy(&jac->kspschur)); 15855f80ce2aSJacob Faibussowitsch CHKERRQ(KSPDestroy(&jac->kspupper)); 15865f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->B)); 15875f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->C)); 15885f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->H)); 15895f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&jac->u)); 15905f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&jac->v)); 15915f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&jac->Hu)); 15925f80ce2aSJacob Faibussowitsch CHKERRQ(VecDestroy(&jac->d)); 15935f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(jac->vecz)); 15945f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerDestroy(&jac->gkbviewer)); 15956dbb499eSCian Wilson jac->isrestrict = PETSC_FALSE; 1596574deadeSBarry Smith PetscFunctionReturn(0); 1597574deadeSBarry Smith } 1598574deadeSBarry Smith 1599574deadeSBarry Smith static PetscErrorCode PCDestroy_FieldSplit(PC pc) 1600574deadeSBarry Smith { 1601574deadeSBarry Smith PetscFunctionBegin; 16025f80ce2aSJacob Faibussowitsch CHKERRQ(PCReset_FieldSplit(pc)); 16035f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(pc->data)); 16045f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSchurGetSubKSP_C",NULL)); 16055f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitGetSubKSP_C",NULL)); 16065f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetFields_C",NULL)); 16075f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetIS_C",NULL)); 16085f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetType_C",NULL)); 16095f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetBlockSize_C",NULL)); 16105f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetSchurPre_C",NULL)); 16115f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitGetSchurPre_C",NULL)); 16125f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetSchurFactType_C",NULL)); 16135f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitRestrictIS_C",NULL)); 16140971522cSBarry Smith PetscFunctionReturn(0); 16150971522cSBarry Smith } 16160971522cSBarry Smith 16174416b707SBarry Smith static PetscErrorCode PCSetFromOptions_FieldSplit(PetscOptionItems *PetscOptionsObject,PC pc) 16180971522cSBarry Smith { 16196c924f48SJed Brown PetscInt bs; 16207b752e3dSPatrick Sanan PetscBool flg; 16219dcbbd2bSBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 16223b224e63SBarry Smith PCCompositeType ctype; 16231b9fc7fcSBarry Smith 16240971522cSBarry Smith PetscFunctionBegin; 16255f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsHead(PetscOptionsObject,"FieldSplit options")); 16265f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsBool("-pc_fieldsplit_dm_splits","Whether to use DMCreateFieldDecomposition() for splits","PCFieldSplitSetDMSplits",jac->dm_splits,&jac->dm_splits,NULL)); 16275f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsInt("-pc_fieldsplit_block_size","Blocksize that defines number of fields","PCFieldSplitSetBlockSize",jac->bs,&bs,&flg)); 162851f519a2SBarry Smith if (flg) { 16295f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetBlockSize(pc,bs)); 163051f519a2SBarry Smith } 16312686e3e9SMatthew G. Knepley jac->diag_use_amat = pc->useAmat; 16325f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsBool("-pc_fieldsplit_diag_use_amat","Use Amat (not Pmat) to extract diagonal fieldsplit blocks", "PCFieldSplitSetDiagUseAmat",jac->diag_use_amat,&jac->diag_use_amat,NULL)); 16332686e3e9SMatthew G. Knepley jac->offdiag_use_amat = pc->useAmat; 16345f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsBool("-pc_fieldsplit_off_diag_use_amat","Use Amat (not Pmat) to extract off-diagonal fieldsplit blocks", "PCFieldSplitSetOffDiagUseAmat",jac->offdiag_use_amat,&jac->offdiag_use_amat,NULL)); 16355f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsBool("-pc_fieldsplit_detect_saddle_point","Form 2-way split by detecting zero diagonal entries", "PCFieldSplitSetDetectSaddlePoint",jac->detect,&jac->detect,NULL)); 16365f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetDetectSaddlePoint(pc,jac->detect)); /* Sets split type and Schur PC type */ 16375f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsEnum("-pc_fieldsplit_type","Type of composition","PCFieldSplitSetType",PCCompositeTypes,(PetscEnum)jac->type,(PetscEnum*)&ctype,&flg)); 16383b224e63SBarry Smith if (flg) { 16395f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetType(pc,ctype)); 16403b224e63SBarry Smith } 1641c30613efSMatthew Knepley /* Only setup fields once */ 1642c30613efSMatthew Knepley if ((jac->bs > 0) && (jac->nsplits == 0)) { 1643d32f9abdSBarry Smith /* only allow user to set fields from command line if bs is already known. 1644d32f9abdSBarry Smith otherwise user can set them in PCFieldSplitSetDefaults() */ 16455f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetRuntimeSplits_Private(pc)); 16465f80ce2aSJacob Faibussowitsch if (jac->splitdefined) CHKERRQ(PetscInfo(pc,"Splits defined using the options database\n")); 1647d32f9abdSBarry Smith } 1648c5d2311dSJed Brown if (jac->type == PC_COMPOSITE_SCHUR) { 16495f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsGetEnum(((PetscObject)pc)->options,((PetscObject)pc)->prefix,"-pc_fieldsplit_schur_factorization_type",PCFieldSplitSchurFactTypes,(PetscEnum*)&jac->schurfactorization,&flg)); 16505f80ce2aSJacob Faibussowitsch if (flg) CHKERRQ(PetscInfo(pc,"Deprecated use of -pc_fieldsplit_schur_factorization_type\n")); 16515f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsEnum("-pc_fieldsplit_schur_fact_type","Which off-diagonal parts of the block factorization to use","PCFieldSplitSetSchurFactType",PCFieldSplitSchurFactTypes,(PetscEnum)jac->schurfactorization,(PetscEnum*)&jac->schurfactorization,NULL)); 16525f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsEnum("-pc_fieldsplit_schur_precondition","How to build preconditioner for Schur complement","PCFieldSplitSetSchurPre",PCFieldSplitSchurPreTypes,(PetscEnum)jac->schurpre,(PetscEnum*)&jac->schurpre,NULL)); 16535f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsScalar("-pc_fieldsplit_schur_scale","Scale Schur complement","PCFieldSplitSetSchurScale",jac->schurscale,&jac->schurscale,NULL)); 1654a51937d4SCarola Kruse } else if (jac->type == PC_COMPOSITE_GKB) { 16555f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsReal("-pc_fieldsplit_gkb_tol","The tolerance for the lower bound stopping criterion","PCFieldSplitGKBTol",jac->gkbtol,&jac->gkbtol,NULL)); 16565f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsInt("-pc_fieldsplit_gkb_delay","The delay value for lower bound criterion","PCFieldSplitGKBDelay",jac->gkbdelay,&jac->gkbdelay,NULL)); 16575f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsReal("-pc_fieldsplit_gkb_nu","Parameter in augmented Lagrangian approach","PCFieldSplitGKBNu",jac->gkbnu,&jac->gkbnu,NULL)); 16582c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->gkbnu < 0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nu cannot be less than 0: value %f",jac->gkbnu); 16595f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsInt("-pc_fieldsplit_gkb_maxit","Maximum allowed number of iterations","PCFieldSplitGKBMaxit",jac->gkbmaxit,&jac->gkbmaxit,NULL)); 16605f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsBool("-pc_fieldsplit_gkb_monitor","Prints number of GKB iterations and error","PCFieldSplitGKB",jac->gkbmonitor,&jac->gkbmonitor,NULL)); 1661c5d2311dSJed Brown } 16625f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsTail()); 16630971522cSBarry Smith PetscFunctionReturn(0); 16640971522cSBarry Smith } 16650971522cSBarry Smith 16660971522cSBarry Smith /*------------------------------------------------------------------------------------*/ 16670971522cSBarry Smith 16681e6b0712SBarry Smith static PetscErrorCode PCFieldSplitSetFields_FieldSplit(PC pc,const char splitname[],PetscInt n,const PetscInt *fields,const PetscInt *fields_col) 16690971522cSBarry Smith { 167097bbdb24SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 16715a9f2f41SSatish Balay PC_FieldSplitLink ilink,next = jac->head; 167269a612a9SBarry Smith char prefix[128]; 16735d4c12cdSJungho Lee PetscInt i; 16740971522cSBarry Smith 16750971522cSBarry Smith PetscFunctionBegin; 16766c924f48SJed Brown if (jac->splitdefined) { 16775f80ce2aSJacob Faibussowitsch CHKERRQ(PetscInfo(pc,"Ignoring new split \"%s\" because the splits have already been defined\n",splitname)); 16786c924f48SJed Brown PetscFunctionReturn(0); 16796c924f48SJed Brown } 168051f519a2SBarry Smith for (i=0; i<n; i++) { 16812c71b3e2SJacob Faibussowitsch PetscCheckFalse(fields[i] >= jac->bs,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Field %D requested but only %D exist",fields[i],jac->bs); 16822c71b3e2SJacob Faibussowitsch PetscCheckFalse(fields[i] < 0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative field %D requested",fields[i]); 168351f519a2SBarry Smith } 16845f80ce2aSJacob Faibussowitsch CHKERRQ(PetscNew(&ilink)); 1685a04f6461SBarry Smith if (splitname) { 16865f80ce2aSJacob Faibussowitsch CHKERRQ(PetscStrallocpy(splitname,&ilink->splitname)); 1687a04f6461SBarry Smith } else { 16885f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(3,&ilink->splitname)); 16895f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(ilink->splitname,2,"%s",jac->nsplits)); 1690a04f6461SBarry Smith } 16919df09d43SBarry Smith ilink->event = jac->nsplits < 5 ? KSP_Solve_FS_0 + jac->nsplits : KSP_Solve_FS_0 + 4; /* Any split great than 4 gets logged in the 4th split */ 16925f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(n,&ilink->fields)); 16935f80ce2aSJacob Faibussowitsch CHKERRQ(PetscArraycpy(ilink->fields,fields,n)); 16945f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(n,&ilink->fields_col)); 16955f80ce2aSJacob Faibussowitsch CHKERRQ(PetscArraycpy(ilink->fields_col,fields_col,n)); 16962fa5cd67SKarl Rupp 16975a9f2f41SSatish Balay ilink->nfields = n; 16980298fd71SBarry Smith ilink->next = NULL; 16995f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCreate(PetscObjectComm((PetscObject)pc),&ilink->ksp)); 17005f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetErrorIfNotConverged(ilink->ksp,pc->erroriffailure)); 17015f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectIncrementTabLevel((PetscObject)ilink->ksp,(PetscObject)pc,1)); 17025f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetType(ilink->ksp,KSPPREONLY)); 17035f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogObjectParent((PetscObject)pc,(PetscObject)ilink->ksp)); 170469a612a9SBarry Smith 17055f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(prefix,sizeof(prefix),"%sfieldsplit_%s_",((PetscObject)pc)->prefix ? ((PetscObject)pc)->prefix : "",ilink->splitname)); 17065f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOptionsPrefix(ilink->ksp,prefix)); 17070971522cSBarry Smith 17080971522cSBarry Smith if (!next) { 17095a9f2f41SSatish Balay jac->head = ilink; 17100298fd71SBarry Smith ilink->previous = NULL; 17110971522cSBarry Smith } else { 17120971522cSBarry Smith while (next->next) { 17130971522cSBarry Smith next = next->next; 17140971522cSBarry Smith } 17155a9f2f41SSatish Balay next->next = ilink; 171651f519a2SBarry Smith ilink->previous = next; 17170971522cSBarry Smith } 17180971522cSBarry Smith jac->nsplits++; 17190971522cSBarry Smith PetscFunctionReturn(0); 17200971522cSBarry Smith } 17210971522cSBarry Smith 1722285fb4e2SStefano Zampini static PetscErrorCode PCFieldSplitSchurGetSubKSP_FieldSplit(PC pc,PetscInt *n,KSP **subksp) 1723285fb4e2SStefano Zampini { 1724285fb4e2SStefano Zampini PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 1725285fb4e2SStefano Zampini 1726285fb4e2SStefano Zampini PetscFunctionBegin; 1727285fb4e2SStefano Zampini *subksp = NULL; 1728285fb4e2SStefano Zampini if (n) *n = 0; 1729285fb4e2SStefano Zampini if (jac->type == PC_COMPOSITE_SCHUR) { 1730285fb4e2SStefano Zampini PetscInt nn; 1731285fb4e2SStefano Zampini 1732*28b400f6SJacob Faibussowitsch PetscCheck(jac->schur,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONGSTATE,"Must call KSPSetUp() or PCSetUp() before calling PCFieldSplitSchurGetSubKSP()"); 17332c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->nsplits != 2,PetscObjectComm((PetscObject)pc),PETSC_ERR_PLIB,"Unexpected number of splits %D != 2",jac->nsplits); 1734285fb4e2SStefano Zampini nn = jac->nsplits + (jac->kspupper != jac->head->ksp ? 1 : 0); 17355f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(nn,subksp)); 1736285fb4e2SStefano Zampini (*subksp)[0] = jac->head->ksp; 1737285fb4e2SStefano Zampini (*subksp)[1] = jac->kspschur; 1738285fb4e2SStefano Zampini if (jac->kspupper != jac->head->ksp) (*subksp)[2] = jac->kspupper; 1739285fb4e2SStefano Zampini if (n) *n = nn; 1740285fb4e2SStefano Zampini } 1741285fb4e2SStefano Zampini PetscFunctionReturn(0); 1742285fb4e2SStefano Zampini } 1743285fb4e2SStefano Zampini 17441e6b0712SBarry Smith static PetscErrorCode PCFieldSplitGetSubKSP_FieldSplit_Schur(PC pc,PetscInt *n,KSP **subksp) 1745e69d4d44SBarry Smith { 1746e69d4d44SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 1747e69d4d44SBarry Smith 1748e69d4d44SBarry Smith PetscFunctionBegin; 1749*28b400f6SJacob Faibussowitsch PetscCheck(jac->schur,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONGSTATE,"Must call KSPSetUp() or PCSetUp() before calling PCFieldSplitGetSubKSP()"); 17505f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(jac->nsplits,subksp)); 17515f80ce2aSJacob Faibussowitsch CHKERRQ(MatSchurComplementGetKSP(jac->schur,*subksp)); 17522fa5cd67SKarl Rupp 1753e69d4d44SBarry Smith (*subksp)[1] = jac->kspschur; 175413e0d083SBarry Smith if (n) *n = jac->nsplits; 1755e69d4d44SBarry Smith PetscFunctionReturn(0); 1756e69d4d44SBarry Smith } 17570971522cSBarry Smith 17581e6b0712SBarry Smith static PetscErrorCode PCFieldSplitGetSubKSP_FieldSplit(PC pc,PetscInt *n,KSP **subksp) 17590971522cSBarry Smith { 17600971522cSBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 17610971522cSBarry Smith PetscInt cnt = 0; 17625a9f2f41SSatish Balay PC_FieldSplitLink ilink = jac->head; 17630971522cSBarry Smith 17640971522cSBarry Smith PetscFunctionBegin; 17655f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(jac->nsplits,subksp)); 17665a9f2f41SSatish Balay while (ilink) { 17675a9f2f41SSatish Balay (*subksp)[cnt++] = ilink->ksp; 17685a9f2f41SSatish Balay ilink = ilink->next; 17690971522cSBarry Smith } 17702c71b3e2SJacob Faibussowitsch PetscCheckFalse(cnt != jac->nsplits,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt PCFIELDSPLIT object: number of splits in linked list %D does not match number in object %D",cnt,jac->nsplits); 177113e0d083SBarry Smith if (n) *n = jac->nsplits; 17720971522cSBarry Smith PetscFunctionReturn(0); 17730971522cSBarry Smith } 17740971522cSBarry Smith 17756dbb499eSCian Wilson /*@C 17766dbb499eSCian Wilson PCFieldSplitRestrictIS - Restricts the fieldsplit ISs to be within a given IS. 17776dbb499eSCian Wilson 17786dbb499eSCian Wilson Input Parameters: 17796dbb499eSCian Wilson + pc - the preconditioner context 1780a2b725a8SWilliam Gropp - is - the index set that defines the indices to which the fieldsplit is to be restricted 17816dbb499eSCian Wilson 17826dbb499eSCian Wilson Level: advanced 17836dbb499eSCian Wilson 17846dbb499eSCian Wilson @*/ 17856dbb499eSCian Wilson PetscErrorCode PCFieldSplitRestrictIS(PC pc,IS isy) 17866dbb499eSCian Wilson { 17876dbb499eSCian Wilson PetscFunctionBegin; 17886dbb499eSCian Wilson PetscValidHeaderSpecific(pc,PC_CLASSID,1); 17896dbb499eSCian Wilson PetscValidHeaderSpecific(isy,IS_CLASSID,2); 17905f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitRestrictIS_C",(PC,IS),(pc,isy))); 17916dbb499eSCian Wilson PetscFunctionReturn(0); 17926dbb499eSCian Wilson } 17936dbb499eSCian Wilson 17946dbb499eSCian Wilson static PetscErrorCode PCFieldSplitRestrictIS_FieldSplit(PC pc, IS isy) 17956dbb499eSCian Wilson { 17966dbb499eSCian Wilson PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 17976dbb499eSCian Wilson PC_FieldSplitLink ilink = jac->head, next; 17986dbb499eSCian Wilson PetscInt localsize,size,sizez,i; 17996dbb499eSCian Wilson const PetscInt *ind, *indz; 18006dbb499eSCian Wilson PetscInt *indc, *indcz; 18016dbb499eSCian Wilson PetscBool flg; 18026dbb499eSCian Wilson 18036dbb499eSCian Wilson PetscFunctionBegin; 18045f80ce2aSJacob Faibussowitsch CHKERRQ(ISGetLocalSize(isy,&localsize)); 18055f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Scan(&localsize,&size,1,MPIU_INT,MPI_SUM,PetscObjectComm((PetscObject)isy))); 18066dbb499eSCian Wilson size -= localsize; 18076dbb499eSCian Wilson while (ilink) { 18086dbb499eSCian Wilson IS isrl,isr; 18091c7cfba8SBarry Smith PC subpc; 18105f80ce2aSJacob Faibussowitsch CHKERRQ(ISEmbed(ilink->is, isy, PETSC_TRUE, &isrl)); 18115f80ce2aSJacob Faibussowitsch CHKERRQ(ISGetLocalSize(isrl,&localsize)); 18125f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(localsize,&indc)); 18135f80ce2aSJacob Faibussowitsch CHKERRQ(ISGetIndices(isrl,&ind)); 18145f80ce2aSJacob Faibussowitsch CHKERRQ(PetscArraycpy(indc,ind,localsize)); 18155f80ce2aSJacob Faibussowitsch CHKERRQ(ISRestoreIndices(isrl,&ind)); 18165f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&isrl)); 18176dbb499eSCian Wilson for (i=0; i<localsize; i++) *(indc+i) += size; 18185f80ce2aSJacob Faibussowitsch CHKERRQ(ISCreateGeneral(PetscObjectComm((PetscObject)isy),localsize,indc,PETSC_OWN_POINTER,&isr)); 18195f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectReference((PetscObject)isr)); 18205f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ilink->is)); 18216dbb499eSCian Wilson ilink->is = isr; 18225f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectReference((PetscObject)isr)); 18235f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ilink->is_col)); 18246dbb499eSCian Wilson ilink->is_col = isr; 18255f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&isr)); 18265f80ce2aSJacob Faibussowitsch CHKERRQ(KSPGetPC(ilink->ksp, &subpc)); 18275f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)subpc,PCFIELDSPLIT,&flg)); 18286dbb499eSCian Wilson if (flg) { 18296dbb499eSCian Wilson IS iszl,isz; 18306dbb499eSCian Wilson MPI_Comm comm; 18315f80ce2aSJacob Faibussowitsch CHKERRQ(ISGetLocalSize(ilink->is,&localsize)); 18326dbb499eSCian Wilson comm = PetscObjectComm((PetscObject)ilink->is); 18335f80ce2aSJacob Faibussowitsch CHKERRQ(ISEmbed(isy, ilink->is, PETSC_TRUE, &iszl)); 18345f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Scan(&localsize,&sizez,1,MPIU_INT,MPI_SUM,comm)); 18356dbb499eSCian Wilson sizez -= localsize; 18365f80ce2aSJacob Faibussowitsch CHKERRQ(ISGetLocalSize(iszl,&localsize)); 18375f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(localsize,&indcz)); 18385f80ce2aSJacob Faibussowitsch CHKERRQ(ISGetIndices(iszl,&indz)); 18395f80ce2aSJacob Faibussowitsch CHKERRQ(PetscArraycpy(indcz,indz,localsize)); 18405f80ce2aSJacob Faibussowitsch CHKERRQ(ISRestoreIndices(iszl,&indz)); 18415f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&iszl)); 18426dbb499eSCian Wilson for (i=0; i<localsize; i++) *(indcz+i) += sizez; 18435f80ce2aSJacob Faibussowitsch CHKERRQ(ISCreateGeneral(comm,localsize,indcz,PETSC_OWN_POINTER,&isz)); 18445f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitRestrictIS(subpc,isz)); 18455f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&isz)); 18466dbb499eSCian Wilson } 18476dbb499eSCian Wilson next = ilink->next; 18486dbb499eSCian Wilson ilink = next; 18496dbb499eSCian Wilson } 18506dbb499eSCian Wilson jac->isrestrict = PETSC_TRUE; 18516dbb499eSCian Wilson PetscFunctionReturn(0); 18526dbb499eSCian Wilson } 18536dbb499eSCian Wilson 18541e6b0712SBarry Smith static PetscErrorCode PCFieldSplitSetIS_FieldSplit(PC pc,const char splitname[],IS is) 1855704ba839SBarry Smith { 1856704ba839SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 1857704ba839SBarry Smith PC_FieldSplitLink ilink, next = jac->head; 1858704ba839SBarry Smith char prefix[128]; 1859704ba839SBarry Smith 1860704ba839SBarry Smith PetscFunctionBegin; 18616c924f48SJed Brown if (jac->splitdefined) { 18625f80ce2aSJacob Faibussowitsch CHKERRQ(PetscInfo(pc,"Ignoring new split \"%s\" because the splits have already been defined\n",splitname)); 18636c924f48SJed Brown PetscFunctionReturn(0); 18646c924f48SJed Brown } 18655f80ce2aSJacob Faibussowitsch CHKERRQ(PetscNew(&ilink)); 1866a04f6461SBarry Smith if (splitname) { 18675f80ce2aSJacob Faibussowitsch CHKERRQ(PetscStrallocpy(splitname,&ilink->splitname)); 1868a04f6461SBarry Smith } else { 18695f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(8,&ilink->splitname)); 18705f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(ilink->splitname,7,"%D",jac->nsplits)); 1871a04f6461SBarry Smith } 18729df09d43SBarry Smith ilink->event = jac->nsplits < 5 ? KSP_Solve_FS_0 + jac->nsplits : KSP_Solve_FS_0 + 4; /* Any split great than 4 gets logged in the 4th split */ 18735f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectReference((PetscObject)is)); 18745f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ilink->is)); 1875b5787286SJed Brown ilink->is = is; 18765f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectReference((PetscObject)is)); 18775f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&ilink->is_col)); 1878b5787286SJed Brown ilink->is_col = is; 18790298fd71SBarry Smith ilink->next = NULL; 18805f80ce2aSJacob Faibussowitsch CHKERRQ(KSPCreate(PetscObjectComm((PetscObject)pc),&ilink->ksp)); 18815f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetErrorIfNotConverged(ilink->ksp,pc->erroriffailure)); 18825f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectIncrementTabLevel((PetscObject)ilink->ksp,(PetscObject)pc,1)); 18835f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetType(ilink->ksp,KSPPREONLY)); 18845f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogObjectParent((PetscObject)pc,(PetscObject)ilink->ksp)); 1885704ba839SBarry Smith 18865f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintf(prefix,sizeof(prefix),"%sfieldsplit_%s_",((PetscObject)pc)->prefix ? ((PetscObject)pc)->prefix : "",ilink->splitname)); 18875f80ce2aSJacob Faibussowitsch CHKERRQ(KSPSetOptionsPrefix(ilink->ksp,prefix)); 1888704ba839SBarry Smith 1889704ba839SBarry Smith if (!next) { 1890704ba839SBarry Smith jac->head = ilink; 18910298fd71SBarry Smith ilink->previous = NULL; 1892704ba839SBarry Smith } else { 1893704ba839SBarry Smith while (next->next) { 1894704ba839SBarry Smith next = next->next; 1895704ba839SBarry Smith } 1896704ba839SBarry Smith next->next = ilink; 1897704ba839SBarry Smith ilink->previous = next; 1898704ba839SBarry Smith } 1899704ba839SBarry Smith jac->nsplits++; 1900704ba839SBarry Smith PetscFunctionReturn(0); 1901704ba839SBarry Smith } 1902704ba839SBarry Smith 190394ef8ddeSSatish Balay /*@C 19040971522cSBarry Smith PCFieldSplitSetFields - Sets the fields for one particular split in the field split preconditioner 19050971522cSBarry Smith 1906ad4df100SBarry Smith Logically Collective on PC 19070971522cSBarry Smith 19080971522cSBarry Smith Input Parameters: 19090971522cSBarry Smith + pc - the preconditioner context 19100298fd71SBarry Smith . splitname - name of this split, if NULL the number of the split is used 19110971522cSBarry Smith . n - the number of fields in this split 1912db4c96c1SJed Brown - fields - the fields in this split 19130971522cSBarry Smith 19140971522cSBarry Smith Level: intermediate 19150971522cSBarry Smith 191695452b02SPatrick Sanan Notes: 191795452b02SPatrick Sanan Use PCFieldSplitSetIS() to set a completely general set of indices as a field. 1918d32f9abdSBarry Smith 19197287d2a3SDmitry Karpeev The PCFieldSplitSetFields() is for defining fields as strided blocks. For example, if the block 1920d32f9abdSBarry Smith size is three then one can define a field as 0, or 1 or 2 or 0,1 or 0,2 or 1,2 which mean 1921d32f9abdSBarry Smith 0xx3xx6xx9xx12 ... x1xx4xx7xx ... xx2xx5xx8xx.. 01x34x67x... 0x1x3x5x7.. x12x45x78x.... 1922d32f9abdSBarry Smith where the numbered entries indicate what is in the field. 1923d32f9abdSBarry Smith 1924db4c96c1SJed Brown This function is called once per split (it creates a new split each time). Solve options 1925db4c96c1SJed Brown for this split will be available under the prefix -fieldsplit_SPLITNAME_. 1926db4c96c1SJed Brown 19275d4c12cdSJungho Lee Developer Note: This routine does not actually create the IS representing the split, that is delayed 19285d4c12cdSJungho Lee until PCSetUp_FieldSplit(), because information about the vector/matrix layouts may not be 19295d4c12cdSJungho Lee available when this routine is called. 19305d4c12cdSJungho Lee 1931d32f9abdSBarry Smith .seealso: PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitSetBlockSize(), PCFieldSplitSetIS() 19320971522cSBarry Smith 19330971522cSBarry Smith @*/ 19345d4c12cdSJungho Lee PetscErrorCode PCFieldSplitSetFields(PC pc,const char splitname[],PetscInt n,const PetscInt *fields,const PetscInt *fields_col) 19350971522cSBarry Smith { 19360971522cSBarry Smith PetscFunctionBegin; 19370700a824SBarry Smith PetscValidHeaderSpecific(pc,PC_CLASSID,1); 1938db4c96c1SJed Brown PetscValidCharPointer(splitname,2); 19392c71b3e2SJacob Faibussowitsch PetscCheckFalse(n < 1,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Provided number of fields %D in split \"%s\" not positive",n,splitname); 1940064a246eSJacob Faibussowitsch PetscValidIntPointer(fields,4); 19415f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetFields_C",(PC,const char[],PetscInt,const PetscInt*,const PetscInt*),(pc,splitname,n,fields,fields_col))); 19420971522cSBarry Smith PetscFunctionReturn(0); 19430971522cSBarry Smith } 19440971522cSBarry Smith 1945c84da90fSDmitry Karpeev /*@ 1946c84da90fSDmitry Karpeev PCFieldSplitSetDiagUseAmat - set flag indicating whether to extract diagonal blocks from Amat (rather than Pmat) 1947c84da90fSDmitry Karpeev 1948c84da90fSDmitry Karpeev Logically Collective on PC 1949c84da90fSDmitry Karpeev 1950c84da90fSDmitry Karpeev Input Parameters: 1951c84da90fSDmitry Karpeev + pc - the preconditioner object 1952c84da90fSDmitry Karpeev - flg - boolean flag indicating whether or not to use Amat to extract the diagonal blocks from 1953c84da90fSDmitry Karpeev 19549506b623SDmitry Karpeev Options Database: 1955147403d9SBarry Smith . -pc_fieldsplit_diag_use_amat - use the Amat to provide the diagonal blocks 1956c84da90fSDmitry Karpeev 1957c84da90fSDmitry Karpeev Level: intermediate 1958c84da90fSDmitry Karpeev 1959c84da90fSDmitry Karpeev .seealso: PCFieldSplitGetDiagUseAmat(), PCFieldSplitSetOffDiagUseAmat(), PCFIELDSPLIT 1960c84da90fSDmitry Karpeev 1961c84da90fSDmitry Karpeev @*/ 1962c84da90fSDmitry Karpeev PetscErrorCode PCFieldSplitSetDiagUseAmat(PC pc,PetscBool flg) 1963c84da90fSDmitry Karpeev { 1964c84da90fSDmitry Karpeev PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 1965c84da90fSDmitry Karpeev PetscBool isfs; 1966c84da90fSDmitry Karpeev 1967c84da90fSDmitry Karpeev PetscFunctionBegin; 1968c84da90fSDmitry Karpeev PetscValidHeaderSpecific(pc,PC_CLASSID,1); 19695f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)pc,PCFIELDSPLIT,&isfs)); 1970*28b400f6SJacob Faibussowitsch PetscCheck(isfs,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"PC not of type %s",PCFIELDSPLIT); 1971c84da90fSDmitry Karpeev jac->diag_use_amat = flg; 1972c84da90fSDmitry Karpeev PetscFunctionReturn(0); 1973c84da90fSDmitry Karpeev } 1974c84da90fSDmitry Karpeev 1975c84da90fSDmitry Karpeev /*@ 1976c84da90fSDmitry Karpeev PCFieldSplitGetDiagUseAmat - get the flag indicating whether to extract diagonal blocks from Amat (rather than Pmat) 1977c84da90fSDmitry Karpeev 1978c84da90fSDmitry Karpeev Logically Collective on PC 1979c84da90fSDmitry Karpeev 1980c84da90fSDmitry Karpeev Input Parameters: 1981c84da90fSDmitry Karpeev . pc - the preconditioner object 1982c84da90fSDmitry Karpeev 1983c84da90fSDmitry Karpeev Output Parameters: 1984c84da90fSDmitry Karpeev . flg - boolean flag indicating whether or not to use Amat to extract the diagonal blocks from 1985c84da90fSDmitry Karpeev 1986c84da90fSDmitry Karpeev Level: intermediate 1987c84da90fSDmitry Karpeev 1988c84da90fSDmitry Karpeev .seealso: PCFieldSplitSetDiagUseAmat(), PCFieldSplitGetOffDiagUseAmat(), PCFIELDSPLIT 1989c84da90fSDmitry Karpeev 1990c84da90fSDmitry Karpeev @*/ 1991c84da90fSDmitry Karpeev PetscErrorCode PCFieldSplitGetDiagUseAmat(PC pc,PetscBool *flg) 1992c84da90fSDmitry Karpeev { 1993c84da90fSDmitry Karpeev PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 1994c84da90fSDmitry Karpeev PetscBool isfs; 1995c84da90fSDmitry Karpeev 1996c84da90fSDmitry Karpeev PetscFunctionBegin; 1997c84da90fSDmitry Karpeev PetscValidHeaderSpecific(pc,PC_CLASSID,1); 1998534a8f05SLisandro Dalcin PetscValidBoolPointer(flg,2); 19995f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)pc,PCFIELDSPLIT,&isfs)); 2000*28b400f6SJacob Faibussowitsch PetscCheck(isfs,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"PC not of type %s",PCFIELDSPLIT); 2001c84da90fSDmitry Karpeev *flg = jac->diag_use_amat; 2002c84da90fSDmitry Karpeev PetscFunctionReturn(0); 2003c84da90fSDmitry Karpeev } 2004c84da90fSDmitry Karpeev 2005c84da90fSDmitry Karpeev /*@ 2006c84da90fSDmitry Karpeev PCFieldSplitSetOffDiagUseAmat - set flag indicating whether to extract off-diagonal blocks from Amat (rather than Pmat) 2007c84da90fSDmitry Karpeev 2008c84da90fSDmitry Karpeev Logically Collective on PC 2009c84da90fSDmitry Karpeev 2010c84da90fSDmitry Karpeev Input Parameters: 2011c84da90fSDmitry Karpeev + pc - the preconditioner object 2012c84da90fSDmitry Karpeev - flg - boolean flag indicating whether or not to use Amat to extract the off-diagonal blocks from 2013c84da90fSDmitry Karpeev 20149506b623SDmitry Karpeev Options Database: 2015147403d9SBarry Smith . -pc_fieldsplit_off_diag_use_amat <bool> - use the Amat to extract the off-diagonal blocks 2016c84da90fSDmitry Karpeev 2017c84da90fSDmitry Karpeev Level: intermediate 2018c84da90fSDmitry Karpeev 2019c84da90fSDmitry Karpeev .seealso: PCFieldSplitGetOffDiagUseAmat(), PCFieldSplitSetDiagUseAmat(), PCFIELDSPLIT 2020c84da90fSDmitry Karpeev 2021c84da90fSDmitry Karpeev @*/ 2022c84da90fSDmitry Karpeev PetscErrorCode PCFieldSplitSetOffDiagUseAmat(PC pc,PetscBool flg) 2023c84da90fSDmitry Karpeev { 2024c84da90fSDmitry Karpeev PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 2025c84da90fSDmitry Karpeev PetscBool isfs; 2026c84da90fSDmitry Karpeev 2027c84da90fSDmitry Karpeev PetscFunctionBegin; 2028c84da90fSDmitry Karpeev PetscValidHeaderSpecific(pc,PC_CLASSID,1); 20295f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)pc,PCFIELDSPLIT,&isfs)); 2030*28b400f6SJacob Faibussowitsch PetscCheck(isfs,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"PC not of type %s",PCFIELDSPLIT); 2031c84da90fSDmitry Karpeev jac->offdiag_use_amat = flg; 2032c84da90fSDmitry Karpeev PetscFunctionReturn(0); 2033c84da90fSDmitry Karpeev } 2034c84da90fSDmitry Karpeev 2035c84da90fSDmitry Karpeev /*@ 2036c84da90fSDmitry Karpeev PCFieldSplitGetOffDiagUseAmat - get the flag indicating whether to extract off-diagonal blocks from Amat (rather than Pmat) 2037c84da90fSDmitry Karpeev 2038c84da90fSDmitry Karpeev Logically Collective on PC 2039c84da90fSDmitry Karpeev 2040c84da90fSDmitry Karpeev Input Parameters: 2041c84da90fSDmitry Karpeev . pc - the preconditioner object 2042c84da90fSDmitry Karpeev 2043c84da90fSDmitry Karpeev Output Parameters: 2044c84da90fSDmitry Karpeev . flg - boolean flag indicating whether or not to use Amat to extract the off-diagonal blocks from 2045c84da90fSDmitry Karpeev 2046c84da90fSDmitry Karpeev Level: intermediate 2047c84da90fSDmitry Karpeev 2048c84da90fSDmitry Karpeev .seealso: PCFieldSplitSetOffDiagUseAmat(), PCFieldSplitGetDiagUseAmat(), PCFIELDSPLIT 2049c84da90fSDmitry Karpeev 2050c84da90fSDmitry Karpeev @*/ 2051c84da90fSDmitry Karpeev PetscErrorCode PCFieldSplitGetOffDiagUseAmat(PC pc,PetscBool *flg) 2052c84da90fSDmitry Karpeev { 2053c84da90fSDmitry Karpeev PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 2054c84da90fSDmitry Karpeev PetscBool isfs; 2055c84da90fSDmitry Karpeev 2056c84da90fSDmitry Karpeev PetscFunctionBegin; 2057c84da90fSDmitry Karpeev PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2058534a8f05SLisandro Dalcin PetscValidBoolPointer(flg,2); 20595f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)pc,PCFIELDSPLIT,&isfs)); 2060*28b400f6SJacob Faibussowitsch PetscCheck(isfs,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"PC not of type %s",PCFIELDSPLIT); 2061c84da90fSDmitry Karpeev *flg = jac->offdiag_use_amat; 2062c84da90fSDmitry Karpeev PetscFunctionReturn(0); 2063c84da90fSDmitry Karpeev } 2064c84da90fSDmitry Karpeev 2065218d4943SBarry Smith /*@C 2066704ba839SBarry Smith PCFieldSplitSetIS - Sets the exact elements for field 2067704ba839SBarry Smith 2068ad4df100SBarry Smith Logically Collective on PC 2069704ba839SBarry Smith 2070704ba839SBarry Smith Input Parameters: 2071704ba839SBarry Smith + pc - the preconditioner context 20720298fd71SBarry Smith . splitname - name of this split, if NULL the number of the split is used 2073db4c96c1SJed Brown - is - the index set that defines the vector elements in this field 2074704ba839SBarry Smith 2075a6ffb8dbSJed Brown Notes: 2076a6ffb8dbSJed Brown Use PCFieldSplitSetFields(), for fields defined by strided types. 2077a6ffb8dbSJed Brown 2078db4c96c1SJed Brown This function is called once per split (it creates a new split each time). Solve options 2079db4c96c1SJed Brown for this split will be available under the prefix -fieldsplit_SPLITNAME_. 2080d32f9abdSBarry Smith 2081704ba839SBarry Smith Level: intermediate 2082704ba839SBarry Smith 2083704ba839SBarry Smith .seealso: PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitSetBlockSize() 2084704ba839SBarry Smith 2085704ba839SBarry Smith @*/ 20867087cfbeSBarry Smith PetscErrorCode PCFieldSplitSetIS(PC pc,const char splitname[],IS is) 2087704ba839SBarry Smith { 2088704ba839SBarry Smith PetscFunctionBegin; 20890700a824SBarry Smith PetscValidHeaderSpecific(pc,PC_CLASSID,1); 20907b62db95SJungho Lee if (splitname) PetscValidCharPointer(splitname,2); 2091db4c96c1SJed Brown PetscValidHeaderSpecific(is,IS_CLASSID,3); 20925f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetIS_C",(PC,const char[],IS),(pc,splitname,is))); 2093704ba839SBarry Smith PetscFunctionReturn(0); 2094704ba839SBarry Smith } 2095704ba839SBarry Smith 209694ef8ddeSSatish Balay /*@C 209757a9adfeSMatthew G Knepley PCFieldSplitGetIS - Retrieves the elements for a field as an IS 209857a9adfeSMatthew G Knepley 209957a9adfeSMatthew G Knepley Logically Collective on PC 210057a9adfeSMatthew G Knepley 210157a9adfeSMatthew G Knepley Input Parameters: 210257a9adfeSMatthew G Knepley + pc - the preconditioner context 210357a9adfeSMatthew G Knepley - splitname - name of this split 210457a9adfeSMatthew G Knepley 210557a9adfeSMatthew G Knepley Output Parameter: 21060298fd71SBarry Smith - is - the index set that defines the vector elements in this field, or NULL if the field is not found 210757a9adfeSMatthew G Knepley 210857a9adfeSMatthew G Knepley Level: intermediate 210957a9adfeSMatthew G Knepley 211057a9adfeSMatthew G Knepley .seealso: PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitSetIS() 211157a9adfeSMatthew G Knepley 211257a9adfeSMatthew G Knepley @*/ 211357a9adfeSMatthew G Knepley PetscErrorCode PCFieldSplitGetIS(PC pc,const char splitname[],IS *is) 211457a9adfeSMatthew G Knepley { 211557a9adfeSMatthew G Knepley PetscFunctionBegin; 211657a9adfeSMatthew G Knepley PetscValidHeaderSpecific(pc,PC_CLASSID,1); 211757a9adfeSMatthew G Knepley PetscValidCharPointer(splitname,2); 211857a9adfeSMatthew G Knepley PetscValidPointer(is,3); 211957a9adfeSMatthew G Knepley { 212057a9adfeSMatthew G Knepley PC_FieldSplit *jac = (PC_FieldSplit*) pc->data; 212157a9adfeSMatthew G Knepley PC_FieldSplitLink ilink = jac->head; 212257a9adfeSMatthew G Knepley PetscBool found; 212357a9adfeSMatthew G Knepley 21240298fd71SBarry Smith *is = NULL; 212557a9adfeSMatthew G Knepley while (ilink) { 21265f80ce2aSJacob Faibussowitsch CHKERRQ(PetscStrcmp(ilink->splitname, splitname, &found)); 212757a9adfeSMatthew G Knepley if (found) { 212857a9adfeSMatthew G Knepley *is = ilink->is; 212957a9adfeSMatthew G Knepley break; 213057a9adfeSMatthew G Knepley } 213157a9adfeSMatthew G Knepley ilink = ilink->next; 213257a9adfeSMatthew G Knepley } 213357a9adfeSMatthew G Knepley } 213457a9adfeSMatthew G Knepley PetscFunctionReturn(0); 213557a9adfeSMatthew G Knepley } 213657a9adfeSMatthew G Knepley 2137998f007dSPierre Jolivet /*@C 2138998f007dSPierre Jolivet PCFieldSplitGetISByIndex - Retrieves the elements for a given index field as an IS 2139998f007dSPierre Jolivet 2140998f007dSPierre Jolivet Logically Collective on PC 2141998f007dSPierre Jolivet 2142998f007dSPierre Jolivet Input Parameters: 2143998f007dSPierre Jolivet + pc - the preconditioner context 2144998f007dSPierre Jolivet - index - index of this split 2145998f007dSPierre Jolivet 2146998f007dSPierre Jolivet Output Parameter: 2147998f007dSPierre Jolivet - is - the index set that defines the vector elements in this field 2148998f007dSPierre Jolivet 2149998f007dSPierre Jolivet Level: intermediate 2150998f007dSPierre Jolivet 2151998f007dSPierre Jolivet .seealso: PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitGetIS(), PCFieldSplitSetIS() 2152998f007dSPierre Jolivet 2153998f007dSPierre Jolivet @*/ 2154998f007dSPierre Jolivet PetscErrorCode PCFieldSplitGetISByIndex(PC pc,PetscInt index,IS *is) 2155998f007dSPierre Jolivet { 2156998f007dSPierre Jolivet PetscFunctionBegin; 21572c71b3e2SJacob Faibussowitsch PetscCheckFalse(index < 0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative field %D requested",index); 2158998f007dSPierre Jolivet PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2159998f007dSPierre Jolivet PetscValidPointer(is,3); 2160998f007dSPierre Jolivet { 2161998f007dSPierre Jolivet PC_FieldSplit *jac = (PC_FieldSplit*) pc->data; 2162998f007dSPierre Jolivet PC_FieldSplitLink ilink = jac->head; 2163998f007dSPierre Jolivet PetscInt i = 0; 21642c71b3e2SJacob Faibussowitsch PetscCheckFalse(index >= jac->nsplits,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Field %D requested but only %D exist",index,jac->nsplits); 2165998f007dSPierre Jolivet 2166998f007dSPierre Jolivet while (i < index) { 2167998f007dSPierre Jolivet ilink = ilink->next; 2168998f007dSPierre Jolivet ++i; 2169998f007dSPierre Jolivet } 21705f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitGetIS(pc,ilink->splitname,is)); 2171998f007dSPierre Jolivet } 2172998f007dSPierre Jolivet PetscFunctionReturn(0); 2173998f007dSPierre Jolivet } 2174998f007dSPierre Jolivet 217551f519a2SBarry Smith /*@ 217651f519a2SBarry Smith PCFieldSplitSetBlockSize - Sets the block size for defining where fields start in the 217751f519a2SBarry Smith fieldsplit preconditioner. If not set the matrix block size is used. 217851f519a2SBarry Smith 2179ad4df100SBarry Smith Logically Collective on PC 218051f519a2SBarry Smith 218151f519a2SBarry Smith Input Parameters: 218251f519a2SBarry Smith + pc - the preconditioner context 218351f519a2SBarry Smith - bs - the block size 218451f519a2SBarry Smith 218551f519a2SBarry Smith Level: intermediate 218651f519a2SBarry Smith 218751f519a2SBarry Smith .seealso: PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitSetFields() 218851f519a2SBarry Smith 218951f519a2SBarry Smith @*/ 21907087cfbeSBarry Smith PetscErrorCode PCFieldSplitSetBlockSize(PC pc,PetscInt bs) 219151f519a2SBarry Smith { 219251f519a2SBarry Smith PetscFunctionBegin; 21930700a824SBarry Smith PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2194c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(pc,bs,2); 21955f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetBlockSize_C",(PC,PetscInt),(pc,bs))); 219651f519a2SBarry Smith PetscFunctionReturn(0); 219751f519a2SBarry Smith } 219851f519a2SBarry Smith 21990971522cSBarry Smith /*@C 220069a612a9SBarry Smith PCFieldSplitGetSubKSP - Gets the KSP contexts for all splits 22010971522cSBarry Smith 220269a612a9SBarry Smith Collective on KSP 22030971522cSBarry Smith 22040971522cSBarry Smith Input Parameter: 22050971522cSBarry Smith . pc - the preconditioner context 22060971522cSBarry Smith 22070971522cSBarry Smith Output Parameters: 220813e0d083SBarry Smith + n - the number of splits 22093111de3cSBarry Smith - subksp - the array of KSP contexts 22100971522cSBarry Smith 22110971522cSBarry Smith Note: 22129a4f7e47SBarry Smith After PCFieldSplitGetSubKSP() the array of KSPs is to be freed by the user with PetscFree() 2213d32f9abdSBarry Smith (not the KSP just the array that contains them). 22140971522cSBarry Smith 2215285fb4e2SStefano Zampini You must call PCSetUp() before calling PCFieldSplitGetSubKSP(). 2216285fb4e2SStefano Zampini 2217285fb4e2SStefano Zampini If the fieldsplit is of type PC_COMPOSITE_SCHUR, it returns the KSP object used inside the 2218285fb4e2SStefano Zampini Schur complement and the KSP object used to iterate over the Schur complement. 2219a51937d4SCarola Kruse To access all the KSP objects used in PC_COMPOSITE_SCHUR, use PCFieldSplitSchurGetSubKSP(). 2220a51937d4SCarola Kruse 2221a51937d4SCarola Kruse If the fieldsplit is of type PC_COMPOSITE_GKB, it returns the KSP object used to solve the 2222a51937d4SCarola Kruse inner linear system defined by the matrix H in each loop. 22230971522cSBarry Smith 2224196cc216SBarry Smith Fortran Usage: You must pass in a KSP array that is large enough to contain all the local KSPs. 22252bf68e3eSBarry Smith You can call PCFieldSplitGetSubKSP(pc,n,PETSC_NULL_KSP,ierr) to determine how large the 2226196cc216SBarry Smith KSP array must be. 2227196cc216SBarry Smith 22280971522cSBarry Smith Level: advanced 22290971522cSBarry Smith 22300971522cSBarry Smith .seealso: PCFIELDSPLIT 22310971522cSBarry Smith @*/ 22327087cfbeSBarry Smith PetscErrorCode PCFieldSplitGetSubKSP(PC pc,PetscInt *n,KSP *subksp[]) 22330971522cSBarry Smith { 22340971522cSBarry Smith PetscFunctionBegin; 22350700a824SBarry Smith PetscValidHeaderSpecific(pc,PC_CLASSID,1); 223613e0d083SBarry Smith if (n) PetscValidIntPointer(n,2); 22375f80ce2aSJacob Faibussowitsch CHKERRQ(PetscUseMethod(pc,"PCFieldSplitGetSubKSP_C",(PC,PetscInt*,KSP **),(pc,n,subksp))); 22380971522cSBarry Smith PetscFunctionReturn(0); 22390971522cSBarry Smith } 22400971522cSBarry Smith 2241285fb4e2SStefano Zampini /*@C 2242285fb4e2SStefano Zampini PCFieldSplitSchurGetSubKSP - Gets the KSP contexts used inside the Schur complement based PCFIELDSPLIT 2243285fb4e2SStefano Zampini 2244285fb4e2SStefano Zampini Collective on KSP 2245285fb4e2SStefano Zampini 2246285fb4e2SStefano Zampini Input Parameter: 2247285fb4e2SStefano Zampini . pc - the preconditioner context 2248285fb4e2SStefano Zampini 2249285fb4e2SStefano Zampini Output Parameters: 2250285fb4e2SStefano Zampini + n - the number of splits 2251285fb4e2SStefano Zampini - subksp - the array of KSP contexts 2252285fb4e2SStefano Zampini 2253285fb4e2SStefano Zampini Note: 2254285fb4e2SStefano Zampini After PCFieldSplitSchurGetSubKSP() the array of KSPs is to be freed by the user with PetscFree() 2255285fb4e2SStefano Zampini (not the KSP just the array that contains them). 2256285fb4e2SStefano Zampini 2257285fb4e2SStefano Zampini You must call PCSetUp() before calling PCFieldSplitSchurGetSubKSP(). 2258285fb4e2SStefano Zampini 2259285fb4e2SStefano Zampini If the fieldsplit type is of type PC_COMPOSITE_SCHUR, it returns (in order) 2260285fb4e2SStefano Zampini - the KSP used for the (1,1) block 2261285fb4e2SStefano Zampini - the KSP used for the Schur complement (not the one used for the interior Schur solver) 2262285fb4e2SStefano Zampini - the KSP used for the (1,1) block in the upper triangular factor (if different from that of the (1,1) block). 2263285fb4e2SStefano Zampini 2264285fb4e2SStefano Zampini It returns a null array if the fieldsplit is not of type PC_COMPOSITE_SCHUR; in this case, you should use PCFieldSplitGetSubKSP(). 2265285fb4e2SStefano Zampini 2266285fb4e2SStefano Zampini Fortran Usage: You must pass in a KSP array that is large enough to contain all the local KSPs. 2267285fb4e2SStefano Zampini You can call PCFieldSplitSchurGetSubKSP(pc,n,PETSC_NULL_KSP,ierr) to determine how large the 2268285fb4e2SStefano Zampini KSP array must be. 2269285fb4e2SStefano Zampini 2270285fb4e2SStefano Zampini Level: advanced 2271285fb4e2SStefano Zampini 2272285fb4e2SStefano Zampini .seealso: PCFIELDSPLIT 2273285fb4e2SStefano Zampini @*/ 2274285fb4e2SStefano Zampini PetscErrorCode PCFieldSplitSchurGetSubKSP(PC pc,PetscInt *n,KSP *subksp[]) 2275285fb4e2SStefano Zampini { 2276285fb4e2SStefano Zampini PetscFunctionBegin; 2277285fb4e2SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2278285fb4e2SStefano Zampini if (n) PetscValidIntPointer(n,2); 22795f80ce2aSJacob Faibussowitsch CHKERRQ(PetscUseMethod(pc,"PCFieldSplitSchurGetSubKSP_C",(PC,PetscInt*,KSP **),(pc,n,subksp))); 2280285fb4e2SStefano Zampini PetscFunctionReturn(0); 2281285fb4e2SStefano Zampini } 2282285fb4e2SStefano Zampini 2283e69d4d44SBarry Smith /*@ 2284a1e3cbf9SBarry Smith PCFieldSplitSetSchurPre - Indicates from what operator the preconditioner is constructucted for the Schur complement. 2285a1e3cbf9SBarry Smith The default is the A11 matrix. 2286e69d4d44SBarry Smith 2287e69d4d44SBarry Smith Collective on PC 2288e69d4d44SBarry Smith 2289e69d4d44SBarry Smith Input Parameters: 2290e69d4d44SBarry Smith + pc - the preconditioner context 22916fdd48a9SBarry Smith . ptype - which matrix to use for preconditioning the Schur complement: PC_FIELDSPLIT_SCHUR_PRE_A11 (default), PC_FIELDSPLIT_SCHUR_PRE_SELF, PC_FIELDSPLIT_SCHUR_PRE_USER 22926fdd48a9SBarry Smith PC_FIELDSPLIT_SCHUR_PRE_SELFP, and PC_FIELDSPLIT_SCHUR_PRE_FULL 22930298fd71SBarry Smith - userpre - matrix to use for preconditioning, or NULL 2294084e4875SJed Brown 2295e69d4d44SBarry Smith Options Database: 2296a1e3cbf9SBarry Smith + -pc_fieldsplit_schur_precondition <self,selfp,user,a11,full> - default is a11. See notes for meaning of various arguments 2297a1e3cbf9SBarry Smith - -fieldsplit_1_pc_type <pctype> - the preconditioner algorithm that is used to construct the preconditioner from the operator 2298e69d4d44SBarry Smith 2299fd1303e9SJungho Lee Notes: 2300fd1303e9SJungho Lee $ If ptype is 2301a1e3cbf9SBarry Smith $ a11 - the preconditioner for the Schur complement is generated from the block diagonal part of the preconditioner 230253f2277eSBarry Smith $ matrix associated with the Schur complement (i.e. A11), not the Schur complement matrix 2303a1e3cbf9SBarry Smith $ self - the preconditioner for the Schur complement is generated from the symbolic representation of the Schur complement matrix: 2304a6a584a2SBarry Smith $ The only preconditioner that currently works with this symbolic respresentation matrix object is the PCLSC 2305fd1303e9SJungho Lee $ preconditioner 2306a1e3cbf9SBarry Smith $ user - the preconditioner for the Schur complement is generated from the user provided matrix (pre argument 23076fdd48a9SBarry Smith $ to this function). 2308a1e3cbf9SBarry Smith $ selfp - the preconditioning for the Schur complement is generated from an explicitly-assembled approximation Sp = A11 - A10 inv(diag(A00)) A01 23091d71051fSDmitry Karpeev $ This is only a good preconditioner when diag(A00) is a good preconditioner for A00. Optionally, A00 can be 23106fdd48a9SBarry Smith $ lumped before extracting the diagonal using the additional option -fieldsplit_1_mat_schur_complement_ainv_type lump 2311a1e3cbf9SBarry Smith $ full - the preconditioner for the Schur complement is generated from the exact Schur complement matrix representation computed internally by PCFIELDSPLIT (this is expensive) 23126fdd48a9SBarry Smith $ useful mostly as a test that the Schur complement approach can work for your problem 2313fd1303e9SJungho Lee 2314e87fab1bSBarry Smith When solving a saddle point problem, where the A11 block is identically zero, using a11 as the ptype only makes sense 2315fd1303e9SJungho Lee with the additional option -fieldsplit_1_pc_type none. Usually for saddle point problems one would use a ptype of self and 2316a7476a74SDmitry Karpeev -fieldsplit_1_pc_type lsc which uses the least squares commutator to compute a preconditioner for the Schur complement. 2317fd1303e9SJungho Lee 2318e69d4d44SBarry Smith Level: intermediate 2319e69d4d44SBarry Smith 23201d71051fSDmitry Karpeev .seealso: PCFieldSplitGetSchurPre(), PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitSetFields(), PCFieldSplitSchurPreType, 23211d71051fSDmitry Karpeev MatSchurComplementSetAinvType(), PCLSC 2322e69d4d44SBarry Smith 2323e69d4d44SBarry Smith @*/ 232429f8a81cSJed Brown PetscErrorCode PCFieldSplitSetSchurPre(PC pc,PCFieldSplitSchurPreType ptype,Mat pre) 2325e69d4d44SBarry Smith { 2326e69d4d44SBarry Smith PetscFunctionBegin; 23270700a824SBarry Smith PetscValidHeaderSpecific(pc,PC_CLASSID,1); 23285f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetSchurPre_C",(PC,PCFieldSplitSchurPreType,Mat),(pc,ptype,pre))); 2329e69d4d44SBarry Smith PetscFunctionReturn(0); 2330e69d4d44SBarry Smith } 2331686bed4dSStefano Zampini 233229f8a81cSJed Brown PetscErrorCode PCFieldSplitSchurPrecondition(PC pc,PCFieldSplitSchurPreType ptype,Mat pre) {return PCFieldSplitSetSchurPre(pc,ptype,pre);} /* Deprecated name */ 2333e69d4d44SBarry Smith 233437a82bf0SJed Brown /*@ 233537a82bf0SJed Brown PCFieldSplitGetSchurPre - For Schur complement fieldsplit, determine how the Schur complement will be 233629f8a81cSJed Brown preconditioned. See PCFieldSplitSetSchurPre() for details. 233737a82bf0SJed Brown 233837a82bf0SJed Brown Logically Collective on PC 233937a82bf0SJed Brown 2340f899ff85SJose E. Roman Input Parameter: 234137a82bf0SJed Brown . pc - the preconditioner context 234237a82bf0SJed Brown 234337a82bf0SJed Brown Output Parameters: 234437a82bf0SJed Brown + ptype - which matrix to use for preconditioning the Schur complement: PC_FIELDSPLIT_SCHUR_PRE_A11, PC_FIELDSPLIT_SCHUR_PRE_SELF, PC_FIELDSPLIT_PRE_USER 234537a82bf0SJed Brown - userpre - matrix to use for preconditioning (with PC_FIELDSPLIT_PRE_USER), or NULL 234637a82bf0SJed Brown 234737a82bf0SJed Brown Level: intermediate 234837a82bf0SJed Brown 234929f8a81cSJed Brown .seealso: PCFieldSplitSetSchurPre(), PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitSetFields(), PCFieldSplitSchurPreType, PCLSC 235037a82bf0SJed Brown 235137a82bf0SJed Brown @*/ 235237a82bf0SJed Brown PetscErrorCode PCFieldSplitGetSchurPre(PC pc,PCFieldSplitSchurPreType *ptype,Mat *pre) 235337a82bf0SJed Brown { 235437a82bf0SJed Brown PetscFunctionBegin; 235537a82bf0SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 23565f80ce2aSJacob Faibussowitsch CHKERRQ(PetscUseMethod(pc,"PCFieldSplitGetSchurPre_C",(PC,PCFieldSplitSchurPreType*,Mat*),(pc,ptype,pre))); 2357e69d4d44SBarry Smith PetscFunctionReturn(0); 2358e69d4d44SBarry Smith } 2359e69d4d44SBarry Smith 236045e7fc46SDmitry Karpeev /*@ 2361470b340bSDmitry Karpeev PCFieldSplitSchurGetS - extract the MatSchurComplement object used by this PC in case it needs to be configured separately 236245e7fc46SDmitry Karpeev 236345e7fc46SDmitry Karpeev Not collective 236445e7fc46SDmitry Karpeev 236545e7fc46SDmitry Karpeev Input Parameter: 236645e7fc46SDmitry Karpeev . pc - the preconditioner context 236745e7fc46SDmitry Karpeev 236845e7fc46SDmitry Karpeev Output Parameter: 2369470b340bSDmitry Karpeev . S - the Schur complement matrix 237045e7fc46SDmitry Karpeev 2371470b340bSDmitry Karpeev Notes: 2372470b340bSDmitry Karpeev This matrix should not be destroyed using MatDestroy(); rather, use PCFieldSplitSchurRestoreS(). 237345e7fc46SDmitry Karpeev 237445e7fc46SDmitry Karpeev Level: advanced 237545e7fc46SDmitry Karpeev 237629f8a81cSJed Brown .seealso: PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitSchurPreType, PCFieldSplitSetSchurPre(), MatSchurComplement, PCFieldSplitSchurRestoreS() 237745e7fc46SDmitry Karpeev 237845e7fc46SDmitry Karpeev @*/ 2379470b340bSDmitry Karpeev PetscErrorCode PCFieldSplitSchurGetS(PC pc,Mat *S) 238045e7fc46SDmitry Karpeev { 238145e7fc46SDmitry Karpeev const char* t; 238245e7fc46SDmitry Karpeev PetscBool isfs; 238345e7fc46SDmitry Karpeev PC_FieldSplit *jac; 238445e7fc46SDmitry Karpeev 238545e7fc46SDmitry Karpeev PetscFunctionBegin; 238645e7fc46SDmitry Karpeev PetscValidHeaderSpecific(pc,PC_CLASSID,1); 23875f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectGetType((PetscObject)pc,&t)); 23885f80ce2aSJacob Faibussowitsch CHKERRQ(PetscStrcmp(t,PCFIELDSPLIT,&isfs)); 2389*28b400f6SJacob Faibussowitsch PetscCheck(isfs,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Expected PC of type PCFIELDSPLIT, got %s instead",t); 239045e7fc46SDmitry Karpeev jac = (PC_FieldSplit*)pc->data; 23912c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->type != PC_COMPOSITE_SCHUR,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Expected PCFIELDSPLIT of type SCHUR, got %D instead",jac->type); 2392470b340bSDmitry Karpeev if (S) *S = jac->schur; 239345e7fc46SDmitry Karpeev PetscFunctionReturn(0); 239445e7fc46SDmitry Karpeev } 239545e7fc46SDmitry Karpeev 2396470b340bSDmitry Karpeev /*@ 2397470b340bSDmitry Karpeev PCFieldSplitSchurRestoreS - restores the MatSchurComplement object used by this PC 2398470b340bSDmitry Karpeev 2399470b340bSDmitry Karpeev Not collective 2400470b340bSDmitry Karpeev 2401470b340bSDmitry Karpeev Input Parameters: 2402470b340bSDmitry Karpeev + pc - the preconditioner context 2403a2b725a8SWilliam Gropp - S - the Schur complement matrix 2404470b340bSDmitry Karpeev 2405470b340bSDmitry Karpeev Level: advanced 2406470b340bSDmitry Karpeev 240729f8a81cSJed Brown .seealso: PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitSchurPreType, PCFieldSplitSetSchurPre(), MatSchurComplement, PCFieldSplitSchurGetS() 2408470b340bSDmitry Karpeev 2409470b340bSDmitry Karpeev @*/ 2410bca69d2bSDmitry Karpeev PetscErrorCode PCFieldSplitSchurRestoreS(PC pc,Mat *S) 2411470b340bSDmitry Karpeev { 2412470b340bSDmitry Karpeev const char* t; 2413470b340bSDmitry Karpeev PetscBool isfs; 2414470b340bSDmitry Karpeev PC_FieldSplit *jac; 2415470b340bSDmitry Karpeev 2416470b340bSDmitry Karpeev PetscFunctionBegin; 2417470b340bSDmitry Karpeev PetscValidHeaderSpecific(pc,PC_CLASSID,1); 24185f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectGetType((PetscObject)pc,&t)); 24195f80ce2aSJacob Faibussowitsch CHKERRQ(PetscStrcmp(t,PCFIELDSPLIT,&isfs)); 2420*28b400f6SJacob Faibussowitsch PetscCheck(isfs,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Expected PC of type PCFIELDSPLIT, got %s instead",t); 2421470b340bSDmitry Karpeev jac = (PC_FieldSplit*)pc->data; 24222c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->type != PC_COMPOSITE_SCHUR,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Expected PCFIELDSPLIT of type SCHUR, got %D instead",jac->type); 24232c71b3e2SJacob Faibussowitsch PetscCheckFalse(!S || *S != jac->schur,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"MatSchurComplement restored is not the same as gotten"); 2424470b340bSDmitry Karpeev PetscFunctionReturn(0); 2425470b340bSDmitry Karpeev } 2426470b340bSDmitry Karpeev 242729f8a81cSJed Brown static PetscErrorCode PCFieldSplitSetSchurPre_FieldSplit(PC pc,PCFieldSplitSchurPreType ptype,Mat pre) 2428e69d4d44SBarry Smith { 2429e69d4d44SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 2430e69d4d44SBarry Smith 2431e69d4d44SBarry Smith PetscFunctionBegin; 2432084e4875SJed Brown jac->schurpre = ptype; 2433a7476a74SDmitry Karpeev if (ptype == PC_FIELDSPLIT_SCHUR_PRE_USER && pre) { 24345f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&jac->schur_user)); 2435084e4875SJed Brown jac->schur_user = pre; 24365f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectReference((PetscObject)jac->schur_user)); 2437084e4875SJed Brown } 2438e69d4d44SBarry Smith PetscFunctionReturn(0); 2439e69d4d44SBarry Smith } 2440e69d4d44SBarry Smith 244137a82bf0SJed Brown static PetscErrorCode PCFieldSplitGetSchurPre_FieldSplit(PC pc,PCFieldSplitSchurPreType *ptype,Mat *pre) 244237a82bf0SJed Brown { 244337a82bf0SJed Brown PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 244437a82bf0SJed Brown 244537a82bf0SJed Brown PetscFunctionBegin; 244637a82bf0SJed Brown *ptype = jac->schurpre; 244737a82bf0SJed Brown *pre = jac->schur_user; 244837a82bf0SJed Brown PetscFunctionReturn(0); 244937a82bf0SJed Brown } 245037a82bf0SJed Brown 2451ab1df9f5SJed Brown /*@ 24520ffb0e17SBarry Smith PCFieldSplitSetSchurFactType - sets which blocks of the approximate block factorization to retain in the preconditioner 2453ab1df9f5SJed Brown 2454ab1df9f5SJed Brown Collective on PC 2455ab1df9f5SJed Brown 2456ab1df9f5SJed Brown Input Parameters: 2457ab1df9f5SJed Brown + pc - the preconditioner context 2458c9c6ffaaSJed Brown - ftype - which blocks of factorization to retain, PC_FIELDSPLIT_SCHUR_FACT_FULL is default 2459ab1df9f5SJed Brown 2460ab1df9f5SJed Brown Options Database: 2461147403d9SBarry Smith . -pc_fieldsplit_schur_fact_type <diag,lower,upper,full> - default is full 2462ab1df9f5SJed Brown 2463ab1df9f5SJed Brown Level: intermediate 2464ab1df9f5SJed Brown 2465ab1df9f5SJed Brown Notes: 2466ab1df9f5SJed Brown The FULL factorization is 2467ab1df9f5SJed Brown 2468147403d9SBarry Smith .vb 2469147403d9SBarry Smith (A B) = (1 0) (A 0) (1 Ainv*B) = L D U 2470147403d9SBarry Smith (C E) (C*Ainv 1) (0 S) (0 1) 2471147403d9SBarry Smith .vb 24720ffb0e17SBarry Smith where S = E - C*Ainv*B. In practice, the full factorization is applied via block triangular solves with the grouping L*(D*U). UPPER uses D*U, LOWER uses L*D, 2473c096484dSStefano Zampini and DIAG is the diagonal part with the sign of S flipped (because this makes the preconditioner positive definite for many formulations, thus allowing the use of KSPMINRES). Sign flipping of S can be turned off with PCFieldSplitSetSchurScale(). 2474ab1df9f5SJed Brown 2475147403d9SBarry Smith If A and S are solved exactly 2476147403d9SBarry Smith .vb 2477147403d9SBarry Smith *) FULL factorization is a direct solver. 2478147403d9SBarry Smith *) The preconditioned operator with LOWER or UPPER has all eigenvalues equal to 1 and minimal polynomial of degree 2, so KSPGMRES converges in 2 iterations. 2479147403d9SBarry Smith *) With DIAG, the preconditioned operator has three distinct nonzero eigenvalues and minimal polynomial of degree at most 4, so KSPGMRES converges in at most 4 iterations. 2480147403d9SBarry Smith .ve 2481ab1df9f5SJed Brown 24820ffb0e17SBarry Smith If the iteration count is very low, consider using KSPFGMRES or KSPGCR which can use one less preconditioner 24830ffb0e17SBarry Smith application in this case. Note that the preconditioned operator may be highly non-normal, so such fast convergence may not be observed in practice. 24840ffb0e17SBarry Smith 24850ffb0e17SBarry Smith For symmetric problems in which A is positive definite and S is negative definite, DIAG can be used with KSPMINRES. 24860ffb0e17SBarry Smith 24870ffb0e17SBarry Smith Note that a flexible method like KSPFGMRES or KSPGCR must be used if the fieldsplit preconditioner is nonlinear (e.g. a few iterations of a Krylov method is used to solve with A or S). 2488ab1df9f5SJed Brown 2489ab1df9f5SJed Brown References: 2490606c0280SSatish Balay + * - Murphy, Golub, and Wathen, A note on preconditioning indefinite linear systems, SIAM J. Sci. Comput., 21 (2000). 2491606c0280SSatish Balay - * - Ipsen, A note on preconditioning nonsymmetric matrices, SIAM J. Sci. Comput., 23 (2001). 2492ab1df9f5SJed Brown 2493c096484dSStefano Zampini .seealso: PCFieldSplitGetSubKSP(), PCFIELDSPLIT, PCFieldSplitSetFields(), PCFieldSplitSchurPreType, PCFieldSplitSetSchurScale() 2494ab1df9f5SJed Brown @*/ 2495c9c6ffaaSJed Brown PetscErrorCode PCFieldSplitSetSchurFactType(PC pc,PCFieldSplitSchurFactType ftype) 2496ab1df9f5SJed Brown { 2497ab1df9f5SJed Brown PetscFunctionBegin; 2498ab1df9f5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 24995f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetSchurFactType_C",(PC,PCFieldSplitSchurFactType),(pc,ftype))); 2500ab1df9f5SJed Brown PetscFunctionReturn(0); 2501ab1df9f5SJed Brown } 2502ab1df9f5SJed Brown 25031e6b0712SBarry Smith static PetscErrorCode PCFieldSplitSetSchurFactType_FieldSplit(PC pc,PCFieldSplitSchurFactType ftype) 2504ab1df9f5SJed Brown { 2505ab1df9f5SJed Brown PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 2506ab1df9f5SJed Brown 2507ab1df9f5SJed Brown PetscFunctionBegin; 2508ab1df9f5SJed Brown jac->schurfactorization = ftype; 2509ab1df9f5SJed Brown PetscFunctionReturn(0); 2510ab1df9f5SJed Brown } 2511ab1df9f5SJed Brown 2512c096484dSStefano Zampini /*@ 2513c096484dSStefano Zampini PCFieldSplitSetSchurScale - Controls the sign flip of S for PC_FIELDSPLIT_SCHUR_FACT_DIAG. 2514c096484dSStefano Zampini 2515c096484dSStefano Zampini Collective on PC 2516c096484dSStefano Zampini 2517c096484dSStefano Zampini Input Parameters: 2518c096484dSStefano Zampini + pc - the preconditioner context 2519c096484dSStefano Zampini - scale - scaling factor for the Schur complement 2520c096484dSStefano Zampini 2521c096484dSStefano Zampini Options Database: 2522c096484dSStefano Zampini . -pc_fieldsplit_schur_scale - default is -1.0 2523c096484dSStefano Zampini 2524c096484dSStefano Zampini Level: intermediate 2525c096484dSStefano Zampini 2526c096484dSStefano Zampini .seealso: PCFIELDSPLIT, PCFieldSplitSetFields(), PCFieldSplitSchurFactType, PCFieldSplitSetSchurScale() 2527c096484dSStefano Zampini @*/ 2528c096484dSStefano Zampini PetscErrorCode PCFieldSplitSetSchurScale(PC pc,PetscScalar scale) 2529c096484dSStefano Zampini { 2530c096484dSStefano Zampini PetscFunctionBegin; 2531c096484dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2532c096484dSStefano Zampini PetscValidLogicalCollectiveScalar(pc,scale,2); 25335f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetSchurScale_C",(PC,PetscScalar),(pc,scale))); 2534c096484dSStefano Zampini PetscFunctionReturn(0); 2535c096484dSStefano Zampini } 2536c096484dSStefano Zampini 2537c096484dSStefano Zampini static PetscErrorCode PCFieldSplitSetSchurScale_FieldSplit(PC pc,PetscScalar scale) 2538c096484dSStefano Zampini { 2539c096484dSStefano Zampini PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 2540c096484dSStefano Zampini 2541c096484dSStefano Zampini PetscFunctionBegin; 2542c096484dSStefano Zampini jac->schurscale = scale; 2543c096484dSStefano Zampini PetscFunctionReturn(0); 2544c096484dSStefano Zampini } 2545c096484dSStefano Zampini 254630ad9308SMatthew Knepley /*@C 25478c03b21aSDmitry Karpeev PCFieldSplitGetSchurBlocks - Gets all matrix blocks for the Schur complement 254830ad9308SMatthew Knepley 254930ad9308SMatthew Knepley Collective on KSP 255030ad9308SMatthew Knepley 255130ad9308SMatthew Knepley Input Parameter: 255230ad9308SMatthew Knepley . pc - the preconditioner context 255330ad9308SMatthew Knepley 255430ad9308SMatthew Knepley Output Parameters: 2555a04f6461SBarry Smith + A00 - the (0,0) block 2556a04f6461SBarry Smith . A01 - the (0,1) block 2557a04f6461SBarry Smith . A10 - the (1,0) block 2558a04f6461SBarry Smith - A11 - the (1,1) block 255930ad9308SMatthew Knepley 256030ad9308SMatthew Knepley Level: advanced 256130ad9308SMatthew Knepley 256230ad9308SMatthew Knepley .seealso: PCFIELDSPLIT 256330ad9308SMatthew Knepley @*/ 2564a04f6461SBarry Smith PetscErrorCode PCFieldSplitGetSchurBlocks(PC pc,Mat *A00,Mat *A01,Mat *A10, Mat *A11) 256530ad9308SMatthew Knepley { 256630ad9308SMatthew Knepley PC_FieldSplit *jac = (PC_FieldSplit*) pc->data; 256730ad9308SMatthew Knepley 256830ad9308SMatthew Knepley PetscFunctionBegin; 25690700a824SBarry Smith PetscValidHeaderSpecific(pc,PC_CLASSID,1); 25702c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->type != PC_COMPOSITE_SCHUR,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG, "FieldSplit is not using a Schur complement approach."); 2571a04f6461SBarry Smith if (A00) *A00 = jac->pmat[0]; 2572a04f6461SBarry Smith if (A01) *A01 = jac->B; 2573a04f6461SBarry Smith if (A10) *A10 = jac->C; 2574a04f6461SBarry Smith if (A11) *A11 = jac->pmat[1]; 257530ad9308SMatthew Knepley PetscFunctionReturn(0); 257630ad9308SMatthew Knepley } 257730ad9308SMatthew Knepley 2578b09e027eSCarola Kruse /*@ 2579e071a0a4SCarola Kruse PCFieldSplitSetGKBTol - Sets the solver tolerance for the generalized Golub-Kahan bidiagonalization preconditioner. 2580b09e027eSCarola Kruse 2581b09e027eSCarola Kruse Collective on PC 2582b09e027eSCarola Kruse 2583e071a0a4SCarola Kruse Notes: 2584e071a0a4SCarola Kruse The generalized GKB algorithm uses a lower bound estimate of the error in energy norm as stopping criterion. 2585e071a0a4SCarola Kruse It stops once the lower bound estimate undershoots the required solver tolerance. Although the actual error might be bigger than 2586e071a0a4SCarola Kruse this estimate, the stopping criterion is satisfactory in practical cases [A13]. 2587e071a0a4SCarola Kruse 2588e071a0a4SCarola Kruse [Ar13] Generalized Golub-Kahan bidiagonalization and stopping criteria, SIAM J. Matrix Anal. Appl., Vol. 34, No. 2, pp. 571-592, 2013. 2589e071a0a4SCarola Kruse 2590b09e027eSCarola Kruse Input Parameters: 2591b09e027eSCarola Kruse + pc - the preconditioner context 2592b09e027eSCarola Kruse - tolerance - the solver tolerance 2593b09e027eSCarola Kruse 2594b09e027eSCarola Kruse Options Database: 2595b09e027eSCarola Kruse . -pc_fieldsplit_gkb_tol - default is 1e-5 2596b09e027eSCarola Kruse 2597b09e027eSCarola Kruse Level: intermediate 2598b09e027eSCarola Kruse 2599b09e027eSCarola Kruse .seealso: PCFIELDSPLIT, PCFieldSplitSetGKBDelay(), PCFieldSplitSetGKBNu(), PCFieldSplitSetGKBMaxit() 2600b09e027eSCarola Kruse @*/ 2601b09e027eSCarola Kruse PetscErrorCode PCFieldSplitSetGKBTol(PC pc,PetscReal tolerance) 2602b09e027eSCarola Kruse { 2603b09e027eSCarola Kruse PetscFunctionBegin; 2604b09e027eSCarola Kruse PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2605de482cd7SCarola Kruse PetscValidLogicalCollectiveReal(pc,tolerance,2); 26065f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetGKBTol_C",(PC,PetscReal),(pc,tolerance))); 2607b09e027eSCarola Kruse PetscFunctionReturn(0); 2608b09e027eSCarola Kruse } 2609b09e027eSCarola Kruse 2610b09e027eSCarola Kruse static PetscErrorCode PCFieldSplitSetGKBTol_FieldSplit(PC pc,PetscReal tolerance) 2611b09e027eSCarola Kruse { 2612b09e027eSCarola Kruse PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 2613b09e027eSCarola Kruse 2614b09e027eSCarola Kruse PetscFunctionBegin; 2615b09e027eSCarola Kruse jac->gkbtol = tolerance; 2616b09e027eSCarola Kruse PetscFunctionReturn(0); 2617b09e027eSCarola Kruse } 2618b09e027eSCarola Kruse 2619b09e027eSCarola Kruse /*@ 2620e071a0a4SCarola Kruse PCFieldSplitSetGKBMaxit - Sets the maximum number of iterations for the generalized Golub-Kahan bidiagonalization 2621e071a0a4SCarola Kruse preconditioner. 2622b09e027eSCarola Kruse 2623b09e027eSCarola Kruse Collective on PC 2624b09e027eSCarola Kruse 2625b09e027eSCarola Kruse Input Parameters: 2626b09e027eSCarola Kruse + pc - the preconditioner context 2627b09e027eSCarola Kruse - maxit - the maximum number of iterations 2628b09e027eSCarola Kruse 2629b09e027eSCarola Kruse Options Database: 2630b09e027eSCarola Kruse . -pc_fieldsplit_gkb_maxit - default is 100 2631b09e027eSCarola Kruse 2632b09e027eSCarola Kruse Level: intermediate 2633b09e027eSCarola Kruse 2634b09e027eSCarola Kruse .seealso: PCFIELDSPLIT, PCFieldSplitSetGKBDelay(), PCFieldSplitSetGKBTol(), PCFieldSplitSetGKBNu() 2635b09e027eSCarola Kruse @*/ 2636b09e027eSCarola Kruse PetscErrorCode PCFieldSplitSetGKBMaxit(PC pc,PetscInt maxit) 2637b09e027eSCarola Kruse { 2638b09e027eSCarola Kruse PetscFunctionBegin; 2639b09e027eSCarola Kruse PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2640de482cd7SCarola Kruse PetscValidLogicalCollectiveInt(pc,maxit,2); 26415f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetGKBMaxit_C",(PC,PetscInt),(pc,maxit))); 2642b09e027eSCarola Kruse PetscFunctionReturn(0); 2643b09e027eSCarola Kruse } 2644b09e027eSCarola Kruse 2645b09e027eSCarola Kruse static PetscErrorCode PCFieldSplitSetGKBMaxit_FieldSplit(PC pc,PetscInt maxit) 2646b09e027eSCarola Kruse { 2647b09e027eSCarola Kruse PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 2648b09e027eSCarola Kruse 2649b09e027eSCarola Kruse PetscFunctionBegin; 2650b09e027eSCarola Kruse jac->gkbmaxit = maxit; 2651b09e027eSCarola Kruse PetscFunctionReturn(0); 2652b09e027eSCarola Kruse } 2653b09e027eSCarola Kruse 2654b09e027eSCarola Kruse /*@ 2655e071a0a4SCarola Kruse PCFieldSplitSetGKBDelay - Sets the delay in the lower bound error estimate in the generalized Golub-Kahan bidiagonalization 2656e071a0a4SCarola Kruse preconditioner. 2657b09e027eSCarola Kruse 2658b09e027eSCarola Kruse Collective on PC 2659b09e027eSCarola Kruse 2660b09e027eSCarola Kruse Notes: 2661e071a0a4SCarola Kruse The algorithm uses a lower bound estimate of the error in energy norm as stopping criterion. The lower bound of the error ||u-u^k||_H 2662e071a0a4SCarola Kruse is expressed as a truncated sum. The error at iteration k can only be measured at iteration (k + delay), and thus the algorithm needs 2663e071a0a4SCarola Kruse at least (delay + 1) iterations to stop. For more details on the generalized Golub-Kahan bidiagonalization method and its lower bound stopping criterion, please refer to 2664b09e027eSCarola Kruse 2665b09e027eSCarola Kruse [Ar13] Generalized Golub-Kahan bidiagonalization and stopping criteria, SIAM J. Matrix Anal. Appl., Vol. 34, No. 2, pp. 571-592, 2013. 2666b09e027eSCarola Kruse 2667b09e027eSCarola Kruse Input Parameters: 2668b09e027eSCarola Kruse + pc - the preconditioner context 2669b09e027eSCarola Kruse - delay - the delay window in the lower bound estimate 2670b09e027eSCarola Kruse 2671b09e027eSCarola Kruse Options Database: 2672b09e027eSCarola Kruse . -pc_fieldsplit_gkb_delay - default is 5 2673b09e027eSCarola Kruse 2674b09e027eSCarola Kruse Level: intermediate 2675b09e027eSCarola Kruse 2676b09e027eSCarola Kruse .seealso: PCFIELDSPLIT, PCFieldSplitSetGKBNu(), PCFieldSplitSetGKBTol(), PCFieldSplitSetGKBMaxit() 2677b09e027eSCarola Kruse @*/ 2678b09e027eSCarola Kruse PetscErrorCode PCFieldSplitSetGKBDelay(PC pc,PetscInt delay) 2679b09e027eSCarola Kruse { 2680b09e027eSCarola Kruse PetscFunctionBegin; 2681b09e027eSCarola Kruse PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2682b09e027eSCarola Kruse PetscValidLogicalCollectiveInt(pc,delay,2); 26835f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetGKBDelay_C",(PC,PetscInt),(pc,delay))); 2684b09e027eSCarola Kruse PetscFunctionReturn(0); 2685b09e027eSCarola Kruse } 2686b09e027eSCarola Kruse 2687b09e027eSCarola Kruse static PetscErrorCode PCFieldSplitSetGKBDelay_FieldSplit(PC pc,PetscInt delay) 2688b09e027eSCarola Kruse { 2689b09e027eSCarola Kruse PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 2690b09e027eSCarola Kruse 2691b09e027eSCarola Kruse PetscFunctionBegin; 2692b09e027eSCarola Kruse jac->gkbdelay = delay; 2693b09e027eSCarola Kruse PetscFunctionReturn(0); 2694b09e027eSCarola Kruse } 2695b09e027eSCarola Kruse 2696b09e027eSCarola Kruse /*@ 2697b09e027eSCarola Kruse PCFieldSplitSetGKBNu - Sets the scalar value nu >= 0 in the transformation H = A00 + nu*A01*A01' of the (1,1) block in the Golub-Kahan bidiagonalization preconditioner. 2698b09e027eSCarola Kruse 2699b09e027eSCarola Kruse Collective on PC 2700b09e027eSCarola Kruse 2701b09e027eSCarola Kruse Notes: 2702e071a0a4SCarola Kruse This shift is in general done to obtain better convergence properties for the outer loop of the algorithm. This is often achieved by chosing nu sufficiently big. However, 2703b09e027eSCarola Kruse if nu is chosen too big, the matrix H might be badly conditioned and the solution of the linear system Hx = b in the inner loop gets difficult. It is therefore 2704b09e027eSCarola Kruse necessary to find a good balance in between the convergence of the inner and outer loop. 2705b09e027eSCarola Kruse 2706b09e027eSCarola Kruse For nu = 0, no shift is done. In this case A00 has to be positive definite. The matrix N in [Ar13] is then chosen as identity. 2707b09e027eSCarola Kruse 2708b09e027eSCarola Kruse [Ar13] Generalized Golub-Kahan bidiagonalization and stopping criteria, SIAM J. Matrix Anal. Appl., Vol. 34, No. 2, pp. 571-592, 2013. 2709b09e027eSCarola Kruse 2710b09e027eSCarola Kruse Input Parameters: 2711b09e027eSCarola Kruse + pc - the preconditioner context 2712b09e027eSCarola Kruse - nu - the shift parameter 2713b09e027eSCarola Kruse 2714b09e027eSCarola Kruse Options Database: 2715b09e027eSCarola Kruse . -pc_fieldsplit_gkb_nu - default is 1 2716b09e027eSCarola Kruse 2717b09e027eSCarola Kruse Level: intermediate 2718b09e027eSCarola Kruse 2719b09e027eSCarola Kruse .seealso: PCFIELDSPLIT, PCFieldSplitSetGKBDelay(), PCFieldSplitSetGKBTol(), PCFieldSplitSetGKBMaxit() 2720b09e027eSCarola Kruse @*/ 2721b09e027eSCarola Kruse PetscErrorCode PCFieldSplitSetGKBNu(PC pc,PetscReal nu) 2722b09e027eSCarola Kruse { 2723b09e027eSCarola Kruse PetscFunctionBegin; 2724b09e027eSCarola Kruse PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2725de482cd7SCarola Kruse PetscValidLogicalCollectiveReal(pc,nu,2); 27265f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetGKBNu_C",(PC,PetscReal),(pc,nu))); 2727b09e027eSCarola Kruse PetscFunctionReturn(0); 2728b09e027eSCarola Kruse } 2729b09e027eSCarola Kruse 2730b09e027eSCarola Kruse static PetscErrorCode PCFieldSplitSetGKBNu_FieldSplit(PC pc,PetscReal nu) 2731b09e027eSCarola Kruse { 2732b09e027eSCarola Kruse PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 2733b09e027eSCarola Kruse 2734b09e027eSCarola Kruse PetscFunctionBegin; 2735b09e027eSCarola Kruse jac->gkbnu = nu; 2736b09e027eSCarola Kruse PetscFunctionReturn(0); 2737b09e027eSCarola Kruse } 2738b09e027eSCarola Kruse 27391e6b0712SBarry Smith static PetscErrorCode PCFieldSplitSetType_FieldSplit(PC pc,PCCompositeType type) 274079416396SBarry Smith { 274179416396SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 274279416396SBarry Smith 274379416396SBarry Smith PetscFunctionBegin; 274479416396SBarry Smith jac->type = type; 2745e071a0a4SCarola Kruse 27465f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitGetSubKSP_C",0)); 27475f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetSchurPre_C",0)); 27485f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitGetSchurPre_C",0)); 27495f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetSchurFactType_C",0)); 27505f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetSchurScale_C",0)); 27515f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetGKBTol_C",0)); 27525f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetGKBMaxit_C",0)); 27535f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetGKBNu_C",0)); 27545f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetGKBDelay_C",0)); 2755e071a0a4SCarola Kruse 27563b224e63SBarry Smith if (type == PC_COMPOSITE_SCHUR) { 27573b224e63SBarry Smith pc->ops->apply = PCApply_FieldSplit_Schur; 27583b224e63SBarry Smith pc->ops->view = PCView_FieldSplit_Schur; 27592fa5cd67SKarl Rupp 27605f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitGetSubKSP_C",PCFieldSplitGetSubKSP_FieldSplit_Schur)); 27615f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetSchurPre_C",PCFieldSplitSetSchurPre_FieldSplit)); 27625f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitGetSchurPre_C",PCFieldSplitGetSchurPre_FieldSplit)); 27635f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetSchurFactType_C",PCFieldSplitSetSchurFactType_FieldSplit)); 27645f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetSchurScale_C",PCFieldSplitSetSchurScale_FieldSplit)); 2765a51937d4SCarola Kruse } else if (type == PC_COMPOSITE_GKB) { 2766a51937d4SCarola Kruse pc->ops->apply = PCApply_FieldSplit_GKB; 2767a51937d4SCarola Kruse pc->ops->view = PCView_FieldSplit_GKB; 2768e69d4d44SBarry Smith 27695f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitGetSubKSP_C",PCFieldSplitGetSubKSP_FieldSplit)); 27705f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetGKBTol_C",PCFieldSplitSetGKBTol_FieldSplit)); 27715f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetGKBMaxit_C",PCFieldSplitSetGKBMaxit_FieldSplit)); 27725f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetGKBNu_C",PCFieldSplitSetGKBNu_FieldSplit)); 27735f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetGKBDelay_C",PCFieldSplitSetGKBDelay_FieldSplit)); 27743b224e63SBarry Smith } else { 27753b224e63SBarry Smith pc->ops->apply = PCApply_FieldSplit; 27763b224e63SBarry Smith pc->ops->view = PCView_FieldSplit; 27772fa5cd67SKarl Rupp 27785f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitGetSubKSP_C",PCFieldSplitGetSubKSP_FieldSplit)); 27793b224e63SBarry Smith } 278079416396SBarry Smith PetscFunctionReturn(0); 278179416396SBarry Smith } 278279416396SBarry Smith 27831e6b0712SBarry Smith static PetscErrorCode PCFieldSplitSetBlockSize_FieldSplit(PC pc,PetscInt bs) 278451f519a2SBarry Smith { 278551f519a2SBarry Smith PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 278651f519a2SBarry Smith 278751f519a2SBarry Smith PetscFunctionBegin; 27882c71b3e2SJacob Faibussowitsch PetscCheckFalse(bs < 1,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Blocksize must be positive, you gave %D",bs); 27892c71b3e2SJacob Faibussowitsch PetscCheckFalse(jac->bs > 0 && jac->bs != bs,PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONGSTATE,"Cannot change fieldsplit blocksize from %D to %D after it has been set",jac->bs,bs); 279051f519a2SBarry Smith jac->bs = bs; 279151f519a2SBarry Smith PetscFunctionReturn(0); 279251f519a2SBarry Smith } 279351f519a2SBarry Smith 27945ddf11f8SNicolas Barnafi static PetscErrorCode PCSetCoordinates_FieldSplit(PC pc, PetscInt dim, PetscInt nloc, PetscReal coords[]) 27955ddf11f8SNicolas Barnafi { 27965ddf11f8SNicolas Barnafi PC_FieldSplit * jac = (PC_FieldSplit*)pc->data; 27975ddf11f8SNicolas Barnafi PC_FieldSplitLink ilink_current = jac->head; 27985ddf11f8SNicolas Barnafi PetscInt ii; 27995ddf11f8SNicolas Barnafi IS is_owned; 28005ddf11f8SNicolas Barnafi 28015ddf11f8SNicolas Barnafi PetscFunctionBegin; 28025ddf11f8SNicolas Barnafi jac->coordinates_set = PETSC_TRUE; // Internal flag 28035f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOwnershipIS(pc->mat,&is_owned,PETSC_NULL)); 28045ddf11f8SNicolas Barnafi 28055ddf11f8SNicolas Barnafi ii=0; 28065ddf11f8SNicolas Barnafi while (ilink_current) { 28075ddf11f8SNicolas Barnafi // For each IS, embed it to get local coords indces 28085ddf11f8SNicolas Barnafi IS is_coords; 28095ddf11f8SNicolas Barnafi PetscInt ndofs_block; 28105ddf11f8SNicolas Barnafi const PetscInt *block_dofs_enumeration; // Numbering of the dofs relevant to the current block 28115ddf11f8SNicolas Barnafi 28125ddf11f8SNicolas Barnafi // Setting drop to true for safety. It should make no difference. 28135f80ce2aSJacob Faibussowitsch CHKERRQ(ISEmbed(ilink_current->is, is_owned, PETSC_TRUE, &is_coords)); 28145f80ce2aSJacob Faibussowitsch CHKERRQ(ISGetLocalSize(is_coords, &ndofs_block)); 28155f80ce2aSJacob Faibussowitsch CHKERRQ(ISGetIndices(is_coords, &block_dofs_enumeration)); 28165ddf11f8SNicolas Barnafi 28175ddf11f8SNicolas Barnafi // Allocate coordinates vector and set it directly 28185f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(ndofs_block * dim, &(ilink_current->coords))); 28195ddf11f8SNicolas Barnafi for (PetscInt dof=0;dof<ndofs_block;++dof) { 28205ddf11f8SNicolas Barnafi for (PetscInt d=0;d<dim;++d) { 28215ddf11f8SNicolas Barnafi (ilink_current->coords)[dim*dof + d] = coords[dim * block_dofs_enumeration[dof] + d]; 28225ddf11f8SNicolas Barnafi } 28235ddf11f8SNicolas Barnafi } 28245ddf11f8SNicolas Barnafi ilink_current->dim = dim; 28255ddf11f8SNicolas Barnafi ilink_current->ndofs = ndofs_block; 28265f80ce2aSJacob Faibussowitsch CHKERRQ(ISRestoreIndices(is_coords, &block_dofs_enumeration)); 28275f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&is_coords)); 28285ddf11f8SNicolas Barnafi ilink_current = ilink_current->next; 28295ddf11f8SNicolas Barnafi ++ii; 28305ddf11f8SNicolas Barnafi } 28315f80ce2aSJacob Faibussowitsch CHKERRQ(ISDestroy(&is_owned)); 28325ddf11f8SNicolas Barnafi PetscFunctionReturn(0); 28335ddf11f8SNicolas Barnafi } 28345ddf11f8SNicolas Barnafi 2835bc08b0f1SBarry Smith /*@ 283679416396SBarry Smith PCFieldSplitSetType - Sets the type of fieldsplit preconditioner. 283779416396SBarry Smith 283879416396SBarry Smith Collective on PC 283979416396SBarry Smith 2840d8d19677SJose E. Roman Input Parameters: 2841a2b725a8SWilliam Gropp + pc - the preconditioner context 2842a2b725a8SWilliam Gropp - type - PC_COMPOSITE_ADDITIVE, PC_COMPOSITE_MULTIPLICATIVE (default), PC_COMPOSITE_SYMMETRIC_MULTIPLICATIVE, PC_COMPOSITE_SPECIAL, PC_COMPOSITE_SCHUR 284379416396SBarry Smith 284479416396SBarry Smith Options Database Key: 2845a4efd8eaSMatthew Knepley . -pc_fieldsplit_type <type: one of multiplicative, additive, symmetric_multiplicative, special, schur> - Sets fieldsplit preconditioner type 284679416396SBarry Smith 2847b02e2d75SMatthew G Knepley Level: Intermediate 284879416396SBarry Smith 284979416396SBarry Smith .seealso: PCCompositeSetType() 285079416396SBarry Smith 285179416396SBarry Smith @*/ 28527087cfbeSBarry Smith PetscErrorCode PCFieldSplitSetType(PC pc,PCCompositeType type) 285379416396SBarry Smith { 285479416396SBarry Smith PetscFunctionBegin; 28550700a824SBarry Smith PetscValidHeaderSpecific(pc,PC_CLASSID,1); 28565f80ce2aSJacob Faibussowitsch CHKERRQ(PetscTryMethod(pc,"PCFieldSplitSetType_C",(PC,PCCompositeType),(pc,type))); 285779416396SBarry Smith PetscFunctionReturn(0); 285879416396SBarry Smith } 285979416396SBarry Smith 2860b02e2d75SMatthew G Knepley /*@ 2861b02e2d75SMatthew G Knepley PCFieldSplitGetType - Gets the type of fieldsplit preconditioner. 2862b02e2d75SMatthew G Knepley 2863b02e2d75SMatthew G Knepley Not collective 2864b02e2d75SMatthew G Knepley 2865b02e2d75SMatthew G Knepley Input Parameter: 2866b02e2d75SMatthew G Knepley . pc - the preconditioner context 2867b02e2d75SMatthew G Knepley 2868b02e2d75SMatthew G Knepley Output Parameter: 2869b02e2d75SMatthew G Knepley . type - PC_COMPOSITE_ADDITIVE, PC_COMPOSITE_MULTIPLICATIVE (default), PC_COMPOSITE_SYMMETRIC_MULTIPLICATIVE, PC_COMPOSITE_SPECIAL, PC_COMPOSITE_SCHUR 2870b02e2d75SMatthew G Knepley 2871b02e2d75SMatthew G Knepley Level: Intermediate 2872b02e2d75SMatthew G Knepley 2873b02e2d75SMatthew G Knepley .seealso: PCCompositeSetType() 2874b02e2d75SMatthew G Knepley @*/ 2875b02e2d75SMatthew G Knepley PetscErrorCode PCFieldSplitGetType(PC pc, PCCompositeType *type) 2876b02e2d75SMatthew G Knepley { 2877b02e2d75SMatthew G Knepley PC_FieldSplit *jac = (PC_FieldSplit*) pc->data; 2878b02e2d75SMatthew G Knepley 2879b02e2d75SMatthew G Knepley PetscFunctionBegin; 2880b02e2d75SMatthew G Knepley PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2881b02e2d75SMatthew G Knepley PetscValidIntPointer(type,2); 2882b02e2d75SMatthew G Knepley *type = jac->type; 2883b02e2d75SMatthew G Knepley PetscFunctionReturn(0); 2884b02e2d75SMatthew G Knepley } 2885b02e2d75SMatthew G Knepley 28864ab8060aSDmitry Karpeev /*@ 28874ab8060aSDmitry Karpeev PCFieldSplitSetDMSplits - Flags whether DMCreateFieldDecomposition() should be used to define the splits, whenever possible. 28884ab8060aSDmitry Karpeev 28894ab8060aSDmitry Karpeev Logically Collective 28904ab8060aSDmitry Karpeev 28914ab8060aSDmitry Karpeev Input Parameters: 28924ab8060aSDmitry Karpeev + pc - the preconditioner context 28934ab8060aSDmitry Karpeev - flg - boolean indicating whether to use field splits defined by the DM 28944ab8060aSDmitry Karpeev 28954ab8060aSDmitry Karpeev Options Database Key: 289667b8a455SSatish Balay . -pc_fieldsplit_dm_splits <bool> - use the field splits defined by the DM 28974ab8060aSDmitry Karpeev 28984ab8060aSDmitry Karpeev Level: Intermediate 28994ab8060aSDmitry Karpeev 29004ab8060aSDmitry Karpeev .seealso: PCFieldSplitGetDMSplits() 29014ab8060aSDmitry Karpeev 29024ab8060aSDmitry Karpeev @*/ 29034ab8060aSDmitry Karpeev PetscErrorCode PCFieldSplitSetDMSplits(PC pc,PetscBool flg) 29044ab8060aSDmitry Karpeev { 29054ab8060aSDmitry Karpeev PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 29064ab8060aSDmitry Karpeev PetscBool isfs; 29074ab8060aSDmitry Karpeev 29084ab8060aSDmitry Karpeev PetscFunctionBegin; 29094ab8060aSDmitry Karpeev PetscValidHeaderSpecific(pc,PC_CLASSID,1); 29104ab8060aSDmitry Karpeev PetscValidLogicalCollectiveBool(pc,flg,2); 29115f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)pc,PCFIELDSPLIT,&isfs)); 29124ab8060aSDmitry Karpeev if (isfs) { 29134ab8060aSDmitry Karpeev jac->dm_splits = flg; 29144ab8060aSDmitry Karpeev } 29154ab8060aSDmitry Karpeev PetscFunctionReturn(0); 29164ab8060aSDmitry Karpeev } 29174ab8060aSDmitry Karpeev 29184ab8060aSDmitry Karpeev /*@ 29194ab8060aSDmitry Karpeev PCFieldSplitGetDMSplits - Returns flag indicating whether DMCreateFieldDecomposition() should be used to define the splits, whenever possible. 29204ab8060aSDmitry Karpeev 29214ab8060aSDmitry Karpeev Logically Collective 29224ab8060aSDmitry Karpeev 29234ab8060aSDmitry Karpeev Input Parameter: 29244ab8060aSDmitry Karpeev . pc - the preconditioner context 29254ab8060aSDmitry Karpeev 29264ab8060aSDmitry Karpeev Output Parameter: 29274ab8060aSDmitry Karpeev . flg - boolean indicating whether to use field splits defined by the DM 29284ab8060aSDmitry Karpeev 29294ab8060aSDmitry Karpeev Level: Intermediate 29304ab8060aSDmitry Karpeev 29314ab8060aSDmitry Karpeev .seealso: PCFieldSplitSetDMSplits() 29324ab8060aSDmitry Karpeev 29334ab8060aSDmitry Karpeev @*/ 29344ab8060aSDmitry Karpeev PetscErrorCode PCFieldSplitGetDMSplits(PC pc,PetscBool* flg) 29354ab8060aSDmitry Karpeev { 29364ab8060aSDmitry Karpeev PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 29374ab8060aSDmitry Karpeev PetscBool isfs; 29384ab8060aSDmitry Karpeev 29394ab8060aSDmitry Karpeev PetscFunctionBegin; 29404ab8060aSDmitry Karpeev PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2941534a8f05SLisandro Dalcin PetscValidBoolPointer(flg,2); 29425f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectTypeCompare((PetscObject)pc,PCFIELDSPLIT,&isfs)); 29434ab8060aSDmitry Karpeev if (isfs) { 29444ab8060aSDmitry Karpeev if (flg) *flg = jac->dm_splits; 29454ab8060aSDmitry Karpeev } 29464ab8060aSDmitry Karpeev PetscFunctionReturn(0); 29474ab8060aSDmitry Karpeev } 29484ab8060aSDmitry Karpeev 29497b752e3dSPatrick Sanan /*@ 29507b752e3dSPatrick Sanan PCFieldSplitGetDetectSaddlePoint - Returns flag indicating whether PCFieldSplit will attempt to automatically determine fields based on zero diagonal entries. 29517b752e3dSPatrick Sanan 29527b752e3dSPatrick Sanan Logically Collective 29537b752e3dSPatrick Sanan 29547b752e3dSPatrick Sanan Input Parameter: 29557b752e3dSPatrick Sanan . pc - the preconditioner context 29567b752e3dSPatrick Sanan 29577b752e3dSPatrick Sanan Output Parameter: 29587b752e3dSPatrick Sanan . flg - boolean indicating whether to detect fields or not 29597b752e3dSPatrick Sanan 29607b752e3dSPatrick Sanan Level: Intermediate 29617b752e3dSPatrick Sanan 29627b752e3dSPatrick Sanan .seealso: PCFIELDSPLIT, PCFieldSplitSetDetectSaddlePoint() 29637b752e3dSPatrick Sanan 29647b752e3dSPatrick Sanan @*/ 29657b752e3dSPatrick Sanan PetscErrorCode PCFieldSplitGetDetectSaddlePoint(PC pc,PetscBool *flg) 29667b752e3dSPatrick Sanan { 29677b752e3dSPatrick Sanan PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 29687b752e3dSPatrick Sanan 29697b752e3dSPatrick Sanan PetscFunctionBegin; 29707b752e3dSPatrick Sanan *flg = jac->detect; 29717b752e3dSPatrick Sanan PetscFunctionReturn(0); 29727b752e3dSPatrick Sanan } 29737b752e3dSPatrick Sanan 29747b752e3dSPatrick Sanan /*@ 29757b752e3dSPatrick Sanan PCFieldSplitSetDetectSaddlePoint - Sets flag indicating whether PCFieldSplit will attempt to automatically determine fields based on zero diagonal entries. 29767b752e3dSPatrick Sanan 29777b752e3dSPatrick Sanan Logically Collective 29787b752e3dSPatrick Sanan 29797b752e3dSPatrick Sanan Input Parameter: 29807b752e3dSPatrick Sanan . pc - the preconditioner context 29817b752e3dSPatrick Sanan 29827b752e3dSPatrick Sanan Output Parameter: 29837b752e3dSPatrick Sanan . flg - boolean indicating whether to detect fields or not 29847b752e3dSPatrick Sanan 29857b752e3dSPatrick Sanan Options Database Key: 2986147403d9SBarry Smith . -pc_fieldsplit_detect_saddle_point <bool> - detect and use the saddle point 2987147403d9SBarry Smith 2988147403d9SBarry Smith Notes: 2989147403d9SBarry Smith Also sets the split type to PC_COMPOSITE_SCHUR (see PCFieldSplitSetType()) and the Schur preconditioner type to PC_FIELDSPLIT_SCHUR_PRE_SELF (see PCFieldSplitSetSchurPre()). 29907b752e3dSPatrick Sanan 29917b752e3dSPatrick Sanan Level: Intermediate 29927b752e3dSPatrick Sanan 29937b752e3dSPatrick Sanan .seealso: PCFIELDSPLIT, PCFieldSplitSetDetectSaddlePoint(), PCFieldSplitSetType(), PCFieldSplitSetSchurPre() 29947b752e3dSPatrick Sanan 29957b752e3dSPatrick Sanan @*/ 29967b752e3dSPatrick Sanan PetscErrorCode PCFieldSplitSetDetectSaddlePoint(PC pc,PetscBool flg) 29977b752e3dSPatrick Sanan { 29987b752e3dSPatrick Sanan PC_FieldSplit *jac = (PC_FieldSplit*)pc->data; 29997b752e3dSPatrick Sanan 30007b752e3dSPatrick Sanan PetscFunctionBegin; 30017b752e3dSPatrick Sanan jac->detect = flg; 30027b752e3dSPatrick Sanan if (jac->detect) { 30035f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetType(pc,PC_COMPOSITE_SCHUR)); 30045f80ce2aSJacob Faibussowitsch CHKERRQ(PCFieldSplitSetSchurPre(pc,PC_FIELDSPLIT_SCHUR_PRE_SELF,NULL)); 30057b752e3dSPatrick Sanan } 30067b752e3dSPatrick Sanan PetscFunctionReturn(0); 30077b752e3dSPatrick Sanan } 30087b752e3dSPatrick Sanan 30090971522cSBarry Smith /* -------------------------------------------------------------------------------------*/ 30100971522cSBarry Smith /*MC 3011a8c7a070SBarry Smith PCFIELDSPLIT - Preconditioner created by combining separate preconditioners for individual 3012a04f6461SBarry Smith fields or groups of fields. See the users manual section "Solving Block Matrices" for more details. 30130971522cSBarry Smith 3014edf189efSBarry Smith To set options on the solvers for each block append -fieldsplit_ to all the PC 3015edf189efSBarry Smith options database keys. For example, -fieldsplit_pc_type ilu -fieldsplit_pc_factor_levels 1 30160971522cSBarry Smith 3017a8c7a070SBarry Smith To set the options on the solvers separate for each block call PCFieldSplitGetSubKSP() 301869a612a9SBarry Smith and set the options directly on the resulting KSP object 30190971522cSBarry Smith 30200971522cSBarry Smith Level: intermediate 30210971522cSBarry Smith 302279416396SBarry Smith Options Database Keys: 302381540f2fSBarry Smith + -pc_fieldsplit_%d_fields <a,b,..> - indicates the fields to be used in the %d'th split 302481540f2fSBarry Smith . -pc_fieldsplit_default - automatically add any fields to additional splits that have not 302581540f2fSBarry Smith been supplied explicitly by -pc_fieldsplit_%d_fields 302681540f2fSBarry Smith . -pc_fieldsplit_block_size <bs> - size of block that defines fields (i.e. there are bs fields) 3027a51937d4SCarola Kruse . -pc_fieldsplit_type <additive,multiplicative,symmetric_multiplicative,schur,gkb> - type of relaxation or factorization splitting 3028a6a584a2SBarry Smith . -pc_fieldsplit_schur_precondition <self,selfp,user,a11,full> - default is a11; see PCFieldSplitSetSchurPre() 3029412be6bfSPatrick Sanan . -pc_fieldsplit_schur_fact_type <diag,lower,upper,full> - set factorization type when using -pc_fieldsplit_type schur; see PCFieldSplitSetSchurFactType() 3030fb6809a2SPatrick Sanan - -pc_fieldsplit_detect_saddle_point - automatically finds rows with zero diagonal and uses Schur complement with no preconditioner as the solver 303179416396SBarry Smith 3032fb6809a2SPatrick Sanan Options prefixes for inner solvers when using the Schur complement preconditioner are -fieldsplit_0_ and -fieldsplit_1_ . 3033fb6809a2SPatrick Sanan For all other solvers they are -fieldsplit_%d_ for the dth field; use -fieldsplit_ for all fields. 3034fb6809a2SPatrick Sanan The options prefix for the inner solver when using the Golub-Kahan biadiagonalization preconditioner is -fieldsplit_0_ 30355d4c12cdSJungho Lee 3036c8a0d604SMatthew G Knepley Notes: 3037c8a0d604SMatthew G Knepley Use PCFieldSplitSetFields() to set fields defined by "strided" entries and PCFieldSplitSetIS() 3038d32f9abdSBarry Smith to define a field by an arbitrary collection of entries. 3039d32f9abdSBarry Smith 3040d32f9abdSBarry Smith If no fields are set the default is used. The fields are defined by entries strided by bs, 3041d32f9abdSBarry Smith beginning at 0 then 1, etc to bs-1. The block size can be set with PCFieldSplitSetBlockSize(), 3042d32f9abdSBarry Smith if this is not called the block size defaults to the blocksize of the second matrix passed 3043d32f9abdSBarry Smith to KSPSetOperators()/PCSetOperators(). 3044d32f9abdSBarry Smith 30452da392ccSBarry Smith $ For the Schur complement preconditioner if J = [ A00 A01] 30462da392ccSBarry Smith $ [ A10 A11] 3047c8a0d604SMatthew G Knepley $ the preconditioner using full factorization is 30482da392ccSBarry Smith $ [ I -ksp(A00) A01] [ inv(A00) 0 ] [ I 0] 30492da392ccSBarry Smith $ [ 0 I ] [ 0 ksp(S)] [ -A10 ksp(A00) I] 305013b05affSJed Brown where the action of inv(A00) is applied using the KSP solver with prefix -fieldsplit_0_. S is the Schur complement 305113b05affSJed Brown $ S = A11 - A10 ksp(A00) A01 305213b05affSJed Brown which is usually dense and not stored explicitly. The action of ksp(S) is computed using the KSP solver with prefix -fieldsplit_splitname_ (where splitname was given 3053686bed4dSStefano Zampini in providing the SECOND split or 1 if not give). For PCFieldSplitGetSubKSP() when field number is 0, 3054c8a0d604SMatthew G Knepley it returns the KSP associated with -fieldsplit_0_ while field number 1 gives -fieldsplit_1_ KSP. By default 3055a6a584a2SBarry Smith A11 is used to construct a preconditioner for S, use PCFieldSplitSetSchurPre() for all the possible ways to construct the preconditioner for S. 30561d71051fSDmitry Karpeev 3057a7476a74SDmitry Karpeev The factorization type is set using -pc_fieldsplit_schur_fact_type <diag, lower, upper, full>. The full is shown above, 30585668aaf4SBarry Smith diag gives 30592da392ccSBarry Smith $ [ inv(A00) 0 ] 30602da392ccSBarry Smith $ [ 0 -ksp(S)] 3061c096484dSStefano Zampini note that slightly counter intuitively there is a negative in front of the ksp(S) so that the preconditioner is positive definite. For SPD matrices J, the sign flip 3062c096484dSStefano Zampini can be turned off with PCFieldSplitSetSchurScale() or by command line -pc_fieldsplit_schur_scale 1.0. The lower factorization is the inverse of 30632da392ccSBarry Smith $ [ A00 0] 30642da392ccSBarry Smith $ [ A10 S] 3065c8a0d604SMatthew G Knepley where the inverses of A00 and S are applied using KSPs. The upper factorization is the inverse of 30662da392ccSBarry Smith $ [ A00 A01] 30672da392ccSBarry Smith $ [ 0 S ] 3068c8a0d604SMatthew G Knepley where again the inverses of A00 and S are applied using KSPs. 3069e69d4d44SBarry Smith 3070edf189efSBarry Smith If only one set of indices (one IS) is provided with PCFieldSplitSetIS() then the complement of that IS 3071edf189efSBarry Smith is used automatically for a second block. 3072edf189efSBarry Smith 3073ff218e97SBarry Smith The fieldsplit preconditioner cannot currently be used with the BAIJ or SBAIJ data formats if the blocksize is larger than 1. 3074ff218e97SBarry Smith Generally it should be used with the AIJ format. 3075ff218e97SBarry Smith 3076ff218e97SBarry Smith The forms of these preconditioners are closely related if not identical to forms derived as "Distributive Iterations", see, 3077ff218e97SBarry Smith for example, page 294 in "Principles of Computational Fluid Dynamics" by Pieter Wesseling. Note that one can also use PCFIELDSPLIT 3078ff218e97SBarry Smith inside a smoother resulting in "Distributive Smoothers". 30790716a85fSBarry Smith 30802da392ccSBarry Smith References: 3081606c0280SSatish Balay . * - A taxonomy and comparison of parallel block multi-level preconditioners for the incompressible Navier-Stokes equations 30823bc631e0SBarry Smith Howard Elman, V.E. Howle, John Shadid, Robert Shuttleworth, Ray Tuminaro, Journal of Computational Physics 227 (2008) 1790--1808 30833bc631e0SBarry Smith http://chess.cs.umd.edu/~elman/papers/tax.pdf 30843bc631e0SBarry Smith 30854d308398SBarry Smith The Constrained Pressure Preconditioner (CPR) can be implemented using PCCOMPOSITE with PCGALERKIN. CPR first solves an R A P subsystem, updates the 30864d308398SBarry Smith residual on all variables (PCCompositeSetType(pc,PC_COMPOSITE_MULTIPLICATIVE)), and then applies a simple ILU like preconditioner on all the variables. 3087a6a584a2SBarry Smith 3088e071a0a4SCarola Kruse The generalized Golub-Kahan bidiagonalization preconditioner (gkb) can be applied to symmetric 2x2 block matrices of the shape 30892da392ccSBarry Smith $ [ A00 A01] 30902da392ccSBarry Smith $ [ A01' 0 ] 3091a51937d4SCarola Kruse with A00 positive semi-definite. The implementation follows [Ar13]. Therein, we choose N := 1/nu * I and the (1,1)-block of the matrix is modified to H = A00 + nu*A01*A01'. 3092e071a0a4SCarola Kruse A linear system Hx = b has to be solved in each iteration of the GKB algorithm. This solver is chosen with the option prefix -fieldsplit_0_. 3093a51937d4SCarola Kruse 3094606c0280SSatish Balay . * - [Ar13] Generalized Golub-Kahan bidiagonalization and stopping criteria, SIAM J. Matrix Anal. Appl., Vol. 34, No. 2, pp. 571-592, 2013. 3095a51937d4SCarola Kruse 3096412be6bfSPatrick Sanan .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, PCLSC, 3097412be6bfSPatrick Sanan PCFieldSplitGetSubKSP(), PCFieldSplitSchurGetSubKSP(), PCFieldSplitSetFields(), 3098412be6bfSPatrick Sanan PCFieldSplitSetType(), PCFieldSplitSetIS(), PCFieldSplitSetSchurPre(), PCFieldSplitSetSchurFactType(), 3099412be6bfSPatrick Sanan MatSchurComplementSetAinvType(), PCFieldSplitSetSchurScale(), PCFieldSplitSetDetectSaddlePoint() 31000971522cSBarry Smith M*/ 31010971522cSBarry Smith 31028cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_FieldSplit(PC pc) 31030971522cSBarry Smith { 31040971522cSBarry Smith PC_FieldSplit *jac; 31050971522cSBarry Smith 31060971522cSBarry Smith PetscFunctionBegin; 31075f80ce2aSJacob Faibussowitsch CHKERRQ(PetscNewLog(pc,&jac)); 31082fa5cd67SKarl Rupp 31090971522cSBarry Smith jac->bs = -1; 31100971522cSBarry Smith jac->nsplits = 0; 31113e197d65SBarry Smith jac->type = PC_COMPOSITE_MULTIPLICATIVE; 3112e6cab6aaSJed Brown jac->schurpre = PC_FIELDSPLIT_SCHUR_PRE_USER; /* Try user preconditioner first, fall back on diagonal */ 3113c9c6ffaaSJed Brown jac->schurfactorization = PC_FIELDSPLIT_SCHUR_FACT_FULL; 3114c096484dSStefano Zampini jac->schurscale = -1.0; 3115fbe7908bSJed Brown jac->dm_splits = PETSC_TRUE; 31167b752e3dSPatrick Sanan jac->detect = PETSC_FALSE; 3117a51937d4SCarola Kruse jac->gkbtol = 1e-5; 3118a51937d4SCarola Kruse jac->gkbdelay = 5; 3119a51937d4SCarola Kruse jac->gkbnu = 1; 3120a51937d4SCarola Kruse jac->gkbmaxit = 100; 3121de482cd7SCarola Kruse jac->gkbmonitor = PETSC_FALSE; 31225ddf11f8SNicolas Barnafi jac->coordinates_set = PETSC_FALSE; 312351f519a2SBarry Smith 31240971522cSBarry Smith pc->data = (void*)jac; 31250971522cSBarry Smith 31260971522cSBarry Smith pc->ops->apply = PCApply_FieldSplit; 3127421e10b8SBarry Smith pc->ops->applytranspose = PCApplyTranspose_FieldSplit; 31280971522cSBarry Smith pc->ops->setup = PCSetUp_FieldSplit; 3129574deadeSBarry Smith pc->ops->reset = PCReset_FieldSplit; 31300971522cSBarry Smith pc->ops->destroy = PCDestroy_FieldSplit; 31310971522cSBarry Smith pc->ops->setfromoptions = PCSetFromOptions_FieldSplit; 31320971522cSBarry Smith pc->ops->view = PCView_FieldSplit; 31330a545947SLisandro Dalcin pc->ops->applyrichardson = NULL; 31340971522cSBarry Smith 31355f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSchurGetSubKSP_C",PCFieldSplitSchurGetSubKSP_FieldSplit)); 31365f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitGetSubKSP_C",PCFieldSplitGetSubKSP_FieldSplit)); 31375f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetFields_C",PCFieldSplitSetFields_FieldSplit)); 31385f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetIS_C",PCFieldSplitSetIS_FieldSplit)); 31395f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetType_C",PCFieldSplitSetType_FieldSplit)); 31405f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitSetBlockSize_C",PCFieldSplitSetBlockSize_FieldSplit)); 31415f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCFieldSplitRestrictIS_C",PCFieldSplitRestrictIS_FieldSplit)); 31425f80ce2aSJacob Faibussowitsch CHKERRQ(PetscObjectComposeFunction((PetscObject)pc,"PCSetCoordinates_C",PCSetCoordinates_FieldSplit)); 31430971522cSBarry Smith PetscFunctionReturn(0); 31440971522cSBarry Smith } 3145