xref: /petsc/src/mat/impls/aij/mpi/mumps/mumps.c (revision adc1d99f45462a3a5c7249b84fb681d9a8f6d761)
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]
24*adc1d99fSHong 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);
261*adc1d99fSHong 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*adc1d99fSHong Zhang   /*
269*adc1d99fSHong Zhang   ierr = PetscViewerASCIIPrintf(viewer,"  RINFO(1) (local estimated flops for the elimination after analysis): %g \n",lu->id.RINFO(1));CHKERRQ(ierr);
270*adc1d99fSHong Zhang   ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] RINFO(2): %g\n",lu->myid,lu->id.RINFO(2));
271*adc1d99fSHong Zhang   */
272*adc1d99fSHong Zhang 
273*adc1d99fSHong Zhang   if (lu->myid == 0){ /* information from the host */
274*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  RINFOG(1) (global estimated flops for the elimination after analysis): %g \n",lu->id.RINFOG(1));CHKERRQ(ierr);
275*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  RINFOG(2) (global estimated flops for the assembly after factorization): %g \n",lu->id.RINFOG(2));CHKERRQ(ierr);
276*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  RINFOG(3) (global estimated flops for the elimination after factorization): %g \n",lu->id.RINFOG(3));CHKERRQ(ierr);
277*adc1d99fSHong Zhang 
278*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(3) (estimated real workspace for factors on all processors after analysis): %d \n",lu->id.INFOG(3));CHKERRQ(ierr);
279*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(4) (estimated integer workspace for factors on all processors after analysis): %d \n",lu->id.INFOG(4));CHKERRQ(ierr);
280*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(5) (estimated maximum front size in the complete tree): %d \n",lu->id.INFOG(5));CHKERRQ(ierr);
281*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(6) (number of nodes in the complete tree): %d \n",lu->id.INFOG(6));CHKERRQ(ierr);
282*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(7) (ordering option effectively uese after analysis): %d \n",lu->id.INFOG(7));CHKERRQ(ierr);
283*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(8) (structural symmetry in percent of the permuted matrix after analysis): %d \n",lu->id.INFOG(8));CHKERRQ(ierr);
284*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(9) (total real space store the matrix factors after analysis): %d \n",lu->id.INFOG(9));CHKERRQ(ierr);
285*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(10) (total integer space store the matrix factors after analysis): %d \n",lu->id.INFOG(10));CHKERRQ(ierr);
286*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(11) (order of largest frontal matrix): %d \n",lu->id.INFOG(11));CHKERRQ(ierr);
287*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(12) (number of off-diagonal pivots): %d \n",lu->id.INFOG(12));CHKERRQ(ierr);
288*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(13) (number of delayed pivots after factorization): %d \n",lu->id.INFOG(13));CHKERRQ(ierr);
289*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(14) (number of memory compress after factorization): %d \n",lu->id.INFOG(14));CHKERRQ(ierr);
290*adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(15) (number of steps of iterative refinement after solution): %d \n",lu->id.INFOG(15));CHKERRQ(ierr);
291*adc1d99fSHong 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);
292*adc1d99fSHong 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);
293*adc1d99fSHong 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);
294*adc1d99fSHong 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);
295*adc1d99fSHong Zhang      ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(20) (estimated number of entries in the factors): %d \n",lu->id.INFOG(20));CHKERRQ(ierr);
296*adc1d99fSHong Zhang   }
297*adc1d99fSHong Zhang 
298397b6df1SKris Buschelman   PetscFunctionReturn(0);
299397b6df1SKris Buschelman }
300397b6df1SKris Buschelman 
301397b6df1SKris Buschelman #undef __FUNCT__
302f0c56d0fSKris Buschelman #define __FUNCT__ "MatView_AIJMUMPS"
303f0c56d0fSKris Buschelman int MatView_AIJMUMPS(Mat A,PetscViewer viewer) {
304397b6df1SKris Buschelman   int               ierr;
305397b6df1SKris Buschelman   PetscTruth        isascii;
306397b6df1SKris Buschelman   PetscViewerFormat format;
307f0c56d0fSKris Buschelman   Mat_MUMPS         *mumps=(Mat_MUMPS*)(A->spptr);
308397b6df1SKris Buschelman 
309397b6df1SKris Buschelman   PetscFunctionBegin;
310397b6df1SKris Buschelman   ierr = (*mumps->MatView)(A,viewer);CHKERRQ(ierr);
311397b6df1SKris Buschelman 
312397b6df1SKris Buschelman   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&isascii);CHKERRQ(ierr);
313397b6df1SKris Buschelman   if (isascii) {
314397b6df1SKris Buschelman     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
315397b6df1SKris Buschelman     if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
316397b6df1SKris Buschelman       ierr = MatFactorInfo_MUMPS(A,viewer);CHKERRQ(ierr);
317397b6df1SKris Buschelman     }
318397b6df1SKris Buschelman   }
319397b6df1SKris Buschelman   PetscFunctionReturn(0);
320397b6df1SKris Buschelman }
321397b6df1SKris Buschelman 
322397b6df1SKris Buschelman #undef __FUNCT__
323f0c56d0fSKris Buschelman #define __FUNCT__ "MatSolve_AIJMUMPS"
324f0c56d0fSKris Buschelman int MatSolve_AIJMUMPS(Mat A,Vec b,Vec x) {
325f0c56d0fSKris Buschelman   Mat_MUMPS   *lu=(Mat_MUMPS*)A->spptr;
326d54de34fSKris Buschelman   PetscScalar *array;
327397b6df1SKris Buschelman   Vec         x_seq;
328397b6df1SKris Buschelman   IS          iden;
329397b6df1SKris Buschelman   VecScatter  scat;
330397b6df1SKris Buschelman   int         ierr;
331397b6df1SKris Buschelman 
332397b6df1SKris Buschelman   PetscFunctionBegin;
333397b6df1SKris Buschelman   if (lu->size > 1){
334397b6df1SKris Buschelman     if (!lu->myid){
335397b6df1SKris Buschelman       ierr = VecCreateSeq(PETSC_COMM_SELF,A->N,&x_seq);CHKERRQ(ierr);
336397b6df1SKris Buschelman       ierr = ISCreateStride(PETSC_COMM_SELF,A->N,0,1,&iden);CHKERRQ(ierr);
337397b6df1SKris Buschelman     } else {
338397b6df1SKris Buschelman       ierr = VecCreateSeq(PETSC_COMM_SELF,0,&x_seq);CHKERRQ(ierr);
339397b6df1SKris Buschelman       ierr = ISCreateStride(PETSC_COMM_SELF,0,0,1,&iden);CHKERRQ(ierr);
340397b6df1SKris Buschelman     }
341397b6df1SKris Buschelman     ierr = VecScatterCreate(b,iden,x_seq,iden,&scat);CHKERRQ(ierr);
342397b6df1SKris Buschelman     ierr = ISDestroy(iden);CHKERRQ(ierr);
343397b6df1SKris Buschelman 
344397b6df1SKris Buschelman     ierr = VecScatterBegin(b,x_seq,INSERT_VALUES,SCATTER_FORWARD,scat);CHKERRQ(ierr);
345397b6df1SKris Buschelman     ierr = VecScatterEnd(b,x_seq,INSERT_VALUES,SCATTER_FORWARD,scat);CHKERRQ(ierr);
346397b6df1SKris Buschelman     if (!lu->myid) {ierr = VecGetArray(x_seq,&array);CHKERRQ(ierr);}
347397b6df1SKris Buschelman   } else {  /* size == 1 */
348397b6df1SKris Buschelman     ierr = VecCopy(b,x);CHKERRQ(ierr);
349397b6df1SKris Buschelman     ierr = VecGetArray(x,&array);CHKERRQ(ierr);
350397b6df1SKris Buschelman   }
351397b6df1SKris Buschelman   if (!lu->myid) { /* define rhs on the host */
352397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
353397b6df1SKris Buschelman     lu->id.rhs = (mumps_double_complex*)array;
354397b6df1SKris Buschelman #else
355397b6df1SKris Buschelman     lu->id.rhs = array;
356397b6df1SKris Buschelman #endif
357397b6df1SKris Buschelman   }
358397b6df1SKris Buschelman 
359397b6df1SKris Buschelman   /* solve phase */
360397b6df1SKris Buschelman   lu->id.job=3;
361397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
362397b6df1SKris Buschelman   zmumps_c(&lu->id);
363397b6df1SKris Buschelman #else
364397b6df1SKris Buschelman   dmumps_c(&lu->id);
365397b6df1SKris Buschelman #endif
366397b6df1SKris Buschelman   if (lu->id.INFOG(1) < 0) {
367397b6df1SKris Buschelman     SETERRQ1(1,"Error reported by MUMPS in solve phase: INFOG(1)=%d\n",lu->id.INFOG(1));
368397b6df1SKris Buschelman   }
369397b6df1SKris Buschelman 
370397b6df1SKris Buschelman   /* convert mumps solution x_seq to petsc mpi x */
371397b6df1SKris Buschelman   if (lu->size > 1) {
372397b6df1SKris Buschelman     if (!lu->myid){
373397b6df1SKris Buschelman       ierr = VecRestoreArray(x_seq,&array);CHKERRQ(ierr);
374397b6df1SKris Buschelman     }
375397b6df1SKris Buschelman     ierr = VecScatterBegin(x_seq,x,INSERT_VALUES,SCATTER_REVERSE,scat);CHKERRQ(ierr);
376397b6df1SKris Buschelman     ierr = VecScatterEnd(x_seq,x,INSERT_VALUES,SCATTER_REVERSE,scat);CHKERRQ(ierr);
377397b6df1SKris Buschelman     ierr = VecScatterDestroy(scat);CHKERRQ(ierr);
378397b6df1SKris Buschelman     ierr = VecDestroy(x_seq);CHKERRQ(ierr);
379397b6df1SKris Buschelman   } else {
380397b6df1SKris Buschelman     ierr = VecRestoreArray(x,&array);CHKERRQ(ierr);
381397b6df1SKris Buschelman   }
382397b6df1SKris Buschelman 
383397b6df1SKris Buschelman   PetscFunctionReturn(0);
384397b6df1SKris Buschelman }
385397b6df1SKris Buschelman 
386397b6df1SKris Buschelman #undef __FUNCT__
387f0c56d0fSKris Buschelman #define __FUNCT__ "MatFactorNumeric_MPIAIJMUMPS"
388f0c56d0fSKris Buschelman int MatFactorNumeric_AIJMUMPS(Mat A,Mat *F) {
389f0c56d0fSKris Buschelman   Mat_MUMPS  *lu =(Mat_MUMPS*)(*F)->spptr;
390f0c56d0fSKris Buschelman   Mat_MUMPS  *lua=(Mat_MUMPS*)(A)->spptr;
391397b6df1SKris Buschelman   int        rnz,nnz,ierr,nz,i,M=A->M,*ai,*aj,icntl;
392397b6df1SKris Buschelman   PetscTruth valOnly,flg;
393397b6df1SKris Buschelman 
394397b6df1SKris Buschelman   PetscFunctionBegin;
395397b6df1SKris Buschelman   if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){
396f0c56d0fSKris Buschelman     (*F)->ops->solve    = MatSolve_AIJMUMPS;
397397b6df1SKris Buschelman 
398397b6df1SKris Buschelman     /* Initialize a MUMPS instance */
399397b6df1SKris Buschelman     ierr = MPI_Comm_rank(A->comm, &lu->myid);
400397b6df1SKris Buschelman     ierr = MPI_Comm_size(A->comm,&lu->size);CHKERRQ(ierr);
40175747be1SHong Zhang     lua->myid = lu->myid; lua->size = lu->size;
402397b6df1SKris Buschelman     lu->id.job = JOB_INIT;
403397b6df1SKris Buschelman     ierr = MPI_Comm_dup(A->comm,&(lu->comm_mumps));CHKERRQ(ierr);
404397b6df1SKris Buschelman     lu->id.comm_fortran = lu->comm_mumps;
405397b6df1SKris Buschelman 
406397b6df1SKris Buschelman     /* Set mumps options */
407397b6df1SKris Buschelman     ierr = PetscOptionsBegin(A->comm,A->prefix,"MUMPS Options","Mat");CHKERRQ(ierr);
408397b6df1SKris Buschelman     lu->id.par=1;  /* host participates factorizaton and solve */
409397b6df1SKris Buschelman     lu->id.sym=lu->sym;
410397b6df1SKris Buschelman     if (lu->sym == 2){
411397b6df1SKris Buschelman       ierr = PetscOptionsInt("-mat_mumps_sym","SYM: (1,2)","None",lu->id.sym,&icntl,&flg);CHKERRQ(ierr);
412397b6df1SKris Buschelman       if (flg && icntl == 1) lu->id.sym=icntl;  /* matrix is spd */
413397b6df1SKris Buschelman     }
414397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
415397b6df1SKris Buschelman   zmumps_c(&lu->id);
416397b6df1SKris Buschelman #else
417397b6df1SKris Buschelman   dmumps_c(&lu->id);
418397b6df1SKris Buschelman #endif
419397b6df1SKris Buschelman 
420397b6df1SKris Buschelman     if (lu->size == 1){
421397b6df1SKris Buschelman       lu->id.ICNTL(18) = 0;   /* centralized assembled matrix input */
422397b6df1SKris Buschelman     } else {
423397b6df1SKris Buschelman       lu->id.ICNTL(18) = 3;   /* distributed assembled matrix input */
424397b6df1SKris Buschelman     }
425397b6df1SKris Buschelman 
426397b6df1SKris Buschelman     icntl=-1;
427397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_4","ICNTL(4): level of printing (0 to 4)","None",lu->id.ICNTL(4),&icntl,&flg);CHKERRQ(ierr);
428397b6df1SKris Buschelman     if (flg && icntl > 0) {
429397b6df1SKris Buschelman       lu->id.ICNTL(4)=icntl; /* and use mumps default icntl(i), i=1,2,3 */
430397b6df1SKris Buschelman     } else { /* no output */
431397b6df1SKris Buschelman       lu->id.ICNTL(1) = 0;  /* error message, default= 6 */
432397b6df1SKris Buschelman       lu->id.ICNTL(2) = -1; /* output stream for diagnostic printing, statistics, and warning. default=0 */
433397b6df1SKris Buschelman       lu->id.ICNTL(3) = -1; /* output stream for global information, default=6 */
434397b6df1SKris Buschelman       lu->id.ICNTL(4) = 0;  /* level of printing, 0,1,2,3,4, default=2 */
435397b6df1SKris Buschelman     }
436397b6df1SKris 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);
437397b6df1SKris Buschelman     icntl=-1;
438397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_7","ICNTL(7): matrix ordering (0 to 7)","None",lu->id.ICNTL(7),&icntl,&flg);CHKERRQ(ierr);
439397b6df1SKris Buschelman     if (flg) {
440397b6df1SKris Buschelman       if (icntl== 1){
441397b6df1SKris Buschelman         SETERRQ(PETSC_ERR_SUP,"pivot order be set by the user in PERM_IN -- not supported by the PETSc/MUMPS interface\n");
442397b6df1SKris Buschelman       } else {
443397b6df1SKris Buschelman         lu->id.ICNTL(7) = icntl;
444397b6df1SKris Buschelman       }
445397b6df1SKris Buschelman     }
446397b6df1SKris 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);
447397b6df1SKris 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);
448397b6df1SKris 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);
449397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_12","ICNTL(12): efficiency control","None",lu->id.ICNTL(12),&lu->id.ICNTL(12),PETSC_NULL);CHKERRQ(ierr);
450397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_13","ICNTL(13): efficiency control","None",lu->id.ICNTL(13),&lu->id.ICNTL(13),PETSC_NULL);CHKERRQ(ierr);
451*adc1d99fSHong 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);
452397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_15","ICNTL(15): efficiency control","None",lu->id.ICNTL(15),&lu->id.ICNTL(15),PETSC_NULL);CHKERRQ(ierr);
453397b6df1SKris Buschelman 
454397b6df1SKris Buschelman     /*
455397b6df1SKris 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);
456397b6df1SKris Buschelman     if (flg){
457397b6df1SKris Buschelman       if (icntl >-1 && icntl <3 ){
458397b6df1SKris Buschelman         if (lu->myid==0) lu->id.ICNTL(16) = icntl;
459397b6df1SKris Buschelman       } else {
460397b6df1SKris Buschelman         SETERRQ1(PETSC_ERR_SUP,"ICNTL(16)=%d -- not supported\n",icntl);
461397b6df1SKris Buschelman       }
462397b6df1SKris Buschelman     }
463397b6df1SKris Buschelman     */
464397b6df1SKris Buschelman 
465397b6df1SKris 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);
466397b6df1SKris 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);
467397b6df1SKris 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);
468397b6df1SKris Buschelman     PetscOptionsEnd();
469397b6df1SKris Buschelman   }
470397b6df1SKris Buschelman 
471397b6df1SKris Buschelman   /* define matrix A */
472397b6df1SKris Buschelman   switch (lu->id.ICNTL(18)){
473397b6df1SKris Buschelman   case 0:  /* centralized assembled matrix input (size=1) */
474397b6df1SKris Buschelman     if (!lu->myid) {
475c36ead0aSKris Buschelman       if (lua->isAIJ){
476397b6df1SKris Buschelman         Mat_SeqAIJ   *aa = (Mat_SeqAIJ*)A->data;
477397b6df1SKris Buschelman         nz               = aa->nz;
478397b6df1SKris Buschelman         ai = aa->i; aj = aa->j; lu->val = aa->a;
479397b6df1SKris Buschelman       } else {
480397b6df1SKris Buschelman         Mat_SeqSBAIJ *aa = (Mat_SeqSBAIJ*)A->data;
481397b6df1SKris Buschelman         nz                  =  aa->s_nz;
482397b6df1SKris Buschelman         ai = aa->i; aj = aa->j; lu->val = aa->a;
483397b6df1SKris Buschelman       }
484397b6df1SKris Buschelman       if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ /* first numeric factorization, get irn and jcn */
485397b6df1SKris Buschelman         ierr = PetscMalloc(nz*sizeof(int),&lu->irn);CHKERRQ(ierr);
486397b6df1SKris Buschelman         ierr = PetscMalloc(nz*sizeof(int),&lu->jcn);CHKERRQ(ierr);
487397b6df1SKris Buschelman         nz = 0;
488397b6df1SKris Buschelman         for (i=0; i<M; i++){
489397b6df1SKris Buschelman           rnz = ai[i+1] - ai[i];
490397b6df1SKris Buschelman           while (rnz--) {  /* Fortran row/col index! */
491397b6df1SKris Buschelman             lu->irn[nz] = i+1; lu->jcn[nz] = (*aj)+1; aj++; nz++;
492397b6df1SKris Buschelman           }
493397b6df1SKris Buschelman         }
494397b6df1SKris Buschelman       }
495397b6df1SKris Buschelman     }
496397b6df1SKris Buschelman     break;
497397b6df1SKris Buschelman   case 3:  /* distributed assembled matrix input (size>1) */
498397b6df1SKris Buschelman     if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){
499397b6df1SKris Buschelman       valOnly = PETSC_FALSE;
500397b6df1SKris Buschelman     } else {
501397b6df1SKris Buschelman       valOnly = PETSC_TRUE; /* only update mat values, not row and col index */
502397b6df1SKris Buschelman     }
503397b6df1SKris Buschelman     ierr = MatConvertToTriples(A,1,valOnly, &nnz, &lu->irn, &lu->jcn, &lu->val);CHKERRQ(ierr);
504397b6df1SKris Buschelman     break;
505397b6df1SKris Buschelman   default: SETERRQ(PETSC_ERR_SUP,"Matrix input format is not supported by MUMPS.");
506397b6df1SKris Buschelman   }
507397b6df1SKris Buschelman 
508397b6df1SKris Buschelman   /* analysis phase */
509397b6df1SKris Buschelman   if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){
510397b6df1SKris Buschelman      lu->id.n = M;
511397b6df1SKris Buschelman     switch (lu->id.ICNTL(18)){
512397b6df1SKris Buschelman     case 0:  /* centralized assembled matrix input */
513397b6df1SKris Buschelman       if (!lu->myid) {
514397b6df1SKris Buschelman         lu->id.nz =nz; lu->id.irn=lu->irn; lu->id.jcn=lu->jcn;
515397b6df1SKris Buschelman         if (lu->id.ICNTL(6)>1){
516397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
517397b6df1SKris Buschelman           lu->id.a = (mumps_double_complex*)lu->val;
518397b6df1SKris Buschelman #else
519397b6df1SKris Buschelman           lu->id.a = lu->val;
520397b6df1SKris Buschelman #endif
521397b6df1SKris Buschelman         }
522397b6df1SKris Buschelman       }
523397b6df1SKris Buschelman       break;
524397b6df1SKris Buschelman     case 3:  /* distributed assembled matrix input (size>1) */
525397b6df1SKris Buschelman       lu->id.nz_loc = nnz;
526397b6df1SKris Buschelman       lu->id.irn_loc=lu->irn; lu->id.jcn_loc=lu->jcn;
527397b6df1SKris Buschelman       if (lu->id.ICNTL(6)>1) {
528397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
529397b6df1SKris Buschelman         lu->id.a_loc = (mumps_double_complex*)lu->val;
530397b6df1SKris Buschelman #else
531397b6df1SKris Buschelman         lu->id.a_loc = lu->val;
532397b6df1SKris Buschelman #endif
533397b6df1SKris Buschelman       }
534397b6df1SKris Buschelman       break;
535397b6df1SKris Buschelman     }
536397b6df1SKris Buschelman     lu->id.job=1;
537397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
538397b6df1SKris Buschelman   zmumps_c(&lu->id);
539397b6df1SKris Buschelman #else
540397b6df1SKris Buschelman   dmumps_c(&lu->id);
541397b6df1SKris Buschelman #endif
542397b6df1SKris Buschelman     if (lu->id.INFOG(1) < 0) {
543397b6df1SKris Buschelman       SETERRQ1(1,"Error reported by MUMPS in analysis phase: INFOG(1)=%d\n",lu->id.INFOG(1));
544397b6df1SKris Buschelman     }
545397b6df1SKris Buschelman   }
546397b6df1SKris Buschelman 
547397b6df1SKris Buschelman   /* numerical factorization phase */
548397b6df1SKris Buschelman   if(lu->id.ICNTL(18) == 0) {
549397b6df1SKris Buschelman     if (lu->myid == 0) {
550397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
551397b6df1SKris Buschelman       lu->id.a = (mumps_double_complex*)lu->val;
552397b6df1SKris Buschelman #else
553397b6df1SKris Buschelman       lu->id.a = lu->val;
554397b6df1SKris Buschelman #endif
555397b6df1SKris Buschelman     }
556397b6df1SKris Buschelman   } else {
557397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
558397b6df1SKris Buschelman     lu->id.a_loc = (mumps_double_complex*)lu->val;
559397b6df1SKris Buschelman #else
560397b6df1SKris Buschelman     lu->id.a_loc = lu->val;
561397b6df1SKris Buschelman #endif
562397b6df1SKris Buschelman   }
563397b6df1SKris Buschelman   lu->id.job=2;
564397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
565397b6df1SKris Buschelman   zmumps_c(&lu->id);
566397b6df1SKris Buschelman #else
567397b6df1SKris Buschelman   dmumps_c(&lu->id);
568397b6df1SKris Buschelman #endif
569397b6df1SKris Buschelman   if (lu->id.INFOG(1) < 0) {
570397b6df1SKris Buschelman     SETERRQ1(1,"1, Error reported by MUMPS in numerical factorization phase: INFOG(1)=%d\n",lu->id.INFOG(1));
571397b6df1SKris Buschelman   }
572397b6df1SKris Buschelman 
573397b6df1SKris Buschelman   if (lu->myid==0 && lu->id.ICNTL(16) > 0){
574397b6df1SKris Buschelman     SETERRQ1(1,"  lu->id.ICNTL(16):=%d\n",lu->id.INFOG(16));
575397b6df1SKris Buschelman   }
576397b6df1SKris Buschelman 
577397b6df1SKris Buschelman   (*F)->assembled  = PETSC_TRUE;
578397b6df1SKris Buschelman   lu->matstruc     = SAME_NONZERO_PATTERN;
579ace87b0dSHong Zhang   lu->CleanUpMUMPS = PETSC_TRUE;
580397b6df1SKris Buschelman   PetscFunctionReturn(0);
581397b6df1SKris Buschelman }
582397b6df1SKris Buschelman 
583397b6df1SKris Buschelman /* Note the Petsc r and c permutations are ignored */
584397b6df1SKris Buschelman #undef __FUNCT__
585f0c56d0fSKris Buschelman #define __FUNCT__ "MatLUFactorSymbolic_AIJMUMPS"
586f0c56d0fSKris Buschelman int MatLUFactorSymbolic_AIJMUMPS(Mat A,IS r,IS c,MatFactorInfo *info,Mat *F) {
587397b6df1SKris Buschelman   Mat       B;
588f0c56d0fSKris Buschelman   Mat_MUMPS *lu;
589397b6df1SKris Buschelman   int       ierr;
590397b6df1SKris Buschelman 
591397b6df1SKris Buschelman   PetscFunctionBegin;
592397b6df1SKris Buschelman 
593397b6df1SKris Buschelman   /* Create the factorization matrix */
594397b6df1SKris Buschelman   ierr = MatCreate(A->comm,A->m,A->n,A->M,A->N,&B);CHKERRQ(ierr);
595397b6df1SKris Buschelman   ierr = MatSetType(B,MATAIJMUMPS);CHKERRQ(ierr);
596397b6df1SKris Buschelman   ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr);
597397b6df1SKris Buschelman   ierr = MatMPIAIJSetPreallocation(B,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr);
598397b6df1SKris Buschelman 
599f0c56d0fSKris Buschelman   B->ops->lufactornumeric = MatFactorNumeric_AIJMUMPS;
600397b6df1SKris Buschelman   B->factor               = FACTOR_LU;
601f0c56d0fSKris Buschelman   lu                      = (Mat_MUMPS*)B->spptr;
602397b6df1SKris Buschelman   lu->sym                 = 0;
603397b6df1SKris Buschelman   lu->matstruc            = DIFFERENT_NONZERO_PATTERN;
604397b6df1SKris Buschelman 
605397b6df1SKris Buschelman   *F = B;
606397b6df1SKris Buschelman   PetscFunctionReturn(0);
607397b6df1SKris Buschelman }
608397b6df1SKris Buschelman 
609397b6df1SKris Buschelman /* Note the Petsc r permutation is ignored */
610397b6df1SKris Buschelman #undef __FUNCT__
611f0c56d0fSKris Buschelman #define __FUNCT__ "MatCholeskyFactorSymbolic_SBAIJMUMPS"
612f0c56d0fSKris Buschelman int MatCholeskyFactorSymbolic_SBAIJMUMPS(Mat A,IS r,MatFactorInfo *info,Mat *F) {
613397b6df1SKris Buschelman   Mat       B;
614f0c56d0fSKris Buschelman   Mat_MUMPS *lu;
615397b6df1SKris Buschelman   int       ierr;
616397b6df1SKris Buschelman 
617397b6df1SKris Buschelman   PetscFunctionBegin;
618397b6df1SKris Buschelman 
619397b6df1SKris Buschelman   /* Create the factorization matrix */
620397b6df1SKris Buschelman   ierr = MatCreate(A->comm,A->m,A->n,A->M,A->N,&B);CHKERRQ(ierr);
621397b6df1SKris Buschelman   ierr = MatSetType(B,MATAIJMUMPS);CHKERRQ(ierr);
622397b6df1SKris Buschelman   ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr);
623397b6df1SKris Buschelman   ierr = MatMPIAIJSetPreallocation(B,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr);
624397b6df1SKris Buschelman 
625f0c56d0fSKris Buschelman   B->ops->choleskyfactornumeric = MatFactorNumeric_AIJMUMPS;
626397b6df1SKris Buschelman   B->factor                     = FACTOR_CHOLESKY;
627f0c56d0fSKris Buschelman   lu                            = (Mat_MUMPS*)B->spptr;
628397b6df1SKris Buschelman   lu->sym                       = 2;
629397b6df1SKris Buschelman   lu->matstruc                  = DIFFERENT_NONZERO_PATTERN;
630397b6df1SKris Buschelman 
631397b6df1SKris Buschelman   *F = B;
632397b6df1SKris Buschelman   PetscFunctionReturn(0);
633397b6df1SKris Buschelman }
634397b6df1SKris Buschelman 
635397b6df1SKris Buschelman #undef __FUNCT__
636f0c56d0fSKris Buschelman #define __FUNCT__ "MatAssemblyEnd_AIJMUMPS"
637f0c56d0fSKris Buschelman int MatAssemblyEnd_AIJMUMPS(Mat A,MatAssemblyType mode) {
638c338a77dSKris Buschelman   int       ierr;
639f0c56d0fSKris Buschelman   Mat_MUMPS *mumps=(Mat_MUMPS*)A->spptr;
640c338a77dSKris Buschelman 
641397b6df1SKris Buschelman   PetscFunctionBegin;
642c338a77dSKris Buschelman   ierr = (*mumps->MatAssemblyEnd)(A,mode);CHKERRQ(ierr);
643f0c56d0fSKris Buschelman 
644c338a77dSKris Buschelman   mumps->MatLUFactorSymbolic       = A->ops->lufactorsymbolic;
645c338a77dSKris Buschelman   mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic;
646f0c56d0fSKris Buschelman   A->ops->lufactorsymbolic         = MatLUFactorSymbolic_AIJMUMPS;
647397b6df1SKris Buschelman   PetscFunctionReturn(0);
648397b6df1SKris Buschelman }
649397b6df1SKris Buschelman 
650c338a77dSKris Buschelman EXTERN_C_BEGIN
651c338a77dSKris Buschelman #undef __FUNCT__
652f0c56d0fSKris Buschelman #define __FUNCT__ "MatConvert_AIJ_AIJMUMPS"
653f0c56d0fSKris Buschelman int MatConvert_AIJ_AIJMUMPS(Mat A,MatType newtype,Mat *newmat) {
654c338a77dSKris Buschelman   int       ierr,size;
655c338a77dSKris Buschelman   MPI_Comm  comm;
656c338a77dSKris Buschelman   Mat       B=*newmat;
657f0c56d0fSKris Buschelman   Mat_MUMPS *mumps;
658397b6df1SKris Buschelman 
659397b6df1SKris Buschelman   PetscFunctionBegin;
660c338a77dSKris Buschelman   if (B != A) {
661c338a77dSKris Buschelman     ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr);
662397b6df1SKris Buschelman   }
663397b6df1SKris Buschelman 
664c338a77dSKris Buschelman   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
665f0c56d0fSKris Buschelman   ierr = PetscNew(Mat_MUMPS,&mumps);CHKERRQ(ierr);
666c338a77dSKris Buschelman 
667f0c56d0fSKris Buschelman   mumps->MatDuplicate              = A->ops->duplicate;
668c338a77dSKris Buschelman   mumps->MatView                   = A->ops->view;
669c338a77dSKris Buschelman   mumps->MatAssemblyEnd            = A->ops->assemblyend;
670c338a77dSKris Buschelman   mumps->MatLUFactorSymbolic       = A->ops->lufactorsymbolic;
671c338a77dSKris Buschelman   mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic;
672c338a77dSKris Buschelman   mumps->MatDestroy                = A->ops->destroy;
673a39386dcSKris Buschelman   mumps->specialdestroy            = MatDestroy_AIJMUMPS;
674c338a77dSKris Buschelman   mumps->CleanUpMUMPS              = PETSC_FALSE;
675f579278aSKris Buschelman   mumps->isAIJ                     = PETSC_TRUE;
676c338a77dSKris Buschelman 
6774b68dd72SKris Buschelman   B->spptr                         = (void *)mumps;
678f0c56d0fSKris Buschelman   B->ops->duplicate                = MatDuplicate_AIJMUMPS;
679f0c56d0fSKris Buschelman   B->ops->view                     = MatView_AIJMUMPS;
680f0c56d0fSKris Buschelman   B->ops->assemblyend              = MatAssemblyEnd_AIJMUMPS;
681f0c56d0fSKris Buschelman   B->ops->lufactorsymbolic         = MatLUFactorSymbolic_AIJMUMPS;
6823924e44cSKris Buschelman   B->ops->destroy                  = MatDestroy_MUMPS;
683c338a77dSKris Buschelman 
684c338a77dSKris Buschelman   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr);
685c338a77dSKris Buschelman   if (size == 1) {
686c338a77dSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqaij_aijmumps_C",
687f0c56d0fSKris Buschelman                                              "MatConvert_AIJ_AIJMUMPS",MatConvert_AIJ_AIJMUMPS);CHKERRQ(ierr);
688c338a77dSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_aijmumps_seqaij_C",
689c338a77dSKris Buschelman                                              "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr);
690c338a77dSKris Buschelman   } else {
691c338a77dSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_mpiaij_aijmumps_C",
692f0c56d0fSKris Buschelman                                              "MatConvert_AIJ_AIJMUMPS",MatConvert_AIJ_AIJMUMPS);CHKERRQ(ierr);
693c338a77dSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_aijmumps_mpiaij_C",
694c338a77dSKris Buschelman                                              "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr);
695c338a77dSKris Buschelman   }
696c338a77dSKris Buschelman 
697f579278aSKris Buschelman   PetscLogInfo(0,"Using MUMPS for LU factorization and solves.");
698c338a77dSKris Buschelman   ierr = PetscObjectChangeTypeName((PetscObject)B,newtype);CHKERRQ(ierr);
699c338a77dSKris Buschelman   *newmat = B;
700397b6df1SKris Buschelman   PetscFunctionReturn(0);
701397b6df1SKris Buschelman }
702c338a77dSKris Buschelman EXTERN_C_END
703397b6df1SKris Buschelman 
704f0c56d0fSKris Buschelman #undef __FUNCT__
705f0c56d0fSKris Buschelman #define __FUNCT__ "MatDuplicate_AIJMUMPS"
706f0c56d0fSKris Buschelman int MatDuplicate_AIJMUMPS(Mat A, MatDuplicateOption op, Mat *M) {
707f0c56d0fSKris Buschelman   int       ierr;
7088f340917SKris Buschelman   Mat_MUMPS *lu=(Mat_MUMPS *)A->spptr;
7098f340917SKris Buschelman 
710f0c56d0fSKris Buschelman   PetscFunctionBegin;
7118f340917SKris Buschelman   ierr = (*lu->MatDuplicate)(A,op,M);CHKERRQ(ierr);
712f0c56d0fSKris Buschelman   ierr = MatConvert_AIJ_AIJMUMPS(*M,MATAIJMUMPS,M);CHKERRQ(ierr);
713a39386dcSKris Buschelman   ierr = PetscMemcpy((*M)->spptr,lu,sizeof(Mat_MUMPS));CHKERRQ(ierr);
714f0c56d0fSKris Buschelman   PetscFunctionReturn(0);
715f0c56d0fSKris Buschelman }
716f0c56d0fSKris Buschelman 
71724b6179bSKris Buschelman /*MC
718fafad747SKris Buschelman   MATAIJMUMPS - MATAIJMUMPS = "aijmumps" - A matrix type providing direct solvers (LU) for distributed
71924b6179bSKris Buschelman   and sequential matrices via the external package MUMPS.
72024b6179bSKris Buschelman 
72124b6179bSKris Buschelman   If MUMPS is installed (see the manual for instructions
72224b6179bSKris Buschelman   on how to declare the existence of external packages),
72324b6179bSKris Buschelman   a matrix type can be constructed which invokes MUMPS solvers.
72424b6179bSKris Buschelman   After calling MatCreate(...,A), simply call MatSetType(A,MATAIJMUMPS).
72524b6179bSKris Buschelman   This matrix type is only supported for double precision real.
72624b6179bSKris Buschelman 
72724b6179bSKris Buschelman   If created with a single process communicator, this matrix type inherits from MATSEQAIJ.
72824b6179bSKris Buschelman   Otherwise, this matrix type inherits from MATMPIAIJ.  Hence for single process communicators,
72924b6179bSKris Buschelman   MatSeqAIJSetPreallocation is supported, and similarly MatMPIAIJSetPreallocation is supported
73024b6179bSKris Buschelman   for communicators controlling multiple processes.  It is recommended that you call both of
73128b08bd3SKris Buschelman   the above preallocation routines for simplicity.  One can also call MatConvert for an inplace
73228b08bd3SKris Buschelman   conversion to or from the MATSEQAIJ or MATMPIAIJ type (depending on the communicator size)
73328b08bd3SKris Buschelman   without data copy.
73424b6179bSKris Buschelman 
73524b6179bSKris Buschelman   Options Database Keys:
7360bad9183SKris Buschelman + -mat_type aijmumps - sets the matrix type to "aijmumps" during a call to MatSetFromOptions()
73724b6179bSKris Buschelman . -mat_mumps_sym <0,1,2> - 0 the matrix is unsymmetric, 1 symmetric positive definite, 2 symmetric
73824b6179bSKris Buschelman . -mat_mumps_icntl_4 <0,1,2,3,4> - print level
73924b6179bSKris Buschelman . -mat_mumps_icntl_6 <0,...,7> - matrix prescaling options (see MUMPS User's Guide)
74024b6179bSKris Buschelman . -mat_mumps_icntl_7 <0,...,7> - matrix orderings (see MUMPS User's Guide)
74124b6179bSKris Buschelman . -mat_mumps_icntl_9 <1,2> - A or A^T x=b to be solved: 1 denotes A, 2 denotes A^T
74224b6179bSKris Buschelman . -mat_mumps_icntl_10 <n> - maximum number of iterative refinements
74324b6179bSKris Buschelman . -mat_mumps_icntl_11 <n> - error analysis, a positive value returns statistics during -sles_view
74424b6179bSKris Buschelman . -mat_mumps_icntl_12 <n> - efficiency control (see MUMPS User's Guide)
74524b6179bSKris Buschelman . -mat_mumps_icntl_13 <n> - efficiency control (see MUMPS User's Guide)
74624b6179bSKris Buschelman . -mat_mumps_icntl_14 <n> - efficiency control (see MUMPS User's Guide)
74724b6179bSKris Buschelman . -mat_mumps_icntl_15 <n> - efficiency control (see MUMPS User's Guide)
74824b6179bSKris Buschelman . -mat_mumps_cntl_1 <delta> - relative pivoting threshold
74924b6179bSKris Buschelman . -mat_mumps_cntl_2 <tol> - stopping criterion for refinement
75024b6179bSKris Buschelman - -mat_mumps_cntl_3 <adelta> - absolute pivoting threshold
75124b6179bSKris Buschelman 
75224b6179bSKris Buschelman   Level: beginner
75324b6179bSKris Buschelman 
75424b6179bSKris Buschelman .seealso: MATSBAIJMUMPS
75524b6179bSKris Buschelman M*/
75624b6179bSKris Buschelman 
757397b6df1SKris Buschelman EXTERN_C_BEGIN
758397b6df1SKris Buschelman #undef __FUNCT__
759f0c56d0fSKris Buschelman #define __FUNCT__ "MatCreate_AIJMUMPS"
760f0c56d0fSKris Buschelman int MatCreate_AIJMUMPS(Mat A) {
761397b6df1SKris Buschelman   int           ierr,size;
762397b6df1SKris Buschelman   MPI_Comm      comm;
763397b6df1SKris Buschelman 
764397b6df1SKris Buschelman   PetscFunctionBegin;
7655441df8eSKris Buschelman   /* Change type name before calling MatSetType to force proper construction of SeqAIJ or MPIAIJ */
7665441df8eSKris Buschelman   /*   and AIJMUMPS types */
7675441df8eSKris Buschelman   ierr = PetscObjectChangeTypeName((PetscObject)A,MATAIJMUMPS);CHKERRQ(ierr);
768397b6df1SKris Buschelman   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
769397b6df1SKris Buschelman   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr);
770397b6df1SKris Buschelman   if (size == 1) {
771397b6df1SKris Buschelman     ierr = MatSetType(A,MATSEQAIJ);CHKERRQ(ierr);
772397b6df1SKris Buschelman   } else {
773397b6df1SKris Buschelman     ierr = MatSetType(A,MATMPIAIJ);CHKERRQ(ierr);
774397b6df1SKris Buschelman   }
775f0c56d0fSKris Buschelman   ierr = MatConvert_AIJ_AIJMUMPS(A,MATAIJMUMPS,&A);CHKERRQ(ierr);
776397b6df1SKris Buschelman   PetscFunctionReturn(0);
777397b6df1SKris Buschelman }
778397b6df1SKris Buschelman EXTERN_C_END
779397b6df1SKris Buschelman 
780f579278aSKris Buschelman #undef __FUNCT__
781f0c56d0fSKris Buschelman #define __FUNCT__ "MatAssemblyEnd_SBAIJMUMPS"
782f0c56d0fSKris Buschelman int MatAssemblyEnd_SBAIJMUMPS(Mat A,MatAssemblyType mode) {
783f579278aSKris Buschelman   int       ierr;
784f0c56d0fSKris Buschelman   Mat_MUMPS *mumps=(Mat_MUMPS*)A->spptr;
785f579278aSKris Buschelman 
786f579278aSKris Buschelman   PetscFunctionBegin;
787f579278aSKris Buschelman   ierr = (*mumps->MatAssemblyEnd)(A,mode);CHKERRQ(ierr);
788f579278aSKris Buschelman   mumps->MatLUFactorSymbolic       = A->ops->lufactorsymbolic;
789f579278aSKris Buschelman   mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic;
790f0c56d0fSKris Buschelman   A->ops->choleskyfactorsymbolic   = MatCholeskyFactorSymbolic_SBAIJMUMPS;
791f579278aSKris Buschelman   PetscFunctionReturn(0);
792f579278aSKris Buschelman }
793f579278aSKris Buschelman 
794f579278aSKris Buschelman EXTERN_C_BEGIN
795f579278aSKris Buschelman #undef __FUNCT__
796f0c56d0fSKris Buschelman #define __FUNCT__ "MatConvert_SBAIJ_SBAIJMUMPS"
797f0c56d0fSKris Buschelman int MatConvert_SBAIJ_SBAIJMUMPS(Mat A,MatType newtype,Mat *newmat) {
798f579278aSKris Buschelman   int       ierr,size;
799f579278aSKris Buschelman   MPI_Comm  comm;
800f579278aSKris Buschelman   Mat       B=*newmat;
801f0c56d0fSKris Buschelman   Mat_MUMPS *mumps;
802f579278aSKris Buschelman 
803f579278aSKris Buschelman   PetscFunctionBegin;
804f579278aSKris Buschelman   if (B != A) {
805f579278aSKris Buschelman     ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr);
806f579278aSKris Buschelman   }
807f579278aSKris Buschelman 
808f579278aSKris Buschelman   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
809f0c56d0fSKris Buschelman   ierr = PetscNew(Mat_MUMPS,&mumps);CHKERRQ(ierr);
810f579278aSKris Buschelman 
811f0c56d0fSKris Buschelman   mumps->MatDuplicate              = A->ops->duplicate;
812f579278aSKris Buschelman   mumps->MatView                   = A->ops->view;
813f579278aSKris Buschelman   mumps->MatAssemblyEnd            = A->ops->assemblyend;
814f579278aSKris Buschelman   mumps->MatLUFactorSymbolic       = A->ops->lufactorsymbolic;
815f579278aSKris Buschelman   mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic;
816f579278aSKris Buschelman   mumps->MatDestroy                = A->ops->destroy;
817a39386dcSKris Buschelman   mumps->specialdestroy            = MatDestroy_SBAIJMUMPS;
818f579278aSKris Buschelman   mumps->CleanUpMUMPS              = PETSC_FALSE;
819f579278aSKris Buschelman   mumps->isAIJ                     = PETSC_FALSE;
820f579278aSKris Buschelman 
821f579278aSKris Buschelman   B->spptr                         = (void *)mumps;
822f0c56d0fSKris Buschelman   B->ops->duplicate                = MatDuplicate_SBAIJMUMPS;
823f0c56d0fSKris Buschelman   B->ops->view                     = MatView_AIJMUMPS;
824f0c56d0fSKris Buschelman   B->ops->assemblyend              = MatAssemblyEnd_SBAIJMUMPS;
825f0c56d0fSKris Buschelman   B->ops->choleskyfactorsymbolic   = MatCholeskyFactorSymbolic_SBAIJMUMPS;
8263924e44cSKris Buschelman   B->ops->destroy                  = MatDestroy_MUMPS;
827f579278aSKris Buschelman 
828f579278aSKris Buschelman   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr);
829f579278aSKris Buschelman   if (size == 1) {
830f0c56d0fSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_sbaijmumps_C",
831f0c56d0fSKris Buschelman                                              "MatConvert_SBAIJ_SBAIJMUMPS",MatConvert_SBAIJ_SBAIJMUMPS);CHKERRQ(ierr);
832f0c56d0fSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_sbaijmumps_seqsbaij_C",
833f579278aSKris Buschelman                                              "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr);
834f579278aSKris Buschelman   } else {
835f0c56d0fSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_mpisbaij_sbaijmumps_C",
836f0c56d0fSKris Buschelman                                              "MatConvert_SBAIJ_SBAIJMUMPS",MatConvert_SBAIJ_SBAIJMUMPS);CHKERRQ(ierr);
837f0c56d0fSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_sbaijmumps_mpisbaij_C",
838f579278aSKris Buschelman                                              "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr);
839f579278aSKris Buschelman   }
840f579278aSKris Buschelman 
841f579278aSKris Buschelman   PetscLogInfo(0,"Using MUMPS for Cholesky factorization and solves.");
842f579278aSKris Buschelman   ierr = PetscObjectChangeTypeName((PetscObject)B,newtype);CHKERRQ(ierr);
843f579278aSKris Buschelman   *newmat = B;
844f579278aSKris Buschelman   PetscFunctionReturn(0);
845f579278aSKris Buschelman }
846f579278aSKris Buschelman EXTERN_C_END
847f579278aSKris Buschelman 
848f0c56d0fSKris Buschelman #undef __FUNCT__
849f0c56d0fSKris Buschelman #define __FUNCT__ "MatDuplicate_SBAIJMUMPS"
850f0c56d0fSKris Buschelman int MatDuplicate_SBAIJMUMPS(Mat A, MatDuplicateOption op, Mat *M) {
851f0c56d0fSKris Buschelman   int       ierr;
8528f340917SKris Buschelman   Mat_MUMPS *lu=(Mat_MUMPS *)A->spptr;
8538f340917SKris Buschelman 
854f0c56d0fSKris Buschelman   PetscFunctionBegin;
8558f340917SKris Buschelman   ierr = (*lu->MatDuplicate)(A,op,M);CHKERRQ(ierr);
856f0c56d0fSKris Buschelman   ierr = MatConvert_SBAIJ_SBAIJMUMPS(*M,MATSBAIJMUMPS,M);CHKERRQ(ierr);
8573f953163SKris Buschelman   ierr = PetscMemcpy((*M)->spptr,lu,sizeof(Mat_MUMPS));CHKERRQ(ierr);
858f0c56d0fSKris Buschelman   PetscFunctionReturn(0);
859f0c56d0fSKris Buschelman }
860f0c56d0fSKris Buschelman 
86124b6179bSKris Buschelman /*MC
862fafad747SKris Buschelman   MATSBAIJMUMPS - MATSBAIJMUMPS = "sbaijmumps" - A symmetric matrix type providing direct solvers (Cholesky) for
86324b6179bSKris Buschelman   distributed and sequential matrices via the external package MUMPS.
86424b6179bSKris Buschelman 
86524b6179bSKris Buschelman   If MUMPS is installed (see the manual for instructions
86624b6179bSKris Buschelman   on how to declare the existence of external packages),
86724b6179bSKris Buschelman   a matrix type can be constructed which invokes MUMPS solvers.
86824b6179bSKris Buschelman   After calling MatCreate(...,A), simply call MatSetType(A,MATSBAIJMUMPS).
86924b6179bSKris Buschelman   This matrix type is only supported for double precision real.
87024b6179bSKris Buschelman 
87124b6179bSKris Buschelman   If created with a single process communicator, this matrix type inherits from MATSEQSBAIJ.
87224b6179bSKris Buschelman   Otherwise, this matrix type inherits from MATMPISBAIJ.  Hence for single process communicators,
87324b6179bSKris Buschelman   MatSeqSBAIJSetPreallocation is supported, and similarly MatMPISBAIJSetPreallocation is supported
87424b6179bSKris Buschelman   for communicators controlling multiple processes.  It is recommended that you call both of
87528b08bd3SKris Buschelman   the above preallocation routines for simplicity.  One can also call MatConvert for an inplace
87628b08bd3SKris Buschelman   conversion to or from the MATSEQSBAIJ or MATMPISBAIJ type (depending on the communicator size)
87728b08bd3SKris Buschelman   without data copy.
87824b6179bSKris Buschelman 
87924b6179bSKris Buschelman   Options Database Keys:
8800bad9183SKris Buschelman + -mat_type sbaijmumps - sets the matrix type to "sbaijmumps" during a call to MatSetFromOptions()
88124b6179bSKris Buschelman . -mat_mumps_sym <0,1,2> - 0 the matrix is unsymmetric, 1 symmetric positive definite, 2 symmetric
88224b6179bSKris Buschelman . -mat_mumps_icntl_4 <0,...,4> - print level
88324b6179bSKris Buschelman . -mat_mumps_icntl_6 <0,...,7> - matrix prescaling options (see MUMPS User's Guide)
88424b6179bSKris Buschelman . -mat_mumps_icntl_7 <0,...,7> - matrix orderings (see MUMPS User's Guide)
88524b6179bSKris Buschelman . -mat_mumps_icntl_9 <1,2> - A or A^T x=b to be solved: 1 denotes A, 2 denotes A^T
88624b6179bSKris Buschelman . -mat_mumps_icntl_10 <n> - maximum number of iterative refinements
88724b6179bSKris Buschelman . -mat_mumps_icntl_11 <n> - error analysis, a positive value returns statistics during -sles_view
88824b6179bSKris Buschelman . -mat_mumps_icntl_12 <n> - efficiency control (see MUMPS User's Guide)
88924b6179bSKris Buschelman . -mat_mumps_icntl_13 <n> - efficiency control (see MUMPS User's Guide)
89024b6179bSKris Buschelman . -mat_mumps_icntl_14 <n> - efficiency control (see MUMPS User's Guide)
89124b6179bSKris Buschelman . -mat_mumps_icntl_15 <n> - efficiency control (see MUMPS User's Guide)
89224b6179bSKris Buschelman . -mat_mumps_cntl_1 <delta> - relative pivoting threshold
89324b6179bSKris Buschelman . -mat_mumps_cntl_2 <tol> - stopping criterion for refinement
89424b6179bSKris Buschelman - -mat_mumps_cntl_3 <adelta> - absolute pivoting threshold
89524b6179bSKris Buschelman 
89624b6179bSKris Buschelman   Level: beginner
89724b6179bSKris Buschelman 
89824b6179bSKris Buschelman .seealso: MATAIJMUMPS
89924b6179bSKris Buschelman M*/
90024b6179bSKris Buschelman 
901397b6df1SKris Buschelman EXTERN_C_BEGIN
902397b6df1SKris Buschelman #undef __FUNCT__
903f0c56d0fSKris Buschelman #define __FUNCT__ "MatCreate_SBAIJMUMPS"
904f0c56d0fSKris Buschelman int MatCreate_SBAIJMUMPS(Mat A) {
905397b6df1SKris Buschelman   int ierr,size;
906397b6df1SKris Buschelman 
907397b6df1SKris Buschelman   PetscFunctionBegin;
9085441df8eSKris Buschelman   /* Change type name before calling MatSetType to force proper construction of SeqSBAIJ or MPISBAIJ */
9095441df8eSKris Buschelman   /*   and SBAIJMUMPS types */
9105441df8eSKris Buschelman   ierr = PetscObjectChangeTypeName((PetscObject)A,MATSBAIJMUMPS);CHKERRQ(ierr);
9115441df8eSKris Buschelman   ierr = MPI_Comm_size(A->comm,&size);CHKERRQ(ierr);CHKERRQ(ierr);
912397b6df1SKris Buschelman   if (size == 1) {
913397b6df1SKris Buschelman     ierr = MatSetType(A,MATSEQSBAIJ);CHKERRQ(ierr);
914397b6df1SKris Buschelman   } else {
915397b6df1SKris Buschelman     ierr = MatSetType(A,MATMPISBAIJ);CHKERRQ(ierr);
916397b6df1SKris Buschelman   }
917f0c56d0fSKris Buschelman   ierr = MatConvert_SBAIJ_SBAIJMUMPS(A,MATSBAIJMUMPS,&A);CHKERRQ(ierr);
918397b6df1SKris Buschelman   PetscFunctionReturn(0);
919397b6df1SKris Buschelman }
920397b6df1SKris Buschelman EXTERN_C_END
921