1 2 #if !defined(__BAIJ_H) 3 #define __BAIJ_H 4 #include "src/mat/matimpl.h" 5 6 7 /* 8 MATSEQBAIJ format - Block compressed row storage. The i[] and j[] 9 arrays start at 0. 10 */ 11 12 /* This header is shared by the SeqSBAIJ matrix */ 13 #define SEQBAIJHEADER \ 14 PetscTruth sorted; /* if true, rows are sorted by increasing columns */ \ 15 PetscTruth roworiented; /* if true, row-oriented input, default */ \ 16 PetscInt nonew; /* 1 don't add new nonzeros, -1 generate error on new */ \ 17 PetscTruth singlemalloc; /* if true a, i, and j have been obtained with \ 18 one big malloc */ \ 19 PetscInt bs2; /* square of block size */ \ 20 PetscInt mbs,nbs; /* rows/bs, columns/bs */ \ 21 PetscInt nz,maxnz; /* nonzeros, allocated nonzeros */ \ 22 PetscInt *diag; /* pointers to diagonal elements */ \ 23 PetscInt *i; /* pointer to beginning of each row */ \ 24 PetscInt *imax; /* maximum space allocated for each row */ \ 25 PetscInt *ilen; /* actual length of each row */ \ 26 PetscInt *j; /* column values: j + i[k] - 1 is start of row k */ \ 27 MatScalar *a; /* nonzero elements */ \ 28 IS row,col,icol; /* index sets, used for reorderings */ \ 29 PetscScalar *solve_work; /* work space used in MatSolve */ \ 30 PetscInt reallocs; /* number of mallocs done during MatSetValues() \ 31 as more values are set then were preallocated */ \ 32 PetscScalar *mult_work; /* work array for matrix vector product*/ \ 33 PetscScalar *saved_values; \ 34 \ 35 PetscTruth keepzeroedrows; /* if true, MatZeroRows() will not change nonzero structure */ \ 36 Mat sbaijMat; /* mat in sbaij format */ \ 37 \ 38 PetscInt setvalueslen; /* only used for single precision */ \ 39 MatScalar *setvaluescopy; /* area double precision values in MatSetValuesXXX() are copied \ 40 before calling MatSetValuesXXX_SeqBAIJ_MatScalar() */ \ 41 \ 42 PetscTruth pivotinblocks; /* pivot inside factorization of each diagonal block */ \ 43 \ 44 PetscInt *xtoy,*xtoyB; /* map nonzero pattern of X into Y's, used by MatAXPY() */ \ 45 Mat XtoY; /* used by MatAXPY() */ \ 46 PetscScalar *idiag; /* inverse of block diagonal */ \ 47 PetscTruth idiagvalid; /* if above has correct/current values */ \ 48 Mat_CompressedRow compressedrow; /* use compressed row format */ 49 50 typedef struct { 51 SEQBAIJHEADER 52 } Mat_SeqBAIJ; 53 54 EXTERN_C_BEGIN 55 EXTERN PetscErrorCode MatSeqBAIJSetPreallocation_SeqBAIJ(Mat,PetscInt,PetscInt,PetscInt*); 56 EXTERN_C_END 57 EXTERN PetscErrorCode MatILUFactorSymbolic_SeqBAIJ(Mat,IS,IS,MatFactorInfo*,Mat *); 58 EXTERN PetscErrorCode MatICCFactorSymbolic_SeqBAIJ(Mat,IS,MatFactorInfo*,Mat *); 59 EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqBAIJ(Mat,IS,MatFactorInfo*,Mat*); 60 EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqBAIJ_N(Mat,MatFactorInfo*,Mat*); 61 EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqBAIJ_N_NaturalOrdering(Mat,MatFactorInfo*,Mat*); 62 EXTERN PetscErrorCode MatDuplicate_SeqBAIJ(Mat,MatDuplicateOption,Mat*); 63 EXTERN PetscErrorCode MatMissingDiagonal_SeqBAIJ(Mat); 64 EXTERN PetscErrorCode MatMarkDiagonal_SeqBAIJ(Mat); 65 66 EXTERN PetscErrorCode MatLUFactorSymbolic_SeqBAIJ(Mat,IS,IS,MatFactorInfo*,Mat*); 67 EXTERN PetscErrorCode MatLUFactor_SeqBAIJ(Mat,IS,IS,MatFactorInfo*); 68 EXTERN PetscErrorCode MatIncreaseOverlap_SeqBAIJ(Mat,PetscInt,IS*,PetscInt); 69 EXTERN PetscErrorCode MatGetSubMatrix_SeqBAIJ(Mat,IS,IS,PetscInt,MatReuse,Mat*); 70 EXTERN PetscErrorCode MatGetSubMatrices_SeqBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]); 71 EXTERN PetscErrorCode MatMultTranspose_SeqBAIJ(Mat,Vec,Vec); 72 EXTERN PetscErrorCode MatMultTransposeAdd_SeqBAIJ(Mat,Vec,Vec,Vec); 73 EXTERN PetscErrorCode MatScale_SeqBAIJ(Mat,PetscScalar); 74 EXTERN PetscErrorCode MatNorm_SeqBAIJ(Mat,NormType,PetscReal *); 75 EXTERN PetscErrorCode MatEqual_SeqBAIJ(Mat,Mat,PetscTruth*); 76 EXTERN PetscErrorCode MatGetDiagonal_SeqBAIJ(Mat,Vec); 77 EXTERN PetscErrorCode MatDiagonalScale_SeqBAIJ(Mat,Vec,Vec); 78 EXTERN PetscErrorCode MatGetInfo_SeqBAIJ(Mat,MatInfoType,MatInfo *); 79 EXTERN PetscErrorCode MatZeroEntries_SeqBAIJ(Mat); 80 81 EXTERN PetscErrorCode MatSeqBAIJ_UpdateFactorNumeric_NaturalOrdering(Mat); 82 EXTERN PetscErrorCode MatSeqBAIJ_UpdateSolvers(Mat); 83 84 EXTERN PetscErrorCode MatSolve_SeqBAIJ_Update(Mat,Vec,Vec); 85 EXTERN PetscErrorCode MatSolve_SeqBAIJ_1(Mat,Vec,Vec); 86 EXTERN PetscErrorCode MatSolve_SeqBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 87 EXTERN PetscErrorCode MatSolve_SeqBAIJ_2(Mat,Vec,Vec); 88 EXTERN PetscErrorCode MatSolve_SeqBAIJ_2_NaturalOrdering(Mat,Vec,Vec); 89 EXTERN PetscErrorCode MatSolve_SeqBAIJ_3(Mat,Vec,Vec); 90 EXTERN PetscErrorCode MatSolve_SeqBAIJ_3_NaturalOrdering(Mat,Vec,Vec); 91 EXTERN PetscErrorCode MatSolve_SeqBAIJ_4(Mat,Vec,Vec); 92 EXTERN PetscErrorCode MatSolve_SeqBAIJ_4_NaturalOrdering(Mat,Vec,Vec); 93 #if defined(PETSC_HAVE_SSE) 94 EXTERN PetscErrorCode MatSolve_SeqBAIJ_4_SSE_Demotion(Mat,Vec,Vec); 95 EXTERN PetscErrorCode MatSolve_SeqBAIJ_4_NaturalOrdering_SSE_Demotion(Mat,Vec,Vec); 96 EXTERN PetscErrorCode MatSolve_SeqBAIJ_4_NaturalOrdering_SSE_Demotion_usj(Mat,Vec,Vec); 97 #endif 98 EXTERN PetscErrorCode MatSolve_SeqBAIJ_5(Mat,Vec,Vec); 99 EXTERN PetscErrorCode MatSolve_SeqBAIJ_5_NaturalOrdering(Mat,Vec,Vec); 100 EXTERN PetscErrorCode MatSolve_SeqBAIJ_6(Mat,Vec,Vec); 101 EXTERN PetscErrorCode MatSolve_SeqBAIJ_6_NaturalOrdering(Mat,Vec,Vec); 102 EXTERN PetscErrorCode MatSolve_SeqBAIJ_7(Mat,Vec,Vec); 103 EXTERN PetscErrorCode MatSolve_SeqBAIJ_7_NaturalOrdering(Mat,Vec,Vec); 104 EXTERN PetscErrorCode MatSolve_SeqBAIJ_N(Mat,Vec,Vec); 105 106 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_Update(Mat,Vec,Vec); 107 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_1(Mat,Vec,Vec); 108 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 109 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_2(Mat,Vec,Vec); 110 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_2_NaturalOrdering(Mat,Vec,Vec); 111 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_3(Mat,Vec,Vec); 112 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_3_NaturalOrdering(Mat,Vec,Vec); 113 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_4(Mat,Vec,Vec); 114 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_4_NaturalOrdering(Mat,Vec,Vec); 115 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_5(Mat,Vec,Vec); 116 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_5_NaturalOrdering(Mat,Vec,Vec); 117 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_6(Mat,Vec,Vec); 118 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_6_NaturalOrdering(Mat,Vec,Vec); 119 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_7(Mat,Vec,Vec); 120 EXTERN PetscErrorCode MatSolveTranspose_SeqBAIJ_7_NaturalOrdering(Mat,Vec,Vec); 121 122 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_1(Mat,MatFactorInfo*,Mat*); 123 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_2(Mat,MatFactorInfo*,Mat*); 124 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering(Mat,MatFactorInfo*,Mat*); 125 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_3(Mat,MatFactorInfo*,Mat*); 126 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_3_NaturalOrdering(Mat,MatFactorInfo*,Mat*); 127 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_4(Mat,MatFactorInfo*,Mat*); 128 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_4_NaturalOrdering(Mat,MatFactorInfo*,Mat*); 129 #if defined(PETSC_HAVE_SSE) 130 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_4_NaturalOrdering_SSE(Mat,MatFactorInfo*,Mat*); 131 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_4_NaturalOrdering_SSE_usj(Mat,MatFactorInfo*,Mat*); 132 #else 133 #endif 134 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_5(Mat,MatFactorInfo*,Mat*); 135 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_5_NaturalOrdering(Mat,MatFactorInfo*,Mat*); 136 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_6(Mat,MatFactorInfo*,Mat*); 137 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_6_NaturalOrdering(Mat,MatFactorInfo*,Mat*); 138 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_7(Mat,MatFactorInfo*,Mat*); 139 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_7_NaturalOrdering(Mat,MatFactorInfo*,Mat*); 140 EXTERN PetscErrorCode MatLUFactorNumeric_SeqBAIJ_N(Mat,MatFactorInfo*,Mat*); 141 142 EXTERN PetscErrorCode MatMult_SeqBAIJ_1(Mat,Vec,Vec); 143 EXTERN PetscErrorCode MatMult_SeqBAIJ_2(Mat,Vec,Vec); 144 EXTERN PetscErrorCode MatMult_SeqBAIJ_3(Mat,Vec,Vec); 145 EXTERN PetscErrorCode MatMult_SeqBAIJ_4(Mat,Vec,Vec); 146 EXTERN PetscErrorCode MatMult_SeqBAIJ_5(Mat,Vec,Vec); 147 EXTERN PetscErrorCode MatMult_SeqBAIJ_6(Mat,Vec,Vec); 148 EXTERN PetscErrorCode MatMult_SeqBAIJ_7(Mat,Vec,Vec); 149 EXTERN PetscErrorCode MatMult_SeqBAIJ_N(Mat,Vec,Vec); 150 151 EXTERN PetscErrorCode MatMultAdd_SeqBAIJ_1(Mat,Vec,Vec,Vec); 152 EXTERN PetscErrorCode MatMultAdd_SeqBAIJ_2(Mat,Vec,Vec,Vec); 153 EXTERN PetscErrorCode MatMultAdd_SeqBAIJ_3(Mat,Vec,Vec,Vec); 154 EXTERN PetscErrorCode MatMultAdd_SeqBAIJ_4(Mat,Vec,Vec,Vec); 155 EXTERN PetscErrorCode MatMultAdd_SeqBAIJ_5(Mat,Vec,Vec,Vec); 156 EXTERN PetscErrorCode MatMultAdd_SeqBAIJ_6(Mat,Vec,Vec,Vec); 157 EXTERN PetscErrorCode MatMultAdd_SeqBAIJ_7(Mat,Vec,Vec,Vec); 158 EXTERN PetscErrorCode MatMultAdd_SeqBAIJ_N(Mat,Vec,Vec,Vec); 159 EXTERN PetscErrorCode MatLoad_SeqBAIJ(PetscViewer, MatType,Mat*); 160 161 #endif 162