Actual source code: mpibaij.h

      1: #pragma once
      2: #include <../src/mat/impls/baij/seq/baij.h>
      3: #include <../src/mat/impls/aij/mpi/mpiaij.h>
      4: #include <petsc/private/hashmapi.h>

      6: #define MPIBAIJHEADER \
      7:   MPIAIJHEADER; \
      8:   PetscInt *rangebs;                            /* rmap->range/bs */ \
      9:   PetscInt  rstartbs, rendbs, cstartbs, cendbs; /* map values / bs  */ \
     10:   PetscInt  bs2;                                /* block size, bs2 = bs*bs */ \
     11:   PetscInt  Mbs, Nbs;                           /* number block rows/cols in matrix; M/bs, N/bs */ \
     12:   PetscInt  mbs, nbs;                           /* number block rows/cols on processor; m/bs, n/bs */ \
     13: \
     14:   /* The following variables are used for matrix assembly */ \
     15:   PetscBool subset_off_proc_entries; /* PETSC_TRUE if assembly will always communicate a subset of the entries communicated the first time */ \
     16: \
     17:   /* The following variable is used by blocked matrix assembly */ \
     18:   MatScalar *barray; /* Block array of size bs2 */ \
     19: \
     20:   /* Some variables to make MatSetValues and others more efficient */ \
     21:   PetscInt    rstart_bs, rend_bs; \
     22:   PetscInt    cstart_bs, cend_bs; \
     23:   PetscInt   *ht; /* Hash table to speed up matrix assembly */ \
     24:   MatScalar **hd; /* Hash table data */ \
     25:   PetscInt    ht_size; \
     26:   PetscInt    ht_total_ct, ht_insert_ct; /* Hash table statistics */ \
     27:   PetscBool   ht_flag;                   /* Flag to indicate if hash tables are used */ \
     28:   double      ht_fact;                   /* Factor to determine the HT size */ \
     29: \
     30:   PetscInt   setvalueslen;  /* only used for single precision computations */ \
     31:   MatScalar *setvaluescopy; /* area double precision values in MatSetValuesXXX() are copied*/ \
     32:                             /* before calling MatSetValuesXXX_MPIBAIJ_MatScalar() */ \
     33:   PetscBool      ijonly;    /* used in  MatCreateSubMatrices_MPIBAIJ_local() for getting ij structure only */ \
     34:   struct _MatOps cops

     36: typedef struct {
     37:   MPIBAIJHEADER;
     38: } Mat_MPIBAIJ;

     40: PETSC_INTERN PetscErrorCode MatView_MPIBAIJ(Mat, PetscViewer);
     41: PETSC_INTERN PetscErrorCode MatLoad_MPIBAIJ(Mat, PetscViewer);
     42: PETSC_INTERN PetscErrorCode MatView_MPIBAIJ_Binary(Mat, PetscViewer);
     43: PETSC_INTERN PetscErrorCode MatLoad_MPIBAIJ_Binary(Mat, PetscViewer);

     45: PETSC_INTERN PetscErrorCode MatCreateColmap_MPIBAIJ_Private(Mat);
     46: PETSC_INTERN PetscErrorCode MatCreateSubMatrices_MPIBAIJ(Mat, PetscInt, const IS[], const IS[], MatReuse, Mat *[]);
     47: PETSC_INTERN PetscErrorCode MatCreateSubMatrices_MPIBAIJ_local(Mat, PetscInt, const IS[], const IS[], MatReuse, Mat *, PetscBool);
     48: PETSC_INTERN PetscErrorCode MatCreateSubMatrix_MPIBAIJ_Private(Mat, IS, IS, PetscInt, MatReuse, Mat *, PetscBool);
     49: PETSC_INTERN PetscErrorCode MatGetMultiProcBlock_MPIBAIJ(Mat, MPI_Comm, MatReuse, Mat *);
     50: PETSC_INTERN PetscErrorCode MatIncreaseOverlap_MPIBAIJ(Mat, PetscInt, IS[], PetscInt);
     51: PETSC_INTERN PetscErrorCode MatIncreaseOverlap_MPIBAIJ_Once(Mat, PetscInt, IS *);
     52: PETSC_INTERN PetscErrorCode MatMPIBAIJSetPreallocation_MPIBAIJ(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt *d_nnz, PetscInt o_nz, const PetscInt *o_nnz);
     53: PETSC_INTERN PetscErrorCode MatAXPYGetPreallocation_MPIBAIJ(Mat, const PetscInt *, Mat, const PetscInt *, PetscInt *);

     55: PETSC_INTERN PetscErrorCode MatConjugate_SeqBAIJ(Mat);