1*be1d678aSKris Buschelman #define PETSCMAT_DLL 214b396bbSKris Buschelman 314b396bbSKris Buschelman /* 475af56d4SHong Zhang Provides an interface to the SuperLU 3.0 sparse solver 514b396bbSKris Buschelman */ 614b396bbSKris Buschelman 714b396bbSKris Buschelman #include "src/mat/impls/aij/seq/aij.h" 814b396bbSKris Buschelman 914b396bbSKris Buschelman EXTERN_C_BEGIN 1014b396bbSKris Buschelman #if defined(PETSC_USE_COMPLEX) 1114b396bbSKris Buschelman #include "zsp_defs.h" 1214b396bbSKris Buschelman #else 1314b396bbSKris Buschelman #include "dsp_defs.h" 1414b396bbSKris Buschelman #endif 1514b396bbSKris Buschelman #include "util.h" 1614b396bbSKris Buschelman EXTERN_C_END 1714b396bbSKris Buschelman 1814b396bbSKris Buschelman typedef struct { 1975af56d4SHong Zhang SuperMatrix A,L,U,B,X; 2075af56d4SHong Zhang superlu_options_t options; 21da7a1d00SHong Zhang PetscInt *perm_c; /* column permutation vector */ 22da7a1d00SHong Zhang PetscInt *perm_r; /* row permutations from partial pivoting */ 23da7a1d00SHong Zhang PetscInt *etree; 24da7a1d00SHong Zhang PetscReal *R, *C; 2575af56d4SHong Zhang char equed[1]; 26da7a1d00SHong Zhang PetscInt lwork; 2775af56d4SHong Zhang void *work; 28da7a1d00SHong Zhang PetscReal rpg, rcond; 2975af56d4SHong Zhang mem_usage_t mem_usage; 3075af56d4SHong Zhang MatStructure flg; 3114b396bbSKris Buschelman 3214b396bbSKris Buschelman /* A few function pointers for inheritance */ 336849ba73SBarry Smith PetscErrorCode (*MatDuplicate)(Mat,MatDuplicateOption,Mat*); 346849ba73SBarry Smith PetscErrorCode (*MatView)(Mat,PetscViewer); 356849ba73SBarry Smith PetscErrorCode (*MatAssemblyEnd)(Mat,MatAssemblyType); 366849ba73SBarry Smith PetscErrorCode (*MatLUFactorSymbolic)(Mat,IS,IS,MatFactorInfo*,Mat*); 376849ba73SBarry Smith PetscErrorCode (*MatDestroy)(Mat); 3814b396bbSKris Buschelman 3914b396bbSKris Buschelman /* Flag to clean up (non-global) SuperLU objects during Destroy */ 4014b396bbSKris Buschelman PetscTruth CleanUpSuperLU; 41f0c56d0fSKris Buschelman } Mat_SuperLU; 4214b396bbSKris Buschelman 4314b396bbSKris Buschelman 44dfbe8321SBarry Smith EXTERN PetscErrorCode MatFactorInfo_SuperLU(Mat,PetscViewer); 45dfbe8321SBarry Smith EXTERN PetscErrorCode MatLUFactorSymbolic_SuperLU(Mat,IS,IS,MatFactorInfo*,Mat*); 46b0592601SKris Buschelman 47b0592601SKris Buschelman EXTERN_C_BEGIN 48*be1d678aSKris Buschelman EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SuperLU_SeqAIJ(Mat,const MatType,MatReuse,Mat*); 49*be1d678aSKris Buschelman EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqAIJ_SuperLU(Mat,const MatType,MatReuse,Mat*); 50b0592601SKris Buschelman EXTERN_C_END 5114b396bbSKris Buschelman 5214b396bbSKris Buschelman #undef __FUNCT__ 53f0c56d0fSKris Buschelman #define __FUNCT__ "MatDestroy_SuperLU" 54dfbe8321SBarry Smith PetscErrorCode MatDestroy_SuperLU(Mat A) 5514b396bbSKris Buschelman { 56dfbe8321SBarry Smith PetscErrorCode ierr; 57f0c56d0fSKris Buschelman Mat_SuperLU *lu=(Mat_SuperLU*)A->spptr; 5814b396bbSKris Buschelman 5914b396bbSKris Buschelman PetscFunctionBegin; 6075af56d4SHong Zhang if (lu->CleanUpSuperLU) { /* Free the SuperLU datastructures */ 6175af56d4SHong Zhang Destroy_SuperMatrix_Store(&lu->A); 6275af56d4SHong Zhang Destroy_SuperMatrix_Store(&lu->B); 6375af56d4SHong Zhang Destroy_SuperMatrix_Store(&lu->X); 649ce50919SHong Zhang 659ce50919SHong Zhang ierr = PetscFree(lu->etree);CHKERRQ(ierr); 669ce50919SHong Zhang ierr = PetscFree(lu->perm_r);CHKERRQ(ierr); 679ce50919SHong Zhang ierr = PetscFree(lu->perm_c);CHKERRQ(ierr); 689ce50919SHong Zhang ierr = PetscFree(lu->R);CHKERRQ(ierr); 699ce50919SHong Zhang ierr = PetscFree(lu->C);CHKERRQ(ierr); 7075af56d4SHong Zhang if ( lu->lwork >= 0 ) { 71fb3e25aaSKris Buschelman Destroy_SuperNode_Matrix(&lu->L); 72fb3e25aaSKris Buschelman Destroy_CompCol_Matrix(&lu->U); 7375af56d4SHong Zhang } 74fb3e25aaSKris Buschelman } 75ceb03754SKris Buschelman ierr = MatConvert_SuperLU_SeqAIJ(A,MATSEQAIJ,MAT_REUSE_MATRIX,&A);CHKERRQ(ierr); 76fb3e25aaSKris Buschelman ierr = (*A->ops->destroy)(A);CHKERRQ(ierr); 7714b396bbSKris Buschelman PetscFunctionReturn(0); 7814b396bbSKris Buschelman } 7914b396bbSKris Buschelman 8014b396bbSKris Buschelman #undef __FUNCT__ 81f0c56d0fSKris Buschelman #define __FUNCT__ "MatView_SuperLU" 82dfbe8321SBarry Smith PetscErrorCode MatView_SuperLU(Mat A,PetscViewer viewer) 8314b396bbSKris Buschelman { 84dfbe8321SBarry Smith PetscErrorCode ierr; 8532077d6dSBarry Smith PetscTruth iascii; 8614b396bbSKris Buschelman PetscViewerFormat format; 87f0c56d0fSKris Buschelman Mat_SuperLU *lu=(Mat_SuperLU*)(A->spptr); 8814b396bbSKris Buschelman 8914b396bbSKris Buschelman PetscFunctionBegin; 9014b396bbSKris Buschelman ierr = (*lu->MatView)(A,viewer);CHKERRQ(ierr); 9114b396bbSKris Buschelman 9232077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 9332077d6dSBarry Smith if (iascii) { 9414b396bbSKris Buschelman ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 9514b396bbSKris Buschelman if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 96f0c56d0fSKris Buschelman ierr = MatFactorInfo_SuperLU(A,viewer);CHKERRQ(ierr); 9714b396bbSKris Buschelman } 9814b396bbSKris Buschelman } 9914b396bbSKris Buschelman PetscFunctionReturn(0); 10014b396bbSKris Buschelman } 10114b396bbSKris Buschelman 10214b396bbSKris Buschelman #undef __FUNCT__ 103f0c56d0fSKris Buschelman #define __FUNCT__ "MatAssemblyEnd_SuperLU" 104dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SuperLU(Mat A,MatAssemblyType mode) { 105dfbe8321SBarry Smith PetscErrorCode ierr; 106f0c56d0fSKris Buschelman Mat_SuperLU *lu=(Mat_SuperLU*)(A->spptr); 10714b396bbSKris Buschelman 10814b396bbSKris Buschelman PetscFunctionBegin; 10914b396bbSKris Buschelman ierr = (*lu->MatAssemblyEnd)(A,mode);CHKERRQ(ierr); 110b0592601SKris Buschelman lu->MatLUFactorSymbolic = A->ops->lufactorsymbolic; 111f0c56d0fSKris Buschelman A->ops->lufactorsymbolic = MatLUFactorSymbolic_SuperLU; 11214b396bbSKris Buschelman PetscFunctionReturn(0); 11314b396bbSKris Buschelman } 11414b396bbSKris Buschelman 11575af56d4SHong Zhang /* This function was written for SuperLU 2.0 by Matthew Knepley. Not tested for SuperLU 3.0! */ 11675af56d4SHong Zhang #ifdef SuperLU2 11714b396bbSKris Buschelman #include "src/mat/impls/dense/seq/dense.h" 11814b396bbSKris Buschelman #undef __FUNCT__ 119f0c56d0fSKris Buschelman #define __FUNCT__ "MatCreateNull_SuperLU" 120dfbe8321SBarry Smith PetscErrorCode MatCreateNull_SuperLU(Mat A,Mat *nullMat) 12114b396bbSKris Buschelman { 122f0c56d0fSKris Buschelman Mat_SuperLU *lu = (Mat_SuperLU*)A->spptr; 123da7a1d00SHong Zhang PetscInt numRows = A->m,numCols = A->n; 12414b396bbSKris Buschelman SCformat *Lstore; 125da7a1d00SHong Zhang PetscInt numNullCols,size; 12675af56d4SHong Zhang SuperLUStat_t stat; 12714b396bbSKris Buschelman #if defined(PETSC_USE_COMPLEX) 12814b396bbSKris Buschelman doublecomplex *nullVals,*workVals; 12914b396bbSKris Buschelman #else 13014b396bbSKris Buschelman PetscScalar *nullVals,*workVals; 13114b396bbSKris Buschelman #endif 132da7a1d00SHong Zhang PetscInt row,newRow,col,newCol,block,b; 1336849ba73SBarry Smith PetscErrorCode ierr; 13414b396bbSKris Buschelman 13514b396bbSKris Buschelman PetscFunctionBegin; 13614b396bbSKris Buschelman if (!A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix"); 13714b396bbSKris Buschelman numNullCols = numCols - numRows; 13814b396bbSKris Buschelman if (numNullCols < 0) SETERRQ(PETSC_ERR_ARG_WRONG,"Function only applies to underdetermined problems"); 139be5d1d56SKris Buschelman /* Create the null matrix using MATSEQDENSE explicitly */ 140878740d9SKris Buschelman ierr = MatCreate(A->comm,numRows,numNullCols,numRows,numNullCols,nullMat);CHKERRQ(ierr); 141878740d9SKris Buschelman ierr = MatSetType(*nullMat,MATSEQDENSE);CHKERRQ(ierr); 142878740d9SKris Buschelman ierr = MatSeqDenseSetPreallocation(*nullMat,PETSC_NULL);CHKERRQ(ierr); 143958c9bccSBarry Smith if (!numNullCols) { 14414b396bbSKris Buschelman ierr = MatAssemblyBegin(*nullMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 14514b396bbSKris Buschelman ierr = MatAssemblyEnd(*nullMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 14614b396bbSKris Buschelman PetscFunctionReturn(0); 14714b396bbSKris Buschelman } 14814b396bbSKris Buschelman #if defined(PETSC_USE_COMPLEX) 14914b396bbSKris Buschelman nullVals = (doublecomplex*)((Mat_SeqDense*)(*nullMat)->data)->v; 15014b396bbSKris Buschelman #else 15114b396bbSKris Buschelman nullVals = ((Mat_SeqDense*)(*nullMat)->data)->v; 15214b396bbSKris Buschelman #endif 15314b396bbSKris Buschelman /* Copy in the columns */ 15414b396bbSKris Buschelman Lstore = (SCformat*)lu->L.Store; 15514b396bbSKris Buschelman for(block = 0; block <= Lstore->nsuper; block++) { 15614b396bbSKris Buschelman newRow = Lstore->sup_to_col[block]; 15714b396bbSKris Buschelman size = Lstore->sup_to_col[block+1] - Lstore->sup_to_col[block]; 15814b396bbSKris Buschelman for(col = Lstore->rowind_colptr[newRow]; col < Lstore->rowind_colptr[newRow+1]; col++) { 15914b396bbSKris Buschelman newCol = Lstore->rowind[col]; 16014b396bbSKris Buschelman if (newCol >= numRows) { 16114b396bbSKris Buschelman for(b = 0; b < size; b++) 16214b396bbSKris Buschelman #if defined(PETSC_USE_COMPLEX) 16314b396bbSKris Buschelman nullVals[(newCol-numRows)*numRows+newRow+b] = ((doublecomplex*)Lstore->nzval)[Lstore->nzval_colptr[newRow+b]+col]; 16414b396bbSKris Buschelman #else 16514b396bbSKris Buschelman nullVals[(newCol-numRows)*numRows+newRow+b] = ((double*)Lstore->nzval)[Lstore->nzval_colptr[newRow+b]+col]; 16614b396bbSKris Buschelman #endif 16714b396bbSKris Buschelman } 16814b396bbSKris Buschelman } 16914b396bbSKris Buschelman } 17014b396bbSKris Buschelman /* Permute rhs to form P^T_c B */ 171da7a1d00SHong Zhang ierr = PetscMalloc(numRows*sizeof(PetscReal),&workVals);CHKERRQ(ierr); 17214b396bbSKris Buschelman for(b = 0; b < numNullCols; b++) { 17314b396bbSKris Buschelman for(row = 0; row < numRows; row++) workVals[lu->perm_c[row]] = nullVals[b*numRows+row]; 17414b396bbSKris Buschelman for(row = 0; row < numRows; row++) nullVals[b*numRows+row] = workVals[row]; 17514b396bbSKris Buschelman } 17614b396bbSKris Buschelman /* Backward solve the upper triangle A x = b */ 17714b396bbSKris Buschelman for(b = 0; b < numNullCols; b++) { 17814b396bbSKris Buschelman #if defined(PETSC_USE_COMPLEX) 17975af56d4SHong Zhang sp_ztrsv("L","T","U",&lu->L,&lu->U,&nullVals[b*numRows],&stat,&ierr); 18014b396bbSKris Buschelman #else 18175af56d4SHong Zhang sp_dtrsv("L","T","U",&lu->L,&lu->U,&nullVals[b*numRows],&stat,&ierr); 18214b396bbSKris Buschelman #endif 18314b396bbSKris Buschelman if (ierr < 0) 18477431f27SBarry Smith SETERRQ1(PETSC_ERR_ARG_WRONG,"The argument %D was invalid",-ierr); 18514b396bbSKris Buschelman } 18614b396bbSKris Buschelman ierr = PetscFree(workVals);CHKERRQ(ierr); 18714b396bbSKris Buschelman 18814b396bbSKris Buschelman ierr = MatAssemblyBegin(*nullMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 18914b396bbSKris Buschelman ierr = MatAssemblyEnd(*nullMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 19014b396bbSKris Buschelman PetscFunctionReturn(0); 19114b396bbSKris Buschelman } 19275af56d4SHong Zhang #endif 19314b396bbSKris Buschelman 19414b396bbSKris Buschelman #undef __FUNCT__ 195f0c56d0fSKris Buschelman #define __FUNCT__ "MatSolve_SuperLU" 196dfbe8321SBarry Smith PetscErrorCode MatSolve_SuperLU(Mat A,Vec b,Vec x) 19714b396bbSKris Buschelman { 198f0c56d0fSKris Buschelman Mat_SuperLU *lu = (Mat_SuperLU*)A->spptr; 19975af56d4SHong Zhang PetscScalar *barray,*xarray; 200dfbe8321SBarry Smith PetscErrorCode ierr; 201da7a1d00SHong Zhang PetscInt info,i; 20275af56d4SHong Zhang SuperLUStat_t stat; 203da7a1d00SHong Zhang PetscReal ferr,berr; 20414b396bbSKris Buschelman 20514b396bbSKris Buschelman PetscFunctionBegin; 206937a6b0eSHong Zhang if ( lu->lwork == -1 ) { 207937a6b0eSHong Zhang PetscFunctionReturn(0); 208937a6b0eSHong Zhang } 20975af56d4SHong Zhang lu->B.ncol = 1; /* Set the number of right-hand side */ 21075af56d4SHong Zhang ierr = VecGetArray(b,&barray);CHKERRQ(ierr); 21175af56d4SHong Zhang ierr = VecGetArray(x,&xarray);CHKERRQ(ierr); 2125fe6150dSHong Zhang 2135fe6150dSHong Zhang #if defined(PETSC_USE_COMPLEX) 2145fe6150dSHong Zhang ((DNformat*)lu->B.Store)->nzval = (doublecomplex*)barray; 2155fe6150dSHong Zhang ((DNformat*)lu->X.Store)->nzval = (doublecomplex*)xarray; 2165fe6150dSHong Zhang #else 2175fe6150dSHong Zhang ((DNformat*)lu->B.Store)->nzval = barray; 21875af56d4SHong Zhang ((DNformat*)lu->X.Store)->nzval = xarray; 2195fe6150dSHong Zhang #endif 22075af56d4SHong Zhang 22175af56d4SHong Zhang /* Initialize the statistics variables. */ 22275af56d4SHong Zhang StatInit(&stat); 22375af56d4SHong Zhang 22475af56d4SHong Zhang lu->options.Fact = FACTORED; /* Indicate the factored form of A is supplied. */ 22575af56d4SHong Zhang lu->options.Trans = TRANS; 2268914a3f7SHong Zhang #if defined(PETSC_USE_COMPLEX) 2278914a3f7SHong Zhang zgssvx(&lu->options, &lu->A, lu->perm_c, lu->perm_r, lu->etree, lu->equed, lu->R, lu->C, 2288914a3f7SHong Zhang &lu->L, &lu->U, lu->work, lu->lwork, &lu->B, &lu->X, &lu->rpg, &lu->rcond, &ferr, &berr, 2298914a3f7SHong Zhang &lu->mem_usage, &stat, &info); 2308914a3f7SHong Zhang #else 23175af56d4SHong Zhang dgssvx(&lu->options, &lu->A, lu->perm_c, lu->perm_r, lu->etree, lu->equed, lu->R, lu->C, 23275af56d4SHong Zhang &lu->L, &lu->U, lu->work, lu->lwork, &lu->B, &lu->X, &lu->rpg, &lu->rcond, &ferr, &berr, 23375af56d4SHong Zhang &lu->mem_usage, &stat, &info); 2348914a3f7SHong Zhang #endif 23575af56d4SHong Zhang ierr = VecRestoreArray(b,&barray);CHKERRQ(ierr); 23675af56d4SHong Zhang ierr = VecRestoreArray(x,&xarray);CHKERRQ(ierr); 23775af56d4SHong Zhang 238958c9bccSBarry Smith if ( !info || info == lu->A.ncol+1 ) { 23975af56d4SHong Zhang if ( lu->options.IterRefine ) { 2408914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF,"Iterative Refinement:\n"); 2418914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," %8s%8s%16s%16s\n", "rhs", "Steps", "FERR", "BERR"); 24275af56d4SHong Zhang for (i = 0; i < 1; ++i) 2438914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," %8d%8d%16e%16e\n", i+1, stat.RefineSteps, ferr, berr); 24475af56d4SHong Zhang } 2458914a3f7SHong Zhang } else if ( info > 0 ){ 2468914a3f7SHong Zhang if ( lu->lwork == -1 ) { 24777431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," ** Estimated memory: %D bytes\n", info - lu->A.ncol); 2488914a3f7SHong Zhang } else { 24977431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," Warning: gssvx() returns info %D\n",info); 2508914a3f7SHong Zhang } 2518914a3f7SHong Zhang } else if (info < 0){ 25277431f27SBarry Smith SETERRQ2(PETSC_ERR_LIB, "info = %D, the %D-th argument in gssvx() had an illegal value", info,-info); 25314b396bbSKris Buschelman } 25414b396bbSKris Buschelman 2558914a3f7SHong Zhang if ( lu->options.PrintStat ) { 2568914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF,"MatSolve__SuperLU():\n"); 2578914a3f7SHong Zhang StatPrint(&stat); 2588914a3f7SHong Zhang } 25975af56d4SHong Zhang StatFree(&stat); 26075af56d4SHong Zhang PetscFunctionReturn(0); 26175af56d4SHong Zhang } 26214b396bbSKris Buschelman 26314b396bbSKris Buschelman #undef __FUNCT__ 264f0c56d0fSKris Buschelman #define __FUNCT__ "MatLUFactorNumeric_SuperLU" 265af281ebdSHong Zhang PetscErrorCode MatLUFactorNumeric_SuperLU(Mat A,MatFactorInfo *info,Mat *F) 26614b396bbSKris Buschelman { 26714b396bbSKris Buschelman Mat_SeqAIJ *aa = (Mat_SeqAIJ*)(A)->data; 268f0c56d0fSKris Buschelman Mat_SuperLU *lu = (Mat_SuperLU*)(*F)->spptr; 269dfbe8321SBarry Smith PetscErrorCode ierr; 270da7a1d00SHong Zhang PetscInt sinfo; 27175af56d4SHong Zhang SuperLUStat_t stat; 272da7a1d00SHong Zhang PetscReal ferr, berr; 2738914a3f7SHong Zhang NCformat *Ustore; 2748914a3f7SHong Zhang SCformat *Lstore; 27514b396bbSKris Buschelman 27614b396bbSKris Buschelman PetscFunctionBegin; 2779ce50919SHong Zhang if (lu->flg == SAME_NONZERO_PATTERN){ /* successing numerical factorization */ 2789ce50919SHong Zhang lu->options.Fact = SamePattern; 27975af56d4SHong Zhang /* Ref: ~SuperLU_3.0/EXAMPLE/dlinsolx2.c */ 28075af56d4SHong Zhang Destroy_SuperMatrix_Store(&lu->A); 2819ce50919SHong Zhang if ( lu->lwork >= 0 ) { 28275af56d4SHong Zhang Destroy_SuperNode_Matrix(&lu->L); 28375af56d4SHong Zhang Destroy_CompCol_Matrix(&lu->U); 28475af56d4SHong Zhang lu->options.Fact = SamePattern; 28514b396bbSKris Buschelman } 28675af56d4SHong Zhang } 28714b396bbSKris Buschelman 28875af56d4SHong Zhang /* Create the SuperMatrix for lu->A=A^T: 28975af56d4SHong Zhang Since SuperLU likes column-oriented matrices,we pass it the transpose, 29075af56d4SHong Zhang and then solve A^T X = B in MatSolve(). */ 29175af56d4SHong Zhang #if defined(PETSC_USE_COMPLEX) 2925fe6150dSHong Zhang zCreate_CompCol_Matrix(&lu->A,A->n,A->m,aa->nz,(doublecomplex*)aa->a,aa->j,aa->i, 29375af56d4SHong Zhang SLU_NC,SLU_Z,SLU_GE); 29475af56d4SHong Zhang #else 29575af56d4SHong Zhang dCreate_CompCol_Matrix(&lu->A,A->n,A->m,aa->nz,aa->a,aa->j,aa->i, 29675af56d4SHong Zhang SLU_NC,SLU_D,SLU_GE); 29775af56d4SHong Zhang #endif 29875af56d4SHong Zhang 29975af56d4SHong Zhang /* Initialize the statistics variables. */ 30075af56d4SHong Zhang StatInit(&stat); 30175af56d4SHong Zhang 3029ce50919SHong Zhang /* Numerical factorization */ 30375af56d4SHong Zhang lu->B.ncol = 0; /* Indicate not to solve the system */ 3048914a3f7SHong Zhang #if defined(PETSC_USE_COMPLEX) 3058914a3f7SHong Zhang zgssvx(&lu->options, &lu->A, lu->perm_c, lu->perm_r, lu->etree, lu->equed, lu->R, lu->C, 3068914a3f7SHong Zhang &lu->L, &lu->U, lu->work, lu->lwork, &lu->B, &lu->X, &lu->rpg, &lu->rcond, &ferr, &berr, 307da7a1d00SHong Zhang &lu->mem_usage, &stat, &sinfo); 3088914a3f7SHong Zhang #else 30975af56d4SHong Zhang dgssvx(&lu->options, &lu->A, lu->perm_c, lu->perm_r, lu->etree, lu->equed, lu->R, lu->C, 31075af56d4SHong Zhang &lu->L, &lu->U, lu->work, lu->lwork, &lu->B, &lu->X, &lu->rpg, &lu->rcond, &ferr, &berr, 311da7a1d00SHong Zhang &lu->mem_usage, &stat, &sinfo); 3128914a3f7SHong Zhang #endif 313da7a1d00SHong Zhang if ( !sinfo || sinfo == lu->A.ncol+1 ) { 3148914a3f7SHong Zhang if ( lu->options.PivotGrowth ) 3158914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," Recip. pivot growth = %e\n", lu->rpg); 31675af56d4SHong Zhang if ( lu->options.ConditionNumber ) 3178914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," Recip. condition number = %e\n", lu->rcond); 318da7a1d00SHong Zhang } else if ( sinfo > 0 ){ 3198914a3f7SHong Zhang if ( lu->lwork == -1 ) { 320da7a1d00SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," ** Estimated memory: %D bytes\n", sinfo - lu->A.ncol); 3218914a3f7SHong Zhang } else { 322da7a1d00SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," Warning: gssvx() returns info %D\n",sinfo); 3238914a3f7SHong Zhang } 324da7a1d00SHong Zhang } else { /* sinfo < 0 */ 325da7a1d00SHong Zhang SETERRQ2(PETSC_ERR_LIB, "info = %D, the %D-th argument in gssvx() had an illegal value", sinfo,-sinfo); 32675af56d4SHong Zhang } 32775af56d4SHong Zhang 3288914a3f7SHong Zhang if ( lu->options.PrintStat ) { 3298914a3f7SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF,"MatLUFactorNumeric_SuperLU():\n"); 3308914a3f7SHong Zhang StatPrint(&stat); 3318914a3f7SHong Zhang Lstore = (SCformat *) lu->L.Store; 3328914a3f7SHong Zhang Ustore = (NCformat *) lu->U.Store; 33377431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," No of nonzeros in factor L = %D\n", Lstore->nnz); 33477431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," No of nonzeros in factor U = %D\n", Ustore->nnz); 33577431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," No of nonzeros in L+U = %D\n", Lstore->nnz + Ustore->nnz - lu->A.ncol); 33677431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," L\\U MB %.3f\ttotal MB needed %.3f\texpansions %D\n", 3378914a3f7SHong Zhang lu->mem_usage.for_lu/1e6, lu->mem_usage.total_needed/1e6, 3388914a3f7SHong Zhang lu->mem_usage.expansions); 3398914a3f7SHong Zhang } 34075af56d4SHong Zhang StatFree(&stat); 34175af56d4SHong Zhang 34275af56d4SHong Zhang lu->flg = SAME_NONZERO_PATTERN; 34375af56d4SHong Zhang PetscFunctionReturn(0); 34414b396bbSKris Buschelman } 34514b396bbSKris Buschelman 34614b396bbSKris Buschelman /* 34714b396bbSKris Buschelman Note the r permutation is ignored 34814b396bbSKris Buschelman */ 34914b396bbSKris Buschelman #undef __FUNCT__ 350f0c56d0fSKris Buschelman #define __FUNCT__ "MatLUFactorSymbolic_SuperLU" 351dfbe8321SBarry Smith PetscErrorCode MatLUFactorSymbolic_SuperLU(Mat A,IS r,IS c,MatFactorInfo *info,Mat *F) 35214b396bbSKris Buschelman { 35314b396bbSKris Buschelman Mat B; 354f0c56d0fSKris Buschelman Mat_SuperLU *lu; 3556849ba73SBarry Smith PetscErrorCode ierr; 356da7a1d00SHong Zhang PetscInt m=A->m,n=A->n,indx; 3579ce50919SHong Zhang PetscTruth flg; 3585ca28756SHong Zhang const char *colperm[]={"NATURAL","MMD_ATA","MMD_AT_PLUS_A","COLAMD"}; /* MY_PERMC - not supported by the petsc interface yet */ 3595ca28756SHong Zhang const char *iterrefine[]={"NOREFINE", "SINGLE", "DOUBLE", "EXTRA"}; 3605ca28756SHong Zhang const char *rowperm[]={"NOROWPERM", "LargeDiag"}; /* MY_PERMC - not supported by the petsc interface yet */ 36114b396bbSKris Buschelman 36214b396bbSKris Buschelman PetscFunctionBegin; 363899d7b4fSKris Buschelman ierr = MatCreate(A->comm,A->m,A->n,PETSC_DETERMINE,PETSC_DETERMINE,&B);CHKERRQ(ierr); 364be5d1d56SKris Buschelman ierr = MatSetType(B,A->type_name);CHKERRQ(ierr); 365899d7b4fSKris Buschelman ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr); 366f0c56d0fSKris Buschelman 367f0c56d0fSKris Buschelman B->ops->lufactornumeric = MatLUFactorNumeric_SuperLU; 368f0c56d0fSKris Buschelman B->ops->solve = MatSolve_SuperLU; 36914b396bbSKris Buschelman B->factor = FACTOR_LU; 37094b7f48cSBarry Smith B->assembled = PETSC_TRUE; /* required by -ksp_view */ 37114b396bbSKris Buschelman 372f0c56d0fSKris Buschelman lu = (Mat_SuperLU*)(B->spptr); 3739ce50919SHong Zhang 3749ce50919SHong Zhang /* Set SuperLU options */ 3759ce50919SHong Zhang /* the default values for options argument: 3769ce50919SHong Zhang options.Fact = DOFACT; 3779ce50919SHong Zhang options.Equil = YES; 3789ce50919SHong Zhang options.ColPerm = COLAMD; 3799ce50919SHong Zhang options.DiagPivotThresh = 1.0; 3809ce50919SHong Zhang options.Trans = NOTRANS; 3819ce50919SHong Zhang options.IterRefine = NOREFINE; 3829ce50919SHong Zhang options.SymmetricMode = NO; 3839ce50919SHong Zhang options.PivotGrowth = NO; 3849ce50919SHong Zhang options.ConditionNumber = NO; 3859ce50919SHong Zhang options.PrintStat = YES; 3869ce50919SHong Zhang */ 3879ce50919SHong Zhang set_default_options(&lu->options); 3888914a3f7SHong Zhang /* equilibration causes error in solve(), thus not supported here. See dgssvx.c for possible reason. */ 3898914a3f7SHong Zhang lu->options.Equil = NO; 3909ce50919SHong Zhang lu->options.PrintStat = NO; 3918914a3f7SHong Zhang lu->lwork = 0; /* allocate space internally by system malloc */ 3929ce50919SHong Zhang 3939ce50919SHong Zhang ierr = PetscOptionsBegin(A->comm,A->prefix,"SuperLU Options","Mat");CHKERRQ(ierr); 3949ce50919SHong Zhang /* 3958914a3f7SHong Zhang ierr = PetscOptionsLogical("-mat_superlu_equil","Equil","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 3968914a3f7SHong Zhang if (flg) lu->options.Equil = YES; -- not supported by the interface !!! 3979ce50919SHong Zhang */ 3988914a3f7SHong Zhang ierr = PetscOptionsEList("-mat_superlu_colperm","ColPerm","None",colperm,4,colperm[3],&indx,&flg);CHKERRQ(ierr); 3999ce50919SHong Zhang if (flg) {lu->options.ColPerm = (colperm_t)indx;} 4008914a3f7SHong Zhang ierr = PetscOptionsEList("-mat_superlu_iterrefine","IterRefine","None",iterrefine,4,iterrefine[0],&indx,&flg);CHKERRQ(ierr); 4019ce50919SHong Zhang if (flg) { lu->options.IterRefine = (IterRefine_t)indx;} 4028914a3f7SHong Zhang ierr = PetscOptionsLogical("-mat_superlu_symmetricmode","SymmetricMode","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4039ce50919SHong Zhang if (flg) lu->options.SymmetricMode = YES; 4048914a3f7SHong Zhang ierr = PetscOptionsReal("-mat_superlu_diagpivotthresh","DiagPivotThresh","None",lu->options.DiagPivotThresh,&lu->options.DiagPivotThresh,PETSC_NULL);CHKERRQ(ierr); 4058914a3f7SHong Zhang ierr = PetscOptionsLogical("-mat_superlu_pivotgrowth","PivotGrowth","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4069ce50919SHong Zhang if (flg) lu->options.PivotGrowth = YES; 4078914a3f7SHong Zhang ierr = PetscOptionsLogical("-mat_superlu_conditionnumber","ConditionNumber","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4089ce50919SHong Zhang if (flg) lu->options.ConditionNumber = YES; 4098914a3f7SHong Zhang ierr = PetscOptionsEList("-mat_superlu_rowperm","rowperm","None",rowperm,2,rowperm[0],&indx,&flg);CHKERRQ(ierr); 4109ce50919SHong Zhang if (flg) {lu->options.RowPerm = (rowperm_t)indx;} 4118914a3f7SHong Zhang ierr = PetscOptionsLogical("-mat_superlu_replacetinypivot","ReplaceTinyPivot","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4129ce50919SHong Zhang if (flg) lu->options.ReplaceTinyPivot = YES; 4138914a3f7SHong Zhang ierr = PetscOptionsLogical("-mat_superlu_printstat","PrintStat","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr); 4149ce50919SHong Zhang if (flg) lu->options.PrintStat = YES; 4158914a3f7SHong Zhang ierr = PetscOptionsInt("-mat_superlu_lwork","size of work array in bytes used by factorization","None",lu->lwork,&lu->lwork,PETSC_NULL);CHKERRQ(ierr); 4165fe6150dSHong Zhang if (lu->lwork > 0 ){ 4175fe6150dSHong Zhang ierr = PetscMalloc(lu->lwork,&lu->work);CHKERRQ(ierr); 4185fe6150dSHong Zhang } else if (lu->lwork != 0 && lu->lwork != -1){ 41977431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF," Warning: lwork %D is not supported by SUPERLU. The default lwork=0 is used.\n",lu->lwork); 4208914a3f7SHong Zhang lu->lwork = 0; 4218914a3f7SHong Zhang } 4229ce50919SHong Zhang PetscOptionsEnd(); 4239ce50919SHong Zhang 42475af56d4SHong Zhang #ifdef SUPERLU2 4252ecb5974SKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatCreateNull","MatCreateNull_SuperLU", 426f0c56d0fSKris Buschelman (void(*)(void))MatCreateNull_SuperLU);CHKERRQ(ierr); 42775af56d4SHong Zhang #endif 42814b396bbSKris Buschelman 42975af56d4SHong Zhang /* Allocate spaces (notice sizes are for the transpose) */ 430da7a1d00SHong Zhang ierr = PetscMalloc(m*sizeof(PetscInt),&lu->etree);CHKERRQ(ierr); 431da7a1d00SHong Zhang ierr = PetscMalloc(n*sizeof(PetscInt),&lu->perm_r);CHKERRQ(ierr); 432da7a1d00SHong Zhang ierr = PetscMalloc(m*sizeof(PetscInt),&lu->perm_c);CHKERRQ(ierr); 433da7a1d00SHong Zhang ierr = PetscMalloc(n*sizeof(PetscInt),&lu->R);CHKERRQ(ierr); 434da7a1d00SHong Zhang ierr = PetscMalloc(m*sizeof(PetscInt),&lu->C);CHKERRQ(ierr); 43575af56d4SHong Zhang 4369ce50919SHong Zhang /* create rhs and solution x without allocate space for .Store */ 4375fe6150dSHong Zhang #if defined(PETSC_USE_COMPLEX) 438937a6b0eSHong Zhang zCreate_Dense_Matrix(&lu->B, m, 1, PETSC_NULL, m, SLU_DN, SLU_Z, SLU_GE); 439937a6b0eSHong Zhang zCreate_Dense_Matrix(&lu->X, m, 1, PETSC_NULL, m, SLU_DN, SLU_Z, SLU_GE); 4405fe6150dSHong Zhang #else 44175af56d4SHong Zhang dCreate_Dense_Matrix(&lu->B, m, 1, PETSC_NULL, m, SLU_DN, SLU_D, SLU_GE); 44275af56d4SHong Zhang dCreate_Dense_Matrix(&lu->X, m, 1, PETSC_NULL, m, SLU_DN, SLU_D, SLU_GE); 4435fe6150dSHong Zhang #endif 44414b396bbSKris Buschelman 44514b396bbSKris Buschelman lu->flg = DIFFERENT_NONZERO_PATTERN; 446e740cb95SKris Buschelman lu->CleanUpSuperLU = PETSC_TRUE; 447899d7b4fSKris Buschelman 448899d7b4fSKris Buschelman *F = B; 449da7a1d00SHong Zhang ierr = PetscLogObjectMemory(B,(A->m+A->n)*sizeof(PetscInt)+sizeof(Mat_SuperLU));CHKERRQ(ierr); 45014b396bbSKris Buschelman PetscFunctionReturn(0); 45114b396bbSKris Buschelman } 45214b396bbSKris Buschelman 45394b7f48cSBarry Smith /* used by -ksp_view */ 45414b396bbSKris Buschelman #undef __FUNCT__ 455f0c56d0fSKris Buschelman #define __FUNCT__ "MatFactorInfo_SuperLU" 456dfbe8321SBarry Smith PetscErrorCode MatFactorInfo_SuperLU(Mat A,PetscViewer viewer) 45714b396bbSKris Buschelman { 458f0c56d0fSKris Buschelman Mat_SuperLU *lu= (Mat_SuperLU*)A->spptr; 459da7a1d00SHong Zhang PetscErrorCode ierr; 4609ce50919SHong Zhang superlu_options_t options; 461f0c56d0fSKris Buschelman 462f0c56d0fSKris Buschelman PetscFunctionBegin; 4639ce50919SHong Zhang /* check if matrix is superlu_dist type */ 4649ce50919SHong Zhang if (A->ops->solve != MatSolve_SuperLU) PetscFunctionReturn(0); 4659ce50919SHong Zhang 4669ce50919SHong Zhang options = lu->options; 467f0c56d0fSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer,"SuperLU run parameters:\n");CHKERRQ(ierr); 4689ce50919SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Equil: %s\n",(options.Equil != NO) ? "YES": "NO");CHKERRQ(ierr); 46977431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ColPerm: %D\n",options.ColPerm);CHKERRQ(ierr); 47077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," IterRefine: %D\n",options.IterRefine);CHKERRQ(ierr); 4719ce50919SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," SymmetricMode: %s\n",(options.SymmetricMode != NO) ? "YES": "NO");CHKERRQ(ierr); 4729ce50919SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," DiagPivotThresh: %g\n",options.DiagPivotThresh);CHKERRQ(ierr); 4739ce50919SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," PivotGrowth: %s\n",(options.PivotGrowth != NO) ? "YES": "NO");CHKERRQ(ierr); 4749ce50919SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ConditionNumber: %s\n",(options.ConditionNumber != NO) ? "YES": "NO");CHKERRQ(ierr); 47577431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," RowPerm: %D\n",options.RowPerm);CHKERRQ(ierr); 4769ce50919SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ReplaceTinyPivot: %s\n",(options.ReplaceTinyPivot != NO) ? "YES": "NO");CHKERRQ(ierr); 4779ce50919SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," PrintStat: %s\n",(options.PrintStat != NO) ? "YES": "NO");CHKERRQ(ierr); 47877431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," lwork: %D\n",lu->lwork);CHKERRQ(ierr); 4799ce50919SHong Zhang 480f0c56d0fSKris Buschelman PetscFunctionReturn(0); 481f0c56d0fSKris Buschelman } 482f0c56d0fSKris Buschelman 483f0c56d0fSKris Buschelman #undef __FUNCT__ 484f0c56d0fSKris Buschelman #define __FUNCT__ "MatDuplicate_SuperLU" 485dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SuperLU(Mat A, MatDuplicateOption op, Mat *M) { 486da7a1d00SHong Zhang PetscErrorCode ierr; 4878f340917SKris Buschelman Mat_SuperLU *lu=(Mat_SuperLU *)A->spptr; 4888f340917SKris Buschelman 48914b396bbSKris Buschelman PetscFunctionBegin; 4908f340917SKris Buschelman ierr = (*lu->MatDuplicate)(A,op,M);CHKERRQ(ierr); 4913f953163SKris Buschelman ierr = PetscMemcpy((*M)->spptr,lu,sizeof(Mat_SuperLU));CHKERRQ(ierr); 49214b396bbSKris Buschelman PetscFunctionReturn(0); 49314b396bbSKris Buschelman } 49414b396bbSKris Buschelman 49514b396bbSKris Buschelman EXTERN_C_BEGIN 49614b396bbSKris Buschelman #undef __FUNCT__ 497b0592601SKris Buschelman #define __FUNCT__ "MatConvert_SuperLU_SeqAIJ" 498*be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SuperLU_SeqAIJ(Mat A,const MatType type,MatReuse reuse,Mat *newmat) 499521d7252SBarry Smith { 500fb3e25aaSKris Buschelman /* This routine is only called to convert an unfactored PETSc-SuperLU matrix */ 501fb3e25aaSKris Buschelman /* to its base PETSc type, so we will ignore 'MatType type'. */ 502da7a1d00SHong Zhang PetscErrorCode ierr; 503b0592601SKris Buschelman Mat B=*newmat; 504f0c56d0fSKris Buschelman Mat_SuperLU *lu=(Mat_SuperLU *)A->spptr; 505b0592601SKris Buschelman 506b0592601SKris Buschelman PetscFunctionBegin; 507ceb03754SKris Buschelman if (reuse == MAT_INITIAL_MATRIX) { 508b0592601SKris Buschelman ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr); 509f0c56d0fSKris Buschelman } 510b0592601SKris Buschelman /* Reset the original function pointers */ 511f0c56d0fSKris Buschelman B->ops->duplicate = lu->MatDuplicate; 512b0592601SKris Buschelman B->ops->view = lu->MatView; 513b0592601SKris Buschelman B->ops->assemblyend = lu->MatAssemblyEnd; 514b0592601SKris Buschelman B->ops->lufactorsymbolic = lu->MatLUFactorSymbolic; 515b0592601SKris Buschelman B->ops->destroy = lu->MatDestroy; 516b0592601SKris Buschelman /* lu is only a function pointer stash unless we've factored the matrix, which we haven't! */ 517b0592601SKris Buschelman ierr = PetscFree(lu);CHKERRQ(ierr); 518901853e0SKris Buschelman 519901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqaij_superlu_C","",PETSC_NULL);CHKERRQ(ierr); 520901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_superlu_seqaij_C","",PETSC_NULL);CHKERRQ(ierr); 521901853e0SKris Buschelman 522b0592601SKris Buschelman ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQAIJ);CHKERRQ(ierr); 523b0592601SKris Buschelman *newmat = B; 524b0592601SKris Buschelman PetscFunctionReturn(0); 525b0592601SKris Buschelman } 526b0592601SKris Buschelman EXTERN_C_END 527b0592601SKris Buschelman 528b0592601SKris Buschelman EXTERN_C_BEGIN 529b0592601SKris Buschelman #undef __FUNCT__ 530b0592601SKris Buschelman #define __FUNCT__ "MatConvert_SeqAIJ_SuperLU" 531*be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqAIJ_SuperLU(Mat A,const MatType type,MatReuse reuse,Mat *newmat) 532521d7252SBarry Smith { 533fb3e25aaSKris Buschelman /* This routine is only called to convert to MATSUPERLU */ 534fb3e25aaSKris Buschelman /* from MATSEQAIJ, so we will ignore 'MatType type'. */ 535da7a1d00SHong Zhang PetscErrorCode ierr; 536b0592601SKris Buschelman Mat B=*newmat; 537f0c56d0fSKris Buschelman Mat_SuperLU *lu; 53814b396bbSKris Buschelman 53914b396bbSKris Buschelman PetscFunctionBegin; 540ceb03754SKris Buschelman if (reuse == MAT_INITIAL_MATRIX) { 541b0592601SKris Buschelman ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr); 542b0592601SKris Buschelman } 54314b396bbSKris Buschelman 544f0c56d0fSKris Buschelman ierr = PetscNew(Mat_SuperLU,&lu);CHKERRQ(ierr); 545f0c56d0fSKris Buschelman lu->MatDuplicate = A->ops->duplicate; 54614b396bbSKris Buschelman lu->MatView = A->ops->view; 54714b396bbSKris Buschelman lu->MatAssemblyEnd = A->ops->assemblyend; 548b0592601SKris Buschelman lu->MatLUFactorSymbolic = A->ops->lufactorsymbolic; 54914b396bbSKris Buschelman lu->MatDestroy = A->ops->destroy; 55014b396bbSKris Buschelman lu->CleanUpSuperLU = PETSC_FALSE; 551b0592601SKris Buschelman 552b0592601SKris Buschelman B->spptr = (void*)lu; 553f0c56d0fSKris Buschelman B->ops->duplicate = MatDuplicate_SuperLU; 554f0c56d0fSKris Buschelman B->ops->view = MatView_SuperLU; 555f0c56d0fSKris Buschelman B->ops->assemblyend = MatAssemblyEnd_SuperLU; 556f0c56d0fSKris Buschelman B->ops->lufactorsymbolic = MatLUFactorSymbolic_SuperLU; 557e4e36384SHong Zhang B->ops->choleskyfactorsymbolic = 0; 558f0c56d0fSKris Buschelman B->ops->destroy = MatDestroy_SuperLU; 559b0592601SKris Buschelman 560b0592601SKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqaij_superlu_C", 561b0592601SKris Buschelman "MatConvert_SeqAIJ_SuperLU",MatConvert_SeqAIJ_SuperLU);CHKERRQ(ierr); 562b0592601SKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_superlu_seqaij_C", 563b0592601SKris Buschelman "MatConvert_SuperLU_SeqAIJ",MatConvert_SuperLU_SeqAIJ);CHKERRQ(ierr); 56463ba0a88SBarry Smith ierr = PetscLogInfo((0,"MatConvert_SeqAIJ_SuperLU:Using SuperLU for SeqAIJ LU factorization and solves.\n"));CHKERRQ(ierr); 565fb3e25aaSKris Buschelman ierr = PetscObjectChangeTypeName((PetscObject)B,MATSUPERLU);CHKERRQ(ierr); 566b0592601SKris Buschelman *newmat = B; 567b0592601SKris Buschelman PetscFunctionReturn(0); 568b0592601SKris Buschelman } 569b0592601SKris Buschelman EXTERN_C_END 570b0592601SKris Buschelman 57124b6179bSKris Buschelman /*MC 572fafad747SKris Buschelman MATSUPERLU - MATSUPERLU = "superlu" - A matrix type providing direct solvers (LU) for sequential matrices 57324b6179bSKris Buschelman via the external package SuperLU. 57424b6179bSKris Buschelman 57524b6179bSKris Buschelman If SuperLU is installed (see the manual for 57624b6179bSKris Buschelman instructions on how to declare the existence of external packages), 57724b6179bSKris Buschelman a matrix type can be constructed which invokes SuperLU solvers. 57824b6179bSKris Buschelman After calling MatCreate(...,A), simply call MatSetType(A,MATSUPERLU). 57924b6179bSKris Buschelman 58024b6179bSKris Buschelman This matrix inherits from MATSEQAIJ. As a result, MatSeqAIJSetPreallocation is 58128b08bd3SKris Buschelman supported for this matrix type. One can also call MatConvert for an inplace conversion to or from 58228b08bd3SKris Buschelman the MATSEQAIJ type without data copy. 58324b6179bSKris Buschelman 58424b6179bSKris Buschelman Options Database Keys: 5850bad9183SKris Buschelman + -mat_type superlu - sets the matrix type to "superlu" during a call to MatSetFromOptions() 58624b6179bSKris Buschelman - -mat_superlu_ordering <0,1,2,3> - 0: natural ordering, 58724b6179bSKris Buschelman 1: MMD applied to A'*A, 58824b6179bSKris Buschelman 2: MMD applied to A'+A, 58924b6179bSKris Buschelman 3: COLAMD, approximate minimum degree column ordering 59024b6179bSKris Buschelman 59124b6179bSKris Buschelman Level: beginner 59224b6179bSKris Buschelman 59324b6179bSKris Buschelman .seealso: PCLU 59424b6179bSKris Buschelman M*/ 59524b6179bSKris Buschelman 596b0592601SKris Buschelman EXTERN_C_BEGIN 597b0592601SKris Buschelman #undef __FUNCT__ 598f0c56d0fSKris Buschelman #define __FUNCT__ "MatCreate_SuperLU" 599*be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_SuperLU(Mat A) 600dfbe8321SBarry Smith { 601dfbe8321SBarry Smith PetscErrorCode ierr; 602b0592601SKris Buschelman 603b0592601SKris Buschelman PetscFunctionBegin; 6045441df8eSKris Buschelman /* Change type name before calling MatSetType to force proper construction of SeqAIJ and SUPERLU types */ 6055441df8eSKris Buschelman ierr = PetscObjectChangeTypeName((PetscObject)A,MATSUPERLU);CHKERRQ(ierr); 606b0592601SKris Buschelman ierr = MatSetType(A,MATSEQAIJ);CHKERRQ(ierr); 607ceb03754SKris Buschelman ierr = MatConvert_SeqAIJ_SuperLU(A,MATSUPERLU,MAT_REUSE_MATRIX,&A);CHKERRQ(ierr); 60814b396bbSKris Buschelman PetscFunctionReturn(0); 60914b396bbSKris Buschelman } 61014b396bbSKris Buschelman EXTERN_C_END 611