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> 8d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) 9d4002b98SHong Zhang #include <immintrin.h> 10d4002b98SHong Zhang 11d4002b98SHong Zhang #if !defined(_MM_SCALE_8) 12d4002b98SHong Zhang #define _MM_SCALE_8 8 13d4002b98SHong Zhang #endif 14d4002b98SHong Zhang 15d4002b98SHong Zhang #if defined(__AVX512F__) 16d4002b98SHong Zhang /* these do not work 17d4002b98SHong Zhang vec_idx = _mm512_loadunpackhi_epi32(vec_idx,acolidx); 18d4002b98SHong Zhang vec_vals = _mm512_loadunpackhi_pd(vec_vals,aval); 19d4002b98SHong Zhang */ 20d4002b98SHong Zhang #define AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y) \ 21d4002b98SHong Zhang /* if the mask bit is set, copy from acolidx, otherwise from vec_idx */ \ 22d4002b98SHong Zhang vec_idx = _mm256_load_si256((__m256i const*)acolidx); \ 23d4002b98SHong Zhang vec_vals = _mm512_load_pd(aval); \ 24d4002b98SHong Zhang vec_x = _mm512_i32gather_pd(vec_idx,x,_MM_SCALE_8); \ 25d4002b98SHong Zhang vec_y = _mm512_fmadd_pd(vec_x,vec_vals,vec_y); 26d4002b98SHong Zhang 27d4002b98SHong Zhang #endif 28d4002b98SHong Zhang #endif /* PETSC_HAVE_IMMINTRIN_H */ 29d4002b98SHong Zhang 30d4002b98SHong Zhang /*@C 31d4002b98SHong Zhang MatSeqSELLSetPreallocation - For good matrix assembly performance 32d4002b98SHong Zhang the user should preallocate the matrix storage by setting the parameter nz 33d4002b98SHong Zhang (or the array nnz). By setting these parameters accurately, performance 34d4002b98SHong Zhang during matrix assembly can be increased significantly. 35d4002b98SHong Zhang 36d4002b98SHong Zhang Collective on MPI_Comm 37d4002b98SHong Zhang 38d4002b98SHong Zhang Input Parameters: 39d4002b98SHong Zhang + B - The matrix 40d4002b98SHong Zhang . nz - number of nonzeros per row (same for all rows) 41d4002b98SHong Zhang - nnz - array containing the number of nonzeros in the various rows 42d4002b98SHong Zhang (possibly different for each row) or NULL 43d4002b98SHong Zhang 44d4002b98SHong Zhang Notes: 45d4002b98SHong Zhang If nnz is given then nz is ignored. 46d4002b98SHong Zhang 47d4002b98SHong Zhang Specify the preallocated storage with either nz or nnz (not both). 48d4002b98SHong Zhang Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 49d4002b98SHong Zhang allocation. For large problems you MUST preallocate memory or you 50d4002b98SHong Zhang will get TERRIBLE performance, see the users' manual chapter on matrices. 51d4002b98SHong Zhang 52d4002b98SHong Zhang You can call MatGetInfo() to get information on how effective the preallocation was; 53d4002b98SHong Zhang for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 54d4002b98SHong Zhang You can also run with the option -info and look for messages with the string 55d4002b98SHong Zhang malloc in them to see if additional memory allocation was needed. 56d4002b98SHong Zhang 57d4002b98SHong Zhang Developers: Use nz of MAT_SKIP_ALLOCATION to not allocate any space for the matrix 58d4002b98SHong Zhang entries or columns indices. 59d4002b98SHong Zhang 60d4002b98SHong Zhang The maximum number of nonzeos in any row should be as accuate as possible. 61d4002b98SHong Zhang If it is underesitmated, you will get bad performance due to reallocation 62d4002b98SHong Zhang (MatSeqXSELLReallocateSELL). 63d4002b98SHong Zhang 64d4002b98SHong Zhang Level: intermediate 65d4002b98SHong Zhang 66d4002b98SHong Zhang .seealso: MatCreate(), MatCreateSELL(), MatSetValues(), MatGetInfo() 67d4002b98SHong Zhang 68d4002b98SHong Zhang @*/ 69d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation(Mat B,PetscInt rlenmax,const PetscInt rlen[]) 70d4002b98SHong Zhang { 71d4002b98SHong Zhang PetscErrorCode ierr; 72d4002b98SHong Zhang 73d4002b98SHong Zhang PetscFunctionBegin; 74d4002b98SHong Zhang PetscValidHeaderSpecific(B,MAT_CLASSID,1); 75d4002b98SHong Zhang PetscValidType(B,1); 76d4002b98SHong Zhang ierr = PetscTryMethod(B,"MatSeqSELLSetPreallocation_C",(Mat,PetscInt,const PetscInt[]),(B,rlenmax,rlen));CHKERRQ(ierr); 77d4002b98SHong Zhang PetscFunctionReturn(0); 78d4002b98SHong Zhang } 79d4002b98SHong Zhang 80d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation_SeqSELL(Mat B,PetscInt maxallocrow,const PetscInt rlen[]) 81d4002b98SHong Zhang { 82d4002b98SHong Zhang Mat_SeqSELL *b; 83d4002b98SHong Zhang PetscInt i,j,totalslices; 84d4002b98SHong Zhang PetscBool skipallocation=PETSC_FALSE,realalloc=PETSC_FALSE; 85d4002b98SHong Zhang PetscErrorCode ierr; 86d4002b98SHong Zhang 87d4002b98SHong Zhang PetscFunctionBegin; 88d4002b98SHong Zhang if (maxallocrow >= 0 || rlen) realalloc = PETSC_TRUE; 89d4002b98SHong Zhang if (maxallocrow == MAT_SKIP_ALLOCATION) { 90d4002b98SHong Zhang skipallocation = PETSC_TRUE; 91d4002b98SHong Zhang maxallocrow = 0; 92d4002b98SHong Zhang } 93d4002b98SHong Zhang 94d4002b98SHong Zhang ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 95d4002b98SHong Zhang ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 96d4002b98SHong Zhang 97d4002b98SHong Zhang /* FIXME: if one preallocates more space than needed, the matrix does not shrink automatically, but for best performance it should */ 98d4002b98SHong Zhang if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 5; 99d4002b98SHong Zhang if (maxallocrow < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"maxallocrow cannot be less than 0: value %D",maxallocrow); 100d4002b98SHong Zhang if (rlen) { 101d4002b98SHong Zhang for (i=0; i<B->rmap->n; i++) { 102d4002b98SHong Zhang if (rlen[i] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"rlen cannot be less than 0: local row %D value %D",i,rlen[i]); 103d4002b98SHong Zhang if (rlen[i] > B->cmap->n) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"rlen cannot be greater than row length: local row %D value %D rowlength %D",i,rlen[i],B->cmap->n); 104d4002b98SHong Zhang } 105d4002b98SHong Zhang } 106d4002b98SHong Zhang 107d4002b98SHong Zhang B->preallocated = PETSC_TRUE; 108d4002b98SHong Zhang 109d4002b98SHong Zhang b = (Mat_SeqSELL*)B->data; 110d4002b98SHong Zhang 111d4002b98SHong Zhang totalslices = B->rmap->n/8+((B->rmap->n & 0x07)?1:0); /* ceil(n/8) */ 112d4002b98SHong Zhang b->totalslices = totalslices; 113d4002b98SHong Zhang if (!skipallocation) { 114d4002b98SHong Zhang if (B->rmap->n & 0x07) PetscInfo1(B,"Padding rows to the SEQSELL matrix because the number of rows is not the multiple of 8 (value %D)\n",B->rmap->n); 115d4002b98SHong Zhang 116d4002b98SHong Zhang if (!b->sliidx) { /* sliidx gives the starting index of each slice, the last element is the total space allocated */ 117d4002b98SHong Zhang ierr = PetscMalloc1(totalslices+1,&b->sliidx);CHKERRQ(ierr); 118d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)B,(totalslices+1)*sizeof(PetscInt));CHKERRQ(ierr); 119d4002b98SHong Zhang } 120d4002b98SHong Zhang if (!rlen) { /* if rlen is not provided, allocate same space for all the slices */ 121d4002b98SHong Zhang if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 10; 122d4002b98SHong Zhang else if (maxallocrow < 0) maxallocrow = 1; 123d4002b98SHong Zhang for (i=0; i<=totalslices; i++) b->sliidx[i] = i*8*maxallocrow; 124d4002b98SHong Zhang } else { 125d4002b98SHong Zhang maxallocrow = 0; 126d4002b98SHong Zhang b->sliidx[0] = 0; 127d4002b98SHong Zhang for (i=1; i<totalslices; i++) { 128d4002b98SHong Zhang b->sliidx[i] = 0; 129d4002b98SHong Zhang for (j=0;j<8;j++) { 130d4002b98SHong Zhang b->sliidx[i] = PetscMax(b->sliidx[i],rlen[8*(i-1)+j]); 131d4002b98SHong Zhang } 132d4002b98SHong Zhang maxallocrow = PetscMax(b->sliidx[i],maxallocrow); 133d4002b98SHong Zhang b->sliidx[i] = b->sliidx[i-1] + 8*b->sliidx[i]; 134d4002b98SHong Zhang } 135d4002b98SHong Zhang /* last slice */ 136d4002b98SHong Zhang b->sliidx[totalslices] = 0; 137d4002b98SHong Zhang for (j=(totalslices-1)*8;j<B->rmap->n;j++) b->sliidx[totalslices] = PetscMax(b->sliidx[totalslices],rlen[j]); 138d4002b98SHong Zhang maxallocrow = PetscMax(b->sliidx[totalslices],maxallocrow); 139d4002b98SHong Zhang b->sliidx[totalslices] = b->sliidx[totalslices-1] + 8*b->sliidx[totalslices]; 140d4002b98SHong Zhang } 141d4002b98SHong Zhang 142d4002b98SHong Zhang /* allocate space for val, colidx, rlen */ 143d4002b98SHong Zhang /* FIXME: should B's old memory be unlogged? */ 144d4002b98SHong Zhang ierr = MatSeqXSELLFreeSELL(B,&b->val,&b->colidx);CHKERRQ(ierr); 145d4002b98SHong Zhang /* FIXME: assuming an element of the bit array takes 8 bits */ 146d4002b98SHong Zhang ierr = PetscMalloc2(b->sliidx[totalslices],&b->val,b->sliidx[totalslices],&b->colidx);CHKERRQ(ierr); 147d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)B,b->sliidx[totalslices]*(sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 148d4002b98SHong 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. */ 149d4002b98SHong Zhang ierr = PetscCalloc1(8*totalslices,&b->rlen);CHKERRQ(ierr); 150d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)B,8*totalslices*sizeof(PetscInt));CHKERRQ(ierr); 151d4002b98SHong Zhang 152d4002b98SHong Zhang b->singlemalloc = PETSC_TRUE; 153d4002b98SHong Zhang b->free_val = PETSC_TRUE; 154d4002b98SHong Zhang b->free_colidx = PETSC_TRUE; 155d4002b98SHong Zhang } else { 156d4002b98SHong Zhang b->free_val = PETSC_FALSE; 157d4002b98SHong Zhang b->free_colidx = PETSC_FALSE; 158d4002b98SHong Zhang } 159d4002b98SHong Zhang 160d4002b98SHong Zhang b->nz = 0; 161d4002b98SHong Zhang b->maxallocrow = maxallocrow; 162d4002b98SHong Zhang b->rlenmax = maxallocrow; 163d4002b98SHong Zhang b->maxallocmat = b->sliidx[totalslices]; 164d4002b98SHong Zhang B->info.nz_unneeded = (double)b->maxallocmat; 165d4002b98SHong Zhang if (realalloc) { 166d4002b98SHong Zhang ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 167d4002b98SHong Zhang } 168d4002b98SHong Zhang PetscFunctionReturn(0); 169d4002b98SHong Zhang } 170d4002b98SHong Zhang 171d4002b98SHong Zhang PetscErrorCode MatConvert_SeqSELL_SeqAIJ(Mat A, MatType newtype,MatReuse reuse,Mat *newmat) 172d4002b98SHong Zhang { 173d4002b98SHong Zhang Mat B; 174d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 175d4002b98SHong Zhang PetscInt i,j,row; 176d4002b98SHong Zhang PetscBool isnonzero; 177d4002b98SHong Zhang PetscErrorCode ierr; 178d4002b98SHong Zhang 179d4002b98SHong Zhang PetscFunctionBegin; 180d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 181d4002b98SHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 182d4002b98SHong Zhang ierr = MatSetType(B,MATSEQAIJ);CHKERRQ(ierr); 183d4002b98SHong Zhang ierr = MatSeqAIJSetPreallocation(B,0,a->rlen);CHKERRQ(ierr); 184d4002b98SHong Zhang ierr = MatSetOption(B,MAT_ROW_ORIENTED,PETSC_FALSE);CHKERRQ(ierr); 185d4002b98SHong Zhang 186d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 187d4002b98SHong Zhang for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 188d4002b98SHong Zhang isnonzero = (PetscBool)((j-a->sliidx[i])/8 < a->rlen[8*i+row]); 189d4002b98SHong Zhang if (isnonzero) { 190d4002b98SHong Zhang ierr = MatSetValue(B,8*i+row,a->colidx[j],a->val[j],INSERT_VALUES);CHKERRQ(ierr); 191d4002b98SHong Zhang } 192d4002b98SHong Zhang } 193d4002b98SHong Zhang } 194d4002b98SHong Zhang ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 195d4002b98SHong Zhang ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 196d4002b98SHong Zhang B->rmap->bs = A->rmap->bs; 197d4002b98SHong Zhang 198d4002b98SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 199d4002b98SHong Zhang ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); 200d4002b98SHong Zhang } else { 201d4002b98SHong Zhang *newmat = B; 202d4002b98SHong Zhang } 203d4002b98SHong Zhang PetscFunctionReturn(0); 204d4002b98SHong Zhang } 205d4002b98SHong Zhang 206d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/aij.h> 207d4002b98SHong Zhang 208d4002b98SHong Zhang PetscErrorCode MatConvert_SeqAIJ_SeqSELL(Mat A,MatType newtype,MatReuse reuse,Mat *newmat) 209d4002b98SHong Zhang { 210d4002b98SHong Zhang Mat B; 211d4002b98SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 212d4002b98SHong Zhang PetscInt *ai=a->i,m=A->rmap->N,n=A->cmap->N,i,*rowlengths,row,ncols; 213d4002b98SHong Zhang const PetscInt *cols; 214d4002b98SHong Zhang const PetscScalar *vals; 215d4002b98SHong Zhang PetscErrorCode ierr; 216d4002b98SHong Zhang 217d4002b98SHong Zhang PetscFunctionBegin; 218d4002b98SHong Zhang if (A->rmap->bs > 1) { 219d4002b98SHong Zhang ierr = MatConvert_Basic(A,newtype,reuse,newmat);CHKERRQ(ierr); 220d4002b98SHong Zhang PetscFunctionReturn(0); 221d4002b98SHong Zhang } 222d4002b98SHong Zhang /* Can we just use ilen? */ 223d4002b98SHong Zhang ierr = PetscMalloc1(m,&rowlengths);CHKERRQ(ierr); 224d4002b98SHong Zhang for (i=0; i<m; i++) { 225d4002b98SHong Zhang rowlengths[i] = ai[i+1] - ai[i]; 226d4002b98SHong Zhang } 227d4002b98SHong Zhang 228d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 229d4002b98SHong Zhang ierr = MatSetSizes(B,m,n,m,n);CHKERRQ(ierr); 230d4002b98SHong Zhang ierr = MatSetType(B,MATSEQSELL);CHKERRQ(ierr); 231d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation(B,0,rowlengths);CHKERRQ(ierr); 232d4002b98SHong Zhang ierr = PetscFree(rowlengths);CHKERRQ(ierr); 233d4002b98SHong Zhang 234d4002b98SHong Zhang ierr = MatSetOption(B,MAT_ROW_ORIENTED,PETSC_TRUE);CHKERRQ(ierr); 235d4002b98SHong Zhang 236d4002b98SHong Zhang for (row=0; row<m; row++) { 237d4002b98SHong Zhang ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 238d4002b98SHong Zhang ierr = MatSetValues(B,1,&row,ncols,cols,vals,INSERT_VALUES);CHKERRQ(ierr); 239d4002b98SHong Zhang ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 240d4002b98SHong Zhang } 241d4002b98SHong Zhang ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 242d4002b98SHong Zhang ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 243d4002b98SHong Zhang B->rmap->bs = A->rmap->bs; 244d4002b98SHong Zhang 245d4002b98SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 246d4002b98SHong Zhang ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); 247d4002b98SHong Zhang } else { 248d4002b98SHong Zhang *newmat = B; 249d4002b98SHong Zhang } 250d4002b98SHong Zhang PetscFunctionReturn(0); 251d4002b98SHong Zhang } 252d4002b98SHong Zhang 253d4002b98SHong Zhang PetscErrorCode MatMult_SeqSELL(Mat A,Vec xx,Vec yy) 254d4002b98SHong Zhang { 255d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 256d4002b98SHong Zhang PetscScalar *y; 257d4002b98SHong Zhang const PetscScalar *x; 258d4002b98SHong Zhang const MatScalar *aval=a->val; 259d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 260d4002b98SHong Zhang const PetscInt *acolidx=a->colidx; 261d4002b98SHong Zhang PetscInt i,j=0; 262d4002b98SHong Zhang PetscErrorCode ierr; 263d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 264d4002b98SHong Zhang __m512d vec_x,vec_y,vec_vals; 265d4002b98SHong Zhang __m256i vec_idx; 266d4002b98SHong Zhang __mmask8 mask; 267d4002b98SHong Zhang __m512d vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4; 268d4002b98SHong Zhang __m256i vec_idx2,vec_idx3,vec_idx4; 269d4002b98SHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) 270d4002b98SHong Zhang __m128d vec_x_tmp; 271d4002b98SHong Zhang __m256d vec_x,vec_y,vec_y2,vec_vals; 272d4002b98SHong Zhang MatScalar yval; 273d4002b98SHong Zhang PetscInt r,rows_left,row,nnz_in_row; 274d4002b98SHong Zhang #else 275d4002b98SHong Zhang PetscScalar sum[8]; 276d4002b98SHong Zhang #endif 277d4002b98SHong Zhang 278d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 279d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval) 280d4002b98SHong Zhang #endif 281d4002b98SHong Zhang 282d4002b98SHong Zhang PetscFunctionBegin; 283d4002b98SHong Zhang ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 284d4002b98SHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 285d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 286d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 287d4002b98SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 288d4002b98SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 289d4002b98SHong Zhang 290d4002b98SHong Zhang vec_y = _mm512_setzero_pd(); 291d4002b98SHong Zhang vec_y2 = _mm512_setzero_pd(); 292d4002b98SHong Zhang vec_y3 = _mm512_setzero_pd(); 293d4002b98SHong Zhang vec_y4 = _mm512_setzero_pd(); 294d4002b98SHong Zhang 295*38efe8efSHong Zhang j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */ 296d4002b98SHong Zhang switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) { 297d4002b98SHong Zhang case 3: 298d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 299d4002b98SHong Zhang acolidx += 8; aval += 8; 300d4002b98SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 301d4002b98SHong Zhang acolidx += 8; aval += 8; 302d4002b98SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 303d4002b98SHong Zhang acolidx += 8; aval += 8; 304d4002b98SHong Zhang j += 3; 305d4002b98SHong Zhang break; 306d4002b98SHong Zhang case 2: 307d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 308d4002b98SHong Zhang acolidx += 8; aval += 8; 309d4002b98SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 310d4002b98SHong Zhang acolidx += 8; aval += 8; 311d4002b98SHong Zhang j += 2; 312d4002b98SHong Zhang break; 313d4002b98SHong Zhang case 1: 314d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 315d4002b98SHong Zhang acolidx += 8; aval += 8; 316d4002b98SHong Zhang j += 1; 317d4002b98SHong Zhang break; 318d4002b98SHong Zhang } 319d4002b98SHong Zhang #pragma novector 320d4002b98SHong Zhang for (; j<(a->sliidx[i+1]>>3); j+=4) { 321d4002b98SHong Zhang /* 322d4002b98SHong Zhang vec_idx = _mm512_loadunpackhi_epi32(vec_idx,acolidx); 323d4002b98SHong Zhang vec_vals = _mm512_loadunpackhi_pd(vec_vals,aval); 324d4002b98SHong Zhang */ 325d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 326d4002b98SHong Zhang acolidx += 8; aval += 8; 327d4002b98SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 328d4002b98SHong Zhang acolidx += 8; aval += 8; 329d4002b98SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 330d4002b98SHong Zhang acolidx += 8; aval += 8; 331d4002b98SHong Zhang AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4); 332d4002b98SHong Zhang acolidx += 8; aval += 8; 333d4002b98SHong Zhang } 334d4002b98SHong Zhang 335d4002b98SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y2); 336d4002b98SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y3); 337d4002b98SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y4); 338d4002b98SHong Zhang if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */ 339d4002b98SHong Zhang mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07))); 340d4002b98SHong Zhang _mm512_mask_store_pd(&y[8*i],mask,vec_y); 341d4002b98SHong Zhang } else { 342d4002b98SHong Zhang _mm512_store_pd(&y[8*i],vec_y); 343d4002b98SHong Zhang } 344d4002b98SHong Zhang } 345d4002b98SHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) 346d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over full slices */ 347d4002b98SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 348d4002b98SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 349d4002b98SHong Zhang 350d4002b98SHong Zhang vec_y = _mm256_setzero_pd(); 351d4002b98SHong Zhang vec_y2 = _mm256_setzero_pd(); 352d4002b98SHong Zhang 353d4002b98SHong Zhang /* last slice may have padding rows. Don't use vectorization. */ 354d4002b98SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 355d4002b98SHong Zhang rows_left = A->rmap->n - 8*i; 356d4002b98SHong Zhang for (r=0; r<rows_left; ++r) { 357d4002b98SHong Zhang yval = (MatScalar)0; 358d4002b98SHong Zhang row = 8*i + r; 359d4002b98SHong Zhang nnz_in_row = a->rlen[row]; 360d4002b98SHong Zhang for (j=0; j<nnz_in_row; ++j) yval += aval[8*j + r] * x[acolidx[8*j + r]]; 361d4002b98SHong Zhang y[row] = yval; 362d4002b98SHong Zhang } 363d4002b98SHong Zhang break; 364d4002b98SHong Zhang } 365d4002b98SHong Zhang 366d4002b98SHong Zhang /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */ 367d4002b98SHong Zhang for (; j<a->sliidx[i+1]; j+=8) { 368d4002b98SHong Zhang vec_vals = _mm256_loadu_pd(aval); 369d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 370d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 371d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 372d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 373d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 374d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 375d4002b98SHong Zhang vec_y = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y); 376d4002b98SHong Zhang aval += 4; 377d4002b98SHong Zhang 378d4002b98SHong Zhang vec_vals = _mm256_loadu_pd(aval); 379d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 380d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 381d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 382d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 383d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 384d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 385d4002b98SHong Zhang vec_y2 = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2); 386d4002b98SHong Zhang aval += 4; 387d4002b98SHong Zhang } 388d4002b98SHong Zhang 389d4002b98SHong Zhang _mm256_storeu_pd(y + i*8, vec_y); 390d4002b98SHong Zhang _mm256_storeu_pd(y + i*8 + 4, vec_y2); 391d4002b98SHong Zhang } 392d4002b98SHong Zhang 393d4002b98SHong Zhang #else 394d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 395d4002b98SHong Zhang for (j=0; j<8; j++) sum[j] = 0.0; 396d4002b98SHong Zhang 397d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 398d4002b98SHong Zhang sum[0] += aval[j]*x[acolidx[j]]; 399d4002b98SHong Zhang sum[1] += aval[j+1]*x[acolidx[j+1]]; 400d4002b98SHong Zhang sum[2] += aval[j+2]*x[acolidx[j+2]]; 401d4002b98SHong Zhang sum[3] += aval[j+3]*x[acolidx[j+3]]; 402d4002b98SHong Zhang sum[4] += aval[j+4]*x[acolidx[j+4]]; 403d4002b98SHong Zhang sum[5] += aval[j+5]*x[acolidx[j+5]]; 404d4002b98SHong Zhang sum[6] += aval[j+6]*x[acolidx[j+6]]; 405d4002b98SHong Zhang sum[7] += aval[j+7]*x[acolidx[j+7]]; 406d4002b98SHong Zhang } 407d4002b98SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */ 408d4002b98SHong Zhang for(j=0;j<(A->rmap->n & 0x07);j++) y[8*i+j] = sum[j]; 409d4002b98SHong Zhang } else { 410d4002b98SHong Zhang y[8*i] = sum[0]; 411d4002b98SHong Zhang y[8*i+1] = sum[1]; 412d4002b98SHong Zhang y[8*i+2] = sum[2]; 413d4002b98SHong Zhang y[8*i+3] = sum[3]; 414d4002b98SHong Zhang y[8*i+4] = sum[4]; 415d4002b98SHong Zhang y[8*i+5] = sum[5]; 416d4002b98SHong Zhang y[8*i+6] = sum[6]; 417d4002b98SHong Zhang y[8*i+7] = sum[7]; 418d4002b98SHong Zhang } 419d4002b98SHong Zhang } 420d4002b98SHong Zhang #endif 421d4002b98SHong Zhang 422d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->nz-a->nonzerorowcnt);CHKERRQ(ierr); /* theoretical minimal FLOPs */ 423d4002b98SHong Zhang ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 424d4002b98SHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 425d4002b98SHong Zhang PetscFunctionReturn(0); 426d4002b98SHong Zhang } 427d4002b98SHong Zhang 428d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/ftn-kernels/fmultadd.h> 429d4002b98SHong Zhang PetscErrorCode MatMultAdd_SeqSELL(Mat A,Vec xx,Vec yy,Vec zz) 430d4002b98SHong Zhang { 431d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 432d4002b98SHong Zhang PetscScalar *y,*z; 433d4002b98SHong Zhang const PetscScalar *x; 434d4002b98SHong Zhang const MatScalar *aval=a->val; 435d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 436d4002b98SHong Zhang const PetscInt *acolidx=a->colidx; 437d4002b98SHong Zhang PetscInt i,j; 438d4002b98SHong Zhang PetscErrorCode ierr; 439d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 440d4002b98SHong Zhang __m512d vec_x,vec_y,vec_z,vec_vals; 441d4002b98SHong Zhang __m256i vec_idx; 442d4002b98SHong Zhang __mmask8 mask; 443d4002b98SHong Zhang #else 444d4002b98SHong Zhang PetscScalar sum[8]; 445d4002b98SHong Zhang #endif 446d4002b98SHong Zhang 447d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 448d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval) 449d4002b98SHong Zhang #endif 450d4002b98SHong Zhang 451d4002b98SHong Zhang PetscFunctionBegin; 452d4002b98SHong Zhang ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 453d4002b98SHong Zhang ierr = VecGetArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 454d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 455d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 456d4002b98SHong Zhang vec_y = _mm512_load_pd(&y[8*i]); 457d4002b98SHong Zhang #pragma novector 458d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 459d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 460d4002b98SHong Zhang acolidx += 8; aval += 8; 461d4002b98SHong Zhang } 462d4002b98SHong Zhang if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */ 463d4002b98SHong Zhang mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07))); 464d4002b98SHong Zhang _mm512_mask_store_pd(&z[8*i],mask,vec_y); 465d4002b98SHong Zhang } else { 466d4002b98SHong Zhang _mm512_store_pd(&z[8*i],vec_y); 467d4002b98SHong Zhang } 468d4002b98SHong Zhang #else 469d4002b98SHong Zhang sum[0] = y[8*i]; 470d4002b98SHong Zhang sum[1] = y[8*i+1]; 471d4002b98SHong Zhang sum[2] = y[8*i+2]; 472d4002b98SHong Zhang sum[3] = y[8*i+3]; 473d4002b98SHong Zhang sum[4] = y[8*i+4]; 474d4002b98SHong Zhang sum[5] = y[8*i+5]; 475d4002b98SHong Zhang sum[6] = y[8*i+6]; 476d4002b98SHong Zhang sum[7] = y[8*i+7]; 477d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 478d4002b98SHong Zhang sum[0] += aval[j]*x[acolidx[j]]; 479d4002b98SHong Zhang sum[1] += aval[j+1]*x[acolidx[j+1]]; 480d4002b98SHong Zhang sum[2] += aval[j+2]*x[acolidx[j+2]]; 481d4002b98SHong Zhang sum[3] += aval[j+3]*x[acolidx[j+3]]; 482d4002b98SHong Zhang sum[4] += aval[j+4]*x[acolidx[j+4]]; 483d4002b98SHong Zhang sum[5] += aval[j+5]*x[acolidx[j+5]]; 484d4002b98SHong Zhang sum[6] += aval[j+6]*x[acolidx[j+6]]; 485d4002b98SHong Zhang sum[7] += aval[j+7]*x[acolidx[j+7]]; 486d4002b98SHong Zhang } 487d4002b98SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */ 488d4002b98SHong Zhang for(j=0;j<(A->rmap->n & 0x07);j++) z[8*i+j] = sum[j]; 489d4002b98SHong Zhang } else { 490d4002b98SHong Zhang z[8*i] = sum[0]; 491d4002b98SHong Zhang z[8*i+1] = sum[1]; 492d4002b98SHong Zhang z[8*i+2] = sum[2]; 493d4002b98SHong Zhang z[8*i+3] = sum[3]; 494d4002b98SHong Zhang z[8*i+4] = sum[4]; 495d4002b98SHong Zhang z[8*i+5] = sum[5]; 496d4002b98SHong Zhang z[8*i+6] = sum[6]; 497d4002b98SHong Zhang z[8*i+7] = sum[7]; 498d4002b98SHong Zhang } 499d4002b98SHong Zhang #endif 500d4002b98SHong Zhang } 501d4002b98SHong Zhang 502d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 503d4002b98SHong Zhang ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 504d4002b98SHong Zhang ierr = VecRestoreArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 505d4002b98SHong Zhang PetscFunctionReturn(0); 506d4002b98SHong Zhang } 507d4002b98SHong Zhang 508d4002b98SHong Zhang PetscErrorCode MatMultTransposeAdd_SeqSELL(Mat A,Vec xx,Vec zz,Vec yy) 509d4002b98SHong Zhang { 510d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 511d4002b98SHong Zhang PetscScalar *y; 512d4002b98SHong Zhang const PetscScalar *x; 513d4002b98SHong Zhang const MatScalar *aval=a->val; 514d4002b98SHong Zhang const PetscInt *acolidx=a->colidx; 515d4002b98SHong Zhang PetscInt i,j,row; 516d4002b98SHong Zhang PetscBool isnonzero; 517d4002b98SHong Zhang PetscErrorCode ierr; 518d4002b98SHong Zhang 519d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 520d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval) 521d4002b98SHong Zhang #endif 522d4002b98SHong Zhang 523d4002b98SHong Zhang PetscFunctionBegin; 524d4002b98SHong Zhang if (zz != yy) { ierr = VecCopy(zz,yy);CHKERRQ(ierr); } 525d4002b98SHong Zhang ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 526d4002b98SHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 527d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 528d4002b98SHong Zhang for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 529d4002b98SHong Zhang isnonzero = (PetscBool)((j-a->sliidx[i])/8 < a->rlen[8*i+row]); 530d4002b98SHong Zhang if (isnonzero) y[acolidx[j]] += aval[j]*x[8*i+row]; 531d4002b98SHong Zhang } 532d4002b98SHong Zhang } 533d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->sliidx[a->totalslices]);CHKERRQ(ierr); 534d4002b98SHong Zhang ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 535d4002b98SHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 536d4002b98SHong Zhang PetscFunctionReturn(0); 537d4002b98SHong Zhang } 538d4002b98SHong Zhang 539d4002b98SHong Zhang PetscErrorCode MatMultTranspose_SeqSELL(Mat A,Vec xx,Vec yy) 540d4002b98SHong Zhang { 541d4002b98SHong Zhang PetscErrorCode ierr; 542d4002b98SHong Zhang 543d4002b98SHong Zhang PetscFunctionBegin; 544d4002b98SHong Zhang ierr = VecSet(yy,0.0);CHKERRQ(ierr); 545d4002b98SHong Zhang ierr = MatMultTransposeAdd_SeqSELL(A,xx,yy,yy);CHKERRQ(ierr); 546d4002b98SHong Zhang PetscFunctionReturn(0); 547d4002b98SHong Zhang } 548d4002b98SHong Zhang 549d4002b98SHong Zhang /* 550d4002b98SHong Zhang Checks for missing diagonals 551d4002b98SHong Zhang */ 552d4002b98SHong Zhang PetscErrorCode MatMissingDiagonal_SeqSELL(Mat A,PetscBool *missing,PetscInt *d) 553d4002b98SHong Zhang { 554d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 555d4002b98SHong Zhang PetscInt *diag,i; 556d4002b98SHong Zhang 557d4002b98SHong Zhang PetscFunctionBegin; 558d4002b98SHong Zhang *missing = PETSC_FALSE; 559d4002b98SHong Zhang if (A->rmap->n > 0 && !(a->colidx)) { 560d4002b98SHong Zhang *missing = PETSC_TRUE; 561d4002b98SHong Zhang if (d) *d = 0; 562d4002b98SHong Zhang PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n"); 563d4002b98SHong Zhang } else { 564d4002b98SHong Zhang diag = a->diag; 565d4002b98SHong Zhang for (i=0; i<A->rmap->n; i++) { 566d4002b98SHong Zhang if (diag[i] == -1) { 567d4002b98SHong Zhang *missing = PETSC_TRUE; 568d4002b98SHong Zhang if (d) *d = i; 569d4002b98SHong Zhang PetscInfo1(A,"Matrix is missing diagonal number %D\n",i); 570d4002b98SHong Zhang break; 571d4002b98SHong Zhang } 572d4002b98SHong Zhang } 573d4002b98SHong Zhang } 574d4002b98SHong Zhang PetscFunctionReturn(0); 575d4002b98SHong Zhang } 576d4002b98SHong Zhang 577d4002b98SHong Zhang PetscErrorCode MatMarkDiagonal_SeqSELL(Mat A) 578d4002b98SHong Zhang { 579d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 580d4002b98SHong Zhang PetscInt i,j,m=A->rmap->n,shift; 581d4002b98SHong Zhang PetscErrorCode ierr; 582d4002b98SHong Zhang 583d4002b98SHong Zhang PetscFunctionBegin; 584d4002b98SHong Zhang if (!a->diag) { 585d4002b98SHong Zhang ierr = PetscMalloc1(m,&a->diag);CHKERRQ(ierr); 586d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)A,m*sizeof(PetscInt));CHKERRQ(ierr); 587d4002b98SHong Zhang a->free_diag = PETSC_TRUE; 588d4002b98SHong Zhang } 589d4002b98SHong Zhang for (i=0; i<m; i++) { /* loop over rows */ 590d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 591d4002b98SHong Zhang a->diag[i] = -1; 592d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 593d4002b98SHong Zhang if (a->colidx[shift+j*8] == i) { 594d4002b98SHong Zhang a->diag[i] = shift+j*8; 595d4002b98SHong Zhang break; 596d4002b98SHong Zhang } 597d4002b98SHong Zhang } 598d4002b98SHong Zhang } 599d4002b98SHong Zhang PetscFunctionReturn(0); 600d4002b98SHong Zhang } 601d4002b98SHong Zhang 602d4002b98SHong Zhang /* 603d4002b98SHong Zhang Negative shift indicates do not generate an error if there is a zero diagonal, just invert it anyways 604d4002b98SHong Zhang */ 605d4002b98SHong Zhang PetscErrorCode MatInvertDiagonal_SeqSELL(Mat A,PetscScalar omega,PetscScalar fshift) 606d4002b98SHong Zhang { 607d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*) A->data; 608d4002b98SHong Zhang PetscInt i,*diag,m = A->rmap->n; 609d4002b98SHong Zhang MatScalar *val = a->val; 610d4002b98SHong Zhang PetscScalar *idiag,*mdiag; 611d4002b98SHong Zhang PetscErrorCode ierr; 612d4002b98SHong Zhang 613d4002b98SHong Zhang PetscFunctionBegin; 614d4002b98SHong Zhang if (a->idiagvalid) PetscFunctionReturn(0); 615d4002b98SHong Zhang ierr = MatMarkDiagonal_SeqSELL(A);CHKERRQ(ierr); 616d4002b98SHong Zhang diag = a->diag; 617d4002b98SHong Zhang if (!a->idiag) { 618d4002b98SHong Zhang ierr = PetscMalloc3(m,&a->idiag,m,&a->mdiag,m,&a->ssor_work);CHKERRQ(ierr); 619d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)A, 3*m*sizeof(PetscScalar));CHKERRQ(ierr); 620d4002b98SHong Zhang val = a->val; 621d4002b98SHong Zhang } 622d4002b98SHong Zhang mdiag = a->mdiag; 623d4002b98SHong Zhang idiag = a->idiag; 624d4002b98SHong Zhang 625d4002b98SHong Zhang if (omega == 1.0 && PetscRealPart(fshift) <= 0.0) { 626d4002b98SHong Zhang for (i=0; i<m; i++) { 627d4002b98SHong Zhang mdiag[i] = val[diag[i]]; 628d4002b98SHong Zhang if (!PetscAbsScalar(mdiag[i])) { /* zero diagonal */ 629d4002b98SHong Zhang if (PetscRealPart(fshift)) { 630d4002b98SHong Zhang ierr = PetscInfo1(A,"Zero diagonal on row %D\n",i);CHKERRQ(ierr); 631d4002b98SHong Zhang A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 632d4002b98SHong Zhang A->factorerror_zeropivot_value = 0.0; 633d4002b98SHong Zhang A->factorerror_zeropivot_row = i; 634d4002b98SHong Zhang } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Zero diagonal on row %D",i); 635d4002b98SHong Zhang } 636d4002b98SHong Zhang idiag[i] = 1.0/val[diag[i]]; 637d4002b98SHong Zhang } 638d4002b98SHong Zhang ierr = PetscLogFlops(m);CHKERRQ(ierr); 639d4002b98SHong Zhang } else { 640d4002b98SHong Zhang for (i=0; i<m; i++) { 641d4002b98SHong Zhang mdiag[i] = val[diag[i]]; 642d4002b98SHong Zhang idiag[i] = omega/(fshift + val[diag[i]]); 643d4002b98SHong Zhang } 644d4002b98SHong Zhang ierr = PetscLogFlops(2.0*m);CHKERRQ(ierr); 645d4002b98SHong Zhang } 646d4002b98SHong Zhang a->idiagvalid = PETSC_TRUE; 647d4002b98SHong Zhang PetscFunctionReturn(0); 648d4002b98SHong Zhang } 649d4002b98SHong Zhang 650d4002b98SHong Zhang PetscErrorCode MatZeroEntries_SeqSELL(Mat A) 651d4002b98SHong Zhang { 652d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 653d4002b98SHong Zhang PetscErrorCode ierr; 654d4002b98SHong Zhang 655d4002b98SHong Zhang PetscFunctionBegin; 656d4002b98SHong Zhang ierr = PetscMemzero(a->val,(a->sliidx[a->totalslices])*sizeof(PetscScalar));CHKERRQ(ierr); 657d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr); 658d4002b98SHong Zhang PetscFunctionReturn(0); 659d4002b98SHong Zhang } 660d4002b98SHong Zhang 661d4002b98SHong Zhang PetscErrorCode MatDestroy_SeqSELL(Mat A) 662d4002b98SHong Zhang { 663d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 664d4002b98SHong Zhang PetscErrorCode ierr; 665d4002b98SHong Zhang 666d4002b98SHong Zhang PetscFunctionBegin; 667d4002b98SHong Zhang #if defined(PETSC_USE_LOG) 668d4002b98SHong Zhang PetscLogObjectState((PetscObject)A,"Rows=%D, Cols=%D, NZ=%D",A->rmap->n,A->cmap->n,a->nz); 669d4002b98SHong Zhang #endif 670d4002b98SHong Zhang ierr = MatSeqXSELLFreeSELL(A,&a->val,&a->colidx);CHKERRQ(ierr); 671d4002b98SHong Zhang ierr = ISDestroy(&a->row);CHKERRQ(ierr); 672d4002b98SHong Zhang ierr = ISDestroy(&a->col);CHKERRQ(ierr); 673d4002b98SHong Zhang ierr = PetscFree(a->diag);CHKERRQ(ierr); 674d4002b98SHong Zhang ierr = PetscFree(a->ibdiag);CHKERRQ(ierr); 675d4002b98SHong Zhang ierr = PetscFree(a->rlen);CHKERRQ(ierr); 676d4002b98SHong Zhang ierr = PetscFree(a->sliidx);CHKERRQ(ierr); 677d4002b98SHong Zhang ierr = PetscFree3(a->idiag,a->mdiag,a->ssor_work);CHKERRQ(ierr); 678d4002b98SHong Zhang ierr = PetscFree(a->solve_work);CHKERRQ(ierr); 679d4002b98SHong Zhang ierr = ISDestroy(&a->icol);CHKERRQ(ierr); 680d4002b98SHong Zhang ierr = PetscFree(a->saved_values);CHKERRQ(ierr); 681d4002b98SHong Zhang 682d4002b98SHong Zhang ierr = PetscFree(A->data);CHKERRQ(ierr); 683d4002b98SHong Zhang 684d4002b98SHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 685d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr); 686d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr); 687d4002b98SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 688d4002b98SHong Zhang #endif 689d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSELLSetPreallocation_C",NULL);CHKERRQ(ierr); 690d4002b98SHong Zhang PetscFunctionReturn(0); 691d4002b98SHong Zhang } 692d4002b98SHong Zhang 693d4002b98SHong Zhang PetscErrorCode MatSetOption_SeqSELL(Mat A,MatOption op,PetscBool flg) 694d4002b98SHong Zhang { 695d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 696d4002b98SHong Zhang PetscErrorCode ierr; 697d4002b98SHong Zhang 698d4002b98SHong Zhang PetscFunctionBegin; 699d4002b98SHong Zhang switch (op) { 700d4002b98SHong Zhang case MAT_ROW_ORIENTED: 701d4002b98SHong Zhang a->roworiented = flg; 702d4002b98SHong Zhang break; 703d4002b98SHong Zhang case MAT_KEEP_NONZERO_PATTERN: 704d4002b98SHong Zhang a->keepnonzeropattern = flg; 705d4002b98SHong Zhang break; 706d4002b98SHong Zhang case MAT_NEW_NONZERO_LOCATIONS: 707d4002b98SHong Zhang a->nonew = (flg ? 0 : 1); 708d4002b98SHong Zhang break; 709d4002b98SHong Zhang case MAT_NEW_NONZERO_LOCATION_ERR: 710d4002b98SHong Zhang a->nonew = (flg ? -1 : 0); 711d4002b98SHong Zhang break; 712d4002b98SHong Zhang case MAT_NEW_NONZERO_ALLOCATION_ERR: 713d4002b98SHong Zhang a->nonew = (flg ? -2 : 0); 714d4002b98SHong Zhang break; 715d4002b98SHong Zhang case MAT_UNUSED_NONZERO_LOCATION_ERR: 716d4002b98SHong Zhang a->nounused = (flg ? -1 : 0); 717d4002b98SHong Zhang break; 718d4002b98SHong Zhang case MAT_NEW_DIAGONALS: 719d4002b98SHong Zhang case MAT_IGNORE_OFF_PROC_ENTRIES: 720d4002b98SHong Zhang case MAT_USE_HASH_TABLE: 721d4002b98SHong Zhang ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 722d4002b98SHong Zhang break; 723d4002b98SHong Zhang case MAT_SPD: 724d4002b98SHong Zhang case MAT_SYMMETRIC: 725d4002b98SHong Zhang case MAT_STRUCTURALLY_SYMMETRIC: 726d4002b98SHong Zhang case MAT_HERMITIAN: 727d4002b98SHong Zhang case MAT_SYMMETRY_ETERNAL: 728d4002b98SHong Zhang /* These options are handled directly by MatSetOption() */ 729d4002b98SHong Zhang break; 730d4002b98SHong Zhang default: 731d4002b98SHong Zhang SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 732d4002b98SHong Zhang } 733d4002b98SHong Zhang PetscFunctionReturn(0); 734d4002b98SHong Zhang } 735d4002b98SHong Zhang 736d4002b98SHong Zhang PetscErrorCode MatGetDiagonal_SeqSELL(Mat A,Vec v) 737d4002b98SHong Zhang { 738d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 739d4002b98SHong Zhang PetscInt i,j,n,shift; 740d4002b98SHong Zhang PetscScalar *x,zero=0.0; 741d4002b98SHong Zhang PetscErrorCode ierr; 742d4002b98SHong Zhang 743d4002b98SHong Zhang PetscFunctionBegin; 744d4002b98SHong Zhang ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 745d4002b98SHong Zhang if (n != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 746d4002b98SHong Zhang 747d4002b98SHong Zhang if (A->factortype == MAT_FACTOR_ILU || A->factortype == MAT_FACTOR_LU) { 748d4002b98SHong Zhang PetscInt *diag=a->diag; 749d4002b98SHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 750d4002b98SHong Zhang for (i=0; i<n; i++) x[i] = 1.0/a->val[diag[i]]; 751d4002b98SHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 752d4002b98SHong Zhang PetscFunctionReturn(0); 753d4002b98SHong Zhang } 754d4002b98SHong Zhang 755d4002b98SHong Zhang ierr = VecSet(v,zero);CHKERRQ(ierr); 756d4002b98SHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 757d4002b98SHong Zhang for (i=0; i<n; i++) { /* loop over rows */ 758d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 759d4002b98SHong Zhang x[i] = 0; 760d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 761d4002b98SHong Zhang if (a->colidx[shift+j*8] == i) { 762d4002b98SHong Zhang x[i] = a->val[shift+j*8]; 763d4002b98SHong Zhang break; 764d4002b98SHong Zhang } 765d4002b98SHong Zhang } 766d4002b98SHong Zhang } 767d4002b98SHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 768d4002b98SHong Zhang PetscFunctionReturn(0); 769d4002b98SHong Zhang } 770d4002b98SHong Zhang 771d4002b98SHong Zhang PetscErrorCode MatDiagonalScale_SeqSELL(Mat A,Vec ll,Vec rr) 772d4002b98SHong Zhang { 773d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 774d4002b98SHong Zhang const PetscScalar *l,*r; 775d4002b98SHong Zhang PetscInt i,j,m,n,row; 776d4002b98SHong Zhang PetscErrorCode ierr; 777d4002b98SHong Zhang 778d4002b98SHong Zhang PetscFunctionBegin; 779d4002b98SHong Zhang if (ll) { 780d4002b98SHong Zhang /* The local size is used so that VecMPI can be passed to this routine 781d4002b98SHong Zhang by MatDiagonalScale_MPISELL */ 782d4002b98SHong Zhang ierr = VecGetLocalSize(ll,&m);CHKERRQ(ierr); 783d4002b98SHong Zhang if (m != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length"); 784d4002b98SHong Zhang ierr = VecGetArrayRead(ll,&l);CHKERRQ(ierr); 785d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 786d4002b98SHong Zhang for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 787d4002b98SHong Zhang a->val[j] *= l[8*i+row]; 788d4002b98SHong Zhang } 789d4002b98SHong Zhang } 790d4002b98SHong Zhang ierr = VecRestoreArrayRead(ll,&l);CHKERRQ(ierr); 791d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 792d4002b98SHong Zhang } 793d4002b98SHong Zhang if (rr) { 794d4002b98SHong Zhang ierr = VecGetLocalSize(rr,&n);CHKERRQ(ierr); 795d4002b98SHong Zhang if (n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length"); 796d4002b98SHong Zhang ierr = VecGetArrayRead(rr,&r);CHKERRQ(ierr); 797d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 798d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j++) { 799d4002b98SHong Zhang a->val[j] *= r[a->colidx[j]]; 800d4002b98SHong Zhang } 801d4002b98SHong Zhang } 802d4002b98SHong Zhang ierr = VecRestoreArrayRead(rr,&r);CHKERRQ(ierr); 803d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 804d4002b98SHong Zhang } 805d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr); 806d4002b98SHong Zhang PetscFunctionReturn(0); 807d4002b98SHong Zhang } 808d4002b98SHong Zhang 809d4002b98SHong Zhang extern PetscErrorCode MatSetValues_SeqSELL(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 810d4002b98SHong Zhang 811d4002b98SHong Zhang PetscErrorCode MatGetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 812d4002b98SHong Zhang { 813d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 814d4002b98SHong Zhang PetscInt *cp,i,k,low,high,t,row,col,l; 815d4002b98SHong Zhang PetscInt shift; 816d4002b98SHong Zhang MatScalar *vp; 817d4002b98SHong Zhang 818d4002b98SHong Zhang PetscFunctionBegin; 819d4002b98SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 820d4002b98SHong Zhang row = im[k]; 821d4002b98SHong Zhang if (row<0) continue; 822d4002b98SHong Zhang #if defined(PETSC_USE_DEBUG) 823d4002b98SHong Zhang if (row >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->n-1); 824d4002b98SHong Zhang #endif 825d4002b98SHong Zhang shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */ 826d4002b98SHong Zhang cp = a->colidx+shift; /* pointer to the row */ 827d4002b98SHong Zhang vp = a->val+shift; /* pointer to the row */ 828d4002b98SHong Zhang for (l=0; l<n; l++) { /* loop over added rows */ 829d4002b98SHong Zhang col = in[l]; 830d4002b98SHong Zhang if (col<0) continue; 831d4002b98SHong Zhang #if defined(PETSC_USE_DEBUG) 832d4002b98SHong Zhang if (col >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: row %D max %D",col,A->cmap->n-1); 833d4002b98SHong Zhang #endif 834d4002b98SHong Zhang high = a->rlen[row]; low = 0; /* assume unsorted */ 835d4002b98SHong Zhang while (high-low > 5) { 836d4002b98SHong Zhang t = (low+high)/2; 837d4002b98SHong Zhang if (*(cp+t*8) > col) high = t; 838d4002b98SHong Zhang else low = t; 839d4002b98SHong Zhang } 840d4002b98SHong Zhang for (i=low; i<high; i++) { 841d4002b98SHong Zhang if (*(cp+8*i) > col) break; 842d4002b98SHong Zhang if (*(cp+8*i) == col) { 843d4002b98SHong Zhang *v++ = *(vp+8*i); 844d4002b98SHong Zhang goto finished; 845d4002b98SHong Zhang } 846d4002b98SHong Zhang } 847d4002b98SHong Zhang *v++ = 0.0; 848d4002b98SHong Zhang finished:; 849d4002b98SHong Zhang } 850d4002b98SHong Zhang } 851d4002b98SHong Zhang PetscFunctionReturn(0); 852d4002b98SHong Zhang } 853d4002b98SHong Zhang 854d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_ASCII(Mat A,PetscViewer viewer) 855d4002b98SHong Zhang { 856d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 857d4002b98SHong Zhang PetscInt i,j,m=A->rmap->n,shift; 858d4002b98SHong Zhang const char *name; 859d4002b98SHong Zhang PetscViewerFormat format; 860d4002b98SHong Zhang PetscErrorCode ierr; 861d4002b98SHong Zhang 862d4002b98SHong Zhang PetscFunctionBegin; 863d4002b98SHong Zhang ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 864d4002b98SHong Zhang if (format == PETSC_VIEWER_ASCII_MATLAB) { 865d4002b98SHong Zhang PetscInt nofinalvalue = 0; 866d4002b98SHong Zhang /* 867d4002b98SHong Zhang if (m && ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != A->cmap->n-1))) { 868d4002b98SHong Zhang nofinalvalue = 1; 869d4002b98SHong Zhang } 870d4002b98SHong Zhang */ 871d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 872d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %D %D \n",m,A->cmap->n);CHKERRQ(ierr); 873d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%% Nonzeros = %D \n",a->nz);CHKERRQ(ierr); 874d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 875d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%D,4);\n",a->nz+nofinalvalue);CHKERRQ(ierr); 876d4002b98SHong Zhang #else 877d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%D,3);\n",a->nz+nofinalvalue);CHKERRQ(ierr); 878d4002b98SHong Zhang #endif 879d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"zzz = [\n");CHKERRQ(ierr); 880d4002b98SHong Zhang 881d4002b98SHong Zhang for (i=0; i<m; i++) { 882d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 883d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 884d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 885d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%D %D %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); 886d4002b98SHong Zhang #else 887d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%D %D %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)a->val[shift+8*j]);CHKERRQ(ierr); 888d4002b98SHong Zhang #endif 889d4002b98SHong Zhang } 890d4002b98SHong Zhang } 891d4002b98SHong Zhang /* 892d4002b98SHong Zhang if (nofinalvalue) { 893d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 894d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%D %D %18.16e %18.16e\n",m,A->cmap->n,0.,0.);CHKERRQ(ierr); 895d4002b98SHong Zhang #else 896d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%D %D %18.16e\n",m,A->cmap->n,0.0);CHKERRQ(ierr); 897d4002b98SHong Zhang #endif 898d4002b98SHong Zhang } 899d4002b98SHong Zhang */ 900d4002b98SHong Zhang ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr); 901d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"];\n %s = spconvert(zzz);\n",name);CHKERRQ(ierr); 902d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 903d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO || format == PETSC_VIEWER_ASCII_INFO) { 904d4002b98SHong Zhang PetscFunctionReturn(0); 905d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_COMMON) { 906d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 907d4002b98SHong Zhang for (i=0; i<m; i++) { 908d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 909d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 910d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 911d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 912d4002b98SHong Zhang if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) { 913d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 914d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) { 915d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 916d4002b98SHong Zhang } else if (PetscRealPart(a->val[shift+8*j]) != 0.0) { 917d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]));CHKERRQ(ierr); 918d4002b98SHong Zhang } 919d4002b98SHong Zhang #else 920d4002b98SHong Zhang if (a->val[shift+8*j] != 0.0) {ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]);CHKERRQ(ierr);} 921d4002b98SHong Zhang #endif 922d4002b98SHong Zhang } 923d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 924d4002b98SHong Zhang } 925d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 926d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_DENSE) { 927d4002b98SHong Zhang PetscInt cnt=0,jcnt; 928d4002b98SHong Zhang PetscScalar value; 929d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 930d4002b98SHong Zhang PetscBool realonly=PETSC_TRUE; 931d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) { 932d4002b98SHong Zhang if (PetscImaginaryPart(a->val[i]) != 0.0) { 933d4002b98SHong Zhang realonly = PETSC_FALSE; 934d4002b98SHong Zhang break; 935d4002b98SHong Zhang } 936d4002b98SHong Zhang } 937d4002b98SHong Zhang #endif 938d4002b98SHong Zhang 939d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 940d4002b98SHong Zhang for (i=0; i<m; i++) { 941d4002b98SHong Zhang jcnt = 0; 942d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 943d4002b98SHong Zhang for (j=0; j<A->cmap->n; j++) { 944d4002b98SHong Zhang if (jcnt < a->rlen[i] && j == a->colidx[shift+8*j]) { 945d4002b98SHong Zhang value = a->val[cnt++]; 946d4002b98SHong Zhang jcnt++; 947d4002b98SHong Zhang } else { 948d4002b98SHong Zhang value = 0.0; 949d4002b98SHong Zhang } 950d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 951d4002b98SHong Zhang if (realonly) { 952d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)PetscRealPart(value));CHKERRQ(ierr); 953d4002b98SHong Zhang } else { 954d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," %7.5e+%7.5e i ",(double)PetscRealPart(value),(double)PetscImaginaryPart(value));CHKERRQ(ierr); 955d4002b98SHong Zhang } 956d4002b98SHong Zhang #else 957d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)value);CHKERRQ(ierr); 958d4002b98SHong Zhang #endif 959d4002b98SHong Zhang } 960d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 961d4002b98SHong Zhang } 962d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 963d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_MATRIXMARKET) { 964d4002b98SHong Zhang PetscInt fshift=1; 965d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 966d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 967d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate complex general\n");CHKERRQ(ierr); 968d4002b98SHong Zhang #else 969d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate real general\n");CHKERRQ(ierr); 970d4002b98SHong Zhang #endif 971d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%D %D %D\n", m, A->cmap->n, a->nz);CHKERRQ(ierr); 972d4002b98SHong Zhang for (i=0; i<m; i++) { 973d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 974d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 975d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 976d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%D %D %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); 977d4002b98SHong Zhang #else 978d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"%D %D %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)a->val[shift+8*j]);CHKERRQ(ierr); 979d4002b98SHong Zhang #endif 980d4002b98SHong Zhang } 981d4002b98SHong Zhang } 982d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 983d4002b98SHong Zhang } else { 984d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 985d4002b98SHong Zhang if (A->factortype) { 986d4002b98SHong Zhang for (i=0; i<m; i++) { 987d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 988d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 989d4002b98SHong Zhang /* L part */ 990d4002b98SHong Zhang for (j=shift; j<a->diag[i]; j+=8) { 991d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 992d4002b98SHong Zhang if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0) { 993d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr); 994d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0) { 995d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])));CHKERRQ(ierr); 996d4002b98SHong Zhang } else { 997d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr); 998d4002b98SHong Zhang } 999d4002b98SHong Zhang #else 1000d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[j],(double)a->val[j]);CHKERRQ(ierr); 1001d4002b98SHong Zhang #endif 1002d4002b98SHong Zhang } 1003d4002b98SHong Zhang /* diagonal */ 1004d4002b98SHong Zhang j = a->diag[i]; 1005d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1006d4002b98SHong Zhang if (PetscImaginaryPart(a->val[j]) > 0.0) { 1007d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)PetscImaginaryPart(1.0/a->val[j]));CHKERRQ(ierr); 1008d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 1009d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)(-PetscImaginaryPart(1.0/a->val[j])));CHKERRQ(ierr); 1010d4002b98SHong Zhang } else { 1011d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]));CHKERRQ(ierr); 1012d4002b98SHong Zhang } 1013d4002b98SHong Zhang #else 1014d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[j],(double)(1.0/a->val[j]));CHKERRQ(ierr); 1015d4002b98SHong Zhang #endif 1016d4002b98SHong Zhang 1017d4002b98SHong Zhang /* U part */ 1018d4002b98SHong Zhang for (j=a->diag[i]+1; j<shift+8*a->rlen[i]; j+=8) { 1019d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1020d4002b98SHong Zhang if (PetscImaginaryPart(a->val[j]) > 0.0) { 1021d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr); 1022d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 1023d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])));CHKERRQ(ierr); 1024d4002b98SHong Zhang } else { 1025d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr); 1026d4002b98SHong Zhang } 1027d4002b98SHong Zhang #else 1028d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[j],(double)a->val[j]);CHKERRQ(ierr); 1029d4002b98SHong Zhang #endif 1030d4002b98SHong Zhang } 1031d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 1032d4002b98SHong Zhang } 1033d4002b98SHong Zhang } else { 1034d4002b98SHong Zhang for (i=0; i<m; i++) { 1035d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1036d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 1037d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1038d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1039d4002b98SHong Zhang if (PetscImaginaryPart(a->val[j]) > 0.0) { 1040d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 1041d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 1042d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr); 1043d4002b98SHong Zhang } else { 1044d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]));CHKERRQ(ierr); 1045d4002b98SHong Zhang } 1046d4002b98SHong Zhang #else 1047d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]);CHKERRQ(ierr); 1048d4002b98SHong Zhang #endif 1049d4002b98SHong Zhang } 1050d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 1051d4002b98SHong Zhang } 1052d4002b98SHong Zhang } 1053d4002b98SHong Zhang ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 1054d4002b98SHong Zhang } 1055d4002b98SHong Zhang ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 1056d4002b98SHong Zhang PetscFunctionReturn(0); 1057d4002b98SHong Zhang } 1058d4002b98SHong Zhang 1059d4002b98SHong Zhang #include <petscdraw.h> 1060d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw_Zoom(PetscDraw draw,void *Aa) 1061d4002b98SHong Zhang { 1062d4002b98SHong Zhang Mat A=(Mat)Aa; 1063d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1064d4002b98SHong Zhang PetscInt i,j,m=A->rmap->n,shift; 1065d4002b98SHong Zhang int color; 1066d4002b98SHong Zhang PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 1067d4002b98SHong Zhang PetscViewer viewer; 1068d4002b98SHong Zhang PetscViewerFormat format; 1069d4002b98SHong Zhang PetscErrorCode ierr; 1070d4002b98SHong Zhang 1071d4002b98SHong Zhang PetscFunctionBegin; 1072d4002b98SHong Zhang ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 1073d4002b98SHong Zhang ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 1074d4002b98SHong Zhang ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 1075d4002b98SHong Zhang 1076d4002b98SHong Zhang /* loop over matrix elements drawing boxes */ 1077d4002b98SHong Zhang 1078d4002b98SHong Zhang if (format != PETSC_VIEWER_DRAW_CONTOUR) { 1079d4002b98SHong Zhang ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 1080d4002b98SHong Zhang /* Blue for negative, Cyan for zero and Red for positive */ 1081d4002b98SHong Zhang color = PETSC_DRAW_BLUE; 1082d4002b98SHong Zhang for (i=0; i<m; i++) { 1083d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1084d4002b98SHong Zhang y_l = m - i - 1.0; y_r = y_l + 1.0; 1085d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1086d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0; 1087d4002b98SHong Zhang if (PetscRealPart(a->val[shift+8*j]) >= 0.) continue; 1088d4002b98SHong Zhang ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1089d4002b98SHong Zhang } 1090d4002b98SHong Zhang } 1091d4002b98SHong Zhang color = PETSC_DRAW_CYAN; 1092d4002b98SHong Zhang for (i=0; i<m; i++) { 1093d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1094d4002b98SHong Zhang y_l = m - i - 1.0; y_r = y_l + 1.0; 1095d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1096d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0; 1097d4002b98SHong Zhang if (a->val[shift+8*j] != 0.) continue; 1098d4002b98SHong Zhang ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1099d4002b98SHong Zhang } 1100d4002b98SHong Zhang } 1101d4002b98SHong Zhang color = PETSC_DRAW_RED; 1102d4002b98SHong Zhang for (i=0; i<m; i++) { 1103d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1104d4002b98SHong Zhang y_l = m - i - 1.0; y_r = y_l + 1.0; 1105d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1106d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0; 1107d4002b98SHong Zhang if (PetscRealPart(a->val[shift+8*j]) <= 0.) continue; 1108d4002b98SHong Zhang ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1109d4002b98SHong Zhang } 1110d4002b98SHong Zhang } 1111d4002b98SHong Zhang ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 1112d4002b98SHong Zhang } else { 1113d4002b98SHong Zhang /* use contour shading to indicate magnitude of values */ 1114d4002b98SHong Zhang /* first determine max of all nonzero values */ 1115d4002b98SHong Zhang PetscReal minv=0.0,maxv=0.0; 1116d4002b98SHong Zhang PetscInt count=0; 1117d4002b98SHong Zhang PetscDraw popup; 1118d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) { 1119d4002b98SHong Zhang if (PetscAbsScalar(a->val[i]) > maxv) maxv = PetscAbsScalar(a->val[i]); 1120d4002b98SHong Zhang } 1121d4002b98SHong Zhang if (minv >= maxv) maxv = minv + PETSC_SMALL; 1122d4002b98SHong Zhang ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr); 1123d4002b98SHong Zhang ierr = PetscDrawScalePopup(popup,minv,maxv);CHKERRQ(ierr); 1124d4002b98SHong Zhang 1125d4002b98SHong Zhang ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 1126d4002b98SHong Zhang for (i=0; i<m; i++) { 1127d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1128d4002b98SHong Zhang y_l = m - i - 1.0; 1129d4002b98SHong Zhang y_r = y_l + 1.0; 1130d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1131d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; 1132d4002b98SHong Zhang x_r = x_l + 1.0; 1133d4002b98SHong Zhang color = PetscDrawRealToColor(PetscAbsScalar(a->val[count]),minv,maxv); 1134d4002b98SHong Zhang ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1135d4002b98SHong Zhang count++; 1136d4002b98SHong Zhang } 1137d4002b98SHong Zhang } 1138d4002b98SHong Zhang ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 1139d4002b98SHong Zhang } 1140d4002b98SHong Zhang PetscFunctionReturn(0); 1141d4002b98SHong Zhang } 1142d4002b98SHong Zhang 1143d4002b98SHong Zhang #include <petscdraw.h> 1144d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw(Mat A,PetscViewer viewer) 1145d4002b98SHong Zhang { 1146d4002b98SHong Zhang PetscDraw draw; 1147d4002b98SHong Zhang PetscReal xr,yr,xl,yl,h,w; 1148d4002b98SHong Zhang PetscBool isnull; 1149d4002b98SHong Zhang PetscErrorCode ierr; 1150d4002b98SHong Zhang 1151d4002b98SHong Zhang PetscFunctionBegin; 1152d4002b98SHong Zhang ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 1153d4002b98SHong Zhang ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 1154d4002b98SHong Zhang if (isnull) PetscFunctionReturn(0); 1155d4002b98SHong Zhang 1156d4002b98SHong Zhang xr = A->cmap->n; yr = A->rmap->n; h = yr/10.0; w = xr/10.0; 1157d4002b98SHong Zhang xr += w; yr += h; xl = -w; yl = -h; 1158d4002b98SHong Zhang ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 1159d4002b98SHong Zhang ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 1160d4002b98SHong Zhang ierr = PetscDrawZoom(draw,MatView_SeqSELL_Draw_Zoom,A);CHKERRQ(ierr); 1161d4002b98SHong Zhang ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr); 1162d4002b98SHong Zhang ierr = PetscDrawSave(draw);CHKERRQ(ierr); 1163d4002b98SHong Zhang PetscFunctionReturn(0); 1164d4002b98SHong Zhang } 1165d4002b98SHong Zhang 1166d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL(Mat A,PetscViewer viewer) 1167d4002b98SHong Zhang { 1168d4002b98SHong Zhang PetscBool iascii,isbinary,isdraw; 1169d4002b98SHong Zhang PetscErrorCode ierr; 1170d4002b98SHong Zhang 1171d4002b98SHong Zhang PetscFunctionBegin; 1172d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 1173d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 1174d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 1175d4002b98SHong Zhang if (iascii) { 1176d4002b98SHong Zhang ierr = MatView_SeqSELL_ASCII(A,viewer);CHKERRQ(ierr); 1177d4002b98SHong Zhang } else if (isbinary) { 1178d4002b98SHong Zhang /* ierr = MatView_SeqSELL_Binary(A,viewer);CHKERRQ(ierr); */ 1179d4002b98SHong Zhang } else if (isdraw) { 1180d4002b98SHong Zhang ierr = MatView_SeqSELL_Draw(A,viewer);CHKERRQ(ierr); 1181d4002b98SHong Zhang } 1182d4002b98SHong Zhang PetscFunctionReturn(0); 1183d4002b98SHong Zhang } 1184d4002b98SHong Zhang 1185d4002b98SHong Zhang PetscErrorCode MatAssemblyEnd_SeqSELL(Mat A,MatAssemblyType mode) 1186d4002b98SHong Zhang { 1187d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1188d4002b98SHong Zhang PetscInt i,shift,row_in_slice,row,nrow,*cp,lastcol,j,k; 1189d4002b98SHong Zhang MatScalar *vp; 1190d4002b98SHong Zhang PetscErrorCode ierr; 1191d4002b98SHong Zhang 1192d4002b98SHong Zhang PetscFunctionBegin; 1193d4002b98SHong Zhang if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 1194d4002b98SHong Zhang /* To do: compress out the unused elements */ 1195d4002b98SHong Zhang ierr = MatMarkDiagonal_SeqSELL(A);CHKERRQ(ierr); 1196d4002b98SHong Zhang ierr = PetscInfo6(A,"Matrix size: %D X %D; storage space: %D allocated %D used (%D nonzeros+%D paddedzeros)\n",A->rmap->n,A->cmap->n,a->maxallocmat,a->sliidx[a->totalslices],a->nz,a->sliidx[a->totalslices]-a->nz);CHKERRQ(ierr); 1197d4002b98SHong Zhang ierr = PetscInfo1(A,"Number of mallocs during MatSetValues() is %D\n",a->reallocs);CHKERRQ(ierr); 1198d4002b98SHong Zhang ierr = PetscInfo1(A,"Maximum nonzeros in any row is %D\n",a->rlenmax);CHKERRQ(ierr); 1199d4002b98SHong 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 */ 1200d4002b98SHong Zhang for (i=0; i<a->totalslices; ++i) { 1201d4002b98SHong Zhang shift = a->sliidx[i]; /* starting index of the slice */ 1202d4002b98SHong Zhang cp = a->colidx+shift; /* pointer to the column indices of the slice */ 1203d4002b98SHong Zhang vp = a->val+shift; /* pointer to the nonzero values of the slice */ 1204d4002b98SHong Zhang for (row_in_slice=0; row_in_slice<8; ++row_in_slice) { /* loop over rows in the slice */ 1205d4002b98SHong Zhang row = 8*i + row_in_slice; 1206d4002b98SHong Zhang nrow = a->rlen[row]; /* number of nonzeros in row */ 1207d4002b98SHong Zhang /* 1208d4002b98SHong Zhang Search for the nearest nonzero. Normally setting the index to zero may cause extra communication. 1209d4002b98SHong Zhang But if the entire slice are empty, it is fine to use 0 since the index will not be loaded. 1210d4002b98SHong Zhang */ 1211d4002b98SHong Zhang lastcol = 0; 1212d4002b98SHong Zhang if (nrow>0) { /* nonempty row */ 1213d4002b98SHong Zhang lastcol = cp[8*(nrow-1)+row_in_slice]; /* use the index from the last nonzero at current row */ 1214d4002b98SHong Zhang } else if (!row_in_slice) { /* first row of the currect slice is empty */ 1215d4002b98SHong Zhang for (j=1;j<8;j++) { 1216d4002b98SHong Zhang if (a->rlen[8*i+j]) { 1217d4002b98SHong Zhang lastcol = cp[j]; 1218d4002b98SHong Zhang break; 1219d4002b98SHong Zhang } 1220d4002b98SHong Zhang } 1221d4002b98SHong Zhang } else { 1222d4002b98SHong Zhang if (a->sliidx[i+1] != shift) lastcol = cp[row_in_slice-1]; /* use the index from the previous row */ 1223d4002b98SHong Zhang } 1224d4002b98SHong Zhang 1225d4002b98SHong Zhang for (k=nrow; k<(a->sliidx[i+1]-shift)/8; ++k) { 1226d4002b98SHong Zhang cp[8*k+row_in_slice] = lastcol; 1227d4002b98SHong Zhang vp[8*k+row_in_slice] = (MatScalar)0; 1228d4002b98SHong Zhang } 1229d4002b98SHong Zhang } 1230d4002b98SHong Zhang } 1231d4002b98SHong Zhang 1232d4002b98SHong Zhang A->info.mallocs += a->reallocs; 1233d4002b98SHong Zhang a->reallocs = 0; 1234d4002b98SHong Zhang 1235d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr); 1236d4002b98SHong Zhang PetscFunctionReturn(0); 1237d4002b98SHong Zhang } 1238d4002b98SHong Zhang 1239d4002b98SHong Zhang PetscErrorCode MatGetInfo_SeqSELL(Mat A,MatInfoType flag,MatInfo *info) 1240d4002b98SHong Zhang { 1241d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1242d4002b98SHong Zhang 1243d4002b98SHong Zhang PetscFunctionBegin; 1244d4002b98SHong Zhang info->block_size = 1.0; 1245d4002b98SHong Zhang info->nz_allocated = (double)a->maxallocmat; 1246d4002b98SHong Zhang info->nz_used = (double)a->sliidx[a->totalslices]; /* include padding zeros */ 1247d4002b98SHong Zhang info->nz_unneeded = (double)(a->maxallocmat-a->sliidx[a->totalslices]); 1248d4002b98SHong Zhang info->assemblies = (double)A->num_ass; 1249d4002b98SHong Zhang info->mallocs = (double)A->info.mallocs; 1250d4002b98SHong Zhang info->memory = ((PetscObject)A)->mem; 1251d4002b98SHong Zhang if (A->factortype) { 1252d4002b98SHong Zhang info->fill_ratio_given = A->info.fill_ratio_given; 1253d4002b98SHong Zhang info->fill_ratio_needed = A->info.fill_ratio_needed; 1254d4002b98SHong Zhang info->factor_mallocs = A->info.factor_mallocs; 1255d4002b98SHong Zhang } else { 1256d4002b98SHong Zhang info->fill_ratio_given = 0; 1257d4002b98SHong Zhang info->fill_ratio_needed = 0; 1258d4002b98SHong Zhang info->factor_mallocs = 0; 1259d4002b98SHong Zhang } 1260d4002b98SHong Zhang PetscFunctionReturn(0); 1261d4002b98SHong Zhang } 1262d4002b98SHong Zhang 1263d4002b98SHong Zhang PetscErrorCode MatSetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 1264d4002b98SHong Zhang { 1265d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1266d4002b98SHong Zhang PetscInt shift,i,k,l,low,high,t,ii,row,col,nrow; 1267d4002b98SHong Zhang PetscInt *cp,nonew=a->nonew,lastcol=-1; 1268d4002b98SHong Zhang MatScalar *vp,value; 1269d4002b98SHong Zhang PetscErrorCode ierr; 1270d4002b98SHong Zhang 1271d4002b98SHong Zhang PetscFunctionBegin; 1272d4002b98SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 1273d4002b98SHong Zhang row = im[k]; 1274d4002b98SHong Zhang if (row < 0) continue; 1275d4002b98SHong Zhang #if defined(PETSC_USE_DEBUG) 1276d4002b98SHong Zhang if (row >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->n-1); 1277d4002b98SHong Zhang #endif 1278d4002b98SHong Zhang shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */ 1279d4002b98SHong Zhang cp = a->colidx+shift; /* pointer to the row */ 1280d4002b98SHong Zhang vp = a->val+shift; /* pointer to the row */ 1281d4002b98SHong Zhang nrow = a->rlen[row]; 1282d4002b98SHong Zhang low = 0; 1283d4002b98SHong Zhang high = nrow; 1284d4002b98SHong Zhang 1285d4002b98SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 1286d4002b98SHong Zhang col = in[l]; 1287d4002b98SHong Zhang if (col<0) continue; 1288d4002b98SHong Zhang #if defined(PETSC_USE_DEBUG) 1289d4002b98SHong Zhang if (col >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Col too large: row %D max %D",col,A->cmap->n-1); 1290d4002b98SHong Zhang #endif 1291d4002b98SHong Zhang if (a->roworiented) { 1292d4002b98SHong Zhang value = v[l+k*n]; 1293d4002b98SHong Zhang } else { 1294d4002b98SHong Zhang value = v[k+l*m]; 1295d4002b98SHong Zhang } 1296d4002b98SHong Zhang if ((value == 0.0 && a->ignorezeroentries) && (is == ADD_VALUES)) continue; 1297d4002b98SHong Zhang 1298d4002b98SHong Zhang /* search in this row for the specified colmun, i indicates the column to be set */ 1299d4002b98SHong Zhang if (col <= lastcol) low = 0; 1300d4002b98SHong Zhang else high = nrow; 1301d4002b98SHong Zhang lastcol = col; 1302d4002b98SHong Zhang while (high-low > 5) { 1303d4002b98SHong Zhang t = (low+high)/2; 1304d4002b98SHong Zhang if (*(cp+t*8) > col) high = t; 1305d4002b98SHong Zhang else low = t; 1306d4002b98SHong Zhang } 1307d4002b98SHong Zhang for (i=low; i<high; i++) { 1308d4002b98SHong Zhang if (*(cp+i*8) > col) break; 1309d4002b98SHong Zhang if (*(cp+i*8) == col) { 1310d4002b98SHong Zhang if (is == ADD_VALUES) *(vp+i*8) += value; 1311d4002b98SHong Zhang else *(vp+i*8) = value; 1312d4002b98SHong Zhang low = i + 1; 1313d4002b98SHong Zhang goto noinsert; 1314d4002b98SHong Zhang } 1315d4002b98SHong Zhang } 1316d4002b98SHong Zhang if (value == 0.0 && a->ignorezeroentries) goto noinsert; 1317d4002b98SHong Zhang if (nonew == 1) goto noinsert; 1318d4002b98SHong Zhang if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 1319d4002b98SHong Zhang /* If the current row length exceeds the slice width (e.g. nrow==slice_width), allocate a new space, otherwise do nothing */ 1320d4002b98SHong Zhang MatSeqXSELLReallocateSELL(A,A->rmap->n,1,nrow,a->sliidx,row/8,row,col,a->colidx,a->val,cp,vp,nonew,MatScalar); 1321d4002b98SHong Zhang /* add the new nonzero to the high position, shift the remaining elements in current row to the right by one slot */ 1322d4002b98SHong Zhang for (ii=nrow-1; ii>=i; ii--) { 1323d4002b98SHong Zhang *(cp+(ii+1)*8) = *(cp+ii*8); 1324d4002b98SHong Zhang *(vp+(ii+1)*8) = *(vp+ii*8); 1325d4002b98SHong Zhang } 1326d4002b98SHong Zhang a->rlen[row]++; 1327d4002b98SHong Zhang *(cp+i*8) = col; 1328d4002b98SHong Zhang *(vp+i*8) = value; 1329d4002b98SHong Zhang a->nz++; 1330d4002b98SHong Zhang A->nonzerostate++; 1331d4002b98SHong Zhang low = i+1; high++; nrow++; 1332d4002b98SHong Zhang noinsert:; 1333d4002b98SHong Zhang } 1334d4002b98SHong Zhang a->rlen[row] = nrow; 1335d4002b98SHong Zhang } 1336d4002b98SHong Zhang PetscFunctionReturn(0); 1337d4002b98SHong Zhang } 1338d4002b98SHong Zhang 1339d4002b98SHong Zhang PetscErrorCode MatCopy_SeqSELL(Mat A,Mat B,MatStructure str) 1340d4002b98SHong Zhang { 1341d4002b98SHong Zhang PetscErrorCode ierr; 1342d4002b98SHong Zhang 1343d4002b98SHong Zhang PetscFunctionBegin; 1344d4002b98SHong Zhang /* If the two matrices have the same copy implementation, use fast copy. */ 1345d4002b98SHong Zhang if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 1346d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1347d4002b98SHong Zhang Mat_SeqSELL *b=(Mat_SeqSELL*)B->data; 1348d4002b98SHong Zhang 1349d4002b98SHong 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"); 1350d4002b98SHong Zhang ierr = PetscMemcpy(b->val,a->val,a->sliidx[a->totalslices]*sizeof(PetscScalar));CHKERRQ(ierr); 1351d4002b98SHong Zhang } else { 1352d4002b98SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1353d4002b98SHong Zhang } 1354d4002b98SHong Zhang PetscFunctionReturn(0); 1355d4002b98SHong Zhang } 1356d4002b98SHong Zhang 1357d4002b98SHong Zhang PetscErrorCode MatSetUp_SeqSELL(Mat A) 1358d4002b98SHong Zhang { 1359d4002b98SHong Zhang PetscErrorCode ierr; 1360d4002b98SHong Zhang 1361d4002b98SHong Zhang PetscFunctionBegin; 1362d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation(A,PETSC_DEFAULT,0);CHKERRQ(ierr); 1363d4002b98SHong Zhang PetscFunctionReturn(0); 1364d4002b98SHong Zhang } 1365d4002b98SHong Zhang 1366d4002b98SHong Zhang PetscErrorCode MatSeqSELLGetArray_SeqSELL(Mat A,PetscScalar *array[]) 1367d4002b98SHong Zhang { 1368d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1369d4002b98SHong Zhang 1370d4002b98SHong Zhang PetscFunctionBegin; 1371d4002b98SHong Zhang *array = a->val; 1372d4002b98SHong Zhang PetscFunctionReturn(0); 1373d4002b98SHong Zhang } 1374d4002b98SHong Zhang 1375d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray_SeqSELL(Mat A,PetscScalar *array[]) 1376d4002b98SHong Zhang { 1377d4002b98SHong Zhang PetscFunctionBegin; 1378d4002b98SHong Zhang PetscFunctionReturn(0); 1379d4002b98SHong Zhang } 1380d4002b98SHong Zhang 1381d4002b98SHong Zhang PetscErrorCode MatRealPart_SeqSELL(Mat A) 1382d4002b98SHong Zhang { 1383d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1384d4002b98SHong Zhang PetscInt i; 1385d4002b98SHong Zhang MatScalar *aval=a->val; 1386d4002b98SHong Zhang 1387d4002b98SHong Zhang PetscFunctionBegin; 1388d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i]=PetscRealPart(aval[i]); 1389d4002b98SHong Zhang PetscFunctionReturn(0); 1390d4002b98SHong Zhang } 1391d4002b98SHong Zhang 1392d4002b98SHong Zhang PetscErrorCode MatImaginaryPart_SeqSELL(Mat A) 1393d4002b98SHong Zhang { 1394d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1395d4002b98SHong Zhang PetscInt i; 1396d4002b98SHong Zhang MatScalar *aval=a->val; 1397d4002b98SHong Zhang PetscErrorCode ierr; 1398d4002b98SHong Zhang 1399d4002b98SHong Zhang PetscFunctionBegin; 1400d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i] = PetscImaginaryPart(aval[i]); 1401d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr); 1402d4002b98SHong Zhang PetscFunctionReturn(0); 1403d4002b98SHong Zhang } 1404d4002b98SHong Zhang 1405d4002b98SHong Zhang PetscErrorCode MatScale_SeqSELL(Mat inA,PetscScalar alpha) 1406d4002b98SHong Zhang { 1407d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)inA->data; 1408d4002b98SHong Zhang MatScalar *aval=a->val; 1409d4002b98SHong Zhang PetscScalar oalpha=alpha; 1410d4002b98SHong Zhang PetscBLASInt one=1,size; 1411d4002b98SHong Zhang PetscErrorCode ierr; 1412d4002b98SHong Zhang 1413d4002b98SHong Zhang PetscFunctionBegin; 1414d4002b98SHong Zhang ierr = PetscBLASIntCast(a->sliidx[a->totalslices],&size);CHKERRQ(ierr); 1415d4002b98SHong Zhang PetscStackCallBLAS("BLASscal",BLASscal_(&size,&oalpha,aval,&one)); 1416d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 1417d4002b98SHong Zhang ierr = MatSeqSELLInvalidateDiagonal(inA);CHKERRQ(ierr); 1418d4002b98SHong Zhang PetscFunctionReturn(0); 1419d4002b98SHong Zhang } 1420d4002b98SHong Zhang 1421d4002b98SHong Zhang PetscErrorCode MatShift_SeqSELL(Mat Y,PetscScalar a) 1422d4002b98SHong Zhang { 1423d4002b98SHong Zhang Mat_SeqSELL *y=(Mat_SeqSELL*)Y->data; 1424d4002b98SHong Zhang PetscErrorCode ierr; 1425d4002b98SHong Zhang 1426d4002b98SHong Zhang PetscFunctionBegin; 1427d4002b98SHong Zhang if (!Y->preallocated || !y->nz) { 1428d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation(Y,1,NULL);CHKERRQ(ierr); 1429d4002b98SHong Zhang } 1430d4002b98SHong Zhang ierr = MatShift_Basic(Y,a);CHKERRQ(ierr); 1431d4002b98SHong Zhang PetscFunctionReturn(0); 1432d4002b98SHong Zhang } 1433d4002b98SHong Zhang 1434d4002b98SHong Zhang PetscErrorCode MatSOR_SeqSELL(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 1435d4002b98SHong Zhang { 1436d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1437d4002b98SHong Zhang PetscScalar *x,sum,*t; 1438d4002b98SHong Zhang const MatScalar *idiag=0,*mdiag; 1439d4002b98SHong Zhang const PetscScalar *b,*xb; 1440d4002b98SHong Zhang PetscInt n,m=A->rmap->n,i,j,shift; 1441d4002b98SHong Zhang const PetscInt *diag; 1442d4002b98SHong Zhang PetscErrorCode ierr; 1443d4002b98SHong Zhang 1444d4002b98SHong Zhang PetscFunctionBegin; 1445d4002b98SHong Zhang its = its*lits; 1446d4002b98SHong Zhang 1447d4002b98SHong Zhang if (fshift != a->fshift || omega != a->omega) a->idiagvalid = PETSC_FALSE; /* must recompute idiag[] */ 1448d4002b98SHong Zhang if (!a->idiagvalid) {ierr = MatInvertDiagonal_SeqSELL(A,omega,fshift);CHKERRQ(ierr);} 1449d4002b98SHong Zhang a->fshift = fshift; 1450d4002b98SHong Zhang a->omega = omega; 1451d4002b98SHong Zhang 1452d4002b98SHong Zhang diag = a->diag; 1453d4002b98SHong Zhang t = a->ssor_work; 1454d4002b98SHong Zhang idiag = a->idiag; 1455d4002b98SHong Zhang mdiag = a->mdiag; 1456d4002b98SHong Zhang 1457d4002b98SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1458d4002b98SHong Zhang ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1459d4002b98SHong Zhang /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */ 1460d4002b98SHong Zhang if (flag == SOR_APPLY_UPPER) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_UPPER is not implemented"); 1461d4002b98SHong Zhang if (flag == SOR_APPLY_LOWER) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_LOWER is not implemented"); 1462d4002b98SHong Zhang if (flag & SOR_EISENSTAT) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support yet for Eisenstat"); 1463d4002b98SHong Zhang 1464d4002b98SHong Zhang if (flag & SOR_ZERO_INITIAL_GUESS) { 1465d4002b98SHong Zhang if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) { 1466d4002b98SHong Zhang for (i=0; i<m; i++) { 1467d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1468d4002b98SHong Zhang sum = b[i]; 1469d4002b98SHong Zhang n = (diag[i]-shift)/8; 1470d4002b98SHong Zhang for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]]; 1471d4002b98SHong Zhang t[i] = sum; 1472d4002b98SHong Zhang x[i] = sum*idiag[i]; 1473d4002b98SHong Zhang } 1474d4002b98SHong Zhang xb = t; 1475d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 1476d4002b98SHong Zhang } else xb = b; 1477d4002b98SHong Zhang if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) { 1478d4002b98SHong Zhang for (i=m-1; i>=0; i--) { 1479d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1480d4002b98SHong Zhang sum = xb[i]; 1481d4002b98SHong Zhang n = a->rlen[i]-(diag[i]-shift)/8-1; 1482d4002b98SHong Zhang for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]]; 1483d4002b98SHong Zhang if (xb == b) { 1484d4002b98SHong Zhang x[i] = sum*idiag[i]; 1485d4002b98SHong Zhang } else { 1486d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+sum*idiag[i]; /* omega in idiag */ 1487d4002b98SHong Zhang } 1488d4002b98SHong Zhang } 1489d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */ 1490d4002b98SHong Zhang } 1491d4002b98SHong Zhang its--; 1492d4002b98SHong Zhang } 1493d4002b98SHong Zhang while (its--) { 1494d4002b98SHong Zhang if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) { 1495d4002b98SHong Zhang for (i=0; i<m; i++) { 1496d4002b98SHong Zhang /* lower */ 1497d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1498d4002b98SHong Zhang sum = b[i]; 1499d4002b98SHong Zhang n = (diag[i]-shift)/8; 1500d4002b98SHong Zhang for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]]; 1501d4002b98SHong Zhang t[i] = sum; /* save application of the lower-triangular part */ 1502d4002b98SHong Zhang /* upper */ 1503d4002b98SHong Zhang n = a->rlen[i]-(diag[i]-shift)/8-1; 1504d4002b98SHong Zhang for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]]; 1505d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+sum*idiag[i]; /* omega in idiag */ 1506d4002b98SHong Zhang } 1507d4002b98SHong Zhang xb = t; 1508d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1509d4002b98SHong Zhang } else xb = b; 1510d4002b98SHong Zhang if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) { 1511d4002b98SHong Zhang for (i=m-1; i>=0; i--) { 1512d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1513d4002b98SHong Zhang sum = xb[i]; 1514d4002b98SHong Zhang if (xb == b) { 1515d4002b98SHong Zhang /* whole matrix (no checkpointing available) */ 1516d4002b98SHong Zhang n = a->rlen[i]; 1517d4002b98SHong Zhang for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]]; 1518d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+(sum+mdiag[i]*x[i])*idiag[i]; 1519d4002b98SHong Zhang } else { /* lower-triangular part has been saved, so only apply upper-triangular */ 1520d4002b98SHong Zhang n = a->rlen[i]-(diag[i]-shift)/8-1; 1521d4002b98SHong Zhang for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]]; 1522d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+sum*idiag[i]; /* omega in idiag */ 1523d4002b98SHong Zhang } 1524d4002b98SHong Zhang } 1525d4002b98SHong Zhang if (xb == b) { 1526d4002b98SHong Zhang ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1527d4002b98SHong Zhang } else { 1528d4002b98SHong Zhang ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */ 1529d4002b98SHong Zhang } 1530d4002b98SHong Zhang } 1531d4002b98SHong Zhang } 1532d4002b98SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1533d4002b98SHong Zhang ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1534d4002b98SHong Zhang PetscFunctionReturn(0); 1535d4002b98SHong Zhang } 1536d4002b98SHong Zhang 1537d4002b98SHong Zhang /* -------------------------------------------------------------------*/ 1538d4002b98SHong Zhang static struct _MatOps MatOps_Values = {MatSetValues_SeqSELL, 1539d4002b98SHong Zhang 0, 1540d4002b98SHong Zhang 0, 1541d4002b98SHong Zhang MatMult_SeqSELL, 1542d4002b98SHong Zhang /* 4*/ MatMultAdd_SeqSELL, 1543d4002b98SHong Zhang MatMultTranspose_SeqSELL, 1544d4002b98SHong Zhang MatMultTransposeAdd_SeqSELL, 1545d4002b98SHong Zhang 0, 1546d4002b98SHong Zhang 0, 1547d4002b98SHong Zhang 0, 1548d4002b98SHong Zhang /* 10*/ 0, 1549d4002b98SHong Zhang 0, 1550d4002b98SHong Zhang 0, 1551d4002b98SHong Zhang MatSOR_SeqSELL, 1552d4002b98SHong Zhang 0, 1553d4002b98SHong Zhang /* 15*/ MatGetInfo_SeqSELL, 1554d4002b98SHong Zhang MatEqual_SeqSELL, 1555d4002b98SHong Zhang MatGetDiagonal_SeqSELL, 1556d4002b98SHong Zhang MatDiagonalScale_SeqSELL, 1557d4002b98SHong Zhang 0, 1558d4002b98SHong Zhang /* 20*/ 0, 1559d4002b98SHong Zhang MatAssemblyEnd_SeqSELL, 1560d4002b98SHong Zhang MatSetOption_SeqSELL, 1561d4002b98SHong Zhang MatZeroEntries_SeqSELL, 1562d4002b98SHong Zhang /* 24*/ 0, 1563d4002b98SHong Zhang 0, 1564d4002b98SHong Zhang 0, 1565d4002b98SHong Zhang 0, 1566d4002b98SHong Zhang 0, 1567d4002b98SHong Zhang /* 29*/ MatSetUp_SeqSELL, 1568d4002b98SHong Zhang 0, 1569d4002b98SHong Zhang 0, 1570d4002b98SHong Zhang 0, 1571d4002b98SHong Zhang 0, 1572d4002b98SHong Zhang /* 34*/ MatDuplicate_SeqSELL, 1573d4002b98SHong Zhang 0, 1574d4002b98SHong Zhang 0, 1575d4002b98SHong Zhang 0, 1576d4002b98SHong Zhang 0, 1577d4002b98SHong Zhang /* 39*/ 0, 1578d4002b98SHong Zhang 0, 1579d4002b98SHong Zhang 0, 1580d4002b98SHong Zhang MatGetValues_SeqSELL, 1581d4002b98SHong Zhang MatCopy_SeqSELL, 1582d4002b98SHong Zhang /* 44*/ 0, 1583d4002b98SHong Zhang MatScale_SeqSELL, 1584d4002b98SHong Zhang MatShift_SeqSELL, 1585d4002b98SHong Zhang 0, 1586d4002b98SHong Zhang 0, 1587d4002b98SHong Zhang /* 49*/ 0, 1588d4002b98SHong Zhang 0, 1589d4002b98SHong Zhang 0, 1590d4002b98SHong Zhang 0, 1591d4002b98SHong Zhang 0, 1592d4002b98SHong Zhang /* 54*/ MatFDColoringCreate_SeqXAIJ, 1593d4002b98SHong Zhang 0, 1594d4002b98SHong Zhang 0, 1595d4002b98SHong Zhang 0, 1596d4002b98SHong Zhang 0, 1597d4002b98SHong Zhang /* 59*/ 0, 1598d4002b98SHong Zhang MatDestroy_SeqSELL, 1599d4002b98SHong Zhang MatView_SeqSELL, 1600d4002b98SHong Zhang 0, 1601d4002b98SHong Zhang 0, 1602d4002b98SHong Zhang /* 64*/ 0, 1603d4002b98SHong Zhang 0, 1604d4002b98SHong Zhang 0, 1605d4002b98SHong Zhang 0, 1606d4002b98SHong Zhang 0, 1607d4002b98SHong Zhang /* 69*/ 0, 1608d4002b98SHong Zhang 0, 1609d4002b98SHong Zhang 0, 1610d4002b98SHong Zhang 0, 1611d4002b98SHong Zhang 0, 1612d4002b98SHong Zhang /* 74*/ 0, 1613d4002b98SHong Zhang MatFDColoringApply_AIJ, /* reuse the FDColoring function for AIJ */ 1614d4002b98SHong Zhang 0, 1615d4002b98SHong Zhang 0, 1616d4002b98SHong Zhang 0, 1617d4002b98SHong Zhang /* 79*/ 0, 1618d4002b98SHong Zhang 0, 1619d4002b98SHong Zhang 0, 1620d4002b98SHong Zhang 0, 1621d4002b98SHong Zhang 0, 1622d4002b98SHong Zhang /* 84*/ 0, 1623d4002b98SHong Zhang 0, 1624d4002b98SHong Zhang 0, 1625d4002b98SHong Zhang 0, 1626d4002b98SHong Zhang 0, 1627d4002b98SHong Zhang /* 89*/ 0, 1628d4002b98SHong Zhang 0, 1629d4002b98SHong Zhang 0, 1630d4002b98SHong Zhang 0, 1631d4002b98SHong Zhang 0, 1632d4002b98SHong Zhang /* 94*/ 0, 1633d4002b98SHong Zhang 0, 1634d4002b98SHong Zhang 0, 1635d4002b98SHong Zhang 0, 1636d4002b98SHong Zhang 0, 1637d4002b98SHong Zhang /* 99*/ 0, 1638d4002b98SHong Zhang 0, 1639d4002b98SHong Zhang 0, 1640d4002b98SHong Zhang MatConjugate_SeqSELL, 1641d4002b98SHong Zhang 0, 1642d4002b98SHong Zhang /*104*/ 0, 1643d4002b98SHong Zhang 0, 1644d4002b98SHong Zhang 0, 1645d4002b98SHong Zhang 0, 1646d4002b98SHong Zhang 0, 1647d4002b98SHong Zhang /*109*/ 0, 1648d4002b98SHong Zhang 0, 1649d4002b98SHong Zhang 0, 1650d4002b98SHong Zhang 0, 1651d4002b98SHong Zhang MatMissingDiagonal_SeqSELL, 1652d4002b98SHong Zhang /*114*/ 0, 1653d4002b98SHong Zhang 0, 1654d4002b98SHong Zhang 0, 1655d4002b98SHong Zhang 0, 1656d4002b98SHong Zhang 0, 1657d4002b98SHong Zhang /*119*/ 0, 1658d4002b98SHong Zhang 0, 1659d4002b98SHong Zhang 0, 1660d4002b98SHong Zhang 0, 1661d4002b98SHong Zhang 0, 1662d4002b98SHong Zhang /*124*/ 0, 1663d4002b98SHong Zhang 0, 1664d4002b98SHong Zhang 0, 1665d4002b98SHong Zhang 0, 1666d4002b98SHong Zhang 0, 1667d4002b98SHong Zhang /*129*/ 0, 1668d4002b98SHong Zhang 0, 1669d4002b98SHong Zhang 0, 1670d4002b98SHong Zhang 0, 1671d4002b98SHong Zhang 0, 1672d4002b98SHong Zhang /*134*/ 0, 1673d4002b98SHong Zhang 0, 1674d4002b98SHong Zhang 0, 1675d4002b98SHong Zhang 0, 1676d4002b98SHong Zhang 0, 1677d4002b98SHong Zhang /*139*/ 0, 1678d4002b98SHong Zhang 0, 1679d4002b98SHong Zhang 0, 1680d4002b98SHong Zhang MatFDColoringSetUp_SeqXAIJ, 1681d4002b98SHong Zhang 0, 1682d4002b98SHong Zhang /*144*/0 1683d4002b98SHong Zhang }; 1684d4002b98SHong Zhang 1685d4002b98SHong Zhang PetscErrorCode MatStoreValues_SeqSELL(Mat mat) 1686d4002b98SHong Zhang { 1687d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)mat->data; 1688d4002b98SHong Zhang PetscErrorCode ierr; 1689d4002b98SHong Zhang 1690d4002b98SHong Zhang PetscFunctionBegin; 1691d4002b98SHong Zhang if (!a->nonew) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1692d4002b98SHong Zhang 1693d4002b98SHong Zhang /* allocate space for values if not already there */ 1694d4002b98SHong Zhang if (!a->saved_values) { 1695d4002b98SHong Zhang ierr = PetscMalloc1(a->sliidx[a->totalslices]+1,&a->saved_values);CHKERRQ(ierr); 1696d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)mat,(a->sliidx[a->totalslices]+1)*sizeof(PetscScalar));CHKERRQ(ierr); 1697d4002b98SHong Zhang } 1698d4002b98SHong Zhang 1699d4002b98SHong Zhang /* copy values over */ 1700d4002b98SHong Zhang ierr = PetscMemcpy(a->saved_values,a->val,a->sliidx[a->totalslices]*sizeof(PetscScalar));CHKERRQ(ierr); 1701d4002b98SHong Zhang PetscFunctionReturn(0); 1702d4002b98SHong Zhang } 1703d4002b98SHong Zhang 1704d4002b98SHong Zhang PetscErrorCode MatRetrieveValues_SeqSELL(Mat mat) 1705d4002b98SHong Zhang { 1706d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)mat->data; 1707d4002b98SHong Zhang PetscErrorCode ierr; 1708d4002b98SHong Zhang 1709d4002b98SHong Zhang PetscFunctionBegin; 1710d4002b98SHong Zhang if (!a->nonew) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1711d4002b98SHong Zhang if (!a->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 1712d4002b98SHong Zhang /* copy values over */ 1713d4002b98SHong Zhang ierr = PetscMemcpy(a->val,a->saved_values,a->sliidx[a->totalslices]*sizeof(PetscScalar));CHKERRQ(ierr); 1714d4002b98SHong Zhang PetscFunctionReturn(0); 1715d4002b98SHong Zhang } 1716d4002b98SHong Zhang 1717d4002b98SHong Zhang /*@C 1718d4002b98SHong Zhang MatSeqSELLRestoreArray - returns access to the array where the data for a MATSEQSELL matrix is stored obtained by MatSeqSELLGetArray() 1719d4002b98SHong Zhang 1720d4002b98SHong Zhang Not Collective 1721d4002b98SHong Zhang 1722d4002b98SHong Zhang Input Parameters: 1723d4002b98SHong Zhang . mat - a MATSEQSELL matrix 1724d4002b98SHong Zhang . array - pointer to the data 1725d4002b98SHong Zhang 1726d4002b98SHong Zhang Level: intermediate 1727d4002b98SHong Zhang 1728d4002b98SHong Zhang .seealso: MatSeqSELLGetArray(), MatSeqSELLRestoreArrayF90() 1729d4002b98SHong Zhang @*/ 1730d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray(Mat A,PetscScalar **array) 1731d4002b98SHong Zhang { 1732d4002b98SHong Zhang PetscErrorCode ierr; 1733d4002b98SHong Zhang 1734d4002b98SHong Zhang PetscFunctionBegin; 1735d4002b98SHong Zhang ierr = PetscUseMethod(A,"MatSeqSELLRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr); 1736d4002b98SHong Zhang PetscFunctionReturn(0); 1737d4002b98SHong Zhang } 1738d4002b98SHong Zhang 1739d4002b98SHong Zhang PETSC_EXTERN PetscErrorCode MatCreate_SeqSELL(Mat B) 1740d4002b98SHong Zhang { 1741d4002b98SHong Zhang Mat_SeqSELL *b; 1742d4002b98SHong Zhang PetscMPIInt size; 1743d4002b98SHong Zhang PetscErrorCode ierr; 1744d4002b98SHong Zhang 1745d4002b98SHong Zhang PetscFunctionBegin; 1746d4002b98SHong Zhang ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr); 1747d4002b98SHong Zhang if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Comm must be of size 1"); 1748d4002b98SHong Zhang 1749d4002b98SHong Zhang ierr = PetscNewLog(B,&b);CHKERRQ(ierr); 1750d4002b98SHong Zhang 1751d4002b98SHong Zhang B->data = (void*)b; 1752d4002b98SHong Zhang 1753d4002b98SHong Zhang ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1754d4002b98SHong Zhang 1755d4002b98SHong Zhang b->row = 0; 1756d4002b98SHong Zhang b->col = 0; 1757d4002b98SHong Zhang b->icol = 0; 1758d4002b98SHong Zhang b->reallocs = 0; 1759d4002b98SHong Zhang b->ignorezeroentries = PETSC_FALSE; 1760d4002b98SHong Zhang b->roworiented = PETSC_TRUE; 1761d4002b98SHong Zhang b->nonew = 0; 1762d4002b98SHong Zhang b->diag = 0; 1763d4002b98SHong Zhang b->solve_work = 0; 1764d4002b98SHong Zhang B->spptr = 0; 1765d4002b98SHong Zhang b->saved_values = 0; 1766d4002b98SHong Zhang b->idiag = 0; 1767d4002b98SHong Zhang b->mdiag = 0; 1768d4002b98SHong Zhang b->ssor_work = 0; 1769d4002b98SHong Zhang b->omega = 1.0; 1770d4002b98SHong Zhang b->fshift = 0.0; 1771d4002b98SHong Zhang b->idiagvalid = PETSC_FALSE; 1772d4002b98SHong Zhang b->ibdiagvalid = PETSC_FALSE; 1773d4002b98SHong Zhang b->keepnonzeropattern = PETSC_FALSE; 1774d4002b98SHong Zhang 1775d4002b98SHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSELL);CHKERRQ(ierr); 1776d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLGetArray_C",MatSeqSELLGetArray_SeqSELL);CHKERRQ(ierr); 1777d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLRestoreArray_C",MatSeqSELLRestoreArray_SeqSELL);CHKERRQ(ierr); 1778d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSELL);CHKERRQ(ierr); 1779d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSELL);CHKERRQ(ierr); 1780d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLSetPreallocation_C",MatSeqSELLSetPreallocation_SeqSELL);CHKERRQ(ierr); 1781d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsell_seqaij_C",MatConvert_SeqSELL_SeqAIJ);CHKERRQ(ierr); 1782d4002b98SHong Zhang PetscFunctionReturn(0); 1783d4002b98SHong Zhang } 1784d4002b98SHong Zhang 1785d4002b98SHong Zhang /* 1786d4002b98SHong Zhang Given a matrix generated with MatGetFactor() duplicates all the information in A into B 1787d4002b98SHong Zhang */ 1788d4002b98SHong Zhang PetscErrorCode MatDuplicateNoCreate_SeqSELL(Mat C,Mat A,MatDuplicateOption cpvalues,PetscBool mallocmatspace) 1789d4002b98SHong Zhang { 1790d4002b98SHong Zhang Mat_SeqSELL *c,*a=(Mat_SeqSELL*)A->data; 1791d4002b98SHong Zhang PetscInt i,m=A->rmap->n; 1792d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 1793d4002b98SHong Zhang PetscErrorCode ierr; 1794d4002b98SHong Zhang 1795d4002b98SHong Zhang PetscFunctionBegin; 1796d4002b98SHong Zhang c = (Mat_SeqSELL*)C->data; 1797d4002b98SHong Zhang 1798d4002b98SHong Zhang C->factortype = A->factortype; 1799d4002b98SHong Zhang c->row = 0; 1800d4002b98SHong Zhang c->col = 0; 1801d4002b98SHong Zhang c->icol = 0; 1802d4002b98SHong Zhang c->reallocs = 0; 1803d4002b98SHong Zhang 1804d4002b98SHong Zhang C->assembled = PETSC_TRUE; 1805d4002b98SHong Zhang 1806d4002b98SHong Zhang ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr); 1807d4002b98SHong Zhang ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr); 1808d4002b98SHong Zhang 1809d4002b98SHong Zhang ierr = PetscMalloc1(8*totalslices,&c->rlen);CHKERRQ(ierr); 1810d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt));CHKERRQ(ierr); 1811d4002b98SHong Zhang ierr = PetscMalloc1(totalslices+1,&c->sliidx);CHKERRQ(ierr); 1812d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)C, (totalslices+1)*sizeof(PetscInt));CHKERRQ(ierr); 1813d4002b98SHong Zhang 1814d4002b98SHong Zhang for (i=0; i<m; i++) c->rlen[i] = a->rlen[i]; 1815d4002b98SHong Zhang for (i=0; i<totalslices+1; i++) c->sliidx[i] = a->sliidx[i]; 1816d4002b98SHong Zhang 1817d4002b98SHong Zhang /* allocate the matrix space */ 1818d4002b98SHong Zhang if (mallocmatspace) { 1819d4002b98SHong Zhang ierr = PetscMalloc2(a->maxallocmat,&c->val,a->maxallocmat,&c->colidx);CHKERRQ(ierr); 1820d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)C,a->maxallocmat*(sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 1821d4002b98SHong Zhang 1822d4002b98SHong Zhang c->singlemalloc = PETSC_TRUE; 1823d4002b98SHong Zhang 1824d4002b98SHong Zhang if (m > 0) { 1825d4002b98SHong Zhang ierr = PetscMemcpy(c->colidx,a->colidx,(a->maxallocmat)*sizeof(PetscInt));CHKERRQ(ierr); 1826d4002b98SHong Zhang if (cpvalues == MAT_COPY_VALUES) { 1827d4002b98SHong Zhang ierr = PetscMemcpy(c->val,a->val,a->maxallocmat*sizeof(PetscScalar));CHKERRQ(ierr); 1828d4002b98SHong Zhang } else { 1829d4002b98SHong Zhang ierr = PetscMemzero(c->val,a->maxallocmat*sizeof(PetscScalar));CHKERRQ(ierr); 1830d4002b98SHong Zhang } 1831d4002b98SHong Zhang } 1832d4002b98SHong Zhang } 1833d4002b98SHong Zhang 1834d4002b98SHong Zhang c->ignorezeroentries = a->ignorezeroentries; 1835d4002b98SHong Zhang c->roworiented = a->roworiented; 1836d4002b98SHong Zhang c->nonew = a->nonew; 1837d4002b98SHong Zhang if (a->diag) { 1838d4002b98SHong Zhang ierr = PetscMalloc1(m,&c->diag);CHKERRQ(ierr); 1839d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt));CHKERRQ(ierr); 1840d4002b98SHong Zhang for (i=0; i<m; i++) { 1841d4002b98SHong Zhang c->diag[i] = a->diag[i]; 1842d4002b98SHong Zhang } 1843d4002b98SHong Zhang } else c->diag = 0; 1844d4002b98SHong Zhang 1845d4002b98SHong Zhang c->solve_work = 0; 1846d4002b98SHong Zhang c->saved_values = 0; 1847d4002b98SHong Zhang c->idiag = 0; 1848d4002b98SHong Zhang c->ssor_work = 0; 1849d4002b98SHong Zhang c->keepnonzeropattern = a->keepnonzeropattern; 1850d4002b98SHong Zhang c->free_val = PETSC_TRUE; 1851d4002b98SHong Zhang c->free_colidx = PETSC_TRUE; 1852d4002b98SHong Zhang 1853d4002b98SHong Zhang c->maxallocmat = a->maxallocmat; 1854d4002b98SHong Zhang c->maxallocrow = a->maxallocrow; 1855d4002b98SHong Zhang c->rlenmax = a->rlenmax; 1856d4002b98SHong Zhang c->nz = a->nz; 1857d4002b98SHong Zhang C->preallocated = PETSC_TRUE; 1858d4002b98SHong Zhang 1859d4002b98SHong Zhang c->nonzerorowcnt = a->nonzerorowcnt; 1860d4002b98SHong Zhang C->nonzerostate = A->nonzerostate; 1861d4002b98SHong Zhang 1862d4002b98SHong Zhang ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr); 1863d4002b98SHong Zhang PetscFunctionReturn(0); 1864d4002b98SHong Zhang } 1865d4002b98SHong Zhang 1866d4002b98SHong Zhang PetscErrorCode MatDuplicate_SeqSELL(Mat A,MatDuplicateOption cpvalues,Mat *B) 1867d4002b98SHong Zhang { 1868d4002b98SHong Zhang PetscErrorCode ierr; 1869d4002b98SHong Zhang 1870d4002b98SHong Zhang PetscFunctionBegin; 1871d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 1872d4002b98SHong Zhang ierr = MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr); 1873d4002b98SHong Zhang if (!(A->rmap->n % A->rmap->bs) && !(A->cmap->n % A->cmap->bs)) { 1874d4002b98SHong Zhang ierr = MatSetBlockSizesFromMats(*B,A,A);CHKERRQ(ierr); 1875d4002b98SHong Zhang } 1876d4002b98SHong Zhang ierr = MatSetType(*B,((PetscObject)A)->type_name);CHKERRQ(ierr); 1877d4002b98SHong Zhang ierr = MatDuplicateNoCreate_SeqSELL(*B,A,cpvalues,PETSC_TRUE);CHKERRQ(ierr); 1878d4002b98SHong Zhang PetscFunctionReturn(0); 1879d4002b98SHong Zhang } 1880d4002b98SHong Zhang 1881d4002b98SHong Zhang /*@C 1882d4002b98SHong Zhang MatCreateSeqSELL - Creates a sparse matrix in SELL format. 1883d4002b98SHong Zhang 1884d4002b98SHong Zhang Collective on MPI_Comm 1885d4002b98SHong Zhang 1886d4002b98SHong Zhang Input Parameters: 1887d4002b98SHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 1888d4002b98SHong Zhang . m - number of rows 1889d4002b98SHong Zhang . n - number of columns 1890d4002b98SHong Zhang . rlenmax - maximum number of nonzeros in a row 1891d4002b98SHong Zhang - rlen - array containing the number of nonzeros in the various rows 1892d4002b98SHong Zhang (possibly different for each row) or NULL 1893d4002b98SHong Zhang 1894d4002b98SHong Zhang Output Parameter: 1895d4002b98SHong Zhang . A - the matrix 1896d4002b98SHong Zhang 1897d4002b98SHong Zhang It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 1898d4002b98SHong Zhang MatXXXXSetPreallocation() paradgm instead of this routine directly. 1899d4002b98SHong Zhang [MatXXXXSetPreallocation() is, for example, MatSeqSELLSetPreallocation] 1900d4002b98SHong Zhang 1901d4002b98SHong Zhang Notes: 1902d4002b98SHong Zhang If nnz is given then nz is ignored 1903d4002b98SHong Zhang 1904d4002b98SHong Zhang Specify the preallocated storage with either rlenmax or rlen (not both). 1905d4002b98SHong Zhang Set rlenmax=PETSC_DEFAULT and rlen=NULL for PETSc to control dynamic memory 1906d4002b98SHong Zhang allocation. For large problems you MUST preallocate memory or you 1907d4002b98SHong Zhang will get TERRIBLE performance, see the users' manual chapter on matrices. 1908d4002b98SHong Zhang 1909d4002b98SHong Zhang Level: intermediate 1910d4002b98SHong Zhang 1911d4002b98SHong Zhang .seealso: MatCreate(), MatCreateSELL(), MatSetValues(), MatCreateSeqSELLWithArrays() 1912d4002b98SHong Zhang 1913d4002b98SHong Zhang @*/ 1914d4002b98SHong Zhang PetscErrorCode MatCreateSeqSELL(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt maxallocrow,const PetscInt rlen[],Mat *A) 1915d4002b98SHong Zhang { 1916d4002b98SHong Zhang PetscErrorCode ierr; 1917d4002b98SHong Zhang 1918d4002b98SHong Zhang PetscFunctionBegin; 1919d4002b98SHong Zhang ierr = MatCreate(comm,A);CHKERRQ(ierr); 1920d4002b98SHong Zhang ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 1921d4002b98SHong Zhang ierr = MatSetType(*A,MATSEQSELL);CHKERRQ(ierr); 1922d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation_SeqSELL(*A,maxallocrow,rlen);CHKERRQ(ierr); 1923d4002b98SHong Zhang PetscFunctionReturn(0); 1924d4002b98SHong Zhang } 1925d4002b98SHong Zhang 1926d4002b98SHong Zhang PetscErrorCode MatEqual_SeqSELL(Mat A,Mat B,PetscBool * flg) 1927d4002b98SHong Zhang { 1928d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data,*b=(Mat_SeqSELL*)B->data; 1929d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 1930d4002b98SHong Zhang PetscErrorCode ierr; 1931d4002b98SHong Zhang 1932d4002b98SHong Zhang PetscFunctionBegin; 1933d4002b98SHong Zhang /* If the matrix dimensions are not equal,or no of nonzeros */ 1934d4002b98SHong Zhang if ((A->rmap->n != B->rmap->n) || (A->cmap->n != B->cmap->n) ||(a->nz != b->nz) || (a->rlenmax != b->rlenmax)) { 1935d4002b98SHong Zhang *flg = PETSC_FALSE; 1936d4002b98SHong Zhang PetscFunctionReturn(0); 1937d4002b98SHong Zhang } 1938d4002b98SHong Zhang /* if the a->colidx are the same */ 1939d4002b98SHong Zhang ierr = PetscMemcmp(a->colidx,b->colidx,a->sliidx[totalslices]*sizeof(PetscInt),flg);CHKERRQ(ierr); 1940d4002b98SHong Zhang if (!*flg) PetscFunctionReturn(0); 1941d4002b98SHong Zhang /* if a->val are the same */ 1942d4002b98SHong Zhang ierr = PetscMemcmp(a->val,b->val,a->sliidx[totalslices]*sizeof(PetscScalar),flg);CHKERRQ(ierr); 1943d4002b98SHong Zhang PetscFunctionReturn(0); 1944d4002b98SHong Zhang } 1945d4002b98SHong Zhang 1946d4002b98SHong Zhang PetscErrorCode MatSeqSELLInvalidateDiagonal(Mat A) 1947d4002b98SHong Zhang { 1948d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1949d4002b98SHong Zhang 1950d4002b98SHong Zhang PetscFunctionBegin; 1951d4002b98SHong Zhang a->idiagvalid = PETSC_FALSE; 1952d4002b98SHong Zhang a->ibdiagvalid = PETSC_FALSE; 1953d4002b98SHong Zhang PetscFunctionReturn(0); 1954d4002b98SHong Zhang } 1955d4002b98SHong Zhang 1956d4002b98SHong Zhang PetscErrorCode MatConjugate_SeqSELL(Mat A) 1957d4002b98SHong Zhang { 1958d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1959d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1960d4002b98SHong Zhang PetscInt i; 1961d4002b98SHong Zhang PetscScalar *val = a->val; 1962d4002b98SHong Zhang 1963d4002b98SHong Zhang PetscFunctionBegin; 1964d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) { 1965d4002b98SHong Zhang val[i] = PetscConj(val[i]); 1966d4002b98SHong Zhang } 1967d4002b98SHong Zhang #else 1968d4002b98SHong Zhang PetscFunctionBegin; 1969d4002b98SHong Zhang #endif 1970d4002b98SHong Zhang PetscFunctionReturn(0); 1971d4002b98SHong Zhang } 1972