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