xref: /petsc/src/mat/impls/aij/mpi/mumps/mumps.c (revision a58c3f205c055707cd7d4adbbe2d866c6911ae59)
1397b6df1SKris Buschelman /*$Id: mumps.c,v 1.10 2001/08/15 15:56:50 bsmith Exp $*/
2397b6df1SKris Buschelman /*
3*a58c3f20SHong Zhang     Provides an interface to the MUMPS_4.3.1 sparse solver
4397b6df1SKris Buschelman */
5397b6df1SKris Buschelman #include "src/mat/impls/aij/seq/aij.h"
6397b6df1SKris Buschelman #include "src/mat/impls/aij/mpi/mpiaij.h"
7397b6df1SKris Buschelman #include "src/mat/impls/sbaij/seq/sbaij.h"
8397b6df1SKris Buschelman #include "src/mat/impls/sbaij/mpi/mpisbaij.h"
9397b6df1SKris Buschelman 
10397b6df1SKris Buschelman EXTERN_C_BEGIN
11397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
12397b6df1SKris Buschelman #include "zmumps_c.h"
13397b6df1SKris Buschelman #else
14397b6df1SKris Buschelman #include "dmumps_c.h"
15397b6df1SKris Buschelman #endif
16397b6df1SKris Buschelman EXTERN_C_END
17397b6df1SKris Buschelman #define JOB_INIT -1
18397b6df1SKris Buschelman #define JOB_END -2
19397b6df1SKris Buschelman /* macros s.t. indices match MUMPS documentation */
20397b6df1SKris Buschelman #define ICNTL(I) icntl[(I)-1]
21397b6df1SKris Buschelman #define CNTL(I) cntl[(I)-1]
22397b6df1SKris Buschelman #define INFOG(I) infog[(I)-1]
23397b6df1SKris Buschelman #define RINFOG(I) rinfog[(I)-1]
24adc1d99fSHong Zhang #define RINFO(I) rinfo[(I)-1]
25397b6df1SKris Buschelman 
26397b6df1SKris Buschelman typedef struct {
27397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
28397b6df1SKris Buschelman   ZMUMPS_STRUC_C id;
29397b6df1SKris Buschelman #else
30397b6df1SKris Buschelman   DMUMPS_STRUC_C id;
31397b6df1SKris Buschelman #endif
32397b6df1SKris Buschelman   MatStructure   matstruc;
33397b6df1SKris Buschelman   int            myid,size,*irn,*jcn,sym;
34397b6df1SKris Buschelman   PetscScalar    *val;
35397b6df1SKris Buschelman   MPI_Comm       comm_mumps;
36397b6df1SKris Buschelman 
37c338a77dSKris Buschelman   PetscTruth     isAIJ,CleanUpMUMPS;
38f0c56d0fSKris Buschelman   int (*MatDuplicate)(Mat,MatDuplicateOption,Mat*);
39c338a77dSKris Buschelman   int (*MatView)(Mat,PetscViewer);
40c338a77dSKris Buschelman   int (*MatAssemblyEnd)(Mat,MatAssemblyType);
41c338a77dSKris Buschelman   int (*MatLUFactorSymbolic)(Mat,IS,IS,MatFactorInfo*,Mat*);
42c338a77dSKris Buschelman   int (*MatCholeskyFactorSymbolic)(Mat,IS,MatFactorInfo*,Mat*);
43c338a77dSKris Buschelman   int (*MatDestroy)(Mat);
44a39386dcSKris Buschelman   int (*specialdestroy)(Mat);
45f0c56d0fSKris Buschelman } Mat_MUMPS;
46f0c56d0fSKris Buschelman 
47f0c56d0fSKris Buschelman EXTERN int MatDuplicate_AIJMUMPS(Mat,MatDuplicateOption,Mat*);
48f0c56d0fSKris Buschelman EXTERN int MatDuplicate_SBAIJMUMPS(Mat,MatDuplicateOption,Mat*);
490e3434eeSKris Buschelman 
50397b6df1SKris Buschelman /* convert Petsc mpiaij matrix to triples: row[nz], col[nz], val[nz] */
51397b6df1SKris Buschelman /*
52397b6df1SKris Buschelman   input:
5375747be1SHong Zhang     A       - matrix in mpiaij or mpisbaij (bs=1) format
54397b6df1SKris Buschelman     shift   - 0: C style output triple; 1: Fortran style output triple.
55397b6df1SKris Buschelman     valOnly - FALSE: spaces are allocated and values are set for the triple
56397b6df1SKris Buschelman               TRUE:  only the values in v array are updated
57397b6df1SKris Buschelman   output:
58397b6df1SKris Buschelman     nnz     - dim of r, c, and v (number of local nonzero entries of A)
59397b6df1SKris Buschelman     r, c, v - row and col index, matrix values (matrix triples)
60397b6df1SKris Buschelman  */
61f0c56d0fSKris Buschelman int MatConvertToTriples(Mat A,int shift,PetscTruth valOnly,int *nnz,int **r, int **c, PetscScalar **v) {
62397b6df1SKris Buschelman   int         *ai, *aj, *bi, *bj, rstart,nz, *garray;
63397b6df1SKris Buschelman   int         ierr,i,j,jj,jB,irow,m=A->m,*ajj,*bjj,countA,countB,colA_start,jcol;
64d54de34fSKris Buschelman   int         *row,*col;
65397b6df1SKris Buschelman   PetscScalar *av, *bv,*val;
66f0c56d0fSKris Buschelman   Mat_MUMPS   *mumps=(Mat_MUMPS*)A->spptr;
67397b6df1SKris Buschelman 
68397b6df1SKris Buschelman   PetscFunctionBegin;
69397b6df1SKris Buschelman   if (mumps->isAIJ){
70397b6df1SKris Buschelman     Mat_MPIAIJ    *mat =  (Mat_MPIAIJ*)A->data;
71397b6df1SKris Buschelman     Mat_SeqAIJ    *aa=(Mat_SeqAIJ*)(mat->A)->data;
72397b6df1SKris Buschelman     Mat_SeqAIJ    *bb=(Mat_SeqAIJ*)(mat->B)->data;
73397b6df1SKris Buschelman     nz = aa->nz + bb->nz;
74397b6df1SKris Buschelman     ai=aa->i; aj=aa->j; bi=bb->i; bj=bb->j; rstart= mat->rstart;
75397b6df1SKris Buschelman     garray = mat->garray;
76397b6df1SKris Buschelman     av=aa->a; bv=bb->a;
77397b6df1SKris Buschelman 
78397b6df1SKris Buschelman   } else {
79397b6df1SKris Buschelman     Mat_MPISBAIJ  *mat =  (Mat_MPISBAIJ*)A->data;
80397b6df1SKris Buschelman     Mat_SeqSBAIJ  *aa=(Mat_SeqSBAIJ*)(mat->A)->data;
81397b6df1SKris Buschelman     Mat_SeqBAIJ    *bb=(Mat_SeqBAIJ*)(mat->B)->data;
82847143adSKris Buschelman     if (mat->bs > 1) SETERRQ1(PETSC_ERR_SUP," bs=%d is not supported yet\n", mat->bs);
83397b6df1SKris Buschelman     nz = aa->s_nz + bb->nz;
84397b6df1SKris Buschelman     ai=aa->i; aj=aa->j; bi=bb->i; bj=bb->j; rstart= mat->rstart;
85397b6df1SKris Buschelman     garray = mat->garray;
86397b6df1SKris Buschelman     av=aa->a; bv=bb->a;
87397b6df1SKris Buschelman   }
88397b6df1SKris Buschelman 
89397b6df1SKris Buschelman   if (!valOnly){
90397b6df1SKris Buschelman     ierr = PetscMalloc(nz*sizeof(int),&row);CHKERRQ(ierr);
91397b6df1SKris Buschelman     ierr = PetscMalloc(nz*sizeof(int),&col);CHKERRQ(ierr);
92397b6df1SKris Buschelman     ierr = PetscMalloc(nz*sizeof(PetscScalar),&val);CHKERRQ(ierr);
93397b6df1SKris Buschelman     *r = row; *c = col; *v = val;
94397b6df1SKris Buschelman   } else {
95397b6df1SKris Buschelman     row = *r; col = *c; val = *v;
96397b6df1SKris Buschelman   }
97397b6df1SKris Buschelman   *nnz = nz;
98397b6df1SKris Buschelman 
99028e57e8SHong Zhang   jj = 0; irow = rstart;
100397b6df1SKris Buschelman   for ( i=0; i<m; i++ ) {
101397b6df1SKris Buschelman     ajj = aj + ai[i];                 /* ptr to the beginning of this row */
102397b6df1SKris Buschelman     countA = ai[i+1] - ai[i];
103397b6df1SKris Buschelman     countB = bi[i+1] - bi[i];
104397b6df1SKris Buschelman     bjj = bj + bi[i];
105397b6df1SKris Buschelman 
106397b6df1SKris Buschelman     /* get jB, the starting local col index for the 2nd B-part */
107397b6df1SKris Buschelman     colA_start = rstart + ajj[0]; /* the smallest col index for A */
10875747be1SHong Zhang     j=-1;
10975747be1SHong Zhang     do {
11075747be1SHong Zhang       j++;
11175747be1SHong Zhang       if (j == countB) break;
112397b6df1SKris Buschelman       jcol = garray[bjj[j]];
11375747be1SHong Zhang     } while (jcol < colA_start);
11475747be1SHong Zhang     jB = j;
115397b6df1SKris Buschelman 
116397b6df1SKris Buschelman     /* B-part, smaller col index */
117397b6df1SKris Buschelman     colA_start = rstart + ajj[0]; /* the smallest col index for A */
118397b6df1SKris Buschelman     for (j=0; j<jB; j++){
119397b6df1SKris Buschelman       jcol = garray[bjj[j]];
120397b6df1SKris Buschelman       if (!valOnly){
121397b6df1SKris Buschelman         row[jj] = irow + shift; col[jj] = jcol + shift;
12275747be1SHong Zhang 
123397b6df1SKris Buschelman       }
124397b6df1SKris Buschelman       val[jj++] = *bv++;
125397b6df1SKris Buschelman     }
126397b6df1SKris Buschelman     /* A-part */
127397b6df1SKris Buschelman     for (j=0; j<countA; j++){
128397b6df1SKris Buschelman       if (!valOnly){
129397b6df1SKris Buschelman         row[jj] = irow + shift; col[jj] = rstart + ajj[j] + shift;
130397b6df1SKris Buschelman       }
131397b6df1SKris Buschelman       val[jj++] = *av++;
132397b6df1SKris Buschelman     }
133397b6df1SKris Buschelman     /* B-part, larger col index */
134397b6df1SKris Buschelman     for (j=jB; j<countB; j++){
135397b6df1SKris Buschelman       if (!valOnly){
136397b6df1SKris Buschelman         row[jj] = irow + shift; col[jj] = garray[bjj[j]] + shift;
137397b6df1SKris Buschelman       }
138397b6df1SKris Buschelman       val[jj++] = *bv++;
139397b6df1SKris Buschelman     }
140397b6df1SKris Buschelman     irow++;
141397b6df1SKris Buschelman   }
142397b6df1SKris Buschelman 
143397b6df1SKris Buschelman   PetscFunctionReturn(0);
144397b6df1SKris Buschelman }
145397b6df1SKris Buschelman 
146c338a77dSKris Buschelman EXTERN_C_BEGIN
147c338a77dSKris Buschelman #undef __FUNCT__
148c338a77dSKris Buschelman #define __FUNCT__ "MatConvert_MUMPS_Base"
149c338a77dSKris Buschelman int MatConvert_MUMPS_Base(Mat A,MatType type,Mat *newmat) {
150c338a77dSKris Buschelman   int       ierr;
151c338a77dSKris Buschelman   Mat       B=*newmat;
152f0c56d0fSKris Buschelman   Mat_MUMPS *mumps=(Mat_MUMPS*)A->spptr;
153c338a77dSKris Buschelman 
154c338a77dSKris Buschelman   PetscFunctionBegin;
155c338a77dSKris Buschelman   if (B != A) {
156c338a77dSKris Buschelman     ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr);
157c338a77dSKris Buschelman   }
158f0c56d0fSKris Buschelman   B->ops->duplicate              = mumps->MatDuplicate;
159f0c56d0fSKris Buschelman   B->ops->view                   = mumps->MatView;
160f0c56d0fSKris Buschelman   B->ops->assemblyend            = mumps->MatAssemblyEnd;
161f0c56d0fSKris Buschelman   B->ops->lufactorsymbolic       = mumps->MatLUFactorSymbolic;
162f0c56d0fSKris Buschelman   B->ops->choleskyfactorsymbolic = mumps->MatCholeskyFactorSymbolic;
163f0c56d0fSKris Buschelman   B->ops->destroy                = mumps->MatDestroy;
1643924e44cSKris Buschelman   ierr = PetscObjectChangeTypeName((PetscObject)B,type);CHKERRQ(ierr);
165f0c56d0fSKris Buschelman   ierr = PetscFree(mumps);CHKERRQ(ierr);
166c338a77dSKris Buschelman   *newmat = B;
167c338a77dSKris Buschelman   PetscFunctionReturn(0);
168c338a77dSKris Buschelman }
169c338a77dSKris Buschelman EXTERN_C_END
170c338a77dSKris Buschelman 
171397b6df1SKris Buschelman #undef __FUNCT__
1723924e44cSKris Buschelman #define __FUNCT__ "MatDestroy_MUMPS"
1733924e44cSKris Buschelman int MatDestroy_MUMPS(Mat A) {
174f0c56d0fSKris Buschelman   Mat_MUMPS *lu=(Mat_MUMPS*)A->spptr;
175c338a77dSKris Buschelman   int       ierr,size=lu->size;
176a39386dcSKris Buschelman   int       (*specialdestroy)(Mat);
177397b6df1SKris Buschelman   PetscFunctionBegin;
178397b6df1SKris Buschelman   if (lu->CleanUpMUMPS) {
179397b6df1SKris Buschelman     /* Terminate instance, deallocate memories */
180397b6df1SKris Buschelman     lu->id.job=JOB_END;
181397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
182397b6df1SKris Buschelman     zmumps_c(&lu->id);
183397b6df1SKris Buschelman #else
184397b6df1SKris Buschelman     dmumps_c(&lu->id);
185397b6df1SKris Buschelman #endif
186c338a77dSKris Buschelman     if (lu->irn) {
187c338a77dSKris Buschelman       ierr = PetscFree(lu->irn);CHKERRQ(ierr);
188c338a77dSKris Buschelman     }
189c338a77dSKris Buschelman     if (lu->jcn) {
190c338a77dSKris Buschelman       ierr = PetscFree(lu->jcn);CHKERRQ(ierr);
191c338a77dSKris Buschelman     }
192c338a77dSKris Buschelman     if (size>1 && lu->val) {
193c338a77dSKris Buschelman       ierr = PetscFree(lu->val);CHKERRQ(ierr);
194c338a77dSKris Buschelman     }
195397b6df1SKris Buschelman     ierr = MPI_Comm_free(&(lu->comm_mumps));CHKERRQ(ierr);
196397b6df1SKris Buschelman   }
197a39386dcSKris Buschelman   specialdestroy = lu->specialdestroy;
198a39386dcSKris Buschelman   ierr = (*specialdestroy)(A);CHKERRQ(ierr);
199c338a77dSKris Buschelman   ierr = (*A->ops->destroy)(A);CHKERRQ(ierr);
200397b6df1SKris Buschelman   PetscFunctionReturn(0);
201397b6df1SKris Buschelman }
202397b6df1SKris Buschelman 
203397b6df1SKris Buschelman #undef __FUNCT__
204a39386dcSKris Buschelman #define __FUNCT__ "MatDestroy_AIJMUMPS"
205a39386dcSKris Buschelman int MatDestroy_AIJMUMPS(Mat A) {
206a39386dcSKris Buschelman   int ierr, size;
207a39386dcSKris Buschelman 
208a39386dcSKris Buschelman   PetscFunctionBegin;
209a39386dcSKris Buschelman   ierr = MPI_Comm_size(A->comm,&size);CHKERRQ(ierr);
210a39386dcSKris Buschelman   if (size==1) {
211a39386dcSKris Buschelman     ierr = MatConvert_MUMPS_Base(A,MATSEQAIJ,&A);CHKERRQ(ierr);
212a39386dcSKris Buschelman   } else {
213a39386dcSKris Buschelman     ierr = MatConvert_MUMPS_Base(A,MATMPIAIJ,&A);CHKERRQ(ierr);
214a39386dcSKris Buschelman   }
215a39386dcSKris Buschelman   PetscFunctionReturn(0);
216a39386dcSKris Buschelman }
217a39386dcSKris Buschelman 
218a39386dcSKris Buschelman #undef __FUNCT__
219a39386dcSKris Buschelman #define __FUNCT__ "MatDestroy_SBAIJMUMPS"
220a39386dcSKris Buschelman int MatDestroy_SBAIJMUMPS(Mat A) {
221a39386dcSKris Buschelman   int ierr, size;
222a39386dcSKris Buschelman 
223a39386dcSKris Buschelman   PetscFunctionBegin;
224a39386dcSKris Buschelman   ierr = MPI_Comm_size(A->comm,&size);CHKERRQ(ierr);
225a39386dcSKris Buschelman   if (size==1) {
226a39386dcSKris Buschelman     ierr = MatConvert_MUMPS_Base(A,MATSEQSBAIJ,&A);CHKERRQ(ierr);
227a39386dcSKris Buschelman   } else {
228a39386dcSKris Buschelman     ierr = MatConvert_MUMPS_Base(A,MATMPISBAIJ,&A);CHKERRQ(ierr);
229a39386dcSKris Buschelman   }
230a39386dcSKris Buschelman   PetscFunctionReturn(0);
231a39386dcSKris Buschelman }
232a39386dcSKris Buschelman 
233a39386dcSKris Buschelman #undef __FUNCT__
234c338a77dSKris Buschelman #define __FUNCT__ "MatFactorInfo_MUMPS"
235f0c56d0fSKris Buschelman int MatFactorInfo_MUMPS(Mat A,PetscViewer viewer) {
236f0c56d0fSKris Buschelman   Mat_MUMPS *lu=(Mat_MUMPS*)A->spptr;
237397b6df1SKris Buschelman   int       ierr;
238397b6df1SKris Buschelman 
239397b6df1SKris Buschelman   PetscFunctionBegin;
240c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"MUMPS run parameters:\n");CHKERRQ(ierr);
241c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  SYM (matrix type):                  %d \n",lu->id.sym);CHKERRQ(ierr);
242c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  PAR (host participation):           %d \n",lu->id.par);CHKERRQ(ierr);
243c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(4) (level of printing):       %d \n",lu->id.ICNTL(4));CHKERRQ(ierr);
244c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(5) (input mat struct):        %d \n",lu->id.ICNTL(5));CHKERRQ(ierr);
245c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(6) (matrix prescaling):       %d \n",lu->id.ICNTL(6));CHKERRQ(ierr);
246c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(7) (matrix ordering):         %d \n",lu->id.ICNTL(7));CHKERRQ(ierr);
247c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(9) (A/A^T x=b is solved):     %d \n",lu->id.ICNTL(9));CHKERRQ(ierr);
248c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(10) (max num of refinements): %d \n",lu->id.ICNTL(10));CHKERRQ(ierr);
249c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(11) (error analysis):         %d \n",lu->id.ICNTL(11));CHKERRQ(ierr);
250c338a77dSKris Buschelman   if (lu->myid == 0 && lu->id.ICNTL(11)>0) {
251c338a77dSKris Buschelman     ierr = PetscPrintf(PETSC_COMM_SELF,"        RINFOG(4) (inf norm of input mat):        %g\n",lu->id.RINFOG(4));CHKERRQ(ierr);
252c338a77dSKris Buschelman     ierr = PetscPrintf(PETSC_COMM_SELF,"        RINFOG(5) (inf norm of solution):         %g\n",lu->id.RINFOG(5));CHKERRQ(ierr);
253c338a77dSKris Buschelman     ierr = PetscPrintf(PETSC_COMM_SELF,"        RINFOG(6) (inf norm of residual):         %g\n",lu->id.RINFOG(6));CHKERRQ(ierr);
254c338a77dSKris Buschelman     ierr = PetscPrintf(PETSC_COMM_SELF,"        RINFOG(7),RINFOG(8) (backward error est): %g, %g\n",lu->id.RINFOG(7),lu->id.RINFOG(8));CHKERRQ(ierr);
255c338a77dSKris Buschelman     ierr = PetscPrintf(PETSC_COMM_SELF,"        RINFOG(9) (error estimate):               %g \n",lu->id.RINFOG(9));CHKERRQ(ierr);
256c338a77dSKris Buschelman     ierr = PetscPrintf(PETSC_COMM_SELF,"        RINFOG(10),RINFOG(11)(condition numbers): %g, %g\n",lu->id.RINFOG(10),lu->id.RINFOG(11));CHKERRQ(ierr);
257c338a77dSKris Buschelman 
258c338a77dSKris Buschelman   }
259c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(12) (efficiency control):                         %d \n",lu->id.ICNTL(12));CHKERRQ(ierr);
260c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(13) (efficiency control):                         %d \n",lu->id.ICNTL(13));CHKERRQ(ierr);
261adc1d99fSHong Zhang   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(14) (percentage of estimated workspace increase): %d \n",lu->id.ICNTL(14));CHKERRQ(ierr);
262c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(15) (efficiency control):                         %d \n",lu->id.ICNTL(15));CHKERRQ(ierr);
263c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  ICNTL(18) (input mat struct):                           %d \n",lu->id.ICNTL(18));CHKERRQ(ierr);
264c338a77dSKris Buschelman 
265c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  CNTL(1) (relative pivoting threshold):      %g \n",lu->id.CNTL(1));CHKERRQ(ierr);
266c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  CNTL(2) (stopping criterion of refinement): %g \n",lu->id.CNTL(2));CHKERRQ(ierr);
267c338a77dSKris Buschelman   ierr = PetscViewerASCIIPrintf(viewer,"  CNTL(3) (absolute pivoting threshold):      %g \n",lu->id.CNTL(3));CHKERRQ(ierr);
26857f0c58bSHong Zhang 
26957f0c58bSHong Zhang   /* infomation local to each processor */
27057f0c58bSHong Zhang   if (lu->myid == 0) ierr = PetscPrintf(PETSC_COMM_SELF, "      RINFO(1) (local estimated flops for the elimination after analysis): \n");CHKERRQ(ierr);
27157f0c58bSHong Zhang   ierr = PetscSynchronizedPrintf(A->comm,"             [%d] %g \n",lu->myid,lu->id.RINFO(1));CHKERRQ(ierr);
27257f0c58bSHong Zhang   ierr = PetscSynchronizedFlush(A->comm);
27357f0c58bSHong Zhang   if (lu->myid == 0) ierr = PetscPrintf(PETSC_COMM_SELF, "      RINFO(2) (local estimated flops for the assembly after factorization): \n");CHKERRQ(ierr);
27457f0c58bSHong Zhang   ierr = PetscSynchronizedPrintf(A->comm,"             [%d]  %g \n",lu->myid,lu->id.RINFO(2));CHKERRQ(ierr);
27557f0c58bSHong Zhang   ierr = PetscSynchronizedFlush(A->comm);
27657f0c58bSHong Zhang   if (lu->myid == 0) ierr = PetscPrintf(PETSC_COMM_SELF, "      RINFO(3) (local estimated flops for the elimination after factorization): \n");CHKERRQ(ierr);
27757f0c58bSHong Zhang   ierr = PetscSynchronizedPrintf(A->comm,"             [%d]  %g \n",lu->myid,lu->id.RINFO(3));CHKERRQ(ierr);
27857f0c58bSHong Zhang   ierr = PetscSynchronizedFlush(A->comm);
279adc1d99fSHong Zhang 
280adc1d99fSHong Zhang   if (lu->myid == 0){ /* information from the host */
281adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  RINFOG(1) (global estimated flops for the elimination after analysis): %g \n",lu->id.RINFOG(1));CHKERRQ(ierr);
282adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  RINFOG(2) (global estimated flops for the assembly after factorization): %g \n",lu->id.RINFOG(2));CHKERRQ(ierr);
283adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  RINFOG(3) (global estimated flops for the elimination after factorization): %g \n",lu->id.RINFOG(3));CHKERRQ(ierr);
284adc1d99fSHong Zhang 
285adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(3) (estimated real workspace for factors on all processors after analysis): %d \n",lu->id.INFOG(3));CHKERRQ(ierr);
286adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(4) (estimated integer workspace for factors on all processors after analysis): %d \n",lu->id.INFOG(4));CHKERRQ(ierr);
287adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(5) (estimated maximum front size in the complete tree): %d \n",lu->id.INFOG(5));CHKERRQ(ierr);
288adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(6) (number of nodes in the complete tree): %d \n",lu->id.INFOG(6));CHKERRQ(ierr);
289adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(7) (ordering option effectively uese after analysis): %d \n",lu->id.INFOG(7));CHKERRQ(ierr);
290adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(8) (structural symmetry in percent of the permuted matrix after analysis): %d \n",lu->id.INFOG(8));CHKERRQ(ierr);
291adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(9) (total real space store the matrix factors after analysis): %d \n",lu->id.INFOG(9));CHKERRQ(ierr);
292adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(10) (total integer space store the matrix factors after analysis): %d \n",lu->id.INFOG(10));CHKERRQ(ierr);
293adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(11) (order of largest frontal matrix): %d \n",lu->id.INFOG(11));CHKERRQ(ierr);
294adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(12) (number of off-diagonal pivots): %d \n",lu->id.INFOG(12));CHKERRQ(ierr);
295adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(13) (number of delayed pivots after factorization): %d \n",lu->id.INFOG(13));CHKERRQ(ierr);
296adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(14) (number of memory compress after factorization): %d \n",lu->id.INFOG(14));CHKERRQ(ierr);
297adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(15) (number of steps of iterative refinement after solution): %d \n",lu->id.INFOG(15));CHKERRQ(ierr);
298adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(16) (estimated size (in million of bytes) of all MUMPS internal data for factorization after analysis: value on the most memory consuming processor): %d \n",lu->id.INFOG(16));CHKERRQ(ierr);
299adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(17) (estimated size of all MUMPS internal data for factorization after analysis: sum over all processors): %d \n",lu->id.INFOG(17));CHKERRQ(ierr);
300adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(18) (size of all MUMPS internal data allocated during factorization: value on the most memory consuming processor): %d \n",lu->id.INFOG(18));CHKERRQ(ierr);
301adc1d99fSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(19) (size of all MUMPS internal data allocated during factorization: sum over all processors): %d \n",lu->id.INFOG(19));CHKERRQ(ierr);
302adc1d99fSHong Zhang      ierr = PetscViewerASCIIPrintf(viewer,"  INFOG(20) (estimated number of entries in the factors): %d \n",lu->id.INFOG(20));CHKERRQ(ierr);
303adc1d99fSHong Zhang   }
304adc1d99fSHong Zhang 
305397b6df1SKris Buschelman   PetscFunctionReturn(0);
306397b6df1SKris Buschelman }
307397b6df1SKris Buschelman 
308397b6df1SKris Buschelman #undef __FUNCT__
309f0c56d0fSKris Buschelman #define __FUNCT__ "MatView_AIJMUMPS"
310f0c56d0fSKris Buschelman int MatView_AIJMUMPS(Mat A,PetscViewer viewer) {
311397b6df1SKris Buschelman   int               ierr;
312397b6df1SKris Buschelman   PetscTruth        isascii;
313397b6df1SKris Buschelman   PetscViewerFormat format;
314f0c56d0fSKris Buschelman   Mat_MUMPS         *mumps=(Mat_MUMPS*)(A->spptr);
315397b6df1SKris Buschelman 
316397b6df1SKris Buschelman   PetscFunctionBegin;
317397b6df1SKris Buschelman   ierr = (*mumps->MatView)(A,viewer);CHKERRQ(ierr);
318397b6df1SKris Buschelman 
319397b6df1SKris Buschelman   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&isascii);CHKERRQ(ierr);
320397b6df1SKris Buschelman   if (isascii) {
321397b6df1SKris Buschelman     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
322397b6df1SKris Buschelman     if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
323397b6df1SKris Buschelman       ierr = MatFactorInfo_MUMPS(A,viewer);CHKERRQ(ierr);
324397b6df1SKris Buschelman     }
325397b6df1SKris Buschelman   }
326397b6df1SKris Buschelman   PetscFunctionReturn(0);
327397b6df1SKris Buschelman }
328397b6df1SKris Buschelman 
329397b6df1SKris Buschelman #undef __FUNCT__
330f0c56d0fSKris Buschelman #define __FUNCT__ "MatSolve_AIJMUMPS"
331f0c56d0fSKris Buschelman int MatSolve_AIJMUMPS(Mat A,Vec b,Vec x) {
332f0c56d0fSKris Buschelman   Mat_MUMPS   *lu=(Mat_MUMPS*)A->spptr;
333d54de34fSKris Buschelman   PetscScalar *array;
334397b6df1SKris Buschelman   Vec         x_seq;
335397b6df1SKris Buschelman   IS          iden;
336397b6df1SKris Buschelman   VecScatter  scat;
337397b6df1SKris Buschelman   int         ierr;
338397b6df1SKris Buschelman 
339397b6df1SKris Buschelman   PetscFunctionBegin;
340397b6df1SKris Buschelman   if (lu->size > 1){
341397b6df1SKris Buschelman     if (!lu->myid){
342397b6df1SKris Buschelman       ierr = VecCreateSeq(PETSC_COMM_SELF,A->N,&x_seq);CHKERRQ(ierr);
343397b6df1SKris Buschelman       ierr = ISCreateStride(PETSC_COMM_SELF,A->N,0,1,&iden);CHKERRQ(ierr);
344397b6df1SKris Buschelman     } else {
345397b6df1SKris Buschelman       ierr = VecCreateSeq(PETSC_COMM_SELF,0,&x_seq);CHKERRQ(ierr);
346397b6df1SKris Buschelman       ierr = ISCreateStride(PETSC_COMM_SELF,0,0,1,&iden);CHKERRQ(ierr);
347397b6df1SKris Buschelman     }
348397b6df1SKris Buschelman     ierr = VecScatterCreate(b,iden,x_seq,iden,&scat);CHKERRQ(ierr);
349397b6df1SKris Buschelman     ierr = ISDestroy(iden);CHKERRQ(ierr);
350397b6df1SKris Buschelman 
351397b6df1SKris Buschelman     ierr = VecScatterBegin(b,x_seq,INSERT_VALUES,SCATTER_FORWARD,scat);CHKERRQ(ierr);
352397b6df1SKris Buschelman     ierr = VecScatterEnd(b,x_seq,INSERT_VALUES,SCATTER_FORWARD,scat);CHKERRQ(ierr);
353397b6df1SKris Buschelman     if (!lu->myid) {ierr = VecGetArray(x_seq,&array);CHKERRQ(ierr);}
354397b6df1SKris Buschelman   } else {  /* size == 1 */
355397b6df1SKris Buschelman     ierr = VecCopy(b,x);CHKERRQ(ierr);
356397b6df1SKris Buschelman     ierr = VecGetArray(x,&array);CHKERRQ(ierr);
357397b6df1SKris Buschelman   }
358397b6df1SKris Buschelman   if (!lu->myid) { /* define rhs on the host */
359397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
360397b6df1SKris Buschelman     lu->id.rhs = (mumps_double_complex*)array;
361397b6df1SKris Buschelman #else
362397b6df1SKris Buschelman     lu->id.rhs = array;
363397b6df1SKris Buschelman #endif
364397b6df1SKris Buschelman   }
365397b6df1SKris Buschelman 
366397b6df1SKris Buschelman   /* solve phase */
367397b6df1SKris Buschelman   lu->id.job=3;
368397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
369397b6df1SKris Buschelman   zmumps_c(&lu->id);
370397b6df1SKris Buschelman #else
371397b6df1SKris Buschelman   dmumps_c(&lu->id);
372397b6df1SKris Buschelman #endif
373397b6df1SKris Buschelman   if (lu->id.INFOG(1) < 0) {
374397b6df1SKris Buschelman     SETERRQ1(1,"Error reported by MUMPS in solve phase: INFOG(1)=%d\n",lu->id.INFOG(1));
375397b6df1SKris Buschelman   }
376397b6df1SKris Buschelman 
377397b6df1SKris Buschelman   /* convert mumps solution x_seq to petsc mpi x */
378397b6df1SKris Buschelman   if (lu->size > 1) {
379397b6df1SKris Buschelman     if (!lu->myid){
380397b6df1SKris Buschelman       ierr = VecRestoreArray(x_seq,&array);CHKERRQ(ierr);
381397b6df1SKris Buschelman     }
382397b6df1SKris Buschelman     ierr = VecScatterBegin(x_seq,x,INSERT_VALUES,SCATTER_REVERSE,scat);CHKERRQ(ierr);
383397b6df1SKris Buschelman     ierr = VecScatterEnd(x_seq,x,INSERT_VALUES,SCATTER_REVERSE,scat);CHKERRQ(ierr);
384397b6df1SKris Buschelman     ierr = VecScatterDestroy(scat);CHKERRQ(ierr);
385397b6df1SKris Buschelman     ierr = VecDestroy(x_seq);CHKERRQ(ierr);
386397b6df1SKris Buschelman   } else {
387397b6df1SKris Buschelman     ierr = VecRestoreArray(x,&array);CHKERRQ(ierr);
388397b6df1SKris Buschelman   }
389397b6df1SKris Buschelman 
390397b6df1SKris Buschelman   PetscFunctionReturn(0);
391397b6df1SKris Buschelman }
392397b6df1SKris Buschelman 
393*a58c3f20SHong Zhang /*
394*a58c3f20SHong Zhang   input:
395*a58c3f20SHong Zhang    F:        numeric factor
396*a58c3f20SHong Zhang   output:
397*a58c3f20SHong Zhang    nneg:     total number of negative pivots
398*a58c3f20SHong Zhang    nzero:    0
399*a58c3f20SHong Zhang    npos:     (global dimension of F) - nneg
400*a58c3f20SHong Zhang */
401*a58c3f20SHong Zhang 
402*a58c3f20SHong Zhang #undef __FUNCT__
403*a58c3f20SHong Zhang #define __FUNCT__ "MatGetInertia_SBAIJMUMPS"
404*a58c3f20SHong Zhang int MatGetInertia_SBAIJMUMPS(Mat F,int *nneg,int *nzero,int *npos)
405*a58c3f20SHong Zhang {
406*a58c3f20SHong Zhang   Mat_MUMPS  *lu =(Mat_MUMPS*)F->spptr;
407*a58c3f20SHong Zhang   int        ierr,neg,zero,pos;
408*a58c3f20SHong Zhang 
409*a58c3f20SHong Zhang   PetscFunctionBegin;
410*a58c3f20SHong Zhang   if (nneg){
411*a58c3f20SHong Zhang     if (!lu->myid){
412*a58c3f20SHong Zhang       *nneg = lu->id.INFOG(12);
413*a58c3f20SHong Zhang     }
414*a58c3f20SHong Zhang     ierr = MPI_Bcast(nneg,1,MPI_INT,0,lu->comm_mumps);
415*a58c3f20SHong Zhang   }
416*a58c3f20SHong Zhang   if (nzero) *nzero = 0;
417*a58c3f20SHong Zhang   if (npos)  *npos  = F->M - (*nneg);
418*a58c3f20SHong Zhang   PetscFunctionReturn(0);
419*a58c3f20SHong Zhang }
420*a58c3f20SHong Zhang 
421397b6df1SKris Buschelman #undef __FUNCT__
422f0c56d0fSKris Buschelman #define __FUNCT__ "MatFactorNumeric_MPIAIJMUMPS"
423f0c56d0fSKris Buschelman int MatFactorNumeric_AIJMUMPS(Mat A,Mat *F) {
424f0c56d0fSKris Buschelman   Mat_MUMPS  *lu =(Mat_MUMPS*)(*F)->spptr;
425f0c56d0fSKris Buschelman   Mat_MUMPS  *lua=(Mat_MUMPS*)(A)->spptr;
426397b6df1SKris Buschelman   int        rnz,nnz,ierr,nz,i,M=A->M,*ai,*aj,icntl;
427397b6df1SKris Buschelman   PetscTruth valOnly,flg;
428397b6df1SKris Buschelman 
429397b6df1SKris Buschelman   PetscFunctionBegin;
430397b6df1SKris Buschelman   if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){
431f0c56d0fSKris Buschelman     (*F)->ops->solve    = MatSolve_AIJMUMPS;
432397b6df1SKris Buschelman 
433397b6df1SKris Buschelman     /* Initialize a MUMPS instance */
434397b6df1SKris Buschelman     ierr = MPI_Comm_rank(A->comm, &lu->myid);
435397b6df1SKris Buschelman     ierr = MPI_Comm_size(A->comm,&lu->size);CHKERRQ(ierr);
43675747be1SHong Zhang     lua->myid = lu->myid; lua->size = lu->size;
437397b6df1SKris Buschelman     lu->id.job = JOB_INIT;
438397b6df1SKris Buschelman     ierr = MPI_Comm_dup(A->comm,&(lu->comm_mumps));CHKERRQ(ierr);
439397b6df1SKris Buschelman     lu->id.comm_fortran = lu->comm_mumps;
440397b6df1SKris Buschelman 
441397b6df1SKris Buschelman     /* Set mumps options */
442397b6df1SKris Buschelman     ierr = PetscOptionsBegin(A->comm,A->prefix,"MUMPS Options","Mat");CHKERRQ(ierr);
443397b6df1SKris Buschelman     lu->id.par=1;  /* host participates factorizaton and solve */
444397b6df1SKris Buschelman     lu->id.sym=lu->sym;
445397b6df1SKris Buschelman     if (lu->sym == 2){
446397b6df1SKris Buschelman       ierr = PetscOptionsInt("-mat_mumps_sym","SYM: (1,2)","None",lu->id.sym,&icntl,&flg);CHKERRQ(ierr);
447397b6df1SKris Buschelman       if (flg && icntl == 1) lu->id.sym=icntl;  /* matrix is spd */
448397b6df1SKris Buschelman     }
449397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
450397b6df1SKris Buschelman   zmumps_c(&lu->id);
451397b6df1SKris Buschelman #else
452397b6df1SKris Buschelman   dmumps_c(&lu->id);
453397b6df1SKris Buschelman #endif
454397b6df1SKris Buschelman 
455397b6df1SKris Buschelman     if (lu->size == 1){
456397b6df1SKris Buschelman       lu->id.ICNTL(18) = 0;   /* centralized assembled matrix input */
457397b6df1SKris Buschelman     } else {
458397b6df1SKris Buschelman       lu->id.ICNTL(18) = 3;   /* distributed assembled matrix input */
459397b6df1SKris Buschelman     }
460397b6df1SKris Buschelman 
461397b6df1SKris Buschelman     icntl=-1;
462397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_4","ICNTL(4): level of printing (0 to 4)","None",lu->id.ICNTL(4),&icntl,&flg);CHKERRQ(ierr);
463397b6df1SKris Buschelman     if (flg && icntl > 0) {
464397b6df1SKris Buschelman       lu->id.ICNTL(4)=icntl; /* and use mumps default icntl(i), i=1,2,3 */
465397b6df1SKris Buschelman     } else { /* no output */
466397b6df1SKris Buschelman       lu->id.ICNTL(1) = 0;  /* error message, default= 6 */
467397b6df1SKris Buschelman       lu->id.ICNTL(2) = -1; /* output stream for diagnostic printing, statistics, and warning. default=0 */
468397b6df1SKris Buschelman       lu->id.ICNTL(3) = -1; /* output stream for global information, default=6 */
469397b6df1SKris Buschelman       lu->id.ICNTL(4) = 0;  /* level of printing, 0,1,2,3,4, default=2 */
470397b6df1SKris Buschelman     }
471397b6df1SKris 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);
472397b6df1SKris Buschelman     icntl=-1;
473397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_7","ICNTL(7): matrix ordering (0 to 7)","None",lu->id.ICNTL(7),&icntl,&flg);CHKERRQ(ierr);
474397b6df1SKris Buschelman     if (flg) {
475397b6df1SKris Buschelman       if (icntl== 1){
476397b6df1SKris Buschelman         SETERRQ(PETSC_ERR_SUP,"pivot order be set by the user in PERM_IN -- not supported by the PETSc/MUMPS interface\n");
477397b6df1SKris Buschelman       } else {
478397b6df1SKris Buschelman         lu->id.ICNTL(7) = icntl;
479397b6df1SKris Buschelman       }
480397b6df1SKris Buschelman     }
481397b6df1SKris 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);
482397b6df1SKris 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);
483397b6df1SKris 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);
484397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_12","ICNTL(12): efficiency control","None",lu->id.ICNTL(12),&lu->id.ICNTL(12),PETSC_NULL);CHKERRQ(ierr);
485397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_13","ICNTL(13): efficiency control","None",lu->id.ICNTL(13),&lu->id.ICNTL(13),PETSC_NULL);CHKERRQ(ierr);
486adc1d99fSHong 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);
487397b6df1SKris Buschelman     ierr = PetscOptionsInt("-mat_mumps_icntl_15","ICNTL(15): efficiency control","None",lu->id.ICNTL(15),&lu->id.ICNTL(15),PETSC_NULL);CHKERRQ(ierr);
488397b6df1SKris Buschelman 
489397b6df1SKris Buschelman     /*
490397b6df1SKris 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);
491397b6df1SKris Buschelman     if (flg){
492397b6df1SKris Buschelman       if (icntl >-1 && icntl <3 ){
493397b6df1SKris Buschelman         if (lu->myid==0) lu->id.ICNTL(16) = icntl;
494397b6df1SKris Buschelman       } else {
495397b6df1SKris Buschelman         SETERRQ1(PETSC_ERR_SUP,"ICNTL(16)=%d -- not supported\n",icntl);
496397b6df1SKris Buschelman       }
497397b6df1SKris Buschelman     }
498397b6df1SKris Buschelman     */
499397b6df1SKris Buschelman 
500397b6df1SKris 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);
501397b6df1SKris 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);
502397b6df1SKris 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);
503397b6df1SKris Buschelman     PetscOptionsEnd();
504397b6df1SKris Buschelman   }
505397b6df1SKris Buschelman 
506397b6df1SKris Buschelman   /* define matrix A */
507397b6df1SKris Buschelman   switch (lu->id.ICNTL(18)){
508397b6df1SKris Buschelman   case 0:  /* centralized assembled matrix input (size=1) */
509397b6df1SKris Buschelman     if (!lu->myid) {
510c36ead0aSKris Buschelman       if (lua->isAIJ){
511397b6df1SKris Buschelman         Mat_SeqAIJ   *aa = (Mat_SeqAIJ*)A->data;
512397b6df1SKris Buschelman         nz               = aa->nz;
513397b6df1SKris Buschelman         ai = aa->i; aj = aa->j; lu->val = aa->a;
514397b6df1SKris Buschelman       } else {
515397b6df1SKris Buschelman         Mat_SeqSBAIJ *aa = (Mat_SeqSBAIJ*)A->data;
516397b6df1SKris Buschelman         nz                  =  aa->s_nz;
517397b6df1SKris Buschelman         ai = aa->i; aj = aa->j; lu->val = aa->a;
518397b6df1SKris Buschelman       }
519397b6df1SKris Buschelman       if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){ /* first numeric factorization, get irn and jcn */
520397b6df1SKris Buschelman         ierr = PetscMalloc(nz*sizeof(int),&lu->irn);CHKERRQ(ierr);
521397b6df1SKris Buschelman         ierr = PetscMalloc(nz*sizeof(int),&lu->jcn);CHKERRQ(ierr);
522397b6df1SKris Buschelman         nz = 0;
523397b6df1SKris Buschelman         for (i=0; i<M; i++){
524397b6df1SKris Buschelman           rnz = ai[i+1] - ai[i];
525397b6df1SKris Buschelman           while (rnz--) {  /* Fortran row/col index! */
526397b6df1SKris Buschelman             lu->irn[nz] = i+1; lu->jcn[nz] = (*aj)+1; aj++; nz++;
527397b6df1SKris Buschelman           }
528397b6df1SKris Buschelman         }
529397b6df1SKris Buschelman       }
530397b6df1SKris Buschelman     }
531397b6df1SKris Buschelman     break;
532397b6df1SKris Buschelman   case 3:  /* distributed assembled matrix input (size>1) */
533397b6df1SKris Buschelman     if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){
534397b6df1SKris Buschelman       valOnly = PETSC_FALSE;
535397b6df1SKris Buschelman     } else {
536397b6df1SKris Buschelman       valOnly = PETSC_TRUE; /* only update mat values, not row and col index */
537397b6df1SKris Buschelman     }
538397b6df1SKris Buschelman     ierr = MatConvertToTriples(A,1,valOnly, &nnz, &lu->irn, &lu->jcn, &lu->val);CHKERRQ(ierr);
539397b6df1SKris Buschelman     break;
540397b6df1SKris Buschelman   default: SETERRQ(PETSC_ERR_SUP,"Matrix input format is not supported by MUMPS.");
541397b6df1SKris Buschelman   }
542397b6df1SKris Buschelman 
543397b6df1SKris Buschelman   /* analysis phase */
544397b6df1SKris Buschelman   if (lu->matstruc == DIFFERENT_NONZERO_PATTERN){
545397b6df1SKris Buschelman      lu->id.n = M;
546397b6df1SKris Buschelman     switch (lu->id.ICNTL(18)){
547397b6df1SKris Buschelman     case 0:  /* centralized assembled matrix input */
548397b6df1SKris Buschelman       if (!lu->myid) {
549397b6df1SKris Buschelman         lu->id.nz =nz; lu->id.irn=lu->irn; lu->id.jcn=lu->jcn;
550397b6df1SKris Buschelman         if (lu->id.ICNTL(6)>1){
551397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
552397b6df1SKris Buschelman           lu->id.a = (mumps_double_complex*)lu->val;
553397b6df1SKris Buschelman #else
554397b6df1SKris Buschelman           lu->id.a = lu->val;
555397b6df1SKris Buschelman #endif
556397b6df1SKris Buschelman         }
557397b6df1SKris Buschelman       }
558397b6df1SKris Buschelman       break;
559397b6df1SKris Buschelman     case 3:  /* distributed assembled matrix input (size>1) */
560397b6df1SKris Buschelman       lu->id.nz_loc = nnz;
561397b6df1SKris Buschelman       lu->id.irn_loc=lu->irn; lu->id.jcn_loc=lu->jcn;
562397b6df1SKris Buschelman       if (lu->id.ICNTL(6)>1) {
563397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
564397b6df1SKris Buschelman         lu->id.a_loc = (mumps_double_complex*)lu->val;
565397b6df1SKris Buschelman #else
566397b6df1SKris Buschelman         lu->id.a_loc = lu->val;
567397b6df1SKris Buschelman #endif
568397b6df1SKris Buschelman       }
569397b6df1SKris Buschelman       break;
570397b6df1SKris Buschelman     }
571397b6df1SKris Buschelman     lu->id.job=1;
572397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
573397b6df1SKris Buschelman   zmumps_c(&lu->id);
574397b6df1SKris Buschelman #else
575397b6df1SKris Buschelman   dmumps_c(&lu->id);
576397b6df1SKris Buschelman #endif
577397b6df1SKris Buschelman     if (lu->id.INFOG(1) < 0) {
578397b6df1SKris Buschelman       SETERRQ1(1,"Error reported by MUMPS in analysis phase: INFOG(1)=%d\n",lu->id.INFOG(1));
579397b6df1SKris Buschelman     }
580397b6df1SKris Buschelman   }
581397b6df1SKris Buschelman 
582397b6df1SKris Buschelman   /* numerical factorization phase */
583397b6df1SKris Buschelman   if(lu->id.ICNTL(18) == 0) {
584397b6df1SKris Buschelman     if (lu->myid == 0) {
585397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
586397b6df1SKris Buschelman       lu->id.a = (mumps_double_complex*)lu->val;
587397b6df1SKris Buschelman #else
588397b6df1SKris Buschelman       lu->id.a = lu->val;
589397b6df1SKris Buschelman #endif
590397b6df1SKris Buschelman     }
591397b6df1SKris Buschelman   } else {
592397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
593397b6df1SKris Buschelman     lu->id.a_loc = (mumps_double_complex*)lu->val;
594397b6df1SKris Buschelman #else
595397b6df1SKris Buschelman     lu->id.a_loc = lu->val;
596397b6df1SKris Buschelman #endif
597397b6df1SKris Buschelman   }
598397b6df1SKris Buschelman   lu->id.job=2;
599397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
600397b6df1SKris Buschelman   zmumps_c(&lu->id);
601397b6df1SKris Buschelman #else
602397b6df1SKris Buschelman   dmumps_c(&lu->id);
603397b6df1SKris Buschelman #endif
604397b6df1SKris Buschelman   if (lu->id.INFOG(1) < 0) {
605397b6df1SKris Buschelman     SETERRQ1(1,"1, Error reported by MUMPS in numerical factorization phase: INFOG(1)=%d\n",lu->id.INFOG(1));
606397b6df1SKris Buschelman   }
607397b6df1SKris Buschelman 
608397b6df1SKris Buschelman   if (lu->myid==0 && lu->id.ICNTL(16) > 0){
609397b6df1SKris Buschelman     SETERRQ1(1,"  lu->id.ICNTL(16):=%d\n",lu->id.INFOG(16));
610397b6df1SKris Buschelman   }
611397b6df1SKris Buschelman 
612397b6df1SKris Buschelman   (*F)->assembled  = PETSC_TRUE;
613397b6df1SKris Buschelman   lu->matstruc     = SAME_NONZERO_PATTERN;
614ace87b0dSHong Zhang   lu->CleanUpMUMPS = PETSC_TRUE;
615397b6df1SKris Buschelman   PetscFunctionReturn(0);
616397b6df1SKris Buschelman }
617397b6df1SKris Buschelman 
618397b6df1SKris Buschelman /* Note the Petsc r and c permutations are ignored */
619397b6df1SKris Buschelman #undef __FUNCT__
620f0c56d0fSKris Buschelman #define __FUNCT__ "MatLUFactorSymbolic_AIJMUMPS"
621f0c56d0fSKris Buschelman int MatLUFactorSymbolic_AIJMUMPS(Mat A,IS r,IS c,MatFactorInfo *info,Mat *F) {
622397b6df1SKris Buschelman   Mat       B;
623f0c56d0fSKris Buschelman   Mat_MUMPS *lu;
624397b6df1SKris Buschelman   int       ierr;
625397b6df1SKris Buschelman 
626397b6df1SKris Buschelman   PetscFunctionBegin;
627397b6df1SKris Buschelman 
628397b6df1SKris Buschelman   /* Create the factorization matrix */
629397b6df1SKris Buschelman   ierr = MatCreate(A->comm,A->m,A->n,A->M,A->N,&B);CHKERRQ(ierr);
630397b6df1SKris Buschelman   ierr = MatSetType(B,MATAIJMUMPS);CHKERRQ(ierr);
631397b6df1SKris Buschelman   ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr);
632397b6df1SKris Buschelman   ierr = MatMPIAIJSetPreallocation(B,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr);
633397b6df1SKris Buschelman 
634f0c56d0fSKris Buschelman   B->ops->lufactornumeric = MatFactorNumeric_AIJMUMPS;
635397b6df1SKris Buschelman   B->factor               = FACTOR_LU;
636f0c56d0fSKris Buschelman   lu                      = (Mat_MUMPS*)B->spptr;
637397b6df1SKris Buschelman   lu->sym                 = 0;
638397b6df1SKris Buschelman   lu->matstruc            = DIFFERENT_NONZERO_PATTERN;
639397b6df1SKris Buschelman 
640397b6df1SKris Buschelman   *F = B;
641397b6df1SKris Buschelman   PetscFunctionReturn(0);
642397b6df1SKris Buschelman }
643397b6df1SKris Buschelman 
644397b6df1SKris Buschelman /* Note the Petsc r permutation is ignored */
645397b6df1SKris Buschelman #undef __FUNCT__
646f0c56d0fSKris Buschelman #define __FUNCT__ "MatCholeskyFactorSymbolic_SBAIJMUMPS"
647f0c56d0fSKris Buschelman int MatCholeskyFactorSymbolic_SBAIJMUMPS(Mat A,IS r,MatFactorInfo *info,Mat *F) {
648397b6df1SKris Buschelman   Mat       B;
649f0c56d0fSKris Buschelman   Mat_MUMPS *lu;
650397b6df1SKris Buschelman   int       ierr;
651397b6df1SKris Buschelman 
652397b6df1SKris Buschelman   PetscFunctionBegin;
653397b6df1SKris Buschelman 
654397b6df1SKris Buschelman   /* Create the factorization matrix */
655397b6df1SKris Buschelman   ierr = MatCreate(A->comm,A->m,A->n,A->M,A->N,&B);CHKERRQ(ierr);
656397b6df1SKris Buschelman   ierr = MatSetType(B,MATAIJMUMPS);CHKERRQ(ierr);
657397b6df1SKris Buschelman   ierr = MatSeqAIJSetPreallocation(B,0,PETSC_NULL);CHKERRQ(ierr);
658397b6df1SKris Buschelman   ierr = MatMPIAIJSetPreallocation(B,0,PETSC_NULL,0,PETSC_NULL);CHKERRQ(ierr);
659397b6df1SKris Buschelman 
660f0c56d0fSKris Buschelman   B->ops->choleskyfactornumeric = MatFactorNumeric_AIJMUMPS;
661*a58c3f20SHong Zhang   B->ops->getinertia            = MatGetInertia_SBAIJMUMPS;
662397b6df1SKris Buschelman   B->factor                     = FACTOR_CHOLESKY;
663f0c56d0fSKris Buschelman   lu                            = (Mat_MUMPS*)B->spptr;
664397b6df1SKris Buschelman   lu->sym                       = 2;
665397b6df1SKris Buschelman   lu->matstruc                  = DIFFERENT_NONZERO_PATTERN;
666397b6df1SKris Buschelman 
667397b6df1SKris Buschelman   *F = B;
668397b6df1SKris Buschelman   PetscFunctionReturn(0);
669397b6df1SKris Buschelman }
670397b6df1SKris Buschelman 
671397b6df1SKris Buschelman #undef __FUNCT__
672f0c56d0fSKris Buschelman #define __FUNCT__ "MatAssemblyEnd_AIJMUMPS"
673f0c56d0fSKris Buschelman int MatAssemblyEnd_AIJMUMPS(Mat A,MatAssemblyType mode) {
674c338a77dSKris Buschelman   int       ierr;
675f0c56d0fSKris Buschelman   Mat_MUMPS *mumps=(Mat_MUMPS*)A->spptr;
676c338a77dSKris Buschelman 
677397b6df1SKris Buschelman   PetscFunctionBegin;
678c338a77dSKris Buschelman   ierr = (*mumps->MatAssemblyEnd)(A,mode);CHKERRQ(ierr);
679f0c56d0fSKris Buschelman 
680c338a77dSKris Buschelman   mumps->MatLUFactorSymbolic       = A->ops->lufactorsymbolic;
681c338a77dSKris Buschelman   mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic;
682f0c56d0fSKris Buschelman   A->ops->lufactorsymbolic         = MatLUFactorSymbolic_AIJMUMPS;
683397b6df1SKris Buschelman   PetscFunctionReturn(0);
684397b6df1SKris Buschelman }
685397b6df1SKris Buschelman 
686c338a77dSKris Buschelman EXTERN_C_BEGIN
687c338a77dSKris Buschelman #undef __FUNCT__
688f0c56d0fSKris Buschelman #define __FUNCT__ "MatConvert_AIJ_AIJMUMPS"
689f0c56d0fSKris Buschelman int MatConvert_AIJ_AIJMUMPS(Mat A,MatType newtype,Mat *newmat) {
690c338a77dSKris Buschelman   int       ierr,size;
691c338a77dSKris Buschelman   MPI_Comm  comm;
692c338a77dSKris Buschelman   Mat       B=*newmat;
693f0c56d0fSKris Buschelman   Mat_MUMPS *mumps;
694397b6df1SKris Buschelman 
695397b6df1SKris Buschelman   PetscFunctionBegin;
696c338a77dSKris Buschelman   if (B != A) {
697c338a77dSKris Buschelman     ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr);
698397b6df1SKris Buschelman   }
699397b6df1SKris Buschelman 
700c338a77dSKris Buschelman   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
701f0c56d0fSKris Buschelman   ierr = PetscNew(Mat_MUMPS,&mumps);CHKERRQ(ierr);
702c338a77dSKris Buschelman 
703f0c56d0fSKris Buschelman   mumps->MatDuplicate              = A->ops->duplicate;
704c338a77dSKris Buschelman   mumps->MatView                   = A->ops->view;
705c338a77dSKris Buschelman   mumps->MatAssemblyEnd            = A->ops->assemblyend;
706c338a77dSKris Buschelman   mumps->MatLUFactorSymbolic       = A->ops->lufactorsymbolic;
707c338a77dSKris Buschelman   mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic;
708c338a77dSKris Buschelman   mumps->MatDestroy                = A->ops->destroy;
709a39386dcSKris Buschelman   mumps->specialdestroy            = MatDestroy_AIJMUMPS;
710c338a77dSKris Buschelman   mumps->CleanUpMUMPS              = PETSC_FALSE;
711f579278aSKris Buschelman   mumps->isAIJ                     = PETSC_TRUE;
712c338a77dSKris Buschelman 
7134b68dd72SKris Buschelman   B->spptr                         = (void *)mumps;
714f0c56d0fSKris Buschelman   B->ops->duplicate                = MatDuplicate_AIJMUMPS;
715f0c56d0fSKris Buschelman   B->ops->view                     = MatView_AIJMUMPS;
716f0c56d0fSKris Buschelman   B->ops->assemblyend              = MatAssemblyEnd_AIJMUMPS;
717f0c56d0fSKris Buschelman   B->ops->lufactorsymbolic         = MatLUFactorSymbolic_AIJMUMPS;
7183924e44cSKris Buschelman   B->ops->destroy                  = MatDestroy_MUMPS;
719c338a77dSKris Buschelman 
720c338a77dSKris Buschelman   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr);
721c338a77dSKris Buschelman   if (size == 1) {
722c338a77dSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqaij_aijmumps_C",
723f0c56d0fSKris Buschelman                                              "MatConvert_AIJ_AIJMUMPS",MatConvert_AIJ_AIJMUMPS);CHKERRQ(ierr);
724c338a77dSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_aijmumps_seqaij_C",
725c338a77dSKris Buschelman                                              "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr);
726c338a77dSKris Buschelman   } else {
727c338a77dSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_mpiaij_aijmumps_C",
728f0c56d0fSKris Buschelman                                              "MatConvert_AIJ_AIJMUMPS",MatConvert_AIJ_AIJMUMPS);CHKERRQ(ierr);
729c338a77dSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_aijmumps_mpiaij_C",
730c338a77dSKris Buschelman                                              "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr);
731c338a77dSKris Buschelman   }
732c338a77dSKris Buschelman 
733f579278aSKris Buschelman   PetscLogInfo(0,"Using MUMPS for LU factorization and solves.");
734c338a77dSKris Buschelman   ierr = PetscObjectChangeTypeName((PetscObject)B,newtype);CHKERRQ(ierr);
735c338a77dSKris Buschelman   *newmat = B;
736397b6df1SKris Buschelman   PetscFunctionReturn(0);
737397b6df1SKris Buschelman }
738c338a77dSKris Buschelman EXTERN_C_END
739397b6df1SKris Buschelman 
740f0c56d0fSKris Buschelman #undef __FUNCT__
741f0c56d0fSKris Buschelman #define __FUNCT__ "MatDuplicate_AIJMUMPS"
742f0c56d0fSKris Buschelman int MatDuplicate_AIJMUMPS(Mat A, MatDuplicateOption op, Mat *M) {
743f0c56d0fSKris Buschelman   int       ierr;
7448f340917SKris Buschelman   Mat_MUMPS *lu=(Mat_MUMPS *)A->spptr;
7458f340917SKris Buschelman 
746f0c56d0fSKris Buschelman   PetscFunctionBegin;
7478f340917SKris Buschelman   ierr = (*lu->MatDuplicate)(A,op,M);CHKERRQ(ierr);
748f0c56d0fSKris Buschelman   ierr = MatConvert_AIJ_AIJMUMPS(*M,MATAIJMUMPS,M);CHKERRQ(ierr);
749a39386dcSKris Buschelman   ierr = PetscMemcpy((*M)->spptr,lu,sizeof(Mat_MUMPS));CHKERRQ(ierr);
750f0c56d0fSKris Buschelman   PetscFunctionReturn(0);
751f0c56d0fSKris Buschelman }
752f0c56d0fSKris Buschelman 
75324b6179bSKris Buschelman /*MC
754fafad747SKris Buschelman   MATAIJMUMPS - MATAIJMUMPS = "aijmumps" - A matrix type providing direct solvers (LU) for distributed
75524b6179bSKris Buschelman   and sequential matrices via the external package MUMPS.
75624b6179bSKris Buschelman 
75724b6179bSKris Buschelman   If MUMPS is installed (see the manual for instructions
75824b6179bSKris Buschelman   on how to declare the existence of external packages),
75924b6179bSKris Buschelman   a matrix type can be constructed which invokes MUMPS solvers.
76024b6179bSKris Buschelman   After calling MatCreate(...,A), simply call MatSetType(A,MATAIJMUMPS).
76124b6179bSKris Buschelman   This matrix type is only supported for double precision real.
76224b6179bSKris Buschelman 
76324b6179bSKris Buschelman   If created with a single process communicator, this matrix type inherits from MATSEQAIJ.
76424b6179bSKris Buschelman   Otherwise, this matrix type inherits from MATMPIAIJ.  Hence for single process communicators,
76524b6179bSKris Buschelman   MatSeqAIJSetPreallocation is supported, and similarly MatMPIAIJSetPreallocation is supported
76624b6179bSKris Buschelman   for communicators controlling multiple processes.  It is recommended that you call both of
76728b08bd3SKris Buschelman   the above preallocation routines for simplicity.  One can also call MatConvert for an inplace
76828b08bd3SKris Buschelman   conversion to or from the MATSEQAIJ or MATMPIAIJ type (depending on the communicator size)
76928b08bd3SKris Buschelman   without data copy.
77024b6179bSKris Buschelman 
77124b6179bSKris Buschelman   Options Database Keys:
7720bad9183SKris Buschelman + -mat_type aijmumps - sets the matrix type to "aijmumps" during a call to MatSetFromOptions()
77324b6179bSKris Buschelman . -mat_mumps_sym <0,1,2> - 0 the matrix is unsymmetric, 1 symmetric positive definite, 2 symmetric
77424b6179bSKris Buschelman . -mat_mumps_icntl_4 <0,1,2,3,4> - print level
77524b6179bSKris Buschelman . -mat_mumps_icntl_6 <0,...,7> - matrix prescaling options (see MUMPS User's Guide)
77624b6179bSKris Buschelman . -mat_mumps_icntl_7 <0,...,7> - matrix orderings (see MUMPS User's Guide)
77724b6179bSKris Buschelman . -mat_mumps_icntl_9 <1,2> - A or A^T x=b to be solved: 1 denotes A, 2 denotes A^T
77824b6179bSKris Buschelman . -mat_mumps_icntl_10 <n> - maximum number of iterative refinements
77924b6179bSKris Buschelman . -mat_mumps_icntl_11 <n> - error analysis, a positive value returns statistics during -sles_view
78024b6179bSKris Buschelman . -mat_mumps_icntl_12 <n> - efficiency control (see MUMPS User's Guide)
78124b6179bSKris Buschelman . -mat_mumps_icntl_13 <n> - efficiency control (see MUMPS User's Guide)
78224b6179bSKris Buschelman . -mat_mumps_icntl_14 <n> - efficiency control (see MUMPS User's Guide)
78324b6179bSKris Buschelman . -mat_mumps_icntl_15 <n> - efficiency control (see MUMPS User's Guide)
78424b6179bSKris Buschelman . -mat_mumps_cntl_1 <delta> - relative pivoting threshold
78524b6179bSKris Buschelman . -mat_mumps_cntl_2 <tol> - stopping criterion for refinement
78624b6179bSKris Buschelman - -mat_mumps_cntl_3 <adelta> - absolute pivoting threshold
78724b6179bSKris Buschelman 
78824b6179bSKris Buschelman   Level: beginner
78924b6179bSKris Buschelman 
79024b6179bSKris Buschelman .seealso: MATSBAIJMUMPS
79124b6179bSKris Buschelman M*/
79224b6179bSKris Buschelman 
793397b6df1SKris Buschelman EXTERN_C_BEGIN
794397b6df1SKris Buschelman #undef __FUNCT__
795f0c56d0fSKris Buschelman #define __FUNCT__ "MatCreate_AIJMUMPS"
796f0c56d0fSKris Buschelman int MatCreate_AIJMUMPS(Mat A) {
797397b6df1SKris Buschelman   int           ierr,size;
798397b6df1SKris Buschelman   MPI_Comm      comm;
799397b6df1SKris Buschelman 
800397b6df1SKris Buschelman   PetscFunctionBegin;
8015441df8eSKris Buschelman   /* Change type name before calling MatSetType to force proper construction of SeqAIJ or MPIAIJ */
8025441df8eSKris Buschelman   /*   and AIJMUMPS types */
8035441df8eSKris Buschelman   ierr = PetscObjectChangeTypeName((PetscObject)A,MATAIJMUMPS);CHKERRQ(ierr);
804397b6df1SKris Buschelman   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
805397b6df1SKris Buschelman   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr);
806397b6df1SKris Buschelman   if (size == 1) {
807397b6df1SKris Buschelman     ierr = MatSetType(A,MATSEQAIJ);CHKERRQ(ierr);
808397b6df1SKris Buschelman   } else {
809397b6df1SKris Buschelman     ierr = MatSetType(A,MATMPIAIJ);CHKERRQ(ierr);
810397b6df1SKris Buschelman   }
811f0c56d0fSKris Buschelman   ierr = MatConvert_AIJ_AIJMUMPS(A,MATAIJMUMPS,&A);CHKERRQ(ierr);
812397b6df1SKris Buschelman   PetscFunctionReturn(0);
813397b6df1SKris Buschelman }
814397b6df1SKris Buschelman EXTERN_C_END
815397b6df1SKris Buschelman 
816f579278aSKris Buschelman #undef __FUNCT__
817f0c56d0fSKris Buschelman #define __FUNCT__ "MatAssemblyEnd_SBAIJMUMPS"
818f0c56d0fSKris Buschelman int MatAssemblyEnd_SBAIJMUMPS(Mat A,MatAssemblyType mode) {
819f579278aSKris Buschelman   int       ierr;
820f0c56d0fSKris Buschelman   Mat_MUMPS *mumps=(Mat_MUMPS*)A->spptr;
821f579278aSKris Buschelman 
822f579278aSKris Buschelman   PetscFunctionBegin;
823f579278aSKris Buschelman   ierr = (*mumps->MatAssemblyEnd)(A,mode);CHKERRQ(ierr);
824f579278aSKris Buschelman   mumps->MatLUFactorSymbolic       = A->ops->lufactorsymbolic;
825f579278aSKris Buschelman   mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic;
826f0c56d0fSKris Buschelman   A->ops->choleskyfactorsymbolic   = MatCholeskyFactorSymbolic_SBAIJMUMPS;
827f579278aSKris Buschelman   PetscFunctionReturn(0);
828f579278aSKris Buschelman }
829f579278aSKris Buschelman 
830f579278aSKris Buschelman EXTERN_C_BEGIN
831f579278aSKris Buschelman #undef __FUNCT__
832f0c56d0fSKris Buschelman #define __FUNCT__ "MatConvert_SBAIJ_SBAIJMUMPS"
833f0c56d0fSKris Buschelman int MatConvert_SBAIJ_SBAIJMUMPS(Mat A,MatType newtype,Mat *newmat) {
834f579278aSKris Buschelman   int       ierr,size;
835f579278aSKris Buschelman   MPI_Comm  comm;
836f579278aSKris Buschelman   Mat       B=*newmat;
837f0c56d0fSKris Buschelman   Mat_MUMPS *mumps;
838f579278aSKris Buschelman 
839f579278aSKris Buschelman   PetscFunctionBegin;
840f579278aSKris Buschelman   if (B != A) {
841f579278aSKris Buschelman     ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr);
842f579278aSKris Buschelman   }
843f579278aSKris Buschelman 
844f579278aSKris Buschelman   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
845f0c56d0fSKris Buschelman   ierr = PetscNew(Mat_MUMPS,&mumps);CHKERRQ(ierr);
846f579278aSKris Buschelman 
847f0c56d0fSKris Buschelman   mumps->MatDuplicate              = A->ops->duplicate;
848f579278aSKris Buschelman   mumps->MatView                   = A->ops->view;
849f579278aSKris Buschelman   mumps->MatAssemblyEnd            = A->ops->assemblyend;
850f579278aSKris Buschelman   mumps->MatLUFactorSymbolic       = A->ops->lufactorsymbolic;
851f579278aSKris Buschelman   mumps->MatCholeskyFactorSymbolic = A->ops->choleskyfactorsymbolic;
852f579278aSKris Buschelman   mumps->MatDestroy                = A->ops->destroy;
853a39386dcSKris Buschelman   mumps->specialdestroy            = MatDestroy_SBAIJMUMPS;
854f579278aSKris Buschelman   mumps->CleanUpMUMPS              = PETSC_FALSE;
855f579278aSKris Buschelman   mumps->isAIJ                     = PETSC_FALSE;
856f579278aSKris Buschelman 
857f579278aSKris Buschelman   B->spptr                         = (void *)mumps;
858f0c56d0fSKris Buschelman   B->ops->duplicate                = MatDuplicate_SBAIJMUMPS;
859f0c56d0fSKris Buschelman   B->ops->view                     = MatView_AIJMUMPS;
860f0c56d0fSKris Buschelman   B->ops->assemblyend              = MatAssemblyEnd_SBAIJMUMPS;
861f0c56d0fSKris Buschelman   B->ops->choleskyfactorsymbolic   = MatCholeskyFactorSymbolic_SBAIJMUMPS;
8623924e44cSKris Buschelman   B->ops->destroy                  = MatDestroy_MUMPS;
863f579278aSKris Buschelman 
864f579278aSKris Buschelman   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);CHKERRQ(ierr);
865f579278aSKris Buschelman   if (size == 1) {
866f0c56d0fSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_sbaijmumps_C",
867f0c56d0fSKris Buschelman                                              "MatConvert_SBAIJ_SBAIJMUMPS",MatConvert_SBAIJ_SBAIJMUMPS);CHKERRQ(ierr);
868f0c56d0fSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_sbaijmumps_seqsbaij_C",
869f579278aSKris Buschelman                                              "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr);
870f579278aSKris Buschelman   } else {
871f0c56d0fSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_mpisbaij_sbaijmumps_C",
872f0c56d0fSKris Buschelman                                              "MatConvert_SBAIJ_SBAIJMUMPS",MatConvert_SBAIJ_SBAIJMUMPS);CHKERRQ(ierr);
873f0c56d0fSKris Buschelman     ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_sbaijmumps_mpisbaij_C",
874f579278aSKris Buschelman                                              "MatConvert_MUMPS_Base",MatConvert_MUMPS_Base);CHKERRQ(ierr);
875f579278aSKris Buschelman   }
876f579278aSKris Buschelman 
877f579278aSKris Buschelman   PetscLogInfo(0,"Using MUMPS for Cholesky factorization and solves.");
878f579278aSKris Buschelman   ierr = PetscObjectChangeTypeName((PetscObject)B,newtype);CHKERRQ(ierr);
879f579278aSKris Buschelman   *newmat = B;
880f579278aSKris Buschelman   PetscFunctionReturn(0);
881f579278aSKris Buschelman }
882f579278aSKris Buschelman EXTERN_C_END
883f579278aSKris Buschelman 
884f0c56d0fSKris Buschelman #undef __FUNCT__
885f0c56d0fSKris Buschelman #define __FUNCT__ "MatDuplicate_SBAIJMUMPS"
886f0c56d0fSKris Buschelman int MatDuplicate_SBAIJMUMPS(Mat A, MatDuplicateOption op, Mat *M) {
887f0c56d0fSKris Buschelman   int       ierr;
8888f340917SKris Buschelman   Mat_MUMPS *lu=(Mat_MUMPS *)A->spptr;
8898f340917SKris Buschelman 
890f0c56d0fSKris Buschelman   PetscFunctionBegin;
8918f340917SKris Buschelman   ierr = (*lu->MatDuplicate)(A,op,M);CHKERRQ(ierr);
892f0c56d0fSKris Buschelman   ierr = MatConvert_SBAIJ_SBAIJMUMPS(*M,MATSBAIJMUMPS,M);CHKERRQ(ierr);
8933f953163SKris Buschelman   ierr = PetscMemcpy((*M)->spptr,lu,sizeof(Mat_MUMPS));CHKERRQ(ierr);
894f0c56d0fSKris Buschelman   PetscFunctionReturn(0);
895f0c56d0fSKris Buschelman }
896f0c56d0fSKris Buschelman 
89724b6179bSKris Buschelman /*MC
898fafad747SKris Buschelman   MATSBAIJMUMPS - MATSBAIJMUMPS = "sbaijmumps" - A symmetric matrix type providing direct solvers (Cholesky) for
89924b6179bSKris Buschelman   distributed and sequential matrices via the external package MUMPS.
90024b6179bSKris Buschelman 
90124b6179bSKris Buschelman   If MUMPS is installed (see the manual for instructions
90224b6179bSKris Buschelman   on how to declare the existence of external packages),
90324b6179bSKris Buschelman   a matrix type can be constructed which invokes MUMPS solvers.
90424b6179bSKris Buschelman   After calling MatCreate(...,A), simply call MatSetType(A,MATSBAIJMUMPS).
90524b6179bSKris Buschelman   This matrix type is only supported for double precision real.
90624b6179bSKris Buschelman 
90724b6179bSKris Buschelman   If created with a single process communicator, this matrix type inherits from MATSEQSBAIJ.
90824b6179bSKris Buschelman   Otherwise, this matrix type inherits from MATMPISBAIJ.  Hence for single process communicators,
90924b6179bSKris Buschelman   MatSeqSBAIJSetPreallocation is supported, and similarly MatMPISBAIJSetPreallocation is supported
91024b6179bSKris Buschelman   for communicators controlling multiple processes.  It is recommended that you call both of
91128b08bd3SKris Buschelman   the above preallocation routines for simplicity.  One can also call MatConvert for an inplace
91228b08bd3SKris Buschelman   conversion to or from the MATSEQSBAIJ or MATMPISBAIJ type (depending on the communicator size)
91328b08bd3SKris Buschelman   without data copy.
91424b6179bSKris Buschelman 
91524b6179bSKris Buschelman   Options Database Keys:
9160bad9183SKris Buschelman + -mat_type sbaijmumps - sets the matrix type to "sbaijmumps" during a call to MatSetFromOptions()
91724b6179bSKris Buschelman . -mat_mumps_sym <0,1,2> - 0 the matrix is unsymmetric, 1 symmetric positive definite, 2 symmetric
91824b6179bSKris Buschelman . -mat_mumps_icntl_4 <0,...,4> - print level
91924b6179bSKris Buschelman . -mat_mumps_icntl_6 <0,...,7> - matrix prescaling options (see MUMPS User's Guide)
92024b6179bSKris Buschelman . -mat_mumps_icntl_7 <0,...,7> - matrix orderings (see MUMPS User's Guide)
92124b6179bSKris Buschelman . -mat_mumps_icntl_9 <1,2> - A or A^T x=b to be solved: 1 denotes A, 2 denotes A^T
92224b6179bSKris Buschelman . -mat_mumps_icntl_10 <n> - maximum number of iterative refinements
92324b6179bSKris Buschelman . -mat_mumps_icntl_11 <n> - error analysis, a positive value returns statistics during -sles_view
92424b6179bSKris Buschelman . -mat_mumps_icntl_12 <n> - efficiency control (see MUMPS User's Guide)
92524b6179bSKris Buschelman . -mat_mumps_icntl_13 <n> - efficiency control (see MUMPS User's Guide)
92624b6179bSKris Buschelman . -mat_mumps_icntl_14 <n> - efficiency control (see MUMPS User's Guide)
92724b6179bSKris Buschelman . -mat_mumps_icntl_15 <n> - efficiency control (see MUMPS User's Guide)
92824b6179bSKris Buschelman . -mat_mumps_cntl_1 <delta> - relative pivoting threshold
92924b6179bSKris Buschelman . -mat_mumps_cntl_2 <tol> - stopping criterion for refinement
93024b6179bSKris Buschelman - -mat_mumps_cntl_3 <adelta> - absolute pivoting threshold
93124b6179bSKris Buschelman 
93224b6179bSKris Buschelman   Level: beginner
93324b6179bSKris Buschelman 
93424b6179bSKris Buschelman .seealso: MATAIJMUMPS
93524b6179bSKris Buschelman M*/
93624b6179bSKris Buschelman 
937397b6df1SKris Buschelman EXTERN_C_BEGIN
938397b6df1SKris Buschelman #undef __FUNCT__
939f0c56d0fSKris Buschelman #define __FUNCT__ "MatCreate_SBAIJMUMPS"
940f0c56d0fSKris Buschelman int MatCreate_SBAIJMUMPS(Mat A) {
941397b6df1SKris Buschelman   int ierr,size;
942397b6df1SKris Buschelman 
943397b6df1SKris Buschelman   PetscFunctionBegin;
9445441df8eSKris Buschelman   /* Change type name before calling MatSetType to force proper construction of SeqSBAIJ or MPISBAIJ */
9455441df8eSKris Buschelman   /*   and SBAIJMUMPS types */
9465441df8eSKris Buschelman   ierr = PetscObjectChangeTypeName((PetscObject)A,MATSBAIJMUMPS);CHKERRQ(ierr);
9475441df8eSKris Buschelman   ierr = MPI_Comm_size(A->comm,&size);CHKERRQ(ierr);CHKERRQ(ierr);
948397b6df1SKris Buschelman   if (size == 1) {
949397b6df1SKris Buschelman     ierr = MatSetType(A,MATSEQSBAIJ);CHKERRQ(ierr);
950397b6df1SKris Buschelman   } else {
951397b6df1SKris Buschelman     ierr = MatSetType(A,MATMPISBAIJ);CHKERRQ(ierr);
952397b6df1SKris Buschelman   }
953f0c56d0fSKris Buschelman   ierr = MatConvert_SBAIJ_SBAIJMUMPS(A,MATSBAIJMUMPS,&A);CHKERRQ(ierr);
954397b6df1SKris Buschelman   PetscFunctionReturn(0);
955397b6df1SKris Buschelman }
956397b6df1SKris Buschelman EXTERN_C_END
957