xref: /petsc/src/ksp/pc/impls/factor/cholesky/cholesky.c (revision 55ba2a515af5dbbcf670f4e560eb9724ec139bf1)
1dba47a55SKris 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 */
86356e834SBarry Smith #include "private/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"
24dba47a55SKris 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"
38dba47a55SKris 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"
56dba47a55SKris 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;
62fbf22428SSatish Balay   if (shift) {
63fbf22428SSatish Balay     dir->info.shift_fraction = 0.0;
64fbf22428SSatish Balay     dir->info.shiftpd = 1.0;
65fbf22428SSatish Balay   } else {
66fbf22428SSatish Balay     dir->info.shiftpd = 0.0;
67fbf22428SSatish Balay   }
68afaefe49SHong Zhang   PetscFunctionReturn(0);
69afaefe49SHong Zhang }
70afaefe49SHong Zhang EXTERN_C_END
71afaefe49SHong Zhang 
72afaefe49SHong Zhang EXTERN_C_BEGIN
73afaefe49SHong Zhang #undef __FUNCT__
749b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetReuseOrdering_Cholesky"
75dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCholeskySetReuseOrdering_Cholesky(PC pc,PetscTruth flag)
769b54502bSHong Zhang {
779b54502bSHong Zhang   PC_Cholesky *lu;
789b54502bSHong Zhang 
799b54502bSHong Zhang   PetscFunctionBegin;
809b54502bSHong Zhang   lu               = (PC_Cholesky*)pc->data;
819b54502bSHong Zhang   lu->reuseordering = flag;
829b54502bSHong Zhang   PetscFunctionReturn(0);
839b54502bSHong Zhang }
849b54502bSHong Zhang EXTERN_C_END
859b54502bSHong Zhang 
869b54502bSHong Zhang EXTERN_C_BEGIN
879b54502bSHong Zhang #undef __FUNCT__
889b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetReuseFill_Cholesky"
89dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCholeskySetReuseFill_Cholesky(PC pc,PetscTruth flag)
909b54502bSHong Zhang {
919b54502bSHong Zhang   PC_Cholesky *lu;
929b54502bSHong Zhang 
939b54502bSHong Zhang   PetscFunctionBegin;
949b54502bSHong Zhang   lu = (PC_Cholesky*)pc->data;
959b54502bSHong Zhang   lu->reusefill = flag;
969b54502bSHong Zhang   PetscFunctionReturn(0);
979b54502bSHong Zhang }
989b54502bSHong Zhang EXTERN_C_END
999b54502bSHong Zhang 
1009b54502bSHong Zhang #undef __FUNCT__
1019b54502bSHong Zhang #define __FUNCT__ "PCSetFromOptions_Cholesky"
1029b54502bSHong Zhang static PetscErrorCode PCSetFromOptions_Cholesky(PC pc)
1039b54502bSHong Zhang {
1049b54502bSHong Zhang   PC_Cholesky    *lu = (PC_Cholesky*)pc->data;
1059b54502bSHong Zhang   PetscErrorCode ierr;
1069b54502bSHong Zhang   PetscTruth     flg;
1079b54502bSHong Zhang   char           tname[256];
1089b54502bSHong Zhang   PetscFList     ordlist;
1099b54502bSHong Zhang 
1109b54502bSHong Zhang   PetscFunctionBegin;
1119b54502bSHong Zhang   ierr = MatOrderingRegisterAll(PETSC_NULL);CHKERRQ(ierr);
1129b54502bSHong Zhang   ierr = PetscOptionsHead("Cholesky options");CHKERRQ(ierr);
1139b54502bSHong Zhang   ierr = PetscOptionsName("-pc_cholesky_in_place","Form Cholesky in the same memory as the matrix","PCCholeskySetUseInPlace",&flg);CHKERRQ(ierr);
1149b54502bSHong Zhang   if (flg) {
1159b54502bSHong Zhang     ierr = PCCholeskySetUseInPlace(pc);CHKERRQ(ierr);
1169b54502bSHong Zhang   }
117*55ba2a51SBarry Smith   ierr = PetscOptionsReal("-pc_factor_fill","Expected non-zeros in Cholesky/non-zeros in matrix","PCFactorSetFill",lu->info.fill,&lu->info.fill,0);CHKERRQ(ierr);
1189b54502bSHong Zhang 
1199b54502bSHong Zhang   ierr = PetscOptionsName("-pc_cholesky_reuse_fill","Use fill from previous factorization","PCCholeskySetReuseFill",&flg);CHKERRQ(ierr);
1209b54502bSHong Zhang   if (flg) {
1219b54502bSHong Zhang     ierr = PCCholeskySetReuseFill(pc,PETSC_TRUE);CHKERRQ(ierr);
1229b54502bSHong Zhang   }
1239b54502bSHong Zhang   ierr = PetscOptionsName("-pc_cholesky_reuse_ordering","Reuse ordering from previous factorization","PCCholeskySetReuseOrdering",&flg);CHKERRQ(ierr);
1249b54502bSHong Zhang   if (flg) {
1259b54502bSHong Zhang     ierr = PCCholeskySetReuseOrdering(pc,PETSC_TRUE);CHKERRQ(ierr);
1269b54502bSHong Zhang   }
1279b54502bSHong Zhang 
1289b54502bSHong Zhang   ierr = MatGetOrderingList(&ordlist);CHKERRQ(ierr);
1299b54502bSHong Zhang   ierr = PetscOptionsList("-pc_cholesky_mat_ordering_type","Reordering to reduce nonzeros in Cholesky","PCCholeskySetMatOrdering",ordlist,lu->ordering,tname,256,&flg);CHKERRQ(ierr);
1309b54502bSHong Zhang   if (flg) {
1319b54502bSHong Zhang     ierr = PCCholeskySetMatOrdering(pc,tname);CHKERRQ(ierr);
1329b54502bSHong Zhang   }
1339f95998fSHong Zhang   ierr = PetscOptionsName("-pc_factor_shift_nonzero","Shift added to diagonal","PCFactorSetShiftNonzero",&flg);CHKERRQ(ierr);
1349b54502bSHong Zhang   if (flg) {
135afaefe49SHong Zhang     ierr = PCFactorSetShiftNonzero(pc,(PetscReal) PETSC_DECIDE);CHKERRQ(ierr);
1369b54502bSHong Zhang   }
1379f95998fSHong Zhang   ierr = PetscOptionsReal("-pc_factor_shift_nonzero","Shift added to diagonal","PCFactorSetShiftNonzero",lu->info.shiftnz,&lu->info.shiftnz,0);CHKERRQ(ierr);
1389f95998fSHong Zhang   ierr = PetscOptionsName("-pc_factor_shift_positive_definite","Manteuffel shift applied to diagonal","PCFactorSetShiftPd",&flg);CHKERRQ(ierr);
1399b54502bSHong Zhang   if (flg) {
140afaefe49SHong Zhang     ierr = PCFactorSetShiftPd(pc,PETSC_TRUE);CHKERRQ(ierr);
1419b54502bSHong Zhang   }
142ee45ca4aSHong Zhang   ierr = PetscOptionsReal("-pc_factor_zeropivot","Pivot is considered zero if less than","PCFactorSetZeroPivot",lu->info.zeropivot,&lu->info.zeropivot,0);CHKERRQ(ierr);
1439b54502bSHong Zhang 
1449b54502bSHong Zhang   ierr = PetscOptionsTail();CHKERRQ(ierr);
1459b54502bSHong Zhang   PetscFunctionReturn(0);
1469b54502bSHong Zhang }
1479b54502bSHong Zhang 
1489b54502bSHong Zhang #undef __FUNCT__
1499b54502bSHong Zhang #define __FUNCT__ "PCView_Cholesky"
1509b54502bSHong Zhang static PetscErrorCode PCView_Cholesky(PC pc,PetscViewer viewer)
1519b54502bSHong Zhang {
1529b54502bSHong Zhang   PC_Cholesky    *lu = (PC_Cholesky*)pc->data;
1539b54502bSHong Zhang   PetscErrorCode ierr;
1549b54502bSHong Zhang   PetscTruth     iascii,isstring;
1559b54502bSHong Zhang 
1569b54502bSHong Zhang   PetscFunctionBegin;
1579b54502bSHong Zhang   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
1589b54502bSHong Zhang   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_STRING,&isstring);CHKERRQ(ierr);
1599b54502bSHong Zhang   if (iascii) {
1609b54502bSHong Zhang     MatInfo info;
1619b54502bSHong Zhang 
1629b54502bSHong Zhang     if (lu->inplace) {ierr = PetscViewerASCIIPrintf(viewer,"  Cholesky: in-place factorization\n");CHKERRQ(ierr);}
1639b54502bSHong Zhang     else             {ierr = PetscViewerASCIIPrintf(viewer,"  Cholesky: out-of-place factorization\n");CHKERRQ(ierr);}
1649b54502bSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"    matrix ordering: %s\n",lu->ordering);CHKERRQ(ierr);
1659b54502bSHong Zhang     if (lu->fact) {
1669b54502bSHong Zhang       ierr = MatGetInfo(lu->fact,MAT_LOCAL,&info);CHKERRQ(ierr);
167a83599f4SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"    Cholesky nonzeros %G\n",info.nz_used);CHKERRQ(ierr);
1689b54502bSHong Zhang       ierr = PetscViewerPushFormat(viewer,PETSC_VIEWER_ASCII_FACTOR_INFO);CHKERRQ(ierr);
1699b54502bSHong Zhang       ierr = MatView(lu->fact,viewer);CHKERRQ(ierr);
1709b54502bSHong Zhang       ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
1719b54502bSHong Zhang     }
1729b54502bSHong Zhang     if (lu->reusefill)    {ierr = PetscViewerASCIIPrintf(viewer,"       Reusing fill from past factorization\n");CHKERRQ(ierr);}
1739b54502bSHong Zhang     if (lu->reuseordering) {ierr = PetscViewerASCIIPrintf(viewer,"       Reusing reordering from past factorization\n");CHKERRQ(ierr);}
1749b54502bSHong Zhang   } else if (isstring) {
1759b54502bSHong Zhang     ierr = PetscViewerStringSPrintf(viewer," order=%s",lu->ordering);CHKERRQ(ierr);CHKERRQ(ierr);
1769b54502bSHong Zhang   } else {
1779b54502bSHong Zhang     SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported for PCCholesky",((PetscObject)viewer)->type_name);
1789b54502bSHong Zhang   }
1799b54502bSHong Zhang   PetscFunctionReturn(0);
1809b54502bSHong Zhang }
1819b54502bSHong Zhang 
1829b54502bSHong Zhang #undef __FUNCT__
1839b54502bSHong Zhang #define __FUNCT__ "PCGetFactoredMatrix_Cholesky"
1849b54502bSHong Zhang static PetscErrorCode PCGetFactoredMatrix_Cholesky(PC pc,Mat *mat)
1859b54502bSHong Zhang {
1869b54502bSHong Zhang   PC_Cholesky *dir = (PC_Cholesky*)pc->data;
1879b54502bSHong Zhang 
1889b54502bSHong Zhang   PetscFunctionBegin;
1899b54502bSHong Zhang   if (!dir->fact) SETERRQ(PETSC_ERR_ORDER,"Matrix not yet factored; call after KSPSetUp() or PCSetUp()");
1909b54502bSHong Zhang   *mat = dir->fact;
1919b54502bSHong Zhang   PetscFunctionReturn(0);
1929b54502bSHong Zhang }
1939b54502bSHong Zhang 
1949b54502bSHong Zhang #undef __FUNCT__
1959b54502bSHong Zhang #define __FUNCT__ "PCSetUp_Cholesky"
1969b54502bSHong Zhang static PetscErrorCode PCSetUp_Cholesky(PC pc)
1979b54502bSHong Zhang {
1989b54502bSHong Zhang   PetscErrorCode ierr;
1999b54502bSHong Zhang   PetscTruth     flg;
2009b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
2019b54502bSHong Zhang 
2029b54502bSHong Zhang   PetscFunctionBegin;
2039b54502bSHong Zhang   if (dir->reusefill && pc->setupcalled) dir->info.fill = dir->actualfill;
2049b54502bSHong Zhang 
2059b54502bSHong Zhang   if (dir->inplace) {
2069b54502bSHong Zhang     if (dir->row && dir->col && (dir->row != dir->col)) {
2079b54502bSHong Zhang       ierr = ISDestroy(dir->row);CHKERRQ(ierr);
2089b54502bSHong Zhang       dir->row = 0;
2099b54502bSHong Zhang     }
2109b54502bSHong Zhang     if (dir->col) {
2119b54502bSHong Zhang       ierr = ISDestroy(dir->col);CHKERRQ(ierr);
2129b54502bSHong Zhang       dir->col = 0;
2139b54502bSHong Zhang     }
2149b54502bSHong Zhang     ierr = MatGetOrdering(pc->pmat,dir->ordering,&dir->row,&dir->col);CHKERRQ(ierr);
2159b54502bSHong Zhang     if (dir->col && (dir->row != dir->col)) {  /* only use row ordering for SBAIJ */
2169b54502bSHong Zhang       ierr = ISDestroy(dir->col);CHKERRQ(ierr);
2179b54502bSHong Zhang       dir->col=0;
2189b54502bSHong Zhang     }
219efee365bSSatish Balay     if (dir->row) {ierr = PetscLogObjectParent(pc,dir->row);CHKERRQ(ierr);}
2209b54502bSHong Zhang     ierr = MatCholeskyFactor(pc->pmat,dir->row,&dir->info);CHKERRQ(ierr);
2219b54502bSHong Zhang     dir->fact = pc->pmat;
2229b54502bSHong Zhang   } else {
2239b54502bSHong Zhang     MatInfo info;
2249b54502bSHong Zhang     if (!pc->setupcalled) {
2259b54502bSHong Zhang       ierr = MatGetOrdering(pc->pmat,dir->ordering,&dir->row,&dir->col);CHKERRQ(ierr);
2269b54502bSHong Zhang       if (dir->col && (dir->row != dir->col)) {  /* only use row ordering for SBAIJ */
2279b54502bSHong Zhang         ierr = ISDestroy(dir->col);CHKERRQ(ierr);
2289b54502bSHong Zhang         dir->col=0;
2299b54502bSHong Zhang       }
2309b54502bSHong Zhang       ierr = PetscOptionsHasName(pc->prefix,"-pc_cholesky_nonzeros_along_diagonal",&flg);CHKERRQ(ierr);
2319b54502bSHong Zhang       if (flg) {
2329b54502bSHong Zhang         PetscReal tol = 1.e-10;
2339b54502bSHong Zhang         ierr = PetscOptionsGetReal(pc->prefix,"-pc_cholesky_nonzeros_along_diagonal",&tol,PETSC_NULL);CHKERRQ(ierr);
2349b54502bSHong Zhang         ierr = MatReorderForNonzeroDiagonal(pc->pmat,tol,dir->row,dir->row);CHKERRQ(ierr);
2359b54502bSHong Zhang       }
236efee365bSSatish Balay       if (dir->row) {ierr = PetscLogObjectParent(pc,dir->row);CHKERRQ(ierr);}
2379b54502bSHong Zhang       ierr = MatCholeskyFactorSymbolic(pc->pmat,dir->row,&dir->info,&dir->fact);CHKERRQ(ierr);
2389b54502bSHong Zhang       ierr = MatGetInfo(dir->fact,MAT_LOCAL,&info);CHKERRQ(ierr);
2399b54502bSHong Zhang       dir->actualfill = info.fill_ratio_needed;
24052e6d16bSBarry Smith       ierr = PetscLogObjectParent(pc,dir->fact);CHKERRQ(ierr);
2419b54502bSHong Zhang     } else if (pc->flag != SAME_NONZERO_PATTERN) {
2429b54502bSHong Zhang       if (!dir->reuseordering) {
2439b54502bSHong Zhang         if (dir->row && dir->col && (dir->row != dir->col)) {
2449b54502bSHong Zhang           ierr = ISDestroy(dir->row);CHKERRQ(ierr);
2459b54502bSHong Zhang           dir->row = 0;
2469b54502bSHong Zhang         }
2479b54502bSHong Zhang         if (dir->col) {
2489b54502bSHong Zhang           ierr = ISDestroy(dir->col);CHKERRQ(ierr);
2499b54502bSHong Zhang           dir->col =0;
2509b54502bSHong Zhang         }
2519b54502bSHong Zhang         ierr = MatGetOrdering(pc->pmat,dir->ordering,&dir->row,&dir->col);CHKERRQ(ierr);
2529b54502bSHong Zhang         if (dir->col && (dir->row != dir->col)) {  /* only use row ordering for SBAIJ */
2539b54502bSHong Zhang           ierr = ISDestroy(dir->col);CHKERRQ(ierr);
2549b54502bSHong Zhang           dir->col=0;
2559b54502bSHong Zhang         }
2569b54502bSHong Zhang         ierr = PetscOptionsHasName(pc->prefix,"-pc_cholesky_nonzeros_along_diagonal",&flg);CHKERRQ(ierr);
2579b54502bSHong Zhang         if (flg) {
2589b54502bSHong Zhang           PetscReal tol = 1.e-10;
2599b54502bSHong Zhang           ierr = PetscOptionsGetReal(pc->prefix,"-pc_cholesky_nonzeros_along_diagonal",&tol,PETSC_NULL);CHKERRQ(ierr);
2609b54502bSHong Zhang           ierr = MatReorderForNonzeroDiagonal(pc->pmat,tol,dir->row,dir->row);CHKERRQ(ierr);
2619b54502bSHong Zhang         }
262efee365bSSatish Balay         if (dir->row) {ierr = PetscLogObjectParent(pc,dir->row);CHKERRQ(ierr);}
2639b54502bSHong Zhang       }
2649b54502bSHong Zhang       ierr = MatDestroy(dir->fact);CHKERRQ(ierr);
2659b54502bSHong Zhang       ierr = MatCholeskyFactorSymbolic(pc->pmat,dir->row,&dir->info,&dir->fact);CHKERRQ(ierr);
2669b54502bSHong Zhang       ierr = MatGetInfo(dir->fact,MAT_LOCAL,&info);CHKERRQ(ierr);
2679b54502bSHong Zhang       dir->actualfill = info.fill_ratio_needed;
26852e6d16bSBarry Smith       ierr = PetscLogObjectParent(pc,dir->fact);CHKERRQ(ierr);
2699b54502bSHong Zhang     }
2709b54502bSHong Zhang     ierr = MatCholeskyFactorNumeric(pc->pmat,&dir->info,&dir->fact);CHKERRQ(ierr);
2719b54502bSHong Zhang   }
2729b54502bSHong Zhang   PetscFunctionReturn(0);
2739b54502bSHong Zhang }
2749b54502bSHong Zhang 
2759b54502bSHong Zhang #undef __FUNCT__
2769b54502bSHong Zhang #define __FUNCT__ "PCDestroy_Cholesky"
2779b54502bSHong Zhang static PetscErrorCode PCDestroy_Cholesky(PC pc)
2789b54502bSHong Zhang {
2799b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
2809b54502bSHong Zhang   PetscErrorCode ierr;
2819b54502bSHong Zhang 
2829b54502bSHong Zhang   PetscFunctionBegin;
2839b54502bSHong Zhang   if (!dir->inplace && dir->fact) {ierr = MatDestroy(dir->fact);CHKERRQ(ierr);}
2849b54502bSHong Zhang   if (dir->row) {ierr = ISDestroy(dir->row);CHKERRQ(ierr);}
2859b54502bSHong Zhang   if (dir->col) {ierr = ISDestroy(dir->col);CHKERRQ(ierr);}
2869b54502bSHong Zhang   ierr = PetscStrfree(dir->ordering);CHKERRQ(ierr);
2879b54502bSHong Zhang   ierr = PetscFree(dir);CHKERRQ(ierr);
2889b54502bSHong Zhang   PetscFunctionReturn(0);
2899b54502bSHong Zhang }
2909b54502bSHong Zhang 
2919b54502bSHong Zhang #undef __FUNCT__
2929b54502bSHong Zhang #define __FUNCT__ "PCApply_Cholesky"
2939b54502bSHong Zhang static PetscErrorCode PCApply_Cholesky(PC pc,Vec x,Vec y)
2949b54502bSHong Zhang {
2959b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
2969b54502bSHong Zhang   PetscErrorCode ierr;
2979b54502bSHong Zhang 
2989b54502bSHong Zhang   PetscFunctionBegin;
2999b54502bSHong Zhang   if (dir->inplace) {ierr = MatSolve(pc->pmat,x,y);CHKERRQ(ierr);}
3009b54502bSHong Zhang   else              {ierr = MatSolve(dir->fact,x,y);CHKERRQ(ierr);}
3019b54502bSHong Zhang   PetscFunctionReturn(0);
3029b54502bSHong Zhang }
3039b54502bSHong Zhang 
3049b54502bSHong Zhang #undef __FUNCT__
3059b54502bSHong Zhang #define __FUNCT__ "PCApplyTranspose_Cholesky"
3069b54502bSHong Zhang static PetscErrorCode PCApplyTranspose_Cholesky(PC pc,Vec x,Vec y)
3079b54502bSHong Zhang {
3089b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
3099b54502bSHong Zhang   PetscErrorCode ierr;
3109b54502bSHong Zhang 
3119b54502bSHong Zhang   PetscFunctionBegin;
3129b54502bSHong Zhang   if (dir->inplace) {ierr = MatSolveTranspose(pc->pmat,x,y);CHKERRQ(ierr);}
3139b54502bSHong Zhang   else              {ierr = MatSolveTranspose(dir->fact,x,y);CHKERRQ(ierr);}
3149b54502bSHong Zhang   PetscFunctionReturn(0);
3159b54502bSHong Zhang }
3169b54502bSHong Zhang 
3179b54502bSHong Zhang /* -----------------------------------------------------------------------------------*/
3189b54502bSHong Zhang 
3199b54502bSHong Zhang EXTERN_C_BEGIN
3209b54502bSHong Zhang #undef __FUNCT__
321*55ba2a51SBarry Smith #define __FUNCT__ "PCFactorSetFill_Cholesky"
322*55ba2a51SBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetFill_Cholesky(PC pc,PetscReal fill)
3239b54502bSHong Zhang {
3249b54502bSHong Zhang   PC_Cholesky *dir;
3259b54502bSHong Zhang 
3269b54502bSHong Zhang   PetscFunctionBegin;
3279b54502bSHong Zhang   dir = (PC_Cholesky*)pc->data;
3289b54502bSHong Zhang   dir->info.fill = fill;
3299b54502bSHong Zhang   PetscFunctionReturn(0);
3309b54502bSHong Zhang }
3319b54502bSHong Zhang EXTERN_C_END
3329b54502bSHong Zhang 
3339b54502bSHong Zhang EXTERN_C_BEGIN
3349b54502bSHong Zhang #undef __FUNCT__
3359b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetUseInPlace_Cholesky"
336dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCholeskySetUseInPlace_Cholesky(PC pc)
3379b54502bSHong Zhang {
3389b54502bSHong Zhang   PC_Cholesky *dir;
3399b54502bSHong Zhang 
3409b54502bSHong Zhang   PetscFunctionBegin;
3419b54502bSHong Zhang   dir = (PC_Cholesky*)pc->data;
3429b54502bSHong Zhang   dir->inplace = PETSC_TRUE;
3439b54502bSHong Zhang   PetscFunctionReturn(0);
3449b54502bSHong Zhang }
3459b54502bSHong Zhang EXTERN_C_END
3469b54502bSHong Zhang 
3479b54502bSHong Zhang EXTERN_C_BEGIN
3489b54502bSHong Zhang #undef __FUNCT__
3499b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetMatOrdering_Cholesky"
350dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCholeskySetMatOrdering_Cholesky(PC pc,MatOrderingType ordering)
3519b54502bSHong Zhang {
3529b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
3539b54502bSHong Zhang   PetscErrorCode ierr;
3549b54502bSHong Zhang 
3559b54502bSHong Zhang   PetscFunctionBegin;
3569b54502bSHong Zhang   ierr = PetscStrfree(dir->ordering);CHKERRQ(ierr);
3579b54502bSHong Zhang   ierr = PetscStrallocpy(ordering,&dir->ordering);CHKERRQ(ierr);
3589b54502bSHong Zhang   PetscFunctionReturn(0);
3599b54502bSHong Zhang }
3609b54502bSHong Zhang EXTERN_C_END
3619b54502bSHong Zhang 
3629b54502bSHong Zhang /* -----------------------------------------------------------------------------------*/
3639b54502bSHong Zhang 
3649b54502bSHong Zhang #undef __FUNCT__
3659b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetReuseOrdering"
3669b54502bSHong Zhang /*@
3679b54502bSHong Zhang    PCCholeskySetReuseOrdering - When similar matrices are factored, this
3689b54502bSHong Zhang    causes the ordering computed in the first factor to be used for all
3699b54502bSHong Zhang    following factors.
3709b54502bSHong Zhang 
3719b54502bSHong Zhang    Collective on PC
3729b54502bSHong Zhang 
3739b54502bSHong Zhang    Input Parameters:
3749b54502bSHong Zhang +  pc - the preconditioner context
3759b54502bSHong Zhang -  flag - PETSC_TRUE to reuse else PETSC_FALSE
3769b54502bSHong Zhang 
3779b54502bSHong Zhang    Options Database Key:
3789b54502bSHong Zhang .  -pc_cholesky_reuse_ordering - Activate PCCholeskySetReuseOrdering()
3799b54502bSHong Zhang 
3809b54502bSHong Zhang    Level: intermediate
3819b54502bSHong Zhang 
3829b54502bSHong Zhang .keywords: PC, levels, reordering, factorization, incomplete, LU
3839b54502bSHong Zhang 
3849b54502bSHong Zhang .seealso: PCCholeskySetReuseFill(), PCICholeskySetReuseOrdering(), PCICholeskyDTSetReuseFill()
3859b54502bSHong Zhang @*/
386dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCholeskySetReuseOrdering(PC pc,PetscTruth flag)
3879b54502bSHong Zhang {
3889b54502bSHong Zhang   PetscErrorCode ierr,(*f)(PC,PetscTruth);
3899b54502bSHong Zhang 
3909b54502bSHong Zhang   PetscFunctionBegin;
3919b54502bSHong Zhang   PetscValidHeaderSpecific(pc,PC_COOKIE,1);
3929b54502bSHong Zhang   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCCholeskySetReuseOrdering_C",(void (**)(void))&f);CHKERRQ(ierr);
3939b54502bSHong Zhang   if (f) {
3949b54502bSHong Zhang     ierr = (*f)(pc,flag);CHKERRQ(ierr);
3959b54502bSHong Zhang   }
3969b54502bSHong Zhang   PetscFunctionReturn(0);
3979b54502bSHong Zhang }
3989b54502bSHong Zhang 
3999b54502bSHong Zhang #undef __FUNCT__
4009b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetReuseFill"
4019b54502bSHong Zhang /*@
4029b54502bSHong Zhang    PCCholeskySetReuseFill - When matrices with same nonzero structure are Cholesky factored,
4039b54502bSHong Zhang    this causes later ones to use the fill computed in the initial factorization.
4049b54502bSHong Zhang 
4059b54502bSHong Zhang    Collective on PC
4069b54502bSHong Zhang 
4079b54502bSHong Zhang    Input Parameters:
4089b54502bSHong Zhang +  pc - the preconditioner context
4099b54502bSHong Zhang -  flag - PETSC_TRUE to reuse else PETSC_FALSE
4109b54502bSHong Zhang 
4119b54502bSHong Zhang    Options Database Key:
4129b54502bSHong Zhang .  -pc_cholesky_reuse_fill - Activates PCCholeskySetReuseFill()
4139b54502bSHong Zhang 
4149b54502bSHong Zhang    Level: intermediate
4159b54502bSHong Zhang 
4169b54502bSHong Zhang .keywords: PC, levels, reordering, factorization, incomplete, Cholesky
4179b54502bSHong Zhang 
4189b54502bSHong Zhang .seealso: PCICholeskySetReuseOrdering(), PCCholeskySetReuseOrdering(), PCICholeskyDTSetReuseFill()
4199b54502bSHong Zhang @*/
420dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCholeskySetReuseFill(PC pc,PetscTruth flag)
4219b54502bSHong Zhang {
4229b54502bSHong Zhang   PetscErrorCode ierr,(*f)(PC,PetscTruth);
4239b54502bSHong Zhang 
4249b54502bSHong Zhang   PetscFunctionBegin;
4259b54502bSHong Zhang   PetscValidHeaderSpecific(pc,PC_COOKIE,2);
4269b54502bSHong Zhang   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCCholeskySetReuseFill_C",(void (**)(void))&f);CHKERRQ(ierr);
4279b54502bSHong Zhang   if (f) {
4289b54502bSHong Zhang     ierr = (*f)(pc,flag);CHKERRQ(ierr);
4299b54502bSHong Zhang   }
4309b54502bSHong Zhang   PetscFunctionReturn(0);
4319b54502bSHong Zhang }
4329b54502bSHong Zhang 
4339b54502bSHong Zhang #undef __FUNCT__
4349b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetUseInPlace"
4359b54502bSHong Zhang /*@
4369b54502bSHong Zhang    PCCholeskySetUseInPlace - Tells the system to do an in-place factorization.
4379b54502bSHong Zhang    For dense matrices, this enables the solution of much larger problems.
4389b54502bSHong Zhang    For sparse matrices the factorization cannot be done truly in-place
4399b54502bSHong Zhang    so this does not save memory during the factorization, but after the matrix
4409b54502bSHong Zhang    is factored, the original unfactored matrix is freed, thus recovering that
4419b54502bSHong Zhang    space.
4429b54502bSHong Zhang 
4439b54502bSHong Zhang    Collective on PC
4449b54502bSHong Zhang 
4459b54502bSHong Zhang    Input Parameters:
4469b54502bSHong Zhang .  pc - the preconditioner context
4479b54502bSHong Zhang 
4489b54502bSHong Zhang    Options Database Key:
4499b54502bSHong Zhang .  -pc_cholesky_in_place - Activates in-place factorization
4509b54502bSHong Zhang 
4519b54502bSHong Zhang    Notes:
4529b54502bSHong Zhang    PCCholeskySetUseInplace() can only be used with the KSP method KSPPREONLY or when
4539b54502bSHong Zhang    a different matrix is provided for the multiply and the preconditioner in
4549b54502bSHong Zhang    a call to KSPSetOperators().
4559b54502bSHong Zhang    This is because the Krylov space methods require an application of the
4569b54502bSHong Zhang    matrix multiplication, which is not possible here because the matrix has
4579b54502bSHong Zhang    been factored in-place, replacing the original matrix.
4589b54502bSHong Zhang 
4599b54502bSHong Zhang    Level: intermediate
4609b54502bSHong Zhang 
4619b54502bSHong Zhang .keywords: PC, set, factorization, direct, inplace, in-place, Cholesky
4629b54502bSHong Zhang 
4639b54502bSHong Zhang .seealso: PCICholeskySetUseInPlace()
4649b54502bSHong Zhang @*/
465dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCholeskySetUseInPlace(PC pc)
4669b54502bSHong Zhang {
4679b54502bSHong Zhang   PetscErrorCode ierr,(*f)(PC);
4689b54502bSHong Zhang 
4699b54502bSHong Zhang   PetscFunctionBegin;
4709b54502bSHong Zhang   PetscValidHeaderSpecific(pc,PC_COOKIE,1);
4719b54502bSHong Zhang   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCCholeskySetUseInPlace_C",(void (**)(void))&f);CHKERRQ(ierr);
4729b54502bSHong Zhang   if (f) {
4739b54502bSHong Zhang     ierr = (*f)(pc);CHKERRQ(ierr);
4749b54502bSHong Zhang   }
4759b54502bSHong Zhang   PetscFunctionReturn(0);
4769b54502bSHong Zhang }
4779b54502bSHong Zhang 
4789b54502bSHong Zhang #undef __FUNCT__
4799b54502bSHong Zhang #define __FUNCT__ "PCCholeskySetMatOrdering"
4809b54502bSHong Zhang /*@
4819b54502bSHong Zhang     PCCholeskySetMatOrdering - Sets the ordering routine (to reduce fill) to
4829b54502bSHong Zhang     be used it the Cholesky factorization.
4839b54502bSHong Zhang 
4849b54502bSHong Zhang     Collective on PC
4859b54502bSHong Zhang 
4869b54502bSHong Zhang     Input Parameters:
4879b54502bSHong Zhang +   pc - the preconditioner context
4889b54502bSHong Zhang -   ordering - the matrix ordering name, for example, MATORDERING_ND or MATORDERING_RCM
4899b54502bSHong Zhang 
4909b54502bSHong Zhang     Options Database Key:
4919b54502bSHong Zhang .   -pc_cholesky_mat_ordering_type <nd,rcm,...> - Sets ordering routine
4929b54502bSHong Zhang 
4939b54502bSHong Zhang     Level: intermediate
4949b54502bSHong Zhang 
4959b54502bSHong Zhang .seealso: PCICholeskySetMatOrdering()
4969b54502bSHong Zhang @*/
497dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCholeskySetMatOrdering(PC pc,MatOrderingType ordering)
4989b54502bSHong Zhang {
4999b54502bSHong Zhang   PetscErrorCode ierr,(*f)(PC,MatOrderingType);
5009b54502bSHong Zhang 
5019b54502bSHong Zhang   PetscFunctionBegin;
5029b54502bSHong Zhang   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCCholeskySetMatOrdering_C",(void (**)(void))&f);CHKERRQ(ierr);
5039b54502bSHong Zhang   if (f) {
5049b54502bSHong Zhang     ierr = (*f)(pc,ordering);CHKERRQ(ierr);
5059b54502bSHong Zhang   }
5069b54502bSHong Zhang   PetscFunctionReturn(0);
5079b54502bSHong Zhang }
5089b54502bSHong Zhang 
5099b54502bSHong Zhang /*MC
5109b54502bSHong Zhang    PCCholesky - Uses a direct solver, based on Cholesky factorization, as a preconditioner
5119b54502bSHong Zhang 
5129b54502bSHong Zhang    Options Database Keys:
5139b54502bSHong Zhang +  -pc_cholesky_reuse_ordering - Activate PCLUSetReuseOrdering()
5149b54502bSHong Zhang .  -pc_cholesky_reuse_fill - Activates PCLUSetReuseFill()
515*55ba2a51SBarry Smith .  -pc_factor_fill <fill> - Sets fill amount
5169b54502bSHong Zhang .  -pc_cholesky_in_place - Activates in-place factorization
517f251bdbdSHong Zhang .  -pc_cholesky_mat_ordering_type <nd,rcm,...> - Sets ordering routine
518f251bdbdSHong Zhang .  -pc_factor_shift_nonzero <shift> - Sets shift amount or PETSC_DECIDE for the default
519f251bdbdSHong Zhang -  -pc_factor_shift_positive_definite [PETSC_TRUE/PETSC_FALSE] - Activate/Deactivate PCFactorSetShiftPd(); the value
520f251bdbdSHong Zhang    is optional with PETSC_TRUE being the default
5219b54502bSHong Zhang 
5229b54502bSHong Zhang    Notes: Not all options work for all matrix formats
5239b54502bSHong Zhang 
5249b54502bSHong Zhang    Level: beginner
5259b54502bSHong Zhang 
5269b54502bSHong Zhang    Concepts: Cholesky factorization, direct solver
5279b54502bSHong Zhang 
5289b54502bSHong Zhang    Notes: Usually this will compute an "exact" solution in one iteration and does
5299b54502bSHong Zhang           not need a Krylov method (i.e. you can use -ksp_type preonly, or
5309b54502bSHong Zhang           KSPSetType(ksp,KSPPREONLY) for the Krylov method
5319b54502bSHong Zhang 
5329b54502bSHong Zhang .seealso:  PCCreate(), PCSetType(), PCType (for list of available types), PC,
5339b54502bSHong Zhang            PCILU, PCLU, PCICC, PCCholeskySetReuseOrdering(), PCCholeskySetReuseFill(), PCGetFactoredMatrix(),
534*55ba2a51SBarry Smith            PCFactorSetFill(), PCFactorSetShiftNonzero(), PCFactorSetShiftPd(),
535f251bdbdSHong Zhang 	   PCCholeskySetUseInPlace(), PCCholeskySetMatOrdering(),PCFactorSetShiftNonzero(),PCFactorSetShiftPd()
5369b54502bSHong Zhang 
5379b54502bSHong Zhang M*/
5389b54502bSHong Zhang 
5399b54502bSHong Zhang EXTERN_C_BEGIN
5409b54502bSHong Zhang #undef __FUNCT__
5419b54502bSHong Zhang #define __FUNCT__ "PCCreate_Cholesky"
542dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCreate_Cholesky(PC pc)
5439b54502bSHong Zhang {
5449b54502bSHong Zhang   PetscErrorCode ierr;
5459b54502bSHong Zhang   PC_Cholesky    *dir;
5469b54502bSHong Zhang 
5479b54502bSHong Zhang   PetscFunctionBegin;
5489b54502bSHong Zhang   ierr = PetscNew(PC_Cholesky,&dir);CHKERRQ(ierr);
54952e6d16bSBarry Smith   ierr = PetscLogObjectMemory(pc,sizeof(PC_Cholesky));CHKERRQ(ierr);
5509b54502bSHong Zhang 
5519b54502bSHong Zhang   dir->fact                   = 0;
5529b54502bSHong Zhang   dir->inplace                = PETSC_FALSE;
5539b54502bSHong Zhang   ierr = MatFactorInfoInitialize(&dir->info);CHKERRQ(ierr);
5549b54502bSHong Zhang   dir->info.fill              = 5.0;
5550a29876aSHong Zhang   dir->info.shiftnz           = 0.0;
556fbf22428SSatish Balay   dir->info.shiftpd           = 0.0; /* false */
5579b54502bSHong Zhang   dir->info.shift_fraction    = 0.0;
5589b54502bSHong Zhang   dir->info.pivotinblocks     = 1.0;
5599b54502bSHong Zhang   dir->col                    = 0;
5609b54502bSHong Zhang   dir->row                    = 0;
5619b54502bSHong Zhang   ierr = PetscStrallocpy(MATORDERING_NATURAL,&dir->ordering);CHKERRQ(ierr);
5629b54502bSHong Zhang   dir->reusefill        = PETSC_FALSE;
5639b54502bSHong Zhang   dir->reuseordering    = PETSC_FALSE;
5649b54502bSHong Zhang   pc->data              = (void*)dir;
5659b54502bSHong Zhang 
5669b54502bSHong Zhang   pc->ops->destroy           = PCDestroy_Cholesky;
5679b54502bSHong Zhang   pc->ops->apply             = PCApply_Cholesky;
5689b54502bSHong Zhang   pc->ops->applytranspose    = PCApplyTranspose_Cholesky;
5699b54502bSHong Zhang   pc->ops->setup             = PCSetUp_Cholesky;
5709b54502bSHong Zhang   pc->ops->setfromoptions    = PCSetFromOptions_Cholesky;
5719b54502bSHong Zhang   pc->ops->view              = PCView_Cholesky;
5729b54502bSHong Zhang   pc->ops->applyrichardson   = 0;
5739b54502bSHong Zhang   pc->ops->getfactoredmatrix = PCGetFactoredMatrix_Cholesky;
5749b54502bSHong Zhang 
575afaefe49SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetZeroPivot_C","PCFactorSetZeroPivot_Cholesky",
576afaefe49SHong Zhang                     PCFactorSetZeroPivot_Cholesky);CHKERRQ(ierr);
577afaefe49SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetShiftNonzero_C","PCFactorSetShiftNonzero_Cholesky",
578afaefe49SHong Zhang                     PCFactorSetShiftNonzero_Cholesky);CHKERRQ(ierr);
579afaefe49SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetShiftPd_C","PCFactorSetShiftPd_Cholesky",
580afaefe49SHong Zhang                     PCFactorSetShiftPd_Cholesky);CHKERRQ(ierr);
581afaefe49SHong Zhang 
582*55ba2a51SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetFill_C","PCFactorSetFill_Cholesky",
583*55ba2a51SBarry Smith                     PCFactorSetFill_Cholesky);CHKERRQ(ierr);
5849b54502bSHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCCholeskySetUseInPlace_C","PCCholeskySetUseInPlace_Cholesky",
5859b54502bSHong Zhang                     PCCholeskySetUseInPlace_Cholesky);CHKERRQ(ierr);
5869b54502bSHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCCholeskySetMatOrdering_C","PCCholeskySetMatOrdering_Cholesky",
5879b54502bSHong Zhang                     PCCholeskySetMatOrdering_Cholesky);CHKERRQ(ierr);
5889b54502bSHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCCholeskySetReuseOrdering_C","PCCholeskySetReuseOrdering_Cholesky",
5899b54502bSHong Zhang                     PCCholeskySetReuseOrdering_Cholesky);CHKERRQ(ierr);
5909b54502bSHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCCholeskySetReuseFill_C","PCCholeskySetReuseFill_Cholesky",
5919b54502bSHong Zhang                     PCCholeskySetReuseFill_Cholesky);CHKERRQ(ierr);
5929b54502bSHong Zhang   PetscFunctionReturn(0);
5939b54502bSHong Zhang }
5949b54502bSHong Zhang EXTERN_C_END
595