Actual source code: baij.h

 4:  #include src/mat/matimpl.h


  7: /*  
  8:   MATSEQBAIJ format - Block compressed row storage. The i[] and j[] 
  9:   arrays start at 0.
 10: */

 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 */

 50: typedef struct {
 51:   SEQBAIJHEADER
 52: } Mat_SeqBAIJ;

 55: EXTERN PetscErrorCode MatSeqBAIJSetPreallocation_SeqBAIJ(Mat,PetscInt,PetscInt,PetscInt*);
 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);

 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);

 81: EXTERN PetscErrorCode MatSeqBAIJ_UpdateFactorNumeric_NaturalOrdering(Mat);
 82: EXTERN PetscErrorCode MatSeqBAIJ_UpdateSolvers(Mat);

 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);

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);

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*);

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);

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*);

161: #endif