xref: /petsc/src/mat/impls/aij/mpi/mumps/mumps.c (revision 4b9405b21326eab36312ca520dd9eb1b4b9102ab)
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 */
57a6053eceSJunchao Zhang     #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 */
6167602552SJunchao Zhang     #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*/
70d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PetscMUMPSIntCast(PetscInt 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
76a6053eceSJunchao Zhang   *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 */
81d71ae5a4SJacob Faibussowitsch 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;
85a6053eceSJunchao Zhang   PetscFunctionBegin;
86a6053eceSJunchao Zhang   /* PetscInt's size should be always >= PetscMUMPSInt's. It is safe to call PetscOptionsInt_Private to read a PetscMUMPSInt */
879566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt_Private(PetscOptionsObject, opt, text, man, (PetscInt)currentvalue, &myval, &myset, lb, ub));
889566063dSJacob Faibussowitsch   if (myset) PetscCall(PetscMUMPSIntCast(myval, value));
89a6053eceSJunchao Zhang   if (set) *set = myset;
903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
91a6053eceSJunchao Zhang }
92a6053eceSJunchao 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)
93a6053eceSJunchao Zhang 
94217d3b1eSJunchao 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 */
953ab56b82SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
963ab56b82SJunchao Zhang   #define PetscMUMPS_c(mumps) \
973ab56b82SJunchao Zhang     do { \
983ab56b82SJunchao Zhang       if (mumps->use_petsc_omp_support) { \
993ab56b82SJunchao Zhang         if (mumps->is_omp_master) { \
1009566063dSJacob Faibussowitsch           PetscCall(PetscOmpCtrlOmpRegionOnMasterBegin(mumps->omp_ctrl)); \
10114ffdc6fSStefano Zampini           PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); \
10214ffdc6fSStefano Zampini           PetscStackCallExternalVoid(PetscStringize(MUMPS_c), MUMPS_c(&mumps->id)); \
10314ffdc6fSStefano Zampini           PetscCall(PetscFPTrapPop()); \
1049566063dSJacob Faibussowitsch           PetscCall(PetscOmpCtrlOmpRegionOnMasterEnd(mumps->omp_ctrl)); \
1053ab56b82SJunchao Zhang         } \
1069566063dSJacob Faibussowitsch         PetscCall(PetscOmpCtrlBarrier(mumps->omp_ctrl)); \
107c3714a1dSJunchao Zhang         /* Global info is same on all processes so we Bcast it within omp_comm. Local info is specific      \
108c3714a1dSJunchao Zhang          to processes, so we only Bcast info[1], an error code and leave others (since they do not have   \
109c3714a1dSJunchao Zhang          an easy translation between omp_comm and petsc_comm). See MUMPS-5.1.2 manual p82.                   \
110c3714a1dSJunchao Zhang          omp_comm is a small shared memory communicator, hence doing multiple Bcast as shown below is OK. \
111c3714a1dSJunchao Zhang       */ \
112338d3105SPierre Jolivet         PetscCallMPI(MPI_Bcast(mumps->id.infog, PETSC_STATIC_ARRAY_LENGTH(mumps->id.infog), MPIU_MUMPSINT, 0, mumps->omp_comm)); \
113338d3105SPierre Jolivet         PetscCallMPI(MPI_Bcast(mumps->id.rinfog, PETSC_STATIC_ARRAY_LENGTH(mumps->id.rinfog), MPIU_REAL, 0, mumps->omp_comm)); \
114338d3105SPierre Jolivet         PetscCallMPI(MPI_Bcast(mumps->id.info, PETSC_STATIC_ARRAY_LENGTH(mumps->id.info), MPIU_MUMPSINT, 0, mumps->omp_comm)); \
115338d3105SPierre Jolivet         PetscCallMPI(MPI_Bcast(mumps->id.rinfo, PETSC_STATIC_ARRAY_LENGTH(mumps->id.rinfo), MPIU_REAL, 0, mumps->omp_comm)); \
1163ab56b82SJunchao Zhang       } else { \
11714ffdc6fSStefano Zampini         PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); \
11814ffdc6fSStefano Zampini         PetscStackCallExternalVoid(PetscStringize(MUMPS_c), MUMPS_c(&mumps->id)); \
11914ffdc6fSStefano Zampini         PetscCall(PetscFPTrapPop()); \
1203ab56b82SJunchao Zhang       } \
1213ab56b82SJunchao Zhang     } while (0)
1223ab56b82SJunchao Zhang #else
1233ab56b82SJunchao Zhang   #define PetscMUMPS_c(mumps) \
124d71ae5a4SJacob Faibussowitsch     do { \
12514ffdc6fSStefano Zampini       PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); \
12614ffdc6fSStefano Zampini       PetscStackCallExternalVoid(PetscStringize(MUMPS_c), MUMPS_c(&mumps->id)); \
12714ffdc6fSStefano Zampini       PetscCall(PetscFPTrapPop()); \
128d71ae5a4SJacob Faibussowitsch     } while (0)
1293ab56b82SJunchao Zhang #endif
1303ab56b82SJunchao Zhang 
131940cd9d6SSatish Balay /* declare MumpsScalar */
132940cd9d6SSatish Balay #if defined(PETSC_USE_COMPLEX)
133940cd9d6SSatish Balay   #if defined(PETSC_USE_REAL_SINGLE)
134940cd9d6SSatish Balay     #define MumpsScalar mumps_complex
135940cd9d6SSatish Balay   #else
136940cd9d6SSatish Balay     #define MumpsScalar mumps_double_complex
137940cd9d6SSatish Balay   #endif
138940cd9d6SSatish Balay #else
139940cd9d6SSatish Balay   #define MumpsScalar PetscScalar
140940cd9d6SSatish Balay #endif
1413d472b54SHong Zhang 
142397b6df1SKris Buschelman /* macros s.t. indices match MUMPS documentation */
143397b6df1SKris Buschelman #define ICNTL(I)  icntl[(I)-1]
144397b6df1SKris Buschelman #define CNTL(I)   cntl[(I)-1]
145397b6df1SKris Buschelman #define INFOG(I)  infog[(I)-1]
146a7aca84bSHong Zhang #define INFO(I)   info[(I)-1]
147397b6df1SKris Buschelman #define RINFOG(I) rinfog[(I)-1]
148adc1d99fSHong Zhang #define RINFO(I)  rinfo[(I)-1]
149397b6df1SKris Buschelman 
150a6053eceSJunchao Zhang typedef struct Mat_MUMPS Mat_MUMPS;
151a6053eceSJunchao Zhang struct Mat_MUMPS {
152397b6df1SKris Buschelman #if defined(PETSC_USE_COMPLEX)
1532907cef9SHong Zhang   #if defined(PETSC_USE_REAL_SINGLE)
1542907cef9SHong Zhang   CMUMPS_STRUC_C id;
1552907cef9SHong Zhang   #else
156397b6df1SKris Buschelman   ZMUMPS_STRUC_C id;
1572907cef9SHong Zhang   #endif
1582907cef9SHong Zhang #else
1592907cef9SHong Zhang   #if defined(PETSC_USE_REAL_SINGLE)
1602907cef9SHong Zhang   SMUMPS_STRUC_C id;
161397b6df1SKris Buschelman   #else
162397b6df1SKris Buschelman   DMUMPS_STRUC_C id;
163397b6df1SKris Buschelman   #endif
1642907cef9SHong Zhang #endif
1652907cef9SHong Zhang 
166397b6df1SKris Buschelman   MatStructure   matstruc;
1672d4298aeSJunchao Zhang   PetscMPIInt    myid, petsc_size;
168a6053eceSJunchao Zhang   PetscMUMPSInt *irn, *jcn;       /* the (i,j,v) triplets passed to mumps. */
169a6053eceSJunchao 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. */
170a6053eceSJunchao Zhang   PetscInt64     nnz;             /* number of nonzeros. The type is called selective 64-bit in mumps */
171a6053eceSJunchao Zhang   PetscMUMPSInt  sym;
1722d4298aeSJunchao Zhang   MPI_Comm       mumps_comm;
173413bcc21SPierre Jolivet   PetscMUMPSInt *ICNTL_pre;
174413bcc21SPierre Jolivet   PetscReal     *CNTL_pre;
175a6053eceSJunchao Zhang   PetscMUMPSInt  ICNTL9_pre;         /* check if ICNTL(9) is changed from previous MatSolve */
176801fbe65SHong Zhang   VecScatter     scat_rhs, scat_sol; /* used by MatSolve() */
17725aac85cSJunchao Zhang   PetscMUMPSInt  ICNTL20;            /* use centralized (0) or distributed (10) dense RHS */
17867602552SJunchao Zhang   PetscMUMPSInt  lrhs_loc, nloc_rhs, *irhs_loc;
17967602552SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
18067602552SJunchao Zhang   PetscInt    *rhs_nrow, max_nrhs;
18167602552SJunchao Zhang   PetscMPIInt *rhs_recvcounts, *rhs_disps;
18267602552SJunchao Zhang   PetscScalar *rhs_loc, *rhs_recvbuf;
18367602552SJunchao Zhang #endif
184801fbe65SHong Zhang   Vec            b_seq, x_seq;
185a6053eceSJunchao Zhang   PetscInt       ninfo, *info; /* which INFO to display */
186b5fa320bSStefano Zampini   PetscInt       sizeredrhs;
18759ac8732SStefano Zampini   PetscScalar   *schur_sol;
18859ac8732SStefano Zampini   PetscInt       schur_sizesol;
189a6053eceSJunchao Zhang   PetscMUMPSInt *ia_alloc, *ja_alloc; /* work arrays used for the CSR struct for sparse rhs */
190a6053eceSJunchao Zhang   PetscInt64     cur_ilen, cur_jlen;  /* current len of ia_alloc[], ja_alloc[] */
191a6053eceSJunchao Zhang   PetscErrorCode (*ConvertToTriples)(Mat, PetscInt, MatReuse, Mat_MUMPS *);
1922205254eSKarl Rupp 
1939d0448ceSStefano Zampini   /* Support for MATNEST */
1949d0448ceSStefano Zampini   PetscErrorCode (**nest_convert_to_triples)(Mat, PetscInt, MatReuse, Mat_MUMPS *);
1959d0448ceSStefano Zampini   PetscInt64  *nest_vals_start;
1969d0448ceSStefano Zampini   PetscScalar *nest_vals;
1979d0448ceSStefano Zampini 
198a6053eceSJunchao Zhang   /* stuff used by petsc/mumps OpenMP support*/
1993ab56b82SJunchao Zhang   PetscBool    use_petsc_omp_support;
200da81f932SPierre Jolivet   PetscOmpCtrl omp_ctrl;             /* an OpenMP controller that blocked processes will release their CPU (MPI_Barrier does not have this guarantee) */
2013ab56b82SJunchao Zhang   MPI_Comm     petsc_comm, omp_comm; /* petsc_comm is petsc matrix's comm */
202a6053eceSJunchao Zhang   PetscInt64  *recvcount;            /* a collection of nnz on omp_master */
203a6053eceSJunchao Zhang   PetscMPIInt  tag, omp_comm_size;
2043ab56b82SJunchao Zhang   PetscBool    is_omp_master; /* is this rank the master of omp_comm */
205a6053eceSJunchao Zhang   MPI_Request *reqs;
206a6053eceSJunchao Zhang };
2073ab56b82SJunchao Zhang 
208a6053eceSJunchao Zhang /* Cast a 1-based CSR represented by (nrow, ia, ja) of type PetscInt to a CSR of type PetscMUMPSInt.
209a6053eceSJunchao Zhang    Here, nrow is number of rows, ia[] is row pointer and ja[] is column indices.
210a6053eceSJunchao Zhang  */
211d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscMUMPSIntCSRCast(Mat_MUMPS *mumps, PetscInt nrow, PetscInt *ia, PetscInt *ja, PetscMUMPSInt **ia_mumps, PetscMUMPSInt **ja_mumps, PetscMUMPSInt *nnz_mumps)
212d71ae5a4SJacob Faibussowitsch {
213a6053eceSJunchao Zhang   PetscInt nnz = ia[nrow] - 1; /* mumps uses 1-based indices. Uses PetscInt instead of PetscInt64 since mumps only uses PetscMUMPSInt for rhs */
214f0c56d0fSKris Buschelman 
215a6053eceSJunchao Zhang   PetscFunctionBegin;
216a6053eceSJunchao Zhang #if defined(PETSC_USE_64BIT_INDICES)
217a6053eceSJunchao Zhang   {
218a6053eceSJunchao Zhang     PetscInt i;
219a6053eceSJunchao Zhang     if (nrow + 1 > mumps->cur_ilen) { /* realloc ia_alloc/ja_alloc to fit ia/ja */
2209566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->ia_alloc));
2219566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(nrow + 1, &mumps->ia_alloc));
222a6053eceSJunchao Zhang       mumps->cur_ilen = nrow + 1;
223a6053eceSJunchao Zhang     }
224a6053eceSJunchao Zhang     if (nnz > mumps->cur_jlen) {
2259566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->ja_alloc));
2269566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(nnz, &mumps->ja_alloc));
227a6053eceSJunchao Zhang       mumps->cur_jlen = nnz;
228a6053eceSJunchao Zhang     }
2299566063dSJacob Faibussowitsch     for (i = 0; i < nrow + 1; i++) PetscCall(PetscMUMPSIntCast(ia[i], &(mumps->ia_alloc[i])));
2309566063dSJacob Faibussowitsch     for (i = 0; i < nnz; i++) PetscCall(PetscMUMPSIntCast(ja[i], &(mumps->ja_alloc[i])));
231a6053eceSJunchao Zhang     *ia_mumps = mumps->ia_alloc;
232a6053eceSJunchao Zhang     *ja_mumps = mumps->ja_alloc;
233a6053eceSJunchao Zhang   }
234a6053eceSJunchao Zhang #else
235a6053eceSJunchao Zhang   *ia_mumps = ia;
236a6053eceSJunchao Zhang   *ja_mumps = ja;
237a6053eceSJunchao Zhang #endif
2389566063dSJacob Faibussowitsch   PetscCall(PetscMUMPSIntCast(nnz, nnz_mumps));
2393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
240a6053eceSJunchao Zhang }
241b24902e0SBarry Smith 
242d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMumpsResetSchur_Private(Mat_MUMPS *mumps)
243d71ae5a4SJacob Faibussowitsch {
244b5fa320bSStefano Zampini   PetscFunctionBegin;
2459566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.listvar_schur));
2469566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.redrhs));
2479566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->schur_sol));
24859ac8732SStefano Zampini   mumps->id.size_schur = 0;
249b3cb21ddSStefano Zampini   mumps->id.schur_lld  = 0;
25059ac8732SStefano Zampini   mumps->id.ICNTL(19)  = 0;
2513ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
25259ac8732SStefano Zampini }
25359ac8732SStefano Zampini 
254b3cb21ddSStefano Zampini /* solve with rhs in mumps->id.redrhs and return in the same location */
255d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMumpsSolveSchur_Private(Mat F)
256d71ae5a4SJacob Faibussowitsch {
257b3cb21ddSStefano Zampini   Mat_MUMPS           *mumps = (Mat_MUMPS *)F->data;
258b3cb21ddSStefano Zampini   Mat                  S, B, X;
259b3cb21ddSStefano Zampini   MatFactorSchurStatus schurstatus;
260b3cb21ddSStefano Zampini   PetscInt             sizesol;
26159ac8732SStefano Zampini 
26259ac8732SStefano Zampini   PetscFunctionBegin;
2639566063dSJacob Faibussowitsch   PetscCall(MatFactorFactorizeSchurComplement(F));
2649566063dSJacob Faibussowitsch   PetscCall(MatFactorGetSchurComplement(F, &S, &schurstatus));
2659566063dSJacob Faibussowitsch   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, mumps->id.size_schur, mumps->id.nrhs, (PetscScalar *)mumps->id.redrhs, &B));
2669566063dSJacob Faibussowitsch   PetscCall(MatSetType(B, ((PetscObject)S)->type_name));
267a3d589ffSStefano Zampini #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
2689566063dSJacob Faibussowitsch   PetscCall(MatBindToCPU(B, S->boundtocpu));
269a3d589ffSStefano Zampini #endif
270b3cb21ddSStefano Zampini   switch (schurstatus) {
271d71ae5a4SJacob Faibussowitsch   case MAT_FACTOR_SCHUR_FACTORED:
272d71ae5a4SJacob Faibussowitsch     PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, mumps->id.size_schur, mumps->id.nrhs, (PetscScalar *)mumps->id.redrhs, &X));
273d71ae5a4SJacob Faibussowitsch     PetscCall(MatSetType(X, ((PetscObject)S)->type_name));
274a3d589ffSStefano Zampini #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
2759566063dSJacob Faibussowitsch     PetscCall(MatBindToCPU(X, S->boundtocpu));
276a3d589ffSStefano Zampini #endif
277b3cb21ddSStefano Zampini     if (!mumps->id.ICNTL(9)) { /* transpose solve */
2789566063dSJacob Faibussowitsch       PetscCall(MatMatSolveTranspose(S, B, X));
27959ac8732SStefano Zampini     } else {
2809566063dSJacob Faibussowitsch       PetscCall(MatMatSolve(S, B, X));
28159ac8732SStefano Zampini     }
282b3cb21ddSStefano Zampini     break;
283b3cb21ddSStefano Zampini   case MAT_FACTOR_SCHUR_INVERTED:
284b3cb21ddSStefano Zampini     sizesol = mumps->id.nrhs * mumps->id.size_schur;
28559ac8732SStefano Zampini     if (!mumps->schur_sol || sizesol > mumps->schur_sizesol) {
2869566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->schur_sol));
2879566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(sizesol, &mumps->schur_sol));
28859ac8732SStefano Zampini       mumps->schur_sizesol = sizesol;
289b5fa320bSStefano Zampini     }
2909566063dSJacob Faibussowitsch     PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, mumps->id.size_schur, mumps->id.nrhs, mumps->schur_sol, &X));
2919566063dSJacob Faibussowitsch     PetscCall(MatSetType(X, ((PetscObject)S)->type_name));
292a3d589ffSStefano Zampini #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
2939566063dSJacob Faibussowitsch     PetscCall(MatBindToCPU(X, S->boundtocpu));
294a3d589ffSStefano Zampini #endif
2959566063dSJacob Faibussowitsch     PetscCall(MatProductCreateWithMat(S, B, NULL, X));
29659ac8732SStefano Zampini     if (!mumps->id.ICNTL(9)) { /* transpose solve */
2979566063dSJacob Faibussowitsch       PetscCall(MatProductSetType(X, MATPRODUCT_AtB));
298b5fa320bSStefano Zampini     } else {
2999566063dSJacob Faibussowitsch       PetscCall(MatProductSetType(X, MATPRODUCT_AB));
300b5fa320bSStefano Zampini     }
3019566063dSJacob Faibussowitsch     PetscCall(MatProductSetFromOptions(X));
3029566063dSJacob Faibussowitsch     PetscCall(MatProductSymbolic(X));
3039566063dSJacob Faibussowitsch     PetscCall(MatProductNumeric(X));
3044417c5e8SHong Zhang 
3059566063dSJacob Faibussowitsch     PetscCall(MatCopy(X, B, SAME_NONZERO_PATTERN));
306b3cb21ddSStefano Zampini     break;
307d71ae5a4SJacob Faibussowitsch   default:
308d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)F), PETSC_ERR_SUP, "Unhandled MatFactorSchurStatus %d", F->schur_status);
30959ac8732SStefano Zampini   }
3109566063dSJacob Faibussowitsch   PetscCall(MatFactorRestoreSchurComplement(F, &S, schurstatus));
3119566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&B));
3129566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&X));
3133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
314b5fa320bSStefano Zampini }
315b5fa320bSStefano Zampini 
316d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMumpsHandleSchur_Private(Mat F, PetscBool expansion)
317d71ae5a4SJacob Faibussowitsch {
318b3cb21ddSStefano Zampini   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
319b5fa320bSStefano Zampini 
320b5fa320bSStefano Zampini   PetscFunctionBegin;
321b5fa320bSStefano Zampini   if (!mumps->id.ICNTL(19)) { /* do nothing when Schur complement has not been computed */
3223ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
323b5fa320bSStefano Zampini   }
324b8f61ee1SStefano Zampini   if (!expansion) { /* prepare for the condensation step */
325b5fa320bSStefano Zampini     PetscInt sizeredrhs = mumps->id.nrhs * mumps->id.size_schur;
326b5fa320bSStefano Zampini     /* allocate MUMPS internal array to store reduced right-hand sides */
327b5fa320bSStefano Zampini     if (!mumps->id.redrhs || sizeredrhs > mumps->sizeredrhs) {
3289566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->id.redrhs));
329b5fa320bSStefano Zampini       mumps->id.lredrhs = mumps->id.size_schur;
3309566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(mumps->id.nrhs * mumps->id.lredrhs, &mumps->id.redrhs));
331b5fa320bSStefano Zampini       mumps->sizeredrhs = mumps->id.nrhs * mumps->id.lredrhs;
332b5fa320bSStefano Zampini     }
333b5fa320bSStefano Zampini   } else { /* prepare for the expansion step */
334b8f61ee1SStefano Zampini     /* solve Schur complement (this has to be done by the MUMPS user, so basically us) */
3359566063dSJacob Faibussowitsch     PetscCall(MatMumpsSolveSchur_Private(F));
336b5fa320bSStefano Zampini     mumps->id.ICNTL(26) = 2; /* expansion phase */
3373ab56b82SJunchao Zhang     PetscMUMPS_c(mumps);
3389261f6e4SBarry 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));
339b5fa320bSStefano Zampini     /* restore defaults */
340b5fa320bSStefano Zampini     mumps->id.ICNTL(26) = -1;
341d3d598ffSStefano Zampini     /* free MUMPS internal array for redrhs if we have solved for multiple rhs in order to save memory space */
342d3d598ffSStefano Zampini     if (mumps->id.nrhs > 1) {
3439566063dSJacob Faibussowitsch       PetscCall(PetscFree(mumps->id.redrhs));
344d3d598ffSStefano Zampini       mumps->id.lredrhs = 0;
345d3d598ffSStefano Zampini       mumps->sizeredrhs = 0;
346d3d598ffSStefano Zampini     }
347b5fa320bSStefano Zampini   }
3483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
349b5fa320bSStefano Zampini }
350b5fa320bSStefano Zampini 
351397b6df1SKris Buschelman /*
352d341cd04SHong Zhang   MatConvertToTriples_A_B - convert Petsc matrix to triples: row[nz], col[nz], val[nz]
353d341cd04SHong Zhang 
354397b6df1SKris Buschelman   input:
35575480915SPierre Jolivet     A       - matrix in aij,baij or sbaij format
356397b6df1SKris Buschelman     shift   - 0: C style output triple; 1: Fortran style output triple.
357bccb9932SShri Abhyankar     reuse   - MAT_INITIAL_MATRIX: spaces are allocated and values are set for the triple
358bccb9932SShri Abhyankar               MAT_REUSE_MATRIX:   only the values in v array are updated
359397b6df1SKris Buschelman   output:
360397b6df1SKris Buschelman     nnz     - dim of r, c, and v (number of local nonzero entries of A)
361397b6df1SKris Buschelman     r, c, v - row and col index, matrix values (matrix triples)
362eb9baa12SBarry Smith 
363eb9baa12SBarry Smith   The returned values r, c, and sometimes v are obtained in a single PetscMalloc(). Then in MatDestroy_MUMPS() it is
3647ee00b23SStefano Zampini   freed with PetscFree(mumps->irn);  This is not ideal code, the fact that v is ONLY sometimes part of mumps->irn means
365eb9baa12SBarry Smith   that the PetscMalloc() cannot easily be replaced with a PetscMalloc3().
366eb9baa12SBarry Smith 
367397b6df1SKris Buschelman  */
36816ebf90aSShri Abhyankar 
36966976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqaij_seqaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
370d71ae5a4SJacob Faibussowitsch {
371a3d589ffSStefano Zampini   const PetscScalar *av;
372185f6596SHong Zhang   const PetscInt    *ai, *aj, *ajj, M = A->rmap->n;
373a6053eceSJunchao Zhang   PetscInt64         nz, rnz, i, j, k;
374a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
37516ebf90aSShri Abhyankar   Mat_SeqAIJ        *aa = (Mat_SeqAIJ *)A->data;
376397b6df1SKris Buschelman 
377397b6df1SKris Buschelman   PetscFunctionBegin;
3789566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(A, &av));
379bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
3802205254eSKarl Rupp     nz = aa->nz;
3812205254eSKarl Rupp     ai = aa->i;
3822205254eSKarl Rupp     aj = aa->j;
3839566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
384a6053eceSJunchao Zhang     for (i = k = 0; i < M; i++) {
38516ebf90aSShri Abhyankar       rnz = ai[i + 1] - ai[i];
38667877ebaSShri Abhyankar       ajj = aj + ai[i];
38767877ebaSShri Abhyankar       for (j = 0; j < rnz; j++) {
3889566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(i + shift, &row[k]));
3899566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[k]));
390a6053eceSJunchao Zhang         k++;
39116ebf90aSShri Abhyankar       }
39216ebf90aSShri Abhyankar     }
39350c845baSStefano Zampini     mumps->val = (PetscScalar *)av;
394a6053eceSJunchao Zhang     mumps->irn = row;
395a6053eceSJunchao Zhang     mumps->jcn = col;
396a6053eceSJunchao Zhang     mumps->nnz = nz;
39750c845baSStefano Zampini   } else PetscCall(PetscArraycpy(mumps->val, av, aa->nz));
3989566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(A, &av));
3993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
40016ebf90aSShri Abhyankar }
401397b6df1SKris Buschelman 
40266976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqsell_seqaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
403d71ae5a4SJacob Faibussowitsch {
404a6053eceSJunchao Zhang   PetscInt64     nz, i, j, k, r;
4057ee00b23SStefano Zampini   Mat_SeqSELL   *a = (Mat_SeqSELL *)A->data;
406a6053eceSJunchao Zhang   PetscMUMPSInt *row, *col;
4077ee00b23SStefano Zampini 
4087ee00b23SStefano Zampini   PetscFunctionBegin;
4097ee00b23SStefano Zampini   nz = a->sliidx[a->totalslices];
41050c845baSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
4119566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
412a6053eceSJunchao Zhang     for (i = k = 0; i < a->totalslices; i++) {
41348a46eb9SPierre 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++]));
4147ee00b23SStefano Zampini     }
4159566063dSJacob Faibussowitsch     for (i = 0; i < nz; i++) PetscCall(PetscMUMPSIntCast(a->colidx[i] + shift, &col[i]));
416a6053eceSJunchao Zhang     mumps->irn = row;
417a6053eceSJunchao Zhang     mumps->jcn = col;
418a6053eceSJunchao Zhang     mumps->nnz = nz;
41950c845baSStefano Zampini     mumps->val = a->val;
42050c845baSStefano Zampini   } else PetscCall(PetscArraycpy(mumps->val, a->val, nz));
4213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4227ee00b23SStefano Zampini }
4237ee00b23SStefano Zampini 
42466976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqbaij_seqaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
425d71ae5a4SJacob Faibussowitsch {
42667877ebaSShri Abhyankar   Mat_SeqBAIJ    *aa = (Mat_SeqBAIJ *)A->data;
42733d57670SJed Brown   const PetscInt *ai, *aj, *ajj, bs2 = aa->bs2;
42850c845baSStefano Zampini   PetscInt64      M, nz = bs2 * aa->nz, idx = 0, rnz, i, j, k, m;
429a6053eceSJunchao Zhang   PetscInt        bs;
430a6053eceSJunchao Zhang   PetscMUMPSInt  *row, *col;
43167877ebaSShri Abhyankar 
43267877ebaSShri Abhyankar   PetscFunctionBegin;
43350c845baSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
4349566063dSJacob Faibussowitsch     PetscCall(MatGetBlockSize(A, &bs));
43533d57670SJed Brown     M  = A->rmap->N / bs;
4369371c9d4SSatish Balay     ai = aa->i;
4379371c9d4SSatish Balay     aj = aa->j;
4389566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
43967877ebaSShri Abhyankar     for (i = 0; i < M; i++) {
44067877ebaSShri Abhyankar       ajj = aj + ai[i];
44167877ebaSShri Abhyankar       rnz = ai[i + 1] - ai[i];
44267877ebaSShri Abhyankar       for (k = 0; k < rnz; k++) {
44367877ebaSShri Abhyankar         for (j = 0; j < bs; j++) {
44467877ebaSShri Abhyankar           for (m = 0; m < bs; m++) {
4459566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(i * bs + m + shift, &row[idx]));
4469566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(bs * ajj[k] + j + shift, &col[idx]));
447a6053eceSJunchao Zhang             idx++;
44867877ebaSShri Abhyankar           }
44967877ebaSShri Abhyankar         }
45067877ebaSShri Abhyankar       }
45167877ebaSShri Abhyankar     }
452a6053eceSJunchao Zhang     mumps->irn = row;
453a6053eceSJunchao Zhang     mumps->jcn = col;
454a6053eceSJunchao Zhang     mumps->nnz = nz;
45550c845baSStefano Zampini     mumps->val = aa->a;
45650c845baSStefano Zampini   } else PetscCall(PetscArraycpy(mumps->val, aa->a, nz));
4573ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
45867877ebaSShri Abhyankar }
45967877ebaSShri Abhyankar 
46066976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqsbaij_seqsbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
461d71ae5a4SJacob Faibussowitsch {
46275480915SPierre Jolivet   const PetscInt *ai, *aj, *ajj;
463a6053eceSJunchao Zhang   PetscInt        bs;
464a6053eceSJunchao Zhang   PetscInt64      nz, rnz, i, j, k, m;
465a6053eceSJunchao Zhang   PetscMUMPSInt  *row, *col;
46675480915SPierre Jolivet   PetscScalar    *val;
46716ebf90aSShri Abhyankar   Mat_SeqSBAIJ   *aa  = (Mat_SeqSBAIJ *)A->data;
46875480915SPierre Jolivet   const PetscInt  bs2 = aa->bs2, mbs = aa->mbs;
46938548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
470b94d7dedSBarry Smith   PetscBool isset, hermitian;
47138548759SBarry Smith #endif
47216ebf90aSShri Abhyankar 
47316ebf90aSShri Abhyankar   PetscFunctionBegin;
47438548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
475b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
476b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
47738548759SBarry Smith #endif
4782205254eSKarl Rupp   ai = aa->i;
4792205254eSKarl Rupp   aj = aa->j;
4809566063dSJacob Faibussowitsch   PetscCall(MatGetBlockSize(A, &bs));
48175480915SPierre Jolivet   if (reuse == MAT_INITIAL_MATRIX) {
482f3fa974cSJacob Faibussowitsch     const PetscInt64 alloc_size = aa->nz * bs2;
483f3fa974cSJacob Faibussowitsch 
484f3fa974cSJacob Faibussowitsch     PetscCall(PetscMalloc2(alloc_size, &row, alloc_size, &col));
485a6053eceSJunchao Zhang     if (bs > 1) {
486f3fa974cSJacob Faibussowitsch       PetscCall(PetscMalloc1(alloc_size, &mumps->val_alloc));
487a6053eceSJunchao Zhang       mumps->val = mumps->val_alloc;
48875480915SPierre Jolivet     } else {
489a6053eceSJunchao Zhang       mumps->val = aa->a;
49075480915SPierre Jolivet     }
491a6053eceSJunchao Zhang     mumps->irn = row;
492a6053eceSJunchao Zhang     mumps->jcn = col;
493a6053eceSJunchao Zhang   } else {
494a6053eceSJunchao Zhang     row = mumps->irn;
495a6053eceSJunchao Zhang     col = mumps->jcn;
496a6053eceSJunchao Zhang   }
497a6053eceSJunchao Zhang   val = mumps->val;
498185f6596SHong Zhang 
49916ebf90aSShri Abhyankar   nz = 0;
500a81fe166SPierre Jolivet   if (bs > 1) {
50175480915SPierre Jolivet     for (i = 0; i < mbs; i++) {
50216ebf90aSShri Abhyankar       rnz = ai[i + 1] - ai[i];
50367877ebaSShri Abhyankar       ajj = aj + ai[i];
50475480915SPierre Jolivet       for (j = 0; j < rnz; j++) {
50575480915SPierre Jolivet         for (k = 0; k < bs; k++) {
50675480915SPierre Jolivet           for (m = 0; m < bs; m++) {
507ec4f40fdSPierre Jolivet             if (ajj[j] > i || k >= m) {
50875480915SPierre Jolivet               if (reuse == MAT_INITIAL_MATRIX) {
5099566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(i * bs + m + shift, &row[nz]));
5109566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(ajj[j] * bs + k + shift, &col[nz]));
51175480915SPierre Jolivet               }
51275480915SPierre Jolivet               val[nz++] = aa->a[(ai[i] + j) * bs2 + m + k * bs];
51375480915SPierre Jolivet             }
51475480915SPierre Jolivet           }
51575480915SPierre Jolivet         }
51675480915SPierre Jolivet       }
51775480915SPierre Jolivet     }
518a81fe166SPierre Jolivet   } else if (reuse == MAT_INITIAL_MATRIX) {
519a81fe166SPierre Jolivet     for (i = 0; i < mbs; i++) {
520a81fe166SPierre Jolivet       rnz = ai[i + 1] - ai[i];
521a81fe166SPierre Jolivet       ajj = aj + ai[i];
522a81fe166SPierre Jolivet       for (j = 0; j < rnz; j++) {
5239566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
5249566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[nz]));
525a6053eceSJunchao Zhang         nz++;
526a81fe166SPierre Jolivet       }
527a81fe166SPierre Jolivet     }
52808401ef6SPierre Jolivet     PetscCheck(nz == aa->nz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Different numbers of nonzeros %" PetscInt64_FMT " != %" PetscInt_FMT, nz, aa->nz);
52950c845baSStefano Zampini   } else PetscCall(PetscArraycpy(mumps->val, aa->a, aa->nz)); /* bs == 1 and MAT_REUSE_MATRIX */
530a6053eceSJunchao Zhang   if (reuse == MAT_INITIAL_MATRIX) mumps->nnz = nz;
5313ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
53216ebf90aSShri Abhyankar }
53316ebf90aSShri Abhyankar 
53466976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqaij_seqsbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
535d71ae5a4SJacob Faibussowitsch {
53667877ebaSShri Abhyankar   const PetscInt    *ai, *aj, *ajj, *adiag, M = A->rmap->n;
537a6053eceSJunchao Zhang   PetscInt64         nz, rnz, i, j;
53867877ebaSShri Abhyankar   const PetscScalar *av, *v1;
53916ebf90aSShri Abhyankar   PetscScalar       *val;
540a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
541829b1710SHong Zhang   Mat_SeqAIJ        *aa = (Mat_SeqAIJ *)A->data;
54229b521d4Sstefano_zampini   PetscBool          missing;
54338548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
544b94d7dedSBarry Smith   PetscBool hermitian, isset;
54538548759SBarry Smith #endif
54616ebf90aSShri Abhyankar 
54716ebf90aSShri Abhyankar   PetscFunctionBegin;
54838548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
549b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
550b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
55138548759SBarry Smith #endif
5529566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(A, &av));
5539371c9d4SSatish Balay   ai    = aa->i;
5549371c9d4SSatish Balay   aj    = aa->j;
55516ebf90aSShri Abhyankar   adiag = aa->diag;
5569566063dSJacob Faibussowitsch   PetscCall(MatMissingDiagonal_SeqAIJ(A, &missing, NULL));
557bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
5587ee00b23SStefano Zampini     /* count nz in the upper triangular part of A */
559829b1710SHong Zhang     nz = 0;
56029b521d4Sstefano_zampini     if (missing) {
56129b521d4Sstefano_zampini       for (i = 0; i < M; i++) {
56229b521d4Sstefano_zampini         if (PetscUnlikely(adiag[i] >= ai[i + 1])) {
56329b521d4Sstefano_zampini           for (j = ai[i]; j < ai[i + 1]; j++) {
56429b521d4Sstefano_zampini             if (aj[j] < i) continue;
56529b521d4Sstefano_zampini             nz++;
56629b521d4Sstefano_zampini           }
56729b521d4Sstefano_zampini         } else {
56829b521d4Sstefano_zampini           nz += ai[i + 1] - adiag[i];
56929b521d4Sstefano_zampini         }
57029b521d4Sstefano_zampini       }
57129b521d4Sstefano_zampini     } else {
572829b1710SHong Zhang       for (i = 0; i < M; i++) nz += ai[i + 1] - adiag[i];
57329b521d4Sstefano_zampini     }
5749566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
5759566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
576a6053eceSJunchao Zhang     mumps->nnz = nz;
577a6053eceSJunchao Zhang     mumps->irn = row;
578a6053eceSJunchao Zhang     mumps->jcn = col;
579a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
580185f6596SHong Zhang 
58116ebf90aSShri Abhyankar     nz = 0;
58229b521d4Sstefano_zampini     if (missing) {
58329b521d4Sstefano_zampini       for (i = 0; i < M; i++) {
58429b521d4Sstefano_zampini         if (PetscUnlikely(adiag[i] >= ai[i + 1])) {
58529b521d4Sstefano_zampini           for (j = ai[i]; j < ai[i + 1]; j++) {
58629b521d4Sstefano_zampini             if (aj[j] < i) continue;
5879566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
5889566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(aj[j] + shift, &col[nz]));
58929b521d4Sstefano_zampini             val[nz] = av[j];
59029b521d4Sstefano_zampini             nz++;
59129b521d4Sstefano_zampini           }
59229b521d4Sstefano_zampini         } else {
59329b521d4Sstefano_zampini           rnz = ai[i + 1] - adiag[i];
59429b521d4Sstefano_zampini           ajj = aj + adiag[i];
59529b521d4Sstefano_zampini           v1  = av + adiag[i];
59629b521d4Sstefano_zampini           for (j = 0; j < rnz; j++) {
5979566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
5989566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[nz]));
599a6053eceSJunchao Zhang             val[nz++] = v1[j];
60029b521d4Sstefano_zampini           }
60129b521d4Sstefano_zampini         }
60229b521d4Sstefano_zampini       }
60329b521d4Sstefano_zampini     } else {
60416ebf90aSShri Abhyankar       for (i = 0; i < M; i++) {
60516ebf90aSShri Abhyankar         rnz = ai[i + 1] - adiag[i];
60667877ebaSShri Abhyankar         ajj = aj + adiag[i];
607cf3759fdSShri Abhyankar         v1  = av + adiag[i];
60867877ebaSShri Abhyankar         for (j = 0; j < rnz; j++) {
6099566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
6109566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[nz]));
611a6053eceSJunchao Zhang           val[nz++] = v1[j];
61216ebf90aSShri Abhyankar         }
61316ebf90aSShri Abhyankar       }
61429b521d4Sstefano_zampini     }
615397b6df1SKris Buschelman   } else {
616a6053eceSJunchao Zhang     nz  = 0;
617a6053eceSJunchao Zhang     val = mumps->val;
61829b521d4Sstefano_zampini     if (missing) {
61916ebf90aSShri Abhyankar       for (i = 0; i < M; i++) {
62029b521d4Sstefano_zampini         if (PetscUnlikely(adiag[i] >= ai[i + 1])) {
62129b521d4Sstefano_zampini           for (j = ai[i]; j < ai[i + 1]; j++) {
62229b521d4Sstefano_zampini             if (aj[j] < i) continue;
62329b521d4Sstefano_zampini             val[nz++] = av[j];
62429b521d4Sstefano_zampini           }
62529b521d4Sstefano_zampini         } else {
62616ebf90aSShri Abhyankar           rnz = ai[i + 1] - adiag[i];
62767877ebaSShri Abhyankar           v1  = av + adiag[i];
628ad540459SPierre Jolivet           for (j = 0; j < rnz; j++) val[nz++] = v1[j];
62916ebf90aSShri Abhyankar         }
63016ebf90aSShri Abhyankar       }
63129b521d4Sstefano_zampini     } else {
63216ebf90aSShri Abhyankar       for (i = 0; i < M; i++) {
63316ebf90aSShri Abhyankar         rnz = ai[i + 1] - adiag[i];
63416ebf90aSShri Abhyankar         v1  = av + adiag[i];
635ad540459SPierre Jolivet         for (j = 0; j < rnz; j++) val[nz++] = v1[j];
63616ebf90aSShri Abhyankar       }
63716ebf90aSShri Abhyankar     }
63829b521d4Sstefano_zampini   }
6399566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(A, &av));
6403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
64116ebf90aSShri Abhyankar }
64216ebf90aSShri Abhyankar 
64366976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpisbaij_mpisbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
644d71ae5a4SJacob Faibussowitsch {
645a6053eceSJunchao Zhang   const PetscInt    *ai, *aj, *bi, *bj, *garray, *ajj, *bjj;
646a6053eceSJunchao Zhang   PetscInt           bs;
647a6053eceSJunchao Zhang   PetscInt64         rstart, nz, i, j, k, m, jj, irow, countA, countB;
648a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
64916ebf90aSShri Abhyankar   const PetscScalar *av, *bv, *v1, *v2;
65016ebf90aSShri Abhyankar   PetscScalar       *val;
651397b6df1SKris Buschelman   Mat_MPISBAIJ      *mat = (Mat_MPISBAIJ *)A->data;
652397b6df1SKris Buschelman   Mat_SeqSBAIJ      *aa  = (Mat_SeqSBAIJ *)(mat->A)->data;
653397b6df1SKris Buschelman   Mat_SeqBAIJ       *bb  = (Mat_SeqBAIJ *)(mat->B)->data;
654ec4f40fdSPierre Jolivet   const PetscInt     bs2 = aa->bs2, mbs = aa->mbs;
65538548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
656b94d7dedSBarry Smith   PetscBool hermitian, isset;
65738548759SBarry Smith #endif
65816ebf90aSShri Abhyankar 
65916ebf90aSShri Abhyankar   PetscFunctionBegin;
66038548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
661b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
662b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
66338548759SBarry Smith #endif
6649566063dSJacob Faibussowitsch   PetscCall(MatGetBlockSize(A, &bs));
66538548759SBarry Smith   rstart = A->rmap->rstart;
66638548759SBarry Smith   ai     = aa->i;
66738548759SBarry Smith   aj     = aa->j;
66838548759SBarry Smith   bi     = bb->i;
66938548759SBarry Smith   bj     = bb->j;
67038548759SBarry Smith   av     = aa->a;
67138548759SBarry Smith   bv     = bb->a;
672397b6df1SKris Buschelman 
6732205254eSKarl Rupp   garray = mat->garray;
6742205254eSKarl Rupp 
675bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
676a6053eceSJunchao Zhang     nz = (aa->nz + bb->nz) * bs2; /* just a conservative estimate */
6779566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
6789566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
679a6053eceSJunchao Zhang     /* can not decide the exact mumps->nnz now because of the SBAIJ */
680a6053eceSJunchao Zhang     mumps->irn = row;
681a6053eceSJunchao Zhang     mumps->jcn = col;
682a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
683397b6df1SKris Buschelman   } else {
684a6053eceSJunchao Zhang     val = mumps->val;
685397b6df1SKris Buschelman   }
686397b6df1SKris Buschelman 
6879371c9d4SSatish Balay   jj   = 0;
6889371c9d4SSatish Balay   irow = rstart;
689ec4f40fdSPierre Jolivet   for (i = 0; i < mbs; i++) {
690397b6df1SKris Buschelman     ajj    = aj + ai[i]; /* ptr to the beginning of this row */
691397b6df1SKris Buschelman     countA = ai[i + 1] - ai[i];
692397b6df1SKris Buschelman     countB = bi[i + 1] - bi[i];
693397b6df1SKris Buschelman     bjj    = bj + bi[i];
694ec4f40fdSPierre Jolivet     v1     = av + ai[i] * bs2;
695ec4f40fdSPierre Jolivet     v2     = bv + bi[i] * bs2;
696397b6df1SKris Buschelman 
697ec4f40fdSPierre Jolivet     if (bs > 1) {
698ec4f40fdSPierre Jolivet       /* A-part */
699ec4f40fdSPierre Jolivet       for (j = 0; j < countA; j++) {
700ec4f40fdSPierre Jolivet         for (k = 0; k < bs; k++) {
701ec4f40fdSPierre Jolivet           for (m = 0; m < bs; m++) {
702ec4f40fdSPierre Jolivet             if (rstart + ajj[j] * bs > irow || k >= m) {
703ec4f40fdSPierre Jolivet               if (reuse == MAT_INITIAL_MATRIX) {
7049566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(irow + m + shift, &row[jj]));
7059566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(rstart + ajj[j] * bs + k + shift, &col[jj]));
706ec4f40fdSPierre Jolivet               }
707ec4f40fdSPierre Jolivet               val[jj++] = v1[j * bs2 + m + k * bs];
708ec4f40fdSPierre Jolivet             }
709ec4f40fdSPierre Jolivet           }
710ec4f40fdSPierre Jolivet         }
711ec4f40fdSPierre Jolivet       }
712ec4f40fdSPierre Jolivet 
713ec4f40fdSPierre Jolivet       /* B-part */
714ec4f40fdSPierre Jolivet       for (j = 0; j < countB; j++) {
715ec4f40fdSPierre Jolivet         for (k = 0; k < bs; k++) {
716ec4f40fdSPierre Jolivet           for (m = 0; m < bs; m++) {
717ec4f40fdSPierre Jolivet             if (reuse == MAT_INITIAL_MATRIX) {
7189566063dSJacob Faibussowitsch               PetscCall(PetscMUMPSIntCast(irow + m + shift, &row[jj]));
7199566063dSJacob Faibussowitsch               PetscCall(PetscMUMPSIntCast(garray[bjj[j]] * bs + k + shift, &col[jj]));
720ec4f40fdSPierre Jolivet             }
721ec4f40fdSPierre Jolivet             val[jj++] = v2[j * bs2 + m + k * bs];
722ec4f40fdSPierre Jolivet           }
723ec4f40fdSPierre Jolivet         }
724ec4f40fdSPierre Jolivet       }
725ec4f40fdSPierre Jolivet     } else {
726397b6df1SKris Buschelman       /* A-part */
727397b6df1SKris Buschelman       for (j = 0; j < countA; j++) {
728bccb9932SShri Abhyankar         if (reuse == MAT_INITIAL_MATRIX) {
7299566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
7309566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(rstart + ajj[j] + shift, &col[jj]));
731397b6df1SKris Buschelman         }
73216ebf90aSShri Abhyankar         val[jj++] = v1[j];
733397b6df1SKris Buschelman       }
73416ebf90aSShri Abhyankar 
73516ebf90aSShri Abhyankar       /* B-part */
73616ebf90aSShri Abhyankar       for (j = 0; j < countB; j++) {
737bccb9932SShri Abhyankar         if (reuse == MAT_INITIAL_MATRIX) {
7389566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
7399566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(garray[bjj[j]] + shift, &col[jj]));
740397b6df1SKris Buschelman         }
74116ebf90aSShri Abhyankar         val[jj++] = v2[j];
74216ebf90aSShri Abhyankar       }
74316ebf90aSShri Abhyankar     }
744ec4f40fdSPierre Jolivet     irow += bs;
745ec4f40fdSPierre Jolivet   }
7465d955bbbSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) mumps->nnz = jj;
7473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
74816ebf90aSShri Abhyankar }
74916ebf90aSShri Abhyankar 
75066976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpiaij_mpiaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
751d71ae5a4SJacob Faibussowitsch {
75216ebf90aSShri Abhyankar   const PetscInt    *ai, *aj, *bi, *bj, *garray, m = A->rmap->n, *ajj, *bjj;
7535d955bbbSStefano Zampini   PetscInt64         rstart, cstart, nz, i, j, jj, irow, countA, countB;
754a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
75516ebf90aSShri Abhyankar   const PetscScalar *av, *bv, *v1, *v2;
75616ebf90aSShri Abhyankar   PetscScalar       *val;
757a3d589ffSStefano Zampini   Mat                Ad, Ao;
758a3d589ffSStefano Zampini   Mat_SeqAIJ        *aa;
759a3d589ffSStefano Zampini   Mat_SeqAIJ        *bb;
76016ebf90aSShri Abhyankar 
76116ebf90aSShri Abhyankar   PetscFunctionBegin;
7629566063dSJacob Faibussowitsch   PetscCall(MatMPIAIJGetSeqAIJ(A, &Ad, &Ao, &garray));
7639566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ad, &av));
7649566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ao, &bv));
765a3d589ffSStefano Zampini 
766a3d589ffSStefano Zampini   aa = (Mat_SeqAIJ *)(Ad)->data;
767a3d589ffSStefano Zampini   bb = (Mat_SeqAIJ *)(Ao)->data;
76838548759SBarry Smith   ai = aa->i;
76938548759SBarry Smith   aj = aa->j;
77038548759SBarry Smith   bi = bb->i;
77138548759SBarry Smith   bj = bb->j;
77216ebf90aSShri Abhyankar 
773a3d589ffSStefano Zampini   rstart = A->rmap->rstart;
7745d955bbbSStefano Zampini   cstart = A->cmap->rstart;
7752205254eSKarl Rupp 
776bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
777a6053eceSJunchao Zhang     nz = (PetscInt64)aa->nz + bb->nz; /* make sure the sum won't overflow PetscInt */
7789566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
7799566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
780a6053eceSJunchao Zhang     mumps->nnz = nz;
781a6053eceSJunchao Zhang     mumps->irn = row;
782a6053eceSJunchao Zhang     mumps->jcn = col;
783a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
78416ebf90aSShri Abhyankar   } else {
785a6053eceSJunchao Zhang     val = mumps->val;
78616ebf90aSShri Abhyankar   }
78716ebf90aSShri Abhyankar 
7889371c9d4SSatish Balay   jj   = 0;
7899371c9d4SSatish Balay   irow = rstart;
79016ebf90aSShri Abhyankar   for (i = 0; i < m; i++) {
79116ebf90aSShri Abhyankar     ajj    = aj + ai[i]; /* ptr to the beginning of this row */
79216ebf90aSShri Abhyankar     countA = ai[i + 1] - ai[i];
79316ebf90aSShri Abhyankar     countB = bi[i + 1] - bi[i];
79416ebf90aSShri Abhyankar     bjj    = bj + bi[i];
79516ebf90aSShri Abhyankar     v1     = av + ai[i];
79616ebf90aSShri Abhyankar     v2     = bv + bi[i];
79716ebf90aSShri Abhyankar 
79816ebf90aSShri Abhyankar     /* A-part */
79916ebf90aSShri Abhyankar     for (j = 0; j < countA; j++) {
800bccb9932SShri Abhyankar       if (reuse == MAT_INITIAL_MATRIX) {
8019566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
8025d955bbbSStefano Zampini         PetscCall(PetscMUMPSIntCast(cstart + ajj[j] + shift, &col[jj]));
80316ebf90aSShri Abhyankar       }
80416ebf90aSShri Abhyankar       val[jj++] = v1[j];
80516ebf90aSShri Abhyankar     }
80616ebf90aSShri Abhyankar 
80716ebf90aSShri Abhyankar     /* B-part */
80816ebf90aSShri Abhyankar     for (j = 0; j < countB; j++) {
809bccb9932SShri Abhyankar       if (reuse == MAT_INITIAL_MATRIX) {
8109566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
8119566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(garray[bjj[j]] + shift, &col[jj]));
81216ebf90aSShri Abhyankar       }
81316ebf90aSShri Abhyankar       val[jj++] = v2[j];
81416ebf90aSShri Abhyankar     }
81516ebf90aSShri Abhyankar     irow++;
81616ebf90aSShri Abhyankar   }
8179566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ad, &av));
8189566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ao, &bv));
8193ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
82016ebf90aSShri Abhyankar }
82116ebf90aSShri Abhyankar 
82266976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpibaij_mpiaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
823d71ae5a4SJacob Faibussowitsch {
82467877ebaSShri Abhyankar   Mat_MPIBAIJ       *mat = (Mat_MPIBAIJ *)A->data;
82567877ebaSShri Abhyankar   Mat_SeqBAIJ       *aa  = (Mat_SeqBAIJ *)(mat->A)->data;
82667877ebaSShri Abhyankar   Mat_SeqBAIJ       *bb  = (Mat_SeqBAIJ *)(mat->B)->data;
82767877ebaSShri Abhyankar   const PetscInt    *ai = aa->i, *bi = bb->i, *aj = aa->j, *bj = bb->j, *ajj, *bjj;
8285d955bbbSStefano Zampini   const PetscInt    *garray = mat->garray, mbs = mat->mbs, rstart = A->rmap->rstart, cstart = A->cmap->rstart;
82933d57670SJed Brown   const PetscInt     bs2 = mat->bs2;
830a6053eceSJunchao Zhang   PetscInt           bs;
831a6053eceSJunchao Zhang   PetscInt64         nz, i, j, k, n, jj, irow, countA, countB, idx;
832a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
83367877ebaSShri Abhyankar   const PetscScalar *av = aa->a, *bv = bb->a, *v1, *v2;
83467877ebaSShri Abhyankar   PetscScalar       *val;
83567877ebaSShri Abhyankar 
83667877ebaSShri Abhyankar   PetscFunctionBegin;
8379566063dSJacob Faibussowitsch   PetscCall(MatGetBlockSize(A, &bs));
838bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
83967877ebaSShri Abhyankar     nz = bs2 * (aa->nz + bb->nz);
8409566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
8419566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
842a6053eceSJunchao Zhang     mumps->nnz = nz;
843a6053eceSJunchao Zhang     mumps->irn = row;
844a6053eceSJunchao Zhang     mumps->jcn = col;
845a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
84667877ebaSShri Abhyankar   } else {
847a6053eceSJunchao Zhang     val = mumps->val;
84867877ebaSShri Abhyankar   }
84967877ebaSShri Abhyankar 
8509371c9d4SSatish Balay   jj   = 0;
8519371c9d4SSatish Balay   irow = rstart;
85267877ebaSShri Abhyankar   for (i = 0; i < mbs; i++) {
85367877ebaSShri Abhyankar     countA = ai[i + 1] - ai[i];
85467877ebaSShri Abhyankar     countB = bi[i + 1] - bi[i];
85567877ebaSShri Abhyankar     ajj    = aj + ai[i];
85667877ebaSShri Abhyankar     bjj    = bj + bi[i];
85767877ebaSShri Abhyankar     v1     = av + bs2 * ai[i];
85867877ebaSShri Abhyankar     v2     = bv + bs2 * bi[i];
85967877ebaSShri Abhyankar 
86067877ebaSShri Abhyankar     idx = 0;
86167877ebaSShri Abhyankar     /* A-part */
86267877ebaSShri Abhyankar     for (k = 0; k < countA; k++) {
86367877ebaSShri Abhyankar       for (j = 0; j < bs; j++) {
86467877ebaSShri Abhyankar         for (n = 0; n < bs; n++) {
865bccb9932SShri Abhyankar           if (reuse == MAT_INITIAL_MATRIX) {
8669566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(irow + n + shift, &row[jj]));
8675d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(cstart + bs * ajj[k] + j + shift, &col[jj]));
86867877ebaSShri Abhyankar           }
86967877ebaSShri Abhyankar           val[jj++] = v1[idx++];
87067877ebaSShri Abhyankar         }
87167877ebaSShri Abhyankar       }
87267877ebaSShri Abhyankar     }
87367877ebaSShri Abhyankar 
87467877ebaSShri Abhyankar     idx = 0;
87567877ebaSShri Abhyankar     /* B-part */
87667877ebaSShri Abhyankar     for (k = 0; k < countB; k++) {
87767877ebaSShri Abhyankar       for (j = 0; j < bs; j++) {
87867877ebaSShri Abhyankar         for (n = 0; n < bs; n++) {
879bccb9932SShri Abhyankar           if (reuse == MAT_INITIAL_MATRIX) {
8809566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(irow + n + shift, &row[jj]));
8819566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(bs * garray[bjj[k]] + j + shift, &col[jj]));
88267877ebaSShri Abhyankar           }
883d985c460SShri Abhyankar           val[jj++] = v2[idx++];
88467877ebaSShri Abhyankar         }
88567877ebaSShri Abhyankar       }
88667877ebaSShri Abhyankar     }
887d985c460SShri Abhyankar     irow += bs;
88867877ebaSShri Abhyankar   }
8893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
89067877ebaSShri Abhyankar }
89167877ebaSShri Abhyankar 
89266976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpiaij_mpisbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
893d71ae5a4SJacob Faibussowitsch {
89416ebf90aSShri Abhyankar   const PetscInt    *ai, *aj, *adiag, *bi, *bj, *garray, m = A->rmap->n, *ajj, *bjj;
895a6053eceSJunchao Zhang   PetscInt64         rstart, nz, nza, nzb, i, j, jj, irow, countA, countB;
896a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
89716ebf90aSShri Abhyankar   const PetscScalar *av, *bv, *v1, *v2;
89816ebf90aSShri Abhyankar   PetscScalar       *val;
899a3d589ffSStefano Zampini   Mat                Ad, Ao;
900a3d589ffSStefano Zampini   Mat_SeqAIJ        *aa;
901a3d589ffSStefano Zampini   Mat_SeqAIJ        *bb;
90238548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
903b94d7dedSBarry Smith   PetscBool hermitian, isset;
90438548759SBarry Smith #endif
90516ebf90aSShri Abhyankar 
90616ebf90aSShri Abhyankar   PetscFunctionBegin;
90738548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
908b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
909b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
91038548759SBarry Smith #endif
9119566063dSJacob Faibussowitsch   PetscCall(MatMPIAIJGetSeqAIJ(A, &Ad, &Ao, &garray));
9129566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ad, &av));
9139566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ao, &bv));
914a3d589ffSStefano Zampini 
915a3d589ffSStefano Zampini   aa    = (Mat_SeqAIJ *)(Ad)->data;
916a3d589ffSStefano Zampini   bb    = (Mat_SeqAIJ *)(Ao)->data;
91738548759SBarry Smith   ai    = aa->i;
91838548759SBarry Smith   aj    = aa->j;
91938548759SBarry Smith   adiag = aa->diag;
92038548759SBarry Smith   bi    = bb->i;
92138548759SBarry Smith   bj    = bb->j;
9222205254eSKarl Rupp 
92316ebf90aSShri Abhyankar   rstart = A->rmap->rstart;
92416ebf90aSShri Abhyankar 
925bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
926e0bace9bSHong Zhang     nza = 0; /* num of upper triangular entries in mat->A, including diagonals */
927e0bace9bSHong Zhang     nzb = 0; /* num of upper triangular entries in mat->B */
92816ebf90aSShri Abhyankar     for (i = 0; i < m; i++) {
929e0bace9bSHong Zhang       nza += (ai[i + 1] - adiag[i]);
93016ebf90aSShri Abhyankar       countB = bi[i + 1] - bi[i];
93116ebf90aSShri Abhyankar       bjj    = bj + bi[i];
932e0bace9bSHong Zhang       for (j = 0; j < countB; j++) {
933e0bace9bSHong Zhang         if (garray[bjj[j]] > rstart) nzb++;
934e0bace9bSHong Zhang       }
935e0bace9bSHong Zhang     }
93616ebf90aSShri Abhyankar 
937e0bace9bSHong Zhang     nz = nza + nzb; /* total nz of upper triangular part of mat */
9389566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
9399566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
940a6053eceSJunchao Zhang     mumps->nnz = nz;
941a6053eceSJunchao Zhang     mumps->irn = row;
942a6053eceSJunchao Zhang     mumps->jcn = col;
943a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
94416ebf90aSShri Abhyankar   } else {
945a6053eceSJunchao Zhang     val = mumps->val;
94616ebf90aSShri Abhyankar   }
94716ebf90aSShri Abhyankar 
9489371c9d4SSatish Balay   jj   = 0;
9499371c9d4SSatish Balay   irow = rstart;
95016ebf90aSShri Abhyankar   for (i = 0; i < m; i++) {
95116ebf90aSShri Abhyankar     ajj    = aj + adiag[i]; /* ptr to the beginning of the diagonal of this row */
95216ebf90aSShri Abhyankar     v1     = av + adiag[i];
95316ebf90aSShri Abhyankar     countA = ai[i + 1] - adiag[i];
95416ebf90aSShri Abhyankar     countB = bi[i + 1] - bi[i];
95516ebf90aSShri Abhyankar     bjj    = bj + bi[i];
95616ebf90aSShri Abhyankar     v2     = bv + bi[i];
95716ebf90aSShri Abhyankar 
95816ebf90aSShri Abhyankar     /* A-part */
95916ebf90aSShri Abhyankar     for (j = 0; j < countA; j++) {
960bccb9932SShri Abhyankar       if (reuse == MAT_INITIAL_MATRIX) {
9619566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
9629566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(rstart + ajj[j] + shift, &col[jj]));
96316ebf90aSShri Abhyankar       }
96416ebf90aSShri Abhyankar       val[jj++] = v1[j];
96516ebf90aSShri Abhyankar     }
96616ebf90aSShri Abhyankar 
96716ebf90aSShri Abhyankar     /* B-part */
96816ebf90aSShri Abhyankar     for (j = 0; j < countB; j++) {
96916ebf90aSShri Abhyankar       if (garray[bjj[j]] > rstart) {
970bccb9932SShri Abhyankar         if (reuse == MAT_INITIAL_MATRIX) {
9719566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
9729566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(garray[bjj[j]] + shift, &col[jj]));
97316ebf90aSShri Abhyankar         }
97416ebf90aSShri Abhyankar         val[jj++] = v2[j];
97516ebf90aSShri Abhyankar       }
976397b6df1SKris Buschelman     }
977397b6df1SKris Buschelman     irow++;
978397b6df1SKris Buschelman   }
9799566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ad, &av));
9809566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ao, &bv));
9813ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
982397b6df1SKris Buschelman }
983397b6df1SKris Buschelman 
984c3e1b152SPierre Jolivet static PetscErrorCode MatConvertToTriples_diagonal_xaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
985c3e1b152SPierre Jolivet {
986c3e1b152SPierre Jolivet   const PetscScalar *av;
987c3e1b152SPierre Jolivet   const PetscInt     M = A->rmap->n;
988c3e1b152SPierre Jolivet   PetscInt64         i;
989c3e1b152SPierre Jolivet   PetscMUMPSInt     *row, *col;
990c3e1b152SPierre Jolivet   Vec                v;
991c3e1b152SPierre Jolivet 
992c3e1b152SPierre Jolivet   PetscFunctionBegin;
993c3e1b152SPierre Jolivet   PetscCall(MatDiagonalGetDiagonal(A, &v));
994c3e1b152SPierre Jolivet   PetscCall(VecGetArrayRead(v, &av));
995c3e1b152SPierre Jolivet   if (reuse == MAT_INITIAL_MATRIX) {
996c3e1b152SPierre Jolivet     PetscCall(PetscMalloc2(M, &row, M, &col));
997c3e1b152SPierre Jolivet     for (i = 0; i < M; i++) {
998c3e1b152SPierre Jolivet       PetscCall(PetscMUMPSIntCast(i + A->rmap->rstart, &row[i]));
999c3e1b152SPierre Jolivet       col[i] = row[i];
1000c3e1b152SPierre Jolivet     }
1001c3e1b152SPierre Jolivet     mumps->val = (PetscScalar *)av;
1002c3e1b152SPierre Jolivet     mumps->irn = row;
1003c3e1b152SPierre Jolivet     mumps->jcn = col;
1004c3e1b152SPierre Jolivet     mumps->nnz = M;
1005c3e1b152SPierre Jolivet   } else PetscCall(PetscArraycpy(mumps->val, av, M));
1006c3e1b152SPierre Jolivet   PetscCall(VecRestoreArrayRead(v, &av));
1007c3e1b152SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
1008c3e1b152SPierre Jolivet }
1009c3e1b152SPierre Jolivet 
1010*4b9405b2SPierre Jolivet static PetscErrorCode MatConvertToTriples_dense_xaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
1011*4b9405b2SPierre Jolivet {
1012*4b9405b2SPierre Jolivet   PetscScalar   *v;
1013*4b9405b2SPierre Jolivet   const PetscInt m = A->rmap->n, N = A->cmap->N;
1014*4b9405b2SPierre Jolivet   PetscInt       lda;
1015*4b9405b2SPierre Jolivet   PetscInt64     i, j;
1016*4b9405b2SPierre Jolivet   PetscMUMPSInt *row, *col;
1017*4b9405b2SPierre Jolivet 
1018*4b9405b2SPierre Jolivet   PetscFunctionBegin;
1019*4b9405b2SPierre Jolivet   PetscCall(MatDenseGetArray(A, &v));
1020*4b9405b2SPierre Jolivet   PetscCall(MatDenseGetLDA(A, &lda));
1021*4b9405b2SPierre Jolivet   if (reuse == MAT_INITIAL_MATRIX) {
1022*4b9405b2SPierre Jolivet     PetscCall(PetscMalloc2(m * N, &row, m * N, &col));
1023*4b9405b2SPierre Jolivet     for (i = 0; i < m; i++) {
1024*4b9405b2SPierre Jolivet       col[i] = 0;
1025*4b9405b2SPierre Jolivet       PetscCall(PetscMUMPSIntCast(i + A->rmap->rstart, &row[i]));
1026*4b9405b2SPierre Jolivet     }
1027*4b9405b2SPierre Jolivet     for (j = 1; j < N; j++) {
1028*4b9405b2SPierre Jolivet       for (i = 0; i < m; i++) PetscCall(PetscMUMPSIntCast(j, col + i + m * j));
1029*4b9405b2SPierre Jolivet       PetscCall(PetscArraycpy(row + m * j, row + m * (j - 1), m));
1030*4b9405b2SPierre Jolivet     }
1031*4b9405b2SPierre Jolivet     if (lda == m) mumps->val = v;
1032*4b9405b2SPierre Jolivet     else {
1033*4b9405b2SPierre Jolivet       PetscCall(PetscMalloc1(m * N, &mumps->val));
1034*4b9405b2SPierre Jolivet       mumps->val_alloc = mumps->val;
1035*4b9405b2SPierre Jolivet       for (j = 0; j < N; j++) PetscCall(PetscArraycpy(mumps->val + m * j, v + lda * j, m));
1036*4b9405b2SPierre Jolivet     }
1037*4b9405b2SPierre Jolivet     mumps->irn = row;
1038*4b9405b2SPierre Jolivet     mumps->jcn = col;
1039*4b9405b2SPierre Jolivet     mumps->nnz = m * N;
1040*4b9405b2SPierre Jolivet   } else {
1041*4b9405b2SPierre Jolivet     if (lda == m && !mumps->nest_vals) mumps->val = v;
1042*4b9405b2SPierre Jolivet     else {
1043*4b9405b2SPierre Jolivet       for (j = 0; j < N; j++) PetscCall(PetscArraycpy(mumps->val + m * j, v + lda * j, m));
1044*4b9405b2SPierre Jolivet     }
1045*4b9405b2SPierre Jolivet   }
1046*4b9405b2SPierre Jolivet   PetscCall(MatDenseRestoreArray(A, &v));
1047*4b9405b2SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
1048*4b9405b2SPierre Jolivet }
1049*4b9405b2SPierre Jolivet 
105066976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_nest_xaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
10519d0448ceSStefano Zampini {
10529d0448ceSStefano Zampini   Mat     **mats;
10539d0448ceSStefano Zampini   PetscInt  nr, nc;
10549d0448ceSStefano Zampini   PetscBool chol = mumps->sym ? PETSC_TRUE : PETSC_FALSE;
10559d0448ceSStefano Zampini 
10569d0448ceSStefano Zampini   PetscFunctionBegin;
10579d0448ceSStefano Zampini   PetscCall(MatNestGetSubMats(A, &nr, &nc, &mats));
10589d0448ceSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
10599d0448ceSStefano Zampini     PetscMUMPSInt *irns, *jcns;
10609d0448ceSStefano Zampini     PetscScalar   *vals;
10619d0448ceSStefano Zampini     PetscInt64     totnnz, cumnnz, maxnnz;
10629d0448ceSStefano Zampini     PetscInt      *pjcns_w;
10639d0448ceSStefano Zampini     IS            *rows, *cols;
10649d0448ceSStefano Zampini     PetscInt     **rows_idx, **cols_idx;
10659d0448ceSStefano Zampini 
10669d0448ceSStefano Zampini     cumnnz = 0;
10679d0448ceSStefano Zampini     maxnnz = 0;
10685d955bbbSStefano Zampini     PetscCall(PetscMalloc2(nr * nc + 1, &mumps->nest_vals_start, nr * nc, &mumps->nest_convert_to_triples));
10699d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) {
10709d0448ceSStefano Zampini       for (PetscInt c = 0; c < nc; c++) {
10719d0448ceSStefano Zampini         Mat sub = mats[r][c];
10729d0448ceSStefano Zampini 
10739d0448ceSStefano Zampini         mumps->nest_convert_to_triples[r * nc + c] = NULL;
10749d0448ceSStefano Zampini         if (chol && c < r) continue; /* skip lower-triangular block for Cholesky */
10759d0448ceSStefano Zampini         if (sub) {
10769d0448ceSStefano Zampini           PetscErrorCode (*convert_to_triples)(Mat, PetscInt, MatReuse, Mat_MUMPS *) = NULL;
1077*4b9405b2SPierre Jolivet           PetscBool isSeqAIJ, isMPIAIJ, isSeqBAIJ, isMPIBAIJ, isSeqSBAIJ, isMPISBAIJ, isTrans, isHTrans = PETSC_FALSE, isDiag, isDense;
10789d0448ceSStefano Zampini           MatInfo   info;
10799d0448ceSStefano Zampini 
10805d955bbbSStefano Zampini           PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATTRANSPOSEVIRTUAL, &isTrans));
10815d955bbbSStefano Zampini           if (isTrans) PetscCall(MatTransposeGetMat(sub, &sub));
10825d955bbbSStefano Zampini           else {
10835d955bbbSStefano Zampini             PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATHERMITIANTRANSPOSEVIRTUAL, &isHTrans));
10845d955bbbSStefano Zampini             if (isHTrans) PetscCall(MatHermitianTransposeGetMat(sub, &sub));
10855d955bbbSStefano Zampini           }
10869d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQAIJ, &isSeqAIJ));
10879d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIAIJ, &isMPIAIJ));
10889d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQBAIJ, &isSeqBAIJ));
10899d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIBAIJ, &isMPIBAIJ));
10909d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQSBAIJ, &isSeqSBAIJ));
10919d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPISBAIJ, &isMPISBAIJ));
1092c3e1b152SPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATDIAGONAL, &isDiag));
1093*4b9405b2SPierre Jolivet           PetscCall(PetscObjectTypeCompareAny((PetscObject)sub, &isDense, MATSEQDENSE, MATMPIDENSE, NULL));
10949d0448ceSStefano Zampini 
10959d0448ceSStefano Zampini           if (chol) {
10969d0448ceSStefano Zampini             if (r == c) {
10979d0448ceSStefano Zampini               if (isSeqAIJ) convert_to_triples = MatConvertToTriples_seqaij_seqsbaij;
10989d0448ceSStefano Zampini               else if (isMPIAIJ) convert_to_triples = MatConvertToTriples_mpiaij_mpisbaij;
10999d0448ceSStefano Zampini               else if (isSeqSBAIJ) convert_to_triples = MatConvertToTriples_seqsbaij_seqsbaij;
11009d0448ceSStefano Zampini               else if (isMPISBAIJ) convert_to_triples = MatConvertToTriples_mpisbaij_mpisbaij;
1101c3e1b152SPierre Jolivet               else if (isDiag) convert_to_triples = MatConvertToTriples_diagonal_xaij;
1102*4b9405b2SPierre Jolivet               else if (isDense) convert_to_triples = MatConvertToTriples_dense_xaij;
11039d0448ceSStefano Zampini             } else {
11049d0448ceSStefano Zampini               if (isSeqAIJ) convert_to_triples = MatConvertToTriples_seqaij_seqaij;
11059d0448ceSStefano Zampini               else if (isMPIAIJ) convert_to_triples = MatConvertToTriples_mpiaij_mpiaij;
11069d0448ceSStefano Zampini               else if (isSeqBAIJ) convert_to_triples = MatConvertToTriples_seqbaij_seqaij;
11079d0448ceSStefano Zampini               else if (isMPIBAIJ) convert_to_triples = MatConvertToTriples_mpibaij_mpiaij;
1108c3e1b152SPierre Jolivet               else if (isDiag) convert_to_triples = MatConvertToTriples_diagonal_xaij;
1109*4b9405b2SPierre Jolivet               else if (isDense) convert_to_triples = MatConvertToTriples_dense_xaij;
11109d0448ceSStefano Zampini             }
11119d0448ceSStefano Zampini           } else {
11129d0448ceSStefano Zampini             if (isSeqAIJ) convert_to_triples = MatConvertToTriples_seqaij_seqaij;
11139d0448ceSStefano Zampini             else if (isMPIAIJ) convert_to_triples = MatConvertToTriples_mpiaij_mpiaij;
11149d0448ceSStefano Zampini             else if (isSeqBAIJ) convert_to_triples = MatConvertToTriples_seqbaij_seqaij;
11159d0448ceSStefano Zampini             else if (isMPIBAIJ) convert_to_triples = MatConvertToTriples_mpibaij_mpiaij;
1116c3e1b152SPierre Jolivet             else if (isDiag) convert_to_triples = MatConvertToTriples_diagonal_xaij;
1117*4b9405b2SPierre Jolivet             else if (isDense) convert_to_triples = MatConvertToTriples_dense_xaij;
11189d0448ceSStefano Zampini           }
11199d0448ceSStefano Zampini           PetscCheck(convert_to_triples, PetscObjectComm((PetscObject)sub), PETSC_ERR_SUP, "Not for block of type %s", ((PetscObject)sub)->type_name);
11209d0448ceSStefano Zampini           mumps->nest_convert_to_triples[r * nc + c] = convert_to_triples;
11219d0448ceSStefano Zampini           PetscCall(MatGetInfo(sub, MAT_LOCAL, &info));
11229d0448ceSStefano Zampini           cumnnz += (PetscInt64)info.nz_used; /* can be overestimated for Cholesky */
11239d0448ceSStefano Zampini           maxnnz = PetscMax(maxnnz, info.nz_used);
11249d0448ceSStefano Zampini         }
11259d0448ceSStefano Zampini       }
11269d0448ceSStefano Zampini     }
11279d0448ceSStefano Zampini 
11289d0448ceSStefano Zampini     /* Allocate total COO */
11299d0448ceSStefano Zampini     totnnz = cumnnz;
11309d0448ceSStefano Zampini     PetscCall(PetscMalloc2(totnnz, &irns, totnnz, &jcns));
11319d0448ceSStefano Zampini     PetscCall(PetscMalloc1(totnnz, &vals));
11329d0448ceSStefano Zampini 
11339d0448ceSStefano Zampini     /* Handle rows and column maps
11349d0448ceSStefano Zampini        We directly map rows and use an SF for the columns */
11359d0448ceSStefano Zampini     PetscCall(PetscMalloc4(nr, &rows, nc, &cols, nr, &rows_idx, nc, &cols_idx));
11369d0448ceSStefano Zampini     PetscCall(MatNestGetISs(A, rows, cols));
11379d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) PetscCall(ISGetIndices(rows[r], (const PetscInt **)&rows_idx[r]));
11389d0448ceSStefano Zampini     for (PetscInt c = 0; c < nc; c++) PetscCall(ISGetIndices(cols[c], (const PetscInt **)&cols_idx[c]));
11399d0448ceSStefano Zampini     if (PetscDefined(USE_64BIT_INDICES)) PetscCall(PetscMalloc1(maxnnz, &pjcns_w));
11405d955bbbSStefano Zampini     else (void)maxnnz;
11419d0448ceSStefano Zampini 
11429d0448ceSStefano Zampini     cumnnz = 0;
11439d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) {
11449d0448ceSStefano Zampini       for (PetscInt c = 0; c < nc; c++) {
11459d0448ceSStefano Zampini         Mat             sub  = mats[r][c];
11469d0448ceSStefano Zampini         const PetscInt *ridx = rows_idx[r];
11475d955bbbSStefano Zampini         const PetscInt *cidx = cols_idx[c];
11489d0448ceSStefano Zampini         PetscInt        rst;
11499d0448ceSStefano Zampini         PetscSF         csf;
11505d955bbbSStefano Zampini         PetscBool       isTrans, isHTrans = PETSC_FALSE, swap;
11515d955bbbSStefano Zampini         PetscLayout     cmap;
11529d0448ceSStefano Zampini 
11539d0448ceSStefano Zampini         mumps->nest_vals_start[r * nc + c] = cumnnz;
11549d0448ceSStefano Zampini         if (!mumps->nest_convert_to_triples[r * nc + c]) continue;
11559d0448ceSStefano Zampini 
11565d955bbbSStefano Zampini         /* Extract inner blocks if needed */
11575d955bbbSStefano Zampini         PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATTRANSPOSEVIRTUAL, &isTrans));
11585d955bbbSStefano Zampini         if (isTrans) PetscCall(MatTransposeGetMat(sub, &sub));
11595d955bbbSStefano Zampini         else {
11605d955bbbSStefano Zampini           PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATHERMITIANTRANSPOSEVIRTUAL, &isHTrans));
11615d955bbbSStefano Zampini           if (isHTrans) PetscCall(MatHermitianTransposeGetMat(sub, &sub));
11625d955bbbSStefano Zampini         }
11635d955bbbSStefano Zampini         swap = (PetscBool)(isTrans || isHTrans);
11645d955bbbSStefano Zampini 
11655d955bbbSStefano Zampini         /* Get column layout to map off-process columns */
11665d955bbbSStefano Zampini         PetscCall(MatGetLayouts(sub, NULL, &cmap));
11675d955bbbSStefano Zampini 
11685d955bbbSStefano Zampini         /* Get row start to map on-process rows */
11695d955bbbSStefano Zampini         PetscCall(MatGetOwnershipRange(sub, &rst, NULL));
11705d955bbbSStefano Zampini 
11719d0448ceSStefano Zampini         /* Directly use the mumps datastructure and use C ordering for now */
11729d0448ceSStefano Zampini         PetscCall((*mumps->nest_convert_to_triples[r * nc + c])(sub, 0, MAT_INITIAL_MATRIX, mumps));
11739d0448ceSStefano Zampini 
11745d955bbbSStefano Zampini         /* Swap the role of rows and columns indices for transposed blocks
11755d955bbbSStefano Zampini            since we need values with global final ordering */
11765d955bbbSStefano Zampini         if (swap) {
11775d955bbbSStefano Zampini           cidx = rows_idx[r];
11785d955bbbSStefano Zampini           ridx = cols_idx[c];
11799d0448ceSStefano Zampini         }
11809d0448ceSStefano Zampini 
11815d955bbbSStefano Zampini         /* Communicate column indices
11825d955bbbSStefano Zampini            This could have been done with a single SF but it would have complicated the code a lot.
11835d955bbbSStefano Zampini            But since we do it only once, we pay the price of setting up an SF for each block */
11845d955bbbSStefano Zampini         if (PetscDefined(USE_64BIT_INDICES)) {
11855d955bbbSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) pjcns_w[k] = mumps->jcn[k];
11865d955bbbSStefano Zampini         } else pjcns_w = (PetscInt *)(mumps->jcn); /* This cast is needed only to silence warnings for 64bit integers builds */
11879d0448ceSStefano Zampini         PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)A), &csf));
11885d955bbbSStefano Zampini         PetscCall(PetscSFSetGraphLayout(csf, cmap, mumps->nnz, NULL, PETSC_OWN_POINTER, pjcns_w));
11895d955bbbSStefano Zampini         PetscCall(PetscSFBcastBegin(csf, MPIU_INT, cidx, pjcns_w, MPI_REPLACE));
11905d955bbbSStefano Zampini         PetscCall(PetscSFBcastEnd(csf, MPIU_INT, cidx, pjcns_w, MPI_REPLACE));
11919d0448ceSStefano Zampini         PetscCall(PetscSFDestroy(&csf));
11929d0448ceSStefano Zampini 
11935d955bbbSStefano Zampini         /* Import indices: use direct map for rows and mapped indices for columns */
11945d955bbbSStefano Zampini         if (swap) {
11955d955bbbSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) {
11965d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(ridx[mumps->irn[k] - rst] + shift, &jcns[cumnnz + k]));
11975d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(pjcns_w[k] + shift, &irns[cumnnz + k]));
11985d955bbbSStefano Zampini           }
11995d955bbbSStefano Zampini         } else {
12005d955bbbSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) {
12015d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(ridx[mumps->irn[k] - rst] + shift, &irns[cumnnz + k]));
12025d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(pjcns_w[k] + shift, &jcns[cumnnz + k]));
12035d955bbbSStefano Zampini           }
12045d955bbbSStefano Zampini         }
12055d955bbbSStefano Zampini 
12065d955bbbSStefano Zampini         /* Import values to full COO */
12075d955bbbSStefano Zampini         PetscCall(PetscArraycpy(vals + cumnnz, mumps->val, mumps->nnz));
120850c845baSStefano Zampini         if (isHTrans) { /* conjugate the entries */
120950c845baSStefano Zampini           PetscScalar *v = vals + cumnnz;
121050c845baSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) v[k] = PetscConj(v[k]);
121150c845baSStefano Zampini         }
12129d0448ceSStefano Zampini 
12139d0448ceSStefano Zampini         /* Shift new starting point and sanity check */
12149d0448ceSStefano Zampini         cumnnz += mumps->nnz;
12159d0448ceSStefano Zampini         PetscCheck(cumnnz <= totnnz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected number of nonzeros %" PetscInt64_FMT " != %" PetscInt64_FMT, cumnnz, totnnz);
12169d0448ceSStefano Zampini 
12179d0448ceSStefano Zampini         /* Free scratch memory */
12189d0448ceSStefano Zampini         PetscCall(PetscFree2(mumps->irn, mumps->jcn));
12199d0448ceSStefano Zampini         PetscCall(PetscFree(mumps->val_alloc));
12209d0448ceSStefano Zampini         mumps->val = NULL;
12219d0448ceSStefano Zampini         mumps->nnz = 0;
12229d0448ceSStefano Zampini       }
12239d0448ceSStefano Zampini     }
12249d0448ceSStefano Zampini     if (PetscDefined(USE_64BIT_INDICES)) PetscCall(PetscFree(pjcns_w));
12259d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) PetscCall(ISRestoreIndices(rows[r], (const PetscInt **)&rows_idx[r]));
12269d0448ceSStefano Zampini     for (PetscInt c = 0; c < nc; c++) PetscCall(ISRestoreIndices(cols[c], (const PetscInt **)&cols_idx[c]));
12279d0448ceSStefano Zampini     PetscCall(PetscFree4(rows, cols, rows_idx, cols_idx));
12289d0448ceSStefano Zampini     if (!chol) PetscCheck(cumnnz == totnnz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Different number of nonzeros %" PetscInt64_FMT " != %" PetscInt64_FMT, cumnnz, totnnz);
12295d955bbbSStefano Zampini     mumps->nest_vals_start[nr * nc] = cumnnz;
12309d0448ceSStefano Zampini 
12319d0448ceSStefano Zampini     /* Set pointers for final MUMPS data structure */
12329d0448ceSStefano Zampini     mumps->nest_vals = vals;
12339d0448ceSStefano Zampini     mumps->val_alloc = NULL; /* do not use val_alloc since it may be reallocated with the OMP callpath */
12349d0448ceSStefano Zampini     mumps->val       = vals;
12359d0448ceSStefano Zampini     mumps->irn       = irns;
12369d0448ceSStefano Zampini     mumps->jcn       = jcns;
12379d0448ceSStefano Zampini     mumps->nnz       = cumnnz;
12389d0448ceSStefano Zampini   } else {
12399d0448ceSStefano Zampini     PetscScalar *oval = mumps->nest_vals;
12409d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) {
12419d0448ceSStefano Zampini       for (PetscInt c = 0; c < nc; c++) {
12425d955bbbSStefano Zampini         PetscBool isTrans, isHTrans = PETSC_FALSE;
12435d955bbbSStefano Zampini         Mat       sub  = mats[r][c];
12445d955bbbSStefano Zampini         PetscInt  midx = r * nc + c;
12455d955bbbSStefano Zampini 
12465d955bbbSStefano Zampini         if (!mumps->nest_convert_to_triples[midx]) continue;
12475d955bbbSStefano Zampini         PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATTRANSPOSEVIRTUAL, &isTrans));
12485d955bbbSStefano Zampini         if (isTrans) PetscCall(MatTransposeGetMat(sub, &sub));
12495d955bbbSStefano Zampini         else {
12505d955bbbSStefano Zampini           PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATHERMITIANTRANSPOSEVIRTUAL, &isHTrans));
12515d955bbbSStefano Zampini           if (isHTrans) PetscCall(MatHermitianTransposeGetMat(sub, &sub));
12525d955bbbSStefano Zampini         }
12535d955bbbSStefano Zampini         mumps->val = oval + mumps->nest_vals_start[midx];
12545d955bbbSStefano Zampini         PetscCall((*mumps->nest_convert_to_triples[midx])(sub, shift, MAT_REUSE_MATRIX, mumps));
12555d955bbbSStefano Zampini         if (isHTrans) {
12565d955bbbSStefano Zampini           PetscInt nnz = mumps->nest_vals_start[midx + 1] - mumps->nest_vals_start[midx];
12575d955bbbSStefano Zampini           for (PetscInt k = 0; k < nnz; k++) mumps->val[k] = PetscConj(mumps->val[k]);
12585d955bbbSStefano Zampini         }
12599d0448ceSStefano Zampini       }
12609d0448ceSStefano Zampini     }
12619d0448ceSStefano Zampini     mumps->val = oval;
12629d0448ceSStefano Zampini   }
12639d0448ceSStefano Zampini   PetscFunctionReturn(PETSC_SUCCESS);
12649d0448ceSStefano Zampini }
12659d0448ceSStefano Zampini 
126666976f2fSJacob Faibussowitsch static PetscErrorCode MatDestroy_MUMPS(Mat A)
1267d71ae5a4SJacob Faibussowitsch {
1268a6053eceSJunchao Zhang   Mat_MUMPS *mumps = (Mat_MUMPS *)A->data;
1269b24902e0SBarry Smith 
1270397b6df1SKris Buschelman   PetscFunctionBegin;
12719566063dSJacob Faibussowitsch   PetscCall(PetscFree2(mumps->id.sol_loc, mumps->id.isol_loc));
12729566063dSJacob Faibussowitsch   PetscCall(VecScatterDestroy(&mumps->scat_rhs));
12739566063dSJacob Faibussowitsch   PetscCall(VecScatterDestroy(&mumps->scat_sol));
12749566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&mumps->b_seq));
12759566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&mumps->x_seq));
12769566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.perm_in));
12779566063dSJacob Faibussowitsch   PetscCall(PetscFree2(mumps->irn, mumps->jcn));
12789566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->val_alloc));
12799566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->info));
1280413bcc21SPierre Jolivet   PetscCall(PetscFree(mumps->ICNTL_pre));
1281413bcc21SPierre Jolivet   PetscCall(PetscFree(mumps->CNTL_pre));
12829566063dSJacob Faibussowitsch   PetscCall(MatMumpsResetSchur_Private(mumps));
1283413bcc21SPierre Jolivet   if (mumps->id.job != JOB_NULL) { /* cannot call PetscMUMPS_c() if JOB_INIT has never been called for this instance */
1284a5e57a09SHong Zhang     mumps->id.job = JOB_END;
12853ab56b82SJunchao Zhang     PetscMUMPS_c(mumps);
12869261f6e4SBarry 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));
1287413bcc21SPierre Jolivet     if (mumps->mumps_comm != MPI_COMM_NULL) {
1288413bcc21SPierre Jolivet       if (PetscDefined(HAVE_OPENMP_SUPPORT) && mumps->use_petsc_omp_support) PetscCallMPI(MPI_Comm_free(&mumps->mumps_comm));
1289413bcc21SPierre Jolivet       else PetscCall(PetscCommRestoreComm(PetscObjectComm((PetscObject)A), &mumps->mumps_comm));
1290413bcc21SPierre Jolivet     }
1291413bcc21SPierre Jolivet   }
12923ab56b82SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
129367602552SJunchao Zhang   if (mumps->use_petsc_omp_support) {
12949566063dSJacob Faibussowitsch     PetscCall(PetscOmpCtrlDestroy(&mumps->omp_ctrl));
12959566063dSJacob Faibussowitsch     PetscCall(PetscFree2(mumps->rhs_loc, mumps->rhs_recvbuf));
12969566063dSJacob Faibussowitsch     PetscCall(PetscFree3(mumps->rhs_nrow, mumps->rhs_recvcounts, mumps->rhs_disps));
129767602552SJunchao Zhang   }
12983ab56b82SJunchao Zhang #endif
12999566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->ia_alloc));
13009566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->ja_alloc));
13019566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->recvcount));
13029566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->reqs));
13039566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->irhs_loc));
13049d0448ceSStefano Zampini   PetscCall(PetscFree2(mumps->nest_vals_start, mumps->nest_convert_to_triples));
13059d0448ceSStefano Zampini   PetscCall(PetscFree(mumps->nest_vals));
13069566063dSJacob Faibussowitsch   PetscCall(PetscFree(A->data));
1307bf0cc555SLisandro Dalcin 
130897969023SHong Zhang   /* clear composed functions */
13099566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorGetSolverType_C", NULL));
13109566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorSetSchurIS_C", NULL));
13119566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorCreateSchurComplement_C", NULL));
13129566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsSetIcntl_C", NULL));
13139566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetIcntl_C", NULL));
13149566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsSetCntl_C", NULL));
13159566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetCntl_C", NULL));
13169566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInfo_C", NULL));
13179566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInfog_C", NULL));
13189566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetRinfo_C", NULL));
13199566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetRinfog_C", NULL));
13205c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetNullPivots_C", NULL));
13219566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInverse_C", NULL));
13229566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInverseTranspose_C", NULL));
13233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1324397b6df1SKris Buschelman }
1325397b6df1SKris Buschelman 
132667602552SJunchao 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. */
1327d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMumpsSetUpDistRHSInfo(Mat A, PetscInt nrhs, const PetscScalar *array)
1328d71ae5a4SJacob Faibussowitsch {
132967602552SJunchao Zhang   Mat_MUMPS        *mumps   = (Mat_MUMPS *)A->data;
133067602552SJunchao Zhang   const PetscMPIInt ompsize = mumps->omp_comm_size;
133167602552SJunchao Zhang   PetscInt          i, m, M, rstart;
133267602552SJunchao Zhang 
133367602552SJunchao Zhang   PetscFunctionBegin;
13349566063dSJacob Faibussowitsch   PetscCall(MatGetSize(A, &M, NULL));
13359566063dSJacob Faibussowitsch   PetscCall(MatGetLocalSize(A, &m, NULL));
133608401ef6SPierre Jolivet   PetscCheck(M <= PETSC_MUMPS_INT_MAX, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "PetscInt too long for PetscMUMPSInt");
133767602552SJunchao Zhang   if (ompsize == 1) {
133867602552SJunchao Zhang     if (!mumps->irhs_loc) {
133967602552SJunchao Zhang       mumps->nloc_rhs = m;
13409566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(m, &mumps->irhs_loc));
13419566063dSJacob Faibussowitsch       PetscCall(MatGetOwnershipRange(A, &rstart, NULL));
134267602552SJunchao Zhang       for (i = 0; i < m; i++) mumps->irhs_loc[i] = rstart + i + 1; /* use 1-based indices */
134367602552SJunchao Zhang     }
134467602552SJunchao Zhang     mumps->id.rhs_loc = (MumpsScalar *)array;
134567602552SJunchao Zhang   } else {
134667602552SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
134767602552SJunchao Zhang     const PetscInt *ranges;
134867602552SJunchao Zhang     PetscMPIInt     j, k, sendcount, *petsc_ranks, *omp_ranks;
134967602552SJunchao Zhang     MPI_Group       petsc_group, omp_group;
135067602552SJunchao Zhang     PetscScalar    *recvbuf = NULL;
135167602552SJunchao Zhang 
135267602552SJunchao Zhang     if (mumps->is_omp_master) {
135367602552SJunchao Zhang       /* Lazily initialize the omp stuff for distributed rhs */
135467602552SJunchao Zhang       if (!mumps->irhs_loc) {
13559566063dSJacob Faibussowitsch         PetscCall(PetscMalloc2(ompsize, &omp_ranks, ompsize, &petsc_ranks));
13569566063dSJacob Faibussowitsch         PetscCall(PetscMalloc3(ompsize, &mumps->rhs_nrow, ompsize, &mumps->rhs_recvcounts, ompsize, &mumps->rhs_disps));
13579566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Comm_group(mumps->petsc_comm, &petsc_group));
13589566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Comm_group(mumps->omp_comm, &omp_group));
135967602552SJunchao Zhang         for (j = 0; j < ompsize; j++) omp_ranks[j] = j;
13609566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Group_translate_ranks(omp_group, ompsize, omp_ranks, petsc_group, petsc_ranks));
136167602552SJunchao Zhang 
136267602552SJunchao Zhang         /* Populate mumps->irhs_loc[], rhs_nrow[] */
136367602552SJunchao Zhang         mumps->nloc_rhs = 0;
13649566063dSJacob Faibussowitsch         PetscCall(MatGetOwnershipRanges(A, &ranges));
136567602552SJunchao Zhang         for (j = 0; j < ompsize; j++) {
136667602552SJunchao Zhang           mumps->rhs_nrow[j] = ranges[petsc_ranks[j] + 1] - ranges[petsc_ranks[j]];
136767602552SJunchao Zhang           mumps->nloc_rhs += mumps->rhs_nrow[j];
136867602552SJunchao Zhang         }
13699566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(mumps->nloc_rhs, &mumps->irhs_loc));
137067602552SJunchao Zhang         for (j = k = 0; j < ompsize; j++) {
137167602552SJunchao 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 */
137267602552SJunchao Zhang         }
137367602552SJunchao Zhang 
13749566063dSJacob Faibussowitsch         PetscCall(PetscFree2(omp_ranks, petsc_ranks));
13759566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Group_free(&petsc_group));
13769566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Group_free(&omp_group));
137767602552SJunchao Zhang       }
137867602552SJunchao Zhang 
137967602552SJunchao Zhang       /* Realloc buffers when current nrhs is bigger than what we have met */
138067602552SJunchao Zhang       if (nrhs > mumps->max_nrhs) {
13819566063dSJacob Faibussowitsch         PetscCall(PetscFree2(mumps->rhs_loc, mumps->rhs_recvbuf));
13829566063dSJacob Faibussowitsch         PetscCall(PetscMalloc2(mumps->nloc_rhs * nrhs, &mumps->rhs_loc, mumps->nloc_rhs * nrhs, &mumps->rhs_recvbuf));
138367602552SJunchao Zhang         mumps->max_nrhs = nrhs;
138467602552SJunchao Zhang       }
138567602552SJunchao Zhang 
138667602552SJunchao Zhang       /* Setup recvcounts[], disps[], recvbuf on omp rank 0 for the upcoming MPI_Gatherv */
13879566063dSJacob Faibussowitsch       for (j = 0; j < ompsize; j++) PetscCall(PetscMPIIntCast(mumps->rhs_nrow[j] * nrhs, &mumps->rhs_recvcounts[j]));
138867602552SJunchao Zhang       mumps->rhs_disps[0] = 0;
138967602552SJunchao Zhang       for (j = 1; j < ompsize; j++) {
139067602552SJunchao Zhang         mumps->rhs_disps[j] = mumps->rhs_disps[j - 1] + mumps->rhs_recvcounts[j - 1];
139108401ef6SPierre Jolivet         PetscCheck(mumps->rhs_disps[j] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "PetscMPIInt overflow!");
139267602552SJunchao Zhang       }
139367602552SJunchao Zhang       recvbuf = (nrhs == 1) ? mumps->rhs_loc : mumps->rhs_recvbuf; /* Directly use rhs_loc[] as recvbuf. Single rhs is common in Ax=b */
139467602552SJunchao Zhang     }
139567602552SJunchao Zhang 
13969566063dSJacob Faibussowitsch     PetscCall(PetscMPIIntCast(m * nrhs, &sendcount));
13979566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Gatherv(array, sendcount, MPIU_SCALAR, recvbuf, mumps->rhs_recvcounts, mumps->rhs_disps, MPIU_SCALAR, 0, mumps->omp_comm));
139867602552SJunchao Zhang 
139967602552SJunchao Zhang     if (mumps->is_omp_master) {
140067602552SJunchao Zhang       if (nrhs > 1) { /* Copy & re-arrange data from rhs_recvbuf[] to mumps->rhs_loc[] only when there are multiple rhs */
140167602552SJunchao Zhang         PetscScalar *dst, *dstbase = mumps->rhs_loc;
140267602552SJunchao Zhang         for (j = 0; j < ompsize; j++) {
140367602552SJunchao Zhang           const PetscScalar *src = mumps->rhs_recvbuf + mumps->rhs_disps[j];
140467602552SJunchao Zhang           dst                    = dstbase;
140567602552SJunchao Zhang           for (i = 0; i < nrhs; i++) {
14069566063dSJacob Faibussowitsch             PetscCall(PetscArraycpy(dst, src, mumps->rhs_nrow[j]));
140767602552SJunchao Zhang             src += mumps->rhs_nrow[j];
140867602552SJunchao Zhang             dst += mumps->nloc_rhs;
140967602552SJunchao Zhang           }
141067602552SJunchao Zhang           dstbase += mumps->rhs_nrow[j];
141167602552SJunchao Zhang         }
141267602552SJunchao Zhang       }
141367602552SJunchao Zhang       mumps->id.rhs_loc = (MumpsScalar *)mumps->rhs_loc;
141467602552SJunchao Zhang     }
141567602552SJunchao Zhang #endif /* PETSC_HAVE_OPENMP_SUPPORT */
141667602552SJunchao Zhang   }
141767602552SJunchao Zhang   mumps->id.nrhs     = nrhs;
141867602552SJunchao Zhang   mumps->id.nloc_rhs = mumps->nloc_rhs;
141967602552SJunchao Zhang   mumps->id.lrhs_loc = mumps->nloc_rhs;
142067602552SJunchao Zhang   mumps->id.irhs_loc = mumps->irhs_loc;
14213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
142267602552SJunchao Zhang }
142367602552SJunchao Zhang 
142466976f2fSJacob Faibussowitsch static PetscErrorCode MatSolve_MUMPS(Mat A, Vec b, Vec x)
1425d71ae5a4SJacob Faibussowitsch {
1426e69c285eSBarry Smith   Mat_MUMPS         *mumps  = (Mat_MUMPS *)A->data;
142725aac85cSJunchao Zhang   const PetscScalar *rarray = NULL;
1428d54de34fSKris Buschelman   PetscScalar       *array;
1429329ec9b3SHong Zhang   IS                 is_iden, is_petsc;
1430329ec9b3SHong Zhang   PetscInt           i;
1431cc86f929SStefano Zampini   PetscBool          second_solve = PETSC_FALSE;
1432883f2eb9SBarry Smith   static PetscBool   cite1 = PETSC_FALSE, cite2 = PETSC_FALSE;
1433397b6df1SKris Buschelman 
1434397b6df1SKris Buschelman   PetscFunctionBegin;
14359371c9d4SSatish 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 "
14369371c9d4SSatish Balay                                    "Journal on Matrix Analysis and Applications},\n  volume = {23},\n  number = {1},\n  pages = {15--41},\n  year = {2001}\n}\n",
14379371c9d4SSatish Balay                                    &cite1));
14389371c9d4SSatish 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 "
14399371c9d4SSatish Balay                                    "Computing},\n  volume = {32},\n  number = {2},\n  pages = {136--156},\n  year = {2006}\n}\n",
14409371c9d4SSatish Balay                                    &cite2));
14412aca8efcSHong Zhang 
1442603e8f96SBarry Smith   if (A->factorerrortype) {
14439566063dSJacob 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)));
14449566063dSJacob Faibussowitsch     PetscCall(VecSetInf(x));
14453ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
14462aca8efcSHong Zhang   }
14472aca8efcSHong Zhang 
1448a5e57a09SHong Zhang   mumps->id.nrhs = 1;
14492d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) {
145025aac85cSJunchao Zhang     if (mumps->ICNTL20 == 10) {
145167602552SJunchao Zhang       mumps->id.ICNTL(20) = 10; /* dense distributed RHS */
14529566063dSJacob Faibussowitsch       PetscCall(VecGetArrayRead(b, &rarray));
14539566063dSJacob Faibussowitsch       PetscCall(MatMumpsSetUpDistRHSInfo(A, 1, rarray));
145425aac85cSJunchao Zhang     } else {
145541ffd417SStefano Zampini       mumps->id.ICNTL(20) = 0; /* dense centralized RHS; Scatter b into a sequential rhs vector*/
14569566063dSJacob Faibussowitsch       PetscCall(VecScatterBegin(mumps->scat_rhs, b, mumps->b_seq, INSERT_VALUES, SCATTER_FORWARD));
14579566063dSJacob Faibussowitsch       PetscCall(VecScatterEnd(mumps->scat_rhs, b, mumps->b_seq, INSERT_VALUES, SCATTER_FORWARD));
145867602552SJunchao Zhang       if (!mumps->myid) {
14599566063dSJacob Faibussowitsch         PetscCall(VecGetArray(mumps->b_seq, &array));
146067602552SJunchao Zhang         mumps->id.rhs = (MumpsScalar *)array;
146167602552SJunchao Zhang       }
146225aac85cSJunchao Zhang     }
14633ab56b82SJunchao Zhang   } else {                   /* petsc_size == 1 */
146467602552SJunchao Zhang     mumps->id.ICNTL(20) = 0; /* dense centralized RHS */
14659566063dSJacob Faibussowitsch     PetscCall(VecCopy(b, x));
14669566063dSJacob Faibussowitsch     PetscCall(VecGetArray(x, &array));
1467940cd9d6SSatish Balay     mumps->id.rhs = (MumpsScalar *)array;
1468397b6df1SKris Buschelman   }
1469397b6df1SKris Buschelman 
1470cc86f929SStefano Zampini   /*
1471cc86f929SStefano Zampini      handle condensation step of Schur complement (if any)
1472cc86f929SStefano Zampini      We set by default ICNTL(26) == -1 when Schur indices have been provided by the user.
1473cc86f929SStefano Zampini      According to MUMPS (5.0.0) manual, any value should be harmful during the factorization phase
1474cc86f929SStefano Zampini      Unless the user provides a valid value for ICNTL(26), MatSolve and MatMatSolve routines solve the full system.
1475cc86f929SStefano Zampini      This requires an extra call to PetscMUMPS_c and the computation of the factors for S
1476cc86f929SStefano Zampini   */
14773e5b40d0SPierre Jolivet   if (mumps->id.size_schur > 0) {
147808401ef6SPierre Jolivet     PetscCheck(mumps->petsc_size <= 1, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Parallel Schur complements not yet supported from PETSc");
14793e5b40d0SPierre Jolivet     if (mumps->id.ICNTL(26) < 0 || mumps->id.ICNTL(26) > 2) {
1480cc86f929SStefano Zampini       second_solve = PETSC_TRUE;
14819566063dSJacob Faibussowitsch       PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
14823e5b40d0SPierre Jolivet       mumps->id.ICNTL(26) = 1; /* condensation phase */
14833e5b40d0SPierre Jolivet     } else if (mumps->id.ICNTL(26) == 1) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
1484cc86f929SStefano Zampini   }
1485397b6df1SKris Buschelman   /* solve phase */
1486a5e57a09SHong Zhang   mumps->id.job = JOB_SOLVE;
14873ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
14889261f6e4SBarry 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));
1489397b6df1SKris Buschelman 
1490b5fa320bSStefano Zampini   /* handle expansion step of Schur complement (if any) */
14911baa6e33SBarry Smith   if (second_solve) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_TRUE));
14923e5b40d0SPierre Jolivet   else if (mumps->id.ICNTL(26) == 1) {
14933e5b40d0SPierre Jolivet     PetscCall(MatMumpsSolveSchur_Private(A));
14943e5b40d0SPierre Jolivet     for (i = 0; i < mumps->id.size_schur; ++i) {
14953e5b40d0SPierre Jolivet #if !defined(PETSC_USE_COMPLEX)
14963e5b40d0SPierre Jolivet       PetscScalar val = mumps->id.redrhs[i];
14973e5b40d0SPierre Jolivet #else
14983e5b40d0SPierre Jolivet       PetscScalar val = mumps->id.redrhs[i].r + PETSC_i * mumps->id.redrhs[i].i;
14993e5b40d0SPierre Jolivet #endif
15003e5b40d0SPierre Jolivet       array[mumps->id.listvar_schur[i] - 1] = val;
15013e5b40d0SPierre Jolivet     }
15023e5b40d0SPierre Jolivet   }
1503b5fa320bSStefano Zampini 
15042d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) { /* convert mumps distributed solution to petsc mpi x */
1505a5e57a09SHong Zhang     if (mumps->scat_sol && mumps->ICNTL9_pre != mumps->id.ICNTL(9)) {
1506a5e57a09SHong Zhang       /* when id.ICNTL(9) changes, the contents of lsol_loc may change (not its size, lsol_loc), recreates scat_sol */
15079566063dSJacob Faibussowitsch       PetscCall(VecScatterDestroy(&mumps->scat_sol));
1508397b6df1SKris Buschelman     }
1509a5e57a09SHong Zhang     if (!mumps->scat_sol) { /* create scatter scat_sol */
1510a6053eceSJunchao Zhang       PetscInt *isol2_loc = NULL;
15119566063dSJacob Faibussowitsch       PetscCall(ISCreateStride(PETSC_COMM_SELF, mumps->id.lsol_loc, 0, 1, &is_iden)); /* from */
15129566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(mumps->id.lsol_loc, &isol2_loc));
1513a6053eceSJunchao 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 */
15149566063dSJacob Faibussowitsch       PetscCall(ISCreateGeneral(PETSC_COMM_SELF, mumps->id.lsol_loc, isol2_loc, PETSC_OWN_POINTER, &is_petsc)); /* to */
15159566063dSJacob Faibussowitsch       PetscCall(VecScatterCreate(mumps->x_seq, is_iden, x, is_petsc, &mumps->scat_sol));
15169566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_iden));
15179566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_petsc));
1518a5e57a09SHong Zhang       mumps->ICNTL9_pre = mumps->id.ICNTL(9); /* save current value of id.ICNTL(9) */
1519397b6df1SKris Buschelman     }
1520a5e57a09SHong Zhang 
15219566063dSJacob Faibussowitsch     PetscCall(VecScatterBegin(mumps->scat_sol, mumps->x_seq, x, INSERT_VALUES, SCATTER_FORWARD));
15229566063dSJacob Faibussowitsch     PetscCall(VecScatterEnd(mumps->scat_sol, mumps->x_seq, x, INSERT_VALUES, SCATTER_FORWARD));
1523329ec9b3SHong Zhang   }
1524353d7d71SJunchao Zhang 
152567602552SJunchao Zhang   if (mumps->petsc_size > 1) {
152625aac85cSJunchao Zhang     if (mumps->ICNTL20 == 10) {
15279566063dSJacob Faibussowitsch       PetscCall(VecRestoreArrayRead(b, &rarray));
152825aac85cSJunchao Zhang     } else if (!mumps->myid) {
15299566063dSJacob Faibussowitsch       PetscCall(VecRestoreArray(mumps->b_seq, &array));
153025aac85cSJunchao Zhang     }
15319566063dSJacob Faibussowitsch   } else PetscCall(VecRestoreArray(x, &array));
1532353d7d71SJunchao Zhang 
153364412097SPierre Jolivet   PetscCall(PetscLogFlops(2.0 * PetscMax(0, (mumps->id.INFO(28) >= 0 ? mumps->id.INFO(28) : -1000000 * mumps->id.INFO(28)) - A->cmap->n)));
15343ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1535397b6df1SKris Buschelman }
1536397b6df1SKris Buschelman 
153766976f2fSJacob Faibussowitsch static PetscErrorCode MatSolveTranspose_MUMPS(Mat A, Vec b, Vec x)
1538d71ae5a4SJacob Faibussowitsch {
1539e69c285eSBarry Smith   Mat_MUMPS          *mumps = (Mat_MUMPS *)A->data;
1540338d3105SPierre Jolivet   const PetscMUMPSInt value = mumps->id.ICNTL(9);
154151d5961aSHong Zhang 
154251d5961aSHong Zhang   PetscFunctionBegin;
1543a5e57a09SHong Zhang   mumps->id.ICNTL(9) = 0;
15449566063dSJacob Faibussowitsch   PetscCall(MatSolve_MUMPS(A, b, x));
1545338d3105SPierre Jolivet   mumps->id.ICNTL(9) = value;
15463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
154751d5961aSHong Zhang }
154851d5961aSHong Zhang 
154966976f2fSJacob Faibussowitsch static PetscErrorCode MatMatSolve_MUMPS(Mat A, Mat B, Mat X)
1550d71ae5a4SJacob Faibussowitsch {
1551b8491c3eSStefano Zampini   Mat                Bt = NULL;
1552a6053eceSJunchao Zhang   PetscBool          denseX, denseB, flg, flgT;
1553e69c285eSBarry Smith   Mat_MUMPS         *mumps = (Mat_MUMPS *)A->data;
1554334c5f61SHong Zhang   PetscInt           i, nrhs, M;
15551683a169SBarry Smith   PetscScalar       *array;
15561683a169SBarry Smith   const PetscScalar *rbray;
1557a6053eceSJunchao Zhang   PetscInt           lsol_loc, nlsol_loc, *idxx, iidx = 0;
1558a6053eceSJunchao Zhang   PetscMUMPSInt     *isol_loc, *isol_loc_save;
15591683a169SBarry Smith   PetscScalar       *bray, *sol_loc, *sol_loc_save;
1560be818407SHong Zhang   IS                 is_to, is_from;
1561beae5ec0SHong Zhang   PetscInt           k, proc, j, m, myrstart;
1562be818407SHong Zhang   const PetscInt    *rstart;
156367602552SJunchao Zhang   Vec                v_mpi, msol_loc;
156467602552SJunchao Zhang   VecScatter         scat_sol;
156567602552SJunchao Zhang   Vec                b_seq;
156667602552SJunchao Zhang   VecScatter         scat_rhs;
1567be818407SHong Zhang   PetscScalar       *aa;
1568be818407SHong Zhang   PetscInt           spnr, *ia, *ja;
1569d56c302dSHong Zhang   Mat_MPIAIJ        *b = NULL;
1570bda8bf91SBarry Smith 
1571e0b74bf9SHong Zhang   PetscFunctionBegin;
15729566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)X, &denseX, MATSEQDENSE, MATMPIDENSE, NULL));
157328b400f6SJacob Faibussowitsch   PetscCheck(denseX, PetscObjectComm((PetscObject)X), PETSC_ERR_ARG_WRONG, "Matrix X must be MATDENSE matrix");
1574be818407SHong Zhang 
15759566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)B, &denseB, MATSEQDENSE, MATMPIDENSE, NULL));
1576a6053eceSJunchao Zhang   if (denseB) {
157708401ef6SPierre Jolivet     PetscCheck(B->rmap->n == X->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Matrix B and X must have same row distribution");
1578be818407SHong Zhang     mumps->id.ICNTL(20) = 0; /* dense RHS */
15790e6b8875SHong Zhang   } else {                   /* sparse B */
158008401ef6SPierre Jolivet     PetscCheck(X != B, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_IDN, "X and B must be different matrices");
1581013e2dc7SBarry Smith     PetscCall(PetscObjectTypeCompare((PetscObject)B, MATTRANSPOSEVIRTUAL, &flgT));
1582da81f932SPierre Jolivet     if (flgT) { /* input B is transpose of actual RHS matrix,
15830e6b8875SHong Zhang                  because mumps requires sparse compressed COLUMN storage! See MatMatTransposeSolve_MUMPS() */
15849566063dSJacob Faibussowitsch       PetscCall(MatTransposeGetMat(B, &Bt));
1585013e2dc7SBarry Smith     } else SETERRQ(PetscObjectComm((PetscObject)B), PETSC_ERR_ARG_WRONG, "Matrix B must be MATTRANSPOSEVIRTUAL matrix");
1586be818407SHong Zhang     mumps->id.ICNTL(20) = 1; /* sparse RHS */
1587b8491c3eSStefano Zampini   }
158887b22cf4SHong Zhang 
15899566063dSJacob Faibussowitsch   PetscCall(MatGetSize(B, &M, &nrhs));
15909481e6e9SHong Zhang   mumps->id.nrhs = nrhs;
15919481e6e9SHong Zhang   mumps->id.lrhs = M;
15922b691707SHong Zhang   mumps->id.rhs  = NULL;
15939481e6e9SHong Zhang 
15942d4298aeSJunchao Zhang   if (mumps->petsc_size == 1) {
1595b8491c3eSStefano Zampini     PetscScalar *aa;
1596b8491c3eSStefano Zampini     PetscInt     spnr, *ia, *ja;
1597e94cce23SStefano Zampini     PetscBool    second_solve = PETSC_FALSE;
1598b8491c3eSStefano Zampini 
15999566063dSJacob Faibussowitsch     PetscCall(MatDenseGetArray(X, &array));
1600b8491c3eSStefano Zampini     mumps->id.rhs = (MumpsScalar *)array;
16012b691707SHong Zhang 
1602a6053eceSJunchao Zhang     if (denseB) {
16032b691707SHong Zhang       /* copy B to X */
16049566063dSJacob Faibussowitsch       PetscCall(MatDenseGetArrayRead(B, &rbray));
16059566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(array, rbray, M * nrhs));
16069566063dSJacob Faibussowitsch       PetscCall(MatDenseRestoreArrayRead(B, &rbray));
16072b691707SHong Zhang     } else { /* sparse B */
16089566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJGetArray(Bt, &aa));
16099566063dSJacob Faibussowitsch       PetscCall(MatGetRowIJ(Bt, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
161028b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
16119566063dSJacob Faibussowitsch       PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs));
1612b8491c3eSStefano Zampini       mumps->id.rhs_sparse = (MumpsScalar *)aa;
1613b8491c3eSStefano Zampini     }
1614e94cce23SStefano Zampini     /* handle condensation step of Schur complement (if any) */
16153e5b40d0SPierre Jolivet     if (mumps->id.size_schur > 0) {
16163e5b40d0SPierre Jolivet       PetscCheck(mumps->petsc_size <= 1, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Parallel Schur complements not yet supported from PETSc");
16173e5b40d0SPierre Jolivet       if (mumps->id.ICNTL(26) < 0 || mumps->id.ICNTL(26) > 2) {
1618e94cce23SStefano Zampini         second_solve = PETSC_TRUE;
16199566063dSJacob Faibussowitsch         PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
16203e5b40d0SPierre Jolivet         mumps->id.ICNTL(26) = 1; /* condensation phase */
16213e5b40d0SPierre Jolivet       } else if (mumps->id.ICNTL(26) == 1) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
1622e94cce23SStefano Zampini     }
16232cd7d884SHong Zhang     /* solve phase */
16242cd7d884SHong Zhang     mumps->id.job = JOB_SOLVE;
16253ab56b82SJunchao Zhang     PetscMUMPS_c(mumps);
16269261f6e4SBarry 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));
1627b5fa320bSStefano Zampini 
1628b5fa320bSStefano Zampini     /* handle expansion step of Schur complement (if any) */
16291baa6e33SBarry Smith     if (second_solve) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_TRUE));
16303e5b40d0SPierre Jolivet     else if (mumps->id.ICNTL(26) == 1) {
16313e5b40d0SPierre Jolivet       PetscCall(MatMumpsSolveSchur_Private(A));
16323e5b40d0SPierre Jolivet       for (j = 0; j < nrhs; ++j)
16333e5b40d0SPierre Jolivet         for (i = 0; i < mumps->id.size_schur; ++i) {
16343e5b40d0SPierre Jolivet #if !defined(PETSC_USE_COMPLEX)
16353e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.redrhs[i + j * mumps->id.lredrhs];
16363e5b40d0SPierre Jolivet #else
16373e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.redrhs[i + j * mumps->id.lredrhs].r + PETSC_i * mumps->id.redrhs[i + j * mumps->id.lredrhs].i;
16383e5b40d0SPierre Jolivet #endif
16393e5b40d0SPierre Jolivet           array[mumps->id.listvar_schur[i] - 1 + j * M] = val;
16403e5b40d0SPierre Jolivet         }
16413e5b40d0SPierre Jolivet     }
1642a6053eceSJunchao Zhang     if (!denseB) { /* sparse B */
16439566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJRestoreArray(Bt, &aa));
16449566063dSJacob Faibussowitsch       PetscCall(MatRestoreRowIJ(Bt, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
164528b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot restore IJ structure");
1646b8491c3eSStefano Zampini     }
16479566063dSJacob Faibussowitsch     PetscCall(MatDenseRestoreArray(X, &array));
16483ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
1649be818407SHong Zhang   }
1650801fbe65SHong Zhang 
16512ef1f0ffSBarry Smith   /* parallel case: MUMPS requires rhs B to be centralized on the host! */
1652aed4548fSBarry Smith   PetscCheck(mumps->petsc_size <= 1 || !mumps->id.ICNTL(19), PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Parallel Schur complements not yet supported from PETSc");
1653241dbb5eSStefano Zampini 
1654beae5ec0SHong Zhang   /* create msol_loc to hold mumps local solution */
16551683a169SBarry Smith   isol_loc_save = mumps->id.isol_loc; /* save it for MatSolve() */
16561683a169SBarry Smith   sol_loc_save  = (PetscScalar *)mumps->id.sol_loc;
1657801fbe65SHong Zhang 
1658a1dfcbd9SJunchao Zhang   lsol_loc  = mumps->id.lsol_loc;
165971aed81dSHong Zhang   nlsol_loc = nrhs * lsol_loc; /* length of sol_loc */
16609566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(nlsol_loc, &sol_loc, lsol_loc, &isol_loc));
1661940cd9d6SSatish Balay   mumps->id.sol_loc  = (MumpsScalar *)sol_loc;
1662801fbe65SHong Zhang   mumps->id.isol_loc = isol_loc;
1663801fbe65SHong Zhang 
16649566063dSJacob Faibussowitsch   PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, nlsol_loc, (PetscScalar *)sol_loc, &msol_loc));
16652cd7d884SHong Zhang 
166667602552SJunchao Zhang   if (denseB) {
166725aac85cSJunchao Zhang     if (mumps->ICNTL20 == 10) {
166867602552SJunchao Zhang       mumps->id.ICNTL(20) = 10; /* dense distributed RHS */
16699566063dSJacob Faibussowitsch       PetscCall(MatDenseGetArrayRead(B, &rbray));
16709566063dSJacob Faibussowitsch       PetscCall(MatMumpsSetUpDistRHSInfo(A, nrhs, rbray));
16719566063dSJacob Faibussowitsch       PetscCall(MatDenseRestoreArrayRead(B, &rbray));
16729566063dSJacob Faibussowitsch       PetscCall(MatGetLocalSize(B, &m, NULL));
16739566063dSJacob Faibussowitsch       PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)B), 1, nrhs * m, nrhs * M, NULL, &v_mpi));
167425aac85cSJunchao Zhang     } else {
167525aac85cSJunchao Zhang       mumps->id.ICNTL(20) = 0; /* dense centralized RHS */
167680577c12SJunchao Zhang       /* TODO: Because of non-contiguous indices, the created vecscatter scat_rhs is not done in MPI_Gather, resulting in
167780577c12SJunchao Zhang         very inefficient communication. An optimization is to use VecScatterCreateToZero to gather B to rank 0. Then on rank
167880577c12SJunchao Zhang         0, re-arrange B into desired order, which is a local operation.
167980577c12SJunchao Zhang       */
168080577c12SJunchao Zhang 
168167602552SJunchao Zhang       /* scatter v_mpi to b_seq because MUMPS before 5.3.0 only supports centralized rhs */
1682be818407SHong Zhang       /* wrap dense rhs matrix B into a vector v_mpi */
16839566063dSJacob Faibussowitsch       PetscCall(MatGetLocalSize(B, &m, NULL));
16849566063dSJacob Faibussowitsch       PetscCall(MatDenseGetArray(B, &bray));
16859566063dSJacob Faibussowitsch       PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)B), 1, nrhs * m, nrhs * M, (const PetscScalar *)bray, &v_mpi));
16869566063dSJacob Faibussowitsch       PetscCall(MatDenseRestoreArray(B, &bray));
16872b691707SHong Zhang 
1688be818407SHong Zhang       /* scatter v_mpi to b_seq in proc[0]. MUMPS requires rhs to be centralized on the host! */
1689801fbe65SHong Zhang       if (!mumps->myid) {
1690beae5ec0SHong Zhang         PetscInt *idx;
1691beae5ec0SHong Zhang         /* idx: maps from k-th index of v_mpi to (i,j)-th global entry of B */
16929566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(nrhs * M, &idx));
16939566063dSJacob Faibussowitsch         PetscCall(MatGetOwnershipRanges(B, &rstart));
1694be818407SHong Zhang         k = 0;
16952d4298aeSJunchao Zhang         for (proc = 0; proc < mumps->petsc_size; proc++) {
1696be818407SHong Zhang           for (j = 0; j < nrhs; j++) {
1697beae5ec0SHong Zhang             for (i = rstart[proc]; i < rstart[proc + 1]; i++) idx[k++] = j * M + i;
1698be818407SHong Zhang           }
1699be818407SHong Zhang         }
1700be818407SHong Zhang 
17019566063dSJacob Faibussowitsch         PetscCall(VecCreateSeq(PETSC_COMM_SELF, nrhs * M, &b_seq));
17029566063dSJacob Faibussowitsch         PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nrhs * M, idx, PETSC_OWN_POINTER, &is_to));
17039566063dSJacob Faibussowitsch         PetscCall(ISCreateStride(PETSC_COMM_SELF, nrhs * M, 0, 1, &is_from));
1704801fbe65SHong Zhang       } else {
17059566063dSJacob Faibussowitsch         PetscCall(VecCreateSeq(PETSC_COMM_SELF, 0, &b_seq));
17069566063dSJacob Faibussowitsch         PetscCall(ISCreateStride(PETSC_COMM_SELF, 0, 0, 1, &is_to));
17079566063dSJacob Faibussowitsch         PetscCall(ISCreateStride(PETSC_COMM_SELF, 0, 0, 1, &is_from));
1708801fbe65SHong Zhang       }
17099566063dSJacob Faibussowitsch       PetscCall(VecScatterCreate(v_mpi, is_from, b_seq, is_to, &scat_rhs));
17109566063dSJacob Faibussowitsch       PetscCall(VecScatterBegin(scat_rhs, v_mpi, b_seq, INSERT_VALUES, SCATTER_FORWARD));
17119566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_to));
17129566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_from));
17139566063dSJacob Faibussowitsch       PetscCall(VecScatterEnd(scat_rhs, v_mpi, b_seq, INSERT_VALUES, SCATTER_FORWARD));
1714801fbe65SHong Zhang 
1715801fbe65SHong Zhang       if (!mumps->myid) { /* define rhs on the host */
17169566063dSJacob Faibussowitsch         PetscCall(VecGetArray(b_seq, &bray));
1717940cd9d6SSatish Balay         mumps->id.rhs = (MumpsScalar *)bray;
17189566063dSJacob Faibussowitsch         PetscCall(VecRestoreArray(b_seq, &bray));
1719801fbe65SHong Zhang       }
172025aac85cSJunchao Zhang     }
17212b691707SHong Zhang   } else { /* sparse B */
17222b691707SHong Zhang     b = (Mat_MPIAIJ *)Bt->data;
17232b691707SHong Zhang 
1724be818407SHong Zhang     /* wrap dense X into a vector v_mpi */
17259566063dSJacob Faibussowitsch     PetscCall(MatGetLocalSize(X, &m, NULL));
17269566063dSJacob Faibussowitsch     PetscCall(MatDenseGetArray(X, &bray));
17279566063dSJacob Faibussowitsch     PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)X), 1, nrhs * m, nrhs * M, (const PetscScalar *)bray, &v_mpi));
17289566063dSJacob Faibussowitsch     PetscCall(MatDenseRestoreArray(X, &bray));
17292b691707SHong Zhang 
17302b691707SHong Zhang     if (!mumps->myid) {
17319566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJGetArray(b->A, &aa));
17329566063dSJacob Faibussowitsch       PetscCall(MatGetRowIJ(b->A, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
173328b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
17349566063dSJacob Faibussowitsch       PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs));
17352b691707SHong Zhang       mumps->id.rhs_sparse = (MumpsScalar *)aa;
17362b691707SHong Zhang     } else {
17372b691707SHong Zhang       mumps->id.irhs_ptr    = NULL;
17382b691707SHong Zhang       mumps->id.irhs_sparse = NULL;
17392b691707SHong Zhang       mumps->id.nz_rhs      = 0;
17402b691707SHong Zhang       mumps->id.rhs_sparse  = NULL;
17412b691707SHong Zhang     }
17422b691707SHong Zhang   }
17432b691707SHong Zhang 
1744801fbe65SHong Zhang   /* solve phase */
1745801fbe65SHong Zhang   mumps->id.job = JOB_SOLVE;
17463ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
17479261f6e4SBarry 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));
1748801fbe65SHong Zhang 
1749334c5f61SHong Zhang   /* scatter mumps distributed solution to petsc vector v_mpi, which shares local arrays with solution matrix X */
17509566063dSJacob Faibussowitsch   PetscCall(MatDenseGetArray(X, &array));
17519566063dSJacob Faibussowitsch   PetscCall(VecPlaceArray(v_mpi, array));
1752801fbe65SHong Zhang 
1753334c5f61SHong Zhang   /* create scatter scat_sol */
17549566063dSJacob Faibussowitsch   PetscCall(MatGetOwnershipRanges(X, &rstart));
1755beae5ec0SHong Zhang   /* iidx: index for scatter mumps solution to petsc X */
1756beae5ec0SHong Zhang 
17579566063dSJacob Faibussowitsch   PetscCall(ISCreateStride(PETSC_COMM_SELF, nlsol_loc, 0, 1, &is_from));
17589566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(nlsol_loc, &idxx));
1759beae5ec0SHong Zhang   for (i = 0; i < lsol_loc; i++) {
1760beae5ec0SHong 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 */
1761beae5ec0SHong Zhang 
17622d4298aeSJunchao Zhang     for (proc = 0; proc < mumps->petsc_size; proc++) {
1763beae5ec0SHong Zhang       if (isol_loc[i] >= rstart[proc] && isol_loc[i] < rstart[proc + 1]) {
1764beae5ec0SHong Zhang         myrstart = rstart[proc];
1765beae5ec0SHong Zhang         k        = isol_loc[i] - myrstart;          /* local index on 1st column of petsc vector X */
1766beae5ec0SHong Zhang         iidx     = k + myrstart * nrhs;             /* maps mumps isol_loc[i] to petsc index in X */
1767beae5ec0SHong Zhang         m        = rstart[proc + 1] - rstart[proc]; /* rows of X for this proc */
1768beae5ec0SHong Zhang         break;
1769be818407SHong Zhang       }
1770be818407SHong Zhang     }
1771be818407SHong Zhang 
1772beae5ec0SHong Zhang     for (j = 0; j < nrhs; j++) idxx[i + j * lsol_loc] = iidx + j * m;
1773801fbe65SHong Zhang   }
17749566063dSJacob Faibussowitsch   PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nlsol_loc, idxx, PETSC_COPY_VALUES, &is_to));
17759566063dSJacob Faibussowitsch   PetscCall(VecScatterCreate(msol_loc, is_from, v_mpi, is_to, &scat_sol));
17769566063dSJacob Faibussowitsch   PetscCall(VecScatterBegin(scat_sol, msol_loc, v_mpi, INSERT_VALUES, SCATTER_FORWARD));
17779566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&is_from));
17789566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&is_to));
17799566063dSJacob Faibussowitsch   PetscCall(VecScatterEnd(scat_sol, msol_loc, v_mpi, INSERT_VALUES, SCATTER_FORWARD));
17809566063dSJacob Faibussowitsch   PetscCall(MatDenseRestoreArray(X, &array));
178171aed81dSHong Zhang 
178271aed81dSHong Zhang   /* free spaces */
17831683a169SBarry Smith   mumps->id.sol_loc  = (MumpsScalar *)sol_loc_save;
178471aed81dSHong Zhang   mumps->id.isol_loc = isol_loc_save;
178571aed81dSHong Zhang 
17869566063dSJacob Faibussowitsch   PetscCall(PetscFree2(sol_loc, isol_loc));
17879566063dSJacob Faibussowitsch   PetscCall(PetscFree(idxx));
17889566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&msol_loc));
17899566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&v_mpi));
1790a6053eceSJunchao Zhang   if (!denseB) {
17912b691707SHong Zhang     if (!mumps->myid) {
1792d56c302dSHong Zhang       b = (Mat_MPIAIJ *)Bt->data;
17939566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJRestoreArray(b->A, &aa));
17949566063dSJacob Faibussowitsch       PetscCall(MatRestoreRowIJ(b->A, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
179528b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot restore IJ structure");
17962b691707SHong Zhang     }
17972b691707SHong Zhang   } else {
179825aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) {
17999566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b_seq));
18009566063dSJacob Faibussowitsch       PetscCall(VecScatterDestroy(&scat_rhs));
180125aac85cSJunchao Zhang     }
18022b691707SHong Zhang   }
18039566063dSJacob Faibussowitsch   PetscCall(VecScatterDestroy(&scat_sol));
180464412097SPierre 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))));
18053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1806e0b74bf9SHong Zhang }
1807e0b74bf9SHong Zhang 
180866976f2fSJacob Faibussowitsch static PetscErrorCode MatMatSolveTranspose_MUMPS(Mat A, Mat B, Mat X)
1809d71ae5a4SJacob Faibussowitsch {
1810b18964edSHong Zhang   Mat_MUMPS          *mumps = (Mat_MUMPS *)A->data;
1811338d3105SPierre Jolivet   const PetscMUMPSInt value = mumps->id.ICNTL(9);
1812b18964edSHong Zhang 
1813b18964edSHong Zhang   PetscFunctionBegin;
1814b18964edSHong Zhang   mumps->id.ICNTL(9) = 0;
1815b18964edSHong Zhang   PetscCall(MatMatSolve_MUMPS(A, B, X));
1816338d3105SPierre Jolivet   mumps->id.ICNTL(9) = value;
18173ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1818b18964edSHong Zhang }
1819b18964edSHong Zhang 
182066976f2fSJacob Faibussowitsch static PetscErrorCode MatMatTransposeSolve_MUMPS(Mat A, Mat Bt, Mat X)
1821d71ae5a4SJacob Faibussowitsch {
1822eb3ef3b2SHong Zhang   PetscBool flg;
1823eb3ef3b2SHong Zhang   Mat       B;
1824eb3ef3b2SHong Zhang 
1825eb3ef3b2SHong Zhang   PetscFunctionBegin;
18269566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)Bt, &flg, MATSEQAIJ, MATMPIAIJ, NULL));
182728b400f6SJacob Faibussowitsch   PetscCheck(flg, PetscObjectComm((PetscObject)Bt), PETSC_ERR_ARG_WRONG, "Matrix Bt must be MATAIJ matrix");
1828eb3ef3b2SHong Zhang 
1829eb3ef3b2SHong Zhang   /* Create B=Bt^T that uses Bt's data structure */
18309566063dSJacob Faibussowitsch   PetscCall(MatCreateTranspose(Bt, &B));
1831eb3ef3b2SHong Zhang 
18329566063dSJacob Faibussowitsch   PetscCall(MatMatSolve_MUMPS(A, B, X));
18339566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&B));
18343ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1835eb3ef3b2SHong Zhang }
1836eb3ef3b2SHong Zhang 
1837ace3df97SHong Zhang #if !defined(PETSC_USE_COMPLEX)
1838a58c3f20SHong Zhang /*
1839a58c3f20SHong Zhang   input:
1840a58c3f20SHong Zhang    F:        numeric factor
1841a58c3f20SHong Zhang   output:
1842a58c3f20SHong Zhang    nneg:     total number of negative pivots
184319d49a3bSHong Zhang    nzero:    total number of zero pivots
184419d49a3bSHong Zhang    npos:     (global dimension of F) - nneg - nzero
1845a58c3f20SHong Zhang */
184666976f2fSJacob Faibussowitsch static PetscErrorCode MatGetInertia_SBAIJMUMPS(Mat F, PetscInt *nneg, PetscInt *nzero, PetscInt *npos)
1847d71ae5a4SJacob Faibussowitsch {
1848e69c285eSBarry Smith   Mat_MUMPS  *mumps = (Mat_MUMPS *)F->data;
1849c1490034SHong Zhang   PetscMPIInt size;
1850a58c3f20SHong Zhang 
1851a58c3f20SHong Zhang   PetscFunctionBegin;
18529566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)F), &size));
1853bcb30aebSHong 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 */
1854aed4548fSBarry 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));
1855ed85ac9fSHong Zhang 
1856710ac8efSHong Zhang   if (nneg) *nneg = mumps->id.INFOG(12);
1857ed85ac9fSHong Zhang   if (nzero || npos) {
185808401ef6SPierre 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");
1859710ac8efSHong Zhang     if (nzero) *nzero = mumps->id.INFOG(28);
1860710ac8efSHong Zhang     if (npos) *npos = F->rmap->N - (mumps->id.INFOG(12) + mumps->id.INFOG(28));
1861a58c3f20SHong Zhang   }
18623ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1863a58c3f20SHong Zhang }
186419d49a3bSHong Zhang #endif
1865a58c3f20SHong Zhang 
186666976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGatherNonzerosOnMaster(MatReuse reuse, Mat_MUMPS *mumps)
1867d71ae5a4SJacob Faibussowitsch {
1868a6053eceSJunchao Zhang   PetscInt       i, nreqs;
1869a6053eceSJunchao Zhang   PetscMUMPSInt *irn, *jcn;
1870a6053eceSJunchao Zhang   PetscMPIInt    count;
1871a6053eceSJunchao Zhang   PetscInt64     totnnz, remain;
1872a6053eceSJunchao Zhang   const PetscInt osize = mumps->omp_comm_size;
1873a6053eceSJunchao Zhang   PetscScalar   *val;
18743ab56b82SJunchao Zhang 
18753ab56b82SJunchao Zhang   PetscFunctionBegin;
1876a6053eceSJunchao Zhang   if (osize > 1) {
18773ab56b82SJunchao Zhang     if (reuse == MAT_INITIAL_MATRIX) {
18783ab56b82SJunchao Zhang       /* master first gathers counts of nonzeros to receive */
18799566063dSJacob Faibussowitsch       if (mumps->is_omp_master) PetscCall(PetscMalloc1(osize, &mumps->recvcount));
18809566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Gather(&mumps->nnz, 1, MPIU_INT64, mumps->recvcount, 1, MPIU_INT64, 0 /*master*/, mumps->omp_comm));
18813ab56b82SJunchao Zhang 
1882a6053eceSJunchao Zhang       /* Then each computes number of send/recvs */
18833ab56b82SJunchao Zhang       if (mumps->is_omp_master) {
1884a6053eceSJunchao Zhang         /* Start from 1 since self communication is not done in MPI */
1885a6053eceSJunchao Zhang         nreqs = 0;
1886a6053eceSJunchao Zhang         for (i = 1; i < osize; i++) nreqs += (mumps->recvcount[i] + PETSC_MPI_INT_MAX - 1) / PETSC_MPI_INT_MAX;
1887a6053eceSJunchao Zhang       } else {
1888a6053eceSJunchao Zhang         nreqs = (mumps->nnz + PETSC_MPI_INT_MAX - 1) / PETSC_MPI_INT_MAX;
18893ab56b82SJunchao Zhang       }
189035cb6cd3SPierre Jolivet       PetscCall(PetscMalloc1(nreqs * 3, &mumps->reqs)); /* Triple the requests since we send irn, jcn and val separately */
18913ab56b82SJunchao Zhang 
1892a6053eceSJunchao Zhang       /* The following code is doing a very simple thing: omp_master rank gathers irn/jcn/val from others.
1893a6053eceSJunchao Zhang          MPI_Gatherv would be enough if it supports big counts > 2^31-1. Since it does not, and mumps->nnz
1894a6053eceSJunchao Zhang          might be a prime number > 2^31-1, we have to slice the message. Note omp_comm_size
1895a6053eceSJunchao Zhang          is very small, the current approach should have no extra overhead compared to MPI_Gatherv.
1896a6053eceSJunchao Zhang        */
1897a6053eceSJunchao Zhang       nreqs = 0; /* counter for actual send/recvs */
18983ab56b82SJunchao Zhang       if (mumps->is_omp_master) {
1899a6053eceSJunchao Zhang         for (i = 0, totnnz = 0; i < osize; i++) totnnz += mumps->recvcount[i]; /* totnnz = sum of nnz over omp_comm */
19009566063dSJacob Faibussowitsch         PetscCall(PetscMalloc2(totnnz, &irn, totnnz, &jcn));
19019566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(totnnz, &val));
1902a6053eceSJunchao Zhang 
1903a6053eceSJunchao Zhang         /* Self communication */
19049566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(irn, mumps->irn, mumps->nnz));
19059566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(jcn, mumps->jcn, mumps->nnz));
19069566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(val, mumps->val, mumps->nnz));
1907a6053eceSJunchao Zhang 
1908a6053eceSJunchao Zhang         /* Replace mumps->irn/jcn etc on master with the newly allocated bigger arrays */
19099566063dSJacob Faibussowitsch         PetscCall(PetscFree2(mumps->irn, mumps->jcn));
19109566063dSJacob Faibussowitsch         PetscCall(PetscFree(mumps->val_alloc));
1911a6053eceSJunchao Zhang         mumps->nnz = totnnz;
19123ab56b82SJunchao Zhang         mumps->irn = irn;
19133ab56b82SJunchao Zhang         mumps->jcn = jcn;
1914a6053eceSJunchao Zhang         mumps->val = mumps->val_alloc = val;
1915a6053eceSJunchao Zhang 
1916a6053eceSJunchao Zhang         irn += mumps->recvcount[0]; /* recvcount[0] is old mumps->nnz on omp rank 0 */
1917a6053eceSJunchao Zhang         jcn += mumps->recvcount[0];
1918a6053eceSJunchao Zhang         val += mumps->recvcount[0];
1919a6053eceSJunchao Zhang 
1920a6053eceSJunchao Zhang         /* Remote communication */
1921a6053eceSJunchao Zhang         for (i = 1; i < osize; i++) {
1922a6053eceSJunchao Zhang           count  = PetscMin(mumps->recvcount[i], PETSC_MPI_INT_MAX);
1923a6053eceSJunchao Zhang           remain = mumps->recvcount[i] - count;
1924a6053eceSJunchao Zhang           while (count > 0) {
19259566063dSJacob Faibussowitsch             PetscCallMPI(MPI_Irecv(irn, count, MPIU_MUMPSINT, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19269566063dSJacob Faibussowitsch             PetscCallMPI(MPI_Irecv(jcn, count, MPIU_MUMPSINT, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19279566063dSJacob Faibussowitsch             PetscCallMPI(MPI_Irecv(val, count, MPIU_SCALAR, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1928a6053eceSJunchao Zhang             irn += count;
1929a6053eceSJunchao Zhang             jcn += count;
1930a6053eceSJunchao Zhang             val += count;
1931a6053eceSJunchao Zhang             count = PetscMin(remain, PETSC_MPI_INT_MAX);
1932a6053eceSJunchao Zhang             remain -= count;
1933a6053eceSJunchao Zhang           }
19343ab56b82SJunchao Zhang         }
19353ab56b82SJunchao Zhang       } else {
1936a6053eceSJunchao Zhang         irn    = mumps->irn;
1937a6053eceSJunchao Zhang         jcn    = mumps->jcn;
1938a6053eceSJunchao Zhang         val    = mumps->val;
1939a6053eceSJunchao Zhang         count  = PetscMin(mumps->nnz, PETSC_MPI_INT_MAX);
1940a6053eceSJunchao Zhang         remain = mumps->nnz - count;
1941a6053eceSJunchao Zhang         while (count > 0) {
19429566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Isend(irn, count, MPIU_MUMPSINT, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19439566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Isend(jcn, count, MPIU_MUMPSINT, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19449566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Isend(val, count, MPIU_SCALAR, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1945a6053eceSJunchao Zhang           irn += count;
1946a6053eceSJunchao Zhang           jcn += count;
1947a6053eceSJunchao Zhang           val += count;
1948a6053eceSJunchao Zhang           count = PetscMin(remain, PETSC_MPI_INT_MAX);
1949a6053eceSJunchao Zhang           remain -= count;
19503ab56b82SJunchao Zhang         }
19513ab56b82SJunchao Zhang       }
1952a6053eceSJunchao Zhang     } else {
1953a6053eceSJunchao Zhang       nreqs = 0;
1954a6053eceSJunchao Zhang       if (mumps->is_omp_master) {
1955a6053eceSJunchao Zhang         val = mumps->val + mumps->recvcount[0];
1956a6053eceSJunchao Zhang         for (i = 1; i < osize; i++) { /* Remote communication only since self data is already in place */
1957a6053eceSJunchao Zhang           count  = PetscMin(mumps->recvcount[i], PETSC_MPI_INT_MAX);
1958a6053eceSJunchao Zhang           remain = mumps->recvcount[i] - count;
1959a6053eceSJunchao Zhang           while (count > 0) {
19609566063dSJacob Faibussowitsch             PetscCallMPI(MPI_Irecv(val, count, MPIU_SCALAR, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1961a6053eceSJunchao Zhang             val += count;
1962a6053eceSJunchao Zhang             count = PetscMin(remain, PETSC_MPI_INT_MAX);
1963a6053eceSJunchao Zhang             remain -= count;
1964a6053eceSJunchao Zhang           }
1965a6053eceSJunchao Zhang         }
1966a6053eceSJunchao Zhang       } else {
1967a6053eceSJunchao Zhang         val    = mumps->val;
1968a6053eceSJunchao Zhang         count  = PetscMin(mumps->nnz, PETSC_MPI_INT_MAX);
1969a6053eceSJunchao Zhang         remain = mumps->nnz - count;
1970a6053eceSJunchao Zhang         while (count > 0) {
19719566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Isend(val, count, MPIU_SCALAR, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1972a6053eceSJunchao Zhang           val += count;
1973a6053eceSJunchao Zhang           count = PetscMin(remain, PETSC_MPI_INT_MAX);
1974a6053eceSJunchao Zhang           remain -= count;
1975a6053eceSJunchao Zhang         }
1976a6053eceSJunchao Zhang       }
1977a6053eceSJunchao Zhang     }
19789566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Waitall(nreqs, mumps->reqs, MPI_STATUSES_IGNORE));
1979a6053eceSJunchao Zhang     mumps->tag++; /* It is totally fine for above send/recvs to share one mpi tag */
1980a6053eceSJunchao Zhang   }
19813ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
19823ab56b82SJunchao Zhang }
19833ab56b82SJunchao Zhang 
198466976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorNumeric_MUMPS(Mat F, Mat A, const MatFactorInfo *info)
1985d71ae5a4SJacob Faibussowitsch {
1986e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)(F)->data;
1987ace3abfcSBarry Smith   PetscBool  isMPIAIJ;
1988397b6df1SKris Buschelman 
1989397b6df1SKris Buschelman   PetscFunctionBegin;
1990dbf6bb8dSprj-   if (mumps->id.INFOG(1) < 0 && !(mumps->id.INFOG(1) == -16 && mumps->id.INFOG(1) == 0)) {
199148a46eb9SPierre 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)));
19929566063dSJacob 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)));
19933ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
19942aca8efcSHong Zhang   }
19956baea169SHong Zhang 
19969566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_REUSE_MATRIX, mumps));
19979566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_REUSE_MATRIX, mumps));
1998397b6df1SKris Buschelman 
1999397b6df1SKris Buschelman   /* numerical factorization phase */
2000a5e57a09SHong Zhang   mumps->id.job = JOB_FACTNUMERIC;
20014e34a73bSHong Zhang   if (!mumps->id.ICNTL(18)) { /* A is centralized */
2002ad540459SPierre Jolivet     if (!mumps->myid) mumps->id.a = (MumpsScalar *)mumps->val;
2003397b6df1SKris Buschelman   } else {
2004940cd9d6SSatish Balay     mumps->id.a_loc = (MumpsScalar *)mumps->val;
2005397b6df1SKris Buschelman   }
20063ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
2007a5e57a09SHong Zhang   if (mumps->id.INFOG(1) < 0) {
20089261f6e4SBarry 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));
20099261f6e4SBarry Smith     if (mumps->id.INFOG(1) == -10) {
20109261f6e4SBarry 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)));
2011603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
2012c0d63f2fSHong Zhang     } else if (mumps->id.INFOG(1) == -13) {
20139261f6e4SBarry 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)));
2014603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OUTMEMORY;
2015c0d63f2fSHong Zhang     } else if (mumps->id.INFOG(1) == -8 || mumps->id.INFOG(1) == -9 || (-16 < mumps->id.INFOG(1) && mumps->id.INFOG(1) < -10)) {
20169261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in numerical factorizatione: INFOG(1)=%d, INFO(2)=%d, problem with work array\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
2017603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OUTMEMORY;
20182aca8efcSHong Zhang     } else {
20199261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in numerical factorization: INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
2020603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OTHER;
2021151787a6SHong Zhang     }
20222aca8efcSHong Zhang   }
20239261f6e4SBarry 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));
2024397b6df1SKris Buschelman 
2025b3cb21ddSStefano Zampini   F->assembled = PETSC_TRUE;
2026d47f36abSHong Zhang 
2027b3cb21ddSStefano Zampini   if (F->schur) { /* reset Schur status to unfactored */
20283cb7dd0eSStefano Zampini #if defined(PETSC_HAVE_CUDA)
2029c70f7ee4SJunchao Zhang     F->schur->offloadmask = PETSC_OFFLOAD_CPU;
20303cb7dd0eSStefano Zampini #endif
2031b3cb21ddSStefano Zampini     if (mumps->id.ICNTL(19) == 1) { /* stored by rows */
2032b3cb21ddSStefano Zampini       mumps->id.ICNTL(19) = 2;
20339566063dSJacob Faibussowitsch       PetscCall(MatTranspose(F->schur, MAT_INPLACE_MATRIX, &F->schur));
2034b3cb21ddSStefano Zampini     }
20359566063dSJacob Faibussowitsch     PetscCall(MatFactorRestoreSchurComplement(F, NULL, MAT_FACTOR_SCHUR_UNFACTORED));
2036b3cb21ddSStefano Zampini   }
203767877ebaSShri Abhyankar 
2038066565c5SStefano Zampini   /* just to be sure that ICNTL(19) value returned by a call from MatMumpsGetIcntl is always consistent */
2039066565c5SStefano Zampini   if (!mumps->sym && mumps->id.ICNTL(19) && mumps->id.ICNTL(19) != 1) mumps->id.ICNTL(19) = 3;
2040066565c5SStefano Zampini 
20413ab56b82SJunchao Zhang   if (!mumps->is_omp_master) mumps->id.INFO(23) = 0;
20422d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) {
204367877ebaSShri Abhyankar     PetscInt     lsol_loc;
204467877ebaSShri Abhyankar     PetscScalar *sol_loc;
20452205254eSKarl Rupp 
20469566063dSJacob Faibussowitsch     PetscCall(PetscObjectTypeCompare((PetscObject)A, MATMPIAIJ, &isMPIAIJ));
2047c2093ab7SHong Zhang 
2048c2093ab7SHong Zhang     /* distributed solution; Create x_seq=sol_loc for repeated use */
2049c2093ab7SHong Zhang     if (mumps->x_seq) {
20509566063dSJacob Faibussowitsch       PetscCall(VecScatterDestroy(&mumps->scat_sol));
20519566063dSJacob Faibussowitsch       PetscCall(PetscFree2(mumps->id.sol_loc, mumps->id.isol_loc));
20529566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mumps->x_seq));
2053c2093ab7SHong Zhang     }
2054a5e57a09SHong Zhang     lsol_loc = mumps->id.INFO(23); /* length of sol_loc */
20559566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(lsol_loc, &sol_loc, lsol_loc, &mumps->id.isol_loc));
2056a5e57a09SHong Zhang     mumps->id.lsol_loc = lsol_loc;
2057940cd9d6SSatish Balay     mumps->id.sol_loc  = (MumpsScalar *)sol_loc;
20589566063dSJacob Faibussowitsch     PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, lsol_loc, sol_loc, &mumps->x_seq));
205967877ebaSShri Abhyankar   }
20609566063dSJacob Faibussowitsch   PetscCall(PetscLogFlops(mumps->id.RINFO(2)));
20613ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2062397b6df1SKris Buschelman }
2063397b6df1SKris Buschelman 
20649a2535b5SHong Zhang /* Sets MUMPS options from the options database */
206566976f2fSJacob Faibussowitsch static PetscErrorCode MatSetFromOptions_MUMPS(Mat F, Mat A)
2066d71ae5a4SJacob Faibussowitsch {
2067e69c285eSBarry Smith   Mat_MUMPS    *mumps = (Mat_MUMPS *)F->data;
2068413bcc21SPierre Jolivet   PetscMUMPSInt icntl = 0, size, *listvar_schur;
206945e3843bSPierre Jolivet   PetscInt      info[80], i, ninfo = 80, rbs, cbs;
2070413bcc21SPierre Jolivet   PetscBool     flg = PETSC_FALSE, schur = (PetscBool)(mumps->id.ICNTL(26) == -1);
2071413bcc21SPierre Jolivet   MumpsScalar  *arr;
2072dcd589f8SShri Abhyankar 
2073dcd589f8SShri Abhyankar   PetscFunctionBegin;
207426cc229bSBarry Smith   PetscOptionsBegin(PetscObjectComm((PetscObject)F), ((PetscObject)F)->prefix, "MUMPS Options", "Mat");
2075413bcc21SPierre Jolivet   if (mumps->id.job == JOB_NULL) { /* MatSetFromOptions_MUMPS() has never been called before */
2076413bcc21SPierre Jolivet     PetscInt nthreads   = 0;
2077413bcc21SPierre Jolivet     PetscInt nCNTL_pre  = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0;
2078413bcc21SPierre Jolivet     PetscInt nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0;
2079413bcc21SPierre Jolivet 
2080413bcc21SPierre Jolivet     mumps->petsc_comm = PetscObjectComm((PetscObject)A);
2081413bcc21SPierre Jolivet     PetscCallMPI(MPI_Comm_size(mumps->petsc_comm, &mumps->petsc_size));
2082413bcc21SPierre Jolivet     PetscCallMPI(MPI_Comm_rank(mumps->petsc_comm, &mumps->myid)); /* "if (!myid)" still works even if mumps_comm is different */
2083413bcc21SPierre Jolivet 
2084413bcc21SPierre Jolivet     PetscCall(PetscOptionsName("-mat_mumps_use_omp_threads", "Convert MPI processes into OpenMP threads", "None", &mumps->use_petsc_omp_support));
2085413bcc21SPierre Jolivet     if (mumps->use_petsc_omp_support) nthreads = -1; /* -1 will let PetscOmpCtrlCreate() guess a proper value when user did not supply one */
2086413bcc21SPierre Jolivet     /* do not use PetscOptionsInt() so that the option -mat_mumps_use_omp_threads is not displayed twice in the help */
2087413bcc21SPierre Jolivet     PetscCall(PetscOptionsGetInt(NULL, ((PetscObject)F)->prefix, "-mat_mumps_use_omp_threads", &nthreads, NULL));
2088413bcc21SPierre Jolivet     if (mumps->use_petsc_omp_support) {
20899371c9d4SSatish Balay       PetscCheck(PetscDefined(HAVE_OPENMP_SUPPORT), 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",
20909371c9d4SSatish Balay                  ((PetscObject)F)->prefix ? ((PetscObject)F)->prefix : "");
2091413bcc21SPierre 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 : "");
2092413bcc21SPierre Jolivet #if defined(PETSC_HAVE_OPENMP_SUPPORT)
2093413bcc21SPierre Jolivet       PetscCall(PetscOmpCtrlCreate(mumps->petsc_comm, nthreads, &mumps->omp_ctrl));
2094413bcc21SPierre Jolivet       PetscCall(PetscOmpCtrlGetOmpComms(mumps->omp_ctrl, &mumps->omp_comm, &mumps->mumps_comm, &mumps->is_omp_master));
2095413bcc21SPierre Jolivet #endif
2096413bcc21SPierre Jolivet     } else {
2097413bcc21SPierre Jolivet       mumps->omp_comm      = PETSC_COMM_SELF;
2098413bcc21SPierre Jolivet       mumps->mumps_comm    = mumps->petsc_comm;
2099413bcc21SPierre Jolivet       mumps->is_omp_master = PETSC_TRUE;
2100413bcc21SPierre Jolivet     }
2101413bcc21SPierre Jolivet     PetscCallMPI(MPI_Comm_size(mumps->omp_comm, &mumps->omp_comm_size));
2102413bcc21SPierre Jolivet     mumps->reqs = NULL;
2103413bcc21SPierre Jolivet     mumps->tag  = 0;
2104413bcc21SPierre Jolivet 
2105413bcc21SPierre Jolivet     if (mumps->mumps_comm != MPI_COMM_NULL) {
2106413bcc21SPierre Jolivet       if (PetscDefined(HAVE_OPENMP_SUPPORT) && mumps->use_petsc_omp_support) {
2107413bcc21SPierre Jolivet         /* It looks like MUMPS does not dup the input comm. Dup a new comm for MUMPS to avoid any tag mismatches. */
2108413bcc21SPierre Jolivet         MPI_Comm comm;
2109413bcc21SPierre Jolivet         PetscCallMPI(MPI_Comm_dup(mumps->mumps_comm, &comm));
2110413bcc21SPierre Jolivet         mumps->mumps_comm = comm;
2111413bcc21SPierre Jolivet       } else PetscCall(PetscCommGetComm(mumps->petsc_comm, &mumps->mumps_comm));
2112413bcc21SPierre Jolivet     }
2113413bcc21SPierre Jolivet 
2114413bcc21SPierre Jolivet     mumps->id.comm_fortran = MPI_Comm_c2f(mumps->mumps_comm);
2115413bcc21SPierre Jolivet     mumps->id.job          = JOB_INIT;
2116413bcc21SPierre Jolivet     mumps->id.par          = 1; /* host participates factorizaton and solve */
2117413bcc21SPierre Jolivet     mumps->id.sym          = mumps->sym;
2118413bcc21SPierre Jolivet 
2119413bcc21SPierre Jolivet     size          = mumps->id.size_schur;
2120413bcc21SPierre Jolivet     arr           = mumps->id.schur;
2121413bcc21SPierre Jolivet     listvar_schur = mumps->id.listvar_schur;
2122413bcc21SPierre Jolivet     PetscMUMPS_c(mumps);
21239261f6e4SBarry Smith     PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
2124413bcc21SPierre Jolivet     /* restore cached ICNTL and CNTL values */
2125413bcc21SPierre Jolivet     for (icntl = 0; icntl < nICNTL_pre; ++icntl) mumps->id.ICNTL(mumps->ICNTL_pre[1 + 2 * icntl]) = mumps->ICNTL_pre[2 + 2 * icntl];
2126413bcc21SPierre Jolivet     for (icntl = 0; icntl < nCNTL_pre; ++icntl) mumps->id.CNTL((PetscInt)mumps->CNTL_pre[1 + 2 * icntl]) = mumps->CNTL_pre[2 + 2 * icntl];
2127413bcc21SPierre Jolivet     PetscCall(PetscFree(mumps->ICNTL_pre));
2128413bcc21SPierre Jolivet     PetscCall(PetscFree(mumps->CNTL_pre));
2129413bcc21SPierre Jolivet 
2130413bcc21SPierre Jolivet     if (schur) {
2131413bcc21SPierre Jolivet       mumps->id.size_schur    = size;
2132413bcc21SPierre Jolivet       mumps->id.schur_lld     = size;
2133413bcc21SPierre Jolivet       mumps->id.schur         = arr;
2134413bcc21SPierre Jolivet       mumps->id.listvar_schur = listvar_schur;
2135413bcc21SPierre Jolivet       if (mumps->petsc_size > 1) {
2136413bcc21SPierre Jolivet         PetscBool gs; /* gs is false if any rank other than root has non-empty IS */
2137413bcc21SPierre Jolivet 
2138413bcc21SPierre Jolivet         mumps->id.ICNTL(19) = 1;                                                                            /* MUMPS returns Schur centralized on the host */
2139413bcc21SPierre 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 */
2140712fec58SPierre Jolivet         PetscCall(MPIU_Allreduce(MPI_IN_PLACE, &gs, 1, MPIU_BOOL, MPI_LAND, mumps->petsc_comm));
2141413bcc21SPierre Jolivet         PetscCheck(gs, PETSC_COMM_SELF, PETSC_ERR_SUP, "MUMPS distributed parallel Schur complements not yet supported from PETSc");
2142413bcc21SPierre Jolivet       } else {
2143413bcc21SPierre Jolivet         if (F->factortype == MAT_FACTOR_LU) {
2144413bcc21SPierre Jolivet           mumps->id.ICNTL(19) = 3; /* MUMPS returns full matrix */
2145413bcc21SPierre Jolivet         } else {
2146413bcc21SPierre Jolivet           mumps->id.ICNTL(19) = 2; /* MUMPS returns lower triangular part */
2147413bcc21SPierre Jolivet         }
2148413bcc21SPierre Jolivet       }
2149413bcc21SPierre Jolivet       mumps->id.ICNTL(26) = -1;
2150413bcc21SPierre Jolivet     }
2151413bcc21SPierre Jolivet 
2152413bcc21SPierre Jolivet     /* copy MUMPS default control values from master to slaves. Although slaves do not call MUMPS, they may access these values in code.
2153413bcc21SPierre Jolivet        For example, ICNTL(9) is initialized to 1 by MUMPS and slaves check ICNTL(9) in MatSolve_MUMPS.
2154413bcc21SPierre Jolivet      */
2155413bcc21SPierre Jolivet     PetscCallMPI(MPI_Bcast(mumps->id.icntl, 40, MPI_INT, 0, mumps->omp_comm));
2156413bcc21SPierre Jolivet     PetscCallMPI(MPI_Bcast(mumps->id.cntl, 15, MPIU_REAL, 0, mumps->omp_comm));
2157413bcc21SPierre Jolivet 
2158413bcc21SPierre Jolivet     mumps->scat_rhs = NULL;
2159413bcc21SPierre Jolivet     mumps->scat_sol = NULL;
2160413bcc21SPierre Jolivet 
2161413bcc21SPierre Jolivet     /* set PETSc-MUMPS default options - override MUMPS default */
2162413bcc21SPierre Jolivet     mumps->id.ICNTL(3) = 0;
2163413bcc21SPierre Jolivet     mumps->id.ICNTL(4) = 0;
2164413bcc21SPierre Jolivet     if (mumps->petsc_size == 1) {
2165413bcc21SPierre Jolivet       mumps->id.ICNTL(18) = 0; /* centralized assembled matrix input */
2166413bcc21SPierre Jolivet       mumps->id.ICNTL(7)  = 7; /* automatic choice of ordering done by the package */
2167413bcc21SPierre Jolivet     } else {
2168413bcc21SPierre Jolivet       mumps->id.ICNTL(18) = 3; /* distributed assembled matrix input */
2169413bcc21SPierre Jolivet       mumps->id.ICNTL(21) = 1; /* distributed solution */
2170413bcc21SPierre Jolivet     }
2171413bcc21SPierre Jolivet   }
21729566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_1", "ICNTL(1): output stream for error messages", "None", mumps->id.ICNTL(1), &icntl, &flg));
21739a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(1) = icntl;
21749566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_2", "ICNTL(2): output stream for diagnostic printing, statistics, and warning", "None", mumps->id.ICNTL(2), &icntl, &flg));
21759a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(2) = icntl;
21769566063dSJacob 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));
21779a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(3) = icntl;
2178dcd589f8SShri Abhyankar 
21799566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_4", "ICNTL(4): level of printing (0 to 4)", "None", mumps->id.ICNTL(4), &icntl, &flg));
21809a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(4) = icntl;
21819a2535b5SHong Zhang   if (mumps->id.ICNTL(4) || PetscLogPrintInfo) mumps->id.ICNTL(3) = 6; /* resume MUMPS default id.ICNTL(3) = 6 */
21829a2535b5SHong Zhang 
21839566063dSJacob 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));
21849a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(6) = icntl;
21859a2535b5SHong Zhang 
21869566063dSJacob 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));
2187dcd589f8SShri Abhyankar   if (flg) {
2188aed4548fSBarry 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");
2189b53c1a7fSBarry Smith     mumps->id.ICNTL(7) = icntl;
2190dcd589f8SShri Abhyankar   }
2191e0b74bf9SHong Zhang 
21929566063dSJacob 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));
21939566063dSJacob 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() */
21949566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_10", "ICNTL(10): max num of refinements", "None", mumps->id.ICNTL(10), &mumps->id.ICNTL(10), NULL));
21959566063dSJacob 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));
21969566063dSJacob 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));
21979566063dSJacob 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));
21989566063dSJacob 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));
219945e3843bSPierre Jolivet   PetscCall(MatGetBlockSizes(A, &rbs, &cbs));
220045e3843bSPierre Jolivet   if (rbs == cbs && rbs > 1) mumps->id.ICNTL(15) = -rbs;
220145e3843bSPierre 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));
220245e3843bSPierre Jolivet   if (flg) {
220345e3843bSPierre Jolivet     PetscCheck(mumps->id.ICNTL(15) <= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Positive -mat_mumps_icntl_15 not handled");
220445e3843bSPierre 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");
220545e3843bSPierre Jolivet   }
22069566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_19", "ICNTL(19): computes the Schur complement", "None", mumps->id.ICNTL(19), &mumps->id.ICNTL(19), NULL));
220759ac8732SStefano Zampini   if (mumps->id.ICNTL(19) <= 0 || mumps->id.ICNTL(19) > 3) { /* reset any schur data (if any) */
22089566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&F->schur));
22099566063dSJacob Faibussowitsch     PetscCall(MatMumpsResetSchur_Private(mumps));
221059ac8732SStefano Zampini   }
221125aac85cSJunchao Zhang 
221243f3b051SJunchao Zhang   /* Two MPICH Fortran MPI_IN_PLACE binding bugs prevented the use of 'mpich + mumps'. One happened with "mpi4py + mpich + mumps",
221343f3b051SJunchao Zhang      and was reported by Firedrake. See https://bitbucket.org/mpi4py/mpi4py/issues/162/mpi4py-initialization-breaks-fortran
221425aac85cSJunchao Zhang      and a petsc-maint mailing list thread with subject 'MUMPS segfaults in parallel because of ...'
221543f3b051SJunchao Zhang      This bug was fixed by https://github.com/pmodels/mpich/pull/4149. But the fix brought a new bug,
221643f3b051SJunchao Zhang      see https://github.com/pmodels/mpich/issues/5589. This bug was fixed by https://github.com/pmodels/mpich/pull/5590.
221741caa250SJunchao Zhang      In short, we could not use distributed RHS until with MPICH v4.0b1 or we enabled a workaround in mumps-5.6.2+
221825aac85cSJunchao Zhang    */
221941caa250SJunchao Zhang #if PETSC_PKG_MUMPS_VERSION_GE(5, 6, 2) && defined(PETSC_HAVE_MUMPS_AVOID_MPI_IN_PLACE)
222041caa250SJunchao Zhang   mumps->ICNTL20 = 10;
222141caa250SJunchao Zhang #elif PETSC_PKG_MUMPS_VERSION_LT(5, 3, 0) || (defined(PETSC_HAVE_MPICH_NUMVERSION) && (PETSC_HAVE_MPICH_NUMVERSION < 40000101))
222225aac85cSJunchao Zhang   mumps->ICNTL20     = 0; /* Centralized dense RHS*/
222343f3b051SJunchao Zhang #else
222443f3b051SJunchao Zhang   mumps->ICNTL20 = 10; /* Distributed dense RHS*/
222525aac85cSJunchao Zhang #endif
22269566063dSJacob 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));
2227aed4548fSBarry 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);
222825aac85cSJunchao Zhang #if PETSC_PKG_MUMPS_VERSION_LT(5, 3, 0)
2229aed4548fSBarry Smith   PetscCheck(!flg || mumps->ICNTL20 != 10, PETSC_COMM_SELF, PETSC_ERR_SUP, "ICNTL(20)=10 is not supported before MUMPS-5.3.0");
223025aac85cSJunchao Zhang #endif
22319566063dSJacob 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 */
22329a2535b5SHong Zhang 
22339566063dSJacob 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));
22349566063dSJacob 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));
22359566063dSJacob 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));
22369371c9d4SSatish Balay   if (mumps->id.ICNTL(24)) { mumps->id.ICNTL(13) = 1; /* turn-off ScaLAPACK to help with the correct detection of null pivots */ }
2237d7ebd59bSHong Zhang 
22389566063dSJacob 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));
22399566063dSJacob 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));
22409566063dSJacob 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));
22419566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_28", "ICNTL(28): use 1 for sequential analysis and ictnl(7) ordering, or 2 for parallel analysis and ictnl(29) ordering", "None", mumps->id.ICNTL(28), &mumps->id.ICNTL(28), NULL));
22429566063dSJacob 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));
22439566063dSJacob 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 */
22449566063dSJacob 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));
2245145b44c9SPierre 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 */
22469566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_33", "ICNTL(33): compute determinant", "None", mumps->id.ICNTL(33), &mumps->id.ICNTL(33), NULL));
22479566063dSJacob 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));
22489566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_36", "ICNTL(36): choice of BLR factorization variant", "None", mumps->id.ICNTL(36), &mumps->id.ICNTL(36), NULL));
22499566063dSJacob 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));
2250146931dbSPierre Jolivet   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_58", "ICNTL(58): defines options for symbolic factorization", "None", mumps->id.ICNTL(58), &mumps->id.ICNTL(58), NULL));
2251dcd589f8SShri Abhyankar 
22529566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_1", "CNTL(1): relative pivoting threshold", "None", mumps->id.CNTL(1), &mumps->id.CNTL(1), NULL));
22539566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_2", "CNTL(2): stopping criterion of refinement", "None", mumps->id.CNTL(2), &mumps->id.CNTL(2), NULL));
22549566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_3", "CNTL(3): absolute pivoting threshold", "None", mumps->id.CNTL(3), &mumps->id.CNTL(3), NULL));
22559566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_4", "CNTL(4): value for static pivoting", "None", mumps->id.CNTL(4), &mumps->id.CNTL(4), NULL));
22569566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_5", "CNTL(5): fixation for null pivots", "None", mumps->id.CNTL(5), &mumps->id.CNTL(5), NULL));
22579566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_7", "CNTL(7): dropping parameter used during BLR", "None", mumps->id.CNTL(7), &mumps->id.CNTL(7), NULL));
2258e5bb22a1SHong Zhang 
22599566063dSJacob 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));
2260b34f08ffSHong Zhang 
22619566063dSJacob Faibussowitsch   PetscCall(PetscOptionsIntArray("-mat_mumps_view_info", "request INFO local to each processor", "", info, &ninfo, NULL));
2262b34f08ffSHong Zhang   if (ninfo) {
226308401ef6SPierre Jolivet     PetscCheck(ninfo <= 80, PETSC_COMM_SELF, PETSC_ERR_USER, "number of INFO %" PetscInt_FMT " must <= 80", ninfo);
22649566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(ninfo, &mumps->info));
2265b34f08ffSHong Zhang     mumps->ninfo = ninfo;
2266b34f08ffSHong Zhang     for (i = 0; i < ninfo; i++) {
2267aed4548fSBarry 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);
2268f7d195e4SLawrence Mitchell       mumps->info[i] = info[i];
2269b34f08ffSHong Zhang     }
2270b34f08ffSHong Zhang   }
2271d0609cedSBarry Smith   PetscOptionsEnd();
22723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2273dcd589f8SShri Abhyankar }
2274dcd589f8SShri Abhyankar 
227566976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorSymbolic_MUMPS_ReportIfError(Mat F, Mat A, const MatFactorInfo *info, Mat_MUMPS *mumps)
2276d71ae5a4SJacob Faibussowitsch {
22775cd7cf9dSHong Zhang   PetscFunctionBegin;
22785cd7cf9dSHong Zhang   if (mumps->id.INFOG(1) < 0) {
22799261f6e4SBarry Smith     PetscCheck(!A->erroriffailure, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in analysis: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
22805cd7cf9dSHong Zhang     if (mumps->id.INFOG(1) == -6) {
22819261f6e4SBarry 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)));
2282603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_STRUCT_ZEROPIVOT;
22835cd7cf9dSHong Zhang     } else if (mumps->id.INFOG(1) == -5 || mumps->id.INFOG(1) == -7) {
22849261f6e4SBarry 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)));
2285603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OUTMEMORY;
2286dbf6bb8dSprj-     } else if (mumps->id.INFOG(1) == -16 && mumps->id.INFOG(1) == 0) {
22879261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in analysis: empty matrix\n"));
22885cd7cf9dSHong Zhang     } else {
22899261f6e4SBarry 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)));
2290603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OTHER;
22915cd7cf9dSHong Zhang     }
22925cd7cf9dSHong Zhang   }
22933ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
22945cd7cf9dSHong Zhang }
22955cd7cf9dSHong Zhang 
229666976f2fSJacob Faibussowitsch static PetscErrorCode MatLUFactorSymbolic_AIJMUMPS(Mat F, Mat A, IS r, IS c, const MatFactorInfo *info)
2297d71ae5a4SJacob Faibussowitsch {
2298e69c285eSBarry Smith   Mat_MUMPS     *mumps = (Mat_MUMPS *)F->data;
229967877ebaSShri Abhyankar   Vec            b;
230067877ebaSShri Abhyankar   const PetscInt M = A->rmap->N;
2301397b6df1SKris Buschelman 
2302397b6df1SKris Buschelman   PetscFunctionBegin;
2303d47f36abSHong Zhang   if (mumps->matstruc == SAME_NONZERO_PATTERN) {
2304d47f36abSHong Zhang     /* F is assembled by a previous call of MatLUFactorSymbolic_AIJMUMPS() */
23053ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2306d47f36abSHong Zhang   }
2307dcd589f8SShri Abhyankar 
23089a2535b5SHong Zhang   /* Set MUMPS options from the options database */
230926cc229bSBarry Smith   PetscCall(MatSetFromOptions_MUMPS(F, A));
2310dcd589f8SShri Abhyankar 
23119566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_INITIAL_MATRIX, mumps));
23129566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_INITIAL_MATRIX, mumps));
2313dcd589f8SShri Abhyankar 
231467877ebaSShri Abhyankar   /* analysis phase */
2315a5e57a09SHong Zhang   mumps->id.job = JOB_FACTSYMBOLIC;
2316a5e57a09SHong Zhang   mumps->id.n   = M;
2317a5e57a09SHong Zhang   switch (mumps->id.ICNTL(18)) {
231867877ebaSShri Abhyankar   case 0: /* centralized assembled matrix input */
2319a5e57a09SHong Zhang     if (!mumps->myid) {
2320a6053eceSJunchao Zhang       mumps->id.nnz = mumps->nnz;
2321a6053eceSJunchao Zhang       mumps->id.irn = mumps->irn;
2322a6053eceSJunchao Zhang       mumps->id.jcn = mumps->jcn;
2323a6053eceSJunchao Zhang       if (mumps->id.ICNTL(6) > 1) mumps->id.a = (MumpsScalar *)mumps->val;
23244ac6704cSBarry Smith       if (r) {
23254ac6704cSBarry Smith         mumps->id.ICNTL(7) = 1;
2326a5e57a09SHong Zhang         if (!mumps->myid) {
2327e0b74bf9SHong Zhang           const PetscInt *idx;
2328a6053eceSJunchao Zhang           PetscInt        i;
23292205254eSKarl Rupp 
23309566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(M, &mumps->id.perm_in));
23319566063dSJacob Faibussowitsch           PetscCall(ISGetIndices(r, &idx));
23329566063dSJacob Faibussowitsch           for (i = 0; i < M; i++) PetscCall(PetscMUMPSIntCast(idx[i] + 1, &(mumps->id.perm_in[i]))); /* perm_in[]: start from 1, not 0! */
23339566063dSJacob Faibussowitsch           PetscCall(ISRestoreIndices(r, &idx));
2334e0b74bf9SHong Zhang         }
2335e0b74bf9SHong Zhang       }
233667877ebaSShri Abhyankar     }
233767877ebaSShri Abhyankar     break;
233867877ebaSShri Abhyankar   case 3: /* distributed assembled matrix input (size>1) */
2339a6053eceSJunchao Zhang     mumps->id.nnz_loc = mumps->nnz;
2340a6053eceSJunchao Zhang     mumps->id.irn_loc = mumps->irn;
2341a6053eceSJunchao Zhang     mumps->id.jcn_loc = mumps->jcn;
2342a6053eceSJunchao Zhang     if (mumps->id.ICNTL(6) > 1) mumps->id.a_loc = (MumpsScalar *)mumps->val;
234325aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) { /* Centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */
23449566063dSJacob Faibussowitsch       PetscCall(MatCreateVecs(A, NULL, &b));
23459566063dSJacob Faibussowitsch       PetscCall(VecScatterCreateToZero(b, &mumps->scat_rhs, &mumps->b_seq));
23469566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b));
234725aac85cSJunchao Zhang     }
234867877ebaSShri Abhyankar     break;
234967877ebaSShri Abhyankar   }
23503ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
23519566063dSJacob Faibussowitsch   PetscCall(MatFactorSymbolic_MUMPS_ReportIfError(F, A, info, mumps));
235267877ebaSShri Abhyankar 
2353719d5645SBarry Smith   F->ops->lufactornumeric   = MatFactorNumeric_MUMPS;
2354dcd589f8SShri Abhyankar   F->ops->solve             = MatSolve_MUMPS;
235551d5961aSHong Zhang   F->ops->solvetranspose    = MatSolveTranspose_MUMPS;
23564e34a73bSHong Zhang   F->ops->matsolve          = MatMatSolve_MUMPS;
2357eb3ef3b2SHong Zhang   F->ops->mattransposesolve = MatMatTransposeSolve_MUMPS;
2358b18964edSHong Zhang   F->ops->matsolvetranspose = MatMatSolveTranspose_MUMPS;
2359d47f36abSHong Zhang 
2360d47f36abSHong Zhang   mumps->matstruc = SAME_NONZERO_PATTERN;
23613ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2362b24902e0SBarry Smith }
2363b24902e0SBarry Smith 
2364450b117fSShri Abhyankar /* Note the Petsc r and c permutations are ignored */
236566976f2fSJacob Faibussowitsch static PetscErrorCode MatLUFactorSymbolic_BAIJMUMPS(Mat F, Mat A, IS r, IS c, const MatFactorInfo *info)
2366d71ae5a4SJacob Faibussowitsch {
2367e69c285eSBarry Smith   Mat_MUMPS     *mumps = (Mat_MUMPS *)F->data;
236867877ebaSShri Abhyankar   Vec            b;
236967877ebaSShri Abhyankar   const PetscInt M = A->rmap->N;
2370450b117fSShri Abhyankar 
2371450b117fSShri Abhyankar   PetscFunctionBegin;
2372d47f36abSHong Zhang   if (mumps->matstruc == SAME_NONZERO_PATTERN) {
2373338d3105SPierre Jolivet     /* F is assembled by a previous call of MatLUFactorSymbolic_BAIJMUMPS() */
23743ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2375d47f36abSHong Zhang   }
2376dcd589f8SShri Abhyankar 
23779a2535b5SHong Zhang   /* Set MUMPS options from the options database */
237826cc229bSBarry Smith   PetscCall(MatSetFromOptions_MUMPS(F, A));
2379dcd589f8SShri Abhyankar 
23809566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_INITIAL_MATRIX, mumps));
23819566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_INITIAL_MATRIX, mumps));
238267877ebaSShri Abhyankar 
238367877ebaSShri Abhyankar   /* analysis phase */
2384a5e57a09SHong Zhang   mumps->id.job = JOB_FACTSYMBOLIC;
2385a5e57a09SHong Zhang   mumps->id.n   = M;
2386a5e57a09SHong Zhang   switch (mumps->id.ICNTL(18)) {
238767877ebaSShri Abhyankar   case 0: /* centralized assembled matrix input */
2388a5e57a09SHong Zhang     if (!mumps->myid) {
2389a6053eceSJunchao Zhang       mumps->id.nnz = mumps->nnz;
2390a6053eceSJunchao Zhang       mumps->id.irn = mumps->irn;
2391a6053eceSJunchao Zhang       mumps->id.jcn = mumps->jcn;
2392ad540459SPierre Jolivet       if (mumps->id.ICNTL(6) > 1) mumps->id.a = (MumpsScalar *)mumps->val;
239367877ebaSShri Abhyankar     }
239467877ebaSShri Abhyankar     break;
239567877ebaSShri Abhyankar   case 3: /* distributed assembled matrix input (size>1) */
2396a6053eceSJunchao Zhang     mumps->id.nnz_loc = mumps->nnz;
2397a6053eceSJunchao Zhang     mumps->id.irn_loc = mumps->irn;
2398a6053eceSJunchao Zhang     mumps->id.jcn_loc = mumps->jcn;
2399ad540459SPierre Jolivet     if (mumps->id.ICNTL(6) > 1) mumps->id.a_loc = (MumpsScalar *)mumps->val;
240025aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) { /* Centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */
24019566063dSJacob Faibussowitsch       PetscCall(MatCreateVecs(A, NULL, &b));
24029566063dSJacob Faibussowitsch       PetscCall(VecScatterCreateToZero(b, &mumps->scat_rhs, &mumps->b_seq));
24039566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b));
240425aac85cSJunchao Zhang     }
240567877ebaSShri Abhyankar     break;
240667877ebaSShri Abhyankar   }
24073ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
24089566063dSJacob Faibussowitsch   PetscCall(MatFactorSymbolic_MUMPS_ReportIfError(F, A, info, mumps));
240967877ebaSShri Abhyankar 
2410450b117fSShri Abhyankar   F->ops->lufactornumeric   = MatFactorNumeric_MUMPS;
2411dcd589f8SShri Abhyankar   F->ops->solve             = MatSolve_MUMPS;
241251d5961aSHong Zhang   F->ops->solvetranspose    = MatSolveTranspose_MUMPS;
2413b18964edSHong Zhang   F->ops->matsolvetranspose = MatMatSolveTranspose_MUMPS;
2414d47f36abSHong Zhang 
2415d47f36abSHong Zhang   mumps->matstruc = SAME_NONZERO_PATTERN;
24163ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2417450b117fSShri Abhyankar }
2418b24902e0SBarry Smith 
2419141f4205SHong Zhang /* Note the Petsc r permutation and factor info are ignored */
242066976f2fSJacob Faibussowitsch static PetscErrorCode MatCholeskyFactorSymbolic_MUMPS(Mat F, Mat A, IS r, const MatFactorInfo *info)
2421d71ae5a4SJacob Faibussowitsch {
2422e69c285eSBarry Smith   Mat_MUMPS     *mumps = (Mat_MUMPS *)F->data;
242367877ebaSShri Abhyankar   Vec            b;
242467877ebaSShri Abhyankar   const PetscInt M = A->rmap->N;
2425397b6df1SKris Buschelman 
2426397b6df1SKris Buschelman   PetscFunctionBegin;
2427d47f36abSHong Zhang   if (mumps->matstruc == SAME_NONZERO_PATTERN) {
2428338d3105SPierre Jolivet     /* F is assembled by a previous call of MatCholeskyFactorSymbolic_MUMPS() */
24293ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2430d47f36abSHong Zhang   }
2431dcd589f8SShri Abhyankar 
24329a2535b5SHong Zhang   /* Set MUMPS options from the options database */
243326cc229bSBarry Smith   PetscCall(MatSetFromOptions_MUMPS(F, A));
2434dcd589f8SShri Abhyankar 
24359566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_INITIAL_MATRIX, mumps));
24369566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_INITIAL_MATRIX, mumps));
2437dcd589f8SShri Abhyankar 
243867877ebaSShri Abhyankar   /* analysis phase */
2439a5e57a09SHong Zhang   mumps->id.job = JOB_FACTSYMBOLIC;
2440a5e57a09SHong Zhang   mumps->id.n   = M;
2441a5e57a09SHong Zhang   switch (mumps->id.ICNTL(18)) {
244267877ebaSShri Abhyankar   case 0: /* centralized assembled matrix input */
2443a5e57a09SHong Zhang     if (!mumps->myid) {
2444a6053eceSJunchao Zhang       mumps->id.nnz = mumps->nnz;
2445a6053eceSJunchao Zhang       mumps->id.irn = mumps->irn;
2446a6053eceSJunchao Zhang       mumps->id.jcn = mumps->jcn;
2447ad540459SPierre Jolivet       if (mumps->id.ICNTL(6) > 1) mumps->id.a = (MumpsScalar *)mumps->val;
244867877ebaSShri Abhyankar     }
244967877ebaSShri Abhyankar     break;
245067877ebaSShri Abhyankar   case 3: /* distributed assembled matrix input (size>1) */
2451a6053eceSJunchao Zhang     mumps->id.nnz_loc = mumps->nnz;
2452a6053eceSJunchao Zhang     mumps->id.irn_loc = mumps->irn;
2453a6053eceSJunchao Zhang     mumps->id.jcn_loc = mumps->jcn;
2454ad540459SPierre Jolivet     if (mumps->id.ICNTL(6) > 1) mumps->id.a_loc = (MumpsScalar *)mumps->val;
245525aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) { /* Centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */
24569566063dSJacob Faibussowitsch       PetscCall(MatCreateVecs(A, NULL, &b));
24579566063dSJacob Faibussowitsch       PetscCall(VecScatterCreateToZero(b, &mumps->scat_rhs, &mumps->b_seq));
24589566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b));
245925aac85cSJunchao Zhang     }
246067877ebaSShri Abhyankar     break;
246167877ebaSShri Abhyankar   }
24623ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
24639566063dSJacob Faibussowitsch   PetscCall(MatFactorSymbolic_MUMPS_ReportIfError(F, A, info, mumps));
24645cd7cf9dSHong Zhang 
24652792810eSHong Zhang   F->ops->choleskyfactornumeric = MatFactorNumeric_MUMPS;
2466dcd589f8SShri Abhyankar   F->ops->solve                 = MatSolve_MUMPS;
246751d5961aSHong Zhang   F->ops->solvetranspose        = MatSolve_MUMPS;
24684e34a73bSHong Zhang   F->ops->matsolve              = MatMatSolve_MUMPS;
246923a5080aSHong Zhang   F->ops->mattransposesolve     = MatMatTransposeSolve_MUMPS;
2470b18964edSHong Zhang   F->ops->matsolvetranspose     = MatMatSolveTranspose_MUMPS;
24714e34a73bSHong Zhang #if defined(PETSC_USE_COMPLEX)
24720298fd71SBarry Smith   F->ops->getinertia = NULL;
24734e34a73bSHong Zhang #else
24744e34a73bSHong Zhang   F->ops->getinertia = MatGetInertia_SBAIJMUMPS;
2475db4efbfdSBarry Smith #endif
2476d47f36abSHong Zhang 
2477d47f36abSHong Zhang   mumps->matstruc = SAME_NONZERO_PATTERN;
24783ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2479b24902e0SBarry Smith }
2480b24902e0SBarry Smith 
248166976f2fSJacob Faibussowitsch static PetscErrorCode MatView_MUMPS(Mat A, PetscViewer viewer)
2482d71ae5a4SJacob Faibussowitsch {
248364e6c443SBarry Smith   PetscBool         iascii;
248464e6c443SBarry Smith   PetscViewerFormat format;
2485e69c285eSBarry Smith   Mat_MUMPS        *mumps = (Mat_MUMPS *)A->data;
2486f6c57405SHong Zhang 
2487f6c57405SHong Zhang   PetscFunctionBegin;
248864e6c443SBarry Smith   /* check if matrix is mumps type */
24893ba16761SJacob Faibussowitsch   if (A->ops->solve != MatSolve_MUMPS) PetscFunctionReturn(PETSC_SUCCESS);
249064e6c443SBarry Smith 
24919566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
249264e6c443SBarry Smith   if (iascii) {
24939566063dSJacob Faibussowitsch     PetscCall(PetscViewerGetFormat(viewer, &format));
24941511cd71SPierre Jolivet     if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
24959566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "MUMPS run parameters:\n"));
24961511cd71SPierre Jolivet       if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
24979566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  SYM (matrix type):                   %d\n", mumps->id.sym));
24989566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  PAR (host participation):            %d\n", mumps->id.par));
24999566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(1) (output for error):         %d\n", mumps->id.ICNTL(1)));
25009566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(2) (output of diagnostic msg): %d\n", mumps->id.ICNTL(2)));
25019566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(3) (output for global info):   %d\n", mumps->id.ICNTL(3)));
25029566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(4) (level of printing):        %d\n", mumps->id.ICNTL(4)));
25039566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(5) (input mat struct):         %d\n", mumps->id.ICNTL(5)));
25049566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(6) (matrix prescaling):        %d\n", mumps->id.ICNTL(6)));
25059566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(7) (sequential matrix ordering):%d\n", mumps->id.ICNTL(7)));
25069566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(8) (scaling strategy):         %d\n", mumps->id.ICNTL(8)));
25079566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(10) (max num of refinements):  %d\n", mumps->id.ICNTL(10)));
25089566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(11) (error analysis):          %d\n", mumps->id.ICNTL(11)));
2509a5e57a09SHong Zhang         if (mumps->id.ICNTL(11) > 0) {
25109566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(4) (inf norm of input mat):        %g\n", mumps->id.RINFOG(4)));
25119566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(5) (inf norm of solution):         %g\n", mumps->id.RINFOG(5)));
25129566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(6) (inf norm of residual):         %g\n", mumps->id.RINFOG(6)));
25139566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(7),RINFOG(8) (backward error est): %g, %g\n", mumps->id.RINFOG(7), mumps->id.RINFOG(8)));
25149566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(9) (error estimate):               %g\n", mumps->id.RINFOG(9)));
25159566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(10),RINFOG(11)(condition numbers): %g, %g\n", mumps->id.RINFOG(10), mumps->id.RINFOG(11)));
2516f6c57405SHong Zhang         }
25179566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(12) (efficiency control):                         %d\n", mumps->id.ICNTL(12)));
25189566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(13) (sequential factorization of the root node):  %d\n", mumps->id.ICNTL(13)));
25199566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(14) (percentage of estimated workspace increase): %d\n", mumps->id.ICNTL(14)));
252045e3843bSPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(15) (compression of the input matrix):            %d\n", mumps->id.ICNTL(15)));
2521f6c57405SHong Zhang         /* ICNTL(15-17) not used */
25229566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(18) (input mat struct):                           %d\n", mumps->id.ICNTL(18)));
25239566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(19) (Schur complement info):                      %d\n", mumps->id.ICNTL(19)));
25249566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(20) (RHS sparse pattern):                         %d\n", mumps->id.ICNTL(20)));
25259566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(21) (solution struct):                            %d\n", mumps->id.ICNTL(21)));
25269566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(22) (in-core/out-of-core facility):               %d\n", mumps->id.ICNTL(22)));
25279566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(23) (max size of memory can be allocated locally):%d\n", mumps->id.ICNTL(23)));
2528c0165424SHong Zhang 
25299566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(24) (detection of null pivot rows):               %d\n", mumps->id.ICNTL(24)));
25309566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(25) (computation of a null space basis):          %d\n", mumps->id.ICNTL(25)));
25319566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(26) (Schur options for RHS or solution):          %d\n", mumps->id.ICNTL(26)));
25329566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(27) (blocking size for multiple RHS):             %d\n", mumps->id.ICNTL(27)));
25339566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(28) (use parallel or sequential ordering):        %d\n", mumps->id.ICNTL(28)));
25349566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(29) (parallel ordering):                          %d\n", mumps->id.ICNTL(29)));
253542179a6aSHong Zhang 
25369566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(30) (user-specified set of entries in inv(A)):    %d\n", mumps->id.ICNTL(30)));
25379566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(31) (factors is discarded in the solve phase):    %d\n", mumps->id.ICNTL(31)));
25389566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(33) (compute determinant):                        %d\n", mumps->id.ICNTL(33)));
25399566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(35) (activate BLR based factorization):           %d\n", mumps->id.ICNTL(35)));
25409566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(36) (choice of BLR factorization variant):        %d\n", mumps->id.ICNTL(36)));
25419566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(38) (estimated compression rate of LU factors):   %d\n", mumps->id.ICNTL(38)));
2542146931dbSPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(58) (options for symbolic factorization):         %d\n", mumps->id.ICNTL(58)));
2543f6c57405SHong Zhang 
25449566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(1) (relative pivoting threshold):      %g\n", mumps->id.CNTL(1)));
25459566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(2) (stopping criterion of refinement): %g\n", mumps->id.CNTL(2)));
25469566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(3) (absolute pivoting threshold):      %g\n", mumps->id.CNTL(3)));
25479566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(4) (value of static pivoting):         %g\n", mumps->id.CNTL(4)));
25489566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(5) (fixation for null pivots):         %g\n", mumps->id.CNTL(5)));
25499566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(7) (dropping parameter for BLR):       %g\n", mumps->id.CNTL(7)));
2550f6c57405SHong Zhang 
2551a5b23f4aSJose E. Roman         /* information local to each processor */
25529566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFO(1) (local estimated flops for the elimination after analysis):\n"));
25539566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPushSynchronized(viewer));
25549566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %g\n", mumps->myid, mumps->id.RINFO(1)));
25559566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
25569566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFO(2) (local estimated flops for the assembly after factorization):\n"));
25579566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %g\n", mumps->myid, mumps->id.RINFO(2)));
25589566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
25599566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFO(3) (local estimated flops for the elimination after factorization):\n"));
25609566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %g\n", mumps->myid, mumps->id.RINFO(3)));
25619566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2562f6c57405SHong Zhang 
25639566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(15) (estimated size of (in MB) MUMPS internal data for running numerical factorization):\n"));
25649566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(15)));
25659566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2566f6c57405SHong Zhang 
25679566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(16) (size of (in MB) MUMPS internal data used during numerical factorization):\n"));
25689566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(16)));
25699566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2570f6c57405SHong Zhang 
25719566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(23) (num of pivots eliminated on this processor after factorization):\n"));
25729566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(23)));
25739566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2574b34f08ffSHong Zhang 
2575a0e18203SThibaut Appel         if (mumps->ninfo && mumps->ninfo <= 80) {
2576b34f08ffSHong Zhang           PetscInt i;
2577b34f08ffSHong Zhang           for (i = 0; i < mumps->ninfo; i++) {
25789566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(%" PetscInt_FMT "):\n", mumps->info[i]));
25799566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(mumps->info[i])));
25809566063dSJacob Faibussowitsch             PetscCall(PetscViewerFlush(viewer));
2581b34f08ffSHong Zhang           }
2582b34f08ffSHong Zhang         }
25839566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPopSynchronized(viewer));
25841511cd71SPierre Jolivet       } else PetscCall(PetscViewerASCIIPrintf(viewer, "  Use -%sksp_view ::ascii_info_detail to display information for all processes\n", ((PetscObject)A)->prefix ? ((PetscObject)A)->prefix : ""));
2585f6c57405SHong Zhang 
25861511cd71SPierre Jolivet       if (mumps->myid == 0) { /* information from the host */
25879566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFOG(1) (global estimated flops for the elimination after analysis): %g\n", mumps->id.RINFOG(1)));
25889566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFOG(2) (global estimated flops for the assembly after factorization): %g\n", mumps->id.RINFOG(2)));
25899566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFOG(3) (global estimated flops for the elimination after factorization): %g\n", mumps->id.RINFOG(3)));
25909566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  (RINFOG(12) RINFOG(13))*2^INFOG(34) (determinant): (%g,%g)*(2^%d)\n", mumps->id.RINFOG(12), mumps->id.RINFOG(13), mumps->id.INFOG(34)));
2591f6c57405SHong Zhang 
25929566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(3) (estimated real workspace for factors on all processors after analysis): %d\n", mumps->id.INFOG(3)));
25939566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(4) (estimated integer workspace for factors on all processors after analysis): %d\n", mumps->id.INFOG(4)));
25949566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(5) (estimated maximum front size in the complete tree): %d\n", mumps->id.INFOG(5)));
25959566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(6) (number of nodes in the complete tree): %d\n", mumps->id.INFOG(6)));
25969566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(7) (ordering option effectively used after analysis): %d\n", mumps->id.INFOG(7)));
25979566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(8) (structural symmetry in percent of the permuted matrix after analysis): %d\n", mumps->id.INFOG(8)));
25989566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(9) (total real/complex workspace to store the matrix factors after factorization): %d\n", mumps->id.INFOG(9)));
25999566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(10) (total integer space store the matrix factors after factorization): %d\n", mumps->id.INFOG(10)));
26009566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(11) (order of largest frontal matrix after factorization): %d\n", mumps->id.INFOG(11)));
26019566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(12) (number of off-diagonal pivots): %d\n", mumps->id.INFOG(12)));
26029566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(13) (number of delayed pivots after factorization): %d\n", mumps->id.INFOG(13)));
26039566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(14) (number of memory compress after factorization): %d\n", mumps->id.INFOG(14)));
26049566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(15) (number of steps of iterative refinement after solution): %d\n", mumps->id.INFOG(15)));
26059566063dSJacob 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)));
26069566063dSJacob 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)));
26079566063dSJacob 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)));
26089566063dSJacob 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)));
26099566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(20) (estimated number of entries in the factors): %d\n", mumps->id.INFOG(20)));
26109566063dSJacob 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)));
26119566063dSJacob 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)));
26129566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(23) (after analysis: value of ICNTL(6) effectively used): %d\n", mumps->id.INFOG(23)));
26139566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(24) (after analysis: value of ICNTL(12) effectively used): %d\n", mumps->id.INFOG(24)));
26149566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(25) (after factorization: number of pivots modified by static pivoting): %d\n", mumps->id.INFOG(25)));
26159566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(28) (after factorization: number of null pivots encountered): %d\n", mumps->id.INFOG(28)));
26169566063dSJacob 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)));
26179566063dSJacob 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)));
26189566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(32) (after analysis: type of analysis done): %d\n", mumps->id.INFOG(32)));
26199566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(33) (value used for ICNTL(8)): %d\n", mumps->id.INFOG(33)));
26209566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(34) (exponent of the determinant if determinant is requested): %d\n", mumps->id.INFOG(34)));
26219566063dSJacob 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)));
26229566063dSJacob 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)));
26239566063dSJacob 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)));
26249566063dSJacob 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)));
26259566063dSJacob 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)));
2626f6c57405SHong Zhang       }
2627f6c57405SHong Zhang     }
2628cb828f0fSHong Zhang   }
26293ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2630f6c57405SHong Zhang }
2631f6c57405SHong Zhang 
263266976f2fSJacob Faibussowitsch static PetscErrorCode MatGetInfo_MUMPS(Mat A, MatInfoType flag, MatInfo *info)
2633d71ae5a4SJacob Faibussowitsch {
2634e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)A->data;
263535bd34faSBarry Smith 
263635bd34faSBarry Smith   PetscFunctionBegin;
263735bd34faSBarry Smith   info->block_size        = 1.0;
263864412097SPierre Jolivet   info->nz_allocated      = mumps->id.INFOG(20) >= 0 ? mumps->id.INFOG(20) : -1000000 * mumps->id.INFOG(20);
263964412097SPierre Jolivet   info->nz_used           = mumps->id.INFOG(20) >= 0 ? mumps->id.INFOG(20) : -1000000 * mumps->id.INFOG(20);
264035bd34faSBarry Smith   info->nz_unneeded       = 0.0;
264135bd34faSBarry Smith   info->assemblies        = 0.0;
264235bd34faSBarry Smith   info->mallocs           = 0.0;
264335bd34faSBarry Smith   info->memory            = 0.0;
264435bd34faSBarry Smith   info->fill_ratio_given  = 0;
264535bd34faSBarry Smith   info->fill_ratio_needed = 0;
264635bd34faSBarry Smith   info->factor_mallocs    = 0;
26473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
264835bd34faSBarry Smith }
264935bd34faSBarry Smith 
265066976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorSetSchurIS_MUMPS(Mat F, IS is)
2651d71ae5a4SJacob Faibussowitsch {
2652e69c285eSBarry Smith   Mat_MUMPS         *mumps = (Mat_MUMPS *)F->data;
2653a3d589ffSStefano Zampini   const PetscScalar *arr;
26548e7ba810SStefano Zampini   const PetscInt    *idxs;
26558e7ba810SStefano Zampini   PetscInt           size, i;
26566444a565SStefano Zampini 
26576444a565SStefano Zampini   PetscFunctionBegin;
26589566063dSJacob Faibussowitsch   PetscCall(ISGetLocalSize(is, &size));
2659b3cb21ddSStefano Zampini   /* Schur complement matrix */
26609566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&F->schur));
26619566063dSJacob Faibussowitsch   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, size, size, NULL, &F->schur));
26629566063dSJacob Faibussowitsch   PetscCall(MatDenseGetArrayRead(F->schur, &arr));
2663a3d589ffSStefano Zampini   mumps->id.schur      = (MumpsScalar *)arr;
2664a3d589ffSStefano Zampini   mumps->id.size_schur = size;
2665a3d589ffSStefano Zampini   mumps->id.schur_lld  = size;
26669566063dSJacob Faibussowitsch   PetscCall(MatDenseRestoreArrayRead(F->schur, &arr));
266748a46eb9SPierre Jolivet   if (mumps->sym == 1) PetscCall(MatSetOption(F->schur, MAT_SPD, PETSC_TRUE));
2668b3cb21ddSStefano Zampini 
2669b3cb21ddSStefano Zampini   /* MUMPS expects Fortran style indices */
26709566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.listvar_schur));
26719566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(size, &mumps->id.listvar_schur));
26729566063dSJacob Faibussowitsch   PetscCall(ISGetIndices(is, &idxs));
26739566063dSJacob Faibussowitsch   for (i = 0; i < size; i++) PetscCall(PetscMUMPSIntCast(idxs[i] + 1, &(mumps->id.listvar_schur[i])));
26749566063dSJacob Faibussowitsch   PetscCall(ISRestoreIndices(is, &idxs));
267559ac8732SStefano Zampini   /* set a special value of ICNTL (not handled my MUMPS) to be used in the solve phase by PETSc */
2676b5fa320bSStefano Zampini   mumps->id.ICNTL(26) = -1;
26773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
26786444a565SStefano Zampini }
267959ac8732SStefano Zampini 
268066976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorCreateSchurComplement_MUMPS(Mat F, Mat *S)
2681d71ae5a4SJacob Faibussowitsch {
26826444a565SStefano Zampini   Mat          St;
2683e69c285eSBarry Smith   Mat_MUMPS   *mumps = (Mat_MUMPS *)F->data;
26846444a565SStefano Zampini   PetscScalar *array;
26856444a565SStefano Zampini 
26866444a565SStefano Zampini   PetscFunctionBegin;
26879261f6e4SBarry Smith   PetscCheck(mumps->id.ICNTL(19), PetscObjectComm((PetscObject)F), PETSC_ERR_ORDER, "Schur complement mode not selected! Call MatFactorSetSchurIS() to enable it");
26889566063dSJacob Faibussowitsch   PetscCall(MatCreate(PETSC_COMM_SELF, &St));
26899566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(St, PETSC_DECIDE, PETSC_DECIDE, mumps->id.size_schur, mumps->id.size_schur));
26909566063dSJacob Faibussowitsch   PetscCall(MatSetType(St, MATDENSE));
26919566063dSJacob Faibussowitsch   PetscCall(MatSetUp(St));
26929566063dSJacob Faibussowitsch   PetscCall(MatDenseGetArray(St, &array));
269359ac8732SStefano Zampini   if (!mumps->sym) {                /* MUMPS always return a full matrix */
26946444a565SStefano Zampini     if (mumps->id.ICNTL(19) == 1) { /* stored by rows */
26956444a565SStefano Zampini       PetscInt i, j, N = mumps->id.size_schur;
26966444a565SStefano Zampini       for (i = 0; i < N; i++) {
26976444a565SStefano Zampini         for (j = 0; j < N; j++) {
26986444a565SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
26996444a565SStefano Zampini           PetscScalar val = mumps->id.schur[i * N + j];
27006444a565SStefano Zampini #else
27013e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.schur[i * N + j].r + PETSC_i * mumps->id.schur[i * N + j].i;
27026444a565SStefano Zampini #endif
27036444a565SStefano Zampini           array[j * N + i] = val;
27046444a565SStefano Zampini         }
27056444a565SStefano Zampini       }
27066444a565SStefano Zampini     } else { /* stored by columns */
27079566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(array, mumps->id.schur, mumps->id.size_schur * mumps->id.size_schur));
27086444a565SStefano Zampini     }
27096444a565SStefano Zampini   } else {                          /* either full or lower-triangular (not packed) */
27106444a565SStefano Zampini     if (mumps->id.ICNTL(19) == 2) { /* lower triangular stored by columns */
27116444a565SStefano Zampini       PetscInt i, j, N = mumps->id.size_schur;
27126444a565SStefano Zampini       for (i = 0; i < N; i++) {
27136444a565SStefano Zampini         for (j = i; j < N; j++) {
27146444a565SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
27156444a565SStefano Zampini           PetscScalar val = mumps->id.schur[i * N + j];
27166444a565SStefano Zampini #else
27173e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.schur[i * N + j].r + PETSC_i * mumps->id.schur[i * N + j].i;
27186444a565SStefano Zampini #endif
27193e5b40d0SPierre Jolivet           array[i * N + j] = array[j * N + i] = val;
27206444a565SStefano Zampini         }
27216444a565SStefano Zampini       }
27226444a565SStefano Zampini     } else if (mumps->id.ICNTL(19) == 3) { /* full matrix */
27239566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(array, mumps->id.schur, mumps->id.size_schur * mumps->id.size_schur));
27246444a565SStefano Zampini     } else { /* ICNTL(19) == 1 lower triangular stored by rows */
27256444a565SStefano Zampini       PetscInt i, j, N = mumps->id.size_schur;
27266444a565SStefano Zampini       for (i = 0; i < N; i++) {
27276444a565SStefano Zampini         for (j = 0; j < i + 1; j++) {
27286444a565SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
27296444a565SStefano Zampini           PetscScalar val = mumps->id.schur[i * N + j];
27306444a565SStefano Zampini #else
27313e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.schur[i * N + j].r + PETSC_i * mumps->id.schur[i * N + j].i;
27326444a565SStefano Zampini #endif
27333e5b40d0SPierre Jolivet           array[i * N + j] = array[j * N + i] = val;
27346444a565SStefano Zampini         }
27356444a565SStefano Zampini       }
27366444a565SStefano Zampini     }
27376444a565SStefano Zampini   }
27389566063dSJacob Faibussowitsch   PetscCall(MatDenseRestoreArray(St, &array));
27396444a565SStefano Zampini   *S = St;
27403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
27416444a565SStefano Zampini }
27426444a565SStefano Zampini 
274366976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsSetIcntl_MUMPS(Mat F, PetscInt icntl, PetscInt ival)
2744d71ae5a4SJacob Faibussowitsch {
2745e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
27465ccb76cbSHong Zhang 
27475ccb76cbSHong Zhang   PetscFunctionBegin;
2748413bcc21SPierre Jolivet   if (mumps->id.job == JOB_NULL) {                                       /* need to cache icntl and ival since PetscMUMPS_c() has never been called */
2749413bcc21SPierre Jolivet     PetscInt i, nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0; /* number of already cached ICNTL */
27509371c9d4SSatish Balay     for (i = 0; i < nICNTL_pre; ++i)
27519371c9d4SSatish Balay       if (mumps->ICNTL_pre[1 + 2 * i] == icntl) break; /* is this ICNTL already cached? */
2752413bcc21SPierre Jolivet     if (i == nICNTL_pre) {                             /* not already cached */
2753413bcc21SPierre Jolivet       if (i > 0) PetscCall(PetscRealloc(sizeof(PetscMUMPSInt) * (2 * nICNTL_pre + 3), &mumps->ICNTL_pre));
2754413bcc21SPierre Jolivet       else PetscCall(PetscCalloc(sizeof(PetscMUMPSInt) * 3, &mumps->ICNTL_pre));
2755413bcc21SPierre Jolivet       mumps->ICNTL_pre[0]++;
2756413bcc21SPierre Jolivet     }
2757413bcc21SPierre Jolivet     mumps->ICNTL_pre[1 + 2 * i] = icntl;
2758413bcc21SPierre Jolivet     PetscCall(PetscMUMPSIntCast(ival, mumps->ICNTL_pre + 2 + 2 * i));
2759413bcc21SPierre Jolivet   } else PetscCall(PetscMUMPSIntCast(ival, &mumps->id.ICNTL(icntl)));
27603ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
27615ccb76cbSHong Zhang }
27625ccb76cbSHong Zhang 
276366976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetIcntl_MUMPS(Mat F, PetscInt icntl, PetscInt *ival)
2764d71ae5a4SJacob Faibussowitsch {
2765e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2766bc6112feSHong Zhang 
2767bc6112feSHong Zhang   PetscFunctionBegin;
276836df9881Sjeremy theler   if (mumps->id.job == JOB_NULL) {
276936df9881Sjeremy theler     PetscInt i, nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0;
277036df9881Sjeremy theler     *ival = 0;
277136df9881Sjeremy theler     for (i = 0; i < nICNTL_pre; ++i) {
277236df9881Sjeremy theler       if (mumps->ICNTL_pre[1 + 2 * i] == icntl) *ival = mumps->ICNTL_pre[2 + 2 * i];
277336df9881Sjeremy theler     }
277436df9881Sjeremy theler   } else *ival = mumps->id.ICNTL(icntl);
27753ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2776bc6112feSHong Zhang }
2777bc6112feSHong Zhang 
27785ccb76cbSHong Zhang /*@
27795ccb76cbSHong Zhang   MatMumpsSetIcntl - Set MUMPS parameter ICNTL()
27805ccb76cbSHong Zhang 
2781c3339decSBarry Smith   Logically Collective
27825ccb76cbSHong Zhang 
27835ccb76cbSHong Zhang   Input Parameters:
278411a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
27855ccb76cbSHong Zhang . icntl - index of MUMPS parameter array ICNTL()
27865ccb76cbSHong Zhang - ival  - value of MUMPS ICNTL(icntl)
27875ccb76cbSHong Zhang 
27883c7db156SBarry Smith   Options Database Key:
2789147403d9SBarry Smith . -mat_mumps_icntl_<icntl> <ival> - change the option numbered icntl to ival
27905ccb76cbSHong Zhang 
27915ccb76cbSHong Zhang   Level: beginner
27925ccb76cbSHong Zhang 
279396a0c994SBarry Smith   References:
2794606c0280SSatish Balay .  * - MUMPS Users' Guide
27955ccb76cbSHong Zhang 
27961cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
27975ccb76cbSHong Zhang @*/
2798d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsSetIcntl(Mat F, PetscInt icntl, PetscInt ival)
2799d71ae5a4SJacob Faibussowitsch {
28005ccb76cbSHong Zhang   PetscFunctionBegin;
28012989dfd4SHong Zhang   PetscValidType(F, 1);
280228b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
28035ccb76cbSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
28045ccb76cbSHong Zhang   PetscValidLogicalCollectiveInt(F, ival, 3);
2805146931dbSPierre Jolivet   PetscCheck((icntl >= 1 && icntl <= 38) || icntl == 58, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported ICNTL value %" PetscInt_FMT, icntl);
2806cac4c232SBarry Smith   PetscTryMethod(F, "MatMumpsSetIcntl_C", (Mat, PetscInt, PetscInt), (F, icntl, ival));
28073ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
28085ccb76cbSHong Zhang }
28095ccb76cbSHong Zhang 
2810a21f80fcSHong Zhang /*@
2811a21f80fcSHong Zhang   MatMumpsGetIcntl - Get MUMPS parameter ICNTL()
2812a21f80fcSHong Zhang 
2813c3339decSBarry Smith   Logically Collective
2814a21f80fcSHong Zhang 
2815a21f80fcSHong Zhang   Input Parameters:
281611a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
2817a21f80fcSHong Zhang - icntl - index of MUMPS parameter array ICNTL()
2818a21f80fcSHong Zhang 
2819a21f80fcSHong Zhang   Output Parameter:
2820a21f80fcSHong Zhang . ival - value of MUMPS ICNTL(icntl)
2821a21f80fcSHong Zhang 
2822a21f80fcSHong Zhang   Level: beginner
2823a21f80fcSHong Zhang 
282496a0c994SBarry Smith   References:
2825606c0280SSatish Balay .  * - MUMPS Users' Guide
2826a21f80fcSHong Zhang 
28271cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
2828a21f80fcSHong Zhang @*/
2829d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetIcntl(Mat F, PetscInt icntl, PetscInt *ival)
2830d71ae5a4SJacob Faibussowitsch {
2831bc6112feSHong Zhang   PetscFunctionBegin;
28322989dfd4SHong Zhang   PetscValidType(F, 1);
283328b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
2834bc6112feSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
28354f572ea9SToby Isaac   PetscAssertPointer(ival, 3);
2836146931dbSPierre Jolivet   PetscCheck((icntl >= 1 && icntl <= 38) || icntl == 58, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported ICNTL value %" PetscInt_FMT, icntl);
2837cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetIcntl_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival));
28383ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2839bc6112feSHong Zhang }
2840bc6112feSHong Zhang 
284166976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsSetCntl_MUMPS(Mat F, PetscInt icntl, PetscReal val)
2842d71ae5a4SJacob Faibussowitsch {
2843e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
28448928b65cSHong Zhang 
28458928b65cSHong Zhang   PetscFunctionBegin;
2846413bcc21SPierre Jolivet   if (mumps->id.job == JOB_NULL) {
2847413bcc21SPierre Jolivet     PetscInt i, nCNTL_pre = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0;
28489371c9d4SSatish Balay     for (i = 0; i < nCNTL_pre; ++i)
28499371c9d4SSatish Balay       if (mumps->CNTL_pre[1 + 2 * i] == icntl) break;
2850413bcc21SPierre Jolivet     if (i == nCNTL_pre) {
2851413bcc21SPierre Jolivet       if (i > 0) PetscCall(PetscRealloc(sizeof(PetscReal) * (2 * nCNTL_pre + 3), &mumps->CNTL_pre));
2852413bcc21SPierre Jolivet       else PetscCall(PetscCalloc(sizeof(PetscReal) * 3, &mumps->CNTL_pre));
2853413bcc21SPierre Jolivet       mumps->CNTL_pre[0]++;
2854413bcc21SPierre Jolivet     }
2855413bcc21SPierre Jolivet     mumps->CNTL_pre[1 + 2 * i] = icntl;
2856413bcc21SPierre Jolivet     mumps->CNTL_pre[2 + 2 * i] = val;
2857413bcc21SPierre Jolivet   } else mumps->id.CNTL(icntl) = val;
28583ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
28598928b65cSHong Zhang }
28608928b65cSHong Zhang 
286166976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetCntl_MUMPS(Mat F, PetscInt icntl, PetscReal *val)
2862d71ae5a4SJacob Faibussowitsch {
2863e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2864bc6112feSHong Zhang 
2865bc6112feSHong Zhang   PetscFunctionBegin;
286636df9881Sjeremy theler   if (mumps->id.job == JOB_NULL) {
286736df9881Sjeremy theler     PetscInt i, nCNTL_pre = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0;
286836df9881Sjeremy theler     *val = 0.0;
286936df9881Sjeremy theler     for (i = 0; i < nCNTL_pre; ++i) {
287036df9881Sjeremy theler       if (mumps->CNTL_pre[1 + 2 * i] == icntl) *val = mumps->CNTL_pre[2 + 2 * i];
287136df9881Sjeremy theler     }
287236df9881Sjeremy theler   } else *val = mumps->id.CNTL(icntl);
28733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2874bc6112feSHong Zhang }
2875bc6112feSHong Zhang 
28768928b65cSHong Zhang /*@
28778928b65cSHong Zhang   MatMumpsSetCntl - Set MUMPS parameter CNTL()
28788928b65cSHong Zhang 
2879c3339decSBarry Smith   Logically Collective
28808928b65cSHong Zhang 
28818928b65cSHong Zhang   Input Parameters:
288211a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
28838928b65cSHong Zhang . icntl - index of MUMPS parameter array CNTL()
28848928b65cSHong Zhang - val   - value of MUMPS CNTL(icntl)
28858928b65cSHong Zhang 
28863c7db156SBarry Smith   Options Database Key:
2887147403d9SBarry Smith . -mat_mumps_cntl_<icntl> <val> - change the option numbered icntl to ival
28888928b65cSHong Zhang 
28898928b65cSHong Zhang   Level: beginner
28908928b65cSHong Zhang 
289196a0c994SBarry Smith   References:
2892606c0280SSatish Balay .  * - MUMPS Users' Guide
28938928b65cSHong Zhang 
28941cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
28958928b65cSHong Zhang @*/
2896d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsSetCntl(Mat F, PetscInt icntl, PetscReal val)
2897d71ae5a4SJacob Faibussowitsch {
28988928b65cSHong Zhang   PetscFunctionBegin;
28992989dfd4SHong Zhang   PetscValidType(F, 1);
290028b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
29018928b65cSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
2902bc6112feSHong Zhang   PetscValidLogicalCollectiveReal(F, val, 3);
2903413bcc21SPierre Jolivet   PetscCheck(icntl >= 1 && icntl <= 7, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported CNTL value %" PetscInt_FMT, icntl);
2904cac4c232SBarry Smith   PetscTryMethod(F, "MatMumpsSetCntl_C", (Mat, PetscInt, PetscReal), (F, icntl, val));
29053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
29068928b65cSHong Zhang }
29078928b65cSHong Zhang 
2908a21f80fcSHong Zhang /*@
2909a21f80fcSHong Zhang   MatMumpsGetCntl - Get MUMPS parameter CNTL()
2910a21f80fcSHong Zhang 
2911c3339decSBarry Smith   Logically Collective
2912a21f80fcSHong Zhang 
2913a21f80fcSHong Zhang   Input Parameters:
291411a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
2915a21f80fcSHong Zhang - icntl - index of MUMPS parameter array CNTL()
2916a21f80fcSHong Zhang 
2917a21f80fcSHong Zhang   Output Parameter:
2918a21f80fcSHong Zhang . val - value of MUMPS CNTL(icntl)
2919a21f80fcSHong Zhang 
2920a21f80fcSHong Zhang   Level: beginner
2921a21f80fcSHong Zhang 
292296a0c994SBarry Smith   References:
2923606c0280SSatish Balay .  * - MUMPS Users' Guide
2924a21f80fcSHong Zhang 
29251cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
2926a21f80fcSHong Zhang @*/
2927d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetCntl(Mat F, PetscInt icntl, PetscReal *val)
2928d71ae5a4SJacob Faibussowitsch {
2929bc6112feSHong Zhang   PetscFunctionBegin;
29302989dfd4SHong Zhang   PetscValidType(F, 1);
293128b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
2932bc6112feSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
29334f572ea9SToby Isaac   PetscAssertPointer(val, 3);
2934413bcc21SPierre Jolivet   PetscCheck(icntl >= 1 && icntl <= 7, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported CNTL value %" PetscInt_FMT, icntl);
2935cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetCntl_C", (Mat, PetscInt, PetscReal *), (F, icntl, val));
29363ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2937bc6112feSHong Zhang }
2938bc6112feSHong Zhang 
293966976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInfo_MUMPS(Mat F, PetscInt icntl, PetscInt *info)
2940d71ae5a4SJacob Faibussowitsch {
2941e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2942bc6112feSHong Zhang 
2943bc6112feSHong Zhang   PetscFunctionBegin;
2944bc6112feSHong Zhang   *info = mumps->id.INFO(icntl);
29453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2946bc6112feSHong Zhang }
2947bc6112feSHong Zhang 
294866976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInfog_MUMPS(Mat F, PetscInt icntl, PetscInt *infog)
2949d71ae5a4SJacob Faibussowitsch {
2950e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2951bc6112feSHong Zhang 
2952bc6112feSHong Zhang   PetscFunctionBegin;
2953bc6112feSHong Zhang   *infog = mumps->id.INFOG(icntl);
29543ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2955bc6112feSHong Zhang }
2956bc6112feSHong Zhang 
295766976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetRinfo_MUMPS(Mat F, PetscInt icntl, PetscReal *rinfo)
2958d71ae5a4SJacob Faibussowitsch {
2959e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2960bc6112feSHong Zhang 
2961bc6112feSHong Zhang   PetscFunctionBegin;
2962bc6112feSHong Zhang   *rinfo = mumps->id.RINFO(icntl);
29633ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2964bc6112feSHong Zhang }
2965bc6112feSHong Zhang 
296666976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetRinfog_MUMPS(Mat F, PetscInt icntl, PetscReal *rinfog)
2967d71ae5a4SJacob Faibussowitsch {
2968e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2969bc6112feSHong Zhang 
2970bc6112feSHong Zhang   PetscFunctionBegin;
2971bc6112feSHong Zhang   *rinfog = mumps->id.RINFOG(icntl);
29723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2973bc6112feSHong Zhang }
2974bc6112feSHong Zhang 
297566976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetNullPivots_MUMPS(Mat F, PetscInt *size, PetscInt **array)
29765c0bae8cSAshish Patel {
29775c0bae8cSAshish Patel   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
29785c0bae8cSAshish Patel 
29795c0bae8cSAshish Patel   PetscFunctionBegin;
29805c0bae8cSAshish 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");
29815c0bae8cSAshish Patel   *size  = 0;
29825c0bae8cSAshish Patel   *array = NULL;
29835c0bae8cSAshish Patel   if (!mumps->myid) {
29845c0bae8cSAshish Patel     *size = mumps->id.INFOG(28);
29855c0bae8cSAshish Patel     PetscCall(PetscMalloc1(*size, array));
29865c0bae8cSAshish Patel     for (int i = 0; i < *size; i++) (*array)[i] = mumps->id.pivnul_list[i] - 1;
29875c0bae8cSAshish Patel   }
29885c0bae8cSAshish Patel   PetscFunctionReturn(PETSC_SUCCESS);
29895c0bae8cSAshish Patel }
29905c0bae8cSAshish Patel 
299166976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInverse_MUMPS(Mat F, Mat spRHS)
2992d71ae5a4SJacob Faibussowitsch {
29930e6b8875SHong Zhang   Mat          Bt = NULL, Btseq = NULL;
29940e6b8875SHong Zhang   PetscBool    flg;
2995bb599dfdSHong Zhang   Mat_MUMPS   *mumps = (Mat_MUMPS *)F->data;
2996bb599dfdSHong Zhang   PetscScalar *aa;
2997f410b75aSHong Zhang   PetscInt     spnr, *ia, *ja, M, nrhs;
2998bb599dfdSHong Zhang 
2999bb599dfdSHong Zhang   PetscFunctionBegin;
30004f572ea9SToby Isaac   PetscAssertPointer(spRHS, 2);
3001013e2dc7SBarry Smith   PetscCall(PetscObjectTypeCompare((PetscObject)spRHS, MATTRANSPOSEVIRTUAL, &flg));
30020e6b8875SHong Zhang   if (flg) {
30039566063dSJacob Faibussowitsch     PetscCall(MatTransposeGetMat(spRHS, &Bt));
3004013e2dc7SBarry Smith   } else SETERRQ(PetscObjectComm((PetscObject)spRHS), PETSC_ERR_ARG_WRONG, "Matrix spRHS must be type MATTRANSPOSEVIRTUAL matrix");
3005bb599dfdSHong Zhang 
30069566063dSJacob Faibussowitsch   PetscCall(MatMumpsSetIcntl(F, 30, 1));
3007bb599dfdSHong Zhang 
30082d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) {
30090e6b8875SHong Zhang     Mat_MPIAIJ *b = (Mat_MPIAIJ *)Bt->data;
30100e6b8875SHong Zhang     Btseq         = b->A;
30110e6b8875SHong Zhang   } else {
30120e6b8875SHong Zhang     Btseq = Bt;
30130e6b8875SHong Zhang   }
30140e6b8875SHong Zhang 
30159566063dSJacob Faibussowitsch   PetscCall(MatGetSize(spRHS, &M, &nrhs));
3016f410b75aSHong Zhang   mumps->id.nrhs = nrhs;
3017f410b75aSHong Zhang   mumps->id.lrhs = M;
3018f410b75aSHong Zhang   mumps->id.rhs  = NULL;
3019f410b75aSHong Zhang 
3020e3f2db6aSHong Zhang   if (!mumps->myid) {
30219566063dSJacob Faibussowitsch     PetscCall(MatSeqAIJGetArray(Btseq, &aa));
30229566063dSJacob Faibussowitsch     PetscCall(MatGetRowIJ(Btseq, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
302328b400f6SJacob Faibussowitsch     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
30249566063dSJacob Faibussowitsch     PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs));
3025bb599dfdSHong Zhang     mumps->id.rhs_sparse = (MumpsScalar *)aa;
3026e3f2db6aSHong Zhang   } else {
3027e3f2db6aSHong Zhang     mumps->id.irhs_ptr    = NULL;
3028e3f2db6aSHong Zhang     mumps->id.irhs_sparse = NULL;
3029e3f2db6aSHong Zhang     mumps->id.nz_rhs      = 0;
3030e3f2db6aSHong Zhang     mumps->id.rhs_sparse  = NULL;
3031e3f2db6aSHong Zhang   }
3032bb599dfdSHong Zhang   mumps->id.ICNTL(20) = 1; /* rhs is sparse */
3033e3f2db6aSHong Zhang   mumps->id.ICNTL(21) = 0; /* solution is in assembled centralized format */
3034bb599dfdSHong Zhang 
3035bb599dfdSHong Zhang   /* solve phase */
3036bb599dfdSHong Zhang   mumps->id.job = JOB_SOLVE;
30373ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
30389261f6e4SBarry 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));
303914267174SHong Zhang 
3040e3f2db6aSHong Zhang   if (!mumps->myid) {
30419566063dSJacob Faibussowitsch     PetscCall(MatSeqAIJRestoreArray(Btseq, &aa));
30429566063dSJacob Faibussowitsch     PetscCall(MatRestoreRowIJ(Btseq, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
304328b400f6SJacob Faibussowitsch     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
3044e3f2db6aSHong Zhang   }
30453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3046bb599dfdSHong Zhang }
3047bb599dfdSHong Zhang 
3048bb599dfdSHong Zhang /*@
30492ef1f0ffSBarry Smith   MatMumpsGetInverse - Get user-specified set of entries in inverse of `A`
3050bb599dfdSHong Zhang 
3051c3339decSBarry Smith   Logically Collective
3052bb599dfdSHong Zhang 
305320f4b53cSBarry Smith   Input Parameter:
305420f4b53cSBarry Smith . F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3055bb599dfdSHong Zhang 
3056bb599dfdSHong Zhang   Output Parameter:
305720f4b53cSBarry Smith . spRHS - sequential sparse matrix in `MATTRANSPOSEVIRTUAL` format with requested entries of inverse of `A`
3058bb599dfdSHong Zhang 
3059bb599dfdSHong Zhang   Level: beginner
3060bb599dfdSHong Zhang 
3061bb599dfdSHong Zhang   References:
3062606c0280SSatish Balay .  * - MUMPS Users' Guide
3063bb599dfdSHong Zhang 
30641cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatCreateTranspose()`
3065bb599dfdSHong Zhang @*/
3066d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInverse(Mat F, Mat spRHS)
3067d71ae5a4SJacob Faibussowitsch {
3068bb599dfdSHong Zhang   PetscFunctionBegin;
3069bb599dfdSHong Zhang   PetscValidType(F, 1);
307028b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
3071cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInverse_C", (Mat, Mat), (F, spRHS));
30723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3073bb599dfdSHong Zhang }
3074bb599dfdSHong Zhang 
307566976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInverseTranspose_MUMPS(Mat F, Mat spRHST)
3076d71ae5a4SJacob Faibussowitsch {
30770e6b8875SHong Zhang   Mat spRHS;
30780e6b8875SHong Zhang 
30790e6b8875SHong Zhang   PetscFunctionBegin;
30809566063dSJacob Faibussowitsch   PetscCall(MatCreateTranspose(spRHST, &spRHS));
30819566063dSJacob Faibussowitsch   PetscCall(MatMumpsGetInverse_MUMPS(F, spRHS));
30829566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&spRHS));
30833ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
30840e6b8875SHong Zhang }
30850e6b8875SHong Zhang 
30860e6b8875SHong Zhang /*@
30872ef1f0ffSBarry Smith   MatMumpsGetInverseTranspose - Get user-specified set of entries in inverse of matrix `A`^T
30880e6b8875SHong Zhang 
3089c3339decSBarry Smith   Logically Collective
30900e6b8875SHong Zhang 
309120f4b53cSBarry Smith   Input Parameter:
309220f4b53cSBarry Smith . F - the factored matrix of A obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
30930e6b8875SHong Zhang 
30940e6b8875SHong Zhang   Output Parameter:
309520f4b53cSBarry Smith . spRHST - sequential sparse matrix in `MATAIJ` format containing the requested entries of inverse of `A`^T
30960e6b8875SHong Zhang 
30970e6b8875SHong Zhang   Level: beginner
30980e6b8875SHong Zhang 
30990e6b8875SHong Zhang   References:
3100606c0280SSatish Balay .  * - MUMPS Users' Guide
31010e6b8875SHong Zhang 
31021cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatCreateTranspose()`, `MatMumpsGetInverse()`
31030e6b8875SHong Zhang @*/
3104d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInverseTranspose(Mat F, Mat spRHST)
3105d71ae5a4SJacob Faibussowitsch {
31060e6b8875SHong Zhang   PetscBool flg;
31070e6b8875SHong Zhang 
31080e6b8875SHong Zhang   PetscFunctionBegin;
31090e6b8875SHong Zhang   PetscValidType(F, 1);
311028b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
31119566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)spRHST, &flg, MATSEQAIJ, MATMPIAIJ, NULL));
311228b400f6SJacob Faibussowitsch   PetscCheck(flg, PetscObjectComm((PetscObject)spRHST), PETSC_ERR_ARG_WRONG, "Matrix spRHST must be MATAIJ matrix");
31130e6b8875SHong Zhang 
3114cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInverseTranspose_C", (Mat, Mat), (F, spRHST));
31153ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
31160e6b8875SHong Zhang }
31170e6b8875SHong Zhang 
3118a21f80fcSHong Zhang /*@
3119a21f80fcSHong Zhang   MatMumpsGetInfo - Get MUMPS parameter INFO()
3120a21f80fcSHong Zhang 
3121c3339decSBarry Smith   Logically Collective
3122a21f80fcSHong Zhang 
3123a21f80fcSHong Zhang   Input Parameters:
312411a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3125a21f80fcSHong Zhang - icntl - index of MUMPS parameter array INFO()
3126a21f80fcSHong Zhang 
3127a21f80fcSHong Zhang   Output Parameter:
3128a21f80fcSHong Zhang . ival - value of MUMPS INFO(icntl)
3129a21f80fcSHong Zhang 
3130a21f80fcSHong Zhang   Level: beginner
3131a21f80fcSHong Zhang 
313296a0c994SBarry Smith   References:
3133606c0280SSatish Balay .  * - MUMPS Users' Guide
3134a21f80fcSHong Zhang 
31351cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
3136a21f80fcSHong Zhang @*/
3137d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInfo(Mat F, PetscInt icntl, PetscInt *ival)
3138d71ae5a4SJacob Faibussowitsch {
3139bc6112feSHong Zhang   PetscFunctionBegin;
31402989dfd4SHong Zhang   PetscValidType(F, 1);
314128b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
31424f572ea9SToby Isaac   PetscAssertPointer(ival, 3);
3143cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInfo_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival));
31443ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3145bc6112feSHong Zhang }
3146bc6112feSHong Zhang 
3147a21f80fcSHong Zhang /*@
3148a21f80fcSHong Zhang   MatMumpsGetInfog - Get MUMPS parameter INFOG()
3149a21f80fcSHong Zhang 
3150c3339decSBarry Smith   Logically Collective
3151a21f80fcSHong Zhang 
3152a21f80fcSHong Zhang   Input Parameters:
315311a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3154a21f80fcSHong Zhang - icntl - index of MUMPS parameter array INFOG()
3155a21f80fcSHong Zhang 
3156a21f80fcSHong Zhang   Output Parameter:
3157a21f80fcSHong Zhang . ival - value of MUMPS INFOG(icntl)
3158a21f80fcSHong Zhang 
3159a21f80fcSHong Zhang   Level: beginner
3160a21f80fcSHong Zhang 
316196a0c994SBarry Smith   References:
3162606c0280SSatish Balay .  * - MUMPS Users' Guide
3163a21f80fcSHong Zhang 
31641cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
3165a21f80fcSHong Zhang @*/
3166d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInfog(Mat F, PetscInt icntl, PetscInt *ival)
3167d71ae5a4SJacob Faibussowitsch {
3168bc6112feSHong Zhang   PetscFunctionBegin;
31692989dfd4SHong Zhang   PetscValidType(F, 1);
317028b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
31714f572ea9SToby Isaac   PetscAssertPointer(ival, 3);
3172cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInfog_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival));
31733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3174bc6112feSHong Zhang }
3175bc6112feSHong Zhang 
3176a21f80fcSHong Zhang /*@
3177a21f80fcSHong Zhang   MatMumpsGetRinfo - Get MUMPS parameter RINFO()
3178a21f80fcSHong Zhang 
3179c3339decSBarry Smith   Logically Collective
3180a21f80fcSHong Zhang 
3181a21f80fcSHong Zhang   Input Parameters:
318211a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3183a21f80fcSHong Zhang - icntl - index of MUMPS parameter array RINFO()
3184a21f80fcSHong Zhang 
3185a21f80fcSHong Zhang   Output Parameter:
3186a21f80fcSHong Zhang . val - value of MUMPS RINFO(icntl)
3187a21f80fcSHong Zhang 
3188a21f80fcSHong Zhang   Level: beginner
3189a21f80fcSHong Zhang 
319096a0c994SBarry Smith   References:
3191606c0280SSatish Balay .  * - MUMPS Users' Guide
3192a21f80fcSHong Zhang 
31931cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfog()`
3194a21f80fcSHong Zhang @*/
3195d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetRinfo(Mat F, PetscInt icntl, PetscReal *val)
3196d71ae5a4SJacob Faibussowitsch {
3197bc6112feSHong Zhang   PetscFunctionBegin;
31982989dfd4SHong Zhang   PetscValidType(F, 1);
319928b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
32004f572ea9SToby Isaac   PetscAssertPointer(val, 3);
3201cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetRinfo_C", (Mat, PetscInt, PetscReal *), (F, icntl, val));
32023ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3203bc6112feSHong Zhang }
3204bc6112feSHong Zhang 
3205a21f80fcSHong Zhang /*@
3206a21f80fcSHong Zhang   MatMumpsGetRinfog - Get MUMPS parameter RINFOG()
3207a21f80fcSHong Zhang 
3208c3339decSBarry Smith   Logically Collective
3209a21f80fcSHong Zhang 
3210a21f80fcSHong Zhang   Input Parameters:
321111a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3212a21f80fcSHong Zhang - icntl - index of MUMPS parameter array RINFOG()
3213a21f80fcSHong Zhang 
3214a21f80fcSHong Zhang   Output Parameter:
3215a21f80fcSHong Zhang . val - value of MUMPS RINFOG(icntl)
3216a21f80fcSHong Zhang 
3217a21f80fcSHong Zhang   Level: beginner
3218a21f80fcSHong Zhang 
321996a0c994SBarry Smith   References:
3220606c0280SSatish Balay .  * - MUMPS Users' Guide
3221a21f80fcSHong Zhang 
32221cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`
3223a21f80fcSHong Zhang @*/
3224d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetRinfog(Mat F, PetscInt icntl, PetscReal *val)
3225d71ae5a4SJacob Faibussowitsch {
3226bc6112feSHong Zhang   PetscFunctionBegin;
32272989dfd4SHong Zhang   PetscValidType(F, 1);
322828b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
32294f572ea9SToby Isaac   PetscAssertPointer(val, 3);
3230cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetRinfog_C", (Mat, PetscInt, PetscReal *), (F, icntl, val));
32313ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3232bc6112feSHong Zhang }
3233bc6112feSHong Zhang 
32345c0bae8cSAshish Patel /*@
32355c0bae8cSAshish Patel   MatMumpsGetNullPivots - Get MUMPS parameter PIVNUL_LIST()
32365c0bae8cSAshish Patel 
32375c0bae8cSAshish Patel   Logically Collective
32385c0bae8cSAshish Patel 
32395c0bae8cSAshish Patel   Input Parameter:
32405c0bae8cSAshish Patel . F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
32415c0bae8cSAshish Patel 
32425c0bae8cSAshish Patel   Output Parameters:
32435c0bae8cSAshish Patel + size  - local size of the array. The size of the array is non-zero only on the host.
32445c0bae8cSAshish 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
32455c0bae8cSAshish Patel            for freeing this array.
32465c0bae8cSAshish Patel 
32475c0bae8cSAshish Patel   Level: beginner
32485c0bae8cSAshish Patel 
32495c0bae8cSAshish Patel   References:
32505c0bae8cSAshish Patel .  * - MUMPS Users' Guide
32515c0bae8cSAshish Patel 
32521cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`
32535c0bae8cSAshish Patel @*/
32545c0bae8cSAshish Patel PetscErrorCode MatMumpsGetNullPivots(Mat F, PetscInt *size, PetscInt **array)
32555c0bae8cSAshish Patel {
32565c0bae8cSAshish Patel   PetscFunctionBegin;
32575c0bae8cSAshish Patel   PetscValidType(F, 1);
32585c0bae8cSAshish Patel   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
32594f572ea9SToby Isaac   PetscAssertPointer(size, 2);
32604f572ea9SToby Isaac   PetscAssertPointer(array, 3);
32615c0bae8cSAshish Patel   PetscUseMethod(F, "MatMumpsGetNullPivots_C", (Mat, PetscInt *, PetscInt **), (F, size, array));
32625c0bae8cSAshish Patel   PetscFunctionReturn(PETSC_SUCCESS);
32635c0bae8cSAshish Patel }
32645c0bae8cSAshish Patel 
326524b6179bSKris Buschelman /*MC
32662692d6eeSBarry Smith   MATSOLVERMUMPS -  A matrix type providing direct solvers (LU and Cholesky) for
326724b6179bSKris Buschelman   distributed and sequential matrices via the external package MUMPS.
326824b6179bSKris Buschelman 
326911a5261eSBarry Smith   Works with `MATAIJ` and `MATSBAIJ` matrices
327024b6179bSKris Buschelman 
3271c2b89b5dSBarry Smith   Use ./configure --download-mumps --download-scalapack --download-parmetis --download-metis --download-ptscotch to have PETSc installed with MUMPS
3272c2b89b5dSBarry Smith 
32732ef1f0ffSBarry 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.
32742ef1f0ffSBarry Smith   See details below.
3275217d3b1eSJunchao Zhang 
32762ef1f0ffSBarry Smith   Use `-pc_type cholesky` or `lu` `-pc_factor_mat_solver_type mumps` to use this direct solver
3277c2b89b5dSBarry Smith 
327824b6179bSKris Buschelman   Options Database Keys:
32794422a9fcSPatrick Sanan +  -mat_mumps_icntl_1 - ICNTL(1): output stream for error messages
32804422a9fcSPatrick Sanan .  -mat_mumps_icntl_2 - ICNTL(2): output stream for diagnostic printing, statistics, and warning
32814422a9fcSPatrick Sanan .  -mat_mumps_icntl_3 -  ICNTL(3): output stream for global information, collected on the host
32824422a9fcSPatrick Sanan .  -mat_mumps_icntl_4 -  ICNTL(4): level of printing (0 to 4)
32834422a9fcSPatrick Sanan .  -mat_mumps_icntl_6 - ICNTL(6): permutes to a zero-free diagonal and/or scale the matrix (0 to 7)
3284b53c1a7fSBarry 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
3285b53c1a7fSBarry Smith                         Use -pc_factor_mat_ordering_type <type> to have PETSc perform the ordering (sequential only)
32864422a9fcSPatrick Sanan .  -mat_mumps_icntl_8  - ICNTL(8): scaling strategy (-2 to 8 or 77)
32874422a9fcSPatrick Sanan .  -mat_mumps_icntl_10  - ICNTL(10): max num of refinements
32884422a9fcSPatrick Sanan .  -mat_mumps_icntl_11  - ICNTL(11): statistics related to an error analysis (via -ksp_view)
32894422a9fcSPatrick Sanan .  -mat_mumps_icntl_12  - ICNTL(12): an ordering strategy for symmetric matrices (0 to 3)
32904422a9fcSPatrick Sanan .  -mat_mumps_icntl_13  - ICNTL(13): parallelism of the root node (enable ScaLAPACK) and its splitting
32914422a9fcSPatrick Sanan .  -mat_mumps_icntl_14  - ICNTL(14): percentage increase in the estimated working space
329245e3843bSPierre Jolivet .  -mat_mumps_icntl_15  - ICNTL(15): compression of the input matrix resulting from a block format
32934422a9fcSPatrick Sanan .  -mat_mumps_icntl_19  - ICNTL(19): computes the Schur complement
329425aac85cSJunchao Zhang .  -mat_mumps_icntl_20  - ICNTL(20): give MUMPS centralized (0) or distributed (10) dense RHS
32954422a9fcSPatrick Sanan .  -mat_mumps_icntl_22  - ICNTL(22): in-core/out-of-core factorization and solve (0 or 1)
32964422a9fcSPatrick Sanan .  -mat_mumps_icntl_23  - ICNTL(23): max size of the working memory (MB) that can allocate per processor
32974422a9fcSPatrick Sanan .  -mat_mumps_icntl_24  - ICNTL(24): detection of null pivot rows (0 or 1)
32984422a9fcSPatrick Sanan .  -mat_mumps_icntl_25  - ICNTL(25): compute a solution of a deficient matrix and a null space basis
32994422a9fcSPatrick Sanan .  -mat_mumps_icntl_26  - ICNTL(26): drives the solution phase if a Schur complement matrix
33004422a9fcSPatrick Sanan .  -mat_mumps_icntl_28  - ICNTL(28): use 1 for sequential analysis and ictnl(7) ordering, or 2 for parallel analysis and ictnl(29) ordering
33014422a9fcSPatrick Sanan .  -mat_mumps_icntl_29 - ICNTL(29): parallel ordering 1 = ptscotch, 2 = parmetis
33024422a9fcSPatrick Sanan .  -mat_mumps_icntl_30 - ICNTL(30): compute user-specified set of entries in inv(A)
33034422a9fcSPatrick Sanan .  -mat_mumps_icntl_31 - ICNTL(31): indicates which factors may be discarded during factorization
33044422a9fcSPatrick Sanan .  -mat_mumps_icntl_33 - ICNTL(33): compute determinant
3305a0e18203SThibaut Appel .  -mat_mumps_icntl_35 - ICNTL(35): level of activation of BLR (Block Low-Rank) feature
3306a0e18203SThibaut Appel .  -mat_mumps_icntl_36 - ICNTL(36): controls the choice of BLR factorization variant
3307a0e18203SThibaut Appel .  -mat_mumps_icntl_38 - ICNTL(38): sets the estimated compression rate of LU factors with BLR
3308146931dbSPierre Jolivet .  -mat_mumps_icntl_58 - ICNTL(58): options for symbolic factorization
33094422a9fcSPatrick Sanan .  -mat_mumps_cntl_1  - CNTL(1): relative pivoting threshold
33104422a9fcSPatrick Sanan .  -mat_mumps_cntl_2  -  CNTL(2): stopping criterion of refinement
33114422a9fcSPatrick Sanan .  -mat_mumps_cntl_3 - CNTL(3): absolute pivoting threshold
33124422a9fcSPatrick Sanan .  -mat_mumps_cntl_4 - CNTL(4): value for static pivoting
3313217d3b1eSJunchao Zhang .  -mat_mumps_cntl_5 - CNTL(5): fixation for null pivots
3314a0e18203SThibaut Appel .  -mat_mumps_cntl_7 - CNTL(7): precision of the dropping parameter used during BLR factorization
3315217d3b1eSJunchao 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.
3316217d3b1eSJunchao Zhang                                    Default might be the number of cores per CPU package (socket) as reported by hwloc and suggested by the MUMPS manual.
331724b6179bSKris Buschelman 
331824b6179bSKris Buschelman   Level: beginner
331924b6179bSKris Buschelman 
332095452b02SPatrick Sanan     Notes:
33212ef1f0ffSBarry 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
33222ef1f0ffSBarry Smith     error if the matrix is Hermitian.
332338548759SBarry Smith 
332426cc229bSBarry 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
332526cc229bSBarry Smith     `MatSetOptionsPrefixFactor()` on the matrix from which the factor was obtained or `MatSetOptionsPrefix()` on the factor matrix.
332626cc229bSBarry Smith 
33272ef1f0ffSBarry 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
33282ef1f0ffSBarry Smith     the failure with
33292ef1f0ffSBarry Smith .vb
33302ef1f0ffSBarry Smith           KSPGetPC(ksp,&pc);
33312ef1f0ffSBarry Smith           PCFactorGetMatrix(pc,&mat);
33322ef1f0ffSBarry Smith           MatMumpsGetInfo(mat,....);
33332ef1f0ffSBarry Smith           MatMumpsGetInfog(mat,....); etc.
33342ef1f0ffSBarry Smith .ve
33352ef1f0ffSBarry Smith     Or run with `-ksp_error_if_not_converged` and the program will be stopped and the information printed in the error message.
33369fc87aa7SBarry Smith 
3337a5399872SJunchao Zhang     MUMPS provides 64-bit integer support in two build modes:
3338a5399872SJunchao Zhang       full 64-bit: here MUMPS is built with C preprocessing flag -DINTSIZE64 and Fortran compiler option -i8, -fdefault-integer-8 or equivalent, and
3339a5399872SJunchao 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).
33408fcaa860SBarry Smith 
3341a5399872SJunchao 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,
3342a5399872SJunchao 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
3343a5399872SJunchao 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
3344a5399872SJunchao Zhang       integer) build of all dependent libraries MPI, ScaLAPACK, LAPACK and BLAS.
3345a5399872SJunchao Zhang 
3346a5399872SJunchao 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.
3347a5399872SJunchao Zhang 
3348a5399872SJunchao Zhang   Two modes to run MUMPS/PETSc with OpenMP
33492ef1f0ffSBarry Smith .vb
33502ef1f0ffSBarry Smith      Set OMP_NUM_THREADS and run with fewer MPI ranks than cores. For example, if you want to have 16 OpenMP
33512ef1f0ffSBarry Smith      threads per rank, then you may use "export OMP_NUM_THREADS=16 && mpirun -n 4 ./test".
33522ef1f0ffSBarry Smith .ve
33538fcaa860SBarry Smith 
33542ef1f0ffSBarry Smith .vb
33552ef1f0ffSBarry Smith      -mat_mumps_use_omp_threads [m] and run your code with as many MPI ranks as the number of cores. For example,
33562ef1f0ffSBarry 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"
33572ef1f0ffSBarry Smith .ve
33588fcaa860SBarry Smith 
33598fcaa860SBarry Smith    To run MUMPS in MPI+OpenMP hybrid mode (i.e., enable multithreading in MUMPS), but still run the non-MUMPS part
33602ef1f0ffSBarry 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`
33612ef1f0ffSBarry 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
33628fcaa860SBarry Smith    libraries, to really get performance, you should have multithreaded BLAS libraries such as Intel MKL, AMD ACML, Cray libSci or OpenBLAS
33638fcaa860SBarry Smith    (PETSc will automatically try to utilized a threaded BLAS if --with-openmp is provided).
3364217d3b1eSJunchao Zhang 
33658fcaa860SBarry 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
3366217d3b1eSJunchao Zhang    processes on each compute node. Listing the processes in rank ascending order, we split processes on a node into consecutive groups of
3367217d3b1eSJunchao 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
3368217d3b1eSJunchao 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
3369217d3b1eSJunchao 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.
3370217d3b1eSJunchao 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,
3371217d3b1eSJunchao 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
3372217d3b1eSJunchao Zhang    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
3373217d3b1eSJunchao 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
3374217d3b1eSJunchao Zhang    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.
33758fcaa860SBarry 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
3376217d3b1eSJunchao Zhang    examine the mapping result.
3377217d3b1eSJunchao Zhang 
337811a5261eSBarry 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,
337911a5261eSBarry 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
338011a5261eSBarry Smith    calls `omp_set_num_threads`(m) internally before calling MUMPS.
3381217d3b1eSJunchao Zhang 
3382217d3b1eSJunchao Zhang    References:
3383606c0280SSatish Balay +  * - Heroux, Michael A., R. Brightwell, and Michael M. Wolf. "Bi-modal MPI and MPI+ threads computing on scalable multicore systems." IJHPCA (Submitted) (2011).
3384606c0280SSatish Balay -  * - Gutierrez, Samuel K., et al. "Accommodating Thread-Level Heterogeneity in Coupled Parallel Applications." Parallel and Distributed Processing Symposium (IPDPS), 2017 IEEE International. IEEE, 2017.
3385217d3b1eSJunchao Zhang 
33861cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `PCFactorSetMatSolverType()`, `MatSolverType`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`, `KSPGetPC()`, `PCFactorGetMatrix()`
338724b6179bSKris Buschelman M*/
338824b6179bSKris Buschelman 
3389d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatFactorGetSolverType_mumps(Mat A, MatSolverType *type)
3390d71ae5a4SJacob Faibussowitsch {
339135bd34faSBarry Smith   PetscFunctionBegin;
33922692d6eeSBarry Smith   *type = MATSOLVERMUMPS;
33933ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
339435bd34faSBarry Smith }
339535bd34faSBarry Smith 
3396bccb9932SShri Abhyankar /* MatGetFactor for Seq and MPI AIJ matrices */
3397d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_aij_mumps(Mat A, MatFactorType ftype, Mat *F)
3398d71ae5a4SJacob Faibussowitsch {
33992877fffaSHong Zhang   Mat         B;
34002877fffaSHong Zhang   Mat_MUMPS  *mumps;
3401*4b9405b2SPierre Jolivet   PetscBool   isSeqAIJ, isDiag, isDense;
34022c7c0729SBarry Smith   PetscMPIInt size;
34032877fffaSHong Zhang 
34042877fffaSHong Zhang   PetscFunctionBegin;
3405eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
340603e5aca4SStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) {
340703e5aca4SStefano Zampini     PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n"));
340803e5aca4SStefano Zampini     *F = NULL;
340903e5aca4SStefano Zampini     PetscFunctionReturn(PETSC_SUCCESS);
341003e5aca4SStefano Zampini   }
3411eb1ec7c1SStefano Zampini #endif
34122877fffaSHong Zhang   /* Create the factorization matrix */
34139566063dSJacob Faibussowitsch   PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATSEQAIJ, &isSeqAIJ));
3414c3e1b152SPierre Jolivet   PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATDIAGONAL, &isDiag));
3415*4b9405b2SPierre Jolivet   PetscCall(PetscObjectTypeCompareAny((PetscObject)A, &isDense, MATSEQDENSE, MATMPIDENSE, NULL));
34169566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
34179566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
34189566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name));
34199566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
34202877fffaSHong Zhang 
34214dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
34222205254eSKarl Rupp 
34232877fffaSHong Zhang   B->ops->view    = MatView_MUMPS;
342435bd34faSBarry Smith   B->ops->getinfo = MatGetInfo_MUMPS;
34252205254eSKarl Rupp 
34269566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
34279566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
34289566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
34299566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
34309566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
34319566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
34329566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
34339566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
34349566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
34359566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
34369566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
34375c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
34389566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
34399566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
34406444a565SStefano Zampini 
3441450b117fSShri Abhyankar   if (ftype == MAT_FACTOR_LU) {
3442450b117fSShri Abhyankar     B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS;
3443d5f3da31SBarry Smith     B->factortype            = MAT_FACTOR_LU;
3444bccb9932SShri Abhyankar     if (isSeqAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqaij_seqaij;
3445c3e1b152SPierre Jolivet     else if (isDiag) mumps->ConvertToTriples = MatConvertToTriples_diagonal_xaij;
3446*4b9405b2SPierre Jolivet     else if (isDense) mumps->ConvertToTriples = MatConvertToTriples_dense_xaij;
3447bccb9932SShri Abhyankar     else mumps->ConvertToTriples = MatConvertToTriples_mpiaij_mpiaij;
34489566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU]));
3449746480a1SHong Zhang     mumps->sym = 0;
3450dcd589f8SShri Abhyankar   } else {
345167877ebaSShri Abhyankar     B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS;
3452450b117fSShri Abhyankar     B->factortype                  = MAT_FACTOR_CHOLESKY;
3453bccb9932SShri Abhyankar     if (isSeqAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqaij_seqsbaij;
3454c3e1b152SPierre Jolivet     else if (isDiag) mumps->ConvertToTriples = MatConvertToTriples_diagonal_xaij;
3455*4b9405b2SPierre Jolivet     else if (isDense) mumps->ConvertToTriples = MatConvertToTriples_dense_xaij;
3456bccb9932SShri Abhyankar     else mumps->ConvertToTriples = MatConvertToTriples_mpiaij_mpisbaij;
34579566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_CHOLESKY]));
345859ac8732SStefano Zampini #if defined(PETSC_USE_COMPLEX)
345959ac8732SStefano Zampini     mumps->sym = 2;
346059ac8732SStefano Zampini #else
3461b94d7dedSBarry Smith     if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1;
34626fdc2a6dSBarry Smith     else mumps->sym = 2;
346359ac8732SStefano Zampini #endif
3464450b117fSShri Abhyankar   }
34652877fffaSHong Zhang 
346600c67f3bSHong Zhang   /* set solvertype */
34679566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
34689566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
34699566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
34702c7c0729SBarry Smith   if (size == 1) {
34714ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
3472f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
34732c7c0729SBarry Smith   }
34742877fffaSHong Zhang   B->ops->destroy = MatDestroy_MUMPS;
3475e69c285eSBarry Smith   B->data         = (void *)mumps;
34762205254eSKarl Rupp 
34772877fffaSHong Zhang   *F               = B;
3478413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3479413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3480413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3481d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
34823ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
34832877fffaSHong Zhang }
34842877fffaSHong Zhang 
3485bccb9932SShri Abhyankar /* MatGetFactor for Seq and MPI SBAIJ matrices */
3486d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_sbaij_mumps(Mat A, MatFactorType ftype, Mat *F)
3487d71ae5a4SJacob Faibussowitsch {
34882877fffaSHong Zhang   Mat         B;
34892877fffaSHong Zhang   Mat_MUMPS  *mumps;
3490ace3abfcSBarry Smith   PetscBool   isSeqSBAIJ;
34912c7c0729SBarry Smith   PetscMPIInt size;
34922877fffaSHong Zhang 
34932877fffaSHong Zhang   PetscFunctionBegin;
3494eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
349503e5aca4SStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) {
349603e5aca4SStefano Zampini     PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n"));
349703e5aca4SStefano Zampini     *F = NULL;
349803e5aca4SStefano Zampini     PetscFunctionReturn(PETSC_SUCCESS);
349903e5aca4SStefano Zampini   }
3500eb1ec7c1SStefano Zampini #endif
35019566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
35029566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
35039566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name));
35049566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
3505e69c285eSBarry Smith 
35064dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
35079566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQSBAIJ, &isSeqSBAIJ));
3508bccb9932SShri Abhyankar   if (isSeqSBAIJ) {
350916ebf90aSShri Abhyankar     mumps->ConvertToTriples = MatConvertToTriples_seqsbaij_seqsbaij;
3510dcd589f8SShri Abhyankar   } else {
3511bccb9932SShri Abhyankar     mumps->ConvertToTriples = MatConvertToTriples_mpisbaij_mpisbaij;
3512bccb9932SShri Abhyankar   }
3513bccb9932SShri Abhyankar 
351467877ebaSShri Abhyankar   B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS;
3515bccb9932SShri Abhyankar   B->ops->view                   = MatView_MUMPS;
3516722b6324SPierre Jolivet   B->ops->getinfo                = MatGetInfo_MUMPS;
35172205254eSKarl Rupp 
35189566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
35199566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
35209566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
35219566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
35229566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
35239566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
35249566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
35259566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
35269566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
35279566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
35289566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
35295c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
35309566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
35319566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
35322205254eSKarl Rupp 
3533f4762488SHong Zhang   B->factortype = MAT_FACTOR_CHOLESKY;
353459ac8732SStefano Zampini #if defined(PETSC_USE_COMPLEX)
353559ac8732SStefano Zampini   mumps->sym = 2;
353659ac8732SStefano Zampini #else
3537b94d7dedSBarry Smith   if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1;
35386fdc2a6dSBarry Smith   else mumps->sym = 2;
353959ac8732SStefano Zampini #endif
3540a214ac2aSShri Abhyankar 
354100c67f3bSHong Zhang   /* set solvertype */
35429566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
35439566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
35449566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
35452c7c0729SBarry Smith   if (size == 1) {
35464ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
3547f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
35482c7c0729SBarry Smith   }
35499566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_CHOLESKY]));
3550f3c0ef26SHong Zhang   B->ops->destroy = MatDestroy_MUMPS;
3551e69c285eSBarry Smith   B->data         = (void *)mumps;
35522205254eSKarl Rupp 
35532877fffaSHong Zhang   *F               = B;
3554413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3555413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3556413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3557d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
35583ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
35592877fffaSHong Zhang }
356097969023SHong Zhang 
3561d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_baij_mumps(Mat A, MatFactorType ftype, Mat *F)
3562d71ae5a4SJacob Faibussowitsch {
356367877ebaSShri Abhyankar   Mat         B;
356467877ebaSShri Abhyankar   Mat_MUMPS  *mumps;
3565ace3abfcSBarry Smith   PetscBool   isSeqBAIJ;
35662c7c0729SBarry Smith   PetscMPIInt size;
356767877ebaSShri Abhyankar 
356867877ebaSShri Abhyankar   PetscFunctionBegin;
356967877ebaSShri Abhyankar   /* Create the factorization matrix */
35709566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQBAIJ, &isSeqBAIJ));
35719566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
35729566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
35739566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name));
35749566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
3575450b117fSShri Abhyankar 
35764dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
3577450b117fSShri Abhyankar   if (ftype == MAT_FACTOR_LU) {
3578450b117fSShri Abhyankar     B->ops->lufactorsymbolic = MatLUFactorSymbolic_BAIJMUMPS;
3579450b117fSShri Abhyankar     B->factortype            = MAT_FACTOR_LU;
3580bccb9932SShri Abhyankar     if (isSeqBAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqbaij_seqaij;
3581bccb9932SShri Abhyankar     else mumps->ConvertToTriples = MatConvertToTriples_mpibaij_mpiaij;
3582746480a1SHong Zhang     mumps->sym = 0;
35839566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU]));
3584546078acSJacob Faibussowitsch   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot use PETSc BAIJ matrices with MUMPS Cholesky, use SBAIJ or AIJ matrix instead");
3585bccb9932SShri Abhyankar 
3586450b117fSShri Abhyankar   B->ops->view    = MatView_MUMPS;
3587722b6324SPierre Jolivet   B->ops->getinfo = MatGetInfo_MUMPS;
35882205254eSKarl Rupp 
35899566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
35909566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
35919566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
35929566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
35939566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
35949566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
35959566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
35969566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
35979566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
35989566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
35999566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
36005c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
36019566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
36029566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
3603450b117fSShri Abhyankar 
360400c67f3bSHong Zhang   /* set solvertype */
36059566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
36069566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
36079566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
36082c7c0729SBarry Smith   if (size == 1) {
36094ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
3610f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
36112c7c0729SBarry Smith   }
36127ee00b23SStefano Zampini   B->ops->destroy = MatDestroy_MUMPS;
36137ee00b23SStefano Zampini   B->data         = (void *)mumps;
36147ee00b23SStefano Zampini 
36157ee00b23SStefano Zampini   *F               = B;
3616413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3617413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3618413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3619d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
36203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
36217ee00b23SStefano Zampini }
36227ee00b23SStefano Zampini 
36237ee00b23SStefano Zampini /* MatGetFactor for Seq and MPI SELL matrices */
3624d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_sell_mumps(Mat A, MatFactorType ftype, Mat *F)
3625d71ae5a4SJacob Faibussowitsch {
36267ee00b23SStefano Zampini   Mat         B;
36277ee00b23SStefano Zampini   Mat_MUMPS  *mumps;
36287ee00b23SStefano Zampini   PetscBool   isSeqSELL;
36292c7c0729SBarry Smith   PetscMPIInt size;
36307ee00b23SStefano Zampini 
36317ee00b23SStefano Zampini   PetscFunctionBegin;
36327ee00b23SStefano Zampini   /* Create the factorization matrix */
36339566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQSELL, &isSeqSELL));
36349566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
36359566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
36369566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name));
36379566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
36387ee00b23SStefano Zampini 
36394dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
36407ee00b23SStefano Zampini 
36417ee00b23SStefano Zampini   B->ops->view    = MatView_MUMPS;
36427ee00b23SStefano Zampini   B->ops->getinfo = MatGetInfo_MUMPS;
36437ee00b23SStefano Zampini 
36449566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
36459566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
36469566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
36479566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
36489566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
36499566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
36509566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
36519566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
36529566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
36539566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
36549566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
36555c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
36567ee00b23SStefano Zampini 
36577ee00b23SStefano Zampini   if (ftype == MAT_FACTOR_LU) {
36587ee00b23SStefano Zampini     B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS;
36597ee00b23SStefano Zampini     B->factortype            = MAT_FACTOR_LU;
36607ee00b23SStefano Zampini     if (isSeqSELL) mumps->ConvertToTriples = MatConvertToTriples_seqsell_seqaij;
36617ee00b23SStefano Zampini     else SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "To be implemented");
36627ee00b23SStefano Zampini     mumps->sym = 0;
36639566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU]));
36647ee00b23SStefano Zampini   } else SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "To be implemented");
36657ee00b23SStefano Zampini 
36667ee00b23SStefano Zampini   /* set solvertype */
36679566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
36689566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
36699566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
36702c7c0729SBarry Smith   if (size == 1) {
36714ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization  */
3672f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
36732c7c0729SBarry Smith   }
3674450b117fSShri Abhyankar   B->ops->destroy = MatDestroy_MUMPS;
3675e69c285eSBarry Smith   B->data         = (void *)mumps;
36762205254eSKarl Rupp 
3677450b117fSShri Abhyankar   *F               = B;
3678413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3679413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3680413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3681d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
36823ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3683450b117fSShri Abhyankar }
368442c9c57cSBarry Smith 
36859d0448ceSStefano Zampini /* MatGetFactor for MATNEST matrices */
36869d0448ceSStefano Zampini static PetscErrorCode MatGetFactor_nest_mumps(Mat A, MatFactorType ftype, Mat *F)
36879d0448ceSStefano Zampini {
36889d0448ceSStefano Zampini   Mat         B, **mats;
36899d0448ceSStefano Zampini   Mat_MUMPS  *mumps;
36909d0448ceSStefano Zampini   PetscInt    nr, nc;
36919d0448ceSStefano Zampini   PetscMPIInt size;
369203e5aca4SStefano Zampini   PetscBool   flg = PETSC_TRUE;
36939d0448ceSStefano Zampini 
36949d0448ceSStefano Zampini   PetscFunctionBegin;
36959d0448ceSStefano Zampini #if defined(PETSC_USE_COMPLEX)
369603e5aca4SStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) {
369703e5aca4SStefano Zampini     PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n"));
369803e5aca4SStefano Zampini     *F = NULL;
369903e5aca4SStefano Zampini     PetscFunctionReturn(PETSC_SUCCESS);
370003e5aca4SStefano Zampini   }
37019d0448ceSStefano Zampini #endif
37029d0448ceSStefano Zampini 
370303e5aca4SStefano Zampini   /* Return if some condition is not satisfied */
370403e5aca4SStefano Zampini   *F = NULL;
37059d0448ceSStefano Zampini   PetscCall(MatNestGetSubMats(A, &nr, &nc, &mats));
37069d0448ceSStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY) {
37079d0448ceSStefano Zampini     IS       *rows, *cols;
37089d0448ceSStefano Zampini     PetscInt *m, *M;
37099d0448ceSStefano Zampini 
37109d0448ceSStefano 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);
37119d0448ceSStefano Zampini     PetscCall(PetscMalloc2(nr, &rows, nc, &cols));
37129d0448ceSStefano Zampini     PetscCall(MatNestGetISs(A, rows, cols));
37139d0448ceSStefano Zampini     for (PetscInt r = 0; flg && r < nr; r++) PetscCall(ISEqualUnsorted(rows[r], cols[r], &flg));
371403e5aca4SStefano Zampini     if (!flg) {
371503e5aca4SStefano Zampini       PetscCall(PetscFree2(rows, cols));
371603e5aca4SStefano Zampini       PetscCall(PetscInfo(A, "MAT_FACTOR_CHOLESKY not supported for unequal row and column maps. Use MAT_FACTOR_LU.\n"));
371703e5aca4SStefano Zampini       PetscFunctionReturn(PETSC_SUCCESS);
371803e5aca4SStefano Zampini     }
37199d0448ceSStefano Zampini     PetscCall(PetscMalloc2(nr, &m, nr, &M));
37209d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) PetscCall(ISGetMinMax(rows[r], &m[r], &M[r]));
37219d0448ceSStefano Zampini     for (PetscInt r = 0; flg && r < nr; r++)
37229d0448ceSStefano Zampini       for (PetscInt k = r + 1; flg && k < nr; k++)
37239d0448ceSStefano Zampini         if ((m[k] <= m[r] && m[r] <= M[k]) || (m[k] <= M[r] && M[r] <= M[k])) flg = PETSC_FALSE;
37249d0448ceSStefano Zampini     PetscCall(PetscFree2(m, M));
37259d0448ceSStefano Zampini     PetscCall(PetscFree2(rows, cols));
372603e5aca4SStefano Zampini     if (!flg) {
372703e5aca4SStefano Zampini       PetscCall(PetscInfo(A, "MAT_FACTOR_CHOLESKY not supported for intersecting row maps. Use MAT_FACTOR_LU.\n"));
372803e5aca4SStefano Zampini       PetscFunctionReturn(PETSC_SUCCESS);
372903e5aca4SStefano Zampini     }
37309d0448ceSStefano Zampini   }
37319d0448ceSStefano Zampini 
37329d0448ceSStefano Zampini   for (PetscInt r = 0; r < nr; r++) {
37339d0448ceSStefano Zampini     for (PetscInt c = 0; c < nc; c++) {
37349d0448ceSStefano Zampini       Mat       sub = mats[r][c];
3735*4b9405b2SPierre Jolivet       PetscBool isSeqAIJ, isMPIAIJ, isSeqBAIJ, isMPIBAIJ, isSeqSBAIJ, isMPISBAIJ, isTrans, isDiag, isDense;
37369d0448ceSStefano Zampini 
37379d0448ceSStefano Zampini       if (!sub || (ftype == MAT_FACTOR_CHOLESKY && c < r)) continue;
37385d955bbbSStefano Zampini       PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATTRANSPOSEVIRTUAL, &isTrans));
37395d955bbbSStefano Zampini       if (isTrans) PetscCall(MatTransposeGetMat(sub, &sub));
37405d955bbbSStefano Zampini       else {
37415d955bbbSStefano Zampini         PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATHERMITIANTRANSPOSEVIRTUAL, &isTrans));
37425d955bbbSStefano Zampini         if (isTrans) PetscCall(MatHermitianTransposeGetMat(sub, &sub));
37435d955bbbSStefano Zampini       }
37449d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQAIJ, &isSeqAIJ));
37459d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIAIJ, &isMPIAIJ));
37469d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQBAIJ, &isSeqBAIJ));
37479d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIBAIJ, &isMPIBAIJ));
37489d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQSBAIJ, &isSeqSBAIJ));
37499d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPISBAIJ, &isMPISBAIJ));
3750c3e1b152SPierre Jolivet       PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATDIAGONAL, &isDiag));
3751*4b9405b2SPierre Jolivet       PetscCall(PetscObjectTypeCompareAny((PetscObject)sub, &isDense, MATSEQDENSE, MATMPIDENSE, NULL));
37529d0448ceSStefano Zampini       if (ftype == MAT_FACTOR_CHOLESKY) {
3753dcab004fSPierre Jolivet         if (r == c) {
3754*4b9405b2SPierre Jolivet           if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isSeqSBAIJ && !isMPISBAIJ && !isDiag && !isDense) {
375503e5aca4SStefano Zampini             PetscCall(PetscInfo(sub, "MAT_CHOLESKY_FACTOR not supported for diagonal block of type %s.\n", ((PetscObject)sub)->type_name));
375603e5aca4SStefano Zampini             flg = PETSC_FALSE;
3757dcab004fSPierre Jolivet           }
3758*4b9405b2SPierre Jolivet         } else if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isDiag && !isDense) {
375903e5aca4SStefano Zampini           PetscCall(PetscInfo(sub, "MAT_CHOLESKY_FACTOR not supported for off-diagonal block of type %s.\n", ((PetscObject)sub)->type_name));
376003e5aca4SStefano Zampini           flg = PETSC_FALSE;
376103e5aca4SStefano Zampini         }
3762*4b9405b2SPierre Jolivet       } else if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isDiag && !isDense) {
376303e5aca4SStefano Zampini         PetscCall(PetscInfo(sub, "MAT_LU_FACTOR not supported for block of type %s.\n", ((PetscObject)sub)->type_name));
376403e5aca4SStefano Zampini         flg = PETSC_FALSE;
37659d0448ceSStefano Zampini       }
37669d0448ceSStefano Zampini     }
376703e5aca4SStefano Zampini   }
376803e5aca4SStefano Zampini   if (!flg) PetscFunctionReturn(PETSC_SUCCESS);
37699d0448ceSStefano Zampini 
37709d0448ceSStefano Zampini   /* Create the factorization matrix */
37719d0448ceSStefano Zampini   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
37729d0448ceSStefano Zampini   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
37739d0448ceSStefano Zampini   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &((PetscObject)B)->type_name));
37749d0448ceSStefano Zampini   PetscCall(MatSetUp(B));
37759d0448ceSStefano Zampini 
37769d0448ceSStefano Zampini   PetscCall(PetscNew(&mumps));
37779d0448ceSStefano Zampini 
37789d0448ceSStefano Zampini   B->ops->view    = MatView_MUMPS;
37799d0448ceSStefano Zampini   B->ops->getinfo = MatGetInfo_MUMPS;
37809d0448ceSStefano Zampini 
37819d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
37829d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
37839d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
37849d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
37859d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
37869d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
37879d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
37889d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
37899d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
37909d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
37919d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
37929d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
37939d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
37949d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
37959d0448ceSStefano Zampini 
37969d0448ceSStefano Zampini   if (ftype == MAT_FACTOR_LU) {
37979d0448ceSStefano Zampini     B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS;
37989d0448ceSStefano Zampini     B->factortype            = MAT_FACTOR_LU;
37999d0448ceSStefano Zampini     mumps->sym               = 0;
38009d0448ceSStefano Zampini   } else {
38019d0448ceSStefano Zampini     B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS;
38029d0448ceSStefano Zampini     B->factortype                  = MAT_FACTOR_CHOLESKY;
38039d0448ceSStefano Zampini #if defined(PETSC_USE_COMPLEX)
38049d0448ceSStefano Zampini     mumps->sym = 2;
38059d0448ceSStefano Zampini #else
38069d0448ceSStefano Zampini     if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1;
38079d0448ceSStefano Zampini     else mumps->sym = 2;
38089d0448ceSStefano Zampini #endif
38099d0448ceSStefano Zampini   }
38109d0448ceSStefano Zampini   mumps->ConvertToTriples = MatConvertToTriples_nest_xaij;
38119d0448ceSStefano Zampini   PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[ftype]));
38129d0448ceSStefano Zampini 
38139d0448ceSStefano Zampini   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
38149d0448ceSStefano Zampini   if (size == 1) {
38159d0448ceSStefano Zampini     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
38169d0448ceSStefano Zampini     B->canuseordering = PETSC_TRUE;
38179d0448ceSStefano Zampini   }
38189d0448ceSStefano Zampini 
38199d0448ceSStefano Zampini   /* set solvertype */
38209d0448ceSStefano Zampini   PetscCall(PetscFree(B->solvertype));
38219d0448ceSStefano Zampini   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
38229d0448ceSStefano Zampini   B->ops->destroy = MatDestroy_MUMPS;
38239d0448ceSStefano Zampini   B->data         = (void *)mumps;
38249d0448ceSStefano Zampini 
38259d0448ceSStefano Zampini   *F               = B;
38269d0448ceSStefano Zampini   mumps->id.job    = JOB_NULL;
38279d0448ceSStefano Zampini   mumps->ICNTL_pre = NULL;
38289d0448ceSStefano Zampini   mumps->CNTL_pre  = NULL;
38299d0448ceSStefano Zampini   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
38309d0448ceSStefano Zampini   PetscFunctionReturn(PETSC_SUCCESS);
38319d0448ceSStefano Zampini }
38329d0448ceSStefano Zampini 
3833d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_MUMPS(void)
3834d71ae5a4SJacob Faibussowitsch {
383542c9c57cSBarry Smith   PetscFunctionBegin;
38369566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIAIJ, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
38379566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
38389566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIBAIJ, MAT_FACTOR_LU, MatGetFactor_baij_mumps));
38399566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_baij_mumps));
38409566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPISBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_sbaij_mumps));
38419566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQAIJ, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
38429566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
38439566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQBAIJ, MAT_FACTOR_LU, MatGetFactor_baij_mumps));
38449566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_baij_mumps));
38459566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQSBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_sbaij_mumps));
38469566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQSELL, MAT_FACTOR_LU, MatGetFactor_sell_mumps));
3847c3e1b152SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATDIAGONAL, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
3848c3e1b152SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATDIAGONAL, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
3849*4b9405b2SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQDENSE, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
3850*4b9405b2SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQDENSE, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
3851*4b9405b2SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIDENSE, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
3852*4b9405b2SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIDENSE, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
38539d0448ceSStefano Zampini   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATNEST, MAT_FACTOR_LU, MatGetFactor_nest_mumps));
38549d0448ceSStefano Zampini   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATNEST, MAT_FACTOR_CHOLESKY, MatGetFactor_nest_mumps));
38553ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
385642c9c57cSBarry Smith }
3857