1397b6df1SKris Buschelman /*$Id: mumps.c,v 1.10 2001/08/15 15:56:50 bsmith Exp $*/ 2397b6df1SKris Buschelman /* 3397b6df1SKris Buschelman Provides an interface to the MUMPS_4.2_beta sparse solver 4397b6df1SKris Buschelman */ 5397b6df1SKris Buschelman 6397b6df1SKris Buschelman #include "src/mat/impls/aij/seq/aij.h" 7397b6df1SKris Buschelman #include "src/mat/impls/aij/mpi/mpiaij.h" 8397b6df1SKris Buschelman #include "src/mat/impls/sbaij/seq/sbaij.h" 9397b6df1SKris Buschelman #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] 24397b6df1SKris Buschelman #define RINFOG(I) rinfog[(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 PetscTruth isAIJ,CleanUpMUMPS; 37397b6df1SKris Buschelman int (*MatDestroy)(Mat); 38397b6df1SKris Buschelman int (*MatAssemblyEnd)(Mat,MatAssemblyType); 39397b6df1SKris Buschelman int (*MatView)(Mat,PetscViewer); 40397b6df1SKris Buschelman } Mat_AIJ_MUMPS; 41397b6df1SKris Buschelman 420e3434eeSKris Buschelman EXTERN int MatFactorInfo_MUMPS(Mat,PetscViewer); 430e3434eeSKris Buschelman 44397b6df1SKris Buschelman /* convert Petsc mpiaij matrix to triples: row[nz], col[nz], val[nz] */ 45397b6df1SKris Buschelman /* 46397b6df1SKris Buschelman input: 47397b6df1SKris Buschelman A - matrix in mpiaij format 48397b6df1SKris Buschelman shift - 0: C style output triple; 1: Fortran style output triple. 49397b6df1SKris Buschelman valOnly - FALSE: spaces are allocated and values are set for the triple 50397b6df1SKris Buschelman TRUE: only the values in v array are updated 51397b6df1SKris Buschelman output: 52397b6df1SKris Buschelman nnz - dim of r, c, and v (number of local nonzero entries of A) 53397b6df1SKris Buschelman r, c, v - row and col index, matrix values (matrix triples) 54397b6df1SKris Buschelman */ 55397b6df1SKris Buschelman int MatConvertToTriples(Mat A,int shift,PetscTruth valOnly,int *nnz,int **r, int **c, PetscScalar **v) 56397b6df1SKris Buschelman { 57397b6df1SKris Buschelman int *ai, *aj, *bi, *bj, rstart,nz, *garray; 58397b6df1SKris Buschelman int ierr,i,j,jj,jB,irow,m=A->m,*ajj,*bjj,countA,countB,colA_start,jcol; 59*d54de34fSKris Buschelman int *row,*col; 60397b6df1SKris Buschelman PetscScalar *av, *bv,*val; 61397b6df1SKris Buschelman Mat_AIJ_MUMPS *mumps = (Mat_AIJ_MUMPS *)A->spptr; 62397b6df1SKris Buschelman 63397b6df1SKris Buschelman PetscFunctionBegin; 64397b6df1SKris Buschelman 65397b6df1SKris Buschelman if (mumps->isAIJ){ 66397b6df1SKris Buschelman Mat_MPIAIJ *mat = (Mat_MPIAIJ*)A->data; 67397b6df1SKris Buschelman Mat_SeqAIJ *aa=(Mat_SeqAIJ*)(mat->A)->data; 68397b6df1SKris Buschelman Mat_SeqAIJ *bb=(Mat_SeqAIJ*)(mat->B)->data; 69397b6df1SKris Buschelman nz = aa->nz + bb->nz; 70397b6df1SKris Buschelman ai=aa->i; aj=aa->j; bi=bb->i; bj=bb->j; rstart= mat->rstart; 71397b6df1SKris Buschelman garray = mat->garray; 72397b6df1SKris Buschelman av=aa->a; bv=bb->a; 73397b6df1SKris Buschelman 74397b6df1SKris Buschelman } else { 75397b6df1SKris Buschelman Mat_MPISBAIJ *mat = (Mat_MPISBAIJ*)A->data; 76397b6df1SKris Buschelman if (mat->bs > 1) SETERRQ1(PETSC_ERR_SUP," bs=%d is not supported yet\n", mat->bs); 77397b6df1SKris Buschelman Mat_SeqSBAIJ *aa=(Mat_SeqSBAIJ*)(mat->A)->data; 78397b6df1SKris Buschelman Mat_SeqBAIJ *bb=(Mat_SeqBAIJ*)(mat->B)->data; 79397b6df1SKris Buschelman nz = aa->s_nz + bb->nz; 80397b6df1SKris Buschelman ai=aa->i; aj=aa->j; bi=bb->i; bj=bb->j; rstart= mat->rstart; 81397b6df1SKris Buschelman garray = mat->garray; 82397b6df1SKris Buschelman av=aa->a; bv=bb->a; 83397b6df1SKris Buschelman } 84397b6df1SKris Buschelman 85397b6df1SKris Buschelman if (!valOnly){ 86397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(int),&row);CHKERRQ(ierr); 87397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(int),&col);CHKERRQ(ierr); 88397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(PetscScalar),&val);CHKERRQ(ierr); 89397b6df1SKris Buschelman *r = row; *c = col; *v = val; 90397b6df1SKris Buschelman } else { 91397b6df1SKris Buschelman row = *r; col = *c; val = *v; 92397b6df1SKris Buschelman } 93397b6df1SKris Buschelman *nnz = nz; 94397b6df1SKris Buschelman 95397b6df1SKris Buschelman jj = 0; jB = 0; irow = rstart; 96397b6df1SKris Buschelman for ( i=0; i<m; i++ ) { 97397b6df1SKris Buschelman ajj = aj + ai[i]; /* ptr to the beginning of this row */ 98397b6df1SKris Buschelman countA = ai[i+1] - ai[i]; 99397b6df1SKris Buschelman countB = bi[i+1] - bi[i]; 100397b6df1SKris Buschelman bjj = bj + bi[i]; 101397b6df1SKris Buschelman 102397b6df1SKris Buschelman /* get jB, the starting local col index for the 2nd B-part */ 103397b6df1SKris Buschelman colA_start = rstart + ajj[0]; /* the smallest col index for A */ 104397b6df1SKris Buschelman for (j=0; j<countB; j++){ 105397b6df1SKris Buschelman jcol = garray[bjj[j]]; 106397b6df1SKris Buschelman if (jcol > colA_start) { jB = j; break; } 107397b6df1SKris Buschelman if (j==countB-1) jB = countB; 108397b6df1SKris Buschelman } 109397b6df1SKris Buschelman 110397b6df1SKris Buschelman /* B-part, smaller col index */ 111397b6df1SKris Buschelman colA_start = rstart + ajj[0]; /* the smallest col index for A */ 112397b6df1SKris Buschelman for (j=0; j<jB; j++){ 113397b6df1SKris Buschelman jcol = garray[bjj[j]]; 114397b6df1SKris Buschelman if (!valOnly){ 115397b6df1SKris Buschelman row[jj] = irow + shift; col[jj] = jcol + shift; 116397b6df1SKris Buschelman } 117397b6df1SKris Buschelman val[jj++] = *bv++; 118397b6df1SKris Buschelman } 119397b6df1SKris Buschelman /* A-part */ 120397b6df1SKris Buschelman for (j=0; j<countA; j++){ 121397b6df1SKris Buschelman if (!valOnly){ 122397b6df1SKris Buschelman row[jj] = irow + shift; col[jj] = rstart + ajj[j] + shift; 123397b6df1SKris Buschelman } 124397b6df1SKris Buschelman val[jj++] = *av++; 125397b6df1SKris Buschelman } 126397b6df1SKris Buschelman /* B-part, larger col index */ 127397b6df1SKris Buschelman for (j=jB; j<countB; j++){ 128397b6df1SKris Buschelman if (!valOnly){ 129397b6df1SKris Buschelman row[jj] = irow + shift; col[jj] = garray[bjj[j]] + shift; 130397b6df1SKris Buschelman } 131397b6df1SKris Buschelman val[jj++] = *bv++; 132397b6df1SKris Buschelman } 133397b6df1SKris Buschelman irow++; 134397b6df1SKris Buschelman } 135397b6df1SKris Buschelman 136397b6df1SKris Buschelman PetscFunctionReturn(0); 137397b6df1SKris Buschelman } 138397b6df1SKris Buschelman 139397b6df1SKris Buschelman #undef __FUNCT__ 140397b6df1SKris Buschelman #define __FUNCT__ "MatDestroy_AIJ_MUMPS" 141397b6df1SKris Buschelman int MatDestroy_AIJ_MUMPS(Mat A) 142397b6df1SKris Buschelman { 143397b6df1SKris Buschelman Mat_AIJ_MUMPS *lu = (Mat_AIJ_MUMPS*)A->spptr; 144397b6df1SKris Buschelman int ierr,size=lu->size,(*destroy)(Mat)=lu->MatDestroy; 145397b6df1SKris Buschelman 146397b6df1SKris Buschelman PetscFunctionBegin; 147397b6df1SKris Buschelman 148397b6df1SKris Buschelman if (lu->CleanUpMUMPS) { 149397b6df1SKris Buschelman /* Terminate instance, deallocate memories */ 150397b6df1SKris Buschelman lu->id.job=JOB_END; 151397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 152397b6df1SKris Buschelman zmumps_c(&lu->id); 153397b6df1SKris Buschelman #else 154397b6df1SKris Buschelman dmumps_c(&lu->id); 155397b6df1SKris Buschelman #endif 156397b6df1SKris Buschelman if (lu->irn) { ierr = PetscFree(lu->irn);CHKERRQ(ierr);} 157397b6df1SKris Buschelman if (lu->jcn) { ierr = PetscFree(lu->jcn);CHKERRQ(ierr);} 158397b6df1SKris Buschelman if (size>1 && lu->val) { ierr = PetscFree(lu->val);CHKERRQ(ierr);} 159397b6df1SKris Buschelman 160397b6df1SKris Buschelman ierr = MPI_Comm_free(&(lu->comm_mumps));CHKERRQ(ierr); 161397b6df1SKris Buschelman } 162397b6df1SKris Buschelman 163397b6df1SKris Buschelman ierr = PetscFree(lu);CHKERRQ(ierr); 164397b6df1SKris Buschelman ierr = (*destroy)(A);CHKERRQ(ierr); 165397b6df1SKris Buschelman PetscFunctionReturn(0); 166397b6df1SKris Buschelman } 167397b6df1SKris Buschelman 168397b6df1SKris Buschelman #undef __FUNCT__ 169397b6df1SKris Buschelman #define __FUNCT__ "MatAssemblyEnd_AIJ_MUMPS" 170397b6df1SKris Buschelman int MatAssemblyEnd_AIJ_MUMPS(Mat A,MatAssemblyType mode) { 171397b6df1SKris Buschelman int ierr; 1720e3434eeSKris Buschelman Mat_AIJ_MUMPS *mumps=(Mat_AIJ_MUMPS*)A->spptr; 173397b6df1SKris Buschelman 174397b6df1SKris Buschelman PetscFunctionBegin; 175397b6df1SKris Buschelman ierr = (*mumps->MatAssemblyEnd)(A,mode);CHKERRQ(ierr); 176397b6df1SKris Buschelman ierr = MatUseMUMPS_MPIAIJ(A);CHKERRQ(ierr); 177397b6df1SKris Buschelman PetscFunctionReturn(0); 178397b6df1SKris Buschelman } 179397b6df1SKris Buschelman 180397b6df1SKris Buschelman #undef __FUNCT__ 181397b6df1SKris Buschelman #define __FUNCT__ "MatView_AIJ_MUMPS" 182397b6df1SKris Buschelman int MatView_AIJ_MUMPS(Mat A,PetscViewer viewer) { 183397b6df1SKris Buschelman int ierr; 184397b6df1SKris Buschelman PetscTruth isascii; 185397b6df1SKris Buschelman PetscViewerFormat format; 186397b6df1SKris Buschelman Mat_AIJ_MUMPS *mumps=(Mat_AIJ_MUMPS*)(A->spptr); 187397b6df1SKris Buschelman 188397b6df1SKris Buschelman PetscFunctionBegin; 189397b6df1SKris Buschelman ierr = (*mumps->MatView)(A,viewer);CHKERRQ(ierr); 190397b6df1SKris Buschelman 191397b6df1SKris Buschelman ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&isascii);CHKERRQ(ierr); 192397b6df1SKris Buschelman if (isascii) { 193397b6df1SKris Buschelman ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 194397b6df1SKris Buschelman if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 195397b6df1SKris Buschelman ierr = MatFactorInfo_MUMPS(A,viewer);CHKERRQ(ierr); 196397b6df1SKris Buschelman } 197397b6df1SKris Buschelman } 198397b6df1SKris Buschelman PetscFunctionReturn(0); 199397b6df1SKris Buschelman } 200397b6df1SKris Buschelman 201397b6df1SKris Buschelman #undef __FUNCT__ 202397b6df1SKris Buschelman #define __FUNCT__ "MatSolve_AIJ_MUMPS" 203397b6df1SKris Buschelman int MatSolve_AIJ_MUMPS(Mat A,Vec b,Vec x) 204397b6df1SKris Buschelman { 205397b6df1SKris Buschelman Mat_AIJ_MUMPS *lu = (Mat_AIJ_MUMPS*)A->spptr; 206*d54de34fSKris Buschelman PetscScalar *array; 207397b6df1SKris Buschelman Vec x_seq; 208397b6df1SKris Buschelman IS iden; 209397b6df1SKris Buschelman VecScatter scat; 210397b6df1SKris Buschelman int ierr; 211397b6df1SKris Buschelman 212397b6df1SKris Buschelman PetscFunctionBegin; 213397b6df1SKris Buschelman if (lu->size > 1){ 214397b6df1SKris Buschelman if (!lu->myid){ 215397b6df1SKris Buschelman ierr = VecCreateSeq(PETSC_COMM_SELF,A->N,&x_seq);CHKERRQ(ierr); 216397b6df1SKris Buschelman ierr = ISCreateStride(PETSC_COMM_SELF,A->N,0,1,&iden);CHKERRQ(ierr); 217397b6df1SKris Buschelman } else { 218397b6df1SKris Buschelman ierr = VecCreateSeq(PETSC_COMM_SELF,0,&x_seq);CHKERRQ(ierr); 219397b6df1SKris Buschelman ierr = ISCreateStride(PETSC_COMM_SELF,0,0,1,&iden);CHKERRQ(ierr); 220397b6df1SKris Buschelman } 221397b6df1SKris Buschelman ierr = VecScatterCreate(b,iden,x_seq,iden,&scat);CHKERRQ(ierr); 222397b6df1SKris Buschelman ierr = ISDestroy(iden);CHKERRQ(ierr); 223397b6df1SKris Buschelman 224397b6df1SKris Buschelman ierr = VecScatterBegin(b,x_seq,INSERT_VALUES,SCATTER_FORWARD,scat);CHKERRQ(ierr); 225397b6df1SKris Buschelman ierr = VecScatterEnd(b,x_seq,INSERT_VALUES,SCATTER_FORWARD,scat);CHKERRQ(ierr); 226397b6df1SKris Buschelman if (!lu->myid) {ierr = VecGetArray(x_seq,&array);CHKERRQ(ierr);} 227397b6df1SKris Buschelman } else { /* size == 1 */ 228397b6df1SKris Buschelman ierr = VecCopy(b,x);CHKERRQ(ierr); 229397b6df1SKris Buschelman ierr = VecGetArray(x,&array);CHKERRQ(ierr); 230397b6df1SKris Buschelman } 231397b6df1SKris Buschelman if (!lu->myid) { /* define rhs on the host */ 232397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 233397b6df1SKris Buschelman lu->id.rhs = (mumps_double_complex*)array; 234397b6df1SKris Buschelman #else 235397b6df1SKris Buschelman lu->id.rhs = array; 236397b6df1SKris Buschelman #endif 237397b6df1SKris Buschelman } 238397b6df1SKris Buschelman 239397b6df1SKris Buschelman /* solve phase */ 240397b6df1SKris Buschelman lu->id.job=3; 241397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 242397b6df1SKris Buschelman zmumps_c(&lu->id); 243397b6df1SKris Buschelman #else 244397b6df1SKris Buschelman dmumps_c(&lu->id); 245397b6df1SKris Buschelman #endif 246397b6df1SKris Buschelman if (lu->id.INFOG(1) < 0) { 247397b6df1SKris Buschelman SETERRQ1(1,"Error reported by MUMPS in solve phase: INFOG(1)=%d\n",lu->id.INFOG(1)); 248397b6df1SKris Buschelman } 249397b6df1SKris Buschelman 250397b6df1SKris Buschelman /* convert mumps solution x_seq to petsc mpi x */ 251397b6df1SKris Buschelman if (lu->size > 1) { 252397b6df1SKris Buschelman if (!lu->myid){ 253397b6df1SKris Buschelman ierr = VecRestoreArray(x_seq,&array);CHKERRQ(ierr); 254397b6df1SKris Buschelman } 255397b6df1SKris Buschelman ierr = VecScatterBegin(x_seq,x,INSERT_VALUES,SCATTER_REVERSE,scat);CHKERRQ(ierr); 256397b6df1SKris Buschelman ierr = VecScatterEnd(x_seq,x,INSERT_VALUES,SCATTER_REVERSE,scat);CHKERRQ(ierr); 257397b6df1SKris Buschelman ierr = VecScatterDestroy(scat);CHKERRQ(ierr); 258397b6df1SKris Buschelman ierr = VecDestroy(x_seq);CHKERRQ(ierr); 259397b6df1SKris Buschelman } else { 260397b6df1SKris Buschelman ierr = VecRestoreArray(x,&array);CHKERRQ(ierr); 261397b6df1SKris Buschelman } 262397b6df1SKris Buschelman 263397b6df1SKris Buschelman PetscFunctionReturn(0); 264397b6df1SKris Buschelman } 265397b6df1SKris Buschelman 266397b6df1SKris Buschelman #undef __FUNCT__ 267397b6df1SKris Buschelman #define __FUNCT__ "MatFactorNumeric_MPIAIJ_MUMPS" 268397b6df1SKris Buschelman int MatFactorNumeric_AIJ_MUMPS(Mat A,Mat *F) 269397b6df1SKris Buschelman { 270397b6df1SKris Buschelman Mat_AIJ_MUMPS *lu = (Mat_AIJ_MUMPS*)(*F)->spptr; 271397b6df1SKris Buschelman int rnz,nnz,ierr,nz,i,M=A->M,*ai,*aj,icntl; 272397b6df1SKris Buschelman PetscTruth valOnly,flg; 273397b6df1SKris Buschelman 274397b6df1SKris Buschelman PetscFunctionBegin; 275397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ 276397b6df1SKris Buschelman (*F)->ops->solve = MatSolve_AIJ_MUMPS; 277397b6df1SKris Buschelman 278397b6df1SKris Buschelman /* Initialize a MUMPS instance */ 279397b6df1SKris Buschelman ierr = MPI_Comm_rank(A->comm, &lu->myid); 280397b6df1SKris Buschelman ierr = MPI_Comm_size(A->comm,&lu->size);CHKERRQ(ierr); 281397b6df1SKris Buschelman lu->id.job = JOB_INIT; 282397b6df1SKris Buschelman ierr = MPI_Comm_dup(A->comm,&(lu->comm_mumps));CHKERRQ(ierr); 283397b6df1SKris Buschelman lu->id.comm_fortran = lu->comm_mumps; 284397b6df1SKris Buschelman 285397b6df1SKris Buschelman /* Set mumps options */ 286397b6df1SKris Buschelman ierr = PetscOptionsBegin(A->comm,A->prefix,"MUMPS Options","Mat");CHKERRQ(ierr); 287397b6df1SKris Buschelman lu->id.par=1; /* host participates factorizaton and solve */ 288397b6df1SKris Buschelman lu->id.sym=lu->sym; 289397b6df1SKris Buschelman if (lu->sym == 2){ 290397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_sym","SYM: (1,2)","None",lu->id.sym,&icntl,&flg);CHKERRQ(ierr); 291397b6df1SKris Buschelman if (flg && icntl == 1) lu->id.sym=icntl; /* matrix is spd */ 292397b6df1SKris Buschelman } 293397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 294397b6df1SKris Buschelman zmumps_c(&lu->id); 295397b6df1SKris Buschelman #else 296397b6df1SKris Buschelman dmumps_c(&lu->id); 297397b6df1SKris Buschelman #endif 298397b6df1SKris Buschelman 299397b6df1SKris Buschelman if (lu->size == 1){ 300397b6df1SKris Buschelman lu->id.ICNTL(18) = 0; /* centralized assembled matrix input */ 301397b6df1SKris Buschelman } else { 302397b6df1SKris Buschelman lu->id.ICNTL(18) = 3; /* distributed assembled matrix input */ 303397b6df1SKris Buschelman } 304397b6df1SKris Buschelman 305397b6df1SKris Buschelman icntl=-1; 306397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_4","ICNTL(4): level of printing (0 to 4)","None",lu->id.ICNTL(4),&icntl,&flg);CHKERRQ(ierr); 307397b6df1SKris Buschelman if (flg && icntl > 0) { 308397b6df1SKris Buschelman lu->id.ICNTL(4)=icntl; /* and use mumps default icntl(i), i=1,2,3 */ 309397b6df1SKris Buschelman } else { /* no output */ 310397b6df1SKris Buschelman lu->id.ICNTL(1) = 0; /* error message, default= 6 */ 311397b6df1SKris Buschelman lu->id.ICNTL(2) = -1; /* output stream for diagnostic printing, statistics, and warning. default=0 */ 312397b6df1SKris Buschelman lu->id.ICNTL(3) = -1; /* output stream for global information, default=6 */ 313397b6df1SKris Buschelman lu->id.ICNTL(4) = 0; /* level of printing, 0,1,2,3,4, default=2 */ 314397b6df1SKris Buschelman } 315397b6df1SKris 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); 316397b6df1SKris Buschelman icntl=-1; 317397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_7","ICNTL(7): matrix ordering (0 to 7)","None",lu->id.ICNTL(7),&icntl,&flg);CHKERRQ(ierr); 318397b6df1SKris Buschelman if (flg) { 319397b6df1SKris Buschelman if (icntl== 1){ 320397b6df1SKris Buschelman SETERRQ(PETSC_ERR_SUP,"pivot order be set by the user in PERM_IN -- not supported by the PETSc/MUMPS interface\n"); 321397b6df1SKris Buschelman } else { 322397b6df1SKris Buschelman lu->id.ICNTL(7) = icntl; 323397b6df1SKris Buschelman } 324397b6df1SKris Buschelman } 325397b6df1SKris 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); 326397b6df1SKris 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); 327397b6df1SKris 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); 328397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_12","ICNTL(12): efficiency control","None",lu->id.ICNTL(12),&lu->id.ICNTL(12),PETSC_NULL);CHKERRQ(ierr); 329397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_13","ICNTL(13): efficiency control","None",lu->id.ICNTL(13),&lu->id.ICNTL(13),PETSC_NULL);CHKERRQ(ierr); 330397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_14","ICNTL(14): efficiency control","None",lu->id.ICNTL(14),&lu->id.ICNTL(14),PETSC_NULL);CHKERRQ(ierr); 331397b6df1SKris Buschelman ierr = PetscOptionsInt("-mat_mumps_icntl_15","ICNTL(15): efficiency control","None",lu->id.ICNTL(15),&lu->id.ICNTL(15),PETSC_NULL);CHKERRQ(ierr); 332397b6df1SKris Buschelman 333397b6df1SKris Buschelman /* 334397b6df1SKris 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); 335397b6df1SKris Buschelman if (flg){ 336397b6df1SKris Buschelman if (icntl >-1 && icntl <3 ){ 337397b6df1SKris Buschelman if (lu->myid==0) lu->id.ICNTL(16) = icntl; 338397b6df1SKris Buschelman } else { 339397b6df1SKris Buschelman SETERRQ1(PETSC_ERR_SUP,"ICNTL(16)=%d -- not supported\n",icntl); 340397b6df1SKris Buschelman } 341397b6df1SKris Buschelman } 342397b6df1SKris Buschelman */ 343397b6df1SKris Buschelman 344397b6df1SKris 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); 345397b6df1SKris 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); 346397b6df1SKris 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); 347397b6df1SKris Buschelman PetscOptionsEnd(); 348397b6df1SKris Buschelman } 349397b6df1SKris Buschelman 350397b6df1SKris Buschelman /* define matrix A */ 351397b6df1SKris Buschelman switch (lu->id.ICNTL(18)){ 352397b6df1SKris Buschelman case 0: /* centralized assembled matrix input (size=1) */ 353397b6df1SKris Buschelman if (!lu->myid) { 354397b6df1SKris Buschelman if (lu->isAIJ){ 355397b6df1SKris Buschelman Mat_SeqAIJ *aa = (Mat_SeqAIJ*)A->data; 356397b6df1SKris Buschelman nz = aa->nz; 357397b6df1SKris Buschelman ai = aa->i; aj = aa->j; lu->val = aa->a; 358397b6df1SKris Buschelman } else { 359397b6df1SKris Buschelman Mat_SeqSBAIJ *aa = (Mat_SeqSBAIJ*)A->data; 360397b6df1SKris Buschelman nz = aa->s_nz; 361397b6df1SKris Buschelman ai = aa->i; aj = aa->j; lu->val = aa->a; 362397b6df1SKris Buschelman } 363397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ /* first numeric factorization, get irn and jcn */ 364397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(int),&lu->irn);CHKERRQ(ierr); 365397b6df1SKris Buschelman ierr = PetscMalloc(nz*sizeof(int),&lu->jcn);CHKERRQ(ierr); 366397b6df1SKris Buschelman nz = 0; 367397b6df1SKris Buschelman for (i=0; i<M; i++){ 368397b6df1SKris Buschelman rnz = ai[i+1] - ai[i]; 369397b6df1SKris Buschelman while (rnz--) { /* Fortran row/col index! */ 370397b6df1SKris Buschelman lu->irn[nz] = i+1; lu->jcn[nz] = (*aj)+1; aj++; nz++; 371397b6df1SKris Buschelman } 372397b6df1SKris Buschelman } 373397b6df1SKris Buschelman } 374397b6df1SKris Buschelman } 375397b6df1SKris Buschelman break; 376397b6df1SKris Buschelman case 3: /* distributed assembled matrix input (size>1) */ 377397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ 378397b6df1SKris Buschelman valOnly = PETSC_FALSE; 379397b6df1SKris Buschelman } else { 380397b6df1SKris Buschelman valOnly = PETSC_TRUE; /* only update mat values, not row and col index */ 381397b6df1SKris Buschelman } 382397b6df1SKris Buschelman ierr = MatConvertToTriples(A,1,valOnly, &nnz, &lu->irn, &lu->jcn, &lu->val);CHKERRQ(ierr); 383397b6df1SKris Buschelman break; 384397b6df1SKris Buschelman default: SETERRQ(PETSC_ERR_SUP,"Matrix input format is not supported by MUMPS."); 385397b6df1SKris Buschelman } 386397b6df1SKris Buschelman 387397b6df1SKris Buschelman /* analysis phase */ 388397b6df1SKris Buschelman if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ 389397b6df1SKris Buschelman lu->id.n = M; 390397b6df1SKris Buschelman switch (lu->id.ICNTL(18)){ 391397b6df1SKris Buschelman case 0: /* centralized assembled matrix input */ 392397b6df1SKris Buschelman if (!lu->myid) { 393397b6df1SKris Buschelman lu->id.nz =nz; lu->id.irn=lu->irn; lu->id.jcn=lu->jcn; 394397b6df1SKris Buschelman if (lu->id.ICNTL(6)>1){ 395397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 396397b6df1SKris Buschelman lu->id.a = (mumps_double_complex*)lu->val; 397397b6df1SKris Buschelman #else 398397b6df1SKris Buschelman lu->id.a = lu->val; 399397b6df1SKris Buschelman #endif 400397b6df1SKris Buschelman } 401397b6df1SKris Buschelman } 402397b6df1SKris Buschelman break; 403397b6df1SKris Buschelman case 3: /* distributed assembled matrix input (size>1) */ 404397b6df1SKris Buschelman lu->id.nz_loc = nnz; 405397b6df1SKris Buschelman lu->id.irn_loc=lu->irn; lu->id.jcn_loc=lu->jcn; 406397b6df1SKris Buschelman if (lu->id.ICNTL(6)>1) { 407397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 408397b6df1SKris Buschelman lu->id.a_loc = (mumps_double_complex*)lu->val; 409397b6df1SKris Buschelman #else 410397b6df1SKris Buschelman lu->id.a_loc = lu->val; 411397b6df1SKris Buschelman #endif 412397b6df1SKris Buschelman } 413397b6df1SKris Buschelman break; 414397b6df1SKris Buschelman } 415397b6df1SKris Buschelman lu->id.job=1; 416397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 417397b6df1SKris Buschelman zmumps_c(&lu->id); 418397b6df1SKris Buschelman #else 419397b6df1SKris Buschelman dmumps_c(&lu->id); 420397b6df1SKris Buschelman #endif 421397b6df1SKris Buschelman if (lu->id.INFOG(1) < 0) { 422397b6df1SKris Buschelman SETERRQ1(1,"Error reported by MUMPS in analysis phase: INFOG(1)=%d\n",lu->id.INFOG(1)); 423397b6df1SKris Buschelman } 424397b6df1SKris Buschelman } 425397b6df1SKris Buschelman 426397b6df1SKris Buschelman /* numerical factorization phase */ 427397b6df1SKris Buschelman if(lu->id.ICNTL(18) == 0) { 428397b6df1SKris Buschelman if (lu->myid == 0) { 429397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 430397b6df1SKris Buschelman lu->id.a = (mumps_double_complex*)lu->val; 431397b6df1SKris Buschelman #else 432397b6df1SKris Buschelman lu->id.a = lu->val; 433397b6df1SKris Buschelman #endif 434397b6df1SKris Buschelman } 435397b6df1SKris Buschelman } else { 436397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 437397b6df1SKris Buschelman lu->id.a_loc = (mumps_double_complex*)lu->val; 438397b6df1SKris Buschelman #else 439397b6df1SKris Buschelman lu->id.a_loc = lu->val; 440397b6df1SKris Buschelman #endif 441397b6df1SKris Buschelman } 442397b6df1SKris Buschelman lu->id.job=2; 443397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX) 444397b6df1SKris Buschelman zmumps_c(&lu->id); 445397b6df1SKris Buschelman #else 446397b6df1SKris Buschelman dmumps_c(&lu->id); 447397b6df1SKris Buschelman #endif 448397b6df1SKris Buschelman if (lu->id.INFOG(1) < 0) { 449397b6df1SKris Buschelman SETERRQ1(1,"1, Error reported by MUMPS in numerical factorization phase: INFOG(1)=%d\n",lu->id.INFOG(1)); 450397b6df1SKris Buschelman } 451397b6df1SKris Buschelman 452397b6df1SKris Buschelman if (lu->myid==0 && lu->id.ICNTL(16) > 0){ 453397b6df1SKris Buschelman SETERRQ1(1," lu->id.ICNTL(16):=%d\n",lu->id.INFOG(16)); 454397b6df1SKris Buschelman } 455397b6df1SKris Buschelman 456397b6df1SKris Buschelman (*F)->assembled = PETSC_TRUE; 457397b6df1SKris Buschelman lu->matstruc = SAME_NONZERO_PATTERN; 458397b6df1SKris Buschelman PetscFunctionReturn(0); 459397b6df1SKris Buschelman } 460397b6df1SKris Buschelman 461397b6df1SKris Buschelman /* Note the Petsc r and c permutations are ignored */ 462397b6df1SKris Buschelman #undef __FUNCT__ 463397b6df1SKris Buschelman #define __FUNCT__ "MatLUFactorSymbolic_AIJ_MUMPS" 464397b6df1SKris Buschelman int MatLUFactorSymbolic_AIJ_MUMPS(Mat A,IS r,IS c,MatFactorInfo *info,Mat *F) 465397b6df1SKris Buschelman { 466397b6df1SKris Buschelman Mat B; 467397b6df1SKris Buschelman Mat_AIJ_MUMPS *lu; 468397b6df1SKris Buschelman int ierr; 469397b6df1SKris Buschelman 470397b6df1SKris Buschelman PetscFunctionBegin; 471397b6df1SKris Buschelman 472397b6df1SKris Buschelman /* Create the factorization matrix */ 473397b6df1SKris Buschelman ierr = MatCreate(A->comm,A->m,A->n,A->M,A->N,&B);CHKERRQ(ierr); 474397b6df1SKris Buschelman ierr = MatSetType(B,MATAIJMUMPS);CHKERRQ(ierr); 475397b6df1SKris Buschelman ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr); 476397b6df1SKris Buschelman ierr = MatMPIAIJSetPreallocation(B,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr); 477397b6df1SKris Buschelman 478397b6df1SKris Buschelman B->ops->lufactornumeric = MatFactorNumeric_AIJ_MUMPS; 479397b6df1SKris Buschelman B->factor = FACTOR_LU; 4800e3434eeSKris Buschelman lu = (Mat_AIJ_MUMPS*)B->spptr; 481397b6df1SKris Buschelman lu->sym = 0; 482397b6df1SKris Buschelman lu->matstruc = DIFFERENT_NONZERO_PATTERN; 483397b6df1SKris Buschelman 484397b6df1SKris Buschelman *F = B; 485397b6df1SKris Buschelman PetscFunctionReturn(0); 486397b6df1SKris Buschelman } 487397b6df1SKris Buschelman 488397b6df1SKris Buschelman /* Note the Petsc r permutation is ignored */ 489397b6df1SKris Buschelman #undef __FUNCT__ 490397b6df1SKris Buschelman #define __FUNCT__ "MatCholeskyFactorSymbolic_AIJ_MUMPS" 491397b6df1SKris Buschelman int MatCholeskyFactorSymbolic_AIJ_MUMPS(Mat A,IS r,MatFactorInfo *info,Mat *F) 492397b6df1SKris Buschelman { 493397b6df1SKris Buschelman Mat B; 494397b6df1SKris Buschelman Mat_AIJ_MUMPS *lu; 495397b6df1SKris Buschelman int ierr; 496397b6df1SKris Buschelman 497397b6df1SKris Buschelman PetscFunctionBegin; 498397b6df1SKris Buschelman 499397b6df1SKris Buschelman /* Create the factorization matrix */ 500397b6df1SKris Buschelman ierr = MatCreate(A->comm,A->m,A->n,A->M,A->N,&B);CHKERRQ(ierr); 501397b6df1SKris Buschelman ierr = MatSetType(B,MATAIJMUMPS);CHKERRQ(ierr); 502397b6df1SKris Buschelman ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr); 503397b6df1SKris Buschelman ierr = MatMPIAIJSetPreallocation(B,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr); 504397b6df1SKris Buschelman 505397b6df1SKris Buschelman B->ops->choleskyfactornumeric = MatFactorNumeric_AIJ_MUMPS; 506397b6df1SKris Buschelman B->factor = FACTOR_CHOLESKY; 507397b6df1SKris Buschelman lu = (Mat_AIJ_MUMPS *)B->spptr; 508397b6df1SKris Buschelman lu->sym = 2; 509397b6df1SKris Buschelman lu->matstruc = DIFFERENT_NONZERO_PATTERN; 510397b6df1SKris Buschelman 511397b6df1SKris Buschelman *F = B; 512397b6df1SKris Buschelman PetscFunctionReturn(0); 513397b6df1SKris Buschelman } 514397b6df1SKris Buschelman 515397b6df1SKris Buschelman #undef __FUNCT__ 516397b6df1SKris Buschelman #define __FUNCT__ "MatUseMUMPS_AIJ" 517397b6df1SKris Buschelman int MatUseMUMPS_AIJ(Mat A) 518397b6df1SKris Buschelman { 519397b6df1SKris Buschelman PetscFunctionBegin; 520397b6df1SKris Buschelman A->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJ_MUMPS; 521397b6df1SKris Buschelman A->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_AIJ_MUMPS; 522397b6df1SKris Buschelman A->ops->lufactornumeric = MatFactorNumeric_AIJ_MUMPS; 523397b6df1SKris Buschelman 524397b6df1SKris Buschelman PetscFunctionReturn(0); 525397b6df1SKris Buschelman } 526397b6df1SKris Buschelman 527397b6df1SKris Buschelman int MatFactorInfo_MUMPS(Mat A,PetscViewer viewer) 528397b6df1SKris Buschelman { 529397b6df1SKris Buschelman Mat_AIJ_MUMPS *lu= (Mat_AIJ_MUMPS*)A->spptr; 530397b6df1SKris Buschelman int ierr; 531397b6df1SKris Buschelman 532397b6df1SKris Buschelman PetscFunctionBegin; 533397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer,"MUMPS run parameters:\n");CHKERRQ(ierr); 534397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," SYM (matrix type): %d \n",lu->id.sym);CHKERRQ(ierr); 535397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," PAR (host participation): %d \n",lu->id.par);CHKERRQ(ierr); 536397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(4) (level of printing): %d \n",lu->id.ICNTL(4));CHKERRQ(ierr); 537397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(5) (input mat struct): %d \n",lu->id.ICNTL(5));CHKERRQ(ierr); 538397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(6) (matrix prescaling): %d \n",lu->id.ICNTL(6));CHKERRQ(ierr); 539397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(7) (matrix ordering): %d \n",lu->id.ICNTL(7));CHKERRQ(ierr); 540397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(9) (A/A^T x=b is solved): %d \n",lu->id.ICNTL(9));CHKERRQ(ierr); 541397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(10) (max num of refinements): %d \n",lu->id.ICNTL(10));CHKERRQ(ierr); 542397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(11) (error analysis): %d \n",lu->id.ICNTL(11));CHKERRQ(ierr); 543397b6df1SKris Buschelman if (lu->myid == 0 && lu->id.ICNTL(11)>0) { 544397b6df1SKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(4) (inf norm of input mat): %g\n",lu->id.RINFOG(4));CHKERRQ(ierr); 545397b6df1SKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(5) (inf norm of solution): %g\n",lu->id.RINFOG(5));CHKERRQ(ierr); 546397b6df1SKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(6) (inf norm of residual): %g\n",lu->id.RINFOG(6));CHKERRQ(ierr); 547397b6df1SKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(7),RINFOG(8) (backward error est): %g\n",lu->id.RINFOG(7),lu->id.RINFOG(8));CHKERRQ(ierr); 548397b6df1SKris Buschelman ierr = PetscPrintf(PETSC_COMM_SELF," RINFOG(9) (error estimate): %g \n",lu->id.RINFOG(9));CHKERRQ(ierr); 549397b6df1SKris 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); 550397b6df1SKris Buschelman 551397b6df1SKris Buschelman } 552397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(12) (efficiency control): %d \n",lu->id.ICNTL(12));CHKERRQ(ierr); 553397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(13) (efficiency control): %d \n",lu->id.ICNTL(13));CHKERRQ(ierr); 554397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(14) (efficiency control): %d \n",lu->id.ICNTL(14));CHKERRQ(ierr); 555397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(15) (efficiency control): %d \n",lu->id.ICNTL(15));CHKERRQ(ierr); 556397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," ICNTL(18) (input mat struct): %d \n",lu->id.ICNTL(18));CHKERRQ(ierr); 557397b6df1SKris Buschelman 558397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," CNTL(1) (relative pivoting threshold): %g \n",lu->id.CNTL(1));CHKERRQ(ierr); 559397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," CNTL(2) (stopping criterion of refinement): %g \n",lu->id.CNTL(2));CHKERRQ(ierr); 560397b6df1SKris Buschelman ierr = PetscViewerASCIIPrintf(viewer," CNTL(3) (absolute pivoting threshold): %g \n",lu->id.CNTL(3));CHKERRQ(ierr); 561397b6df1SKris Buschelman PetscFunctionReturn(0); 562397b6df1SKris Buschelman } 563397b6df1SKris Buschelman 564397b6df1SKris Buschelman EXTERN_C_BEGIN 565397b6df1SKris Buschelman #undef __FUNCT__ 566397b6df1SKris Buschelman #define __FUNCT__ "MatCreate_AIJ_MUMPS" 567397b6df1SKris Buschelman int MatCreate_AIJ_MUMPS(Mat A) { 568397b6df1SKris Buschelman int ierr,size; 569397b6df1SKris Buschelman MPI_Comm comm; 570397b6df1SKris Buschelman Mat_AIJ_MUMPS *mumps; 571397b6df1SKris Buschelman 572397b6df1SKris Buschelman PetscFunctionBegin; 573397b6df1SKris Buschelman ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 574397b6df1SKris Buschelman ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr); 575397b6df1SKris Buschelman if (size == 1) { 576397b6df1SKris Buschelman ierr = MatSetType(A,MATSEQAIJ);CHKERRQ(ierr); 577397b6df1SKris Buschelman } else { 578397b6df1SKris Buschelman ierr = MatSetType(A,MATMPIAIJ);CHKERRQ(ierr); 579397b6df1SKris Buschelman } 580397b6df1SKris Buschelman ierr = MatUseMUMPS_AIJ(A); 581397b6df1SKris Buschelman 582397b6df1SKris Buschelman ierr = PetscNew(Mat_AIJ_MUMPS,&mumps);CHKERRQ(ierr); 583397b6df1SKris Buschelman mumps->MatDestroy = A->ops->destroy; 584397b6df1SKris Buschelman mumps->MatAssemblyEnd = A->ops->assemblyend; 585397b6df1SKris Buschelman mumps->MatView = A->ops->view; 586397b6df1SKris Buschelman mumps->CleanUpMUMPS = PETSC_FALSE; 587397b6df1SKris Buschelman A->spptr = (void *)mumps; 588397b6df1SKris Buschelman A->ops->destroy = MatDestroy_AIJ_MUMPS; 589397b6df1SKris Buschelman A->ops->assemblyend = MatAssemblyEnd_AIJ_MUMPS; 590397b6df1SKris Buschelman A->ops->view = MatView_AIJ_MUMPS; 591397b6df1SKris Buschelman PetscFunctionReturn(0); 592397b6df1SKris Buschelman } 593397b6df1SKris Buschelman EXTERN_C_END 594397b6df1SKris Buschelman 595397b6df1SKris Buschelman EXTERN_C_BEGIN 596397b6df1SKris Buschelman #undef __FUNCT__ 597397b6df1SKris Buschelman #define __FUNCT__ "MatCreate_SBAIJ_MUMPS" 598397b6df1SKris Buschelman int MatCreate_SBAIJ_MUMPS(Mat A) { 599397b6df1SKris Buschelman int ierr,size; 600397b6df1SKris Buschelman MPI_Comm comm; 601397b6df1SKris Buschelman Mat_AIJ_MUMPS *mumps; 602397b6df1SKris Buschelman 603397b6df1SKris Buschelman PetscFunctionBegin; 604397b6df1SKris Buschelman ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 605397b6df1SKris Buschelman ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr); 606397b6df1SKris Buschelman if (size == 1) { 607397b6df1SKris Buschelman ierr = MatSetType(A,MATSEQSBAIJ);CHKERRQ(ierr); 608397b6df1SKris Buschelman } else { 609397b6df1SKris Buschelman ierr = MatSetType(A,MATMPISBAIJ);CHKERRQ(ierr); 610397b6df1SKris Buschelman } 611397b6df1SKris Buschelman ierr=MatUseMUMPS_AIJ(A); 612397b6df1SKris Buschelman 613397b6df1SKris Buschelman ierr = PetscNew(Mat_AIJ_MUMPS,&mumps);CHKERRQ(ierr); 614397b6df1SKris Buschelman mumps->MatDestroy = A->ops->destroy; 615397b6df1SKris Buschelman mumps->MatAssemblyEnd = A->ops->assemblyend; 616397b6df1SKris Buschelman mumps->CleanUpMUMPS = PETSC_FALSE; 617397b6df1SKris Buschelman A->spptr = (void *)mumps; 618397b6df1SKris Buschelman A->ops->destroy = MatDestroy_AIJ_MUMPS; 619397b6df1SKris Buschelman A->ops->assemblyend = MatAssemblyEnd_AIJ_MUMPS; 620397b6df1SKris Buschelman 621397b6df1SKris Buschelman PetscFunctionReturn(0); 622397b6df1SKris Buschelman } 623397b6df1SKris Buschelman EXTERN_C_END 624397b6df1SKris Buschelman 625397b6df1SKris Buschelman EXTERN_C_BEGIN 626397b6df1SKris Buschelman #undef __FUNCT__ 627397b6df1SKris Buschelman #define __FUNCT__ "MatLoad_AIJ_MUMPS" 628397b6df1SKris Buschelman int MatLoad_AIJ_MUMPS(PetscViewer viewer,MatType type,Mat *A) { 629397b6df1SKris Buschelman int ierr,size,(*r)(PetscViewer,MatType,Mat*); 630397b6df1SKris Buschelman MPI_Comm comm; 631397b6df1SKris Buschelman 632397b6df1SKris Buschelman PetscFunctionBegin; 633397b6df1SKris Buschelman ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 634397b6df1SKris Buschelman ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 635397b6df1SKris Buschelman if (size == 1) { 636397b6df1SKris Buschelman ierr = PetscFListFind(comm,MatLoadList,MATSEQAIJ,(void(**)(void))&r);CHKERRQ(ierr); 637397b6df1SKris Buschelman } else { 638397b6df1SKris Buschelman ierr = PetscFListFind(comm,MatLoadList,MATMPIAIJ,(void(**)(void))&r);CHKERRQ(ierr); 639397b6df1SKris Buschelman } 640397b6df1SKris Buschelman ierr = (*r)(viewer,type,A);CHKERRQ(ierr); 641397b6df1SKris Buschelman PetscFunctionReturn(0); 642397b6df1SKris Buschelman } 643397b6df1SKris Buschelman EXTERN_C_END 644397b6df1SKris Buschelman 645397b6df1SKris Buschelman EXTERN_C_BEGIN 646397b6df1SKris Buschelman #undef __FUNCT__ 647397b6df1SKris Buschelman #define __FUNCT__ "MatLoad_SBAIJ_MUMPS" 648397b6df1SKris Buschelman int MatLoad_SBAIJ_MUMPS(PetscViewer viewer,MatType type,Mat *A) { 649397b6df1SKris Buschelman int ierr,size,(*r)(PetscViewer,MatType,Mat*); 650397b6df1SKris Buschelman MPI_Comm comm; 651397b6df1SKris Buschelman 652397b6df1SKris Buschelman PetscFunctionBegin; 653397b6df1SKris Buschelman ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 654397b6df1SKris Buschelman ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 655397b6df1SKris Buschelman if (size == 1) { 656397b6df1SKris Buschelman ierr = PetscFListFind(comm,MatLoadList,MATSEQSBAIJ,(void(**)(void))&r);CHKERRQ(ierr); 657397b6df1SKris Buschelman } else { 658397b6df1SKris Buschelman ierr = PetscFListFind(comm,MatLoadList,MATMPISBAIJ,(void(**)(void))&r);CHKERRQ(ierr); 659397b6df1SKris Buschelman } 660397b6df1SKris Buschelman ierr = (*r)(viewer,type,A);CHKERRQ(ierr); 661397b6df1SKris Buschelman PetscFunctionReturn(0); 662397b6df1SKris Buschelman } 663397b6df1SKris Buschelman EXTERN_C_END 664