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 92d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatDestroy_UMFPACK(Mat A) 93d71ae5a4SJacob Faibussowitsch { 94b9c12af5SBarry Smith Mat_UMFPACK *lu = (Mat_UMFPACK *)A->data; 951677d0b8SKris Buschelman 961677d0b8SKris Buschelman PetscFunctionBegin; 97b9c12af5SBarry Smith if (lu->CleanUpUMFPACK) { 988592901bSBarry Smith umfpack_UMF_free_symbolic(&lu->Symbolic); 998592901bSBarry Smith umfpack_UMF_free_numeric(&lu->Numeric); 1009566063dSJacob Faibussowitsch PetscCall(PetscFree(lu->Wi)); 1019566063dSJacob Faibussowitsch PetscCall(PetscFree(lu->W)); 1029566063dSJacob Faibussowitsch PetscCall(PetscFree(lu->perm_c)); 1031677d0b8SKris Buschelman } 1049566063dSJacob Faibussowitsch PetscCall(MatDestroy(&lu->A)); 1052e956fe4SStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorGetSolverType_C", NULL)); 1069566063dSJacob Faibussowitsch PetscCall(PetscFree(A->data)); 107*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1081677d0b8SKris Buschelman } 1091677d0b8SKris Buschelman 110d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_UMFPACK_Private(Mat A, Vec b, Vec x, int uflag) 111d71ae5a4SJacob Faibussowitsch { 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; 116ae26a541SJed Brown PetscInt *ai = a->i, *aj = a->j, status; 117fcf6e9abSJed Brown static PetscBool cite = PETSC_FALSE; 1181677d0b8SKris Buschelman 1191677d0b8SKris Buschelman PetscFunctionBegin; 120*3ba16761SJacob Faibussowitsch if (!A->rmap->n) PetscFunctionReturn(PETSC_SUCCESS); 1219371c9d4SSatish Balay PetscCall(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 " 1229371c9d4SSatish Balay "volume={30},\n number={2},\n pages={196--199},\n year={2004},\n publisher={ACM}\n}\n", 1239371c9d4SSatish Balay &cite)); 1242d4e2982SHong Zhang /* solve Ax = b by umfpack_*_wsolve */ 1251677d0b8SKris Buschelman /* ----------------------------------*/ 1262d4e2982SHong Zhang 127ae26a541SJed Brown if (!lu->Wi) { /* first time, allocate working space for wsolve */ 1289566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(A->rmap->n, &lu->Wi)); 1299566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(5 * A->rmap->n, &lu->W)); 130ae26a541SJed Brown } 131ae26a541SJed Brown 1329566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(b, &ba)); 1339566063dSJacob Faibussowitsch PetscCall(VecGetArray(x, &xa)); 13479c34000SHong Zhang #if defined(PETSC_USE_COMPLEX) 135b0043f70SBarry 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); 13679c34000SHong Zhang #else 1374b019bd2SJed Brown status = umfpack_UMF_wsolve(uflag, ai, aj, av, xa, ba, lu->Numeric, lu->Control, lu->Info, lu->Wi, lu->W); 13879c34000SHong Zhang #endif 1398592901bSBarry Smith umfpack_UMF_report_info(lu->Control, lu->Info); 1401677d0b8SKris Buschelman if (status < 0) { 1418592901bSBarry Smith umfpack_UMF_report_status(lu->Control, status); 142e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF, PETSC_ERR_LIB, "umfpack_UMF_wsolve failed"); 1431677d0b8SKris Buschelman } 1441677d0b8SKris Buschelman 1459566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(b, &ba)); 1469566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(x, &xa)); 147*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1484b019bd2SJed Brown } 1492a325a84SHong Zhang 150d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_UMFPACK(Mat A, Vec b, Vec x) 151d71ae5a4SJacob Faibussowitsch { 1524b019bd2SJed Brown PetscFunctionBegin; 1534b019bd2SJed Brown /* We gave UMFPACK the algebraic transpose (because it assumes column alignment) */ 1549566063dSJacob Faibussowitsch PetscCall(MatSolve_UMFPACK_Private(A, b, x, UMFPACK_Aat)); 155*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1564b019bd2SJed Brown } 1574b019bd2SJed Brown 158d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolveTranspose_UMFPACK(Mat A, Vec b, Vec x) 159d71ae5a4SJacob Faibussowitsch { 1604b019bd2SJed Brown PetscFunctionBegin; 1614b019bd2SJed Brown /* We gave UMFPACK the algebraic transpose (because it assumes column alignment) */ 1629566063dSJacob Faibussowitsch PetscCall(MatSolve_UMFPACK_Private(A, b, x, UMFPACK_A)); 163*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1641677d0b8SKris Buschelman } 1651677d0b8SKris Buschelman 166d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatLUFactorNumeric_UMFPACK(Mat F, Mat A, const MatFactorInfo *info) 167d71ae5a4SJacob Faibussowitsch { 168b9c12af5SBarry Smith Mat_UMFPACK *lu = (Mat_UMFPACK *)(F)->data; 169ae26a541SJed Brown Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data; 170ae26a541SJed Brown PetscInt *ai = a->i, *aj = a->j, status; 171ae26a541SJed Brown PetscScalar *av = a->a; 1721677d0b8SKris Buschelman 1731677d0b8SKris Buschelman PetscFunctionBegin; 174*3ba16761SJacob Faibussowitsch if (!A->rmap->n) PetscFunctionReturn(PETSC_SUCCESS); 1751677d0b8SKris Buschelman /* numeric factorization of A' */ 1761677d0b8SKris Buschelman /* ----------------------------*/ 1772d4e2982SHong Zhang 178ad540459SPierre Jolivet if (lu->flg == SAME_NONZERO_PATTERN && lu->Numeric) umfpack_UMF_free_numeric(&lu->Numeric); 1792d4e2982SHong Zhang #if defined(PETSC_USE_COMPLEX) 1808592901bSBarry Smith status = umfpack_UMF_numeric(ai, aj, (double *)av, NULL, lu->Symbolic, &lu->Numeric, lu->Control, lu->Info); 1812d4e2982SHong Zhang #else 1828592901bSBarry Smith status = umfpack_UMF_numeric(ai, aj, av, lu->Symbolic, &lu->Numeric, lu->Control, lu->Info); 1838592901bSBarry Smith #endif 1849f42a82aSMatthew Knepley if (status < 0) { 1858592901bSBarry Smith umfpack_UMF_report_status(lu->Control, status); 186e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF, PETSC_ERR_LIB, "umfpack_UMF_numeric failed"); 1879f42a82aSMatthew Knepley } 1882d4e2982SHong Zhang /* report numeric factorization of A' when Control[PRL] > 3 */ 1898592901bSBarry Smith (void)umfpack_UMF_report_numeric(lu->Numeric, lu->Control); 1901677d0b8SKris Buschelman 1919566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)A)); 1929566063dSJacob Faibussowitsch PetscCall(MatDestroy(&lu->A)); 1932205254eSKarl Rupp 194ae26a541SJed Brown lu->A = A; 1952a325a84SHong Zhang lu->flg = SAME_NONZERO_PATTERN; 1962a325a84SHong Zhang lu->CleanUpUMFPACK = PETSC_TRUE; 1974b019bd2SJed Brown F->ops->solve = MatSolve_UMFPACK; 1984b019bd2SJed Brown F->ops->solvetranspose = MatSolveTranspose_UMFPACK; 199*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2001677d0b8SKris Buschelman } 2011677d0b8SKris Buschelman 202d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatLUFactorSymbolic_UMFPACK(Mat F, Mat A, IS r, IS c, const MatFactorInfo *info) 203d71ae5a4SJacob Faibussowitsch { 204ae26a541SJed Brown Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data; 205b9c12af5SBarry Smith Mat_UMFPACK *lu = (Mat_UMFPACK *)(F->data); 20626cc229bSBarry Smith PetscInt i, *ai = a->i, *aj = a->j, m = A->rmap->n, n = A->cmap->n, status, idx; 207989213a4SSatish Balay #if !defined(PETSC_USE_COMPLEX) 208ae26a541SJed Brown PetscScalar *av = a->a; 209989213a4SSatish Balay #endif 2105d0c19d7SBarry Smith const PetscInt *ra; 21126cc229bSBarry Smith const char *strategy[] = {"AUTO", "UNSYMMETRIC", "SYMMETRIC"}; 21226cc229bSBarry Smith const char *scale[] = {"NONE", "SUM", "MAX"}; 21326cc229bSBarry Smith PetscBool flg; 2141677d0b8SKris Buschelman 2151677d0b8SKris Buschelman PetscFunctionBegin; 216a643c07aSBarry Smith (F)->ops->lufactornumeric = MatLUFactorNumeric_UMFPACK; 217*3ba16761SJacob Faibussowitsch if (!n) PetscFunctionReturn(PETSC_SUCCESS); 21826cc229bSBarry Smith 21926cc229bSBarry Smith /* Set options to F */ 22026cc229bSBarry Smith PetscOptionsBegin(PetscObjectComm((PetscObject)F), ((PetscObject)F)->prefix, "UMFPACK Options", "Mat"); 22126cc229bSBarry Smith /* Control parameters used by reporting routiones */ 22226cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_prl", "Control[UMFPACK_PRL]", "None", lu->Control[UMFPACK_PRL], &lu->Control[UMFPACK_PRL], NULL)); 22326cc229bSBarry Smith 22426cc229bSBarry Smith /* Control parameters for symbolic factorization */ 22526cc229bSBarry Smith PetscCall(PetscOptionsEList("-mat_umfpack_strategy", "ordering and pivoting strategy", "None", strategy, 3, strategy[0], &idx, &flg)); 22626cc229bSBarry Smith if (flg) { 22726cc229bSBarry Smith switch (idx) { 228d71ae5a4SJacob Faibussowitsch case 0: 229d71ae5a4SJacob Faibussowitsch lu->Control[UMFPACK_STRATEGY] = UMFPACK_STRATEGY_AUTO; 230d71ae5a4SJacob Faibussowitsch break; 231d71ae5a4SJacob Faibussowitsch case 1: 232d71ae5a4SJacob Faibussowitsch lu->Control[UMFPACK_STRATEGY] = UMFPACK_STRATEGY_UNSYMMETRIC; 233d71ae5a4SJacob Faibussowitsch break; 234d71ae5a4SJacob Faibussowitsch case 2: 235d71ae5a4SJacob Faibussowitsch lu->Control[UMFPACK_STRATEGY] = UMFPACK_STRATEGY_SYMMETRIC; 236d71ae5a4SJacob Faibussowitsch break; 23726cc229bSBarry Smith } 23826cc229bSBarry Smith } 23926cc229bSBarry Smith PetscCall(PetscOptionsEList("-mat_umfpack_ordering", "Internal ordering method", "None", UmfpackOrderingTypes, PETSC_STATIC_ARRAY_LENGTH(UmfpackOrderingTypes), UmfpackOrderingTypes[(int)lu->Control[UMFPACK_ORDERING]], &idx, &flg)); 24026cc229bSBarry Smith if (flg) lu->Control[UMFPACK_ORDERING] = (int)idx; 24126cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_dense_col", "Control[UMFPACK_DENSE_COL]", "None", lu->Control[UMFPACK_DENSE_COL], &lu->Control[UMFPACK_DENSE_COL], NULL)); 24226cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_dense_row", "Control[UMFPACK_DENSE_ROW]", "None", lu->Control[UMFPACK_DENSE_ROW], &lu->Control[UMFPACK_DENSE_ROW], NULL)); 24326cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_amd_dense", "Control[UMFPACK_AMD_DENSE]", "None", lu->Control[UMFPACK_AMD_DENSE], &lu->Control[UMFPACK_AMD_DENSE], NULL)); 24426cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_block_size", "Control[UMFPACK_BLOCK_SIZE]", "None", lu->Control[UMFPACK_BLOCK_SIZE], &lu->Control[UMFPACK_BLOCK_SIZE], NULL)); 24526cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_fixq", "Control[UMFPACK_FIXQ]", "None", lu->Control[UMFPACK_FIXQ], &lu->Control[UMFPACK_FIXQ], NULL)); 24626cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_aggressive", "Control[UMFPACK_AGGRESSIVE]", "None", lu->Control[UMFPACK_AGGRESSIVE], &lu->Control[UMFPACK_AGGRESSIVE], NULL)); 24726cc229bSBarry Smith 24826cc229bSBarry Smith /* Control parameters used by numeric factorization */ 24926cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_pivot_tolerance", "Control[UMFPACK_PIVOT_TOLERANCE]", "None", lu->Control[UMFPACK_PIVOT_TOLERANCE], &lu->Control[UMFPACK_PIVOT_TOLERANCE], NULL)); 25026cc229bSBarry Smith PetscCall(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)); 25126cc229bSBarry Smith PetscCall(PetscOptionsEList("-mat_umfpack_scale", "Control[UMFPACK_SCALE]", "None", scale, 3, scale[0], &idx, &flg)); 25226cc229bSBarry Smith if (flg) { 25326cc229bSBarry Smith switch (idx) { 254d71ae5a4SJacob Faibussowitsch case 0: 255d71ae5a4SJacob Faibussowitsch lu->Control[UMFPACK_SCALE] = UMFPACK_SCALE_NONE; 256d71ae5a4SJacob Faibussowitsch break; 257d71ae5a4SJacob Faibussowitsch case 1: 258d71ae5a4SJacob Faibussowitsch lu->Control[UMFPACK_SCALE] = UMFPACK_SCALE_SUM; 259d71ae5a4SJacob Faibussowitsch break; 260d71ae5a4SJacob Faibussowitsch case 2: 261d71ae5a4SJacob Faibussowitsch lu->Control[UMFPACK_SCALE] = UMFPACK_SCALE_MAX; 262d71ae5a4SJacob Faibussowitsch break; 26326cc229bSBarry Smith } 26426cc229bSBarry Smith } 26526cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_alloc_init", "Control[UMFPACK_ALLOC_INIT]", "None", lu->Control[UMFPACK_ALLOC_INIT], &lu->Control[UMFPACK_ALLOC_INIT], NULL)); 26626cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_front_alloc_init", "Control[UMFPACK_FRONT_ALLOC_INIT]", "None", lu->Control[UMFPACK_FRONT_ALLOC_INIT], &lu->Control[UMFPACK_ALLOC_INIT], NULL)); 26726cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_droptol", "Control[UMFPACK_DROPTOL]", "None", lu->Control[UMFPACK_DROPTOL], &lu->Control[UMFPACK_DROPTOL], NULL)); 26826cc229bSBarry Smith 26926cc229bSBarry Smith /* Control parameters used by solve */ 27026cc229bSBarry Smith PetscCall(PetscOptionsReal("-mat_umfpack_irstep", "Control[UMFPACK_IRSTEP]", "None", lu->Control[UMFPACK_IRSTEP], &lu->Control[UMFPACK_IRSTEP], NULL)); 27126cc229bSBarry Smith PetscOptionsEnd(); 27226cc229bSBarry Smith 2732c7c0729SBarry Smith if (r) { 2749566063dSJacob Faibussowitsch PetscCall(ISGetIndices(r, &ra)); 2759566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(m, &lu->perm_c)); 2762d4e2982SHong Zhang /* we cannot simply memcpy on 64 bit archs */ 2772d4e2982SHong Zhang for (i = 0; i < m; i++) lu->perm_c[i] = ra[i]; 2789566063dSJacob Faibussowitsch PetscCall(ISRestoreIndices(r, &ra)); 2791677d0b8SKris Buschelman } 2801677d0b8SKris Buschelman 2811677d0b8SKris Buschelman /* print the control parameters */ 2828592901bSBarry Smith if (lu->Control[UMFPACK_PRL] > 1) umfpack_UMF_report_control(lu->Control); 2831677d0b8SKris Buschelman 2841677d0b8SKris Buschelman /* symbolic factorization of A' */ 2851677d0b8SKris Buschelman /* ---------------------------------------------------------------------- */ 2862c7c0729SBarry Smith if (r) { /* use Petsc row ordering */ 2878592901bSBarry Smith #if !defined(PETSC_USE_COMPLEX) 288ae26a541SJed Brown status = umfpack_UMF_qsymbolic(n, m, ai, aj, av, lu->perm_c, &lu->Symbolic, lu->Control, lu->Info); 2892d4e2982SHong Zhang #else 290b0043f70SBarry Smith status = umfpack_UMF_qsymbolic(n, m, ai, aj, NULL, NULL, lu->perm_c, &lu->Symbolic, lu->Control, lu->Info); 2918592901bSBarry Smith #endif 2922d4e2982SHong Zhang } else { /* use Umfpack col ordering */ 2938592901bSBarry Smith #if !defined(PETSC_USE_COMPLEX) 294ae26a541SJed Brown status = umfpack_UMF_symbolic(n, m, ai, aj, av, &lu->Symbolic, lu->Control, lu->Info); 2958592901bSBarry Smith #else 296ae26a541SJed Brown status = umfpack_UMF_symbolic(n, m, ai, aj, NULL, NULL, &lu->Symbolic, lu->Control, lu->Info); 2978592901bSBarry Smith #endif 2982d4e2982SHong Zhang } 2992d4e2982SHong Zhang if (status < 0) { 3008592901bSBarry Smith umfpack_UMF_report_info(lu->Control, lu->Info); 3018592901bSBarry Smith umfpack_UMF_report_status(lu->Control, status); 302e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF, PETSC_ERR_LIB, "umfpack_UMF_symbolic failed"); 3032d4e2982SHong Zhang } 3042d4e2982SHong Zhang /* report sumbolic factorization of A' when Control[PRL] > 3 */ 3058592901bSBarry Smith (void)umfpack_UMF_report_symbolic(lu->Symbolic, lu->Control); 3061677d0b8SKris Buschelman 3071677d0b8SKris Buschelman lu->flg = DIFFERENT_NONZERO_PATTERN; 308e1b4c3a1SKris Buschelman lu->CleanUpUMFPACK = PETSC_TRUE; 309*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3101677d0b8SKris Buschelman } 3111677d0b8SKris Buschelman 312d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatView_Info_UMFPACK(Mat A, PetscViewer viewer) 313d71ae5a4SJacob Faibussowitsch { 314b9c12af5SBarry Smith Mat_UMFPACK *lu = (Mat_UMFPACK *)A->data; 315f0c56d0fSKris Buschelman 3161677d0b8SKris Buschelman PetscFunctionBegin; 3171677d0b8SKris Buschelman /* check if matrix is UMFPACK type */ 318*3ba16761SJacob Faibussowitsch if (A->ops->solve != MatSolve_UMFPACK) PetscFunctionReturn(PETSC_SUCCESS); 3191677d0b8SKris Buschelman 3209566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "UMFPACK run parameters:\n")); 3211677d0b8SKris Buschelman /* Control parameters used by reporting routiones */ 3229566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_PRL]: %g\n", lu->Control[UMFPACK_PRL])); 3231677d0b8SKris Buschelman 3241677d0b8SKris Buschelman /* Control parameters used by symbolic factorization */ 3259566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_STRATEGY]: %g\n", lu->Control[UMFPACK_STRATEGY])); 3269566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_DENSE_COL]: %g\n", lu->Control[UMFPACK_DENSE_COL])); 3279566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_DENSE_ROW]: %g\n", lu->Control[UMFPACK_DENSE_ROW])); 3289566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_AMD_DENSE]: %g\n", lu->Control[UMFPACK_AMD_DENSE])); 3299566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_BLOCK_SIZE]: %g\n", lu->Control[UMFPACK_BLOCK_SIZE])); 3309566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_FIXQ]: %g\n", lu->Control[UMFPACK_FIXQ])); 3319566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_AGGRESSIVE]: %g\n", lu->Control[UMFPACK_AGGRESSIVE])); 3321677d0b8SKris Buschelman 3331677d0b8SKris Buschelman /* Control parameters used by numeric factorization */ 3349566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_PIVOT_TOLERANCE]: %g\n", lu->Control[UMFPACK_PIVOT_TOLERANCE])); 3359566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_SYM_PIVOT_TOLERANCE]: %g\n", lu->Control[UMFPACK_SYM_PIVOT_TOLERANCE])); 3369566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_SCALE]: %g\n", lu->Control[UMFPACK_SCALE])); 3379566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_ALLOC_INIT]: %g\n", lu->Control[UMFPACK_ALLOC_INIT])); 3389566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_DROPTOL]: %g\n", lu->Control[UMFPACK_DROPTOL])); 3391677d0b8SKris Buschelman 3401677d0b8SKris Buschelman /* Control parameters used by solve */ 3419566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_IRSTEP]: %g\n", lu->Control[UMFPACK_IRSTEP])); 3421677d0b8SKris Buschelman 3431677d0b8SKris Buschelman /* mat ordering */ 34448a46eb9SPierre Jolivet if (!lu->perm_c) PetscCall(PetscViewerASCIIPrintf(viewer, " Control[UMFPACK_ORDERING]: %s (not using the PETSc ordering)\n", UmfpackOrderingTypes[(int)lu->Control[UMFPACK_ORDERING]])); 345*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3461677d0b8SKris Buschelman } 3471677d0b8SKris Buschelman 348d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatView_UMFPACK(Mat A, PetscViewer viewer) 349d71ae5a4SJacob Faibussowitsch { 350ace3abfcSBarry Smith PetscBool iascii; 351d844382dSKris Buschelman PetscViewerFormat format; 352d844382dSKris Buschelman 353d844382dSKris Buschelman PetscFunctionBegin; 3549566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii)); 35532077d6dSBarry Smith if (iascii) { 3569566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(viewer, &format)); 35748a46eb9SPierre Jolivet if (format == PETSC_VIEWER_ASCII_INFO) PetscCall(MatView_Info_UMFPACK(A, viewer)); 358d844382dSKris Buschelman } 359*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 360d844382dSKris Buschelman } 361d844382dSKris Buschelman 362d71ae5a4SJacob Faibussowitsch PetscErrorCode MatFactorGetSolverType_seqaij_umfpack(Mat A, MatSolverType *type) 363d71ae5a4SJacob Faibussowitsch { 36435bd34faSBarry Smith PetscFunctionBegin; 3652692d6eeSBarry Smith *type = MATSOLVERUMFPACK; 366*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 36735bd34faSBarry Smith } 36835bd34faSBarry Smith 3692f71e704SKris Buschelman /*MC 37011a5261eSBarry Smith MATSOLVERUMFPACK = "umfpack" - A matrix type providing direct solvers, LU, for sequential matrices 3712f71e704SKris Buschelman via the external package UMFPACK. 3722f71e704SKris Buschelman 373c2b89b5dSBarry Smith Use ./configure --download-suitesparse to install PETSc to use UMFPACK 374c2b89b5dSBarry Smith 3753ca39a21SBarry Smith Use -pc_type lu -pc_factor_mat_solver_type umfpack to use this direct solver 3762f71e704SKris Buschelman 3772f71e704SKris Buschelman Consult UMFPACK documentation for more information about the Control parameters 3782f71e704SKris Buschelman which correspond to the options database keys below. 3792f71e704SKris Buschelman 3802f71e704SKris Buschelman Options Database Keys: 381ba41fbb6SBarry Smith + -mat_umfpack_ordering - CHOLMOD, AMD, GIVEN, METIS, BEST, NONE 382ba41fbb6SBarry Smith . -mat_umfpack_prl - UMFPACK print level: Control[UMFPACK_PRL] 383e08999f5SMatthew G Knepley . -mat_umfpack_strategy <AUTO> - (choose one of) AUTO UNSYMMETRIC SYMMETRIC 2BY2 3842f71e704SKris Buschelman . -mat_umfpack_dense_col <alpha_c> - UMFPACK dense column threshold: Control[UMFPACK_DENSE_COL] 385e08999f5SMatthew G Knepley . -mat_umfpack_dense_row <0.2> - Control[UMFPACK_DENSE_ROW] 386e08999f5SMatthew G Knepley . -mat_umfpack_amd_dense <10> - Control[UMFPACK_AMD_DENSE] 3872f71e704SKris Buschelman . -mat_umfpack_block_size <bs> - UMFPACK block size for BLAS-Level 3 calls: Control[UMFPACK_BLOCK_SIZE] 388e08999f5SMatthew G Knepley . -mat_umfpack_2by2_tolerance <0.01> - Control[UMFPACK_2BY2_TOLERANCE] 389e08999f5SMatthew G Knepley . -mat_umfpack_fixq <0> - Control[UMFPACK_FIXQ] 390e08999f5SMatthew G Knepley . -mat_umfpack_aggressive <1> - Control[UMFPACK_AGGRESSIVE] 3912f71e704SKris Buschelman . -mat_umfpack_pivot_tolerance <delta> - UMFPACK partial pivot tolerance: Control[UMFPACK_PIVOT_TOLERANCE] 392e08999f5SMatthew G Knepley . -mat_umfpack_sym_pivot_tolerance <0.001> - Control[UMFPACK_SYM_PIVOT_TOLERANCE] 393e08999f5SMatthew G Knepley . -mat_umfpack_scale <NONE> - (choose one of) NONE SUM MAX 3942f71e704SKris Buschelman . -mat_umfpack_alloc_init <delta> - UMFPACK factorized matrix allocation modifier: Control[UMFPACK_ALLOC_INIT] 395e08999f5SMatthew G Knepley . -mat_umfpack_droptol <0> - Control[UMFPACK_DROPTOL] 3962f71e704SKris Buschelman - -mat_umfpack_irstep <maxit> - UMFPACK maximum number of iterative refinement steps: Control[UMFPACK_IRSTEP] 3972f71e704SKris Buschelman 3982f71e704SKris Buschelman Level: beginner 399a364b7d2SBarry Smith 400a364b7d2SBarry Smith Note: UMFPACK is part of SuiteSparse http://faculty.cse.tamu.edu/davis/suitesparse.html 4012f71e704SKris Buschelman 402db781477SPatrick Sanan .seealso: `PCLU`, `MATSOLVERSUPERLU`, `MATSOLVERMUMPS`, `PCFactorSetMatSolverType()`, `MatSolverType` 4032f71e704SKris Buschelman M*/ 404b2573a8aSBarry Smith 405d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode MatGetFactor_seqaij_umfpack(Mat A, MatFactorType ftype, Mat *F) 406d71ae5a4SJacob Faibussowitsch { 4073519fcdcSHong Zhang Mat B; 4083519fcdcSHong Zhang Mat_UMFPACK *lu; 40926cc229bSBarry Smith PetscInt m = A->rmap->n, n = A->cmap->n; 4102d4e2982SHong Zhang 4113519fcdcSHong Zhang PetscFunctionBegin; 4123519fcdcSHong Zhang /* Create the factorization matrix F */ 4139566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B)); 4149566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B, PETSC_DECIDE, PETSC_DECIDE, m, n)); 4159566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy("umfpack", &((PetscObject)B)->type_name)); 4169566063dSJacob Faibussowitsch PetscCall(MatSetUp(B)); 417b9c12af5SBarry Smith 4184dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&lu)); 4192205254eSKarl Rupp 420b9c12af5SBarry Smith B->data = lu; 421b9c12af5SBarry Smith B->ops->getinfo = MatGetInfo_External; 4223519fcdcSHong Zhang B->ops->lufactorsymbolic = MatLUFactorSymbolic_UMFPACK; 4233519fcdcSHong Zhang B->ops->destroy = MatDestroy_UMFPACK; 4243519fcdcSHong Zhang B->ops->view = MatView_UMFPACK; 4251aef8b4cSStefano Zampini B->ops->matsolve = NULL; 4262205254eSKarl Rupp 4279566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_seqaij_umfpack)); 4282205254eSKarl Rupp 429d5f3da31SBarry Smith B->factortype = MAT_FACTOR_LU; 4303519fcdcSHong Zhang B->assembled = PETSC_TRUE; /* required by -ksp_view */ 4313519fcdcSHong Zhang B->preallocated = PETSC_TRUE; 4323519fcdcSHong Zhang 4339566063dSJacob Faibussowitsch PetscCall(PetscFree(B->solvertype)); 4349566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATSOLVERUMFPACK, &B->solvertype)); 435f73b0415SBarry Smith B->canuseordering = PETSC_TRUE; 4369566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU])); 43700c67f3bSHong Zhang 4383519fcdcSHong Zhang /* initializations */ 4393519fcdcSHong Zhang /* ------------------------------------------------*/ 4403519fcdcSHong Zhang /* get the default control parameters */ 4418592901bSBarry Smith umfpack_UMF_defaults(lu->Control); 4420298fd71SBarry Smith lu->perm_c = NULL; /* use defaul UMFPACK col permutation */ 4433519fcdcSHong Zhang lu->Control[UMFPACK_IRSTEP] = 0; /* max num of iterative refinement steps to attempt */ 4443519fcdcSHong Zhang 4453519fcdcSHong Zhang *F = B; 446*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 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 454d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_SuiteSparse(void) 455d71ae5a4SJacob Faibussowitsch { 45642c9c57cSBarry Smith PetscFunctionBegin; 4579566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERUMFPACK, MATSEQAIJ, MAT_FACTOR_LU, MatGetFactor_seqaij_umfpack)); 4589566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERCHOLMOD, MATSEQAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_seqaij_cholmod)); 4599566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERCHOLMOD, MATSEQSBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_seqsbaij_cholmod)); 4609566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERKLU, MATSEQAIJ, MAT_FACTOR_LU, MatGetFactor_seqaij_klu)); 4619566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERSPQR, MATSEQAIJ, MAT_FACTOR_QR, MatGetFactor_seqaij_spqr)); 46248a46eb9SPierre Jolivet if (!PetscDefined(USE_COMPLEX)) PetscCall(MatSolverTypeRegister(MATSOLVERSPQR, MATNORMAL, MAT_FACTOR_QR, MatGetFactor_seqaij_spqr)); 4639566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERSPQR, MATNORMALHERMITIAN, MAT_FACTOR_QR, MatGetFactor_seqaij_spqr)); 464*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 46542c9c57cSBarry Smith } 466