1be1d678aSKris Buschelman #define PETSCMAT_DLL 21c2a3de1SBarry Smith 3397b6df1SKris Buschelman /* 4c2b5dc30SHong Zhang Provides an interface to the MUMPS sparse solver 5397b6df1SKris Buschelman */ 67c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h" 77c4f633dSBarry Smith #include "../src/mat/impls/aij/mpi/mpiaij.h" 87c4f633dSBarry Smith #include "../src/mat/impls/sbaij/seq/sbaij.h" 97c4f633dSBarry Smith #include "../src/mat/impls/sbaij/mpi/mpisbaij.h" 10397b6df1SKris Buschelman 11397b6df1SKris Buschelman EXTERN_C_BEGIN 12397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 13397b6df1SKris Buschelman #include "zmumps_c.h" 14397b6df1SKris Buschelman #else 15397b6df1SKris Buschelman #include "dmumps_c.h" 16397b6df1SKris Buschelman #endif 17397b6df1SKris Buschelman EXTERN_C_END 18397b6df1SKris Buschelman #define JOB_INIT -1 19397b6df1SKris Buschelman #define JOB_END -2 20397b6df1SKris Buschelman /* macros s.t. indices match MUMPS documentation */ 21397b6df1SKris Buschelman #define ICNTL(I) icntl[(I)-1] 22397b6df1SKris Buschelman #define CNTL(I) cntl[(I)-1] 23397b6df1SKris Buschelman #define INFOG(I) infog[(I)-1] 24a7aca84bSHong Zhang #define INFO(I) info[(I)-1] 25397b6df1SKris Buschelman #define RINFOG(I) rinfog[(I)-1] 26adc1d99fSHong Zhang #define RINFO(I) rinfo[(I)-1] 27397b6df1SKris Buschelman 28397b6df1SKris Buschelman typedef struct { 29397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 30397b6df1SKris Buschelman ZMUMPS_STRUC_C id; 31397b6df1SKris Buschelman #else 32397b6df1SKris Buschelman DMUMPS_STRUC_C id; 33397b6df1SKris Buschelman #endif 34397b6df1SKris Buschelman MatStructure matstruc; 35c1490034SHong Zhang PetscMPIInt myid,size; 36329ec9b3SHong Zhang PetscInt *irn,*jcn,sym,nSolve; 37397b6df1SKris Buschelman PetscScalar *val; 38397b6df1SKris Buschelman MPI_Comm comm_mumps; 39329ec9b3SHong Zhang VecScatter scat_rhs, scat_sol; 40c338a77dSKris Buschelman PetscTruth isAIJ,CleanUpMUMPS; 41329ec9b3SHong Zhang Vec b_seq,x_seq; 4267334b25SHong Zhang PetscErrorCode (*MatDestroy)(Mat); 43f0c56d0fSKris Buschelman } Mat_MUMPS; 44f0c56d0fSKris Buschelman 45dfbe8321SBarry Smith EXTERN PetscErrorCode MatDuplicate_MUMPS(Mat,MatDuplicateOption,Mat*); 46b24902e0SBarry Smith 47397b6df1SKris Buschelman /* convert Petsc mpiaij matrix to triples: row[nz], col[nz], val[nz] */ 48397b6df1SKris Buschelman /* 49397b6df1SKris Buschelman input: 5075747be1SHong Zhang A - matrix in mpiaij or mpisbaij (bs=1) format 51397b6df1SKris Buschelman shift - 0: C style output triple; 1: Fortran style output triple. 52397b6df1SKris Buschelman valOnly - FALSE: spaces are allocated and values are set for the triple 53397b6df1SKris Buschelman TRUE: only the values in v array are updated 54397b6df1SKris Buschelman output: 55397b6df1SKris Buschelman nnz - dim of r, c, and v (number of local nonzero entries of A) 56397b6df1SKris Buschelman r, c, v - row and col index, matrix values (matrix triples) 57397b6df1SKris Buschelman */ 58b24902e0SBarry Smith PetscErrorCode MatConvertToTriples(Mat A,int shift,PetscTruth valOnly,int *nnz,int **r, int **c, PetscScalar **v) 59b24902e0SBarry Smith { 60c1490034SHong Zhang PetscInt *ai, *aj, *bi, *bj, rstart,nz, *garray; 61dfbe8321SBarry Smith PetscErrorCode ierr; 62d0f46423SBarry Smith PetscInt i,j,jj,jB,irow,m=A->rmap->n,*ajj,*bjj,countA,countB,colA_start,jcol; 63c1490034SHong Zhang PetscInt *row,*col; 64397b6df1SKris Buschelman PetscScalar *av, *bv,*val; 655c9eb25fSBarry Smith PetscTruth isAIJ; 66397b6df1SKris Buschelman 67397b6df1SKris Buschelman PetscFunctionBegin; 685c9eb25fSBarry Smith ierr = PetscTypeCompare((PetscObject)A,MATMPIAIJ,&isAIJ);CHKERRQ(ierr); 695c9eb25fSBarry Smith if (isAIJ){ 70397b6df1SKris Buschelman Mat_MPIAIJ *mat = (Mat_MPIAIJ*)A->data; 71397b6df1SKris Buschelman Mat_SeqAIJ *aa=(Mat_SeqAIJ*)(mat->A)->data; 72397b6df1SKris Buschelman Mat_SeqAIJ *bb=(Mat_SeqAIJ*)(mat->B)->data; 73397b6df1SKris Buschelman nz = aa->nz + bb->nz; 74d0f46423SBarry Smith ai=aa->i; aj=aa->j; bi=bb->i; bj=bb->j; rstart= A->rmap->rstart; 75397b6df1SKris Buschelman garray = mat->garray; 76397b6df1SKris Buschelman av=aa->a; bv=bb->a; 77397b6df1SKris Buschelman 78397b6df1SKris Buschelman } else { 79397b6df1SKris Buschelman Mat_MPISBAIJ *mat = (Mat_MPISBAIJ*)A->data; 80397b6df1SKris Buschelman Mat_SeqSBAIJ *aa=(Mat_SeqSBAIJ*)(mat->A)->data; 81397b6df1SKris Buschelman Mat_SeqBAIJ *bb=(Mat_SeqBAIJ*)(mat->B)->data; 82d0f46423SBarry Smith if (A->rmap->bs > 1) SETERRQ1(PETSC_ERR_SUP," bs=%d is not supported yet\n", A->rmap->bs); 836c6c5352SBarry Smith nz = aa->nz + bb->nz; 84d0f46423SBarry Smith ai=aa->i; aj=aa->j; bi=bb->i; bj=bb->j; rstart= A->rmap->rstart; 85397b6df1SKris Buschelman garray = mat->garray; 86397b6df1SKris Buschelman av=aa->a; bv=bb->a; 87397b6df1SKris Buschelman } 88397b6df1SKris Buschelman 89397b6df1SKris Buschelman if (!valOnly){ 907c307921SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt) ,&row);CHKERRQ(ierr); 917c307921SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&col);CHKERRQ(ierr); 92397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(PetscScalar),&val);CHKERRQ(ierr); 93397b6df1SKris Buschelman *r = row; *c = col; *v = val; 94397b6df1SKris Buschelman } else { 95397b6df1SKris Buschelman row = *r; col = *c; val = *v; 96397b6df1SKris Buschelman } 97397b6df1SKris Buschelman *nnz = nz; 98397b6df1SKris Buschelman 99028e57e8SHong Zhang jj = 0; irow = rstart; 100397b6df1SKris Buschelman for ( i=0; i<m; i++ ) { 101397b6df1SKris Buschelman ajj = aj + ai[i]; /* ptr to the beginning of this row */ 102397b6df1SKris Buschelman countA = ai[i+1] - ai[i]; 103397b6df1SKris Buschelman countB = bi[i+1] - bi[i]; 104397b6df1SKris Buschelman bjj = bj + bi[i]; 105397b6df1SKris Buschelman 106397b6df1SKris Buschelman /* get jB, the starting local col index for the 2nd B-part */ 107397b6df1SKris Buschelman colA_start = rstart + ajj[0]; /* the smallest col index for A */ 10875747be1SHong Zhang j=-1; 10975747be1SHong Zhang do { 11075747be1SHong Zhang j++; 11175747be1SHong Zhang if (j == countB) break; 112397b6df1SKris Buschelman jcol = garray[bjj[j]]; 11375747be1SHong Zhang } while (jcol < colA_start); 11475747be1SHong Zhang jB = j; 115397b6df1SKris Buschelman 116397b6df1SKris Buschelman /* B-part, smaller col index */ 117397b6df1SKris Buschelman colA_start = rstart + ajj[0]; /* the smallest col index for A */ 118397b6df1SKris Buschelman for (j=0; j<jB; j++){ 119397b6df1SKris Buschelman jcol = garray[bjj[j]]; 120397b6df1SKris Buschelman if (!valOnly){ 121397b6df1SKris Buschelman row[jj] = irow + shift; col[jj] = jcol + shift; 12275747be1SHong Zhang 123397b6df1SKris Buschelman } 124397b6df1SKris Buschelman val[jj++] = *bv++; 125397b6df1SKris Buschelman } 126397b6df1SKris Buschelman /* A-part */ 127397b6df1SKris Buschelman for (j=0; j<countA; j++){ 128397b6df1SKris Buschelman if (!valOnly){ 129397b6df1SKris Buschelman row[jj] = irow + shift; col[jj] = rstart + ajj[j] + shift; 130397b6df1SKris Buschelman } 131397b6df1SKris Buschelman val[jj++] = *av++; 132397b6df1SKris Buschelman } 133397b6df1SKris Buschelman /* B-part, larger col index */ 134397b6df1SKris Buschelman for (j=jB; j<countB; j++){ 135397b6df1SKris Buschelman if (!valOnly){ 136397b6df1SKris Buschelman row[jj] = irow + shift; col[jj] = garray[bjj[j]] + shift; 137397b6df1SKris Buschelman } 138397b6df1SKris Buschelman val[jj++] = *bv++; 139397b6df1SKris Buschelman } 140397b6df1SKris Buschelman irow++; 141397b6df1SKris Buschelman } 142397b6df1SKris Buschelman 143397b6df1SKris Buschelman PetscFunctionReturn(0); 144397b6df1SKris Buschelman } 145397b6df1SKris Buschelman 146397b6df1SKris Buschelman #undef __FUNCT__ 1473924e44cSKris Buschelman #define __FUNCT__ "MatDestroy_MUMPS" 148dfbe8321SBarry Smith PetscErrorCode MatDestroy_MUMPS(Mat A) 149dfbe8321SBarry Smith { 150f0c56d0fSKris Buschelman Mat_MUMPS *lu=(Mat_MUMPS*)A->spptr; 151dfbe8321SBarry Smith PetscErrorCode ierr; 152c1490034SHong Zhang PetscMPIInt size=lu->size; 153b24902e0SBarry Smith 154397b6df1SKris Buschelman PetscFunctionBegin; 155397b6df1SKris Buschelman if (lu->CleanUpMUMPS) { 156397b6df1SKris Buschelman /* Terminate instance, deallocate memories */ 157329ec9b3SHong Zhang if (size > 1){ 158329ec9b3SHong Zhang ierr = PetscFree(lu->id.sol_loc);CHKERRQ(ierr); 159329ec9b3SHong Zhang ierr = VecScatterDestroy(lu->scat_rhs);CHKERRQ(ierr); 160329ec9b3SHong Zhang ierr = VecDestroy(lu->b_seq);CHKERRQ(ierr); 1612750af12SHong Zhang if (lu->nSolve && lu->scat_sol){ierr = VecScatterDestroy(lu->scat_sol);CHKERRQ(ierr);} 1622750af12SHong Zhang if (lu->nSolve && lu->x_seq){ierr = VecDestroy(lu->x_seq);CHKERRQ(ierr);} 163329ec9b3SHong Zhang ierr = PetscFree(lu->val);CHKERRQ(ierr); 164329ec9b3SHong Zhang } 165397b6df1SKris Buschelman lu->id.job=JOB_END; 166397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 167397b6df1SKris Buschelman zmumps_c(&lu->id); 168397b6df1SKris Buschelman #else 169397b6df1SKris Buschelman dmumps_c(&lu->id); 170397b6df1SKris Buschelman #endif 171c338a77dSKris Buschelman ierr = PetscFree(lu->irn);CHKERRQ(ierr); 172c338a77dSKris Buschelman ierr = PetscFree(lu->jcn);CHKERRQ(ierr); 173397b6df1SKris Buschelman ierr = MPI_Comm_free(&(lu->comm_mumps));CHKERRQ(ierr); 174397b6df1SKris Buschelman } 17567334b25SHong Zhang ierr = (lu->MatDestroy)(A);CHKERRQ(ierr); 176397b6df1SKris Buschelman PetscFunctionReturn(0); 177397b6df1SKris Buschelman } 178397b6df1SKris Buschelman 179397b6df1SKris Buschelman #undef __FUNCT__ 180f6c57405SHong Zhang #define __FUNCT__ "MatSolve_MUMPS" 181b24902e0SBarry Smith PetscErrorCode MatSolve_MUMPS(Mat A,Vec b,Vec x) 182b24902e0SBarry Smith { 183f0c56d0fSKris Buschelman Mat_MUMPS *lu=(Mat_MUMPS*)A->spptr; 184d54de34fSKris Buschelman PetscScalar *array; 185397b6df1SKris Buschelman Vec x_seq; 186329ec9b3SHong Zhang IS is_iden,is_petsc; 187dfbe8321SBarry Smith PetscErrorCode ierr; 188329ec9b3SHong Zhang PetscInt i; 189397b6df1SKris Buschelman 190397b6df1SKris Buschelman PetscFunctionBegin; 191329ec9b3SHong Zhang lu->id.nrhs = 1; 192329ec9b3SHong Zhang x_seq = lu->b_seq; 193397b6df1SKris Buschelman if (lu->size > 1){ 194329ec9b3SHong Zhang /* MUMPS only supports centralized rhs. Scatter b into a seqential rhs vector */ 195f6cfb2d1SLisandro Dalcin ierr = VecScatterBegin(lu->scat_rhs,b,x_seq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 196f6cfb2d1SLisandro Dalcin ierr = VecScatterEnd(lu->scat_rhs,b,x_seq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 197397b6df1SKris Buschelman if (!lu->myid) {ierr = VecGetArray(x_seq,&array);CHKERRQ(ierr);} 198397b6df1SKris Buschelman } else { /* size == 1 */ 199397b6df1SKris Buschelman ierr = VecCopy(b,x);CHKERRQ(ierr); 200397b6df1SKris Buschelman ierr = VecGetArray(x,&array);CHKERRQ(ierr); 201397b6df1SKris Buschelman } 202397b6df1SKris Buschelman if (!lu->myid) { /* define rhs on the host */ 203*8278f211SHong Zhang lu->id.nrhs = 1; 204397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 205397b6df1SKris Buschelman lu->id.rhs = (mumps_double_complex*)array; 206397b6df1SKris Buschelman #else 207397b6df1SKris Buschelman lu->id.rhs = array; 208397b6df1SKris Buschelman #endif 209397b6df1SKris Buschelman } 210329ec9b3SHong Zhang if (lu->size == 1){ 211329ec9b3SHong Zhang ierr = VecRestoreArray(x,&array);CHKERRQ(ierr); 212329ec9b3SHong Zhang } else if (!lu->myid){ 213329ec9b3SHong Zhang ierr = VecRestoreArray(x_seq,&array);CHKERRQ(ierr); 214329ec9b3SHong Zhang } 215329ec9b3SHong Zhang 216329ec9b3SHong Zhang if (lu->size > 1){ 217329ec9b3SHong Zhang /* distributed solution */ 218329ec9b3SHong Zhang lu->id.ICNTL(21) = 1; 219329ec9b3SHong Zhang if (!lu->nSolve){ 220329ec9b3SHong Zhang /* Create x_seq=sol_loc for repeated use */ 221329ec9b3SHong Zhang PetscInt lsol_loc; 222329ec9b3SHong Zhang PetscScalar *sol_loc; 223329ec9b3SHong Zhang lsol_loc = lu->id.INFO(23); /* length of sol_loc */ 224329ec9b3SHong Zhang ierr = PetscMalloc((1+lsol_loc)*(sizeof(PetscScalar)+sizeof(PetscInt)),&sol_loc);CHKERRQ(ierr); 225329ec9b3SHong Zhang lu->id.isol_loc = (PetscInt *)(sol_loc + lsol_loc); 226329ec9b3SHong Zhang lu->id.lsol_loc = lsol_loc; 22744ea04b1SSatish Balay #if defined(PETSC_USE_COMPLEX) 22844ea04b1SSatish Balay lu->id.sol_loc = (mumps_double_complex*)sol_loc; 22944ea04b1SSatish Balay #else 23044ea04b1SSatish Balay lu->id.sol_loc = sol_loc; 23144ea04b1SSatish Balay #endif 232329ec9b3SHong Zhang ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,lsol_loc,sol_loc,&lu->x_seq);CHKERRQ(ierr); 233329ec9b3SHong Zhang } 234329ec9b3SHong Zhang } 235397b6df1SKris Buschelman 236397b6df1SKris Buschelman /* solve phase */ 237329ec9b3SHong Zhang /*-------------*/ 238397b6df1SKris Buschelman lu->id.job = 3; 239397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 240397b6df1SKris Buschelman zmumps_c(&lu->id); 241397b6df1SKris Buschelman #else 242397b6df1SKris Buschelman dmumps_c(&lu->id); 243397b6df1SKris Buschelman #endif 244397b6df1SKris Buschelman if (lu->id.INFOG(1) < 0) { 24579a5c55eSBarry Smith SETERRQ1(PETSC_ERR_LIB,"Error reported by MUMPS in solve phase: INFOG(1)=%d\n",lu->id.INFOG(1)); 246397b6df1SKris Buschelman } 247397b6df1SKris Buschelman 248329ec9b3SHong Zhang if (lu->size > 1) { /* convert mumps distributed solution to petsc mpi x */ 249329ec9b3SHong Zhang if (!lu->nSolve){ /* create scatter scat_sol */ 250329ec9b3SHong Zhang ierr = ISCreateStride(PETSC_COMM_SELF,lu->id.lsol_loc,0,1,&is_iden);CHKERRQ(ierr); /* from */ 251329ec9b3SHong Zhang for (i=0; i<lu->id.lsol_loc; i++){ 252329ec9b3SHong Zhang lu->id.isol_loc[i] -= 1; /* change Fortran style to C style */ 253397b6df1SKris Buschelman } 254329ec9b3SHong Zhang ierr = ISCreateGeneral(PETSC_COMM_SELF,lu->id.lsol_loc,lu->id.isol_loc,&is_petsc);CHKERRQ(ierr); /* to */ 255329ec9b3SHong Zhang ierr = VecScatterCreate(lu->x_seq,is_iden,x,is_petsc,&lu->scat_sol);CHKERRQ(ierr); 256329ec9b3SHong Zhang ierr = ISDestroy(is_iden);CHKERRQ(ierr); 257329ec9b3SHong Zhang ierr = ISDestroy(is_petsc);CHKERRQ(ierr); 258397b6df1SKris Buschelman } 259ca9f406cSSatish Balay ierr = VecScatterBegin(lu->scat_sol,lu->x_seq,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 260ca9f406cSSatish Balay ierr = VecScatterEnd(lu->scat_sol,lu->x_seq,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 261329ec9b3SHong Zhang } 262329ec9b3SHong Zhang lu->nSolve++; 263397b6df1SKris Buschelman PetscFunctionReturn(0); 264397b6df1SKris Buschelman } 265397b6df1SKris Buschelman 266ace3df97SHong Zhang #if !defined(PETSC_USE_COMPLEX) 267a58c3f20SHong Zhang /* 268a58c3f20SHong Zhang input: 269a58c3f20SHong Zhang F: numeric factor 270a58c3f20SHong Zhang output: 271a58c3f20SHong Zhang nneg: total number of negative pivots 272a58c3f20SHong Zhang nzero: 0 273a58c3f20SHong Zhang npos: (global dimension of F) - nneg 274a58c3f20SHong Zhang */ 275a58c3f20SHong Zhang 276a58c3f20SHong Zhang #undef __FUNCT__ 277a58c3f20SHong Zhang #define __FUNCT__ "MatGetInertia_SBAIJMUMPS" 278dfbe8321SBarry Smith PetscErrorCode MatGetInertia_SBAIJMUMPS(Mat F,int *nneg,int *nzero,int *npos) 279a58c3f20SHong Zhang { 280a58c3f20SHong Zhang Mat_MUMPS *lu =(Mat_MUMPS*)F->spptr; 281dfbe8321SBarry Smith PetscErrorCode ierr; 282c1490034SHong Zhang PetscMPIInt size; 283a58c3f20SHong Zhang 284a58c3f20SHong Zhang PetscFunctionBegin; 2857adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)F)->comm,&size);CHKERRQ(ierr); 286bcb30aebSHong Zhang /* MUMPS 4.3.1 calls ScaLAPACK when ICNTL(13)=0 (default), which does not offer the possibility to compute the inertia of a dense matrix. Set ICNTL(13)=1 to skip ScaLAPACK */ 287bcb30aebSHong Zhang if (size > 1 && lu->id.ICNTL(13) != 1){ 28879a5c55eSBarry Smith SETERRQ1(PETSC_ERR_ARG_WRONG,"ICNTL(13)=%d. -mat_mumps_icntl_13 must be set as 1 for correct global matrix inertia\n",lu->id.INFOG(13)); 289bcb30aebSHong Zhang } 290a58c3f20SHong Zhang if (nneg){ 291a58c3f20SHong Zhang if (!lu->myid){ 292a58c3f20SHong Zhang *nneg = lu->id.INFOG(12); 293a58c3f20SHong Zhang } 294bcb30aebSHong Zhang ierr = MPI_Bcast(nneg,1,MPI_INT,0,lu->comm_mumps);CHKERRQ(ierr); 295a58c3f20SHong Zhang } 296a58c3f20SHong Zhang if (nzero) *nzero = 0; 297d0f46423SBarry Smith if (npos) *npos = F->rmap->N - (*nneg); 298a58c3f20SHong Zhang PetscFunctionReturn(0); 299a58c3f20SHong Zhang } 300ace3df97SHong Zhang #endif /* !defined(PETSC_USE_COMPLEX) */ 301a58c3f20SHong Zhang 302397b6df1SKris Buschelman #undef __FUNCT__ 303f6c57405SHong Zhang #define __FUNCT__ "MatFactorNumeric_MUMPS" 3040481f469SBarry Smith PetscErrorCode MatFactorNumeric_MUMPS(Mat F,Mat A,const MatFactorInfo *info) 305af281ebdSHong Zhang { 306719d5645SBarry Smith Mat_MUMPS *lu =(Mat_MUMPS*)(F)->spptr; 3076849ba73SBarry Smith PetscErrorCode ierr; 308d0f46423SBarry Smith PetscInt rnz,nnz,nz=0,i,M=A->rmap->N,*ai,*aj,icntl; 309397b6df1SKris Buschelman PetscTruth valOnly,flg; 310e09efc27SHong Zhang Mat F_diag; 311c349612cSHong Zhang IS is_iden; 312c349612cSHong Zhang Vec b; 3135c9eb25fSBarry Smith PetscTruth isSeqAIJ,isSeqSBAIJ; 314397b6df1SKris Buschelman 315397b6df1SKris Buschelman PetscFunctionBegin; 3165c9eb25fSBarry Smith ierr = PetscTypeCompare((PetscObject)A,MATSEQAIJ,&isSeqAIJ);CHKERRQ(ierr); 3175c9eb25fSBarry Smith ierr = PetscTypeCompare((PetscObject)A,MATSEQSBAIJ,&isSeqSBAIJ);CHKERRQ(ierr); 318397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ 319719d5645SBarry Smith (F)->ops->solve = MatSolve_MUMPS; 320397b6df1SKris Buschelman 321397b6df1SKris Buschelman /* Initialize a MUMPS instance */ 3227adad957SLisandro Dalcin ierr = MPI_Comm_rank(((PetscObject)A)->comm, &lu->myid); 3237adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)A)->comm,&lu->size);CHKERRQ(ierr); 324397b6df1SKris Buschelman lu->id.job = JOB_INIT; 3257adad957SLisandro Dalcin ierr = MPI_Comm_dup(((PetscObject)A)->comm,&(lu->comm_mumps));CHKERRQ(ierr); 3266a1dac61SBarry Smith lu->id.comm_fortran = MPI_Comm_c2f(lu->comm_mumps); 327397b6df1SKris Buschelman 328397b6df1SKris Buschelman /* Set mumps options */ 3297adad957SLisandro Dalcin ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"MUMPS Options","Mat");CHKERRQ(ierr); 330397b6df1SKris Buschelman lu->id.par=1; /* host participates factorizaton and solve */ 331397b6df1SKris Buschelman lu->id.sym=lu->sym; 332397b6df1SKris Buschelman if (lu->sym == 2){ 333397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_sym","SYM: (1,2)","None",lu->id.sym,&icntl,&flg);CHKERRQ(ierr); 334397b6df1SKris Buschelman if (flg && icntl == 1) lu->id.sym=icntl; /* matrix is spd */ 335397b6df1SKris Buschelman } 336397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 337397b6df1SKris Buschelman zmumps_c(&lu->id); 338397b6df1SKris Buschelman #else 339397b6df1SKris Buschelman dmumps_c(&lu->id); 340397b6df1SKris Buschelman #endif 341397b6df1SKris Buschelman 3425c9eb25fSBarry Smith if (isSeqAIJ || isSeqSBAIJ){ 343397b6df1SKris Buschelman lu->id.ICNTL(18) = 0; /* centralized assembled matrix input */ 344397b6df1SKris Buschelman } else { 345397b6df1SKris Buschelman lu->id.ICNTL(18) = 3; /* distributed assembled matrix input */ 346397b6df1SKris Buschelman } 347397b6df1SKris Buschelman 348397b6df1SKris Buschelman icntl=-1; 349c0165424SHong Zhang lu->id.ICNTL(4) = 0; /* level of printing; overwrite mumps default ICNTL(4)=2 */ 350397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_4","ICNTL(4): level of printing (0 to 4)","None",lu->id.ICNTL(4),&icntl,&flg);CHKERRQ(ierr); 35119facb7aSBarry Smith if ((flg && icntl > 0) || PetscLogPrintInfo) { 352397b6df1SKris Buschelman lu->id.ICNTL(4)=icntl; /* and use mumps default icntl(i), i=1,2,3 */ 353397b6df1SKris Buschelman } else { /* no output */ 354397b6df1SKris Buschelman lu->id.ICNTL(1) = 0; /* error message, default= 6 */ 3553823f358SHong Zhang lu->id.ICNTL(2) = 0; /* output stream for diagnostic printing, statistics, and warning. default=0 */ 3563823f358SHong Zhang lu->id.ICNTL(3) = 0; /* output stream for global information, default=6 */ 357397b6df1SKris Buschelman } 3583823f358SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_6","ICNTL(6): column permutation and/or scaling to get a zero-free diagonal (0 to 7)","None",lu->id.ICNTL(6),&lu->id.ICNTL(6),PETSC_NULL);CHKERRQ(ierr); 359397b6df1SKris Buschelman icntl=-1; 360397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_7","ICNTL(7): matrix ordering (0 to 7)","None",lu->id.ICNTL(7),&icntl,&flg);CHKERRQ(ierr); 361397b6df1SKris Buschelman if (flg) { 362397b6df1SKris Buschelman if (icntl== 1){ 363397b6df1SKris Buschelman SETERRQ(PETSC_ERR_SUP,"pivot order be set by the user in PERM_IN -- not supported by the PETSc/MUMPS interface\n"); 364397b6df1SKris Buschelman } else { 365397b6df1SKris Buschelman lu->id.ICNTL(7) = icntl; 366397b6df1SKris Buschelman } 367397b6df1SKris Buschelman } 3683823f358SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_8","ICNTL(8): scaling strategy (-2 to 7 or 77)","None",lu->id.ICNTL(8),&lu->id.ICNTL(8),PETSC_NULL);CHKERRQ(ierr); 369397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_9","ICNTL(9): A or A^T x=b to be solved. 1: A; otherwise: A^T","None",lu->id.ICNTL(9),&lu->id.ICNTL(9),PETSC_NULL);CHKERRQ(ierr); 370397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_10","ICNTL(10): max num of refinements","None",lu->id.ICNTL(10),&lu->id.ICNTL(10),PETSC_NULL);CHKERRQ(ierr); 371c0165424SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_11","ICNTL(11): statistics related to the linear system solved (via -ksp_view)","None",lu->id.ICNTL(11),&lu->id.ICNTL(11),PETSC_NULL);CHKERRQ(ierr); 3723823f358SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_12","ICNTL(12): efficiency control: defines the ordering strategy with scaling constraints (0 to 3","None",lu->id.ICNTL(12),&lu->id.ICNTL(12),PETSC_NULL);CHKERRQ(ierr); 3733823f358SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_13","ICNTL(13): efficiency control: with or without ScaLAPACK","None",lu->id.ICNTL(13),&lu->id.ICNTL(13),PETSC_NULL);CHKERRQ(ierr); 374adc1d99fSHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_14","ICNTL(14): percentage of estimated workspace increase","None",lu->id.ICNTL(14),&lu->id.ICNTL(14),PETSC_NULL);CHKERRQ(ierr); 3753823f358SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_19","ICNTL(19): Schur complement","None",lu->id.ICNTL(19),&lu->id.ICNTL(19),PETSC_NULL);CHKERRQ(ierr); 3763823f358SHong Zhang 377c0165424SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_22","ICNTL(22): in-core/out-of-core facility (0 or 1)","None",lu->id.ICNTL(22),&lu->id.ICNTL(22),PETSC_NULL);CHKERRQ(ierr); 378c0165424SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_23","ICNTL(23): max size of the working memory (MB) that can allocate per processor","None",lu->id.ICNTL(23),&lu->id.ICNTL(23),PETSC_NULL);CHKERRQ(ierr); 379c0165424SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_24","ICNTL(24): detection of null pivot rows (0 or 1)","None",lu->id.ICNTL(24),&lu->id.ICNTL(24),PETSC_NULL);CHKERRQ(ierr); 380c0165424SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_25","ICNTL(25): computation of a null space basis","None",lu->id.ICNTL(25),&lu->id.ICNTL(25),PETSC_NULL);CHKERRQ(ierr); 381c0165424SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_26","ICNTL(26): Schur options for right-hand side or solution vector","None",lu->id.ICNTL(26),&lu->id.ICNTL(26),PETSC_NULL);CHKERRQ(ierr); 382c0165424SHong Zhang ierr = PetscOptionsInt("-mat_mumps_icntl_27","ICNTL(27): experimental parameter","None",lu->id.ICNTL(27),&lu->id.ICNTL(27),PETSC_NULL);CHKERRQ(ierr); 383397b6df1SKris Buschelman 384397b6df1SKris Buschelman ierr = PetscOptionsReal("-mat_mumps_cntl_1","CNTL(1): relative pivoting threshold","None",lu->id.CNTL(1),&lu->id.CNTL(1),PETSC_NULL);CHKERRQ(ierr); 385397b6df1SKris Buschelman ierr = PetscOptionsReal("-mat_mumps_cntl_2","CNTL(2): stopping criterion of refinement","None",lu->id.CNTL(2),&lu->id.CNTL(2),PETSC_NULL);CHKERRQ(ierr); 386397b6df1SKris Buschelman ierr = PetscOptionsReal("-mat_mumps_cntl_3","CNTL(3): absolute pivoting threshold","None",lu->id.CNTL(3),&lu->id.CNTL(3),PETSC_NULL);CHKERRQ(ierr); 38725f9c88cSHong Zhang ierr = PetscOptionsReal("-mat_mumps_cntl_4","CNTL(4): value for static pivoting","None",lu->id.CNTL(4),&lu->id.CNTL(4),PETSC_NULL);CHKERRQ(ierr); 388c0165424SHong Zhang ierr = PetscOptionsReal("-mat_mumps_cntl_5","CNTL(5): fixation for null pivots","None",lu->id.CNTL(5),&lu->id.CNTL(5),PETSC_NULL);CHKERRQ(ierr); 389397b6df1SKris Buschelman PetscOptionsEnd(); 390397b6df1SKris Buschelman } 391397b6df1SKris Buschelman 392397b6df1SKris Buschelman /* define matrix A */ 393397b6df1SKris Buschelman switch (lu->id.ICNTL(18)){ 394397b6df1SKris Buschelman case 0: /* centralized assembled matrix input (size=1) */ 395397b6df1SKris Buschelman if (!lu->myid) { 3965c9eb25fSBarry Smith if (isSeqAIJ){ 397397b6df1SKris Buschelman Mat_SeqAIJ *aa = (Mat_SeqAIJ*)A->data; 398397b6df1SKris Buschelman nz = aa->nz; 399397b6df1SKris Buschelman ai = aa->i; aj = aa->j; lu->val = aa->a; 4005c9eb25fSBarry Smith } else if (isSeqSBAIJ) { 401397b6df1SKris Buschelman Mat_SeqSBAIJ *aa = (Mat_SeqSBAIJ*)A->data; 4026c6c5352SBarry Smith nz = aa->nz; 403397b6df1SKris Buschelman ai = aa->i; aj = aa->j; lu->val = aa->a; 4045c9eb25fSBarry Smith } else { 4055c9eb25fSBarry Smith SETERRQ(PETSC_ERR_SUP,"No mumps factorization for this matrix type"); 406397b6df1SKris Buschelman } 407397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ /* first numeric factorization, get irn and jcn */ 4087c307921SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&lu->irn);CHKERRQ(ierr); 4097c307921SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&lu->jcn);CHKERRQ(ierr); 410397b6df1SKris Buschelman nz = 0; 411397b6df1SKris Buschelman for (i=0; i<M; i++){ 412397b6df1SKris Buschelman rnz = ai[i+1] - ai[i]; 413397b6df1SKris Buschelman while (rnz--) { /* Fortran row/col index! */ 414397b6df1SKris Buschelman lu->irn[nz] = i+1; lu->jcn[nz] = (*aj)+1; aj++; nz++; 415397b6df1SKris Buschelman } 416397b6df1SKris Buschelman } 417397b6df1SKris Buschelman } 418397b6df1SKris Buschelman } 419397b6df1SKris Buschelman break; 420397b6df1SKris Buschelman case 3: /* distributed assembled matrix input (size>1) */ 421397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ 422397b6df1SKris Buschelman valOnly = PETSC_FALSE; 423397b6df1SKris Buschelman } else { 424397b6df1SKris Buschelman valOnly = PETSC_TRUE; /* only update mat values, not row and col index */ 425397b6df1SKris Buschelman } 426397b6df1SKris Buschelman ierr = MatConvertToTriples(A,1,valOnly, &nnz, &lu->irn, &lu->jcn, &lu->val);CHKERRQ(ierr); 427397b6df1SKris Buschelman break; 428397b6df1SKris Buschelman default: SETERRQ(PETSC_ERR_SUP,"Matrix input format is not supported by MUMPS."); 429397b6df1SKris Buschelman } 430397b6df1SKris Buschelman 431397b6df1SKris Buschelman /* analysis phase */ 432329ec9b3SHong Zhang /*----------------*/ 433397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ 434329ec9b3SHong Zhang lu->id.job = 1; 435329ec9b3SHong Zhang 436397b6df1SKris Buschelman lu->id.n = M; 437397b6df1SKris Buschelman switch (lu->id.ICNTL(18)){ 438397b6df1SKris Buschelman case 0: /* centralized assembled matrix input */ 439397b6df1SKris Buschelman if (!lu->myid) { 440397b6df1SKris Buschelman lu->id.nz =nz; lu->id.irn=lu->irn; lu->id.jcn=lu->jcn; 441397b6df1SKris Buschelman if (lu->id.ICNTL(6)>1){ 442397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 443397b6df1SKris Buschelman lu->id.a = (mumps_double_complex*)lu->val; 444397b6df1SKris Buschelman #else 445397b6df1SKris Buschelman lu->id.a = lu->val; 446397b6df1SKris Buschelman #endif 447397b6df1SKris Buschelman } 448397b6df1SKris Buschelman } 449397b6df1SKris Buschelman break; 450397b6df1SKris Buschelman case 3: /* distributed assembled matrix input (size>1) */ 451397b6df1SKris Buschelman lu->id.nz_loc = nnz; 452397b6df1SKris Buschelman lu->id.irn_loc=lu->irn; lu->id.jcn_loc=lu->jcn; 453397b6df1SKris Buschelman if (lu->id.ICNTL(6)>1) { 454397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 455397b6df1SKris Buschelman lu->id.a_loc = (mumps_double_complex*)lu->val; 456397b6df1SKris Buschelman #else 457397b6df1SKris Buschelman lu->id.a_loc = lu->val; 458397b6df1SKris Buschelman #endif 459397b6df1SKris Buschelman } 460329ec9b3SHong Zhang /* MUMPS only supports centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */ 461329ec9b3SHong Zhang if (!lu->myid){ 462d0f46423SBarry Smith ierr = VecCreateSeq(PETSC_COMM_SELF,A->cmap->N,&lu->b_seq);CHKERRQ(ierr); 463d0f46423SBarry Smith ierr = ISCreateStride(PETSC_COMM_SELF,A->cmap->N,0,1,&is_iden);CHKERRQ(ierr); 464329ec9b3SHong Zhang } else { 465329ec9b3SHong Zhang ierr = VecCreateSeq(PETSC_COMM_SELF,0,&lu->b_seq);CHKERRQ(ierr); 466329ec9b3SHong Zhang ierr = ISCreateStride(PETSC_COMM_SELF,0,0,1,&is_iden);CHKERRQ(ierr); 467329ec9b3SHong Zhang } 4687adad957SLisandro Dalcin ierr = VecCreate(((PetscObject)A)->comm,&b);CHKERRQ(ierr); 469d0f46423SBarry Smith ierr = VecSetSizes(b,A->rmap->n,PETSC_DECIDE);CHKERRQ(ierr); 470329ec9b3SHong Zhang ierr = VecSetFromOptions(b);CHKERRQ(ierr); 471329ec9b3SHong Zhang 472329ec9b3SHong Zhang ierr = VecScatterCreate(b,is_iden,lu->b_seq,is_iden,&lu->scat_rhs);CHKERRQ(ierr); 473329ec9b3SHong Zhang ierr = ISDestroy(is_iden);CHKERRQ(ierr); 474329ec9b3SHong Zhang ierr = VecDestroy(b);CHKERRQ(ierr); 475397b6df1SKris Buschelman break; 476397b6df1SKris Buschelman } 477397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 478397b6df1SKris Buschelman zmumps_c(&lu->id); 479397b6df1SKris Buschelman #else 480397b6df1SKris Buschelman dmumps_c(&lu->id); 481397b6df1SKris Buschelman #endif 482397b6df1SKris Buschelman if (lu->id.INFOG(1) < 0) { 48379a5c55eSBarry Smith SETERRQ1(PETSC_ERR_LIB,"Error reported by MUMPS in analysis phase: INFOG(1)=%d\n",lu->id.INFOG(1)); 484397b6df1SKris Buschelman } 485397b6df1SKris Buschelman } 486397b6df1SKris Buschelman 487397b6df1SKris Buschelman /* numerical factorization phase */ 488329ec9b3SHong Zhang /*-------------------------------*/ 489329ec9b3SHong Zhang lu->id.job = 2; 490958c9bccSBarry Smith if(!lu->id.ICNTL(18)) { 491a7aca84bSHong Zhang if (!lu->myid) { 492397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 493397b6df1SKris Buschelman lu->id.a = (mumps_double_complex*)lu->val; 494397b6df1SKris Buschelman #else 495397b6df1SKris Buschelman lu->id.a = lu->val; 496397b6df1SKris Buschelman #endif 497397b6df1SKris Buschelman } 498397b6df1SKris Buschelman } else { 499397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 500397b6df1SKris Buschelman lu->id.a_loc = (mumps_double_complex*)lu->val; 501397b6df1SKris Buschelman #else 502397b6df1SKris Buschelman lu->id.a_loc = lu->val; 503397b6df1SKris Buschelman #endif 504397b6df1SKris Buschelman } 505397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 506397b6df1SKris Buschelman zmumps_c(&lu->id); 507397b6df1SKris Buschelman #else 508397b6df1SKris Buschelman dmumps_c(&lu->id); 509397b6df1SKris Buschelman #endif 510397b6df1SKris Buschelman if (lu->id.INFOG(1) < 0) { 51119facb7aSBarry Smith if (lu->id.INFO(1) == -13) { 51219facb7aSBarry Smith SETERRQ1(PETSC_ERR_LIB,"Error reported by MUMPS in numerical factorization phase: Cannot allocate required memory %d megabytes\n",lu->id.INFO(2)); 51319facb7aSBarry Smith } else { 51479a5c55eSBarry Smith SETERRQ2(PETSC_ERR_LIB,"Error reported by MUMPS in numerical factorization phase: INFO(1)=%d, INFO(2)=%d\n",lu->id.INFO(1),lu->id.INFO(2)); 515397b6df1SKris Buschelman } 51619facb7aSBarry Smith } 517397b6df1SKris Buschelman 51819facb7aSBarry Smith if (!lu->myid && lu->id.ICNTL(16) > 0){ 51979a5c55eSBarry Smith SETERRQ1(PETSC_ERR_LIB," lu->id.ICNTL(16):=%d\n",lu->id.INFOG(16)); 520397b6df1SKris Buschelman } 521397b6df1SKris Buschelman 5228ada1bb4SHong Zhang if (lu->size > 1){ 523719d5645SBarry Smith if ((F)->factor == MAT_FACTOR_LU){ 524719d5645SBarry Smith F_diag = ((Mat_MPIAIJ *)(F)->data)->A; 525e09efc27SHong Zhang } else { 526719d5645SBarry Smith F_diag = ((Mat_MPISBAIJ *)(F)->data)->A; 527e09efc27SHong Zhang } 528e09efc27SHong Zhang F_diag->assembled = PETSC_TRUE; 529329ec9b3SHong Zhang if (lu->nSolve){ 530329ec9b3SHong Zhang ierr = VecScatterDestroy(lu->scat_sol);CHKERRQ(ierr); 531329ec9b3SHong Zhang ierr = PetscFree(lu->id.sol_loc);CHKERRQ(ierr); 532329ec9b3SHong Zhang ierr = VecDestroy(lu->x_seq);CHKERRQ(ierr); 533329ec9b3SHong Zhang } 5348ada1bb4SHong Zhang } 535719d5645SBarry Smith (F)->assembled = PETSC_TRUE; 536397b6df1SKris Buschelman lu->matstruc = SAME_NONZERO_PATTERN; 537ace87b0dSHong Zhang lu->CleanUpMUMPS = PETSC_TRUE; 538329ec9b3SHong Zhang lu->nSolve = 0; 539397b6df1SKris Buschelman PetscFunctionReturn(0); 540397b6df1SKris Buschelman } 541397b6df1SKris Buschelman 542b24902e0SBarry Smith 543397b6df1SKris Buschelman /* Note the Petsc r and c permutations are ignored */ 544397b6df1SKris Buschelman #undef __FUNCT__ 545f0c56d0fSKris Buschelman #define __FUNCT__ "MatLUFactorSymbolic_AIJMUMPS" 5460481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_AIJMUMPS(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 547b24902e0SBarry Smith { 548719d5645SBarry Smith Mat_MUMPS *lu = (Mat_MUMPS*)F->spptr; 549397b6df1SKris Buschelman 550397b6df1SKris Buschelman PetscFunctionBegin; 551b24902e0SBarry Smith lu->sym = 0; 552b24902e0SBarry Smith lu->matstruc = DIFFERENT_NONZERO_PATTERN; 553719d5645SBarry Smith F->ops->lufactornumeric = MatFactorNumeric_MUMPS; 554b24902e0SBarry Smith PetscFunctionReturn(0); 555b24902e0SBarry Smith } 556b24902e0SBarry Smith 557b24902e0SBarry Smith 558397b6df1SKris Buschelman /* Note the Petsc r permutation is ignored */ 559397b6df1SKris Buschelman #undef __FUNCT__ 560f0c56d0fSKris Buschelman #define __FUNCT__ "MatCholeskyFactorSymbolic_SBAIJMUMPS" 5610481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SBAIJMUMPS(Mat F,Mat A,IS r,const MatFactorInfo *info) 562b24902e0SBarry Smith { 563719d5645SBarry Smith Mat_MUMPS *lu = (Mat_MUMPS*)(F)->spptr; 564397b6df1SKris Buschelman 565397b6df1SKris Buschelman PetscFunctionBegin; 566b24902e0SBarry Smith lu->sym = 2; 567b24902e0SBarry Smith lu->matstruc = DIFFERENT_NONZERO_PATTERN; 568719d5645SBarry Smith (F)->ops->choleskyfactornumeric = MatFactorNumeric_MUMPS; 569db4efbfdSBarry Smith #if !defined(PETSC_USE_COMPLEX) 570719d5645SBarry Smith (F)->ops->getinertia = MatGetInertia_SBAIJMUMPS; 571db4efbfdSBarry Smith #endif 572b24902e0SBarry Smith PetscFunctionReturn(0); 573b24902e0SBarry Smith } 574b24902e0SBarry Smith 575397b6df1SKris Buschelman #undef __FUNCT__ 576f6c57405SHong Zhang #define __FUNCT__ "MatFactorInfo_MUMPS" 577f6c57405SHong Zhang PetscErrorCode MatFactorInfo_MUMPS(Mat A,PetscViewer viewer) { 578f6c57405SHong Zhang Mat_MUMPS *lu=(Mat_MUMPS*)A->spptr; 579f6c57405SHong Zhang PetscErrorCode ierr; 580f6c57405SHong Zhang 581f6c57405SHong Zhang PetscFunctionBegin; 582f6c57405SHong Zhang /* check if matrix is mumps type */ 583f6c57405SHong Zhang if (A->ops->solve != MatSolve_MUMPS) PetscFunctionReturn(0); 584f6c57405SHong Zhang 585f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"MUMPS run parameters:\n");CHKERRQ(ierr); 586f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," SYM (matrix type): %d \n",lu->id.sym);CHKERRQ(ierr); 587f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," PAR (host participation): %d \n",lu->id.par);CHKERRQ(ierr); 588f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(1) (output for error): %d \n",lu->id.ICNTL(1));CHKERRQ(ierr); 589f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(2) (output of diagnostic msg):%d \n",lu->id.ICNTL(2));CHKERRQ(ierr); 590f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(3) (output for global info): %d \n",lu->id.ICNTL(3));CHKERRQ(ierr); 591f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(4) (level of printing): %d \n",lu->id.ICNTL(4));CHKERRQ(ierr); 592f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(5) (input mat struct): %d \n",lu->id.ICNTL(5));CHKERRQ(ierr); 593f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(6) (matrix prescaling): %d \n",lu->id.ICNTL(6));CHKERRQ(ierr); 594f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(7) (matrix ordering): %d \n",lu->id.ICNTL(7));CHKERRQ(ierr); 595f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(8) (scalling strategy): %d \n",lu->id.ICNTL(8));CHKERRQ(ierr); 596f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(9) (A/A^T x=b is solved): %d \n",lu->id.ICNTL(9));CHKERRQ(ierr); 597f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(10) (max num of refinements): %d \n",lu->id.ICNTL(10));CHKERRQ(ierr); 598f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(11) (error analysis): %d \n",lu->id.ICNTL(11));CHKERRQ(ierr); 599f6c57405SHong Zhang if (!lu->myid && lu->id.ICNTL(11)>0) { 600f6c57405SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(4) (inf norm of input mat): %g\n",lu->id.RINFOG(4));CHKERRQ(ierr); 601f6c57405SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(5) (inf norm of solution): %g\n",lu->id.RINFOG(5));CHKERRQ(ierr); 602f6c57405SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(6) (inf norm of residual): %g\n",lu->id.RINFOG(6));CHKERRQ(ierr); 603f6c57405SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(7),RINFOG(8) (backward error est): %g, %g\n",lu->id.RINFOG(7),lu->id.RINFOG(8));CHKERRQ(ierr); 604f6c57405SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(9) (error estimate): %g \n",lu->id.RINFOG(9));CHKERRQ(ierr); 605f6c57405SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(10),RINFOG(11)(condition numbers): %g, %g\n",lu->id.RINFOG(10),lu->id.RINFOG(11));CHKERRQ(ierr); 606f6c57405SHong Zhang 607f6c57405SHong Zhang } 608f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(12) (efficiency control): %d \n",lu->id.ICNTL(12));CHKERRQ(ierr); 609f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(13) (efficiency control): %d \n",lu->id.ICNTL(13));CHKERRQ(ierr); 610f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(14) (percentage of estimated workspace increase): %d \n",lu->id.ICNTL(14));CHKERRQ(ierr); 611f6c57405SHong Zhang /* ICNTL(15-17) not used */ 612f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(18) (input mat struct): %d \n",lu->id.ICNTL(18));CHKERRQ(ierr); 613f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(19) (Shur complement info): %d \n",lu->id.ICNTL(19));CHKERRQ(ierr); 614f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(20) (rhs sparse pattern): %d \n",lu->id.ICNTL(20));CHKERRQ(ierr); 615f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(21) (solution struct): %d \n",lu->id.ICNTL(21));CHKERRQ(ierr); 616c0165424SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(22) (in-core/out-of-core facility): %d \n",lu->id.ICNTL(22));CHKERRQ(ierr); 617c0165424SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(23) (max size of memory can be allocated locally):%d \n",lu->id.ICNTL(23));CHKERRQ(ierr); 618c0165424SHong Zhang 619c0165424SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(24) (detection of null pivot rows): %d \n",lu->id.ICNTL(24));CHKERRQ(ierr); 620c0165424SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(25) (computation of a null space basis): %d \n",lu->id.ICNTL(25));CHKERRQ(ierr); 621c0165424SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(26) (Schur options for rhs or solution): %d \n",lu->id.ICNTL(26));CHKERRQ(ierr); 622c0165424SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(27) (experimental parameter): %d \n",lu->id.ICNTL(27));CHKERRQ(ierr); 623f6c57405SHong Zhang 624f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," CNTL(1) (relative pivoting threshold): %g \n",lu->id.CNTL(1));CHKERRQ(ierr); 625f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," CNTL(2) (stopping criterion of refinement): %g \n",lu->id.CNTL(2));CHKERRQ(ierr); 626f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," CNTL(3) (absolute pivoting threshold): %g \n",lu->id.CNTL(3));CHKERRQ(ierr); 627f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," CNTL(4) (value of static pivoting): %g \n",lu->id.CNTL(4));CHKERRQ(ierr); 628c0165424SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," CNTL(5) (fixation for null pivots): %g \n",lu->id.CNTL(5));CHKERRQ(ierr); 629f6c57405SHong Zhang 630f6c57405SHong Zhang /* infomation local to each processor */ 631f6c57405SHong Zhang if (!lu->myid) {ierr = PetscPrintf(PETSC_COMM_SELF, " RINFO(1) (local estimated flops for the elimination after analysis): \n");CHKERRQ(ierr);} 6327adad957SLisandro Dalcin ierr = PetscSynchronizedPrintf(((PetscObject)A)->comm," [%d] %g \n",lu->myid,lu->id.RINFO(1));CHKERRQ(ierr); 6337adad957SLisandro Dalcin ierr = PetscSynchronizedFlush(((PetscObject)A)->comm); 634f6c57405SHong Zhang if (!lu->myid) {ierr = PetscPrintf(PETSC_COMM_SELF, " RINFO(2) (local estimated flops for the assembly after factorization): \n");CHKERRQ(ierr);} 6357adad957SLisandro Dalcin ierr = PetscSynchronizedPrintf(((PetscObject)A)->comm," [%d] %g \n",lu->myid,lu->id.RINFO(2));CHKERRQ(ierr); 6367adad957SLisandro Dalcin ierr = PetscSynchronizedFlush(((PetscObject)A)->comm); 637f6c57405SHong Zhang if (!lu->myid) {ierr = PetscPrintf(PETSC_COMM_SELF, " RINFO(3) (local estimated flops for the elimination after factorization): \n");CHKERRQ(ierr);} 6387adad957SLisandro Dalcin ierr = PetscSynchronizedPrintf(((PetscObject)A)->comm," [%d] %g \n",lu->myid,lu->id.RINFO(3));CHKERRQ(ierr); 6397adad957SLisandro Dalcin ierr = PetscSynchronizedFlush(((PetscObject)A)->comm); 640f6c57405SHong Zhang 641f6c57405SHong Zhang if (!lu->myid) {ierr = PetscPrintf(PETSC_COMM_SELF, " INFO(15) (estimated size of (in MB) MUMPS internal data for running numerical factorization): \n");CHKERRQ(ierr);} 6427adad957SLisandro Dalcin ierr = PetscSynchronizedPrintf(((PetscObject)A)->comm," [%d] %d \n",lu->myid,lu->id.INFO(15));CHKERRQ(ierr); 6437adad957SLisandro Dalcin ierr = PetscSynchronizedFlush(((PetscObject)A)->comm); 644f6c57405SHong Zhang 645f6c57405SHong Zhang if (!lu->myid) {ierr = PetscPrintf(PETSC_COMM_SELF, " INFO(16) (size of (in MB) MUMPS internal data used during numerical factorization): \n");CHKERRQ(ierr);} 6467adad957SLisandro Dalcin ierr = PetscSynchronizedPrintf(((PetscObject)A)->comm," [%d] %d \n",lu->myid,lu->id.INFO(16));CHKERRQ(ierr); 6477adad957SLisandro Dalcin ierr = PetscSynchronizedFlush(((PetscObject)A)->comm); 648f6c57405SHong Zhang 649f6c57405SHong Zhang if (!lu->myid) {ierr = PetscPrintf(PETSC_COMM_SELF, " INFO(23) (num of pivots eliminated on this processor after factorization): \n");CHKERRQ(ierr);} 6507adad957SLisandro Dalcin ierr = PetscSynchronizedPrintf(((PetscObject)A)->comm," [%d] %d \n",lu->myid,lu->id.INFO(23));CHKERRQ(ierr); 6517adad957SLisandro Dalcin ierr = PetscSynchronizedFlush(((PetscObject)A)->comm); 652f6c57405SHong Zhang 653f6c57405SHong Zhang if (!lu->myid){ /* information from the host */ 654f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," RINFOG(1) (global estimated flops for the elimination after analysis): %g \n",lu->id.RINFOG(1));CHKERRQ(ierr); 655f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," RINFOG(2) (global estimated flops for the assembly after factorization): %g \n",lu->id.RINFOG(2));CHKERRQ(ierr); 656f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," RINFOG(3) (global estimated flops for the elimination after factorization): %g \n",lu->id.RINFOG(3));CHKERRQ(ierr); 657f6c57405SHong Zhang 658f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(3) (estimated real workspace for factors on all processors after analysis): %d \n",lu->id.INFOG(3));CHKERRQ(ierr); 659f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(4) (estimated integer workspace for factors on all processors after analysis): %d \n",lu->id.INFOG(4));CHKERRQ(ierr); 660f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(5) (estimated maximum front size in the complete tree): %d \n",lu->id.INFOG(5));CHKERRQ(ierr); 661f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(6) (number of nodes in the complete tree): %d \n",lu->id.INFOG(6));CHKERRQ(ierr); 662f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(7) (ordering option effectively uese after analysis): %d \n",lu->id.INFOG(7));CHKERRQ(ierr); 663f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(8) (structural symmetry in percent of the permuted matrix after analysis): %d \n",lu->id.INFOG(8));CHKERRQ(ierr); 664f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(9) (total real/complex workspace to store the matrix factors after factorization): %d \n",lu->id.INFOG(9));CHKERRQ(ierr); 665f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(10) (total integer space store the matrix factors after factorization): %d \n",lu->id.INFOG(10));CHKERRQ(ierr); 666f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(11) (order of largest frontal matrix after factorization): %d \n",lu->id.INFOG(11));CHKERRQ(ierr); 667f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(12) (number of off-diagonal pivots): %d \n",lu->id.INFOG(12));CHKERRQ(ierr); 668f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(13) (number of delayed pivots after factorization): %d \n",lu->id.INFOG(13));CHKERRQ(ierr); 669f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(14) (number of memory compress after factorization): %d \n",lu->id.INFOG(14));CHKERRQ(ierr); 670f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(15) (number of steps of iterative refinement after solution): %d \n",lu->id.INFOG(15));CHKERRQ(ierr); 671f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(16) (estimated size (in MB) of all MUMPS internal data for factorization after analysis: value on the most memory consuming processor): %d \n",lu->id.INFOG(16));CHKERRQ(ierr); 672f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(17) (estimated size of all MUMPS internal data for factorization after analysis: sum over all processors): %d \n",lu->id.INFOG(17));CHKERRQ(ierr); 673f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(18) (size of all MUMPS internal data allocated during factorization: value on the most memory consuming processor): %d \n",lu->id.INFOG(18));CHKERRQ(ierr); 674f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(19) (size of all MUMPS internal data allocated during factorization: sum over all processors): %d \n",lu->id.INFOG(19));CHKERRQ(ierr); 675f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(20) (estimated number of entries in the factors): %d \n",lu->id.INFOG(20));CHKERRQ(ierr); 676f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(21) (size in MB of memory effectively used during factorization - value on the most memory consuming processor): %d \n",lu->id.INFOG(21));CHKERRQ(ierr); 677f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(22) (size in MB of memory effectively used during factorization - sum over all processors): %d \n",lu->id.INFOG(22));CHKERRQ(ierr); 678f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(23) (after analysis: value of ICNTL(6) effectively used): %d \n",lu->id.INFOG(23));CHKERRQ(ierr); 679f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(24) (after analysis: value of ICNTL(12) effectively used): %d \n",lu->id.INFOG(24));CHKERRQ(ierr); 680f6c57405SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(25) (after factorization: number of pivots modified by static pivoting): %d \n",lu->id.INFOG(25));CHKERRQ(ierr); 681f6c57405SHong Zhang } 682f6c57405SHong Zhang 683f6c57405SHong Zhang PetscFunctionReturn(0); 684f6c57405SHong Zhang } 685f6c57405SHong Zhang 686f6c57405SHong Zhang #undef __FUNCT__ 687f6c57405SHong Zhang #define __FUNCT__ "MatView_MUMPS" 688b24902e0SBarry Smith PetscErrorCode MatView_MUMPS(Mat A,PetscViewer viewer) 689b24902e0SBarry Smith { 690f6c57405SHong Zhang PetscErrorCode ierr; 691f6c57405SHong Zhang PetscTruth iascii; 692f6c57405SHong Zhang PetscViewerFormat format; 693f6c57405SHong Zhang 694f6c57405SHong Zhang PetscFunctionBegin; 695f6c57405SHong Zhang ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 696f6c57405SHong Zhang if (iascii) { 697f6c57405SHong Zhang ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 698f6c57405SHong Zhang if (format == PETSC_VIEWER_ASCII_INFO){ 699f6c57405SHong Zhang ierr = MatFactorInfo_MUMPS(A,viewer);CHKERRQ(ierr); 700f6c57405SHong Zhang } 701f6c57405SHong Zhang } 702f6c57405SHong Zhang PetscFunctionReturn(0); 703f6c57405SHong Zhang } 704f6c57405SHong Zhang 70535bd34faSBarry Smith #undef __FUNCT__ 70635bd34faSBarry Smith #define __FUNCT__ "MatGetInfo_MUMPS" 70735bd34faSBarry Smith PetscErrorCode MatGetInfo_MUMPS(Mat A,MatInfoType flag,MatInfo *info) 70835bd34faSBarry Smith { 70935bd34faSBarry Smith Mat_MUMPS *lu =(Mat_MUMPS*)A->spptr; 71035bd34faSBarry Smith 71135bd34faSBarry Smith PetscFunctionBegin; 71235bd34faSBarry Smith info->block_size = 1.0; 71335bd34faSBarry Smith info->nz_allocated = lu->id.INFOG(20); 71435bd34faSBarry Smith info->nz_used = lu->id.INFOG(20); 71535bd34faSBarry Smith info->nz_unneeded = 0.0; 71635bd34faSBarry Smith info->assemblies = 0.0; 71735bd34faSBarry Smith info->mallocs = 0.0; 71835bd34faSBarry Smith info->memory = 0.0; 71935bd34faSBarry Smith info->fill_ratio_given = 0; 72035bd34faSBarry Smith info->fill_ratio_needed = 0; 72135bd34faSBarry Smith info->factor_mallocs = 0; 72235bd34faSBarry Smith PetscFunctionReturn(0); 72335bd34faSBarry Smith } 72435bd34faSBarry Smith 72524b6179bSKris Buschelman /*MC 72641c8de11SBarry Smith MAT_SOLVER_MUMPS - A matrix type providing direct solvers (LU and Cholesky) for 72724b6179bSKris Buschelman distributed and sequential matrices via the external package MUMPS. 72824b6179bSKris Buschelman 72941c8de11SBarry Smith Works with MATAIJ and MATSBAIJ matrices 73024b6179bSKris Buschelman 73124b6179bSKris Buschelman Options Database Keys: 73241c8de11SBarry Smith + -mat_mumps_sym <0,1,2> - 0 the matrix is unsymmetric, 1 symmetric positive definite, 2 symmetric 73324b6179bSKris Buschelman . -mat_mumps_icntl_4 <0,...,4> - print level 73424b6179bSKris Buschelman . -mat_mumps_icntl_6 <0,...,7> - matrix prescaling options (see MUMPS User's Guide) 73524b6179bSKris Buschelman . -mat_mumps_icntl_7 <0,...,7> - matrix orderings (see MUMPS User's Guide) 73624b6179bSKris Buschelman . -mat_mumps_icntl_9 <1,2> - A or A^T x=b to be solved: 1 denotes A, 2 denotes A^T 73724b6179bSKris Buschelman . -mat_mumps_icntl_10 <n> - maximum number of iterative refinements 73894b7f48cSBarry Smith . -mat_mumps_icntl_11 <n> - error analysis, a positive value returns statistics during -ksp_view 73924b6179bSKris Buschelman . -mat_mumps_icntl_12 <n> - efficiency control (see MUMPS User's Guide) 74024b6179bSKris Buschelman . -mat_mumps_icntl_13 <n> - efficiency control (see MUMPS User's Guide) 74124b6179bSKris Buschelman . -mat_mumps_icntl_14 <n> - efficiency control (see MUMPS User's Guide) 74224b6179bSKris Buschelman . -mat_mumps_icntl_15 <n> - efficiency control (see MUMPS User's Guide) 74324b6179bSKris Buschelman . -mat_mumps_cntl_1 <delta> - relative pivoting threshold 74424b6179bSKris Buschelman . -mat_mumps_cntl_2 <tol> - stopping criterion for refinement 74524b6179bSKris Buschelman - -mat_mumps_cntl_3 <adelta> - absolute pivoting threshold 74624b6179bSKris Buschelman 74724b6179bSKris Buschelman Level: beginner 74824b6179bSKris Buschelman 74941c8de11SBarry Smith .seealso: PCFactorSetMatSolverPackage(), MatSolverPackage 75041c8de11SBarry Smith 75124b6179bSKris Buschelman M*/ 75224b6179bSKris Buschelman 7532877fffaSHong Zhang EXTERN_C_BEGIN 75435bd34faSBarry Smith #undef __FUNCT__ 75535bd34faSBarry Smith #define __FUNCT__ "MatFactorGetSolverPackage_mumps" 75635bd34faSBarry Smith PetscErrorCode MatFactorGetSolverPackage_mumps(Mat A,const MatSolverPackage *type) 75735bd34faSBarry Smith { 75835bd34faSBarry Smith PetscFunctionBegin; 75935bd34faSBarry Smith *type = MAT_SOLVER_MUMPS; 76035bd34faSBarry Smith PetscFunctionReturn(0); 76135bd34faSBarry Smith } 76235bd34faSBarry Smith EXTERN_C_END 76335bd34faSBarry Smith 76435bd34faSBarry Smith EXTERN_C_BEGIN 7652877fffaSHong Zhang /* 7662877fffaSHong Zhang The seq and mpi versions of this function are the same 7672877fffaSHong Zhang */ 7682877fffaSHong Zhang #undef __FUNCT__ 7692877fffaSHong Zhang #define __FUNCT__ "MatGetFactor_seqaij_mumps" 7702877fffaSHong Zhang PetscErrorCode MatGetFactor_seqaij_mumps(Mat A,MatFactorType ftype,Mat *F) 7712877fffaSHong Zhang { 7722877fffaSHong Zhang Mat B; 7732877fffaSHong Zhang PetscErrorCode ierr; 7742877fffaSHong Zhang Mat_MUMPS *mumps; 7752877fffaSHong Zhang 7762877fffaSHong Zhang PetscFunctionBegin; 7772877fffaSHong Zhang if (ftype != MAT_FACTOR_LU) { 7782877fffaSHong Zhang SETERRQ(PETSC_ERR_SUP,"Cannot use PETSc AIJ matrices with MUMPS Cholesky, use SBAIJ matrix"); 7792877fffaSHong Zhang } 7802877fffaSHong Zhang /* Create the factorization matrix */ 7812877fffaSHong Zhang ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 7822877fffaSHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 7832877fffaSHong Zhang ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 7842877fffaSHong Zhang ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr); 7852877fffaSHong Zhang 7862877fffaSHong Zhang B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS; 7872877fffaSHong Zhang B->ops->view = MatView_MUMPS; 78835bd34faSBarry Smith B->ops->getinfo = MatGetInfo_MUMPS; 78935bd34faSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_mumps",MatFactorGetSolverPackage_mumps);CHKERRQ(ierr); 7902877fffaSHong Zhang B->factor = MAT_FACTOR_LU; 7912877fffaSHong Zhang 7922877fffaSHong Zhang ierr = PetscNewLog(B,Mat_MUMPS,&mumps);CHKERRQ(ierr); 7932877fffaSHong Zhang mumps->CleanUpMUMPS = PETSC_FALSE; 7942877fffaSHong Zhang mumps->isAIJ = PETSC_TRUE; 7952877fffaSHong Zhang mumps->scat_rhs = PETSC_NULL; 7962877fffaSHong Zhang mumps->scat_sol = PETSC_NULL; 7972877fffaSHong Zhang mumps->nSolve = 0; 7982877fffaSHong Zhang mumps->MatDestroy = B->ops->destroy; 7992877fffaSHong Zhang B->ops->destroy = MatDestroy_MUMPS; 8002877fffaSHong Zhang B->spptr = (void*)mumps; 8012877fffaSHong Zhang 8022877fffaSHong Zhang *F = B; 8032877fffaSHong Zhang PetscFunctionReturn(0); 8042877fffaSHong Zhang } 8052877fffaSHong Zhang EXTERN_C_END 8062877fffaSHong Zhang 8072877fffaSHong Zhang EXTERN_C_BEGIN 8082877fffaSHong Zhang #undef __FUNCT__ 8092877fffaSHong Zhang #define __FUNCT__ "MatGetFactor_mpiaij_mumps" 8102877fffaSHong Zhang PetscErrorCode MatGetFactor_mpiaij_mumps(Mat A,MatFactorType ftype,Mat *F) 8112877fffaSHong Zhang { 8122877fffaSHong Zhang Mat B; 8132877fffaSHong Zhang PetscErrorCode ierr; 8142877fffaSHong Zhang Mat_MUMPS *mumps; 8152877fffaSHong Zhang 8162877fffaSHong Zhang PetscFunctionBegin; 8172877fffaSHong Zhang if (ftype != MAT_FACTOR_LU) { 8182877fffaSHong Zhang SETERRQ(PETSC_ERR_SUP,"Cannot use PETSc AIJ matrices with MUMPS Cholesky, use SBAIJ matrix"); 8192877fffaSHong Zhang } 8202877fffaSHong Zhang /* Create the factorization matrix */ 8212877fffaSHong Zhang ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 8222877fffaSHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 8232877fffaSHong Zhang ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 8242877fffaSHong Zhang ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr); 8252877fffaSHong Zhang ierr = MatMPIAIJSetPreallocation(B,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr); 8262877fffaSHong Zhang 8272877fffaSHong Zhang B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS; 8282877fffaSHong Zhang B->ops->view = MatView_MUMPS; 82935bd34faSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_mumps",MatFactorGetSolverPackage_mumps);CHKERRQ(ierr); 8302877fffaSHong Zhang B->factor = MAT_FACTOR_LU; 8312877fffaSHong Zhang 8322877fffaSHong Zhang ierr = PetscNewLog(B,Mat_MUMPS,&mumps);CHKERRQ(ierr); 8332877fffaSHong Zhang mumps->CleanUpMUMPS = PETSC_FALSE; 8342877fffaSHong Zhang mumps->isAIJ = PETSC_TRUE; 8352877fffaSHong Zhang mumps->scat_rhs = PETSC_NULL; 8362877fffaSHong Zhang mumps->scat_sol = PETSC_NULL; 8372877fffaSHong Zhang mumps->nSolve = 0; 838f3c0ef26SHong Zhang mumps->MatDestroy = B->ops->destroy; 839f3c0ef26SHong Zhang B->ops->destroy = MatDestroy_MUMPS; 8402877fffaSHong Zhang B->spptr = (void*)mumps; 8412877fffaSHong Zhang 8422877fffaSHong Zhang *F = B; 8432877fffaSHong Zhang PetscFunctionReturn(0); 8442877fffaSHong Zhang } 8452877fffaSHong Zhang EXTERN_C_END 8462877fffaSHong Zhang 8472877fffaSHong Zhang EXTERN_C_BEGIN 8482877fffaSHong Zhang #undef __FUNCT__ 8492877fffaSHong Zhang #define __FUNCT__ "MatGetFactor_seqsbaij_mumps" 8502877fffaSHong Zhang PetscErrorCode MatGetFactor_seqsbaij_mumps(Mat A,MatFactorType ftype,Mat *F) 8512877fffaSHong Zhang { 8522877fffaSHong Zhang Mat B; 8532877fffaSHong Zhang PetscErrorCode ierr; 8542877fffaSHong Zhang Mat_MUMPS *mumps; 8552877fffaSHong Zhang 8562877fffaSHong Zhang PetscFunctionBegin; 8572877fffaSHong Zhang if (ftype != MAT_FACTOR_CHOLESKY) { 8582877fffaSHong Zhang SETERRQ(PETSC_ERR_SUP,"Cannot use PETSc SBAIJ matrices with MUMPS LU, use AIJ matrix"); 8592877fffaSHong Zhang } 8602877fffaSHong Zhang /* Create the factorization matrix */ 8612877fffaSHong Zhang ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 8622877fffaSHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 8632877fffaSHong Zhang ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 8642877fffaSHong Zhang ierr = MatSeqSBAIJSetPreallocation(B,1,0,PETSC_NULL);CHKERRQ(ierr); 8652877fffaSHong Zhang ierr = MatMPISBAIJSetPreallocation(B,1,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr); 8662877fffaSHong Zhang 8672877fffaSHong Zhang B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SBAIJMUMPS; 8682877fffaSHong Zhang B->ops->view = MatView_MUMPS; 86935bd34faSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_mumps",MatFactorGetSolverPackage_mumps);CHKERRQ(ierr); 87035bd34faSBarry Smith 8712877fffaSHong Zhang B->factor = MAT_FACTOR_CHOLESKY; 8722877fffaSHong Zhang 8732877fffaSHong Zhang ierr = PetscNewLog(B,Mat_MUMPS,&mumps);CHKERRQ(ierr); 8742877fffaSHong Zhang mumps->CleanUpMUMPS = PETSC_FALSE; 8752877fffaSHong Zhang mumps->isAIJ = PETSC_TRUE; 8762877fffaSHong Zhang mumps->scat_rhs = PETSC_NULL; 8772877fffaSHong Zhang mumps->scat_sol = PETSC_NULL; 8782877fffaSHong Zhang mumps->nSolve = 0; 8792877fffaSHong Zhang mumps->MatDestroy = B->ops->destroy; 8802877fffaSHong Zhang B->ops->destroy = MatDestroy_MUMPS; 8812877fffaSHong Zhang B->spptr = (void*)mumps; 882f3c0ef26SHong Zhang 8832877fffaSHong Zhang *F = B; 8842877fffaSHong Zhang PetscFunctionReturn(0); 8852877fffaSHong Zhang } 8862877fffaSHong Zhang EXTERN_C_END 8872877fffaSHong Zhang 8882877fffaSHong Zhang EXTERN_C_BEGIN 8892877fffaSHong Zhang #undef __FUNCT__ 8902877fffaSHong Zhang #define __FUNCT__ "MatGetFactor_mpisbaij_mumps" 8912877fffaSHong Zhang PetscErrorCode MatGetFactor_mpisbaij_mumps(Mat A,MatFactorType ftype,Mat *F) 8922877fffaSHong Zhang { 8932877fffaSHong Zhang Mat B; 8942877fffaSHong Zhang PetscErrorCode ierr; 8952877fffaSHong Zhang Mat_MUMPS *mumps; 8962877fffaSHong Zhang 8972877fffaSHong Zhang PetscFunctionBegin; 8982877fffaSHong Zhang if (ftype != MAT_FACTOR_CHOLESKY) { 8992877fffaSHong Zhang SETERRQ(PETSC_ERR_SUP,"Cannot use PETSc SBAIJ matrices with MUMPS LU, use AIJ matrix"); 9002877fffaSHong Zhang } 9012877fffaSHong Zhang /* Create the factorization matrix */ 9022877fffaSHong Zhang ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 9032877fffaSHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 9042877fffaSHong Zhang ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 9052877fffaSHong Zhang ierr = MatSeqSBAIJSetPreallocation(B,1,0,PETSC_NULL);CHKERRQ(ierr); 9062877fffaSHong Zhang ierr = MatMPISBAIJSetPreallocation(B,1,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr); 9072877fffaSHong Zhang 9082877fffaSHong Zhang B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SBAIJMUMPS; 9092877fffaSHong Zhang B->ops->view = MatView_MUMPS; 91035bd34faSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_mumps",MatFactorGetSolverPackage_mumps);CHKERRQ(ierr); 9112877fffaSHong Zhang B->factor = MAT_FACTOR_CHOLESKY; 9122877fffaSHong Zhang 9132877fffaSHong Zhang ierr = PetscNewLog(B,Mat_MUMPS,&mumps);CHKERRQ(ierr); 9142877fffaSHong Zhang mumps->CleanUpMUMPS = PETSC_FALSE; 9152877fffaSHong Zhang mumps->isAIJ = PETSC_TRUE; 9162877fffaSHong Zhang mumps->scat_rhs = PETSC_NULL; 9172877fffaSHong Zhang mumps->scat_sol = PETSC_NULL; 9182877fffaSHong Zhang mumps->nSolve = 0; 919f3c0ef26SHong Zhang mumps->MatDestroy = B->ops->destroy; 920f3c0ef26SHong Zhang B->ops->destroy = MatDestroy_MUMPS; 9212877fffaSHong Zhang B->spptr = (void*)mumps; 922f3c0ef26SHong Zhang 9232877fffaSHong Zhang *F = B; 9242877fffaSHong Zhang PetscFunctionReturn(0); 9252877fffaSHong Zhang } 9262877fffaSHong Zhang EXTERN_C_END 927