xref: /petsc/src/mat/impls/sbaij/seq/sbaij2.c (revision b1d4fb267e72f6b087ed013bbfbbee1418c3f503)
173f4d377SMatthew Knepley /*$Id: sbaij2.c,v 1.32 2001/08/07 03:03:01 balay Exp $*/
249b5e25fSSatish Balay 
349b5e25fSSatish Balay #include "src/mat/impls/baij/seq/baij.h"
449b5e25fSSatish Balay #include "src/inline/spops.h"
549b5e25fSSatish Balay #include "src/inline/ilu.h"
649b5e25fSSatish Balay #include "petscbt.h"
73a7fca6bSBarry Smith #include "src/mat/impls/sbaij/seq/sbaij.h"
849b5e25fSSatish Balay 
94a2ae208SSatish Balay #undef __FUNCT__
104a2ae208SSatish Balay #define __FUNCT__ "MatIncreaseOverlap_SeqSBAIJ"
11268466fbSBarry Smith int MatIncreaseOverlap_SeqSBAIJ(Mat A,int is_max,IS is[],int ov)
1249b5e25fSSatish Balay {
1349b5e25fSSatish Balay   PetscFunctionBegin;
140eaf17bfSBarry Smith   SETERRQ(1,"Function not yet written for SBAIJ format");
155535bfb1SHong Zhang   /* PetscFunctionReturn(0); */
1649b5e25fSSatish Balay }
1749b5e25fSSatish Balay 
184a2ae208SSatish Balay #undef __FUNCT__
194a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqSBAIJ_Private"
2049b5e25fSSatish Balay int MatGetSubMatrix_SeqSBAIJ_Private(Mat A,IS isrow,IS iscol,int cs,MatReuse scall,Mat *B)
2149b5e25fSSatish Balay {
2249b5e25fSSatish Balay   Mat_SeqSBAIJ  *a = (Mat_SeqSBAIJ*)A->data,*c;
2349b5e25fSSatish Balay   int          *smap,i,k,kstart,kend,ierr,oldcols = a->mbs,*lens;
2449b5e25fSSatish Balay   int          row,mat_i,*mat_j,tcol,*mat_ilen;
2549b5e25fSSatish Balay   int          *irow,nrows,*ssmap,bs=a->bs,bs2=a->bs2;
2649b5e25fSSatish Balay   int          *aj = a->j,*ai = a->i;
2749b5e25fSSatish Balay   MatScalar    *mat_a;
2849b5e25fSSatish Balay   Mat          C;
2949b5e25fSSatish Balay   PetscTruth   flag;
3049b5e25fSSatish Balay 
3149b5e25fSSatish Balay   PetscFunctionBegin;
3249b5e25fSSatish Balay 
33ac355199SBarry Smith   if (isrow != iscol) SETERRQ(1,"MatGetSubmatrices_SeqSBAIJ: For symm. format, iscol must equal isro");
3449b5e25fSSatish Balay   ierr = ISSorted(iscol,(PetscTruth*)&i);CHKERRQ(ierr);
35347d480fSBarry Smith   if (!i) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"IS is not sorted");
3649b5e25fSSatish Balay 
3749b5e25fSSatish Balay   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
3849b5e25fSSatish Balay   ierr = ISGetSize(isrow,&nrows);CHKERRQ(ierr);
3949b5e25fSSatish Balay 
40b0a32e0cSBarry Smith   ierr  = PetscMalloc((1+oldcols)*sizeof(int),&smap);CHKERRQ(ierr);
4149b5e25fSSatish Balay   ssmap = smap;
42b0a32e0cSBarry Smith   ierr  = PetscMalloc((1+nrows)*sizeof(int),&lens);CHKERRQ(ierr);
4349b5e25fSSatish Balay   ierr  = PetscMemzero(smap,oldcols*sizeof(int));CHKERRQ(ierr);
4449b5e25fSSatish Balay   for (i=0; i<nrows; i++) smap[irow[i]] = i+1; /* nrows = ncols */
4549b5e25fSSatish Balay   /* determine lens of each row */
4649b5e25fSSatish Balay   for (i=0; i<nrows; i++) {
4749b5e25fSSatish Balay     kstart  = ai[irow[i]];
4849b5e25fSSatish Balay     kend    = kstart + a->ilen[irow[i]];
4949b5e25fSSatish Balay     lens[i] = 0;
5049b5e25fSSatish Balay       for (k=kstart; k<kend; k++) {
5149b5e25fSSatish Balay         if (ssmap[aj[k]]) {
5249b5e25fSSatish Balay           lens[i]++;
5349b5e25fSSatish Balay         }
5449b5e25fSSatish Balay       }
5549b5e25fSSatish Balay     }
5649b5e25fSSatish Balay   /* Create and fill new matrix */
5749b5e25fSSatish Balay   if (scall == MAT_REUSE_MATRIX) {
5849b5e25fSSatish Balay     c = (Mat_SeqSBAIJ *)((*B)->data);
5949b5e25fSSatish Balay 
60347d480fSBarry Smith     if (c->mbs!=nrows || c->bs!=bs) SETERRQ(PETSC_ERR_ARG_SIZ,"Submatrix wrong size");
6149b5e25fSSatish Balay     ierr = PetscMemcmp(c->ilen,lens,c->mbs *sizeof(int),&flag);CHKERRQ(ierr);
6249b5e25fSSatish Balay     if (flag == PETSC_FALSE) {
63347d480fSBarry Smith       SETERRQ(PETSC_ERR_ARG_SIZ,"Cannot reuse matrix. wrong no of nonzeros");
6449b5e25fSSatish Balay     }
6549b5e25fSSatish Balay     ierr = PetscMemzero(c->ilen,c->mbs*sizeof(int));CHKERRQ(ierr);
6649b5e25fSSatish Balay     C = *B;
6749b5e25fSSatish Balay   } else {
6849b5e25fSSatish Balay     ierr = MatCreateSeqSBAIJ(A->comm,bs,nrows*bs,nrows*bs,0,lens,&C);CHKERRQ(ierr);
6949b5e25fSSatish Balay   }
7049b5e25fSSatish Balay   c = (Mat_SeqSBAIJ *)(C->data);
7149b5e25fSSatish Balay   for (i=0; i<nrows; i++) {
7249b5e25fSSatish Balay     row    = irow[i];
7349b5e25fSSatish Balay     kstart = ai[row];
7449b5e25fSSatish Balay     kend   = kstart + a->ilen[row];
7549b5e25fSSatish Balay     mat_i  = c->i[i];
7649b5e25fSSatish Balay     mat_j  = c->j + mat_i;
7749b5e25fSSatish Balay     mat_a  = c->a + mat_i*bs2;
7849b5e25fSSatish Balay     mat_ilen = c->ilen + i;
7949b5e25fSSatish Balay     for (k=kstart; k<kend; k++) {
8049b5e25fSSatish Balay       if ((tcol=ssmap[a->j[k]])) {
8149b5e25fSSatish Balay         *mat_j++ = tcol - 1;
8249b5e25fSSatish Balay         ierr     = PetscMemcpy(mat_a,a->a+k*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
8349b5e25fSSatish Balay         mat_a   += bs2;
8449b5e25fSSatish Balay         (*mat_ilen)++;
8549b5e25fSSatish Balay       }
8649b5e25fSSatish Balay     }
8749b5e25fSSatish Balay   }
8849b5e25fSSatish Balay 
8949b5e25fSSatish Balay   /* Free work space */
9049b5e25fSSatish Balay   ierr = PetscFree(smap);CHKERRQ(ierr);
9149b5e25fSSatish Balay   ierr = PetscFree(lens);CHKERRQ(ierr);
9249b5e25fSSatish Balay   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
9349b5e25fSSatish Balay   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
9449b5e25fSSatish Balay 
9549b5e25fSSatish Balay   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
9649b5e25fSSatish Balay   *B = C;
9749b5e25fSSatish Balay   PetscFunctionReturn(0);
9849b5e25fSSatish Balay }
9949b5e25fSSatish Balay 
1004a2ae208SSatish Balay #undef __FUNCT__
1014a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqSBAIJ"
10249b5e25fSSatish Balay int MatGetSubMatrix_SeqSBAIJ(Mat A,IS isrow,IS iscol,int cs,MatReuse scall,Mat *B)
10349b5e25fSSatish Balay {
10449b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
10549b5e25fSSatish Balay   IS          is1;
10649b5e25fSSatish Balay   int         *vary,*iary,*irow,nrows,i,ierr,bs=a->bs,count;
10749b5e25fSSatish Balay 
10849b5e25fSSatish Balay   PetscFunctionBegin;
109ac355199SBarry Smith   if (isrow != iscol) SETERRQ(1,"MatGetSubmatrices_SeqSBAIJ: For symm. format, iscol must equal isro");
11049b5e25fSSatish Balay 
11149b5e25fSSatish Balay   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
11249b5e25fSSatish Balay   ierr = ISGetSize(isrow,&nrows);CHKERRQ(ierr);
11349b5e25fSSatish Balay 
11449b5e25fSSatish Balay   /* Verify if the indices corespond to each element in a block
11549b5e25fSSatish Balay    and form the IS with compressed IS */
11682502324SSatish Balay   ierr = PetscMalloc(2*(a->mbs+1)*sizeof(int),&vary);CHKERRQ(ierr);
11749b5e25fSSatish Balay   iary = vary + a->mbs;
11849b5e25fSSatish Balay   ierr = PetscMemzero(vary,(a->mbs)*sizeof(int));CHKERRQ(ierr);
11949b5e25fSSatish Balay   for (i=0; i<nrows; i++) vary[irow[i]/bs]++;
12049b5e25fSSatish Balay 
12149b5e25fSSatish Balay   count = 0;
12249b5e25fSSatish Balay   for (i=0; i<a->mbs; i++) {
123ac355199SBarry Smith     if (vary[i]!=0 && vary[i]!=bs) SETERRQ(1,"Index set does not match blocks");
12449b5e25fSSatish Balay     if (vary[i]==bs) iary[count++] = i;
12549b5e25fSSatish Balay   }
12649b5e25fSSatish Balay   ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is1);CHKERRQ(ierr);
12749b5e25fSSatish Balay 
12849b5e25fSSatish Balay   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
12949b5e25fSSatish Balay   ierr = PetscFree(vary);CHKERRQ(ierr);
13049b5e25fSSatish Balay 
13149b5e25fSSatish Balay   ierr = MatGetSubMatrix_SeqSBAIJ_Private(A,is1,is1,cs,scall,B);CHKERRQ(ierr);
13249b5e25fSSatish Balay   ISDestroy(is1);
13349b5e25fSSatish Balay   PetscFunctionReturn(0);
13449b5e25fSSatish Balay }
13549b5e25fSSatish Balay 
1364a2ae208SSatish Balay #undef __FUNCT__
1374a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqSBAIJ"
138268466fbSBarry Smith int MatGetSubMatrices_SeqSBAIJ(Mat A,int n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[])
13949b5e25fSSatish Balay {
14049b5e25fSSatish Balay   int ierr,i;
14149b5e25fSSatish Balay 
14249b5e25fSSatish Balay   PetscFunctionBegin;
14349b5e25fSSatish Balay   if (scall == MAT_INITIAL_MATRIX) {
14482502324SSatish Balay     ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr);
14549b5e25fSSatish Balay   }
14649b5e25fSSatish Balay 
14749b5e25fSSatish Balay   for (i=0; i<n; i++) {
14849b5e25fSSatish Balay     ierr = MatGetSubMatrix_SeqSBAIJ(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr);
14949b5e25fSSatish Balay   }
15049b5e25fSSatish Balay   PetscFunctionReturn(0);
15149b5e25fSSatish Balay }
15249b5e25fSSatish Balay 
15349b5e25fSSatish Balay /* -------------------------------------------------------*/
15449b5e25fSSatish Balay /* Should check that shapes of vectors and matrices match */
15549b5e25fSSatish Balay /* -------------------------------------------------------*/
156d9eff348SSatish Balay #include "petscblaslapack.h"
15749b5e25fSSatish Balay 
1584a2ae208SSatish Balay #undef __FUNCT__
1594a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqSBAIJ_1"
16049b5e25fSSatish Balay int MatMult_SeqSBAIJ_1(Mat A,Vec xx,Vec zz)
16149b5e25fSSatish Balay {
16249b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
16387828ca2SBarry Smith   PetscScalar     *x,*z,*xb,x1,zero=0.0;
16449b5e25fSSatish Balay   MatScalar       *v;
165831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
16649b5e25fSSatish Balay 
16749b5e25fSSatish Balay   PetscFunctionBegin;
16849b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
169*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
170*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
17149b5e25fSSatish Balay 
17249b5e25fSSatish Balay   v  = a->a;
17349b5e25fSSatish Balay   xb = x;
17449b5e25fSSatish Balay 
17549b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
17649b5e25fSSatish Balay     n  = ai[1] - ai[0];  /* length of i_th row of A */
17749b5e25fSSatish Balay     x1 = xb[0];
17849b5e25fSSatish Balay     ib = aj + *ai;
179831a3094SHong Zhang     jmin = 0;
180831a3094SHong Zhang     if (*ib == i) {      /* (diag of A)*x */
181831a3094SHong Zhang       z[i] += *v++ * x[*ib++];
182831a3094SHong Zhang       jmin++;
183831a3094SHong Zhang     }
184831a3094SHong Zhang     for (j=jmin; j<n; j++) {
18549b5e25fSSatish Balay       cval    = *ib;
18649b5e25fSSatish Balay       z[cval] += *v * x1;      /* (strict lower triangular part of A)*x  */
18749b5e25fSSatish Balay       z[i] += *v++ * x[*ib++]; /* (strict upper triangular part of A)*x  */
18849b5e25fSSatish Balay     }
18949b5e25fSSatish Balay     xb++; ai++;
19049b5e25fSSatish Balay   }
19149b5e25fSSatish Balay 
192*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
193*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
1946c6c5352SBarry Smith   PetscLogFlops(2*(a->nz*2 - A->m) - A->m);  /* nz = (nz+m)/2 */
19549b5e25fSSatish Balay   PetscFunctionReturn(0);
19649b5e25fSSatish Balay }
19749b5e25fSSatish Balay 
1984a2ae208SSatish Balay #undef __FUNCT__
1994a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqSBAIJ_2"
20049b5e25fSSatish Balay int MatMult_SeqSBAIJ_2(Mat A,Vec xx,Vec zz)
20149b5e25fSSatish Balay {
20249b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
20387828ca2SBarry Smith   PetscScalar      *x,*z,*xb,x1,x2,zero=0.0;
20449b5e25fSSatish Balay   MatScalar        *v;
205831a3094SHong Zhang   int              mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
20649b5e25fSSatish Balay 
20749b5e25fSSatish Balay 
20849b5e25fSSatish Balay   PetscFunctionBegin;
20949b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
210*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
211*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
21249b5e25fSSatish Balay 
21349b5e25fSSatish Balay   v     = a->a;
21449b5e25fSSatish Balay   xb = x;
21549b5e25fSSatish Balay 
21649b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
21749b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
21849b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1];
21949b5e25fSSatish Balay     ib = aj + *ai;
220831a3094SHong Zhang     jmin = 0;
2217fbae186SHong Zhang     if (*ib == i){     /* (diag of A)*x */
22249b5e25fSSatish Balay       z[2*i]   += v[0]*x1 + v[2]*x2;
22349b5e25fSSatish Balay       z[2*i+1] += v[2]*x1 + v[3]*x2;
224831a3094SHong Zhang       v += 4; jmin++;
2257fbae186SHong Zhang     }
226831a3094SHong Zhang     for (j=jmin; j<n; j++) {
22749b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
22849b5e25fSSatish Balay       cval       = ib[j]*2;
22949b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2;
23049b5e25fSSatish Balay       z[cval+1]   += v[2]*x1 + v[3]*x2;
23149b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
23249b5e25fSSatish Balay       z[2*i]   += v[0]*x[cval] + v[2]*x[cval+1];
23349b5e25fSSatish Balay       z[2*i+1] += v[1]*x[cval] + v[3]*x[cval+1];
23449b5e25fSSatish Balay       v  += 4;
23549b5e25fSSatish Balay     }
23649b5e25fSSatish Balay     xb +=2; ai++;
23749b5e25fSSatish Balay   }
23849b5e25fSSatish Balay 
239*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
240*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
2416c6c5352SBarry Smith   PetscLogFlops(8*(a->nz*2 - A->m) - A->m);
24249b5e25fSSatish Balay   PetscFunctionReturn(0);
24349b5e25fSSatish Balay }
24449b5e25fSSatish Balay 
2454a2ae208SSatish Balay #undef __FUNCT__
2464a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqSBAIJ_3"
24749b5e25fSSatish Balay int MatMult_SeqSBAIJ_3(Mat A,Vec xx,Vec zz)
24849b5e25fSSatish Balay {
24949b5e25fSSatish Balay   Mat_SeqSBAIJ  *a = (Mat_SeqSBAIJ*)A->data;
25087828ca2SBarry Smith   PetscScalar   *x,*z,*xb,x1,x2,x3,zero=0.0;
25149b5e25fSSatish Balay   MatScalar     *v;
252831a3094SHong Zhang   int           mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
25349b5e25fSSatish Balay 
25449b5e25fSSatish Balay 
25549b5e25fSSatish Balay   PetscFunctionBegin;
25649b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
257*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
258*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
25949b5e25fSSatish Balay 
26049b5e25fSSatish Balay   v     = a->a;
26149b5e25fSSatish Balay   xb = x;
26249b5e25fSSatish Balay 
26349b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
26449b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
26549b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
26649b5e25fSSatish Balay     ib = aj + *ai;
267831a3094SHong Zhang     jmin = 0;
2687fbae186SHong Zhang     if (*ib == i){     /* (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;
272831a3094SHong Zhang       v += 9; jmin++;
2737fbae186SHong Zhang     }
274831a3094SHong Zhang     for (j=jmin; j<n; j++) {
27549b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
27649b5e25fSSatish Balay       cval       = ib[j]*3;
27749b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3;
27849b5e25fSSatish Balay       z[cval+1]   += v[3]*x1 + v[4]*x2 + v[5]*x3;
27949b5e25fSSatish Balay       z[cval+2]   += v[6]*x1 + v[7]*x2 + v[8]*x3;
28049b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
28149b5e25fSSatish Balay       z[3*i]   += v[0]*x[cval] + v[3]*x[cval+1]+ v[6]*x[cval+2];
28249b5e25fSSatish Balay       z[3*i+1] += v[1]*x[cval] + v[4]*x[cval+1]+ v[7]*x[cval+2];
28349b5e25fSSatish Balay       z[3*i+2] += v[2]*x[cval] + v[5]*x[cval+1]+ v[8]*x[cval+2];
28449b5e25fSSatish Balay       v  += 9;
28549b5e25fSSatish Balay     }
28649b5e25fSSatish Balay     xb +=3; ai++;
28749b5e25fSSatish Balay   }
28849b5e25fSSatish Balay 
289*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
290*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
2916c6c5352SBarry Smith   PetscLogFlops(18*(a->nz*2 - A->m) - A->m);
29249b5e25fSSatish Balay   PetscFunctionReturn(0);
29349b5e25fSSatish Balay }
29449b5e25fSSatish Balay 
2954a2ae208SSatish Balay #undef __FUNCT__
2964a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqSBAIJ_4"
29749b5e25fSSatish Balay int MatMult_SeqSBAIJ_4(Mat A,Vec xx,Vec zz)
29849b5e25fSSatish Balay {
29949b5e25fSSatish Balay   Mat_SeqSBAIJ     *a = (Mat_SeqSBAIJ*)A->data;
30087828ca2SBarry Smith   PetscScalar      *x,*z,*xb,x1,x2,x3,x4,zero=0.0;
30149b5e25fSSatish Balay   MatScalar        *v;
302831a3094SHong Zhang   int              mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
30349b5e25fSSatish Balay 
30449b5e25fSSatish Balay   PetscFunctionBegin;
30549b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
306*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
307*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
30849b5e25fSSatish Balay 
30949b5e25fSSatish Balay   v     = a->a;
31049b5e25fSSatish Balay   xb = x;
31149b5e25fSSatish Balay 
31249b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
31349b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
31449b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
31549b5e25fSSatish Balay     ib = aj + *ai;
316831a3094SHong Zhang     jmin = 0;
3177fbae186SHong Zhang     if (*ib == i){     /* (diag of A)*x */
31849b5e25fSSatish Balay       z[4*i]   += v[0]*x1 + v[4]*x2 +  v[8]*x3 + v[12]*x4;
31949b5e25fSSatish Balay       z[4*i+1] += v[4]*x1 + v[5]*x2 +  v[9]*x3 + v[13]*x4;
32049b5e25fSSatish Balay       z[4*i+2] += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[14]*x4;
32149b5e25fSSatish Balay       z[4*i+3] += v[12]*x1+ v[13]*x2+ v[14]*x3 + v[15]*x4;
322831a3094SHong Zhang       v += 16; jmin++;
3237fbae186SHong Zhang     }
324831a3094SHong Zhang     for (j=jmin; j<n; j++) {
32549b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
32649b5e25fSSatish Balay       cval       = ib[j]*4;
32749b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4;
32849b5e25fSSatish Balay       z[cval+1]   += v[4]*x1 + v[5]*x2 + v[6]*x3 + v[7]*x4;
32949b5e25fSSatish Balay       z[cval+2]   += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[11]*x4;
33049b5e25fSSatish Balay       z[cval+3]   += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4;
33149b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
33249b5e25fSSatish Balay       z[4*i]   += v[0]*x[cval] + v[4]*x[cval+1]+ v[8]*x[cval+2] + v[12]*x[cval+3];
33349b5e25fSSatish Balay       z[4*i+1] += v[1]*x[cval] + v[5]*x[cval+1]+ v[9]*x[cval+2] + v[13]*x[cval+3];
33449b5e25fSSatish Balay       z[4*i+2] += v[2]*x[cval] + v[6]*x[cval+1]+ v[10]*x[cval+2]+ v[14]*x[cval+3];
33549b5e25fSSatish Balay       z[4*i+3] += v[3]*x[cval] + v[7]*x[cval+1]+ v[11]*x[cval+2]+ v[15]*x[cval+3];
33649b5e25fSSatish Balay       v  += 16;
33749b5e25fSSatish Balay     }
33849b5e25fSSatish Balay     xb +=4; ai++;
33949b5e25fSSatish Balay   }
34049b5e25fSSatish Balay 
341*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
342*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
3436c6c5352SBarry Smith   PetscLogFlops(32*(a->nz*2 - A->m) - A->m);
34449b5e25fSSatish Balay   PetscFunctionReturn(0);
34549b5e25fSSatish Balay }
34649b5e25fSSatish Balay 
3474a2ae208SSatish Balay #undef __FUNCT__
3484a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqSBAIJ_5"
34949b5e25fSSatish Balay int MatMult_SeqSBAIJ_5(Mat A,Vec xx,Vec zz)
35049b5e25fSSatish Balay {
35149b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
35287828ca2SBarry Smith   PetscScalar     *x,*z,*xb,x1,x2,x3,x4,x5,zero=0.0;
35349b5e25fSSatish Balay   MatScalar       *v;
354831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
35549b5e25fSSatish Balay 
35649b5e25fSSatish Balay   PetscFunctionBegin;
35749b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
358*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
359*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
36049b5e25fSSatish Balay 
36149b5e25fSSatish Balay   v     = a->a;
36249b5e25fSSatish Balay   xb = x;
36349b5e25fSSatish Balay 
36449b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
36549b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
36649b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4];
36749b5e25fSSatish Balay     ib = aj + *ai;
368831a3094SHong Zhang     jmin = 0;
3697fbae186SHong Zhang     if (*ib == i){      /* (diag of A)*x */
37049b5e25fSSatish Balay       z[5*i]   += v[0]*x1  + v[5]*x2 + v[10]*x3 + v[15]*x4+ v[20]*x5;
37149b5e25fSSatish Balay       z[5*i+1] += v[5]*x1  + v[6]*x2 + v[11]*x3 + v[16]*x4+ v[21]*x5;
37249b5e25fSSatish Balay       z[5*i+2] += v[10]*x1 +v[11]*x2 + v[12]*x3 + v[17]*x4+ v[22]*x5;
37349b5e25fSSatish Balay       z[5*i+3] += v[15]*x1 +v[16]*x2 + v[17]*x3 + v[18]*x4+ v[23]*x5;
37449b5e25fSSatish Balay       z[5*i+4] += v[20]*x1 +v[21]*x2 + v[22]*x3 + v[23]*x4+ v[24]*x5;
375831a3094SHong Zhang       v += 25; jmin++;
3767fbae186SHong Zhang     }
377831a3094SHong Zhang     for (j=jmin; j<n; j++) {
37849b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
37949b5e25fSSatish Balay       cval       = ib[j]*5;
38049b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4 + v[4]*x5;
38149b5e25fSSatish Balay       z[cval+1]   += v[5]*x1 + v[6]*x2 + v[7]*x3 + v[8]*x4 + v[9]*x5;
38249b5e25fSSatish Balay       z[cval+2]   += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4+ v[14]*x5;
38349b5e25fSSatish Balay       z[cval+3]   += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4+ v[19]*x5;
38449b5e25fSSatish Balay       z[cval+4]   += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4+ v[24]*x5;
38549b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
38649b5e25fSSatish 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];
38749b5e25fSSatish 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];
38849b5e25fSSatish 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];
38949b5e25fSSatish 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];
39049b5e25fSSatish 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];
39149b5e25fSSatish Balay       v  += 25;
39249b5e25fSSatish Balay     }
39349b5e25fSSatish Balay     xb +=5; ai++;
39449b5e25fSSatish Balay   }
39549b5e25fSSatish Balay 
396*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
397*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
3986c6c5352SBarry Smith   PetscLogFlops(50*(a->nz*2 - A->m) - A->m);
39949b5e25fSSatish Balay   PetscFunctionReturn(0);
40049b5e25fSSatish Balay }
40149b5e25fSSatish Balay 
40249b5e25fSSatish Balay 
4034a2ae208SSatish Balay #undef __FUNCT__
4044a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqSBAIJ_6"
40549b5e25fSSatish Balay int MatMult_SeqSBAIJ_6(Mat A,Vec xx,Vec zz)
40649b5e25fSSatish Balay {
40749b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
40887828ca2SBarry Smith   PetscScalar     *x,*z,*xb,x1,x2,x3,x4,x5,x6,zero=0.0;
40949b5e25fSSatish Balay   MatScalar       *v;
410831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
41149b5e25fSSatish Balay 
41249b5e25fSSatish Balay   PetscFunctionBegin;
41349b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
414*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
415*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
41649b5e25fSSatish Balay 
41749b5e25fSSatish Balay   v     = a->a;
41849b5e25fSSatish Balay   xb = x;
41949b5e25fSSatish Balay 
42049b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
42149b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
42249b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4]; x6=xb[5];
42349b5e25fSSatish Balay     ib = aj + *ai;
424831a3094SHong Zhang     jmin = 0;
4257fbae186SHong Zhang     if (*ib == i){      /* (diag of A)*x */
42649b5e25fSSatish Balay       z[6*i]   += v[0]*x1  + v[6]*x2 + v[12]*x3 + v[18]*x4+ v[24]*x5 + v[30]*x6;
42749b5e25fSSatish Balay       z[6*i+1] += v[6]*x1  + v[7]*x2 + v[13]*x3 + v[19]*x4+ v[25]*x5 + v[31]*x6;
42849b5e25fSSatish Balay       z[6*i+2] += v[12]*x1 +v[13]*x2 + v[14]*x3 + v[20]*x4+ v[26]*x5 + v[32]*x6;
42949b5e25fSSatish Balay       z[6*i+3] += v[18]*x1 +v[19]*x2 + v[20]*x3 + v[21]*x4+ v[27]*x5 + v[33]*x6;
43049b5e25fSSatish Balay       z[6*i+4] += v[24]*x1 +v[25]*x2 + v[26]*x3 + v[27]*x4+ v[28]*x5 + v[34]*x6;
43149b5e25fSSatish Balay       z[6*i+5] += v[30]*x1 +v[31]*x2 + v[32]*x3 + v[33]*x4+ v[34]*x5 + v[35]*x6;
432831a3094SHong Zhang       v += 36; jmin++;
4337fbae186SHong Zhang     }
434831a3094SHong Zhang     for (j=jmin; j<n; j++) {
43549b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
43649b5e25fSSatish Balay       cval       = ib[j]*6;
43749b5e25fSSatish Balay       z[cval]   += v[0]*x1  + v[1]*x2 + v[2]*x3 + v[3]*x4+ v[4]*x5 + v[5]*x6;
43849b5e25fSSatish Balay       z[cval+1] += v[6]*x1  + v[7]*x2 + v[8]*x3 + v[9]*x4+ v[10]*x5 + v[11]*x6;
43949b5e25fSSatish Balay       z[cval+2] += v[12]*x1  + v[13]*x2 + v[14]*x3 + v[15]*x4+ v[16]*x5 + v[17]*x6;
44049b5e25fSSatish Balay       z[cval+3] += v[18]*x1  + v[19]*x2 + v[20]*x3 + v[21]*x4+ v[22]*x5 + v[23]*x6;
44149b5e25fSSatish Balay       z[cval+4] += v[24]*x1  + v[25]*x2 + v[26]*x3 + v[27]*x4+ v[28]*x5 + v[29]*x6;
44249b5e25fSSatish Balay       z[cval+5] += v[30]*x1  + v[31]*x2 + v[32]*x3 + v[33]*x4+ v[34]*x5 + v[35]*x6;
44349b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
44449b5e25fSSatish 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];
44549b5e25fSSatish 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];
44649b5e25fSSatish 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];
44749b5e25fSSatish 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];
44849b5e25fSSatish 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];
44949b5e25fSSatish 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];
45049b5e25fSSatish Balay       v  += 36;
45149b5e25fSSatish Balay     }
45249b5e25fSSatish Balay     xb +=6; ai++;
45349b5e25fSSatish Balay   }
45449b5e25fSSatish Balay 
455*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
456*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
4576c6c5352SBarry Smith   PetscLogFlops(72*(a->nz*2 - A->m) - A->m);
45849b5e25fSSatish Balay   PetscFunctionReturn(0);
45949b5e25fSSatish Balay }
4604a2ae208SSatish Balay #undef __FUNCT__
4614a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqSBAIJ_7"
46249b5e25fSSatish Balay int MatMult_SeqSBAIJ_7(Mat A,Vec xx,Vec zz)
46349b5e25fSSatish Balay {
46449b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
46587828ca2SBarry Smith   PetscScalar     *x,*z,*xb,x1,x2,x3,x4,x5,x6,x7,zero=0.0;
46649b5e25fSSatish Balay   MatScalar       *v;
467831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
46849b5e25fSSatish Balay 
46949b5e25fSSatish Balay   PetscFunctionBegin;
47049b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
471*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
472*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
47349b5e25fSSatish Balay 
47449b5e25fSSatish Balay   v     = a->a;
47549b5e25fSSatish Balay   xb = x;
47649b5e25fSSatish Balay 
47749b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
47849b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
47949b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4]; x6=xb[5]; x7=xb[6];
48049b5e25fSSatish Balay     ib = aj + *ai;
481831a3094SHong Zhang     jmin = 0;
4827fbae186SHong Zhang     if (*ib == i){      /* (diag of A)*x */
48349b5e25fSSatish 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;
48449b5e25fSSatish 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;
48549b5e25fSSatish 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;
48649b5e25fSSatish 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;
48749b5e25fSSatish 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;
48849b5e25fSSatish 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;
48949b5e25fSSatish 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;
490831a3094SHong Zhang       v += 49; jmin++;
4917fbae186SHong Zhang     }
492831a3094SHong Zhang     for (j=jmin; j<n; j++) {
49349b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
49449b5e25fSSatish Balay       cval       = ib[j]*7;
49549b5e25fSSatish Balay       z[cval]   += v[0]*x1  + v[1]*x2 + v[2]*x3 + v[3]*x4+ v[4]*x5 + v[5]*x6+ v[6]*x7;
49649b5e25fSSatish 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;
49749b5e25fSSatish 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;
49849b5e25fSSatish 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;
49949b5e25fSSatish 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;
50049b5e25fSSatish 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;
50149b5e25fSSatish 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;
50249b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
50349b5e25fSSatish 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];
50449b5e25fSSatish 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];
50549b5e25fSSatish 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];
50649b5e25fSSatish 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];
50749b5e25fSSatish 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];
50849b5e25fSSatish 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];
50949b5e25fSSatish 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];
51049b5e25fSSatish Balay       v  += 49;
51149b5e25fSSatish Balay     }
51249b5e25fSSatish Balay     xb +=7; ai++;
51349b5e25fSSatish Balay   }
514*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
515*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
5166c6c5352SBarry Smith   PetscLogFlops(98*(a->nz*2 - A->m) - A->m);
51749b5e25fSSatish Balay   PetscFunctionReturn(0);
51849b5e25fSSatish Balay }
51949b5e25fSSatish Balay 
52049b5e25fSSatish Balay /*
52149b5e25fSSatish Balay     This will not work with MatScalar == float because it calls the BLAS
52249b5e25fSSatish Balay */
5234a2ae208SSatish Balay #undef __FUNCT__
5244a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqSBAIJ_N"
52549b5e25fSSatish Balay int MatMult_SeqSBAIJ_N(Mat A,Vec xx,Vec zz)
52649b5e25fSSatish Balay {
52749b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
52887828ca2SBarry Smith   PetscScalar     *x,*x_ptr,*z,*z_ptr,*xb,*zb,*work,*workt,zero=0.0;
5290b60a74dSHong Zhang   MatScalar       *v;
530066653e3SSatish Balay   int             ierr,mbs=a->mbs,i,*idx,*aj,*ii,bs=a->bs,j,n,bs2=a->bs2;
5310b60a74dSHong Zhang   int             ncols,k;
53249b5e25fSSatish Balay 
53349b5e25fSSatish Balay   PetscFunctionBegin;
53449b5e25fSSatish Balay   ierr = VecSet(&zero,zz);CHKERRQ(ierr);
535*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr); x_ptr=x;
536*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr); z_ptr=z;
53749b5e25fSSatish Balay 
53849b5e25fSSatish Balay   aj   = a->j;
53949b5e25fSSatish Balay   v    = a->a;
54049b5e25fSSatish Balay   ii   = a->i;
54149b5e25fSSatish Balay 
54249b5e25fSSatish Balay   if (!a->mult_work) {
54387828ca2SBarry Smith     ierr = PetscMalloc((A->m+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
54449b5e25fSSatish Balay   }
54549b5e25fSSatish Balay   work = a->mult_work;
54649b5e25fSSatish Balay 
54749b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
54849b5e25fSSatish Balay     n     = ii[1] - ii[0]; ncols = n*bs;
54949b5e25fSSatish Balay     workt = work; idx=aj+ii[0];
55049b5e25fSSatish Balay 
55149b5e25fSSatish Balay     /* upper triangular part */
55249b5e25fSSatish Balay     for (j=0; j<n; j++) {
55349b5e25fSSatish Balay       xb = x_ptr + bs*(*idx++);
55449b5e25fSSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
55549b5e25fSSatish Balay       workt += bs;
55649b5e25fSSatish Balay     }
55749b5e25fSSatish Balay     /* z(i*bs:(i+1)*bs-1) += A(i,:)*x */
55849b5e25fSSatish Balay     Kernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z);
55949b5e25fSSatish Balay 
56049b5e25fSSatish Balay     /* strict lower triangular part */
561831a3094SHong Zhang     idx = aj+ii[0];
562831a3094SHong Zhang     if (*idx == i){
56396b9376eSHong Zhang       ncols -= bs; v += bs2; idx++; n--;
564831a3094SHong Zhang     }
56596b9376eSHong Zhang 
56649b5e25fSSatish Balay     if (ncols > 0){
56749b5e25fSSatish Balay       workt = work;
56887828ca2SBarry Smith       ierr  = PetscMemzero(workt,ncols*sizeof(PetscScalar));CHKERRQ(ierr);
569831a3094SHong Zhang       Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,x,v,workt);
570831a3094SHong Zhang       for (j=0; j<n; j++) {
571831a3094SHong Zhang         zb = z_ptr + bs*(*idx++);
57249b5e25fSSatish Balay         for (k=0; k<bs; k++) zb[k] += workt[k] ;
57349b5e25fSSatish Balay         workt += bs;
57449b5e25fSSatish Balay       }
57549b5e25fSSatish Balay     }
57649b5e25fSSatish Balay     x += bs; v += n*bs2; z += bs; ii++;
57749b5e25fSSatish Balay   }
57849b5e25fSSatish Balay 
579*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
580*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
5816c6c5352SBarry Smith   PetscLogFlops(2*(a->nz*2 - A->m)*bs2 - A->m);
58249b5e25fSSatish Balay   PetscFunctionReturn(0);
58349b5e25fSSatish Balay }
58449b5e25fSSatish Balay 
5854a2ae208SSatish Balay #undef __FUNCT__
5864a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqSBAIJ_1"
58749b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_1(Mat A,Vec xx,Vec yy,Vec zz)
58849b5e25fSSatish Balay {
58949b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
59087828ca2SBarry Smith   PetscScalar     *x,*y,*z,*xb,x1;
59149b5e25fSSatish Balay   MatScalar       *v;
592831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
59349b5e25fSSatish Balay 
59449b5e25fSSatish Balay   PetscFunctionBegin;
595*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
59649b5e25fSSatish Balay   if (yy != xx) {
597*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(yy,&y);CHKERRQ(ierr);
59849b5e25fSSatish Balay   } else {
59949b5e25fSSatish Balay     y = x;
60049b5e25fSSatish Balay   }
60149b5e25fSSatish Balay   if (zz != yy) {
602a9d4b620SHong Zhang     /* ierr = VecCopy(yy,zz);CHKERRQ(ierr); */
603*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
604a9d4b620SHong Zhang     ierr = PetscMemcpy(z,y,yy->n*sizeof(PetscScalar));CHKERRQ(ierr);
60549b5e25fSSatish Balay   } else {
60649b5e25fSSatish Balay     z = y;
60749b5e25fSSatish Balay   }
60849b5e25fSSatish Balay 
60949b5e25fSSatish Balay   v  = a->a;
61049b5e25fSSatish Balay   xb = x;
61149b5e25fSSatish Balay 
61249b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
61349b5e25fSSatish Balay     n  = ai[1] - ai[0];  /* length of i_th row of A */
61449b5e25fSSatish Balay     x1 = xb[0];
61549b5e25fSSatish Balay     ib = aj + *ai;
616831a3094SHong Zhang     jmin = 0;
617831a3094SHong Zhang     if (*ib == i) {            /* (diag of A)*x */
618831a3094SHong Zhang       z[i] += *v++ * x[*ib++]; jmin++;
619831a3094SHong Zhang     }
620831a3094SHong Zhang     for (j=jmin; j<n; j++) {
62149b5e25fSSatish Balay       cval    = *ib;
62249b5e25fSSatish Balay       z[cval] += *v * x1;      /* (strict lower triangular part of A)*x  */
62349b5e25fSSatish Balay       z[i] += *v++ * x[*ib++]; /* (strict upper triangular part of A)*x  */
62449b5e25fSSatish Balay     }
62549b5e25fSSatish Balay     xb++; ai++;
62649b5e25fSSatish Balay   }
62749b5e25fSSatish Balay 
628*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
629*b1d4fb26SBarry Smith   if (yy != xx) ierr = VecRestoreArrayFast(yy,&y);CHKERRQ(ierr);
630*b1d4fb26SBarry Smith   if (zz != yy) ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
63149b5e25fSSatish Balay 
6326c6c5352SBarry Smith   PetscLogFlops(2*(a->nz*2 - A->m));
63349b5e25fSSatish Balay   PetscFunctionReturn(0);
63449b5e25fSSatish Balay }
63549b5e25fSSatish Balay 
6364a2ae208SSatish Balay #undef __FUNCT__
6374a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqSBAIJ_2"
63849b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_2(Mat A,Vec xx,Vec yy,Vec zz)
63949b5e25fSSatish Balay {
64049b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
64187828ca2SBarry Smith   PetscScalar     *x,*y,*z,*xb,x1,x2;
64249b5e25fSSatish Balay   MatScalar       *v;
643831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
64449b5e25fSSatish Balay 
64549b5e25fSSatish Balay   PetscFunctionBegin;
646*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
64749b5e25fSSatish Balay   if (yy != xx) {
648*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(yy,&y);CHKERRQ(ierr);
64949b5e25fSSatish Balay   } else {
65049b5e25fSSatish Balay     y = x;
65149b5e25fSSatish Balay   }
65249b5e25fSSatish Balay   if (zz != yy) {
653a9d4b620SHong Zhang     /* ierr = VecCopy(yy,zz);CHKERRQ(ierr); */
654*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
655a9d4b620SHong Zhang     ierr = PetscMemcpy(z,y,yy->n*sizeof(PetscScalar));CHKERRQ(ierr);
65649b5e25fSSatish Balay   } else {
65749b5e25fSSatish Balay     z = y;
65849b5e25fSSatish Balay   }
65949b5e25fSSatish Balay 
66049b5e25fSSatish Balay   v     = a->a;
66149b5e25fSSatish Balay   xb = x;
66249b5e25fSSatish Balay 
66349b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
66449b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
66549b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1];
66649b5e25fSSatish Balay     ib = aj + *ai;
667831a3094SHong Zhang     jmin = 0;
6687fbae186SHong Zhang     if (*ib == i){      /* (diag of A)*x */
66949b5e25fSSatish Balay       z[2*i]   += v[0]*x1 + v[2]*x2;
67049b5e25fSSatish Balay       z[2*i+1] += v[2]*x1 + v[3]*x2;
671831a3094SHong Zhang       v += 4; jmin++;
6727fbae186SHong Zhang     }
673831a3094SHong Zhang     for (j=jmin; j<n; j++) {
67449b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
67549b5e25fSSatish Balay       cval       = ib[j]*2;
67649b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2;
67749b5e25fSSatish Balay       z[cval+1]   += v[2]*x1 + v[3]*x2;
67849b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
67949b5e25fSSatish Balay       z[2*i]   += v[0]*x[cval] + v[2]*x[cval+1];
68049b5e25fSSatish Balay       z[2*i+1] += v[1]*x[cval] + v[3]*x[cval+1];
68149b5e25fSSatish Balay       v  += 4;
68249b5e25fSSatish Balay     }
68349b5e25fSSatish Balay     xb +=2; ai++;
68449b5e25fSSatish Balay   }
68549b5e25fSSatish Balay 
686*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
687*b1d4fb26SBarry Smith   if (yy != xx) ierr = VecRestoreArrayFast(yy,&y);CHKERRQ(ierr);
688*b1d4fb26SBarry Smith   if (zz != yy) ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
68949b5e25fSSatish Balay 
6906c6c5352SBarry Smith   PetscLogFlops(4*(a->nz*2 - A->m));
69149b5e25fSSatish Balay   PetscFunctionReturn(0);
69249b5e25fSSatish Balay }
69349b5e25fSSatish Balay 
6944a2ae208SSatish Balay #undef __FUNCT__
6954a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqSBAIJ_3"
69649b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_3(Mat A,Vec xx,Vec yy,Vec zz)
69749b5e25fSSatish Balay {
69849b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
69987828ca2SBarry Smith   PetscScalar     *x,*y,*z,*xb,x1,x2,x3;
70049b5e25fSSatish Balay   MatScalar       *v;
701831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
70249b5e25fSSatish Balay 
70349b5e25fSSatish Balay   PetscFunctionBegin;
704*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
70549b5e25fSSatish Balay   if (yy != xx) {
706*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(yy,&y);CHKERRQ(ierr);
70749b5e25fSSatish Balay   } else {
70849b5e25fSSatish Balay     y = x;
70949b5e25fSSatish Balay   }
71049b5e25fSSatish Balay   if (zz != yy) {
711a9d4b620SHong Zhang     /* ierr = VecCopy(yy,zz);CHKERRQ(ierr); */
712*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
713a9d4b620SHong Zhang     ierr = PetscMemcpy(z,y,yy->n*sizeof(PetscScalar));CHKERRQ(ierr);
71449b5e25fSSatish Balay   } else {
71549b5e25fSSatish Balay     z = y;
71649b5e25fSSatish Balay   }
71749b5e25fSSatish Balay 
71849b5e25fSSatish Balay   v     = a->a;
71949b5e25fSSatish Balay   xb = x;
72049b5e25fSSatish Balay 
72149b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
72249b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
72349b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2];
72449b5e25fSSatish Balay     ib = aj + *ai;
725831a3094SHong Zhang     jmin = 0;
7267fbae186SHong Zhang     if (*ib == i){     /* (diag of A)*x */
72749b5e25fSSatish Balay      z[3*i]   += v[0]*x1 + v[3]*x2 + v[6]*x3;
72849b5e25fSSatish Balay      z[3*i+1] += v[3]*x1 + v[4]*x2 + v[7]*x3;
72949b5e25fSSatish Balay      z[3*i+2] += v[6]*x1 + v[7]*x2 + v[8]*x3;
730831a3094SHong Zhang      v += 9; jmin++;
7317fbae186SHong Zhang     }
732831a3094SHong Zhang     for (j=jmin; j<n; j++) {
73349b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
73449b5e25fSSatish Balay       cval       = ib[j]*3;
73549b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3;
73649b5e25fSSatish Balay       z[cval+1]   += v[3]*x1 + v[4]*x2 + v[5]*x3;
73749b5e25fSSatish Balay       z[cval+2]   += v[6]*x1 + v[7]*x2 + v[8]*x3;
73849b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
73949b5e25fSSatish Balay       z[3*i]   += v[0]*x[cval] + v[3]*x[cval+1]+ v[6]*x[cval+2];
74049b5e25fSSatish Balay       z[3*i+1] += v[1]*x[cval] + v[4]*x[cval+1]+ v[7]*x[cval+2];
74149b5e25fSSatish Balay       z[3*i+2] += v[2]*x[cval] + v[5]*x[cval+1]+ v[8]*x[cval+2];
74249b5e25fSSatish Balay       v  += 9;
74349b5e25fSSatish Balay     }
74449b5e25fSSatish Balay     xb +=3; ai++;
74549b5e25fSSatish Balay   }
74649b5e25fSSatish Balay 
747*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
748*b1d4fb26SBarry Smith   if (yy != xx) ierr = VecRestoreArrayFast(yy,&y);CHKERRQ(ierr);
749*b1d4fb26SBarry Smith   if (zz != yy) ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
75049b5e25fSSatish Balay 
7516c6c5352SBarry Smith   PetscLogFlops(18*(a->nz*2 - A->m));
75249b5e25fSSatish Balay   PetscFunctionReturn(0);
75349b5e25fSSatish Balay }
75449b5e25fSSatish Balay 
7554a2ae208SSatish Balay #undef __FUNCT__
7564a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqSBAIJ_4"
75749b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_4(Mat A,Vec xx,Vec yy,Vec zz)
75849b5e25fSSatish Balay {
75949b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
76087828ca2SBarry Smith   PetscScalar     *x,*y,*z,*xb,x1,x2,x3,x4;
76149b5e25fSSatish Balay   MatScalar       *v;
762831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
76349b5e25fSSatish Balay 
76449b5e25fSSatish Balay   PetscFunctionBegin;
765*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
76649b5e25fSSatish Balay   if (yy != xx) {
767*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(yy,&y);CHKERRQ(ierr);
76849b5e25fSSatish Balay   } else {
76949b5e25fSSatish Balay     y = x;
77049b5e25fSSatish Balay   }
77149b5e25fSSatish Balay   if (zz != yy) {
772a9d4b620SHong Zhang     /* ierr = VecCopy(yy,zz);CHKERRQ(ierr); */
773*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
774a9d4b620SHong Zhang     ierr = PetscMemcpy(z,y,yy->n*sizeof(PetscScalar));CHKERRQ(ierr);
77549b5e25fSSatish Balay   } else {
77649b5e25fSSatish Balay     z = y;
77749b5e25fSSatish Balay   }
77849b5e25fSSatish Balay 
77949b5e25fSSatish Balay   v     = a->a;
78049b5e25fSSatish Balay   xb = x;
78149b5e25fSSatish Balay 
78249b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
78349b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
78449b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3];
78549b5e25fSSatish Balay     ib = aj + *ai;
786831a3094SHong Zhang     jmin = 0;
7877fbae186SHong Zhang     if (*ib == i){      /* (diag of A)*x */
78849b5e25fSSatish Balay       z[4*i]   += v[0]*x1 + v[4]*x2 +  v[8]*x3 + v[12]*x4;
78949b5e25fSSatish Balay       z[4*i+1] += v[4]*x1 + v[5]*x2 +  v[9]*x3 + v[13]*x4;
79049b5e25fSSatish Balay       z[4*i+2] += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[14]*x4;
79149b5e25fSSatish Balay       z[4*i+3] += v[12]*x1+ v[13]*x2+ v[14]*x3 + v[15]*x4;
792831a3094SHong Zhang       v += 16; jmin++;
7937fbae186SHong Zhang     }
794831a3094SHong Zhang     for (j=jmin; j<n; j++) {
79549b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
79649b5e25fSSatish Balay       cval       = ib[j]*4;
79749b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4;
79849b5e25fSSatish Balay       z[cval+1]   += v[4]*x1 + v[5]*x2 + v[6]*x3 + v[7]*x4;
79949b5e25fSSatish Balay       z[cval+2]   += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[11]*x4;
80049b5e25fSSatish Balay       z[cval+3]   += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4;
80149b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
80249b5e25fSSatish Balay       z[4*i]   += v[0]*x[cval] + v[4]*x[cval+1]+ v[8]*x[cval+2] + v[12]*x[cval+3];
80349b5e25fSSatish Balay       z[4*i+1] += v[1]*x[cval] + v[5]*x[cval+1]+ v[9]*x[cval+2] + v[13]*x[cval+3];
80449b5e25fSSatish Balay       z[4*i+2] += v[2]*x[cval] + v[6]*x[cval+1]+ v[10]*x[cval+2]+ v[14]*x[cval+3];
80549b5e25fSSatish Balay       z[4*i+3] += v[3]*x[cval] + v[7]*x[cval+1]+ v[11]*x[cval+2]+ v[15]*x[cval+3];
80649b5e25fSSatish Balay       v  += 16;
80749b5e25fSSatish Balay     }
80849b5e25fSSatish Balay     xb +=4; ai++;
80949b5e25fSSatish Balay   }
81049b5e25fSSatish Balay 
811*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
812*b1d4fb26SBarry Smith   if (yy != xx) ierr = VecRestoreArrayFast(yy,&y);CHKERRQ(ierr);
813*b1d4fb26SBarry Smith   if (zz != yy) ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
81449b5e25fSSatish Balay 
8156c6c5352SBarry Smith   PetscLogFlops(32*(a->nz*2 - A->m));
81649b5e25fSSatish Balay   PetscFunctionReturn(0);
81749b5e25fSSatish Balay }
81849b5e25fSSatish Balay 
8194a2ae208SSatish Balay #undef __FUNCT__
8204a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqSBAIJ_5"
82149b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_5(Mat A,Vec xx,Vec yy,Vec zz)
82249b5e25fSSatish Balay {
82349b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
82487828ca2SBarry Smith   PetscScalar     *x,*y,*z,*xb,x1,x2,x3,x4,x5;
82549b5e25fSSatish Balay   MatScalar       *v;
826831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
82749b5e25fSSatish Balay 
82849b5e25fSSatish Balay   PetscFunctionBegin;
829*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
83049b5e25fSSatish Balay   if (yy != xx) {
831*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(yy,&y);CHKERRQ(ierr);
83249b5e25fSSatish Balay   } else {
83349b5e25fSSatish Balay     y = x;
83449b5e25fSSatish Balay   }
83549b5e25fSSatish Balay   if (zz != yy) {
836a9d4b620SHong Zhang     /* ierr = VecCopy(yy,zz);CHKERRQ(ierr); */
837*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
838a9d4b620SHong Zhang     ierr = PetscMemcpy(z,y,yy->n*sizeof(PetscScalar));CHKERRQ(ierr);
83949b5e25fSSatish Balay   } else {
84049b5e25fSSatish Balay     z = y;
84149b5e25fSSatish Balay   }
84249b5e25fSSatish Balay 
84349b5e25fSSatish Balay   v     = a->a;
84449b5e25fSSatish Balay   xb = x;
84549b5e25fSSatish Balay 
84649b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
84749b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
84849b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4];
84949b5e25fSSatish Balay     ib = aj + *ai;
850831a3094SHong Zhang     jmin = 0;
8517fbae186SHong Zhang     if (*ib == i){      /* (diag of A)*x */
85249b5e25fSSatish Balay       z[5*i]   += v[0]*x1  + v[5]*x2 + v[10]*x3 + v[15]*x4+ v[20]*x5;
85349b5e25fSSatish Balay       z[5*i+1] += v[5]*x1  + v[6]*x2 + v[11]*x3 + v[16]*x4+ v[21]*x5;
85449b5e25fSSatish Balay       z[5*i+2] += v[10]*x1 +v[11]*x2 + v[12]*x3 + v[17]*x4+ v[22]*x5;
85549b5e25fSSatish Balay       z[5*i+3] += v[15]*x1 +v[16]*x2 + v[17]*x3 + v[18]*x4+ v[23]*x5;
85649b5e25fSSatish Balay       z[5*i+4] += v[20]*x1 +v[21]*x2 + v[22]*x3 + v[23]*x4+ v[24]*x5;
857831a3094SHong Zhang       v += 25; jmin++;
8587fbae186SHong Zhang     }
859831a3094SHong Zhang     for (j=jmin; j<n; j++) {
86049b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
86149b5e25fSSatish Balay       cval       = ib[j]*5;
86249b5e25fSSatish Balay       z[cval]     += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4 + v[4]*x5;
86349b5e25fSSatish Balay       z[cval+1]   += v[5]*x1 + v[6]*x2 + v[7]*x3 + v[8]*x4 + v[9]*x5;
86449b5e25fSSatish Balay       z[cval+2]   += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4+ v[14]*x5;
86549b5e25fSSatish Balay       z[cval+3]   += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4+ v[19]*x5;
86649b5e25fSSatish Balay       z[cval+4]   += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4+ v[24]*x5;
86749b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
86849b5e25fSSatish 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];
86949b5e25fSSatish 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];
87049b5e25fSSatish 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];
87149b5e25fSSatish 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];
87249b5e25fSSatish 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];
87349b5e25fSSatish Balay       v  += 25;
87449b5e25fSSatish Balay     }
87549b5e25fSSatish Balay     xb +=5; ai++;
87649b5e25fSSatish Balay   }
87749b5e25fSSatish Balay 
878*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
879*b1d4fb26SBarry Smith   if (yy != xx) ierr = VecRestoreArrayFast(yy,&y);CHKERRQ(ierr);
880*b1d4fb26SBarry Smith   if (zz != yy) ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
88149b5e25fSSatish Balay 
8826c6c5352SBarry Smith   PetscLogFlops(50*(a->nz*2 - A->m));
88349b5e25fSSatish Balay   PetscFunctionReturn(0);
88449b5e25fSSatish Balay }
8854a2ae208SSatish Balay #undef __FUNCT__
8864a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqSBAIJ_6"
88749b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_6(Mat A,Vec xx,Vec yy,Vec zz)
88849b5e25fSSatish Balay {
88949b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
89087828ca2SBarry Smith   PetscScalar     *x,*y,*z,*xb,x1,x2,x3,x4,x5,x6;
89149b5e25fSSatish Balay   MatScalar       *v;
892831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
89349b5e25fSSatish Balay 
89449b5e25fSSatish Balay   PetscFunctionBegin;
895*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
89649b5e25fSSatish Balay   if (yy != xx) {
897*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(yy,&y);CHKERRQ(ierr);
89849b5e25fSSatish Balay   } else {
89949b5e25fSSatish Balay     y = x;
90049b5e25fSSatish Balay   }
90149b5e25fSSatish Balay   if (zz != yy) {
902a9d4b620SHong Zhang     /* ierr = VecCopy(yy,zz);CHKERRQ(ierr); */
903*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
904a9d4b620SHong Zhang     ierr = PetscMemcpy(z,y,yy->n*sizeof(PetscScalar));CHKERRQ(ierr);
90549b5e25fSSatish Balay   } else {
90649b5e25fSSatish Balay     z = y;
90749b5e25fSSatish Balay   }
90849b5e25fSSatish Balay 
90949b5e25fSSatish Balay   v     = a->a;
91049b5e25fSSatish Balay   xb = x;
91149b5e25fSSatish Balay 
91249b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
91349b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
91449b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4]; x6=xb[5];
91549b5e25fSSatish Balay     ib = aj + *ai;
916831a3094SHong Zhang     jmin = 0;
9177fbae186SHong Zhang     if (*ib == i){     /* (diag of A)*x */
91849b5e25fSSatish Balay       z[6*i]   += v[0]*x1  + v[6]*x2 + v[12]*x3 + v[18]*x4+ v[24]*x5 + v[30]*x6;
91949b5e25fSSatish Balay       z[6*i+1] += v[6]*x1  + v[7]*x2 + v[13]*x3 + v[19]*x4+ v[25]*x5 + v[31]*x6;
92049b5e25fSSatish Balay       z[6*i+2] += v[12]*x1 +v[13]*x2 + v[14]*x3 + v[20]*x4+ v[26]*x5 + v[32]*x6;
92149b5e25fSSatish Balay       z[6*i+3] += v[18]*x1 +v[19]*x2 + v[20]*x3 + v[21]*x4+ v[27]*x5 + v[33]*x6;
92249b5e25fSSatish Balay       z[6*i+4] += v[24]*x1 +v[25]*x2 + v[26]*x3 + v[27]*x4+ v[28]*x5 + v[34]*x6;
92349b5e25fSSatish Balay       z[6*i+5] += v[30]*x1 +v[31]*x2 + v[32]*x3 + v[33]*x4+ v[34]*x5 + v[35]*x6;
924831a3094SHong Zhang       v += 36; jmin++;
9257fbae186SHong Zhang     }
926831a3094SHong Zhang     for (j=jmin; j<n; j++) {
92749b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
92849b5e25fSSatish Balay       cval       = ib[j]*6;
92949b5e25fSSatish Balay       z[cval]   += v[0]*x1  + v[1]*x2 + v[2]*x3 + v[3]*x4+ v[4]*x5 + v[5]*x6;
93049b5e25fSSatish Balay       z[cval+1] += v[6]*x1  + v[7]*x2 + v[8]*x3 + v[9]*x4+ v[10]*x5 + v[11]*x6;
93149b5e25fSSatish Balay       z[cval+2] += v[12]*x1  + v[13]*x2 + v[14]*x3 + v[15]*x4+ v[16]*x5 + v[17]*x6;
93249b5e25fSSatish Balay       z[cval+3] += v[18]*x1  + v[19]*x2 + v[20]*x3 + v[21]*x4+ v[22]*x5 + v[23]*x6;
93349b5e25fSSatish Balay       z[cval+4] += v[24]*x1  + v[25]*x2 + v[26]*x3 + v[27]*x4+ v[28]*x5 + v[29]*x6;
93449b5e25fSSatish Balay       z[cval+5] += v[30]*x1  + v[31]*x2 + v[32]*x3 + v[33]*x4+ v[34]*x5 + v[35]*x6;
93549b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
93649b5e25fSSatish 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];
93749b5e25fSSatish 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];
93849b5e25fSSatish 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];
93949b5e25fSSatish 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];
94049b5e25fSSatish 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];
94149b5e25fSSatish 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];
94249b5e25fSSatish Balay       v  += 36;
94349b5e25fSSatish Balay     }
94449b5e25fSSatish Balay     xb +=6; ai++;
94549b5e25fSSatish Balay   }
94649b5e25fSSatish Balay 
947*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
948*b1d4fb26SBarry Smith   if (yy != xx) ierr = VecRestoreArrayFast(yy,&y);CHKERRQ(ierr);
949*b1d4fb26SBarry Smith   if (zz != yy) ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
95049b5e25fSSatish Balay 
9516c6c5352SBarry Smith   PetscLogFlops(72*(a->nz*2 - A->m));
95249b5e25fSSatish Balay   PetscFunctionReturn(0);
95349b5e25fSSatish Balay }
95449b5e25fSSatish Balay 
9554a2ae208SSatish Balay #undef __FUNCT__
9564a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqSBAIJ_7"
95749b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_7(Mat A,Vec xx,Vec yy,Vec zz)
95849b5e25fSSatish Balay {
95949b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
96087828ca2SBarry Smith   PetscScalar     *x,*y,*z,*xb,x1,x2,x3,x4,x5,x6,x7;
96149b5e25fSSatish Balay   MatScalar       *v;
962831a3094SHong Zhang   int             mbs=a->mbs,i,*aj=a->j,*ai=a->i,n,ierr,*ib,cval,j,jmin;
96349b5e25fSSatish Balay 
96449b5e25fSSatish Balay   PetscFunctionBegin;
965*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr);
96649b5e25fSSatish Balay   if (yy != xx) {
967*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(yy,&y);CHKERRQ(ierr);
96849b5e25fSSatish Balay   } else {
96949b5e25fSSatish Balay     y = x;
97049b5e25fSSatish Balay   }
97149b5e25fSSatish Balay   if (zz != yy) {
972a9d4b620SHong Zhang     /* ierr = VecCopy(yy,zz);CHKERRQ(ierr); */
973*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr);
974a9d4b620SHong Zhang     ierr = PetscMemcpy(z,y,yy->n*sizeof(PetscScalar));CHKERRQ(ierr);
97549b5e25fSSatish Balay   } else {
97649b5e25fSSatish Balay     z = y;
97749b5e25fSSatish Balay   }
97849b5e25fSSatish Balay 
97949b5e25fSSatish Balay   v     = a->a;
98049b5e25fSSatish Balay   xb = x;
98149b5e25fSSatish Balay 
98249b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
98349b5e25fSSatish Balay     n  = ai[1] - ai[0]; /* length of i_th block row of A */
98449b5e25fSSatish Balay     x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5=xb[4]; x6=xb[5]; x7=xb[6];
98549b5e25fSSatish Balay     ib = aj + *ai;
986831a3094SHong Zhang     jmin = 0;
9877fbae186SHong Zhang     if (*ib == i){     /* (diag of A)*x */
98849b5e25fSSatish 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;
98949b5e25fSSatish 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;
99049b5e25fSSatish 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;
99149b5e25fSSatish 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;
99249b5e25fSSatish 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;
99349b5e25fSSatish 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;
99449b5e25fSSatish 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;
995831a3094SHong Zhang       v += 49; jmin++;
9967fbae186SHong Zhang     }
997831a3094SHong Zhang     for (j=jmin; j<n; j++) {
99849b5e25fSSatish Balay       /* (strict lower triangular part of A)*x  */
99949b5e25fSSatish Balay       cval       = ib[j]*7;
100049b5e25fSSatish Balay       z[cval]   += v[0]*x1  + v[1]*x2 + v[2]*x3 + v[3]*x4+ v[4]*x5 + v[5]*x6+ v[6]*x7;
100149b5e25fSSatish 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;
100249b5e25fSSatish 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;
100349b5e25fSSatish 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;
100449b5e25fSSatish 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;
100549b5e25fSSatish 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;
100649b5e25fSSatish 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;
100749b5e25fSSatish Balay       /* (strict upper triangular part of A)*x  */
100849b5e25fSSatish 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];
100949b5e25fSSatish 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];
101049b5e25fSSatish 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];
101149b5e25fSSatish 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];
101249b5e25fSSatish 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];
101349b5e25fSSatish 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];
101449b5e25fSSatish 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];
101549b5e25fSSatish Balay       v  += 49;
101649b5e25fSSatish Balay     }
101749b5e25fSSatish Balay     xb +=7; ai++;
101849b5e25fSSatish Balay   }
101949b5e25fSSatish Balay 
1020*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
1021*b1d4fb26SBarry Smith   if (yy != xx) ierr = VecRestoreArrayFast(yy,&y);CHKERRQ(ierr);
1022*b1d4fb26SBarry Smith   if (zz != yy) ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
102349b5e25fSSatish Balay 
10246c6c5352SBarry Smith   PetscLogFlops(98*(a->nz*2 - A->m));
102549b5e25fSSatish Balay   PetscFunctionReturn(0);
102649b5e25fSSatish Balay }
102749b5e25fSSatish Balay 
10284a2ae208SSatish Balay #undef __FUNCT__
10294a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqSBAIJ_N"
103049b5e25fSSatish Balay int MatMultAdd_SeqSBAIJ_N(Mat A,Vec xx,Vec yy,Vec zz)
103149b5e25fSSatish Balay {
103249b5e25fSSatish Balay   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
103387828ca2SBarry Smith   PetscScalar     *x,*x_ptr,*y,*z,*z_ptr=0,*xb,*zb,*work,*workt;
1034066653e3SSatish Balay   MatScalar       *v;
1035066653e3SSatish Balay   int             ierr,mbs=a->mbs,i,*idx,*aj,*ii,bs=a->bs,j,n,bs2=a->bs2;
1036066653e3SSatish Balay   int             ncols,k;
103749b5e25fSSatish Balay 
103849b5e25fSSatish Balay   PetscFunctionBegin;
1039*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(xx,&x);CHKERRQ(ierr); x_ptr=x;
104049b5e25fSSatish Balay   if (yy != xx) {
1041*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(yy,&y);CHKERRQ(ierr);
104249b5e25fSSatish Balay   } else {
104349b5e25fSSatish Balay     y = x;
104449b5e25fSSatish Balay   }
104549b5e25fSSatish Balay   if (zz != yy) {
1046a9d4b620SHong Zhang     /* ierr = VecCopy(yy,zz);CHKERRQ(ierr); */
1047*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(zz,&z);CHKERRQ(ierr); z_ptr=z;
1048a9d4b620SHong Zhang     ierr = PetscMemcpy(z,y,yy->n*sizeof(PetscScalar));CHKERRQ(ierr);
104949b5e25fSSatish Balay   } else {
105049b5e25fSSatish Balay     z = y;
105149b5e25fSSatish Balay   }
105249b5e25fSSatish Balay 
105349b5e25fSSatish Balay   aj   = a->j;
105449b5e25fSSatish Balay   v    = a->a;
105549b5e25fSSatish Balay   ii   = a->i;
105649b5e25fSSatish Balay 
105749b5e25fSSatish Balay   if (!a->mult_work) {
105887828ca2SBarry Smith     ierr = PetscMalloc((A->m+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr);
105949b5e25fSSatish Balay   }
106049b5e25fSSatish Balay   work = a->mult_work;
106149b5e25fSSatish Balay 
106249b5e25fSSatish Balay 
106349b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
106449b5e25fSSatish Balay     n     = ii[1] - ii[0]; ncols = n*bs;
106549b5e25fSSatish Balay     workt = work; idx=aj+ii[0];
106649b5e25fSSatish Balay 
106749b5e25fSSatish Balay     /* upper triangular part */
106849b5e25fSSatish Balay     for (j=0; j<n; j++) {
106949b5e25fSSatish Balay       xb = x_ptr + bs*(*idx++);
107049b5e25fSSatish Balay       for (k=0; k<bs; k++) workt[k] = xb[k];
107149b5e25fSSatish Balay       workt += bs;
107249b5e25fSSatish Balay     }
107349b5e25fSSatish Balay     /* z(i*bs:(i+1)*bs-1) += A(i,:)*x */
107449b5e25fSSatish Balay     Kernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z);
107549b5e25fSSatish Balay 
107649b5e25fSSatish Balay     /* strict lower triangular part */
1077831a3094SHong Zhang     idx = aj+ii[0];
1078831a3094SHong Zhang     if (*idx == i){
107996b9376eSHong Zhang       ncols -= bs; v += bs2; idx++; n--;
1080831a3094SHong Zhang     }
108149b5e25fSSatish Balay     if (ncols > 0){
108249b5e25fSSatish Balay       workt = work;
108387828ca2SBarry Smith       ierr  = PetscMemzero(workt,ncols*sizeof(PetscScalar));CHKERRQ(ierr);
1084831a3094SHong Zhang       Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,x,v,workt);
1085831a3094SHong Zhang       for (j=0; j<n; j++) {
1086831a3094SHong Zhang         zb = z_ptr + bs*(*idx++);
1087831a3094SHong Zhang         /* idx++; */
108849b5e25fSSatish Balay         for (k=0; k<bs; k++) zb[k] += workt[k] ;
108949b5e25fSSatish Balay         workt += bs;
109049b5e25fSSatish Balay       }
109149b5e25fSSatish Balay     }
109249b5e25fSSatish Balay 
109349b5e25fSSatish Balay     x += bs; v += n*bs2; z += bs; ii++;
109449b5e25fSSatish Balay   }
109549b5e25fSSatish Balay 
1096*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(xx,&x);CHKERRQ(ierr);
1097*b1d4fb26SBarry Smith   if (yy != xx) ierr = VecRestoreArrayFast(yy,&y);CHKERRQ(ierr);
1098*b1d4fb26SBarry Smith   if (zz != yy) ierr = VecRestoreArrayFast(zz,&z);CHKERRQ(ierr);
109949b5e25fSSatish Balay 
11006c6c5352SBarry Smith   PetscLogFlops(2*(a->nz*2 - A->m));
110149b5e25fSSatish Balay   PetscFunctionReturn(0);
110249b5e25fSSatish Balay }
110349b5e25fSSatish Balay 
11044a2ae208SSatish Balay #undef __FUNCT__
11054a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqSBAIJ"
110649b5e25fSSatish Balay int MatMultTranspose_SeqSBAIJ(Mat A,Vec xx,Vec zz)
110749b5e25fSSatish Balay {
1108eeeff2ecSHong Zhang   int ierr;
1109eeeff2ecSHong Zhang 
111049b5e25fSSatish Balay   PetscFunctionBegin;
1111eeeff2ecSHong Zhang   ierr = MatMult(A,xx,zz);CHKERRQ(ierr);
1112eeeff2ecSHong Zhang   PetscFunctionReturn(0);
111349b5e25fSSatish Balay }
111449b5e25fSSatish Balay 
11154a2ae208SSatish Balay #undef __FUNCT__
11164a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqSBAIJ"
111749b5e25fSSatish Balay int MatMultTransposeAdd_SeqSBAIJ(Mat A,Vec xx,Vec yy,Vec zz)
111849b5e25fSSatish Balay 
111949b5e25fSSatish Balay {
1120eeeff2ecSHong Zhang   int ierr;
1121eeeff2ecSHong Zhang 
112249b5e25fSSatish Balay   PetscFunctionBegin;
1123eeeff2ecSHong Zhang   ierr = MatMultAdd(A,xx,yy,zz);CHKERRQ(ierr);
1124eeeff2ecSHong Zhang   PetscFunctionReturn(0);
112549b5e25fSSatish Balay }
112649b5e25fSSatish Balay 
11274a2ae208SSatish Balay #undef __FUNCT__
11284a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqSBAIJ"
1129268466fbSBarry Smith int MatScale_SeqSBAIJ(const PetscScalar *alpha,Mat inA)
113049b5e25fSSatish Balay {
113149b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data;
11326c6c5352SBarry Smith   int         one = 1,totalnz = a->bs2*a->nz;
113349b5e25fSSatish Balay 
113449b5e25fSSatish Balay   PetscFunctionBegin;
1135268466fbSBarry Smith   BLscal_(&totalnz,(PetscScalar*)alpha,a->a,&one);
1136b0a32e0cSBarry Smith   PetscLogFlops(totalnz);
113749b5e25fSSatish Balay   PetscFunctionReturn(0);
113849b5e25fSSatish Balay }
113949b5e25fSSatish Balay 
11404a2ae208SSatish Balay #undef __FUNCT__
11414a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqSBAIJ"
114249b5e25fSSatish Balay int MatNorm_SeqSBAIJ(Mat A,NormType type,PetscReal *norm)
114349b5e25fSSatish Balay {
114449b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
114549b5e25fSSatish Balay   MatScalar   *v = a->a;
114649b5e25fSSatish Balay   PetscReal   sum_diag = 0.0, sum_off = 0.0, *sum;
1147831a3094SHong Zhang   int         i,j,k,bs = a->bs,bs2=a->bs2,k1,mbs=a->mbs,*aj=a->j;
1148831a3094SHong Zhang   int         *jl,*il,jmin,jmax,ierr,nexti,ik,*col;
114949b5e25fSSatish Balay 
115049b5e25fSSatish Balay   PetscFunctionBegin;
115149b5e25fSSatish Balay   if (type == NORM_FROBENIUS) {
115249b5e25fSSatish Balay     for (k=0; k<mbs; k++){
115349b5e25fSSatish Balay       jmin = a->i[k]; jmax = a->i[k+1];
1154831a3094SHong Zhang       col  = aj + jmin;
1155831a3094SHong Zhang       if (*col == k){         /* diagonal block */
115649b5e25fSSatish Balay         for (i=0; i<bs2; i++){
115749b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
115849b5e25fSSatish Balay           sum_diag += PetscRealPart(PetscConj(*v)*(*v)); v++;
115949b5e25fSSatish Balay #else
116049b5e25fSSatish Balay           sum_diag += (*v)*(*v); v++;
116149b5e25fSSatish Balay #endif
116249b5e25fSSatish Balay         }
1163831a3094SHong Zhang         jmin++;
1164831a3094SHong Zhang       }
1165831a3094SHong Zhang       for (j=jmin; j<jmax; j++){  /* off-diagonal blocks */
116649b5e25fSSatish Balay         for (i=0; i<bs2; i++){
116749b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
116849b5e25fSSatish Balay           sum_off += PetscRealPart(PetscConj(*v)*(*v)); v++;
116949b5e25fSSatish Balay #else
117049b5e25fSSatish Balay           sum_off += (*v)*(*v); v++;
117149b5e25fSSatish Balay #endif
117249b5e25fSSatish Balay         }
117349b5e25fSSatish Balay       }
117449b5e25fSSatish Balay     }
117549b5e25fSSatish Balay     *norm = sqrt(sum_diag + 2*sum_off);
117649b5e25fSSatish Balay 
117749b5e25fSSatish Balay   }  else if (type == NORM_INFINITY) { /* maximum row sum */
117882502324SSatish Balay     ierr = PetscMalloc(mbs*sizeof(int),&il);CHKERRQ(ierr);
117982502324SSatish Balay     ierr = PetscMalloc(mbs*sizeof(int),&jl);CHKERRQ(ierr);
118082502324SSatish Balay     ierr = PetscMalloc(bs*sizeof(PetscReal),&sum);CHKERRQ(ierr);
118149b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
118249b5e25fSSatish Balay       jl[i] = mbs; il[0] = 0;
118349b5e25fSSatish Balay     }
118449b5e25fSSatish Balay 
118549b5e25fSSatish Balay     *norm = 0.0;
118649b5e25fSSatish Balay     for (k=0; k<mbs; k++) { /* k_th block row */
118749b5e25fSSatish Balay       for (j=0; j<bs; j++) sum[j]=0.0;
118849b5e25fSSatish Balay 
118949b5e25fSSatish Balay       /*-- col sum --*/
119049b5e25fSSatish Balay       i = jl[k]; /* first |A(i,k)| to be added */
1191831a3094SHong Zhang       /* jl[k]=i: first nozero element in row i for submatrix A(1:k,k:n) (active window)
1192831a3094SHong Zhang                   at step k */
119349b5e25fSSatish Balay       while (i<mbs){
119449b5e25fSSatish Balay         nexti = jl[i];  /* next block row to be added */
119549b5e25fSSatish Balay         ik    = il[i];  /* block index of A(i,k) in the array a */
119649b5e25fSSatish Balay         for (j=0; j<bs; j++){
119749b5e25fSSatish Balay           v = a->a + ik*bs2 + j*bs;
119849b5e25fSSatish Balay           for (k1=0; k1<bs; k1++) {
119949b5e25fSSatish Balay             sum[j] += PetscAbsScalar(*v); v++;
120049b5e25fSSatish Balay           }
120149b5e25fSSatish Balay         }
120249b5e25fSSatish Balay         /* update il, jl */
1203831a3094SHong Zhang         jmin = ik + 1; /* block index of array a: points to the next nonzero of A in row i */
1204831a3094SHong Zhang         jmax = a->i[i+1];
120549b5e25fSSatish Balay         if (jmin < jmax){
120649b5e25fSSatish Balay           il[i] = jmin;
120749b5e25fSSatish Balay           j   = a->j[jmin];
120849b5e25fSSatish Balay           jl[i] = jl[j]; jl[j]=i;
120949b5e25fSSatish Balay         }
121049b5e25fSSatish Balay         i = nexti;
121149b5e25fSSatish Balay       }
121249b5e25fSSatish Balay 
121349b5e25fSSatish Balay       /*-- row sum --*/
121449b5e25fSSatish Balay       jmin = a->i[k]; jmax = a->i[k+1];
121549b5e25fSSatish Balay       for (i=jmin; i<jmax; i++) {
121649b5e25fSSatish Balay         for (j=0; j<bs; j++){
121749b5e25fSSatish Balay           v = a->a + i*bs2 + j;
121849b5e25fSSatish Balay           for (k1=0; k1<bs; k1++){
121949b5e25fSSatish Balay             sum[j] += PetscAbsScalar(*v);
122049b5e25fSSatish Balay             v   += bs;
122149b5e25fSSatish Balay           }
122249b5e25fSSatish Balay         }
122349b5e25fSSatish Balay       }
122449b5e25fSSatish Balay       /* add k_th block row to il, jl */
1225831a3094SHong Zhang       col = aj+jmin;
1226831a3094SHong Zhang       if (*col == k) jmin++;
122749b5e25fSSatish Balay       if (jmin < jmax){
122849b5e25fSSatish Balay         il[k] = jmin;
122949b5e25fSSatish Balay         j   = a->j[jmin];
123049b5e25fSSatish Balay         jl[k] = jl[j]; jl[j] = k;
123149b5e25fSSatish Balay       }
123249b5e25fSSatish Balay       for (j=0; j<bs; j++){
123349b5e25fSSatish Balay         if (sum[j] > *norm) *norm = sum[j];
123449b5e25fSSatish Balay       }
123549b5e25fSSatish Balay     }
123649b5e25fSSatish Balay     ierr = PetscFree(il);CHKERRQ(ierr);
123749b5e25fSSatish Balay     ierr = PetscFree(jl);CHKERRQ(ierr);
123849b5e25fSSatish Balay     ierr = PetscFree(sum);CHKERRQ(ierr);
123949b5e25fSSatish Balay   } else {
1240347d480fSBarry Smith     SETERRQ(PETSC_ERR_SUP,"No support for this norm yet");
124149b5e25fSSatish Balay   }
124249b5e25fSSatish Balay   PetscFunctionReturn(0);
124349b5e25fSSatish Balay }
124449b5e25fSSatish Balay 
12454a2ae208SSatish Balay #undef __FUNCT__
12464a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqSBAIJ"
124749b5e25fSSatish Balay int MatEqual_SeqSBAIJ(Mat A,Mat B,PetscTruth* flg)
124849b5e25fSSatish Balay {
124949b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ *)A->data,*b = (Mat_SeqSBAIJ *)B->data;
125049b5e25fSSatish Balay   int          ierr;
125149b5e25fSSatish Balay 
125249b5e25fSSatish Balay   PetscFunctionBegin;
125349b5e25fSSatish Balay 
125449b5e25fSSatish Balay   /* If the  matrix/block dimensions are not equal, or no of nonzeros or shift */
12556c6c5352SBarry Smith   if ((A->m != B->m) || (A->n != B->n) || (a->bs != b->bs)|| (a->nz != b->nz)) {
1256ef511fbeSHong Zhang     *flg = PETSC_FALSE;
1257ef511fbeSHong Zhang     PetscFunctionReturn(0);
125849b5e25fSSatish Balay   }
125949b5e25fSSatish Balay 
126049b5e25fSSatish Balay   /* if the a->i are the same */
126149b5e25fSSatish Balay   ierr = PetscMemcmp(a->i,b->i,(a->mbs+1)*sizeof(int),flg);CHKERRQ(ierr);
126249b5e25fSSatish Balay   if (*flg == PETSC_FALSE) {
126349b5e25fSSatish Balay     PetscFunctionReturn(0);
126449b5e25fSSatish Balay   }
126549b5e25fSSatish Balay 
126649b5e25fSSatish Balay   /* if a->j are the same */
12676c6c5352SBarry Smith   ierr = PetscMemcmp(a->j,b->j,(a->nz)*sizeof(int),flg);CHKERRQ(ierr);
126849b5e25fSSatish Balay   if (*flg == PETSC_FALSE) {
126949b5e25fSSatish Balay     PetscFunctionReturn(0);
127049b5e25fSSatish Balay   }
127149b5e25fSSatish Balay   /* if a->a are the same */
12726c6c5352SBarry Smith   ierr = PetscMemcmp(a->a,b->a,(a->nz)*(a->bs)*(a->bs)*sizeof(PetscScalar),flg);CHKERRQ(ierr);
127349b5e25fSSatish Balay 
1274935af2e7SHong Zhang   PetscFunctionReturn(0);
127549b5e25fSSatish Balay }
127649b5e25fSSatish Balay 
12774a2ae208SSatish Balay #undef __FUNCT__
12784a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqSBAIJ"
127949b5e25fSSatish Balay int MatGetDiagonal_SeqSBAIJ(Mat A,Vec v)
128049b5e25fSSatish Balay {
128149b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
128249b5e25fSSatish Balay   int          ierr,i,j,k,n,row,bs,*ai,*aj,ambs,bs2;
128387828ca2SBarry Smith   PetscScalar  *x,zero = 0.0;
128449b5e25fSSatish Balay   MatScalar    *aa,*aa_j;
128549b5e25fSSatish Balay 
128649b5e25fSSatish Balay   PetscFunctionBegin;
128749b5e25fSSatish Balay   bs   = a->bs;
128882799104SHong Zhang   if (A->factor && bs>1) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix with bs>1");
128982799104SHong Zhang 
129049b5e25fSSatish Balay   aa   = a->a;
129149b5e25fSSatish Balay   ai   = a->i;
129249b5e25fSSatish Balay   aj   = a->j;
129349b5e25fSSatish Balay   ambs = a->mbs;
129449b5e25fSSatish Balay   bs2  = a->bs2;
129549b5e25fSSatish Balay 
129649b5e25fSSatish Balay   ierr = VecSet(&zero,v);CHKERRQ(ierr);
1297*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(v,&x);CHKERRQ(ierr);
129849b5e25fSSatish Balay   ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr);
1299ef511fbeSHong Zhang   if (n != A->m) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
130049b5e25fSSatish Balay   for (i=0; i<ambs; i++) {
130149b5e25fSSatish Balay     j=ai[i];
130249b5e25fSSatish Balay     if (aj[j] == i) {             /* if this is a diagonal element */
130349b5e25fSSatish Balay       row  = i*bs;
130449b5e25fSSatish Balay       aa_j = aa + j*bs2;
130582799104SHong Zhang       if (A->factor && bs==1){
130682799104SHong Zhang         for (k=0; k<bs2; k+=(bs+1),row++) x[row] = 1.0/aa_j[k];
130782799104SHong Zhang       } else {
130849b5e25fSSatish Balay         for (k=0; k<bs2; k+=(bs+1),row++) x[row] = aa_j[k];
130949b5e25fSSatish Balay       }
131049b5e25fSSatish Balay     }
131182799104SHong Zhang   }
131282799104SHong Zhang 
1313*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(v,&x);CHKERRQ(ierr);
131449b5e25fSSatish Balay   PetscFunctionReturn(0);
131549b5e25fSSatish Balay }
131649b5e25fSSatish Balay 
13174a2ae208SSatish Balay #undef __FUNCT__
13184a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqSBAIJ"
131949b5e25fSSatish Balay int MatDiagonalScale_SeqSBAIJ(Mat A,Vec ll,Vec rr)
132049b5e25fSSatish Balay {
132149b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
132287828ca2SBarry Smith   PetscScalar  *l,*r,x,*li,*ri;
132349b5e25fSSatish Balay   MatScalar    *aa,*v;
1324066653e3SSatish Balay   int          ierr,i,j,k,lm,rn,M,m,*ai,*aj,mbs,tmp,bs,bs2;
132549b5e25fSSatish Balay 
132649b5e25fSSatish Balay   PetscFunctionBegin;
132749b5e25fSSatish Balay   ai  = a->i;
132849b5e25fSSatish Balay   aj  = a->j;
132949b5e25fSSatish Balay   aa  = a->a;
1330ef511fbeSHong Zhang   m   = A->m;
133149b5e25fSSatish Balay   bs  = a->bs;
133249b5e25fSSatish Balay   mbs = a->mbs;
133349b5e25fSSatish Balay   bs2 = a->bs2;
133449b5e25fSSatish Balay 
133549b5e25fSSatish Balay   if (ll != rr) {
1336347d480fSBarry Smith     SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"For symmetric format, left and right scaling vectors must be same\n");
133749b5e25fSSatish Balay   }
133849b5e25fSSatish Balay   if (ll) {
1339*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(ll,&l);CHKERRQ(ierr);
134049b5e25fSSatish Balay     ierr = VecGetLocalSize(ll,&lm);CHKERRQ(ierr);
1341347d480fSBarry Smith     if (lm != m) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length");
134249b5e25fSSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
134349b5e25fSSatish Balay       M  = ai[i+1] - ai[i];
134449b5e25fSSatish Balay       li = l + i*bs;
134549b5e25fSSatish Balay       v  = aa + bs2*ai[i];
134649b5e25fSSatish Balay       for (j=0; j<M; j++) { /* for each block */
134749b5e25fSSatish Balay         for (k=0; k<bs2; k++) {
134849b5e25fSSatish Balay           (*v++) *= li[k%bs];
134949b5e25fSSatish Balay         }
135049b5e25fSSatish Balay #ifdef CONT
135149b5e25fSSatish Balay         /* will be used to replace the above loop */
135249b5e25fSSatish Balay         ri = l + bs*aj[ai[i]+j];
135349b5e25fSSatish Balay         for (k=0; k<bs; k++) { /* column value */
135449b5e25fSSatish Balay           x = ri[k];
135549b5e25fSSatish Balay           for (tmp=0; tmp<bs; tmp++) (*v++) *= li[tmp]*x;
135649b5e25fSSatish Balay         }
135749b5e25fSSatish Balay #endif
135849b5e25fSSatish Balay 
135949b5e25fSSatish Balay       }
136049b5e25fSSatish Balay     }
1361*b1d4fb26SBarry Smith     ierr = VecRestoreArrayFast(ll,&l);CHKERRQ(ierr);
13626c6c5352SBarry Smith     PetscLogFlops(2*a->nz);
136349b5e25fSSatish Balay   }
136449b5e25fSSatish Balay   /* will be deleted */
136549b5e25fSSatish Balay   if (rr) {
1366*b1d4fb26SBarry Smith     ierr = VecGetArrayFast(rr,&r);CHKERRQ(ierr);
136749b5e25fSSatish Balay     ierr = VecGetLocalSize(rr,&rn);CHKERRQ(ierr);
1368347d480fSBarry Smith     if (rn != m) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length");
136949b5e25fSSatish Balay     for (i=0; i<mbs; i++) { /* for each block row */
137049b5e25fSSatish Balay       M  = ai[i+1] - ai[i];
137149b5e25fSSatish Balay       v  = aa + bs2*ai[i];
137249b5e25fSSatish Balay       for (j=0; j<M; j++) { /* for each block */
137349b5e25fSSatish Balay         ri = r + bs*aj[ai[i]+j];
137449b5e25fSSatish Balay         for (k=0; k<bs; k++) {
137549b5e25fSSatish Balay           x = ri[k];
137649b5e25fSSatish Balay           for (tmp=0; tmp<bs; tmp++) (*v++) *= x;
137749b5e25fSSatish Balay         }
137849b5e25fSSatish Balay       }
137949b5e25fSSatish Balay     }
1380*b1d4fb26SBarry Smith     ierr = VecRestoreArrayFast(rr,&r);CHKERRQ(ierr);
13816c6c5352SBarry Smith     PetscLogFlops(a->nz);
138249b5e25fSSatish Balay   }
138349b5e25fSSatish Balay   PetscFunctionReturn(0);
138449b5e25fSSatish Balay }
138549b5e25fSSatish Balay 
13864a2ae208SSatish Balay #undef __FUNCT__
13874a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqSBAIJ"
138849b5e25fSSatish Balay int MatGetInfo_SeqSBAIJ(Mat A,MatInfoType flag,MatInfo *info)
138949b5e25fSSatish Balay {
139049b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
139149b5e25fSSatish Balay 
139249b5e25fSSatish Balay   PetscFunctionBegin;
1393ef511fbeSHong Zhang   info->rows_global    = (double)A->m;
1394ef511fbeSHong Zhang   info->columns_global = (double)A->m;
1395ef511fbeSHong Zhang   info->rows_local     = (double)A->m;
1396ef511fbeSHong Zhang   info->columns_local  = (double)A->m;
139749b5e25fSSatish Balay   info->block_size     = a->bs2;
13986c6c5352SBarry Smith   info->nz_allocated   = a->maxnz; /*num. of nonzeros in upper triangular part */
13996c6c5352SBarry Smith   info->nz_used        = a->bs2*a->nz; /*num. of nonzeros in upper triangular part */
140049b5e25fSSatish Balay   info->nz_unneeded    = (double)(info->nz_allocated - info->nz_used);
140149b5e25fSSatish Balay   info->assemblies   = A->num_ass;
140249b5e25fSSatish Balay   info->mallocs      = a->reallocs;
140349b5e25fSSatish Balay   info->memory       = A->mem;
140449b5e25fSSatish Balay   if (A->factor) {
140549b5e25fSSatish Balay     info->fill_ratio_given  = A->info.fill_ratio_given;
140649b5e25fSSatish Balay     info->fill_ratio_needed = A->info.fill_ratio_needed;
140749b5e25fSSatish Balay     info->factor_mallocs    = A->info.factor_mallocs;
140849b5e25fSSatish Balay   } else {
140949b5e25fSSatish Balay     info->fill_ratio_given  = 0;
141049b5e25fSSatish Balay     info->fill_ratio_needed = 0;
141149b5e25fSSatish Balay     info->factor_mallocs    = 0;
141249b5e25fSSatish Balay   }
141349b5e25fSSatish Balay   PetscFunctionReturn(0);
141449b5e25fSSatish Balay }
141549b5e25fSSatish Balay 
141649b5e25fSSatish Balay 
14174a2ae208SSatish Balay #undef __FUNCT__
14184a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqSBAIJ"
141949b5e25fSSatish Balay int MatZeroEntries_SeqSBAIJ(Mat A)
142049b5e25fSSatish Balay {
142149b5e25fSSatish Balay   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
142249b5e25fSSatish Balay   int         ierr;
142349b5e25fSSatish Balay 
142449b5e25fSSatish Balay   PetscFunctionBegin;
142549b5e25fSSatish Balay   ierr = PetscMemzero(a->a,a->bs2*a->i[a->mbs]*sizeof(MatScalar));CHKERRQ(ierr);
142649b5e25fSSatish Balay   PetscFunctionReturn(0);
142749b5e25fSSatish Balay }
1428dc354874SHong Zhang 
14294a2ae208SSatish Balay #undef __FUNCT__
14304a2ae208SSatish Balay #define __FUNCT__ "MatGetRowMax_SeqSBAIJ"
1431dc354874SHong Zhang int MatGetRowMax_SeqSBAIJ(Mat A,Vec v)
1432dc354874SHong Zhang {
1433dc354874SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
1434d0f6400bSHong Zhang   int          ierr,i,j,n,row,col,bs,*ai,*aj,mbs;
1435c3fca9a7SHong Zhang   PetscReal    atmp;
1436273d9f13SBarry Smith   MatScalar    *aa;
143787828ca2SBarry Smith   PetscScalar  zero = 0.0,*x;
1438d0f6400bSHong Zhang   int          ncols,brow,bcol,krow,kcol;
1439dc354874SHong Zhang 
1440dc354874SHong Zhang   PetscFunctionBegin;
1441dc354874SHong Zhang   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1442dc354874SHong Zhang   bs   = a->bs;
1443dc354874SHong Zhang   aa   = a->a;
1444dc354874SHong Zhang   ai   = a->i;
1445dc354874SHong Zhang   aj   = a->j;
144644117c81SHong Zhang   mbs = a->mbs;
1447dc354874SHong Zhang 
1448dc354874SHong Zhang   ierr = VecSet(&zero,v);CHKERRQ(ierr);
1449*b1d4fb26SBarry Smith   ierr = VecGetArrayFast(v,&x);CHKERRQ(ierr);
1450dc354874SHong Zhang   ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr);
1451ef511fbeSHong Zhang   if (n != A->m) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
145244117c81SHong Zhang   for (i=0; i<mbs; i++) {
1453d0f6400bSHong Zhang     ncols = ai[1] - ai[0]; ai++;
1454d0f6400bSHong Zhang     brow  = bs*i;
145544117c81SHong Zhang     for (j=0; j<ncols; j++){
1456d0f6400bSHong Zhang       bcol = bs*(*aj);
145744117c81SHong Zhang       for (kcol=0; kcol<bs; kcol++){
1458d0f6400bSHong Zhang         col = bcol + kcol;      /* col index */
145944117c81SHong Zhang         for (krow=0; krow<bs; krow++){
1460d0f6400bSHong Zhang           atmp = PetscAbsScalar(*aa); aa++;
1461d0f6400bSHong Zhang           row = brow + krow;    /* row index */
146244117c81SHong Zhang           /* printf("val[%d,%d]: %g\n",row,col,atmp); */
1463c3fca9a7SHong Zhang           if (PetscRealPart(x[row]) < atmp) x[row] = atmp;
1464c3fca9a7SHong Zhang           if (*aj > i && PetscRealPart(x[col]) < atmp) x[col] = atmp;
146544117c81SHong Zhang         }
146644117c81SHong Zhang       }
1467d0f6400bSHong Zhang       aj++;
1468dc354874SHong Zhang     }
1469dc354874SHong Zhang   }
1470*b1d4fb26SBarry Smith   ierr = VecRestoreArrayFast(v,&x);CHKERRQ(ierr);
1471dc354874SHong Zhang   PetscFunctionReturn(0);
1472dc354874SHong Zhang }
1473