1d4002b98SHong Zhang 2d4002b98SHong Zhang /* 3d4002b98SHong Zhang Defines the basic matrix operations for the SELL matrix storage format. 4d4002b98SHong Zhang */ 5d4002b98SHong Zhang #include <../src/mat/impls/sell/seq/sell.h> /*I "petscmat.h" I*/ 6d4002b98SHong Zhang #include <petscblaslapack.h> 7d4002b98SHong Zhang #include <petsc/private/kernels/blocktranspose.h> 85f70456aSHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && (defined(__AVX512F__) || (defined(__AVX2__) && defined(__FMA__)) || defined(__AVX__)) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 94243e2ceSHong Zhang 10d4002b98SHong Zhang #include <immintrin.h> 11d4002b98SHong Zhang 12d4002b98SHong Zhang #if !defined(_MM_SCALE_8) 13d4002b98SHong Zhang #define _MM_SCALE_8 8 14d4002b98SHong Zhang #endif 15d4002b98SHong Zhang 16d4002b98SHong Zhang #if defined(__AVX512F__) 17d4002b98SHong Zhang /* these do not work 18d4002b98SHong Zhang vec_idx = _mm512_loadunpackhi_epi32(vec_idx,acolidx); 19d4002b98SHong Zhang vec_vals = _mm512_loadunpackhi_pd(vec_vals,aval); 20d4002b98SHong Zhang */ 21d4002b98SHong Zhang #define AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y) \ 22d4002b98SHong Zhang /* if the mask bit is set, copy from acolidx, otherwise from vec_idx */ \ 23ef588d5cSRichard Tran Mills vec_idx = _mm256_loadu_si256((__m256i const*)acolidx); \ 24ef588d5cSRichard Tran Mills vec_vals = _mm512_loadu_pd(aval); \ 25d4002b98SHong Zhang vec_x = _mm512_i32gather_pd(vec_idx,x,_MM_SCALE_8); \ 26a48a6482SHong Zhang vec_y = _mm512_fmadd_pd(vec_x,vec_vals,vec_y) 275f70456aSHong Zhang #elif defined(__AVX2__) && defined(__FMA__) 28a48a6482SHong Zhang #define AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y) \ 29ef588d5cSRichard Tran Mills vec_vals = _mm256_loadu_pd(aval); \ 30ef588d5cSRichard Tran Mills vec_idx = _mm_loadu_si128((__m128i const*)acolidx); /* SSE2 */ \ 31a48a6482SHong Zhang vec_x = _mm256_i32gather_pd(x,vec_idx,_MM_SCALE_8); \ 32a48a6482SHong Zhang vec_y = _mm256_fmadd_pd(vec_x,vec_vals,vec_y) 33d4002b98SHong Zhang #endif 34d4002b98SHong Zhang #endif /* PETSC_HAVE_IMMINTRIN_H */ 35d4002b98SHong Zhang 36d4002b98SHong Zhang /*@C 37d4002b98SHong Zhang MatSeqSELLSetPreallocation - For good matrix assembly performance 38d4002b98SHong Zhang the user should preallocate the matrix storage by setting the parameter nz 39d4002b98SHong Zhang (or the array nnz). By setting these parameters accurately, performance 40d4002b98SHong Zhang during matrix assembly can be increased significantly. 41d4002b98SHong Zhang 42d083f849SBarry Smith Collective 43d4002b98SHong Zhang 44d4002b98SHong Zhang Input Parameters: 45d4002b98SHong Zhang + B - The matrix 46d4002b98SHong Zhang . nz - number of nonzeros per row (same for all rows) 47d4002b98SHong Zhang - nnz - array containing the number of nonzeros in the various rows 48d4002b98SHong Zhang (possibly different for each row) or NULL 49d4002b98SHong Zhang 50d4002b98SHong Zhang Notes: 51d4002b98SHong Zhang If nnz is given then nz is ignored. 52d4002b98SHong Zhang 53d4002b98SHong Zhang Specify the preallocated storage with either nz or nnz (not both). 54d4002b98SHong Zhang Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 55d4002b98SHong Zhang allocation. For large problems you MUST preallocate memory or you 56d4002b98SHong Zhang will get TERRIBLE performance, see the users' manual chapter on matrices. 57d4002b98SHong Zhang 58d4002b98SHong Zhang You can call MatGetInfo() to get information on how effective the preallocation was; 59d4002b98SHong Zhang for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 60d4002b98SHong Zhang You can also run with the option -info and look for messages with the string 61d4002b98SHong Zhang malloc in them to see if additional memory allocation was needed. 62d4002b98SHong Zhang 63d4002b98SHong Zhang Developers: Use nz of MAT_SKIP_ALLOCATION to not allocate any space for the matrix 64d4002b98SHong Zhang entries or columns indices. 65d4002b98SHong Zhang 66c7ee91abSRichard Tran Mills The maximum number of nonzeos in any row should be as accurate as possible. 67c7ee91abSRichard Tran Mills If it is underestimated, you will get bad performance due to reallocation 68d4002b98SHong Zhang (MatSeqXSELLReallocateSELL). 69d4002b98SHong Zhang 70d4002b98SHong Zhang Level: intermediate 71d4002b98SHong Zhang 72d4002b98SHong Zhang .seealso: MatCreate(), MatCreateSELL(), MatSetValues(), MatGetInfo() 73d4002b98SHong Zhang 74d4002b98SHong Zhang @*/ 75d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation(Mat B,PetscInt rlenmax,const PetscInt rlen[]) 76d4002b98SHong Zhang { 77d4002b98SHong Zhang PetscErrorCode ierr; 78d4002b98SHong Zhang 79d4002b98SHong Zhang PetscFunctionBegin; 80d4002b98SHong Zhang PetscValidHeaderSpecific(B,MAT_CLASSID,1); 81d4002b98SHong Zhang PetscValidType(B,1); 82d4002b98SHong Zhang ierr = PetscTryMethod(B,"MatSeqSELLSetPreallocation_C",(Mat,PetscInt,const PetscInt[]),(B,rlenmax,rlen));CHKERRQ(ierr); 83d4002b98SHong Zhang PetscFunctionReturn(0); 84d4002b98SHong Zhang } 85d4002b98SHong Zhang 86d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation_SeqSELL(Mat B,PetscInt maxallocrow,const PetscInt rlen[]) 87d4002b98SHong Zhang { 88d4002b98SHong Zhang Mat_SeqSELL *b; 89d4002b98SHong Zhang PetscInt i,j,totalslices; 90d4002b98SHong Zhang PetscBool skipallocation=PETSC_FALSE,realalloc=PETSC_FALSE; 91d4002b98SHong Zhang PetscErrorCode ierr; 92d4002b98SHong Zhang 93d4002b98SHong Zhang PetscFunctionBegin; 94d4002b98SHong Zhang if (maxallocrow >= 0 || rlen) realalloc = PETSC_TRUE; 95d4002b98SHong Zhang if (maxallocrow == MAT_SKIP_ALLOCATION) { 96d4002b98SHong Zhang skipallocation = PETSC_TRUE; 97d4002b98SHong Zhang maxallocrow = 0; 98d4002b98SHong Zhang } 99d4002b98SHong Zhang 100d4002b98SHong Zhang ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 101d4002b98SHong Zhang ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 102d4002b98SHong Zhang 103d4002b98SHong Zhang /* FIXME: if one preallocates more space than needed, the matrix does not shrink automatically, but for best performance it should */ 104d4002b98SHong Zhang if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 5; 105*c0aa6a63SJacob Faibussowitsch if (maxallocrow < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"maxallocrow cannot be less than 0: value %" PetscInt_FMT,maxallocrow); 106d4002b98SHong Zhang if (rlen) { 107d4002b98SHong Zhang for (i=0; i<B->rmap->n; i++) { 108*c0aa6a63SJacob Faibussowitsch if (rlen[i] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"rlen cannot be less than 0: local row %" PetscInt_FMT " value %" PetscInt_FMT,i,rlen[i]); 109*c0aa6a63SJacob Faibussowitsch if (rlen[i] > B->cmap->n) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"rlen cannot be greater than row length: local row %" PetscInt_FMT " value %" PetscInt_FMT " rowlength %" PetscInt_FMT,i,rlen[i],B->cmap->n); 110d4002b98SHong Zhang } 111d4002b98SHong Zhang } 112d4002b98SHong Zhang 113d4002b98SHong Zhang B->preallocated = PETSC_TRUE; 114d4002b98SHong Zhang 115d4002b98SHong Zhang b = (Mat_SeqSELL*)B->data; 116d4002b98SHong Zhang 117d4002b98SHong Zhang totalslices = B->rmap->n/8+((B->rmap->n & 0x07)?1:0); /* ceil(n/8) */ 118d4002b98SHong Zhang b->totalslices = totalslices; 119d4002b98SHong Zhang if (!skipallocation) { 120*c0aa6a63SJacob Faibussowitsch if (B->rmap->n & 0x07) {ierr = PetscInfo1(B,"Padding rows to the SEQSELL matrix because the number of rows is not the multiple of 8 (value %" PetscInt_FMT ")\n",B->rmap->n);CHKERRQ(ierr);} 121d4002b98SHong Zhang 122d4002b98SHong Zhang if (!b->sliidx) { /* sliidx gives the starting index of each slice, the last element is the total space allocated */ 123d4002b98SHong Zhang ierr = PetscMalloc1(totalslices+1,&b->sliidx);CHKERRQ(ierr); 124d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)B,(totalslices+1)*sizeof(PetscInt));CHKERRQ(ierr); 125d4002b98SHong Zhang } 126d4002b98SHong Zhang if (!rlen) { /* if rlen is not provided, allocate same space for all the slices */ 127d4002b98SHong Zhang if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 10; 128d4002b98SHong Zhang else if (maxallocrow < 0) maxallocrow = 1; 129d4002b98SHong Zhang for (i=0; i<=totalslices; i++) b->sliidx[i] = i*8*maxallocrow; 130d4002b98SHong Zhang } else { 131d4002b98SHong Zhang maxallocrow = 0; 132d4002b98SHong Zhang b->sliidx[0] = 0; 133d4002b98SHong Zhang for (i=1; i<totalslices; i++) { 134d4002b98SHong Zhang b->sliidx[i] = 0; 135d4002b98SHong Zhang for (j=0;j<8;j++) { 136d4002b98SHong Zhang b->sliidx[i] = PetscMax(b->sliidx[i],rlen[8*(i-1)+j]); 137d4002b98SHong Zhang } 138d4002b98SHong Zhang maxallocrow = PetscMax(b->sliidx[i],maxallocrow); 139c73702f5SBarry Smith ierr = PetscIntSumError(b->sliidx[i-1],8*b->sliidx[i],&b->sliidx[i]);CHKERRQ(ierr); 140d4002b98SHong Zhang } 141d4002b98SHong Zhang /* last slice */ 142d4002b98SHong Zhang b->sliidx[totalslices] = 0; 143d4002b98SHong Zhang for (j=(totalslices-1)*8;j<B->rmap->n;j++) b->sliidx[totalslices] = PetscMax(b->sliidx[totalslices],rlen[j]); 144d4002b98SHong Zhang maxallocrow = PetscMax(b->sliidx[totalslices],maxallocrow); 145d4002b98SHong Zhang b->sliidx[totalslices] = b->sliidx[totalslices-1] + 8*b->sliidx[totalslices]; 146d4002b98SHong Zhang } 147d4002b98SHong Zhang 148d4002b98SHong Zhang /* allocate space for val, colidx, rlen */ 149d4002b98SHong Zhang /* FIXME: should B's old memory be unlogged? */ 150d4002b98SHong Zhang ierr = MatSeqXSELLFreeSELL(B,&b->val,&b->colidx);CHKERRQ(ierr); 151d4002b98SHong Zhang /* FIXME: assuming an element of the bit array takes 8 bits */ 152d4002b98SHong Zhang ierr = PetscMalloc2(b->sliidx[totalslices],&b->val,b->sliidx[totalslices],&b->colidx);CHKERRQ(ierr); 153d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)B,b->sliidx[totalslices]*(sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 154d4002b98SHong Zhang /* b->rlen will count nonzeros in each row so far. We dont copy rlen to b->rlen because the matrix has not been set. */ 155d4002b98SHong Zhang ierr = PetscCalloc1(8*totalslices,&b->rlen);CHKERRQ(ierr); 156d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)B,8*totalslices*sizeof(PetscInt));CHKERRQ(ierr); 157d4002b98SHong Zhang 158d4002b98SHong Zhang b->singlemalloc = PETSC_TRUE; 159d4002b98SHong Zhang b->free_val = PETSC_TRUE; 160d4002b98SHong Zhang b->free_colidx = PETSC_TRUE; 161d4002b98SHong Zhang } else { 162d4002b98SHong Zhang b->free_val = PETSC_FALSE; 163d4002b98SHong Zhang b->free_colidx = PETSC_FALSE; 164d4002b98SHong Zhang } 165d4002b98SHong Zhang 166d4002b98SHong Zhang b->nz = 0; 167d4002b98SHong Zhang b->maxallocrow = maxallocrow; 168d4002b98SHong Zhang b->rlenmax = maxallocrow; 169d4002b98SHong Zhang b->maxallocmat = b->sliidx[totalslices]; 170d4002b98SHong Zhang B->info.nz_unneeded = (double)b->maxallocmat; 171d4002b98SHong Zhang if (realalloc) { 172d4002b98SHong Zhang ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 173d4002b98SHong Zhang } 174d4002b98SHong Zhang PetscFunctionReturn(0); 175d4002b98SHong Zhang } 176d4002b98SHong Zhang 1776108893eSStefano Zampini PetscErrorCode MatGetRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 1786108893eSStefano Zampini { 1796108893eSStefano Zampini Mat_SeqSELL *a = (Mat_SeqSELL*)A->data; 1806108893eSStefano Zampini PetscInt shift; 1816108893eSStefano Zampini 1826108893eSStefano Zampini PetscFunctionBegin; 183*c0aa6a63SJacob Faibussowitsch if (row < 0 || row >= A->rmap->n) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %" PetscInt_FMT " out of range",row); 1846108893eSStefano Zampini if (nz) *nz = a->rlen[row]; 1856108893eSStefano Zampini shift = a->sliidx[row>>3]+(row&0x07); 1866108893eSStefano Zampini if (!a->getrowcols) { 1876108893eSStefano Zampini PetscErrorCode ierr; 1886108893eSStefano Zampini 1896108893eSStefano Zampini ierr = PetscMalloc2(a->rlenmax,&a->getrowcols,a->rlenmax,&a->getrowvals);CHKERRQ(ierr); 1906108893eSStefano Zampini } 1916108893eSStefano Zampini if (idx) { 1926108893eSStefano Zampini PetscInt j; 1936108893eSStefano Zampini for (j=0; j<a->rlen[row]; j++) a->getrowcols[j] = a->colidx[shift+8*j]; 1946108893eSStefano Zampini *idx = a->getrowcols; 1956108893eSStefano Zampini } 1966108893eSStefano Zampini if (v) { 1976108893eSStefano Zampini PetscInt j; 1986108893eSStefano Zampini for (j=0; j<a->rlen[row]; j++) a->getrowvals[j] = a->val[shift+8*j]; 1996108893eSStefano Zampini *v = a->getrowvals; 2006108893eSStefano Zampini } 2016108893eSStefano Zampini PetscFunctionReturn(0); 2026108893eSStefano Zampini } 2036108893eSStefano Zampini 2046108893eSStefano Zampini PetscErrorCode MatRestoreRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 2056108893eSStefano Zampini { 2066108893eSStefano Zampini PetscFunctionBegin; 2076108893eSStefano Zampini PetscFunctionReturn(0); 2086108893eSStefano Zampini } 2096108893eSStefano Zampini 210d4002b98SHong Zhang PetscErrorCode MatConvert_SeqSELL_SeqAIJ(Mat A, MatType newtype,MatReuse reuse,Mat *newmat) 211d4002b98SHong Zhang { 212d4002b98SHong Zhang Mat B; 213d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 214e3f1f374SStefano Zampini PetscInt i; 215d4002b98SHong Zhang PetscErrorCode ierr; 216d4002b98SHong Zhang 217d4002b98SHong Zhang PetscFunctionBegin; 218ad013a7bSRichard Tran Mills if (reuse == MAT_REUSE_MATRIX) { 219ad013a7bSRichard Tran Mills B = *newmat; 220e3f1f374SStefano Zampini ierr = MatZeroEntries(B);CHKERRQ(ierr); 221ad013a7bSRichard Tran Mills } else { 222d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 223d4002b98SHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 224d4002b98SHong Zhang ierr = MatSetType(B,MATSEQAIJ);CHKERRQ(ierr); 225d4002b98SHong Zhang ierr = MatSeqAIJSetPreallocation(B,0,a->rlen);CHKERRQ(ierr); 226ad013a7bSRichard Tran Mills } 227d4002b98SHong Zhang 228e3f1f374SStefano Zampini for (i=0; i<A->rmap->n; i++) { 229e108cb99SStefano Zampini PetscInt nz = 0,*cols = NULL; 230e108cb99SStefano Zampini PetscScalar *vals = NULL; 231e3f1f374SStefano Zampini 232e3f1f374SStefano Zampini ierr = MatGetRow_SeqSELL(A,i,&nz,&cols,&vals);CHKERRQ(ierr); 233e3f1f374SStefano Zampini ierr = MatSetValues(B,1,&i,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr); 234e3f1f374SStefano Zampini ierr = MatRestoreRow_SeqSELL(A,i,&nz,&cols,&vals);CHKERRQ(ierr); 235d4002b98SHong Zhang } 236e3f1f374SStefano Zampini 237d4002b98SHong Zhang ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 238d4002b98SHong Zhang ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 239d4002b98SHong Zhang B->rmap->bs = A->rmap->bs; 240d4002b98SHong Zhang 241d4002b98SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 242d4002b98SHong Zhang ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); 243d4002b98SHong Zhang } else { 244d4002b98SHong Zhang *newmat = B; 245d4002b98SHong Zhang } 246d4002b98SHong Zhang PetscFunctionReturn(0); 247d4002b98SHong Zhang } 248d4002b98SHong Zhang 249d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/aij.h> 250d4002b98SHong Zhang 251d4002b98SHong Zhang PetscErrorCode MatConvert_SeqAIJ_SeqSELL(Mat A,MatType newtype,MatReuse reuse,Mat *newmat) 252d4002b98SHong Zhang { 253d4002b98SHong Zhang Mat B; 254d4002b98SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 255d4002b98SHong Zhang PetscInt *ai=a->i,m=A->rmap->N,n=A->cmap->N,i,*rowlengths,row,ncols; 256d4002b98SHong Zhang const PetscInt *cols; 257d4002b98SHong Zhang const PetscScalar *vals; 258d4002b98SHong Zhang PetscErrorCode ierr; 259d4002b98SHong Zhang 260d4002b98SHong Zhang PetscFunctionBegin; 261ad013a7bSRichard Tran Mills 262ad013a7bSRichard Tran Mills if (reuse == MAT_REUSE_MATRIX) { 263ad013a7bSRichard Tran Mills B = *newmat; 264ad013a7bSRichard Tran Mills } else { 265d4002b98SHong Zhang /* Can we just use ilen? */ 266d4002b98SHong Zhang ierr = PetscMalloc1(m,&rowlengths);CHKERRQ(ierr); 267d4002b98SHong Zhang for (i=0; i<m; i++) { 268d4002b98SHong Zhang rowlengths[i] = ai[i+1] - ai[i]; 269d4002b98SHong Zhang } 270d4002b98SHong Zhang 271d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 272d4002b98SHong Zhang ierr = MatSetSizes(B,m,n,m,n);CHKERRQ(ierr); 273d4002b98SHong Zhang ierr = MatSetType(B,MATSEQSELL);CHKERRQ(ierr); 274d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation(B,0,rowlengths);CHKERRQ(ierr); 275d4002b98SHong Zhang ierr = PetscFree(rowlengths);CHKERRQ(ierr); 276ad013a7bSRichard Tran Mills } 277d4002b98SHong Zhang 278d4002b98SHong Zhang for (row=0; row<m; row++) { 279d4002b98SHong Zhang ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 280d4002b98SHong Zhang ierr = MatSetValues(B,1,&row,ncols,cols,vals,INSERT_VALUES);CHKERRQ(ierr); 281d4002b98SHong Zhang ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 282d4002b98SHong Zhang } 283d4002b98SHong Zhang ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 284d4002b98SHong Zhang ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 285d4002b98SHong Zhang B->rmap->bs = A->rmap->bs; 286d4002b98SHong Zhang 287d4002b98SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 288d4002b98SHong Zhang ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); 289d4002b98SHong Zhang } else { 290d4002b98SHong Zhang *newmat = B; 291d4002b98SHong Zhang } 292d4002b98SHong Zhang PetscFunctionReturn(0); 293d4002b98SHong Zhang } 294d4002b98SHong Zhang 295d4002b98SHong Zhang PetscErrorCode MatMult_SeqSELL(Mat A,Vec xx,Vec yy) 296d4002b98SHong Zhang { 297d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 298d4002b98SHong Zhang PetscScalar *y; 299d4002b98SHong Zhang const PetscScalar *x; 300d4002b98SHong Zhang const MatScalar *aval=a->val; 301d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 302d4002b98SHong Zhang const PetscInt *acolidx=a->colidx; 3037285fed1SHong Zhang PetscInt i,j; 304d4002b98SHong Zhang PetscErrorCode ierr; 305d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 306d4002b98SHong Zhang __m512d vec_x,vec_y,vec_vals; 307d4002b98SHong Zhang __m256i vec_idx; 308d4002b98SHong Zhang __mmask8 mask; 309d4002b98SHong Zhang __m512d vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4; 310d4002b98SHong Zhang __m256i vec_idx2,vec_idx3,vec_idx4; 3115f70456aSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX2__) && defined(__FMA__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 312a48a6482SHong Zhang __m128i vec_idx; 313a48a6482SHong Zhang __m256d vec_x,vec_y,vec_y2,vec_vals; 314a48a6482SHong Zhang MatScalar yval; 315a48a6482SHong Zhang PetscInt r,rows_left,row,nnz_in_row; 31621cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 317d4002b98SHong Zhang __m128d vec_x_tmp; 318d4002b98SHong Zhang __m256d vec_x,vec_y,vec_y2,vec_vals; 319d4002b98SHong Zhang MatScalar yval; 320d4002b98SHong Zhang PetscInt r,rows_left,row,nnz_in_row; 321d4002b98SHong Zhang #else 322d4002b98SHong Zhang PetscScalar sum[8]; 323d4002b98SHong Zhang #endif 324d4002b98SHong Zhang 325d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 326d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval) 327d4002b98SHong Zhang #endif 328d4002b98SHong Zhang 329d4002b98SHong Zhang PetscFunctionBegin; 330d4002b98SHong Zhang ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 331d4002b98SHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 332d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 333d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 334d4002b98SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 335d4002b98SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 336d4002b98SHong Zhang 337d4002b98SHong Zhang vec_y = _mm512_setzero_pd(); 338d4002b98SHong Zhang vec_y2 = _mm512_setzero_pd(); 339d4002b98SHong Zhang vec_y3 = _mm512_setzero_pd(); 340d4002b98SHong Zhang vec_y4 = _mm512_setzero_pd(); 341d4002b98SHong Zhang 34238efe8efSHong Zhang j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */ 343d4002b98SHong Zhang switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) { 344d4002b98SHong Zhang case 3: 345d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 346d4002b98SHong Zhang acolidx += 8; aval += 8; 347d4002b98SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 348d4002b98SHong Zhang acolidx += 8; aval += 8; 349d4002b98SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 350d4002b98SHong Zhang acolidx += 8; aval += 8; 351d4002b98SHong Zhang j += 3; 352d4002b98SHong Zhang break; 353d4002b98SHong Zhang case 2: 354d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 355d4002b98SHong Zhang acolidx += 8; aval += 8; 356d4002b98SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 357d4002b98SHong Zhang acolidx += 8; aval += 8; 358d4002b98SHong Zhang j += 2; 359d4002b98SHong Zhang break; 360d4002b98SHong Zhang case 1: 361d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 362d4002b98SHong Zhang acolidx += 8; aval += 8; 363d4002b98SHong Zhang j += 1; 364d4002b98SHong Zhang break; 365d4002b98SHong Zhang } 366d4002b98SHong Zhang #pragma novector 367d4002b98SHong Zhang for (; j<(a->sliidx[i+1]>>3); j+=4) { 368d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 369d4002b98SHong Zhang acolidx += 8; aval += 8; 370d4002b98SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 371d4002b98SHong Zhang acolidx += 8; aval += 8; 372d4002b98SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 373d4002b98SHong Zhang acolidx += 8; aval += 8; 374d4002b98SHong Zhang AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4); 375d4002b98SHong Zhang acolidx += 8; aval += 8; 376d4002b98SHong Zhang } 377d4002b98SHong Zhang 378d4002b98SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y2); 379d4002b98SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y3); 380d4002b98SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y4); 381d4002b98SHong Zhang if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */ 382d4002b98SHong Zhang mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07))); 383ef588d5cSRichard Tran Mills _mm512_mask_storeu_pd(&y[8*i],mask,vec_y); 384d4002b98SHong Zhang } else { 385ef588d5cSRichard Tran Mills _mm512_storeu_pd(&y[8*i],vec_y); 386d4002b98SHong Zhang } 387d4002b98SHong Zhang } 3885f70456aSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX2__) && defined(__FMA__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 389a48a6482SHong Zhang for (i=0; i<totalslices; i++) { /* loop over full slices */ 390a48a6482SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 391a48a6482SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 392a48a6482SHong Zhang 393a48a6482SHong Zhang /* last slice may have padding rows. Don't use vectorization. */ 394a48a6482SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 395a48a6482SHong Zhang rows_left = A->rmap->n - 8*i; 396a48a6482SHong Zhang for (r=0; r<rows_left; ++r) { 397a48a6482SHong Zhang yval = (MatScalar)0; 398a48a6482SHong Zhang row = 8*i + r; 399a48a6482SHong Zhang nnz_in_row = a->rlen[row]; 400a48a6482SHong Zhang for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]]; 401a48a6482SHong Zhang y[row] = yval; 402a48a6482SHong Zhang } 403a48a6482SHong Zhang break; 404a48a6482SHong Zhang } 405a48a6482SHong Zhang 406a48a6482SHong Zhang vec_y = _mm256_setzero_pd(); 407a48a6482SHong Zhang vec_y2 = _mm256_setzero_pd(); 408a48a6482SHong Zhang 409a48a6482SHong Zhang /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */ 410a48a6482SHong Zhang #pragma novector 411a48a6482SHong Zhang #pragma unroll(2) 412a48a6482SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 413a48a6482SHong Zhang AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 414a48a6482SHong Zhang aval += 4; acolidx += 4; 415a48a6482SHong Zhang AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y2); 416a48a6482SHong Zhang aval += 4; acolidx += 4; 417a48a6482SHong Zhang } 418a48a6482SHong Zhang 419ef588d5cSRichard Tran Mills _mm256_storeu_pd(y+i*8,vec_y); 420ef588d5cSRichard Tran Mills _mm256_storeu_pd(y+i*8+4,vec_y2); 421a48a6482SHong Zhang } 42221cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 423d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over full slices */ 424d4002b98SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 425d4002b98SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 426d4002b98SHong Zhang 427d4002b98SHong Zhang vec_y = _mm256_setzero_pd(); 428d4002b98SHong Zhang vec_y2 = _mm256_setzero_pd(); 429d4002b98SHong Zhang 430d4002b98SHong Zhang /* last slice may have padding rows. Don't use vectorization. */ 431d4002b98SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 432d4002b98SHong Zhang rows_left = A->rmap->n - 8*i; 433d4002b98SHong Zhang for (r=0; r<rows_left; ++r) { 434d4002b98SHong Zhang yval = (MatScalar)0; 435d4002b98SHong Zhang row = 8*i + r; 436d4002b98SHong Zhang nnz_in_row = a->rlen[row]; 437d4002b98SHong Zhang for (j=0; j<nnz_in_row; ++j) yval += aval[8*j + r] * x[acolidx[8*j + r]]; 438d4002b98SHong Zhang y[row] = yval; 439d4002b98SHong Zhang } 440d4002b98SHong Zhang break; 441d4002b98SHong Zhang } 442d4002b98SHong Zhang 443d4002b98SHong Zhang /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */ 444a48a6482SHong Zhang #pragma novector 445a48a6482SHong Zhang #pragma unroll(2) 4467285fed1SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 447d4002b98SHong Zhang vec_vals = _mm256_loadu_pd(aval); 448d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 449d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 450d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 451d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 452d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 453d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 454d4002b98SHong Zhang vec_y = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y); 455d4002b98SHong Zhang aval += 4; 456d4002b98SHong Zhang 457d4002b98SHong Zhang vec_vals = _mm256_loadu_pd(aval); 458d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 459d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 460d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 461d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 462d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 463d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 464d4002b98SHong Zhang vec_y2 = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2); 465d4002b98SHong Zhang aval += 4; 466d4002b98SHong Zhang } 467d4002b98SHong Zhang 468d4002b98SHong Zhang _mm256_storeu_pd(y + i*8, vec_y); 469d4002b98SHong Zhang _mm256_storeu_pd(y + i*8 + 4, vec_y2); 470d4002b98SHong Zhang } 471d4002b98SHong Zhang #else 472d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 473d4002b98SHong Zhang for (j=0; j<8; j++) sum[j] = 0.0; 474d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 475d4002b98SHong Zhang sum[0] += aval[j] * x[acolidx[j]]; 476d4002b98SHong Zhang sum[1] += aval[j+1] * x[acolidx[j+1]]; 477d4002b98SHong Zhang sum[2] += aval[j+2] * x[acolidx[j+2]]; 478d4002b98SHong Zhang sum[3] += aval[j+3] * x[acolidx[j+3]]; 479d4002b98SHong Zhang sum[4] += aval[j+4] * x[acolidx[j+4]]; 480d4002b98SHong Zhang sum[5] += aval[j+5] * x[acolidx[j+5]]; 481d4002b98SHong Zhang sum[6] += aval[j+6] * x[acolidx[j+6]]; 482d4002b98SHong Zhang sum[7] += aval[j+7] * x[acolidx[j+7]]; 483d4002b98SHong Zhang } 484d4002b98SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */ 485d4002b98SHong Zhang for (j=0; j<(A->rmap->n & 0x07); j++) y[8*i+j] = sum[j]; 486d4002b98SHong Zhang } else { 4877285fed1SHong Zhang for (j=0; j<8; j++) y[8*i+j] = sum[j]; 488d4002b98SHong Zhang } 489d4002b98SHong Zhang } 490d4002b98SHong Zhang #endif 491d4002b98SHong Zhang 492d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->nz-a->nonzerorowcnt);CHKERRQ(ierr); /* theoretical minimal FLOPs */ 493d4002b98SHong Zhang ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 494d4002b98SHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 495d4002b98SHong Zhang PetscFunctionReturn(0); 496d4002b98SHong Zhang } 497d4002b98SHong Zhang 498d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/ftn-kernels/fmultadd.h> 499d4002b98SHong Zhang PetscErrorCode MatMultAdd_SeqSELL(Mat A,Vec xx,Vec yy,Vec zz) 500d4002b98SHong Zhang { 501d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 502d4002b98SHong Zhang PetscScalar *y,*z; 503d4002b98SHong Zhang const PetscScalar *x; 504d4002b98SHong Zhang const MatScalar *aval=a->val; 505d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 506d4002b98SHong Zhang const PetscInt *acolidx=a->colidx; 507d4002b98SHong Zhang PetscInt i,j; 508d4002b98SHong Zhang PetscErrorCode ierr; 509d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 5107285fed1SHong Zhang __m512d vec_x,vec_y,vec_vals; 511d4002b98SHong Zhang __m256i vec_idx; 512d4002b98SHong Zhang __mmask8 mask; 5137285fed1SHong Zhang __m512d vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4; 5147285fed1SHong Zhang __m256i vec_idx2,vec_idx3,vec_idx4; 51521cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 5167285fed1SHong Zhang __m128d vec_x_tmp; 5177285fed1SHong Zhang __m256d vec_x,vec_y,vec_y2,vec_vals; 5187285fed1SHong Zhang MatScalar yval; 5197285fed1SHong Zhang PetscInt r,row,nnz_in_row; 520d4002b98SHong Zhang #else 521d4002b98SHong Zhang PetscScalar sum[8]; 522d4002b98SHong Zhang #endif 523d4002b98SHong Zhang 524d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 525d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval) 526d4002b98SHong Zhang #endif 527d4002b98SHong Zhang 528d4002b98SHong Zhang PetscFunctionBegin; 529d4002b98SHong Zhang ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 530d4002b98SHong Zhang ierr = VecGetArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 531d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 5327285fed1SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 5337285fed1SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 5347285fed1SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 5357285fed1SHong Zhang 536d4002b98SHong Zhang if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */ 537d4002b98SHong Zhang mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07))); 538ef588d5cSRichard Tran Mills vec_y = _mm512_mask_loadu_pd(vec_y,mask,&y[8*i]); 5397285fed1SHong Zhang } else { 540ef588d5cSRichard Tran Mills vec_y = _mm512_loadu_pd(&y[8*i]); 5417285fed1SHong Zhang } 5427285fed1SHong Zhang vec_y2 = _mm512_setzero_pd(); 5437285fed1SHong Zhang vec_y3 = _mm512_setzero_pd(); 5447285fed1SHong Zhang vec_y4 = _mm512_setzero_pd(); 5457285fed1SHong Zhang 5467285fed1SHong Zhang j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */ 5477285fed1SHong Zhang switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) { 5487285fed1SHong Zhang case 3: 5497285fed1SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 5507285fed1SHong Zhang acolidx += 8; aval += 8; 5517285fed1SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 5527285fed1SHong Zhang acolidx += 8; aval += 8; 5537285fed1SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 5547285fed1SHong Zhang acolidx += 8; aval += 8; 5557285fed1SHong Zhang j += 3; 5567285fed1SHong Zhang break; 5577285fed1SHong Zhang case 2: 5587285fed1SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 5597285fed1SHong Zhang acolidx += 8; aval += 8; 5607285fed1SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 5617285fed1SHong Zhang acolidx += 8; aval += 8; 5627285fed1SHong Zhang j += 2; 5637285fed1SHong Zhang break; 5647285fed1SHong Zhang case 1: 5657285fed1SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 5667285fed1SHong Zhang acolidx += 8; aval += 8; 5677285fed1SHong Zhang j += 1; 5687285fed1SHong Zhang break; 5697285fed1SHong Zhang } 5707285fed1SHong Zhang #pragma novector 5717285fed1SHong Zhang for (; j<(a->sliidx[i+1]>>3); j+=4) { 5727285fed1SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 5737285fed1SHong Zhang acolidx += 8; aval += 8; 5747285fed1SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 5757285fed1SHong Zhang acolidx += 8; aval += 8; 5767285fed1SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 5777285fed1SHong Zhang acolidx += 8; aval += 8; 5787285fed1SHong Zhang AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4); 5797285fed1SHong Zhang acolidx += 8; aval += 8; 5807285fed1SHong Zhang } 5817285fed1SHong Zhang 5827285fed1SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y2); 5837285fed1SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y3); 5847285fed1SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y4); 5857285fed1SHong Zhang if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */ 586ef588d5cSRichard Tran Mills _mm512_mask_storeu_pd(&z[8*i],mask,vec_y); 587d4002b98SHong Zhang } else { 588ef588d5cSRichard Tran Mills _mm512_storeu_pd(&z[8*i],vec_y); 589d4002b98SHong Zhang } 5907285fed1SHong Zhang } 59121cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 5927285fed1SHong Zhang for (i=0; i<totalslices; i++) { /* loop over full slices */ 5937285fed1SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 5947285fed1SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 5957285fed1SHong Zhang 5967285fed1SHong Zhang /* last slice may have padding rows. Don't use vectorization. */ 5977285fed1SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 5987285fed1SHong Zhang for (r=0; r<(A->rmap->n & 0x07); ++r) { 5997285fed1SHong Zhang row = 8*i + r; 6007285fed1SHong Zhang yval = (MatScalar)0.0; 6017285fed1SHong Zhang nnz_in_row = a->rlen[row]; 6027285fed1SHong Zhang for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]]; 6037285fed1SHong Zhang z[row] = y[row] + yval; 6047285fed1SHong Zhang } 6057285fed1SHong Zhang break; 6067285fed1SHong Zhang } 6077285fed1SHong Zhang 6087285fed1SHong Zhang vec_y = _mm256_loadu_pd(y+8*i); 6097285fed1SHong Zhang vec_y2 = _mm256_loadu_pd(y+8*i+4); 6107285fed1SHong Zhang 6117285fed1SHong Zhang /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */ 6127285fed1SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 6137285fed1SHong Zhang vec_vals = _mm256_loadu_pd(aval); 6147285fed1SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 6157285fed1SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 6167285fed1SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 6177285fed1SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 6187285fed1SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 6197285fed1SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 6207285fed1SHong Zhang vec_y = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y); 6217285fed1SHong Zhang aval += 4; 6227285fed1SHong Zhang 6237285fed1SHong Zhang vec_vals = _mm256_loadu_pd(aval); 6247285fed1SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 6257285fed1SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 6267285fed1SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 6277285fed1SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 6287285fed1SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 6297285fed1SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 6307285fed1SHong Zhang vec_y2 = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2); 6317285fed1SHong Zhang aval += 4; 6327285fed1SHong Zhang } 6337285fed1SHong Zhang 6347285fed1SHong Zhang _mm256_storeu_pd(z+i*8,vec_y); 6357285fed1SHong Zhang _mm256_storeu_pd(z+i*8+4,vec_y2); 6367285fed1SHong Zhang } 637d4002b98SHong Zhang #else 6387285fed1SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 6397285fed1SHong Zhang for (j=0; j<8; j++) sum[j] = 0.0; 640d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 641d4002b98SHong Zhang sum[0] += aval[j] * x[acolidx[j]]; 642d4002b98SHong Zhang sum[1] += aval[j+1] * x[acolidx[j+1]]; 643d4002b98SHong Zhang sum[2] += aval[j+2] * x[acolidx[j+2]]; 644d4002b98SHong Zhang sum[3] += aval[j+3] * x[acolidx[j+3]]; 645d4002b98SHong Zhang sum[4] += aval[j+4] * x[acolidx[j+4]]; 646d4002b98SHong Zhang sum[5] += aval[j+5] * x[acolidx[j+5]]; 647d4002b98SHong Zhang sum[6] += aval[j+6] * x[acolidx[j+6]]; 648d4002b98SHong Zhang sum[7] += aval[j+7] * x[acolidx[j+7]]; 649d4002b98SHong Zhang } 6507285fed1SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 6517285fed1SHong Zhang for (j=0; j<(A->rmap->n & 0x07); j++) z[8*i+j] = y[8*i+j] + sum[j]; 652d4002b98SHong Zhang } else { 6537285fed1SHong Zhang for (j=0; j<8; j++) z[8*i+j] = y[8*i+j] + sum[j]; 6547285fed1SHong Zhang } 655d4002b98SHong Zhang } 656d4002b98SHong Zhang #endif 657d4002b98SHong Zhang 658d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 659d4002b98SHong Zhang ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 660d4002b98SHong Zhang ierr = VecRestoreArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 661d4002b98SHong Zhang PetscFunctionReturn(0); 662d4002b98SHong Zhang } 663d4002b98SHong Zhang 664d4002b98SHong Zhang PetscErrorCode MatMultTransposeAdd_SeqSELL(Mat A,Vec xx,Vec zz,Vec yy) 665d4002b98SHong Zhang { 666d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 667d4002b98SHong Zhang PetscScalar *y; 668d4002b98SHong Zhang const PetscScalar *x; 669d4002b98SHong Zhang const MatScalar *aval=a->val; 670d4002b98SHong Zhang const PetscInt *acolidx=a->colidx; 6717285fed1SHong Zhang PetscInt i,j,r,row,nnz_in_row,totalslices=a->totalslices; 672d4002b98SHong Zhang PetscErrorCode ierr; 673d4002b98SHong Zhang 674d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 675d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval) 676d4002b98SHong Zhang #endif 677d4002b98SHong Zhang 678d4002b98SHong Zhang PetscFunctionBegin; 6799fc32365SStefano Zampini if (A->symmetric) { 6809fc32365SStefano Zampini ierr = MatMultAdd_SeqSELL(A,xx,zz,yy);CHKERRQ(ierr); 6819fc32365SStefano Zampini PetscFunctionReturn(0); 6829fc32365SStefano Zampini } 683d4002b98SHong Zhang if (zz != yy) { ierr = VecCopy(zz,yy);CHKERRQ(ierr); } 684d4002b98SHong Zhang ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 685d4002b98SHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 686d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 6877285fed1SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 6887285fed1SHong Zhang for (r=0; r<(A->rmap->n & 0x07); ++r) { 6897285fed1SHong Zhang row = 8*i + r; 6907285fed1SHong Zhang nnz_in_row = a->rlen[row]; 6917285fed1SHong Zhang for (j=0; j<nnz_in_row; ++j) y[acolidx[8*j+r]] += aval[8*j+r] * x[row]; 6927285fed1SHong Zhang } 6937285fed1SHong Zhang break; 6947285fed1SHong Zhang } 6957285fed1SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 6967285fed1SHong Zhang y[acolidx[j]] += aval[j] * x[8*i]; 6977285fed1SHong Zhang y[acolidx[j+1]] += aval[j+1] * x[8*i+1]; 6987285fed1SHong Zhang y[acolidx[j+2]] += aval[j+2] * x[8*i+2]; 6997285fed1SHong Zhang y[acolidx[j+3]] += aval[j+3] * x[8*i+3]; 7007285fed1SHong Zhang y[acolidx[j+4]] += aval[j+4] * x[8*i+4]; 7017285fed1SHong Zhang y[acolidx[j+5]] += aval[j+5] * x[8*i+5]; 7027285fed1SHong Zhang y[acolidx[j+6]] += aval[j+6] * x[8*i+6]; 7037285fed1SHong Zhang y[acolidx[j+7]] += aval[j+7] * x[8*i+7]; 704d4002b98SHong Zhang } 705d4002b98SHong Zhang } 706d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->sliidx[a->totalslices]);CHKERRQ(ierr); 707d4002b98SHong Zhang ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 708d4002b98SHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 709d4002b98SHong Zhang PetscFunctionReturn(0); 710d4002b98SHong Zhang } 711d4002b98SHong Zhang 712d4002b98SHong Zhang PetscErrorCode MatMultTranspose_SeqSELL(Mat A,Vec xx,Vec yy) 713d4002b98SHong Zhang { 714d4002b98SHong Zhang PetscErrorCode ierr; 715d4002b98SHong Zhang 716d4002b98SHong Zhang PetscFunctionBegin; 7179fc32365SStefano Zampini if (A->symmetric) { 7189fc32365SStefano Zampini ierr = MatMult_SeqSELL(A,xx,yy);CHKERRQ(ierr); 7199fc32365SStefano Zampini } else { 720d4002b98SHong Zhang ierr = VecSet(yy,0.0);CHKERRQ(ierr); 721d4002b98SHong Zhang ierr = MatMultTransposeAdd_SeqSELL(A,xx,yy,yy);CHKERRQ(ierr); 7229fc32365SStefano Zampini } 723d4002b98SHong Zhang PetscFunctionReturn(0); 724d4002b98SHong Zhang } 725d4002b98SHong Zhang 726d4002b98SHong Zhang /* 727d4002b98SHong Zhang Checks for missing diagonals 728d4002b98SHong Zhang */ 729d4002b98SHong Zhang PetscErrorCode MatMissingDiagonal_SeqSELL(Mat A,PetscBool *missing,PetscInt *d) 730d4002b98SHong Zhang { 731d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 732d4002b98SHong Zhang PetscInt *diag,i; 733994fe344SLisandro Dalcin PetscErrorCode ierr; 734d4002b98SHong Zhang 735d4002b98SHong Zhang PetscFunctionBegin; 736d4002b98SHong Zhang *missing = PETSC_FALSE; 737d4002b98SHong Zhang if (A->rmap->n > 0 && !(a->colidx)) { 738d4002b98SHong Zhang *missing = PETSC_TRUE; 739d4002b98SHong Zhang if (d) *d = 0; 740994fe344SLisandro Dalcin ierr = PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n");CHKERRQ(ierr); 741d4002b98SHong Zhang } else { 742d4002b98SHong Zhang diag = a->diag; 743d4002b98SHong Zhang for (i=0; i<A->rmap->n; i++) { 744d4002b98SHong Zhang if (diag[i] == -1) { 745d4002b98SHong Zhang *missing = PETSC_TRUE; 746d4002b98SHong Zhang if (d) *d = i; 747*c0aa6a63SJacob Faibussowitsch ierr = PetscInfo1(A,"Matrix is missing diagonal number %" PetscInt_FMT "\n",i);CHKERRQ(ierr); 748d4002b98SHong Zhang break; 749d4002b98SHong Zhang } 750d4002b98SHong Zhang } 751d4002b98SHong Zhang } 752d4002b98SHong Zhang PetscFunctionReturn(0); 753d4002b98SHong Zhang } 754d4002b98SHong Zhang 755d4002b98SHong Zhang PetscErrorCode MatMarkDiagonal_SeqSELL(Mat A) 756d4002b98SHong Zhang { 757d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 758d4002b98SHong Zhang PetscInt i,j,m=A->rmap->n,shift; 759d4002b98SHong Zhang PetscErrorCode ierr; 760d4002b98SHong Zhang 761d4002b98SHong Zhang PetscFunctionBegin; 762d4002b98SHong Zhang if (!a->diag) { 763d4002b98SHong Zhang ierr = PetscMalloc1(m,&a->diag);CHKERRQ(ierr); 764d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)A,m*sizeof(PetscInt));CHKERRQ(ierr); 765d4002b98SHong Zhang a->free_diag = PETSC_TRUE; 766d4002b98SHong Zhang } 767d4002b98SHong Zhang for (i=0; i<m; i++) { /* loop over rows */ 768d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 769d4002b98SHong Zhang a->diag[i] = -1; 770d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 771d4002b98SHong Zhang if (a->colidx[shift+j*8] == i) { 772d4002b98SHong Zhang a->diag[i] = shift+j*8; 773d4002b98SHong Zhang break; 774d4002b98SHong Zhang } 775d4002b98SHong Zhang } 776d4002b98SHong Zhang } 777d4002b98SHong Zhang PetscFunctionReturn(0); 778d4002b98SHong Zhang } 779d4002b98SHong Zhang 780d4002b98SHong Zhang /* 781d4002b98SHong Zhang Negative shift indicates do not generate an error if there is a zero diagonal, just invert it anyways 782d4002b98SHong Zhang */ 783d4002b98SHong Zhang PetscErrorCode MatInvertDiagonal_SeqSELL(Mat A,PetscScalar omega,PetscScalar fshift) 784d4002b98SHong Zhang { 785d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*) A->data; 786d4002b98SHong Zhang PetscInt i,*diag,m = A->rmap->n; 787d4002b98SHong Zhang MatScalar *val = a->val; 788d4002b98SHong Zhang PetscScalar *idiag,*mdiag; 789d4002b98SHong Zhang PetscErrorCode ierr; 790d4002b98SHong Zhang 791d4002b98SHong Zhang PetscFunctionBegin; 792d4002b98SHong Zhang if (a->idiagvalid) PetscFunctionReturn(0); 793d4002b98SHong Zhang ierr = MatMarkDiagonal_SeqSELL(A);CHKERRQ(ierr); 794d4002b98SHong Zhang diag = a->diag; 795d4002b98SHong Zhang if (!a->idiag) { 796d4002b98SHong Zhang ierr = PetscMalloc3(m,&a->idiag,m,&a->mdiag,m,&a->ssor_work);CHKERRQ(ierr); 797d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)A, 3*m*sizeof(PetscScalar));CHKERRQ(ierr); 798d4002b98SHong Zhang val = a->val; 799d4002b98SHong Zhang } 800d4002b98SHong Zhang mdiag = a->mdiag; 801d4002b98SHong Zhang idiag = a->idiag; 802d4002b98SHong Zhang 803d4002b98SHong Zhang if (omega == 1.0 && PetscRealPart(fshift) <= 0.0) { 804d4002b98SHong Zhang for (i=0; i<m; i++) { 805d4002b98SHong Zhang mdiag[i] = val[diag[i]]; 806d4002b98SHong Zhang if (!PetscAbsScalar(mdiag[i])) { /* zero diagonal */ 807d4002b98SHong Zhang if (PetscRealPart(fshift)) { 808*c0aa6a63SJacob Faibussowitsch ierr = PetscInfo1(A,"Zero diagonal on row %" PetscInt_FMT "\n",i);CHKERRQ(ierr); 809d4002b98SHong Zhang A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 810d4002b98SHong Zhang A->factorerror_zeropivot_value = 0.0; 811d4002b98SHong Zhang A->factorerror_zeropivot_row = i; 812*c0aa6a63SJacob Faibussowitsch } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Zero diagonal on row %" PetscInt_FMT,i); 813d4002b98SHong Zhang } 814d4002b98SHong Zhang idiag[i] = 1.0/val[diag[i]]; 815d4002b98SHong Zhang } 816d4002b98SHong Zhang ierr = PetscLogFlops(m);CHKERRQ(ierr); 817d4002b98SHong Zhang } else { 818d4002b98SHong Zhang for (i=0; i<m; i++) { 819d4002b98SHong Zhang mdiag[i] = val[diag[i]]; 820d4002b98SHong Zhang idiag[i] = omega/(fshift + val[diag[i]]); 821d4002b98SHong Zhang } 822d4002b98SHong Zhang ierr = PetscLogFlops(2.0*m);CHKERRQ(ierr); 823d4002b98SHong Zhang } 824d4002b98SHong Zhang a->idiagvalid = PETSC_TRUE; 825d4002b98SHong Zhang PetscFunctionReturn(0); 826d4002b98SHong Zhang } 827d4002b98SHong Zhang 828d4002b98SHong Zhang PetscErrorCode MatZeroEntries_SeqSELL(Mat A) 829d4002b98SHong Zhang { 830d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 831d4002b98SHong Zhang PetscErrorCode ierr; 832d4002b98SHong Zhang 833d4002b98SHong Zhang PetscFunctionBegin; 834580bdb30SBarry Smith ierr = PetscArrayzero(a->val,a->sliidx[a->totalslices]);CHKERRQ(ierr); 835d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr); 836d4002b98SHong Zhang PetscFunctionReturn(0); 837d4002b98SHong Zhang } 838d4002b98SHong Zhang 839d4002b98SHong Zhang PetscErrorCode MatDestroy_SeqSELL(Mat A) 840d4002b98SHong Zhang { 841d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 842d4002b98SHong Zhang PetscErrorCode ierr; 843d4002b98SHong Zhang 844d4002b98SHong Zhang PetscFunctionBegin; 845d4002b98SHong Zhang #if defined(PETSC_USE_LOG) 846*c0aa6a63SJacob Faibussowitsch PetscLogObjectState((PetscObject)A,"Rows=%" PetscInt_FMT ", Cols=%" PetscInt_FMT ", NZ=%" PetscInt_FMT,A->rmap->n,A->cmap->n,a->nz); 847d4002b98SHong Zhang #endif 848d4002b98SHong Zhang ierr = MatSeqXSELLFreeSELL(A,&a->val,&a->colidx);CHKERRQ(ierr); 849d4002b98SHong Zhang ierr = ISDestroy(&a->row);CHKERRQ(ierr); 850d4002b98SHong Zhang ierr = ISDestroy(&a->col);CHKERRQ(ierr); 851d4002b98SHong Zhang ierr = PetscFree(a->diag);CHKERRQ(ierr); 852d4002b98SHong Zhang ierr = PetscFree(a->rlen);CHKERRQ(ierr); 853d4002b98SHong Zhang ierr = PetscFree(a->sliidx);CHKERRQ(ierr); 854d4002b98SHong Zhang ierr = PetscFree3(a->idiag,a->mdiag,a->ssor_work);CHKERRQ(ierr); 855d4002b98SHong Zhang ierr = PetscFree(a->solve_work);CHKERRQ(ierr); 856d4002b98SHong Zhang ierr = ISDestroy(&a->icol);CHKERRQ(ierr); 857d4002b98SHong Zhang ierr = PetscFree(a->saved_values);CHKERRQ(ierr); 8586108893eSStefano Zampini ierr = PetscFree2(a->getrowcols,a->getrowvals);CHKERRQ(ierr); 859d4002b98SHong Zhang 860d4002b98SHong Zhang ierr = PetscFree(A->data);CHKERRQ(ierr); 861d4002b98SHong Zhang 862f4259b30SLisandro Dalcin ierr = PetscObjectChangeTypeName((PetscObject)A,NULL);CHKERRQ(ierr); 863d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr); 864d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr); 865d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSELLSetPreallocation_C",NULL);CHKERRQ(ierr); 866d4002b98SHong Zhang PetscFunctionReturn(0); 867d4002b98SHong Zhang } 868d4002b98SHong Zhang 869d4002b98SHong Zhang PetscErrorCode MatSetOption_SeqSELL(Mat A,MatOption op,PetscBool flg) 870d4002b98SHong Zhang { 871d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 872d4002b98SHong Zhang PetscErrorCode ierr; 873d4002b98SHong Zhang 874d4002b98SHong Zhang PetscFunctionBegin; 875d4002b98SHong Zhang switch (op) { 876d4002b98SHong Zhang case MAT_ROW_ORIENTED: 877d4002b98SHong Zhang a->roworiented = flg; 878d4002b98SHong Zhang break; 879d4002b98SHong Zhang case MAT_KEEP_NONZERO_PATTERN: 880d4002b98SHong Zhang a->keepnonzeropattern = flg; 881d4002b98SHong Zhang break; 882d4002b98SHong Zhang case MAT_NEW_NONZERO_LOCATIONS: 883d4002b98SHong Zhang a->nonew = (flg ? 0 : 1); 884d4002b98SHong Zhang break; 885d4002b98SHong Zhang case MAT_NEW_NONZERO_LOCATION_ERR: 886d4002b98SHong Zhang a->nonew = (flg ? -1 : 0); 887d4002b98SHong Zhang break; 888d4002b98SHong Zhang case MAT_NEW_NONZERO_ALLOCATION_ERR: 889d4002b98SHong Zhang a->nonew = (flg ? -2 : 0); 890d4002b98SHong Zhang break; 891d4002b98SHong Zhang case MAT_UNUSED_NONZERO_LOCATION_ERR: 892d4002b98SHong Zhang a->nounused = (flg ? -1 : 0); 893d4002b98SHong Zhang break; 8948c78258cSHong Zhang case MAT_FORCE_DIAGONAL_ENTRIES: 895d4002b98SHong Zhang case MAT_IGNORE_OFF_PROC_ENTRIES: 896d4002b98SHong Zhang case MAT_USE_HASH_TABLE: 897071fcb05SBarry Smith case MAT_SORTED_FULL: 898d4002b98SHong Zhang ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 899d4002b98SHong Zhang break; 900d4002b98SHong Zhang case MAT_SPD: 901d4002b98SHong Zhang case MAT_SYMMETRIC: 902d4002b98SHong Zhang case MAT_STRUCTURALLY_SYMMETRIC: 903d4002b98SHong Zhang case MAT_HERMITIAN: 904d4002b98SHong Zhang case MAT_SYMMETRY_ETERNAL: 905d4002b98SHong Zhang /* These options are handled directly by MatSetOption() */ 906d4002b98SHong Zhang break; 907d4002b98SHong Zhang default: 908d4002b98SHong Zhang SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 909d4002b98SHong Zhang } 910d4002b98SHong Zhang PetscFunctionReturn(0); 911d4002b98SHong Zhang } 912d4002b98SHong Zhang 913d4002b98SHong Zhang PetscErrorCode MatGetDiagonal_SeqSELL(Mat A,Vec v) 914d4002b98SHong Zhang { 915d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 916d4002b98SHong Zhang PetscInt i,j,n,shift; 917d4002b98SHong Zhang PetscScalar *x,zero=0.0; 918d4002b98SHong Zhang PetscErrorCode ierr; 919d4002b98SHong Zhang 920d4002b98SHong Zhang PetscFunctionBegin; 921d4002b98SHong Zhang ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 922d4002b98SHong Zhang if (n != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 923d4002b98SHong Zhang 924d4002b98SHong Zhang if (A->factortype == MAT_FACTOR_ILU || A->factortype == MAT_FACTOR_LU) { 925d4002b98SHong Zhang PetscInt *diag=a->diag; 926d4002b98SHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 927d4002b98SHong Zhang for (i=0; i<n; i++) x[i] = 1.0/a->val[diag[i]]; 928d4002b98SHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 929d4002b98SHong Zhang PetscFunctionReturn(0); 930d4002b98SHong Zhang } 931d4002b98SHong Zhang 932d4002b98SHong Zhang ierr = VecSet(v,zero);CHKERRQ(ierr); 933d4002b98SHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 934d4002b98SHong Zhang for (i=0; i<n; i++) { /* loop over rows */ 935d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 936d4002b98SHong Zhang x[i] = 0; 937d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 938d4002b98SHong Zhang if (a->colidx[shift+j*8] == i) { 939d4002b98SHong Zhang x[i] = a->val[shift+j*8]; 940d4002b98SHong Zhang break; 941d4002b98SHong Zhang } 942d4002b98SHong Zhang } 943d4002b98SHong Zhang } 944d4002b98SHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 945d4002b98SHong Zhang PetscFunctionReturn(0); 946d4002b98SHong Zhang } 947d4002b98SHong Zhang 948d4002b98SHong Zhang PetscErrorCode MatDiagonalScale_SeqSELL(Mat A,Vec ll,Vec rr) 949d4002b98SHong Zhang { 950d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 951d4002b98SHong Zhang const PetscScalar *l,*r; 952d4002b98SHong Zhang PetscInt i,j,m,n,row; 953d4002b98SHong Zhang PetscErrorCode ierr; 954d4002b98SHong Zhang 955d4002b98SHong Zhang PetscFunctionBegin; 956d4002b98SHong Zhang if (ll) { 957d4002b98SHong Zhang /* The local size is used so that VecMPI can be passed to this routine 958d4002b98SHong Zhang by MatDiagonalScale_MPISELL */ 959d4002b98SHong Zhang ierr = VecGetLocalSize(ll,&m);CHKERRQ(ierr); 960d4002b98SHong Zhang if (m != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length"); 961d4002b98SHong Zhang ierr = VecGetArrayRead(ll,&l);CHKERRQ(ierr); 962d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 963dab86139SHong Zhang if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */ 964dab86139SHong Zhang for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 965dab86139SHong Zhang if (row < (A->rmap->n & 0x07)) a->val[j] *= l[8*i+row]; 966dab86139SHong Zhang } 967dab86139SHong Zhang } else { 968d4002b98SHong Zhang for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 969d4002b98SHong Zhang a->val[j] *= l[8*i+row]; 970d4002b98SHong Zhang } 971d4002b98SHong Zhang } 972dab86139SHong Zhang } 973d4002b98SHong Zhang ierr = VecRestoreArrayRead(ll,&l);CHKERRQ(ierr); 974d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 975d4002b98SHong Zhang } 976d4002b98SHong Zhang if (rr) { 977d4002b98SHong Zhang ierr = VecGetLocalSize(rr,&n);CHKERRQ(ierr); 978d4002b98SHong Zhang if (n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length"); 979d4002b98SHong Zhang ierr = VecGetArrayRead(rr,&r);CHKERRQ(ierr); 980d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 981dab86139SHong Zhang if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */ 982dab86139SHong Zhang for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 983dab86139SHong Zhang if (row < (A->rmap->n & 0x07)) a->val[j] *= r[a->colidx[j]]; 984dab86139SHong Zhang } 985dab86139SHong Zhang } else { 986d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j++) { 987d4002b98SHong Zhang a->val[j] *= r[a->colidx[j]]; 988d4002b98SHong Zhang } 989d4002b98SHong Zhang } 990dab86139SHong Zhang } 991d4002b98SHong Zhang ierr = VecRestoreArrayRead(rr,&r);CHKERRQ(ierr); 992d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 993d4002b98SHong Zhang } 994d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr); 995d4002b98SHong Zhang PetscFunctionReturn(0); 996d4002b98SHong Zhang } 997d4002b98SHong Zhang 998d4002b98SHong Zhang extern PetscErrorCode MatSetValues_SeqSELL(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 999d4002b98SHong Zhang 1000d4002b98SHong Zhang PetscErrorCode MatGetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 1001d4002b98SHong Zhang { 1002d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1003d4002b98SHong Zhang PetscInt *cp,i,k,low,high,t,row,col,l; 1004d4002b98SHong Zhang PetscInt shift; 1005d4002b98SHong Zhang MatScalar *vp; 1006d4002b98SHong Zhang 1007d4002b98SHong Zhang PetscFunctionBegin; 100868aafef3SStefano Zampini for (k=0; k<m; k++) { /* loop over requested rows */ 1009d4002b98SHong Zhang row = im[k]; 1010d4002b98SHong Zhang if (row<0) continue; 1011*c0aa6a63SJacob Faibussowitsch if (PetscUnlikelyDebug(row >= A->rmap->n)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->n-1); 1012d4002b98SHong Zhang shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */ 1013d4002b98SHong Zhang cp = a->colidx+shift; /* pointer to the row */ 1014d4002b98SHong Zhang vp = a->val+shift; /* pointer to the row */ 101568aafef3SStefano Zampini for (l=0; l<n; l++) { /* loop over requested columns */ 1016d4002b98SHong Zhang col = in[l]; 1017d4002b98SHong Zhang if (col<0) continue; 1018*c0aa6a63SJacob Faibussowitsch if (PetscUnlikelyDebug(col >= A->cmap->n)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: row %" PetscInt_FMT " max %" PetscInt_FMT,col,A->cmap->n-1); 1019d4002b98SHong Zhang high = a->rlen[row]; low = 0; /* assume unsorted */ 1020d4002b98SHong Zhang while (high-low > 5) { 1021d4002b98SHong Zhang t = (low+high)/2; 1022d4002b98SHong Zhang if (*(cp+t*8) > col) high = t; 1023d4002b98SHong Zhang else low = t; 1024d4002b98SHong Zhang } 1025d4002b98SHong Zhang for (i=low; i<high; i++) { 1026d4002b98SHong Zhang if (*(cp+8*i) > col) break; 1027d4002b98SHong Zhang if (*(cp+8*i) == col) { 1028d4002b98SHong Zhang *v++ = *(vp+8*i); 1029d4002b98SHong Zhang goto finished; 1030d4002b98SHong Zhang } 1031d4002b98SHong Zhang } 1032d4002b98SHong Zhang *v++ = 0.0; 1033d4002b98SHong Zhang finished:; 1034d4002b98SHong Zhang } 1035d4002b98SHong Zhang } 1036d4002b98SHong Zhang PetscFunctionReturn(0); 1037d4002b98SHong Zhang } 1038d4002b98SHong Zhang 1039d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_ASCII(Mat A,PetscViewer viewer) 1040d4002b98SHong Zhang { 1041d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1042d4002b98SHong Zhang PetscInt i,j,m=A->rmap->n,shift; 1043d4002b98SHong Zhang const char *name; 1044d4002b98SHong Zhang PetscViewerFormat format; 1045d4002b98SHong Zhang PetscErrorCode ierr; 1046d4002b98SHong Zhang 1047d4002b98SHong Zhang PetscFunctionBegin; 1048d4002b98SHong Zhang ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 1049d4002b98SHong Zhang if (format == PETSC_VIEWER_ASCII_MATLAB) { 1050d4002b98SHong Zhang PetscInt nofinalvalue = 0; 1051d4002b98SHong Zhang /* 1052d4002b98SHong Zhang if (m && ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != A->cmap->n-1))) { 1053d4002b98SHong Zhang nofinalvalue = 1; 1054d4002b98SHong Zhang } 1055d4002b98SHong Zhang */ 1056d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 1057*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %" PetscInt_FMT " %" PetscInt_FMT " \n",m,A->cmap->n);CHKERRQ(ierr); 1058*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"%% Nonzeros = %" PetscInt_FMT " \n",a->nz);CHKERRQ(ierr); 1059d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1060*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",4);\n",a->nz+nofinalvalue);CHKERRQ(ierr); 1061d4002b98SHong Zhang #else 1062*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",3);\n",a->nz+nofinalvalue);CHKERRQ(ierr); 1063d4002b98SHong Zhang #endif 1064d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"zzz = [\n");CHKERRQ(ierr); 1065d4002b98SHong Zhang 1066d4002b98SHong Zhang for (i=0; i<m; i++) { 1067d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1068d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1069d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1070*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %18.16e %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 1071d4002b98SHong Zhang #else 1072*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)a->val[shift+8*j]);CHKERRQ(ierr); 1073d4002b98SHong Zhang #endif 1074d4002b98SHong Zhang } 1075d4002b98SHong Zhang } 1076d4002b98SHong Zhang /* 1077d4002b98SHong Zhang if (nofinalvalue) { 1078d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1079*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %18.16e %18.16e\n",m,A->cmap->n,0.,0.);CHKERRQ(ierr); 1080d4002b98SHong Zhang #else 1081*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %18.16e\n",m,A->cmap->n,0.0);CHKERRQ(ierr); 1082d4002b98SHong Zhang #endif 1083d4002b98SHong Zhang } 1084d4002b98SHong Zhang */ 1085d4002b98SHong Zhang ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr); 1086d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"];\n %s = spconvert(zzz);\n",name);CHKERRQ(ierr); 1087d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 1088d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO || format == PETSC_VIEWER_ASCII_INFO) { 1089d4002b98SHong Zhang PetscFunctionReturn(0); 1090d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_COMMON) { 1091d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 1092d4002b98SHong Zhang for (i=0; i<m; i++) { 1093*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr); 1094d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1095d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1096d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1097d4002b98SHong Zhang if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) { 1098*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 1099d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) { 1100*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 1101d4002b98SHong Zhang } else if (PetscRealPart(a->val[shift+8*j]) != 0.0) { 1102*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]));CHKERRQ(ierr); 1103d4002b98SHong Zhang } 1104d4002b98SHong Zhang #else 1105*c0aa6a63SJacob Faibussowitsch if (a->val[shift+8*j] != 0.0) {ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]);CHKERRQ(ierr);} 1106d4002b98SHong Zhang #endif 1107d4002b98SHong Zhang } 1108d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 1109d4002b98SHong Zhang } 1110d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 1111d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_DENSE) { 1112d4002b98SHong Zhang PetscInt cnt=0,jcnt; 1113d4002b98SHong Zhang PetscScalar value; 1114d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1115d4002b98SHong Zhang PetscBool realonly=PETSC_TRUE; 1116d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) { 1117d4002b98SHong Zhang if (PetscImaginaryPart(a->val[i]) != 0.0) { 1118d4002b98SHong Zhang realonly = PETSC_FALSE; 1119d4002b98SHong Zhang break; 1120d4002b98SHong Zhang } 1121d4002b98SHong Zhang } 1122d4002b98SHong Zhang #endif 1123d4002b98SHong Zhang 1124d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 1125d4002b98SHong Zhang for (i=0; i<m; i++) { 1126d4002b98SHong Zhang jcnt = 0; 1127d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1128d4002b98SHong Zhang for (j=0; j<A->cmap->n; j++) { 1129d4002b98SHong Zhang if (jcnt < a->rlen[i] && j == a->colidx[shift+8*j]) { 1130d4002b98SHong Zhang value = a->val[cnt++]; 1131d4002b98SHong Zhang jcnt++; 1132d4002b98SHong Zhang } else { 1133d4002b98SHong Zhang value = 0.0; 1134d4002b98SHong Zhang } 1135d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1136d4002b98SHong Zhang if (realonly) { 1137d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)PetscRealPart(value));CHKERRQ(ierr); 1138d4002b98SHong Zhang } else { 1139d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," %7.5e+%7.5e i ",(double)PetscRealPart(value),(double)PetscImaginaryPart(value));CHKERRQ(ierr); 1140d4002b98SHong Zhang } 1141d4002b98SHong Zhang #else 1142d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)value);CHKERRQ(ierr); 1143d4002b98SHong Zhang #endif 1144d4002b98SHong Zhang } 1145d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 1146d4002b98SHong Zhang } 1147d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 1148d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_MATRIXMARKET) { 1149d4002b98SHong Zhang PetscInt fshift=1; 1150d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 1151d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1152d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate complex general\n");CHKERRQ(ierr); 1153d4002b98SHong Zhang #else 1154d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate real general\n");CHKERRQ(ierr); 1155d4002b98SHong Zhang #endif 1156*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %" PetscInt_FMT "\n", m, A->cmap->n, a->nz);CHKERRQ(ierr); 1157d4002b98SHong Zhang for (i=0; i<m; i++) { 1158d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1159d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1160d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1161*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 1162d4002b98SHong Zhang #else 1163*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)a->val[shift+8*j]);CHKERRQ(ierr); 1164d4002b98SHong Zhang #endif 1165d4002b98SHong Zhang } 1166d4002b98SHong Zhang } 1167d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 116868aafef3SStefano Zampini } else if (format == PETSC_VIEWER_NATIVE) { 116968aafef3SStefano Zampini for (i=0; i<a->totalslices; i++) { /* loop over slices */ 117068aafef3SStefano Zampini PetscInt row; 1171*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"slice %" PetscInt_FMT ": %" PetscInt_FMT " %" PetscInt_FMT "\n",i,a->sliidx[i],a->sliidx[i+1]);CHKERRQ(ierr); 117268aafef3SStefano Zampini for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 117368aafef3SStefano Zampini #if defined(PETSC_USE_COMPLEX) 117468aafef3SStefano Zampini if (PetscImaginaryPart(a->val[j]) > 0.0) { 1175*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," %" PetscInt_FMT " %" PetscInt_FMT " %g + %g i\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr); 117668aafef3SStefano Zampini } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 1177*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," %" PetscInt_FMT " %" PetscInt_FMT " %g - %g i\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]),-(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr); 117868aafef3SStefano Zampini } else { 1179*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr); 118068aafef3SStefano Zampini } 118168aafef3SStefano Zampini #else 1182*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)a->val[j]);CHKERRQ(ierr); 118368aafef3SStefano Zampini #endif 118468aafef3SStefano Zampini } 118568aafef3SStefano Zampini } 1186d4002b98SHong Zhang } else { 1187d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 1188d4002b98SHong Zhang if (A->factortype) { 1189d4002b98SHong Zhang for (i=0; i<m; i++) { 1190d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1191*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr); 1192d4002b98SHong Zhang /* L part */ 1193d4002b98SHong Zhang for (j=shift; j<a->diag[i]; j+=8) { 1194d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1195d4002b98SHong Zhang if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0) { 1196*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr); 1197d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0) { 1198*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])));CHKERRQ(ierr); 1199d4002b98SHong Zhang } else { 1200*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr); 1201d4002b98SHong Zhang } 1202d4002b98SHong Zhang #else 1203*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]);CHKERRQ(ierr); 1204d4002b98SHong Zhang #endif 1205d4002b98SHong Zhang } 1206d4002b98SHong Zhang /* diagonal */ 1207d4002b98SHong Zhang j = a->diag[i]; 1208d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1209d4002b98SHong Zhang if (PetscImaginaryPart(a->val[j]) > 0.0) { 1210*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)PetscImaginaryPart(1.0/a->val[j]));CHKERRQ(ierr); 1211d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 1212*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)(-PetscImaginaryPart(1.0/a->val[j])));CHKERRQ(ierr); 1213d4002b98SHong Zhang } else { 1214*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]));CHKERRQ(ierr); 1215d4002b98SHong Zhang } 1216d4002b98SHong Zhang #else 1217*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)(1.0/a->val[j]));CHKERRQ(ierr); 1218d4002b98SHong Zhang #endif 1219d4002b98SHong Zhang 1220d4002b98SHong Zhang /* U part */ 1221d4002b98SHong Zhang for (j=a->diag[i]+1; j<shift+8*a->rlen[i]; j+=8) { 1222d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1223d4002b98SHong Zhang if (PetscImaginaryPart(a->val[j]) > 0.0) { 1224*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr); 1225d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 1226*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])));CHKERRQ(ierr); 1227d4002b98SHong Zhang } else { 1228*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr); 1229d4002b98SHong Zhang } 1230d4002b98SHong Zhang #else 1231*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]);CHKERRQ(ierr); 1232d4002b98SHong Zhang #endif 1233d4002b98SHong Zhang } 1234d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 1235d4002b98SHong Zhang } 1236d4002b98SHong Zhang } else { 1237d4002b98SHong Zhang for (i=0; i<m; i++) { 1238d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1239*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr); 1240d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1241d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1242d4002b98SHong Zhang if (PetscImaginaryPart(a->val[j]) > 0.0) { 1243*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 1244d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 1245*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 1246d4002b98SHong Zhang } else { 1247*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]));CHKERRQ(ierr); 1248d4002b98SHong Zhang } 1249d4002b98SHong Zhang #else 1250*c0aa6a63SJacob Faibussowitsch ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]);CHKERRQ(ierr); 1251d4002b98SHong Zhang #endif 1252d4002b98SHong Zhang } 1253d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 1254d4002b98SHong Zhang } 1255d4002b98SHong Zhang } 1256d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 1257d4002b98SHong Zhang } 1258d4002b98SHong Zhang ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 1259d4002b98SHong Zhang PetscFunctionReturn(0); 1260d4002b98SHong Zhang } 1261d4002b98SHong Zhang 1262d4002b98SHong Zhang #include <petscdraw.h> 1263d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw_Zoom(PetscDraw draw,void *Aa) 1264d4002b98SHong Zhang { 1265d4002b98SHong Zhang Mat A=(Mat)Aa; 1266d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1267d4002b98SHong Zhang PetscInt i,j,m=A->rmap->n,shift; 1268d4002b98SHong Zhang int color; 1269d4002b98SHong Zhang PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 1270d4002b98SHong Zhang PetscViewer viewer; 1271d4002b98SHong Zhang PetscViewerFormat format; 1272d4002b98SHong Zhang PetscErrorCode ierr; 1273d4002b98SHong Zhang 1274d4002b98SHong Zhang PetscFunctionBegin; 1275d4002b98SHong Zhang ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 1276d4002b98SHong Zhang ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 1277d4002b98SHong Zhang ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 1278d4002b98SHong Zhang 1279d4002b98SHong Zhang /* loop over matrix elements drawing boxes */ 1280d4002b98SHong Zhang 1281d4002b98SHong Zhang if (format != PETSC_VIEWER_DRAW_CONTOUR) { 1282d4002b98SHong Zhang ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 1283d4002b98SHong Zhang /* Blue for negative, Cyan for zero and Red for positive */ 1284d4002b98SHong Zhang color = PETSC_DRAW_BLUE; 1285d4002b98SHong Zhang for (i=0; i<m; i++) { 1286d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1287d4002b98SHong Zhang y_l = m - i - 1.0; y_r = y_l + 1.0; 1288d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1289d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0; 1290d4002b98SHong Zhang if (PetscRealPart(a->val[shift+8*j]) >= 0.) continue; 1291d4002b98SHong Zhang ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1292d4002b98SHong Zhang } 1293d4002b98SHong Zhang } 1294d4002b98SHong Zhang color = PETSC_DRAW_CYAN; 1295d4002b98SHong Zhang for (i=0; i<m; i++) { 1296d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1297d4002b98SHong Zhang y_l = m - i - 1.0; y_r = y_l + 1.0; 1298d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1299d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0; 1300d4002b98SHong Zhang if (a->val[shift+8*j] != 0.) continue; 1301d4002b98SHong Zhang ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1302d4002b98SHong Zhang } 1303d4002b98SHong Zhang } 1304d4002b98SHong Zhang color = PETSC_DRAW_RED; 1305d4002b98SHong Zhang for (i=0; i<m; i++) { 1306d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1307d4002b98SHong Zhang y_l = m - i - 1.0; y_r = y_l + 1.0; 1308d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1309d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0; 1310d4002b98SHong Zhang if (PetscRealPart(a->val[shift+8*j]) <= 0.) continue; 1311d4002b98SHong Zhang ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1312d4002b98SHong Zhang } 1313d4002b98SHong Zhang } 1314d4002b98SHong Zhang ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 1315d4002b98SHong Zhang } else { 1316d4002b98SHong Zhang /* use contour shading to indicate magnitude of values */ 1317d4002b98SHong Zhang /* first determine max of all nonzero values */ 1318d4002b98SHong Zhang PetscReal minv=0.0,maxv=0.0; 1319d4002b98SHong Zhang PetscInt count=0; 1320d4002b98SHong Zhang PetscDraw popup; 1321d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) { 1322d4002b98SHong Zhang if (PetscAbsScalar(a->val[i]) > maxv) maxv = PetscAbsScalar(a->val[i]); 1323d4002b98SHong Zhang } 1324d4002b98SHong Zhang if (minv >= maxv) maxv = minv + PETSC_SMALL; 1325d4002b98SHong Zhang ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr); 1326d4002b98SHong Zhang ierr = PetscDrawScalePopup(popup,minv,maxv);CHKERRQ(ierr); 1327d4002b98SHong Zhang 1328d4002b98SHong Zhang ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 1329d4002b98SHong Zhang for (i=0; i<m; i++) { 1330d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1331d4002b98SHong Zhang y_l = m - i - 1.0; 1332d4002b98SHong Zhang y_r = y_l + 1.0; 1333d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1334d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; 1335d4002b98SHong Zhang x_r = x_l + 1.0; 1336d4002b98SHong Zhang color = PetscDrawRealToColor(PetscAbsScalar(a->val[count]),minv,maxv); 1337d4002b98SHong Zhang ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1338d4002b98SHong Zhang count++; 1339d4002b98SHong Zhang } 1340d4002b98SHong Zhang } 1341d4002b98SHong Zhang ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 1342d4002b98SHong Zhang } 1343d4002b98SHong Zhang PetscFunctionReturn(0); 1344d4002b98SHong Zhang } 1345d4002b98SHong Zhang 1346d4002b98SHong Zhang #include <petscdraw.h> 1347d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw(Mat A,PetscViewer viewer) 1348d4002b98SHong Zhang { 1349d4002b98SHong Zhang PetscDraw draw; 1350d4002b98SHong Zhang PetscReal xr,yr,xl,yl,h,w; 1351d4002b98SHong Zhang PetscBool isnull; 1352d4002b98SHong Zhang PetscErrorCode ierr; 1353d4002b98SHong Zhang 1354d4002b98SHong Zhang PetscFunctionBegin; 1355d4002b98SHong Zhang ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 1356d4002b98SHong Zhang ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 1357d4002b98SHong Zhang if (isnull) PetscFunctionReturn(0); 1358d4002b98SHong Zhang 1359d4002b98SHong Zhang xr = A->cmap->n; yr = A->rmap->n; h = yr/10.0; w = xr/10.0; 1360d4002b98SHong Zhang xr += w; yr += h; xl = -w; yl = -h; 1361d4002b98SHong Zhang ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 1362d4002b98SHong Zhang ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 1363d4002b98SHong Zhang ierr = PetscDrawZoom(draw,MatView_SeqSELL_Draw_Zoom,A);CHKERRQ(ierr); 1364d4002b98SHong Zhang ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr); 1365d4002b98SHong Zhang ierr = PetscDrawSave(draw);CHKERRQ(ierr); 1366d4002b98SHong Zhang PetscFunctionReturn(0); 1367d4002b98SHong Zhang } 1368d4002b98SHong Zhang 1369d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL(Mat A,PetscViewer viewer) 1370d4002b98SHong Zhang { 1371d4002b98SHong Zhang PetscBool iascii,isbinary,isdraw; 1372d4002b98SHong Zhang PetscErrorCode ierr; 1373d4002b98SHong Zhang 1374d4002b98SHong Zhang PetscFunctionBegin; 1375d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 1376d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 1377d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 1378d4002b98SHong Zhang if (iascii) { 1379d4002b98SHong Zhang ierr = MatView_SeqSELL_ASCII(A,viewer);CHKERRQ(ierr); 1380d4002b98SHong Zhang } else if (isbinary) { 1381d4002b98SHong Zhang /* ierr = MatView_SeqSELL_Binary(A,viewer);CHKERRQ(ierr); */ 1382d4002b98SHong Zhang } else if (isdraw) { 1383d4002b98SHong Zhang ierr = MatView_SeqSELL_Draw(A,viewer);CHKERRQ(ierr); 1384d4002b98SHong Zhang } 1385d4002b98SHong Zhang PetscFunctionReturn(0); 1386d4002b98SHong Zhang } 1387d4002b98SHong Zhang 1388d4002b98SHong Zhang PetscErrorCode MatAssemblyEnd_SeqSELL(Mat A,MatAssemblyType mode) 1389d4002b98SHong Zhang { 1390d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1391d4002b98SHong Zhang PetscInt i,shift,row_in_slice,row,nrow,*cp,lastcol,j,k; 1392d4002b98SHong Zhang MatScalar *vp; 1393d4002b98SHong Zhang PetscErrorCode ierr; 1394d4002b98SHong Zhang 1395d4002b98SHong Zhang PetscFunctionBegin; 1396d4002b98SHong Zhang if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 1397d4002b98SHong Zhang /* To do: compress out the unused elements */ 1398d4002b98SHong Zhang ierr = MatMarkDiagonal_SeqSELL(A);CHKERRQ(ierr); 1399*c0aa6a63SJacob Faibussowitsch ierr = PetscInfo6(A,"Matrix size: %" PetscInt_FMT " X %" PetscInt_FMT "; storage space: %" PetscInt_FMT " allocated %" PetscInt_FMT " used (%" PetscInt_FMT " nonzeros+%" PetscInt_FMT " paddedzeros)\n",A->rmap->n,A->cmap->n,a->maxallocmat,a->sliidx[a->totalslices],a->nz,a->sliidx[a->totalslices]-a->nz);CHKERRQ(ierr); 1400*c0aa6a63SJacob Faibussowitsch ierr = PetscInfo1(A,"Number of mallocs during MatSetValues() is %" PetscInt_FMT "\n",a->reallocs);CHKERRQ(ierr); 1401*c0aa6a63SJacob Faibussowitsch ierr = PetscInfo1(A,"Maximum nonzeros in any row is %" PetscInt_FMT "\n",a->rlenmax);CHKERRQ(ierr); 1402d4002b98SHong Zhang /* Set unused slots for column indices to last valid column index. Set unused slots for values to zero. This allows for a use of unmasked intrinsics -> higher performance */ 1403d4002b98SHong Zhang for (i=0; i<a->totalslices; ++i) { 1404d4002b98SHong Zhang shift = a->sliidx[i]; /* starting index of the slice */ 1405d4002b98SHong Zhang cp = a->colidx+shift; /* pointer to the column indices of the slice */ 1406d4002b98SHong Zhang vp = a->val+shift; /* pointer to the nonzero values of the slice */ 1407d4002b98SHong Zhang for (row_in_slice=0; row_in_slice<8; ++row_in_slice) { /* loop over rows in the slice */ 1408d4002b98SHong Zhang row = 8*i + row_in_slice; 1409d4002b98SHong Zhang nrow = a->rlen[row]; /* number of nonzeros in row */ 1410d4002b98SHong Zhang /* 1411d4002b98SHong Zhang Search for the nearest nonzero. Normally setting the index to zero may cause extra communication. 1412d4002b98SHong Zhang But if the entire slice are empty, it is fine to use 0 since the index will not be loaded. 1413d4002b98SHong Zhang */ 1414d4002b98SHong Zhang lastcol = 0; 1415d4002b98SHong Zhang if (nrow>0) { /* nonempty row */ 1416d4002b98SHong Zhang lastcol = cp[8*(nrow-1)+row_in_slice]; /* use the index from the last nonzero at current row */ 1417d4002b98SHong Zhang } else if (!row_in_slice) { /* first row of the currect slice is empty */ 1418d4002b98SHong Zhang for (j=1;j<8;j++) { 1419d4002b98SHong Zhang if (a->rlen[8*i+j]) { 1420d4002b98SHong Zhang lastcol = cp[j]; 1421d4002b98SHong Zhang break; 1422d4002b98SHong Zhang } 1423d4002b98SHong Zhang } 1424d4002b98SHong Zhang } else { 1425d4002b98SHong Zhang if (a->sliidx[i+1] != shift) lastcol = cp[row_in_slice-1]; /* use the index from the previous row */ 1426d4002b98SHong Zhang } 1427d4002b98SHong Zhang 1428d4002b98SHong Zhang for (k=nrow; k<(a->sliidx[i+1]-shift)/8; ++k) { 1429d4002b98SHong Zhang cp[8*k+row_in_slice] = lastcol; 1430d4002b98SHong Zhang vp[8*k+row_in_slice] = (MatScalar)0; 1431d4002b98SHong Zhang } 1432d4002b98SHong Zhang } 1433d4002b98SHong Zhang } 1434d4002b98SHong Zhang 1435d4002b98SHong Zhang A->info.mallocs += a->reallocs; 1436d4002b98SHong Zhang a->reallocs = 0; 1437d4002b98SHong Zhang 1438d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr); 1439d4002b98SHong Zhang PetscFunctionReturn(0); 1440d4002b98SHong Zhang } 1441d4002b98SHong Zhang 1442d4002b98SHong Zhang PetscErrorCode MatGetInfo_SeqSELL(Mat A,MatInfoType flag,MatInfo *info) 1443d4002b98SHong Zhang { 1444d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1445d4002b98SHong Zhang 1446d4002b98SHong Zhang PetscFunctionBegin; 1447d4002b98SHong Zhang info->block_size = 1.0; 14483966268fSBarry Smith info->nz_allocated = a->maxallocmat; 14493966268fSBarry Smith info->nz_used = a->sliidx[a->totalslices]; /* include padding zeros */ 14503966268fSBarry Smith info->nz_unneeded = (a->maxallocmat-a->sliidx[a->totalslices]); 14513966268fSBarry Smith info->assemblies = A->num_ass; 14523966268fSBarry Smith info->mallocs = A->info.mallocs; 1453d4002b98SHong Zhang info->memory = ((PetscObject)A)->mem; 1454d4002b98SHong Zhang if (A->factortype) { 1455d4002b98SHong Zhang info->fill_ratio_given = A->info.fill_ratio_given; 1456d4002b98SHong Zhang info->fill_ratio_needed = A->info.fill_ratio_needed; 1457d4002b98SHong Zhang info->factor_mallocs = A->info.factor_mallocs; 1458d4002b98SHong Zhang } else { 1459d4002b98SHong Zhang info->fill_ratio_given = 0; 1460d4002b98SHong Zhang info->fill_ratio_needed = 0; 1461d4002b98SHong Zhang info->factor_mallocs = 0; 1462d4002b98SHong Zhang } 1463d4002b98SHong Zhang PetscFunctionReturn(0); 1464d4002b98SHong Zhang } 1465d4002b98SHong Zhang 1466d4002b98SHong Zhang PetscErrorCode MatSetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 1467d4002b98SHong Zhang { 1468d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1469d4002b98SHong Zhang PetscInt shift,i,k,l,low,high,t,ii,row,col,nrow; 1470d4002b98SHong Zhang PetscInt *cp,nonew=a->nonew,lastcol=-1; 1471d4002b98SHong Zhang MatScalar *vp,value; 1472d4002b98SHong Zhang PetscErrorCode ierr; 1473d4002b98SHong Zhang 1474d4002b98SHong Zhang PetscFunctionBegin; 1475d4002b98SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 1476d4002b98SHong Zhang row = im[k]; 1477d4002b98SHong Zhang if (row < 0) continue; 1478*c0aa6a63SJacob Faibussowitsch if (PetscUnlikelyDebug(row >= A->rmap->n)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->n-1); 1479d4002b98SHong Zhang shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */ 1480d4002b98SHong Zhang cp = a->colidx+shift; /* pointer to the row */ 1481d4002b98SHong Zhang vp = a->val+shift; /* pointer to the row */ 1482d4002b98SHong Zhang nrow = a->rlen[row]; 1483d4002b98SHong Zhang low = 0; 1484d4002b98SHong Zhang high = nrow; 1485d4002b98SHong Zhang 1486d4002b98SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 1487d4002b98SHong Zhang col = in[l]; 1488d4002b98SHong Zhang if (col<0) continue; 1489*c0aa6a63SJacob Faibussowitsch if (PetscUnlikelyDebug(col >= A->cmap->n)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Col too large: row %" PetscInt_FMT " max %" PetscInt_FMT,col,A->cmap->n-1); 1490d4002b98SHong Zhang if (a->roworiented) { 1491d4002b98SHong Zhang value = v[l+k*n]; 1492d4002b98SHong Zhang } else { 1493d4002b98SHong Zhang value = v[k+l*m]; 1494d4002b98SHong Zhang } 1495d4002b98SHong Zhang if ((value == 0.0 && a->ignorezeroentries) && (is == ADD_VALUES)) continue; 1496d4002b98SHong Zhang 1497d4002b98SHong Zhang /* search in this row for the specified colmun, i indicates the column to be set */ 1498d4002b98SHong Zhang if (col <= lastcol) low = 0; 1499d4002b98SHong Zhang else high = nrow; 1500d4002b98SHong Zhang lastcol = col; 1501d4002b98SHong Zhang while (high-low > 5) { 1502d4002b98SHong Zhang t = (low+high)/2; 1503d4002b98SHong Zhang if (*(cp+t*8) > col) high = t; 1504d4002b98SHong Zhang else low = t; 1505d4002b98SHong Zhang } 1506d4002b98SHong Zhang for (i=low; i<high; i++) { 1507d4002b98SHong Zhang if (*(cp+i*8) > col) break; 1508d4002b98SHong Zhang if (*(cp+i*8) == col) { 1509d4002b98SHong Zhang if (is == ADD_VALUES) *(vp+i*8) += value; 1510d4002b98SHong Zhang else *(vp+i*8) = value; 1511d4002b98SHong Zhang low = i + 1; 1512d4002b98SHong Zhang goto noinsert; 1513d4002b98SHong Zhang } 1514d4002b98SHong Zhang } 1515d4002b98SHong Zhang if (value == 0.0 && a->ignorezeroentries) goto noinsert; 1516d4002b98SHong Zhang if (nonew == 1) goto noinsert; 1517*c0aa6a63SJacob Faibussowitsch if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%" PetscInt_FMT ", %" PetscInt_FMT ") in the matrix", row, col); 1518d4002b98SHong Zhang /* If the current row length exceeds the slice width (e.g. nrow==slice_width), allocate a new space, otherwise do nothing */ 1519d4002b98SHong Zhang MatSeqXSELLReallocateSELL(A,A->rmap->n,1,nrow,a->sliidx,row/8,row,col,a->colidx,a->val,cp,vp,nonew,MatScalar); 1520d4002b98SHong Zhang /* add the new nonzero to the high position, shift the remaining elements in current row to the right by one slot */ 1521d4002b98SHong Zhang for (ii=nrow-1; ii>=i; ii--) { 1522d4002b98SHong Zhang *(cp+(ii+1)*8) = *(cp+ii*8); 1523d4002b98SHong Zhang *(vp+(ii+1)*8) = *(vp+ii*8); 1524d4002b98SHong Zhang } 1525d4002b98SHong Zhang a->rlen[row]++; 1526d4002b98SHong Zhang *(cp+i*8) = col; 1527d4002b98SHong Zhang *(vp+i*8) = value; 1528d4002b98SHong Zhang a->nz++; 1529d4002b98SHong Zhang A->nonzerostate++; 1530d4002b98SHong Zhang low = i+1; high++; nrow++; 1531d4002b98SHong Zhang noinsert:; 1532d4002b98SHong Zhang } 1533d4002b98SHong Zhang a->rlen[row] = nrow; 1534d4002b98SHong Zhang } 1535d4002b98SHong Zhang PetscFunctionReturn(0); 1536d4002b98SHong Zhang } 1537d4002b98SHong Zhang 1538d4002b98SHong Zhang PetscErrorCode MatCopy_SeqSELL(Mat A,Mat B,MatStructure str) 1539d4002b98SHong Zhang { 1540d4002b98SHong Zhang PetscErrorCode ierr; 1541d4002b98SHong Zhang 1542d4002b98SHong Zhang PetscFunctionBegin; 1543d4002b98SHong Zhang /* If the two matrices have the same copy implementation, use fast copy. */ 1544d4002b98SHong Zhang if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 1545d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1546d4002b98SHong Zhang Mat_SeqSELL *b=(Mat_SeqSELL*)B->data; 1547d4002b98SHong Zhang 1548d4002b98SHong Zhang if (a->sliidx[a->totalslices] != b->sliidx[b->totalslices]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different"); 1549580bdb30SBarry Smith ierr = PetscArraycpy(b->val,a->val,a->sliidx[a->totalslices]);CHKERRQ(ierr); 1550d4002b98SHong Zhang } else { 1551d4002b98SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1552d4002b98SHong Zhang } 1553d4002b98SHong Zhang PetscFunctionReturn(0); 1554d4002b98SHong Zhang } 1555d4002b98SHong Zhang 1556d4002b98SHong Zhang PetscErrorCode MatSetUp_SeqSELL(Mat A) 1557d4002b98SHong Zhang { 1558d4002b98SHong Zhang PetscErrorCode ierr; 1559d4002b98SHong Zhang 1560d4002b98SHong Zhang PetscFunctionBegin; 1561f4259b30SLisandro Dalcin ierr = MatSeqSELLSetPreallocation(A,PETSC_DEFAULT,NULL);CHKERRQ(ierr); 1562d4002b98SHong Zhang PetscFunctionReturn(0); 1563d4002b98SHong Zhang } 1564d4002b98SHong Zhang 1565d4002b98SHong Zhang PetscErrorCode MatSeqSELLGetArray_SeqSELL(Mat A,PetscScalar *array[]) 1566d4002b98SHong Zhang { 1567d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1568d4002b98SHong Zhang 1569d4002b98SHong Zhang PetscFunctionBegin; 1570d4002b98SHong Zhang *array = a->val; 1571d4002b98SHong Zhang PetscFunctionReturn(0); 1572d4002b98SHong Zhang } 1573d4002b98SHong Zhang 1574d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray_SeqSELL(Mat A,PetscScalar *array[]) 1575d4002b98SHong Zhang { 1576d4002b98SHong Zhang PetscFunctionBegin; 1577d4002b98SHong Zhang PetscFunctionReturn(0); 1578d4002b98SHong Zhang } 1579d4002b98SHong Zhang 1580d4002b98SHong Zhang PetscErrorCode MatRealPart_SeqSELL(Mat A) 1581d4002b98SHong Zhang { 1582d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1583d4002b98SHong Zhang PetscInt i; 1584d4002b98SHong Zhang MatScalar *aval=a->val; 1585d4002b98SHong Zhang 1586d4002b98SHong Zhang PetscFunctionBegin; 1587d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i]=PetscRealPart(aval[i]); 1588d4002b98SHong Zhang PetscFunctionReturn(0); 1589d4002b98SHong Zhang } 1590d4002b98SHong Zhang 1591d4002b98SHong Zhang PetscErrorCode MatImaginaryPart_SeqSELL(Mat A) 1592d4002b98SHong Zhang { 1593d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1594d4002b98SHong Zhang PetscInt i; 1595d4002b98SHong Zhang MatScalar *aval=a->val; 1596d4002b98SHong Zhang PetscErrorCode ierr; 1597d4002b98SHong Zhang 1598d4002b98SHong Zhang PetscFunctionBegin; 1599d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i] = PetscImaginaryPart(aval[i]); 1600d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr); 1601d4002b98SHong Zhang PetscFunctionReturn(0); 1602d4002b98SHong Zhang } 1603d4002b98SHong Zhang 1604d4002b98SHong Zhang PetscErrorCode MatScale_SeqSELL(Mat inA,PetscScalar alpha) 1605d4002b98SHong Zhang { 1606d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)inA->data; 1607d4002b98SHong Zhang MatScalar *aval=a->val; 1608d4002b98SHong Zhang PetscScalar oalpha=alpha; 1609d4002b98SHong Zhang PetscBLASInt one=1,size; 1610d4002b98SHong Zhang PetscErrorCode ierr; 1611d4002b98SHong Zhang 1612d4002b98SHong Zhang PetscFunctionBegin; 1613d4002b98SHong Zhang ierr = PetscBLASIntCast(a->sliidx[a->totalslices],&size);CHKERRQ(ierr); 1614d4002b98SHong Zhang PetscStackCallBLAS("BLASscal",BLASscal_(&size,&oalpha,aval,&one)); 1615d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 1616d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(inA);CHKERRQ(ierr); 1617d4002b98SHong Zhang PetscFunctionReturn(0); 1618d4002b98SHong Zhang } 1619d4002b98SHong Zhang 1620d4002b98SHong Zhang PetscErrorCode MatShift_SeqSELL(Mat Y,PetscScalar a) 1621d4002b98SHong Zhang { 1622d4002b98SHong Zhang Mat_SeqSELL *y=(Mat_SeqSELL*)Y->data; 1623d4002b98SHong Zhang PetscErrorCode ierr; 1624d4002b98SHong Zhang 1625d4002b98SHong Zhang PetscFunctionBegin; 1626d4002b98SHong Zhang if (!Y->preallocated || !y->nz) { 1627d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation(Y,1,NULL);CHKERRQ(ierr); 1628d4002b98SHong Zhang } 1629d4002b98SHong Zhang ierr = MatShift_Basic(Y,a);CHKERRQ(ierr); 1630d4002b98SHong Zhang PetscFunctionReturn(0); 1631d4002b98SHong Zhang } 1632d4002b98SHong Zhang 1633d4002b98SHong Zhang PetscErrorCode MatSOR_SeqSELL(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 1634d4002b98SHong Zhang { 1635d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1636d4002b98SHong Zhang PetscScalar *x,sum,*t; 1637f4259b30SLisandro Dalcin const MatScalar *idiag=NULL,*mdiag; 1638d4002b98SHong Zhang const PetscScalar *b,*xb; 1639d4002b98SHong Zhang PetscInt n,m=A->rmap->n,i,j,shift; 1640d4002b98SHong Zhang const PetscInt *diag; 1641d4002b98SHong Zhang PetscErrorCode ierr; 1642d4002b98SHong Zhang 1643d4002b98SHong Zhang PetscFunctionBegin; 1644d4002b98SHong Zhang its = its*lits; 1645d4002b98SHong Zhang 1646d4002b98SHong Zhang if (fshift != a->fshift || omega != a->omega) a->idiagvalid = PETSC_FALSE; /* must recompute idiag[] */ 1647d4002b98SHong Zhang if (!a->idiagvalid) {ierr = MatInvertDiagonal_SeqSELL(A,omega,fshift);CHKERRQ(ierr);} 1648d4002b98SHong Zhang a->fshift = fshift; 1649d4002b98SHong Zhang a->omega = omega; 1650d4002b98SHong Zhang 1651d4002b98SHong Zhang diag = a->diag; 1652d4002b98SHong Zhang t = a->ssor_work; 1653d4002b98SHong Zhang idiag = a->idiag; 1654d4002b98SHong Zhang mdiag = a->mdiag; 1655d4002b98SHong Zhang 1656d4002b98SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1657d4002b98SHong Zhang ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1658d4002b98SHong Zhang /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */ 1659d4002b98SHong Zhang if (flag == SOR_APPLY_UPPER) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_UPPER is not implemented"); 1660d4002b98SHong Zhang if (flag == SOR_APPLY_LOWER) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_LOWER is not implemented"); 1661d4002b98SHong Zhang if (flag & SOR_EISENSTAT) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support yet for Eisenstat"); 1662d4002b98SHong Zhang 1663d4002b98SHong Zhang if (flag & SOR_ZERO_INITIAL_GUESS) { 1664d4002b98SHong Zhang if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) { 1665d4002b98SHong Zhang for (i=0; i<m; i++) { 1666d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1667d4002b98SHong Zhang sum = b[i]; 1668d4002b98SHong Zhang n = (diag[i]-shift)/8; 1669d4002b98SHong Zhang for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]]; 1670d4002b98SHong Zhang t[i] = sum; 1671d4002b98SHong Zhang x[i] = sum*idiag[i]; 1672d4002b98SHong Zhang } 1673d4002b98SHong Zhang xb = t; 1674d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 1675d4002b98SHong Zhang } else xb = b; 1676d4002b98SHong Zhang if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) { 1677d4002b98SHong Zhang for (i=m-1; i>=0; i--) { 1678d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1679d4002b98SHong Zhang sum = xb[i]; 1680d4002b98SHong Zhang n = a->rlen[i]-(diag[i]-shift)/8-1; 1681d4002b98SHong Zhang for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]]; 1682d4002b98SHong Zhang if (xb == b) { 1683d4002b98SHong Zhang x[i] = sum*idiag[i]; 1684d4002b98SHong Zhang } else { 1685d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+sum*idiag[i]; /* omega in idiag */ 1686d4002b98SHong Zhang } 1687d4002b98SHong Zhang } 1688d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */ 1689d4002b98SHong Zhang } 1690d4002b98SHong Zhang its--; 1691d4002b98SHong Zhang } 1692d4002b98SHong Zhang while (its--) { 1693d4002b98SHong Zhang if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) { 1694d4002b98SHong Zhang for (i=0; i<m; i++) { 1695d4002b98SHong Zhang /* lower */ 1696d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1697d4002b98SHong Zhang sum = b[i]; 1698d4002b98SHong Zhang n = (diag[i]-shift)/8; 1699d4002b98SHong Zhang for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]]; 1700d4002b98SHong Zhang t[i] = sum; /* save application of the lower-triangular part */ 1701d4002b98SHong Zhang /* upper */ 1702d4002b98SHong Zhang n = a->rlen[i]-(diag[i]-shift)/8-1; 1703d4002b98SHong Zhang for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]]; 1704d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+sum*idiag[i]; /* omega in idiag */ 1705d4002b98SHong Zhang } 1706d4002b98SHong Zhang xb = t; 1707d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1708d4002b98SHong Zhang } else xb = b; 1709d4002b98SHong Zhang if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) { 1710d4002b98SHong Zhang for (i=m-1; i>=0; i--) { 1711d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1712d4002b98SHong Zhang sum = xb[i]; 1713d4002b98SHong Zhang if (xb == b) { 1714d4002b98SHong Zhang /* whole matrix (no checkpointing available) */ 1715d4002b98SHong Zhang n = a->rlen[i]; 1716d4002b98SHong Zhang for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]]; 1717d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+(sum+mdiag[i]*x[i])*idiag[i]; 1718d4002b98SHong Zhang } else { /* lower-triangular part has been saved, so only apply upper-triangular */ 1719d4002b98SHong Zhang n = a->rlen[i]-(diag[i]-shift)/8-1; 1720d4002b98SHong Zhang for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]]; 1721d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+sum*idiag[i]; /* omega in idiag */ 1722d4002b98SHong Zhang } 1723d4002b98SHong Zhang } 1724d4002b98SHong Zhang if (xb == b) { 1725d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1726d4002b98SHong Zhang } else { 1727d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */ 1728d4002b98SHong Zhang } 1729d4002b98SHong Zhang } 1730d4002b98SHong Zhang } 1731d4002b98SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1732d4002b98SHong Zhang ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1733d4002b98SHong Zhang PetscFunctionReturn(0); 1734d4002b98SHong Zhang } 1735d4002b98SHong Zhang 1736d4002b98SHong Zhang /* -------------------------------------------------------------------*/ 1737d4002b98SHong Zhang static struct _MatOps MatOps_Values = {MatSetValues_SeqSELL, 17386108893eSStefano Zampini MatGetRow_SeqSELL, 17396108893eSStefano Zampini MatRestoreRow_SeqSELL, 1740d4002b98SHong Zhang MatMult_SeqSELL, 1741d4002b98SHong Zhang /* 4*/ MatMultAdd_SeqSELL, 1742d4002b98SHong Zhang MatMultTranspose_SeqSELL, 1743d4002b98SHong Zhang MatMultTransposeAdd_SeqSELL, 1744f4259b30SLisandro Dalcin NULL, 1745f4259b30SLisandro Dalcin NULL, 1746f4259b30SLisandro Dalcin NULL, 1747f4259b30SLisandro Dalcin /* 10*/ NULL, 1748f4259b30SLisandro Dalcin NULL, 1749f4259b30SLisandro Dalcin NULL, 1750d4002b98SHong Zhang MatSOR_SeqSELL, 1751f4259b30SLisandro Dalcin NULL, 1752d4002b98SHong Zhang /* 15*/ MatGetInfo_SeqSELL, 1753d4002b98SHong Zhang MatEqual_SeqSELL, 1754d4002b98SHong Zhang MatGetDiagonal_SeqSELL, 1755d4002b98SHong Zhang MatDiagonalScale_SeqSELL, 1756f4259b30SLisandro Dalcin NULL, 1757f4259b30SLisandro Dalcin /* 20*/ NULL, 1758d4002b98SHong Zhang MatAssemblyEnd_SeqSELL, 1759d4002b98SHong Zhang MatSetOption_SeqSELL, 1760d4002b98SHong Zhang MatZeroEntries_SeqSELL, 1761f4259b30SLisandro Dalcin /* 24*/ NULL, 1762f4259b30SLisandro Dalcin NULL, 1763f4259b30SLisandro Dalcin NULL, 1764f4259b30SLisandro Dalcin NULL, 1765f4259b30SLisandro Dalcin NULL, 1766d4002b98SHong Zhang /* 29*/ MatSetUp_SeqSELL, 1767f4259b30SLisandro Dalcin NULL, 1768f4259b30SLisandro Dalcin NULL, 1769f4259b30SLisandro Dalcin NULL, 1770f4259b30SLisandro Dalcin NULL, 1771d4002b98SHong Zhang /* 34*/ MatDuplicate_SeqSELL, 1772f4259b30SLisandro Dalcin NULL, 1773f4259b30SLisandro Dalcin NULL, 1774f4259b30SLisandro Dalcin NULL, 1775f4259b30SLisandro Dalcin NULL, 1776f4259b30SLisandro Dalcin /* 39*/ NULL, 1777f4259b30SLisandro Dalcin NULL, 1778f4259b30SLisandro Dalcin NULL, 1779d4002b98SHong Zhang MatGetValues_SeqSELL, 1780d4002b98SHong Zhang MatCopy_SeqSELL, 1781f4259b30SLisandro Dalcin /* 44*/ NULL, 1782d4002b98SHong Zhang MatScale_SeqSELL, 1783d4002b98SHong Zhang MatShift_SeqSELL, 1784f4259b30SLisandro Dalcin NULL, 1785f4259b30SLisandro Dalcin NULL, 1786f4259b30SLisandro Dalcin /* 49*/ NULL, 1787f4259b30SLisandro Dalcin NULL, 1788f4259b30SLisandro Dalcin NULL, 1789f4259b30SLisandro Dalcin NULL, 1790f4259b30SLisandro Dalcin NULL, 1791d4002b98SHong Zhang /* 54*/ MatFDColoringCreate_SeqXAIJ, 1792f4259b30SLisandro Dalcin NULL, 1793f4259b30SLisandro Dalcin NULL, 1794f4259b30SLisandro Dalcin NULL, 1795f4259b30SLisandro Dalcin NULL, 1796f4259b30SLisandro Dalcin /* 59*/ NULL, 1797d4002b98SHong Zhang MatDestroy_SeqSELL, 1798d4002b98SHong Zhang MatView_SeqSELL, 1799f4259b30SLisandro Dalcin NULL, 1800f4259b30SLisandro Dalcin NULL, 1801f4259b30SLisandro Dalcin /* 64*/ NULL, 1802f4259b30SLisandro Dalcin NULL, 1803f4259b30SLisandro Dalcin NULL, 1804f4259b30SLisandro Dalcin NULL, 1805f4259b30SLisandro Dalcin NULL, 1806f4259b30SLisandro Dalcin /* 69*/ NULL, 1807f4259b30SLisandro Dalcin NULL, 1808f4259b30SLisandro Dalcin NULL, 1809f4259b30SLisandro Dalcin NULL, 1810f4259b30SLisandro Dalcin NULL, 1811f4259b30SLisandro Dalcin /* 74*/ NULL, 1812d4002b98SHong Zhang MatFDColoringApply_AIJ, /* reuse the FDColoring function for AIJ */ 1813f4259b30SLisandro Dalcin NULL, 1814f4259b30SLisandro Dalcin NULL, 1815f4259b30SLisandro Dalcin NULL, 1816f4259b30SLisandro Dalcin /* 79*/ NULL, 1817f4259b30SLisandro Dalcin NULL, 1818f4259b30SLisandro Dalcin NULL, 1819f4259b30SLisandro Dalcin NULL, 1820f4259b30SLisandro Dalcin NULL, 1821f4259b30SLisandro Dalcin /* 84*/ NULL, 1822f4259b30SLisandro Dalcin NULL, 1823f4259b30SLisandro Dalcin NULL, 1824f4259b30SLisandro Dalcin NULL, 1825f4259b30SLisandro Dalcin NULL, 1826f4259b30SLisandro Dalcin /* 89*/ NULL, 1827f4259b30SLisandro Dalcin NULL, 1828f4259b30SLisandro Dalcin NULL, 1829f4259b30SLisandro Dalcin NULL, 1830f4259b30SLisandro Dalcin NULL, 1831f4259b30SLisandro Dalcin /* 94*/ NULL, 1832f4259b30SLisandro Dalcin NULL, 1833f4259b30SLisandro Dalcin NULL, 1834f4259b30SLisandro Dalcin NULL, 1835f4259b30SLisandro Dalcin NULL, 1836f4259b30SLisandro Dalcin /* 99*/ NULL, 1837f4259b30SLisandro Dalcin NULL, 1838f4259b30SLisandro Dalcin NULL, 1839d4002b98SHong Zhang MatConjugate_SeqSELL, 1840f4259b30SLisandro Dalcin NULL, 1841f4259b30SLisandro Dalcin /*104*/ NULL, 1842f4259b30SLisandro Dalcin NULL, 1843f4259b30SLisandro Dalcin NULL, 1844f4259b30SLisandro Dalcin NULL, 1845f4259b30SLisandro Dalcin NULL, 1846f4259b30SLisandro Dalcin /*109*/ NULL, 1847f4259b30SLisandro Dalcin NULL, 1848f4259b30SLisandro Dalcin NULL, 1849f4259b30SLisandro Dalcin NULL, 1850d4002b98SHong Zhang MatMissingDiagonal_SeqSELL, 1851f4259b30SLisandro Dalcin /*114*/ NULL, 1852f4259b30SLisandro Dalcin NULL, 1853f4259b30SLisandro Dalcin NULL, 1854f4259b30SLisandro Dalcin NULL, 1855f4259b30SLisandro Dalcin NULL, 1856f4259b30SLisandro Dalcin /*119*/ NULL, 1857f4259b30SLisandro Dalcin NULL, 1858f4259b30SLisandro Dalcin NULL, 1859f4259b30SLisandro Dalcin NULL, 1860f4259b30SLisandro Dalcin NULL, 1861f4259b30SLisandro Dalcin /*124*/ NULL, 1862f4259b30SLisandro Dalcin NULL, 1863f4259b30SLisandro Dalcin NULL, 1864f4259b30SLisandro Dalcin NULL, 1865f4259b30SLisandro Dalcin NULL, 1866f4259b30SLisandro Dalcin /*129*/ NULL, 1867f4259b30SLisandro Dalcin NULL, 1868f4259b30SLisandro Dalcin NULL, 1869f4259b30SLisandro Dalcin NULL, 1870f4259b30SLisandro Dalcin NULL, 1871f4259b30SLisandro Dalcin /*134*/ NULL, 1872f4259b30SLisandro Dalcin NULL, 1873f4259b30SLisandro Dalcin NULL, 1874f4259b30SLisandro Dalcin NULL, 1875f4259b30SLisandro Dalcin NULL, 1876f4259b30SLisandro Dalcin /*139*/ NULL, 1877f4259b30SLisandro Dalcin NULL, 1878f4259b30SLisandro Dalcin NULL, 1879d4002b98SHong Zhang MatFDColoringSetUp_SeqXAIJ, 1880f4259b30SLisandro Dalcin NULL, 1881f4259b30SLisandro Dalcin /*144*/ NULL 1882d4002b98SHong Zhang }; 1883d4002b98SHong Zhang 1884d4002b98SHong Zhang PetscErrorCode MatStoreValues_SeqSELL(Mat mat) 1885d4002b98SHong Zhang { 1886d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)mat->data; 1887d4002b98SHong Zhang PetscErrorCode ierr; 1888d4002b98SHong Zhang 1889d4002b98SHong Zhang PetscFunctionBegin; 1890d4002b98SHong Zhang if (!a->nonew) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1891d4002b98SHong Zhang 1892d4002b98SHong Zhang /* allocate space for values if not already there */ 1893d4002b98SHong Zhang if (!a->saved_values) { 1894d4002b98SHong Zhang ierr = PetscMalloc1(a->sliidx[a->totalslices]+1,&a->saved_values);CHKERRQ(ierr); 1895d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)mat,(a->sliidx[a->totalslices]+1)*sizeof(PetscScalar));CHKERRQ(ierr); 1896d4002b98SHong Zhang } 1897d4002b98SHong Zhang 1898d4002b98SHong Zhang /* copy values over */ 1899580bdb30SBarry Smith ierr = PetscArraycpy(a->saved_values,a->val,a->sliidx[a->totalslices]);CHKERRQ(ierr); 1900d4002b98SHong Zhang PetscFunctionReturn(0); 1901d4002b98SHong Zhang } 1902d4002b98SHong Zhang 1903d4002b98SHong Zhang PetscErrorCode MatRetrieveValues_SeqSELL(Mat mat) 1904d4002b98SHong Zhang { 1905d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)mat->data; 1906d4002b98SHong Zhang PetscErrorCode ierr; 1907d4002b98SHong Zhang 1908d4002b98SHong Zhang PetscFunctionBegin; 1909d4002b98SHong Zhang if (!a->nonew) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1910d4002b98SHong Zhang if (!a->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 1911580bdb30SBarry Smith ierr = PetscArraycpy(a->val,a->saved_values,a->sliidx[a->totalslices]);CHKERRQ(ierr); 1912d4002b98SHong Zhang PetscFunctionReturn(0); 1913d4002b98SHong Zhang } 1914d4002b98SHong Zhang 1915d4002b98SHong Zhang /*@C 1916d4002b98SHong Zhang MatSeqSELLRestoreArray - returns access to the array where the data for a MATSEQSELL matrix is stored obtained by MatSeqSELLGetArray() 1917d4002b98SHong Zhang 1918d4002b98SHong Zhang Not Collective 1919d4002b98SHong Zhang 1920d4002b98SHong Zhang Input Parameters: 1921d4002b98SHong Zhang . mat - a MATSEQSELL matrix 1922d4002b98SHong Zhang . array - pointer to the data 1923d4002b98SHong Zhang 1924d4002b98SHong Zhang Level: intermediate 1925d4002b98SHong Zhang 1926d4002b98SHong Zhang .seealso: MatSeqSELLGetArray(), MatSeqSELLRestoreArrayF90() 1927d4002b98SHong Zhang @*/ 1928d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray(Mat A,PetscScalar **array) 1929d4002b98SHong Zhang { 1930d4002b98SHong Zhang PetscErrorCode ierr; 1931d4002b98SHong Zhang 1932d4002b98SHong Zhang PetscFunctionBegin; 1933d4002b98SHong Zhang ierr = PetscUseMethod(A,"MatSeqSELLRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr); 1934d4002b98SHong Zhang PetscFunctionReturn(0); 1935d4002b98SHong Zhang } 1936d4002b98SHong Zhang 1937d4002b98SHong Zhang PETSC_EXTERN PetscErrorCode MatCreate_SeqSELL(Mat B) 1938d4002b98SHong Zhang { 1939d4002b98SHong Zhang Mat_SeqSELL *b; 1940d4002b98SHong Zhang PetscMPIInt size; 1941d4002b98SHong Zhang PetscErrorCode ierr; 1942d4002b98SHong Zhang 1943d4002b98SHong Zhang PetscFunctionBegin; 1944ffc4695bSBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRMPI(ierr); 1945d4002b98SHong Zhang if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Comm must be of size 1"); 1946d4002b98SHong Zhang 1947d4002b98SHong Zhang ierr = PetscNewLog(B,&b);CHKERRQ(ierr); 1948d4002b98SHong Zhang 1949d4002b98SHong Zhang B->data = (void*)b; 1950d4002b98SHong Zhang 1951d4002b98SHong Zhang ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1952d4002b98SHong Zhang 1953f4259b30SLisandro Dalcin b->row = NULL; 1954f4259b30SLisandro Dalcin b->col = NULL; 1955f4259b30SLisandro Dalcin b->icol = NULL; 1956d4002b98SHong Zhang b->reallocs = 0; 1957d4002b98SHong Zhang b->ignorezeroentries = PETSC_FALSE; 1958d4002b98SHong Zhang b->roworiented = PETSC_TRUE; 1959d4002b98SHong Zhang b->nonew = 0; 1960f4259b30SLisandro Dalcin b->diag = NULL; 1961f4259b30SLisandro Dalcin b->solve_work = NULL; 1962f4259b30SLisandro Dalcin B->spptr = NULL; 1963f4259b30SLisandro Dalcin b->saved_values = NULL; 1964f4259b30SLisandro Dalcin b->idiag = NULL; 1965f4259b30SLisandro Dalcin b->mdiag = NULL; 1966f4259b30SLisandro Dalcin b->ssor_work = NULL; 1967d4002b98SHong Zhang b->omega = 1.0; 1968d4002b98SHong Zhang b->fshift = 0.0; 1969d4002b98SHong Zhang b->idiagvalid = PETSC_FALSE; 1970d4002b98SHong Zhang b->keepnonzeropattern = PETSC_FALSE; 1971d4002b98SHong Zhang 1972d4002b98SHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSELL);CHKERRQ(ierr); 1973d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLGetArray_C",MatSeqSELLGetArray_SeqSELL);CHKERRQ(ierr); 1974d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLRestoreArray_C",MatSeqSELLRestoreArray_SeqSELL);CHKERRQ(ierr); 1975d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSELL);CHKERRQ(ierr); 1976d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSELL);CHKERRQ(ierr); 1977d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLSetPreallocation_C",MatSeqSELLSetPreallocation_SeqSELL);CHKERRQ(ierr); 1978d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsell_seqaij_C",MatConvert_SeqSELL_SeqAIJ);CHKERRQ(ierr); 1979d4002b98SHong Zhang PetscFunctionReturn(0); 1980d4002b98SHong Zhang } 1981d4002b98SHong Zhang 1982d4002b98SHong Zhang /* 1983d4002b98SHong Zhang Given a matrix generated with MatGetFactor() duplicates all the information in A into B 1984d4002b98SHong Zhang */ 1985d4002b98SHong Zhang PetscErrorCode MatDuplicateNoCreate_SeqSELL(Mat C,Mat A,MatDuplicateOption cpvalues,PetscBool mallocmatspace) 1986d4002b98SHong Zhang { 1987d4002b98SHong Zhang Mat_SeqSELL *c,*a=(Mat_SeqSELL*)A->data; 1988d4002b98SHong Zhang PetscInt i,m=A->rmap->n; 1989d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 1990d4002b98SHong Zhang PetscErrorCode ierr; 1991d4002b98SHong Zhang 1992d4002b98SHong Zhang PetscFunctionBegin; 1993d4002b98SHong Zhang c = (Mat_SeqSELL*)C->data; 1994d4002b98SHong Zhang 1995d4002b98SHong Zhang C->factortype = A->factortype; 1996f4259b30SLisandro Dalcin c->row = NULL; 1997f4259b30SLisandro Dalcin c->col = NULL; 1998f4259b30SLisandro Dalcin c->icol = NULL; 1999d4002b98SHong Zhang c->reallocs = 0; 2000d4002b98SHong Zhang 2001d4002b98SHong Zhang C->assembled = PETSC_TRUE; 2002d4002b98SHong Zhang 2003d4002b98SHong Zhang ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr); 2004d4002b98SHong Zhang ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr); 2005d4002b98SHong Zhang 2006d4002b98SHong Zhang ierr = PetscMalloc1(8*totalslices,&c->rlen);CHKERRQ(ierr); 2007d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt));CHKERRQ(ierr); 2008d4002b98SHong Zhang ierr = PetscMalloc1(totalslices+1,&c->sliidx);CHKERRQ(ierr); 2009d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)C, (totalslices+1)*sizeof(PetscInt));CHKERRQ(ierr); 2010d4002b98SHong Zhang 2011d4002b98SHong Zhang for (i=0; i<m; i++) c->rlen[i] = a->rlen[i]; 2012d4002b98SHong Zhang for (i=0; i<totalslices+1; i++) c->sliidx[i] = a->sliidx[i]; 2013d4002b98SHong Zhang 2014d4002b98SHong Zhang /* allocate the matrix space */ 2015d4002b98SHong Zhang if (mallocmatspace) { 2016d4002b98SHong Zhang ierr = PetscMalloc2(a->maxallocmat,&c->val,a->maxallocmat,&c->colidx);CHKERRQ(ierr); 2017d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)C,a->maxallocmat*(sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 2018d4002b98SHong Zhang 2019d4002b98SHong Zhang c->singlemalloc = PETSC_TRUE; 2020d4002b98SHong Zhang 2021d4002b98SHong Zhang if (m > 0) { 2022580bdb30SBarry Smith ierr = PetscArraycpy(c->colidx,a->colidx,a->maxallocmat);CHKERRQ(ierr); 2023d4002b98SHong Zhang if (cpvalues == MAT_COPY_VALUES) { 2024580bdb30SBarry Smith ierr = PetscArraycpy(c->val,a->val,a->maxallocmat);CHKERRQ(ierr); 2025d4002b98SHong Zhang } else { 2026580bdb30SBarry Smith ierr = PetscArrayzero(c->val,a->maxallocmat);CHKERRQ(ierr); 2027d4002b98SHong Zhang } 2028d4002b98SHong Zhang } 2029d4002b98SHong Zhang } 2030d4002b98SHong Zhang 2031d4002b98SHong Zhang c->ignorezeroentries = a->ignorezeroentries; 2032d4002b98SHong Zhang c->roworiented = a->roworiented; 2033d4002b98SHong Zhang c->nonew = a->nonew; 2034d4002b98SHong Zhang if (a->diag) { 2035d4002b98SHong Zhang ierr = PetscMalloc1(m,&c->diag);CHKERRQ(ierr); 2036d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt));CHKERRQ(ierr); 2037d4002b98SHong Zhang for (i=0; i<m; i++) { 2038d4002b98SHong Zhang c->diag[i] = a->diag[i]; 2039d4002b98SHong Zhang } 2040f4259b30SLisandro Dalcin } else c->diag = NULL; 2041d4002b98SHong Zhang 2042f4259b30SLisandro Dalcin c->solve_work = NULL; 2043f4259b30SLisandro Dalcin c->saved_values = NULL; 2044f4259b30SLisandro Dalcin c->idiag = NULL; 2045f4259b30SLisandro Dalcin c->ssor_work = NULL; 2046d4002b98SHong Zhang c->keepnonzeropattern = a->keepnonzeropattern; 2047d4002b98SHong Zhang c->free_val = PETSC_TRUE; 2048d4002b98SHong Zhang c->free_colidx = PETSC_TRUE; 2049d4002b98SHong Zhang 2050d4002b98SHong Zhang c->maxallocmat = a->maxallocmat; 2051d4002b98SHong Zhang c->maxallocrow = a->maxallocrow; 2052d4002b98SHong Zhang c->rlenmax = a->rlenmax; 2053d4002b98SHong Zhang c->nz = a->nz; 2054d4002b98SHong Zhang C->preallocated = PETSC_TRUE; 2055d4002b98SHong Zhang 2056d4002b98SHong Zhang c->nonzerorowcnt = a->nonzerorowcnt; 2057d4002b98SHong Zhang C->nonzerostate = A->nonzerostate; 2058d4002b98SHong Zhang 2059d4002b98SHong Zhang ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr); 2060d4002b98SHong Zhang PetscFunctionReturn(0); 2061d4002b98SHong Zhang } 2062d4002b98SHong Zhang 2063d4002b98SHong Zhang PetscErrorCode MatDuplicate_SeqSELL(Mat A,MatDuplicateOption cpvalues,Mat *B) 2064d4002b98SHong Zhang { 2065d4002b98SHong Zhang PetscErrorCode ierr; 2066d4002b98SHong Zhang 2067d4002b98SHong Zhang PetscFunctionBegin; 2068d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 2069d4002b98SHong Zhang ierr = MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr); 2070d4002b98SHong Zhang if (!(A->rmap->n % A->rmap->bs) && !(A->cmap->n % A->cmap->bs)) { 2071d4002b98SHong Zhang ierr = MatSetBlockSizesFromMats(*B,A,A);CHKERRQ(ierr); 2072d4002b98SHong Zhang } 2073d4002b98SHong Zhang ierr = MatSetType(*B,((PetscObject)A)->type_name);CHKERRQ(ierr); 2074d4002b98SHong Zhang ierr = MatDuplicateNoCreate_SeqSELL(*B,A,cpvalues,PETSC_TRUE);CHKERRQ(ierr); 2075d4002b98SHong Zhang PetscFunctionReturn(0); 2076d4002b98SHong Zhang } 2077d4002b98SHong Zhang 2078d4002b98SHong Zhang /*@C 2079d4002b98SHong Zhang MatCreateSeqSELL - Creates a sparse matrix in SELL format. 2080d4002b98SHong Zhang 2081d083f849SBarry Smith Collective 2082d4002b98SHong Zhang 2083d4002b98SHong Zhang Input Parameters: 2084d4002b98SHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 2085d4002b98SHong Zhang . m - number of rows 2086d4002b98SHong Zhang . n - number of columns 2087d4002b98SHong Zhang . rlenmax - maximum number of nonzeros in a row 2088d4002b98SHong Zhang - rlen - array containing the number of nonzeros in the various rows 2089d4002b98SHong Zhang (possibly different for each row) or NULL 2090d4002b98SHong Zhang 2091d4002b98SHong Zhang Output Parameter: 2092d4002b98SHong Zhang . A - the matrix 2093d4002b98SHong Zhang 2094d4002b98SHong Zhang It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 2095f6f02116SRichard Tran Mills MatXXXXSetPreallocation() paradigm instead of this routine directly. 2096d4002b98SHong Zhang [MatXXXXSetPreallocation() is, for example, MatSeqSELLSetPreallocation] 2097d4002b98SHong Zhang 2098d4002b98SHong Zhang Notes: 2099d4002b98SHong Zhang If nnz is given then nz is ignored 2100d4002b98SHong Zhang 2101d4002b98SHong Zhang Specify the preallocated storage with either rlenmax or rlen (not both). 2102d4002b98SHong Zhang Set rlenmax=PETSC_DEFAULT and rlen=NULL for PETSc to control dynamic memory 2103d4002b98SHong Zhang allocation. For large problems you MUST preallocate memory or you 2104d4002b98SHong Zhang will get TERRIBLE performance, see the users' manual chapter on matrices. 2105d4002b98SHong Zhang 2106d4002b98SHong Zhang Level: intermediate 2107d4002b98SHong Zhang 21083eec2f6aSPierre Jolivet .seealso: MatCreate(), MatCreateSELL(), MatSetValues() 2109d4002b98SHong Zhang 2110d4002b98SHong Zhang @*/ 2111d4002b98SHong Zhang PetscErrorCode MatCreateSeqSELL(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt maxallocrow,const PetscInt rlen[],Mat *A) 2112d4002b98SHong Zhang { 2113d4002b98SHong Zhang PetscErrorCode ierr; 2114d4002b98SHong Zhang 2115d4002b98SHong Zhang PetscFunctionBegin; 2116d4002b98SHong Zhang ierr = MatCreate(comm,A);CHKERRQ(ierr); 2117d4002b98SHong Zhang ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 2118d4002b98SHong Zhang ierr = MatSetType(*A,MATSEQSELL);CHKERRQ(ierr); 2119d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation_SeqSELL(*A,maxallocrow,rlen);CHKERRQ(ierr); 2120d4002b98SHong Zhang PetscFunctionReturn(0); 2121d4002b98SHong Zhang } 2122d4002b98SHong Zhang 2123d4002b98SHong Zhang PetscErrorCode MatEqual_SeqSELL(Mat A,Mat B,PetscBool * flg) 2124d4002b98SHong Zhang { 2125d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data,*b=(Mat_SeqSELL*)B->data; 2126d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 2127d4002b98SHong Zhang PetscErrorCode ierr; 2128d4002b98SHong Zhang 2129d4002b98SHong Zhang PetscFunctionBegin; 2130d4002b98SHong Zhang /* If the matrix dimensions are not equal,or no of nonzeros */ 2131d4002b98SHong Zhang if ((A->rmap->n != B->rmap->n) || (A->cmap->n != B->cmap->n) ||(a->nz != b->nz) || (a->rlenmax != b->rlenmax)) { 2132d4002b98SHong Zhang *flg = PETSC_FALSE; 2133d4002b98SHong Zhang PetscFunctionReturn(0); 2134d4002b98SHong Zhang } 2135d4002b98SHong Zhang /* if the a->colidx are the same */ 2136580bdb30SBarry Smith ierr = PetscArraycmp(a->colidx,b->colidx,a->sliidx[totalslices],flg);CHKERRQ(ierr); 2137d4002b98SHong Zhang if (!*flg) PetscFunctionReturn(0); 2138d4002b98SHong Zhang /* if a->val are the same */ 2139580bdb30SBarry Smith ierr = PetscArraycmp(a->val,b->val,a->sliidx[totalslices],flg);CHKERRQ(ierr); 2140d4002b98SHong Zhang PetscFunctionReturn(0); 2141d4002b98SHong Zhang } 2142d4002b98SHong Zhang 2143d4002b98SHong Zhang PetscErrorCode MatSeqSELLInvalidateDiagonal(Mat A) 2144d4002b98SHong Zhang { 2145d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 2146d4002b98SHong Zhang 2147d4002b98SHong Zhang PetscFunctionBegin; 2148d4002b98SHong Zhang a->idiagvalid = PETSC_FALSE; 2149d4002b98SHong Zhang PetscFunctionReturn(0); 2150d4002b98SHong Zhang } 2151d4002b98SHong Zhang 2152d4002b98SHong Zhang PetscErrorCode MatConjugate_SeqSELL(Mat A) 2153d4002b98SHong Zhang { 2154d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 2155d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 2156d4002b98SHong Zhang PetscInt i; 2157d4002b98SHong Zhang PetscScalar *val = a->val; 2158d4002b98SHong Zhang 2159d4002b98SHong Zhang PetscFunctionBegin; 2160d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) { 2161d4002b98SHong Zhang val[i] = PetscConj(val[i]); 2162d4002b98SHong Zhang } 2163d4002b98SHong Zhang #else 2164d4002b98SHong Zhang PetscFunctionBegin; 2165d4002b98SHong Zhang #endif 2166d4002b98SHong Zhang PetscFunctionReturn(0); 2167d4002b98SHong Zhang } 2168