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