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 */ 180fca92195SBarry Smith ierr = PetscMalloc2(a->mbs,PetscInt,&vary,a->mbs,PetscInt,&iary);CHKERRQ(ierr); 181fca92195SBarry Smith ierr = PetscMemzero(vary,a->mbs*sizeof(PetscInt));CHKERRQ(ierr); 182218c64b6SSatish Balay for (i=0; i<nrows; i++) vary[irow[i]/bs]++; 183218c64b6SSatish Balay count = 0; 184218c64b6SSatish Balay for (i=0; i<a->mbs; i++) { 185634064b4SBarry Smith if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_ERR_ARG_SIZ,"Index set does not match blocks"); 186218c64b6SSatish Balay if (vary[i]==bs) iary[count++] = i; 187218c64b6SSatish Balay } 188029af93fSBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is1);CHKERRQ(ierr); 189218c64b6SSatish Balay 190690b6cddSBarry Smith ierr = PetscMemzero(vary,(a->mbs)*sizeof(PetscInt));CHKERRQ(ierr); 191218c64b6SSatish Balay for (i=0; i<ncols; i++) vary[icol[i]/bs]++; 192218c64b6SSatish Balay count = 0; 193218c64b6SSatish Balay for (i=0; i<a->mbs; i++) { 194634064b4SBarry Smith if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_ERR_PLIB,"Internal error in PETSc"); 195218c64b6SSatish Balay if (vary[i]==bs) iary[count++] = i; 196218c64b6SSatish Balay } 197029af93fSBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,&is2);CHKERRQ(ierr); 198218c64b6SSatish Balay ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 199218c64b6SSatish Balay ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 200fca92195SBarry Smith ierr = PetscFree2(vary,iary);CHKERRQ(ierr); 201218c64b6SSatish Balay 2024aa3045dSJed Brown ierr = MatGetSubMatrix_SeqBAIJ_Private(A,is1,is2,scall,B);CHKERRQ(ierr); 203218c64b6SSatish Balay ISDestroy(is1); 204218c64b6SSatish Balay ISDestroy(is2); 2053a40ed3dSBarry Smith PetscFunctionReturn(0); 206218c64b6SSatish Balay } 207218c64b6SSatish Balay 2084a2ae208SSatish Balay #undef __FUNCT__ 2094a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqBAIJ" 210690b6cddSBarry Smith PetscErrorCode MatGetSubMatrices_SeqBAIJ(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[]) 211736121d4SSatish Balay { 2126849ba73SBarry Smith PetscErrorCode ierr; 213690b6cddSBarry Smith PetscInt i; 214736121d4SSatish Balay 2153a40ed3dSBarry Smith PetscFunctionBegin; 216736121d4SSatish Balay if (scall == MAT_INITIAL_MATRIX) { 21782502324SSatish Balay ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr); 218736121d4SSatish Balay } 219736121d4SSatish Balay 220736121d4SSatish Balay for (i=0; i<n; i++) { 2214aa3045dSJed Brown ierr = MatGetSubMatrix_SeqBAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr); 222736121d4SSatish Balay } 2233a40ed3dSBarry Smith PetscFunctionReturn(0); 224736121d4SSatish Balay } 225218c64b6SSatish Balay 226218c64b6SSatish Balay 2272d61bbb3SSatish Balay /* -------------------------------------------------------*/ 2282d61bbb3SSatish Balay /* Should check that shapes of vectors and matrices match */ 2292d61bbb3SSatish Balay /* -------------------------------------------------------*/ 230d9eff348SSatish Balay #include "petscblaslapack.h" 2312d61bbb3SSatish Balay 2324a2ae208SSatish Balay #undef __FUNCT__ 2334a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_1" 234dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_1(Mat A,Vec xx,Vec zz) 2352d61bbb3SSatish Balay { 2362d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 237d9fead3dSBarry Smith PetscScalar *z,sum; 238d9fead3dSBarry Smith const PetscScalar *x; 239d9fead3dSBarry Smith const MatScalar *v; 2406849ba73SBarry Smith PetscErrorCode ierr; 2412162cab8SBarry Smith PetscInt mbs,i,n,nonzerorow=0; 2422162cab8SBarry Smith const PetscInt *idx,*ii,*ridx=PETSC_NULL; 24326e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 2442d61bbb3SSatish Balay 2452d61bbb3SSatish Balay PetscFunctionBegin; 246d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 2471ebc52fbSHong Zhang ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 2482d61bbb3SSatish Balay 24926e093fcSHong Zhang if (usecprow){ 25026e093fcSHong Zhang mbs = a->compressedrow.nrows; 25126e093fcSHong Zhang ii = a->compressedrow.i; 2527b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 253a1c3900fSBarry Smith ierr = PetscMemzero(z,mbs*sizeof(PetscScalar));CHKERRQ(ierr); 25426e093fcSHong Zhang } else { 25526e093fcSHong Zhang mbs = a->mbs; 2562d61bbb3SSatish Balay ii = a->i; 25726e093fcSHong Zhang } 2582d61bbb3SSatish Balay 2592d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 260ee54c7eeSHong Zhang n = ii[1] - ii[0]; 261ee54c7eeSHong Zhang v = a->a + ii[0]; 262ee54c7eeSHong Zhang idx = a->j + ii[0]; 263ee54c7eeSHong Zhang ii++; 2642d61bbb3SSatish Balay sum = 0.0; 2652162cab8SBarry Smith PetscSparseDensePlusDot(sum,x,v,idx,n); 26626e093fcSHong Zhang if (usecprow){ 2677b2bb3b9SHong Zhang z[ridx[i]] = sum; 26826e093fcSHong Zhang } else { 269122f12eaSBarry Smith nonzerorow += (n>0); 2702d61bbb3SSatish Balay z[i] = sum; 2712d61bbb3SSatish Balay } 27226e093fcSHong Zhang } 273d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 2741ebc52fbSHong Zhang ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 275dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr); 2762d61bbb3SSatish Balay PetscFunctionReturn(0); 2772d61bbb3SSatish Balay } 2782d61bbb3SSatish Balay 2794a2ae208SSatish Balay #undef __FUNCT__ 2804a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_2" 281dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_2(Mat A,Vec xx,Vec zz) 2822d61bbb3SSatish Balay { 2832d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 284d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,*zarray; 285d9fead3dSBarry Smith const PetscScalar *x,*xb; 28687828ca2SBarry Smith PetscScalar x1,x2; 287d9fead3dSBarry Smith const MatScalar *v; 288dfbe8321SBarry Smith PetscErrorCode ierr; 28998c9bda7SSatish Balay PetscInt mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 29026e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 2912d61bbb3SSatish Balay 2922d61bbb3SSatish Balay PetscFunctionBegin; 293d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 29426e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 2952d61bbb3SSatish Balay 2962d61bbb3SSatish Balay idx = a->j; 2972d61bbb3SSatish Balay v = a->a; 29826e093fcSHong Zhang if (usecprow){ 29926e093fcSHong Zhang mbs = a->compressedrow.nrows; 30026e093fcSHong Zhang ii = a->compressedrow.i; 3017b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 30226e093fcSHong Zhang } else { 30326e093fcSHong Zhang mbs = a->mbs; 3042d61bbb3SSatish Balay ii = a->i; 30526e093fcSHong Zhang z = zarray; 30626e093fcSHong Zhang } 3072d61bbb3SSatish Balay 3082d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 3092d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 3102d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; 31198c9bda7SSatish Balay nonzerorow += (n>0); 3122d61bbb3SSatish Balay for (j=0; j<n; j++) { 3132d61bbb3SSatish Balay xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1]; 3142d61bbb3SSatish Balay sum1 += v[0]*x1 + v[2]*x2; 3152d61bbb3SSatish Balay sum2 += v[1]*x1 + v[3]*x2; 3162d61bbb3SSatish Balay v += 4; 3172d61bbb3SSatish Balay } 3187b2bb3b9SHong Zhang if (usecprow) z = zarray + 2*ridx[i]; 3192d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; 32026e093fcSHong Zhang if (!usecprow) z += 2; 3212d61bbb3SSatish Balay } 322d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 32326e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 324dc0b31edSSatish Balay ierr = PetscLogFlops(8.0*a->nz - 2.0*nonzerorow);CHKERRQ(ierr); 3252d61bbb3SSatish Balay PetscFunctionReturn(0); 3262d61bbb3SSatish Balay } 3272d61bbb3SSatish Balay 3284a2ae208SSatish Balay #undef __FUNCT__ 3294a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_3" 330dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_3(Mat A,Vec xx,Vec zz) 3312d61bbb3SSatish Balay { 3322d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 333d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,x1,x2,x3,*zarray; 334d9fead3dSBarry Smith const PetscScalar *x,*xb; 335d9fead3dSBarry Smith const MatScalar *v; 336dfbe8321SBarry Smith PetscErrorCode ierr; 337028cd4eaSSatish Balay PetscInt mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 338028cd4eaSSatish Balay PetscTruth usecprow=a->compressedrow.use; 33926e093fcSHong Zhang 3402d61bbb3SSatish Balay 341b6410449SSatish Balay #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 342fee21e36SBarry Smith #pragma disjoint(*v,*z,*xb) 343fee21e36SBarry Smith #endif 344fee21e36SBarry Smith 3452d61bbb3SSatish Balay PetscFunctionBegin; 346d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 34726e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 3482d61bbb3SSatish Balay 3492d61bbb3SSatish Balay idx = a->j; 3502d61bbb3SSatish Balay v = a->a; 35126e093fcSHong Zhang if (usecprow){ 35226e093fcSHong Zhang mbs = a->compressedrow.nrows; 35326e093fcSHong Zhang ii = a->compressedrow.i; 3547b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 35526e093fcSHong Zhang } else { 35626e093fcSHong Zhang mbs = a->mbs; 3572d61bbb3SSatish Balay ii = a->i; 35826e093fcSHong Zhang z = zarray; 35926e093fcSHong Zhang } 3602d61bbb3SSatish Balay 3612d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 3622d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 3632d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; 36498c9bda7SSatish Balay nonzerorow += (n>0); 3652d61bbb3SSatish Balay for (j=0; j<n; j++) { 3662d61bbb3SSatish Balay xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 3672d61bbb3SSatish Balay sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3; 3682d61bbb3SSatish Balay sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3; 3692d61bbb3SSatish Balay sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3; 3702d61bbb3SSatish Balay v += 9; 3712d61bbb3SSatish Balay } 3727b2bb3b9SHong Zhang if (usecprow) z = zarray + 3*ridx[i]; 3732d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; 37426e093fcSHong Zhang if (!usecprow) z += 3; 3752d61bbb3SSatish Balay } 376d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 37726e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 378dc0b31edSSatish Balay ierr = PetscLogFlops(18.0*a->nz - 3.0*nonzerorow);CHKERRQ(ierr); 3792d61bbb3SSatish Balay PetscFunctionReturn(0); 3802d61bbb3SSatish Balay } 3812d61bbb3SSatish Balay 3824a2ae208SSatish Balay #undef __FUNCT__ 3834a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_4" 384dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_4(Mat A,Vec xx,Vec zz) 3852d61bbb3SSatish Balay { 3862d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 387d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*zarray; 388d9fead3dSBarry Smith const PetscScalar *x,*xb; 389d9fead3dSBarry Smith const MatScalar *v; 390dfbe8321SBarry Smith PetscErrorCode ierr; 39198c9bda7SSatish Balay PetscInt mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 39226e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 3932d61bbb3SSatish Balay 3942d61bbb3SSatish Balay PetscFunctionBegin; 395d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 39626e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 3972d61bbb3SSatish Balay 3982d61bbb3SSatish Balay idx = a->j; 3992d61bbb3SSatish Balay v = a->a; 40026e093fcSHong Zhang if (usecprow){ 40126e093fcSHong Zhang mbs = a->compressedrow.nrows; 40226e093fcSHong Zhang ii = a->compressedrow.i; 4037b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 40426e093fcSHong Zhang } else { 40526e093fcSHong Zhang mbs = a->mbs; 4062d61bbb3SSatish Balay ii = a->i; 40726e093fcSHong Zhang z = zarray; 40826e093fcSHong Zhang } 4092d61bbb3SSatish Balay 4102d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 4112d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 4122d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; 41398c9bda7SSatish Balay nonzerorow += (n>0); 4142d61bbb3SSatish Balay for (j=0; j<n; j++) { 4152d61bbb3SSatish Balay xb = x + 4*(*idx++); 4162d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 4172d61bbb3SSatish Balay sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3 + v[12]*x4; 4182d61bbb3SSatish Balay sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3 + v[13]*x4; 4192d61bbb3SSatish Balay sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3 + v[14]*x4; 4202d61bbb3SSatish Balay sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3 + v[15]*x4; 4212d61bbb3SSatish Balay v += 16; 4222d61bbb3SSatish Balay } 4237b2bb3b9SHong Zhang if (usecprow) z = zarray + 4*ridx[i]; 4242d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; 42526e093fcSHong Zhang if (!usecprow) z += 4; 4262d61bbb3SSatish Balay } 427d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 42826e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 429dc0b31edSSatish Balay ierr = PetscLogFlops(32.0*a->nz - 4.0*nonzerorow);CHKERRQ(ierr); 4302d61bbb3SSatish Balay PetscFunctionReturn(0); 4312d61bbb3SSatish Balay } 4322d61bbb3SSatish Balay 4334a2ae208SSatish Balay #undef __FUNCT__ 4344a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_5" 435dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_5(Mat A,Vec xx,Vec zz) 4362d61bbb3SSatish Balay { 4372d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 438d9fead3dSBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5,*z = 0,*zarray; 439d9fead3dSBarry Smith const PetscScalar *xb,*x; 440d9fead3dSBarry Smith const MatScalar *v; 441dfbe8321SBarry Smith PetscErrorCode ierr; 442766f9fbaSBarry Smith const PetscInt *idx,*ii,*ridx=PETSC_NULL; 443766f9fbaSBarry Smith PetscInt mbs,i,j,n,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 } 541*8ab949d8SShri Abhyankar 5424a2ae208SSatish Balay #undef __FUNCT__ 5434a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_7" 544dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_7(Mat A,Vec xx,Vec zz) 5452d61bbb3SSatish Balay { 5462d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 547d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7; 548d9fead3dSBarry Smith const PetscScalar *x,*xb; 54926e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,x7,*zarray; 550d9fead3dSBarry Smith const MatScalar *v; 551dfbe8321SBarry Smith PetscErrorCode ierr; 55298c9bda7SSatish Balay PetscInt mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL,nonzerorow=0; 55326e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 5542d61bbb3SSatish Balay 555433994e6SBarry Smith PetscFunctionBegin; 556d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 55726e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 5582d61bbb3SSatish Balay 5592d61bbb3SSatish Balay idx = a->j; 5602d61bbb3SSatish Balay v = a->a; 56126e093fcSHong Zhang if (usecprow){ 56226e093fcSHong Zhang mbs = a->compressedrow.nrows; 56326e093fcSHong Zhang ii = a->compressedrow.i; 5647b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 56526e093fcSHong Zhang } else { 56626e093fcSHong Zhang mbs = a->mbs; 5672d61bbb3SSatish Balay ii = a->i; 56826e093fcSHong Zhang z = zarray; 56926e093fcSHong Zhang } 5702d61bbb3SSatish Balay 5712d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 5722d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 5732d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 57498c9bda7SSatish Balay nonzerorow += (n>0); 5752d61bbb3SSatish Balay for (j=0; j<n; j++) { 5762d61bbb3SSatish Balay xb = x + 7*(*idx++); 5772d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 5782d61bbb3SSatish Balay sum1 += v[0]*x1 + v[7]*x2 + v[14]*x3 + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7; 5792d61bbb3SSatish Balay sum2 += v[1]*x1 + v[8]*x2 + v[15]*x3 + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7; 5802d61bbb3SSatish Balay sum3 += v[2]*x1 + v[9]*x2 + v[16]*x3 + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7; 5812d61bbb3SSatish Balay sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3 + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7; 5822d61bbb3SSatish Balay sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3 + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7; 5832d61bbb3SSatish Balay sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3 + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7; 5842d61bbb3SSatish Balay sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3 + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7; 5852d61bbb3SSatish Balay v += 49; 5862d61bbb3SSatish Balay } 5877b2bb3b9SHong Zhang if (usecprow) z = zarray + 7*ridx[i]; 5882d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 58926e093fcSHong Zhang if (!usecprow) z += 7; 5902d61bbb3SSatish Balay } 5912d61bbb3SSatish Balay 592d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 59326e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 594dc0b31edSSatish Balay ierr = PetscLogFlops(98.0*a->nz - 7.0*nonzerorow);CHKERRQ(ierr); 5952d61bbb3SSatish Balay PetscFunctionReturn(0); 5962d61bbb3SSatish Balay } 5972d61bbb3SSatish Balay 598*8ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15 version 1: Columns in the block are accessed one at a time */ 599*8ab949d8SShri Abhyankar 600*8ab949d8SShri Abhyankar #undef __FUNCT__ 601*8ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver1" 602*8ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver1(Mat A,Vec xx,Vec zz) 603*8ab949d8SShri Abhyankar { 604*8ab949d8SShri Abhyankar Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 605*8ab949d8SShri Abhyankar PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15; 606*8ab949d8SShri Abhyankar const PetscScalar *x,*xb; 607*8ab949d8SShri Abhyankar PetscScalar x1,x2,x3,x4,x5,x6,x7,x8,x9,x10,x11,x12,x13,x14,x15,*zarray; 608*8ab949d8SShri Abhyankar const MatScalar *v; 609*8ab949d8SShri Abhyankar PetscErrorCode ierr; 610*8ab949d8SShri Abhyankar const PetscInt *ii,*ij=a->j,*idx; 611*8ab949d8SShri Abhyankar PetscInt mbs,i,j,k,n,*ridx=PETSC_NULL,nonzerorow=0; 612*8ab949d8SShri Abhyankar PetscTruth usecprow=a->compressedrow.use; 613*8ab949d8SShri Abhyankar 614*8ab949d8SShri Abhyankar PetscFunctionBegin; 615*8ab949d8SShri Abhyankar ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 616*8ab949d8SShri Abhyankar ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 617*8ab949d8SShri Abhyankar 618*8ab949d8SShri Abhyankar v = a->a; 619*8ab949d8SShri Abhyankar if (usecprow){ 620*8ab949d8SShri Abhyankar mbs = a->compressedrow.nrows; 621*8ab949d8SShri Abhyankar ii = a->compressedrow.i; 622*8ab949d8SShri Abhyankar ridx = a->compressedrow.rindex; 623*8ab949d8SShri Abhyankar } else { 624*8ab949d8SShri Abhyankar mbs = a->mbs; 625*8ab949d8SShri Abhyankar ii = a->i; 626*8ab949d8SShri Abhyankar z = zarray; 627*8ab949d8SShri Abhyankar } 628*8ab949d8SShri Abhyankar 629*8ab949d8SShri Abhyankar for (i=0; i<mbs; i++) { 630*8ab949d8SShri Abhyankar n = ii[i+1] - ii[i]; 631*8ab949d8SShri Abhyankar idx = ij + ii[i]; 632*8ab949d8SShri Abhyankar sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 633*8ab949d8SShri Abhyankar sum8 = 0.0; sum9 = 0.0; sum10 = 0.0; sum11 = 0.0; sum12 = 0.0; sum13 = 0.0; sum14 = 0.0;sum15 = 0.0; 634*8ab949d8SShri Abhyankar 635*8ab949d8SShri Abhyankar nonzerorow += (n>0); 636*8ab949d8SShri Abhyankar for (j=0; j<n; j++) { 637*8ab949d8SShri Abhyankar xb = x + 15*(idx[j]); 638*8ab949d8SShri Abhyankar x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 639*8ab949d8SShri Abhyankar x8 = xb[7]; x9 = xb[8]; x10 = xb[9]; x11 = xb[10]; x12 = xb[11]; x13 = xb[12]; x14 = xb[13];x15 = xb[14]; 640*8ab949d8SShri Abhyankar 641*8ab949d8SShri Abhyankar for(k=0;k<15;k++){ 642*8ab949d8SShri Abhyankar sum1 += v[0]*xb[k]; 643*8ab949d8SShri Abhyankar sum2 += v[1]*xb[k]; 644*8ab949d8SShri Abhyankar sum3 += v[2]*xb[k]; 645*8ab949d8SShri Abhyankar sum4 += v[3]*xb[k]; 646*8ab949d8SShri Abhyankar sum5 += v[4]*xb[k]; 647*8ab949d8SShri Abhyankar sum6 += v[5]*xb[k]; 648*8ab949d8SShri Abhyankar sum7 += v[6]*xb[k]; 649*8ab949d8SShri Abhyankar sum8 += v[7]*xb[k]; 650*8ab949d8SShri Abhyankar sum9 += v[8]*xb[k]; 651*8ab949d8SShri Abhyankar sum10 += v[9]*xb[k]; 652*8ab949d8SShri Abhyankar sum11 += v[10]*xb[k]; 653*8ab949d8SShri Abhyankar sum12 += v[11]*xb[k]; 654*8ab949d8SShri Abhyankar sum13 += v[12]*xb[k]; 655*8ab949d8SShri Abhyankar sum14 += v[13]*xb[k]; 656*8ab949d8SShri Abhyankar sum15 += v[14]*xb[k]; 657*8ab949d8SShri Abhyankar v += 15; 658*8ab949d8SShri Abhyankar } 659*8ab949d8SShri Abhyankar } 660*8ab949d8SShri Abhyankar if (usecprow) z = zarray + 15*ridx[i]; 661*8ab949d8SShri Abhyankar z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 662*8ab949d8SShri Abhyankar z[7] = sum8; z[8] = sum9; z[9] = sum10; z[10] = sum11; z[11] = sum12; z[12] = sum13; z[13] = sum14;z[14] = sum15; 663*8ab949d8SShri Abhyankar 664*8ab949d8SShri Abhyankar if (!usecprow) z += 15; 665*8ab949d8SShri Abhyankar } 666*8ab949d8SShri Abhyankar 667*8ab949d8SShri Abhyankar ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 668*8ab949d8SShri Abhyankar ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 669*8ab949d8SShri Abhyankar ierr = PetscLogFlops(450.0*a->nz - 15.0*nonzerorow);CHKERRQ(ierr); 670*8ab949d8SShri Abhyankar PetscFunctionReturn(0); 671*8ab949d8SShri Abhyankar } 672*8ab949d8SShri Abhyankar 673*8ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15_ver2 : Columns in the block are accessed in sets of 4,4,4,3 */ 674*8ab949d8SShri Abhyankar #undef __FUNCT__ 675*8ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver2" 676*8ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver2(Mat A,Vec xx,Vec zz) 677*8ab949d8SShri Abhyankar { 678*8ab949d8SShri Abhyankar Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 679*8ab949d8SShri Abhyankar PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15; 680*8ab949d8SShri Abhyankar const PetscScalar *x,*xb; 681*8ab949d8SShri Abhyankar PetscScalar x1,x2,x3,x4,x5,x6,x7,x8,x9,x10,x11,x12,x13,x14,x15,*zarray; 682*8ab949d8SShri Abhyankar const MatScalar *v; 683*8ab949d8SShri Abhyankar PetscErrorCode ierr; 684*8ab949d8SShri Abhyankar const PetscInt *ii,*ij=a->j,*idx; 685*8ab949d8SShri Abhyankar PetscInt mbs,i,j,n,*ridx=PETSC_NULL,nonzerorow=0; 686*8ab949d8SShri Abhyankar PetscTruth usecprow=a->compressedrow.use; 687*8ab949d8SShri Abhyankar 688*8ab949d8SShri Abhyankar PetscFunctionBegin; 689*8ab949d8SShri Abhyankar ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 690*8ab949d8SShri Abhyankar ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 691*8ab949d8SShri Abhyankar 692*8ab949d8SShri Abhyankar v = a->a; 693*8ab949d8SShri Abhyankar if (usecprow){ 694*8ab949d8SShri Abhyankar mbs = a->compressedrow.nrows; 695*8ab949d8SShri Abhyankar ii = a->compressedrow.i; 696*8ab949d8SShri Abhyankar ridx = a->compressedrow.rindex; 697*8ab949d8SShri Abhyankar } else { 698*8ab949d8SShri Abhyankar mbs = a->mbs; 699*8ab949d8SShri Abhyankar ii = a->i; 700*8ab949d8SShri Abhyankar z = zarray; 701*8ab949d8SShri Abhyankar } 702*8ab949d8SShri Abhyankar 703*8ab949d8SShri Abhyankar for (i=0; i<mbs; i++) { 704*8ab949d8SShri Abhyankar n = ii[i+1] - ii[i]; 705*8ab949d8SShri Abhyankar idx = ij + ii[i]; 706*8ab949d8SShri Abhyankar sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 707*8ab949d8SShri Abhyankar sum8 = 0.0; sum9 = 0.0; sum10 = 0.0; sum11 = 0.0; sum12 = 0.0; sum13 = 0.0; sum14 = 0.0;sum15 = 0.0; 708*8ab949d8SShri Abhyankar 709*8ab949d8SShri Abhyankar nonzerorow += (n>0); 710*8ab949d8SShri Abhyankar for (j=0; j<n; j++) { 711*8ab949d8SShri Abhyankar xb = x + 15*(idx[j]); 712*8ab949d8SShri Abhyankar x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 713*8ab949d8SShri Abhyankar x8 = xb[7]; x9 = xb[8]; x10 = xb[9]; x11 = xb[10]; x12 = xb[11]; x13 = xb[12]; x14 = xb[13];x15 = xb[14]; 714*8ab949d8SShri Abhyankar 715*8ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4; 716*8ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4; 717*8ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4; 718*8ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4; 719*8ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4; 720*8ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4; 721*8ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4; 722*8ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4; 723*8ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4; 724*8ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4; 725*8ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4; 726*8ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4; 727*8ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4; 728*8ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4; 729*8ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4; 730*8ab949d8SShri Abhyankar 731*8ab949d8SShri Abhyankar v += 60; 732*8ab949d8SShri Abhyankar 733*8ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4; 734*8ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4; 735*8ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4; 736*8ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4; 737*8ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4; 738*8ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4; 739*8ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4; 740*8ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4; 741*8ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4; 742*8ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4; 743*8ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4; 744*8ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4; 745*8ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4; 746*8ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4; 747*8ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4; 748*8ab949d8SShri Abhyankar v += 60; 749*8ab949d8SShri Abhyankar 750*8ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4; 751*8ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4; 752*8ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4; 753*8ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4; 754*8ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4; 755*8ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4; 756*8ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4; 757*8ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4; 758*8ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4; 759*8ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4; 760*8ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4; 761*8ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4; 762*8ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4; 763*8ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4; 764*8ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4; 765*8ab949d8SShri Abhyankar v += 60; 766*8ab949d8SShri Abhyankar 767*8ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3; 768*8ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3; 769*8ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3; 770*8ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3; 771*8ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3; 772*8ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3; 773*8ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3; 774*8ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3; 775*8ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3; 776*8ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3; 777*8ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3; 778*8ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3; 779*8ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3; 780*8ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3; 781*8ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3; 782*8ab949d8SShri Abhyankar v += 45; 783*8ab949d8SShri Abhyankar } 784*8ab949d8SShri Abhyankar if (usecprow) z = zarray + 15*ridx[i]; 785*8ab949d8SShri Abhyankar z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 786*8ab949d8SShri Abhyankar z[7] = sum8; z[8] = sum9; z[9] = sum10; z[10] = sum11; z[11] = sum12; z[12] = sum13; z[13] = sum14;z[14] = sum15; 787*8ab949d8SShri Abhyankar 788*8ab949d8SShri Abhyankar if (!usecprow) z += 15; 789*8ab949d8SShri Abhyankar } 790*8ab949d8SShri Abhyankar 791*8ab949d8SShri Abhyankar ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 792*8ab949d8SShri Abhyankar ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 793*8ab949d8SShri Abhyankar ierr = PetscLogFlops(450.0*a->nz - 15.0*nonzerorow);CHKERRQ(ierr); 794*8ab949d8SShri Abhyankar PetscFunctionReturn(0); 795*8ab949d8SShri Abhyankar } 796*8ab949d8SShri Abhyankar 797*8ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15_ver3 : Columns in the block are accessed in sets of 8,7 */ 798*8ab949d8SShri Abhyankar #undef __FUNCT__ 799*8ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver3" 800*8ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver3(Mat A,Vec xx,Vec zz) 801*8ab949d8SShri Abhyankar { 802*8ab949d8SShri Abhyankar Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 803*8ab949d8SShri Abhyankar PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15; 804*8ab949d8SShri Abhyankar const PetscScalar *x,*xb; 805*8ab949d8SShri Abhyankar PetscScalar x1,x2,x3,x4,x5,x6,x7,x8,x9,x10,x11,x12,x13,x14,x15,*zarray; 806*8ab949d8SShri Abhyankar const MatScalar *v; 807*8ab949d8SShri Abhyankar PetscErrorCode ierr; 808*8ab949d8SShri Abhyankar const PetscInt *ii,*ij=a->j,*idx; 809*8ab949d8SShri Abhyankar PetscInt mbs,i,j,n,*ridx=PETSC_NULL,nonzerorow=0; 810*8ab949d8SShri Abhyankar PetscTruth usecprow=a->compressedrow.use; 811*8ab949d8SShri Abhyankar 812*8ab949d8SShri Abhyankar PetscFunctionBegin; 813*8ab949d8SShri Abhyankar ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 814*8ab949d8SShri Abhyankar ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 815*8ab949d8SShri Abhyankar 816*8ab949d8SShri Abhyankar v = a->a; 817*8ab949d8SShri Abhyankar if (usecprow){ 818*8ab949d8SShri Abhyankar mbs = a->compressedrow.nrows; 819*8ab949d8SShri Abhyankar ii = a->compressedrow.i; 820*8ab949d8SShri Abhyankar ridx = a->compressedrow.rindex; 821*8ab949d8SShri Abhyankar } else { 822*8ab949d8SShri Abhyankar mbs = a->mbs; 823*8ab949d8SShri Abhyankar ii = a->i; 824*8ab949d8SShri Abhyankar z = zarray; 825*8ab949d8SShri Abhyankar } 826*8ab949d8SShri Abhyankar 827*8ab949d8SShri Abhyankar for (i=0; i<mbs; i++) { 828*8ab949d8SShri Abhyankar n = ii[i+1] - ii[i]; 829*8ab949d8SShri Abhyankar idx = ij + ii[i]; 830*8ab949d8SShri Abhyankar sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 831*8ab949d8SShri Abhyankar sum8 = 0.0; sum9 = 0.0; sum10 = 0.0; sum11 = 0.0; sum12 = 0.0; sum13 = 0.0; sum14 = 0.0;sum15 = 0.0; 832*8ab949d8SShri Abhyankar 833*8ab949d8SShri Abhyankar nonzerorow += (n>0); 834*8ab949d8SShri Abhyankar for (j=0; j<n; j++) { 835*8ab949d8SShri Abhyankar xb = x + 15*(idx[j]); 836*8ab949d8SShri Abhyankar x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 837*8ab949d8SShri Abhyankar x8 = xb[7]; x9 = xb[8]; x10 = xb[9]; x11 = xb[10]; x12 = xb[11]; x13 = xb[12]; x14 = xb[13];x15 = xb[14]; 838*8ab949d8SShri Abhyankar 839*8ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4 + v[60]*x5 + v[75]*x6 + v[90]*x7 + v[105]*x8; 840*8ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4 + v[61]*x5 + v[76]*x6 + v[91]*x7 + v[106]*x8; 841*8ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4 + v[62]*x5 + v[77]*x6 + v[92]*x7 + v[107]*x8; 842*8ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4 + v[63]*x5 + v[78]*x6 + v[93]*x7 + v[108]*x8; 843*8ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4 + v[64]*x5 + v[79]*x6 + v[94]*x7 + v[109]*x8; 844*8ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4 + v[65]*x5 + v[80]*x6 + v[95]*x7 + v[110]*x8; 845*8ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4 + v[66]*x5 + v[81]*x6 + v[96]*x7 + v[111]*x8; 846*8ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4 + v[67]*x5 + v[82]*x6 + v[97]*x7 + v[112]*x8; 847*8ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4 + v[68]*x5 + v[83]*x6 + v[98]*x7 + v[113]*x8; 848*8ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4 + v[69]*x5 + v[84]*x6 + v[99]*x7 + v[114]*x8; 849*8ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4 + v[70]*x5 + v[85]*x6 + v[100]*x7 + v[115]*x8; 850*8ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4 + v[71]*x5 + v[86]*x6 + v[101]*x7 + v[116]*x8; 851*8ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4 + v[72]*x5 + v[87]*x6 + v[102]*x7 + v[117]*x8; 852*8ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4 + v[73]*x5 + v[88]*x6 + v[103]*x7 + v[118]*x8; 853*8ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4 + v[74]*x5 + v[89]*x6 + v[104]*x7 + v[119]*x8; 854*8ab949d8SShri Abhyankar v += 120; 855*8ab949d8SShri Abhyankar 856*8ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4 + v[60]*x5 + v[75]*x6 + v[90]*x7; 857*8ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4 + v[61]*x5 + v[76]*x6 + v[91]*x7; 858*8ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4 + v[62]*x5 + v[77]*x6 + v[92]*x7; 859*8ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4 + v[63]*x5 + v[78]*x6 + v[93]*x7; 860*8ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4 + v[64]*x5 + v[79]*x6 + v[94]*x7; 861*8ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4 + v[65]*x5 + v[80]*x6 + v[95]*x7; 862*8ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4 + v[66]*x5 + v[81]*x6 + v[96]*x7; 863*8ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4 + v[67]*x5 + v[82]*x6 + v[97]*x7; 864*8ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4 + v[68]*x5 + v[83]*x6 + v[98]*x7; 865*8ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4 + v[69]*x5 + v[84]*x6 + v[99]*x7; 866*8ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4 + v[70]*x5 + v[85]*x6 + v[100]*x7; 867*8ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4 + v[71]*x5 + v[86]*x6 + v[101]*x7; 868*8ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4 + v[72]*x5 + v[87]*x6 + v[102]*x7; 869*8ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4 + v[73]*x5 + v[88]*x6 + v[103]*x7; 870*8ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4 + v[74]*x5 + v[89]*x6 + v[104]*x7; 871*8ab949d8SShri Abhyankar v += 105; 872*8ab949d8SShri Abhyankar } 873*8ab949d8SShri Abhyankar if (usecprow) z = zarray + 15*ridx[i]; 874*8ab949d8SShri Abhyankar z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 875*8ab949d8SShri Abhyankar z[7] = sum8; z[8] = sum9; z[9] = sum10; z[10] = sum11; z[11] = sum12; z[12] = sum13; z[13] = sum14;z[14] = sum15; 876*8ab949d8SShri Abhyankar 877*8ab949d8SShri Abhyankar if (!usecprow) z += 15; 878*8ab949d8SShri Abhyankar } 879*8ab949d8SShri Abhyankar 880*8ab949d8SShri Abhyankar ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 881*8ab949d8SShri Abhyankar ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 882*8ab949d8SShri Abhyankar ierr = PetscLogFlops(450.0*a->nz - 15.0*nonzerorow);CHKERRQ(ierr); 883*8ab949d8SShri Abhyankar PetscFunctionReturn(0); 884*8ab949d8SShri Abhyankar } 885*8ab949d8SShri Abhyankar 886*8ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15_ver4 : All columns in the block are accessed at once */ 887*8ab949d8SShri Abhyankar 888*8ab949d8SShri Abhyankar #undef __FUNCT__ 889*8ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver4" 890*8ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver4(Mat A,Vec xx,Vec zz) 891*8ab949d8SShri Abhyankar { 892*8ab949d8SShri Abhyankar Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 893*8ab949d8SShri Abhyankar PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15; 894*8ab949d8SShri Abhyankar const PetscScalar *x,*xb; 895*8ab949d8SShri Abhyankar PetscScalar x1,x2,x3,x4,x5,x6,x7,x8,x9,x10,x11,x12,x13,x14,x15,*zarray; 896*8ab949d8SShri Abhyankar const MatScalar *v; 897*8ab949d8SShri Abhyankar PetscErrorCode ierr; 898*8ab949d8SShri Abhyankar const PetscInt *ii,*ij=a->j,*idx; 899*8ab949d8SShri Abhyankar PetscInt mbs,i,j,n,*ridx=PETSC_NULL,nonzerorow=0; 900*8ab949d8SShri Abhyankar PetscTruth usecprow=a->compressedrow.use; 901*8ab949d8SShri Abhyankar 902*8ab949d8SShri Abhyankar PetscFunctionBegin; 903*8ab949d8SShri Abhyankar ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 904*8ab949d8SShri Abhyankar ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 905*8ab949d8SShri Abhyankar 906*8ab949d8SShri Abhyankar v = a->a; 907*8ab949d8SShri Abhyankar if (usecprow){ 908*8ab949d8SShri Abhyankar mbs = a->compressedrow.nrows; 909*8ab949d8SShri Abhyankar ii = a->compressedrow.i; 910*8ab949d8SShri Abhyankar ridx = a->compressedrow.rindex; 911*8ab949d8SShri Abhyankar } else { 912*8ab949d8SShri Abhyankar mbs = a->mbs; 913*8ab949d8SShri Abhyankar ii = a->i; 914*8ab949d8SShri Abhyankar z = zarray; 915*8ab949d8SShri Abhyankar } 916*8ab949d8SShri Abhyankar 917*8ab949d8SShri Abhyankar for (i=0; i<mbs; i++) { 918*8ab949d8SShri Abhyankar n = ii[i+1] - ii[i]; 919*8ab949d8SShri Abhyankar idx = ij + ii[i]; 920*8ab949d8SShri Abhyankar sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 921*8ab949d8SShri Abhyankar sum8 = 0.0; sum9 = 0.0; sum10 = 0.0; sum11 = 0.0; sum12 = 0.0; sum13 = 0.0; sum14 = 0.0;sum15 = 0.0; 922*8ab949d8SShri Abhyankar 923*8ab949d8SShri Abhyankar nonzerorow += (n>0); 924*8ab949d8SShri Abhyankar for (j=0; j<n; j++) { 925*8ab949d8SShri Abhyankar xb = x + 15*(idx[j]); 926*8ab949d8SShri Abhyankar x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 927*8ab949d8SShri Abhyankar x8 = xb[7]; x9 = xb[8]; x10 = xb[9]; x11 = xb[10]; x12 = xb[11]; x13 = xb[12]; x14 = xb[13];x15 = xb[14]; 928*8ab949d8SShri Abhyankar 929*8ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4 + v[60]*x5 + v[75]*x6 + v[90]*x7 + v[105]*x8 + v[120]*x9 + v[135]*x10 + v[150]*x11 + v[165]*x12 + v[180]*x13 + v[195]*x14 + v[210]*x15; 930*8ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4 + v[61]*x5 + v[76]*x6 + v[91]*x7 + v[106]*x8 + v[121]*x9 + v[136]*x10 + v[151]*x11 + v[166]*x12 + v[181]*x13 + v[196]*x14 + v[211]*x15; 931*8ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4 + v[62]*x5 + v[77]*x6 + v[92]*x7 + v[107]*x8 + v[122]*x9 + v[137]*x10 + v[152]*x11 + v[167]*x12 + v[182]*x13 + v[197]*x14 + v[212]*x15; 932*8ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4 + v[63]*x5 + v[78]*x6 + v[93]*x7 + v[108]*x8 + v[123]*x9 + v[138]*x10 + v[153]*x11 + v[168]*x12 + v[183]*x13 + v[198]*x14 + v[213]*x15; 933*8ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4 + v[64]*x5 + v[79]*x6 + v[94]*x7 + v[109]*x8 + v[124]*x9 + v[139]*x10 + v[154]*x11 + v[169]*x12 + v[184]*x13 + v[199]*x14 + v[214]*x15; 934*8ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4 + v[65]*x5 + v[80]*x6 + v[95]*x7 + v[110]*x8 + v[125]*x9 + v[140]*x10 + v[155]*x11 + v[170]*x12 + v[185]*x13 + v[200]*x14 + v[215]*x15; 935*8ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4 + v[66]*x5 + v[81]*x6 + v[96]*x7 + v[111]*x8 + v[126]*x9 + v[141]*x10 + v[156]*x11 + v[171]*x12 + v[186]*x13 + v[201]*x14 + v[216]*x15; 936*8ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4 + v[67]*x5 + v[82]*x6 + v[97]*x7 + v[112]*x8 + v[127]*x9 + v[142]*x10 + v[157]*x11 + v[172]*x12 + v[187]*x13 + v[202]*x14 + v[217]*x15; 937*8ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4 + v[68]*x5 + v[83]*x6 + v[98]*x7 + v[113]*x8 + v[128]*x9 + v[143]*x10 + v[158]*x11 + v[173]*x12 + v[188]*x13 + v[203]*x14 + v[218]*x15; 938*8ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4 + v[69]*x5 + v[84]*x6 + v[99]*x7 + v[114]*x8 + v[129]*x9 + v[144]*x10 + v[159]*x11 + v[174]*x12 + v[189]*x13 + v[204]*x14 + v[219]*x15; 939*8ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4 + v[70]*x5 + v[85]*x6 + v[100]*x7 + v[115]*x8 + v[130]*x9 + v[145]*x10 + v[160]*x11 + v[175]*x12 + v[190]*x13 + v[205]*x14 + v[220]*x15; 940*8ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4 + v[71]*x5 + v[86]*x6 + v[101]*x7 + v[116]*x8 + v[131]*x9 + v[146]*x10 + v[161]*x11 + v[176]*x12 + v[191]*x13 + v[206]*x14 + v[221]*x15; 941*8ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4 + v[72]*x5 + v[87]*x6 + v[102]*x7 + v[117]*x8 + v[132]*x9 + v[147]*x10 + v[162]*x11 + v[177]*x12 + v[192]*x13 + v[207]*x14 + v[222]*x15; 942*8ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4 + v[73]*x5 + v[88]*x6 + v[103]*x7 + v[118]*x8 + v[133]*x9 + v[148]*x10 + v[163]*x11 + v[178]*x12 + v[193]*x13 + v[208]*x14 + v[223]*x15; 943*8ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4 + v[74]*x5 + v[89]*x6 + v[104]*x7 + v[119]*x8 + v[134]*x9 + v[149]*x10 + v[164]*x11 + v[179]*x12 + v[194]*x13 + v[209]*x14 + v[224]*x15; 944*8ab949d8SShri Abhyankar v += 225; 945*8ab949d8SShri Abhyankar } 946*8ab949d8SShri Abhyankar if (usecprow) z = zarray + 15*ridx[i]; 947*8ab949d8SShri Abhyankar z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 948*8ab949d8SShri Abhyankar z[7] = sum8; z[8] = sum9; z[9] = sum10; z[10] = sum11; z[11] = sum12; z[12] = sum13; z[13] = sum14;z[14] = sum15; 949*8ab949d8SShri Abhyankar 950*8ab949d8SShri Abhyankar if (!usecprow) z += 15; 951*8ab949d8SShri Abhyankar } 952*8ab949d8SShri Abhyankar 953*8ab949d8SShri Abhyankar ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 954*8ab949d8SShri Abhyankar ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 955*8ab949d8SShri Abhyankar ierr = PetscLogFlops(450.0*a->nz - 15.0*nonzerorow);CHKERRQ(ierr); 956*8ab949d8SShri Abhyankar PetscFunctionReturn(0); 957*8ab949d8SShri Abhyankar } 958*8ab949d8SShri Abhyankar 959*8ab949d8SShri Abhyankar 9603f1db9ecSBarry Smith /* 9613f1db9ecSBarry Smith This will not work with MatScalar == float because it calls the BLAS 9623f1db9ecSBarry Smith */ 9634a2ae208SSatish Balay #undef __FUNCT__ 9644a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_N" 965dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_N(Mat A,Vec xx,Vec zz) 9662d61bbb3SSatish Balay { 9672d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 968dcf5cc72SBarry Smith PetscScalar *x,*z = 0,*xb,*work,*workt,*zarray; 9693f1db9ecSBarry Smith MatScalar *v; 970dfbe8321SBarry Smith PetscErrorCode ierr; 971d0f46423SBarry Smith PetscInt mbs=a->mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2; 97298c9bda7SSatish Balay PetscInt ncols,k,*ridx=PETSC_NULL,nonzerorow=0; 97326e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 9742d61bbb3SSatish Balay 9752d61bbb3SSatish Balay PetscFunctionBegin; 9761ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 97726e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 9782d61bbb3SSatish Balay 9792d61bbb3SSatish Balay idx = a->j; 9802d61bbb3SSatish Balay v = a->a; 98126e093fcSHong Zhang if (usecprow){ 98226e093fcSHong Zhang mbs = a->compressedrow.nrows; 98326e093fcSHong Zhang ii = a->compressedrow.i; 9847b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 98526e093fcSHong Zhang } else { 98626e093fcSHong Zhang mbs = a->mbs; 9872d61bbb3SSatish Balay ii = a->i; 98826e093fcSHong Zhang z = zarray; 98926e093fcSHong Zhang } 990218c64b6SSatish Balay 9912d61bbb3SSatish Balay if (!a->mult_work) { 992d0f46423SBarry Smith k = PetscMax(A->rmap->n,A->cmap->n); 99387828ca2SBarry Smith ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr); 9942d61bbb3SSatish Balay } 9952d61bbb3SSatish Balay work = a->mult_work; 9962d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 9972d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 9982d61bbb3SSatish Balay ncols = n*bs; 9992d61bbb3SSatish Balay workt = work; 100098c9bda7SSatish Balay nonzerorow += (n>0); 10012d61bbb3SSatish Balay for (j=0; j<n; j++) { 10022d61bbb3SSatish Balay xb = x + bs*(*idx++); 10032d61bbb3SSatish Balay for (k=0; k<bs; k++) workt[k] = xb[k]; 10042d61bbb3SSatish Balay workt += bs; 10052d61bbb3SSatish Balay } 10067b2bb3b9SHong Zhang if (usecprow) z = zarray + bs*ridx[i]; 1007737d121aSSatish Balay Kernel_w_gets_Ar_times_v(bs,ncols,work,v,z); 100871044d3cSBarry Smith /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DZero,z,&_One); */ 10092d61bbb3SSatish Balay v += n*bs2; 101026e093fcSHong Zhang if (!usecprow) z += bs; 10112d61bbb3SSatish Balay } 10121ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 101326e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 1014dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz*bs2 - bs*nonzerorow);CHKERRQ(ierr); 10152d61bbb3SSatish Balay PetscFunctionReturn(0); 10162d61bbb3SSatish Balay } 10172d61bbb3SSatish Balay 10186a65c766SHong Zhang extern PetscErrorCode VecCopy_Seq(Vec,Vec); 10194a2ae208SSatish Balay #undef __FUNCT__ 10204a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_1" 1021dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_1(Mat A,Vec xx,Vec yy,Vec zz) 10222d61bbb3SSatish Balay { 10232d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1024122f12eaSBarry Smith const PetscScalar *x; 1025122f12eaSBarry Smith PetscScalar *y,*z,sum; 1026122f12eaSBarry Smith const MatScalar *v; 1027dfbe8321SBarry Smith PetscErrorCode ierr; 1028122f12eaSBarry Smith PetscInt mbs=a->mbs,i,n,*ridx=PETSC_NULL,nonzerorow=0; 1029122f12eaSBarry Smith const PetscInt *idx,*ii; 103026e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 10312d61bbb3SSatish Balay 10322d61bbb3SSatish Balay PetscFunctionBegin; 1033122f12eaSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 1034122f12eaSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 10352e8a6d31SBarry Smith if (zz != yy) { 1036122f12eaSBarry Smith ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 10372e8a6d31SBarry Smith } else { 1038122f12eaSBarry Smith z = y; 10392e8a6d31SBarry Smith } 10402d61bbb3SSatish Balay 10412d61bbb3SSatish Balay idx = a->j; 10422d61bbb3SSatish Balay v = a->a; 104326e093fcSHong Zhang if (usecprow){ 10444eb6d288SHong Zhang if (zz != yy){ 1045122f12eaSBarry Smith ierr = PetscMemcpy(z,y,mbs*sizeof(PetscScalar));CHKERRQ(ierr); 10464eb6d288SHong Zhang } 104726e093fcSHong Zhang mbs = a->compressedrow.nrows; 104826e093fcSHong Zhang ii = a->compressedrow.i; 10497b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 105026e093fcSHong Zhang } else { 10512d61bbb3SSatish Balay ii = a->i; 105226e093fcSHong Zhang } 10532d61bbb3SSatish Balay 10542d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 1055122f12eaSBarry Smith n = ii[1] - ii[0]; 1056122f12eaSBarry Smith ii++; 105726e093fcSHong Zhang if (!usecprow){ 1058122f12eaSBarry Smith nonzerorow += (n>0); 1059122f12eaSBarry Smith sum = y[i]; 1060122f12eaSBarry Smith } else { 1061122f12eaSBarry Smith sum = y[ridx[i]]; 1062122f12eaSBarry Smith } 1063122f12eaSBarry Smith PetscSparseDensePlusDot(sum,x,v,idx,n); 1064122f12eaSBarry Smith v += n; 1065122f12eaSBarry Smith idx += n; 1066122f12eaSBarry Smith if (usecprow){ 1067122f12eaSBarry Smith z[ridx[i]] = sum; 1068122f12eaSBarry Smith } else { 1069122f12eaSBarry Smith z[i] = sum; 107026e093fcSHong Zhang } 10712d61bbb3SSatish Balay } 1072122f12eaSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 1073122f12eaSBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 10742e8a6d31SBarry Smith if (zz != yy) { 1075122f12eaSBarry Smith ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 10762e8a6d31SBarry Smith } 1077122f12eaSBarry Smith ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr); 10782d61bbb3SSatish Balay PetscFunctionReturn(0); 10792d61bbb3SSatish Balay } 10802d61bbb3SSatish Balay 10814a2ae208SSatish Balay #undef __FUNCT__ 10824a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_2" 1083dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_2(Mat A,Vec xx,Vec yy,Vec zz) 10842d61bbb3SSatish Balay { 10852d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1086dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2; 108726e093fcSHong Zhang PetscScalar x1,x2,*yarray,*zarray; 10883f1db9ecSBarry Smith MatScalar *v; 1089dfbe8321SBarry Smith PetscErrorCode ierr; 10904eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 109126e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 10922d61bbb3SSatish Balay 10932d61bbb3SSatish Balay PetscFunctionBegin; 10941ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 109526e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 10962e8a6d31SBarry Smith if (zz != yy) { 109726e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 10982e8a6d31SBarry Smith } else { 109926e093fcSHong Zhang zarray = yarray; 11002e8a6d31SBarry Smith } 11012d61bbb3SSatish Balay 11022d61bbb3SSatish Balay idx = a->j; 11032d61bbb3SSatish Balay v = a->a; 110426e093fcSHong Zhang if (usecprow){ 11054eb6d288SHong Zhang if (zz != yy){ 11064eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,2*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 11074eb6d288SHong Zhang } 110826e093fcSHong Zhang mbs = a->compressedrow.nrows; 110926e093fcSHong Zhang ii = a->compressedrow.i; 11107b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 11114eb6d288SHong Zhang if (zz != yy){ 11124eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,a->mbs*sizeof(PetscScalar));CHKERRQ(ierr); 11134eb6d288SHong Zhang } 111426e093fcSHong Zhang } else { 11152d61bbb3SSatish Balay ii = a->i; 111626e093fcSHong Zhang y = yarray; 111726e093fcSHong Zhang z = zarray; 111826e093fcSHong Zhang } 11192d61bbb3SSatish Balay 11202d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 11212d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 112226e093fcSHong Zhang if (usecprow){ 11237b2bb3b9SHong Zhang z = zarray + 2*ridx[i]; 11247b2bb3b9SHong Zhang y = yarray + 2*ridx[i]; 112526e093fcSHong Zhang } 11262d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; 11272d61bbb3SSatish Balay for (j=0; j<n; j++) { 11282d61bbb3SSatish Balay xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1]; 11292d61bbb3SSatish Balay sum1 += v[0]*x1 + v[2]*x2; 11302d61bbb3SSatish Balay sum2 += v[1]*x1 + v[3]*x2; 11312d61bbb3SSatish Balay v += 4; 11322d61bbb3SSatish Balay } 11332d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; 113426e093fcSHong Zhang if (!usecprow){ 11352d61bbb3SSatish Balay z += 2; y += 2; 11362d61bbb3SSatish Balay } 113726e093fcSHong Zhang } 11381ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 113926e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 11402e8a6d31SBarry Smith if (zz != yy) { 114126e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 11422e8a6d31SBarry Smith } 1143dc0b31edSSatish Balay ierr = PetscLogFlops(4.0*a->nz);CHKERRQ(ierr); 11442d61bbb3SSatish Balay PetscFunctionReturn(0); 11452d61bbb3SSatish Balay } 11462d61bbb3SSatish Balay 11474a2ae208SSatish Balay #undef __FUNCT__ 11484a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_3" 1149dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_3(Mat A,Vec xx,Vec yy,Vec zz) 11502d61bbb3SSatish Balay { 11512d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1152dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,x1,x2,x3,*yarray,*zarray; 11533f1db9ecSBarry Smith MatScalar *v; 1154dfbe8321SBarry Smith PetscErrorCode ierr; 11554eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 115626e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 11572d61bbb3SSatish Balay 11582d61bbb3SSatish Balay PetscFunctionBegin; 11591ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 116026e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 11612e8a6d31SBarry Smith if (zz != yy) { 116226e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 11632e8a6d31SBarry Smith } else { 116426e093fcSHong Zhang zarray = yarray; 11652e8a6d31SBarry Smith } 11662d61bbb3SSatish Balay 11672d61bbb3SSatish Balay idx = a->j; 11682d61bbb3SSatish Balay v = a->a; 116926e093fcSHong Zhang if (usecprow){ 11704eb6d288SHong Zhang if (zz != yy){ 11714eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,3*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 11724eb6d288SHong Zhang } 117326e093fcSHong Zhang mbs = a->compressedrow.nrows; 117426e093fcSHong Zhang ii = a->compressedrow.i; 11757b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 117626e093fcSHong Zhang } else { 11772d61bbb3SSatish Balay ii = a->i; 117826e093fcSHong Zhang y = yarray; 117926e093fcSHong Zhang z = zarray; 118026e093fcSHong Zhang } 11812d61bbb3SSatish Balay 11822d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 11832d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 118426e093fcSHong Zhang if (usecprow){ 11857b2bb3b9SHong Zhang z = zarray + 3*ridx[i]; 11867b2bb3b9SHong Zhang y = yarray + 3*ridx[i]; 118726e093fcSHong Zhang } 11882d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; 11892d61bbb3SSatish Balay for (j=0; j<n; j++) { 11902d61bbb3SSatish Balay xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 11912d61bbb3SSatish Balay sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3; 11922d61bbb3SSatish Balay sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3; 11932d61bbb3SSatish Balay sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3; 11942d61bbb3SSatish Balay v += 9; 11952d61bbb3SSatish Balay } 11962d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; 119726e093fcSHong Zhang if (!usecprow){ 11982d61bbb3SSatish Balay z += 3; y += 3; 11992d61bbb3SSatish Balay } 120026e093fcSHong Zhang } 12011ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 120226e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 12032e8a6d31SBarry Smith if (zz != yy) { 120426e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 12052e8a6d31SBarry Smith } 1206dc0b31edSSatish Balay ierr = PetscLogFlops(18.0*a->nz);CHKERRQ(ierr); 12072d61bbb3SSatish Balay PetscFunctionReturn(0); 12082d61bbb3SSatish Balay } 12092d61bbb3SSatish Balay 12104a2ae208SSatish Balay #undef __FUNCT__ 12114a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_4" 1212dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_4(Mat A,Vec xx,Vec yy,Vec zz) 12132d61bbb3SSatish Balay { 12142d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1215dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*yarray,*zarray; 12163f1db9ecSBarry Smith MatScalar *v; 1217dfbe8321SBarry Smith PetscErrorCode ierr; 12184eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 121926e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 12202d61bbb3SSatish Balay 12212d61bbb3SSatish Balay PetscFunctionBegin; 12221ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 122326e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 12242e8a6d31SBarry Smith if (zz != yy) { 122526e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 12262e8a6d31SBarry Smith } else { 122726e093fcSHong Zhang zarray = yarray; 12282e8a6d31SBarry Smith } 12292d61bbb3SSatish Balay 12302d61bbb3SSatish Balay idx = a->j; 12312d61bbb3SSatish Balay v = a->a; 123226e093fcSHong Zhang if (usecprow){ 12334eb6d288SHong Zhang if (zz != yy){ 12344eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,4*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 12354eb6d288SHong Zhang } 123626e093fcSHong Zhang mbs = a->compressedrow.nrows; 123726e093fcSHong Zhang ii = a->compressedrow.i; 12387b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 123926e093fcSHong Zhang } else { 12402d61bbb3SSatish Balay ii = a->i; 124126e093fcSHong Zhang y = yarray; 124226e093fcSHong Zhang z = zarray; 124326e093fcSHong Zhang } 12442d61bbb3SSatish Balay 12452d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 12462d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 124726e093fcSHong Zhang if (usecprow){ 12487b2bb3b9SHong Zhang z = zarray + 4*ridx[i]; 12497b2bb3b9SHong Zhang y = yarray + 4*ridx[i]; 125026e093fcSHong Zhang } 12512d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; 12522d61bbb3SSatish Balay for (j=0; j<n; j++) { 12532d61bbb3SSatish Balay xb = x + 4*(*idx++); 12542d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 12552d61bbb3SSatish Balay sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3 + v[12]*x4; 12562d61bbb3SSatish Balay sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3 + v[13]*x4; 12572d61bbb3SSatish Balay sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3 + v[14]*x4; 12582d61bbb3SSatish Balay sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3 + v[15]*x4; 12592d61bbb3SSatish Balay v += 16; 12602d61bbb3SSatish Balay } 12612d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; 126226e093fcSHong Zhang if (!usecprow){ 12632d61bbb3SSatish Balay z += 4; y += 4; 12642d61bbb3SSatish Balay } 126526e093fcSHong Zhang } 12661ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 126726e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 12682e8a6d31SBarry Smith if (zz != yy) { 126926e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 12702e8a6d31SBarry Smith } 1271dc0b31edSSatish Balay ierr = PetscLogFlops(32.0*a->nz);CHKERRQ(ierr); 12722d61bbb3SSatish Balay PetscFunctionReturn(0); 12732d61bbb3SSatish Balay } 12742d61bbb3SSatish Balay 12754a2ae208SSatish Balay #undef __FUNCT__ 12764a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_5" 1277dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_5(Mat A,Vec xx,Vec yy,Vec zz) 12782d61bbb3SSatish Balay { 12792d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1280dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5; 128126e093fcSHong Zhang PetscScalar *yarray,*zarray; 12823f1db9ecSBarry Smith MatScalar *v; 1283dfbe8321SBarry Smith PetscErrorCode ierr; 12844eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 128526e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 12862d61bbb3SSatish Balay 12872d61bbb3SSatish Balay PetscFunctionBegin; 12881ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 128926e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 12902e8a6d31SBarry Smith if (zz != yy) { 129126e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 12922e8a6d31SBarry Smith } else { 129326e093fcSHong Zhang zarray = yarray; 12942e8a6d31SBarry Smith } 12952d61bbb3SSatish Balay 12962d61bbb3SSatish Balay idx = a->j; 12972d61bbb3SSatish Balay v = a->a; 129826e093fcSHong Zhang if (usecprow){ 12994eb6d288SHong Zhang if (zz != yy){ 13004eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,5*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 13014eb6d288SHong Zhang } 130226e093fcSHong Zhang mbs = a->compressedrow.nrows; 130326e093fcSHong Zhang ii = a->compressedrow.i; 13047b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 130526e093fcSHong Zhang } else { 13062d61bbb3SSatish Balay ii = a->i; 130726e093fcSHong Zhang y = yarray; 130826e093fcSHong Zhang z = zarray; 130926e093fcSHong Zhang } 13102d61bbb3SSatish Balay 13112d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 13122d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 131326e093fcSHong Zhang if (usecprow){ 13147b2bb3b9SHong Zhang z = zarray + 5*ridx[i]; 13157b2bb3b9SHong Zhang y = yarray + 5*ridx[i]; 131626e093fcSHong Zhang } 13172d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; 13182d61bbb3SSatish Balay for (j=0; j<n; j++) { 13192d61bbb3SSatish Balay xb = x + 5*(*idx++); 13202d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; 13212d61bbb3SSatish Balay sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3 + v[15]*x4 + v[20]*x5; 13222d61bbb3SSatish Balay sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3 + v[16]*x4 + v[21]*x5; 13232d61bbb3SSatish Balay sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3 + v[17]*x4 + v[22]*x5; 13242d61bbb3SSatish Balay sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3 + v[18]*x4 + v[23]*x5; 13252d61bbb3SSatish Balay sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3 + v[19]*x4 + v[24]*x5; 13262d61bbb3SSatish Balay v += 25; 13272d61bbb3SSatish Balay } 13282d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; 132926e093fcSHong Zhang if (!usecprow){ 13302d61bbb3SSatish Balay z += 5; y += 5; 13312d61bbb3SSatish Balay } 133226e093fcSHong Zhang } 13331ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 133426e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 13352e8a6d31SBarry Smith if (zz != yy) { 133626e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 13372e8a6d31SBarry Smith } 1338dc0b31edSSatish Balay ierr = PetscLogFlops(50.0*a->nz);CHKERRQ(ierr); 13392d61bbb3SSatish Balay PetscFunctionReturn(0); 13402d61bbb3SSatish Balay } 13414a2ae208SSatish Balay #undef __FUNCT__ 13424a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_6" 1343dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_6(Mat A,Vec xx,Vec yy,Vec zz) 134415091d37SBarry Smith { 134515091d37SBarry Smith Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1346dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6; 134726e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,*yarray,*zarray; 134815091d37SBarry Smith MatScalar *v; 1349dfbe8321SBarry Smith PetscErrorCode ierr; 13504eb6d288SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 135126e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 135215091d37SBarry Smith 135315091d37SBarry Smith PetscFunctionBegin; 13541ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 135526e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 135615091d37SBarry Smith if (zz != yy) { 135726e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 135815091d37SBarry Smith } else { 135926e093fcSHong Zhang zarray = yarray; 136015091d37SBarry Smith } 136115091d37SBarry Smith 136215091d37SBarry Smith idx = a->j; 136315091d37SBarry Smith v = a->a; 136426e093fcSHong Zhang if (usecprow){ 13654eb6d288SHong Zhang if (zz != yy){ 13664eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,6*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 13674eb6d288SHong Zhang } 136826e093fcSHong Zhang mbs = a->compressedrow.nrows; 136926e093fcSHong Zhang ii = a->compressedrow.i; 13707b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 137126e093fcSHong Zhang } else { 137215091d37SBarry Smith ii = a->i; 137326e093fcSHong Zhang y = yarray; 137426e093fcSHong Zhang z = zarray; 137526e093fcSHong Zhang } 137615091d37SBarry Smith 137715091d37SBarry Smith for (i=0; i<mbs; i++) { 137815091d37SBarry Smith n = ii[1] - ii[0]; ii++; 137926e093fcSHong Zhang if (usecprow){ 13807b2bb3b9SHong Zhang z = zarray + 6*ridx[i]; 13817b2bb3b9SHong Zhang y = yarray + 6*ridx[i]; 138226e093fcSHong Zhang } 138315091d37SBarry Smith sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5]; 138415091d37SBarry Smith for (j=0; j<n; j++) { 13853b95cb0eSSatish Balay xb = x + 6*(*idx++); 138615091d37SBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; 138715091d37SBarry Smith sum1 += v[0]*x1 + v[6]*x2 + v[12]*x3 + v[18]*x4 + v[24]*x5 + v[30]*x6; 138815091d37SBarry Smith sum2 += v[1]*x1 + v[7]*x2 + v[13]*x3 + v[19]*x4 + v[25]*x5 + v[31]*x6; 138915091d37SBarry Smith sum3 += v[2]*x1 + v[8]*x2 + v[14]*x3 + v[20]*x4 + v[26]*x5 + v[32]*x6; 139015091d37SBarry Smith sum4 += v[3]*x1 + v[9]*x2 + v[15]*x3 + v[21]*x4 + v[27]*x5 + v[33]*x6; 139115091d37SBarry Smith sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3 + v[22]*x4 + v[28]*x5 + v[34]*x6; 139215091d37SBarry Smith sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3 + v[23]*x4 + v[29]*x5 + v[35]*x6; 139315091d37SBarry Smith v += 36; 139415091d37SBarry Smith } 139515091d37SBarry Smith z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; 139626e093fcSHong Zhang if (!usecprow){ 139715091d37SBarry Smith z += 6; y += 6; 139815091d37SBarry Smith } 139926e093fcSHong Zhang } 14001ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 140126e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 140215091d37SBarry Smith if (zz != yy) { 140326e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 140415091d37SBarry Smith } 1405dc0b31edSSatish Balay ierr = PetscLogFlops(72.0*a->nz);CHKERRQ(ierr); 140615091d37SBarry Smith PetscFunctionReturn(0); 140715091d37SBarry Smith } 14082d61bbb3SSatish Balay 14094a2ae208SSatish Balay #undef __FUNCT__ 14104a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_7" 1411dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_7(Mat A,Vec xx,Vec yy,Vec zz) 14122d61bbb3SSatish Balay { 14132d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1414dcf5cc72SBarry Smith PetscScalar *x,*y = 0,*z = 0,*xb,sum1,sum2,sum3,sum4,sum5,sum6,sum7; 141526e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,x7,*yarray,*zarray; 14163f1db9ecSBarry Smith MatScalar *v; 1417dfbe8321SBarry Smith PetscErrorCode ierr; 14187b2bb3b9SHong Zhang PetscInt mbs=a->mbs,i,*idx,*ii,j,n,*ridx=PETSC_NULL; 141926e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 14202d61bbb3SSatish Balay 14212d61bbb3SSatish Balay PetscFunctionBegin; 14221ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 142326e093fcSHong Zhang ierr = VecGetArray(yy,&yarray);CHKERRQ(ierr); 14242e8a6d31SBarry Smith if (zz != yy) { 142526e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 14262e8a6d31SBarry Smith } else { 142726e093fcSHong Zhang zarray = yarray; 14282e8a6d31SBarry Smith } 14292d61bbb3SSatish Balay 14302d61bbb3SSatish Balay idx = a->j; 14312d61bbb3SSatish Balay v = a->a; 143226e093fcSHong Zhang if (usecprow){ 14334eb6d288SHong Zhang if (zz != yy){ 14344eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,7*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 14354eb6d288SHong Zhang } 143626e093fcSHong Zhang mbs = a->compressedrow.nrows; 143726e093fcSHong Zhang ii = a->compressedrow.i; 14387b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 143926e093fcSHong Zhang } else { 14402d61bbb3SSatish Balay ii = a->i; 144126e093fcSHong Zhang y = yarray; 144226e093fcSHong Zhang z = zarray; 144326e093fcSHong Zhang } 14442d61bbb3SSatish Balay 14452d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 14462d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 144726e093fcSHong Zhang if (usecprow){ 14487b2bb3b9SHong Zhang z = zarray + 7*ridx[i]; 14497b2bb3b9SHong Zhang y = yarray + 7*ridx[i]; 145026e093fcSHong Zhang } 14512d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5]; sum7 = y[6]; 14522d61bbb3SSatish Balay for (j=0; j<n; j++) { 14532d61bbb3SSatish Balay xb = x + 7*(*idx++); 14542d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 14552d61bbb3SSatish Balay sum1 += v[0]*x1 + v[7]*x2 + v[14]*x3 + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7; 14562d61bbb3SSatish Balay sum2 += v[1]*x1 + v[8]*x2 + v[15]*x3 + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7; 14572d61bbb3SSatish Balay sum3 += v[2]*x1 + v[9]*x2 + v[16]*x3 + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7; 14582d61bbb3SSatish Balay sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3 + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7; 14592d61bbb3SSatish Balay sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3 + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7; 14602d61bbb3SSatish Balay sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3 + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7; 14612d61bbb3SSatish Balay sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3 + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7; 14622d61bbb3SSatish Balay v += 49; 14632d61bbb3SSatish Balay } 14642d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 146526e093fcSHong Zhang if (!usecprow){ 14662d61bbb3SSatish Balay z += 7; y += 7; 14672d61bbb3SSatish Balay } 146826e093fcSHong Zhang } 14691ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 147026e093fcSHong Zhang ierr = VecRestoreArray(yy,&yarray);CHKERRQ(ierr); 14712e8a6d31SBarry Smith if (zz != yy) { 147226e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 14732e8a6d31SBarry Smith } 1474dc0b31edSSatish Balay ierr = PetscLogFlops(98.0*a->nz);CHKERRQ(ierr); 14752d61bbb3SSatish Balay PetscFunctionReturn(0); 14762d61bbb3SSatish Balay } 1477218c64b6SSatish Balay 14784a2ae208SSatish Balay #undef __FUNCT__ 14794a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_N" 1480dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_N(Mat A,Vec xx,Vec yy,Vec zz) 14812d61bbb3SSatish Balay { 14822d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1483bba805e6SBarry Smith PetscScalar *x,*z = 0,*xb,*work,*workt,*zarray; 14843f1db9ecSBarry Smith MatScalar *v; 14856849ba73SBarry Smith PetscErrorCode ierr; 1486d0f46423SBarry Smith PetscInt mbs,i,*idx,*ii,bs=A->rmap->bs,j,n,bs2=a->bs2; 14877b2bb3b9SHong Zhang PetscInt ncols,k,*ridx=PETSC_NULL; 148826e093fcSHong Zhang PetscTruth usecprow=a->compressedrow.use; 1489218c64b6SSatish Balay 14902d61bbb3SSatish Balay PetscFunctionBegin; 14916a65c766SHong Zhang ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr); 14921ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 149326e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 14942d61bbb3SSatish Balay 14952d61bbb3SSatish Balay idx = a->j; 14962d61bbb3SSatish Balay v = a->a; 149726e093fcSHong Zhang if (usecprow){ 149826e093fcSHong Zhang mbs = a->compressedrow.nrows; 149926e093fcSHong Zhang ii = a->compressedrow.i; 15007b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 150126e093fcSHong Zhang } else { 150226e093fcSHong Zhang mbs = a->mbs; 15032d61bbb3SSatish Balay ii = a->i; 150426e093fcSHong Zhang z = zarray; 150526e093fcSHong Zhang } 15062d61bbb3SSatish Balay 15072d61bbb3SSatish Balay if (!a->mult_work) { 1508d0f46423SBarry Smith k = PetscMax(A->rmap->n,A->cmap->n); 150987828ca2SBarry Smith ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr); 15102d61bbb3SSatish Balay } 15112d61bbb3SSatish Balay work = a->mult_work; 15122d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 15132d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 15142d61bbb3SSatish Balay ncols = n*bs; 15152d61bbb3SSatish Balay workt = work; 15162d61bbb3SSatish Balay for (j=0; j<n; j++) { 15172d61bbb3SSatish Balay xb = x + bs*(*idx++); 15182d61bbb3SSatish Balay for (k=0; k<bs; k++) workt[k] = xb[k]; 15192d61bbb3SSatish Balay workt += bs; 15202d61bbb3SSatish Balay } 15217b2bb3b9SHong Zhang if (usecprow) z = zarray + bs*ridx[i]; 15223f1db9ecSBarry Smith Kernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z); 152371044d3cSBarry Smith /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DOne,z,&_One); */ 15242d61bbb3SSatish Balay v += n*bs2; 152526e093fcSHong Zhang if (!usecprow){ 15262d61bbb3SSatish Balay z += bs; 15272d61bbb3SSatish Balay } 152826e093fcSHong Zhang } 15291ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 153026e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 1531dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz*bs2);CHKERRQ(ierr); 15322d61bbb3SSatish Balay PetscFunctionReturn(0); 15332d61bbb3SSatish Balay } 15342d61bbb3SSatish Balay 15354a2ae208SSatish Balay #undef __FUNCT__ 1536547795f9SHong Zhang #define __FUNCT__ "MatMultHermitianTranspose_SeqBAIJ" 1537547795f9SHong Zhang PetscErrorCode MatMultHermitianTranspose_SeqBAIJ(Mat A,Vec xx,Vec zz) 1538547795f9SHong Zhang { 1539547795f9SHong Zhang PetscScalar zero = 0.0; 1540547795f9SHong Zhang PetscErrorCode ierr; 1541547795f9SHong Zhang 1542547795f9SHong Zhang PetscFunctionBegin; 1543547795f9SHong Zhang ierr = VecSet(zz,zero);CHKERRQ(ierr); 1544547795f9SHong Zhang ierr = MatMultHermitianTransposeAdd_SeqBAIJ(A,xx,zz,zz);CHKERRQ(ierr); 1545547795f9SHong Zhang PetscFunctionReturn(0); 1546547795f9SHong Zhang } 1547547795f9SHong Zhang 1548547795f9SHong Zhang #undef __FUNCT__ 15494a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqBAIJ" 1550dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqBAIJ(Mat A,Vec xx,Vec zz) 15512d61bbb3SSatish Balay { 15523447b6efSHong Zhang PetscScalar zero = 0.0; 15536849ba73SBarry Smith PetscErrorCode ierr; 15542d61bbb3SSatish Balay 15552d61bbb3SSatish Balay PetscFunctionBegin; 15562dcb1b2aSMatthew Knepley ierr = VecSet(zz,zero);CHKERRQ(ierr); 15573447b6efSHong Zhang ierr = MatMultTransposeAdd_SeqBAIJ(A,xx,zz,zz);CHKERRQ(ierr); 15582d61bbb3SSatish Balay PetscFunctionReturn(0); 15592d61bbb3SSatish Balay } 15602d61bbb3SSatish Balay 15614a2ae208SSatish Balay #undef __FUNCT__ 1562547795f9SHong Zhang #define __FUNCT__ "MatMultHermitianTransposeAdd_SeqBAIJ" 1563547795f9SHong Zhang PetscErrorCode MatMultHermitianTransposeAdd_SeqBAIJ(Mat A,Vec xx,Vec yy,Vec zz) 1564547795f9SHong Zhang 1565547795f9SHong Zhang { 1566547795f9SHong Zhang Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1567547795f9SHong Zhang PetscScalar *zb,*x,*z,*xb = 0,x1,x2,x3,x4,x5; 1568547795f9SHong Zhang MatScalar *v; 1569547795f9SHong Zhang PetscErrorCode ierr; 1570547795f9SHong Zhang PetscInt mbs,i,*idx,*ii,rval,bs=A->rmap->bs,j,n,bs2=a->bs2,*ib,*ridx=PETSC_NULL; 1571547795f9SHong Zhang Mat_CompressedRow cprow = a->compressedrow; 1572547795f9SHong Zhang PetscTruth usecprow=cprow.use; 1573547795f9SHong Zhang 1574547795f9SHong Zhang PetscFunctionBegin; 1575547795f9SHong Zhang if (yy != zz) { ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr); } 1576547795f9SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1577547795f9SHong Zhang ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 1578547795f9SHong Zhang 1579547795f9SHong Zhang idx = a->j; 1580547795f9SHong Zhang v = a->a; 1581547795f9SHong Zhang if (usecprow){ 1582547795f9SHong Zhang mbs = cprow.nrows; 1583547795f9SHong Zhang ii = cprow.i; 1584547795f9SHong Zhang ridx = cprow.rindex; 1585547795f9SHong Zhang } else { 1586547795f9SHong Zhang mbs=a->mbs; 1587547795f9SHong Zhang ii = a->i; 1588547795f9SHong Zhang xb = x; 1589547795f9SHong Zhang } 1590547795f9SHong Zhang 1591547795f9SHong Zhang switch (bs) { 1592547795f9SHong Zhang case 1: 1593547795f9SHong Zhang for (i=0; i<mbs; i++) { 1594547795f9SHong Zhang if (usecprow) xb = x + ridx[i]; 1595547795f9SHong Zhang x1 = xb[0]; 1596547795f9SHong Zhang ib = idx + ii[0]; 1597547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 1598547795f9SHong Zhang for (j=0; j<n; j++) { 1599547795f9SHong Zhang rval = ib[j]; 1600547795f9SHong Zhang z[rval] += PetscConj(*v) * x1; 1601547795f9SHong Zhang v++; 1602547795f9SHong Zhang } 1603547795f9SHong Zhang if (!usecprow) xb++; 1604547795f9SHong Zhang } 1605547795f9SHong Zhang break; 1606547795f9SHong Zhang case 2: 1607547795f9SHong Zhang for (i=0; i<mbs; i++) { 1608547795f9SHong Zhang if (usecprow) xb = x + 2*ridx[i]; 1609547795f9SHong Zhang x1 = xb[0]; x2 = xb[1]; 1610547795f9SHong Zhang ib = idx + ii[0]; 1611547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 1612547795f9SHong Zhang for (j=0; j<n; j++) { 1613547795f9SHong Zhang rval = ib[j]*2; 1614547795f9SHong Zhang z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2; 1615547795f9SHong Zhang z[rval++] += PetscConj(v[2])*x1 + PetscConj(v[3])*x2; 1616547795f9SHong Zhang v += 4; 1617547795f9SHong Zhang } 1618547795f9SHong Zhang if (!usecprow) xb += 2; 1619547795f9SHong Zhang } 1620547795f9SHong Zhang break; 1621547795f9SHong Zhang case 3: 1622547795f9SHong Zhang for (i=0; i<mbs; i++) { 1623547795f9SHong Zhang if (usecprow) xb = x + 3*ridx[i]; 1624547795f9SHong Zhang x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 1625547795f9SHong Zhang ib = idx + ii[0]; 1626547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 1627547795f9SHong Zhang for (j=0; j<n; j++) { 1628547795f9SHong Zhang rval = ib[j]*3; 1629547795f9SHong Zhang z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2 + PetscConj(v[2])*x3; 1630547795f9SHong Zhang z[rval++] += PetscConj(v[3])*x1 + PetscConj(v[4])*x2 + PetscConj(v[5])*x3; 1631547795f9SHong Zhang z[rval++] += PetscConj(v[6])*x1 + PetscConj(v[7])*x2 + PetscConj(v[8])*x3; 1632547795f9SHong Zhang v += 9; 1633547795f9SHong Zhang } 1634547795f9SHong Zhang if (!usecprow) xb += 3; 1635547795f9SHong Zhang } 1636547795f9SHong Zhang break; 1637547795f9SHong Zhang case 4: 1638547795f9SHong Zhang for (i=0; i<mbs; i++) { 1639547795f9SHong Zhang if (usecprow) xb = x + 4*ridx[i]; 1640547795f9SHong Zhang x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 1641547795f9SHong Zhang ib = idx + ii[0]; 1642547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 1643547795f9SHong Zhang for (j=0; j<n; j++) { 1644547795f9SHong Zhang rval = ib[j]*4; 1645547795f9SHong Zhang z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2 + PetscConj(v[2])*x3 + PetscConj(v[3])*x4; 1646547795f9SHong Zhang z[rval++] += PetscConj(v[4])*x1 + PetscConj(v[5])*x2 + PetscConj(v[6])*x3 + PetscConj(v[7])*x4; 1647547795f9SHong Zhang z[rval++] += PetscConj(v[8])*x1 + PetscConj(v[9])*x2 + PetscConj(v[10])*x3 + PetscConj(v[11])*x4; 1648547795f9SHong Zhang z[rval++] += PetscConj(v[12])*x1 + PetscConj(v[13])*x2 + PetscConj(v[14])*x3 + PetscConj(v[15])*x4; 1649547795f9SHong Zhang v += 16; 1650547795f9SHong Zhang } 1651547795f9SHong Zhang if (!usecprow) xb += 4; 1652547795f9SHong Zhang } 1653547795f9SHong Zhang break; 1654547795f9SHong Zhang case 5: 1655547795f9SHong Zhang for (i=0; i<mbs; i++) { 1656547795f9SHong Zhang if (usecprow) xb = x + 5*ridx[i]; 1657547795f9SHong Zhang x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 1658547795f9SHong Zhang x4 = xb[3]; x5 = xb[4]; 1659547795f9SHong Zhang ib = idx + ii[0]; 1660547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 1661547795f9SHong Zhang for (j=0; j<n; j++) { 1662547795f9SHong Zhang rval = ib[j]*5; 1663547795f9SHong Zhang z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2 + PetscConj(v[2])*x3 + PetscConj(v[3])*x4 + PetscConj(v[4])*x5; 1664547795f9SHong Zhang z[rval++] += PetscConj(v[5])*x1 + PetscConj(v[6])*x2 + PetscConj(v[7])*x3 + PetscConj(v[8])*x4 + PetscConj(v[9])*x5; 1665547795f9SHong Zhang z[rval++] += PetscConj(v[10])*x1 + PetscConj(v[11])*x2 + PetscConj(v[12])*x3 + PetscConj(v[13])*x4 + PetscConj(v[14])*x5; 1666547795f9SHong Zhang z[rval++] += PetscConj(v[15])*x1 + PetscConj(v[16])*x2 + PetscConj(v[17])*x3 + PetscConj(v[18])*x4 + PetscConj(v[19])*x5; 1667547795f9SHong Zhang z[rval++] += PetscConj(v[20])*x1 + PetscConj(v[21])*x2 + PetscConj(v[22])*x3 + PetscConj(v[23])*x4 + PetscConj(v[24])*x5; 1668547795f9SHong Zhang v += 25; 1669547795f9SHong Zhang } 1670547795f9SHong Zhang if (!usecprow) xb += 5; 1671547795f9SHong Zhang } 1672547795f9SHong Zhang break; 1673547795f9SHong Zhang default: { /* block sizes larger than 5 by 5 are handled by BLAS */ 1674547795f9SHong Zhang PetscInt ncols,k; 1675547795f9SHong Zhang PetscScalar *work,*workt,*xtmp; 1676547795f9SHong Zhang 1677547795f9SHong Zhang SETERRQ(PETSC_ERR_SUP,"block size larger than 5 is not supported yet"); 1678547795f9SHong Zhang if (!a->mult_work) { 1679547795f9SHong Zhang k = PetscMax(A->rmap->n,A->cmap->n); 1680547795f9SHong Zhang ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr); 1681547795f9SHong Zhang } 1682547795f9SHong Zhang work = a->mult_work; 1683547795f9SHong Zhang xtmp = x; 1684547795f9SHong Zhang for (i=0; i<mbs; i++) { 1685547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 1686547795f9SHong Zhang ncols = n*bs; 1687547795f9SHong Zhang ierr = PetscMemzero(work,ncols*sizeof(PetscScalar));CHKERRQ(ierr); 1688547795f9SHong Zhang if (usecprow) { 1689547795f9SHong Zhang xtmp = x + bs*ridx[i]; 1690547795f9SHong Zhang } 1691547795f9SHong Zhang Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,xtmp,v,work); 1692547795f9SHong Zhang /* BLASgemv_("T",&bs,&ncols,&_DOne,v,&bs,xtmp,&_One,&_DOne,work,&_One); */ 1693547795f9SHong Zhang v += n*bs2; 1694547795f9SHong Zhang if (!usecprow) xtmp += bs; 1695547795f9SHong Zhang workt = work; 1696547795f9SHong Zhang for (j=0; j<n; j++) { 1697547795f9SHong Zhang zb = z + bs*(*idx++); 1698547795f9SHong Zhang for (k=0; k<bs; k++) zb[k] += workt[k] ; 1699547795f9SHong Zhang workt += bs; 1700547795f9SHong Zhang } 1701547795f9SHong Zhang } 1702547795f9SHong Zhang } 1703547795f9SHong Zhang } 1704547795f9SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1705547795f9SHong Zhang ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 1706547795f9SHong Zhang ierr = PetscLogFlops(2.0*a->nz*a->bs2);CHKERRQ(ierr); 1707547795f9SHong Zhang PetscFunctionReturn(0); 1708547795f9SHong Zhang } 1709547795f9SHong Zhang 1710547795f9SHong Zhang #undef __FUNCT__ 17114a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqBAIJ" 1712dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqBAIJ(Mat A,Vec xx,Vec yy,Vec zz) 17132d61bbb3SSatish Balay { 17142d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1715dcf5cc72SBarry Smith PetscScalar *zb,*x,*z,*xb = 0,x1,x2,x3,x4,x5; 17163f1db9ecSBarry Smith MatScalar *v; 17176849ba73SBarry Smith PetscErrorCode ierr; 1718d0f46423SBarry Smith PetscInt mbs,i,*idx,*ii,rval,bs=A->rmap->bs,j,n,bs2=a->bs2,*ib,*ridx=PETSC_NULL; 17193447b6efSHong Zhang Mat_CompressedRow cprow = a->compressedrow; 17203447b6efSHong Zhang PetscTruth usecprow=cprow.use; 17212d61bbb3SSatish Balay 17222d61bbb3SSatish Balay PetscFunctionBegin; 17236a65c766SHong Zhang if (yy != zz) { ierr = VecCopy_Seq(yy,zz);CHKERRQ(ierr); } 17243447b6efSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 17253447b6efSHong Zhang ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 17262d61bbb3SSatish Balay 17272d61bbb3SSatish Balay idx = a->j; 17282d61bbb3SSatish Balay v = a->a; 17293447b6efSHong Zhang if (usecprow){ 17303447b6efSHong Zhang mbs = cprow.nrows; 17313447b6efSHong Zhang ii = cprow.i; 17327b2bb3b9SHong Zhang ridx = cprow.rindex; 17333447b6efSHong Zhang } else { 17343447b6efSHong Zhang mbs=a->mbs; 17352d61bbb3SSatish Balay ii = a->i; 1736f1af5d2fSBarry Smith xb = x; 17373447b6efSHong Zhang } 17382d61bbb3SSatish Balay 17392d61bbb3SSatish Balay switch (bs) { 17402d61bbb3SSatish Balay case 1: 17412d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 17427b2bb3b9SHong Zhang if (usecprow) xb = x + ridx[i]; 1743f1af5d2fSBarry Smith x1 = xb[0]; 17443447b6efSHong Zhang ib = idx + ii[0]; 17453447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 17462d61bbb3SSatish Balay for (j=0; j<n; j++) { 17472d61bbb3SSatish Balay rval = ib[j]; 1748f1af5d2fSBarry Smith z[rval] += *v * x1; 1749f1af5d2fSBarry Smith v++; 17502d61bbb3SSatish Balay } 17513447b6efSHong Zhang if (!usecprow) xb++; 17522d61bbb3SSatish Balay } 17532d61bbb3SSatish Balay break; 17542d61bbb3SSatish Balay case 2: 17552d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 17567b2bb3b9SHong Zhang if (usecprow) xb = x + 2*ridx[i]; 1757f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; 17583447b6efSHong Zhang ib = idx + ii[0]; 17593447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 17602d61bbb3SSatish Balay for (j=0; j<n; j++) { 17612d61bbb3SSatish Balay rval = ib[j]*2; 17622d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2; 17632d61bbb3SSatish Balay z[rval++] += v[2]*x1 + v[3]*x2; 17642d61bbb3SSatish Balay v += 4; 17652d61bbb3SSatish Balay } 17663447b6efSHong Zhang if (!usecprow) xb += 2; 17672d61bbb3SSatish Balay } 17682d61bbb3SSatish Balay break; 17692d61bbb3SSatish Balay case 3: 17702d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 17717b2bb3b9SHong Zhang if (usecprow) xb = x + 3*ridx[i]; 1772f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 17733447b6efSHong Zhang ib = idx + ii[0]; 17743447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 17752d61bbb3SSatish Balay for (j=0; j<n; j++) { 17762d61bbb3SSatish Balay rval = ib[j]*3; 17772d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3; 17782d61bbb3SSatish Balay z[rval++] += v[3]*x1 + v[4]*x2 + v[5]*x3; 17792d61bbb3SSatish Balay z[rval++] += v[6]*x1 + v[7]*x2 + v[8]*x3; 17802d61bbb3SSatish Balay v += 9; 17812d61bbb3SSatish Balay } 17823447b6efSHong Zhang if (!usecprow) xb += 3; 17832d61bbb3SSatish Balay } 17842d61bbb3SSatish Balay break; 17852d61bbb3SSatish Balay case 4: 17862d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 17877b2bb3b9SHong Zhang if (usecprow) xb = x + 4*ridx[i]; 1788f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 17893447b6efSHong Zhang ib = idx + ii[0]; 17903447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 17912d61bbb3SSatish Balay for (j=0; j<n; j++) { 17922d61bbb3SSatish Balay rval = ib[j]*4; 17932d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4; 17942d61bbb3SSatish Balay z[rval++] += v[4]*x1 + v[5]*x2 + v[6]*x3 + v[7]*x4; 17952d61bbb3SSatish Balay z[rval++] += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[11]*x4; 17962d61bbb3SSatish Balay z[rval++] += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4; 17972d61bbb3SSatish Balay v += 16; 17982d61bbb3SSatish Balay } 17993447b6efSHong Zhang if (!usecprow) xb += 4; 18002d61bbb3SSatish Balay } 18012d61bbb3SSatish Balay break; 18022d61bbb3SSatish Balay case 5: 18032d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 18047b2bb3b9SHong Zhang if (usecprow) xb = x + 5*ridx[i]; 1805f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 18062d61bbb3SSatish Balay x4 = xb[3]; x5 = xb[4]; 18073447b6efSHong Zhang ib = idx + ii[0]; 18083447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 18092d61bbb3SSatish Balay for (j=0; j<n; j++) { 18102d61bbb3SSatish Balay rval = ib[j]*5; 18112d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4 + v[4]*x5; 18122d61bbb3SSatish Balay z[rval++] += v[5]*x1 + v[6]*x2 + v[7]*x3 + v[8]*x4 + v[9]*x5; 18132d61bbb3SSatish Balay z[rval++] += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4 + v[14]*x5; 18142d61bbb3SSatish Balay z[rval++] += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4 + v[19]*x5; 18152d61bbb3SSatish Balay z[rval++] += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4 + v[24]*x5; 18162d61bbb3SSatish Balay v += 25; 18172d61bbb3SSatish Balay } 18183447b6efSHong Zhang if (!usecprow) xb += 5; 18192d61bbb3SSatish Balay } 18202d61bbb3SSatish Balay break; 1821f1af5d2fSBarry Smith default: { /* block sizes larger then 5 by 5 are handled by BLAS */ 1822690b6cddSBarry Smith PetscInt ncols,k; 18233447b6efSHong Zhang PetscScalar *work,*workt,*xtmp; 18243f1db9ecSBarry Smith 18252d61bbb3SSatish Balay if (!a->mult_work) { 1826d0f46423SBarry Smith k = PetscMax(A->rmap->n,A->cmap->n); 182787828ca2SBarry Smith ierr = PetscMalloc((k+1)*sizeof(PetscScalar),&a->mult_work);CHKERRQ(ierr); 18282d61bbb3SSatish Balay } 18292d61bbb3SSatish Balay work = a->mult_work; 18303447b6efSHong Zhang xtmp = x; 18312d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 18322d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 18332d61bbb3SSatish Balay ncols = n*bs; 183487828ca2SBarry Smith ierr = PetscMemzero(work,ncols*sizeof(PetscScalar));CHKERRQ(ierr); 18353447b6efSHong Zhang if (usecprow) { 18367b2bb3b9SHong Zhang xtmp = x + bs*ridx[i]; 18373447b6efSHong Zhang } 18383447b6efSHong Zhang Kernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,xtmp,v,work); 183971044d3cSBarry Smith /* BLASgemv_("T",&bs,&ncols,&_DOne,v,&bs,xtmp,&_One,&_DOne,work,&_One); */ 18402d61bbb3SSatish Balay v += n*bs2; 18413447b6efSHong Zhang if (!usecprow) xtmp += bs; 18422d61bbb3SSatish Balay workt = work; 18432d61bbb3SSatish Balay for (j=0; j<n; j++) { 18442d61bbb3SSatish Balay zb = z + bs*(*idx++); 18452d61bbb3SSatish Balay for (k=0; k<bs; k++) zb[k] += workt[k] ; 18462d61bbb3SSatish Balay workt += bs; 18472d61bbb3SSatish Balay } 18482d61bbb3SSatish Balay } 18492d61bbb3SSatish Balay } 18502d61bbb3SSatish Balay } 18513447b6efSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 18523447b6efSHong Zhang ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 1853dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz*a->bs2);CHKERRQ(ierr); 18542d61bbb3SSatish Balay PetscFunctionReturn(0); 18552d61bbb3SSatish Balay } 18562d61bbb3SSatish Balay 18574a2ae208SSatish Balay #undef __FUNCT__ 18584a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqBAIJ" 1859f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqBAIJ(Mat inA,PetscScalar alpha) 18602d61bbb3SSatish Balay { 18612d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)inA->data; 1862690b6cddSBarry Smith PetscInt totalnz = a->bs2*a->nz; 1863f4df32b1SMatthew Knepley PetscScalar oalpha = alpha; 1864efee365bSSatish Balay PetscErrorCode ierr; 18650805154bSBarry Smith PetscBLASInt one = 1,tnz = PetscBLASIntCast(totalnz); 18662d61bbb3SSatish Balay 18672d61bbb3SSatish Balay PetscFunctionBegin; 1868f4df32b1SMatthew Knepley BLASscal_(&tnz,&oalpha,a->a,&one); 1869efee365bSSatish Balay ierr = PetscLogFlops(totalnz);CHKERRQ(ierr); 18702d61bbb3SSatish Balay PetscFunctionReturn(0); 18712d61bbb3SSatish Balay } 18722d61bbb3SSatish Balay 18734a2ae208SSatish Balay #undef __FUNCT__ 18744a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqBAIJ" 1875dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqBAIJ(Mat A,NormType type,PetscReal *norm) 18762d61bbb3SSatish Balay { 18778a62d963SHong Zhang PetscErrorCode ierr; 18782d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 18793f1db9ecSBarry Smith MatScalar *v = a->a; 1880329f5518SBarry Smith PetscReal sum = 0.0; 1881d0f46423SBarry Smith PetscInt i,j,k,bs=A->rmap->bs,nz=a->nz,bs2=a->bs2,k1; 18822d61bbb3SSatish Balay 18832d61bbb3SSatish Balay PetscFunctionBegin; 18842d61bbb3SSatish Balay if (type == NORM_FROBENIUS) { 18852d61bbb3SSatish Balay for (i=0; i< bs2*nz; i++) { 1886aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 1887329f5518SBarry Smith sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 18882d61bbb3SSatish Balay #else 18892d61bbb3SSatish Balay sum += (*v)*(*v); v++; 18902d61bbb3SSatish Balay #endif 18912d61bbb3SSatish Balay } 18922d61bbb3SSatish Balay *norm = sqrt(sum); 18938a62d963SHong Zhang } else if (type == NORM_1) { /* maximum column sum */ 18948a62d963SHong Zhang PetscReal *tmp; 18958a62d963SHong Zhang PetscInt *bcol = a->j; 1896d0f46423SBarry Smith ierr = PetscMalloc((A->cmap->n+1)*sizeof(PetscReal),&tmp);CHKERRQ(ierr); 1897d0f46423SBarry Smith ierr = PetscMemzero(tmp,A->cmap->n*sizeof(PetscReal));CHKERRQ(ierr); 18988a62d963SHong Zhang for (i=0; i<nz; i++){ 18998a62d963SHong Zhang for (j=0; j<bs; j++){ 19008a62d963SHong Zhang k1 = bs*(*bcol) + j; /* column index */ 19018a62d963SHong Zhang for (k=0; k<bs; k++){ 19028a62d963SHong Zhang tmp[k1] += PetscAbsScalar(*v); v++; 19038a62d963SHong Zhang } 19048a62d963SHong Zhang } 19058a62d963SHong Zhang bcol++; 19068a62d963SHong Zhang } 19078a62d963SHong Zhang *norm = 0.0; 1908d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 19098a62d963SHong Zhang if (tmp[j] > *norm) *norm = tmp[j]; 19108a62d963SHong Zhang } 19118a62d963SHong Zhang ierr = PetscFree(tmp);CHKERRQ(ierr); 1912596552b5SBarry Smith } else if (type == NORM_INFINITY) { /* maximum row sum */ 1913596552b5SBarry Smith *norm = 0.0; 1914596552b5SBarry Smith for (k=0; k<bs; k++) { 191574f84c7bSSatish Balay for (j=0; j<a->mbs; j++) { 1916596552b5SBarry Smith v = a->a + bs2*a->i[j] + k; 1917596552b5SBarry Smith sum = 0.0; 1918596552b5SBarry Smith for (i=0; i<a->i[j+1]-a->i[j]; i++) { 19190e90e235SBarry Smith for (k1=0; k1<bs; k1++){ 1920596552b5SBarry Smith sum += PetscAbsScalar(*v); 1921596552b5SBarry Smith v += bs; 19222d61bbb3SSatish Balay } 19230e90e235SBarry Smith } 1924596552b5SBarry Smith if (sum > *norm) *norm = sum; 1925596552b5SBarry Smith } 1926596552b5SBarry Smith } 1927596552b5SBarry Smith } else { 192829bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"No support for this norm yet"); 19292d61bbb3SSatish Balay } 19302d61bbb3SSatish Balay PetscFunctionReturn(0); 19312d61bbb3SSatish Balay } 19322d61bbb3SSatish Balay 19332d61bbb3SSatish Balay 19344a2ae208SSatish Balay #undef __FUNCT__ 19354a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqBAIJ" 1936dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqBAIJ(Mat A,Mat B,PetscTruth* flg) 19372d61bbb3SSatish Balay { 19382d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ *)A->data,*b = (Mat_SeqBAIJ *)B->data; 1939dfbe8321SBarry Smith PetscErrorCode ierr; 19402d61bbb3SSatish Balay 19412d61bbb3SSatish Balay PetscFunctionBegin; 19422d61bbb3SSatish Balay /* If the matrix/block dimensions are not equal, or no of nonzeros or shift */ 1943d0f46423SBarry Smith if ((A->rmap->N != B->rmap->N) || (A->cmap->n != B->cmap->n) || (A->rmap->bs != B->rmap->bs)|| (a->nz != b->nz)) { 1944273d9f13SBarry Smith *flg = PETSC_FALSE; 1945273d9f13SBarry Smith PetscFunctionReturn(0); 19462d61bbb3SSatish Balay } 19472d61bbb3SSatish Balay 19482d61bbb3SSatish Balay /* if the a->i are the same */ 1949690b6cddSBarry Smith ierr = PetscMemcmp(a->i,b->i,(a->mbs+1)*sizeof(PetscInt),flg);CHKERRQ(ierr); 1950abc0a331SBarry Smith if (!*flg) { 19510f5bd95cSBarry Smith PetscFunctionReturn(0); 19522d61bbb3SSatish Balay } 19532d61bbb3SSatish Balay 19542d61bbb3SSatish Balay /* if a->j are the same */ 1955690b6cddSBarry Smith ierr = PetscMemcmp(a->j,b->j,(a->nz)*sizeof(PetscInt),flg);CHKERRQ(ierr); 1956abc0a331SBarry Smith if (!*flg) { 19570f5bd95cSBarry Smith PetscFunctionReturn(0); 19582d61bbb3SSatish Balay } 19592d61bbb3SSatish Balay /* if a->a are the same */ 1960d0f46423SBarry Smith ierr = PetscMemcmp(a->a,b->a,(a->nz)*(A->rmap->bs)*(B->rmap->bs)*sizeof(PetscScalar),flg);CHKERRQ(ierr); 19612d61bbb3SSatish Balay PetscFunctionReturn(0); 19622d61bbb3SSatish Balay 19632d61bbb3SSatish Balay } 19642d61bbb3SSatish Balay 19654a2ae208SSatish Balay #undef __FUNCT__ 19664a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqBAIJ" 1967dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqBAIJ(Mat A,Vec v) 19682d61bbb3SSatish Balay { 19692d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1970dfbe8321SBarry Smith PetscErrorCode ierr; 1971690b6cddSBarry Smith PetscInt i,j,k,n,row,bs,*ai,*aj,ambs,bs2; 197287828ca2SBarry Smith PetscScalar *x,zero = 0.0; 19733f1db9ecSBarry Smith MatScalar *aa,*aa_j; 19742d61bbb3SSatish Balay 19752d61bbb3SSatish Balay PetscFunctionBegin; 197629bbc08cSBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 1977d0f46423SBarry Smith bs = A->rmap->bs; 19782d61bbb3SSatish Balay aa = a->a; 19792d61bbb3SSatish Balay ai = a->i; 19802d61bbb3SSatish Balay aj = a->j; 19812d61bbb3SSatish Balay ambs = a->mbs; 19822d61bbb3SSatish Balay bs2 = a->bs2; 19832d61bbb3SSatish Balay 19842dcb1b2aSMatthew Knepley ierr = VecSet(v,zero);CHKERRQ(ierr); 19851ebc52fbSHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1986e1311b90SBarry Smith ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 1987d0f46423SBarry Smith if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 19882d61bbb3SSatish Balay for (i=0; i<ambs; i++) { 19892d61bbb3SSatish Balay for (j=ai[i]; j<ai[i+1]; j++) { 19902d61bbb3SSatish Balay if (aj[j] == i) { 19912d61bbb3SSatish Balay row = i*bs; 19922d61bbb3SSatish Balay aa_j = aa+j*bs2; 19932d61bbb3SSatish Balay for (k=0; k<bs2; k+=(bs+1),row++) x[row] = aa_j[k]; 19942d61bbb3SSatish Balay break; 19952d61bbb3SSatish Balay } 19962d61bbb3SSatish Balay } 19972d61bbb3SSatish Balay } 19981ebc52fbSHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 19992d61bbb3SSatish Balay PetscFunctionReturn(0); 20002d61bbb3SSatish Balay } 20012d61bbb3SSatish Balay 20024a2ae208SSatish Balay #undef __FUNCT__ 20034a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqBAIJ" 2004dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqBAIJ(Mat A,Vec ll,Vec rr) 20052d61bbb3SSatish Balay { 20062d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 200787828ca2SBarry Smith PetscScalar *l,*r,x,*li,*ri; 20083f1db9ecSBarry Smith MatScalar *aa,*v; 2009dfbe8321SBarry Smith PetscErrorCode ierr; 2010690b6cddSBarry Smith PetscInt i,j,k,lm,rn,M,m,n,*ai,*aj,mbs,tmp,bs,bs2; 20112d61bbb3SSatish Balay 20122d61bbb3SSatish Balay PetscFunctionBegin; 20132d61bbb3SSatish Balay ai = a->i; 20142d61bbb3SSatish Balay aj = a->j; 20152d61bbb3SSatish Balay aa = a->a; 2016d0f46423SBarry Smith m = A->rmap->n; 2017d0f46423SBarry Smith n = A->cmap->n; 2018d0f46423SBarry Smith bs = A->rmap->bs; 20192d61bbb3SSatish Balay mbs = a->mbs; 20202d61bbb3SSatish Balay bs2 = a->bs2; 20212d61bbb3SSatish Balay if (ll) { 20221ebc52fbSHong Zhang ierr = VecGetArray(ll,&l);CHKERRQ(ierr); 2023e1311b90SBarry Smith ierr = VecGetLocalSize(ll,&lm);CHKERRQ(ierr); 202429bbc08cSBarry Smith if (lm != m) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length"); 20252d61bbb3SSatish Balay for (i=0; i<mbs; i++) { /* for each block row */ 20262d61bbb3SSatish Balay M = ai[i+1] - ai[i]; 20272d61bbb3SSatish Balay li = l + i*bs; 20282d61bbb3SSatish Balay v = aa + bs2*ai[i]; 20292d61bbb3SSatish Balay for (j=0; j<M; j++) { /* for each block */ 20302d61bbb3SSatish Balay for (k=0; k<bs2; k++) { 20312d61bbb3SSatish Balay (*v++) *= li[k%bs]; 20322d61bbb3SSatish Balay } 20332d61bbb3SSatish Balay } 20342d61bbb3SSatish Balay } 20351ebc52fbSHong Zhang ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr); 2036efee365bSSatish Balay ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 20372d61bbb3SSatish Balay } 20382d61bbb3SSatish Balay 20392d61bbb3SSatish Balay if (rr) { 20401ebc52fbSHong Zhang ierr = VecGetArray(rr,&r);CHKERRQ(ierr); 2041e1311b90SBarry Smith ierr = VecGetLocalSize(rr,&rn);CHKERRQ(ierr); 204229bbc08cSBarry Smith if (rn != n) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length"); 20432d61bbb3SSatish Balay for (i=0; i<mbs; i++) { /* for each block row */ 20442d61bbb3SSatish Balay M = ai[i+1] - ai[i]; 20452d61bbb3SSatish Balay v = aa + bs2*ai[i]; 20462d61bbb3SSatish Balay for (j=0; j<M; j++) { /* for each block */ 20472d61bbb3SSatish Balay ri = r + bs*aj[ai[i]+j]; 20482d61bbb3SSatish Balay for (k=0; k<bs; k++) { 20492d61bbb3SSatish Balay x = ri[k]; 20502d61bbb3SSatish Balay for (tmp=0; tmp<bs; tmp++) (*v++) *= x; 20512d61bbb3SSatish Balay } 20522d61bbb3SSatish Balay } 20532d61bbb3SSatish Balay } 20541ebc52fbSHong Zhang ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr); 2055efee365bSSatish Balay ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 20562d61bbb3SSatish Balay } 20572d61bbb3SSatish Balay PetscFunctionReturn(0); 20582d61bbb3SSatish Balay } 20592d61bbb3SSatish Balay 20602d61bbb3SSatish Balay 20614a2ae208SSatish Balay #undef __FUNCT__ 20624a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqBAIJ" 2063dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqBAIJ(Mat A,MatInfoType flag,MatInfo *info) 20642d61bbb3SSatish Balay { 20652d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 20662d61bbb3SSatish Balay 20672d61bbb3SSatish Balay PetscFunctionBegin; 20682d61bbb3SSatish Balay info->block_size = a->bs2; 20692d61bbb3SSatish Balay info->nz_allocated = a->maxnz; 20702d61bbb3SSatish Balay info->nz_used = a->bs2*a->nz; 20712d61bbb3SSatish Balay info->nz_unneeded = (double)(info->nz_allocated - info->nz_used); 20722d61bbb3SSatish Balay info->assemblies = A->num_ass; 20732d61bbb3SSatish Balay info->mallocs = a->reallocs; 20747adad957SLisandro Dalcin info->memory = ((PetscObject)A)->mem; 20752d61bbb3SSatish Balay if (A->factor) { 20762d61bbb3SSatish Balay info->fill_ratio_given = A->info.fill_ratio_given; 20772d61bbb3SSatish Balay info->fill_ratio_needed = A->info.fill_ratio_needed; 20782d61bbb3SSatish Balay info->factor_mallocs = A->info.factor_mallocs; 20792d61bbb3SSatish Balay } else { 20802d61bbb3SSatish Balay info->fill_ratio_given = 0; 20812d61bbb3SSatish Balay info->fill_ratio_needed = 0; 20822d61bbb3SSatish Balay info->factor_mallocs = 0; 20832d61bbb3SSatish Balay } 20842d61bbb3SSatish Balay PetscFunctionReturn(0); 20852d61bbb3SSatish Balay } 20862d61bbb3SSatish Balay 20872d61bbb3SSatish Balay 20884a2ae208SSatish Balay #undef __FUNCT__ 20894a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqBAIJ" 2090dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqBAIJ(Mat A) 20912d61bbb3SSatish Balay { 20922d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 2093dfbe8321SBarry Smith PetscErrorCode ierr; 20942d61bbb3SSatish Balay 20952d61bbb3SSatish Balay PetscFunctionBegin; 2096549d3d68SSatish Balay ierr = PetscMemzero(a->a,a->bs2*a->i[a->mbs]*sizeof(MatScalar));CHKERRQ(ierr); 20972d61bbb3SSatish Balay PetscFunctionReturn(0); 20982d61bbb3SSatish Balay } 2099