19b54502bSHong Zhang 29b54502bSHong Zhang /* 39b54502bSHong Zhang Defines a direct factorization preconditioner for any Mat implementation 49b54502bSHong Zhang Note: this need not be consided a preconditioner since it supplies 59b54502bSHong Zhang a direct solver. 69b54502bSHong Zhang */ 79b54502bSHong Zhang #include "src/ksp/pc/pcimpl.h" /*I "petscpc.h" I*/ 89b54502bSHong Zhang 99b54502bSHong Zhang typedef struct { 109b54502bSHong Zhang Mat fact; /* factored matrix */ 119b54502bSHong Zhang PetscReal actualfill; /* actual fill in factor */ 129b54502bSHong Zhang PetscTruth inplace; /* flag indicating in-place factorization */ 139b54502bSHong Zhang IS row,col; /* index sets used for reordering */ 149b54502bSHong Zhang MatOrderingType ordering; /* matrix ordering */ 159b54502bSHong Zhang PetscTruth reuseordering; /* reuses previous reordering computed */ 169b54502bSHong Zhang PetscTruth reusefill; /* reuse fill from previous Cholesky */ 179b54502bSHong Zhang MatFactorInfo info; 189b54502bSHong Zhang } PC_Cholesky; 199b54502bSHong Zhang 209b54502bSHong Zhang EXTERN_C_BEGIN 219b54502bSHong Zhang #undef __FUNCT__ 22afaefe49SHong Zhang #define __FUNCT__ "PCFactorSetZeroPivot_Cholesky" 23afaefe49SHong Zhang PetscErrorCode PCFactorSetZeroPivot_Cholesky(PC pc,PetscReal z) 24afaefe49SHong Zhang { 25afaefe49SHong Zhang PC_Cholesky *ch; 26afaefe49SHong Zhang 27afaefe49SHong Zhang PetscFunctionBegin; 28afaefe49SHong Zhang ch = (PC_Cholesky*)pc->data; 29afaefe49SHong Zhang ch->info.zeropivot = z; 30afaefe49SHong Zhang PetscFunctionReturn(0); 31afaefe49SHong Zhang } 32afaefe49SHong Zhang EXTERN_C_END 33afaefe49SHong Zhang 34afaefe49SHong Zhang EXTERN_C_BEGIN 35afaefe49SHong Zhang #undef __FUNCT__ 36afaefe49SHong Zhang #define __FUNCT__ "PCFactorSetShiftNonzero_Cholesky" 37afaefe49SHong Zhang PetscErrorCode PCFactorSetShiftNonzero_Cholesky(PC pc,PetscReal shift) 38afaefe49SHong Zhang { 39afaefe49SHong Zhang PC_Cholesky *dir; 40afaefe49SHong Zhang 41afaefe49SHong Zhang PetscFunctionBegin; 42afaefe49SHong Zhang dir = (PC_Cholesky*)pc->data; 43afaefe49SHong Zhang if (shift == (PetscReal) PETSC_DECIDE) { 44afaefe49SHong Zhang dir->info.shiftnz = 1.e-12; 45afaefe49SHong Zhang } else { 46afaefe49SHong Zhang dir->info.shiftnz = shift; 47afaefe49SHong Zhang } 48afaefe49SHong Zhang PetscFunctionReturn(0); 49afaefe49SHong Zhang } 50afaefe49SHong Zhang EXTERN_C_END 51afaefe49SHong Zhang 52afaefe49SHong Zhang EXTERN_C_BEGIN 53afaefe49SHong Zhang #undef __FUNCT__ 54afaefe49SHong Zhang #define __FUNCT__ "PCFactorSetShiftPd_Cholesky" 55afaefe49SHong Zhang PetscErrorCode PCFactorSetShiftPd_Cholesky(PC pc,PetscTruth shift) 56afaefe49SHong Zhang { 57afaefe49SHong Zhang PC_Cholesky *dir; 58afaefe49SHong Zhang 59afaefe49SHong Zhang PetscFunctionBegin; 60afaefe49SHong Zhang dir = (PC_Cholesky*)pc->data; 61afaefe49SHong Zhang dir->info.shiftpd = shift; 62afaefe49SHong Zhang if (shift) dir->info.shift_fraction = 0.0; 63afaefe49SHong Zhang PetscFunctionReturn(0); 64afaefe49SHong Zhang } 65afaefe49SHong Zhang EXTERN_C_END 66afaefe49SHong Zhang 67afaefe49SHong Zhang EXTERN_C_BEGIN 68afaefe49SHong Zhang #undef __FUNCT__ 699b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetReuseOrdering_Cholesky" 709b54502bSHong Zhang PetscErrorCode PCCholeskySetReuseOrdering_Cholesky(PC pc,PetscTruth flag) 719b54502bSHong Zhang { 729b54502bSHong Zhang PC_Cholesky *lu; 739b54502bSHong Zhang 749b54502bSHong Zhang PetscFunctionBegin; 759b54502bSHong Zhang lu = (PC_Cholesky*)pc->data; 769b54502bSHong Zhang lu->reuseordering = flag; 779b54502bSHong Zhang PetscFunctionReturn(0); 789b54502bSHong Zhang } 799b54502bSHong Zhang EXTERN_C_END 809b54502bSHong Zhang 819b54502bSHong Zhang EXTERN_C_BEGIN 829b54502bSHong Zhang #undef __FUNCT__ 839b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetReuseFill_Cholesky" 849b54502bSHong Zhang PetscErrorCode PCCholeskySetReuseFill_Cholesky(PC pc,PetscTruth flag) 859b54502bSHong Zhang { 869b54502bSHong Zhang PC_Cholesky *lu; 879b54502bSHong Zhang 889b54502bSHong Zhang PetscFunctionBegin; 899b54502bSHong Zhang lu = (PC_Cholesky*)pc->data; 909b54502bSHong Zhang lu->reusefill = flag; 919b54502bSHong Zhang PetscFunctionReturn(0); 929b54502bSHong Zhang } 939b54502bSHong Zhang EXTERN_C_END 949b54502bSHong Zhang 959b54502bSHong Zhang #undef __FUNCT__ 969b54502bSHong Zhang #define __FUNCT__ "PCSetFromOptions_Cholesky" 979b54502bSHong Zhang static PetscErrorCode PCSetFromOptions_Cholesky(PC pc) 989b54502bSHong Zhang { 999b54502bSHong Zhang PC_Cholesky *lu = (PC_Cholesky*)pc->data; 1009b54502bSHong Zhang PetscErrorCode ierr; 1019b54502bSHong Zhang PetscTruth flg; 1029b54502bSHong Zhang char tname[256]; 1039b54502bSHong Zhang PetscFList ordlist; 1049b54502bSHong Zhang 1059b54502bSHong Zhang PetscFunctionBegin; 1069b54502bSHong Zhang ierr = MatOrderingRegisterAll(PETSC_NULL);CHKERRQ(ierr); 1079b54502bSHong Zhang ierr = PetscOptionsHead("Cholesky options");CHKERRQ(ierr); 1089b54502bSHong Zhang ierr = PetscOptionsName("-pc_cholesky_in_place","Form Cholesky in the same memory as the matrix","PCCholeskySetUseInPlace",&flg);CHKERRQ(ierr); 1099b54502bSHong Zhang if (flg) { 1109b54502bSHong Zhang ierr = PCCholeskySetUseInPlace(pc);CHKERRQ(ierr); 1119b54502bSHong Zhang } 1129b54502bSHong Zhang ierr = PetscOptionsReal("-pc_cholesky_fill","Expected non-zeros in Cholesky/non-zeros in matrix","PCCholeskySetFill",lu->info.fill,&lu->info.fill,0);CHKERRQ(ierr); 1139b54502bSHong Zhang 1149b54502bSHong Zhang ierr = PetscOptionsName("-pc_cholesky_reuse_fill","Use fill from previous factorization","PCCholeskySetReuseFill",&flg);CHKERRQ(ierr); 1159b54502bSHong Zhang if (flg) { 1169b54502bSHong Zhang ierr = PCCholeskySetReuseFill(pc,PETSC_TRUE);CHKERRQ(ierr); 1179b54502bSHong Zhang } 1189b54502bSHong Zhang ierr = PetscOptionsName("-pc_cholesky_reuse_ordering","Reuse ordering from previous factorization","PCCholeskySetReuseOrdering",&flg);CHKERRQ(ierr); 1199b54502bSHong Zhang if (flg) { 1209b54502bSHong Zhang ierr = PCCholeskySetReuseOrdering(pc,PETSC_TRUE);CHKERRQ(ierr); 1219b54502bSHong Zhang } 1229b54502bSHong Zhang 1239b54502bSHong Zhang ierr = MatGetOrderingList(&ordlist);CHKERRQ(ierr); 1249b54502bSHong Zhang ierr = PetscOptionsList("-pc_cholesky_mat_ordering_type","Reordering to reduce nonzeros in Cholesky","PCCholeskySetMatOrdering",ordlist,lu->ordering,tname,256,&flg);CHKERRQ(ierr); 1259b54502bSHong Zhang if (flg) { 1269b54502bSHong Zhang ierr = PCCholeskySetMatOrdering(pc,tname);CHKERRQ(ierr); 1279b54502bSHong Zhang } 1280a29876aSHong Zhang ierr = PetscOptionsName("-pc_factor_shiftnonzero","Shift added to diagonal","PCFactorSetShiftNonzero",&flg);CHKERRQ(ierr); 1299b54502bSHong Zhang if (flg) { 130afaefe49SHong Zhang ierr = PCFactorSetShiftNonzero(pc,(PetscReal) PETSC_DECIDE);CHKERRQ(ierr); 1319b54502bSHong Zhang } 1320a29876aSHong Zhang ierr = PetscOptionsReal("-pc_factor_shiftnonzero","Shift added to diagonal","PCFactorSetShiftNonzero",lu->info.shiftnz,&lu->info.shiftnz,0);CHKERRQ(ierr); 1330a29876aSHong Zhang ierr = PetscOptionsName("-pc_factor_shiftpd","Manteuffel shift applied to diagonal","PCFactorSetShiftPd",&flg);CHKERRQ(ierr); 1349b54502bSHong Zhang if (flg) { 135afaefe49SHong Zhang ierr = PCFactorSetShiftPd(pc,PETSC_TRUE);CHKERRQ(ierr); 1369b54502bSHong Zhang } 137ee45ca4aSHong Zhang ierr = PetscOptionsReal("-pc_factor_zeropivot","Pivot is considered zero if less than","PCFactorSetZeroPivot",lu->info.zeropivot,&lu->info.zeropivot,0);CHKERRQ(ierr); 1389b54502bSHong Zhang 1399b54502bSHong Zhang ierr = PetscOptionsTail();CHKERRQ(ierr); 1409b54502bSHong Zhang PetscFunctionReturn(0); 1419b54502bSHong Zhang } 1429b54502bSHong Zhang 1439b54502bSHong Zhang #undef __FUNCT__ 1449b54502bSHong Zhang #define __FUNCT__ "PCView_Cholesky" 1459b54502bSHong Zhang static PetscErrorCode PCView_Cholesky(PC pc,PetscViewer viewer) 1469b54502bSHong Zhang { 1479b54502bSHong Zhang PC_Cholesky *lu = (PC_Cholesky*)pc->data; 1489b54502bSHong Zhang PetscErrorCode ierr; 1499b54502bSHong Zhang PetscTruth iascii,isstring; 1509b54502bSHong Zhang 1519b54502bSHong Zhang PetscFunctionBegin; 1529b54502bSHong Zhang ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 1539b54502bSHong Zhang ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_STRING,&isstring);CHKERRQ(ierr); 1549b54502bSHong Zhang if (iascii) { 1559b54502bSHong Zhang MatInfo info; 1569b54502bSHong Zhang 1579b54502bSHong Zhang if (lu->inplace) {ierr = PetscViewerASCIIPrintf(viewer," Cholesky: in-place factorization\n");CHKERRQ(ierr);} 1589b54502bSHong Zhang else {ierr = PetscViewerASCIIPrintf(viewer," Cholesky: out-of-place factorization\n");CHKERRQ(ierr);} 1599b54502bSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," matrix ordering: %s\n",lu->ordering);CHKERRQ(ierr); 1609b54502bSHong Zhang if (lu->fact) { 1619b54502bSHong Zhang ierr = MatGetInfo(lu->fact,MAT_LOCAL,&info);CHKERRQ(ierr); 1629b54502bSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Cholesky nonzeros %g\n",info.nz_used);CHKERRQ(ierr); 1639b54502bSHong Zhang ierr = PetscViewerPushFormat(viewer,PETSC_VIEWER_ASCII_FACTOR_INFO);CHKERRQ(ierr); 1649b54502bSHong Zhang ierr = MatView(lu->fact,viewer);CHKERRQ(ierr); 1659b54502bSHong Zhang ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr); 1669b54502bSHong Zhang } 1679b54502bSHong Zhang if (lu->reusefill) {ierr = PetscViewerASCIIPrintf(viewer," Reusing fill from past factorization\n");CHKERRQ(ierr);} 1689b54502bSHong Zhang if (lu->reuseordering) {ierr = PetscViewerASCIIPrintf(viewer," Reusing reordering from past factorization\n");CHKERRQ(ierr);} 1699b54502bSHong Zhang } else if (isstring) { 1709b54502bSHong Zhang ierr = PetscViewerStringSPrintf(viewer," order=%s",lu->ordering);CHKERRQ(ierr);CHKERRQ(ierr); 1719b54502bSHong Zhang } else { 1729b54502bSHong Zhang SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported for PCCholesky",((PetscObject)viewer)->type_name); 1739b54502bSHong Zhang } 1749b54502bSHong Zhang PetscFunctionReturn(0); 1759b54502bSHong Zhang } 1769b54502bSHong Zhang 1779b54502bSHong Zhang #undef __FUNCT__ 1789b54502bSHong Zhang #define __FUNCT__ "PCGetFactoredMatrix_Cholesky" 1799b54502bSHong Zhang static PetscErrorCode PCGetFactoredMatrix_Cholesky(PC pc,Mat *mat) 1809b54502bSHong Zhang { 1819b54502bSHong Zhang PC_Cholesky *dir = (PC_Cholesky*)pc->data; 1829b54502bSHong Zhang 1839b54502bSHong Zhang PetscFunctionBegin; 1849b54502bSHong Zhang if (!dir->fact) SETERRQ(PETSC_ERR_ORDER,"Matrix not yet factored; call after KSPSetUp() or PCSetUp()"); 1859b54502bSHong Zhang *mat = dir->fact; 1869b54502bSHong Zhang PetscFunctionReturn(0); 1879b54502bSHong Zhang } 1889b54502bSHong Zhang 1899b54502bSHong Zhang #undef __FUNCT__ 1909b54502bSHong Zhang #define __FUNCT__ "PCSetUp_Cholesky" 1919b54502bSHong Zhang static PetscErrorCode PCSetUp_Cholesky(PC pc) 1929b54502bSHong Zhang { 1939b54502bSHong Zhang PetscErrorCode ierr; 1949b54502bSHong Zhang PetscTruth flg; 1959b54502bSHong Zhang PC_Cholesky *dir = (PC_Cholesky*)pc->data; 1969b54502bSHong Zhang 1979b54502bSHong Zhang PetscFunctionBegin; 1989b54502bSHong Zhang if (dir->reusefill && pc->setupcalled) dir->info.fill = dir->actualfill; 1999b54502bSHong Zhang 2009b54502bSHong Zhang if (dir->inplace) { 2019b54502bSHong Zhang if (dir->row && dir->col && (dir->row != dir->col)) { 2029b54502bSHong Zhang ierr = ISDestroy(dir->row);CHKERRQ(ierr); 2039b54502bSHong Zhang dir->row = 0; 2049b54502bSHong Zhang } 2059b54502bSHong Zhang if (dir->col) { 2069b54502bSHong Zhang ierr = ISDestroy(dir->col);CHKERRQ(ierr); 2079b54502bSHong Zhang dir->col = 0; 2089b54502bSHong Zhang } 2099b54502bSHong Zhang ierr = MatGetOrdering(pc->pmat,dir->ordering,&dir->row,&dir->col);CHKERRQ(ierr); 2109b54502bSHong Zhang if (dir->col && (dir->row != dir->col)) { /* only use row ordering for SBAIJ */ 2119b54502bSHong Zhang ierr = ISDestroy(dir->col);CHKERRQ(ierr); 2129b54502bSHong Zhang dir->col=0; 2139b54502bSHong Zhang } 2149b54502bSHong Zhang if (dir->row) {PetscLogObjectParent(pc,dir->row);} 2159b54502bSHong Zhang ierr = MatCholeskyFactor(pc->pmat,dir->row,&dir->info);CHKERRQ(ierr); 2169b54502bSHong Zhang dir->fact = pc->pmat; 2179b54502bSHong Zhang } else { 2189b54502bSHong Zhang MatInfo info; 2199b54502bSHong Zhang if (!pc->setupcalled) { 2209b54502bSHong Zhang ierr = MatGetOrdering(pc->pmat,dir->ordering,&dir->row,&dir->col);CHKERRQ(ierr); 2219b54502bSHong Zhang if (dir->col && (dir->row != dir->col)) { /* only use row ordering for SBAIJ */ 2229b54502bSHong Zhang ierr = ISDestroy(dir->col);CHKERRQ(ierr); 2239b54502bSHong Zhang dir->col=0; 2249b54502bSHong Zhang } 2259b54502bSHong Zhang ierr = PetscOptionsHasName(pc->prefix,"-pc_cholesky_nonzeros_along_diagonal",&flg);CHKERRQ(ierr); 2269b54502bSHong Zhang if (flg) { 2279b54502bSHong Zhang PetscReal tol = 1.e-10; 2289b54502bSHong Zhang ierr = PetscOptionsGetReal(pc->prefix,"-pc_cholesky_nonzeros_along_diagonal",&tol,PETSC_NULL);CHKERRQ(ierr); 2299b54502bSHong Zhang ierr = MatReorderForNonzeroDiagonal(pc->pmat,tol,dir->row,dir->row);CHKERRQ(ierr); 2309b54502bSHong Zhang } 2319b54502bSHong Zhang if (dir->row) {PetscLogObjectParent(pc,dir->row);} 2329b54502bSHong Zhang ierr = MatCholeskyFactorSymbolic(pc->pmat,dir->row,&dir->info,&dir->fact);CHKERRQ(ierr); 2339b54502bSHong Zhang ierr = MatGetInfo(dir->fact,MAT_LOCAL,&info);CHKERRQ(ierr); 2349b54502bSHong Zhang dir->actualfill = info.fill_ratio_needed; 235*52e6d16bSBarry Smith ierr = PetscLogObjectParent(pc,dir->fact);CHKERRQ(ierr); 2369b54502bSHong Zhang } else if (pc->flag != SAME_NONZERO_PATTERN) { 2379b54502bSHong Zhang if (!dir->reuseordering) { 2389b54502bSHong Zhang if (dir->row && dir->col && (dir->row != dir->col)) { 2399b54502bSHong Zhang ierr = ISDestroy(dir->row);CHKERRQ(ierr); 2409b54502bSHong Zhang dir->row = 0; 2419b54502bSHong Zhang } 2429b54502bSHong Zhang if (dir->col) { 2439b54502bSHong Zhang ierr = ISDestroy(dir->col);CHKERRQ(ierr); 2449b54502bSHong Zhang dir->col =0; 2459b54502bSHong Zhang } 2469b54502bSHong Zhang ierr = MatGetOrdering(pc->pmat,dir->ordering,&dir->row,&dir->col);CHKERRQ(ierr); 2479b54502bSHong Zhang if (dir->col && (dir->row != dir->col)) { /* only use row ordering for SBAIJ */ 2489b54502bSHong Zhang ierr = ISDestroy(dir->col);CHKERRQ(ierr); 2499b54502bSHong Zhang dir->col=0; 2509b54502bSHong Zhang } 2519b54502bSHong Zhang ierr = PetscOptionsHasName(pc->prefix,"-pc_cholesky_nonzeros_along_diagonal",&flg);CHKERRQ(ierr); 2529b54502bSHong Zhang if (flg) { 2539b54502bSHong Zhang PetscReal tol = 1.e-10; 2549b54502bSHong Zhang ierr = PetscOptionsGetReal(pc->prefix,"-pc_cholesky_nonzeros_along_diagonal",&tol,PETSC_NULL);CHKERRQ(ierr); 2559b54502bSHong Zhang ierr = MatReorderForNonzeroDiagonal(pc->pmat,tol,dir->row,dir->row);CHKERRQ(ierr); 2569b54502bSHong Zhang } 2579b54502bSHong Zhang if (dir->row) {PetscLogObjectParent(pc,dir->row);} 2589b54502bSHong Zhang } 2599b54502bSHong Zhang ierr = MatDestroy(dir->fact);CHKERRQ(ierr); 2609b54502bSHong Zhang ierr = MatCholeskyFactorSymbolic(pc->pmat,dir->row,&dir->info,&dir->fact);CHKERRQ(ierr); 2619b54502bSHong Zhang ierr = MatGetInfo(dir->fact,MAT_LOCAL,&info);CHKERRQ(ierr); 2629b54502bSHong Zhang dir->actualfill = info.fill_ratio_needed; 263*52e6d16bSBarry Smith ierr = PetscLogObjectParent(pc,dir->fact);CHKERRQ(ierr); 2649b54502bSHong Zhang } 2659b54502bSHong Zhang ierr = MatCholeskyFactorNumeric(pc->pmat,&dir->info,&dir->fact);CHKERRQ(ierr); 2669b54502bSHong Zhang } 2679b54502bSHong Zhang PetscFunctionReturn(0); 2689b54502bSHong Zhang } 2699b54502bSHong Zhang 2709b54502bSHong Zhang #undef __FUNCT__ 2719b54502bSHong Zhang #define __FUNCT__ "PCDestroy_Cholesky" 2729b54502bSHong Zhang static PetscErrorCode PCDestroy_Cholesky(PC pc) 2739b54502bSHong Zhang { 2749b54502bSHong Zhang PC_Cholesky *dir = (PC_Cholesky*)pc->data; 2759b54502bSHong Zhang PetscErrorCode ierr; 2769b54502bSHong Zhang 2779b54502bSHong Zhang PetscFunctionBegin; 2789b54502bSHong Zhang if (!dir->inplace && dir->fact) {ierr = MatDestroy(dir->fact);CHKERRQ(ierr);} 2799b54502bSHong Zhang if (dir->row) {ierr = ISDestroy(dir->row);CHKERRQ(ierr);} 2809b54502bSHong Zhang if (dir->col) {ierr = ISDestroy(dir->col);CHKERRQ(ierr);} 2819b54502bSHong Zhang ierr = PetscStrfree(dir->ordering);CHKERRQ(ierr); 2829b54502bSHong Zhang ierr = PetscFree(dir);CHKERRQ(ierr); 2839b54502bSHong Zhang PetscFunctionReturn(0); 2849b54502bSHong Zhang } 2859b54502bSHong Zhang 2869b54502bSHong Zhang #undef __FUNCT__ 2879b54502bSHong Zhang #define __FUNCT__ "PCApply_Cholesky" 2889b54502bSHong Zhang static PetscErrorCode PCApply_Cholesky(PC pc,Vec x,Vec y) 2899b54502bSHong Zhang { 2909b54502bSHong Zhang PC_Cholesky *dir = (PC_Cholesky*)pc->data; 2919b54502bSHong Zhang PetscErrorCode ierr; 2929b54502bSHong Zhang 2939b54502bSHong Zhang PetscFunctionBegin; 2949b54502bSHong Zhang if (dir->inplace) {ierr = MatSolve(pc->pmat,x,y);CHKERRQ(ierr);} 2959b54502bSHong Zhang else {ierr = MatSolve(dir->fact,x,y);CHKERRQ(ierr);} 2969b54502bSHong Zhang PetscFunctionReturn(0); 2979b54502bSHong Zhang } 2989b54502bSHong Zhang 2999b54502bSHong Zhang #undef __FUNCT__ 3009b54502bSHong Zhang #define __FUNCT__ "PCApplyTranspose_Cholesky" 3019b54502bSHong Zhang static PetscErrorCode PCApplyTranspose_Cholesky(PC pc,Vec x,Vec y) 3029b54502bSHong Zhang { 3039b54502bSHong Zhang PC_Cholesky *dir = (PC_Cholesky*)pc->data; 3049b54502bSHong Zhang PetscErrorCode ierr; 3059b54502bSHong Zhang 3069b54502bSHong Zhang PetscFunctionBegin; 3079b54502bSHong Zhang if (dir->inplace) {ierr = MatSolveTranspose(pc->pmat,x,y);CHKERRQ(ierr);} 3089b54502bSHong Zhang else {ierr = MatSolveTranspose(dir->fact,x,y);CHKERRQ(ierr);} 3099b54502bSHong Zhang PetscFunctionReturn(0); 3109b54502bSHong Zhang } 3119b54502bSHong Zhang 3129b54502bSHong Zhang /* -----------------------------------------------------------------------------------*/ 3139b54502bSHong Zhang 3149b54502bSHong Zhang EXTERN_C_BEGIN 3159b54502bSHong Zhang #undef __FUNCT__ 3169b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetFill_Cholesky" 3179b54502bSHong Zhang PetscErrorCode PCCholeskySetFill_Cholesky(PC pc,PetscReal fill) 3189b54502bSHong Zhang { 3199b54502bSHong Zhang PC_Cholesky *dir; 3209b54502bSHong Zhang 3219b54502bSHong Zhang PetscFunctionBegin; 3229b54502bSHong Zhang dir = (PC_Cholesky*)pc->data; 3239b54502bSHong Zhang dir->info.fill = fill; 3249b54502bSHong Zhang PetscFunctionReturn(0); 3259b54502bSHong Zhang } 3269b54502bSHong Zhang EXTERN_C_END 3279b54502bSHong Zhang 3289b54502bSHong Zhang EXTERN_C_BEGIN 3299b54502bSHong Zhang #undef __FUNCT__ 3309b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetUseInPlace_Cholesky" 3319b54502bSHong Zhang PetscErrorCode PCCholeskySetUseInPlace_Cholesky(PC pc) 3329b54502bSHong Zhang { 3339b54502bSHong Zhang PC_Cholesky *dir; 3349b54502bSHong Zhang 3359b54502bSHong Zhang PetscFunctionBegin; 3369b54502bSHong Zhang dir = (PC_Cholesky*)pc->data; 3379b54502bSHong Zhang dir->inplace = PETSC_TRUE; 3389b54502bSHong Zhang PetscFunctionReturn(0); 3399b54502bSHong Zhang } 3409b54502bSHong Zhang EXTERN_C_END 3419b54502bSHong Zhang 3429b54502bSHong Zhang EXTERN_C_BEGIN 3439b54502bSHong Zhang #undef __FUNCT__ 3449b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetMatOrdering_Cholesky" 3459b54502bSHong Zhang PetscErrorCode PCCholeskySetMatOrdering_Cholesky(PC pc,MatOrderingType ordering) 3469b54502bSHong Zhang { 3479b54502bSHong Zhang PC_Cholesky *dir = (PC_Cholesky*)pc->data; 3489b54502bSHong Zhang PetscErrorCode ierr; 3499b54502bSHong Zhang 3509b54502bSHong Zhang PetscFunctionBegin; 3519b54502bSHong Zhang ierr = PetscStrfree(dir->ordering);CHKERRQ(ierr); 3529b54502bSHong Zhang ierr = PetscStrallocpy(ordering,&dir->ordering);CHKERRQ(ierr); 3539b54502bSHong Zhang PetscFunctionReturn(0); 3549b54502bSHong Zhang } 3559b54502bSHong Zhang EXTERN_C_END 3569b54502bSHong Zhang 3579b54502bSHong Zhang /* -----------------------------------------------------------------------------------*/ 3589b54502bSHong Zhang 3599b54502bSHong Zhang #undef __FUNCT__ 3609b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetReuseOrdering" 3619b54502bSHong Zhang /*@ 3629b54502bSHong Zhang PCCholeskySetReuseOrdering - When similar matrices are factored, this 3639b54502bSHong Zhang causes the ordering computed in the first factor to be used for all 3649b54502bSHong Zhang following factors. 3659b54502bSHong Zhang 3669b54502bSHong Zhang Collective on PC 3679b54502bSHong Zhang 3689b54502bSHong Zhang Input Parameters: 3699b54502bSHong Zhang + pc - the preconditioner context 3709b54502bSHong Zhang - flag - PETSC_TRUE to reuse else PETSC_FALSE 3719b54502bSHong Zhang 3729b54502bSHong Zhang Options Database Key: 3739b54502bSHong Zhang . -pc_cholesky_reuse_ordering - Activate PCCholeskySetReuseOrdering() 3749b54502bSHong Zhang 3759b54502bSHong Zhang Level: intermediate 3769b54502bSHong Zhang 3779b54502bSHong Zhang .keywords: PC, levels, reordering, factorization, incomplete, LU 3789b54502bSHong Zhang 3799b54502bSHong Zhang .seealso: PCCholeskySetReuseFill(), PCICholeskySetReuseOrdering(), PCICholeskyDTSetReuseFill() 3809b54502bSHong Zhang @*/ 3819b54502bSHong Zhang PetscErrorCode PCCholeskySetReuseOrdering(PC pc,PetscTruth flag) 3829b54502bSHong Zhang { 3839b54502bSHong Zhang PetscErrorCode ierr,(*f)(PC,PetscTruth); 3849b54502bSHong Zhang 3859b54502bSHong Zhang PetscFunctionBegin; 3869b54502bSHong Zhang PetscValidHeaderSpecific(pc,PC_COOKIE,1); 3879b54502bSHong Zhang ierr = PetscObjectQueryFunction((PetscObject)pc,"PCCholeskySetReuseOrdering_C",(void (**)(void))&f);CHKERRQ(ierr); 3889b54502bSHong Zhang if (f) { 3899b54502bSHong Zhang ierr = (*f)(pc,flag);CHKERRQ(ierr); 3909b54502bSHong Zhang } 3919b54502bSHong Zhang PetscFunctionReturn(0); 3929b54502bSHong Zhang } 3939b54502bSHong Zhang 3949b54502bSHong Zhang #undef __FUNCT__ 3959b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetReuseFill" 3969b54502bSHong Zhang /*@ 3979b54502bSHong Zhang PCCholeskySetReuseFill - When matrices with same nonzero structure are Cholesky factored, 3989b54502bSHong Zhang this causes later ones to use the fill computed in the initial factorization. 3999b54502bSHong Zhang 4009b54502bSHong Zhang Collective on PC 4019b54502bSHong Zhang 4029b54502bSHong Zhang Input Parameters: 4039b54502bSHong Zhang + pc - the preconditioner context 4049b54502bSHong Zhang - flag - PETSC_TRUE to reuse else PETSC_FALSE 4059b54502bSHong Zhang 4069b54502bSHong Zhang Options Database Key: 4079b54502bSHong Zhang . -pc_cholesky_reuse_fill - Activates PCCholeskySetReuseFill() 4089b54502bSHong Zhang 4099b54502bSHong Zhang Level: intermediate 4109b54502bSHong Zhang 4119b54502bSHong Zhang .keywords: PC, levels, reordering, factorization, incomplete, Cholesky 4129b54502bSHong Zhang 4139b54502bSHong Zhang .seealso: PCICholeskySetReuseOrdering(), PCCholeskySetReuseOrdering(), PCICholeskyDTSetReuseFill() 4149b54502bSHong Zhang @*/ 4159b54502bSHong Zhang PetscErrorCode PCCholeskySetReuseFill(PC pc,PetscTruth flag) 4169b54502bSHong Zhang { 4179b54502bSHong Zhang PetscErrorCode ierr,(*f)(PC,PetscTruth); 4189b54502bSHong Zhang 4199b54502bSHong Zhang PetscFunctionBegin; 4209b54502bSHong Zhang PetscValidHeaderSpecific(pc,PC_COOKIE,2); 4219b54502bSHong Zhang ierr = PetscObjectQueryFunction((PetscObject)pc,"PCCholeskySetReuseFill_C",(void (**)(void))&f);CHKERRQ(ierr); 4229b54502bSHong Zhang if (f) { 4239b54502bSHong Zhang ierr = (*f)(pc,flag);CHKERRQ(ierr); 4249b54502bSHong Zhang } 4259b54502bSHong Zhang PetscFunctionReturn(0); 4269b54502bSHong Zhang } 4279b54502bSHong Zhang 4289b54502bSHong Zhang #undef __FUNCT__ 4299b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetFill" 4309b54502bSHong Zhang /*@ 4319b54502bSHong Zhang PCCholeskySetFill - Indicates the amount of fill you expect in the factored matrix, 4329b54502bSHong Zhang fill = number nonzeros in factor/number nonzeros in original matrix. 4339b54502bSHong Zhang 4349b54502bSHong Zhang Collective on PC 4359b54502bSHong Zhang 4369b54502bSHong Zhang Input Parameters: 4379b54502bSHong Zhang + pc - the preconditioner context 4389b54502bSHong Zhang - fill - amount of expected fill 4399b54502bSHong Zhang 4409b54502bSHong Zhang Options Database Key: 4419b54502bSHong Zhang . -pc_cholesky_fill <fill> - Sets fill amount 4429b54502bSHong Zhang 4439b54502bSHong Zhang Level: intermediate 4449b54502bSHong Zhang 4459b54502bSHong Zhang Note: 4469b54502bSHong Zhang For sparse matrix factorizations it is difficult to predict how much 4479b54502bSHong Zhang fill to expect. By running with the option -log_info PETSc will print the 4489b54502bSHong Zhang actual amount of fill used; allowing you to set the value accurately for 4499b54502bSHong Zhang future runs. Default PETSc uses a value of 5.0 4509b54502bSHong Zhang 4519b54502bSHong Zhang .keywords: PC, set, factorization, direct, fill 4529b54502bSHong Zhang 4539b54502bSHong Zhang .seealso: PCILUSetFill() 4549b54502bSHong Zhang @*/ 4559b54502bSHong Zhang PetscErrorCode PCCholeskySetFill(PC pc,PetscReal fill) 4569b54502bSHong Zhang { 4579b54502bSHong Zhang PetscErrorCode ierr,(*f)(PC,PetscReal); 4589b54502bSHong Zhang 4599b54502bSHong Zhang PetscFunctionBegin; 4609b54502bSHong Zhang PetscValidHeaderSpecific(pc,PC_COOKIE,1); 4619b54502bSHong Zhang if (fill < 1.0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Fill factor cannot be less then 1.0"); 4629b54502bSHong Zhang ierr = PetscObjectQueryFunction((PetscObject)pc,"PCCholeskySetFill_C",(void (**)(void))&f);CHKERRQ(ierr); 4639b54502bSHong Zhang if (f) { 4649b54502bSHong Zhang ierr = (*f)(pc,fill);CHKERRQ(ierr); 4659b54502bSHong Zhang } 4669b54502bSHong Zhang PetscFunctionReturn(0); 4679b54502bSHong Zhang } 4689b54502bSHong Zhang 4699b54502bSHong Zhang #undef __FUNCT__ 4709b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetUseInPlace" 4719b54502bSHong Zhang /*@ 4729b54502bSHong Zhang PCCholeskySetUseInPlace - Tells the system to do an in-place factorization. 4739b54502bSHong Zhang For dense matrices, this enables the solution of much larger problems. 4749b54502bSHong Zhang For sparse matrices the factorization cannot be done truly in-place 4759b54502bSHong Zhang so this does not save memory during the factorization, but after the matrix 4769b54502bSHong Zhang is factored, the original unfactored matrix is freed, thus recovering that 4779b54502bSHong Zhang space. 4789b54502bSHong Zhang 4799b54502bSHong Zhang Collective on PC 4809b54502bSHong Zhang 4819b54502bSHong Zhang Input Parameters: 4829b54502bSHong Zhang . pc - the preconditioner context 4839b54502bSHong Zhang 4849b54502bSHong Zhang Options Database Key: 4859b54502bSHong Zhang . -pc_cholesky_in_place - Activates in-place factorization 4869b54502bSHong Zhang 4879b54502bSHong Zhang Notes: 4889b54502bSHong Zhang PCCholeskySetUseInplace() can only be used with the KSP method KSPPREONLY or when 4899b54502bSHong Zhang a different matrix is provided for the multiply and the preconditioner in 4909b54502bSHong Zhang a call to KSPSetOperators(). 4919b54502bSHong Zhang This is because the Krylov space methods require an application of the 4929b54502bSHong Zhang matrix multiplication, which is not possible here because the matrix has 4939b54502bSHong Zhang been factored in-place, replacing the original matrix. 4949b54502bSHong Zhang 4959b54502bSHong Zhang Level: intermediate 4969b54502bSHong Zhang 4979b54502bSHong Zhang .keywords: PC, set, factorization, direct, inplace, in-place, Cholesky 4989b54502bSHong Zhang 4999b54502bSHong Zhang .seealso: PCICholeskySetUseInPlace() 5009b54502bSHong Zhang @*/ 5019b54502bSHong Zhang PetscErrorCode PCCholeskySetUseInPlace(PC pc) 5029b54502bSHong Zhang { 5039b54502bSHong Zhang PetscErrorCode ierr,(*f)(PC); 5049b54502bSHong Zhang 5059b54502bSHong Zhang PetscFunctionBegin; 5069b54502bSHong Zhang PetscValidHeaderSpecific(pc,PC_COOKIE,1); 5079b54502bSHong Zhang ierr = PetscObjectQueryFunction((PetscObject)pc,"PCCholeskySetUseInPlace_C",(void (**)(void))&f);CHKERRQ(ierr); 5089b54502bSHong Zhang if (f) { 5099b54502bSHong Zhang ierr = (*f)(pc);CHKERRQ(ierr); 5109b54502bSHong Zhang } 5119b54502bSHong Zhang PetscFunctionReturn(0); 5129b54502bSHong Zhang } 5139b54502bSHong Zhang 5149b54502bSHong Zhang #undef __FUNCT__ 5159b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetMatOrdering" 5169b54502bSHong Zhang /*@ 5179b54502bSHong Zhang PCCholeskySetMatOrdering - Sets the ordering routine (to reduce fill) to 5189b54502bSHong Zhang be used it the Cholesky factorization. 5199b54502bSHong Zhang 5209b54502bSHong Zhang Collective on PC 5219b54502bSHong Zhang 5229b54502bSHong Zhang Input Parameters: 5239b54502bSHong Zhang + pc - the preconditioner context 5249b54502bSHong Zhang - ordering - the matrix ordering name, for example, MATORDERING_ND or MATORDERING_RCM 5259b54502bSHong Zhang 5269b54502bSHong Zhang Options Database Key: 5279b54502bSHong Zhang . -pc_cholesky_mat_ordering_type <nd,rcm,...> - Sets ordering routine 5289b54502bSHong Zhang 5299b54502bSHong Zhang Level: intermediate 5309b54502bSHong Zhang 5319b54502bSHong Zhang .seealso: PCICholeskySetMatOrdering() 5329b54502bSHong Zhang @*/ 5339b54502bSHong Zhang PetscErrorCode PCCholeskySetMatOrdering(PC pc,MatOrderingType ordering) 5349b54502bSHong Zhang { 5359b54502bSHong Zhang PetscErrorCode ierr,(*f)(PC,MatOrderingType); 5369b54502bSHong Zhang 5379b54502bSHong Zhang PetscFunctionBegin; 5389b54502bSHong Zhang ierr = PetscObjectQueryFunction((PetscObject)pc,"PCCholeskySetMatOrdering_C",(void (**)(void))&f);CHKERRQ(ierr); 5399b54502bSHong Zhang if (f) { 5409b54502bSHong Zhang ierr = (*f)(pc,ordering);CHKERRQ(ierr); 5419b54502bSHong Zhang } 5429b54502bSHong Zhang PetscFunctionReturn(0); 5439b54502bSHong Zhang } 5449b54502bSHong Zhang 5459b54502bSHong Zhang /*MC 5469b54502bSHong Zhang PCCholesky - Uses a direct solver, based on Cholesky factorization, as a preconditioner 5479b54502bSHong Zhang 5489b54502bSHong Zhang Options Database Keys: 5499b54502bSHong Zhang + -pc_cholesky_reuse_ordering - Activate PCLUSetReuseOrdering() 5509b54502bSHong Zhang . -pc_cholesky_reuse_fill - Activates PCLUSetReuseFill() 5519b54502bSHong Zhang . -pc_cholesky_fill <fill> - Sets fill amount 5529b54502bSHong Zhang . -pc_cholesky_in_place - Activates in-place factorization 5539b54502bSHong Zhang - -pc_cholesky_mat_ordering_type <nd,rcm,...> - Sets ordering routine 5549b54502bSHong Zhang 5559b54502bSHong Zhang Notes: Not all options work for all matrix formats 5569b54502bSHong Zhang 5579b54502bSHong Zhang Level: beginner 5589b54502bSHong Zhang 5599b54502bSHong Zhang Concepts: Cholesky factorization, direct solver 5609b54502bSHong Zhang 5619b54502bSHong Zhang Notes: Usually this will compute an "exact" solution in one iteration and does 5629b54502bSHong Zhang not need a Krylov method (i.e. you can use -ksp_type preonly, or 5639b54502bSHong Zhang KSPSetType(ksp,KSPPREONLY) for the Krylov method 5649b54502bSHong Zhang 5659b54502bSHong Zhang .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, 5669b54502bSHong Zhang PCILU, PCLU, PCICC, PCCholeskySetReuseOrdering(), PCCholeskySetReuseFill(), PCGetFactoredMatrix(), 567ee45ca4aSHong Zhang PCCholeskySetFill(), PCFactorSetShiftNonzero(), PCFactorSetShiftPd(), 5689b54502bSHong Zhang PCCholeskySetUseInPlace(), PCCholeskySetMatOrdering() 5699b54502bSHong Zhang 5709b54502bSHong Zhang M*/ 5719b54502bSHong Zhang 5729b54502bSHong Zhang EXTERN_C_BEGIN 5739b54502bSHong Zhang #undef __FUNCT__ 5749b54502bSHong Zhang #define __FUNCT__ "PCCreate_Cholesky" 5759b54502bSHong Zhang PetscErrorCode PCCreate_Cholesky(PC pc) 5769b54502bSHong Zhang { 5779b54502bSHong Zhang PetscErrorCode ierr; 5789b54502bSHong Zhang PC_Cholesky *dir; 5799b54502bSHong Zhang 5809b54502bSHong Zhang PetscFunctionBegin; 5819b54502bSHong Zhang ierr = PetscNew(PC_Cholesky,&dir);CHKERRQ(ierr); 582*52e6d16bSBarry Smith ierr = PetscLogObjectMemory(pc,sizeof(PC_Cholesky));CHKERRQ(ierr); 5839b54502bSHong Zhang 5849b54502bSHong Zhang dir->fact = 0; 5859b54502bSHong Zhang dir->inplace = PETSC_FALSE; 5869b54502bSHong Zhang ierr = MatFactorInfoInitialize(&dir->info);CHKERRQ(ierr); 5879b54502bSHong Zhang dir->info.fill = 5.0; 5880a29876aSHong Zhang dir->info.shiftnz = 0.0; 5890a29876aSHong Zhang dir->info.shiftpd = PETSC_FALSE; 5909b54502bSHong Zhang dir->info.shift_fraction = 0.0; 5919b54502bSHong Zhang dir->info.pivotinblocks = 1.0; 5929b54502bSHong Zhang dir->col = 0; 5939b54502bSHong Zhang dir->row = 0; 5949b54502bSHong Zhang ierr = PetscStrallocpy(MATORDERING_NATURAL,&dir->ordering);CHKERRQ(ierr); 5959b54502bSHong Zhang dir->reusefill = PETSC_FALSE; 5969b54502bSHong Zhang dir->reuseordering = PETSC_FALSE; 5979b54502bSHong Zhang pc->data = (void*)dir; 5989b54502bSHong Zhang 5999b54502bSHong Zhang pc->ops->destroy = PCDestroy_Cholesky; 6009b54502bSHong Zhang pc->ops->apply = PCApply_Cholesky; 6019b54502bSHong Zhang pc->ops->applytranspose = PCApplyTranspose_Cholesky; 6029b54502bSHong Zhang pc->ops->setup = PCSetUp_Cholesky; 6039b54502bSHong Zhang pc->ops->setfromoptions = PCSetFromOptions_Cholesky; 6049b54502bSHong Zhang pc->ops->view = PCView_Cholesky; 6059b54502bSHong Zhang pc->ops->applyrichardson = 0; 6069b54502bSHong Zhang pc->ops->getfactoredmatrix = PCGetFactoredMatrix_Cholesky; 6079b54502bSHong Zhang 608afaefe49SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetZeroPivot_C","PCFactorSetZeroPivot_Cholesky", 609afaefe49SHong Zhang PCFactorSetZeroPivot_Cholesky);CHKERRQ(ierr); 610afaefe49SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetShiftNonzero_C","PCFactorSetShiftNonzero_Cholesky", 611afaefe49SHong Zhang PCFactorSetShiftNonzero_Cholesky);CHKERRQ(ierr); 612afaefe49SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetShiftPd_C","PCFactorSetShiftPd_Cholesky", 613afaefe49SHong Zhang PCFactorSetShiftPd_Cholesky);CHKERRQ(ierr); 614afaefe49SHong Zhang 6159b54502bSHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCCholeskySetFill_C","PCCholeskySetFill_Cholesky", 6169b54502bSHong Zhang PCCholeskySetFill_Cholesky);CHKERRQ(ierr); 6179b54502bSHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCCholeskySetUseInPlace_C","PCCholeskySetUseInPlace_Cholesky", 6189b54502bSHong Zhang PCCholeskySetUseInPlace_Cholesky);CHKERRQ(ierr); 6199b54502bSHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCCholeskySetMatOrdering_C","PCCholeskySetMatOrdering_Cholesky", 6209b54502bSHong Zhang PCCholeskySetMatOrdering_Cholesky);CHKERRQ(ierr); 6219b54502bSHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCCholeskySetReuseOrdering_C","PCCholeskySetReuseOrdering_Cholesky", 6229b54502bSHong Zhang PCCholeskySetReuseOrdering_Cholesky);CHKERRQ(ierr); 6239b54502bSHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCCholeskySetReuseFill_C","PCCholeskySetReuseFill_Cholesky", 6249b54502bSHong Zhang PCCholeskySetReuseFill_Cholesky);CHKERRQ(ierr); 6259b54502bSHong Zhang PetscFunctionReturn(0); 6269b54502bSHong Zhang } 6279b54502bSHong Zhang EXTERN_C_END 628