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