xref: /petsc/src/mat/impls/aij/seq/umfpack/umfpack.c (revision 65f45395ee41b0220cb8038ff86f18021a88432f)
11677d0b8SKris Buschelman 
21677d0b8SKris Buschelman /*
3d515b9b4SShri Abhyankar    Provides an interface to the UMFPACK sparse solver available through SuiteSparse version 4.2.1
42d4e2982SHong Zhang 
59e475b0dSSatish Balay    When build with PETSC_USE_64BIT_INDICES this will use Suitesparse_long as the
68592901bSBarry Smith    integer type in UMFPACK, otherwise it will use int. This means
78592901bSBarry Smith    all integers in this file as simply declared as PetscInt. Also it means
89e475b0dSSatish Balay    that one cannot use 64BIT_INDICES on 32bit machines [as Suitesparse_long is 32bit only]
92d4e2982SHong Zhang 
101677d0b8SKris Buschelman */
11c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h>
121677d0b8SKris Buschelman 
138592901bSBarry Smith #if defined(PETSC_USE_64BIT_INDICES)
148592901bSBarry Smith #if defined(PETSC_USE_COMPLEX)
158592901bSBarry Smith #define umfpack_UMF_free_symbolic                      umfpack_zl_free_symbolic
168592901bSBarry Smith #define umfpack_UMF_free_numeric                       umfpack_zl_free_numeric
17d755ee67SBarry Smith /* the type casts are needed because PetscInt is long long while SuiteSparse_long is long and compilers warn even when they are identical */
18d755ee67SBarry Smith #define umfpack_UMF_wsolve(a,b,c,d,e,f,g,h,i,j,k,l,m,n) umfpack_zl_wsolve(a,(SuiteSparse_long*)b,(SuiteSparse_long*)c,d,e,f,g,h,i,(SuiteSparse_long*)j,k,l,(SuiteSparse_long*)m,n)
19d755ee67SBarry Smith #define umfpack_UMF_numeric(a,b,c,d,e,f,g,h)          umfpack_zl_numeric((SuiteSparse_long*)a,(SuiteSparse_long*)b,c,d,e,f,g,h)
208592901bSBarry Smith #define umfpack_UMF_report_numeric                    umfpack_zl_report_numeric
218592901bSBarry Smith #define umfpack_UMF_report_control                    umfpack_zl_report_control
228592901bSBarry Smith #define umfpack_UMF_report_status                     umfpack_zl_report_status
238592901bSBarry Smith #define umfpack_UMF_report_info                       umfpack_zl_report_info
248592901bSBarry Smith #define umfpack_UMF_report_symbolic                   umfpack_zl_report_symbolic
25d755ee67SBarry Smith #define umfpack_UMF_qsymbolic(a,b,c,d,e,f,g,h,i,j)    umfpack_zl_qsymbolic(a,b,(SuiteSparse_long*)c,(SuiteSparse_long*)d,e,f,(SuiteSparse_long*)g,h,i,j)
26d755ee67SBarry Smith #define umfpack_UMF_symbolic(a,b,c,d,e,f,g,h,i)       umfpack_zl_symbolic(a,b,(SuiteSparse_long*)c,(SuiteSparse_long*)d,e,f,g,h,i)
278592901bSBarry Smith #define umfpack_UMF_defaults                          umfpack_zl_defaults
288592901bSBarry Smith 
298592901bSBarry Smith #else
308592901bSBarry Smith #define umfpack_UMF_free_symbolic                  umfpack_dl_free_symbolic
318592901bSBarry Smith #define umfpack_UMF_free_numeric                   umfpack_dl_free_numeric
32d755ee67SBarry Smith #define umfpack_UMF_wsolve(a,b,c,d,e,f,g,h,i,j,k)  umfpack_dl_wsolve(a,(SuiteSparse_long*)b,(SuiteSparse_long*)c,d,e,f,g,h,i,(SuiteSparse_long*)j,k)
33d755ee67SBarry Smith #define umfpack_UMF_numeric(a,b,c,d,e,f,g)         umfpack_dl_numeric((SuiteSparse_long*)a,(SuiteSparse_long*)b,c,d,e,f,g)
348592901bSBarry Smith #define umfpack_UMF_report_numeric                 umfpack_dl_report_numeric
358592901bSBarry Smith #define umfpack_UMF_report_control                 umfpack_dl_report_control
368592901bSBarry Smith #define umfpack_UMF_report_status                  umfpack_dl_report_status
378592901bSBarry Smith #define umfpack_UMF_report_info                    umfpack_dl_report_info
388592901bSBarry Smith #define umfpack_UMF_report_symbolic                umfpack_dl_report_symbolic
39d755ee67SBarry Smith #define umfpack_UMF_qsymbolic(a,b,c,d,e,f,g,h,i)   umfpack_dl_qsymbolic(a,b,(SuiteSparse_long*)c,(SuiteSparse_long*)d,e,(SuiteSparse_long*)f,g,h,i)
40d755ee67SBarry Smith #define umfpack_UMF_symbolic(a,b,c,d,e,f,g,h)      umfpack_dl_symbolic(a,b,(SuiteSparse_long*)c,(SuiteSparse_long*)d,e,f,g,h)
418592901bSBarry Smith #define umfpack_UMF_defaults                       umfpack_dl_defaults
428592901bSBarry Smith #endif
438592901bSBarry Smith 
448592901bSBarry Smith #else
458592901bSBarry Smith #if defined(PETSC_USE_COMPLEX)
468592901bSBarry Smith #define umfpack_UMF_free_symbolic   umfpack_zi_free_symbolic
478592901bSBarry Smith #define umfpack_UMF_free_numeric    umfpack_zi_free_numeric
488592901bSBarry Smith #define umfpack_UMF_wsolve          umfpack_zi_wsolve
498592901bSBarry Smith #define umfpack_UMF_numeric         umfpack_zi_numeric
508592901bSBarry Smith #define umfpack_UMF_report_numeric  umfpack_zi_report_numeric
518592901bSBarry Smith #define umfpack_UMF_report_control  umfpack_zi_report_control
528592901bSBarry Smith #define umfpack_UMF_report_status   umfpack_zi_report_status
538592901bSBarry Smith #define umfpack_UMF_report_info     umfpack_zi_report_info
548592901bSBarry Smith #define umfpack_UMF_report_symbolic umfpack_zi_report_symbolic
558592901bSBarry Smith #define umfpack_UMF_qsymbolic       umfpack_zi_qsymbolic
568592901bSBarry Smith #define umfpack_UMF_symbolic        umfpack_zi_symbolic
578592901bSBarry Smith #define umfpack_UMF_defaults        umfpack_zi_defaults
588592901bSBarry Smith 
598592901bSBarry Smith #else
608592901bSBarry Smith #define umfpack_UMF_free_symbolic   umfpack_di_free_symbolic
618592901bSBarry Smith #define umfpack_UMF_free_numeric    umfpack_di_free_numeric
628592901bSBarry Smith #define umfpack_UMF_wsolve          umfpack_di_wsolve
638592901bSBarry Smith #define umfpack_UMF_numeric         umfpack_di_numeric
648592901bSBarry Smith #define umfpack_UMF_report_numeric  umfpack_di_report_numeric
658592901bSBarry Smith #define umfpack_UMF_report_control  umfpack_di_report_control
668592901bSBarry Smith #define umfpack_UMF_report_status   umfpack_di_report_status
678592901bSBarry Smith #define umfpack_UMF_report_info     umfpack_di_report_info
688592901bSBarry Smith #define umfpack_UMF_report_symbolic umfpack_di_report_symbolic
698592901bSBarry Smith #define umfpack_UMF_qsymbolic       umfpack_di_qsymbolic
708592901bSBarry Smith #define umfpack_UMF_symbolic        umfpack_di_symbolic
718592901bSBarry Smith #define umfpack_UMF_defaults        umfpack_di_defaults
728592901bSBarry Smith #endif
738592901bSBarry Smith #endif
748592901bSBarry Smith 
751677d0b8SKris Buschelman EXTERN_C_BEGIN
76c6db04a5SJed Brown #include <umfpack.h>
771677d0b8SKris Buschelman EXTERN_C_END
781677d0b8SKris Buschelman 
79fcfd50ebSBarry Smith static const char *const UmfpackOrderingTypes[] = {"CHOLMOD","AMD","GIVEN","METIS","BEST","NONE","USER","UmfpackOrderingTypes","UMFPACK_ORDERING_",0};
80fcfd50ebSBarry Smith 
811677d0b8SKris Buschelman typedef struct {
821677d0b8SKris Buschelman   void         *Symbolic, *Numeric;
831677d0b8SKris Buschelman   double       Info[UMFPACK_INFO], Control[UMFPACK_CONTROL],*W;
84ae26a541SJed Brown   PetscInt     *Wi,*perm_c;
85ae26a541SJed Brown   Mat          A;               /* Matrix used for factorization */
861677d0b8SKris Buschelman   MatStructure flg;
871677d0b8SKris Buschelman 
881677d0b8SKris Buschelman   /* Flag to clean up UMFPACK objects during Destroy */
89ace3abfcSBarry Smith   PetscBool CleanUpUMFPACK;
90f0c56d0fSKris Buschelman } Mat_UMFPACK;
91f0c56d0fSKris Buschelman 
924b019bd2SJed Brown static PetscErrorCode MatDestroy_UMFPACK(Mat A)
93dfbe8321SBarry Smith {
94dfbe8321SBarry Smith   PetscErrorCode ierr;
95b9c12af5SBarry Smith   Mat_UMFPACK    *lu=(Mat_UMFPACK*)A->data;
961677d0b8SKris Buschelman 
971677d0b8SKris Buschelman   PetscFunctionBegin;
98b9c12af5SBarry Smith   if (lu->CleanUpUMFPACK) {
998592901bSBarry Smith     umfpack_UMF_free_symbolic(&lu->Symbolic);
1008592901bSBarry Smith     umfpack_UMF_free_numeric(&lu->Numeric);
1011677d0b8SKris Buschelman     ierr = PetscFree(lu->Wi);CHKERRQ(ierr);
1021677d0b8SKris Buschelman     ierr = PetscFree(lu->W);CHKERRQ(ierr);
1031677d0b8SKris Buschelman     ierr = PetscFree(lu->perm_c);CHKERRQ(ierr);
1041677d0b8SKris Buschelman   }
105ae26a541SJed Brown   ierr = MatDestroy(&lu->A);CHKERRQ(ierr);
106b9c12af5SBarry Smith   ierr = PetscFree(A->data);CHKERRQ(ierr);
1071677d0b8SKris Buschelman   PetscFunctionReturn(0);
1081677d0b8SKris Buschelman }
1091677d0b8SKris Buschelman 
1104b019bd2SJed Brown static PetscErrorCode MatSolve_UMFPACK_Private(Mat A,Vec b,Vec x,int uflag)
1116849ba73SBarry Smith {
112b9c12af5SBarry Smith   Mat_UMFPACK       *lu = (Mat_UMFPACK*)A->data;
113ae26a541SJed Brown   Mat_SeqAIJ        *a  = (Mat_SeqAIJ*)lu->A->data;
114d9ca1df4SBarry Smith   PetscScalar       *av = a->a,*xa;
115d9ca1df4SBarry Smith   const PetscScalar *ba;
116dfbe8321SBarry Smith   PetscErrorCode    ierr;
117ae26a541SJed Brown   PetscInt          *ai = a->i,*aj = a->j,status;
118fcf6e9abSJed Brown   static PetscBool  cite = PETSC_FALSE;
1191677d0b8SKris Buschelman 
1201677d0b8SKris Buschelman   PetscFunctionBegin;
121a643c07aSBarry Smith   if (!A->rmap->n) PetscFunctionReturn(0);
122fcf6e9abSJed Brown   ierr = PetscCitationsRegister("@article{davis2004algorithm,\n  title={Algorithm 832: {UMFPACK} V4.3---An Unsymmetric-Pattern Multifrontal Method},\n  author={Davis, Timothy A},\n  journal={ACM Transactions on Mathematical Software (TOMS)},\n  volume={30},\n  number={2},\n  pages={196--199},\n  year={2004},\n  publisher={ACM}\n}\n",&cite);CHKERRQ(ierr);
1232d4e2982SHong Zhang   /* solve Ax = b by umfpack_*_wsolve */
1241677d0b8SKris Buschelman   /* ----------------------------------*/
1252d4e2982SHong Zhang 
126ae26a541SJed Brown   if (!lu->Wi) {  /* first time, allocate working space for wsolve */
127785e854fSJed Brown     ierr = PetscMalloc1(A->rmap->n,&lu->Wi);CHKERRQ(ierr);
128785e854fSJed Brown     ierr = PetscMalloc1(5*A->rmap->n,&lu->W);CHKERRQ(ierr);
129ae26a541SJed Brown   }
130ae26a541SJed Brown 
1311e1ea65dSPierre Jolivet   ierr = VecGetArrayRead(b,&ba);CHKERRQ(ierr);
1321e1ea65dSPierre Jolivet   ierr = VecGetArray(x,&xa);CHKERRQ(ierr);
13379c34000SHong Zhang #if defined(PETSC_USE_COMPLEX)
134b0043f70SBarry Smith   status = umfpack_UMF_wsolve(uflag,ai,aj,(PetscReal*)av,NULL,(PetscReal*)xa,NULL,(PetscReal*)ba,NULL,lu->Numeric,lu->Control,lu->Info,lu->Wi,lu->W);
13579c34000SHong Zhang #else
1364b019bd2SJed Brown   status = umfpack_UMF_wsolve(uflag,ai,aj,av,xa,ba,lu->Numeric,lu->Control,lu->Info,lu->Wi,lu->W);
13779c34000SHong Zhang #endif
1388592901bSBarry Smith   umfpack_UMF_report_info(lu->Control, lu->Info);
1391677d0b8SKris Buschelman   if (status < 0) {
1408592901bSBarry Smith     umfpack_UMF_report_status(lu->Control, status);
141e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"umfpack_UMF_wsolve failed");
1421677d0b8SKris Buschelman   }
1431677d0b8SKris Buschelman 
144d9ca1df4SBarry Smith   ierr = VecRestoreArrayRead(b,&ba);CHKERRQ(ierr);
145057dcbc9SJed Brown   ierr = VecRestoreArray(x,&xa);CHKERRQ(ierr);
1464b019bd2SJed Brown   PetscFunctionReturn(0);
1474b019bd2SJed Brown }
1482a325a84SHong Zhang 
1494b019bd2SJed Brown static PetscErrorCode MatSolve_UMFPACK(Mat A,Vec b,Vec x)
1504b019bd2SJed Brown {
1514b019bd2SJed Brown   PetscErrorCode ierr;
1524b019bd2SJed Brown 
1534b019bd2SJed Brown   PetscFunctionBegin;
1544b019bd2SJed Brown   /* We gave UMFPACK the algebraic transpose (because it assumes column alignment) */
1554b019bd2SJed Brown   ierr = MatSolve_UMFPACK_Private(A,b,x,UMFPACK_Aat);CHKERRQ(ierr);
1564b019bd2SJed Brown   PetscFunctionReturn(0);
1574b019bd2SJed Brown }
1584b019bd2SJed Brown 
1594b019bd2SJed Brown static PetscErrorCode MatSolveTranspose_UMFPACK(Mat A,Vec b,Vec x)
1604b019bd2SJed Brown {
1614b019bd2SJed Brown   PetscErrorCode ierr;
1624b019bd2SJed Brown 
1634b019bd2SJed Brown   PetscFunctionBegin;
1644b019bd2SJed Brown   /* We gave UMFPACK the algebraic transpose (because it assumes column alignment) */
1654b019bd2SJed Brown   ierr = MatSolve_UMFPACK_Private(A,b,x,UMFPACK_A);CHKERRQ(ierr);
1661677d0b8SKris Buschelman   PetscFunctionReturn(0);
1671677d0b8SKris Buschelman }
1681677d0b8SKris Buschelman 
1694b019bd2SJed Brown static PetscErrorCode MatLUFactorNumeric_UMFPACK(Mat F,Mat A,const MatFactorInfo *info)
1706849ba73SBarry Smith {
171b9c12af5SBarry Smith   Mat_UMFPACK    *lu = (Mat_UMFPACK*)(F)->data;
172ae26a541SJed Brown   Mat_SeqAIJ     *a  = (Mat_SeqAIJ*)A->data;
173ae26a541SJed Brown   PetscInt       *ai = a->i,*aj=a->j,status;
174ae26a541SJed Brown   PetscScalar    *av = a->a;
1756849ba73SBarry Smith   PetscErrorCode ierr;
1761677d0b8SKris Buschelman 
1771677d0b8SKris Buschelman   PetscFunctionBegin;
178a643c07aSBarry Smith   if (!A->rmap->n) PetscFunctionReturn(0);
1791677d0b8SKris Buschelman   /* numeric factorization of A' */
1801677d0b8SKris Buschelman   /* ----------------------------*/
1812d4e2982SHong Zhang 
1828592901bSBarry Smith   if (lu->flg == SAME_NONZERO_PATTERN && lu->Numeric) {
1838592901bSBarry Smith     umfpack_UMF_free_numeric(&lu->Numeric);
1848592901bSBarry Smith   }
1852d4e2982SHong Zhang #if defined(PETSC_USE_COMPLEX)
1868592901bSBarry Smith   status = umfpack_UMF_numeric(ai,aj,(double*)av,NULL,lu->Symbolic,&lu->Numeric,lu->Control,lu->Info);
1872d4e2982SHong Zhang #else
1888592901bSBarry Smith   status = umfpack_UMF_numeric(ai,aj,av,lu->Symbolic,&lu->Numeric,lu->Control,lu->Info);
1898592901bSBarry Smith #endif
1909f42a82aSMatthew Knepley   if (status < 0) {
1918592901bSBarry Smith     umfpack_UMF_report_status(lu->Control, status);
192e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"umfpack_UMF_numeric failed");
1939f42a82aSMatthew Knepley   }
1942d4e2982SHong Zhang   /* report numeric factorization of A' when Control[PRL] > 3 */
1958592901bSBarry Smith   (void) umfpack_UMF_report_numeric(lu->Numeric, lu->Control);
1961677d0b8SKris Buschelman 
197ae26a541SJed Brown   ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
198ae26a541SJed Brown   ierr = MatDestroy(&lu->A);CHKERRQ(ierr);
1992205254eSKarl Rupp 
200ae26a541SJed Brown   lu->A                  = A;
2012a325a84SHong Zhang   lu->flg                = SAME_NONZERO_PATTERN;
2022a325a84SHong Zhang   lu->CleanUpUMFPACK     = PETSC_TRUE;
2034b019bd2SJed Brown   F->ops->solve          = MatSolve_UMFPACK;
2044b019bd2SJed Brown   F->ops->solvetranspose = MatSolveTranspose_UMFPACK;
2051677d0b8SKris Buschelman   PetscFunctionReturn(0);
2061677d0b8SKris Buschelman }
2071677d0b8SKris Buschelman 
2084b019bd2SJed Brown static PetscErrorCode MatLUFactorSymbolic_UMFPACK(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
2096849ba73SBarry Smith {
210ae26a541SJed Brown   Mat_SeqAIJ     *a  = (Mat_SeqAIJ*)A->data;
211b9c12af5SBarry Smith   Mat_UMFPACK    *lu = (Mat_UMFPACK*)(F->data);
212dfbe8321SBarry Smith   PetscErrorCode ierr;
213ae26a541SJed Brown   PetscInt       i,*ai = a->i,*aj = a->j,m=A->rmap->n,n=A->cmap->n;
214989213a4SSatish Balay #if !defined(PETSC_USE_COMPLEX)
215ae26a541SJed Brown   PetscScalar    *av = a->a;
216989213a4SSatish Balay #endif
2175d0c19d7SBarry Smith   const PetscInt *ra;
2188592901bSBarry Smith   PetscInt       status;
2191677d0b8SKris Buschelman 
2201677d0b8SKris Buschelman   PetscFunctionBegin;
221a643c07aSBarry Smith   (F)->ops->lufactornumeric = MatLUFactorNumeric_UMFPACK;
222a643c07aSBarry Smith   if (!n) PetscFunctionReturn(0);
2232c7c0729SBarry Smith   if (r) {
2240f6efc10SHong Zhang     ierr = ISGetIndices(r,&ra);CHKERRQ(ierr);
225785e854fSJed Brown     ierr = PetscMalloc1(m,&lu->perm_c);CHKERRQ(ierr);
2262d4e2982SHong Zhang     /* we cannot simply memcpy on 64 bit archs */
2272d4e2982SHong Zhang     for (i = 0; i < m; i++) lu->perm_c[i] = ra[i];
2280f6efc10SHong Zhang     ierr = ISRestoreIndices(r,&ra);CHKERRQ(ierr);
2291677d0b8SKris Buschelman   }
2301677d0b8SKris Buschelman 
2311677d0b8SKris Buschelman   /* print the control parameters */
2328592901bSBarry Smith   if (lu->Control[UMFPACK_PRL] > 1) umfpack_UMF_report_control(lu->Control);
2331677d0b8SKris Buschelman 
2341677d0b8SKris Buschelman   /* symbolic factorization of A' */
2351677d0b8SKris Buschelman   /* ---------------------------------------------------------------------- */
2362c7c0729SBarry Smith   if (r) { /* use Petsc row ordering */
2378592901bSBarry Smith #if !defined(PETSC_USE_COMPLEX)
238ae26a541SJed Brown     status = umfpack_UMF_qsymbolic(n,m,ai,aj,av,lu->perm_c,&lu->Symbolic,lu->Control,lu->Info);
2392d4e2982SHong Zhang #else
240b0043f70SBarry Smith     status = umfpack_UMF_qsymbolic(n,m,ai,aj,NULL,NULL,lu->perm_c,&lu->Symbolic,lu->Control,lu->Info);
2418592901bSBarry Smith #endif
2422d4e2982SHong Zhang   } else { /* use Umfpack col ordering */
2438592901bSBarry Smith #if !defined(PETSC_USE_COMPLEX)
244ae26a541SJed Brown     status = umfpack_UMF_symbolic(n,m,ai,aj,av,&lu->Symbolic,lu->Control,lu->Info);
2458592901bSBarry Smith #else
246ae26a541SJed Brown     status = umfpack_UMF_symbolic(n,m,ai,aj,NULL,NULL,&lu->Symbolic,lu->Control,lu->Info);
2478592901bSBarry Smith #endif
2482d4e2982SHong Zhang   }
2492d4e2982SHong Zhang   if (status < 0) {
2508592901bSBarry Smith     umfpack_UMF_report_info(lu->Control, lu->Info);
2518592901bSBarry Smith     umfpack_UMF_report_status(lu->Control, status);
252e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"umfpack_UMF_symbolic failed");
2532d4e2982SHong Zhang   }
2542d4e2982SHong Zhang   /* report sumbolic factorization of A' when Control[PRL] > 3 */
2558592901bSBarry Smith   (void) umfpack_UMF_report_symbolic(lu->Symbolic, lu->Control);
2561677d0b8SKris Buschelman 
2571677d0b8SKris Buschelman   lu->flg                   = DIFFERENT_NONZERO_PATTERN;
258e1b4c3a1SKris Buschelman   lu->CleanUpUMFPACK        = PETSC_TRUE;
2591677d0b8SKris Buschelman   PetscFunctionReturn(0);
2601677d0b8SKris Buschelman }
2611677d0b8SKris Buschelman 
262860c79edSBarry Smith static PetscErrorCode MatView_Info_UMFPACK(Mat A,PetscViewer viewer)
2636849ba73SBarry Smith {
264b9c12af5SBarry Smith   Mat_UMFPACK    *lu= (Mat_UMFPACK*)A->data;
265dfbe8321SBarry Smith   PetscErrorCode ierr;
266f0c56d0fSKris Buschelman 
2671677d0b8SKris Buschelman   PetscFunctionBegin;
2681677d0b8SKris Buschelman   /* check if matrix is UMFPACK type */
269f0c56d0fSKris Buschelman   if (A->ops->solve != MatSolve_UMFPACK) PetscFunctionReturn(0);
2701677d0b8SKris Buschelman 
2711677d0b8SKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"UMFPACK run parameters:\n");CHKERRQ(ierr);
2721677d0b8SKris Buschelman   /* Control parameters used by reporting routiones */
2731677d0b8SKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_PRL]: %g\n",lu->Control[UMFPACK_PRL]);CHKERRQ(ierr);
2741677d0b8SKris Buschelman 
2751677d0b8SKris Buschelman   /* Control parameters used by symbolic factorization */
2760f6efc10SHong Zhang   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_STRATEGY]: %g\n",lu->Control[UMFPACK_STRATEGY]);CHKERRQ(ierr);
2771677d0b8SKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_DENSE_COL]: %g\n",lu->Control[UMFPACK_DENSE_COL]);CHKERRQ(ierr);
2781677d0b8SKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_DENSE_ROW]: %g\n",lu->Control[UMFPACK_DENSE_ROW]);CHKERRQ(ierr);
2790f6efc10SHong Zhang   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_AMD_DENSE]: %g\n",lu->Control[UMFPACK_AMD_DENSE]);CHKERRQ(ierr);
2801677d0b8SKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_BLOCK_SIZE]: %g\n",lu->Control[UMFPACK_BLOCK_SIZE]);CHKERRQ(ierr);
2810f6efc10SHong Zhang   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_FIXQ]: %g\n",lu->Control[UMFPACK_FIXQ]);CHKERRQ(ierr);
2820f6efc10SHong Zhang   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_AGGRESSIVE]: %g\n",lu->Control[UMFPACK_AGGRESSIVE]);CHKERRQ(ierr);
2831677d0b8SKris Buschelman 
2841677d0b8SKris Buschelman   /* Control parameters used by numeric factorization */
2851677d0b8SKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_PIVOT_TOLERANCE]: %g\n",lu->Control[UMFPACK_PIVOT_TOLERANCE]);CHKERRQ(ierr);
2860f6efc10SHong Zhang   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_SYM_PIVOT_TOLERANCE]: %g\n",lu->Control[UMFPACK_SYM_PIVOT_TOLERANCE]);CHKERRQ(ierr);
2870f6efc10SHong Zhang   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_SCALE]: %g\n",lu->Control[UMFPACK_SCALE]);CHKERRQ(ierr);
2881677d0b8SKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_ALLOC_INIT]: %g\n",lu->Control[UMFPACK_ALLOC_INIT]);CHKERRQ(ierr);
2890f6efc10SHong Zhang   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_DROPTOL]: %g\n",lu->Control[UMFPACK_DROPTOL]);CHKERRQ(ierr);
2901677d0b8SKris Buschelman 
2911677d0b8SKris Buschelman   /* Control parameters used by solve */
2921677d0b8SKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_IRSTEP]: %g\n",lu->Control[UMFPACK_IRSTEP]);CHKERRQ(ierr);
2931677d0b8SKris Buschelman 
2941677d0b8SKris Buschelman   /* mat ordering */
2952c7c0729SBarry Smith   if (!lu->perm_c) {
2962c7c0729SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Control[UMFPACK_ORDERING]: AMD (not using the PETSc ordering)\n",UmfpackOrderingTypes[(int)lu->Control[UMFPACK_ORDERING]]);CHKERRQ(ierr);
297fcfd50ebSBarry Smith   }
2981677d0b8SKris Buschelman   PetscFunctionReturn(0);
2991677d0b8SKris Buschelman }
3001677d0b8SKris Buschelman 
3014b019bd2SJed Brown static PetscErrorCode MatView_UMFPACK(Mat A,PetscViewer viewer)
3026849ba73SBarry Smith {
303dfbe8321SBarry Smith   PetscErrorCode    ierr;
304ace3abfcSBarry Smith   PetscBool         iascii;
305d844382dSKris Buschelman   PetscViewerFormat format;
306d844382dSKris Buschelman 
307d844382dSKris Buschelman   PetscFunctionBegin;
308251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
30932077d6dSBarry Smith   if (iascii) {
310d844382dSKris Buschelman     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
3112f59403fSHong Zhang     if (format == PETSC_VIEWER_ASCII_INFO) {
312860c79edSBarry Smith       ierr = MatView_Info_UMFPACK(A,viewer);CHKERRQ(ierr);
313d844382dSKris Buschelman     }
314d844382dSKris Buschelman   }
315d844382dSKris Buschelman   PetscFunctionReturn(0);
316d844382dSKris Buschelman }
317d844382dSKris Buschelman 
318ea799195SBarry Smith PetscErrorCode MatFactorGetSolverType_seqaij_umfpack(Mat A,MatSolverType *type)
31935bd34faSBarry Smith {
32035bd34faSBarry Smith   PetscFunctionBegin;
3212692d6eeSBarry Smith   *type = MATSOLVERUMFPACK;
32235bd34faSBarry Smith   PetscFunctionReturn(0);
32335bd34faSBarry Smith }
32435bd34faSBarry Smith 
3252f71e704SKris Buschelman /*MC
3262692d6eeSBarry Smith   MATSOLVERUMFPACK = "umfpack" - A matrix type providing direct solvers (LU) for sequential matrices
3272f71e704SKris Buschelman   via the external package UMFPACK.
3282f71e704SKris Buschelman 
329c2b89b5dSBarry Smith   Use ./configure --download-suitesparse to install PETSc to use UMFPACK
330c2b89b5dSBarry Smith 
3313ca39a21SBarry Smith   Use -pc_type lu -pc_factor_mat_solver_type umfpack to use this direct solver
3322f71e704SKris Buschelman 
3332f71e704SKris Buschelman   Consult UMFPACK documentation for more information about the Control parameters
3342f71e704SKris Buschelman   which correspond to the options database keys below.
3352f71e704SKris Buschelman 
3362f71e704SKris Buschelman   Options Database Keys:
337ba41fbb6SBarry Smith + -mat_umfpack_ordering                - CHOLMOD, AMD, GIVEN, METIS, BEST, NONE
338ba41fbb6SBarry Smith . -mat_umfpack_prl                     - UMFPACK print level: Control[UMFPACK_PRL]
339e08999f5SMatthew G Knepley . -mat_umfpack_strategy <AUTO>         - (choose one of) AUTO UNSYMMETRIC SYMMETRIC 2BY2
3402f71e704SKris Buschelman . -mat_umfpack_dense_col <alpha_c>     - UMFPACK dense column threshold: Control[UMFPACK_DENSE_COL]
341e08999f5SMatthew G Knepley . -mat_umfpack_dense_row <0.2>         - Control[UMFPACK_DENSE_ROW]
342e08999f5SMatthew G Knepley . -mat_umfpack_amd_dense <10>          - Control[UMFPACK_AMD_DENSE]
3432f71e704SKris Buschelman . -mat_umfpack_block_size <bs>         - UMFPACK block size for BLAS-Level 3 calls: Control[UMFPACK_BLOCK_SIZE]
344e08999f5SMatthew G Knepley . -mat_umfpack_2by2_tolerance <0.01>   - Control[UMFPACK_2BY2_TOLERANCE]
345e08999f5SMatthew G Knepley . -mat_umfpack_fixq <0>                - Control[UMFPACK_FIXQ]
346e08999f5SMatthew G Knepley . -mat_umfpack_aggressive <1>          - Control[UMFPACK_AGGRESSIVE]
3472f71e704SKris Buschelman . -mat_umfpack_pivot_tolerance <delta> - UMFPACK partial pivot tolerance: Control[UMFPACK_PIVOT_TOLERANCE]
348e08999f5SMatthew G Knepley . -mat_umfpack_sym_pivot_tolerance <0.001> - Control[UMFPACK_SYM_PIVOT_TOLERANCE]
349e08999f5SMatthew G Knepley . -mat_umfpack_scale <NONE>           - (choose one of) NONE SUM MAX
3502f71e704SKris Buschelman . -mat_umfpack_alloc_init <delta>      - UMFPACK factorized matrix allocation modifier: Control[UMFPACK_ALLOC_INIT]
351e08999f5SMatthew G Knepley . -mat_umfpack_droptol <0>            - Control[UMFPACK_DROPTOL]
3522f71e704SKris Buschelman - -mat_umfpack_irstep <maxit>          - UMFPACK maximum number of iterative refinement steps: Control[UMFPACK_IRSTEP]
3532f71e704SKris Buschelman 
3542f71e704SKris Buschelman    Level: beginner
355a364b7d2SBarry Smith 
356a364b7d2SBarry Smith    Note: UMFPACK is part of SuiteSparse http://faculty.cse.tamu.edu/davis/suitesparse.html
3572f71e704SKris Buschelman 
3583ca39a21SBarry Smith .seealso: PCLU, MATSOLVERSUPERLU, MATSOLVERMUMPS, PCFactorSetMatSolverType(), MatSolverType
3592f71e704SKris Buschelman M*/
360b2573a8aSBarry Smith 
3618cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqaij_umfpack(Mat A,MatFactorType ftype,Mat *F)
3623519fcdcSHong Zhang {
3633519fcdcSHong Zhang   Mat            B;
3643519fcdcSHong Zhang   Mat_UMFPACK    *lu;
3653519fcdcSHong Zhang   PetscErrorCode ierr;
3668592901bSBarry Smith   PetscInt       m=A->rmap->n,n=A->cmap->n,idx;
3672205254eSKarl Rupp   const char     *strategy[]={"AUTO","UNSYMMETRIC","SYMMETRIC"};
3682205254eSKarl Rupp   const char     *scale[]   ={"NONE","SUM","MAX"};
369ace3abfcSBarry Smith   PetscBool      flg;
3702d4e2982SHong Zhang 
3713519fcdcSHong Zhang   PetscFunctionBegin;
3723519fcdcSHong Zhang   /* Create the factorization matrix F */
373ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
3743519fcdcSHong Zhang   ierr = MatSetSizes(B,PETSC_DECIDE,PETSC_DECIDE,m,n);CHKERRQ(ierr);
375b9c12af5SBarry Smith   ierr = PetscStrallocpy("umfpack",&((PetscObject)B)->type_name);CHKERRQ(ierr);
376b9c12af5SBarry Smith   ierr = MatSetUp(B);CHKERRQ(ierr);
377b9c12af5SBarry Smith 
378b00a9115SJed Brown   ierr = PetscNewLog(B,&lu);CHKERRQ(ierr);
3792205254eSKarl Rupp 
380b9c12af5SBarry Smith   B->data                   = lu;
381b9c12af5SBarry Smith   B->ops->getinfo          = MatGetInfo_External;
3823519fcdcSHong Zhang   B->ops->lufactorsymbolic = MatLUFactorSymbolic_UMFPACK;
3833519fcdcSHong Zhang   B->ops->destroy          = MatDestroy_UMFPACK;
3843519fcdcSHong Zhang   B->ops->view             = MatView_UMFPACK;
3851aef8b4cSStefano Zampini   B->ops->matsolve         = NULL;
3862205254eSKarl Rupp 
3873ca39a21SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverType_C",MatFactorGetSolverType_seqaij_umfpack);CHKERRQ(ierr);
3882205254eSKarl Rupp 
389d5f3da31SBarry Smith   B->factortype   = MAT_FACTOR_LU;
3903519fcdcSHong Zhang   B->assembled    = PETSC_TRUE;           /* required by -ksp_view */
3913519fcdcSHong Zhang   B->preallocated = PETSC_TRUE;
3923519fcdcSHong Zhang 
39300c67f3bSHong Zhang   ierr = PetscFree(B->solvertype);CHKERRQ(ierr);
39400c67f3bSHong Zhang   ierr = PetscStrallocpy(MATSOLVERUMFPACK,&B->solvertype);CHKERRQ(ierr);
395f73b0415SBarry Smith   B->canuseordering = PETSC_TRUE;
3964ac6704cSBarry Smith   ierr = PetscStrallocpy(MATORDERINGEXTERNAL,(char**)&B->preferredordering[MAT_FACTOR_LU]);CHKERRQ(ierr);
39700c67f3bSHong Zhang 
3983519fcdcSHong Zhang   /* initializations */
3993519fcdcSHong Zhang   /* ------------------------------------------------*/
4003519fcdcSHong Zhang   /* get the default control parameters */
4018592901bSBarry Smith   umfpack_UMF_defaults(lu->Control);
4020298fd71SBarry Smith   lu->perm_c                  = NULL; /* use defaul UMFPACK col permutation */
4033519fcdcSHong Zhang   lu->Control[UMFPACK_IRSTEP] = 0;          /* max num of iterative refinement steps to attempt */
4043519fcdcSHong Zhang 
405ce94432eSBarry Smith   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)A),((PetscObject)A)->prefix,"UMFPACK Options","Mat");CHKERRQ(ierr);
4063519fcdcSHong Zhang   /* Control parameters used by reporting routiones */
4070298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_prl","Control[UMFPACK_PRL]","None",lu->Control[UMFPACK_PRL],&lu->Control[UMFPACK_PRL],NULL);CHKERRQ(ierr);
4083519fcdcSHong Zhang 
4093519fcdcSHong Zhang   /* Control parameters for symbolic factorization */
410fcfd50ebSBarry Smith   ierr = PetscOptionsEList("-mat_umfpack_strategy","ordering and pivoting strategy","None",strategy,3,strategy[0],&idx,&flg);CHKERRQ(ierr);
4113519fcdcSHong Zhang   if (flg) {
4123519fcdcSHong Zhang     switch (idx) {
4133519fcdcSHong Zhang     case 0: lu->Control[UMFPACK_STRATEGY] = UMFPACK_STRATEGY_AUTO; break;
4143519fcdcSHong Zhang     case 1: lu->Control[UMFPACK_STRATEGY] = UMFPACK_STRATEGY_UNSYMMETRIC; break;
4153519fcdcSHong Zhang     case 2: lu->Control[UMFPACK_STRATEGY] = UMFPACK_STRATEGY_SYMMETRIC; break;
4163519fcdcSHong Zhang     }
4173519fcdcSHong Zhang   }
4188caf3d72SBarry Smith   ierr = PetscOptionsEList("-mat_umfpack_ordering","Internal ordering method","None",UmfpackOrderingTypes,sizeof(UmfpackOrderingTypes)/sizeof(UmfpackOrderingTypes[0]),UmfpackOrderingTypes[(int)lu->Control[UMFPACK_ORDERING]],&idx,&flg);CHKERRQ(ierr);
419fcfd50ebSBarry Smith   if (flg) lu->Control[UMFPACK_ORDERING] = (int)idx;
4200298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_dense_col","Control[UMFPACK_DENSE_COL]","None",lu->Control[UMFPACK_DENSE_COL],&lu->Control[UMFPACK_DENSE_COL],NULL);CHKERRQ(ierr);
4210298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_dense_row","Control[UMFPACK_DENSE_ROW]","None",lu->Control[UMFPACK_DENSE_ROW],&lu->Control[UMFPACK_DENSE_ROW],NULL);CHKERRQ(ierr);
4220298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_amd_dense","Control[UMFPACK_AMD_DENSE]","None",lu->Control[UMFPACK_AMD_DENSE],&lu->Control[UMFPACK_AMD_DENSE],NULL);CHKERRQ(ierr);
4230298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_block_size","Control[UMFPACK_BLOCK_SIZE]","None",lu->Control[UMFPACK_BLOCK_SIZE],&lu->Control[UMFPACK_BLOCK_SIZE],NULL);CHKERRQ(ierr);
4240298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_fixq","Control[UMFPACK_FIXQ]","None",lu->Control[UMFPACK_FIXQ],&lu->Control[UMFPACK_FIXQ],NULL);CHKERRQ(ierr);
4250298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_aggressive","Control[UMFPACK_AGGRESSIVE]","None",lu->Control[UMFPACK_AGGRESSIVE],&lu->Control[UMFPACK_AGGRESSIVE],NULL);CHKERRQ(ierr);
4263519fcdcSHong Zhang 
4273519fcdcSHong Zhang   /* Control parameters used by numeric factorization */
4280298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_pivot_tolerance","Control[UMFPACK_PIVOT_TOLERANCE]","None",lu->Control[UMFPACK_PIVOT_TOLERANCE],&lu->Control[UMFPACK_PIVOT_TOLERANCE],NULL);CHKERRQ(ierr);
4290298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_sym_pivot_tolerance","Control[UMFPACK_SYM_PIVOT_TOLERANCE]","None",lu->Control[UMFPACK_SYM_PIVOT_TOLERANCE],&lu->Control[UMFPACK_SYM_PIVOT_TOLERANCE],NULL);CHKERRQ(ierr);
4303519fcdcSHong Zhang   ierr = PetscOptionsEList("-mat_umfpack_scale","Control[UMFPACK_SCALE]","None",scale,3,scale[0],&idx,&flg);CHKERRQ(ierr);
4313519fcdcSHong Zhang   if (flg) {
4323519fcdcSHong Zhang     switch (idx) {
4333519fcdcSHong Zhang     case 0: lu->Control[UMFPACK_SCALE] = UMFPACK_SCALE_NONE; break;
4343519fcdcSHong Zhang     case 1: lu->Control[UMFPACK_SCALE] = UMFPACK_SCALE_SUM; break;
4353519fcdcSHong Zhang     case 2: lu->Control[UMFPACK_SCALE] = UMFPACK_SCALE_MAX; break;
4363519fcdcSHong Zhang     }
4373519fcdcSHong Zhang   }
4380298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_alloc_init","Control[UMFPACK_ALLOC_INIT]","None",lu->Control[UMFPACK_ALLOC_INIT],&lu->Control[UMFPACK_ALLOC_INIT],NULL);CHKERRQ(ierr);
4390298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_front_alloc_init","Control[UMFPACK_FRONT_ALLOC_INIT]","None",lu->Control[UMFPACK_FRONT_ALLOC_INIT],&lu->Control[UMFPACK_ALLOC_INIT],NULL);CHKERRQ(ierr);
4400298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_droptol","Control[UMFPACK_DROPTOL]","None",lu->Control[UMFPACK_DROPTOL],&lu->Control[UMFPACK_DROPTOL],NULL);CHKERRQ(ierr);
4413519fcdcSHong Zhang 
4423519fcdcSHong Zhang   /* Control parameters used by solve */
4430298fd71SBarry Smith   ierr = PetscOptionsReal("-mat_umfpack_irstep","Control[UMFPACK_IRSTEP]","None",lu->Control[UMFPACK_IRSTEP],&lu->Control[UMFPACK_IRSTEP],NULL);CHKERRQ(ierr);
4443519fcdcSHong Zhang   PetscOptionsEnd();
4453519fcdcSHong Zhang   *F = B;
4463519fcdcSHong Zhang   PetscFunctionReturn(0);
4473519fcdcSHong Zhang }
448b2573a8aSBarry Smith 
449db87b0f2SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqaij_cholmod(Mat,MatFactorType,Mat*);
450db87b0f2SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat,MatFactorType,Mat*);
451db87b0f2SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqaij_klu(Mat,MatFactorType,Mat*);
452a2fc1e05SToby Isaac PETSC_INTERN PetscErrorCode MatGetFactor_seqaij_spqr(Mat,MatFactorType,Mat*);
4532d4e2982SHong Zhang 
4543ca39a21SBarry Smith PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_SuiteSparse(void)
45542c9c57cSBarry Smith {
45642c9c57cSBarry Smith   PetscErrorCode ierr;
45742c9c57cSBarry Smith 
45842c9c57cSBarry Smith   PetscFunctionBegin;
4593ca39a21SBarry Smith   ierr = MatSolverTypeRegister(MATSOLVERUMFPACK,MATSEQAIJ,  MAT_FACTOR_LU,MatGetFactor_seqaij_umfpack);CHKERRQ(ierr);
4603ca39a21SBarry Smith   ierr = MatSolverTypeRegister(MATSOLVERCHOLMOD,MATSEQAIJ,  MAT_FACTOR_CHOLESKY,MatGetFactor_seqaij_cholmod);CHKERRQ(ierr);
4613ca39a21SBarry Smith   ierr = MatSolverTypeRegister(MATSOLVERCHOLMOD,MATSEQSBAIJ,MAT_FACTOR_CHOLESKY,MatGetFactor_seqsbaij_cholmod);CHKERRQ(ierr);
4623ca39a21SBarry Smith   ierr = MatSolverTypeRegister(MATSOLVERKLU,MATSEQAIJ,      MAT_FACTOR_LU,MatGetFactor_seqaij_klu);CHKERRQ(ierr);
463a2fc1e05SToby Isaac   ierr = MatSolverTypeRegister(MATSOLVERSPQR,MATSEQAIJ,     MAT_FACTOR_QR,MatGetFactor_seqaij_spqr);CHKERRQ(ierr);
464*65f45395SPierre Jolivet   if (!PetscDefined(USE_COMPLEX)) {
46502ef7dfaSPierre Jolivet     ierr = MatSolverTypeRegister(MATSOLVERSPQR,MATNORMAL,   MAT_FACTOR_QR,MatGetFactor_seqaij_spqr);CHKERRQ(ierr);
466*65f45395SPierre Jolivet   }
467*65f45395SPierre Jolivet   ierr = MatSolverTypeRegister(MATSOLVERSPQR,MATNORMALHERMITIAN, MAT_FACTOR_QR,MatGetFactor_seqaij_spqr);CHKERRQ(ierr);
46842c9c57cSBarry Smith   PetscFunctionReturn(0);
46942c9c57cSBarry Smith }
470