xref: /petsc/src/mat/impls/aij/mpi/mumps/mumps.c (revision 127cd2765bc40a30cdf86a6ae3964053688c352b)
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;
397*127cd276SPierre Jolivet   } else if (mumps->nest_vals) PetscCall(PetscArraycpy(mumps->val, av, aa->nz)); /* MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_seqaij_seqaij(), so one needs to copy the memory */
398*127cd276SPierre Jolivet   else mumps->val = (PetscScalar *)av;                                           /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
3999566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(A, &av));
4003ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
40116ebf90aSShri Abhyankar }
402397b6df1SKris Buschelman 
40366976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqsell_seqaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
404d71ae5a4SJacob Faibussowitsch {
405a6053eceSJunchao Zhang   PetscInt64     nz, i, j, k, r;
4067ee00b23SStefano Zampini   Mat_SeqSELL   *a = (Mat_SeqSELL *)A->data;
407a6053eceSJunchao Zhang   PetscMUMPSInt *row, *col;
4087ee00b23SStefano Zampini 
4097ee00b23SStefano Zampini   PetscFunctionBegin;
4107ee00b23SStefano Zampini   nz = a->sliidx[a->totalslices];
41150c845baSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
4129566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
413a6053eceSJunchao Zhang     for (i = k = 0; i < a->totalslices; i++) {
41448a46eb9SPierre 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++]));
4157ee00b23SStefano Zampini     }
4169566063dSJacob Faibussowitsch     for (i = 0; i < nz; i++) PetscCall(PetscMUMPSIntCast(a->colidx[i] + shift, &col[i]));
417a6053eceSJunchao Zhang     mumps->irn = row;
418a6053eceSJunchao Zhang     mumps->jcn = col;
419a6053eceSJunchao Zhang     mumps->nnz = nz;
42050c845baSStefano Zampini     mumps->val = a->val;
421*127cd276SPierre Jolivet   } else if (mumps->nest_vals) PetscCall(PetscArraycpy(mumps->val, a->val, nz)); /* MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_seqsell_seqaij(), so one needs to copy the memory */
422*127cd276SPierre Jolivet   else mumps->val = a->val;                                                      /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
4233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4247ee00b23SStefano Zampini }
4257ee00b23SStefano Zampini 
42666976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqbaij_seqaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
427d71ae5a4SJacob Faibussowitsch {
42867877ebaSShri Abhyankar   Mat_SeqBAIJ    *aa = (Mat_SeqBAIJ *)A->data;
42933d57670SJed Brown   const PetscInt *ai, *aj, *ajj, bs2 = aa->bs2;
43050c845baSStefano Zampini   PetscInt64      M, nz = bs2 * aa->nz, idx = 0, rnz, i, j, k, m;
431a6053eceSJunchao Zhang   PetscInt        bs;
432a6053eceSJunchao Zhang   PetscMUMPSInt  *row, *col;
43367877ebaSShri Abhyankar 
43467877ebaSShri Abhyankar   PetscFunctionBegin;
43550c845baSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
4369566063dSJacob Faibussowitsch     PetscCall(MatGetBlockSize(A, &bs));
43733d57670SJed Brown     M  = A->rmap->N / bs;
4389371c9d4SSatish Balay     ai = aa->i;
4399371c9d4SSatish Balay     aj = aa->j;
4409566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
44167877ebaSShri Abhyankar     for (i = 0; i < M; i++) {
44267877ebaSShri Abhyankar       ajj = aj + ai[i];
44367877ebaSShri Abhyankar       rnz = ai[i + 1] - ai[i];
44467877ebaSShri Abhyankar       for (k = 0; k < rnz; k++) {
44567877ebaSShri Abhyankar         for (j = 0; j < bs; j++) {
44667877ebaSShri Abhyankar           for (m = 0; m < bs; m++) {
4479566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(i * bs + m + shift, &row[idx]));
4489566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(bs * ajj[k] + j + shift, &col[idx]));
449a6053eceSJunchao Zhang             idx++;
45067877ebaSShri Abhyankar           }
45167877ebaSShri Abhyankar         }
45267877ebaSShri Abhyankar       }
45367877ebaSShri Abhyankar     }
454a6053eceSJunchao Zhang     mumps->irn = row;
455a6053eceSJunchao Zhang     mumps->jcn = col;
456a6053eceSJunchao Zhang     mumps->nnz = nz;
45750c845baSStefano Zampini     mumps->val = aa->a;
458*127cd276SPierre Jolivet   } else if (mumps->nest_vals) PetscCall(PetscArraycpy(mumps->val, aa->a, nz)); /* MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_seqbaij_seqaij(), so one needs to copy the memory */
459*127cd276SPierre Jolivet   else mumps->val = aa->a;                                                      /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
4603ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
46167877ebaSShri Abhyankar }
46267877ebaSShri Abhyankar 
46366976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqsbaij_seqsbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
464d71ae5a4SJacob Faibussowitsch {
46575480915SPierre Jolivet   const PetscInt *ai, *aj, *ajj;
466a6053eceSJunchao Zhang   PetscInt        bs;
467a6053eceSJunchao Zhang   PetscInt64      nz, rnz, i, j, k, m;
468a6053eceSJunchao Zhang   PetscMUMPSInt  *row, *col;
46975480915SPierre Jolivet   PetscScalar    *val;
47016ebf90aSShri Abhyankar   Mat_SeqSBAIJ   *aa  = (Mat_SeqSBAIJ *)A->data;
47175480915SPierre Jolivet   const PetscInt  bs2 = aa->bs2, mbs = aa->mbs;
47238548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
473b94d7dedSBarry Smith   PetscBool isset, hermitian;
47438548759SBarry Smith #endif
47516ebf90aSShri Abhyankar 
47616ebf90aSShri Abhyankar   PetscFunctionBegin;
47738548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
478b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
479b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
48038548759SBarry Smith #endif
4812205254eSKarl Rupp   ai = aa->i;
4822205254eSKarl Rupp   aj = aa->j;
4839566063dSJacob Faibussowitsch   PetscCall(MatGetBlockSize(A, &bs));
48475480915SPierre Jolivet   if (reuse == MAT_INITIAL_MATRIX) {
485f3fa974cSJacob Faibussowitsch     const PetscInt64 alloc_size = aa->nz * bs2;
486f3fa974cSJacob Faibussowitsch 
487f3fa974cSJacob Faibussowitsch     PetscCall(PetscMalloc2(alloc_size, &row, alloc_size, &col));
488a6053eceSJunchao Zhang     if (bs > 1) {
489f3fa974cSJacob Faibussowitsch       PetscCall(PetscMalloc1(alloc_size, &mumps->val_alloc));
490a6053eceSJunchao Zhang       mumps->val = mumps->val_alloc;
49175480915SPierre Jolivet     } else {
492a6053eceSJunchao Zhang       mumps->val = aa->a;
49375480915SPierre Jolivet     }
494a6053eceSJunchao Zhang     mumps->irn = row;
495a6053eceSJunchao Zhang     mumps->jcn = col;
496a6053eceSJunchao Zhang   } else {
497a6053eceSJunchao Zhang     row = mumps->irn;
498a6053eceSJunchao Zhang     col = mumps->jcn;
499a6053eceSJunchao Zhang   }
500a6053eceSJunchao Zhang   val = mumps->val;
501185f6596SHong Zhang 
50216ebf90aSShri Abhyankar   nz = 0;
503a81fe166SPierre Jolivet   if (bs > 1) {
50475480915SPierre Jolivet     for (i = 0; i < mbs; i++) {
50516ebf90aSShri Abhyankar       rnz = ai[i + 1] - ai[i];
50667877ebaSShri Abhyankar       ajj = aj + ai[i];
50775480915SPierre Jolivet       for (j = 0; j < rnz; j++) {
50875480915SPierre Jolivet         for (k = 0; k < bs; k++) {
50975480915SPierre Jolivet           for (m = 0; m < bs; m++) {
510ec4f40fdSPierre Jolivet             if (ajj[j] > i || k >= m) {
51175480915SPierre Jolivet               if (reuse == MAT_INITIAL_MATRIX) {
5129566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(i * bs + m + shift, &row[nz]));
5139566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(ajj[j] * bs + k + shift, &col[nz]));
51475480915SPierre Jolivet               }
51575480915SPierre Jolivet               val[nz++] = aa->a[(ai[i] + j) * bs2 + m + k * bs];
51675480915SPierre Jolivet             }
51775480915SPierre Jolivet           }
51875480915SPierre Jolivet         }
51975480915SPierre Jolivet       }
52075480915SPierre Jolivet     }
521a81fe166SPierre Jolivet   } else if (reuse == MAT_INITIAL_MATRIX) {
522a81fe166SPierre Jolivet     for (i = 0; i < mbs; i++) {
523a81fe166SPierre Jolivet       rnz = ai[i + 1] - ai[i];
524a81fe166SPierre Jolivet       ajj = aj + ai[i];
525a81fe166SPierre Jolivet       for (j = 0; j < rnz; j++) {
5269566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
5279566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[nz]));
528a6053eceSJunchao Zhang         nz++;
529a81fe166SPierre Jolivet       }
530a81fe166SPierre Jolivet     }
53108401ef6SPierre Jolivet     PetscCheck(nz == aa->nz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Different numbers of nonzeros %" PetscInt64_FMT " != %" PetscInt_FMT, nz, aa->nz);
532*127cd276SPierre Jolivet   } else if (mumps->nest_vals)
533*127cd276SPierre Jolivet     PetscCall(PetscArraycpy(mumps->val, aa->a, aa->nz)); /* bs == 1 and MAT_REUSE_MATRIX, MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_seqsbaij_seqsbaij(), so one needs to copy the memory */
534*127cd276SPierre Jolivet   else mumps->val = aa->a;                               /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
535a6053eceSJunchao Zhang   if (reuse == MAT_INITIAL_MATRIX) mumps->nnz = nz;
5363ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
53716ebf90aSShri Abhyankar }
53816ebf90aSShri Abhyankar 
53966976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_seqaij_seqsbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
540d71ae5a4SJacob Faibussowitsch {
54167877ebaSShri Abhyankar   const PetscInt    *ai, *aj, *ajj, *adiag, M = A->rmap->n;
542a6053eceSJunchao Zhang   PetscInt64         nz, rnz, i, j;
54367877ebaSShri Abhyankar   const PetscScalar *av, *v1;
54416ebf90aSShri Abhyankar   PetscScalar       *val;
545a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
546829b1710SHong Zhang   Mat_SeqAIJ        *aa = (Mat_SeqAIJ *)A->data;
54729b521d4Sstefano_zampini   PetscBool          missing;
54838548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
549b94d7dedSBarry Smith   PetscBool hermitian, isset;
55038548759SBarry Smith #endif
55116ebf90aSShri Abhyankar 
55216ebf90aSShri Abhyankar   PetscFunctionBegin;
55338548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
554b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
555b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
55638548759SBarry Smith #endif
5579566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(A, &av));
5589371c9d4SSatish Balay   ai    = aa->i;
5599371c9d4SSatish Balay   aj    = aa->j;
56016ebf90aSShri Abhyankar   adiag = aa->diag;
5619566063dSJacob Faibussowitsch   PetscCall(MatMissingDiagonal_SeqAIJ(A, &missing, NULL));
562bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
5637ee00b23SStefano Zampini     /* count nz in the upper triangular part of A */
564829b1710SHong Zhang     nz = 0;
56529b521d4Sstefano_zampini     if (missing) {
56629b521d4Sstefano_zampini       for (i = 0; i < M; i++) {
56729b521d4Sstefano_zampini         if (PetscUnlikely(adiag[i] >= ai[i + 1])) {
56829b521d4Sstefano_zampini           for (j = ai[i]; j < ai[i + 1]; j++) {
56929b521d4Sstefano_zampini             if (aj[j] < i) continue;
57029b521d4Sstefano_zampini             nz++;
57129b521d4Sstefano_zampini           }
57229b521d4Sstefano_zampini         } else {
57329b521d4Sstefano_zampini           nz += ai[i + 1] - adiag[i];
57429b521d4Sstefano_zampini         }
57529b521d4Sstefano_zampini       }
57629b521d4Sstefano_zampini     } else {
577829b1710SHong Zhang       for (i = 0; i < M; i++) nz += ai[i + 1] - adiag[i];
57829b521d4Sstefano_zampini     }
5799566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
5809566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
581a6053eceSJunchao Zhang     mumps->nnz = nz;
582a6053eceSJunchao Zhang     mumps->irn = row;
583a6053eceSJunchao Zhang     mumps->jcn = col;
584a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
585185f6596SHong Zhang 
58616ebf90aSShri Abhyankar     nz = 0;
58729b521d4Sstefano_zampini     if (missing) {
58829b521d4Sstefano_zampini       for (i = 0; i < M; i++) {
58929b521d4Sstefano_zampini         if (PetscUnlikely(adiag[i] >= ai[i + 1])) {
59029b521d4Sstefano_zampini           for (j = ai[i]; j < ai[i + 1]; j++) {
59129b521d4Sstefano_zampini             if (aj[j] < i) continue;
5929566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
5939566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(aj[j] + shift, &col[nz]));
59429b521d4Sstefano_zampini             val[nz] = av[j];
59529b521d4Sstefano_zampini             nz++;
59629b521d4Sstefano_zampini           }
59729b521d4Sstefano_zampini         } else {
59829b521d4Sstefano_zampini           rnz = ai[i + 1] - adiag[i];
59929b521d4Sstefano_zampini           ajj = aj + adiag[i];
60029b521d4Sstefano_zampini           v1  = av + adiag[i];
60129b521d4Sstefano_zampini           for (j = 0; j < rnz; j++) {
6029566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
6039566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[nz]));
604a6053eceSJunchao Zhang             val[nz++] = v1[j];
60529b521d4Sstefano_zampini           }
60629b521d4Sstefano_zampini         }
60729b521d4Sstefano_zampini       }
60829b521d4Sstefano_zampini     } else {
60916ebf90aSShri Abhyankar       for (i = 0; i < M; i++) {
61016ebf90aSShri Abhyankar         rnz = ai[i + 1] - adiag[i];
61167877ebaSShri Abhyankar         ajj = aj + adiag[i];
612cf3759fdSShri Abhyankar         v1  = av + adiag[i];
61367877ebaSShri Abhyankar         for (j = 0; j < rnz; j++) {
6149566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(i + shift, &row[nz]));
6159566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(ajj[j] + shift, &col[nz]));
616a6053eceSJunchao Zhang           val[nz++] = v1[j];
61716ebf90aSShri Abhyankar         }
61816ebf90aSShri Abhyankar       }
61929b521d4Sstefano_zampini     }
620397b6df1SKris Buschelman   } else {
621a6053eceSJunchao Zhang     nz  = 0;
622a6053eceSJunchao Zhang     val = mumps->val;
62329b521d4Sstefano_zampini     if (missing) {
62416ebf90aSShri Abhyankar       for (i = 0; i < M; i++) {
62529b521d4Sstefano_zampini         if (PetscUnlikely(adiag[i] >= ai[i + 1])) {
62629b521d4Sstefano_zampini           for (j = ai[i]; j < ai[i + 1]; j++) {
62729b521d4Sstefano_zampini             if (aj[j] < i) continue;
62829b521d4Sstefano_zampini             val[nz++] = av[j];
62929b521d4Sstefano_zampini           }
63029b521d4Sstefano_zampini         } else {
63116ebf90aSShri Abhyankar           rnz = ai[i + 1] - adiag[i];
63267877ebaSShri Abhyankar           v1  = av + adiag[i];
633ad540459SPierre Jolivet           for (j = 0; j < rnz; j++) val[nz++] = v1[j];
63416ebf90aSShri Abhyankar         }
63516ebf90aSShri Abhyankar       }
63629b521d4Sstefano_zampini     } else {
63716ebf90aSShri Abhyankar       for (i = 0; i < M; i++) {
63816ebf90aSShri Abhyankar         rnz = ai[i + 1] - adiag[i];
63916ebf90aSShri Abhyankar         v1  = av + adiag[i];
640ad540459SPierre Jolivet         for (j = 0; j < rnz; j++) val[nz++] = v1[j];
64116ebf90aSShri Abhyankar       }
64216ebf90aSShri Abhyankar     }
64329b521d4Sstefano_zampini   }
6449566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(A, &av));
6453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
64616ebf90aSShri Abhyankar }
64716ebf90aSShri Abhyankar 
64866976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpisbaij_mpisbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
649d71ae5a4SJacob Faibussowitsch {
650a6053eceSJunchao Zhang   const PetscInt    *ai, *aj, *bi, *bj, *garray, *ajj, *bjj;
651a6053eceSJunchao Zhang   PetscInt           bs;
652a6053eceSJunchao Zhang   PetscInt64         rstart, nz, i, j, k, m, jj, irow, countA, countB;
653a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
65416ebf90aSShri Abhyankar   const PetscScalar *av, *bv, *v1, *v2;
65516ebf90aSShri Abhyankar   PetscScalar       *val;
656397b6df1SKris Buschelman   Mat_MPISBAIJ      *mat = (Mat_MPISBAIJ *)A->data;
657397b6df1SKris Buschelman   Mat_SeqSBAIJ      *aa  = (Mat_SeqSBAIJ *)(mat->A)->data;
658397b6df1SKris Buschelman   Mat_SeqBAIJ       *bb  = (Mat_SeqBAIJ *)(mat->B)->data;
659ec4f40fdSPierre Jolivet   const PetscInt     bs2 = aa->bs2, mbs = aa->mbs;
66038548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
661b94d7dedSBarry Smith   PetscBool hermitian, isset;
66238548759SBarry Smith #endif
66316ebf90aSShri Abhyankar 
66416ebf90aSShri Abhyankar   PetscFunctionBegin;
66538548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
666b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
667b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
66838548759SBarry Smith #endif
6699566063dSJacob Faibussowitsch   PetscCall(MatGetBlockSize(A, &bs));
67038548759SBarry Smith   rstart = A->rmap->rstart;
67138548759SBarry Smith   ai     = aa->i;
67238548759SBarry Smith   aj     = aa->j;
67338548759SBarry Smith   bi     = bb->i;
67438548759SBarry Smith   bj     = bb->j;
67538548759SBarry Smith   av     = aa->a;
67638548759SBarry Smith   bv     = bb->a;
677397b6df1SKris Buschelman 
6782205254eSKarl Rupp   garray = mat->garray;
6792205254eSKarl Rupp 
680bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
681a6053eceSJunchao Zhang     nz = (aa->nz + bb->nz) * bs2; /* just a conservative estimate */
6829566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
6839566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
684a6053eceSJunchao Zhang     /* can not decide the exact mumps->nnz now because of the SBAIJ */
685a6053eceSJunchao Zhang     mumps->irn = row;
686a6053eceSJunchao Zhang     mumps->jcn = col;
687a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
688397b6df1SKris Buschelman   } else {
689a6053eceSJunchao Zhang     val = mumps->val;
690397b6df1SKris Buschelman   }
691397b6df1SKris Buschelman 
6929371c9d4SSatish Balay   jj   = 0;
6939371c9d4SSatish Balay   irow = rstart;
694ec4f40fdSPierre Jolivet   for (i = 0; i < mbs; i++) {
695397b6df1SKris Buschelman     ajj    = aj + ai[i]; /* ptr to the beginning of this row */
696397b6df1SKris Buschelman     countA = ai[i + 1] - ai[i];
697397b6df1SKris Buschelman     countB = bi[i + 1] - bi[i];
698397b6df1SKris Buschelman     bjj    = bj + bi[i];
699ec4f40fdSPierre Jolivet     v1     = av + ai[i] * bs2;
700ec4f40fdSPierre Jolivet     v2     = bv + bi[i] * bs2;
701397b6df1SKris Buschelman 
702ec4f40fdSPierre Jolivet     if (bs > 1) {
703ec4f40fdSPierre Jolivet       /* A-part */
704ec4f40fdSPierre Jolivet       for (j = 0; j < countA; j++) {
705ec4f40fdSPierre Jolivet         for (k = 0; k < bs; k++) {
706ec4f40fdSPierre Jolivet           for (m = 0; m < bs; m++) {
707ec4f40fdSPierre Jolivet             if (rstart + ajj[j] * bs > irow || k >= m) {
708ec4f40fdSPierre Jolivet               if (reuse == MAT_INITIAL_MATRIX) {
7099566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(irow + m + shift, &row[jj]));
7109566063dSJacob Faibussowitsch                 PetscCall(PetscMUMPSIntCast(rstart + ajj[j] * bs + k + shift, &col[jj]));
711ec4f40fdSPierre Jolivet               }
712ec4f40fdSPierre Jolivet               val[jj++] = v1[j * bs2 + m + k * bs];
713ec4f40fdSPierre Jolivet             }
714ec4f40fdSPierre Jolivet           }
715ec4f40fdSPierre Jolivet         }
716ec4f40fdSPierre Jolivet       }
717ec4f40fdSPierre Jolivet 
718ec4f40fdSPierre Jolivet       /* B-part */
719ec4f40fdSPierre Jolivet       for (j = 0; j < countB; j++) {
720ec4f40fdSPierre Jolivet         for (k = 0; k < bs; k++) {
721ec4f40fdSPierre Jolivet           for (m = 0; m < bs; m++) {
722ec4f40fdSPierre Jolivet             if (reuse == MAT_INITIAL_MATRIX) {
7239566063dSJacob Faibussowitsch               PetscCall(PetscMUMPSIntCast(irow + m + shift, &row[jj]));
7249566063dSJacob Faibussowitsch               PetscCall(PetscMUMPSIntCast(garray[bjj[j]] * bs + k + shift, &col[jj]));
725ec4f40fdSPierre Jolivet             }
726ec4f40fdSPierre Jolivet             val[jj++] = v2[j * bs2 + m + k * bs];
727ec4f40fdSPierre Jolivet           }
728ec4f40fdSPierre Jolivet         }
729ec4f40fdSPierre Jolivet       }
730ec4f40fdSPierre Jolivet     } else {
731397b6df1SKris Buschelman       /* A-part */
732397b6df1SKris Buschelman       for (j = 0; j < countA; j++) {
733bccb9932SShri Abhyankar         if (reuse == MAT_INITIAL_MATRIX) {
7349566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
7359566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(rstart + ajj[j] + shift, &col[jj]));
736397b6df1SKris Buschelman         }
73716ebf90aSShri Abhyankar         val[jj++] = v1[j];
738397b6df1SKris Buschelman       }
73916ebf90aSShri Abhyankar 
74016ebf90aSShri Abhyankar       /* B-part */
74116ebf90aSShri Abhyankar       for (j = 0; j < countB; j++) {
742bccb9932SShri Abhyankar         if (reuse == MAT_INITIAL_MATRIX) {
7439566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
7449566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(garray[bjj[j]] + shift, &col[jj]));
745397b6df1SKris Buschelman         }
74616ebf90aSShri Abhyankar         val[jj++] = v2[j];
74716ebf90aSShri Abhyankar       }
74816ebf90aSShri Abhyankar     }
749ec4f40fdSPierre Jolivet     irow += bs;
750ec4f40fdSPierre Jolivet   }
7515d955bbbSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) mumps->nnz = jj;
7523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
75316ebf90aSShri Abhyankar }
75416ebf90aSShri Abhyankar 
75566976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpiaij_mpiaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
756d71ae5a4SJacob Faibussowitsch {
75716ebf90aSShri Abhyankar   const PetscInt    *ai, *aj, *bi, *bj, *garray, m = A->rmap->n, *ajj, *bjj;
7585d955bbbSStefano Zampini   PetscInt64         rstart, cstart, nz, i, j, jj, irow, countA, countB;
759a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
76016ebf90aSShri Abhyankar   const PetscScalar *av, *bv, *v1, *v2;
76116ebf90aSShri Abhyankar   PetscScalar       *val;
762a3d589ffSStefano Zampini   Mat                Ad, Ao;
763a3d589ffSStefano Zampini   Mat_SeqAIJ        *aa;
764a3d589ffSStefano Zampini   Mat_SeqAIJ        *bb;
76516ebf90aSShri Abhyankar 
76616ebf90aSShri Abhyankar   PetscFunctionBegin;
7679566063dSJacob Faibussowitsch   PetscCall(MatMPIAIJGetSeqAIJ(A, &Ad, &Ao, &garray));
7689566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ad, &av));
7699566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ao, &bv));
770a3d589ffSStefano Zampini 
771a3d589ffSStefano Zampini   aa = (Mat_SeqAIJ *)(Ad)->data;
772a3d589ffSStefano Zampini   bb = (Mat_SeqAIJ *)(Ao)->data;
77338548759SBarry Smith   ai = aa->i;
77438548759SBarry Smith   aj = aa->j;
77538548759SBarry Smith   bi = bb->i;
77638548759SBarry Smith   bj = bb->j;
77716ebf90aSShri Abhyankar 
778a3d589ffSStefano Zampini   rstart = A->rmap->rstart;
7795d955bbbSStefano Zampini   cstart = A->cmap->rstart;
7802205254eSKarl Rupp 
781bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
782a6053eceSJunchao Zhang     nz = (PetscInt64)aa->nz + bb->nz; /* make sure the sum won't overflow PetscInt */
7839566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
7849566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
785a6053eceSJunchao Zhang     mumps->nnz = nz;
786a6053eceSJunchao Zhang     mumps->irn = row;
787a6053eceSJunchao Zhang     mumps->jcn = col;
788a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
78916ebf90aSShri Abhyankar   } else {
790a6053eceSJunchao Zhang     val = mumps->val;
79116ebf90aSShri Abhyankar   }
79216ebf90aSShri Abhyankar 
7939371c9d4SSatish Balay   jj   = 0;
7949371c9d4SSatish Balay   irow = rstart;
79516ebf90aSShri Abhyankar   for (i = 0; i < m; i++) {
79616ebf90aSShri Abhyankar     ajj    = aj + ai[i]; /* ptr to the beginning of this row */
79716ebf90aSShri Abhyankar     countA = ai[i + 1] - ai[i];
79816ebf90aSShri Abhyankar     countB = bi[i + 1] - bi[i];
79916ebf90aSShri Abhyankar     bjj    = bj + bi[i];
80016ebf90aSShri Abhyankar     v1     = av + ai[i];
80116ebf90aSShri Abhyankar     v2     = bv + bi[i];
80216ebf90aSShri Abhyankar 
80316ebf90aSShri Abhyankar     /* A-part */
80416ebf90aSShri Abhyankar     for (j = 0; j < countA; j++) {
805bccb9932SShri Abhyankar       if (reuse == MAT_INITIAL_MATRIX) {
8069566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
8075d955bbbSStefano Zampini         PetscCall(PetscMUMPSIntCast(cstart + ajj[j] + shift, &col[jj]));
80816ebf90aSShri Abhyankar       }
80916ebf90aSShri Abhyankar       val[jj++] = v1[j];
81016ebf90aSShri Abhyankar     }
81116ebf90aSShri Abhyankar 
81216ebf90aSShri Abhyankar     /* B-part */
81316ebf90aSShri Abhyankar     for (j = 0; j < countB; j++) {
814bccb9932SShri Abhyankar       if (reuse == MAT_INITIAL_MATRIX) {
8159566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
8169566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(garray[bjj[j]] + shift, &col[jj]));
81716ebf90aSShri Abhyankar       }
81816ebf90aSShri Abhyankar       val[jj++] = v2[j];
81916ebf90aSShri Abhyankar     }
82016ebf90aSShri Abhyankar     irow++;
82116ebf90aSShri Abhyankar   }
8229566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ad, &av));
8239566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ao, &bv));
8243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
82516ebf90aSShri Abhyankar }
82616ebf90aSShri Abhyankar 
82766976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpibaij_mpiaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
828d71ae5a4SJacob Faibussowitsch {
82967877ebaSShri Abhyankar   Mat_MPIBAIJ       *mat = (Mat_MPIBAIJ *)A->data;
83067877ebaSShri Abhyankar   Mat_SeqBAIJ       *aa  = (Mat_SeqBAIJ *)(mat->A)->data;
83167877ebaSShri Abhyankar   Mat_SeqBAIJ       *bb  = (Mat_SeqBAIJ *)(mat->B)->data;
83267877ebaSShri Abhyankar   const PetscInt    *ai = aa->i, *bi = bb->i, *aj = aa->j, *bj = bb->j, *ajj, *bjj;
8335d955bbbSStefano Zampini   const PetscInt    *garray = mat->garray, mbs = mat->mbs, rstart = A->rmap->rstart, cstart = A->cmap->rstart;
83433d57670SJed Brown   const PetscInt     bs2 = mat->bs2;
835a6053eceSJunchao Zhang   PetscInt           bs;
836a6053eceSJunchao Zhang   PetscInt64         nz, i, j, k, n, jj, irow, countA, countB, idx;
837a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
83867877ebaSShri Abhyankar   const PetscScalar *av = aa->a, *bv = bb->a, *v1, *v2;
83967877ebaSShri Abhyankar   PetscScalar       *val;
84067877ebaSShri Abhyankar 
84167877ebaSShri Abhyankar   PetscFunctionBegin;
8429566063dSJacob Faibussowitsch   PetscCall(MatGetBlockSize(A, &bs));
843bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
84467877ebaSShri Abhyankar     nz = bs2 * (aa->nz + bb->nz);
8459566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
8469566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
847a6053eceSJunchao Zhang     mumps->nnz = nz;
848a6053eceSJunchao Zhang     mumps->irn = row;
849a6053eceSJunchao Zhang     mumps->jcn = col;
850a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
85167877ebaSShri Abhyankar   } else {
852a6053eceSJunchao Zhang     val = mumps->val;
85367877ebaSShri Abhyankar   }
85467877ebaSShri Abhyankar 
8559371c9d4SSatish Balay   jj   = 0;
8569371c9d4SSatish Balay   irow = rstart;
85767877ebaSShri Abhyankar   for (i = 0; i < mbs; i++) {
85867877ebaSShri Abhyankar     countA = ai[i + 1] - ai[i];
85967877ebaSShri Abhyankar     countB = bi[i + 1] - bi[i];
86067877ebaSShri Abhyankar     ajj    = aj + ai[i];
86167877ebaSShri Abhyankar     bjj    = bj + bi[i];
86267877ebaSShri Abhyankar     v1     = av + bs2 * ai[i];
86367877ebaSShri Abhyankar     v2     = bv + bs2 * bi[i];
86467877ebaSShri Abhyankar 
86567877ebaSShri Abhyankar     idx = 0;
86667877ebaSShri Abhyankar     /* A-part */
86767877ebaSShri Abhyankar     for (k = 0; k < countA; k++) {
86867877ebaSShri Abhyankar       for (j = 0; j < bs; j++) {
86967877ebaSShri Abhyankar         for (n = 0; n < bs; n++) {
870bccb9932SShri Abhyankar           if (reuse == MAT_INITIAL_MATRIX) {
8719566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(irow + n + shift, &row[jj]));
8725d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(cstart + bs * ajj[k] + j + shift, &col[jj]));
87367877ebaSShri Abhyankar           }
87467877ebaSShri Abhyankar           val[jj++] = v1[idx++];
87567877ebaSShri Abhyankar         }
87667877ebaSShri Abhyankar       }
87767877ebaSShri Abhyankar     }
87867877ebaSShri Abhyankar 
87967877ebaSShri Abhyankar     idx = 0;
88067877ebaSShri Abhyankar     /* B-part */
88167877ebaSShri Abhyankar     for (k = 0; k < countB; k++) {
88267877ebaSShri Abhyankar       for (j = 0; j < bs; j++) {
88367877ebaSShri Abhyankar         for (n = 0; n < bs; n++) {
884bccb9932SShri Abhyankar           if (reuse == MAT_INITIAL_MATRIX) {
8859566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(irow + n + shift, &row[jj]));
8869566063dSJacob Faibussowitsch             PetscCall(PetscMUMPSIntCast(bs * garray[bjj[k]] + j + shift, &col[jj]));
88767877ebaSShri Abhyankar           }
888d985c460SShri Abhyankar           val[jj++] = v2[idx++];
88967877ebaSShri Abhyankar         }
89067877ebaSShri Abhyankar       }
89167877ebaSShri Abhyankar     }
892d985c460SShri Abhyankar     irow += bs;
89367877ebaSShri Abhyankar   }
8943ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
89567877ebaSShri Abhyankar }
89667877ebaSShri Abhyankar 
89766976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_mpiaij_mpisbaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
898d71ae5a4SJacob Faibussowitsch {
89916ebf90aSShri Abhyankar   const PetscInt    *ai, *aj, *adiag, *bi, *bj, *garray, m = A->rmap->n, *ajj, *bjj;
900a6053eceSJunchao Zhang   PetscInt64         rstart, nz, nza, nzb, i, j, jj, irow, countA, countB;
901a6053eceSJunchao Zhang   PetscMUMPSInt     *row, *col;
90216ebf90aSShri Abhyankar   const PetscScalar *av, *bv, *v1, *v2;
90316ebf90aSShri Abhyankar   PetscScalar       *val;
904a3d589ffSStefano Zampini   Mat                Ad, Ao;
905a3d589ffSStefano Zampini   Mat_SeqAIJ        *aa;
906a3d589ffSStefano Zampini   Mat_SeqAIJ        *bb;
90738548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
908b94d7dedSBarry Smith   PetscBool hermitian, isset;
90938548759SBarry Smith #endif
91016ebf90aSShri Abhyankar 
91116ebf90aSShri Abhyankar   PetscFunctionBegin;
91238548759SBarry Smith #if defined(PETSC_USE_COMPLEX)
913b94d7dedSBarry Smith   PetscCall(MatIsHermitianKnown(A, &isset, &hermitian));
914b94d7dedSBarry Smith   PetscCheck(!isset || !hermitian, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MUMPS does not support Hermitian symmetric matrices for Choleksy");
91538548759SBarry Smith #endif
9169566063dSJacob Faibussowitsch   PetscCall(MatMPIAIJGetSeqAIJ(A, &Ad, &Ao, &garray));
9179566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ad, &av));
9189566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJGetArrayRead(Ao, &bv));
919a3d589ffSStefano Zampini 
920a3d589ffSStefano Zampini   aa    = (Mat_SeqAIJ *)(Ad)->data;
921a3d589ffSStefano Zampini   bb    = (Mat_SeqAIJ *)(Ao)->data;
92238548759SBarry Smith   ai    = aa->i;
92338548759SBarry Smith   aj    = aa->j;
92438548759SBarry Smith   adiag = aa->diag;
92538548759SBarry Smith   bi    = bb->i;
92638548759SBarry Smith   bj    = bb->j;
9272205254eSKarl Rupp 
92816ebf90aSShri Abhyankar   rstart = A->rmap->rstart;
92916ebf90aSShri Abhyankar 
930bccb9932SShri Abhyankar   if (reuse == MAT_INITIAL_MATRIX) {
931e0bace9bSHong Zhang     nza = 0; /* num of upper triangular entries in mat->A, including diagonals */
932e0bace9bSHong Zhang     nzb = 0; /* num of upper triangular entries in mat->B */
93316ebf90aSShri Abhyankar     for (i = 0; i < m; i++) {
934e0bace9bSHong Zhang       nza += (ai[i + 1] - adiag[i]);
93516ebf90aSShri Abhyankar       countB = bi[i + 1] - bi[i];
93616ebf90aSShri Abhyankar       bjj    = bj + bi[i];
937e0bace9bSHong Zhang       for (j = 0; j < countB; j++) {
938e0bace9bSHong Zhang         if (garray[bjj[j]] > rstart) nzb++;
939e0bace9bSHong Zhang       }
940e0bace9bSHong Zhang     }
94116ebf90aSShri Abhyankar 
942e0bace9bSHong Zhang     nz = nza + nzb; /* total nz of upper triangular part of mat */
9439566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nz, &row, nz, &col));
9449566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nz, &val));
945a6053eceSJunchao Zhang     mumps->nnz = nz;
946a6053eceSJunchao Zhang     mumps->irn = row;
947a6053eceSJunchao Zhang     mumps->jcn = col;
948a6053eceSJunchao Zhang     mumps->val = mumps->val_alloc = val;
94916ebf90aSShri Abhyankar   } else {
950a6053eceSJunchao Zhang     val = mumps->val;
95116ebf90aSShri Abhyankar   }
95216ebf90aSShri Abhyankar 
9539371c9d4SSatish Balay   jj   = 0;
9549371c9d4SSatish Balay   irow = rstart;
95516ebf90aSShri Abhyankar   for (i = 0; i < m; i++) {
95616ebf90aSShri Abhyankar     ajj    = aj + adiag[i]; /* ptr to the beginning of the diagonal of this row */
95716ebf90aSShri Abhyankar     v1     = av + adiag[i];
95816ebf90aSShri Abhyankar     countA = ai[i + 1] - adiag[i];
95916ebf90aSShri Abhyankar     countB = bi[i + 1] - bi[i];
96016ebf90aSShri Abhyankar     bjj    = bj + bi[i];
96116ebf90aSShri Abhyankar     v2     = bv + bi[i];
96216ebf90aSShri Abhyankar 
96316ebf90aSShri Abhyankar     /* A-part */
96416ebf90aSShri Abhyankar     for (j = 0; j < countA; j++) {
965bccb9932SShri Abhyankar       if (reuse == MAT_INITIAL_MATRIX) {
9669566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
9679566063dSJacob Faibussowitsch         PetscCall(PetscMUMPSIntCast(rstart + ajj[j] + shift, &col[jj]));
96816ebf90aSShri Abhyankar       }
96916ebf90aSShri Abhyankar       val[jj++] = v1[j];
97016ebf90aSShri Abhyankar     }
97116ebf90aSShri Abhyankar 
97216ebf90aSShri Abhyankar     /* B-part */
97316ebf90aSShri Abhyankar     for (j = 0; j < countB; j++) {
97416ebf90aSShri Abhyankar       if (garray[bjj[j]] > rstart) {
975bccb9932SShri Abhyankar         if (reuse == MAT_INITIAL_MATRIX) {
9769566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(irow + shift, &row[jj]));
9779566063dSJacob Faibussowitsch           PetscCall(PetscMUMPSIntCast(garray[bjj[j]] + shift, &col[jj]));
97816ebf90aSShri Abhyankar         }
97916ebf90aSShri Abhyankar         val[jj++] = v2[j];
98016ebf90aSShri Abhyankar       }
981397b6df1SKris Buschelman     }
982397b6df1SKris Buschelman     irow++;
983397b6df1SKris Buschelman   }
9849566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ad, &av));
9859566063dSJacob Faibussowitsch   PetscCall(MatSeqAIJRestoreArrayRead(Ao, &bv));
9863ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
987397b6df1SKris Buschelman }
988397b6df1SKris Buschelman 
989c3e1b152SPierre Jolivet static PetscErrorCode MatConvertToTriples_diagonal_xaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
990c3e1b152SPierre Jolivet {
991c3e1b152SPierre Jolivet   const PetscScalar *av;
992c3e1b152SPierre Jolivet   const PetscInt     M = A->rmap->n;
993c3e1b152SPierre Jolivet   PetscInt64         i;
994c3e1b152SPierre Jolivet   PetscMUMPSInt     *row, *col;
995c3e1b152SPierre Jolivet   Vec                v;
996c3e1b152SPierre Jolivet 
997c3e1b152SPierre Jolivet   PetscFunctionBegin;
998c3e1b152SPierre Jolivet   PetscCall(MatDiagonalGetDiagonal(A, &v));
999c3e1b152SPierre Jolivet   PetscCall(VecGetArrayRead(v, &av));
1000c3e1b152SPierre Jolivet   if (reuse == MAT_INITIAL_MATRIX) {
1001c3e1b152SPierre Jolivet     PetscCall(PetscMalloc2(M, &row, M, &col));
1002c3e1b152SPierre Jolivet     for (i = 0; i < M; i++) {
1003c3e1b152SPierre Jolivet       PetscCall(PetscMUMPSIntCast(i + A->rmap->rstart, &row[i]));
1004c3e1b152SPierre Jolivet       col[i] = row[i];
1005c3e1b152SPierre Jolivet     }
1006c3e1b152SPierre Jolivet     mumps->val = (PetscScalar *)av;
1007c3e1b152SPierre Jolivet     mumps->irn = row;
1008c3e1b152SPierre Jolivet     mumps->jcn = col;
1009c3e1b152SPierre Jolivet     mumps->nnz = M;
1010*127cd276SPierre Jolivet   } else if (mumps->nest_vals) PetscCall(PetscArraycpy(mumps->val, av, M)); /* MatConvertToTriples_nest_xaij() allocates mumps->val outside of MatConvertToTriples_diagonal_xaij(), so one needs to copy the memory */
1011*127cd276SPierre Jolivet   else mumps->val = (PetscScalar *)av;                                      /* in the default case, mumps->val is never allocated, one just needs to update the mumps->val pointer */
1012c3e1b152SPierre Jolivet   PetscCall(VecRestoreArrayRead(v, &av));
1013c3e1b152SPierre Jolivet   PetscFunctionReturn(PETSC_SUCCESS);
1014c3e1b152SPierre Jolivet }
1015c3e1b152SPierre Jolivet 
101666976f2fSJacob Faibussowitsch static PetscErrorCode MatConvertToTriples_nest_xaij(Mat A, PetscInt shift, MatReuse reuse, Mat_MUMPS *mumps)
10179d0448ceSStefano Zampini {
10189d0448ceSStefano Zampini   Mat     **mats;
10199d0448ceSStefano Zampini   PetscInt  nr, nc;
10209d0448ceSStefano Zampini   PetscBool chol = mumps->sym ? PETSC_TRUE : PETSC_FALSE;
10219d0448ceSStefano Zampini 
10229d0448ceSStefano Zampini   PetscFunctionBegin;
10239d0448ceSStefano Zampini   PetscCall(MatNestGetSubMats(A, &nr, &nc, &mats));
10249d0448ceSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
10259d0448ceSStefano Zampini     PetscMUMPSInt *irns, *jcns;
10269d0448ceSStefano Zampini     PetscScalar   *vals;
10279d0448ceSStefano Zampini     PetscInt64     totnnz, cumnnz, maxnnz;
10289d0448ceSStefano Zampini     PetscInt      *pjcns_w;
10299d0448ceSStefano Zampini     IS            *rows, *cols;
10309d0448ceSStefano Zampini     PetscInt     **rows_idx, **cols_idx;
10319d0448ceSStefano Zampini 
10329d0448ceSStefano Zampini     cumnnz = 0;
10339d0448ceSStefano Zampini     maxnnz = 0;
10345d955bbbSStefano Zampini     PetscCall(PetscMalloc2(nr * nc + 1, &mumps->nest_vals_start, nr * nc, &mumps->nest_convert_to_triples));
10359d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) {
10369d0448ceSStefano Zampini       for (PetscInt c = 0; c < nc; c++) {
10379d0448ceSStefano Zampini         Mat sub = mats[r][c];
10389d0448ceSStefano Zampini 
10399d0448ceSStefano Zampini         mumps->nest_convert_to_triples[r * nc + c] = NULL;
10409d0448ceSStefano Zampini         if (chol && c < r) continue; /* skip lower-triangular block for Cholesky */
10419d0448ceSStefano Zampini         if (sub) {
10429d0448ceSStefano Zampini           PetscErrorCode (*convert_to_triples)(Mat, PetscInt, MatReuse, Mat_MUMPS *) = NULL;
1043c3e1b152SPierre Jolivet           PetscBool isSeqAIJ, isMPIAIJ, isSeqBAIJ, isMPIBAIJ, isSeqSBAIJ, isMPISBAIJ, isTrans, isHTrans = PETSC_FALSE, isDiag;
10449d0448ceSStefano Zampini           MatInfo   info;
10459d0448ceSStefano Zampini 
10465d955bbbSStefano Zampini           PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATTRANSPOSEVIRTUAL, &isTrans));
10475d955bbbSStefano Zampini           if (isTrans) PetscCall(MatTransposeGetMat(sub, &sub));
10485d955bbbSStefano Zampini           else {
10495d955bbbSStefano Zampini             PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATHERMITIANTRANSPOSEVIRTUAL, &isHTrans));
10505d955bbbSStefano Zampini             if (isHTrans) PetscCall(MatHermitianTransposeGetMat(sub, &sub));
10515d955bbbSStefano Zampini           }
10529d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQAIJ, &isSeqAIJ));
10539d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIAIJ, &isMPIAIJ));
10549d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQBAIJ, &isSeqBAIJ));
10559d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIBAIJ, &isMPIBAIJ));
10569d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQSBAIJ, &isSeqSBAIJ));
10579d0448ceSStefano Zampini           PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPISBAIJ, &isMPISBAIJ));
1058c3e1b152SPierre Jolivet           PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATDIAGONAL, &isDiag));
10599d0448ceSStefano Zampini 
10609d0448ceSStefano Zampini           if (chol) {
10619d0448ceSStefano Zampini             if (r == c) {
10629d0448ceSStefano Zampini               if (isSeqAIJ) convert_to_triples = MatConvertToTriples_seqaij_seqsbaij;
10639d0448ceSStefano Zampini               else if (isMPIAIJ) convert_to_triples = MatConvertToTriples_mpiaij_mpisbaij;
10649d0448ceSStefano Zampini               else if (isSeqSBAIJ) convert_to_triples = MatConvertToTriples_seqsbaij_seqsbaij;
10659d0448ceSStefano Zampini               else if (isMPISBAIJ) convert_to_triples = MatConvertToTriples_mpisbaij_mpisbaij;
1066c3e1b152SPierre Jolivet               else if (isDiag) convert_to_triples = MatConvertToTriples_diagonal_xaij;
10679d0448ceSStefano Zampini             } else {
10689d0448ceSStefano Zampini               if (isSeqAIJ) convert_to_triples = MatConvertToTriples_seqaij_seqaij;
10699d0448ceSStefano Zampini               else if (isMPIAIJ) convert_to_triples = MatConvertToTriples_mpiaij_mpiaij;
10709d0448ceSStefano Zampini               else if (isSeqBAIJ) convert_to_triples = MatConvertToTriples_seqbaij_seqaij;
10719d0448ceSStefano Zampini               else if (isMPIBAIJ) convert_to_triples = MatConvertToTriples_mpibaij_mpiaij;
1072c3e1b152SPierre Jolivet               else if (isDiag) convert_to_triples = MatConvertToTriples_diagonal_xaij;
10739d0448ceSStefano Zampini             }
10749d0448ceSStefano Zampini           } else {
10759d0448ceSStefano Zampini             if (isSeqAIJ) convert_to_triples = MatConvertToTriples_seqaij_seqaij;
10769d0448ceSStefano Zampini             else if (isMPIAIJ) convert_to_triples = MatConvertToTriples_mpiaij_mpiaij;
10779d0448ceSStefano Zampini             else if (isSeqBAIJ) convert_to_triples = MatConvertToTriples_seqbaij_seqaij;
10789d0448ceSStefano Zampini             else if (isMPIBAIJ) convert_to_triples = MatConvertToTriples_mpibaij_mpiaij;
1079c3e1b152SPierre Jolivet             else if (isDiag) convert_to_triples = MatConvertToTriples_diagonal_xaij;
10809d0448ceSStefano Zampini           }
10819d0448ceSStefano Zampini           PetscCheck(convert_to_triples, PetscObjectComm((PetscObject)sub), PETSC_ERR_SUP, "Not for block of type %s", ((PetscObject)sub)->type_name);
10829d0448ceSStefano Zampini           mumps->nest_convert_to_triples[r * nc + c] = convert_to_triples;
10839d0448ceSStefano Zampini           PetscCall(MatGetInfo(sub, MAT_LOCAL, &info));
10849d0448ceSStefano Zampini           cumnnz += (PetscInt64)info.nz_used; /* can be overestimated for Cholesky */
10859d0448ceSStefano Zampini           maxnnz = PetscMax(maxnnz, info.nz_used);
10869d0448ceSStefano Zampini         }
10879d0448ceSStefano Zampini       }
10889d0448ceSStefano Zampini     }
10899d0448ceSStefano Zampini 
10909d0448ceSStefano Zampini     /* Allocate total COO */
10919d0448ceSStefano Zampini     totnnz = cumnnz;
10929d0448ceSStefano Zampini     PetscCall(PetscMalloc2(totnnz, &irns, totnnz, &jcns));
10939d0448ceSStefano Zampini     PetscCall(PetscMalloc1(totnnz, &vals));
10949d0448ceSStefano Zampini 
10959d0448ceSStefano Zampini     /* Handle rows and column maps
10969d0448ceSStefano Zampini        We directly map rows and use an SF for the columns */
10979d0448ceSStefano Zampini     PetscCall(PetscMalloc4(nr, &rows, nc, &cols, nr, &rows_idx, nc, &cols_idx));
10989d0448ceSStefano Zampini     PetscCall(MatNestGetISs(A, rows, cols));
10999d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) PetscCall(ISGetIndices(rows[r], (const PetscInt **)&rows_idx[r]));
11009d0448ceSStefano Zampini     for (PetscInt c = 0; c < nc; c++) PetscCall(ISGetIndices(cols[c], (const PetscInt **)&cols_idx[c]));
11019d0448ceSStefano Zampini     if (PetscDefined(USE_64BIT_INDICES)) PetscCall(PetscMalloc1(maxnnz, &pjcns_w));
11025d955bbbSStefano Zampini     else (void)maxnnz;
11039d0448ceSStefano Zampini 
11049d0448ceSStefano Zampini     cumnnz = 0;
11059d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) {
11069d0448ceSStefano Zampini       for (PetscInt c = 0; c < nc; c++) {
11079d0448ceSStefano Zampini         Mat             sub  = mats[r][c];
11089d0448ceSStefano Zampini         const PetscInt *ridx = rows_idx[r];
11095d955bbbSStefano Zampini         const PetscInt *cidx = cols_idx[c];
11109d0448ceSStefano Zampini         PetscInt        rst;
11119d0448ceSStefano Zampini         PetscSF         csf;
11125d955bbbSStefano Zampini         PetscBool       isTrans, isHTrans = PETSC_FALSE, swap;
11135d955bbbSStefano Zampini         PetscLayout     cmap;
11149d0448ceSStefano Zampini 
11159d0448ceSStefano Zampini         mumps->nest_vals_start[r * nc + c] = cumnnz;
11169d0448ceSStefano Zampini         if (!mumps->nest_convert_to_triples[r * nc + c]) continue;
11179d0448ceSStefano Zampini 
11185d955bbbSStefano Zampini         /* Extract inner blocks if needed */
11195d955bbbSStefano Zampini         PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATTRANSPOSEVIRTUAL, &isTrans));
11205d955bbbSStefano Zampini         if (isTrans) PetscCall(MatTransposeGetMat(sub, &sub));
11215d955bbbSStefano Zampini         else {
11225d955bbbSStefano Zampini           PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATHERMITIANTRANSPOSEVIRTUAL, &isHTrans));
11235d955bbbSStefano Zampini           if (isHTrans) PetscCall(MatHermitianTransposeGetMat(sub, &sub));
11245d955bbbSStefano Zampini         }
11255d955bbbSStefano Zampini         swap = (PetscBool)(isTrans || isHTrans);
11265d955bbbSStefano Zampini 
11275d955bbbSStefano Zampini         /* Get column layout to map off-process columns */
11285d955bbbSStefano Zampini         PetscCall(MatGetLayouts(sub, NULL, &cmap));
11295d955bbbSStefano Zampini 
11305d955bbbSStefano Zampini         /* Get row start to map on-process rows */
11315d955bbbSStefano Zampini         PetscCall(MatGetOwnershipRange(sub, &rst, NULL));
11325d955bbbSStefano Zampini 
11339d0448ceSStefano Zampini         /* Directly use the mumps datastructure and use C ordering for now */
11349d0448ceSStefano Zampini         PetscCall((*mumps->nest_convert_to_triples[r * nc + c])(sub, 0, MAT_INITIAL_MATRIX, mumps));
11359d0448ceSStefano Zampini 
11365d955bbbSStefano Zampini         /* Swap the role of rows and columns indices for transposed blocks
11375d955bbbSStefano Zampini            since we need values with global final ordering */
11385d955bbbSStefano Zampini         if (swap) {
11395d955bbbSStefano Zampini           cidx = rows_idx[r];
11405d955bbbSStefano Zampini           ridx = cols_idx[c];
11419d0448ceSStefano Zampini         }
11429d0448ceSStefano Zampini 
11435d955bbbSStefano Zampini         /* Communicate column indices
11445d955bbbSStefano Zampini            This could have been done with a single SF but it would have complicated the code a lot.
11455d955bbbSStefano Zampini            But since we do it only once, we pay the price of setting up an SF for each block */
11465d955bbbSStefano Zampini         if (PetscDefined(USE_64BIT_INDICES)) {
11475d955bbbSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) pjcns_w[k] = mumps->jcn[k];
11485d955bbbSStefano Zampini         } else pjcns_w = (PetscInt *)(mumps->jcn); /* This cast is needed only to silence warnings for 64bit integers builds */
11499d0448ceSStefano Zampini         PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)A), &csf));
11505d955bbbSStefano Zampini         PetscCall(PetscSFSetGraphLayout(csf, cmap, mumps->nnz, NULL, PETSC_OWN_POINTER, pjcns_w));
11515d955bbbSStefano Zampini         PetscCall(PetscSFBcastBegin(csf, MPIU_INT, cidx, pjcns_w, MPI_REPLACE));
11525d955bbbSStefano Zampini         PetscCall(PetscSFBcastEnd(csf, MPIU_INT, cidx, pjcns_w, MPI_REPLACE));
11539d0448ceSStefano Zampini         PetscCall(PetscSFDestroy(&csf));
11549d0448ceSStefano Zampini 
11555d955bbbSStefano Zampini         /* Import indices: use direct map for rows and mapped indices for columns */
11565d955bbbSStefano Zampini         if (swap) {
11575d955bbbSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) {
11585d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(ridx[mumps->irn[k] - rst] + shift, &jcns[cumnnz + k]));
11595d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(pjcns_w[k] + shift, &irns[cumnnz + k]));
11605d955bbbSStefano Zampini           }
11615d955bbbSStefano Zampini         } else {
11625d955bbbSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) {
11635d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(ridx[mumps->irn[k] - rst] + shift, &irns[cumnnz + k]));
11645d955bbbSStefano Zampini             PetscCall(PetscMUMPSIntCast(pjcns_w[k] + shift, &jcns[cumnnz + k]));
11655d955bbbSStefano Zampini           }
11665d955bbbSStefano Zampini         }
11675d955bbbSStefano Zampini 
11685d955bbbSStefano Zampini         /* Import values to full COO */
11695d955bbbSStefano Zampini         PetscCall(PetscArraycpy(vals + cumnnz, mumps->val, mumps->nnz));
117050c845baSStefano Zampini         if (isHTrans) { /* conjugate the entries */
117150c845baSStefano Zampini           PetscScalar *v = vals + cumnnz;
117250c845baSStefano Zampini           for (PetscInt k = 0; k < mumps->nnz; k++) v[k] = PetscConj(v[k]);
117350c845baSStefano Zampini         }
11749d0448ceSStefano Zampini 
11759d0448ceSStefano Zampini         /* Shift new starting point and sanity check */
11769d0448ceSStefano Zampini         cumnnz += mumps->nnz;
11779d0448ceSStefano Zampini         PetscCheck(cumnnz <= totnnz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected number of nonzeros %" PetscInt64_FMT " != %" PetscInt64_FMT, cumnnz, totnnz);
11789d0448ceSStefano Zampini 
11799d0448ceSStefano Zampini         /* Free scratch memory */
11809d0448ceSStefano Zampini         PetscCall(PetscFree2(mumps->irn, mumps->jcn));
11819d0448ceSStefano Zampini         PetscCall(PetscFree(mumps->val_alloc));
11829d0448ceSStefano Zampini         mumps->val = NULL;
11839d0448ceSStefano Zampini         mumps->nnz = 0;
11849d0448ceSStefano Zampini       }
11859d0448ceSStefano Zampini     }
11869d0448ceSStefano Zampini     if (PetscDefined(USE_64BIT_INDICES)) PetscCall(PetscFree(pjcns_w));
11879d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) PetscCall(ISRestoreIndices(rows[r], (const PetscInt **)&rows_idx[r]));
11889d0448ceSStefano Zampini     for (PetscInt c = 0; c < nc; c++) PetscCall(ISRestoreIndices(cols[c], (const PetscInt **)&cols_idx[c]));
11899d0448ceSStefano Zampini     PetscCall(PetscFree4(rows, cols, rows_idx, cols_idx));
11909d0448ceSStefano Zampini     if (!chol) PetscCheck(cumnnz == totnnz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Different number of nonzeros %" PetscInt64_FMT " != %" PetscInt64_FMT, cumnnz, totnnz);
11915d955bbbSStefano Zampini     mumps->nest_vals_start[nr * nc] = cumnnz;
11929d0448ceSStefano Zampini 
11939d0448ceSStefano Zampini     /* Set pointers for final MUMPS data structure */
11949d0448ceSStefano Zampini     mumps->nest_vals = vals;
11959d0448ceSStefano Zampini     mumps->val_alloc = NULL; /* do not use val_alloc since it may be reallocated with the OMP callpath */
11969d0448ceSStefano Zampini     mumps->val       = vals;
11979d0448ceSStefano Zampini     mumps->irn       = irns;
11989d0448ceSStefano Zampini     mumps->jcn       = jcns;
11999d0448ceSStefano Zampini     mumps->nnz       = cumnnz;
12009d0448ceSStefano Zampini   } else {
12019d0448ceSStefano Zampini     PetscScalar *oval = mumps->nest_vals;
12029d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) {
12039d0448ceSStefano Zampini       for (PetscInt c = 0; c < nc; c++) {
12045d955bbbSStefano Zampini         PetscBool isTrans, isHTrans = PETSC_FALSE;
12055d955bbbSStefano Zampini         Mat       sub  = mats[r][c];
12065d955bbbSStefano Zampini         PetscInt  midx = r * nc + c;
12075d955bbbSStefano Zampini 
12085d955bbbSStefano Zampini         if (!mumps->nest_convert_to_triples[midx]) continue;
12095d955bbbSStefano Zampini         PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATTRANSPOSEVIRTUAL, &isTrans));
12105d955bbbSStefano Zampini         if (isTrans) PetscCall(MatTransposeGetMat(sub, &sub));
12115d955bbbSStefano Zampini         else {
12125d955bbbSStefano Zampini           PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATHERMITIANTRANSPOSEVIRTUAL, &isHTrans));
12135d955bbbSStefano Zampini           if (isHTrans) PetscCall(MatHermitianTransposeGetMat(sub, &sub));
12145d955bbbSStefano Zampini         }
12155d955bbbSStefano Zampini         mumps->val = oval + mumps->nest_vals_start[midx];
12165d955bbbSStefano Zampini         PetscCall((*mumps->nest_convert_to_triples[midx])(sub, shift, MAT_REUSE_MATRIX, mumps));
12175d955bbbSStefano Zampini         if (isHTrans) {
12185d955bbbSStefano Zampini           PetscInt nnz = mumps->nest_vals_start[midx + 1] - mumps->nest_vals_start[midx];
12195d955bbbSStefano Zampini           for (PetscInt k = 0; k < nnz; k++) mumps->val[k] = PetscConj(mumps->val[k]);
12205d955bbbSStefano Zampini         }
12219d0448ceSStefano Zampini       }
12229d0448ceSStefano Zampini     }
12239d0448ceSStefano Zampini     mumps->val = oval;
12249d0448ceSStefano Zampini   }
12259d0448ceSStefano Zampini   PetscFunctionReturn(PETSC_SUCCESS);
12269d0448ceSStefano Zampini }
12279d0448ceSStefano Zampini 
122866976f2fSJacob Faibussowitsch static PetscErrorCode MatDestroy_MUMPS(Mat A)
1229d71ae5a4SJacob Faibussowitsch {
1230a6053eceSJunchao Zhang   Mat_MUMPS *mumps = (Mat_MUMPS *)A->data;
1231b24902e0SBarry Smith 
1232397b6df1SKris Buschelman   PetscFunctionBegin;
12339566063dSJacob Faibussowitsch   PetscCall(PetscFree2(mumps->id.sol_loc, mumps->id.isol_loc));
12349566063dSJacob Faibussowitsch   PetscCall(VecScatterDestroy(&mumps->scat_rhs));
12359566063dSJacob Faibussowitsch   PetscCall(VecScatterDestroy(&mumps->scat_sol));
12369566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&mumps->b_seq));
12379566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&mumps->x_seq));
12389566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.perm_in));
12399566063dSJacob Faibussowitsch   PetscCall(PetscFree2(mumps->irn, mumps->jcn));
12409566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->val_alloc));
12419566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->info));
1242413bcc21SPierre Jolivet   PetscCall(PetscFree(mumps->ICNTL_pre));
1243413bcc21SPierre Jolivet   PetscCall(PetscFree(mumps->CNTL_pre));
12449566063dSJacob Faibussowitsch   PetscCall(MatMumpsResetSchur_Private(mumps));
1245413bcc21SPierre Jolivet   if (mumps->id.job != JOB_NULL) { /* cannot call PetscMUMPS_c() if JOB_INIT has never been called for this instance */
1246a5e57a09SHong Zhang     mumps->id.job = JOB_END;
12473ab56b82SJunchao Zhang     PetscMUMPS_c(mumps);
12489261f6e4SBarry 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));
1249413bcc21SPierre Jolivet     if (mumps->mumps_comm != MPI_COMM_NULL) {
1250413bcc21SPierre Jolivet       if (PetscDefined(HAVE_OPENMP_SUPPORT) && mumps->use_petsc_omp_support) PetscCallMPI(MPI_Comm_free(&mumps->mumps_comm));
1251413bcc21SPierre Jolivet       else PetscCall(PetscCommRestoreComm(PetscObjectComm((PetscObject)A), &mumps->mumps_comm));
1252413bcc21SPierre Jolivet     }
1253413bcc21SPierre Jolivet   }
12543ab56b82SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
125567602552SJunchao Zhang   if (mumps->use_petsc_omp_support) {
12569566063dSJacob Faibussowitsch     PetscCall(PetscOmpCtrlDestroy(&mumps->omp_ctrl));
12579566063dSJacob Faibussowitsch     PetscCall(PetscFree2(mumps->rhs_loc, mumps->rhs_recvbuf));
12589566063dSJacob Faibussowitsch     PetscCall(PetscFree3(mumps->rhs_nrow, mumps->rhs_recvcounts, mumps->rhs_disps));
125967602552SJunchao Zhang   }
12603ab56b82SJunchao Zhang #endif
12619566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->ia_alloc));
12629566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->ja_alloc));
12639566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->recvcount));
12649566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->reqs));
12659566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->irhs_loc));
12669d0448ceSStefano Zampini   PetscCall(PetscFree2(mumps->nest_vals_start, mumps->nest_convert_to_triples));
12679d0448ceSStefano Zampini   PetscCall(PetscFree(mumps->nest_vals));
12689566063dSJacob Faibussowitsch   PetscCall(PetscFree(A->data));
1269bf0cc555SLisandro Dalcin 
127097969023SHong Zhang   /* clear composed functions */
12719566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorGetSolverType_C", NULL));
12729566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorSetSchurIS_C", NULL));
12739566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatFactorCreateSchurComplement_C", NULL));
12749566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsSetIcntl_C", NULL));
12759566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetIcntl_C", NULL));
12769566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsSetCntl_C", NULL));
12779566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetCntl_C", NULL));
12789566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInfo_C", NULL));
12799566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInfog_C", NULL));
12809566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetRinfo_C", NULL));
12819566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetRinfog_C", NULL));
12825c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetNullPivots_C", NULL));
12839566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInverse_C", NULL));
12849566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMumpsGetInverseTranspose_C", NULL));
12853ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1286397b6df1SKris Buschelman }
1287397b6df1SKris Buschelman 
128867602552SJunchao 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. */
1289d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMumpsSetUpDistRHSInfo(Mat A, PetscInt nrhs, const PetscScalar *array)
1290d71ae5a4SJacob Faibussowitsch {
129167602552SJunchao Zhang   Mat_MUMPS        *mumps   = (Mat_MUMPS *)A->data;
129267602552SJunchao Zhang   const PetscMPIInt ompsize = mumps->omp_comm_size;
129367602552SJunchao Zhang   PetscInt          i, m, M, rstart;
129467602552SJunchao Zhang 
129567602552SJunchao Zhang   PetscFunctionBegin;
12969566063dSJacob Faibussowitsch   PetscCall(MatGetSize(A, &M, NULL));
12979566063dSJacob Faibussowitsch   PetscCall(MatGetLocalSize(A, &m, NULL));
129808401ef6SPierre Jolivet   PetscCheck(M <= PETSC_MUMPS_INT_MAX, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "PetscInt too long for PetscMUMPSInt");
129967602552SJunchao Zhang   if (ompsize == 1) {
130067602552SJunchao Zhang     if (!mumps->irhs_loc) {
130167602552SJunchao Zhang       mumps->nloc_rhs = m;
13029566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(m, &mumps->irhs_loc));
13039566063dSJacob Faibussowitsch       PetscCall(MatGetOwnershipRange(A, &rstart, NULL));
130467602552SJunchao Zhang       for (i = 0; i < m; i++) mumps->irhs_loc[i] = rstart + i + 1; /* use 1-based indices */
130567602552SJunchao Zhang     }
130667602552SJunchao Zhang     mumps->id.rhs_loc = (MumpsScalar *)array;
130767602552SJunchao Zhang   } else {
130867602552SJunchao Zhang #if defined(PETSC_HAVE_OPENMP_SUPPORT)
130967602552SJunchao Zhang     const PetscInt *ranges;
131067602552SJunchao Zhang     PetscMPIInt     j, k, sendcount, *petsc_ranks, *omp_ranks;
131167602552SJunchao Zhang     MPI_Group       petsc_group, omp_group;
131267602552SJunchao Zhang     PetscScalar    *recvbuf = NULL;
131367602552SJunchao Zhang 
131467602552SJunchao Zhang     if (mumps->is_omp_master) {
131567602552SJunchao Zhang       /* Lazily initialize the omp stuff for distributed rhs */
131667602552SJunchao Zhang       if (!mumps->irhs_loc) {
13179566063dSJacob Faibussowitsch         PetscCall(PetscMalloc2(ompsize, &omp_ranks, ompsize, &petsc_ranks));
13189566063dSJacob Faibussowitsch         PetscCall(PetscMalloc3(ompsize, &mumps->rhs_nrow, ompsize, &mumps->rhs_recvcounts, ompsize, &mumps->rhs_disps));
13199566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Comm_group(mumps->petsc_comm, &petsc_group));
13209566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Comm_group(mumps->omp_comm, &omp_group));
132167602552SJunchao Zhang         for (j = 0; j < ompsize; j++) omp_ranks[j] = j;
13229566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Group_translate_ranks(omp_group, ompsize, omp_ranks, petsc_group, petsc_ranks));
132367602552SJunchao Zhang 
132467602552SJunchao Zhang         /* Populate mumps->irhs_loc[], rhs_nrow[] */
132567602552SJunchao Zhang         mumps->nloc_rhs = 0;
13269566063dSJacob Faibussowitsch         PetscCall(MatGetOwnershipRanges(A, &ranges));
132767602552SJunchao Zhang         for (j = 0; j < ompsize; j++) {
132867602552SJunchao Zhang           mumps->rhs_nrow[j] = ranges[petsc_ranks[j] + 1] - ranges[petsc_ranks[j]];
132967602552SJunchao Zhang           mumps->nloc_rhs += mumps->rhs_nrow[j];
133067602552SJunchao Zhang         }
13319566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(mumps->nloc_rhs, &mumps->irhs_loc));
133267602552SJunchao Zhang         for (j = k = 0; j < ompsize; j++) {
133367602552SJunchao 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 */
133467602552SJunchao Zhang         }
133567602552SJunchao Zhang 
13369566063dSJacob Faibussowitsch         PetscCall(PetscFree2(omp_ranks, petsc_ranks));
13379566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Group_free(&petsc_group));
13389566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Group_free(&omp_group));
133967602552SJunchao Zhang       }
134067602552SJunchao Zhang 
134167602552SJunchao Zhang       /* Realloc buffers when current nrhs is bigger than what we have met */
134267602552SJunchao Zhang       if (nrhs > mumps->max_nrhs) {
13439566063dSJacob Faibussowitsch         PetscCall(PetscFree2(mumps->rhs_loc, mumps->rhs_recvbuf));
13449566063dSJacob Faibussowitsch         PetscCall(PetscMalloc2(mumps->nloc_rhs * nrhs, &mumps->rhs_loc, mumps->nloc_rhs * nrhs, &mumps->rhs_recvbuf));
134567602552SJunchao Zhang         mumps->max_nrhs = nrhs;
134667602552SJunchao Zhang       }
134767602552SJunchao Zhang 
134867602552SJunchao Zhang       /* Setup recvcounts[], disps[], recvbuf on omp rank 0 for the upcoming MPI_Gatherv */
13499566063dSJacob Faibussowitsch       for (j = 0; j < ompsize; j++) PetscCall(PetscMPIIntCast(mumps->rhs_nrow[j] * nrhs, &mumps->rhs_recvcounts[j]));
135067602552SJunchao Zhang       mumps->rhs_disps[0] = 0;
135167602552SJunchao Zhang       for (j = 1; j < ompsize; j++) {
135267602552SJunchao Zhang         mumps->rhs_disps[j] = mumps->rhs_disps[j - 1] + mumps->rhs_recvcounts[j - 1];
135308401ef6SPierre Jolivet         PetscCheck(mumps->rhs_disps[j] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "PetscMPIInt overflow!");
135467602552SJunchao Zhang       }
135567602552SJunchao Zhang       recvbuf = (nrhs == 1) ? mumps->rhs_loc : mumps->rhs_recvbuf; /* Directly use rhs_loc[] as recvbuf. Single rhs is common in Ax=b */
135667602552SJunchao Zhang     }
135767602552SJunchao Zhang 
13589566063dSJacob Faibussowitsch     PetscCall(PetscMPIIntCast(m * nrhs, &sendcount));
13599566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Gatherv(array, sendcount, MPIU_SCALAR, recvbuf, mumps->rhs_recvcounts, mumps->rhs_disps, MPIU_SCALAR, 0, mumps->omp_comm));
136067602552SJunchao Zhang 
136167602552SJunchao Zhang     if (mumps->is_omp_master) {
136267602552SJunchao Zhang       if (nrhs > 1) { /* Copy & re-arrange data from rhs_recvbuf[] to mumps->rhs_loc[] only when there are multiple rhs */
136367602552SJunchao Zhang         PetscScalar *dst, *dstbase = mumps->rhs_loc;
136467602552SJunchao Zhang         for (j = 0; j < ompsize; j++) {
136567602552SJunchao Zhang           const PetscScalar *src = mumps->rhs_recvbuf + mumps->rhs_disps[j];
136667602552SJunchao Zhang           dst                    = dstbase;
136767602552SJunchao Zhang           for (i = 0; i < nrhs; i++) {
13689566063dSJacob Faibussowitsch             PetscCall(PetscArraycpy(dst, src, mumps->rhs_nrow[j]));
136967602552SJunchao Zhang             src += mumps->rhs_nrow[j];
137067602552SJunchao Zhang             dst += mumps->nloc_rhs;
137167602552SJunchao Zhang           }
137267602552SJunchao Zhang           dstbase += mumps->rhs_nrow[j];
137367602552SJunchao Zhang         }
137467602552SJunchao Zhang       }
137567602552SJunchao Zhang       mumps->id.rhs_loc = (MumpsScalar *)mumps->rhs_loc;
137667602552SJunchao Zhang     }
137767602552SJunchao Zhang #endif /* PETSC_HAVE_OPENMP_SUPPORT */
137867602552SJunchao Zhang   }
137967602552SJunchao Zhang   mumps->id.nrhs     = nrhs;
138067602552SJunchao Zhang   mumps->id.nloc_rhs = mumps->nloc_rhs;
138167602552SJunchao Zhang   mumps->id.lrhs_loc = mumps->nloc_rhs;
138267602552SJunchao Zhang   mumps->id.irhs_loc = mumps->irhs_loc;
13833ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
138467602552SJunchao Zhang }
138567602552SJunchao Zhang 
138666976f2fSJacob Faibussowitsch static PetscErrorCode MatSolve_MUMPS(Mat A, Vec b, Vec x)
1387d71ae5a4SJacob Faibussowitsch {
1388e69c285eSBarry Smith   Mat_MUMPS         *mumps  = (Mat_MUMPS *)A->data;
138925aac85cSJunchao Zhang   const PetscScalar *rarray = NULL;
1390d54de34fSKris Buschelman   PetscScalar       *array;
1391329ec9b3SHong Zhang   IS                 is_iden, is_petsc;
1392329ec9b3SHong Zhang   PetscInt           i;
1393cc86f929SStefano Zampini   PetscBool          second_solve = PETSC_FALSE;
1394883f2eb9SBarry Smith   static PetscBool   cite1 = PETSC_FALSE, cite2 = PETSC_FALSE;
1395397b6df1SKris Buschelman 
1396397b6df1SKris Buschelman   PetscFunctionBegin;
13979371c9d4SSatish 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 "
13989371c9d4SSatish Balay                                    "Journal on Matrix Analysis and Applications},\n  volume = {23},\n  number = {1},\n  pages = {15--41},\n  year = {2001}\n}\n",
13999371c9d4SSatish Balay                                    &cite1));
14009371c9d4SSatish 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 "
14019371c9d4SSatish Balay                                    "Computing},\n  volume = {32},\n  number = {2},\n  pages = {136--156},\n  year = {2006}\n}\n",
14029371c9d4SSatish Balay                                    &cite2));
14032aca8efcSHong Zhang 
1404603e8f96SBarry Smith   if (A->factorerrortype) {
14059566063dSJacob 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)));
14069566063dSJacob Faibussowitsch     PetscCall(VecSetInf(x));
14073ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
14082aca8efcSHong Zhang   }
14092aca8efcSHong Zhang 
1410a5e57a09SHong Zhang   mumps->id.nrhs = 1;
14112d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) {
141225aac85cSJunchao Zhang     if (mumps->ICNTL20 == 10) {
141367602552SJunchao Zhang       mumps->id.ICNTL(20) = 10; /* dense distributed RHS */
14149566063dSJacob Faibussowitsch       PetscCall(VecGetArrayRead(b, &rarray));
14159566063dSJacob Faibussowitsch       PetscCall(MatMumpsSetUpDistRHSInfo(A, 1, rarray));
141625aac85cSJunchao Zhang     } else {
141741ffd417SStefano Zampini       mumps->id.ICNTL(20) = 0; /* dense centralized RHS; Scatter b into a sequential rhs vector*/
14189566063dSJacob Faibussowitsch       PetscCall(VecScatterBegin(mumps->scat_rhs, b, mumps->b_seq, INSERT_VALUES, SCATTER_FORWARD));
14199566063dSJacob Faibussowitsch       PetscCall(VecScatterEnd(mumps->scat_rhs, b, mumps->b_seq, INSERT_VALUES, SCATTER_FORWARD));
142067602552SJunchao Zhang       if (!mumps->myid) {
14219566063dSJacob Faibussowitsch         PetscCall(VecGetArray(mumps->b_seq, &array));
142267602552SJunchao Zhang         mumps->id.rhs = (MumpsScalar *)array;
142367602552SJunchao Zhang       }
142425aac85cSJunchao Zhang     }
14253ab56b82SJunchao Zhang   } else {                   /* petsc_size == 1 */
142667602552SJunchao Zhang     mumps->id.ICNTL(20) = 0; /* dense centralized RHS */
14279566063dSJacob Faibussowitsch     PetscCall(VecCopy(b, x));
14289566063dSJacob Faibussowitsch     PetscCall(VecGetArray(x, &array));
1429940cd9d6SSatish Balay     mumps->id.rhs = (MumpsScalar *)array;
1430397b6df1SKris Buschelman   }
1431397b6df1SKris Buschelman 
1432cc86f929SStefano Zampini   /*
1433cc86f929SStefano Zampini      handle condensation step of Schur complement (if any)
1434cc86f929SStefano Zampini      We set by default ICNTL(26) == -1 when Schur indices have been provided by the user.
1435cc86f929SStefano Zampini      According to MUMPS (5.0.0) manual, any value should be harmful during the factorization phase
1436cc86f929SStefano Zampini      Unless the user provides a valid value for ICNTL(26), MatSolve and MatMatSolve routines solve the full system.
1437cc86f929SStefano Zampini      This requires an extra call to PetscMUMPS_c and the computation of the factors for S
1438cc86f929SStefano Zampini   */
14393e5b40d0SPierre Jolivet   if (mumps->id.size_schur > 0) {
144008401ef6SPierre Jolivet     PetscCheck(mumps->petsc_size <= 1, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Parallel Schur complements not yet supported from PETSc");
14413e5b40d0SPierre Jolivet     if (mumps->id.ICNTL(26) < 0 || mumps->id.ICNTL(26) > 2) {
1442cc86f929SStefano Zampini       second_solve = PETSC_TRUE;
14439566063dSJacob Faibussowitsch       PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
14443e5b40d0SPierre Jolivet       mumps->id.ICNTL(26) = 1; /* condensation phase */
14453e5b40d0SPierre Jolivet     } else if (mumps->id.ICNTL(26) == 1) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
1446cc86f929SStefano Zampini   }
1447397b6df1SKris Buschelman   /* solve phase */
1448a5e57a09SHong Zhang   mumps->id.job = JOB_SOLVE;
14493ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
14509261f6e4SBarry 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));
1451397b6df1SKris Buschelman 
1452b5fa320bSStefano Zampini   /* handle expansion step of Schur complement (if any) */
14531baa6e33SBarry Smith   if (second_solve) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_TRUE));
14543e5b40d0SPierre Jolivet   else if (mumps->id.ICNTL(26) == 1) {
14553e5b40d0SPierre Jolivet     PetscCall(MatMumpsSolveSchur_Private(A));
14563e5b40d0SPierre Jolivet     for (i = 0; i < mumps->id.size_schur; ++i) {
14573e5b40d0SPierre Jolivet #if !defined(PETSC_USE_COMPLEX)
14583e5b40d0SPierre Jolivet       PetscScalar val = mumps->id.redrhs[i];
14593e5b40d0SPierre Jolivet #else
14603e5b40d0SPierre Jolivet       PetscScalar val = mumps->id.redrhs[i].r + PETSC_i * mumps->id.redrhs[i].i;
14613e5b40d0SPierre Jolivet #endif
14623e5b40d0SPierre Jolivet       array[mumps->id.listvar_schur[i] - 1] = val;
14633e5b40d0SPierre Jolivet     }
14643e5b40d0SPierre Jolivet   }
1465b5fa320bSStefano Zampini 
14662d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) { /* convert mumps distributed solution to petsc mpi x */
1467a5e57a09SHong Zhang     if (mumps->scat_sol && mumps->ICNTL9_pre != mumps->id.ICNTL(9)) {
1468a5e57a09SHong Zhang       /* when id.ICNTL(9) changes, the contents of lsol_loc may change (not its size, lsol_loc), recreates scat_sol */
14699566063dSJacob Faibussowitsch       PetscCall(VecScatterDestroy(&mumps->scat_sol));
1470397b6df1SKris Buschelman     }
1471a5e57a09SHong Zhang     if (!mumps->scat_sol) { /* create scatter scat_sol */
1472a6053eceSJunchao Zhang       PetscInt *isol2_loc = NULL;
14739566063dSJacob Faibussowitsch       PetscCall(ISCreateStride(PETSC_COMM_SELF, mumps->id.lsol_loc, 0, 1, &is_iden)); /* from */
14749566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(mumps->id.lsol_loc, &isol2_loc));
1475a6053eceSJunchao 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 */
14769566063dSJacob Faibussowitsch       PetscCall(ISCreateGeneral(PETSC_COMM_SELF, mumps->id.lsol_loc, isol2_loc, PETSC_OWN_POINTER, &is_petsc)); /* to */
14779566063dSJacob Faibussowitsch       PetscCall(VecScatterCreate(mumps->x_seq, is_iden, x, is_petsc, &mumps->scat_sol));
14789566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_iden));
14799566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_petsc));
1480a5e57a09SHong Zhang       mumps->ICNTL9_pre = mumps->id.ICNTL(9); /* save current value of id.ICNTL(9) */
1481397b6df1SKris Buschelman     }
1482a5e57a09SHong Zhang 
14839566063dSJacob Faibussowitsch     PetscCall(VecScatterBegin(mumps->scat_sol, mumps->x_seq, x, INSERT_VALUES, SCATTER_FORWARD));
14849566063dSJacob Faibussowitsch     PetscCall(VecScatterEnd(mumps->scat_sol, mumps->x_seq, x, INSERT_VALUES, SCATTER_FORWARD));
1485329ec9b3SHong Zhang   }
1486353d7d71SJunchao Zhang 
148767602552SJunchao Zhang   if (mumps->petsc_size > 1) {
148825aac85cSJunchao Zhang     if (mumps->ICNTL20 == 10) {
14899566063dSJacob Faibussowitsch       PetscCall(VecRestoreArrayRead(b, &rarray));
149025aac85cSJunchao Zhang     } else if (!mumps->myid) {
14919566063dSJacob Faibussowitsch       PetscCall(VecRestoreArray(mumps->b_seq, &array));
149225aac85cSJunchao Zhang     }
14939566063dSJacob Faibussowitsch   } else PetscCall(VecRestoreArray(x, &array));
1494353d7d71SJunchao Zhang 
149564412097SPierre Jolivet   PetscCall(PetscLogFlops(2.0 * PetscMax(0, (mumps->id.INFO(28) >= 0 ? mumps->id.INFO(28) : -1000000 * mumps->id.INFO(28)) - A->cmap->n)));
14963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1497397b6df1SKris Buschelman }
1498397b6df1SKris Buschelman 
149966976f2fSJacob Faibussowitsch static PetscErrorCode MatSolveTranspose_MUMPS(Mat A, Vec b, Vec x)
1500d71ae5a4SJacob Faibussowitsch {
1501e69c285eSBarry Smith   Mat_MUMPS          *mumps = (Mat_MUMPS *)A->data;
1502338d3105SPierre Jolivet   const PetscMUMPSInt value = mumps->id.ICNTL(9);
150351d5961aSHong Zhang 
150451d5961aSHong Zhang   PetscFunctionBegin;
1505a5e57a09SHong Zhang   mumps->id.ICNTL(9) = 0;
15069566063dSJacob Faibussowitsch   PetscCall(MatSolve_MUMPS(A, b, x));
1507338d3105SPierre Jolivet   mumps->id.ICNTL(9) = value;
15083ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
150951d5961aSHong Zhang }
151051d5961aSHong Zhang 
151166976f2fSJacob Faibussowitsch static PetscErrorCode MatMatSolve_MUMPS(Mat A, Mat B, Mat X)
1512d71ae5a4SJacob Faibussowitsch {
1513b8491c3eSStefano Zampini   Mat                Bt = NULL;
1514a6053eceSJunchao Zhang   PetscBool          denseX, denseB, flg, flgT;
1515e69c285eSBarry Smith   Mat_MUMPS         *mumps = (Mat_MUMPS *)A->data;
1516334c5f61SHong Zhang   PetscInt           i, nrhs, M;
15171683a169SBarry Smith   PetscScalar       *array;
15181683a169SBarry Smith   const PetscScalar *rbray;
1519a6053eceSJunchao Zhang   PetscInt           lsol_loc, nlsol_loc, *idxx, iidx = 0;
1520a6053eceSJunchao Zhang   PetscMUMPSInt     *isol_loc, *isol_loc_save;
15211683a169SBarry Smith   PetscScalar       *bray, *sol_loc, *sol_loc_save;
1522be818407SHong Zhang   IS                 is_to, is_from;
1523beae5ec0SHong Zhang   PetscInt           k, proc, j, m, myrstart;
1524be818407SHong Zhang   const PetscInt    *rstart;
152567602552SJunchao Zhang   Vec                v_mpi, msol_loc;
152667602552SJunchao Zhang   VecScatter         scat_sol;
152767602552SJunchao Zhang   Vec                b_seq;
152867602552SJunchao Zhang   VecScatter         scat_rhs;
1529be818407SHong Zhang   PetscScalar       *aa;
1530be818407SHong Zhang   PetscInt           spnr, *ia, *ja;
1531d56c302dSHong Zhang   Mat_MPIAIJ        *b = NULL;
1532bda8bf91SBarry Smith 
1533e0b74bf9SHong Zhang   PetscFunctionBegin;
15349566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)X, &denseX, MATSEQDENSE, MATMPIDENSE, NULL));
153528b400f6SJacob Faibussowitsch   PetscCheck(denseX, PetscObjectComm((PetscObject)X), PETSC_ERR_ARG_WRONG, "Matrix X must be MATDENSE matrix");
1536be818407SHong Zhang 
15379566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)B, &denseB, MATSEQDENSE, MATMPIDENSE, NULL));
1538a6053eceSJunchao Zhang   if (denseB) {
153908401ef6SPierre Jolivet     PetscCheck(B->rmap->n == X->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Matrix B and X must have same row distribution");
1540be818407SHong Zhang     mumps->id.ICNTL(20) = 0; /* dense RHS */
15410e6b8875SHong Zhang   } else {                   /* sparse B */
154208401ef6SPierre Jolivet     PetscCheck(X != B, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_IDN, "X and B must be different matrices");
1543013e2dc7SBarry Smith     PetscCall(PetscObjectTypeCompare((PetscObject)B, MATTRANSPOSEVIRTUAL, &flgT));
1544da81f932SPierre Jolivet     if (flgT) { /* input B is transpose of actual RHS matrix,
15450e6b8875SHong Zhang                  because mumps requires sparse compressed COLUMN storage! See MatMatTransposeSolve_MUMPS() */
15469566063dSJacob Faibussowitsch       PetscCall(MatTransposeGetMat(B, &Bt));
1547013e2dc7SBarry Smith     } else SETERRQ(PetscObjectComm((PetscObject)B), PETSC_ERR_ARG_WRONG, "Matrix B must be MATTRANSPOSEVIRTUAL matrix");
1548be818407SHong Zhang     mumps->id.ICNTL(20) = 1; /* sparse RHS */
1549b8491c3eSStefano Zampini   }
155087b22cf4SHong Zhang 
15519566063dSJacob Faibussowitsch   PetscCall(MatGetSize(B, &M, &nrhs));
15529481e6e9SHong Zhang   mumps->id.nrhs = nrhs;
15539481e6e9SHong Zhang   mumps->id.lrhs = M;
15542b691707SHong Zhang   mumps->id.rhs  = NULL;
15559481e6e9SHong Zhang 
15562d4298aeSJunchao Zhang   if (mumps->petsc_size == 1) {
1557b8491c3eSStefano Zampini     PetscScalar *aa;
1558b8491c3eSStefano Zampini     PetscInt     spnr, *ia, *ja;
1559e94cce23SStefano Zampini     PetscBool    second_solve = PETSC_FALSE;
1560b8491c3eSStefano Zampini 
15619566063dSJacob Faibussowitsch     PetscCall(MatDenseGetArray(X, &array));
1562b8491c3eSStefano Zampini     mumps->id.rhs = (MumpsScalar *)array;
15632b691707SHong Zhang 
1564a6053eceSJunchao Zhang     if (denseB) {
15652b691707SHong Zhang       /* copy B to X */
15669566063dSJacob Faibussowitsch       PetscCall(MatDenseGetArrayRead(B, &rbray));
15679566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(array, rbray, M * nrhs));
15689566063dSJacob Faibussowitsch       PetscCall(MatDenseRestoreArrayRead(B, &rbray));
15692b691707SHong Zhang     } else { /* sparse B */
15709566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJGetArray(Bt, &aa));
15719566063dSJacob Faibussowitsch       PetscCall(MatGetRowIJ(Bt, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
157228b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
15739566063dSJacob Faibussowitsch       PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs));
1574b8491c3eSStefano Zampini       mumps->id.rhs_sparse = (MumpsScalar *)aa;
1575b8491c3eSStefano Zampini     }
1576e94cce23SStefano Zampini     /* handle condensation step of Schur complement (if any) */
15773e5b40d0SPierre Jolivet     if (mumps->id.size_schur > 0) {
15783e5b40d0SPierre Jolivet       PetscCheck(mumps->petsc_size <= 1, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Parallel Schur complements not yet supported from PETSc");
15793e5b40d0SPierre Jolivet       if (mumps->id.ICNTL(26) < 0 || mumps->id.ICNTL(26) > 2) {
1580e94cce23SStefano Zampini         second_solve = PETSC_TRUE;
15819566063dSJacob Faibussowitsch         PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
15823e5b40d0SPierre Jolivet         mumps->id.ICNTL(26) = 1; /* condensation phase */
15833e5b40d0SPierre Jolivet       } else if (mumps->id.ICNTL(26) == 1) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE));
1584e94cce23SStefano Zampini     }
15852cd7d884SHong Zhang     /* solve phase */
15862cd7d884SHong Zhang     mumps->id.job = JOB_SOLVE;
15873ab56b82SJunchao Zhang     PetscMUMPS_c(mumps);
15889261f6e4SBarry 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));
1589b5fa320bSStefano Zampini 
1590b5fa320bSStefano Zampini     /* handle expansion step of Schur complement (if any) */
15911baa6e33SBarry Smith     if (second_solve) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_TRUE));
15923e5b40d0SPierre Jolivet     else if (mumps->id.ICNTL(26) == 1) {
15933e5b40d0SPierre Jolivet       PetscCall(MatMumpsSolveSchur_Private(A));
15943e5b40d0SPierre Jolivet       for (j = 0; j < nrhs; ++j)
15953e5b40d0SPierre Jolivet         for (i = 0; i < mumps->id.size_schur; ++i) {
15963e5b40d0SPierre Jolivet #if !defined(PETSC_USE_COMPLEX)
15973e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.redrhs[i + j * mumps->id.lredrhs];
15983e5b40d0SPierre Jolivet #else
15993e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.redrhs[i + j * mumps->id.lredrhs].r + PETSC_i * mumps->id.redrhs[i + j * mumps->id.lredrhs].i;
16003e5b40d0SPierre Jolivet #endif
16013e5b40d0SPierre Jolivet           array[mumps->id.listvar_schur[i] - 1 + j * M] = val;
16023e5b40d0SPierre Jolivet         }
16033e5b40d0SPierre Jolivet     }
1604a6053eceSJunchao Zhang     if (!denseB) { /* sparse B */
16059566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJRestoreArray(Bt, &aa));
16069566063dSJacob Faibussowitsch       PetscCall(MatRestoreRowIJ(Bt, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
160728b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot restore IJ structure");
1608b8491c3eSStefano Zampini     }
16099566063dSJacob Faibussowitsch     PetscCall(MatDenseRestoreArray(X, &array));
16103ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
1611be818407SHong Zhang   }
1612801fbe65SHong Zhang 
16132ef1f0ffSBarry Smith   /* parallel case: MUMPS requires rhs B to be centralized on the host! */
1614aed4548fSBarry Smith   PetscCheck(mumps->petsc_size <= 1 || !mumps->id.ICNTL(19), PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Parallel Schur complements not yet supported from PETSc");
1615241dbb5eSStefano Zampini 
1616beae5ec0SHong Zhang   /* create msol_loc to hold mumps local solution */
16171683a169SBarry Smith   isol_loc_save = mumps->id.isol_loc; /* save it for MatSolve() */
16181683a169SBarry Smith   sol_loc_save  = (PetscScalar *)mumps->id.sol_loc;
1619801fbe65SHong Zhang 
1620a1dfcbd9SJunchao Zhang   lsol_loc  = mumps->id.lsol_loc;
162171aed81dSHong Zhang   nlsol_loc = nrhs * lsol_loc; /* length of sol_loc */
16229566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(nlsol_loc, &sol_loc, lsol_loc, &isol_loc));
1623940cd9d6SSatish Balay   mumps->id.sol_loc  = (MumpsScalar *)sol_loc;
1624801fbe65SHong Zhang   mumps->id.isol_loc = isol_loc;
1625801fbe65SHong Zhang 
16269566063dSJacob Faibussowitsch   PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, nlsol_loc, (PetscScalar *)sol_loc, &msol_loc));
16272cd7d884SHong Zhang 
162867602552SJunchao Zhang   if (denseB) {
162925aac85cSJunchao Zhang     if (mumps->ICNTL20 == 10) {
163067602552SJunchao Zhang       mumps->id.ICNTL(20) = 10; /* dense distributed RHS */
16319566063dSJacob Faibussowitsch       PetscCall(MatDenseGetArrayRead(B, &rbray));
16329566063dSJacob Faibussowitsch       PetscCall(MatMumpsSetUpDistRHSInfo(A, nrhs, rbray));
16339566063dSJacob Faibussowitsch       PetscCall(MatDenseRestoreArrayRead(B, &rbray));
16349566063dSJacob Faibussowitsch       PetscCall(MatGetLocalSize(B, &m, NULL));
16359566063dSJacob Faibussowitsch       PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)B), 1, nrhs * m, nrhs * M, NULL, &v_mpi));
163625aac85cSJunchao Zhang     } else {
163725aac85cSJunchao Zhang       mumps->id.ICNTL(20) = 0; /* dense centralized RHS */
163880577c12SJunchao Zhang       /* TODO: Because of non-contiguous indices, the created vecscatter scat_rhs is not done in MPI_Gather, resulting in
163980577c12SJunchao Zhang         very inefficient communication. An optimization is to use VecScatterCreateToZero to gather B to rank 0. Then on rank
164080577c12SJunchao Zhang         0, re-arrange B into desired order, which is a local operation.
164180577c12SJunchao Zhang       */
164280577c12SJunchao Zhang 
164367602552SJunchao Zhang       /* scatter v_mpi to b_seq because MUMPS before 5.3.0 only supports centralized rhs */
1644be818407SHong Zhang       /* wrap dense rhs matrix B into a vector v_mpi */
16459566063dSJacob Faibussowitsch       PetscCall(MatGetLocalSize(B, &m, NULL));
16469566063dSJacob Faibussowitsch       PetscCall(MatDenseGetArray(B, &bray));
16479566063dSJacob Faibussowitsch       PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)B), 1, nrhs * m, nrhs * M, (const PetscScalar *)bray, &v_mpi));
16489566063dSJacob Faibussowitsch       PetscCall(MatDenseRestoreArray(B, &bray));
16492b691707SHong Zhang 
1650be818407SHong Zhang       /* scatter v_mpi to b_seq in proc[0]. MUMPS requires rhs to be centralized on the host! */
1651801fbe65SHong Zhang       if (!mumps->myid) {
1652beae5ec0SHong Zhang         PetscInt *idx;
1653beae5ec0SHong Zhang         /* idx: maps from k-th index of v_mpi to (i,j)-th global entry of B */
16549566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(nrhs * M, &idx));
16559566063dSJacob Faibussowitsch         PetscCall(MatGetOwnershipRanges(B, &rstart));
1656be818407SHong Zhang         k = 0;
16572d4298aeSJunchao Zhang         for (proc = 0; proc < mumps->petsc_size; proc++) {
1658be818407SHong Zhang           for (j = 0; j < nrhs; j++) {
1659beae5ec0SHong Zhang             for (i = rstart[proc]; i < rstart[proc + 1]; i++) idx[k++] = j * M + i;
1660be818407SHong Zhang           }
1661be818407SHong Zhang         }
1662be818407SHong Zhang 
16639566063dSJacob Faibussowitsch         PetscCall(VecCreateSeq(PETSC_COMM_SELF, nrhs * M, &b_seq));
16649566063dSJacob Faibussowitsch         PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nrhs * M, idx, PETSC_OWN_POINTER, &is_to));
16659566063dSJacob Faibussowitsch         PetscCall(ISCreateStride(PETSC_COMM_SELF, nrhs * M, 0, 1, &is_from));
1666801fbe65SHong Zhang       } else {
16679566063dSJacob Faibussowitsch         PetscCall(VecCreateSeq(PETSC_COMM_SELF, 0, &b_seq));
16689566063dSJacob Faibussowitsch         PetscCall(ISCreateStride(PETSC_COMM_SELF, 0, 0, 1, &is_to));
16699566063dSJacob Faibussowitsch         PetscCall(ISCreateStride(PETSC_COMM_SELF, 0, 0, 1, &is_from));
1670801fbe65SHong Zhang       }
16719566063dSJacob Faibussowitsch       PetscCall(VecScatterCreate(v_mpi, is_from, b_seq, is_to, &scat_rhs));
16729566063dSJacob Faibussowitsch       PetscCall(VecScatterBegin(scat_rhs, v_mpi, b_seq, INSERT_VALUES, SCATTER_FORWARD));
16739566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_to));
16749566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&is_from));
16759566063dSJacob Faibussowitsch       PetscCall(VecScatterEnd(scat_rhs, v_mpi, b_seq, INSERT_VALUES, SCATTER_FORWARD));
1676801fbe65SHong Zhang 
1677801fbe65SHong Zhang       if (!mumps->myid) { /* define rhs on the host */
16789566063dSJacob Faibussowitsch         PetscCall(VecGetArray(b_seq, &bray));
1679940cd9d6SSatish Balay         mumps->id.rhs = (MumpsScalar *)bray;
16809566063dSJacob Faibussowitsch         PetscCall(VecRestoreArray(b_seq, &bray));
1681801fbe65SHong Zhang       }
168225aac85cSJunchao Zhang     }
16832b691707SHong Zhang   } else { /* sparse B */
16842b691707SHong Zhang     b = (Mat_MPIAIJ *)Bt->data;
16852b691707SHong Zhang 
1686be818407SHong Zhang     /* wrap dense X into a vector v_mpi */
16879566063dSJacob Faibussowitsch     PetscCall(MatGetLocalSize(X, &m, NULL));
16889566063dSJacob Faibussowitsch     PetscCall(MatDenseGetArray(X, &bray));
16899566063dSJacob Faibussowitsch     PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)X), 1, nrhs * m, nrhs * M, (const PetscScalar *)bray, &v_mpi));
16909566063dSJacob Faibussowitsch     PetscCall(MatDenseRestoreArray(X, &bray));
16912b691707SHong Zhang 
16922b691707SHong Zhang     if (!mumps->myid) {
16939566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJGetArray(b->A, &aa));
16949566063dSJacob Faibussowitsch       PetscCall(MatGetRowIJ(b->A, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
169528b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
16969566063dSJacob Faibussowitsch       PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs));
16972b691707SHong Zhang       mumps->id.rhs_sparse = (MumpsScalar *)aa;
16982b691707SHong Zhang     } else {
16992b691707SHong Zhang       mumps->id.irhs_ptr    = NULL;
17002b691707SHong Zhang       mumps->id.irhs_sparse = NULL;
17012b691707SHong Zhang       mumps->id.nz_rhs      = 0;
17022b691707SHong Zhang       mumps->id.rhs_sparse  = NULL;
17032b691707SHong Zhang     }
17042b691707SHong Zhang   }
17052b691707SHong Zhang 
1706801fbe65SHong Zhang   /* solve phase */
1707801fbe65SHong Zhang   mumps->id.job = JOB_SOLVE;
17083ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
17099261f6e4SBarry 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));
1710801fbe65SHong Zhang 
1711334c5f61SHong Zhang   /* scatter mumps distributed solution to petsc vector v_mpi, which shares local arrays with solution matrix X */
17129566063dSJacob Faibussowitsch   PetscCall(MatDenseGetArray(X, &array));
17139566063dSJacob Faibussowitsch   PetscCall(VecPlaceArray(v_mpi, array));
1714801fbe65SHong Zhang 
1715334c5f61SHong Zhang   /* create scatter scat_sol */
17169566063dSJacob Faibussowitsch   PetscCall(MatGetOwnershipRanges(X, &rstart));
1717beae5ec0SHong Zhang   /* iidx: index for scatter mumps solution to petsc X */
1718beae5ec0SHong Zhang 
17199566063dSJacob Faibussowitsch   PetscCall(ISCreateStride(PETSC_COMM_SELF, nlsol_loc, 0, 1, &is_from));
17209566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(nlsol_loc, &idxx));
1721beae5ec0SHong Zhang   for (i = 0; i < lsol_loc; i++) {
1722beae5ec0SHong 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 */
1723beae5ec0SHong Zhang 
17242d4298aeSJunchao Zhang     for (proc = 0; proc < mumps->petsc_size; proc++) {
1725beae5ec0SHong Zhang       if (isol_loc[i] >= rstart[proc] && isol_loc[i] < rstart[proc + 1]) {
1726beae5ec0SHong Zhang         myrstart = rstart[proc];
1727beae5ec0SHong Zhang         k        = isol_loc[i] - myrstart;          /* local index on 1st column of petsc vector X */
1728beae5ec0SHong Zhang         iidx     = k + myrstart * nrhs;             /* maps mumps isol_loc[i] to petsc index in X */
1729beae5ec0SHong Zhang         m        = rstart[proc + 1] - rstart[proc]; /* rows of X for this proc */
1730beae5ec0SHong Zhang         break;
1731be818407SHong Zhang       }
1732be818407SHong Zhang     }
1733be818407SHong Zhang 
1734beae5ec0SHong Zhang     for (j = 0; j < nrhs; j++) idxx[i + j * lsol_loc] = iidx + j * m;
1735801fbe65SHong Zhang   }
17369566063dSJacob Faibussowitsch   PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nlsol_loc, idxx, PETSC_COPY_VALUES, &is_to));
17379566063dSJacob Faibussowitsch   PetscCall(VecScatterCreate(msol_loc, is_from, v_mpi, is_to, &scat_sol));
17389566063dSJacob Faibussowitsch   PetscCall(VecScatterBegin(scat_sol, msol_loc, v_mpi, INSERT_VALUES, SCATTER_FORWARD));
17399566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&is_from));
17409566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&is_to));
17419566063dSJacob Faibussowitsch   PetscCall(VecScatterEnd(scat_sol, msol_loc, v_mpi, INSERT_VALUES, SCATTER_FORWARD));
17429566063dSJacob Faibussowitsch   PetscCall(MatDenseRestoreArray(X, &array));
174371aed81dSHong Zhang 
174471aed81dSHong Zhang   /* free spaces */
17451683a169SBarry Smith   mumps->id.sol_loc  = (MumpsScalar *)sol_loc_save;
174671aed81dSHong Zhang   mumps->id.isol_loc = isol_loc_save;
174771aed81dSHong Zhang 
17489566063dSJacob Faibussowitsch   PetscCall(PetscFree2(sol_loc, isol_loc));
17499566063dSJacob Faibussowitsch   PetscCall(PetscFree(idxx));
17509566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&msol_loc));
17519566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&v_mpi));
1752a6053eceSJunchao Zhang   if (!denseB) {
17532b691707SHong Zhang     if (!mumps->myid) {
1754d56c302dSHong Zhang       b = (Mat_MPIAIJ *)Bt->data;
17559566063dSJacob Faibussowitsch       PetscCall(MatSeqAIJRestoreArray(b->A, &aa));
17569566063dSJacob Faibussowitsch       PetscCall(MatRestoreRowIJ(b->A, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
175728b400f6SJacob Faibussowitsch       PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot restore IJ structure");
17582b691707SHong Zhang     }
17592b691707SHong Zhang   } else {
176025aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) {
17619566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b_seq));
17629566063dSJacob Faibussowitsch       PetscCall(VecScatterDestroy(&scat_rhs));
176325aac85cSJunchao Zhang     }
17642b691707SHong Zhang   }
17659566063dSJacob Faibussowitsch   PetscCall(VecScatterDestroy(&scat_sol));
176664412097SPierre 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))));
17673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1768e0b74bf9SHong Zhang }
1769e0b74bf9SHong Zhang 
177066976f2fSJacob Faibussowitsch static PetscErrorCode MatMatSolveTranspose_MUMPS(Mat A, Mat B, Mat X)
1771d71ae5a4SJacob Faibussowitsch {
1772b18964edSHong Zhang   Mat_MUMPS          *mumps = (Mat_MUMPS *)A->data;
1773338d3105SPierre Jolivet   const PetscMUMPSInt value = mumps->id.ICNTL(9);
1774b18964edSHong Zhang 
1775b18964edSHong Zhang   PetscFunctionBegin;
1776b18964edSHong Zhang   mumps->id.ICNTL(9) = 0;
1777b18964edSHong Zhang   PetscCall(MatMatSolve_MUMPS(A, B, X));
1778338d3105SPierre Jolivet   mumps->id.ICNTL(9) = value;
17793ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1780b18964edSHong Zhang }
1781b18964edSHong Zhang 
178266976f2fSJacob Faibussowitsch static PetscErrorCode MatMatTransposeSolve_MUMPS(Mat A, Mat Bt, Mat X)
1783d71ae5a4SJacob Faibussowitsch {
1784eb3ef3b2SHong Zhang   PetscBool flg;
1785eb3ef3b2SHong Zhang   Mat       B;
1786eb3ef3b2SHong Zhang 
1787eb3ef3b2SHong Zhang   PetscFunctionBegin;
17889566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)Bt, &flg, MATSEQAIJ, MATMPIAIJ, NULL));
178928b400f6SJacob Faibussowitsch   PetscCheck(flg, PetscObjectComm((PetscObject)Bt), PETSC_ERR_ARG_WRONG, "Matrix Bt must be MATAIJ matrix");
1790eb3ef3b2SHong Zhang 
1791eb3ef3b2SHong Zhang   /* Create B=Bt^T that uses Bt's data structure */
17929566063dSJacob Faibussowitsch   PetscCall(MatCreateTranspose(Bt, &B));
1793eb3ef3b2SHong Zhang 
17949566063dSJacob Faibussowitsch   PetscCall(MatMatSolve_MUMPS(A, B, X));
17959566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&B));
17963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1797eb3ef3b2SHong Zhang }
1798eb3ef3b2SHong Zhang 
1799ace3df97SHong Zhang #if !defined(PETSC_USE_COMPLEX)
1800a58c3f20SHong Zhang /*
1801a58c3f20SHong Zhang   input:
1802a58c3f20SHong Zhang    F:        numeric factor
1803a58c3f20SHong Zhang   output:
1804a58c3f20SHong Zhang    nneg:     total number of negative pivots
180519d49a3bSHong Zhang    nzero:    total number of zero pivots
180619d49a3bSHong Zhang    npos:     (global dimension of F) - nneg - nzero
1807a58c3f20SHong Zhang */
180866976f2fSJacob Faibussowitsch static PetscErrorCode MatGetInertia_SBAIJMUMPS(Mat F, PetscInt *nneg, PetscInt *nzero, PetscInt *npos)
1809d71ae5a4SJacob Faibussowitsch {
1810e69c285eSBarry Smith   Mat_MUMPS  *mumps = (Mat_MUMPS *)F->data;
1811c1490034SHong Zhang   PetscMPIInt size;
1812a58c3f20SHong Zhang 
1813a58c3f20SHong Zhang   PetscFunctionBegin;
18149566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)F), &size));
1815bcb30aebSHong 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 */
1816aed4548fSBarry 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));
1817ed85ac9fSHong Zhang 
1818710ac8efSHong Zhang   if (nneg) *nneg = mumps->id.INFOG(12);
1819ed85ac9fSHong Zhang   if (nzero || npos) {
182008401ef6SPierre 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");
1821710ac8efSHong Zhang     if (nzero) *nzero = mumps->id.INFOG(28);
1822710ac8efSHong Zhang     if (npos) *npos = F->rmap->N - (mumps->id.INFOG(12) + mumps->id.INFOG(28));
1823a58c3f20SHong Zhang   }
18243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1825a58c3f20SHong Zhang }
182619d49a3bSHong Zhang #endif
1827a58c3f20SHong Zhang 
182866976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGatherNonzerosOnMaster(MatReuse reuse, Mat_MUMPS *mumps)
1829d71ae5a4SJacob Faibussowitsch {
1830a6053eceSJunchao Zhang   PetscInt       i, nreqs;
1831a6053eceSJunchao Zhang   PetscMUMPSInt *irn, *jcn;
1832a6053eceSJunchao Zhang   PetscMPIInt    count;
1833a6053eceSJunchao Zhang   PetscInt64     totnnz, remain;
1834a6053eceSJunchao Zhang   const PetscInt osize = mumps->omp_comm_size;
1835a6053eceSJunchao Zhang   PetscScalar   *val;
18363ab56b82SJunchao Zhang 
18373ab56b82SJunchao Zhang   PetscFunctionBegin;
1838a6053eceSJunchao Zhang   if (osize > 1) {
18393ab56b82SJunchao Zhang     if (reuse == MAT_INITIAL_MATRIX) {
18403ab56b82SJunchao Zhang       /* master first gathers counts of nonzeros to receive */
18419566063dSJacob Faibussowitsch       if (mumps->is_omp_master) PetscCall(PetscMalloc1(osize, &mumps->recvcount));
18429566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Gather(&mumps->nnz, 1, MPIU_INT64, mumps->recvcount, 1, MPIU_INT64, 0 /*master*/, mumps->omp_comm));
18433ab56b82SJunchao Zhang 
1844a6053eceSJunchao Zhang       /* Then each computes number of send/recvs */
18453ab56b82SJunchao Zhang       if (mumps->is_omp_master) {
1846a6053eceSJunchao Zhang         /* Start from 1 since self communication is not done in MPI */
1847a6053eceSJunchao Zhang         nreqs = 0;
1848a6053eceSJunchao Zhang         for (i = 1; i < osize; i++) nreqs += (mumps->recvcount[i] + PETSC_MPI_INT_MAX - 1) / PETSC_MPI_INT_MAX;
1849a6053eceSJunchao Zhang       } else {
1850a6053eceSJunchao Zhang         nreqs = (mumps->nnz + PETSC_MPI_INT_MAX - 1) / PETSC_MPI_INT_MAX;
18513ab56b82SJunchao Zhang       }
185235cb6cd3SPierre Jolivet       PetscCall(PetscMalloc1(nreqs * 3, &mumps->reqs)); /* Triple the requests since we send irn, jcn and val separately */
18533ab56b82SJunchao Zhang 
1854a6053eceSJunchao Zhang       /* The following code is doing a very simple thing: omp_master rank gathers irn/jcn/val from others.
1855a6053eceSJunchao Zhang          MPI_Gatherv would be enough if it supports big counts > 2^31-1. Since it does not, and mumps->nnz
1856a6053eceSJunchao Zhang          might be a prime number > 2^31-1, we have to slice the message. Note omp_comm_size
1857a6053eceSJunchao Zhang          is very small, the current approach should have no extra overhead compared to MPI_Gatherv.
1858a6053eceSJunchao Zhang        */
1859a6053eceSJunchao Zhang       nreqs = 0; /* counter for actual send/recvs */
18603ab56b82SJunchao Zhang       if (mumps->is_omp_master) {
1861a6053eceSJunchao Zhang         for (i = 0, totnnz = 0; i < osize; i++) totnnz += mumps->recvcount[i]; /* totnnz = sum of nnz over omp_comm */
18629566063dSJacob Faibussowitsch         PetscCall(PetscMalloc2(totnnz, &irn, totnnz, &jcn));
18639566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(totnnz, &val));
1864a6053eceSJunchao Zhang 
1865a6053eceSJunchao Zhang         /* Self communication */
18669566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(irn, mumps->irn, mumps->nnz));
18679566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(jcn, mumps->jcn, mumps->nnz));
18689566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(val, mumps->val, mumps->nnz));
1869a6053eceSJunchao Zhang 
1870a6053eceSJunchao Zhang         /* Replace mumps->irn/jcn etc on master with the newly allocated bigger arrays */
18719566063dSJacob Faibussowitsch         PetscCall(PetscFree2(mumps->irn, mumps->jcn));
18729566063dSJacob Faibussowitsch         PetscCall(PetscFree(mumps->val_alloc));
1873a6053eceSJunchao Zhang         mumps->nnz = totnnz;
18743ab56b82SJunchao Zhang         mumps->irn = irn;
18753ab56b82SJunchao Zhang         mumps->jcn = jcn;
1876a6053eceSJunchao Zhang         mumps->val = mumps->val_alloc = val;
1877a6053eceSJunchao Zhang 
1878a6053eceSJunchao Zhang         irn += mumps->recvcount[0]; /* recvcount[0] is old mumps->nnz on omp rank 0 */
1879a6053eceSJunchao Zhang         jcn += mumps->recvcount[0];
1880a6053eceSJunchao Zhang         val += mumps->recvcount[0];
1881a6053eceSJunchao Zhang 
1882a6053eceSJunchao Zhang         /* Remote communication */
1883a6053eceSJunchao Zhang         for (i = 1; i < osize; i++) {
1884a6053eceSJunchao Zhang           count  = PetscMin(mumps->recvcount[i], PETSC_MPI_INT_MAX);
1885a6053eceSJunchao Zhang           remain = mumps->recvcount[i] - count;
1886a6053eceSJunchao Zhang           while (count > 0) {
18879566063dSJacob Faibussowitsch             PetscCallMPI(MPI_Irecv(irn, count, MPIU_MUMPSINT, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
18889566063dSJacob Faibussowitsch             PetscCallMPI(MPI_Irecv(jcn, count, MPIU_MUMPSINT, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
18899566063dSJacob Faibussowitsch             PetscCallMPI(MPI_Irecv(val, count, MPIU_SCALAR, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1890a6053eceSJunchao Zhang             irn += count;
1891a6053eceSJunchao Zhang             jcn += count;
1892a6053eceSJunchao Zhang             val += count;
1893a6053eceSJunchao Zhang             count = PetscMin(remain, PETSC_MPI_INT_MAX);
1894a6053eceSJunchao Zhang             remain -= count;
1895a6053eceSJunchao Zhang           }
18963ab56b82SJunchao Zhang         }
18973ab56b82SJunchao Zhang       } else {
1898a6053eceSJunchao Zhang         irn    = mumps->irn;
1899a6053eceSJunchao Zhang         jcn    = mumps->jcn;
1900a6053eceSJunchao Zhang         val    = mumps->val;
1901a6053eceSJunchao Zhang         count  = PetscMin(mumps->nnz, PETSC_MPI_INT_MAX);
1902a6053eceSJunchao Zhang         remain = mumps->nnz - count;
1903a6053eceSJunchao Zhang         while (count > 0) {
19049566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Isend(irn, count, MPIU_MUMPSINT, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19059566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Isend(jcn, count, MPIU_MUMPSINT, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
19069566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Isend(val, count, MPIU_SCALAR, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1907a6053eceSJunchao Zhang           irn += count;
1908a6053eceSJunchao Zhang           jcn += count;
1909a6053eceSJunchao Zhang           val += count;
1910a6053eceSJunchao Zhang           count = PetscMin(remain, PETSC_MPI_INT_MAX);
1911a6053eceSJunchao Zhang           remain -= count;
19123ab56b82SJunchao Zhang         }
19133ab56b82SJunchao Zhang       }
1914a6053eceSJunchao Zhang     } else {
1915a6053eceSJunchao Zhang       nreqs = 0;
1916a6053eceSJunchao Zhang       if (mumps->is_omp_master) {
1917a6053eceSJunchao Zhang         val = mumps->val + mumps->recvcount[0];
1918a6053eceSJunchao Zhang         for (i = 1; i < osize; i++) { /* Remote communication only since self data is already in place */
1919a6053eceSJunchao Zhang           count  = PetscMin(mumps->recvcount[i], PETSC_MPI_INT_MAX);
1920a6053eceSJunchao Zhang           remain = mumps->recvcount[i] - count;
1921a6053eceSJunchao Zhang           while (count > 0) {
19229566063dSJacob Faibussowitsch             PetscCallMPI(MPI_Irecv(val, count, MPIU_SCALAR, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1923a6053eceSJunchao Zhang             val += count;
1924a6053eceSJunchao Zhang             count = PetscMin(remain, PETSC_MPI_INT_MAX);
1925a6053eceSJunchao Zhang             remain -= count;
1926a6053eceSJunchao Zhang           }
1927a6053eceSJunchao Zhang         }
1928a6053eceSJunchao Zhang       } else {
1929a6053eceSJunchao Zhang         val    = mumps->val;
1930a6053eceSJunchao Zhang         count  = PetscMin(mumps->nnz, PETSC_MPI_INT_MAX);
1931a6053eceSJunchao Zhang         remain = mumps->nnz - count;
1932a6053eceSJunchao Zhang         while (count > 0) {
19339566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Isend(val, count, MPIU_SCALAR, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++]));
1934a6053eceSJunchao Zhang           val += count;
1935a6053eceSJunchao Zhang           count = PetscMin(remain, PETSC_MPI_INT_MAX);
1936a6053eceSJunchao Zhang           remain -= count;
1937a6053eceSJunchao Zhang         }
1938a6053eceSJunchao Zhang       }
1939a6053eceSJunchao Zhang     }
19409566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Waitall(nreqs, mumps->reqs, MPI_STATUSES_IGNORE));
1941a6053eceSJunchao Zhang     mumps->tag++; /* It is totally fine for above send/recvs to share one mpi tag */
1942a6053eceSJunchao Zhang   }
19433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
19443ab56b82SJunchao Zhang }
19453ab56b82SJunchao Zhang 
194666976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorNumeric_MUMPS(Mat F, Mat A, const MatFactorInfo *info)
1947d71ae5a4SJacob Faibussowitsch {
1948e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)(F)->data;
1949ace3abfcSBarry Smith   PetscBool  isMPIAIJ;
1950397b6df1SKris Buschelman 
1951397b6df1SKris Buschelman   PetscFunctionBegin;
1952dbf6bb8dSprj-   if (mumps->id.INFOG(1) < 0 && !(mumps->id.INFOG(1) == -16 && mumps->id.INFOG(1) == 0)) {
195348a46eb9SPierre 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)));
19549566063dSJacob 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)));
19553ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
19562aca8efcSHong Zhang   }
19576baea169SHong Zhang 
19589566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_REUSE_MATRIX, mumps));
19599566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_REUSE_MATRIX, mumps));
1960397b6df1SKris Buschelman 
1961397b6df1SKris Buschelman   /* numerical factorization phase */
1962a5e57a09SHong Zhang   mumps->id.job = JOB_FACTNUMERIC;
19634e34a73bSHong Zhang   if (!mumps->id.ICNTL(18)) { /* A is centralized */
1964ad540459SPierre Jolivet     if (!mumps->myid) mumps->id.a = (MumpsScalar *)mumps->val;
1965397b6df1SKris Buschelman   } else {
1966940cd9d6SSatish Balay     mumps->id.a_loc = (MumpsScalar *)mumps->val;
1967397b6df1SKris Buschelman   }
19683ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
1969a5e57a09SHong Zhang   if (mumps->id.INFOG(1) < 0) {
19709261f6e4SBarry 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));
19719261f6e4SBarry Smith     if (mumps->id.INFOG(1) == -10) {
19729261f6e4SBarry 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)));
1973603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
1974c0d63f2fSHong Zhang     } else if (mumps->id.INFOG(1) == -13) {
19759261f6e4SBarry 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)));
1976603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OUTMEMORY;
1977c0d63f2fSHong Zhang     } else if (mumps->id.INFOG(1) == -8 || mumps->id.INFOG(1) == -9 || (-16 < mumps->id.INFOG(1) && mumps->id.INFOG(1) < -10)) {
19789261f6e4SBarry 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)));
1979603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OUTMEMORY;
19802aca8efcSHong Zhang     } else {
19819261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in numerical factorization: INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2)));
1982603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OTHER;
1983151787a6SHong Zhang     }
19842aca8efcSHong Zhang   }
19859261f6e4SBarry 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));
1986397b6df1SKris Buschelman 
1987b3cb21ddSStefano Zampini   F->assembled = PETSC_TRUE;
1988d47f36abSHong Zhang 
1989b3cb21ddSStefano Zampini   if (F->schur) { /* reset Schur status to unfactored */
19903cb7dd0eSStefano Zampini #if defined(PETSC_HAVE_CUDA)
1991c70f7ee4SJunchao Zhang     F->schur->offloadmask = PETSC_OFFLOAD_CPU;
19923cb7dd0eSStefano Zampini #endif
1993b3cb21ddSStefano Zampini     if (mumps->id.ICNTL(19) == 1) { /* stored by rows */
1994b3cb21ddSStefano Zampini       mumps->id.ICNTL(19) = 2;
19959566063dSJacob Faibussowitsch       PetscCall(MatTranspose(F->schur, MAT_INPLACE_MATRIX, &F->schur));
1996b3cb21ddSStefano Zampini     }
19979566063dSJacob Faibussowitsch     PetscCall(MatFactorRestoreSchurComplement(F, NULL, MAT_FACTOR_SCHUR_UNFACTORED));
1998b3cb21ddSStefano Zampini   }
199967877ebaSShri Abhyankar 
2000066565c5SStefano Zampini   /* just to be sure that ICNTL(19) value returned by a call from MatMumpsGetIcntl is always consistent */
2001066565c5SStefano Zampini   if (!mumps->sym && mumps->id.ICNTL(19) && mumps->id.ICNTL(19) != 1) mumps->id.ICNTL(19) = 3;
2002066565c5SStefano Zampini 
20033ab56b82SJunchao Zhang   if (!mumps->is_omp_master) mumps->id.INFO(23) = 0;
20042d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) {
200567877ebaSShri Abhyankar     PetscInt     lsol_loc;
200667877ebaSShri Abhyankar     PetscScalar *sol_loc;
20072205254eSKarl Rupp 
20089566063dSJacob Faibussowitsch     PetscCall(PetscObjectTypeCompare((PetscObject)A, MATMPIAIJ, &isMPIAIJ));
2009c2093ab7SHong Zhang 
2010c2093ab7SHong Zhang     /* distributed solution; Create x_seq=sol_loc for repeated use */
2011c2093ab7SHong Zhang     if (mumps->x_seq) {
20129566063dSJacob Faibussowitsch       PetscCall(VecScatterDestroy(&mumps->scat_sol));
20139566063dSJacob Faibussowitsch       PetscCall(PetscFree2(mumps->id.sol_loc, mumps->id.isol_loc));
20149566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mumps->x_seq));
2015c2093ab7SHong Zhang     }
2016a5e57a09SHong Zhang     lsol_loc = mumps->id.INFO(23); /* length of sol_loc */
20179566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(lsol_loc, &sol_loc, lsol_loc, &mumps->id.isol_loc));
2018a5e57a09SHong Zhang     mumps->id.lsol_loc = lsol_loc;
2019940cd9d6SSatish Balay     mumps->id.sol_loc  = (MumpsScalar *)sol_loc;
20209566063dSJacob Faibussowitsch     PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, lsol_loc, sol_loc, &mumps->x_seq));
202167877ebaSShri Abhyankar   }
20229566063dSJacob Faibussowitsch   PetscCall(PetscLogFlops(mumps->id.RINFO(2)));
20233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2024397b6df1SKris Buschelman }
2025397b6df1SKris Buschelman 
20269a2535b5SHong Zhang /* Sets MUMPS options from the options database */
202766976f2fSJacob Faibussowitsch static PetscErrorCode MatSetFromOptions_MUMPS(Mat F, Mat A)
2028d71ae5a4SJacob Faibussowitsch {
2029e69c285eSBarry Smith   Mat_MUMPS    *mumps = (Mat_MUMPS *)F->data;
2030413bcc21SPierre Jolivet   PetscMUMPSInt icntl = 0, size, *listvar_schur;
203145e3843bSPierre Jolivet   PetscInt      info[80], i, ninfo = 80, rbs, cbs;
2032413bcc21SPierre Jolivet   PetscBool     flg = PETSC_FALSE, schur = (PetscBool)(mumps->id.ICNTL(26) == -1);
2033413bcc21SPierre Jolivet   MumpsScalar  *arr;
2034dcd589f8SShri Abhyankar 
2035dcd589f8SShri Abhyankar   PetscFunctionBegin;
203626cc229bSBarry Smith   PetscOptionsBegin(PetscObjectComm((PetscObject)F), ((PetscObject)F)->prefix, "MUMPS Options", "Mat");
2037413bcc21SPierre Jolivet   if (mumps->id.job == JOB_NULL) { /* MatSetFromOptions_MUMPS() has never been called before */
2038413bcc21SPierre Jolivet     PetscInt nthreads   = 0;
2039413bcc21SPierre Jolivet     PetscInt nCNTL_pre  = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0;
2040413bcc21SPierre Jolivet     PetscInt nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0;
2041413bcc21SPierre Jolivet 
2042413bcc21SPierre Jolivet     mumps->petsc_comm = PetscObjectComm((PetscObject)A);
2043413bcc21SPierre Jolivet     PetscCallMPI(MPI_Comm_size(mumps->petsc_comm, &mumps->petsc_size));
2044413bcc21SPierre Jolivet     PetscCallMPI(MPI_Comm_rank(mumps->petsc_comm, &mumps->myid)); /* "if (!myid)" still works even if mumps_comm is different */
2045413bcc21SPierre Jolivet 
2046413bcc21SPierre Jolivet     PetscCall(PetscOptionsName("-mat_mumps_use_omp_threads", "Convert MPI processes into OpenMP threads", "None", &mumps->use_petsc_omp_support));
2047413bcc21SPierre Jolivet     if (mumps->use_petsc_omp_support) nthreads = -1; /* -1 will let PetscOmpCtrlCreate() guess a proper value when user did not supply one */
2048413bcc21SPierre Jolivet     /* do not use PetscOptionsInt() so that the option -mat_mumps_use_omp_threads is not displayed twice in the help */
2049413bcc21SPierre Jolivet     PetscCall(PetscOptionsGetInt(NULL, ((PetscObject)F)->prefix, "-mat_mumps_use_omp_threads", &nthreads, NULL));
2050413bcc21SPierre Jolivet     if (mumps->use_petsc_omp_support) {
20519371c9d4SSatish 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",
20529371c9d4SSatish Balay                  ((PetscObject)F)->prefix ? ((PetscObject)F)->prefix : "");
2053413bcc21SPierre 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 : "");
2054413bcc21SPierre Jolivet #if defined(PETSC_HAVE_OPENMP_SUPPORT)
2055413bcc21SPierre Jolivet       PetscCall(PetscOmpCtrlCreate(mumps->petsc_comm, nthreads, &mumps->omp_ctrl));
2056413bcc21SPierre Jolivet       PetscCall(PetscOmpCtrlGetOmpComms(mumps->omp_ctrl, &mumps->omp_comm, &mumps->mumps_comm, &mumps->is_omp_master));
2057413bcc21SPierre Jolivet #endif
2058413bcc21SPierre Jolivet     } else {
2059413bcc21SPierre Jolivet       mumps->omp_comm      = PETSC_COMM_SELF;
2060413bcc21SPierre Jolivet       mumps->mumps_comm    = mumps->petsc_comm;
2061413bcc21SPierre Jolivet       mumps->is_omp_master = PETSC_TRUE;
2062413bcc21SPierre Jolivet     }
2063413bcc21SPierre Jolivet     PetscCallMPI(MPI_Comm_size(mumps->omp_comm, &mumps->omp_comm_size));
2064413bcc21SPierre Jolivet     mumps->reqs = NULL;
2065413bcc21SPierre Jolivet     mumps->tag  = 0;
2066413bcc21SPierre Jolivet 
2067413bcc21SPierre Jolivet     if (mumps->mumps_comm != MPI_COMM_NULL) {
2068413bcc21SPierre Jolivet       if (PetscDefined(HAVE_OPENMP_SUPPORT) && mumps->use_petsc_omp_support) {
2069413bcc21SPierre Jolivet         /* It looks like MUMPS does not dup the input comm. Dup a new comm for MUMPS to avoid any tag mismatches. */
2070413bcc21SPierre Jolivet         MPI_Comm comm;
2071413bcc21SPierre Jolivet         PetscCallMPI(MPI_Comm_dup(mumps->mumps_comm, &comm));
2072413bcc21SPierre Jolivet         mumps->mumps_comm = comm;
2073413bcc21SPierre Jolivet       } else PetscCall(PetscCommGetComm(mumps->petsc_comm, &mumps->mumps_comm));
2074413bcc21SPierre Jolivet     }
2075413bcc21SPierre Jolivet 
2076413bcc21SPierre Jolivet     mumps->id.comm_fortran = MPI_Comm_c2f(mumps->mumps_comm);
2077413bcc21SPierre Jolivet     mumps->id.job          = JOB_INIT;
2078413bcc21SPierre Jolivet     mumps->id.par          = 1; /* host participates factorizaton and solve */
2079413bcc21SPierre Jolivet     mumps->id.sym          = mumps->sym;
2080413bcc21SPierre Jolivet 
2081413bcc21SPierre Jolivet     size          = mumps->id.size_schur;
2082413bcc21SPierre Jolivet     arr           = mumps->id.schur;
2083413bcc21SPierre Jolivet     listvar_schur = mumps->id.listvar_schur;
2084413bcc21SPierre Jolivet     PetscMUMPS_c(mumps);
20859261f6e4SBarry Smith     PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
2086413bcc21SPierre Jolivet     /* restore cached ICNTL and CNTL values */
2087413bcc21SPierre Jolivet     for (icntl = 0; icntl < nICNTL_pre; ++icntl) mumps->id.ICNTL(mumps->ICNTL_pre[1 + 2 * icntl]) = mumps->ICNTL_pre[2 + 2 * icntl];
2088413bcc21SPierre Jolivet     for (icntl = 0; icntl < nCNTL_pre; ++icntl) mumps->id.CNTL((PetscInt)mumps->CNTL_pre[1 + 2 * icntl]) = mumps->CNTL_pre[2 + 2 * icntl];
2089413bcc21SPierre Jolivet     PetscCall(PetscFree(mumps->ICNTL_pre));
2090413bcc21SPierre Jolivet     PetscCall(PetscFree(mumps->CNTL_pre));
2091413bcc21SPierre Jolivet 
2092413bcc21SPierre Jolivet     if (schur) {
2093413bcc21SPierre Jolivet       mumps->id.size_schur    = size;
2094413bcc21SPierre Jolivet       mumps->id.schur_lld     = size;
2095413bcc21SPierre Jolivet       mumps->id.schur         = arr;
2096413bcc21SPierre Jolivet       mumps->id.listvar_schur = listvar_schur;
2097413bcc21SPierre Jolivet       if (mumps->petsc_size > 1) {
2098413bcc21SPierre Jolivet         PetscBool gs; /* gs is false if any rank other than root has non-empty IS */
2099413bcc21SPierre Jolivet 
2100413bcc21SPierre Jolivet         mumps->id.ICNTL(19) = 1;                                                                            /* MUMPS returns Schur centralized on the host */
2101413bcc21SPierre 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 */
2102712fec58SPierre Jolivet         PetscCall(MPIU_Allreduce(MPI_IN_PLACE, &gs, 1, MPIU_BOOL, MPI_LAND, mumps->petsc_comm));
2103413bcc21SPierre Jolivet         PetscCheck(gs, PETSC_COMM_SELF, PETSC_ERR_SUP, "MUMPS distributed parallel Schur complements not yet supported from PETSc");
2104413bcc21SPierre Jolivet       } else {
2105413bcc21SPierre Jolivet         if (F->factortype == MAT_FACTOR_LU) {
2106413bcc21SPierre Jolivet           mumps->id.ICNTL(19) = 3; /* MUMPS returns full matrix */
2107413bcc21SPierre Jolivet         } else {
2108413bcc21SPierre Jolivet           mumps->id.ICNTL(19) = 2; /* MUMPS returns lower triangular part */
2109413bcc21SPierre Jolivet         }
2110413bcc21SPierre Jolivet       }
2111413bcc21SPierre Jolivet       mumps->id.ICNTL(26) = -1;
2112413bcc21SPierre Jolivet     }
2113413bcc21SPierre Jolivet 
2114413bcc21SPierre Jolivet     /* copy MUMPS default control values from master to slaves. Although slaves do not call MUMPS, they may access these values in code.
2115413bcc21SPierre Jolivet        For example, ICNTL(9) is initialized to 1 by MUMPS and slaves check ICNTL(9) in MatSolve_MUMPS.
2116413bcc21SPierre Jolivet      */
2117413bcc21SPierre Jolivet     PetscCallMPI(MPI_Bcast(mumps->id.icntl, 40, MPI_INT, 0, mumps->omp_comm));
2118413bcc21SPierre Jolivet     PetscCallMPI(MPI_Bcast(mumps->id.cntl, 15, MPIU_REAL, 0, mumps->omp_comm));
2119413bcc21SPierre Jolivet 
2120413bcc21SPierre Jolivet     mumps->scat_rhs = NULL;
2121413bcc21SPierre Jolivet     mumps->scat_sol = NULL;
2122413bcc21SPierre Jolivet 
2123413bcc21SPierre Jolivet     /* set PETSc-MUMPS default options - override MUMPS default */
2124413bcc21SPierre Jolivet     mumps->id.ICNTL(3) = 0;
2125413bcc21SPierre Jolivet     mumps->id.ICNTL(4) = 0;
2126413bcc21SPierre Jolivet     if (mumps->petsc_size == 1) {
2127413bcc21SPierre Jolivet       mumps->id.ICNTL(18) = 0; /* centralized assembled matrix input */
2128413bcc21SPierre Jolivet       mumps->id.ICNTL(7)  = 7; /* automatic choice of ordering done by the package */
2129413bcc21SPierre Jolivet     } else {
2130413bcc21SPierre Jolivet       mumps->id.ICNTL(18) = 3; /* distributed assembled matrix input */
2131413bcc21SPierre Jolivet       mumps->id.ICNTL(21) = 1; /* distributed solution */
2132413bcc21SPierre Jolivet     }
2133413bcc21SPierre Jolivet   }
21349566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_1", "ICNTL(1): output stream for error messages", "None", mumps->id.ICNTL(1), &icntl, &flg));
21359a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(1) = icntl;
21369566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_2", "ICNTL(2): output stream for diagnostic printing, statistics, and warning", "None", mumps->id.ICNTL(2), &icntl, &flg));
21379a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(2) = icntl;
21389566063dSJacob 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));
21399a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(3) = icntl;
2140dcd589f8SShri Abhyankar 
21419566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_4", "ICNTL(4): level of printing (0 to 4)", "None", mumps->id.ICNTL(4), &icntl, &flg));
21429a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(4) = icntl;
21439a2535b5SHong Zhang   if (mumps->id.ICNTL(4) || PetscLogPrintInfo) mumps->id.ICNTL(3) = 6; /* resume MUMPS default id.ICNTL(3) = 6 */
21449a2535b5SHong Zhang 
21459566063dSJacob 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));
21469a2535b5SHong Zhang   if (flg) mumps->id.ICNTL(6) = icntl;
21479a2535b5SHong Zhang 
21489566063dSJacob 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));
2149dcd589f8SShri Abhyankar   if (flg) {
2150aed4548fSBarry 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");
2151b53c1a7fSBarry Smith     mumps->id.ICNTL(7) = icntl;
2152dcd589f8SShri Abhyankar   }
2153e0b74bf9SHong Zhang 
21549566063dSJacob 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));
21559566063dSJacob 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() */
21569566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_10", "ICNTL(10): max num of refinements", "None", mumps->id.ICNTL(10), &mumps->id.ICNTL(10), NULL));
21579566063dSJacob 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));
21589566063dSJacob 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));
21599566063dSJacob 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));
21609566063dSJacob 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));
216145e3843bSPierre Jolivet   PetscCall(MatGetBlockSizes(A, &rbs, &cbs));
216245e3843bSPierre Jolivet   if (rbs == cbs && rbs > 1) mumps->id.ICNTL(15) = -rbs;
216345e3843bSPierre 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));
216445e3843bSPierre Jolivet   if (flg) {
216545e3843bSPierre Jolivet     PetscCheck(mumps->id.ICNTL(15) <= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Positive -mat_mumps_icntl_15 not handled");
216645e3843bSPierre 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");
216745e3843bSPierre Jolivet   }
21689566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_19", "ICNTL(19): computes the Schur complement", "None", mumps->id.ICNTL(19), &mumps->id.ICNTL(19), NULL));
216959ac8732SStefano Zampini   if (mumps->id.ICNTL(19) <= 0 || mumps->id.ICNTL(19) > 3) { /* reset any schur data (if any) */
21709566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&F->schur));
21719566063dSJacob Faibussowitsch     PetscCall(MatMumpsResetSchur_Private(mumps));
217259ac8732SStefano Zampini   }
217325aac85cSJunchao Zhang 
217443f3b051SJunchao Zhang   /* Two MPICH Fortran MPI_IN_PLACE binding bugs prevented the use of 'mpich + mumps'. One happened with "mpi4py + mpich + mumps",
217543f3b051SJunchao Zhang      and was reported by Firedrake. See https://bitbucket.org/mpi4py/mpi4py/issues/162/mpi4py-initialization-breaks-fortran
217625aac85cSJunchao Zhang      and a petsc-maint mailing list thread with subject 'MUMPS segfaults in parallel because of ...'
217743f3b051SJunchao Zhang      This bug was fixed by https://github.com/pmodels/mpich/pull/4149. But the fix brought a new bug,
217843f3b051SJunchao Zhang      see https://github.com/pmodels/mpich/issues/5589. This bug was fixed by https://github.com/pmodels/mpich/pull/5590.
217943f3b051SJunchao Zhang      In short, we could not use distributed RHS with MPICH until v4.0b1.
218025aac85cSJunchao Zhang    */
218143f3b051SJunchao Zhang #if PETSC_PKG_MUMPS_VERSION_LT(5, 3, 0) || (defined(PETSC_HAVE_MPICH_NUMVERSION) && (PETSC_HAVE_MPICH_NUMVERSION < 40000101))
218225aac85cSJunchao Zhang   mumps->ICNTL20 = 0; /* Centralized dense RHS*/
218343f3b051SJunchao Zhang #else
218443f3b051SJunchao Zhang   mumps->ICNTL20     = 10; /* Distributed dense RHS*/
218525aac85cSJunchao Zhang #endif
21869566063dSJacob 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));
2187aed4548fSBarry 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);
218825aac85cSJunchao Zhang #if PETSC_PKG_MUMPS_VERSION_LT(5, 3, 0)
2189aed4548fSBarry Smith   PetscCheck(!flg || mumps->ICNTL20 != 10, PETSC_COMM_SELF, PETSC_ERR_SUP, "ICNTL(20)=10 is not supported before MUMPS-5.3.0");
219025aac85cSJunchao Zhang #endif
21919566063dSJacob 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 */
21929a2535b5SHong Zhang 
21939566063dSJacob 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));
21949566063dSJacob 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));
21959566063dSJacob 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));
21969371c9d4SSatish Balay   if (mumps->id.ICNTL(24)) { mumps->id.ICNTL(13) = 1; /* turn-off ScaLAPACK to help with the correct detection of null pivots */ }
2197d7ebd59bSHong Zhang 
21989566063dSJacob 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));
21999566063dSJacob 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));
22009566063dSJacob 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));
22019566063dSJacob 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));
22029566063dSJacob 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));
22039566063dSJacob 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 */
22049566063dSJacob 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));
2205145b44c9SPierre 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 */
22069566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_33", "ICNTL(33): compute determinant", "None", mumps->id.ICNTL(33), &mumps->id.ICNTL(33), NULL));
22079566063dSJacob 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));
22089566063dSJacob Faibussowitsch   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_36", "ICNTL(36): choice of BLR factorization variant", "None", mumps->id.ICNTL(36), &mumps->id.ICNTL(36), NULL));
22099566063dSJacob 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));
2210146931dbSPierre Jolivet   PetscCall(PetscOptionsMUMPSInt("-mat_mumps_icntl_58", "ICNTL(58): defines options for symbolic factorization", "None", mumps->id.ICNTL(58), &mumps->id.ICNTL(58), NULL));
2211dcd589f8SShri Abhyankar 
22129566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_1", "CNTL(1): relative pivoting threshold", "None", mumps->id.CNTL(1), &mumps->id.CNTL(1), NULL));
22139566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_2", "CNTL(2): stopping criterion of refinement", "None", mumps->id.CNTL(2), &mumps->id.CNTL(2), NULL));
22149566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_3", "CNTL(3): absolute pivoting threshold", "None", mumps->id.CNTL(3), &mumps->id.CNTL(3), NULL));
22159566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_4", "CNTL(4): value for static pivoting", "None", mumps->id.CNTL(4), &mumps->id.CNTL(4), NULL));
22169566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_5", "CNTL(5): fixation for null pivots", "None", mumps->id.CNTL(5), &mumps->id.CNTL(5), NULL));
22179566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-mat_mumps_cntl_7", "CNTL(7): dropping parameter used during BLR", "None", mumps->id.CNTL(7), &mumps->id.CNTL(7), NULL));
2218e5bb22a1SHong Zhang 
22199566063dSJacob 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));
2220b34f08ffSHong Zhang 
22219566063dSJacob Faibussowitsch   PetscCall(PetscOptionsIntArray("-mat_mumps_view_info", "request INFO local to each processor", "", info, &ninfo, NULL));
2222b34f08ffSHong Zhang   if (ninfo) {
222308401ef6SPierre Jolivet     PetscCheck(ninfo <= 80, PETSC_COMM_SELF, PETSC_ERR_USER, "number of INFO %" PetscInt_FMT " must <= 80", ninfo);
22249566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(ninfo, &mumps->info));
2225b34f08ffSHong Zhang     mumps->ninfo = ninfo;
2226b34f08ffSHong Zhang     for (i = 0; i < ninfo; i++) {
2227aed4548fSBarry 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);
2228f7d195e4SLawrence Mitchell       mumps->info[i] = info[i];
2229b34f08ffSHong Zhang     }
2230b34f08ffSHong Zhang   }
2231d0609cedSBarry Smith   PetscOptionsEnd();
22323ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2233dcd589f8SShri Abhyankar }
2234dcd589f8SShri Abhyankar 
223566976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorSymbolic_MUMPS_ReportIfError(Mat F, Mat A, const MatFactorInfo *info, Mat_MUMPS *mumps)
2236d71ae5a4SJacob Faibussowitsch {
22375cd7cf9dSHong Zhang   PetscFunctionBegin;
22385cd7cf9dSHong Zhang   if (mumps->id.INFOG(1) < 0) {
22399261f6e4SBarry Smith     PetscCheck(!A->erroriffailure, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error in analysis: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1));
22405cd7cf9dSHong Zhang     if (mumps->id.INFOG(1) == -6) {
22419261f6e4SBarry 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)));
2242603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_STRUCT_ZEROPIVOT;
22435cd7cf9dSHong Zhang     } else if (mumps->id.INFOG(1) == -5 || mumps->id.INFOG(1) == -7) {
22449261f6e4SBarry 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)));
2245603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OUTMEMORY;
2246dbf6bb8dSprj-     } else if (mumps->id.INFOG(1) == -16 && mumps->id.INFOG(1) == 0) {
22479261f6e4SBarry Smith       PetscCall(PetscInfo(F, "MUMPS error in analysis: empty matrix\n"));
22485cd7cf9dSHong Zhang     } else {
22499261f6e4SBarry 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)));
2250603e8f96SBarry Smith       F->factorerrortype = MAT_FACTOR_OTHER;
22515cd7cf9dSHong Zhang     }
22525cd7cf9dSHong Zhang   }
22533ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
22545cd7cf9dSHong Zhang }
22555cd7cf9dSHong Zhang 
225666976f2fSJacob Faibussowitsch static PetscErrorCode MatLUFactorSymbolic_AIJMUMPS(Mat F, Mat A, IS r, IS c, const MatFactorInfo *info)
2257d71ae5a4SJacob Faibussowitsch {
2258e69c285eSBarry Smith   Mat_MUMPS     *mumps = (Mat_MUMPS *)F->data;
225967877ebaSShri Abhyankar   Vec            b;
226067877ebaSShri Abhyankar   const PetscInt M = A->rmap->N;
2261397b6df1SKris Buschelman 
2262397b6df1SKris Buschelman   PetscFunctionBegin;
2263d47f36abSHong Zhang   if (mumps->matstruc == SAME_NONZERO_PATTERN) {
2264d47f36abSHong Zhang     /* F is assembled by a previous call of MatLUFactorSymbolic_AIJMUMPS() */
22653ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2266d47f36abSHong Zhang   }
2267dcd589f8SShri Abhyankar 
22689a2535b5SHong Zhang   /* Set MUMPS options from the options database */
226926cc229bSBarry Smith   PetscCall(MatSetFromOptions_MUMPS(F, A));
2270dcd589f8SShri Abhyankar 
22719566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_INITIAL_MATRIX, mumps));
22729566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_INITIAL_MATRIX, mumps));
2273dcd589f8SShri Abhyankar 
227467877ebaSShri Abhyankar   /* analysis phase */
2275a5e57a09SHong Zhang   mumps->id.job = JOB_FACTSYMBOLIC;
2276a5e57a09SHong Zhang   mumps->id.n   = M;
2277a5e57a09SHong Zhang   switch (mumps->id.ICNTL(18)) {
227867877ebaSShri Abhyankar   case 0: /* centralized assembled matrix input */
2279a5e57a09SHong Zhang     if (!mumps->myid) {
2280a6053eceSJunchao Zhang       mumps->id.nnz = mumps->nnz;
2281a6053eceSJunchao Zhang       mumps->id.irn = mumps->irn;
2282a6053eceSJunchao Zhang       mumps->id.jcn = mumps->jcn;
2283a6053eceSJunchao Zhang       if (mumps->id.ICNTL(6) > 1) mumps->id.a = (MumpsScalar *)mumps->val;
22844ac6704cSBarry Smith       if (r) {
22854ac6704cSBarry Smith         mumps->id.ICNTL(7) = 1;
2286a5e57a09SHong Zhang         if (!mumps->myid) {
2287e0b74bf9SHong Zhang           const PetscInt *idx;
2288a6053eceSJunchao Zhang           PetscInt        i;
22892205254eSKarl Rupp 
22909566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(M, &mumps->id.perm_in));
22919566063dSJacob Faibussowitsch           PetscCall(ISGetIndices(r, &idx));
22929566063dSJacob Faibussowitsch           for (i = 0; i < M; i++) PetscCall(PetscMUMPSIntCast(idx[i] + 1, &(mumps->id.perm_in[i]))); /* perm_in[]: start from 1, not 0! */
22939566063dSJacob Faibussowitsch           PetscCall(ISRestoreIndices(r, &idx));
2294e0b74bf9SHong Zhang         }
2295e0b74bf9SHong Zhang       }
229667877ebaSShri Abhyankar     }
229767877ebaSShri Abhyankar     break;
229867877ebaSShri Abhyankar   case 3: /* distributed assembled matrix input (size>1) */
2299a6053eceSJunchao Zhang     mumps->id.nnz_loc = mumps->nnz;
2300a6053eceSJunchao Zhang     mumps->id.irn_loc = mumps->irn;
2301a6053eceSJunchao Zhang     mumps->id.jcn_loc = mumps->jcn;
2302a6053eceSJunchao Zhang     if (mumps->id.ICNTL(6) > 1) mumps->id.a_loc = (MumpsScalar *)mumps->val;
230325aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) { /* Centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */
23049566063dSJacob Faibussowitsch       PetscCall(MatCreateVecs(A, NULL, &b));
23059566063dSJacob Faibussowitsch       PetscCall(VecScatterCreateToZero(b, &mumps->scat_rhs, &mumps->b_seq));
23069566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b));
230725aac85cSJunchao Zhang     }
230867877ebaSShri Abhyankar     break;
230967877ebaSShri Abhyankar   }
23103ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
23119566063dSJacob Faibussowitsch   PetscCall(MatFactorSymbolic_MUMPS_ReportIfError(F, A, info, mumps));
231267877ebaSShri Abhyankar 
2313719d5645SBarry Smith   F->ops->lufactornumeric   = MatFactorNumeric_MUMPS;
2314dcd589f8SShri Abhyankar   F->ops->solve             = MatSolve_MUMPS;
231551d5961aSHong Zhang   F->ops->solvetranspose    = MatSolveTranspose_MUMPS;
23164e34a73bSHong Zhang   F->ops->matsolve          = MatMatSolve_MUMPS;
2317eb3ef3b2SHong Zhang   F->ops->mattransposesolve = MatMatTransposeSolve_MUMPS;
2318b18964edSHong Zhang   F->ops->matsolvetranspose = MatMatSolveTranspose_MUMPS;
2319d47f36abSHong Zhang 
2320d47f36abSHong Zhang   mumps->matstruc = SAME_NONZERO_PATTERN;
23213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2322b24902e0SBarry Smith }
2323b24902e0SBarry Smith 
2324450b117fSShri Abhyankar /* Note the Petsc r and c permutations are ignored */
232566976f2fSJacob Faibussowitsch static PetscErrorCode MatLUFactorSymbolic_BAIJMUMPS(Mat F, Mat A, IS r, IS c, const MatFactorInfo *info)
2326d71ae5a4SJacob Faibussowitsch {
2327e69c285eSBarry Smith   Mat_MUMPS     *mumps = (Mat_MUMPS *)F->data;
232867877ebaSShri Abhyankar   Vec            b;
232967877ebaSShri Abhyankar   const PetscInt M = A->rmap->N;
2330450b117fSShri Abhyankar 
2331450b117fSShri Abhyankar   PetscFunctionBegin;
2332d47f36abSHong Zhang   if (mumps->matstruc == SAME_NONZERO_PATTERN) {
2333338d3105SPierre Jolivet     /* F is assembled by a previous call of MatLUFactorSymbolic_BAIJMUMPS() */
23343ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2335d47f36abSHong Zhang   }
2336dcd589f8SShri Abhyankar 
23379a2535b5SHong Zhang   /* Set MUMPS options from the options database */
233826cc229bSBarry Smith   PetscCall(MatSetFromOptions_MUMPS(F, A));
2339dcd589f8SShri Abhyankar 
23409566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_INITIAL_MATRIX, mumps));
23419566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_INITIAL_MATRIX, mumps));
234267877ebaSShri Abhyankar 
234367877ebaSShri Abhyankar   /* analysis phase */
2344a5e57a09SHong Zhang   mumps->id.job = JOB_FACTSYMBOLIC;
2345a5e57a09SHong Zhang   mumps->id.n   = M;
2346a5e57a09SHong Zhang   switch (mumps->id.ICNTL(18)) {
234767877ebaSShri Abhyankar   case 0: /* centralized assembled matrix input */
2348a5e57a09SHong Zhang     if (!mumps->myid) {
2349a6053eceSJunchao Zhang       mumps->id.nnz = mumps->nnz;
2350a6053eceSJunchao Zhang       mumps->id.irn = mumps->irn;
2351a6053eceSJunchao Zhang       mumps->id.jcn = mumps->jcn;
2352ad540459SPierre Jolivet       if (mumps->id.ICNTL(6) > 1) mumps->id.a = (MumpsScalar *)mumps->val;
235367877ebaSShri Abhyankar     }
235467877ebaSShri Abhyankar     break;
235567877ebaSShri Abhyankar   case 3: /* distributed assembled matrix input (size>1) */
2356a6053eceSJunchao Zhang     mumps->id.nnz_loc = mumps->nnz;
2357a6053eceSJunchao Zhang     mumps->id.irn_loc = mumps->irn;
2358a6053eceSJunchao Zhang     mumps->id.jcn_loc = mumps->jcn;
2359ad540459SPierre Jolivet     if (mumps->id.ICNTL(6) > 1) mumps->id.a_loc = (MumpsScalar *)mumps->val;
236025aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) { /* Centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */
23619566063dSJacob Faibussowitsch       PetscCall(MatCreateVecs(A, NULL, &b));
23629566063dSJacob Faibussowitsch       PetscCall(VecScatterCreateToZero(b, &mumps->scat_rhs, &mumps->b_seq));
23639566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b));
236425aac85cSJunchao Zhang     }
236567877ebaSShri Abhyankar     break;
236667877ebaSShri Abhyankar   }
23673ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
23689566063dSJacob Faibussowitsch   PetscCall(MatFactorSymbolic_MUMPS_ReportIfError(F, A, info, mumps));
236967877ebaSShri Abhyankar 
2370450b117fSShri Abhyankar   F->ops->lufactornumeric   = MatFactorNumeric_MUMPS;
2371dcd589f8SShri Abhyankar   F->ops->solve             = MatSolve_MUMPS;
237251d5961aSHong Zhang   F->ops->solvetranspose    = MatSolveTranspose_MUMPS;
2373b18964edSHong Zhang   F->ops->matsolvetranspose = MatMatSolveTranspose_MUMPS;
2374d47f36abSHong Zhang 
2375d47f36abSHong Zhang   mumps->matstruc = SAME_NONZERO_PATTERN;
23763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2377450b117fSShri Abhyankar }
2378b24902e0SBarry Smith 
2379141f4205SHong Zhang /* Note the Petsc r permutation and factor info are ignored */
238066976f2fSJacob Faibussowitsch static PetscErrorCode MatCholeskyFactorSymbolic_MUMPS(Mat F, Mat A, IS r, const MatFactorInfo *info)
2381d71ae5a4SJacob Faibussowitsch {
2382e69c285eSBarry Smith   Mat_MUMPS     *mumps = (Mat_MUMPS *)F->data;
238367877ebaSShri Abhyankar   Vec            b;
238467877ebaSShri Abhyankar   const PetscInt M = A->rmap->N;
2385397b6df1SKris Buschelman 
2386397b6df1SKris Buschelman   PetscFunctionBegin;
2387d47f36abSHong Zhang   if (mumps->matstruc == SAME_NONZERO_PATTERN) {
2388338d3105SPierre Jolivet     /* F is assembled by a previous call of MatCholeskyFactorSymbolic_MUMPS() */
23893ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2390d47f36abSHong Zhang   }
2391dcd589f8SShri Abhyankar 
23929a2535b5SHong Zhang   /* Set MUMPS options from the options database */
239326cc229bSBarry Smith   PetscCall(MatSetFromOptions_MUMPS(F, A));
2394dcd589f8SShri Abhyankar 
23959566063dSJacob Faibussowitsch   PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_INITIAL_MATRIX, mumps));
23969566063dSJacob Faibussowitsch   PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_INITIAL_MATRIX, mumps));
2397dcd589f8SShri Abhyankar 
239867877ebaSShri Abhyankar   /* analysis phase */
2399a5e57a09SHong Zhang   mumps->id.job = JOB_FACTSYMBOLIC;
2400a5e57a09SHong Zhang   mumps->id.n   = M;
2401a5e57a09SHong Zhang   switch (mumps->id.ICNTL(18)) {
240267877ebaSShri Abhyankar   case 0: /* centralized assembled matrix input */
2403a5e57a09SHong Zhang     if (!mumps->myid) {
2404a6053eceSJunchao Zhang       mumps->id.nnz = mumps->nnz;
2405a6053eceSJunchao Zhang       mumps->id.irn = mumps->irn;
2406a6053eceSJunchao Zhang       mumps->id.jcn = mumps->jcn;
2407ad540459SPierre Jolivet       if (mumps->id.ICNTL(6) > 1) mumps->id.a = (MumpsScalar *)mumps->val;
240867877ebaSShri Abhyankar     }
240967877ebaSShri Abhyankar     break;
241067877ebaSShri Abhyankar   case 3: /* distributed assembled matrix input (size>1) */
2411a6053eceSJunchao Zhang     mumps->id.nnz_loc = mumps->nnz;
2412a6053eceSJunchao Zhang     mumps->id.irn_loc = mumps->irn;
2413a6053eceSJunchao Zhang     mumps->id.jcn_loc = mumps->jcn;
2414ad540459SPierre Jolivet     if (mumps->id.ICNTL(6) > 1) mumps->id.a_loc = (MumpsScalar *)mumps->val;
241525aac85cSJunchao Zhang     if (mumps->ICNTL20 == 0) { /* Centralized rhs. Create scatter scat_rhs for repeated use in MatSolve() */
24169566063dSJacob Faibussowitsch       PetscCall(MatCreateVecs(A, NULL, &b));
24179566063dSJacob Faibussowitsch       PetscCall(VecScatterCreateToZero(b, &mumps->scat_rhs, &mumps->b_seq));
24189566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&b));
241925aac85cSJunchao Zhang     }
242067877ebaSShri Abhyankar     break;
242167877ebaSShri Abhyankar   }
24223ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
24239566063dSJacob Faibussowitsch   PetscCall(MatFactorSymbolic_MUMPS_ReportIfError(F, A, info, mumps));
24245cd7cf9dSHong Zhang 
24252792810eSHong Zhang   F->ops->choleskyfactornumeric = MatFactorNumeric_MUMPS;
2426dcd589f8SShri Abhyankar   F->ops->solve                 = MatSolve_MUMPS;
242751d5961aSHong Zhang   F->ops->solvetranspose        = MatSolve_MUMPS;
24284e34a73bSHong Zhang   F->ops->matsolve              = MatMatSolve_MUMPS;
242923a5080aSHong Zhang   F->ops->mattransposesolve     = MatMatTransposeSolve_MUMPS;
2430b18964edSHong Zhang   F->ops->matsolvetranspose     = MatMatSolveTranspose_MUMPS;
24314e34a73bSHong Zhang #if defined(PETSC_USE_COMPLEX)
24320298fd71SBarry Smith   F->ops->getinertia = NULL;
24334e34a73bSHong Zhang #else
24344e34a73bSHong Zhang   F->ops->getinertia = MatGetInertia_SBAIJMUMPS;
2435db4efbfdSBarry Smith #endif
2436d47f36abSHong Zhang 
2437d47f36abSHong Zhang   mumps->matstruc = SAME_NONZERO_PATTERN;
24383ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2439b24902e0SBarry Smith }
2440b24902e0SBarry Smith 
244166976f2fSJacob Faibussowitsch static PetscErrorCode MatView_MUMPS(Mat A, PetscViewer viewer)
2442d71ae5a4SJacob Faibussowitsch {
244364e6c443SBarry Smith   PetscBool         iascii;
244464e6c443SBarry Smith   PetscViewerFormat format;
2445e69c285eSBarry Smith   Mat_MUMPS        *mumps = (Mat_MUMPS *)A->data;
2446f6c57405SHong Zhang 
2447f6c57405SHong Zhang   PetscFunctionBegin;
244864e6c443SBarry Smith   /* check if matrix is mumps type */
24493ba16761SJacob Faibussowitsch   if (A->ops->solve != MatSolve_MUMPS) PetscFunctionReturn(PETSC_SUCCESS);
245064e6c443SBarry Smith 
24519566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
245264e6c443SBarry Smith   if (iascii) {
24539566063dSJacob Faibussowitsch     PetscCall(PetscViewerGetFormat(viewer, &format));
24541511cd71SPierre Jolivet     if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
24559566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "MUMPS run parameters:\n"));
24561511cd71SPierre Jolivet       if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
24579566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  SYM (matrix type):                   %d\n", mumps->id.sym));
24589566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  PAR (host participation):            %d\n", mumps->id.par));
24599566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(1) (output for error):         %d\n", mumps->id.ICNTL(1)));
24609566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(2) (output of diagnostic msg): %d\n", mumps->id.ICNTL(2)));
24619566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(3) (output for global info):   %d\n", mumps->id.ICNTL(3)));
24629566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(4) (level of printing):        %d\n", mumps->id.ICNTL(4)));
24639566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(5) (input mat struct):         %d\n", mumps->id.ICNTL(5)));
24649566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(6) (matrix prescaling):        %d\n", mumps->id.ICNTL(6)));
24659566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(7) (sequential matrix ordering):%d\n", mumps->id.ICNTL(7)));
24669566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(8) (scaling strategy):         %d\n", mumps->id.ICNTL(8)));
24679566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(10) (max num of refinements):  %d\n", mumps->id.ICNTL(10)));
24689566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(11) (error analysis):          %d\n", mumps->id.ICNTL(11)));
2469a5e57a09SHong Zhang         if (mumps->id.ICNTL(11) > 0) {
24709566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(4) (inf norm of input mat):        %g\n", mumps->id.RINFOG(4)));
24719566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(5) (inf norm of solution):         %g\n", mumps->id.RINFOG(5)));
24729566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(6) (inf norm of residual):         %g\n", mumps->id.RINFOG(6)));
24739566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(7),RINFOG(8) (backward error est): %g, %g\n", mumps->id.RINFOG(7), mumps->id.RINFOG(8)));
24749566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(9) (error estimate):               %g\n", mumps->id.RINFOG(9)));
24759566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer, "    RINFOG(10),RINFOG(11)(condition numbers): %g, %g\n", mumps->id.RINFOG(10), mumps->id.RINFOG(11)));
2476f6c57405SHong Zhang         }
24779566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(12) (efficiency control):                         %d\n", mumps->id.ICNTL(12)));
24789566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(13) (sequential factorization of the root node):  %d\n", mumps->id.ICNTL(13)));
24799566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(14) (percentage of estimated workspace increase): %d\n", mumps->id.ICNTL(14)));
248045e3843bSPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(15) (compression of the input matrix):            %d\n", mumps->id.ICNTL(15)));
2481f6c57405SHong Zhang         /* ICNTL(15-17) not used */
24829566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(18) (input mat struct):                           %d\n", mumps->id.ICNTL(18)));
24839566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(19) (Schur complement info):                      %d\n", mumps->id.ICNTL(19)));
24849566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(20) (RHS sparse pattern):                         %d\n", mumps->id.ICNTL(20)));
24859566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(21) (solution struct):                            %d\n", mumps->id.ICNTL(21)));
24869566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(22) (in-core/out-of-core facility):               %d\n", mumps->id.ICNTL(22)));
24879566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(23) (max size of memory can be allocated locally):%d\n", mumps->id.ICNTL(23)));
2488c0165424SHong Zhang 
24899566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(24) (detection of null pivot rows):               %d\n", mumps->id.ICNTL(24)));
24909566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(25) (computation of a null space basis):          %d\n", mumps->id.ICNTL(25)));
24919566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(26) (Schur options for RHS or solution):          %d\n", mumps->id.ICNTL(26)));
24929566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(27) (blocking size for multiple RHS):             %d\n", mumps->id.ICNTL(27)));
24939566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(28) (use parallel or sequential ordering):        %d\n", mumps->id.ICNTL(28)));
24949566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(29) (parallel ordering):                          %d\n", mumps->id.ICNTL(29)));
249542179a6aSHong Zhang 
24969566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(30) (user-specified set of entries in inv(A)):    %d\n", mumps->id.ICNTL(30)));
24979566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(31) (factors is discarded in the solve phase):    %d\n", mumps->id.ICNTL(31)));
24989566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(33) (compute determinant):                        %d\n", mumps->id.ICNTL(33)));
24999566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(35) (activate BLR based factorization):           %d\n", mumps->id.ICNTL(35)));
25009566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(36) (choice of BLR factorization variant):        %d\n", mumps->id.ICNTL(36)));
25019566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(38) (estimated compression rate of LU factors):   %d\n", mumps->id.ICNTL(38)));
2502146931dbSPierre Jolivet         PetscCall(PetscViewerASCIIPrintf(viewer, "  ICNTL(58) (options for symbolic factorization):         %d\n", mumps->id.ICNTL(58)));
2503f6c57405SHong Zhang 
25049566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(1) (relative pivoting threshold):      %g\n", mumps->id.CNTL(1)));
25059566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(2) (stopping criterion of refinement): %g\n", mumps->id.CNTL(2)));
25069566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(3) (absolute pivoting threshold):      %g\n", mumps->id.CNTL(3)));
25079566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(4) (value of static pivoting):         %g\n", mumps->id.CNTL(4)));
25089566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(5) (fixation for null pivots):         %g\n", mumps->id.CNTL(5)));
25099566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  CNTL(7) (dropping parameter for BLR):       %g\n", mumps->id.CNTL(7)));
2510f6c57405SHong Zhang 
2511a5b23f4aSJose E. Roman         /* information local to each processor */
25129566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFO(1) (local estimated flops for the elimination after analysis):\n"));
25139566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPushSynchronized(viewer));
25149566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %g\n", mumps->myid, mumps->id.RINFO(1)));
25159566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
25169566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFO(2) (local estimated flops for the assembly after factorization):\n"));
25179566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %g\n", mumps->myid, mumps->id.RINFO(2)));
25189566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
25199566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFO(3) (local estimated flops for the elimination after factorization):\n"));
25209566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %g\n", mumps->myid, mumps->id.RINFO(3)));
25219566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2522f6c57405SHong Zhang 
25239566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(15) (estimated size of (in MB) MUMPS internal data for running numerical factorization):\n"));
25249566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(15)));
25259566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2526f6c57405SHong Zhang 
25279566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(16) (size of (in MB) MUMPS internal data used during numerical factorization):\n"));
25289566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(16)));
25299566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2530f6c57405SHong Zhang 
25319566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(23) (num of pivots eliminated on this processor after factorization):\n"));
25329566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(23)));
25339566063dSJacob Faibussowitsch         PetscCall(PetscViewerFlush(viewer));
2534b34f08ffSHong Zhang 
2535a0e18203SThibaut Appel         if (mumps->ninfo && mumps->ninfo <= 80) {
2536b34f08ffSHong Zhang           PetscInt i;
2537b34f08ffSHong Zhang           for (i = 0; i < mumps->ninfo; i++) {
25389566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer, "  INFO(%" PetscInt_FMT "):\n", mumps->info[i]));
25399566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "    [%d] %d\n", mumps->myid, mumps->id.INFO(mumps->info[i])));
25409566063dSJacob Faibussowitsch             PetscCall(PetscViewerFlush(viewer));
2541b34f08ffSHong Zhang           }
2542b34f08ffSHong Zhang         }
25439566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPopSynchronized(viewer));
25441511cd71SPierre Jolivet       } else PetscCall(PetscViewerASCIIPrintf(viewer, "  Use -%sksp_view ::ascii_info_detail to display information for all processes\n", ((PetscObject)A)->prefix ? ((PetscObject)A)->prefix : ""));
2545f6c57405SHong Zhang 
25461511cd71SPierre Jolivet       if (mumps->myid == 0) { /* information from the host */
25479566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFOG(1) (global estimated flops for the elimination after analysis): %g\n", mumps->id.RINFOG(1)));
25489566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFOG(2) (global estimated flops for the assembly after factorization): %g\n", mumps->id.RINFOG(2)));
25499566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  RINFOG(3) (global estimated flops for the elimination after factorization): %g\n", mumps->id.RINFOG(3)));
25509566063dSJacob 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)));
2551f6c57405SHong Zhang 
25529566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(3) (estimated real workspace for factors on all processors after analysis): %d\n", mumps->id.INFOG(3)));
25539566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(4) (estimated integer workspace for factors on all processors after analysis): %d\n", mumps->id.INFOG(4)));
25549566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(5) (estimated maximum front size in the complete tree): %d\n", mumps->id.INFOG(5)));
25559566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(6) (number of nodes in the complete tree): %d\n", mumps->id.INFOG(6)));
25569566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(7) (ordering option effectively used after analysis): %d\n", mumps->id.INFOG(7)));
25579566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(8) (structural symmetry in percent of the permuted matrix after analysis): %d\n", mumps->id.INFOG(8)));
25589566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(9) (total real/complex workspace to store the matrix factors after factorization): %d\n", mumps->id.INFOG(9)));
25599566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(10) (total integer space store the matrix factors after factorization): %d\n", mumps->id.INFOG(10)));
25609566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(11) (order of largest frontal matrix after factorization): %d\n", mumps->id.INFOG(11)));
25619566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(12) (number of off-diagonal pivots): %d\n", mumps->id.INFOG(12)));
25629566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(13) (number of delayed pivots after factorization): %d\n", mumps->id.INFOG(13)));
25639566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(14) (number of memory compress after factorization): %d\n", mumps->id.INFOG(14)));
25649566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(15) (number of steps of iterative refinement after solution): %d\n", mumps->id.INFOG(15)));
25659566063dSJacob 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)));
25669566063dSJacob 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)));
25679566063dSJacob 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)));
25689566063dSJacob 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)));
25699566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(20) (estimated number of entries in the factors): %d\n", mumps->id.INFOG(20)));
25709566063dSJacob 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)));
25719566063dSJacob 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)));
25729566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(23) (after analysis: value of ICNTL(6) effectively used): %d\n", mumps->id.INFOG(23)));
25739566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(24) (after analysis: value of ICNTL(12) effectively used): %d\n", mumps->id.INFOG(24)));
25749566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(25) (after factorization: number of pivots modified by static pivoting): %d\n", mumps->id.INFOG(25)));
25759566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(28) (after factorization: number of null pivots encountered): %d\n", mumps->id.INFOG(28)));
25769566063dSJacob 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)));
25779566063dSJacob 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)));
25789566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(32) (after analysis: type of analysis done): %d\n", mumps->id.INFOG(32)));
25799566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(33) (value used for ICNTL(8)): %d\n", mumps->id.INFOG(33)));
25809566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "  INFOG(34) (exponent of the determinant if determinant is requested): %d\n", mumps->id.INFOG(34)));
25819566063dSJacob 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)));
25829566063dSJacob 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)));
25839566063dSJacob 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)));
25849566063dSJacob 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)));
25859566063dSJacob 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)));
2586f6c57405SHong Zhang       }
2587f6c57405SHong Zhang     }
2588cb828f0fSHong Zhang   }
25893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2590f6c57405SHong Zhang }
2591f6c57405SHong Zhang 
259266976f2fSJacob Faibussowitsch static PetscErrorCode MatGetInfo_MUMPS(Mat A, MatInfoType flag, MatInfo *info)
2593d71ae5a4SJacob Faibussowitsch {
2594e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)A->data;
259535bd34faSBarry Smith 
259635bd34faSBarry Smith   PetscFunctionBegin;
259735bd34faSBarry Smith   info->block_size        = 1.0;
259864412097SPierre Jolivet   info->nz_allocated      = mumps->id.INFOG(20) >= 0 ? mumps->id.INFOG(20) : -1000000 * mumps->id.INFOG(20);
259964412097SPierre Jolivet   info->nz_used           = mumps->id.INFOG(20) >= 0 ? mumps->id.INFOG(20) : -1000000 * mumps->id.INFOG(20);
260035bd34faSBarry Smith   info->nz_unneeded       = 0.0;
260135bd34faSBarry Smith   info->assemblies        = 0.0;
260235bd34faSBarry Smith   info->mallocs           = 0.0;
260335bd34faSBarry Smith   info->memory            = 0.0;
260435bd34faSBarry Smith   info->fill_ratio_given  = 0;
260535bd34faSBarry Smith   info->fill_ratio_needed = 0;
260635bd34faSBarry Smith   info->factor_mallocs    = 0;
26073ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
260835bd34faSBarry Smith }
260935bd34faSBarry Smith 
261066976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorSetSchurIS_MUMPS(Mat F, IS is)
2611d71ae5a4SJacob Faibussowitsch {
2612e69c285eSBarry Smith   Mat_MUMPS         *mumps = (Mat_MUMPS *)F->data;
2613a3d589ffSStefano Zampini   const PetscScalar *arr;
26148e7ba810SStefano Zampini   const PetscInt    *idxs;
26158e7ba810SStefano Zampini   PetscInt           size, i;
26166444a565SStefano Zampini 
26176444a565SStefano Zampini   PetscFunctionBegin;
26189566063dSJacob Faibussowitsch   PetscCall(ISGetLocalSize(is, &size));
2619b3cb21ddSStefano Zampini   /* Schur complement matrix */
26209566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&F->schur));
26219566063dSJacob Faibussowitsch   PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, size, size, NULL, &F->schur));
26229566063dSJacob Faibussowitsch   PetscCall(MatDenseGetArrayRead(F->schur, &arr));
2623a3d589ffSStefano Zampini   mumps->id.schur      = (MumpsScalar *)arr;
2624a3d589ffSStefano Zampini   mumps->id.size_schur = size;
2625a3d589ffSStefano Zampini   mumps->id.schur_lld  = size;
26269566063dSJacob Faibussowitsch   PetscCall(MatDenseRestoreArrayRead(F->schur, &arr));
262748a46eb9SPierre Jolivet   if (mumps->sym == 1) PetscCall(MatSetOption(F->schur, MAT_SPD, PETSC_TRUE));
2628b3cb21ddSStefano Zampini 
2629b3cb21ddSStefano Zampini   /* MUMPS expects Fortran style indices */
26309566063dSJacob Faibussowitsch   PetscCall(PetscFree(mumps->id.listvar_schur));
26319566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(size, &mumps->id.listvar_schur));
26329566063dSJacob Faibussowitsch   PetscCall(ISGetIndices(is, &idxs));
26339566063dSJacob Faibussowitsch   for (i = 0; i < size; i++) PetscCall(PetscMUMPSIntCast(idxs[i] + 1, &(mumps->id.listvar_schur[i])));
26349566063dSJacob Faibussowitsch   PetscCall(ISRestoreIndices(is, &idxs));
263559ac8732SStefano Zampini   /* set a special value of ICNTL (not handled my MUMPS) to be used in the solve phase by PETSc */
2636b5fa320bSStefano Zampini   mumps->id.ICNTL(26) = -1;
26373ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
26386444a565SStefano Zampini }
263959ac8732SStefano Zampini 
264066976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorCreateSchurComplement_MUMPS(Mat F, Mat *S)
2641d71ae5a4SJacob Faibussowitsch {
26426444a565SStefano Zampini   Mat          St;
2643e69c285eSBarry Smith   Mat_MUMPS   *mumps = (Mat_MUMPS *)F->data;
26446444a565SStefano Zampini   PetscScalar *array;
26456444a565SStefano Zampini 
26466444a565SStefano Zampini   PetscFunctionBegin;
26479261f6e4SBarry Smith   PetscCheck(mumps->id.ICNTL(19), PetscObjectComm((PetscObject)F), PETSC_ERR_ORDER, "Schur complement mode not selected! Call MatFactorSetSchurIS() to enable it");
26489566063dSJacob Faibussowitsch   PetscCall(MatCreate(PETSC_COMM_SELF, &St));
26499566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(St, PETSC_DECIDE, PETSC_DECIDE, mumps->id.size_schur, mumps->id.size_schur));
26509566063dSJacob Faibussowitsch   PetscCall(MatSetType(St, MATDENSE));
26519566063dSJacob Faibussowitsch   PetscCall(MatSetUp(St));
26529566063dSJacob Faibussowitsch   PetscCall(MatDenseGetArray(St, &array));
265359ac8732SStefano Zampini   if (!mumps->sym) {                /* MUMPS always return a full matrix */
26546444a565SStefano Zampini     if (mumps->id.ICNTL(19) == 1) { /* stored by rows */
26556444a565SStefano Zampini       PetscInt i, j, N = mumps->id.size_schur;
26566444a565SStefano Zampini       for (i = 0; i < N; i++) {
26576444a565SStefano Zampini         for (j = 0; j < N; j++) {
26586444a565SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
26596444a565SStefano Zampini           PetscScalar val = mumps->id.schur[i * N + j];
26606444a565SStefano Zampini #else
26613e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.schur[i * N + j].r + PETSC_i * mumps->id.schur[i * N + j].i;
26626444a565SStefano Zampini #endif
26636444a565SStefano Zampini           array[j * N + i] = val;
26646444a565SStefano Zampini         }
26656444a565SStefano Zampini       }
26666444a565SStefano Zampini     } else { /* stored by columns */
26679566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(array, mumps->id.schur, mumps->id.size_schur * mumps->id.size_schur));
26686444a565SStefano Zampini     }
26696444a565SStefano Zampini   } else {                          /* either full or lower-triangular (not packed) */
26706444a565SStefano Zampini     if (mumps->id.ICNTL(19) == 2) { /* lower triangular stored by columns */
26716444a565SStefano Zampini       PetscInt i, j, N = mumps->id.size_schur;
26726444a565SStefano Zampini       for (i = 0; i < N; i++) {
26736444a565SStefano Zampini         for (j = i; j < N; j++) {
26746444a565SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
26756444a565SStefano Zampini           PetscScalar val = mumps->id.schur[i * N + j];
26766444a565SStefano Zampini #else
26773e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.schur[i * N + j].r + PETSC_i * mumps->id.schur[i * N + j].i;
26786444a565SStefano Zampini #endif
26793e5b40d0SPierre Jolivet           array[i * N + j] = array[j * N + i] = val;
26806444a565SStefano Zampini         }
26816444a565SStefano Zampini       }
26826444a565SStefano Zampini     } else if (mumps->id.ICNTL(19) == 3) { /* full matrix */
26839566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(array, mumps->id.schur, mumps->id.size_schur * mumps->id.size_schur));
26846444a565SStefano Zampini     } else { /* ICNTL(19) == 1 lower triangular stored by rows */
26856444a565SStefano Zampini       PetscInt i, j, N = mumps->id.size_schur;
26866444a565SStefano Zampini       for (i = 0; i < N; i++) {
26876444a565SStefano Zampini         for (j = 0; j < i + 1; j++) {
26886444a565SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
26896444a565SStefano Zampini           PetscScalar val = mumps->id.schur[i * N + j];
26906444a565SStefano Zampini #else
26913e5b40d0SPierre Jolivet           PetscScalar val = mumps->id.schur[i * N + j].r + PETSC_i * mumps->id.schur[i * N + j].i;
26926444a565SStefano Zampini #endif
26933e5b40d0SPierre Jolivet           array[i * N + j] = array[j * N + i] = val;
26946444a565SStefano Zampini         }
26956444a565SStefano Zampini       }
26966444a565SStefano Zampini     }
26976444a565SStefano Zampini   }
26989566063dSJacob Faibussowitsch   PetscCall(MatDenseRestoreArray(St, &array));
26996444a565SStefano Zampini   *S = St;
27003ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
27016444a565SStefano Zampini }
27026444a565SStefano Zampini 
270366976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsSetIcntl_MUMPS(Mat F, PetscInt icntl, PetscInt ival)
2704d71ae5a4SJacob Faibussowitsch {
2705e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
27065ccb76cbSHong Zhang 
27075ccb76cbSHong Zhang   PetscFunctionBegin;
2708413bcc21SPierre Jolivet   if (mumps->id.job == JOB_NULL) {                                       /* need to cache icntl and ival since PetscMUMPS_c() has never been called */
2709413bcc21SPierre Jolivet     PetscInt i, nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0; /* number of already cached ICNTL */
27109371c9d4SSatish Balay     for (i = 0; i < nICNTL_pre; ++i)
27119371c9d4SSatish Balay       if (mumps->ICNTL_pre[1 + 2 * i] == icntl) break; /* is this ICNTL already cached? */
2712413bcc21SPierre Jolivet     if (i == nICNTL_pre) {                             /* not already cached */
2713413bcc21SPierre Jolivet       if (i > 0) PetscCall(PetscRealloc(sizeof(PetscMUMPSInt) * (2 * nICNTL_pre + 3), &mumps->ICNTL_pre));
2714413bcc21SPierre Jolivet       else PetscCall(PetscCalloc(sizeof(PetscMUMPSInt) * 3, &mumps->ICNTL_pre));
2715413bcc21SPierre Jolivet       mumps->ICNTL_pre[0]++;
2716413bcc21SPierre Jolivet     }
2717413bcc21SPierre Jolivet     mumps->ICNTL_pre[1 + 2 * i] = icntl;
2718413bcc21SPierre Jolivet     PetscCall(PetscMUMPSIntCast(ival, mumps->ICNTL_pre + 2 + 2 * i));
2719413bcc21SPierre Jolivet   } else PetscCall(PetscMUMPSIntCast(ival, &mumps->id.ICNTL(icntl)));
27203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
27215ccb76cbSHong Zhang }
27225ccb76cbSHong Zhang 
272366976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetIcntl_MUMPS(Mat F, PetscInt icntl, PetscInt *ival)
2724d71ae5a4SJacob Faibussowitsch {
2725e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2726bc6112feSHong Zhang 
2727bc6112feSHong Zhang   PetscFunctionBegin;
272836df9881Sjeremy theler   if (mumps->id.job == JOB_NULL) {
272936df9881Sjeremy theler     PetscInt i, nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0;
273036df9881Sjeremy theler     *ival = 0;
273136df9881Sjeremy theler     for (i = 0; i < nICNTL_pre; ++i) {
273236df9881Sjeremy theler       if (mumps->ICNTL_pre[1 + 2 * i] == icntl) *ival = mumps->ICNTL_pre[2 + 2 * i];
273336df9881Sjeremy theler     }
273436df9881Sjeremy theler   } else *ival = mumps->id.ICNTL(icntl);
27353ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2736bc6112feSHong Zhang }
2737bc6112feSHong Zhang 
27385ccb76cbSHong Zhang /*@
27395ccb76cbSHong Zhang   MatMumpsSetIcntl - Set MUMPS parameter ICNTL()
27405ccb76cbSHong Zhang 
2741c3339decSBarry Smith   Logically Collective
27425ccb76cbSHong Zhang 
27435ccb76cbSHong Zhang   Input Parameters:
274411a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
27455ccb76cbSHong Zhang . icntl - index of MUMPS parameter array ICNTL()
27465ccb76cbSHong Zhang - ival  - value of MUMPS ICNTL(icntl)
27475ccb76cbSHong Zhang 
27483c7db156SBarry Smith   Options Database Key:
2749147403d9SBarry Smith . -mat_mumps_icntl_<icntl> <ival> - change the option numbered icntl to ival
27505ccb76cbSHong Zhang 
27515ccb76cbSHong Zhang   Level: beginner
27525ccb76cbSHong Zhang 
275396a0c994SBarry Smith   References:
2754606c0280SSatish Balay .  * - MUMPS Users' Guide
27555ccb76cbSHong Zhang 
27561cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
27575ccb76cbSHong Zhang @*/
2758d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsSetIcntl(Mat F, PetscInt icntl, PetscInt ival)
2759d71ae5a4SJacob Faibussowitsch {
27605ccb76cbSHong Zhang   PetscFunctionBegin;
27612989dfd4SHong Zhang   PetscValidType(F, 1);
276228b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
27635ccb76cbSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
27645ccb76cbSHong Zhang   PetscValidLogicalCollectiveInt(F, ival, 3);
2765146931dbSPierre Jolivet   PetscCheck((icntl >= 1 && icntl <= 38) || icntl == 58, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported ICNTL value %" PetscInt_FMT, icntl);
2766cac4c232SBarry Smith   PetscTryMethod(F, "MatMumpsSetIcntl_C", (Mat, PetscInt, PetscInt), (F, icntl, ival));
27673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
27685ccb76cbSHong Zhang }
27695ccb76cbSHong Zhang 
2770a21f80fcSHong Zhang /*@
2771a21f80fcSHong Zhang   MatMumpsGetIcntl - Get MUMPS parameter ICNTL()
2772a21f80fcSHong Zhang 
2773c3339decSBarry Smith   Logically Collective
2774a21f80fcSHong Zhang 
2775a21f80fcSHong Zhang   Input Parameters:
277611a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
2777a21f80fcSHong Zhang - icntl - index of MUMPS parameter array ICNTL()
2778a21f80fcSHong Zhang 
2779a21f80fcSHong Zhang   Output Parameter:
2780a21f80fcSHong Zhang . ival - value of MUMPS ICNTL(icntl)
2781a21f80fcSHong Zhang 
2782a21f80fcSHong Zhang   Level: beginner
2783a21f80fcSHong Zhang 
278496a0c994SBarry Smith   References:
2785606c0280SSatish Balay .  * - MUMPS Users' Guide
2786a21f80fcSHong Zhang 
27871cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
2788a21f80fcSHong Zhang @*/
2789d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetIcntl(Mat F, PetscInt icntl, PetscInt *ival)
2790d71ae5a4SJacob Faibussowitsch {
2791bc6112feSHong Zhang   PetscFunctionBegin;
27922989dfd4SHong Zhang   PetscValidType(F, 1);
279328b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
2794bc6112feSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
27954f572ea9SToby Isaac   PetscAssertPointer(ival, 3);
2796146931dbSPierre Jolivet   PetscCheck((icntl >= 1 && icntl <= 38) || icntl == 58, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported ICNTL value %" PetscInt_FMT, icntl);
2797cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetIcntl_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival));
27983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2799bc6112feSHong Zhang }
2800bc6112feSHong Zhang 
280166976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsSetCntl_MUMPS(Mat F, PetscInt icntl, PetscReal val)
2802d71ae5a4SJacob Faibussowitsch {
2803e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
28048928b65cSHong Zhang 
28058928b65cSHong Zhang   PetscFunctionBegin;
2806413bcc21SPierre Jolivet   if (mumps->id.job == JOB_NULL) {
2807413bcc21SPierre Jolivet     PetscInt i, nCNTL_pre = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0;
28089371c9d4SSatish Balay     for (i = 0; i < nCNTL_pre; ++i)
28099371c9d4SSatish Balay       if (mumps->CNTL_pre[1 + 2 * i] == icntl) break;
2810413bcc21SPierre Jolivet     if (i == nCNTL_pre) {
2811413bcc21SPierre Jolivet       if (i > 0) PetscCall(PetscRealloc(sizeof(PetscReal) * (2 * nCNTL_pre + 3), &mumps->CNTL_pre));
2812413bcc21SPierre Jolivet       else PetscCall(PetscCalloc(sizeof(PetscReal) * 3, &mumps->CNTL_pre));
2813413bcc21SPierre Jolivet       mumps->CNTL_pre[0]++;
2814413bcc21SPierre Jolivet     }
2815413bcc21SPierre Jolivet     mumps->CNTL_pre[1 + 2 * i] = icntl;
2816413bcc21SPierre Jolivet     mumps->CNTL_pre[2 + 2 * i] = val;
2817413bcc21SPierre Jolivet   } else mumps->id.CNTL(icntl) = val;
28183ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
28198928b65cSHong Zhang }
28208928b65cSHong Zhang 
282166976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetCntl_MUMPS(Mat F, PetscInt icntl, PetscReal *val)
2822d71ae5a4SJacob Faibussowitsch {
2823e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2824bc6112feSHong Zhang 
2825bc6112feSHong Zhang   PetscFunctionBegin;
282636df9881Sjeremy theler   if (mumps->id.job == JOB_NULL) {
282736df9881Sjeremy theler     PetscInt i, nCNTL_pre = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0;
282836df9881Sjeremy theler     *val = 0.0;
282936df9881Sjeremy theler     for (i = 0; i < nCNTL_pre; ++i) {
283036df9881Sjeremy theler       if (mumps->CNTL_pre[1 + 2 * i] == icntl) *val = mumps->CNTL_pre[2 + 2 * i];
283136df9881Sjeremy theler     }
283236df9881Sjeremy theler   } else *val = mumps->id.CNTL(icntl);
28333ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2834bc6112feSHong Zhang }
2835bc6112feSHong Zhang 
28368928b65cSHong Zhang /*@
28378928b65cSHong Zhang   MatMumpsSetCntl - Set MUMPS parameter CNTL()
28388928b65cSHong Zhang 
2839c3339decSBarry Smith   Logically Collective
28408928b65cSHong Zhang 
28418928b65cSHong Zhang   Input Parameters:
284211a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
28438928b65cSHong Zhang . icntl - index of MUMPS parameter array CNTL()
28448928b65cSHong Zhang - val   - value of MUMPS CNTL(icntl)
28458928b65cSHong Zhang 
28463c7db156SBarry Smith   Options Database Key:
2847147403d9SBarry Smith . -mat_mumps_cntl_<icntl> <val> - change the option numbered icntl to ival
28488928b65cSHong Zhang 
28498928b65cSHong Zhang   Level: beginner
28508928b65cSHong Zhang 
285196a0c994SBarry Smith   References:
2852606c0280SSatish Balay .  * - MUMPS Users' Guide
28538928b65cSHong Zhang 
28541cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
28558928b65cSHong Zhang @*/
2856d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsSetCntl(Mat F, PetscInt icntl, PetscReal val)
2857d71ae5a4SJacob Faibussowitsch {
28588928b65cSHong Zhang   PetscFunctionBegin;
28592989dfd4SHong Zhang   PetscValidType(F, 1);
286028b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
28618928b65cSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
2862bc6112feSHong Zhang   PetscValidLogicalCollectiveReal(F, val, 3);
2863413bcc21SPierre Jolivet   PetscCheck(icntl >= 1 && icntl <= 7, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported CNTL value %" PetscInt_FMT, icntl);
2864cac4c232SBarry Smith   PetscTryMethod(F, "MatMumpsSetCntl_C", (Mat, PetscInt, PetscReal), (F, icntl, val));
28653ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
28668928b65cSHong Zhang }
28678928b65cSHong Zhang 
2868a21f80fcSHong Zhang /*@
2869a21f80fcSHong Zhang   MatMumpsGetCntl - Get MUMPS parameter CNTL()
2870a21f80fcSHong Zhang 
2871c3339decSBarry Smith   Logically Collective
2872a21f80fcSHong Zhang 
2873a21f80fcSHong Zhang   Input Parameters:
287411a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
2875a21f80fcSHong Zhang - icntl - index of MUMPS parameter array CNTL()
2876a21f80fcSHong Zhang 
2877a21f80fcSHong Zhang   Output Parameter:
2878a21f80fcSHong Zhang . val - value of MUMPS CNTL(icntl)
2879a21f80fcSHong Zhang 
2880a21f80fcSHong Zhang   Level: beginner
2881a21f80fcSHong Zhang 
288296a0c994SBarry Smith   References:
2883606c0280SSatish Balay .  * - MUMPS Users' Guide
2884a21f80fcSHong Zhang 
28851cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
2886a21f80fcSHong Zhang @*/
2887d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetCntl(Mat F, PetscInt icntl, PetscReal *val)
2888d71ae5a4SJacob Faibussowitsch {
2889bc6112feSHong Zhang   PetscFunctionBegin;
28902989dfd4SHong Zhang   PetscValidType(F, 1);
289128b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
2892bc6112feSHong Zhang   PetscValidLogicalCollectiveInt(F, icntl, 2);
28934f572ea9SToby Isaac   PetscAssertPointer(val, 3);
2894413bcc21SPierre Jolivet   PetscCheck(icntl >= 1 && icntl <= 7, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported CNTL value %" PetscInt_FMT, icntl);
2895cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetCntl_C", (Mat, PetscInt, PetscReal *), (F, icntl, val));
28963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2897bc6112feSHong Zhang }
2898bc6112feSHong Zhang 
289966976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInfo_MUMPS(Mat F, PetscInt icntl, PetscInt *info)
2900d71ae5a4SJacob Faibussowitsch {
2901e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2902bc6112feSHong Zhang 
2903bc6112feSHong Zhang   PetscFunctionBegin;
2904bc6112feSHong Zhang   *info = mumps->id.INFO(icntl);
29053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2906bc6112feSHong Zhang }
2907bc6112feSHong Zhang 
290866976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInfog_MUMPS(Mat F, PetscInt icntl, PetscInt *infog)
2909d71ae5a4SJacob Faibussowitsch {
2910e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2911bc6112feSHong Zhang 
2912bc6112feSHong Zhang   PetscFunctionBegin;
2913bc6112feSHong Zhang   *infog = mumps->id.INFOG(icntl);
29143ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2915bc6112feSHong Zhang }
2916bc6112feSHong Zhang 
291766976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetRinfo_MUMPS(Mat F, PetscInt icntl, PetscReal *rinfo)
2918d71ae5a4SJacob Faibussowitsch {
2919e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2920bc6112feSHong Zhang 
2921bc6112feSHong Zhang   PetscFunctionBegin;
2922bc6112feSHong Zhang   *rinfo = mumps->id.RINFO(icntl);
29233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2924bc6112feSHong Zhang }
2925bc6112feSHong Zhang 
292666976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetRinfog_MUMPS(Mat F, PetscInt icntl, PetscReal *rinfog)
2927d71ae5a4SJacob Faibussowitsch {
2928e69c285eSBarry Smith   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
2929bc6112feSHong Zhang 
2930bc6112feSHong Zhang   PetscFunctionBegin;
2931bc6112feSHong Zhang   *rinfog = mumps->id.RINFOG(icntl);
29323ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2933bc6112feSHong Zhang }
2934bc6112feSHong Zhang 
293566976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetNullPivots_MUMPS(Mat F, PetscInt *size, PetscInt **array)
29365c0bae8cSAshish Patel {
29375c0bae8cSAshish Patel   Mat_MUMPS *mumps = (Mat_MUMPS *)F->data;
29385c0bae8cSAshish Patel 
29395c0bae8cSAshish Patel   PetscFunctionBegin;
29405c0bae8cSAshish 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");
29415c0bae8cSAshish Patel   *size  = 0;
29425c0bae8cSAshish Patel   *array = NULL;
29435c0bae8cSAshish Patel   if (!mumps->myid) {
29445c0bae8cSAshish Patel     *size = mumps->id.INFOG(28);
29455c0bae8cSAshish Patel     PetscCall(PetscMalloc1(*size, array));
29465c0bae8cSAshish Patel     for (int i = 0; i < *size; i++) (*array)[i] = mumps->id.pivnul_list[i] - 1;
29475c0bae8cSAshish Patel   }
29485c0bae8cSAshish Patel   PetscFunctionReturn(PETSC_SUCCESS);
29495c0bae8cSAshish Patel }
29505c0bae8cSAshish Patel 
295166976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInverse_MUMPS(Mat F, Mat spRHS)
2952d71ae5a4SJacob Faibussowitsch {
29530e6b8875SHong Zhang   Mat          Bt = NULL, Btseq = NULL;
29540e6b8875SHong Zhang   PetscBool    flg;
2955bb599dfdSHong Zhang   Mat_MUMPS   *mumps = (Mat_MUMPS *)F->data;
2956bb599dfdSHong Zhang   PetscScalar *aa;
2957f410b75aSHong Zhang   PetscInt     spnr, *ia, *ja, M, nrhs;
2958bb599dfdSHong Zhang 
2959bb599dfdSHong Zhang   PetscFunctionBegin;
29604f572ea9SToby Isaac   PetscAssertPointer(spRHS, 2);
2961013e2dc7SBarry Smith   PetscCall(PetscObjectTypeCompare((PetscObject)spRHS, MATTRANSPOSEVIRTUAL, &flg));
29620e6b8875SHong Zhang   if (flg) {
29639566063dSJacob Faibussowitsch     PetscCall(MatTransposeGetMat(spRHS, &Bt));
2964013e2dc7SBarry Smith   } else SETERRQ(PetscObjectComm((PetscObject)spRHS), PETSC_ERR_ARG_WRONG, "Matrix spRHS must be type MATTRANSPOSEVIRTUAL matrix");
2965bb599dfdSHong Zhang 
29669566063dSJacob Faibussowitsch   PetscCall(MatMumpsSetIcntl(F, 30, 1));
2967bb599dfdSHong Zhang 
29682d4298aeSJunchao Zhang   if (mumps->petsc_size > 1) {
29690e6b8875SHong Zhang     Mat_MPIAIJ *b = (Mat_MPIAIJ *)Bt->data;
29700e6b8875SHong Zhang     Btseq         = b->A;
29710e6b8875SHong Zhang   } else {
29720e6b8875SHong Zhang     Btseq = Bt;
29730e6b8875SHong Zhang   }
29740e6b8875SHong Zhang 
29759566063dSJacob Faibussowitsch   PetscCall(MatGetSize(spRHS, &M, &nrhs));
2976f410b75aSHong Zhang   mumps->id.nrhs = nrhs;
2977f410b75aSHong Zhang   mumps->id.lrhs = M;
2978f410b75aSHong Zhang   mumps->id.rhs  = NULL;
2979f410b75aSHong Zhang 
2980e3f2db6aSHong Zhang   if (!mumps->myid) {
29819566063dSJacob Faibussowitsch     PetscCall(MatSeqAIJGetArray(Btseq, &aa));
29829566063dSJacob Faibussowitsch     PetscCall(MatGetRowIJ(Btseq, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
298328b400f6SJacob Faibussowitsch     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
29849566063dSJacob Faibussowitsch     PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs));
2985bb599dfdSHong Zhang     mumps->id.rhs_sparse = (MumpsScalar *)aa;
2986e3f2db6aSHong Zhang   } else {
2987e3f2db6aSHong Zhang     mumps->id.irhs_ptr    = NULL;
2988e3f2db6aSHong Zhang     mumps->id.irhs_sparse = NULL;
2989e3f2db6aSHong Zhang     mumps->id.nz_rhs      = 0;
2990e3f2db6aSHong Zhang     mumps->id.rhs_sparse  = NULL;
2991e3f2db6aSHong Zhang   }
2992bb599dfdSHong Zhang   mumps->id.ICNTL(20) = 1; /* rhs is sparse */
2993e3f2db6aSHong Zhang   mumps->id.ICNTL(21) = 0; /* solution is in assembled centralized format */
2994bb599dfdSHong Zhang 
2995bb599dfdSHong Zhang   /* solve phase */
2996bb599dfdSHong Zhang   mumps->id.job = JOB_SOLVE;
29973ab56b82SJunchao Zhang   PetscMUMPS_c(mumps);
29989261f6e4SBarry 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));
299914267174SHong Zhang 
3000e3f2db6aSHong Zhang   if (!mumps->myid) {
30019566063dSJacob Faibussowitsch     PetscCall(MatSeqAIJRestoreArray(Btseq, &aa));
30029566063dSJacob Faibussowitsch     PetscCall(MatRestoreRowIJ(Btseq, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg));
300328b400f6SJacob Faibussowitsch     PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure");
3004e3f2db6aSHong Zhang   }
30053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3006bb599dfdSHong Zhang }
3007bb599dfdSHong Zhang 
3008bb599dfdSHong Zhang /*@
30092ef1f0ffSBarry Smith   MatMumpsGetInverse - Get user-specified set of entries in inverse of `A`
3010bb599dfdSHong Zhang 
3011c3339decSBarry Smith   Logically Collective
3012bb599dfdSHong Zhang 
301320f4b53cSBarry Smith   Input Parameter:
301420f4b53cSBarry Smith . F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3015bb599dfdSHong Zhang 
3016bb599dfdSHong Zhang   Output Parameter:
301720f4b53cSBarry Smith . spRHS - sequential sparse matrix in `MATTRANSPOSEVIRTUAL` format with requested entries of inverse of `A`
3018bb599dfdSHong Zhang 
3019bb599dfdSHong Zhang   Level: beginner
3020bb599dfdSHong Zhang 
3021bb599dfdSHong Zhang   References:
3022606c0280SSatish Balay .  * - MUMPS Users' Guide
3023bb599dfdSHong Zhang 
30241cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatCreateTranspose()`
3025bb599dfdSHong Zhang @*/
3026d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInverse(Mat F, Mat spRHS)
3027d71ae5a4SJacob Faibussowitsch {
3028bb599dfdSHong Zhang   PetscFunctionBegin;
3029bb599dfdSHong Zhang   PetscValidType(F, 1);
303028b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
3031cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInverse_C", (Mat, Mat), (F, spRHS));
30323ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3033bb599dfdSHong Zhang }
3034bb599dfdSHong Zhang 
303566976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInverseTranspose_MUMPS(Mat F, Mat spRHST)
3036d71ae5a4SJacob Faibussowitsch {
30370e6b8875SHong Zhang   Mat spRHS;
30380e6b8875SHong Zhang 
30390e6b8875SHong Zhang   PetscFunctionBegin;
30409566063dSJacob Faibussowitsch   PetscCall(MatCreateTranspose(spRHST, &spRHS));
30419566063dSJacob Faibussowitsch   PetscCall(MatMumpsGetInverse_MUMPS(F, spRHS));
30429566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&spRHS));
30433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
30440e6b8875SHong Zhang }
30450e6b8875SHong Zhang 
30460e6b8875SHong Zhang /*@
30472ef1f0ffSBarry Smith   MatMumpsGetInverseTranspose - Get user-specified set of entries in inverse of matrix `A`^T
30480e6b8875SHong Zhang 
3049c3339decSBarry Smith   Logically Collective
30500e6b8875SHong Zhang 
305120f4b53cSBarry Smith   Input Parameter:
305220f4b53cSBarry Smith . F - the factored matrix of A obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
30530e6b8875SHong Zhang 
30540e6b8875SHong Zhang   Output Parameter:
305520f4b53cSBarry Smith . spRHST - sequential sparse matrix in `MATAIJ` format containing the requested entries of inverse of `A`^T
30560e6b8875SHong Zhang 
30570e6b8875SHong Zhang   Level: beginner
30580e6b8875SHong Zhang 
30590e6b8875SHong Zhang   References:
3060606c0280SSatish Balay .  * - MUMPS Users' Guide
30610e6b8875SHong Zhang 
30621cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatCreateTranspose()`, `MatMumpsGetInverse()`
30630e6b8875SHong Zhang @*/
3064d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInverseTranspose(Mat F, Mat spRHST)
3065d71ae5a4SJacob Faibussowitsch {
30660e6b8875SHong Zhang   PetscBool flg;
30670e6b8875SHong Zhang 
30680e6b8875SHong Zhang   PetscFunctionBegin;
30690e6b8875SHong Zhang   PetscValidType(F, 1);
307028b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
30719566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompareAny((PetscObject)spRHST, &flg, MATSEQAIJ, MATMPIAIJ, NULL));
307228b400f6SJacob Faibussowitsch   PetscCheck(flg, PetscObjectComm((PetscObject)spRHST), PETSC_ERR_ARG_WRONG, "Matrix spRHST must be MATAIJ matrix");
30730e6b8875SHong Zhang 
3074cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInverseTranspose_C", (Mat, Mat), (F, spRHST));
30753ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
30760e6b8875SHong Zhang }
30770e6b8875SHong Zhang 
3078a21f80fcSHong Zhang /*@
3079a21f80fcSHong Zhang   MatMumpsGetInfo - Get MUMPS parameter INFO()
3080a21f80fcSHong Zhang 
3081c3339decSBarry Smith   Logically Collective
3082a21f80fcSHong Zhang 
3083a21f80fcSHong Zhang   Input Parameters:
308411a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3085a21f80fcSHong Zhang - icntl - index of MUMPS parameter array INFO()
3086a21f80fcSHong Zhang 
3087a21f80fcSHong Zhang   Output Parameter:
3088a21f80fcSHong Zhang . ival - value of MUMPS INFO(icntl)
3089a21f80fcSHong Zhang 
3090a21f80fcSHong Zhang   Level: beginner
3091a21f80fcSHong Zhang 
309296a0c994SBarry Smith   References:
3093606c0280SSatish Balay .  * - MUMPS Users' Guide
3094a21f80fcSHong Zhang 
30951cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
3096a21f80fcSHong Zhang @*/
3097d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInfo(Mat F, PetscInt icntl, PetscInt *ival)
3098d71ae5a4SJacob Faibussowitsch {
3099bc6112feSHong Zhang   PetscFunctionBegin;
31002989dfd4SHong Zhang   PetscValidType(F, 1);
310128b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
31024f572ea9SToby Isaac   PetscAssertPointer(ival, 3);
3103cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInfo_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival));
31043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3105bc6112feSHong Zhang }
3106bc6112feSHong Zhang 
3107a21f80fcSHong Zhang /*@
3108a21f80fcSHong Zhang   MatMumpsGetInfog - Get MUMPS parameter INFOG()
3109a21f80fcSHong Zhang 
3110c3339decSBarry Smith   Logically Collective
3111a21f80fcSHong Zhang 
3112a21f80fcSHong Zhang   Input Parameters:
311311a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3114a21f80fcSHong Zhang - icntl - index of MUMPS parameter array INFOG()
3115a21f80fcSHong Zhang 
3116a21f80fcSHong Zhang   Output Parameter:
3117a21f80fcSHong Zhang . ival - value of MUMPS INFOG(icntl)
3118a21f80fcSHong Zhang 
3119a21f80fcSHong Zhang   Level: beginner
3120a21f80fcSHong Zhang 
312196a0c994SBarry Smith   References:
3122606c0280SSatish Balay .  * - MUMPS Users' Guide
3123a21f80fcSHong Zhang 
31241cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`
3125a21f80fcSHong Zhang @*/
3126d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInfog(Mat F, PetscInt icntl, PetscInt *ival)
3127d71ae5a4SJacob Faibussowitsch {
3128bc6112feSHong Zhang   PetscFunctionBegin;
31292989dfd4SHong Zhang   PetscValidType(F, 1);
313028b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
31314f572ea9SToby Isaac   PetscAssertPointer(ival, 3);
3132cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetInfog_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival));
31333ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3134bc6112feSHong Zhang }
3135bc6112feSHong Zhang 
3136a21f80fcSHong Zhang /*@
3137a21f80fcSHong Zhang   MatMumpsGetRinfo - Get MUMPS parameter RINFO()
3138a21f80fcSHong Zhang 
3139c3339decSBarry Smith   Logically Collective
3140a21f80fcSHong Zhang 
3141a21f80fcSHong Zhang   Input Parameters:
314211a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3143a21f80fcSHong Zhang - icntl - index of MUMPS parameter array RINFO()
3144a21f80fcSHong Zhang 
3145a21f80fcSHong Zhang   Output Parameter:
3146a21f80fcSHong Zhang . val - value of MUMPS RINFO(icntl)
3147a21f80fcSHong Zhang 
3148a21f80fcSHong Zhang   Level: beginner
3149a21f80fcSHong Zhang 
315096a0c994SBarry Smith   References:
3151606c0280SSatish Balay .  * - MUMPS Users' Guide
3152a21f80fcSHong Zhang 
31531cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfog()`
3154a21f80fcSHong Zhang @*/
3155d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetRinfo(Mat F, PetscInt icntl, PetscReal *val)
3156d71ae5a4SJacob Faibussowitsch {
3157bc6112feSHong Zhang   PetscFunctionBegin;
31582989dfd4SHong Zhang   PetscValidType(F, 1);
315928b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
31604f572ea9SToby Isaac   PetscAssertPointer(val, 3);
3161cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetRinfo_C", (Mat, PetscInt, PetscReal *), (F, icntl, val));
31623ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3163bc6112feSHong Zhang }
3164bc6112feSHong Zhang 
3165a21f80fcSHong Zhang /*@
3166a21f80fcSHong Zhang   MatMumpsGetRinfog - Get MUMPS parameter RINFOG()
3167a21f80fcSHong Zhang 
3168c3339decSBarry Smith   Logically Collective
3169a21f80fcSHong Zhang 
3170a21f80fcSHong Zhang   Input Parameters:
317111a5261eSBarry Smith + F     - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
3172a21f80fcSHong Zhang - icntl - index of MUMPS parameter array RINFOG()
3173a21f80fcSHong Zhang 
3174a21f80fcSHong Zhang   Output Parameter:
3175a21f80fcSHong Zhang . val - value of MUMPS RINFOG(icntl)
3176a21f80fcSHong Zhang 
3177a21f80fcSHong Zhang   Level: beginner
3178a21f80fcSHong Zhang 
317996a0c994SBarry Smith   References:
3180606c0280SSatish Balay .  * - MUMPS Users' Guide
3181a21f80fcSHong Zhang 
31821cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`
3183a21f80fcSHong Zhang @*/
3184d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetRinfog(Mat F, PetscInt icntl, PetscReal *val)
3185d71ae5a4SJacob Faibussowitsch {
3186bc6112feSHong Zhang   PetscFunctionBegin;
31872989dfd4SHong Zhang   PetscValidType(F, 1);
318828b400f6SJacob Faibussowitsch   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
31894f572ea9SToby Isaac   PetscAssertPointer(val, 3);
3190cac4c232SBarry Smith   PetscUseMethod(F, "MatMumpsGetRinfog_C", (Mat, PetscInt, PetscReal *), (F, icntl, val));
31913ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3192bc6112feSHong Zhang }
3193bc6112feSHong Zhang 
31945c0bae8cSAshish Patel /*@
31955c0bae8cSAshish Patel   MatMumpsGetNullPivots - Get MUMPS parameter PIVNUL_LIST()
31965c0bae8cSAshish Patel 
31975c0bae8cSAshish Patel   Logically Collective
31985c0bae8cSAshish Patel 
31995c0bae8cSAshish Patel   Input Parameter:
32005c0bae8cSAshish Patel . F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface
32015c0bae8cSAshish Patel 
32025c0bae8cSAshish Patel   Output Parameters:
32035c0bae8cSAshish Patel + size  - local size of the array. The size of the array is non-zero only on the host.
32045c0bae8cSAshish 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
32055c0bae8cSAshish Patel            for freeing this array.
32065c0bae8cSAshish Patel 
32075c0bae8cSAshish Patel   Level: beginner
32085c0bae8cSAshish Patel 
32095c0bae8cSAshish Patel   References:
32105c0bae8cSAshish Patel .  * - MUMPS Users' Guide
32115c0bae8cSAshish Patel 
32121cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`
32135c0bae8cSAshish Patel @*/
32145c0bae8cSAshish Patel PetscErrorCode MatMumpsGetNullPivots(Mat F, PetscInt *size, PetscInt **array)
32155c0bae8cSAshish Patel {
32165c0bae8cSAshish Patel   PetscFunctionBegin;
32175c0bae8cSAshish Patel   PetscValidType(F, 1);
32185c0bae8cSAshish Patel   PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix");
32194f572ea9SToby Isaac   PetscAssertPointer(size, 2);
32204f572ea9SToby Isaac   PetscAssertPointer(array, 3);
32215c0bae8cSAshish Patel   PetscUseMethod(F, "MatMumpsGetNullPivots_C", (Mat, PetscInt *, PetscInt **), (F, size, array));
32225c0bae8cSAshish Patel   PetscFunctionReturn(PETSC_SUCCESS);
32235c0bae8cSAshish Patel }
32245c0bae8cSAshish Patel 
322524b6179bSKris Buschelman /*MC
32262692d6eeSBarry Smith   MATSOLVERMUMPS -  A matrix type providing direct solvers (LU and Cholesky) for
322724b6179bSKris Buschelman   distributed and sequential matrices via the external package MUMPS.
322824b6179bSKris Buschelman 
322911a5261eSBarry Smith   Works with `MATAIJ` and `MATSBAIJ` matrices
323024b6179bSKris Buschelman 
3231c2b89b5dSBarry Smith   Use ./configure --download-mumps --download-scalapack --download-parmetis --download-metis --download-ptscotch to have PETSc installed with MUMPS
3232c2b89b5dSBarry Smith 
32332ef1f0ffSBarry 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.
32342ef1f0ffSBarry Smith   See details below.
3235217d3b1eSJunchao Zhang 
32362ef1f0ffSBarry Smith   Use `-pc_type cholesky` or `lu` `-pc_factor_mat_solver_type mumps` to use this direct solver
3237c2b89b5dSBarry Smith 
323824b6179bSKris Buschelman   Options Database Keys:
32394422a9fcSPatrick Sanan +  -mat_mumps_icntl_1 - ICNTL(1): output stream for error messages
32404422a9fcSPatrick Sanan .  -mat_mumps_icntl_2 - ICNTL(2): output stream for diagnostic printing, statistics, and warning
32414422a9fcSPatrick Sanan .  -mat_mumps_icntl_3 -  ICNTL(3): output stream for global information, collected on the host
32424422a9fcSPatrick Sanan .  -mat_mumps_icntl_4 -  ICNTL(4): level of printing (0 to 4)
32434422a9fcSPatrick Sanan .  -mat_mumps_icntl_6 - ICNTL(6): permutes to a zero-free diagonal and/or scale the matrix (0 to 7)
3244b53c1a7fSBarry 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
3245b53c1a7fSBarry Smith                         Use -pc_factor_mat_ordering_type <type> to have PETSc perform the ordering (sequential only)
32464422a9fcSPatrick Sanan .  -mat_mumps_icntl_8  - ICNTL(8): scaling strategy (-2 to 8 or 77)
32474422a9fcSPatrick Sanan .  -mat_mumps_icntl_10  - ICNTL(10): max num of refinements
32484422a9fcSPatrick Sanan .  -mat_mumps_icntl_11  - ICNTL(11): statistics related to an error analysis (via -ksp_view)
32494422a9fcSPatrick Sanan .  -mat_mumps_icntl_12  - ICNTL(12): an ordering strategy for symmetric matrices (0 to 3)
32504422a9fcSPatrick Sanan .  -mat_mumps_icntl_13  - ICNTL(13): parallelism of the root node (enable ScaLAPACK) and its splitting
32514422a9fcSPatrick Sanan .  -mat_mumps_icntl_14  - ICNTL(14): percentage increase in the estimated working space
325245e3843bSPierre Jolivet .  -mat_mumps_icntl_15  - ICNTL(15): compression of the input matrix resulting from a block format
32534422a9fcSPatrick Sanan .  -mat_mumps_icntl_19  - ICNTL(19): computes the Schur complement
325425aac85cSJunchao Zhang .  -mat_mumps_icntl_20  - ICNTL(20): give MUMPS centralized (0) or distributed (10) dense RHS
32554422a9fcSPatrick Sanan .  -mat_mumps_icntl_22  - ICNTL(22): in-core/out-of-core factorization and solve (0 or 1)
32564422a9fcSPatrick Sanan .  -mat_mumps_icntl_23  - ICNTL(23): max size of the working memory (MB) that can allocate per processor
32574422a9fcSPatrick Sanan .  -mat_mumps_icntl_24  - ICNTL(24): detection of null pivot rows (0 or 1)
32584422a9fcSPatrick Sanan .  -mat_mumps_icntl_25  - ICNTL(25): compute a solution of a deficient matrix and a null space basis
32594422a9fcSPatrick Sanan .  -mat_mumps_icntl_26  - ICNTL(26): drives the solution phase if a Schur complement matrix
32604422a9fcSPatrick 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
32614422a9fcSPatrick Sanan .  -mat_mumps_icntl_29 - ICNTL(29): parallel ordering 1 = ptscotch, 2 = parmetis
32624422a9fcSPatrick Sanan .  -mat_mumps_icntl_30 - ICNTL(30): compute user-specified set of entries in inv(A)
32634422a9fcSPatrick Sanan .  -mat_mumps_icntl_31 - ICNTL(31): indicates which factors may be discarded during factorization
32644422a9fcSPatrick Sanan .  -mat_mumps_icntl_33 - ICNTL(33): compute determinant
3265a0e18203SThibaut Appel .  -mat_mumps_icntl_35 - ICNTL(35): level of activation of BLR (Block Low-Rank) feature
3266a0e18203SThibaut Appel .  -mat_mumps_icntl_36 - ICNTL(36): controls the choice of BLR factorization variant
3267a0e18203SThibaut Appel .  -mat_mumps_icntl_38 - ICNTL(38): sets the estimated compression rate of LU factors with BLR
3268146931dbSPierre Jolivet .  -mat_mumps_icntl_58 - ICNTL(58): options for symbolic factorization
32694422a9fcSPatrick Sanan .  -mat_mumps_cntl_1  - CNTL(1): relative pivoting threshold
32704422a9fcSPatrick Sanan .  -mat_mumps_cntl_2  -  CNTL(2): stopping criterion of refinement
32714422a9fcSPatrick Sanan .  -mat_mumps_cntl_3 - CNTL(3): absolute pivoting threshold
32724422a9fcSPatrick Sanan .  -mat_mumps_cntl_4 - CNTL(4): value for static pivoting
3273217d3b1eSJunchao Zhang .  -mat_mumps_cntl_5 - CNTL(5): fixation for null pivots
3274a0e18203SThibaut Appel .  -mat_mumps_cntl_7 - CNTL(7): precision of the dropping parameter used during BLR factorization
3275217d3b1eSJunchao 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.
3276217d3b1eSJunchao Zhang                                    Default might be the number of cores per CPU package (socket) as reported by hwloc and suggested by the MUMPS manual.
327724b6179bSKris Buschelman 
327824b6179bSKris Buschelman   Level: beginner
327924b6179bSKris Buschelman 
328095452b02SPatrick Sanan     Notes:
32812ef1f0ffSBarry 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
32822ef1f0ffSBarry Smith     error if the matrix is Hermitian.
328338548759SBarry Smith 
328426cc229bSBarry 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
328526cc229bSBarry Smith     `MatSetOptionsPrefixFactor()` on the matrix from which the factor was obtained or `MatSetOptionsPrefix()` on the factor matrix.
328626cc229bSBarry Smith 
32872ef1f0ffSBarry 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
32882ef1f0ffSBarry Smith     the failure with
32892ef1f0ffSBarry Smith .vb
32902ef1f0ffSBarry Smith           KSPGetPC(ksp,&pc);
32912ef1f0ffSBarry Smith           PCFactorGetMatrix(pc,&mat);
32922ef1f0ffSBarry Smith           MatMumpsGetInfo(mat,....);
32932ef1f0ffSBarry Smith           MatMumpsGetInfog(mat,....); etc.
32942ef1f0ffSBarry Smith .ve
32952ef1f0ffSBarry Smith     Or run with `-ksp_error_if_not_converged` and the program will be stopped and the information printed in the error message.
32969fc87aa7SBarry Smith 
3297a5399872SJunchao Zhang     MUMPS provides 64-bit integer support in two build modes:
3298a5399872SJunchao Zhang       full 64-bit: here MUMPS is built with C preprocessing flag -DINTSIZE64 and Fortran compiler option -i8, -fdefault-integer-8 or equivalent, and
3299a5399872SJunchao 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).
33008fcaa860SBarry Smith 
3301a5399872SJunchao 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,
3302a5399872SJunchao 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
3303a5399872SJunchao 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
3304a5399872SJunchao Zhang       integer) build of all dependent libraries MPI, ScaLAPACK, LAPACK and BLAS.
3305a5399872SJunchao Zhang 
3306a5399872SJunchao 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.
3307a5399872SJunchao Zhang 
3308a5399872SJunchao Zhang   Two modes to run MUMPS/PETSc with OpenMP
33092ef1f0ffSBarry Smith .vb
33102ef1f0ffSBarry Smith      Set OMP_NUM_THREADS and run with fewer MPI ranks than cores. For example, if you want to have 16 OpenMP
33112ef1f0ffSBarry Smith      threads per rank, then you may use "export OMP_NUM_THREADS=16 && mpirun -n 4 ./test".
33122ef1f0ffSBarry Smith .ve
33138fcaa860SBarry Smith 
33142ef1f0ffSBarry Smith .vb
33152ef1f0ffSBarry Smith      -mat_mumps_use_omp_threads [m] and run your code with as many MPI ranks as the number of cores. For example,
33162ef1f0ffSBarry 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"
33172ef1f0ffSBarry Smith .ve
33188fcaa860SBarry Smith 
33198fcaa860SBarry Smith    To run MUMPS in MPI+OpenMP hybrid mode (i.e., enable multithreading in MUMPS), but still run the non-MUMPS part
33202ef1f0ffSBarry 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`
33212ef1f0ffSBarry 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
33228fcaa860SBarry Smith    libraries, to really get performance, you should have multithreaded BLAS libraries such as Intel MKL, AMD ACML, Cray libSci or OpenBLAS
33238fcaa860SBarry Smith    (PETSc will automatically try to utilized a threaded BLAS if --with-openmp is provided).
3324217d3b1eSJunchao Zhang 
33258fcaa860SBarry 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
3326217d3b1eSJunchao Zhang    processes on each compute node. Listing the processes in rank ascending order, we split processes on a node into consecutive groups of
3327217d3b1eSJunchao 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
3328217d3b1eSJunchao 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
3329217d3b1eSJunchao 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.
3330217d3b1eSJunchao 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,
3331217d3b1eSJunchao 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
3332217d3b1eSJunchao 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
3333217d3b1eSJunchao 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
3334217d3b1eSJunchao 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.
33358fcaa860SBarry 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
3336217d3b1eSJunchao Zhang    examine the mapping result.
3337217d3b1eSJunchao Zhang 
333811a5261eSBarry 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,
333911a5261eSBarry 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
334011a5261eSBarry Smith    calls `omp_set_num_threads`(m) internally before calling MUMPS.
3341217d3b1eSJunchao Zhang 
3342217d3b1eSJunchao Zhang    References:
3343606c0280SSatish Balay +  * - Heroux, Michael A., R. Brightwell, and Michael M. Wolf. "Bi-modal MPI and MPI+ threads computing on scalable multicore systems." IJHPCA (Submitted) (2011).
3344606c0280SSatish 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.
3345217d3b1eSJunchao Zhang 
33461cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `PCFactorSetMatSolverType()`, `MatSolverType`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`, `KSPGetPC()`, `PCFactorGetMatrix()`
334724b6179bSKris Buschelman M*/
334824b6179bSKris Buschelman 
3349d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatFactorGetSolverType_mumps(Mat A, MatSolverType *type)
3350d71ae5a4SJacob Faibussowitsch {
335135bd34faSBarry Smith   PetscFunctionBegin;
33522692d6eeSBarry Smith   *type = MATSOLVERMUMPS;
33533ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
335435bd34faSBarry Smith }
335535bd34faSBarry Smith 
3356bccb9932SShri Abhyankar /* MatGetFactor for Seq and MPI AIJ matrices */
3357d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_aij_mumps(Mat A, MatFactorType ftype, Mat *F)
3358d71ae5a4SJacob Faibussowitsch {
33592877fffaSHong Zhang   Mat         B;
33602877fffaSHong Zhang   Mat_MUMPS  *mumps;
3361c3e1b152SPierre Jolivet   PetscBool   isSeqAIJ, isDiag;
33622c7c0729SBarry Smith   PetscMPIInt size;
33632877fffaSHong Zhang 
33642877fffaSHong Zhang   PetscFunctionBegin;
3365eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
336603e5aca4SStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) {
336703e5aca4SStefano Zampini     PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n"));
336803e5aca4SStefano Zampini     *F = NULL;
336903e5aca4SStefano Zampini     PetscFunctionReturn(PETSC_SUCCESS);
337003e5aca4SStefano Zampini   }
3371eb1ec7c1SStefano Zampini #endif
33722877fffaSHong Zhang   /* Create the factorization matrix */
33739566063dSJacob Faibussowitsch   PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATSEQAIJ, &isSeqAIJ));
3374c3e1b152SPierre Jolivet   PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATDIAGONAL, &isDiag));
33759566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
33769566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
33779566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name));
33789566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
33792877fffaSHong Zhang 
33804dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
33812205254eSKarl Rupp 
33822877fffaSHong Zhang   B->ops->view    = MatView_MUMPS;
338335bd34faSBarry Smith   B->ops->getinfo = MatGetInfo_MUMPS;
33842205254eSKarl Rupp 
33859566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
33869566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
33879566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
33889566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
33899566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
33909566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
33919566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
33929566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
33939566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
33949566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
33959566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
33965c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
33979566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
33989566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
33996444a565SStefano Zampini 
3400450b117fSShri Abhyankar   if (ftype == MAT_FACTOR_LU) {
3401450b117fSShri Abhyankar     B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS;
3402d5f3da31SBarry Smith     B->factortype            = MAT_FACTOR_LU;
3403bccb9932SShri Abhyankar     if (isSeqAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqaij_seqaij;
3404c3e1b152SPierre Jolivet     else if (isDiag) mumps->ConvertToTriples = MatConvertToTriples_diagonal_xaij;
3405bccb9932SShri Abhyankar     else mumps->ConvertToTriples = MatConvertToTriples_mpiaij_mpiaij;
34069566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU]));
3407746480a1SHong Zhang     mumps->sym = 0;
3408dcd589f8SShri Abhyankar   } else {
340967877ebaSShri Abhyankar     B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS;
3410450b117fSShri Abhyankar     B->factortype                  = MAT_FACTOR_CHOLESKY;
3411bccb9932SShri Abhyankar     if (isSeqAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqaij_seqsbaij;
3412c3e1b152SPierre Jolivet     else if (isDiag) mumps->ConvertToTriples = MatConvertToTriples_diagonal_xaij;
3413bccb9932SShri Abhyankar     else mumps->ConvertToTriples = MatConvertToTriples_mpiaij_mpisbaij;
34149566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_CHOLESKY]));
341559ac8732SStefano Zampini #if defined(PETSC_USE_COMPLEX)
341659ac8732SStefano Zampini     mumps->sym = 2;
341759ac8732SStefano Zampini #else
3418b94d7dedSBarry Smith     if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1;
34196fdc2a6dSBarry Smith     else mumps->sym = 2;
342059ac8732SStefano Zampini #endif
3421450b117fSShri Abhyankar   }
34222877fffaSHong Zhang 
342300c67f3bSHong Zhang   /* set solvertype */
34249566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
34259566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
34269566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
34272c7c0729SBarry Smith   if (size == 1) {
34284ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
3429f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
34302c7c0729SBarry Smith   }
34312877fffaSHong Zhang   B->ops->destroy = MatDestroy_MUMPS;
3432e69c285eSBarry Smith   B->data         = (void *)mumps;
34332205254eSKarl Rupp 
34342877fffaSHong Zhang   *F               = B;
3435413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3436413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3437413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3438d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
34393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
34402877fffaSHong Zhang }
34412877fffaSHong Zhang 
3442bccb9932SShri Abhyankar /* MatGetFactor for Seq and MPI SBAIJ matrices */
3443d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_sbaij_mumps(Mat A, MatFactorType ftype, Mat *F)
3444d71ae5a4SJacob Faibussowitsch {
34452877fffaSHong Zhang   Mat         B;
34462877fffaSHong Zhang   Mat_MUMPS  *mumps;
3447ace3abfcSBarry Smith   PetscBool   isSeqSBAIJ;
34482c7c0729SBarry Smith   PetscMPIInt size;
34492877fffaSHong Zhang 
34502877fffaSHong Zhang   PetscFunctionBegin;
3451eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
345203e5aca4SStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) {
345303e5aca4SStefano Zampini     PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n"));
345403e5aca4SStefano Zampini     *F = NULL;
345503e5aca4SStefano Zampini     PetscFunctionReturn(PETSC_SUCCESS);
345603e5aca4SStefano Zampini   }
3457eb1ec7c1SStefano Zampini #endif
34589566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
34599566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
34609566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name));
34619566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
3462e69c285eSBarry Smith 
34634dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
34649566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQSBAIJ, &isSeqSBAIJ));
3465bccb9932SShri Abhyankar   if (isSeqSBAIJ) {
346616ebf90aSShri Abhyankar     mumps->ConvertToTriples = MatConvertToTriples_seqsbaij_seqsbaij;
3467dcd589f8SShri Abhyankar   } else {
3468bccb9932SShri Abhyankar     mumps->ConvertToTriples = MatConvertToTriples_mpisbaij_mpisbaij;
3469bccb9932SShri Abhyankar   }
3470bccb9932SShri Abhyankar 
347167877ebaSShri Abhyankar   B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS;
3472bccb9932SShri Abhyankar   B->ops->view                   = MatView_MUMPS;
3473722b6324SPierre Jolivet   B->ops->getinfo                = MatGetInfo_MUMPS;
34742205254eSKarl Rupp 
34759566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
34769566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
34779566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
34789566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
34799566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
34809566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
34819566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
34829566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
34839566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
34849566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
34859566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
34865c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
34879566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
34889566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
34892205254eSKarl Rupp 
3490f4762488SHong Zhang   B->factortype = MAT_FACTOR_CHOLESKY;
349159ac8732SStefano Zampini #if defined(PETSC_USE_COMPLEX)
349259ac8732SStefano Zampini   mumps->sym = 2;
349359ac8732SStefano Zampini #else
3494b94d7dedSBarry Smith   if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1;
34956fdc2a6dSBarry Smith   else mumps->sym = 2;
349659ac8732SStefano Zampini #endif
3497a214ac2aSShri Abhyankar 
349800c67f3bSHong Zhang   /* set solvertype */
34999566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
35009566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
35019566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
35022c7c0729SBarry Smith   if (size == 1) {
35034ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
3504f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
35052c7c0729SBarry Smith   }
35069566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_CHOLESKY]));
3507f3c0ef26SHong Zhang   B->ops->destroy = MatDestroy_MUMPS;
3508e69c285eSBarry Smith   B->data         = (void *)mumps;
35092205254eSKarl Rupp 
35102877fffaSHong Zhang   *F               = B;
3511413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3512413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3513413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3514d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
35153ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
35162877fffaSHong Zhang }
351797969023SHong Zhang 
3518d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_baij_mumps(Mat A, MatFactorType ftype, Mat *F)
3519d71ae5a4SJacob Faibussowitsch {
352067877ebaSShri Abhyankar   Mat         B;
352167877ebaSShri Abhyankar   Mat_MUMPS  *mumps;
3522ace3abfcSBarry Smith   PetscBool   isSeqBAIJ;
35232c7c0729SBarry Smith   PetscMPIInt size;
352467877ebaSShri Abhyankar 
352567877ebaSShri Abhyankar   PetscFunctionBegin;
352667877ebaSShri Abhyankar   /* Create the factorization matrix */
35279566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQBAIJ, &isSeqBAIJ));
35289566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
35299566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
35309566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name));
35319566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
3532450b117fSShri Abhyankar 
35334dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
3534450b117fSShri Abhyankar   if (ftype == MAT_FACTOR_LU) {
3535450b117fSShri Abhyankar     B->ops->lufactorsymbolic = MatLUFactorSymbolic_BAIJMUMPS;
3536450b117fSShri Abhyankar     B->factortype            = MAT_FACTOR_LU;
3537bccb9932SShri Abhyankar     if (isSeqBAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqbaij_seqaij;
3538bccb9932SShri Abhyankar     else mumps->ConvertToTriples = MatConvertToTriples_mpibaij_mpiaij;
3539746480a1SHong Zhang     mumps->sym = 0;
35409566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU]));
3541546078acSJacob Faibussowitsch   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot use PETSc BAIJ matrices with MUMPS Cholesky, use SBAIJ or AIJ matrix instead");
3542bccb9932SShri Abhyankar 
3543450b117fSShri Abhyankar   B->ops->view    = MatView_MUMPS;
3544722b6324SPierre Jolivet   B->ops->getinfo = MatGetInfo_MUMPS;
35452205254eSKarl Rupp 
35469566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
35479566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
35489566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
35499566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
35509566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
35519566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
35529566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
35539566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
35549566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
35559566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
35569566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
35575c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
35589566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
35599566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
3560450b117fSShri Abhyankar 
356100c67f3bSHong Zhang   /* set solvertype */
35629566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
35639566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
35649566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
35652c7c0729SBarry Smith   if (size == 1) {
35664ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
3567f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
35682c7c0729SBarry Smith   }
35697ee00b23SStefano Zampini   B->ops->destroy = MatDestroy_MUMPS;
35707ee00b23SStefano Zampini   B->data         = (void *)mumps;
35717ee00b23SStefano Zampini 
35727ee00b23SStefano Zampini   *F               = B;
3573413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3574413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3575413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3576d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
35773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
35787ee00b23SStefano Zampini }
35797ee00b23SStefano Zampini 
35807ee00b23SStefano Zampini /* MatGetFactor for Seq and MPI SELL matrices */
3581d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_sell_mumps(Mat A, MatFactorType ftype, Mat *F)
3582d71ae5a4SJacob Faibussowitsch {
35837ee00b23SStefano Zampini   Mat         B;
35847ee00b23SStefano Zampini   Mat_MUMPS  *mumps;
35857ee00b23SStefano Zampini   PetscBool   isSeqSELL;
35862c7c0729SBarry Smith   PetscMPIInt size;
35877ee00b23SStefano Zampini 
35887ee00b23SStefano Zampini   PetscFunctionBegin;
35897ee00b23SStefano Zampini   /* Create the factorization matrix */
35909566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQSELL, &isSeqSELL));
35919566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
35929566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
35939566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name));
35949566063dSJacob Faibussowitsch   PetscCall(MatSetUp(B));
35957ee00b23SStefano Zampini 
35964dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mumps));
35977ee00b23SStefano Zampini 
35987ee00b23SStefano Zampini   B->ops->view    = MatView_MUMPS;
35997ee00b23SStefano Zampini   B->ops->getinfo = MatGetInfo_MUMPS;
36007ee00b23SStefano Zampini 
36019566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
36029566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
36039566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
36049566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
36059566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
36069566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
36079566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
36089566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
36099566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
36109566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
36119566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
36125c0bae8cSAshish Patel   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
36137ee00b23SStefano Zampini 
36147ee00b23SStefano Zampini   if (ftype == MAT_FACTOR_LU) {
36157ee00b23SStefano Zampini     B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS;
36167ee00b23SStefano Zampini     B->factortype            = MAT_FACTOR_LU;
36177ee00b23SStefano Zampini     if (isSeqSELL) mumps->ConvertToTriples = MatConvertToTriples_seqsell_seqaij;
36187ee00b23SStefano Zampini     else SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "To be implemented");
36197ee00b23SStefano Zampini     mumps->sym = 0;
36209566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU]));
36217ee00b23SStefano Zampini   } else SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "To be implemented");
36227ee00b23SStefano Zampini 
36237ee00b23SStefano Zampini   /* set solvertype */
36249566063dSJacob Faibussowitsch   PetscCall(PetscFree(B->solvertype));
36259566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
36269566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
36272c7c0729SBarry Smith   if (size == 1) {
36284ac6704cSBarry Smith     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization  */
3629f73b0415SBarry Smith     B->canuseordering = PETSC_TRUE;
36302c7c0729SBarry Smith   }
3631450b117fSShri Abhyankar   B->ops->destroy = MatDestroy_MUMPS;
3632e69c285eSBarry Smith   B->data         = (void *)mumps;
36332205254eSKarl Rupp 
3634450b117fSShri Abhyankar   *F               = B;
3635413bcc21SPierre Jolivet   mumps->id.job    = JOB_NULL;
3636413bcc21SPierre Jolivet   mumps->ICNTL_pre = NULL;
3637413bcc21SPierre Jolivet   mumps->CNTL_pre  = NULL;
3638d47f36abSHong Zhang   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
36393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3640450b117fSShri Abhyankar }
364142c9c57cSBarry Smith 
36429d0448ceSStefano Zampini /* MatGetFactor for MATNEST matrices */
36439d0448ceSStefano Zampini static PetscErrorCode MatGetFactor_nest_mumps(Mat A, MatFactorType ftype, Mat *F)
36449d0448ceSStefano Zampini {
36459d0448ceSStefano Zampini   Mat         B, **mats;
36469d0448ceSStefano Zampini   Mat_MUMPS  *mumps;
36479d0448ceSStefano Zampini   PetscInt    nr, nc;
36489d0448ceSStefano Zampini   PetscMPIInt size;
364903e5aca4SStefano Zampini   PetscBool   flg = PETSC_TRUE;
36509d0448ceSStefano Zampini 
36519d0448ceSStefano Zampini   PetscFunctionBegin;
36529d0448ceSStefano Zampini #if defined(PETSC_USE_COMPLEX)
365303e5aca4SStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) {
365403e5aca4SStefano Zampini     PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n"));
365503e5aca4SStefano Zampini     *F = NULL;
365603e5aca4SStefano Zampini     PetscFunctionReturn(PETSC_SUCCESS);
365703e5aca4SStefano Zampini   }
36589d0448ceSStefano Zampini #endif
36599d0448ceSStefano Zampini 
366003e5aca4SStefano Zampini   /* Return if some condition is not satisfied */
366103e5aca4SStefano Zampini   *F = NULL;
36629d0448ceSStefano Zampini   PetscCall(MatNestGetSubMats(A, &nr, &nc, &mats));
36639d0448ceSStefano Zampini   if (ftype == MAT_FACTOR_CHOLESKY) {
36649d0448ceSStefano Zampini     IS       *rows, *cols;
36659d0448ceSStefano Zampini     PetscInt *m, *M;
36669d0448ceSStefano Zampini 
36679d0448ceSStefano 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);
36689d0448ceSStefano Zampini     PetscCall(PetscMalloc2(nr, &rows, nc, &cols));
36699d0448ceSStefano Zampini     PetscCall(MatNestGetISs(A, rows, cols));
36709d0448ceSStefano Zampini     for (PetscInt r = 0; flg && r < nr; r++) PetscCall(ISEqualUnsorted(rows[r], cols[r], &flg));
367103e5aca4SStefano Zampini     if (!flg) {
367203e5aca4SStefano Zampini       PetscCall(PetscFree2(rows, cols));
367303e5aca4SStefano Zampini       PetscCall(PetscInfo(A, "MAT_FACTOR_CHOLESKY not supported for unequal row and column maps. Use MAT_FACTOR_LU.\n"));
367403e5aca4SStefano Zampini       PetscFunctionReturn(PETSC_SUCCESS);
367503e5aca4SStefano Zampini     }
36769d0448ceSStefano Zampini     PetscCall(PetscMalloc2(nr, &m, nr, &M));
36779d0448ceSStefano Zampini     for (PetscInt r = 0; r < nr; r++) PetscCall(ISGetMinMax(rows[r], &m[r], &M[r]));
36789d0448ceSStefano Zampini     for (PetscInt r = 0; flg && r < nr; r++)
36799d0448ceSStefano Zampini       for (PetscInt k = r + 1; flg && k < nr; k++)
36809d0448ceSStefano Zampini         if ((m[k] <= m[r] && m[r] <= M[k]) || (m[k] <= M[r] && M[r] <= M[k])) flg = PETSC_FALSE;
36819d0448ceSStefano Zampini     PetscCall(PetscFree2(m, M));
36829d0448ceSStefano Zampini     PetscCall(PetscFree2(rows, cols));
368303e5aca4SStefano Zampini     if (!flg) {
368403e5aca4SStefano Zampini       PetscCall(PetscInfo(A, "MAT_FACTOR_CHOLESKY not supported for intersecting row maps. Use MAT_FACTOR_LU.\n"));
368503e5aca4SStefano Zampini       PetscFunctionReturn(PETSC_SUCCESS);
368603e5aca4SStefano Zampini     }
36879d0448ceSStefano Zampini   }
36889d0448ceSStefano Zampini 
36899d0448ceSStefano Zampini   for (PetscInt r = 0; r < nr; r++) {
36909d0448ceSStefano Zampini     for (PetscInt c = 0; c < nc; c++) {
36919d0448ceSStefano Zampini       Mat       sub = mats[r][c];
3692c3e1b152SPierre Jolivet       PetscBool isSeqAIJ, isMPIAIJ, isSeqBAIJ, isMPIBAIJ, isSeqSBAIJ, isMPISBAIJ, isTrans, isDiag;
36939d0448ceSStefano Zampini 
36949d0448ceSStefano Zampini       if (!sub || (ftype == MAT_FACTOR_CHOLESKY && c < r)) continue;
36955d955bbbSStefano Zampini       PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATTRANSPOSEVIRTUAL, &isTrans));
36965d955bbbSStefano Zampini       if (isTrans) PetscCall(MatTransposeGetMat(sub, &sub));
36975d955bbbSStefano Zampini       else {
36985d955bbbSStefano Zampini         PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATHERMITIANTRANSPOSEVIRTUAL, &isTrans));
36995d955bbbSStefano Zampini         if (isTrans) PetscCall(MatHermitianTransposeGetMat(sub, &sub));
37005d955bbbSStefano Zampini       }
37019d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQAIJ, &isSeqAIJ));
37029d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIAIJ, &isMPIAIJ));
37039d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQBAIJ, &isSeqBAIJ));
37049d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIBAIJ, &isMPIBAIJ));
37059d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQSBAIJ, &isSeqSBAIJ));
37069d0448ceSStefano Zampini       PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPISBAIJ, &isMPISBAIJ));
3707c3e1b152SPierre Jolivet       PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATDIAGONAL, &isDiag));
37089d0448ceSStefano Zampini       if (ftype == MAT_FACTOR_CHOLESKY) {
3709dcab004fSPierre Jolivet         if (r == c) {
3710c3e1b152SPierre Jolivet           if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isSeqSBAIJ && !isMPISBAIJ && !isDiag) {
371103e5aca4SStefano Zampini             PetscCall(PetscInfo(sub, "MAT_CHOLESKY_FACTOR not supported for diagonal block of type %s.\n", ((PetscObject)sub)->type_name));
371203e5aca4SStefano Zampini             flg = PETSC_FALSE;
3713dcab004fSPierre Jolivet           }
3714c3e1b152SPierre Jolivet         } else if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isDiag) {
371503e5aca4SStefano Zampini           PetscCall(PetscInfo(sub, "MAT_CHOLESKY_FACTOR not supported for off-diagonal block of type %s.\n", ((PetscObject)sub)->type_name));
371603e5aca4SStefano Zampini           flg = PETSC_FALSE;
371703e5aca4SStefano Zampini         }
3718c3e1b152SPierre Jolivet       } else if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isDiag) {
371903e5aca4SStefano Zampini         PetscCall(PetscInfo(sub, "MAT_LU_FACTOR not supported for block of type %s.\n", ((PetscObject)sub)->type_name));
372003e5aca4SStefano Zampini         flg = PETSC_FALSE;
37219d0448ceSStefano Zampini       }
37229d0448ceSStefano Zampini     }
372303e5aca4SStefano Zampini   }
372403e5aca4SStefano Zampini   if (!flg) PetscFunctionReturn(PETSC_SUCCESS);
37259d0448ceSStefano Zampini 
37269d0448ceSStefano Zampini   /* Create the factorization matrix */
37279d0448ceSStefano Zampini   PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B));
37289d0448ceSStefano Zampini   PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N));
37299d0448ceSStefano Zampini   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &((PetscObject)B)->type_name));
37309d0448ceSStefano Zampini   PetscCall(MatSetUp(B));
37319d0448ceSStefano Zampini 
37329d0448ceSStefano Zampini   PetscCall(PetscNew(&mumps));
37339d0448ceSStefano Zampini 
37349d0448ceSStefano Zampini   B->ops->view    = MatView_MUMPS;
37359d0448ceSStefano Zampini   B->ops->getinfo = MatGetInfo_MUMPS;
37369d0448ceSStefano Zampini 
37379d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps));
37389d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS));
37399d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS));
37409d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS));
37419d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS));
37429d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS));
37439d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS));
37449d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS));
37459d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS));
37469d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS));
37479d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS));
37489d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS));
37499d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS));
37509d0448ceSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS));
37519d0448ceSStefano Zampini 
37529d0448ceSStefano Zampini   if (ftype == MAT_FACTOR_LU) {
37539d0448ceSStefano Zampini     B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS;
37549d0448ceSStefano Zampini     B->factortype            = MAT_FACTOR_LU;
37559d0448ceSStefano Zampini     mumps->sym               = 0;
37569d0448ceSStefano Zampini   } else {
37579d0448ceSStefano Zampini     B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS;
37589d0448ceSStefano Zampini     B->factortype                  = MAT_FACTOR_CHOLESKY;
37599d0448ceSStefano Zampini #if defined(PETSC_USE_COMPLEX)
37609d0448ceSStefano Zampini     mumps->sym = 2;
37619d0448ceSStefano Zampini #else
37629d0448ceSStefano Zampini     if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1;
37639d0448ceSStefano Zampini     else mumps->sym = 2;
37649d0448ceSStefano Zampini #endif
37659d0448ceSStefano Zampini   }
37669d0448ceSStefano Zampini   mumps->ConvertToTriples = MatConvertToTriples_nest_xaij;
37679d0448ceSStefano Zampini   PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[ftype]));
37689d0448ceSStefano Zampini 
37699d0448ceSStefano Zampini   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size));
37709d0448ceSStefano Zampini   if (size == 1) {
37719d0448ceSStefano Zampini     /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */
37729d0448ceSStefano Zampini     B->canuseordering = PETSC_TRUE;
37739d0448ceSStefano Zampini   }
37749d0448ceSStefano Zampini 
37759d0448ceSStefano Zampini   /* set solvertype */
37769d0448ceSStefano Zampini   PetscCall(PetscFree(B->solvertype));
37779d0448ceSStefano Zampini   PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype));
37789d0448ceSStefano Zampini   B->ops->destroy = MatDestroy_MUMPS;
37799d0448ceSStefano Zampini   B->data         = (void *)mumps;
37809d0448ceSStefano Zampini 
37819d0448ceSStefano Zampini   *F               = B;
37829d0448ceSStefano Zampini   mumps->id.job    = JOB_NULL;
37839d0448ceSStefano Zampini   mumps->ICNTL_pre = NULL;
37849d0448ceSStefano Zampini   mumps->CNTL_pre  = NULL;
37859d0448ceSStefano Zampini   mumps->matstruc  = DIFFERENT_NONZERO_PATTERN;
37869d0448ceSStefano Zampini   PetscFunctionReturn(PETSC_SUCCESS);
37879d0448ceSStefano Zampini }
37889d0448ceSStefano Zampini 
3789d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_MUMPS(void)
3790d71ae5a4SJacob Faibussowitsch {
379142c9c57cSBarry Smith   PetscFunctionBegin;
37929566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIAIJ, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
37939566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
37949566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIBAIJ, MAT_FACTOR_LU, MatGetFactor_baij_mumps));
37959566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_baij_mumps));
37969566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPISBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_sbaij_mumps));
37979566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQAIJ, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
37989566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
37999566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQBAIJ, MAT_FACTOR_LU, MatGetFactor_baij_mumps));
38009566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_baij_mumps));
38019566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQSBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_sbaij_mumps));
38029566063dSJacob Faibussowitsch   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQSELL, MAT_FACTOR_LU, MatGetFactor_sell_mumps));
3803c3e1b152SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATDIAGONAL, MAT_FACTOR_LU, MatGetFactor_aij_mumps));
3804c3e1b152SPierre Jolivet   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATDIAGONAL, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps));
38059d0448ceSStefano Zampini   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATNEST, MAT_FACTOR_LU, MatGetFactor_nest_mumps));
38069d0448ceSStefano Zampini   PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATNEST, MAT_FACTOR_CHOLESKY, MatGetFactor_nest_mumps));
38073ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
380842c9c57cSBarry Smith }
3809