xref: /petsc/src/mat/impls/baij/seq/baij2.c (revision a1c3900f8205a3251f25486089597715c7b73577)
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;
254*a1c3900fSBarry Smith     ierr = PetscMemzero(z,mbs*sizeof(PetscScalar));CHKERRQ(ierr);
25526e093fcSHong Zhang   } else {
25626e093fcSHong Zhang     mbs = a->mbs;
2572d61bbb3SSatish Balay     ii  = a->i;
25826e093fcSHong Zhang   }
2592d61bbb3SSatish Balay 
2602d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
261ee54c7eeSHong Zhang     n    = ii[1] - ii[0];
262ee54c7eeSHong Zhang     v    = a->a + ii[0];
263ee54c7eeSHong Zhang     idx  = a->j + ii[0];
264ee54c7eeSHong Zhang     ii++;
2652d61bbb3SSatish Balay     sum  = 0.0;
26698c9bda7SSatish Balay     nonzerorow += (n>0);
2672162cab8SBarry Smith     PetscSparseDensePlusDot(sum,x,v,idx,n);
26826e093fcSHong Zhang     if (usecprow){
2697b2bb3b9SHong Zhang       z[ridx[i]] = sum;
27026e093fcSHong Zhang     } else {
2712d61bbb3SSatish Balay       z[i] = sum;
2722d61bbb3SSatish Balay     }
27326e093fcSHong Zhang   }
274d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
2751ebc52fbSHong Zhang   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
276dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr);
2772d61bbb3SSatish Balay   PetscFunctionReturn(0);
2782d61bbb3SSatish Balay }
2792d61bbb3SSatish Balay 
2804a2ae208SSatish Balay #undef __FUNCT__
2814a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_2"
282dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_2(Mat A,Vec xx,Vec zz)
2832d61bbb3SSatish Balay {
2842d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
285d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,*zarray;
286d9fead3dSBarry Smith   const PetscScalar *x,*xb;
28787828ca2SBarry Smith   PetscScalar       x1,x2;
288d9fead3dSBarry Smith   const MatScalar   *v;
289dfbe8321SBarry Smith   PetscErrorCode    ierr;
29098c9bda7SSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
29126e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
2922d61bbb3SSatish Balay 
2932d61bbb3SSatish Balay   PetscFunctionBegin;
294d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
29526e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
2962d61bbb3SSatish Balay 
2972d61bbb3SSatish Balay   idx = a->j;
2982d61bbb3SSatish Balay   v   = a->a;
29926e093fcSHong Zhang   if (usecprow){
30026e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
30126e093fcSHong Zhang     ii   = a->compressedrow.i;
3027b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
30326e093fcSHong Zhang   } else {
30426e093fcSHong Zhang     mbs = a->mbs;
3052d61bbb3SSatish Balay     ii  = a->i;
30626e093fcSHong Zhang     z   = zarray;
30726e093fcSHong Zhang   }
3082d61bbb3SSatish Balay 
3092d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
3102d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
3112d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0;
31298c9bda7SSatish Balay     nonzerorow += (n>0);
3132d61bbb3SSatish Balay     for (j=0; j<n; j++) {
3142d61bbb3SSatish Balay       xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1];
3152d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[2]*x2;
3162d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[3]*x2;
3172d61bbb3SSatish Balay       v += 4;
3182d61bbb3SSatish Balay     }
3197b2bb3b9SHong Zhang     if (usecprow) z = zarray + 2*ridx[i];
3202d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2;
32126e093fcSHong Zhang     if (!usecprow) z += 2;
3222d61bbb3SSatish Balay   }
323d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
32426e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
325dc0b31edSSatish Balay   ierr = PetscLogFlops(8.0*a->nz - 2.0*nonzerorow);CHKERRQ(ierr);
3262d61bbb3SSatish Balay   PetscFunctionReturn(0);
3272d61bbb3SSatish Balay }
3282d61bbb3SSatish Balay 
3294a2ae208SSatish Balay #undef __FUNCT__
3304a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_3"
331dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_3(Mat A,Vec xx,Vec zz)
3322d61bbb3SSatish Balay {
3332d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
334d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,sum3,x1,x2,x3,*zarray;
335d9fead3dSBarry Smith   const PetscScalar *x,*xb;
336d9fead3dSBarry Smith   const MatScalar   *v;
337dfbe8321SBarry Smith   PetscErrorCode    ierr;
338028cd4eaSSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
339028cd4eaSSatish Balay   PetscTruth        usecprow=a->compressedrow.use;
34026e093fcSHong Zhang 
3412d61bbb3SSatish Balay 
342b6410449SSatish Balay #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
343fee21e36SBarry Smith #pragma disjoint(*v,*z,*xb)
344fee21e36SBarry Smith #endif
345fee21e36SBarry Smith 
3462d61bbb3SSatish Balay   PetscFunctionBegin;
347d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
34826e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
3492d61bbb3SSatish Balay 
3502d61bbb3SSatish Balay   idx = a->j;
3512d61bbb3SSatish Balay   v   = a->a;
35226e093fcSHong Zhang   if (usecprow){
35326e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
35426e093fcSHong Zhang     ii   = a->compressedrow.i;
3557b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
35626e093fcSHong Zhang   } else {
35726e093fcSHong Zhang     mbs = a->mbs;
3582d61bbb3SSatish Balay     ii  = a->i;
35926e093fcSHong Zhang     z   = zarray;
36026e093fcSHong Zhang   }
3612d61bbb3SSatish Balay 
3622d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
3632d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
3642d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0;
36598c9bda7SSatish Balay     nonzerorow += (n>0);
3662d61bbb3SSatish Balay     for (j=0; j<n; j++) {
3672d61bbb3SSatish Balay       xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
3682d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3;
3692d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3;
3702d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3;
3712d61bbb3SSatish Balay       v += 9;
3722d61bbb3SSatish Balay     }
3737b2bb3b9SHong Zhang     if (usecprow) z = zarray + 3*ridx[i];
3742d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3;
37526e093fcSHong Zhang     if (!usecprow) z += 3;
3762d61bbb3SSatish Balay   }
377d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
37826e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
379dc0b31edSSatish Balay   ierr = PetscLogFlops(18.0*a->nz - 3.0*nonzerorow);CHKERRQ(ierr);
3802d61bbb3SSatish Balay   PetscFunctionReturn(0);
3812d61bbb3SSatish Balay }
3822d61bbb3SSatish Balay 
3834a2ae208SSatish Balay #undef __FUNCT__
3844a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_4"
385dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_4(Mat A,Vec xx,Vec zz)
3862d61bbb3SSatish Balay {
3872d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
388d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*zarray;
389d9fead3dSBarry Smith   const PetscScalar *x,*xb;
390d9fead3dSBarry Smith   const MatScalar   *v;
391dfbe8321SBarry Smith   PetscErrorCode    ierr;
39298c9bda7SSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
39326e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
3942d61bbb3SSatish Balay 
3952d61bbb3SSatish Balay   PetscFunctionBegin;
396d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
39726e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
3982d61bbb3SSatish Balay 
3992d61bbb3SSatish Balay   idx = a->j;
4002d61bbb3SSatish Balay   v   = a->a;
40126e093fcSHong Zhang   if (usecprow){
40226e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
40326e093fcSHong Zhang     ii   = a->compressedrow.i;
4047b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
40526e093fcSHong Zhang   } else {
40626e093fcSHong Zhang     mbs = a->mbs;
4072d61bbb3SSatish Balay     ii  = a->i;
40826e093fcSHong Zhang     z   = zarray;
40926e093fcSHong Zhang   }
4102d61bbb3SSatish Balay 
4112d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
4122d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
4132d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0;
41498c9bda7SSatish Balay     nonzerorow += (n>0);
4152d61bbb3SSatish Balay     for (j=0; j<n; j++) {
4162d61bbb3SSatish Balay       xb = x + 4*(*idx++);
4172d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
4182d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3   + v[12]*x4;
4192d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3   + v[13]*x4;
4202d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3  + v[14]*x4;
4212d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3  + v[15]*x4;
4222d61bbb3SSatish Balay       v += 16;
4232d61bbb3SSatish Balay     }
4247b2bb3b9SHong Zhang     if (usecprow) z = zarray + 4*ridx[i];
4252d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4;
42626e093fcSHong Zhang     if (!usecprow) z += 4;
4272d61bbb3SSatish Balay   }
428d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
42926e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
430dc0b31edSSatish Balay   ierr = PetscLogFlops(32.0*a->nz - 4.0*nonzerorow);CHKERRQ(ierr);
4312d61bbb3SSatish Balay   PetscFunctionReturn(0);
4322d61bbb3SSatish Balay }
4332d61bbb3SSatish Balay 
4344a2ae208SSatish Balay #undef __FUNCT__
4354a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_5"
436dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_5(Mat A,Vec xx,Vec zz)
4372d61bbb3SSatish Balay {
4382d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
439d9fead3dSBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5,*z = 0,*zarray;
440d9fead3dSBarry Smith   const PetscScalar *xb,*x;
441d9fead3dSBarry Smith   const MatScalar   *v;
442dfbe8321SBarry Smith   PetscErrorCode    ierr;
44398c9bda7SSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
44426e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
4452d61bbb3SSatish Balay 
446433994e6SBarry Smith   PetscFunctionBegin;
447d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
44826e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
4492d61bbb3SSatish Balay 
4502d61bbb3SSatish Balay   idx = a->j;
4512d61bbb3SSatish Balay   v   = a->a;
45226e093fcSHong Zhang   if (usecprow){
45326e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
45426e093fcSHong Zhang     ii   = a->compressedrow.i;
4557b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
45626e093fcSHong Zhang   } else {
45726e093fcSHong Zhang     mbs = a->mbs;
4582d61bbb3SSatish Balay     ii  = a->i;
45926e093fcSHong Zhang     z   = zarray;
46026e093fcSHong Zhang   }
4612d61bbb3SSatish Balay 
4622d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
4632d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
4642d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0;
46598c9bda7SSatish Balay     nonzerorow += (n>0);
4662d61bbb3SSatish Balay     for (j=0; j<n; j++) {
4672d61bbb3SSatish Balay       xb = x + 5*(*idx++);
4682d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4];
4692d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3  + v[15]*x4 + v[20]*x5;
4702d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3  + v[16]*x4 + v[21]*x5;
4712d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3  + v[17]*x4 + v[22]*x5;
4722d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3  + v[18]*x4 + v[23]*x5;
4732d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3  + v[19]*x4 + v[24]*x5;
4742d61bbb3SSatish Balay       v += 25;
4752d61bbb3SSatish Balay     }
4767b2bb3b9SHong Zhang     if (usecprow) z = zarray + 5*ridx[i];
4772d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5;
47826e093fcSHong Zhang     if (!usecprow) z += 5;
4792d61bbb3SSatish Balay   }
480d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
48126e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
482dc0b31edSSatish Balay   ierr = PetscLogFlops(50.0*a->nz - 5.0*nonzerorow);CHKERRQ(ierr);
4832d61bbb3SSatish Balay   PetscFunctionReturn(0);
4842d61bbb3SSatish Balay }
4852d61bbb3SSatish Balay 
48615091d37SBarry Smith 
4874a2ae208SSatish Balay #undef __FUNCT__
4884a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_6"
489dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_6(Mat A,Vec xx,Vec zz)
49015091d37SBarry Smith {
49115091d37SBarry Smith   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
492d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,sum5,sum6;
493d9fead3dSBarry Smith   const PetscScalar *x,*xb;
49426e093fcSHong Zhang   PetscScalar       x1,x2,x3,x4,x5,x6,*zarray;
495d9fead3dSBarry Smith   const MatScalar   *v;
496dfbe8321SBarry Smith   PetscErrorCode    ierr;
49798c9bda7SSatish Balay   PetscInt          mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
49826e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
49915091d37SBarry Smith 
500433994e6SBarry Smith   PetscFunctionBegin;
501d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
50226e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
50315091d37SBarry Smith 
50415091d37SBarry Smith   idx = a->j;
50515091d37SBarry Smith   v   = a->a;
50626e093fcSHong Zhang   if (usecprow){
50726e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
50826e093fcSHong Zhang     ii   = a->compressedrow.i;
5097b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
51026e093fcSHong Zhang   } else {
51126e093fcSHong Zhang     mbs = a->mbs;
51215091d37SBarry Smith     ii  = a->i;
51326e093fcSHong Zhang     z   = zarray;
51426e093fcSHong Zhang   }
51515091d37SBarry Smith 
51615091d37SBarry Smith   for (i=0; i<mbs; i++) {
51715091d37SBarry Smith     n  = ii[1] - ii[0]; ii++;
51815091d37SBarry Smith     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0;
51998c9bda7SSatish Balay     nonzerorow += (n>0);
52015091d37SBarry Smith     for (j=0; j<n; j++) {
52115091d37SBarry Smith       xb = x + 6*(*idx++);
52215091d37SBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5];
52315091d37SBarry Smith       sum1 += v[0]*x1 + v[6]*x2  + v[12]*x3  + v[18]*x4 + v[24]*x5 + v[30]*x6;
52415091d37SBarry Smith       sum2 += v[1]*x1 + v[7]*x2  + v[13]*x3  + v[19]*x4 + v[25]*x5 + v[31]*x6;
52515091d37SBarry Smith       sum3 += v[2]*x1 + v[8]*x2  + v[14]*x3  + v[20]*x4 + v[26]*x5 + v[32]*x6;
52615091d37SBarry Smith       sum4 += v[3]*x1 + v[9]*x2  + v[15]*x3  + v[21]*x4 + v[27]*x5 + v[33]*x6;
52715091d37SBarry Smith       sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3  + v[22]*x4 + v[28]*x5 + v[34]*x6;
52815091d37SBarry Smith       sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3  + v[23]*x4 + v[29]*x5 + v[35]*x6;
52915091d37SBarry Smith       v += 36;
53015091d37SBarry Smith     }
5317b2bb3b9SHong Zhang     if (usecprow) z = zarray + 6*ridx[i];
53215091d37SBarry Smith     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6;
53326e093fcSHong Zhang     if (!usecprow) z += 6;
53415091d37SBarry Smith   }
53515091d37SBarry Smith 
536d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
53726e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
538dc0b31edSSatish Balay   ierr = PetscLogFlops(72.0*a->nz - 6.0*nonzerorow);CHKERRQ(ierr);
53915091d37SBarry Smith   PetscFunctionReturn(0);
54015091d37SBarry Smith }
5414a2ae208SSatish Balay #undef __FUNCT__
5424a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_7"
543dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_7(Mat A,Vec xx,Vec zz)
5442d61bbb3SSatish Balay {
5452d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
546d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7;
547d9fead3dSBarry Smith   const PetscScalar *x,*xb;
54826e093fcSHong Zhang   PetscScalar       x1,x2,x3,x4,x5,x6,x7,*zarray;
549d9fead3dSBarry Smith   const MatScalar   *v;
550dfbe8321SBarry Smith   PetscErrorCode    ierr;
55198c9bda7SSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
55226e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
5532d61bbb3SSatish Balay 
554433994e6SBarry Smith   PetscFunctionBegin;
555d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
55626e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
5572d61bbb3SSatish Balay 
5582d61bbb3SSatish Balay   idx = a->j;
5592d61bbb3SSatish Balay   v   = a->a;
56026e093fcSHong Zhang   if (usecprow){
56126e093fcSHong Zhang     mbs    = a->compressedrow.nrows;
56226e093fcSHong Zhang     ii     = a->compressedrow.i;
5637b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
56426e093fcSHong Zhang   } else {
56526e093fcSHong Zhang     mbs = a->mbs;
5662d61bbb3SSatish Balay     ii  = a->i;
56726e093fcSHong Zhang     z   = zarray;
56826e093fcSHong Zhang   }
5692d61bbb3SSatish Balay 
5702d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
5712d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
5722d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0;
57398c9bda7SSatish Balay     nonzerorow += (n>0);
5742d61bbb3SSatish Balay     for (j=0; j<n; j++) {
5752d61bbb3SSatish Balay       xb = x + 7*(*idx++);
5762d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
5772d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[7]*x2  + v[14]*x3  + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7;
5782d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[8]*x2  + v[15]*x3  + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7;
5792d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[9]*x2  + v[16]*x3  + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7;
5802d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3  + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7;
5812d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3  + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7;
5822d61bbb3SSatish Balay       sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3  + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7;
5832d61bbb3SSatish Balay       sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3  + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7;
5842d61bbb3SSatish Balay       v += 49;
5852d61bbb3SSatish Balay     }
5867b2bb3b9SHong Zhang     if (usecprow) z = zarray + 7*ridx[i];
5872d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
58826e093fcSHong Zhang     if (!usecprow) z += 7;
5892d61bbb3SSatish Balay   }
5902d61bbb3SSatish Balay 
591d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
59226e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
593dc0b31edSSatish Balay   ierr = PetscLogFlops(98.0*a->nz - 7.0*nonzerorow);CHKERRQ(ierr);
5942d61bbb3SSatish Balay   PetscFunctionReturn(0);
5952d61bbb3SSatish Balay }
5962d61bbb3SSatish Balay 
5973f1db9ecSBarry Smith /*
5983f1db9ecSBarry Smith     This will not work with MatScalar == float because it calls the BLAS
5993f1db9ecSBarry Smith */
6004a2ae208SSatish Balay #undef __FUNCT__
6014a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_N"
602dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_N(Mat A,Vec xx,Vec zz)
6032d61bbb3SSatish Balay {
6042d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
605dcf5cc72SBarry Smith   PetscScalar    *x,*z = 0,*xb,*work,*workt,*zarray;
6063f1db9ecSBarry Smith   MatScalar      *v;
607dfbe8321SBarry Smith   PetscErrorCode ierr;
608d0f46423SBarry Smith   PetscInt       mbs=a->mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2;
60998c9bda7SSatish Balay   PetscInt       ncols,k,*ridx=PETSC_NULL,nonzerorow=0;
61026e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
6112d61bbb3SSatish Balay 
6122d61bbb3SSatish Balay   PetscFunctionBegin;
6131ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
61426e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
6152d61bbb3SSatish Balay 
6162d61bbb3SSatish Balay   idx = a->j;
6172d61bbb3SSatish Balay   v   = a->a;
61826e093fcSHong Zhang   if (usecprow){
61926e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
62026e093fcSHong Zhang     ii   = a->compressedrow.i;
6217b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
62226e093fcSHong Zhang   } else {
62326e093fcSHong Zhang     mbs = a->mbs;
6242d61bbb3SSatish Balay     ii  = a->i;
62526e093fcSHong Zhang     z   = zarray;
62626e093fcSHong Zhang   }
627218c64b6SSatish Balay 
6282d61bbb3SSatish Balay   if (!a->mult_work) {
629d0f46423SBarry Smith     k    = PetscMax(A->rmap->n,A->cmap->n);
63087828ca2SBarry Smith     ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
6312d61bbb3SSatish Balay   }
6322d61bbb3SSatish Balay   work = a->mult_work;
6332d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
6342d61bbb3SSatish Balay     n     = ii[1] - ii[0]; ii++;
6352d61bbb3SSatish Balay     ncols = n*bs;
6362d61bbb3SSatish Balay     workt = work;
63798c9bda7SSatish Balay     nonzerorow += (n>0);
6382d61bbb3SSatish Balay     for (j=0; j<n; j++) {
6392d61bbb3SSatish Balay       xb = x + bs*(*idx++);
6402d61bbb3SSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
6412d61bbb3SSatish Balay       workt += bs;
6422d61bbb3SSatish Balay     }
6437b2bb3b9SHong Zhang     if (usecprow) z = zarray + bs*ridx[i];
644737d121aSSatish Balay     Kernel_w_gets_Ar_times_v(bs,ncols,work,v,z);
64571044d3cSBarry Smith     /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DZero,z,&_One); */
6462d61bbb3SSatish Balay     v += n*bs2;
64726e093fcSHong Zhang     if (!usecprow) z += bs;
6482d61bbb3SSatish Balay   }
6491ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
65026e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
651dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz*bs2 - bs*nonzerorow);CHKERRQ(ierr);
6522d61bbb3SSatish Balay   PetscFunctionReturn(0);
6532d61bbb3SSatish Balay }
6542d61bbb3SSatish Balay 
6556a65c766SHong Zhang extern PetscErrorCode VecCopy_Seq(Vec,Vec);
6564a2ae208SSatish Balay #undef __FUNCT__
6574a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_1"
658dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_1(Mat A,Vec xx,Vec yy,Vec zz)
6592d61bbb3SSatish Balay {
6602d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
661dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,sum,*yarray,*zarray;
6623f1db9ecSBarry Smith   MatScalar      *v;
663dfbe8321SBarry Smith   PetscErrorCode ierr;
6644eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,n,*ridx=PETSC_NULL;
66526e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
6662d61bbb3SSatish Balay 
6672d61bbb3SSatish Balay   PetscFunctionBegin;
6681ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
66926e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
6702e8a6d31SBarry Smith   if (zz != yy) {
67126e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
6722e8a6d31SBarry Smith   } else {
67326e093fcSHong Zhang     zarray = yarray;
6742e8a6d31SBarry Smith   }
6752d61bbb3SSatish Balay 
6762d61bbb3SSatish Balay   idx = a->j;
6772d61bbb3SSatish Balay   v   = a->a;
67826e093fcSHong Zhang   if (usecprow){
6794eb6d288SHong Zhang     if (zz != yy){
6804eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,mbs*sizeof(PetscScalar));CHKERRQ(ierr);
6814eb6d288SHong Zhang     }
68226e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
68326e093fcSHong Zhang     ii   = a->compressedrow.i;
6847b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
68526e093fcSHong Zhang   } else {
6862d61bbb3SSatish Balay     ii  = a->i;
68726e093fcSHong Zhang     y   = yarray;
68826e093fcSHong Zhang     z   = zarray;
68926e093fcSHong Zhang   }
6902d61bbb3SSatish Balay 
6912d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
6922d61bbb3SSatish Balay     n    = ii[1] - ii[0]; ii++;
69326e093fcSHong Zhang     if (usecprow){
6947b2bb3b9SHong Zhang       z = zarray + ridx[i];
6957b2bb3b9SHong Zhang       y = yarray + ridx[i];
69626e093fcSHong Zhang     }
69726e093fcSHong Zhang     sum = y[0];
6982d61bbb3SSatish Balay     while (n--) sum += *v++ * x[*idx++];
69926e093fcSHong Zhang     z[0] = sum;
70026e093fcSHong Zhang     if (!usecprow){
70126e093fcSHong Zhang       z++; y++;
70226e093fcSHong Zhang     }
7032d61bbb3SSatish Balay   }
7041ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
70526e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
7062e8a6d31SBarry Smith   if (zz != yy) {
70726e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
7082e8a6d31SBarry Smith   }
709dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
7102d61bbb3SSatish Balay   PetscFunctionReturn(0);
7112d61bbb3SSatish Balay }
7122d61bbb3SSatish Balay 
7134a2ae208SSatish Balay #undef __FUNCT__
7144a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_2"
715dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_2(Mat A,Vec xx,Vec yy,Vec zz)
7162d61bbb3SSatish Balay {
7172d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
718dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2;
71926e093fcSHong Zhang   PetscScalar    x1,x2,*yarray,*zarray;
7203f1db9ecSBarry Smith   MatScalar      *v;
721dfbe8321SBarry Smith   PetscErrorCode ierr;
7224eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
72326e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
7242d61bbb3SSatish Balay 
7252d61bbb3SSatish Balay   PetscFunctionBegin;
7261ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
72726e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
7282e8a6d31SBarry Smith   if (zz != yy) {
72926e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
7302e8a6d31SBarry Smith   } else {
73126e093fcSHong Zhang     zarray = yarray;
7322e8a6d31SBarry Smith   }
7332d61bbb3SSatish Balay 
7342d61bbb3SSatish Balay   idx = a->j;
7352d61bbb3SSatish Balay   v   = a->a;
73626e093fcSHong Zhang   if (usecprow){
7374eb6d288SHong Zhang     if (zz != yy){
7384eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,2*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
7394eb6d288SHong Zhang     }
74026e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
74126e093fcSHong Zhang     ii   = a->compressedrow.i;
7427b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
7434eb6d288SHong Zhang     if (zz != yy){
7444eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,a->mbs*sizeof(PetscScalar));CHKERRQ(ierr);
7454eb6d288SHong Zhang     }
74626e093fcSHong Zhang   } else {
7472d61bbb3SSatish Balay     ii  = a->i;
74826e093fcSHong Zhang     y   = yarray;
74926e093fcSHong Zhang     z   = zarray;
75026e093fcSHong Zhang   }
7512d61bbb3SSatish Balay 
7522d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
7532d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
75426e093fcSHong Zhang     if (usecprow){
7557b2bb3b9SHong Zhang       z = zarray + 2*ridx[i];
7567b2bb3b9SHong Zhang       y = yarray + 2*ridx[i];
75726e093fcSHong Zhang     }
7582d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1];
7592d61bbb3SSatish Balay     for (j=0; j<n; j++) {
7602d61bbb3SSatish Balay       xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1];
7612d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[2]*x2;
7622d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[3]*x2;
7632d61bbb3SSatish Balay       v += 4;
7642d61bbb3SSatish Balay     }
7652d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2;
76626e093fcSHong Zhang     if (!usecprow){
7672d61bbb3SSatish Balay       z += 2; y += 2;
7682d61bbb3SSatish Balay     }
76926e093fcSHong Zhang   }
7701ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
77126e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
7722e8a6d31SBarry Smith   if (zz != yy) {
77326e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
7742e8a6d31SBarry Smith   }
775dc0b31edSSatish Balay   ierr = PetscLogFlops(4.0*a->nz);CHKERRQ(ierr);
7762d61bbb3SSatish Balay   PetscFunctionReturn(0);
7772d61bbb3SSatish Balay }
7782d61bbb3SSatish Balay 
7794a2ae208SSatish Balay #undef __FUNCT__
7804a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_3"
781dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_3(Mat A,Vec xx,Vec yy,Vec zz)
7822d61bbb3SSatish Balay {
7832d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
784dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,x1,x2,x3,*yarray,*zarray;
7853f1db9ecSBarry Smith   MatScalar      *v;
786dfbe8321SBarry Smith   PetscErrorCode ierr;
7874eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
78826e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
7892d61bbb3SSatish Balay 
7902d61bbb3SSatish Balay   PetscFunctionBegin;
7911ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
79226e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
7932e8a6d31SBarry Smith   if (zz != yy) {
79426e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
7952e8a6d31SBarry Smith   } else {
79626e093fcSHong Zhang     zarray = yarray;
7972e8a6d31SBarry Smith   }
7982d61bbb3SSatish Balay 
7992d61bbb3SSatish Balay   idx = a->j;
8002d61bbb3SSatish Balay   v   = a->a;
80126e093fcSHong Zhang   if (usecprow){
8024eb6d288SHong Zhang     if (zz != yy){
8034eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,3*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
8044eb6d288SHong Zhang     }
80526e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
80626e093fcSHong Zhang     ii   = a->compressedrow.i;
8077b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
80826e093fcSHong Zhang   } else {
8092d61bbb3SSatish Balay     ii  = a->i;
81026e093fcSHong Zhang     y   = yarray;
81126e093fcSHong Zhang     z   = zarray;
81226e093fcSHong Zhang   }
8132d61bbb3SSatish Balay 
8142d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
8152d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
81626e093fcSHong Zhang     if (usecprow){
8177b2bb3b9SHong Zhang       z = zarray + 3*ridx[i];
8187b2bb3b9SHong Zhang       y = yarray + 3*ridx[i];
81926e093fcSHong Zhang     }
8202d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2];
8212d61bbb3SSatish Balay     for (j=0; j<n; j++) {
8222d61bbb3SSatish Balay       xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
8232d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3;
8242d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3;
8252d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3;
8262d61bbb3SSatish Balay       v += 9;
8272d61bbb3SSatish Balay     }
8282d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3;
82926e093fcSHong Zhang     if (!usecprow){
8302d61bbb3SSatish Balay       z += 3; y += 3;
8312d61bbb3SSatish Balay     }
83226e093fcSHong Zhang   }
8331ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
83426e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
8352e8a6d31SBarry Smith   if (zz != yy) {
83626e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
8372e8a6d31SBarry Smith   }
838dc0b31edSSatish Balay   ierr = PetscLogFlops(18.0*a->nz);CHKERRQ(ierr);
8392d61bbb3SSatish Balay   PetscFunctionReturn(0);
8402d61bbb3SSatish Balay }
8412d61bbb3SSatish Balay 
8424a2ae208SSatish Balay #undef __FUNCT__
8434a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_4"
844dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_4(Mat A,Vec xx,Vec yy,Vec zz)
8452d61bbb3SSatish Balay {
8462d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
847dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*yarray,*zarray;
8483f1db9ecSBarry Smith   MatScalar      *v;
849dfbe8321SBarry Smith   PetscErrorCode ierr;
8504eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
85126e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
8522d61bbb3SSatish Balay 
8532d61bbb3SSatish Balay   PetscFunctionBegin;
8541ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
85526e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
8562e8a6d31SBarry Smith   if (zz != yy) {
85726e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
8582e8a6d31SBarry Smith   } else {
85926e093fcSHong Zhang     zarray = yarray;
8602e8a6d31SBarry Smith   }
8612d61bbb3SSatish Balay 
8622d61bbb3SSatish Balay   idx   = a->j;
8632d61bbb3SSatish Balay   v     = a->a;
86426e093fcSHong Zhang   if (usecprow){
8654eb6d288SHong Zhang     if (zz != yy){
8664eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,4*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
8674eb6d288SHong Zhang     }
86826e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
86926e093fcSHong Zhang     ii   = a->compressedrow.i;
8707b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
87126e093fcSHong Zhang   } else {
8722d61bbb3SSatish Balay     ii  = a->i;
87326e093fcSHong Zhang     y   = yarray;
87426e093fcSHong Zhang     z   = zarray;
87526e093fcSHong Zhang   }
8762d61bbb3SSatish Balay 
8772d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
8782d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
87926e093fcSHong Zhang     if (usecprow){
8807b2bb3b9SHong Zhang       z = zarray + 4*ridx[i];
8817b2bb3b9SHong Zhang       y = yarray + 4*ridx[i];
88226e093fcSHong Zhang     }
8832d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3];
8842d61bbb3SSatish Balay     for (j=0; j<n; j++) {
8852d61bbb3SSatish Balay       xb = x + 4*(*idx++);
8862d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
8872d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3   + v[12]*x4;
8882d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3   + v[13]*x4;
8892d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3  + v[14]*x4;
8902d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3  + v[15]*x4;
8912d61bbb3SSatish Balay       v += 16;
8922d61bbb3SSatish Balay     }
8932d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4;
89426e093fcSHong Zhang     if (!usecprow){
8952d61bbb3SSatish Balay       z += 4; y += 4;
8962d61bbb3SSatish Balay     }
89726e093fcSHong Zhang   }
8981ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
89926e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
9002e8a6d31SBarry Smith   if (zz != yy) {
90126e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
9022e8a6d31SBarry Smith   }
903dc0b31edSSatish Balay   ierr = PetscLogFlops(32.0*a->nz);CHKERRQ(ierr);
9042d61bbb3SSatish Balay   PetscFunctionReturn(0);
9052d61bbb3SSatish Balay }
9062d61bbb3SSatish Balay 
9074a2ae208SSatish Balay #undef __FUNCT__
9084a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_5"
909dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_5(Mat A,Vec xx,Vec yy,Vec zz)
9102d61bbb3SSatish Balay {
9112d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
912dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5;
91326e093fcSHong Zhang   PetscScalar    *yarray,*zarray;
9143f1db9ecSBarry Smith   MatScalar      *v;
915dfbe8321SBarry Smith   PetscErrorCode ierr;
9164eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
91726e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
9182d61bbb3SSatish Balay 
9192d61bbb3SSatish Balay   PetscFunctionBegin;
9201ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
92126e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
9222e8a6d31SBarry Smith   if (zz != yy) {
92326e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
9242e8a6d31SBarry Smith   } else {
92526e093fcSHong Zhang     zarray = yarray;
9262e8a6d31SBarry Smith   }
9272d61bbb3SSatish Balay 
9282d61bbb3SSatish Balay   idx = a->j;
9292d61bbb3SSatish Balay   v   = a->a;
93026e093fcSHong Zhang   if (usecprow){
9314eb6d288SHong Zhang     if (zz != yy){
9324eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,5*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
9334eb6d288SHong Zhang     }
93426e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
93526e093fcSHong Zhang     ii   = a->compressedrow.i;
9367b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
93726e093fcSHong Zhang   } else {
9382d61bbb3SSatish Balay     ii  = a->i;
93926e093fcSHong Zhang     y   = yarray;
94026e093fcSHong Zhang     z   = zarray;
94126e093fcSHong Zhang   }
9422d61bbb3SSatish Balay 
9432d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
9442d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
94526e093fcSHong Zhang     if (usecprow){
9467b2bb3b9SHong Zhang       z = zarray + 5*ridx[i];
9477b2bb3b9SHong Zhang       y = yarray + 5*ridx[i];
94826e093fcSHong Zhang     }
9492d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4];
9502d61bbb3SSatish Balay     for (j=0; j<n; j++) {
9512d61bbb3SSatish Balay       xb = x + 5*(*idx++);
9522d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4];
9532d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3  + v[15]*x4 + v[20]*x5;
9542d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3  + v[16]*x4 + v[21]*x5;
9552d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3  + v[17]*x4 + v[22]*x5;
9562d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3  + v[18]*x4 + v[23]*x5;
9572d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3  + v[19]*x4 + v[24]*x5;
9582d61bbb3SSatish Balay       v += 25;
9592d61bbb3SSatish Balay     }
9602d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5;
96126e093fcSHong Zhang     if (!usecprow){
9622d61bbb3SSatish Balay       z += 5; y += 5;
9632d61bbb3SSatish Balay     }
96426e093fcSHong Zhang   }
9651ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
96626e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
9672e8a6d31SBarry Smith   if (zz != yy) {
96826e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
9692e8a6d31SBarry Smith   }
970dc0b31edSSatish Balay   ierr = PetscLogFlops(50.0*a->nz);CHKERRQ(ierr);
9712d61bbb3SSatish Balay   PetscFunctionReturn(0);
9722d61bbb3SSatish Balay }
9734a2ae208SSatish Balay #undef __FUNCT__
9744a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_6"
975dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_6(Mat A,Vec xx,Vec yy,Vec zz)
97615091d37SBarry Smith {
97715091d37SBarry Smith   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
978dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6;
97926e093fcSHong Zhang   PetscScalar    x1,x2,x3,x4,x5,x6,*yarray,*zarray;
98015091d37SBarry Smith   MatScalar      *v;
981dfbe8321SBarry Smith   PetscErrorCode ierr;
9824eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
98326e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
98415091d37SBarry Smith 
98515091d37SBarry Smith   PetscFunctionBegin;
9861ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
98726e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
98815091d37SBarry Smith   if (zz != yy) {
98926e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
99015091d37SBarry Smith   } else {
99126e093fcSHong Zhang     zarray = yarray;
99215091d37SBarry Smith   }
99315091d37SBarry Smith 
99415091d37SBarry Smith   idx = a->j;
99515091d37SBarry Smith   v   = a->a;
99626e093fcSHong Zhang   if (usecprow){
9974eb6d288SHong Zhang     if (zz != yy){
9984eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,6*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
9994eb6d288SHong Zhang     }
100026e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
100126e093fcSHong Zhang     ii   = a->compressedrow.i;
10027b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
100326e093fcSHong Zhang   } else {
100415091d37SBarry Smith     ii  = a->i;
100526e093fcSHong Zhang     y   = yarray;
100626e093fcSHong Zhang     z   = zarray;
100726e093fcSHong Zhang   }
100815091d37SBarry Smith 
100915091d37SBarry Smith   for (i=0; i<mbs; i++) {
101015091d37SBarry Smith     n  = ii[1] - ii[0]; ii++;
101126e093fcSHong Zhang     if (usecprow){
10127b2bb3b9SHong Zhang       z = zarray + 6*ridx[i];
10137b2bb3b9SHong Zhang       y = yarray + 6*ridx[i];
101426e093fcSHong Zhang     }
101515091d37SBarry Smith     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5];
101615091d37SBarry Smith     for (j=0; j<n; j++) {
10173b95cb0eSSatish Balay       xb = x + 6*(*idx++);
101815091d37SBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5];
101915091d37SBarry Smith       sum1 += v[0]*x1 + v[6]*x2  + v[12]*x3  + v[18]*x4 + v[24]*x5 + v[30]*x6;
102015091d37SBarry Smith       sum2 += v[1]*x1 + v[7]*x2  + v[13]*x3  + v[19]*x4 + v[25]*x5 + v[31]*x6;
102115091d37SBarry Smith       sum3 += v[2]*x1 + v[8]*x2  + v[14]*x3  + v[20]*x4 + v[26]*x5 + v[32]*x6;
102215091d37SBarry Smith       sum4 += v[3]*x1 + v[9]*x2  + v[15]*x3  + v[21]*x4 + v[27]*x5 + v[33]*x6;
102315091d37SBarry Smith       sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3  + v[22]*x4 + v[28]*x5 + v[34]*x6;
102415091d37SBarry Smith       sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3  + v[23]*x4 + v[29]*x5 + v[35]*x6;
102515091d37SBarry Smith       v += 36;
102615091d37SBarry Smith     }
102715091d37SBarry Smith     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6;
102826e093fcSHong Zhang     if (!usecprow){
102915091d37SBarry Smith       z += 6; y += 6;
103015091d37SBarry Smith     }
103126e093fcSHong Zhang   }
10321ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
103326e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
103415091d37SBarry Smith   if (zz != yy) {
103526e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
103615091d37SBarry Smith   }
1037dc0b31edSSatish Balay   ierr = PetscLogFlops(72.0*a->nz);CHKERRQ(ierr);
103815091d37SBarry Smith   PetscFunctionReturn(0);
103915091d37SBarry Smith }
10402d61bbb3SSatish Balay 
10414a2ae208SSatish Balay #undef __FUNCT__
10424a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_7"
1043dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_7(Mat A,Vec xx,Vec yy,Vec zz)
10442d61bbb3SSatish Balay {
10452d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1046dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6,sum7;
104726e093fcSHong Zhang   PetscScalar    x1,x2,x3,x4,x5,x6,x7,*yarray,*zarray;
10483f1db9ecSBarry Smith   MatScalar      *v;
1049dfbe8321SBarry Smith   PetscErrorCode ierr;
10507b2bb3b9SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
105126e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
10522d61bbb3SSatish Balay 
10532d61bbb3SSatish Balay   PetscFunctionBegin;
10541ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
105526e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
10562e8a6d31SBarry Smith   if (zz != yy) {
105726e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
10582e8a6d31SBarry Smith   } else {
105926e093fcSHong Zhang     zarray = yarray;
10602e8a6d31SBarry Smith   }
10612d61bbb3SSatish Balay 
10622d61bbb3SSatish Balay   idx = a->j;
10632d61bbb3SSatish Balay   v   = a->a;
106426e093fcSHong Zhang   if (usecprow){
10654eb6d288SHong Zhang     if (zz != yy){
10664eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,7*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
10674eb6d288SHong Zhang     }
106826e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
106926e093fcSHong Zhang     ii   = a->compressedrow.i;
10707b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
107126e093fcSHong Zhang   } else {
10722d61bbb3SSatish Balay     ii  = a->i;
107326e093fcSHong Zhang     y   = yarray;
107426e093fcSHong Zhang     z   = zarray;
107526e093fcSHong Zhang   }
10762d61bbb3SSatish Balay 
10772d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
10782d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
107926e093fcSHong Zhang     if (usecprow){
10807b2bb3b9SHong Zhang       z = zarray + 7*ridx[i];
10817b2bb3b9SHong Zhang       y = yarray + 7*ridx[i];
108226e093fcSHong Zhang     }
10832d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5]; sum7 = y[6];
10842d61bbb3SSatish Balay     for (j=0; j<n; j++) {
10852d61bbb3SSatish Balay       xb = x + 7*(*idx++);
10862d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
10872d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[7]*x2  + v[14]*x3  + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7;
10882d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[8]*x2  + v[15]*x3  + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7;
10892d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[9]*x2  + v[16]*x3  + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7;
10902d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3  + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7;
10912d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3  + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7;
10922d61bbb3SSatish Balay       sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3  + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7;
10932d61bbb3SSatish Balay       sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3  + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7;
10942d61bbb3SSatish Balay       v += 49;
10952d61bbb3SSatish Balay     }
10962d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
109726e093fcSHong Zhang     if (!usecprow){
10982d61bbb3SSatish Balay       z += 7; y += 7;
10992d61bbb3SSatish Balay     }
110026e093fcSHong Zhang   }
11011ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
110226e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
11032e8a6d31SBarry Smith   if (zz != yy) {
110426e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
11052e8a6d31SBarry Smith   }
1106dc0b31edSSatish Balay   ierr = PetscLogFlops(98.0*a->nz);CHKERRQ(ierr);
11072d61bbb3SSatish Balay   PetscFunctionReturn(0);
11082d61bbb3SSatish Balay }
1109218c64b6SSatish Balay 
11104a2ae208SSatish Balay #undef __FUNCT__
11114a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_N"
1112dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_N(Mat A,Vec xx,Vec yy,Vec zz)
11132d61bbb3SSatish Balay {
11142d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1115bba805e6SBarry Smith   PetscScalar    *x,*z = 0,*xb,*work,*workt,*zarray;
11163f1db9ecSBarry Smith   MatScalar      *v;
11176849ba73SBarry Smith   PetscErrorCode ierr;
1118d0f46423SBarry Smith   PetscInt       mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2;
11197b2bb3b9SHong Zhang   PetscInt       ncols,k,*ridx=PETSC_NULL;
112026e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
1121218c64b6SSatish Balay 
11222d61bbb3SSatish Balay   PetscFunctionBegin;
11236a65c766SHong Zhang   ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr);
11241ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
112526e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
11262d61bbb3SSatish Balay 
11272d61bbb3SSatish Balay   idx = a->j;
11282d61bbb3SSatish Balay   v   = a->a;
112926e093fcSHong Zhang   if (usecprow){
113026e093fcSHong Zhang     mbs    = a->compressedrow.nrows;
113126e093fcSHong Zhang     ii     = a->compressedrow.i;
11327b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
113326e093fcSHong Zhang   } else {
113426e093fcSHong Zhang     mbs = a->mbs;
11352d61bbb3SSatish Balay     ii  = a->i;
113626e093fcSHong Zhang     z   = zarray;
113726e093fcSHong Zhang   }
11382d61bbb3SSatish Balay 
11392d61bbb3SSatish Balay   if (!a->mult_work) {
1140d0f46423SBarry Smith     k    = PetscMax(A->rmap->n,A->cmap->n);
114187828ca2SBarry Smith     ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
11422d61bbb3SSatish Balay   }
11432d61bbb3SSatish Balay   work = a->mult_work;
11442d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
11452d61bbb3SSatish Balay     n     = ii[1] - ii[0]; ii++;
11462d61bbb3SSatish Balay     ncols = n*bs;
11472d61bbb3SSatish Balay     workt = work;
11482d61bbb3SSatish Balay     for (j=0; j<n; j++) {
11492d61bbb3SSatish Balay       xb = x + bs*(*idx++);
11502d61bbb3SSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
11512d61bbb3SSatish Balay       workt += bs;
11522d61bbb3SSatish Balay     }
11537b2bb3b9SHong Zhang     if (usecprow) z = zarray + bs*ridx[i];
11543f1db9ecSBarry Smith     Kernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z);
115571044d3cSBarry Smith     /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DOne,z,&_One); */
11562d61bbb3SSatish Balay     v += n*bs2;
115726e093fcSHong Zhang     if (!usecprow){
11582d61bbb3SSatish Balay       z += bs;
11592d61bbb3SSatish Balay     }
116026e093fcSHong Zhang   }
11611ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
116226e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
1163dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz*bs2);CHKERRQ(ierr);
11642d61bbb3SSatish Balay   PetscFunctionReturn(0);
11652d61bbb3SSatish Balay }
11662d61bbb3SSatish Balay 
11674a2ae208SSatish Balay #undef __FUNCT__
11684a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqBAIJ"
1169dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqBAIJ(Mat A,Vec xx,Vec zz)
11702d61bbb3SSatish Balay {
11713447b6efSHong Zhang   PetscScalar    zero = 0.0;
11726849ba73SBarry Smith   PetscErrorCode ierr;
11732d61bbb3SSatish Balay 
11742d61bbb3SSatish Balay   PetscFunctionBegin;
11752dcb1b2aSMatthew Knepley   ierr = VecSet(zz,zero);CHKERRQ(ierr);
11763447b6efSHong Zhang   ierr = MatMultTransposeAdd_SeqBAIJ(A,xx,zz,zz);CHKERRQ(ierr);
11772d61bbb3SSatish Balay   PetscFunctionReturn(0);
11782d61bbb3SSatish Balay }
11792d61bbb3SSatish Balay 
11804a2ae208SSatish Balay #undef __FUNCT__
11814a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqBAIJ"
1182dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqBAIJ(Mat A,Vec xx,Vec yy,Vec zz)
11832d61bbb3SSatish Balay 
11842d61bbb3SSatish Balay {
11852d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
1186dcf5cc72SBarry Smith   PetscScalar       *zb,*x,*z,*xb = 0,x1,x2,x3,x4,x5;
11873f1db9ecSBarry Smith   MatScalar         *v;
11886849ba73SBarry Smith   PetscErrorCode    ierr;
1189d0f46423SBarry Smith   PetscInt          mbs,i,*idx,*ii,rval,bs=A->rmap->bs,j,n,bs2=a->bs2,*ib,*ridx=PETSC_NULL;
11903447b6efSHong Zhang   Mat_CompressedRow cprow = a->compressedrow;
11913447b6efSHong Zhang   PetscTruth        usecprow=cprow.use;
11922d61bbb3SSatish Balay 
11932d61bbb3SSatish Balay   PetscFunctionBegin;
11946a65c766SHong Zhang   if (yy != zz) { ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr); }
11953447b6efSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
11963447b6efSHong Zhang   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
11972d61bbb3SSatish Balay 
11982d61bbb3SSatish Balay   idx = a->j;
11992d61bbb3SSatish Balay   v   = a->a;
12003447b6efSHong Zhang   if (usecprow){
12013447b6efSHong Zhang     mbs  = cprow.nrows;
12023447b6efSHong Zhang     ii   = cprow.i;
12037b2bb3b9SHong Zhang     ridx = cprow.rindex;
12043447b6efSHong Zhang   } else {
12053447b6efSHong Zhang     mbs=a->mbs;
12062d61bbb3SSatish Balay     ii = a->i;
1207f1af5d2fSBarry Smith     xb = x;
12083447b6efSHong Zhang   }
12092d61bbb3SSatish Balay 
12102d61bbb3SSatish Balay   switch (bs) {
12112d61bbb3SSatish Balay   case 1:
12122d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12137b2bb3b9SHong Zhang       if (usecprow) xb = x + ridx[i];
1214f1af5d2fSBarry Smith       x1 = xb[0];
12153447b6efSHong Zhang       ib = idx + ii[0];
12163447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12172d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12182d61bbb3SSatish Balay         rval    = ib[j];
1219f1af5d2fSBarry Smith         z[rval] += *v * x1;
1220f1af5d2fSBarry Smith         v++;
12212d61bbb3SSatish Balay       }
12223447b6efSHong Zhang       if (!usecprow) xb++;
12232d61bbb3SSatish Balay     }
12242d61bbb3SSatish Balay     break;
12252d61bbb3SSatish Balay   case 2:
12262d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12277b2bb3b9SHong Zhang       if (usecprow) xb = x + 2*ridx[i];
1228f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1];
12293447b6efSHong Zhang       ib = idx + ii[0];
12303447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12312d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12322d61bbb3SSatish Balay         rval      = ib[j]*2;
12332d61bbb3SSatish Balay         z[rval++] += v[0]*x1 + v[1]*x2;
12342d61bbb3SSatish Balay         z[rval++] += v[2]*x1 + v[3]*x2;
12352d61bbb3SSatish Balay         v  += 4;
12362d61bbb3SSatish Balay       }
12373447b6efSHong Zhang       if (!usecprow) xb += 2;
12382d61bbb3SSatish Balay     }
12392d61bbb3SSatish Balay     break;
12402d61bbb3SSatish Balay   case 3:
12412d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12427b2bb3b9SHong Zhang       if (usecprow) xb = x + 3*ridx[i];
1243f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
12443447b6efSHong Zhang       ib = idx + ii[0];
12453447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12462d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12472d61bbb3SSatish Balay         rval      = ib[j]*3;
12482d61bbb3SSatish Balay         z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3;
12492d61bbb3SSatish Balay         z[rval++] += v[3]*x1 + v[4]*x2 + v[5]*x3;
12502d61bbb3SSatish Balay         z[rval++] += v[6]*x1 + v[7]*x2 + v[8]*x3;
12512d61bbb3SSatish Balay         v  += 9;
12522d61bbb3SSatish Balay       }
12533447b6efSHong Zhang       if (!usecprow) xb += 3;
12542d61bbb3SSatish Balay     }
12552d61bbb3SSatish Balay     break;
12562d61bbb3SSatish Balay   case 4:
12572d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12587b2bb3b9SHong Zhang       if (usecprow) xb = x + 4*ridx[i];
1259f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
12603447b6efSHong Zhang       ib = idx + ii[0];
12613447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12622d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12632d61bbb3SSatish Balay         rval      = ib[j]*4;
12642d61bbb3SSatish Balay         z[rval++] +=  v[0]*x1 +  v[1]*x2 +  v[2]*x3 +  v[3]*x4;
12652d61bbb3SSatish Balay         z[rval++] +=  v[4]*x1 +  v[5]*x2 +  v[6]*x3 +  v[7]*x4;
12662d61bbb3SSatish Balay         z[rval++] +=  v[8]*x1 +  v[9]*x2 + v[10]*x3 + v[11]*x4;
12672d61bbb3SSatish Balay         z[rval++] += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4;
12682d61bbb3SSatish Balay         v  += 16;
12692d61bbb3SSatish Balay       }
12703447b6efSHong Zhang       if (!usecprow) xb += 4;
12712d61bbb3SSatish Balay     }
12722d61bbb3SSatish Balay     break;
12732d61bbb3SSatish Balay   case 5:
12742d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
12757b2bb3b9SHong Zhang       if (usecprow) xb = x + 5*ridx[i];
1276f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
12772d61bbb3SSatish Balay       x4 = xb[3]; x5 = xb[4];
12783447b6efSHong Zhang       ib = idx + ii[0];
12793447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
12802d61bbb3SSatish Balay       for (j=0; j<n; j++) {
12812d61bbb3SSatish Balay         rval      = ib[j]*5;
12822d61bbb3SSatish Balay         z[rval++] +=  v[0]*x1 +  v[1]*x2 +  v[2]*x3 +  v[3]*x4 +  v[4]*x5;
12832d61bbb3SSatish Balay         z[rval++] +=  v[5]*x1 +  v[6]*x2 +  v[7]*x3 +  v[8]*x4 +  v[9]*x5;
12842d61bbb3SSatish Balay         z[rval++] += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4 + v[14]*x5;
12852d61bbb3SSatish Balay         z[rval++] += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4 + v[19]*x5;
12862d61bbb3SSatish Balay         z[rval++] += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4 + v[24]*x5;
12872d61bbb3SSatish Balay         v  += 25;
12882d61bbb3SSatish Balay       }
12893447b6efSHong Zhang       if (!usecprow) xb += 5;
12902d61bbb3SSatish Balay     }
12912d61bbb3SSatish Balay     break;
1292f1af5d2fSBarry Smith   default: {      /* block sizes larger then 5 by 5 are handled by BLAS */
1293690b6cddSBarry Smith       PetscInt     ncols,k;
12943447b6efSHong Zhang       PetscScalar  *work,*workt,*xtmp;
12953f1db9ecSBarry Smith 
12962d61bbb3SSatish Balay       if (!a->mult_work) {
1297d0f46423SBarry Smith         k = PetscMax(A->rmap->n,A->cmap->n);
129887828ca2SBarry Smith         ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
12992d61bbb3SSatish Balay       }
13002d61bbb3SSatish Balay       work = a->mult_work;
13013447b6efSHong Zhang       xtmp = x;
13022d61bbb3SSatish Balay       for (i=0; i<mbs; i++) {
13032d61bbb3SSatish Balay         n     = ii[1] - ii[0]; ii++;
13042d61bbb3SSatish Balay         ncols = n*bs;
130587828ca2SBarry Smith         ierr  = PetscMemzero(work,ncols*sizeof(PetscScalar));CHKERRQ(ierr);
13063447b6efSHong Zhang         if (usecprow) {
13077b2bb3b9SHong Zhang           xtmp = x + bs*ridx[i];
13083447b6efSHong Zhang         }
13093447b6efSHong Zhang         Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,xtmp,v,work);
131071044d3cSBarry Smith         /* BLASgemv_("T",&bs,&ncols,&_DOne,v,&bs,xtmp,&_One,&_DOne,work,&_One); */
13112d61bbb3SSatish Balay         v += n*bs2;
13123447b6efSHong Zhang         if (!usecprow) xtmp += bs;
13132d61bbb3SSatish Balay         workt = work;
13142d61bbb3SSatish Balay         for (j=0; j<n; j++) {
13152d61bbb3SSatish Balay           zb = z + bs*(*idx++);
13162d61bbb3SSatish Balay           for (k=0; k<bs; k++) zb[k] += workt[k] ;
13172d61bbb3SSatish Balay           workt += bs;
13182d61bbb3SSatish Balay         }
13192d61bbb3SSatish Balay       }
13202d61bbb3SSatish Balay     }
13212d61bbb3SSatish Balay   }
13223447b6efSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13233447b6efSHong Zhang   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
1324dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz*a->bs2);CHKERRQ(ierr);
13252d61bbb3SSatish Balay   PetscFunctionReturn(0);
13262d61bbb3SSatish Balay }
13272d61bbb3SSatish Balay 
13284a2ae208SSatish Balay #undef __FUNCT__
13294a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqBAIJ"
1330f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqBAIJ(Mat inA,PetscScalar alpha)
13312d61bbb3SSatish Balay {
13322d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)inA->data;
1333690b6cddSBarry Smith   PetscInt       totalnz = a->bs2*a->nz;
1334f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
1335efee365bSSatish Balay   PetscErrorCode ierr;
13360805154bSBarry Smith   PetscBLASInt   one = 1,tnz = PetscBLASIntCast(totalnz);
13372d61bbb3SSatish Balay 
13382d61bbb3SSatish Balay   PetscFunctionBegin;
1339f4df32b1SMatthew Knepley   BLASscal_(&tnz,&oalpha,a->a,&one);
1340efee365bSSatish Balay   ierr = PetscLogFlops(totalnz);CHKERRQ(ierr);
13412d61bbb3SSatish Balay   PetscFunctionReturn(0);
13422d61bbb3SSatish Balay }
13432d61bbb3SSatish Balay 
13444a2ae208SSatish Balay #undef __FUNCT__
13454a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqBAIJ"
1346dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqBAIJ(Mat A,NormType type,PetscReal *norm)
13472d61bbb3SSatish Balay {
13488a62d963SHong Zhang   PetscErrorCode ierr;
13492d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
13503f1db9ecSBarry Smith   MatScalar      *v = a->a;
1351329f5518SBarry Smith   PetscReal      sum = 0.0;
1352d0f46423SBarry Smith   PetscInt       i,j,k,bs=A->rmap->bs,nz=a->nz,bs2=a->bs2,k1;
13532d61bbb3SSatish Balay 
13542d61bbb3SSatish Balay   PetscFunctionBegin;
13552d61bbb3SSatish Balay   if (type == NORM_FROBENIUS) {
13562d61bbb3SSatish Balay     for (i=0; i< bs2*nz; i++) {
1357aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
1358329f5518SBarry Smith       sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
13592d61bbb3SSatish Balay #else
13602d61bbb3SSatish Balay       sum += (*v)*(*v); v++;
13612d61bbb3SSatish Balay #endif
13622d61bbb3SSatish Balay     }
13632d61bbb3SSatish Balay     *norm = sqrt(sum);
13648a62d963SHong Zhang   } else if (type == NORM_1) { /* maximum column sum */
13658a62d963SHong Zhang     PetscReal *tmp;
13668a62d963SHong Zhang     PetscInt  *bcol = a->j;
1367d0f46423SBarry Smith     ierr = PetscMalloc((A->cmap->n+1)*sizeof(PetscReal),&tmp);CHKERRQ(ierr);
1368d0f46423SBarry Smith     ierr = PetscMemzero(tmp,A->cmap->n*sizeof(PetscReal));CHKERRQ(ierr);
13698a62d963SHong Zhang     for (i=0; i<nz; i++){
13708a62d963SHong Zhang       for (j=0; j<bs; j++){
13718a62d963SHong Zhang         k1 = bs*(*bcol) + j; /* column index */
13728a62d963SHong Zhang         for (k=0; k<bs; k++){
13738a62d963SHong Zhang           tmp[k1] += PetscAbsScalar(*v); v++;
13748a62d963SHong Zhang         }
13758a62d963SHong Zhang       }
13768a62d963SHong Zhang       bcol++;
13778a62d963SHong Zhang     }
13788a62d963SHong Zhang     *norm = 0.0;
1379d0f46423SBarry Smith     for (j=0; j<A->cmap->n; j++) {
13808a62d963SHong Zhang       if (tmp[j] > *norm) *norm = tmp[j];
13818a62d963SHong Zhang     }
13828a62d963SHong Zhang     ierr = PetscFree(tmp);CHKERRQ(ierr);
1383596552b5SBarry Smith   } else if (type == NORM_INFINITY) { /* maximum row sum */
1384596552b5SBarry Smith     *norm = 0.0;
1385596552b5SBarry Smith     for (k=0; k<bs; k++) {
138674f84c7bSSatish Balay       for (j=0; j<a->mbs; j++) {
1387596552b5SBarry Smith         v = a->a + bs2*a->i[j] + k;
1388596552b5SBarry Smith         sum = 0.0;
1389596552b5SBarry Smith         for (i=0; i<a->i[j+1]-a->i[j]; i++) {
13900e90e235SBarry Smith           for (k1=0; k1<bs; k1++){
1391596552b5SBarry Smith             sum += PetscAbsScalar(*v);
1392596552b5SBarry Smith             v   += bs;
13932d61bbb3SSatish Balay           }
13940e90e235SBarry Smith         }
1395596552b5SBarry Smith         if (sum > *norm) *norm = sum;
1396596552b5SBarry Smith       }
1397596552b5SBarry Smith     }
1398596552b5SBarry Smith   } else {
139929bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"No support for this norm yet");
14002d61bbb3SSatish Balay   }
14012d61bbb3SSatish Balay   PetscFunctionReturn(0);
14022d61bbb3SSatish Balay }
14032d61bbb3SSatish Balay 
14042d61bbb3SSatish Balay 
14054a2ae208SSatish Balay #undef __FUNCT__
14064a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqBAIJ"
1407dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqBAIJ(Mat A,Mat B,PetscTruth* flg)
14082d61bbb3SSatish Balay {
14092d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ *)A->data,*b = (Mat_SeqBAIJ *)B->data;
1410dfbe8321SBarry Smith   PetscErrorCode ierr;
14112d61bbb3SSatish Balay 
14122d61bbb3SSatish Balay   PetscFunctionBegin;
14132d61bbb3SSatish Balay   /* If the  matrix/block dimensions are not equal, or no of nonzeros or shift */
1414d0f46423SBarry Smith   if ((A->rmap->N != B->rmap->N) || (A->cmap->n != B->cmap->n) || (A->rmap->bs != B->rmap->bs)|| (a->nz != b->nz)) {
1415273d9f13SBarry Smith     *flg = PETSC_FALSE;
1416273d9f13SBarry Smith     PetscFunctionReturn(0);
14172d61bbb3SSatish Balay   }
14182d61bbb3SSatish Balay 
14192d61bbb3SSatish Balay   /* if the a->i are the same */
1420690b6cddSBarry Smith   ierr = PetscMemcmp(a->i,b->i,(a->mbs+1)*sizeof(PetscInt),flg);CHKERRQ(ierr);
1421abc0a331SBarry Smith   if (!*flg) {
14220f5bd95cSBarry Smith     PetscFunctionReturn(0);
14232d61bbb3SSatish Balay   }
14242d61bbb3SSatish Balay 
14252d61bbb3SSatish Balay   /* if a->j are the same */
1426690b6cddSBarry Smith   ierr = PetscMemcmp(a->j,b->j,(a->nz)*sizeof(PetscInt),flg);CHKERRQ(ierr);
1427abc0a331SBarry Smith   if (!*flg) {
14280f5bd95cSBarry Smith     PetscFunctionReturn(0);
14292d61bbb3SSatish Balay   }
14302d61bbb3SSatish Balay   /* if a->a are the same */
1431d0f46423SBarry Smith   ierr = PetscMemcmp(a->a,b->a,(a->nz)*(A->rmap->bs)*(B->rmap->bs)*sizeof(PetscScalar),flg);CHKERRQ(ierr);
14322d61bbb3SSatish Balay   PetscFunctionReturn(0);
14332d61bbb3SSatish Balay 
14342d61bbb3SSatish Balay }
14352d61bbb3SSatish Balay 
14364a2ae208SSatish Balay #undef __FUNCT__
14374a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqBAIJ"
1438dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqBAIJ(Mat A,Vec v)
14392d61bbb3SSatish Balay {
14402d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1441dfbe8321SBarry Smith   PetscErrorCode ierr;
1442690b6cddSBarry Smith   PetscInt       i,j,k,n,row,bs,*ai,*aj,ambs,bs2;
144387828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
14443f1db9ecSBarry Smith   MatScalar      *aa,*aa_j;
14452d61bbb3SSatish Balay 
14462d61bbb3SSatish Balay   PetscFunctionBegin;
144729bbc08cSBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1448d0f46423SBarry Smith   bs   = A->rmap->bs;
14492d61bbb3SSatish Balay   aa   = a->a;
14502d61bbb3SSatish Balay   ai   = a->i;
14512d61bbb3SSatish Balay   aj   = a->j;
14522d61bbb3SSatish Balay   ambs = a->mbs;
14532d61bbb3SSatish Balay   bs2  = a->bs2;
14542d61bbb3SSatish Balay 
14552dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
14561ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1457e1311b90SBarry Smith   ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr);
1458d0f46423SBarry Smith   if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
14592d61bbb3SSatish Balay   for (i=0; i<ambs; i++) {
14602d61bbb3SSatish Balay     for (j=ai[i]; j<ai[i+1]; j++) {
14612d61bbb3SSatish Balay       if (aj[j] == i) {
14622d61bbb3SSatish Balay         row  = i*bs;
14632d61bbb3SSatish Balay         aa_j = aa+j*bs2;
14642d61bbb3SSatish Balay         for (k=0; k<bs2; k+=(bs+1),row++) x[row] = aa_j[k];
14652d61bbb3SSatish Balay         break;
14662d61bbb3SSatish Balay       }
14672d61bbb3SSatish Balay     }
14682d61bbb3SSatish Balay   }
14691ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
14702d61bbb3SSatish Balay   PetscFunctionReturn(0);
14712d61bbb3SSatish Balay }
14722d61bbb3SSatish Balay 
14734a2ae208SSatish Balay #undef __FUNCT__
14744a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqBAIJ"
1475dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqBAIJ(Mat A,Vec ll,Vec rr)
14762d61bbb3SSatish Balay {
14772d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
147887828ca2SBarry Smith   PetscScalar    *l,*r,x,*li,*ri;
14793f1db9ecSBarry Smith   MatScalar      *aa,*v;
1480dfbe8321SBarry Smith   PetscErrorCode ierr;
1481690b6cddSBarry Smith   PetscInt       i,j,k,lm,rn,M,m,n,*ai,*aj,mbs,tmp,bs,bs2;
14822d61bbb3SSatish Balay 
14832d61bbb3SSatish Balay   PetscFunctionBegin;
14842d61bbb3SSatish Balay   ai  = a->i;
14852d61bbb3SSatish Balay   aj  = a->j;
14862d61bbb3SSatish Balay   aa  = a->a;
1487d0f46423SBarry Smith   m   = A->rmap->n;
1488d0f46423SBarry Smith   n   = A->cmap->n;
1489d0f46423SBarry Smith   bs  = A->rmap->bs;
14902d61bbb3SSatish Balay   mbs = a->mbs;
14912d61bbb3SSatish Balay   bs2 = a->bs2;
14922d61bbb3SSatish Balay   if (ll) {
14931ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
1494e1311b90SBarry Smith     ierr = VecGetLocalSize(ll,&lm);CHKERRQ(ierr);
149529bbc08cSBarry Smith     if (lm != m) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length");
14962d61bbb3SSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
14972d61bbb3SSatish Balay       M  = ai[i+1] - ai[i];
14982d61bbb3SSatish Balay       li = l + i*bs;
14992d61bbb3SSatish Balay       v  = aa + bs2*ai[i];
15002d61bbb3SSatish Balay       for (j=0; j<M; j++) { /* for each block */
15012d61bbb3SSatish Balay         for (k=0; k<bs2; k++) {
15022d61bbb3SSatish Balay           (*v++) *= li[k%bs];
15032d61bbb3SSatish Balay         }
15042d61bbb3SSatish Balay       }
15052d61bbb3SSatish Balay     }
15061ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
1507efee365bSSatish Balay     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
15082d61bbb3SSatish Balay   }
15092d61bbb3SSatish Balay 
15102d61bbb3SSatish Balay   if (rr) {
15111ebc52fbSHong Zhang     ierr = VecGetArray(rr,&r);CHKERRQ(ierr);
1512e1311b90SBarry Smith     ierr = VecGetLocalSize(rr,&rn);CHKERRQ(ierr);
151329bbc08cSBarry Smith     if (rn != n) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length");
15142d61bbb3SSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
15152d61bbb3SSatish Balay       M  = ai[i+1] - ai[i];
15162d61bbb3SSatish Balay       v  = aa + bs2*ai[i];
15172d61bbb3SSatish Balay       for (j=0; j<M; j++) { /* for each block */
15182d61bbb3SSatish Balay         ri = r + bs*aj[ai[i]+j];
15192d61bbb3SSatish Balay         for (k=0; k<bs; k++) {
15202d61bbb3SSatish Balay           x = ri[k];
15212d61bbb3SSatish Balay           for (tmp=0; tmp<bs; tmp++) (*v++) *= x;
15222d61bbb3SSatish Balay         }
15232d61bbb3SSatish Balay       }
15242d61bbb3SSatish Balay     }
15251ebc52fbSHong Zhang     ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr);
1526efee365bSSatish Balay     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
15272d61bbb3SSatish Balay   }
15282d61bbb3SSatish Balay   PetscFunctionReturn(0);
15292d61bbb3SSatish Balay }
15302d61bbb3SSatish Balay 
15312d61bbb3SSatish Balay 
15324a2ae208SSatish Balay #undef __FUNCT__
15334a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqBAIJ"
1534dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqBAIJ(Mat A,MatInfoType flag,MatInfo *info)
15352d61bbb3SSatish Balay {
15362d61bbb3SSatish Balay   Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data;
15372d61bbb3SSatish Balay 
15382d61bbb3SSatish Balay   PetscFunctionBegin;
15392d61bbb3SSatish Balay   info->block_size     = a->bs2;
15402d61bbb3SSatish Balay   info->nz_allocated   = a->maxnz;
15412d61bbb3SSatish Balay   info->nz_used        = a->bs2*a->nz;
15422d61bbb3SSatish Balay   info->nz_unneeded    = (double)(info->nz_allocated - info->nz_used);
15432d61bbb3SSatish Balay   info->assemblies   = A->num_ass;
15442d61bbb3SSatish Balay   info->mallocs      = a->reallocs;
15457adad957SLisandro Dalcin   info->memory       = ((PetscObject)A)->mem;
15462d61bbb3SSatish Balay   if (A->factor) {
15472d61bbb3SSatish Balay     info->fill_ratio_given  = A->info.fill_ratio_given;
15482d61bbb3SSatish Balay     info->fill_ratio_needed = A->info.fill_ratio_needed;
15492d61bbb3SSatish Balay     info->factor_mallocs    = A->info.factor_mallocs;
15502d61bbb3SSatish Balay   } else {
15512d61bbb3SSatish Balay     info->fill_ratio_given  = 0;
15522d61bbb3SSatish Balay     info->fill_ratio_needed = 0;
15532d61bbb3SSatish Balay     info->factor_mallocs    = 0;
15542d61bbb3SSatish Balay   }
15552d61bbb3SSatish Balay   PetscFunctionReturn(0);
15562d61bbb3SSatish Balay }
15572d61bbb3SSatish Balay 
15582d61bbb3SSatish Balay 
15594a2ae208SSatish Balay #undef __FUNCT__
15604a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqBAIJ"
1561dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqBAIJ(Mat A)
15622d61bbb3SSatish Balay {
15632d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1564dfbe8321SBarry Smith   PetscErrorCode ierr;
15652d61bbb3SSatish Balay 
15662d61bbb3SSatish Balay   PetscFunctionBegin;
1567549d3d68SSatish Balay   ierr = PetscMemzero(a->a,a->bs2*a->i[a->mbs]*sizeof(MatScalar));CHKERRQ(ierr);
15682d61bbb3SSatish Balay   PetscFunctionReturn(0);
15692d61bbb3SSatish Balay }
1570