Actual source code: dlregismat.c
1: /* Portions of this code are under:
2: Copyright (c) 2022 Advanced Micro Devices, Inc. All rights reserved.
3: */
4: #include <petsc/private/matimpl.h>
6: const char *MatOptions_Shifted[] = {"UNUSED_NONZERO_LOCATION_ERR", "ROW_ORIENTED", "NOT_A_VALID_OPTION", "SYMMETRIC", "STRUCTURALLY_SYMMETRIC", "FORCE_DIAGONAL_ENTRIES", "IGNORE_OFF_PROC_ENTRIES", "USE_HASH_TABLE", "KEEP_NONZERO_PATTERN", "IGNORE_ZERO_ENTRIES", "USE_INODES", "HERMITIAN", "SYMMETRY_ETERNAL", "NEW_NONZERO_LOCATION_ERR", "IGNORE_LOWER_TRIANGULAR", "ERROR_LOWER_TRIANGULAR", "GETROW_UPPERTRIANGULAR", "SPD", "NO_OFF_PROC_ZERO_ROWS", "NO_OFF_PROC_ENTRIES", "NEW_NONZERO_LOCATIONS", "NEW_NONZERO_ALLOCATION_ERR", "SUBSET_OFF_PROC_ENTRIES", "SUBMAT_SINGLEIS", "STRUCTURE_ONLY", "SORTED_FULL", "FORM_EXPLICIT_TRANSPOSE", "STRUCTURAL_SYMMETRY_ETERNAL", "SPD_ETERNAL", "MatOption", "MAT_", NULL};
7: const char *const *MatOptions = MatOptions_Shifted + 2;
8: const char *const MatFactorShiftTypes[] = {"NONE", "NONZERO", "POSITIVE_DEFINITE", "INBLOCKS", "MatFactorShiftType", "PC_FACTOR_", NULL};
9: const char *const MatStructures[] = {"DIFFERENT", "SUBSET", "SAME", "UNKNOWN", "MatStructure", "MAT_STRUCTURE_", NULL};
10: const char *const MatFactorShiftTypesDetail[] = {NULL, "diagonal shift to prevent zero pivot", "Manteuffel shift", "diagonal shift on blocks to prevent zero pivot"};
11: const char *const MPPTScotchStrategyTypes[] = {"DEFAULT", "QUALITY", "SPEED", "BALANCE", "SAFETY", "SCALABILITY", "MPPTScotchStrategyType", "MP_PTSCOTCH_", NULL};
12: const char *const MPChacoGlobalTypes[] = {"", "MULTILEVEL", "SPECTRAL", "", "LINEAR", "RANDOM", "SCATTERED", "MPChacoGlobalType", "MP_CHACO_", NULL};
13: const char *const MPChacoLocalTypes[] = {"", "KERNIGHAN", "NONE", "MPChacoLocalType", "MP_CHACO_", NULL};
14: const char *const MPChacoEigenTypes[] = {"LANCZOS", "RQI", "MPChacoEigenType", "MP_CHACO_", NULL};
16: extern PetscErrorCode MatMFFDInitializePackage(void);
17: extern PetscErrorCode MatSolverTypeDestroy(void);
18: static PetscBool MatPackageInitialized = PETSC_FALSE;
19: /*@C
20: MatFinalizePackage - This function destroys everything in the Petsc interface to the `Mat` package. It is
21: called from `PetscFinalize()`.
23: Level: developer
25: .seealso: `Mat`, `PetscFinalize()`, `MatInitializePackage()`
26: @*/
27: PetscErrorCode MatFinalizePackage(void)
28: {
29: MatRootName nnames, names = MatRootNameList;
31: MatSolverTypeDestroy();
32: while (names) {
33: nnames = names->next;
34: PetscFree(names->rname);
35: PetscFree(names->sname);
36: PetscFree(names->mname);
37: PetscFree(names);
38: names = nnames;
39: }
40: PetscFunctionListDestroy(&MatList);
41: PetscFunctionListDestroy(&MatOrderingList);
42: PetscFunctionListDestroy(&MatColoringList);
43: PetscFunctionListDestroy(&MatPartitioningList);
44: PetscFunctionListDestroy(&MatCoarsenList);
45: MatRootNameList = NULL;
46: MatPackageInitialized = PETSC_FALSE;
47: MatRegisterAllCalled = PETSC_FALSE;
48: MatOrderingRegisterAllCalled = PETSC_FALSE;
49: MatColoringRegisterAllCalled = PETSC_FALSE;
50: MatPartitioningRegisterAllCalled = PETSC_FALSE;
51: MatCoarsenRegisterAllCalled = PETSC_FALSE;
52: /* this is not ideal because it exposes SeqAIJ implementation details directly into the base Mat code */
53: PetscFunctionListDestroy(&MatSeqAIJList);
54: MatSeqAIJRegisterAllCalled = PETSC_FALSE;
55: return 0;
56: }
58: #if defined(PETSC_HAVE_MUMPS)
59: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_MUMPS(void);
60: #endif
61: #if defined(PETSC_HAVE_CUDA)
62: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_CUSPARSE(void);
63: #endif
64: #if defined(PETSC_HAVE_HIP)
65: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_HIPSPARSE(void);
66: #endif
67: #if defined(PETSC_HAVE_KOKKOS_KERNELS)
68: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_KOKKOS(void);
69: #endif
70: #if defined(PETSC_HAVE_VIENNACL)
71: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_ViennaCL(void);
72: #endif
73: #if defined(PETSC_HAVE_ELEMENTAL)
74: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_Elemental(void);
75: #endif
76: #if defined(PETSC_HAVE_SCALAPACK)
77: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_ScaLAPACK(void);
78: #endif
79: #if defined(PETSC_HAVE_MATLAB)
80: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_Matlab(void);
81: #endif
82: #if defined(PETSC_HAVE_ESSL)
83: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_Essl(void);
84: #endif
85: #if defined(PETSC_HAVE_SUPERLU)
86: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_SuperLU(void);
87: #endif
88: #if defined(PETSC_HAVE_STRUMPACK)
89: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_STRUMPACK(void);
90: #endif
91: #if defined(PETSC_HAVE_PASTIX)
92: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_Pastix(void);
93: #endif
94: #if defined(PETSC_HAVE_SUPERLU_DIST)
95: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_SuperLU_DIST(void);
96: #endif
97: #if defined(PETSC_HAVE_ELEMENTAL)
98: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_SparseElemental(void);
99: #endif
100: #if defined(PETSC_HAVE_MKL_PARDISO)
101: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_MKL_Pardiso(void);
102: #endif
103: #if defined(PETSC_HAVE_MKL_CPARDISO)
104: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_MKL_CPardiso(void);
105: #endif
106: #if defined(PETSC_HAVE_SUITESPARSE)
107: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_SuiteSparse(void);
108: #endif
109: #if defined(PETSC_HAVE_LUSOL)
110: PETSC_EXTERN PetscErrorCode MatSolverTypeRegister_Lusol(void);
111: #endif
113: PETSC_INTERN PetscErrorCode MatGetFactor_seqaij_petsc(Mat, MatFactorType, Mat *);
114: PETSC_INTERN PetscErrorCode MatGetFactor_seqbaij_petsc(Mat, MatFactorType, Mat *);
115: PETSC_INTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat, MatFactorType, Mat *);
116: PETSC_INTERN PetscErrorCode MatGetFactor_seqdense_petsc(Mat, MatFactorType, Mat *);
117: #if defined(PETSC_HAVE_CUDA)
118: PETSC_INTERN PetscErrorCode MatGetFactor_seqdense_cuda(Mat, MatFactorType, Mat *);
119: #endif
120: #if defined(PETSC_HAVE_HIP)
121: PETSC_INTERN PetscErrorCode MatGetFactor_seqdense_hip(Mat, MatFactorType, Mat *);
122: #endif
123: PETSC_INTERN PetscErrorCode MatGetFactor_constantdiagonal_petsc(Mat, MatFactorType, Mat *);
124: PETSC_INTERN PetscErrorCode MatGetFactor_seqaij_bas(Mat, MatFactorType, Mat *);
126: /*@C
127: MatInitializePackage - This function initializes everything in the `Mat` package. It is called
128: from `PetscDLLibraryRegister_petscmat()` when using dynamic libraries, and on the first call to `MatCreate()`
129: when using shared or static libraries.
131: Level: developer
133: .seealso: `Mat`, `PetscInitialize()`, `MatFinalizePackage()`
134: @*/
135: PetscErrorCode MatInitializePackage(void)
136: {
137: char logList[256];
138: PetscBool opt, pkg;
140: if (MatPackageInitialized) return 0;
141: MatPackageInitialized = PETSC_TRUE;
142: /* Initialize subpackage */
143: MatMFFDInitializePackage();
144: /* Register Classes */
145: PetscClassIdRegister("Matrix", &MAT_CLASSID);
146: PetscClassIdRegister("Matrix FD Coloring", &MAT_FDCOLORING_CLASSID);
147: PetscClassIdRegister("Matrix Coloring", &MAT_COLORING_CLASSID);
148: PetscClassIdRegister("Matrix MatTranspose Coloring", &MAT_TRANSPOSECOLORING_CLASSID);
149: PetscClassIdRegister("Matrix Partitioning", &MAT_PARTITIONING_CLASSID);
150: PetscClassIdRegister("Matrix Coarsen", &MAT_COARSEN_CLASSID);
151: PetscClassIdRegister("Matrix Null Space", &MAT_NULLSPACE_CLASSID);
152: /* Register Constructors */
153: MatRegisterAll();
154: MatOrderingRegisterAll();
155: MatColoringRegisterAll();
156: MatPartitioningRegisterAll();
157: MatCoarsenRegisterAll();
158: MatSeqAIJRegisterAll();
159: /* Register Events */
160: PetscLogEventRegister("MatMult", MAT_CLASSID, &MAT_Mult);
161: PetscLogEventRegister("MatMults", MAT_CLASSID, &MAT_Mults);
162: PetscLogEventRegister("MatMultAdd", MAT_CLASSID, &MAT_MultAdd);
163: PetscLogEventRegister("MatMultTranspose", MAT_CLASSID, &MAT_MultTranspose);
164: PetscLogEventRegister("MatMultTrAdd", MAT_CLASSID, &MAT_MultTransposeAdd);
165: PetscLogEventRegister("MatSolve", MAT_CLASSID, &MAT_Solve);
166: PetscLogEventRegister("MatSolves", MAT_CLASSID, &MAT_Solves);
167: PetscLogEventRegister("MatSolveAdd", MAT_CLASSID, &MAT_SolveAdd);
168: PetscLogEventRegister("MatSolveTranspos", MAT_CLASSID, &MAT_SolveTranspose);
169: PetscLogEventRegister("MatSolveTrAdd", MAT_CLASSID, &MAT_SolveTransposeAdd);
170: PetscLogEventRegister("MatSOR", MAT_CLASSID, &MAT_SOR);
171: PetscLogEventRegister("MatForwardSolve", MAT_CLASSID, &MAT_ForwardSolve);
172: PetscLogEventRegister("MatBackwardSolve", MAT_CLASSID, &MAT_BackwardSolve);
173: PetscLogEventRegister("MatLUFactor", MAT_CLASSID, &MAT_LUFactor);
174: PetscLogEventRegister("MatLUFactorSym", MAT_CLASSID, &MAT_LUFactorSymbolic);
175: PetscLogEventRegister("MatLUFactorNum", MAT_CLASSID, &MAT_LUFactorNumeric);
176: PetscLogEventRegister("MatQRFactor", MAT_CLASSID, &MAT_QRFactor);
177: PetscLogEventRegister("MatQRFactorSym", MAT_CLASSID, &MAT_QRFactorSymbolic);
178: PetscLogEventRegister("MatQRFactorNum", MAT_CLASSID, &MAT_QRFactorNumeric);
179: PetscLogEventRegister("MatCholeskyFctr", MAT_CLASSID, &MAT_CholeskyFactor);
180: PetscLogEventRegister("MatCholFctrSym", MAT_CLASSID, &MAT_CholeskyFactorSymbolic);
181: PetscLogEventRegister("MatCholFctrNum", MAT_CLASSID, &MAT_CholeskyFactorNumeric);
182: PetscLogEventRegister("MatFctrFactSchur", MAT_CLASSID, &MAT_FactorFactS);
183: PetscLogEventRegister("MatFctrInvSchur", MAT_CLASSID, &MAT_FactorInvS);
184: PetscLogEventRegister("MatILUFactor", MAT_CLASSID, &MAT_ILUFactor);
185: PetscLogEventRegister("MatILUFactorSym", MAT_CLASSID, &MAT_ILUFactorSymbolic);
186: PetscLogEventRegister("MatICCFactorSym", MAT_CLASSID, &MAT_ICCFactorSymbolic);
187: PetscLogEventRegister("MatCopy", MAT_CLASSID, &MAT_Copy);
188: PetscLogEventRegister("MatConvert", MAT_CLASSID, &MAT_Convert);
189: PetscLogEventRegister("MatScale", MAT_CLASSID, &MAT_Scale);
190: PetscLogEventRegister("MatResidual", MAT_CLASSID, &MAT_Residual);
191: PetscLogEventRegister("MatAssemblyBegin", MAT_CLASSID, &MAT_AssemblyBegin);
192: PetscLogEventRegister("MatAssemblyEnd", MAT_CLASSID, &MAT_AssemblyEnd);
193: PetscLogEventRegister("MatSetValues", MAT_CLASSID, &MAT_SetValues);
194: PetscLogEventRegister("MatGetValues", MAT_CLASSID, &MAT_GetValues);
195: PetscLogEventRegister("MatGetRow", MAT_CLASSID, &MAT_GetRow);
196: PetscLogEventRegister("MatGetRowIJ", MAT_CLASSID, &MAT_GetRowIJ);
197: PetscLogEventRegister("MatCreateSubMats", MAT_CLASSID, &MAT_CreateSubMats);
198: PetscLogEventRegister("MatCreateSubMat", MAT_CLASSID, &MAT_CreateSubMat);
199: PetscLogEventRegister("MatGetOrdering", MAT_CLASSID, &MAT_GetOrdering);
200: PetscLogEventRegister("MatIncreaseOvrlp", MAT_CLASSID, &MAT_IncreaseOverlap);
201: PetscLogEventRegister("MatPartitioning", MAT_PARTITIONING_CLASSID, &MAT_Partitioning);
202: PetscLogEventRegister("MatPartitioningND", MAT_PARTITIONING_CLASSID, &MAT_PartitioningND);
203: PetscLogEventRegister("MatCoarsen", MAT_COARSEN_CLASSID, &MAT_Coarsen);
204: PetscLogEventRegister("MatZeroEntries", MAT_CLASSID, &MAT_ZeroEntries);
205: PetscLogEventRegister("MatLoad", MAT_CLASSID, &MAT_Load);
206: PetscLogEventRegister("MatView", MAT_CLASSID, &MAT_View);
207: PetscLogEventRegister("MatAXPY", MAT_CLASSID, &MAT_AXPY);
208: PetscLogEventRegister("MatFDColorCreate", MAT_FDCOLORING_CLASSID, &MAT_FDColoringCreate);
209: PetscLogEventRegister("MatFDColorSetUp", MAT_FDCOLORING_CLASSID, &MAT_FDColoringSetUp);
210: PetscLogEventRegister("MatFDColorApply", MAT_FDCOLORING_CLASSID, &MAT_FDColoringApply);
211: PetscLogEventRegister("MatFDColorFunc", MAT_FDCOLORING_CLASSID, &MAT_FDColoringFunction);
212: PetscLogEventRegister("MatTranspose", MAT_CLASSID, &MAT_Transpose);
213: PetscLogEventRegister("MatMatSolve", MAT_CLASSID, &MAT_MatSolve);
214: PetscLogEventRegister("MatMatTrSolve", MAT_CLASSID, &MAT_MatTrSolve);
215: PetscLogEventRegister("MatMatMultSym", MAT_CLASSID, &MAT_MatMultSymbolic);
216: PetscLogEventRegister("MatMatMultNum", MAT_CLASSID, &MAT_MatMultNumeric);
217: PetscLogEventRegister("MatMatMatMultSym", MAT_CLASSID, &MAT_MatMatMultSymbolic);
218: PetscLogEventRegister("MatMatMatMultNum", MAT_CLASSID, &MAT_MatMatMultNumeric);
219: PetscLogEventRegister("MatPtAPSymbolic", MAT_CLASSID, &MAT_PtAPSymbolic);
220: PetscLogEventRegister("MatPtAPNumeric", MAT_CLASSID, &MAT_PtAPNumeric);
221: PetscLogEventRegister("MatRARtSym", MAT_CLASSID, &MAT_RARtSymbolic);
222: PetscLogEventRegister("MatRARtNum", MAT_CLASSID, &MAT_RARtNumeric);
223: PetscLogEventRegister("MatMatTrnMultSym", MAT_CLASSID, &MAT_MatTransposeMultSymbolic);
224: PetscLogEventRegister("MatMatTrnMultNum", MAT_CLASSID, &MAT_MatTransposeMultNumeric);
225: PetscLogEventRegister("MatTrnMatMultSym", MAT_CLASSID, &MAT_TransposeMatMultSymbolic);
226: PetscLogEventRegister("MatTrnMatMultNum", MAT_CLASSID, &MAT_TransposeMatMultNumeric);
227: PetscLogEventRegister("MatTrnColorCreate", MAT_CLASSID, &MAT_TransposeColoringCreate);
228: PetscLogEventRegister("MatRedundantMat", MAT_CLASSID, &MAT_RedundantMat);
229: PetscLogEventRegister("MatGetSeqNZStrct", MAT_CLASSID, &MAT_GetSequentialNonzeroStructure);
230: PetscLogEventRegister("MatGetMultiProcB", MAT_CLASSID, &MAT_GetMultiProcBlock);
231: PetscLogEventRegister("MatSetRandom", MAT_CLASSID, &MAT_SetRandom);
233: /* these may be specific to MPIAIJ matrices */
234: PetscLogEventRegister("MatMPISumSeqNumeric", MAT_CLASSID, &MAT_Seqstompinum);
235: PetscLogEventRegister("MatMPISumSeqSymbolic", MAT_CLASSID, &MAT_Seqstompisym);
236: PetscLogEventRegister("MatMPISumSeq", MAT_CLASSID, &MAT_Seqstompi);
237: PetscLogEventRegister("MatMPIConcateSeq", MAT_CLASSID, &MAT_Merge);
238: PetscLogEventRegister("MatGetLocalMat", MAT_CLASSID, &MAT_Getlocalmat);
239: PetscLogEventRegister("MatGetLocalMatCondensed", MAT_CLASSID, &MAT_Getlocalmatcondensed);
240: PetscLogEventRegister("MatGetBrowsOfAcols", MAT_CLASSID, &MAT_GetBrowsOfAcols);
241: PetscLogEventRegister("MatGetBrAoCol", MAT_CLASSID, &MAT_GetBrowsOfAocols);
243: PetscLogEventRegister("MatApplyPAPt_Symbolic", MAT_CLASSID, &MAT_Applypapt_symbolic);
244: PetscLogEventRegister("MatApplyPAPt_Numeric", MAT_CLASSID, &MAT_Applypapt_numeric);
245: PetscLogEventRegister("MatApplyPAPt", MAT_CLASSID, &MAT_Applypapt);
247: PetscLogEventRegister("MatGetSymTrans", MAT_CLASSID, &MAT_Getsymtranspose);
248: PetscLogEventRegister("MatGetSymTransR", MAT_CLASSID, &MAT_Getsymtransreduced);
249: PetscLogEventRegister("MatCUSPARSCopyTo", MAT_CLASSID, &MAT_CUSPARSECopyToGPU);
250: PetscLogEventRegister("MatCUSPARSCopyFr", MAT_CLASSID, &MAT_CUSPARSECopyFromGPU);
251: PetscLogEventRegister("MatCUSPARSSolAnl", MAT_CLASSID, &MAT_CUSPARSESolveAnalysis);
252: PetscLogEventRegister("MatCUSPARSGenT", MAT_CLASSID, &MAT_CUSPARSEGenerateTranspose);
253: PetscLogEventRegister("MatHIPSPARSCopyTo", MAT_CLASSID, &MAT_HIPSPARSECopyToGPU);
254: PetscLogEventRegister("MatHIPSPARSCopyFr", MAT_CLASSID, &MAT_HIPSPARSECopyFromGPU);
255: PetscLogEventRegister("MatHIPSPARSSolAnl", MAT_CLASSID, &MAT_HIPSPARSESolveAnalysis);
256: PetscLogEventRegister("MatHIPSPARSGenT", MAT_CLASSID, &MAT_HIPSPARSEGenerateTranspose);
257: PetscLogEventRegister("MatVCLCopyTo", MAT_CLASSID, &MAT_ViennaCLCopyToGPU);
258: PetscLogEventRegister("MatDenseCopyTo", MAT_CLASSID, &MAT_DenseCopyToGPU);
259: PetscLogEventRegister("MatDenseCopyFrom", MAT_CLASSID, &MAT_DenseCopyFromGPU);
260: PetscLogEventRegister("MatSetValBatch", MAT_CLASSID, &MAT_SetValuesBatch);
262: PetscLogEventRegister("MatColoringApply", MAT_COLORING_CLASSID, &MATCOLORING_Apply);
263: PetscLogEventRegister("MatColoringComm", MAT_COLORING_CLASSID, &MATCOLORING_Comm);
264: PetscLogEventRegister("MatColoringLocal", MAT_COLORING_CLASSID, &MATCOLORING_Local);
265: PetscLogEventRegister("MatColoringIS", MAT_COLORING_CLASSID, &MATCOLORING_ISCreate);
266: PetscLogEventRegister("MatColoringSetUp", MAT_COLORING_CLASSID, &MATCOLORING_SetUp);
267: PetscLogEventRegister("MatColoringWeights", MAT_COLORING_CLASSID, &MATCOLORING_Weights);
269: PetscLogEventRegister("MatSetPreallCOO", MAT_CLASSID, &MAT_PreallCOO);
270: PetscLogEventRegister("MatSetValuesCOO", MAT_CLASSID, &MAT_SetVCOO);
272: PetscLogEventRegister("MatH2OpusBuild", MAT_CLASSID, &MAT_H2Opus_Build);
273: PetscLogEventRegister("MatH2OpusComp", MAT_CLASSID, &MAT_H2Opus_Compress);
274: PetscLogEventRegister("MatH2OpusOrth", MAT_CLASSID, &MAT_H2Opus_Orthog);
275: PetscLogEventRegister("MatH2OpusLR", MAT_CLASSID, &MAT_H2Opus_LR);
277: /* Mark non-collective events */
278: PetscLogEventSetCollective(MAT_SetValues, PETSC_FALSE);
279: PetscLogEventSetCollective(MAT_SetValuesBatch, PETSC_FALSE);
280: PetscLogEventSetCollective(MAT_GetRow, PETSC_FALSE);
281: /* Turn off high traffic events by default */
282: PetscLogEventSetActiveAll(MAT_SetValues, PETSC_FALSE);
283: PetscLogEventSetActiveAll(MAT_GetValues, PETSC_FALSE);
284: PetscLogEventSetActiveAll(MAT_GetRow, PETSC_FALSE);
285: /* Process Info */
286: {
287: PetscClassId classids[7];
289: classids[0] = MAT_CLASSID;
290: classids[1] = MAT_FDCOLORING_CLASSID;
291: classids[2] = MAT_COLORING_CLASSID;
292: classids[3] = MAT_TRANSPOSECOLORING_CLASSID;
293: classids[4] = MAT_PARTITIONING_CLASSID;
294: classids[5] = MAT_COARSEN_CLASSID;
295: classids[6] = MAT_NULLSPACE_CLASSID;
296: PetscInfoProcessClass("mat", 7, classids);
297: }
299: /* Process summary exclusions */
300: PetscOptionsGetString(NULL, NULL, "-log_exclude", logList, sizeof(logList), &opt);
301: if (opt) {
302: PetscStrInList("mat", logList, ',', &pkg);
303: if (pkg) PetscLogEventExcludeClass(MAT_CLASSID);
304: if (pkg) PetscLogEventExcludeClass(MAT_FDCOLORING_CLASSID);
305: if (pkg) PetscLogEventExcludeClass(MAT_COLORING_CLASSID);
306: if (pkg) PetscLogEventExcludeClass(MAT_TRANSPOSECOLORING_CLASSID);
307: if (pkg) PetscLogEventExcludeClass(MAT_PARTITIONING_CLASSID);
308: if (pkg) PetscLogEventExcludeClass(MAT_COARSEN_CLASSID);
309: if (pkg) PetscLogEventExcludeClass(MAT_NULLSPACE_CLASSID);
310: }
312: /* Register the PETSc built in factorization based solvers */
313: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJ, MAT_FACTOR_LU, MatGetFactor_seqaij_petsc);
314: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_seqaij_petsc);
315: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJ, MAT_FACTOR_ILU, MatGetFactor_seqaij_petsc);
316: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJ, MAT_FACTOR_ICC, MatGetFactor_seqaij_petsc);
318: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJPERM, MAT_FACTOR_LU, MatGetFactor_seqaij_petsc);
319: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJPERM, MAT_FACTOR_CHOLESKY, MatGetFactor_seqaij_petsc);
320: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJPERM, MAT_FACTOR_ILU, MatGetFactor_seqaij_petsc);
321: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJPERM, MAT_FACTOR_ICC, MatGetFactor_seqaij_petsc);
323: MatSolverTypeRegister(MATSOLVERPETSC, MATCONSTANTDIAGONAL, MAT_FACTOR_LU, MatGetFactor_constantdiagonal_petsc);
324: MatSolverTypeRegister(MATSOLVERPETSC, MATCONSTANTDIAGONAL, MAT_FACTOR_CHOLESKY, MatGetFactor_constantdiagonal_petsc);
325: MatSolverTypeRegister(MATSOLVERPETSC, MATCONSTANTDIAGONAL, MAT_FACTOR_ILU, MatGetFactor_constantdiagonal_petsc);
326: MatSolverTypeRegister(MATSOLVERPETSC, MATCONSTANTDIAGONAL, MAT_FACTOR_ICC, MatGetFactor_constantdiagonal_petsc);
328: #if defined(PETSC_HAVE_MKL_SPARSE)
329: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJMKL, MAT_FACTOR_LU, MatGetFactor_seqaij_petsc);
330: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJMKL, MAT_FACTOR_CHOLESKY, MatGetFactor_seqaij_petsc);
331: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJMKL, MAT_FACTOR_ILU, MatGetFactor_seqaij_petsc);
332: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJMKL, MAT_FACTOR_ICC, MatGetFactor_seqaij_petsc);
334: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQBAIJMKL, MAT_FACTOR_LU, MatGetFactor_seqbaij_petsc);
335: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQBAIJMKL, MAT_FACTOR_CHOLESKY, MatGetFactor_seqbaij_petsc);
336: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQBAIJMKL, MAT_FACTOR_ILU, MatGetFactor_seqbaij_petsc);
337: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQBAIJMKL, MAT_FACTOR_ICC, MatGetFactor_seqbaij_petsc);
338: #endif
339: /* Above, we register the PETSc built-in factorization solvers for MATSEQAIJMKL. In the future, we may want to use
340: * some of the MKL-provided ones instead. */
342: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJCRL, MAT_FACTOR_LU, MatGetFactor_seqaij_petsc);
343: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJCRL, MAT_FACTOR_CHOLESKY, MatGetFactor_seqaij_petsc);
344: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJCRL, MAT_FACTOR_ILU, MatGetFactor_seqaij_petsc);
345: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQAIJCRL, MAT_FACTOR_ICC, MatGetFactor_seqaij_petsc);
347: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQBAIJ, MAT_FACTOR_LU, MatGetFactor_seqbaij_petsc);
348: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_seqbaij_petsc);
349: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQBAIJ, MAT_FACTOR_ILU, MatGetFactor_seqbaij_petsc);
350: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQBAIJ, MAT_FACTOR_ICC, MatGetFactor_seqbaij_petsc);
352: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQSBAIJ, MAT_FACTOR_CHOLESKY, MatGetFactor_seqsbaij_petsc);
353: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQSBAIJ, MAT_FACTOR_ICC, MatGetFactor_seqsbaij_petsc);
355: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQDENSE, MAT_FACTOR_LU, MatGetFactor_seqdense_petsc);
356: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQDENSE, MAT_FACTOR_ILU, MatGetFactor_seqdense_petsc);
357: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQDENSE, MAT_FACTOR_CHOLESKY, MatGetFactor_seqdense_petsc);
358: MatSolverTypeRegister(MATSOLVERPETSC, MATSEQDENSE, MAT_FACTOR_QR, MatGetFactor_seqdense_petsc);
359: #if defined(PETSC_HAVE_CUDA)
360: MatSolverTypeRegister(MATSOLVERCUDA, MATSEQDENSE, MAT_FACTOR_LU, MatGetFactor_seqdense_cuda);
361: MatSolverTypeRegister(MATSOLVERCUDA, MATSEQDENSE, MAT_FACTOR_CHOLESKY, MatGetFactor_seqdense_cuda);
362: MatSolverTypeRegister(MATSOLVERCUDA, MATSEQDENSE, MAT_FACTOR_QR, MatGetFactor_seqdense_cuda);
363: MatSolverTypeRegister(MATSOLVERCUDA, MATSEQDENSECUDA, MAT_FACTOR_LU, MatGetFactor_seqdense_cuda);
364: MatSolverTypeRegister(MATSOLVERCUDA, MATSEQDENSECUDA, MAT_FACTOR_CHOLESKY, MatGetFactor_seqdense_cuda);
365: MatSolverTypeRegister(MATSOLVERCUDA, MATSEQDENSECUDA, MAT_FACTOR_QR, MatGetFactor_seqdense_cuda);
366: #endif
367: #if defined(PETSC_HAVE_HIP)
368: MatSolverTypeRegister(MATSOLVERHIP, MATSEQDENSE, MAT_FACTOR_LU, MatGetFactor_seqdense_hip);
369: MatSolverTypeRegister(MATSOLVERHIP, MATSEQDENSE, MAT_FACTOR_CHOLESKY, MatGetFactor_seqdense_hip);
370: MatSolverTypeRegister(MATSOLVERHIP, MATSEQDENSE, MAT_FACTOR_QR, MatGetFactor_seqdense_hip);
371: MatSolverTypeRegister(MATSOLVERHIP, MATSEQDENSEHIP, MAT_FACTOR_LU, MatGetFactor_seqdense_hip);
372: MatSolverTypeRegister(MATSOLVERHIP, MATSEQDENSEHIP, MAT_FACTOR_CHOLESKY, MatGetFactor_seqdense_hip);
373: MatSolverTypeRegister(MATSOLVERHIP, MATSEQDENSEHIP, MAT_FACTOR_QR, MatGetFactor_seqdense_hip);
374: #endif
376: MatSolverTypeRegister(MATSOLVERBAS, MATSEQAIJ, MAT_FACTOR_ICC, MatGetFactor_seqaij_bas);
378: /*
379: Register the external package factorization based solvers
380: Eventually we don't want to have these hardwired here at compile time of PETSc
381: */
382: #if defined(PETSC_HAVE_MUMPS)
383: MatSolverTypeRegister_MUMPS();
384: #endif
385: #if defined(PETSC_HAVE_CUDA)
386: MatSolverTypeRegister_CUSPARSE();
387: #endif
388: #if defined(PETSC_HAVE_HIP)
389: MatSolverTypeRegister_HIPSPARSE();
390: #endif
391: #if defined(PETSC_HAVE_KOKKOS_KERNELS)
392: MatSolverTypeRegister_KOKKOS();
393: #endif
394: #if defined(PETSC_HAVE_VIENNACL)
395: MatSolverTypeRegister_ViennaCL();
396: #endif
397: #if defined(PETSC_HAVE_ELEMENTAL)
398: MatSolverTypeRegister_Elemental();
399: #endif
400: #if defined(PETSC_HAVE_SCALAPACK)
401: MatSolverTypeRegister_ScaLAPACK();
402: #endif
403: #if defined(PETSC_HAVE_MATLAB)
404: MatSolverTypeRegister_Matlab();
405: #endif
406: #if defined(PETSC_HAVE_ESSL)
407: MatSolverTypeRegister_Essl();
408: #endif
409: #if defined(PETSC_HAVE_SUPERLU)
410: MatSolverTypeRegister_SuperLU();
411: #endif
412: #if defined(PETSC_HAVE_STRUMPACK)
413: MatSolverTypeRegister_STRUMPACK();
414: #endif
415: #if defined(PETSC_HAVE_PASTIX)
416: MatSolverTypeRegister_Pastix();
417: #endif
418: #if defined(PETSC_HAVE_SUPERLU_DIST)
419: MatSolverTypeRegister_SuperLU_DIST();
420: #endif
421: #if defined(PETSC_HAVE_ELEMENTAL)
422: MatSolverTypeRegister_SparseElemental();
423: #endif
424: #if defined(PETSC_HAVE_MKL_PARDISO)
425: MatSolverTypeRegister_MKL_Pardiso();
426: #endif
427: #if defined(PETSC_HAVE_MKL_CPARDISO)
428: MatSolverTypeRegister_MKL_CPardiso();
429: #endif
430: #if defined(PETSC_HAVE_SUITESPARSE)
431: MatSolverTypeRegister_SuiteSparse();
432: #endif
433: #if defined(PETSC_HAVE_LUSOL)
434: MatSolverTypeRegister_Lusol();
435: #endif
436: /* Register package finalizer */
437: PetscRegisterFinalize(MatFinalizePackage);
438: return 0;
439: }
441: #if defined(PETSC_HAVE_DYNAMIC_LIBRARIES)
442: /*
443: PetscDLLibraryRegister - This function is called when the dynamic library it is in is opened.
445: This one registers all the matrix methods that are in the basic PETSc Matrix library.
447: */
448: PETSC_EXTERN PetscErrorCode PetscDLLibraryRegister_petscmat(void)
449: {
450: MatInitializePackage();
451: return 0;
452: }
454: #endif /* PETSC_HAVE_DYNAMIC_LIBRARIES */