xref: /petsc/src/mat/impls/sbaij/seq/sbaij2.c (revision f102d2d3ba4052999c8b59e4b1ab65578f0fd3c9)
1*f102d2d3SHong Zhang /*$Id: sbaij2.c,v 1.6 2000/08/16 15:15:02 balay Exp hzhang $*/
249b5e25fSSatish Balay 
349b5e25fSSatish Balay #include "petscsys.h"
449b5e25fSSatish Balay #include "src/mat/impls/baij/seq/baij.h"
549b5e25fSSatish Balay #include "src/vec/vecimpl.h"
649b5e25fSSatish Balay #include "src/inline/spops.h"
749b5e25fSSatish Balay #include "src/inline/ilu.h"
849b5e25fSSatish Balay #include "petscbt.h"
949b5e25fSSatish Balay #include "sbaij.h"
1049b5e25fSSatish Balay 
1149b5e25fSSatish Balay #undef __FUNC__
1249b5e25fSSatish Balay #define __FUNC__ "MatIncreaseOverlap_SeqSBAIJ"
1349b5e25fSSatish Balay int MatIncreaseOverlap_SeqSBAIJ(Mat A,int is_max,IS *is,int ov)
1449b5e25fSSatish Balay {
15*f102d2d3SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
1649b5e25fSSatish Balay   int         row,i,j,k,l,m,n,*idx,ierr,*nidx,isz,val,ival;
1749b5e25fSSatish Balay   int         start,end,*ai,*aj,bs,*nidx2;
1849b5e25fSSatish Balay   PetscBT     table;
1949b5e25fSSatish Balay 
2049b5e25fSSatish Balay   PetscFunctionBegin;
21*f102d2d3SHong Zhang   SETERRQ(1,1,"Function not yet written for SBAIJ format");
2249b5e25fSSatish Balay   PetscFunctionReturn(0);
2349b5e25fSSatish Balay }
2449b5e25fSSatish Balay 
2549b5e25fSSatish Balay #undef __FUNC__
2649b5e25fSSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqSBAIJ_Private"
2749b5e25fSSatish Balay int MatGetSubMatrix_SeqSBAIJ_Private(Mat A,IS isrow,IS iscol,int cs,MatReuse scall,Mat *B)
2849b5e25fSSatish Balay {
2949b5e25fSSatish Balay   Mat_SeqSBAIJ  *a = (Mat_SeqSBAIJ*)A->data,*c;
3049b5e25fSSatish Balay   int          *smap,i,k,kstart,kend,ierr,oldcols = a->mbs,*lens;
3149b5e25fSSatish Balay   int          row,mat_i,*mat_j,tcol,*mat_ilen;
3249b5e25fSSatish Balay   int          *irow,nrows,*ssmap,bs=a->bs,bs2=a->bs2;
3349b5e25fSSatish Balay   int          *aj = a->j,*ai = a->i;
3449b5e25fSSatish Balay   MatScalar    *mat_a;
3549b5e25fSSatish Balay   Mat          C;
3649b5e25fSSatish Balay   PetscTruth   flag;
3749b5e25fSSatish Balay 
3849b5e25fSSatish Balay   PetscFunctionBegin;
3949b5e25fSSatish Balay 
4049b5e25fSSatish Balay   if (isrow != iscol) SETERRA(1,0,"MatGetSubmatrices_SeqSBAIJ: For symm. format, iscol must equal isro");
4149b5e25fSSatish Balay   ierr = ISSorted(iscol,(PetscTruth*)&i);CHKERRQ(ierr);
4249b5e25fSSatish Balay   if (!i) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"IS is not sorted");
4349b5e25fSSatish Balay 
4449b5e25fSSatish Balay   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
4549b5e25fSSatish Balay   ierr = ISGetSize(isrow,&nrows);CHKERRQ(ierr);
4649b5e25fSSatish Balay 
4749b5e25fSSatish Balay   smap  = (int*)PetscMalloc((1+oldcols)*sizeof(int));CHKPTRQ(smap);
4849b5e25fSSatish Balay   ssmap = smap;
4949b5e25fSSatish Balay   lens  = (int*)PetscMalloc((1+nrows)*sizeof(int));CHKPTRQ(lens);
5049b5e25fSSatish Balay   ierr  = PetscMemzero(smap,oldcols*sizeof(int));CHKERRQ(ierr);
5149b5e25fSSatish Balay   for (i=0; i<nrows; i++) smap[irow[i]] = i+1; /* nrows = ncols */
5249b5e25fSSatish Balay   /* determine lens of each row */
5349b5e25fSSatish Balay   for (i=0; i<nrows; i++) {
5449b5e25fSSatish Balay     kstart  = ai[irow[i]];
5549b5e25fSSatish Balay     kend    = kstart + a->ilen[irow[i]];
5649b5e25fSSatish Balay     lens[i] = 0;
5749b5e25fSSatish Balay       for (k=kstart; k<kend; k++) {
5849b5e25fSSatish Balay         if (ssmap[aj[k]]) {
5949b5e25fSSatish Balay           lens[i]++;
6049b5e25fSSatish Balay         }
6149b5e25fSSatish Balay       }
6249b5e25fSSatish Balay     }
6349b5e25fSSatish Balay   /* Create and fill new matrix */
6449b5e25fSSatish Balay   if (scall == MAT_REUSE_MATRIX) {
6549b5e25fSSatish Balay     c = (Mat_SeqSBAIJ *)((*B)->data);
6649b5e25fSSatish Balay 
6749b5e25fSSatish Balay     if (c->mbs!=nrows || c->bs!=bs) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Submatrix wrong size");
6849b5e25fSSatish Balay     ierr = PetscMemcmp(c->ilen,lens,c->mbs *sizeof(int),&flag);CHKERRQ(ierr);
6949b5e25fSSatish Balay     if (flag == PETSC_FALSE) {
7049b5e25fSSatish Balay       SETERRQ(PETSC_ERR_ARG_SIZ,0,"Cannot reuse matrix. wrong no of nonzeros");
7149b5e25fSSatish Balay     }
7249b5e25fSSatish Balay     ierr = PetscMemzero(c->ilen,c->mbs*sizeof(int));CHKERRQ(ierr);
7349b5e25fSSatish Balay     C = *B;
7449b5e25fSSatish Balay   } else {
7549b5e25fSSatish Balay     ierr = MatCreateSeqSBAIJ(A->comm,bs,nrows*bs,nrows*bs,0,lens,&C);CHKERRQ(ierr);
7649b5e25fSSatish Balay   }
7749b5e25fSSatish Balay   c = (Mat_SeqSBAIJ *)(C->data);
7849b5e25fSSatish Balay   for (i=0; i<nrows; i++) {
7949b5e25fSSatish Balay     row    = irow[i];
8049b5e25fSSatish Balay     kstart = ai[row];
8149b5e25fSSatish Balay     kend   = kstart + a->ilen[row];
8249b5e25fSSatish Balay     mat_i  = c->i[i];
8349b5e25fSSatish Balay     mat_j  = c->j + mat_i;
8449b5e25fSSatish Balay     mat_a  = c->a + mat_i*bs2;
8549b5e25fSSatish Balay     mat_ilen = c->ilen + i;
8649b5e25fSSatish Balay     for (k=kstart; k<kend; k++) {
8749b5e25fSSatish Balay       if ((tcol=ssmap[a->j[k]])) {
8849b5e25fSSatish Balay         *mat_j++ = tcol - 1;
8949b5e25fSSatish Balay         ierr     = PetscMemcpy(mat_a,a->a+k*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
9049b5e25fSSatish Balay         mat_a   += bs2;
9149b5e25fSSatish Balay         (*mat_ilen)++;
9249b5e25fSSatish Balay       }
9349b5e25fSSatish Balay     }
9449b5e25fSSatish Balay   }
9549b5e25fSSatish Balay 
9649b5e25fSSatish Balay   /* Free work space */
9749b5e25fSSatish Balay   ierr = PetscFree(smap);CHKERRQ(ierr);
9849b5e25fSSatish Balay   ierr = PetscFree(lens);CHKERRQ(ierr);
9949b5e25fSSatish Balay   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10049b5e25fSSatish Balay   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10149b5e25fSSatish Balay 
10249b5e25fSSatish Balay   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
10349b5e25fSSatish Balay   *B = C;
10449b5e25fSSatish Balay   PetscFunctionReturn(0);
10549b5e25fSSatish Balay }
10649b5e25fSSatish Balay 
10749b5e25fSSatish Balay #undef __FUNC__
10849b5e25fSSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqSBAIJ"
10949b5e25fSSatish Balay int MatGetSubMatrix_SeqSBAIJ(Mat A,IS isrow,IS iscol,int cs,MatReuse scall,Mat *B)
11049b5e25fSSatish Balay {
11149b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11249b5e25fSSatish Balay   IS          is1;
11349b5e25fSSatish Balay   int         *vary,*iary,*irow,nrows,i,ierr,bs=a->bs,count;
11449b5e25fSSatish Balay 
11549b5e25fSSatish Balay   PetscFunctionBegin;
11649b5e25fSSatish Balay   if (isrow != iscol) SETERRA(1,0,"MatGetSubmatrices_SeqSBAIJ: For symm. format, iscol must equal isro");
11749b5e25fSSatish Balay 
11849b5e25fSSatish Balay   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
11949b5e25fSSatish Balay   ierr = ISGetSize(isrow,&nrows);CHKERRQ(ierr);
12049b5e25fSSatish Balay 
12149b5e25fSSatish Balay   /* Verify if the indices corespond to each element in a block
12249b5e25fSSatish Balay    and form the IS with compressed IS */
12349b5e25fSSatish Balay   vary = (int*)PetscMalloc(2*(a->mbs+1)*sizeof(int));CHKPTRQ(vary);
12449b5e25fSSatish Balay   iary = vary + a->mbs;
12549b5e25fSSatish Balay   ierr = PetscMemzero(vary,(a->mbs)*sizeof(int));CHKERRQ(ierr);
12649b5e25fSSatish Balay   for (i=0; i<nrows; i++) vary[irow[i]/bs]++;
12749b5e25fSSatish Balay 
12849b5e25fSSatish Balay   count = 0;
12949b5e25fSSatish Balay   for (i=0; i<a->mbs; i++) {
13049b5e25fSSatish Balay     if (vary[i]!=0 && vary[i]!=bs) SETERRA(1,0,"Index set does not match blocks");
13149b5e25fSSatish Balay     if (vary[i]==bs) iary[count++] = i;
13249b5e25fSSatish Balay   }
13349b5e25fSSatish Balay   ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is1);CHKERRQ(ierr);
13449b5e25fSSatish Balay 
13549b5e25fSSatish Balay   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
13649b5e25fSSatish Balay   ierr = PetscFree(vary);CHKERRQ(ierr);
13749b5e25fSSatish Balay 
13849b5e25fSSatish Balay   ierr = MatGetSubMatrix_SeqSBAIJ_Private(A,is1,is1,cs,scall,B);CHKERRQ(ierr);
13949b5e25fSSatish Balay   ISDestroy(is1);
14049b5e25fSSatish Balay   PetscFunctionReturn(0);
14149b5e25fSSatish Balay }
14249b5e25fSSatish Balay 
14349b5e25fSSatish Balay #undef __FUNC__
14449b5e25fSSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqSBAIJ"
14549b5e25fSSatish Balay int MatGetSubMatrices_SeqSBAIJ(Mat A,int n,IS *irow,IS *icol,MatReuse scall,Mat **B)
14649b5e25fSSatish Balay {
14749b5e25fSSatish Balay   int ierr,i;
14849b5e25fSSatish Balay 
14949b5e25fSSatish Balay   PetscFunctionBegin;
15049b5e25fSSatish Balay   if (scall == MAT_INITIAL_MATRIX) {
15149b5e25fSSatish Balay     *B = (Mat*)PetscMalloc((n+1)*sizeof(Mat));CHKPTRQ(*B);
15249b5e25fSSatish Balay   }
15349b5e25fSSatish Balay 
15449b5e25fSSatish Balay   for (i=0; i<n; i++) {
15549b5e25fSSatish Balay     ierr = MatGetSubMatrix_SeqSBAIJ(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr);
15649b5e25fSSatish Balay   }
15749b5e25fSSatish Balay   PetscFunctionReturn(0);
15849b5e25fSSatish Balay }
15949b5e25fSSatish Balay 
16049b5e25fSSatish Balay /* -------------------------------------------------------*/
16149b5e25fSSatish Balay /* Should check that shapes of vectors and matrices match */
16249b5e25fSSatish Balay /* -------------------------------------------------------*/
163d9eff348SSatish Balay #include "petscblaslapack.h"
16449b5e25fSSatish Balay 
16549b5e25fSSatish Balay #undef __FUNC__
16649b5e25fSSatish Balay #define __FUNC__ "MatMult_SeqSBAIJ_1"
16749b5e25fSSatish Balay int MatMult_SeqSBAIJ_1(Mat A,Vec xx,Vec zz)
16849b5e25fSSatish Balay {
16949b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
17049b5e25fSSatish Balay   Scalar          *x,*z,*xb,x1,zero=0.0;
17149b5e25fSSatish Balay   MatScalar       *v;
17249b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
17349b5e25fSSatish Balay 
17449b5e25fSSatish Balay   PetscFunctionBegin;
17549b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
17649b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
17749b5e25fSSatish Balay   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
17849b5e25fSSatish Balay 
17949b5e25fSSatish Balay   v  = a->a;
18049b5e25fSSatish Balay   xb = x;
18149b5e25fSSatish Balay 
18249b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
18349b5e25fSSatish Balay     n  = ai[1] - ai[0];  /* length of i_th row of A */
18449b5e25fSSatish Balay     x1 = xb[0];
18549b5e25fSSatish Balay     ib = aj + *ai;
18649b5e25fSSatish Balay     z[i] += *v++ * x[*ib++];   /* (diag of A)*x */
18749b5e25fSSatish Balay     for (j=1; j<n; j++) {
18849b5e25fSSatish Balay       cval    = *ib;
18949b5e25fSSatish Balay       z[cval] += *v * x1;      /* (strict lower triangular part of A)*x  */
19049b5e25fSSatish Balay       z[i] += *v++ * x[*ib++]; /* (strict upper triangular part of A)*x  */
19149b5e25fSSatish Balay     }
19249b5e25fSSatish Balay     xb++; ai++;
19349b5e25fSSatish Balay   }
19449b5e25fSSatish Balay 
19549b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
19649b5e25fSSatish Balay   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
19749b5e25fSSatish Balay   PLogFlops(2*(a->s_nz*2 - a->m) - a->m);  /* s_nz = (nz+m)/2 */
19849b5e25fSSatish Balay   PetscFunctionReturn(0);
19949b5e25fSSatish Balay }
20049b5e25fSSatish Balay 
20149b5e25fSSatish Balay #undef __FUNC__
20249b5e25fSSatish Balay #define __FUNC__ "MatMult_SeqSBAIJ_2"
20349b5e25fSSatish Balay int MatMult_SeqSBAIJ_2(Mat A,Vec xx,Vec zz)
20449b5e25fSSatish Balay {
20549b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
20649b5e25fSSatish Balay   Scalar          *x,*z,*xb,x1,x2,zero=0.0;
20749b5e25fSSatish Balay   MatScalar       *v;
20849b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
20949b5e25fSSatish Balay 
21049b5e25fSSatish Balay 
21149b5e25fSSatish Balay   PetscFunctionBegin;
21249b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
21349b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
21449b5e25fSSatish Balay   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
21549b5e25fSSatish Balay 
21649b5e25fSSatish Balay   v     = a->a;
21749b5e25fSSatish Balay   xb = x;
21849b5e25fSSatish Balay 
21949b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
22049b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
22149b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1];
22249b5e25fSSatish Balay     ib = aj + *ai;
22349b5e25fSSatish Balay     /* (diag of A)*x */
22449b5e25fSSatish Balay     z[2*i]   += v[0]*x1 + v[2]*x2;
22549b5e25fSSatish Balay     z[2*i+1] += v[2]*x1 + v[3]*x2;
22649b5e25fSSatish Balay     v += 4;
22749b5e25fSSatish Balay     for (j=1; j<n; j++) {
22849b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
22949b5e25fSSatish Balay       cval       = ib[j]*2;
23049b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2;
23149b5e25fSSatish Balay       z[cval+1]   += v[2]*x1 + v[3]*x2;
23249b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
23349b5e25fSSatish Balay       z[2*i]   += v[0]*x[cval] + v[2]*x[cval+1];
23449b5e25fSSatish Balay       z[2*i+1] += v[1]*x[cval] + v[3]*x[cval+1];
23549b5e25fSSatish Balay       v  += 4;
23649b5e25fSSatish Balay     }
23749b5e25fSSatish Balay     xb +=2; ai++;
23849b5e25fSSatish Balay   }
23949b5e25fSSatish Balay 
24049b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
24149b5e25fSSatish Balay   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
24249b5e25fSSatish Balay   PLogFlops(8*(a->s_nz*2 - a->m) - a->m);
24349b5e25fSSatish Balay   PetscFunctionReturn(0);
24449b5e25fSSatish Balay }
24549b5e25fSSatish Balay 
24649b5e25fSSatish Balay #undef __FUNC__
24749b5e25fSSatish Balay #define __FUNC__ "MatMult_SeqSBAIJ_3"
24849b5e25fSSatish Balay int MatMult_SeqSBAIJ_3(Mat A,Vec xx,Vec zz)
24949b5e25fSSatish Balay {
25049b5e25fSSatish Balay   Mat_SeqSBAIJ  *a = (Mat_SeqSBAIJ*)A->data;
25149b5e25fSSatish Balay   Scalar          *x,*z,*xb,x1,x2,x3,zero=0.0;
25249b5e25fSSatish Balay   MatScalar       *v;
25349b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
25449b5e25fSSatish Balay 
25549b5e25fSSatish Balay 
25649b5e25fSSatish Balay   PetscFunctionBegin;
25749b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
25849b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
25949b5e25fSSatish Balay   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
26049b5e25fSSatish Balay 
26149b5e25fSSatish Balay   v     = a->a;
26249b5e25fSSatish Balay   xb = x;
26349b5e25fSSatish Balay 
26449b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
26549b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
26649b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
26749b5e25fSSatish Balay     ib = aj + *ai;
26849b5e25fSSatish Balay     /* (diag of A)*x */
26949b5e25fSSatish Balay     z[3*i]   += v[0]*x1 + v[3]*x2 + v[6]*x3;
27049b5e25fSSatish Balay     z[3*i+1] += v[3]*x1 + v[4]*x2 + v[7]*x3;
27149b5e25fSSatish Balay     z[3*i+2] += v[6]*x1 + v[7]*x2 + v[8]*x3;
27249b5e25fSSatish Balay     v += 9;
27349b5e25fSSatish Balay     for (j=1; j<n; j++) {
27449b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
27549b5e25fSSatish Balay       cval       = ib[j]*3;
27649b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3;
27749b5e25fSSatish Balay       z[cval+1]   += v[3]*x1 + v[4]*x2 + v[5]*x3;
27849b5e25fSSatish Balay       z[cval+2]   += v[6]*x1 + v[7]*x2 + v[8]*x3;
27949b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
28049b5e25fSSatish Balay       z[3*i]   += v[0]*x[cval] + v[3]*x[cval+1]+ v[6]*x[cval+2];
28149b5e25fSSatish Balay       z[3*i+1] += v[1]*x[cval] + v[4]*x[cval+1]+ v[7]*x[cval+2];
28249b5e25fSSatish Balay       z[3*i+2] += v[2]*x[cval] + v[5]*x[cval+1]+ v[8]*x[cval+2];
28349b5e25fSSatish Balay       v  += 9;
28449b5e25fSSatish Balay     }
28549b5e25fSSatish Balay     xb +=3; ai++;
28649b5e25fSSatish Balay   }
28749b5e25fSSatish Balay 
28849b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
28949b5e25fSSatish Balay   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
29049b5e25fSSatish Balay   PLogFlops(18*(a->s_nz*2 - a->m) - a->m);
29149b5e25fSSatish Balay   PetscFunctionReturn(0);
29249b5e25fSSatish Balay }
29349b5e25fSSatish Balay 
29449b5e25fSSatish Balay #undef __FUNC__
29549b5e25fSSatish Balay #define __FUNC__ "MatMult_SeqSBAIJ_4"
29649b5e25fSSatish Balay int MatMult_SeqSBAIJ_4(Mat A,Vec xx,Vec zz)
29749b5e25fSSatish Balay {
29849b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
29949b5e25fSSatish Balay   Scalar          *x,*z,*xb,x1,x2,x3,x4,zero=0.0;
30049b5e25fSSatish Balay   MatScalar       *v;
30149b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
30249b5e25fSSatish Balay 
30349b5e25fSSatish Balay   PetscFunctionBegin;
30449b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
30549b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
30649b5e25fSSatish Balay   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
30749b5e25fSSatish Balay 
30849b5e25fSSatish Balay   v     = a->a;
30949b5e25fSSatish Balay   xb = x;
31049b5e25fSSatish Balay 
31149b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
31249b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
31349b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
31449b5e25fSSatish Balay     ib = aj + *ai;
31549b5e25fSSatish Balay     /* (diag of A)*x */
31649b5e25fSSatish Balay     z[4*i]   += v[0]*x1 + v[4]*x2 +  v[8]*x3 + v[12]*x4;
31749b5e25fSSatish Balay     z[4*i+1] += v[4]*x1 + v[5]*x2 +  v[9]*x3 + v[13]*x4;
31849b5e25fSSatish Balay     z[4*i+2] += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[14]*x4;
31949b5e25fSSatish Balay     z[4*i+3] += v[12]*x1+ v[13]*x2+ v[14]*x3 + v[15]*x4;
32049b5e25fSSatish Balay     v += 16;
32149b5e25fSSatish Balay     for (j=1; j<n; j++) {
32249b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
32349b5e25fSSatish Balay       cval       = ib[j]*4;
32449b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4;
32549b5e25fSSatish Balay       z[cval+1]   += v[4]*x1 + v[5]*x2 + v[6]*x3 + v[7]*x4;
32649b5e25fSSatish Balay       z[cval+2]   += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[11]*x4;
32749b5e25fSSatish Balay       z[cval+3]   += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4;
32849b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
32949b5e25fSSatish Balay       z[4*i]   += v[0]*x[cval] + v[4]*x[cval+1]+ v[8]*x[cval+2] + v[12]*x[cval+3];
33049b5e25fSSatish Balay       z[4*i+1] += v[1]*x[cval] + v[5]*x[cval+1]+ v[9]*x[cval+2] + v[13]*x[cval+3];
33149b5e25fSSatish Balay       z[4*i+2] += v[2]*x[cval] + v[6]*x[cval+1]+ v[10]*x[cval+2]+ v[14]*x[cval+3];
33249b5e25fSSatish Balay       z[4*i+3] += v[3]*x[cval] + v[7]*x[cval+1]+ v[11]*x[cval+2]+ v[15]*x[cval+3];
33349b5e25fSSatish Balay       v  += 16;
33449b5e25fSSatish Balay     }
33549b5e25fSSatish Balay     xb +=4; ai++;
33649b5e25fSSatish Balay   }
33749b5e25fSSatish Balay 
33849b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
33949b5e25fSSatish Balay   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
34049b5e25fSSatish Balay   PLogFlops(32*(a->s_nz*2 - a->m) - a->m);
34149b5e25fSSatish Balay   PetscFunctionReturn(0);
34249b5e25fSSatish Balay }
34349b5e25fSSatish Balay 
34449b5e25fSSatish Balay #undef __FUNC__
34549b5e25fSSatish Balay #define __FUNC__ "MatMult_SeqSBAIJ_5"
34649b5e25fSSatish Balay int MatMult_SeqSBAIJ_5(Mat A,Vec xx,Vec zz)
34749b5e25fSSatish Balay {
34849b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
34949b5e25fSSatish Balay   Scalar          *x,*z,*xb,x1,x2,x3,x4,x5,zero=0.0;
35049b5e25fSSatish Balay   MatScalar       *v;
35149b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
35249b5e25fSSatish Balay 
35349b5e25fSSatish Balay   PetscFunctionBegin;
35449b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
35549b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
35649b5e25fSSatish Balay   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
35749b5e25fSSatish Balay 
35849b5e25fSSatish Balay   v     = a->a;
35949b5e25fSSatish Balay   xb = x;
36049b5e25fSSatish Balay 
36149b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
36249b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
36349b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4];
36449b5e25fSSatish Balay     ib = aj + *ai;
36549b5e25fSSatish Balay     /* (diag of A)*x */
36649b5e25fSSatish Balay     z[5*i]   += v[0]*x1  + v[5]*x2 + v[10]*x3 + v[15]*x4+ v[20]*x5;
36749b5e25fSSatish Balay     z[5*i+1] += v[5]*x1  + v[6]*x2 + v[11]*x3 + v[16]*x4+ v[21]*x5;
36849b5e25fSSatish Balay     z[5*i+2] += v[10]*x1 +v[11]*x2 + v[12]*x3 + v[17]*x4+ v[22]*x5;
36949b5e25fSSatish Balay     z[5*i+3] += v[15]*x1 +v[16]*x2 + v[17]*x3 + v[18]*x4+ v[23]*x5;
37049b5e25fSSatish Balay     z[5*i+4] += v[20]*x1 +v[21]*x2 + v[22]*x3 + v[23]*x4+ v[24]*x5;
37149b5e25fSSatish Balay     v += 25;
37249b5e25fSSatish Balay     for (j=1; j<n; j++) {
37349b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
37449b5e25fSSatish Balay       cval       = ib[j]*5;
37549b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4 + v[4]*x5;
37649b5e25fSSatish Balay       z[cval+1]   += v[5]*x1 + v[6]*x2 + v[7]*x3 + v[8]*x4 + v[9]*x5;
37749b5e25fSSatish Balay       z[cval+2]   += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4+ v[14]*x5;
37849b5e25fSSatish Balay       z[cval+3]   += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4+ v[19]*x5;
37949b5e25fSSatish Balay       z[cval+4]   += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4+ v[24]*x5;
38049b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
38149b5e25fSSatish Balay       z[5*i]   +=v[0]*x[cval]+v[5]*x[cval+1]+v[10]*x[cval+2]+v[15]*x[cval+3]+v[20]*x[cval+4];
38249b5e25fSSatish Balay       z[5*i+1] +=v[1]*x[cval]+v[6]*x[cval+1]+v[11]*x[cval+2]+v[16]*x[cval+3]+v[21]*x[cval+4];
38349b5e25fSSatish Balay       z[5*i+2] +=v[2]*x[cval]+v[7]*x[cval+1]+v[12]*x[cval+2]+v[17]*x[cval+3]+v[22]*x[cval+4];
38449b5e25fSSatish Balay       z[5*i+3] +=v[3]*x[cval]+v[8]*x[cval+1]+v[13]*x[cval+2]+v[18]*x[cval+3]+v[23]*x[cval+4];
38549b5e25fSSatish Balay       z[5*i+4] +=v[4]*x[cval]+v[9]*x[cval+1]+v[14]*x[cval+2]+v[19]*x[cval+3]+v[24]*x[cval+4];
38649b5e25fSSatish Balay       v  += 25;
38749b5e25fSSatish Balay     }
38849b5e25fSSatish Balay     xb +=5; ai++;
38949b5e25fSSatish Balay   }
39049b5e25fSSatish Balay 
39149b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
39249b5e25fSSatish Balay   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
39349b5e25fSSatish Balay   PLogFlops(50*(a->s_nz*2 - a->m) - a->m);
39449b5e25fSSatish Balay   PetscFunctionReturn(0);
39549b5e25fSSatish Balay }
39649b5e25fSSatish Balay 
39749b5e25fSSatish Balay 
39849b5e25fSSatish Balay #undef __FUNC__
39949b5e25fSSatish Balay #define __FUNC__ "MatMult_SeqSBAIJ_6"
40049b5e25fSSatish Balay int MatMult_SeqSBAIJ_6(Mat A,Vec xx,Vec zz)
40149b5e25fSSatish Balay {
40249b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
40349b5e25fSSatish Balay   Scalar          *x,*z,*xb,x1,x2,x3,x4,x5,x6,zero=0.0;
40449b5e25fSSatish Balay   MatScalar       *v;
40549b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
40649b5e25fSSatish Balay 
40749b5e25fSSatish Balay   PetscFunctionBegin;
40849b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
40949b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
41049b5e25fSSatish Balay   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
41149b5e25fSSatish Balay 
41249b5e25fSSatish Balay   v     = a->a;
41349b5e25fSSatish Balay   xb = x;
41449b5e25fSSatish Balay 
41549b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
41649b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
41749b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4]; x6=xb[5];
41849b5e25fSSatish Balay     ib = aj + *ai;
41949b5e25fSSatish Balay     /* (diag of A)*x */
42049b5e25fSSatish Balay     z[6*i]   += v[0]*x1  + v[6]*x2 + v[12]*x3 + v[18]*x4+ v[24]*x5 + v[30]*x6;
42149b5e25fSSatish Balay     z[6*i+1] += v[6]*x1  + v[7]*x2 + v[13]*x3 + v[19]*x4+ v[25]*x5 + v[31]*x6;
42249b5e25fSSatish Balay     z[6*i+2] += v[12]*x1 +v[13]*x2 + v[14]*x3 + v[20]*x4+ v[26]*x5 + v[32]*x6;
42349b5e25fSSatish Balay     z[6*i+3] += v[18]*x1 +v[19]*x2 + v[20]*x3 + v[21]*x4+ v[27]*x5 + v[33]*x6;
42449b5e25fSSatish Balay     z[6*i+4] += v[24]*x1 +v[25]*x2 + v[26]*x3 + v[27]*x4+ v[28]*x5 + v[34]*x6;
42549b5e25fSSatish Balay     z[6*i+5] += v[30]*x1 +v[31]*x2 + v[32]*x3 + v[33]*x4+ v[34]*x5 + v[35]*x6;
42649b5e25fSSatish Balay     v += 36;
42749b5e25fSSatish Balay     for (j=1; j<n; j++) {
42849b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
42949b5e25fSSatish Balay       cval       = ib[j]*6;
43049b5e25fSSatish Balay       z[cval]   += v[0]*x1  + v[1]*x2 + v[2]*x3 + v[3]*x4+ v[4]*x5 + v[5]*x6;
43149b5e25fSSatish Balay       z[cval+1] += v[6]*x1  + v[7]*x2 + v[8]*x3 + v[9]*x4+ v[10]*x5 + v[11]*x6;
43249b5e25fSSatish Balay       z[cval+2] += v[12]*x1  + v[13]*x2 + v[14]*x3 + v[15]*x4+ v[16]*x5 + v[17]*x6;
43349b5e25fSSatish Balay       z[cval+3] += v[18]*x1  + v[19]*x2 + v[20]*x3 + v[21]*x4+ v[22]*x5 + v[23]*x6;
43449b5e25fSSatish Balay       z[cval+4] += v[24]*x1  + v[25]*x2 + v[26]*x3 + v[27]*x4+ v[28]*x5 + v[29]*x6;
43549b5e25fSSatish Balay       z[cval+5] += v[30]*x1  + v[31]*x2 + v[32]*x3 + v[33]*x4+ v[34]*x5 + v[35]*x6;
43649b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
43749b5e25fSSatish Balay       z[6*i]   +=v[0]*x[cval]+v[6]*x[cval+1]+v[12]*x[cval+2]+v[18]*x[cval+3]+v[24]*x[cval+4]+v[30]*x[cval+5];
43849b5e25fSSatish Balay       z[6*i+1] +=v[1]*x[cval]+v[7]*x[cval+1]+v[13]*x[cval+2]+v[19]*x[cval+3]+v[25]*x[cval+4]+v[31]*x[cval+5];
43949b5e25fSSatish Balay       z[6*i+2] +=v[2]*x[cval]+v[8]*x[cval+1]+v[14]*x[cval+2]+v[20]*x[cval+3]+v[26]*x[cval+4]+v[32]*x[cval+5];
44049b5e25fSSatish Balay       z[6*i+3] +=v[3]*x[cval]+v[9]*x[cval+1]+v[15]*x[cval+2]+v[21]*x[cval+3]+v[27]*x[cval+4]+v[33]*x[cval+5];
44149b5e25fSSatish Balay       z[6*i+4] +=v[4]*x[cval]+v[10]*x[cval+1]+v[16]*x[cval+2]+v[22]*x[cval+3]+v[28]*x[cval+4]+v[34]*x[cval+5];
44249b5e25fSSatish Balay       z[6*i+5] +=v[5]*x[cval]+v[11]*x[cval+1]+v[17]*x[cval+2]+v[23]*x[cval+3]+v[29]*x[cval+4]+v[35]*x[cval+5];
44349b5e25fSSatish Balay       v  += 36;
44449b5e25fSSatish Balay     }
44549b5e25fSSatish Balay     xb +=6; ai++;
44649b5e25fSSatish Balay   }
44749b5e25fSSatish Balay 
44849b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
44949b5e25fSSatish Balay   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
45049b5e25fSSatish Balay   PLogFlops(72*(a->s_nz*2 - a->m) - a->m);
45149b5e25fSSatish Balay   PetscFunctionReturn(0);
45249b5e25fSSatish Balay }
45349b5e25fSSatish Balay #undef __FUNC__
45449b5e25fSSatish Balay #define __FUNC__ "MatMult_SeqSBAIJ_7"
45549b5e25fSSatish Balay int MatMult_SeqSBAIJ_7(Mat A,Vec xx,Vec zz)
45649b5e25fSSatish Balay {
45749b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
45849b5e25fSSatish Balay   Scalar          *x,*z,*xb,x1,x2,x3,x4,x5,x6,x7,zero=0.0;
45949b5e25fSSatish Balay   MatScalar       *v;
46049b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
46149b5e25fSSatish Balay 
46249b5e25fSSatish Balay   PetscFunctionBegin;
46349b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
46449b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
46549b5e25fSSatish Balay   ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
46649b5e25fSSatish Balay 
46749b5e25fSSatish Balay   v     = a->a;
46849b5e25fSSatish Balay   xb = x;
46949b5e25fSSatish Balay 
47049b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
47149b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
47249b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4]; x6=xb[5]; x7=xb[6];
47349b5e25fSSatish Balay     ib = aj + *ai;
47449b5e25fSSatish Balay     /* (diag of A)*x */
47549b5e25fSSatish Balay     z[7*i]   += v[0]*x1 + v[7]*x2 + v[14]*x3 + v[21]*x4+ v[28]*x5 + v[35]*x6+ v[42]*x7;
47649b5e25fSSatish Balay     z[7*i+1] += v[7]*x1 + v[8]*x2 + v[15]*x3 + v[22]*x4+ v[29]*x5 + v[36]*x6+ v[43]*x7;
47749b5e25fSSatish Balay     z[7*i+2] += v[14]*x1+ v[15]*x2 +v[16]*x3 + v[23]*x4+ v[30]*x5 + v[37]*x6+ v[44]*x7;
47849b5e25fSSatish Balay     z[7*i+3] += v[21]*x1+ v[22]*x2 +v[23]*x3 + v[24]*x4+ v[31]*x5 + v[38]*x6+ v[45]*x7;
47949b5e25fSSatish Balay     z[7*i+4] += v[28]*x1+ v[29]*x2 +v[30]*x3 + v[31]*x4+ v[32]*x5 + v[39]*x6+ v[46]*x7;
48049b5e25fSSatish Balay     z[7*i+5] += v[35]*x1+ v[36]*x2 +v[37]*x3 + v[38]*x4+ v[39]*x5 + v[40]*x6+ v[47]*x7;
48149b5e25fSSatish Balay     z[7*i+6] += v[42]*x1+ v[43]*x2 +v[44]*x3 + v[45]*x4+ v[46]*x5 + v[47]*x6+ v[48]*x7;
48249b5e25fSSatish Balay     v += 49;
48349b5e25fSSatish Balay     for (j=1; j<n; j++) {
48449b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
48549b5e25fSSatish Balay       cval       = ib[j]*7;
48649b5e25fSSatish Balay       z[cval]   += v[0]*x1  + v[1]*x2 + v[2]*x3 + v[3]*x4+ v[4]*x5 + v[5]*x6+ v[6]*x7;
48749b5e25fSSatish Balay       z[cval+1] += v[7]*x1  + v[8]*x2 + v[9]*x3 + v[10]*x4+ v[11]*x5 + v[12]*x6+ v[13]*x7;
48849b5e25fSSatish Balay       z[cval+2] += v[14]*x1  + v[15]*x2 + v[16]*x3 + v[17]*x4+ v[18]*x5 + v[19]*x6+ v[20]*x7;
48949b5e25fSSatish Balay       z[cval+3] += v[21]*x1  + v[22]*x2 + v[23]*x3 + v[24]*x4+ v[25]*x5 + v[26]*x6+ v[27]*x7;
49049b5e25fSSatish Balay       z[cval+4] += v[28]*x1  + v[29]*x2 + v[30]*x3 + v[31]*x4+ v[32]*x5 + v[33]*x6+ v[34]*x7;
49149b5e25fSSatish Balay       z[cval+5] += v[35]*x1  + v[36]*x2 + v[37]*x3 + v[38]*x4+ v[39]*x5 + v[40]*x6+ v[41]*x7;
49249b5e25fSSatish Balay       z[cval+6] += v[42]*x1  + v[43]*x2 + v[44]*x3 + v[45]*x4+ v[46]*x5 + v[47]*x6+ v[48]*x7;
49349b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
49449b5e25fSSatish Balay       z[7*i]  +=v[0]*x[cval]+v[7]*x[cval+1]+v[14]*x[cval+2]+v[21]*x[cval+3]+v[28]*x[cval+4]+v[35]*x[cval+5]+v[42]*x[cval+6];
49549b5e25fSSatish Balay       z[7*i+1]+=v[1]*x[cval]+v[8]*x[cval+1]+v[15]*x[cval+2]+v[22]*x[cval+3]+v[29]*x[cval+4]+v[36]*x[cval+5]+v[43]*x[cval+6];
49649b5e25fSSatish Balay       z[7*i+2]+=v[2]*x[cval]+v[9]*x[cval+1]+v[16]*x[cval+2]+v[23]*x[cval+3]+v[30]*x[cval+4]+v[37]*x[cval+5]+v[44]*x[cval+6];
49749b5e25fSSatish Balay       z[7*i+3]+=v[3]*x[cval]+v[10]*x[cval+1]+v[17]*x[cval+2]+v[24]*x[cval+3]+v[31]*x[cval+4]+v[38]*x[cval+5]+v[45]*x[cval+6];
49849b5e25fSSatish Balay       z[7*i+4]+=v[4]*x[cval]+v[11]*x[cval+1]+v[18]*x[cval+2]+v[25]*x[cval+3]+v[32]*x[cval+4]+v[39]*x[cval+5]+v[46]*x[cval+6];
49949b5e25fSSatish Balay       z[7*i+5]+=v[5]*x[cval]+v[12]*x[cval+1]+v[19]*x[cval+2]+v[26]*x[cval+3]+v[33]*x[cval+4]+v[40]*x[cval+5]+v[47]*x[cval+6];
50049b5e25fSSatish Balay       z[7*i+6]+=v[6]*x[cval]+v[13]*x[cval+1]+v[20]*x[cval+2]+v[27]*x[cval+3]+v[34]*x[cval+4]+v[41]*x[cval+5]+v[48]*x[cval+6];
50149b5e25fSSatish Balay       v  += 49;
50249b5e25fSSatish Balay     }
50349b5e25fSSatish Balay     xb +=7; ai++;
50449b5e25fSSatish Balay   }
50549b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
50649b5e25fSSatish Balay   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
50749b5e25fSSatish Balay   PLogFlops(98*(a->s_nz*2 - a->m) - a->m);
50849b5e25fSSatish Balay   PetscFunctionReturn(0);
50949b5e25fSSatish Balay }
51049b5e25fSSatish Balay 
51149b5e25fSSatish Balay /*
51249b5e25fSSatish Balay     This will not work with MatScalar == float because it calls the BLAS
51349b5e25fSSatish Balay */
51449b5e25fSSatish Balay #undef __FUNC__
51549b5e25fSSatish Balay #define __FUNC__ "MatMult_SeqSBAIJ_N"
51649b5e25fSSatish Balay int MatMult_SeqSBAIJ_N(Mat A,Vec xx,Vec zz)
51749b5e25fSSatish Balay {
51849b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
519066653e3SSatish Balay   Scalar          *x,*x_ptr,*z,*z_ptr,*xb,*zb,*work,*workt,zero=0.0;
520066653e3SSatish Balay   MatScalar       *v;
521066653e3SSatish Balay   int             ierr,mbs=a->mbs,i,*idx,*aj,*ii,bs=a->bs,j,n,bs2=a->bs2;
522066653e3SSatish Balay   int             ncols,k;
52349b5e25fSSatish Balay 
52449b5e25fSSatish Balay   PetscFunctionBegin;
52549b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
52649b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr); x_ptr=x;
52749b5e25fSSatish Balay   ierr = VecGetArray(zz,&z);CHKERRQ(ierr); z_ptr=z;
52849b5e25fSSatish Balay 
52949b5e25fSSatish Balay   aj   = a->j;
53049b5e25fSSatish Balay   v    = a->a;
53149b5e25fSSatish Balay   ii   = a->i;
53249b5e25fSSatish Balay 
53349b5e25fSSatish Balay   if (!a->mult_work) {
53449b5e25fSSatish Balay     k = a->m;
53549b5e25fSSatish Balay     a->mult_work = (Scalar*)PetscMalloc((k+1)*sizeof(Scalar));CHKPTRQ(a->mult_work);
53649b5e25fSSatish Balay   }
53749b5e25fSSatish Balay   work = a->mult_work;
53849b5e25fSSatish Balay 
53949b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
54049b5e25fSSatish Balay     n     = ii[1] - ii[0]; ncols = n*bs;
54149b5e25fSSatish Balay     workt = work; idx=aj+ii[0];
54249b5e25fSSatish Balay 
54349b5e25fSSatish Balay     /* upper triangular part */
54449b5e25fSSatish Balay     for (j=0; j<n; j++) {
545066653e3SSatish Balay       /* col = bs*(*idx); */
54649b5e25fSSatish Balay       xb = x_ptr + bs*(*idx++);
54749b5e25fSSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
54849b5e25fSSatish Balay       workt += bs;
54949b5e25fSSatish Balay     }
55049b5e25fSSatish Balay     /* z(i*bs:(i+1)*bs-1) += A(i,:)*x */
55149b5e25fSSatish Balay     Kernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z);
55249b5e25fSSatish Balay 
55349b5e25fSSatish Balay     /* strict lower triangular part */
55449b5e25fSSatish Balay     ncols -= bs;
55549b5e25fSSatish Balay     if (ncols > 0){
55649b5e25fSSatish Balay       workt = work;
55749b5e25fSSatish Balay       ierr  = PetscMemzero(workt,ncols*sizeof(Scalar));CHKERRQ(ierr);
55849b5e25fSSatish Balay       Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,x,v+bs2,workt);
55949b5e25fSSatish Balay 
56049b5e25fSSatish Balay       idx=aj+ii[0]+1;
56149b5e25fSSatish Balay       for (j=1; j<n; j++) {
56249b5e25fSSatish Balay         zb = z_ptr + bs*(*idx);
56349b5e25fSSatish Balay         idx++;
56449b5e25fSSatish Balay         for (k=0; k<bs; k++) zb[k] += workt[k] ;
56549b5e25fSSatish Balay         workt += bs;
56649b5e25fSSatish Balay       }
56749b5e25fSSatish Balay     }
56849b5e25fSSatish Balay 
56949b5e25fSSatish Balay     x += bs; v += n*bs2; z += bs; ii++;
57049b5e25fSSatish Balay   }
57149b5e25fSSatish Balay 
57249b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
57349b5e25fSSatish Balay   ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
57449b5e25fSSatish Balay   PLogFlops(2*(a->s_nz*2 - a->m)*bs2 - a->m);
57549b5e25fSSatish Balay   PetscFunctionReturn(0);
57649b5e25fSSatish Balay }
57749b5e25fSSatish Balay 
57849b5e25fSSatish Balay #undef __FUNC__
57949b5e25fSSatish Balay #define __FUNC__ "MatMultAdd_SeqSBAIJ_1"
58049b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_1(Mat A,Vec xx,Vec yy,Vec zz)
58149b5e25fSSatish Balay {
58249b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
58349b5e25fSSatish Balay   Scalar          *x,*y,*z,*xb,x1;
58449b5e25fSSatish Balay   MatScalar       *v;
58549b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
58649b5e25fSSatish Balay 
58749b5e25fSSatish Balay   PetscFunctionBegin;
58849b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
58949b5e25fSSatish Balay   if (yy != xx) {
59049b5e25fSSatish Balay     ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
59149b5e25fSSatish Balay   } else {
59249b5e25fSSatish Balay     y = x;
59349b5e25fSSatish Balay   }
59449b5e25fSSatish Balay   if (zz != yy) {
59549b5e25fSSatish Balay     ierr = VecCopy(yy,zz);CHKERRQ(ierr);
59649b5e25fSSatish Balay     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
59749b5e25fSSatish Balay   } else {
59849b5e25fSSatish Balay     z = y;
59949b5e25fSSatish Balay   }
60049b5e25fSSatish Balay 
60149b5e25fSSatish Balay   v  = a->a;
60249b5e25fSSatish Balay   xb = x;
60349b5e25fSSatish Balay 
60449b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
60549b5e25fSSatish Balay     n  = ai[1] - ai[0];  /* length of i_th row of A */
60649b5e25fSSatish Balay     x1 = xb[0];
60749b5e25fSSatish Balay     ib = aj + *ai;
60849b5e25fSSatish Balay     z[i] += *v++ * x[*ib++];   /* (diag of A)*x */
60949b5e25fSSatish Balay     for (j=1; j<n; j++) {
61049b5e25fSSatish Balay       cval    = *ib;
61149b5e25fSSatish Balay       z[cval] += *v * x1;      /* (strict lower triangular part of A)*x  */
61249b5e25fSSatish Balay       z[i] += *v++ * x[*ib++]; /* (strict upper triangular part of A)*x  */
61349b5e25fSSatish Balay     }
61449b5e25fSSatish Balay     xb++; ai++;
61549b5e25fSSatish Balay   }
61649b5e25fSSatish Balay 
61749b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
61849b5e25fSSatish Balay   if (yy != xx) ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
61949b5e25fSSatish Balay   if (zz != yy) ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
62049b5e25fSSatish Balay 
62149b5e25fSSatish Balay   PLogFlops(2*(a->s_nz*2 - a->m));
62249b5e25fSSatish Balay   PetscFunctionReturn(0);
62349b5e25fSSatish Balay }
62449b5e25fSSatish Balay 
62549b5e25fSSatish Balay #undef __FUNC__
62649b5e25fSSatish Balay #define __FUNC__ "MatMultAdd_SeqSBAIJ_2"
62749b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_2(Mat A,Vec xx,Vec yy,Vec zz)
62849b5e25fSSatish Balay {
62949b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
63049b5e25fSSatish Balay   Scalar          *x,*y,*z,*xb,x1,x2;
63149b5e25fSSatish Balay   MatScalar       *v;
63249b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
63349b5e25fSSatish Balay 
63449b5e25fSSatish Balay   PetscFunctionBegin;
63549b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
63649b5e25fSSatish Balay   if (yy != xx) {
63749b5e25fSSatish Balay     ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
63849b5e25fSSatish Balay   } else {
63949b5e25fSSatish Balay     y = x;
64049b5e25fSSatish Balay   }
64149b5e25fSSatish Balay   if (zz != yy) {
64249b5e25fSSatish Balay     ierr = VecCopy(yy,zz);CHKERRQ(ierr);
64349b5e25fSSatish Balay     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
64449b5e25fSSatish Balay   } else {
64549b5e25fSSatish Balay     z = y;
64649b5e25fSSatish Balay   }
64749b5e25fSSatish Balay 
64849b5e25fSSatish Balay   v     = a->a;
64949b5e25fSSatish Balay   xb = x;
65049b5e25fSSatish Balay 
65149b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
65249b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
65349b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1];
65449b5e25fSSatish Balay     ib = aj + *ai;
65549b5e25fSSatish Balay     /* (diag of A)*x */
65649b5e25fSSatish Balay     z[2*i]   += v[0]*x1 + v[2]*x2;
65749b5e25fSSatish Balay     z[2*i+1] += v[2]*x1 + v[3]*x2;
65849b5e25fSSatish Balay     v += 4;
65949b5e25fSSatish Balay     for (j=1; j<n; j++) {
66049b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
66149b5e25fSSatish Balay       cval       = ib[j]*2;
66249b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2;
66349b5e25fSSatish Balay       z[cval+1]   += v[2]*x1 + v[3]*x2;
66449b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
66549b5e25fSSatish Balay       z[2*i]   += v[0]*x[cval] + v[2]*x[cval+1];
66649b5e25fSSatish Balay       z[2*i+1] += v[1]*x[cval] + v[3]*x[cval+1];
66749b5e25fSSatish Balay       v  += 4;
66849b5e25fSSatish Balay     }
66949b5e25fSSatish Balay     xb +=2; ai++;
67049b5e25fSSatish Balay   }
67149b5e25fSSatish Balay 
67249b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
67349b5e25fSSatish Balay   if (yy != xx) ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
67449b5e25fSSatish Balay   if (zz != yy) ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
67549b5e25fSSatish Balay 
67649b5e25fSSatish Balay   PLogFlops(4*(a->s_nz*2 - a->m));
67749b5e25fSSatish Balay   PetscFunctionReturn(0);
67849b5e25fSSatish Balay }
67949b5e25fSSatish Balay 
68049b5e25fSSatish Balay #undef __FUNC__
68149b5e25fSSatish Balay #define __FUNC__ "MatMultAdd_SeqSBAIJ_3"
68249b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_3(Mat A,Vec xx,Vec yy,Vec zz)
68349b5e25fSSatish Balay {
68449b5e25fSSatish Balay   Mat_SeqSBAIJ  *a = (Mat_SeqSBAIJ*)A->data;
68549b5e25fSSatish Balay   Scalar          *x,*y,*z,*xb,x1,x2,x3;
68649b5e25fSSatish Balay   MatScalar       *v;
68749b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
68849b5e25fSSatish Balay 
68949b5e25fSSatish Balay   PetscFunctionBegin;
69049b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
69149b5e25fSSatish Balay   if (yy != xx) {
69249b5e25fSSatish Balay     ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
69349b5e25fSSatish Balay   } else {
69449b5e25fSSatish Balay     y = x;
69549b5e25fSSatish Balay   }
69649b5e25fSSatish Balay   if (zz != yy) {
69749b5e25fSSatish Balay     ierr = VecCopy(yy,zz);CHKERRQ(ierr);
69849b5e25fSSatish Balay     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
69949b5e25fSSatish Balay   } else {
70049b5e25fSSatish Balay     z = y;
70149b5e25fSSatish Balay   }
70249b5e25fSSatish Balay 
70349b5e25fSSatish Balay   v     = a->a;
70449b5e25fSSatish Balay   xb = x;
70549b5e25fSSatish Balay 
70649b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
70749b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
70849b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
70949b5e25fSSatish Balay     ib = aj + *ai;
71049b5e25fSSatish Balay     /* (diag of A)*x */
71149b5e25fSSatish Balay     z[3*i]   += v[0]*x1 + v[3]*x2 + v[6]*x3;
71249b5e25fSSatish Balay     z[3*i+1] += v[3]*x1 + v[4]*x2 + v[7]*x3;
71349b5e25fSSatish Balay     z[3*i+2] += v[6]*x1 + v[7]*x2 + v[8]*x3;
71449b5e25fSSatish Balay     v += 9;
71549b5e25fSSatish Balay     for (j=1; j<n; j++) {
71649b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
71749b5e25fSSatish Balay       cval       = ib[j]*3;
71849b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3;
71949b5e25fSSatish Balay       z[cval+1]   += v[3]*x1 + v[4]*x2 + v[5]*x3;
72049b5e25fSSatish Balay       z[cval+2]   += v[6]*x1 + v[7]*x2 + v[8]*x3;
72149b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
72249b5e25fSSatish Balay       z[3*i]   += v[0]*x[cval] + v[3]*x[cval+1]+ v[6]*x[cval+2];
72349b5e25fSSatish Balay       z[3*i+1] += v[1]*x[cval] + v[4]*x[cval+1]+ v[7]*x[cval+2];
72449b5e25fSSatish Balay       z[3*i+2] += v[2]*x[cval] + v[5]*x[cval+1]+ v[8]*x[cval+2];
72549b5e25fSSatish Balay       v  += 9;
72649b5e25fSSatish Balay     }
72749b5e25fSSatish Balay     xb +=3; ai++;
72849b5e25fSSatish Balay   }
72949b5e25fSSatish Balay 
73049b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
73149b5e25fSSatish Balay   if (yy != xx) ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
73249b5e25fSSatish Balay   if (zz != yy) ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
73349b5e25fSSatish Balay 
73449b5e25fSSatish Balay   PLogFlops(18*(a->s_nz*2 - a->m));
73549b5e25fSSatish Balay   PetscFunctionReturn(0);
73649b5e25fSSatish Balay }
73749b5e25fSSatish Balay 
73849b5e25fSSatish Balay #undef __FUNC__
73949b5e25fSSatish Balay #define __FUNC__ "MatMultAdd_SeqSBAIJ_4"
74049b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_4(Mat A,Vec xx,Vec yy,Vec zz)
74149b5e25fSSatish Balay {
74249b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
74349b5e25fSSatish Balay   Scalar          *x,*y,*z,*xb,x1,x2,x3,x4;
74449b5e25fSSatish Balay   MatScalar       *v;
74549b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
74649b5e25fSSatish Balay 
74749b5e25fSSatish Balay   PetscFunctionBegin;
74849b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
74949b5e25fSSatish Balay   if (yy != xx) {
75049b5e25fSSatish Balay     ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
75149b5e25fSSatish Balay   } else {
75249b5e25fSSatish Balay     y = x;
75349b5e25fSSatish Balay   }
75449b5e25fSSatish Balay   if (zz != yy) {
75549b5e25fSSatish Balay     ierr = VecCopy(yy,zz);CHKERRQ(ierr);
75649b5e25fSSatish Balay     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
75749b5e25fSSatish Balay   } else {
75849b5e25fSSatish Balay     z = y;
75949b5e25fSSatish Balay   }
76049b5e25fSSatish Balay 
76149b5e25fSSatish Balay   v     = a->a;
76249b5e25fSSatish Balay   xb = x;
76349b5e25fSSatish Balay 
76449b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
76549b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
76649b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
76749b5e25fSSatish Balay     ib = aj + *ai;
76849b5e25fSSatish Balay     /* (diag of A)*x */
76949b5e25fSSatish Balay     z[4*i]   += v[0]*x1 + v[4]*x2 +  v[8]*x3 + v[12]*x4;
77049b5e25fSSatish Balay     z[4*i+1] += v[4]*x1 + v[5]*x2 +  v[9]*x3 + v[13]*x4;
77149b5e25fSSatish Balay     z[4*i+2] += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[14]*x4;
77249b5e25fSSatish Balay     z[4*i+3] += v[12]*x1+ v[13]*x2+ v[14]*x3 + v[15]*x4;
77349b5e25fSSatish Balay     v += 16;
77449b5e25fSSatish Balay     for (j=1; j<n; j++) {
77549b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
77649b5e25fSSatish Balay       cval       = ib[j]*4;
77749b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4;
77849b5e25fSSatish Balay       z[cval+1]   += v[4]*x1 + v[5]*x2 + v[6]*x3 + v[7]*x4;
77949b5e25fSSatish Balay       z[cval+2]   += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[11]*x4;
78049b5e25fSSatish Balay       z[cval+3]   += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4;
78149b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
78249b5e25fSSatish Balay       z[4*i]   += v[0]*x[cval] + v[4]*x[cval+1]+ v[8]*x[cval+2] + v[12]*x[cval+3];
78349b5e25fSSatish Balay       z[4*i+1] += v[1]*x[cval] + v[5]*x[cval+1]+ v[9]*x[cval+2] + v[13]*x[cval+3];
78449b5e25fSSatish Balay       z[4*i+2] += v[2]*x[cval] + v[6]*x[cval+1]+ v[10]*x[cval+2]+ v[14]*x[cval+3];
78549b5e25fSSatish Balay       z[4*i+3] += v[3]*x[cval] + v[7]*x[cval+1]+ v[11]*x[cval+2]+ v[15]*x[cval+3];
78649b5e25fSSatish Balay       v  += 16;
78749b5e25fSSatish Balay     }
78849b5e25fSSatish Balay     xb +=4; ai++;
78949b5e25fSSatish Balay   }
79049b5e25fSSatish Balay 
79149b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
79249b5e25fSSatish Balay   if (yy != xx) ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
79349b5e25fSSatish Balay   if (zz != yy) ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
79449b5e25fSSatish Balay 
79549b5e25fSSatish Balay   PLogFlops(32*(a->s_nz*2 - a->m));
79649b5e25fSSatish Balay   PetscFunctionReturn(0);
79749b5e25fSSatish Balay }
79849b5e25fSSatish Balay 
79949b5e25fSSatish Balay #undef __FUNC__
80049b5e25fSSatish Balay #define __FUNC__ "MatMultAdd_SeqSBAIJ_5"
80149b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_5(Mat A,Vec xx,Vec yy,Vec zz)
80249b5e25fSSatish Balay {
80349b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
80449b5e25fSSatish Balay   Scalar          *x,*y,*z,*xb,x1,x2,x3,x4,x5;
80549b5e25fSSatish Balay   MatScalar       *v;
80649b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
80749b5e25fSSatish Balay 
80849b5e25fSSatish Balay   PetscFunctionBegin;
80949b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
81049b5e25fSSatish Balay   if (yy != xx) {
81149b5e25fSSatish Balay     ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
81249b5e25fSSatish Balay   } else {
81349b5e25fSSatish Balay     y = x;
81449b5e25fSSatish Balay   }
81549b5e25fSSatish Balay   if (zz != yy) {
81649b5e25fSSatish Balay     ierr = VecCopy(yy,zz);CHKERRQ(ierr);
81749b5e25fSSatish Balay     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
81849b5e25fSSatish Balay   } else {
81949b5e25fSSatish Balay     z = y;
82049b5e25fSSatish Balay   }
82149b5e25fSSatish Balay 
82249b5e25fSSatish Balay   v     = a->a;
82349b5e25fSSatish Balay   xb = x;
82449b5e25fSSatish Balay 
82549b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
82649b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
82749b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4];
82849b5e25fSSatish Balay     ib = aj + *ai;
82949b5e25fSSatish Balay     /* (diag of A)*x */
83049b5e25fSSatish Balay     z[5*i]   += v[0]*x1  + v[5]*x2 + v[10]*x3 + v[15]*x4+ v[20]*x5;
83149b5e25fSSatish Balay     z[5*i+1] += v[5]*x1  + v[6]*x2 + v[11]*x3 + v[16]*x4+ v[21]*x5;
83249b5e25fSSatish Balay     z[5*i+2] += v[10]*x1 +v[11]*x2 + v[12]*x3 + v[17]*x4+ v[22]*x5;
83349b5e25fSSatish Balay     z[5*i+3] += v[15]*x1 +v[16]*x2 + v[17]*x3 + v[18]*x4+ v[23]*x5;
83449b5e25fSSatish Balay     z[5*i+4] += v[20]*x1 +v[21]*x2 + v[22]*x3 + v[23]*x4+ v[24]*x5;
83549b5e25fSSatish Balay     v += 25;
83649b5e25fSSatish Balay     for (j=1; j<n; j++) {
83749b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
83849b5e25fSSatish Balay       cval       = ib[j]*5;
83949b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4 + v[4]*x5;
84049b5e25fSSatish Balay       z[cval+1]   += v[5]*x1 + v[6]*x2 + v[7]*x3 + v[8]*x4 + v[9]*x5;
84149b5e25fSSatish Balay       z[cval+2]   += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4+ v[14]*x5;
84249b5e25fSSatish Balay       z[cval+3]   += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4+ v[19]*x5;
84349b5e25fSSatish Balay       z[cval+4]   += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4+ v[24]*x5;
84449b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
84549b5e25fSSatish Balay       z[5*i]   +=v[0]*x[cval]+v[5]*x[cval+1]+v[10]*x[cval+2]+v[15]*x[cval+3]+v[20]*x[cval+4];
84649b5e25fSSatish Balay       z[5*i+1] +=v[1]*x[cval]+v[6]*x[cval+1]+v[11]*x[cval+2]+v[16]*x[cval+3]+v[21]*x[cval+4];
84749b5e25fSSatish Balay       z[5*i+2] +=v[2]*x[cval]+v[7]*x[cval+1]+v[12]*x[cval+2]+v[17]*x[cval+3]+v[22]*x[cval+4];
84849b5e25fSSatish Balay       z[5*i+3] +=v[3]*x[cval]+v[8]*x[cval+1]+v[13]*x[cval+2]+v[18]*x[cval+3]+v[23]*x[cval+4];
84949b5e25fSSatish Balay       z[5*i+4] +=v[4]*x[cval]+v[9]*x[cval+1]+v[14]*x[cval+2]+v[19]*x[cval+3]+v[24]*x[cval+4];
85049b5e25fSSatish Balay       v  += 25;
85149b5e25fSSatish Balay     }
85249b5e25fSSatish Balay     xb +=5; ai++;
85349b5e25fSSatish Balay   }
85449b5e25fSSatish Balay 
85549b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
85649b5e25fSSatish Balay   if (yy != xx) ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
85749b5e25fSSatish Balay   if (zz != yy) ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
85849b5e25fSSatish Balay 
85949b5e25fSSatish Balay   PLogFlops(50*(a->s_nz*2 - a->m));
86049b5e25fSSatish Balay   PetscFunctionReturn(0);
86149b5e25fSSatish Balay }
86249b5e25fSSatish Balay #undef __FUNC__
86349b5e25fSSatish Balay #define __FUNC__ "MatMultAdd_SeqSBAIJ_6"
86449b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_6(Mat A,Vec xx,Vec yy,Vec zz)
86549b5e25fSSatish Balay {
86649b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
86749b5e25fSSatish Balay   Scalar          *x,*y,*z,*xb,x1,x2,x3,x4,x5,x6;
86849b5e25fSSatish Balay   MatScalar       *v;
86949b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
87049b5e25fSSatish Balay 
87149b5e25fSSatish Balay   PetscFunctionBegin;
87249b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
87349b5e25fSSatish Balay   if (yy != xx) {
87449b5e25fSSatish Balay     ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
87549b5e25fSSatish Balay   } else {
87649b5e25fSSatish Balay     y = x;
87749b5e25fSSatish Balay   }
87849b5e25fSSatish Balay   if (zz != yy) {
87949b5e25fSSatish Balay     ierr = VecCopy(yy,zz);CHKERRQ(ierr);
88049b5e25fSSatish Balay     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
88149b5e25fSSatish Balay   } else {
88249b5e25fSSatish Balay     z = y;
88349b5e25fSSatish Balay   }
88449b5e25fSSatish Balay 
88549b5e25fSSatish Balay   v     = a->a;
88649b5e25fSSatish Balay   xb = x;
88749b5e25fSSatish Balay 
88849b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
88949b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
89049b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4]; x6=xb[5];
89149b5e25fSSatish Balay     ib = aj + *ai;
89249b5e25fSSatish Balay     /* (diag of A)*x */
89349b5e25fSSatish Balay     z[6*i]   += v[0]*x1  + v[6]*x2 + v[12]*x3 + v[18]*x4+ v[24]*x5 + v[30]*x6;
89449b5e25fSSatish Balay     z[6*i+1] += v[6]*x1  + v[7]*x2 + v[13]*x3 + v[19]*x4+ v[25]*x5 + v[31]*x6;
89549b5e25fSSatish Balay     z[6*i+2] += v[12]*x1 +v[13]*x2 + v[14]*x3 + v[20]*x4+ v[26]*x5 + v[32]*x6;
89649b5e25fSSatish Balay     z[6*i+3] += v[18]*x1 +v[19]*x2 + v[20]*x3 + v[21]*x4+ v[27]*x5 + v[33]*x6;
89749b5e25fSSatish Balay     z[6*i+4] += v[24]*x1 +v[25]*x2 + v[26]*x3 + v[27]*x4+ v[28]*x5 + v[34]*x6;
89849b5e25fSSatish Balay     z[6*i+5] += v[30]*x1 +v[31]*x2 + v[32]*x3 + v[33]*x4+ v[34]*x5 + v[35]*x6;
89949b5e25fSSatish Balay     v += 36;
90049b5e25fSSatish Balay     for (j=1; j<n; j++) {
90149b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
90249b5e25fSSatish Balay       cval       = ib[j]*6;
90349b5e25fSSatish Balay       z[cval]   += v[0]*x1  + v[1]*x2 + v[2]*x3 + v[3]*x4+ v[4]*x5 + v[5]*x6;
90449b5e25fSSatish Balay       z[cval+1] += v[6]*x1  + v[7]*x2 + v[8]*x3 + v[9]*x4+ v[10]*x5 + v[11]*x6;
90549b5e25fSSatish Balay       z[cval+2] += v[12]*x1  + v[13]*x2 + v[14]*x3 + v[15]*x4+ v[16]*x5 + v[17]*x6;
90649b5e25fSSatish Balay       z[cval+3] += v[18]*x1  + v[19]*x2 + v[20]*x3 + v[21]*x4+ v[22]*x5 + v[23]*x6;
90749b5e25fSSatish Balay       z[cval+4] += v[24]*x1  + v[25]*x2 + v[26]*x3 + v[27]*x4+ v[28]*x5 + v[29]*x6;
90849b5e25fSSatish Balay       z[cval+5] += v[30]*x1  + v[31]*x2 + v[32]*x3 + v[33]*x4+ v[34]*x5 + v[35]*x6;
90949b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
91049b5e25fSSatish Balay       z[6*i]   +=v[0]*x[cval]+v[6]*x[cval+1]+v[12]*x[cval+2]+v[18]*x[cval+3]+v[24]*x[cval+4]+v[30]*x[cval+5];
91149b5e25fSSatish Balay       z[6*i+1] +=v[1]*x[cval]+v[7]*x[cval+1]+v[13]*x[cval+2]+v[19]*x[cval+3]+v[25]*x[cval+4]+v[31]*x[cval+5];
91249b5e25fSSatish Balay       z[6*i+2] +=v[2]*x[cval]+v[8]*x[cval+1]+v[14]*x[cval+2]+v[20]*x[cval+3]+v[26]*x[cval+4]+v[32]*x[cval+5];
91349b5e25fSSatish Balay       z[6*i+3] +=v[3]*x[cval]+v[9]*x[cval+1]+v[15]*x[cval+2]+v[21]*x[cval+3]+v[27]*x[cval+4]+v[33]*x[cval+5];
91449b5e25fSSatish Balay       z[6*i+4] +=v[4]*x[cval]+v[10]*x[cval+1]+v[16]*x[cval+2]+v[22]*x[cval+3]+v[28]*x[cval+4]+v[34]*x[cval+5];
91549b5e25fSSatish Balay       z[6*i+5] +=v[5]*x[cval]+v[11]*x[cval+1]+v[17]*x[cval+2]+v[23]*x[cval+3]+v[29]*x[cval+4]+v[35]*x[cval+5];
91649b5e25fSSatish Balay       v  += 36;
91749b5e25fSSatish Balay     }
91849b5e25fSSatish Balay     xb +=6; ai++;
91949b5e25fSSatish Balay   }
92049b5e25fSSatish Balay 
92149b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
92249b5e25fSSatish Balay   if (yy != xx) ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
92349b5e25fSSatish Balay   if (zz != yy) ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
92449b5e25fSSatish Balay 
92549b5e25fSSatish Balay   PLogFlops(72*(a->s_nz*2 - a->m));
92649b5e25fSSatish Balay   PetscFunctionReturn(0);
92749b5e25fSSatish Balay }
92849b5e25fSSatish Balay 
92949b5e25fSSatish Balay #undef __FUNC__
93049b5e25fSSatish Balay #define __FUNC__ "MatMultAdd_SeqSBAIJ_7"
93149b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_7(Mat A,Vec xx,Vec yy,Vec zz)
93249b5e25fSSatish Balay {
93349b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
93449b5e25fSSatish Balay   Scalar          *x,*y,*z,*xb,x1,x2,x3,x4,x5,x6,x7;
93549b5e25fSSatish Balay   MatScalar       *v;
93649b5e25fSSatish Balay   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j;
93749b5e25fSSatish Balay 
93849b5e25fSSatish Balay   PetscFunctionBegin;
93949b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
94049b5e25fSSatish Balay   if (yy != xx) {
94149b5e25fSSatish Balay     ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
94249b5e25fSSatish Balay   } else {
94349b5e25fSSatish Balay     y = x;
94449b5e25fSSatish Balay   }
94549b5e25fSSatish Balay   if (zz != yy) {
94649b5e25fSSatish Balay     ierr = VecCopy(yy,zz);CHKERRQ(ierr);
94749b5e25fSSatish Balay     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
94849b5e25fSSatish Balay   } else {
94949b5e25fSSatish Balay     z = y;
95049b5e25fSSatish Balay   }
95149b5e25fSSatish Balay 
95249b5e25fSSatish Balay   v     = a->a;
95349b5e25fSSatish Balay   xb = x;
95449b5e25fSSatish Balay 
95549b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
95649b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
95749b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4]; x6=xb[5]; x7=xb[6];
95849b5e25fSSatish Balay     ib = aj + *ai;
95949b5e25fSSatish Balay     /* (diag of A)*x */
96049b5e25fSSatish Balay     z[7*i]   += v[0]*x1 + v[7]*x2 + v[14]*x3 + v[21]*x4+ v[28]*x5 + v[35]*x6+ v[42]*x7;
96149b5e25fSSatish Balay     z[7*i+1] += v[7]*x1 + v[8]*x2 + v[15]*x3 + v[22]*x4+ v[29]*x5 + v[36]*x6+ v[43]*x7;
96249b5e25fSSatish Balay     z[7*i+2] += v[14]*x1+ v[15]*x2 +v[16]*x3 + v[23]*x4+ v[30]*x5 + v[37]*x6+ v[44]*x7;
96349b5e25fSSatish Balay     z[7*i+3] += v[21]*x1+ v[22]*x2 +v[23]*x3 + v[24]*x4+ v[31]*x5 + v[38]*x6+ v[45]*x7;
96449b5e25fSSatish Balay     z[7*i+4] += v[28]*x1+ v[29]*x2 +v[30]*x3 + v[31]*x4+ v[32]*x5 + v[39]*x6+ v[46]*x7;
96549b5e25fSSatish Balay     z[7*i+5] += v[35]*x1+ v[36]*x2 +v[37]*x3 + v[38]*x4+ v[39]*x5 + v[40]*x6+ v[47]*x7;
96649b5e25fSSatish Balay     z[7*i+6] += v[42]*x1+ v[43]*x2 +v[44]*x3 + v[45]*x4+ v[46]*x5 + v[47]*x6+ v[48]*x7;
96749b5e25fSSatish Balay     v += 49;
96849b5e25fSSatish Balay     for (j=1; j<n; j++) {
96949b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
97049b5e25fSSatish Balay       cval       = ib[j]*7;
97149b5e25fSSatish Balay       z[cval]   += v[0]*x1  + v[1]*x2 + v[2]*x3 + v[3]*x4+ v[4]*x5 + v[5]*x6+ v[6]*x7;
97249b5e25fSSatish Balay       z[cval+1] += v[7]*x1  + v[8]*x2 + v[9]*x3 + v[10]*x4+ v[11]*x5 + v[12]*x6+ v[13]*x7;
97349b5e25fSSatish Balay       z[cval+2] += v[14]*x1  + v[15]*x2 + v[16]*x3 + v[17]*x4+ v[18]*x5 + v[19]*x6+ v[20]*x7;
97449b5e25fSSatish Balay       z[cval+3] += v[21]*x1  + v[22]*x2 + v[23]*x3 + v[24]*x4+ v[25]*x5 + v[26]*x6+ v[27]*x7;
97549b5e25fSSatish Balay       z[cval+4] += v[28]*x1  + v[29]*x2 + v[30]*x3 + v[31]*x4+ v[32]*x5 + v[33]*x6+ v[34]*x7;
97649b5e25fSSatish Balay       z[cval+5] += v[35]*x1  + v[36]*x2 + v[37]*x3 + v[38]*x4+ v[39]*x5 + v[40]*x6+ v[41]*x7;
97749b5e25fSSatish Balay       z[cval+6] += v[42]*x1  + v[43]*x2 + v[44]*x3 + v[45]*x4+ v[46]*x5 + v[47]*x6+ v[48]*x7;
97849b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
97949b5e25fSSatish Balay       z[7*i]  +=v[0]*x[cval]+v[7]*x[cval+1]+v[14]*x[cval+2]+v[21]*x[cval+3]+v[28]*x[cval+4]+v[35]*x[cval+5]+v[42]*x[cval+6];
98049b5e25fSSatish Balay       z[7*i+1]+=v[1]*x[cval]+v[8]*x[cval+1]+v[15]*x[cval+2]+v[22]*x[cval+3]+v[29]*x[cval+4]+v[36]*x[cval+5]+v[43]*x[cval+6];
98149b5e25fSSatish Balay       z[7*i+2]+=v[2]*x[cval]+v[9]*x[cval+1]+v[16]*x[cval+2]+v[23]*x[cval+3]+v[30]*x[cval+4]+v[37]*x[cval+5]+v[44]*x[cval+6];
98249b5e25fSSatish Balay       z[7*i+3]+=v[3]*x[cval]+v[10]*x[cval+1]+v[17]*x[cval+2]+v[24]*x[cval+3]+v[31]*x[cval+4]+v[38]*x[cval+5]+v[45]*x[cval+6];
98349b5e25fSSatish Balay       z[7*i+4]+=v[4]*x[cval]+v[11]*x[cval+1]+v[18]*x[cval+2]+v[25]*x[cval+3]+v[32]*x[cval+4]+v[39]*x[cval+5]+v[46]*x[cval+6];
98449b5e25fSSatish Balay       z[7*i+5]+=v[5]*x[cval]+v[12]*x[cval+1]+v[19]*x[cval+2]+v[26]*x[cval+3]+v[33]*x[cval+4]+v[40]*x[cval+5]+v[47]*x[cval+6];
98549b5e25fSSatish Balay       z[7*i+6]+=v[6]*x[cval]+v[13]*x[cval+1]+v[20]*x[cval+2]+v[27]*x[cval+3]+v[34]*x[cval+4]+v[41]*x[cval+5]+v[48]*x[cval+6];
98649b5e25fSSatish Balay       v  += 49;
98749b5e25fSSatish Balay     }
98849b5e25fSSatish Balay     xb +=7; ai++;
98949b5e25fSSatish Balay   }
99049b5e25fSSatish Balay 
99149b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
99249b5e25fSSatish Balay   if (yy != xx) ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
99349b5e25fSSatish Balay   if (zz != yy) ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
99449b5e25fSSatish Balay 
99549b5e25fSSatish Balay   PLogFlops(98*(a->s_nz*2 - a->m));
99649b5e25fSSatish Balay   PetscFunctionReturn(0);
99749b5e25fSSatish Balay }
99849b5e25fSSatish Balay 
99949b5e25fSSatish Balay #undef __FUNC__
100049b5e25fSSatish Balay #define __FUNC__ "MatMultAdd_SeqSBAIJ_N"
100149b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_N(Mat A,Vec xx,Vec yy,Vec zz)
100249b5e25fSSatish Balay {
100349b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
1004ecfa4421SSatish Balay   Scalar          *x,*x_ptr,*y,*z,*z_ptr=0,*xb,*zb,*work,*workt;
1005066653e3SSatish Balay   MatScalar       *v;
1006066653e3SSatish Balay   int             ierr,mbs=a->mbs,i,*idx,*aj,*ii,bs=a->bs,j,n,bs2=a->bs2;
1007066653e3SSatish Balay   int             ncols,k;
100849b5e25fSSatish Balay 
100949b5e25fSSatish Balay   PetscFunctionBegin;
101049b5e25fSSatish Balay   ierr = VecGetArray(xx,&x);CHKERRQ(ierr); x_ptr=x;
101149b5e25fSSatish Balay   if (yy != xx) {
101249b5e25fSSatish Balay     ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
101349b5e25fSSatish Balay   } else {
101449b5e25fSSatish Balay     y = x;
101549b5e25fSSatish Balay   }
101649b5e25fSSatish Balay   if (zz != yy) {
101749b5e25fSSatish Balay     ierr = VecCopy(yy,zz);CHKERRQ(ierr);
101849b5e25fSSatish Balay     ierr = VecGetArray(zz,&z);CHKERRQ(ierr); z_ptr=z;
101949b5e25fSSatish Balay   } else {
102049b5e25fSSatish Balay     z = y;
102149b5e25fSSatish Balay   }
102249b5e25fSSatish Balay 
102349b5e25fSSatish Balay   aj   = a->j;
102449b5e25fSSatish Balay   v    = a->a;
102549b5e25fSSatish Balay   ii   = a->i;
102649b5e25fSSatish Balay 
102749b5e25fSSatish Balay   if (!a->mult_work) {
102849b5e25fSSatish Balay     k = a->m;
102949b5e25fSSatish Balay     a->mult_work = (Scalar*)PetscMalloc((k+1)*sizeof(Scalar));CHKPTRQ(a->mult_work);
103049b5e25fSSatish Balay   }
103149b5e25fSSatish Balay   work = a->mult_work;
103249b5e25fSSatish Balay 
103349b5e25fSSatish Balay 
103449b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
103549b5e25fSSatish Balay     n     = ii[1] - ii[0]; ncols = n*bs;
103649b5e25fSSatish Balay     workt = work; idx=aj+ii[0];
103749b5e25fSSatish Balay 
103849b5e25fSSatish Balay     /* upper triangular part */
103949b5e25fSSatish Balay     for (j=0; j<n; j++) {
1040066653e3SSatish Balay       /* col = bs*(*idx); */
104149b5e25fSSatish Balay       xb = x_ptr + bs*(*idx++);
104249b5e25fSSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
104349b5e25fSSatish Balay       workt += bs;
104449b5e25fSSatish Balay     }
104549b5e25fSSatish Balay     /* z(i*bs:(i+1)*bs-1) += A(i,:)*x */
104649b5e25fSSatish Balay     Kernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z);
104749b5e25fSSatish Balay 
104849b5e25fSSatish Balay     /* strict lower triangular part */
104949b5e25fSSatish Balay     ncols -= bs;
105049b5e25fSSatish Balay     if (ncols > 0){
105149b5e25fSSatish Balay       workt = work;
105249b5e25fSSatish Balay       ierr  = PetscMemzero(workt,ncols*sizeof(Scalar));CHKERRQ(ierr);
105349b5e25fSSatish Balay       Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,x,v+bs2,workt);
105449b5e25fSSatish Balay 
105549b5e25fSSatish Balay       idx=aj+ii[0]+1;
105649b5e25fSSatish Balay       for (j=1; j<n; j++) {
105749b5e25fSSatish Balay         zb = z_ptr + bs*(*idx);
105849b5e25fSSatish Balay         idx++;
105949b5e25fSSatish Balay         for (k=0; k<bs; k++) zb[k] += workt[k] ;
106049b5e25fSSatish Balay         workt += bs;
106149b5e25fSSatish Balay       }
106249b5e25fSSatish Balay     }
106349b5e25fSSatish Balay 
106449b5e25fSSatish Balay     x += bs; v += n*bs2; z += bs; ii++;
106549b5e25fSSatish Balay   }
106649b5e25fSSatish Balay 
106749b5e25fSSatish Balay   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
106849b5e25fSSatish Balay   if (yy != xx) ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
106949b5e25fSSatish Balay   if (zz != yy) ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
107049b5e25fSSatish Balay 
107149b5e25fSSatish Balay   PLogFlops(2*(a->s_nz*2 - a->m));
107249b5e25fSSatish Balay   PetscFunctionReturn(0);
107349b5e25fSSatish Balay }
107449b5e25fSSatish Balay 
107549b5e25fSSatish Balay #undef __FUNC__
107649b5e25fSSatish Balay #define __FUNC__ "MatMultTranspose_SeqSBAIJ"
107749b5e25fSSatish Balay int MatMultTranspose_SeqSBAIJ(Mat A,Vec xx,Vec zz)
107849b5e25fSSatish Balay {
1079*f102d2d3SHong Zhang   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
108049b5e25fSSatish Balay   Scalar          *xg,*zg,*zb,zero = 0.0;
108149b5e25fSSatish Balay   Scalar          *x,*z,*xb,x1,x2,x3,x4,x5,x6,x7;
108249b5e25fSSatish Balay   MatScalar       *v;
108349b5e25fSSatish Balay   int             mbs=a->mbs,i,*idx,*ii,*ai=a->i,rval;
108449b5e25fSSatish Balay   int             bs=a->bs,j,n,bs2=a->bs2,*ib,ierr;
108549b5e25fSSatish Balay 
108649b5e25fSSatish Balay   PetscFunctionBegin;
1087*f102d2d3SHong Zhang   SETERRQ(1,1,"Matrix is symmetric. Call MatMult().");
108849b5e25fSSatish Balay   PetscFunctionReturn(0);
108949b5e25fSSatish Balay }
109049b5e25fSSatish Balay 
109149b5e25fSSatish Balay #undef __FUNC__
109249b5e25fSSatish Balay #define __FUNC__ "MatMultTransposeAdd_SeqSBAIJ"
109349b5e25fSSatish Balay int MatMultTransposeAdd_SeqSBAIJ(Mat A,Vec xx,Vec yy,Vec zz)
109449b5e25fSSatish Balay 
109549b5e25fSSatish Balay {
109649b5e25fSSatish Balay   Mat_SeqBAIJ     *a = (Mat_SeqBAIJ*)A->data;
109749b5e25fSSatish Balay   Scalar          *xg,*zg,*zb,*x,*z,*xb,x1,x2,x3,x4,x5;
109849b5e25fSSatish Balay   MatScalar       *v;
109949b5e25fSSatish Balay   int             mbs=a->mbs,i,*idx,*ii,*ai=a->i,rval,bs=a->bs,j,n,bs2=a->bs2,*ib,ierr;
110049b5e25fSSatish Balay 
110149b5e25fSSatish Balay   PetscFunctionBegin;
1102*f102d2d3SHong Zhang   SETERRQ(1,1,"Matrix is symmetric. Call MatMultAdd().");
110349b5e25fSSatish Balay   PetscFunctionReturn(0);
110449b5e25fSSatish Balay }
110549b5e25fSSatish Balay 
110649b5e25fSSatish Balay #undef __FUNC__
110749b5e25fSSatish Balay #define __FUNC__ "MatScale_SeqSBAIJ"
110849b5e25fSSatish Balay int MatScale_SeqSBAIJ(Scalar *alpha,Mat inA)
110949b5e25fSSatish Balay {
111049b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data;
111149b5e25fSSatish Balay   int         one = 1,totalnz = a->bs2*a->s_nz;
111249b5e25fSSatish Balay 
111349b5e25fSSatish Balay   PetscFunctionBegin;
111449b5e25fSSatish Balay   BLscal_(&totalnz,alpha,a->a,&one);
111549b5e25fSSatish Balay   PLogFlops(totalnz);
111649b5e25fSSatish Balay   PetscFunctionReturn(0);
111749b5e25fSSatish Balay }
111849b5e25fSSatish Balay 
111949b5e25fSSatish Balay #undef __FUNC__
112049b5e25fSSatish Balay #define __FUNC__ "MatNorm_SeqSBAIJ"
112149b5e25fSSatish Balay int MatNorm_SeqSBAIJ(Mat A,NormType type,PetscReal *norm)
112249b5e25fSSatish Balay {
112349b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
112449b5e25fSSatish Balay   MatScalar   *v = a->a;
112549b5e25fSSatish Balay   PetscReal   sum_diag = 0.0, sum_off = 0.0, *sum;
1126066653e3SSatish Balay   int         i,j,k,bs = a->bs,bs2=a->bs2,k1,mbs=a->mbs;
112749b5e25fSSatish Balay   int         *jl,*il,jmin,jmax,ierr,nexti,ik;
112849b5e25fSSatish Balay 
112949b5e25fSSatish Balay   PetscFunctionBegin;
113049b5e25fSSatish Balay   if (type == NORM_FROBENIUS) {
113149b5e25fSSatish Balay     for (k=0; k<mbs; k++){
113249b5e25fSSatish Balay       jmin = a->i[k]; jmax = a->i[k+1];
113349b5e25fSSatish Balay       /* diagonal block */
113449b5e25fSSatish Balay       for (i=0; i<bs2; i++){
113549b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
113649b5e25fSSatish Balay         sum_diag += PetscRealPart(PetscConj(*v)*(*v)); v++;
113749b5e25fSSatish Balay #else
113849b5e25fSSatish Balay         sum_diag += (*v)*(*v); v++;
113949b5e25fSSatish Balay #endif
114049b5e25fSSatish Balay       }
114149b5e25fSSatish Balay       for (j=jmin+1; j<jmax; j++){  /* off-diagonal blocks */
114249b5e25fSSatish Balay         for (i=0; i<bs2; i++){
114349b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
114449b5e25fSSatish Balay           sum_off += PetscRealPart(PetscConj(*v)*(*v)); v++;
114549b5e25fSSatish Balay #else
114649b5e25fSSatish Balay           sum_off += (*v)*(*v); v++;
114749b5e25fSSatish Balay #endif
114849b5e25fSSatish Balay         }
114949b5e25fSSatish Balay       }
115049b5e25fSSatish Balay     }
115149b5e25fSSatish Balay     *norm = sqrt(sum_diag + 2*sum_off);
115249b5e25fSSatish Balay 
115349b5e25fSSatish Balay   }  else if (type == NORM_INFINITY) { /* maximum row sum */
115449b5e25fSSatish Balay     il = (int*)PetscMalloc(mbs*sizeof(int));CHKPTRQ(il);
115549b5e25fSSatish Balay     jl = (int*)PetscMalloc(mbs*sizeof(int));CHKPTRQ(jl);
1156f3dd7b2dSKris Buschelman     sum = (PetscReal*)PetscMalloc(bs*sizeof(PetscReal));CHKPTRQ(sum);
115749b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
115849b5e25fSSatish Balay       jl[i] = mbs; il[0] = 0;
115949b5e25fSSatish Balay     }
116049b5e25fSSatish Balay 
116149b5e25fSSatish Balay     *norm = 0.0;
116249b5e25fSSatish Balay     for (k=0; k<mbs; k++) { /* k_th block row */
116349b5e25fSSatish Balay       for (j=0; j<bs; j++) sum[j]=0.0;
116449b5e25fSSatish Balay 
116549b5e25fSSatish Balay       /*-- col sum --*/
116649b5e25fSSatish Balay       i = jl[k]; /* first |A(i,k)| to be added */
116749b5e25fSSatish Balay       while (i<mbs){
116849b5e25fSSatish Balay         nexti = jl[i];  /* next block row to be added */
116949b5e25fSSatish Balay         ik    = il[i];  /* block index of A(i,k) in the array a */
117049b5e25fSSatish Balay         for (j=0; j<bs; j++){
117149b5e25fSSatish Balay           v = a->a + ik*bs2 + j*bs;
117249b5e25fSSatish Balay           for (k1=0; k1<bs; k1++) {
117349b5e25fSSatish Balay             sum[j] += PetscAbsScalar(*v); v++;
117449b5e25fSSatish Balay           }
117549b5e25fSSatish Balay         }
117649b5e25fSSatish Balay         /* update il, jl */
117749b5e25fSSatish Balay         jmin = ik + 1; jmax = a->i[i+1];
117849b5e25fSSatish Balay         if (jmin < jmax){
117949b5e25fSSatish Balay           il[i] = jmin;
118049b5e25fSSatish Balay           j   = a->j[jmin];
118149b5e25fSSatish Balay           jl[i] = jl[j]; jl[j]=i;
118249b5e25fSSatish Balay         }
118349b5e25fSSatish Balay         i = nexti;
118449b5e25fSSatish Balay       }
118549b5e25fSSatish Balay 
118649b5e25fSSatish Balay       /*-- row sum --*/
118749b5e25fSSatish Balay       jmin = a->i[k]; jmax = a->i[k+1];
118849b5e25fSSatish Balay       for (i=jmin; i<jmax; i++) {
118949b5e25fSSatish Balay         for (j=0; j<bs; j++){
119049b5e25fSSatish Balay           v = a->a + i*bs2 + j;
119149b5e25fSSatish Balay           for (k1=0; k1<bs; k1++){
119249b5e25fSSatish Balay             sum[j] += PetscAbsScalar(*v);
119349b5e25fSSatish Balay             v   += bs;
119449b5e25fSSatish Balay           }
119549b5e25fSSatish Balay         }
119649b5e25fSSatish Balay       }
119749b5e25fSSatish Balay       /* add k_th block row to il, jl */
119849b5e25fSSatish Balay       jmin++;
119949b5e25fSSatish Balay       if (jmin < jmax){
120049b5e25fSSatish Balay         il[k] = jmin;
120149b5e25fSSatish Balay         j   = a->j[jmin];
120249b5e25fSSatish Balay         jl[k] = jl[j]; jl[j] = k;
120349b5e25fSSatish Balay       }
120449b5e25fSSatish Balay       for (j=0; j<bs; j++){
120549b5e25fSSatish Balay         if (sum[j] > *norm) *norm = sum[j];
120649b5e25fSSatish Balay       }
120749b5e25fSSatish Balay     }
120849b5e25fSSatish Balay     ierr = PetscFree(il);CHKERRQ(ierr);
120949b5e25fSSatish Balay     ierr = PetscFree(jl);CHKERRQ(ierr);
121049b5e25fSSatish Balay     ierr = PetscFree(sum);CHKERRQ(ierr);
121149b5e25fSSatish Balay   } else {
121249b5e25fSSatish Balay     SETERRQ(PETSC_ERR_SUP,0,"No support for this norm yet");
121349b5e25fSSatish Balay   }
121449b5e25fSSatish Balay   PetscFunctionReturn(0);
121549b5e25fSSatish Balay }
121649b5e25fSSatish Balay 
121749b5e25fSSatish Balay #undef __FUNC__
121849b5e25fSSatish Balay #define __FUNC__ "MatEqual_SeqSBAIJ"
121949b5e25fSSatish Balay int MatEqual_SeqSBAIJ(Mat A,Mat B,PetscTruth* flg)
122049b5e25fSSatish Balay {
122149b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ *)A->data,*b = (Mat_SeqSBAIJ *)B->data;
122249b5e25fSSatish Balay   int         ierr;
122349b5e25fSSatish Balay 
122449b5e25fSSatish Balay   PetscFunctionBegin;
122549b5e25fSSatish Balay   if (B->type !=MATSEQSBAIJ) SETERRQ(PETSC_ERR_ARG_INCOMP,0,"Matrices must be same type");
122649b5e25fSSatish Balay 
122749b5e25fSSatish Balay   /* If the  matrix/block dimensions are not equal, or no of nonzeros or shift */
122849b5e25fSSatish Balay   if ((a->m != b->m) || (a->n !=b->n) || (a->bs != b->bs)|| (a->s_nz != b->s_nz)) {
122949b5e25fSSatish Balay     *flg = PETSC_FALSE; PetscFunctionReturn(0);
123049b5e25fSSatish Balay   }
123149b5e25fSSatish Balay 
123249b5e25fSSatish Balay   /* if the a->i are the same */
123349b5e25fSSatish Balay   ierr = PetscMemcmp(a->i,b->i,(a->mbs+1)*sizeof(int),flg);CHKERRQ(ierr);
123449b5e25fSSatish Balay   if (*flg == PETSC_FALSE) {
123549b5e25fSSatish Balay     PetscFunctionReturn(0);
123649b5e25fSSatish Balay   }
123749b5e25fSSatish Balay 
123849b5e25fSSatish Balay   /* if a->j are the same */
123949b5e25fSSatish Balay   ierr = PetscMemcmp(a->j,b->j,(a->s_nz)*sizeof(int),flg);CHKERRQ(ierr);
124049b5e25fSSatish Balay   if (*flg == PETSC_FALSE) {
124149b5e25fSSatish Balay     PetscFunctionReturn(0);
124249b5e25fSSatish Balay   }
124349b5e25fSSatish Balay   /* if a->a are the same */
124449b5e25fSSatish Balay   ierr = PetscMemcmp(a->a,b->a,(a->s_nz)*(a->bs)*(a->bs)*sizeof(Scalar),flg);CHKERRQ(ierr);
124549b5e25fSSatish Balay   PetscFunctionReturn(0);
124649b5e25fSSatish Balay 
124749b5e25fSSatish Balay }
124849b5e25fSSatish Balay 
124949b5e25fSSatish Balay #undef __FUNC__
125049b5e25fSSatish Balay #define __FUNC__ "MatGetDiagonal_SeqSBAIJ"
125149b5e25fSSatish Balay int MatGetDiagonal_SeqSBAIJ(Mat A,Vec v)
125249b5e25fSSatish Balay {
125349b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
125449b5e25fSSatish Balay   int         ierr,i,j,k,n,row,bs,*ai,*aj,ambs,bs2;
125549b5e25fSSatish Balay   Scalar      *x,zero = 0.0;
125649b5e25fSSatish Balay   MatScalar   *aa,*aa_j;
125749b5e25fSSatish Balay 
125849b5e25fSSatish Balay   PetscFunctionBegin;
125949b5e25fSSatish Balay   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Not for factored matrix");
126049b5e25fSSatish Balay   bs   = a->bs;
126149b5e25fSSatish Balay   aa   = a->a;
126249b5e25fSSatish Balay   ai   = a->i;
126349b5e25fSSatish Balay   aj   = a->j;
126449b5e25fSSatish Balay   ambs = a->mbs;
126549b5e25fSSatish Balay   bs2  = a->bs2;
126649b5e25fSSatish Balay 
126749b5e25fSSatish Balay   ierr = VecSet(&zero,v);CHKERRQ(ierr);
126849b5e25fSSatish Balay   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
126949b5e25fSSatish Balay   ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr);
127049b5e25fSSatish Balay   /* if (n != a->m) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Nonconforming matrix and vector");*/
127149b5e25fSSatish Balay   for (i=0; i<ambs; i++) {
127249b5e25fSSatish Balay     j=ai[i];
127349b5e25fSSatish Balay     if (aj[j] == i) {             /* if this is a diagonal element */
127449b5e25fSSatish Balay       row  = i*bs;
127549b5e25fSSatish Balay       aa_j = aa + j*bs2;
127649b5e25fSSatish Balay       for (k=0; k<bs2; k+=(bs+1),row++) x[row] = aa_j[k];
127749b5e25fSSatish Balay     }
127849b5e25fSSatish Balay   }
127949b5e25fSSatish Balay   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
128049b5e25fSSatish Balay   PetscFunctionReturn(0);
128149b5e25fSSatish Balay }
128249b5e25fSSatish Balay 
128349b5e25fSSatish Balay #undef __FUNC__
128449b5e25fSSatish Balay #define __FUNC__ "MatDiagonalScale_SeqSBAIJ"
128549b5e25fSSatish Balay int MatDiagonalScale_SeqSBAIJ(Mat A,Vec ll,Vec rr)
128649b5e25fSSatish Balay {
128749b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
128849b5e25fSSatish Balay   Scalar      *l,*r,x,*li,*ri;
128949b5e25fSSatish Balay   MatScalar   *aa,*v;
1290066653e3SSatish Balay   int         ierr,i,j,k,lm,rn,M,m,*ai,*aj,mbs,tmp,bs,bs2;
129149b5e25fSSatish Balay 
129249b5e25fSSatish Balay   PetscFunctionBegin;
129349b5e25fSSatish Balay   ai  = a->i;
129449b5e25fSSatish Balay   aj  = a->j;
129549b5e25fSSatish Balay   aa  = a->a;
129649b5e25fSSatish Balay   m   = a->m;
129749b5e25fSSatish Balay   bs  = a->bs;
129849b5e25fSSatish Balay   mbs = a->mbs;
129949b5e25fSSatish Balay   bs2 = a->bs2;
130049b5e25fSSatish Balay 
130149b5e25fSSatish Balay   if (ll != rr) {
130249b5e25fSSatish Balay     SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"For symmetric format, left and right scaling vectors must be same\n");
130349b5e25fSSatish Balay   }
130449b5e25fSSatish Balay   if (ll) {
130549b5e25fSSatish Balay     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
130649b5e25fSSatish Balay     ierr = VecGetLocalSize(ll,&lm);CHKERRQ(ierr);
130749b5e25fSSatish Balay     if (lm != m) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Left scaling vector wrong length");
130849b5e25fSSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
130949b5e25fSSatish Balay       M  = ai[i+1] - ai[i];
131049b5e25fSSatish Balay       li = l + i*bs;
131149b5e25fSSatish Balay       v  = aa + bs2*ai[i];
131249b5e25fSSatish Balay       for (j=0; j<M; j++) { /* for each block */
131349b5e25fSSatish Balay         for (k=0; k<bs2; k++) {
131449b5e25fSSatish Balay           (*v++) *= li[k%bs];
131549b5e25fSSatish Balay         }
131649b5e25fSSatish Balay #ifdef CONT
131749b5e25fSSatish Balay         /* will be used to replace the above loop */
131849b5e25fSSatish Balay         ri = l + bs*aj[ai[i]+j];
131949b5e25fSSatish Balay         for (k=0; k<bs; k++) { /* column value */
132049b5e25fSSatish Balay           x = ri[k];
132149b5e25fSSatish Balay           for (tmp=0; tmp<bs; tmp++) (*v++) *= li[tmp]*x;
132249b5e25fSSatish Balay         }
132349b5e25fSSatish Balay #endif
132449b5e25fSSatish Balay 
132549b5e25fSSatish Balay       }
132649b5e25fSSatish Balay     }
132749b5e25fSSatish Balay     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
132849b5e25fSSatish Balay     PLogFlops(2*a->s_nz);
132949b5e25fSSatish Balay   }
133049b5e25fSSatish Balay   /* will be deleted */
133149b5e25fSSatish Balay   if (rr) {
133249b5e25fSSatish Balay     ierr = VecGetArray(rr,&r);CHKERRQ(ierr);
133349b5e25fSSatish Balay     ierr = VecGetLocalSize(rr,&rn);CHKERRQ(ierr);
133449b5e25fSSatish Balay     if (rn != m) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Right scaling vector wrong length");
133549b5e25fSSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
133649b5e25fSSatish Balay       M  = ai[i+1] - ai[i];
133749b5e25fSSatish Balay       v  = aa + bs2*ai[i];
133849b5e25fSSatish Balay       for (j=0; j<M; j++) { /* for each block */
133949b5e25fSSatish Balay         ri = r + bs*aj[ai[i]+j];
134049b5e25fSSatish Balay         for (k=0; k<bs; k++) {
134149b5e25fSSatish Balay           x = ri[k];
134249b5e25fSSatish Balay           for (tmp=0; tmp<bs; tmp++) (*v++) *= x;
134349b5e25fSSatish Balay         }
134449b5e25fSSatish Balay       }
134549b5e25fSSatish Balay     }
134649b5e25fSSatish Balay     ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr);
134749b5e25fSSatish Balay     PLogFlops(a->s_nz);
134849b5e25fSSatish Balay   }
134949b5e25fSSatish Balay   PetscFunctionReturn(0);
135049b5e25fSSatish Balay }
135149b5e25fSSatish Balay 
135249b5e25fSSatish Balay #undef __FUNC__
135349b5e25fSSatish Balay #define __FUNC__ "MatGetInfo_SeqSBAIJ"
135449b5e25fSSatish Balay int MatGetInfo_SeqSBAIJ(Mat A,MatInfoType flag,MatInfo *info)
135549b5e25fSSatish Balay {
135649b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
135749b5e25fSSatish Balay 
135849b5e25fSSatish Balay   PetscFunctionBegin;
135949b5e25fSSatish Balay   info->rows_global    = (double)a->m;
136049b5e25fSSatish Balay   info->columns_global = (double)a->m;
136149b5e25fSSatish Balay   info->rows_local     = (double)a->m;
136249b5e25fSSatish Balay   info->columns_local  = (double)a->m;
136349b5e25fSSatish Balay   info->block_size     = a->bs2;
136449b5e25fSSatish Balay   info->nz_allocated   = a->s_maxnz; /*num. of nonzeros in upper triangular part */
136549b5e25fSSatish Balay   info->nz_used        = a->bs2*a->s_nz; /*num. of nonzeros in upper triangular part */
136649b5e25fSSatish Balay   info->nz_unneeded    = (double)(info->nz_allocated - info->nz_used);
136749b5e25fSSatish Balay   info->assemblies   = A->num_ass;
136849b5e25fSSatish Balay   info->mallocs      = a->reallocs;
136949b5e25fSSatish Balay   info->memory       = A->mem;
137049b5e25fSSatish Balay   if (A->factor) {
137149b5e25fSSatish Balay     info->fill_ratio_given  = A->info.fill_ratio_given;
137249b5e25fSSatish Balay     info->fill_ratio_needed = A->info.fill_ratio_needed;
137349b5e25fSSatish Balay     info->factor_mallocs    = A->info.factor_mallocs;
137449b5e25fSSatish Balay   } else {
137549b5e25fSSatish Balay     info->fill_ratio_given  = 0;
137649b5e25fSSatish Balay     info->fill_ratio_needed = 0;
137749b5e25fSSatish Balay     info->factor_mallocs    = 0;
137849b5e25fSSatish Balay   }
137949b5e25fSSatish Balay   PetscFunctionReturn(0);
138049b5e25fSSatish Balay }
138149b5e25fSSatish Balay 
138249b5e25fSSatish Balay 
138349b5e25fSSatish Balay #undef __FUNC__
138449b5e25fSSatish Balay #define __FUNC__ "MatZeroEntries_SeqSBAIJ"
138549b5e25fSSatish Balay int MatZeroEntries_SeqSBAIJ(Mat A)
138649b5e25fSSatish Balay {
138749b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
138849b5e25fSSatish Balay   int         ierr;
138949b5e25fSSatish Balay 
139049b5e25fSSatish Balay   PetscFunctionBegin;
139149b5e25fSSatish Balay   ierr = PetscMemzero(a->a,a->bs2*a->i[a->mbs]*sizeof(MatScalar));CHKERRQ(ierr);
139249b5e25fSSatish Balay   PetscFunctionReturn(0);
139349b5e25fSSatish Balay }
1394