1397b6df1SKris Buschelman /*$Id: mumps.c,v 1.10 2001/08/15 15:56:50 bsmith Exp $*/ 2397b6df1SKris Buschelman /* 33a0aa19bSHong Zhang Provides an interface to the MUMPS_4.3 sparse solver 4397b6df1SKris Buschelman */ 5397b6df1SKris Buschelman #include "src/mat/impls/aij/seq/aij.h" 6397b6df1SKris Buschelman #include "src/mat/impls/aij/mpi/mpiaij.h" 7397b6df1SKris Buschelman #include "src/mat/impls/sbaij/seq/sbaij.h" 8397b6df1SKris Buschelman #include "src/mat/impls/sbaij/mpi/mpisbaij.h" 9397b6df1SKris Buschelman 10397b6df1SKris Buschelman EXTERN_C_BEGIN 11397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 12397b6df1SKris Buschelman #include "zmumps_c.h" 13397b6df1SKris Buschelman #else 14397b6df1SKris Buschelman #include "dmumps_c.h" 15397b6df1SKris Buschelman #endif 16397b6df1SKris Buschelman EXTERN_C_END 17397b6df1SKris Buschelman #define JOB_INIT -1 18397b6df1SKris Buschelman #define JOB_END -2 19397b6df1SKris Buschelman /* macros s.t. indices match MUMPS documentation */ 20397b6df1SKris Buschelman #define ICNTL(I) icntl[(I)-1] 21397b6df1SKris Buschelman #define CNTL(I) cntl[(I)-1] 22397b6df1SKris Buschelman #define INFOG(I) infog[(I)-1] 23397b6df1SKris Buschelman #define RINFOG(I) rinfog[(I)-1] 24adc1d99fSHong Zhang #define RINFO(I) rinfo[(I)-1] 25397b6df1SKris Buschelman 26397b6df1SKris Buschelman typedef struct { 27397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 28397b6df1SKris Buschelman ZMUMPS_STRUC_C id; 29397b6df1SKris Buschelman #else 30397b6df1SKris Buschelman DMUMPS_STRUC_C id; 31397b6df1SKris Buschelman #endif 32397b6df1SKris Buschelman MatStructure matstruc; 33397b6df1SKris Buschelman int myid,size,*irn,*jcn,sym; 34397b6df1SKris Buschelman PetscScalar *val; 35397b6df1SKris Buschelman MPI_Comm comm_mumps; 36397b6df1SKris Buschelman 37c338a77dSKris Buschelman PetscTruth isAIJ,CleanUpMUMPS; 38f0c56d0fSKris Buschelman int (*MatDuplicate)(Mat,MatDuplicateOption,Mat*); 39c338a77dSKris Buschelman int (*MatView)(Mat,PetscViewer); 40c338a77dSKris Buschelman int (*MatAssemblyEnd)(Mat,MatAssemblyType); 41c338a77dSKris Buschelman int (*MatLUFactorSymbolic)(Mat,IS,IS,MatFactorInfo*,Mat*); 42c338a77dSKris Buschelman int (*MatCholeskyFactorSymbolic)(Mat,IS,MatFactorInfo*,Mat*); 43c338a77dSKris Buschelman int (*MatDestroy)(Mat); 44a39386dcSKris Buschelman int (*specialdestroy)(Mat); 45f0c56d0fSKris Buschelman } Mat_MUMPS; 46f0c56d0fSKris Buschelman 47f0c56d0fSKris Buschelman EXTERN int MatDuplicate_AIJMUMPS(Mat,MatDuplicateOption,Mat*); 48f0c56d0fSKris Buschelman EXTERN int MatDuplicate_SBAIJMUMPS(Mat,MatDuplicateOption,Mat*); 490e3434eeSKris Buschelman 50397b6df1SKris Buschelman /* convert Petsc mpiaij matrix to triples: row[nz], col[nz], val[nz] */ 51397b6df1SKris Buschelman /* 52397b6df1SKris Buschelman input: 5375747be1SHong Zhang A - matrix in mpiaij or mpisbaij (bs=1) format 54397b6df1SKris Buschelman shift - 0: C style output triple; 1: Fortran style output triple. 55397b6df1SKris Buschelman valOnly - FALSE: spaces are allocated and values are set for the triple 56397b6df1SKris Buschelman TRUE: only the values in v array are updated 57397b6df1SKris Buschelman output: 58397b6df1SKris Buschelman nnz - dim of r, c, and v (number of local nonzero entries of A) 59397b6df1SKris Buschelman r, c, v - row and col index, matrix values (matrix triples) 60397b6df1SKris Buschelman */ 61f0c56d0fSKris Buschelman int MatConvertToTriples(Mat A,int shift,PetscTruth valOnly,int *nnz,int **r, int **c, PetscScalar **v) { 62397b6df1SKris Buschelman int *ai, *aj, *bi, *bj, rstart,nz, *garray; 63397b6df1SKris Buschelman int ierr,i,j,jj,jB,irow,m=A->m,*ajj,*bjj,countA,countB,colA_start,jcol; 64d54de34fSKris Buschelman int *row,*col; 65397b6df1SKris Buschelman PetscScalar *av, *bv,*val; 66f0c56d0fSKris Buschelman Mat_MUMPS *mumps=(Mat_MUMPS*)A->spptr; 67397b6df1SKris Buschelman 68397b6df1SKris Buschelman PetscFunctionBegin; 69397b6df1SKris Buschelman if (mumps->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; 74397b6df1SKris Buschelman ai=aa->i; aj=aa->j; bi=bb->i; bj=bb->j; rstart= mat->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; 82847143adSKris Buschelman if (mat->bs > 1) SETERRQ1(PETSC_ERR_SUP," bs=%d is not supported yet\n", mat->bs); 83397b6df1SKris Buschelman nz = aa->s_nz + bb->nz; 84397b6df1SKris Buschelman ai=aa->i; aj=aa->j; bi=bb->i; bj=bb->j; rstart= mat->rstart; 85397b6df1SKris Buschelman garray = mat->garray; 86397b6df1SKris Buschelman av=aa->a; bv=bb->a; 87397b6df1SKris Buschelman } 88397b6df1SKris Buschelman 89397b6df1SKris Buschelman if (!valOnly){ 90397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(int),&row);CHKERRQ(ierr); 91397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(int),&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 146c338a77dSKris Buschelman EXTERN_C_BEGIN 147c338a77dSKris Buschelman #undef __FUNCT__ 148c338a77dSKris Buschelman #define __FUNCT__ "MatConvert_MUMPS_Base" 149c338a77dSKris Buschelman int MatConvert_MUMPS_Base(Mat A,MatType type,Mat *newmat) { 150c338a77dSKris Buschelman int ierr; 151c338a77dSKris Buschelman Mat B=*newmat; 152f0c56d0fSKris Buschelman Mat_MUMPS *mumps=(Mat_MUMPS*)A->spptr; 153c338a77dSKris Buschelman 154c338a77dSKris Buschelman PetscFunctionBegin; 155c338a77dSKris Buschelman if (B != A) { 156c338a77dSKris Buschelman ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr); 157c338a77dSKris Buschelman } 158f0c56d0fSKris Buschelman B->ops->duplicate = mumps->MatDuplicate; 159f0c56d0fSKris Buschelman B->ops->view = mumps->MatView; 160f0c56d0fSKris Buschelman B->ops->assemblyend = mumps->MatAssemblyEnd; 161f0c56d0fSKris Buschelman B->ops->lufactorsymbolic = mumps->MatLUFactorSymbolic; 162f0c56d0fSKris Buschelman B->ops->choleskyfactorsymbolic = mumps->MatCholeskyFactorSymbolic; 163f0c56d0fSKris Buschelman B->ops->destroy = mumps->MatDestroy; 1643924e44cSKris Buschelman ierr = PetscObjectChangeTypeName((PetscObject)B,type);CHKERRQ(ierr); 165f0c56d0fSKris Buschelman ierr = PetscFree(mumps);CHKERRQ(ierr); 166c338a77dSKris Buschelman *newmat = B; 167c338a77dSKris Buschelman PetscFunctionReturn(0); 168c338a77dSKris Buschelman } 169c338a77dSKris Buschelman EXTERN_C_END 170c338a77dSKris Buschelman 171397b6df1SKris Buschelman #undef __FUNCT__ 1723924e44cSKris Buschelman #define __FUNCT__ "MatDestroy_MUMPS" 1733924e44cSKris Buschelman int MatDestroy_MUMPS(Mat A) { 174f0c56d0fSKris Buschelman Mat_MUMPS *lu=(Mat_MUMPS*)A->spptr; 175c338a77dSKris Buschelman int ierr,size=lu->size; 176a39386dcSKris Buschelman int (*specialdestroy)(Mat); 177397b6df1SKris Buschelman PetscFunctionBegin; 178397b6df1SKris Buschelman if (lu->CleanUpMUMPS) { 179397b6df1SKris Buschelman /* Terminate instance, deallocate memories */ 180397b6df1SKris Buschelman lu->id.job=JOB_END; 181397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 182397b6df1SKris Buschelman zmumps_c(&lu->id); 183397b6df1SKris Buschelman #else 184397b6df1SKris Buschelman dmumps_c(&lu->id); 185397b6df1SKris Buschelman #endif 186c338a77dSKris Buschelman if (lu->irn) { 187c338a77dSKris Buschelman ierr = PetscFree(lu->irn);CHKERRQ(ierr); 188c338a77dSKris Buschelman } 189c338a77dSKris Buschelman if (lu->jcn) { 190c338a77dSKris Buschelman ierr = PetscFree(lu->jcn);CHKERRQ(ierr); 191c338a77dSKris Buschelman } 192c338a77dSKris Buschelman if (size>1 && lu->val) { 193c338a77dSKris Buschelman ierr = PetscFree(lu->val);CHKERRQ(ierr); 194c338a77dSKris Buschelman } 195397b6df1SKris Buschelman ierr = MPI_Comm_free(&(lu->comm_mumps));CHKERRQ(ierr); 196397b6df1SKris Buschelman } 197a39386dcSKris Buschelman specialdestroy = lu->specialdestroy; 198a39386dcSKris Buschelman ierr = (*specialdestroy)(A);CHKERRQ(ierr); 199c338a77dSKris Buschelman ierr = (*A->ops->destroy)(A);CHKERRQ(ierr); 200397b6df1SKris Buschelman PetscFunctionReturn(0); 201397b6df1SKris Buschelman } 202397b6df1SKris Buschelman 203397b6df1SKris Buschelman #undef __FUNCT__ 204a39386dcSKris Buschelman #define __FUNCT__ "MatDestroy_AIJMUMPS" 205a39386dcSKris Buschelman int MatDestroy_AIJMUMPS(Mat A) { 206a39386dcSKris Buschelman int ierr, size; 207a39386dcSKris Buschelman 208a39386dcSKris Buschelman PetscFunctionBegin; 209a39386dcSKris Buschelman ierr = MPI_Comm_size(A->comm,&size);CHKERRQ(ierr); 210a39386dcSKris Buschelman if (size==1) { 211a39386dcSKris Buschelman ierr = MatConvert_MUMPS_Base(A,MATSEQAIJ,&A);CHKERRQ(ierr); 212a39386dcSKris Buschelman } else { 213a39386dcSKris Buschelman ierr = MatConvert_MUMPS_Base(A,MATMPIAIJ,&A);CHKERRQ(ierr); 214a39386dcSKris Buschelman } 215a39386dcSKris Buschelman PetscFunctionReturn(0); 216a39386dcSKris Buschelman } 217a39386dcSKris Buschelman 218a39386dcSKris Buschelman #undef __FUNCT__ 219a39386dcSKris Buschelman #define __FUNCT__ "MatDestroy_SBAIJMUMPS" 220a39386dcSKris Buschelman int MatDestroy_SBAIJMUMPS(Mat A) { 221a39386dcSKris Buschelman int ierr, size; 222a39386dcSKris Buschelman 223a39386dcSKris Buschelman PetscFunctionBegin; 224a39386dcSKris Buschelman ierr = MPI_Comm_size(A->comm,&size);CHKERRQ(ierr); 225a39386dcSKris Buschelman if (size==1) { 226a39386dcSKris Buschelman ierr = MatConvert_MUMPS_Base(A,MATSEQSBAIJ,&A);CHKERRQ(ierr); 227a39386dcSKris Buschelman } else { 228a39386dcSKris Buschelman ierr = MatConvert_MUMPS_Base(A,MATMPISBAIJ,&A);CHKERRQ(ierr); 229a39386dcSKris Buschelman } 230a39386dcSKris Buschelman PetscFunctionReturn(0); 231a39386dcSKris Buschelman } 232a39386dcSKris Buschelman 233a39386dcSKris Buschelman #undef __FUNCT__ 234c338a77dSKris Buschelman #define __FUNCT__ "MatFactorInfo_MUMPS" 235f0c56d0fSKris Buschelman int MatFactorInfo_MUMPS(Mat A,PetscViewer viewer) { 236f0c56d0fSKris Buschelman Mat_MUMPS *lu=(Mat_MUMPS*)A->spptr; 237397b6df1SKris Buschelman int ierr; 238397b6df1SKris Buschelman 239397b6df1SKris Buschelman PetscFunctionBegin; 240c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer,"MUMPS run parameters:\n");CHKERRQ(ierr); 241c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," SYM (matrix type): %d \n",lu->id.sym);CHKERRQ(ierr); 242c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," PAR (host participation): %d \n",lu->id.par);CHKERRQ(ierr); 243c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(4) (level of printing): %d \n",lu->id.ICNTL(4));CHKERRQ(ierr); 244c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(5) (input mat struct): %d \n",lu->id.ICNTL(5));CHKERRQ(ierr); 245c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(6) (matrix prescaling): %d \n",lu->id.ICNTL(6));CHKERRQ(ierr); 246c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(7) (matrix ordering): %d \n",lu->id.ICNTL(7));CHKERRQ(ierr); 247c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(9) (A/A^T x=b is solved): %d \n",lu->id.ICNTL(9));CHKERRQ(ierr); 248c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(10) (max num of refinements): %d \n",lu->id.ICNTL(10));CHKERRQ(ierr); 249c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(11) (error analysis): %d \n",lu->id.ICNTL(11));CHKERRQ(ierr); 250c338a77dSKris Buschelman if (lu->myid == 0 && lu->id.ICNTL(11)>0) { 251c338a77dSKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(4) (inf norm of input mat): %g\n",lu->id.RINFOG(4));CHKERRQ(ierr); 252c338a77dSKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(5) (inf norm of solution): %g\n",lu->id.RINFOG(5));CHKERRQ(ierr); 253c338a77dSKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(6) (inf norm of residual): %g\n",lu->id.RINFOG(6));CHKERRQ(ierr); 254c338a77dSKris Buschelman 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); 255c338a77dSKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(9) (error estimate): %g \n",lu->id.RINFOG(9));CHKERRQ(ierr); 256c338a77dSKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(10),RINFOG(11)(condition numbers): %g, %g\n",lu->id.RINFOG(10),lu->id.RINFOG(11));CHKERRQ(ierr); 257c338a77dSKris Buschelman 258c338a77dSKris Buschelman } 259c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(12) (efficiency control): %d \n",lu->id.ICNTL(12));CHKERRQ(ierr); 260c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(13) (efficiency control): %d \n",lu->id.ICNTL(13));CHKERRQ(ierr); 261adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," ICNTL(14) (percentage of estimated workspace increase): %d \n",lu->id.ICNTL(14));CHKERRQ(ierr); 262c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(15) (efficiency control): %d \n",lu->id.ICNTL(15));CHKERRQ(ierr); 263c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(18) (input mat struct): %d \n",lu->id.ICNTL(18));CHKERRQ(ierr); 264c338a77dSKris Buschelman 265c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," CNTL(1) (relative pivoting threshold): %g \n",lu->id.CNTL(1));CHKERRQ(ierr); 266c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," CNTL(2) (stopping criterion of refinement): %g \n",lu->id.CNTL(2));CHKERRQ(ierr); 267c338a77dSKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," CNTL(3) (absolute pivoting threshold): %g \n",lu->id.CNTL(3));CHKERRQ(ierr); 268*57f0c58bSHong Zhang 269*57f0c58bSHong Zhang /* infomation local to each processor */ 270*57f0c58bSHong Zhang if (lu->myid == 0) ierr = PetscPrintf(PETSC_COMM_SELF, " RINFO(1) (local estimated flops for the elimination after analysis): \n");CHKERRQ(ierr); 271*57f0c58bSHong Zhang ierr = PetscSynchronizedPrintf(A->comm," [%d] %g \n",lu->myid,lu->id.RINFO(1));CHKERRQ(ierr); 272*57f0c58bSHong Zhang ierr = PetscSynchronizedFlush(A->comm); 273*57f0c58bSHong Zhang if (lu->myid == 0) ierr = PetscPrintf(PETSC_COMM_SELF, " RINFO(2) (local estimated flops for the assembly after factorization): \n");CHKERRQ(ierr); 274*57f0c58bSHong Zhang ierr = PetscSynchronizedPrintf(A->comm," [%d] %g \n",lu->myid,lu->id.RINFO(2));CHKERRQ(ierr); 275*57f0c58bSHong Zhang ierr = PetscSynchronizedFlush(A->comm); 276*57f0c58bSHong Zhang if (lu->myid == 0) ierr = PetscPrintf(PETSC_COMM_SELF, " RINFO(3) (local estimated flops for the elimination after factorization): \n");CHKERRQ(ierr); 277*57f0c58bSHong Zhang ierr = PetscSynchronizedPrintf(A->comm," [%d] %g \n",lu->myid,lu->id.RINFO(3));CHKERRQ(ierr); 278*57f0c58bSHong Zhang ierr = PetscSynchronizedFlush(A->comm); 279adc1d99fSHong Zhang 280adc1d99fSHong Zhang if (lu->myid == 0){ /* information from the host */ 281adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," RINFOG(1) (global estimated flops for the elimination after analysis): %g \n",lu->id.RINFOG(1));CHKERRQ(ierr); 282adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," RINFOG(2) (global estimated flops for the assembly after factorization): %g \n",lu->id.RINFOG(2));CHKERRQ(ierr); 283adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," RINFOG(3) (global estimated flops for the elimination after factorization): %g \n",lu->id.RINFOG(3));CHKERRQ(ierr); 284adc1d99fSHong Zhang 285adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(3) (estimated real workspace for factors on all processors after analysis): %d \n",lu->id.INFOG(3));CHKERRQ(ierr); 286adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(4) (estimated integer workspace for factors on all processors after analysis): %d \n",lu->id.INFOG(4));CHKERRQ(ierr); 287adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(5) (estimated maximum front size in the complete tree): %d \n",lu->id.INFOG(5));CHKERRQ(ierr); 288adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(6) (number of nodes in the complete tree): %d \n",lu->id.INFOG(6));CHKERRQ(ierr); 289adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(7) (ordering option effectively uese after analysis): %d \n",lu->id.INFOG(7));CHKERRQ(ierr); 290adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(8) (structural symmetry in percent of the permuted matrix after analysis): %d \n",lu->id.INFOG(8));CHKERRQ(ierr); 291adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(9) (total real space store the matrix factors after analysis): %d \n",lu->id.INFOG(9));CHKERRQ(ierr); 292adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(10) (total integer space store the matrix factors after analysis): %d \n",lu->id.INFOG(10));CHKERRQ(ierr); 293adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(11) (order of largest frontal matrix): %d \n",lu->id.INFOG(11));CHKERRQ(ierr); 294adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(12) (number of off-diagonal pivots): %d \n",lu->id.INFOG(12));CHKERRQ(ierr); 295adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(13) (number of delayed pivots after factorization): %d \n",lu->id.INFOG(13));CHKERRQ(ierr); 296adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(14) (number of memory compress after factorization): %d \n",lu->id.INFOG(14));CHKERRQ(ierr); 297adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(15) (number of steps of iterative refinement after solution): %d \n",lu->id.INFOG(15));CHKERRQ(ierr); 298adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(16) (estimated size (in million of bytes) of all MUMPS internal data for factorization after analysis: value on the most memory consuming processor): %d \n",lu->id.INFOG(16));CHKERRQ(ierr); 299adc1d99fSHong 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); 300adc1d99fSHong 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); 301adc1d99fSHong 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); 302adc1d99fSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," INFOG(20) (estimated number of entries in the factors): %d \n",lu->id.INFOG(20));CHKERRQ(ierr); 303adc1d99fSHong Zhang } 304adc1d99fSHong Zhang 305397b6df1SKris Buschelman PetscFunctionReturn(0); 306397b6df1SKris Buschelman } 307397b6df1SKris Buschelman 308397b6df1SKris Buschelman #undef __FUNCT__ 309f0c56d0fSKris Buschelman #define __FUNCT__ "MatView_AIJMUMPS" 310f0c56d0fSKris Buschelman int MatView_AIJMUMPS(Mat A,PetscViewer viewer) { 311397b6df1SKris Buschelman int ierr; 312397b6df1SKris Buschelman PetscTruth isascii; 313397b6df1SKris Buschelman PetscViewerFormat format; 314f0c56d0fSKris Buschelman Mat_MUMPS *mumps=(Mat_MUMPS*)(A->spptr); 315397b6df1SKris Buschelman 316397b6df1SKris Buschelman PetscFunctionBegin; 317397b6df1SKris Buschelman ierr = (*mumps->MatView)(A,viewer);CHKERRQ(ierr); 318397b6df1SKris Buschelman 319397b6df1SKris Buschelman ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&isascii);CHKERRQ(ierr); 320397b6df1SKris Buschelman if (isascii) { 321397b6df1SKris Buschelman ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 322397b6df1SKris Buschelman if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 323397b6df1SKris Buschelman ierr = MatFactorInfo_MUMPS(A,viewer);CHKERRQ(ierr); 324397b6df1SKris Buschelman } 325397b6df1SKris Buschelman } 326397b6df1SKris Buschelman PetscFunctionReturn(0); 327397b6df1SKris Buschelman } 328397b6df1SKris Buschelman 329397b6df1SKris Buschelman #undef __FUNCT__ 330f0c56d0fSKris Buschelman #define __FUNCT__ "MatSolve_AIJMUMPS" 331f0c56d0fSKris Buschelman int MatSolve_AIJMUMPS(Mat A,Vec b,Vec x) { 332f0c56d0fSKris Buschelman Mat_MUMPS *lu=(Mat_MUMPS*)A->spptr; 333d54de34fSKris Buschelman PetscScalar *array; 334397b6df1SKris Buschelman Vec x_seq; 335397b6df1SKris Buschelman IS iden; 336397b6df1SKris Buschelman VecScatter scat; 337397b6df1SKris Buschelman int ierr; 338397b6df1SKris Buschelman 339397b6df1SKris Buschelman PetscFunctionBegin; 340397b6df1SKris Buschelman if (lu->size > 1){ 341397b6df1SKris Buschelman if (!lu->myid){ 342397b6df1SKris Buschelman ierr = VecCreateSeq(PETSC_COMM_SELF,A->N,&x_seq);CHKERRQ(ierr); 343397b6df1SKris Buschelman ierr = ISCreateStride(PETSC_COMM_SELF,A->N,0,1,&iden);CHKERRQ(ierr); 344397b6df1SKris Buschelman } else { 345397b6df1SKris Buschelman ierr = VecCreateSeq(PETSC_COMM_SELF,0,&x_seq);CHKERRQ(ierr); 346397b6df1SKris Buschelman ierr = ISCreateStride(PETSC_COMM_SELF,0,0,1,&iden);CHKERRQ(ierr); 347397b6df1SKris Buschelman } 348397b6df1SKris Buschelman ierr = VecScatterCreate(b,iden,x_seq,iden,&scat);CHKERRQ(ierr); 349397b6df1SKris Buschelman ierr = ISDestroy(iden);CHKERRQ(ierr); 350397b6df1SKris Buschelman 351397b6df1SKris Buschelman ierr = VecScatterBegin(b,x_seq,INSERT_VALUES,SCATTER_FORWARD,scat);CHKERRQ(ierr); 352397b6df1SKris Buschelman ierr = VecScatterEnd(b,x_seq,INSERT_VALUES,SCATTER_FORWARD,scat);CHKERRQ(ierr); 353397b6df1SKris Buschelman if (!lu->myid) {ierr = VecGetArray(x_seq,&array);CHKERRQ(ierr);} 354397b6df1SKris Buschelman } else { /* size == 1 */ 355397b6df1SKris Buschelman ierr = VecCopy(b,x);CHKERRQ(ierr); 356397b6df1SKris Buschelman ierr = VecGetArray(x,&array);CHKERRQ(ierr); 357397b6df1SKris Buschelman } 358397b6df1SKris Buschelman if (!lu->myid) { /* define rhs on the host */ 359397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 360397b6df1SKris Buschelman lu->id.rhs = (mumps_double_complex*)array; 361397b6df1SKris Buschelman #else 362397b6df1SKris Buschelman lu->id.rhs = array; 363397b6df1SKris Buschelman #endif 364397b6df1SKris Buschelman } 365397b6df1SKris Buschelman 366397b6df1SKris Buschelman /* solve phase */ 367397b6df1SKris Buschelman lu->id.job=3; 368397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 369397b6df1SKris Buschelman zmumps_c(&lu->id); 370397b6df1SKris Buschelman #else 371397b6df1SKris Buschelman dmumps_c(&lu->id); 372397b6df1SKris Buschelman #endif 373397b6df1SKris Buschelman if (lu->id.INFOG(1) < 0) { 374397b6df1SKris Buschelman SETERRQ1(1,"Error reported by MUMPS in solve phase: INFOG(1)=%d\n",lu->id.INFOG(1)); 375397b6df1SKris Buschelman } 376397b6df1SKris Buschelman 377397b6df1SKris Buschelman /* convert mumps solution x_seq to petsc mpi x */ 378397b6df1SKris Buschelman if (lu->size > 1) { 379397b6df1SKris Buschelman if (!lu->myid){ 380397b6df1SKris Buschelman ierr = VecRestoreArray(x_seq,&array);CHKERRQ(ierr); 381397b6df1SKris Buschelman } 382397b6df1SKris Buschelman ierr = VecScatterBegin(x_seq,x,INSERT_VALUES,SCATTER_REVERSE,scat);CHKERRQ(ierr); 383397b6df1SKris Buschelman ierr = VecScatterEnd(x_seq,x,INSERT_VALUES,SCATTER_REVERSE,scat);CHKERRQ(ierr); 384397b6df1SKris Buschelman ierr = VecScatterDestroy(scat);CHKERRQ(ierr); 385397b6df1SKris Buschelman ierr = VecDestroy(x_seq);CHKERRQ(ierr); 386397b6df1SKris Buschelman } else { 387397b6df1SKris Buschelman ierr = VecRestoreArray(x,&array);CHKERRQ(ierr); 388397b6df1SKris Buschelman } 389397b6df1SKris Buschelman 390397b6df1SKris Buschelman PetscFunctionReturn(0); 391397b6df1SKris Buschelman } 392397b6df1SKris Buschelman 393397b6df1SKris Buschelman #undef __FUNCT__ 394f0c56d0fSKris Buschelman #define __FUNCT__ "MatFactorNumeric_MPIAIJMUMPS" 395f0c56d0fSKris Buschelman int MatFactorNumeric_AIJMUMPS(Mat A,Mat *F) { 396f0c56d0fSKris Buschelman Mat_MUMPS *lu =(Mat_MUMPS*)(*F)->spptr; 397f0c56d0fSKris Buschelman Mat_MUMPS *lua=(Mat_MUMPS*)(A)->spptr; 398397b6df1SKris Buschelman int rnz,nnz,ierr,nz,i,M=A->M,*ai,*aj,icntl; 399397b6df1SKris Buschelman PetscTruth valOnly,flg; 400397b6df1SKris Buschelman 401397b6df1SKris Buschelman PetscFunctionBegin; 402397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ 403f0c56d0fSKris Buschelman (*F)->ops->solve = MatSolve_AIJMUMPS; 404397b6df1SKris Buschelman 405397b6df1SKris Buschelman /* Initialize a MUMPS instance */ 406397b6df1SKris Buschelman ierr = MPI_Comm_rank(A->comm, &lu->myid); 407397b6df1SKris Buschelman ierr = MPI_Comm_size(A->comm,&lu->size);CHKERRQ(ierr); 40875747be1SHong Zhang lua->myid = lu->myid; lua->size = lu->size; 409397b6df1SKris Buschelman lu->id.job = JOB_INIT; 410397b6df1SKris Buschelman ierr = MPI_Comm_dup(A->comm,&(lu->comm_mumps));CHKERRQ(ierr); 411397b6df1SKris Buschelman lu->id.comm_fortran = lu->comm_mumps; 412397b6df1SKris Buschelman 413397b6df1SKris Buschelman /* Set mumps options */ 414397b6df1SKris Buschelman ierr = PetscOptionsBegin(A->comm,A->prefix,"MUMPS Options","Mat");CHKERRQ(ierr); 415397b6df1SKris Buschelman lu->id.par=1; /* host participates factorizaton and solve */ 416397b6df1SKris Buschelman lu->id.sym=lu->sym; 417397b6df1SKris Buschelman if (lu->sym == 2){ 418397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_sym","SYM: (1,2)","None",lu->id.sym,&icntl,&flg);CHKERRQ(ierr); 419397b6df1SKris Buschelman if (flg && icntl == 1) lu->id.sym=icntl; /* matrix is spd */ 420397b6df1SKris Buschelman } 421397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 422397b6df1SKris Buschelman zmumps_c(&lu->id); 423397b6df1SKris Buschelman #else 424397b6df1SKris Buschelman dmumps_c(&lu->id); 425397b6df1SKris Buschelman #endif 426397b6df1SKris Buschelman 427397b6df1SKris Buschelman if (lu->size == 1){ 428397b6df1SKris Buschelman lu->id.ICNTL(18) = 0; /* centralized assembled matrix input */ 429397b6df1SKris Buschelman } else { 430397b6df1SKris Buschelman lu->id.ICNTL(18) = 3; /* distributed assembled matrix input */ 431397b6df1SKris Buschelman } 432397b6df1SKris Buschelman 433397b6df1SKris Buschelman icntl=-1; 434397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_4","ICNTL(4): level of printing (0 to 4)","None",lu->id.ICNTL(4),&icntl,&flg);CHKERRQ(ierr); 435397b6df1SKris Buschelman if (flg && icntl > 0) { 436397b6df1SKris Buschelman lu->id.ICNTL(4)=icntl; /* and use mumps default icntl(i), i=1,2,3 */ 437397b6df1SKris Buschelman } else { /* no output */ 438397b6df1SKris Buschelman lu->id.ICNTL(1) = 0; /* error message, default= 6 */ 439397b6df1SKris Buschelman lu->id.ICNTL(2) = -1; /* output stream for diagnostic printing, statistics, and warning. default=0 */ 440397b6df1SKris Buschelman lu->id.ICNTL(3) = -1; /* output stream for global information, default=6 */ 441397b6df1SKris Buschelman lu->id.ICNTL(4) = 0; /* level of printing, 0,1,2,3,4, default=2 */ 442397b6df1SKris Buschelman } 443397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_6","ICNTL(6): matrix prescaling (0 to 7)","None",lu->id.ICNTL(6),&lu->id.ICNTL(6),PETSC_NULL);CHKERRQ(ierr); 444397b6df1SKris Buschelman icntl=-1; 445397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_7","ICNTL(7): matrix ordering (0 to 7)","None",lu->id.ICNTL(7),&icntl,&flg);CHKERRQ(ierr); 446397b6df1SKris Buschelman if (flg) { 447397b6df1SKris Buschelman if (icntl== 1){ 448397b6df1SKris Buschelman SETERRQ(PETSC_ERR_SUP,"pivot order be set by the user in PERM_IN -- not supported by the PETSc/MUMPS interface\n"); 449397b6df1SKris Buschelman } else { 450397b6df1SKris Buschelman lu->id.ICNTL(7) = icntl; 451397b6df1SKris Buschelman } 452397b6df1SKris Buschelman } 453397b6df1SKris 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); 454397b6df1SKris 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); 455397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_11","ICNTL(11): error analysis, a positive value returns statistics (by -sles_view)","None",lu->id.ICNTL(11),&lu->id.ICNTL(11),PETSC_NULL);CHKERRQ(ierr); 456397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_12","ICNTL(12): efficiency control","None",lu->id.ICNTL(12),&lu->id.ICNTL(12),PETSC_NULL);CHKERRQ(ierr); 457397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_13","ICNTL(13): efficiency control","None",lu->id.ICNTL(13),&lu->id.ICNTL(13),PETSC_NULL);CHKERRQ(ierr); 458adc1d99fSHong 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); 459397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_15","ICNTL(15): efficiency control","None",lu->id.ICNTL(15),&lu->id.ICNTL(15),PETSC_NULL);CHKERRQ(ierr); 460397b6df1SKris Buschelman 461397b6df1SKris Buschelman /* 462397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_16","ICNTL(16): 1: rank detection; 2: rank detection and nullspace","None",lu->id.ICNTL(16),&icntl,&flg);CHKERRQ(ierr); 463397b6df1SKris Buschelman if (flg){ 464397b6df1SKris Buschelman if (icntl >-1 && icntl <3 ){ 465397b6df1SKris Buschelman if (lu->myid==0) lu->id.ICNTL(16) = icntl; 466397b6df1SKris Buschelman } else { 467397b6df1SKris Buschelman SETERRQ1(PETSC_ERR_SUP,"ICNTL(16)=%d -- not supported\n",icntl); 468397b6df1SKris Buschelman } 469397b6df1SKris Buschelman } 470397b6df1SKris Buschelman */ 471397b6df1SKris Buschelman 472397b6df1SKris 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); 473397b6df1SKris 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); 474397b6df1SKris 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); 475397b6df1SKris Buschelman PetscOptionsEnd(); 476397b6df1SKris Buschelman } 477397b6df1SKris Buschelman 478397b6df1SKris Buschelman /* define matrix A */ 479397b6df1SKris Buschelman switch (lu->id.ICNTL(18)){ 480397b6df1SKris Buschelman case 0: /* centralized assembled matrix input (size=1) */ 481397b6df1SKris Buschelman if (!lu->myid) { 482c36ead0aSKris Buschelman if (lua->isAIJ){ 483397b6df1SKris Buschelman Mat_SeqAIJ *aa = (Mat_SeqAIJ*)A->data; 484397b6df1SKris Buschelman nz = aa->nz; 485397b6df1SKris Buschelman ai = aa->i; aj = aa->j; lu->val = aa->a; 486397b6df1SKris Buschelman } else { 487397b6df1SKris Buschelman Mat_SeqSBAIJ *aa = (Mat_SeqSBAIJ*)A->data; 488397b6df1SKris Buschelman nz = aa->s_nz; 489397b6df1SKris Buschelman ai = aa->i; aj = aa->j; lu->val = aa->a; 490397b6df1SKris Buschelman } 491397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ /* first numeric factorization, get irn and jcn */ 492397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(int),&lu->irn);CHKERRQ(ierr); 493397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(int),&lu->jcn);CHKERRQ(ierr); 494397b6df1SKris Buschelman nz = 0; 495397b6df1SKris Buschelman for (i=0; i<M; i++){ 496397b6df1SKris Buschelman rnz = ai[i+1] - ai[i]; 497397b6df1SKris Buschelman while (rnz--) { /* Fortran row/col index! */ 498397b6df1SKris Buschelman lu->irn[nz] = i+1; lu->jcn[nz] = (*aj)+1; aj++; nz++; 499397b6df1SKris Buschelman } 500397b6df1SKris Buschelman } 501397b6df1SKris Buschelman } 502397b6df1SKris Buschelman } 503397b6df1SKris Buschelman break; 504397b6df1SKris Buschelman case 3: /* distributed assembled matrix input (size>1) */ 505397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ 506397b6df1SKris Buschelman valOnly = PETSC_FALSE; 507397b6df1SKris Buschelman } else { 508397b6df1SKris Buschelman valOnly = PETSC_TRUE; /* only update mat values, not row and col index */ 509397b6df1SKris Buschelman } 510397b6df1SKris Buschelman ierr = MatConvertToTriples(A,1,valOnly, &nnz, &lu->irn, &lu->jcn, &lu->val);CHKERRQ(ierr); 511397b6df1SKris Buschelman break; 512397b6df1SKris Buschelman default: SETERRQ(PETSC_ERR_SUP,"Matrix input format is not supported by MUMPS."); 513397b6df1SKris Buschelman } 514397b6df1SKris Buschelman 515397b6df1SKris Buschelman /* analysis phase */ 516397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ 517397b6df1SKris Buschelman lu->id.n = M; 518397b6df1SKris Buschelman switch (lu->id.ICNTL(18)){ 519397b6df1SKris Buschelman case 0: /* centralized assembled matrix input */ 520397b6df1SKris Buschelman if (!lu->myid) { 521397b6df1SKris Buschelman lu->id.nz =nz; lu->id.irn=lu->irn; lu->id.jcn=lu->jcn; 522397b6df1SKris Buschelman if (lu->id.ICNTL(6)>1){ 523397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 524397b6df1SKris Buschelman lu->id.a = (mumps_double_complex*)lu->val; 525397b6df1SKris Buschelman #else 526397b6df1SKris Buschelman lu->id.a = lu->val; 527397b6df1SKris Buschelman #endif 528397b6df1SKris Buschelman } 529397b6df1SKris Buschelman } 530397b6df1SKris Buschelman break; 531397b6df1SKris Buschelman case 3: /* distributed assembled matrix input (size>1) */ 532397b6df1SKris Buschelman lu->id.nz_loc = nnz; 533397b6df1SKris Buschelman lu->id.irn_loc=lu->irn; lu->id.jcn_loc=lu->jcn; 534397b6df1SKris Buschelman if (lu->id.ICNTL(6)>1) { 535397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 536397b6df1SKris Buschelman lu->id.a_loc = (mumps_double_complex*)lu->val; 537397b6df1SKris Buschelman #else 538397b6df1SKris Buschelman lu->id.a_loc = lu->val; 539397b6df1SKris Buschelman #endif 540397b6df1SKris Buschelman } 541397b6df1SKris Buschelman break; 542397b6df1SKris Buschelman } 543397b6df1SKris Buschelman lu->id.job=1; 544397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 545397b6df1SKris Buschelman zmumps_c(&lu->id); 546397b6df1SKris Buschelman #else 547397b6df1SKris Buschelman dmumps_c(&lu->id); 548397b6df1SKris Buschelman #endif 549397b6df1SKris Buschelman if (lu->id.INFOG(1) < 0) { 550397b6df1SKris Buschelman SETERRQ1(1,"Error reported by MUMPS in analysis phase: INFOG(1)=%d\n",lu->id.INFOG(1)); 551397b6df1SKris Buschelman } 552397b6df1SKris Buschelman } 553397b6df1SKris Buschelman 554397b6df1SKris Buschelman /* numerical factorization phase */ 555397b6df1SKris Buschelman if(lu->id.ICNTL(18) == 0) { 556397b6df1SKris Buschelman if (lu->myid == 0) { 557397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 558397b6df1SKris Buschelman lu->id.a = (mumps_double_complex*)lu->val; 559397b6df1SKris Buschelman #else 560397b6df1SKris Buschelman lu->id.a = lu->val; 561397b6df1SKris Buschelman #endif 562397b6df1SKris Buschelman } 563397b6df1SKris Buschelman } else { 564397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 565397b6df1SKris Buschelman lu->id.a_loc = (mumps_double_complex*)lu->val; 566397b6df1SKris Buschelman #else 567397b6df1SKris Buschelman lu->id.a_loc = lu->val; 568397b6df1SKris Buschelman #endif 569397b6df1SKris Buschelman } 570397b6df1SKris Buschelman lu->id.job=2; 571397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 572397b6df1SKris Buschelman zmumps_c(&lu->id); 573397b6df1SKris Buschelman #else 574397b6df1SKris Buschelman dmumps_c(&lu->id); 575397b6df1SKris Buschelman #endif 576397b6df1SKris Buschelman if (lu->id.INFOG(1) < 0) { 577397b6df1SKris Buschelman SETERRQ1(1,"1, Error reported by MUMPS in numerical factorization phase: INFOG(1)=%d\n",lu->id.INFOG(1)); 578397b6df1SKris Buschelman } 579397b6df1SKris Buschelman 580397b6df1SKris Buschelman if (lu->myid==0 && lu->id.ICNTL(16) > 0){ 581397b6df1SKris Buschelman SETERRQ1(1," lu->id.ICNTL(16):=%d\n",lu->id.INFOG(16)); 582397b6df1SKris Buschelman } 583397b6df1SKris Buschelman 584397b6df1SKris Buschelman (*F)->assembled = PETSC_TRUE; 585397b6df1SKris Buschelman lu->matstruc = SAME_NONZERO_PATTERN; 586ace87b0dSHong Zhang lu->CleanUpMUMPS = PETSC_TRUE; 587397b6df1SKris Buschelman PetscFunctionReturn(0); 588397b6df1SKris Buschelman } 589397b6df1SKris Buschelman 590397b6df1SKris Buschelman /* Note the Petsc r and c permutations are ignored */ 591397b6df1SKris Buschelman #undef __FUNCT__ 592f0c56d0fSKris Buschelman #define __FUNCT__ "MatLUFactorSymbolic_AIJMUMPS" 593f0c56d0fSKris Buschelman int MatLUFactorSymbolic_AIJMUMPS(Mat A,IS r,IS c,MatFactorInfo *info,Mat *F) { 594397b6df1SKris Buschelman Mat B; 595f0c56d0fSKris Buschelman Mat_MUMPS *lu; 596397b6df1SKris Buschelman int ierr; 597397b6df1SKris Buschelman 598397b6df1SKris Buschelman PetscFunctionBegin; 599397b6df1SKris Buschelman 600397b6df1SKris Buschelman /* Create the factorization matrix */ 601397b6df1SKris Buschelman ierr = MatCreate(A->comm,A->m,A->n,A->M,A->N,&B);CHKERRQ(ierr); 602397b6df1SKris Buschelman ierr = MatSetType(B,MATAIJMUMPS);CHKERRQ(ierr); 603397b6df1SKris Buschelman ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr); 604397b6df1SKris Buschelman ierr = MatMPIAIJSetPreallocation(B,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr); 605397b6df1SKris Buschelman 606f0c56d0fSKris Buschelman B->ops->lufactornumeric = MatFactorNumeric_AIJMUMPS; 607397b6df1SKris Buschelman B->factor = FACTOR_LU; 608f0c56d0fSKris Buschelman lu = (Mat_MUMPS*)B->spptr; 609397b6df1SKris Buschelman lu->sym = 0; 610397b6df1SKris Buschelman lu->matstruc = DIFFERENT_NONZERO_PATTERN; 611397b6df1SKris Buschelman 612397b6df1SKris Buschelman *F = B; 613397b6df1SKris Buschelman PetscFunctionReturn(0); 614397b6df1SKris Buschelman } 615397b6df1SKris Buschelman 616397b6df1SKris Buschelman /* Note the Petsc r permutation is ignored */ 617397b6df1SKris Buschelman #undef __FUNCT__ 618f0c56d0fSKris Buschelman #define __FUNCT__ "MatCholeskyFactorSymbolic_SBAIJMUMPS" 619f0c56d0fSKris Buschelman int MatCholeskyFactorSymbolic_SBAIJMUMPS(Mat A,IS r,MatFactorInfo *info,Mat *F) { 620397b6df1SKris Buschelman Mat B; 621f0c56d0fSKris Buschelman Mat_MUMPS *lu; 622397b6df1SKris Buschelman int ierr; 623397b6df1SKris Buschelman 624397b6df1SKris Buschelman PetscFunctionBegin; 625397b6df1SKris Buschelman 626397b6df1SKris Buschelman /* Create the factorization matrix */ 627397b6df1SKris Buschelman ierr = MatCreate(A->comm,A->m,A->n,A->M,A->N,&B);CHKERRQ(ierr); 628397b6df1SKris Buschelman ierr = MatSetType(B,MATAIJMUMPS);CHKERRQ(ierr); 629397b6df1SKris Buschelman ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr); 630397b6df1SKris Buschelman ierr = MatMPIAIJSetPreallocation(B,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr); 631397b6df1SKris Buschelman 632f0c56d0fSKris Buschelman B->ops->choleskyfactornumeric = MatFactorNumeric_AIJMUMPS; 633397b6df1SKris Buschelman B->factor = FACTOR_CHOLESKY; 634f0c56d0fSKris Buschelman lu = (Mat_MUMPS*)B->spptr; 635397b6df1SKris Buschelman lu->sym = 2; 636397b6df1SKris Buschelman lu->matstruc = DIFFERENT_NONZERO_PATTERN; 637397b6df1SKris Buschelman 638397b6df1SKris Buschelman *F = B; 639397b6df1SKris Buschelman PetscFunctionReturn(0); 640397b6df1SKris Buschelman } 641397b6df1SKris Buschelman 642397b6df1SKris Buschelman #undef __FUNCT__ 643f0c56d0fSKris Buschelman #define __FUNCT__ "MatAssemblyEnd_AIJMUMPS" 644f0c56d0fSKris Buschelman int MatAssemblyEnd_AIJMUMPS(Mat A,MatAssemblyType mode) { 645c338a77dSKris Buschelman int ierr; 646f0c56d0fSKris Buschelman Mat_MUMPS *mumps=(Mat_MUMPS*)A->spptr; 647c338a77dSKris Buschelman 648397b6df1SKris Buschelman PetscFunctionBegin; 649c338a77dSKris Buschelman ierr = (*mumps->MatAssemblyEnd)(A,mode);CHKERRQ(ierr); 650f0c56d0fSKris Buschelman 651c338a77dSKris Buschelman mumps->MatLUFactorSymbolic = A->ops->lufactorsymbolic; 652c338a77dSKris Buschelman mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic; 653f0c56d0fSKris Buschelman A->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS; 654397b6df1SKris Buschelman PetscFunctionReturn(0); 655397b6df1SKris Buschelman } 656397b6df1SKris Buschelman 657c338a77dSKris Buschelman EXTERN_C_BEGIN 658c338a77dSKris Buschelman #undef __FUNCT__ 659f0c56d0fSKris Buschelman #define __FUNCT__ "MatConvert_AIJ_AIJMUMPS" 660f0c56d0fSKris Buschelman int MatConvert_AIJ_AIJMUMPS(Mat A,MatType newtype,Mat *newmat) { 661c338a77dSKris Buschelman int ierr,size; 662c338a77dSKris Buschelman MPI_Comm comm; 663c338a77dSKris Buschelman Mat B=*newmat; 664f0c56d0fSKris Buschelman Mat_MUMPS *mumps; 665397b6df1SKris Buschelman 666397b6df1SKris Buschelman PetscFunctionBegin; 667c338a77dSKris Buschelman if (B != A) { 668c338a77dSKris Buschelman ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr); 669397b6df1SKris Buschelman } 670397b6df1SKris Buschelman 671c338a77dSKris Buschelman ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 672f0c56d0fSKris Buschelman ierr = PetscNew(Mat_MUMPS,&mumps);CHKERRQ(ierr); 673c338a77dSKris Buschelman 674f0c56d0fSKris Buschelman mumps->MatDuplicate = A->ops->duplicate; 675c338a77dSKris Buschelman mumps->MatView = A->ops->view; 676c338a77dSKris Buschelman mumps->MatAssemblyEnd = A->ops->assemblyend; 677c338a77dSKris Buschelman mumps->MatLUFactorSymbolic = A->ops->lufactorsymbolic; 678c338a77dSKris Buschelman mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic; 679c338a77dSKris Buschelman mumps->MatDestroy = A->ops->destroy; 680a39386dcSKris Buschelman mumps->specialdestroy = MatDestroy_AIJMUMPS; 681c338a77dSKris Buschelman mumps->CleanUpMUMPS = PETSC_FALSE; 682f579278aSKris Buschelman mumps->isAIJ = PETSC_TRUE; 683c338a77dSKris Buschelman 6844b68dd72SKris Buschelman B->spptr = (void *)mumps; 685f0c56d0fSKris Buschelman B->ops->duplicate = MatDuplicate_AIJMUMPS; 686f0c56d0fSKris Buschelman B->ops->view = MatView_AIJMUMPS; 687f0c56d0fSKris Buschelman B->ops->assemblyend = MatAssemblyEnd_AIJMUMPS; 688f0c56d0fSKris Buschelman B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS; 6893924e44cSKris Buschelman B->ops->destroy = MatDestroy_MUMPS; 690c338a77dSKris Buschelman 691c338a77dSKris Buschelman ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr); 692c338a77dSKris Buschelman if (size == 1) { 693c338a77dSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqaij_aijmumps_C", 694f0c56d0fSKris Buschelman "MatConvert_AIJ_AIJMUMPS",MatConvert_AIJ_AIJMUMPS);CHKERRQ(ierr); 695c338a77dSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_aijmumps_seqaij_C", 696c338a77dSKris Buschelman "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr); 697c338a77dSKris Buschelman } else { 698c338a77dSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_mpiaij_aijmumps_C", 699f0c56d0fSKris Buschelman "MatConvert_AIJ_AIJMUMPS",MatConvert_AIJ_AIJMUMPS);CHKERRQ(ierr); 700c338a77dSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_aijmumps_mpiaij_C", 701c338a77dSKris Buschelman "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr); 702c338a77dSKris Buschelman } 703c338a77dSKris Buschelman 704f579278aSKris Buschelman PetscLogInfo(0,"Using MUMPS for LU factorization and solves."); 705c338a77dSKris Buschelman ierr = PetscObjectChangeTypeName((PetscObject)B,newtype);CHKERRQ(ierr); 706c338a77dSKris Buschelman *newmat = B; 707397b6df1SKris Buschelman PetscFunctionReturn(0); 708397b6df1SKris Buschelman } 709c338a77dSKris Buschelman EXTERN_C_END 710397b6df1SKris Buschelman 711f0c56d0fSKris Buschelman #undef __FUNCT__ 712f0c56d0fSKris Buschelman #define __FUNCT__ "MatDuplicate_AIJMUMPS" 713f0c56d0fSKris Buschelman int MatDuplicate_AIJMUMPS(Mat A, MatDuplicateOption op, Mat *M) { 714f0c56d0fSKris Buschelman int ierr; 7158f340917SKris Buschelman Mat_MUMPS *lu=(Mat_MUMPS *)A->spptr; 7168f340917SKris Buschelman 717f0c56d0fSKris Buschelman PetscFunctionBegin; 7188f340917SKris Buschelman ierr = (*lu->MatDuplicate)(A,op,M);CHKERRQ(ierr); 719f0c56d0fSKris Buschelman ierr = MatConvert_AIJ_AIJMUMPS(*M,MATAIJMUMPS,M);CHKERRQ(ierr); 720a39386dcSKris Buschelman ierr = PetscMemcpy((*M)->spptr,lu,sizeof(Mat_MUMPS));CHKERRQ(ierr); 721f0c56d0fSKris Buschelman PetscFunctionReturn(0); 722f0c56d0fSKris Buschelman } 723f0c56d0fSKris Buschelman 72424b6179bSKris Buschelman /*MC 725fafad747SKris Buschelman MATAIJMUMPS - MATAIJMUMPS = "aijmumps" - A matrix type providing direct solvers (LU) for distributed 72624b6179bSKris Buschelman and sequential matrices via the external package MUMPS. 72724b6179bSKris Buschelman 72824b6179bSKris Buschelman If MUMPS is installed (see the manual for instructions 72924b6179bSKris Buschelman on how to declare the existence of external packages), 73024b6179bSKris Buschelman a matrix type can be constructed which invokes MUMPS solvers. 73124b6179bSKris Buschelman After calling MatCreate(...,A), simply call MatSetType(A,MATAIJMUMPS). 73224b6179bSKris Buschelman This matrix type is only supported for double precision real. 73324b6179bSKris Buschelman 73424b6179bSKris Buschelman If created with a single process communicator, this matrix type inherits from MATSEQAIJ. 73524b6179bSKris Buschelman Otherwise, this matrix type inherits from MATMPIAIJ. Hence for single process communicators, 73624b6179bSKris Buschelman MatSeqAIJSetPreallocation is supported, and similarly MatMPIAIJSetPreallocation is supported 73724b6179bSKris Buschelman for communicators controlling multiple processes. It is recommended that you call both of 73828b08bd3SKris Buschelman the above preallocation routines for simplicity. One can also call MatConvert for an inplace 73928b08bd3SKris Buschelman conversion to or from the MATSEQAIJ or MATMPIAIJ type (depending on the communicator size) 74028b08bd3SKris Buschelman without data copy. 74124b6179bSKris Buschelman 74224b6179bSKris Buschelman Options Database Keys: 7430bad9183SKris Buschelman + -mat_type aijmumps - sets the matrix type to "aijmumps" during a call to MatSetFromOptions() 74424b6179bSKris Buschelman . -mat_mumps_sym <0,1,2> - 0 the matrix is unsymmetric, 1 symmetric positive definite, 2 symmetric 74524b6179bSKris Buschelman . -mat_mumps_icntl_4 <0,1,2,3,4> - print level 74624b6179bSKris Buschelman . -mat_mumps_icntl_6 <0,...,7> - matrix prescaling options (see MUMPS User's Guide) 74724b6179bSKris Buschelman . -mat_mumps_icntl_7 <0,...,7> - matrix orderings (see MUMPS User's Guide) 74824b6179bSKris Buschelman . -mat_mumps_icntl_9 <1,2> - A or A^T x=b to be solved: 1 denotes A, 2 denotes A^T 74924b6179bSKris Buschelman . -mat_mumps_icntl_10 <n> - maximum number of iterative refinements 75024b6179bSKris Buschelman . -mat_mumps_icntl_11 <n> - error analysis, a positive value returns statistics during -sles_view 75124b6179bSKris Buschelman . -mat_mumps_icntl_12 <n> - efficiency control (see MUMPS User's Guide) 75224b6179bSKris Buschelman . -mat_mumps_icntl_13 <n> - efficiency control (see MUMPS User's Guide) 75324b6179bSKris Buschelman . -mat_mumps_icntl_14 <n> - efficiency control (see MUMPS User's Guide) 75424b6179bSKris Buschelman . -mat_mumps_icntl_15 <n> - efficiency control (see MUMPS User's Guide) 75524b6179bSKris Buschelman . -mat_mumps_cntl_1 <delta> - relative pivoting threshold 75624b6179bSKris Buschelman . -mat_mumps_cntl_2 <tol> - stopping criterion for refinement 75724b6179bSKris Buschelman - -mat_mumps_cntl_3 <adelta> - absolute pivoting threshold 75824b6179bSKris Buschelman 75924b6179bSKris Buschelman Level: beginner 76024b6179bSKris Buschelman 76124b6179bSKris Buschelman .seealso: MATSBAIJMUMPS 76224b6179bSKris Buschelman M*/ 76324b6179bSKris Buschelman 764397b6df1SKris Buschelman EXTERN_C_BEGIN 765397b6df1SKris Buschelman #undef __FUNCT__ 766f0c56d0fSKris Buschelman #define __FUNCT__ "MatCreate_AIJMUMPS" 767f0c56d0fSKris Buschelman int MatCreate_AIJMUMPS(Mat A) { 768397b6df1SKris Buschelman int ierr,size; 769397b6df1SKris Buschelman MPI_Comm comm; 770397b6df1SKris Buschelman 771397b6df1SKris Buschelman PetscFunctionBegin; 7725441df8eSKris Buschelman /* Change type name before calling MatSetType to force proper construction of SeqAIJ or MPIAIJ */ 7735441df8eSKris Buschelman /* and AIJMUMPS types */ 7745441df8eSKris Buschelman ierr = PetscObjectChangeTypeName((PetscObject)A,MATAIJMUMPS);CHKERRQ(ierr); 775397b6df1SKris Buschelman ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 776397b6df1SKris Buschelman ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr); 777397b6df1SKris Buschelman if (size == 1) { 778397b6df1SKris Buschelman ierr = MatSetType(A,MATSEQAIJ);CHKERRQ(ierr); 779397b6df1SKris Buschelman } else { 780397b6df1SKris Buschelman ierr = MatSetType(A,MATMPIAIJ);CHKERRQ(ierr); 781397b6df1SKris Buschelman } 782f0c56d0fSKris Buschelman ierr = MatConvert_AIJ_AIJMUMPS(A,MATAIJMUMPS,&A);CHKERRQ(ierr); 783397b6df1SKris Buschelman PetscFunctionReturn(0); 784397b6df1SKris Buschelman } 785397b6df1SKris Buschelman EXTERN_C_END 786397b6df1SKris Buschelman 787f579278aSKris Buschelman #undef __FUNCT__ 788f0c56d0fSKris Buschelman #define __FUNCT__ "MatAssemblyEnd_SBAIJMUMPS" 789f0c56d0fSKris Buschelman int MatAssemblyEnd_SBAIJMUMPS(Mat A,MatAssemblyType mode) { 790f579278aSKris Buschelman int ierr; 791f0c56d0fSKris Buschelman Mat_MUMPS *mumps=(Mat_MUMPS*)A->spptr; 792f579278aSKris Buschelman 793f579278aSKris Buschelman PetscFunctionBegin; 794f579278aSKris Buschelman ierr = (*mumps->MatAssemblyEnd)(A,mode);CHKERRQ(ierr); 795f579278aSKris Buschelman mumps->MatLUFactorSymbolic = A->ops->lufactorsymbolic; 796f579278aSKris Buschelman mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic; 797f0c56d0fSKris Buschelman A->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SBAIJMUMPS; 798f579278aSKris Buschelman PetscFunctionReturn(0); 799f579278aSKris Buschelman } 800f579278aSKris Buschelman 801f579278aSKris Buschelman EXTERN_C_BEGIN 802f579278aSKris Buschelman #undef __FUNCT__ 803f0c56d0fSKris Buschelman #define __FUNCT__ "MatConvert_SBAIJ_SBAIJMUMPS" 804f0c56d0fSKris Buschelman int MatConvert_SBAIJ_SBAIJMUMPS(Mat A,MatType newtype,Mat *newmat) { 805f579278aSKris Buschelman int ierr,size; 806f579278aSKris Buschelman MPI_Comm comm; 807f579278aSKris Buschelman Mat B=*newmat; 808f0c56d0fSKris Buschelman Mat_MUMPS *mumps; 809f579278aSKris Buschelman 810f579278aSKris Buschelman PetscFunctionBegin; 811f579278aSKris Buschelman if (B != A) { 812f579278aSKris Buschelman ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr); 813f579278aSKris Buschelman } 814f579278aSKris Buschelman 815f579278aSKris Buschelman ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 816f0c56d0fSKris Buschelman ierr = PetscNew(Mat_MUMPS,&mumps);CHKERRQ(ierr); 817f579278aSKris Buschelman 818f0c56d0fSKris Buschelman mumps->MatDuplicate = A->ops->duplicate; 819f579278aSKris Buschelman mumps->MatView = A->ops->view; 820f579278aSKris Buschelman mumps->MatAssemblyEnd = A->ops->assemblyend; 821f579278aSKris Buschelman mumps->MatLUFactorSymbolic = A->ops->lufactorsymbolic; 822f579278aSKris Buschelman mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic; 823f579278aSKris Buschelman mumps->MatDestroy = A->ops->destroy; 824a39386dcSKris Buschelman mumps->specialdestroy = MatDestroy_SBAIJMUMPS; 825f579278aSKris Buschelman mumps->CleanUpMUMPS = PETSC_FALSE; 826f579278aSKris Buschelman mumps->isAIJ = PETSC_FALSE; 827f579278aSKris Buschelman 828f579278aSKris Buschelman B->spptr = (void *)mumps; 829f0c56d0fSKris Buschelman B->ops->duplicate = MatDuplicate_SBAIJMUMPS; 830f0c56d0fSKris Buschelman B->ops->view = MatView_AIJMUMPS; 831f0c56d0fSKris Buschelman B->ops->assemblyend = MatAssemblyEnd_SBAIJMUMPS; 832f0c56d0fSKris Buschelman B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SBAIJMUMPS; 8333924e44cSKris Buschelman B->ops->destroy = MatDestroy_MUMPS; 834f579278aSKris Buschelman 835f579278aSKris Buschelman ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr); 836f579278aSKris Buschelman if (size == 1) { 837f0c56d0fSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_sbaijmumps_C", 838f0c56d0fSKris Buschelman "MatConvert_SBAIJ_SBAIJMUMPS",MatConvert_SBAIJ_SBAIJMUMPS);CHKERRQ(ierr); 839f0c56d0fSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_sbaijmumps_seqsbaij_C", 840f579278aSKris Buschelman "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr); 841f579278aSKris Buschelman } else { 842f0c56d0fSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_mpisbaij_sbaijmumps_C", 843f0c56d0fSKris Buschelman "MatConvert_SBAIJ_SBAIJMUMPS",MatConvert_SBAIJ_SBAIJMUMPS);CHKERRQ(ierr); 844f0c56d0fSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_sbaijmumps_mpisbaij_C", 845f579278aSKris Buschelman "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr); 846f579278aSKris Buschelman } 847f579278aSKris Buschelman 848f579278aSKris Buschelman PetscLogInfo(0,"Using MUMPS for Cholesky factorization and solves."); 849f579278aSKris Buschelman ierr = PetscObjectChangeTypeName((PetscObject)B,newtype);CHKERRQ(ierr); 850f579278aSKris Buschelman *newmat = B; 851f579278aSKris Buschelman PetscFunctionReturn(0); 852f579278aSKris Buschelman } 853f579278aSKris Buschelman EXTERN_C_END 854f579278aSKris Buschelman 855f0c56d0fSKris Buschelman #undef __FUNCT__ 856f0c56d0fSKris Buschelman #define __FUNCT__ "MatDuplicate_SBAIJMUMPS" 857f0c56d0fSKris Buschelman int MatDuplicate_SBAIJMUMPS(Mat A, MatDuplicateOption op, Mat *M) { 858f0c56d0fSKris Buschelman int ierr; 8598f340917SKris Buschelman Mat_MUMPS *lu=(Mat_MUMPS *)A->spptr; 8608f340917SKris Buschelman 861f0c56d0fSKris Buschelman PetscFunctionBegin; 8628f340917SKris Buschelman ierr = (*lu->MatDuplicate)(A,op,M);CHKERRQ(ierr); 863f0c56d0fSKris Buschelman ierr = MatConvert_SBAIJ_SBAIJMUMPS(*M,MATSBAIJMUMPS,M);CHKERRQ(ierr); 8643f953163SKris Buschelman ierr = PetscMemcpy((*M)->spptr,lu,sizeof(Mat_MUMPS));CHKERRQ(ierr); 865f0c56d0fSKris Buschelman PetscFunctionReturn(0); 866f0c56d0fSKris Buschelman } 867f0c56d0fSKris Buschelman 86824b6179bSKris Buschelman /*MC 869fafad747SKris Buschelman MATSBAIJMUMPS - MATSBAIJMUMPS = "sbaijmumps" - A symmetric matrix type providing direct solvers (Cholesky) for 87024b6179bSKris Buschelman distributed and sequential matrices via the external package MUMPS. 87124b6179bSKris Buschelman 87224b6179bSKris Buschelman If MUMPS is installed (see the manual for instructions 87324b6179bSKris Buschelman on how to declare the existence of external packages), 87424b6179bSKris Buschelman a matrix type can be constructed which invokes MUMPS solvers. 87524b6179bSKris Buschelman After calling MatCreate(...,A), simply call MatSetType(A,MATSBAIJMUMPS). 87624b6179bSKris Buschelman This matrix type is only supported for double precision real. 87724b6179bSKris Buschelman 87824b6179bSKris Buschelman If created with a single process communicator, this matrix type inherits from MATSEQSBAIJ. 87924b6179bSKris Buschelman Otherwise, this matrix type inherits from MATMPISBAIJ. Hence for single process communicators, 88024b6179bSKris Buschelman MatSeqSBAIJSetPreallocation is supported, and similarly MatMPISBAIJSetPreallocation is supported 88124b6179bSKris Buschelman for communicators controlling multiple processes. It is recommended that you call both of 88228b08bd3SKris Buschelman the above preallocation routines for simplicity. One can also call MatConvert for an inplace 88328b08bd3SKris Buschelman conversion to or from the MATSEQSBAIJ or MATMPISBAIJ type (depending on the communicator size) 88428b08bd3SKris Buschelman without data copy. 88524b6179bSKris Buschelman 88624b6179bSKris Buschelman Options Database Keys: 8870bad9183SKris Buschelman + -mat_type sbaijmumps - sets the matrix type to "sbaijmumps" during a call to MatSetFromOptions() 88824b6179bSKris Buschelman . -mat_mumps_sym <0,1,2> - 0 the matrix is unsymmetric, 1 symmetric positive definite, 2 symmetric 88924b6179bSKris Buschelman . -mat_mumps_icntl_4 <0,...,4> - print level 89024b6179bSKris Buschelman . -mat_mumps_icntl_6 <0,...,7> - matrix prescaling options (see MUMPS User's Guide) 89124b6179bSKris Buschelman . -mat_mumps_icntl_7 <0,...,7> - matrix orderings (see MUMPS User's Guide) 89224b6179bSKris Buschelman . -mat_mumps_icntl_9 <1,2> - A or A^T x=b to be solved: 1 denotes A, 2 denotes A^T 89324b6179bSKris Buschelman . -mat_mumps_icntl_10 <n> - maximum number of iterative refinements 89424b6179bSKris Buschelman . -mat_mumps_icntl_11 <n> - error analysis, a positive value returns statistics during -sles_view 89524b6179bSKris Buschelman . -mat_mumps_icntl_12 <n> - efficiency control (see MUMPS User's Guide) 89624b6179bSKris Buschelman . -mat_mumps_icntl_13 <n> - efficiency control (see MUMPS User's Guide) 89724b6179bSKris Buschelman . -mat_mumps_icntl_14 <n> - efficiency control (see MUMPS User's Guide) 89824b6179bSKris Buschelman . -mat_mumps_icntl_15 <n> - efficiency control (see MUMPS User's Guide) 89924b6179bSKris Buschelman . -mat_mumps_cntl_1 <delta> - relative pivoting threshold 90024b6179bSKris Buschelman . -mat_mumps_cntl_2 <tol> - stopping criterion for refinement 90124b6179bSKris Buschelman - -mat_mumps_cntl_3 <adelta> - absolute pivoting threshold 90224b6179bSKris Buschelman 90324b6179bSKris Buschelman Level: beginner 90424b6179bSKris Buschelman 90524b6179bSKris Buschelman .seealso: MATAIJMUMPS 90624b6179bSKris Buschelman M*/ 90724b6179bSKris Buschelman 908397b6df1SKris Buschelman EXTERN_C_BEGIN 909397b6df1SKris Buschelman #undef __FUNCT__ 910f0c56d0fSKris Buschelman #define __FUNCT__ "MatCreate_SBAIJMUMPS" 911f0c56d0fSKris Buschelman int MatCreate_SBAIJMUMPS(Mat A) { 912397b6df1SKris Buschelman int ierr,size; 913397b6df1SKris Buschelman 914397b6df1SKris Buschelman PetscFunctionBegin; 9155441df8eSKris Buschelman /* Change type name before calling MatSetType to force proper construction of SeqSBAIJ or MPISBAIJ */ 9165441df8eSKris Buschelman /* and SBAIJMUMPS types */ 9175441df8eSKris Buschelman ierr = PetscObjectChangeTypeName((PetscObject)A,MATSBAIJMUMPS);CHKERRQ(ierr); 9185441df8eSKris Buschelman ierr = MPI_Comm_size(A->comm,&size);CHKERRQ(ierr);CHKERRQ(ierr); 919397b6df1SKris Buschelman if (size == 1) { 920397b6df1SKris Buschelman ierr = MatSetType(A,MATSEQSBAIJ);CHKERRQ(ierr); 921397b6df1SKris Buschelman } else { 922397b6df1SKris Buschelman ierr = MatSetType(A,MATMPISBAIJ);CHKERRQ(ierr); 923397b6df1SKris Buschelman } 924f0c56d0fSKris Buschelman ierr = MatConvert_SBAIJ_SBAIJMUMPS(A,MATSBAIJMUMPS,&A);CHKERRQ(ierr); 925397b6df1SKris Buschelman PetscFunctionReturn(0); 926397b6df1SKris Buschelman } 927397b6df1SKris Buschelman EXTERN_C_END 928