149b5e25fSSatish Balay 249b5e25fSSatish Balay #if !defined(__SBAIJ_H) 349b5e25fSSatish Balay #define __SBAIJ_H 4*b45d2f2cSJed Brown #include <petsc-private/matimpl.h> 5c6db04a5SJed Brown #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 */ 2013f74950SBarry Smith PetscInt *a2anew; /* map used for symm permutation */ 21ace3abfcSBarry Smith PetscBool permute; /* if true, a non-trivial permutation is used for factorization */ 22ace3abfcSBarry Smith PetscBool ignore_ltriangular; /* if true, ignore the lower triangular values inserted by users */ 23ace3abfcSBarry Smith PetscBool getrow_utriangular; /* if true, MatGetRow_SeqSBAIJ() is enabled to get the upper part of the row */ 244108e4d5SBarry Smith Mat_SeqAIJ_Inode inode; 2538702af4SBarry Smith unsigned short *jshort; 26ace3abfcSBarry Smith PetscBool free_jshort; 2749b5e25fSSatish Balay } Mat_SeqSBAIJ; 2849b5e25fSSatish Balay 29f349cc81SSatish Balay EXTERN_C_BEGIN 3009573ac7SBarry Smith extern PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat,PetscInt,PetscInt,PetscInt*); 31f349cc81SSatish Balay EXTERN_C_END 3209573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,Mat,IS,const MatFactorInfo*); 3309573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ_inplace(Mat,Mat,IS,const MatFactorInfo*); 3409573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,const MatFactorInfo*); 3509573ac7SBarry Smith extern PetscErrorCode MatICCFactorSymbolic_SeqSBAIJ(Mat,Mat,IS,const MatFactorInfo*); 3609573ac7SBarry Smith extern PetscErrorCode MatICCFactorSymbolic_SeqSBAIJ_inplace(Mat,Mat,IS,const MatFactorInfo*); 3709573ac7SBarry Smith extern PetscErrorCode MatDuplicate_SeqSBAIJ(Mat,MatDuplicateOption,Mat*); 3809573ac7SBarry Smith extern PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat); 3909573ac7SBarry Smith extern PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt); 4009573ac7SBarry Smith extern PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,MatReuse,Mat*); 4109573ac7SBarry Smith extern PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]); 4209573ac7SBarry Smith extern PetscErrorCode MatScale_SeqSBAIJ(Mat,PetscScalar); 4309573ac7SBarry Smith extern PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *); 4409573ac7SBarry Smith extern PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscBool *); 4509573ac7SBarry Smith extern PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec); 4609573ac7SBarry Smith extern PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec); 4709573ac7SBarry Smith extern PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *); 4809573ac7SBarry Smith extern PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat); 4909573ac7SBarry Smith extern PetscErrorCode MatGetRowMaxAbs_SeqSBAIJ(Mat,Vec,PetscInt[]); 5009573ac7SBarry Smith extern PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*); 5109573ac7SBarry Smith extern PetscErrorCode MatDestroy_SeqSBAIJ(Mat); 5209573ac7SBarry Smith extern PetscErrorCode MatView_SeqSBAIJ(Mat,PetscViewer); 5349b5e25fSSatish Balay 5409573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 5509573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Mat,const MatFactorInfo*); 5609573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Vec,Vec); 5709573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 580a3c351aSHong Zhang 5909573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Vec,Vec); 6009573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Vec,Vec); 6109573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_1_inplace(Mat,Vec,Vec); 6209573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_1_inplace(Mat,Vec,Vec); 630a3c351aSHong Zhang 6409573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 6509573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 6609573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_1(Mat,Vec,Vec); 6709573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_1(Mat,Vec,Vec); 68266135f8SHong Zhang 6909573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 7009573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering_inplace(Mat,Vec,Vec); 7109573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_2_NaturalOrdering_inplace(Mat,Vec,Vec); 7209573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering_inplace(Mat,Vec,Vec); 73266135f8SHong Zhang 7409573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 7509573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering_inplace(Mat,Vec,Vec); 7609573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_3_NaturalOrdering_inplace(Mat,Vec,Vec); 7709573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering_inplace(Mat,Vec,Vec); 78266135f8SHong Zhang 7909573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 8009573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering_inplace(Mat,Vec,Vec); 8109573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_4_NaturalOrdering_inplace(Mat,Vec,Vec); 8209573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering_inplace(Mat,Vec,Vec); 83266135f8SHong Zhang 8409573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 8509573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering_inplace(Mat,Vec,Vec); 8609573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_5_NaturalOrdering_inplace(Mat,Vec,Vec); 8709573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering_inplace(Mat,Vec,Vec); 88266135f8SHong Zhang 8909573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 9009573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering_inplace(Mat,Vec,Vec); 9109573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_6_NaturalOrdering_inplace(Mat,Vec,Vec); 9209573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering_inplace(Mat,Vec,Vec); 93266135f8SHong Zhang 9409573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 9509573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering_inplace(Mat,Vec,Vec); 9609573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_7_NaturalOrdering_inplace(Mat,Vec,Vec); 9709573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering_inplace(Mat,Vec,Vec); 9849b5e25fSSatish Balay 9909573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 10009573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering_inplace(Mat,Vec,Vec); 10109573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_N_NaturalOrdering_inplace(Mat,Vec,Vec); 10209573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering_inplace(Mat,Vec,Vec); 10309573ac7SBarry Smith extern PetscErrorCode MatForwardSolve_SeqSBAIJ_N_inplace(Mat,Vec,Vec); 10409573ac7SBarry Smith extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_N_inplace(Mat,Vec,Vec); 1059495ac64SHong Zhang 10609573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,Mat,const MatFactorInfo*); 10709573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace(Mat,Mat,const MatFactorInfo*); 10809573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,Mat,const MatFactorInfo*); 10909573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,Mat,const MatFactorInfo*); 11009573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,Mat,const MatFactorInfo*); 11109573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,Mat,const MatFactorInfo*); 11209573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,Mat,const MatFactorInfo*); 11309573ac7SBarry Smith extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,Mat,const MatFactorInfo*); 1149495ac64SHong Zhang 11509573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_N_inplace(Mat,Vec,Vec); 11609573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_1_inplace(Mat,Vec,Vec); 11709573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec); 11809573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_2_inplace(Mat,Vec,Vec); 11909573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_3_inplace(Mat,Vec,Vec); 12009573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_4_inplace(Mat,Vec,Vec); 12109573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_5_inplace(Mat,Vec,Vec); 12209573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_6_inplace(Mat,Vec,Vec); 12309573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_7_inplace(Mat,Vec,Vec); 1249495ac64SHong Zhang 12509573ac7SBarry Smith extern PetscErrorCode MatSolves_SeqSBAIJ_1_inplace(Mat,Vecs,Vecs); 12609573ac7SBarry Smith extern PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs); 1279495ac64SHong Zhang 12809573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Vec,Vec); 12909573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 13009573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering_inplace(Mat,Vec,Vec); 13109573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering_inplace(Mat,Vec,Vec); 13209573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering_inplace(Mat,Vec,Vec); 13309573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering_inplace(Mat,Vec,Vec); 13409573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering_inplace(Mat,Vec,Vec); 13509573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering_inplace(Mat,Vec,Vec); 13609573ac7SBarry Smith extern PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering_inplace(Mat,Vec,Vec); 1379495ac64SHong Zhang 13809573ac7SBarry Smith extern PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec); 13909573ac7SBarry Smith extern PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec); 14009573ac7SBarry Smith extern PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec); 14109573ac7SBarry Smith extern PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec); 14209573ac7SBarry Smith extern PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec); 14309573ac7SBarry Smith extern PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec); 14409573ac7SBarry Smith extern PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec); 14509573ac7SBarry Smith extern PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec); 1469495ac64SHong Zhang 14709573ac7SBarry Smith extern PetscErrorCode MatMult_SeqSBAIJ_1_Hermitian(Mat,Vec,Vec); 148eeffb40dSHong Zhang 14909573ac7SBarry Smith extern PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec); 15009573ac7SBarry Smith extern PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec); 15109573ac7SBarry Smith extern PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec); 15209573ac7SBarry Smith extern PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec); 15309573ac7SBarry Smith extern PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec); 15409573ac7SBarry Smith extern PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec); 15509573ac7SBarry Smith extern PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec); 15609573ac7SBarry Smith extern PetscErrorCode MatMultAdd_SeqSBAIJ_N(Mat,Vec,Vec,Vec); 157671cb588SHong Zhang 15809573ac7SBarry Smith extern PetscErrorCode MatSOR_SeqSBAIJ(Mat,Vec,PetscReal,MatSORType,PetscReal,PetscInt,PetscInt,Vec); 15909573ac7SBarry Smith extern PetscErrorCode MatLoad_SeqSBAIJ(Mat,PetscViewer); 16009573ac7SBarry Smith extern PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat,PetscBool ); 161d595f711SHong Zhang 16249b5e25fSSatish Balay #endif 163