xref: /petsc/src/mat/impls/aij/mpi/mumps/mumps.c (revision 91b026ca51aebd71e678a8998e7922b3a964a025)
1397b6df1SKris Buschelman /*
2c2b5dc30SHong Zhang     Provides an interface to the MUMPS sparse solver
3397b6df1SKris Buschelman */
467602552SJunchao Zhang #include <petscpkg_version.h>
59d0448ceSStefano Zampini #include <petscsf.h>
6c6db04a5SJed Brown #include <../src/mat/impls/aij/mpi/mpiaij.h> /*I  "petscmat.h"  I*/
7c6db04a5SJed Brown #include <../src/mat/impls/sbaij/mpi/mpisbaij.h>
87ee00b23SStefano Zampini #include <../src/mat/impls/sell/mpi/mpisell.h>
9397b6df1SKris Buschelman 
109261f6e4SBarry Smith #define MUMPS_MANUALS "(see users manual https://mumps-solver.org/index.php?page=doc \"Error and warning diagnostics\")"
119261f6e4SBarry Smith 
12397b6df1SKris Buschelman EXTERN_C_BEGIN
13397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
142907cef9SHong Zhang   #if defined(PETSC_USE_REAL_SINGLE)
152907cef9SHong Zhang     #include <cmumps_c.h>
162907cef9SHong Zhang   #else
17c6db04a5SJed Brown     #include <zmumps_c.h>
182907cef9SHong Zhang   #endif
192907cef9SHong Zhang #else
202907cef9SHong Zhang   #if defined(PETSC_USE_REAL_SINGLE)
212907cef9SHong Zhang     #include <smumps_c.h>
22397b6df1SKris Buschelman   #else
23c6db04a5SJed Brown     #include <dmumps_c.h>
24397b6df1SKris Buschelman   #endif
252907cef9SHong Zhang #endif
26397b6df1SKris Buschelman EXTERN_C_END
27397b6df1SKris Buschelman #define JOB_INIT         -1
28413bcc21SPierre Jolivet #define JOB_NULL         0
293d472b54SHong Zhang #define JOB_FACTSYMBOLIC 1
303d472b54SHong Zhang #define JOB_FACTNUMERIC  2
313d472b54SHong Zhang #define JOB_SOLVE        3
32397b6df1SKris Buschelman #define JOB_END          -2
333d472b54SHong Zhang 
342907cef9SHong Zhang /* calls to MUMPS */
352907cef9SHong Zhang #if defined(PETSC_USE_COMPLEX)
362907cef9SHong Zhang   #if defined(PETSC_USE_REAL_SINGLE)
373ab56b82SJunchao Zhang     #define MUMPS_c cmumps_c
382907cef9SHong Zhang   #else
393ab56b82SJunchao Zhang     #define MUMPS_c zmumps_c
402907cef9SHong Zhang   #endif
412907cef9SHong Zhang #else
422907cef9SHong Zhang   #if defined(PETSC_USE_REAL_SINGLE)
433ab56b82SJunchao Zhang     #define MUMPS_c smumps_c
442907cef9SHong Zhang   #else
453ab56b82SJunchao Zhang     #define MUMPS_c dmumps_c
462907cef9SHong Zhang   #endif
472907cef9SHong Zhang #endif
482907cef9SHong Zhang 
49a6053eceSJunchao Zhang /* MUMPS uses MUMPS_INT for nonzero indices such as irn/jcn, irn_loc/jcn_loc and uses int64_t for
50a6053eceSJunchao Zhang    number of nonzeros such as nnz, nnz_loc. We typedef MUMPS_INT to PetscMUMPSInt to follow the
51a6053eceSJunchao Zhang    naming convention in PetscMPIInt, PetscBLASInt etc.
52a6053eceSJunchao Zhang */
53a6053eceSJunchao Zhang typedef MUMPS_INT PetscMUMPSInt;
54a6053eceSJunchao Zhang 
5567602552SJunchao Zhang #if PETSC_PKG_MUMPS_VERSION_GE(5, 3, 0)
5667602552SJunchao Zhang   #if defined(MUMPS_INTSIZE64) /* MUMPS_INTSIZE64 is in MUMPS headers if it is built in full 64-bit mode, therefore the macro is more reliable */
57f0b74427SPierre Jolivet     #error "PETSc has not been tested with full 64-bit MUMPS and we choose to error out"
5867602552SJunchao Zhang   #endif
59a6053eceSJunchao Zhang #else
6067602552SJunchao Zhang   #if defined(INTSIZE64) /* INTSIZE64 is a command line macro one used to build MUMPS in full 64-bit mode */
61f0b74427SPierre Jolivet     #error "PETSc has not been tested with full 64-bit MUMPS and we choose to error out"
6267602552SJunchao Zhang   #endif
6367602552SJunchao Zhang #endif
6467602552SJunchao Zhang 
65a6053eceSJunchao Zhang #define MPIU_MUMPSINT       MPI_INT
66a6053eceSJunchao Zhang #define PETSC_MUMPS_INT_MAX 2147483647
67a6053eceSJunchao Zhang #define PETSC_MUMPS_INT_MIN -2147483648
68a6053eceSJunchao Zhang 
69a6053eceSJunchao Zhang /* Cast PetscInt to PetscMUMPSInt. Usually there is no overflow since <a> is row/col indices or some small integers*/
706497c311SBarry Smith static inline PetscErrorCode PetscMUMPSIntCast(PetscCount a, PetscMUMPSInt *b)
71d71ae5a4SJacob Faibussowitsch {
72a6053eceSJunchao Zhang   PetscFunctionBegin;
73ece88022SPierre Jolivet #if PetscDefined(USE_64BIT_INDICES)
742c71b3e2SJacob Faibussowitsch   PetscAssert(a <= PETSC_MUMPS_INT_MAX && a >= PETSC_MUMPS_INT_MIN, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "PetscInt too long for PetscMUMPSInt");
75ece88022SPierre Jolivet #endif
7657508eceSPierre Jolivet   *b = (PetscMUMPSInt)a;
773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
78a6053eceSJunchao Zhang }
79a6053eceSJunchao Zhang 
80a6053eceSJunchao Zhang /* Put these utility routines here since they are only used in this file */
81ce78bad3SBarry Smith static inline PetscErrorCode PetscOptionsMUMPSInt_Private(PetscOptionItems PetscOptionsObject, const char opt[], const char text[], const char man[], PetscMUMPSInt currentvalue, PetscMUMPSInt *value, PetscBool *set, PetscMUMPSInt lb, PetscMUMPSInt ub)
82d71ae5a4SJacob Faibussowitsch {
83a6053eceSJunchao Zhang   PetscInt  myval;
84a6053eceSJunchao Zhang   PetscBool myset;
854d86920dSPierre Jolivet 
86a6053eceSJunchao Zhang   PetscFunctionBegin;
87a6053eceSJunchao Zhang   /* PetscInt's size should be always >= PetscMUMPSInt's. It is safe to call PetscOptionsInt_Private to read a PetscMUMPSInt */
889566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt_Private(PetscOptionsObject, opt, text, man, (PetscInt)currentvalue, &myval, &myset, lb, ub));
899566063dSJacob Faibussowitsch   if (myset) PetscCall(PetscMUMPSIntCast(myval, value));
90a6053eceSJunchao Zhang   if (set) *set = myset;
913ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
92a6053eceSJunchao Zhang }
93a6053eceSJunchao Zhang #define PetscOptionsMUMPSInt(a, b, c, d, e, f) PetscOptionsMUMPSInt_Private(PetscOptionsObject, a, b, c, d, e, f, PETSC_MUMPS_INT_MIN, PETSC_MUMPS_INT_MAX)
94a6053eceSJunchao Zhang 
95217d3b1eSJunchao Zhang /* if using PETSc OpenMP support, we only call MUMPS on master ranks. Before/after the call, we change/restore CPUs the master ranks can run on */
963ab56b82SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
973ab56b82SJunchao Zhang   #define PetscMUMPS_c(mumps) \
983ab56b82SJunchao Zhang     do { \
993ab56b82SJunchao Zhang       if (mumps->use_petsc_omp_support) { \
1003ab56b82SJunchao Zhang         if (mumps->is_omp_master) { \
1019566063dSJacob Faibussowitsch           PetscCall(PetscOmpCtrlOmpRegionOnMasterBegin(mumps->omp_ctrl)); \
10214ffdc6fSStefano Zampini           PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); \
10314ffdc6fSStefano Zampini           PetscStackCallExternalVoid(PetscStringize(MUMPS_c), MUMPS_c(&mumps->id)); \
10414ffdc6fSStefano Zampini           PetscCall(PetscFPTrapPop()); \
1059566063dSJacob Faibussowitsch           PetscCall(PetscOmpCtrlOmpRegionOnMasterEnd(mumps->omp_ctrl)); \
1063ab56b82SJunchao Zhang         } \
1079566063dSJacob Faibussowitsch         PetscCall(PetscOmpCtrlBarrier(mumps->omp_ctrl)); \
108c3714a1dSJunchao Zhang         /* Global info is same on all processes so we Bcast it within omp_comm. Local info is specific      \
109c3714a1dSJunchao Zhang          to processes, so we only Bcast info[1], an error code and leave others (since they do not have   \
110c3714a1dSJunchao Zhang          an easy translation between omp_comm and petsc_comm). See MUMPS-5.1.2 manual p82.                   \
111c3714a1dSJunchao Zhang          omp_comm is a small shared memory communicator, hence doing multiple Bcast as shown below is OK. \
112c3714a1dSJunchao Zhang       */ \
113338d3105SPierre Jolivet         PetscCallMPI(MPI_Bcast(mumps->id.infog, PETSC_STATIC_ARRAY_LENGTH(mumps->id.infog), MPIU_MUMPSINT, 0, mumps->omp_comm)); \
114338d3105SPierre Jolivet         PetscCallMPI(MPI_Bcast(mumps->id.rinfog, PETSC_STATIC_ARRAY_LENGTH(mumps->id.rinfog), MPIU_REAL, 0, mumps->omp_comm)); \
115338d3105SPierre Jolivet         PetscCallMPI(MPI_Bcast(mumps->id.info, PETSC_STATIC_ARRAY_LENGTH(mumps->id.info), MPIU_MUMPSINT, 0, mumps->omp_comm)); \
116338d3105SPierre Jolivet         PetscCallMPI(MPI_Bcast(mumps->id.rinfo, PETSC_STATIC_ARRAY_LENGTH(mumps->id.rinfo), MPIU_REAL, 0, mumps->omp_comm)); \
1173ab56b82SJunchao Zhang       } else { \
11814ffdc6fSStefano Zampini         PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); \
11914ffdc6fSStefano Zampini         PetscStackCallExternalVoid(PetscStringize(MUMPS_c), MUMPS_c(&mumps->id)); \
12014ffdc6fSStefano Zampini         PetscCall(PetscFPTrapPop()); \
1213ab56b82SJunchao Zhang       } \
1223ab56b82SJunchao Zhang     } while (0)
1233ab56b82SJunchao Zhang #else
1243ab56b82SJunchao Zhang   #define PetscMUMPS_c(mumps) \
125d71ae5a4SJacob Faibussowitsch     do { \
12614ffdc6fSStefano Zampini       PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); \
12714ffdc6fSStefano Zampini       PetscStackCallExternalVoid(PetscStringize(MUMPS_c), MUMPS_c(&mumps->id)); \
12814ffdc6fSStefano Zampini       PetscCall(PetscFPTrapPop()); \
129d71ae5a4SJacob Faibussowitsch     } while (0)
1303ab56b82SJunchao Zhang #endif
1313ab56b82SJunchao Zhang 
132940cd9d6SSatish Balay /* declare MumpsScalar */
133940cd9d6SSatish Balay #if defined(PETSC_USE_COMPLEX)
134940cd9d6SSatish Balay   #if defined(PETSC_USE_REAL_SINGLE)
135940cd9d6SSatish Balay     #define MumpsScalar mumps_complex
136940cd9d6SSatish Balay   #else
137940cd9d6SSatish Balay     #define MumpsScalar mumps_double_complex
138940cd9d6SSatish Balay   #endif
139940cd9d6SSatish Balay #else
140940cd9d6SSatish Balay   #define MumpsScalar PetscScalar
141940cd9d6SSatish Balay #endif
1423d472b54SHong Zhang 
143397b6df1SKris Buschelman /* macros s.t. indices match MUMPS documentation */
144397b6df1SKris Buschelman #define ICNTL(I)  icntl[(I) - 1]
145397b6df1SKris Buschelman #define CNTL(I)   cntl[(I) - 1]
146397b6df1SKris Buschelman #define INFOG(I)  infog[(I) - 1]
147a7aca84bSHong Zhang #define INFO(I)   info[(I) - 1]
148397b6df1SKris Buschelman #define RINFOG(I) rinfog[(I) - 1]
149adc1d99fSHong Zhang #define RINFO(I)  rinfo[(I) - 1]
150397b6df1SKris Buschelman 
151a6053eceSJunchao Zhang typedef struct Mat_MUMPS Mat_MUMPS;
152a6053eceSJunchao Zhang struct Mat_MUMPS {
153397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
1542907cef9SHong Zhang   #if defined(PETSC_USE_REAL_SINGLE)
1552907cef9SHong Zhang   CMUMPS_STRUC_C id;
1562907cef9SHong Zhang   #else
157397b6df1SKris Buschelman   ZMUMPS_STRUC_C id;
1582907cef9SHong Zhang   #endif
1592907cef9SHong Zhang #else
1602907cef9SHong Zhang   #if defined(PETSC_USE_REAL_SINGLE)
1612907cef9SHong Zhang   SMUMPS_STRUC_C id;
162397b6df1SKris Buschelman   #else
163397b6df1SKris Buschelman   DMUMPS_STRUC_C id;
164397b6df1SKris Buschelman   #endif
1652907cef9SHong Zhang #endif
1662907cef9SHong Zhang 
167397b6df1SKris Buschelman   MatStructure   matstruc;
1682d4298aeSJunchao Zhang   PetscMPIInt    myid, petsc_size;
169a6053eceSJunchao Zhang   PetscMUMPSInt *irn, *jcn;       /* the (i,j,v) triplets passed to mumps. */
170a6053eceSJunchao Zhang   PetscScalar   *val, *val_alloc; /* For some matrices, we can directly access their data array without a buffer. For others, we need a buffer. So comes val_alloc. */
1716497c311SBarry Smith   PetscCount     nnz;             /* number of nonzeros. The type is called selective 64-bit in mumps */
172a6053eceSJunchao Zhang   PetscMUMPSInt  sym;
1732d4298aeSJunchao Zhang   MPI_Comm       mumps_comm;
174413bcc21SPierre Jolivet   PetscMUMPSInt *ICNTL_pre;
175413bcc21SPierre Jolivet   PetscReal     *CNTL_pre;
176a6053eceSJunchao Zhang   PetscMUMPSInt  ICNTL9_pre;         /* check if ICNTL(9) is changed from previous MatSolve */
177801fbe65SHong Zhang   VecScatter     scat_rhs, scat_sol; /* used by MatSolve() */
17825aac85cSJunchao Zhang   PetscMUMPSInt  ICNTL20;            /* use centralized (0) or distributed (10) dense RHS */
17967602552SJunchao Zhang   PetscMUMPSInt  lrhs_loc, nloc_rhs, *irhs_loc;
18067602552SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
18167602552SJunchao Zhang   PetscInt    *rhs_nrow, max_nrhs;
18267602552SJunchao Zhang   PetscMPIInt *rhs_recvcounts, *rhs_disps;
18367602552SJunchao Zhang   PetscScalar *rhs_loc, *rhs_recvbuf;
18467602552SJunchao Zhang #endif
185801fbe65SHong Zhang   Vec            b_seq, x_seq;
186a6053eceSJunchao Zhang   PetscInt       ninfo, *info; /* which INFO to display */
187b5fa320bSStefano Zampini   PetscInt       sizeredrhs;
18859ac8732SStefano Zampini   PetscScalar   *schur_sol;
18959ac8732SStefano Zampini   PetscInt       schur_sizesol;
190a6053eceSJunchao Zhang   PetscMUMPSInt *ia_alloc, *ja_alloc; /* work arrays used for the CSR struct for sparse rhs */
1916497c311SBarry Smith   PetscCount     cur_ilen, cur_jlen;  /* current len of ia_alloc[], ja_alloc[] */
192a6053eceSJunchao Zhang   PetscErrorCode (*ConvertToTriples)(Mat, PetscInt, MatReuse, Mat_MUMPS *);
1932205254eSKarl Rupp 
1949d0448ceSStefano Zampini   /* Support for MATNEST */
1959d0448ceSStefano Zampini   PetscErrorCode (**nest_convert_to_triples)(Mat, PetscInt, MatReuse, Mat_MUMPS *);
1966497c311SBarry Smith   PetscCount  *nest_vals_start;
1979d0448ceSStefano Zampini   PetscScalar *nest_vals;
1989d0448ceSStefano Zampini 
199a6053eceSJunchao Zhang   /* stuff used by petsc/mumps OpenMP support*/
2003ab56b82SJunchao Zhang   PetscBool    use_petsc_omp_support;
201da81f932SPierre Jolivet   PetscOmpCtrl omp_ctrl;             /* an OpenMP controller that blocked processes will release their CPU (MPI_Barrier does not have this guarantee) */
202f0b74427SPierre Jolivet   MPI_Comm     petsc_comm, omp_comm; /* petsc_comm is PETSc matrix's comm */
2036497c311SBarry Smith   PetscCount  *recvcount;            /* a collection of nnz on omp_master */
204a6053eceSJunchao Zhang   PetscMPIInt  tag, omp_comm_size;
2053ab56b82SJunchao Zhang   PetscBool    is_omp_master; /* is this rank the master of omp_comm */
206a6053eceSJunchao Zhang   MPI_Request *reqs;
207a6053eceSJunchao Zhang };
2083ab56b82SJunchao Zhang 
209a6053eceSJunchao Zhang /* Cast a 1-based CSR represented by (nrow, ia, ja) of type PetscInt to a CSR of type PetscMUMPSInt.
210a6053eceSJunchao Zhang    Here, nrow is number of rows, ia[] is row pointer and ja[] is column indices.
211a6053eceSJunchao Zhang  */
212d2a308c1SPierre Jolivet static PetscErrorCode PetscMUMPSIntCSRCast(PETSC_UNUSED Mat_MUMPS *mumps, PetscInt nrow, PetscInt *ia, PetscInt *ja, PetscMUMPSInt **ia_mumps, PetscMUMPSInt **ja_mumps, PetscMUMPSInt *nnz_mumps)
213d71ae5a4SJacob Faibussowitsch {
2146497c311SBarry Smith   PetscInt nnz = ia[nrow] - 1; /* mumps uses 1-based indices. Uses PetscInt instead of PetscCount since mumps only uses PetscMUMPSInt for rhs */
215f0c56d0fSKris Buschelman 
216a6053eceSJunchao Zhang   PetscFunctionBegin;
217a6053eceSJunchao Zhang #if defined(PETSC_USE_64BIT_INDICES)
218a6053eceSJunchao Zhang   {
219a6053eceSJunchao Zhang     PetscInt i;
220a6053eceSJunchao Zhang     if (nrow + 1 > mumps->cur_ilen) { /* realloc ia_alloc/ja_alloc to fit ia/ja */
2219566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->ia_alloc));
2229566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(nrow + 1, &mumps->ia_alloc));
223a6053eceSJunchao Zhang       mumps->cur_ilen = nrow + 1;
224a6053eceSJunchao Zhang     }
225a6053eceSJunchao Zhang     if (nnz > mumps->cur_jlen) {
2269566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->ja_alloc));
2279566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(nnz, &mumps->ja_alloc));
228a6053eceSJunchao Zhang       mumps->cur_jlen = nnz;
229a6053eceSJunchao Zhang     }
230f4f49eeaSPierre Jolivet     for (i = 0; i < nrow + 1; i++) PetscCall(PetscMUMPSIntCast(ia[i], &mumps->ia_alloc[i]));
231f4f49eeaSPierre Jolivet     for (i = 0; i < nnz; i++) PetscCall(PetscMUMPSIntCast(ja[i], &mumps->ja_alloc[i]));
232a6053eceSJunchao Zhang     *ia_mumps = mumps->ia_alloc;
233a6053eceSJunchao Zhang     *ja_mumps = mumps->ja_alloc;
234a6053eceSJunchao Zhang   }
235a6053eceSJunchao Zhang #else
236a6053eceSJunchao Zhang   *ia_mumps = ia;
237a6053eceSJunchao Zhang   *ja_mumps = ja;
238a6053eceSJunchao Zhang #endif
2399566063dSJacob Faibussowitsch   PetscCall(PetscMUMPSIntCast(nnz, nnz_mumps));
2403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
241a6053eceSJunchao Zhang }
242b24902e0SBarry Smith 
243d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMumpsResetSchur_Private(Mat_MUMPS *mumps)
244d71ae5a4SJacob Faibussowitsch {
245b5fa320bSStefano Zampini   PetscFunctionBegin;
2469566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.listvar_schur));
2479566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.redrhs));
2489566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->schur_sol));
24959ac8732SStefano Zampini   mumps->id.size_schur = 0;
250b3cb21ddSStefano Zampini   mumps->id.schur_lld  = 0;
25159ac8732SStefano Zampini   mumps->id.ICNTL(19)  = 0;
2523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
25359ac8732SStefano Zampini }
25459ac8732SStefano Zampini 
255b3cb21ddSStefano Zampini /* solve with rhs in mumps->id.redrhs and return in the same location */
256d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMumpsSolveSchur_Private(Mat F)
257d71ae5a4SJacob Faibussowitsch {
258b3cb21ddSStefano Zampini   Mat_MUMPS           *mumps = (Mat_MUMPS *)F->data;
259b3cb21ddSStefano Zampini   Mat                  S, B, X;
260b3cb21ddSStefano Zampini   MatFactorSchurStatus schurstatus;
261b3cb21ddSStefano Zampini   PetscInt             sizesol;
26259ac8732SStefano Zampini 
26359ac8732SStefano Zampini   PetscFunctionBegin;
2649566063dSJacob Faibussowitsch   PetscCall(MatFactorFactorizeSchurComplement(F));
2659566063dSJacob Faibussowitsch   PetscCall(MatFactorGetSchurComplement(F, &S, &schurstatus));
2669566063dSJacob Faibussowitsch   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, mumps->id.size_schur, mumps->id.nrhs, (PetscScalar *)mumps->id.redrhs, &B));
2679566063dSJacob Faibussowitsch   PetscCall(MatSetType(B, ((PetscObject)S)->type_name));
268a3d589ffSStefano Zampini #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
2699566063dSJacob Faibussowitsch   PetscCall(MatBindToCPU(B, S->boundtocpu));
270a3d589ffSStefano Zampini #endif
271b3cb21ddSStefano Zampini   switch (schurstatus) {
272d71ae5a4SJacob Faibussowitsch   case MAT_FACTOR_SCHUR_FACTORED:
273d71ae5a4SJacob Faibussowitsch     PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, mumps->id.size_schur, mumps->id.nrhs, (PetscScalar *)mumps->id.redrhs, &X));
274d71ae5a4SJacob Faibussowitsch     PetscCall(MatSetType(X, ((PetscObject)S)->type_name));
275a3d589ffSStefano Zampini #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
2769566063dSJacob Faibussowitsch     PetscCall(MatBindToCPU(X, S->boundtocpu));
277a3d589ffSStefano Zampini #endif
278b3cb21ddSStefano Zampini     if (!mumps->id.ICNTL(9)) { /* transpose solve */
2799566063dSJacob Faibussowitsch       PetscCall(MatMatSolveTranspose(S, B, X));
28059ac8732SStefano Zampini     } else {
2819566063dSJacob Faibussowitsch       PetscCall(MatMatSolve(S, B, X));
28259ac8732SStefano Zampini     }
283b3cb21ddSStefano Zampini     break;
284b3cb21ddSStefano Zampini   case MAT_FACTOR_SCHUR_INVERTED:
285b3cb21ddSStefano Zampini     sizesol = mumps->id.nrhs * mumps->id.size_schur;
28659ac8732SStefano Zampini     if (!mumps->schur_sol || sizesol > mumps->schur_sizesol) {
2879566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->schur_sol));
2889566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(sizesol, &mumps->schur_sol));
28959ac8732SStefano Zampini       mumps->schur_sizesol = sizesol;
290b5fa320bSStefano Zampini     }
2919566063dSJacob Faibussowitsch     PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, mumps->id.size_schur, mumps->id.nrhs, mumps->schur_sol, &X));
2929566063dSJacob Faibussowitsch     PetscCall(MatSetType(X, ((PetscObject)S)->type_name));
293a3d589ffSStefano Zampini #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
2949566063dSJacob Faibussowitsch     PetscCall(MatBindToCPU(X, S->boundtocpu));
295a3d589ffSStefano Zampini #endif
2969566063dSJacob Faibussowitsch     PetscCall(MatProductCreateWithMat(S, B, NULL, X));
29759ac8732SStefano Zampini     if (!mumps->id.ICNTL(9)) { /* transpose solve */
2989566063dSJacob Faibussowitsch       PetscCall(MatProductSetType(X, MATPRODUCT_AtB));
299b5fa320bSStefano Zampini     } else {
3009566063dSJacob Faibussowitsch       PetscCall(MatProductSetType(X, MATPRODUCT_AB));
301b5fa320bSStefano Zampini     }
3029566063dSJacob Faibussowitsch     PetscCall(MatProductSetFromOptions(X));
3039566063dSJacob Faibussowitsch     PetscCall(MatProductSymbolic(X));
3049566063dSJacob Faibussowitsch     PetscCall(MatProductNumeric(X));
3054417c5e8SHong Zhang 
3069566063dSJacob Faibussowitsch     PetscCall(MatCopy(X, B, SAME_NONZERO_PATTERN));
307b3cb21ddSStefano Zampini     break;
308d71ae5a4SJacob Faibussowitsch   default:
309d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)F), PETSC_ERR_SUP, "Unhandled MatFactorSchurStatus %d", F->schur_status);
31059ac8732SStefano Zampini   }
3119566063dSJacob Faibussowitsch   PetscCall(MatFactorRestoreSchurComplement(F, &S, schurstatus));
3129566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&B));
3139566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&X));
3143ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
315b5fa320bSStefano Zampini }
316b5fa320bSStefano Zampini 
317d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMumpsHandleSchur_Private(Mat F, PetscBool expansion)
318d71ae5a4SJacob Faibussowitsch {
319b3cb21ddSStefano Zampini   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
320b5fa320bSStefano Zampini 
321b5fa320bSStefano Zampini   PetscFunctionBegin;
322b5fa320bSStefano Zampini   if (!mumps->id.ICNTL(19)) { /* do nothing when Schur complement has not been computed */
3233ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
324b5fa320bSStefano Zampini   }
325b8f61ee1SStefano Zampini   if (!expansion) { /* prepare for the condensation step */
326b5fa320bSStefano Zampini     PetscInt sizeredrhs = mumps->id.nrhs * mumps->id.size_schur;
327b5fa320bSStefano Zampini     /* allocate MUMPS internal array to store reduced right-hand sides */
328b5fa320bSStefano Zampini     if (!mumps->id.redrhs || sizeredrhs > mumps->sizeredrhs) {
3299566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->id.redrhs));
330b5fa320bSStefano Zampini       mumps->id.lredrhs = mumps->id.size_schur;
3319566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(mumps->id.nrhs * mumps->id.lredrhs, &mumps->id.redrhs));
332b5fa320bSStefano Zampini       mumps->sizeredrhs = mumps->id.nrhs * mumps->id.lredrhs;
333b5fa320bSStefano Zampini     }
334b5fa320bSStefano Zampini   } else { /* prepare for the expansion step */
335b8f61ee1SStefano Zampini     /* solve Schur complement (this has to be done by the MUMPS user, so basically us) */
3369566063dSJacob Faibussowitsch     PetscCall(MatMumpsSolveSchur_Private(F));
337b5fa320bSStefano Zampini     mumps->id.ICNTL(26) = 2; /* expansion phase */
3383ab56b82SJunchao Zhang     PetscMUMPS_c(mumps);
3399261f6e4SBarry Smith     PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in solve: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
340b5fa320bSStefano Zampini     /* restore defaults */
341b5fa320bSStefano Zampini     mumps->id.ICNTL(26) = -1;
342d3d598ffSStefano Zampini     /* free MUMPS internal array for redrhs if we have solved for multiple rhs in order to save memory space */
343d3d598ffSStefano Zampini     if (mumps->id.nrhs > 1) {
3449566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->id.redrhs));
345d3d598ffSStefano Zampini       mumps->id.lredrhs = 0;
346d3d598ffSStefano Zampini       mumps->sizeredrhs = 0;
347d3d598ffSStefano Zampini     }
348b5fa320bSStefano Zampini   }
3493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
350b5fa320bSStefano Zampini }
351b5fa320bSStefano Zampini 
352397b6df1SKris Buschelman /*
353f0b74427SPierre Jolivet   MatConvertToTriples_A_B - convert PETSc matrix to triples: row[nz], col[nz], val[nz]
354d341cd04SHong Zhang 
355397b6df1SKris Buschelman   input:
35675480915SPierre Jolivet     A       - matrix in aij,baij or sbaij format
357397b6df1SKris Buschelman     shift   - 0: C style output triple; 1: Fortran style output triple.
358bccb9932SShri Abhyankar     reuse   - MAT_INITIAL_MATRIX: spaces are allocated and values are set for the triple
359bccb9932SShri Abhyankar               MAT_REUSE_MATRIX:   only the values in v array are updated
360397b6df1SKris Buschelman   output:
361397b6df1SKris Buschelman     nnz     - dim of r, c, and v (number of local nonzero entries of A)
362397b6df1SKris Buschelman     r, c, v - row and col index, matrix values (matrix triples)
363eb9baa12SBarry Smith 
364eb9baa12SBarry Smith   The returned values r, c, and sometimes v are obtained in a single PetscMalloc(). Then in MatDestroy_MUMPS() it is
3657ee00b23SStefano Zampini   freed with PetscFree(mumps->irn);  This is not ideal code, the fact that v is ONLY sometimes part of mumps->irn means
366eb9baa12SBarry Smith   that the PetscMalloc() cannot easily be replaced with a PetscMalloc3().
367eb9baa12SBarry Smith 
368397b6df1SKris Buschelman  */
36916ebf90aSShri Abhyankar 
37066976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqaij_seqaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
371d71ae5a4SJacob Faibussowitsch {
372a3d589ffSStefano Zampini   const PetscScalar *av;
373185f6596SHong Zhang   const PetscInt    *ai, *aj, *ajj, M = A->rmap->n;
3746497c311SBarry Smith   PetscCount         nz, rnz, k;
375a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
37616ebf90aSShri Abhyankar   Mat_SeqAIJ        *aa = (Mat_SeqAIJ *)A->data;
377397b6df1SKris Buschelman 
378397b6df1SKris Buschelman   PetscFunctionBegin;
3799566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(A, &av));
380bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
3812205254eSKarl Rupp     nz = aa->nz;
3822205254eSKarl Rupp     ai = aa->i;
3832205254eSKarl Rupp     aj = aa->j;
3849566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
3856497c311SBarry Smith     for (PetscCount i = k = 0; i < M; i++) {
38616ebf90aSShri Abhyankar       rnz = ai[i + 1] - ai[i];
38767877ebaSShri Abhyankar       ajj = aj + ai[i];
3886497c311SBarry Smith       for (PetscCount j = 0; j < rnz; j++) {
3899566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(i + shift, &row[k]));
3909566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[k]));
391a6053eceSJunchao Zhang         k++;
39216ebf90aSShri Abhyankar       }
39316ebf90aSShri Abhyankar     }
39450c845baSStefano Zampini     mumps->val = (PetscScalar *)av;
395a6053eceSJunchao Zhang     mumps->irn = row;
396a6053eceSJunchao Zhang     mumps->jcn = col;
397a6053eceSJunchao Zhang     mumps->nnz = nz;
398127cd276SPierre Jolivet   } else if (mumps->nest_vals) PetscCall(PetscArraycpy(mumps->val, av, aa->nz)); /* MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_seqaij_seqaij(), so one needs to copy the memory */
399127cd276SPierre Jolivet   else mumps->val = (PetscScalar *)av;                                           /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
4009566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(A, &av));
4013ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
40216ebf90aSShri Abhyankar }
403397b6df1SKris Buschelman 
40466976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqsell_seqaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
405d71ae5a4SJacob Faibussowitsch {
4066497c311SBarry Smith   PetscCount     nz, i, j, k, r;
4077ee00b23SStefano Zampini   Mat_SeqSELL   *a = (Mat_SeqSELL *)A->data;
408a6053eceSJunchao Zhang   PetscMUMPSInt *row, *col;
4097ee00b23SStefano Zampini 
4107ee00b23SStefano Zampini   PetscFunctionBegin;
4117ee00b23SStefano Zampini   nz = a->sliidx[a->totalslices];
41250c845baSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
4139566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
414a6053eceSJunchao Zhang     for (i = k = 0; i < a->totalslices; i++) {
41548a46eb9SPierre Jolivet       for (j = a->sliidx[i], r = 0; j < a->sliidx[i + 1]; j++, r = ((r + 1) & 0x07)) PetscCall(PetscMUMPSIntCast(8 * i + r + shift, &row[k++]));
4167ee00b23SStefano Zampini     }
4179566063dSJacob Faibussowitsch     for (i = 0; i < nz; i++) PetscCall(PetscMUMPSIntCast(a->colidx[i] + shift, &col[i]));
418a6053eceSJunchao Zhang     mumps->irn = row;
419a6053eceSJunchao Zhang     mumps->jcn = col;
420a6053eceSJunchao Zhang     mumps->nnz = nz;
42150c845baSStefano Zampini     mumps->val = a->val;
422127cd276SPierre Jolivet   } else if (mumps->nest_vals) PetscCall(PetscArraycpy(mumps->val, a->val, nz)); /* MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_seqsell_seqaij(), so one needs to copy the memory */
423127cd276SPierre Jolivet   else mumps->val = a->val;                                                      /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
4243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4257ee00b23SStefano Zampini }
4267ee00b23SStefano Zampini 
42766976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqbaij_seqaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
428d71ae5a4SJacob Faibussowitsch {
42967877ebaSShri Abhyankar   Mat_SeqBAIJ    *aa = (Mat_SeqBAIJ *)A->data;
43033d57670SJed Brown   const PetscInt *ai, *aj, *ajj, bs2 = aa->bs2;
4316497c311SBarry Smith   PetscCount      M, nz = bs2 * aa->nz, idx = 0, rnz, i, j, k, m;
432a6053eceSJunchao Zhang   PetscInt        bs;
433a6053eceSJunchao Zhang   PetscMUMPSInt  *row, *col;
43467877ebaSShri Abhyankar 
43567877ebaSShri Abhyankar   PetscFunctionBegin;
43650c845baSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
4379566063dSJacob Faibussowitsch     PetscCall(MatGetBlockSize(A, &bs));
43833d57670SJed Brown     M  = A->rmap->N / bs;
4399371c9d4SSatish Balay     ai = aa->i;
4409371c9d4SSatish Balay     aj = aa->j;
4419566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
44267877ebaSShri Abhyankar     for (i = 0; i < M; i++) {
44367877ebaSShri Abhyankar       ajj = aj + ai[i];
44467877ebaSShri Abhyankar       rnz = ai[i + 1] - ai[i];
44567877ebaSShri Abhyankar       for (k = 0; k < rnz; k++) {
44667877ebaSShri Abhyankar         for (j = 0; j < bs; j++) {
44767877ebaSShri Abhyankar           for (m = 0; m < bs; m++) {
4489566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(i * bs + m + shift, &row[idx]));
4499566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(bs * ajj[k] + j + shift, &col[idx]));
450a6053eceSJunchao Zhang             idx++;
45167877ebaSShri Abhyankar           }
45267877ebaSShri Abhyankar         }
45367877ebaSShri Abhyankar       }
45467877ebaSShri Abhyankar     }
455a6053eceSJunchao Zhang     mumps->irn = row;
456a6053eceSJunchao Zhang     mumps->jcn = col;
457a6053eceSJunchao Zhang     mumps->nnz = nz;
45850c845baSStefano Zampini     mumps->val = aa->a;
459127cd276SPierre Jolivet   } else if (mumps->nest_vals) PetscCall(PetscArraycpy(mumps->val, aa->a, nz)); /* MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_seqbaij_seqaij(), so one needs to copy the memory */
460127cd276SPierre Jolivet   else mumps->val = aa->a;                                                      /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
4613ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
46267877ebaSShri Abhyankar }
46367877ebaSShri Abhyankar 
46466976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqsbaij_seqsbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
465d71ae5a4SJacob Faibussowitsch {
46675480915SPierre Jolivet   const PetscInt *ai, *aj, *ajj;
467a6053eceSJunchao Zhang   PetscInt        bs;
4686497c311SBarry Smith   PetscCount      nz, rnz, i, j, k, m;
469a6053eceSJunchao Zhang   PetscMUMPSInt  *row, *col;
47075480915SPierre Jolivet   PetscScalar    *val;
47116ebf90aSShri Abhyankar   Mat_SeqSBAIJ   *aa  = (Mat_SeqSBAIJ *)A->data;
47275480915SPierre Jolivet   const PetscInt  bs2 = aa->bs2, mbs = aa->mbs;
47338548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
474b94d7dedSBarry Smith   PetscBool isset, hermitian;
47538548759SBarry Smith #endif
47616ebf90aSShri Abhyankar 
47716ebf90aSShri Abhyankar   PetscFunctionBegin;
47838548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
479b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
480b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
48138548759SBarry Smith #endif
4822205254eSKarl Rupp   ai = aa->i;
4832205254eSKarl Rupp   aj = aa->j;
4849566063dSJacob Faibussowitsch   PetscCall(MatGetBlockSize(A, &bs));
48575480915SPierre Jolivet   if (reuse == MAT_INITIAL_MATRIX) {
4866497c311SBarry Smith     const PetscCount alloc_size = aa->nz * bs2;
487f3fa974cSJacob Faibussowitsch 
488f3fa974cSJacob Faibussowitsch     PetscCall(PetscMalloc2(alloc_size, &row, alloc_size, &col));
489a6053eceSJunchao Zhang     if (bs > 1) {
490f3fa974cSJacob Faibussowitsch       PetscCall(PetscMalloc1(alloc_size, &mumps->val_alloc));
491a6053eceSJunchao Zhang       mumps->val = mumps->val_alloc;
49275480915SPierre Jolivet     } else {
493a6053eceSJunchao Zhang       mumps->val = aa->a;
49475480915SPierre Jolivet     }
495a6053eceSJunchao Zhang     mumps->irn = row;
496a6053eceSJunchao Zhang     mumps->jcn = col;
497a6053eceSJunchao Zhang   } else {
498a6053eceSJunchao Zhang     row = mumps->irn;
499a6053eceSJunchao Zhang     col = mumps->jcn;
500a6053eceSJunchao Zhang   }
501a6053eceSJunchao Zhang   val = mumps->val;
502185f6596SHong Zhang 
50316ebf90aSShri Abhyankar   nz = 0;
504a81fe166SPierre Jolivet   if (bs > 1) {
50575480915SPierre Jolivet     for (i = 0; i < mbs; i++) {
50616ebf90aSShri Abhyankar       rnz = ai[i + 1] - ai[i];
50767877ebaSShri Abhyankar       ajj = aj + ai[i];
50875480915SPierre Jolivet       for (j = 0; j < rnz; j++) {
50975480915SPierre Jolivet         for (k = 0; k < bs; k++) {
51075480915SPierre Jolivet           for (m = 0; m < bs; m++) {
511ec4f40fdSPierre Jolivet             if (ajj[j] > i || k >= m) {
51275480915SPierre Jolivet               if (reuse == MAT_INITIAL_MATRIX) {
5139566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(i * bs + m + shift, &row[nz]));
5149566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(ajj[j] * bs + k + shift, &col[nz]));
51575480915SPierre Jolivet               }
51675480915SPierre Jolivet               val[nz++] = aa->a[(ai[i] + j) * bs2 + m + k * bs];
51775480915SPierre Jolivet             }
51875480915SPierre Jolivet           }
51975480915SPierre Jolivet         }
52075480915SPierre Jolivet       }
52175480915SPierre Jolivet     }
522a81fe166SPierre Jolivet   } else if (reuse == MAT_INITIAL_MATRIX) {
523a81fe166SPierre Jolivet     for (i = 0; i < mbs; i++) {
524a81fe166SPierre Jolivet       rnz = ai[i + 1] - ai[i];
525a81fe166SPierre Jolivet       ajj = aj + ai[i];
526a81fe166SPierre Jolivet       for (j = 0; j < rnz; j++) {
5279566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
5289566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[nz]));
529a6053eceSJunchao Zhang         nz++;
530a81fe166SPierre Jolivet       }
531a81fe166SPierre Jolivet     }
5326497c311SBarry Smith     PetscCheck(nz == aa->nz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Different numbers of nonzeros %" PetscCount_FMT " != %" PetscInt_FMT, nz, aa->nz);
533127cd276SPierre Jolivet   } else if (mumps->nest_vals)
534127cd276SPierre Jolivet     PetscCall(PetscArraycpy(mumps->val, aa->a, aa->nz)); /* bs == 1 and MAT_REUSE_MATRIX, MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_seqsbaij_seqsbaij(), so one needs to copy the memory */
535127cd276SPierre Jolivet   else mumps->val = aa->a;                               /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
536a6053eceSJunchao Zhang   if (reuse == MAT_INITIAL_MATRIX) mumps->nnz = nz;
5373ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
53816ebf90aSShri Abhyankar }
53916ebf90aSShri Abhyankar 
54066976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqaij_seqsbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
541d71ae5a4SJacob Faibussowitsch {
54267877ebaSShri Abhyankar   const PetscInt    *ai, *aj, *ajj, *adiag, M = A->rmap->n;
5436497c311SBarry Smith   PetscCount         nz, rnz, i, j;
54467877ebaSShri Abhyankar   const PetscScalar *av, *v1;
54516ebf90aSShri Abhyankar   PetscScalar       *val;
546a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
547829b1710SHong Zhang   Mat_SeqAIJ        *aa = (Mat_SeqAIJ *)A->data;
54829b521d4Sstefano_zampini   PetscBool          missing;
54938548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
550b94d7dedSBarry Smith   PetscBool hermitian, isset;
55138548759SBarry Smith #endif
55216ebf90aSShri Abhyankar 
55316ebf90aSShri Abhyankar   PetscFunctionBegin;
55438548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
555b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
556b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
55738548759SBarry Smith #endif
5589566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(A, &av));
5599371c9d4SSatish Balay   ai    = aa->i;
5609371c9d4SSatish Balay   aj    = aa->j;
56116ebf90aSShri Abhyankar   adiag = aa->diag;
5629566063dSJacob Faibussowitsch   PetscCall(MatMissingDiagonal_SeqAIJ(A, &missing, NULL));
563bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
5647ee00b23SStefano Zampini     /* count nz in the upper triangular part of A */
565829b1710SHong Zhang     nz = 0;
56629b521d4Sstefano_zampini     if (missing) {
56729b521d4Sstefano_zampini       for (i = 0; i < M; i++) {
56829b521d4Sstefano_zampini         if (PetscUnlikely(adiag[i] >= ai[i + 1])) {
56929b521d4Sstefano_zampini           for (j = ai[i]; j < ai[i + 1]; j++) {
57029b521d4Sstefano_zampini             if (aj[j] < i) continue;
57129b521d4Sstefano_zampini             nz++;
57229b521d4Sstefano_zampini           }
57329b521d4Sstefano_zampini         } else {
57429b521d4Sstefano_zampini           nz += ai[i + 1] - adiag[i];
57529b521d4Sstefano_zampini         }
57629b521d4Sstefano_zampini       }
57729b521d4Sstefano_zampini     } else {
578829b1710SHong Zhang       for (i = 0; i < M; i++) nz += ai[i + 1] - adiag[i];
57929b521d4Sstefano_zampini     }
5809566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
5819566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
582a6053eceSJunchao Zhang     mumps->nnz = nz;
583a6053eceSJunchao Zhang     mumps->irn = row;
584a6053eceSJunchao Zhang     mumps->jcn = col;
585a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
586185f6596SHong Zhang 
58716ebf90aSShri Abhyankar     nz = 0;
58829b521d4Sstefano_zampini     if (missing) {
58929b521d4Sstefano_zampini       for (i = 0; i < M; i++) {
59029b521d4Sstefano_zampini         if (PetscUnlikely(adiag[i] >= ai[i + 1])) {
59129b521d4Sstefano_zampini           for (j = ai[i]; j < ai[i + 1]; j++) {
59229b521d4Sstefano_zampini             if (aj[j] < i) continue;
5939566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
5949566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(aj[j] + shift, &col[nz]));
59529b521d4Sstefano_zampini             val[nz] = av[j];
59629b521d4Sstefano_zampini             nz++;
59729b521d4Sstefano_zampini           }
59829b521d4Sstefano_zampini         } else {
59929b521d4Sstefano_zampini           rnz = ai[i + 1] - adiag[i];
60029b521d4Sstefano_zampini           ajj = aj + adiag[i];
60129b521d4Sstefano_zampini           v1  = av + adiag[i];
60229b521d4Sstefano_zampini           for (j = 0; j < rnz; j++) {
6039566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
6049566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[nz]));
605a6053eceSJunchao Zhang             val[nz++] = v1[j];
60629b521d4Sstefano_zampini           }
60729b521d4Sstefano_zampini         }
60829b521d4Sstefano_zampini       }
60929b521d4Sstefano_zampini     } else {
61016ebf90aSShri Abhyankar       for (i = 0; i < M; i++) {
61116ebf90aSShri Abhyankar         rnz = ai[i + 1] - adiag[i];
61267877ebaSShri Abhyankar         ajj = aj + adiag[i];
613cf3759fdSShri Abhyankar         v1  = av + adiag[i];
61467877ebaSShri Abhyankar         for (j = 0; j < rnz; j++) {
6159566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
6169566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[nz]));
617a6053eceSJunchao Zhang           val[nz++] = v1[j];
61816ebf90aSShri Abhyankar         }
61916ebf90aSShri Abhyankar       }
62029b521d4Sstefano_zampini     }
621397b6df1SKris Buschelman   } else {
622a6053eceSJunchao Zhang     nz  = 0;
623a6053eceSJunchao Zhang     val = mumps->val;
62429b521d4Sstefano_zampini     if (missing) {
62516ebf90aSShri Abhyankar       for (i = 0; i < M; i++) {
62629b521d4Sstefano_zampini         if (PetscUnlikely(adiag[i] >= ai[i + 1])) {
62729b521d4Sstefano_zampini           for (j = ai[i]; j < ai[i + 1]; j++) {
62829b521d4Sstefano_zampini             if (aj[j] < i) continue;
62929b521d4Sstefano_zampini             val[nz++] = av[j];
63029b521d4Sstefano_zampini           }
63129b521d4Sstefano_zampini         } else {
63216ebf90aSShri Abhyankar           rnz = ai[i + 1] - adiag[i];
63367877ebaSShri Abhyankar           v1  = av + adiag[i];
634ad540459SPierre Jolivet           for (j = 0; j < rnz; j++) val[nz++] = v1[j];
63516ebf90aSShri Abhyankar         }
63616ebf90aSShri Abhyankar       }
63729b521d4Sstefano_zampini     } else {
63816ebf90aSShri Abhyankar       for (i = 0; i < M; i++) {
63916ebf90aSShri Abhyankar         rnz = ai[i + 1] - adiag[i];
64016ebf90aSShri Abhyankar         v1  = av + adiag[i];
641ad540459SPierre Jolivet         for (j = 0; j < rnz; j++) val[nz++] = v1[j];
64216ebf90aSShri Abhyankar       }
64316ebf90aSShri Abhyankar     }
64429b521d4Sstefano_zampini   }
6459566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(A, &av));
6463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
64716ebf90aSShri Abhyankar }
64816ebf90aSShri Abhyankar 
64966976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpisbaij_mpisbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
650d71ae5a4SJacob Faibussowitsch {
651a6053eceSJunchao Zhang   const PetscInt    *ai, *aj, *bi, *bj, *garray, *ajj, *bjj;
652a6053eceSJunchao Zhang   PetscInt           bs;
6536497c311SBarry Smith   PetscCount         rstart, nz, i, j, k, m, jj, irow, countA, countB;
654a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
65516ebf90aSShri Abhyankar   const PetscScalar *av, *bv, *v1, *v2;
65616ebf90aSShri Abhyankar   PetscScalar       *val;
657397b6df1SKris Buschelman   Mat_MPISBAIJ      *mat = (Mat_MPISBAIJ *)A->data;
658f4f49eeaSPierre Jolivet   Mat_SeqSBAIJ      *aa  = (Mat_SeqSBAIJ *)mat->A->data;
659f4f49eeaSPierre Jolivet   Mat_SeqBAIJ       *bb  = (Mat_SeqBAIJ *)mat->B->data;
660ec4f40fdSPierre Jolivet   const PetscInt     bs2 = aa->bs2, mbs = aa->mbs;
66138548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
662b94d7dedSBarry Smith   PetscBool hermitian, isset;
66338548759SBarry Smith #endif
66416ebf90aSShri Abhyankar 
66516ebf90aSShri Abhyankar   PetscFunctionBegin;
66638548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
667b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
668b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
66938548759SBarry Smith #endif
6709566063dSJacob Faibussowitsch   PetscCall(MatGetBlockSize(A, &bs));
67138548759SBarry Smith   rstart = A->rmap->rstart;
67238548759SBarry Smith   ai     = aa->i;
67338548759SBarry Smith   aj     = aa->j;
67438548759SBarry Smith   bi     = bb->i;
67538548759SBarry Smith   bj     = bb->j;
67638548759SBarry Smith   av     = aa->a;
67738548759SBarry Smith   bv     = bb->a;
678397b6df1SKris Buschelman 
6792205254eSKarl Rupp   garray = mat->garray;
6802205254eSKarl Rupp 
681bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
682a6053eceSJunchao Zhang     nz = (aa->nz + bb->nz) * bs2; /* just a conservative estimate */
6839566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
6849566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
685a6053eceSJunchao Zhang     /* can not decide the exact mumps->nnz now because of the SBAIJ */
686a6053eceSJunchao Zhang     mumps->irn = row;
687a6053eceSJunchao Zhang     mumps->jcn = col;
688a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
689397b6df1SKris Buschelman   } else {
690a6053eceSJunchao Zhang     val = mumps->val;
691397b6df1SKris Buschelman   }
692397b6df1SKris Buschelman 
6939371c9d4SSatish Balay   jj   = 0;
6949371c9d4SSatish Balay   irow = rstart;
695ec4f40fdSPierre Jolivet   for (i = 0; i < mbs; i++) {
696397b6df1SKris Buschelman     ajj    = aj + ai[i]; /* ptr to the beginning of this row */
697397b6df1SKris Buschelman     countA = ai[i + 1] - ai[i];
698397b6df1SKris Buschelman     countB = bi[i + 1] - bi[i];
699397b6df1SKris Buschelman     bjj    = bj + bi[i];
700ec4f40fdSPierre Jolivet     v1     = av + ai[i] * bs2;
701ec4f40fdSPierre Jolivet     v2     = bv + bi[i] * bs2;
702397b6df1SKris Buschelman 
703ec4f40fdSPierre Jolivet     if (bs > 1) {
704ec4f40fdSPierre Jolivet       /* A-part */
705ec4f40fdSPierre Jolivet       for (j = 0; j < countA; j++) {
706ec4f40fdSPierre Jolivet         for (k = 0; k < bs; k++) {
707ec4f40fdSPierre Jolivet           for (m = 0; m < bs; m++) {
708ec4f40fdSPierre Jolivet             if (rstart + ajj[j] * bs > irow || k >= m) {
709ec4f40fdSPierre Jolivet               if (reuse == MAT_INITIAL_MATRIX) {
7109566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(irow + m + shift, &row[jj]));
7119566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(rstart + ajj[j] * bs + k + shift, &col[jj]));
712ec4f40fdSPierre Jolivet               }
713ec4f40fdSPierre Jolivet               val[jj++] = v1[j * bs2 + m + k * bs];
714ec4f40fdSPierre Jolivet             }
715ec4f40fdSPierre Jolivet           }
716ec4f40fdSPierre Jolivet         }
717ec4f40fdSPierre Jolivet       }
718ec4f40fdSPierre Jolivet 
719ec4f40fdSPierre Jolivet       /* B-part */
720ec4f40fdSPierre Jolivet       for (j = 0; j < countB; j++) {
721ec4f40fdSPierre Jolivet         for (k = 0; k < bs; k++) {
722ec4f40fdSPierre Jolivet           for (m = 0; m < bs; m++) {
723ec4f40fdSPierre Jolivet             if (reuse == MAT_INITIAL_MATRIX) {
7249566063dSJacob Faibussowitsch               PetscCall(PetscMUMPSIntCast(irow + m + shift, &row[jj]));
7259566063dSJacob Faibussowitsch               PetscCall(PetscMUMPSIntCast(garray[bjj[j]] * bs + k + shift, &col[jj]));
726ec4f40fdSPierre Jolivet             }
727ec4f40fdSPierre Jolivet             val[jj++] = v2[j * bs2 + m + k * bs];
728ec4f40fdSPierre Jolivet           }
729ec4f40fdSPierre Jolivet         }
730ec4f40fdSPierre Jolivet       }
731ec4f40fdSPierre Jolivet     } else {
732397b6df1SKris Buschelman       /* A-part */
733397b6df1SKris Buschelman       for (j = 0; j < countA; j++) {
734bccb9932SShri Abhyankar         if (reuse == MAT_INITIAL_MATRIX) {
7359566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
7369566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(rstart + ajj[j] + shift, &col[jj]));
737397b6df1SKris Buschelman         }
73816ebf90aSShri Abhyankar         val[jj++] = v1[j];
739397b6df1SKris Buschelman       }
74016ebf90aSShri Abhyankar 
74116ebf90aSShri Abhyankar       /* B-part */
74216ebf90aSShri Abhyankar       for (j = 0; j < countB; j++) {
743bccb9932SShri Abhyankar         if (reuse == MAT_INITIAL_MATRIX) {
7449566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
7459566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(garray[bjj[j]] + shift, &col[jj]));
746397b6df1SKris Buschelman         }
74716ebf90aSShri Abhyankar         val[jj++] = v2[j];
74816ebf90aSShri Abhyankar       }
74916ebf90aSShri Abhyankar     }
750ec4f40fdSPierre Jolivet     irow += bs;
751ec4f40fdSPierre Jolivet   }
7525d955bbbSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) mumps->nnz = jj;
7533ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
75416ebf90aSShri Abhyankar }
75516ebf90aSShri Abhyankar 
75666976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpiaij_mpiaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
757d71ae5a4SJacob Faibussowitsch {
75816ebf90aSShri Abhyankar   const PetscInt    *ai, *aj, *bi, *bj, *garray, m = A->rmap->n, *ajj, *bjj;
7596497c311SBarry Smith   PetscCount         rstart, cstart, nz, i, j, jj, irow, countA, countB;
760a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
76116ebf90aSShri Abhyankar   const PetscScalar *av, *bv, *v1, *v2;
76216ebf90aSShri Abhyankar   PetscScalar       *val;
763a3d589ffSStefano Zampini   Mat                Ad, Ao;
764a3d589ffSStefano Zampini   Mat_SeqAIJ        *aa;
765a3d589ffSStefano Zampini   Mat_SeqAIJ        *bb;
76616ebf90aSShri Abhyankar 
76716ebf90aSShri Abhyankar   PetscFunctionBegin;
7689566063dSJacob Faibussowitsch   PetscCall(MatMPIAIJGetSeqAIJ(A, &Ad, &Ao, &garray));
7699566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ad, &av));
7709566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ao, &bv));
771a3d589ffSStefano Zampini 
77257508eceSPierre Jolivet   aa = (Mat_SeqAIJ *)Ad->data;
77357508eceSPierre Jolivet   bb = (Mat_SeqAIJ *)Ao->data;
77438548759SBarry Smith   ai = aa->i;
77538548759SBarry Smith   aj = aa->j;
77638548759SBarry Smith   bi = bb->i;
77738548759SBarry Smith   bj = bb->j;
77816ebf90aSShri Abhyankar 
779a3d589ffSStefano Zampini   rstart = A->rmap->rstart;
7805d955bbbSStefano Zampini   cstart = A->cmap->rstart;
7812205254eSKarl Rupp 
782bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
7836497c311SBarry Smith     nz = (PetscCount)aa->nz + bb->nz; /* make sure the sum won't overflow PetscInt */
7849566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
7859566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
786a6053eceSJunchao Zhang     mumps->nnz = nz;
787a6053eceSJunchao Zhang     mumps->irn = row;
788a6053eceSJunchao Zhang     mumps->jcn = col;
789a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
79016ebf90aSShri Abhyankar   } else {
791a6053eceSJunchao Zhang     val = mumps->val;
79216ebf90aSShri Abhyankar   }
79316ebf90aSShri Abhyankar 
7949371c9d4SSatish Balay   jj   = 0;
7959371c9d4SSatish Balay   irow = rstart;
79616ebf90aSShri Abhyankar   for (i = 0; i < m; i++) {
79716ebf90aSShri Abhyankar     ajj    = aj + ai[i]; /* ptr to the beginning of this row */
79816ebf90aSShri Abhyankar     countA = ai[i + 1] - ai[i];
79916ebf90aSShri Abhyankar     countB = bi[i + 1] - bi[i];
80016ebf90aSShri Abhyankar     bjj    = bj + bi[i];
80116ebf90aSShri Abhyankar     v1     = av + ai[i];
80216ebf90aSShri Abhyankar     v2     = bv + bi[i];
80316ebf90aSShri Abhyankar 
80416ebf90aSShri Abhyankar     /* A-part */
80516ebf90aSShri Abhyankar     for (j = 0; j < countA; j++) {
806bccb9932SShri Abhyankar       if (reuse == MAT_INITIAL_MATRIX) {
8079566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
8085d955bbbSStefano Zampini         PetscCall(PetscMUMPSIntCast(cstart + ajj[j] + shift, &col[jj]));
80916ebf90aSShri Abhyankar       }
81016ebf90aSShri Abhyankar       val[jj++] = v1[j];
81116ebf90aSShri Abhyankar     }
81216ebf90aSShri Abhyankar 
81316ebf90aSShri Abhyankar     /* B-part */
81416ebf90aSShri Abhyankar     for (j = 0; j < countB; j++) {
815bccb9932SShri Abhyankar       if (reuse == MAT_INITIAL_MATRIX) {
8169566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
8179566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(garray[bjj[j]] + shift, &col[jj]));
81816ebf90aSShri Abhyankar       }
81916ebf90aSShri Abhyankar       val[jj++] = v2[j];
82016ebf90aSShri Abhyankar     }
82116ebf90aSShri Abhyankar     irow++;
82216ebf90aSShri Abhyankar   }
8239566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ad, &av));
8249566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ao, &bv));
8253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
82616ebf90aSShri Abhyankar }
82716ebf90aSShri Abhyankar 
82866976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpibaij_mpiaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
829d71ae5a4SJacob Faibussowitsch {
83067877ebaSShri Abhyankar   Mat_MPIBAIJ       *mat = (Mat_MPIBAIJ *)A->data;
831f4f49eeaSPierre Jolivet   Mat_SeqBAIJ       *aa  = (Mat_SeqBAIJ *)mat->A->data;
832f4f49eeaSPierre Jolivet   Mat_SeqBAIJ       *bb  = (Mat_SeqBAIJ *)mat->B->data;
83367877ebaSShri Abhyankar   const PetscInt    *ai = aa->i, *bi = bb->i, *aj = aa->j, *bj = bb->j, *ajj, *bjj;
8345d955bbbSStefano Zampini   const PetscInt    *garray = mat->garray, mbs = mat->mbs, rstart = A->rmap->rstart, cstart = A->cmap->rstart;
83533d57670SJed Brown   const PetscInt     bs2 = mat->bs2;
836a6053eceSJunchao Zhang   PetscInt           bs;
8376497c311SBarry Smith   PetscCount         nz, i, j, k, n, jj, irow, countA, countB, idx;
838a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
83967877ebaSShri Abhyankar   const PetscScalar *av = aa->a, *bv = bb->a, *v1, *v2;
84067877ebaSShri Abhyankar   PetscScalar       *val;
84167877ebaSShri Abhyankar 
84267877ebaSShri Abhyankar   PetscFunctionBegin;
8439566063dSJacob Faibussowitsch   PetscCall(MatGetBlockSize(A, &bs));
844bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
84567877ebaSShri Abhyankar     nz = bs2 * (aa->nz + bb->nz);
8469566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
8479566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
848a6053eceSJunchao Zhang     mumps->nnz = nz;
849a6053eceSJunchao Zhang     mumps->irn = row;
850a6053eceSJunchao Zhang     mumps->jcn = col;
851a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
85267877ebaSShri Abhyankar   } else {
853a6053eceSJunchao Zhang     val = mumps->val;
85467877ebaSShri Abhyankar   }
85567877ebaSShri Abhyankar 
8569371c9d4SSatish Balay   jj   = 0;
8579371c9d4SSatish Balay   irow = rstart;
85867877ebaSShri Abhyankar   for (i = 0; i < mbs; i++) {
85967877ebaSShri Abhyankar     countA = ai[i + 1] - ai[i];
86067877ebaSShri Abhyankar     countB = bi[i + 1] - bi[i];
86167877ebaSShri Abhyankar     ajj    = aj + ai[i];
86267877ebaSShri Abhyankar     bjj    = bj + bi[i];
86367877ebaSShri Abhyankar     v1     = av + bs2 * ai[i];
86467877ebaSShri Abhyankar     v2     = bv + bs2 * bi[i];
86567877ebaSShri Abhyankar 
86667877ebaSShri Abhyankar     idx = 0;
86767877ebaSShri Abhyankar     /* A-part */
86867877ebaSShri Abhyankar     for (k = 0; k < countA; k++) {
86967877ebaSShri Abhyankar       for (j = 0; j < bs; j++) {
87067877ebaSShri Abhyankar         for (n = 0; n < bs; n++) {
871bccb9932SShri Abhyankar           if (reuse == MAT_INITIAL_MATRIX) {
8729566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(irow + n + shift, &row[jj]));
8735d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(cstart + bs * ajj[k] + j + shift, &col[jj]));
87467877ebaSShri Abhyankar           }
87567877ebaSShri Abhyankar           val[jj++] = v1[idx++];
87667877ebaSShri Abhyankar         }
87767877ebaSShri Abhyankar       }
87867877ebaSShri Abhyankar     }
87967877ebaSShri Abhyankar 
88067877ebaSShri Abhyankar     idx = 0;
88167877ebaSShri Abhyankar     /* B-part */
88267877ebaSShri Abhyankar     for (k = 0; k < countB; k++) {
88367877ebaSShri Abhyankar       for (j = 0; j < bs; j++) {
88467877ebaSShri Abhyankar         for (n = 0; n < bs; n++) {
885bccb9932SShri Abhyankar           if (reuse == MAT_INITIAL_MATRIX) {
8869566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(irow + n + shift, &row[jj]));
8879566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(bs * garray[bjj[k]] + j + shift, &col[jj]));
88867877ebaSShri Abhyankar           }
889d985c460SShri Abhyankar           val[jj++] = v2[idx++];
89067877ebaSShri Abhyankar         }
89167877ebaSShri Abhyankar       }
89267877ebaSShri Abhyankar     }
893d985c460SShri Abhyankar     irow += bs;
89467877ebaSShri Abhyankar   }
8953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
89667877ebaSShri Abhyankar }
89767877ebaSShri Abhyankar 
89866976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpiaij_mpisbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
899d71ae5a4SJacob Faibussowitsch {
90016ebf90aSShri Abhyankar   const PetscInt    *ai, *aj, *adiag, *bi, *bj, *garray, m = A->rmap->n, *ajj, *bjj;
9016497c311SBarry Smith   PetscCount         rstart, nz, nza, nzb, i, j, jj, irow, countA, countB;
902a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
90316ebf90aSShri Abhyankar   const PetscScalar *av, *bv, *v1, *v2;
90416ebf90aSShri Abhyankar   PetscScalar       *val;
905a3d589ffSStefano Zampini   Mat                Ad, Ao;
906a3d589ffSStefano Zampini   Mat_SeqAIJ        *aa;
907a3d589ffSStefano Zampini   Mat_SeqAIJ        *bb;
90838548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
909b94d7dedSBarry Smith   PetscBool hermitian, isset;
91038548759SBarry Smith #endif
91116ebf90aSShri Abhyankar 
91216ebf90aSShri Abhyankar   PetscFunctionBegin;
91338548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
914b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
915b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
91638548759SBarry Smith #endif
9179566063dSJacob Faibussowitsch   PetscCall(MatMPIAIJGetSeqAIJ(A, &Ad, &Ao, &garray));
9189566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ad, &av));
9199566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ao, &bv));
920a3d589ffSStefano Zampini 
92157508eceSPierre Jolivet   aa    = (Mat_SeqAIJ *)Ad->data;
92257508eceSPierre Jolivet   bb    = (Mat_SeqAIJ *)Ao->data;
92338548759SBarry Smith   ai    = aa->i;
92438548759SBarry Smith   aj    = aa->j;
92538548759SBarry Smith   adiag = aa->diag;
92638548759SBarry Smith   bi    = bb->i;
92738548759SBarry Smith   bj    = bb->j;
9282205254eSKarl Rupp 
92916ebf90aSShri Abhyankar   rstart = A->rmap->rstart;
93016ebf90aSShri Abhyankar 
931bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
932e0bace9bSHong Zhang     nza = 0; /* num of upper triangular entries in mat->A, including diagonals */
933e0bace9bSHong Zhang     nzb = 0; /* num of upper triangular entries in mat->B */
93416ebf90aSShri Abhyankar     for (i = 0; i < m; i++) {
935e0bace9bSHong Zhang       nza += (ai[i + 1] - adiag[i]);
93616ebf90aSShri Abhyankar       countB = bi[i + 1] - bi[i];
93716ebf90aSShri Abhyankar       bjj    = bj + bi[i];
938e0bace9bSHong Zhang       for (j = 0; j < countB; j++) {
939e0bace9bSHong Zhang         if (garray[bjj[j]] > rstart) nzb++;
940e0bace9bSHong Zhang       }
941e0bace9bSHong Zhang     }
94216ebf90aSShri Abhyankar 
943e0bace9bSHong Zhang     nz = nza + nzb; /* total nz of upper triangular part of mat */
9449566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
9459566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
946a6053eceSJunchao Zhang     mumps->nnz = nz;
947a6053eceSJunchao Zhang     mumps->irn = row;
948a6053eceSJunchao Zhang     mumps->jcn = col;
949a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
95016ebf90aSShri Abhyankar   } else {
951a6053eceSJunchao Zhang     val = mumps->val;
95216ebf90aSShri Abhyankar   }
95316ebf90aSShri Abhyankar 
9549371c9d4SSatish Balay   jj   = 0;
9559371c9d4SSatish Balay   irow = rstart;
95616ebf90aSShri Abhyankar   for (i = 0; i < m; i++) {
95716ebf90aSShri Abhyankar     ajj    = aj + adiag[i]; /* ptr to the beginning of the diagonal of this row */
95816ebf90aSShri Abhyankar     v1     = av + adiag[i];
95916ebf90aSShri Abhyankar     countA = ai[i + 1] - adiag[i];
96016ebf90aSShri Abhyankar     countB = bi[i + 1] - bi[i];
96116ebf90aSShri Abhyankar     bjj    = bj + bi[i];
96216ebf90aSShri Abhyankar     v2     = bv + bi[i];
96316ebf90aSShri Abhyankar 
96416ebf90aSShri Abhyankar     /* A-part */
96516ebf90aSShri Abhyankar     for (j = 0; j < countA; j++) {
966bccb9932SShri Abhyankar       if (reuse == MAT_INITIAL_MATRIX) {
9679566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
9689566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(rstart + ajj[j] + shift, &col[jj]));
96916ebf90aSShri Abhyankar       }
97016ebf90aSShri Abhyankar       val[jj++] = v1[j];
97116ebf90aSShri Abhyankar     }
97216ebf90aSShri Abhyankar 
97316ebf90aSShri Abhyankar     /* B-part */
97416ebf90aSShri Abhyankar     for (j = 0; j < countB; j++) {
97516ebf90aSShri Abhyankar       if (garray[bjj[j]] > rstart) {
976bccb9932SShri Abhyankar         if (reuse == MAT_INITIAL_MATRIX) {
9779566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
9789566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(garray[bjj[j]] + shift, &col[jj]));
97916ebf90aSShri Abhyankar         }
98016ebf90aSShri Abhyankar         val[jj++] = v2[j];
98116ebf90aSShri Abhyankar       }
982397b6df1SKris Buschelman     }
983397b6df1SKris Buschelman     irow++;
984397b6df1SKris Buschelman   }
9859566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ad, &av));
9869566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ao, &bv));
9873ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
988397b6df1SKris Buschelman }
989397b6df1SKris Buschelman 
990d2a308c1SPierre Jolivet static PetscErrorCode MatConvertToTriples_diagonal_xaij(Mat A, PETSC_UNUSED PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
991c3e1b152SPierre Jolivet {
992c3e1b152SPierre Jolivet   const PetscScalar *av;
993c3e1b152SPierre Jolivet   const PetscInt     M = A->rmap->n;
9946497c311SBarry Smith   PetscCount         i;
995c3e1b152SPierre Jolivet   PetscMUMPSInt     *row, *col;
996c3e1b152SPierre Jolivet   Vec                v;
997c3e1b152SPierre Jolivet 
998c3e1b152SPierre Jolivet   PetscFunctionBegin;
999c3e1b152SPierre Jolivet   PetscCall(MatDiagonalGetDiagonal(A, &v));
1000c3e1b152SPierre Jolivet   PetscCall(VecGetArrayRead(v, &av));
1001c3e1b152SPierre Jolivet   if (reuse == MAT_INITIAL_MATRIX) {
1002c3e1b152SPierre Jolivet     PetscCall(PetscMalloc2(M, &row, M, &col));
1003c3e1b152SPierre Jolivet     for (i = 0; i < M; i++) {
1004c3e1b152SPierre Jolivet       PetscCall(PetscMUMPSIntCast(i + A->rmap->rstart, &row[i]));
1005c3e1b152SPierre Jolivet       col[i] = row[i];
1006c3e1b152SPierre Jolivet     }
1007c3e1b152SPierre Jolivet     mumps->val = (PetscScalar *)av;
1008c3e1b152SPierre Jolivet     mumps->irn = row;
1009c3e1b152SPierre Jolivet     mumps->jcn = col;
1010c3e1b152SPierre Jolivet     mumps->nnz = M;
1011127cd276SPierre Jolivet   } else if (mumps->nest_vals) PetscCall(PetscArraycpy(mumps->val, av, M)); /* MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_diagonal_xaij(), so one needs to copy the memory */
1012127cd276SPierre Jolivet   else mumps->val = (PetscScalar *)av;                                      /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
1013c3e1b152SPierre Jolivet   PetscCall(VecRestoreArrayRead(v, &av));
1014c3e1b152SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
1015c3e1b152SPierre Jolivet }
1016c3e1b152SPierre Jolivet 
1017d2a308c1SPierre Jolivet static PetscErrorCode MatConvertToTriples_dense_xaij(Mat A, PETSC_UNUSED PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
10184b9405b2SPierre Jolivet {
10194b9405b2SPierre Jolivet   PetscScalar   *v;
10204b9405b2SPierre Jolivet   const PetscInt m = A->rmap->n, N = A->cmap->N;
10214b9405b2SPierre Jolivet   PetscInt       lda;
10226497c311SBarry Smith   PetscCount     i, j;
10234b9405b2SPierre Jolivet   PetscMUMPSInt *row, *col;
10244b9405b2SPierre Jolivet 
10254b9405b2SPierre Jolivet   PetscFunctionBegin;
10264b9405b2SPierre Jolivet   PetscCall(MatDenseGetArray(A, &v));
10274b9405b2SPierre Jolivet   PetscCall(MatDenseGetLDA(A, &lda));
10284b9405b2SPierre Jolivet   if (reuse == MAT_INITIAL_MATRIX) {
10294b9405b2SPierre Jolivet     PetscCall(PetscMalloc2(m * N, &row, m * N, &col));
10304b9405b2SPierre Jolivet     for (i = 0; i < m; i++) {
10314b9405b2SPierre Jolivet       col[i] = 0;
10324b9405b2SPierre Jolivet       PetscCall(PetscMUMPSIntCast(i + A->rmap->rstart, &row[i]));
10334b9405b2SPierre Jolivet     }
10344b9405b2SPierre Jolivet     for (j = 1; j < N; j++) {
10354b9405b2SPierre Jolivet       for (i = 0; i < m; i++) PetscCall(PetscMUMPSIntCast(j, col + i + m * j));
10364b9405b2SPierre Jolivet       PetscCall(PetscArraycpy(row + m * j, row + m * (j - 1), m));
10374b9405b2SPierre Jolivet     }
10384b9405b2SPierre Jolivet     if (lda == m) mumps->val = v;
10394b9405b2SPierre Jolivet     else {
10404b9405b2SPierre Jolivet       PetscCall(PetscMalloc1(m * N, &mumps->val));
10414b9405b2SPierre Jolivet       mumps->val_alloc = mumps->val;
10424b9405b2SPierre Jolivet       for (j = 0; j < N; j++) PetscCall(PetscArraycpy(mumps->val + m * j, v + lda * j, m));
10434b9405b2SPierre Jolivet     }
10444b9405b2SPierre Jolivet     mumps->irn = row;
10454b9405b2SPierre Jolivet     mumps->jcn = col;
10464b9405b2SPierre Jolivet     mumps->nnz = m * N;
10474b9405b2SPierre Jolivet   } else {
10484b9405b2SPierre Jolivet     if (lda == m && !mumps->nest_vals) mumps->val = v;
10494b9405b2SPierre Jolivet     else {
10504b9405b2SPierre Jolivet       for (j = 0; j < N; j++) PetscCall(PetscArraycpy(mumps->val + m * j, v + lda * j, m));
10514b9405b2SPierre Jolivet     }
10524b9405b2SPierre Jolivet   }
10534b9405b2SPierre Jolivet   PetscCall(MatDenseRestoreArray(A, &v));
10544b9405b2SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
10554b9405b2SPierre Jolivet }
10564b9405b2SPierre Jolivet 
105753587d93SPierre Jolivet // If the input Mat (sub) is either MATTRANSPOSEVIRTUAL or MATHERMITIANTRANSPOSEVIRTUAL, this function gets the parent Mat until it is not a
105853587d93SPierre Jolivet // MATTRANSPOSEVIRTUAL or MATHERMITIANTRANSPOSEVIRTUAL itself and returns the appropriate shift, scaling, and whether the parent Mat should be conjugated
105953587d93SPierre Jolivet // and its rows and columns permuted
106053587d93SPierre Jolivet // TODO FIXME: this should not be in this file and should instead be refactored where the same logic applies, e.g., MatAXPY_Dense_Nest()
106153587d93SPierre Jolivet static PetscErrorCode MatGetTranspose_TransposeVirtual(Mat *sub, PetscBool *conjugate, PetscScalar *vshift, PetscScalar *vscale, PetscBool *swap)
106253587d93SPierre Jolivet {
106353587d93SPierre Jolivet   Mat         A;
106453587d93SPierre Jolivet   PetscScalar s[2];
106553587d93SPierre Jolivet   PetscBool   isTrans, isHTrans, compare;
106653587d93SPierre Jolivet 
106753587d93SPierre Jolivet   PetscFunctionBegin;
106853587d93SPierre Jolivet   do {
106953587d93SPierre Jolivet     PetscCall(PetscObjectTypeCompare((PetscObject)*sub, MATTRANSPOSEVIRTUAL, &isTrans));
107053587d93SPierre Jolivet     if (isTrans) {
107153587d93SPierre Jolivet       PetscCall(MatTransposeGetMat(*sub, &A));
107253587d93SPierre Jolivet       isHTrans = PETSC_FALSE;
107353587d93SPierre Jolivet     } else {
107453587d93SPierre Jolivet       PetscCall(PetscObjectTypeCompare((PetscObject)*sub, MATHERMITIANTRANSPOSEVIRTUAL, &isHTrans));
107553587d93SPierre Jolivet       if (isHTrans) PetscCall(MatHermitianTransposeGetMat(*sub, &A));
107653587d93SPierre Jolivet     }
107753587d93SPierre Jolivet     compare = (PetscBool)(isTrans || isHTrans);
107853587d93SPierre Jolivet     if (compare) {
107953587d93SPierre Jolivet       if (vshift && vscale) {
108053587d93SPierre Jolivet         PetscCall(MatShellGetScalingShifts(*sub, s, s + 1, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Mat *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED));
108153587d93SPierre Jolivet         if (!*conjugate) {
108253587d93SPierre Jolivet           *vshift += s[0] * *vscale;
108353587d93SPierre Jolivet           *vscale *= s[1];
108453587d93SPierre Jolivet         } else {
108553587d93SPierre Jolivet           *vshift += PetscConj(s[0]) * *vscale;
108653587d93SPierre Jolivet           *vscale *= PetscConj(s[1]);
108753587d93SPierre Jolivet         }
108853587d93SPierre Jolivet       }
108953587d93SPierre Jolivet       if (swap) *swap = (PetscBool)!*swap;
109053587d93SPierre Jolivet       if (isHTrans && conjugate) *conjugate = (PetscBool)!*conjugate;
109153587d93SPierre Jolivet       *sub = A;
109253587d93SPierre Jolivet     }
109353587d93SPierre Jolivet   } while (compare);
109453587d93SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
109553587d93SPierre Jolivet }
109653587d93SPierre Jolivet 
109766976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_nest_xaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
10989d0448ceSStefano Zampini {
10999d0448ceSStefano Zampini   Mat     **mats;
11009d0448ceSStefano Zampini   PetscInt  nr, nc;
11019d0448ceSStefano Zampini   PetscBool chol = mumps->sym ? PETSC_TRUE : PETSC_FALSE;
11029d0448ceSStefano Zampini 
11039d0448ceSStefano Zampini   PetscFunctionBegin;
11049d0448ceSStefano Zampini   PetscCall(MatNestGetSubMats(A, &nr, &nc, &mats));
11059d0448ceSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
11069d0448ceSStefano Zampini     PetscMUMPSInt *irns, *jcns;
11079d0448ceSStefano Zampini     PetscScalar   *vals;
11086497c311SBarry Smith     PetscCount     totnnz, cumnnz, maxnnz;
11099d0448ceSStefano Zampini     PetscInt      *pjcns_w;
11109d0448ceSStefano Zampini     IS            *rows, *cols;
11119d0448ceSStefano Zampini     PetscInt     **rows_idx, **cols_idx;
11129d0448ceSStefano Zampini 
11139d0448ceSStefano Zampini     cumnnz = 0;
11149d0448ceSStefano Zampini     maxnnz = 0;
11155d955bbbSStefano Zampini     PetscCall(PetscMalloc2(nr * nc + 1, &mumps->nest_vals_start, nr * nc, &mumps->nest_convert_to_triples));
11169d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) {
11179d0448ceSStefano Zampini       for (PetscInt c = 0; c < nc; c++) {
11189d0448ceSStefano Zampini         Mat sub = mats[r][c];
11199d0448ceSStefano Zampini 
11209d0448ceSStefano Zampini         mumps->nest_convert_to_triples[r * nc + c] = NULL;
11219d0448ceSStefano Zampini         if (chol && c < r) continue; /* skip lower-triangular block for Cholesky */
11229d0448ceSStefano Zampini         if (sub) {
11239d0448ceSStefano Zampini           PetscErrorCode (*convert_to_triples)(Mat, PetscInt, MatReuse, Mat_MUMPS *) = NULL;
112453587d93SPierre Jolivet           PetscBool isSeqAIJ, isMPIAIJ, isSeqBAIJ, isMPIBAIJ, isSeqSBAIJ, isMPISBAIJ, isDiag, isDense;
11259d0448ceSStefano Zampini           MatInfo   info;
11269d0448ceSStefano Zampini 
112753587d93SPierre Jolivet           PetscCall(MatGetTranspose_TransposeVirtual(&sub, NULL, NULL, NULL, NULL));
11289d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQAIJ, &isSeqAIJ));
11299d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIAIJ, &isMPIAIJ));
11309d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQBAIJ, &isSeqBAIJ));
11319d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIBAIJ, &isMPIBAIJ));
11329d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQSBAIJ, &isSeqSBAIJ));
11339d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPISBAIJ, &isMPISBAIJ));
1134c3e1b152SPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATDIAGONAL, &isDiag));
11354b9405b2SPierre Jolivet           PetscCall(PetscObjectTypeCompareAny((PetscObject)sub, &isDense, MATSEQDENSE, MATMPIDENSE, NULL));
11369d0448ceSStefano Zampini 
11379d0448ceSStefano Zampini           if (chol) {
11389d0448ceSStefano Zampini             if (r == c) {
11399d0448ceSStefano Zampini               if (isSeqAIJ) convert_to_triples = MatConvertToTriples_seqaij_seqsbaij;
11409d0448ceSStefano Zampini               else if (isMPIAIJ) convert_to_triples = MatConvertToTriples_mpiaij_mpisbaij;
11419d0448ceSStefano Zampini               else if (isSeqSBAIJ) convert_to_triples = MatConvertToTriples_seqsbaij_seqsbaij;
11429d0448ceSStefano Zampini               else if (isMPISBAIJ) convert_to_triples = MatConvertToTriples_mpisbaij_mpisbaij;
1143c3e1b152SPierre Jolivet               else if (isDiag) convert_to_triples = MatConvertToTriples_diagonal_xaij;
11444b9405b2SPierre Jolivet               else if (isDense) convert_to_triples = MatConvertToTriples_dense_xaij;
11459d0448ceSStefano Zampini             } else {
11469d0448ceSStefano Zampini               if (isSeqAIJ) convert_to_triples = MatConvertToTriples_seqaij_seqaij;
11479d0448ceSStefano Zampini               else if (isMPIAIJ) convert_to_triples = MatConvertToTriples_mpiaij_mpiaij;
11489d0448ceSStefano Zampini               else if (isSeqBAIJ) convert_to_triples = MatConvertToTriples_seqbaij_seqaij;
11499d0448ceSStefano Zampini               else if (isMPIBAIJ) convert_to_triples = MatConvertToTriples_mpibaij_mpiaij;
1150c3e1b152SPierre Jolivet               else if (isDiag) convert_to_triples = MatConvertToTriples_diagonal_xaij;
11514b9405b2SPierre Jolivet               else if (isDense) convert_to_triples = MatConvertToTriples_dense_xaij;
11529d0448ceSStefano Zampini             }
11539d0448ceSStefano Zampini           } else {
11549d0448ceSStefano Zampini             if (isSeqAIJ) convert_to_triples = MatConvertToTriples_seqaij_seqaij;
11559d0448ceSStefano Zampini             else if (isMPIAIJ) convert_to_triples = MatConvertToTriples_mpiaij_mpiaij;
11569d0448ceSStefano Zampini             else if (isSeqBAIJ) convert_to_triples = MatConvertToTriples_seqbaij_seqaij;
11579d0448ceSStefano Zampini             else if (isMPIBAIJ) convert_to_triples = MatConvertToTriples_mpibaij_mpiaij;
1158c3e1b152SPierre Jolivet             else if (isDiag) convert_to_triples = MatConvertToTriples_diagonal_xaij;
11594b9405b2SPierre Jolivet             else if (isDense) convert_to_triples = MatConvertToTriples_dense_xaij;
11609d0448ceSStefano Zampini           }
11619d0448ceSStefano Zampini           PetscCheck(convert_to_triples, PetscObjectComm((PetscObject)sub), PETSC_ERR_SUP, "Not for block of type %s", ((PetscObject)sub)->type_name);
11629d0448ceSStefano Zampini           mumps->nest_convert_to_triples[r * nc + c] = convert_to_triples;
11639d0448ceSStefano Zampini           PetscCall(MatGetInfo(sub, MAT_LOCAL, &info));
11646497c311SBarry Smith           cumnnz += (PetscCount)info.nz_used; /* can be overestimated for Cholesky */
11659d0448ceSStefano Zampini           maxnnz = PetscMax(maxnnz, info.nz_used);
11669d0448ceSStefano Zampini         }
11679d0448ceSStefano Zampini       }
11689d0448ceSStefano Zampini     }
11699d0448ceSStefano Zampini 
11709d0448ceSStefano Zampini     /* Allocate total COO */
11719d0448ceSStefano Zampini     totnnz = cumnnz;
11729d0448ceSStefano Zampini     PetscCall(PetscMalloc2(totnnz, &irns, totnnz, &jcns));
11739d0448ceSStefano Zampini     PetscCall(PetscMalloc1(totnnz, &vals));
11749d0448ceSStefano Zampini 
11759d0448ceSStefano Zampini     /* Handle rows and column maps
11769d0448ceSStefano Zampini        We directly map rows and use an SF for the columns */
11779d0448ceSStefano Zampini     PetscCall(PetscMalloc4(nr, &rows, nc, &cols, nr, &rows_idx, nc, &cols_idx));
11789d0448ceSStefano Zampini     PetscCall(MatNestGetISs(A, rows, cols));
11799d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) PetscCall(ISGetIndices(rows[r], (const PetscInt **)&rows_idx[r]));
11809d0448ceSStefano Zampini     for (PetscInt c = 0; c < nc; c++) PetscCall(ISGetIndices(cols[c], (const PetscInt **)&cols_idx[c]));
11819d0448ceSStefano Zampini     if (PetscDefined(USE_64BIT_INDICES)) PetscCall(PetscMalloc1(maxnnz, &pjcns_w));
11825d955bbbSStefano Zampini     else (void)maxnnz;
11839d0448ceSStefano Zampini 
11849d0448ceSStefano Zampini     cumnnz = 0;
11859d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) {
11869d0448ceSStefano Zampini       for (PetscInt c = 0; c < nc; c++) {
11879d0448ceSStefano Zampini         Mat             sub    = mats[r][c];
11889d0448ceSStefano Zampini         const PetscInt *ridx   = rows_idx[r];
11895d955bbbSStefano Zampini         const PetscInt *cidx   = cols_idx[c];
119053587d93SPierre Jolivet         PetscScalar     vscale = 1.0, vshift = 0.0;
11919d0448ceSStefano Zampini         PetscInt        rst;
11929d0448ceSStefano Zampini         PetscSF         csf;
119353587d93SPierre Jolivet         PetscBool       conjugate = PETSC_FALSE, swap = PETSC_FALSE;
11945d955bbbSStefano Zampini         PetscLayout     cmap;
11956497c311SBarry Smith         PetscInt        innz;
11969d0448ceSStefano Zampini 
11979d0448ceSStefano Zampini         mumps->nest_vals_start[r * nc + c] = cumnnz;
11989d0448ceSStefano Zampini         if (!mumps->nest_convert_to_triples[r * nc + c]) continue;
11999d0448ceSStefano Zampini 
12005d955bbbSStefano Zampini         /* Extract inner blocks if needed */
120153587d93SPierre Jolivet         PetscCall(MatGetTranspose_TransposeVirtual(&sub, &conjugate, &vshift, &vscale, &swap));
120253587d93SPierre Jolivet         PetscCheck(vshift == 0.0, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Nonzero shift in parent MatShell");
12035d955bbbSStefano Zampini 
12045d955bbbSStefano Zampini         /* Get column layout to map off-process columns */
12055d955bbbSStefano Zampini         PetscCall(MatGetLayouts(sub, NULL, &cmap));
12065d955bbbSStefano Zampini 
12075d955bbbSStefano Zampini         /* Get row start to map on-process rows */
12085d955bbbSStefano Zampini         PetscCall(MatGetOwnershipRange(sub, &rst, NULL));
12095d955bbbSStefano Zampini 
12109d0448ceSStefano Zampini         /* Directly use the mumps datastructure and use C ordering for now */
12119d0448ceSStefano Zampini         PetscCall((*mumps->nest_convert_to_triples[r * nc + c])(sub, 0, MAT_INITIAL_MATRIX, mumps));
12129d0448ceSStefano Zampini 
12135d955bbbSStefano Zampini         /* Swap the role of rows and columns indices for transposed blocks
12145d955bbbSStefano Zampini            since we need values with global final ordering */
12155d955bbbSStefano Zampini         if (swap) {
12165d955bbbSStefano Zampini           cidx = rows_idx[r];
12175d955bbbSStefano Zampini           ridx = cols_idx[c];
12189d0448ceSStefano Zampini         }
12199d0448ceSStefano Zampini 
12205d955bbbSStefano Zampini         /* Communicate column indices
12215d955bbbSStefano Zampini            This could have been done with a single SF but it would have complicated the code a lot.
12225d955bbbSStefano Zampini            But since we do it only once, we pay the price of setting up an SF for each block */
12235d955bbbSStefano Zampini         if (PetscDefined(USE_64BIT_INDICES)) {
12245d955bbbSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) pjcns_w[k] = mumps->jcn[k];
1225f4f49eeaSPierre Jolivet         } else pjcns_w = (PetscInt *)mumps->jcn; /* This cast is needed only to silence warnings for 64bit integers builds */
12269d0448ceSStefano Zampini         PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)A), &csf));
12276497c311SBarry Smith         PetscCall(PetscIntCast(mumps->nnz, &innz));
12286497c311SBarry Smith         PetscCall(PetscSFSetGraphLayout(csf, cmap, innz, NULL, PETSC_OWN_POINTER, pjcns_w));
12295d955bbbSStefano Zampini         PetscCall(PetscSFBcastBegin(csf, MPIU_INT, cidx, pjcns_w, MPI_REPLACE));
12305d955bbbSStefano Zampini         PetscCall(PetscSFBcastEnd(csf, MPIU_INT, cidx, pjcns_w, MPI_REPLACE));
12319d0448ceSStefano Zampini         PetscCall(PetscSFDestroy(&csf));
12329d0448ceSStefano Zampini 
12335d955bbbSStefano Zampini         /* Import indices: use direct map for rows and mapped indices for columns */
12345d955bbbSStefano Zampini         if (swap) {
12355d955bbbSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) {
12365d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(ridx[mumps->irn[k] - rst] + shift, &jcns[cumnnz + k]));
12375d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(pjcns_w[k] + shift, &irns[cumnnz + k]));
12385d955bbbSStefano Zampini           }
12395d955bbbSStefano Zampini         } else {
12405d955bbbSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) {
12415d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(ridx[mumps->irn[k] - rst] + shift, &irns[cumnnz + k]));
12425d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(pjcns_w[k] + shift, &jcns[cumnnz + k]));
12435d955bbbSStefano Zampini           }
12445d955bbbSStefano Zampini         }
12455d955bbbSStefano Zampini 
12465d955bbbSStefano Zampini         /* Import values to full COO */
124753587d93SPierre Jolivet         if (conjugate) { /* conjugate the entries */
124850c845baSStefano Zampini           PetscScalar *v = vals + cumnnz;
124953587d93SPierre Jolivet           for (PetscInt k = 0; k < mumps->nnz; k++) v[k] = vscale * PetscConj(mumps->val[k]);
125053587d93SPierre Jolivet         } else if (vscale != 1.0) {
125153587d93SPierre Jolivet           PetscScalar *v = vals + cumnnz;
125253587d93SPierre Jolivet           for (PetscInt k = 0; k < mumps->nnz; k++) v[k] = vscale * mumps->val[k];
125353587d93SPierre Jolivet         } else PetscCall(PetscArraycpy(vals + cumnnz, mumps->val, mumps->nnz));
12549d0448ceSStefano Zampini 
12559d0448ceSStefano Zampini         /* Shift new starting point and sanity check */
12569d0448ceSStefano Zampini         cumnnz += mumps->nnz;
12576497c311SBarry Smith         PetscCheck(cumnnz <= totnnz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected number of nonzeros %" PetscCount_FMT " != %" PetscCount_FMT, cumnnz, totnnz);
12589d0448ceSStefano Zampini 
12599d0448ceSStefano Zampini         /* Free scratch memory */
12609d0448ceSStefano Zampini         PetscCall(PetscFree2(mumps->irn, mumps->jcn));
12619d0448ceSStefano Zampini         PetscCall(PetscFree(mumps->val_alloc));
12629d0448ceSStefano Zampini         mumps->val = NULL;
12639d0448ceSStefano Zampini         mumps->nnz = 0;
12649d0448ceSStefano Zampini       }
12659d0448ceSStefano Zampini     }
12669d0448ceSStefano Zampini     if (PetscDefined(USE_64BIT_INDICES)) PetscCall(PetscFree(pjcns_w));
12679d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) PetscCall(ISRestoreIndices(rows[r], (const PetscInt **)&rows_idx[r]));
12689d0448ceSStefano Zampini     for (PetscInt c = 0; c < nc; c++) PetscCall(ISRestoreIndices(cols[c], (const PetscInt **)&cols_idx[c]));
12699d0448ceSStefano Zampini     PetscCall(PetscFree4(rows, cols, rows_idx, cols_idx));
12706497c311SBarry Smith     if (!chol) PetscCheck(cumnnz == totnnz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Different number of nonzeros %" PetscCount_FMT " != %" PetscCount_FMT, cumnnz, totnnz);
12715d955bbbSStefano Zampini     mumps->nest_vals_start[nr * nc] = cumnnz;
12729d0448ceSStefano Zampini 
12739d0448ceSStefano Zampini     /* Set pointers for final MUMPS data structure */
12749d0448ceSStefano Zampini     mumps->nest_vals = vals;
12759d0448ceSStefano Zampini     mumps->val_alloc = NULL; /* do not use val_alloc since it may be reallocated with the OMP callpath */
12769d0448ceSStefano Zampini     mumps->val       = vals;
12779d0448ceSStefano Zampini     mumps->irn       = irns;
12789d0448ceSStefano Zampini     mumps->jcn       = jcns;
12799d0448ceSStefano Zampini     mumps->nnz       = cumnnz;
12809d0448ceSStefano Zampini   } else {
12819d0448ceSStefano Zampini     PetscScalar *oval = mumps->nest_vals;
12829d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) {
12839d0448ceSStefano Zampini       for (PetscInt c = 0; c < nc; c++) {
128453587d93SPierre Jolivet         PetscBool   conjugate = PETSC_FALSE;
12855d955bbbSStefano Zampini         Mat         sub       = mats[r][c];
128653587d93SPierre Jolivet         PetscScalar vscale = 1.0, vshift = 0.0;
12875d955bbbSStefano Zampini         PetscInt    midx = r * nc + c;
12885d955bbbSStefano Zampini 
12895d955bbbSStefano Zampini         if (!mumps->nest_convert_to_triples[midx]) continue;
129053587d93SPierre Jolivet         PetscCall(MatGetTranspose_TransposeVirtual(&sub, &conjugate, &vshift, &vscale, NULL));
129153587d93SPierre Jolivet         PetscCheck(vshift == 0.0, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Nonzero shift in parent MatShell");
12925d955bbbSStefano Zampini         mumps->val = oval + mumps->nest_vals_start[midx];
12935d955bbbSStefano Zampini         PetscCall((*mumps->nest_convert_to_triples[midx])(sub, shift, MAT_REUSE_MATRIX, mumps));
129453587d93SPierre Jolivet         if (conjugate) {
12956497c311SBarry Smith           PetscCount nnz = mumps->nest_vals_start[midx + 1] - mumps->nest_vals_start[midx];
129653587d93SPierre Jolivet           for (PetscCount k = 0; k < nnz; k++) mumps->val[k] = vscale * PetscConj(mumps->val[k]);
129753587d93SPierre Jolivet         } else if (vscale != 1.0) {
129853587d93SPierre Jolivet           PetscCount nnz = mumps->nest_vals_start[midx + 1] - mumps->nest_vals_start[midx];
129953587d93SPierre Jolivet           for (PetscCount k = 0; k < nnz; k++) mumps->val[k] *= vscale;
13005d955bbbSStefano Zampini         }
13019d0448ceSStefano Zampini       }
13029d0448ceSStefano Zampini     }
13039d0448ceSStefano Zampini     mumps->val = oval;
13049d0448ceSStefano Zampini   }
13059d0448ceSStefano Zampini   PetscFunctionReturn(PETSC_SUCCESS);
13069d0448ceSStefano Zampini }
13079d0448ceSStefano Zampini 
130866976f2fSJacob Faibussowitsch static PetscErrorCode MatDestroy_MUMPS(Mat A)
1309d71ae5a4SJacob Faibussowitsch {
1310a6053eceSJunchao Zhang   Mat_MUMPS *mumps = (Mat_MUMPS *)A->data;
1311b24902e0SBarry Smith 
1312397b6df1SKris Buschelman   PetscFunctionBegin;
13139566063dSJacob Faibussowitsch   PetscCall(PetscFree2(mumps->id.sol_loc, mumps->id.isol_loc));
13149566063dSJacob Faibussowitsch   PetscCall(VecScatterDestroy(&mumps->scat_rhs));
13159566063dSJacob Faibussowitsch   PetscCall(VecScatterDestroy(&mumps->scat_sol));
13169566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&mumps->b_seq));
13179566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&mumps->x_seq));
13189566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.perm_in));
13199566063dSJacob Faibussowitsch   PetscCall(PetscFree2(mumps->irn, mumps->jcn));
13209566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->val_alloc));
13219566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->info));
1322413bcc21SPierre Jolivet   PetscCall(PetscFree(mumps->ICNTL_pre));
1323413bcc21SPierre Jolivet   PetscCall(PetscFree(mumps->CNTL_pre));
13249566063dSJacob Faibussowitsch   PetscCall(MatMumpsResetSchur_Private(mumps));
1325413bcc21SPierre Jolivet   if (mumps->id.job != JOB_NULL) { /* cannot call PetscMUMPS_c() if JOB_INIT has never been called for this instance */
1326a5e57a09SHong Zhang     mumps->id.job = JOB_END;
13273ab56b82SJunchao Zhang     PetscMUMPS_c(mumps);
13289261f6e4SBarry Smith     PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in termination: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
1329413bcc21SPierre Jolivet     if (mumps->mumps_comm != MPI_COMM_NULL) {
1330413bcc21SPierre Jolivet       if (PetscDefined(HAVE_OPENMP_SUPPORT) && mumps->use_petsc_omp_support) PetscCallMPI(MPI_Comm_free(&mumps->mumps_comm));
1331413bcc21SPierre Jolivet       else PetscCall(PetscCommRestoreComm(PetscObjectComm((PetscObject)A), &mumps->mumps_comm));
1332413bcc21SPierre Jolivet     }
1333413bcc21SPierre Jolivet   }
13343ab56b82SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
133567602552SJunchao Zhang   if (mumps->use_petsc_omp_support) {
13369566063dSJacob Faibussowitsch     PetscCall(PetscOmpCtrlDestroy(&mumps->omp_ctrl));
13379566063dSJacob Faibussowitsch     PetscCall(PetscFree2(mumps->rhs_loc, mumps->rhs_recvbuf));
13389566063dSJacob Faibussowitsch     PetscCall(PetscFree3(mumps->rhs_nrow, mumps->rhs_recvcounts, mumps->rhs_disps));
133967602552SJunchao Zhang   }
13403ab56b82SJunchao Zhang #endif
13419566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->ia_alloc));
13429566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->ja_alloc));
13439566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->recvcount));
13449566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->reqs));
13459566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->irhs_loc));
13469d0448ceSStefano Zampini   PetscCall(PetscFree2(mumps->nest_vals_start, mumps->nest_convert_to_triples));
13479d0448ceSStefano Zampini   PetscCall(PetscFree(mumps->nest_vals));
13489566063dSJacob Faibussowitsch   PetscCall(PetscFree(A->data));
1349bf0cc555SLisandro Dalcin 
135097969023SHong Zhang   /* clear composed functions */
13519566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorGetSolverType_C", NULL));
13529566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorSetSchurIS_C", NULL));
13539566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorCreateSchurComplement_C", NULL));
13549566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsSetIcntl_C", NULL));
13559566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetIcntl_C", NULL));
13569566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsSetCntl_C", NULL));
13579566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetCntl_C", NULL));
13589566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInfo_C", NULL));
13599566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInfog_C", NULL));
13609566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetRinfo_C", NULL));
13619566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetRinfog_C", NULL));
13625c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetNullPivots_C", NULL));
13639566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInverse_C", NULL));
13649566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInverseTranspose_C", NULL));
13653ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1366397b6df1SKris Buschelman }
1367397b6df1SKris Buschelman 
136867602552SJunchao Zhang /* Set up the distributed RHS info for MUMPS. <nrhs> is the number of RHS. <array> points to start of RHS on the local processor. */
1369d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMumpsSetUpDistRHSInfo(Mat A, PetscInt nrhs, const PetscScalar *array)
1370d71ae5a4SJacob Faibussowitsch {
137167602552SJunchao Zhang   Mat_MUMPS        *mumps   = (Mat_MUMPS *)A->data;
137267602552SJunchao Zhang   const PetscMPIInt ompsize = mumps->omp_comm_size;
137367602552SJunchao Zhang   PetscInt          i, m, M, rstart;
137467602552SJunchao Zhang 
137567602552SJunchao Zhang   PetscFunctionBegin;
13769566063dSJacob Faibussowitsch   PetscCall(MatGetSize(A, &M, NULL));
13779566063dSJacob Faibussowitsch   PetscCall(MatGetLocalSize(A, &m, NULL));
137808401ef6SPierre Jolivet   PetscCheck(M <= PETSC_MUMPS_INT_MAX, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "PetscInt too long for PetscMUMPSInt");
137967602552SJunchao Zhang   if (ompsize == 1) {
138067602552SJunchao Zhang     if (!mumps->irhs_loc) {
13816497c311SBarry Smith       mumps->nloc_rhs = (PetscMUMPSInt)m;
13829566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(m, &mumps->irhs_loc));
13839566063dSJacob Faibussowitsch       PetscCall(MatGetOwnershipRange(A, &rstart, NULL));
13846497c311SBarry Smith       for (i = 0; i < m; i++) PetscCall(PetscMUMPSIntCast(rstart + i + 1, &mumps->irhs_loc[i])); /* use 1-based indices */
138567602552SJunchao Zhang     }
138667602552SJunchao Zhang     mumps->id.rhs_loc = (MumpsScalar *)array;
138767602552SJunchao Zhang   } else {
138867602552SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
138967602552SJunchao Zhang     const PetscInt *ranges;
139067602552SJunchao Zhang     PetscMPIInt     j, k, sendcount, *petsc_ranks, *omp_ranks;
139167602552SJunchao Zhang     MPI_Group       petsc_group, omp_group;
139267602552SJunchao Zhang     PetscScalar    *recvbuf = NULL;
139367602552SJunchao Zhang 
139467602552SJunchao Zhang     if (mumps->is_omp_master) {
139567602552SJunchao Zhang       /* Lazily initialize the omp stuff for distributed rhs */
139667602552SJunchao Zhang       if (!mumps->irhs_loc) {
13979566063dSJacob Faibussowitsch         PetscCall(PetscMalloc2(ompsize, &omp_ranks, ompsize, &petsc_ranks));
13989566063dSJacob Faibussowitsch         PetscCall(PetscMalloc3(ompsize, &mumps->rhs_nrow, ompsize, &mumps->rhs_recvcounts, ompsize, &mumps->rhs_disps));
13999566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Comm_group(mumps->petsc_comm, &petsc_group));
14009566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Comm_group(mumps->omp_comm, &omp_group));
140167602552SJunchao Zhang         for (j = 0; j < ompsize; j++) omp_ranks[j] = j;
14029566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Group_translate_ranks(omp_group, ompsize, omp_ranks, petsc_group, petsc_ranks));
140367602552SJunchao Zhang 
140467602552SJunchao Zhang         /* Populate mumps->irhs_loc[], rhs_nrow[] */
140567602552SJunchao Zhang         mumps->nloc_rhs = 0;
14069566063dSJacob Faibussowitsch         PetscCall(MatGetOwnershipRanges(A, &ranges));
140767602552SJunchao Zhang         for (j = 0; j < ompsize; j++) {
140867602552SJunchao Zhang           mumps->rhs_nrow[j] = ranges[petsc_ranks[j] + 1] - ranges[petsc_ranks[j]];
140967602552SJunchao Zhang           mumps->nloc_rhs += mumps->rhs_nrow[j];
141067602552SJunchao Zhang         }
14119566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(mumps->nloc_rhs, &mumps->irhs_loc));
141267602552SJunchao Zhang         for (j = k = 0; j < ompsize; j++) {
141367602552SJunchao Zhang           for (i = ranges[petsc_ranks[j]]; i < ranges[petsc_ranks[j] + 1]; i++, k++) mumps->irhs_loc[k] = i + 1; /* uses 1-based indices */
141467602552SJunchao Zhang         }
141567602552SJunchao Zhang 
14169566063dSJacob Faibussowitsch         PetscCall(PetscFree2(omp_ranks, petsc_ranks));
14179566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Group_free(&petsc_group));
14189566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Group_free(&omp_group));
141967602552SJunchao Zhang       }
142067602552SJunchao Zhang 
142167602552SJunchao Zhang       /* Realloc buffers when current nrhs is bigger than what we have met */
142267602552SJunchao Zhang       if (nrhs > mumps->max_nrhs) {
14239566063dSJacob Faibussowitsch         PetscCall(PetscFree2(mumps->rhs_loc, mumps->rhs_recvbuf));
14249566063dSJacob Faibussowitsch         PetscCall(PetscMalloc2(mumps->nloc_rhs * nrhs, &mumps->rhs_loc, mumps->nloc_rhs * nrhs, &mumps->rhs_recvbuf));
142567602552SJunchao Zhang         mumps->max_nrhs = nrhs;
142667602552SJunchao Zhang       }
142767602552SJunchao Zhang 
142867602552SJunchao Zhang       /* Setup recvcounts[], disps[], recvbuf on omp rank 0 for the upcoming MPI_Gatherv */
14299566063dSJacob Faibussowitsch       for (j = 0; j < ompsize; j++) PetscCall(PetscMPIIntCast(mumps->rhs_nrow[j] * nrhs, &mumps->rhs_recvcounts[j]));
143067602552SJunchao Zhang       mumps->rhs_disps[0] = 0;
143167602552SJunchao Zhang       for (j = 1; j < ompsize; j++) {
143267602552SJunchao Zhang         mumps->rhs_disps[j] = mumps->rhs_disps[j - 1] + mumps->rhs_recvcounts[j - 1];
143308401ef6SPierre Jolivet         PetscCheck(mumps->rhs_disps[j] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "PetscMPIInt overflow!");
143467602552SJunchao Zhang       }
143567602552SJunchao Zhang       recvbuf = (nrhs == 1) ? mumps->rhs_loc : mumps->rhs_recvbuf; /* Directly use rhs_loc[] as recvbuf. Single rhs is common in Ax=b */
143667602552SJunchao Zhang     }
143767602552SJunchao Zhang 
14389566063dSJacob Faibussowitsch     PetscCall(PetscMPIIntCast(m * nrhs, &sendcount));
14399566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Gatherv(array, sendcount, MPIU_SCALAR, recvbuf, mumps->rhs_recvcounts, mumps->rhs_disps, MPIU_SCALAR, 0, mumps->omp_comm));
144067602552SJunchao Zhang 
144167602552SJunchao Zhang     if (mumps->is_omp_master) {
144267602552SJunchao Zhang       if (nrhs > 1) { /* Copy & re-arrange data from rhs_recvbuf[] to mumps->rhs_loc[] only when there are multiple rhs */
144367602552SJunchao Zhang         PetscScalar *dst, *dstbase = mumps->rhs_loc;
144467602552SJunchao Zhang         for (j = 0; j < ompsize; j++) {
144567602552SJunchao Zhang           const PetscScalar *src = mumps->rhs_recvbuf + mumps->rhs_disps[j];
144667602552SJunchao Zhang           dst                    = dstbase;
144767602552SJunchao Zhang           for (i = 0; i < nrhs; i++) {
14489566063dSJacob Faibussowitsch             PetscCall(PetscArraycpy(dst, src, mumps->rhs_nrow[j]));
144967602552SJunchao Zhang             src += mumps->rhs_nrow[j];
145067602552SJunchao Zhang             dst += mumps->nloc_rhs;
145167602552SJunchao Zhang           }
145267602552SJunchao Zhang           dstbase += mumps->rhs_nrow[j];
145367602552SJunchao Zhang         }
145467602552SJunchao Zhang       }
145567602552SJunchao Zhang       mumps->id.rhs_loc = (MumpsScalar *)mumps->rhs_loc;
145667602552SJunchao Zhang     }
145767602552SJunchao Zhang #endif /* PETSC_HAVE_OPENMP_SUPPORT */
145867602552SJunchao Zhang   }
14596497c311SBarry Smith   mumps->id.nrhs     = (PetscMUMPSInt)nrhs;
14606497c311SBarry Smith   mumps->id.nloc_rhs = (PetscMUMPSInt)mumps->nloc_rhs;
146167602552SJunchao Zhang   mumps->id.lrhs_loc = mumps->nloc_rhs;
146267602552SJunchao Zhang   mumps->id.irhs_loc = mumps->irhs_loc;
14633ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
146467602552SJunchao Zhang }
146567602552SJunchao Zhang 
146666976f2fSJacob Faibussowitsch static PetscErrorCode MatSolve_MUMPS(Mat A, Vec b, Vec x)
1467d71ae5a4SJacob Faibussowitsch {
1468e69c285eSBarry Smith   Mat_MUMPS         *mumps  = (Mat_MUMPS *)A->data;
146925aac85cSJunchao Zhang   const PetscScalar *rarray = NULL;
1470d54de34fSKris Buschelman   PetscScalar       *array;
1471329ec9b3SHong Zhang   IS                 is_iden, is_petsc;
1472329ec9b3SHong Zhang   PetscInt           i;
1473cc86f929SStefano Zampini   PetscBool          second_solve = PETSC_FALSE;
1474883f2eb9SBarry Smith   static PetscBool   cite1 = PETSC_FALSE, cite2 = PETSC_FALSE;
1475397b6df1SKris Buschelman 
1476397b6df1SKris Buschelman   PetscFunctionBegin;
14779371c9d4SSatish Balay   PetscCall(PetscCitationsRegister("@article{MUMPS01,\n  author = {P.~R. Amestoy and I.~S. Duff and J.-Y. L'Excellent and J. Koster},\n  title = {A fully asynchronous multifrontal solver using distributed dynamic scheduling},\n  journal = {SIAM "
14789371c9d4SSatish Balay                                    "Journal on Matrix Analysis and Applications},\n  volume = {23},\n  number = {1},\n  pages = {15--41},\n  year = {2001}\n}\n",
14799371c9d4SSatish Balay                                    &cite1));
14809371c9d4SSatish Balay   PetscCall(PetscCitationsRegister("@article{MUMPS02,\n  author = {P.~R. Amestoy and A. Guermouche and J.-Y. L'Excellent and S. Pralet},\n  title = {Hybrid scheduling for the parallel solution of linear systems},\n  journal = {Parallel "
14819371c9d4SSatish Balay                                    "Computing},\n  volume = {32},\n  number = {2},\n  pages = {136--156},\n  year = {2006}\n}\n",
14829371c9d4SSatish Balay                                    &cite2));
14832aca8efcSHong Zhang 
1484f480ea8aSBarry Smith   PetscCall(VecFlag(x, A->factorerrortype));
1485603e8f96SBarry Smith   if (A->factorerrortype) {
14869566063dSJacob Faibussowitsch     PetscCall(PetscInfo(A, "MatSolve is called with singular matrix factor, INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
14873ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
14882aca8efcSHong Zhang   }
14892aca8efcSHong Zhang 
1490a5e57a09SHong Zhang   mumps->id.nrhs = 1;
14912d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) {
149225aac85cSJunchao Zhang     if (mumps->ICNTL20 == 10) {
149367602552SJunchao Zhang       mumps->id.ICNTL(20) = 10; /* dense distributed RHS */
14949566063dSJacob Faibussowitsch       PetscCall(VecGetArrayRead(b, &rarray));
14959566063dSJacob Faibussowitsch       PetscCall(MatMumpsSetUpDistRHSInfo(A, 1, rarray));
149625aac85cSJunchao Zhang     } else {
149741ffd417SStefano Zampini       mumps->id.ICNTL(20) = 0; /* dense centralized RHS; Scatter b into a sequential rhs vector*/
14989566063dSJacob Faibussowitsch       PetscCall(VecScatterBegin(mumps->scat_rhs, b, mumps->b_seq, INSERT_VALUES, SCATTER_FORWARD));
14999566063dSJacob Faibussowitsch       PetscCall(VecScatterEnd(mumps->scat_rhs, b, mumps->b_seq, INSERT_VALUES, SCATTER_FORWARD));
150067602552SJunchao Zhang       if (!mumps->myid) {
15019566063dSJacob Faibussowitsch         PetscCall(VecGetArray(mumps->b_seq, &array));
150267602552SJunchao Zhang         mumps->id.rhs = (MumpsScalar *)array;
150367602552SJunchao Zhang       }
150425aac85cSJunchao Zhang     }
15053ab56b82SJunchao Zhang   } else {                   /* petsc_size == 1 */
150667602552SJunchao Zhang     mumps->id.ICNTL(20) = 0; /* dense centralized RHS */
15079566063dSJacob Faibussowitsch     PetscCall(VecCopy(b, x));
15089566063dSJacob Faibussowitsch     PetscCall(VecGetArray(x, &array));
1509940cd9d6SSatish Balay     mumps->id.rhs = (MumpsScalar *)array;
1510397b6df1SKris Buschelman   }
1511397b6df1SKris Buschelman 
1512cc86f929SStefano Zampini   /*
1513cc86f929SStefano Zampini      handle condensation step of Schur complement (if any)
1514cc86f929SStefano Zampini      We set by default ICNTL(26) == -1 when Schur indices have been provided by the user.
1515cc86f929SStefano Zampini      According to MUMPS (5.0.0) manual, any value should be harmful during the factorization phase
1516cc86f929SStefano Zampini      Unless the user provides a valid value for ICNTL(26), MatSolve and MatMatSolve routines solve the full system.
1517cc86f929SStefano Zampini      This requires an extra call to PetscMUMPS_c and the computation of the factors for S
1518cc86f929SStefano Zampini   */
15193e5b40d0SPierre Jolivet   if (mumps->id.size_schur > 0) {
152008401ef6SPierre Jolivet     PetscCheck(mumps->petsc_size <= 1, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Parallel Schur complements not yet supported from PETSc");
15213e5b40d0SPierre Jolivet     if (mumps->id.ICNTL(26) < 0 || mumps->id.ICNTL(26) > 2) {
1522cc86f929SStefano Zampini       second_solve = PETSC_TRUE;
15239566063dSJacob Faibussowitsch       PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
15243e5b40d0SPierre Jolivet       mumps->id.ICNTL(26) = 1; /* condensation phase */
15253e5b40d0SPierre Jolivet     } else if (mumps->id.ICNTL(26) == 1) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
1526cc86f929SStefano Zampini   }
1527397b6df1SKris Buschelman   /* solve phase */
1528a5e57a09SHong Zhang   mumps->id.job = JOB_SOLVE;
15293ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
15309261f6e4SBarry Smith   PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in solve: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
1531397b6df1SKris Buschelman 
1532b5fa320bSStefano Zampini   /* handle expansion step of Schur complement (if any) */
15331baa6e33SBarry Smith   if (second_solve) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_TRUE));
15343e5b40d0SPierre Jolivet   else if (mumps->id.ICNTL(26) == 1) {
15353e5b40d0SPierre Jolivet     PetscCall(MatMumpsSolveSchur_Private(A));
15363e5b40d0SPierre Jolivet     for (i = 0; i < mumps->id.size_schur; ++i) {
15373e5b40d0SPierre Jolivet #if !defined(PETSC_USE_COMPLEX)
15383e5b40d0SPierre Jolivet       PetscScalar val = mumps->id.redrhs[i];
15393e5b40d0SPierre Jolivet #else
15403e5b40d0SPierre Jolivet       PetscScalar val = mumps->id.redrhs[i].r + PETSC_i * mumps->id.redrhs[i].i;
15413e5b40d0SPierre Jolivet #endif
15423e5b40d0SPierre Jolivet       array[mumps->id.listvar_schur[i] - 1] = val;
15433e5b40d0SPierre Jolivet     }
15443e5b40d0SPierre Jolivet   }
1545b5fa320bSStefano Zampini 
1546f0b74427SPierre Jolivet   if (mumps->petsc_size > 1) { /* convert mumps distributed solution to PETSc mpi x */
1547a5e57a09SHong Zhang     if (mumps->scat_sol && mumps->ICNTL9_pre != mumps->id.ICNTL(9)) {
1548a5e57a09SHong Zhang       /* when id.ICNTL(9) changes, the contents of lsol_loc may change (not its size, lsol_loc), recreates scat_sol */
15499566063dSJacob Faibussowitsch       PetscCall(VecScatterDestroy(&mumps->scat_sol));
1550397b6df1SKris Buschelman     }
1551a5e57a09SHong Zhang     if (!mumps->scat_sol) { /* create scatter scat_sol */
1552a6053eceSJunchao Zhang       PetscInt *isol2_loc = NULL;
15539566063dSJacob Faibussowitsch       PetscCall(ISCreateStride(PETSC_COMM_SELF, mumps->id.lsol_loc, 0, 1, &is_iden)); /* from */
15549566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(mumps->id.lsol_loc, &isol2_loc));
1555a6053eceSJunchao Zhang       for (i = 0; i < mumps->id.lsol_loc; i++) isol2_loc[i] = mumps->id.isol_loc[i] - 1;                        /* change Fortran style to C style */
15569566063dSJacob Faibussowitsch       PetscCall(ISCreateGeneral(PETSC_COMM_SELF, mumps->id.lsol_loc, isol2_loc, PETSC_OWN_POINTER, &is_petsc)); /* to */
15579566063dSJacob Faibussowitsch       PetscCall(VecScatterCreate(mumps->x_seq, is_iden, x, is_petsc, &mumps->scat_sol));
15589566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_iden));
15599566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_petsc));
1560a5e57a09SHong Zhang       mumps->ICNTL9_pre = mumps->id.ICNTL(9); /* save current value of id.ICNTL(9) */
1561397b6df1SKris Buschelman     }
1562a5e57a09SHong Zhang 
15639566063dSJacob Faibussowitsch     PetscCall(VecScatterBegin(mumps->scat_sol, mumps->x_seq, x, INSERT_VALUES, SCATTER_FORWARD));
15649566063dSJacob Faibussowitsch     PetscCall(VecScatterEnd(mumps->scat_sol, mumps->x_seq, x, INSERT_VALUES, SCATTER_FORWARD));
1565329ec9b3SHong Zhang   }
1566353d7d71SJunchao Zhang 
156767602552SJunchao Zhang   if (mumps->petsc_size > 1) {
156825aac85cSJunchao Zhang     if (mumps->ICNTL20 == 10) {
15699566063dSJacob Faibussowitsch       PetscCall(VecRestoreArrayRead(b, &rarray));
157025aac85cSJunchao Zhang     } else if (!mumps->myid) {
15719566063dSJacob Faibussowitsch       PetscCall(VecRestoreArray(mumps->b_seq, &array));
157225aac85cSJunchao Zhang     }
15739566063dSJacob Faibussowitsch   } else PetscCall(VecRestoreArray(x, &array));
1574353d7d71SJunchao Zhang 
157564412097SPierre Jolivet   PetscCall(PetscLogFlops(2.0 * PetscMax(0, (mumps->id.INFO(28) >= 0 ? mumps->id.INFO(28) : -1000000 * mumps->id.INFO(28)) - A->cmap->n)));
15763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1577397b6df1SKris Buschelman }
1578397b6df1SKris Buschelman 
157966976f2fSJacob Faibussowitsch static PetscErrorCode MatSolveTranspose_MUMPS(Mat A, Vec b, Vec x)
1580d71ae5a4SJacob Faibussowitsch {
1581e69c285eSBarry Smith   Mat_MUMPS          *mumps = (Mat_MUMPS *)A->data;
1582338d3105SPierre Jolivet   const PetscMUMPSInt value = mumps->id.ICNTL(9);
158351d5961aSHong Zhang 
158451d5961aSHong Zhang   PetscFunctionBegin;
1585a5e57a09SHong Zhang   mumps->id.ICNTL(9) = 0;
15869566063dSJacob Faibussowitsch   PetscCall(MatSolve_MUMPS(A, b, x));
1587338d3105SPierre Jolivet   mumps->id.ICNTL(9) = value;
15883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
158951d5961aSHong Zhang }
159051d5961aSHong Zhang 
159166976f2fSJacob Faibussowitsch static PetscErrorCode MatMatSolve_MUMPS(Mat A, Mat B, Mat X)
1592d71ae5a4SJacob Faibussowitsch {
1593b8491c3eSStefano Zampini   Mat                Bt = NULL;
1594a6053eceSJunchao Zhang   PetscBool          denseX, denseB, flg, flgT;
1595e69c285eSBarry Smith   Mat_MUMPS         *mumps = (Mat_MUMPS *)A->data;
1596917c3dccSPierre Jolivet   PetscInt           i, nrhs, M, nrhsM;
15971683a169SBarry Smith   PetscScalar       *array;
15981683a169SBarry Smith   const PetscScalar *rbray;
1599a6053eceSJunchao Zhang   PetscInt           lsol_loc, nlsol_loc, *idxx, iidx = 0;
1600a6053eceSJunchao Zhang   PetscMUMPSInt     *isol_loc, *isol_loc_save;
16011683a169SBarry Smith   PetscScalar       *bray, *sol_loc, *sol_loc_save;
1602be818407SHong Zhang   IS                 is_to, is_from;
1603beae5ec0SHong Zhang   PetscInt           k, proc, j, m, myrstart;
1604be818407SHong Zhang   const PetscInt    *rstart;
160567602552SJunchao Zhang   Vec                v_mpi, msol_loc;
160667602552SJunchao Zhang   VecScatter         scat_sol;
160767602552SJunchao Zhang   Vec                b_seq;
160867602552SJunchao Zhang   VecScatter         scat_rhs;
1609be818407SHong Zhang   PetscScalar       *aa;
1610be818407SHong Zhang   PetscInt           spnr, *ia, *ja;
1611d56c302dSHong Zhang   Mat_MPIAIJ        *b = NULL;
1612bda8bf91SBarry Smith 
1613e0b74bf9SHong Zhang   PetscFunctionBegin;
16149566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)X, &denseX, MATSEQDENSE, MATMPIDENSE, NULL));
161528b400f6SJacob Faibussowitsch   PetscCheck(denseX, PetscObjectComm((PetscObject)X), PETSC_ERR_ARG_WRONG, "Matrix X must be MATDENSE matrix");
1616be818407SHong Zhang 
16179566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)B, &denseB, MATSEQDENSE, MATMPIDENSE, NULL));
1618a6053eceSJunchao Zhang   if (denseB) {
161908401ef6SPierre Jolivet     PetscCheck(B->rmap->n == X->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Matrix B and X must have same row distribution");
1620be818407SHong Zhang     mumps->id.ICNTL(20) = 0; /* dense RHS */
16210e6b8875SHong Zhang   } else {                   /* sparse B */
162208401ef6SPierre Jolivet     PetscCheck(X != B, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_IDN, "X and B must be different matrices");
1623013e2dc7SBarry Smith     PetscCall(PetscObjectTypeCompare((PetscObject)B, MATTRANSPOSEVIRTUAL, &flgT));
162453587d93SPierre Jolivet     PetscCheck(flgT, PetscObjectComm((PetscObject)B), PETSC_ERR_ARG_WRONG, "Matrix B must be MATTRANSPOSEVIRTUAL matrix");
162553587d93SPierre Jolivet     PetscCall(MatShellGetScalingShifts(B, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Mat *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED));
162653587d93SPierre Jolivet     /* input B is transpose of actual RHS matrix,
16270e6b8875SHong Zhang      because mumps requires sparse compressed COLUMN storage! See MatMatTransposeSolve_MUMPS() */
16289566063dSJacob Faibussowitsch     PetscCall(MatTransposeGetMat(B, &Bt));
1629be818407SHong Zhang     mumps->id.ICNTL(20) = 1; /* sparse RHS */
1630b8491c3eSStefano Zampini   }
163187b22cf4SHong Zhang 
16329566063dSJacob Faibussowitsch   PetscCall(MatGetSize(B, &M, &nrhs));
1633917c3dccSPierre Jolivet   PetscCall(PetscIntMultError(nrhs, M, &nrhsM));
16346497c311SBarry Smith   mumps->id.nrhs = (PetscMUMPSInt)nrhs;
16356497c311SBarry Smith   mumps->id.lrhs = (PetscMUMPSInt)M;
16362b691707SHong Zhang   mumps->id.rhs  = NULL;
16379481e6e9SHong Zhang 
16382d4298aeSJunchao Zhang   if (mumps->petsc_size == 1) {
1639b8491c3eSStefano Zampini     PetscScalar *aa;
1640b8491c3eSStefano Zampini     PetscInt     spnr, *ia, *ja;
1641e94cce23SStefano Zampini     PetscBool    second_solve = PETSC_FALSE;
1642b8491c3eSStefano Zampini 
16439566063dSJacob Faibussowitsch     PetscCall(MatDenseGetArray(X, &array));
1644b8491c3eSStefano Zampini     mumps->id.rhs = (MumpsScalar *)array;
16452b691707SHong Zhang 
1646a6053eceSJunchao Zhang     if (denseB) {
16472b691707SHong Zhang       /* copy B to X */
16489566063dSJacob Faibussowitsch       PetscCall(MatDenseGetArrayRead(B, &rbray));
1649917c3dccSPierre Jolivet       PetscCall(PetscArraycpy(array, rbray, nrhsM));
16509566063dSJacob Faibussowitsch       PetscCall(MatDenseRestoreArrayRead(B, &rbray));
16512b691707SHong Zhang     } else { /* sparse B */
16529566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJGetArray(Bt, &aa));
16539566063dSJacob Faibussowitsch       PetscCall(MatGetRowIJ(Bt, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
165428b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
16559566063dSJacob Faibussowitsch       PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs));
1656b8491c3eSStefano Zampini       mumps->id.rhs_sparse = (MumpsScalar *)aa;
1657b8491c3eSStefano Zampini     }
1658e94cce23SStefano Zampini     /* handle condensation step of Schur complement (if any) */
16593e5b40d0SPierre Jolivet     if (mumps->id.size_schur > 0) {
16603e5b40d0SPierre Jolivet       if (mumps->id.ICNTL(26) < 0 || mumps->id.ICNTL(26) > 2) {
1661e94cce23SStefano Zampini         second_solve = PETSC_TRUE;
16629566063dSJacob Faibussowitsch         PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
16633e5b40d0SPierre Jolivet         mumps->id.ICNTL(26) = 1; /* condensation phase */
16643e5b40d0SPierre Jolivet       } else if (mumps->id.ICNTL(26) == 1) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
1665e94cce23SStefano Zampini     }
16662cd7d884SHong Zhang     /* solve phase */
16672cd7d884SHong Zhang     mumps->id.job = JOB_SOLVE;
16683ab56b82SJunchao Zhang     PetscMUMPS_c(mumps);
16699261f6e4SBarry Smith     PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in solve: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
1670b5fa320bSStefano Zampini 
1671b5fa320bSStefano Zampini     /* handle expansion step of Schur complement (if any) */
16721baa6e33SBarry Smith     if (second_solve) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_TRUE));
16733e5b40d0SPierre Jolivet     else if (mumps->id.ICNTL(26) == 1) {
16743e5b40d0SPierre Jolivet       PetscCall(MatMumpsSolveSchur_Private(A));
16753e5b40d0SPierre Jolivet       for (j = 0; j < nrhs; ++j)
16763e5b40d0SPierre Jolivet         for (i = 0; i < mumps->id.size_schur; ++i) {
16773e5b40d0SPierre Jolivet #if !defined(PETSC_USE_COMPLEX)
16783e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.redrhs[i + j * mumps->id.lredrhs];
16793e5b40d0SPierre Jolivet #else
16803e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.redrhs[i + j * mumps->id.lredrhs].r + PETSC_i * mumps->id.redrhs[i + j * mumps->id.lredrhs].i;
16813e5b40d0SPierre Jolivet #endif
16823e5b40d0SPierre Jolivet           array[mumps->id.listvar_schur[i] - 1 + j * M] = val;
16833e5b40d0SPierre Jolivet         }
16843e5b40d0SPierre Jolivet     }
1685a6053eceSJunchao Zhang     if (!denseB) { /* sparse B */
16869566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJRestoreArray(Bt, &aa));
16879566063dSJacob Faibussowitsch       PetscCall(MatRestoreRowIJ(Bt, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
168828b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot restore IJ structure");
1689b8491c3eSStefano Zampini     }
16909566063dSJacob Faibussowitsch     PetscCall(MatDenseRestoreArray(X, &array));
16913ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
1692be818407SHong Zhang   }
1693801fbe65SHong Zhang 
16942ef1f0ffSBarry Smith   /* parallel case: MUMPS requires rhs B to be centralized on the host! */
169550a7cd33SPierre Jolivet   PetscCheck(!mumps->id.ICNTL(19), PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Parallel Schur complements not yet supported from PETSc");
1696241dbb5eSStefano Zampini 
1697beae5ec0SHong Zhang   /* create msol_loc to hold mumps local solution */
16981683a169SBarry Smith   isol_loc_save = mumps->id.isol_loc; /* save it for MatSolve() */
16991683a169SBarry Smith   sol_loc_save  = (PetscScalar *)mumps->id.sol_loc;
1700801fbe65SHong Zhang 
1701a1dfcbd9SJunchao Zhang   lsol_loc = mumps->id.lsol_loc;
1702917c3dccSPierre Jolivet   PetscCall(PetscIntMultError(nrhs, lsol_loc, &nlsol_loc)); /* length of sol_loc */
17039566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(nlsol_loc, &sol_loc, lsol_loc, &isol_loc));
1704940cd9d6SSatish Balay   mumps->id.sol_loc  = (MumpsScalar *)sol_loc;
1705801fbe65SHong Zhang   mumps->id.isol_loc = isol_loc;
1706801fbe65SHong Zhang 
17079566063dSJacob Faibussowitsch   PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, nlsol_loc, (PetscScalar *)sol_loc, &msol_loc));
17082cd7d884SHong Zhang 
170967602552SJunchao Zhang   if (denseB) {
171025aac85cSJunchao Zhang     if (mumps->ICNTL20 == 10) {
171167602552SJunchao Zhang       mumps->id.ICNTL(20) = 10; /* dense distributed RHS */
17129566063dSJacob Faibussowitsch       PetscCall(MatDenseGetArrayRead(B, &rbray));
17139566063dSJacob Faibussowitsch       PetscCall(MatMumpsSetUpDistRHSInfo(A, nrhs, rbray));
17149566063dSJacob Faibussowitsch       PetscCall(MatDenseRestoreArrayRead(B, &rbray));
17159566063dSJacob Faibussowitsch       PetscCall(MatGetLocalSize(B, &m, NULL));
1716917c3dccSPierre Jolivet       PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)B), 1, nrhs * m, nrhsM, NULL, &v_mpi));
171725aac85cSJunchao Zhang     } else {
171825aac85cSJunchao Zhang       mumps->id.ICNTL(20) = 0; /* dense centralized RHS */
171980577c12SJunchao Zhang       /* TODO: Because of non-contiguous indices, the created vecscatter scat_rhs is not done in MPI_Gather, resulting in
172080577c12SJunchao Zhang         very inefficient communication. An optimization is to use VecScatterCreateToZero to gather B to rank 0. Then on rank
172180577c12SJunchao Zhang         0, re-arrange B into desired order, which is a local operation.
172280577c12SJunchao Zhang       */
172380577c12SJunchao Zhang 
172467602552SJunchao Zhang       /* scatter v_mpi to b_seq because MUMPS before 5.3.0 only supports centralized rhs */
1725be818407SHong Zhang       /* wrap dense rhs matrix B into a vector v_mpi */
17269566063dSJacob Faibussowitsch       PetscCall(MatGetLocalSize(B, &m, NULL));
17279566063dSJacob Faibussowitsch       PetscCall(MatDenseGetArray(B, &bray));
1728917c3dccSPierre Jolivet       PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)B), 1, nrhs * m, nrhsM, (const PetscScalar *)bray, &v_mpi));
17299566063dSJacob Faibussowitsch       PetscCall(MatDenseRestoreArray(B, &bray));
17302b691707SHong Zhang 
1731be818407SHong Zhang       /* scatter v_mpi to b_seq in proc[0]. MUMPS requires rhs to be centralized on the host! */
1732801fbe65SHong Zhang       if (!mumps->myid) {
1733beae5ec0SHong Zhang         PetscInt *idx;
1734beae5ec0SHong Zhang         /* idx: maps from k-th index of v_mpi to (i,j)-th global entry of B */
1735917c3dccSPierre Jolivet         PetscCall(PetscMalloc1(nrhsM, &idx));
17369566063dSJacob Faibussowitsch         PetscCall(MatGetOwnershipRanges(B, &rstart));
1737917c3dccSPierre Jolivet         for (proc = 0, k = 0; proc < mumps->petsc_size; proc++) {
1738be818407SHong Zhang           for (j = 0; j < nrhs; j++) {
1739beae5ec0SHong Zhang             for (i = rstart[proc]; i < rstart[proc + 1]; i++) idx[k++] = j * M + i;
1740be818407SHong Zhang           }
1741be818407SHong Zhang         }
1742be818407SHong Zhang 
1743917c3dccSPierre Jolivet         PetscCall(VecCreateSeq(PETSC_COMM_SELF, nrhsM, &b_seq));
1744917c3dccSPierre Jolivet         PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nrhsM, idx, PETSC_OWN_POINTER, &is_to));
1745917c3dccSPierre Jolivet         PetscCall(ISCreateStride(PETSC_COMM_SELF, nrhsM, 0, 1, &is_from));
1746801fbe65SHong Zhang       } else {
17479566063dSJacob Faibussowitsch         PetscCall(VecCreateSeq(PETSC_COMM_SELF, 0, &b_seq));
17489566063dSJacob Faibussowitsch         PetscCall(ISCreateStride(PETSC_COMM_SELF, 0, 0, 1, &is_to));
17499566063dSJacob Faibussowitsch         PetscCall(ISCreateStride(PETSC_COMM_SELF, 0, 0, 1, &is_from));
1750801fbe65SHong Zhang       }
17519566063dSJacob Faibussowitsch       PetscCall(VecScatterCreate(v_mpi, is_from, b_seq, is_to, &scat_rhs));
17529566063dSJacob Faibussowitsch       PetscCall(VecScatterBegin(scat_rhs, v_mpi, b_seq, INSERT_VALUES, SCATTER_FORWARD));
17539566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_to));
17549566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_from));
17559566063dSJacob Faibussowitsch       PetscCall(VecScatterEnd(scat_rhs, v_mpi, b_seq, INSERT_VALUES, SCATTER_FORWARD));
1756801fbe65SHong Zhang 
1757801fbe65SHong Zhang       if (!mumps->myid) { /* define rhs on the host */
17589566063dSJacob Faibussowitsch         PetscCall(VecGetArray(b_seq, &bray));
1759940cd9d6SSatish Balay         mumps->id.rhs = (MumpsScalar *)bray;
17609566063dSJacob Faibussowitsch         PetscCall(VecRestoreArray(b_seq, &bray));
1761801fbe65SHong Zhang       }
176225aac85cSJunchao Zhang     }
17632b691707SHong Zhang   } else { /* sparse B */
17642b691707SHong Zhang     b = (Mat_MPIAIJ *)Bt->data;
17652b691707SHong Zhang 
1766be818407SHong Zhang     /* wrap dense X into a vector v_mpi */
17679566063dSJacob Faibussowitsch     PetscCall(MatGetLocalSize(X, &m, NULL));
17689566063dSJacob Faibussowitsch     PetscCall(MatDenseGetArray(X, &bray));
1769917c3dccSPierre Jolivet     PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)X), 1, nrhs * m, nrhsM, (const PetscScalar *)bray, &v_mpi));
17709566063dSJacob Faibussowitsch     PetscCall(MatDenseRestoreArray(X, &bray));
17712b691707SHong Zhang 
17722b691707SHong Zhang     if (!mumps->myid) {
17739566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJGetArray(b->A, &aa));
17749566063dSJacob Faibussowitsch       PetscCall(MatGetRowIJ(b->A, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
177528b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
17769566063dSJacob Faibussowitsch       PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs));
17772b691707SHong Zhang       mumps->id.rhs_sparse = (MumpsScalar *)aa;
17782b691707SHong Zhang     } else {
17792b691707SHong Zhang       mumps->id.irhs_ptr    = NULL;
17802b691707SHong Zhang       mumps->id.irhs_sparse = NULL;
17812b691707SHong Zhang       mumps->id.nz_rhs      = 0;
17822b691707SHong Zhang       mumps->id.rhs_sparse  = NULL;
17832b691707SHong Zhang     }
17842b691707SHong Zhang   }
17852b691707SHong Zhang 
1786801fbe65SHong Zhang   /* solve phase */
1787801fbe65SHong Zhang   mumps->id.job = JOB_SOLVE;
17883ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
17899261f6e4SBarry Smith   PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in solve: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
1790801fbe65SHong Zhang 
1791f0b74427SPierre Jolivet   /* scatter mumps distributed solution to PETSc vector v_mpi, which shares local arrays with solution matrix X */
17929566063dSJacob Faibussowitsch   PetscCall(MatDenseGetArray(X, &array));
17939566063dSJacob Faibussowitsch   PetscCall(VecPlaceArray(v_mpi, array));
1794801fbe65SHong Zhang 
1795334c5f61SHong Zhang   /* create scatter scat_sol */
17969566063dSJacob Faibussowitsch   PetscCall(MatGetOwnershipRanges(X, &rstart));
1797f0b74427SPierre Jolivet   /* iidx: index for scatter mumps solution to PETSc X */
1798beae5ec0SHong Zhang 
17999566063dSJacob Faibussowitsch   PetscCall(ISCreateStride(PETSC_COMM_SELF, nlsol_loc, 0, 1, &is_from));
18009566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(nlsol_loc, &idxx));
1801beae5ec0SHong Zhang   for (i = 0; i < lsol_loc; i++) {
1802beae5ec0SHong Zhang     isol_loc[i] -= 1; /* change Fortran style to C style. isol_loc[i+j*lsol_loc] contains x[isol_loc[i]] in j-th vector */
1803beae5ec0SHong Zhang 
18042d4298aeSJunchao Zhang     for (proc = 0; proc < mumps->petsc_size; proc++) {
1805beae5ec0SHong Zhang       if (isol_loc[i] >= rstart[proc] && isol_loc[i] < rstart[proc + 1]) {
1806beae5ec0SHong Zhang         myrstart = rstart[proc];
1807f0b74427SPierre Jolivet         k        = isol_loc[i] - myrstart;          /* local index on 1st column of PETSc vector X */
1808f0b74427SPierre Jolivet         iidx     = k + myrstart * nrhs;             /* maps mumps isol_loc[i] to PETSc index in X */
1809beae5ec0SHong Zhang         m        = rstart[proc + 1] - rstart[proc]; /* rows of X for this proc */
1810beae5ec0SHong Zhang         break;
1811be818407SHong Zhang       }
1812be818407SHong Zhang     }
1813be818407SHong Zhang 
1814beae5ec0SHong Zhang     for (j = 0; j < nrhs; j++) idxx[i + j * lsol_loc] = iidx + j * m;
1815801fbe65SHong Zhang   }
18169566063dSJacob Faibussowitsch   PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nlsol_loc, idxx, PETSC_COPY_VALUES, &is_to));
18179566063dSJacob Faibussowitsch   PetscCall(VecScatterCreate(msol_loc, is_from, v_mpi, is_to, &scat_sol));
18189566063dSJacob Faibussowitsch   PetscCall(VecScatterBegin(scat_sol, msol_loc, v_mpi, INSERT_VALUES, SCATTER_FORWARD));
18199566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&is_from));
18209566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&is_to));
18219566063dSJacob Faibussowitsch   PetscCall(VecScatterEnd(scat_sol, msol_loc, v_mpi, INSERT_VALUES, SCATTER_FORWARD));
18229566063dSJacob Faibussowitsch   PetscCall(MatDenseRestoreArray(X, &array));
182371aed81dSHong Zhang 
182471aed81dSHong Zhang   /* free spaces */
18251683a169SBarry Smith   mumps->id.sol_loc  = (MumpsScalar *)sol_loc_save;
182671aed81dSHong Zhang   mumps->id.isol_loc = isol_loc_save;
182771aed81dSHong Zhang 
18289566063dSJacob Faibussowitsch   PetscCall(PetscFree2(sol_loc, isol_loc));
18299566063dSJacob Faibussowitsch   PetscCall(PetscFree(idxx));
18309566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&msol_loc));
18319566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&v_mpi));
1832a6053eceSJunchao Zhang   if (!denseB) {
18332b691707SHong Zhang     if (!mumps->myid) {
1834d56c302dSHong Zhang       b = (Mat_MPIAIJ *)Bt->data;
18359566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJRestoreArray(b->A, &aa));
18369566063dSJacob Faibussowitsch       PetscCall(MatRestoreRowIJ(b->A, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
183728b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot restore IJ structure");
18382b691707SHong Zhang     }
18392b691707SHong Zhang   } else {
184025aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) {
18419566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b_seq));
18429566063dSJacob Faibussowitsch       PetscCall(VecScatterDestroy(&scat_rhs));
184325aac85cSJunchao Zhang     }
18442b691707SHong Zhang   }
18459566063dSJacob Faibussowitsch   PetscCall(VecScatterDestroy(&scat_sol));
184657508eceSPierre Jolivet   PetscCall(PetscLogFlops(nrhs * PetscMax(0, 2.0 * (mumps->id.INFO(28) >= 0 ? mumps->id.INFO(28) : -1000000 * mumps->id.INFO(28)) - A->cmap->n)));
18473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1848e0b74bf9SHong Zhang }
1849e0b74bf9SHong Zhang 
185066976f2fSJacob Faibussowitsch static PetscErrorCode MatMatSolveTranspose_MUMPS(Mat A, Mat B, Mat X)
1851d71ae5a4SJacob Faibussowitsch {
1852b18964edSHong Zhang   Mat_MUMPS          *mumps = (Mat_MUMPS *)A->data;
1853338d3105SPierre Jolivet   const PetscMUMPSInt value = mumps->id.ICNTL(9);
1854b18964edSHong Zhang 
1855b18964edSHong Zhang   PetscFunctionBegin;
1856b18964edSHong Zhang   mumps->id.ICNTL(9) = 0;
1857b18964edSHong Zhang   PetscCall(MatMatSolve_MUMPS(A, B, X));
1858338d3105SPierre Jolivet   mumps->id.ICNTL(9) = value;
18593ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1860b18964edSHong Zhang }
1861b18964edSHong Zhang 
186266976f2fSJacob Faibussowitsch static PetscErrorCode MatMatTransposeSolve_MUMPS(Mat A, Mat Bt, Mat X)
1863d71ae5a4SJacob Faibussowitsch {
1864eb3ef3b2SHong Zhang   PetscBool flg;
1865eb3ef3b2SHong Zhang   Mat       B;
1866eb3ef3b2SHong Zhang 
1867eb3ef3b2SHong Zhang   PetscFunctionBegin;
18689566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)Bt, &flg, MATSEQAIJ, MATMPIAIJ, NULL));
186928b400f6SJacob Faibussowitsch   PetscCheck(flg, PetscObjectComm((PetscObject)Bt), PETSC_ERR_ARG_WRONG, "Matrix Bt must be MATAIJ matrix");
1870eb3ef3b2SHong Zhang 
1871eb3ef3b2SHong Zhang   /* Create B=Bt^T that uses Bt's data structure */
18729566063dSJacob Faibussowitsch   PetscCall(MatCreateTranspose(Bt, &B));
1873eb3ef3b2SHong Zhang 
18749566063dSJacob Faibussowitsch   PetscCall(MatMatSolve_MUMPS(A, B, X));
18759566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&B));
18763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1877eb3ef3b2SHong Zhang }
1878eb3ef3b2SHong Zhang 
1879ace3df97SHong Zhang #if !defined(PETSC_USE_COMPLEX)
1880a58c3f20SHong Zhang /*
1881a58c3f20SHong Zhang   input:
1882a58c3f20SHong Zhang    F:        numeric factor
1883a58c3f20SHong Zhang   output:
1884a58c3f20SHong Zhang    nneg:     total number of negative pivots
188519d49a3bSHong Zhang    nzero:    total number of zero pivots
188619d49a3bSHong Zhang    npos:     (global dimension of F) - nneg - nzero
1887a58c3f20SHong Zhang */
188866976f2fSJacob Faibussowitsch static PetscErrorCode MatGetInertia_SBAIJMUMPS(Mat F, PetscInt *nneg, PetscInt *nzero, PetscInt *npos)
1889d71ae5a4SJacob Faibussowitsch {
1890e69c285eSBarry Smith   Mat_MUMPS  *mumps = (Mat_MUMPS *)F->data;
1891c1490034SHong Zhang   PetscMPIInt size;
1892a58c3f20SHong Zhang 
1893a58c3f20SHong Zhang   PetscFunctionBegin;
18949566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)F), &size));
1895bcb30aebSHong Zhang   /* MUMPS 4.3.1 calls ScaLAPACK when ICNTL(13)=0 (default), which does not offer the possibility to compute the inertia of a dense matrix. Set ICNTL(13)=1 to skip ScaLAPACK */
1896aed4548fSBarry Smith   PetscCheck(size <= 1 || mumps->id.ICNTL(13) == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "ICNTL(13)=%d. -mat_mumps_icntl_13 must be set as 1 for correct global matrix inertia", mumps->id.INFOG(13));
1897ed85ac9fSHong Zhang 
1898710ac8efSHong Zhang   if (nneg) *nneg = mumps->id.INFOG(12);
1899ed85ac9fSHong Zhang   if (nzero || npos) {
190008401ef6SPierre Jolivet     PetscCheck(mumps->id.ICNTL(24) == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "-mat_mumps_icntl_24 must be set as 1 for null pivot row detection");
1901710ac8efSHong Zhang     if (nzero) *nzero = mumps->id.INFOG(28);
1902710ac8efSHong Zhang     if (npos) *npos = F->rmap->N - (mumps->id.INFOG(12) + mumps->id.INFOG(28));
1903a58c3f20SHong Zhang   }
19043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1905a58c3f20SHong Zhang }
190619d49a3bSHong Zhang #endif
1907a58c3f20SHong Zhang 
190866976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGatherNonzerosOnMaster(MatReuse reuse, Mat_MUMPS *mumps)
1909d71ae5a4SJacob Faibussowitsch {
19106497c311SBarry Smith   PetscMPIInt    nreqs;
1911a6053eceSJunchao Zhang   PetscMUMPSInt *irn, *jcn;
1912a6053eceSJunchao Zhang   PetscMPIInt    count;
19136497c311SBarry Smith   PetscCount     totnnz, remain;
1914a6053eceSJunchao Zhang   const PetscInt osize = mumps->omp_comm_size;
1915a6053eceSJunchao Zhang   PetscScalar   *val;
19163ab56b82SJunchao Zhang 
19173ab56b82SJunchao Zhang   PetscFunctionBegin;
1918a6053eceSJunchao Zhang   if (osize > 1) {
19193ab56b82SJunchao Zhang     if (reuse == MAT_INITIAL_MATRIX) {
19203ab56b82SJunchao Zhang       /* master first gathers counts of nonzeros to receive */
19219566063dSJacob Faibussowitsch       if (mumps->is_omp_master) PetscCall(PetscMalloc1(osize, &mumps->recvcount));
19229566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Gather(&mumps->nnz, 1, MPIU_INT64, mumps->recvcount, 1, MPIU_INT64, 0 /*master*/, mumps->omp_comm));
19233ab56b82SJunchao Zhang 
1924a6053eceSJunchao Zhang       /* Then each computes number of send/recvs */
19253ab56b82SJunchao Zhang       if (mumps->is_omp_master) {
1926a6053eceSJunchao Zhang         /* Start from 1 since self communication is not done in MPI */
1927a6053eceSJunchao Zhang         nreqs = 0;
19286497c311SBarry Smith         for (PetscMPIInt i = 1; i < osize; i++) nreqs += (mumps->recvcount[i] + PETSC_MPI_INT_MAX - 1) / PETSC_MPI_INT_MAX;
1929a6053eceSJunchao Zhang       } else {
19306497c311SBarry Smith         nreqs = (PetscMPIInt)(((mumps->nnz + PETSC_MPI_INT_MAX - 1) / PETSC_MPI_INT_MAX));
19313ab56b82SJunchao Zhang       }
193235cb6cd3SPierre Jolivet       PetscCall(PetscMalloc1(nreqs * 3, &mumps->reqs)); /* Triple the requests since we send irn, jcn and val separately */
19333ab56b82SJunchao Zhang 
1934a6053eceSJunchao Zhang       /* The following code is doing a very simple thing: omp_master rank gathers irn/jcn/val from others.
1935a6053eceSJunchao Zhang          MPI_Gatherv would be enough if it supports big counts > 2^31-1. Since it does not, and mumps->nnz
1936a6053eceSJunchao Zhang          might be a prime number > 2^31-1, we have to slice the message. Note omp_comm_size
1937a6053eceSJunchao Zhang          is very small, the current approach should have no extra overhead compared to MPI_Gatherv.
1938a6053eceSJunchao Zhang        */
1939a6053eceSJunchao Zhang       nreqs = 0; /* counter for actual send/recvs */
19403ab56b82SJunchao Zhang       if (mumps->is_omp_master) {
19416497c311SBarry Smith         totnnz = 0;
19426497c311SBarry Smith 
19436497c311SBarry Smith         for (PetscMPIInt i = 0; i < osize; i++) totnnz += mumps->recvcount[i]; /* totnnz = sum of nnz over omp_comm */
19449566063dSJacob Faibussowitsch         PetscCall(PetscMalloc2(totnnz, &irn, totnnz, &jcn));
19459566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(totnnz, &val));
1946a6053eceSJunchao Zhang 
1947a6053eceSJunchao Zhang         /* Self communication */
19489566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(irn, mumps->irn, mumps->nnz));
19499566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(jcn, mumps->jcn, mumps->nnz));
19509566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(val, mumps->val, mumps->nnz));
1951a6053eceSJunchao Zhang 
1952a6053eceSJunchao Zhang         /* Replace mumps->irn/jcn etc on master with the newly allocated bigger arrays */
19539566063dSJacob Faibussowitsch         PetscCall(PetscFree2(mumps->irn, mumps->jcn));
19549566063dSJacob Faibussowitsch         PetscCall(PetscFree(mumps->val_alloc));
1955a6053eceSJunchao Zhang         mumps->nnz = totnnz;
19563ab56b82SJunchao Zhang         mumps->irn = irn;
19573ab56b82SJunchao Zhang         mumps->jcn = jcn;
1958a6053eceSJunchao Zhang         mumps->val = mumps->val_alloc = val;
1959a6053eceSJunchao Zhang 
1960a6053eceSJunchao Zhang         irn += mumps->recvcount[0]; /* recvcount[0] is old mumps->nnz on omp rank 0 */
1961a6053eceSJunchao Zhang         jcn += mumps->recvcount[0];
1962a6053eceSJunchao Zhang         val += mumps->recvcount[0];
1963a6053eceSJunchao Zhang 
1964a6053eceSJunchao Zhang         /* Remote communication */
19656497c311SBarry Smith         for (PetscMPIInt i = 1; i < osize; i++) {
19666497c311SBarry Smith           count  = (PetscMPIInt)PetscMin(mumps->recvcount[i], (PetscMPIInt)PETSC_MPI_INT_MAX);
1967a6053eceSJunchao Zhang           remain = mumps->recvcount[i] - count;
1968a6053eceSJunchao Zhang           while (count > 0) {
19696497c311SBarry Smith             PetscCallMPI(MPIU_Irecv(irn, count, MPIU_MUMPSINT, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19706497c311SBarry Smith             PetscCallMPI(MPIU_Irecv(jcn, count, MPIU_MUMPSINT, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19716497c311SBarry Smith             PetscCallMPI(MPIU_Irecv(val, count, MPIU_SCALAR, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1972a6053eceSJunchao Zhang             irn += count;
1973a6053eceSJunchao Zhang             jcn += count;
1974a6053eceSJunchao Zhang             val += count;
19756497c311SBarry Smith             count = (PetscMPIInt)PetscMin(remain, (PetscMPIInt)PETSC_MPI_INT_MAX);
1976a6053eceSJunchao Zhang             remain -= count;
1977a6053eceSJunchao Zhang           }
19783ab56b82SJunchao Zhang         }
19793ab56b82SJunchao Zhang       } else {
1980a6053eceSJunchao Zhang         irn    = mumps->irn;
1981a6053eceSJunchao Zhang         jcn    = mumps->jcn;
1982a6053eceSJunchao Zhang         val    = mumps->val;
19836497c311SBarry Smith         count  = (PetscMPIInt)PetscMin(mumps->nnz, (PetscMPIInt)PETSC_MPI_INT_MAX);
1984a6053eceSJunchao Zhang         remain = mumps->nnz - count;
1985a6053eceSJunchao Zhang         while (count > 0) {
19866497c311SBarry Smith           PetscCallMPI(MPIU_Isend(irn, count, MPIU_MUMPSINT, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19876497c311SBarry Smith           PetscCallMPI(MPIU_Isend(jcn, count, MPIU_MUMPSINT, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19886497c311SBarry Smith           PetscCallMPI(MPIU_Isend(val, count, MPIU_SCALAR, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1989a6053eceSJunchao Zhang           irn += count;
1990a6053eceSJunchao Zhang           jcn += count;
1991a6053eceSJunchao Zhang           val += count;
19926497c311SBarry Smith           count = (PetscMPIInt)PetscMin(remain, (PetscMPIInt)PETSC_MPI_INT_MAX);
1993a6053eceSJunchao Zhang           remain -= count;
19943ab56b82SJunchao Zhang         }
19953ab56b82SJunchao Zhang       }
1996a6053eceSJunchao Zhang     } else {
1997a6053eceSJunchao Zhang       nreqs = 0;
1998a6053eceSJunchao Zhang       if (mumps->is_omp_master) {
1999a6053eceSJunchao Zhang         val = mumps->val + mumps->recvcount[0];
20006497c311SBarry Smith         for (PetscMPIInt i = 1; i < osize; i++) { /* Remote communication only since self data is already in place */
20016497c311SBarry Smith           count  = (PetscMPIInt)PetscMin(mumps->recvcount[i], (PetscMPIInt)PETSC_MPI_INT_MAX);
2002a6053eceSJunchao Zhang           remain = mumps->recvcount[i] - count;
2003a6053eceSJunchao Zhang           while (count > 0) {
20046497c311SBarry Smith             PetscCallMPI(MPIU_Irecv(val, count, MPIU_SCALAR, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
2005a6053eceSJunchao Zhang             val += count;
20066497c311SBarry Smith             count = (PetscMPIInt)PetscMin(remain, (PetscMPIInt)PETSC_MPI_INT_MAX);
2007a6053eceSJunchao Zhang             remain -= count;
2008a6053eceSJunchao Zhang           }
2009a6053eceSJunchao Zhang         }
2010a6053eceSJunchao Zhang       } else {
2011a6053eceSJunchao Zhang         val    = mumps->val;
20126497c311SBarry Smith         count  = (PetscMPIInt)PetscMin(mumps->nnz, (PetscMPIInt)PETSC_MPI_INT_MAX);
2013a6053eceSJunchao Zhang         remain = mumps->nnz - count;
2014a6053eceSJunchao Zhang         while (count > 0) {
20156497c311SBarry Smith           PetscCallMPI(MPIU_Isend(val, count, MPIU_SCALAR, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
2016a6053eceSJunchao Zhang           val += count;
20176497c311SBarry Smith           count = (PetscMPIInt)PetscMin(remain, (PetscMPIInt)PETSC_MPI_INT_MAX);
2018a6053eceSJunchao Zhang           remain -= count;
2019a6053eceSJunchao Zhang         }
2020a6053eceSJunchao Zhang       }
2021a6053eceSJunchao Zhang     }
20229566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Waitall(nreqs, mumps->reqs, MPI_STATUSES_IGNORE));
2023a6053eceSJunchao Zhang     mumps->tag++; /* It is totally fine for above send/recvs to share one mpi tag */
2024a6053eceSJunchao Zhang   }
20253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
20263ab56b82SJunchao Zhang }
20273ab56b82SJunchao Zhang 
2028d2a308c1SPierre Jolivet static PetscErrorCode MatFactorNumeric_MUMPS(Mat F, Mat A, PETSC_UNUSED const MatFactorInfo *info)
2029d71ae5a4SJacob Faibussowitsch {
203057508eceSPierre Jolivet   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2031ace3abfcSBarry Smith   PetscBool  isMPIAIJ;
2032397b6df1SKris Buschelman 
2033397b6df1SKris Buschelman   PetscFunctionBegin;
2034dbf6bb8dSprj-   if (mumps->id.INFOG(1) < 0 && !(mumps->id.INFOG(1) == -16 && mumps->id.INFOG(1) == 0)) {
203548a46eb9SPierre Jolivet     if (mumps->id.INFOG(1) == -6) PetscCall(PetscInfo(A, "MatFactorNumeric is called with singular matrix structure, INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
20369566063dSJacob Faibussowitsch     PetscCall(PetscInfo(A, "MatFactorNumeric is called after analysis phase fails, INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
20373ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
20382aca8efcSHong Zhang   }
20396baea169SHong Zhang 
20409566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_REUSE_MATRIX, mumps));
20419566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_REUSE_MATRIX, mumps));
2042397b6df1SKris Buschelman 
2043397b6df1SKris Buschelman   /* numerical factorization phase */
2044a5e57a09SHong Zhang   mumps->id.job = JOB_FACTNUMERIC;
20454e34a73bSHong Zhang   if (!mumps->id.ICNTL(18)) { /* A is centralized */
2046ad540459SPierre Jolivet     if (!mumps->myid) mumps->id.a = (MumpsScalar *)mumps->val;
2047397b6df1SKris Buschelman   } else {
2048940cd9d6SSatish Balay     mumps->id.a_loc = (MumpsScalar *)mumps->val;
2049397b6df1SKris Buschelman   }
20503ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
2051a5e57a09SHong Zhang   if (mumps->id.INFOG(1) < 0) {
20529261f6e4SBarry Smith     PetscCheck(!A->erroriffailure, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in numerical factorization: INFOG(1)=%d, INFO(2)=%d " MUMPS_MANUALS, mumps->id.INFOG(1), mumps->id.INFO(2));
20539261f6e4SBarry Smith     if (mumps->id.INFOG(1) == -10) {
20549261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in numerical factorization: matrix is numerically singular, INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
2055603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
2056c0d63f2fSHong Zhang     } else if (mumps->id.INFOG(1) == -13) {
20579261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in numerical factorization: INFOG(1)=%d, cannot allocate required memory %d megabytes\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
2058603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OUTMEMORY;
2059c0d63f2fSHong Zhang     } else if (mumps->id.INFOG(1) == -8 || mumps->id.INFOG(1) == -9 || (-16 < mumps->id.INFOG(1) && mumps->id.INFOG(1) < -10)) {
2060bdcd51b8SPierre Jolivet       PetscCall(PetscInfo(F, "MUMPS error in numerical factorization: INFOG(1)=%d, INFO(2)=%d, problem with work array\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
2061603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OUTMEMORY;
20622aca8efcSHong Zhang     } else {
20639261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in numerical factorization: INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
2064603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OTHER;
2065151787a6SHong Zhang     }
20662aca8efcSHong Zhang   }
20679261f6e4SBarry Smith   PetscCheck(mumps->myid || mumps->id.ICNTL(16) <= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in numerical factorization: ICNTL(16)=%d " MUMPS_MANUALS, mumps->id.INFOG(16));
2068397b6df1SKris Buschelman 
2069b3cb21ddSStefano Zampini   F->assembled = PETSC_TRUE;
2070d47f36abSHong Zhang 
2071b3cb21ddSStefano Zampini   if (F->schur) { /* reset Schur status to unfactored */
20723cb7dd0eSStefano Zampini #if defined(PETSC_HAVE_CUDA)
2073c70f7ee4SJunchao Zhang     F->schur->offloadmask = PETSC_OFFLOAD_CPU;
20743cb7dd0eSStefano Zampini #endif
2075b3cb21ddSStefano Zampini     if (mumps->id.ICNTL(19) == 1) { /* stored by rows */
2076b3cb21ddSStefano Zampini       mumps->id.ICNTL(19) = 2;
20779566063dSJacob Faibussowitsch       PetscCall(MatTranspose(F->schur, MAT_INPLACE_MATRIX, &F->schur));
2078b3cb21ddSStefano Zampini     }
20799566063dSJacob Faibussowitsch     PetscCall(MatFactorRestoreSchurComplement(F, NULL, MAT_FACTOR_SCHUR_UNFACTORED));
2080b3cb21ddSStefano Zampini   }
208167877ebaSShri Abhyankar 
2082066565c5SStefano Zampini   /* just to be sure that ICNTL(19) value returned by a call from MatMumpsGetIcntl is always consistent */
2083066565c5SStefano Zampini   if (!mumps->sym && mumps->id.ICNTL(19) && mumps->id.ICNTL(19) != 1) mumps->id.ICNTL(19) = 3;
2084066565c5SStefano Zampini 
20853ab56b82SJunchao Zhang   if (!mumps->is_omp_master) mumps->id.INFO(23) = 0;
20862d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) {
208767877ebaSShri Abhyankar     PetscInt     lsol_loc;
208867877ebaSShri Abhyankar     PetscScalar *sol_loc;
20892205254eSKarl Rupp 
20909566063dSJacob Faibussowitsch     PetscCall(PetscObjectTypeCompare((PetscObject)A, MATMPIAIJ, &isMPIAIJ));
2091c2093ab7SHong Zhang 
2092c2093ab7SHong Zhang     /* distributed solution; Create x_seq=sol_loc for repeated use */
2093c2093ab7SHong Zhang     if (mumps->x_seq) {
20949566063dSJacob Faibussowitsch       PetscCall(VecScatterDestroy(&mumps->scat_sol));
20959566063dSJacob Faibussowitsch       PetscCall(PetscFree2(mumps->id.sol_loc, mumps->id.isol_loc));
20969566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mumps->x_seq));
2097c2093ab7SHong Zhang     }
2098a5e57a09SHong Zhang     lsol_loc = mumps->id.INFO(23); /* length of sol_loc */
20999566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(lsol_loc, &sol_loc, lsol_loc, &mumps->id.isol_loc));
21006497c311SBarry Smith     mumps->id.lsol_loc = (PetscMUMPSInt)lsol_loc;
2101940cd9d6SSatish Balay     mumps->id.sol_loc  = (MumpsScalar *)sol_loc;
21029566063dSJacob Faibussowitsch     PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, lsol_loc, sol_loc, &mumps->x_seq));
210367877ebaSShri Abhyankar   }
2104d2a308c1SPierre Jolivet   PetscCall(PetscLogFlops((double)mumps->id.RINFO(2)));
21053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2106397b6df1SKris Buschelman }
2107397b6df1SKris Buschelman 
21089a2535b5SHong Zhang /* Sets MUMPS options from the options database */
210966976f2fSJacob Faibussowitsch static PetscErrorCode MatSetFromOptions_MUMPS(Mat F, Mat A)
2110d71ae5a4SJacob Faibussowitsch {
2111e69c285eSBarry Smith   Mat_MUMPS    *mumps = (Mat_MUMPS *)F->data;
2112413bcc21SPierre Jolivet   PetscMUMPSInt icntl = 0, size, *listvar_schur;
211345e3843bSPierre Jolivet   PetscInt      info[80], i, ninfo = 80, rbs, cbs;
2114413bcc21SPierre Jolivet   PetscBool     flg = PETSC_FALSE, schur = (PetscBool)(mumps->id.ICNTL(26) == -1);
2115413bcc21SPierre Jolivet   MumpsScalar  *arr;
2116dcd589f8SShri Abhyankar 
2117dcd589f8SShri Abhyankar   PetscFunctionBegin;
211826cc229bSBarry Smith   PetscOptionsBegin(PetscObjectComm((PetscObject)F), ((PetscObject)F)->prefix, "MUMPS Options", "Mat");
2119413bcc21SPierre Jolivet   if (mumps->id.job == JOB_NULL) { /* MatSetFromOptions_MUMPS() has never been called before */
2120413bcc21SPierre Jolivet     PetscInt nthreads   = 0;
2121413bcc21SPierre Jolivet     PetscInt nCNTL_pre  = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0;
2122413bcc21SPierre Jolivet     PetscInt nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0;
2123413bcc21SPierre Jolivet 
2124413bcc21SPierre Jolivet     mumps->petsc_comm = PetscObjectComm((PetscObject)A);
2125413bcc21SPierre Jolivet     PetscCallMPI(MPI_Comm_size(mumps->petsc_comm, &mumps->petsc_size));
2126413bcc21SPierre Jolivet     PetscCallMPI(MPI_Comm_rank(mumps->petsc_comm, &mumps->myid)); /* "if (!myid)" still works even if mumps_comm is different */
2127413bcc21SPierre Jolivet 
2128413bcc21SPierre Jolivet     PetscCall(PetscOptionsName("-mat_mumps_use_omp_threads", "Convert MPI processes into OpenMP threads", "None", &mumps->use_petsc_omp_support));
2129413bcc21SPierre Jolivet     if (mumps->use_petsc_omp_support) nthreads = -1; /* -1 will let PetscOmpCtrlCreate() guess a proper value when user did not supply one */
2130413bcc21SPierre Jolivet     /* do not use PetscOptionsInt() so that the option -mat_mumps_use_omp_threads is not displayed twice in the help */
2131413bcc21SPierre Jolivet     PetscCall(PetscOptionsGetInt(NULL, ((PetscObject)F)->prefix, "-mat_mumps_use_omp_threads", &nthreads, NULL));
2132413bcc21SPierre Jolivet     if (mumps->use_petsc_omp_support) {
2133413bcc21SPierre Jolivet       PetscCheck(!schur, PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot use -%smat_mumps_use_omp_threads with the Schur complement feature", ((PetscObject)F)->prefix ? ((PetscObject)F)->prefix : "");
2134413bcc21SPierre Jolivet #if defined(PETSC_HAVE_OPENMP_SUPPORT)
2135413bcc21SPierre Jolivet       PetscCall(PetscOmpCtrlCreate(mumps->petsc_comm, nthreads, &mumps->omp_ctrl));
2136413bcc21SPierre Jolivet       PetscCall(PetscOmpCtrlGetOmpComms(mumps->omp_ctrl, &mumps->omp_comm, &mumps->mumps_comm, &mumps->is_omp_master));
2137ea17275aSJose E. Roman #else
2138ea17275aSJose E. Roman       SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP_SYS, "The system does not have PETSc OpenMP support but you added the -%smat_mumps_use_omp_threads option. Configure PETSc with --with-openmp --download-hwloc (or --with-hwloc) to enable it, see more in MATSOLVERMUMPS manual",
2139ea17275aSJose E. Roman               ((PetscObject)F)->prefix ? ((PetscObject)F)->prefix : "");
2140413bcc21SPierre Jolivet #endif
2141413bcc21SPierre Jolivet     } else {
2142413bcc21SPierre Jolivet       mumps->omp_comm      = PETSC_COMM_SELF;
2143413bcc21SPierre Jolivet       mumps->mumps_comm    = mumps->petsc_comm;
2144413bcc21SPierre Jolivet       mumps->is_omp_master = PETSC_TRUE;
2145413bcc21SPierre Jolivet     }
2146413bcc21SPierre Jolivet     PetscCallMPI(MPI_Comm_size(mumps->omp_comm, &mumps->omp_comm_size));
2147413bcc21SPierre Jolivet     mumps->reqs = NULL;
2148413bcc21SPierre Jolivet     mumps->tag  = 0;
2149413bcc21SPierre Jolivet 
2150413bcc21SPierre Jolivet     if (mumps->mumps_comm != MPI_COMM_NULL) {
2151413bcc21SPierre Jolivet       if (PetscDefined(HAVE_OPENMP_SUPPORT) && mumps->use_petsc_omp_support) {
2152413bcc21SPierre Jolivet         /* It looks like MUMPS does not dup the input comm. Dup a new comm for MUMPS to avoid any tag mismatches. */
2153413bcc21SPierre Jolivet         MPI_Comm comm;
2154413bcc21SPierre Jolivet         PetscCallMPI(MPI_Comm_dup(mumps->mumps_comm, &comm));
2155413bcc21SPierre Jolivet         mumps->mumps_comm = comm;
2156413bcc21SPierre Jolivet       } else PetscCall(PetscCommGetComm(mumps->petsc_comm, &mumps->mumps_comm));
2157413bcc21SPierre Jolivet     }
2158413bcc21SPierre Jolivet 
2159413bcc21SPierre Jolivet     mumps->id.comm_fortran = MPI_Comm_c2f(mumps->mumps_comm);
2160413bcc21SPierre Jolivet     mumps->id.job          = JOB_INIT;
2161413bcc21SPierre Jolivet     mumps->id.par          = 1; /* host participates factorizaton and solve */
2162413bcc21SPierre Jolivet     mumps->id.sym          = mumps->sym;
2163413bcc21SPierre Jolivet 
2164413bcc21SPierre Jolivet     size          = mumps->id.size_schur;
2165413bcc21SPierre Jolivet     arr           = mumps->id.schur;
2166413bcc21SPierre Jolivet     listvar_schur = mumps->id.listvar_schur;
2167413bcc21SPierre Jolivet     PetscMUMPS_c(mumps);
21689261f6e4SBarry Smith     PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
216951ad14ebSPierre Jolivet 
217051ad14ebSPierre Jolivet     /* set PETSc-MUMPS default options - override MUMPS default */
217151ad14ebSPierre Jolivet     mumps->id.ICNTL(3) = 0;
217251ad14ebSPierre Jolivet     mumps->id.ICNTL(4) = 0;
217351ad14ebSPierre Jolivet     if (mumps->petsc_size == 1) {
217451ad14ebSPierre Jolivet       mumps->id.ICNTL(18) = 0; /* centralized assembled matrix input */
217551ad14ebSPierre Jolivet       mumps->id.ICNTL(7)  = 7; /* automatic choice of ordering done by the package */
217651ad14ebSPierre Jolivet     } else {
217751ad14ebSPierre Jolivet       mumps->id.ICNTL(18) = 3; /* distributed assembled matrix input */
217851ad14ebSPierre Jolivet       mumps->id.ICNTL(21) = 1; /* distributed solution */
217951ad14ebSPierre Jolivet     }
218051ad14ebSPierre Jolivet 
2181413bcc21SPierre Jolivet     /* restore cached ICNTL and CNTL values */
2182413bcc21SPierre Jolivet     for (icntl = 0; icntl < nICNTL_pre; ++icntl) mumps->id.ICNTL(mumps->ICNTL_pre[1 + 2 * icntl]) = mumps->ICNTL_pre[2 + 2 * icntl];
2183413bcc21SPierre Jolivet     for (icntl = 0; icntl < nCNTL_pre; ++icntl) mumps->id.CNTL((PetscInt)mumps->CNTL_pre[1 + 2 * icntl]) = mumps->CNTL_pre[2 + 2 * icntl];
2184413bcc21SPierre Jolivet     PetscCall(PetscFree(mumps->ICNTL_pre));
2185413bcc21SPierre Jolivet     PetscCall(PetscFree(mumps->CNTL_pre));
2186413bcc21SPierre Jolivet 
2187413bcc21SPierre Jolivet     if (schur) {
2188413bcc21SPierre Jolivet       mumps->id.size_schur    = size;
2189413bcc21SPierre Jolivet       mumps->id.schur_lld     = size;
2190413bcc21SPierre Jolivet       mumps->id.schur         = arr;
2191413bcc21SPierre Jolivet       mumps->id.listvar_schur = listvar_schur;
2192413bcc21SPierre Jolivet       if (mumps->petsc_size > 1) {
2193413bcc21SPierre Jolivet         PetscBool gs; /* gs is false if any rank other than root has non-empty IS */
2194413bcc21SPierre Jolivet 
2195413bcc21SPierre Jolivet         mumps->id.ICNTL(19) = 1;                                                                            /* MUMPS returns Schur centralized on the host */
2196413bcc21SPierre Jolivet         gs                  = mumps->myid ? (mumps->id.size_schur ? PETSC_FALSE : PETSC_TRUE) : PETSC_TRUE; /* always true on root; false on others if their size != 0 */
2197462c564dSBarry Smith         PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &gs, 1, MPIU_BOOL, MPI_LAND, mumps->petsc_comm));
2198413bcc21SPierre Jolivet         PetscCheck(gs, PETSC_COMM_SELF, PETSC_ERR_SUP, "MUMPS distributed parallel Schur complements not yet supported from PETSc");
2199413bcc21SPierre Jolivet       } else {
2200413bcc21SPierre Jolivet         if (F->factortype == MAT_FACTOR_LU) {
2201413bcc21SPierre Jolivet           mumps->id.ICNTL(19) = 3; /* MUMPS returns full matrix */
2202413bcc21SPierre Jolivet         } else {
2203413bcc21SPierre Jolivet           mumps->id.ICNTL(19) = 2; /* MUMPS returns lower triangular part */
2204413bcc21SPierre Jolivet         }
2205413bcc21SPierre Jolivet       }
2206413bcc21SPierre Jolivet       mumps->id.ICNTL(26) = -1;
2207413bcc21SPierre Jolivet     }
2208413bcc21SPierre Jolivet 
2209413bcc21SPierre Jolivet     /* copy MUMPS default control values from master to slaves. Although slaves do not call MUMPS, they may access these values in code.
2210413bcc21SPierre Jolivet        For example, ICNTL(9) is initialized to 1 by MUMPS and slaves check ICNTL(9) in MatSolve_MUMPS.
2211413bcc21SPierre Jolivet      */
2212413bcc21SPierre Jolivet     PetscCallMPI(MPI_Bcast(mumps->id.icntl, 40, MPI_INT, 0, mumps->omp_comm));
2213413bcc21SPierre Jolivet     PetscCallMPI(MPI_Bcast(mumps->id.cntl, 15, MPIU_REAL, 0, mumps->omp_comm));
2214413bcc21SPierre Jolivet 
2215413bcc21SPierre Jolivet     mumps->scat_rhs = NULL;
2216413bcc21SPierre Jolivet     mumps->scat_sol = NULL;
2217413bcc21SPierre Jolivet   }
22189566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_1", "ICNTL(1): output stream for error messages", "None", mumps->id.ICNTL(1), &icntl, &flg));
22199a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(1) = icntl;
22209566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_2", "ICNTL(2): output stream for diagnostic printing, statistics, and warning", "None", mumps->id.ICNTL(2), &icntl, &flg));
22219a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(2) = icntl;
22229566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_3", "ICNTL(3): output stream for global information, collected on the host", "None", mumps->id.ICNTL(3), &icntl, &flg));
22239a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(3) = icntl;
2224dcd589f8SShri Abhyankar 
22259566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_4", "ICNTL(4): level of printing (0 to 4)", "None", mumps->id.ICNTL(4), &icntl, &flg));
22269a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(4) = icntl;
22279a2535b5SHong Zhang   if (mumps->id.ICNTL(4) || PetscLogPrintInfo) mumps->id.ICNTL(3) = 6; /* resume MUMPS default id.ICNTL(3) = 6 */
22289a2535b5SHong Zhang 
22299566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_6", "ICNTL(6): permutes to a zero-free diagonal and/or scale the matrix (0 to 7)", "None", mumps->id.ICNTL(6), &icntl, &flg));
22309a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(6) = icntl;
22319a2535b5SHong Zhang 
22329566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_7", "ICNTL(7): computes a symmetric permutation in sequential analysis. 0=AMD, 2=AMF, 3=Scotch, 4=PORD, 5=Metis, 6=QAMD, and 7=auto(default)", "None", mumps->id.ICNTL(7), &icntl, &flg));
2233dcd589f8SShri Abhyankar   if (flg) {
2234aed4548fSBarry Smith     PetscCheck(icntl != 1 && icntl >= 0 && icntl <= 7, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Valid values are 0=AMD, 2=AMF, 3=Scotch, 4=PORD, 5=Metis, 6=QAMD, and 7=auto");
2235b53c1a7fSBarry Smith     mumps->id.ICNTL(7) = icntl;
2236dcd589f8SShri Abhyankar   }
2237e0b74bf9SHong Zhang 
22389566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_8", "ICNTL(8): scaling strategy (-2 to 8 or 77)", "None", mumps->id.ICNTL(8), &mumps->id.ICNTL(8), NULL));
22399566063dSJacob Faibussowitsch   /* PetscCall(PetscOptionsInt("-mat_mumps_icntl_9","ICNTL(9): computes the solution using A or A^T","None",mumps->id.ICNTL(9),&mumps->id.ICNTL(9),NULL)); handled by MatSolveTranspose_MUMPS() */
22409566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_10", "ICNTL(10): max num of refinements", "None", mumps->id.ICNTL(10), &mumps->id.ICNTL(10), NULL));
22419566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_11", "ICNTL(11): statistics related to an error analysis (via -ksp_view)", "None", mumps->id.ICNTL(11), &mumps->id.ICNTL(11), NULL));
22429566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_12", "ICNTL(12): an ordering strategy for symmetric matrices (0 to 3)", "None", mumps->id.ICNTL(12), &mumps->id.ICNTL(12), NULL));
22439566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_13", "ICNTL(13): parallelism of the root node (enable ScaLAPACK) and its splitting", "None", mumps->id.ICNTL(13), &mumps->id.ICNTL(13), NULL));
22449566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_14", "ICNTL(14): percentage increase in the estimated working space", "None", mumps->id.ICNTL(14), &mumps->id.ICNTL(14), NULL));
224545e3843bSPierre Jolivet   PetscCall(MatGetBlockSizes(A, &rbs, &cbs));
22466497c311SBarry Smith   if (rbs == cbs && rbs > 1) mumps->id.ICNTL(15) = (PetscMUMPSInt)-rbs;
224745e3843bSPierre Jolivet   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_15", "ICNTL(15): compression of the input matrix resulting from a block format", "None", mumps->id.ICNTL(15), &mumps->id.ICNTL(15), &flg));
224845e3843bSPierre Jolivet   if (flg) {
224945e3843bSPierre Jolivet     PetscCheck(mumps->id.ICNTL(15) <= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Positive -mat_mumps_icntl_15 not handled");
225045e3843bSPierre Jolivet     PetscCheck((-mumps->id.ICNTL(15) % cbs == 0) && (-mumps->id.ICNTL(15) % rbs == 0), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The opposite of -mat_mumps_icntl_15 must be a multiple of the column and row blocksizes");
225145e3843bSPierre Jolivet   }
22529566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_19", "ICNTL(19): computes the Schur complement", "None", mumps->id.ICNTL(19), &mumps->id.ICNTL(19), NULL));
225359ac8732SStefano Zampini   if (mumps->id.ICNTL(19) <= 0 || mumps->id.ICNTL(19) > 3) { /* reset any schur data (if any) */
22549566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&F->schur));
22559566063dSJacob Faibussowitsch     PetscCall(MatMumpsResetSchur_Private(mumps));
225659ac8732SStefano Zampini   }
225725aac85cSJunchao Zhang 
225843f3b051SJunchao Zhang   /* Two MPICH Fortran MPI_IN_PLACE binding bugs prevented the use of 'mpich + mumps'. One happened with "mpi4py + mpich + mumps",
225943f3b051SJunchao Zhang      and was reported by Firedrake. See https://bitbucket.org/mpi4py/mpi4py/issues/162/mpi4py-initialization-breaks-fortran
226025aac85cSJunchao Zhang      and a petsc-maint mailing list thread with subject 'MUMPS segfaults in parallel because of ...'
226143f3b051SJunchao Zhang      This bug was fixed by https://github.com/pmodels/mpich/pull/4149. But the fix brought a new bug,
226243f3b051SJunchao Zhang      see https://github.com/pmodels/mpich/issues/5589. This bug was fixed by https://github.com/pmodels/mpich/pull/5590.
226341caa250SJunchao Zhang      In short, we could not use distributed RHS until with MPICH v4.0b1 or we enabled a workaround in mumps-5.6.2+
226425aac85cSJunchao Zhang    */
2265c183326eSPierre Jolivet   mumps->ICNTL20 = 10; /* Distributed dense RHS, by default */
2266c183326eSPierre Jolivet #if PETSC_PKG_MUMPS_VERSION_LT(5, 3, 0) || (PetscDefined(HAVE_MPICH) && MPICH_NUMVERSION < 40000101) || PetscDefined(HAVE_MSMPI)
2267c183326eSPierre Jolivet   mumps->ICNTL20 = 0; /* Centralized dense RHS, if need be */
226825aac85cSJunchao Zhang #endif
22699566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_20", "ICNTL(20): give mumps centralized (0) or distributed (10) dense right-hand sides", "None", mumps->ICNTL20, &mumps->ICNTL20, &flg));
2270aed4548fSBarry Smith   PetscCheck(!flg || mumps->ICNTL20 == 10 || mumps->ICNTL20 == 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "ICNTL(20)=%d is not supported by the PETSc/MUMPS interface. Allowed values are 0, 10", (int)mumps->ICNTL20);
227125aac85cSJunchao Zhang #if PETSC_PKG_MUMPS_VERSION_LT(5, 3, 0)
2272aed4548fSBarry Smith   PetscCheck(!flg || mumps->ICNTL20 != 10, PETSC_COMM_SELF, PETSC_ERR_SUP, "ICNTL(20)=10 is not supported before MUMPS-5.3.0");
227325aac85cSJunchao Zhang #endif
22749566063dSJacob Faibussowitsch   /* PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_21","ICNTL(21): the distribution (centralized or distributed) of the solution vectors","None",mumps->id.ICNTL(21),&mumps->id.ICNTL(21),NULL)); we only use distributed solution vector */
22759a2535b5SHong Zhang 
22769566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_22", "ICNTL(22): in-core/out-of-core factorization and solve (0 or 1)", "None", mumps->id.ICNTL(22), &mumps->id.ICNTL(22), NULL));
22779566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_23", "ICNTL(23): max size of the working memory (MB) that can allocate per processor", "None", mumps->id.ICNTL(23), &mumps->id.ICNTL(23), NULL));
22789566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_24", "ICNTL(24): detection of null pivot rows (0 or 1)", "None", mumps->id.ICNTL(24), &mumps->id.ICNTL(24), NULL));
22799371c9d4SSatish Balay   if (mumps->id.ICNTL(24)) { mumps->id.ICNTL(13) = 1; /* turn-off ScaLAPACK to help with the correct detection of null pivots */ }
2280d7ebd59bSHong Zhang 
22819566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_25", "ICNTL(25): computes a solution of a deficient matrix and a null space basis", "None", mumps->id.ICNTL(25), &mumps->id.ICNTL(25), NULL));
22829566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_26", "ICNTL(26): drives the solution phase if a Schur complement matrix", "None", mumps->id.ICNTL(26), &mumps->id.ICNTL(26), NULL));
22839566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_27", "ICNTL(27): controls the blocking size for multiple right-hand sides", "None", mumps->id.ICNTL(27), &mumps->id.ICNTL(27), NULL));
2284fa6fd9d0SPierre Jolivet   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_28", "ICNTL(28): use 1 for sequential analysis and ICNTL(7) ordering, or 2 for parallel analysis and ICNTL(29) ordering", "None", mumps->id.ICNTL(28), &mumps->id.ICNTL(28), NULL));
22859566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_29", "ICNTL(29): parallel ordering 1 = ptscotch, 2 = parmetis", "None", mumps->id.ICNTL(29), &mumps->id.ICNTL(29), NULL));
22869566063dSJacob Faibussowitsch   /* PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_30","ICNTL(30): compute user-specified set of entries in inv(A)","None",mumps->id.ICNTL(30),&mumps->id.ICNTL(30),NULL)); */ /* call MatMumpsGetInverse() directly */
22879566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_31", "ICNTL(31): indicates which factors may be discarded during factorization", "None", mumps->id.ICNTL(31), &mumps->id.ICNTL(31), NULL));
2288145b44c9SPierre Jolivet   /* PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_32","ICNTL(32): performs the forward elimination of the right-hand sides during factorization","None",mumps->id.ICNTL(32),&mumps->id.ICNTL(32),NULL));  -- not supported by PETSc API */
22899566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_33", "ICNTL(33): compute determinant", "None", mumps->id.ICNTL(33), &mumps->id.ICNTL(33), NULL));
22909566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_35", "ICNTL(35): activates Block Low Rank (BLR) based factorization", "None", mumps->id.ICNTL(35), &mumps->id.ICNTL(35), NULL));
22919566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_36", "ICNTL(36): choice of BLR factorization variant", "None", mumps->id.ICNTL(36), &mumps->id.ICNTL(36), NULL));
229250ea2040Saszaboa   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_37", "ICNTL(37): compression of the contribution blocks (CB)", "None", mumps->id.ICNTL(37), &mumps->id.ICNTL(37), NULL));
22939566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_38", "ICNTL(38): estimated compression rate of LU factors with BLR", "None", mumps->id.ICNTL(38), &mumps->id.ICNTL(38), NULL));
2294c92b4f89SPierre Jolivet   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_48", "ICNTL(48): multithreading with tree parallelism", "None", mumps->id.ICNTL(48), &mumps->id.ICNTL(48), NULL));
2295*91b026caSPierre Jolivet   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_56", "ICNTL(56): postponing and rank-revealing factorization", "None", mumps->id.ICNTL(56), &mumps->id.ICNTL(56), NULL));
2296146931dbSPierre Jolivet   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_58", "ICNTL(58): defines options for symbolic factorization", "None", mumps->id.ICNTL(58), &mumps->id.ICNTL(58), NULL));
2297dcd589f8SShri Abhyankar 
22989566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_1", "CNTL(1): relative pivoting threshold", "None", mumps->id.CNTL(1), &mumps->id.CNTL(1), NULL));
22999566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_2", "CNTL(2): stopping criterion of refinement", "None", mumps->id.CNTL(2), &mumps->id.CNTL(2), NULL));
23009566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_3", "CNTL(3): absolute pivoting threshold", "None", mumps->id.CNTL(3), &mumps->id.CNTL(3), NULL));
23019566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_4", "CNTL(4): value for static pivoting", "None", mumps->id.CNTL(4), &mumps->id.CNTL(4), NULL));
23029566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_5", "CNTL(5): fixation for null pivots", "None", mumps->id.CNTL(5), &mumps->id.CNTL(5), NULL));
23039566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_7", "CNTL(7): dropping parameter used during BLR", "None", mumps->id.CNTL(7), &mumps->id.CNTL(7), NULL));
2304e5bb22a1SHong Zhang 
23059566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-mat_mumps_ooc_tmpdir", "out of core directory", "None", mumps->id.ooc_tmpdir, mumps->id.ooc_tmpdir, sizeof(mumps->id.ooc_tmpdir), NULL));
2306b34f08ffSHong Zhang 
23079566063dSJacob Faibussowitsch   PetscCall(PetscOptionsIntArray("-mat_mumps_view_info", "request INFO local to each processor", "", info, &ninfo, NULL));
2308b34f08ffSHong Zhang   if (ninfo) {
230908401ef6SPierre Jolivet     PetscCheck(ninfo <= 80, PETSC_COMM_SELF, PETSC_ERR_USER, "number of INFO %" PetscInt_FMT " must <= 80", ninfo);
23109566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(ninfo, &mumps->info));
2311b34f08ffSHong Zhang     mumps->ninfo = ninfo;
2312b34f08ffSHong Zhang     for (i = 0; i < ninfo; i++) {
2313aed4548fSBarry Smith       PetscCheck(info[i] >= 0 && info[i] <= 80, PETSC_COMM_SELF, PETSC_ERR_USER, "index of INFO %" PetscInt_FMT " must between 1 and 80", ninfo);
2314f7d195e4SLawrence Mitchell       mumps->info[i] = info[i];
2315b34f08ffSHong Zhang     }
2316b34f08ffSHong Zhang   }
2317d0609cedSBarry Smith   PetscOptionsEnd();
23183ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2319dcd589f8SShri Abhyankar }
2320dcd589f8SShri Abhyankar 
2321d2a308c1SPierre Jolivet static PetscErrorCode MatFactorSymbolic_MUMPS_ReportIfError(Mat F, Mat A, PETSC_UNUSED const MatFactorInfo *info, Mat_MUMPS *mumps)
2322d71ae5a4SJacob Faibussowitsch {
23235cd7cf9dSHong Zhang   PetscFunctionBegin;
23245cd7cf9dSHong Zhang   if (mumps->id.INFOG(1) < 0) {
23259261f6e4SBarry Smith     PetscCheck(!A->erroriffailure, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in analysis: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
23265cd7cf9dSHong Zhang     if (mumps->id.INFOG(1) == -6) {
23279261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in analysis: matrix is singular, INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
2328603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_STRUCT_ZEROPIVOT;
23295cd7cf9dSHong Zhang     } else if (mumps->id.INFOG(1) == -5 || mumps->id.INFOG(1) == -7) {
23309261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in analysis: problem with work array, INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
2331603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OUTMEMORY;
2332dbf6bb8dSprj-     } else if (mumps->id.INFOG(1) == -16 && mumps->id.INFOG(1) == 0) {
23339261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in analysis: empty matrix\n"));
23345cd7cf9dSHong Zhang     } else {
23359261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in analysis: INFOG(1)=%d, INFO(2)=%d " MUMPS_MANUALS "\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
2336603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OTHER;
23375cd7cf9dSHong Zhang     }
23385cd7cf9dSHong Zhang   }
233972b150d8SStefano Zampini   if (!mumps->id.n) F->factorerrortype = MAT_FACTOR_NOERROR;
23403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
23415cd7cf9dSHong Zhang }
23425cd7cf9dSHong Zhang 
2343d2a308c1SPierre Jolivet static PetscErrorCode MatLUFactorSymbolic_AIJMUMPS(Mat F, Mat A, IS r, PETSC_UNUSED IS c, const MatFactorInfo *info)
2344d71ae5a4SJacob Faibussowitsch {
2345e69c285eSBarry Smith   Mat_MUMPS     *mumps = (Mat_MUMPS *)F->data;
234667877ebaSShri Abhyankar   Vec            b;
234767877ebaSShri Abhyankar   const PetscInt M = A->rmap->N;
2348397b6df1SKris Buschelman 
2349397b6df1SKris Buschelman   PetscFunctionBegin;
2350d47f36abSHong Zhang   if (mumps->matstruc == SAME_NONZERO_PATTERN) {
2351d47f36abSHong Zhang     /* F is assembled by a previous call of MatLUFactorSymbolic_AIJMUMPS() */
23523ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2353d47f36abSHong Zhang   }
2354dcd589f8SShri Abhyankar 
23559a2535b5SHong Zhang   /* Set MUMPS options from the options database */
235626cc229bSBarry Smith   PetscCall(MatSetFromOptions_MUMPS(F, A));
2357dcd589f8SShri Abhyankar 
23589566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_INITIAL_MATRIX, mumps));
23599566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_INITIAL_MATRIX, mumps));
2360dcd589f8SShri Abhyankar 
236167877ebaSShri Abhyankar   /* analysis phase */
2362a5e57a09SHong Zhang   mumps->id.job = JOB_FACTSYMBOLIC;
23636497c311SBarry Smith   PetscCall(PetscMUMPSIntCast(M, &mumps->id.n));
2364a5e57a09SHong Zhang   switch (mumps->id.ICNTL(18)) {
236567877ebaSShri Abhyankar   case 0: /* centralized assembled matrix input */
2366a5e57a09SHong Zhang     if (!mumps->myid) {
2367a6053eceSJunchao Zhang       mumps->id.nnz = mumps->nnz;
2368a6053eceSJunchao Zhang       mumps->id.irn = mumps->irn;
2369a6053eceSJunchao Zhang       mumps->id.jcn = mumps->jcn;
2370a6053eceSJunchao Zhang       if (mumps->id.ICNTL(6) > 1) mumps->id.a = (MumpsScalar *)mumps->val;
237151ad14ebSPierre Jolivet       if (r && mumps->id.ICNTL(7) == 7) {
23724ac6704cSBarry Smith         mumps->id.ICNTL(7) = 1;
2373a5e57a09SHong Zhang         if (!mumps->myid) {
2374e0b74bf9SHong Zhang           const PetscInt *idx;
2375a6053eceSJunchao Zhang           PetscInt        i;
23762205254eSKarl Rupp 
23779566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(M, &mumps->id.perm_in));
23789566063dSJacob Faibussowitsch           PetscCall(ISGetIndices(r, &idx));
2379f4f49eeaSPierre Jolivet           for (i = 0; i < M; i++) PetscCall(PetscMUMPSIntCast(idx[i] + 1, &mumps->id.perm_in[i])); /* perm_in[]: start from 1, not 0! */
23809566063dSJacob Faibussowitsch           PetscCall(ISRestoreIndices(r, &idx));
2381e0b74bf9SHong Zhang         }
2382e0b74bf9SHong Zhang       }
238367877ebaSShri Abhyankar     }
238467877ebaSShri Abhyankar     break;
238567877ebaSShri Abhyankar   case 3: /* distributed assembled matrix input (size>1) */
2386a6053eceSJunchao Zhang     mumps->id.nnz_loc = mumps->nnz;
2387a6053eceSJunchao Zhang     mumps->id.irn_loc = mumps->irn;
2388a6053eceSJunchao Zhang     mumps->id.jcn_loc = mumps->jcn;
2389a6053eceSJunchao Zhang     if (mumps->id.ICNTL(6) > 1) mumps->id.a_loc = (MumpsScalar *)mumps->val;
239025aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) { /* Centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */
23919566063dSJacob Faibussowitsch       PetscCall(MatCreateVecs(A, NULL, &b));
23929566063dSJacob Faibussowitsch       PetscCall(VecScatterCreateToZero(b, &mumps->scat_rhs, &mumps->b_seq));
23939566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b));
239425aac85cSJunchao Zhang     }
239567877ebaSShri Abhyankar     break;
239667877ebaSShri Abhyankar   }
23973ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
23989566063dSJacob Faibussowitsch   PetscCall(MatFactorSymbolic_MUMPS_ReportIfError(F, A, info, mumps));
239967877ebaSShri Abhyankar 
2400719d5645SBarry Smith   F->ops->lufactornumeric   = MatFactorNumeric_MUMPS;
2401dcd589f8SShri Abhyankar   F->ops->solve             = MatSolve_MUMPS;
240251d5961aSHong Zhang   F->ops->solvetranspose    = MatSolveTranspose_MUMPS;
24034e34a73bSHong Zhang   F->ops->matsolve          = MatMatSolve_MUMPS;
2404eb3ef3b2SHong Zhang   F->ops->mattransposesolve = MatMatTransposeSolve_MUMPS;
2405b18964edSHong Zhang   F->ops->matsolvetranspose = MatMatSolveTranspose_MUMPS;
2406d47f36abSHong Zhang 
2407d47f36abSHong Zhang   mumps->matstruc = SAME_NONZERO_PATTERN;
24083ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2409b24902e0SBarry Smith }
2410b24902e0SBarry Smith 
2411f0b74427SPierre Jolivet /* Note the PETSc r and c permutations are ignored */
2412d2a308c1SPierre Jolivet static PetscErrorCode MatLUFactorSymbolic_BAIJMUMPS(Mat F, Mat A, PETSC_UNUSED IS r, PETSC_UNUSED IS c, const MatFactorInfo *info)
2413d71ae5a4SJacob Faibussowitsch {
2414e69c285eSBarry Smith   Mat_MUMPS     *mumps = (Mat_MUMPS *)F->data;
241567877ebaSShri Abhyankar   Vec            b;
241667877ebaSShri Abhyankar   const PetscInt M = A->rmap->N;
2417450b117fSShri Abhyankar 
2418450b117fSShri Abhyankar   PetscFunctionBegin;
2419d47f36abSHong Zhang   if (mumps->matstruc == SAME_NONZERO_PATTERN) {
2420338d3105SPierre Jolivet     /* F is assembled by a previous call of MatLUFactorSymbolic_BAIJMUMPS() */
24213ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2422d47f36abSHong Zhang   }
2423dcd589f8SShri Abhyankar 
24249a2535b5SHong Zhang   /* Set MUMPS options from the options database */
242526cc229bSBarry Smith   PetscCall(MatSetFromOptions_MUMPS(F, A));
2426dcd589f8SShri Abhyankar 
24279566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_INITIAL_MATRIX, mumps));
24289566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_INITIAL_MATRIX, mumps));
242967877ebaSShri Abhyankar 
243067877ebaSShri Abhyankar   /* analysis phase */
2431a5e57a09SHong Zhang   mumps->id.job = JOB_FACTSYMBOLIC;
24326497c311SBarry Smith   PetscCall(PetscMUMPSIntCast(M, &mumps->id.n));
2433a5e57a09SHong Zhang   switch (mumps->id.ICNTL(18)) {
243467877ebaSShri Abhyankar   case 0: /* centralized assembled matrix input */
2435a5e57a09SHong Zhang     if (!mumps->myid) {
2436a6053eceSJunchao Zhang       mumps->id.nnz = mumps->nnz;
2437a6053eceSJunchao Zhang       mumps->id.irn = mumps->irn;
2438a6053eceSJunchao Zhang       mumps->id.jcn = mumps->jcn;
2439ad540459SPierre Jolivet       if (mumps->id.ICNTL(6) > 1) mumps->id.a = (MumpsScalar *)mumps->val;
244067877ebaSShri Abhyankar     }
244167877ebaSShri Abhyankar     break;
244267877ebaSShri Abhyankar   case 3: /* distributed assembled matrix input (size>1) */
2443a6053eceSJunchao Zhang     mumps->id.nnz_loc = mumps->nnz;
2444a6053eceSJunchao Zhang     mumps->id.irn_loc = mumps->irn;
2445a6053eceSJunchao Zhang     mumps->id.jcn_loc = mumps->jcn;
2446ad540459SPierre Jolivet     if (mumps->id.ICNTL(6) > 1) mumps->id.a_loc = (MumpsScalar *)mumps->val;
244725aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) { /* Centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */
24489566063dSJacob Faibussowitsch       PetscCall(MatCreateVecs(A, NULL, &b));
24499566063dSJacob Faibussowitsch       PetscCall(VecScatterCreateToZero(b, &mumps->scat_rhs, &mumps->b_seq));
24509566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b));
245125aac85cSJunchao Zhang     }
245267877ebaSShri Abhyankar     break;
245367877ebaSShri Abhyankar   }
24543ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
24559566063dSJacob Faibussowitsch   PetscCall(MatFactorSymbolic_MUMPS_ReportIfError(F, A, info, mumps));
245667877ebaSShri Abhyankar 
2457450b117fSShri Abhyankar   F->ops->lufactornumeric   = MatFactorNumeric_MUMPS;
2458dcd589f8SShri Abhyankar   F->ops->solve             = MatSolve_MUMPS;
245951d5961aSHong Zhang   F->ops->solvetranspose    = MatSolveTranspose_MUMPS;
2460b18964edSHong Zhang   F->ops->matsolvetranspose = MatMatSolveTranspose_MUMPS;
2461d47f36abSHong Zhang 
2462d47f36abSHong Zhang   mumps->matstruc = SAME_NONZERO_PATTERN;
24633ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2464450b117fSShri Abhyankar }
2465b24902e0SBarry Smith 
2466f0b74427SPierre Jolivet /* Note the PETSc r permutation and factor info are ignored */
2467d2a308c1SPierre Jolivet static PetscErrorCode MatCholeskyFactorSymbolic_MUMPS(Mat F, Mat A, PETSC_UNUSED IS r, const MatFactorInfo *info)
2468d71ae5a4SJacob Faibussowitsch {
2469e69c285eSBarry Smith   Mat_MUMPS     *mumps = (Mat_MUMPS *)F->data;
247067877ebaSShri Abhyankar   Vec            b;
247167877ebaSShri Abhyankar   const PetscInt M = A->rmap->N;
2472397b6df1SKris Buschelman 
2473397b6df1SKris Buschelman   PetscFunctionBegin;
2474d47f36abSHong Zhang   if (mumps->matstruc == SAME_NONZERO_PATTERN) {
2475338d3105SPierre Jolivet     /* F is assembled by a previous call of MatCholeskyFactorSymbolic_MUMPS() */
24763ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2477d47f36abSHong Zhang   }
2478dcd589f8SShri Abhyankar 
24799a2535b5SHong Zhang   /* Set MUMPS options from the options database */
248026cc229bSBarry Smith   PetscCall(MatSetFromOptions_MUMPS(F, A));
2481dcd589f8SShri Abhyankar 
24829566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_INITIAL_MATRIX, mumps));
24839566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_INITIAL_MATRIX, mumps));
2484dcd589f8SShri Abhyankar 
248567877ebaSShri Abhyankar   /* analysis phase */
2486a5e57a09SHong Zhang   mumps->id.job = JOB_FACTSYMBOLIC;
24876497c311SBarry Smith   PetscCall(PetscMUMPSIntCast(M, &mumps->id.n));
2488a5e57a09SHong Zhang   switch (mumps->id.ICNTL(18)) {
248967877ebaSShri Abhyankar   case 0: /* centralized assembled matrix input */
2490a5e57a09SHong Zhang     if (!mumps->myid) {
2491a6053eceSJunchao Zhang       mumps->id.nnz = mumps->nnz;
2492a6053eceSJunchao Zhang       mumps->id.irn = mumps->irn;
2493a6053eceSJunchao Zhang       mumps->id.jcn = mumps->jcn;
2494ad540459SPierre Jolivet       if (mumps->id.ICNTL(6) > 1) mumps->id.a = (MumpsScalar *)mumps->val;
249567877ebaSShri Abhyankar     }
249667877ebaSShri Abhyankar     break;
249767877ebaSShri Abhyankar   case 3: /* distributed assembled matrix input (size>1) */
2498a6053eceSJunchao Zhang     mumps->id.nnz_loc = mumps->nnz;
2499a6053eceSJunchao Zhang     mumps->id.irn_loc = mumps->irn;
2500a6053eceSJunchao Zhang     mumps->id.jcn_loc = mumps->jcn;
2501ad540459SPierre Jolivet     if (mumps->id.ICNTL(6) > 1) mumps->id.a_loc = (MumpsScalar *)mumps->val;
250225aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) { /* Centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */
25039566063dSJacob Faibussowitsch       PetscCall(MatCreateVecs(A, NULL, &b));
25049566063dSJacob Faibussowitsch       PetscCall(VecScatterCreateToZero(b, &mumps->scat_rhs, &mumps->b_seq));
25059566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b));
250625aac85cSJunchao Zhang     }
250767877ebaSShri Abhyankar     break;
250867877ebaSShri Abhyankar   }
25093ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
25109566063dSJacob Faibussowitsch   PetscCall(MatFactorSymbolic_MUMPS_ReportIfError(F, A, info, mumps));
25115cd7cf9dSHong Zhang 
25122792810eSHong Zhang   F->ops->choleskyfactornumeric = MatFactorNumeric_MUMPS;
2513dcd589f8SShri Abhyankar   F->ops->solve                 = MatSolve_MUMPS;
251451d5961aSHong Zhang   F->ops->solvetranspose        = MatSolve_MUMPS;
25154e34a73bSHong Zhang   F->ops->matsolve              = MatMatSolve_MUMPS;
251623a5080aSHong Zhang   F->ops->mattransposesolve     = MatMatTransposeSolve_MUMPS;
2517b18964edSHong Zhang   F->ops->matsolvetranspose     = MatMatSolveTranspose_MUMPS;
25184e34a73bSHong Zhang #if defined(PETSC_USE_COMPLEX)
25190298fd71SBarry Smith   F->ops->getinertia = NULL;
25204e34a73bSHong Zhang #else
25214e34a73bSHong Zhang   F->ops->getinertia = MatGetInertia_SBAIJMUMPS;
2522db4efbfdSBarry Smith #endif
2523d47f36abSHong Zhang 
2524d47f36abSHong Zhang   mumps->matstruc = SAME_NONZERO_PATTERN;
25253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2526b24902e0SBarry Smith }
2527b24902e0SBarry Smith 
252866976f2fSJacob Faibussowitsch static PetscErrorCode MatView_MUMPS(Mat A, PetscViewer viewer)
2529d71ae5a4SJacob Faibussowitsch {
253064e6c443SBarry Smith   PetscBool         iascii;
253164e6c443SBarry Smith   PetscViewerFormat format;
2532e69c285eSBarry Smith   Mat_MUMPS        *mumps = (Mat_MUMPS *)A->data;
2533f6c57405SHong Zhang 
2534f6c57405SHong Zhang   PetscFunctionBegin;
253564e6c443SBarry Smith   /* check if matrix is mumps type */
25363ba16761SJacob Faibussowitsch   if (A->ops->solve != MatSolve_MUMPS) PetscFunctionReturn(PETSC_SUCCESS);
253764e6c443SBarry Smith 
25389566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
253964e6c443SBarry Smith   if (iascii) {
25409566063dSJacob Faibussowitsch     PetscCall(PetscViewerGetFormat(viewer, &format));
25411511cd71SPierre Jolivet     if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
25429566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "MUMPS run parameters:\n"));
25431511cd71SPierre Jolivet       if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
25449566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  SYM (matrix type):                   %d\n", mumps->id.sym));
25459566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  PAR (host participation):            %d\n", mumps->id.par));
25469566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(1) (output for error):         %d\n", mumps->id.ICNTL(1)));
25479566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(2) (output of diagnostic msg): %d\n", mumps->id.ICNTL(2)));
25489566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(3) (output for global info):   %d\n", mumps->id.ICNTL(3)));
25499566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(4) (level of printing):        %d\n", mumps->id.ICNTL(4)));
25509566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(5) (input mat struct):         %d\n", mumps->id.ICNTL(5)));
25519566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(6) (matrix prescaling):        %d\n", mumps->id.ICNTL(6)));
25529566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(7) (sequential matrix ordering):%d\n", mumps->id.ICNTL(7)));
25539566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(8) (scaling strategy):         %d\n", mumps->id.ICNTL(8)));
25549566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(10) (max num of refinements):  %d\n", mumps->id.ICNTL(10)));
25559566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(11) (error analysis):          %d\n", mumps->id.ICNTL(11)));
2556a5e57a09SHong Zhang         if (mumps->id.ICNTL(11) > 0) {
2557d2a308c1SPierre Jolivet           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(4) (inf norm of input mat):        %g\n", (double)mumps->id.RINFOG(4)));
2558d2a308c1SPierre Jolivet           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(5) (inf norm of solution):         %g\n", (double)mumps->id.RINFOG(5)));
2559d2a308c1SPierre Jolivet           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(6) (inf norm of residual):         %g\n", (double)mumps->id.RINFOG(6)));
2560d2a308c1SPierre Jolivet           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(7),RINFOG(8) (backward error est): %g, %g\n", (double)mumps->id.RINFOG(7), (double)mumps->id.RINFOG(8)));
2561d2a308c1SPierre Jolivet           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(9) (error estimate):               %g\n", (double)mumps->id.RINFOG(9)));
2562d2a308c1SPierre Jolivet           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(10),RINFOG(11)(condition numbers): %g, %g\n", (double)mumps->id.RINFOG(10), (double)mumps->id.RINFOG(11)));
2563f6c57405SHong Zhang         }
25649566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(12) (efficiency control):                         %d\n", mumps->id.ICNTL(12)));
25659566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(13) (sequential factorization of the root node):  %d\n", mumps->id.ICNTL(13)));
25669566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(14) (percentage of estimated workspace increase): %d\n", mumps->id.ICNTL(14)));
256745e3843bSPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(15) (compression of the input matrix):            %d\n", mumps->id.ICNTL(15)));
2568f6c57405SHong Zhang         /* ICNTL(15-17) not used */
25699566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(18) (input mat struct):                           %d\n", mumps->id.ICNTL(18)));
25709566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(19) (Schur complement info):                      %d\n", mumps->id.ICNTL(19)));
25719566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(20) (RHS sparse pattern):                         %d\n", mumps->id.ICNTL(20)));
25729566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(21) (solution struct):                            %d\n", mumps->id.ICNTL(21)));
25739566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(22) (in-core/out-of-core facility):               %d\n", mumps->id.ICNTL(22)));
25749566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(23) (max size of memory can be allocated locally):%d\n", mumps->id.ICNTL(23)));
2575c0165424SHong Zhang 
25769566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(24) (detection of null pivot rows):               %d\n", mumps->id.ICNTL(24)));
25779566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(25) (computation of a null space basis):          %d\n", mumps->id.ICNTL(25)));
25789566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(26) (Schur options for RHS or solution):          %d\n", mumps->id.ICNTL(26)));
25799566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(27) (blocking size for multiple RHS):             %d\n", mumps->id.ICNTL(27)));
25809566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(28) (use parallel or sequential ordering):        %d\n", mumps->id.ICNTL(28)));
25819566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(29) (parallel ordering):                          %d\n", mumps->id.ICNTL(29)));
258242179a6aSHong Zhang 
25839566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(30) (user-specified set of entries in inv(A)):    %d\n", mumps->id.ICNTL(30)));
25849566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(31) (factors is discarded in the solve phase):    %d\n", mumps->id.ICNTL(31)));
25859566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(33) (compute determinant):                        %d\n", mumps->id.ICNTL(33)));
25869566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(35) (activate BLR based factorization):           %d\n", mumps->id.ICNTL(35)));
25879566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(36) (choice of BLR factorization variant):        %d\n", mumps->id.ICNTL(36)));
258850ea2040Saszaboa         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(37) (compression of the contribution blocks):     %d\n", mumps->id.ICNTL(37)));
25899566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(38) (estimated compression rate of LU factors):   %d\n", mumps->id.ICNTL(38)));
2590c92b4f89SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(48) (multithreading with tree parallelism):       %d\n", mumps->id.ICNTL(48)));
2591*91b026caSPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(56) (postponing and rank-revealing factorization):%d\n", mumps->id.ICNTL(56)));
2592146931dbSPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(58) (options for symbolic factorization):         %d\n", mumps->id.ICNTL(58)));
2593f6c57405SHong Zhang 
2594d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(1) (relative pivoting threshold):      %g\n", (double)mumps->id.CNTL(1)));
2595d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(2) (stopping criterion of refinement): %g\n", (double)mumps->id.CNTL(2)));
2596d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(3) (absolute pivoting threshold):      %g\n", (double)mumps->id.CNTL(3)));
2597d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(4) (value of static pivoting):         %g\n", (double)mumps->id.CNTL(4)));
2598d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(5) (fixation for null pivots):         %g\n", (double)mumps->id.CNTL(5)));
2599d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(7) (dropping parameter for BLR):       %g\n", (double)mumps->id.CNTL(7)));
2600f6c57405SHong Zhang 
2601a5b23f4aSJose E. Roman         /* information local to each processor */
26029566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFO(1) (local estimated flops for the elimination after analysis):\n"));
26039566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPushSynchronized(viewer));
2604d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %g\n", mumps->myid, (double)mumps->id.RINFO(1)));
26059566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
26069566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFO(2) (local estimated flops for the assembly after factorization):\n"));
2607d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %g\n", mumps->myid, (double)mumps->id.RINFO(2)));
26089566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
26099566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFO(3) (local estimated flops for the elimination after factorization):\n"));
2610d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %g\n", mumps->myid, (double)mumps->id.RINFO(3)));
26119566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2612f6c57405SHong Zhang 
26139566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(15) (estimated size of (in MB) MUMPS internal data for running numerical factorization):\n"));
26149566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(15)));
26159566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2616f6c57405SHong Zhang 
26179566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(16) (size of (in MB) MUMPS internal data used during numerical factorization):\n"));
26189566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(16)));
26199566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2620f6c57405SHong Zhang 
26219566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(23) (num of pivots eliminated on this processor after factorization):\n"));
26229566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(23)));
26239566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2624b34f08ffSHong Zhang 
2625a0e18203SThibaut Appel         if (mumps->ninfo && mumps->ninfo <= 80) {
2626b34f08ffSHong Zhang           PetscInt i;
2627b34f08ffSHong Zhang           for (i = 0; i < mumps->ninfo; i++) {
26289566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(%" PetscInt_FMT "):\n", mumps->info[i]));
26299566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(mumps->info[i])));
26309566063dSJacob Faibussowitsch             PetscCall(PetscViewerFlush(viewer));
2631b34f08ffSHong Zhang           }
2632b34f08ffSHong Zhang         }
26339566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPopSynchronized(viewer));
26341511cd71SPierre Jolivet       } else PetscCall(PetscViewerASCIIPrintf(viewer, "  Use -%sksp_view ::ascii_info_detail to display information for all processes\n", ((PetscObject)A)->prefix ? ((PetscObject)A)->prefix : ""));
2635f6c57405SHong Zhang 
26361511cd71SPierre Jolivet       if (mumps->myid == 0) { /* information from the host */
2637d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFOG(1) (global estimated flops for the elimination after analysis): %g\n", (double)mumps->id.RINFOG(1)));
2638d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFOG(2) (global estimated flops for the assembly after factorization): %g\n", (double)mumps->id.RINFOG(2)));
2639d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFOG(3) (global estimated flops for the elimination after factorization): %g\n", (double)mumps->id.RINFOG(3)));
2640d2a308c1SPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  (RINFOG(12) RINFOG(13))*2^INFOG(34) (determinant): (%g,%g)*(2^%d)\n", (double)mumps->id.RINFOG(12), (double)mumps->id.RINFOG(13), mumps->id.INFOG(34)));
2641f6c57405SHong Zhang 
26429566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(3) (estimated real workspace for factors on all processors after analysis): %d\n", mumps->id.INFOG(3)));
26439566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(4) (estimated integer workspace for factors on all processors after analysis): %d\n", mumps->id.INFOG(4)));
26449566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(5) (estimated maximum front size in the complete tree): %d\n", mumps->id.INFOG(5)));
26459566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(6) (number of nodes in the complete tree): %d\n", mumps->id.INFOG(6)));
26469566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(7) (ordering option effectively used after analysis): %d\n", mumps->id.INFOG(7)));
26479566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(8) (structural symmetry in percent of the permuted matrix after analysis): %d\n", mumps->id.INFOG(8)));
26489566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(9) (total real/complex workspace to store the matrix factors after factorization): %d\n", mumps->id.INFOG(9)));
26499566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(10) (total integer space store the matrix factors after factorization): %d\n", mumps->id.INFOG(10)));
26509566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(11) (order of largest frontal matrix after factorization): %d\n", mumps->id.INFOG(11)));
26519566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(12) (number of off-diagonal pivots): %d\n", mumps->id.INFOG(12)));
26529566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(13) (number of delayed pivots after factorization): %d\n", mumps->id.INFOG(13)));
26539566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(14) (number of memory compress after factorization): %d\n", mumps->id.INFOG(14)));
26549566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(15) (number of steps of iterative refinement after solution): %d\n", mumps->id.INFOG(15)));
26559566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(16) (estimated size (in MB) of all MUMPS internal data for factorization after analysis: value on the most memory consuming processor): %d\n", mumps->id.INFOG(16)));
26569566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(17) (estimated size of all MUMPS internal data for factorization after analysis: sum over all processors): %d\n", mumps->id.INFOG(17)));
26579566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(18) (size of all MUMPS internal data allocated during factorization: value on the most memory consuming processor): %d\n", mumps->id.INFOG(18)));
26589566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(19) (size of all MUMPS internal data allocated during factorization: sum over all processors): %d\n", mumps->id.INFOG(19)));
26599566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(20) (estimated number of entries in the factors): %d\n", mumps->id.INFOG(20)));
26609566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(21) (size in MB of memory effectively used during factorization - value on the most memory consuming processor): %d\n", mumps->id.INFOG(21)));
26619566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(22) (size in MB of memory effectively used during factorization - sum over all processors): %d\n", mumps->id.INFOG(22)));
26629566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(23) (after analysis: value of ICNTL(6) effectively used): %d\n", mumps->id.INFOG(23)));
26639566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(24) (after analysis: value of ICNTL(12) effectively used): %d\n", mumps->id.INFOG(24)));
26649566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(25) (after factorization: number of pivots modified by static pivoting): %d\n", mumps->id.INFOG(25)));
26659566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(28) (after factorization: number of null pivots encountered): %d\n", mumps->id.INFOG(28)));
26669566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(29) (after factorization: effective number of entries in the factors (sum over all processors)): %d\n", mumps->id.INFOG(29)));
26679566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(30, 31) (after solution: size in Mbytes of memory used during solution phase): %d, %d\n", mumps->id.INFOG(30), mumps->id.INFOG(31)));
26689566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(32) (after analysis: type of analysis done): %d\n", mumps->id.INFOG(32)));
26699566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(33) (value used for ICNTL(8)): %d\n", mumps->id.INFOG(33)));
26709566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(34) (exponent of the determinant if determinant is requested): %d\n", mumps->id.INFOG(34)));
26719566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(35) (after factorization: number of entries taking into account BLR factor compression - sum over all processors): %d\n", mumps->id.INFOG(35)));
26729566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(36) (after analysis: estimated size of all MUMPS internal data for running BLR in-core - value on the most memory consuming processor): %d\n", mumps->id.INFOG(36)));
26739566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(37) (after analysis: estimated size of all MUMPS internal data for running BLR in-core - sum over all processors): %d\n", mumps->id.INFOG(37)));
26749566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(38) (after analysis: estimated size of all MUMPS internal data for running BLR out-of-core - value on the most memory consuming processor): %d\n", mumps->id.INFOG(38)));
26759566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(39) (after analysis: estimated size of all MUMPS internal data for running BLR out-of-core - sum over all processors): %d\n", mumps->id.INFOG(39)));
2676f6c57405SHong Zhang       }
2677f6c57405SHong Zhang     }
2678cb828f0fSHong Zhang   }
26793ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2680f6c57405SHong Zhang }
2681f6c57405SHong Zhang 
2682d2a308c1SPierre Jolivet static PetscErrorCode MatGetInfo_MUMPS(Mat A, PETSC_UNUSED MatInfoType flag, MatInfo *info)
2683d71ae5a4SJacob Faibussowitsch {
2684e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)A->data;
268535bd34faSBarry Smith 
268635bd34faSBarry Smith   PetscFunctionBegin;
268735bd34faSBarry Smith   info->block_size        = 1.0;
268864412097SPierre Jolivet   info->nz_allocated      = mumps->id.INFOG(20) >= 0 ? mumps->id.INFOG(20) : -1000000 * mumps->id.INFOG(20);
268964412097SPierre Jolivet   info->nz_used           = mumps->id.INFOG(20) >= 0 ? mumps->id.INFOG(20) : -1000000 * mumps->id.INFOG(20);
269035bd34faSBarry Smith   info->nz_unneeded       = 0.0;
269135bd34faSBarry Smith   info->assemblies        = 0.0;
269235bd34faSBarry Smith   info->mallocs           = 0.0;
269335bd34faSBarry Smith   info->memory            = 0.0;
269435bd34faSBarry Smith   info->fill_ratio_given  = 0;
269535bd34faSBarry Smith   info->fill_ratio_needed = 0;
269635bd34faSBarry Smith   info->factor_mallocs    = 0;
26973ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
269835bd34faSBarry Smith }
269935bd34faSBarry Smith 
270066976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorSetSchurIS_MUMPS(Mat F, IS is)
2701d71ae5a4SJacob Faibussowitsch {
2702e69c285eSBarry Smith   Mat_MUMPS         *mumps = (Mat_MUMPS *)F->data;
2703a3d589ffSStefano Zampini   const PetscScalar *arr;
27048e7ba810SStefano Zampini   const PetscInt    *idxs;
27058e7ba810SStefano Zampini   PetscInt           size, i;
27066444a565SStefano Zampini 
27076444a565SStefano Zampini   PetscFunctionBegin;
27089566063dSJacob Faibussowitsch   PetscCall(ISGetLocalSize(is, &size));
2709b3cb21ddSStefano Zampini   /* Schur complement matrix */
27109566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&F->schur));
27119566063dSJacob Faibussowitsch   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, size, size, NULL, &F->schur));
27129566063dSJacob Faibussowitsch   PetscCall(MatDenseGetArrayRead(F->schur, &arr));
2713a3d589ffSStefano Zampini   mumps->id.schur = (MumpsScalar *)arr;
27146497c311SBarry Smith   PetscCall(PetscMUMPSIntCast(size, &mumps->id.size_schur));
27156497c311SBarry Smith   PetscCall(PetscMUMPSIntCast(size, &mumps->id.schur_lld));
27169566063dSJacob Faibussowitsch   PetscCall(MatDenseRestoreArrayRead(F->schur, &arr));
271748a46eb9SPierre Jolivet   if (mumps->sym == 1) PetscCall(MatSetOption(F->schur, MAT_SPD, PETSC_TRUE));
2718b3cb21ddSStefano Zampini 
2719b3cb21ddSStefano Zampini   /* MUMPS expects Fortran style indices */
27209566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.listvar_schur));
27219566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(size, &mumps->id.listvar_schur));
27229566063dSJacob Faibussowitsch   PetscCall(ISGetIndices(is, &idxs));
2723f4f49eeaSPierre Jolivet   for (i = 0; i < size; i++) PetscCall(PetscMUMPSIntCast(idxs[i] + 1, &mumps->id.listvar_schur[i]));
27249566063dSJacob Faibussowitsch   PetscCall(ISRestoreIndices(is, &idxs));
272559ac8732SStefano Zampini   /* set a special value of ICNTL (not handled my MUMPS) to be used in the solve phase by PETSc */
2726b5fa320bSStefano Zampini   mumps->id.ICNTL(26) = -1;
27273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
27286444a565SStefano Zampini }
272959ac8732SStefano Zampini 
273066976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorCreateSchurComplement_MUMPS(Mat F, Mat *S)
2731d71ae5a4SJacob Faibussowitsch {
27326444a565SStefano Zampini   Mat          St;
2733e69c285eSBarry Smith   Mat_MUMPS   *mumps = (Mat_MUMPS *)F->data;
27346444a565SStefano Zampini   PetscScalar *array;
27356444a565SStefano Zampini 
27366444a565SStefano Zampini   PetscFunctionBegin;
27379261f6e4SBarry Smith   PetscCheck(mumps->id.ICNTL(19), PetscObjectComm((PetscObject)F), PETSC_ERR_ORDER, "Schur complement mode not selected! Call MatFactorSetSchurIS() to enable it");
27389566063dSJacob Faibussowitsch   PetscCall(MatCreate(PETSC_COMM_SELF, &St));
27399566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(St, PETSC_DECIDE, PETSC_DECIDE, mumps->id.size_schur, mumps->id.size_schur));
27409566063dSJacob Faibussowitsch   PetscCall(MatSetType(St, MATDENSE));
27419566063dSJacob Faibussowitsch   PetscCall(MatSetUp(St));
27429566063dSJacob Faibussowitsch   PetscCall(MatDenseGetArray(St, &array));
274359ac8732SStefano Zampini   if (!mumps->sym) {                /* MUMPS always return a full matrix */
27446444a565SStefano Zampini     if (mumps->id.ICNTL(19) == 1) { /* stored by rows */
27456444a565SStefano Zampini       PetscInt i, j, N = mumps->id.size_schur;
27466444a565SStefano Zampini       for (i = 0; i < N; i++) {
27476444a565SStefano Zampini         for (j = 0; j < N; j++) {
27486444a565SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
27496444a565SStefano Zampini           PetscScalar val = mumps->id.schur[i * N + j];
27506444a565SStefano Zampini #else
27513e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.schur[i * N + j].r + PETSC_i * mumps->id.schur[i * N + j].i;
27526444a565SStefano Zampini #endif
27536444a565SStefano Zampini           array[j * N + i] = val;
27546444a565SStefano Zampini         }
27556444a565SStefano Zampini       }
27566444a565SStefano Zampini     } else { /* stored by columns */
27579566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(array, mumps->id.schur, mumps->id.size_schur * mumps->id.size_schur));
27586444a565SStefano Zampini     }
27596444a565SStefano Zampini   } else {                          /* either full or lower-triangular (not packed) */
27606444a565SStefano Zampini     if (mumps->id.ICNTL(19) == 2) { /* lower triangular stored by columns */
27616444a565SStefano Zampini       PetscInt i, j, N = mumps->id.size_schur;
27626444a565SStefano Zampini       for (i = 0; i < N; i++) {
27636444a565SStefano Zampini         for (j = i; j < N; j++) {
27646444a565SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
27656444a565SStefano Zampini           PetscScalar val = mumps->id.schur[i * N + j];
27666444a565SStefano Zampini #else
27673e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.schur[i * N + j].r + PETSC_i * mumps->id.schur[i * N + j].i;
27686444a565SStefano Zampini #endif
27693e5b40d0SPierre Jolivet           array[i * N + j] = array[j * N + i] = val;
27706444a565SStefano Zampini         }
27716444a565SStefano Zampini       }
27726444a565SStefano Zampini     } else if (mumps->id.ICNTL(19) == 3) { /* full matrix */
27739566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(array, mumps->id.schur, mumps->id.size_schur * mumps->id.size_schur));
27746444a565SStefano Zampini     } else { /* ICNTL(19) == 1 lower triangular stored by rows */
27756444a565SStefano Zampini       PetscInt i, j, N = mumps->id.size_schur;
27766444a565SStefano Zampini       for (i = 0; i < N; i++) {
27776444a565SStefano Zampini         for (j = 0; j < i + 1; j++) {
27786444a565SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
27796444a565SStefano Zampini           PetscScalar val = mumps->id.schur[i * N + j];
27806444a565SStefano Zampini #else
27813e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.schur[i * N + j].r + PETSC_i * mumps->id.schur[i * N + j].i;
27826444a565SStefano Zampini #endif
27833e5b40d0SPierre Jolivet           array[i * N + j] = array[j * N + i] = val;
27846444a565SStefano Zampini         }
27856444a565SStefano Zampini       }
27866444a565SStefano Zampini     }
27876444a565SStefano Zampini   }
27889566063dSJacob Faibussowitsch   PetscCall(MatDenseRestoreArray(St, &array));
27896444a565SStefano Zampini   *S = St;
27903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
27916444a565SStefano Zampini }
27926444a565SStefano Zampini 
279366976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsSetIcntl_MUMPS(Mat F, PetscInt icntl, PetscInt ival)
2794d71ae5a4SJacob Faibussowitsch {
2795e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
27965ccb76cbSHong Zhang 
27975ccb76cbSHong Zhang   PetscFunctionBegin;
2798413bcc21SPierre Jolivet   if (mumps->id.job == JOB_NULL) {                                            /* need to cache icntl and ival since PetscMUMPS_c() has never been called */
27996497c311SBarry Smith     PetscMUMPSInt i, nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0; /* number of already cached ICNTL */
28009371c9d4SSatish Balay     for (i = 0; i < nICNTL_pre; ++i)
28019371c9d4SSatish Balay       if (mumps->ICNTL_pre[1 + 2 * i] == icntl) break; /* is this ICNTL already cached? */
2802413bcc21SPierre Jolivet     if (i == nICNTL_pre) {                             /* not already cached */
2803413bcc21SPierre Jolivet       if (i > 0) PetscCall(PetscRealloc(sizeof(PetscMUMPSInt) * (2 * nICNTL_pre + 3), &mumps->ICNTL_pre));
2804413bcc21SPierre Jolivet       else PetscCall(PetscCalloc(sizeof(PetscMUMPSInt) * 3, &mumps->ICNTL_pre));
2805413bcc21SPierre Jolivet       mumps->ICNTL_pre[0]++;
2806413bcc21SPierre Jolivet     }
28076497c311SBarry Smith     mumps->ICNTL_pre[1 + 2 * i] = (PetscMUMPSInt)icntl;
2808413bcc21SPierre Jolivet     PetscCall(PetscMUMPSIntCast(ival, mumps->ICNTL_pre + 2 + 2 * i));
2809413bcc21SPierre Jolivet   } else PetscCall(PetscMUMPSIntCast(ival, &mumps->id.ICNTL(icntl)));
28103ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
28115ccb76cbSHong Zhang }
28125ccb76cbSHong Zhang 
281366976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetIcntl_MUMPS(Mat F, PetscInt icntl, PetscInt *ival)
2814d71ae5a4SJacob Faibussowitsch {
2815e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2816bc6112feSHong Zhang 
2817bc6112feSHong Zhang   PetscFunctionBegin;
281836df9881Sjeremy theler   if (mumps->id.job == JOB_NULL) {
281936df9881Sjeremy theler     PetscInt i, nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0;
282036df9881Sjeremy theler     *ival = 0;
282136df9881Sjeremy theler     for (i = 0; i < nICNTL_pre; ++i) {
282236df9881Sjeremy theler       if (mumps->ICNTL_pre[1 + 2 * i] == icntl) *ival = mumps->ICNTL_pre[2 + 2 * i];
282336df9881Sjeremy theler     }
282436df9881Sjeremy theler   } else *ival = mumps->id.ICNTL(icntl);
28253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2826bc6112feSHong Zhang }
2827bc6112feSHong Zhang 
28285ccb76cbSHong Zhang /*@
28291d27aa22SBarry Smith   MatMumpsSetIcntl - Set MUMPS parameter ICNTL() <https://mumps-solver.org/index.php?page=doc>
28305ccb76cbSHong Zhang 
2831c3339decSBarry Smith   Logically Collective
28325ccb76cbSHong Zhang 
28335ccb76cbSHong Zhang   Input Parameters:
28340b4b7b1cSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
283579578405SBarry Smith . icntl - index of MUMPS parameter array `ICNTL()`
283679578405SBarry Smith - ival  - value of MUMPS `ICNTL(icntl)`
28375ccb76cbSHong Zhang 
28383c7db156SBarry Smith   Options Database Key:
283979578405SBarry Smith . -mat_mumps_icntl_<icntl> <ival> - change the option numbered `icntl` to `ival`
28405ccb76cbSHong Zhang 
28415ccb76cbSHong Zhang   Level: beginner
28425ccb76cbSHong Zhang 
284379578405SBarry Smith   Note:
284479578405SBarry Smith   Ignored if MUMPS is not installed or `F` is not a MUMPS matrix
284579578405SBarry Smith 
28461cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
28475ccb76cbSHong Zhang @*/
2848d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsSetIcntl(Mat F, PetscInt icntl, PetscInt ival)
2849d71ae5a4SJacob Faibussowitsch {
28505ccb76cbSHong Zhang   PetscFunctionBegin;
28512989dfd4SHong Zhang   PetscValidType(F, 1);
285228b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
28535ccb76cbSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
28545ccb76cbSHong Zhang   PetscValidLogicalCollectiveInt(F, ival, 3);
2855*91b026caSPierre Jolivet   PetscCheck((icntl >= 1 && icntl <= 38) || icntl == 48 || icntl == 56 || icntl == 58, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported ICNTL value %" PetscInt_FMT, icntl);
2856cac4c232SBarry Smith   PetscTryMethod(F, "MatMumpsSetIcntl_C", (Mat, PetscInt, PetscInt), (F, icntl, ival));
28573ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
28585ccb76cbSHong Zhang }
28595ccb76cbSHong Zhang 
2860a21f80fcSHong Zhang /*@
28611d27aa22SBarry Smith   MatMumpsGetIcntl - Get MUMPS parameter ICNTL() <https://mumps-solver.org/index.php?page=doc>
2862a21f80fcSHong Zhang 
2863c3339decSBarry Smith   Logically Collective
2864a21f80fcSHong Zhang 
2865a21f80fcSHong Zhang   Input Parameters:
28660b4b7b1cSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
2867a21f80fcSHong Zhang - icntl - index of MUMPS parameter array ICNTL()
2868a21f80fcSHong Zhang 
2869a21f80fcSHong Zhang   Output Parameter:
2870a21f80fcSHong Zhang . ival - value of MUMPS ICNTL(icntl)
2871a21f80fcSHong Zhang 
2872a21f80fcSHong Zhang   Level: beginner
2873a21f80fcSHong Zhang 
28741cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
2875a21f80fcSHong Zhang @*/
2876d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetIcntl(Mat F, PetscInt icntl, PetscInt *ival)
2877d71ae5a4SJacob Faibussowitsch {
2878bc6112feSHong Zhang   PetscFunctionBegin;
28792989dfd4SHong Zhang   PetscValidType(F, 1);
288028b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
2881bc6112feSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
28824f572ea9SToby Isaac   PetscAssertPointer(ival, 3);
2883c92b4f89SPierre Jolivet   PetscCheck((icntl >= 1 && icntl <= 38) || icntl == 48 || icntl == 58, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported ICNTL value %" PetscInt_FMT, icntl);
2884cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetIcntl_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival));
28853ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2886bc6112feSHong Zhang }
2887bc6112feSHong Zhang 
288866976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsSetCntl_MUMPS(Mat F, PetscInt icntl, PetscReal val)
2889d71ae5a4SJacob Faibussowitsch {
2890e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
28918928b65cSHong Zhang 
28928928b65cSHong Zhang   PetscFunctionBegin;
2893413bcc21SPierre Jolivet   if (mumps->id.job == JOB_NULL) {
2894413bcc21SPierre Jolivet     PetscInt i, nCNTL_pre = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0;
28959371c9d4SSatish Balay     for (i = 0; i < nCNTL_pre; ++i)
28969371c9d4SSatish Balay       if (mumps->CNTL_pre[1 + 2 * i] == icntl) break;
2897413bcc21SPierre Jolivet     if (i == nCNTL_pre) {
2898413bcc21SPierre Jolivet       if (i > 0) PetscCall(PetscRealloc(sizeof(PetscReal) * (2 * nCNTL_pre + 3), &mumps->CNTL_pre));
2899413bcc21SPierre Jolivet       else PetscCall(PetscCalloc(sizeof(PetscReal) * 3, &mumps->CNTL_pre));
2900413bcc21SPierre Jolivet       mumps->CNTL_pre[0]++;
2901413bcc21SPierre Jolivet     }
2902413bcc21SPierre Jolivet     mumps->CNTL_pre[1 + 2 * i] = icntl;
2903413bcc21SPierre Jolivet     mumps->CNTL_pre[2 + 2 * i] = val;
2904413bcc21SPierre Jolivet   } else mumps->id.CNTL(icntl) = val;
29053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
29068928b65cSHong Zhang }
29078928b65cSHong Zhang 
290866976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetCntl_MUMPS(Mat F, PetscInt icntl, PetscReal *val)
2909d71ae5a4SJacob Faibussowitsch {
2910e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2911bc6112feSHong Zhang 
2912bc6112feSHong Zhang   PetscFunctionBegin;
291336df9881Sjeremy theler   if (mumps->id.job == JOB_NULL) {
291436df9881Sjeremy theler     PetscInt i, nCNTL_pre = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0;
291536df9881Sjeremy theler     *val = 0.0;
291636df9881Sjeremy theler     for (i = 0; i < nCNTL_pre; ++i) {
291736df9881Sjeremy theler       if (mumps->CNTL_pre[1 + 2 * i] == icntl) *val = mumps->CNTL_pre[2 + 2 * i];
291836df9881Sjeremy theler     }
291936df9881Sjeremy theler   } else *val = mumps->id.CNTL(icntl);
29203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2921bc6112feSHong Zhang }
2922bc6112feSHong Zhang 
29238928b65cSHong Zhang /*@
29241d27aa22SBarry Smith   MatMumpsSetCntl - Set MUMPS parameter CNTL() <https://mumps-solver.org/index.php?page=doc>
29258928b65cSHong Zhang 
2926c3339decSBarry Smith   Logically Collective
29278928b65cSHong Zhang 
29288928b65cSHong Zhang   Input Parameters:
29290b4b7b1cSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
293079578405SBarry Smith . icntl - index of MUMPS parameter array `CNTL()`
293179578405SBarry Smith - val   - value of MUMPS `CNTL(icntl)`
29328928b65cSHong Zhang 
29333c7db156SBarry Smith   Options Database Key:
2934147403d9SBarry Smith . -mat_mumps_cntl_<icntl> <val> - change the option numbered icntl to ival
29358928b65cSHong Zhang 
29368928b65cSHong Zhang   Level: beginner
29378928b65cSHong Zhang 
293879578405SBarry Smith   Note:
293979578405SBarry Smith   Ignored if MUMPS is not installed or `F` is not a MUMPS matrix
294079578405SBarry Smith 
29411cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
29428928b65cSHong Zhang @*/
2943d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsSetCntl(Mat F, PetscInt icntl, PetscReal val)
2944d71ae5a4SJacob Faibussowitsch {
29458928b65cSHong Zhang   PetscFunctionBegin;
29462989dfd4SHong Zhang   PetscValidType(F, 1);
294728b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
29488928b65cSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
2949bc6112feSHong Zhang   PetscValidLogicalCollectiveReal(F, val, 3);
2950413bcc21SPierre Jolivet   PetscCheck(icntl >= 1 && icntl <= 7, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported CNTL value %" PetscInt_FMT, icntl);
2951cac4c232SBarry Smith   PetscTryMethod(F, "MatMumpsSetCntl_C", (Mat, PetscInt, PetscReal), (F, icntl, val));
29523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
29538928b65cSHong Zhang }
29548928b65cSHong Zhang 
2955a21f80fcSHong Zhang /*@
29561d27aa22SBarry Smith   MatMumpsGetCntl - Get MUMPS parameter CNTL() <https://mumps-solver.org/index.php?page=doc>
2957a21f80fcSHong Zhang 
2958c3339decSBarry Smith   Logically Collective
2959a21f80fcSHong Zhang 
2960a21f80fcSHong Zhang   Input Parameters:
29610b4b7b1cSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
2962a21f80fcSHong Zhang - icntl - index of MUMPS parameter array CNTL()
2963a21f80fcSHong Zhang 
2964a21f80fcSHong Zhang   Output Parameter:
2965a21f80fcSHong Zhang . val - value of MUMPS CNTL(icntl)
2966a21f80fcSHong Zhang 
2967a21f80fcSHong Zhang   Level: beginner
2968a21f80fcSHong Zhang 
29691cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
2970a21f80fcSHong Zhang @*/
2971d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetCntl(Mat F, PetscInt icntl, PetscReal *val)
2972d71ae5a4SJacob Faibussowitsch {
2973bc6112feSHong Zhang   PetscFunctionBegin;
29742989dfd4SHong Zhang   PetscValidType(F, 1);
297528b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
2976bc6112feSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
29774f572ea9SToby Isaac   PetscAssertPointer(val, 3);
2978413bcc21SPierre Jolivet   PetscCheck(icntl >= 1 && icntl <= 7, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported CNTL value %" PetscInt_FMT, icntl);
2979cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetCntl_C", (Mat, PetscInt, PetscReal *), (F, icntl, val));
29803ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2981bc6112feSHong Zhang }
2982bc6112feSHong Zhang 
298366976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInfo_MUMPS(Mat F, PetscInt icntl, PetscInt *info)
2984d71ae5a4SJacob Faibussowitsch {
2985e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2986bc6112feSHong Zhang 
2987bc6112feSHong Zhang   PetscFunctionBegin;
2988bc6112feSHong Zhang   *info = mumps->id.INFO(icntl);
29893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2990bc6112feSHong Zhang }
2991bc6112feSHong Zhang 
299266976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInfog_MUMPS(Mat F, PetscInt icntl, PetscInt *infog)
2993d71ae5a4SJacob Faibussowitsch {
2994e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2995bc6112feSHong Zhang 
2996bc6112feSHong Zhang   PetscFunctionBegin;
2997bc6112feSHong Zhang   *infog = mumps->id.INFOG(icntl);
29983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2999bc6112feSHong Zhang }
3000bc6112feSHong Zhang 
300166976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetRinfo_MUMPS(Mat F, PetscInt icntl, PetscReal *rinfo)
3002d71ae5a4SJacob Faibussowitsch {
3003e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
3004bc6112feSHong Zhang 
3005bc6112feSHong Zhang   PetscFunctionBegin;
3006bc6112feSHong Zhang   *rinfo = mumps->id.RINFO(icntl);
30073ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3008bc6112feSHong Zhang }
3009bc6112feSHong Zhang 
301066976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetRinfog_MUMPS(Mat F, PetscInt icntl, PetscReal *rinfog)
3011d71ae5a4SJacob Faibussowitsch {
3012e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
3013bc6112feSHong Zhang 
3014bc6112feSHong Zhang   PetscFunctionBegin;
3015bc6112feSHong Zhang   *rinfog = mumps->id.RINFOG(icntl);
30163ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3017bc6112feSHong Zhang }
3018bc6112feSHong Zhang 
301966976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetNullPivots_MUMPS(Mat F, PetscInt *size, PetscInt **array)
30205c0bae8cSAshish Patel {
30215c0bae8cSAshish Patel   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
30225c0bae8cSAshish Patel 
30235c0bae8cSAshish Patel   PetscFunctionBegin;
30245c0bae8cSAshish Patel   PetscCheck(mumps->id.ICNTL(24) == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "-mat_mumps_icntl_24 must be set as 1 for null pivot row detection");
30255c0bae8cSAshish Patel   *size  = 0;
30265c0bae8cSAshish Patel   *array = NULL;
30275c0bae8cSAshish Patel   if (!mumps->myid) {
30285c0bae8cSAshish Patel     *size = mumps->id.INFOG(28);
30295c0bae8cSAshish Patel     PetscCall(PetscMalloc1(*size, array));
30305c0bae8cSAshish Patel     for (int i = 0; i < *size; i++) (*array)[i] = mumps->id.pivnul_list[i] - 1;
30315c0bae8cSAshish Patel   }
30325c0bae8cSAshish Patel   PetscFunctionReturn(PETSC_SUCCESS);
30335c0bae8cSAshish Patel }
30345c0bae8cSAshish Patel 
303566976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInverse_MUMPS(Mat F, Mat spRHS)
3036d71ae5a4SJacob Faibussowitsch {
30370e6b8875SHong Zhang   Mat          Bt = NULL, Btseq = NULL;
30380e6b8875SHong Zhang   PetscBool    flg;
3039bb599dfdSHong Zhang   Mat_MUMPS   *mumps = (Mat_MUMPS *)F->data;
3040bb599dfdSHong Zhang   PetscScalar *aa;
3041f410b75aSHong Zhang   PetscInt     spnr, *ia, *ja, M, nrhs;
3042bb599dfdSHong Zhang 
3043bb599dfdSHong Zhang   PetscFunctionBegin;
30444f572ea9SToby Isaac   PetscAssertPointer(spRHS, 2);
3045013e2dc7SBarry Smith   PetscCall(PetscObjectTypeCompare((PetscObject)spRHS, MATTRANSPOSEVIRTUAL, &flg));
304653587d93SPierre Jolivet   PetscCheck(flg, PetscObjectComm((PetscObject)spRHS), PETSC_ERR_ARG_WRONG, "Matrix spRHS must be type MATTRANSPOSEVIRTUAL matrix");
304753587d93SPierre Jolivet   PetscCall(MatShellGetScalingShifts(spRHS, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (PetscScalar *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Vec *)MAT_SHELL_NOT_ALLOWED, (Mat *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED, (IS *)MAT_SHELL_NOT_ALLOWED));
30489566063dSJacob Faibussowitsch   PetscCall(MatTransposeGetMat(spRHS, &Bt));
3049bb599dfdSHong Zhang 
30509566063dSJacob Faibussowitsch   PetscCall(MatMumpsSetIcntl(F, 30, 1));
3051bb599dfdSHong Zhang 
30522d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) {
30530e6b8875SHong Zhang     Mat_MPIAIJ *b = (Mat_MPIAIJ *)Bt->data;
30540e6b8875SHong Zhang     Btseq         = b->A;
30550e6b8875SHong Zhang   } else {
30560e6b8875SHong Zhang     Btseq = Bt;
30570e6b8875SHong Zhang   }
30580e6b8875SHong Zhang 
30599566063dSJacob Faibussowitsch   PetscCall(MatGetSize(spRHS, &M, &nrhs));
30606497c311SBarry Smith   mumps->id.nrhs = (PetscMUMPSInt)nrhs;
30616497c311SBarry Smith   PetscCall(PetscMUMPSIntCast(M, &mumps->id.lrhs));
3062f410b75aSHong Zhang   mumps->id.rhs = NULL;
3063f410b75aSHong Zhang 
3064e3f2db6aSHong Zhang   if (!mumps->myid) {
30659566063dSJacob Faibussowitsch     PetscCall(MatSeqAIJGetArray(Btseq, &aa));
30669566063dSJacob Faibussowitsch     PetscCall(MatGetRowIJ(Btseq, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
306728b400f6SJacob Faibussowitsch     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
30689566063dSJacob Faibussowitsch     PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs));
3069bb599dfdSHong Zhang     mumps->id.rhs_sparse = (MumpsScalar *)aa;
3070e3f2db6aSHong Zhang   } else {
3071e3f2db6aSHong Zhang     mumps->id.irhs_ptr    = NULL;
3072e3f2db6aSHong Zhang     mumps->id.irhs_sparse = NULL;
3073e3f2db6aSHong Zhang     mumps->id.nz_rhs      = 0;
3074e3f2db6aSHong Zhang     mumps->id.rhs_sparse  = NULL;
3075e3f2db6aSHong Zhang   }
3076bb599dfdSHong Zhang   mumps->id.ICNTL(20) = 1; /* rhs is sparse */
3077e3f2db6aSHong Zhang   mumps->id.ICNTL(21) = 0; /* solution is in assembled centralized format */
3078bb599dfdSHong Zhang 
3079bb599dfdSHong Zhang   /* solve phase */
3080bb599dfdSHong Zhang   mumps->id.job = JOB_SOLVE;
30813ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
30829261f6e4SBarry Smith   PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in solve: INFOG(1)=%d INFO(2)=%d " MUMPS_MANUALS, mumps->id.INFOG(1), mumps->id.INFO(2));
308314267174SHong Zhang 
3084e3f2db6aSHong Zhang   if (!mumps->myid) {
30859566063dSJacob Faibussowitsch     PetscCall(MatSeqAIJRestoreArray(Btseq, &aa));
30869566063dSJacob Faibussowitsch     PetscCall(MatRestoreRowIJ(Btseq, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
308728b400f6SJacob Faibussowitsch     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
3088e3f2db6aSHong Zhang   }
30893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3090bb599dfdSHong Zhang }
3091bb599dfdSHong Zhang 
3092bb599dfdSHong Zhang /*@
30931d27aa22SBarry Smith   MatMumpsGetInverse - Get user-specified set of entries in inverse of `A` <https://mumps-solver.org/index.php?page=doc>
3094bb599dfdSHong Zhang 
3095c3339decSBarry Smith   Logically Collective
3096bb599dfdSHong Zhang 
309720f4b53cSBarry Smith   Input Parameter:
30980b4b7b1cSBarry Smith . F - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
3099bb599dfdSHong Zhang 
3100bb599dfdSHong Zhang   Output Parameter:
310120f4b53cSBarry Smith . spRHS - sequential sparse matrix in `MATTRANSPOSEVIRTUAL` format with requested entries of inverse of `A`
3102bb599dfdSHong Zhang 
3103bb599dfdSHong Zhang   Level: beginner
3104bb599dfdSHong Zhang 
31051cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatCreateTranspose()`
3106bb599dfdSHong Zhang @*/
3107d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInverse(Mat F, Mat spRHS)
3108d71ae5a4SJacob Faibussowitsch {
3109bb599dfdSHong Zhang   PetscFunctionBegin;
3110bb599dfdSHong Zhang   PetscValidType(F, 1);
311128b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
3112cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInverse_C", (Mat, Mat), (F, spRHS));
31133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3114bb599dfdSHong Zhang }
3115bb599dfdSHong Zhang 
311666976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInverseTranspose_MUMPS(Mat F, Mat spRHST)
3117d71ae5a4SJacob Faibussowitsch {
31180e6b8875SHong Zhang   Mat spRHS;
31190e6b8875SHong Zhang 
31200e6b8875SHong Zhang   PetscFunctionBegin;
31219566063dSJacob Faibussowitsch   PetscCall(MatCreateTranspose(spRHST, &spRHS));
31229566063dSJacob Faibussowitsch   PetscCall(MatMumpsGetInverse_MUMPS(F, spRHS));
31239566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&spRHS));
31243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
31250e6b8875SHong Zhang }
31260e6b8875SHong Zhang 
31270e6b8875SHong Zhang /*@
31281d27aa22SBarry Smith   MatMumpsGetInverseTranspose - Get user-specified set of entries in inverse of matrix $A^T $ <https://mumps-solver.org/index.php?page=doc>
31290e6b8875SHong Zhang 
3130c3339decSBarry Smith   Logically Collective
31310e6b8875SHong Zhang 
313220f4b53cSBarry Smith   Input Parameter:
31330b4b7b1cSBarry Smith . F - the factored matrix of A obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
31340e6b8875SHong Zhang 
31350e6b8875SHong Zhang   Output Parameter:
313620f4b53cSBarry Smith . spRHST - sequential sparse matrix in `MATAIJ` format containing the requested entries of inverse of `A`^T
31370e6b8875SHong Zhang 
31380e6b8875SHong Zhang   Level: beginner
31390e6b8875SHong Zhang 
31401cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatCreateTranspose()`, `MatMumpsGetInverse()`
31410e6b8875SHong Zhang @*/
3142d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInverseTranspose(Mat F, Mat spRHST)
3143d71ae5a4SJacob Faibussowitsch {
31440e6b8875SHong Zhang   PetscBool flg;
31450e6b8875SHong Zhang 
31460e6b8875SHong Zhang   PetscFunctionBegin;
31470e6b8875SHong Zhang   PetscValidType(F, 1);
314828b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
31499566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)spRHST, &flg, MATSEQAIJ, MATMPIAIJ, NULL));
315028b400f6SJacob Faibussowitsch   PetscCheck(flg, PetscObjectComm((PetscObject)spRHST), PETSC_ERR_ARG_WRONG, "Matrix spRHST must be MATAIJ matrix");
3151cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInverseTranspose_C", (Mat, Mat), (F, spRHST));
31523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
31530e6b8875SHong Zhang }
31540e6b8875SHong Zhang 
3155a21f80fcSHong Zhang /*@
31561d27aa22SBarry Smith   MatMumpsGetInfo - Get MUMPS parameter INFO() <https://mumps-solver.org/index.php?page=doc>
3157a21f80fcSHong Zhang 
3158c3339decSBarry Smith   Logically Collective
3159a21f80fcSHong Zhang 
3160a21f80fcSHong Zhang   Input Parameters:
31610b4b7b1cSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
3162a21f80fcSHong Zhang - icntl - index of MUMPS parameter array INFO()
3163a21f80fcSHong Zhang 
3164a21f80fcSHong Zhang   Output Parameter:
3165a21f80fcSHong Zhang . ival - value of MUMPS INFO(icntl)
3166a21f80fcSHong Zhang 
3167a21f80fcSHong Zhang   Level: beginner
3168a21f80fcSHong Zhang 
31691cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
3170a21f80fcSHong Zhang @*/
3171d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInfo(Mat F, PetscInt icntl, PetscInt *ival)
3172d71ae5a4SJacob Faibussowitsch {
3173bc6112feSHong Zhang   PetscFunctionBegin;
31742989dfd4SHong Zhang   PetscValidType(F, 1);
317528b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
31764f572ea9SToby Isaac   PetscAssertPointer(ival, 3);
3177cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInfo_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival));
31783ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3179bc6112feSHong Zhang }
3180bc6112feSHong Zhang 
3181a21f80fcSHong Zhang /*@
31821d27aa22SBarry Smith   MatMumpsGetInfog - Get MUMPS parameter INFOG() <https://mumps-solver.org/index.php?page=doc>
3183a21f80fcSHong Zhang 
3184c3339decSBarry Smith   Logically Collective
3185a21f80fcSHong Zhang 
3186a21f80fcSHong Zhang   Input Parameters:
31870b4b7b1cSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
3188a21f80fcSHong Zhang - icntl - index of MUMPS parameter array INFOG()
3189a21f80fcSHong Zhang 
3190a21f80fcSHong Zhang   Output Parameter:
3191a21f80fcSHong Zhang . ival - value of MUMPS INFOG(icntl)
3192a21f80fcSHong Zhang 
3193a21f80fcSHong Zhang   Level: beginner
3194a21f80fcSHong Zhang 
31951cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
3196a21f80fcSHong Zhang @*/
3197d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInfog(Mat F, PetscInt icntl, PetscInt *ival)
3198d71ae5a4SJacob Faibussowitsch {
3199bc6112feSHong Zhang   PetscFunctionBegin;
32002989dfd4SHong Zhang   PetscValidType(F, 1);
320128b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
32024f572ea9SToby Isaac   PetscAssertPointer(ival, 3);
3203cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInfog_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival));
32043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3205bc6112feSHong Zhang }
3206bc6112feSHong Zhang 
3207a21f80fcSHong Zhang /*@
32081d27aa22SBarry Smith   MatMumpsGetRinfo - Get MUMPS parameter RINFO() <https://mumps-solver.org/index.php?page=doc>
3209a21f80fcSHong Zhang 
3210c3339decSBarry Smith   Logically Collective
3211a21f80fcSHong Zhang 
3212a21f80fcSHong Zhang   Input Parameters:
32130b4b7b1cSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
3214a21f80fcSHong Zhang - icntl - index of MUMPS parameter array RINFO()
3215a21f80fcSHong Zhang 
3216a21f80fcSHong Zhang   Output Parameter:
3217a21f80fcSHong Zhang . val - value of MUMPS RINFO(icntl)
3218a21f80fcSHong Zhang 
3219a21f80fcSHong Zhang   Level: beginner
3220a21f80fcSHong Zhang 
32211cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfog()`
3222a21f80fcSHong Zhang @*/
3223d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetRinfo(Mat F, PetscInt icntl, PetscReal *val)
3224d71ae5a4SJacob Faibussowitsch {
3225bc6112feSHong Zhang   PetscFunctionBegin;
32262989dfd4SHong Zhang   PetscValidType(F, 1);
322728b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
32284f572ea9SToby Isaac   PetscAssertPointer(val, 3);
3229cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetRinfo_C", (Mat, PetscInt, PetscReal *), (F, icntl, val));
32303ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3231bc6112feSHong Zhang }
3232bc6112feSHong Zhang 
3233a21f80fcSHong Zhang /*@
32341d27aa22SBarry Smith   MatMumpsGetRinfog - Get MUMPS parameter RINFOG() <https://mumps-solver.org/index.php?page=doc>
3235a21f80fcSHong Zhang 
3236c3339decSBarry Smith   Logically Collective
3237a21f80fcSHong Zhang 
3238a21f80fcSHong Zhang   Input Parameters:
32390b4b7b1cSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
3240a21f80fcSHong Zhang - icntl - index of MUMPS parameter array RINFOG()
3241a21f80fcSHong Zhang 
3242a21f80fcSHong Zhang   Output Parameter:
3243a21f80fcSHong Zhang . val - value of MUMPS RINFOG(icntl)
3244a21f80fcSHong Zhang 
3245a21f80fcSHong Zhang   Level: beginner
3246a21f80fcSHong Zhang 
32471cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`
3248a21f80fcSHong Zhang @*/
3249d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetRinfog(Mat F, PetscInt icntl, PetscReal *val)
3250d71ae5a4SJacob Faibussowitsch {
3251bc6112feSHong Zhang   PetscFunctionBegin;
32522989dfd4SHong Zhang   PetscValidType(F, 1);
325328b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
32544f572ea9SToby Isaac   PetscAssertPointer(val, 3);
3255cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetRinfog_C", (Mat, PetscInt, PetscReal *), (F, icntl, val));
32563ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3257bc6112feSHong Zhang }
3258bc6112feSHong Zhang 
32595c0bae8cSAshish Patel /*@
32601d27aa22SBarry Smith   MatMumpsGetNullPivots - Get MUMPS parameter PIVNUL_LIST() <https://mumps-solver.org/index.php?page=doc>
32615c0bae8cSAshish Patel 
32625c0bae8cSAshish Patel   Logically Collective
32635c0bae8cSAshish Patel 
32645c0bae8cSAshish Patel   Input Parameter:
32650b4b7b1cSBarry Smith . F - the factored matrix obtained by calling `MatGetFactor()` with a `MatSolverType` of `MATSOLVERMUMPS` and a `MatFactorType` of `MAT_FACTOR_LU` or `MAT_FACTOR_CHOLESKY`
32665c0bae8cSAshish Patel 
32675c0bae8cSAshish Patel   Output Parameters:
32680b4b7b1cSBarry Smith + size  - local size of the array. The size of the array is non-zero only on MPI rank 0
32695c0bae8cSAshish Patel - array - array of rows with null pivot, these rows follow 0-based indexing. The array gets allocated within the function and the user is responsible
32705c0bae8cSAshish Patel           for freeing this array.
32715c0bae8cSAshish Patel 
32725c0bae8cSAshish Patel   Level: beginner
32735c0bae8cSAshish Patel 
32741cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`
32755c0bae8cSAshish Patel @*/
32765c0bae8cSAshish Patel PetscErrorCode MatMumpsGetNullPivots(Mat F, PetscInt *size, PetscInt **array)
32775c0bae8cSAshish Patel {
32785c0bae8cSAshish Patel   PetscFunctionBegin;
32795c0bae8cSAshish Patel   PetscValidType(F, 1);
32805c0bae8cSAshish Patel   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
32814f572ea9SToby Isaac   PetscAssertPointer(size, 2);
32824f572ea9SToby Isaac   PetscAssertPointer(array, 3);
32835c0bae8cSAshish Patel   PetscUseMethod(F, "MatMumpsGetNullPivots_C", (Mat, PetscInt *, PetscInt **), (F, size, array));
32845c0bae8cSAshish Patel   PetscFunctionReturn(PETSC_SUCCESS);
32855c0bae8cSAshish Patel }
32865c0bae8cSAshish Patel 
328724b6179bSKris Buschelman /*MC
32882692d6eeSBarry Smith   MATSOLVERMUMPS -  A matrix type providing direct solvers (LU and Cholesky) for
32890b4b7b1cSBarry Smith   MPI distributed and sequential matrices via the external package MUMPS <https://mumps-solver.org/index.php?page=doc>
329024b6179bSKris Buschelman 
329111a5261eSBarry Smith   Works with `MATAIJ` and `MATSBAIJ` matrices
329224b6179bSKris Buschelman 
3293c2b89b5dSBarry Smith   Use ./configure --download-mumps --download-scalapack --download-parmetis --download-metis --download-ptscotch to have PETSc installed with MUMPS
3294c2b89b5dSBarry Smith 
32952ef1f0ffSBarry Smith   Use ./configure --with-openmp --download-hwloc (or --with-hwloc) to enable running MUMPS in MPI+OpenMP hybrid mode and non-MUMPS in flat-MPI mode.
32962ef1f0ffSBarry Smith   See details below.
3297217d3b1eSJunchao Zhang 
32982ef1f0ffSBarry Smith   Use `-pc_type cholesky` or `lu` `-pc_factor_mat_solver_type mumps` to use this direct solver
3299c2b89b5dSBarry Smith 
330024b6179bSKris Buschelman   Options Database Keys:
33014422a9fcSPatrick Sanan +  -mat_mumps_icntl_1  - ICNTL(1): output stream for error messages
33024422a9fcSPatrick Sanan .  -mat_mumps_icntl_2  - ICNTL(2): output stream for diagnostic printing, statistics, and warning
33034422a9fcSPatrick Sanan .  -mat_mumps_icntl_3  - ICNTL(3): output stream for global information, collected on the host
33044422a9fcSPatrick Sanan .  -mat_mumps_icntl_4  - ICNTL(4): level of printing (0 to 4)
33054422a9fcSPatrick Sanan .  -mat_mumps_icntl_6  - ICNTL(6): permutes to a zero-free diagonal and/or scale the matrix (0 to 7)
3306b53c1a7fSBarry Smith .  -mat_mumps_icntl_7  - ICNTL(7): computes a symmetric permutation in sequential analysis, 0=AMD, 2=AMF, 3=Scotch, 4=PORD, 5=Metis, 6=QAMD, and 7=auto
3307b53c1a7fSBarry Smith                           Use -pc_factor_mat_ordering_type <type> to have PETSc perform the ordering (sequential only)
33084422a9fcSPatrick Sanan .  -mat_mumps_icntl_8  - ICNTL(8): scaling strategy (-2 to 8 or 77)
33094422a9fcSPatrick Sanan .  -mat_mumps_icntl_10 - ICNTL(10): max num of refinements
33104422a9fcSPatrick Sanan .  -mat_mumps_icntl_11 - ICNTL(11): statistics related to an error analysis (via -ksp_view)
33114422a9fcSPatrick Sanan .  -mat_mumps_icntl_12 - ICNTL(12): an ordering strategy for symmetric matrices (0 to 3)
33124422a9fcSPatrick Sanan .  -mat_mumps_icntl_13 - ICNTL(13): parallelism of the root node (enable ScaLAPACK) and its splitting
33134422a9fcSPatrick Sanan .  -mat_mumps_icntl_14 - ICNTL(14): percentage increase in the estimated working space
331445e3843bSPierre Jolivet .  -mat_mumps_icntl_15 - ICNTL(15): compression of the input matrix resulting from a block format
33154422a9fcSPatrick Sanan .  -mat_mumps_icntl_19 - ICNTL(19): computes the Schur complement
331625aac85cSJunchao Zhang .  -mat_mumps_icntl_20 - ICNTL(20): give MUMPS centralized (0) or distributed (10) dense RHS
33174422a9fcSPatrick Sanan .  -mat_mumps_icntl_22 - ICNTL(22): in-core/out-of-core factorization and solve (0 or 1)
33184422a9fcSPatrick Sanan .  -mat_mumps_icntl_23 - ICNTL(23): max size of the working memory (MB) that can allocate per processor
33194422a9fcSPatrick Sanan .  -mat_mumps_icntl_24 - ICNTL(24): detection of null pivot rows (0 or 1)
33204422a9fcSPatrick Sanan .  -mat_mumps_icntl_25 - ICNTL(25): compute a solution of a deficient matrix and a null space basis
33214422a9fcSPatrick Sanan .  -mat_mumps_icntl_26 - ICNTL(26): drives the solution phase if a Schur complement matrix
3322fa6fd9d0SPierre Jolivet .  -mat_mumps_icntl_28 - ICNTL(28): use 1 for sequential analysis and ICNTL(7) ordering, or 2 for parallel analysis and ICNTL(29) ordering
33234422a9fcSPatrick Sanan .  -mat_mumps_icntl_29 - ICNTL(29): parallel ordering 1 = ptscotch, 2 = parmetis
33244422a9fcSPatrick Sanan .  -mat_mumps_icntl_30 - ICNTL(30): compute user-specified set of entries in inv(A)
33254422a9fcSPatrick Sanan .  -mat_mumps_icntl_31 - ICNTL(31): indicates which factors may be discarded during factorization
33264422a9fcSPatrick Sanan .  -mat_mumps_icntl_33 - ICNTL(33): compute determinant
3327a0e18203SThibaut Appel .  -mat_mumps_icntl_35 - ICNTL(35): level of activation of BLR (Block Low-Rank) feature
3328a0e18203SThibaut Appel .  -mat_mumps_icntl_36 - ICNTL(36): controls the choice of BLR factorization variant
332950ea2040Saszaboa .  -mat_mumps_icntl_37 - ICNTL(37): compression of the contribution blocks (CB)
3330a0e18203SThibaut Appel .  -mat_mumps_icntl_38 - ICNTL(38): sets the estimated compression rate of LU factors with BLR
3331c92b4f89SPierre Jolivet .  -mat_mumps_icntl_48 - ICNTL(48): multithreading with tree parallelism
3332146931dbSPierre Jolivet .  -mat_mumps_icntl_58 - ICNTL(58): options for symbolic factorization
33334422a9fcSPatrick Sanan .  -mat_mumps_cntl_1   - CNTL(1): relative pivoting threshold
33344422a9fcSPatrick Sanan .  -mat_mumps_cntl_2   - CNTL(2): stopping criterion of refinement
33354422a9fcSPatrick Sanan .  -mat_mumps_cntl_3   - CNTL(3): absolute pivoting threshold
33364422a9fcSPatrick Sanan .  -mat_mumps_cntl_4   - CNTL(4): value for static pivoting
3337217d3b1eSJunchao Zhang .  -mat_mumps_cntl_5   - CNTL(5): fixation for null pivots
3338a0e18203SThibaut Appel .  -mat_mumps_cntl_7   - CNTL(7): precision of the dropping parameter used during BLR factorization
3339217d3b1eSJunchao Zhang -  -mat_mumps_use_omp_threads [m] - run MUMPS in MPI+OpenMP hybrid mode as if omp_set_num_threads(m) is called before calling MUMPS.
3340217d3b1eSJunchao Zhang                                     Default might be the number of cores per CPU package (socket) as reported by hwloc and suggested by the MUMPS manual.
334124b6179bSKris Buschelman 
334224b6179bSKris Buschelman   Level: beginner
334324b6179bSKris Buschelman 
334495452b02SPatrick Sanan   Notes:
33451d27aa22SBarry Smith   MUMPS Cholesky does not handle (complex) Hermitian matrices (see User's Guide at <https://mumps-solver.org/index.php?page=doc>) so using it will
33462ef1f0ffSBarry Smith   error if the matrix is Hermitian.
334738548759SBarry Smith 
334826cc229bSBarry Smith   When used within a `KSP`/`PC` solve the options are prefixed with that of the `PC`. Otherwise one can set the options prefix by calling
334926cc229bSBarry Smith   `MatSetOptionsPrefixFactor()` on the matrix from which the factor was obtained or `MatSetOptionsPrefix()` on the factor matrix.
335026cc229bSBarry Smith 
33512ef1f0ffSBarry Smith   When a MUMPS factorization fails inside a KSP solve, for example with a `KSP_DIVERGED_PC_FAILED`, one can find the MUMPS information about
33522ef1f0ffSBarry Smith   the failure with
33532ef1f0ffSBarry Smith .vb
33542ef1f0ffSBarry Smith           KSPGetPC(ksp,&pc);
33552ef1f0ffSBarry Smith           PCFactorGetMatrix(pc,&mat);
33562ef1f0ffSBarry Smith           MatMumpsGetInfo(mat,....);
33572ef1f0ffSBarry Smith           MatMumpsGetInfog(mat,....); etc.
33582ef1f0ffSBarry Smith .ve
33592ef1f0ffSBarry Smith   Or run with `-ksp_error_if_not_converged` and the program will be stopped and the information printed in the error message.
33609fc87aa7SBarry Smith 
3361a5399872SJunchao Zhang   MUMPS provides 64-bit integer support in two build modes:
3362a5399872SJunchao Zhang   full 64-bit: here MUMPS is built with C preprocessing flag -DINTSIZE64 and Fortran compiler option -i8, -fdefault-integer-8 or equivalent, and
3363a5399872SJunchao Zhang   requires all dependent libraries MPI, ScaLAPACK, LAPACK and BLAS built the same way with 64-bit integers (for example ILP64 Intel MKL and MPI).
33648fcaa860SBarry Smith 
3365a5399872SJunchao Zhang   selective 64-bit: with the default MUMPS build, 64-bit integers have been introduced where needed. In compressed sparse row (CSR) storage of matrices,
3366a5399872SJunchao Zhang   MUMPS stores column indices in 32-bit, but row offsets in 64-bit, so you can have a huge number of non-zeros, but must have less than 2^31 rows and
3367a5399872SJunchao Zhang   columns. This can lead to significant memory and performance gains with respect to a full 64-bit integer MUMPS version. This requires a regular (32-bit
3368a5399872SJunchao Zhang   integer) build of all dependent libraries MPI, ScaLAPACK, LAPACK and BLAS.
3369a5399872SJunchao Zhang 
3370a5399872SJunchao Zhang   With --download-mumps=1, PETSc always build MUMPS in selective 64-bit mode, which can be used by both --with-64-bit-indices=0/1 variants of PETSc.
3371a5399872SJunchao Zhang 
3372a5399872SJunchao Zhang   Two modes to run MUMPS/PETSc with OpenMP
33732ef1f0ffSBarry Smith .vb
33740b4b7b1cSBarry Smith    Set `OMP_NUM_THREADS` and run with fewer MPI ranks than cores. For example, if you want to have 16 OpenMP
33750b4b7b1cSBarry Smith    threads per rank, then you may use "export `OMP_NUM_THREADS` = 16 && mpirun -n 4 ./test".
33762ef1f0ffSBarry Smith .ve
33778fcaa860SBarry Smith 
33782ef1f0ffSBarry Smith .vb
33790b4b7b1cSBarry Smith    `-mat_mumps_use_omp_threads` [m] and run your code with as many MPI ranks as the number of cores. For example,
33802ef1f0ffSBarry Smith    if a compute node has 32 cores and you run on two nodes, you may use "mpirun -n 64 ./test -mat_mumps_use_omp_threads 16"
33812ef1f0ffSBarry Smith .ve
33828fcaa860SBarry Smith 
33838fcaa860SBarry Smith    To run MUMPS in MPI+OpenMP hybrid mode (i.e., enable multithreading in MUMPS), but still run the non-MUMPS part
33842ef1f0ffSBarry Smith    (i.e., PETSc part) of your code in the so-called flat-MPI (aka pure-MPI) mode, you need to configure PETSc with `--with-openmp` `--download-hwloc`
33852ef1f0ffSBarry Smith    (or `--with-hwloc`), and have an MPI that supports MPI-3.0's process shared memory (which is usually available). Since MUMPS calls BLAS
33868fcaa860SBarry Smith    libraries, to really get performance, you should have multithreaded BLAS libraries such as Intel MKL, AMD ACML, Cray libSci or OpenBLAS
33870b4b7b1cSBarry Smith    (PETSc will automatically try to utilized a threaded BLAS if `--with-openmp` is provided).
3388217d3b1eSJunchao Zhang 
33898fcaa860SBarry Smith    If you run your code through a job submission system, there are caveats in MPI rank mapping. We use MPI_Comm_split_type() to obtain MPI
3390217d3b1eSJunchao Zhang    processes on each compute node. Listing the processes in rank ascending order, we split processes on a node into consecutive groups of
3391217d3b1eSJunchao Zhang    size m and create a communicator called omp_comm for each group. Rank 0 in an omp_comm is called the master rank, and others in the omp_comm
3392217d3b1eSJunchao Zhang    are called slave ranks (or slaves). Only master ranks are seen to MUMPS and slaves are not. We will free CPUs assigned to slaves (might be set
3393217d3b1eSJunchao Zhang    by CPU binding policies in job scripts) and make the CPUs available to the master so that OMP threads spawned by MUMPS can run on the CPUs.
3394217d3b1eSJunchao Zhang    In a multi-socket compute node, MPI rank mapping is an issue. Still use the above example and suppose your compute node has two sockets,
3395217d3b1eSJunchao Zhang    if you interleave MPI ranks on the two sockets, in other words, even ranks are placed on socket 0, and odd ranks are on socket 1, and bind
33960b4b7b1cSBarry Smith    MPI ranks to cores, then with `-mat_mumps_use_omp_threads` 16, a master rank (and threads it spawns) will use half cores in socket 0, and half
3397217d3b1eSJunchao Zhang    cores in socket 1, that definitely hurts locality. On the other hand, if you map MPI ranks consecutively on the two sockets, then the
33980b4b7b1cSBarry Smith    problem will not happen. Therefore, when you use `-mat_mumps_use_omp_threads`, you need to keep an eye on your MPI rank mapping and CPU binding.
33990b4b7b1cSBarry Smith    For example, with the Slurm job scheduler, one can use srun `--cpu-bind`=verbose -m block:block to map consecutive MPI ranks to sockets and
3400217d3b1eSJunchao Zhang    examine the mapping result.
3401217d3b1eSJunchao Zhang 
340211a5261eSBarry Smith    PETSc does not control thread binding in MUMPS. So to get best performance, one still has to set `OMP_PROC_BIND` and `OMP_PLACES` in job scripts,
340311a5261eSBarry Smith    for example, export `OMP_PLACES`=threads and export `OMP_PROC_BIND`=spread. One does not need to export `OMP_NUM_THREADS`=m in job scripts as PETSc
340411a5261eSBarry Smith    calls `omp_set_num_threads`(m) internally before calling MUMPS.
3405217d3b1eSJunchao Zhang 
34061d27aa22SBarry Smith    See {cite}`heroux2011bi` and {cite}`gutierrez2017accommodating`
3407217d3b1eSJunchao Zhang 
34081cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `PCFactorSetMatSolverType()`, `MatSolverType`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`, `KSPGetPC()`, `PCFactorGetMatrix()`
340924b6179bSKris Buschelman M*/
341024b6179bSKris Buschelman 
3411d2a308c1SPierre Jolivet static PetscErrorCode MatFactorGetSolverType_mumps(PETSC_UNUSED Mat A, MatSolverType *type)
3412d71ae5a4SJacob Faibussowitsch {
341335bd34faSBarry Smith   PetscFunctionBegin;
34142692d6eeSBarry Smith   *type = MATSOLVERMUMPS;
34153ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
341635bd34faSBarry Smith }
341735bd34faSBarry Smith 
3418bccb9932SShri Abhyankar /* MatGetFactor for Seq and MPI AIJ matrices */
3419d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_aij_mumps(Mat A, MatFactorType ftype, Mat *F)
3420d71ae5a4SJacob Faibussowitsch {
34212877fffaSHong Zhang   Mat         B;
34222877fffaSHong Zhang   Mat_MUMPS  *mumps;
34234b9405b2SPierre Jolivet   PetscBool   isSeqAIJ, isDiag, isDense;
34242c7c0729SBarry Smith   PetscMPIInt size;
34252877fffaSHong Zhang 
34262877fffaSHong Zhang   PetscFunctionBegin;
3427eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
342803e5aca4SStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) {
342903e5aca4SStefano Zampini     PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n"));
343003e5aca4SStefano Zampini     *F = NULL;
343103e5aca4SStefano Zampini     PetscFunctionReturn(PETSC_SUCCESS);
343203e5aca4SStefano Zampini   }
3433eb1ec7c1SStefano Zampini #endif
34342877fffaSHong Zhang   /* Create the factorization matrix */
34359566063dSJacob Faibussowitsch   PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATSEQAIJ, &isSeqAIJ));
3436c3e1b152SPierre Jolivet   PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATDIAGONAL, &isDiag));
34374b9405b2SPierre Jolivet   PetscCall(PetscObjectTypeCompareAny((PetscObject)A, &isDense, MATSEQDENSE, MATMPIDENSE, NULL));
34389566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
34399566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
3440d2a308c1SPierre Jolivet   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &((PetscObject)B)->type_name));
34419566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
34422877fffaSHong Zhang 
34434dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
34442205254eSKarl Rupp 
34452877fffaSHong Zhang   B->ops->view    = MatView_MUMPS;
344635bd34faSBarry Smith   B->ops->getinfo = MatGetInfo_MUMPS;
34472205254eSKarl Rupp 
34489566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
34499566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
34509566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
34519566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
34529566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
34539566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
34549566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
34559566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
34569566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
34579566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
34589566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
34595c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
34609566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
34619566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
34626444a565SStefano Zampini 
3463450b117fSShri Abhyankar   if (ftype == MAT_FACTOR_LU) {
3464450b117fSShri Abhyankar     B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS;
3465d5f3da31SBarry Smith     B->factortype            = MAT_FACTOR_LU;
3466bccb9932SShri Abhyankar     if (isSeqAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqaij_seqaij;
3467c3e1b152SPierre Jolivet     else if (isDiag) mumps->ConvertToTriples = MatConvertToTriples_diagonal_xaij;
34684b9405b2SPierre Jolivet     else if (isDense) mumps->ConvertToTriples = MatConvertToTriples_dense_xaij;
3469bccb9932SShri Abhyankar     else mumps->ConvertToTriples = MatConvertToTriples_mpiaij_mpiaij;
34709566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU]));
3471746480a1SHong Zhang     mumps->sym = 0;
3472dcd589f8SShri Abhyankar   } else {
347367877ebaSShri Abhyankar     B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS;
3474450b117fSShri Abhyankar     B->factortype                  = MAT_FACTOR_CHOLESKY;
3475bccb9932SShri Abhyankar     if (isSeqAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqaij_seqsbaij;
3476c3e1b152SPierre Jolivet     else if (isDiag) mumps->ConvertToTriples = MatConvertToTriples_diagonal_xaij;
34774b9405b2SPierre Jolivet     else if (isDense) mumps->ConvertToTriples = MatConvertToTriples_dense_xaij;
3478bccb9932SShri Abhyankar     else mumps->ConvertToTriples = MatConvertToTriples_mpiaij_mpisbaij;
34799566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_CHOLESKY]));
348059ac8732SStefano Zampini #if defined(PETSC_USE_COMPLEX)
348159ac8732SStefano Zampini     mumps->sym = 2;
348259ac8732SStefano Zampini #else
3483b94d7dedSBarry Smith     if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1;
34846fdc2a6dSBarry Smith     else mumps->sym = 2;
348559ac8732SStefano Zampini #endif
3486450b117fSShri Abhyankar   }
34872877fffaSHong Zhang 
348800c67f3bSHong Zhang   /* set solvertype */
34899566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
34909566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
34919566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
34922c7c0729SBarry Smith   if (size == 1) {
34934ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
3494f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
34952c7c0729SBarry Smith   }
34962877fffaSHong Zhang   B->ops->destroy = MatDestroy_MUMPS;
3497e69c285eSBarry Smith   B->data         = (void *)mumps;
34982205254eSKarl Rupp 
34992877fffaSHong Zhang   *F               = B;
3500413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3501413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3502413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3503d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
35043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
35052877fffaSHong Zhang }
35062877fffaSHong Zhang 
3507bccb9932SShri Abhyankar /* MatGetFactor for Seq and MPI SBAIJ matrices */
3508d2a308c1SPierre Jolivet static PetscErrorCode MatGetFactor_sbaij_mumps(Mat A, PETSC_UNUSED MatFactorType ftype, Mat *F)
3509d71ae5a4SJacob Faibussowitsch {
35102877fffaSHong Zhang   Mat         B;
35112877fffaSHong Zhang   Mat_MUMPS  *mumps;
3512ace3abfcSBarry Smith   PetscBool   isSeqSBAIJ;
35132c7c0729SBarry Smith   PetscMPIInt size;
35142877fffaSHong Zhang 
35152877fffaSHong Zhang   PetscFunctionBegin;
3516eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
351703e5aca4SStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) {
351803e5aca4SStefano Zampini     PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n"));
351903e5aca4SStefano Zampini     *F = NULL;
352003e5aca4SStefano Zampini     PetscFunctionReturn(PETSC_SUCCESS);
352103e5aca4SStefano Zampini   }
3522eb1ec7c1SStefano Zampini #endif
35239566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
35249566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
3525d2a308c1SPierre Jolivet   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &((PetscObject)B)->type_name));
35269566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
3527e69c285eSBarry Smith 
35284dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
35299566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQSBAIJ, &isSeqSBAIJ));
3530bccb9932SShri Abhyankar   if (isSeqSBAIJ) {
353116ebf90aSShri Abhyankar     mumps->ConvertToTriples = MatConvertToTriples_seqsbaij_seqsbaij;
3532dcd589f8SShri Abhyankar   } else {
3533bccb9932SShri Abhyankar     mumps->ConvertToTriples = MatConvertToTriples_mpisbaij_mpisbaij;
3534bccb9932SShri Abhyankar   }
3535bccb9932SShri Abhyankar 
353667877ebaSShri Abhyankar   B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS;
3537bccb9932SShri Abhyankar   B->ops->view                   = MatView_MUMPS;
3538722b6324SPierre Jolivet   B->ops->getinfo                = MatGetInfo_MUMPS;
35392205254eSKarl Rupp 
35409566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
35419566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
35429566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
35439566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
35449566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
35459566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
35469566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
35479566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
35489566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
35499566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
35509566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
35515c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
35529566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
35539566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
35542205254eSKarl Rupp 
3555f4762488SHong Zhang   B->factortype = MAT_FACTOR_CHOLESKY;
355659ac8732SStefano Zampini #if defined(PETSC_USE_COMPLEX)
355759ac8732SStefano Zampini   mumps->sym = 2;
355859ac8732SStefano Zampini #else
3559b94d7dedSBarry Smith   if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1;
35606fdc2a6dSBarry Smith   else mumps->sym = 2;
356159ac8732SStefano Zampini #endif
3562a214ac2aSShri Abhyankar 
356300c67f3bSHong Zhang   /* set solvertype */
35649566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
35659566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
35669566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
35672c7c0729SBarry Smith   if (size == 1) {
35684ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
3569f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
35702c7c0729SBarry Smith   }
35719566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_CHOLESKY]));
3572f3c0ef26SHong Zhang   B->ops->destroy = MatDestroy_MUMPS;
3573e69c285eSBarry Smith   B->data         = (void *)mumps;
35742205254eSKarl Rupp 
35752877fffaSHong Zhang   *F               = B;
3576413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3577413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3578413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3579d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
35803ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
35812877fffaSHong Zhang }
358297969023SHong Zhang 
3583d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_baij_mumps(Mat A, MatFactorType ftype, Mat *F)
3584d71ae5a4SJacob Faibussowitsch {
358567877ebaSShri Abhyankar   Mat         B;
358667877ebaSShri Abhyankar   Mat_MUMPS  *mumps;
3587ace3abfcSBarry Smith   PetscBool   isSeqBAIJ;
35882c7c0729SBarry Smith   PetscMPIInt size;
358967877ebaSShri Abhyankar 
359067877ebaSShri Abhyankar   PetscFunctionBegin;
359167877ebaSShri Abhyankar   /* Create the factorization matrix */
35929566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQBAIJ, &isSeqBAIJ));
35939566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
35949566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
3595d2a308c1SPierre Jolivet   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &((PetscObject)B)->type_name));
35969566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
3597450b117fSShri Abhyankar 
35984dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
3599450b117fSShri Abhyankar   if (ftype == MAT_FACTOR_LU) {
3600450b117fSShri Abhyankar     B->ops->lufactorsymbolic = MatLUFactorSymbolic_BAIJMUMPS;
3601450b117fSShri Abhyankar     B->factortype            = MAT_FACTOR_LU;
3602bccb9932SShri Abhyankar     if (isSeqBAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqbaij_seqaij;
3603bccb9932SShri Abhyankar     else mumps->ConvertToTriples = MatConvertToTriples_mpibaij_mpiaij;
3604746480a1SHong Zhang     mumps->sym = 0;
36059566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU]));
3606546078acSJacob Faibussowitsch   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot use PETSc BAIJ matrices with MUMPS Cholesky, use SBAIJ or AIJ matrix instead");
3607bccb9932SShri Abhyankar 
3608450b117fSShri Abhyankar   B->ops->view    = MatView_MUMPS;
3609722b6324SPierre Jolivet   B->ops->getinfo = MatGetInfo_MUMPS;
36102205254eSKarl Rupp 
36119566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
36129566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
36139566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
36149566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
36159566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
36169566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
36179566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
36189566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
36199566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
36209566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
36219566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
36225c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
36239566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
36249566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
3625450b117fSShri Abhyankar 
362600c67f3bSHong Zhang   /* set solvertype */
36279566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
36289566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
36299566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
36302c7c0729SBarry Smith   if (size == 1) {
36314ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
3632f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
36332c7c0729SBarry Smith   }
36347ee00b23SStefano Zampini   B->ops->destroy = MatDestroy_MUMPS;
36357ee00b23SStefano Zampini   B->data         = (void *)mumps;
36367ee00b23SStefano Zampini 
36377ee00b23SStefano Zampini   *F               = B;
3638413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3639413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3640413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3641d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
36423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
36437ee00b23SStefano Zampini }
36447ee00b23SStefano Zampini 
36457ee00b23SStefano Zampini /* MatGetFactor for Seq and MPI SELL matrices */
3646d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_sell_mumps(Mat A, MatFactorType ftype, Mat *F)
3647d71ae5a4SJacob Faibussowitsch {
36487ee00b23SStefano Zampini   Mat         B;
36497ee00b23SStefano Zampini   Mat_MUMPS  *mumps;
36507ee00b23SStefano Zampini   PetscBool   isSeqSELL;
36512c7c0729SBarry Smith   PetscMPIInt size;
36527ee00b23SStefano Zampini 
36537ee00b23SStefano Zampini   PetscFunctionBegin;
36547ee00b23SStefano Zampini   /* Create the factorization matrix */
36559566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQSELL, &isSeqSELL));
36569566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
36579566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
3658d2a308c1SPierre Jolivet   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &((PetscObject)B)->type_name));
36599566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
36607ee00b23SStefano Zampini 
36614dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
36627ee00b23SStefano Zampini 
36637ee00b23SStefano Zampini   B->ops->view    = MatView_MUMPS;
36647ee00b23SStefano Zampini   B->ops->getinfo = MatGetInfo_MUMPS;
36657ee00b23SStefano Zampini 
36669566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
36679566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
36689566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
36699566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
36709566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
36719566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
36729566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
36739566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
36749566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
36759566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
36769566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
36775c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
36787ee00b23SStefano Zampini 
36797ee00b23SStefano Zampini   if (ftype == MAT_FACTOR_LU) {
36807ee00b23SStefano Zampini     B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS;
36817ee00b23SStefano Zampini     B->factortype            = MAT_FACTOR_LU;
36827ee00b23SStefano Zampini     if (isSeqSELL) mumps->ConvertToTriples = MatConvertToTriples_seqsell_seqaij;
36837ee00b23SStefano Zampini     else SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "To be implemented");
36847ee00b23SStefano Zampini     mumps->sym = 0;
36859566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU]));
36867ee00b23SStefano Zampini   } else SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "To be implemented");
36877ee00b23SStefano Zampini 
36887ee00b23SStefano Zampini   /* set solvertype */
36899566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
36909566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
36919566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
36922c7c0729SBarry Smith   if (size == 1) {
36934ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization  */
3694f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
36952c7c0729SBarry Smith   }
3696450b117fSShri Abhyankar   B->ops->destroy = MatDestroy_MUMPS;
3697e69c285eSBarry Smith   B->data         = (void *)mumps;
36982205254eSKarl Rupp 
3699450b117fSShri Abhyankar   *F               = B;
3700413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3701413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3702413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3703d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
37043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3705450b117fSShri Abhyankar }
370642c9c57cSBarry Smith 
37079d0448ceSStefano Zampini /* MatGetFactor for MATNEST matrices */
37089d0448ceSStefano Zampini static PetscErrorCode MatGetFactor_nest_mumps(Mat A, MatFactorType ftype, Mat *F)
37099d0448ceSStefano Zampini {
37109d0448ceSStefano Zampini   Mat         B, **mats;
37119d0448ceSStefano Zampini   Mat_MUMPS  *mumps;
37129d0448ceSStefano Zampini   PetscInt    nr, nc;
37139d0448ceSStefano Zampini   PetscMPIInt size;
371403e5aca4SStefano Zampini   PetscBool   flg = PETSC_TRUE;
37159d0448ceSStefano Zampini 
37169d0448ceSStefano Zampini   PetscFunctionBegin;
37179d0448ceSStefano Zampini #if defined(PETSC_USE_COMPLEX)
371803e5aca4SStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) {
371903e5aca4SStefano Zampini     PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n"));
372003e5aca4SStefano Zampini     *F = NULL;
372103e5aca4SStefano Zampini     PetscFunctionReturn(PETSC_SUCCESS);
372203e5aca4SStefano Zampini   }
37239d0448ceSStefano Zampini #endif
37249d0448ceSStefano Zampini 
372503e5aca4SStefano Zampini   /* Return if some condition is not satisfied */
372603e5aca4SStefano Zampini   *F = NULL;
37279d0448ceSStefano Zampini   PetscCall(MatNestGetSubMats(A, &nr, &nc, &mats));
37289d0448ceSStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY) {
37299d0448ceSStefano Zampini     IS       *rows, *cols;
37309d0448ceSStefano Zampini     PetscInt *m, *M;
37319d0448ceSStefano Zampini 
37329d0448ceSStefano Zampini     PetscCheck(nr == nc, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MAT_FACTOR_CHOLESKY not supported for nest sizes %" PetscInt_FMT " != %" PetscInt_FMT ". Use MAT_FACTOR_LU.", nr, nc);
37339d0448ceSStefano Zampini     PetscCall(PetscMalloc2(nr, &rows, nc, &cols));
37349d0448ceSStefano Zampini     PetscCall(MatNestGetISs(A, rows, cols));
37359d0448ceSStefano Zampini     for (PetscInt r = 0; flg && r < nr; r++) PetscCall(ISEqualUnsorted(rows[r], cols[r], &flg));
373603e5aca4SStefano Zampini     if (!flg) {
373703e5aca4SStefano Zampini       PetscCall(PetscFree2(rows, cols));
373803e5aca4SStefano Zampini       PetscCall(PetscInfo(A, "MAT_FACTOR_CHOLESKY not supported for unequal row and column maps. Use MAT_FACTOR_LU.\n"));
373903e5aca4SStefano Zampini       PetscFunctionReturn(PETSC_SUCCESS);
374003e5aca4SStefano Zampini     }
37419d0448ceSStefano Zampini     PetscCall(PetscMalloc2(nr, &m, nr, &M));
37429d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) PetscCall(ISGetMinMax(rows[r], &m[r], &M[r]));
37439d0448ceSStefano Zampini     for (PetscInt r = 0; flg && r < nr; r++)
37449d0448ceSStefano Zampini       for (PetscInt k = r + 1; flg && k < nr; k++)
37459d0448ceSStefano Zampini         if ((m[k] <= m[r] && m[r] <= M[k]) || (m[k] <= M[r] && M[r] <= M[k])) flg = PETSC_FALSE;
37469d0448ceSStefano Zampini     PetscCall(PetscFree2(m, M));
37479d0448ceSStefano Zampini     PetscCall(PetscFree2(rows, cols));
374803e5aca4SStefano Zampini     if (!flg) {
374903e5aca4SStefano Zampini       PetscCall(PetscInfo(A, "MAT_FACTOR_CHOLESKY not supported for intersecting row maps. Use MAT_FACTOR_LU.\n"));
375003e5aca4SStefano Zampini       PetscFunctionReturn(PETSC_SUCCESS);
375103e5aca4SStefano Zampini     }
37529d0448ceSStefano Zampini   }
37539d0448ceSStefano Zampini 
37549d0448ceSStefano Zampini   for (PetscInt r = 0; r < nr; r++) {
37559d0448ceSStefano Zampini     for (PetscInt c = 0; c < nc; c++) {
37569d0448ceSStefano Zampini       Mat       sub = mats[r][c];
375753587d93SPierre Jolivet       PetscBool isSeqAIJ, isMPIAIJ, isSeqBAIJ, isMPIBAIJ, isSeqSBAIJ, isMPISBAIJ, isDiag, isDense;
37589d0448ceSStefano Zampini 
37599d0448ceSStefano Zampini       if (!sub || (ftype == MAT_FACTOR_CHOLESKY && c < r)) continue;
376053587d93SPierre Jolivet       PetscCall(MatGetTranspose_TransposeVirtual(&sub, NULL, NULL, NULL, NULL));
37619d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQAIJ, &isSeqAIJ));
37629d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIAIJ, &isMPIAIJ));
37639d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQBAIJ, &isSeqBAIJ));
37649d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIBAIJ, &isMPIBAIJ));
37659d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQSBAIJ, &isSeqSBAIJ));
37669d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPISBAIJ, &isMPISBAIJ));
3767c3e1b152SPierre Jolivet       PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATDIAGONAL, &isDiag));
37684b9405b2SPierre Jolivet       PetscCall(PetscObjectTypeCompareAny((PetscObject)sub, &isDense, MATSEQDENSE, MATMPIDENSE, NULL));
37699d0448ceSStefano Zampini       if (ftype == MAT_FACTOR_CHOLESKY) {
3770dcab004fSPierre Jolivet         if (r == c) {
37714b9405b2SPierre Jolivet           if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isSeqSBAIJ && !isMPISBAIJ && !isDiag && !isDense) {
377240afc089SBarry Smith             PetscCall(PetscInfo(sub, "MAT_FACTOR_CHOLESKY not supported for diagonal block of type %s.\n", ((PetscObject)sub)->type_name));
377303e5aca4SStefano Zampini             flg = PETSC_FALSE;
3774dcab004fSPierre Jolivet           }
37754b9405b2SPierre Jolivet         } else if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isDiag && !isDense) {
377640afc089SBarry Smith           PetscCall(PetscInfo(sub, "MAT_FACTOR_CHOLESKY not supported for off-diagonal block of type %s.\n", ((PetscObject)sub)->type_name));
377703e5aca4SStefano Zampini           flg = PETSC_FALSE;
377803e5aca4SStefano Zampini         }
37794b9405b2SPierre Jolivet       } else if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isDiag && !isDense) {
37809afb9c56SPierre Jolivet         PetscCall(PetscInfo(sub, "MAT_FACTOR_LU not supported for block of type %s.\n", ((PetscObject)sub)->type_name));
378103e5aca4SStefano Zampini         flg = PETSC_FALSE;
37829d0448ceSStefano Zampini       }
37839d0448ceSStefano Zampini     }
378403e5aca4SStefano Zampini   }
378503e5aca4SStefano Zampini   if (!flg) PetscFunctionReturn(PETSC_SUCCESS);
37869d0448ceSStefano Zampini 
37879d0448ceSStefano Zampini   /* Create the factorization matrix */
37889d0448ceSStefano Zampini   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
37899d0448ceSStefano Zampini   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
37909d0448ceSStefano Zampini   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &((PetscObject)B)->type_name));
37919d0448ceSStefano Zampini   PetscCall(MatSetUp(B));
37929d0448ceSStefano Zampini 
37939d0448ceSStefano Zampini   PetscCall(PetscNew(&mumps));
37949d0448ceSStefano Zampini 
37959d0448ceSStefano Zampini   B->ops->view    = MatView_MUMPS;
37969d0448ceSStefano Zampini   B->ops->getinfo = MatGetInfo_MUMPS;
37979d0448ceSStefano Zampini 
37989d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
37999d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
38009d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
38019d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
38029d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
38039d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
38049d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
38059d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
38069d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
38079d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
38089d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
38099d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
38109d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
38119d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
38129d0448ceSStefano Zampini 
38139d0448ceSStefano Zampini   if (ftype == MAT_FACTOR_LU) {
38149d0448ceSStefano Zampini     B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS;
38159d0448ceSStefano Zampini     B->factortype            = MAT_FACTOR_LU;
38169d0448ceSStefano Zampini     mumps->sym               = 0;
38179d0448ceSStefano Zampini   } else {
38189d0448ceSStefano Zampini     B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS;
38199d0448ceSStefano Zampini     B->factortype                  = MAT_FACTOR_CHOLESKY;
38209d0448ceSStefano Zampini #if defined(PETSC_USE_COMPLEX)
38219d0448ceSStefano Zampini     mumps->sym = 2;
38229d0448ceSStefano Zampini #else
38239d0448ceSStefano Zampini     if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1;
38249d0448ceSStefano Zampini     else mumps->sym = 2;
38259d0448ceSStefano Zampini #endif
38269d0448ceSStefano Zampini   }
38279d0448ceSStefano Zampini   mumps->ConvertToTriples = MatConvertToTriples_nest_xaij;
38289d0448ceSStefano Zampini   PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[ftype]));
38299d0448ceSStefano Zampini 
38309d0448ceSStefano Zampini   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
38319d0448ceSStefano Zampini   if (size == 1) {
38329d0448ceSStefano Zampini     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
38339d0448ceSStefano Zampini     B->canuseordering = PETSC_TRUE;
38349d0448ceSStefano Zampini   }
38359d0448ceSStefano Zampini 
38369d0448ceSStefano Zampini   /* set solvertype */
38379d0448ceSStefano Zampini   PetscCall(PetscFree(B->solvertype));
38389d0448ceSStefano Zampini   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
38399d0448ceSStefano Zampini   B->ops->destroy = MatDestroy_MUMPS;
38409d0448ceSStefano Zampini   B->data         = (void *)mumps;
38419d0448ceSStefano Zampini 
38429d0448ceSStefano Zampini   *F               = B;
38439d0448ceSStefano Zampini   mumps->id.job    = JOB_NULL;
38449d0448ceSStefano Zampini   mumps->ICNTL_pre = NULL;
38459d0448ceSStefano Zampini   mumps->CNTL_pre  = NULL;
38469d0448ceSStefano Zampini   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
38479d0448ceSStefano Zampini   PetscFunctionReturn(PETSC_SUCCESS);
38489d0448ceSStefano Zampini }
38499d0448ceSStefano Zampini 
3850d1f0640dSPierre Jolivet PETSC_INTERN PetscErrorCode MatSolverTypeRegister_MUMPS(void)
3851d71ae5a4SJacob Faibussowitsch {
385242c9c57cSBarry Smith   PetscFunctionBegin;
38539566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIAIJ, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
38549566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
38559566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIBAIJ, MAT_FACTOR_LU, MatGetFactor_baij_mumps));
38569566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_baij_mumps));
38579566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPISBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_sbaij_mumps));
38589566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQAIJ, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
38599566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
38609566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQBAIJ, MAT_FACTOR_LU, MatGetFactor_baij_mumps));
38619566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_baij_mumps));
38629566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQSBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_sbaij_mumps));
38639566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQSELL, MAT_FACTOR_LU, MatGetFactor_sell_mumps));
3864c3e1b152SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATDIAGONAL, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
3865c3e1b152SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATDIAGONAL, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
38664b9405b2SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQDENSE, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
38674b9405b2SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQDENSE, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
38684b9405b2SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIDENSE, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
38694b9405b2SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIDENSE, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
38709d0448ceSStefano Zampini   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATNEST, MAT_FACTOR_LU, MatGetFactor_nest_mumps));
38719d0448ceSStefano Zampini   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATNEST, MAT_FACTOR_CHOLESKY, MatGetFactor_nest_mumps));
38723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
387342c9c57cSBarry Smith }
3874