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