1be1d678aSKris Buschelman #define PETSCMAT_DLL 214b396bbSKris Buschelman 35a46d3faSBarry Smith /* -------------------------------------------------------------------- 45a46d3faSBarry Smith 55a46d3faSBarry Smith This file implements a subclass of the SeqAIJ matrix class that uses 65a46d3faSBarry Smith the SuperLU 3.0 sparse solver. You can use this as a starting point for 75a46d3faSBarry Smith implementing your own subclass of a PETSc matrix class. 85a46d3faSBarry Smith 95a46d3faSBarry Smith This demonstrates a way to make an implementation inheritence of a PETSc 105a46d3faSBarry Smith matrix type. This means constructing a new matrix type (SuperLU) by changing some 115a46d3faSBarry Smith of the methods of the previous type (SeqAIJ), adding additional data, and possibly 125a46d3faSBarry Smith additional method. (See any book on object oriented programming). 1314b396bbSKris Buschelman */ 1414b396bbSKris Buschelman 155a46d3faSBarry Smith /* 165a46d3faSBarry Smith Defines the data structure for the base matrix type (SeqAIJ) 175a46d3faSBarry Smith */ 1814b396bbSKris Buschelman #include "src/mat/impls/aij/seq/aij.h" 1914b396bbSKris Buschelman 205a46d3faSBarry Smith /* 215a46d3faSBarry Smith SuperLU include files 225a46d3faSBarry Smith */ 2314b396bbSKris Buschelman EXTERN_C_BEGIN 2414b396bbSKris Buschelman #if defined(PETSC_USE_COMPLEX) 2566f57492SHong Zhang #include "slu_zdefs.h" 2614b396bbSKris Buschelman #else 2766f57492SHong Zhang #include "slu_ddefs.h" 2814b396bbSKris Buschelman #endif 2966f57492SHong Zhang #include "slu_util.h" 3014b396bbSKris Buschelman EXTERN_C_END 3114b396bbSKris Buschelman 325a46d3faSBarry Smith /* 335a46d3faSBarry Smith This is the data we are "ADDING" to the SeqAIJ matrix type to get the SuperLU matrix type 345a46d3faSBarry Smith */ 3514b396bbSKris Buschelman typedef struct { 3675af56d4SHong Zhang SuperMatrix A,L,U,B,X; 3775af56d4SHong Zhang superlu_options_t options; 38da7a1d00SHong Zhang PetscInt *perm_c; /* column permutation vector */ 39da7a1d00SHong Zhang PetscInt *perm_r; /* row permutations from partial pivoting */ 40da7a1d00SHong Zhang PetscInt *etree; 41da7a1d00SHong Zhang PetscReal *R, *C; 4275af56d4SHong Zhang char equed[1]; 43da7a1d00SHong Zhang PetscInt lwork; 4475af56d4SHong Zhang void *work; 45da7a1d00SHong Zhang PetscReal rpg, rcond; 4675af56d4SHong Zhang mem_usage_t mem_usage; 4775af56d4SHong Zhang MatStructure flg; 4814b396bbSKris Buschelman 4914b396bbSKris Buschelman /* Flag to clean up (non-global) SuperLU objects during Destroy */ 5014b396bbSKris Buschelman PetscTruth CleanUpSuperLU; 51f0c56d0fSKris Buschelman } Mat_SuperLU; 5214b396bbSKris Buschelman 53e0e586b9SHong Zhang extern PetscErrorCode MatFactorInfo_SuperLU(Mat,PetscViewer); 54e0e586b9SHong Zhang extern PetscErrorCode MatLUFactorNumeric_SuperLU(Mat,MatFactorInfo *,Mat *); 55e0e586b9SHong Zhang extern PetscErrorCode MatDestroy_SuperLU(Mat); 56e0e586b9SHong Zhang extern PetscErrorCode MatView_SuperLU(Mat,PetscViewer); 57e0e586b9SHong Zhang extern PetscErrorCode MatAssemblyEnd_SuperLU(Mat,MatAssemblyType); 58e0e586b9SHong Zhang extern PetscErrorCode MatSolve_SuperLU(Mat,Vec,Vec); 59e0e586b9SHong Zhang extern PetscErrorCode MatSolveTranspose_SuperLU(Mat,Vec,Vec); 60e0e586b9SHong Zhang extern PetscErrorCode MatLUFactorSymbolic_SuperLU(Mat,IS,IS,MatFactorInfo *,Mat *); 61e0e586b9SHong Zhang extern PetscErrorCode MatDuplicate_SuperLU(Mat, MatDuplicateOption, Mat *); 62e0e586b9SHong Zhang 635a46d3faSBarry Smith /* 645a46d3faSBarry Smith Utility function 655a46d3faSBarry Smith */ 665a46d3faSBarry Smith #undef __FUNCT__ 675a46d3faSBarry Smith #define __FUNCT__ "MatFactorInfo_SuperLU" 685a46d3faSBarry Smith PetscErrorCode MatFactorInfo_SuperLU(Mat A,PetscViewer viewer) 695a46d3faSBarry Smith { 705a46d3faSBarry Smith Mat_SuperLU *lu= (Mat_SuperLU*)A->spptr; 715a46d3faSBarry Smith PetscErrorCode ierr; 725a46d3faSBarry Smith superlu_options_t options; 735a46d3faSBarry Smith 745a46d3faSBarry Smith PetscFunctionBegin; 755a46d3faSBarry Smith /* check if matrix is superlu_dist type */ 765a46d3faSBarry Smith if (A->ops->solve != MatSolve_SuperLU) PetscFunctionReturn(0); 775a46d3faSBarry Smith 785a46d3faSBarry Smith options = lu->options; 795a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"SuperLU run parameters:\n");CHKERRQ(ierr); 805a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Equil: %s\n",(options.Equil != NO) ? "YES": "NO");CHKERRQ(ierr); 815a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ColPerm: %D\n",options.ColPerm);CHKERRQ(ierr); 825a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," IterRefine: %D\n",options.IterRefine);CHKERRQ(ierr); 835a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," SymmetricMode: %s\n",(options.SymmetricMode != NO) ? "YES": "NO");CHKERRQ(ierr); 845a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," DiagPivotThresh: %g\n",options.DiagPivotThresh);CHKERRQ(ierr); 855a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," PivotGrowth: %s\n",(options.PivotGrowth != NO) ? "YES": "NO");CHKERRQ(ierr); 865a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ConditionNumber: %s\n",(options.ConditionNumber != NO) ? "YES": "NO");CHKERRQ(ierr); 875a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," RowPerm: %D\n",options.RowPerm);CHKERRQ(ierr); 885a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ReplaceTinyPivot: %s\n",(options.ReplaceTinyPivot != NO) ? "YES": "NO");CHKERRQ(ierr); 895a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," PrintStat: %s\n",(options.PrintStat != NO) ? "YES": "NO");CHKERRQ(ierr); 905a46d3faSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," lwork: %D\n",lu->lwork);CHKERRQ(ierr); 915a46d3faSBarry Smith 925a46d3faSBarry Smith PetscFunctionReturn(0); 935a46d3faSBarry Smith } 945a46d3faSBarry Smith 955a46d3faSBarry Smith /* 965a46d3faSBarry Smith These are the methods provided to REPLACE the corresponding methods of the 975a46d3faSBarry Smith SeqAIJ matrix class. Other methods of SeqAIJ are not replaced 985a46d3faSBarry Smith */ 995a46d3faSBarry Smith #undef __FUNCT__ 1005a46d3faSBarry Smith #define __FUNCT__ "MatLUFactorNumeric_SuperLU" 1015a46d3faSBarry Smith PetscErrorCode MatLUFactorNumeric_SuperLU(Mat A,MatFactorInfo *info,Mat *F) 1025a46d3faSBarry Smith { 1035a46d3faSBarry Smith Mat_SeqAIJ *aa = (Mat_SeqAIJ*)(A)->data; 1045a46d3faSBarry Smith Mat_SuperLU *lu = (Mat_SuperLU*)(*F)->spptr; 1055a46d3faSBarry Smith PetscErrorCode ierr; 1065a46d3faSBarry Smith PetscInt sinfo; 1075a46d3faSBarry Smith SuperLUStat_t stat; 1085a46d3faSBarry Smith PetscReal ferr, berr; 1095a46d3faSBarry Smith NCformat *Ustore; 1105a46d3faSBarry Smith SCformat *Lstore; 1115a46d3faSBarry Smith 1125a46d3faSBarry Smith PetscFunctionBegin; 1135a46d3faSBarry Smith if (lu->flg == SAME_NONZERO_PATTERN){ /* successing numerical factorization */ 1145a46d3faSBarry Smith lu->options.Fact = SamePattern; 1155a46d3faSBarry Smith /* Ref: ~SuperLU_3.0/EXAMPLE/dlinsolx2.c */ 1165a46d3faSBarry Smith Destroy_SuperMatrix_Store(&lu->A); 1175a46d3faSBarry Smith if ( lu->lwork >= 0 ) { 1185a46d3faSBarry Smith Destroy_SuperNode_Matrix(&lu->L); 1195a46d3faSBarry Smith Destroy_CompCol_Matrix(&lu->U); 1205a46d3faSBarry Smith lu->options.Fact = SamePattern; 1215a46d3faSBarry Smith } 1225a46d3faSBarry Smith } 1235a46d3faSBarry Smith 1245a46d3faSBarry Smith /* Create the SuperMatrix for lu->A=A^T: 1255a46d3faSBarry Smith Since SuperLU likes column-oriented matrices,we pass it the transpose, 1265a46d3faSBarry Smith and then solve A^T X = B in MatSolve(). */ 1275a46d3faSBarry Smith #if defined(PETSC_USE_COMPLEX) 1285a46d3faSBarry Smith zCreate_CompCol_Matrix(&lu->A,A->cmap.n,A->rmap.n,aa->nz,(doublecomplex*)aa->a,aa->j,aa->i, 1295a46d3faSBarry Smith SLU_NC,SLU_Z,SLU_GE); 1305a46d3faSBarry Smith #else 1315a46d3faSBarry Smith dCreate_CompCol_Matrix(&lu->A,A->cmap.n,A->rmap.n,aa->nz,aa->a,aa->j,aa->i, 1325a46d3faSBarry Smith SLU_NC,SLU_D,SLU_GE); 1335a46d3faSBarry Smith #endif 1345a46d3faSBarry Smith 1355a46d3faSBarry Smith /* Initialize the statistics variables. */ 1365a46d3faSBarry Smith StatInit(&stat); 1375a46d3faSBarry Smith 1385a46d3faSBarry Smith /* Numerical factorization */ 1395a46d3faSBarry Smith lu->B.ncol = 0; /* Indicate not to solve the system */ 1405a46d3faSBarry Smith #if defined(PETSC_USE_COMPLEX) 1415a46d3faSBarry Smith zgssvx(&lu->options, &lu->A, lu->perm_c, lu->perm_r, lu->etree, lu->equed, lu->R, lu->C, 1425a46d3faSBarry Smith &lu->L, &lu->U, lu->work, lu->lwork, &lu->B, &lu->X, &lu->rpg, &lu->rcond, &ferr, &berr, 1435a46d3faSBarry Smith &lu->mem_usage, &stat, &sinfo); 1445a46d3faSBarry Smith #else 1455a46d3faSBarry Smith dgssvx(&lu->options, &lu->A, lu->perm_c, lu->perm_r, lu->etree, lu->equed, lu->R, lu->C, 1465a46d3faSBarry Smith &lu->L, &lu->U, lu->work, lu->lwork, &lu->B, &lu->X, &lu->rpg, &lu->rcond, &ferr, &berr, 1475a46d3faSBarry Smith &lu->mem_usage, &stat, &sinfo); 1485a46d3faSBarry Smith #endif 1495a46d3faSBarry Smith if ( !sinfo || sinfo == lu->A.ncol+1 ) { 1505a46d3faSBarry Smith if ( lu->options.PivotGrowth ) 1515a46d3faSBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," Recip. pivot growth = %e\n", lu->rpg); 1525a46d3faSBarry Smith if ( lu->options.ConditionNumber ) 1535a46d3faSBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," Recip. condition number = %e\n", lu->rcond); 1545a46d3faSBarry Smith } else if ( sinfo > 0 ){ 1555a46d3faSBarry Smith if ( lu->lwork == -1 ) { 1565a46d3faSBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," ** Estimated memory: %D bytes\n", sinfo - lu->A.ncol); 1575a46d3faSBarry Smith } else { 1585a46d3faSBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," Warning: gssvx() returns info %D\n",sinfo); 1595a46d3faSBarry Smith } 1605a46d3faSBarry Smith } else { /* sinfo < 0 */ 1615a46d3faSBarry Smith SETERRQ2(PETSC_ERR_LIB, "info = %D, the %D-th argument in gssvx() had an illegal value", sinfo,-sinfo); 1625a46d3faSBarry Smith } 1635a46d3faSBarry Smith 1645a46d3faSBarry Smith if ( lu->options.PrintStat ) { 1655a46d3faSBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF,"MatLUFactorNumeric_SuperLU():\n"); 1665a46d3faSBarry Smith StatPrint(&stat); 1675a46d3faSBarry Smith Lstore = (SCformat *) lu->L.Store; 1685a46d3faSBarry Smith Ustore = (NCformat *) lu->U.Store; 1695a46d3faSBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," No of nonzeros in factor L = %D\n", Lstore->nnz); 1705a46d3faSBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," No of nonzeros in factor U = %D\n", Ustore->nnz); 1715a46d3faSBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," No of nonzeros in L+U = %D\n", Lstore->nnz + Ustore->nnz - lu->A.ncol); 1725a46d3faSBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," L\\U MB %.3f\ttotal MB needed %.3f\texpansions %D\n", 1735a46d3faSBarry Smith lu->mem_usage.for_lu/1e6, lu->mem_usage.total_needed/1e6, 1745a46d3faSBarry Smith lu->mem_usage.expansions); 1755a46d3faSBarry Smith } 1765a46d3faSBarry Smith StatFree(&stat); 1775a46d3faSBarry Smith 1785a46d3faSBarry Smith lu->flg = SAME_NONZERO_PATTERN; 1795a46d3faSBarry Smith PetscFunctionReturn(0); 1805a46d3faSBarry Smith } 1815a46d3faSBarry Smith 18214b396bbSKris Buschelman #undef __FUNCT__ 183f0c56d0fSKris Buschelman #define __FUNCT__ "MatDestroy_SuperLU" 184dfbe8321SBarry Smith PetscErrorCode MatDestroy_SuperLU(Mat A) 18514b396bbSKris Buschelman { 186dfbe8321SBarry Smith PetscErrorCode ierr; 187f0c56d0fSKris Buschelman Mat_SuperLU *lu=(Mat_SuperLU*)A->spptr; 18814b396bbSKris Buschelman 18914b396bbSKris Buschelman PetscFunctionBegin; 19075af56d4SHong Zhang if (lu->CleanUpSuperLU) { /* Free the SuperLU datastructures */ 19175af56d4SHong Zhang Destroy_SuperMatrix_Store(&lu->A); 19275af56d4SHong Zhang Destroy_SuperMatrix_Store(&lu->B); 19375af56d4SHong Zhang Destroy_SuperMatrix_Store(&lu->X); 1949ce50919SHong Zhang 1959ce50919SHong Zhang ierr = PetscFree(lu->etree);CHKERRQ(ierr); 1969ce50919SHong Zhang ierr = PetscFree(lu->perm_r);CHKERRQ(ierr); 1979ce50919SHong Zhang ierr = PetscFree(lu->perm_c);CHKERRQ(ierr); 1989ce50919SHong Zhang ierr = PetscFree(lu->R);CHKERRQ(ierr); 1999ce50919SHong Zhang ierr = PetscFree(lu->C);CHKERRQ(ierr); 20075af56d4SHong Zhang if ( lu->lwork >= 0 ) { 201fb3e25aaSKris Buschelman Destroy_SuperNode_Matrix(&lu->L); 202fb3e25aaSKris Buschelman Destroy_CompCol_Matrix(&lu->U); 20375af56d4SHong Zhang } 204fb3e25aaSKris Buschelman } 205b24902e0SBarry Smith ierr = MatDestroy_SeqAIJ(A);CHKERRQ(ierr); 20614b396bbSKris Buschelman PetscFunctionReturn(0); 20714b396bbSKris Buschelman } 20814b396bbSKris Buschelman 20914b396bbSKris Buschelman #undef __FUNCT__ 210f0c56d0fSKris Buschelman #define __FUNCT__ "MatView_SuperLU" 211dfbe8321SBarry Smith PetscErrorCode MatView_SuperLU(Mat A,PetscViewer viewer) 21214b396bbSKris Buschelman { 213dfbe8321SBarry Smith PetscErrorCode ierr; 21432077d6dSBarry Smith PetscTruth iascii; 21514b396bbSKris Buschelman PetscViewerFormat format; 216f0c56d0fSKris Buschelman Mat_SuperLU *lu=(Mat_SuperLU*)(A->spptr); 21714b396bbSKris Buschelman 21814b396bbSKris Buschelman PetscFunctionBegin; 219b24902e0SBarry Smith ierr = MatView_SeqAIJ(A,viewer);CHKERRQ(ierr); 22014b396bbSKris Buschelman 22132077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 22232077d6dSBarry Smith if (iascii) { 22314b396bbSKris Buschelman ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 2242f59403fSHong Zhang if (format == PETSC_VIEWER_ASCII_INFO) { 225f0c56d0fSKris Buschelman ierr = MatFactorInfo_SuperLU(A,viewer);CHKERRQ(ierr); 22614b396bbSKris Buschelman } 22714b396bbSKris Buschelman } 22814b396bbSKris Buschelman PetscFunctionReturn(0); 22914b396bbSKris Buschelman } 23014b396bbSKris Buschelman 23114b396bbSKris Buschelman 23214b396bbSKris Buschelman #undef __FUNCT__ 233c29e884eSHong Zhang #define __FUNCT__ "MatSolve_SuperLU_Private" 234c29e884eSHong Zhang PetscErrorCode MatSolve_SuperLU_Private(Mat A,Vec b,Vec x) 23514b396bbSKris Buschelman { 236f0c56d0fSKris Buschelman Mat_SuperLU *lu = (Mat_SuperLU*)A->spptr; 23775af56d4SHong Zhang PetscScalar *barray,*xarray; 238dfbe8321SBarry Smith PetscErrorCode ierr; 239da7a1d00SHong Zhang PetscInt info,i; 24075af56d4SHong Zhang SuperLUStat_t stat; 241da7a1d00SHong Zhang PetscReal ferr,berr; 24214b396bbSKris Buschelman 24314b396bbSKris Buschelman PetscFunctionBegin; 244937a6b0eSHong Zhang if ( lu->lwork == -1 ) { 245937a6b0eSHong Zhang PetscFunctionReturn(0); 246937a6b0eSHong Zhang } 24775af56d4SHong Zhang lu->B.ncol = 1; /* Set the number of right-hand side */ 24875af56d4SHong Zhang ierr = VecGetArray(b,&barray);CHKERRQ(ierr); 24975af56d4SHong Zhang ierr = VecGetArray(x,&xarray);CHKERRQ(ierr); 2505fe6150dSHong Zhang 2515fe6150dSHong Zhang #if defined(PETSC_USE_COMPLEX) 2525fe6150dSHong Zhang ((DNformat*)lu->B.Store)->nzval = (doublecomplex*)barray; 2535fe6150dSHong Zhang ((DNformat*)lu->X.Store)->nzval = (doublecomplex*)xarray; 2545fe6150dSHong Zhang #else 2555fe6150dSHong Zhang ((DNformat*)lu->B.Store)->nzval = barray; 25675af56d4SHong Zhang ((DNformat*)lu->X.Store)->nzval = xarray; 2575fe6150dSHong Zhang #endif 25875af56d4SHong Zhang 25975af56d4SHong Zhang /* Initialize the statistics variables. */ 26075af56d4SHong Zhang StatInit(&stat); 26175af56d4SHong Zhang 26275af56d4SHong Zhang lu->options.Fact = FACTORED; /* Indicate the factored form of A is supplied. */ 2638914a3f7SHong Zhang #if defined(PETSC_USE_COMPLEX) 2648914a3f7SHong Zhang zgssvx(&lu->options, &lu->A, lu->perm_c, lu->perm_r, lu->etree, lu->equed, lu->R, lu->C, 2658914a3f7SHong Zhang &lu->L, &lu->U, lu->work, lu->lwork, &lu->B, &lu->X, &lu->rpg, &lu->rcond, &ferr, &berr, 2668914a3f7SHong Zhang &lu->mem_usage, &stat, &info); 2678914a3f7SHong Zhang #else 26875af56d4SHong Zhang dgssvx(&lu->options, &lu->A, lu->perm_c, lu->perm_r, lu->etree, lu->equed, lu->R, lu->C, 26975af56d4SHong Zhang &lu->L, &lu->U, lu->work, lu->lwork, &lu->B, &lu->X, &lu->rpg, &lu->rcond, &ferr, &berr, 27075af56d4SHong Zhang &lu->mem_usage, &stat, &info); 2718914a3f7SHong Zhang #endif 27275af56d4SHong Zhang ierr = VecRestoreArray(b,&barray);CHKERRQ(ierr); 27375af56d4SHong Zhang ierr = VecRestoreArray(x,&xarray);CHKERRQ(ierr); 27475af56d4SHong Zhang 275958c9bccSBarry Smith if ( !info || info == lu->A.ncol+1 ) { 27675af56d4SHong Zhang if ( lu->options.IterRefine ) { 2778914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF,"Iterative Refinement:\n"); 2788914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," %8s%8s%16s%16s\n", "rhs", "Steps", "FERR", "BERR"); 27975af56d4SHong Zhang for (i = 0; i < 1; ++i) 2808914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," %8d%8d%16e%16e\n", i+1, stat.RefineSteps, ferr, berr); 28175af56d4SHong Zhang } 2828914a3f7SHong Zhang } else if ( info > 0 ){ 2838914a3f7SHong Zhang if ( lu->lwork == -1 ) { 28477431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," ** Estimated memory: %D bytes\n", info - lu->A.ncol); 2858914a3f7SHong Zhang } else { 28677431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," Warning: gssvx() returns info %D\n",info); 2878914a3f7SHong Zhang } 2888914a3f7SHong Zhang } else if (info < 0){ 28977431f27SBarry Smith SETERRQ2(PETSC_ERR_LIB, "info = %D, the %D-th argument in gssvx() had an illegal value", info,-info); 29014b396bbSKris Buschelman } 29114b396bbSKris Buschelman 2928914a3f7SHong Zhang if ( lu->options.PrintStat ) { 2938914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF,"MatSolve__SuperLU():\n"); 2948914a3f7SHong Zhang StatPrint(&stat); 2958914a3f7SHong Zhang } 29675af56d4SHong Zhang StatFree(&stat); 29775af56d4SHong Zhang PetscFunctionReturn(0); 29875af56d4SHong Zhang } 29914b396bbSKris Buschelman 30014b396bbSKris Buschelman #undef __FUNCT__ 301c29e884eSHong Zhang #define __FUNCT__ "MatSolve_SuperLU" 302c29e884eSHong Zhang PetscErrorCode MatSolve_SuperLU(Mat A,Vec b,Vec x) 303c29e884eSHong Zhang { 304c29e884eSHong Zhang Mat_SuperLU *lu = (Mat_SuperLU*)A->spptr; 305c29e884eSHong Zhang PetscErrorCode ierr; 306c29e884eSHong Zhang 307c29e884eSHong Zhang PetscFunctionBegin; 308c29e884eSHong Zhang lu->options.Trans = TRANS; 309c29e884eSHong Zhang ierr = MatSolve_SuperLU_Private(A,b,x);CHKERRQ(ierr); 310c29e884eSHong Zhang PetscFunctionReturn(0); 311c29e884eSHong Zhang } 312c29e884eSHong Zhang 313c29e884eSHong Zhang #undef __FUNCT__ 314c7c1fe80SHong Zhang #define __FUNCT__ "MatSolveTranspose_SuperLU" 315c7c1fe80SHong Zhang PetscErrorCode MatSolveTranspose_SuperLU(Mat A,Vec b,Vec x) 316c7c1fe80SHong Zhang { 317c7c1fe80SHong Zhang Mat_SuperLU *lu = (Mat_SuperLU*)A->spptr; 318c7c1fe80SHong Zhang PetscErrorCode ierr; 319c7c1fe80SHong Zhang 320c7c1fe80SHong Zhang PetscFunctionBegin; 321c7c1fe80SHong Zhang lu->options.Trans = NOTRANS; 322c29e884eSHong Zhang ierr = MatSolve_SuperLU_Private(A,b,x);CHKERRQ(ierr); 323c7c1fe80SHong Zhang PetscFunctionReturn(0); 324c7c1fe80SHong Zhang } 325c7c1fe80SHong Zhang 32614b396bbSKris Buschelman 32714b396bbSKris Buschelman /* 32814b396bbSKris Buschelman Note the r permutation is ignored 32914b396bbSKris Buschelman */ 33014b396bbSKris Buschelman #undef __FUNCT__ 331f0c56d0fSKris Buschelman #define __FUNCT__ "MatLUFactorSymbolic_SuperLU" 332dfbe8321SBarry Smith PetscErrorCode MatLUFactorSymbolic_SuperLU(Mat A,IS r,IS c,MatFactorInfo *info,Mat *F) 33314b396bbSKris Buschelman { 334b24902e0SBarry Smith Mat_SuperLU *lu = (Mat_SuperLU*)((*F)->spptr); 335b24902e0SBarry Smith PetscErrorCode ierr; 336b24902e0SBarry Smith PetscInt m=A->rmap.n,n=A->cmap.n,indx; 337b24902e0SBarry Smith 338b24902e0SBarry Smith PetscFunctionBegin; 339b24902e0SBarry Smith 340b24902e0SBarry Smith /* Allocate spaces (notice sizes are for the transpose) */ 341b24902e0SBarry Smith ierr = PetscMalloc(m*sizeof(PetscInt),&lu->etree);CHKERRQ(ierr); 342b24902e0SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&lu->perm_r);CHKERRQ(ierr); 343b24902e0SBarry Smith ierr = PetscMalloc(m*sizeof(PetscInt),&lu->perm_c);CHKERRQ(ierr); 344b24902e0SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&lu->R);CHKERRQ(ierr); 345b24902e0SBarry Smith ierr = PetscMalloc(m*sizeof(PetscInt),&lu->C);CHKERRQ(ierr); 346b24902e0SBarry Smith 347b24902e0SBarry Smith /* create rhs and solution x without allocate space for .Store */ 348b24902e0SBarry Smith #if defined(PETSC_USE_COMPLEX) 349b24902e0SBarry Smith zCreate_Dense_Matrix(&lu->B, m, 1, PETSC_NULL, m, SLU_DN, SLU_Z, SLU_GE); 350b24902e0SBarry Smith zCreate_Dense_Matrix(&lu->X, m, 1, PETSC_NULL, m, SLU_DN, SLU_Z, SLU_GE); 351b24902e0SBarry Smith #else 352b24902e0SBarry Smith dCreate_Dense_Matrix(&lu->B, m, 1, PETSC_NULL, m, SLU_DN, SLU_D, SLU_GE); 353b24902e0SBarry Smith dCreate_Dense_Matrix(&lu->X, m, 1, PETSC_NULL, m, SLU_DN, SLU_D, SLU_GE); 354b24902e0SBarry Smith #endif 355b24902e0SBarry Smith 356b24902e0SBarry Smith lu->flg = DIFFERENT_NONZERO_PATTERN; 357b24902e0SBarry Smith lu->CleanUpSuperLU = PETSC_TRUE; 358b24902e0SBarry Smith PetscFunctionReturn(0); 359b24902e0SBarry Smith } 360b24902e0SBarry Smith 361b24902e0SBarry Smith 362b24902e0SBarry Smith /*MC 363*5c9eb25fSBarry Smith MAT_SOLVER_SUPERLU = "superlu" - A matrix type providing direct solvers (LU) for sequential matrices 364b24902e0SBarry Smith via the external package SuperLU. 365b24902e0SBarry Smith 366*5c9eb25fSBarry Smith Use config/configure.py --download-superlu to have PETSc installed with SuperLU 367b24902e0SBarry Smith 368b24902e0SBarry Smith Options Database Keys: 369*5c9eb25fSBarry Smith + -mat_superlu_ordering <0,1,2,3> - 0: natural ordering, 370b24902e0SBarry Smith 1: MMD applied to A'*A, 371b24902e0SBarry Smith 2: MMD applied to A'+A, 372b24902e0SBarry Smith 3: COLAMD, approximate minimum degree column ordering 373b24902e0SBarry Smith . -mat_superlu_iterrefine - have SuperLU do iterative refinement after the triangular solve 374b24902e0SBarry Smith choices: NOREFINE, SINGLE, DOUBLE, EXTRA; default is NOREFINE 375b24902e0SBarry Smith - -mat_superlu_printstat - print SuperLU statistics about the factorization 376b24902e0SBarry Smith 377*5c9eb25fSBarry Smith Notes: Do not confuse this with MAT_SOLVER_SUPERLU_DIST which is for parallel sparse solves 378*5c9eb25fSBarry Smith 379b24902e0SBarry Smith Level: beginner 380b24902e0SBarry Smith 381*5c9eb25fSBarry Smith .seealso: PCLU, MAT_SOLVER_SUPERLU_DIST, MAT_SOLVER_MUMPS, MAT_SOLVER_SPOOLES 382b24902e0SBarry Smith M*/ 383b24902e0SBarry Smith 384*5c9eb25fSBarry Smith EXTERN_C_BEGIN 385b24902e0SBarry Smith #undef __FUNCT__ 386b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_superlu" 387*5c9eb25fSBarry Smith PetscErrorCode MatGetFactor_seqaij_superlu(Mat A,MatFactorType ftype,Mat *F) 388b24902e0SBarry Smith { 38914b396bbSKris Buschelman Mat B; 390f0c56d0fSKris Buschelman Mat_SuperLU *lu; 3916849ba73SBarry Smith PetscErrorCode ierr; 3922a4c71feSBarry Smith PetscInt m=A->rmap.n,n=A->cmap.n,indx; 3939ce50919SHong Zhang PetscTruth flg; 3945ca28756SHong Zhang const char *colperm[]={"NATURAL","MMD_ATA","MMD_AT_PLUS_A","COLAMD"}; /* MY_PERMC - not supported by the petsc interface yet */ 3955ca28756SHong Zhang const char *iterrefine[]={"NOREFINE", "SINGLE", "DOUBLE", "EXTRA"}; 3965ca28756SHong Zhang const char *rowperm[]={"NOROWPERM", "LargeDiag"}; /* MY_PERMC - not supported by the petsc interface yet */ 39714b396bbSKris Buschelman 39814b396bbSKris Buschelman PetscFunctionBegin; 3997adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 4002a4c71feSBarry Smith ierr = MatSetSizes(B,A->rmap.n,A->cmap.n,PETSC_DETERMINE,PETSC_DETERMINE);CHKERRQ(ierr); 4017adad957SLisandro Dalcin ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 402899d7b4fSKris Buschelman ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr); 403f0c56d0fSKris Buschelman 404f0c56d0fSKris Buschelman B->ops->lufactornumeric = MatLUFactorNumeric_SuperLU; 405b24902e0SBarry Smith B->ops->lufactorsymbolic = MatLUFactorSymbolic_SuperLU; 406f0c56d0fSKris Buschelman B->ops->solve = MatSolve_SuperLU; 407c7c1fe80SHong Zhang B->ops->solvetranspose = MatSolveTranspose_SuperLU; 408b24902e0SBarry Smith B->ops->destroy = MatDestroy_SuperLU; 409*5c9eb25fSBarry Smith B->factor = MAT_FACTOR_LU; 41094b7f48cSBarry Smith B->assembled = PETSC_TRUE; /* required by -ksp_view */ 411*5c9eb25fSBarry Smith B->preallocated = PETSC_TRUE; 41214b396bbSKris Buschelman 413b24902e0SBarry Smith ierr = PetscNewLog(B,Mat_SuperLU,&lu);CHKERRQ(ierr); 4149ce50919SHong Zhang set_default_options(&lu->options); 4158914a3f7SHong Zhang /* equilibration causes error in solve(), thus not supported here. See dgssvx.c for possible reason. */ 4168914a3f7SHong Zhang lu->options.Equil = NO; 4179ce50919SHong Zhang lu->options.PrintStat = NO; 4188914a3f7SHong Zhang lu->lwork = 0; /* allocate space internally by system malloc */ 4199ce50919SHong Zhang 4207adad957SLisandro Dalcin ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"SuperLU Options","Mat");CHKERRQ(ierr); 4218914a3f7SHong Zhang ierr = PetscOptionsEList("-mat_superlu_colperm","ColPerm","None",colperm,4,colperm[3],&indx,&flg);CHKERRQ(ierr); 4229ce50919SHong Zhang if (flg) {lu->options.ColPerm = (colperm_t)indx;} 4238914a3f7SHong Zhang ierr = PetscOptionsEList("-mat_superlu_iterrefine","IterRefine","None",iterrefine,4,iterrefine[0],&indx,&flg);CHKERRQ(ierr); 4249ce50919SHong Zhang if (flg) { lu->options.IterRefine = (IterRefine_t)indx;} 4254dc4c822SBarry Smith ierr = PetscOptionsTruth("-mat_superlu_symmetricmode","SymmetricMode","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4269ce50919SHong Zhang if (flg) lu->options.SymmetricMode = YES; 4278914a3f7SHong Zhang ierr = PetscOptionsReal("-mat_superlu_diagpivotthresh","DiagPivotThresh","None",lu->options.DiagPivotThresh,&lu->options.DiagPivotThresh,PETSC_NULL);CHKERRQ(ierr); 4284dc4c822SBarry Smith ierr = PetscOptionsTruth("-mat_superlu_pivotgrowth","PivotGrowth","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4299ce50919SHong Zhang if (flg) lu->options.PivotGrowth = YES; 4304dc4c822SBarry Smith ierr = PetscOptionsTruth("-mat_superlu_conditionnumber","ConditionNumber","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4319ce50919SHong Zhang if (flg) lu->options.ConditionNumber = YES; 4328914a3f7SHong Zhang ierr = PetscOptionsEList("-mat_superlu_rowperm","rowperm","None",rowperm,2,rowperm[0],&indx,&flg);CHKERRQ(ierr); 4339ce50919SHong Zhang if (flg) {lu->options.RowPerm = (rowperm_t)indx;} 4344dc4c822SBarry Smith ierr = PetscOptionsTruth("-mat_superlu_replacetinypivot","ReplaceTinyPivot","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4359ce50919SHong Zhang if (flg) lu->options.ReplaceTinyPivot = YES; 4364dc4c822SBarry Smith ierr = PetscOptionsTruth("-mat_superlu_printstat","PrintStat","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4379ce50919SHong Zhang if (flg) lu->options.PrintStat = YES; 4388914a3f7SHong Zhang ierr = PetscOptionsInt("-mat_superlu_lwork","size of work array in bytes used by factorization","None",lu->lwork,&lu->lwork,PETSC_NULL);CHKERRQ(ierr); 4395fe6150dSHong Zhang if (lu->lwork > 0 ){ 4405fe6150dSHong Zhang ierr = PetscMalloc(lu->lwork,&lu->work);CHKERRQ(ierr); 4415fe6150dSHong Zhang } else if (lu->lwork != 0 && lu->lwork != -1){ 44277431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," Warning: lwork %D is not supported by SUPERLU. The default lwork=0 is used.\n",lu->lwork); 4438914a3f7SHong Zhang lu->lwork = 0; 4448914a3f7SHong Zhang } 4459ce50919SHong Zhang PetscOptionsEnd(); 4469ce50919SHong Zhang 44775af56d4SHong Zhang #ifdef SUPERLU2 448*5c9eb25fSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatCreateNull","MatCreateNull_SuperLU",(void(*)(void))MatCreateNull_SuperLU);CHKERRQ(ierr); 44975af56d4SHong Zhang #endif 450*5c9eb25fSBarry Smith B->spptr = lu; 451899d7b4fSKris Buschelman *F = B; 45214b396bbSKris Buschelman PetscFunctionReturn(0); 45314b396bbSKris Buschelman } 454*5c9eb25fSBarry Smith EXTERN_C_END 455