1a4963045SJacob Faibussowitsch #pragma once 2c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h> 3001ddc4fSHong Zhang #include <../src/mat/impls/aij/mpi/mpiaij.h> 4eec179cfSJacob Faibussowitsch #include <petsc/private/hashmapi.h> 579bdfe76SSatish Balay 66c6c5352SBarry Smith #define MPIBAIJHEADER \ 7*68d6bbd5SJunchao Zhang MPIAIJHEADER; \ 8d0f46423SBarry Smith PetscInt *rangebs; /* rmap->range/bs */ \ 9899cda47SBarry Smith PetscInt rstartbs, rendbs, cstartbs, cendbs; /* map values / bs */ \ 10521d7252SBarry Smith PetscInt bs2; /* block size, bs2 = bs*bs */ \ 11b24ad042SBarry Smith PetscInt Mbs, Nbs; /* number block rows/cols in matrix; M/bs, N/bs */ \ 12b24ad042SBarry Smith PetscInt mbs, nbs; /* number block rows/cols on processor; m/bs, n/bs */ \ 136c6c5352SBarry Smith \ 146c6c5352SBarry Smith /* The following variables are used for matrix assembly */ \ 1597da8949SJed Brown PetscBool subset_off_proc_entries; /* PETSC_TRUE if assembly will always communicate a subset of the entries communicated the first time */ \ 166c6c5352SBarry Smith \ 176c6c5352SBarry Smith /* The following variable is used by blocked matrix assembly */ \ 186c6c5352SBarry Smith MatScalar *barray; /* Block array of size bs2 */ \ 196c6c5352SBarry Smith \ 206c6c5352SBarry Smith /* Some variables to make MatSetValues and others more efficient */ \ 21b24ad042SBarry Smith PetscInt rstart_bs, rend_bs; \ 22b24ad042SBarry Smith PetscInt cstart_bs, cend_bs; \ 23b24ad042SBarry Smith PetscInt *ht; /* Hash table to speed up matrix assembly */ \ 246c6c5352SBarry Smith MatScalar **hd; /* Hash table data */ \ 25b24ad042SBarry Smith PetscInt ht_size; \ 26b24ad042SBarry Smith PetscInt ht_total_ct, ht_insert_ct; /* Hash table statistics */ \ 27ace3abfcSBarry Smith PetscBool ht_flag; /* Flag to indicate if hash tables are used */ \ 286c6c5352SBarry Smith double ht_fact; /* Factor to determine the HT size */ \ 296c6c5352SBarry Smith \ 30b24ad042SBarry Smith PetscInt setvalueslen; /* only used for single precision computations */ \ 317a868f3eSHong Zhang MatScalar *setvaluescopy; /* area double precision values in MatSetValuesXXX() are copied*/ \ 327a868f3eSHong Zhang /* before calling MatSetValuesXXX_MPIBAIJ_MatScalar() */ \ 3326cec326SBarry Smith PetscBool ijonly; /* used in MatCreateSubMatrices_MPIBAIJ_local() for getting ij structure only */ \ 3426cec326SBarry Smith struct _MatOps cops 35ab9863d7SBarry Smith 366c6c5352SBarry Smith typedef struct { 371d18e487SKris Buschelman MPIBAIJHEADER; 38e8823a70SSatish Balay } Mat_MPIBAIJ; 3979bdfe76SSatish Balay 40b51a4376SLisandro Dalcin PETSC_INTERN PetscErrorCode MatView_MPIBAIJ(Mat, PetscViewer); 415a576424SJed Brown PETSC_INTERN PetscErrorCode MatLoad_MPIBAIJ(Mat, PetscViewer); 42b51a4376SLisandro Dalcin PETSC_INTERN PetscErrorCode MatView_MPIBAIJ_Binary(Mat, PetscViewer); 43b51a4376SLisandro Dalcin PETSC_INTERN PetscErrorCode MatLoad_MPIBAIJ_Binary(Mat, PetscViewer); 44b51a4376SLisandro Dalcin 455a576424SJed Brown PETSC_INTERN PetscErrorCode MatCreateColmap_MPIBAIJ_Private(Mat); 467dae84e0SHong Zhang PETSC_INTERN PetscErrorCode MatCreateSubMatrices_MPIBAIJ(Mat, PetscInt, const IS[], const IS[], MatReuse, Mat *[]); 47fdfbdca6SPierre Jolivet PETSC_INTERN PetscErrorCode MatCreateSubMatrices_MPIBAIJ_local(Mat, PetscInt, const IS[], const IS[], MatReuse, Mat *, PetscBool); 48fdfbdca6SPierre Jolivet PETSC_INTERN PetscErrorCode MatCreateSubMatrix_MPIBAIJ_Private(Mat, IS, IS, PetscInt, MatReuse, Mat *, PetscBool); 49e8271787SHong Zhang PETSC_INTERN PetscErrorCode MatGetMultiProcBlock_MPIBAIJ(Mat, MPI_Comm, MatReuse, Mat *); 505a576424SJed Brown PETSC_INTERN PetscErrorCode MatIncreaseOverlap_MPIBAIJ(Mat, PetscInt, IS[], PetscInt); 515a576424SJed Brown PETSC_INTERN PetscErrorCode MatIncreaseOverlap_MPIBAIJ_Once(Mat, PetscInt, IS *); 52b5b72c8aSIrina Sokolova PETSC_INTERN PetscErrorCode MatMPIBAIJSetPreallocation_MPIBAIJ(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt *d_nnz, PetscInt o_nz, const PetscInt *o_nnz); 534de5dceeSHong Zhang PETSC_INTERN PetscErrorCode MatAXPYGetPreallocation_MPIBAIJ(Mat, const PetscInt *, Mat, const PetscInt *, PetscInt *); 542726fb6dSPierre Jolivet 552726fb6dSPierre Jolivet PETSC_INTERN PetscErrorCode MatConjugate_SeqBAIJ(Mat); 56