1e8823a70SSatish Balay 279bdfe76SSatish Balay #if !defined(__MPIBAIJ_H) 379bdfe76SSatish Balay #define __MPIBAIJ_H 4c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h> 582c86c8fSBarry Smith #include <petscctable.h> 679bdfe76SSatish Balay 7aa482453SBarry Smith #if defined(PETSC_USE_CTABLE) 8c66d48e9SBarry Smith #define PETSCTABLE PetscTable 932b39ab2SSatish Balay #else 10b24ad042SBarry Smith #define PETSCTABLE PetscInt* 1132b39ab2SSatish Balay #endif 12e8823a70SSatish Balay 136c6c5352SBarry Smith #define MPIBAIJHEADER \ 14d0f46423SBarry Smith PetscInt *rangebs; /* rmap->range/bs */ \ 15899cda47SBarry Smith PetscInt rstartbs,rendbs,cstartbs,cendbs; /* map values / bs */ \ 161d18e487SKris Buschelman Mat A,B; /* local submatrices: A (diag part), B (off-diag part) */ \ 17b24ad042SBarry Smith PetscMPIInt size; /* size of communicator */ \ 18b24ad042SBarry Smith PetscMPIInt rank; /* rank of proc in communicator */ \ 19521d7252SBarry Smith PetscInt bs2; /* block size, bs2 = bs*bs */ \ 20b24ad042SBarry Smith PetscInt Mbs,Nbs; /* number block rows/cols in matrix; M/bs, N/bs */ \ 21b24ad042SBarry Smith PetscInt mbs,nbs; /* number block rows/cols on processor; m/bs, n/bs */ \ 226c6c5352SBarry Smith \ 236c6c5352SBarry Smith /* The following variables are used for matrix assembly */ \ 246c6c5352SBarry Smith \ 25ace3abfcSBarry Smith PetscBool donotstash; /* if 1, off processor entries dropped */ \ 266c6c5352SBarry Smith MPI_Request *send_waits; /* array of send requests */ \ 276c6c5352SBarry Smith MPI_Request *recv_waits; /* array of receive requests */ \ 28b24ad042SBarry Smith PetscInt nsends,nrecvs; /* numbers of sends and receives */ \ 296c6c5352SBarry Smith MatScalar *svalues,*rvalues; /* sending and receiving data */ \ 30b24ad042SBarry Smith PetscInt rmax; /* maximum message length */ \ 316c6c5352SBarry Smith PETSCTABLE colmap; /* local col number of off-diag col */ \ 326c6c5352SBarry Smith \ 33b24ad042SBarry Smith PetscInt *garray; /* work array */ \ 346c6c5352SBarry Smith \ 356c6c5352SBarry Smith /* The following variable is used by blocked matrix assembly */ \ 366c6c5352SBarry Smith MatScalar *barray; /* Block array of size bs2 */ \ 376c6c5352SBarry Smith \ 386c6c5352SBarry Smith /* The following variables are used for matrix-vector products */ \ 396c6c5352SBarry Smith \ 406c6c5352SBarry Smith Vec lvec; /* local vector */ \ 416c6c5352SBarry Smith VecScatter Mvctx; /* scatter context for vector */ \ 42ace3abfcSBarry Smith PetscBool roworiented; /* if true, row-oriented input, default true */ \ 436c6c5352SBarry Smith \ 446c6c5352SBarry Smith /* The following variables are for MatGetRow() */ \ 456c6c5352SBarry Smith \ 46b24ad042SBarry Smith PetscInt *rowindices; /* column indices for row */ \ 476c6c5352SBarry Smith PetscScalar *rowvalues; /* nonzero values in row */ \ 48ace3abfcSBarry Smith PetscBool getrowactive; /* indicates MatGetRow(), not restored */ \ 496c6c5352SBarry Smith \ 506c6c5352SBarry Smith /* Some variables to make MatSetValues and others more efficient */ \ 51b24ad042SBarry Smith PetscInt rstart_bs,rend_bs; \ 52b24ad042SBarry Smith PetscInt cstart_bs,cend_bs; \ 53b24ad042SBarry Smith PetscInt *ht; /* Hash table to speed up matrix assembly */ \ 546c6c5352SBarry Smith MatScalar **hd; /* Hash table data */ \ 55b24ad042SBarry Smith PetscInt ht_size; \ 56b24ad042SBarry Smith PetscInt ht_total_ct,ht_insert_ct; /* Hash table statistics */ \ 57ace3abfcSBarry Smith PetscBool ht_flag; /* Flag to indicate if hash tables are used */ \ 586c6c5352SBarry Smith double ht_fact; /* Factor to determine the HT size */ \ 596c6c5352SBarry Smith \ 60b24ad042SBarry Smith PetscInt setvalueslen; /* only used for single precision computations */ \ 617a868f3eSHong Zhang MatScalar *setvaluescopy; /* area double precision values in MatSetValuesXXX() are copied*/ \ 627a868f3eSHong Zhang /* before calling MatSetValuesXXX_MPIBAIJ_MatScalar() */ \ 637a868f3eSHong Zhang PetscBool ijonly /* used in MatGetSubMatrices_MPIBAIJ_local() for getting ij structure only */ 64ab9863d7SBarry Smith 656c6c5352SBarry Smith typedef struct { 661d18e487SKris Buschelman MPIBAIJHEADER; 67e8823a70SSatish Balay } Mat_MPIBAIJ; 6879bdfe76SSatish Balay 695a576424SJed Brown PETSC_INTERN PetscErrorCode MatLoad_MPIBAIJ(Mat,PetscViewer); 705a576424SJed Brown PETSC_INTERN PetscErrorCode MatCreateColmap_MPIBAIJ_Private(Mat); 715a576424SJed Brown PETSC_INTERN PetscErrorCode MatGetSubMatrices_MPIBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]); 725a576424SJed Brown PETSC_INTERN PetscErrorCode MatGetSubMatrices_MPIBAIJ_local(Mat,PetscInt,const IS[],const IS[],MatReuse,PetscBool*,PetscBool*,Mat*); 735a576424SJed Brown PETSC_INTERN PetscErrorCode MatGetSubMatrix_MPIBAIJ_Private(Mat,IS,IS,PetscInt,MatReuse,Mat*); 74e8271787SHong Zhang PETSC_INTERN PetscErrorCode MatGetMultiProcBlock_MPIBAIJ(Mat,MPI_Comm,MatReuse,Mat*); 755a576424SJed Brown PETSC_INTERN PetscErrorCode MatIncreaseOverlap_MPIBAIJ(Mat,PetscInt,IS[],PetscInt); 765a576424SJed Brown PETSC_INTERN PetscErrorCode MatIncreaseOverlap_MPIBAIJ_Once(Mat,PetscInt,IS*); 7793dfae19SHong Zhang PETSC_INTERN PetscErrorCode MatFDColoringCreate_MPIXAIJ(Mat,ISColoring,MatFDColoring); 78f86b9fbaSHong Zhang PETSC_INTERN PetscErrorCode MatFDColoringSetUp_MPIXAIJ(Mat,ISColoring,MatFDColoring); 79d1e9a80fSBarry Smith PETSC_INTERN PetscErrorCode MatFDColoringApply_BAIJ(Mat,MatFDColoring,Vec,void*); 80*4de5dceeSHong Zhang 81*4de5dceeSHong Zhang PETSC_INTERN PetscErrorCode MatAXPYGetPreallocation_MPIBAIJ(Mat,const PetscInt *,Mat,const PetscInt*,PetscInt*); 8279bdfe76SSatish Balay #endif 83