xref: /petsc/src/mat/impls/sbaij/seq/sbaij.h (revision 9495ac641afb6c63677b6f6914aed8e1b2a2d6d9) 
149b5e25fSSatish Balay 
249b5e25fSSatish Balay #if !defined(__SBAIJ_H)
349b5e25fSSatish Balay #define __SBAIJ_H
4b9147fbbSdalcinl #include "include/private/matimpl.h"
56c6c5352SBarry Smith #include "src/mat/impls/baij/seq/baij.h"
649b5e25fSSatish Balay 
749b5e25fSSatish Balay /*
849b5e25fSSatish Balay   MATSEQSBAIJ format - Block compressed row storage. The i[] and j[]
949b5e25fSSatish Balay   arrays start at 0.
1049b5e25fSSatish Balay */
1149b5e25fSSatish Balay 
1249b5e25fSSatish Balay typedef struct {
13421e10b8SBarry Smith   SEQAIJHEADER(PetscScalar);
14e6b907acSBarry Smith   SEQBAIJHEADER;
1513f74950SBarry Smith   PetscInt         *inew;        /* pointer to beginning of each row of reordered matrix */
1613f74950SBarry Smith   PetscInt         *jnew;        /* column values: jnew + i[k] is start of row k */
17a6175056SHong Zhang   MatScalar        *anew;        /* nonzero diagonal and superdiagonal elements of reordered matrix */
18d59c15a7SBarry Smith   PetscScalar      *solves_work; /* work space used in MatSolves */
1913f74950SBarry Smith   PetscInt         solves_work_n;/* size of solves_work */
20e2ee2a47SBarry Smith   PetscScalar      *relax_work;
2113f74950SBarry Smith   PetscInt         *a2anew;        /* map used for symm permutation */
22d36791b2SHong Zhang   PetscTruth       permute;        /* if true, a non-trivial permutation is used for factorization */
23941593c8SHong Zhang   PetscTruth       ignore_ltriangular; /* if true, ignore the lower triangular values inserted by users */
24f5edf698SHong Zhang   PetscTruth       getrow_utriangular; /* if true, MatGetRow_SeqSBAIJ() is enabled to get the upper part of the row */
2549b5e25fSSatish Balay } Mat_SeqSBAIJ;
2649b5e25fSSatish Balay 
27f349cc81SSatish Balay EXTERN_C_BEGIN
28ab93d7beSBarry Smith EXTERN PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat,PetscInt,PetscInt,PetscInt*);
29f349cc81SSatish Balay EXTERN_C_END
30*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,IS,MatFactorInfo*,Mat*);
31*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,MatFactorInfo*);
32dfbe8321SBarry Smith EXTERN PetscErrorCode MatICCFactorSymbolic_SeqSBAIJ(Mat,IS,MatFactorInfo*,Mat *);
33dfbe8321SBarry Smith EXTERN PetscErrorCode MatDuplicate_SeqSBAIJ(Mat,MatDuplicateOption,Mat*);
34dfbe8321SBarry Smith EXTERN PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat);
35*9495ac64SHong Zhang EXTERN PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt);
36*9495ac64SHong Zhang EXTERN PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,PetscInt,MatReuse,Mat*);
37*9495ac64SHong Zhang EXTERN PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]);
38*9495ac64SHong Zhang EXTERN PetscErrorCode MatScale_SeqSBAIJ(Mat,PetscScalar);
39*9495ac64SHong Zhang EXTERN PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *);
40*9495ac64SHong Zhang EXTERN PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscTruth*);
41*9495ac64SHong Zhang EXTERN PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec);
42*9495ac64SHong Zhang EXTERN PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec);
43*9495ac64SHong Zhang EXTERN PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *);
44*9495ac64SHong Zhang EXTERN PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat);
45*9495ac64SHong Zhang EXTERN PetscErrorCode MatGetRowMax_SeqSBAIJ(Mat,Vec);
46*9495ac64SHong Zhang EXTERN PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*);
4749b5e25fSSatish Balay 
48af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering(Mat,MatFactorInfo*,Mat*);
49dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
504078b935SHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
514078b935SHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
52*9495ac64SHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_1(Mat,Vec,Vec);
53*9495ac64SHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_1(Mat,Vec,Vec);
54266135f8SHong Zhang 
55af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering(Mat,MatFactorInfo*,Mat*);
56dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
57759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
58759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
59266135f8SHong Zhang 
60af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering(Mat,MatFactorInfo*,Mat*);
61dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
62759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
63759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
64266135f8SHong Zhang 
65af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering(Mat,MatFactorInfo*,Mat*);
66dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
67759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
68759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
69266135f8SHong Zhang 
70af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering(Mat,MatFactorInfo*,Mat*);
71dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
72759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
73759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
74266135f8SHong Zhang 
75af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering(Mat,MatFactorInfo*,Mat*);
76dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
77759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
78759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
79266135f8SHong Zhang 
80af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering(Mat,MatFactorInfo*,Mat*);
81dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
82759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
83759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
8449b5e25fSSatish Balay 
85af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering(Mat,MatFactorInfo*,Mat*);
86dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
87759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
88759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
89*9495ac64SHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_N(Mat,Vec,Vec);
90*9495ac64SHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_N(Mat,Vec,Vec);
91*9495ac64SHong Zhang 
92*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,MatFactorInfo*,Mat*);
93*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1(Mat,MatFactorInfo*,Mat*);
94*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,MatFactorInfo*,Mat*);
95*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,MatFactorInfo*,Mat*);
96*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,MatFactorInfo*,Mat*);
97*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,MatFactorInfo*,Mat*);
98*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,MatFactorInfo*,Mat*);
99*9495ac64SHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,MatFactorInfo*,Mat*);
100*9495ac64SHong Zhang 
101*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N(Mat,Vec,Vec);
102*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec);
103*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2(Mat,Vec,Vec);
104*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3(Mat,Vec,Vec);
105*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4(Mat,Vec,Vec);
106*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5(Mat,Vec,Vec);
107*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6(Mat,Vec,Vec);
108*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7(Mat,Vec,Vec);
109*9495ac64SHong Zhang 
110*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs);
111*9495ac64SHong Zhang 
112*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
113*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
114*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
115*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
116*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
117*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
118*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
119*9495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
120*9495ac64SHong Zhang 
121*9495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec);
122*9495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec);
123*9495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec);
124*9495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec);
125*9495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec);
126*9495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec);
127*9495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec);
128*9495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec);
129*9495ac64SHong Zhang 
130*9495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec);
131*9495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec);
132*9495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec);
133*9495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec);
134*9495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec);
135*9495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec);
136*9495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec);
137*9495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_N(Mat,Vec,Vec,Vec);
138671cb588SHong Zhang 
13913f74950SBarry Smith EXTERN PetscErrorCode MatRelax_SeqSBAIJ(Mat,Vec,PetscReal,MatSORType,PetscReal,PetscInt,PetscInt,Vec);
140f69a0ea3SMatthew Knepley EXTERN PetscErrorCode MatLoad_SeqSBAIJ(PetscViewer, MatType,Mat*);
141d06b337dSHong Zhang 
14249b5e25fSSatish Balay #endif
143