1be1d678aSKris Buschelman #define PETSCMAT_DLL 2cac129eeSSatish Balay 37c4f633dSBarry Smith #include "../src/mat/impls/baij/seq/baij.h" 4c60f0209SBarry Smith #include "../src/mat/blockinvert.h" 50a835dfdSSatish Balay #include "petscbt.h" 6cac129eeSSatish Balay 74a2ae208SSatish Balay #undef __FUNCT__ 84a2ae208SSatish Balay #define __FUNCT__ "MatIncreaseOverlap_SeqBAIJ" 9690b6cddSBarry Smith PetscErrorCode MatIncreaseOverlap_SeqBAIJ(Mat A,PetscInt is_max,IS is[],PetscInt ov) 10a3192f15SSatish Balay { 11a3192f15SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 126849ba73SBarry Smith PetscErrorCode ierr; 135d0c19d7SBarry Smith PetscInt row,i,j,k,l,m,n,*nidx,isz,val,ival; 145d0c19d7SBarry Smith const PetscInt *idx; 15690b6cddSBarry Smith PetscInt start,end,*ai,*aj,bs,*nidx2; 16f1af5d2fSBarry Smith PetscBT table; 17a3192f15SSatish Balay 183a40ed3dSBarry Smith PetscFunctionBegin; 19a3192f15SSatish Balay m = a->mbs; 20a3192f15SSatish Balay ai = a->i; 21a3192f15SSatish Balay aj = a->j; 22d0f46423SBarry Smith bs = A->rmap->bs; 23a3192f15SSatish Balay 2429bbc08cSBarry Smith if (ov < 0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative overlap specified"); 25a3192f15SSatish Balay 266831982aSBarry Smith ierr = PetscBTCreate(m,table);CHKERRQ(ierr); 27690b6cddSBarry Smith ierr = PetscMalloc((m+1)*sizeof(PetscInt),&nidx);CHKERRQ(ierr); 28d0f46423SBarry Smith ierr = PetscMalloc((A->rmap->N+1)*sizeof(PetscInt),&nidx2);CHKERRQ(ierr); 29a3192f15SSatish Balay 30a3192f15SSatish Balay for (i=0; i<is_max; i++) { 31a3192f15SSatish Balay /* Initialise the two local arrays */ 32a3192f15SSatish Balay isz = 0; 336831982aSBarry Smith ierr = PetscBTMemzero(m,table);CHKERRQ(ierr); 34a3192f15SSatish Balay 35a3192f15SSatish Balay /* Extract the indices, assume there can be duplicate entries */ 36a3192f15SSatish Balay ierr = ISGetIndices(is[i],&idx);CHKERRQ(ierr); 37b9b97703SBarry Smith ierr = ISGetLocalSize(is[i],&n);CHKERRQ(ierr); 38a3192f15SSatish Balay 39a3192f15SSatish Balay /* Enter these into the temp arrays i.e mark table[row], enter row into new index */ 40a3192f15SSatish Balay for (j=0; j<n ; ++j){ 41218c64b6SSatish Balay ival = idx[j]/bs; /* convert the indices into block indices */ 425eee224dSHong Zhang if (ival>=m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"index greater than mat-dim"); 43f1af5d2fSBarry Smith if(!PetscBTLookupSet(table,ival)) { nidx[isz++] = ival;} 44a3192f15SSatish Balay } 45a3192f15SSatish Balay ierr = ISRestoreIndices(is[i],&idx);CHKERRQ(ierr); 46a3192f15SSatish Balay ierr = ISDestroy(is[i]);CHKERRQ(ierr); 47a3192f15SSatish Balay 48a3192f15SSatish Balay k = 0; 49a3192f15SSatish Balay for (j=0; j<ov; j++){ /* for each overlap*/ 50a3192f15SSatish Balay n = isz; 51a3192f15SSatish Balay for (; k<n ; k++){ /* do only those rows in nidx[k], which are not done yet */ 52a3192f15SSatish Balay row = nidx[k]; 53a3192f15SSatish Balay start = ai[row]; 54a3192f15SSatish Balay end = ai[row+1]; 55a3192f15SSatish Balay for (l = start; l<end ; l++){ 56a3192f15SSatish Balay val = aj[l]; 57f1af5d2fSBarry Smith if (!PetscBTLookupSet(table,val)) {nidx[isz++] = val;} 58a3192f15SSatish Balay } 59a3192f15SSatish Balay } 60a3192f15SSatish Balay } 61218c64b6SSatish Balay /* expand the Index Set */ 62218c64b6SSatish Balay for (j=0; j<isz; j++) { 63218c64b6SSatish Balay for (k=0; k<bs; k++) 64218c64b6SSatish Balay nidx2[j*bs+k] = nidx[j]*bs+k; 65218c64b6SSatish Balay } 66329f5518SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,isz*bs,nidx2,is+i);CHKERRQ(ierr); 67a3192f15SSatish Balay } 686831982aSBarry Smith ierr = PetscBTDestroy(table);CHKERRQ(ierr); 69606d414cSSatish Balay ierr = PetscFree(nidx);CHKERRQ(ierr); 70606d414cSSatish Balay ierr = PetscFree(nidx2);CHKERRQ(ierr); 713a40ed3dSBarry Smith PetscFunctionReturn(0); 72a3192f15SSatish Balay } 731c351548SSatish Balay 744a2ae208SSatish Balay #undef __FUNCT__ 754a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqBAIJ_Private" 764aa3045dSJed Brown PetscErrorCode MatGetSubMatrix_SeqBAIJ_Private(Mat A,IS isrow,IS iscol,MatReuse scall,Mat *B) 77736121d4SSatish Balay { 78736121d4SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data,*c; 796849ba73SBarry Smith PetscErrorCode ierr; 80690b6cddSBarry Smith PetscInt *smap,i,k,kstart,kend,oldcols = a->nbs,*lens; 81690b6cddSBarry Smith PetscInt row,mat_i,*mat_j,tcol,*mat_ilen; 825d0c19d7SBarry Smith const PetscInt *irow,*icol; 835d0c19d7SBarry Smith PetscInt nrows,ncols,*ssmap,bs=A->rmap->bs,bs2=a->bs2; 84690b6cddSBarry Smith PetscInt *aj = a->j,*ai = a->i; 853f1db9ecSBarry Smith MatScalar *mat_a; 86736121d4SSatish Balay Mat C; 8714ca34e6SBarry Smith PetscTruth flag,sorted; 88736121d4SSatish Balay 893a40ed3dSBarry Smith PetscFunctionBegin; 9014ca34e6SBarry Smith ierr = ISSorted(iscol,&sorted);CHKERRQ(ierr); 9114ca34e6SBarry Smith if (!sorted) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"IS is not sorted"); 92736121d4SSatish Balay 93736121d4SSatish Balay ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr); 94218c64b6SSatish Balay ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr); 95b9b97703SBarry Smith ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr); 96b9b97703SBarry Smith ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr); 97736121d4SSatish Balay 98690b6cddSBarry Smith ierr = PetscMalloc((1+oldcols)*sizeof(PetscInt),&smap);CHKERRQ(ierr); 99736121d4SSatish Balay ssmap = smap; 100690b6cddSBarry Smith ierr = PetscMalloc((1+nrows)*sizeof(PetscInt),&lens);CHKERRQ(ierr); 101690b6cddSBarry Smith ierr = PetscMemzero(smap,oldcols*sizeof(PetscInt));CHKERRQ(ierr); 102736121d4SSatish Balay for (i=0; i<ncols; i++) smap[icol[i]] = i+1; 103736121d4SSatish Balay /* determine lens of each row */ 104736121d4SSatish Balay for (i=0; i<nrows; i++) { 105736121d4SSatish Balay kstart = ai[irow[i]]; 106736121d4SSatish Balay kend = kstart + a->ilen[irow[i]]; 107736121d4SSatish Balay lens[i] = 0; 108736121d4SSatish Balay for (k=kstart; k<kend; k++) { 109736121d4SSatish Balay if (ssmap[aj[k]]) { 110736121d4SSatish Balay lens[i]++; 111736121d4SSatish Balay } 112736121d4SSatish Balay } 113736121d4SSatish Balay } 114736121d4SSatish Balay /* Create and fill new matrix */ 115736121d4SSatish Balay if (scall == MAT_REUSE_MATRIX) { 116736121d4SSatish Balay c = (Mat_SeqBAIJ *)((*B)->data); 117736121d4SSatish Balay 118d0f46423SBarry Smith if (c->mbs!=nrows || c->nbs!=ncols || (*B)->rmap->bs!=bs) SETERRQ(PETSC_ERR_ARG_SIZ,"Submatrix wrong size"); 119690b6cddSBarry Smith ierr = PetscMemcmp(c->ilen,lens,c->mbs *sizeof(PetscInt),&flag);CHKERRQ(ierr); 120abc0a331SBarry Smith if (!flag) { 12129bbc08cSBarry Smith SETERRQ(PETSC_ERR_ARG_SIZ,"Cannot reuse matrix. wrong no of nonzeros"); 122736121d4SSatish Balay } 123690b6cddSBarry Smith ierr = PetscMemzero(c->ilen,c->mbs*sizeof(PetscInt));CHKERRQ(ierr); 124736121d4SSatish Balay C = *B; 1253a40ed3dSBarry Smith } else { 1267adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&C);CHKERRQ(ierr); 127f69a0ea3SMatthew Knepley ierr = MatSetSizes(C,nrows*bs,ncols*bs,PETSC_DETERMINE,PETSC_DETERMINE);CHKERRQ(ierr); 1287adad957SLisandro Dalcin ierr = MatSetType(C,((PetscObject)A)->type_name);CHKERRQ(ierr); 129ab93d7beSBarry Smith ierr = MatSeqBAIJSetPreallocation_SeqBAIJ(C,bs,0,lens);CHKERRQ(ierr); 130736121d4SSatish Balay } 131736121d4SSatish Balay c = (Mat_SeqBAIJ *)(C->data); 132736121d4SSatish Balay for (i=0; i<nrows; i++) { 133736121d4SSatish Balay row = irow[i]; 134736121d4SSatish Balay kstart = ai[row]; 135736121d4SSatish Balay kend = kstart + a->ilen[row]; 136736121d4SSatish Balay mat_i = c->i[i]; 137736121d4SSatish Balay mat_j = c->j + mat_i; 138218c64b6SSatish Balay mat_a = c->a + mat_i*bs2; 139736121d4SSatish Balay mat_ilen = c->ilen + i; 140736121d4SSatish Balay for (k=kstart; k<kend; k++) { 141736121d4SSatish Balay if ((tcol=ssmap[a->j[k]])) { 142736121d4SSatish Balay *mat_j++ = tcol - 1; 143549d3d68SSatish Balay ierr = PetscMemcpy(mat_a,a->a+k*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 144549d3d68SSatish Balay mat_a += bs2; 145736121d4SSatish Balay (*mat_ilen)++; 146736121d4SSatish Balay } 147736121d4SSatish Balay } 148736121d4SSatish Balay } 149218c64b6SSatish Balay 150736121d4SSatish Balay /* Free work space */ 151736121d4SSatish Balay ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 152606d414cSSatish Balay ierr = PetscFree(smap);CHKERRQ(ierr); 153606d414cSSatish Balay ierr = PetscFree(lens);CHKERRQ(ierr); 1546d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1556d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 156736121d4SSatish Balay 157736121d4SSatish Balay ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 158736121d4SSatish Balay *B = C; 1593a40ed3dSBarry Smith PetscFunctionReturn(0); 160736121d4SSatish Balay } 161736121d4SSatish Balay 1624a2ae208SSatish Balay #undef __FUNCT__ 1634a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqBAIJ" 1644aa3045dSJed Brown PetscErrorCode MatGetSubMatrix_SeqBAIJ(Mat A,IS isrow,IS iscol,MatReuse scall,Mat *B) 165218c64b6SSatish Balay { 166218c64b6SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 167218c64b6SSatish Balay IS is1,is2; 1686849ba73SBarry Smith PetscErrorCode ierr; 1695d0c19d7SBarry Smith PetscInt *vary,*iary,nrows,ncols,i,bs=A->rmap->bs,count; 1705d0c19d7SBarry Smith const PetscInt *irow,*icol; 171218c64b6SSatish Balay 1723a40ed3dSBarry Smith PetscFunctionBegin; 173218c64b6SSatish Balay ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr); 174218c64b6SSatish Balay ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr); 175b9b97703SBarry Smith ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr); 176b9b97703SBarry Smith ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr); 177218c64b6SSatish Balay 178218c64b6SSatish Balay /* Verify if the indices corespond to each element in a block 179218c64b6SSatish Balay and form the IS with compressed IS */ 180690b6cddSBarry Smith ierr = PetscMalloc(2*(a->mbs+1)*sizeof(PetscInt),&vary);CHKERRQ(ierr); 181218c64b6SSatish Balay iary = vary + a->mbs; 182690b6cddSBarry Smith ierr = PetscMemzero(vary,(a->mbs)*sizeof(PetscInt));CHKERRQ(ierr); 183218c64b6SSatish Balay for (i=0; i<nrows; i++) vary[irow[i]/bs]++; 184218c64b6SSatish Balay count = 0; 185218c64b6SSatish Balay for (i=0; i<a->mbs; i++) { 186634064b4SBarry Smith if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_ERR_ARG_SIZ,"Index set does not match blocks"); 187218c64b6SSatish Balay if (vary[i]==bs) iary[count++] = i; 188218c64b6SSatish Balay } 189029af93fSBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is1);CHKERRQ(ierr); 190218c64b6SSatish Balay 191690b6cddSBarry Smith ierr = PetscMemzero(vary,(a->mbs)*sizeof(PetscInt));CHKERRQ(ierr); 192218c64b6SSatish Balay for (i=0; i<ncols; i++) vary[icol[i]/bs]++; 193218c64b6SSatish Balay count = 0; 194218c64b6SSatish Balay for (i=0; i<a->mbs; i++) { 195634064b4SBarry Smith if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_ERR_PLIB,"Internal error in PETSc"); 196218c64b6SSatish Balay if (vary[i]==bs) iary[count++] = i; 197218c64b6SSatish Balay } 198029af93fSBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is2);CHKERRQ(ierr); 199218c64b6SSatish Balay ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 200218c64b6SSatish Balay ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 201606d414cSSatish Balay ierr = PetscFree(vary);CHKERRQ(ierr); 202218c64b6SSatish Balay 2034aa3045dSJed Brown ierr = MatGetSubMatrix_SeqBAIJ_Private(A,is1,is2,scall,B);CHKERRQ(ierr); 204218c64b6SSatish Balay ISDestroy(is1); 205218c64b6SSatish Balay ISDestroy(is2); 2063a40ed3dSBarry Smith PetscFunctionReturn(0); 207218c64b6SSatish Balay } 208218c64b6SSatish Balay 2094a2ae208SSatish Balay #undef __FUNCT__ 2104a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqBAIJ" 211690b6cddSBarry Smith PetscErrorCode MatGetSubMatrices_SeqBAIJ(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[]) 212736121d4SSatish Balay { 2136849ba73SBarry Smith PetscErrorCode ierr; 214690b6cddSBarry Smith PetscInt i; 215736121d4SSatish Balay 2163a40ed3dSBarry Smith PetscFunctionBegin; 217736121d4SSatish Balay if (scall == MAT_INITIAL_MATRIX) { 21882502324SSatish Balay ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr); 219736121d4SSatish Balay } 220736121d4SSatish Balay 221736121d4SSatish Balay for (i=0; i<n; i++) { 2224aa3045dSJed Brown ierr = MatGetSubMatrix_SeqBAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr); 223736121d4SSatish Balay } 2243a40ed3dSBarry Smith PetscFunctionReturn(0); 225736121d4SSatish Balay } 226218c64b6SSatish Balay 227218c64b6SSatish Balay 2282d61bbb3SSatish Balay /* -------------------------------------------------------*/ 2292d61bbb3SSatish Balay /* Should check that shapes of vectors and matrices match */ 2302d61bbb3SSatish Balay /* -------------------------------------------------------*/ 231d9eff348SSatish Balay #include "petscblaslapack.h" 2322d61bbb3SSatish Balay 2334a2ae208SSatish Balay #undef __FUNCT__ 2344a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_1" 235dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_1(Mat A,Vec xx,Vec zz) 2362d61bbb3SSatish Balay { 2372d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 238d9fead3dSBarry Smith PetscScalar *z,sum; 239d9fead3dSBarry Smith const PetscScalar *x; 240d9fead3dSBarry Smith const MatScalar *v; 2416849ba73SBarry Smith PetscErrorCode ierr; 2422162cab8SBarry Smith PetscInt mbs,i,n,nonzerorow=0; 2432162cab8SBarry Smith const PetscInt *idx,*ii,*ridx=PETSC_NULL; 24426e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 2452d61bbb3SSatish Balay 2462d61bbb3SSatish Balay PetscFunctionBegin; 247d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 2481ebc52fbSHong Zhang ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 2492d61bbb3SSatish Balay 25026e093fcSHong Zhang if (usecprow){ 25126e093fcSHong Zhang mbs = a->compressedrow.nrows; 25226e093fcSHong Zhang ii = a->compressedrow.i; 2537b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 254*a1c3900fSBarry Smith ierr = PetscMemzero(z,mbs*sizeof(PetscScalar));CHKERRQ(ierr); 25526e093fcSHong Zhang } else { 25626e093fcSHong Zhang mbs = a->mbs; 2572d61bbb3SSatish Balay ii = a->i; 25826e093fcSHong Zhang } 2592d61bbb3SSatish Balay 2602d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 261ee54c7eeSHong Zhang n = ii[1] - ii[0]; 262ee54c7eeSHong Zhang v = a->a + ii[0]; 263ee54c7eeSHong Zhang idx = a->j + ii[0]; 264ee54c7eeSHong Zhang ii++; 2652d61bbb3SSatish Balay sum = 0.0; 26698c9bda7SSatish Balay nonzerorow += (n>0); 2672162cab8SBarry Smith PetscSparseDensePlusDot(sum,x,v,idx,n); 26826e093fcSHong Zhang if (usecprow){ 2697b2bb3b9SHong Zhang z[ridx[i]] = sum; 27026e093fcSHong Zhang } else { 2712d61bbb3SSatish Balay z[i] = sum; 2722d61bbb3SSatish Balay } 27326e093fcSHong Zhang } 274d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 2751ebc52fbSHong Zhang ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 276dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr); 2772d61bbb3SSatish Balay PetscFunctionReturn(0); 2782d61bbb3SSatish Balay } 2792d61bbb3SSatish Balay 2804a2ae208SSatish Balay #undef __FUNCT__ 2814a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_2" 282dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_2(Mat A,Vec xx,Vec zz) 2832d61bbb3SSatish Balay { 2842d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 285d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,*zarray; 286d9fead3dSBarry Smith const PetscScalar *x,*xb; 28787828ca2SBarry Smith PetscScalar x1,x2; 288d9fead3dSBarry Smith const MatScalar *v; 289dfbe8321SBarry Smith PetscErrorCode ierr; 29098c9bda7SSatish Balay PetscInt mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 29126e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 2922d61bbb3SSatish Balay 2932d61bbb3SSatish Balay PetscFunctionBegin; 294d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 29526e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 2962d61bbb3SSatish Balay 2972d61bbb3SSatish Balay idx = a->j; 2982d61bbb3SSatish Balay v = a->a; 29926e093fcSHong Zhang if (usecprow){ 30026e093fcSHong Zhang mbs = a->compressedrow.nrows; 30126e093fcSHong Zhang ii = a->compressedrow.i; 3027b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 30326e093fcSHong Zhang } else { 30426e093fcSHong Zhang mbs = a->mbs; 3052d61bbb3SSatish Balay ii = a->i; 30626e093fcSHong Zhang z = zarray; 30726e093fcSHong Zhang } 3082d61bbb3SSatish Balay 3092d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 3102d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 3112d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; 31298c9bda7SSatish Balay nonzerorow += (n>0); 3132d61bbb3SSatish Balay for (j=0; j<n; j++) { 3142d61bbb3SSatish Balay xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1]; 3152d61bbb3SSatish Balay sum1 += v[0]*x1 + v[2]*x2; 3162d61bbb3SSatish Balay sum2 += v[1]*x1 + v[3]*x2; 3172d61bbb3SSatish Balay v += 4; 3182d61bbb3SSatish Balay } 3197b2bb3b9SHong Zhang if (usecprow) z = zarray + 2*ridx[i]; 3202d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; 32126e093fcSHong Zhang if (!usecprow) z += 2; 3222d61bbb3SSatish Balay } 323d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 32426e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 325dc0b31edSSatish Balay ierr = PetscLogFlops(8.0*a->nz - 2.0*nonzerorow);CHKERRQ(ierr); 3262d61bbb3SSatish Balay PetscFunctionReturn(0); 3272d61bbb3SSatish Balay } 3282d61bbb3SSatish Balay 3294a2ae208SSatish Balay #undef __FUNCT__ 3304a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_3" 331dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_3(Mat A,Vec xx,Vec zz) 3322d61bbb3SSatish Balay { 3332d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 334d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,x1,x2,x3,*zarray; 335d9fead3dSBarry Smith const PetscScalar *x,*xb; 336d9fead3dSBarry Smith const MatScalar *v; 337dfbe8321SBarry Smith PetscErrorCode ierr; 338028cd4eaSSatish Balay PetscInt mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 339028cd4eaSSatish Balay PetscTruth usecprow=a->compressedrow.use; 34026e093fcSHong Zhang 3412d61bbb3SSatish Balay 342b6410449SSatish Balay #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 343fee21e36SBarry Smith #pragma disjoint(*v,*z,*xb) 344fee21e36SBarry Smith #endif 345fee21e36SBarry Smith 3462d61bbb3SSatish Balay PetscFunctionBegin; 347d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 34826e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 3492d61bbb3SSatish Balay 3502d61bbb3SSatish Balay idx = a->j; 3512d61bbb3SSatish Balay v = a->a; 35226e093fcSHong Zhang if (usecprow){ 35326e093fcSHong Zhang mbs = a->compressedrow.nrows; 35426e093fcSHong Zhang ii = a->compressedrow.i; 3557b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 35626e093fcSHong Zhang } else { 35726e093fcSHong Zhang mbs = a->mbs; 3582d61bbb3SSatish Balay ii = a->i; 35926e093fcSHong Zhang z = zarray; 36026e093fcSHong Zhang } 3612d61bbb3SSatish Balay 3622d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 3632d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 3642d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; 36598c9bda7SSatish Balay nonzerorow += (n>0); 3662d61bbb3SSatish Balay for (j=0; j<n; j++) { 3672d61bbb3SSatish Balay xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 3682d61bbb3SSatish Balay sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3; 3692d61bbb3SSatish Balay sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3; 3702d61bbb3SSatish Balay sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3; 3712d61bbb3SSatish Balay v += 9; 3722d61bbb3SSatish Balay } 3737b2bb3b9SHong Zhang if (usecprow) z = zarray + 3*ridx[i]; 3742d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; 37526e093fcSHong Zhang if (!usecprow) z += 3; 3762d61bbb3SSatish Balay } 377d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 37826e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 379dc0b31edSSatish Balay ierr = PetscLogFlops(18.0*a->nz - 3.0*nonzerorow);CHKERRQ(ierr); 3802d61bbb3SSatish Balay PetscFunctionReturn(0); 3812d61bbb3SSatish Balay } 3822d61bbb3SSatish Balay 3834a2ae208SSatish Balay #undef __FUNCT__ 3844a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_4" 385dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_4(Mat A,Vec xx,Vec zz) 3862d61bbb3SSatish Balay { 3872d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 388d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*zarray; 389d9fead3dSBarry Smith const PetscScalar *x,*xb; 390d9fead3dSBarry Smith const MatScalar *v; 391dfbe8321SBarry Smith PetscErrorCode ierr; 39298c9bda7SSatish Balay PetscInt mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 39326e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 3942d61bbb3SSatish Balay 3952d61bbb3SSatish Balay PetscFunctionBegin; 396d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 39726e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 3982d61bbb3SSatish Balay 3992d61bbb3SSatish Balay idx = a->j; 4002d61bbb3SSatish Balay v = a->a; 40126e093fcSHong Zhang if (usecprow){ 40226e093fcSHong Zhang mbs = a->compressedrow.nrows; 40326e093fcSHong Zhang ii = a->compressedrow.i; 4047b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 40526e093fcSHong Zhang } else { 40626e093fcSHong Zhang mbs = a->mbs; 4072d61bbb3SSatish Balay ii = a->i; 40826e093fcSHong Zhang z = zarray; 40926e093fcSHong Zhang } 4102d61bbb3SSatish Balay 4112d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 4122d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 4132d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; 41498c9bda7SSatish Balay nonzerorow += (n>0); 4152d61bbb3SSatish Balay for (j=0; j<n; j++) { 4162d61bbb3SSatish Balay xb = x + 4*(*idx++); 4172d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 4182d61bbb3SSatish Balay sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3 + v[12]*x4; 4192d61bbb3SSatish Balay sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3 + v[13]*x4; 4202d61bbb3SSatish Balay sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3 + v[14]*x4; 4212d61bbb3SSatish Balay sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3 + v[15]*x4; 4222d61bbb3SSatish Balay v += 16; 4232d61bbb3SSatish Balay } 4247b2bb3b9SHong Zhang if (usecprow) z = zarray + 4*ridx[i]; 4252d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; 42626e093fcSHong Zhang if (!usecprow) z += 4; 4272d61bbb3SSatish Balay } 428d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 42926e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 430dc0b31edSSatish Balay ierr = PetscLogFlops(32.0*a->nz - 4.0*nonzerorow);CHKERRQ(ierr); 4312d61bbb3SSatish Balay PetscFunctionReturn(0); 4322d61bbb3SSatish Balay } 4332d61bbb3SSatish Balay 4344a2ae208SSatish Balay #undef __FUNCT__ 4354a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_5" 436dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_5(Mat A,Vec xx,Vec zz) 4372d61bbb3SSatish Balay { 4382d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 439d9fead3dSBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5,*z = 0,*zarray; 440d9fead3dSBarry Smith const PetscScalar *xb,*x; 441d9fead3dSBarry Smith const MatScalar *v; 442dfbe8321SBarry Smith PetscErrorCode ierr; 44398c9bda7SSatish Balay PetscInt mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 44426e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 4452d61bbb3SSatish Balay 446433994e6SBarry Smith PetscFunctionBegin; 447d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 44826e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 4492d61bbb3SSatish Balay 4502d61bbb3SSatish Balay idx = a->j; 4512d61bbb3SSatish Balay v = a->a; 45226e093fcSHong Zhang if (usecprow){ 45326e093fcSHong Zhang mbs = a->compressedrow.nrows; 45426e093fcSHong Zhang ii = a->compressedrow.i; 4557b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 45626e093fcSHong Zhang } else { 45726e093fcSHong Zhang mbs = a->mbs; 4582d61bbb3SSatish Balay ii = a->i; 45926e093fcSHong Zhang z = zarray; 46026e093fcSHong Zhang } 4612d61bbb3SSatish Balay 4622d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 4632d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 4642d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; 46598c9bda7SSatish Balay nonzerorow += (n>0); 4662d61bbb3SSatish Balay for (j=0; j<n; j++) { 4672d61bbb3SSatish Balay xb = x + 5*(*idx++); 4682d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; 4692d61bbb3SSatish Balay sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3 + v[15]*x4 + v[20]*x5; 4702d61bbb3SSatish Balay sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3 + v[16]*x4 + v[21]*x5; 4712d61bbb3SSatish Balay sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3 + v[17]*x4 + v[22]*x5; 4722d61bbb3SSatish Balay sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3 + v[18]*x4 + v[23]*x5; 4732d61bbb3SSatish Balay sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3 + v[19]*x4 + v[24]*x5; 4742d61bbb3SSatish Balay v += 25; 4752d61bbb3SSatish Balay } 4767b2bb3b9SHong Zhang if (usecprow) z = zarray + 5*ridx[i]; 4772d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; 47826e093fcSHong Zhang if (!usecprow) z += 5; 4792d61bbb3SSatish Balay } 480d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 48126e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 482dc0b31edSSatish Balay ierr = PetscLogFlops(50.0*a->nz - 5.0*nonzerorow);CHKERRQ(ierr); 4832d61bbb3SSatish Balay PetscFunctionReturn(0); 4842d61bbb3SSatish Balay } 4852d61bbb3SSatish Balay 48615091d37SBarry Smith 4874a2ae208SSatish Balay #undef __FUNCT__ 4884a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_6" 489dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_6(Mat A,Vec xx,Vec zz) 49015091d37SBarry Smith { 49115091d37SBarry Smith Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 492d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6; 493d9fead3dSBarry Smith const PetscScalar *x,*xb; 49426e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,*zarray; 495d9fead3dSBarry Smith const MatScalar *v; 496dfbe8321SBarry Smith PetscErrorCode ierr; 49798c9bda7SSatish Balay PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 49826e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 49915091d37SBarry Smith 500433994e6SBarry Smith PetscFunctionBegin; 501d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 50226e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 50315091d37SBarry Smith 50415091d37SBarry Smith idx = a->j; 50515091d37SBarry Smith v = a->a; 50626e093fcSHong Zhang if (usecprow){ 50726e093fcSHong Zhang mbs = a->compressedrow.nrows; 50826e093fcSHong Zhang ii = a->compressedrow.i; 5097b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 51026e093fcSHong Zhang } else { 51126e093fcSHong Zhang mbs = a->mbs; 51215091d37SBarry Smith ii = a->i; 51326e093fcSHong Zhang z = zarray; 51426e093fcSHong Zhang } 51515091d37SBarry Smith 51615091d37SBarry Smith for (i=0; i<mbs; i++) { 51715091d37SBarry Smith n = ii[1] - ii[0]; ii++; 51815091d37SBarry Smith sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; 51998c9bda7SSatish Balay nonzerorow += (n>0); 52015091d37SBarry Smith for (j=0; j<n; j++) { 52115091d37SBarry Smith xb = x + 6*(*idx++); 52215091d37SBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; 52315091d37SBarry Smith sum1 += v[0]*x1 + v[6]*x2 + v[12]*x3 + v[18]*x4 + v[24]*x5 + v[30]*x6; 52415091d37SBarry Smith sum2 += v[1]*x1 + v[7]*x2 + v[13]*x3 + v[19]*x4 + v[25]*x5 + v[31]*x6; 52515091d37SBarry Smith sum3 += v[2]*x1 + v[8]*x2 + v[14]*x3 + v[20]*x4 + v[26]*x5 + v[32]*x6; 52615091d37SBarry Smith sum4 += v[3]*x1 + v[9]*x2 + v[15]*x3 + v[21]*x4 + v[27]*x5 + v[33]*x6; 52715091d37SBarry Smith sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3 + v[22]*x4 + v[28]*x5 + v[34]*x6; 52815091d37SBarry Smith sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3 + v[23]*x4 + v[29]*x5 + v[35]*x6; 52915091d37SBarry Smith v += 36; 53015091d37SBarry Smith } 5317b2bb3b9SHong Zhang if (usecprow) z = zarray + 6*ridx[i]; 53215091d37SBarry Smith z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; 53326e093fcSHong Zhang if (!usecprow) z += 6; 53415091d37SBarry Smith } 53515091d37SBarry Smith 536d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 53726e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 538dc0b31edSSatish Balay ierr = PetscLogFlops(72.0*a->nz - 6.0*nonzerorow);CHKERRQ(ierr); 53915091d37SBarry Smith PetscFunctionReturn(0); 54015091d37SBarry Smith } 5414a2ae208SSatish Balay #undef __FUNCT__ 5424a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_7" 543dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_7(Mat A,Vec xx,Vec zz) 5442d61bbb3SSatish Balay { 5452d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 546d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7; 547d9fead3dSBarry Smith const PetscScalar *x,*xb; 54826e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,x7,*zarray; 549d9fead3dSBarry Smith const MatScalar *v; 550dfbe8321SBarry Smith PetscErrorCode ierr; 55198c9bda7SSatish Balay PetscInt mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 55226e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 5532d61bbb3SSatish Balay 554433994e6SBarry Smith PetscFunctionBegin; 555d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 55626e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 5572d61bbb3SSatish Balay 5582d61bbb3SSatish Balay idx = a->j; 5592d61bbb3SSatish Balay v = a->a; 56026e093fcSHong Zhang if (usecprow){ 56126e093fcSHong Zhang mbs = a->compressedrow.nrows; 56226e093fcSHong Zhang ii = a->compressedrow.i; 5637b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 56426e093fcSHong Zhang } else { 56526e093fcSHong Zhang mbs = a->mbs; 5662d61bbb3SSatish Balay ii = a->i; 56726e093fcSHong Zhang z = zarray; 56826e093fcSHong Zhang } 5692d61bbb3SSatish Balay 5702d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 5712d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 5722d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 57398c9bda7SSatish Balay nonzerorow += (n>0); 5742d61bbb3SSatish Balay for (j=0; j<n; j++) { 5752d61bbb3SSatish Balay xb = x + 7*(*idx++); 5762d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 5772d61bbb3SSatish Balay sum1 += v[0]*x1 + v[7]*x2 + v[14]*x3 + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7; 5782d61bbb3SSatish Balay sum2 += v[1]*x1 + v[8]*x2 + v[15]*x3 + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7; 5792d61bbb3SSatish Balay sum3 += v[2]*x1 + v[9]*x2 + v[16]*x3 + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7; 5802d61bbb3SSatish Balay sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3 + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7; 5812d61bbb3SSatish Balay sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3 + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7; 5822d61bbb3SSatish Balay sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3 + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7; 5832d61bbb3SSatish Balay sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3 + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7; 5842d61bbb3SSatish Balay v += 49; 5852d61bbb3SSatish Balay } 5867b2bb3b9SHong Zhang if (usecprow) z = zarray + 7*ridx[i]; 5872d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 58826e093fcSHong Zhang if (!usecprow) z += 7; 5892d61bbb3SSatish Balay } 5902d61bbb3SSatish Balay 591d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 59226e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 593dc0b31edSSatish Balay ierr = PetscLogFlops(98.0*a->nz - 7.0*nonzerorow);CHKERRQ(ierr); 5942d61bbb3SSatish Balay PetscFunctionReturn(0); 5952d61bbb3SSatish Balay } 5962d61bbb3SSatish Balay 5973f1db9ecSBarry Smith /* 5983f1db9ecSBarry Smith This will not work with MatScalar == float because it calls the BLAS 5993f1db9ecSBarry Smith */ 6004a2ae208SSatish Balay #undef __FUNCT__ 6014a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_N" 602dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_N(Mat A,Vec xx,Vec zz) 6032d61bbb3SSatish Balay { 6042d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 605dcf5cc72SBarry Smith PetscScalar *x,*z = 0,*xb,*work,*workt,*zarray; 6063f1db9ecSBarry Smith MatScalar *v; 607dfbe8321SBarry Smith PetscErrorCode ierr; 608d0f46423SBarry Smith PetscInt mbs=a->mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2; 60998c9bda7SSatish Balay PetscInt ncols,k,*ridx=PETSC_NULL,nonzerorow=0; 61026e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 6112d61bbb3SSatish Balay 6122d61bbb3SSatish Balay PetscFunctionBegin; 6131ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 61426e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 6152d61bbb3SSatish Balay 6162d61bbb3SSatish Balay idx = a->j; 6172d61bbb3SSatish Balay v = a->a; 61826e093fcSHong Zhang if (usecprow){ 61926e093fcSHong Zhang mbs = a->compressedrow.nrows; 62026e093fcSHong Zhang ii = a->compressedrow.i; 6217b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 62226e093fcSHong Zhang } else { 62326e093fcSHong Zhang mbs = a->mbs; 6242d61bbb3SSatish Balay ii = a->i; 62526e093fcSHong Zhang z = zarray; 62626e093fcSHong Zhang } 627218c64b6SSatish Balay 6282d61bbb3SSatish Balay if (!a->mult_work) { 629d0f46423SBarry Smith k = PetscMax(A->rmap->n,A->cmap->n); 63087828ca2SBarry Smith ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr); 6312d61bbb3SSatish Balay } 6322d61bbb3SSatish Balay work = a->mult_work; 6332d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 6342d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 6352d61bbb3SSatish Balay ncols = n*bs; 6362d61bbb3SSatish Balay workt = work; 63798c9bda7SSatish Balay nonzerorow += (n>0); 6382d61bbb3SSatish Balay for (j=0; j<n; j++) { 6392d61bbb3SSatish Balay xb = x + bs*(*idx++); 6402d61bbb3SSatish Balay for (k=0; k<bs; k++) workt[k] = xb[k]; 6412d61bbb3SSatish Balay workt += bs; 6422d61bbb3SSatish Balay } 6437b2bb3b9SHong Zhang if (usecprow) z = zarray + bs*ridx[i]; 644737d121aSSatish Balay Kernel_w_gets_Ar_times_v(bs,ncols,work,v,z); 64571044d3cSBarry Smith /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DZero,z,&_One); */ 6462d61bbb3SSatish Balay v += n*bs2; 64726e093fcSHong Zhang if (!usecprow) z += bs; 6482d61bbb3SSatish Balay } 6491ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 65026e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 651dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz*bs2 - bs*nonzerorow);CHKERRQ(ierr); 6522d61bbb3SSatish Balay PetscFunctionReturn(0); 6532d61bbb3SSatish Balay } 6542d61bbb3SSatish Balay 6556a65c766SHong Zhang extern PetscErrorCode VecCopy_Seq(Vec,Vec); 6564a2ae208SSatish Balay #undef __FUNCT__ 6574a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_1" 658dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_1(Mat A,Vec xx,Vec yy,Vec zz) 6592d61bbb3SSatish Balay { 6602d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 661dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,sum,*yarray,*zarray; 6623f1db9ecSBarry Smith MatScalar *v; 663dfbe8321SBarry Smith PetscErrorCode ierr; 6644eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,n,*ridx=PETSC_NULL; 66526e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 6662d61bbb3SSatish Balay 6672d61bbb3SSatish Balay PetscFunctionBegin; 6681ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 66926e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 6702e8a6d31SBarry Smith if (zz != yy) { 67126e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 6722e8a6d31SBarry Smith } else { 67326e093fcSHong Zhang zarray = yarray; 6742e8a6d31SBarry Smith } 6752d61bbb3SSatish Balay 6762d61bbb3SSatish Balay idx = a->j; 6772d61bbb3SSatish Balay v = a->a; 67826e093fcSHong Zhang if (usecprow){ 6794eb6d288SHong Zhang if (zz != yy){ 6804eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,mbs*sizeof(PetscScalar));CHKERRQ(ierr); 6814eb6d288SHong Zhang } 68226e093fcSHong Zhang mbs = a->compressedrow.nrows; 68326e093fcSHong Zhang ii = a->compressedrow.i; 6847b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 68526e093fcSHong Zhang } else { 6862d61bbb3SSatish Balay ii = a->i; 68726e093fcSHong Zhang y = yarray; 68826e093fcSHong Zhang z = zarray; 68926e093fcSHong Zhang } 6902d61bbb3SSatish Balay 6912d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 6922d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 69326e093fcSHong Zhang if (usecprow){ 6947b2bb3b9SHong Zhang z = zarray + ridx[i]; 6957b2bb3b9SHong Zhang y = yarray + ridx[i]; 69626e093fcSHong Zhang } 69726e093fcSHong Zhang sum = y[0]; 6982d61bbb3SSatish Balay while (n--) sum += *v++ * x[*idx++]; 69926e093fcSHong Zhang z[0] = sum; 70026e093fcSHong Zhang if (!usecprow){ 70126e093fcSHong Zhang z++; y++; 70226e093fcSHong Zhang } 7032d61bbb3SSatish Balay } 7041ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 70526e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 7062e8a6d31SBarry Smith if (zz != yy) { 70726e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 7082e8a6d31SBarry Smith } 709dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 7102d61bbb3SSatish Balay PetscFunctionReturn(0); 7112d61bbb3SSatish Balay } 7122d61bbb3SSatish Balay 7134a2ae208SSatish Balay #undef __FUNCT__ 7144a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_2" 715dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_2(Mat A,Vec xx,Vec yy,Vec zz) 7162d61bbb3SSatish Balay { 7172d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 718dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2; 71926e093fcSHong Zhang PetscScalar x1,x2,*yarray,*zarray; 7203f1db9ecSBarry Smith MatScalar *v; 721dfbe8321SBarry Smith PetscErrorCode ierr; 7224eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 72326e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 7242d61bbb3SSatish Balay 7252d61bbb3SSatish Balay PetscFunctionBegin; 7261ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 72726e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 7282e8a6d31SBarry Smith if (zz != yy) { 72926e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 7302e8a6d31SBarry Smith } else { 73126e093fcSHong Zhang zarray = yarray; 7322e8a6d31SBarry Smith } 7332d61bbb3SSatish Balay 7342d61bbb3SSatish Balay idx = a->j; 7352d61bbb3SSatish Balay v = a->a; 73626e093fcSHong Zhang if (usecprow){ 7374eb6d288SHong Zhang if (zz != yy){ 7384eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,2*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 7394eb6d288SHong Zhang } 74026e093fcSHong Zhang mbs = a->compressedrow.nrows; 74126e093fcSHong Zhang ii = a->compressedrow.i; 7427b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 7434eb6d288SHong Zhang if (zz != yy){ 7444eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,a->mbs*sizeof(PetscScalar));CHKERRQ(ierr); 7454eb6d288SHong Zhang } 74626e093fcSHong Zhang } else { 7472d61bbb3SSatish Balay ii = a->i; 74826e093fcSHong Zhang y = yarray; 74926e093fcSHong Zhang z = zarray; 75026e093fcSHong Zhang } 7512d61bbb3SSatish Balay 7522d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 7532d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 75426e093fcSHong Zhang if (usecprow){ 7557b2bb3b9SHong Zhang z = zarray + 2*ridx[i]; 7567b2bb3b9SHong Zhang y = yarray + 2*ridx[i]; 75726e093fcSHong Zhang } 7582d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; 7592d61bbb3SSatish Balay for (j=0; j<n; j++) { 7602d61bbb3SSatish Balay xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1]; 7612d61bbb3SSatish Balay sum1 += v[0]*x1 + v[2]*x2; 7622d61bbb3SSatish Balay sum2 += v[1]*x1 + v[3]*x2; 7632d61bbb3SSatish Balay v += 4; 7642d61bbb3SSatish Balay } 7652d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; 76626e093fcSHong Zhang if (!usecprow){ 7672d61bbb3SSatish Balay z += 2; y += 2; 7682d61bbb3SSatish Balay } 76926e093fcSHong Zhang } 7701ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 77126e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 7722e8a6d31SBarry Smith if (zz != yy) { 77326e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 7742e8a6d31SBarry Smith } 775dc0b31edSSatish Balay ierr = PetscLogFlops(4.0*a->nz);CHKERRQ(ierr); 7762d61bbb3SSatish Balay PetscFunctionReturn(0); 7772d61bbb3SSatish Balay } 7782d61bbb3SSatish Balay 7794a2ae208SSatish Balay #undef __FUNCT__ 7804a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_3" 781dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_3(Mat A,Vec xx,Vec yy,Vec zz) 7822d61bbb3SSatish Balay { 7832d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 784dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,x1,x2,x3,*yarray,*zarray; 7853f1db9ecSBarry Smith MatScalar *v; 786dfbe8321SBarry Smith PetscErrorCode ierr; 7874eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 78826e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 7892d61bbb3SSatish Balay 7902d61bbb3SSatish Balay PetscFunctionBegin; 7911ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 79226e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 7932e8a6d31SBarry Smith if (zz != yy) { 79426e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 7952e8a6d31SBarry Smith } else { 79626e093fcSHong Zhang zarray = yarray; 7972e8a6d31SBarry Smith } 7982d61bbb3SSatish Balay 7992d61bbb3SSatish Balay idx = a->j; 8002d61bbb3SSatish Balay v = a->a; 80126e093fcSHong Zhang if (usecprow){ 8024eb6d288SHong Zhang if (zz != yy){ 8034eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,3*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 8044eb6d288SHong Zhang } 80526e093fcSHong Zhang mbs = a->compressedrow.nrows; 80626e093fcSHong Zhang ii = a->compressedrow.i; 8077b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 80826e093fcSHong Zhang } else { 8092d61bbb3SSatish Balay ii = a->i; 81026e093fcSHong Zhang y = yarray; 81126e093fcSHong Zhang z = zarray; 81226e093fcSHong Zhang } 8132d61bbb3SSatish Balay 8142d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 8152d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 81626e093fcSHong Zhang if (usecprow){ 8177b2bb3b9SHong Zhang z = zarray + 3*ridx[i]; 8187b2bb3b9SHong Zhang y = yarray + 3*ridx[i]; 81926e093fcSHong Zhang } 8202d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; 8212d61bbb3SSatish Balay for (j=0; j<n; j++) { 8222d61bbb3SSatish Balay xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 8232d61bbb3SSatish Balay sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3; 8242d61bbb3SSatish Balay sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3; 8252d61bbb3SSatish Balay sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3; 8262d61bbb3SSatish Balay v += 9; 8272d61bbb3SSatish Balay } 8282d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; 82926e093fcSHong Zhang if (!usecprow){ 8302d61bbb3SSatish Balay z += 3; y += 3; 8312d61bbb3SSatish Balay } 83226e093fcSHong Zhang } 8331ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 83426e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 8352e8a6d31SBarry Smith if (zz != yy) { 83626e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 8372e8a6d31SBarry Smith } 838dc0b31edSSatish Balay ierr = PetscLogFlops(18.0*a->nz);CHKERRQ(ierr); 8392d61bbb3SSatish Balay PetscFunctionReturn(0); 8402d61bbb3SSatish Balay } 8412d61bbb3SSatish Balay 8424a2ae208SSatish Balay #undef __FUNCT__ 8434a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_4" 844dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_4(Mat A,Vec xx,Vec yy,Vec zz) 8452d61bbb3SSatish Balay { 8462d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 847dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*yarray,*zarray; 8483f1db9ecSBarry Smith MatScalar *v; 849dfbe8321SBarry Smith PetscErrorCode ierr; 8504eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 85126e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 8522d61bbb3SSatish Balay 8532d61bbb3SSatish Balay PetscFunctionBegin; 8541ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 85526e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 8562e8a6d31SBarry Smith if (zz != yy) { 85726e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 8582e8a6d31SBarry Smith } else { 85926e093fcSHong Zhang zarray = yarray; 8602e8a6d31SBarry Smith } 8612d61bbb3SSatish Balay 8622d61bbb3SSatish Balay idx = a->j; 8632d61bbb3SSatish Balay v = a->a; 86426e093fcSHong Zhang if (usecprow){ 8654eb6d288SHong Zhang if (zz != yy){ 8664eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,4*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 8674eb6d288SHong Zhang } 86826e093fcSHong Zhang mbs = a->compressedrow.nrows; 86926e093fcSHong Zhang ii = a->compressedrow.i; 8707b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 87126e093fcSHong Zhang } else { 8722d61bbb3SSatish Balay ii = a->i; 87326e093fcSHong Zhang y = yarray; 87426e093fcSHong Zhang z = zarray; 87526e093fcSHong Zhang } 8762d61bbb3SSatish Balay 8772d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 8782d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 87926e093fcSHong Zhang if (usecprow){ 8807b2bb3b9SHong Zhang z = zarray + 4*ridx[i]; 8817b2bb3b9SHong Zhang y = yarray + 4*ridx[i]; 88226e093fcSHong Zhang } 8832d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; 8842d61bbb3SSatish Balay for (j=0; j<n; j++) { 8852d61bbb3SSatish Balay xb = x + 4*(*idx++); 8862d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 8872d61bbb3SSatish Balay sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3 + v[12]*x4; 8882d61bbb3SSatish Balay sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3 + v[13]*x4; 8892d61bbb3SSatish Balay sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3 + v[14]*x4; 8902d61bbb3SSatish Balay sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3 + v[15]*x4; 8912d61bbb3SSatish Balay v += 16; 8922d61bbb3SSatish Balay } 8932d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; 89426e093fcSHong Zhang if (!usecprow){ 8952d61bbb3SSatish Balay z += 4; y += 4; 8962d61bbb3SSatish Balay } 89726e093fcSHong Zhang } 8981ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 89926e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 9002e8a6d31SBarry Smith if (zz != yy) { 90126e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 9022e8a6d31SBarry Smith } 903dc0b31edSSatish Balay ierr = PetscLogFlops(32.0*a->nz);CHKERRQ(ierr); 9042d61bbb3SSatish Balay PetscFunctionReturn(0); 9052d61bbb3SSatish Balay } 9062d61bbb3SSatish Balay 9074a2ae208SSatish Balay #undef __FUNCT__ 9084a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_5" 909dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_5(Mat A,Vec xx,Vec yy,Vec zz) 9102d61bbb3SSatish Balay { 9112d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 912dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5; 91326e093fcSHong Zhang PetscScalar *yarray,*zarray; 9143f1db9ecSBarry Smith MatScalar *v; 915dfbe8321SBarry Smith PetscErrorCode ierr; 9164eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 91726e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 9182d61bbb3SSatish Balay 9192d61bbb3SSatish Balay PetscFunctionBegin; 9201ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 92126e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 9222e8a6d31SBarry Smith if (zz != yy) { 92326e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 9242e8a6d31SBarry Smith } else { 92526e093fcSHong Zhang zarray = yarray; 9262e8a6d31SBarry Smith } 9272d61bbb3SSatish Balay 9282d61bbb3SSatish Balay idx = a->j; 9292d61bbb3SSatish Balay v = a->a; 93026e093fcSHong Zhang if (usecprow){ 9314eb6d288SHong Zhang if (zz != yy){ 9324eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,5*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 9334eb6d288SHong Zhang } 93426e093fcSHong Zhang mbs = a->compressedrow.nrows; 93526e093fcSHong Zhang ii = a->compressedrow.i; 9367b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 93726e093fcSHong Zhang } else { 9382d61bbb3SSatish Balay ii = a->i; 93926e093fcSHong Zhang y = yarray; 94026e093fcSHong Zhang z = zarray; 94126e093fcSHong Zhang } 9422d61bbb3SSatish Balay 9432d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 9442d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 94526e093fcSHong Zhang if (usecprow){ 9467b2bb3b9SHong Zhang z = zarray + 5*ridx[i]; 9477b2bb3b9SHong Zhang y = yarray + 5*ridx[i]; 94826e093fcSHong Zhang } 9492d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; 9502d61bbb3SSatish Balay for (j=0; j<n; j++) { 9512d61bbb3SSatish Balay xb = x + 5*(*idx++); 9522d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; 9532d61bbb3SSatish Balay sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3 + v[15]*x4 + v[20]*x5; 9542d61bbb3SSatish Balay sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3 + v[16]*x4 + v[21]*x5; 9552d61bbb3SSatish Balay sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3 + v[17]*x4 + v[22]*x5; 9562d61bbb3SSatish Balay sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3 + v[18]*x4 + v[23]*x5; 9572d61bbb3SSatish Balay sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3 + v[19]*x4 + v[24]*x5; 9582d61bbb3SSatish Balay v += 25; 9592d61bbb3SSatish Balay } 9602d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; 96126e093fcSHong Zhang if (!usecprow){ 9622d61bbb3SSatish Balay z += 5; y += 5; 9632d61bbb3SSatish Balay } 96426e093fcSHong Zhang } 9651ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 96626e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 9672e8a6d31SBarry Smith if (zz != yy) { 96826e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 9692e8a6d31SBarry Smith } 970dc0b31edSSatish Balay ierr = PetscLogFlops(50.0*a->nz);CHKERRQ(ierr); 9712d61bbb3SSatish Balay PetscFunctionReturn(0); 9722d61bbb3SSatish Balay } 9734a2ae208SSatish Balay #undef __FUNCT__ 9744a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_6" 975dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_6(Mat A,Vec xx,Vec yy,Vec zz) 97615091d37SBarry Smith { 97715091d37SBarry Smith Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 978dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6; 97926e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,*yarray,*zarray; 98015091d37SBarry Smith MatScalar *v; 981dfbe8321SBarry Smith PetscErrorCode ierr; 9824eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 98326e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 98415091d37SBarry Smith 98515091d37SBarry Smith PetscFunctionBegin; 9861ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 98726e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 98815091d37SBarry Smith if (zz != yy) { 98926e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 99015091d37SBarry Smith } else { 99126e093fcSHong Zhang zarray = yarray; 99215091d37SBarry Smith } 99315091d37SBarry Smith 99415091d37SBarry Smith idx = a->j; 99515091d37SBarry Smith v = a->a; 99626e093fcSHong Zhang if (usecprow){ 9974eb6d288SHong Zhang if (zz != yy){ 9984eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,6*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 9994eb6d288SHong Zhang } 100026e093fcSHong Zhang mbs = a->compressedrow.nrows; 100126e093fcSHong Zhang ii = a->compressedrow.i; 10027b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 100326e093fcSHong Zhang } else { 100415091d37SBarry Smith ii = a->i; 100526e093fcSHong Zhang y = yarray; 100626e093fcSHong Zhang z = zarray; 100726e093fcSHong Zhang } 100815091d37SBarry Smith 100915091d37SBarry Smith for (i=0; i<mbs; i++) { 101015091d37SBarry Smith n = ii[1] - ii[0]; ii++; 101126e093fcSHong Zhang if (usecprow){ 10127b2bb3b9SHong Zhang z = zarray + 6*ridx[i]; 10137b2bb3b9SHong Zhang y = yarray + 6*ridx[i]; 101426e093fcSHong Zhang } 101515091d37SBarry Smith sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5]; 101615091d37SBarry Smith for (j=0; j<n; j++) { 10173b95cb0eSSatish Balay xb = x + 6*(*idx++); 101815091d37SBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; 101915091d37SBarry Smith sum1 += v[0]*x1 + v[6]*x2 + v[12]*x3 + v[18]*x4 + v[24]*x5 + v[30]*x6; 102015091d37SBarry Smith sum2 += v[1]*x1 + v[7]*x2 + v[13]*x3 + v[19]*x4 + v[25]*x5 + v[31]*x6; 102115091d37SBarry Smith sum3 += v[2]*x1 + v[8]*x2 + v[14]*x3 + v[20]*x4 + v[26]*x5 + v[32]*x6; 102215091d37SBarry Smith sum4 += v[3]*x1 + v[9]*x2 + v[15]*x3 + v[21]*x4 + v[27]*x5 + v[33]*x6; 102315091d37SBarry Smith sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3 + v[22]*x4 + v[28]*x5 + v[34]*x6; 102415091d37SBarry Smith sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3 + v[23]*x4 + v[29]*x5 + v[35]*x6; 102515091d37SBarry Smith v += 36; 102615091d37SBarry Smith } 102715091d37SBarry Smith z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; 102826e093fcSHong Zhang if (!usecprow){ 102915091d37SBarry Smith z += 6; y += 6; 103015091d37SBarry Smith } 103126e093fcSHong Zhang } 10321ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 103326e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 103415091d37SBarry Smith if (zz != yy) { 103526e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 103615091d37SBarry Smith } 1037dc0b31edSSatish Balay ierr = PetscLogFlops(72.0*a->nz);CHKERRQ(ierr); 103815091d37SBarry Smith PetscFunctionReturn(0); 103915091d37SBarry Smith } 10402d61bbb3SSatish Balay 10414a2ae208SSatish Balay #undef __FUNCT__ 10424a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_7" 1043dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_7(Mat A,Vec xx,Vec yy,Vec zz) 10442d61bbb3SSatish Balay { 10452d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1046dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6,sum7; 104726e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,x7,*yarray,*zarray; 10483f1db9ecSBarry Smith MatScalar *v; 1049dfbe8321SBarry Smith PetscErrorCode ierr; 10507b2bb3b9SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 105126e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 10522d61bbb3SSatish Balay 10532d61bbb3SSatish Balay PetscFunctionBegin; 10541ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 105526e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 10562e8a6d31SBarry Smith if (zz != yy) { 105726e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 10582e8a6d31SBarry Smith } else { 105926e093fcSHong Zhang zarray = yarray; 10602e8a6d31SBarry Smith } 10612d61bbb3SSatish Balay 10622d61bbb3SSatish Balay idx = a->j; 10632d61bbb3SSatish Balay v = a->a; 106426e093fcSHong Zhang if (usecprow){ 10654eb6d288SHong Zhang if (zz != yy){ 10664eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,7*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 10674eb6d288SHong Zhang } 106826e093fcSHong Zhang mbs = a->compressedrow.nrows; 106926e093fcSHong Zhang ii = a->compressedrow.i; 10707b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 107126e093fcSHong Zhang } else { 10722d61bbb3SSatish Balay ii = a->i; 107326e093fcSHong Zhang y = yarray; 107426e093fcSHong Zhang z = zarray; 107526e093fcSHong Zhang } 10762d61bbb3SSatish Balay 10772d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 10782d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 107926e093fcSHong Zhang if (usecprow){ 10807b2bb3b9SHong Zhang z = zarray + 7*ridx[i]; 10817b2bb3b9SHong Zhang y = yarray + 7*ridx[i]; 108226e093fcSHong Zhang } 10832d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5]; sum7 = y[6]; 10842d61bbb3SSatish Balay for (j=0; j<n; j++) { 10852d61bbb3SSatish Balay xb = x + 7*(*idx++); 10862d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 10872d61bbb3SSatish Balay sum1 += v[0]*x1 + v[7]*x2 + v[14]*x3 + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7; 10882d61bbb3SSatish Balay sum2 += v[1]*x1 + v[8]*x2 + v[15]*x3 + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7; 10892d61bbb3SSatish Balay sum3 += v[2]*x1 + v[9]*x2 + v[16]*x3 + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7; 10902d61bbb3SSatish Balay sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3 + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7; 10912d61bbb3SSatish Balay sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3 + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7; 10922d61bbb3SSatish Balay sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3 + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7; 10932d61bbb3SSatish Balay sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3 + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7; 10942d61bbb3SSatish Balay v += 49; 10952d61bbb3SSatish Balay } 10962d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 109726e093fcSHong Zhang if (!usecprow){ 10982d61bbb3SSatish Balay z += 7; y += 7; 10992d61bbb3SSatish Balay } 110026e093fcSHong Zhang } 11011ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 110226e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 11032e8a6d31SBarry Smith if (zz != yy) { 110426e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 11052e8a6d31SBarry Smith } 1106dc0b31edSSatish Balay ierr = PetscLogFlops(98.0*a->nz);CHKERRQ(ierr); 11072d61bbb3SSatish Balay PetscFunctionReturn(0); 11082d61bbb3SSatish Balay } 1109218c64b6SSatish Balay 11104a2ae208SSatish Balay #undef __FUNCT__ 11114a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_N" 1112dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_N(Mat A,Vec xx,Vec yy,Vec zz) 11132d61bbb3SSatish Balay { 11142d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1115bba805e6SBarry Smith PetscScalar *x,*z = 0,*xb,*work,*workt,*zarray; 11163f1db9ecSBarry Smith MatScalar *v; 11176849ba73SBarry Smith PetscErrorCode ierr; 1118d0f46423SBarry Smith PetscInt mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2; 11197b2bb3b9SHong Zhang PetscInt ncols,k,*ridx=PETSC_NULL; 112026e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 1121218c64b6SSatish Balay 11222d61bbb3SSatish Balay PetscFunctionBegin; 11236a65c766SHong Zhang ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr); 11241ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 112526e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 11262d61bbb3SSatish Balay 11272d61bbb3SSatish Balay idx = a->j; 11282d61bbb3SSatish Balay v = a->a; 112926e093fcSHong Zhang if (usecprow){ 113026e093fcSHong Zhang mbs = a->compressedrow.nrows; 113126e093fcSHong Zhang ii = a->compressedrow.i; 11327b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 113326e093fcSHong Zhang } else { 113426e093fcSHong Zhang mbs = a->mbs; 11352d61bbb3SSatish Balay ii = a->i; 113626e093fcSHong Zhang z = zarray; 113726e093fcSHong Zhang } 11382d61bbb3SSatish Balay 11392d61bbb3SSatish Balay if (!a->mult_work) { 1140d0f46423SBarry Smith k = PetscMax(A->rmap->n,A->cmap->n); 114187828ca2SBarry Smith ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr); 11422d61bbb3SSatish Balay } 11432d61bbb3SSatish Balay work = a->mult_work; 11442d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 11452d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 11462d61bbb3SSatish Balay ncols = n*bs; 11472d61bbb3SSatish Balay workt = work; 11482d61bbb3SSatish Balay for (j=0; j<n; j++) { 11492d61bbb3SSatish Balay xb = x + bs*(*idx++); 11502d61bbb3SSatish Balay for (k=0; k<bs; k++) workt[k] = xb[k]; 11512d61bbb3SSatish Balay workt += bs; 11522d61bbb3SSatish Balay } 11537b2bb3b9SHong Zhang if (usecprow) z = zarray + bs*ridx[i]; 11543f1db9ecSBarry Smith Kernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z); 115571044d3cSBarry Smith /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DOne,z,&_One); */ 11562d61bbb3SSatish Balay v += n*bs2; 115726e093fcSHong Zhang if (!usecprow){ 11582d61bbb3SSatish Balay z += bs; 11592d61bbb3SSatish Balay } 116026e093fcSHong Zhang } 11611ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 116226e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 1163dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz*bs2);CHKERRQ(ierr); 11642d61bbb3SSatish Balay PetscFunctionReturn(0); 11652d61bbb3SSatish Balay } 11662d61bbb3SSatish Balay 11674a2ae208SSatish Balay #undef __FUNCT__ 11684a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqBAIJ" 1169dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqBAIJ(Mat A,Vec xx,Vec zz) 11702d61bbb3SSatish Balay { 11713447b6efSHong Zhang PetscScalar zero = 0.0; 11726849ba73SBarry Smith PetscErrorCode ierr; 11732d61bbb3SSatish Balay 11742d61bbb3SSatish Balay PetscFunctionBegin; 11752dcb1b2aSMatthew Knepley ierr = VecSet(zz,zero);CHKERRQ(ierr); 11763447b6efSHong Zhang ierr = MatMultTransposeAdd_SeqBAIJ(A,xx,zz,zz);CHKERRQ(ierr); 11772d61bbb3SSatish Balay PetscFunctionReturn(0); 11782d61bbb3SSatish Balay } 11792d61bbb3SSatish Balay 11804a2ae208SSatish Balay #undef __FUNCT__ 11814a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqBAIJ" 1182dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqBAIJ(Mat A,Vec xx,Vec yy,Vec zz) 11832d61bbb3SSatish Balay 11842d61bbb3SSatish Balay { 11852d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1186dcf5cc72SBarry Smith PetscScalar *zb,*x,*z,*xb = 0,x1,x2,x3,x4,x5; 11873f1db9ecSBarry Smith MatScalar *v; 11886849ba73SBarry Smith PetscErrorCode ierr; 1189d0f46423SBarry Smith PetscInt mbs,i,*idx,*ii,rval,bs=A->rmap->bs,j,n,bs2=a->bs2,*ib,*ridx=PETSC_NULL; 11903447b6efSHong Zhang Mat_CompressedRow cprow = a->compressedrow; 11913447b6efSHong Zhang PetscTruth usecprow=cprow.use; 11922d61bbb3SSatish Balay 11932d61bbb3SSatish Balay PetscFunctionBegin; 11946a65c766SHong Zhang if (yy != zz) { ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr); } 11953447b6efSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 11963447b6efSHong Zhang ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 11972d61bbb3SSatish Balay 11982d61bbb3SSatish Balay idx = a->j; 11992d61bbb3SSatish Balay v = a->a; 12003447b6efSHong Zhang if (usecprow){ 12013447b6efSHong Zhang mbs = cprow.nrows; 12023447b6efSHong Zhang ii = cprow.i; 12037b2bb3b9SHong Zhang ridx = cprow.rindex; 12043447b6efSHong Zhang } else { 12053447b6efSHong Zhang mbs=a->mbs; 12062d61bbb3SSatish Balay ii = a->i; 1207f1af5d2fSBarry Smith xb = x; 12083447b6efSHong Zhang } 12092d61bbb3SSatish Balay 12102d61bbb3SSatish Balay switch (bs) { 12112d61bbb3SSatish Balay case 1: 12122d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 12137b2bb3b9SHong Zhang if (usecprow) xb = x + ridx[i]; 1214f1af5d2fSBarry Smith x1 = xb[0]; 12153447b6efSHong Zhang ib = idx + ii[0]; 12163447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 12172d61bbb3SSatish Balay for (j=0; j<n; j++) { 12182d61bbb3SSatish Balay rval = ib[j]; 1219f1af5d2fSBarry Smith z[rval] += *v * x1; 1220f1af5d2fSBarry Smith v++; 12212d61bbb3SSatish Balay } 12223447b6efSHong Zhang if (!usecprow) xb++; 12232d61bbb3SSatish Balay } 12242d61bbb3SSatish Balay break; 12252d61bbb3SSatish Balay case 2: 12262d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 12277b2bb3b9SHong Zhang if (usecprow) xb = x + 2*ridx[i]; 1228f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; 12293447b6efSHong Zhang ib = idx + ii[0]; 12303447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 12312d61bbb3SSatish Balay for (j=0; j<n; j++) { 12322d61bbb3SSatish Balay rval = ib[j]*2; 12332d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2; 12342d61bbb3SSatish Balay z[rval++] += v[2]*x1 + v[3]*x2; 12352d61bbb3SSatish Balay v += 4; 12362d61bbb3SSatish Balay } 12373447b6efSHong Zhang if (!usecprow) xb += 2; 12382d61bbb3SSatish Balay } 12392d61bbb3SSatish Balay break; 12402d61bbb3SSatish Balay case 3: 12412d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 12427b2bb3b9SHong Zhang if (usecprow) xb = x + 3*ridx[i]; 1243f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 12443447b6efSHong Zhang ib = idx + ii[0]; 12453447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 12462d61bbb3SSatish Balay for (j=0; j<n; j++) { 12472d61bbb3SSatish Balay rval = ib[j]*3; 12482d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3; 12492d61bbb3SSatish Balay z[rval++] += v[3]*x1 + v[4]*x2 + v[5]*x3; 12502d61bbb3SSatish Balay z[rval++] += v[6]*x1 + v[7]*x2 + v[8]*x3; 12512d61bbb3SSatish Balay v += 9; 12522d61bbb3SSatish Balay } 12533447b6efSHong Zhang if (!usecprow) xb += 3; 12542d61bbb3SSatish Balay } 12552d61bbb3SSatish Balay break; 12562d61bbb3SSatish Balay case 4: 12572d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 12587b2bb3b9SHong Zhang if (usecprow) xb = x + 4*ridx[i]; 1259f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 12603447b6efSHong Zhang ib = idx + ii[0]; 12613447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 12622d61bbb3SSatish Balay for (j=0; j<n; j++) { 12632d61bbb3SSatish Balay rval = ib[j]*4; 12642d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4; 12652d61bbb3SSatish Balay z[rval++] += v[4]*x1 + v[5]*x2 + v[6]*x3 + v[7]*x4; 12662d61bbb3SSatish Balay z[rval++] += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[11]*x4; 12672d61bbb3SSatish Balay z[rval++] += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4; 12682d61bbb3SSatish Balay v += 16; 12692d61bbb3SSatish Balay } 12703447b6efSHong Zhang if (!usecprow) xb += 4; 12712d61bbb3SSatish Balay } 12722d61bbb3SSatish Balay break; 12732d61bbb3SSatish Balay case 5: 12742d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 12757b2bb3b9SHong Zhang if (usecprow) xb = x + 5*ridx[i]; 1276f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 12772d61bbb3SSatish Balay x4 = xb[3]; x5 = xb[4]; 12783447b6efSHong Zhang ib = idx + ii[0]; 12793447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 12802d61bbb3SSatish Balay for (j=0; j<n; j++) { 12812d61bbb3SSatish Balay rval = ib[j]*5; 12822d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4 + v[4]*x5; 12832d61bbb3SSatish Balay z[rval++] += v[5]*x1 + v[6]*x2 + v[7]*x3 + v[8]*x4 + v[9]*x5; 12842d61bbb3SSatish Balay z[rval++] += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4 + v[14]*x5; 12852d61bbb3SSatish Balay z[rval++] += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4 + v[19]*x5; 12862d61bbb3SSatish Balay z[rval++] += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4 + v[24]*x5; 12872d61bbb3SSatish Balay v += 25; 12882d61bbb3SSatish Balay } 12893447b6efSHong Zhang if (!usecprow) xb += 5; 12902d61bbb3SSatish Balay } 12912d61bbb3SSatish Balay break; 1292f1af5d2fSBarry Smith default: { /* block sizes larger then 5 by 5 are handled by BLAS */ 1293690b6cddSBarry Smith PetscInt ncols,k; 12943447b6efSHong Zhang PetscScalar *work,*workt,*xtmp; 12953f1db9ecSBarry Smith 12962d61bbb3SSatish Balay if (!a->mult_work) { 1297d0f46423SBarry Smith k = PetscMax(A->rmap->n,A->cmap->n); 129887828ca2SBarry Smith ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr); 12992d61bbb3SSatish Balay } 13002d61bbb3SSatish Balay work = a->mult_work; 13013447b6efSHong Zhang xtmp = x; 13022d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 13032d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 13042d61bbb3SSatish Balay ncols = n*bs; 130587828ca2SBarry Smith ierr = PetscMemzero(work,ncols*sizeof(PetscScalar));CHKERRQ(ierr); 13063447b6efSHong Zhang if (usecprow) { 13077b2bb3b9SHong Zhang xtmp = x + bs*ridx[i]; 13083447b6efSHong Zhang } 13093447b6efSHong Zhang Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,xtmp,v,work); 131071044d3cSBarry Smith /* BLASgemv_("T",&bs,&ncols,&_DOne,v,&bs,xtmp,&_One,&_DOne,work,&_One); */ 13112d61bbb3SSatish Balay v += n*bs2; 13123447b6efSHong Zhang if (!usecprow) xtmp += bs; 13132d61bbb3SSatish Balay workt = work; 13142d61bbb3SSatish Balay for (j=0; j<n; j++) { 13152d61bbb3SSatish Balay zb = z + bs*(*idx++); 13162d61bbb3SSatish Balay for (k=0; k<bs; k++) zb[k] += workt[k] ; 13172d61bbb3SSatish Balay workt += bs; 13182d61bbb3SSatish Balay } 13192d61bbb3SSatish Balay } 13202d61bbb3SSatish Balay } 13212d61bbb3SSatish Balay } 13223447b6efSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 13233447b6efSHong Zhang ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 1324dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz*a->bs2);CHKERRQ(ierr); 13252d61bbb3SSatish Balay PetscFunctionReturn(0); 13262d61bbb3SSatish Balay } 13272d61bbb3SSatish Balay 13284a2ae208SSatish Balay #undef __FUNCT__ 13294a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqBAIJ" 1330f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqBAIJ(Mat inA,PetscScalar alpha) 13312d61bbb3SSatish Balay { 13322d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)inA->data; 1333690b6cddSBarry Smith PetscInt totalnz = a->bs2*a->nz; 1334f4df32b1SMatthew Knepley PetscScalar oalpha = alpha; 1335efee365bSSatish Balay PetscErrorCode ierr; 13360805154bSBarry Smith PetscBLASInt one = 1,tnz = PetscBLASIntCast(totalnz); 13372d61bbb3SSatish Balay 13382d61bbb3SSatish Balay PetscFunctionBegin; 1339f4df32b1SMatthew Knepley BLASscal_(&tnz,&oalpha,a->a,&one); 1340efee365bSSatish Balay ierr = PetscLogFlops(totalnz);CHKERRQ(ierr); 13412d61bbb3SSatish Balay PetscFunctionReturn(0); 13422d61bbb3SSatish Balay } 13432d61bbb3SSatish Balay 13444a2ae208SSatish Balay #undef __FUNCT__ 13454a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqBAIJ" 1346dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqBAIJ(Mat A,NormType type,PetscReal *norm) 13472d61bbb3SSatish Balay { 13488a62d963SHong Zhang PetscErrorCode ierr; 13492d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 13503f1db9ecSBarry Smith MatScalar *v = a->a; 1351329f5518SBarry Smith PetscReal sum = 0.0; 1352d0f46423SBarry Smith PetscInt i,j,k,bs=A->rmap->bs,nz=a->nz,bs2=a->bs2,k1; 13532d61bbb3SSatish Balay 13542d61bbb3SSatish Balay PetscFunctionBegin; 13552d61bbb3SSatish Balay if (type == NORM_FROBENIUS) { 13562d61bbb3SSatish Balay for (i=0; i< bs2*nz; i++) { 1357aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 1358329f5518SBarry Smith sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 13592d61bbb3SSatish Balay #else 13602d61bbb3SSatish Balay sum += (*v)*(*v); v++; 13612d61bbb3SSatish Balay #endif 13622d61bbb3SSatish Balay } 13632d61bbb3SSatish Balay *norm = sqrt(sum); 13648a62d963SHong Zhang } else if (type == NORM_1) { /* maximum column sum */ 13658a62d963SHong Zhang PetscReal *tmp; 13668a62d963SHong Zhang PetscInt *bcol = a->j; 1367d0f46423SBarry Smith ierr = PetscMalloc((A->cmap->n+1)*sizeof(PetscReal),&tmp);CHKERRQ(ierr); 1368d0f46423SBarry Smith ierr = PetscMemzero(tmp,A->cmap->n*sizeof(PetscReal));CHKERRQ(ierr); 13698a62d963SHong Zhang for (i=0; i<nz; i++){ 13708a62d963SHong Zhang for (j=0; j<bs; j++){ 13718a62d963SHong Zhang k1 = bs*(*bcol) + j; /* column index */ 13728a62d963SHong Zhang for (k=0; k<bs; k++){ 13738a62d963SHong Zhang tmp[k1] += PetscAbsScalar(*v); v++; 13748a62d963SHong Zhang } 13758a62d963SHong Zhang } 13768a62d963SHong Zhang bcol++; 13778a62d963SHong Zhang } 13788a62d963SHong Zhang *norm = 0.0; 1379d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 13808a62d963SHong Zhang if (tmp[j] > *norm) *norm = tmp[j]; 13818a62d963SHong Zhang } 13828a62d963SHong Zhang ierr = PetscFree(tmp);CHKERRQ(ierr); 1383596552b5SBarry Smith } else if (type == NORM_INFINITY) { /* maximum row sum */ 1384596552b5SBarry Smith *norm = 0.0; 1385596552b5SBarry Smith for (k=0; k<bs; k++) { 138674f84c7bSSatish Balay for (j=0; j<a->mbs; j++) { 1387596552b5SBarry Smith v = a->a + bs2*a->i[j] + k; 1388596552b5SBarry Smith sum = 0.0; 1389596552b5SBarry Smith for (i=0; i<a->i[j+1]-a->i[j]; i++) { 13900e90e235SBarry Smith for (k1=0; k1<bs; k1++){ 1391596552b5SBarry Smith sum += PetscAbsScalar(*v); 1392596552b5SBarry Smith v += bs; 13932d61bbb3SSatish Balay } 13940e90e235SBarry Smith } 1395596552b5SBarry Smith if (sum > *norm) *norm = sum; 1396596552b5SBarry Smith } 1397596552b5SBarry Smith } 1398596552b5SBarry Smith } else { 139929bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"No support for this norm yet"); 14002d61bbb3SSatish Balay } 14012d61bbb3SSatish Balay PetscFunctionReturn(0); 14022d61bbb3SSatish Balay } 14032d61bbb3SSatish Balay 14042d61bbb3SSatish Balay 14054a2ae208SSatish Balay #undef __FUNCT__ 14064a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqBAIJ" 1407dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqBAIJ(Mat A,Mat B,PetscTruth* flg) 14082d61bbb3SSatish Balay { 14092d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ *)A->data,*b = (Mat_SeqBAIJ *)B->data; 1410dfbe8321SBarry Smith PetscErrorCode ierr; 14112d61bbb3SSatish Balay 14122d61bbb3SSatish Balay PetscFunctionBegin; 14132d61bbb3SSatish Balay /* If the matrix/block dimensions are not equal, or no of nonzeros or shift */ 1414d0f46423SBarry Smith if ((A->rmap->N != B->rmap->N) || (A->cmap->n != B->cmap->n) || (A->rmap->bs != B->rmap->bs)|| (a->nz != b->nz)) { 1415273d9f13SBarry Smith *flg = PETSC_FALSE; 1416273d9f13SBarry Smith PetscFunctionReturn(0); 14172d61bbb3SSatish Balay } 14182d61bbb3SSatish Balay 14192d61bbb3SSatish Balay /* if the a->i are the same */ 1420690b6cddSBarry Smith ierr = PetscMemcmp(a->i,b->i,(a->mbs+1)*sizeof(PetscInt),flg);CHKERRQ(ierr); 1421abc0a331SBarry Smith if (!*flg) { 14220f5bd95cSBarry Smith PetscFunctionReturn(0); 14232d61bbb3SSatish Balay } 14242d61bbb3SSatish Balay 14252d61bbb3SSatish Balay /* if a->j are the same */ 1426690b6cddSBarry Smith ierr = PetscMemcmp(a->j,b->j,(a->nz)*sizeof(PetscInt),flg);CHKERRQ(ierr); 1427abc0a331SBarry Smith if (!*flg) { 14280f5bd95cSBarry Smith PetscFunctionReturn(0); 14292d61bbb3SSatish Balay } 14302d61bbb3SSatish Balay /* if a->a are the same */ 1431d0f46423SBarry Smith ierr = PetscMemcmp(a->a,b->a,(a->nz)*(A->rmap->bs)*(B->rmap->bs)*sizeof(PetscScalar),flg);CHKERRQ(ierr); 14322d61bbb3SSatish Balay PetscFunctionReturn(0); 14332d61bbb3SSatish Balay 14342d61bbb3SSatish Balay } 14352d61bbb3SSatish Balay 14364a2ae208SSatish Balay #undef __FUNCT__ 14374a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqBAIJ" 1438dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqBAIJ(Mat A,Vec v) 14392d61bbb3SSatish Balay { 14402d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1441dfbe8321SBarry Smith PetscErrorCode ierr; 1442690b6cddSBarry Smith PetscInt i,j,k,n,row,bs,*ai,*aj,ambs,bs2; 144387828ca2SBarry Smith PetscScalar *x,zero = 0.0; 14443f1db9ecSBarry Smith MatScalar *aa,*aa_j; 14452d61bbb3SSatish Balay 14462d61bbb3SSatish Balay PetscFunctionBegin; 144729bbc08cSBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 1448d0f46423SBarry Smith bs = A->rmap->bs; 14492d61bbb3SSatish Balay aa = a->a; 14502d61bbb3SSatish Balay ai = a->i; 14512d61bbb3SSatish Balay aj = a->j; 14522d61bbb3SSatish Balay ambs = a->mbs; 14532d61bbb3SSatish Balay bs2 = a->bs2; 14542d61bbb3SSatish Balay 14552dcb1b2aSMatthew Knepley ierr = VecSet(v,zero);CHKERRQ(ierr); 14561ebc52fbSHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1457e1311b90SBarry Smith ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 1458d0f46423SBarry Smith if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 14592d61bbb3SSatish Balay for (i=0; i<ambs; i++) { 14602d61bbb3SSatish Balay for (j=ai[i]; j<ai[i+1]; j++) { 14612d61bbb3SSatish Balay if (aj[j] == i) { 14622d61bbb3SSatish Balay row = i*bs; 14632d61bbb3SSatish Balay aa_j = aa+j*bs2; 14642d61bbb3SSatish Balay for (k=0; k<bs2; k+=(bs+1),row++) x[row] = aa_j[k]; 14652d61bbb3SSatish Balay break; 14662d61bbb3SSatish Balay } 14672d61bbb3SSatish Balay } 14682d61bbb3SSatish Balay } 14691ebc52fbSHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 14702d61bbb3SSatish Balay PetscFunctionReturn(0); 14712d61bbb3SSatish Balay } 14722d61bbb3SSatish Balay 14734a2ae208SSatish Balay #undef __FUNCT__ 14744a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqBAIJ" 1475dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqBAIJ(Mat A,Vec ll,Vec rr) 14762d61bbb3SSatish Balay { 14772d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 147887828ca2SBarry Smith PetscScalar *l,*r,x,*li,*ri; 14793f1db9ecSBarry Smith MatScalar *aa,*v; 1480dfbe8321SBarry Smith PetscErrorCode ierr; 1481690b6cddSBarry Smith PetscInt i,j,k,lm,rn,M,m,n,*ai,*aj,mbs,tmp,bs,bs2; 14822d61bbb3SSatish Balay 14832d61bbb3SSatish Balay PetscFunctionBegin; 14842d61bbb3SSatish Balay ai = a->i; 14852d61bbb3SSatish Balay aj = a->j; 14862d61bbb3SSatish Balay aa = a->a; 1487d0f46423SBarry Smith m = A->rmap->n; 1488d0f46423SBarry Smith n = A->cmap->n; 1489d0f46423SBarry Smith bs = A->rmap->bs; 14902d61bbb3SSatish Balay mbs = a->mbs; 14912d61bbb3SSatish Balay bs2 = a->bs2; 14922d61bbb3SSatish Balay if (ll) { 14931ebc52fbSHong Zhang ierr = VecGetArray(ll,&l);CHKERRQ(ierr); 1494e1311b90SBarry Smith ierr = VecGetLocalSize(ll,&lm);CHKERRQ(ierr); 149529bbc08cSBarry Smith if (lm != m) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length"); 14962d61bbb3SSatish Balay for (i=0; i<mbs; i++) { /* for each block row */ 14972d61bbb3SSatish Balay M = ai[i+1] - ai[i]; 14982d61bbb3SSatish Balay li = l + i*bs; 14992d61bbb3SSatish Balay v = aa + bs2*ai[i]; 15002d61bbb3SSatish Balay for (j=0; j<M; j++) { /* for each block */ 15012d61bbb3SSatish Balay for (k=0; k<bs2; k++) { 15022d61bbb3SSatish Balay (*v++) *= li[k%bs]; 15032d61bbb3SSatish Balay } 15042d61bbb3SSatish Balay } 15052d61bbb3SSatish Balay } 15061ebc52fbSHong Zhang ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr); 1507efee365bSSatish Balay ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 15082d61bbb3SSatish Balay } 15092d61bbb3SSatish Balay 15102d61bbb3SSatish Balay if (rr) { 15111ebc52fbSHong Zhang ierr = VecGetArray(rr,&r);CHKERRQ(ierr); 1512e1311b90SBarry Smith ierr = VecGetLocalSize(rr,&rn);CHKERRQ(ierr); 151329bbc08cSBarry Smith if (rn != n) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length"); 15142d61bbb3SSatish Balay for (i=0; i<mbs; i++) { /* for each block row */ 15152d61bbb3SSatish Balay M = ai[i+1] - ai[i]; 15162d61bbb3SSatish Balay v = aa + bs2*ai[i]; 15172d61bbb3SSatish Balay for (j=0; j<M; j++) { /* for each block */ 15182d61bbb3SSatish Balay ri = r + bs*aj[ai[i]+j]; 15192d61bbb3SSatish Balay for (k=0; k<bs; k++) { 15202d61bbb3SSatish Balay x = ri[k]; 15212d61bbb3SSatish Balay for (tmp=0; tmp<bs; tmp++) (*v++) *= x; 15222d61bbb3SSatish Balay } 15232d61bbb3SSatish Balay } 15242d61bbb3SSatish Balay } 15251ebc52fbSHong Zhang ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr); 1526efee365bSSatish Balay ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 15272d61bbb3SSatish Balay } 15282d61bbb3SSatish Balay PetscFunctionReturn(0); 15292d61bbb3SSatish Balay } 15302d61bbb3SSatish Balay 15312d61bbb3SSatish Balay 15324a2ae208SSatish Balay #undef __FUNCT__ 15334a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqBAIJ" 1534dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqBAIJ(Mat A,MatInfoType flag,MatInfo *info) 15352d61bbb3SSatish Balay { 15362d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 15372d61bbb3SSatish Balay 15382d61bbb3SSatish Balay PetscFunctionBegin; 15392d61bbb3SSatish Balay info->block_size = a->bs2; 15402d61bbb3SSatish Balay info->nz_allocated = a->maxnz; 15412d61bbb3SSatish Balay info->nz_used = a->bs2*a->nz; 15422d61bbb3SSatish Balay info->nz_unneeded = (double)(info->nz_allocated - info->nz_used); 15432d61bbb3SSatish Balay info->assemblies = A->num_ass; 15442d61bbb3SSatish Balay info->mallocs = a->reallocs; 15457adad957SLisandro Dalcin info->memory = ((PetscObject)A)->mem; 15462d61bbb3SSatish Balay if (A->factor) { 15472d61bbb3SSatish Balay info->fill_ratio_given = A->info.fill_ratio_given; 15482d61bbb3SSatish Balay info->fill_ratio_needed = A->info.fill_ratio_needed; 15492d61bbb3SSatish Balay info->factor_mallocs = A->info.factor_mallocs; 15502d61bbb3SSatish Balay } else { 15512d61bbb3SSatish Balay info->fill_ratio_given = 0; 15522d61bbb3SSatish Balay info->fill_ratio_needed = 0; 15532d61bbb3SSatish Balay info->factor_mallocs = 0; 15542d61bbb3SSatish Balay } 15552d61bbb3SSatish Balay PetscFunctionReturn(0); 15562d61bbb3SSatish Balay } 15572d61bbb3SSatish Balay 15582d61bbb3SSatish Balay 15594a2ae208SSatish Balay #undef __FUNCT__ 15604a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqBAIJ" 1561dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqBAIJ(Mat A) 15622d61bbb3SSatish Balay { 15632d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1564dfbe8321SBarry Smith PetscErrorCode ierr; 15652d61bbb3SSatish Balay 15662d61bbb3SSatish Balay PetscFunctionBegin; 1567549d3d68SSatish Balay ierr = PetscMemzero(a->a,a->bs2*a->i[a->mbs]*sizeof(MatScalar));CHKERRQ(ierr); 15682d61bbb3SSatish Balay PetscFunctionReturn(0); 15692d61bbb3SSatish Balay } 1570