1cac129eeSSatish Balay 2c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h> 306873bf2SBarry Smith #include <petsc-private/kernels/blockinvert.h> 4c6db04a5SJed Brown #include <petscbt.h> 5c6db04a5SJed Brown #include <petscblaslapack.h> 6d6232bceSMichael Lange #if defined(PETSC_THREADCOMM_ACTIVE) 7d6232bceSMichael Lange #include <petscthreadcomm.h> 8d6232bceSMichael Lange #endif 9cac129eeSSatish Balay 104a2ae208SSatish Balay #undef __FUNCT__ 114a2ae208SSatish Balay #define __FUNCT__ "MatIncreaseOverlap_SeqBAIJ" 12690b6cddSBarry Smith PetscErrorCode MatIncreaseOverlap_SeqBAIJ(Mat A,PetscInt is_max,IS is[],PetscInt ov) 13a3192f15SSatish Balay { 14a3192f15SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 156849ba73SBarry Smith PetscErrorCode ierr; 165d0c19d7SBarry Smith PetscInt row,i,j,k,l,m,n,*nidx,isz,val,ival; 175d0c19d7SBarry Smith const PetscInt *idx; 18690b6cddSBarry Smith PetscInt start,end,*ai,*aj,bs,*nidx2; 19f1af5d2fSBarry Smith PetscBT table; 20a3192f15SSatish Balay 213a40ed3dSBarry Smith PetscFunctionBegin; 22a3192f15SSatish Balay m = a->mbs; 23a3192f15SSatish Balay ai = a->i; 24a3192f15SSatish Balay aj = a->j; 25d0f46423SBarry Smith bs = A->rmap->bs; 26a3192f15SSatish Balay 27e32f2f54SBarry Smith if (ov < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative overlap specified"); 28a3192f15SSatish Balay 2953b8de81SBarry Smith ierr = PetscBTCreate(m,&table);CHKERRQ(ierr); 30854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&nidx);CHKERRQ(ierr); 31854ce69bSBarry Smith ierr = PetscMalloc1(A->rmap->N+1,&nidx2);CHKERRQ(ierr); 32a3192f15SSatish Balay 33a3192f15SSatish Balay for (i=0; i<is_max; i++) { 34a3192f15SSatish Balay /* Initialise the two local arrays */ 35a3192f15SSatish Balay isz = 0; 366831982aSBarry Smith ierr = PetscBTMemzero(m,table);CHKERRQ(ierr); 37a3192f15SSatish Balay 38a3192f15SSatish Balay /* Extract the indices, assume there can be duplicate entries */ 39a3192f15SSatish Balay ierr = ISGetIndices(is[i],&idx);CHKERRQ(ierr); 40b9b97703SBarry Smith ierr = ISGetLocalSize(is[i],&n);CHKERRQ(ierr); 41a3192f15SSatish Balay 42a3192f15SSatish Balay /* Enter these into the temp arrays i.e mark table[row], enter row into new index */ 43a3192f15SSatish Balay for (j=0; j<n; ++j) { 44218c64b6SSatish Balay ival = idx[j]/bs; /* convert the indices into block indices */ 45e32f2f54SBarry Smith if (ival>=m) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"index greater than mat-dim"); 4626fbe8dcSKarl Rupp if (!PetscBTLookupSet(table,ival)) nidx[isz++] = ival; 47a3192f15SSatish Balay } 48a3192f15SSatish Balay ierr = ISRestoreIndices(is[i],&idx);CHKERRQ(ierr); 496bf464f9SBarry Smith ierr = ISDestroy(&is[i]);CHKERRQ(ierr); 50a3192f15SSatish Balay 51a3192f15SSatish Balay k = 0; 52a3192f15SSatish Balay for (j=0; j<ov; j++) { /* for each overlap*/ 53a3192f15SSatish Balay n = isz; 54a3192f15SSatish Balay for (; k<n; k++) { /* do only those rows in nidx[k], which are not done yet */ 55a3192f15SSatish Balay row = nidx[k]; 56a3192f15SSatish Balay start = ai[row]; 57a3192f15SSatish Balay end = ai[row+1]; 58a3192f15SSatish Balay for (l = start; l<end; l++) { 59a3192f15SSatish Balay val = aj[l]; 6026fbe8dcSKarl Rupp if (!PetscBTLookupSet(table,val)) nidx[isz++] = val; 61a3192f15SSatish Balay } 62a3192f15SSatish Balay } 63a3192f15SSatish Balay } 64218c64b6SSatish Balay /* expand the Index Set */ 65218c64b6SSatish Balay for (j=0; j<isz; j++) { 6626fbe8dcSKarl Rupp for (k=0; k<bs; k++) nidx2[j*bs+k] = nidx[j]*bs+k; 67218c64b6SSatish Balay } 6870b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,isz*bs,nidx2,PETSC_COPY_VALUES,is+i);CHKERRQ(ierr); 69a3192f15SSatish Balay } 7094bacf5dSBarry Smith ierr = PetscBTDestroy(&table);CHKERRQ(ierr); 71606d414cSSatish Balay ierr = PetscFree(nidx);CHKERRQ(ierr); 72606d414cSSatish Balay ierr = PetscFree(nidx2);CHKERRQ(ierr); 733a40ed3dSBarry Smith PetscFunctionReturn(0); 74a3192f15SSatish Balay } 751c351548SSatish Balay 764a2ae208SSatish Balay #undef __FUNCT__ 774a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqBAIJ_Private" 784aa3045dSJed Brown PetscErrorCode MatGetSubMatrix_SeqBAIJ_Private(Mat A,IS isrow,IS iscol,MatReuse scall,Mat *B) 79736121d4SSatish Balay { 80736121d4SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data,*c; 816849ba73SBarry Smith PetscErrorCode ierr; 82690b6cddSBarry Smith PetscInt *smap,i,k,kstart,kend,oldcols = a->nbs,*lens; 83690b6cddSBarry Smith PetscInt row,mat_i,*mat_j,tcol,*mat_ilen; 845d0c19d7SBarry Smith const PetscInt *irow,*icol; 855d0c19d7SBarry Smith PetscInt nrows,ncols,*ssmap,bs=A->rmap->bs,bs2=a->bs2; 86690b6cddSBarry Smith PetscInt *aj = a->j,*ai = a->i; 873f1db9ecSBarry Smith MatScalar *mat_a; 88736121d4SSatish Balay Mat C; 896041f1b1SToby Isaac PetscBool flag; 90736121d4SSatish Balay 913a40ed3dSBarry Smith PetscFunctionBegin; 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 98854ce69bSBarry Smith ierr = PetscCalloc1(1+oldcols,&smap);CHKERRQ(ierr); 99736121d4SSatish Balay ssmap = smap; 100854ce69bSBarry Smith ierr = PetscMalloc1(1+nrows,&lens);CHKERRQ(ierr); 101736121d4SSatish Balay for (i=0; i<ncols; i++) smap[icol[i]] = i+1; 102736121d4SSatish Balay /* determine lens of each row */ 103736121d4SSatish Balay for (i=0; i<nrows; i++) { 104736121d4SSatish Balay kstart = ai[irow[i]]; 105736121d4SSatish Balay kend = kstart + a->ilen[irow[i]]; 106736121d4SSatish Balay lens[i] = 0; 107736121d4SSatish Balay for (k=kstart; k<kend; k++) { 10826fbe8dcSKarl Rupp if (ssmap[aj[k]]) lens[i]++; 109736121d4SSatish Balay } 110736121d4SSatish Balay } 111736121d4SSatish Balay /* Create and fill new matrix */ 112736121d4SSatish Balay if (scall == MAT_REUSE_MATRIX) { 113736121d4SSatish Balay c = (Mat_SeqBAIJ*)((*B)->data); 114736121d4SSatish Balay 115e32f2f54SBarry Smith if (c->mbs!=nrows || c->nbs!=ncols || (*B)->rmap->bs!=bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Submatrix wrong size"); 116690b6cddSBarry Smith ierr = PetscMemcmp(c->ilen,lens,c->mbs *sizeof(PetscInt),&flag);CHKERRQ(ierr); 117e7e72b3dSBarry Smith if (!flag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Cannot reuse matrix. wrong no of nonzeros"); 118690b6cddSBarry Smith ierr = PetscMemzero(c->ilen,c->mbs*sizeof(PetscInt));CHKERRQ(ierr); 119736121d4SSatish Balay C = *B; 1203a40ed3dSBarry Smith } else { 121ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 122f69a0ea3SMatthew Knepley ierr = MatSetSizes(C,nrows*bs,ncols*bs,PETSC_DETERMINE,PETSC_DETERMINE);CHKERRQ(ierr); 1237adad957SLisandro Dalcin ierr = MatSetType(C,((PetscObject)A)->type_name);CHKERRQ(ierr); 124ab93d7beSBarry Smith ierr = MatSeqBAIJSetPreallocation_SeqBAIJ(C,bs,0,lens);CHKERRQ(ierr); 125736121d4SSatish Balay } 126736121d4SSatish Balay c = (Mat_SeqBAIJ*)(C->data); 127736121d4SSatish Balay for (i=0; i<nrows; i++) { 128736121d4SSatish Balay row = irow[i]; 129736121d4SSatish Balay kstart = ai[row]; 130736121d4SSatish Balay kend = kstart + a->ilen[row]; 131736121d4SSatish Balay mat_i = c->i[i]; 132736121d4SSatish Balay mat_j = c->j + mat_i; 133218c64b6SSatish Balay mat_a = c->a + mat_i*bs2; 134736121d4SSatish Balay mat_ilen = c->ilen + i; 135736121d4SSatish Balay for (k=kstart; k<kend; k++) { 136736121d4SSatish Balay if ((tcol=ssmap[a->j[k]])) { 137736121d4SSatish Balay *mat_j++ = tcol - 1; 138549d3d68SSatish Balay ierr = PetscMemcpy(mat_a,a->a+k*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 139549d3d68SSatish Balay mat_a += bs2; 140736121d4SSatish Balay (*mat_ilen)++; 141736121d4SSatish Balay } 142736121d4SSatish Balay } 143736121d4SSatish Balay } 144cdc6f3adSToby Isaac /* sort */ 145cdc6f3adSToby Isaac { 146cdc6f3adSToby Isaac MatScalar *work; 147cdc6f3adSToby Isaac 148cdc6f3adSToby Isaac ierr = PetscMalloc1(bs2,&work);CHKERRQ(ierr); 149cdc6f3adSToby Isaac for (i=0; i<nrows; i++) { 150cdc6f3adSToby Isaac PetscInt ilen; 151cdc6f3adSToby Isaac mat_i = c->i[i]; 152cdc6f3adSToby Isaac mat_j = c->j + mat_i; 153cdc6f3adSToby Isaac mat_a = c->a + mat_i*bs2; 154cdc6f3adSToby Isaac ilen = c->ilen[i]; 155cdc6f3adSToby Isaac ierr = PetscSortIntWithDataArray(ilen,mat_j,mat_a,bs2*sizeof(MatScalar),work);CHKERRQ(ierr); 156cdc6f3adSToby Isaac } 157cdc6f3adSToby Isaac ierr = PetscFree(work);CHKERRQ(ierr); 158cdc6f3adSToby Isaac } 159218c64b6SSatish Balay 160736121d4SSatish Balay /* Free work space */ 161736121d4SSatish Balay ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 162606d414cSSatish Balay ierr = PetscFree(smap);CHKERRQ(ierr); 163606d414cSSatish Balay ierr = PetscFree(lens);CHKERRQ(ierr); 1646d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1656d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 166736121d4SSatish Balay 167736121d4SSatish Balay ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 168736121d4SSatish Balay *B = C; 1693a40ed3dSBarry Smith PetscFunctionReturn(0); 170736121d4SSatish Balay } 171736121d4SSatish Balay 1724a2ae208SSatish Balay #undef __FUNCT__ 1734a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqBAIJ" 1744aa3045dSJed Brown PetscErrorCode MatGetSubMatrix_SeqBAIJ(Mat A,IS isrow,IS iscol,MatReuse scall,Mat *B) 175218c64b6SSatish Balay { 176218c64b6SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 177218c64b6SSatish Balay IS is1,is2; 1786849ba73SBarry Smith PetscErrorCode ierr; 1795d0c19d7SBarry Smith PetscInt *vary,*iary,nrows,ncols,i,bs=A->rmap->bs,count; 1805d0c19d7SBarry Smith const PetscInt *irow,*icol; 181218c64b6SSatish Balay 1823a40ed3dSBarry Smith PetscFunctionBegin; 183218c64b6SSatish Balay ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr); 184218c64b6SSatish Balay ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr); 185b9b97703SBarry Smith ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr); 186b9b97703SBarry Smith ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr); 187218c64b6SSatish Balay 188218c64b6SSatish Balay /* Verify if the indices corespond to each element in a block 189218c64b6SSatish Balay and form the IS with compressed IS */ 190dcca6d9dSJed Brown ierr = PetscMalloc2(a->mbs,&vary,a->mbs,&iary);CHKERRQ(ierr); 191fca92195SBarry Smith ierr = PetscMemzero(vary,a->mbs*sizeof(PetscInt));CHKERRQ(ierr); 192218c64b6SSatish Balay for (i=0; i<nrows; i++) vary[irow[i]/bs]++; 193218c64b6SSatish Balay for (i=0; i<a->mbs; i++) { 194e32f2f54SBarry Smith if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Index set does not match blocks"); 1956041f1b1SToby Isaac } 1966041f1b1SToby Isaac count = 0; 1976041f1b1SToby Isaac for (i=0; i<nrows; i++) { 1986041f1b1SToby Isaac PetscInt j = irow[i] / bs; 1996041f1b1SToby Isaac if ((vary[j]--)==bs) iary[count++] = j; 200218c64b6SSatish Balay } 20170b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,PETSC_COPY_VALUES,&is1);CHKERRQ(ierr); 202218c64b6SSatish Balay 203690b6cddSBarry Smith ierr = PetscMemzero(vary,(a->mbs)*sizeof(PetscInt));CHKERRQ(ierr); 204218c64b6SSatish Balay for (i=0; i<ncols; i++) vary[icol[i]/bs]++; 205218c64b6SSatish Balay for (i=0; i<a->mbs; i++) { 206e32f2f54SBarry Smith if (vary[i]!=0 && vary[i]!=bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error in PETSc"); 207218c64b6SSatish Balay } 2086041f1b1SToby Isaac count = 0; 2096041f1b1SToby Isaac for (i=0; i<ncols; i++) { 2106041f1b1SToby Isaac PetscInt j = icol[i] / bs; 2116041f1b1SToby Isaac if ((vary[j]--)==bs) iary[count++] = j; 2126041f1b1SToby Isaac } 21370b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,count,iary,PETSC_COPY_VALUES,&is2);CHKERRQ(ierr); 214218c64b6SSatish Balay ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 215218c64b6SSatish Balay ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 216fca92195SBarry Smith ierr = PetscFree2(vary,iary);CHKERRQ(ierr); 217218c64b6SSatish Balay 2184aa3045dSJed Brown ierr = MatGetSubMatrix_SeqBAIJ_Private(A,is1,is2,scall,B);CHKERRQ(ierr); 2196bf464f9SBarry Smith ierr = ISDestroy(&is1);CHKERRQ(ierr); 2206bf464f9SBarry Smith ierr = ISDestroy(&is2);CHKERRQ(ierr); 2213a40ed3dSBarry Smith PetscFunctionReturn(0); 222218c64b6SSatish Balay } 223218c64b6SSatish Balay 2244a2ae208SSatish Balay #undef __FUNCT__ 2254a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqBAIJ" 226690b6cddSBarry Smith PetscErrorCode MatGetSubMatrices_SeqBAIJ(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[]) 227736121d4SSatish Balay { 2286849ba73SBarry Smith PetscErrorCode ierr; 229690b6cddSBarry Smith PetscInt i; 230736121d4SSatish Balay 2313a40ed3dSBarry Smith PetscFunctionBegin; 232736121d4SSatish Balay if (scall == MAT_INITIAL_MATRIX) { 233854ce69bSBarry Smith ierr = PetscMalloc1(n+1,B);CHKERRQ(ierr); 234736121d4SSatish Balay } 235736121d4SSatish Balay 236736121d4SSatish Balay for (i=0; i<n; i++) { 2374aa3045dSJed Brown ierr = MatGetSubMatrix_SeqBAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr); 238736121d4SSatish Balay } 2393a40ed3dSBarry Smith PetscFunctionReturn(0); 240736121d4SSatish Balay } 241218c64b6SSatish Balay 242218c64b6SSatish Balay 2432d61bbb3SSatish Balay /* -------------------------------------------------------*/ 2442d61bbb3SSatish Balay /* Should check that shapes of vectors and matrices match */ 2452d61bbb3SSatish Balay /* -------------------------------------------------------*/ 2462d61bbb3SSatish Balay 247d6232bceSMichael Lange #if defined(PETSC_THREADCOMM_ACTIVE) 248d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_1_Kernel(PetscInt thread_id,Mat A,Vec xx,Vec zz) 249d6232bceSMichael Lange { 250d6232bceSMichael Lange PetscErrorCode ierr; 251d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 252d6232bceSMichael Lange PetscScalar *z; 253d6232bceSMichael Lange const PetscScalar *x; 254d6232bceSMichael Lange const MatScalar *aa; 255d6232bceSMichael Lange PetscInt *trstarts=A->rmap->trstarts; 256d6232bceSMichael Lange PetscInt n,start,end,i; 257d6232bceSMichael Lange const PetscInt *aj,*ai; 258d6232bceSMichael Lange PetscScalar sum; 259d6232bceSMichael Lange 260d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 261d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 262d6232bceSMichael Lange start = trstarts[thread_id]; 263d6232bceSMichael Lange end = trstarts[thread_id+1]; 264d6232bceSMichael Lange ai = a->i; 265d6232bceSMichael Lange for (i=start; i<end; i++) { 266d6232bceSMichael Lange n = ai[i+1] - ai[i]; 267d6232bceSMichael Lange aj = a->j + ai[i]; 268d6232bceSMichael Lange aa = a->a + ai[i]; 269d6232bceSMichael Lange sum = 0.0; 270d6232bceSMichael Lange PetscSparseDensePlusDot(sum,x,aa,aj,n); 271d6232bceSMichael Lange z[i] = sum; 272d6232bceSMichael Lange } 273d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 274d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 275d6232bceSMichael Lange return 0; 276d6232bceSMichael Lange } 277d6232bceSMichael Lange 278d6232bceSMichael Lange #undef __FUNCT__ 279d6232bceSMichael Lange #define __FUNCT__ "MatMult_SeqBAIJ_1" 280d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_1(Mat A,Vec xx,Vec zz) 281d6232bceSMichael Lange { 282d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 283d6232bceSMichael Lange PetscScalar *z,sum; 284d6232bceSMichael Lange const PetscScalar *x; 285d6232bceSMichael Lange const MatScalar *v; 286d6232bceSMichael Lange PetscErrorCode ierr; 287d6232bceSMichael Lange PetscInt mbs,i,n; 288d6232bceSMichael Lange const PetscInt *idx,*ii,*ridx=NULL; 289d6232bceSMichael Lange PetscBool usecprow=a->compressedrow.use; 290d6232bceSMichael Lange 291d6232bceSMichael Lange PetscFunctionBegin; 292d6232bceSMichael Lange 293d6232bceSMichael Lange if (usecprow) { 294d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 295d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 296d6232bceSMichael Lange mbs = a->compressedrow.nrows; 297d6232bceSMichael Lange ii = a->compressedrow.i; 298d6232bceSMichael Lange ridx = a->compressedrow.rindex; 299d6232bceSMichael Lange ierr = PetscMemzero(z,mbs*sizeof(PetscScalar));CHKERRQ(ierr); 300d6232bceSMichael Lange for (i=0; i<mbs; i++) { 301d6232bceSMichael Lange n = ii[i+1] - ii[i]; 302d6232bceSMichael Lange v = a->a + ii[i]; 303d6232bceSMichael Lange idx = a->j + ii[i]; 304d6232bceSMichael Lange PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 305d6232bceSMichael Lange PetscPrefetchBlock(v+1*n,1*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 306d6232bceSMichael Lange sum = 0.0; 307d6232bceSMichael Lange PetscSparseDensePlusDot(sum,x,v,idx,n); 308d6232bceSMichael Lange z[ridx[i]] = sum; 309d6232bceSMichael Lange } 310d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 311d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 312d6232bceSMichael Lange } else { 313d6232bceSMichael Lange ierr = PetscThreadCommRunKernel(PetscObjectComm((PetscObject)A),(PetscThreadKernel)MatMult_SeqBAIJ_1_Kernel,3,A,xx,zz);CHKERRQ(ierr); 314d6232bceSMichael Lange } 315d6232bceSMichael Lange ierr = PetscLogFlops(2.0*a->nz - a->nonzerorowcnt);CHKERRQ(ierr); 316d6232bceSMichael Lange PetscFunctionReturn(0); 317d6232bceSMichael Lange } 318d6232bceSMichael Lange #else 3194a2ae208SSatish Balay #undef __FUNCT__ 3204a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_1" 321dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_1(Mat A,Vec xx,Vec zz) 3222d61bbb3SSatish Balay { 3232d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 324d9fead3dSBarry Smith PetscScalar *z,sum; 325d9fead3dSBarry Smith const PetscScalar *x; 326d9fead3dSBarry Smith const MatScalar *v; 3276849ba73SBarry Smith PetscErrorCode ierr; 3287c565772SBarry Smith PetscInt mbs,i,n; 3290298fd71SBarry Smith const PetscInt *idx,*ii,*ridx=NULL; 330ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 3312d61bbb3SSatish Balay 3322d61bbb3SSatish Balay PetscFunctionBegin; 3333649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 3341ebc52fbSHong Zhang ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 3352d61bbb3SSatish Balay 33626e093fcSHong Zhang if (usecprow) { 33726e093fcSHong Zhang mbs = a->compressedrow.nrows; 33826e093fcSHong Zhang ii = a->compressedrow.i; 3397b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 340a1c3900fSBarry Smith ierr = PetscMemzero(z,mbs*sizeof(PetscScalar));CHKERRQ(ierr); 34126e093fcSHong Zhang } else { 34226e093fcSHong Zhang mbs = a->mbs; 3432d61bbb3SSatish Balay ii = a->i; 34426e093fcSHong Zhang } 3452d61bbb3SSatish Balay 3462d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 347ee54c7eeSHong Zhang n = ii[1] - ii[0]; 348ee54c7eeSHong Zhang v = a->a + ii[0]; 349ee54c7eeSHong Zhang idx = a->j + ii[0]; 350ee54c7eeSHong Zhang ii++; 351444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 352444d8c10SJed Brown PetscPrefetchBlock(v+1*n,1*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 3532d61bbb3SSatish Balay sum = 0.0; 3542162cab8SBarry Smith PetscSparseDensePlusDot(sum,x,v,idx,n); 35526e093fcSHong Zhang if (usecprow) { 3567b2bb3b9SHong Zhang z[ridx[i]] = sum; 35726e093fcSHong Zhang } else { 3582d61bbb3SSatish Balay z[i] = sum; 3592d61bbb3SSatish Balay } 36026e093fcSHong Zhang } 3613649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 3621ebc52fbSHong Zhang ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 3637c565772SBarry Smith ierr = PetscLogFlops(2.0*a->nz - a->nonzerorowcnt);CHKERRQ(ierr); 3642d61bbb3SSatish Balay PetscFunctionReturn(0); 3652d61bbb3SSatish Balay } 366d6232bceSMichael Lange #endif 3672d61bbb3SSatish Balay 368d6232bceSMichael Lange #if defined(PETSC_THREADCOMM_ACTIVE) 369d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_2_Kernel(PetscInt thread_id,Mat A,Vec xx,Vec zz) 370d6232bceSMichael Lange { 371d6232bceSMichael Lange PetscErrorCode ierr; 372d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 373d6232bceSMichael Lange PetscScalar *z,x1,x2,sum1,sum2; 374d6232bceSMichael Lange const PetscScalar *x,*xb; 375d6232bceSMichael Lange const MatScalar *aa; 376d6232bceSMichael Lange PetscInt *trstarts=A->rmap->trstarts; 377d6232bceSMichael Lange PetscInt n,start,end,i,j; 378d6232bceSMichael Lange const PetscInt *aj,*ai; 379d6232bceSMichael Lange 380d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 381d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 382d6232bceSMichael Lange start = trstarts[thread_id] / 2; 383d6232bceSMichael Lange end = trstarts[thread_id+1] / 2; 384d6232bceSMichael Lange ai = a->i; 385d6232bceSMichael Lange for (i=start; i<end; i++) { 386d6232bceSMichael Lange n = ai[i+1] - ai[i]; 387d6232bceSMichael Lange aj = a->j + ai[i]; 388d6232bceSMichael Lange aa = a->a + ai[i]*4; 389d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; 390d6232bceSMichael Lange for (j=0; j<n; j++) { 391d6232bceSMichael Lange xb = x + 2*aj[j]; x1 = xb[0]; x2 = xb[1]; 392d6232bceSMichael Lange sum1 += aa[4*j]*x1 + aa[4*j+2]*x2; 393d6232bceSMichael Lange sum2 += aa[4*j+1]*x1 + aa[4*j+3]*x2; 394d6232bceSMichael Lange } 395d6232bceSMichael Lange z[2*i] = sum1; z[2*i+1] = sum2; 396d6232bceSMichael Lange } 397d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 398d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 399d6232bceSMichael Lange return 0; 400d6232bceSMichael Lange } 401d6232bceSMichael Lange 402d6232bceSMichael Lange #undef __FUNCT__ 403d6232bceSMichael Lange #define __FUNCT__ "MatMult_SeqBAIJ_2" 404d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_2(Mat A,Vec xx,Vec zz) 405d6232bceSMichael Lange { 406d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 407d6232bceSMichael Lange PetscScalar *z,x1,x2,sum1,sum2; 408d6232bceSMichael Lange const PetscScalar *x,*xb; 409d6232bceSMichael Lange const MatScalar *v; 410d6232bceSMichael Lange PetscErrorCode ierr; 411d6232bceSMichael Lange PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 412d6232bceSMichael Lange PetscBool usecprow=a->compressedrow.use; 413d6232bceSMichael Lange 414d6232bceSMichael Lange PetscFunctionBegin; 415d6232bceSMichael Lange 416d6232bceSMichael Lange if (usecprow) { 417d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 418d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 419d6232bceSMichael Lange mbs = a->compressedrow.nrows; 420d6232bceSMichael Lange ii = a->compressedrow.i; 421d6232bceSMichael Lange ridx = a->compressedrow.rindex; 422d6232bceSMichael Lange for (i=0; i<mbs; i++) { 423d6232bceSMichael Lange n = ii[i+1] - ii[i]; 424d6232bceSMichael Lange idx = a->j + ii[i]; 425d6232bceSMichael Lange v = a->a + ii[i]*4; 426d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; 427d6232bceSMichael Lange PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 428d6232bceSMichael Lange PetscPrefetchBlock(v+4*n,4*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 429d6232bceSMichael Lange for (j=0; j<n; j++) { 430d6232bceSMichael Lange xb = x + 2*idx[j]; x1 = xb[0]; x2 = xb[1]; 431d6232bceSMichael Lange sum1 += v[4*j]*x1 + v[4*j+2]*x2; 432d6232bceSMichael Lange sum2 += v[4*j+1]*x1 + v[4*j+3]*x2; 433d6232bceSMichael Lange } 434d6232bceSMichael Lange z[2*ridx[i]] = sum1; z[2*ridx[i]+1] = sum2; 435d6232bceSMichael Lange } 436d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 437d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 438d6232bceSMichael Lange } else { 439d6232bceSMichael Lange ierr = PetscThreadCommRunKernel(PetscObjectComm((PetscObject)A),(PetscThreadKernel)MatMult_SeqBAIJ_2_Kernel,3,A,xx,zz);CHKERRQ(ierr); 440d6232bceSMichael Lange } 441d6232bceSMichael Lange ierr = PetscLogFlops(8.0*a->nz - 2.0*a->nonzerorowcnt);CHKERRQ(ierr); 442d6232bceSMichael Lange PetscFunctionReturn(0); 443d6232bceSMichael Lange } 444d6232bceSMichael Lange #else 4454a2ae208SSatish Balay #undef __FUNCT__ 4464a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_2" 447dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_2(Mat A,Vec xx,Vec zz) 4482d61bbb3SSatish Balay { 4492d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 450d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,*zarray; 451d9fead3dSBarry Smith const PetscScalar *x,*xb; 45287828ca2SBarry Smith PetscScalar x1,x2; 453d9fead3dSBarry Smith const MatScalar *v; 454dfbe8321SBarry Smith PetscErrorCode ierr; 4557c565772SBarry Smith PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 456ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 4572d61bbb3SSatish Balay 4582d61bbb3SSatish Balay PetscFunctionBegin; 4593649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 46026e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 4612d61bbb3SSatish Balay 4622d61bbb3SSatish Balay idx = a->j; 4632d61bbb3SSatish Balay v = a->a; 46426e093fcSHong Zhang if (usecprow) { 46526e093fcSHong Zhang mbs = a->compressedrow.nrows; 46626e093fcSHong Zhang ii = a->compressedrow.i; 4677b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 46826e093fcSHong Zhang } else { 46926e093fcSHong Zhang mbs = a->mbs; 4702d61bbb3SSatish Balay ii = a->i; 47126e093fcSHong Zhang z = zarray; 47226e093fcSHong Zhang } 4732d61bbb3SSatish Balay 4742d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 4752d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 4762d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; 477444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 478444d8c10SJed Brown PetscPrefetchBlock(v+4*n,4*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 4792d61bbb3SSatish Balay for (j=0; j<n; j++) { 4802d61bbb3SSatish Balay xb = x + 2*(*idx++); x1 = xb[0]; x2 = xb[1]; 4812d61bbb3SSatish Balay sum1 += v[0]*x1 + v[2]*x2; 4822d61bbb3SSatish Balay sum2 += v[1]*x1 + v[3]*x2; 4832d61bbb3SSatish Balay v += 4; 4842d61bbb3SSatish Balay } 4857b2bb3b9SHong Zhang if (usecprow) z = zarray + 2*ridx[i]; 4862d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; 48726e093fcSHong Zhang if (!usecprow) z += 2; 4882d61bbb3SSatish Balay } 4893649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 49026e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 4917c565772SBarry Smith ierr = PetscLogFlops(8.0*a->nz - 2.0*a->nonzerorowcnt);CHKERRQ(ierr); 4922d61bbb3SSatish Balay PetscFunctionReturn(0); 4932d61bbb3SSatish Balay } 494d6232bceSMichael Lange #endif 4952d61bbb3SSatish Balay 496d6232bceSMichael Lange #if defined(PETSC_THREADCOMM_ACTIVE) 497d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_3_Kernel(PetscInt thread_id,Mat A,Vec xx,Vec zz) 498d6232bceSMichael Lange { 499d6232bceSMichael Lange PetscErrorCode ierr; 500d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 501d6232bceSMichael Lange PetscScalar *z,x1,x2,x3,sum1,sum2,sum3; 502d6232bceSMichael Lange const PetscScalar *x,*xb; 503d6232bceSMichael Lange const MatScalar *aa; 504d6232bceSMichael Lange PetscInt *trstarts=A->rmap->trstarts; 505d6232bceSMichael Lange PetscInt n,start,end,i,j; 506d6232bceSMichael Lange const PetscInt *aj,*ai; 507d6232bceSMichael Lange 508d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 509d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 510d6232bceSMichael Lange start = trstarts[thread_id] / 3; 511d6232bceSMichael Lange end = trstarts[thread_id+1] / 3; 512d6232bceSMichael Lange ai = a->i; 513d6232bceSMichael Lange for (i=start; i<end; i++) { 514d6232bceSMichael Lange n = ai[i+1] - ai[i]; 515d6232bceSMichael Lange aj = a->j + ai[i]; 516d6232bceSMichael Lange aa = a->a + ai[i]*9; 517d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; 518d6232bceSMichael Lange for (j=0; j<n; j++) { 519d6232bceSMichael Lange xb = x + 3*aj[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 520d6232bceSMichael Lange sum1 += aa[9*j]*x1 + aa[9*j+3]*x2 + aa[9*j+6]*x3; 521d6232bceSMichael Lange sum2 += aa[9*j+1]*x1 + aa[9*j+4]*x2 + aa[9*j+7]*x3; 522d6232bceSMichael Lange sum3 += aa[9*j+2]*x1 + aa[9*j+5]*x2 + aa[9*j+8]*x3; 523d6232bceSMichael Lange } 524d6232bceSMichael Lange z[3*i] = sum1; z[3*i+1] = sum2; z[3*i+2] = sum3; 525d6232bceSMichael Lange } 526d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 527d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 528d6232bceSMichael Lange return 0; 529d6232bceSMichael Lange } 530d6232bceSMichael Lange 531d6232bceSMichael Lange #undef __FUNCT__ 532d6232bceSMichael Lange #define __FUNCT__ "MatMult_SeqBAIJ_3" 533d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_3(Mat A,Vec xx,Vec zz) 534d6232bceSMichael Lange { 535d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 536d6232bceSMichael Lange PetscScalar *z,sum1,sum2,sum3,x1,x2,x3; 537d6232bceSMichael Lange const PetscScalar *x,*xb; 538d6232bceSMichael Lange const MatScalar *v; 539d6232bceSMichael Lange PetscErrorCode ierr; 540d6232bceSMichael Lange PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 541d6232bceSMichael Lange PetscBool usecprow=a->compressedrow.use; 542d6232bceSMichael Lange 543d6232bceSMichael Lange 544d6232bceSMichael Lange #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 545d6232bceSMichael Lange #pragma disjoint(*v,*z,*xb) 546d6232bceSMichael Lange #endif 547d6232bceSMichael Lange 548d6232bceSMichael Lange PetscFunctionBegin; 549d6232bceSMichael Lange 550d6232bceSMichael Lange if (usecprow) { 551d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 552d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 553d6232bceSMichael Lange mbs = a->compressedrow.nrows; 554d6232bceSMichael Lange ii = a->compressedrow.i; 555d6232bceSMichael Lange ridx = a->compressedrow.rindex; 556d6232bceSMichael Lange for (i=0; i<mbs; i++) { 557d6232bceSMichael Lange n = ii[i+1] - ii[i]; 558d6232bceSMichael Lange idx = a->j + ii[i]; 559d6232bceSMichael Lange v = a->a + ii[i]*9; 560d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; 561d6232bceSMichael Lange PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 562d6232bceSMichael Lange PetscPrefetchBlock(v+9*n,9*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 563d6232bceSMichael Lange for (j=0; j<n; j++) { 564d6232bceSMichael Lange xb = x + 3*idx[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 565d6232bceSMichael Lange sum1 += v[9*j]*x1 + v[9*j+3]*x2 + v[9*j+6]*x3; 566d6232bceSMichael Lange sum2 += v[9*j+1]*x1 + v[9*j+4]*x2 + v[9*j+7]*x3; 567d6232bceSMichael Lange sum3 += v[9*j+2]*x1 + v[9*j+5]*x2 + v[9*j+8]*x3; 568d6232bceSMichael Lange } 569d6232bceSMichael Lange z[3*ridx[i]] = sum1; z[3*ridx[i]+1] = sum2; z[3*ridx[i]+2] = sum3; 570d6232bceSMichael Lange } 571d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 572d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 573d6232bceSMichael Lange } else { 574d6232bceSMichael Lange ierr = PetscThreadCommRunKernel(PetscObjectComm((PetscObject)A),(PetscThreadKernel)MatMult_SeqBAIJ_3_Kernel,3,A,xx,zz);CHKERRQ(ierr); 575d6232bceSMichael Lange } 576d6232bceSMichael Lange ierr = PetscLogFlops(18.0*a->nz - 3.0*a->nonzerorowcnt);CHKERRQ(ierr); 577d6232bceSMichael Lange PetscFunctionReturn(0); 578d6232bceSMichael Lange } 579d6232bceSMichael Lange #else 5804a2ae208SSatish Balay #undef __FUNCT__ 5814a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_3" 582dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_3(Mat A,Vec xx,Vec zz) 5832d61bbb3SSatish Balay { 5842d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 585d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,x1,x2,x3,*zarray; 586d9fead3dSBarry Smith const PetscScalar *x,*xb; 587d9fead3dSBarry Smith const MatScalar *v; 588dfbe8321SBarry Smith PetscErrorCode ierr; 5897c565772SBarry Smith PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 590ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 59126e093fcSHong Zhang 592b6410449SSatish Balay #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 593fee21e36SBarry Smith #pragma disjoint(*v,*z,*xb) 594fee21e36SBarry Smith #endif 595fee21e36SBarry Smith 5962d61bbb3SSatish Balay PetscFunctionBegin; 5973649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 59826e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 5992d61bbb3SSatish Balay 6002d61bbb3SSatish Balay idx = a->j; 6012d61bbb3SSatish Balay v = a->a; 60226e093fcSHong Zhang if (usecprow) { 60326e093fcSHong Zhang mbs = a->compressedrow.nrows; 60426e093fcSHong Zhang ii = a->compressedrow.i; 6057b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 60626e093fcSHong Zhang } else { 60726e093fcSHong Zhang mbs = a->mbs; 6082d61bbb3SSatish Balay ii = a->i; 60926e093fcSHong Zhang z = zarray; 61026e093fcSHong Zhang } 6112d61bbb3SSatish Balay 6122d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 6132d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 6142d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; 615444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 616444d8c10SJed Brown PetscPrefetchBlock(v+9*n,9*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 6172d61bbb3SSatish Balay for (j=0; j<n; j++) { 61826fbe8dcSKarl Rupp xb = x + 3*(*idx++); 61926fbe8dcSKarl Rupp x1 = xb[0]; 62026fbe8dcSKarl Rupp x2 = xb[1]; 62126fbe8dcSKarl Rupp x3 = xb[2]; 62226fbe8dcSKarl Rupp 6232d61bbb3SSatish Balay sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3; 6242d61bbb3SSatish Balay sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3; 6252d61bbb3SSatish Balay sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3; 6262d61bbb3SSatish Balay v += 9; 6272d61bbb3SSatish Balay } 6287b2bb3b9SHong Zhang if (usecprow) z = zarray + 3*ridx[i]; 6292d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; 63026e093fcSHong Zhang if (!usecprow) z += 3; 6312d61bbb3SSatish Balay } 6323649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 63326e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 6347c565772SBarry Smith ierr = PetscLogFlops(18.0*a->nz - 3.0*a->nonzerorowcnt);CHKERRQ(ierr); 6352d61bbb3SSatish Balay PetscFunctionReturn(0); 6362d61bbb3SSatish Balay } 637d6232bceSMichael Lange #endif 6382d61bbb3SSatish Balay 639d6232bceSMichael Lange #if defined(PETSC_THREADCOMM_ACTIVE) 640d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_4_Kernel(PetscInt thread_id,Mat A,Vec xx,Vec zz) 641d6232bceSMichael Lange { 642d6232bceSMichael Lange PetscErrorCode ierr; 643d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 644d6232bceSMichael Lange PetscScalar *z,x1,x2,x3,x4,sum1,sum2,sum3,sum4; 645d6232bceSMichael Lange const PetscScalar *x,*xb; 646d6232bceSMichael Lange const MatScalar *aa; 647d6232bceSMichael Lange PetscInt *trstarts=A->rmap->trstarts; 648d6232bceSMichael Lange PetscInt n,start,end,i,j; 649d6232bceSMichael Lange const PetscInt *aj,*ai; 650d6232bceSMichael Lange 651d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 652d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 653d6232bceSMichael Lange start = trstarts[thread_id] / 4; 654d6232bceSMichael Lange end = trstarts[thread_id+1] / 4; 655d6232bceSMichael Lange ai = a->i; 656d6232bceSMichael Lange for (i=start; i<end; i++) { 657d6232bceSMichael Lange n = ai[i+1] - ai[i]; 658d6232bceSMichael Lange aj = a->j + ai[i]; 659d6232bceSMichael Lange aa = a->a + ai[i]*16; 660d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; 661d6232bceSMichael Lange for (j=0; j<n; j++) { 662d6232bceSMichael Lange xb = x + 4*aj[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 663d6232bceSMichael Lange sum1 += aa[16*j]*x1 + aa[16*j+4]*x2 + aa[16*j+8]*x3 + aa[16*j+12]*x4; 664d6232bceSMichael Lange sum2 += aa[16*j+1]*x1 + aa[16*j+5]*x2 + aa[16*j+9]*x3 + aa[16*j+13]*x4; 665d6232bceSMichael Lange sum3 += aa[16*j+2]*x1 + aa[16*j+6]*x2 + aa[16*j+10]*x3 + aa[16*j+14]*x4; 666d6232bceSMichael Lange sum4 += aa[16*j+3]*x1 + aa[16*j+7]*x2 + aa[16*j+11]*x3 + aa[16*j+15]*x4; 667d6232bceSMichael Lange } 668d6232bceSMichael Lange z[4*i] = sum1; z[4*i+1] = sum2; 669d6232bceSMichael Lange z[4*i+2] = sum3; z[4*i+3] = sum4; 670d6232bceSMichael Lange } 671d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 672d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 673d6232bceSMichael Lange return 0; 674d6232bceSMichael Lange } 675d6232bceSMichael Lange 676d6232bceSMichael Lange #undef __FUNCT__ 677d6232bceSMichael Lange #define __FUNCT__ "MatMult_SeqBAIJ_4" 678d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_4(Mat A,Vec xx,Vec zz) 679d6232bceSMichael Lange { 680d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 681d6232bceSMichael Lange PetscScalar *z,x1,x2,x3,x4,sum1,sum2,sum3,sum4; 682d6232bceSMichael Lange const PetscScalar *x,*xb; 683d6232bceSMichael Lange const MatScalar *v; 684d6232bceSMichael Lange PetscErrorCode ierr; 685d6232bceSMichael Lange PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 686d6232bceSMichael Lange PetscBool usecprow=a->compressedrow.use; 687d6232bceSMichael Lange 688d6232bceSMichael Lange PetscFunctionBegin; 689d6232bceSMichael Lange 690d6232bceSMichael Lange if (usecprow) { 691d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 692d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 693d6232bceSMichael Lange mbs = a->compressedrow.nrows; 694d6232bceSMichael Lange ii = a->compressedrow.i; 695d6232bceSMichael Lange ridx = a->compressedrow.rindex; 696d6232bceSMichael Lange for (i=0; i<mbs; i++) { 697d6232bceSMichael Lange n = ii[i+1] - ii[1]; 698d6232bceSMichael Lange idx = a->j + ii[i]; 699d6232bceSMichael Lange v = a->a + ii[i]*16; 700d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; 701d6232bceSMichael Lange PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 702d6232bceSMichael Lange PetscPrefetchBlock(v+16*n,16*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 703d6232bceSMichael Lange for (j=0; j<n; j++) { 704d6232bceSMichael Lange xb = x + 4*idx[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 705d6232bceSMichael Lange sum1 += v[16*j]*x1 + v[16*j+4]*x2 + v[16*j+8]*x3 + v[16*j+12]*x4; 706d6232bceSMichael Lange sum2 += v[16*j+1]*x1 + v[16*j+5]*x2 + v[16*j+9]*x3 + v[16*j+13]*x4; 707d6232bceSMichael Lange sum3 += v[16*j+2]*x1 + v[16*j+6]*x2 + v[16*j+10]*x3 + v[16*j+14]*x4; 708d6232bceSMichael Lange sum4 += v[16*j+3]*x1 + v[16*j+7]*x2 + v[16*j+11]*x3 + v[16*j+15]*x4; 709d6232bceSMichael Lange } 710d6232bceSMichael Lange z[4*ridx[i]] = sum1; z[4*ridx[i]+1] = sum2; 711d6232bceSMichael Lange z[4*ridx[i]+2] = sum3; z[4*ridx[i]+3] = sum4; 712d6232bceSMichael Lange } 713d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 714d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 715d6232bceSMichael Lange } else { 716d6232bceSMichael Lange ierr = PetscThreadCommRunKernel(PetscObjectComm((PetscObject)A),(PetscThreadKernel)MatMult_SeqBAIJ_4_Kernel,3,A,xx,zz);CHKERRQ(ierr); 717d6232bceSMichael Lange } 718d6232bceSMichael Lange ierr = PetscLogFlops(32.0*a->nz - 4.0*a->nonzerorowcnt);CHKERRQ(ierr); 719d6232bceSMichael Lange PetscFunctionReturn(0); 720d6232bceSMichael Lange } 721d6232bceSMichael Lange #else 7224a2ae208SSatish Balay #undef __FUNCT__ 7234a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_4" 724dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_4(Mat A,Vec xx,Vec zz) 7252d61bbb3SSatish Balay { 7262d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 727d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*zarray; 728d9fead3dSBarry Smith const PetscScalar *x,*xb; 729d9fead3dSBarry Smith const MatScalar *v; 730dfbe8321SBarry Smith PetscErrorCode ierr; 7317c565772SBarry Smith PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 732ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 7332d61bbb3SSatish Balay 7342d61bbb3SSatish Balay PetscFunctionBegin; 7353649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 73626e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 7372d61bbb3SSatish Balay 7382d61bbb3SSatish Balay idx = a->j; 7392d61bbb3SSatish Balay v = a->a; 74026e093fcSHong Zhang if (usecprow) { 74126e093fcSHong Zhang mbs = a->compressedrow.nrows; 74226e093fcSHong Zhang ii = a->compressedrow.i; 7437b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 74426e093fcSHong Zhang } else { 74526e093fcSHong Zhang mbs = a->mbs; 7462d61bbb3SSatish Balay ii = a->i; 74726e093fcSHong Zhang z = zarray; 74826e093fcSHong Zhang } 7492d61bbb3SSatish Balay 7502d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 75126fbe8dcSKarl Rupp n = ii[1] - ii[0]; 75226fbe8dcSKarl Rupp ii++; 75326fbe8dcSKarl Rupp sum1 = 0.0; 75426fbe8dcSKarl Rupp sum2 = 0.0; 75526fbe8dcSKarl Rupp sum3 = 0.0; 75626fbe8dcSKarl Rupp sum4 = 0.0; 75726fbe8dcSKarl Rupp 758444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 759444d8c10SJed Brown PetscPrefetchBlock(v+16*n,16*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 7602d61bbb3SSatish Balay for (j=0; j<n; j++) { 7612d61bbb3SSatish Balay xb = x + 4*(*idx++); 7622d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 7632d61bbb3SSatish Balay sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3 + v[12]*x4; 7642d61bbb3SSatish Balay sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3 + v[13]*x4; 7652d61bbb3SSatish Balay sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3 + v[14]*x4; 7662d61bbb3SSatish Balay sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3 + v[15]*x4; 7672d61bbb3SSatish Balay v += 16; 7682d61bbb3SSatish Balay } 7697b2bb3b9SHong Zhang if (usecprow) z = zarray + 4*ridx[i]; 7702d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; 77126e093fcSHong Zhang if (!usecprow) z += 4; 7722d61bbb3SSatish Balay } 7733649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 77426e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 7757c565772SBarry Smith ierr = PetscLogFlops(32.0*a->nz - 4.0*a->nonzerorowcnt);CHKERRQ(ierr); 7762d61bbb3SSatish Balay PetscFunctionReturn(0); 7772d61bbb3SSatish Balay } 778d6232bceSMichael Lange #endif 7792d61bbb3SSatish Balay 780d6232bceSMichael Lange #if defined(PETSC_THREADCOMM_ACTIVE) 781d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_5_Kernel(PetscInt thread_id,Mat A,Vec xx,Vec zz) 782d6232bceSMichael Lange { 783d6232bceSMichael Lange PetscErrorCode ierr; 784d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 785d6232bceSMichael Lange PetscScalar *z,x1,x2,x3,x4,x5,sum1,sum2,sum3,sum4,sum5; 786d6232bceSMichael Lange const PetscScalar *x,*xb; 787d6232bceSMichael Lange const MatScalar *aa; 788d6232bceSMichael Lange PetscInt *trstarts=A->rmap->trstarts; 789d6232bceSMichael Lange PetscInt n,start,end,i,j; 790d6232bceSMichael Lange const PetscInt *aj,*ai; 791d6232bceSMichael Lange 792d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 793d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 794d6232bceSMichael Lange start = trstarts[thread_id] / 5; 795d6232bceSMichael Lange end = trstarts[thread_id+1] / 5; 796d6232bceSMichael Lange ai = a->i; 797d6232bceSMichael Lange for (i=start; i<end; i++) { 798d6232bceSMichael Lange n = ai[i+1] - ai[i]; 799d6232bceSMichael Lange aj = a->j + ai[i]; 800d6232bceSMichael Lange aa = a->a + ai[i]*25; 801d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; 802d6232bceSMichael Lange for (j=0; j<n; j++) { 803d6232bceSMichael Lange xb = x + 5*aj[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; 804d6232bceSMichael Lange sum1 += aa[25*j]*x1 + aa[25*j+5]*x2 + aa[25*j+10]*x3 + aa[25*j+15]*x4 + aa[25*j+20]*x5; 805d6232bceSMichael Lange sum2 += aa[25*j+1]*x1 + aa[25*j+6]*x2 + aa[25*j+11]*x3 + aa[25*j+16]*x4 + aa[25*j+21]*x5; 806d6232bceSMichael Lange sum3 += aa[25*j+2]*x1 + aa[25*j+7]*x2 + aa[25*j+12]*x3 + aa[25*j+17]*x4 + aa[25*j+22]*x5; 807d6232bceSMichael Lange sum4 += aa[25*j+3]*x1 + aa[25*j+8]*x2 + aa[25*j+13]*x3 + aa[25*j+18]*x4 + aa[25*j+23]*x5; 808d6232bceSMichael Lange sum5 += aa[25*j+4]*x1 + aa[25*j+9]*x2 + aa[25*j+14]*x3 + aa[25*j+19]*x4 + aa[25*j+24]*x5; 809d6232bceSMichael Lange } 810d6232bceSMichael Lange z[5*i] = sum1; z[5*i+1] = sum2; z[5*i+2] = sum3; 811d6232bceSMichael Lange z[5*i+3] = sum4; z[5*i+4] = sum5; 812d6232bceSMichael Lange } 813d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 814d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 815d6232bceSMichael Lange return 0; 816d6232bceSMichael Lange } 817d6232bceSMichael Lange 818d6232bceSMichael Lange #undef __FUNCT__ 819d6232bceSMichael Lange #define __FUNCT__ "MatMult_SeqBAIJ_5" 820d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_5(Mat A,Vec xx,Vec zz) 821d6232bceSMichael Lange { 822d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 823d6232bceSMichael Lange PetscScalar *z,x1,x2,x3,x4,x5,sum1,sum2,sum3,sum4,sum5; 824d6232bceSMichael Lange const PetscScalar *xb,*x; 825d6232bceSMichael Lange const MatScalar *v; 826d6232bceSMichael Lange PetscErrorCode ierr; 827d6232bceSMichael Lange const PetscInt *idx,*ii,*ridx=NULL; 828d6232bceSMichael Lange PetscInt mbs,i,j,n; 829d6232bceSMichael Lange PetscBool usecprow=a->compressedrow.use; 830d6232bceSMichael Lange 831d6232bceSMichael Lange PetscFunctionBegin; 832d6232bceSMichael Lange 833d6232bceSMichael Lange if (usecprow) { 834d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 835d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 836d6232bceSMichael Lange mbs = a->compressedrow.nrows; 837d6232bceSMichael Lange ii = a->compressedrow.i; 838d6232bceSMichael Lange ridx = a->compressedrow.rindex; 839d6232bceSMichael Lange for (i=0; i<mbs; i++) { 840d6232bceSMichael Lange n = ii[i+1] - ii[i]; 841d6232bceSMichael Lange idx = a->j + ii[i]; 842d6232bceSMichael Lange v = a->a + ii[i]*25; 843d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; 844d6232bceSMichael Lange PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 845d6232bceSMichael Lange PetscPrefetchBlock(v+25*n,25*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 846d6232bceSMichael Lange for (j=0; j<n; j++) { 847d6232bceSMichael Lange xb = x + 5*idx[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; 848d6232bceSMichael Lange sum1 += v[25*j]*x1 + v[25*j+5]*x2 + v[25*j+10]*x3 + v[25*j+15]*x4 + v[25*j+20]*x5; 849d6232bceSMichael Lange sum2 += v[25*j+1]*x1 + v[25*j+6]*x2 + v[25*j+11]*x3 + v[25*j+16]*x4 + v[25*j+21]*x5; 850d6232bceSMichael Lange sum3 += v[25*j+2]*x1 + v[25*j+7]*x2 + v[25*j+12]*x3 + v[25*j+17]*x4 + v[25*j+22]*x5; 851d6232bceSMichael Lange sum4 += v[25*j+3]*x1 + v[25*j+8]*x2 + v[25*j+13]*x3 + v[25*j+18]*x4 + v[25*j+23]*x5; 852d6232bceSMichael Lange sum5 += v[25*j+4]*x1 + v[25*j+9]*x2 + v[25*j+14]*x3 + v[25*j+19]*x4 + v[25*j+24]*x5; 853d6232bceSMichael Lange } 854d6232bceSMichael Lange z[5*ridx[i]] = sum1; z[5*ridx[i]+1] = sum2; z[5*ridx[i]+2] = sum3; 855d6232bceSMichael Lange z[5*ridx[i]+3] = sum4; z[5*ridx[i]+4] = sum5; 856d6232bceSMichael Lange } 857d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 858d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 859d6232bceSMichael Lange } else { 860d6232bceSMichael Lange ierr = PetscThreadCommRunKernel(PetscObjectComm((PetscObject)A),(PetscThreadKernel)MatMult_SeqBAIJ_5_Kernel,3,A,xx,zz);CHKERRQ(ierr); 861d6232bceSMichael Lange } 862d6232bceSMichael Lange ierr = PetscLogFlops(50.0*a->nz - 5.0*a->nonzerorowcnt);CHKERRQ(ierr); 863d6232bceSMichael Lange PetscFunctionReturn(0); 864d6232bceSMichael Lange } 865d6232bceSMichael Lange #else 8664a2ae208SSatish Balay #undef __FUNCT__ 8674a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_5" 868dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_5(Mat A,Vec xx,Vec zz) 8692d61bbb3SSatish Balay { 8702d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 871d9fead3dSBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5,*z = 0,*zarray; 872d9fead3dSBarry Smith const PetscScalar *xb,*x; 873d9fead3dSBarry Smith const MatScalar *v; 874dfbe8321SBarry Smith PetscErrorCode ierr; 8750298fd71SBarry Smith const PetscInt *idx,*ii,*ridx=NULL; 8767c565772SBarry Smith PetscInt mbs,i,j,n; 877ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 8782d61bbb3SSatish Balay 879433994e6SBarry Smith PetscFunctionBegin; 8803649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 88126e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 8822d61bbb3SSatish Balay 8832d61bbb3SSatish Balay idx = a->j; 8842d61bbb3SSatish Balay v = a->a; 88526e093fcSHong Zhang if (usecprow) { 88626e093fcSHong Zhang mbs = a->compressedrow.nrows; 88726e093fcSHong Zhang ii = a->compressedrow.i; 8887b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 88926e093fcSHong Zhang } else { 89026e093fcSHong Zhang mbs = a->mbs; 8912d61bbb3SSatish Balay ii = a->i; 89226e093fcSHong Zhang z = zarray; 89326e093fcSHong Zhang } 8942d61bbb3SSatish Balay 8952d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 8962d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 8972d61bbb3SSatish Balay sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; 898444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 899444d8c10SJed Brown PetscPrefetchBlock(v+25*n,25*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 9002d61bbb3SSatish Balay for (j=0; j<n; j++) { 9012d61bbb3SSatish Balay xb = x + 5*(*idx++); 9022d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; 9032d61bbb3SSatish Balay sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3 + v[15]*x4 + v[20]*x5; 9042d61bbb3SSatish Balay sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3 + v[16]*x4 + v[21]*x5; 9052d61bbb3SSatish Balay sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3 + v[17]*x4 + v[22]*x5; 9062d61bbb3SSatish Balay sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3 + v[18]*x4 + v[23]*x5; 9072d61bbb3SSatish Balay sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3 + v[19]*x4 + v[24]*x5; 9082d61bbb3SSatish Balay v += 25; 9092d61bbb3SSatish Balay } 9107b2bb3b9SHong Zhang if (usecprow) z = zarray + 5*ridx[i]; 9112d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; 91226e093fcSHong Zhang if (!usecprow) z += 5; 9132d61bbb3SSatish Balay } 9143649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 91526e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 9167c565772SBarry Smith ierr = PetscLogFlops(50.0*a->nz - 5.0*a->nonzerorowcnt);CHKERRQ(ierr); 9172d61bbb3SSatish Balay PetscFunctionReturn(0); 9182d61bbb3SSatish Balay } 919d6232bceSMichael Lange #endif 9202d61bbb3SSatish Balay 92115091d37SBarry Smith 922d6232bceSMichael Lange #if defined(PETSC_THREADCOMM_ACTIVE) 923d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_6_Kernel(PetscInt thread_id,Mat A,Vec xx,Vec zz) 924d6232bceSMichael Lange { 925d6232bceSMichael Lange PetscErrorCode ierr; 926d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 927d6232bceSMichael Lange PetscScalar *z,x1,x2,x3,x4,x5,x6,sum1,sum2,sum3,sum4,sum5,sum6; 928d6232bceSMichael Lange const PetscScalar *x,*xb; 929d6232bceSMichael Lange const MatScalar *aa; 930d6232bceSMichael Lange PetscInt *trstarts=A->rmap->trstarts; 931d6232bceSMichael Lange PetscInt n,start,end,i,j; 932d6232bceSMichael Lange const PetscInt *aj,*ai; 933d6232bceSMichael Lange 934d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 935d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 936d6232bceSMichael Lange start = trstarts[thread_id] / 6; 937d6232bceSMichael Lange end = trstarts[thread_id+1] / 6; 938d6232bceSMichael Lange ai = a->i; 939d6232bceSMichael Lange for (i=start; i<end; i++) { 940d6232bceSMichael Lange n = ai[i+1] - ai[i]; 941d6232bceSMichael Lange aj = a->j + ai[i]; 942d6232bceSMichael Lange aa = a->a + ai[i]*36; 943d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; 944d6232bceSMichael Lange for (j=0; j<n; j++) { 945d6232bceSMichael Lange xb = x + 6*aj[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; 946d6232bceSMichael Lange sum1 += aa[36*j]*x1 + aa[36*j+6]*x2 + aa[36*j+12]*x3 + aa[36*j+18]*x4 + aa[36*j+24]*x5 + aa[36*j+30]*x6; 947d6232bceSMichael Lange sum2 += aa[36*j+1]*x1 + aa[36*j+7]*x2 + aa[36*j+13]*x3 + aa[36*j+19]*x4 + aa[36*j+25]*x5 + aa[36*j+31]*x6; 948d6232bceSMichael Lange sum3 += aa[36*j+2]*x1 + aa[36*j+8]*x2 + aa[36*j+14]*x3 + aa[36*j+20]*x4 + aa[36*j+26]*x5 + aa[36*j+32]*x6; 949d6232bceSMichael Lange sum4 += aa[36*j+3]*x1 + aa[36*j+9]*x2 + aa[36*j+15]*x3 + aa[36*j+21]*x4 + aa[36*j+27]*x5 + aa[36*j+33]*x6; 950d6232bceSMichael Lange sum5 += aa[36*j+4]*x1 + aa[36*j+10]*x2 + aa[36*j+16]*x3 + aa[36*j+22]*x4 + aa[36*j+28]*x5 + aa[36*j+34]*x6; 951d6232bceSMichael Lange sum6 += aa[36*j+5]*x1 + aa[36*j+11]*x2 + aa[36*j+17]*x3 + aa[36*j+23]*x4 + aa[36*j+29]*x5 + aa[36*j+35]*x6; 952d6232bceSMichael Lange } 953d6232bceSMichael Lange z[6*i] = sum1; z[6*i+1] = sum2; z[6*i+2] = sum3; 954d6232bceSMichael Lange z[6*i+3] = sum4; z[6*i+4] = sum5; z[6*i+5] = sum6; 955d6232bceSMichael Lange } 956d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 957d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 958d6232bceSMichael Lange return 0; 959d6232bceSMichael Lange } 960d6232bceSMichael Lange 961d6232bceSMichael Lange #undef __FUNCT__ 962d6232bceSMichael Lange #define __FUNCT__ "MatMult_SeqBAIJ_6" 963d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_6(Mat A,Vec xx,Vec zz) 964d6232bceSMichael Lange { 965d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 966d6232bceSMichael Lange PetscScalar *z,x1,x2,x3,x4,x5,x6,sum1,sum2,sum3,sum4,sum5,sum6; 967d6232bceSMichael Lange const PetscScalar *x,*xb; 968d6232bceSMichael Lange const MatScalar *v; 969d6232bceSMichael Lange PetscErrorCode ierr; 970d6232bceSMichael Lange PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 971d6232bceSMichael Lange PetscBool usecprow=a->compressedrow.use; 972d6232bceSMichael Lange 973d6232bceSMichael Lange PetscFunctionBegin; 974d6232bceSMichael Lange 975d6232bceSMichael Lange if (usecprow) { 976d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 977d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 978d6232bceSMichael Lange mbs = a->compressedrow.nrows; 979d6232bceSMichael Lange ii = a->compressedrow.i; 980d6232bceSMichael Lange ridx = a->compressedrow.rindex; 981d6232bceSMichael Lange for (i=0; i<mbs; i++) { 982d6232bceSMichael Lange n = ii[i+1] - ii[i]; 983d6232bceSMichael Lange idx = a->j + ii[i]; 984d6232bceSMichael Lange v = a->a + ii[i]*36; 985d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; 986d6232bceSMichael Lange PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 987d6232bceSMichael Lange PetscPrefetchBlock(v+36*n,36*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 988d6232bceSMichael Lange for (j=0; j<n; j++) { 989d6232bceSMichael Lange xb = x + 6*idx[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; 990d6232bceSMichael Lange sum1 += v[36*j]*x1 + v[36*j+6]*x2 + v[36*j+12]*x3 + v[36*j+18]*x4 + v[36*j+24]*x5 + v[36*j+30]*x6; 991d6232bceSMichael Lange sum2 += v[36*j+1]*x1 + v[36*j+7]*x2 + v[36*j+13]*x3 + v[36*j+19]*x4 + v[36*j+25]*x5 + v[36*j+31]*x6; 992d6232bceSMichael Lange sum3 += v[36*j+2]*x1 + v[36*j+8]*x2 + v[36*j+14]*x3 + v[36*j+20]*x4 + v[36*j+26]*x5 + v[36*j+32]*x6; 993d6232bceSMichael Lange sum4 += v[36*j+3]*x1 + v[36*j+9]*x2 + v[36*j+15]*x3 + v[36*j+21]*x4 + v[36*j+27]*x5 + v[36*j+33]*x6; 994d6232bceSMichael Lange sum5 += v[36*j+4]*x1 + v[36*j+10]*x2 + v[36*j+16]*x3 + v[36*j+22]*x4 + v[36*j+28]*x5 + v[36*j+34]*x6; 995d6232bceSMichael Lange sum6 += v[36*j+5]*x1 + v[36*j+11]*x2 + v[36*j+17]*x3 + v[36*j+23]*x4 + v[36*j+29]*x5 + v[36*j+35]*x6; 996d6232bceSMichael Lange } 997d6232bceSMichael Lange z[6*ridx[i]] = sum1; z[6*ridx[i]+1] = sum2; z[6*ridx[i]+2] = sum3; 998d6232bceSMichael Lange z[6*ridx[i]+3] = sum4; z[6*ridx[i]+4] = sum5; z[6*ridx[i]+5] = sum6; 999d6232bceSMichael Lange } 1000d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1001d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 1002d6232bceSMichael Lange } else { 1003d6232bceSMichael Lange ierr = PetscThreadCommRunKernel(PetscObjectComm((PetscObject)A),(PetscThreadKernel)MatMult_SeqBAIJ_6_Kernel,3,A,xx,zz);CHKERRQ(ierr); 1004d6232bceSMichael Lange } 1005d6232bceSMichael Lange ierr = PetscLogFlops(72.0*a->nz - 6.0*a->nonzerorowcnt);CHKERRQ(ierr); 1006d6232bceSMichael Lange PetscFunctionReturn(0); 1007d6232bceSMichael Lange } 1008d6232bceSMichael Lange #else 10094a2ae208SSatish Balay #undef __FUNCT__ 10104a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_6" 1011dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_6(Mat A,Vec xx,Vec zz) 101215091d37SBarry Smith { 101315091d37SBarry Smith Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1014d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6; 1015d9fead3dSBarry Smith const PetscScalar *x,*xb; 101626e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,*zarray; 1017d9fead3dSBarry Smith const MatScalar *v; 1018dfbe8321SBarry Smith PetscErrorCode ierr; 10197c565772SBarry Smith PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 1020ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 102115091d37SBarry Smith 1022433994e6SBarry Smith PetscFunctionBegin; 10233649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 102426e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 102515091d37SBarry Smith 102615091d37SBarry Smith idx = a->j; 102715091d37SBarry Smith v = a->a; 102826e093fcSHong Zhang if (usecprow) { 102926e093fcSHong Zhang mbs = a->compressedrow.nrows; 103026e093fcSHong Zhang ii = a->compressedrow.i; 10317b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 103226e093fcSHong Zhang } else { 103326e093fcSHong Zhang mbs = a->mbs; 103415091d37SBarry Smith ii = a->i; 103526e093fcSHong Zhang z = zarray; 103626e093fcSHong Zhang } 103715091d37SBarry Smith 103815091d37SBarry Smith for (i=0; i<mbs; i++) { 103926fbe8dcSKarl Rupp n = ii[1] - ii[0]; 104026fbe8dcSKarl Rupp ii++; 104126fbe8dcSKarl Rupp sum1 = 0.0; 104226fbe8dcSKarl Rupp sum2 = 0.0; 104326fbe8dcSKarl Rupp sum3 = 0.0; 104426fbe8dcSKarl Rupp sum4 = 0.0; 104526fbe8dcSKarl Rupp sum5 = 0.0; 104626fbe8dcSKarl Rupp sum6 = 0.0; 104726fbe8dcSKarl Rupp 1048444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 1049444d8c10SJed Brown PetscPrefetchBlock(v+36*n,36*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 105015091d37SBarry Smith for (j=0; j<n; j++) { 105115091d37SBarry Smith xb = x + 6*(*idx++); 105215091d37SBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; 105315091d37SBarry Smith sum1 += v[0]*x1 + v[6]*x2 + v[12]*x3 + v[18]*x4 + v[24]*x5 + v[30]*x6; 105415091d37SBarry Smith sum2 += v[1]*x1 + v[7]*x2 + v[13]*x3 + v[19]*x4 + v[25]*x5 + v[31]*x6; 105515091d37SBarry Smith sum3 += v[2]*x1 + v[8]*x2 + v[14]*x3 + v[20]*x4 + v[26]*x5 + v[32]*x6; 105615091d37SBarry Smith sum4 += v[3]*x1 + v[9]*x2 + v[15]*x3 + v[21]*x4 + v[27]*x5 + v[33]*x6; 105715091d37SBarry Smith sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3 + v[22]*x4 + v[28]*x5 + v[34]*x6; 105815091d37SBarry Smith sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3 + v[23]*x4 + v[29]*x5 + v[35]*x6; 105915091d37SBarry Smith v += 36; 106015091d37SBarry Smith } 10617b2bb3b9SHong Zhang if (usecprow) z = zarray + 6*ridx[i]; 106215091d37SBarry Smith z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; 106326e093fcSHong Zhang if (!usecprow) z += 6; 106415091d37SBarry Smith } 106515091d37SBarry Smith 10663649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 106726e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 10687c565772SBarry Smith ierr = PetscLogFlops(72.0*a->nz - 6.0*a->nonzerorowcnt);CHKERRQ(ierr); 106915091d37SBarry Smith PetscFunctionReturn(0); 107015091d37SBarry Smith } 1071d6232bceSMichael Lange #endif 10728ab949d8SShri Abhyankar 1073d6232bceSMichael Lange #if defined(PETSC_THREADCOMM_ACTIVE) 1074d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_7_Kernel(PetscInt thread_id,Mat A,Vec xx,Vec zz) 1075d6232bceSMichael Lange { 1076d6232bceSMichael Lange PetscErrorCode ierr; 1077d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1078d6232bceSMichael Lange PetscScalar *z,x1,x2,x3,x4,x5,x6,x7,sum1,sum2,sum3,sum4,sum5,sum6,sum7; 1079d6232bceSMichael Lange const PetscScalar *x,*xb; 1080d6232bceSMichael Lange const MatScalar *aa; 1081d6232bceSMichael Lange PetscInt *trstarts=A->rmap->trstarts; 1082d6232bceSMichael Lange PetscInt n,start,end,i,j; 1083d6232bceSMichael Lange const PetscInt *aj,*ai; 1084d6232bceSMichael Lange 1085d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1086d6232bceSMichael Lange ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 1087d6232bceSMichael Lange start = trstarts[thread_id] / 7; 1088d6232bceSMichael Lange end = trstarts[thread_id+1] / 7; 1089d6232bceSMichael Lange ai = a->i; 1090d6232bceSMichael Lange for (i=start; i<end; i++) { 1091d6232bceSMichael Lange n = ai[i+1] - ai[i]; 1092d6232bceSMichael Lange aj = a->j + ai[i]; 1093d6232bceSMichael Lange aa = a->a + ai[i]*49; 1094d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 1095d6232bceSMichael Lange for (j=0; j<n; j++) { 1096d6232bceSMichael Lange xb = x + 7*aj[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 1097d6232bceSMichael Lange sum1 += aa[49*j]*x1 + aa[49*j+7]*x2 + aa[49*j+14]*x3 + aa[49*j+21]*x4 + aa[49*j+28]*x5 + aa[49*j+35]*x6 + aa[49*j+42]*x7; 1098d6232bceSMichael Lange sum2 += aa[49*j+1]*x1 + aa[49*j+8]*x2 + aa[49*j+15]*x3 + aa[49*j+22]*x4 + aa[49*j+29]*x5 + aa[49*j+36]*x6 + aa[49*j+43]*x7; 1099d6232bceSMichael Lange sum3 += aa[49*j+2]*x1 + aa[49*j+9]*x2 + aa[49*j+16]*x3 + aa[49*j+23]*x4 + aa[49*j+30]*x5 + aa[49*j+37]*x6 + aa[49*j+44]*x7; 1100d6232bceSMichael Lange sum4 += aa[49*j+3]*x1 + aa[49*j+10]*x2 + aa[49*j+17]*x3 + aa[49*j+24]*x4 + aa[49*j+31]*x5 + aa[49*j+38]*x6 + aa[49*j+45]*x7; 1101d6232bceSMichael Lange sum5 += aa[49*j+4]*x1 + aa[49*j+11]*x2 + aa[49*j+18]*x3 + aa[49*j+25]*x4 + aa[49*j+32]*x5 + aa[49*j+39]*x6 + aa[49*j+46]*x7; 1102d6232bceSMichael Lange sum6 += aa[49*j+5]*x1 + aa[49*j+12]*x2 + aa[49*j+19]*x3 + aa[49*j+26]*x4 + aa[49*j+33]*x5 + aa[49*j+40]*x6 + aa[49*j+47]*x7; 1103d6232bceSMichael Lange sum7 += aa[49*j+6]*x1 + aa[49*j+13]*x2 + aa[49*j+20]*x3 + aa[49*j+27]*x4 + aa[49*j+34]*x5 + aa[49*j+41]*x6 + aa[49*j+48]*x7; 1104d6232bceSMichael Lange } 1105d6232bceSMichael Lange z[7*i] = sum1; z[7*i+1] = sum2; z[7*i+2] = sum3; z[7*i+3] = sum4; 1106d6232bceSMichael Lange z[7*i+4] = sum5; z[7*i+5] = sum6; z[7*i+6] = sum7; 1107d6232bceSMichael Lange } 1108d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1109d6232bceSMichael Lange ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 1110d6232bceSMichael Lange return 0; 1111d6232bceSMichael Lange } 1112d6232bceSMichael Lange 1113d6232bceSMichael Lange #undef __FUNCT__ 1114d6232bceSMichael Lange #define __FUNCT__ "MatMult_SeqBAIJ_7" 1115d6232bceSMichael Lange PetscErrorCode MatMult_SeqBAIJ_7(Mat A,Vec xx,Vec zz) 1116d6232bceSMichael Lange { 1117d6232bceSMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1118d6232bceSMichael Lange PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7; 1119d6232bceSMichael Lange const PetscScalar *x,*xb; 1120d6232bceSMichael Lange PetscScalar x1,x2,x3,x4,x5,x6,x7,*zarray; 1121d6232bceSMichael Lange const MatScalar *v; 1122d6232bceSMichael Lange PetscErrorCode ierr; 1123d6232bceSMichael Lange PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 1124d6232bceSMichael Lange PetscBool usecprow=a->compressedrow.use; 1125d6232bceSMichael Lange 1126d6232bceSMichael Lange PetscFunctionBegin; 1127d6232bceSMichael Lange 1128d6232bceSMichael Lange if (usecprow) { 1129d6232bceSMichael Lange ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1130d6232bceSMichael Lange ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 1131d6232bceSMichael Lange mbs = a->compressedrow.nrows; 1132d6232bceSMichael Lange ii = a->compressedrow.i; 1133d6232bceSMichael Lange ridx = a->compressedrow.rindex; 1134d6232bceSMichael Lange for (i=0; i<mbs; i++) { 1135d6232bceSMichael Lange n = ii[i+1] - ii[i]; 1136d6232bceSMichael Lange idx = a->j + ii[i]; 1137d6232bceSMichael Lange v = a->a + ii[i]*49; 1138d6232bceSMichael Lange sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 1139d6232bceSMichael Lange PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 1140d6232bceSMichael Lange PetscPrefetchBlock(v+49*n,49*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 1141d6232bceSMichael Lange for (j=0; j<n; j++) { 1142d6232bceSMichael Lange xb = x + 7*idx[j]; x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 1143d6232bceSMichael Lange sum1 += v[49*j]*x1 + v[49*j+7]*x2 + v[49*j+14]*x3 + v[49*j+21]*x4 + v[49*j+28]*x5 + v[49*j+35]*x6 + v[49*j+42]*x7; 1144d6232bceSMichael Lange sum2 += v[49*j+1]*x1 + v[49*j+8]*x2 + v[49*j+15]*x3 + v[49*j+22]*x4 + v[49*j+29]*x5 + v[49*j+36]*x6 + v[49*j+43]*x7; 1145d6232bceSMichael Lange sum3 += v[49*j+2]*x1 + v[49*j+9]*x2 + v[49*j+16]*x3 + v[49*j+23]*x4 + v[49*j+30]*x5 + v[49*j+37]*x6 + v[49*j+44]*x7; 1146d6232bceSMichael Lange sum4 += v[49*j+3]*x1 + v[49*j+10]*x2 + v[49*j+17]*x3 + v[49*j+24]*x4 + v[49*j+31]*x5 + v[49*j+38]*x6 + v[49*j+45]*x7; 1147d6232bceSMichael Lange sum5 += v[49*j+4]*x1 + v[49*j+11]*x2 + v[49*j+18]*x3 + v[49*j+25]*x4 + v[49*j+32]*x5 + v[49*j+39]*x6 + v[49*j+46]*x7; 1148d6232bceSMichael Lange sum6 += v[49*j+5]*x1 + v[49*j+12]*x2 + v[49*j+19]*x3 + v[49*j+26]*x4 + v[49*j+33]*x5 + v[49*j+40]*x6 + v[49*j+47]*x7; 1149d6232bceSMichael Lange sum7 += v[49*j+6]*x1 + v[49*j+13]*x2 + v[49*j+20]*x3 + v[49*j+27]*x4 + v[49*j+34]*x5 + v[49*j+41]*x6 + v[49*j+48]*x7; 1150d6232bceSMichael Lange } 1151d6232bceSMichael Lange z[7*ridx[i]] = sum1; z[7*ridx[i]+1] = sum2; z[7*ridx[i]+2] = sum3; z[7*ridx[i]+3] = sum4; 1152d6232bceSMichael Lange z[7*ridx[i]+4] = sum5; z[7*ridx[i]+5] = sum6; z[7*ridx[i]+6] = sum7; 1153d6232bceSMichael Lange } 1154d6232bceSMichael Lange ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1155d6232bceSMichael Lange ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 1156d6232bceSMichael Lange } else { 1157d6232bceSMichael Lange ierr = PetscThreadCommRunKernel(PetscObjectComm((PetscObject)A),(PetscThreadKernel)MatMult_SeqBAIJ_7_Kernel,3,A,xx,zz);CHKERRQ(ierr); 1158d6232bceSMichael Lange } 1159d6232bceSMichael Lange ierr = PetscLogFlops(98.0*a->nz - 7.0*a->nonzerorowcnt);CHKERRQ(ierr); 1160d6232bceSMichael Lange PetscFunctionReturn(0); 1161d6232bceSMichael Lange } 1162d6232bceSMichael Lange #else 11634a2ae208SSatish Balay #undef __FUNCT__ 11644a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_7" 1165dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_7(Mat A,Vec xx,Vec zz) 11662d61bbb3SSatish Balay { 11672d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1168d9fead3dSBarry Smith PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7; 1169d9fead3dSBarry Smith const PetscScalar *x,*xb; 117026e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,x7,*zarray; 1171d9fead3dSBarry Smith const MatScalar *v; 1172dfbe8321SBarry Smith PetscErrorCode ierr; 11737c565772SBarry Smith PetscInt mbs,i,*idx,*ii,j,n,*ridx=NULL; 1174ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 11752d61bbb3SSatish Balay 1176433994e6SBarry Smith PetscFunctionBegin; 11773649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 117826e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 11792d61bbb3SSatish Balay 11802d61bbb3SSatish Balay idx = a->j; 11812d61bbb3SSatish Balay v = a->a; 118226e093fcSHong Zhang if (usecprow) { 118326e093fcSHong Zhang mbs = a->compressedrow.nrows; 118426e093fcSHong Zhang ii = a->compressedrow.i; 11857b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 118626e093fcSHong Zhang } else { 118726e093fcSHong Zhang mbs = a->mbs; 11882d61bbb3SSatish Balay ii = a->i; 118926e093fcSHong Zhang z = zarray; 119026e093fcSHong Zhang } 11912d61bbb3SSatish Balay 11922d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 119326fbe8dcSKarl Rupp n = ii[1] - ii[0]; 119426fbe8dcSKarl Rupp ii++; 119526fbe8dcSKarl Rupp sum1 = 0.0; 119626fbe8dcSKarl Rupp sum2 = 0.0; 119726fbe8dcSKarl Rupp sum3 = 0.0; 119826fbe8dcSKarl Rupp sum4 = 0.0; 119926fbe8dcSKarl Rupp sum5 = 0.0; 120026fbe8dcSKarl Rupp sum6 = 0.0; 120126fbe8dcSKarl Rupp sum7 = 0.0; 120226fbe8dcSKarl Rupp 1203444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 1204444d8c10SJed Brown PetscPrefetchBlock(v+49*n,49*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 12052d61bbb3SSatish Balay for (j=0; j<n; j++) { 12062d61bbb3SSatish Balay xb = x + 7*(*idx++); 12072d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 12082d61bbb3SSatish Balay sum1 += v[0]*x1 + v[7]*x2 + v[14]*x3 + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7; 12092d61bbb3SSatish Balay sum2 += v[1]*x1 + v[8]*x2 + v[15]*x3 + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7; 12102d61bbb3SSatish Balay sum3 += v[2]*x1 + v[9]*x2 + v[16]*x3 + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7; 12112d61bbb3SSatish Balay sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3 + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7; 12122d61bbb3SSatish Balay sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3 + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7; 12132d61bbb3SSatish Balay sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3 + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7; 12142d61bbb3SSatish Balay sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3 + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7; 12152d61bbb3SSatish Balay v += 49; 12162d61bbb3SSatish Balay } 12177b2bb3b9SHong Zhang if (usecprow) z = zarray + 7*ridx[i]; 12182d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 121926e093fcSHong Zhang if (!usecprow) z += 7; 12202d61bbb3SSatish Balay } 12212d61bbb3SSatish Balay 12223649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 122326e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 12247c565772SBarry Smith ierr = PetscLogFlops(98.0*a->nz - 7.0*a->nonzerorowcnt);CHKERRQ(ierr); 12252d61bbb3SSatish Balay PetscFunctionReturn(0); 12262d61bbb3SSatish Balay } 1227d6232bceSMichael Lange #endif 12282d61bbb3SSatish Balay 12298ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15 version 1: Columns in the block are accessed one at a time */ 1230832cc040SShri Abhyankar /* Default MatMult for block size 15 */ 12318ab949d8SShri Abhyankar 12328ab949d8SShri Abhyankar #undef __FUNCT__ 12338ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver1" 12348ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver1(Mat A,Vec xx,Vec zz) 12358ab949d8SShri Abhyankar { 12368ab949d8SShri Abhyankar Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 12378ab949d8SShri Abhyankar PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15; 12388ab949d8SShri Abhyankar const PetscScalar *x,*xb; 123953ef36baSBarry Smith PetscScalar *zarray,xv; 12408ab949d8SShri Abhyankar const MatScalar *v; 12418ab949d8SShri Abhyankar PetscErrorCode ierr; 12428ab949d8SShri Abhyankar const PetscInt *ii,*ij=a->j,*idx; 12437c565772SBarry Smith PetscInt mbs,i,j,k,n,*ridx=NULL; 1244ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 12458ab949d8SShri Abhyankar 12468ab949d8SShri Abhyankar PetscFunctionBegin; 12473649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 12488ab949d8SShri Abhyankar ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 12498ab949d8SShri Abhyankar 12508ab949d8SShri Abhyankar v = a->a; 12518ab949d8SShri Abhyankar if (usecprow) { 12528ab949d8SShri Abhyankar mbs = a->compressedrow.nrows; 12538ab949d8SShri Abhyankar ii = a->compressedrow.i; 12548ab949d8SShri Abhyankar ridx = a->compressedrow.rindex; 12558ab949d8SShri Abhyankar } else { 12568ab949d8SShri Abhyankar mbs = a->mbs; 12578ab949d8SShri Abhyankar ii = a->i; 12588ab949d8SShri Abhyankar z = zarray; 12598ab949d8SShri Abhyankar } 12608ab949d8SShri Abhyankar 12618ab949d8SShri Abhyankar for (i=0; i<mbs; i++) { 12628ab949d8SShri Abhyankar n = ii[i+1] - ii[i]; 12638ab949d8SShri Abhyankar idx = ij + ii[i]; 12648ab949d8SShri Abhyankar sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 12658ab949d8SShri 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; 12668ab949d8SShri Abhyankar 12678ab949d8SShri Abhyankar for (j=0; j<n; j++) { 12688ab949d8SShri Abhyankar xb = x + 15*(idx[j]); 12698ab949d8SShri Abhyankar 12708ab949d8SShri Abhyankar for (k=0; k<15; k++) { 127153ef36baSBarry Smith xv = xb[k]; 127253ef36baSBarry Smith sum1 += v[0]*xv; 127353ef36baSBarry Smith sum2 += v[1]*xv; 127453ef36baSBarry Smith sum3 += v[2]*xv; 127553ef36baSBarry Smith sum4 += v[3]*xv; 127653ef36baSBarry Smith sum5 += v[4]*xv; 127753ef36baSBarry Smith sum6 += v[5]*xv; 127853ef36baSBarry Smith sum7 += v[6]*xv; 127953ef36baSBarry Smith sum8 += v[7]*xv; 128053ef36baSBarry Smith sum9 += v[8]*xv; 128153ef36baSBarry Smith sum10 += v[9]*xv; 128253ef36baSBarry Smith sum11 += v[10]*xv; 128353ef36baSBarry Smith sum12 += v[11]*xv; 128453ef36baSBarry Smith sum13 += v[12]*xv; 128553ef36baSBarry Smith sum14 += v[13]*xv; 128653ef36baSBarry Smith sum15 += v[14]*xv; 12878ab949d8SShri Abhyankar v += 15; 12888ab949d8SShri Abhyankar } 12898ab949d8SShri Abhyankar } 12908ab949d8SShri Abhyankar if (usecprow) z = zarray + 15*ridx[i]; 12918ab949d8SShri Abhyankar z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 12928ab949d8SShri 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; 12938ab949d8SShri Abhyankar 12948ab949d8SShri Abhyankar if (!usecprow) z += 15; 12958ab949d8SShri Abhyankar } 12968ab949d8SShri Abhyankar 12973649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 12988ab949d8SShri Abhyankar ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 12997c565772SBarry Smith ierr = PetscLogFlops(450.0*a->nz - 15.0*a->nonzerorowcnt);CHKERRQ(ierr); 13008ab949d8SShri Abhyankar PetscFunctionReturn(0); 13018ab949d8SShri Abhyankar } 13028ab949d8SShri Abhyankar 13038ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15_ver2 : Columns in the block are accessed in sets of 4,4,4,3 */ 13048ab949d8SShri Abhyankar #undef __FUNCT__ 13058ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver2" 13068ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver2(Mat A,Vec xx,Vec zz) 13078ab949d8SShri Abhyankar { 13088ab949d8SShri Abhyankar Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 13098ab949d8SShri Abhyankar PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15; 13108ab949d8SShri Abhyankar const PetscScalar *x,*xb; 13110b8f6341SShri Abhyankar PetscScalar x1,x2,x3,x4,*zarray; 13128ab949d8SShri Abhyankar const MatScalar *v; 13138ab949d8SShri Abhyankar PetscErrorCode ierr; 13148ab949d8SShri Abhyankar const PetscInt *ii,*ij=a->j,*idx; 13157c565772SBarry Smith PetscInt mbs,i,j,n,*ridx=NULL; 1316ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 13178ab949d8SShri Abhyankar 13188ab949d8SShri Abhyankar PetscFunctionBegin; 13193649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 13208ab949d8SShri Abhyankar ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 13218ab949d8SShri Abhyankar 13228ab949d8SShri Abhyankar v = a->a; 13238ab949d8SShri Abhyankar if (usecprow) { 13248ab949d8SShri Abhyankar mbs = a->compressedrow.nrows; 13258ab949d8SShri Abhyankar ii = a->compressedrow.i; 13268ab949d8SShri Abhyankar ridx = a->compressedrow.rindex; 13278ab949d8SShri Abhyankar } else { 13288ab949d8SShri Abhyankar mbs = a->mbs; 13298ab949d8SShri Abhyankar ii = a->i; 13308ab949d8SShri Abhyankar z = zarray; 13318ab949d8SShri Abhyankar } 13328ab949d8SShri Abhyankar 13338ab949d8SShri Abhyankar for (i=0; i<mbs; i++) { 13348ab949d8SShri Abhyankar n = ii[i+1] - ii[i]; 13358ab949d8SShri Abhyankar idx = ij + ii[i]; 13368ab949d8SShri Abhyankar sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 13378ab949d8SShri 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; 13388ab949d8SShri Abhyankar 13398ab949d8SShri Abhyankar for (j=0; j<n; j++) { 13408ab949d8SShri Abhyankar xb = x + 15*(idx[j]); 13410b8f6341SShri Abhyankar x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 13428ab949d8SShri Abhyankar 13438ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4; 13448ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4; 13458ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4; 13468ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4; 13478ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4; 13488ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4; 13498ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4; 13508ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4; 13518ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4; 13528ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4; 13538ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4; 13548ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4; 13558ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4; 13568ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4; 13578ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4; 13588ab949d8SShri Abhyankar 13598ab949d8SShri Abhyankar v += 60; 13608ab949d8SShri Abhyankar 13610b8f6341SShri Abhyankar x1 = xb[4]; x2 = xb[5]; x3 = xb[6]; x4 = xb[7]; 13628ab949d8SShri Abhyankar 13638ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4; 13648ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4; 13658ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4; 13668ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4; 13678ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4; 13688ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4; 13698ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4; 13708ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4; 13718ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4; 13728ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4; 13738ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4; 13748ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4; 13758ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4; 13768ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4; 13778ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4; 13788ab949d8SShri Abhyankar v += 60; 13798ab949d8SShri Abhyankar 13800b8f6341SShri Abhyankar x1 = xb[8]; x2 = xb[9]; x3 = xb[10]; x4 = xb[11]; 13810b8f6341SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4; 13820b8f6341SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4; 13830b8f6341SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4; 13840b8f6341SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4; 13850b8f6341SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4; 13860b8f6341SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4; 13870b8f6341SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4; 13880b8f6341SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4; 13890b8f6341SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4; 13900b8f6341SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4; 13910b8f6341SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4; 13920b8f6341SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4; 13930b8f6341SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4; 13940b8f6341SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4; 13950b8f6341SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4; 13960b8f6341SShri Abhyankar v += 60; 13970b8f6341SShri Abhyankar 13980b8f6341SShri Abhyankar x1 = xb[12]; x2 = xb[13]; x3 = xb[14]; 13998ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3; 14008ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3; 14018ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3; 14028ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3; 14038ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3; 14048ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3; 14058ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3; 14068ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3; 14078ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3; 14088ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3; 14098ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3; 14108ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3; 14118ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3; 14128ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3; 14138ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3; 14148ab949d8SShri Abhyankar v += 45; 14158ab949d8SShri Abhyankar } 14168ab949d8SShri Abhyankar if (usecprow) z = zarray + 15*ridx[i]; 14178ab949d8SShri Abhyankar z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 14188ab949d8SShri 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; 14198ab949d8SShri Abhyankar 14208ab949d8SShri Abhyankar if (!usecprow) z += 15; 14218ab949d8SShri Abhyankar } 14228ab949d8SShri Abhyankar 14233649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 14248ab949d8SShri Abhyankar ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 14257c565772SBarry Smith ierr = PetscLogFlops(450.0*a->nz - 15.0*a->nonzerorowcnt);CHKERRQ(ierr); 14268ab949d8SShri Abhyankar PetscFunctionReturn(0); 14278ab949d8SShri Abhyankar } 14288ab949d8SShri Abhyankar 14298ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15_ver3 : Columns in the block are accessed in sets of 8,7 */ 14308ab949d8SShri Abhyankar #undef __FUNCT__ 14318ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver3" 14328ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver3(Mat A,Vec xx,Vec zz) 14338ab949d8SShri Abhyankar { 14348ab949d8SShri Abhyankar Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 14358ab949d8SShri Abhyankar PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15; 14368ab949d8SShri Abhyankar const PetscScalar *x,*xb; 14370b8f6341SShri Abhyankar PetscScalar x1,x2,x3,x4,x5,x6,x7,x8,*zarray; 14388ab949d8SShri Abhyankar const MatScalar *v; 14398ab949d8SShri Abhyankar PetscErrorCode ierr; 14408ab949d8SShri Abhyankar const PetscInt *ii,*ij=a->j,*idx; 14417c565772SBarry Smith PetscInt mbs,i,j,n,*ridx=NULL; 1442ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 14438ab949d8SShri Abhyankar 14448ab949d8SShri Abhyankar PetscFunctionBegin; 14453649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 14468ab949d8SShri Abhyankar ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 14478ab949d8SShri Abhyankar 14488ab949d8SShri Abhyankar v = a->a; 14498ab949d8SShri Abhyankar if (usecprow) { 14508ab949d8SShri Abhyankar mbs = a->compressedrow.nrows; 14518ab949d8SShri Abhyankar ii = a->compressedrow.i; 14528ab949d8SShri Abhyankar ridx = a->compressedrow.rindex; 14538ab949d8SShri Abhyankar } else { 14548ab949d8SShri Abhyankar mbs = a->mbs; 14558ab949d8SShri Abhyankar ii = a->i; 14568ab949d8SShri Abhyankar z = zarray; 14578ab949d8SShri Abhyankar } 14588ab949d8SShri Abhyankar 14598ab949d8SShri Abhyankar for (i=0; i<mbs; i++) { 14608ab949d8SShri Abhyankar n = ii[i+1] - ii[i]; 14618ab949d8SShri Abhyankar idx = ij + ii[i]; 14628ab949d8SShri Abhyankar sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 14638ab949d8SShri 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; 14648ab949d8SShri Abhyankar 14658ab949d8SShri Abhyankar for (j=0; j<n; j++) { 14668ab949d8SShri Abhyankar xb = x + 15*(idx[j]); 14678ab949d8SShri Abhyankar x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 14680b8f6341SShri Abhyankar x8 = xb[7]; 14698ab949d8SShri Abhyankar 14708ab949d8SShri 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; 14718ab949d8SShri 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; 14728ab949d8SShri 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; 14738ab949d8SShri 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; 14748ab949d8SShri 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; 14758ab949d8SShri 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; 14768ab949d8SShri 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; 14778ab949d8SShri 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; 14788ab949d8SShri 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; 14798ab949d8SShri 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; 14808ab949d8SShri 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; 14818ab949d8SShri 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; 14828ab949d8SShri 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; 14838ab949d8SShri 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; 14848ab949d8SShri 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; 14858ab949d8SShri Abhyankar v += 120; 14868ab949d8SShri Abhyankar 14870b8f6341SShri Abhyankar x1 = xb[8]; x2 = xb[9]; x3 = xb[10]; x4 = xb[11]; x5 = xb[12]; x6 = xb[13]; x7 = xb[14]; 14880b8f6341SShri Abhyankar 14898ab949d8SShri Abhyankar sum1 += v[0]*x1 + v[15]*x2 + v[30]*x3 + v[45]*x4 + v[60]*x5 + v[75]*x6 + v[90]*x7; 14908ab949d8SShri Abhyankar sum2 += v[1]*x1 + v[16]*x2 + v[31]*x3 + v[46]*x4 + v[61]*x5 + v[76]*x6 + v[91]*x7; 14918ab949d8SShri Abhyankar sum3 += v[2]*x1 + v[17]*x2 + v[32]*x3 + v[47]*x4 + v[62]*x5 + v[77]*x6 + v[92]*x7; 14928ab949d8SShri Abhyankar sum4 += v[3]*x1 + v[18]*x2 + v[33]*x3 + v[48]*x4 + v[63]*x5 + v[78]*x6 + v[93]*x7; 14938ab949d8SShri Abhyankar sum5 += v[4]*x1 + v[19]*x2 + v[34]*x3 + v[49]*x4 + v[64]*x5 + v[79]*x6 + v[94]*x7; 14948ab949d8SShri Abhyankar sum6 += v[5]*x1 + v[20]*x2 + v[35]*x3 + v[50]*x4 + v[65]*x5 + v[80]*x6 + v[95]*x7; 14958ab949d8SShri Abhyankar sum7 += v[6]*x1 + v[21]*x2 + v[36]*x3 + v[51]*x4 + v[66]*x5 + v[81]*x6 + v[96]*x7; 14968ab949d8SShri Abhyankar sum8 += v[7]*x1 + v[22]*x2 + v[37]*x3 + v[52]*x4 + v[67]*x5 + v[82]*x6 + v[97]*x7; 14978ab949d8SShri Abhyankar sum9 += v[8]*x1 + v[23]*x2 + v[38]*x3 + v[53]*x4 + v[68]*x5 + v[83]*x6 + v[98]*x7; 14988ab949d8SShri Abhyankar sum10 += v[9]*x1 + v[24]*x2 + v[39]*x3 + v[54]*x4 + v[69]*x5 + v[84]*x6 + v[99]*x7; 14998ab949d8SShri Abhyankar sum11 += v[10]*x1 + v[25]*x2 + v[40]*x3 + v[55]*x4 + v[70]*x5 + v[85]*x6 + v[100]*x7; 15008ab949d8SShri Abhyankar sum12 += v[11]*x1 + v[26]*x2 + v[41]*x3 + v[56]*x4 + v[71]*x5 + v[86]*x6 + v[101]*x7; 15018ab949d8SShri Abhyankar sum13 += v[12]*x1 + v[27]*x2 + v[42]*x3 + v[57]*x4 + v[72]*x5 + v[87]*x6 + v[102]*x7; 15028ab949d8SShri Abhyankar sum14 += v[13]*x1 + v[28]*x2 + v[43]*x3 + v[58]*x4 + v[73]*x5 + v[88]*x6 + v[103]*x7; 15038ab949d8SShri Abhyankar sum15 += v[14]*x1 + v[29]*x2 + v[44]*x3 + v[59]*x4 + v[74]*x5 + v[89]*x6 + v[104]*x7; 15048ab949d8SShri Abhyankar v += 105; 15058ab949d8SShri Abhyankar } 15068ab949d8SShri Abhyankar if (usecprow) z = zarray + 15*ridx[i]; 15078ab949d8SShri Abhyankar z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 15088ab949d8SShri 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; 15098ab949d8SShri Abhyankar 15108ab949d8SShri Abhyankar if (!usecprow) z += 15; 15118ab949d8SShri Abhyankar } 15128ab949d8SShri Abhyankar 15133649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 15148ab949d8SShri Abhyankar ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 15157c565772SBarry Smith ierr = PetscLogFlops(450.0*a->nz - 15.0*a->nonzerorowcnt);CHKERRQ(ierr); 15168ab949d8SShri Abhyankar PetscFunctionReturn(0); 15178ab949d8SShri Abhyankar } 15188ab949d8SShri Abhyankar 15198ab949d8SShri Abhyankar /* MatMult_SeqBAIJ_15_ver4 : All columns in the block are accessed at once */ 15208ab949d8SShri Abhyankar 15218ab949d8SShri Abhyankar #undef __FUNCT__ 15228ab949d8SShri Abhyankar #define __FUNCT__ "MatMult_SeqBAIJ_15_ver4" 15238ab949d8SShri Abhyankar PetscErrorCode MatMult_SeqBAIJ_15_ver4(Mat A,Vec xx,Vec zz) 15248ab949d8SShri Abhyankar { 15258ab949d8SShri Abhyankar Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 15268ab949d8SShri Abhyankar PetscScalar *z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7,sum8,sum9,sum10,sum11,sum12,sum13,sum14,sum15; 15278ab949d8SShri Abhyankar const PetscScalar *x,*xb; 15288ab949d8SShri Abhyankar PetscScalar x1,x2,x3,x4,x5,x6,x7,x8,x9,x10,x11,x12,x13,x14,x15,*zarray; 15298ab949d8SShri Abhyankar const MatScalar *v; 15308ab949d8SShri Abhyankar PetscErrorCode ierr; 15318ab949d8SShri Abhyankar const PetscInt *ii,*ij=a->j,*idx; 15327c565772SBarry Smith PetscInt mbs,i,j,n,*ridx=NULL; 1533ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 15348ab949d8SShri Abhyankar 15358ab949d8SShri Abhyankar PetscFunctionBegin; 15363649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 15378ab949d8SShri Abhyankar ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 15388ab949d8SShri Abhyankar 15398ab949d8SShri Abhyankar v = a->a; 15408ab949d8SShri Abhyankar if (usecprow) { 15418ab949d8SShri Abhyankar mbs = a->compressedrow.nrows; 15428ab949d8SShri Abhyankar ii = a->compressedrow.i; 15438ab949d8SShri Abhyankar ridx = a->compressedrow.rindex; 15448ab949d8SShri Abhyankar } else { 15458ab949d8SShri Abhyankar mbs = a->mbs; 15468ab949d8SShri Abhyankar ii = a->i; 15478ab949d8SShri Abhyankar z = zarray; 15488ab949d8SShri Abhyankar } 15498ab949d8SShri Abhyankar 15508ab949d8SShri Abhyankar for (i=0; i<mbs; i++) { 15518ab949d8SShri Abhyankar n = ii[i+1] - ii[i]; 15528ab949d8SShri Abhyankar idx = ij + ii[i]; 15538ab949d8SShri Abhyankar sum1 = 0.0; sum2 = 0.0; sum3 = 0.0; sum4 = 0.0; sum5 = 0.0; sum6 = 0.0; sum7 = 0.0; 15548ab949d8SShri 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; 15558ab949d8SShri Abhyankar 15568ab949d8SShri Abhyankar for (j=0; j<n; j++) { 15578ab949d8SShri Abhyankar xb = x + 15*(idx[j]); 15588ab949d8SShri Abhyankar x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 15598ab949d8SShri 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]; 15608ab949d8SShri Abhyankar 15618ab949d8SShri 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; 15628ab949d8SShri 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; 15638ab949d8SShri 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; 15648ab949d8SShri 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; 15658ab949d8SShri 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; 15668ab949d8SShri 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; 15678ab949d8SShri 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; 15688ab949d8SShri 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; 15698ab949d8SShri 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; 15708ab949d8SShri 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; 15718ab949d8SShri 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; 15728ab949d8SShri 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; 15738ab949d8SShri 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; 15748ab949d8SShri 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; 15758ab949d8SShri 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; 15768ab949d8SShri Abhyankar v += 225; 15778ab949d8SShri Abhyankar } 15788ab949d8SShri Abhyankar if (usecprow) z = zarray + 15*ridx[i]; 15798ab949d8SShri Abhyankar z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 15808ab949d8SShri 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; 15818ab949d8SShri Abhyankar 15828ab949d8SShri Abhyankar if (!usecprow) z += 15; 15838ab949d8SShri Abhyankar } 15848ab949d8SShri Abhyankar 15853649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 15868ab949d8SShri Abhyankar ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 15877c565772SBarry Smith ierr = PetscLogFlops(450.0*a->nz - 15.0*a->nonzerorowcnt);CHKERRQ(ierr); 15888ab949d8SShri Abhyankar PetscFunctionReturn(0); 15898ab949d8SShri Abhyankar } 15908ab949d8SShri Abhyankar 15918ab949d8SShri Abhyankar 15923f1db9ecSBarry Smith /* 15933f1db9ecSBarry Smith This will not work with MatScalar == float because it calls the BLAS 15943f1db9ecSBarry Smith */ 15954a2ae208SSatish Balay #undef __FUNCT__ 15964a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqBAIJ_N" 1597dfbe8321SBarry Smith PetscErrorCode MatMult_SeqBAIJ_N(Mat A,Vec xx,Vec zz) 15982d61bbb3SSatish Balay { 15992d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1600d9ca1df4SBarry Smith PetscScalar *z = 0,*work,*workt,*zarray; 1601d9ca1df4SBarry Smith const PetscScalar *x,*xb; 1602d9ca1df4SBarry Smith const MatScalar *v; 1603dfbe8321SBarry Smith PetscErrorCode ierr; 1604d9ca1df4SBarry Smith PetscInt mbs,i,bs=A->rmap->bs,j,n,bs2=a->bs2; 1605d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ridx=NULL; 1606d9ca1df4SBarry Smith PetscInt ncols,k; 1607ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 16082d61bbb3SSatish Balay 16092d61bbb3SSatish Balay PetscFunctionBegin; 1610d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 161126e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 16122d61bbb3SSatish Balay 16132d61bbb3SSatish Balay idx = a->j; 16142d61bbb3SSatish Balay v = a->a; 161526e093fcSHong Zhang if (usecprow) { 161626e093fcSHong Zhang mbs = a->compressedrow.nrows; 161726e093fcSHong Zhang ii = a->compressedrow.i; 16187b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 161926e093fcSHong Zhang } else { 162026e093fcSHong Zhang mbs = a->mbs; 16212d61bbb3SSatish Balay ii = a->i; 162226e093fcSHong Zhang z = zarray; 162326e093fcSHong Zhang } 1624218c64b6SSatish Balay 16252d61bbb3SSatish Balay if (!a->mult_work) { 1626d0f46423SBarry Smith k = PetscMax(A->rmap->n,A->cmap->n); 1627854ce69bSBarry Smith ierr = PetscMalloc1(k+1,&a->mult_work);CHKERRQ(ierr); 16282d61bbb3SSatish Balay } 16292d61bbb3SSatish Balay work = a->mult_work; 16302d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 16312d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 16322d61bbb3SSatish Balay ncols = n*bs; 16332d61bbb3SSatish Balay workt = work; 16342d61bbb3SSatish Balay for (j=0; j<n; j++) { 16352d61bbb3SSatish Balay xb = x + bs*(*idx++); 16362d61bbb3SSatish Balay for (k=0; k<bs; k++) workt[k] = xb[k]; 16372d61bbb3SSatish Balay workt += bs; 16382d61bbb3SSatish Balay } 16397b2bb3b9SHong Zhang if (usecprow) z = zarray + bs*ridx[i]; 164096b95a6bSBarry Smith PetscKernel_w_gets_Ar_times_v(bs,ncols,work,v,z); 164171044d3cSBarry Smith /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DZero,z,&_One); */ 16422d61bbb3SSatish Balay v += n*bs2; 164326e093fcSHong Zhang if (!usecprow) z += bs; 16442d61bbb3SSatish Balay } 1645d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 164626e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 16477c565772SBarry Smith ierr = PetscLogFlops(2.0*a->nz*bs2 - bs*a->nonzerorowcnt);CHKERRQ(ierr); 16482d61bbb3SSatish Balay PetscFunctionReturn(0); 16492d61bbb3SSatish Balay } 16502d61bbb3SSatish Balay 16514a2ae208SSatish Balay #undef __FUNCT__ 16524a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_1" 1653dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_1(Mat A,Vec xx,Vec yy,Vec zz) 16542d61bbb3SSatish Balay { 16552d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1656122f12eaSBarry Smith const PetscScalar *x; 1657122f12eaSBarry Smith PetscScalar *y,*z,sum; 1658122f12eaSBarry Smith const MatScalar *v; 1659dfbe8321SBarry Smith PetscErrorCode ierr; 16607c565772SBarry Smith PetscInt mbs=a->mbs,i,n,*ridx=NULL; 1661122f12eaSBarry Smith const PetscInt *idx,*ii; 1662ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 16632d61bbb3SSatish Balay 16642d61bbb3SSatish Balay PetscFunctionBegin; 16653649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1666d9ca1df4SBarry Smith ierr = VecGetArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 16672d61bbb3SSatish Balay 16682d61bbb3SSatish Balay idx = a->j; 16692d61bbb3SSatish Balay v = a->a; 167026e093fcSHong Zhang if (usecprow) { 16714eb6d288SHong Zhang if (zz != yy) { 1672122f12eaSBarry Smith ierr = PetscMemcpy(z,y,mbs*sizeof(PetscScalar));CHKERRQ(ierr); 16734eb6d288SHong Zhang } 167426e093fcSHong Zhang mbs = a->compressedrow.nrows; 167526e093fcSHong Zhang ii = a->compressedrow.i; 16767b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 167726e093fcSHong Zhang } else { 16782d61bbb3SSatish Balay ii = a->i; 167926e093fcSHong Zhang } 16802d61bbb3SSatish Balay 16812d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 1682122f12eaSBarry Smith n = ii[1] - ii[0]; 1683122f12eaSBarry Smith ii++; 168426e093fcSHong Zhang if (!usecprow) { 1685122f12eaSBarry Smith sum = y[i]; 1686122f12eaSBarry Smith } else { 1687122f12eaSBarry Smith sum = y[ridx[i]]; 1688122f12eaSBarry Smith } 1689444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 1690444d8c10SJed Brown PetscPrefetchBlock(v+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 1691122f12eaSBarry Smith PetscSparseDensePlusDot(sum,x,v,idx,n); 1692122f12eaSBarry Smith v += n; 1693122f12eaSBarry Smith idx += n; 1694122f12eaSBarry Smith if (usecprow) { 1695122f12eaSBarry Smith z[ridx[i]] = sum; 1696122f12eaSBarry Smith } else { 1697122f12eaSBarry Smith z[i] = sum; 169826e093fcSHong Zhang } 16992d61bbb3SSatish Balay } 17003649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1701d9ca1df4SBarry Smith ierr = VecRestoreArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 17027c565772SBarry Smith ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 17032d61bbb3SSatish Balay PetscFunctionReturn(0); 17042d61bbb3SSatish Balay } 17052d61bbb3SSatish Balay 17064a2ae208SSatish Balay #undef __FUNCT__ 17074a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_2" 1708dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_2(Mat A,Vec xx,Vec yy,Vec zz) 17092d61bbb3SSatish Balay { 17102d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1711d9ca1df4SBarry Smith PetscScalar *y = 0,*z = 0,sum1,sum2; 1712d9ca1df4SBarry Smith const PetscScalar *x,*xb; 171326e093fcSHong Zhang PetscScalar x1,x2,*yarray,*zarray; 1714d9ca1df4SBarry Smith const MatScalar *v; 1715dfbe8321SBarry Smith PetscErrorCode ierr; 1716d9ca1df4SBarry Smith PetscInt mbs = a->mbs,i,n,j; 1717d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ridx = NULL; 1718ace3abfcSBarry Smith PetscBool usecprow = a->compressedrow.use; 17192d61bbb3SSatish Balay 17202d61bbb3SSatish Balay PetscFunctionBegin; 1721d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1722d9ca1df4SBarry Smith ierr = VecGetArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 17232d61bbb3SSatish Balay 17242d61bbb3SSatish Balay idx = a->j; 17252d61bbb3SSatish Balay v = a->a; 172626e093fcSHong Zhang if (usecprow) { 17274eb6d288SHong Zhang if (zz != yy) { 17284eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,2*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 17294eb6d288SHong Zhang } 173026e093fcSHong Zhang mbs = a->compressedrow.nrows; 173126e093fcSHong Zhang ii = a->compressedrow.i; 17327b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 17334eb6d288SHong Zhang if (zz != yy) { 17344eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,a->mbs*sizeof(PetscScalar));CHKERRQ(ierr); 17354eb6d288SHong Zhang } 173626e093fcSHong Zhang } else { 17372d61bbb3SSatish Balay ii = a->i; 173826e093fcSHong Zhang y = yarray; 173926e093fcSHong Zhang z = zarray; 174026e093fcSHong Zhang } 17412d61bbb3SSatish Balay 17422d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 17432d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 174426e093fcSHong Zhang if (usecprow) { 17457b2bb3b9SHong Zhang z = zarray + 2*ridx[i]; 17467b2bb3b9SHong Zhang y = yarray + 2*ridx[i]; 174726e093fcSHong Zhang } 17482d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; 1749444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 1750444d8c10SJed Brown PetscPrefetchBlock(v+4*n,4*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 17512d61bbb3SSatish Balay for (j=0; j<n; j++) { 175226fbe8dcSKarl Rupp xb = x + 2*(*idx++); 175326fbe8dcSKarl Rupp x1 = xb[0]; 175426fbe8dcSKarl Rupp x2 = xb[1]; 175526fbe8dcSKarl Rupp 17562d61bbb3SSatish Balay sum1 += v[0]*x1 + v[2]*x2; 17572d61bbb3SSatish Balay sum2 += v[1]*x1 + v[3]*x2; 17582d61bbb3SSatish Balay v += 4; 17592d61bbb3SSatish Balay } 17602d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; 176126e093fcSHong Zhang if (!usecprow) { 17622d61bbb3SSatish Balay z += 2; y += 2; 17632d61bbb3SSatish Balay } 176426e093fcSHong Zhang } 1765d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1766d9ca1df4SBarry Smith ierr = VecRestoreArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 1767dc0b31edSSatish Balay ierr = PetscLogFlops(4.0*a->nz);CHKERRQ(ierr); 17682d61bbb3SSatish Balay PetscFunctionReturn(0); 17692d61bbb3SSatish Balay } 17702d61bbb3SSatish Balay 17714a2ae208SSatish Balay #undef __FUNCT__ 17724a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_3" 1773dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_3(Mat A,Vec xx,Vec yy,Vec zz) 17742d61bbb3SSatish Balay { 17752d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1776d9ca1df4SBarry Smith PetscScalar *y = 0,*z = 0,sum1,sum2,sum3,x1,x2,x3,*yarray,*zarray; 1777d9ca1df4SBarry Smith const PetscScalar *x,*xb; 1778d9ca1df4SBarry Smith const MatScalar *v; 1779dfbe8321SBarry Smith PetscErrorCode ierr; 1780d9ca1df4SBarry Smith PetscInt mbs = a->mbs,i,j,n; 1781d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ridx = NULL; 1782ace3abfcSBarry Smith PetscBool usecprow = a->compressedrow.use; 17832d61bbb3SSatish Balay 17842d61bbb3SSatish Balay PetscFunctionBegin; 1785d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1786d9ca1df4SBarry Smith ierr = VecGetArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 17872d61bbb3SSatish Balay 17882d61bbb3SSatish Balay idx = a->j; 17892d61bbb3SSatish Balay v = a->a; 179026e093fcSHong Zhang if (usecprow) { 17914eb6d288SHong Zhang if (zz != yy) { 17924eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,3*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 17934eb6d288SHong Zhang } 179426e093fcSHong Zhang mbs = a->compressedrow.nrows; 179526e093fcSHong Zhang ii = a->compressedrow.i; 17967b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 179726e093fcSHong Zhang } else { 17982d61bbb3SSatish Balay ii = a->i; 179926e093fcSHong Zhang y = yarray; 180026e093fcSHong Zhang z = zarray; 180126e093fcSHong Zhang } 18022d61bbb3SSatish Balay 18032d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 18042d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 180526e093fcSHong Zhang if (usecprow) { 18067b2bb3b9SHong Zhang z = zarray + 3*ridx[i]; 18077b2bb3b9SHong Zhang y = yarray + 3*ridx[i]; 180826e093fcSHong Zhang } 18092d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; 1810444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 1811444d8c10SJed Brown PetscPrefetchBlock(v+9*n,9*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 18122d61bbb3SSatish Balay for (j=0; j<n; j++) { 18132d61bbb3SSatish Balay xb = x + 3*(*idx++); x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 18142d61bbb3SSatish Balay sum1 += v[0]*x1 + v[3]*x2 + v[6]*x3; 18152d61bbb3SSatish Balay sum2 += v[1]*x1 + v[4]*x2 + v[7]*x3; 18162d61bbb3SSatish Balay sum3 += v[2]*x1 + v[5]*x2 + v[8]*x3; 18172d61bbb3SSatish Balay v += 9; 18182d61bbb3SSatish Balay } 18192d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; 182026e093fcSHong Zhang if (!usecprow) { 18212d61bbb3SSatish Balay z += 3; y += 3; 18222d61bbb3SSatish Balay } 182326e093fcSHong Zhang } 1824d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1825d9ca1df4SBarry Smith ierr = VecRestoreArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 1826dc0b31edSSatish Balay ierr = PetscLogFlops(18.0*a->nz);CHKERRQ(ierr); 18272d61bbb3SSatish Balay PetscFunctionReturn(0); 18282d61bbb3SSatish Balay } 18292d61bbb3SSatish Balay 18304a2ae208SSatish Balay #undef __FUNCT__ 18314a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_4" 1832dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_4(Mat A,Vec xx,Vec yy,Vec zz) 18332d61bbb3SSatish Balay { 18342d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1835d9ca1df4SBarry Smith PetscScalar *y = 0,*z = 0,sum1,sum2,sum3,sum4,x1,x2,x3,x4,*yarray,*zarray; 1836d9ca1df4SBarry Smith const PetscScalar *x,*xb; 1837d9ca1df4SBarry Smith const MatScalar *v; 1838dfbe8321SBarry Smith PetscErrorCode ierr; 1839d9ca1df4SBarry Smith PetscInt mbs = a->mbs,i,j,n; 1840d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ridx=NULL; 1841ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 18422d61bbb3SSatish Balay 18432d61bbb3SSatish Balay PetscFunctionBegin; 1844d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1845d9ca1df4SBarry Smith ierr = VecGetArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 18462d61bbb3SSatish Balay 18472d61bbb3SSatish Balay idx = a->j; 18482d61bbb3SSatish Balay v = a->a; 184926e093fcSHong Zhang if (usecprow) { 18504eb6d288SHong Zhang if (zz != yy) { 18514eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,4*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 18524eb6d288SHong Zhang } 185326e093fcSHong Zhang mbs = a->compressedrow.nrows; 185426e093fcSHong Zhang ii = a->compressedrow.i; 18557b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 185626e093fcSHong Zhang } else { 18572d61bbb3SSatish Balay ii = a->i; 185826e093fcSHong Zhang y = yarray; 185926e093fcSHong Zhang z = zarray; 186026e093fcSHong Zhang } 18612d61bbb3SSatish Balay 18622d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 18632d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 186426e093fcSHong Zhang if (usecprow) { 18657b2bb3b9SHong Zhang z = zarray + 4*ridx[i]; 18667b2bb3b9SHong Zhang y = yarray + 4*ridx[i]; 186726e093fcSHong Zhang } 18682d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; 1869444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 1870444d8c10SJed Brown PetscPrefetchBlock(v+16*n,16*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 18712d61bbb3SSatish Balay for (j=0; j<n; j++) { 18722d61bbb3SSatish Balay xb = x + 4*(*idx++); 18732d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 18742d61bbb3SSatish Balay sum1 += v[0]*x1 + v[4]*x2 + v[8]*x3 + v[12]*x4; 18752d61bbb3SSatish Balay sum2 += v[1]*x1 + v[5]*x2 + v[9]*x3 + v[13]*x4; 18762d61bbb3SSatish Balay sum3 += v[2]*x1 + v[6]*x2 + v[10]*x3 + v[14]*x4; 18772d61bbb3SSatish Balay sum4 += v[3]*x1 + v[7]*x2 + v[11]*x3 + v[15]*x4; 18782d61bbb3SSatish Balay v += 16; 18792d61bbb3SSatish Balay } 18802d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; 188126e093fcSHong Zhang if (!usecprow) { 18822d61bbb3SSatish Balay z += 4; y += 4; 18832d61bbb3SSatish Balay } 188426e093fcSHong Zhang } 1885d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1886d9ca1df4SBarry Smith ierr = VecRestoreArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 1887dc0b31edSSatish Balay ierr = PetscLogFlops(32.0*a->nz);CHKERRQ(ierr); 18882d61bbb3SSatish Balay PetscFunctionReturn(0); 18892d61bbb3SSatish Balay } 18902d61bbb3SSatish Balay 18914a2ae208SSatish Balay #undef __FUNCT__ 18924a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_5" 1893dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_5(Mat A,Vec xx,Vec yy,Vec zz) 18942d61bbb3SSatish Balay { 18952d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1896d9ca1df4SBarry Smith PetscScalar *y = 0,*z = 0,sum1,sum2,sum3,sum4,sum5,x1,x2,x3,x4,x5; 1897d9ca1df4SBarry Smith const PetscScalar *x,*xb; 189826e093fcSHong Zhang PetscScalar *yarray,*zarray; 1899d9ca1df4SBarry Smith const MatScalar *v; 1900dfbe8321SBarry Smith PetscErrorCode ierr; 1901d9ca1df4SBarry Smith PetscInt mbs = a->mbs,i,j,n; 1902d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ridx = NULL; 1903ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 19042d61bbb3SSatish Balay 19052d61bbb3SSatish Balay PetscFunctionBegin; 1906d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1907d9ca1df4SBarry Smith ierr = VecGetArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 19082d61bbb3SSatish Balay 19092d61bbb3SSatish Balay idx = a->j; 19102d61bbb3SSatish Balay v = a->a; 191126e093fcSHong Zhang if (usecprow) { 19124eb6d288SHong Zhang if (zz != yy) { 19134eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,5*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 19144eb6d288SHong Zhang } 191526e093fcSHong Zhang mbs = a->compressedrow.nrows; 191626e093fcSHong Zhang ii = a->compressedrow.i; 19177b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 191826e093fcSHong Zhang } else { 19192d61bbb3SSatish Balay ii = a->i; 192026e093fcSHong Zhang y = yarray; 192126e093fcSHong Zhang z = zarray; 192226e093fcSHong Zhang } 19232d61bbb3SSatish Balay 19242d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 19252d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 192626e093fcSHong Zhang if (usecprow) { 19277b2bb3b9SHong Zhang z = zarray + 5*ridx[i]; 19287b2bb3b9SHong Zhang y = yarray + 5*ridx[i]; 192926e093fcSHong Zhang } 19302d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; 1931444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 1932444d8c10SJed Brown PetscPrefetchBlock(v+25*n,25*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 19332d61bbb3SSatish Balay for (j=0; j<n; j++) { 19342d61bbb3SSatish Balay xb = x + 5*(*idx++); 19352d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; 19362d61bbb3SSatish Balay sum1 += v[0]*x1 + v[5]*x2 + v[10]*x3 + v[15]*x4 + v[20]*x5; 19372d61bbb3SSatish Balay sum2 += v[1]*x1 + v[6]*x2 + v[11]*x3 + v[16]*x4 + v[21]*x5; 19382d61bbb3SSatish Balay sum3 += v[2]*x1 + v[7]*x2 + v[12]*x3 + v[17]*x4 + v[22]*x5; 19392d61bbb3SSatish Balay sum4 += v[3]*x1 + v[8]*x2 + v[13]*x3 + v[18]*x4 + v[23]*x5; 19402d61bbb3SSatish Balay sum5 += v[4]*x1 + v[9]*x2 + v[14]*x3 + v[19]*x4 + v[24]*x5; 19412d61bbb3SSatish Balay v += 25; 19422d61bbb3SSatish Balay } 19432d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; 194426e093fcSHong Zhang if (!usecprow) { 19452d61bbb3SSatish Balay z += 5; y += 5; 19462d61bbb3SSatish Balay } 194726e093fcSHong Zhang } 1948d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1949d9ca1df4SBarry Smith ierr = VecRestoreArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 1950dc0b31edSSatish Balay ierr = PetscLogFlops(50.0*a->nz);CHKERRQ(ierr); 19512d61bbb3SSatish Balay PetscFunctionReturn(0); 19522d61bbb3SSatish Balay } 19534a2ae208SSatish Balay #undef __FUNCT__ 19544a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_6" 1955dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_6(Mat A,Vec xx,Vec yy,Vec zz) 195615091d37SBarry Smith { 195715091d37SBarry Smith Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 1958d9ca1df4SBarry Smith PetscScalar *y = 0,*z = 0,sum1,sum2,sum3,sum4,sum5,sum6; 1959d9ca1df4SBarry Smith const PetscScalar *x,*xb; 196026e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,*yarray,*zarray; 1961d9ca1df4SBarry Smith const MatScalar *v; 1962dfbe8321SBarry Smith PetscErrorCode ierr; 1963d9ca1df4SBarry Smith PetscInt mbs = a->mbs,i,j,n; 1964d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ridx=NULL; 1965ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 196615091d37SBarry Smith 196715091d37SBarry Smith PetscFunctionBegin; 1968d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1969d9ca1df4SBarry Smith ierr = VecGetArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 197015091d37SBarry Smith 197115091d37SBarry Smith idx = a->j; 197215091d37SBarry Smith v = a->a; 197326e093fcSHong Zhang if (usecprow) { 19744eb6d288SHong Zhang if (zz != yy) { 19754eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,6*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 19764eb6d288SHong Zhang } 197726e093fcSHong Zhang mbs = a->compressedrow.nrows; 197826e093fcSHong Zhang ii = a->compressedrow.i; 19797b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 198026e093fcSHong Zhang } else { 198115091d37SBarry Smith ii = a->i; 198226e093fcSHong Zhang y = yarray; 198326e093fcSHong Zhang z = zarray; 198426e093fcSHong Zhang } 198515091d37SBarry Smith 198615091d37SBarry Smith for (i=0; i<mbs; i++) { 198715091d37SBarry Smith n = ii[1] - ii[0]; ii++; 198826e093fcSHong Zhang if (usecprow) { 19897b2bb3b9SHong Zhang z = zarray + 6*ridx[i]; 19907b2bb3b9SHong Zhang y = yarray + 6*ridx[i]; 199126e093fcSHong Zhang } 199215091d37SBarry Smith sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5]; 1993444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 1994444d8c10SJed Brown PetscPrefetchBlock(v+36*n,36*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 199515091d37SBarry Smith for (j=0; j<n; j++) { 19963b95cb0eSSatish Balay xb = x + 6*(*idx++); 199715091d37SBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; 199815091d37SBarry Smith sum1 += v[0]*x1 + v[6]*x2 + v[12]*x3 + v[18]*x4 + v[24]*x5 + v[30]*x6; 199915091d37SBarry Smith sum2 += v[1]*x1 + v[7]*x2 + v[13]*x3 + v[19]*x4 + v[25]*x5 + v[31]*x6; 200015091d37SBarry Smith sum3 += v[2]*x1 + v[8]*x2 + v[14]*x3 + v[20]*x4 + v[26]*x5 + v[32]*x6; 200115091d37SBarry Smith sum4 += v[3]*x1 + v[9]*x2 + v[15]*x3 + v[21]*x4 + v[27]*x5 + v[33]*x6; 200215091d37SBarry Smith sum5 += v[4]*x1 + v[10]*x2 + v[16]*x3 + v[22]*x4 + v[28]*x5 + v[34]*x6; 200315091d37SBarry Smith sum6 += v[5]*x1 + v[11]*x2 + v[17]*x3 + v[23]*x4 + v[29]*x5 + v[35]*x6; 200415091d37SBarry Smith v += 36; 200515091d37SBarry Smith } 200615091d37SBarry Smith z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; 200726e093fcSHong Zhang if (!usecprow) { 200815091d37SBarry Smith z += 6; y += 6; 200915091d37SBarry Smith } 201026e093fcSHong Zhang } 2011d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 2012d9ca1df4SBarry Smith ierr = VecRestoreArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 2013dc0b31edSSatish Balay ierr = PetscLogFlops(72.0*a->nz);CHKERRQ(ierr); 201415091d37SBarry Smith PetscFunctionReturn(0); 201515091d37SBarry Smith } 20162d61bbb3SSatish Balay 20174a2ae208SSatish Balay #undef __FUNCT__ 20184a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_7" 2019dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_7(Mat A,Vec xx,Vec yy,Vec zz) 20202d61bbb3SSatish Balay { 20212d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 2022d9ca1df4SBarry Smith PetscScalar *y = 0,*z = 0,sum1,sum2,sum3,sum4,sum5,sum6,sum7; 2023d9ca1df4SBarry Smith const PetscScalar *x,*xb; 202426e093fcSHong Zhang PetscScalar x1,x2,x3,x4,x5,x6,x7,*yarray,*zarray; 2025d9ca1df4SBarry Smith const MatScalar *v; 2026dfbe8321SBarry Smith PetscErrorCode ierr; 2027d9ca1df4SBarry Smith PetscInt mbs = a->mbs,i,j,n; 2028d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ridx = NULL; 2029ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 20302d61bbb3SSatish Balay 20312d61bbb3SSatish Balay PetscFunctionBegin; 2032d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 2033d9ca1df4SBarry Smith ierr = VecGetArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 20342d61bbb3SSatish Balay 20352d61bbb3SSatish Balay idx = a->j; 20362d61bbb3SSatish Balay v = a->a; 203726e093fcSHong Zhang if (usecprow) { 20384eb6d288SHong Zhang if (zz != yy) { 20394eb6d288SHong Zhang ierr = PetscMemcpy(zarray,yarray,7*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 20404eb6d288SHong Zhang } 204126e093fcSHong Zhang mbs = a->compressedrow.nrows; 204226e093fcSHong Zhang ii = a->compressedrow.i; 20437b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 204426e093fcSHong Zhang } else { 20452d61bbb3SSatish Balay ii = a->i; 204626e093fcSHong Zhang y = yarray; 204726e093fcSHong Zhang z = zarray; 204826e093fcSHong Zhang } 20492d61bbb3SSatish Balay 20502d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 20512d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 205226e093fcSHong Zhang if (usecprow) { 20537b2bb3b9SHong Zhang z = zarray + 7*ridx[i]; 20547b2bb3b9SHong Zhang y = yarray + 7*ridx[i]; 205526e093fcSHong Zhang } 20562d61bbb3SSatish Balay sum1 = y[0]; sum2 = y[1]; sum3 = y[2]; sum4 = y[3]; sum5 = y[4]; sum6 = y[5]; sum7 = y[6]; 2057444d8c10SJed Brown PetscPrefetchBlock(idx+n,n,0,PETSC_PREFETCH_HINT_NTA); /* Indices for the next row (assumes same size as this one) */ 2058444d8c10SJed Brown PetscPrefetchBlock(v+49*n,49*n,0,PETSC_PREFETCH_HINT_NTA); /* Entries for the next row */ 20592d61bbb3SSatish Balay for (j=0; j<n; j++) { 20602d61bbb3SSatish Balay xb = x + 7*(*idx++); 20612d61bbb3SSatish Balay x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; x5 = xb[4]; x6 = xb[5]; x7 = xb[6]; 20622d61bbb3SSatish Balay sum1 += v[0]*x1 + v[7]*x2 + v[14]*x3 + v[21]*x4 + v[28]*x5 + v[35]*x6 + v[42]*x7; 20632d61bbb3SSatish Balay sum2 += v[1]*x1 + v[8]*x2 + v[15]*x3 + v[22]*x4 + v[29]*x5 + v[36]*x6 + v[43]*x7; 20642d61bbb3SSatish Balay sum3 += v[2]*x1 + v[9]*x2 + v[16]*x3 + v[23]*x4 + v[30]*x5 + v[37]*x6 + v[44]*x7; 20652d61bbb3SSatish Balay sum4 += v[3]*x1 + v[10]*x2 + v[17]*x3 + v[24]*x4 + v[31]*x5 + v[38]*x6 + v[45]*x7; 20662d61bbb3SSatish Balay sum5 += v[4]*x1 + v[11]*x2 + v[18]*x3 + v[25]*x4 + v[32]*x5 + v[39]*x6 + v[46]*x7; 20672d61bbb3SSatish Balay sum6 += v[5]*x1 + v[12]*x2 + v[19]*x3 + v[26]*x4 + v[33]*x5 + v[40]*x6 + v[47]*x7; 20682d61bbb3SSatish Balay sum7 += v[6]*x1 + v[13]*x2 + v[20]*x3 + v[27]*x4 + v[34]*x5 + v[41]*x6 + v[48]*x7; 20692d61bbb3SSatish Balay v += 49; 20702d61bbb3SSatish Balay } 20712d61bbb3SSatish Balay z[0] = sum1; z[1] = sum2; z[2] = sum3; z[3] = sum4; z[4] = sum5; z[5] = sum6; z[6] = sum7; 207226e093fcSHong Zhang if (!usecprow) { 20732d61bbb3SSatish Balay z += 7; y += 7; 20742d61bbb3SSatish Balay } 207526e093fcSHong Zhang } 2076d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 2077d9ca1df4SBarry Smith ierr = VecRestoreArrayPair(yy,zz,&yarray,&zarray);CHKERRQ(ierr); 2078dc0b31edSSatish Balay ierr = PetscLogFlops(98.0*a->nz);CHKERRQ(ierr); 20792d61bbb3SSatish Balay PetscFunctionReturn(0); 20802d61bbb3SSatish Balay } 2081218c64b6SSatish Balay 20824a2ae208SSatish Balay #undef __FUNCT__ 20834a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqBAIJ_N" 2084dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqBAIJ_N(Mat A,Vec xx,Vec yy,Vec zz) 20852d61bbb3SSatish Balay { 20862d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 2087d9ca1df4SBarry Smith PetscScalar *z = 0,*work,*workt,*zarray; 2088d9ca1df4SBarry Smith const PetscScalar *x,*xb; 2089d9ca1df4SBarry Smith const MatScalar *v; 20906849ba73SBarry Smith PetscErrorCode ierr; 2091d9ca1df4SBarry Smith PetscInt mbs,i,bs=A->rmap->bs,j,n,bs2=a->bs2; 2092d9ca1df4SBarry Smith PetscInt ncols,k; 2093d9ca1df4SBarry Smith const PetscInt *ridx = NULL,*idx,*ii; 2094ace3abfcSBarry Smith PetscBool usecprow = a->compressedrow.use; 2095218c64b6SSatish Balay 20962d61bbb3SSatish Balay PetscFunctionBegin; 2097343551c4SBarry Smith ierr = VecCopy(yy,zz);CHKERRQ(ierr); 2098d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 209926e093fcSHong Zhang ierr = VecGetArray(zz,&zarray);CHKERRQ(ierr); 21002d61bbb3SSatish Balay 21012d61bbb3SSatish Balay idx = a->j; 21022d61bbb3SSatish Balay v = a->a; 210326e093fcSHong Zhang if (usecprow) { 210426e093fcSHong Zhang mbs = a->compressedrow.nrows; 210526e093fcSHong Zhang ii = a->compressedrow.i; 21067b2bb3b9SHong Zhang ridx = a->compressedrow.rindex; 210726e093fcSHong Zhang } else { 210826e093fcSHong Zhang mbs = a->mbs; 21092d61bbb3SSatish Balay ii = a->i; 211026e093fcSHong Zhang z = zarray; 211126e093fcSHong Zhang } 21122d61bbb3SSatish Balay 21132d61bbb3SSatish Balay if (!a->mult_work) { 2114d0f46423SBarry Smith k = PetscMax(A->rmap->n,A->cmap->n); 2115854ce69bSBarry Smith ierr = PetscMalloc1(k+1,&a->mult_work);CHKERRQ(ierr); 21162d61bbb3SSatish Balay } 21172d61bbb3SSatish Balay work = a->mult_work; 21182d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 21192d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 21202d61bbb3SSatish Balay ncols = n*bs; 21212d61bbb3SSatish Balay workt = work; 21222d61bbb3SSatish Balay for (j=0; j<n; j++) { 21232d61bbb3SSatish Balay xb = x + bs*(*idx++); 21242d61bbb3SSatish Balay for (k=0; k<bs; k++) workt[k] = xb[k]; 21252d61bbb3SSatish Balay workt += bs; 21262d61bbb3SSatish Balay } 21277b2bb3b9SHong Zhang if (usecprow) z = zarray + bs*ridx[i]; 212896b95a6bSBarry Smith PetscKernel_w_gets_w_plus_Ar_times_v(bs,ncols,work,v,z); 212971044d3cSBarry Smith /* BLASgemv_("N",&bs,&ncols,&_DOne,v,&bs,work,&_One,&_DOne,z,&_One); */ 21302d61bbb3SSatish Balay v += n*bs2; 213126fbe8dcSKarl Rupp if (!usecprow) z += bs; 213226e093fcSHong Zhang } 2133d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 213426e093fcSHong Zhang ierr = VecRestoreArray(zz,&zarray);CHKERRQ(ierr); 2135dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz*bs2);CHKERRQ(ierr); 21362d61bbb3SSatish Balay PetscFunctionReturn(0); 21372d61bbb3SSatish Balay } 21382d61bbb3SSatish Balay 21394a2ae208SSatish Balay #undef __FUNCT__ 2140547795f9SHong Zhang #define __FUNCT__ "MatMultHermitianTranspose_SeqBAIJ" 2141547795f9SHong Zhang PetscErrorCode MatMultHermitianTranspose_SeqBAIJ(Mat A,Vec xx,Vec zz) 2142547795f9SHong Zhang { 2143547795f9SHong Zhang PetscScalar zero = 0.0; 2144547795f9SHong Zhang PetscErrorCode ierr; 2145547795f9SHong Zhang 2146547795f9SHong Zhang PetscFunctionBegin; 2147547795f9SHong Zhang ierr = VecSet(zz,zero);CHKERRQ(ierr); 2148547795f9SHong Zhang ierr = MatMultHermitianTransposeAdd_SeqBAIJ(A,xx,zz,zz);CHKERRQ(ierr); 2149547795f9SHong Zhang PetscFunctionReturn(0); 2150547795f9SHong Zhang } 2151547795f9SHong Zhang 2152547795f9SHong Zhang #undef __FUNCT__ 21534a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqBAIJ" 2154dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqBAIJ(Mat A,Vec xx,Vec zz) 21552d61bbb3SSatish Balay { 21563447b6efSHong Zhang PetscScalar zero = 0.0; 21576849ba73SBarry Smith PetscErrorCode ierr; 21582d61bbb3SSatish Balay 21592d61bbb3SSatish Balay PetscFunctionBegin; 21602dcb1b2aSMatthew Knepley ierr = VecSet(zz,zero);CHKERRQ(ierr); 21613447b6efSHong Zhang ierr = MatMultTransposeAdd_SeqBAIJ(A,xx,zz,zz);CHKERRQ(ierr); 21622d61bbb3SSatish Balay PetscFunctionReturn(0); 21632d61bbb3SSatish Balay } 21642d61bbb3SSatish Balay 21654a2ae208SSatish Balay #undef __FUNCT__ 2166547795f9SHong Zhang #define __FUNCT__ "MatMultHermitianTransposeAdd_SeqBAIJ" 2167547795f9SHong Zhang PetscErrorCode MatMultHermitianTransposeAdd_SeqBAIJ(Mat A,Vec xx,Vec yy,Vec zz) 2168547795f9SHong Zhang { 2169547795f9SHong Zhang Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 2170d9ca1df4SBarry Smith PetscScalar *zb,*z,x1,x2,x3,x4,x5; 2171d9ca1df4SBarry Smith const PetscScalar *x,*xb = NULL; 2172d9ca1df4SBarry Smith const MatScalar *v; 2173547795f9SHong Zhang PetscErrorCode ierr; 2174d9ca1df4SBarry Smith PetscInt mbs,i,rval,bs=A->rmap->bs,j,n,bs2=a->bs2; 2175d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ib,*ridx = NULL; 2176547795f9SHong Zhang Mat_CompressedRow cprow = a->compressedrow; 2177ace3abfcSBarry Smith PetscBool usecprow = cprow.use; 2178547795f9SHong Zhang 2179547795f9SHong Zhang PetscFunctionBegin; 2180343551c4SBarry Smith if (yy != zz) { ierr = VecCopy(yy,zz);CHKERRQ(ierr); } 2181d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 2182547795f9SHong Zhang ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 2183547795f9SHong Zhang 2184547795f9SHong Zhang idx = a->j; 2185547795f9SHong Zhang v = a->a; 2186547795f9SHong Zhang if (usecprow) { 2187547795f9SHong Zhang mbs = cprow.nrows; 2188547795f9SHong Zhang ii = cprow.i; 2189547795f9SHong Zhang ridx = cprow.rindex; 2190547795f9SHong Zhang } else { 2191547795f9SHong Zhang mbs=a->mbs; 2192547795f9SHong Zhang ii = a->i; 2193547795f9SHong Zhang xb = x; 2194547795f9SHong Zhang } 2195547795f9SHong Zhang 2196547795f9SHong Zhang switch (bs) { 2197547795f9SHong Zhang case 1: 2198547795f9SHong Zhang for (i=0; i<mbs; i++) { 2199547795f9SHong Zhang if (usecprow) xb = x + ridx[i]; 2200547795f9SHong Zhang x1 = xb[0]; 2201547795f9SHong Zhang ib = idx + ii[0]; 2202547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 2203547795f9SHong Zhang for (j=0; j<n; j++) { 2204547795f9SHong Zhang rval = ib[j]; 2205547795f9SHong Zhang z[rval] += PetscConj(*v) * x1; 2206547795f9SHong Zhang v++; 2207547795f9SHong Zhang } 2208547795f9SHong Zhang if (!usecprow) xb++; 2209547795f9SHong Zhang } 2210547795f9SHong Zhang break; 2211547795f9SHong Zhang case 2: 2212547795f9SHong Zhang for (i=0; i<mbs; i++) { 2213547795f9SHong Zhang if (usecprow) xb = x + 2*ridx[i]; 2214547795f9SHong Zhang x1 = xb[0]; x2 = xb[1]; 2215547795f9SHong Zhang ib = idx + ii[0]; 2216547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 2217547795f9SHong Zhang for (j=0; j<n; j++) { 2218547795f9SHong Zhang rval = ib[j]*2; 2219547795f9SHong Zhang z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2; 2220547795f9SHong Zhang z[rval++] += PetscConj(v[2])*x1 + PetscConj(v[3])*x2; 2221547795f9SHong Zhang v += 4; 2222547795f9SHong Zhang } 2223547795f9SHong Zhang if (!usecprow) xb += 2; 2224547795f9SHong Zhang } 2225547795f9SHong Zhang break; 2226547795f9SHong Zhang case 3: 2227547795f9SHong Zhang for (i=0; i<mbs; i++) { 2228547795f9SHong Zhang if (usecprow) xb = x + 3*ridx[i]; 2229547795f9SHong Zhang x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 2230547795f9SHong Zhang ib = idx + ii[0]; 2231547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 2232547795f9SHong Zhang for (j=0; j<n; j++) { 2233547795f9SHong Zhang rval = ib[j]*3; 2234547795f9SHong Zhang z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2 + PetscConj(v[2])*x3; 2235547795f9SHong Zhang z[rval++] += PetscConj(v[3])*x1 + PetscConj(v[4])*x2 + PetscConj(v[5])*x3; 2236547795f9SHong Zhang z[rval++] += PetscConj(v[6])*x1 + PetscConj(v[7])*x2 + PetscConj(v[8])*x3; 2237547795f9SHong Zhang v += 9; 2238547795f9SHong Zhang } 2239547795f9SHong Zhang if (!usecprow) xb += 3; 2240547795f9SHong Zhang } 2241547795f9SHong Zhang break; 2242547795f9SHong Zhang case 4: 2243547795f9SHong Zhang for (i=0; i<mbs; i++) { 2244547795f9SHong Zhang if (usecprow) xb = x + 4*ridx[i]; 2245547795f9SHong Zhang x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 2246547795f9SHong Zhang ib = idx + ii[0]; 2247547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 2248547795f9SHong Zhang for (j=0; j<n; j++) { 2249547795f9SHong Zhang rval = ib[j]*4; 2250547795f9SHong Zhang z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2 + PetscConj(v[2])*x3 + PetscConj(v[3])*x4; 2251547795f9SHong Zhang z[rval++] += PetscConj(v[4])*x1 + PetscConj(v[5])*x2 + PetscConj(v[6])*x3 + PetscConj(v[7])*x4; 2252547795f9SHong Zhang z[rval++] += PetscConj(v[8])*x1 + PetscConj(v[9])*x2 + PetscConj(v[10])*x3 + PetscConj(v[11])*x4; 2253547795f9SHong Zhang z[rval++] += PetscConj(v[12])*x1 + PetscConj(v[13])*x2 + PetscConj(v[14])*x3 + PetscConj(v[15])*x4; 2254547795f9SHong Zhang v += 16; 2255547795f9SHong Zhang } 2256547795f9SHong Zhang if (!usecprow) xb += 4; 2257547795f9SHong Zhang } 2258547795f9SHong Zhang break; 2259547795f9SHong Zhang case 5: 2260547795f9SHong Zhang for (i=0; i<mbs; i++) { 2261547795f9SHong Zhang if (usecprow) xb = x + 5*ridx[i]; 2262547795f9SHong Zhang x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 2263547795f9SHong Zhang x4 = xb[3]; x5 = xb[4]; 2264547795f9SHong Zhang ib = idx + ii[0]; 2265547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 2266547795f9SHong Zhang for (j=0; j<n; j++) { 2267547795f9SHong Zhang rval = ib[j]*5; 2268547795f9SHong Zhang z[rval++] += PetscConj(v[0])*x1 + PetscConj(v[1])*x2 + PetscConj(v[2])*x3 + PetscConj(v[3])*x4 + PetscConj(v[4])*x5; 2269547795f9SHong Zhang z[rval++] += PetscConj(v[5])*x1 + PetscConj(v[6])*x2 + PetscConj(v[7])*x3 + PetscConj(v[8])*x4 + PetscConj(v[9])*x5; 2270547795f9SHong Zhang z[rval++] += PetscConj(v[10])*x1 + PetscConj(v[11])*x2 + PetscConj(v[12])*x3 + PetscConj(v[13])*x4 + PetscConj(v[14])*x5; 2271547795f9SHong Zhang z[rval++] += PetscConj(v[15])*x1 + PetscConj(v[16])*x2 + PetscConj(v[17])*x3 + PetscConj(v[18])*x4 + PetscConj(v[19])*x5; 2272547795f9SHong Zhang z[rval++] += PetscConj(v[20])*x1 + PetscConj(v[21])*x2 + PetscConj(v[22])*x3 + PetscConj(v[23])*x4 + PetscConj(v[24])*x5; 2273547795f9SHong Zhang v += 25; 2274547795f9SHong Zhang } 2275547795f9SHong Zhang if (!usecprow) xb += 5; 2276547795f9SHong Zhang } 2277547795f9SHong Zhang break; 2278*968ae2c8SSatish Balay default: /* block sizes larger than 5 by 5 are handled by BLAS */ 2279*968ae2c8SSatish Balay SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size larger than 5 is not supported yet"); 2280*968ae2c8SSatish Balay #if 0 2281*968ae2c8SSatish Balay { 2282547795f9SHong Zhang PetscInt ncols,k; 2283d9ca1df4SBarry Smith PetscScalar *work,*workt; 2284d9ca1df4SBarry Smith const PetscScalar *xtmp; 2285547795f9SHong Zhang if (!a->mult_work) { 2286547795f9SHong Zhang k = PetscMax(A->rmap->n,A->cmap->n); 2287854ce69bSBarry Smith ierr = PetscMalloc1(k+1,&a->mult_work);CHKERRQ(ierr); 2288547795f9SHong Zhang } 2289547795f9SHong Zhang work = a->mult_work; 2290547795f9SHong Zhang xtmp = x; 2291547795f9SHong Zhang for (i=0; i<mbs; i++) { 2292547795f9SHong Zhang n = ii[1] - ii[0]; ii++; 2293547795f9SHong Zhang ncols = n*bs; 2294547795f9SHong Zhang ierr = PetscMemzero(work,ncols*sizeof(PetscScalar));CHKERRQ(ierr); 229526fbe8dcSKarl Rupp if (usecprow) xtmp = x + bs*ridx[i]; 229696b95a6bSBarry Smith PetscKernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,xtmp,v,work); 2297547795f9SHong Zhang /* BLASgemv_("T",&bs,&ncols,&_DOne,v,&bs,xtmp,&_One,&_DOne,work,&_One); */ 2298547795f9SHong Zhang v += n*bs2; 2299547795f9SHong Zhang if (!usecprow) xtmp += bs; 2300547795f9SHong Zhang workt = work; 2301547795f9SHong Zhang for (j=0; j<n; j++) { 2302547795f9SHong Zhang zb = z + bs*(*idx++); 2303547795f9SHong Zhang for (k=0; k<bs; k++) zb[k] += workt[k] ; 2304547795f9SHong Zhang workt += bs; 2305547795f9SHong Zhang } 2306547795f9SHong Zhang } 2307547795f9SHong Zhang } 2308*968ae2c8SSatish Balay #endif 2309547795f9SHong Zhang } 2310d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 2311547795f9SHong Zhang ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 2312547795f9SHong Zhang ierr = PetscLogFlops(2.0*a->nz*a->bs2);CHKERRQ(ierr); 2313547795f9SHong Zhang PetscFunctionReturn(0); 2314547795f9SHong Zhang } 2315547795f9SHong Zhang 2316547795f9SHong Zhang #undef __FUNCT__ 23174a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqBAIJ" 2318dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqBAIJ(Mat A,Vec xx,Vec yy,Vec zz) 23192d61bbb3SSatish Balay { 23202d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 2321d9ca1df4SBarry Smith PetscScalar *zb,*z,x1,x2,x3,x4,x5; 2322d9ca1df4SBarry Smith const PetscScalar *x,*xb = 0; 2323d9ca1df4SBarry Smith const MatScalar *v; 23246849ba73SBarry Smith PetscErrorCode ierr; 2325d9ca1df4SBarry Smith PetscInt mbs,i,rval,bs=A->rmap->bs,j,n,bs2=a->bs2; 2326d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ib,*ridx = NULL; 23273447b6efSHong Zhang Mat_CompressedRow cprow = a->compressedrow; 2328ace3abfcSBarry Smith PetscBool usecprow=cprow.use; 23292d61bbb3SSatish Balay 23302d61bbb3SSatish Balay PetscFunctionBegin; 2331343551c4SBarry Smith if (yy != zz) { ierr = VecCopy(yy,zz);CHKERRQ(ierr); } 2332d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 23333447b6efSHong Zhang ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 23342d61bbb3SSatish Balay 23352d61bbb3SSatish Balay idx = a->j; 23362d61bbb3SSatish Balay v = a->a; 23373447b6efSHong Zhang if (usecprow) { 23383447b6efSHong Zhang mbs = cprow.nrows; 23393447b6efSHong Zhang ii = cprow.i; 23407b2bb3b9SHong Zhang ridx = cprow.rindex; 23413447b6efSHong Zhang } else { 23423447b6efSHong Zhang mbs=a->mbs; 23432d61bbb3SSatish Balay ii = a->i; 2344f1af5d2fSBarry Smith xb = x; 23453447b6efSHong Zhang } 23462d61bbb3SSatish Balay 23472d61bbb3SSatish Balay switch (bs) { 23482d61bbb3SSatish Balay case 1: 23492d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 23507b2bb3b9SHong Zhang if (usecprow) xb = x + ridx[i]; 2351f1af5d2fSBarry Smith x1 = xb[0]; 23523447b6efSHong Zhang ib = idx + ii[0]; 23533447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 23542d61bbb3SSatish Balay for (j=0; j<n; j++) { 23552d61bbb3SSatish Balay rval = ib[j]; 2356f1af5d2fSBarry Smith z[rval] += *v * x1; 2357f1af5d2fSBarry Smith v++; 23582d61bbb3SSatish Balay } 23593447b6efSHong Zhang if (!usecprow) xb++; 23602d61bbb3SSatish Balay } 23612d61bbb3SSatish Balay break; 23622d61bbb3SSatish Balay case 2: 23632d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 23647b2bb3b9SHong Zhang if (usecprow) xb = x + 2*ridx[i]; 2365f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; 23663447b6efSHong Zhang ib = idx + ii[0]; 23673447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 23682d61bbb3SSatish Balay for (j=0; j<n; j++) { 23692d61bbb3SSatish Balay rval = ib[j]*2; 23702d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2; 23712d61bbb3SSatish Balay z[rval++] += v[2]*x1 + v[3]*x2; 23722d61bbb3SSatish Balay v += 4; 23732d61bbb3SSatish Balay } 23743447b6efSHong Zhang if (!usecprow) xb += 2; 23752d61bbb3SSatish Balay } 23762d61bbb3SSatish Balay break; 23772d61bbb3SSatish Balay case 3: 23782d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 23797b2bb3b9SHong Zhang if (usecprow) xb = x + 3*ridx[i]; 2380f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 23813447b6efSHong Zhang ib = idx + ii[0]; 23823447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 23832d61bbb3SSatish Balay for (j=0; j<n; j++) { 23842d61bbb3SSatish Balay rval = ib[j]*3; 23852d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3; 23862d61bbb3SSatish Balay z[rval++] += v[3]*x1 + v[4]*x2 + v[5]*x3; 23872d61bbb3SSatish Balay z[rval++] += v[6]*x1 + v[7]*x2 + v[8]*x3; 23882d61bbb3SSatish Balay v += 9; 23892d61bbb3SSatish Balay } 23903447b6efSHong Zhang if (!usecprow) xb += 3; 23912d61bbb3SSatish Balay } 23922d61bbb3SSatish Balay break; 23932d61bbb3SSatish Balay case 4: 23942d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 23957b2bb3b9SHong Zhang if (usecprow) xb = x + 4*ridx[i]; 2396f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; x4 = xb[3]; 23973447b6efSHong Zhang ib = idx + ii[0]; 23983447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 23992d61bbb3SSatish Balay for (j=0; j<n; j++) { 24002d61bbb3SSatish Balay rval = ib[j]*4; 24012d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4; 24022d61bbb3SSatish Balay z[rval++] += v[4]*x1 + v[5]*x2 + v[6]*x3 + v[7]*x4; 24032d61bbb3SSatish Balay z[rval++] += v[8]*x1 + v[9]*x2 + v[10]*x3 + v[11]*x4; 24042d61bbb3SSatish Balay z[rval++] += v[12]*x1 + v[13]*x2 + v[14]*x3 + v[15]*x4; 24052d61bbb3SSatish Balay v += 16; 24062d61bbb3SSatish Balay } 24073447b6efSHong Zhang if (!usecprow) xb += 4; 24082d61bbb3SSatish Balay } 24092d61bbb3SSatish Balay break; 24102d61bbb3SSatish Balay case 5: 24112d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 24127b2bb3b9SHong Zhang if (usecprow) xb = x + 5*ridx[i]; 2413f1af5d2fSBarry Smith x1 = xb[0]; x2 = xb[1]; x3 = xb[2]; 24142d61bbb3SSatish Balay x4 = xb[3]; x5 = xb[4]; 24153447b6efSHong Zhang ib = idx + ii[0]; 24163447b6efSHong Zhang n = ii[1] - ii[0]; ii++; 24172d61bbb3SSatish Balay for (j=0; j<n; j++) { 24182d61bbb3SSatish Balay rval = ib[j]*5; 24192d61bbb3SSatish Balay z[rval++] += v[0]*x1 + v[1]*x2 + v[2]*x3 + v[3]*x4 + v[4]*x5; 24202d61bbb3SSatish Balay z[rval++] += v[5]*x1 + v[6]*x2 + v[7]*x3 + v[8]*x4 + v[9]*x5; 24212d61bbb3SSatish Balay z[rval++] += v[10]*x1 + v[11]*x2 + v[12]*x3 + v[13]*x4 + v[14]*x5; 24222d61bbb3SSatish Balay z[rval++] += v[15]*x1 + v[16]*x2 + v[17]*x3 + v[18]*x4 + v[19]*x5; 24232d61bbb3SSatish Balay z[rval++] += v[20]*x1 + v[21]*x2 + v[22]*x3 + v[23]*x4 + v[24]*x5; 24242d61bbb3SSatish Balay v += 25; 24252d61bbb3SSatish Balay } 24263447b6efSHong Zhang if (!usecprow) xb += 5; 24272d61bbb3SSatish Balay } 24282d61bbb3SSatish Balay break; 2429f1af5d2fSBarry Smith default: { /* block sizes larger then 5 by 5 are handled by BLAS */ 2430690b6cddSBarry Smith PetscInt ncols,k; 2431d9ca1df4SBarry Smith PetscScalar *work,*workt; 2432d9ca1df4SBarry Smith const PetscScalar *xtmp; 24332d61bbb3SSatish Balay if (!a->mult_work) { 2434d0f46423SBarry Smith k = PetscMax(A->rmap->n,A->cmap->n); 2435854ce69bSBarry Smith ierr = PetscMalloc1(k+1,&a->mult_work);CHKERRQ(ierr); 24362d61bbb3SSatish Balay } 24372d61bbb3SSatish Balay work = a->mult_work; 24383447b6efSHong Zhang xtmp = x; 24392d61bbb3SSatish Balay for (i=0; i<mbs; i++) { 24402d61bbb3SSatish Balay n = ii[1] - ii[0]; ii++; 24412d61bbb3SSatish Balay ncols = n*bs; 244287828ca2SBarry Smith ierr = PetscMemzero(work,ncols*sizeof(PetscScalar));CHKERRQ(ierr); 244326fbe8dcSKarl Rupp if (usecprow) xtmp = x + bs*ridx[i]; 244496b95a6bSBarry Smith PetscKernel_w_gets_w_plus_trans_Ar_times_v(bs,ncols,xtmp,v,work); 244571044d3cSBarry Smith /* BLASgemv_("T",&bs,&ncols,&_DOne,v,&bs,xtmp,&_One,&_DOne,work,&_One); */ 24462d61bbb3SSatish Balay v += n*bs2; 24473447b6efSHong Zhang if (!usecprow) xtmp += bs; 24482d61bbb3SSatish Balay workt = work; 24492d61bbb3SSatish Balay for (j=0; j<n; j++) { 24502d61bbb3SSatish Balay zb = z + bs*(*idx++); 24512d61bbb3SSatish Balay for (k=0; k<bs; k++) zb[k] += workt[k]; 24522d61bbb3SSatish Balay workt += bs; 24532d61bbb3SSatish Balay } 24542d61bbb3SSatish Balay } 24552d61bbb3SSatish Balay } 24562d61bbb3SSatish Balay } 2457d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 24583447b6efSHong Zhang ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 2459dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz*a->bs2);CHKERRQ(ierr); 24602d61bbb3SSatish Balay PetscFunctionReturn(0); 24612d61bbb3SSatish Balay } 24622d61bbb3SSatish Balay 24634a2ae208SSatish Balay #undef __FUNCT__ 24644a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqBAIJ" 2465f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqBAIJ(Mat inA,PetscScalar alpha) 24662d61bbb3SSatish Balay { 24672d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)inA->data; 2468690b6cddSBarry Smith PetscInt totalnz = a->bs2*a->nz; 2469f4df32b1SMatthew Knepley PetscScalar oalpha = alpha; 2470efee365bSSatish Balay PetscErrorCode ierr; 2471c5df96a5SBarry Smith PetscBLASInt one = 1,tnz; 24722d61bbb3SSatish Balay 24732d61bbb3SSatish Balay PetscFunctionBegin; 2474c5df96a5SBarry Smith ierr = PetscBLASIntCast(totalnz,&tnz);CHKERRQ(ierr); 24758b83055fSJed Brown PetscStackCallBLAS("BLASscal",BLASscal_(&tnz,&oalpha,a->a,&one)); 2476efee365bSSatish Balay ierr = PetscLogFlops(totalnz);CHKERRQ(ierr); 24772d61bbb3SSatish Balay PetscFunctionReturn(0); 24782d61bbb3SSatish Balay } 24792d61bbb3SSatish Balay 24804a2ae208SSatish Balay #undef __FUNCT__ 24814a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqBAIJ" 2482dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqBAIJ(Mat A,NormType type,PetscReal *norm) 24832d61bbb3SSatish Balay { 24848a62d963SHong Zhang PetscErrorCode ierr; 24852d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 24863f1db9ecSBarry Smith MatScalar *v = a->a; 2487329f5518SBarry Smith PetscReal sum = 0.0; 2488d0f46423SBarry Smith PetscInt i,j,k,bs=A->rmap->bs,nz=a->nz,bs2=a->bs2,k1; 24892d61bbb3SSatish Balay 24902d61bbb3SSatish Balay PetscFunctionBegin; 24912d61bbb3SSatish Balay if (type == NORM_FROBENIUS) { 24922d61bbb3SSatish Balay for (i=0; i< bs2*nz; i++) { 2493329f5518SBarry Smith sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 24942d61bbb3SSatish Balay } 24958f1a2a5eSBarry Smith *norm = PetscSqrtReal(sum); 24968a62d963SHong Zhang } else if (type == NORM_1) { /* maximum column sum */ 24978a62d963SHong Zhang PetscReal *tmp; 24988a62d963SHong Zhang PetscInt *bcol = a->j; 2499854ce69bSBarry Smith ierr = PetscCalloc1(A->cmap->n+1,&tmp);CHKERRQ(ierr); 25008a62d963SHong Zhang for (i=0; i<nz; i++) { 25018a62d963SHong Zhang for (j=0; j<bs; j++) { 25028a62d963SHong Zhang k1 = bs*(*bcol) + j; /* column index */ 25038a62d963SHong Zhang for (k=0; k<bs; k++) { 25048a62d963SHong Zhang tmp[k1] += PetscAbsScalar(*v); v++; 25058a62d963SHong Zhang } 25068a62d963SHong Zhang } 25078a62d963SHong Zhang bcol++; 25088a62d963SHong Zhang } 25098a62d963SHong Zhang *norm = 0.0; 2510d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 25118a62d963SHong Zhang if (tmp[j] > *norm) *norm = tmp[j]; 25128a62d963SHong Zhang } 25138a62d963SHong Zhang ierr = PetscFree(tmp);CHKERRQ(ierr); 2514596552b5SBarry Smith } else if (type == NORM_INFINITY) { /* maximum row sum */ 2515596552b5SBarry Smith *norm = 0.0; 2516596552b5SBarry Smith for (k=0; k<bs; k++) { 251774f84c7bSSatish Balay for (j=0; j<a->mbs; j++) { 2518596552b5SBarry Smith v = a->a + bs2*a->i[j] + k; 2519596552b5SBarry Smith sum = 0.0; 2520596552b5SBarry Smith for (i=0; i<a->i[j+1]-a->i[j]; i++) { 25210e90e235SBarry Smith for (k1=0; k1<bs; k1++) { 2522596552b5SBarry Smith sum += PetscAbsScalar(*v); 2523596552b5SBarry Smith v += bs; 25242d61bbb3SSatish Balay } 25250e90e235SBarry Smith } 2526596552b5SBarry Smith if (sum > *norm) *norm = sum; 2527596552b5SBarry Smith } 2528596552b5SBarry Smith } 2529e7e72b3dSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for this norm yet"); 25302d61bbb3SSatish Balay PetscFunctionReturn(0); 25312d61bbb3SSatish Balay } 25322d61bbb3SSatish Balay 25332d61bbb3SSatish Balay 25344a2ae208SSatish Balay #undef __FUNCT__ 25354a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqBAIJ" 2536ace3abfcSBarry Smith PetscErrorCode MatEqual_SeqBAIJ(Mat A,Mat B,PetscBool * flg) 25372d61bbb3SSatish Balay { 25382d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data,*b = (Mat_SeqBAIJ*)B->data; 2539dfbe8321SBarry Smith PetscErrorCode ierr; 25402d61bbb3SSatish Balay 25412d61bbb3SSatish Balay PetscFunctionBegin; 25422d61bbb3SSatish Balay /* If the matrix/block dimensions are not equal, or no of nonzeros or shift */ 2543d0f46423SBarry Smith if ((A->rmap->N != B->rmap->N) || (A->cmap->n != B->cmap->n) || (A->rmap->bs != B->rmap->bs)|| (a->nz != b->nz)) { 2544273d9f13SBarry Smith *flg = PETSC_FALSE; 2545273d9f13SBarry Smith PetscFunctionReturn(0); 25462d61bbb3SSatish Balay } 25472d61bbb3SSatish Balay 25482d61bbb3SSatish Balay /* if the a->i are the same */ 2549690b6cddSBarry Smith ierr = PetscMemcmp(a->i,b->i,(a->mbs+1)*sizeof(PetscInt),flg);CHKERRQ(ierr); 255026fbe8dcSKarl Rupp if (!*flg) PetscFunctionReturn(0); 25512d61bbb3SSatish Balay 25522d61bbb3SSatish Balay /* if a->j are the same */ 2553690b6cddSBarry Smith ierr = PetscMemcmp(a->j,b->j,(a->nz)*sizeof(PetscInt),flg);CHKERRQ(ierr); 255426fbe8dcSKarl Rupp if (!*flg) PetscFunctionReturn(0); 255526fbe8dcSKarl Rupp 25562d61bbb3SSatish Balay /* if a->a are the same */ 2557d0f46423SBarry Smith ierr = PetscMemcmp(a->a,b->a,(a->nz)*(A->rmap->bs)*(B->rmap->bs)*sizeof(PetscScalar),flg);CHKERRQ(ierr); 25582d61bbb3SSatish Balay PetscFunctionReturn(0); 25592d61bbb3SSatish Balay 25602d61bbb3SSatish Balay } 25612d61bbb3SSatish Balay 25624a2ae208SSatish Balay #undef __FUNCT__ 25634a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqBAIJ" 2564dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqBAIJ(Mat A,Vec v) 25652d61bbb3SSatish Balay { 25662d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 2567dfbe8321SBarry Smith PetscErrorCode ierr; 2568690b6cddSBarry Smith PetscInt i,j,k,n,row,bs,*ai,*aj,ambs,bs2; 256987828ca2SBarry Smith PetscScalar *x,zero = 0.0; 25703f1db9ecSBarry Smith MatScalar *aa,*aa_j; 25712d61bbb3SSatish Balay 25722d61bbb3SSatish Balay PetscFunctionBegin; 2573e32f2f54SBarry Smith if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 2574d0f46423SBarry Smith bs = A->rmap->bs; 25752d61bbb3SSatish Balay aa = a->a; 25762d61bbb3SSatish Balay ai = a->i; 25772d61bbb3SSatish Balay aj = a->j; 25782d61bbb3SSatish Balay ambs = a->mbs; 25792d61bbb3SSatish Balay bs2 = a->bs2; 25802d61bbb3SSatish Balay 25812dcb1b2aSMatthew Knepley ierr = VecSet(v,zero);CHKERRQ(ierr); 25821ebc52fbSHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 2583e1311b90SBarry Smith ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 2584e32f2f54SBarry Smith if (n != A->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 25852d61bbb3SSatish Balay for (i=0; i<ambs; i++) { 25862d61bbb3SSatish Balay for (j=ai[i]; j<ai[i+1]; j++) { 25872d61bbb3SSatish Balay if (aj[j] == i) { 25882d61bbb3SSatish Balay row = i*bs; 25892d61bbb3SSatish Balay aa_j = aa+j*bs2; 25902d61bbb3SSatish Balay for (k=0; k<bs2; k+=(bs+1),row++) x[row] = aa_j[k]; 25912d61bbb3SSatish Balay break; 25922d61bbb3SSatish Balay } 25932d61bbb3SSatish Balay } 25942d61bbb3SSatish Balay } 25951ebc52fbSHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 25962d61bbb3SSatish Balay PetscFunctionReturn(0); 25972d61bbb3SSatish Balay } 25982d61bbb3SSatish Balay 25994a2ae208SSatish Balay #undef __FUNCT__ 26004a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqBAIJ" 2601dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqBAIJ(Mat A,Vec ll,Vec rr) 26022d61bbb3SSatish Balay { 26032d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 260453ef36baSBarry Smith const PetscScalar *l,*r,*li,*ri; 260553ef36baSBarry Smith PetscScalar x; 26063f1db9ecSBarry Smith MatScalar *aa, *v; 2607dfbe8321SBarry Smith PetscErrorCode ierr; 260853ef36baSBarry Smith PetscInt i,j,k,lm,rn,M,m,n,mbs,tmp,bs,bs2,iai; 260953ef36baSBarry Smith const PetscInt *ai,*aj; 26102d61bbb3SSatish Balay 26112d61bbb3SSatish Balay PetscFunctionBegin; 26122d61bbb3SSatish Balay ai = a->i; 26132d61bbb3SSatish Balay aj = a->j; 26142d61bbb3SSatish Balay aa = a->a; 2615d0f46423SBarry Smith m = A->rmap->n; 2616d0f46423SBarry Smith n = A->cmap->n; 2617d0f46423SBarry Smith bs = A->rmap->bs; 26182d61bbb3SSatish Balay mbs = a->mbs; 26192d61bbb3SSatish Balay bs2 = a->bs2; 26202d61bbb3SSatish Balay if (ll) { 262153ef36baSBarry Smith ierr = VecGetArrayRead(ll,&l);CHKERRQ(ierr); 2622e1311b90SBarry Smith ierr = VecGetLocalSize(ll,&lm);CHKERRQ(ierr); 2623e32f2f54SBarry Smith if (lm != m) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length"); 26242d61bbb3SSatish Balay for (i=0; i<mbs; i++) { /* for each block row */ 26252d61bbb3SSatish Balay M = ai[i+1] - ai[i]; 26262d61bbb3SSatish Balay li = l + i*bs; 26272d61bbb3SSatish Balay v = aa + bs2*ai[i]; 26282d61bbb3SSatish Balay for (j=0; j<M; j++) { /* for each block */ 26292d61bbb3SSatish Balay for (k=0; k<bs2; k++) { 26302d61bbb3SSatish Balay (*v++) *= li[k%bs]; 26312d61bbb3SSatish Balay } 26322d61bbb3SSatish Balay } 26332d61bbb3SSatish Balay } 263453ef36baSBarry Smith ierr = VecRestoreArrayRead(ll,&l);CHKERRQ(ierr); 2635efee365bSSatish Balay ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 26362d61bbb3SSatish Balay } 26372d61bbb3SSatish Balay 26382d61bbb3SSatish Balay if (rr) { 263953ef36baSBarry Smith ierr = VecGetArrayRead(rr,&r);CHKERRQ(ierr); 2640e1311b90SBarry Smith ierr = VecGetLocalSize(rr,&rn);CHKERRQ(ierr); 2641e32f2f54SBarry Smith if (rn != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length"); 26422d61bbb3SSatish Balay for (i=0; i<mbs; i++) { /* for each block row */ 264353ef36baSBarry Smith iai = ai[i]; 264453ef36baSBarry Smith M = ai[i+1] - iai; 264553ef36baSBarry Smith v = aa + bs2*iai; 26462d61bbb3SSatish Balay for (j=0; j<M; j++) { /* for each block */ 264753ef36baSBarry Smith ri = r + bs*aj[iai+j]; 26482d61bbb3SSatish Balay for (k=0; k<bs; k++) { 26492d61bbb3SSatish Balay x = ri[k]; 265053ef36baSBarry Smith for (tmp=0; tmp<bs; tmp++) v[tmp] *= x; 265153ef36baSBarry Smith v += bs; 26522d61bbb3SSatish Balay } 26532d61bbb3SSatish Balay } 26542d61bbb3SSatish Balay } 265553ef36baSBarry Smith ierr = VecRestoreArrayRead(rr,&r);CHKERRQ(ierr); 2656efee365bSSatish Balay ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 26572d61bbb3SSatish Balay } 26582d61bbb3SSatish Balay PetscFunctionReturn(0); 26592d61bbb3SSatish Balay } 26602d61bbb3SSatish Balay 26612d61bbb3SSatish Balay 26624a2ae208SSatish Balay #undef __FUNCT__ 26634a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqBAIJ" 2664dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqBAIJ(Mat A,MatInfoType flag,MatInfo *info) 26652d61bbb3SSatish Balay { 26662d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 26672d61bbb3SSatish Balay 26682d61bbb3SSatish Balay PetscFunctionBegin; 26692d61bbb3SSatish Balay info->block_size = a->bs2; 2670ceed8ce5SJed Brown info->nz_allocated = a->bs2*a->maxnz; 26712d61bbb3SSatish Balay info->nz_used = a->bs2*a->nz; 26722d61bbb3SSatish Balay info->nz_unneeded = (double)(info->nz_allocated - info->nz_used); 26732d61bbb3SSatish Balay info->assemblies = A->num_ass; 26748e58a170SBarry Smith info->mallocs = A->info.mallocs; 26757adad957SLisandro Dalcin info->memory = ((PetscObject)A)->mem; 2676d5f3da31SBarry Smith if (A->factortype) { 26772d61bbb3SSatish Balay info->fill_ratio_given = A->info.fill_ratio_given; 26782d61bbb3SSatish Balay info->fill_ratio_needed = A->info.fill_ratio_needed; 26792d61bbb3SSatish Balay info->factor_mallocs = A->info.factor_mallocs; 26802d61bbb3SSatish Balay } else { 26812d61bbb3SSatish Balay info->fill_ratio_given = 0; 26822d61bbb3SSatish Balay info->fill_ratio_needed = 0; 26832d61bbb3SSatish Balay info->factor_mallocs = 0; 26842d61bbb3SSatish Balay } 26852d61bbb3SSatish Balay PetscFunctionReturn(0); 26862d61bbb3SSatish Balay } 26872d61bbb3SSatish Balay 26882d61bbb3SSatish Balay 26899bfb1392SMichael Lange #if defined(PETSC_THREADCOMM_ACTIVE) 26909bfb1392SMichael Lange PetscErrorCode MatZeroEntries_SeqBAIJ_Kernel(PetscInt thread_id,Mat A) 26919bfb1392SMichael Lange { 26929bfb1392SMichael Lange PetscErrorCode ierr; 26939bfb1392SMichael Lange PetscInt *trstarts=A->rmap->trstarts; 26949bfb1392SMichael Lange PetscInt n,start,end; 26959bfb1392SMichael Lange Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 26969bfb1392SMichael Lange 26973295f6cbSLawrence Mitchell start = trstarts[thread_id]/A->rmap->bs; 26983295f6cbSLawrence Mitchell end = trstarts[thread_id+1]/A->rmap->bs; 26999bfb1392SMichael Lange n = a->i[end] - a->i[start]; 27003295f6cbSLawrence Mitchell ierr = PetscMemzero(a->a+a->bs2*a->i[start],a->bs2*n*sizeof(MatScalar));CHKERRQ(ierr); 27019bfb1392SMichael Lange return 0; 27029bfb1392SMichael Lange } 27039bfb1392SMichael Lange 27049bfb1392SMichael Lange #undef __FUNCT__ 27059bfb1392SMichael Lange #define __FUNCT__ "MatZeroEntries_SeqBAIJ" 27069bfb1392SMichael Lange PetscErrorCode MatZeroEntries_SeqBAIJ(Mat A) 27079bfb1392SMichael Lange { 27089bfb1392SMichael Lange PetscErrorCode ierr; 27099bfb1392SMichael Lange 27109bfb1392SMichael Lange 27119bfb1392SMichael Lange PetscFunctionBegin; 27129bfb1392SMichael Lange ierr = PetscThreadCommRunKernel(PetscObjectComm((PetscObject)A),(PetscThreadKernel)MatZeroEntries_SeqBAIJ_Kernel,1,A);CHKERRQ(ierr); 27139bfb1392SMichael Lange PetscFunctionReturn(0); 27149bfb1392SMichael Lange } 27159bfb1392SMichael Lange #else 27164a2ae208SSatish Balay #undef __FUNCT__ 27174a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqBAIJ" 2718dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqBAIJ(Mat A) 27192d61bbb3SSatish Balay { 27202d61bbb3SSatish Balay Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 2721dfbe8321SBarry Smith PetscErrorCode ierr; 27222d61bbb3SSatish Balay 27232d61bbb3SSatish Balay PetscFunctionBegin; 2724549d3d68SSatish Balay ierr = PetscMemzero(a->a,a->bs2*a->i[a->mbs]*sizeof(MatScalar));CHKERRQ(ierr); 27252d61bbb3SSatish Balay PetscFunctionReturn(0); 27262d61bbb3SSatish Balay } 27279bfb1392SMichael Lange #endif 2728