xref: /petsc/src/mat/impls/baij/seq/baij2.c (revision 8ab949d8a86daa69f82db2e3a2e7f07f1f4d47d0)
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 */
180fca92195SBarry Smith   ierr = PetscMalloc2(a->mbs,PetscInt,&vary,a->mbs,PetscInt,&iary);CHKERRQ(ierr);
181fca92195SBarry Smith   ierr = PetscMemzero(vary,a->mbs*sizeof(PetscInt));CHKERRQ(ierr);
182218c64b6SSatish Balay   for (i=0; i<nrows; i++) vary[irow[i]/bs]++;
183218c64b6SSatish Balay   count = 0;
184218c64b6SSatish Balay   for (i=0; i<a->mbs; i++) {
185634064b4SBarry Smith     if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_ERR_ARG_SIZ,"Index set does not match blocks");
186218c64b6SSatish Balay     if (vary[i]==bs) iary[count++] = i;
187218c64b6SSatish Balay   }
188029af93fSBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is1);CHKERRQ(ierr);
189218c64b6SSatish Balay 
190690b6cddSBarry Smith   ierr = PetscMemzero(vary,(a->mbs)*sizeof(PetscInt));CHKERRQ(ierr);
191218c64b6SSatish Balay   for (i=0; i<ncols; i++) vary[icol[i]/bs]++;
192218c64b6SSatish Balay   count = 0;
193218c64b6SSatish Balay   for (i=0; i<a->mbs; i++) {
194634064b4SBarry Smith     if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_ERR_PLIB,"Internal error in PETSc");
195218c64b6SSatish Balay     if (vary[i]==bs) iary[count++] = i;
196218c64b6SSatish Balay   }
197029af93fSBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is2);CHKERRQ(ierr);
198218c64b6SSatish Balay   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
199218c64b6SSatish Balay   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
200fca92195SBarry Smith   ierr = PetscFree2(vary,iary);CHKERRQ(ierr);
201218c64b6SSatish Balay 
2024aa3045dSJed Brown   ierr = MatGetSubMatrix_SeqBAIJ_Private(A,is1,is2,scall,B);CHKERRQ(ierr);
203218c64b6SSatish Balay   ISDestroy(is1);
204218c64b6SSatish Balay   ISDestroy(is2);
2053a40ed3dSBarry Smith   PetscFunctionReturn(0);
206218c64b6SSatish Balay }
207218c64b6SSatish Balay 
2084a2ae208SSatish Balay #undef __FUNCT__
2094a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqBAIJ"
210690b6cddSBarry Smith PetscErrorCode MatGetSubMatrices_SeqBAIJ(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[])
211736121d4SSatish Balay {
2126849ba73SBarry Smith   PetscErrorCode ierr;
213690b6cddSBarry Smith   PetscInt       i;
214736121d4SSatish Balay 
2153a40ed3dSBarry Smith   PetscFunctionBegin;
216736121d4SSatish Balay   if (scall == MAT_INITIAL_MATRIX) {
21782502324SSatish Balay     ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr);
218736121d4SSatish Balay   }
219736121d4SSatish Balay 
220736121d4SSatish Balay   for (i=0; i<n; i++) {
2214aa3045dSJed Brown     ierr = MatGetSubMatrix_SeqBAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr);
222736121d4SSatish Balay   }
2233a40ed3dSBarry Smith   PetscFunctionReturn(0);
224736121d4SSatish Balay }
225218c64b6SSatish Balay 
226218c64b6SSatish Balay 
2272d61bbb3SSatish Balay /* -------------------------------------------------------*/
2282d61bbb3SSatish Balay /* Should check that shapes of vectors and matrices match */
2292d61bbb3SSatish Balay /* -------------------------------------------------------*/
230d9eff348SSatish Balay #include "petscblaslapack.h"
2312d61bbb3SSatish Balay 
2324a2ae208SSatish Balay #undef __FUNCT__
2334a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_1"
234dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_1(Mat A,Vec xx,Vec zz)
2352d61bbb3SSatish Balay {
2362d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
237d9fead3dSBarry Smith   PetscScalar       *z,sum;
238d9fead3dSBarry Smith   const PetscScalar *x;
239d9fead3dSBarry Smith   const MatScalar   *v;
2406849ba73SBarry Smith   PetscErrorCode    ierr;
2412162cab8SBarry Smith   PetscInt          mbs,i,n,nonzerorow=0;
2422162cab8SBarry Smith   const PetscInt    *idx,*ii,*ridx=PETSC_NULL;
24326e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
2442d61bbb3SSatish Balay 
2452d61bbb3SSatish Balay   PetscFunctionBegin;
246d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
2471ebc52fbSHong Zhang   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
2482d61bbb3SSatish Balay 
24926e093fcSHong Zhang   if (usecprow){
25026e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
25126e093fcSHong Zhang     ii   = a->compressedrow.i;
2527b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
253a1c3900fSBarry Smith     ierr = PetscMemzero(z,mbs*sizeof(PetscScalar));CHKERRQ(ierr);
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++) {
260ee54c7eeSHong Zhang     n    = ii[1] - ii[0];
261ee54c7eeSHong Zhang     v    = a->a + ii[0];
262ee54c7eeSHong Zhang     idx  = a->j + ii[0];
263ee54c7eeSHong Zhang     ii++;
2642d61bbb3SSatish Balay     sum  = 0.0;
2652162cab8SBarry Smith     PetscSparseDensePlusDot(sum,x,v,idx,n);
26626e093fcSHong Zhang     if (usecprow){
2677b2bb3b9SHong Zhang       z[ridx[i]] = sum;
26826e093fcSHong Zhang     } else {
269122f12eaSBarry Smith       nonzerorow += (n>0);
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;
442766f9fbaSBarry Smith   const PetscInt    *idx,*ii,*ridx=PETSC_NULL;
443766f9fbaSBarry Smith   PetscInt          mbs,i,j,n,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 }
541*8ab949d8SShri Abhyankar 
5424a2ae208SSatish Balay #undef __FUNCT__
5434a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_7"
544dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_7(Mat A,Vec xx,Vec zz)
5452d61bbb3SSatish Balay {
5462d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
547d9fead3dSBarry Smith   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7;
548d9fead3dSBarry Smith   const PetscScalar *x,*xb;
54926e093fcSHong Zhang   PetscScalar       x1,x2,x3,x4,x5,x6,x7,*zarray;
550d9fead3dSBarry Smith   const MatScalar   *v;
551dfbe8321SBarry Smith   PetscErrorCode    ierr;
55298c9bda7SSatish Balay   PetscInt          mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0;
55326e093fcSHong Zhang   PetscTruth        usecprow=a->compressedrow.use;
5542d61bbb3SSatish Balay 
555433994e6SBarry Smith   PetscFunctionBegin;
556d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
55726e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
5582d61bbb3SSatish Balay 
5592d61bbb3SSatish Balay   idx = a->j;
5602d61bbb3SSatish Balay   v   = a->a;
56126e093fcSHong Zhang   if (usecprow){
56226e093fcSHong Zhang     mbs    = a->compressedrow.nrows;
56326e093fcSHong Zhang     ii     = a->compressedrow.i;
5647b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
56526e093fcSHong Zhang   } else {
56626e093fcSHong Zhang     mbs = a->mbs;
5672d61bbb3SSatish Balay     ii  = a->i;
56826e093fcSHong Zhang     z   = zarray;
56926e093fcSHong Zhang   }
5702d61bbb3SSatish Balay 
5712d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
5722d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
5732d61bbb3SSatish Balay     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0;
57498c9bda7SSatish Balay     nonzerorow += (n>0);
5752d61bbb3SSatish Balay     for (j=0; j<n; j++) {
5762d61bbb3SSatish Balay       xb = x + 7*(*idx++);
5772d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
5782d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[7]*x2  + v[14]*x3  + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7;
5792d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[8]*x2  + v[15]*x3  + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7;
5802d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[9]*x2  + v[16]*x3  + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7;
5812d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3  + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7;
5822d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3  + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7;
5832d61bbb3SSatish Balay       sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3  + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7;
5842d61bbb3SSatish Balay       sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3  + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7;
5852d61bbb3SSatish Balay       v += 49;
5862d61bbb3SSatish Balay     }
5877b2bb3b9SHong Zhang     if (usecprow) z = zarray + 7*ridx[i];
5882d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
58926e093fcSHong Zhang     if (!usecprow) z += 7;
5902d61bbb3SSatish Balay   }
5912d61bbb3SSatish Balay 
592d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
59326e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
594dc0b31edSSatish Balay   ierr = PetscLogFlops(98.0*a->nz - 7.0*nonzerorow);CHKERRQ(ierr);
5952d61bbb3SSatish Balay   PetscFunctionReturn(0);
5962d61bbb3SSatish Balay }
5972d61bbb3SSatish Balay 
598*8ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15 version 1: Columns in the block are accessed one at a time */
599*8ab949d8SShri Abhyankar 
600*8ab949d8SShri Abhyankar #undef __FUNCT__
601*8ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver1"
602*8ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver1(Mat A,Vec xx,Vec zz)
603*8ab949d8SShri Abhyankar {
604*8ab949d8SShri Abhyankar   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
605*8ab949d8SShri Abhyankar   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15;
606*8ab949d8SShri Abhyankar   const PetscScalar *x,*xb;
607*8ab949d8SShri Abhyankar   PetscScalar        x1,x2,x3,x4,x5,x6,x7,x8,x9,x10,x11,x12,x13,x14,x15,*zarray;
608*8ab949d8SShri Abhyankar   const MatScalar   *v;
609*8ab949d8SShri Abhyankar   PetscErrorCode    ierr;
610*8ab949d8SShri Abhyankar   const PetscInt    *ii,*ij=a->j,*idx;
611*8ab949d8SShri Abhyankar   PetscInt          mbs,i,j,k,n,*ridx=PETSC_NULL,nonzerorow=0;
612*8ab949d8SShri Abhyankar   PetscTruth        usecprow=a->compressedrow.use;
613*8ab949d8SShri Abhyankar 
614*8ab949d8SShri Abhyankar   PetscFunctionBegin;
615*8ab949d8SShri Abhyankar   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
616*8ab949d8SShri Abhyankar   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
617*8ab949d8SShri Abhyankar 
618*8ab949d8SShri Abhyankar   v   = a->a;
619*8ab949d8SShri Abhyankar   if (usecprow){
620*8ab949d8SShri Abhyankar     mbs    = a->compressedrow.nrows;
621*8ab949d8SShri Abhyankar     ii     = a->compressedrow.i;
622*8ab949d8SShri Abhyankar     ridx = a->compressedrow.rindex;
623*8ab949d8SShri Abhyankar   } else {
624*8ab949d8SShri Abhyankar     mbs = a->mbs;
625*8ab949d8SShri Abhyankar     ii  = a->i;
626*8ab949d8SShri Abhyankar     z   = zarray;
627*8ab949d8SShri Abhyankar   }
628*8ab949d8SShri Abhyankar 
629*8ab949d8SShri Abhyankar   for (i=0; i<mbs; i++) {
630*8ab949d8SShri Abhyankar     n  = ii[i+1] - ii[i];
631*8ab949d8SShri Abhyankar     idx = ij + ii[i];
632*8ab949d8SShri Abhyankar     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0;
633*8ab949d8SShri Abhyankar     sum8 = 0.0; sum9 = 0.0; sum10 = 0.0; sum11 = 0.0; sum12 = 0.0; sum13 = 0.0; sum14 = 0.0;sum15 = 0.0;
634*8ab949d8SShri Abhyankar 
635*8ab949d8SShri Abhyankar     nonzerorow += (n>0);
636*8ab949d8SShri Abhyankar     for (j=0; j<n; j++) {
637*8ab949d8SShri Abhyankar       xb = x + 15*(idx[j]);
638*8ab949d8SShri Abhyankar       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
639*8ab949d8SShri Abhyankar       x8 = xb[7]; x9 = xb[8]; x10 = xb[9]; x11 = xb[10]; x12 = xb[11]; x13 = xb[12]; x14 = xb[13];x15 = xb[14];
640*8ab949d8SShri Abhyankar 
641*8ab949d8SShri Abhyankar       for(k=0;k<15;k++){
642*8ab949d8SShri Abhyankar 	sum1  += v[0]*xb[k];
643*8ab949d8SShri Abhyankar         sum2  += v[1]*xb[k];
644*8ab949d8SShri Abhyankar 	sum3  += v[2]*xb[k];
645*8ab949d8SShri Abhyankar 	sum4  += v[3]*xb[k];
646*8ab949d8SShri Abhyankar 	sum5  += v[4]*xb[k];
647*8ab949d8SShri Abhyankar         sum6  += v[5]*xb[k];
648*8ab949d8SShri Abhyankar 	sum7  += v[6]*xb[k];
649*8ab949d8SShri Abhyankar 	sum8  += v[7]*xb[k];
650*8ab949d8SShri Abhyankar         sum9  += v[8]*xb[k];
651*8ab949d8SShri Abhyankar         sum10 += v[9]*xb[k];
652*8ab949d8SShri Abhyankar 	sum11 += v[10]*xb[k];
653*8ab949d8SShri Abhyankar 	sum12 += v[11]*xb[k];
654*8ab949d8SShri Abhyankar 	sum13 += v[12]*xb[k];
655*8ab949d8SShri Abhyankar         sum14 += v[13]*xb[k];
656*8ab949d8SShri Abhyankar 	sum15 += v[14]*xb[k];
657*8ab949d8SShri Abhyankar 	v += 15;
658*8ab949d8SShri Abhyankar       }
659*8ab949d8SShri Abhyankar     }
660*8ab949d8SShri Abhyankar     if (usecprow) z = zarray + 15*ridx[i];
661*8ab949d8SShri Abhyankar     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
662*8ab949d8SShri Abhyankar     z[7] = sum8; z[8] = sum9; z[9] = sum10; z[10] = sum11; z[11] = sum12; z[12] = sum13; z[13] = sum14;z[14] = sum15;
663*8ab949d8SShri Abhyankar 
664*8ab949d8SShri Abhyankar     if (!usecprow) z += 15;
665*8ab949d8SShri Abhyankar   }
666*8ab949d8SShri Abhyankar 
667*8ab949d8SShri Abhyankar   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
668*8ab949d8SShri Abhyankar   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
669*8ab949d8SShri Abhyankar   ierr = PetscLogFlops(450.0*a->nz - 15.0*nonzerorow);CHKERRQ(ierr);
670*8ab949d8SShri Abhyankar   PetscFunctionReturn(0);
671*8ab949d8SShri Abhyankar }
672*8ab949d8SShri Abhyankar 
673*8ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15_ver2 : Columns in the block are accessed in sets of 4,4,4,3 */
674*8ab949d8SShri Abhyankar #undef __FUNCT__
675*8ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver2"
676*8ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver2(Mat A,Vec xx,Vec zz)
677*8ab949d8SShri Abhyankar {
678*8ab949d8SShri Abhyankar   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
679*8ab949d8SShri Abhyankar   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15;
680*8ab949d8SShri Abhyankar   const PetscScalar *x,*xb;
681*8ab949d8SShri Abhyankar   PetscScalar        x1,x2,x3,x4,x5,x6,x7,x8,x9,x10,x11,x12,x13,x14,x15,*zarray;
682*8ab949d8SShri Abhyankar   const MatScalar   *v;
683*8ab949d8SShri Abhyankar   PetscErrorCode    ierr;
684*8ab949d8SShri Abhyankar   const PetscInt    *ii,*ij=a->j,*idx;
685*8ab949d8SShri Abhyankar   PetscInt          mbs,i,j,n,*ridx=PETSC_NULL,nonzerorow=0;
686*8ab949d8SShri Abhyankar   PetscTruth        usecprow=a->compressedrow.use;
687*8ab949d8SShri Abhyankar 
688*8ab949d8SShri Abhyankar   PetscFunctionBegin;
689*8ab949d8SShri Abhyankar   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
690*8ab949d8SShri Abhyankar   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
691*8ab949d8SShri Abhyankar 
692*8ab949d8SShri Abhyankar   v   = a->a;
693*8ab949d8SShri Abhyankar   if (usecprow){
694*8ab949d8SShri Abhyankar     mbs    = a->compressedrow.nrows;
695*8ab949d8SShri Abhyankar     ii     = a->compressedrow.i;
696*8ab949d8SShri Abhyankar     ridx = a->compressedrow.rindex;
697*8ab949d8SShri Abhyankar   } else {
698*8ab949d8SShri Abhyankar     mbs = a->mbs;
699*8ab949d8SShri Abhyankar     ii  = a->i;
700*8ab949d8SShri Abhyankar     z   = zarray;
701*8ab949d8SShri Abhyankar   }
702*8ab949d8SShri Abhyankar 
703*8ab949d8SShri Abhyankar   for (i=0; i<mbs; i++) {
704*8ab949d8SShri Abhyankar     n  = ii[i+1] - ii[i];
705*8ab949d8SShri Abhyankar     idx = ij + ii[i];
706*8ab949d8SShri Abhyankar     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0;
707*8ab949d8SShri Abhyankar     sum8 = 0.0; sum9 = 0.0; sum10 = 0.0; sum11 = 0.0; sum12 = 0.0; sum13 = 0.0; sum14 = 0.0;sum15 = 0.0;
708*8ab949d8SShri Abhyankar 
709*8ab949d8SShri Abhyankar     nonzerorow += (n>0);
710*8ab949d8SShri Abhyankar     for (j=0; j<n; j++) {
711*8ab949d8SShri Abhyankar       xb = x + 15*(idx[j]);
712*8ab949d8SShri Abhyankar       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
713*8ab949d8SShri Abhyankar       x8 = xb[7]; x9 = xb[8]; x10 = xb[9]; x11 = xb[10]; x12 = xb[11]; x13 = xb[12]; x14 = xb[13];x15 = xb[14];
714*8ab949d8SShri Abhyankar 
715*8ab949d8SShri Abhyankar       sum1  += v[0]*x1 + v[15]*x2 + v[30]*x3   + v[45]*x4;
716*8ab949d8SShri Abhyankar       sum2  += v[1]*x1 + v[16]*x2 + v[31]*x3   + v[46]*x4;
717*8ab949d8SShri Abhyankar       sum3  += v[2]*x1 + v[17]*x2 + v[32]*x3  + v[47]*x4;
718*8ab949d8SShri Abhyankar       sum4  += v[3]*x1 + v[18]*x2 + v[33]*x3  + v[48]*x4;
719*8ab949d8SShri Abhyankar       sum5  += v[4]*x1 + v[19]*x2 + v[34]*x3   + v[49]*x4;
720*8ab949d8SShri Abhyankar       sum6  += v[5]*x1 + v[20]*x2 + v[35]*x3   + v[50]*x4;
721*8ab949d8SShri Abhyankar       sum7  += v[6]*x1 + v[21]*x2 + v[36]*x3  + v[51]*x4;
722*8ab949d8SShri Abhyankar       sum8  += v[7]*x1 + v[22]*x2 + v[37]*x3  + v[52]*x4;
723*8ab949d8SShri Abhyankar       sum9  += v[8]*x1 + v[23]*x2 + v[38]*x3   + v[53]*x4;
724*8ab949d8SShri Abhyankar       sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3   + v[54]*x4;
725*8ab949d8SShri Abhyankar       sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3  + v[55]*x4;
726*8ab949d8SShri Abhyankar       sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3  + v[56]*x4;
727*8ab949d8SShri Abhyankar       sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3   + v[57]*x4;
728*8ab949d8SShri Abhyankar       sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3   + v[58]*x4;
729*8ab949d8SShri Abhyankar       sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3  + v[59]*x4;
730*8ab949d8SShri Abhyankar 
731*8ab949d8SShri Abhyankar       v += 60;
732*8ab949d8SShri Abhyankar 
733*8ab949d8SShri Abhyankar       sum1  += v[0]*x1 + v[15]*x2 + v[30]*x3   + v[45]*x4;
734*8ab949d8SShri Abhyankar       sum2  += v[1]*x1 + v[16]*x2 + v[31]*x3   + v[46]*x4;
735*8ab949d8SShri Abhyankar       sum3  += v[2]*x1 + v[17]*x2 + v[32]*x3  + v[47]*x4;
736*8ab949d8SShri Abhyankar       sum4  += v[3]*x1 + v[18]*x2 + v[33]*x3  + v[48]*x4;
737*8ab949d8SShri Abhyankar       sum5  += v[4]*x1 + v[19]*x2 + v[34]*x3   + v[49]*x4;
738*8ab949d8SShri Abhyankar       sum6  += v[5]*x1 + v[20]*x2 + v[35]*x3   + v[50]*x4;
739*8ab949d8SShri Abhyankar       sum7  += v[6]*x1 + v[21]*x2 + v[36]*x3  + v[51]*x4;
740*8ab949d8SShri Abhyankar       sum8  += v[7]*x1 + v[22]*x2 + v[37]*x3  + v[52]*x4;
741*8ab949d8SShri Abhyankar       sum9  += v[8]*x1 + v[23]*x2 + v[38]*x3   + v[53]*x4;
742*8ab949d8SShri Abhyankar       sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3   + v[54]*x4;
743*8ab949d8SShri Abhyankar       sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3  + v[55]*x4;
744*8ab949d8SShri Abhyankar       sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3  + v[56]*x4;
745*8ab949d8SShri Abhyankar       sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3   + v[57]*x4;
746*8ab949d8SShri Abhyankar       sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3   + v[58]*x4;
747*8ab949d8SShri Abhyankar       sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3  + v[59]*x4;
748*8ab949d8SShri Abhyankar       v += 60;
749*8ab949d8SShri Abhyankar 
750*8ab949d8SShri Abhyankar       sum1  += v[0]*x1 + v[15]*x2 + v[30]*x3   + v[45]*x4;
751*8ab949d8SShri Abhyankar       sum2  += v[1]*x1 + v[16]*x2 + v[31]*x3   + v[46]*x4;
752*8ab949d8SShri Abhyankar       sum3  += v[2]*x1 + v[17]*x2 + v[32]*x3  + v[47]*x4;
753*8ab949d8SShri Abhyankar       sum4  += v[3]*x1 + v[18]*x2 + v[33]*x3  + v[48]*x4;
754*8ab949d8SShri Abhyankar       sum5  += v[4]*x1 + v[19]*x2 + v[34]*x3   + v[49]*x4;
755*8ab949d8SShri Abhyankar       sum6  += v[5]*x1 + v[20]*x2 + v[35]*x3   + v[50]*x4;
756*8ab949d8SShri Abhyankar       sum7  += v[6]*x1 + v[21]*x2 + v[36]*x3  + v[51]*x4;
757*8ab949d8SShri Abhyankar       sum8  += v[7]*x1 + v[22]*x2 + v[37]*x3  + v[52]*x4;
758*8ab949d8SShri Abhyankar       sum9  += v[8]*x1 + v[23]*x2 + v[38]*x3   + v[53]*x4;
759*8ab949d8SShri Abhyankar       sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3   + v[54]*x4;
760*8ab949d8SShri Abhyankar       sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3  + v[55]*x4;
761*8ab949d8SShri Abhyankar       sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3  + v[56]*x4;
762*8ab949d8SShri Abhyankar       sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3   + v[57]*x4;
763*8ab949d8SShri Abhyankar       sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3   + v[58]*x4;
764*8ab949d8SShri Abhyankar       sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3  + v[59]*x4;
765*8ab949d8SShri Abhyankar       v  += 60;
766*8ab949d8SShri Abhyankar 
767*8ab949d8SShri Abhyankar       sum1  += v[0]*x1 + v[15]*x2 + v[30]*x3;
768*8ab949d8SShri Abhyankar       sum2  += v[1]*x1 + v[16]*x2 + v[31]*x3;
769*8ab949d8SShri Abhyankar       sum3  += v[2]*x1 + v[17]*x2 + v[32]*x3;
770*8ab949d8SShri Abhyankar       sum4  += v[3]*x1 + v[18]*x2 + v[33]*x3;
771*8ab949d8SShri Abhyankar       sum5  += v[4]*x1 + v[19]*x2 + v[34]*x3;
772*8ab949d8SShri Abhyankar       sum6  += v[5]*x1 + v[20]*x2 + v[35]*x3;
773*8ab949d8SShri Abhyankar       sum7  += v[6]*x1 + v[21]*x2 + v[36]*x3;
774*8ab949d8SShri Abhyankar       sum8  += v[7]*x1 + v[22]*x2 + v[37]*x3;
775*8ab949d8SShri Abhyankar       sum9  += v[8]*x1 + v[23]*x2 + v[38]*x3;
776*8ab949d8SShri Abhyankar       sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3;
777*8ab949d8SShri Abhyankar       sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3;
778*8ab949d8SShri Abhyankar       sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3;
779*8ab949d8SShri Abhyankar       sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3;
780*8ab949d8SShri Abhyankar       sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3;
781*8ab949d8SShri Abhyankar       sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3;
782*8ab949d8SShri Abhyankar       v += 45;
783*8ab949d8SShri Abhyankar     }
784*8ab949d8SShri Abhyankar     if (usecprow) z = zarray + 15*ridx[i];
785*8ab949d8SShri Abhyankar     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
786*8ab949d8SShri Abhyankar     z[7] = sum8; z[8] = sum9; z[9] = sum10; z[10] = sum11; z[11] = sum12; z[12] = sum13; z[13] = sum14;z[14] = sum15;
787*8ab949d8SShri Abhyankar 
788*8ab949d8SShri Abhyankar     if (!usecprow) z += 15;
789*8ab949d8SShri Abhyankar   }
790*8ab949d8SShri Abhyankar 
791*8ab949d8SShri Abhyankar   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
792*8ab949d8SShri Abhyankar   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
793*8ab949d8SShri Abhyankar   ierr = PetscLogFlops(450.0*a->nz - 15.0*nonzerorow);CHKERRQ(ierr);
794*8ab949d8SShri Abhyankar   PetscFunctionReturn(0);
795*8ab949d8SShri Abhyankar }
796*8ab949d8SShri Abhyankar 
797*8ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15_ver3 : Columns in the block are accessed in sets of 8,7 */
798*8ab949d8SShri Abhyankar #undef __FUNCT__
799*8ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver3"
800*8ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver3(Mat A,Vec xx,Vec zz)
801*8ab949d8SShri Abhyankar {
802*8ab949d8SShri Abhyankar   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
803*8ab949d8SShri Abhyankar   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15;
804*8ab949d8SShri Abhyankar   const PetscScalar *x,*xb;
805*8ab949d8SShri Abhyankar   PetscScalar        x1,x2,x3,x4,x5,x6,x7,x8,x9,x10,x11,x12,x13,x14,x15,*zarray;
806*8ab949d8SShri Abhyankar   const MatScalar   *v;
807*8ab949d8SShri Abhyankar   PetscErrorCode    ierr;
808*8ab949d8SShri Abhyankar   const PetscInt    *ii,*ij=a->j,*idx;
809*8ab949d8SShri Abhyankar   PetscInt          mbs,i,j,n,*ridx=PETSC_NULL,nonzerorow=0;
810*8ab949d8SShri Abhyankar   PetscTruth        usecprow=a->compressedrow.use;
811*8ab949d8SShri Abhyankar 
812*8ab949d8SShri Abhyankar   PetscFunctionBegin;
813*8ab949d8SShri Abhyankar   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
814*8ab949d8SShri Abhyankar   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
815*8ab949d8SShri Abhyankar 
816*8ab949d8SShri Abhyankar   v   = a->a;
817*8ab949d8SShri Abhyankar   if (usecprow){
818*8ab949d8SShri Abhyankar     mbs    = a->compressedrow.nrows;
819*8ab949d8SShri Abhyankar     ii     = a->compressedrow.i;
820*8ab949d8SShri Abhyankar     ridx = a->compressedrow.rindex;
821*8ab949d8SShri Abhyankar   } else {
822*8ab949d8SShri Abhyankar     mbs = a->mbs;
823*8ab949d8SShri Abhyankar     ii  = a->i;
824*8ab949d8SShri Abhyankar     z   = zarray;
825*8ab949d8SShri Abhyankar   }
826*8ab949d8SShri Abhyankar 
827*8ab949d8SShri Abhyankar   for (i=0; i<mbs; i++) {
828*8ab949d8SShri Abhyankar     n  = ii[i+1] - ii[i];
829*8ab949d8SShri Abhyankar     idx = ij + ii[i];
830*8ab949d8SShri Abhyankar     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0;
831*8ab949d8SShri Abhyankar     sum8 = 0.0; sum9 = 0.0; sum10 = 0.0; sum11 = 0.0; sum12 = 0.0; sum13 = 0.0; sum14 = 0.0;sum15 = 0.0;
832*8ab949d8SShri Abhyankar 
833*8ab949d8SShri Abhyankar     nonzerorow += (n>0);
834*8ab949d8SShri Abhyankar     for (j=0; j<n; j++) {
835*8ab949d8SShri Abhyankar       xb = x + 15*(idx[j]);
836*8ab949d8SShri Abhyankar       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
837*8ab949d8SShri Abhyankar       x8 = xb[7]; x9 = xb[8]; x10 = xb[9]; x11 = xb[10]; x12 = xb[11]; x13 = xb[12]; x14 = xb[13];x15 = xb[14];
838*8ab949d8SShri Abhyankar 
839*8ab949d8SShri Abhyankar       sum1  += v[0]*x1 + v[15]*x2  + v[30]*x3  + v[45]*x4 + v[60]*x5 + v[75]*x6 + v[90]*x7 + v[105]*x8;
840*8ab949d8SShri Abhyankar       sum2  += v[1]*x1 + v[16]*x2  + v[31]*x3  + v[46]*x4 + v[61]*x5 + v[76]*x6 + v[91]*x7 + v[106]*x8;
841*8ab949d8SShri Abhyankar       sum3  += v[2]*x1 + v[17]*x2  + v[32]*x3  + v[47]*x4 + v[62]*x5 + v[77]*x6 + v[92]*x7 + v[107]*x8;
842*8ab949d8SShri Abhyankar       sum4  += v[3]*x1 + v[18]*x2 + v[33]*x3  + v[48]*x4 + v[63]*x5 + v[78]*x6 + v[93]*x7 + v[108]*x8;
843*8ab949d8SShri Abhyankar       sum5  += v[4]*x1 + v[19]*x2 + v[34]*x3  + v[49]*x4 + v[64]*x5 + v[79]*x6 + v[94]*x7 + v[109]*x8;
844*8ab949d8SShri Abhyankar       sum6  += v[5]*x1 + v[20]*x2 + v[35]*x3  + v[50]*x4 + v[65]*x5 + v[80]*x6 + v[95]*x7 + v[110]*x8;
845*8ab949d8SShri Abhyankar       sum7  += v[6]*x1 + v[21]*x2 + v[36]*x3  + v[51]*x4 + v[66]*x5 + v[81]*x6 + v[96]*x7 + v[111]*x8;
846*8ab949d8SShri Abhyankar       sum8  += v[7]*x1 + v[22]*x2  + v[37]*x3  + v[52]*x4 + v[67]*x5 + v[82]*x6 + v[97]*x7 + v[112]*x8;
847*8ab949d8SShri Abhyankar       sum9  += v[8]*x1 + v[23]*x2  + v[38]*x3  + v[53]*x4 + v[68]*x5 + v[83]*x6 + v[98]*x7 + v[113]*x8;
848*8ab949d8SShri Abhyankar       sum10 += v[9]*x1 + v[24]*x2  + v[39]*x3  + v[54]*x4 + v[69]*x5 + v[84]*x6 + v[99]*x7 + v[114]*x8;
849*8ab949d8SShri Abhyankar       sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3  + v[55]*x4 + v[70]*x5 + v[85]*x6 + v[100]*x7 + v[115]*x8;
850*8ab949d8SShri Abhyankar       sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3  + v[56]*x4 + v[71]*x5 + v[86]*x6 + v[101]*x7 + v[116]*x8;
851*8ab949d8SShri Abhyankar       sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3  + v[57]*x4 + v[72]*x5 + v[87]*x6 + v[102]*x7 + v[117]*x8;
852*8ab949d8SShri Abhyankar       sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3  + v[58]*x4 + v[73]*x5 + v[88]*x6 + v[103]*x7 + v[118]*x8;
853*8ab949d8SShri Abhyankar       sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3  + v[59]*x4 + v[74]*x5 + v[89]*x6 + v[104]*x7 + v[119]*x8;
854*8ab949d8SShri Abhyankar       v += 120;
855*8ab949d8SShri Abhyankar 
856*8ab949d8SShri Abhyankar       sum1  += v[0]*x1 + v[15]*x2  + v[30]*x3  + v[45]*x4 + v[60]*x5 + v[75]*x6 + v[90]*x7;
857*8ab949d8SShri Abhyankar       sum2  += v[1]*x1 + v[16]*x2  + v[31]*x3  + v[46]*x4 + v[61]*x5 + v[76]*x6 + v[91]*x7;
858*8ab949d8SShri Abhyankar       sum3  += v[2]*x1 + v[17]*x2  + v[32]*x3  + v[47]*x4 + v[62]*x5 + v[77]*x6 + v[92]*x7;
859*8ab949d8SShri Abhyankar       sum4  += v[3]*x1 + v[18]*x2 + v[33]*x3  + v[48]*x4 + v[63]*x5 + v[78]*x6 + v[93]*x7;
860*8ab949d8SShri Abhyankar       sum5  += v[4]*x1 + v[19]*x2 + v[34]*x3  + v[49]*x4 + v[64]*x5 + v[79]*x6 + v[94]*x7;
861*8ab949d8SShri Abhyankar       sum6  += v[5]*x1 + v[20]*x2 + v[35]*x3  + v[50]*x4 + v[65]*x5 + v[80]*x6 + v[95]*x7;
862*8ab949d8SShri Abhyankar       sum7  += v[6]*x1 + v[21]*x2 + v[36]*x3  + v[51]*x4 + v[66]*x5 + v[81]*x6 + v[96]*x7;
863*8ab949d8SShri Abhyankar       sum8  += v[7]*x1 + v[22]*x2  + v[37]*x3  + v[52]*x4 + v[67]*x5 + v[82]*x6 + v[97]*x7;
864*8ab949d8SShri Abhyankar       sum9  += v[8]*x1 + v[23]*x2  + v[38]*x3  + v[53]*x4 + v[68]*x5 + v[83]*x6 + v[98]*x7;
865*8ab949d8SShri Abhyankar       sum10 += v[9]*x1 + v[24]*x2  + v[39]*x3  + v[54]*x4 + v[69]*x5 + v[84]*x6 + v[99]*x7;
866*8ab949d8SShri Abhyankar       sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3  + v[55]*x4 + v[70]*x5 + v[85]*x6 + v[100]*x7;
867*8ab949d8SShri Abhyankar       sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3  + v[56]*x4 + v[71]*x5 + v[86]*x6 + v[101]*x7;
868*8ab949d8SShri Abhyankar       sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3  + v[57]*x4 + v[72]*x5 + v[87]*x6 + v[102]*x7;
869*8ab949d8SShri Abhyankar       sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3  + v[58]*x4 + v[73]*x5 + v[88]*x6 + v[103]*x7;
870*8ab949d8SShri Abhyankar       sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3  + v[59]*x4 + v[74]*x5 + v[89]*x6 + v[104]*x7;
871*8ab949d8SShri Abhyankar       v += 105;
872*8ab949d8SShri Abhyankar     }
873*8ab949d8SShri Abhyankar     if (usecprow) z = zarray + 15*ridx[i];
874*8ab949d8SShri Abhyankar     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
875*8ab949d8SShri Abhyankar     z[7] = sum8; z[8] = sum9; z[9] = sum10; z[10] = sum11; z[11] = sum12; z[12] = sum13; z[13] = sum14;z[14] = sum15;
876*8ab949d8SShri Abhyankar 
877*8ab949d8SShri Abhyankar     if (!usecprow) z += 15;
878*8ab949d8SShri Abhyankar   }
879*8ab949d8SShri Abhyankar 
880*8ab949d8SShri Abhyankar   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
881*8ab949d8SShri Abhyankar   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
882*8ab949d8SShri Abhyankar   ierr = PetscLogFlops(450.0*a->nz - 15.0*nonzerorow);CHKERRQ(ierr);
883*8ab949d8SShri Abhyankar   PetscFunctionReturn(0);
884*8ab949d8SShri Abhyankar }
885*8ab949d8SShri Abhyankar 
886*8ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15_ver4 : All columns in the block are accessed at once */
887*8ab949d8SShri Abhyankar 
888*8ab949d8SShri Abhyankar #undef __FUNCT__
889*8ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver4"
890*8ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver4(Mat A,Vec xx,Vec zz)
891*8ab949d8SShri Abhyankar {
892*8ab949d8SShri Abhyankar   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
893*8ab949d8SShri Abhyankar   PetscScalar       *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15;
894*8ab949d8SShri Abhyankar   const PetscScalar *x,*xb;
895*8ab949d8SShri Abhyankar   PetscScalar        x1,x2,x3,x4,x5,x6,x7,x8,x9,x10,x11,x12,x13,x14,x15,*zarray;
896*8ab949d8SShri Abhyankar   const MatScalar   *v;
897*8ab949d8SShri Abhyankar   PetscErrorCode    ierr;
898*8ab949d8SShri Abhyankar   const PetscInt    *ii,*ij=a->j,*idx;
899*8ab949d8SShri Abhyankar   PetscInt          mbs,i,j,n,*ridx=PETSC_NULL,nonzerorow=0;
900*8ab949d8SShri Abhyankar   PetscTruth        usecprow=a->compressedrow.use;
901*8ab949d8SShri Abhyankar 
902*8ab949d8SShri Abhyankar   PetscFunctionBegin;
903*8ab949d8SShri Abhyankar   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
904*8ab949d8SShri Abhyankar   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
905*8ab949d8SShri Abhyankar 
906*8ab949d8SShri Abhyankar   v   = a->a;
907*8ab949d8SShri Abhyankar   if (usecprow){
908*8ab949d8SShri Abhyankar     mbs    = a->compressedrow.nrows;
909*8ab949d8SShri Abhyankar     ii     = a->compressedrow.i;
910*8ab949d8SShri Abhyankar     ridx = a->compressedrow.rindex;
911*8ab949d8SShri Abhyankar   } else {
912*8ab949d8SShri Abhyankar     mbs = a->mbs;
913*8ab949d8SShri Abhyankar     ii  = a->i;
914*8ab949d8SShri Abhyankar     z   = zarray;
915*8ab949d8SShri Abhyankar   }
916*8ab949d8SShri Abhyankar 
917*8ab949d8SShri Abhyankar   for (i=0; i<mbs; i++) {
918*8ab949d8SShri Abhyankar     n  = ii[i+1] - ii[i];
919*8ab949d8SShri Abhyankar     idx = ij + ii[i];
920*8ab949d8SShri Abhyankar     sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0;
921*8ab949d8SShri Abhyankar     sum8 = 0.0; sum9 = 0.0; sum10 = 0.0; sum11 = 0.0; sum12 = 0.0; sum13 = 0.0; sum14 = 0.0;sum15 = 0.0;
922*8ab949d8SShri Abhyankar 
923*8ab949d8SShri Abhyankar     nonzerorow += (n>0);
924*8ab949d8SShri Abhyankar     for (j=0; j<n; j++) {
925*8ab949d8SShri Abhyankar       xb = x + 15*(idx[j]);
926*8ab949d8SShri Abhyankar       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
927*8ab949d8SShri Abhyankar       x8 = xb[7]; x9 = xb[8]; x10 = xb[9]; x11 = xb[10]; x12 = xb[11]; x13 = xb[12]; x14 = xb[13];x15 = xb[14];
928*8ab949d8SShri Abhyankar 
929*8ab949d8SShri Abhyankar       sum1 +=  v[0]*x1  + v[15]*x2 + v[30]*x3 + v[45]*x4 + v[60]*x5 + v[75]*x6 + v[90]*x7  + v[105]*x8 + v[120]*x9 + v[135]*x10 + v[150]*x11 + v[165]*x12 + v[180]*x13 + v[195]*x14 + v[210]*x15;
930*8ab949d8SShri Abhyankar       sum2 +=  v[1]*x1  + v[16]*x2 + v[31]*x3 + v[46]*x4 + v[61]*x5 + v[76]*x6 + v[91]*x7  + v[106]*x8 + v[121]*x9 + v[136]*x10 + v[151]*x11 + v[166]*x12 + v[181]*x13 + v[196]*x14 + v[211]*x15;
931*8ab949d8SShri Abhyankar       sum3 +=  v[2]*x1  + v[17]*x2 + v[32]*x3 + v[47]*x4 + v[62]*x5 + v[77]*x6 + v[92]*x7  + v[107]*x8 + v[122]*x9 + v[137]*x10 + v[152]*x11 + v[167]*x12 + v[182]*x13 + v[197]*x14 + v[212]*x15;
932*8ab949d8SShri Abhyankar       sum4 +=  v[3]*x1  + v[18]*x2 + v[33]*x3 + v[48]*x4 + v[63]*x5 + v[78]*x6 + v[93]*x7  + v[108]*x8 + v[123]*x9 + v[138]*x10 + v[153]*x11 + v[168]*x12 + v[183]*x13 + v[198]*x14 + v[213]*x15;
933*8ab949d8SShri Abhyankar       sum5  += v[4]*x1  + v[19]*x2 + v[34]*x3 + v[49]*x4 + v[64]*x5 + v[79]*x6 + v[94]*x7  + v[109]*x8 + v[124]*x9 + v[139]*x10 + v[154]*x11 + v[169]*x12 + v[184]*x13 + v[199]*x14 + v[214]*x15;
934*8ab949d8SShri Abhyankar       sum6  += v[5]*x1  + v[20]*x2 + v[35]*x3 + v[50]*x4 + v[65]*x5 + v[80]*x6 + v[95]*x7  + v[110]*x8 + v[125]*x9 + v[140]*x10 + v[155]*x11 + v[170]*x12 + v[185]*x13 + v[200]*x14 + v[215]*x15;
935*8ab949d8SShri Abhyankar       sum7  += v[6]*x1  + v[21]*x2 + v[36]*x3 + v[51]*x4 + v[66]*x5 + v[81]*x6 + v[96]*x7  + v[111]*x8 + v[126]*x9 + v[141]*x10 + v[156]*x11 + v[171]*x12 + v[186]*x13 + v[201]*x14 + v[216]*x15;
936*8ab949d8SShri Abhyankar       sum8  += v[7]*x1  + v[22]*x2 + v[37]*x3 + v[52]*x4 + v[67]*x5 + v[82]*x6 + v[97]*x7  + v[112]*x8 + v[127]*x9 + v[142]*x10 + v[157]*x11 + v[172]*x12 + v[187]*x13 + v[202]*x14 + v[217]*x15;
937*8ab949d8SShri Abhyankar       sum9  += v[8]*x1  + v[23]*x2 + v[38]*x3 + v[53]*x4 + v[68]*x5 + v[83]*x6 + v[98]*x7  + v[113]*x8 + v[128]*x9 + v[143]*x10 + v[158]*x11 + v[173]*x12 + v[188]*x13 + v[203]*x14 + v[218]*x15;
938*8ab949d8SShri Abhyankar       sum10 += v[9]*x1  + v[24]*x2 + v[39]*x3 + v[54]*x4 + v[69]*x5 + v[84]*x6 + v[99]*x7  + v[114]*x8 + v[129]*x9 + v[144]*x10 + v[159]*x11 + v[174]*x12 + v[189]*x13 + v[204]*x14 + v[219]*x15;
939*8ab949d8SShri Abhyankar       sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4 + v[70]*x5 + v[85]*x6 + v[100]*x7 + v[115]*x8 + v[130]*x9 + v[145]*x10 + v[160]*x11 + v[175]*x12 + v[190]*x13 + v[205]*x14 + v[220]*x15;
940*8ab949d8SShri Abhyankar       sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4 + v[71]*x5 + v[86]*x6 + v[101]*x7 + v[116]*x8 + v[131]*x9 + v[146]*x10 + v[161]*x11 + v[176]*x12 + v[191]*x13 + v[206]*x14 + v[221]*x15;
941*8ab949d8SShri Abhyankar       sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4 + v[72]*x5 + v[87]*x6 + v[102]*x7 + v[117]*x8 + v[132]*x9 + v[147]*x10 + v[162]*x11 + v[177]*x12 + v[192]*x13 + v[207]*x14 + v[222]*x15;
942*8ab949d8SShri Abhyankar       sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4 + v[73]*x5 + v[88]*x6 + v[103]*x7 + v[118]*x8 + v[133]*x9 + v[148]*x10 + v[163]*x11 + v[178]*x12 + v[193]*x13 + v[208]*x14 + v[223]*x15;
943*8ab949d8SShri Abhyankar       sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4 + v[74]*x5 + v[89]*x6 + v[104]*x7 + v[119]*x8 + v[134]*x9 + v[149]*x10 + v[164]*x11 + v[179]*x12 + v[194]*x13 + v[209]*x14 + v[224]*x15;
944*8ab949d8SShri Abhyankar       v += 225;
945*8ab949d8SShri Abhyankar     }
946*8ab949d8SShri Abhyankar     if (usecprow) z = zarray + 15*ridx[i];
947*8ab949d8SShri Abhyankar     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
948*8ab949d8SShri Abhyankar     z[7] = sum8; z[8] = sum9; z[9] = sum10; z[10] = sum11; z[11] = sum12; z[12] = sum13; z[13] = sum14;z[14] = sum15;
949*8ab949d8SShri Abhyankar 
950*8ab949d8SShri Abhyankar     if (!usecprow) z += 15;
951*8ab949d8SShri Abhyankar   }
952*8ab949d8SShri Abhyankar 
953*8ab949d8SShri Abhyankar   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
954*8ab949d8SShri Abhyankar   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
955*8ab949d8SShri Abhyankar   ierr = PetscLogFlops(450.0*a->nz - 15.0*nonzerorow);CHKERRQ(ierr);
956*8ab949d8SShri Abhyankar   PetscFunctionReturn(0);
957*8ab949d8SShri Abhyankar }
958*8ab949d8SShri Abhyankar 
959*8ab949d8SShri Abhyankar 
9603f1db9ecSBarry Smith /*
9613f1db9ecSBarry Smith     This will not work with MatScalar == float because it calls the BLAS
9623f1db9ecSBarry Smith */
9634a2ae208SSatish Balay #undef __FUNCT__
9644a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_N"
965dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_N(Mat A,Vec xx,Vec zz)
9662d61bbb3SSatish Balay {
9672d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
968dcf5cc72SBarry Smith   PetscScalar    *x,*z = 0,*xb,*work,*workt,*zarray;
9693f1db9ecSBarry Smith   MatScalar      *v;
970dfbe8321SBarry Smith   PetscErrorCode ierr;
971d0f46423SBarry Smith   PetscInt       mbs=a->mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2;
97298c9bda7SSatish Balay   PetscInt       ncols,k,*ridx=PETSC_NULL,nonzerorow=0;
97326e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
9742d61bbb3SSatish Balay 
9752d61bbb3SSatish Balay   PetscFunctionBegin;
9761ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
97726e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
9782d61bbb3SSatish Balay 
9792d61bbb3SSatish Balay   idx = a->j;
9802d61bbb3SSatish Balay   v   = a->a;
98126e093fcSHong Zhang   if (usecprow){
98226e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
98326e093fcSHong Zhang     ii   = a->compressedrow.i;
9847b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
98526e093fcSHong Zhang   } else {
98626e093fcSHong Zhang     mbs = a->mbs;
9872d61bbb3SSatish Balay     ii  = a->i;
98826e093fcSHong Zhang     z   = zarray;
98926e093fcSHong Zhang   }
990218c64b6SSatish Balay 
9912d61bbb3SSatish Balay   if (!a->mult_work) {
992d0f46423SBarry Smith     k    = PetscMax(A->rmap->n,A->cmap->n);
99387828ca2SBarry Smith     ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
9942d61bbb3SSatish Balay   }
9952d61bbb3SSatish Balay   work = a->mult_work;
9962d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
9972d61bbb3SSatish Balay     n     = ii[1] - ii[0]; ii++;
9982d61bbb3SSatish Balay     ncols = n*bs;
9992d61bbb3SSatish Balay     workt = work;
100098c9bda7SSatish Balay     nonzerorow += (n>0);
10012d61bbb3SSatish Balay     for (j=0; j<n; j++) {
10022d61bbb3SSatish Balay       xb = x + bs*(*idx++);
10032d61bbb3SSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
10042d61bbb3SSatish Balay       workt += bs;
10052d61bbb3SSatish Balay     }
10067b2bb3b9SHong Zhang     if (usecprow) z = zarray + bs*ridx[i];
1007737d121aSSatish Balay     Kernel_w_gets_Ar_times_v(bs,ncols,work,v,z);
100871044d3cSBarry Smith     /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DZero,z,&_One); */
10092d61bbb3SSatish Balay     v += n*bs2;
101026e093fcSHong Zhang     if (!usecprow) z += bs;
10112d61bbb3SSatish Balay   }
10121ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
101326e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
1014dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz*bs2 - bs*nonzerorow);CHKERRQ(ierr);
10152d61bbb3SSatish Balay   PetscFunctionReturn(0);
10162d61bbb3SSatish Balay }
10172d61bbb3SSatish Balay 
10186a65c766SHong Zhang extern PetscErrorCode VecCopy_Seq(Vec,Vec);
10194a2ae208SSatish Balay #undef __FUNCT__
10204a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_1"
1021dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_1(Mat A,Vec xx,Vec yy,Vec zz)
10222d61bbb3SSatish Balay {
10232d61bbb3SSatish Balay   Mat_SeqBAIJ        *a = (Mat_SeqBAIJ*)A->data;
1024122f12eaSBarry Smith   const PetscScalar  *x;
1025122f12eaSBarry Smith   PetscScalar        *y,*z,sum;
1026122f12eaSBarry Smith   const MatScalar    *v;
1027dfbe8321SBarry Smith   PetscErrorCode     ierr;
1028122f12eaSBarry Smith   PetscInt           mbs=a->mbs,i,n,*ridx=PETSC_NULL,nonzerorow=0;
1029122f12eaSBarry Smith   const PetscInt     *idx,*ii;
103026e093fcSHong Zhang   PetscTruth         usecprow=a->compressedrow.use;
10312d61bbb3SSatish Balay 
10322d61bbb3SSatish Balay   PetscFunctionBegin;
1033122f12eaSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
1034122f12eaSBarry Smith   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
10352e8a6d31SBarry Smith   if (zz != yy) {
1036122f12eaSBarry Smith     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
10372e8a6d31SBarry Smith   } else {
1038122f12eaSBarry Smith     z = y;
10392e8a6d31SBarry Smith   }
10402d61bbb3SSatish Balay 
10412d61bbb3SSatish Balay   idx = a->j;
10422d61bbb3SSatish Balay   v   = a->a;
104326e093fcSHong Zhang   if (usecprow){
10444eb6d288SHong Zhang     if (zz != yy){
1045122f12eaSBarry Smith       ierr = PetscMemcpy(z,y,mbs*sizeof(PetscScalar));CHKERRQ(ierr);
10464eb6d288SHong Zhang     }
104726e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
104826e093fcSHong Zhang     ii   = a->compressedrow.i;
10497b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
105026e093fcSHong Zhang   } else {
10512d61bbb3SSatish Balay     ii  = a->i;
105226e093fcSHong Zhang   }
10532d61bbb3SSatish Balay 
10542d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
1055122f12eaSBarry Smith     n    = ii[1] - ii[0];
1056122f12eaSBarry Smith     ii++;
105726e093fcSHong Zhang     if (!usecprow){
1058122f12eaSBarry Smith       nonzerorow += (n>0);
1059122f12eaSBarry Smith       sum = y[i];
1060122f12eaSBarry Smith     } else {
1061122f12eaSBarry Smith       sum = y[ridx[i]];
1062122f12eaSBarry Smith     }
1063122f12eaSBarry Smith     PetscSparseDensePlusDot(sum,x,v,idx,n);
1064122f12eaSBarry Smith     v += n;
1065122f12eaSBarry Smith     idx += n;
1066122f12eaSBarry Smith     if (usecprow){
1067122f12eaSBarry Smith       z[ridx[i]] = sum;
1068122f12eaSBarry Smith     } else {
1069122f12eaSBarry Smith       z[i] = sum;
107026e093fcSHong Zhang     }
10712d61bbb3SSatish Balay   }
1072122f12eaSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
1073122f12eaSBarry Smith   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
10742e8a6d31SBarry Smith   if (zz != yy) {
1075122f12eaSBarry Smith     ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
10762e8a6d31SBarry Smith   }
1077122f12eaSBarry Smith   ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr);
10782d61bbb3SSatish Balay   PetscFunctionReturn(0);
10792d61bbb3SSatish Balay }
10802d61bbb3SSatish Balay 
10814a2ae208SSatish Balay #undef __FUNCT__
10824a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_2"
1083dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_2(Mat A,Vec xx,Vec yy,Vec zz)
10842d61bbb3SSatish Balay {
10852d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1086dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2;
108726e093fcSHong Zhang   PetscScalar    x1,x2,*yarray,*zarray;
10883f1db9ecSBarry Smith   MatScalar      *v;
1089dfbe8321SBarry Smith   PetscErrorCode ierr;
10904eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
109126e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
10922d61bbb3SSatish Balay 
10932d61bbb3SSatish Balay   PetscFunctionBegin;
10941ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
109526e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
10962e8a6d31SBarry Smith   if (zz != yy) {
109726e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
10982e8a6d31SBarry Smith   } else {
109926e093fcSHong Zhang     zarray = yarray;
11002e8a6d31SBarry Smith   }
11012d61bbb3SSatish Balay 
11022d61bbb3SSatish Balay   idx = a->j;
11032d61bbb3SSatish Balay   v   = a->a;
110426e093fcSHong Zhang   if (usecprow){
11054eb6d288SHong Zhang     if (zz != yy){
11064eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,2*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
11074eb6d288SHong Zhang     }
110826e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
110926e093fcSHong Zhang     ii   = a->compressedrow.i;
11107b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
11114eb6d288SHong Zhang     if (zz != yy){
11124eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,a->mbs*sizeof(PetscScalar));CHKERRQ(ierr);
11134eb6d288SHong Zhang     }
111426e093fcSHong Zhang   } else {
11152d61bbb3SSatish Balay     ii  = a->i;
111626e093fcSHong Zhang     y   = yarray;
111726e093fcSHong Zhang     z   = zarray;
111826e093fcSHong Zhang   }
11192d61bbb3SSatish Balay 
11202d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
11212d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
112226e093fcSHong Zhang     if (usecprow){
11237b2bb3b9SHong Zhang       z = zarray + 2*ridx[i];
11247b2bb3b9SHong Zhang       y = yarray + 2*ridx[i];
112526e093fcSHong Zhang     }
11262d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1];
11272d61bbb3SSatish Balay     for (j=0; j<n; j++) {
11282d61bbb3SSatish Balay       xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1];
11292d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[2]*x2;
11302d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[3]*x2;
11312d61bbb3SSatish Balay       v += 4;
11322d61bbb3SSatish Balay     }
11332d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2;
113426e093fcSHong Zhang     if (!usecprow){
11352d61bbb3SSatish Balay       z += 2; y += 2;
11362d61bbb3SSatish Balay     }
113726e093fcSHong Zhang   }
11381ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
113926e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
11402e8a6d31SBarry Smith   if (zz != yy) {
114126e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
11422e8a6d31SBarry Smith   }
1143dc0b31edSSatish Balay   ierr = PetscLogFlops(4.0*a->nz);CHKERRQ(ierr);
11442d61bbb3SSatish Balay   PetscFunctionReturn(0);
11452d61bbb3SSatish Balay }
11462d61bbb3SSatish Balay 
11474a2ae208SSatish Balay #undef __FUNCT__
11484a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_3"
1149dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_3(Mat A,Vec xx,Vec yy,Vec zz)
11502d61bbb3SSatish Balay {
11512d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1152dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,x1,x2,x3,*yarray,*zarray;
11533f1db9ecSBarry Smith   MatScalar      *v;
1154dfbe8321SBarry Smith   PetscErrorCode ierr;
11554eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
115626e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
11572d61bbb3SSatish Balay 
11582d61bbb3SSatish Balay   PetscFunctionBegin;
11591ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
116026e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
11612e8a6d31SBarry Smith   if (zz != yy) {
116226e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
11632e8a6d31SBarry Smith   } else {
116426e093fcSHong Zhang     zarray = yarray;
11652e8a6d31SBarry Smith   }
11662d61bbb3SSatish Balay 
11672d61bbb3SSatish Balay   idx = a->j;
11682d61bbb3SSatish Balay   v   = a->a;
116926e093fcSHong Zhang   if (usecprow){
11704eb6d288SHong Zhang     if (zz != yy){
11714eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,3*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
11724eb6d288SHong Zhang     }
117326e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
117426e093fcSHong Zhang     ii   = a->compressedrow.i;
11757b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
117626e093fcSHong Zhang   } else {
11772d61bbb3SSatish Balay     ii  = a->i;
117826e093fcSHong Zhang     y   = yarray;
117926e093fcSHong Zhang     z   = zarray;
118026e093fcSHong Zhang   }
11812d61bbb3SSatish Balay 
11822d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
11832d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
118426e093fcSHong Zhang     if (usecprow){
11857b2bb3b9SHong Zhang       z = zarray + 3*ridx[i];
11867b2bb3b9SHong Zhang       y = yarray + 3*ridx[i];
118726e093fcSHong Zhang     }
11882d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2];
11892d61bbb3SSatish Balay     for (j=0; j<n; j++) {
11902d61bbb3SSatish Balay       xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
11912d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3;
11922d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3;
11932d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3;
11942d61bbb3SSatish Balay       v += 9;
11952d61bbb3SSatish Balay     }
11962d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3;
119726e093fcSHong Zhang     if (!usecprow){
11982d61bbb3SSatish Balay       z += 3; y += 3;
11992d61bbb3SSatish Balay     }
120026e093fcSHong Zhang   }
12011ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
120226e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
12032e8a6d31SBarry Smith   if (zz != yy) {
120426e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
12052e8a6d31SBarry Smith   }
1206dc0b31edSSatish Balay   ierr = PetscLogFlops(18.0*a->nz);CHKERRQ(ierr);
12072d61bbb3SSatish Balay   PetscFunctionReturn(0);
12082d61bbb3SSatish Balay }
12092d61bbb3SSatish Balay 
12104a2ae208SSatish Balay #undef __FUNCT__
12114a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_4"
1212dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_4(Mat A,Vec xx,Vec yy,Vec zz)
12132d61bbb3SSatish Balay {
12142d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1215dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*yarray,*zarray;
12163f1db9ecSBarry Smith   MatScalar      *v;
1217dfbe8321SBarry Smith   PetscErrorCode ierr;
12184eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
121926e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
12202d61bbb3SSatish Balay 
12212d61bbb3SSatish Balay   PetscFunctionBegin;
12221ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
122326e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
12242e8a6d31SBarry Smith   if (zz != yy) {
122526e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
12262e8a6d31SBarry Smith   } else {
122726e093fcSHong Zhang     zarray = yarray;
12282e8a6d31SBarry Smith   }
12292d61bbb3SSatish Balay 
12302d61bbb3SSatish Balay   idx   = a->j;
12312d61bbb3SSatish Balay   v     = a->a;
123226e093fcSHong Zhang   if (usecprow){
12334eb6d288SHong Zhang     if (zz != yy){
12344eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,4*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
12354eb6d288SHong Zhang     }
123626e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
123726e093fcSHong Zhang     ii   = a->compressedrow.i;
12387b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
123926e093fcSHong Zhang   } else {
12402d61bbb3SSatish Balay     ii  = a->i;
124126e093fcSHong Zhang     y   = yarray;
124226e093fcSHong Zhang     z   = zarray;
124326e093fcSHong Zhang   }
12442d61bbb3SSatish Balay 
12452d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
12462d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
124726e093fcSHong Zhang     if (usecprow){
12487b2bb3b9SHong Zhang       z = zarray + 4*ridx[i];
12497b2bb3b9SHong Zhang       y = yarray + 4*ridx[i];
125026e093fcSHong Zhang     }
12512d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3];
12522d61bbb3SSatish Balay     for (j=0; j<n; j++) {
12532d61bbb3SSatish Balay       xb = x + 4*(*idx++);
12542d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
12552d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3   + v[12]*x4;
12562d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3   + v[13]*x4;
12572d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3  + v[14]*x4;
12582d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3  + v[15]*x4;
12592d61bbb3SSatish Balay       v += 16;
12602d61bbb3SSatish Balay     }
12612d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4;
126226e093fcSHong Zhang     if (!usecprow){
12632d61bbb3SSatish Balay       z += 4; y += 4;
12642d61bbb3SSatish Balay     }
126526e093fcSHong Zhang   }
12661ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
126726e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
12682e8a6d31SBarry Smith   if (zz != yy) {
126926e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
12702e8a6d31SBarry Smith   }
1271dc0b31edSSatish Balay   ierr = PetscLogFlops(32.0*a->nz);CHKERRQ(ierr);
12722d61bbb3SSatish Balay   PetscFunctionReturn(0);
12732d61bbb3SSatish Balay }
12742d61bbb3SSatish Balay 
12754a2ae208SSatish Balay #undef __FUNCT__
12764a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_5"
1277dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_5(Mat A,Vec xx,Vec yy,Vec zz)
12782d61bbb3SSatish Balay {
12792d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1280dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5;
128126e093fcSHong Zhang   PetscScalar    *yarray,*zarray;
12823f1db9ecSBarry Smith   MatScalar      *v;
1283dfbe8321SBarry Smith   PetscErrorCode ierr;
12844eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
128526e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
12862d61bbb3SSatish Balay 
12872d61bbb3SSatish Balay   PetscFunctionBegin;
12881ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
128926e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
12902e8a6d31SBarry Smith   if (zz != yy) {
129126e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
12922e8a6d31SBarry Smith   } else {
129326e093fcSHong Zhang     zarray = yarray;
12942e8a6d31SBarry Smith   }
12952d61bbb3SSatish Balay 
12962d61bbb3SSatish Balay   idx = a->j;
12972d61bbb3SSatish Balay   v   = a->a;
129826e093fcSHong Zhang   if (usecprow){
12994eb6d288SHong Zhang     if (zz != yy){
13004eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,5*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
13014eb6d288SHong Zhang     }
130226e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
130326e093fcSHong Zhang     ii   = a->compressedrow.i;
13047b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
130526e093fcSHong Zhang   } else {
13062d61bbb3SSatish Balay     ii  = a->i;
130726e093fcSHong Zhang     y   = yarray;
130826e093fcSHong Zhang     z   = zarray;
130926e093fcSHong Zhang   }
13102d61bbb3SSatish Balay 
13112d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
13122d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
131326e093fcSHong Zhang     if (usecprow){
13147b2bb3b9SHong Zhang       z = zarray + 5*ridx[i];
13157b2bb3b9SHong Zhang       y = yarray + 5*ridx[i];
131626e093fcSHong Zhang     }
13172d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4];
13182d61bbb3SSatish Balay     for (j=0; j<n; j++) {
13192d61bbb3SSatish Balay       xb = x + 5*(*idx++);
13202d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4];
13212d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3  + v[15]*x4 + v[20]*x5;
13222d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3  + v[16]*x4 + v[21]*x5;
13232d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3  + v[17]*x4 + v[22]*x5;
13242d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3  + v[18]*x4 + v[23]*x5;
13252d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3  + v[19]*x4 + v[24]*x5;
13262d61bbb3SSatish Balay       v += 25;
13272d61bbb3SSatish Balay     }
13282d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5;
132926e093fcSHong Zhang     if (!usecprow){
13302d61bbb3SSatish Balay       z += 5; y += 5;
13312d61bbb3SSatish Balay     }
133226e093fcSHong Zhang   }
13331ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
133426e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
13352e8a6d31SBarry Smith   if (zz != yy) {
133626e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
13372e8a6d31SBarry Smith   }
1338dc0b31edSSatish Balay   ierr = PetscLogFlops(50.0*a->nz);CHKERRQ(ierr);
13392d61bbb3SSatish Balay   PetscFunctionReturn(0);
13402d61bbb3SSatish Balay }
13414a2ae208SSatish Balay #undef __FUNCT__
13424a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_6"
1343dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_6(Mat A,Vec xx,Vec yy,Vec zz)
134415091d37SBarry Smith {
134515091d37SBarry Smith   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1346dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6;
134726e093fcSHong Zhang   PetscScalar    x1,x2,x3,x4,x5,x6,*yarray,*zarray;
134815091d37SBarry Smith   MatScalar      *v;
1349dfbe8321SBarry Smith   PetscErrorCode ierr;
13504eb6d288SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
135126e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
135215091d37SBarry Smith 
135315091d37SBarry Smith   PetscFunctionBegin;
13541ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
135526e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
135615091d37SBarry Smith   if (zz != yy) {
135726e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
135815091d37SBarry Smith   } else {
135926e093fcSHong Zhang     zarray = yarray;
136015091d37SBarry Smith   }
136115091d37SBarry Smith 
136215091d37SBarry Smith   idx = a->j;
136315091d37SBarry Smith   v   = a->a;
136426e093fcSHong Zhang   if (usecprow){
13654eb6d288SHong Zhang     if (zz != yy){
13664eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,6*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
13674eb6d288SHong Zhang     }
136826e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
136926e093fcSHong Zhang     ii   = a->compressedrow.i;
13707b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
137126e093fcSHong Zhang   } else {
137215091d37SBarry Smith     ii  = a->i;
137326e093fcSHong Zhang     y   = yarray;
137426e093fcSHong Zhang     z   = zarray;
137526e093fcSHong Zhang   }
137615091d37SBarry Smith 
137715091d37SBarry Smith   for (i=0; i<mbs; i++) {
137815091d37SBarry Smith     n  = ii[1] - ii[0]; ii++;
137926e093fcSHong Zhang     if (usecprow){
13807b2bb3b9SHong Zhang       z = zarray + 6*ridx[i];
13817b2bb3b9SHong Zhang       y = yarray + 6*ridx[i];
138226e093fcSHong Zhang     }
138315091d37SBarry Smith     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5];
138415091d37SBarry Smith     for (j=0; j<n; j++) {
13853b95cb0eSSatish Balay       xb = x + 6*(*idx++);
138615091d37SBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5];
138715091d37SBarry Smith       sum1 += v[0]*x1 + v[6]*x2  + v[12]*x3  + v[18]*x4 + v[24]*x5 + v[30]*x6;
138815091d37SBarry Smith       sum2 += v[1]*x1 + v[7]*x2  + v[13]*x3  + v[19]*x4 + v[25]*x5 + v[31]*x6;
138915091d37SBarry Smith       sum3 += v[2]*x1 + v[8]*x2  + v[14]*x3  + v[20]*x4 + v[26]*x5 + v[32]*x6;
139015091d37SBarry Smith       sum4 += v[3]*x1 + v[9]*x2  + v[15]*x3  + v[21]*x4 + v[27]*x5 + v[33]*x6;
139115091d37SBarry Smith       sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3  + v[22]*x4 + v[28]*x5 + v[34]*x6;
139215091d37SBarry Smith       sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3  + v[23]*x4 + v[29]*x5 + v[35]*x6;
139315091d37SBarry Smith       v += 36;
139415091d37SBarry Smith     }
139515091d37SBarry Smith     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6;
139626e093fcSHong Zhang     if (!usecprow){
139715091d37SBarry Smith       z += 6; y += 6;
139815091d37SBarry Smith     }
139926e093fcSHong Zhang   }
14001ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
140126e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
140215091d37SBarry Smith   if (zz != yy) {
140326e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
140415091d37SBarry Smith   }
1405dc0b31edSSatish Balay   ierr = PetscLogFlops(72.0*a->nz);CHKERRQ(ierr);
140615091d37SBarry Smith   PetscFunctionReturn(0);
140715091d37SBarry Smith }
14082d61bbb3SSatish Balay 
14094a2ae208SSatish Balay #undef __FUNCT__
14104a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_7"
1411dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_7(Mat A,Vec xx,Vec yy,Vec zz)
14122d61bbb3SSatish Balay {
14132d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1414dcf5cc72SBarry Smith   PetscScalar    *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6,sum7;
141526e093fcSHong Zhang   PetscScalar    x1,x2,x3,x4,x5,x6,x7,*yarray,*zarray;
14163f1db9ecSBarry Smith   MatScalar      *v;
1417dfbe8321SBarry Smith   PetscErrorCode ierr;
14187b2bb3b9SHong Zhang   PetscInt       mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL;
141926e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
14202d61bbb3SSatish Balay 
14212d61bbb3SSatish Balay   PetscFunctionBegin;
14221ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
142326e093fcSHong Zhang   ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr);
14242e8a6d31SBarry Smith   if (zz != yy) {
142526e093fcSHong Zhang     ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
14262e8a6d31SBarry Smith   } else {
142726e093fcSHong Zhang     zarray = yarray;
14282e8a6d31SBarry Smith   }
14292d61bbb3SSatish Balay 
14302d61bbb3SSatish Balay   idx = a->j;
14312d61bbb3SSatish Balay   v   = a->a;
143226e093fcSHong Zhang   if (usecprow){
14334eb6d288SHong Zhang     if (zz != yy){
14344eb6d288SHong Zhang       ierr = PetscMemcpy(zarray,yarray,7*mbs*sizeof(PetscScalar));CHKERRQ(ierr);
14354eb6d288SHong Zhang     }
143626e093fcSHong Zhang     mbs  = a->compressedrow.nrows;
143726e093fcSHong Zhang     ii   = a->compressedrow.i;
14387b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
143926e093fcSHong Zhang   } else {
14402d61bbb3SSatish Balay     ii  = a->i;
144126e093fcSHong Zhang     y   = yarray;
144226e093fcSHong Zhang     z   = zarray;
144326e093fcSHong Zhang   }
14442d61bbb3SSatish Balay 
14452d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
14462d61bbb3SSatish Balay     n  = ii[1] - ii[0]; ii++;
144726e093fcSHong Zhang     if (usecprow){
14487b2bb3b9SHong Zhang       z = zarray + 7*ridx[i];
14497b2bb3b9SHong Zhang       y = yarray + 7*ridx[i];
145026e093fcSHong Zhang     }
14512d61bbb3SSatish Balay     sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5]; sum7 = y[6];
14522d61bbb3SSatish Balay     for (j=0; j<n; j++) {
14532d61bbb3SSatish Balay       xb = x + 7*(*idx++);
14542d61bbb3SSatish Balay       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6];
14552d61bbb3SSatish Balay       sum1 += v[0]*x1 + v[7]*x2  + v[14]*x3  + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7;
14562d61bbb3SSatish Balay       sum2 += v[1]*x1 + v[8]*x2  + v[15]*x3  + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7;
14572d61bbb3SSatish Balay       sum3 += v[2]*x1 + v[9]*x2  + v[16]*x3  + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7;
14582d61bbb3SSatish Balay       sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3  + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7;
14592d61bbb3SSatish Balay       sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3  + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7;
14602d61bbb3SSatish Balay       sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3  + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7;
14612d61bbb3SSatish Balay       sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3  + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7;
14622d61bbb3SSatish Balay       v += 49;
14632d61bbb3SSatish Balay     }
14642d61bbb3SSatish Balay     z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7;
146526e093fcSHong Zhang     if (!usecprow){
14662d61bbb3SSatish Balay       z += 7; y += 7;
14672d61bbb3SSatish Balay     }
146826e093fcSHong Zhang   }
14691ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
147026e093fcSHong Zhang   ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr);
14712e8a6d31SBarry Smith   if (zz != yy) {
147226e093fcSHong Zhang     ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
14732e8a6d31SBarry Smith   }
1474dc0b31edSSatish Balay   ierr = PetscLogFlops(98.0*a->nz);CHKERRQ(ierr);
14752d61bbb3SSatish Balay   PetscFunctionReturn(0);
14762d61bbb3SSatish Balay }
1477218c64b6SSatish Balay 
14784a2ae208SSatish Balay #undef __FUNCT__
14794a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_N"
1480dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_N(Mat A,Vec xx,Vec yy,Vec zz)
14812d61bbb3SSatish Balay {
14822d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1483bba805e6SBarry Smith   PetscScalar    *x,*z = 0,*xb,*work,*workt,*zarray;
14843f1db9ecSBarry Smith   MatScalar      *v;
14856849ba73SBarry Smith   PetscErrorCode ierr;
1486d0f46423SBarry Smith   PetscInt       mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2;
14877b2bb3b9SHong Zhang   PetscInt       ncols,k,*ridx=PETSC_NULL;
148826e093fcSHong Zhang   PetscTruth     usecprow=a->compressedrow.use;
1489218c64b6SSatish Balay 
14902d61bbb3SSatish Balay   PetscFunctionBegin;
14916a65c766SHong Zhang   ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr);
14921ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
149326e093fcSHong Zhang   ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr);
14942d61bbb3SSatish Balay 
14952d61bbb3SSatish Balay   idx = a->j;
14962d61bbb3SSatish Balay   v   = a->a;
149726e093fcSHong Zhang   if (usecprow){
149826e093fcSHong Zhang     mbs    = a->compressedrow.nrows;
149926e093fcSHong Zhang     ii     = a->compressedrow.i;
15007b2bb3b9SHong Zhang     ridx = a->compressedrow.rindex;
150126e093fcSHong Zhang   } else {
150226e093fcSHong Zhang     mbs = a->mbs;
15032d61bbb3SSatish Balay     ii  = a->i;
150426e093fcSHong Zhang     z   = zarray;
150526e093fcSHong Zhang   }
15062d61bbb3SSatish Balay 
15072d61bbb3SSatish Balay   if (!a->mult_work) {
1508d0f46423SBarry Smith     k    = PetscMax(A->rmap->n,A->cmap->n);
150987828ca2SBarry Smith     ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
15102d61bbb3SSatish Balay   }
15112d61bbb3SSatish Balay   work = a->mult_work;
15122d61bbb3SSatish Balay   for (i=0; i<mbs; i++) {
15132d61bbb3SSatish Balay     n     = ii[1] - ii[0]; ii++;
15142d61bbb3SSatish Balay     ncols = n*bs;
15152d61bbb3SSatish Balay     workt = work;
15162d61bbb3SSatish Balay     for (j=0; j<n; j++) {
15172d61bbb3SSatish Balay       xb = x + bs*(*idx++);
15182d61bbb3SSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
15192d61bbb3SSatish Balay       workt += bs;
15202d61bbb3SSatish Balay     }
15217b2bb3b9SHong Zhang     if (usecprow) z = zarray + bs*ridx[i];
15223f1db9ecSBarry Smith     Kernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z);
152371044d3cSBarry Smith     /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DOne,z,&_One); */
15242d61bbb3SSatish Balay     v += n*bs2;
152526e093fcSHong Zhang     if (!usecprow){
15262d61bbb3SSatish Balay       z += bs;
15272d61bbb3SSatish Balay     }
152826e093fcSHong Zhang   }
15291ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
153026e093fcSHong Zhang   ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr);
1531dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz*bs2);CHKERRQ(ierr);
15322d61bbb3SSatish Balay   PetscFunctionReturn(0);
15332d61bbb3SSatish Balay }
15342d61bbb3SSatish Balay 
15354a2ae208SSatish Balay #undef __FUNCT__
1536547795f9SHong Zhang #define __FUNCT__ "MatMultHermitianTranspose_SeqBAIJ"
1537547795f9SHong Zhang PetscErrorCode MatMultHermitianTranspose_SeqBAIJ(Mat A,Vec xx,Vec zz)
1538547795f9SHong Zhang {
1539547795f9SHong Zhang   PetscScalar    zero = 0.0;
1540547795f9SHong Zhang   PetscErrorCode ierr;
1541547795f9SHong Zhang 
1542547795f9SHong Zhang   PetscFunctionBegin;
1543547795f9SHong Zhang   ierr = VecSet(zz,zero);CHKERRQ(ierr);
1544547795f9SHong Zhang   ierr = MatMultHermitianTransposeAdd_SeqBAIJ(A,xx,zz,zz);CHKERRQ(ierr);
1545547795f9SHong Zhang   PetscFunctionReturn(0);
1546547795f9SHong Zhang }
1547547795f9SHong Zhang 
1548547795f9SHong Zhang #undef __FUNCT__
15494a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqBAIJ"
1550dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqBAIJ(Mat A,Vec xx,Vec zz)
15512d61bbb3SSatish Balay {
15523447b6efSHong Zhang   PetscScalar    zero = 0.0;
15536849ba73SBarry Smith   PetscErrorCode ierr;
15542d61bbb3SSatish Balay 
15552d61bbb3SSatish Balay   PetscFunctionBegin;
15562dcb1b2aSMatthew Knepley   ierr = VecSet(zz,zero);CHKERRQ(ierr);
15573447b6efSHong Zhang   ierr = MatMultTransposeAdd_SeqBAIJ(A,xx,zz,zz);CHKERRQ(ierr);
15582d61bbb3SSatish Balay   PetscFunctionReturn(0);
15592d61bbb3SSatish Balay }
15602d61bbb3SSatish Balay 
15614a2ae208SSatish Balay #undef __FUNCT__
1562547795f9SHong Zhang #define __FUNCT__ "MatMultHermitianTransposeAdd_SeqBAIJ"
1563547795f9SHong Zhang PetscErrorCode MatMultHermitianTransposeAdd_SeqBAIJ(Mat A,Vec xx,Vec yy,Vec zz)
1564547795f9SHong Zhang 
1565547795f9SHong Zhang {
1566547795f9SHong Zhang   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
1567547795f9SHong Zhang   PetscScalar       *zb,*x,*z,*xb = 0,x1,x2,x3,x4,x5;
1568547795f9SHong Zhang   MatScalar         *v;
1569547795f9SHong Zhang   PetscErrorCode    ierr;
1570547795f9SHong Zhang   PetscInt          mbs,i,*idx,*ii,rval,bs=A->rmap->bs,j,n,bs2=a->bs2,*ib,*ridx=PETSC_NULL;
1571547795f9SHong Zhang   Mat_CompressedRow cprow = a->compressedrow;
1572547795f9SHong Zhang   PetscTruth        usecprow=cprow.use;
1573547795f9SHong Zhang 
1574547795f9SHong Zhang   PetscFunctionBegin;
1575547795f9SHong Zhang   if (yy != zz) { ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr); }
1576547795f9SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1577547795f9SHong Zhang   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
1578547795f9SHong Zhang 
1579547795f9SHong Zhang   idx = a->j;
1580547795f9SHong Zhang   v   = a->a;
1581547795f9SHong Zhang   if (usecprow){
1582547795f9SHong Zhang     mbs  = cprow.nrows;
1583547795f9SHong Zhang     ii   = cprow.i;
1584547795f9SHong Zhang     ridx = cprow.rindex;
1585547795f9SHong Zhang   } else {
1586547795f9SHong Zhang     mbs=a->mbs;
1587547795f9SHong Zhang     ii = a->i;
1588547795f9SHong Zhang     xb = x;
1589547795f9SHong Zhang   }
1590547795f9SHong Zhang 
1591547795f9SHong Zhang   switch (bs) {
1592547795f9SHong Zhang   case 1:
1593547795f9SHong Zhang     for (i=0; i<mbs; i++) {
1594547795f9SHong Zhang       if (usecprow) xb = x + ridx[i];
1595547795f9SHong Zhang       x1 = xb[0];
1596547795f9SHong Zhang       ib = idx + ii[0];
1597547795f9SHong Zhang       n  = ii[1] - ii[0]; ii++;
1598547795f9SHong Zhang       for (j=0; j<n; j++) {
1599547795f9SHong Zhang         rval    = ib[j];
1600547795f9SHong Zhang         z[rval] += PetscConj(*v) * x1;
1601547795f9SHong Zhang         v++;
1602547795f9SHong Zhang       }
1603547795f9SHong Zhang       if (!usecprow) xb++;
1604547795f9SHong Zhang     }
1605547795f9SHong Zhang     break;
1606547795f9SHong Zhang   case 2:
1607547795f9SHong Zhang     for (i=0; i<mbs; i++) {
1608547795f9SHong Zhang       if (usecprow) xb = x + 2*ridx[i];
1609547795f9SHong Zhang       x1 = xb[0]; x2 = xb[1];
1610547795f9SHong Zhang       ib = idx + ii[0];
1611547795f9SHong Zhang       n  = ii[1] - ii[0]; ii++;
1612547795f9SHong Zhang       for (j=0; j<n; j++) {
1613547795f9SHong Zhang         rval      = ib[j]*2;
1614547795f9SHong Zhang         z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2;
1615547795f9SHong Zhang         z[rval++] += PetscConj(v[2])*x1 + PetscConj(v[3])*x2;
1616547795f9SHong Zhang         v  += 4;
1617547795f9SHong Zhang       }
1618547795f9SHong Zhang       if (!usecprow) xb += 2;
1619547795f9SHong Zhang     }
1620547795f9SHong Zhang     break;
1621547795f9SHong Zhang   case 3:
1622547795f9SHong Zhang     for (i=0; i<mbs; i++) {
1623547795f9SHong Zhang       if (usecprow) xb = x + 3*ridx[i];
1624547795f9SHong Zhang       x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
1625547795f9SHong Zhang       ib = idx + ii[0];
1626547795f9SHong Zhang       n  = ii[1] - ii[0]; ii++;
1627547795f9SHong Zhang       for (j=0; j<n; j++) {
1628547795f9SHong Zhang         rval      = ib[j]*3;
1629547795f9SHong Zhang         z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2 + PetscConj(v[2])*x3;
1630547795f9SHong Zhang         z[rval++] += PetscConj(v[3])*x1 + PetscConj(v[4])*x2 + PetscConj(v[5])*x3;
1631547795f9SHong Zhang         z[rval++] += PetscConj(v[6])*x1 + PetscConj(v[7])*x2 + PetscConj(v[8])*x3;
1632547795f9SHong Zhang         v  += 9;
1633547795f9SHong Zhang       }
1634547795f9SHong Zhang       if (!usecprow) xb += 3;
1635547795f9SHong Zhang     }
1636547795f9SHong Zhang     break;
1637547795f9SHong Zhang   case 4:
1638547795f9SHong Zhang     for (i=0; i<mbs; i++) {
1639547795f9SHong Zhang       if (usecprow) xb = x + 4*ridx[i];
1640547795f9SHong Zhang       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
1641547795f9SHong Zhang       ib = idx + ii[0];
1642547795f9SHong Zhang       n  = ii[1] - ii[0]; ii++;
1643547795f9SHong Zhang       for (j=0; j<n; j++) {
1644547795f9SHong Zhang         rval      = ib[j]*4;
1645547795f9SHong Zhang         z[rval++] +=  PetscConj(v[0])*x1 + PetscConj(v[1])*x2  + PetscConj(v[2])*x3  + PetscConj(v[3])*x4;
1646547795f9SHong Zhang         z[rval++] +=  PetscConj(v[4])*x1 + PetscConj(v[5])*x2  + PetscConj(v[6])*x3  + PetscConj(v[7])*x4;
1647547795f9SHong Zhang         z[rval++] +=  PetscConj(v[8])*x1 + PetscConj(v[9])*x2  + PetscConj(v[10])*x3 + PetscConj(v[11])*x4;
1648547795f9SHong Zhang         z[rval++] += PetscConj(v[12])*x1 + PetscConj(v[13])*x2 + PetscConj(v[14])*x3 + PetscConj(v[15])*x4;
1649547795f9SHong Zhang         v  += 16;
1650547795f9SHong Zhang       }
1651547795f9SHong Zhang       if (!usecprow) xb += 4;
1652547795f9SHong Zhang     }
1653547795f9SHong Zhang     break;
1654547795f9SHong Zhang   case 5:
1655547795f9SHong Zhang     for (i=0; i<mbs; i++) {
1656547795f9SHong Zhang       if (usecprow) xb = x + 5*ridx[i];
1657547795f9SHong Zhang       x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
1658547795f9SHong Zhang       x4 = xb[3]; x5 = xb[4];
1659547795f9SHong Zhang       ib = idx + ii[0];
1660547795f9SHong Zhang       n  = ii[1] - ii[0]; ii++;
1661547795f9SHong Zhang       for (j=0; j<n; j++) {
1662547795f9SHong Zhang         rval      = ib[j]*5;
1663547795f9SHong Zhang         z[rval++] +=  PetscConj(v[0])*x1 +  PetscConj(v[1])*x2 +  PetscConj(v[2])*x3 +  PetscConj(v[3])*x4 +  PetscConj(v[4])*x5;
1664547795f9SHong Zhang         z[rval++] +=  PetscConj(v[5])*x1 +  PetscConj(v[6])*x2 +  PetscConj(v[7])*x3 +  PetscConj(v[8])*x4 +  PetscConj(v[9])*x5;
1665547795f9SHong Zhang         z[rval++] += PetscConj(v[10])*x1 + PetscConj(v[11])*x2 + PetscConj(v[12])*x3 + PetscConj(v[13])*x4 + PetscConj(v[14])*x5;
1666547795f9SHong Zhang         z[rval++] += PetscConj(v[15])*x1 + PetscConj(v[16])*x2 + PetscConj(v[17])*x3 + PetscConj(v[18])*x4 + PetscConj(v[19])*x5;
1667547795f9SHong Zhang         z[rval++] += PetscConj(v[20])*x1 + PetscConj(v[21])*x2 + PetscConj(v[22])*x3 + PetscConj(v[23])*x4 + PetscConj(v[24])*x5;
1668547795f9SHong Zhang         v  += 25;
1669547795f9SHong Zhang       }
1670547795f9SHong Zhang       if (!usecprow) xb += 5;
1671547795f9SHong Zhang     }
1672547795f9SHong Zhang     break;
1673547795f9SHong Zhang   default: {      /* block sizes larger than 5 by 5 are handled by BLAS */
1674547795f9SHong Zhang       PetscInt     ncols,k;
1675547795f9SHong Zhang       PetscScalar  *work,*workt,*xtmp;
1676547795f9SHong Zhang 
1677547795f9SHong Zhang       SETERRQ(PETSC_ERR_SUP,"block size larger than 5 is not supported yet");
1678547795f9SHong Zhang       if (!a->mult_work) {
1679547795f9SHong Zhang         k = PetscMax(A->rmap->n,A->cmap->n);
1680547795f9SHong Zhang         ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
1681547795f9SHong Zhang       }
1682547795f9SHong Zhang       work = a->mult_work;
1683547795f9SHong Zhang       xtmp = x;
1684547795f9SHong Zhang       for (i=0; i<mbs; i++) {
1685547795f9SHong Zhang         n     = ii[1] - ii[0]; ii++;
1686547795f9SHong Zhang         ncols = n*bs;
1687547795f9SHong Zhang         ierr  = PetscMemzero(work,ncols*sizeof(PetscScalar));CHKERRQ(ierr);
1688547795f9SHong Zhang         if (usecprow) {
1689547795f9SHong Zhang           xtmp = x + bs*ridx[i];
1690547795f9SHong Zhang         }
1691547795f9SHong Zhang         Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,xtmp,v,work);
1692547795f9SHong Zhang         /* BLASgemv_("T",&bs,&ncols,&_DOne,v,&bs,xtmp,&_One,&_DOne,work,&_One); */
1693547795f9SHong Zhang         v += n*bs2;
1694547795f9SHong Zhang         if (!usecprow) xtmp += bs;
1695547795f9SHong Zhang         workt = work;
1696547795f9SHong Zhang         for (j=0; j<n; j++) {
1697547795f9SHong Zhang           zb = z + bs*(*idx++);
1698547795f9SHong Zhang           for (k=0; k<bs; k++) zb[k] += workt[k] ;
1699547795f9SHong Zhang           workt += bs;
1700547795f9SHong Zhang         }
1701547795f9SHong Zhang       }
1702547795f9SHong Zhang     }
1703547795f9SHong Zhang   }
1704547795f9SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1705547795f9SHong Zhang   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
1706547795f9SHong Zhang   ierr = PetscLogFlops(2.0*a->nz*a->bs2);CHKERRQ(ierr);
1707547795f9SHong Zhang   PetscFunctionReturn(0);
1708547795f9SHong Zhang }
1709547795f9SHong Zhang 
1710547795f9SHong Zhang #undef __FUNCT__
17114a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqBAIJ"
1712dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqBAIJ(Mat A,Vec xx,Vec yy,Vec zz)
17132d61bbb3SSatish Balay {
17142d61bbb3SSatish Balay   Mat_SeqBAIJ       *a = (Mat_SeqBAIJ*)A->data;
1715dcf5cc72SBarry Smith   PetscScalar       *zb,*x,*z,*xb = 0,x1,x2,x3,x4,x5;
17163f1db9ecSBarry Smith   MatScalar         *v;
17176849ba73SBarry Smith   PetscErrorCode    ierr;
1718d0f46423SBarry Smith   PetscInt          mbs,i,*idx,*ii,rval,bs=A->rmap->bs,j,n,bs2=a->bs2,*ib,*ridx=PETSC_NULL;
17193447b6efSHong Zhang   Mat_CompressedRow cprow = a->compressedrow;
17203447b6efSHong Zhang   PetscTruth        usecprow=cprow.use;
17212d61bbb3SSatish Balay 
17222d61bbb3SSatish Balay   PetscFunctionBegin;
17236a65c766SHong Zhang   if (yy != zz) { ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr); }
17243447b6efSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
17253447b6efSHong Zhang   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
17262d61bbb3SSatish Balay 
17272d61bbb3SSatish Balay   idx = a->j;
17282d61bbb3SSatish Balay   v   = a->a;
17293447b6efSHong Zhang   if (usecprow){
17303447b6efSHong Zhang     mbs  = cprow.nrows;
17313447b6efSHong Zhang     ii   = cprow.i;
17327b2bb3b9SHong Zhang     ridx = cprow.rindex;
17333447b6efSHong Zhang   } else {
17343447b6efSHong Zhang     mbs=a->mbs;
17352d61bbb3SSatish Balay     ii = a->i;
1736f1af5d2fSBarry Smith     xb = x;
17373447b6efSHong Zhang   }
17382d61bbb3SSatish Balay 
17392d61bbb3SSatish Balay   switch (bs) {
17402d61bbb3SSatish Balay   case 1:
17412d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
17427b2bb3b9SHong Zhang       if (usecprow) xb = x + ridx[i];
1743f1af5d2fSBarry Smith       x1 = xb[0];
17443447b6efSHong Zhang       ib = idx + ii[0];
17453447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
17462d61bbb3SSatish Balay       for (j=0; j<n; j++) {
17472d61bbb3SSatish Balay         rval    = ib[j];
1748f1af5d2fSBarry Smith         z[rval] += *v * x1;
1749f1af5d2fSBarry Smith         v++;
17502d61bbb3SSatish Balay       }
17513447b6efSHong Zhang       if (!usecprow) xb++;
17522d61bbb3SSatish Balay     }
17532d61bbb3SSatish Balay     break;
17542d61bbb3SSatish Balay   case 2:
17552d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
17567b2bb3b9SHong Zhang       if (usecprow) xb = x + 2*ridx[i];
1757f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1];
17583447b6efSHong Zhang       ib = idx + ii[0];
17593447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
17602d61bbb3SSatish Balay       for (j=0; j<n; j++) {
17612d61bbb3SSatish Balay         rval      = ib[j]*2;
17622d61bbb3SSatish Balay         z[rval++] += v[0]*x1 + v[1]*x2;
17632d61bbb3SSatish Balay         z[rval++] += v[2]*x1 + v[3]*x2;
17642d61bbb3SSatish Balay         v  += 4;
17652d61bbb3SSatish Balay       }
17663447b6efSHong Zhang       if (!usecprow) xb += 2;
17672d61bbb3SSatish Balay     }
17682d61bbb3SSatish Balay     break;
17692d61bbb3SSatish Balay   case 3:
17702d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
17717b2bb3b9SHong Zhang       if (usecprow) xb = x + 3*ridx[i];
1772f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
17733447b6efSHong Zhang       ib = idx + ii[0];
17743447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
17752d61bbb3SSatish Balay       for (j=0; j<n; j++) {
17762d61bbb3SSatish Balay         rval      = ib[j]*3;
17772d61bbb3SSatish Balay         z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3;
17782d61bbb3SSatish Balay         z[rval++] += v[3]*x1 + v[4]*x2 + v[5]*x3;
17792d61bbb3SSatish Balay         z[rval++] += v[6]*x1 + v[7]*x2 + v[8]*x3;
17802d61bbb3SSatish Balay         v  += 9;
17812d61bbb3SSatish Balay       }
17823447b6efSHong Zhang       if (!usecprow) xb += 3;
17832d61bbb3SSatish Balay     }
17842d61bbb3SSatish Balay     break;
17852d61bbb3SSatish Balay   case 4:
17862d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
17877b2bb3b9SHong Zhang       if (usecprow) xb = x + 4*ridx[i];
1788f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
17893447b6efSHong Zhang       ib = idx + ii[0];
17903447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
17912d61bbb3SSatish Balay       for (j=0; j<n; j++) {
17922d61bbb3SSatish Balay         rval      = ib[j]*4;
17932d61bbb3SSatish Balay         z[rval++] +=  v[0]*x1 +  v[1]*x2 +  v[2]*x3 +  v[3]*x4;
17942d61bbb3SSatish Balay         z[rval++] +=  v[4]*x1 +  v[5]*x2 +  v[6]*x3 +  v[7]*x4;
17952d61bbb3SSatish Balay         z[rval++] +=  v[8]*x1 +  v[9]*x2 + v[10]*x3 + v[11]*x4;
17962d61bbb3SSatish Balay         z[rval++] += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4;
17972d61bbb3SSatish Balay         v  += 16;
17982d61bbb3SSatish Balay       }
17993447b6efSHong Zhang       if (!usecprow) xb += 4;
18002d61bbb3SSatish Balay     }
18012d61bbb3SSatish Balay     break;
18022d61bbb3SSatish Balay   case 5:
18032d61bbb3SSatish Balay     for (i=0; i<mbs; i++) {
18047b2bb3b9SHong Zhang       if (usecprow) xb = x + 5*ridx[i];
1805f1af5d2fSBarry Smith       x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
18062d61bbb3SSatish Balay       x4 = xb[3]; x5 = xb[4];
18073447b6efSHong Zhang       ib = idx + ii[0];
18083447b6efSHong Zhang       n  = ii[1] - ii[0]; ii++;
18092d61bbb3SSatish Balay       for (j=0; j<n; j++) {
18102d61bbb3SSatish Balay         rval      = ib[j]*5;
18112d61bbb3SSatish Balay         z[rval++] +=  v[0]*x1 +  v[1]*x2 +  v[2]*x3 +  v[3]*x4 +  v[4]*x5;
18122d61bbb3SSatish Balay         z[rval++] +=  v[5]*x1 +  v[6]*x2 +  v[7]*x3 +  v[8]*x4 +  v[9]*x5;
18132d61bbb3SSatish Balay         z[rval++] += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4 + v[14]*x5;
18142d61bbb3SSatish Balay         z[rval++] += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4 + v[19]*x5;
18152d61bbb3SSatish Balay         z[rval++] += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4 + v[24]*x5;
18162d61bbb3SSatish Balay         v  += 25;
18172d61bbb3SSatish Balay       }
18183447b6efSHong Zhang       if (!usecprow) xb += 5;
18192d61bbb3SSatish Balay     }
18202d61bbb3SSatish Balay     break;
1821f1af5d2fSBarry Smith   default: {      /* block sizes larger then 5 by 5 are handled by BLAS */
1822690b6cddSBarry Smith       PetscInt     ncols,k;
18233447b6efSHong Zhang       PetscScalar  *work,*workt,*xtmp;
18243f1db9ecSBarry Smith 
18252d61bbb3SSatish Balay       if (!a->mult_work) {
1826d0f46423SBarry Smith         k = PetscMax(A->rmap->n,A->cmap->n);
182787828ca2SBarry Smith         ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
18282d61bbb3SSatish Balay       }
18292d61bbb3SSatish Balay       work = a->mult_work;
18303447b6efSHong Zhang       xtmp = x;
18312d61bbb3SSatish Balay       for (i=0; i<mbs; i++) {
18322d61bbb3SSatish Balay         n     = ii[1] - ii[0]; ii++;
18332d61bbb3SSatish Balay         ncols = n*bs;
183487828ca2SBarry Smith         ierr  = PetscMemzero(work,ncols*sizeof(PetscScalar));CHKERRQ(ierr);
18353447b6efSHong Zhang         if (usecprow) {
18367b2bb3b9SHong Zhang           xtmp = x + bs*ridx[i];
18373447b6efSHong Zhang         }
18383447b6efSHong Zhang         Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,xtmp,v,work);
183971044d3cSBarry Smith         /* BLASgemv_("T",&bs,&ncols,&_DOne,v,&bs,xtmp,&_One,&_DOne,work,&_One); */
18402d61bbb3SSatish Balay         v += n*bs2;
18413447b6efSHong Zhang         if (!usecprow) xtmp += bs;
18422d61bbb3SSatish Balay         workt = work;
18432d61bbb3SSatish Balay         for (j=0; j<n; j++) {
18442d61bbb3SSatish Balay           zb = z + bs*(*idx++);
18452d61bbb3SSatish Balay           for (k=0; k<bs; k++) zb[k] += workt[k] ;
18462d61bbb3SSatish Balay           workt += bs;
18472d61bbb3SSatish Balay         }
18482d61bbb3SSatish Balay       }
18492d61bbb3SSatish Balay     }
18502d61bbb3SSatish Balay   }
18513447b6efSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
18523447b6efSHong Zhang   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
1853dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz*a->bs2);CHKERRQ(ierr);
18542d61bbb3SSatish Balay   PetscFunctionReturn(0);
18552d61bbb3SSatish Balay }
18562d61bbb3SSatish Balay 
18574a2ae208SSatish Balay #undef __FUNCT__
18584a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqBAIJ"
1859f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqBAIJ(Mat inA,PetscScalar alpha)
18602d61bbb3SSatish Balay {
18612d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)inA->data;
1862690b6cddSBarry Smith   PetscInt       totalnz = a->bs2*a->nz;
1863f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
1864efee365bSSatish Balay   PetscErrorCode ierr;
18650805154bSBarry Smith   PetscBLASInt   one = 1,tnz = PetscBLASIntCast(totalnz);
18662d61bbb3SSatish Balay 
18672d61bbb3SSatish Balay   PetscFunctionBegin;
1868f4df32b1SMatthew Knepley   BLASscal_(&tnz,&oalpha,a->a,&one);
1869efee365bSSatish Balay   ierr = PetscLogFlops(totalnz);CHKERRQ(ierr);
18702d61bbb3SSatish Balay   PetscFunctionReturn(0);
18712d61bbb3SSatish Balay }
18722d61bbb3SSatish Balay 
18734a2ae208SSatish Balay #undef __FUNCT__
18744a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqBAIJ"
1875dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqBAIJ(Mat A,NormType type,PetscReal *norm)
18762d61bbb3SSatish Balay {
18778a62d963SHong Zhang   PetscErrorCode ierr;
18782d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
18793f1db9ecSBarry Smith   MatScalar      *v = a->a;
1880329f5518SBarry Smith   PetscReal      sum = 0.0;
1881d0f46423SBarry Smith   PetscInt       i,j,k,bs=A->rmap->bs,nz=a->nz,bs2=a->bs2,k1;
18822d61bbb3SSatish Balay 
18832d61bbb3SSatish Balay   PetscFunctionBegin;
18842d61bbb3SSatish Balay   if (type == NORM_FROBENIUS) {
18852d61bbb3SSatish Balay     for (i=0; i< bs2*nz; i++) {
1886aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
1887329f5518SBarry Smith       sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
18882d61bbb3SSatish Balay #else
18892d61bbb3SSatish Balay       sum += (*v)*(*v); v++;
18902d61bbb3SSatish Balay #endif
18912d61bbb3SSatish Balay     }
18922d61bbb3SSatish Balay     *norm = sqrt(sum);
18938a62d963SHong Zhang   } else if (type == NORM_1) { /* maximum column sum */
18948a62d963SHong Zhang     PetscReal *tmp;
18958a62d963SHong Zhang     PetscInt  *bcol = a->j;
1896d0f46423SBarry Smith     ierr = PetscMalloc((A->cmap->n+1)*sizeof(PetscReal),&tmp);CHKERRQ(ierr);
1897d0f46423SBarry Smith     ierr = PetscMemzero(tmp,A->cmap->n*sizeof(PetscReal));CHKERRQ(ierr);
18988a62d963SHong Zhang     for (i=0; i<nz; i++){
18998a62d963SHong Zhang       for (j=0; j<bs; j++){
19008a62d963SHong Zhang         k1 = bs*(*bcol) + j; /* column index */
19018a62d963SHong Zhang         for (k=0; k<bs; k++){
19028a62d963SHong Zhang           tmp[k1] += PetscAbsScalar(*v); v++;
19038a62d963SHong Zhang         }
19048a62d963SHong Zhang       }
19058a62d963SHong Zhang       bcol++;
19068a62d963SHong Zhang     }
19078a62d963SHong Zhang     *norm = 0.0;
1908d0f46423SBarry Smith     for (j=0; j<A->cmap->n; j++) {
19098a62d963SHong Zhang       if (tmp[j] > *norm) *norm = tmp[j];
19108a62d963SHong Zhang     }
19118a62d963SHong Zhang     ierr = PetscFree(tmp);CHKERRQ(ierr);
1912596552b5SBarry Smith   } else if (type == NORM_INFINITY) { /* maximum row sum */
1913596552b5SBarry Smith     *norm = 0.0;
1914596552b5SBarry Smith     for (k=0; k<bs; k++) {
191574f84c7bSSatish Balay       for (j=0; j<a->mbs; j++) {
1916596552b5SBarry Smith         v = a->a + bs2*a->i[j] + k;
1917596552b5SBarry Smith         sum = 0.0;
1918596552b5SBarry Smith         for (i=0; i<a->i[j+1]-a->i[j]; i++) {
19190e90e235SBarry Smith           for (k1=0; k1<bs; k1++){
1920596552b5SBarry Smith             sum += PetscAbsScalar(*v);
1921596552b5SBarry Smith             v   += bs;
19222d61bbb3SSatish Balay           }
19230e90e235SBarry Smith         }
1924596552b5SBarry Smith         if (sum > *norm) *norm = sum;
1925596552b5SBarry Smith       }
1926596552b5SBarry Smith     }
1927596552b5SBarry Smith   } else {
192829bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"No support for this norm yet");
19292d61bbb3SSatish Balay   }
19302d61bbb3SSatish Balay   PetscFunctionReturn(0);
19312d61bbb3SSatish Balay }
19322d61bbb3SSatish Balay 
19332d61bbb3SSatish Balay 
19344a2ae208SSatish Balay #undef __FUNCT__
19354a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqBAIJ"
1936dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqBAIJ(Mat A,Mat B,PetscTruth* flg)
19372d61bbb3SSatish Balay {
19382d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ *)A->data,*b = (Mat_SeqBAIJ *)B->data;
1939dfbe8321SBarry Smith   PetscErrorCode ierr;
19402d61bbb3SSatish Balay 
19412d61bbb3SSatish Balay   PetscFunctionBegin;
19422d61bbb3SSatish Balay   /* If the  matrix/block dimensions are not equal, or no of nonzeros or shift */
1943d0f46423SBarry Smith   if ((A->rmap->N != B->rmap->N) || (A->cmap->n != B->cmap->n) || (A->rmap->bs != B->rmap->bs)|| (a->nz != b->nz)) {
1944273d9f13SBarry Smith     *flg = PETSC_FALSE;
1945273d9f13SBarry Smith     PetscFunctionReturn(0);
19462d61bbb3SSatish Balay   }
19472d61bbb3SSatish Balay 
19482d61bbb3SSatish Balay   /* if the a->i are the same */
1949690b6cddSBarry Smith   ierr = PetscMemcmp(a->i,b->i,(a->mbs+1)*sizeof(PetscInt),flg);CHKERRQ(ierr);
1950abc0a331SBarry Smith   if (!*flg) {
19510f5bd95cSBarry Smith     PetscFunctionReturn(0);
19522d61bbb3SSatish Balay   }
19532d61bbb3SSatish Balay 
19542d61bbb3SSatish Balay   /* if a->j are the same */
1955690b6cddSBarry Smith   ierr = PetscMemcmp(a->j,b->j,(a->nz)*sizeof(PetscInt),flg);CHKERRQ(ierr);
1956abc0a331SBarry Smith   if (!*flg) {
19570f5bd95cSBarry Smith     PetscFunctionReturn(0);
19582d61bbb3SSatish Balay   }
19592d61bbb3SSatish Balay   /* if a->a are the same */
1960d0f46423SBarry Smith   ierr = PetscMemcmp(a->a,b->a,(a->nz)*(A->rmap->bs)*(B->rmap->bs)*sizeof(PetscScalar),flg);CHKERRQ(ierr);
19612d61bbb3SSatish Balay   PetscFunctionReturn(0);
19622d61bbb3SSatish Balay 
19632d61bbb3SSatish Balay }
19642d61bbb3SSatish Balay 
19654a2ae208SSatish Balay #undef __FUNCT__
19664a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqBAIJ"
1967dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqBAIJ(Mat A,Vec v)
19682d61bbb3SSatish Balay {
19692d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
1970dfbe8321SBarry Smith   PetscErrorCode ierr;
1971690b6cddSBarry Smith   PetscInt       i,j,k,n,row,bs,*ai,*aj,ambs,bs2;
197287828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
19733f1db9ecSBarry Smith   MatScalar      *aa,*aa_j;
19742d61bbb3SSatish Balay 
19752d61bbb3SSatish Balay   PetscFunctionBegin;
197629bbc08cSBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1977d0f46423SBarry Smith   bs   = A->rmap->bs;
19782d61bbb3SSatish Balay   aa   = a->a;
19792d61bbb3SSatish Balay   ai   = a->i;
19802d61bbb3SSatish Balay   aj   = a->j;
19812d61bbb3SSatish Balay   ambs = a->mbs;
19822d61bbb3SSatish Balay   bs2  = a->bs2;
19832d61bbb3SSatish Balay 
19842dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
19851ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1986e1311b90SBarry Smith   ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr);
1987d0f46423SBarry Smith   if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
19882d61bbb3SSatish Balay   for (i=0; i<ambs; i++) {
19892d61bbb3SSatish Balay     for (j=ai[i]; j<ai[i+1]; j++) {
19902d61bbb3SSatish Balay       if (aj[j] == i) {
19912d61bbb3SSatish Balay         row  = i*bs;
19922d61bbb3SSatish Balay         aa_j = aa+j*bs2;
19932d61bbb3SSatish Balay         for (k=0; k<bs2; k+=(bs+1),row++) x[row] = aa_j[k];
19942d61bbb3SSatish Balay         break;
19952d61bbb3SSatish Balay       }
19962d61bbb3SSatish Balay     }
19972d61bbb3SSatish Balay   }
19981ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
19992d61bbb3SSatish Balay   PetscFunctionReturn(0);
20002d61bbb3SSatish Balay }
20012d61bbb3SSatish Balay 
20024a2ae208SSatish Balay #undef __FUNCT__
20034a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqBAIJ"
2004dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqBAIJ(Mat A,Vec ll,Vec rr)
20052d61bbb3SSatish Balay {
20062d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
200787828ca2SBarry Smith   PetscScalar    *l,*r,x,*li,*ri;
20083f1db9ecSBarry Smith   MatScalar      *aa,*v;
2009dfbe8321SBarry Smith   PetscErrorCode ierr;
2010690b6cddSBarry Smith   PetscInt       i,j,k,lm,rn,M,m,n,*ai,*aj,mbs,tmp,bs,bs2;
20112d61bbb3SSatish Balay 
20122d61bbb3SSatish Balay   PetscFunctionBegin;
20132d61bbb3SSatish Balay   ai  = a->i;
20142d61bbb3SSatish Balay   aj  = a->j;
20152d61bbb3SSatish Balay   aa  = a->a;
2016d0f46423SBarry Smith   m   = A->rmap->n;
2017d0f46423SBarry Smith   n   = A->cmap->n;
2018d0f46423SBarry Smith   bs  = A->rmap->bs;
20192d61bbb3SSatish Balay   mbs = a->mbs;
20202d61bbb3SSatish Balay   bs2 = a->bs2;
20212d61bbb3SSatish Balay   if (ll) {
20221ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
2023e1311b90SBarry Smith     ierr = VecGetLocalSize(ll,&lm);CHKERRQ(ierr);
202429bbc08cSBarry Smith     if (lm != m) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length");
20252d61bbb3SSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
20262d61bbb3SSatish Balay       M  = ai[i+1] - ai[i];
20272d61bbb3SSatish Balay       li = l + i*bs;
20282d61bbb3SSatish Balay       v  = aa + bs2*ai[i];
20292d61bbb3SSatish Balay       for (j=0; j<M; j++) { /* for each block */
20302d61bbb3SSatish Balay         for (k=0; k<bs2; k++) {
20312d61bbb3SSatish Balay           (*v++) *= li[k%bs];
20322d61bbb3SSatish Balay         }
20332d61bbb3SSatish Balay       }
20342d61bbb3SSatish Balay     }
20351ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
2036efee365bSSatish Balay     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
20372d61bbb3SSatish Balay   }
20382d61bbb3SSatish Balay 
20392d61bbb3SSatish Balay   if (rr) {
20401ebc52fbSHong Zhang     ierr = VecGetArray(rr,&r);CHKERRQ(ierr);
2041e1311b90SBarry Smith     ierr = VecGetLocalSize(rr,&rn);CHKERRQ(ierr);
204229bbc08cSBarry Smith     if (rn != n) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length");
20432d61bbb3SSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
20442d61bbb3SSatish Balay       M  = ai[i+1] - ai[i];
20452d61bbb3SSatish Balay       v  = aa + bs2*ai[i];
20462d61bbb3SSatish Balay       for (j=0; j<M; j++) { /* for each block */
20472d61bbb3SSatish Balay         ri = r + bs*aj[ai[i]+j];
20482d61bbb3SSatish Balay         for (k=0; k<bs; k++) {
20492d61bbb3SSatish Balay           x = ri[k];
20502d61bbb3SSatish Balay           for (tmp=0; tmp<bs; tmp++) (*v++) *= x;
20512d61bbb3SSatish Balay         }
20522d61bbb3SSatish Balay       }
20532d61bbb3SSatish Balay     }
20541ebc52fbSHong Zhang     ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr);
2055efee365bSSatish Balay     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
20562d61bbb3SSatish Balay   }
20572d61bbb3SSatish Balay   PetscFunctionReturn(0);
20582d61bbb3SSatish Balay }
20592d61bbb3SSatish Balay 
20602d61bbb3SSatish Balay 
20614a2ae208SSatish Balay #undef __FUNCT__
20624a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqBAIJ"
2063dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqBAIJ(Mat A,MatInfoType flag,MatInfo *info)
20642d61bbb3SSatish Balay {
20652d61bbb3SSatish Balay   Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data;
20662d61bbb3SSatish Balay 
20672d61bbb3SSatish Balay   PetscFunctionBegin;
20682d61bbb3SSatish Balay   info->block_size     = a->bs2;
20692d61bbb3SSatish Balay   info->nz_allocated   = a->maxnz;
20702d61bbb3SSatish Balay   info->nz_used        = a->bs2*a->nz;
20712d61bbb3SSatish Balay   info->nz_unneeded    = (double)(info->nz_allocated - info->nz_used);
20722d61bbb3SSatish Balay   info->assemblies   = A->num_ass;
20732d61bbb3SSatish Balay   info->mallocs      = a->reallocs;
20747adad957SLisandro Dalcin   info->memory       = ((PetscObject)A)->mem;
20752d61bbb3SSatish Balay   if (A->factor) {
20762d61bbb3SSatish Balay     info->fill_ratio_given  = A->info.fill_ratio_given;
20772d61bbb3SSatish Balay     info->fill_ratio_needed = A->info.fill_ratio_needed;
20782d61bbb3SSatish Balay     info->factor_mallocs    = A->info.factor_mallocs;
20792d61bbb3SSatish Balay   } else {
20802d61bbb3SSatish Balay     info->fill_ratio_given  = 0;
20812d61bbb3SSatish Balay     info->fill_ratio_needed = 0;
20822d61bbb3SSatish Balay     info->factor_mallocs    = 0;
20832d61bbb3SSatish Balay   }
20842d61bbb3SSatish Balay   PetscFunctionReturn(0);
20852d61bbb3SSatish Balay }
20862d61bbb3SSatish Balay 
20872d61bbb3SSatish Balay 
20884a2ae208SSatish Balay #undef __FUNCT__
20894a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqBAIJ"
2090dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqBAIJ(Mat A)
20912d61bbb3SSatish Balay {
20922d61bbb3SSatish Balay   Mat_SeqBAIJ    *a = (Mat_SeqBAIJ*)A->data;
2093dfbe8321SBarry Smith   PetscErrorCode ierr;
20942d61bbb3SSatish Balay 
20952d61bbb3SSatish Balay   PetscFunctionBegin;
2096549d3d68SSatish Balay   ierr = PetscMemzero(a->a,a->bs2*a->i[a->mbs]*sizeof(MatScalar));CHKERRQ(ierr);
20972d61bbb3SSatish Balay   PetscFunctionReturn(0);
20982d61bbb3SSatish Balay }
2099