1*a30f8f8cSSatish Balay /* $Id: mpisbaij.h,v 1.26 2000/04/15 04:16:32 bsmith Exp bsmith $ */ 2*a30f8f8cSSatish Balay 3*a30f8f8cSSatish Balay #include "src/mat/impls/baij/seq/baij.h" 4*a30f8f8cSSatish Balay #include "src/sys/ctable.h" 5*a30f8f8cSSatish Balay #include "src/mat/impls/sbaij/seq/sbaij.h" 6*a30f8f8cSSatish Balay 7*a30f8f8cSSatish Balay #if !defined(__MPISBAIJ_H) 8*a30f8f8cSSatish Balay #define __MPISBAIJ_H 9*a30f8f8cSSatish Balay 10*a30f8f8cSSatish Balay typedef struct { 11*a30f8f8cSSatish Balay int *rowners,*cowners; /* ranges owned by each processor */ 12*a30f8f8cSSatish Balay int *rowners_bs; /* rowners*bs */ 13*a30f8f8cSSatish Balay int m,n; /* local rows/columns, m=n */ 14*a30f8f8cSSatish Balay int M,N; /* global rows and columns, M=N */ 15*a30f8f8cSSatish Balay int rstart,rend; /* starting and ending owned rows */ 16*a30f8f8cSSatish Balay int cstart,cend; /* starting and ending owned columns */ 17*a30f8f8cSSatish Balay Mat A,B; /* local submatrices: 18*a30f8f8cSSatish Balay A (diag part) in SeqSBAIJ format, 19*a30f8f8cSSatish Balay B (supper off-diag part) in SeqBAIJ format */ 20*a30f8f8cSSatish Balay int size; /* size of communicator */ 21*a30f8f8cSSatish Balay int rank; /* rank of proc in communicator */ 22*a30f8f8cSSatish Balay int bs,bs2; /* block size, bs2 = bs*bs */ 23*a30f8f8cSSatish Balay int Mbs,Nbs; /* number block rows/cols in matrix; M/bs */ 24*a30f8f8cSSatish Balay int mbs,nbs; /* number block rows/cols on processor; m/bs=n/bs */ 25*a30f8f8cSSatish Balay 26*a30f8f8cSSatish Balay /* The following variables are used for matrix assembly */ 27*a30f8f8cSSatish Balay 28*a30f8f8cSSatish Balay PetscTruth donotstash; /* if 1, off processor entries dropped */ 29*a30f8f8cSSatish Balay MPI_Request *send_waits; /* array of send requests */ 30*a30f8f8cSSatish Balay MPI_Request *recv_waits; /* array of receive requests */ 31*a30f8f8cSSatish Balay int nsends,nrecvs; /* numbers of sends and receives */ 32*a30f8f8cSSatish Balay MatScalar *svalues,*rvalues; /* sending and receiving data */ 33*a30f8f8cSSatish Balay int rmax; /* maximum message length */ 34*a30f8f8cSSatish Balay #if defined (PETSC_USE_CTABLE) 35*a30f8f8cSSatish Balay PetscTable colmap; 36*a30f8f8cSSatish Balay #else 37*a30f8f8cSSatish Balay int *colmap; /* local col number of off-diag col */ 38*a30f8f8cSSatish Balay #endif 39*a30f8f8cSSatish Balay int *garray; /* work array */ 40*a30f8f8cSSatish Balay 41*a30f8f8cSSatish Balay /* The following variable is used by blocked matrix assembly */ 42*a30f8f8cSSatish Balay MatScalar *barray; /* Block array of size bs2 */ 43*a30f8f8cSSatish Balay 44*a30f8f8cSSatish Balay /* The following variables are used for matrix-vector products */ 45*a30f8f8cSSatish Balay 46*a30f8f8cSSatish Balay Vec lvec; /* local vector */ 47*a30f8f8cSSatish Balay VecScatter Mvctx; /* scatter context for vector */ 48*a30f8f8cSSatish Balay PetscTruth roworiented; /* if true, row-oriented input, default true */ 49*a30f8f8cSSatish Balay 50*a30f8f8cSSatish Balay /* The following variables are for MatGetRow() */ 51*a30f8f8cSSatish Balay 52*a30f8f8cSSatish Balay int *rowindices; /* column indices for row */ 53*a30f8f8cSSatish Balay Scalar *rowvalues; /* nonzero values in row */ 54*a30f8f8cSSatish Balay PetscTruth getrowactive; /* indicates MatGetRow(), not restored */ 55*a30f8f8cSSatish Balay 56*a30f8f8cSSatish Balay /* Some variables to make MatSetValues and others more efficient */ 57*a30f8f8cSSatish Balay int rstart_bs,rend_bs; 58*a30f8f8cSSatish Balay int cstart_bs,cend_bs; 59*a30f8f8cSSatish Balay int *ht; /* Hash table to speed up matrix assembly */ 60*a30f8f8cSSatish Balay MatScalar **hd; /* Hash table data */ 61*a30f8f8cSSatish Balay int ht_size; 62*a30f8f8cSSatish Balay int ht_total_ct,ht_insert_ct; /* Hash table statistics */ 63*a30f8f8cSSatish Balay PetscTruth ht_flag; /* Flag to indicate if hash tables are used */ 64*a30f8f8cSSatish Balay double ht_fact; /* Factor to determine the HT size */ 65*a30f8f8cSSatish Balay 66*a30f8f8cSSatish Balay #if defined(PETSC_USE_MAT_SINGLE) 67*a30f8f8cSSatish Balay int setvalueslen; 68*a30f8f8cSSatish Balay MatScalar *setvaluescopy; /* area double precision values in MatSetValuesXXX() are copied 69*a30f8f8cSSatish Balay before calling MatSetValuesXXX_MPISBAIJ_MatScalar() */ 70*a30f8f8cSSatish Balay #endif 71*a30f8f8cSSatish Balay } Mat_MPISBAIJ; 72*a30f8f8cSSatish Balay 73*a30f8f8cSSatish Balay #endif 74