xref: /petsc/src/mat/impls/baij/seq/baij2.c (revision ee54c7eeedc0e12ac48b27e5e83e6f2d15f2ca1c)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2cac129eeSSatish Balay 
37c4f633dSBarry Smith #include "../src/mat/impls/baij/seq/baij.h"
4c60f0209SBarry Smith #include "../src/mat/blockinvert.h"
50a835dfdSSatish Balay #include "petscbt.h"
6cac129eeSSatish Balay 
74a2ae208SSatish Balay #undef __FUNCT__
84a2ae208SSatish Balay #define __FUNCT__ "MatIncreaseOverlap_SeqBAIJ"
9690b6cddSBarry Smith PetscErrorCode MatIncreaseOverlap_SeqBAIJ(Mat A,PetscInt is_max,IS is[],PetscInt ov)
10a3192f15SSatish Balay {
11a3192f15SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
126849ba73SBarry Smith   PetscErrorCode ierr;
135d0c19d7SBarry Smith   PetscInt       row,i,j,k,l,m,n,*nidx,isz,val,ival;
145d0c19d7SBarry Smith   const PetscInt *idx;
15690b6cddSBarry Smith   PetscInt       start,end,*ai,*aj,bs,*nidx2;
16f1af5d2fSBarry Smith   PetscBT        table;
17a3192f15SSatish Balay 
183a40ed3dSBarry Smith   PetscFunctionBegin;
19a3192f15SSatish Balay   m     = a->mbs;
20a3192f15SSatish Balay   ai    = a->i;
21a3192f15SSatish Balay   aj    = a->j;
22d0f46423SBarry Smith   bs    = A->rmap->bs;
23a3192f15SSatish Balay 
2429bbc08cSBarry Smith   if (ov < 0)  SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative overlap specified");
25a3192f15SSatish Balay 
266831982aSBarry Smith   ierr = PetscBTCreate(m,table);CHKERRQ(ierr);
27690b6cddSBarry Smith   ierr = PetscMalloc((m+1)*sizeof(PetscInt),&nidx);CHKERRQ(ierr);
28d0f46423SBarry Smith   ierr = PetscMalloc((A->rmap->N+1)*sizeof(PetscInt),&nidx2);CHKERRQ(ierr);
29a3192f15SSatish Balay 
30a3192f15SSatish Balay   for (i=0; i<is_max; i++) {
31a3192f15SSatish Balay     /* Initialise the two local arrays */
32a3192f15SSatish Balay     isz  = 0;
336831982aSBarry Smith     ierr = PetscBTMemzero(m,table);CHKERRQ(ierr);
34a3192f15SSatish Balay 
35a3192f15SSatish Balay     /* Extract the indices, assume there can be duplicate entries */
36a3192f15SSatish Balay     ierr = ISGetIndices(is[i],&idx);CHKERRQ(ierr);
37b9b97703SBarry Smith     ierr = ISGetLocalSize(is[i],&n);CHKERRQ(ierr);
38a3192f15SSatish Balay 
39a3192f15SSatish Balay     /* Enter these into the temp arrays i.e mark table[row], enter row into new index */
40a3192f15SSatish Balay     for (j=0; j<n ; ++j){
41218c64b6SSatish Balay       ival = idx[j]/bs; /* convert the indices into block indices */
425eee224dSHong Zhang       if (ival>=m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"index greater than mat-dim");
43f1af5d2fSBarry Smith       if(!PetscBTLookupSet(table,ival)) { nidx[isz++] = ival;}
44a3192f15SSatish Balay     }
45a3192f15SSatish Balay     ierr = ISRestoreIndices(is[i],&idx);CHKERRQ(ierr);
46a3192f15SSatish Balay     ierr = ISDestroy(is[i]);CHKERRQ(ierr);
47a3192f15SSatish Balay 
48a3192f15SSatish Balay     k = 0;
49a3192f15SSatish Balay     for (j=0; j<ov; j++){ /* for each overlap*/
50a3192f15SSatish Balay       n = isz;
51a3192f15SSatish Balay       for (; k<n ; k++){ /* do only those rows in nidx[k], which are not done yet */
52a3192f15SSatish Balay         row   = nidx[k];
53a3192f15SSatish Balay         start = ai[row];
54a3192f15SSatish Balay         end   = ai[row+1];
55a3192f15SSatish Balay         for (l = start; l<end ; l++){
56a3192f15SSatish Balay           val = aj[l];
57f1af5d2fSBarry Smith           if (!PetscBTLookupSet(table,val)) {nidx[isz++] = val;}
58a3192f15SSatish Balay         }
59a3192f15SSatish Balay       }
60a3192f15SSatish Balay     }
61218c64b6SSatish Balay     /* expand the Index Set */
62218c64b6SSatish Balay     for (j=0; j<isz; j++) {
63218c64b6SSatish Balay       for (k=0; k<bs; k++)
64218c64b6SSatish Balay         nidx2[j*bs+k] = nidx[j]*bs+k;
65218c64b6SSatish Balay     }
66329f5518SBarry Smith     ierr = ISCreateGeneral(PETSC_COMM_SELF,isz*bs,nidx2,is+i);CHKERRQ(ierr);
67a3192f15SSatish Balay   }
686831982aSBarry Smith   ierr = PetscBTDestroy(table);CHKERRQ(ierr);
69606d414cSSatish Balay   ierr = PetscFree(nidx);CHKERRQ(ierr);
70606d414cSSatish Balay   ierr = PetscFree(nidx2);CHKERRQ(ierr);
713a40ed3dSBarry Smith   PetscFunctionReturn(0);
72a3192f15SSatish Balay }
731c351548SSatish Balay 
744a2ae208SSatish Balay #undef __FUNCT__
754a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqBAIJ_Private"
764aa3045dSJed Brown PetscErrorCode MatGetSubMatrix_SeqBAIJ_Private(Mat A,IS isrow,IS iscol,MatReuse scall,Mat *B)
77736121d4SSatish Balay {
78736121d4SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data,*c;
796849ba73SBarry Smith   PetscErrorCode ierr;
80690b6cddSBarry Smith   PetscInt       *smap,i,k,kstart,kend,oldcols = a->nbs,*lens;
81690b6cddSBarry Smith   PetscInt       row,mat_i,*mat_j,tcol,*mat_ilen;
825d0c19d7SBarry Smith   const PetscInt *irow,*icol;
835d0c19d7SBarry Smith   PetscInt       nrows,ncols,*ssmap,bs=A->rmap->bs,bs2=a->bs2;
84690b6cddSBarry Smith   PetscInt       *aj = a->j,*ai = a->i;
853f1db9ecSBarry Smith   MatScalar      *mat_a;
86736121d4SSatish Balay   Mat            C;
8714ca34e6SBarry Smith   PetscTruth     flag,sorted;
88736121d4SSatish Balay 
893a40ed3dSBarry Smith   PetscFunctionBegin;
9014ca34e6SBarry Smith   ierr = ISSorted(iscol,&sorted);CHKERRQ(ierr);
9114ca34e6SBarry Smith   if (!sorted) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"IS is not sorted");
92736121d4SSatish Balay 
93736121d4SSatish Balay   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
94218c64b6SSatish Balay   ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr);
95b9b97703SBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
96b9b97703SBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
97736121d4SSatish Balay 
98690b6cddSBarry Smith   ierr = PetscMalloc((1+oldcols)*sizeof(PetscInt),&smap);CHKERRQ(ierr);
99736121d4SSatish Balay   ssmap = smap;
100690b6cddSBarry Smith   ierr = PetscMalloc((1+nrows)*sizeof(PetscInt),&lens);CHKERRQ(ierr);
101690b6cddSBarry Smith   ierr  = PetscMemzero(smap,oldcols*sizeof(PetscInt));CHKERRQ(ierr);
102736121d4SSatish Balay   for (i=0; i<ncols; i++) smap[icol[i]] = i+1;
103736121d4SSatish Balay   /* determine lens of each row */
104736121d4SSatish Balay   for (i=0; i<nrows; i++) {
105736121d4SSatish Balay     kstart  = ai[irow[i]];
106736121d4SSatish Balay     kend    = kstart + a->ilen[irow[i]];
107736121d4SSatish Balay     lens[i] = 0;
108736121d4SSatish Balay       for (k=kstart; k<kend; k++) {
109736121d4SSatish Balay         if (ssmap[aj[k]]) {
110736121d4SSatish Balay           lens[i]++;
111736121d4SSatish Balay         }
112736121d4SSatish Balay       }
113736121d4SSatish Balay     }
114736121d4SSatish Balay   /* Create and fill new matrix */
115736121d4SSatish Balay   if (scall == MAT_REUSE_MATRIX) {
116736121d4SSatish Balay     c = (Mat_SeqBAIJ *)((*B)->data);
117736121d4SSatish Balay 
118d0f46423SBarry Smith     if (c->mbs!=nrows || c->nbs!=ncols || (*B)->rmap->bs!=bs) SETERRQ(PETSC_ERR_ARG_SIZ,"Submatrix wrong size");
119690b6cddSBarry Smith     ierr = PetscMemcmp(c->ilen,lens,c->mbs *sizeof(PetscInt),&flag);CHKERRQ(ierr);
120abc0a331SBarry Smith     if (!flag) {
12129bbc08cSBarry Smith       SETERRQ(PETSC_ERR_ARG_SIZ,"Cannot reuse matrix. wrong no of nonzeros");
122736121d4SSatish Balay     }
123690b6cddSBarry Smith     ierr = PetscMemzero(c->ilen,c->mbs*sizeof(PetscInt));CHKERRQ(ierr);
124736121d4SSatish Balay     C = *B;
1253a40ed3dSBarry Smith   } else {
1267adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&C);CHKERRQ(ierr);
127f69a0ea3SMatthew Knepley     ierr = MatSetSizes(C,nrows*bs,ncols*bs,PETSC_DETERMINE,PETSC_DETERMINE);CHKERRQ(ierr);
1287adad957SLisandro Dalcin     ierr = MatSetType(C,((PetscObject)A)->type_name);CHKERRQ(ierr);
129ab93d7beSBarry Smith     ierr = MatSeqBAIJSetPreallocation_SeqBAIJ(C,bs,0,lens);CHKERRQ(ierr);
130736121d4SSatish Balay   }
131736121d4SSatish Balay   c = (Mat_SeqBAIJ *)(C->data);
132736121d4SSatish Balay   for (i=0; i<nrows; i++) {
133736121d4SSatish Balay     row    = irow[i];
134736121d4SSatish Balay     kstart = ai[row];
135736121d4SSatish Balay     kend   = kstart + a->ilen[row];
136736121d4SSatish Balay     mat_i  = c->i[i];
137736121d4SSatish Balay     mat_j  = c->j + mat_i;
138218c64b6SSatish Balay     mat_a  = c->a + mat_i*bs2;
139736121d4SSatish Balay     mat_ilen = c->ilen + i;
140736121d4SSatish Balay     for (k=kstart; k<kend; k++) {
141736121d4SSatish Balay       if ((tcol=ssmap[a->j[k]])) {
142736121d4SSatish Balay         *mat_j++ = tcol - 1;
143549d3d68SSatish Balay         ierr     = PetscMemcpy(mat_a,a->a+k*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
144549d3d68SSatish Balay         mat_a   += bs2;
145736121d4SSatish Balay         (*mat_ilen)++;
146736121d4SSatish Balay       }
147736121d4SSatish Balay     }
148736121d4SSatish Balay   }
149218c64b6SSatish Balay 
150736121d4SSatish Balay   /* Free work space */
151736121d4SSatish Balay   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
152606d414cSSatish Balay   ierr = PetscFree(smap);CHKERRQ(ierr);
153606d414cSSatish Balay   ierr = PetscFree(lens);CHKERRQ(ierr);
1546d4a8577SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1556d4a8577SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
156736121d4SSatish Balay 
157736121d4SSatish Balay   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
158736121d4SSatish Balay   *B = C;
1593a40ed3dSBarry Smith   PetscFunctionReturn(0);
160736121d4SSatish Balay }
161736121d4SSatish Balay 
1624a2ae208SSatish Balay #undef __FUNCT__
1634a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqBAIJ"
1644aa3045dSJed Brown PetscErrorCode MatGetSubMatrix_SeqBAIJ(Mat A,IS isrow,IS iscol,MatReuse scall,Mat *B)
165218c64b6SSatish Balay {
166218c64b6SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
167218c64b6SSatish Balay   IS             is1,is2;
1686849ba73SBarry Smith   PetscErrorCode ierr;
1695d0c19d7SBarry Smith   PetscInt       *vary,*iary,nrows,ncols,i,bs=A->rmap->bs,count;
1705d0c19d7SBarry Smith   const PetscInt *irow,*icol;
171218c64b6SSatish Balay 
1723a40ed3dSBarry Smith   PetscFunctionBegin;
173218c64b6SSatish Balay   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
174218c64b6SSatish Balay   ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr);
175b9b97703SBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
176b9b97703SBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
177218c64b6SSatish Balay 
178218c64b6SSatish Balay   /* Verify if the indices corespond to each element in a block
179218c64b6SSatish Balay    and form the IS with compressed IS */
180690b6cddSBarry Smith   ierr = PetscMalloc(2*(a->mbs+1)*sizeof(PetscInt),&vary);CHKERRQ(ierr);
181218c64b6SSatish Balay   iary = vary + a->mbs;
182690b6cddSBarry Smith   ierr = PetscMemzero(vary,(a->mbs)*sizeof(PetscInt));CHKERRQ(ierr);
183218c64b6SSatish Balay   for (i=0; i<nrows; i++) vary[irow[i]/bs]++;
184218c64b6SSatish Balay   count = 0;
185218c64b6SSatish Balay   for (i=0; i<a->mbs; i++) {
186634064b4SBarry Smith     if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_ERR_ARG_SIZ,"Index set does not match blocks");
187218c64b6SSatish Balay     if (vary[i]==bs) iary[count++] = i;
188218c64b6SSatish Balay   }
189029af93fSBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is1);CHKERRQ(ierr);
190218c64b6SSatish Balay 
191690b6cddSBarry Smith   ierr = PetscMemzero(vary,(a->mbs)*sizeof(PetscInt));CHKERRQ(ierr);
192218c64b6SSatish Balay   for (i=0; i<ncols; i++) vary[icol[i]/bs]++;
193218c64b6SSatish Balay   count = 0;
194218c64b6SSatish Balay   for (i=0; i<a->mbs; i++) {
195634064b4SBarry Smith     if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_ERR_PLIB,"Internal error in PETSc");
196218c64b6SSatish Balay     if (vary[i]==bs) iary[count++] = i;
197218c64b6SSatish Balay   }
198029af93fSBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is2);CHKERRQ(ierr);
199218c64b6SSatish Balay   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
200218c64b6SSatish Balay   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
201606d414cSSatish Balay   ierr = PetscFree(vary);CHKERRQ(ierr);
202218c64b6SSatish Balay 
2034aa3045dSJed Brown   ierr = MatGetSubMatrix_SeqBAIJ_Private(A,is1,is2,scall,B);CHKERRQ(ierr);
204218c64b6SSatish Balay   ISDestroy(is1);
205218c64b6SSatish Balay   ISDestroy(is2);
2063a40ed3dSBarry Smith   PetscFunctionReturn(0);
207218c64b6SSatish Balay }
208218c64b6SSatish Balay 
2094a2ae208SSatish Balay #undef __FUNCT__
2104a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqBAIJ"
211690b6cddSBarry Smith PetscErrorCode MatGetSubMatrices_SeqBAIJ(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[])
212736121d4SSatish Balay {
2136849ba73SBarry Smith   PetscErrorCode ierr;
214690b6cddSBarry Smith   PetscInt       i;
215736121d4SSatish Balay 
2163a40ed3dSBarry Smith   PetscFunctionBegin;
217736121d4SSatish Balay   if (scall == MAT_INITIAL_MATRIX) {
21882502324SSatish Balay     ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr);
219736121d4SSatish Balay   }
220736121d4SSatish Balay 
221736121d4SSatish Balay   for (i=0; i<n; i++) {
2224aa3045dSJed Brown     ierr = MatGetSubMatrix_SeqBAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr);
223736121d4SSatish Balay   }
2243a40ed3dSBarry Smith   PetscFunctionReturn(0);
225736121d4SSatish Balay }
226218c64b6SSatish Balay 
227218c64b6SSatish Balay 
2282d61bbb3SSatish Balay /* -------------------------------------------------------*/
2292d61bbb3SSatish Balay /* Should check that shapes of vectors and matrices match */
2302d61bbb3SSatish Balay /* -------------------------------------------------------*/
231d9eff348SSatish Balay #include "petscblaslapack.h"
2322d61bbb3SSatish Balay 
2334a2ae208SSatish Balay #undef __FUNCT__
2344a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_1"
235dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_1(Mat A,Vec xx,Vec zz)
2362d61bbb3SSatish Balay {
2372d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
238d9fead3dSBarry Smith   PetscScalar       *z,sum;
239d9fead3dSBarry Smith   const PetscScalar *x;
240d9fead3dSBarry Smith   const MatScalar   *v;
2416849ba73SBarry Smith   PetscErrorCode    ierr;
2422162cab8SBarry Smith   PetscInt          mbs,i,n,nonzerorow=0;
2432162cab8SBarry Smith   const PetscInt    *idx,*ii,*ridx=PETSC_NULL;
24426e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
2452d61bbb3SSatish Balay 
2462d61bbb3SSatish Balay   PetscFunctionBegin;
247d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
2481ebc52fbSHong Zhang   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
2492d61bbb3SSatish Balay 
25026e093fcSHong Zhang   if (usecprow){
25126e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
25226e093fcSHong Zhang     ii   = a->compressedrow.i;
2537b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
25426e093fcSHong Zhang   } else {
25526e093fcSHong Zhang     mbs = a->mbs;
2562d61bbb3SSatish Balay     ii  = a->i;
25726e093fcSHong Zhang   }
2582d61bbb3SSatish Balay 
2592d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
260*ee54c7eeSHong Zhang     n    = ii[1] - ii[0];
261*ee54c7eeSHong Zhang     v    = a->a + ii[0];
262*ee54c7eeSHong Zhang     idx  = a->j + ii[0];
263*ee54c7eeSHong Zhang     ii++;
2642d61bbb3SSatish Balay     sum  = 0.0;
26598c9bda7SSatish Balay     nonzerorow += (n>0);
2662162cab8SBarry Smith     PetscSparseDensePlusDot(sum,x,v,idx,n);
26726e093fcSHong Zhang     if (usecprow){
2687b2bb3b9SHong Zhang       z[ridx[i]] = sum;
26926e093fcSHong Zhang     } else {
2702d61bbb3SSatish Balay       z[i] = sum;
2712d61bbb3SSatish Balay     }
27226e093fcSHong Zhang   }
273d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
2741ebc52fbSHong Zhang   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
275dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr);
2762d61bbb3SSatish Balay   PetscFunctionReturn(0);
2772d61bbb3SSatish Balay }
2782d61bbb3SSatish Balay 
2794a2ae208SSatish Balay #undef __FUNCT__
2804a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_2"
281dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_2(Mat A,Vec xx,Vec zz)
2822d61bbb3SSatish Balay {
2832d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
284d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,*zarray;
285d9fead3dSBarry Smith   const PetscScalar *x,*xb;
28687828ca2SBarry Smith   PetscScalar       x1,x2;
287d9fead3dSBarry Smith   const MatScalar   *v;
288dfbe8321SBarry Smith   PetscErrorCode    ierr;
28998c9bda7SSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
29026e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
2912d61bbb3SSatish Balay 
2922d61bbb3SSatish Balay   PetscFunctionBegin;
293d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
29426e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
2952d61bbb3SSatish Balay 
2962d61bbb3SSatish Balay   idx = a->j;
2972d61bbb3SSatish Balay   v   = a->a;
29826e093fcSHong Zhang   if (usecprow){
29926e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
30026e093fcSHong Zhang     ii   = a->compressedrow.i;
3017b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
30226e093fcSHong Zhang   } else {
30326e093fcSHong Zhang     mbs = a->mbs;
3042d61bbb3SSatish Balay     ii  = a->i;
30526e093fcSHong Zhang     z   = zarray;
30626e093fcSHong Zhang   }
3072d61bbb3SSatish Balay 
3082d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
3092d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
3102d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0;
31198c9bda7SSatish Balay     nonzerorow += (n>0);
3122d61bbb3SSatish Balay     for (j=0; j<n; j++) {
3132d61bbb3SSatish Balay       xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1];
3142d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[2]*x2;
3152d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[3]*x2;
3162d61bbb3SSatish Balay       v += 4;
3172d61bbb3SSatish Balay     }
3187b2bb3b9SHong Zhang     if (usecprow) z = zarray + 2*ridx[i];
3192d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2;
32026e093fcSHong Zhang     if (!usecprow) z += 2;
3212d61bbb3SSatish Balay   }
322d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
32326e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
324dc0b31edSSatish Balay   ierr = PetscLogFlops(8.0*a->nz - 2.0*nonzerorow);CHKERRQ(ierr);
3252d61bbb3SSatish Balay   PetscFunctionReturn(0);
3262d61bbb3SSatish Balay }
3272d61bbb3SSatish Balay 
3284a2ae208SSatish Balay #undef __FUNCT__
3294a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_3"
330dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_3(Mat A,Vec xx,Vec zz)
3312d61bbb3SSatish Balay {
3322d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
333d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,sum3,x1,x2,x3,*zarray;
334d9fead3dSBarry Smith   const PetscScalar *x,*xb;
335d9fead3dSBarry Smith   const MatScalar   *v;
336dfbe8321SBarry Smith   PetscErrorCode    ierr;
337028cd4eaSSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
338028cd4eaSSatish Balay   PetscTruth        usecprow=a->compressedrow.use;
33926e093fcSHong Zhang 
3402d61bbb3SSatish Balay 
341b6410449SSatish Balay #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
342fee21e36SBarry Smith #pragma disjoint(*v,*z,*xb)
343fee21e36SBarry Smith #endif
344fee21e36SBarry Smith 
3452d61bbb3SSatish Balay   PetscFunctionBegin;
346d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
34726e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
3482d61bbb3SSatish Balay 
3492d61bbb3SSatish Balay   idx = a->j;
3502d61bbb3SSatish Balay   v   = a->a;
35126e093fcSHong Zhang   if (usecprow){
35226e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
35326e093fcSHong Zhang     ii   = a->compressedrow.i;
3547b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
35526e093fcSHong Zhang   } else {
35626e093fcSHong Zhang     mbs = a->mbs;
3572d61bbb3SSatish Balay     ii  = a->i;
35826e093fcSHong Zhang     z   = zarray;
35926e093fcSHong Zhang   }
3602d61bbb3SSatish Balay 
3612d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
3622d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
3632d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0;
36498c9bda7SSatish Balay     nonzerorow += (n>0);
3652d61bbb3SSatish Balay     for (j=0; j<n; j++) {
3662d61bbb3SSatish Balay       xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
3672d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3;
3682d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3;
3692d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3;
3702d61bbb3SSatish Balay       v += 9;
3712d61bbb3SSatish Balay     }
3727b2bb3b9SHong Zhang     if (usecprow) z = zarray + 3*ridx[i];
3732d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3;
37426e093fcSHong Zhang     if (!usecprow) z += 3;
3752d61bbb3SSatish Balay   }
376d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
37726e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
378dc0b31edSSatish Balay   ierr = PetscLogFlops(18.0*a->nz - 3.0*nonzerorow);CHKERRQ(ierr);
3792d61bbb3SSatish Balay   PetscFunctionReturn(0);
3802d61bbb3SSatish Balay }
3812d61bbb3SSatish Balay 
3824a2ae208SSatish Balay #undef __FUNCT__
3834a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_4"
384dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_4(Mat A,Vec xx,Vec zz)
3852d61bbb3SSatish Balay {
3862d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
387d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*zarray;
388d9fead3dSBarry Smith   const PetscScalar *x,*xb;
389d9fead3dSBarry Smith   const MatScalar   *v;
390dfbe8321SBarry Smith   PetscErrorCode    ierr;
39198c9bda7SSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
39226e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
3932d61bbb3SSatish Balay 
3942d61bbb3SSatish Balay   PetscFunctionBegin;
395d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
39626e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
3972d61bbb3SSatish Balay 
3982d61bbb3SSatish Balay   idx = a->j;
3992d61bbb3SSatish Balay   v   = a->a;
40026e093fcSHong Zhang   if (usecprow){
40126e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
40226e093fcSHong Zhang     ii   = a->compressedrow.i;
4037b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
40426e093fcSHong Zhang   } else {
40526e093fcSHong Zhang     mbs = a->mbs;
4062d61bbb3SSatish Balay     ii  = a->i;
40726e093fcSHong Zhang     z   = zarray;
40826e093fcSHong Zhang   }
4092d61bbb3SSatish Balay 
4102d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
4112d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
4122d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0;
41398c9bda7SSatish Balay     nonzerorow += (n>0);
4142d61bbb3SSatish Balay     for (j=0; j<n; j++) {
4152d61bbb3SSatish Balay       xb = x + 4*(*idx++);
4162d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
4172d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3   + v[12]*x4;
4182d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3   + v[13]*x4;
4192d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3  + v[14]*x4;
4202d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3  + v[15]*x4;
4212d61bbb3SSatish Balay       v += 16;
4222d61bbb3SSatish Balay     }
4237b2bb3b9SHong Zhang     if (usecprow) z = zarray + 4*ridx[i];
4242d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4;
42526e093fcSHong Zhang     if (!usecprow) z += 4;
4262d61bbb3SSatish Balay   }
427d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
42826e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
429dc0b31edSSatish Balay   ierr = PetscLogFlops(32.0*a->nz - 4.0*nonzerorow);CHKERRQ(ierr);
4302d61bbb3SSatish Balay   PetscFunctionReturn(0);
4312d61bbb3SSatish Balay }
4322d61bbb3SSatish Balay 
4334a2ae208SSatish Balay #undef __FUNCT__
4344a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_5"
435dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_5(Mat A,Vec xx,Vec zz)
4362d61bbb3SSatish Balay {
4372d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
438d9fead3dSBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5,*z = 0,*zarray;
439d9fead3dSBarry Smith   const PetscScalar *xb,*x;
440d9fead3dSBarry Smith   const MatScalar   *v;
441dfbe8321SBarry Smith   PetscErrorCode    ierr;
44298c9bda7SSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
44326e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
4442d61bbb3SSatish Balay 
445433994e6SBarry Smith   PetscFunctionBegin;
446d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
44726e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
4482d61bbb3SSatish Balay 
4492d61bbb3SSatish Balay   idx = a->j;
4502d61bbb3SSatish Balay   v   = a->a;
45126e093fcSHong Zhang   if (usecprow){
45226e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
45326e093fcSHong Zhang     ii   = a->compressedrow.i;
4547b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
45526e093fcSHong Zhang   } else {
45626e093fcSHong Zhang     mbs = a->mbs;
4572d61bbb3SSatish Balay     ii  = a->i;
45826e093fcSHong Zhang     z   = zarray;
45926e093fcSHong Zhang   }
4602d61bbb3SSatish Balay 
4612d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
4622d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
4632d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0;
46498c9bda7SSatish Balay     nonzerorow += (n>0);
4652d61bbb3SSatish Balay     for (j=0; j<n; j++) {
4662d61bbb3SSatish Balay       xb = x + 5*(*idx++);
4672d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4];
4682d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3  + v[15]*x4 + v[20]*x5;
4692d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3  + v[16]*x4 + v[21]*x5;
4702d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3  + v[17]*x4 + v[22]*x5;
4712d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3  + v[18]*x4 + v[23]*x5;
4722d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3  + v[19]*x4 + v[24]*x5;
4732d61bbb3SSatish Balay       v += 25;
4742d61bbb3SSatish Balay     }
4757b2bb3b9SHong Zhang     if (usecprow) z = zarray + 5*ridx[i];
4762d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5;
47726e093fcSHong Zhang     if (!usecprow) z += 5;
4782d61bbb3SSatish Balay   }
479d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
48026e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
481dc0b31edSSatish Balay   ierr = PetscLogFlops(50.0*a->nz - 5.0*nonzerorow);CHKERRQ(ierr);
4822d61bbb3SSatish Balay   PetscFunctionReturn(0);
4832d61bbb3SSatish Balay }
4842d61bbb3SSatish Balay 
48515091d37SBarry Smith 
4864a2ae208SSatish Balay #undef __FUNCT__
4874a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_6"
488dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_6(Mat A,Vec xx,Vec zz)
48915091d37SBarry Smith {
49015091d37SBarry Smith   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
491d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,sum5,sum6;
492d9fead3dSBarry Smith   const PetscScalar *x,*xb;
49326e093fcSHong Zhang   PetscScalar       x1,x2,x3,x4,x5,x6,*zarray;
494d9fead3dSBarry Smith   const MatScalar   *v;
495dfbe8321SBarry Smith   PetscErrorCode    ierr;
49698c9bda7SSatish Balay   PetscInt          mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
49726e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
49815091d37SBarry Smith 
499433994e6SBarry Smith   PetscFunctionBegin;
500d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
50126e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
50215091d37SBarry Smith 
50315091d37SBarry Smith   idx = a->j;
50415091d37SBarry Smith   v   = a->a;
50526e093fcSHong Zhang   if (usecprow){
50626e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
50726e093fcSHong Zhang     ii   = a->compressedrow.i;
5087b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
50926e093fcSHong Zhang   } else {
51026e093fcSHong Zhang     mbs = a->mbs;
51115091d37SBarry Smith     ii  = a->i;
51226e093fcSHong Zhang     z   = zarray;
51326e093fcSHong Zhang   }
51415091d37SBarry Smith 
51515091d37SBarry Smith   for (i=0; i<mbs; i++) {
51615091d37SBarry Smith     n  = ii[1] - ii[0]; ii++;
51715091d37SBarry Smith     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0;
51898c9bda7SSatish Balay     nonzerorow += (n>0);
51915091d37SBarry Smith     for (j=0; j<n; j++) {
52015091d37SBarry Smith       xb = x + 6*(*idx++);
52115091d37SBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5];
52215091d37SBarry Smith       sum1 += v[0]*x1 + v[6]*x2  + v[12]*x3  + v[18]*x4 + v[24]*x5 + v[30]*x6;
52315091d37SBarry Smith       sum2 += v[1]*x1 + v[7]*x2  + v[13]*x3  + v[19]*x4 + v[25]*x5 + v[31]*x6;
52415091d37SBarry Smith       sum3 += v[2]*x1 + v[8]*x2  + v[14]*x3  + v[20]*x4 + v[26]*x5 + v[32]*x6;
52515091d37SBarry Smith       sum4 += v[3]*x1 + v[9]*x2  + v[15]*x3  + v[21]*x4 + v[27]*x5 + v[33]*x6;
52615091d37SBarry Smith       sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3  + v[22]*x4 + v[28]*x5 + v[34]*x6;
52715091d37SBarry Smith       sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3  + v[23]*x4 + v[29]*x5 + v[35]*x6;
52815091d37SBarry Smith       v += 36;
52915091d37SBarry Smith     }
5307b2bb3b9SHong Zhang     if (usecprow) z = zarray + 6*ridx[i];
53115091d37SBarry Smith     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6;
53226e093fcSHong Zhang     if (!usecprow) z += 6;
53315091d37SBarry Smith   }
53415091d37SBarry Smith 
535d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
53626e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
537dc0b31edSSatish Balay   ierr = PetscLogFlops(72.0*a->nz - 6.0*nonzerorow);CHKERRQ(ierr);
53815091d37SBarry Smith   PetscFunctionReturn(0);
53915091d37SBarry Smith }
5404a2ae208SSatish Balay #undef __FUNCT__
5414a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_7"
542dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_7(Mat A,Vec xx,Vec zz)
5432d61bbb3SSatish Balay {
5442d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
545d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7;
546d9fead3dSBarry Smith   const PetscScalar *x,*xb;
54726e093fcSHong Zhang   PetscScalar       x1,x2,x3,x4,x5,x6,x7,*zarray;
548d9fead3dSBarry Smith   const MatScalar   *v;
549dfbe8321SBarry Smith   PetscErrorCode    ierr;
55098c9bda7SSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
55126e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
5522d61bbb3SSatish Balay 
553433994e6SBarry Smith   PetscFunctionBegin;
554d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
55526e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
5562d61bbb3SSatish Balay 
5572d61bbb3SSatish Balay   idx = a->j;
5582d61bbb3SSatish Balay   v   = a->a;
55926e093fcSHong Zhang   if (usecprow){
56026e093fcSHong Zhang     mbs    = a->compressedrow.nrows;
56126e093fcSHong Zhang     ii     = a->compressedrow.i;
5627b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
56326e093fcSHong Zhang   } else {
56426e093fcSHong Zhang     mbs = a->mbs;
5652d61bbb3SSatish Balay     ii  = a->i;
56626e093fcSHong Zhang     z   = zarray;
56726e093fcSHong Zhang   }
5682d61bbb3SSatish Balay 
5692d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
5702d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
5712d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0;
57298c9bda7SSatish Balay     nonzerorow += (n>0);
5732d61bbb3SSatish Balay     for (j=0; j<n; j++) {
5742d61bbb3SSatish Balay       xb = x + 7*(*idx++);
5752d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
5762d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[7]*x2  + v[14]*x3  + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7;
5772d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[8]*x2  + v[15]*x3  + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7;
5782d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[9]*x2  + v[16]*x3  + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7;
5792d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3  + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7;
5802d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3  + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7;
5812d61bbb3SSatish Balay       sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3  + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7;
5822d61bbb3SSatish Balay       sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3  + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7;
5832d61bbb3SSatish Balay       v += 49;
5842d61bbb3SSatish Balay     }
5857b2bb3b9SHong Zhang     if (usecprow) z = zarray + 7*ridx[i];
5862d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
58726e093fcSHong Zhang     if (!usecprow) z += 7;
5882d61bbb3SSatish Balay   }
5892d61bbb3SSatish Balay 
590d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
59126e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
592dc0b31edSSatish Balay   ierr = PetscLogFlops(98.0*a->nz - 7.0*nonzerorow);CHKERRQ(ierr);
5932d61bbb3SSatish Balay   PetscFunctionReturn(0);
5942d61bbb3SSatish Balay }
5952d61bbb3SSatish Balay 
5963f1db9ecSBarry Smith /*
5973f1db9ecSBarry Smith     This will not work with MatScalar == float because it calls the BLAS
5983f1db9ecSBarry Smith */
5994a2ae208SSatish Balay #undef __FUNCT__
6004a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_N"
601dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_N(Mat A,Vec xx,Vec zz)
6022d61bbb3SSatish Balay {
6032d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
604dcf5cc72SBarry Smith   PetscScalar    *x,*z = 0,*xb,*work,*workt,*zarray;
6053f1db9ecSBarry Smith   MatScalar      *v;
606dfbe8321SBarry Smith   PetscErrorCode ierr;
607d0f46423SBarry Smith   PetscInt       mbs=a->mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2;
60898c9bda7SSatish Balay   PetscInt       ncols,k,*ridx=PETSC_NULL,nonzerorow=0;
60926e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
6102d61bbb3SSatish Balay 
6112d61bbb3SSatish Balay   PetscFunctionBegin;
6121ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
61326e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
6142d61bbb3SSatish Balay 
6152d61bbb3SSatish Balay   idx = a->j;
6162d61bbb3SSatish Balay   v   = a->a;
61726e093fcSHong Zhang   if (usecprow){
61826e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
61926e093fcSHong Zhang     ii   = a->compressedrow.i;
6207b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
62126e093fcSHong Zhang   } else {
62226e093fcSHong Zhang     mbs = a->mbs;
6232d61bbb3SSatish Balay     ii  = a->i;
62426e093fcSHong Zhang     z   = zarray;
62526e093fcSHong Zhang   }
626218c64b6SSatish Balay 
6272d61bbb3SSatish Balay   if (!a->mult_work) {
628d0f46423SBarry Smith     k    = PetscMax(A->rmap->n,A->cmap->n);
62987828ca2SBarry Smith     ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
6302d61bbb3SSatish Balay   }
6312d61bbb3SSatish Balay   work = a->mult_work;
6322d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
6332d61bbb3SSatish Balay     n     = ii[1] - ii[0]; ii++;
6342d61bbb3SSatish Balay     ncols = n*bs;
6352d61bbb3SSatish Balay     workt = work;
63698c9bda7SSatish Balay     nonzerorow += (n>0);
6372d61bbb3SSatish Balay     for (j=0; j<n; j++) {
6382d61bbb3SSatish Balay       xb = x + bs*(*idx++);
6392d61bbb3SSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
6402d61bbb3SSatish Balay       workt += bs;
6412d61bbb3SSatish Balay     }
6427b2bb3b9SHong Zhang     if (usecprow) z = zarray + bs*ridx[i];
643737d121aSSatish Balay     Kernel_w_gets_Ar_times_v(bs,ncols,work,v,z);
64471044d3cSBarry Smith     /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DZero,z,&_One); */
6452d61bbb3SSatish Balay     v += n*bs2;
64626e093fcSHong Zhang     if (!usecprow) z += bs;
6472d61bbb3SSatish Balay   }
6481ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
64926e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
650dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz*bs2 - bs*nonzerorow);CHKERRQ(ierr);
6512d61bbb3SSatish Balay   PetscFunctionReturn(0);
6522d61bbb3SSatish Balay }
6532d61bbb3SSatish Balay 
6546a65c766SHong Zhang extern PetscErrorCode VecCopy_Seq(Vec,Vec);
6554a2ae208SSatish Balay #undef __FUNCT__
6564a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_1"
657dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_1(Mat A,Vec xx,Vec yy,Vec zz)
6582d61bbb3SSatish Balay {
6592d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
660dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,sum,*yarray,*zarray;
6613f1db9ecSBarry Smith   MatScalar      *v;
662dfbe8321SBarry Smith   PetscErrorCode ierr;
6634eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,n,*ridx=PETSC_NULL;
66426e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
6652d61bbb3SSatish Balay 
6662d61bbb3SSatish Balay   PetscFunctionBegin;
6671ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
66826e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
6692e8a6d31SBarry Smith   if (zz != yy) {
67026e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
6712e8a6d31SBarry Smith   } else {
67226e093fcSHong Zhang     zarray = yarray;
6732e8a6d31SBarry Smith   }
6742d61bbb3SSatish Balay 
6752d61bbb3SSatish Balay   idx = a->j;
6762d61bbb3SSatish Balay   v   = a->a;
67726e093fcSHong Zhang   if (usecprow){
6784eb6d288SHong Zhang     if (zz != yy){
6794eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,mbs*sizeof(PetscScalar));CHKERRQ(ierr);
6804eb6d288SHong Zhang     }
68126e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
68226e093fcSHong Zhang     ii   = a->compressedrow.i;
6837b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
68426e093fcSHong Zhang   } else {
6852d61bbb3SSatish Balay     ii  = a->i;
68626e093fcSHong Zhang     y   = yarray;
68726e093fcSHong Zhang     z   = zarray;
68826e093fcSHong Zhang   }
6892d61bbb3SSatish Balay 
6902d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
6912d61bbb3SSatish Balay     n    = ii[1] - ii[0]; ii++;
69226e093fcSHong Zhang     if (usecprow){
6937b2bb3b9SHong Zhang       z = zarray + ridx[i];
6947b2bb3b9SHong Zhang       y = yarray + ridx[i];
69526e093fcSHong Zhang     }
69626e093fcSHong Zhang     sum = y[0];
6972d61bbb3SSatish Balay     while (n--) sum += *v++ * x[*idx++];
69826e093fcSHong Zhang     z[0] = sum;
69926e093fcSHong Zhang     if (!usecprow){
70026e093fcSHong Zhang       z++; y++;
70126e093fcSHong Zhang     }
7022d61bbb3SSatish Balay   }
7031ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
70426e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
7052e8a6d31SBarry Smith   if (zz != yy) {
70626e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
7072e8a6d31SBarry Smith   }
708dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
7092d61bbb3SSatish Balay   PetscFunctionReturn(0);
7102d61bbb3SSatish Balay }
7112d61bbb3SSatish Balay 
7124a2ae208SSatish Balay #undef __FUNCT__
7134a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_2"
714dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_2(Mat A,Vec xx,Vec yy,Vec zz)
7152d61bbb3SSatish Balay {
7162d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
717dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2;
71826e093fcSHong Zhang   PetscScalar    x1,x2,*yarray,*zarray;
7193f1db9ecSBarry Smith   MatScalar      *v;
720dfbe8321SBarry Smith   PetscErrorCode ierr;
7214eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
72226e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
7232d61bbb3SSatish Balay 
7242d61bbb3SSatish Balay   PetscFunctionBegin;
7251ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
72626e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
7272e8a6d31SBarry Smith   if (zz != yy) {
72826e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
7292e8a6d31SBarry Smith   } else {
73026e093fcSHong Zhang     zarray = yarray;
7312e8a6d31SBarry Smith   }
7322d61bbb3SSatish Balay 
7332d61bbb3SSatish Balay   idx = a->j;
7342d61bbb3SSatish Balay   v   = a->a;
73526e093fcSHong Zhang   if (usecprow){
7364eb6d288SHong Zhang     if (zz != yy){
7374eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,2*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
7384eb6d288SHong Zhang     }
73926e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
74026e093fcSHong Zhang     ii   = a->compressedrow.i;
7417b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
7424eb6d288SHong Zhang     if (zz != yy){
7434eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,a->mbs*sizeof(PetscScalar));CHKERRQ(ierr);
7444eb6d288SHong Zhang     }
74526e093fcSHong Zhang   } else {
7462d61bbb3SSatish Balay     ii  = a->i;
74726e093fcSHong Zhang     y   = yarray;
74826e093fcSHong Zhang     z   = zarray;
74926e093fcSHong Zhang   }
7502d61bbb3SSatish Balay 
7512d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
7522d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
75326e093fcSHong Zhang     if (usecprow){
7547b2bb3b9SHong Zhang       z = zarray + 2*ridx[i];
7557b2bb3b9SHong Zhang       y = yarray + 2*ridx[i];
75626e093fcSHong Zhang     }
7572d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1];
7582d61bbb3SSatish Balay     for (j=0; j<n; j++) {
7592d61bbb3SSatish Balay       xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1];
7602d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[2]*x2;
7612d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[3]*x2;
7622d61bbb3SSatish Balay       v += 4;
7632d61bbb3SSatish Balay     }
7642d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2;
76526e093fcSHong Zhang     if (!usecprow){
7662d61bbb3SSatish Balay       z += 2; y += 2;
7672d61bbb3SSatish Balay     }
76826e093fcSHong Zhang   }
7691ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
77026e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
7712e8a6d31SBarry Smith   if (zz != yy) {
77226e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
7732e8a6d31SBarry Smith   }
774dc0b31edSSatish Balay   ierr = PetscLogFlops(4.0*a->nz);CHKERRQ(ierr);
7752d61bbb3SSatish Balay   PetscFunctionReturn(0);
7762d61bbb3SSatish Balay }
7772d61bbb3SSatish Balay 
7784a2ae208SSatish Balay #undef __FUNCT__
7794a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_3"
780dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_3(Mat A,Vec xx,Vec yy,Vec zz)
7812d61bbb3SSatish Balay {
7822d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
783dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,x1,x2,x3,*yarray,*zarray;
7843f1db9ecSBarry Smith   MatScalar      *v;
785dfbe8321SBarry Smith   PetscErrorCode ierr;
7864eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
78726e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
7882d61bbb3SSatish Balay 
7892d61bbb3SSatish Balay   PetscFunctionBegin;
7901ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
79126e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
7922e8a6d31SBarry Smith   if (zz != yy) {
79326e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
7942e8a6d31SBarry Smith   } else {
79526e093fcSHong Zhang     zarray = yarray;
7962e8a6d31SBarry Smith   }
7972d61bbb3SSatish Balay 
7982d61bbb3SSatish Balay   idx = a->j;
7992d61bbb3SSatish Balay   v   = a->a;
80026e093fcSHong Zhang   if (usecprow){
8014eb6d288SHong Zhang     if (zz != yy){
8024eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,3*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
8034eb6d288SHong Zhang     }
80426e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
80526e093fcSHong Zhang     ii   = a->compressedrow.i;
8067b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
80726e093fcSHong Zhang   } else {
8082d61bbb3SSatish Balay     ii  = a->i;
80926e093fcSHong Zhang     y   = yarray;
81026e093fcSHong Zhang     z   = zarray;
81126e093fcSHong Zhang   }
8122d61bbb3SSatish Balay 
8132d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
8142d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
81526e093fcSHong Zhang     if (usecprow){
8167b2bb3b9SHong Zhang       z = zarray + 3*ridx[i];
8177b2bb3b9SHong Zhang       y = yarray + 3*ridx[i];
81826e093fcSHong Zhang     }
8192d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2];
8202d61bbb3SSatish Balay     for (j=0; j<n; j++) {
8212d61bbb3SSatish Balay       xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
8222d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3;
8232d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3;
8242d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3;
8252d61bbb3SSatish Balay       v += 9;
8262d61bbb3SSatish Balay     }
8272d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3;
82826e093fcSHong Zhang     if (!usecprow){
8292d61bbb3SSatish Balay       z += 3; y += 3;
8302d61bbb3SSatish Balay     }
83126e093fcSHong Zhang   }
8321ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
83326e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
8342e8a6d31SBarry Smith   if (zz != yy) {
83526e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
8362e8a6d31SBarry Smith   }
837dc0b31edSSatish Balay   ierr = PetscLogFlops(18.0*a->nz);CHKERRQ(ierr);
8382d61bbb3SSatish Balay   PetscFunctionReturn(0);
8392d61bbb3SSatish Balay }
8402d61bbb3SSatish Balay 
8414a2ae208SSatish Balay #undef __FUNCT__
8424a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_4"
843dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_4(Mat A,Vec xx,Vec yy,Vec zz)
8442d61bbb3SSatish Balay {
8452d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
846dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*yarray,*zarray;
8473f1db9ecSBarry Smith   MatScalar      *v;
848dfbe8321SBarry Smith   PetscErrorCode ierr;
8494eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
85026e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
8512d61bbb3SSatish Balay 
8522d61bbb3SSatish Balay   PetscFunctionBegin;
8531ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
85426e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
8552e8a6d31SBarry Smith   if (zz != yy) {
85626e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
8572e8a6d31SBarry Smith   } else {
85826e093fcSHong Zhang     zarray = yarray;
8592e8a6d31SBarry Smith   }
8602d61bbb3SSatish Balay 
8612d61bbb3SSatish Balay   idx   = a->j;
8622d61bbb3SSatish Balay   v     = a->a;
86326e093fcSHong Zhang   if (usecprow){
8644eb6d288SHong Zhang     if (zz != yy){
8654eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,4*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
8664eb6d288SHong Zhang     }
86726e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
86826e093fcSHong Zhang     ii   = a->compressedrow.i;
8697b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
87026e093fcSHong Zhang   } else {
8712d61bbb3SSatish Balay     ii  = a->i;
87226e093fcSHong Zhang     y   = yarray;
87326e093fcSHong Zhang     z   = zarray;
87426e093fcSHong Zhang   }
8752d61bbb3SSatish Balay 
8762d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
8772d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
87826e093fcSHong Zhang     if (usecprow){
8797b2bb3b9SHong Zhang       z = zarray + 4*ridx[i];
8807b2bb3b9SHong Zhang       y = yarray + 4*ridx[i];
88126e093fcSHong Zhang     }
8822d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3];
8832d61bbb3SSatish Balay     for (j=0; j<n; j++) {
8842d61bbb3SSatish Balay       xb = x + 4*(*idx++);
8852d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
8862d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3   + v[12]*x4;
8872d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3   + v[13]*x4;
8882d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3  + v[14]*x4;
8892d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3  + v[15]*x4;
8902d61bbb3SSatish Balay       v += 16;
8912d61bbb3SSatish Balay     }
8922d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4;
89326e093fcSHong Zhang     if (!usecprow){
8942d61bbb3SSatish Balay       z += 4; y += 4;
8952d61bbb3SSatish Balay     }
89626e093fcSHong Zhang   }
8971ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
89826e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
8992e8a6d31SBarry Smith   if (zz != yy) {
90026e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
9012e8a6d31SBarry Smith   }
902dc0b31edSSatish Balay   ierr = PetscLogFlops(32.0*a->nz);CHKERRQ(ierr);
9032d61bbb3SSatish Balay   PetscFunctionReturn(0);
9042d61bbb3SSatish Balay }
9052d61bbb3SSatish Balay 
9064a2ae208SSatish Balay #undef __FUNCT__
9074a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_5"
908dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_5(Mat A,Vec xx,Vec yy,Vec zz)
9092d61bbb3SSatish Balay {
9102d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
911dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5;
91226e093fcSHong Zhang   PetscScalar    *yarray,*zarray;
9133f1db9ecSBarry Smith   MatScalar      *v;
914dfbe8321SBarry Smith   PetscErrorCode ierr;
9154eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
91626e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
9172d61bbb3SSatish Balay 
9182d61bbb3SSatish Balay   PetscFunctionBegin;
9191ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
92026e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
9212e8a6d31SBarry Smith   if (zz != yy) {
92226e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
9232e8a6d31SBarry Smith   } else {
92426e093fcSHong Zhang     zarray = yarray;
9252e8a6d31SBarry Smith   }
9262d61bbb3SSatish Balay 
9272d61bbb3SSatish Balay   idx = a->j;
9282d61bbb3SSatish Balay   v   = a->a;
92926e093fcSHong Zhang   if (usecprow){
9304eb6d288SHong Zhang     if (zz != yy){
9314eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,5*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
9324eb6d288SHong Zhang     }
93326e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
93426e093fcSHong Zhang     ii   = a->compressedrow.i;
9357b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
93626e093fcSHong Zhang   } else {
9372d61bbb3SSatish Balay     ii  = a->i;
93826e093fcSHong Zhang     y   = yarray;
93926e093fcSHong Zhang     z   = zarray;
94026e093fcSHong Zhang   }
9412d61bbb3SSatish Balay 
9422d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
9432d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
94426e093fcSHong Zhang     if (usecprow){
9457b2bb3b9SHong Zhang       z = zarray + 5*ridx[i];
9467b2bb3b9SHong Zhang       y = yarray + 5*ridx[i];
94726e093fcSHong Zhang     }
9482d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4];
9492d61bbb3SSatish Balay     for (j=0; j<n; j++) {
9502d61bbb3SSatish Balay       xb = x + 5*(*idx++);
9512d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4];
9522d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3  + v[15]*x4 + v[20]*x5;
9532d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3  + v[16]*x4 + v[21]*x5;
9542d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3  + v[17]*x4 + v[22]*x5;
9552d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3  + v[18]*x4 + v[23]*x5;
9562d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3  + v[19]*x4 + v[24]*x5;
9572d61bbb3SSatish Balay       v += 25;
9582d61bbb3SSatish Balay     }
9592d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5;
96026e093fcSHong Zhang     if (!usecprow){
9612d61bbb3SSatish Balay       z += 5; y += 5;
9622d61bbb3SSatish Balay     }
96326e093fcSHong Zhang   }
9641ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
96526e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
9662e8a6d31SBarry Smith   if (zz != yy) {
96726e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
9682e8a6d31SBarry Smith   }
969dc0b31edSSatish Balay   ierr = PetscLogFlops(50.0*a->nz);CHKERRQ(ierr);
9702d61bbb3SSatish Balay   PetscFunctionReturn(0);
9712d61bbb3SSatish Balay }
9724a2ae208SSatish Balay #undef __FUNCT__
9734a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_6"
974dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_6(Mat A,Vec xx,Vec yy,Vec zz)
97515091d37SBarry Smith {
97615091d37SBarry Smith   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
977dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6;
97826e093fcSHong Zhang   PetscScalar    x1,x2,x3,x4,x5,x6,*yarray,*zarray;
97915091d37SBarry Smith   MatScalar      *v;
980dfbe8321SBarry Smith   PetscErrorCode ierr;
9814eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
98226e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
98315091d37SBarry Smith 
98415091d37SBarry Smith   PetscFunctionBegin;
9851ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
98626e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
98715091d37SBarry Smith   if (zz != yy) {
98826e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
98915091d37SBarry Smith   } else {
99026e093fcSHong Zhang     zarray = yarray;
99115091d37SBarry Smith   }
99215091d37SBarry Smith 
99315091d37SBarry Smith   idx = a->j;
99415091d37SBarry Smith   v   = a->a;
99526e093fcSHong Zhang   if (usecprow){
9964eb6d288SHong Zhang     if (zz != yy){
9974eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,6*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
9984eb6d288SHong Zhang     }
99926e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
100026e093fcSHong Zhang     ii   = a->compressedrow.i;
10017b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
100226e093fcSHong Zhang   } else {
100315091d37SBarry Smith     ii  = a->i;
100426e093fcSHong Zhang     y   = yarray;
100526e093fcSHong Zhang     z   = zarray;
100626e093fcSHong Zhang   }
100715091d37SBarry Smith 
100815091d37SBarry Smith   for (i=0; i<mbs; i++) {
100915091d37SBarry Smith     n  = ii[1] - ii[0]; ii++;
101026e093fcSHong Zhang     if (usecprow){
10117b2bb3b9SHong Zhang       z = zarray + 6*ridx[i];
10127b2bb3b9SHong Zhang       y = yarray + 6*ridx[i];
101326e093fcSHong Zhang     }
101415091d37SBarry Smith     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5];
101515091d37SBarry Smith     for (j=0; j<n; j++) {
10163b95cb0eSSatish Balay       xb = x + 6*(*idx++);
101715091d37SBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5];
101815091d37SBarry Smith       sum1 += v[0]*x1 + v[6]*x2  + v[12]*x3  + v[18]*x4 + v[24]*x5 + v[30]*x6;
101915091d37SBarry Smith       sum2 += v[1]*x1 + v[7]*x2  + v[13]*x3  + v[19]*x4 + v[25]*x5 + v[31]*x6;
102015091d37SBarry Smith       sum3 += v[2]*x1 + v[8]*x2  + v[14]*x3  + v[20]*x4 + v[26]*x5 + v[32]*x6;
102115091d37SBarry Smith       sum4 += v[3]*x1 + v[9]*x2  + v[15]*x3  + v[21]*x4 + v[27]*x5 + v[33]*x6;
102215091d37SBarry Smith       sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3  + v[22]*x4 + v[28]*x5 + v[34]*x6;
102315091d37SBarry Smith       sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3  + v[23]*x4 + v[29]*x5 + v[35]*x6;
102415091d37SBarry Smith       v += 36;
102515091d37SBarry Smith     }
102615091d37SBarry Smith     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6;
102726e093fcSHong Zhang     if (!usecprow){
102815091d37SBarry Smith       z += 6; y += 6;
102915091d37SBarry Smith     }
103026e093fcSHong Zhang   }
10311ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
103226e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
103315091d37SBarry Smith   if (zz != yy) {
103426e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
103515091d37SBarry Smith   }
1036dc0b31edSSatish Balay   ierr = PetscLogFlops(72.0*a->nz);CHKERRQ(ierr);
103715091d37SBarry Smith   PetscFunctionReturn(0);
103815091d37SBarry Smith }
10392d61bbb3SSatish Balay 
10404a2ae208SSatish Balay #undef __FUNCT__
10414a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_7"
1042dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_7(Mat A,Vec xx,Vec yy,Vec zz)
10432d61bbb3SSatish Balay {
10442d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1045dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6,sum7;
104626e093fcSHong Zhang   PetscScalar    x1,x2,x3,x4,x5,x6,x7,*yarray,*zarray;
10473f1db9ecSBarry Smith   MatScalar      *v;
1048dfbe8321SBarry Smith   PetscErrorCode ierr;
10497b2bb3b9SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
105026e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
10512d61bbb3SSatish Balay 
10522d61bbb3SSatish Balay   PetscFunctionBegin;
10531ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
105426e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
10552e8a6d31SBarry Smith   if (zz != yy) {
105626e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
10572e8a6d31SBarry Smith   } else {
105826e093fcSHong Zhang     zarray = yarray;
10592e8a6d31SBarry Smith   }
10602d61bbb3SSatish Balay 
10612d61bbb3SSatish Balay   idx = a->j;
10622d61bbb3SSatish Balay   v   = a->a;
106326e093fcSHong Zhang   if (usecprow){
10644eb6d288SHong Zhang     if (zz != yy){
10654eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,7*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
10664eb6d288SHong Zhang     }
106726e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
106826e093fcSHong Zhang     ii   = a->compressedrow.i;
10697b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
107026e093fcSHong Zhang   } else {
10712d61bbb3SSatish Balay     ii  = a->i;
107226e093fcSHong Zhang     y   = yarray;
107326e093fcSHong Zhang     z   = zarray;
107426e093fcSHong Zhang   }
10752d61bbb3SSatish Balay 
10762d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
10772d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
107826e093fcSHong Zhang     if (usecprow){
10797b2bb3b9SHong Zhang       z = zarray + 7*ridx[i];
10807b2bb3b9SHong Zhang       y = yarray + 7*ridx[i];
108126e093fcSHong Zhang     }
10822d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5]; sum7 = y[6];
10832d61bbb3SSatish Balay     for (j=0; j<n; j++) {
10842d61bbb3SSatish Balay       xb = x + 7*(*idx++);
10852d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
10862d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[7]*x2  + v[14]*x3  + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7;
10872d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[8]*x2  + v[15]*x3  + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7;
10882d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[9]*x2  + v[16]*x3  + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7;
10892d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3  + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7;
10902d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3  + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7;
10912d61bbb3SSatish Balay       sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3  + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7;
10922d61bbb3SSatish Balay       sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3  + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7;
10932d61bbb3SSatish Balay       v += 49;
10942d61bbb3SSatish Balay     }
10952d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
109626e093fcSHong Zhang     if (!usecprow){
10972d61bbb3SSatish Balay       z += 7; y += 7;
10982d61bbb3SSatish Balay     }
109926e093fcSHong Zhang   }
11001ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
110126e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
11022e8a6d31SBarry Smith   if (zz != yy) {
110326e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
11042e8a6d31SBarry Smith   }
1105dc0b31edSSatish Balay   ierr = PetscLogFlops(98.0*a->nz);CHKERRQ(ierr);
11062d61bbb3SSatish Balay   PetscFunctionReturn(0);
11072d61bbb3SSatish Balay }
1108218c64b6SSatish Balay 
11094a2ae208SSatish Balay #undef __FUNCT__
11104a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_N"
1111dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_N(Mat A,Vec xx,Vec yy,Vec zz)
11122d61bbb3SSatish Balay {
11132d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1114bba805e6SBarry Smith   PetscScalar    *x,*z = 0,*xb,*work,*workt,*zarray;
11153f1db9ecSBarry Smith   MatScalar      *v;
11166849ba73SBarry Smith   PetscErrorCode ierr;
1117d0f46423SBarry Smith   PetscInt       mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2;
11187b2bb3b9SHong Zhang   PetscInt       ncols,k,*ridx=PETSC_NULL;
111926e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
1120218c64b6SSatish Balay 
11212d61bbb3SSatish Balay   PetscFunctionBegin;
11226a65c766SHong Zhang   ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr);
11231ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
112426e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
11252d61bbb3SSatish Balay 
11262d61bbb3SSatish Balay   idx = a->j;
11272d61bbb3SSatish Balay   v   = a->a;
112826e093fcSHong Zhang   if (usecprow){
112926e093fcSHong Zhang     mbs    = a->compressedrow.nrows;
113026e093fcSHong Zhang     ii     = a->compressedrow.i;
11317b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
113226e093fcSHong Zhang   } else {
113326e093fcSHong Zhang     mbs = a->mbs;
11342d61bbb3SSatish Balay     ii  = a->i;
113526e093fcSHong Zhang     z   = zarray;
113626e093fcSHong Zhang   }
11372d61bbb3SSatish Balay 
11382d61bbb3SSatish Balay   if (!a->mult_work) {
1139d0f46423SBarry Smith     k    = PetscMax(A->rmap->n,A->cmap->n);
114087828ca2SBarry Smith     ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
11412d61bbb3SSatish Balay   }
11422d61bbb3SSatish Balay   work = a->mult_work;
11432d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
11442d61bbb3SSatish Balay     n     = ii[1] - ii[0]; ii++;
11452d61bbb3SSatish Balay     ncols = n*bs;
11462d61bbb3SSatish Balay     workt = work;
11472d61bbb3SSatish Balay     for (j=0; j<n; j++) {
11482d61bbb3SSatish Balay       xb = x + bs*(*idx++);
11492d61bbb3SSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
11502d61bbb3SSatish Balay       workt += bs;
11512d61bbb3SSatish Balay     }
11527b2bb3b9SHong Zhang     if (usecprow) z = zarray + bs*ridx[i];
11533f1db9ecSBarry Smith     Kernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z);
115471044d3cSBarry Smith     /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DOne,z,&_One); */
11552d61bbb3SSatish Balay     v += n*bs2;
115626e093fcSHong Zhang     if (!usecprow){
11572d61bbb3SSatish Balay       z += bs;
11582d61bbb3SSatish Balay     }
115926e093fcSHong Zhang   }
11601ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
116126e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
1162dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz*bs2);CHKERRQ(ierr);
11632d61bbb3SSatish Balay   PetscFunctionReturn(0);
11642d61bbb3SSatish Balay }
11652d61bbb3SSatish Balay 
11664a2ae208SSatish Balay #undef __FUNCT__
11674a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqBAIJ"
1168dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqBAIJ(Mat A,Vec xx,Vec zz)
11692d61bbb3SSatish Balay {
11703447b6efSHong Zhang   PetscScalar    zero = 0.0;
11716849ba73SBarry Smith   PetscErrorCode ierr;
11722d61bbb3SSatish Balay 
11732d61bbb3SSatish Balay   PetscFunctionBegin;
11742dcb1b2aSMatthew Knepley   ierr = VecSet(zz,zero);CHKERRQ(ierr);
11753447b6efSHong Zhang   ierr = MatMultTransposeAdd_SeqBAIJ(A,xx,zz,zz);CHKERRQ(ierr);
11762d61bbb3SSatish Balay   PetscFunctionReturn(0);
11772d61bbb3SSatish Balay }
11782d61bbb3SSatish Balay 
11794a2ae208SSatish Balay #undef __FUNCT__
11804a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqBAIJ"
1181dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqBAIJ(Mat A,Vec xx,Vec yy,Vec zz)
11822d61bbb3SSatish Balay 
11832d61bbb3SSatish Balay {
11842d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
1185dcf5cc72SBarry Smith   PetscScalar       *zb,*x,*z,*xb = 0,x1,x2,x3,x4,x5;
11863f1db9ecSBarry Smith   MatScalar         *v;
11876849ba73SBarry Smith   PetscErrorCode    ierr;
1188d0f46423SBarry Smith   PetscInt          mbs,i,*idx,*ii,rval,bs=A->rmap->bs,j,n,bs2=a->bs2,*ib,*ridx=PETSC_NULL;
11893447b6efSHong Zhang   Mat_CompressedRow cprow = a->compressedrow;
11903447b6efSHong Zhang   PetscTruth        usecprow=cprow.use;
11912d61bbb3SSatish Balay 
11922d61bbb3SSatish Balay   PetscFunctionBegin;
11936a65c766SHong Zhang   if (yy != zz) { ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr); }
11943447b6efSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
11953447b6efSHong Zhang   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
11962d61bbb3SSatish Balay 
11972d61bbb3SSatish Balay   idx = a->j;
11982d61bbb3SSatish Balay   v   = a->a;
11993447b6efSHong Zhang   if (usecprow){
12003447b6efSHong Zhang     mbs  = cprow.nrows;
12013447b6efSHong Zhang     ii   = cprow.i;
12027b2bb3b9SHong Zhang     ridx = cprow.rindex;
12033447b6efSHong Zhang   } else {
12043447b6efSHong Zhang     mbs=a->mbs;
12052d61bbb3SSatish Balay     ii = a->i;
1206f1af5d2fSBarry Smith     xb = x;
12073447b6efSHong Zhang   }
12082d61bbb3SSatish Balay 
12092d61bbb3SSatish Balay   switch (bs) {
12102d61bbb3SSatish Balay   case 1:
12112d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12127b2bb3b9SHong Zhang       if (usecprow) xb = x + ridx[i];
1213f1af5d2fSBarry Smith       x1 = xb[0];
12143447b6efSHong Zhang       ib = idx + ii[0];
12153447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12162d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12172d61bbb3SSatish Balay         rval    = ib[j];
1218f1af5d2fSBarry Smith         z[rval] += *v * x1;
1219f1af5d2fSBarry Smith         v++;
12202d61bbb3SSatish Balay       }
12213447b6efSHong Zhang       if (!usecprow) xb++;
12222d61bbb3SSatish Balay     }
12232d61bbb3SSatish Balay     break;
12242d61bbb3SSatish Balay   case 2:
12252d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12267b2bb3b9SHong Zhang       if (usecprow) xb = x + 2*ridx[i];
1227f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1];
12283447b6efSHong Zhang       ib = idx + ii[0];
12293447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12302d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12312d61bbb3SSatish Balay         rval      = ib[j]*2;
12322d61bbb3SSatish Balay         z[rval++] += v[0]*x1 + v[1]*x2;
12332d61bbb3SSatish Balay         z[rval++] += v[2]*x1 + v[3]*x2;
12342d61bbb3SSatish Balay         v  += 4;
12352d61bbb3SSatish Balay       }
12363447b6efSHong Zhang       if (!usecprow) xb += 2;
12372d61bbb3SSatish Balay     }
12382d61bbb3SSatish Balay     break;
12392d61bbb3SSatish Balay   case 3:
12402d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12417b2bb3b9SHong Zhang       if (usecprow) xb = x + 3*ridx[i];
1242f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
12433447b6efSHong Zhang       ib = idx + ii[0];
12443447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12452d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12462d61bbb3SSatish Balay         rval      = ib[j]*3;
12472d61bbb3SSatish Balay         z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3;
12482d61bbb3SSatish Balay         z[rval++] += v[3]*x1 + v[4]*x2 + v[5]*x3;
12492d61bbb3SSatish Balay         z[rval++] += v[6]*x1 + v[7]*x2 + v[8]*x3;
12502d61bbb3SSatish Balay         v  += 9;
12512d61bbb3SSatish Balay       }
12523447b6efSHong Zhang       if (!usecprow) xb += 3;
12532d61bbb3SSatish Balay     }
12542d61bbb3SSatish Balay     break;
12552d61bbb3SSatish Balay   case 4:
12562d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12577b2bb3b9SHong Zhang       if (usecprow) xb = x + 4*ridx[i];
1258f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
12593447b6efSHong Zhang       ib = idx + ii[0];
12603447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12612d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12622d61bbb3SSatish Balay         rval      = ib[j]*4;
12632d61bbb3SSatish Balay         z[rval++] +=  v[0]*x1 +  v[1]*x2 +  v[2]*x3 +  v[3]*x4;
12642d61bbb3SSatish Balay         z[rval++] +=  v[4]*x1 +  v[5]*x2 +  v[6]*x3 +  v[7]*x4;
12652d61bbb3SSatish Balay         z[rval++] +=  v[8]*x1 +  v[9]*x2 + v[10]*x3 + v[11]*x4;
12662d61bbb3SSatish Balay         z[rval++] += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4;
12672d61bbb3SSatish Balay         v  += 16;
12682d61bbb3SSatish Balay       }
12693447b6efSHong Zhang       if (!usecprow) xb += 4;
12702d61bbb3SSatish Balay     }
12712d61bbb3SSatish Balay     break;
12722d61bbb3SSatish Balay   case 5:
12732d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12747b2bb3b9SHong Zhang       if (usecprow) xb = x + 5*ridx[i];
1275f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
12762d61bbb3SSatish Balay       x4 = xb[3]; x5 = xb[4];
12773447b6efSHong Zhang       ib = idx + ii[0];
12783447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12792d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12802d61bbb3SSatish Balay         rval      = ib[j]*5;
12812d61bbb3SSatish Balay         z[rval++] +=  v[0]*x1 +  v[1]*x2 +  v[2]*x3 +  v[3]*x4 +  v[4]*x5;
12822d61bbb3SSatish Balay         z[rval++] +=  v[5]*x1 +  v[6]*x2 +  v[7]*x3 +  v[8]*x4 +  v[9]*x5;
12832d61bbb3SSatish Balay         z[rval++] += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4 + v[14]*x5;
12842d61bbb3SSatish Balay         z[rval++] += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4 + v[19]*x5;
12852d61bbb3SSatish Balay         z[rval++] += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4 + v[24]*x5;
12862d61bbb3SSatish Balay         v  += 25;
12872d61bbb3SSatish Balay       }
12883447b6efSHong Zhang       if (!usecprow) xb += 5;
12892d61bbb3SSatish Balay     }
12902d61bbb3SSatish Balay     break;
1291f1af5d2fSBarry Smith   default: {      /* block sizes larger then 5 by 5 are handled by BLAS */
1292690b6cddSBarry Smith       PetscInt     ncols,k;
12933447b6efSHong Zhang       PetscScalar  *work,*workt,*xtmp;
12943f1db9ecSBarry Smith 
12952d61bbb3SSatish Balay       if (!a->mult_work) {
1296d0f46423SBarry Smith         k = PetscMax(A->rmap->n,A->cmap->n);
129787828ca2SBarry Smith         ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
12982d61bbb3SSatish Balay       }
12992d61bbb3SSatish Balay       work = a->mult_work;
13003447b6efSHong Zhang       xtmp = x;
13012d61bbb3SSatish Balay       for (i=0; i<mbs; i++) {
13022d61bbb3SSatish Balay         n     = ii[1] - ii[0]; ii++;
13032d61bbb3SSatish Balay         ncols = n*bs;
130487828ca2SBarry Smith         ierr  = PetscMemzero(work,ncols*sizeof(PetscScalar));CHKERRQ(ierr);
13053447b6efSHong Zhang         if (usecprow) {
13067b2bb3b9SHong Zhang           xtmp = x + bs*ridx[i];
13073447b6efSHong Zhang         }
13083447b6efSHong Zhang         Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,xtmp,v,work);
130971044d3cSBarry Smith         /* BLASgemv_("T",&bs,&ncols,&_DOne,v,&bs,xtmp,&_One,&_DOne,work,&_One); */
13102d61bbb3SSatish Balay         v += n*bs2;
13113447b6efSHong Zhang         if (!usecprow) xtmp += bs;
13122d61bbb3SSatish Balay         workt = work;
13132d61bbb3SSatish Balay         for (j=0; j<n; j++) {
13142d61bbb3SSatish Balay           zb = z + bs*(*idx++);
13152d61bbb3SSatish Balay           for (k=0; k<bs; k++) zb[k] += workt[k] ;
13162d61bbb3SSatish Balay           workt += bs;
13172d61bbb3SSatish Balay         }
13182d61bbb3SSatish Balay       }
13192d61bbb3SSatish Balay     }
13202d61bbb3SSatish Balay   }
13213447b6efSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13223447b6efSHong Zhang   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
1323dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz*a->bs2);CHKERRQ(ierr);
13242d61bbb3SSatish Balay   PetscFunctionReturn(0);
13252d61bbb3SSatish Balay }
13262d61bbb3SSatish Balay 
13274a2ae208SSatish Balay #undef __FUNCT__
13284a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqBAIJ"
1329f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqBAIJ(Mat inA,PetscScalar alpha)
13302d61bbb3SSatish Balay {
13312d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)inA->data;
1332690b6cddSBarry Smith   PetscInt       totalnz = a->bs2*a->nz;
1333f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
1334efee365bSSatish Balay   PetscErrorCode ierr;
13350805154bSBarry Smith   PetscBLASInt   one = 1,tnz = PetscBLASIntCast(totalnz);
13362d61bbb3SSatish Balay 
13372d61bbb3SSatish Balay   PetscFunctionBegin;
1338f4df32b1SMatthew Knepley   BLASscal_(&tnz,&oalpha,a->a,&one);
1339efee365bSSatish Balay   ierr = PetscLogFlops(totalnz);CHKERRQ(ierr);
13402d61bbb3SSatish Balay   PetscFunctionReturn(0);
13412d61bbb3SSatish Balay }
13422d61bbb3SSatish Balay 
13434a2ae208SSatish Balay #undef __FUNCT__
13444a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqBAIJ"
1345dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqBAIJ(Mat A,NormType type,PetscReal *norm)
13462d61bbb3SSatish Balay {
13478a62d963SHong Zhang   PetscErrorCode ierr;
13482d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
13493f1db9ecSBarry Smith   MatScalar      *v = a->a;
1350329f5518SBarry Smith   PetscReal      sum = 0.0;
1351d0f46423SBarry Smith   PetscInt       i,j,k,bs=A->rmap->bs,nz=a->nz,bs2=a->bs2,k1;
13522d61bbb3SSatish Balay 
13532d61bbb3SSatish Balay   PetscFunctionBegin;
13542d61bbb3SSatish Balay   if (type == NORM_FROBENIUS) {
13552d61bbb3SSatish Balay     for (i=0; i< bs2*nz; i++) {
1356aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
1357329f5518SBarry Smith       sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
13582d61bbb3SSatish Balay #else
13592d61bbb3SSatish Balay       sum += (*v)*(*v); v++;
13602d61bbb3SSatish Balay #endif
13612d61bbb3SSatish Balay     }
13622d61bbb3SSatish Balay     *norm = sqrt(sum);
13638a62d963SHong Zhang   } else if (type == NORM_1) { /* maximum column sum */
13648a62d963SHong Zhang     PetscReal *tmp;
13658a62d963SHong Zhang     PetscInt  *bcol = a->j;
1366d0f46423SBarry Smith     ierr = PetscMalloc((A->cmap->n+1)*sizeof(PetscReal),&tmp);CHKERRQ(ierr);
1367d0f46423SBarry Smith     ierr = PetscMemzero(tmp,A->cmap->n*sizeof(PetscReal));CHKERRQ(ierr);
13688a62d963SHong Zhang     for (i=0; i<nz; i++){
13698a62d963SHong Zhang       for (j=0; j<bs; j++){
13708a62d963SHong Zhang         k1 = bs*(*bcol) + j; /* column index */
13718a62d963SHong Zhang         for (k=0; k<bs; k++){
13728a62d963SHong Zhang           tmp[k1] += PetscAbsScalar(*v); v++;
13738a62d963SHong Zhang         }
13748a62d963SHong Zhang       }
13758a62d963SHong Zhang       bcol++;
13768a62d963SHong Zhang     }
13778a62d963SHong Zhang     *norm = 0.0;
1378d0f46423SBarry Smith     for (j=0; j<A->cmap->n; j++) {
13798a62d963SHong Zhang       if (tmp[j] > *norm) *norm = tmp[j];
13808a62d963SHong Zhang     }
13818a62d963SHong Zhang     ierr = PetscFree(tmp);CHKERRQ(ierr);
1382596552b5SBarry Smith   } else if (type == NORM_INFINITY) { /* maximum row sum */
1383596552b5SBarry Smith     *norm = 0.0;
1384596552b5SBarry Smith     for (k=0; k<bs; k++) {
138574f84c7bSSatish Balay       for (j=0; j<a->mbs; j++) {
1386596552b5SBarry Smith         v = a->a + bs2*a->i[j] + k;
1387596552b5SBarry Smith         sum = 0.0;
1388596552b5SBarry Smith         for (i=0; i<a->i[j+1]-a->i[j]; i++) {
13890e90e235SBarry Smith           for (k1=0; k1<bs; k1++){
1390596552b5SBarry Smith             sum += PetscAbsScalar(*v);
1391596552b5SBarry Smith             v   += bs;
13922d61bbb3SSatish Balay           }
13930e90e235SBarry Smith         }
1394596552b5SBarry Smith         if (sum > *norm) *norm = sum;
1395596552b5SBarry Smith       }
1396596552b5SBarry Smith     }
1397596552b5SBarry Smith   } else {
139829bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"No support for this norm yet");
13992d61bbb3SSatish Balay   }
14002d61bbb3SSatish Balay   PetscFunctionReturn(0);
14012d61bbb3SSatish Balay }
14022d61bbb3SSatish Balay 
14032d61bbb3SSatish Balay 
14044a2ae208SSatish Balay #undef __FUNCT__
14054a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqBAIJ"
1406dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqBAIJ(Mat A,Mat B,PetscTruth* flg)
14072d61bbb3SSatish Balay {
14082d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ *)A->data,*b = (Mat_SeqBAIJ *)B->data;
1409dfbe8321SBarry Smith   PetscErrorCode ierr;
14102d61bbb3SSatish Balay 
14112d61bbb3SSatish Balay   PetscFunctionBegin;
14122d61bbb3SSatish Balay   /* If the  matrix/block dimensions are not equal, or no of nonzeros or shift */
1413d0f46423SBarry Smith   if ((A->rmap->N != B->rmap->N) || (A->cmap->n != B->cmap->n) || (A->rmap->bs != B->rmap->bs)|| (a->nz != b->nz)) {
1414273d9f13SBarry Smith     *flg = PETSC_FALSE;
1415273d9f13SBarry Smith     PetscFunctionReturn(0);
14162d61bbb3SSatish Balay   }
14172d61bbb3SSatish Balay 
14182d61bbb3SSatish Balay   /* if the a->i are the same */
1419690b6cddSBarry Smith   ierr = PetscMemcmp(a->i,b->i,(a->mbs+1)*sizeof(PetscInt),flg);CHKERRQ(ierr);
1420abc0a331SBarry Smith   if (!*flg) {
14210f5bd95cSBarry Smith     PetscFunctionReturn(0);
14222d61bbb3SSatish Balay   }
14232d61bbb3SSatish Balay 
14242d61bbb3SSatish Balay   /* if a->j are the same */
1425690b6cddSBarry Smith   ierr = PetscMemcmp(a->j,b->j,(a->nz)*sizeof(PetscInt),flg);CHKERRQ(ierr);
1426abc0a331SBarry Smith   if (!*flg) {
14270f5bd95cSBarry Smith     PetscFunctionReturn(0);
14282d61bbb3SSatish Balay   }
14292d61bbb3SSatish Balay   /* if a->a are the same */
1430d0f46423SBarry Smith   ierr = PetscMemcmp(a->a,b->a,(a->nz)*(A->rmap->bs)*(B->rmap->bs)*sizeof(PetscScalar),flg);CHKERRQ(ierr);
14312d61bbb3SSatish Balay   PetscFunctionReturn(0);
14322d61bbb3SSatish Balay 
14332d61bbb3SSatish Balay }
14342d61bbb3SSatish Balay 
14354a2ae208SSatish Balay #undef __FUNCT__
14364a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqBAIJ"
1437dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqBAIJ(Mat A,Vec v)
14382d61bbb3SSatish Balay {
14392d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1440dfbe8321SBarry Smith   PetscErrorCode ierr;
1441690b6cddSBarry Smith   PetscInt       i,j,k,n,row,bs,*ai,*aj,ambs,bs2;
144287828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
14433f1db9ecSBarry Smith   MatScalar      *aa,*aa_j;
14442d61bbb3SSatish Balay 
14452d61bbb3SSatish Balay   PetscFunctionBegin;
144629bbc08cSBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1447d0f46423SBarry Smith   bs   = A->rmap->bs;
14482d61bbb3SSatish Balay   aa   = a->a;
14492d61bbb3SSatish Balay   ai   = a->i;
14502d61bbb3SSatish Balay   aj   = a->j;
14512d61bbb3SSatish Balay   ambs = a->mbs;
14522d61bbb3SSatish Balay   bs2  = a->bs2;
14532d61bbb3SSatish Balay 
14542dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
14551ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1456e1311b90SBarry Smith   ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr);
1457d0f46423SBarry Smith   if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
14582d61bbb3SSatish Balay   for (i=0; i<ambs; i++) {
14592d61bbb3SSatish Balay     for (j=ai[i]; j<ai[i+1]; j++) {
14602d61bbb3SSatish Balay       if (aj[j] == i) {
14612d61bbb3SSatish Balay         row  = i*bs;
14622d61bbb3SSatish Balay         aa_j = aa+j*bs2;
14632d61bbb3SSatish Balay         for (k=0; k<bs2; k+=(bs+1),row++) x[row] = aa_j[k];
14642d61bbb3SSatish Balay         break;
14652d61bbb3SSatish Balay       }
14662d61bbb3SSatish Balay     }
14672d61bbb3SSatish Balay   }
14681ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
14692d61bbb3SSatish Balay   PetscFunctionReturn(0);
14702d61bbb3SSatish Balay }
14712d61bbb3SSatish Balay 
14724a2ae208SSatish Balay #undef __FUNCT__
14734a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqBAIJ"
1474dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqBAIJ(Mat A,Vec ll,Vec rr)
14752d61bbb3SSatish Balay {
14762d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
147787828ca2SBarry Smith   PetscScalar    *l,*r,x,*li,*ri;
14783f1db9ecSBarry Smith   MatScalar      *aa,*v;
1479dfbe8321SBarry Smith   PetscErrorCode ierr;
1480690b6cddSBarry Smith   PetscInt       i,j,k,lm,rn,M,m,n,*ai,*aj,mbs,tmp,bs,bs2;
14812d61bbb3SSatish Balay 
14822d61bbb3SSatish Balay   PetscFunctionBegin;
14832d61bbb3SSatish Balay   ai  = a->i;
14842d61bbb3SSatish Balay   aj  = a->j;
14852d61bbb3SSatish Balay   aa  = a->a;
1486d0f46423SBarry Smith   m   = A->rmap->n;
1487d0f46423SBarry Smith   n   = A->cmap->n;
1488d0f46423SBarry Smith   bs  = A->rmap->bs;
14892d61bbb3SSatish Balay   mbs = a->mbs;
14902d61bbb3SSatish Balay   bs2 = a->bs2;
14912d61bbb3SSatish Balay   if (ll) {
14921ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
1493e1311b90SBarry Smith     ierr = VecGetLocalSize(ll,&lm);CHKERRQ(ierr);
149429bbc08cSBarry Smith     if (lm != m) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length");
14952d61bbb3SSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
14962d61bbb3SSatish Balay       M  = ai[i+1] - ai[i];
14972d61bbb3SSatish Balay       li = l + i*bs;
14982d61bbb3SSatish Balay       v  = aa + bs2*ai[i];
14992d61bbb3SSatish Balay       for (j=0; j<M; j++) { /* for each block */
15002d61bbb3SSatish Balay         for (k=0; k<bs2; k++) {
15012d61bbb3SSatish Balay           (*v++) *= li[k%bs];
15022d61bbb3SSatish Balay         }
15032d61bbb3SSatish Balay       }
15042d61bbb3SSatish Balay     }
15051ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
1506efee365bSSatish Balay     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
15072d61bbb3SSatish Balay   }
15082d61bbb3SSatish Balay 
15092d61bbb3SSatish Balay   if (rr) {
15101ebc52fbSHong Zhang     ierr = VecGetArray(rr,&r);CHKERRQ(ierr);
1511e1311b90SBarry Smith     ierr = VecGetLocalSize(rr,&rn);CHKERRQ(ierr);
151229bbc08cSBarry Smith     if (rn != n) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length");
15132d61bbb3SSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
15142d61bbb3SSatish Balay       M  = ai[i+1] - ai[i];
15152d61bbb3SSatish Balay       v  = aa + bs2*ai[i];
15162d61bbb3SSatish Balay       for (j=0; j<M; j++) { /* for each block */
15172d61bbb3SSatish Balay         ri = r + bs*aj[ai[i]+j];
15182d61bbb3SSatish Balay         for (k=0; k<bs; k++) {
15192d61bbb3SSatish Balay           x = ri[k];
15202d61bbb3SSatish Balay           for (tmp=0; tmp<bs; tmp++) (*v++) *= x;
15212d61bbb3SSatish Balay         }
15222d61bbb3SSatish Balay       }
15232d61bbb3SSatish Balay     }
15241ebc52fbSHong Zhang     ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr);
1525efee365bSSatish Balay     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
15262d61bbb3SSatish Balay   }
15272d61bbb3SSatish Balay   PetscFunctionReturn(0);
15282d61bbb3SSatish Balay }
15292d61bbb3SSatish Balay 
15302d61bbb3SSatish Balay 
15314a2ae208SSatish Balay #undef __FUNCT__
15324a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqBAIJ"
1533dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqBAIJ(Mat A,MatInfoType flag,MatInfo *info)
15342d61bbb3SSatish Balay {
15352d61bbb3SSatish Balay   Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data;
15362d61bbb3SSatish Balay 
15372d61bbb3SSatish Balay   PetscFunctionBegin;
15382d61bbb3SSatish Balay   info->block_size     = a->bs2;
15392d61bbb3SSatish Balay   info->nz_allocated   = a->maxnz;
15402d61bbb3SSatish Balay   info->nz_used        = a->bs2*a->nz;
15412d61bbb3SSatish Balay   info->nz_unneeded    = (double)(info->nz_allocated - info->nz_used);
15422d61bbb3SSatish Balay   info->assemblies   = A->num_ass;
15432d61bbb3SSatish Balay   info->mallocs      = a->reallocs;
15447adad957SLisandro Dalcin   info->memory       = ((PetscObject)A)->mem;
15452d61bbb3SSatish Balay   if (A->factor) {
15462d61bbb3SSatish Balay     info->fill_ratio_given  = A->info.fill_ratio_given;
15472d61bbb3SSatish Balay     info->fill_ratio_needed = A->info.fill_ratio_needed;
15482d61bbb3SSatish Balay     info->factor_mallocs    = A->info.factor_mallocs;
15492d61bbb3SSatish Balay   } else {
15502d61bbb3SSatish Balay     info->fill_ratio_given  = 0;
15512d61bbb3SSatish Balay     info->fill_ratio_needed = 0;
15522d61bbb3SSatish Balay     info->factor_mallocs    = 0;
15532d61bbb3SSatish Balay   }
15542d61bbb3SSatish Balay   PetscFunctionReturn(0);
15552d61bbb3SSatish Balay }
15562d61bbb3SSatish Balay 
15572d61bbb3SSatish Balay 
15584a2ae208SSatish Balay #undef __FUNCT__
15594a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqBAIJ"
1560dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqBAIJ(Mat A)
15612d61bbb3SSatish Balay {
15622d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1563dfbe8321SBarry Smith   PetscErrorCode ierr;
15642d61bbb3SSatish Balay 
15652d61bbb3SSatish Balay   PetscFunctionBegin;
1566549d3d68SSatish Balay   ierr = PetscMemzero(a->a,a->bs2*a->i[a->mbs]*sizeof(MatScalar));CHKERRQ(ierr);
15672d61bbb3SSatish Balay   PetscFunctionReturn(0);
15682d61bbb3SSatish Balay }
1569