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> 5 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 35 36 typedef struct { 37 MPIBAIJHEADER; 38 } Mat_MPIBAIJ; 39 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); 44 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 *); 54 55 PETSC_INTERN PetscErrorCode MatConjugate_SeqBAIJ(Mat); 56