baij/seq/baij.h

*54138f6bSKris Buschelman/* $Id: baij.h,v 1.31 2001/07/12 00:37:47 buschelm Exp buschelm $ */
2593348eSBarry Smith
70f55243SBarry Smith#include "src/mat/matimpl.h"
2593348eSBarry Smith
35aab85fSBarry Smith#if !defined(__BAIJ_H)
35aab85fSBarry Smith#define __BAIJ_H
2593348eSBarry Smith
2593348eSBarry Smith/*
35aab85fSBarry Smith  MATSEQBAIJ format - Block compressed row storage. The i[] and j[]
e24b481bSBarry Smith  arrays start at 0.
2593348eSBarry Smith*/
2593348eSBarry Smith
2593348eSBarry Smithtypedef struct {
c4992f7dSBarry Smith  PetscTruth       sorted;       /* if true, rows are sorted by increasing columns */
c4992f7dSBarry Smith  PetscTruth       roworiented;  /* if true, row-oriented input, default */
3369ce9aSBarry Smith  int              nonew;        /* 1 don't add new nonzeros, -1 generate error on new */
c4992f7dSBarry Smith  PetscTruth       singlemalloc; /* if true a, i, and j have been obtained with
2593348eSBarry Smith                                        one big malloc */
2a90d852SSatish Balay  int              bs,bs2;       /* block size, square of block size */
35aab85fSBarry Smith  int              mbs,nbs;      /* rows/bs, columns/bs */
2593348eSBarry Smith  int              nz,maxnz;     /* nonzeros, allocated nonzeros */
2593348eSBarry Smith  int              *diag;        /* pointers to diagonal elements */
2593348eSBarry Smith  int              *i;           /* pointer to beginning of each row */
2593348eSBarry Smith  int              *imax;        /* maximum space allocated for each row */
2593348eSBarry Smith  int              *ilen;        /* actual length of each row */
2593348eSBarry Smith  int              *j;           /* column values: j + i[k] - 1 is start of row k */
3f1db9ecSBarry Smith  MatScalar        *a;           /* nonzero elements */
e51c0b9cSSatish Balay  IS               row,col,icol; /* index sets, used for reorderings */
2593348eSBarry Smith  Scalar           *solve_work;  /* work space used in MatSolve */
2593348eSBarry Smith  void             *spptr;       /* pointer for special library like SuperLU */
2593348eSBarry Smith  int              reallocs;     /* number of mallocs done during MatSetValues()
47b4a8eaSLois Curfman McInnes                                    as more values are set then were preallocated */
4eeb42bcSBarry Smith  Scalar           *mult_work;   /* work array for matrix vector product*/
7fc3c18eSBarry Smith  Scalar           *saved_values;
c4992f7dSBarry Smith
c4992f7dSBarry Smith  PetscTruth       keepzeroedrows; /* if true, MatZeroRows() will not change nonzero structure */
69a4a418SKris Buschelman  PetscTruth       single_precision_solves; /* if true, use single precision for triangular solves */
a377f70aSBarry Smith
a377f70aSBarry Smith#if defined(PETSC_USE_MAT_SINGLE)
e7592fafSBarry Smith  int              setvalueslen;
a377f70aSBarry Smith  MatScalar        *setvaluescopy; /* area double precision values in MatSetValuesXXX() are copied
a377f70aSBarry Smith                                      before calling MatSetValuesXXX_SeqBAIJ_MatScalar() */
a377f70aSBarry Smith#endif
35aab85fSBarry Smith} Mat_SeqBAIJ;
2593348eSBarry Smith
ca44d042SBarry SmithEXTERN int MatILUFactorSymbolic_SeqBAIJ(Mat,IS,IS,MatILUInfo*,Mat *);
ca44d042SBarry SmithEXTERN int MatDuplicate_SeqBAIJ(Mat,MatDuplicateOption,Mat*);
ca44d042SBarry SmithEXTERN int MatMarkDiagonal_SeqBAIJ(Mat);
be3590efSBarry Smith
*54138f6bSKris BuschelmanEXTERN int MatLUFactorSymbolic_SeqBAIJ(Mat,IS,IS,MatLUInfo*,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactor_SeqBAIJ(Mat,IS,IS,MatLUInfo*);
*54138f6bSKris BuschelmanEXTERN int MatIncreaseOverlap_SeqBAIJ(Mat,int,IS*,int);
*54138f6bSKris BuschelmanEXTERN int MatGetSubMatrix_SeqBAIJ(Mat,IS,IS,int,MatReuse,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatGetSubMatrices_SeqBAIJ(Mat,int,IS*,IS*,MatReuse,Mat**);
*54138f6bSKris BuschelmanEXTERN int MatMultTranspose_SeqBAIJ(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMultTransposeAdd_SeqBAIJ(Mat,Vec,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatScale_SeqBAIJ(Scalar*,Mat);
*54138f6bSKris BuschelmanEXTERN int MatNorm_SeqBAIJ(Mat,NormType,PetscReal *);
*54138f6bSKris BuschelmanEXTERN int MatEqual_SeqBAIJ(Mat,Mat,PetscTruth*);
*54138f6bSKris BuschelmanEXTERN int MatGetDiagonal_SeqBAIJ(Mat,Vec);
*54138f6bSKris BuschelmanEXTERN int MatDiagonalScale_SeqBAIJ(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatGetInfo_SeqBAIJ(Mat,MatInfoType,MatInfo *);
*54138f6bSKris BuschelmanEXTERN int MatZeroEntries_SeqBAIJ(Mat);
*54138f6bSKris Buschelman
*54138f6bSKris BuschelmanEXTERN int MatSeqBAIJ_UpdateFactorNumeric_NaturalOrdering(Mat);
*54138f6bSKris BuschelmanEXTERN int MatSeqBAIJ_UpdateSolvers(Mat);
*54138f6bSKris Buschelman
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_Update(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_1(Mat,Vec,Vec);
e37c484bSBarry SmithEXTERN int MatSolve_SeqBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_2(Mat,Vec,Vec);
ca44d042SBarry SmithEXTERN int MatSolve_SeqBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_3(Mat,Vec,Vec);
ca44d042SBarry SmithEXTERN int MatSolve_SeqBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_4(Mat,Vec,Vec);
ca44d042SBarry SmithEXTERN int MatSolve_SeqBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris Buschelman#if defined(PETSC_HAVE_SSE)
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_4_SSE_Demotion(Mat,Vec,Vec);
4110d75bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_4_NaturalOrdering_SSE_Demotion(Mat,Vec,Vec);
*54138f6bSKris Buschelman#else
*54138f6bSKris Buschelman#define MatSolve_SeqBAIJ_4_SSE_Demotion(a,b,c) MatSolve_SeqBAIJ_4(a,b,c)
*54138f6bSKris Buschelman#define MatSolve_SeqBAIJ_4_NaturalOrdering_SSE_Demotion(a,b,c)  MatSolve_SeqBAIJ_4_NaturalOrdering(a,b,c)
4110d75bSKris Buschelman#endif
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_5(Mat,Vec,Vec);
ca44d042SBarry SmithEXTERN int MatSolve_SeqBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_6(Mat,Vec,Vec);
ca44d042SBarry SmithEXTERN int MatSolve_SeqBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_7(Mat,Vec,Vec);
ca44d042SBarry SmithEXTERN int MatSolve_SeqBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolve_SeqBAIJ_N(Mat,Vec,Vec);
*54138f6bSKris Buschelman
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_Update(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_1(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_2(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_3(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_4(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_5(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_6(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatSolveTranspose_SeqBAIJ_7(Mat,Vec,Vec);
ca44d042SBarry SmithEXTERN int MatSolveTranspose_SeqBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
*54138f6bSKris Buschelman
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_1(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_2(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_3(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_3_NaturalOrdering(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_4(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_4_NaturalOrdering(Mat,Mat*);
*54138f6bSKris Buschelman#if defined(PETSC_HAVE_SSE)
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_4_NaturalOrdering_SSE(Mat,Mat*);
*54138f6bSKris Buschelman#else
*54138f6bSKris Buschelman#define MatLUFactorNumeric_SeqBAIJ_4_NaturalOrdering_SSE(a,b) MatLUFactorNumeric_SeqBAIJ_4_NaturalOrdering(a,b)
*54138f6bSKris Buschelman#endif
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_5(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_5_NaturalOrdering(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_6(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_6_NaturalOrdering(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_7(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_7_NaturalOrdering(Mat,Mat*);
*54138f6bSKris BuschelmanEXTERN int MatLUFactorNumeric_SeqBAIJ_N(Mat,Mat*);
*54138f6bSKris Buschelman
*54138f6bSKris BuschelmanEXTERN int MatMult_SeqBAIJ_1(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMult_SeqBAIJ_2(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMult_SeqBAIJ_3(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMult_SeqBAIJ_4(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMult_SeqBAIJ_5(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMult_SeqBAIJ_6(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMult_SeqBAIJ_7(Mat,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMult_SeqBAIJ_N(Mat,Vec,Vec);
*54138f6bSKris Buschelman
*54138f6bSKris BuschelmanEXTERN int MatMultAdd_SeqBAIJ_1(Mat,Vec,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMultAdd_SeqBAIJ_2(Mat,Vec,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMultAdd_SeqBAIJ_3(Mat,Vec,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMultAdd_SeqBAIJ_4(Mat,Vec,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMultAdd_SeqBAIJ_5(Mat,Vec,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMultAdd_SeqBAIJ_6(Mat,Vec,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMultAdd_SeqBAIJ_7(Mat,Vec,Vec,Vec);
*54138f6bSKris BuschelmanEXTERN int MatMultAdd_SeqBAIJ_N(Mat,Vec,Vec,Vec);
2593348eSBarry Smith
2593348eSBarry Smith#endif