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; 397127cd276SPierre 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 */ 398127cd276SPierre 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; 421127cd276SPierre 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 */ 422127cd276SPierre 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; 458127cd276SPierre 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 */ 459127cd276SPierre 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); 532127cd276SPierre Jolivet } else if (mumps->nest_vals) 533127cd276SPierre 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 */ 534127cd276SPierre 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; 1010127cd276SPierre 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 */ 1011127cd276SPierre 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 if (mumps->id.ICNTL(26) < 0 || mumps->id.ICNTL(26) > 2) { 1579e94cce23SStefano Zampini second_solve = PETSC_TRUE; 15809566063dSJacob Faibussowitsch PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE)); 15813e5b40d0SPierre Jolivet mumps->id.ICNTL(26) = 1; /* condensation phase */ 15823e5b40d0SPierre Jolivet } else if (mumps->id.ICNTL(26) == 1) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_FALSE)); 1583e94cce23SStefano Zampini } 15842cd7d884SHong Zhang /* solve phase */ 15852cd7d884SHong Zhang mumps->id.job = JOB_SOLVE; 15863ab56b82SJunchao Zhang PetscMUMPS_c(mumps); 15879261f6e4SBarry 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)); 1588b5fa320bSStefano Zampini 1589b5fa320bSStefano Zampini /* handle expansion step of Schur complement (if any) */ 15901baa6e33SBarry Smith if (second_solve) PetscCall(MatMumpsHandleSchur_Private(A, PETSC_TRUE)); 15913e5b40d0SPierre Jolivet else if (mumps->id.ICNTL(26) == 1) { 15923e5b40d0SPierre Jolivet PetscCall(MatMumpsSolveSchur_Private(A)); 15933e5b40d0SPierre Jolivet for (j = 0; j < nrhs; ++j) 15943e5b40d0SPierre Jolivet for (i = 0; i < mumps->id.size_schur; ++i) { 15953e5b40d0SPierre Jolivet #if !defined(PETSC_USE_COMPLEX) 15963e5b40d0SPierre Jolivet PetscScalar val = mumps->id.redrhs[i + j * mumps->id.lredrhs]; 15973e5b40d0SPierre Jolivet #else 15983e5b40d0SPierre Jolivet PetscScalar val = mumps->id.redrhs[i + j * mumps->id.lredrhs].r + PETSC_i * mumps->id.redrhs[i + j * mumps->id.lredrhs].i; 15993e5b40d0SPierre Jolivet #endif 16003e5b40d0SPierre Jolivet array[mumps->id.listvar_schur[i] - 1 + j * M] = val; 16013e5b40d0SPierre Jolivet } 16023e5b40d0SPierre Jolivet } 1603a6053eceSJunchao Zhang if (!denseB) { /* sparse B */ 16049566063dSJacob Faibussowitsch PetscCall(MatSeqAIJRestoreArray(Bt, &aa)); 16059566063dSJacob Faibussowitsch PetscCall(MatRestoreRowIJ(Bt, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg)); 160628b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot restore IJ structure"); 1607b8491c3eSStefano Zampini } 16089566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(X, &array)); 16093ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1610be818407SHong Zhang } 1611801fbe65SHong Zhang 16122ef1f0ffSBarry Smith /* parallel case: MUMPS requires rhs B to be centralized on the host! */ 161350a7cd33SPierre Jolivet PetscCheck(!mumps->id.ICNTL(19), PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Parallel Schur complements not yet supported from PETSc"); 1614241dbb5eSStefano Zampini 1615beae5ec0SHong Zhang /* create msol_loc to hold mumps local solution */ 16161683a169SBarry Smith isol_loc_save = mumps->id.isol_loc; /* save it for MatSolve() */ 16171683a169SBarry Smith sol_loc_save = (PetscScalar *)mumps->id.sol_loc; 1618801fbe65SHong Zhang 1619a1dfcbd9SJunchao Zhang lsol_loc = mumps->id.lsol_loc; 162071aed81dSHong Zhang nlsol_loc = nrhs * lsol_loc; /* length of sol_loc */ 16219566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(nlsol_loc, &sol_loc, lsol_loc, &isol_loc)); 1622940cd9d6SSatish Balay mumps->id.sol_loc = (MumpsScalar *)sol_loc; 1623801fbe65SHong Zhang mumps->id.isol_loc = isol_loc; 1624801fbe65SHong Zhang 16259566063dSJacob Faibussowitsch PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, nlsol_loc, (PetscScalar *)sol_loc, &msol_loc)); 16262cd7d884SHong Zhang 162767602552SJunchao Zhang if (denseB) { 162825aac85cSJunchao Zhang if (mumps->ICNTL20 == 10) { 162967602552SJunchao Zhang mumps->id.ICNTL(20) = 10; /* dense distributed RHS */ 16309566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(B, &rbray)); 16319566063dSJacob Faibussowitsch PetscCall(MatMumpsSetUpDistRHSInfo(A, nrhs, rbray)); 16329566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(B, &rbray)); 16339566063dSJacob Faibussowitsch PetscCall(MatGetLocalSize(B, &m, NULL)); 16349566063dSJacob Faibussowitsch PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)B), 1, nrhs * m, nrhs * M, NULL, &v_mpi)); 163525aac85cSJunchao Zhang } else { 163625aac85cSJunchao Zhang mumps->id.ICNTL(20) = 0; /* dense centralized RHS */ 163780577c12SJunchao Zhang /* TODO: Because of non-contiguous indices, the created vecscatter scat_rhs is not done in MPI_Gather, resulting in 163880577c12SJunchao Zhang very inefficient communication. An optimization is to use VecScatterCreateToZero to gather B to rank 0. Then on rank 163980577c12SJunchao Zhang 0, re-arrange B into desired order, which is a local operation. 164080577c12SJunchao Zhang */ 164180577c12SJunchao Zhang 164267602552SJunchao Zhang /* scatter v_mpi to b_seq because MUMPS before 5.3.0 only supports centralized rhs */ 1643be818407SHong Zhang /* wrap dense rhs matrix B into a vector v_mpi */ 16449566063dSJacob Faibussowitsch PetscCall(MatGetLocalSize(B, &m, NULL)); 16459566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(B, &bray)); 16469566063dSJacob Faibussowitsch PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)B), 1, nrhs * m, nrhs * M, (const PetscScalar *)bray, &v_mpi)); 16479566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(B, &bray)); 16482b691707SHong Zhang 1649be818407SHong Zhang /* scatter v_mpi to b_seq in proc[0]. MUMPS requires rhs to be centralized on the host! */ 1650801fbe65SHong Zhang if (!mumps->myid) { 1651beae5ec0SHong Zhang PetscInt *idx; 1652beae5ec0SHong Zhang /* idx: maps from k-th index of v_mpi to (i,j)-th global entry of B */ 16539566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(nrhs * M, &idx)); 16549566063dSJacob Faibussowitsch PetscCall(MatGetOwnershipRanges(B, &rstart)); 1655be818407SHong Zhang k = 0; 16562d4298aeSJunchao Zhang for (proc = 0; proc < mumps->petsc_size; proc++) { 1657be818407SHong Zhang for (j = 0; j < nrhs; j++) { 1658beae5ec0SHong Zhang for (i = rstart[proc]; i < rstart[proc + 1]; i++) idx[k++] = j * M + i; 1659be818407SHong Zhang } 1660be818407SHong Zhang } 1661be818407SHong Zhang 16629566063dSJacob Faibussowitsch PetscCall(VecCreateSeq(PETSC_COMM_SELF, nrhs * M, &b_seq)); 16639566063dSJacob Faibussowitsch PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nrhs * M, idx, PETSC_OWN_POINTER, &is_to)); 16649566063dSJacob Faibussowitsch PetscCall(ISCreateStride(PETSC_COMM_SELF, nrhs * M, 0, 1, &is_from)); 1665801fbe65SHong Zhang } else { 16669566063dSJacob Faibussowitsch PetscCall(VecCreateSeq(PETSC_COMM_SELF, 0, &b_seq)); 16679566063dSJacob Faibussowitsch PetscCall(ISCreateStride(PETSC_COMM_SELF, 0, 0, 1, &is_to)); 16689566063dSJacob Faibussowitsch PetscCall(ISCreateStride(PETSC_COMM_SELF, 0, 0, 1, &is_from)); 1669801fbe65SHong Zhang } 16709566063dSJacob Faibussowitsch PetscCall(VecScatterCreate(v_mpi, is_from, b_seq, is_to, &scat_rhs)); 16719566063dSJacob Faibussowitsch PetscCall(VecScatterBegin(scat_rhs, v_mpi, b_seq, INSERT_VALUES, SCATTER_FORWARD)); 16729566063dSJacob Faibussowitsch PetscCall(ISDestroy(&is_to)); 16739566063dSJacob Faibussowitsch PetscCall(ISDestroy(&is_from)); 16749566063dSJacob Faibussowitsch PetscCall(VecScatterEnd(scat_rhs, v_mpi, b_seq, INSERT_VALUES, SCATTER_FORWARD)); 1675801fbe65SHong Zhang 1676801fbe65SHong Zhang if (!mumps->myid) { /* define rhs on the host */ 16779566063dSJacob Faibussowitsch PetscCall(VecGetArray(b_seq, &bray)); 1678940cd9d6SSatish Balay mumps->id.rhs = (MumpsScalar *)bray; 16799566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(b_seq, &bray)); 1680801fbe65SHong Zhang } 168125aac85cSJunchao Zhang } 16822b691707SHong Zhang } else { /* sparse B */ 16832b691707SHong Zhang b = (Mat_MPIAIJ *)Bt->data; 16842b691707SHong Zhang 1685be818407SHong Zhang /* wrap dense X into a vector v_mpi */ 16869566063dSJacob Faibussowitsch PetscCall(MatGetLocalSize(X, &m, NULL)); 16879566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(X, &bray)); 16889566063dSJacob Faibussowitsch PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)X), 1, nrhs * m, nrhs * M, (const PetscScalar *)bray, &v_mpi)); 16899566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(X, &bray)); 16902b691707SHong Zhang 16912b691707SHong Zhang if (!mumps->myid) { 16929566063dSJacob Faibussowitsch PetscCall(MatSeqAIJGetArray(b->A, &aa)); 16939566063dSJacob Faibussowitsch PetscCall(MatGetRowIJ(b->A, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg)); 169428b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure"); 16959566063dSJacob Faibussowitsch PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs)); 16962b691707SHong Zhang mumps->id.rhs_sparse = (MumpsScalar *)aa; 16972b691707SHong Zhang } else { 16982b691707SHong Zhang mumps->id.irhs_ptr = NULL; 16992b691707SHong Zhang mumps->id.irhs_sparse = NULL; 17002b691707SHong Zhang mumps->id.nz_rhs = 0; 17012b691707SHong Zhang mumps->id.rhs_sparse = NULL; 17022b691707SHong Zhang } 17032b691707SHong Zhang } 17042b691707SHong Zhang 1705801fbe65SHong Zhang /* solve phase */ 1706801fbe65SHong Zhang mumps->id.job = JOB_SOLVE; 17073ab56b82SJunchao Zhang PetscMUMPS_c(mumps); 17089261f6e4SBarry 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)); 1709801fbe65SHong Zhang 1710334c5f61SHong Zhang /* scatter mumps distributed solution to petsc vector v_mpi, which shares local arrays with solution matrix X */ 17119566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(X, &array)); 17129566063dSJacob Faibussowitsch PetscCall(VecPlaceArray(v_mpi, array)); 1713801fbe65SHong Zhang 1714334c5f61SHong Zhang /* create scatter scat_sol */ 17159566063dSJacob Faibussowitsch PetscCall(MatGetOwnershipRanges(X, &rstart)); 1716beae5ec0SHong Zhang /* iidx: index for scatter mumps solution to petsc X */ 1717beae5ec0SHong Zhang 17189566063dSJacob Faibussowitsch PetscCall(ISCreateStride(PETSC_COMM_SELF, nlsol_loc, 0, 1, &is_from)); 17199566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(nlsol_loc, &idxx)); 1720beae5ec0SHong Zhang for (i = 0; i < lsol_loc; i++) { 1721beae5ec0SHong 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 */ 1722beae5ec0SHong Zhang 17232d4298aeSJunchao Zhang for (proc = 0; proc < mumps->petsc_size; proc++) { 1724beae5ec0SHong Zhang if (isol_loc[i] >= rstart[proc] && isol_loc[i] < rstart[proc + 1]) { 1725beae5ec0SHong Zhang myrstart = rstart[proc]; 1726beae5ec0SHong Zhang k = isol_loc[i] - myrstart; /* local index on 1st column of petsc vector X */ 1727beae5ec0SHong Zhang iidx = k + myrstart * nrhs; /* maps mumps isol_loc[i] to petsc index in X */ 1728beae5ec0SHong Zhang m = rstart[proc + 1] - rstart[proc]; /* rows of X for this proc */ 1729beae5ec0SHong Zhang break; 1730be818407SHong Zhang } 1731be818407SHong Zhang } 1732be818407SHong Zhang 1733beae5ec0SHong Zhang for (j = 0; j < nrhs; j++) idxx[i + j * lsol_loc] = iidx + j * m; 1734801fbe65SHong Zhang } 17359566063dSJacob Faibussowitsch PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nlsol_loc, idxx, PETSC_COPY_VALUES, &is_to)); 17369566063dSJacob Faibussowitsch PetscCall(VecScatterCreate(msol_loc, is_from, v_mpi, is_to, &scat_sol)); 17379566063dSJacob Faibussowitsch PetscCall(VecScatterBegin(scat_sol, msol_loc, v_mpi, INSERT_VALUES, SCATTER_FORWARD)); 17389566063dSJacob Faibussowitsch PetscCall(ISDestroy(&is_from)); 17399566063dSJacob Faibussowitsch PetscCall(ISDestroy(&is_to)); 17409566063dSJacob Faibussowitsch PetscCall(VecScatterEnd(scat_sol, msol_loc, v_mpi, INSERT_VALUES, SCATTER_FORWARD)); 17419566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(X, &array)); 174271aed81dSHong Zhang 174371aed81dSHong Zhang /* free spaces */ 17441683a169SBarry Smith mumps->id.sol_loc = (MumpsScalar *)sol_loc_save; 174571aed81dSHong Zhang mumps->id.isol_loc = isol_loc_save; 174671aed81dSHong Zhang 17479566063dSJacob Faibussowitsch PetscCall(PetscFree2(sol_loc, isol_loc)); 17489566063dSJacob Faibussowitsch PetscCall(PetscFree(idxx)); 17499566063dSJacob Faibussowitsch PetscCall(VecDestroy(&msol_loc)); 17509566063dSJacob Faibussowitsch PetscCall(VecDestroy(&v_mpi)); 1751a6053eceSJunchao Zhang if (!denseB) { 17522b691707SHong Zhang if (!mumps->myid) { 1753d56c302dSHong Zhang b = (Mat_MPIAIJ *)Bt->data; 17549566063dSJacob Faibussowitsch PetscCall(MatSeqAIJRestoreArray(b->A, &aa)); 17559566063dSJacob Faibussowitsch PetscCall(MatRestoreRowIJ(b->A, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg)); 175628b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot restore IJ structure"); 17572b691707SHong Zhang } 17582b691707SHong Zhang } else { 175925aac85cSJunchao Zhang if (mumps->ICNTL20 == 0) { 17609566063dSJacob Faibussowitsch PetscCall(VecDestroy(&b_seq)); 17619566063dSJacob Faibussowitsch PetscCall(VecScatterDestroy(&scat_rhs)); 176225aac85cSJunchao Zhang } 17632b691707SHong Zhang } 17649566063dSJacob Faibussowitsch PetscCall(VecScatterDestroy(&scat_sol)); 176564412097SPierre 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)))); 17663ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1767e0b74bf9SHong Zhang } 1768e0b74bf9SHong Zhang 176966976f2fSJacob Faibussowitsch static PetscErrorCode MatMatSolveTranspose_MUMPS(Mat A, Mat B, Mat X) 1770d71ae5a4SJacob Faibussowitsch { 1771b18964edSHong Zhang Mat_MUMPS *mumps = (Mat_MUMPS *)A->data; 1772338d3105SPierre Jolivet const PetscMUMPSInt value = mumps->id.ICNTL(9); 1773b18964edSHong Zhang 1774b18964edSHong Zhang PetscFunctionBegin; 1775b18964edSHong Zhang mumps->id.ICNTL(9) = 0; 1776b18964edSHong Zhang PetscCall(MatMatSolve_MUMPS(A, B, X)); 1777338d3105SPierre Jolivet mumps->id.ICNTL(9) = value; 17783ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1779b18964edSHong Zhang } 1780b18964edSHong Zhang 178166976f2fSJacob Faibussowitsch static PetscErrorCode MatMatTransposeSolve_MUMPS(Mat A, Mat Bt, Mat X) 1782d71ae5a4SJacob Faibussowitsch { 1783eb3ef3b2SHong Zhang PetscBool flg; 1784eb3ef3b2SHong Zhang Mat B; 1785eb3ef3b2SHong Zhang 1786eb3ef3b2SHong Zhang PetscFunctionBegin; 17879566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompareAny((PetscObject)Bt, &flg, MATSEQAIJ, MATMPIAIJ, NULL)); 178828b400f6SJacob Faibussowitsch PetscCheck(flg, PetscObjectComm((PetscObject)Bt), PETSC_ERR_ARG_WRONG, "Matrix Bt must be MATAIJ matrix"); 1789eb3ef3b2SHong Zhang 1790eb3ef3b2SHong Zhang /* Create B=Bt^T that uses Bt's data structure */ 17919566063dSJacob Faibussowitsch PetscCall(MatCreateTranspose(Bt, &B)); 1792eb3ef3b2SHong Zhang 17939566063dSJacob Faibussowitsch PetscCall(MatMatSolve_MUMPS(A, B, X)); 17949566063dSJacob Faibussowitsch PetscCall(MatDestroy(&B)); 17953ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1796eb3ef3b2SHong Zhang } 1797eb3ef3b2SHong Zhang 1798ace3df97SHong Zhang #if !defined(PETSC_USE_COMPLEX) 1799a58c3f20SHong Zhang /* 1800a58c3f20SHong Zhang input: 1801a58c3f20SHong Zhang F: numeric factor 1802a58c3f20SHong Zhang output: 1803a58c3f20SHong Zhang nneg: total number of negative pivots 180419d49a3bSHong Zhang nzero: total number of zero pivots 180519d49a3bSHong Zhang npos: (global dimension of F) - nneg - nzero 1806a58c3f20SHong Zhang */ 180766976f2fSJacob Faibussowitsch static PetscErrorCode MatGetInertia_SBAIJMUMPS(Mat F, PetscInt *nneg, PetscInt *nzero, PetscInt *npos) 1808d71ae5a4SJacob Faibussowitsch { 1809e69c285eSBarry Smith Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 1810c1490034SHong Zhang PetscMPIInt size; 1811a58c3f20SHong Zhang 1812a58c3f20SHong Zhang PetscFunctionBegin; 18139566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)F), &size)); 1814bcb30aebSHong 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 */ 1815aed4548fSBarry 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)); 1816ed85ac9fSHong Zhang 1817710ac8efSHong Zhang if (nneg) *nneg = mumps->id.INFOG(12); 1818ed85ac9fSHong Zhang if (nzero || npos) { 181908401ef6SPierre 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"); 1820710ac8efSHong Zhang if (nzero) *nzero = mumps->id.INFOG(28); 1821710ac8efSHong Zhang if (npos) *npos = F->rmap->N - (mumps->id.INFOG(12) + mumps->id.INFOG(28)); 1822a58c3f20SHong Zhang } 18233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1824a58c3f20SHong Zhang } 182519d49a3bSHong Zhang #endif 1826a58c3f20SHong Zhang 182766976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGatherNonzerosOnMaster(MatReuse reuse, Mat_MUMPS *mumps) 1828d71ae5a4SJacob Faibussowitsch { 1829a6053eceSJunchao Zhang PetscInt i, nreqs; 1830a6053eceSJunchao Zhang PetscMUMPSInt *irn, *jcn; 1831a6053eceSJunchao Zhang PetscMPIInt count; 1832a6053eceSJunchao Zhang PetscInt64 totnnz, remain; 1833a6053eceSJunchao Zhang const PetscInt osize = mumps->omp_comm_size; 1834a6053eceSJunchao Zhang PetscScalar *val; 18353ab56b82SJunchao Zhang 18363ab56b82SJunchao Zhang PetscFunctionBegin; 1837a6053eceSJunchao Zhang if (osize > 1) { 18383ab56b82SJunchao Zhang if (reuse == MAT_INITIAL_MATRIX) { 18393ab56b82SJunchao Zhang /* master first gathers counts of nonzeros to receive */ 18409566063dSJacob Faibussowitsch if (mumps->is_omp_master) PetscCall(PetscMalloc1(osize, &mumps->recvcount)); 18419566063dSJacob Faibussowitsch PetscCallMPI(MPI_Gather(&mumps->nnz, 1, MPIU_INT64, mumps->recvcount, 1, MPIU_INT64, 0 /*master*/, mumps->omp_comm)); 18423ab56b82SJunchao Zhang 1843a6053eceSJunchao Zhang /* Then each computes number of send/recvs */ 18443ab56b82SJunchao Zhang if (mumps->is_omp_master) { 1845a6053eceSJunchao Zhang /* Start from 1 since self communication is not done in MPI */ 1846a6053eceSJunchao Zhang nreqs = 0; 1847a6053eceSJunchao Zhang for (i = 1; i < osize; i++) nreqs += (mumps->recvcount[i] + PETSC_MPI_INT_MAX - 1) / PETSC_MPI_INT_MAX; 1848a6053eceSJunchao Zhang } else { 1849a6053eceSJunchao Zhang nreqs = (mumps->nnz + PETSC_MPI_INT_MAX - 1) / PETSC_MPI_INT_MAX; 18503ab56b82SJunchao Zhang } 185135cb6cd3SPierre Jolivet PetscCall(PetscMalloc1(nreqs * 3, &mumps->reqs)); /* Triple the requests since we send irn, jcn and val separately */ 18523ab56b82SJunchao Zhang 1853a6053eceSJunchao Zhang /* The following code is doing a very simple thing: omp_master rank gathers irn/jcn/val from others. 1854a6053eceSJunchao Zhang MPI_Gatherv would be enough if it supports big counts > 2^31-1. Since it does not, and mumps->nnz 1855a6053eceSJunchao Zhang might be a prime number > 2^31-1, we have to slice the message. Note omp_comm_size 1856a6053eceSJunchao Zhang is very small, the current approach should have no extra overhead compared to MPI_Gatherv. 1857a6053eceSJunchao Zhang */ 1858a6053eceSJunchao Zhang nreqs = 0; /* counter for actual send/recvs */ 18593ab56b82SJunchao Zhang if (mumps->is_omp_master) { 1860a6053eceSJunchao Zhang for (i = 0, totnnz = 0; i < osize; i++) totnnz += mumps->recvcount[i]; /* totnnz = sum of nnz over omp_comm */ 18619566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(totnnz, &irn, totnnz, &jcn)); 18629566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totnnz, &val)); 1863a6053eceSJunchao Zhang 1864a6053eceSJunchao Zhang /* Self communication */ 18659566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(irn, mumps->irn, mumps->nnz)); 18669566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(jcn, mumps->jcn, mumps->nnz)); 18679566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(val, mumps->val, mumps->nnz)); 1868a6053eceSJunchao Zhang 1869a6053eceSJunchao Zhang /* Replace mumps->irn/jcn etc on master with the newly allocated bigger arrays */ 18709566063dSJacob Faibussowitsch PetscCall(PetscFree2(mumps->irn, mumps->jcn)); 18719566063dSJacob Faibussowitsch PetscCall(PetscFree(mumps->val_alloc)); 1872a6053eceSJunchao Zhang mumps->nnz = totnnz; 18733ab56b82SJunchao Zhang mumps->irn = irn; 18743ab56b82SJunchao Zhang mumps->jcn = jcn; 1875a6053eceSJunchao Zhang mumps->val = mumps->val_alloc = val; 1876a6053eceSJunchao Zhang 1877a6053eceSJunchao Zhang irn += mumps->recvcount[0]; /* recvcount[0] is old mumps->nnz on omp rank 0 */ 1878a6053eceSJunchao Zhang jcn += mumps->recvcount[0]; 1879a6053eceSJunchao Zhang val += mumps->recvcount[0]; 1880a6053eceSJunchao Zhang 1881a6053eceSJunchao Zhang /* Remote communication */ 1882a6053eceSJunchao Zhang for (i = 1; i < osize; i++) { 1883a6053eceSJunchao Zhang count = PetscMin(mumps->recvcount[i], PETSC_MPI_INT_MAX); 1884a6053eceSJunchao Zhang remain = mumps->recvcount[i] - count; 1885a6053eceSJunchao Zhang while (count > 0) { 18869566063dSJacob Faibussowitsch PetscCallMPI(MPI_Irecv(irn, count, MPIU_MUMPSINT, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++])); 18879566063dSJacob Faibussowitsch PetscCallMPI(MPI_Irecv(jcn, count, MPIU_MUMPSINT, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++])); 18889566063dSJacob Faibussowitsch PetscCallMPI(MPI_Irecv(val, count, MPIU_SCALAR, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++])); 1889a6053eceSJunchao Zhang irn += count; 1890a6053eceSJunchao Zhang jcn += count; 1891a6053eceSJunchao Zhang val += count; 1892a6053eceSJunchao Zhang count = PetscMin(remain, PETSC_MPI_INT_MAX); 1893a6053eceSJunchao Zhang remain -= count; 1894a6053eceSJunchao Zhang } 18953ab56b82SJunchao Zhang } 18963ab56b82SJunchao Zhang } else { 1897a6053eceSJunchao Zhang irn = mumps->irn; 1898a6053eceSJunchao Zhang jcn = mumps->jcn; 1899a6053eceSJunchao Zhang val = mumps->val; 1900a6053eceSJunchao Zhang count = PetscMin(mumps->nnz, PETSC_MPI_INT_MAX); 1901a6053eceSJunchao Zhang remain = mumps->nnz - count; 1902a6053eceSJunchao Zhang while (count > 0) { 19039566063dSJacob Faibussowitsch PetscCallMPI(MPI_Isend(irn, count, MPIU_MUMPSINT, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++])); 19049566063dSJacob Faibussowitsch PetscCallMPI(MPI_Isend(jcn, count, MPIU_MUMPSINT, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++])); 19059566063dSJacob Faibussowitsch PetscCallMPI(MPI_Isend(val, count, MPIU_SCALAR, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++])); 1906a6053eceSJunchao Zhang irn += count; 1907a6053eceSJunchao Zhang jcn += count; 1908a6053eceSJunchao Zhang val += count; 1909a6053eceSJunchao Zhang count = PetscMin(remain, PETSC_MPI_INT_MAX); 1910a6053eceSJunchao Zhang remain -= count; 19113ab56b82SJunchao Zhang } 19123ab56b82SJunchao Zhang } 1913a6053eceSJunchao Zhang } else { 1914a6053eceSJunchao Zhang nreqs = 0; 1915a6053eceSJunchao Zhang if (mumps->is_omp_master) { 1916a6053eceSJunchao Zhang val = mumps->val + mumps->recvcount[0]; 1917a6053eceSJunchao Zhang for (i = 1; i < osize; i++) { /* Remote communication only since self data is already in place */ 1918a6053eceSJunchao Zhang count = PetscMin(mumps->recvcount[i], PETSC_MPI_INT_MAX); 1919a6053eceSJunchao Zhang remain = mumps->recvcount[i] - count; 1920a6053eceSJunchao Zhang while (count > 0) { 19219566063dSJacob Faibussowitsch PetscCallMPI(MPI_Irecv(val, count, MPIU_SCALAR, i, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++])); 1922a6053eceSJunchao Zhang val += count; 1923a6053eceSJunchao Zhang count = PetscMin(remain, PETSC_MPI_INT_MAX); 1924a6053eceSJunchao Zhang remain -= count; 1925a6053eceSJunchao Zhang } 1926a6053eceSJunchao Zhang } 1927a6053eceSJunchao Zhang } else { 1928a6053eceSJunchao Zhang val = mumps->val; 1929a6053eceSJunchao Zhang count = PetscMin(mumps->nnz, PETSC_MPI_INT_MAX); 1930a6053eceSJunchao Zhang remain = mumps->nnz - count; 1931a6053eceSJunchao Zhang while (count > 0) { 19329566063dSJacob Faibussowitsch PetscCallMPI(MPI_Isend(val, count, MPIU_SCALAR, 0, mumps->tag, mumps->omp_comm, &mumps->reqs[nreqs++])); 1933a6053eceSJunchao Zhang val += count; 1934a6053eceSJunchao Zhang count = PetscMin(remain, PETSC_MPI_INT_MAX); 1935a6053eceSJunchao Zhang remain -= count; 1936a6053eceSJunchao Zhang } 1937a6053eceSJunchao Zhang } 1938a6053eceSJunchao Zhang } 19399566063dSJacob Faibussowitsch PetscCallMPI(MPI_Waitall(nreqs, mumps->reqs, MPI_STATUSES_IGNORE)); 1940a6053eceSJunchao Zhang mumps->tag++; /* It is totally fine for above send/recvs to share one mpi tag */ 1941a6053eceSJunchao Zhang } 19423ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 19433ab56b82SJunchao Zhang } 19443ab56b82SJunchao Zhang 194566976f2fSJacob Faibussowitsch static PetscErrorCode MatFactorNumeric_MUMPS(Mat F, Mat A, const MatFactorInfo *info) 1946d71ae5a4SJacob Faibussowitsch { 1947e69c285eSBarry Smith Mat_MUMPS *mumps = (Mat_MUMPS *)(F)->data; 1948ace3abfcSBarry Smith PetscBool isMPIAIJ; 1949397b6df1SKris Buschelman 1950397b6df1SKris Buschelman PetscFunctionBegin; 1951dbf6bb8dSprj- if (mumps->id.INFOG(1) < 0 && !(mumps->id.INFOG(1) == -16 && mumps->id.INFOG(1) == 0)) { 195248a46eb9SPierre 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))); 19539566063dSJacob 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))); 19543ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 19552aca8efcSHong Zhang } 19566baea169SHong Zhang 19579566063dSJacob Faibussowitsch PetscCall((*mumps->ConvertToTriples)(A, 1, MAT_REUSE_MATRIX, mumps)); 19589566063dSJacob Faibussowitsch PetscCall(MatMumpsGatherNonzerosOnMaster(MAT_REUSE_MATRIX, mumps)); 1959397b6df1SKris Buschelman 1960397b6df1SKris Buschelman /* numerical factorization phase */ 1961a5e57a09SHong Zhang mumps->id.job = JOB_FACTNUMERIC; 19624e34a73bSHong Zhang if (!mumps->id.ICNTL(18)) { /* A is centralized */ 1963ad540459SPierre Jolivet if (!mumps->myid) mumps->id.a = (MumpsScalar *)mumps->val; 1964397b6df1SKris Buschelman } else { 1965940cd9d6SSatish Balay mumps->id.a_loc = (MumpsScalar *)mumps->val; 1966397b6df1SKris Buschelman } 19673ab56b82SJunchao Zhang PetscMUMPS_c(mumps); 1968a5e57a09SHong Zhang if (mumps->id.INFOG(1) < 0) { 19699261f6e4SBarry 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)); 19709261f6e4SBarry Smith if (mumps->id.INFOG(1) == -10) { 19719261f6e4SBarry 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))); 1972603e8f96SBarry Smith F->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 1973c0d63f2fSHong Zhang } else if (mumps->id.INFOG(1) == -13) { 19749261f6e4SBarry 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))); 1975603e8f96SBarry Smith F->factorerrortype = MAT_FACTOR_OUTMEMORY; 1976c0d63f2fSHong Zhang } else if (mumps->id.INFOG(1) == -8 || mumps->id.INFOG(1) == -9 || (-16 < mumps->id.INFOG(1) && mumps->id.INFOG(1) < -10)) { 19779261f6e4SBarry 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))); 1978603e8f96SBarry Smith F->factorerrortype = MAT_FACTOR_OUTMEMORY; 19792aca8efcSHong Zhang } else { 19809261f6e4SBarry Smith PetscCall(PetscInfo(F, "MUMPS error in numerical factorization: INFOG(1)=%d, INFO(2)=%d\n", mumps->id.INFOG(1), mumps->id.INFO(2))); 1981603e8f96SBarry Smith F->factorerrortype = MAT_FACTOR_OTHER; 1982151787a6SHong Zhang } 19832aca8efcSHong Zhang } 19849261f6e4SBarry 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)); 1985397b6df1SKris Buschelman 1986b3cb21ddSStefano Zampini F->assembled = PETSC_TRUE; 1987d47f36abSHong Zhang 1988b3cb21ddSStefano Zampini if (F->schur) { /* reset Schur status to unfactored */ 19893cb7dd0eSStefano Zampini #if defined(PETSC_HAVE_CUDA) 1990c70f7ee4SJunchao Zhang F->schur->offloadmask = PETSC_OFFLOAD_CPU; 19913cb7dd0eSStefano Zampini #endif 1992b3cb21ddSStefano Zampini if (mumps->id.ICNTL(19) == 1) { /* stored by rows */ 1993b3cb21ddSStefano Zampini mumps->id.ICNTL(19) = 2; 19949566063dSJacob Faibussowitsch PetscCall(MatTranspose(F->schur, MAT_INPLACE_MATRIX, &F->schur)); 1995b3cb21ddSStefano Zampini } 19969566063dSJacob Faibussowitsch PetscCall(MatFactorRestoreSchurComplement(F, NULL, MAT_FACTOR_SCHUR_UNFACTORED)); 1997b3cb21ddSStefano Zampini } 199867877ebaSShri Abhyankar 1999066565c5SStefano Zampini /* just to be sure that ICNTL(19) value returned by a call from MatMumpsGetIcntl is always consistent */ 2000066565c5SStefano Zampini if (!mumps->sym && mumps->id.ICNTL(19) && mumps->id.ICNTL(19) != 1) mumps->id.ICNTL(19) = 3; 2001066565c5SStefano Zampini 20023ab56b82SJunchao Zhang if (!mumps->is_omp_master) mumps->id.INFO(23) = 0; 20032d4298aeSJunchao Zhang if (mumps->petsc_size > 1) { 200467877ebaSShri Abhyankar PetscInt lsol_loc; 200567877ebaSShri Abhyankar PetscScalar *sol_loc; 20062205254eSKarl Rupp 20079566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)A, MATMPIAIJ, &isMPIAIJ)); 2008c2093ab7SHong Zhang 2009c2093ab7SHong Zhang /* distributed solution; Create x_seq=sol_loc for repeated use */ 2010c2093ab7SHong Zhang if (mumps->x_seq) { 20119566063dSJacob Faibussowitsch PetscCall(VecScatterDestroy(&mumps->scat_sol)); 20129566063dSJacob Faibussowitsch PetscCall(PetscFree2(mumps->id.sol_loc, mumps->id.isol_loc)); 20139566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mumps->x_seq)); 2014c2093ab7SHong Zhang } 2015a5e57a09SHong Zhang lsol_loc = mumps->id.INFO(23); /* length of sol_loc */ 20169566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(lsol_loc, &sol_loc, lsol_loc, &mumps->id.isol_loc)); 2017a5e57a09SHong Zhang mumps->id.lsol_loc = lsol_loc; 2018940cd9d6SSatish Balay mumps->id.sol_loc = (MumpsScalar *)sol_loc; 20199566063dSJacob Faibussowitsch PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, lsol_loc, sol_loc, &mumps->x_seq)); 202067877ebaSShri Abhyankar } 20219566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(mumps->id.RINFO(2))); 20223ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2023397b6df1SKris Buschelman } 2024397b6df1SKris Buschelman 20259a2535b5SHong Zhang /* Sets MUMPS options from the options database */ 202666976f2fSJacob Faibussowitsch static PetscErrorCode MatSetFromOptions_MUMPS(Mat F, Mat A) 2027d71ae5a4SJacob Faibussowitsch { 2028e69c285eSBarry Smith Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 2029413bcc21SPierre Jolivet PetscMUMPSInt icntl = 0, size, *listvar_schur; 203045e3843bSPierre Jolivet PetscInt info[80], i, ninfo = 80, rbs, cbs; 2031413bcc21SPierre Jolivet PetscBool flg = PETSC_FALSE, schur = (PetscBool)(mumps->id.ICNTL(26) == -1); 2032413bcc21SPierre Jolivet MumpsScalar *arr; 2033dcd589f8SShri Abhyankar 2034dcd589f8SShri Abhyankar PetscFunctionBegin; 203526cc229bSBarry Smith PetscOptionsBegin(PetscObjectComm((PetscObject)F), ((PetscObject)F)->prefix, "MUMPS Options", "Mat"); 2036413bcc21SPierre Jolivet if (mumps->id.job == JOB_NULL) { /* MatSetFromOptions_MUMPS() has never been called before */ 2037413bcc21SPierre Jolivet PetscInt nthreads = 0; 2038413bcc21SPierre Jolivet PetscInt nCNTL_pre = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0; 2039413bcc21SPierre Jolivet PetscInt nICNTL_pre = mumps->ICNTL_pre ? mumps->ICNTL_pre[0] : 0; 2040413bcc21SPierre Jolivet 2041413bcc21SPierre Jolivet mumps->petsc_comm = PetscObjectComm((PetscObject)A); 2042413bcc21SPierre Jolivet PetscCallMPI(MPI_Comm_size(mumps->petsc_comm, &mumps->petsc_size)); 2043413bcc21SPierre Jolivet PetscCallMPI(MPI_Comm_rank(mumps->petsc_comm, &mumps->myid)); /* "if (!myid)" still works even if mumps_comm is different */ 2044413bcc21SPierre Jolivet 2045413bcc21SPierre Jolivet PetscCall(PetscOptionsName("-mat_mumps_use_omp_threads", "Convert MPI processes into OpenMP threads", "None", &mumps->use_petsc_omp_support)); 2046413bcc21SPierre Jolivet if (mumps->use_petsc_omp_support) nthreads = -1; /* -1 will let PetscOmpCtrlCreate() guess a proper value when user did not supply one */ 2047413bcc21SPierre Jolivet /* do not use PetscOptionsInt() so that the option -mat_mumps_use_omp_threads is not displayed twice in the help */ 2048413bcc21SPierre Jolivet PetscCall(PetscOptionsGetInt(NULL, ((PetscObject)F)->prefix, "-mat_mumps_use_omp_threads", &nthreads, NULL)); 2049413bcc21SPierre Jolivet if (mumps->use_petsc_omp_support) { 20509371c9d4SSatish 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", 20519371c9d4SSatish Balay ((PetscObject)F)->prefix ? ((PetscObject)F)->prefix : ""); 2052413bcc21SPierre 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 : ""); 2053413bcc21SPierre Jolivet #if defined(PETSC_HAVE_OPENMP_SUPPORT) 2054413bcc21SPierre Jolivet PetscCall(PetscOmpCtrlCreate(mumps->petsc_comm, nthreads, &mumps->omp_ctrl)); 2055413bcc21SPierre Jolivet PetscCall(PetscOmpCtrlGetOmpComms(mumps->omp_ctrl, &mumps->omp_comm, &mumps->mumps_comm, &mumps->is_omp_master)); 2056413bcc21SPierre Jolivet #endif 2057413bcc21SPierre Jolivet } else { 2058413bcc21SPierre Jolivet mumps->omp_comm = PETSC_COMM_SELF; 2059413bcc21SPierre Jolivet mumps->mumps_comm = mumps->petsc_comm; 2060413bcc21SPierre Jolivet mumps->is_omp_master = PETSC_TRUE; 2061413bcc21SPierre Jolivet } 2062413bcc21SPierre Jolivet PetscCallMPI(MPI_Comm_size(mumps->omp_comm, &mumps->omp_comm_size)); 2063413bcc21SPierre Jolivet mumps->reqs = NULL; 2064413bcc21SPierre Jolivet mumps->tag = 0; 2065413bcc21SPierre Jolivet 2066413bcc21SPierre Jolivet if (mumps->mumps_comm != MPI_COMM_NULL) { 2067413bcc21SPierre Jolivet if (PetscDefined(HAVE_OPENMP_SUPPORT) && mumps->use_petsc_omp_support) { 2068413bcc21SPierre Jolivet /* It looks like MUMPS does not dup the input comm. Dup a new comm for MUMPS to avoid any tag mismatches. */ 2069413bcc21SPierre Jolivet MPI_Comm comm; 2070413bcc21SPierre Jolivet PetscCallMPI(MPI_Comm_dup(mumps->mumps_comm, &comm)); 2071413bcc21SPierre Jolivet mumps->mumps_comm = comm; 2072413bcc21SPierre Jolivet } else PetscCall(PetscCommGetComm(mumps->petsc_comm, &mumps->mumps_comm)); 2073413bcc21SPierre Jolivet } 2074413bcc21SPierre Jolivet 2075413bcc21SPierre Jolivet mumps->id.comm_fortran = MPI_Comm_c2f(mumps->mumps_comm); 2076413bcc21SPierre Jolivet mumps->id.job = JOB_INIT; 2077413bcc21SPierre Jolivet mumps->id.par = 1; /* host participates factorizaton and solve */ 2078413bcc21SPierre Jolivet mumps->id.sym = mumps->sym; 2079413bcc21SPierre Jolivet 2080413bcc21SPierre Jolivet size = mumps->id.size_schur; 2081413bcc21SPierre Jolivet arr = mumps->id.schur; 2082413bcc21SPierre Jolivet listvar_schur = mumps->id.listvar_schur; 2083413bcc21SPierre Jolivet PetscMUMPS_c(mumps); 20849261f6e4SBarry Smith PetscCheck(mumps->id.INFOG(1) >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "MUMPS error: INFOG(1)=%d " MUMPS_MANUALS, mumps->id.INFOG(1)); 2085*51ad14ebSPierre Jolivet 2086*51ad14ebSPierre Jolivet /* set PETSc-MUMPS default options - override MUMPS default */ 2087*51ad14ebSPierre Jolivet mumps->id.ICNTL(3) = 0; 2088*51ad14ebSPierre Jolivet mumps->id.ICNTL(4) = 0; 2089*51ad14ebSPierre Jolivet if (mumps->petsc_size == 1) { 2090*51ad14ebSPierre Jolivet mumps->id.ICNTL(18) = 0; /* centralized assembled matrix input */ 2091*51ad14ebSPierre Jolivet mumps->id.ICNTL(7) = 7; /* automatic choice of ordering done by the package */ 2092*51ad14ebSPierre Jolivet } else { 2093*51ad14ebSPierre Jolivet mumps->id.ICNTL(18) = 3; /* distributed assembled matrix input */ 2094*51ad14ebSPierre Jolivet mumps->id.ICNTL(21) = 1; /* distributed solution */ 2095*51ad14ebSPierre Jolivet } 2096*51ad14ebSPierre Jolivet 2097413bcc21SPierre Jolivet /* restore cached ICNTL and CNTL values */ 2098413bcc21SPierre Jolivet for (icntl = 0; icntl < nICNTL_pre; ++icntl) mumps->id.ICNTL(mumps->ICNTL_pre[1 + 2 * icntl]) = mumps->ICNTL_pre[2 + 2 * icntl]; 2099413bcc21SPierre Jolivet for (icntl = 0; icntl < nCNTL_pre; ++icntl) mumps->id.CNTL((PetscInt)mumps->CNTL_pre[1 + 2 * icntl]) = mumps->CNTL_pre[2 + 2 * icntl]; 2100413bcc21SPierre Jolivet PetscCall(PetscFree(mumps->ICNTL_pre)); 2101413bcc21SPierre Jolivet PetscCall(PetscFree(mumps->CNTL_pre)); 2102413bcc21SPierre Jolivet 2103413bcc21SPierre Jolivet if (schur) { 2104413bcc21SPierre Jolivet mumps->id.size_schur = size; 2105413bcc21SPierre Jolivet mumps->id.schur_lld = size; 2106413bcc21SPierre Jolivet mumps->id.schur = arr; 2107413bcc21SPierre Jolivet mumps->id.listvar_schur = listvar_schur; 2108413bcc21SPierre Jolivet if (mumps->petsc_size > 1) { 2109413bcc21SPierre Jolivet PetscBool gs; /* gs is false if any rank other than root has non-empty IS */ 2110413bcc21SPierre Jolivet 2111413bcc21SPierre Jolivet mumps->id.ICNTL(19) = 1; /* MUMPS returns Schur centralized on the host */ 2112413bcc21SPierre 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 */ 2113712fec58SPierre Jolivet PetscCall(MPIU_Allreduce(MPI_IN_PLACE, &gs, 1, MPIU_BOOL, MPI_LAND, mumps->petsc_comm)); 2114413bcc21SPierre Jolivet PetscCheck(gs, PETSC_COMM_SELF, PETSC_ERR_SUP, "MUMPS distributed parallel Schur complements not yet supported from PETSc"); 2115413bcc21SPierre Jolivet } else { 2116413bcc21SPierre Jolivet if (F->factortype == MAT_FACTOR_LU) { 2117413bcc21SPierre Jolivet mumps->id.ICNTL(19) = 3; /* MUMPS returns full matrix */ 2118413bcc21SPierre Jolivet } else { 2119413bcc21SPierre Jolivet mumps->id.ICNTL(19) = 2; /* MUMPS returns lower triangular part */ 2120413bcc21SPierre Jolivet } 2121413bcc21SPierre Jolivet } 2122413bcc21SPierre Jolivet mumps->id.ICNTL(26) = -1; 2123413bcc21SPierre Jolivet } 2124413bcc21SPierre Jolivet 2125413bcc21SPierre Jolivet /* copy MUMPS default control values from master to slaves. Although slaves do not call MUMPS, they may access these values in code. 2126413bcc21SPierre Jolivet For example, ICNTL(9) is initialized to 1 by MUMPS and slaves check ICNTL(9) in MatSolve_MUMPS. 2127413bcc21SPierre Jolivet */ 2128413bcc21SPierre Jolivet PetscCallMPI(MPI_Bcast(mumps->id.icntl, 40, MPI_INT, 0, mumps->omp_comm)); 2129413bcc21SPierre Jolivet PetscCallMPI(MPI_Bcast(mumps->id.cntl, 15, MPIU_REAL, 0, mumps->omp_comm)); 2130413bcc21SPierre Jolivet 2131413bcc21SPierre Jolivet mumps->scat_rhs = NULL; 2132413bcc21SPierre Jolivet mumps->scat_sol = NULL; 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; 2284*51ad14ebSPierre Jolivet if (r && mumps->id.ICNTL(7) == 7) { 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 /*@ 27391d27aa22SBarry Smith MatMumpsSetIcntl - Set MUMPS parameter ICNTL() <https://mumps-solver.org/index.php?page=doc> 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 27531cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()` 27545ccb76cbSHong Zhang @*/ 2755d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsSetIcntl(Mat F, PetscInt icntl, PetscInt ival) 2756d71ae5a4SJacob Faibussowitsch { 27575ccb76cbSHong Zhang PetscFunctionBegin; 27582989dfd4SHong Zhang PetscValidType(F, 1); 275928b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 27605ccb76cbSHong Zhang PetscValidLogicalCollectiveInt(F, icntl, 2); 27615ccb76cbSHong Zhang PetscValidLogicalCollectiveInt(F, ival, 3); 2762146931dbSPierre Jolivet PetscCheck((icntl >= 1 && icntl <= 38) || icntl == 58, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported ICNTL value %" PetscInt_FMT, icntl); 2763cac4c232SBarry Smith PetscTryMethod(F, "MatMumpsSetIcntl_C", (Mat, PetscInt, PetscInt), (F, icntl, ival)); 27643ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 27655ccb76cbSHong Zhang } 27665ccb76cbSHong Zhang 2767a21f80fcSHong Zhang /*@ 27681d27aa22SBarry Smith MatMumpsGetIcntl - Get MUMPS parameter ICNTL() <https://mumps-solver.org/index.php?page=doc> 2769a21f80fcSHong Zhang 2770c3339decSBarry Smith Logically Collective 2771a21f80fcSHong Zhang 2772a21f80fcSHong Zhang Input Parameters: 277311a5261eSBarry Smith + F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 2774a21f80fcSHong Zhang - icntl - index of MUMPS parameter array ICNTL() 2775a21f80fcSHong Zhang 2776a21f80fcSHong Zhang Output Parameter: 2777a21f80fcSHong Zhang . ival - value of MUMPS ICNTL(icntl) 2778a21f80fcSHong Zhang 2779a21f80fcSHong Zhang Level: beginner 2780a21f80fcSHong Zhang 27811cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()` 2782a21f80fcSHong Zhang @*/ 2783d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetIcntl(Mat F, PetscInt icntl, PetscInt *ival) 2784d71ae5a4SJacob Faibussowitsch { 2785bc6112feSHong Zhang PetscFunctionBegin; 27862989dfd4SHong Zhang PetscValidType(F, 1); 278728b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 2788bc6112feSHong Zhang PetscValidLogicalCollectiveInt(F, icntl, 2); 27894f572ea9SToby Isaac PetscAssertPointer(ival, 3); 2790146931dbSPierre Jolivet PetscCheck((icntl >= 1 && icntl <= 38) || icntl == 58, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported ICNTL value %" PetscInt_FMT, icntl); 2791cac4c232SBarry Smith PetscUseMethod(F, "MatMumpsGetIcntl_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival)); 27923ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2793bc6112feSHong Zhang } 2794bc6112feSHong Zhang 279566976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsSetCntl_MUMPS(Mat F, PetscInt icntl, PetscReal val) 2796d71ae5a4SJacob Faibussowitsch { 2797e69c285eSBarry Smith Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 27988928b65cSHong Zhang 27998928b65cSHong Zhang PetscFunctionBegin; 2800413bcc21SPierre Jolivet if (mumps->id.job == JOB_NULL) { 2801413bcc21SPierre Jolivet PetscInt i, nCNTL_pre = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0; 28029371c9d4SSatish Balay for (i = 0; i < nCNTL_pre; ++i) 28039371c9d4SSatish Balay if (mumps->CNTL_pre[1 + 2 * i] == icntl) break; 2804413bcc21SPierre Jolivet if (i == nCNTL_pre) { 2805413bcc21SPierre Jolivet if (i > 0) PetscCall(PetscRealloc(sizeof(PetscReal) * (2 * nCNTL_pre + 3), &mumps->CNTL_pre)); 2806413bcc21SPierre Jolivet else PetscCall(PetscCalloc(sizeof(PetscReal) * 3, &mumps->CNTL_pre)); 2807413bcc21SPierre Jolivet mumps->CNTL_pre[0]++; 2808413bcc21SPierre Jolivet } 2809413bcc21SPierre Jolivet mumps->CNTL_pre[1 + 2 * i] = icntl; 2810413bcc21SPierre Jolivet mumps->CNTL_pre[2 + 2 * i] = val; 2811413bcc21SPierre Jolivet } else mumps->id.CNTL(icntl) = val; 28123ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 28138928b65cSHong Zhang } 28148928b65cSHong Zhang 281566976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetCntl_MUMPS(Mat F, PetscInt icntl, PetscReal *val) 2816d71ae5a4SJacob Faibussowitsch { 2817e69c285eSBarry Smith Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 2818bc6112feSHong Zhang 2819bc6112feSHong Zhang PetscFunctionBegin; 282036df9881Sjeremy theler if (mumps->id.job == JOB_NULL) { 282136df9881Sjeremy theler PetscInt i, nCNTL_pre = mumps->CNTL_pre ? mumps->CNTL_pre[0] : 0; 282236df9881Sjeremy theler *val = 0.0; 282336df9881Sjeremy theler for (i = 0; i < nCNTL_pre; ++i) { 282436df9881Sjeremy theler if (mumps->CNTL_pre[1 + 2 * i] == icntl) *val = mumps->CNTL_pre[2 + 2 * i]; 282536df9881Sjeremy theler } 282636df9881Sjeremy theler } else *val = mumps->id.CNTL(icntl); 28273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2828bc6112feSHong Zhang } 2829bc6112feSHong Zhang 28308928b65cSHong Zhang /*@ 28311d27aa22SBarry Smith MatMumpsSetCntl - Set MUMPS parameter CNTL() <https://mumps-solver.org/index.php?page=doc> 28328928b65cSHong Zhang 2833c3339decSBarry Smith Logically Collective 28348928b65cSHong Zhang 28358928b65cSHong Zhang Input Parameters: 283611a5261eSBarry Smith + F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 28378928b65cSHong Zhang . icntl - index of MUMPS parameter array CNTL() 28388928b65cSHong Zhang - val - value of MUMPS CNTL(icntl) 28398928b65cSHong Zhang 28403c7db156SBarry Smith Options Database Key: 2841147403d9SBarry Smith . -mat_mumps_cntl_<icntl> <val> - change the option numbered icntl to ival 28428928b65cSHong Zhang 28438928b65cSHong Zhang Level: beginner 28448928b65cSHong Zhang 28451cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()` 28468928b65cSHong Zhang @*/ 2847d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsSetCntl(Mat F, PetscInt icntl, PetscReal val) 2848d71ae5a4SJacob Faibussowitsch { 28498928b65cSHong Zhang PetscFunctionBegin; 28502989dfd4SHong Zhang PetscValidType(F, 1); 285128b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 28528928b65cSHong Zhang PetscValidLogicalCollectiveInt(F, icntl, 2); 2853bc6112feSHong Zhang PetscValidLogicalCollectiveReal(F, val, 3); 2854413bcc21SPierre Jolivet PetscCheck(icntl >= 1 && icntl <= 7, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported CNTL value %" PetscInt_FMT, icntl); 2855cac4c232SBarry Smith PetscTryMethod(F, "MatMumpsSetCntl_C", (Mat, PetscInt, PetscReal), (F, icntl, val)); 28563ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 28578928b65cSHong Zhang } 28588928b65cSHong Zhang 2859a21f80fcSHong Zhang /*@ 28601d27aa22SBarry Smith MatMumpsGetCntl - Get MUMPS parameter CNTL() <https://mumps-solver.org/index.php?page=doc> 2861a21f80fcSHong Zhang 2862c3339decSBarry Smith Logically Collective 2863a21f80fcSHong Zhang 2864a21f80fcSHong Zhang Input Parameters: 286511a5261eSBarry Smith + F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 2866a21f80fcSHong Zhang - icntl - index of MUMPS parameter array CNTL() 2867a21f80fcSHong Zhang 2868a21f80fcSHong Zhang Output Parameter: 2869a21f80fcSHong Zhang . val - value of MUMPS CNTL(icntl) 2870a21f80fcSHong Zhang 2871a21f80fcSHong Zhang Level: beginner 2872a21f80fcSHong Zhang 28731cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()` 2874a21f80fcSHong Zhang @*/ 2875d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetCntl(Mat F, PetscInt icntl, PetscReal *val) 2876d71ae5a4SJacob Faibussowitsch { 2877bc6112feSHong Zhang PetscFunctionBegin; 28782989dfd4SHong Zhang PetscValidType(F, 1); 287928b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 2880bc6112feSHong Zhang PetscValidLogicalCollectiveInt(F, icntl, 2); 28814f572ea9SToby Isaac PetscAssertPointer(val, 3); 2882413bcc21SPierre Jolivet PetscCheck(icntl >= 1 && icntl <= 7, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONG, "Unsupported CNTL value %" PetscInt_FMT, icntl); 2883cac4c232SBarry Smith PetscUseMethod(F, "MatMumpsGetCntl_C", (Mat, PetscInt, PetscReal *), (F, icntl, val)); 28843ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2885bc6112feSHong Zhang } 2886bc6112feSHong Zhang 288766976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInfo_MUMPS(Mat F, PetscInt icntl, PetscInt *info) 2888d71ae5a4SJacob Faibussowitsch { 2889e69c285eSBarry Smith Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 2890bc6112feSHong Zhang 2891bc6112feSHong Zhang PetscFunctionBegin; 2892bc6112feSHong Zhang *info = mumps->id.INFO(icntl); 28933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2894bc6112feSHong Zhang } 2895bc6112feSHong Zhang 289666976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInfog_MUMPS(Mat F, PetscInt icntl, PetscInt *infog) 2897d71ae5a4SJacob Faibussowitsch { 2898e69c285eSBarry Smith Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 2899bc6112feSHong Zhang 2900bc6112feSHong Zhang PetscFunctionBegin; 2901bc6112feSHong Zhang *infog = mumps->id.INFOG(icntl); 29023ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2903bc6112feSHong Zhang } 2904bc6112feSHong Zhang 290566976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetRinfo_MUMPS(Mat F, PetscInt icntl, PetscReal *rinfo) 2906d71ae5a4SJacob Faibussowitsch { 2907e69c285eSBarry Smith Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 2908bc6112feSHong Zhang 2909bc6112feSHong Zhang PetscFunctionBegin; 2910bc6112feSHong Zhang *rinfo = mumps->id.RINFO(icntl); 29113ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2912bc6112feSHong Zhang } 2913bc6112feSHong Zhang 291466976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetRinfog_MUMPS(Mat F, PetscInt icntl, PetscReal *rinfog) 2915d71ae5a4SJacob Faibussowitsch { 2916e69c285eSBarry Smith Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 2917bc6112feSHong Zhang 2918bc6112feSHong Zhang PetscFunctionBegin; 2919bc6112feSHong Zhang *rinfog = mumps->id.RINFOG(icntl); 29203ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2921bc6112feSHong Zhang } 2922bc6112feSHong Zhang 292366976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetNullPivots_MUMPS(Mat F, PetscInt *size, PetscInt **array) 29245c0bae8cSAshish Patel { 29255c0bae8cSAshish Patel Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 29265c0bae8cSAshish Patel 29275c0bae8cSAshish Patel PetscFunctionBegin; 29285c0bae8cSAshish 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"); 29295c0bae8cSAshish Patel *size = 0; 29305c0bae8cSAshish Patel *array = NULL; 29315c0bae8cSAshish Patel if (!mumps->myid) { 29325c0bae8cSAshish Patel *size = mumps->id.INFOG(28); 29335c0bae8cSAshish Patel PetscCall(PetscMalloc1(*size, array)); 29345c0bae8cSAshish Patel for (int i = 0; i < *size; i++) (*array)[i] = mumps->id.pivnul_list[i] - 1; 29355c0bae8cSAshish Patel } 29365c0bae8cSAshish Patel PetscFunctionReturn(PETSC_SUCCESS); 29375c0bae8cSAshish Patel } 29385c0bae8cSAshish Patel 293966976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInverse_MUMPS(Mat F, Mat spRHS) 2940d71ae5a4SJacob Faibussowitsch { 29410e6b8875SHong Zhang Mat Bt = NULL, Btseq = NULL; 29420e6b8875SHong Zhang PetscBool flg; 2943bb599dfdSHong Zhang Mat_MUMPS *mumps = (Mat_MUMPS *)F->data; 2944bb599dfdSHong Zhang PetscScalar *aa; 2945f410b75aSHong Zhang PetscInt spnr, *ia, *ja, M, nrhs; 2946bb599dfdSHong Zhang 2947bb599dfdSHong Zhang PetscFunctionBegin; 29484f572ea9SToby Isaac PetscAssertPointer(spRHS, 2); 2949013e2dc7SBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)spRHS, MATTRANSPOSEVIRTUAL, &flg)); 29500e6b8875SHong Zhang if (flg) { 29519566063dSJacob Faibussowitsch PetscCall(MatTransposeGetMat(spRHS, &Bt)); 2952013e2dc7SBarry Smith } else SETERRQ(PetscObjectComm((PetscObject)spRHS), PETSC_ERR_ARG_WRONG, "Matrix spRHS must be type MATTRANSPOSEVIRTUAL matrix"); 2953bb599dfdSHong Zhang 29549566063dSJacob Faibussowitsch PetscCall(MatMumpsSetIcntl(F, 30, 1)); 2955bb599dfdSHong Zhang 29562d4298aeSJunchao Zhang if (mumps->petsc_size > 1) { 29570e6b8875SHong Zhang Mat_MPIAIJ *b = (Mat_MPIAIJ *)Bt->data; 29580e6b8875SHong Zhang Btseq = b->A; 29590e6b8875SHong Zhang } else { 29600e6b8875SHong Zhang Btseq = Bt; 29610e6b8875SHong Zhang } 29620e6b8875SHong Zhang 29639566063dSJacob Faibussowitsch PetscCall(MatGetSize(spRHS, &M, &nrhs)); 2964f410b75aSHong Zhang mumps->id.nrhs = nrhs; 2965f410b75aSHong Zhang mumps->id.lrhs = M; 2966f410b75aSHong Zhang mumps->id.rhs = NULL; 2967f410b75aSHong Zhang 2968e3f2db6aSHong Zhang if (!mumps->myid) { 29699566063dSJacob Faibussowitsch PetscCall(MatSeqAIJGetArray(Btseq, &aa)); 29709566063dSJacob Faibussowitsch PetscCall(MatGetRowIJ(Btseq, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg)); 297128b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure"); 29729566063dSJacob Faibussowitsch PetscCall(PetscMUMPSIntCSRCast(mumps, spnr, ia, ja, &mumps->id.irhs_ptr, &mumps->id.irhs_sparse, &mumps->id.nz_rhs)); 2973bb599dfdSHong Zhang mumps->id.rhs_sparse = (MumpsScalar *)aa; 2974e3f2db6aSHong Zhang } else { 2975e3f2db6aSHong Zhang mumps->id.irhs_ptr = NULL; 2976e3f2db6aSHong Zhang mumps->id.irhs_sparse = NULL; 2977e3f2db6aSHong Zhang mumps->id.nz_rhs = 0; 2978e3f2db6aSHong Zhang mumps->id.rhs_sparse = NULL; 2979e3f2db6aSHong Zhang } 2980bb599dfdSHong Zhang mumps->id.ICNTL(20) = 1; /* rhs is sparse */ 2981e3f2db6aSHong Zhang mumps->id.ICNTL(21) = 0; /* solution is in assembled centralized format */ 2982bb599dfdSHong Zhang 2983bb599dfdSHong Zhang /* solve phase */ 2984bb599dfdSHong Zhang mumps->id.job = JOB_SOLVE; 29853ab56b82SJunchao Zhang PetscMUMPS_c(mumps); 29869261f6e4SBarry 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)); 298714267174SHong Zhang 2988e3f2db6aSHong Zhang if (!mumps->myid) { 29899566063dSJacob Faibussowitsch PetscCall(MatSeqAIJRestoreArray(Btseq, &aa)); 29909566063dSJacob Faibussowitsch PetscCall(MatRestoreRowIJ(Btseq, 1, PETSC_FALSE, PETSC_FALSE, &spnr, (const PetscInt **)&ia, (const PetscInt **)&ja, &flg)); 299128b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot get IJ structure"); 2992e3f2db6aSHong Zhang } 29933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2994bb599dfdSHong Zhang } 2995bb599dfdSHong Zhang 2996bb599dfdSHong Zhang /*@ 29971d27aa22SBarry Smith MatMumpsGetInverse - Get user-specified set of entries in inverse of `A` <https://mumps-solver.org/index.php?page=doc> 2998bb599dfdSHong Zhang 2999c3339decSBarry Smith Logically Collective 3000bb599dfdSHong Zhang 300120f4b53cSBarry Smith Input Parameter: 300220f4b53cSBarry Smith . F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 3003bb599dfdSHong Zhang 3004bb599dfdSHong Zhang Output Parameter: 300520f4b53cSBarry Smith . spRHS - sequential sparse matrix in `MATTRANSPOSEVIRTUAL` format with requested entries of inverse of `A` 3006bb599dfdSHong Zhang 3007bb599dfdSHong Zhang Level: beginner 3008bb599dfdSHong Zhang 30091cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatCreateTranspose()` 3010bb599dfdSHong Zhang @*/ 3011d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInverse(Mat F, Mat spRHS) 3012d71ae5a4SJacob Faibussowitsch { 3013bb599dfdSHong Zhang PetscFunctionBegin; 3014bb599dfdSHong Zhang PetscValidType(F, 1); 301528b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 3016cac4c232SBarry Smith PetscUseMethod(F, "MatMumpsGetInverse_C", (Mat, Mat), (F, spRHS)); 30173ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3018bb599dfdSHong Zhang } 3019bb599dfdSHong Zhang 302066976f2fSJacob Faibussowitsch static PetscErrorCode MatMumpsGetInverseTranspose_MUMPS(Mat F, Mat spRHST) 3021d71ae5a4SJacob Faibussowitsch { 30220e6b8875SHong Zhang Mat spRHS; 30230e6b8875SHong Zhang 30240e6b8875SHong Zhang PetscFunctionBegin; 30259566063dSJacob Faibussowitsch PetscCall(MatCreateTranspose(spRHST, &spRHS)); 30269566063dSJacob Faibussowitsch PetscCall(MatMumpsGetInverse_MUMPS(F, spRHS)); 30279566063dSJacob Faibussowitsch PetscCall(MatDestroy(&spRHS)); 30283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 30290e6b8875SHong Zhang } 30300e6b8875SHong Zhang 30310e6b8875SHong Zhang /*@ 30321d27aa22SBarry Smith MatMumpsGetInverseTranspose - Get user-specified set of entries in inverse of matrix $A^T $ <https://mumps-solver.org/index.php?page=doc> 30330e6b8875SHong Zhang 3034c3339decSBarry Smith Logically Collective 30350e6b8875SHong Zhang 303620f4b53cSBarry Smith Input Parameter: 303720f4b53cSBarry Smith . F - the factored matrix of A obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 30380e6b8875SHong Zhang 30390e6b8875SHong Zhang Output Parameter: 304020f4b53cSBarry Smith . spRHST - sequential sparse matrix in `MATAIJ` format containing the requested entries of inverse of `A`^T 30410e6b8875SHong Zhang 30420e6b8875SHong Zhang Level: beginner 30430e6b8875SHong Zhang 30441cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatCreateTranspose()`, `MatMumpsGetInverse()` 30450e6b8875SHong Zhang @*/ 3046d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInverseTranspose(Mat F, Mat spRHST) 3047d71ae5a4SJacob Faibussowitsch { 30480e6b8875SHong Zhang PetscBool flg; 30490e6b8875SHong Zhang 30500e6b8875SHong Zhang PetscFunctionBegin; 30510e6b8875SHong Zhang PetscValidType(F, 1); 305228b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 30539566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompareAny((PetscObject)spRHST, &flg, MATSEQAIJ, MATMPIAIJ, NULL)); 305428b400f6SJacob Faibussowitsch PetscCheck(flg, PetscObjectComm((PetscObject)spRHST), PETSC_ERR_ARG_WRONG, "Matrix spRHST must be MATAIJ matrix"); 30550e6b8875SHong Zhang 3056cac4c232SBarry Smith PetscUseMethod(F, "MatMumpsGetInverseTranspose_C", (Mat, Mat), (F, spRHST)); 30573ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 30580e6b8875SHong Zhang } 30590e6b8875SHong Zhang 3060a21f80fcSHong Zhang /*@ 30611d27aa22SBarry Smith MatMumpsGetInfo - Get MUMPS parameter INFO() <https://mumps-solver.org/index.php?page=doc> 3062a21f80fcSHong Zhang 3063c3339decSBarry Smith Logically Collective 3064a21f80fcSHong Zhang 3065a21f80fcSHong Zhang Input Parameters: 306611a5261eSBarry Smith + F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 3067a21f80fcSHong Zhang - icntl - index of MUMPS parameter array INFO() 3068a21f80fcSHong Zhang 3069a21f80fcSHong Zhang Output Parameter: 3070a21f80fcSHong Zhang . ival - value of MUMPS INFO(icntl) 3071a21f80fcSHong Zhang 3072a21f80fcSHong Zhang Level: beginner 3073a21f80fcSHong Zhang 30741cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()` 3075a21f80fcSHong Zhang @*/ 3076d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInfo(Mat F, PetscInt icntl, PetscInt *ival) 3077d71ae5a4SJacob Faibussowitsch { 3078bc6112feSHong Zhang PetscFunctionBegin; 30792989dfd4SHong Zhang PetscValidType(F, 1); 308028b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 30814f572ea9SToby Isaac PetscAssertPointer(ival, 3); 3082cac4c232SBarry Smith PetscUseMethod(F, "MatMumpsGetInfo_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival)); 30833ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3084bc6112feSHong Zhang } 3085bc6112feSHong Zhang 3086a21f80fcSHong Zhang /*@ 30871d27aa22SBarry Smith MatMumpsGetInfog - Get MUMPS parameter INFOG() <https://mumps-solver.org/index.php?page=doc> 3088a21f80fcSHong Zhang 3089c3339decSBarry Smith Logically Collective 3090a21f80fcSHong Zhang 3091a21f80fcSHong Zhang Input Parameters: 309211a5261eSBarry Smith + F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 3093a21f80fcSHong Zhang - icntl - index of MUMPS parameter array INFOG() 3094a21f80fcSHong Zhang 3095a21f80fcSHong Zhang Output Parameter: 3096a21f80fcSHong Zhang . ival - value of MUMPS INFOG(icntl) 3097a21f80fcSHong Zhang 3098a21f80fcSHong Zhang Level: beginner 3099a21f80fcSHong Zhang 31001cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()` 3101a21f80fcSHong Zhang @*/ 3102d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetInfog(Mat F, PetscInt icntl, PetscInt *ival) 3103d71ae5a4SJacob Faibussowitsch { 3104bc6112feSHong Zhang PetscFunctionBegin; 31052989dfd4SHong Zhang PetscValidType(F, 1); 310628b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 31074f572ea9SToby Isaac PetscAssertPointer(ival, 3); 3108cac4c232SBarry Smith PetscUseMethod(F, "MatMumpsGetInfog_C", (Mat, PetscInt, PetscInt *), (F, icntl, ival)); 31093ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3110bc6112feSHong Zhang } 3111bc6112feSHong Zhang 3112a21f80fcSHong Zhang /*@ 31131d27aa22SBarry Smith MatMumpsGetRinfo - Get MUMPS parameter RINFO() <https://mumps-solver.org/index.php?page=doc> 3114a21f80fcSHong Zhang 3115c3339decSBarry Smith Logically Collective 3116a21f80fcSHong Zhang 3117a21f80fcSHong Zhang Input Parameters: 311811a5261eSBarry Smith + F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 3119a21f80fcSHong Zhang - icntl - index of MUMPS parameter array RINFO() 3120a21f80fcSHong Zhang 3121a21f80fcSHong Zhang Output Parameter: 3122a21f80fcSHong Zhang . val - value of MUMPS RINFO(icntl) 3123a21f80fcSHong Zhang 3124a21f80fcSHong Zhang Level: beginner 3125a21f80fcSHong Zhang 31261cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfog()` 3127a21f80fcSHong Zhang @*/ 3128d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetRinfo(Mat F, PetscInt icntl, PetscReal *val) 3129d71ae5a4SJacob Faibussowitsch { 3130bc6112feSHong Zhang PetscFunctionBegin; 31312989dfd4SHong Zhang PetscValidType(F, 1); 313228b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 31334f572ea9SToby Isaac PetscAssertPointer(val, 3); 3134cac4c232SBarry Smith PetscUseMethod(F, "MatMumpsGetRinfo_C", (Mat, PetscInt, PetscReal *), (F, icntl, val)); 31353ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3136bc6112feSHong Zhang } 3137bc6112feSHong Zhang 3138a21f80fcSHong Zhang /*@ 31391d27aa22SBarry Smith MatMumpsGetRinfog - Get MUMPS parameter RINFOG() <https://mumps-solver.org/index.php?page=doc> 3140a21f80fcSHong Zhang 3141c3339decSBarry Smith Logically Collective 3142a21f80fcSHong Zhang 3143a21f80fcSHong Zhang Input Parameters: 314411a5261eSBarry Smith + F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 3145a21f80fcSHong Zhang - icntl - index of MUMPS parameter array RINFOG() 3146a21f80fcSHong Zhang 3147a21f80fcSHong Zhang Output Parameter: 3148a21f80fcSHong Zhang . val - value of MUMPS RINFOG(icntl) 3149a21f80fcSHong Zhang 3150a21f80fcSHong Zhang Level: beginner 3151a21f80fcSHong Zhang 31521cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()` 3153a21f80fcSHong Zhang @*/ 3154d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMumpsGetRinfog(Mat F, PetscInt icntl, PetscReal *val) 3155d71ae5a4SJacob Faibussowitsch { 3156bc6112feSHong Zhang PetscFunctionBegin; 31572989dfd4SHong Zhang PetscValidType(F, 1); 315828b400f6SJacob Faibussowitsch PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 31594f572ea9SToby Isaac PetscAssertPointer(val, 3); 3160cac4c232SBarry Smith PetscUseMethod(F, "MatMumpsGetRinfog_C", (Mat, PetscInt, PetscReal *), (F, icntl, val)); 31613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3162bc6112feSHong Zhang } 3163bc6112feSHong Zhang 31645c0bae8cSAshish Patel /*@ 31651d27aa22SBarry Smith MatMumpsGetNullPivots - Get MUMPS parameter PIVNUL_LIST() <https://mumps-solver.org/index.php?page=doc> 31665c0bae8cSAshish Patel 31675c0bae8cSAshish Patel Logically Collective 31685c0bae8cSAshish Patel 31695c0bae8cSAshish Patel Input Parameter: 31705c0bae8cSAshish Patel . F - the factored matrix obtained by calling `MatGetFactor()` from PETSc-MUMPS interface 31715c0bae8cSAshish Patel 31725c0bae8cSAshish Patel Output Parameters: 31735c0bae8cSAshish Patel + size - local size of the array. The size of the array is non-zero only on the host. 31745c0bae8cSAshish 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 31755c0bae8cSAshish Patel for freeing this array. 31765c0bae8cSAshish Patel 31775c0bae8cSAshish Patel Level: beginner 31785c0bae8cSAshish Patel 31791cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MatGetFactor()`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()` 31805c0bae8cSAshish Patel @*/ 31815c0bae8cSAshish Patel PetscErrorCode MatMumpsGetNullPivots(Mat F, PetscInt *size, PetscInt **array) 31825c0bae8cSAshish Patel { 31835c0bae8cSAshish Patel PetscFunctionBegin; 31845c0bae8cSAshish Patel PetscValidType(F, 1); 31855c0bae8cSAshish Patel PetscCheck(F->factortype, PetscObjectComm((PetscObject)F), PETSC_ERR_ARG_WRONGSTATE, "Only for factored matrix"); 31864f572ea9SToby Isaac PetscAssertPointer(size, 2); 31874f572ea9SToby Isaac PetscAssertPointer(array, 3); 31885c0bae8cSAshish Patel PetscUseMethod(F, "MatMumpsGetNullPivots_C", (Mat, PetscInt *, PetscInt **), (F, size, array)); 31895c0bae8cSAshish Patel PetscFunctionReturn(PETSC_SUCCESS); 31905c0bae8cSAshish Patel } 31915c0bae8cSAshish Patel 319224b6179bSKris Buschelman /*MC 31932692d6eeSBarry Smith MATSOLVERMUMPS - A matrix type providing direct solvers (LU and Cholesky) for 31941d27aa22SBarry Smith distributed and sequential matrices via the external package MUMPS <https://mumps-solver.org/index.php?page=doc> 319524b6179bSKris Buschelman 319611a5261eSBarry Smith Works with `MATAIJ` and `MATSBAIJ` matrices 319724b6179bSKris Buschelman 3198c2b89b5dSBarry Smith Use ./configure --download-mumps --download-scalapack --download-parmetis --download-metis --download-ptscotch to have PETSc installed with MUMPS 3199c2b89b5dSBarry Smith 32002ef1f0ffSBarry 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. 32012ef1f0ffSBarry Smith See details below. 3202217d3b1eSJunchao Zhang 32032ef1f0ffSBarry Smith Use `-pc_type cholesky` or `lu` `-pc_factor_mat_solver_type mumps` to use this direct solver 3204c2b89b5dSBarry Smith 320524b6179bSKris Buschelman Options Database Keys: 32064422a9fcSPatrick Sanan + -mat_mumps_icntl_1 - ICNTL(1): output stream for error messages 32074422a9fcSPatrick Sanan . -mat_mumps_icntl_2 - ICNTL(2): output stream for diagnostic printing, statistics, and warning 32084422a9fcSPatrick Sanan . -mat_mumps_icntl_3 - ICNTL(3): output stream for global information, collected on the host 32094422a9fcSPatrick Sanan . -mat_mumps_icntl_4 - ICNTL(4): level of printing (0 to 4) 32104422a9fcSPatrick Sanan . -mat_mumps_icntl_6 - ICNTL(6): permutes to a zero-free diagonal and/or scale the matrix (0 to 7) 3211b53c1a7fSBarry 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 3212b53c1a7fSBarry Smith Use -pc_factor_mat_ordering_type <type> to have PETSc perform the ordering (sequential only) 32134422a9fcSPatrick Sanan . -mat_mumps_icntl_8 - ICNTL(8): scaling strategy (-2 to 8 or 77) 32144422a9fcSPatrick Sanan . -mat_mumps_icntl_10 - ICNTL(10): max num of refinements 32154422a9fcSPatrick Sanan . -mat_mumps_icntl_11 - ICNTL(11): statistics related to an error analysis (via -ksp_view) 32164422a9fcSPatrick Sanan . -mat_mumps_icntl_12 - ICNTL(12): an ordering strategy for symmetric matrices (0 to 3) 32174422a9fcSPatrick Sanan . -mat_mumps_icntl_13 - ICNTL(13): parallelism of the root node (enable ScaLAPACK) and its splitting 32184422a9fcSPatrick Sanan . -mat_mumps_icntl_14 - ICNTL(14): percentage increase in the estimated working space 321945e3843bSPierre Jolivet . -mat_mumps_icntl_15 - ICNTL(15): compression of the input matrix resulting from a block format 32204422a9fcSPatrick Sanan . -mat_mumps_icntl_19 - ICNTL(19): computes the Schur complement 322125aac85cSJunchao Zhang . -mat_mumps_icntl_20 - ICNTL(20): give MUMPS centralized (0) or distributed (10) dense RHS 32224422a9fcSPatrick Sanan . -mat_mumps_icntl_22 - ICNTL(22): in-core/out-of-core factorization and solve (0 or 1) 32234422a9fcSPatrick Sanan . -mat_mumps_icntl_23 - ICNTL(23): max size of the working memory (MB) that can allocate per processor 32244422a9fcSPatrick Sanan . -mat_mumps_icntl_24 - ICNTL(24): detection of null pivot rows (0 or 1) 32254422a9fcSPatrick Sanan . -mat_mumps_icntl_25 - ICNTL(25): compute a solution of a deficient matrix and a null space basis 32264422a9fcSPatrick Sanan . -mat_mumps_icntl_26 - ICNTL(26): drives the solution phase if a Schur complement matrix 32274422a9fcSPatrick 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 32284422a9fcSPatrick Sanan . -mat_mumps_icntl_29 - ICNTL(29): parallel ordering 1 = ptscotch, 2 = parmetis 32294422a9fcSPatrick Sanan . -mat_mumps_icntl_30 - ICNTL(30): compute user-specified set of entries in inv(A) 32304422a9fcSPatrick Sanan . -mat_mumps_icntl_31 - ICNTL(31): indicates which factors may be discarded during factorization 32314422a9fcSPatrick Sanan . -mat_mumps_icntl_33 - ICNTL(33): compute determinant 3232a0e18203SThibaut Appel . -mat_mumps_icntl_35 - ICNTL(35): level of activation of BLR (Block Low-Rank) feature 3233a0e18203SThibaut Appel . -mat_mumps_icntl_36 - ICNTL(36): controls the choice of BLR factorization variant 3234a0e18203SThibaut Appel . -mat_mumps_icntl_38 - ICNTL(38): sets the estimated compression rate of LU factors with BLR 3235146931dbSPierre Jolivet . -mat_mumps_icntl_58 - ICNTL(58): options for symbolic factorization 32364422a9fcSPatrick Sanan . -mat_mumps_cntl_1 - CNTL(1): relative pivoting threshold 32374422a9fcSPatrick Sanan . -mat_mumps_cntl_2 - CNTL(2): stopping criterion of refinement 32384422a9fcSPatrick Sanan . -mat_mumps_cntl_3 - CNTL(3): absolute pivoting threshold 32394422a9fcSPatrick Sanan . -mat_mumps_cntl_4 - CNTL(4): value for static pivoting 3240217d3b1eSJunchao Zhang . -mat_mumps_cntl_5 - CNTL(5): fixation for null pivots 3241a0e18203SThibaut Appel . -mat_mumps_cntl_7 - CNTL(7): precision of the dropping parameter used during BLR factorization 3242217d3b1eSJunchao 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. 3243217d3b1eSJunchao Zhang Default might be the number of cores per CPU package (socket) as reported by hwloc and suggested by the MUMPS manual. 324424b6179bSKris Buschelman 324524b6179bSKris Buschelman Level: beginner 324624b6179bSKris Buschelman 324795452b02SPatrick Sanan Notes: 32481d27aa22SBarry 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 32492ef1f0ffSBarry Smith error if the matrix is Hermitian. 325038548759SBarry Smith 325126cc229bSBarry 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 325226cc229bSBarry Smith `MatSetOptionsPrefixFactor()` on the matrix from which the factor was obtained or `MatSetOptionsPrefix()` on the factor matrix. 325326cc229bSBarry Smith 32542ef1f0ffSBarry 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 32552ef1f0ffSBarry Smith the failure with 32562ef1f0ffSBarry Smith .vb 32572ef1f0ffSBarry Smith KSPGetPC(ksp,&pc); 32582ef1f0ffSBarry Smith PCFactorGetMatrix(pc,&mat); 32592ef1f0ffSBarry Smith MatMumpsGetInfo(mat,....); 32602ef1f0ffSBarry Smith MatMumpsGetInfog(mat,....); etc. 32612ef1f0ffSBarry Smith .ve 32622ef1f0ffSBarry Smith Or run with `-ksp_error_if_not_converged` and the program will be stopped and the information printed in the error message. 32639fc87aa7SBarry Smith 3264a5399872SJunchao Zhang MUMPS provides 64-bit integer support in two build modes: 3265a5399872SJunchao Zhang full 64-bit: here MUMPS is built with C preprocessing flag -DINTSIZE64 and Fortran compiler option -i8, -fdefault-integer-8 or equivalent, and 3266a5399872SJunchao 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). 32678fcaa860SBarry Smith 3268a5399872SJunchao 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, 3269a5399872SJunchao 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 3270a5399872SJunchao 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 3271a5399872SJunchao Zhang integer) build of all dependent libraries MPI, ScaLAPACK, LAPACK and BLAS. 3272a5399872SJunchao Zhang 3273a5399872SJunchao 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. 3274a5399872SJunchao Zhang 3275a5399872SJunchao Zhang Two modes to run MUMPS/PETSc with OpenMP 32762ef1f0ffSBarry Smith .vb 32772ef1f0ffSBarry Smith Set OMP_NUM_THREADS and run with fewer MPI ranks than cores. For example, if you want to have 16 OpenMP 32782ef1f0ffSBarry Smith threads per rank, then you may use "export OMP_NUM_THREADS=16 && mpirun -n 4 ./test". 32792ef1f0ffSBarry Smith .ve 32808fcaa860SBarry Smith 32812ef1f0ffSBarry Smith .vb 32822ef1f0ffSBarry Smith -mat_mumps_use_omp_threads [m] and run your code with as many MPI ranks as the number of cores. For example, 32832ef1f0ffSBarry 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" 32842ef1f0ffSBarry Smith .ve 32858fcaa860SBarry Smith 32868fcaa860SBarry Smith To run MUMPS in MPI+OpenMP hybrid mode (i.e., enable multithreading in MUMPS), but still run the non-MUMPS part 32872ef1f0ffSBarry 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` 32882ef1f0ffSBarry 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 32898fcaa860SBarry Smith libraries, to really get performance, you should have multithreaded BLAS libraries such as Intel MKL, AMD ACML, Cray libSci or OpenBLAS 32908fcaa860SBarry Smith (PETSc will automatically try to utilized a threaded BLAS if --with-openmp is provided). 3291217d3b1eSJunchao Zhang 32928fcaa860SBarry 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 3293217d3b1eSJunchao Zhang processes on each compute node. Listing the processes in rank ascending order, we split processes on a node into consecutive groups of 3294217d3b1eSJunchao 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 3295217d3b1eSJunchao 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 3296217d3b1eSJunchao 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. 3297217d3b1eSJunchao 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, 3298217d3b1eSJunchao 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 3299217d3b1eSJunchao 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 3300217d3b1eSJunchao 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 3301217d3b1eSJunchao 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. 33028fcaa860SBarry 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 3303217d3b1eSJunchao Zhang examine the mapping result. 3304217d3b1eSJunchao Zhang 330511a5261eSBarry 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, 330611a5261eSBarry 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 330711a5261eSBarry Smith calls `omp_set_num_threads`(m) internally before calling MUMPS. 3308217d3b1eSJunchao Zhang 33091d27aa22SBarry Smith See {cite}`heroux2011bi` and {cite}`gutierrez2017accommodating` 3310217d3b1eSJunchao Zhang 33111cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `PCFactorSetMatSolverType()`, `MatSolverType`, `MatMumpsSetIcntl()`, `MatMumpsGetIcntl()`, `MatMumpsSetCntl()`, `MatMumpsGetCntl()`, `MatMumpsGetInfo()`, `MatMumpsGetInfog()`, `MatMumpsGetRinfo()`, `MatMumpsGetRinfog()`, `KSPGetPC()`, `PCFactorGetMatrix()` 331224b6179bSKris Buschelman M*/ 331324b6179bSKris Buschelman 3314d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatFactorGetSolverType_mumps(Mat A, MatSolverType *type) 3315d71ae5a4SJacob Faibussowitsch { 331635bd34faSBarry Smith PetscFunctionBegin; 33172692d6eeSBarry Smith *type = MATSOLVERMUMPS; 33183ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 331935bd34faSBarry Smith } 332035bd34faSBarry Smith 3321bccb9932SShri Abhyankar /* MatGetFactor for Seq and MPI AIJ matrices */ 3322d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_aij_mumps(Mat A, MatFactorType ftype, Mat *F) 3323d71ae5a4SJacob Faibussowitsch { 33242877fffaSHong Zhang Mat B; 33252877fffaSHong Zhang Mat_MUMPS *mumps; 3326c3e1b152SPierre Jolivet PetscBool isSeqAIJ, isDiag; 33272c7c0729SBarry Smith PetscMPIInt size; 33282877fffaSHong Zhang 33292877fffaSHong Zhang PetscFunctionBegin; 3330eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX) 333103e5aca4SStefano Zampini if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) { 333203e5aca4SStefano Zampini PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n")); 333303e5aca4SStefano Zampini *F = NULL; 333403e5aca4SStefano Zampini PetscFunctionReturn(PETSC_SUCCESS); 333503e5aca4SStefano Zampini } 3336eb1ec7c1SStefano Zampini #endif 33372877fffaSHong Zhang /* Create the factorization matrix */ 33389566063dSJacob Faibussowitsch PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATSEQAIJ, &isSeqAIJ)); 3339c3e1b152SPierre Jolivet PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATDIAGONAL, &isDiag)); 33409566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B)); 33419566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); 33429566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name)); 33439566063dSJacob Faibussowitsch PetscCall(MatSetUp(B)); 33442877fffaSHong Zhang 33454dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&mumps)); 33462205254eSKarl Rupp 33472877fffaSHong Zhang B->ops->view = MatView_MUMPS; 334835bd34faSBarry Smith B->ops->getinfo = MatGetInfo_MUMPS; 33492205254eSKarl Rupp 33509566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps)); 33519566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS)); 33529566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS)); 33539566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS)); 33549566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS)); 33559566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS)); 33569566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS)); 33579566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS)); 33589566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS)); 33599566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS)); 33609566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS)); 33615c0bae8cSAshish Patel PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS)); 33629566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS)); 33639566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS)); 33646444a565SStefano Zampini 3365450b117fSShri Abhyankar if (ftype == MAT_FACTOR_LU) { 3366450b117fSShri Abhyankar B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS; 3367d5f3da31SBarry Smith B->factortype = MAT_FACTOR_LU; 3368bccb9932SShri Abhyankar if (isSeqAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqaij_seqaij; 3369c3e1b152SPierre Jolivet else if (isDiag) mumps->ConvertToTriples = MatConvertToTriples_diagonal_xaij; 3370bccb9932SShri Abhyankar else mumps->ConvertToTriples = MatConvertToTriples_mpiaij_mpiaij; 33719566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU])); 3372746480a1SHong Zhang mumps->sym = 0; 3373dcd589f8SShri Abhyankar } else { 337467877ebaSShri Abhyankar B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS; 3375450b117fSShri Abhyankar B->factortype = MAT_FACTOR_CHOLESKY; 3376bccb9932SShri Abhyankar if (isSeqAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqaij_seqsbaij; 3377c3e1b152SPierre Jolivet else if (isDiag) mumps->ConvertToTriples = MatConvertToTriples_diagonal_xaij; 3378bccb9932SShri Abhyankar else mumps->ConvertToTriples = MatConvertToTriples_mpiaij_mpisbaij; 33799566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_CHOLESKY])); 338059ac8732SStefano Zampini #if defined(PETSC_USE_COMPLEX) 338159ac8732SStefano Zampini mumps->sym = 2; 338259ac8732SStefano Zampini #else 3383b94d7dedSBarry Smith if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1; 33846fdc2a6dSBarry Smith else mumps->sym = 2; 338559ac8732SStefano Zampini #endif 3386450b117fSShri Abhyankar } 33872877fffaSHong Zhang 338800c67f3bSHong Zhang /* set solvertype */ 33899566063dSJacob Faibussowitsch PetscCall(PetscFree(B->solvertype)); 33909566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype)); 33919566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size)); 33922c7c0729SBarry Smith if (size == 1) { 33934ac6704cSBarry Smith /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */ 3394f73b0415SBarry Smith B->canuseordering = PETSC_TRUE; 33952c7c0729SBarry Smith } 33962877fffaSHong Zhang B->ops->destroy = MatDestroy_MUMPS; 3397e69c285eSBarry Smith B->data = (void *)mumps; 33982205254eSKarl Rupp 33992877fffaSHong Zhang *F = B; 3400413bcc21SPierre Jolivet mumps->id.job = JOB_NULL; 3401413bcc21SPierre Jolivet mumps->ICNTL_pre = NULL; 3402413bcc21SPierre Jolivet mumps->CNTL_pre = NULL; 3403d47f36abSHong Zhang mumps->matstruc = DIFFERENT_NONZERO_PATTERN; 34043ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 34052877fffaSHong Zhang } 34062877fffaSHong Zhang 3407bccb9932SShri Abhyankar /* MatGetFactor for Seq and MPI SBAIJ matrices */ 3408d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_sbaij_mumps(Mat A, MatFactorType ftype, Mat *F) 3409d71ae5a4SJacob Faibussowitsch { 34102877fffaSHong Zhang Mat B; 34112877fffaSHong Zhang Mat_MUMPS *mumps; 3412ace3abfcSBarry Smith PetscBool isSeqSBAIJ; 34132c7c0729SBarry Smith PetscMPIInt size; 34142877fffaSHong Zhang 34152877fffaSHong Zhang PetscFunctionBegin; 3416eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX) 341703e5aca4SStefano Zampini if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) { 341803e5aca4SStefano Zampini PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n")); 341903e5aca4SStefano Zampini *F = NULL; 342003e5aca4SStefano Zampini PetscFunctionReturn(PETSC_SUCCESS); 342103e5aca4SStefano Zampini } 3422eb1ec7c1SStefano Zampini #endif 34239566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B)); 34249566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); 34259566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name)); 34269566063dSJacob Faibussowitsch PetscCall(MatSetUp(B)); 3427e69c285eSBarry Smith 34284dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&mumps)); 34299566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQSBAIJ, &isSeqSBAIJ)); 3430bccb9932SShri Abhyankar if (isSeqSBAIJ) { 343116ebf90aSShri Abhyankar mumps->ConvertToTriples = MatConvertToTriples_seqsbaij_seqsbaij; 3432dcd589f8SShri Abhyankar } else { 3433bccb9932SShri Abhyankar mumps->ConvertToTriples = MatConvertToTriples_mpisbaij_mpisbaij; 3434bccb9932SShri Abhyankar } 3435bccb9932SShri Abhyankar 343667877ebaSShri Abhyankar B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS; 3437bccb9932SShri Abhyankar B->ops->view = MatView_MUMPS; 3438722b6324SPierre Jolivet B->ops->getinfo = MatGetInfo_MUMPS; 34392205254eSKarl Rupp 34409566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps)); 34419566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS)); 34429566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS)); 34439566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS)); 34449566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS)); 34459566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS)); 34469566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS)); 34479566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS)); 34489566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS)); 34499566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS)); 34509566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS)); 34515c0bae8cSAshish Patel PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS)); 34529566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS)); 34539566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS)); 34542205254eSKarl Rupp 3455f4762488SHong Zhang B->factortype = MAT_FACTOR_CHOLESKY; 345659ac8732SStefano Zampini #if defined(PETSC_USE_COMPLEX) 345759ac8732SStefano Zampini mumps->sym = 2; 345859ac8732SStefano Zampini #else 3459b94d7dedSBarry Smith if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1; 34606fdc2a6dSBarry Smith else mumps->sym = 2; 346159ac8732SStefano Zampini #endif 3462a214ac2aSShri Abhyankar 346300c67f3bSHong Zhang /* set solvertype */ 34649566063dSJacob Faibussowitsch PetscCall(PetscFree(B->solvertype)); 34659566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype)); 34669566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size)); 34672c7c0729SBarry Smith if (size == 1) { 34684ac6704cSBarry Smith /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */ 3469f73b0415SBarry Smith B->canuseordering = PETSC_TRUE; 34702c7c0729SBarry Smith } 34719566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_CHOLESKY])); 3472f3c0ef26SHong Zhang B->ops->destroy = MatDestroy_MUMPS; 3473e69c285eSBarry Smith B->data = (void *)mumps; 34742205254eSKarl Rupp 34752877fffaSHong Zhang *F = B; 3476413bcc21SPierre Jolivet mumps->id.job = JOB_NULL; 3477413bcc21SPierre Jolivet mumps->ICNTL_pre = NULL; 3478413bcc21SPierre Jolivet mumps->CNTL_pre = NULL; 3479d47f36abSHong Zhang mumps->matstruc = DIFFERENT_NONZERO_PATTERN; 34803ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 34812877fffaSHong Zhang } 348297969023SHong Zhang 3483d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_baij_mumps(Mat A, MatFactorType ftype, Mat *F) 3484d71ae5a4SJacob Faibussowitsch { 348567877ebaSShri Abhyankar Mat B; 348667877ebaSShri Abhyankar Mat_MUMPS *mumps; 3487ace3abfcSBarry Smith PetscBool isSeqBAIJ; 34882c7c0729SBarry Smith PetscMPIInt size; 348967877ebaSShri Abhyankar 349067877ebaSShri Abhyankar PetscFunctionBegin; 349167877ebaSShri Abhyankar /* Create the factorization matrix */ 34929566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQBAIJ, &isSeqBAIJ)); 34939566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B)); 34949566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); 34959566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name)); 34969566063dSJacob Faibussowitsch PetscCall(MatSetUp(B)); 3497450b117fSShri Abhyankar 34984dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&mumps)); 3499450b117fSShri Abhyankar if (ftype == MAT_FACTOR_LU) { 3500450b117fSShri Abhyankar B->ops->lufactorsymbolic = MatLUFactorSymbolic_BAIJMUMPS; 3501450b117fSShri Abhyankar B->factortype = MAT_FACTOR_LU; 3502bccb9932SShri Abhyankar if (isSeqBAIJ) mumps->ConvertToTriples = MatConvertToTriples_seqbaij_seqaij; 3503bccb9932SShri Abhyankar else mumps->ConvertToTriples = MatConvertToTriples_mpibaij_mpiaij; 3504746480a1SHong Zhang mumps->sym = 0; 35059566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU])); 3506546078acSJacob Faibussowitsch } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot use PETSc BAIJ matrices with MUMPS Cholesky, use SBAIJ or AIJ matrix instead"); 3507bccb9932SShri Abhyankar 3508450b117fSShri Abhyankar B->ops->view = MatView_MUMPS; 3509722b6324SPierre Jolivet B->ops->getinfo = MatGetInfo_MUMPS; 35102205254eSKarl Rupp 35119566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps)); 35129566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS)); 35139566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS)); 35149566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS)); 35159566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS)); 35169566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS)); 35179566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS)); 35189566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS)); 35199566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS)); 35209566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS)); 35219566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS)); 35225c0bae8cSAshish Patel PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS)); 35239566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS)); 35249566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS)); 3525450b117fSShri Abhyankar 352600c67f3bSHong Zhang /* set solvertype */ 35279566063dSJacob Faibussowitsch PetscCall(PetscFree(B->solvertype)); 35289566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype)); 35299566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size)); 35302c7c0729SBarry Smith if (size == 1) { 35314ac6704cSBarry Smith /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */ 3532f73b0415SBarry Smith B->canuseordering = PETSC_TRUE; 35332c7c0729SBarry Smith } 35347ee00b23SStefano Zampini B->ops->destroy = MatDestroy_MUMPS; 35357ee00b23SStefano Zampini B->data = (void *)mumps; 35367ee00b23SStefano Zampini 35377ee00b23SStefano Zampini *F = B; 3538413bcc21SPierre Jolivet mumps->id.job = JOB_NULL; 3539413bcc21SPierre Jolivet mumps->ICNTL_pre = NULL; 3540413bcc21SPierre Jolivet mumps->CNTL_pre = NULL; 3541d47f36abSHong Zhang mumps->matstruc = DIFFERENT_NONZERO_PATTERN; 35423ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 35437ee00b23SStefano Zampini } 35447ee00b23SStefano Zampini 35457ee00b23SStefano Zampini /* MatGetFactor for Seq and MPI SELL matrices */ 3546d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetFactor_sell_mumps(Mat A, MatFactorType ftype, Mat *F) 3547d71ae5a4SJacob Faibussowitsch { 35487ee00b23SStefano Zampini Mat B; 35497ee00b23SStefano Zampini Mat_MUMPS *mumps; 35507ee00b23SStefano Zampini PetscBool isSeqSELL; 35512c7c0729SBarry Smith PetscMPIInt size; 35527ee00b23SStefano Zampini 35537ee00b23SStefano Zampini PetscFunctionBegin; 35547ee00b23SStefano Zampini /* Create the factorization matrix */ 35559566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQSELL, &isSeqSELL)); 35569566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B)); 35579566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); 35589566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy("mumps", &((PetscObject)B)->type_name)); 35599566063dSJacob Faibussowitsch PetscCall(MatSetUp(B)); 35607ee00b23SStefano Zampini 35614dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&mumps)); 35627ee00b23SStefano Zampini 35637ee00b23SStefano Zampini B->ops->view = MatView_MUMPS; 35647ee00b23SStefano Zampini B->ops->getinfo = MatGetInfo_MUMPS; 35657ee00b23SStefano Zampini 35669566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps)); 35679566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS)); 35689566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS)); 35699566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS)); 35709566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS)); 35719566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS)); 35729566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS)); 35739566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS)); 35749566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS)); 35759566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS)); 35769566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS)); 35775c0bae8cSAshish Patel PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS)); 35787ee00b23SStefano Zampini 35797ee00b23SStefano Zampini if (ftype == MAT_FACTOR_LU) { 35807ee00b23SStefano Zampini B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS; 35817ee00b23SStefano Zampini B->factortype = MAT_FACTOR_LU; 35827ee00b23SStefano Zampini if (isSeqSELL) mumps->ConvertToTriples = MatConvertToTriples_seqsell_seqaij; 35837ee00b23SStefano Zampini else SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "To be implemented"); 35847ee00b23SStefano Zampini mumps->sym = 0; 35859566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[MAT_FACTOR_LU])); 35867ee00b23SStefano Zampini } else SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "To be implemented"); 35877ee00b23SStefano Zampini 35887ee00b23SStefano Zampini /* set solvertype */ 35899566063dSJacob Faibussowitsch PetscCall(PetscFree(B->solvertype)); 35909566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype)); 35919566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size)); 35922c7c0729SBarry Smith if (size == 1) { 35934ac6704cSBarry Smith /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */ 3594f73b0415SBarry Smith B->canuseordering = PETSC_TRUE; 35952c7c0729SBarry Smith } 3596450b117fSShri Abhyankar B->ops->destroy = MatDestroy_MUMPS; 3597e69c285eSBarry Smith B->data = (void *)mumps; 35982205254eSKarl Rupp 3599450b117fSShri Abhyankar *F = B; 3600413bcc21SPierre Jolivet mumps->id.job = JOB_NULL; 3601413bcc21SPierre Jolivet mumps->ICNTL_pre = NULL; 3602413bcc21SPierre Jolivet mumps->CNTL_pre = NULL; 3603d47f36abSHong Zhang mumps->matstruc = DIFFERENT_NONZERO_PATTERN; 36043ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3605450b117fSShri Abhyankar } 360642c9c57cSBarry Smith 36079d0448ceSStefano Zampini /* MatGetFactor for MATNEST matrices */ 36089d0448ceSStefano Zampini static PetscErrorCode MatGetFactor_nest_mumps(Mat A, MatFactorType ftype, Mat *F) 36099d0448ceSStefano Zampini { 36109d0448ceSStefano Zampini Mat B, **mats; 36119d0448ceSStefano Zampini Mat_MUMPS *mumps; 36129d0448ceSStefano Zampini PetscInt nr, nc; 36139d0448ceSStefano Zampini PetscMPIInt size; 361403e5aca4SStefano Zampini PetscBool flg = PETSC_TRUE; 36159d0448ceSStefano Zampini 36169d0448ceSStefano Zampini PetscFunctionBegin; 36179d0448ceSStefano Zampini #if defined(PETSC_USE_COMPLEX) 361803e5aca4SStefano Zampini if (ftype == MAT_FACTOR_CHOLESKY && A->hermitian == PETSC_BOOL3_TRUE && A->symmetric != PETSC_BOOL3_TRUE) { 361903e5aca4SStefano Zampini PetscCall(PetscInfo(A, "Hermitian MAT_FACTOR_CHOLESKY is not supported. Use MAT_FACTOR_LU instead.\n")); 362003e5aca4SStefano Zampini *F = NULL; 362103e5aca4SStefano Zampini PetscFunctionReturn(PETSC_SUCCESS); 362203e5aca4SStefano Zampini } 36239d0448ceSStefano Zampini #endif 36249d0448ceSStefano Zampini 362503e5aca4SStefano Zampini /* Return if some condition is not satisfied */ 362603e5aca4SStefano Zampini *F = NULL; 36279d0448ceSStefano Zampini PetscCall(MatNestGetSubMats(A, &nr, &nc, &mats)); 36289d0448ceSStefano Zampini if (ftype == MAT_FACTOR_CHOLESKY) { 36299d0448ceSStefano Zampini IS *rows, *cols; 36309d0448ceSStefano Zampini PetscInt *m, *M; 36319d0448ceSStefano Zampini 36329d0448ceSStefano 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); 36339d0448ceSStefano Zampini PetscCall(PetscMalloc2(nr, &rows, nc, &cols)); 36349d0448ceSStefano Zampini PetscCall(MatNestGetISs(A, rows, cols)); 36359d0448ceSStefano Zampini for (PetscInt r = 0; flg && r < nr; r++) PetscCall(ISEqualUnsorted(rows[r], cols[r], &flg)); 363603e5aca4SStefano Zampini if (!flg) { 363703e5aca4SStefano Zampini PetscCall(PetscFree2(rows, cols)); 363803e5aca4SStefano Zampini PetscCall(PetscInfo(A, "MAT_FACTOR_CHOLESKY not supported for unequal row and column maps. Use MAT_FACTOR_LU.\n")); 363903e5aca4SStefano Zampini PetscFunctionReturn(PETSC_SUCCESS); 364003e5aca4SStefano Zampini } 36419d0448ceSStefano Zampini PetscCall(PetscMalloc2(nr, &m, nr, &M)); 36429d0448ceSStefano Zampini for (PetscInt r = 0; r < nr; r++) PetscCall(ISGetMinMax(rows[r], &m[r], &M[r])); 36439d0448ceSStefano Zampini for (PetscInt r = 0; flg && r < nr; r++) 36449d0448ceSStefano Zampini for (PetscInt k = r + 1; flg && k < nr; k++) 36459d0448ceSStefano Zampini if ((m[k] <= m[r] && m[r] <= M[k]) || (m[k] <= M[r] && M[r] <= M[k])) flg = PETSC_FALSE; 36469d0448ceSStefano Zampini PetscCall(PetscFree2(m, M)); 36479d0448ceSStefano Zampini PetscCall(PetscFree2(rows, cols)); 364803e5aca4SStefano Zampini if (!flg) { 364903e5aca4SStefano Zampini PetscCall(PetscInfo(A, "MAT_FACTOR_CHOLESKY not supported for intersecting row maps. Use MAT_FACTOR_LU.\n")); 365003e5aca4SStefano Zampini PetscFunctionReturn(PETSC_SUCCESS); 365103e5aca4SStefano Zampini } 36529d0448ceSStefano Zampini } 36539d0448ceSStefano Zampini 36549d0448ceSStefano Zampini for (PetscInt r = 0; r < nr; r++) { 36559d0448ceSStefano Zampini for (PetscInt c = 0; c < nc; c++) { 36569d0448ceSStefano Zampini Mat sub = mats[r][c]; 3657c3e1b152SPierre Jolivet PetscBool isSeqAIJ, isMPIAIJ, isSeqBAIJ, isMPIBAIJ, isSeqSBAIJ, isMPISBAIJ, isTrans, isDiag; 36589d0448ceSStefano Zampini 36599d0448ceSStefano Zampini if (!sub || (ftype == MAT_FACTOR_CHOLESKY && c < r)) continue; 36605d955bbbSStefano Zampini PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATTRANSPOSEVIRTUAL, &isTrans)); 36615d955bbbSStefano Zampini if (isTrans) PetscCall(MatTransposeGetMat(sub, &sub)); 36625d955bbbSStefano Zampini else { 36635d955bbbSStefano Zampini PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATHERMITIANTRANSPOSEVIRTUAL, &isTrans)); 36645d955bbbSStefano Zampini if (isTrans) PetscCall(MatHermitianTransposeGetMat(sub, &sub)); 36655d955bbbSStefano Zampini } 36669d0448ceSStefano Zampini PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQAIJ, &isSeqAIJ)); 36679d0448ceSStefano Zampini PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIAIJ, &isMPIAIJ)); 36689d0448ceSStefano Zampini PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQBAIJ, &isSeqBAIJ)); 36699d0448ceSStefano Zampini PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPIBAIJ, &isMPIBAIJ)); 36709d0448ceSStefano Zampini PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATSEQSBAIJ, &isSeqSBAIJ)); 36719d0448ceSStefano Zampini PetscCall(PetscObjectBaseTypeCompare((PetscObject)sub, MATMPISBAIJ, &isMPISBAIJ)); 3672c3e1b152SPierre Jolivet PetscCall(PetscObjectTypeCompare((PetscObject)sub, MATDIAGONAL, &isDiag)); 36739d0448ceSStefano Zampini if (ftype == MAT_FACTOR_CHOLESKY) { 3674dcab004fSPierre Jolivet if (r == c) { 3675c3e1b152SPierre Jolivet if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isSeqSBAIJ && !isMPISBAIJ && !isDiag) { 367603e5aca4SStefano Zampini PetscCall(PetscInfo(sub, "MAT_CHOLESKY_FACTOR not supported for diagonal block of type %s.\n", ((PetscObject)sub)->type_name)); 367703e5aca4SStefano Zampini flg = PETSC_FALSE; 3678dcab004fSPierre Jolivet } 3679c3e1b152SPierre Jolivet } else if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isDiag) { 368003e5aca4SStefano Zampini PetscCall(PetscInfo(sub, "MAT_CHOLESKY_FACTOR not supported for off-diagonal block of type %s.\n", ((PetscObject)sub)->type_name)); 368103e5aca4SStefano Zampini flg = PETSC_FALSE; 368203e5aca4SStefano Zampini } 3683c3e1b152SPierre Jolivet } else if (!isSeqAIJ && !isMPIAIJ && !isSeqBAIJ && !isMPIBAIJ && !isDiag) { 368403e5aca4SStefano Zampini PetscCall(PetscInfo(sub, "MAT_LU_FACTOR not supported for block of type %s.\n", ((PetscObject)sub)->type_name)); 368503e5aca4SStefano Zampini flg = PETSC_FALSE; 36869d0448ceSStefano Zampini } 36879d0448ceSStefano Zampini } 368803e5aca4SStefano Zampini } 368903e5aca4SStefano Zampini if (!flg) PetscFunctionReturn(PETSC_SUCCESS); 36909d0448ceSStefano Zampini 36919d0448ceSStefano Zampini /* Create the factorization matrix */ 36929d0448ceSStefano Zampini PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B)); 36939d0448ceSStefano Zampini PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); 36949d0448ceSStefano Zampini PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &((PetscObject)B)->type_name)); 36959d0448ceSStefano Zampini PetscCall(MatSetUp(B)); 36969d0448ceSStefano Zampini 36979d0448ceSStefano Zampini PetscCall(PetscNew(&mumps)); 36989d0448ceSStefano Zampini 36999d0448ceSStefano Zampini B->ops->view = MatView_MUMPS; 37009d0448ceSStefano Zampini B->ops->getinfo = MatGetInfo_MUMPS; 37019d0448ceSStefano Zampini 37029d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorGetSolverType_C", MatFactorGetSolverType_mumps)); 37039d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorSetSchurIS_C", MatFactorSetSchurIS_MUMPS)); 37049d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatFactorCreateSchurComplement_C", MatFactorCreateSchurComplement_MUMPS)); 37059d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetIcntl_C", MatMumpsSetIcntl_MUMPS)); 37069d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetIcntl_C", MatMumpsGetIcntl_MUMPS)); 37079d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsSetCntl_C", MatMumpsSetCntl_MUMPS)); 37089d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetCntl_C", MatMumpsGetCntl_MUMPS)); 37099d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfo_C", MatMumpsGetInfo_MUMPS)); 37109d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInfog_C", MatMumpsGetInfog_MUMPS)); 37119d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfo_C", MatMumpsGetRinfo_MUMPS)); 37129d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetRinfog_C", MatMumpsGetRinfog_MUMPS)); 37139d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetNullPivots_C", MatMumpsGetNullPivots_MUMPS)); 37149d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverse_C", MatMumpsGetInverse_MUMPS)); 37159d0448ceSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatMumpsGetInverseTranspose_C", MatMumpsGetInverseTranspose_MUMPS)); 37169d0448ceSStefano Zampini 37179d0448ceSStefano Zampini if (ftype == MAT_FACTOR_LU) { 37189d0448ceSStefano Zampini B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMUMPS; 37199d0448ceSStefano Zampini B->factortype = MAT_FACTOR_LU; 37209d0448ceSStefano Zampini mumps->sym = 0; 37219d0448ceSStefano Zampini } else { 37229d0448ceSStefano Zampini B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MUMPS; 37239d0448ceSStefano Zampini B->factortype = MAT_FACTOR_CHOLESKY; 37249d0448ceSStefano Zampini #if defined(PETSC_USE_COMPLEX) 37259d0448ceSStefano Zampini mumps->sym = 2; 37269d0448ceSStefano Zampini #else 37279d0448ceSStefano Zampini if (A->spd == PETSC_BOOL3_TRUE) mumps->sym = 1; 37289d0448ceSStefano Zampini else mumps->sym = 2; 37299d0448ceSStefano Zampini #endif 37309d0448ceSStefano Zampini } 37319d0448ceSStefano Zampini mumps->ConvertToTriples = MatConvertToTriples_nest_xaij; 37329d0448ceSStefano Zampini PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&B->preferredordering[ftype])); 37339d0448ceSStefano Zampini 37349d0448ceSStefano Zampini PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)A), &size)); 37359d0448ceSStefano Zampini if (size == 1) { 37369d0448ceSStefano Zampini /* MUMPS option -mat_mumps_icntl_7 1 is automatically set if PETSc ordering is passed into symbolic factorization */ 37379d0448ceSStefano Zampini B->canuseordering = PETSC_TRUE; 37389d0448ceSStefano Zampini } 37399d0448ceSStefano Zampini 37409d0448ceSStefano Zampini /* set solvertype */ 37419d0448ceSStefano Zampini PetscCall(PetscFree(B->solvertype)); 37429d0448ceSStefano Zampini PetscCall(PetscStrallocpy(MATSOLVERMUMPS, &B->solvertype)); 37439d0448ceSStefano Zampini B->ops->destroy = MatDestroy_MUMPS; 37449d0448ceSStefano Zampini B->data = (void *)mumps; 37459d0448ceSStefano Zampini 37469d0448ceSStefano Zampini *F = B; 37479d0448ceSStefano Zampini mumps->id.job = JOB_NULL; 37489d0448ceSStefano Zampini mumps->ICNTL_pre = NULL; 37499d0448ceSStefano Zampini mumps->CNTL_pre = NULL; 37509d0448ceSStefano Zampini mumps->matstruc = DIFFERENT_NONZERO_PATTERN; 37519d0448ceSStefano Zampini PetscFunctionReturn(PETSC_SUCCESS); 37529d0448ceSStefano Zampini } 37539d0448ceSStefano Zampini 3754d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_MUMPS(void) 3755d71ae5a4SJacob Faibussowitsch { 375642c9c57cSBarry Smith PetscFunctionBegin; 37579566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIAIJ, MAT_FACTOR_LU, MatGetFactor_aij_mumps)); 37589566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps)); 37599566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIBAIJ, MAT_FACTOR_LU, MatGetFactor_baij_mumps)); 37609566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPIBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_baij_mumps)); 37619566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATMPISBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_sbaij_mumps)); 37629566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQAIJ, MAT_FACTOR_LU, MatGetFactor_aij_mumps)); 37639566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps)); 37649566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQBAIJ, MAT_FACTOR_LU, MatGetFactor_baij_mumps)); 37659566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_baij_mumps)); 37669566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQSBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_sbaij_mumps)); 37679566063dSJacob Faibussowitsch PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATSEQSELL, MAT_FACTOR_LU, MatGetFactor_sell_mumps)); 3768c3e1b152SPierre Jolivet PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATDIAGONAL, MAT_FACTOR_LU, MatGetFactor_aij_mumps)); 3769c3e1b152SPierre Jolivet PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATDIAGONAL, MAT_FACTOR_CHOLESKY, MatGetFactor_aij_mumps)); 37709d0448ceSStefano Zampini PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATNEST, MAT_FACTOR_LU, MatGetFactor_nest_mumps)); 37719d0448ceSStefano Zampini PetscCall(MatSolverTypeRegister(MATSOLVERMUMPS, MATNEST, MAT_FACTOR_CHOLESKY, MatGetFactor_nest_mumps)); 37723ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 377342c9c57cSBarry Smith } 3774