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> 8ed73aabaSBarry Smith 9ed73aabaSBarry Smith static PetscBool cited = PETSC_FALSE; 10ed73aabaSBarry Smith static const char citation[] = 11ed73aabaSBarry Smith "@inproceedings{ZhangELLPACK2018,\n" 12ed73aabaSBarry Smith " author = {Hong Zhang and Richard T. Mills and Karl Rupp and Barry F. Smith},\n" 13ed73aabaSBarry Smith " title = {Vectorized Parallel Sparse Matrix-Vector Multiplication in {PETSc} Using {AVX-512}},\n" 14ed73aabaSBarry Smith " booktitle = {Proceedings of the 47th International Conference on Parallel Processing},\n" 15ed73aabaSBarry Smith " year = 2018\n" 16ed73aabaSBarry Smith "}\n"; 17ed73aabaSBarry Smith 185f70456aSHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && (defined(__AVX512F__) || (defined(__AVX2__) && defined(__FMA__)) || defined(__AVX__)) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 194243e2ceSHong Zhang 20d4002b98SHong Zhang #include <immintrin.h> 21d4002b98SHong Zhang 22d4002b98SHong Zhang #if !defined(_MM_SCALE_8) 23d4002b98SHong Zhang #define _MM_SCALE_8 8 24d4002b98SHong Zhang #endif 25d4002b98SHong Zhang 26d4002b98SHong Zhang #if defined(__AVX512F__) 27d4002b98SHong Zhang /* these do not work 28d4002b98SHong Zhang vec_idx = _mm512_loadunpackhi_epi32(vec_idx,acolidx); 29d4002b98SHong Zhang vec_vals = _mm512_loadunpackhi_pd(vec_vals,aval); 30d4002b98SHong Zhang */ 31d4002b98SHong Zhang #define AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y) \ 32d4002b98SHong Zhang /* if the mask bit is set, copy from acolidx, otherwise from vec_idx */ \ 33ef588d5cSRichard Tran Mills vec_idx = _mm256_loadu_si256((__m256i const*)acolidx); \ 34ef588d5cSRichard Tran Mills vec_vals = _mm512_loadu_pd(aval); \ 35d4002b98SHong Zhang vec_x = _mm512_i32gather_pd(vec_idx,x,_MM_SCALE_8); \ 36a48a6482SHong Zhang vec_y = _mm512_fmadd_pd(vec_x,vec_vals,vec_y) 375f70456aSHong Zhang #elif defined(__AVX2__) && defined(__FMA__) 38a48a6482SHong Zhang #define AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y) \ 39ef588d5cSRichard Tran Mills vec_vals = _mm256_loadu_pd(aval); \ 40ef588d5cSRichard Tran Mills vec_idx = _mm_loadu_si128((__m128i const*)acolidx); /* SSE2 */ \ 41a48a6482SHong Zhang vec_x = _mm256_i32gather_pd(x,vec_idx,_MM_SCALE_8); \ 42a48a6482SHong Zhang vec_y = _mm256_fmadd_pd(vec_x,vec_vals,vec_y) 43d4002b98SHong Zhang #endif 44d4002b98SHong Zhang #endif /* PETSC_HAVE_IMMINTRIN_H */ 45d4002b98SHong Zhang 46d4002b98SHong Zhang /*@C 47d4002b98SHong Zhang MatSeqSELLSetPreallocation - For good matrix assembly performance 48d4002b98SHong Zhang the user should preallocate the matrix storage by setting the parameter nz 49d4002b98SHong Zhang (or the array nnz). By setting these parameters accurately, performance 50d4002b98SHong Zhang during matrix assembly can be increased significantly. 51d4002b98SHong Zhang 52d083f849SBarry Smith Collective 53d4002b98SHong Zhang 54d4002b98SHong Zhang Input Parameters: 55d4002b98SHong Zhang + B - The matrix 56d4002b98SHong Zhang . nz - number of nonzeros per row (same for all rows) 57d4002b98SHong Zhang - nnz - array containing the number of nonzeros in the various rows 58d4002b98SHong Zhang (possibly different for each row) or NULL 59d4002b98SHong Zhang 60d4002b98SHong Zhang Notes: 61d4002b98SHong Zhang If nnz is given then nz is ignored. 62d4002b98SHong Zhang 63d4002b98SHong Zhang Specify the preallocated storage with either nz or nnz (not both). 64d4002b98SHong Zhang Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 65d4002b98SHong Zhang allocation. For large problems you MUST preallocate memory or you 66d4002b98SHong Zhang will get TERRIBLE performance, see the users' manual chapter on matrices. 67d4002b98SHong Zhang 68d4002b98SHong Zhang You can call MatGetInfo() to get information on how effective the preallocation was; 69d4002b98SHong Zhang for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 70d4002b98SHong Zhang You can also run with the option -info and look for messages with the string 71d4002b98SHong Zhang malloc in them to see if additional memory allocation was needed. 72d4002b98SHong Zhang 73d4002b98SHong Zhang Developers: Use nz of MAT_SKIP_ALLOCATION to not allocate any space for the matrix 74d4002b98SHong Zhang entries or columns indices. 75d4002b98SHong Zhang 76c7ee91abSRichard Tran Mills The maximum number of nonzeos in any row should be as accurate as possible. 77c7ee91abSRichard Tran Mills If it is underestimated, you will get bad performance due to reallocation 78d4002b98SHong Zhang (MatSeqXSELLReallocateSELL). 79d4002b98SHong Zhang 80d4002b98SHong Zhang Level: intermediate 81d4002b98SHong Zhang 82db781477SPatrick Sanan .seealso: `MatCreate()`, `MatCreateSELL()`, `MatSetValues()`, `MatGetInfo()` 83d4002b98SHong Zhang 84d4002b98SHong Zhang @*/ 85d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation(Mat B,PetscInt rlenmax,const PetscInt rlen[]) 86d4002b98SHong Zhang { 87d4002b98SHong Zhang PetscFunctionBegin; 88d4002b98SHong Zhang PetscValidHeaderSpecific(B,MAT_CLASSID,1); 89d4002b98SHong Zhang PetscValidType(B,1); 90cac4c232SBarry Smith PetscTryMethod(B,"MatSeqSELLSetPreallocation_C",(Mat,PetscInt,const PetscInt[]),(B,rlenmax,rlen)); 91d4002b98SHong Zhang PetscFunctionReturn(0); 92d4002b98SHong Zhang } 93d4002b98SHong Zhang 94d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation_SeqSELL(Mat B,PetscInt maxallocrow,const PetscInt rlen[]) 95d4002b98SHong Zhang { 96d4002b98SHong Zhang Mat_SeqSELL *b; 97d4002b98SHong Zhang PetscInt i,j,totalslices; 98d4002b98SHong Zhang PetscBool skipallocation=PETSC_FALSE,realalloc=PETSC_FALSE; 99d4002b98SHong Zhang 100d4002b98SHong Zhang PetscFunctionBegin; 101d4002b98SHong Zhang if (maxallocrow >= 0 || rlen) realalloc = PETSC_TRUE; 102d4002b98SHong Zhang if (maxallocrow == MAT_SKIP_ALLOCATION) { 103d4002b98SHong Zhang skipallocation = PETSC_TRUE; 104d4002b98SHong Zhang maxallocrow = 0; 105d4002b98SHong Zhang } 106d4002b98SHong Zhang 1079566063dSJacob Faibussowitsch PetscCall(PetscLayoutSetUp(B->rmap)); 1089566063dSJacob Faibussowitsch PetscCall(PetscLayoutSetUp(B->cmap)); 109d4002b98SHong Zhang 110d4002b98SHong Zhang /* FIXME: if one preallocates more space than needed, the matrix does not shrink automatically, but for best performance it should */ 111d4002b98SHong Zhang if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 5; 11208401ef6SPierre Jolivet PetscCheck(maxallocrow >= 0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"maxallocrow cannot be less than 0: value %" PetscInt_FMT,maxallocrow); 113d4002b98SHong Zhang if (rlen) { 114d4002b98SHong Zhang for (i=0; i<B->rmap->n; i++) { 11508401ef6SPierre Jolivet PetscCheck(rlen[i] >= 0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"rlen cannot be less than 0: local row %" PetscInt_FMT " value %" PetscInt_FMT,i,rlen[i]); 11608401ef6SPierre Jolivet PetscCheck(rlen[i] <= B->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"rlen cannot be greater than row length: local row %" PetscInt_FMT " value %" PetscInt_FMT " rowlength %" PetscInt_FMT,i,rlen[i],B->cmap->n); 117d4002b98SHong Zhang } 118d4002b98SHong Zhang } 119d4002b98SHong Zhang 120d4002b98SHong Zhang B->preallocated = PETSC_TRUE; 121d4002b98SHong Zhang 122d4002b98SHong Zhang b = (Mat_SeqSELL*)B->data; 123d4002b98SHong Zhang 124faa75363SBarry Smith totalslices = PetscCeilInt(B->rmap->n,8); 125d4002b98SHong Zhang b->totalslices = totalslices; 126d4002b98SHong Zhang if (!skipallocation) { 1279566063dSJacob Faibussowitsch if (B->rmap->n & 0x07) PetscCall(PetscInfo(B,"Padding rows to the SEQSELL matrix because the number of rows is not the multiple of 8 (value %" PetscInt_FMT ")\n",B->rmap->n)); 128d4002b98SHong Zhang 129d4002b98SHong Zhang if (!b->sliidx) { /* sliidx gives the starting index of each slice, the last element is the total space allocated */ 1309566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totalslices+1,&b->sliidx)); 1319566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)B,(totalslices+1)*sizeof(PetscInt))); 132d4002b98SHong Zhang } 133d4002b98SHong Zhang if (!rlen) { /* if rlen is not provided, allocate same space for all the slices */ 134d4002b98SHong Zhang if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 10; 135d4002b98SHong Zhang else if (maxallocrow < 0) maxallocrow = 1; 136d4002b98SHong Zhang for (i=0; i<=totalslices; i++) b->sliidx[i] = i*8*maxallocrow; 137d4002b98SHong Zhang } else { 138d4002b98SHong Zhang maxallocrow = 0; 139d4002b98SHong Zhang b->sliidx[0] = 0; 140d4002b98SHong Zhang for (i=1; i<totalslices; i++) { 141d4002b98SHong Zhang b->sliidx[i] = 0; 142d4002b98SHong Zhang for (j=0;j<8;j++) { 143d4002b98SHong Zhang b->sliidx[i] = PetscMax(b->sliidx[i],rlen[8*(i-1)+j]); 144d4002b98SHong Zhang } 145d4002b98SHong Zhang maxallocrow = PetscMax(b->sliidx[i],maxallocrow); 1469566063dSJacob Faibussowitsch PetscCall(PetscIntSumError(b->sliidx[i-1],8*b->sliidx[i],&b->sliidx[i])); 147d4002b98SHong Zhang } 148d4002b98SHong Zhang /* last slice */ 149d4002b98SHong Zhang b->sliidx[totalslices] = 0; 150d4002b98SHong Zhang for (j=(totalslices-1)*8;j<B->rmap->n;j++) b->sliidx[totalslices] = PetscMax(b->sliidx[totalslices],rlen[j]); 151d4002b98SHong Zhang maxallocrow = PetscMax(b->sliidx[totalslices],maxallocrow); 152d4002b98SHong Zhang b->sliidx[totalslices] = b->sliidx[totalslices-1] + 8*b->sliidx[totalslices]; 153d4002b98SHong Zhang } 154d4002b98SHong Zhang 155d4002b98SHong Zhang /* allocate space for val, colidx, rlen */ 156d4002b98SHong Zhang /* FIXME: should B's old memory be unlogged? */ 1579566063dSJacob Faibussowitsch PetscCall(MatSeqXSELLFreeSELL(B,&b->val,&b->colidx)); 158d4002b98SHong Zhang /* FIXME: assuming an element of the bit array takes 8 bits */ 1599566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(b->sliidx[totalslices],&b->val,b->sliidx[totalslices],&b->colidx)); 1609566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)B,b->sliidx[totalslices]*(sizeof(PetscScalar)+sizeof(PetscInt)))); 161d4002b98SHong 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. */ 1629566063dSJacob Faibussowitsch PetscCall(PetscCalloc1(8*totalslices,&b->rlen)); 1639566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)B,8*totalslices*sizeof(PetscInt))); 164d4002b98SHong Zhang 165d4002b98SHong Zhang b->singlemalloc = PETSC_TRUE; 166d4002b98SHong Zhang b->free_val = PETSC_TRUE; 167d4002b98SHong Zhang b->free_colidx = PETSC_TRUE; 168d4002b98SHong Zhang } else { 169d4002b98SHong Zhang b->free_val = PETSC_FALSE; 170d4002b98SHong Zhang b->free_colidx = PETSC_FALSE; 171d4002b98SHong Zhang } 172d4002b98SHong Zhang 173d4002b98SHong Zhang b->nz = 0; 174d4002b98SHong Zhang b->maxallocrow = maxallocrow; 175d4002b98SHong Zhang b->rlenmax = maxallocrow; 176d4002b98SHong Zhang b->maxallocmat = b->sliidx[totalslices]; 177d4002b98SHong Zhang B->info.nz_unneeded = (double)b->maxallocmat; 178d4002b98SHong Zhang if (realalloc) { 1799566063dSJacob Faibussowitsch PetscCall(MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE)); 180d4002b98SHong Zhang } 181d4002b98SHong Zhang PetscFunctionReturn(0); 182d4002b98SHong Zhang } 183d4002b98SHong Zhang 1846108893eSStefano Zampini PetscErrorCode MatGetRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 1856108893eSStefano Zampini { 1866108893eSStefano Zampini Mat_SeqSELL *a = (Mat_SeqSELL*)A->data; 1876108893eSStefano Zampini PetscInt shift; 1886108893eSStefano Zampini 1896108893eSStefano Zampini PetscFunctionBegin; 190aed4548fSBarry Smith PetscCheck(row >= 0 && row < A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %" PetscInt_FMT " out of range",row); 1916108893eSStefano Zampini if (nz) *nz = a->rlen[row]; 1926108893eSStefano Zampini shift = a->sliidx[row>>3]+(row&0x07); 1936108893eSStefano Zampini if (!a->getrowcols) { 1946108893eSStefano Zampini 1959566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(a->rlenmax,&a->getrowcols,a->rlenmax,&a->getrowvals)); 1966108893eSStefano Zampini } 1976108893eSStefano Zampini if (idx) { 1986108893eSStefano Zampini PetscInt j; 1996108893eSStefano Zampini for (j=0; j<a->rlen[row]; j++) a->getrowcols[j] = a->colidx[shift+8*j]; 2006108893eSStefano Zampini *idx = a->getrowcols; 2016108893eSStefano Zampini } 2026108893eSStefano Zampini if (v) { 2036108893eSStefano Zampini PetscInt j; 2046108893eSStefano Zampini for (j=0; j<a->rlen[row]; j++) a->getrowvals[j] = a->val[shift+8*j]; 2056108893eSStefano Zampini *v = a->getrowvals; 2066108893eSStefano Zampini } 2076108893eSStefano Zampini PetscFunctionReturn(0); 2086108893eSStefano Zampini } 2096108893eSStefano Zampini 2106108893eSStefano Zampini PetscErrorCode MatRestoreRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 2116108893eSStefano Zampini { 2126108893eSStefano Zampini PetscFunctionBegin; 2136108893eSStefano Zampini PetscFunctionReturn(0); 2146108893eSStefano Zampini } 2156108893eSStefano Zampini 216d4002b98SHong Zhang PetscErrorCode MatConvert_SeqSELL_SeqAIJ(Mat A, MatType newtype,MatReuse reuse,Mat *newmat) 217d4002b98SHong Zhang { 218d4002b98SHong Zhang Mat B; 219d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 220e3f1f374SStefano Zampini PetscInt i; 221d4002b98SHong Zhang 222d4002b98SHong Zhang PetscFunctionBegin; 223ad013a7bSRichard Tran Mills if (reuse == MAT_REUSE_MATRIX) { 224ad013a7bSRichard Tran Mills B = *newmat; 2259566063dSJacob Faibussowitsch PetscCall(MatZeroEntries(B)); 226ad013a7bSRichard Tran Mills } else { 2279566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A),&B)); 2289566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N)); 2299566063dSJacob Faibussowitsch PetscCall(MatSetType(B,MATSEQAIJ)); 2309566063dSJacob Faibussowitsch PetscCall(MatSeqAIJSetPreallocation(B,0,a->rlen)); 231ad013a7bSRichard Tran Mills } 232d4002b98SHong Zhang 233e3f1f374SStefano Zampini for (i=0; i<A->rmap->n; i++) { 234e108cb99SStefano Zampini PetscInt nz = 0,*cols = NULL; 235e108cb99SStefano Zampini PetscScalar *vals = NULL; 236e3f1f374SStefano Zampini 2379566063dSJacob Faibussowitsch PetscCall(MatGetRow_SeqSELL(A,i,&nz,&cols,&vals)); 2389566063dSJacob Faibussowitsch PetscCall(MatSetValues(B,1,&i,nz,cols,vals,INSERT_VALUES)); 2399566063dSJacob Faibussowitsch PetscCall(MatRestoreRow_SeqSELL(A,i,&nz,&cols,&vals)); 240d4002b98SHong Zhang } 241e3f1f374SStefano Zampini 2429566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY)); 2439566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY)); 244d4002b98SHong Zhang B->rmap->bs = A->rmap->bs; 245d4002b98SHong Zhang 246d4002b98SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 2479566063dSJacob Faibussowitsch PetscCall(MatHeaderReplace(A,&B)); 248d4002b98SHong Zhang } else { 249d4002b98SHong Zhang *newmat = B; 250d4002b98SHong Zhang } 251d4002b98SHong Zhang PetscFunctionReturn(0); 252d4002b98SHong Zhang } 253d4002b98SHong Zhang 254d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/aij.h> 255d4002b98SHong Zhang 256d4002b98SHong Zhang PetscErrorCode MatConvert_SeqAIJ_SeqSELL(Mat A,MatType newtype,MatReuse reuse,Mat *newmat) 257d4002b98SHong Zhang { 258d4002b98SHong Zhang Mat B; 259d4002b98SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 260d4002b98SHong Zhang PetscInt *ai=a->i,m=A->rmap->N,n=A->cmap->N,i,*rowlengths,row,ncols; 261d4002b98SHong Zhang const PetscInt *cols; 262d4002b98SHong Zhang const PetscScalar *vals; 263d4002b98SHong Zhang 264d4002b98SHong Zhang PetscFunctionBegin; 265ad013a7bSRichard Tran Mills 266ad013a7bSRichard Tran Mills if (reuse == MAT_REUSE_MATRIX) { 267ad013a7bSRichard Tran Mills B = *newmat; 268ad013a7bSRichard Tran Mills } else { 269d5e5b2e5SBarry Smith if (PetscDefined(USE_DEBUG) || !a->ilen) { 2709566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(m,&rowlengths)); 271d4002b98SHong Zhang for (i=0; i<m; i++) { 272d4002b98SHong Zhang rowlengths[i] = ai[i+1] - ai[i]; 273d4002b98SHong Zhang } 274d5e5b2e5SBarry Smith } 275d5e5b2e5SBarry Smith if (PetscDefined(USE_DEBUG) && a->ilen) { 276d5e5b2e5SBarry Smith PetscBool eq; 2779566063dSJacob Faibussowitsch PetscCall(PetscMemcmp(rowlengths,a->ilen,m*sizeof(PetscInt),&eq)); 27828b400f6SJacob Faibussowitsch PetscCheck(eq,PETSC_COMM_SELF,PETSC_ERR_PLIB,"SeqAIJ ilen array incorrect"); 2799566063dSJacob Faibussowitsch PetscCall(PetscFree(rowlengths)); 280d5e5b2e5SBarry Smith rowlengths = a->ilen; 281d5e5b2e5SBarry Smith } else if (a->ilen) rowlengths = a->ilen; 2829566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A),&B)); 2839566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B,m,n,m,n)); 2849566063dSJacob Faibussowitsch PetscCall(MatSetType(B,MATSEQSELL)); 2859566063dSJacob Faibussowitsch PetscCall(MatSeqSELLSetPreallocation(B,0,rowlengths)); 2869566063dSJacob Faibussowitsch if (rowlengths != a->ilen) PetscCall(PetscFree(rowlengths)); 287ad013a7bSRichard Tran Mills } 288d4002b98SHong Zhang 289d4002b98SHong Zhang for (row=0; row<m; row++) { 2909566063dSJacob Faibussowitsch PetscCall(MatGetRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals)); 2919566063dSJacob Faibussowitsch PetscCall(MatSetValues_SeqSELL(B,1,&row,ncols,cols,vals,INSERT_VALUES)); 2929566063dSJacob Faibussowitsch PetscCall(MatRestoreRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals)); 293d4002b98SHong Zhang } 2949566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY)); 2959566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY)); 296d4002b98SHong Zhang B->rmap->bs = A->rmap->bs; 297d4002b98SHong Zhang 298d4002b98SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 2999566063dSJacob Faibussowitsch PetscCall(MatHeaderReplace(A,&B)); 300d4002b98SHong Zhang } else { 301d4002b98SHong Zhang *newmat = B; 302d4002b98SHong Zhang } 303d4002b98SHong Zhang PetscFunctionReturn(0); 304d4002b98SHong Zhang } 305d4002b98SHong Zhang 306d4002b98SHong Zhang PetscErrorCode MatMult_SeqSELL(Mat A,Vec xx,Vec yy) 307d4002b98SHong Zhang { 308d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 309d4002b98SHong Zhang PetscScalar *y; 310d4002b98SHong Zhang const PetscScalar *x; 311d4002b98SHong Zhang const MatScalar *aval=a->val; 312d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 313d4002b98SHong Zhang const PetscInt *acolidx=a->colidx; 3147285fed1SHong Zhang PetscInt i,j; 315d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 316d4002b98SHong Zhang __m512d vec_x,vec_y,vec_vals; 317d4002b98SHong Zhang __m256i vec_idx; 318d4002b98SHong Zhang __mmask8 mask; 319d4002b98SHong Zhang __m512d vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4; 320d4002b98SHong Zhang __m256i vec_idx2,vec_idx3,vec_idx4; 3215f70456aSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX2__) && defined(__FMA__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 322a48a6482SHong Zhang __m128i vec_idx; 323a48a6482SHong Zhang __m256d vec_x,vec_y,vec_y2,vec_vals; 324a48a6482SHong Zhang MatScalar yval; 325a48a6482SHong Zhang PetscInt r,rows_left,row,nnz_in_row; 32621cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 327d4002b98SHong Zhang __m128d vec_x_tmp; 328d4002b98SHong Zhang __m256d vec_x,vec_y,vec_y2,vec_vals; 329d4002b98SHong Zhang MatScalar yval; 330d4002b98SHong Zhang PetscInt r,rows_left,row,nnz_in_row; 331d4002b98SHong Zhang #else 332d4002b98SHong Zhang PetscScalar sum[8]; 333d4002b98SHong Zhang #endif 334d4002b98SHong Zhang 335d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 336d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval) 337d4002b98SHong Zhang #endif 338d4002b98SHong Zhang 339d4002b98SHong Zhang PetscFunctionBegin; 3409566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(xx,&x)); 3419566063dSJacob Faibussowitsch PetscCall(VecGetArray(yy,&y)); 342d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 343d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 344d4002b98SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 345d4002b98SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 346d4002b98SHong Zhang 347d4002b98SHong Zhang vec_y = _mm512_setzero_pd(); 348d4002b98SHong Zhang vec_y2 = _mm512_setzero_pd(); 349d4002b98SHong Zhang vec_y3 = _mm512_setzero_pd(); 350d4002b98SHong Zhang vec_y4 = _mm512_setzero_pd(); 351d4002b98SHong Zhang 35238efe8efSHong Zhang j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */ 353d4002b98SHong Zhang switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) { 354d4002b98SHong Zhang case 3: 355d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 356d4002b98SHong Zhang acolidx += 8; aval += 8; 357d4002b98SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 358d4002b98SHong Zhang acolidx += 8; aval += 8; 359d4002b98SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 360d4002b98SHong Zhang acolidx += 8; aval += 8; 361d4002b98SHong Zhang j += 3; 362d4002b98SHong Zhang break; 363d4002b98SHong Zhang case 2: 364d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 365d4002b98SHong Zhang acolidx += 8; aval += 8; 366d4002b98SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 367d4002b98SHong Zhang acolidx += 8; aval += 8; 368d4002b98SHong Zhang j += 2; 369d4002b98SHong Zhang break; 370d4002b98SHong Zhang case 1: 371d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 372d4002b98SHong Zhang acolidx += 8; aval += 8; 373d4002b98SHong Zhang j += 1; 374d4002b98SHong Zhang break; 375d4002b98SHong Zhang } 376d4002b98SHong Zhang #pragma novector 377d4002b98SHong Zhang for (; j<(a->sliidx[i+1]>>3); j+=4) { 378d4002b98SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 379d4002b98SHong Zhang acolidx += 8; aval += 8; 380d4002b98SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 381d4002b98SHong Zhang acolidx += 8; aval += 8; 382d4002b98SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 383d4002b98SHong Zhang acolidx += 8; aval += 8; 384d4002b98SHong Zhang AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4); 385d4002b98SHong Zhang acolidx += 8; aval += 8; 386d4002b98SHong Zhang } 387d4002b98SHong Zhang 388d4002b98SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y2); 389d4002b98SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y3); 390d4002b98SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y4); 391d4002b98SHong Zhang if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */ 392d4002b98SHong Zhang mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07))); 393ef588d5cSRichard Tran Mills _mm512_mask_storeu_pd(&y[8*i],mask,vec_y); 394d4002b98SHong Zhang } else { 395ef588d5cSRichard Tran Mills _mm512_storeu_pd(&y[8*i],vec_y); 396d4002b98SHong Zhang } 397d4002b98SHong Zhang } 3985f70456aSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX2__) && defined(__FMA__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 399a48a6482SHong Zhang for (i=0; i<totalslices; i++) { /* loop over full slices */ 400a48a6482SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 401a48a6482SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 402a48a6482SHong Zhang 403a48a6482SHong Zhang /* last slice may have padding rows. Don't use vectorization. */ 404a48a6482SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 405a48a6482SHong Zhang rows_left = A->rmap->n - 8*i; 406a48a6482SHong Zhang for (r=0; r<rows_left; ++r) { 407a48a6482SHong Zhang yval = (MatScalar)0; 408a48a6482SHong Zhang row = 8*i + r; 409a48a6482SHong Zhang nnz_in_row = a->rlen[row]; 410a48a6482SHong Zhang for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]]; 411a48a6482SHong Zhang y[row] = yval; 412a48a6482SHong Zhang } 413a48a6482SHong Zhang break; 414a48a6482SHong Zhang } 415a48a6482SHong Zhang 416a48a6482SHong Zhang vec_y = _mm256_setzero_pd(); 417a48a6482SHong Zhang vec_y2 = _mm256_setzero_pd(); 418a48a6482SHong Zhang 419a48a6482SHong Zhang /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */ 420a48a6482SHong Zhang #pragma novector 421a48a6482SHong Zhang #pragma unroll(2) 422a48a6482SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 423a48a6482SHong Zhang AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 424a48a6482SHong Zhang aval += 4; acolidx += 4; 425a48a6482SHong Zhang AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y2); 426a48a6482SHong Zhang aval += 4; acolidx += 4; 427a48a6482SHong Zhang } 428a48a6482SHong Zhang 429ef588d5cSRichard Tran Mills _mm256_storeu_pd(y+i*8,vec_y); 430ef588d5cSRichard Tran Mills _mm256_storeu_pd(y+i*8+4,vec_y2); 431a48a6482SHong Zhang } 43221cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 433d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over full slices */ 434d4002b98SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 435d4002b98SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 436d4002b98SHong Zhang 437d4002b98SHong Zhang vec_y = _mm256_setzero_pd(); 438d4002b98SHong Zhang vec_y2 = _mm256_setzero_pd(); 439d4002b98SHong Zhang 440d4002b98SHong Zhang /* last slice may have padding rows. Don't use vectorization. */ 441d4002b98SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 442d4002b98SHong Zhang rows_left = A->rmap->n - 8*i; 443d4002b98SHong Zhang for (r=0; r<rows_left; ++r) { 444d4002b98SHong Zhang yval = (MatScalar)0; 445d4002b98SHong Zhang row = 8*i + r; 446d4002b98SHong Zhang nnz_in_row = a->rlen[row]; 447d4002b98SHong Zhang for (j=0; j<nnz_in_row; ++j) yval += aval[8*j + r] * x[acolidx[8*j + r]]; 448d4002b98SHong Zhang y[row] = yval; 449d4002b98SHong Zhang } 450d4002b98SHong Zhang break; 451d4002b98SHong Zhang } 452d4002b98SHong Zhang 453d4002b98SHong Zhang /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */ 454a48a6482SHong Zhang #pragma novector 455a48a6482SHong Zhang #pragma unroll(2) 4567285fed1SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 457d4002b98SHong Zhang vec_vals = _mm256_loadu_pd(aval); 458*165f9cc3SJed Brown vec_x_tmp = _mm_setzero_pd(); 459d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 460d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 461d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 462d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 463d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 464d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 465d4002b98SHong Zhang vec_y = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y); 466d4002b98SHong Zhang aval += 4; 467d4002b98SHong Zhang 468d4002b98SHong Zhang vec_vals = _mm256_loadu_pd(aval); 469d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 470d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 471d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 472d4002b98SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 473d4002b98SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 474d4002b98SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 475d4002b98SHong Zhang vec_y2 = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2); 476d4002b98SHong Zhang aval += 4; 477d4002b98SHong Zhang } 478d4002b98SHong Zhang 479d4002b98SHong Zhang _mm256_storeu_pd(y + i*8, vec_y); 480d4002b98SHong Zhang _mm256_storeu_pd(y + i*8 + 4, vec_y2); 481d4002b98SHong Zhang } 482d4002b98SHong Zhang #else 483d4002b98SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 484d4002b98SHong Zhang for (j=0; j<8; j++) sum[j] = 0.0; 485d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 486d4002b98SHong Zhang sum[0] += aval[j] * x[acolidx[j]]; 487d4002b98SHong Zhang sum[1] += aval[j+1] * x[acolidx[j+1]]; 488d4002b98SHong Zhang sum[2] += aval[j+2] * x[acolidx[j+2]]; 489d4002b98SHong Zhang sum[3] += aval[j+3] * x[acolidx[j+3]]; 490d4002b98SHong Zhang sum[4] += aval[j+4] * x[acolidx[j+4]]; 491d4002b98SHong Zhang sum[5] += aval[j+5] * x[acolidx[j+5]]; 492d4002b98SHong Zhang sum[6] += aval[j+6] * x[acolidx[j+6]]; 493d4002b98SHong Zhang sum[7] += aval[j+7] * x[acolidx[j+7]]; 494d4002b98SHong Zhang } 495d4002b98SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */ 496d4002b98SHong Zhang for (j=0; j<(A->rmap->n & 0x07); j++) y[8*i+j] = sum[j]; 497d4002b98SHong Zhang } else { 4987285fed1SHong Zhang for (j=0; j<8; j++) y[8*i+j] = sum[j]; 499d4002b98SHong Zhang } 500d4002b98SHong Zhang } 501d4002b98SHong Zhang #endif 502d4002b98SHong Zhang 5039566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0*a->nz-a->nonzerorowcnt)); /* theoretical minimal FLOPs */ 5049566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(xx,&x)); 5059566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(yy,&y)); 506d4002b98SHong Zhang PetscFunctionReturn(0); 507d4002b98SHong Zhang } 508d4002b98SHong Zhang 509d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/ftn-kernels/fmultadd.h> 510d4002b98SHong Zhang PetscErrorCode MatMultAdd_SeqSELL(Mat A,Vec xx,Vec yy,Vec zz) 511d4002b98SHong Zhang { 512d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 513d4002b98SHong Zhang PetscScalar *y,*z; 514d4002b98SHong Zhang const PetscScalar *x; 515d4002b98SHong Zhang const MatScalar *aval=a->val; 516d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 517d4002b98SHong Zhang const PetscInt *acolidx=a->colidx; 518d4002b98SHong Zhang PetscInt i,j; 519d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 5207285fed1SHong Zhang __m512d vec_x,vec_y,vec_vals; 521d4002b98SHong Zhang __m256i vec_idx; 522d4002b98SHong Zhang __mmask8 mask; 5237285fed1SHong Zhang __m512d vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4; 5247285fed1SHong Zhang __m256i vec_idx2,vec_idx3,vec_idx4; 52521cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 5267285fed1SHong Zhang __m128d vec_x_tmp; 5277285fed1SHong Zhang __m256d vec_x,vec_y,vec_y2,vec_vals; 5287285fed1SHong Zhang MatScalar yval; 5297285fed1SHong Zhang PetscInt r,row,nnz_in_row; 530d4002b98SHong Zhang #else 531d4002b98SHong Zhang PetscScalar sum[8]; 532d4002b98SHong Zhang #endif 533d4002b98SHong Zhang 534d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 535d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval) 536d4002b98SHong Zhang #endif 537d4002b98SHong Zhang 538d4002b98SHong Zhang PetscFunctionBegin; 5399566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(xx,&x)); 5409566063dSJacob Faibussowitsch PetscCall(VecGetArrayPair(yy,zz,&y,&z)); 541d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 5427285fed1SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 5437285fed1SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 5447285fed1SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 5457285fed1SHong Zhang 546d4002b98SHong Zhang if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */ 547d4002b98SHong Zhang mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07))); 548ef588d5cSRichard Tran Mills vec_y = _mm512_mask_loadu_pd(vec_y,mask,&y[8*i]); 5497285fed1SHong Zhang } else { 550ef588d5cSRichard Tran Mills vec_y = _mm512_loadu_pd(&y[8*i]); 5517285fed1SHong Zhang } 5527285fed1SHong Zhang vec_y2 = _mm512_setzero_pd(); 5537285fed1SHong Zhang vec_y3 = _mm512_setzero_pd(); 5547285fed1SHong Zhang vec_y4 = _mm512_setzero_pd(); 5557285fed1SHong Zhang 5567285fed1SHong Zhang j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */ 5577285fed1SHong Zhang switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) { 5587285fed1SHong Zhang case 3: 5597285fed1SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 5607285fed1SHong Zhang acolidx += 8; aval += 8; 5617285fed1SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 5627285fed1SHong Zhang acolidx += 8; aval += 8; 5637285fed1SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 5647285fed1SHong Zhang acolidx += 8; aval += 8; 5657285fed1SHong Zhang j += 3; 5667285fed1SHong Zhang break; 5677285fed1SHong Zhang case 2: 5687285fed1SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 5697285fed1SHong Zhang acolidx += 8; aval += 8; 5707285fed1SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 5717285fed1SHong Zhang acolidx += 8; aval += 8; 5727285fed1SHong Zhang j += 2; 5737285fed1SHong Zhang break; 5747285fed1SHong Zhang case 1: 5757285fed1SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 5767285fed1SHong Zhang acolidx += 8; aval += 8; 5777285fed1SHong Zhang j += 1; 5787285fed1SHong Zhang break; 5797285fed1SHong Zhang } 5807285fed1SHong Zhang #pragma novector 5817285fed1SHong Zhang for (; j<(a->sliidx[i+1]>>3); j+=4) { 5827285fed1SHong Zhang AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y); 5837285fed1SHong Zhang acolidx += 8; aval += 8; 5847285fed1SHong Zhang AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2); 5857285fed1SHong Zhang acolidx += 8; aval += 8; 5867285fed1SHong Zhang AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3); 5877285fed1SHong Zhang acolidx += 8; aval += 8; 5887285fed1SHong Zhang AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4); 5897285fed1SHong Zhang acolidx += 8; aval += 8; 5907285fed1SHong Zhang } 5917285fed1SHong Zhang 5927285fed1SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y2); 5937285fed1SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y3); 5947285fed1SHong Zhang vec_y = _mm512_add_pd(vec_y,vec_y4); 5957285fed1SHong Zhang if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */ 596ef588d5cSRichard Tran Mills _mm512_mask_storeu_pd(&z[8*i],mask,vec_y); 597d4002b98SHong Zhang } else { 598ef588d5cSRichard Tran Mills _mm512_storeu_pd(&z[8*i],vec_y); 599d4002b98SHong Zhang } 6007285fed1SHong Zhang } 60121cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES) 6027285fed1SHong Zhang for (i=0; i<totalslices; i++) { /* loop over full slices */ 6037285fed1SHong Zhang PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 6047285fed1SHong Zhang PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0); 6057285fed1SHong Zhang 6067285fed1SHong Zhang /* last slice may have padding rows. Don't use vectorization. */ 6077285fed1SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 6087285fed1SHong Zhang for (r=0; r<(A->rmap->n & 0x07); ++r) { 6097285fed1SHong Zhang row = 8*i + r; 6107285fed1SHong Zhang yval = (MatScalar)0.0; 6117285fed1SHong Zhang nnz_in_row = a->rlen[row]; 6127285fed1SHong Zhang for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]]; 6137285fed1SHong Zhang z[row] = y[row] + yval; 6147285fed1SHong Zhang } 6157285fed1SHong Zhang break; 6167285fed1SHong Zhang } 6177285fed1SHong Zhang 6187285fed1SHong Zhang vec_y = _mm256_loadu_pd(y+8*i); 6197285fed1SHong Zhang vec_y2 = _mm256_loadu_pd(y+8*i+4); 6207285fed1SHong Zhang 6217285fed1SHong Zhang /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */ 6227285fed1SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 6237285fed1SHong Zhang vec_vals = _mm256_loadu_pd(aval); 624*165f9cc3SJed Brown vec_x_tmp = _mm_setzero_pd(); 6257285fed1SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 6267285fed1SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 627*165f9cc3SJed Brown vec_x = _mm256_setzero_pd(); 6287285fed1SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 6297285fed1SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 6307285fed1SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 6317285fed1SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 6327285fed1SHong Zhang vec_y = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y); 6337285fed1SHong Zhang aval += 4; 6347285fed1SHong Zhang 6357285fed1SHong Zhang vec_vals = _mm256_loadu_pd(aval); 6367285fed1SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 6377285fed1SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 6387285fed1SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,0); 6397285fed1SHong Zhang vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++); 6407285fed1SHong Zhang vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++); 6417285fed1SHong Zhang vec_x = _mm256_insertf128_pd(vec_x,vec_x_tmp,1); 6427285fed1SHong Zhang vec_y2 = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2); 6437285fed1SHong Zhang aval += 4; 6447285fed1SHong Zhang } 6457285fed1SHong Zhang 6467285fed1SHong Zhang _mm256_storeu_pd(z+i*8,vec_y); 6477285fed1SHong Zhang _mm256_storeu_pd(z+i*8+4,vec_y2); 6487285fed1SHong Zhang } 649d4002b98SHong Zhang #else 6507285fed1SHong Zhang for (i=0; i<totalslices; i++) { /* loop over slices */ 6517285fed1SHong Zhang for (j=0; j<8; j++) sum[j] = 0.0; 652d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 653d4002b98SHong Zhang sum[0] += aval[j] * x[acolidx[j]]; 654d4002b98SHong Zhang sum[1] += aval[j+1] * x[acolidx[j+1]]; 655d4002b98SHong Zhang sum[2] += aval[j+2] * x[acolidx[j+2]]; 656d4002b98SHong Zhang sum[3] += aval[j+3] * x[acolidx[j+3]]; 657d4002b98SHong Zhang sum[4] += aval[j+4] * x[acolidx[j+4]]; 658d4002b98SHong Zhang sum[5] += aval[j+5] * x[acolidx[j+5]]; 659d4002b98SHong Zhang sum[6] += aval[j+6] * x[acolidx[j+6]]; 660d4002b98SHong Zhang sum[7] += aval[j+7] * x[acolidx[j+7]]; 661d4002b98SHong Zhang } 6627285fed1SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 6637285fed1SHong Zhang for (j=0; j<(A->rmap->n & 0x07); j++) z[8*i+j] = y[8*i+j] + sum[j]; 664d4002b98SHong Zhang } else { 6657285fed1SHong Zhang for (j=0; j<8; j++) z[8*i+j] = y[8*i+j] + sum[j]; 6667285fed1SHong Zhang } 667d4002b98SHong Zhang } 668d4002b98SHong Zhang #endif 669d4002b98SHong Zhang 6709566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0*a->nz)); 6719566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(xx,&x)); 6729566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayPair(yy,zz,&y,&z)); 673d4002b98SHong Zhang PetscFunctionReturn(0); 674d4002b98SHong Zhang } 675d4002b98SHong Zhang 676d4002b98SHong Zhang PetscErrorCode MatMultTransposeAdd_SeqSELL(Mat A,Vec xx,Vec zz,Vec yy) 677d4002b98SHong Zhang { 678d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 679d4002b98SHong Zhang PetscScalar *y; 680d4002b98SHong Zhang const PetscScalar *x; 681d4002b98SHong Zhang const MatScalar *aval=a->val; 682d4002b98SHong Zhang const PetscInt *acolidx=a->colidx; 6837285fed1SHong Zhang PetscInt i,j,r,row,nnz_in_row,totalslices=a->totalslices; 684d4002b98SHong Zhang 685d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 686d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval) 687d4002b98SHong Zhang #endif 688d4002b98SHong Zhang 689d4002b98SHong Zhang PetscFunctionBegin; 6909fc32365SStefano Zampini if (A->symmetric) { 6919566063dSJacob Faibussowitsch PetscCall(MatMultAdd_SeqSELL(A,xx,zz,yy)); 6929fc32365SStefano Zampini PetscFunctionReturn(0); 6939fc32365SStefano Zampini } 6949566063dSJacob Faibussowitsch if (zz != yy) PetscCall(VecCopy(zz,yy)); 6959566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(xx,&x)); 6969566063dSJacob Faibussowitsch PetscCall(VecGetArray(yy,&y)); 697d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 6987285fed1SHong Zhang if (i == totalslices-1 && (A->rmap->n & 0x07)) { 6997285fed1SHong Zhang for (r=0; r<(A->rmap->n & 0x07); ++r) { 7007285fed1SHong Zhang row = 8*i + r; 7017285fed1SHong Zhang nnz_in_row = a->rlen[row]; 7027285fed1SHong Zhang for (j=0; j<nnz_in_row; ++j) y[acolidx[8*j+r]] += aval[8*j+r] * x[row]; 7037285fed1SHong Zhang } 7047285fed1SHong Zhang break; 7057285fed1SHong Zhang } 7067285fed1SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) { 7077285fed1SHong Zhang y[acolidx[j]] += aval[j] * x[8*i]; 7087285fed1SHong Zhang y[acolidx[j+1]] += aval[j+1] * x[8*i+1]; 7097285fed1SHong Zhang y[acolidx[j+2]] += aval[j+2] * x[8*i+2]; 7107285fed1SHong Zhang y[acolidx[j+3]] += aval[j+3] * x[8*i+3]; 7117285fed1SHong Zhang y[acolidx[j+4]] += aval[j+4] * x[8*i+4]; 7127285fed1SHong Zhang y[acolidx[j+5]] += aval[j+5] * x[8*i+5]; 7137285fed1SHong Zhang y[acolidx[j+6]] += aval[j+6] * x[8*i+6]; 7147285fed1SHong Zhang y[acolidx[j+7]] += aval[j+7] * x[8*i+7]; 715d4002b98SHong Zhang } 716d4002b98SHong Zhang } 7179566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0*a->sliidx[a->totalslices])); 7189566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(xx,&x)); 7199566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(yy,&y)); 720d4002b98SHong Zhang PetscFunctionReturn(0); 721d4002b98SHong Zhang } 722d4002b98SHong Zhang 723d4002b98SHong Zhang PetscErrorCode MatMultTranspose_SeqSELL(Mat A,Vec xx,Vec yy) 724d4002b98SHong Zhang { 725d4002b98SHong Zhang PetscFunctionBegin; 7269fc32365SStefano Zampini if (A->symmetric) { 7279566063dSJacob Faibussowitsch PetscCall(MatMult_SeqSELL(A,xx,yy)); 7289fc32365SStefano Zampini } else { 7299566063dSJacob Faibussowitsch PetscCall(VecSet(yy,0.0)); 7309566063dSJacob Faibussowitsch PetscCall(MatMultTransposeAdd_SeqSELL(A,xx,yy,yy)); 7319fc32365SStefano Zampini } 732d4002b98SHong Zhang PetscFunctionReturn(0); 733d4002b98SHong Zhang } 734d4002b98SHong Zhang 735d4002b98SHong Zhang /* 736d4002b98SHong Zhang Checks for missing diagonals 737d4002b98SHong Zhang */ 738d4002b98SHong Zhang PetscErrorCode MatMissingDiagonal_SeqSELL(Mat A,PetscBool *missing,PetscInt *d) 739d4002b98SHong Zhang { 740d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 741d4002b98SHong Zhang PetscInt *diag,i; 742d4002b98SHong Zhang 743d4002b98SHong Zhang PetscFunctionBegin; 744d4002b98SHong Zhang *missing = PETSC_FALSE; 745d4002b98SHong Zhang if (A->rmap->n > 0 && !(a->colidx)) { 746d4002b98SHong Zhang *missing = PETSC_TRUE; 747d4002b98SHong Zhang if (d) *d = 0; 7489566063dSJacob Faibussowitsch PetscCall(PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n")); 749d4002b98SHong Zhang } else { 750d4002b98SHong Zhang diag = a->diag; 751d4002b98SHong Zhang for (i=0; i<A->rmap->n; i++) { 752d4002b98SHong Zhang if (diag[i] == -1) { 753d4002b98SHong Zhang *missing = PETSC_TRUE; 754d4002b98SHong Zhang if (d) *d = i; 7559566063dSJacob Faibussowitsch PetscCall(PetscInfo(A,"Matrix is missing diagonal number %" PetscInt_FMT "\n",i)); 756d4002b98SHong Zhang break; 757d4002b98SHong Zhang } 758d4002b98SHong Zhang } 759d4002b98SHong Zhang } 760d4002b98SHong Zhang PetscFunctionReturn(0); 761d4002b98SHong Zhang } 762d4002b98SHong Zhang 763d4002b98SHong Zhang PetscErrorCode MatMarkDiagonal_SeqSELL(Mat A) 764d4002b98SHong Zhang { 765d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 766d4002b98SHong Zhang PetscInt i,j,m=A->rmap->n,shift; 767d4002b98SHong Zhang 768d4002b98SHong Zhang PetscFunctionBegin; 769d4002b98SHong Zhang if (!a->diag) { 7709566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(m,&a->diag)); 7719566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)A,m*sizeof(PetscInt))); 772d4002b98SHong Zhang a->free_diag = PETSC_TRUE; 773d4002b98SHong Zhang } 774d4002b98SHong Zhang for (i=0; i<m; i++) { /* loop over rows */ 775d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 776d4002b98SHong Zhang a->diag[i] = -1; 777d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 778d4002b98SHong Zhang if (a->colidx[shift+j*8] == i) { 779d4002b98SHong Zhang a->diag[i] = shift+j*8; 780d4002b98SHong Zhang break; 781d4002b98SHong Zhang } 782d4002b98SHong Zhang } 783d4002b98SHong Zhang } 784d4002b98SHong Zhang PetscFunctionReturn(0); 785d4002b98SHong Zhang } 786d4002b98SHong Zhang 787d4002b98SHong Zhang /* 788d4002b98SHong Zhang Negative shift indicates do not generate an error if there is a zero diagonal, just invert it anyways 789d4002b98SHong Zhang */ 790d4002b98SHong Zhang PetscErrorCode MatInvertDiagonal_SeqSELL(Mat A,PetscScalar omega,PetscScalar fshift) 791d4002b98SHong Zhang { 792d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*) A->data; 793d4002b98SHong Zhang PetscInt i,*diag,m = A->rmap->n; 794d4002b98SHong Zhang MatScalar *val = a->val; 795d4002b98SHong Zhang PetscScalar *idiag,*mdiag; 796d4002b98SHong Zhang 797d4002b98SHong Zhang PetscFunctionBegin; 798d4002b98SHong Zhang if (a->idiagvalid) PetscFunctionReturn(0); 7999566063dSJacob Faibussowitsch PetscCall(MatMarkDiagonal_SeqSELL(A)); 800d4002b98SHong Zhang diag = a->diag; 801d4002b98SHong Zhang if (!a->idiag) { 8029566063dSJacob Faibussowitsch PetscCall(PetscMalloc3(m,&a->idiag,m,&a->mdiag,m,&a->ssor_work)); 8039566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)A, 3*m*sizeof(PetscScalar))); 804d4002b98SHong Zhang val = a->val; 805d4002b98SHong Zhang } 806d4002b98SHong Zhang mdiag = a->mdiag; 807d4002b98SHong Zhang idiag = a->idiag; 808d4002b98SHong Zhang 809d4002b98SHong Zhang if (omega == 1.0 && PetscRealPart(fshift) <= 0.0) { 810d4002b98SHong Zhang for (i=0; i<m; i++) { 811d4002b98SHong Zhang mdiag[i] = val[diag[i]]; 812d4002b98SHong Zhang if (!PetscAbsScalar(mdiag[i])) { /* zero diagonal */ 813d4002b98SHong Zhang if (PetscRealPart(fshift)) { 8149566063dSJacob Faibussowitsch PetscCall(PetscInfo(A,"Zero diagonal on row %" PetscInt_FMT "\n",i)); 815d4002b98SHong Zhang A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 816d4002b98SHong Zhang A->factorerror_zeropivot_value = 0.0; 817d4002b98SHong Zhang A->factorerror_zeropivot_row = i; 81898921bdaSJacob Faibussowitsch } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Zero diagonal on row %" PetscInt_FMT,i); 819d4002b98SHong Zhang } 820d4002b98SHong Zhang idiag[i] = 1.0/val[diag[i]]; 821d4002b98SHong Zhang } 8229566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(m)); 823d4002b98SHong Zhang } else { 824d4002b98SHong Zhang for (i=0; i<m; i++) { 825d4002b98SHong Zhang mdiag[i] = val[diag[i]]; 826d4002b98SHong Zhang idiag[i] = omega/(fshift + val[diag[i]]); 827d4002b98SHong Zhang } 8289566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0*m)); 829d4002b98SHong Zhang } 830d4002b98SHong Zhang a->idiagvalid = PETSC_TRUE; 831d4002b98SHong Zhang PetscFunctionReturn(0); 832d4002b98SHong Zhang } 833d4002b98SHong Zhang 834d4002b98SHong Zhang PetscErrorCode MatZeroEntries_SeqSELL(Mat A) 835d4002b98SHong Zhang { 836d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 837d4002b98SHong Zhang 838d4002b98SHong Zhang PetscFunctionBegin; 8399566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(a->val,a->sliidx[a->totalslices])); 8409566063dSJacob Faibussowitsch PetscCall(MatSeqSELLInvalidateDiagonal(A)); 841d4002b98SHong Zhang PetscFunctionReturn(0); 842d4002b98SHong Zhang } 843d4002b98SHong Zhang 844d4002b98SHong Zhang PetscErrorCode MatDestroy_SeqSELL(Mat A) 845d4002b98SHong Zhang { 846d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 847d4002b98SHong Zhang 848d4002b98SHong Zhang PetscFunctionBegin; 849d4002b98SHong Zhang #if defined(PETSC_USE_LOG) 850c0aa6a63SJacob Faibussowitsch PetscLogObjectState((PetscObject)A,"Rows=%" PetscInt_FMT ", Cols=%" PetscInt_FMT ", NZ=%" PetscInt_FMT,A->rmap->n,A->cmap->n,a->nz); 851d4002b98SHong Zhang #endif 8529566063dSJacob Faibussowitsch PetscCall(MatSeqXSELLFreeSELL(A,&a->val,&a->colidx)); 8539566063dSJacob Faibussowitsch PetscCall(ISDestroy(&a->row)); 8549566063dSJacob Faibussowitsch PetscCall(ISDestroy(&a->col)); 8559566063dSJacob Faibussowitsch PetscCall(PetscFree(a->diag)); 8569566063dSJacob Faibussowitsch PetscCall(PetscFree(a->rlen)); 8579566063dSJacob Faibussowitsch PetscCall(PetscFree(a->sliidx)); 8589566063dSJacob Faibussowitsch PetscCall(PetscFree3(a->idiag,a->mdiag,a->ssor_work)); 8599566063dSJacob Faibussowitsch PetscCall(PetscFree(a->solve_work)); 8609566063dSJacob Faibussowitsch PetscCall(ISDestroy(&a->icol)); 8619566063dSJacob Faibussowitsch PetscCall(PetscFree(a->saved_values)); 8629566063dSJacob Faibussowitsch PetscCall(PetscFree2(a->getrowcols,a->getrowvals)); 863d4002b98SHong Zhang 8649566063dSJacob Faibussowitsch PetscCall(PetscFree(A->data)); 865d4002b98SHong Zhang 8669566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)A,NULL)); 8679566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL)); 8689566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL)); 8699566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)A,"MatSeqSELLSetPreallocation_C",NULL)); 870d4002b98SHong Zhang PetscFunctionReturn(0); 871d4002b98SHong Zhang } 872d4002b98SHong Zhang 873d4002b98SHong Zhang PetscErrorCode MatSetOption_SeqSELL(Mat A,MatOption op,PetscBool flg) 874d4002b98SHong Zhang { 875d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 876d4002b98SHong Zhang 877d4002b98SHong Zhang PetscFunctionBegin; 878d4002b98SHong Zhang switch (op) { 879d4002b98SHong Zhang case MAT_ROW_ORIENTED: 880d4002b98SHong Zhang a->roworiented = flg; 881d4002b98SHong Zhang break; 882d4002b98SHong Zhang case MAT_KEEP_NONZERO_PATTERN: 883d4002b98SHong Zhang a->keepnonzeropattern = flg; 884d4002b98SHong Zhang break; 885d4002b98SHong Zhang case MAT_NEW_NONZERO_LOCATIONS: 886d4002b98SHong Zhang a->nonew = (flg ? 0 : 1); 887d4002b98SHong Zhang break; 888d4002b98SHong Zhang case MAT_NEW_NONZERO_LOCATION_ERR: 889d4002b98SHong Zhang a->nonew = (flg ? -1 : 0); 890d4002b98SHong Zhang break; 891d4002b98SHong Zhang case MAT_NEW_NONZERO_ALLOCATION_ERR: 892d4002b98SHong Zhang a->nonew = (flg ? -2 : 0); 893d4002b98SHong Zhang break; 894d4002b98SHong Zhang case MAT_UNUSED_NONZERO_LOCATION_ERR: 895d4002b98SHong Zhang a->nounused = (flg ? -1 : 0); 896d4002b98SHong Zhang break; 8978c78258cSHong Zhang case MAT_FORCE_DIAGONAL_ENTRIES: 898d4002b98SHong Zhang case MAT_IGNORE_OFF_PROC_ENTRIES: 899d4002b98SHong Zhang case MAT_USE_HASH_TABLE: 900071fcb05SBarry Smith case MAT_SORTED_FULL: 9019566063dSJacob Faibussowitsch PetscCall(PetscInfo(A,"Option %s ignored\n",MatOptions[op])); 902d4002b98SHong Zhang break; 903d4002b98SHong Zhang case MAT_SPD: 904d4002b98SHong Zhang case MAT_SYMMETRIC: 905d4002b98SHong Zhang case MAT_STRUCTURALLY_SYMMETRIC: 906d4002b98SHong Zhang case MAT_HERMITIAN: 907d4002b98SHong Zhang case MAT_SYMMETRY_ETERNAL: 908d4002b98SHong Zhang /* These options are handled directly by MatSetOption() */ 909d4002b98SHong Zhang break; 910d4002b98SHong Zhang default: 91198921bdaSJacob Faibussowitsch SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 912d4002b98SHong Zhang } 913d4002b98SHong Zhang PetscFunctionReturn(0); 914d4002b98SHong Zhang } 915d4002b98SHong Zhang 916d4002b98SHong Zhang PetscErrorCode MatGetDiagonal_SeqSELL(Mat A,Vec v) 917d4002b98SHong Zhang { 918d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 919d4002b98SHong Zhang PetscInt i,j,n,shift; 920d4002b98SHong Zhang PetscScalar *x,zero=0.0; 921d4002b98SHong Zhang 922d4002b98SHong Zhang PetscFunctionBegin; 9239566063dSJacob Faibussowitsch PetscCall(VecGetLocalSize(v,&n)); 92408401ef6SPierre Jolivet PetscCheck(n == A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 925d4002b98SHong Zhang 926d4002b98SHong Zhang if (A->factortype == MAT_FACTOR_ILU || A->factortype == MAT_FACTOR_LU) { 927d4002b98SHong Zhang PetscInt *diag=a->diag; 9289566063dSJacob Faibussowitsch PetscCall(VecGetArray(v,&x)); 929d4002b98SHong Zhang for (i=0; i<n; i++) x[i] = 1.0/a->val[diag[i]]; 9309566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(v,&x)); 931d4002b98SHong Zhang PetscFunctionReturn(0); 932d4002b98SHong Zhang } 933d4002b98SHong Zhang 9349566063dSJacob Faibussowitsch PetscCall(VecSet(v,zero)); 9359566063dSJacob Faibussowitsch PetscCall(VecGetArray(v,&x)); 936d4002b98SHong Zhang for (i=0; i<n; i++) { /* loop over rows */ 937d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 938d4002b98SHong Zhang x[i] = 0; 939d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 940d4002b98SHong Zhang if (a->colidx[shift+j*8] == i) { 941d4002b98SHong Zhang x[i] = a->val[shift+j*8]; 942d4002b98SHong Zhang break; 943d4002b98SHong Zhang } 944d4002b98SHong Zhang } 945d4002b98SHong Zhang } 9469566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(v,&x)); 947d4002b98SHong Zhang PetscFunctionReturn(0); 948d4002b98SHong Zhang } 949d4002b98SHong Zhang 950d4002b98SHong Zhang PetscErrorCode MatDiagonalScale_SeqSELL(Mat A,Vec ll,Vec rr) 951d4002b98SHong Zhang { 952d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 953d4002b98SHong Zhang const PetscScalar *l,*r; 954d4002b98SHong Zhang PetscInt i,j,m,n,row; 955d4002b98SHong Zhang 956d4002b98SHong Zhang PetscFunctionBegin; 957d4002b98SHong Zhang if (ll) { 958d4002b98SHong Zhang /* The local size is used so that VecMPI can be passed to this routine 959d4002b98SHong Zhang by MatDiagonalScale_MPISELL */ 9609566063dSJacob Faibussowitsch PetscCall(VecGetLocalSize(ll,&m)); 96108401ef6SPierre Jolivet PetscCheck(m == A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length"); 9629566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(ll,&l)); 963d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 964dab86139SHong Zhang if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */ 965dab86139SHong Zhang for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 966dab86139SHong Zhang if (row < (A->rmap->n & 0x07)) a->val[j] *= l[8*i+row]; 967dab86139SHong Zhang } 968dab86139SHong Zhang } else { 969d4002b98SHong Zhang for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 970d4002b98SHong Zhang a->val[j] *= l[8*i+row]; 971d4002b98SHong Zhang } 972d4002b98SHong Zhang } 973dab86139SHong Zhang } 9749566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(ll,&l)); 9759566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(a->nz)); 976d4002b98SHong Zhang } 977d4002b98SHong Zhang if (rr) { 9789566063dSJacob Faibussowitsch PetscCall(VecGetLocalSize(rr,&n)); 97908401ef6SPierre Jolivet PetscCheck(n == A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length"); 9809566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(rr,&r)); 981d4002b98SHong Zhang for (i=0; i<a->totalslices; i++) { /* loop over slices */ 982dab86139SHong Zhang if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */ 983dab86139SHong Zhang for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 984dab86139SHong Zhang if (row < (A->rmap->n & 0x07)) a->val[j] *= r[a->colidx[j]]; 985dab86139SHong Zhang } 986dab86139SHong Zhang } else { 987d4002b98SHong Zhang for (j=a->sliidx[i]; j<a->sliidx[i+1]; j++) { 988d4002b98SHong Zhang a->val[j] *= r[a->colidx[j]]; 989d4002b98SHong Zhang } 990d4002b98SHong Zhang } 991dab86139SHong Zhang } 9929566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(rr,&r)); 9939566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(a->nz)); 994d4002b98SHong Zhang } 9959566063dSJacob Faibussowitsch PetscCall(MatSeqSELLInvalidateDiagonal(A)); 996d4002b98SHong Zhang PetscFunctionReturn(0); 997d4002b98SHong Zhang } 998d4002b98SHong Zhang 999d4002b98SHong Zhang PetscErrorCode MatGetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 1000d4002b98SHong Zhang { 1001d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1002d4002b98SHong Zhang PetscInt *cp,i,k,low,high,t,row,col,l; 1003d4002b98SHong Zhang PetscInt shift; 1004d4002b98SHong Zhang MatScalar *vp; 1005d4002b98SHong Zhang 1006d4002b98SHong Zhang PetscFunctionBegin; 100768aafef3SStefano Zampini for (k=0; k<m; k++) { /* loop over requested rows */ 1008d4002b98SHong Zhang row = im[k]; 1009d4002b98SHong Zhang if (row<0) continue; 10106bdcaf15SBarry Smith PetscCheck(row < A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->n-1); 1011d4002b98SHong Zhang shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */ 1012d4002b98SHong Zhang cp = a->colidx+shift; /* pointer to the row */ 1013d4002b98SHong Zhang vp = a->val+shift; /* pointer to the row */ 101468aafef3SStefano Zampini for (l=0; l<n; l++) { /* loop over requested columns */ 1015d4002b98SHong Zhang col = in[l]; 1016d4002b98SHong Zhang if (col<0) continue; 10176bdcaf15SBarry Smith PetscCheck(col < A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: row %" PetscInt_FMT " max %" PetscInt_FMT,col,A->cmap->n-1); 1018d4002b98SHong Zhang high = a->rlen[row]; low = 0; /* assume unsorted */ 1019d4002b98SHong Zhang while (high-low > 5) { 1020d4002b98SHong Zhang t = (low+high)/2; 1021d4002b98SHong Zhang if (*(cp+t*8) > col) high = t; 1022d4002b98SHong Zhang else low = t; 1023d4002b98SHong Zhang } 1024d4002b98SHong Zhang for (i=low; i<high; i++) { 1025d4002b98SHong Zhang if (*(cp+8*i) > col) break; 1026d4002b98SHong Zhang if (*(cp+8*i) == col) { 1027d4002b98SHong Zhang *v++ = *(vp+8*i); 1028d4002b98SHong Zhang goto finished; 1029d4002b98SHong Zhang } 1030d4002b98SHong Zhang } 1031d4002b98SHong Zhang *v++ = 0.0; 1032d4002b98SHong Zhang finished:; 1033d4002b98SHong Zhang } 1034d4002b98SHong Zhang } 1035d4002b98SHong Zhang PetscFunctionReturn(0); 1036d4002b98SHong Zhang } 1037d4002b98SHong Zhang 1038d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_ASCII(Mat A,PetscViewer viewer) 1039d4002b98SHong Zhang { 1040d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1041d4002b98SHong Zhang PetscInt i,j,m=A->rmap->n,shift; 1042d4002b98SHong Zhang const char *name; 1043d4002b98SHong Zhang PetscViewerFormat format; 1044d4002b98SHong Zhang 1045d4002b98SHong Zhang PetscFunctionBegin; 10469566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(viewer,&format)); 1047d4002b98SHong Zhang if (format == PETSC_VIEWER_ASCII_MATLAB) { 1048d4002b98SHong Zhang PetscInt nofinalvalue = 0; 1049d4002b98SHong Zhang /* 1050d4002b98SHong Zhang if (m && ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != A->cmap->n-1))) { 1051d4002b98SHong Zhang nofinalvalue = 1; 1052d4002b98SHong Zhang } 1053d4002b98SHong Zhang */ 10549566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE)); 10559566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%% Size = %" PetscInt_FMT " %" PetscInt_FMT " \n",m,A->cmap->n)); 10569566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%% Nonzeros = %" PetscInt_FMT " \n",a->nz)); 1057d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 10589566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",4);\n",a->nz+nofinalvalue)); 1059d4002b98SHong Zhang #else 10609566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",3);\n",a->nz+nofinalvalue)); 1061d4002b98SHong Zhang #endif 10629566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"zzz = [\n")); 1063d4002b98SHong Zhang 1064d4002b98SHong Zhang for (i=0; i<m; i++) { 1065d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1066d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1067d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 10689566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %18.16e %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]))); 1069d4002b98SHong Zhang #else 10709566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)a->val[shift+8*j])); 1071d4002b98SHong Zhang #endif 1072d4002b98SHong Zhang } 1073d4002b98SHong Zhang } 1074d4002b98SHong Zhang /* 1075d4002b98SHong Zhang if (nofinalvalue) { 1076d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 10779566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %18.16e %18.16e\n",m,A->cmap->n,0.,0.)); 1078d4002b98SHong Zhang #else 10799566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %18.16e\n",m,A->cmap->n,0.0)); 1080d4002b98SHong Zhang #endif 1081d4002b98SHong Zhang } 1082d4002b98SHong Zhang */ 10839566063dSJacob Faibussowitsch PetscCall(PetscObjectGetName((PetscObject)A,&name)); 10849566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"];\n %s = spconvert(zzz);\n",name)); 10859566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE)); 1086d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO || format == PETSC_VIEWER_ASCII_INFO) { 1087d4002b98SHong Zhang PetscFunctionReturn(0); 1088d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_COMMON) { 10899566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE)); 1090d4002b98SHong Zhang for (i=0; i<m; i++) { 10919566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i)); 1092d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1093d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1094d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1095d4002b98SHong Zhang if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) { 10969566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]))); 1097d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) { 10989566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]))); 1099d4002b98SHong Zhang } else if (PetscRealPart(a->val[shift+8*j]) != 0.0) { 11009566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]))); 1101d4002b98SHong Zhang } 1102d4002b98SHong Zhang #else 11039566063dSJacob Faibussowitsch if (a->val[shift+8*j] != 0.0) PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j])); 1104d4002b98SHong Zhang #endif 1105d4002b98SHong Zhang } 11069566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"\n")); 1107d4002b98SHong Zhang } 11089566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE)); 1109d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_DENSE) { 1110d4002b98SHong Zhang PetscInt cnt=0,jcnt; 1111d4002b98SHong Zhang PetscScalar value; 1112d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1113d4002b98SHong Zhang PetscBool realonly=PETSC_TRUE; 1114d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) { 1115d4002b98SHong Zhang if (PetscImaginaryPart(a->val[i]) != 0.0) { 1116d4002b98SHong Zhang realonly = PETSC_FALSE; 1117d4002b98SHong Zhang break; 1118d4002b98SHong Zhang } 1119d4002b98SHong Zhang } 1120d4002b98SHong Zhang #endif 1121d4002b98SHong Zhang 11229566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE)); 1123d4002b98SHong Zhang for (i=0; i<m; i++) { 1124d4002b98SHong Zhang jcnt = 0; 1125d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1126d4002b98SHong Zhang for (j=0; j<A->cmap->n; j++) { 1127d4002b98SHong Zhang if (jcnt < a->rlen[i] && j == a->colidx[shift+8*j]) { 1128d4002b98SHong Zhang value = a->val[cnt++]; 1129d4002b98SHong Zhang jcnt++; 1130d4002b98SHong Zhang } else { 1131d4002b98SHong Zhang value = 0.0; 1132d4002b98SHong Zhang } 1133d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1134d4002b98SHong Zhang if (realonly) { 11359566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %7.5e ",(double)PetscRealPart(value))); 1136d4002b98SHong Zhang } else { 11379566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %7.5e+%7.5e i ",(double)PetscRealPart(value),(double)PetscImaginaryPart(value))); 1138d4002b98SHong Zhang } 1139d4002b98SHong Zhang #else 11409566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %7.5e ",(double)value)); 1141d4002b98SHong Zhang #endif 1142d4002b98SHong Zhang } 11439566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"\n")); 1144d4002b98SHong Zhang } 11459566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE)); 1146d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_MATRIXMARKET) { 1147d4002b98SHong Zhang PetscInt fshift=1; 11489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE)); 1149d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 11509566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate complex general\n")); 1151d4002b98SHong Zhang #else 11529566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate real general\n")); 1153d4002b98SHong Zhang #endif 11549566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %" PetscInt_FMT "\n", m, A->cmap->n, a->nz)); 1155d4002b98SHong Zhang for (i=0; i<m; i++) { 1156d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1157d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1158d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 11599566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]))); 1160d4002b98SHong Zhang #else 11619566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)a->val[shift+8*j])); 1162d4002b98SHong Zhang #endif 1163d4002b98SHong Zhang } 1164d4002b98SHong Zhang } 11659566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE)); 116668aafef3SStefano Zampini } else if (format == PETSC_VIEWER_NATIVE) { 116768aafef3SStefano Zampini for (i=0; i<a->totalslices; i++) { /* loop over slices */ 116868aafef3SStefano Zampini PetscInt row; 11699566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"slice %" PetscInt_FMT ": %" PetscInt_FMT " %" PetscInt_FMT "\n",i,a->sliidx[i],a->sliidx[i+1])); 117068aafef3SStefano Zampini for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) { 117168aafef3SStefano Zampini #if defined(PETSC_USE_COMPLEX) 117268aafef3SStefano Zampini if (PetscImaginaryPart(a->val[j]) > 0.0) { 11739566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %" PetscInt_FMT " %" PetscInt_FMT " %g + %g i\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]))); 117468aafef3SStefano Zampini } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 11759566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %" PetscInt_FMT " %" PetscInt_FMT " %g - %g i\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]),-(double)PetscImaginaryPart(a->val[j]))); 117668aafef3SStefano Zampini } else { 11779566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]))); 117868aafef3SStefano Zampini } 117968aafef3SStefano Zampini #else 11809566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)a->val[j])); 118168aafef3SStefano Zampini #endif 118268aafef3SStefano Zampini } 118368aafef3SStefano Zampini } 1184d4002b98SHong Zhang } else { 11859566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE)); 1186d4002b98SHong Zhang if (A->factortype) { 1187d4002b98SHong Zhang for (i=0; i<m; i++) { 1188d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 11899566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i)); 1190d4002b98SHong Zhang /* L part */ 1191d4002b98SHong Zhang for (j=shift; j<a->diag[i]; j+=8) { 1192d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1193d4002b98SHong Zhang if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0) { 11949566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]))); 1195d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0) { 11969566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])))); 1197d4002b98SHong Zhang } else { 11989566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]))); 1199d4002b98SHong Zhang } 1200d4002b98SHong Zhang #else 12019566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j])); 1202d4002b98SHong Zhang #endif 1203d4002b98SHong Zhang } 1204d4002b98SHong Zhang /* diagonal */ 1205d4002b98SHong Zhang j = a->diag[i]; 1206d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1207d4002b98SHong Zhang if (PetscImaginaryPart(a->val[j]) > 0.0) { 12089566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)PetscImaginaryPart(1.0/a->val[j]))); 1209d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 12109566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)(-PetscImaginaryPart(1.0/a->val[j])))); 1211d4002b98SHong Zhang } else { 12129566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]))); 1213d4002b98SHong Zhang } 1214d4002b98SHong Zhang #else 12159566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)(1.0/a->val[j]))); 1216d4002b98SHong Zhang #endif 1217d4002b98SHong Zhang 1218d4002b98SHong Zhang /* U part */ 1219d4002b98SHong Zhang for (j=a->diag[i]+1; j<shift+8*a->rlen[i]; j+=8) { 1220d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1221d4002b98SHong Zhang if (PetscImaginaryPart(a->val[j]) > 0.0) { 12229566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]))); 1223d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 12249566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])))); 1225d4002b98SHong Zhang } else { 12269566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]))); 1227d4002b98SHong Zhang } 1228d4002b98SHong Zhang #else 12299566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j])); 1230d4002b98SHong Zhang #endif 1231d4002b98SHong Zhang } 12329566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"\n")); 1233d4002b98SHong Zhang } 1234d4002b98SHong Zhang } else { 1235d4002b98SHong Zhang for (i=0; i<m; i++) { 1236d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 12379566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i)); 1238d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1239d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 1240d4002b98SHong Zhang if (PetscImaginaryPart(a->val[j]) > 0.0) { 12419566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]))); 1242d4002b98SHong Zhang } else if (PetscImaginaryPart(a->val[j]) < 0.0) { 12439566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]))); 1244d4002b98SHong Zhang } else { 12459566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]))); 1246d4002b98SHong Zhang } 1247d4002b98SHong Zhang #else 12489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j])); 1249d4002b98SHong Zhang #endif 1250d4002b98SHong Zhang } 12519566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"\n")); 1252d4002b98SHong Zhang } 1253d4002b98SHong Zhang } 12549566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE)); 1255d4002b98SHong Zhang } 12569566063dSJacob Faibussowitsch PetscCall(PetscViewerFlush(viewer)); 1257d4002b98SHong Zhang PetscFunctionReturn(0); 1258d4002b98SHong Zhang } 1259d4002b98SHong Zhang 1260d4002b98SHong Zhang #include <petscdraw.h> 1261d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw_Zoom(PetscDraw draw,void *Aa) 1262d4002b98SHong Zhang { 1263d4002b98SHong Zhang Mat A=(Mat)Aa; 1264d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1265d4002b98SHong Zhang PetscInt i,j,m=A->rmap->n,shift; 1266d4002b98SHong Zhang int color; 1267d4002b98SHong Zhang PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 1268d4002b98SHong Zhang PetscViewer viewer; 1269d4002b98SHong Zhang PetscViewerFormat format; 1270d4002b98SHong Zhang 1271d4002b98SHong Zhang PetscFunctionBegin; 12729566063dSJacob Faibussowitsch PetscCall(PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer)); 12739566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(viewer,&format)); 12749566063dSJacob Faibussowitsch PetscCall(PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr)); 1275d4002b98SHong Zhang 1276d4002b98SHong Zhang /* loop over matrix elements drawing boxes */ 1277d4002b98SHong Zhang 1278d4002b98SHong Zhang if (format != PETSC_VIEWER_DRAW_CONTOUR) { 1279d0609cedSBarry Smith PetscDrawCollectiveBegin(draw); 1280d4002b98SHong Zhang /* Blue for negative, Cyan for zero and Red for positive */ 1281d4002b98SHong Zhang color = PETSC_DRAW_BLUE; 1282d4002b98SHong Zhang for (i=0; i<m; i++) { 1283d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1284d4002b98SHong Zhang y_l = m - i - 1.0; y_r = y_l + 1.0; 1285d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1286d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0; 1287d4002b98SHong Zhang if (PetscRealPart(a->val[shift+8*j]) >= 0.) continue; 12889566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color)); 1289d4002b98SHong Zhang } 1290d4002b98SHong Zhang } 1291d4002b98SHong Zhang color = PETSC_DRAW_CYAN; 1292d4002b98SHong Zhang for (i=0; i<m; i++) { 1293d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1294d4002b98SHong Zhang y_l = m - i - 1.0; y_r = y_l + 1.0; 1295d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1296d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0; 1297d4002b98SHong Zhang if (a->val[shift+8*j] != 0.) continue; 12989566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color)); 1299d4002b98SHong Zhang } 1300d4002b98SHong Zhang } 1301d4002b98SHong Zhang color = PETSC_DRAW_RED; 1302d4002b98SHong Zhang for (i=0; i<m; i++) { 1303d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1304d4002b98SHong Zhang y_l = m - i - 1.0; y_r = y_l + 1.0; 1305d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1306d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0; 1307d4002b98SHong Zhang if (PetscRealPart(a->val[shift+8*j]) <= 0.) continue; 13089566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color)); 1309d4002b98SHong Zhang } 1310d4002b98SHong Zhang } 1311d0609cedSBarry Smith PetscDrawCollectiveEnd(draw); 1312d4002b98SHong Zhang } else { 1313d4002b98SHong Zhang /* use contour shading to indicate magnitude of values */ 1314d4002b98SHong Zhang /* first determine max of all nonzero values */ 1315d4002b98SHong Zhang PetscReal minv=0.0,maxv=0.0; 1316d4002b98SHong Zhang PetscInt count=0; 1317d4002b98SHong Zhang PetscDraw popup; 1318d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) { 1319d4002b98SHong Zhang if (PetscAbsScalar(a->val[i]) > maxv) maxv = PetscAbsScalar(a->val[i]); 1320d4002b98SHong Zhang } 1321d4002b98SHong Zhang if (minv >= maxv) maxv = minv + PETSC_SMALL; 13229566063dSJacob Faibussowitsch PetscCall(PetscDrawGetPopup(draw,&popup)); 13239566063dSJacob Faibussowitsch PetscCall(PetscDrawScalePopup(popup,minv,maxv)); 1324d4002b98SHong Zhang 1325d0609cedSBarry Smith PetscDrawCollectiveBegin(draw); 1326d4002b98SHong Zhang for (i=0; i<m; i++) { 1327d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); 1328d4002b98SHong Zhang y_l = m - i - 1.0; 1329d4002b98SHong Zhang y_r = y_l + 1.0; 1330d4002b98SHong Zhang for (j=0; j<a->rlen[i]; j++) { 1331d4002b98SHong Zhang x_l = a->colidx[shift+j*8]; 1332d4002b98SHong Zhang x_r = x_l + 1.0; 1333d4002b98SHong Zhang color = PetscDrawRealToColor(PetscAbsScalar(a->val[count]),minv,maxv); 13349566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color)); 1335d4002b98SHong Zhang count++; 1336d4002b98SHong Zhang } 1337d4002b98SHong Zhang } 1338d0609cedSBarry Smith PetscDrawCollectiveEnd(draw); 1339d4002b98SHong Zhang } 1340d4002b98SHong Zhang PetscFunctionReturn(0); 1341d4002b98SHong Zhang } 1342d4002b98SHong Zhang 1343d4002b98SHong Zhang #include <petscdraw.h> 1344d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw(Mat A,PetscViewer viewer) 1345d4002b98SHong Zhang { 1346d4002b98SHong Zhang PetscDraw draw; 1347d4002b98SHong Zhang PetscReal xr,yr,xl,yl,h,w; 1348d4002b98SHong Zhang PetscBool isnull; 1349d4002b98SHong Zhang 1350d4002b98SHong Zhang PetscFunctionBegin; 13519566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDraw(viewer,0,&draw)); 13529566063dSJacob Faibussowitsch PetscCall(PetscDrawIsNull(draw,&isnull)); 1353d4002b98SHong Zhang if (isnull) PetscFunctionReturn(0); 1354d4002b98SHong Zhang 1355d4002b98SHong Zhang xr = A->cmap->n; yr = A->rmap->n; h = yr/10.0; w = xr/10.0; 1356d4002b98SHong Zhang xr += w; yr += h; xl = -w; yl = -h; 13579566063dSJacob Faibussowitsch PetscCall(PetscDrawSetCoordinates(draw,xl,yl,xr,yr)); 13589566063dSJacob Faibussowitsch PetscCall(PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer)); 13599566063dSJacob Faibussowitsch PetscCall(PetscDrawZoom(draw,MatView_SeqSELL_Draw_Zoom,A)); 13609566063dSJacob Faibussowitsch PetscCall(PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL)); 13619566063dSJacob Faibussowitsch PetscCall(PetscDrawSave(draw)); 1362d4002b98SHong Zhang PetscFunctionReturn(0); 1363d4002b98SHong Zhang } 1364d4002b98SHong Zhang 1365d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL(Mat A,PetscViewer viewer) 1366d4002b98SHong Zhang { 1367d4002b98SHong Zhang PetscBool iascii,isbinary,isdraw; 1368d4002b98SHong Zhang 1369d4002b98SHong Zhang PetscFunctionBegin; 13709566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii)); 13719566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary)); 13729566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw)); 1373d4002b98SHong Zhang if (iascii) { 13749566063dSJacob Faibussowitsch PetscCall(MatView_SeqSELL_ASCII(A,viewer)); 1375d4002b98SHong Zhang } else if (isbinary) { 13769566063dSJacob Faibussowitsch /* PetscCall(MatView_SeqSELL_Binary(A,viewer)); */ 1377d4002b98SHong Zhang } else if (isdraw) { 13789566063dSJacob Faibussowitsch PetscCall(MatView_SeqSELL_Draw(A,viewer)); 1379d4002b98SHong Zhang } 1380d4002b98SHong Zhang PetscFunctionReturn(0); 1381d4002b98SHong Zhang } 1382d4002b98SHong Zhang 1383d4002b98SHong Zhang PetscErrorCode MatAssemblyEnd_SeqSELL(Mat A,MatAssemblyType mode) 1384d4002b98SHong Zhang { 1385d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1386d4002b98SHong Zhang PetscInt i,shift,row_in_slice,row,nrow,*cp,lastcol,j,k; 1387d4002b98SHong Zhang MatScalar *vp; 1388d4002b98SHong Zhang 1389d4002b98SHong Zhang PetscFunctionBegin; 1390d4002b98SHong Zhang if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 1391d4002b98SHong Zhang /* To do: compress out the unused elements */ 13929566063dSJacob Faibussowitsch PetscCall(MatMarkDiagonal_SeqSELL(A)); 13939566063dSJacob Faibussowitsch PetscCall(PetscInfo(A,"Matrix size: %" PetscInt_FMT " X %" PetscInt_FMT "; storage space: %" PetscInt_FMT " allocated %" PetscInt_FMT " used (%" PetscInt_FMT " nonzeros+%" PetscInt_FMT " paddedzeros)\n",A->rmap->n,A->cmap->n,a->maxallocmat,a->sliidx[a->totalslices],a->nz,a->sliidx[a->totalslices]-a->nz)); 13949566063dSJacob Faibussowitsch PetscCall(PetscInfo(A,"Number of mallocs during MatSetValues() is %" PetscInt_FMT "\n",a->reallocs)); 13959566063dSJacob Faibussowitsch PetscCall(PetscInfo(A,"Maximum nonzeros in any row is %" PetscInt_FMT "\n",a->rlenmax)); 1396d4002b98SHong 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 */ 1397d4002b98SHong Zhang for (i=0; i<a->totalslices; ++i) { 1398d4002b98SHong Zhang shift = a->sliidx[i]; /* starting index of the slice */ 1399d4002b98SHong Zhang cp = a->colidx+shift; /* pointer to the column indices of the slice */ 1400d4002b98SHong Zhang vp = a->val+shift; /* pointer to the nonzero values of the slice */ 1401d4002b98SHong Zhang for (row_in_slice=0; row_in_slice<8; ++row_in_slice) { /* loop over rows in the slice */ 1402d4002b98SHong Zhang row = 8*i + row_in_slice; 1403d4002b98SHong Zhang nrow = a->rlen[row]; /* number of nonzeros in row */ 1404d4002b98SHong Zhang /* 1405d4002b98SHong Zhang Search for the nearest nonzero. Normally setting the index to zero may cause extra communication. 1406d4002b98SHong Zhang But if the entire slice are empty, it is fine to use 0 since the index will not be loaded. 1407d4002b98SHong Zhang */ 1408d4002b98SHong Zhang lastcol = 0; 1409d4002b98SHong Zhang if (nrow>0) { /* nonempty row */ 1410d4002b98SHong Zhang lastcol = cp[8*(nrow-1)+row_in_slice]; /* use the index from the last nonzero at current row */ 1411d4002b98SHong Zhang } else if (!row_in_slice) { /* first row of the currect slice is empty */ 1412d4002b98SHong Zhang for (j=1;j<8;j++) { 1413d4002b98SHong Zhang if (a->rlen[8*i+j]) { 1414d4002b98SHong Zhang lastcol = cp[j]; 1415d4002b98SHong Zhang break; 1416d4002b98SHong Zhang } 1417d4002b98SHong Zhang } 1418d4002b98SHong Zhang } else { 1419d4002b98SHong Zhang if (a->sliidx[i+1] != shift) lastcol = cp[row_in_slice-1]; /* use the index from the previous row */ 1420d4002b98SHong Zhang } 1421d4002b98SHong Zhang 1422d4002b98SHong Zhang for (k=nrow; k<(a->sliidx[i+1]-shift)/8; ++k) { 1423d4002b98SHong Zhang cp[8*k+row_in_slice] = lastcol; 1424d4002b98SHong Zhang vp[8*k+row_in_slice] = (MatScalar)0; 1425d4002b98SHong Zhang } 1426d4002b98SHong Zhang } 1427d4002b98SHong Zhang } 1428d4002b98SHong Zhang 1429d4002b98SHong Zhang A->info.mallocs += a->reallocs; 1430d4002b98SHong Zhang a->reallocs = 0; 1431d4002b98SHong Zhang 14329566063dSJacob Faibussowitsch PetscCall(MatSeqSELLInvalidateDiagonal(A)); 1433d4002b98SHong Zhang PetscFunctionReturn(0); 1434d4002b98SHong Zhang } 1435d4002b98SHong Zhang 1436d4002b98SHong Zhang PetscErrorCode MatGetInfo_SeqSELL(Mat A,MatInfoType flag,MatInfo *info) 1437d4002b98SHong Zhang { 1438d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1439d4002b98SHong Zhang 1440d4002b98SHong Zhang PetscFunctionBegin; 1441d4002b98SHong Zhang info->block_size = 1.0; 14423966268fSBarry Smith info->nz_allocated = a->maxallocmat; 14433966268fSBarry Smith info->nz_used = a->sliidx[a->totalslices]; /* include padding zeros */ 14443966268fSBarry Smith info->nz_unneeded = (a->maxallocmat-a->sliidx[a->totalslices]); 14453966268fSBarry Smith info->assemblies = A->num_ass; 14463966268fSBarry Smith info->mallocs = A->info.mallocs; 1447d4002b98SHong Zhang info->memory = ((PetscObject)A)->mem; 1448d4002b98SHong Zhang if (A->factortype) { 1449d4002b98SHong Zhang info->fill_ratio_given = A->info.fill_ratio_given; 1450d4002b98SHong Zhang info->fill_ratio_needed = A->info.fill_ratio_needed; 1451d4002b98SHong Zhang info->factor_mallocs = A->info.factor_mallocs; 1452d4002b98SHong Zhang } else { 1453d4002b98SHong Zhang info->fill_ratio_given = 0; 1454d4002b98SHong Zhang info->fill_ratio_needed = 0; 1455d4002b98SHong Zhang info->factor_mallocs = 0; 1456d4002b98SHong Zhang } 1457d4002b98SHong Zhang PetscFunctionReturn(0); 1458d4002b98SHong Zhang } 1459d4002b98SHong Zhang 1460d4002b98SHong Zhang PetscErrorCode MatSetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 1461d4002b98SHong Zhang { 1462d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1463d4002b98SHong Zhang PetscInt shift,i,k,l,low,high,t,ii,row,col,nrow; 1464d4002b98SHong Zhang PetscInt *cp,nonew=a->nonew,lastcol=-1; 1465d4002b98SHong Zhang MatScalar *vp,value; 1466d4002b98SHong Zhang 1467d4002b98SHong Zhang PetscFunctionBegin; 1468d4002b98SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 1469d4002b98SHong Zhang row = im[k]; 1470d4002b98SHong Zhang if (row < 0) continue; 14716bdcaf15SBarry Smith PetscCheck(row < A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->n-1); 1472d4002b98SHong Zhang shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */ 1473d4002b98SHong Zhang cp = a->colidx+shift; /* pointer to the row */ 1474d4002b98SHong Zhang vp = a->val+shift; /* pointer to the row */ 1475d4002b98SHong Zhang nrow = a->rlen[row]; 1476d4002b98SHong Zhang low = 0; 1477d4002b98SHong Zhang high = nrow; 1478d4002b98SHong Zhang 1479d4002b98SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 1480d4002b98SHong Zhang col = in[l]; 1481d4002b98SHong Zhang if (col<0) continue; 14826bdcaf15SBarry Smith PetscCheck(col < A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Col too large: row %" PetscInt_FMT " max %" PetscInt_FMT,col,A->cmap->n-1); 1483d4002b98SHong Zhang if (a->roworiented) { 1484d4002b98SHong Zhang value = v[l+k*n]; 1485d4002b98SHong Zhang } else { 1486d4002b98SHong Zhang value = v[k+l*m]; 1487d4002b98SHong Zhang } 1488d4002b98SHong Zhang if ((value == 0.0 && a->ignorezeroentries) && (is == ADD_VALUES)) continue; 1489d4002b98SHong Zhang 1490ed73aabaSBarry Smith /* search in this row for the specified column, i indicates the column to be set */ 1491d4002b98SHong Zhang if (col <= lastcol) low = 0; 1492d4002b98SHong Zhang else high = nrow; 1493d4002b98SHong Zhang lastcol = col; 1494d4002b98SHong Zhang while (high-low > 5) { 1495d4002b98SHong Zhang t = (low+high)/2; 1496d4002b98SHong Zhang if (*(cp+t*8) > col) high = t; 1497d4002b98SHong Zhang else low = t; 1498d4002b98SHong Zhang } 1499d4002b98SHong Zhang for (i=low; i<high; i++) { 1500d4002b98SHong Zhang if (*(cp+i*8) > col) break; 1501d4002b98SHong Zhang if (*(cp+i*8) == col) { 1502d4002b98SHong Zhang if (is == ADD_VALUES) *(vp+i*8) += value; 1503d4002b98SHong Zhang else *(vp+i*8) = value; 1504d4002b98SHong Zhang low = i + 1; 1505d4002b98SHong Zhang goto noinsert; 1506d4002b98SHong Zhang } 1507d4002b98SHong Zhang } 1508d4002b98SHong Zhang if (value == 0.0 && a->ignorezeroentries) goto noinsert; 1509d4002b98SHong Zhang if (nonew == 1) goto noinsert; 151008401ef6SPierre Jolivet PetscCheck(nonew != -1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%" PetscInt_FMT ", %" PetscInt_FMT ") in the matrix", row, col); 1511d4002b98SHong Zhang /* If the current row length exceeds the slice width (e.g. nrow==slice_width), allocate a new space, otherwise do nothing */ 1512d4002b98SHong Zhang MatSeqXSELLReallocateSELL(A,A->rmap->n,1,nrow,a->sliidx,row/8,row,col,a->colidx,a->val,cp,vp,nonew,MatScalar); 1513d4002b98SHong Zhang /* add the new nonzero to the high position, shift the remaining elements in current row to the right by one slot */ 1514d4002b98SHong Zhang for (ii=nrow-1; ii>=i; ii--) { 1515d4002b98SHong Zhang *(cp+(ii+1)*8) = *(cp+ii*8); 1516d4002b98SHong Zhang *(vp+(ii+1)*8) = *(vp+ii*8); 1517d4002b98SHong Zhang } 1518d4002b98SHong Zhang a->rlen[row]++; 1519d4002b98SHong Zhang *(cp+i*8) = col; 1520d4002b98SHong Zhang *(vp+i*8) = value; 1521d4002b98SHong Zhang a->nz++; 1522d4002b98SHong Zhang A->nonzerostate++; 1523d4002b98SHong Zhang low = i+1; high++; nrow++; 1524d4002b98SHong Zhang noinsert:; 1525d4002b98SHong Zhang } 1526d4002b98SHong Zhang a->rlen[row] = nrow; 1527d4002b98SHong Zhang } 1528d4002b98SHong Zhang PetscFunctionReturn(0); 1529d4002b98SHong Zhang } 1530d4002b98SHong Zhang 1531d4002b98SHong Zhang PetscErrorCode MatCopy_SeqSELL(Mat A,Mat B,MatStructure str) 1532d4002b98SHong Zhang { 1533d4002b98SHong Zhang PetscFunctionBegin; 1534d4002b98SHong Zhang /* If the two matrices have the same copy implementation, use fast copy. */ 1535d4002b98SHong Zhang if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 1536d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1537d4002b98SHong Zhang Mat_SeqSELL *b=(Mat_SeqSELL*)B->data; 1538d4002b98SHong Zhang 153908401ef6SPierre Jolivet PetscCheck(a->sliidx[a->totalslices] == b->sliidx[b->totalslices],PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different"); 15409566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(b->val,a->val,a->sliidx[a->totalslices])); 1541d4002b98SHong Zhang } else { 15429566063dSJacob Faibussowitsch PetscCall(MatCopy_Basic(A,B,str)); 1543d4002b98SHong Zhang } 1544d4002b98SHong Zhang PetscFunctionReturn(0); 1545d4002b98SHong Zhang } 1546d4002b98SHong Zhang 1547d4002b98SHong Zhang PetscErrorCode MatSetUp_SeqSELL(Mat A) 1548d4002b98SHong Zhang { 1549d4002b98SHong Zhang PetscFunctionBegin; 15509566063dSJacob Faibussowitsch PetscCall(MatSeqSELLSetPreallocation(A,PETSC_DEFAULT,NULL)); 1551d4002b98SHong Zhang PetscFunctionReturn(0); 1552d4002b98SHong Zhang } 1553d4002b98SHong Zhang 1554d4002b98SHong Zhang PetscErrorCode MatSeqSELLGetArray_SeqSELL(Mat A,PetscScalar *array[]) 1555d4002b98SHong Zhang { 1556d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1557d4002b98SHong Zhang 1558d4002b98SHong Zhang PetscFunctionBegin; 1559d4002b98SHong Zhang *array = a->val; 1560d4002b98SHong Zhang PetscFunctionReturn(0); 1561d4002b98SHong Zhang } 1562d4002b98SHong Zhang 1563d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray_SeqSELL(Mat A,PetscScalar *array[]) 1564d4002b98SHong Zhang { 1565d4002b98SHong Zhang PetscFunctionBegin; 1566d4002b98SHong Zhang PetscFunctionReturn(0); 1567d4002b98SHong Zhang } 1568d4002b98SHong Zhang 1569d4002b98SHong Zhang PetscErrorCode MatRealPart_SeqSELL(Mat A) 1570d4002b98SHong Zhang { 1571d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1572d4002b98SHong Zhang PetscInt i; 1573d4002b98SHong Zhang MatScalar *aval=a->val; 1574d4002b98SHong Zhang 1575d4002b98SHong Zhang PetscFunctionBegin; 1576d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i]=PetscRealPart(aval[i]); 1577d4002b98SHong Zhang PetscFunctionReturn(0); 1578d4002b98SHong Zhang } 1579d4002b98SHong Zhang 1580d4002b98SHong Zhang PetscErrorCode MatImaginaryPart_SeqSELL(Mat A) 1581d4002b98SHong Zhang { 1582d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1583d4002b98SHong Zhang PetscInt i; 1584d4002b98SHong Zhang MatScalar *aval=a->val; 1585d4002b98SHong Zhang 1586d4002b98SHong Zhang PetscFunctionBegin; 1587d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i] = PetscImaginaryPart(aval[i]); 15889566063dSJacob Faibussowitsch PetscCall(MatSeqSELLInvalidateDiagonal(A)); 1589d4002b98SHong Zhang PetscFunctionReturn(0); 1590d4002b98SHong Zhang } 1591d4002b98SHong Zhang 1592d4002b98SHong Zhang PetscErrorCode MatScale_SeqSELL(Mat inA,PetscScalar alpha) 1593d4002b98SHong Zhang { 1594d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)inA->data; 1595d4002b98SHong Zhang MatScalar *aval=a->val; 1596d4002b98SHong Zhang PetscScalar oalpha=alpha; 1597d4002b98SHong Zhang PetscBLASInt one=1,size; 1598d4002b98SHong Zhang 1599d4002b98SHong Zhang PetscFunctionBegin; 16009566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(a->sliidx[a->totalslices],&size)); 1601d4002b98SHong Zhang PetscStackCallBLAS("BLASscal",BLASscal_(&size,&oalpha,aval,&one)); 16029566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(a->nz)); 16039566063dSJacob Faibussowitsch PetscCall(MatSeqSELLInvalidateDiagonal(inA)); 1604d4002b98SHong Zhang PetscFunctionReturn(0); 1605d4002b98SHong Zhang } 1606d4002b98SHong Zhang 1607d4002b98SHong Zhang PetscErrorCode MatShift_SeqSELL(Mat Y,PetscScalar a) 1608d4002b98SHong Zhang { 1609d4002b98SHong Zhang Mat_SeqSELL *y=(Mat_SeqSELL*)Y->data; 1610d4002b98SHong Zhang 1611d4002b98SHong Zhang PetscFunctionBegin; 1612d4002b98SHong Zhang if (!Y->preallocated || !y->nz) { 16139566063dSJacob Faibussowitsch PetscCall(MatSeqSELLSetPreallocation(Y,1,NULL)); 1614d4002b98SHong Zhang } 16159566063dSJacob Faibussowitsch PetscCall(MatShift_Basic(Y,a)); 1616d4002b98SHong Zhang PetscFunctionReturn(0); 1617d4002b98SHong Zhang } 1618d4002b98SHong Zhang 1619d4002b98SHong Zhang PetscErrorCode MatSOR_SeqSELL(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 1620d4002b98SHong Zhang { 1621d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 1622d4002b98SHong Zhang PetscScalar *x,sum,*t; 1623f4259b30SLisandro Dalcin const MatScalar *idiag=NULL,*mdiag; 1624d4002b98SHong Zhang const PetscScalar *b,*xb; 1625d4002b98SHong Zhang PetscInt n,m=A->rmap->n,i,j,shift; 1626d4002b98SHong Zhang const PetscInt *diag; 1627d4002b98SHong Zhang 1628d4002b98SHong Zhang PetscFunctionBegin; 1629d4002b98SHong Zhang its = its*lits; 1630d4002b98SHong Zhang 1631d4002b98SHong Zhang if (fshift != a->fshift || omega != a->omega) a->idiagvalid = PETSC_FALSE; /* must recompute idiag[] */ 16329566063dSJacob Faibussowitsch if (!a->idiagvalid) PetscCall(MatInvertDiagonal_SeqSELL(A,omega,fshift)); 1633d4002b98SHong Zhang a->fshift = fshift; 1634d4002b98SHong Zhang a->omega = omega; 1635d4002b98SHong Zhang 1636d4002b98SHong Zhang diag = a->diag; 1637d4002b98SHong Zhang t = a->ssor_work; 1638d4002b98SHong Zhang idiag = a->idiag; 1639d4002b98SHong Zhang mdiag = a->mdiag; 1640d4002b98SHong Zhang 16419566063dSJacob Faibussowitsch PetscCall(VecGetArray(xx,&x)); 16429566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(bb,&b)); 1643d4002b98SHong Zhang /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */ 164408401ef6SPierre Jolivet PetscCheck(flag != SOR_APPLY_UPPER,PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_UPPER is not implemented"); 164508401ef6SPierre Jolivet PetscCheck(flag != SOR_APPLY_LOWER,PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_LOWER is not implemented"); 1646aed4548fSBarry Smith PetscCheck(!(flag & SOR_EISENSTAT),PETSC_COMM_SELF,PETSC_ERR_SUP,"No support yet for Eisenstat"); 1647d4002b98SHong Zhang 1648d4002b98SHong Zhang if (flag & SOR_ZERO_INITIAL_GUESS) { 1649d4002b98SHong Zhang if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) { 1650d4002b98SHong Zhang for (i=0; i<m; i++) { 1651d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1652d4002b98SHong Zhang sum = b[i]; 1653d4002b98SHong Zhang n = (diag[i]-shift)/8; 1654d4002b98SHong Zhang for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]]; 1655d4002b98SHong Zhang t[i] = sum; 1656d4002b98SHong Zhang x[i] = sum*idiag[i]; 1657d4002b98SHong Zhang } 1658d4002b98SHong Zhang xb = t; 16599566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(a->nz)); 1660d4002b98SHong Zhang } else xb = b; 1661d4002b98SHong Zhang if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) { 1662d4002b98SHong Zhang for (i=m-1; i>=0; i--) { 1663d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1664d4002b98SHong Zhang sum = xb[i]; 1665d4002b98SHong Zhang n = a->rlen[i]-(diag[i]-shift)/8-1; 1666d4002b98SHong Zhang for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]]; 1667d4002b98SHong Zhang if (xb == b) { 1668d4002b98SHong Zhang x[i] = sum*idiag[i]; 1669d4002b98SHong Zhang } else { 1670d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+sum*idiag[i]; /* omega in idiag */ 1671d4002b98SHong Zhang } 1672d4002b98SHong Zhang } 16739566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(a->nz)); /* assumes 1/2 in upper */ 1674d4002b98SHong Zhang } 1675d4002b98SHong Zhang its--; 1676d4002b98SHong Zhang } 1677d4002b98SHong Zhang while (its--) { 1678d4002b98SHong Zhang if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) { 1679d4002b98SHong Zhang for (i=0; i<m; i++) { 1680d4002b98SHong Zhang /* lower */ 1681d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1682d4002b98SHong Zhang sum = b[i]; 1683d4002b98SHong Zhang n = (diag[i]-shift)/8; 1684d4002b98SHong Zhang for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]]; 1685d4002b98SHong Zhang t[i] = sum; /* save application of the lower-triangular part */ 1686d4002b98SHong Zhang /* upper */ 1687d4002b98SHong Zhang n = a->rlen[i]-(diag[i]-shift)/8-1; 1688d4002b98SHong Zhang for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]]; 1689d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+sum*idiag[i]; /* omega in idiag */ 1690d4002b98SHong Zhang } 1691d4002b98SHong Zhang xb = t; 16929566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0*a->nz)); 1693d4002b98SHong Zhang } else xb = b; 1694d4002b98SHong Zhang if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) { 1695d4002b98SHong Zhang for (i=m-1; i>=0; i--) { 1696d4002b98SHong Zhang shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */ 1697d4002b98SHong Zhang sum = xb[i]; 1698d4002b98SHong Zhang if (xb == b) { 1699d4002b98SHong Zhang /* whole matrix (no checkpointing available) */ 1700d4002b98SHong Zhang n = a->rlen[i]; 1701d4002b98SHong Zhang for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]]; 1702d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+(sum+mdiag[i]*x[i])*idiag[i]; 1703d4002b98SHong Zhang } else { /* lower-triangular part has been saved, so only apply upper-triangular */ 1704d4002b98SHong Zhang n = a->rlen[i]-(diag[i]-shift)/8-1; 1705d4002b98SHong Zhang for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]]; 1706d4002b98SHong Zhang x[i] = (1.-omega)*x[i]+sum*idiag[i]; /* omega in idiag */ 1707d4002b98SHong Zhang } 1708d4002b98SHong Zhang } 1709d4002b98SHong Zhang if (xb == b) { 17109566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0*a->nz)); 1711d4002b98SHong Zhang } else { 17129566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(a->nz)); /* assumes 1/2 in upper */ 1713d4002b98SHong Zhang } 1714d4002b98SHong Zhang } 1715d4002b98SHong Zhang } 17169566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(xx,&x)); 17179566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(bb,&b)); 1718d4002b98SHong Zhang PetscFunctionReturn(0); 1719d4002b98SHong Zhang } 1720d4002b98SHong Zhang 1721d4002b98SHong Zhang /* -------------------------------------------------------------------*/ 1722d4002b98SHong Zhang static struct _MatOps MatOps_Values = {MatSetValues_SeqSELL, 17236108893eSStefano Zampini MatGetRow_SeqSELL, 17246108893eSStefano Zampini MatRestoreRow_SeqSELL, 1725d4002b98SHong Zhang MatMult_SeqSELL, 1726d4002b98SHong Zhang /* 4*/ MatMultAdd_SeqSELL, 1727d4002b98SHong Zhang MatMultTranspose_SeqSELL, 1728d4002b98SHong Zhang MatMultTransposeAdd_SeqSELL, 1729f4259b30SLisandro Dalcin NULL, 1730f4259b30SLisandro Dalcin NULL, 1731f4259b30SLisandro Dalcin NULL, 1732f4259b30SLisandro Dalcin /* 10*/ NULL, 1733f4259b30SLisandro Dalcin NULL, 1734f4259b30SLisandro Dalcin NULL, 1735d4002b98SHong Zhang MatSOR_SeqSELL, 1736f4259b30SLisandro Dalcin NULL, 1737d4002b98SHong Zhang /* 15*/ MatGetInfo_SeqSELL, 1738d4002b98SHong Zhang MatEqual_SeqSELL, 1739d4002b98SHong Zhang MatGetDiagonal_SeqSELL, 1740d4002b98SHong Zhang MatDiagonalScale_SeqSELL, 1741f4259b30SLisandro Dalcin NULL, 1742f4259b30SLisandro Dalcin /* 20*/ NULL, 1743d4002b98SHong Zhang MatAssemblyEnd_SeqSELL, 1744d4002b98SHong Zhang MatSetOption_SeqSELL, 1745d4002b98SHong Zhang MatZeroEntries_SeqSELL, 1746f4259b30SLisandro Dalcin /* 24*/ NULL, 1747f4259b30SLisandro Dalcin NULL, 1748f4259b30SLisandro Dalcin NULL, 1749f4259b30SLisandro Dalcin NULL, 1750f4259b30SLisandro Dalcin NULL, 1751d4002b98SHong Zhang /* 29*/ MatSetUp_SeqSELL, 1752f4259b30SLisandro Dalcin NULL, 1753f4259b30SLisandro Dalcin NULL, 1754f4259b30SLisandro Dalcin NULL, 1755f4259b30SLisandro Dalcin NULL, 1756d4002b98SHong Zhang /* 34*/ MatDuplicate_SeqSELL, 1757f4259b30SLisandro Dalcin NULL, 1758f4259b30SLisandro Dalcin NULL, 1759f4259b30SLisandro Dalcin NULL, 1760f4259b30SLisandro Dalcin NULL, 1761f4259b30SLisandro Dalcin /* 39*/ NULL, 1762f4259b30SLisandro Dalcin NULL, 1763f4259b30SLisandro Dalcin NULL, 1764d4002b98SHong Zhang MatGetValues_SeqSELL, 1765d4002b98SHong Zhang MatCopy_SeqSELL, 1766f4259b30SLisandro Dalcin /* 44*/ NULL, 1767d4002b98SHong Zhang MatScale_SeqSELL, 1768d4002b98SHong Zhang MatShift_SeqSELL, 1769f4259b30SLisandro Dalcin NULL, 1770f4259b30SLisandro Dalcin NULL, 1771f4259b30SLisandro Dalcin /* 49*/ NULL, 1772f4259b30SLisandro Dalcin NULL, 1773f4259b30SLisandro Dalcin NULL, 1774f4259b30SLisandro Dalcin NULL, 1775f4259b30SLisandro Dalcin NULL, 1776d4002b98SHong Zhang /* 54*/ MatFDColoringCreate_SeqXAIJ, 1777f4259b30SLisandro Dalcin NULL, 1778f4259b30SLisandro Dalcin NULL, 1779f4259b30SLisandro Dalcin NULL, 1780f4259b30SLisandro Dalcin NULL, 1781f4259b30SLisandro Dalcin /* 59*/ NULL, 1782d4002b98SHong Zhang MatDestroy_SeqSELL, 1783d4002b98SHong Zhang MatView_SeqSELL, 1784f4259b30SLisandro Dalcin NULL, 1785f4259b30SLisandro Dalcin NULL, 1786f4259b30SLisandro Dalcin /* 64*/ NULL, 1787f4259b30SLisandro Dalcin NULL, 1788f4259b30SLisandro Dalcin NULL, 1789f4259b30SLisandro Dalcin NULL, 1790f4259b30SLisandro Dalcin NULL, 1791f4259b30SLisandro Dalcin /* 69*/ NULL, 1792f4259b30SLisandro Dalcin NULL, 1793f4259b30SLisandro Dalcin NULL, 1794f4259b30SLisandro Dalcin NULL, 1795f4259b30SLisandro Dalcin NULL, 1796f4259b30SLisandro Dalcin /* 74*/ NULL, 1797d4002b98SHong Zhang MatFDColoringApply_AIJ, /* reuse the FDColoring function for AIJ */ 1798f4259b30SLisandro Dalcin NULL, 1799f4259b30SLisandro Dalcin NULL, 1800f4259b30SLisandro Dalcin NULL, 1801f4259b30SLisandro Dalcin /* 79*/ NULL, 1802f4259b30SLisandro Dalcin NULL, 1803f4259b30SLisandro Dalcin NULL, 1804f4259b30SLisandro Dalcin NULL, 1805f4259b30SLisandro Dalcin NULL, 1806f4259b30SLisandro Dalcin /* 84*/ NULL, 1807f4259b30SLisandro Dalcin NULL, 1808f4259b30SLisandro Dalcin NULL, 1809f4259b30SLisandro Dalcin NULL, 1810f4259b30SLisandro Dalcin NULL, 1811f4259b30SLisandro Dalcin /* 89*/ NULL, 1812f4259b30SLisandro Dalcin NULL, 1813f4259b30SLisandro Dalcin NULL, 1814f4259b30SLisandro Dalcin NULL, 1815f4259b30SLisandro Dalcin NULL, 1816f4259b30SLisandro Dalcin /* 94*/ NULL, 1817f4259b30SLisandro Dalcin NULL, 1818f4259b30SLisandro Dalcin NULL, 1819f4259b30SLisandro Dalcin NULL, 1820f4259b30SLisandro Dalcin NULL, 1821f4259b30SLisandro Dalcin /* 99*/ NULL, 1822f4259b30SLisandro Dalcin NULL, 1823f4259b30SLisandro Dalcin NULL, 1824d4002b98SHong Zhang MatConjugate_SeqSELL, 1825f4259b30SLisandro Dalcin NULL, 1826f4259b30SLisandro Dalcin /*104*/ NULL, 1827f4259b30SLisandro Dalcin NULL, 1828f4259b30SLisandro Dalcin NULL, 1829f4259b30SLisandro Dalcin NULL, 1830f4259b30SLisandro Dalcin NULL, 1831f4259b30SLisandro Dalcin /*109*/ NULL, 1832f4259b30SLisandro Dalcin NULL, 1833f4259b30SLisandro Dalcin NULL, 1834f4259b30SLisandro Dalcin NULL, 1835d4002b98SHong Zhang MatMissingDiagonal_SeqSELL, 1836f4259b30SLisandro Dalcin /*114*/ NULL, 1837f4259b30SLisandro Dalcin NULL, 1838f4259b30SLisandro Dalcin NULL, 1839f4259b30SLisandro Dalcin NULL, 1840f4259b30SLisandro Dalcin NULL, 1841f4259b30SLisandro Dalcin /*119*/ NULL, 1842f4259b30SLisandro Dalcin NULL, 1843f4259b30SLisandro Dalcin NULL, 1844f4259b30SLisandro Dalcin NULL, 1845f4259b30SLisandro Dalcin NULL, 1846f4259b30SLisandro Dalcin /*124*/ NULL, 1847f4259b30SLisandro Dalcin NULL, 1848f4259b30SLisandro Dalcin NULL, 1849f4259b30SLisandro Dalcin NULL, 1850f4259b30SLisandro Dalcin NULL, 1851f4259b30SLisandro Dalcin /*129*/ NULL, 1852f4259b30SLisandro Dalcin NULL, 1853f4259b30SLisandro Dalcin NULL, 1854f4259b30SLisandro Dalcin NULL, 1855f4259b30SLisandro Dalcin NULL, 1856f4259b30SLisandro Dalcin /*134*/ NULL, 1857f4259b30SLisandro Dalcin NULL, 1858f4259b30SLisandro Dalcin NULL, 1859f4259b30SLisandro Dalcin NULL, 1860f4259b30SLisandro Dalcin NULL, 1861f4259b30SLisandro Dalcin /*139*/ NULL, 1862f4259b30SLisandro Dalcin NULL, 1863f4259b30SLisandro Dalcin NULL, 1864d4002b98SHong Zhang MatFDColoringSetUp_SeqXAIJ, 1865f4259b30SLisandro Dalcin NULL, 1866d70f29a3SPierre Jolivet /*144*/ NULL, 1867d70f29a3SPierre Jolivet NULL, 1868d70f29a3SPierre Jolivet NULL, 1869d70f29a3SPierre Jolivet NULL 1870d4002b98SHong Zhang }; 1871d4002b98SHong Zhang 1872d4002b98SHong Zhang PetscErrorCode MatStoreValues_SeqSELL(Mat mat) 1873d4002b98SHong Zhang { 1874d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)mat->data; 1875d4002b98SHong Zhang 1876d4002b98SHong Zhang PetscFunctionBegin; 187728b400f6SJacob Faibussowitsch PetscCheck(a->nonew,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1878d4002b98SHong Zhang 1879d4002b98SHong Zhang /* allocate space for values if not already there */ 1880d4002b98SHong Zhang if (!a->saved_values) { 18819566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(a->sliidx[a->totalslices]+1,&a->saved_values)); 18829566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)mat,(a->sliidx[a->totalslices]+1)*sizeof(PetscScalar))); 1883d4002b98SHong Zhang } 1884d4002b98SHong Zhang 1885d4002b98SHong Zhang /* copy values over */ 18869566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(a->saved_values,a->val,a->sliidx[a->totalslices])); 1887d4002b98SHong Zhang PetscFunctionReturn(0); 1888d4002b98SHong Zhang } 1889d4002b98SHong Zhang 1890d4002b98SHong Zhang PetscErrorCode MatRetrieveValues_SeqSELL(Mat mat) 1891d4002b98SHong Zhang { 1892d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)mat->data; 1893d4002b98SHong Zhang 1894d4002b98SHong Zhang PetscFunctionBegin; 189528b400f6SJacob Faibussowitsch PetscCheck(a->nonew,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 189628b400f6SJacob Faibussowitsch PetscCheck(a->saved_values,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 18979566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(a->val,a->saved_values,a->sliidx[a->totalslices])); 1898d4002b98SHong Zhang PetscFunctionReturn(0); 1899d4002b98SHong Zhang } 1900d4002b98SHong Zhang 1901d4002b98SHong Zhang /*@C 1902d4002b98SHong Zhang MatSeqSELLRestoreArray - returns access to the array where the data for a MATSEQSELL matrix is stored obtained by MatSeqSELLGetArray() 1903d4002b98SHong Zhang 1904d4002b98SHong Zhang Not Collective 1905d4002b98SHong Zhang 1906d4002b98SHong Zhang Input Parameters: 1907d4002b98SHong Zhang . mat - a MATSEQSELL matrix 1908d4002b98SHong Zhang . array - pointer to the data 1909d4002b98SHong Zhang 1910d4002b98SHong Zhang Level: intermediate 1911d4002b98SHong Zhang 1912db781477SPatrick Sanan .seealso: `MatSeqSELLGetArray()`, `MatSeqSELLRestoreArrayF90()` 1913d4002b98SHong Zhang @*/ 1914d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray(Mat A,PetscScalar **array) 1915d4002b98SHong Zhang { 1916d4002b98SHong Zhang PetscFunctionBegin; 1917cac4c232SBarry Smith PetscUseMethod(A,"MatSeqSELLRestoreArray_C",(Mat,PetscScalar**),(A,array)); 1918d4002b98SHong Zhang PetscFunctionReturn(0); 1919d4002b98SHong Zhang } 1920d4002b98SHong Zhang 1921d4002b98SHong Zhang PETSC_EXTERN PetscErrorCode MatCreate_SeqSELL(Mat B) 1922d4002b98SHong Zhang { 1923d4002b98SHong Zhang Mat_SeqSELL *b; 1924d4002b98SHong Zhang PetscMPIInt size; 1925d4002b98SHong Zhang 1926d4002b98SHong Zhang PetscFunctionBegin; 19279566063dSJacob Faibussowitsch PetscCall(PetscCitationsRegister(citation,&cited)); 19289566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)B),&size)); 192908401ef6SPierre Jolivet PetscCheck(size <= 1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Comm must be of size 1"); 1930d4002b98SHong Zhang 19319566063dSJacob Faibussowitsch PetscCall(PetscNewLog(B,&b)); 1932d4002b98SHong Zhang 1933d4002b98SHong Zhang B->data = (void*)b; 1934d4002b98SHong Zhang 19359566063dSJacob Faibussowitsch PetscCall(PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps))); 1936d4002b98SHong Zhang 1937f4259b30SLisandro Dalcin b->row = NULL; 1938f4259b30SLisandro Dalcin b->col = NULL; 1939f4259b30SLisandro Dalcin b->icol = NULL; 1940d4002b98SHong Zhang b->reallocs = 0; 1941d4002b98SHong Zhang b->ignorezeroentries = PETSC_FALSE; 1942d4002b98SHong Zhang b->roworiented = PETSC_TRUE; 1943d4002b98SHong Zhang b->nonew = 0; 1944f4259b30SLisandro Dalcin b->diag = NULL; 1945f4259b30SLisandro Dalcin b->solve_work = NULL; 1946f4259b30SLisandro Dalcin B->spptr = NULL; 1947f4259b30SLisandro Dalcin b->saved_values = NULL; 1948f4259b30SLisandro Dalcin b->idiag = NULL; 1949f4259b30SLisandro Dalcin b->mdiag = NULL; 1950f4259b30SLisandro Dalcin b->ssor_work = NULL; 1951d4002b98SHong Zhang b->omega = 1.0; 1952d4002b98SHong Zhang b->fshift = 0.0; 1953d4002b98SHong Zhang b->idiagvalid = PETSC_FALSE; 1954d4002b98SHong Zhang b->keepnonzeropattern = PETSC_FALSE; 1955d4002b98SHong Zhang 19569566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)B,MATSEQSELL)); 19579566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLGetArray_C",MatSeqSELLGetArray_SeqSELL)); 19589566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLRestoreArray_C",MatSeqSELLRestoreArray_SeqSELL)); 19599566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSELL)); 19609566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSELL)); 19619566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLSetPreallocation_C",MatSeqSELLSetPreallocation_SeqSELL)); 19629566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsell_seqaij_C",MatConvert_SeqSELL_SeqAIJ)); 1963d4002b98SHong Zhang PetscFunctionReturn(0); 1964d4002b98SHong Zhang } 1965d4002b98SHong Zhang 1966d4002b98SHong Zhang /* 1967d4002b98SHong Zhang Given a matrix generated with MatGetFactor() duplicates all the information in A into B 1968d4002b98SHong Zhang */ 1969d4002b98SHong Zhang PetscErrorCode MatDuplicateNoCreate_SeqSELL(Mat C,Mat A,MatDuplicateOption cpvalues,PetscBool mallocmatspace) 1970d4002b98SHong Zhang { 1971ed73aabaSBarry Smith Mat_SeqSELL *c = (Mat_SeqSELL*)C->data,*a = (Mat_SeqSELL*)A->data; 1972d4002b98SHong Zhang PetscInt i,m=A->rmap->n; 1973d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 1974d4002b98SHong Zhang 1975d4002b98SHong Zhang PetscFunctionBegin; 1976d4002b98SHong Zhang C->factortype = A->factortype; 1977f4259b30SLisandro Dalcin c->row = NULL; 1978f4259b30SLisandro Dalcin c->col = NULL; 1979f4259b30SLisandro Dalcin c->icol = NULL; 1980d4002b98SHong Zhang c->reallocs = 0; 1981d4002b98SHong Zhang C->assembled = PETSC_TRUE; 1982d4002b98SHong Zhang 19839566063dSJacob Faibussowitsch PetscCall(PetscLayoutReference(A->rmap,&C->rmap)); 19849566063dSJacob Faibussowitsch PetscCall(PetscLayoutReference(A->cmap,&C->cmap)); 1985d4002b98SHong Zhang 19869566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(8*totalslices,&c->rlen)); 19879566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt))); 19889566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totalslices+1,&c->sliidx)); 19899566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)C, (totalslices+1)*sizeof(PetscInt))); 1990d4002b98SHong Zhang 1991d4002b98SHong Zhang for (i=0; i<m; i++) c->rlen[i] = a->rlen[i]; 1992d4002b98SHong Zhang for (i=0; i<totalslices+1; i++) c->sliidx[i] = a->sliidx[i]; 1993d4002b98SHong Zhang 1994d4002b98SHong Zhang /* allocate the matrix space */ 1995d4002b98SHong Zhang if (mallocmatspace) { 19969566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(a->maxallocmat,&c->val,a->maxallocmat,&c->colidx)); 19979566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)C,a->maxallocmat*(sizeof(PetscScalar)+sizeof(PetscInt)))); 1998d4002b98SHong Zhang 1999d4002b98SHong Zhang c->singlemalloc = PETSC_TRUE; 2000d4002b98SHong Zhang 2001d4002b98SHong Zhang if (m > 0) { 20029566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(c->colidx,a->colidx,a->maxallocmat)); 2003d4002b98SHong Zhang if (cpvalues == MAT_COPY_VALUES) { 20049566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(c->val,a->val,a->maxallocmat)); 2005d4002b98SHong Zhang } else { 20069566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(c->val,a->maxallocmat)); 2007d4002b98SHong Zhang } 2008d4002b98SHong Zhang } 2009d4002b98SHong Zhang } 2010d4002b98SHong Zhang 2011d4002b98SHong Zhang c->ignorezeroentries = a->ignorezeroentries; 2012d4002b98SHong Zhang c->roworiented = a->roworiented; 2013d4002b98SHong Zhang c->nonew = a->nonew; 2014d4002b98SHong Zhang if (a->diag) { 20159566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(m,&c->diag)); 20169566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt))); 2017d4002b98SHong Zhang for (i=0; i<m; i++) { 2018d4002b98SHong Zhang c->diag[i] = a->diag[i]; 2019d4002b98SHong Zhang } 2020f4259b30SLisandro Dalcin } else c->diag = NULL; 2021d4002b98SHong Zhang 2022f4259b30SLisandro Dalcin c->solve_work = NULL; 2023f4259b30SLisandro Dalcin c->saved_values = NULL; 2024f4259b30SLisandro Dalcin c->idiag = NULL; 2025f4259b30SLisandro Dalcin c->ssor_work = NULL; 2026d4002b98SHong Zhang c->keepnonzeropattern = a->keepnonzeropattern; 2027d4002b98SHong Zhang c->free_val = PETSC_TRUE; 2028d4002b98SHong Zhang c->free_colidx = PETSC_TRUE; 2029d4002b98SHong Zhang 2030d4002b98SHong Zhang c->maxallocmat = a->maxallocmat; 2031d4002b98SHong Zhang c->maxallocrow = a->maxallocrow; 2032d4002b98SHong Zhang c->rlenmax = a->rlenmax; 2033d4002b98SHong Zhang c->nz = a->nz; 2034d4002b98SHong Zhang C->preallocated = PETSC_TRUE; 2035d4002b98SHong Zhang 2036d4002b98SHong Zhang c->nonzerorowcnt = a->nonzerorowcnt; 2037d4002b98SHong Zhang C->nonzerostate = A->nonzerostate; 2038d4002b98SHong Zhang 20399566063dSJacob Faibussowitsch PetscCall(PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist)); 2040d4002b98SHong Zhang PetscFunctionReturn(0); 2041d4002b98SHong Zhang } 2042d4002b98SHong Zhang 2043d4002b98SHong Zhang PetscErrorCode MatDuplicate_SeqSELL(Mat A,MatDuplicateOption cpvalues,Mat *B) 2044d4002b98SHong Zhang { 2045d4002b98SHong Zhang PetscFunctionBegin; 20469566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A),B)); 20479566063dSJacob Faibussowitsch PetscCall(MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n)); 2048d4002b98SHong Zhang if (!(A->rmap->n % A->rmap->bs) && !(A->cmap->n % A->cmap->bs)) { 20499566063dSJacob Faibussowitsch PetscCall(MatSetBlockSizesFromMats(*B,A,A)); 2050d4002b98SHong Zhang } 20519566063dSJacob Faibussowitsch PetscCall(MatSetType(*B,((PetscObject)A)->type_name)); 20529566063dSJacob Faibussowitsch PetscCall(MatDuplicateNoCreate_SeqSELL(*B,A,cpvalues,PETSC_TRUE)); 2053d4002b98SHong Zhang PetscFunctionReturn(0); 2054d4002b98SHong Zhang } 2055d4002b98SHong Zhang 2056ed73aabaSBarry Smith /*MC 2057ed73aabaSBarry Smith MATSEQSELL - MATSEQSELL = "seqsell" - A matrix type to be used for sequential sparse matrices, 2058ed73aabaSBarry Smith based on the sliced Ellpack format 2059ed73aabaSBarry Smith 2060ed73aabaSBarry Smith Options Database Keys: 2061ed73aabaSBarry Smith . -mat_type seqsell - sets the matrix type to "seqsell" during a call to MatSetFromOptions() 2062ed73aabaSBarry Smith 2063ed73aabaSBarry Smith Level: beginner 2064ed73aabaSBarry Smith 2065db781477SPatrick Sanan .seealso: `MatCreateSeqSell()`, `MATSELL`, `MATMPISELL`, `MATSEQAIJ`, `MATAIJ`, `MATMPIAIJ` 2066ed73aabaSBarry Smith M*/ 2067ed73aabaSBarry Smith 2068ed73aabaSBarry Smith /*MC 2069ed73aabaSBarry Smith MATSELL - MATSELL = "sell" - A matrix type to be used for sparse matrices. 2070ed73aabaSBarry Smith 2071ed73aabaSBarry Smith This matrix type is identical to MATSEQSELL when constructed with a single process communicator, 2072ed73aabaSBarry Smith and MATMPISELL otherwise. As a result, for single process communicators, 2073ed73aabaSBarry Smith MatSeqSELLSetPreallocation() is supported, and similarly MatMPISELLSetPreallocation() is supported 2074ed73aabaSBarry Smith for communicators controlling multiple processes. It is recommended that you call both of 2075ed73aabaSBarry Smith the above preallocation routines for simplicity. 2076ed73aabaSBarry Smith 2077ed73aabaSBarry Smith Options Database Keys: 2078ed73aabaSBarry Smith . -mat_type sell - sets the matrix type to "sell" during a call to MatSetFromOptions() 2079ed73aabaSBarry Smith 2080ed73aabaSBarry Smith Level: beginner 2081ed73aabaSBarry Smith 2082ed73aabaSBarry Smith Notes: 2083ed73aabaSBarry Smith This format is only supported for real scalars, double precision, and 32 bit indices (the defaults). 2084ed73aabaSBarry Smith 2085ed73aabaSBarry Smith It can provide better performance on Intel and AMD processes with AVX2 or AVX512 support for matrices that have a similar number of 2086ed73aabaSBarry Smith non-zeros in contiguous groups of rows. However if the computation is memory bandwidth limited it may not provide much improvement. 2087ed73aabaSBarry Smith 2088ed73aabaSBarry Smith Developer Notes: 2089ed73aabaSBarry Smith On Intel (and AMD) systems some of the matrix operations use SIMD (AVX) instructions to achieve higher performance. 2090ed73aabaSBarry Smith 2091ed73aabaSBarry Smith The sparse matrix format is as follows. For simplicity we assume a slice size of 2, it is actually 8 2092ed73aabaSBarry Smith .vb 2093ed73aabaSBarry Smith (2 0 3 4) 2094ed73aabaSBarry Smith Consider the matrix A = (5 0 6 0) 2095ed73aabaSBarry Smith (0 0 7 8) 2096ed73aabaSBarry Smith (0 0 9 9) 2097ed73aabaSBarry Smith 2098ed73aabaSBarry Smith symbolically the Ellpack format can be written as 2099ed73aabaSBarry Smith 2100ed73aabaSBarry Smith (2 3 4 |) (0 2 3 |) 2101ed73aabaSBarry Smith v = (5 6 0 |) colidx = (0 2 2 |) 2102ed73aabaSBarry Smith -------- --------- 2103ed73aabaSBarry Smith (7 8 |) (2 3 |) 2104ed73aabaSBarry Smith (9 9 |) (2 3 |) 2105ed73aabaSBarry Smith 2106ed73aabaSBarry Smith The data for 2 contiguous rows of the matrix are stored together (in column-major format) (with any left-over rows handled as a special case). 2107ed73aabaSBarry Smith Any of the rows in a slice fewer columns than the rest of the slice (row 1 above) are padded with a previous valid column in their "extra" colidx[] locations and 2108ed73aabaSBarry Smith zeros in their "extra" v locations so that the matrix operations do not need special code to handle different length rows within the 2 rows in a slice. 2109ed73aabaSBarry Smith 2110ed73aabaSBarry Smith The one-dimensional representation of v used in the code is (2 5 3 6 4 0 7 9 8 9) and for colidx is (0 0 2 2 3 2 2 2 3 3) 2111ed73aabaSBarry Smith 2112ed73aabaSBarry Smith .ve 2113ed73aabaSBarry Smith 2114ed73aabaSBarry Smith See MatMult_SeqSELL() for how this format is used with the SIMD operations to achieve high performance. 2115ed73aabaSBarry Smith 2116ed73aabaSBarry Smith References: 2117606c0280SSatish Balay . * - Hong Zhang, Richard T. Mills, Karl Rupp, and Barry F. Smith, Vectorized Parallel Sparse Matrix-Vector Multiplication in {PETSc} Using {AVX-512}, 2118ed73aabaSBarry Smith Proceedings of the 47th International Conference on Parallel Processing, 2018. 2119ed73aabaSBarry Smith 2120db781477SPatrick Sanan .seealso: `MatCreateSeqSELL()`, `MatCreateSeqAIJ()`, `MatCreateSell()`, `MATSEQSELL`, `MATMPISELL`, `MATSEQAIJ`, `MATMPIAIJ`, `MATAIJ` 2121ed73aabaSBarry Smith M*/ 2122ed73aabaSBarry Smith 2123d4002b98SHong Zhang /*@C 2124d4002b98SHong Zhang MatCreateSeqSELL - Creates a sparse matrix in SELL format. 2125d4002b98SHong Zhang 2126ed73aabaSBarry Smith Collective on comm 2127d4002b98SHong Zhang 2128d4002b98SHong Zhang Input Parameters: 2129d4002b98SHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 2130d4002b98SHong Zhang . m - number of rows 2131d4002b98SHong Zhang . n - number of columns 2132d4002b98SHong Zhang . rlenmax - maximum number of nonzeros in a row 2133d4002b98SHong Zhang - rlen - array containing the number of nonzeros in the various rows 2134d4002b98SHong Zhang (possibly different for each row) or NULL 2135d4002b98SHong Zhang 2136d4002b98SHong Zhang Output Parameter: 2137d4002b98SHong Zhang . A - the matrix 2138d4002b98SHong Zhang 2139d4002b98SHong Zhang It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 2140f6f02116SRichard Tran Mills MatXXXXSetPreallocation() paradigm instead of this routine directly. 2141d4002b98SHong Zhang [MatXXXXSetPreallocation() is, for example, MatSeqSELLSetPreallocation] 2142d4002b98SHong Zhang 2143d4002b98SHong Zhang Notes: 2144d4002b98SHong Zhang If nnz is given then nz is ignored 2145d4002b98SHong Zhang 2146d4002b98SHong Zhang Specify the preallocated storage with either rlenmax or rlen (not both). 2147d4002b98SHong Zhang Set rlenmax=PETSC_DEFAULT and rlen=NULL for PETSc to control dynamic memory 2148d4002b98SHong Zhang allocation. For large problems you MUST preallocate memory or you 2149d4002b98SHong Zhang will get TERRIBLE performance, see the users' manual chapter on matrices. 2150d4002b98SHong Zhang 2151d4002b98SHong Zhang Level: intermediate 2152d4002b98SHong Zhang 2153db781477SPatrick Sanan .seealso: `MatCreate()`, `MatCreateSELL()`, `MatSetValues()`, `MatSeqSELLSetPreallocation()`, `MATSELL`, `MATSEQSELL`, `MATMPISELL` 2154d4002b98SHong Zhang 2155d4002b98SHong Zhang @*/ 2156d4002b98SHong Zhang PetscErrorCode MatCreateSeqSELL(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt maxallocrow,const PetscInt rlen[],Mat *A) 2157d4002b98SHong Zhang { 2158d4002b98SHong Zhang PetscFunctionBegin; 21599566063dSJacob Faibussowitsch PetscCall(MatCreate(comm,A)); 21609566063dSJacob Faibussowitsch PetscCall(MatSetSizes(*A,m,n,m,n)); 21619566063dSJacob Faibussowitsch PetscCall(MatSetType(*A,MATSEQSELL)); 21629566063dSJacob Faibussowitsch PetscCall(MatSeqSELLSetPreallocation_SeqSELL(*A,maxallocrow,rlen)); 2163d4002b98SHong Zhang PetscFunctionReturn(0); 2164d4002b98SHong Zhang } 2165d4002b98SHong Zhang 2166d4002b98SHong Zhang PetscErrorCode MatEqual_SeqSELL(Mat A,Mat B,PetscBool * flg) 2167d4002b98SHong Zhang { 2168d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data,*b=(Mat_SeqSELL*)B->data; 2169d4002b98SHong Zhang PetscInt totalslices=a->totalslices; 2170d4002b98SHong Zhang 2171d4002b98SHong Zhang PetscFunctionBegin; 2172d4002b98SHong Zhang /* If the matrix dimensions are not equal,or no of nonzeros */ 2173d4002b98SHong Zhang if ((A->rmap->n != B->rmap->n) || (A->cmap->n != B->cmap->n) ||(a->nz != b->nz) || (a->rlenmax != b->rlenmax)) { 2174d4002b98SHong Zhang *flg = PETSC_FALSE; 2175d4002b98SHong Zhang PetscFunctionReturn(0); 2176d4002b98SHong Zhang } 2177d4002b98SHong Zhang /* if the a->colidx are the same */ 21789566063dSJacob Faibussowitsch PetscCall(PetscArraycmp(a->colidx,b->colidx,a->sliidx[totalslices],flg)); 2179d4002b98SHong Zhang if (!*flg) PetscFunctionReturn(0); 2180d4002b98SHong Zhang /* if a->val are the same */ 21819566063dSJacob Faibussowitsch PetscCall(PetscArraycmp(a->val,b->val,a->sliidx[totalslices],flg)); 2182d4002b98SHong Zhang PetscFunctionReturn(0); 2183d4002b98SHong Zhang } 2184d4002b98SHong Zhang 2185d4002b98SHong Zhang PetscErrorCode MatSeqSELLInvalidateDiagonal(Mat A) 2186d4002b98SHong Zhang { 2187d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 2188d4002b98SHong Zhang 2189d4002b98SHong Zhang PetscFunctionBegin; 2190d4002b98SHong Zhang a->idiagvalid = PETSC_FALSE; 2191d4002b98SHong Zhang PetscFunctionReturn(0); 2192d4002b98SHong Zhang } 2193d4002b98SHong Zhang 2194d4002b98SHong Zhang PetscErrorCode MatConjugate_SeqSELL(Mat A) 2195d4002b98SHong Zhang { 2196d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 2197d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 2198d4002b98SHong Zhang PetscInt i; 2199d4002b98SHong Zhang PetscScalar *val = a->val; 2200d4002b98SHong Zhang 2201d4002b98SHong Zhang PetscFunctionBegin; 2202d4002b98SHong Zhang for (i=0; i<a->sliidx[a->totalslices]; i++) { 2203d4002b98SHong Zhang val[i] = PetscConj(val[i]); 2204d4002b98SHong Zhang } 2205d4002b98SHong Zhang #else 2206d4002b98SHong Zhang PetscFunctionBegin; 2207d4002b98SHong Zhang #endif 2208d4002b98SHong Zhang PetscFunctionReturn(0); 2209d4002b98SHong Zhang } 2210