xref: /petsc/src/mat/interface/matregis.c (revision fc8a9adeb7fcdc98711d755fa2dc544ddccf0f3e)
1 
2 #include <petsc/private/matimpl.h>  /*I "petscmat.h" I*/
3 
4 PETSC_EXTERN PetscErrorCode MatCreate_MFFD(Mat);
5 PETSC_EXTERN PetscErrorCode MatCreate_MAIJ(Mat);
6 PETSC_EXTERN PetscErrorCode MatCreate_IS(Mat);
7 
8 PETSC_EXTERN PetscErrorCode MatCreate_SeqAIJ(Mat);
9 PETSC_EXTERN PetscErrorCode MatCreate_MPIAIJ(Mat);
10 
11 PETSC_EXTERN PetscErrorCode MatCreate_SeqBAIJ(Mat);
12 PETSC_EXTERN PetscErrorCode MatCreate_MPIBAIJ(Mat);
13 
14 PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat);
15 PETSC_EXTERN PetscErrorCode MatCreate_MPISBAIJ(Mat);
16 
17 PETSC_EXTERN PetscErrorCode MatCreate_SeqDense(Mat);
18 PETSC_EXTERN PetscErrorCode MatCreate_MPIDense(Mat);
19 
20 PETSC_EXTERN PetscErrorCode MatCreate_MPIAdj(Mat);
21 PETSC_EXTERN PetscErrorCode MatCreate_Shell(Mat);
22 PETSC_EXTERN PetscErrorCode MatCreate_Composite(Mat);
23 
24 PETSC_EXTERN PetscErrorCode MatCreate_SeqAIJPERM(Mat);
25 PETSC_EXTERN PetscErrorCode MatCreate_MPIAIJPERM(Mat);
26 
27 PETSC_EXTERN PetscErrorCode MatCreate_SeqAIJSELL(Mat);
28 PETSC_EXTERN PetscErrorCode MatCreate_MPIAIJSELL(Mat);
29 
30 #if defined PETSC_HAVE_MKL_SPARSE
31 PETSC_EXTERN PetscErrorCode MatCreate_SeqAIJMKL(Mat);
32 PETSC_EXTERN PetscErrorCode MatCreate_MPIAIJMKL(Mat);
33 
34 PETSC_EXTERN PetscErrorCode MatCreate_SeqBAIJMKL(Mat);
35 PETSC_EXTERN PetscErrorCode MatCreate_MPIBAIJMKL(Mat);
36 #endif
37 
38 PETSC_EXTERN PetscErrorCode MatCreate_SeqAIJCRL(Mat);
39 PETSC_EXTERN PetscErrorCode MatCreate_MPIAIJCRL(Mat);
40 
41 PETSC_EXTERN PetscErrorCode MatCreate_Scatter(Mat);
42 PETSC_EXTERN PetscErrorCode MatCreate_BlockMat(Mat);
43 PETSC_EXTERN PetscErrorCode MatCreate_Nest(Mat);
44 
45 PETSC_EXTERN PetscErrorCode MatCreate_SeqSELL(Mat);
46 PETSC_EXTERN PetscErrorCode MatCreate_MPISELL(Mat);
47 
48 #if defined PETSC_HAVE_CUDA
49 PETSC_EXTERN PetscErrorCode MatCreate_SeqAIJCUSPARSE(Mat);
50 PETSC_EXTERN PetscErrorCode MatCreate_MPIAIJCUSPARSE(Mat);
51 #endif
52 
53 #if defined PETSC_HAVE_VIENNACL
54 PETSC_EXTERN PetscErrorCode MatCreate_SeqAIJViennaCL(Mat);
55 PETSC_EXTERN PetscErrorCode MatCreate_MPIAIJViennaCL(Mat);
56 #endif
57 
58 #if defined PETSC_HAVE_FFTW
59 PETSC_EXTERN PetscErrorCode MatCreate_FFTW(Mat);
60 #endif
61 PETSC_EXTERN PetscErrorCode MatCreate_Elemental(Mat);
62 
63 PETSC_EXTERN PetscErrorCode MatCreate_Preallocator(Mat);
64 PETSC_EXTERN PetscErrorCode MatCreate_Dummy(Mat);
65 
66 #if defined PETSC_HAVE_HYPRE
67 PETSC_EXTERN PetscErrorCode MatCreate_HYPRE(Mat);
68 #endif
69 
70 PETSC_EXTERN PetscErrorCode MatCreate_ConstantDiagonal(Mat);
71 /*@C
72   MatRegisterAll - Registers all of the matrix types in PETSc
73 
74   Not Collective
75 
76   Level: advanced
77 
78 .keywords: KSP, register, all
79 
80 .seealso:  MatRegister()
81 @*/
82 PetscErrorCode  MatRegisterAll(void)
83 {
84   PetscErrorCode ierr;
85 
86   PetscFunctionBegin;
87   if (MatRegisterAllCalled) PetscFunctionReturn(0);
88   MatRegisterAllCalled = PETSC_TRUE;
89 
90   ierr = MatRegister(MATMFFD,           MatCreate_MFFD);CHKERRQ(ierr);
91 
92   ierr = MatRegister(MATMPIMAIJ,        MatCreate_MAIJ);CHKERRQ(ierr);
93   ierr = MatRegister(MATSEQMAIJ,        MatCreate_MAIJ);CHKERRQ(ierr);
94   ierr = MatRegister(MATMAIJ,           MatCreate_MAIJ);CHKERRQ(ierr);
95 
96   ierr = MatRegister(MATIS,             MatCreate_IS);CHKERRQ(ierr);
97   ierr = MatRegister(MATSHELL,          MatCreate_Shell);CHKERRQ(ierr);
98   ierr = MatRegister(MATCOMPOSITE,      MatCreate_Composite);CHKERRQ(ierr);
99 
100   ierr = MatRegisterRootName(MATAIJ,MATSEQAIJ,MATMPIAIJ);CHKERRQ(ierr);
101   ierr = MatRegister(MATMPIAIJ,         MatCreate_MPIAIJ);CHKERRQ(ierr);
102   ierr = MatRegister(MATSEQAIJ,         MatCreate_SeqAIJ);CHKERRQ(ierr);
103 
104   ierr = MatRegisterRootName(MATAIJPERM,MATSEQAIJPERM,MATMPIAIJPERM);CHKERRQ(ierr);
105   ierr = MatRegister(MATMPIAIJPERM,     MatCreate_MPIAIJPERM);CHKERRQ(ierr);
106   ierr = MatRegister(MATSEQAIJPERM,     MatCreate_SeqAIJPERM);CHKERRQ(ierr);
107 
108   ierr = MatRegisterRootName(MATAIJSELL,MATSEQAIJSELL,MATMPIAIJSELL);CHKERRQ(ierr);
109   ierr = MatRegister(MATMPIAIJSELL,     MatCreate_MPIAIJSELL);CHKERRQ(ierr);
110   ierr = MatRegister(MATSEQAIJSELL,     MatCreate_SeqAIJSELL);CHKERRQ(ierr);
111 
112 #if defined PETSC_HAVE_MKL_SPARSE
113   ierr = MatRegisterRootName(MATAIJMKL, MATSEQAIJMKL,MATMPIAIJMKL);CHKERRQ(ierr);
114   ierr = MatRegister(MATMPIAIJMKL,      MatCreate_MPIAIJMKL);CHKERRQ(ierr);
115   ierr = MatRegister(MATSEQAIJMKL,      MatCreate_SeqAIJMKL);CHKERRQ(ierr);
116 
117   ierr = MatRegisterRootName(MATBAIJMKL,MATSEQBAIJMKL,MATMPIBAIJMKL);CHKERRQ(ierr);
118   ierr = MatRegister(MATMPIBAIJMKL,      MatCreate_MPIBAIJMKL);CHKERRQ(ierr);
119   ierr = MatRegister(MATSEQBAIJMKL,      MatCreate_SeqBAIJMKL);CHKERRQ(ierr);
120 #endif
121 
122   ierr = MatRegisterRootName(MATAIJCRL,MATSEQAIJCRL,MATMPIAIJCRL);CHKERRQ(ierr);
123   ierr = MatRegister(MATSEQAIJCRL,      MatCreate_SeqAIJCRL);CHKERRQ(ierr);
124   ierr = MatRegister(MATMPIAIJCRL,      MatCreate_MPIAIJCRL);CHKERRQ(ierr);
125 
126   ierr = MatRegisterRootName(MATBAIJ,MATSEQBAIJ,MATMPIBAIJ);CHKERRQ(ierr);
127   ierr = MatRegister(MATMPIBAIJ,        MatCreate_MPIBAIJ);CHKERRQ(ierr);
128   ierr = MatRegister(MATSEQBAIJ,        MatCreate_SeqBAIJ);CHKERRQ(ierr);
129 
130   ierr = MatRegisterRootName(MATSBAIJ,MATSEQSBAIJ,MATMPISBAIJ);CHKERRQ(ierr);
131   ierr = MatRegister(MATMPISBAIJ,       MatCreate_MPISBAIJ);CHKERRQ(ierr);
132   ierr = MatRegister(MATSEQSBAIJ,       MatCreate_SeqSBAIJ);CHKERRQ(ierr);
133 
134   ierr = MatRegisterRootName(MATDENSE,MATSEQDENSE,MATMPIDENSE);CHKERRQ(ierr);
135   ierr = MatRegister(MATMPIDENSE,       MatCreate_MPIDense);CHKERRQ(ierr);
136   ierr = MatRegister(MATSEQDENSE,       MatCreate_SeqDense);CHKERRQ(ierr);
137 
138   ierr = MatRegister(MATMPIADJ,         MatCreate_MPIAdj);CHKERRQ(ierr);
139   ierr = MatRegister(MATSCATTER,        MatCreate_Scatter);CHKERRQ(ierr);
140   ierr = MatRegister(MATBLOCKMAT,       MatCreate_BlockMat);CHKERRQ(ierr);
141   ierr = MatRegister(MATNEST,           MatCreate_Nest);CHKERRQ(ierr);
142 
143   ierr = MatRegisterRootName(MATSELL,MATSEQSELL,MATMPISELL);CHKERRQ(ierr);
144   ierr = MatRegister(MATMPISELL,         MatCreate_MPISELL);CHKERRQ(ierr);
145   ierr = MatRegister(MATSEQSELL,         MatCreate_SeqSELL);CHKERRQ(ierr);
146 
147 #if defined PETSC_HAVE_CUDA
148   ierr = MatRegisterRootName(MATAIJCUSPARSE,MATSEQAIJCUSPARSE,MATMPIAIJCUSPARSE);CHKERRQ(ierr);
149   ierr = MatRegister(MATSEQAIJCUSPARSE, MatCreate_SeqAIJCUSPARSE);CHKERRQ(ierr);
150   ierr = MatRegister(MATMPIAIJCUSPARSE, MatCreate_MPIAIJCUSPARSE);CHKERRQ(ierr);
151 #endif
152 
153 #if defined PETSC_HAVE_VIENNACL
154   ierr = MatRegisterRootName(MATAIJVIENNACL,MATSEQAIJVIENNACL,MATMPIAIJVIENNACL);CHKERRQ(ierr);
155   ierr = MatRegister(MATSEQAIJVIENNACL, MatCreate_SeqAIJViennaCL);CHKERRQ(ierr);
156   ierr = MatRegister(MATMPIAIJVIENNACL, MatCreate_MPIAIJViennaCL);CHKERRQ(ierr);
157 #endif
158 
159 #if defined PETSC_HAVE_FFTW
160   ierr = MatRegister(MATFFTW,           MatCreate_FFTW);CHKERRQ(ierr);
161 #endif
162 #if defined PETSC_HAVE_ELEMENTAL
163   ierr = MatRegister(MATELEMENTAL,      MatCreate_Elemental);CHKERRQ(ierr);
164 #endif
165 
166   ierr = MatRegister(MATPREALLOCATOR,   MatCreate_Preallocator);CHKERRQ(ierr);
167   ierr = MatRegister(MATDUMMY,          MatCreate_Dummy);CHKERRQ(ierr);
168 
169   ierr = MatRegister(MATCONSTANTDIAGONAL,MatCreate_ConstantDiagonal);CHKERRQ(ierr);
170 
171 #if defined PETSC_HAVE_HYPRE
172   ierr = MatRegister(MATHYPRE,          MatCreate_HYPRE);CHKERRQ(ierr);
173 #endif
174   PetscFunctionReturn(0);
175 }
176