149b5e25fSSatish Balay 249b5e25fSSatish Balay #if !defined(__SBAIJ_H) 349b5e25fSSatish Balay #define __SBAIJ_H 4*7c4f633dSBarry Smith #include "private/matimpl.h" 5*7c4f633dSBarry 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 { 13dd6ea824SBarry Smith SEQAIJHEADER(MatScalar); 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 300481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,Mat,IS,const MatFactorInfo*); 310481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,const MatFactorInfo*); 320481f469SBarry Smith EXTERN PetscErrorCode MatICCFactorSymbolic_SeqSBAIJ(Mat,Mat,IS,const MatFactorInfo*); 33dfbe8321SBarry Smith EXTERN PetscErrorCode MatDuplicate_SeqSBAIJ(Mat,MatDuplicateOption,Mat*); 34dfbe8321SBarry Smith EXTERN PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat); 359495ac64SHong Zhang EXTERN PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt); 369495ac64SHong Zhang EXTERN PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,PetscInt,MatReuse,Mat*); 379495ac64SHong Zhang EXTERN PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]); 389495ac64SHong Zhang EXTERN PetscErrorCode MatScale_SeqSBAIJ(Mat,PetscScalar); 399495ac64SHong Zhang EXTERN PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *); 409495ac64SHong Zhang EXTERN PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscTruth*); 419495ac64SHong Zhang EXTERN PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec); 429495ac64SHong Zhang EXTERN PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec); 439495ac64SHong Zhang EXTERN PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *); 449495ac64SHong Zhang EXTERN PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat); 45985db425SBarry Smith EXTERN PetscErrorCode MatGetRowMaxAbs_SeqSBAIJ(Mat,Vec,PetscInt[]); 469495ac64SHong Zhang EXTERN PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*); 4749b5e25fSSatish Balay 480481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 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); 529495ac64SHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_1(Mat,Vec,Vec); 539495ac64SHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_1(Mat,Vec,Vec); 54266135f8SHong Zhang 550481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 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 600481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 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 650481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 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 700481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 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 750481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 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 800481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 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 850481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 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); 899495ac64SHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_N(Mat,Vec,Vec); 909495ac64SHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_N(Mat,Vec,Vec); 919495ac64SHong Zhang 920481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,Mat,const MatFactorInfo*); 930481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1(Mat,Mat,const MatFactorInfo*); 940481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,Mat,const MatFactorInfo*); 950481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,Mat,const MatFactorInfo*); 960481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,Mat,const MatFactorInfo*); 970481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,Mat,const MatFactorInfo*); 980481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,Mat,const MatFactorInfo*); 990481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,Mat,const MatFactorInfo*); 1009495ac64SHong Zhang 1019495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N(Mat,Vec,Vec); 1029495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec); 1039495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2(Mat,Vec,Vec); 1049495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3(Mat,Vec,Vec); 1059495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4(Mat,Vec,Vec); 1069495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5(Mat,Vec,Vec); 1079495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6(Mat,Vec,Vec); 1089495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7(Mat,Vec,Vec); 1099495ac64SHong Zhang 1109495ac64SHong Zhang EXTERN PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs); 1119495ac64SHong Zhang 1129495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 1139495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec); 1149495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec); 1159495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec); 1169495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec); 1179495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec); 1189495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec); 1199495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec); 1209495ac64SHong Zhang 1219495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec); 1229495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec); 1239495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec); 1249495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec); 1259495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec); 1269495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec); 1279495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec); 1289495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec); 1299495ac64SHong Zhang 1309495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec); 1319495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec); 1329495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec); 1339495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec); 1349495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec); 1359495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec); 1369495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec); 1379495ac64SHong 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); 140a313700dSBarry Smith EXTERN PetscErrorCode MatLoad_SeqSBAIJ(PetscViewer, const MatType,Mat*); 141d06b337dSHong Zhang 14249b5e25fSSatish Balay #endif 143