xref: /petsc/src/mat/impls/sell/seq/sell.c (revision 99a7f59ec9f9ccee3cdf9fd88ce16d7f72105d85)
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;
1781baa6e33SBarry Smith   if (realalloc) PetscCall(MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE));
179d4002b98SHong Zhang   PetscFunctionReturn(0);
180d4002b98SHong Zhang }
181d4002b98SHong Zhang 
1826108893eSStefano Zampini PetscErrorCode MatGetRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
1836108893eSStefano Zampini {
1846108893eSStefano Zampini   Mat_SeqSELL *a = (Mat_SeqSELL*)A->data;
1856108893eSStefano Zampini   PetscInt    shift;
1866108893eSStefano Zampini 
1876108893eSStefano Zampini   PetscFunctionBegin;
188aed4548fSBarry Smith   PetscCheck(row >= 0 && row < A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %" PetscInt_FMT " out of range",row);
1896108893eSStefano Zampini   if (nz) *nz = a->rlen[row];
1906108893eSStefano Zampini   shift = a->sliidx[row>>3]+(row&0x07);
1916108893eSStefano Zampini   if (!a->getrowcols) {
1926108893eSStefano Zampini 
1939566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(a->rlenmax,&a->getrowcols,a->rlenmax,&a->getrowvals));
1946108893eSStefano Zampini   }
1956108893eSStefano Zampini   if (idx) {
1966108893eSStefano Zampini     PetscInt j;
1976108893eSStefano Zampini     for (j=0; j<a->rlen[row]; j++) a->getrowcols[j] = a->colidx[shift+8*j];
1986108893eSStefano Zampini     *idx = a->getrowcols;
1996108893eSStefano Zampini   }
2006108893eSStefano Zampini   if (v) {
2016108893eSStefano Zampini     PetscInt j;
2026108893eSStefano Zampini     for (j=0; j<a->rlen[row]; j++) a->getrowvals[j] = a->val[shift+8*j];
2036108893eSStefano Zampini     *v = a->getrowvals;
2046108893eSStefano Zampini   }
2056108893eSStefano Zampini   PetscFunctionReturn(0);
2066108893eSStefano Zampini }
2076108893eSStefano Zampini 
2086108893eSStefano Zampini PetscErrorCode MatRestoreRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
2096108893eSStefano Zampini {
2106108893eSStefano Zampini   PetscFunctionBegin;
2116108893eSStefano Zampini   PetscFunctionReturn(0);
2126108893eSStefano Zampini }
2136108893eSStefano Zampini 
214d4002b98SHong Zhang PetscErrorCode MatConvert_SeqSELL_SeqAIJ(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
215d4002b98SHong Zhang {
216d4002b98SHong Zhang   Mat            B;
217d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
218e3f1f374SStefano Zampini   PetscInt       i;
219d4002b98SHong Zhang 
220d4002b98SHong Zhang   PetscFunctionBegin;
221ad013a7bSRichard Tran Mills   if (reuse == MAT_REUSE_MATRIX) {
222ad013a7bSRichard Tran Mills     B    = *newmat;
2239566063dSJacob Faibussowitsch     PetscCall(MatZeroEntries(B));
224ad013a7bSRichard Tran Mills   } else {
2259566063dSJacob Faibussowitsch     PetscCall(MatCreate(PetscObjectComm((PetscObject)A),&B));
2269566063dSJacob Faibussowitsch     PetscCall(MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N));
2279566063dSJacob Faibussowitsch     PetscCall(MatSetType(B,MATSEQAIJ));
2289566063dSJacob Faibussowitsch     PetscCall(MatSeqAIJSetPreallocation(B,0,a->rlen));
229ad013a7bSRichard Tran Mills   }
230d4002b98SHong Zhang 
231e3f1f374SStefano Zampini   for (i=0; i<A->rmap->n; i++) {
232e108cb99SStefano Zampini     PetscInt    nz = 0,*cols = NULL;
233e108cb99SStefano Zampini     PetscScalar *vals = NULL;
234e3f1f374SStefano Zampini 
2359566063dSJacob Faibussowitsch     PetscCall(MatGetRow_SeqSELL(A,i,&nz,&cols,&vals));
2369566063dSJacob Faibussowitsch     PetscCall(MatSetValues(B,1,&i,nz,cols,vals,INSERT_VALUES));
2379566063dSJacob Faibussowitsch     PetscCall(MatRestoreRow_SeqSELL(A,i,&nz,&cols,&vals));
238d4002b98SHong Zhang   }
239e3f1f374SStefano Zampini 
2409566063dSJacob Faibussowitsch   PetscCall(MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY));
2419566063dSJacob Faibussowitsch   PetscCall(MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY));
242d4002b98SHong Zhang   B->rmap->bs = A->rmap->bs;
243d4002b98SHong Zhang 
244d4002b98SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
2459566063dSJacob Faibussowitsch     PetscCall(MatHeaderReplace(A,&B));
246d4002b98SHong Zhang   } else {
247d4002b98SHong Zhang     *newmat = B;
248d4002b98SHong Zhang   }
249d4002b98SHong Zhang   PetscFunctionReturn(0);
250d4002b98SHong Zhang }
251d4002b98SHong Zhang 
252d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/aij.h>
253d4002b98SHong Zhang 
254d4002b98SHong Zhang PetscErrorCode MatConvert_SeqAIJ_SeqSELL(Mat A,MatType newtype,MatReuse reuse,Mat *newmat)
255d4002b98SHong Zhang {
256d4002b98SHong Zhang   Mat               B;
257d4002b98SHong Zhang   Mat_SeqAIJ        *a=(Mat_SeqAIJ*)A->data;
258d4002b98SHong Zhang   PetscInt          *ai=a->i,m=A->rmap->N,n=A->cmap->N,i,*rowlengths,row,ncols;
259d4002b98SHong Zhang   const PetscInt    *cols;
260d4002b98SHong Zhang   const PetscScalar *vals;
261d4002b98SHong Zhang 
262d4002b98SHong Zhang   PetscFunctionBegin;
263ad013a7bSRichard Tran Mills 
264ad013a7bSRichard Tran Mills   if (reuse == MAT_REUSE_MATRIX) {
265ad013a7bSRichard Tran Mills     B = *newmat;
266ad013a7bSRichard Tran Mills   } else {
267d5e5b2e5SBarry Smith     if (PetscDefined(USE_DEBUG) || !a->ilen) {
2689566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(m,&rowlengths));
269d4002b98SHong Zhang       for (i=0; i<m; i++) {
270d4002b98SHong Zhang         rowlengths[i] = ai[i+1] - ai[i];
271d4002b98SHong Zhang       }
272d5e5b2e5SBarry Smith     }
273d5e5b2e5SBarry Smith     if (PetscDefined(USE_DEBUG) && a->ilen) {
274d5e5b2e5SBarry Smith       PetscBool eq;
2759566063dSJacob Faibussowitsch       PetscCall(PetscMemcmp(rowlengths,a->ilen,m*sizeof(PetscInt),&eq));
27628b400f6SJacob Faibussowitsch       PetscCheck(eq,PETSC_COMM_SELF,PETSC_ERR_PLIB,"SeqAIJ ilen array incorrect");
2779566063dSJacob Faibussowitsch       PetscCall(PetscFree(rowlengths));
278d5e5b2e5SBarry Smith       rowlengths = a->ilen;
279d5e5b2e5SBarry Smith     } else if (a->ilen) rowlengths = a->ilen;
2809566063dSJacob Faibussowitsch     PetscCall(MatCreate(PetscObjectComm((PetscObject)A),&B));
2819566063dSJacob Faibussowitsch     PetscCall(MatSetSizes(B,m,n,m,n));
2829566063dSJacob Faibussowitsch     PetscCall(MatSetType(B,MATSEQSELL));
2839566063dSJacob Faibussowitsch     PetscCall(MatSeqSELLSetPreallocation(B,0,rowlengths));
2849566063dSJacob Faibussowitsch     if (rowlengths != a->ilen) PetscCall(PetscFree(rowlengths));
285ad013a7bSRichard Tran Mills   }
286d4002b98SHong Zhang 
287d4002b98SHong Zhang   for (row=0; row<m; row++) {
2889566063dSJacob Faibussowitsch     PetscCall(MatGetRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals));
2899566063dSJacob Faibussowitsch     PetscCall(MatSetValues_SeqSELL(B,1,&row,ncols,cols,vals,INSERT_VALUES));
2909566063dSJacob Faibussowitsch     PetscCall(MatRestoreRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals));
291d4002b98SHong Zhang   }
2929566063dSJacob Faibussowitsch   PetscCall(MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY));
2939566063dSJacob Faibussowitsch   PetscCall(MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY));
294d4002b98SHong Zhang   B->rmap->bs = A->rmap->bs;
295d4002b98SHong Zhang 
296d4002b98SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
2979566063dSJacob Faibussowitsch     PetscCall(MatHeaderReplace(A,&B));
298d4002b98SHong Zhang   } else {
299d4002b98SHong Zhang     *newmat = B;
300d4002b98SHong Zhang   }
301d4002b98SHong Zhang   PetscFunctionReturn(0);
302d4002b98SHong Zhang }
303d4002b98SHong Zhang 
304d4002b98SHong Zhang PetscErrorCode MatMult_SeqSELL(Mat A,Vec xx,Vec yy)
305d4002b98SHong Zhang {
306d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
307d4002b98SHong Zhang   PetscScalar       *y;
308d4002b98SHong Zhang   const PetscScalar *x;
309d4002b98SHong Zhang   const MatScalar   *aval=a->val;
310d4002b98SHong Zhang   PetscInt          totalslices=a->totalslices;
311d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
3127285fed1SHong Zhang   PetscInt          i,j;
313d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
314d4002b98SHong Zhang   __m512d           vec_x,vec_y,vec_vals;
315d4002b98SHong Zhang   __m256i           vec_idx;
316d4002b98SHong Zhang   __mmask8          mask;
317d4002b98SHong Zhang   __m512d           vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4;
318d4002b98SHong Zhang   __m256i           vec_idx2,vec_idx3,vec_idx4;
3195f70456aSHong 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)
320a48a6482SHong Zhang   __m128i           vec_idx;
321a48a6482SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
322a48a6482SHong Zhang   MatScalar         yval;
323a48a6482SHong Zhang   PetscInt          r,rows_left,row,nnz_in_row;
32421cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
325d4002b98SHong Zhang   __m128d           vec_x_tmp;
326d4002b98SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
327d4002b98SHong Zhang   MatScalar         yval;
328d4002b98SHong Zhang   PetscInt          r,rows_left,row,nnz_in_row;
329d4002b98SHong Zhang #else
330d4002b98SHong Zhang   PetscScalar       sum[8];
331d4002b98SHong Zhang #endif
332d4002b98SHong Zhang 
333d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
334d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
335d4002b98SHong Zhang #endif
336d4002b98SHong Zhang 
337d4002b98SHong Zhang   PetscFunctionBegin;
3389566063dSJacob Faibussowitsch   PetscCall(VecGetArrayRead(xx,&x));
3399566063dSJacob Faibussowitsch   PetscCall(VecGetArray(yy,&y));
340d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
341d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
342d4002b98SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
343d4002b98SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
344d4002b98SHong Zhang 
345d4002b98SHong Zhang     vec_y  = _mm512_setzero_pd();
346d4002b98SHong Zhang     vec_y2 = _mm512_setzero_pd();
347d4002b98SHong Zhang     vec_y3 = _mm512_setzero_pd();
348d4002b98SHong Zhang     vec_y4 = _mm512_setzero_pd();
349d4002b98SHong Zhang 
35038efe8efSHong Zhang     j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */
351d4002b98SHong Zhang     switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) {
352d4002b98SHong Zhang     case 3:
353d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
354d4002b98SHong Zhang       acolidx += 8; aval += 8;
355d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
356d4002b98SHong Zhang       acolidx += 8; aval += 8;
357d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
358d4002b98SHong Zhang       acolidx += 8; aval += 8;
359d4002b98SHong Zhang       j += 3;
360d4002b98SHong Zhang       break;
361d4002b98SHong Zhang     case 2:
362d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
363d4002b98SHong Zhang       acolidx += 8; aval += 8;
364d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
365d4002b98SHong Zhang       acolidx += 8; aval += 8;
366d4002b98SHong Zhang       j += 2;
367d4002b98SHong Zhang       break;
368d4002b98SHong Zhang     case 1:
369d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
370d4002b98SHong Zhang       acolidx += 8; aval += 8;
371d4002b98SHong Zhang       j += 1;
372d4002b98SHong Zhang       break;
373d4002b98SHong Zhang     }
374d4002b98SHong Zhang     #pragma novector
375d4002b98SHong Zhang     for (; j<(a->sliidx[i+1]>>3); j+=4) {
376d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
377d4002b98SHong Zhang       acolidx += 8; aval += 8;
378d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
379d4002b98SHong Zhang       acolidx += 8; aval += 8;
380d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
381d4002b98SHong Zhang       acolidx += 8; aval += 8;
382d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4);
383d4002b98SHong Zhang       acolidx += 8; aval += 8;
384d4002b98SHong Zhang     }
385d4002b98SHong Zhang 
386d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y2);
387d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y3);
388d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y4);
389d4002b98SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
390d4002b98SHong Zhang       mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07)));
391ef588d5cSRichard Tran Mills       _mm512_mask_storeu_pd(&y[8*i],mask,vec_y);
392d4002b98SHong Zhang     } else {
393ef588d5cSRichard Tran Mills       _mm512_storeu_pd(&y[8*i],vec_y);
394d4002b98SHong Zhang     }
395d4002b98SHong Zhang   }
3965f70456aSHong 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)
397a48a6482SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
398a48a6482SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
399a48a6482SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
400a48a6482SHong Zhang 
401a48a6482SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
402a48a6482SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
403a48a6482SHong Zhang       rows_left = A->rmap->n - 8*i;
404a48a6482SHong Zhang       for (r=0; r<rows_left; ++r) {
405a48a6482SHong Zhang         yval = (MatScalar)0;
406a48a6482SHong Zhang         row = 8*i + r;
407a48a6482SHong Zhang         nnz_in_row = a->rlen[row];
408a48a6482SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]];
409a48a6482SHong Zhang         y[row] = yval;
410a48a6482SHong Zhang       }
411a48a6482SHong Zhang       break;
412a48a6482SHong Zhang     }
413a48a6482SHong Zhang 
414a48a6482SHong Zhang     vec_y  = _mm256_setzero_pd();
415a48a6482SHong Zhang     vec_y2 = _mm256_setzero_pd();
416a48a6482SHong Zhang 
417a48a6482SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
418a48a6482SHong Zhang     #pragma novector
419a48a6482SHong Zhang     #pragma unroll(2)
420a48a6482SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
421a48a6482SHong Zhang       AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
422a48a6482SHong Zhang       aval += 4; acolidx += 4;
423a48a6482SHong Zhang       AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y2);
424a48a6482SHong Zhang       aval += 4; acolidx += 4;
425a48a6482SHong Zhang     }
426a48a6482SHong Zhang 
427ef588d5cSRichard Tran Mills     _mm256_storeu_pd(y+i*8,vec_y);
428ef588d5cSRichard Tran Mills     _mm256_storeu_pd(y+i*8+4,vec_y2);
429a48a6482SHong Zhang   }
43021cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
431d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
432d4002b98SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
433d4002b98SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
434d4002b98SHong Zhang 
435d4002b98SHong Zhang     vec_y  = _mm256_setzero_pd();
436d4002b98SHong Zhang     vec_y2 = _mm256_setzero_pd();
437d4002b98SHong Zhang 
438d4002b98SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
439d4002b98SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
440d4002b98SHong Zhang       rows_left = A->rmap->n - 8*i;
441d4002b98SHong Zhang       for (r=0; r<rows_left; ++r) {
442d4002b98SHong Zhang         yval = (MatScalar)0;
443d4002b98SHong Zhang         row = 8*i + r;
444d4002b98SHong Zhang         nnz_in_row = a->rlen[row];
445d4002b98SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j + r] * x[acolidx[8*j + r]];
446d4002b98SHong Zhang         y[row] = yval;
447d4002b98SHong Zhang       }
448d4002b98SHong Zhang       break;
449d4002b98SHong Zhang     }
450d4002b98SHong Zhang 
451d4002b98SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
452a48a6482SHong Zhang     #pragma novector
453a48a6482SHong Zhang     #pragma unroll(2)
4547285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
455d4002b98SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
456165f9cc3SJed Brown       vec_x_tmp = _mm_setzero_pd();
457d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
458d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
459d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
460d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
461d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
462d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
463d4002b98SHong Zhang       vec_y     = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y);
464d4002b98SHong Zhang       aval     += 4;
465d4002b98SHong Zhang 
466d4002b98SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
467d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
468d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
469d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
470d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
471d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
472d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
473d4002b98SHong Zhang       vec_y2    = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2);
474d4002b98SHong Zhang       aval     += 4;
475d4002b98SHong Zhang     }
476d4002b98SHong Zhang 
477d4002b98SHong Zhang     _mm256_storeu_pd(y + i*8,     vec_y);
478d4002b98SHong Zhang     _mm256_storeu_pd(y + i*8 + 4, vec_y2);
479d4002b98SHong Zhang   }
480d4002b98SHong Zhang #else
481d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
482d4002b98SHong Zhang     for (j=0; j<8; j++) sum[j] = 0.0;
483d4002b98SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
484d4002b98SHong Zhang       sum[0] += aval[j] * x[acolidx[j]];
485d4002b98SHong Zhang       sum[1] += aval[j+1] * x[acolidx[j+1]];
486d4002b98SHong Zhang       sum[2] += aval[j+2] * x[acolidx[j+2]];
487d4002b98SHong Zhang       sum[3] += aval[j+3] * x[acolidx[j+3]];
488d4002b98SHong Zhang       sum[4] += aval[j+4] * x[acolidx[j+4]];
489d4002b98SHong Zhang       sum[5] += aval[j+5] * x[acolidx[j+5]];
490d4002b98SHong Zhang       sum[6] += aval[j+6] * x[acolidx[j+6]];
491d4002b98SHong Zhang       sum[7] += aval[j+7] * x[acolidx[j+7]];
492d4002b98SHong Zhang     }
493d4002b98SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
494d4002b98SHong Zhang       for (j=0; j<(A->rmap->n & 0x07); j++) y[8*i+j] = sum[j];
495d4002b98SHong Zhang     } else {
4967285fed1SHong Zhang       for (j=0; j<8; j++) y[8*i+j] = sum[j];
497d4002b98SHong Zhang     }
498d4002b98SHong Zhang   }
499d4002b98SHong Zhang #endif
500d4002b98SHong Zhang 
5019566063dSJacob Faibussowitsch   PetscCall(PetscLogFlops(2.0*a->nz-a->nonzerorowcnt)); /* theoretical minimal FLOPs */
5029566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayRead(xx,&x));
5039566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(yy,&y));
504d4002b98SHong Zhang   PetscFunctionReturn(0);
505d4002b98SHong Zhang }
506d4002b98SHong Zhang 
507d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/ftn-kernels/fmultadd.h>
508d4002b98SHong Zhang PetscErrorCode MatMultAdd_SeqSELL(Mat A,Vec xx,Vec yy,Vec zz)
509d4002b98SHong Zhang {
510d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
511d4002b98SHong Zhang   PetscScalar       *y,*z;
512d4002b98SHong Zhang   const PetscScalar *x;
513d4002b98SHong Zhang   const MatScalar   *aval=a->val;
514d4002b98SHong Zhang   PetscInt          totalslices=a->totalslices;
515d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
516d4002b98SHong Zhang   PetscInt          i,j;
517d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5187285fed1SHong Zhang   __m512d           vec_x,vec_y,vec_vals;
519d4002b98SHong Zhang   __m256i           vec_idx;
520d4002b98SHong Zhang   __mmask8          mask;
5217285fed1SHong Zhang   __m512d           vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4;
5227285fed1SHong Zhang   __m256i           vec_idx2,vec_idx3,vec_idx4;
52321cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5247285fed1SHong Zhang   __m128d           vec_x_tmp;
5257285fed1SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
5267285fed1SHong Zhang   MatScalar         yval;
5277285fed1SHong Zhang   PetscInt          r,row,nnz_in_row;
528d4002b98SHong Zhang #else
529d4002b98SHong Zhang   PetscScalar       sum[8];
530d4002b98SHong Zhang #endif
531d4002b98SHong Zhang 
532d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
533d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
534d4002b98SHong Zhang #endif
535d4002b98SHong Zhang 
536d4002b98SHong Zhang   PetscFunctionBegin;
5379566063dSJacob Faibussowitsch   PetscCall(VecGetArrayRead(xx,&x));
5389566063dSJacob Faibussowitsch   PetscCall(VecGetArrayPair(yy,zz,&y,&z));
539d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5407285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
5417285fed1SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
5427285fed1SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
5437285fed1SHong Zhang 
544d4002b98SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
545d4002b98SHong Zhang       mask   = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07)));
546ef588d5cSRichard Tran Mills       vec_y  = _mm512_mask_loadu_pd(vec_y,mask,&y[8*i]);
5477285fed1SHong Zhang     } else {
548ef588d5cSRichard Tran Mills       vec_y  = _mm512_loadu_pd(&y[8*i]);
5497285fed1SHong Zhang     }
5507285fed1SHong Zhang     vec_y2 = _mm512_setzero_pd();
5517285fed1SHong Zhang     vec_y3 = _mm512_setzero_pd();
5527285fed1SHong Zhang     vec_y4 = _mm512_setzero_pd();
5537285fed1SHong Zhang 
5547285fed1SHong Zhang     j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */
5557285fed1SHong Zhang     switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) {
5567285fed1SHong Zhang     case 3:
5577285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5587285fed1SHong Zhang       acolidx += 8; aval += 8;
5597285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5607285fed1SHong Zhang       acolidx += 8; aval += 8;
5617285fed1SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
5627285fed1SHong Zhang       acolidx += 8; aval += 8;
5637285fed1SHong Zhang       j += 3;
5647285fed1SHong Zhang       break;
5657285fed1SHong Zhang     case 2:
5667285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5677285fed1SHong Zhang       acolidx += 8; aval += 8;
5687285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5697285fed1SHong Zhang       acolidx += 8; aval += 8;
5707285fed1SHong Zhang       j += 2;
5717285fed1SHong Zhang       break;
5727285fed1SHong Zhang     case 1:
5737285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5747285fed1SHong Zhang       acolidx += 8; aval += 8;
5757285fed1SHong Zhang       j += 1;
5767285fed1SHong Zhang       break;
5777285fed1SHong Zhang     }
5787285fed1SHong Zhang     #pragma novector
5797285fed1SHong Zhang     for (; j<(a->sliidx[i+1]>>3); j+=4) {
5807285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5817285fed1SHong Zhang       acolidx += 8; aval += 8;
5827285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5837285fed1SHong Zhang       acolidx += 8; aval += 8;
5847285fed1SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
5857285fed1SHong Zhang       acolidx += 8; aval += 8;
5867285fed1SHong Zhang       AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4);
5877285fed1SHong Zhang       acolidx += 8; aval += 8;
5887285fed1SHong Zhang     }
5897285fed1SHong Zhang 
5907285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y2);
5917285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y3);
5927285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y4);
5937285fed1SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
594ef588d5cSRichard Tran Mills       _mm512_mask_storeu_pd(&z[8*i],mask,vec_y);
595d4002b98SHong Zhang     } else {
596ef588d5cSRichard Tran Mills       _mm512_storeu_pd(&z[8*i],vec_y);
597d4002b98SHong Zhang     }
5987285fed1SHong Zhang   }
59921cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
6007285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
6017285fed1SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
6027285fed1SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
6037285fed1SHong Zhang 
6047285fed1SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
6057285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6067285fed1SHong Zhang       for (r=0; r<(A->rmap->n & 0x07); ++r) {
6077285fed1SHong Zhang         row        = 8*i + r;
6087285fed1SHong Zhang         yval       = (MatScalar)0.0;
6097285fed1SHong Zhang         nnz_in_row = a->rlen[row];
6107285fed1SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]];
6117285fed1SHong Zhang         z[row] = y[row] + yval;
6127285fed1SHong Zhang       }
6137285fed1SHong Zhang       break;
6147285fed1SHong Zhang     }
6157285fed1SHong Zhang 
6167285fed1SHong Zhang     vec_y  = _mm256_loadu_pd(y+8*i);
6177285fed1SHong Zhang     vec_y2 = _mm256_loadu_pd(y+8*i+4);
6187285fed1SHong Zhang 
6197285fed1SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
6207285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
6217285fed1SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
622165f9cc3SJed Brown       vec_x_tmp = _mm_setzero_pd();
6237285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6247285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
625165f9cc3SJed Brown       vec_x     = _mm256_setzero_pd();
6267285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
6277285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6287285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6297285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
6307285fed1SHong Zhang       vec_y     = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y);
6317285fed1SHong Zhang       aval     += 4;
6327285fed1SHong Zhang 
6337285fed1SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
6347285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6357285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6367285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
6377285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6387285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6397285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
6407285fed1SHong Zhang       vec_y2    = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2);
6417285fed1SHong Zhang       aval     += 4;
6427285fed1SHong Zhang     }
6437285fed1SHong Zhang 
6447285fed1SHong Zhang     _mm256_storeu_pd(z+i*8,vec_y);
6457285fed1SHong Zhang     _mm256_storeu_pd(z+i*8+4,vec_y2);
6467285fed1SHong Zhang   }
647d4002b98SHong Zhang #else
6487285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
6497285fed1SHong Zhang     for (j=0; j<8; j++) sum[j] = 0.0;
650d4002b98SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
651d4002b98SHong Zhang       sum[0] += aval[j] * x[acolidx[j]];
652d4002b98SHong Zhang       sum[1] += aval[j+1] * x[acolidx[j+1]];
653d4002b98SHong Zhang       sum[2] += aval[j+2] * x[acolidx[j+2]];
654d4002b98SHong Zhang       sum[3] += aval[j+3] * x[acolidx[j+3]];
655d4002b98SHong Zhang       sum[4] += aval[j+4] * x[acolidx[j+4]];
656d4002b98SHong Zhang       sum[5] += aval[j+5] * x[acolidx[j+5]];
657d4002b98SHong Zhang       sum[6] += aval[j+6] * x[acolidx[j+6]];
658d4002b98SHong Zhang       sum[7] += aval[j+7] * x[acolidx[j+7]];
659d4002b98SHong Zhang     }
6607285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6617285fed1SHong Zhang       for (j=0; j<(A->rmap->n & 0x07); j++) z[8*i+j] = y[8*i+j] + sum[j];
662d4002b98SHong Zhang     } else {
6637285fed1SHong Zhang       for (j=0; j<8; j++) z[8*i+j] = y[8*i+j] + sum[j];
6647285fed1SHong Zhang     }
665d4002b98SHong Zhang   }
666d4002b98SHong Zhang #endif
667d4002b98SHong Zhang 
6689566063dSJacob Faibussowitsch   PetscCall(PetscLogFlops(2.0*a->nz));
6699566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayRead(xx,&x));
6709566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayPair(yy,zz,&y,&z));
671d4002b98SHong Zhang   PetscFunctionReturn(0);
672d4002b98SHong Zhang }
673d4002b98SHong Zhang 
674d4002b98SHong Zhang PetscErrorCode MatMultTransposeAdd_SeqSELL(Mat A,Vec xx,Vec zz,Vec yy)
675d4002b98SHong Zhang {
676d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
677d4002b98SHong Zhang   PetscScalar       *y;
678d4002b98SHong Zhang   const PetscScalar *x;
679d4002b98SHong Zhang   const MatScalar   *aval=a->val;
680d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
6817285fed1SHong Zhang   PetscInt          i,j,r,row,nnz_in_row,totalslices=a->totalslices;
682d4002b98SHong Zhang 
683d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
684d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
685d4002b98SHong Zhang #endif
686d4002b98SHong Zhang 
687d4002b98SHong Zhang   PetscFunctionBegin;
6889fc32365SStefano Zampini   if (A->symmetric) {
6899566063dSJacob Faibussowitsch     PetscCall(MatMultAdd_SeqSELL(A,xx,zz,yy));
6909fc32365SStefano Zampini     PetscFunctionReturn(0);
6919fc32365SStefano Zampini   }
6929566063dSJacob Faibussowitsch   if (zz != yy) PetscCall(VecCopy(zz,yy));
6939566063dSJacob Faibussowitsch   PetscCall(VecGetArrayRead(xx,&x));
6949566063dSJacob Faibussowitsch   PetscCall(VecGetArray(yy,&y));
695d4002b98SHong Zhang   for (i=0; i<a->totalslices; i++) { /* loop over slices */
6967285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6977285fed1SHong Zhang       for (r=0; r<(A->rmap->n & 0x07); ++r) {
6987285fed1SHong Zhang         row        = 8*i + r;
6997285fed1SHong Zhang         nnz_in_row = a->rlen[row];
7007285fed1SHong Zhang         for (j=0; j<nnz_in_row; ++j) y[acolidx[8*j+r]] += aval[8*j+r] * x[row];
7017285fed1SHong Zhang       }
7027285fed1SHong Zhang       break;
7037285fed1SHong Zhang     }
7047285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
7057285fed1SHong Zhang       y[acolidx[j]]   += aval[j] * x[8*i];
7067285fed1SHong Zhang       y[acolidx[j+1]] += aval[j+1] * x[8*i+1];
7077285fed1SHong Zhang       y[acolidx[j+2]] += aval[j+2] * x[8*i+2];
7087285fed1SHong Zhang       y[acolidx[j+3]] += aval[j+3] * x[8*i+3];
7097285fed1SHong Zhang       y[acolidx[j+4]] += aval[j+4] * x[8*i+4];
7107285fed1SHong Zhang       y[acolidx[j+5]] += aval[j+5] * x[8*i+5];
7117285fed1SHong Zhang       y[acolidx[j+6]] += aval[j+6] * x[8*i+6];
7127285fed1SHong Zhang       y[acolidx[j+7]] += aval[j+7] * x[8*i+7];
713d4002b98SHong Zhang     }
714d4002b98SHong Zhang   }
7159566063dSJacob Faibussowitsch   PetscCall(PetscLogFlops(2.0*a->sliidx[a->totalslices]));
7169566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayRead(xx,&x));
7179566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(yy,&y));
718d4002b98SHong Zhang   PetscFunctionReturn(0);
719d4002b98SHong Zhang }
720d4002b98SHong Zhang 
721d4002b98SHong Zhang PetscErrorCode MatMultTranspose_SeqSELL(Mat A,Vec xx,Vec yy)
722d4002b98SHong Zhang {
723d4002b98SHong Zhang   PetscFunctionBegin;
7249fc32365SStefano Zampini   if (A->symmetric) {
7259566063dSJacob Faibussowitsch     PetscCall(MatMult_SeqSELL(A,xx,yy));
7269fc32365SStefano Zampini   } else {
7279566063dSJacob Faibussowitsch     PetscCall(VecSet(yy,0.0));
7289566063dSJacob Faibussowitsch     PetscCall(MatMultTransposeAdd_SeqSELL(A,xx,yy,yy));
7299fc32365SStefano Zampini   }
730d4002b98SHong Zhang   PetscFunctionReturn(0);
731d4002b98SHong Zhang }
732d4002b98SHong Zhang 
733d4002b98SHong Zhang /*
734d4002b98SHong Zhang      Checks for missing diagonals
735d4002b98SHong Zhang */
736d4002b98SHong Zhang PetscErrorCode MatMissingDiagonal_SeqSELL(Mat A,PetscBool  *missing,PetscInt *d)
737d4002b98SHong Zhang {
738d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
739d4002b98SHong Zhang   PetscInt       *diag,i;
740d4002b98SHong Zhang 
741d4002b98SHong Zhang   PetscFunctionBegin;
742d4002b98SHong Zhang   *missing = PETSC_FALSE;
743d4002b98SHong Zhang   if (A->rmap->n > 0 && !(a->colidx)) {
744d4002b98SHong Zhang     *missing = PETSC_TRUE;
745d4002b98SHong Zhang     if (d) *d = 0;
7469566063dSJacob Faibussowitsch     PetscCall(PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n"));
747d4002b98SHong Zhang   } else {
748d4002b98SHong Zhang     diag = a->diag;
749d4002b98SHong Zhang     for (i=0; i<A->rmap->n; i++) {
750d4002b98SHong Zhang       if (diag[i] == -1) {
751d4002b98SHong Zhang         *missing = PETSC_TRUE;
752d4002b98SHong Zhang         if (d) *d = i;
7539566063dSJacob Faibussowitsch         PetscCall(PetscInfo(A,"Matrix is missing diagonal number %" PetscInt_FMT "\n",i));
754d4002b98SHong Zhang         break;
755d4002b98SHong Zhang       }
756d4002b98SHong Zhang     }
757d4002b98SHong Zhang   }
758d4002b98SHong Zhang   PetscFunctionReturn(0);
759d4002b98SHong Zhang }
760d4002b98SHong Zhang 
761d4002b98SHong Zhang PetscErrorCode MatMarkDiagonal_SeqSELL(Mat A)
762d4002b98SHong Zhang {
763d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
764d4002b98SHong Zhang   PetscInt       i,j,m=A->rmap->n,shift;
765d4002b98SHong Zhang 
766d4002b98SHong Zhang   PetscFunctionBegin;
767d4002b98SHong Zhang   if (!a->diag) {
7689566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(m,&a->diag));
7699566063dSJacob Faibussowitsch     PetscCall(PetscLogObjectMemory((PetscObject)A,m*sizeof(PetscInt)));
770d4002b98SHong Zhang     a->free_diag = PETSC_TRUE;
771d4002b98SHong Zhang   }
772d4002b98SHong Zhang   for (i=0; i<m; i++) { /* loop over rows */
773d4002b98SHong Zhang     shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
774d4002b98SHong Zhang     a->diag[i] = -1;
775d4002b98SHong Zhang     for (j=0; j<a->rlen[i]; j++) {
776d4002b98SHong Zhang       if (a->colidx[shift+j*8] == i) {
777d4002b98SHong Zhang         a->diag[i] = shift+j*8;
778d4002b98SHong Zhang         break;
779d4002b98SHong Zhang       }
780d4002b98SHong Zhang     }
781d4002b98SHong Zhang   }
782d4002b98SHong Zhang   PetscFunctionReturn(0);
783d4002b98SHong Zhang }
784d4002b98SHong Zhang 
785d4002b98SHong Zhang /*
786d4002b98SHong Zhang   Negative shift indicates do not generate an error if there is a zero diagonal, just invert it anyways
787d4002b98SHong Zhang */
788d4002b98SHong Zhang PetscErrorCode MatInvertDiagonal_SeqSELL(Mat A,PetscScalar omega,PetscScalar fshift)
789d4002b98SHong Zhang {
790d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*) A->data;
791d4002b98SHong Zhang   PetscInt       i,*diag,m = A->rmap->n;
792d4002b98SHong Zhang   MatScalar      *val = a->val;
793d4002b98SHong Zhang   PetscScalar    *idiag,*mdiag;
794d4002b98SHong Zhang 
795d4002b98SHong Zhang   PetscFunctionBegin;
796d4002b98SHong Zhang   if (a->idiagvalid) PetscFunctionReturn(0);
7979566063dSJacob Faibussowitsch   PetscCall(MatMarkDiagonal_SeqSELL(A));
798d4002b98SHong Zhang   diag = a->diag;
799d4002b98SHong Zhang   if (!a->idiag) {
8009566063dSJacob Faibussowitsch     PetscCall(PetscMalloc3(m,&a->idiag,m,&a->mdiag,m,&a->ssor_work));
8019566063dSJacob Faibussowitsch     PetscCall(PetscLogObjectMemory((PetscObject)A, 3*m*sizeof(PetscScalar)));
802d4002b98SHong Zhang     val  = a->val;
803d4002b98SHong Zhang   }
804d4002b98SHong Zhang   mdiag = a->mdiag;
805d4002b98SHong Zhang   idiag = a->idiag;
806d4002b98SHong Zhang 
807d4002b98SHong Zhang   if (omega == 1.0 && PetscRealPart(fshift) <= 0.0) {
808d4002b98SHong Zhang     for (i=0; i<m; i++) {
809d4002b98SHong Zhang       mdiag[i] = val[diag[i]];
810d4002b98SHong Zhang       if (!PetscAbsScalar(mdiag[i])) { /* zero diagonal */
811d4002b98SHong Zhang         if (PetscRealPart(fshift)) {
8129566063dSJacob Faibussowitsch           PetscCall(PetscInfo(A,"Zero diagonal on row %" PetscInt_FMT "\n",i));
813d4002b98SHong Zhang           A->factorerrortype             = MAT_FACTOR_NUMERIC_ZEROPIVOT;
814d4002b98SHong Zhang           A->factorerror_zeropivot_value = 0.0;
815d4002b98SHong Zhang           A->factorerror_zeropivot_row   = i;
81698921bdaSJacob Faibussowitsch         } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Zero diagonal on row %" PetscInt_FMT,i);
817d4002b98SHong Zhang       }
818d4002b98SHong Zhang       idiag[i] = 1.0/val[diag[i]];
819d4002b98SHong Zhang     }
8209566063dSJacob Faibussowitsch     PetscCall(PetscLogFlops(m));
821d4002b98SHong Zhang   } else {
822d4002b98SHong Zhang     for (i=0; i<m; i++) {
823d4002b98SHong Zhang       mdiag[i] = val[diag[i]];
824d4002b98SHong Zhang       idiag[i] = omega/(fshift + val[diag[i]]);
825d4002b98SHong Zhang     }
8269566063dSJacob Faibussowitsch     PetscCall(PetscLogFlops(2.0*m));
827d4002b98SHong Zhang   }
828d4002b98SHong Zhang   a->idiagvalid = PETSC_TRUE;
829d4002b98SHong Zhang   PetscFunctionReturn(0);
830d4002b98SHong Zhang }
831d4002b98SHong Zhang 
832d4002b98SHong Zhang PetscErrorCode MatZeroEntries_SeqSELL(Mat A)
833d4002b98SHong Zhang {
834d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
835d4002b98SHong Zhang 
836d4002b98SHong Zhang   PetscFunctionBegin;
8379566063dSJacob Faibussowitsch   PetscCall(PetscArrayzero(a->val,a->sliidx[a->totalslices]));
8389566063dSJacob Faibussowitsch   PetscCall(MatSeqSELLInvalidateDiagonal(A));
839d4002b98SHong Zhang   PetscFunctionReturn(0);
840d4002b98SHong Zhang }
841d4002b98SHong Zhang 
842d4002b98SHong Zhang PetscErrorCode MatDestroy_SeqSELL(Mat A)
843d4002b98SHong Zhang {
844d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
845d4002b98SHong Zhang 
846d4002b98SHong Zhang   PetscFunctionBegin;
847d4002b98SHong Zhang #if defined(PETSC_USE_LOG)
848c0aa6a63SJacob Faibussowitsch   PetscLogObjectState((PetscObject)A,"Rows=%" PetscInt_FMT ", Cols=%" PetscInt_FMT ", NZ=%" PetscInt_FMT,A->rmap->n,A->cmap->n,a->nz);
849d4002b98SHong Zhang #endif
8509566063dSJacob Faibussowitsch   PetscCall(MatSeqXSELLFreeSELL(A,&a->val,&a->colidx));
8519566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&a->row));
8529566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&a->col));
8539566063dSJacob Faibussowitsch   PetscCall(PetscFree(a->diag));
8549566063dSJacob Faibussowitsch   PetscCall(PetscFree(a->rlen));
8559566063dSJacob Faibussowitsch   PetscCall(PetscFree(a->sliidx));
8569566063dSJacob Faibussowitsch   PetscCall(PetscFree3(a->idiag,a->mdiag,a->ssor_work));
8579566063dSJacob Faibussowitsch   PetscCall(PetscFree(a->solve_work));
8589566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&a->icol));
8599566063dSJacob Faibussowitsch   PetscCall(PetscFree(a->saved_values));
8609566063dSJacob Faibussowitsch   PetscCall(PetscFree2(a->getrowcols,a->getrowvals));
861d4002b98SHong Zhang 
8629566063dSJacob Faibussowitsch   PetscCall(PetscFree(A->data));
863d4002b98SHong Zhang 
8649566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)A,NULL));
8659566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL));
8669566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL));
8679566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)A,"MatSeqSELLSetPreallocation_C",NULL));
8682e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)A,"MatSeqSELLGetArray_C",NULL));
8692e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)A,"MatSeqSELLRestoreArray_C",NULL));
8702e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsell_seqaij_C",NULL));
871d4002b98SHong Zhang   PetscFunctionReturn(0);
872d4002b98SHong Zhang }
873d4002b98SHong Zhang 
874d4002b98SHong Zhang PetscErrorCode MatSetOption_SeqSELL(Mat A,MatOption op,PetscBool flg)
875d4002b98SHong Zhang {
876d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
877d4002b98SHong Zhang 
878d4002b98SHong Zhang   PetscFunctionBegin;
879d4002b98SHong Zhang   switch (op) {
880d4002b98SHong Zhang   case MAT_ROW_ORIENTED:
881d4002b98SHong Zhang     a->roworiented = flg;
882d4002b98SHong Zhang     break;
883d4002b98SHong Zhang   case MAT_KEEP_NONZERO_PATTERN:
884d4002b98SHong Zhang     a->keepnonzeropattern = flg;
885d4002b98SHong Zhang     break;
886d4002b98SHong Zhang   case MAT_NEW_NONZERO_LOCATIONS:
887d4002b98SHong Zhang     a->nonew = (flg ? 0 : 1);
888d4002b98SHong Zhang     break;
889d4002b98SHong Zhang   case MAT_NEW_NONZERO_LOCATION_ERR:
890d4002b98SHong Zhang     a->nonew = (flg ? -1 : 0);
891d4002b98SHong Zhang     break;
892d4002b98SHong Zhang   case MAT_NEW_NONZERO_ALLOCATION_ERR:
893d4002b98SHong Zhang     a->nonew = (flg ? -2 : 0);
894d4002b98SHong Zhang     break;
895d4002b98SHong Zhang   case MAT_UNUSED_NONZERO_LOCATION_ERR:
896d4002b98SHong Zhang     a->nounused = (flg ? -1 : 0);
897d4002b98SHong Zhang     break;
8988c78258cSHong Zhang   case MAT_FORCE_DIAGONAL_ENTRIES:
899d4002b98SHong Zhang   case MAT_IGNORE_OFF_PROC_ENTRIES:
900d4002b98SHong Zhang   case MAT_USE_HASH_TABLE:
901071fcb05SBarry Smith   case MAT_SORTED_FULL:
9029566063dSJacob Faibussowitsch     PetscCall(PetscInfo(A,"Option %s ignored\n",MatOptions[op]));
903d4002b98SHong Zhang     break;
904d4002b98SHong Zhang   case MAT_SPD:
905d4002b98SHong Zhang   case MAT_SYMMETRIC:
906d4002b98SHong Zhang   case MAT_STRUCTURALLY_SYMMETRIC:
907d4002b98SHong Zhang   case MAT_HERMITIAN:
908d4002b98SHong Zhang   case MAT_SYMMETRY_ETERNAL:
909d4002b98SHong Zhang     /* These options are handled directly by MatSetOption() */
910d4002b98SHong Zhang     break;
911d4002b98SHong Zhang   default:
91298921bdaSJacob Faibussowitsch     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
913d4002b98SHong Zhang   }
914d4002b98SHong Zhang   PetscFunctionReturn(0);
915d4002b98SHong Zhang }
916d4002b98SHong Zhang 
917d4002b98SHong Zhang PetscErrorCode MatGetDiagonal_SeqSELL(Mat A,Vec v)
918d4002b98SHong Zhang {
919d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
920d4002b98SHong Zhang   PetscInt       i,j,n,shift;
921d4002b98SHong Zhang   PetscScalar    *x,zero=0.0;
922d4002b98SHong Zhang 
923d4002b98SHong Zhang   PetscFunctionBegin;
9249566063dSJacob Faibussowitsch   PetscCall(VecGetLocalSize(v,&n));
92508401ef6SPierre Jolivet   PetscCheck(n == A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
926d4002b98SHong Zhang 
927d4002b98SHong Zhang   if (A->factortype == MAT_FACTOR_ILU || A->factortype == MAT_FACTOR_LU) {
928d4002b98SHong Zhang     PetscInt *diag=a->diag;
9299566063dSJacob Faibussowitsch     PetscCall(VecGetArray(v,&x));
930d4002b98SHong Zhang     for (i=0; i<n; i++) x[i] = 1.0/a->val[diag[i]];
9319566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(v,&x));
932d4002b98SHong Zhang     PetscFunctionReturn(0);
933d4002b98SHong Zhang   }
934d4002b98SHong Zhang 
9359566063dSJacob Faibussowitsch   PetscCall(VecSet(v,zero));
9369566063dSJacob Faibussowitsch   PetscCall(VecGetArray(v,&x));
937d4002b98SHong Zhang   for (i=0; i<n; i++) { /* loop over rows */
938d4002b98SHong Zhang     shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
939d4002b98SHong Zhang     x[i] = 0;
940d4002b98SHong Zhang     for (j=0; j<a->rlen[i]; j++) {
941d4002b98SHong Zhang       if (a->colidx[shift+j*8] == i) {
942d4002b98SHong Zhang         x[i] = a->val[shift+j*8];
943d4002b98SHong Zhang         break;
944d4002b98SHong Zhang       }
945d4002b98SHong Zhang     }
946d4002b98SHong Zhang   }
9479566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(v,&x));
948d4002b98SHong Zhang   PetscFunctionReturn(0);
949d4002b98SHong Zhang }
950d4002b98SHong Zhang 
951d4002b98SHong Zhang PetscErrorCode MatDiagonalScale_SeqSELL(Mat A,Vec ll,Vec rr)
952d4002b98SHong Zhang {
953d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
954d4002b98SHong Zhang   const PetscScalar *l,*r;
955d4002b98SHong Zhang   PetscInt          i,j,m,n,row;
956d4002b98SHong Zhang 
957d4002b98SHong Zhang   PetscFunctionBegin;
958d4002b98SHong Zhang   if (ll) {
959d4002b98SHong Zhang     /* The local size is used so that VecMPI can be passed to this routine
960d4002b98SHong Zhang        by MatDiagonalScale_MPISELL */
9619566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(ll,&m));
96208401ef6SPierre Jolivet     PetscCheck(m == A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length");
9639566063dSJacob Faibussowitsch     PetscCall(VecGetArrayRead(ll,&l));
964d4002b98SHong Zhang     for (i=0; i<a->totalslices; i++) { /* loop over slices */
965dab86139SHong Zhang       if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
966dab86139SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
967dab86139SHong Zhang           if (row < (A->rmap->n & 0x07)) a->val[j] *= l[8*i+row];
968dab86139SHong Zhang         }
969dab86139SHong Zhang       } else {
970d4002b98SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
971d4002b98SHong Zhang           a->val[j] *= l[8*i+row];
972d4002b98SHong Zhang         }
973d4002b98SHong Zhang       }
974dab86139SHong Zhang     }
9759566063dSJacob Faibussowitsch     PetscCall(VecRestoreArrayRead(ll,&l));
9769566063dSJacob Faibussowitsch     PetscCall(PetscLogFlops(a->nz));
977d4002b98SHong Zhang   }
978d4002b98SHong Zhang   if (rr) {
9799566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(rr,&n));
98008401ef6SPierre Jolivet     PetscCheck(n == A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length");
9819566063dSJacob Faibussowitsch     PetscCall(VecGetArrayRead(rr,&r));
982d4002b98SHong Zhang     for (i=0; i<a->totalslices; i++) { /* loop over slices */
983dab86139SHong Zhang       if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
984dab86139SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
985dab86139SHong Zhang           if (row < (A->rmap->n & 0x07)) a->val[j] *= r[a->colidx[j]];
986dab86139SHong Zhang         }
987dab86139SHong Zhang       } else {
988d4002b98SHong Zhang         for (j=a->sliidx[i]; j<a->sliidx[i+1]; j++) {
989d4002b98SHong Zhang           a->val[j] *= r[a->colidx[j]];
990d4002b98SHong Zhang         }
991d4002b98SHong Zhang       }
992dab86139SHong Zhang     }
9939566063dSJacob Faibussowitsch     PetscCall(VecRestoreArrayRead(rr,&r));
9949566063dSJacob Faibussowitsch     PetscCall(PetscLogFlops(a->nz));
995d4002b98SHong Zhang   }
9969566063dSJacob Faibussowitsch   PetscCall(MatSeqSELLInvalidateDiagonal(A));
997d4002b98SHong Zhang   PetscFunctionReturn(0);
998d4002b98SHong Zhang }
999d4002b98SHong Zhang 
1000d4002b98SHong Zhang PetscErrorCode MatGetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
1001d4002b98SHong Zhang {
1002d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1003d4002b98SHong Zhang   PetscInt    *cp,i,k,low,high,t,row,col,l;
1004d4002b98SHong Zhang   PetscInt    shift;
1005d4002b98SHong Zhang   MatScalar   *vp;
1006d4002b98SHong Zhang 
1007d4002b98SHong Zhang   PetscFunctionBegin;
100868aafef3SStefano Zampini   for (k=0; k<m; k++) { /* loop over requested rows */
1009d4002b98SHong Zhang     row = im[k];
1010d4002b98SHong Zhang     if (row<0) continue;
10116bdcaf15SBarry 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);
1012d4002b98SHong Zhang     shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */
1013d4002b98SHong Zhang     cp = a->colidx+shift; /* pointer to the row */
1014d4002b98SHong Zhang     vp = a->val+shift; /* pointer to the row */
101568aafef3SStefano Zampini     for (l=0; l<n; l++) { /* loop over requested columns */
1016d4002b98SHong Zhang       col = in[l];
1017d4002b98SHong Zhang       if (col<0) continue;
10186bdcaf15SBarry 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);
1019d4002b98SHong Zhang       high = a->rlen[row]; low = 0; /* assume unsorted */
1020d4002b98SHong Zhang       while (high-low > 5) {
1021d4002b98SHong Zhang         t = (low+high)/2;
1022d4002b98SHong Zhang         if (*(cp+t*8) > col) high = t;
1023d4002b98SHong Zhang         else low = t;
1024d4002b98SHong Zhang       }
1025d4002b98SHong Zhang       for (i=low; i<high; i++) {
1026d4002b98SHong Zhang         if (*(cp+8*i) > col) break;
1027d4002b98SHong Zhang         if (*(cp+8*i) == col) {
1028d4002b98SHong Zhang           *v++ = *(vp+8*i);
1029d4002b98SHong Zhang           goto finished;
1030d4002b98SHong Zhang         }
1031d4002b98SHong Zhang       }
1032d4002b98SHong Zhang       *v++ = 0.0;
1033d4002b98SHong Zhang     finished:;
1034d4002b98SHong Zhang     }
1035d4002b98SHong Zhang   }
1036d4002b98SHong Zhang   PetscFunctionReturn(0);
1037d4002b98SHong Zhang }
1038d4002b98SHong Zhang 
1039d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_ASCII(Mat A,PetscViewer viewer)
1040d4002b98SHong Zhang {
1041d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1042d4002b98SHong Zhang   PetscInt          i,j,m=A->rmap->n,shift;
1043d4002b98SHong Zhang   const char        *name;
1044d4002b98SHong Zhang   PetscViewerFormat format;
1045d4002b98SHong Zhang 
1046d4002b98SHong Zhang   PetscFunctionBegin;
10479566063dSJacob Faibussowitsch   PetscCall(PetscViewerGetFormat(viewer,&format));
1048d4002b98SHong Zhang   if (format == PETSC_VIEWER_ASCII_MATLAB) {
1049d4002b98SHong Zhang     PetscInt nofinalvalue = 0;
1050d4002b98SHong Zhang     /*
1051d4002b98SHong Zhang     if (m && ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != A->cmap->n-1))) {
1052d4002b98SHong Zhang       nofinalvalue = 1;
1053d4002b98SHong Zhang     }
1054d4002b98SHong Zhang     */
10559566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
10569566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%% Size = %" PetscInt_FMT " %" PetscInt_FMT " \n",m,A->cmap->n));
10579566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%% Nonzeros = %" PetscInt_FMT " \n",a->nz));
1058d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
10599566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",4);\n",a->nz+nofinalvalue));
1060d4002b98SHong Zhang #else
10619566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",3);\n",a->nz+nofinalvalue));
1062d4002b98SHong Zhang #endif
10639566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"zzz = [\n"));
1064d4002b98SHong Zhang 
1065d4002b98SHong Zhang     for (i=0; i<m; i++) {
1066d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1067d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1068d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
10699566063dSJacob 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])));
1070d4002b98SHong Zhang #else
10719566063dSJacob 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]));
1072d4002b98SHong Zhang #endif
1073d4002b98SHong Zhang       }
1074d4002b98SHong Zhang     }
1075d4002b98SHong Zhang     /*
1076d4002b98SHong Zhang     if (nofinalvalue) {
1077d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
10789566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e %18.16e\n",m,A->cmap->n,0.,0.));
1079d4002b98SHong Zhang #else
10809566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e\n",m,A->cmap->n,0.0));
1081d4002b98SHong Zhang #endif
1082d4002b98SHong Zhang     }
1083d4002b98SHong Zhang     */
10849566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)A,&name));
10859566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"];\n %s = spconvert(zzz);\n",name));
10869566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
1087d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO || format == PETSC_VIEWER_ASCII_INFO) {
1088d4002b98SHong Zhang     PetscFunctionReturn(0);
1089d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
10909566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
1091d4002b98SHong Zhang     for (i=0; i<m; i++) {
10929566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i));
1093d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1094d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1095d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1096d4002b98SHong Zhang         if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) {
10979566063dSJacob 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])));
1098d4002b98SHong Zhang         } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) {
10999566063dSJacob 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])));
1100d4002b98SHong Zhang         } else if (PetscRealPart(a->val[shift+8*j]) != 0.0) {
11019566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j])));
1102d4002b98SHong Zhang         }
1103d4002b98SHong Zhang #else
11049566063dSJacob 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]));
1105d4002b98SHong Zhang #endif
1106d4002b98SHong Zhang       }
11079566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
1108d4002b98SHong Zhang     }
11099566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
1110d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_DENSE) {
1111d4002b98SHong Zhang     PetscInt    cnt=0,jcnt;
1112d4002b98SHong Zhang     PetscScalar value;
1113d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1114d4002b98SHong Zhang     PetscBool   realonly=PETSC_TRUE;
1115d4002b98SHong Zhang     for (i=0; i<a->sliidx[a->totalslices]; i++) {
1116d4002b98SHong Zhang       if (PetscImaginaryPart(a->val[i]) != 0.0) {
1117d4002b98SHong Zhang         realonly = PETSC_FALSE;
1118d4002b98SHong Zhang         break;
1119d4002b98SHong Zhang       }
1120d4002b98SHong Zhang     }
1121d4002b98SHong Zhang #endif
1122d4002b98SHong Zhang 
11239566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
1124d4002b98SHong Zhang     for (i=0; i<m; i++) {
1125d4002b98SHong Zhang       jcnt = 0;
1126d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1127d4002b98SHong Zhang       for (j=0; j<A->cmap->n; j++) {
1128d4002b98SHong Zhang         if (jcnt < a->rlen[i] && j == a->colidx[shift+8*j]) {
1129d4002b98SHong Zhang           value = a->val[cnt++];
1130d4002b98SHong Zhang           jcnt++;
1131d4002b98SHong Zhang         } else {
1132d4002b98SHong Zhang           value = 0.0;
1133d4002b98SHong Zhang         }
1134d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1135d4002b98SHong Zhang         if (realonly) {
11369566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer," %7.5e ",(double)PetscRealPart(value)));
1137d4002b98SHong Zhang         } else {
11389566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer," %7.5e+%7.5e i ",(double)PetscRealPart(value),(double)PetscImaginaryPart(value)));
1139d4002b98SHong Zhang         }
1140d4002b98SHong Zhang #else
11419566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," %7.5e ",(double)value));
1142d4002b98SHong Zhang #endif
1143d4002b98SHong Zhang       }
11449566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
1145d4002b98SHong Zhang     }
11469566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
1147d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_MATRIXMARKET) {
1148d4002b98SHong Zhang     PetscInt fshift=1;
11499566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
1150d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
11519566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate complex general\n"));
1152d4002b98SHong Zhang #else
11539566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate real general\n"));
1154d4002b98SHong Zhang #endif
11559566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %" PetscInt_FMT "\n", m, A->cmap->n, a->nz));
1156d4002b98SHong Zhang     for (i=0; i<m; i++) {
1157d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1158d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1159d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
11609566063dSJacob 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])));
1161d4002b98SHong Zhang #else
11629566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)a->val[shift+8*j]));
1163d4002b98SHong Zhang #endif
1164d4002b98SHong Zhang       }
1165d4002b98SHong Zhang     }
11669566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
116768aafef3SStefano Zampini   } else if (format == PETSC_VIEWER_NATIVE) {
116868aafef3SStefano Zampini     for (i=0; i<a->totalslices; i++) { /* loop over slices */
116968aafef3SStefano Zampini       PetscInt row;
11709566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"slice %" PetscInt_FMT ": %" PetscInt_FMT " %" PetscInt_FMT "\n",i,a->sliidx[i],a->sliidx[i+1]));
117168aafef3SStefano Zampini       for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
117268aafef3SStefano Zampini #if defined(PETSC_USE_COMPLEX)
117368aafef3SStefano Zampini         if (PetscImaginaryPart(a->val[j]) > 0.0) {
11749566063dSJacob 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])));
117568aafef3SStefano Zampini         } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
11769566063dSJacob 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])));
117768aafef3SStefano Zampini         } else {
11789566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j])));
117968aafef3SStefano Zampini         }
118068aafef3SStefano Zampini #else
11819566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)a->val[j]));
118268aafef3SStefano Zampini #endif
118368aafef3SStefano Zampini       }
118468aafef3SStefano Zampini     }
1185d4002b98SHong Zhang   } else {
11869566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
1187d4002b98SHong Zhang     if (A->factortype) {
1188d4002b98SHong Zhang       for (i=0; i<m; i++) {
1189d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07);
11909566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i));
1191d4002b98SHong Zhang         /* L part */
1192d4002b98SHong Zhang         for (j=shift; j<a->diag[i]; j+=8) {
1193d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1194d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0) {
11959566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j])));
1196d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0) {
11979566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j]))));
1198d4002b98SHong Zhang           } else {
11999566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j])));
1200d4002b98SHong Zhang           }
1201d4002b98SHong Zhang #else
12029566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]));
1203d4002b98SHong Zhang #endif
1204d4002b98SHong Zhang         }
1205d4002b98SHong Zhang         /* diagonal */
1206d4002b98SHong Zhang         j = a->diag[i];
1207d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1208d4002b98SHong Zhang         if (PetscImaginaryPart(a->val[j]) > 0.0) {
12099566063dSJacob 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])));
1210d4002b98SHong Zhang         } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
12119566063dSJacob 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]))));
1212d4002b98SHong Zhang         } else {
12139566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(1.0/a->val[j])));
1214d4002b98SHong Zhang         }
1215d4002b98SHong Zhang #else
12169566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)(1.0/a->val[j])));
1217d4002b98SHong Zhang #endif
1218d4002b98SHong Zhang 
1219d4002b98SHong Zhang         /* U part */
1220d4002b98SHong Zhang         for (j=a->diag[i]+1; j<shift+8*a->rlen[i]; j+=8) {
1221d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1222d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[j]) > 0.0) {
12239566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j])));
1224d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
12259566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j]))));
1226d4002b98SHong Zhang           } else {
12279566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j])));
1228d4002b98SHong Zhang           }
1229d4002b98SHong Zhang #else
12309566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]));
1231d4002b98SHong Zhang #endif
1232d4002b98SHong Zhang         }
12339566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
1234d4002b98SHong Zhang       }
1235d4002b98SHong Zhang     } else {
1236d4002b98SHong Zhang       for (i=0; i<m; i++) {
1237d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07);
12389566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i));
1239d4002b98SHong Zhang         for (j=0; j<a->rlen[i]; j++) {
1240d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1241d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[j]) > 0.0) {
12429566063dSJacob 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])));
1243d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
12449566063dSJacob 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])));
1245d4002b98SHong Zhang           } else {
12469566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j])));
1247d4002b98SHong Zhang           }
1248d4002b98SHong Zhang #else
12499566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]));
1250d4002b98SHong Zhang #endif
1251d4002b98SHong Zhang         }
12529566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
1253d4002b98SHong Zhang       }
1254d4002b98SHong Zhang     }
12559566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
1256d4002b98SHong Zhang   }
12579566063dSJacob Faibussowitsch   PetscCall(PetscViewerFlush(viewer));
1258d4002b98SHong Zhang   PetscFunctionReturn(0);
1259d4002b98SHong Zhang }
1260d4002b98SHong Zhang 
1261d4002b98SHong Zhang #include <petscdraw.h>
1262d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw_Zoom(PetscDraw draw,void *Aa)
1263d4002b98SHong Zhang {
1264d4002b98SHong Zhang   Mat               A=(Mat)Aa;
1265d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1266d4002b98SHong Zhang   PetscInt          i,j,m=A->rmap->n,shift;
1267d4002b98SHong Zhang   int               color;
1268d4002b98SHong Zhang   PetscReal         xl,yl,xr,yr,x_l,x_r,y_l,y_r;
1269d4002b98SHong Zhang   PetscViewer       viewer;
1270d4002b98SHong Zhang   PetscViewerFormat format;
1271d4002b98SHong Zhang 
1272d4002b98SHong Zhang   PetscFunctionBegin;
12739566063dSJacob Faibussowitsch   PetscCall(PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer));
12749566063dSJacob Faibussowitsch   PetscCall(PetscViewerGetFormat(viewer,&format));
12759566063dSJacob Faibussowitsch   PetscCall(PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr));
1276d4002b98SHong Zhang 
1277d4002b98SHong Zhang   /* loop over matrix elements drawing boxes */
1278d4002b98SHong Zhang 
1279d4002b98SHong Zhang   if (format != PETSC_VIEWER_DRAW_CONTOUR) {
1280d0609cedSBarry Smith     PetscDrawCollectiveBegin(draw);
1281d4002b98SHong Zhang     /* Blue for negative, Cyan for zero and  Red for positive */
1282d4002b98SHong Zhang     color = PETSC_DRAW_BLUE;
1283d4002b98SHong Zhang     for (i=0; i<m; i++) {
1284d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1285d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1286d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1287d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1288d4002b98SHong Zhang         if (PetscRealPart(a->val[shift+8*j]) >=  0.) continue;
12899566063dSJacob Faibussowitsch         PetscCall(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color));
1290d4002b98SHong Zhang       }
1291d4002b98SHong Zhang     }
1292d4002b98SHong Zhang     color = PETSC_DRAW_CYAN;
1293d4002b98SHong Zhang     for (i=0; i<m; i++) {
1294d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1295d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1296d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1297d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1298d4002b98SHong Zhang         if (a->val[shift+8*j] !=  0.) continue;
12999566063dSJacob Faibussowitsch         PetscCall(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color));
1300d4002b98SHong Zhang       }
1301d4002b98SHong Zhang     }
1302d4002b98SHong Zhang     color = PETSC_DRAW_RED;
1303d4002b98SHong Zhang     for (i=0; i<m; i++) {
1304d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1305d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1306d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1307d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1308d4002b98SHong Zhang         if (PetscRealPart(a->val[shift+8*j]) <=  0.) continue;
13099566063dSJacob Faibussowitsch         PetscCall(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color));
1310d4002b98SHong Zhang       }
1311d4002b98SHong Zhang     }
1312d0609cedSBarry Smith     PetscDrawCollectiveEnd(draw);
1313d4002b98SHong Zhang   } else {
1314d4002b98SHong Zhang     /* use contour shading to indicate magnitude of values */
1315d4002b98SHong Zhang     /* first determine max of all nonzero values */
1316d4002b98SHong Zhang     PetscReal minv=0.0,maxv=0.0;
1317d4002b98SHong Zhang     PetscInt  count=0;
1318d4002b98SHong Zhang     PetscDraw popup;
1319d4002b98SHong Zhang     for (i=0; i<a->sliidx[a->totalslices]; i++) {
1320d4002b98SHong Zhang       if (PetscAbsScalar(a->val[i]) > maxv) maxv = PetscAbsScalar(a->val[i]);
1321d4002b98SHong Zhang     }
1322d4002b98SHong Zhang     if (minv >= maxv) maxv = minv + PETSC_SMALL;
13239566063dSJacob Faibussowitsch     PetscCall(PetscDrawGetPopup(draw,&popup));
13249566063dSJacob Faibussowitsch     PetscCall(PetscDrawScalePopup(popup,minv,maxv));
1325d4002b98SHong Zhang 
1326d0609cedSBarry Smith     PetscDrawCollectiveBegin(draw);
1327d4002b98SHong Zhang     for (i=0; i<m; i++) {
1328d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1329d4002b98SHong Zhang       y_l = m - i - 1.0;
1330d4002b98SHong Zhang       y_r = y_l + 1.0;
1331d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1332d4002b98SHong Zhang         x_l = a->colidx[shift+j*8];
1333d4002b98SHong Zhang         x_r = x_l + 1.0;
1334d4002b98SHong Zhang         color = PetscDrawRealToColor(PetscAbsScalar(a->val[count]),minv,maxv);
13359566063dSJacob Faibussowitsch         PetscCall(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color));
1336d4002b98SHong Zhang         count++;
1337d4002b98SHong Zhang       }
1338d4002b98SHong Zhang     }
1339d0609cedSBarry Smith     PetscDrawCollectiveEnd(draw);
1340d4002b98SHong Zhang   }
1341d4002b98SHong Zhang   PetscFunctionReturn(0);
1342d4002b98SHong Zhang }
1343d4002b98SHong Zhang 
1344d4002b98SHong Zhang #include <petscdraw.h>
1345d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw(Mat A,PetscViewer viewer)
1346d4002b98SHong Zhang {
1347d4002b98SHong Zhang   PetscDraw      draw;
1348d4002b98SHong Zhang   PetscReal      xr,yr,xl,yl,h,w;
1349d4002b98SHong Zhang   PetscBool      isnull;
1350d4002b98SHong Zhang 
1351d4002b98SHong Zhang   PetscFunctionBegin;
13529566063dSJacob Faibussowitsch   PetscCall(PetscViewerDrawGetDraw(viewer,0,&draw));
13539566063dSJacob Faibussowitsch   PetscCall(PetscDrawIsNull(draw,&isnull));
1354d4002b98SHong Zhang   if (isnull) PetscFunctionReturn(0);
1355d4002b98SHong Zhang 
1356d4002b98SHong Zhang   xr   = A->cmap->n; yr  = A->rmap->n; h = yr/10.0; w = xr/10.0;
1357d4002b98SHong Zhang   xr  += w;          yr += h;         xl = -w;     yl = -h;
13589566063dSJacob Faibussowitsch   PetscCall(PetscDrawSetCoordinates(draw,xl,yl,xr,yr));
13599566063dSJacob Faibussowitsch   PetscCall(PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer));
13609566063dSJacob Faibussowitsch   PetscCall(PetscDrawZoom(draw,MatView_SeqSELL_Draw_Zoom,A));
13619566063dSJacob Faibussowitsch   PetscCall(PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL));
13629566063dSJacob Faibussowitsch   PetscCall(PetscDrawSave(draw));
1363d4002b98SHong Zhang   PetscFunctionReturn(0);
1364d4002b98SHong Zhang }
1365d4002b98SHong Zhang 
1366d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL(Mat A,PetscViewer viewer)
1367d4002b98SHong Zhang {
1368d4002b98SHong Zhang   PetscBool      iascii,isbinary,isdraw;
1369d4002b98SHong Zhang 
1370d4002b98SHong Zhang   PetscFunctionBegin;
13719566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii));
13729566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary));
13739566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw));
1374d4002b98SHong Zhang   if (iascii) {
13759566063dSJacob Faibussowitsch     PetscCall(MatView_SeqSELL_ASCII(A,viewer));
1376d4002b98SHong Zhang   } else if (isbinary) {
13779566063dSJacob Faibussowitsch     /* PetscCall(MatView_SeqSELL_Binary(A,viewer)); */
13781baa6e33SBarry Smith   } else if (isdraw) PetscCall(MatView_SeqSELL_Draw(A,viewer));
1379d4002b98SHong Zhang   PetscFunctionReturn(0);
1380d4002b98SHong Zhang }
1381d4002b98SHong Zhang 
1382d4002b98SHong Zhang PetscErrorCode MatAssemblyEnd_SeqSELL(Mat A,MatAssemblyType mode)
1383d4002b98SHong Zhang {
1384d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1385d4002b98SHong Zhang   PetscInt       i,shift,row_in_slice,row,nrow,*cp,lastcol,j,k;
1386d4002b98SHong Zhang   MatScalar      *vp;
1387d4002b98SHong Zhang 
1388d4002b98SHong Zhang   PetscFunctionBegin;
1389d4002b98SHong Zhang   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
1390d4002b98SHong Zhang   /* To do: compress out the unused elements */
13919566063dSJacob Faibussowitsch   PetscCall(MatMarkDiagonal_SeqSELL(A));
13929566063dSJacob 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));
13939566063dSJacob Faibussowitsch   PetscCall(PetscInfo(A,"Number of mallocs during MatSetValues() is %" PetscInt_FMT "\n",a->reallocs));
13949566063dSJacob Faibussowitsch   PetscCall(PetscInfo(A,"Maximum nonzeros in any row is %" PetscInt_FMT "\n",a->rlenmax));
1395d4002b98SHong 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 */
1396d4002b98SHong Zhang   for (i=0; i<a->totalslices; ++i) {
1397d4002b98SHong Zhang     shift = a->sliidx[i];    /* starting index of the slice */
1398d4002b98SHong Zhang     cp    = a->colidx+shift; /* pointer to the column indices of the slice */
1399d4002b98SHong Zhang     vp    = a->val+shift;    /* pointer to the nonzero values of the slice */
1400d4002b98SHong Zhang     for (row_in_slice=0; row_in_slice<8; ++row_in_slice) { /* loop over rows in the slice */
1401d4002b98SHong Zhang       row  = 8*i + row_in_slice;
1402d4002b98SHong Zhang       nrow = a->rlen[row]; /* number of nonzeros in row */
1403d4002b98SHong Zhang       /*
1404d4002b98SHong Zhang         Search for the nearest nonzero. Normally setting the index to zero may cause extra communication.
1405d4002b98SHong Zhang         But if the entire slice are empty, it is fine to use 0 since the index will not be loaded.
1406d4002b98SHong Zhang       */
1407d4002b98SHong Zhang       lastcol = 0;
1408d4002b98SHong Zhang       if (nrow>0) { /* nonempty row */
1409d4002b98SHong Zhang         lastcol = cp[8*(nrow-1)+row_in_slice]; /* use the index from the last nonzero at current row */
1410d4002b98SHong Zhang       } else if (!row_in_slice) { /* first row of the currect slice is empty */
1411d4002b98SHong Zhang         for (j=1;j<8;j++) {
1412d4002b98SHong Zhang           if (a->rlen[8*i+j]) {
1413d4002b98SHong Zhang             lastcol = cp[j];
1414d4002b98SHong Zhang             break;
1415d4002b98SHong Zhang           }
1416d4002b98SHong Zhang         }
1417d4002b98SHong Zhang       } else {
1418d4002b98SHong Zhang         if (a->sliidx[i+1] != shift) lastcol = cp[row_in_slice-1]; /* use the index from the previous row */
1419d4002b98SHong Zhang       }
1420d4002b98SHong Zhang 
1421d4002b98SHong Zhang       for (k=nrow; k<(a->sliidx[i+1]-shift)/8; ++k) {
1422d4002b98SHong Zhang         cp[8*k+row_in_slice] = lastcol;
1423d4002b98SHong Zhang         vp[8*k+row_in_slice] = (MatScalar)0;
1424d4002b98SHong Zhang       }
1425d4002b98SHong Zhang     }
1426d4002b98SHong Zhang   }
1427d4002b98SHong Zhang 
1428d4002b98SHong Zhang   A->info.mallocs += a->reallocs;
1429d4002b98SHong Zhang   a->reallocs      = 0;
1430d4002b98SHong Zhang 
14319566063dSJacob Faibussowitsch   PetscCall(MatSeqSELLInvalidateDiagonal(A));
1432d4002b98SHong Zhang   PetscFunctionReturn(0);
1433d4002b98SHong Zhang }
1434d4002b98SHong Zhang 
1435d4002b98SHong Zhang PetscErrorCode MatGetInfo_SeqSELL(Mat A,MatInfoType flag,MatInfo *info)
1436d4002b98SHong Zhang {
1437d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1438d4002b98SHong Zhang 
1439d4002b98SHong Zhang   PetscFunctionBegin;
1440d4002b98SHong Zhang   info->block_size   = 1.0;
14413966268fSBarry Smith   info->nz_allocated = a->maxallocmat;
14423966268fSBarry Smith   info->nz_used      = a->sliidx[a->totalslices]; /* include padding zeros */
14433966268fSBarry Smith   info->nz_unneeded  = (a->maxallocmat-a->sliidx[a->totalslices]);
14443966268fSBarry Smith   info->assemblies   = A->num_ass;
14453966268fSBarry Smith   info->mallocs      = A->info.mallocs;
1446d4002b98SHong Zhang   info->memory       = ((PetscObject)A)->mem;
1447d4002b98SHong Zhang   if (A->factortype) {
1448d4002b98SHong Zhang     info->fill_ratio_given  = A->info.fill_ratio_given;
1449d4002b98SHong Zhang     info->fill_ratio_needed = A->info.fill_ratio_needed;
1450d4002b98SHong Zhang     info->factor_mallocs    = A->info.factor_mallocs;
1451d4002b98SHong Zhang   } else {
1452d4002b98SHong Zhang     info->fill_ratio_given  = 0;
1453d4002b98SHong Zhang     info->fill_ratio_needed = 0;
1454d4002b98SHong Zhang     info->factor_mallocs    = 0;
1455d4002b98SHong Zhang   }
1456d4002b98SHong Zhang   PetscFunctionReturn(0);
1457d4002b98SHong Zhang }
1458d4002b98SHong Zhang 
1459d4002b98SHong Zhang PetscErrorCode MatSetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
1460d4002b98SHong Zhang {
1461d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1462d4002b98SHong Zhang   PetscInt       shift,i,k,l,low,high,t,ii,row,col,nrow;
1463d4002b98SHong Zhang   PetscInt       *cp,nonew=a->nonew,lastcol=-1;
1464d4002b98SHong Zhang   MatScalar      *vp,value;
1465d4002b98SHong Zhang 
1466d4002b98SHong Zhang   PetscFunctionBegin;
1467d4002b98SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
1468d4002b98SHong Zhang     row = im[k];
1469d4002b98SHong Zhang     if (row < 0) continue;
14706bdcaf15SBarry 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);
1471d4002b98SHong Zhang     shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */
1472d4002b98SHong Zhang     cp    = a->colidx+shift; /* pointer to the row */
1473d4002b98SHong Zhang     vp    = a->val+shift; /* pointer to the row */
1474d4002b98SHong Zhang     nrow  = a->rlen[row];
1475d4002b98SHong Zhang     low   = 0;
1476d4002b98SHong Zhang     high  = nrow;
1477d4002b98SHong Zhang 
1478d4002b98SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
1479d4002b98SHong Zhang       col = in[l];
1480d4002b98SHong Zhang       if (col<0) continue;
14816bdcaf15SBarry 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);
1482d4002b98SHong Zhang       if (a->roworiented) {
1483d4002b98SHong Zhang         value = v[l+k*n];
1484d4002b98SHong Zhang       } else {
1485d4002b98SHong Zhang         value = v[k+l*m];
1486d4002b98SHong Zhang       }
1487d4002b98SHong Zhang       if ((value == 0.0 && a->ignorezeroentries) && (is == ADD_VALUES)) continue;
1488d4002b98SHong Zhang 
1489ed73aabaSBarry Smith       /* search in this row for the specified column, i indicates the column to be set */
1490d4002b98SHong Zhang       if (col <= lastcol) low = 0;
1491d4002b98SHong Zhang       else high = nrow;
1492d4002b98SHong Zhang       lastcol = col;
1493d4002b98SHong Zhang       while (high-low > 5) {
1494d4002b98SHong Zhang         t = (low+high)/2;
1495d4002b98SHong Zhang         if (*(cp+t*8) > col) high = t;
1496d4002b98SHong Zhang         else low = t;
1497d4002b98SHong Zhang       }
1498d4002b98SHong Zhang       for (i=low; i<high; i++) {
1499d4002b98SHong Zhang         if (*(cp+i*8) > col) break;
1500d4002b98SHong Zhang         if (*(cp+i*8) == col) {
1501d4002b98SHong Zhang           if (is == ADD_VALUES) *(vp+i*8) += value;
1502d4002b98SHong Zhang           else *(vp+i*8) = value;
1503d4002b98SHong Zhang           low = i + 1;
1504d4002b98SHong Zhang           goto noinsert;
1505d4002b98SHong Zhang         }
1506d4002b98SHong Zhang       }
1507d4002b98SHong Zhang       if (value == 0.0 && a->ignorezeroentries) goto noinsert;
1508d4002b98SHong Zhang       if (nonew == 1) goto noinsert;
150908401ef6SPierre Jolivet       PetscCheck(nonew != -1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%" PetscInt_FMT ", %" PetscInt_FMT ") in the matrix", row, col);
1510d4002b98SHong Zhang       /* If the current row length exceeds the slice width (e.g. nrow==slice_width), allocate a new space, otherwise do nothing */
1511d4002b98SHong Zhang       MatSeqXSELLReallocateSELL(A,A->rmap->n,1,nrow,a->sliidx,row/8,row,col,a->colidx,a->val,cp,vp,nonew,MatScalar);
1512d4002b98SHong Zhang       /* add the new nonzero to the high position, shift the remaining elements in current row to the right by one slot */
1513d4002b98SHong Zhang       for (ii=nrow-1; ii>=i; ii--) {
1514d4002b98SHong Zhang         *(cp+(ii+1)*8) = *(cp+ii*8);
1515d4002b98SHong Zhang         *(vp+(ii+1)*8) = *(vp+ii*8);
1516d4002b98SHong Zhang       }
1517d4002b98SHong Zhang       a->rlen[row]++;
1518d4002b98SHong Zhang       *(cp+i*8) = col;
1519d4002b98SHong Zhang       *(vp+i*8) = value;
1520d4002b98SHong Zhang       a->nz++;
1521d4002b98SHong Zhang       A->nonzerostate++;
1522d4002b98SHong Zhang       low = i+1; high++; nrow++;
1523d4002b98SHong Zhang noinsert:;
1524d4002b98SHong Zhang     }
1525d4002b98SHong Zhang     a->rlen[row] = nrow;
1526d4002b98SHong Zhang   }
1527d4002b98SHong Zhang   PetscFunctionReturn(0);
1528d4002b98SHong Zhang }
1529d4002b98SHong Zhang 
1530d4002b98SHong Zhang PetscErrorCode MatCopy_SeqSELL(Mat A,Mat B,MatStructure str)
1531d4002b98SHong Zhang {
1532d4002b98SHong Zhang   PetscFunctionBegin;
1533d4002b98SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
1534d4002b98SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
1535d4002b98SHong Zhang     Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1536d4002b98SHong Zhang     Mat_SeqSELL *b=(Mat_SeqSELL*)B->data;
1537d4002b98SHong Zhang 
153808401ef6SPierre 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");
15399566063dSJacob Faibussowitsch     PetscCall(PetscArraycpy(b->val,a->val,a->sliidx[a->totalslices]));
1540d4002b98SHong Zhang   } else {
15419566063dSJacob Faibussowitsch     PetscCall(MatCopy_Basic(A,B,str));
1542d4002b98SHong Zhang   }
1543d4002b98SHong Zhang   PetscFunctionReturn(0);
1544d4002b98SHong Zhang }
1545d4002b98SHong Zhang 
1546d4002b98SHong Zhang PetscErrorCode MatSetUp_SeqSELL(Mat A)
1547d4002b98SHong Zhang {
1548d4002b98SHong Zhang   PetscFunctionBegin;
15499566063dSJacob Faibussowitsch   PetscCall(MatSeqSELLSetPreallocation(A,PETSC_DEFAULT,NULL));
1550d4002b98SHong Zhang   PetscFunctionReturn(0);
1551d4002b98SHong Zhang }
1552d4002b98SHong Zhang 
1553d4002b98SHong Zhang PetscErrorCode MatSeqSELLGetArray_SeqSELL(Mat A,PetscScalar *array[])
1554d4002b98SHong Zhang {
1555d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1556d4002b98SHong Zhang 
1557d4002b98SHong Zhang   PetscFunctionBegin;
1558d4002b98SHong Zhang   *array = a->val;
1559d4002b98SHong Zhang   PetscFunctionReturn(0);
1560d4002b98SHong Zhang }
1561d4002b98SHong Zhang 
1562d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray_SeqSELL(Mat A,PetscScalar *array[])
1563d4002b98SHong Zhang {
1564d4002b98SHong Zhang   PetscFunctionBegin;
1565d4002b98SHong Zhang   PetscFunctionReturn(0);
1566d4002b98SHong Zhang }
1567d4002b98SHong Zhang 
1568d4002b98SHong Zhang PetscErrorCode MatRealPart_SeqSELL(Mat A)
1569d4002b98SHong Zhang {
1570d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1571d4002b98SHong Zhang   PetscInt    i;
1572d4002b98SHong Zhang   MatScalar   *aval=a->val;
1573d4002b98SHong Zhang 
1574d4002b98SHong Zhang   PetscFunctionBegin;
1575d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i]=PetscRealPart(aval[i]);
1576d4002b98SHong Zhang   PetscFunctionReturn(0);
1577d4002b98SHong Zhang }
1578d4002b98SHong Zhang 
1579d4002b98SHong Zhang PetscErrorCode MatImaginaryPart_SeqSELL(Mat A)
1580d4002b98SHong Zhang {
1581d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1582d4002b98SHong Zhang   PetscInt       i;
1583d4002b98SHong Zhang   MatScalar      *aval=a->val;
1584d4002b98SHong Zhang 
1585d4002b98SHong Zhang   PetscFunctionBegin;
1586d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i] = PetscImaginaryPart(aval[i]);
15879566063dSJacob Faibussowitsch   PetscCall(MatSeqSELLInvalidateDiagonal(A));
1588d4002b98SHong Zhang   PetscFunctionReturn(0);
1589d4002b98SHong Zhang }
1590d4002b98SHong Zhang 
1591d4002b98SHong Zhang PetscErrorCode MatScale_SeqSELL(Mat inA,PetscScalar alpha)
1592d4002b98SHong Zhang {
1593d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)inA->data;
1594d4002b98SHong Zhang   MatScalar      *aval=a->val;
1595d4002b98SHong Zhang   PetscScalar    oalpha=alpha;
1596d4002b98SHong Zhang   PetscBLASInt   one=1,size;
1597d4002b98SHong Zhang 
1598d4002b98SHong Zhang   PetscFunctionBegin;
15999566063dSJacob Faibussowitsch   PetscCall(PetscBLASIntCast(a->sliidx[a->totalslices],&size));
1600d4002b98SHong Zhang   PetscStackCallBLAS("BLASscal",BLASscal_(&size,&oalpha,aval,&one));
16019566063dSJacob Faibussowitsch   PetscCall(PetscLogFlops(a->nz));
16029566063dSJacob Faibussowitsch   PetscCall(MatSeqSELLInvalidateDiagonal(inA));
1603d4002b98SHong Zhang   PetscFunctionReturn(0);
1604d4002b98SHong Zhang }
1605d4002b98SHong Zhang 
1606d4002b98SHong Zhang PetscErrorCode MatShift_SeqSELL(Mat Y,PetscScalar a)
1607d4002b98SHong Zhang {
1608d4002b98SHong Zhang   Mat_SeqSELL    *y=(Mat_SeqSELL*)Y->data;
1609d4002b98SHong Zhang 
1610d4002b98SHong Zhang   PetscFunctionBegin;
1611d4002b98SHong Zhang   if (!Y->preallocated || !y->nz) {
16129566063dSJacob Faibussowitsch     PetscCall(MatSeqSELLSetPreallocation(Y,1,NULL));
1613d4002b98SHong Zhang   }
16149566063dSJacob Faibussowitsch   PetscCall(MatShift_Basic(Y,a));
1615d4002b98SHong Zhang   PetscFunctionReturn(0);
1616d4002b98SHong Zhang }
1617d4002b98SHong Zhang 
1618d4002b98SHong Zhang PetscErrorCode MatSOR_SeqSELL(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
1619d4002b98SHong Zhang {
1620d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1621d4002b98SHong Zhang   PetscScalar       *x,sum,*t;
1622f4259b30SLisandro Dalcin   const MatScalar   *idiag=NULL,*mdiag;
1623d4002b98SHong Zhang   const PetscScalar *b,*xb;
1624d4002b98SHong Zhang   PetscInt          n,m=A->rmap->n,i,j,shift;
1625d4002b98SHong Zhang   const PetscInt    *diag;
1626d4002b98SHong Zhang 
1627d4002b98SHong Zhang   PetscFunctionBegin;
1628d4002b98SHong Zhang   its = its*lits;
1629d4002b98SHong Zhang 
1630d4002b98SHong Zhang   if (fshift != a->fshift || omega != a->omega) a->idiagvalid = PETSC_FALSE; /* must recompute idiag[] */
16319566063dSJacob Faibussowitsch   if (!a->idiagvalid) PetscCall(MatInvertDiagonal_SeqSELL(A,omega,fshift));
1632d4002b98SHong Zhang   a->fshift = fshift;
1633d4002b98SHong Zhang   a->omega  = omega;
1634d4002b98SHong Zhang 
1635d4002b98SHong Zhang   diag  = a->diag;
1636d4002b98SHong Zhang   t     = a->ssor_work;
1637d4002b98SHong Zhang   idiag = a->idiag;
1638d4002b98SHong Zhang   mdiag = a->mdiag;
1639d4002b98SHong Zhang 
16409566063dSJacob Faibussowitsch   PetscCall(VecGetArray(xx,&x));
16419566063dSJacob Faibussowitsch   PetscCall(VecGetArrayRead(bb,&b));
1642d4002b98SHong Zhang   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
164308401ef6SPierre Jolivet   PetscCheck(flag != SOR_APPLY_UPPER,PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_UPPER is not implemented");
164408401ef6SPierre Jolivet   PetscCheck(flag != SOR_APPLY_LOWER,PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_LOWER is not implemented");
1645aed4548fSBarry Smith   PetscCheck(!(flag & SOR_EISENSTAT),PETSC_COMM_SELF,PETSC_ERR_SUP,"No support yet for Eisenstat");
1646d4002b98SHong Zhang 
1647d4002b98SHong Zhang   if (flag & SOR_ZERO_INITIAL_GUESS) {
1648d4002b98SHong Zhang     if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) {
1649d4002b98SHong Zhang       for (i=0; i<m; i++) {
1650d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1651d4002b98SHong Zhang         sum   = b[i];
1652d4002b98SHong Zhang         n     = (diag[i]-shift)/8;
1653d4002b98SHong Zhang         for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1654d4002b98SHong Zhang         t[i]  = sum;
1655d4002b98SHong Zhang         x[i]  = sum*idiag[i];
1656d4002b98SHong Zhang       }
1657d4002b98SHong Zhang       xb   = t;
16589566063dSJacob Faibussowitsch       PetscCall(PetscLogFlops(a->nz));
1659d4002b98SHong Zhang     } else xb = b;
1660d4002b98SHong Zhang     if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) {
1661d4002b98SHong Zhang       for (i=m-1; i>=0; i--) {
1662d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1663d4002b98SHong Zhang         sum   = xb[i];
1664d4002b98SHong Zhang         n     = a->rlen[i]-(diag[i]-shift)/8-1;
1665d4002b98SHong Zhang         for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1666d4002b98SHong Zhang         if (xb == b) {
1667d4002b98SHong Zhang           x[i] = sum*idiag[i];
1668d4002b98SHong Zhang         } else {
1669d4002b98SHong Zhang           x[i] = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1670d4002b98SHong Zhang         }
1671d4002b98SHong Zhang       }
16729566063dSJacob Faibussowitsch       PetscCall(PetscLogFlops(a->nz)); /* assumes 1/2 in upper */
1673d4002b98SHong Zhang     }
1674d4002b98SHong Zhang     its--;
1675d4002b98SHong Zhang   }
1676d4002b98SHong Zhang   while (its--) {
1677d4002b98SHong Zhang     if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) {
1678d4002b98SHong Zhang       for (i=0; i<m; i++) {
1679d4002b98SHong Zhang         /* lower */
1680d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1681d4002b98SHong Zhang         sum   = b[i];
1682d4002b98SHong Zhang         n     = (diag[i]-shift)/8;
1683d4002b98SHong Zhang         for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1684d4002b98SHong Zhang         t[i]  = sum;             /* save application of the lower-triangular part */
1685d4002b98SHong Zhang         /* upper */
1686d4002b98SHong Zhang         n     = a->rlen[i]-(diag[i]-shift)/8-1;
1687d4002b98SHong Zhang         for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1688d4002b98SHong Zhang         x[i]  = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1689d4002b98SHong Zhang       }
1690d4002b98SHong Zhang       xb   = t;
16919566063dSJacob Faibussowitsch       PetscCall(PetscLogFlops(2.0*a->nz));
1692d4002b98SHong Zhang     } else xb = b;
1693d4002b98SHong Zhang     if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) {
1694d4002b98SHong Zhang       for (i=m-1; i>=0; i--) {
1695d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1696d4002b98SHong Zhang         sum = xb[i];
1697d4002b98SHong Zhang         if (xb == b) {
1698d4002b98SHong Zhang           /* whole matrix (no checkpointing available) */
1699d4002b98SHong Zhang           n     = a->rlen[i];
1700d4002b98SHong Zhang           for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1701d4002b98SHong Zhang           x[i] = (1.-omega)*x[i]+(sum+mdiag[i]*x[i])*idiag[i];
1702d4002b98SHong Zhang         } else { /* lower-triangular part has been saved, so only apply upper-triangular */
1703d4002b98SHong Zhang           n     = a->rlen[i]-(diag[i]-shift)/8-1;
1704d4002b98SHong Zhang           for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1705d4002b98SHong Zhang           x[i]  = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1706d4002b98SHong Zhang         }
1707d4002b98SHong Zhang       }
1708d4002b98SHong Zhang       if (xb == b) {
17099566063dSJacob Faibussowitsch         PetscCall(PetscLogFlops(2.0*a->nz));
1710d4002b98SHong Zhang       } else {
17119566063dSJacob Faibussowitsch         PetscCall(PetscLogFlops(a->nz)); /* assumes 1/2 in upper */
1712d4002b98SHong Zhang       }
1713d4002b98SHong Zhang     }
1714d4002b98SHong Zhang   }
17159566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(xx,&x));
17169566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayRead(bb,&b));
1717d4002b98SHong Zhang   PetscFunctionReturn(0);
1718d4002b98SHong Zhang }
1719d4002b98SHong Zhang 
1720d4002b98SHong Zhang /* -------------------------------------------------------------------*/
1721d4002b98SHong Zhang static struct _MatOps MatOps_Values = {MatSetValues_SeqSELL,
17226108893eSStefano Zampini                                        MatGetRow_SeqSELL,
17236108893eSStefano Zampini                                        MatRestoreRow_SeqSELL,
1724d4002b98SHong Zhang                                        MatMult_SeqSELL,
1725d4002b98SHong Zhang                                /* 4*/  MatMultAdd_SeqSELL,
1726d4002b98SHong Zhang                                        MatMultTranspose_SeqSELL,
1727d4002b98SHong Zhang                                        MatMultTransposeAdd_SeqSELL,
1728f4259b30SLisandro Dalcin                                        NULL,
1729f4259b30SLisandro Dalcin                                        NULL,
1730f4259b30SLisandro Dalcin                                        NULL,
1731f4259b30SLisandro Dalcin                                /* 10*/ NULL,
1732f4259b30SLisandro Dalcin                                        NULL,
1733f4259b30SLisandro Dalcin                                        NULL,
1734d4002b98SHong Zhang                                        MatSOR_SeqSELL,
1735f4259b30SLisandro Dalcin                                        NULL,
1736d4002b98SHong Zhang                                /* 15*/ MatGetInfo_SeqSELL,
1737d4002b98SHong Zhang                                        MatEqual_SeqSELL,
1738d4002b98SHong Zhang                                        MatGetDiagonal_SeqSELL,
1739d4002b98SHong Zhang                                        MatDiagonalScale_SeqSELL,
1740f4259b30SLisandro Dalcin                                        NULL,
1741f4259b30SLisandro Dalcin                                /* 20*/ NULL,
1742d4002b98SHong Zhang                                        MatAssemblyEnd_SeqSELL,
1743d4002b98SHong Zhang                                        MatSetOption_SeqSELL,
1744d4002b98SHong Zhang                                        MatZeroEntries_SeqSELL,
1745f4259b30SLisandro Dalcin                                /* 24*/ NULL,
1746f4259b30SLisandro Dalcin                                        NULL,
1747f4259b30SLisandro Dalcin                                        NULL,
1748f4259b30SLisandro Dalcin                                        NULL,
1749f4259b30SLisandro Dalcin                                        NULL,
1750d4002b98SHong Zhang                                /* 29*/ MatSetUp_SeqSELL,
1751f4259b30SLisandro Dalcin                                        NULL,
1752f4259b30SLisandro Dalcin                                        NULL,
1753f4259b30SLisandro Dalcin                                        NULL,
1754f4259b30SLisandro Dalcin                                        NULL,
1755d4002b98SHong Zhang                                /* 34*/ MatDuplicate_SeqSELL,
1756f4259b30SLisandro Dalcin                                        NULL,
1757f4259b30SLisandro Dalcin                                        NULL,
1758f4259b30SLisandro Dalcin                                        NULL,
1759f4259b30SLisandro Dalcin                                        NULL,
1760f4259b30SLisandro Dalcin                                /* 39*/ NULL,
1761f4259b30SLisandro Dalcin                                        NULL,
1762f4259b30SLisandro Dalcin                                        NULL,
1763d4002b98SHong Zhang                                        MatGetValues_SeqSELL,
1764d4002b98SHong Zhang                                        MatCopy_SeqSELL,
1765f4259b30SLisandro Dalcin                                /* 44*/ NULL,
1766d4002b98SHong Zhang                                        MatScale_SeqSELL,
1767d4002b98SHong Zhang                                        MatShift_SeqSELL,
1768f4259b30SLisandro Dalcin                                        NULL,
1769f4259b30SLisandro Dalcin                                        NULL,
1770f4259b30SLisandro Dalcin                                /* 49*/ NULL,
1771f4259b30SLisandro Dalcin                                        NULL,
1772f4259b30SLisandro Dalcin                                        NULL,
1773f4259b30SLisandro Dalcin                                        NULL,
1774f4259b30SLisandro Dalcin                                        NULL,
1775d4002b98SHong Zhang                                /* 54*/ MatFDColoringCreate_SeqXAIJ,
1776f4259b30SLisandro Dalcin                                        NULL,
1777f4259b30SLisandro Dalcin                                        NULL,
1778f4259b30SLisandro Dalcin                                        NULL,
1779f4259b30SLisandro Dalcin                                        NULL,
1780f4259b30SLisandro Dalcin                                /* 59*/ NULL,
1781d4002b98SHong Zhang                                        MatDestroy_SeqSELL,
1782d4002b98SHong Zhang                                        MatView_SeqSELL,
1783f4259b30SLisandro Dalcin                                        NULL,
1784f4259b30SLisandro Dalcin                                        NULL,
1785f4259b30SLisandro Dalcin                                /* 64*/ NULL,
1786f4259b30SLisandro Dalcin                                        NULL,
1787f4259b30SLisandro Dalcin                                        NULL,
1788f4259b30SLisandro Dalcin                                        NULL,
1789f4259b30SLisandro Dalcin                                        NULL,
1790f4259b30SLisandro Dalcin                                /* 69*/ NULL,
1791f4259b30SLisandro Dalcin                                        NULL,
1792f4259b30SLisandro Dalcin                                        NULL,
1793f4259b30SLisandro Dalcin                                        NULL,
1794f4259b30SLisandro Dalcin                                        NULL,
1795f4259b30SLisandro Dalcin                                /* 74*/ NULL,
1796d4002b98SHong Zhang                                        MatFDColoringApply_AIJ, /* reuse the FDColoring function for AIJ */
1797f4259b30SLisandro Dalcin                                        NULL,
1798f4259b30SLisandro Dalcin                                        NULL,
1799f4259b30SLisandro Dalcin                                        NULL,
1800f4259b30SLisandro Dalcin                                /* 79*/ NULL,
1801f4259b30SLisandro Dalcin                                        NULL,
1802f4259b30SLisandro Dalcin                                        NULL,
1803f4259b30SLisandro Dalcin                                        NULL,
1804f4259b30SLisandro Dalcin                                        NULL,
1805f4259b30SLisandro Dalcin                                /* 84*/ NULL,
1806f4259b30SLisandro Dalcin                                        NULL,
1807f4259b30SLisandro Dalcin                                        NULL,
1808f4259b30SLisandro Dalcin                                        NULL,
1809f4259b30SLisandro Dalcin                                        NULL,
1810f4259b30SLisandro Dalcin                                /* 89*/ NULL,
1811f4259b30SLisandro Dalcin                                        NULL,
1812f4259b30SLisandro Dalcin                                        NULL,
1813f4259b30SLisandro Dalcin                                        NULL,
1814f4259b30SLisandro Dalcin                                        NULL,
1815f4259b30SLisandro Dalcin                                /* 94*/ NULL,
1816f4259b30SLisandro Dalcin                                        NULL,
1817f4259b30SLisandro Dalcin                                        NULL,
1818f4259b30SLisandro Dalcin                                        NULL,
1819f4259b30SLisandro Dalcin                                        NULL,
1820f4259b30SLisandro Dalcin                                /* 99*/ NULL,
1821f4259b30SLisandro Dalcin                                        NULL,
1822f4259b30SLisandro Dalcin                                        NULL,
1823d4002b98SHong Zhang                                        MatConjugate_SeqSELL,
1824f4259b30SLisandro Dalcin                                        NULL,
1825f4259b30SLisandro Dalcin                                /*104*/ NULL,
1826f4259b30SLisandro Dalcin                                        NULL,
1827f4259b30SLisandro Dalcin                                        NULL,
1828f4259b30SLisandro Dalcin                                        NULL,
1829f4259b30SLisandro Dalcin                                        NULL,
1830f4259b30SLisandro Dalcin                                /*109*/ NULL,
1831f4259b30SLisandro Dalcin                                        NULL,
1832f4259b30SLisandro Dalcin                                        NULL,
1833f4259b30SLisandro Dalcin                                        NULL,
1834d4002b98SHong Zhang                                        MatMissingDiagonal_SeqSELL,
1835f4259b30SLisandro Dalcin                                /*114*/ NULL,
1836f4259b30SLisandro Dalcin                                        NULL,
1837f4259b30SLisandro Dalcin                                        NULL,
1838f4259b30SLisandro Dalcin                                        NULL,
1839f4259b30SLisandro Dalcin                                        NULL,
1840f4259b30SLisandro Dalcin                                /*119*/ NULL,
1841f4259b30SLisandro Dalcin                                        NULL,
1842f4259b30SLisandro Dalcin                                        NULL,
1843f4259b30SLisandro Dalcin                                        NULL,
1844f4259b30SLisandro Dalcin                                        NULL,
1845f4259b30SLisandro Dalcin                                /*124*/ NULL,
1846f4259b30SLisandro Dalcin                                        NULL,
1847f4259b30SLisandro Dalcin                                        NULL,
1848f4259b30SLisandro Dalcin                                        NULL,
1849f4259b30SLisandro Dalcin                                        NULL,
1850f4259b30SLisandro Dalcin                                /*129*/ NULL,
1851f4259b30SLisandro Dalcin                                        NULL,
1852f4259b30SLisandro Dalcin                                        NULL,
1853f4259b30SLisandro Dalcin                                        NULL,
1854f4259b30SLisandro Dalcin                                        NULL,
1855f4259b30SLisandro Dalcin                                /*134*/ NULL,
1856f4259b30SLisandro Dalcin                                        NULL,
1857f4259b30SLisandro Dalcin                                        NULL,
1858f4259b30SLisandro Dalcin                                        NULL,
1859f4259b30SLisandro Dalcin                                        NULL,
1860f4259b30SLisandro Dalcin                                /*139*/ NULL,
1861f4259b30SLisandro Dalcin                                        NULL,
1862f4259b30SLisandro Dalcin                                        NULL,
1863d4002b98SHong Zhang                                        MatFDColoringSetUp_SeqXAIJ,
1864f4259b30SLisandro Dalcin                                        NULL,
1865d70f29a3SPierre Jolivet                                /*144*/ NULL,
1866d70f29a3SPierre Jolivet                                        NULL,
1867d70f29a3SPierre Jolivet                                        NULL,
1868*99a7f59eSMark Adams                                        NULL,
1869*99a7f59eSMark Adams                                        NULL,
1870d70f29a3SPierre Jolivet                                        NULL
1871d4002b98SHong Zhang };
1872d4002b98SHong Zhang 
1873d4002b98SHong Zhang PetscErrorCode MatStoreValues_SeqSELL(Mat mat)
1874d4002b98SHong Zhang {
1875d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)mat->data;
1876d4002b98SHong Zhang 
1877d4002b98SHong Zhang   PetscFunctionBegin;
187828b400f6SJacob Faibussowitsch   PetscCheck(a->nonew,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1879d4002b98SHong Zhang 
1880d4002b98SHong Zhang   /* allocate space for values if not already there */
1881d4002b98SHong Zhang   if (!a->saved_values) {
18829566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(a->sliidx[a->totalslices]+1,&a->saved_values));
18839566063dSJacob Faibussowitsch     PetscCall(PetscLogObjectMemory((PetscObject)mat,(a->sliidx[a->totalslices]+1)*sizeof(PetscScalar)));
1884d4002b98SHong Zhang   }
1885d4002b98SHong Zhang 
1886d4002b98SHong Zhang   /* copy values over */
18879566063dSJacob Faibussowitsch   PetscCall(PetscArraycpy(a->saved_values,a->val,a->sliidx[a->totalslices]));
1888d4002b98SHong Zhang   PetscFunctionReturn(0);
1889d4002b98SHong Zhang }
1890d4002b98SHong Zhang 
1891d4002b98SHong Zhang PetscErrorCode MatRetrieveValues_SeqSELL(Mat mat)
1892d4002b98SHong Zhang {
1893d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)mat->data;
1894d4002b98SHong Zhang 
1895d4002b98SHong Zhang   PetscFunctionBegin;
189628b400f6SJacob Faibussowitsch   PetscCheck(a->nonew,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
189728b400f6SJacob Faibussowitsch   PetscCheck(a->saved_values,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
18989566063dSJacob Faibussowitsch   PetscCall(PetscArraycpy(a->val,a->saved_values,a->sliidx[a->totalslices]));
1899d4002b98SHong Zhang   PetscFunctionReturn(0);
1900d4002b98SHong Zhang }
1901d4002b98SHong Zhang 
1902d4002b98SHong Zhang /*@C
1903d4002b98SHong Zhang  MatSeqSELLRestoreArray - returns access to the array where the data for a MATSEQSELL matrix is stored obtained by MatSeqSELLGetArray()
1904d4002b98SHong Zhang 
1905d4002b98SHong Zhang  Not Collective
1906d4002b98SHong Zhang 
1907d4002b98SHong Zhang  Input Parameters:
1908d4002b98SHong Zhang  .  mat - a MATSEQSELL matrix
1909d4002b98SHong Zhang  .  array - pointer to the data
1910d4002b98SHong Zhang 
1911d4002b98SHong Zhang  Level: intermediate
1912d4002b98SHong Zhang 
1913db781477SPatrick Sanan  .seealso: `MatSeqSELLGetArray()`, `MatSeqSELLRestoreArrayF90()`
1914d4002b98SHong Zhang  @*/
1915d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray(Mat A,PetscScalar **array)
1916d4002b98SHong Zhang {
1917d4002b98SHong Zhang   PetscFunctionBegin;
1918cac4c232SBarry Smith   PetscUseMethod(A,"MatSeqSELLRestoreArray_C",(Mat,PetscScalar**),(A,array));
1919d4002b98SHong Zhang   PetscFunctionReturn(0);
1920d4002b98SHong Zhang }
1921d4002b98SHong Zhang 
1922d4002b98SHong Zhang PETSC_EXTERN PetscErrorCode MatCreate_SeqSELL(Mat B)
1923d4002b98SHong Zhang {
1924d4002b98SHong Zhang   Mat_SeqSELL    *b;
1925d4002b98SHong Zhang   PetscMPIInt    size;
1926d4002b98SHong Zhang 
1927d4002b98SHong Zhang   PetscFunctionBegin;
19289566063dSJacob Faibussowitsch   PetscCall(PetscCitationsRegister(citation,&cited));
19299566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)B),&size));
193008401ef6SPierre Jolivet   PetscCheck(size <= 1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Comm must be of size 1");
1931d4002b98SHong Zhang 
19329566063dSJacob Faibussowitsch   PetscCall(PetscNewLog(B,&b));
1933d4002b98SHong Zhang 
1934d4002b98SHong Zhang   B->data = (void*)b;
1935d4002b98SHong Zhang 
19369566063dSJacob Faibussowitsch   PetscCall(PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps)));
1937d4002b98SHong Zhang 
1938f4259b30SLisandro Dalcin   b->row                = NULL;
1939f4259b30SLisandro Dalcin   b->col                = NULL;
1940f4259b30SLisandro Dalcin   b->icol               = NULL;
1941d4002b98SHong Zhang   b->reallocs           = 0;
1942d4002b98SHong Zhang   b->ignorezeroentries  = PETSC_FALSE;
1943d4002b98SHong Zhang   b->roworiented        = PETSC_TRUE;
1944d4002b98SHong Zhang   b->nonew              = 0;
1945f4259b30SLisandro Dalcin   b->diag               = NULL;
1946f4259b30SLisandro Dalcin   b->solve_work         = NULL;
1947f4259b30SLisandro Dalcin   B->spptr              = NULL;
1948f4259b30SLisandro Dalcin   b->saved_values       = NULL;
1949f4259b30SLisandro Dalcin   b->idiag              = NULL;
1950f4259b30SLisandro Dalcin   b->mdiag              = NULL;
1951f4259b30SLisandro Dalcin   b->ssor_work          = NULL;
1952d4002b98SHong Zhang   b->omega              = 1.0;
1953d4002b98SHong Zhang   b->fshift             = 0.0;
1954d4002b98SHong Zhang   b->idiagvalid         = PETSC_FALSE;
1955d4002b98SHong Zhang   b->keepnonzeropattern = PETSC_FALSE;
1956d4002b98SHong Zhang 
19579566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)B,MATSEQSELL));
19589566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLGetArray_C",MatSeqSELLGetArray_SeqSELL));
19599566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLRestoreArray_C",MatSeqSELLRestoreArray_SeqSELL));
19609566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSELL));
19619566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSELL));
19629566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLSetPreallocation_C",MatSeqSELLSetPreallocation_SeqSELL));
19639566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsell_seqaij_C",MatConvert_SeqSELL_SeqAIJ));
1964d4002b98SHong Zhang   PetscFunctionReturn(0);
1965d4002b98SHong Zhang }
1966d4002b98SHong Zhang 
1967d4002b98SHong Zhang /*
1968d4002b98SHong Zhang  Given a matrix generated with MatGetFactor() duplicates all the information in A into B
1969d4002b98SHong Zhang  */
1970d4002b98SHong Zhang PetscErrorCode MatDuplicateNoCreate_SeqSELL(Mat C,Mat A,MatDuplicateOption cpvalues,PetscBool mallocmatspace)
1971d4002b98SHong Zhang {
1972ed73aabaSBarry Smith   Mat_SeqSELL    *c = (Mat_SeqSELL*)C->data,*a = (Mat_SeqSELL*)A->data;
1973d4002b98SHong Zhang   PetscInt       i,m=A->rmap->n;
1974d4002b98SHong Zhang   PetscInt       totalslices=a->totalslices;
1975d4002b98SHong Zhang 
1976d4002b98SHong Zhang   PetscFunctionBegin;
1977d4002b98SHong Zhang   C->factortype = A->factortype;
1978f4259b30SLisandro Dalcin   c->row        = NULL;
1979f4259b30SLisandro Dalcin   c->col        = NULL;
1980f4259b30SLisandro Dalcin   c->icol       = NULL;
1981d4002b98SHong Zhang   c->reallocs   = 0;
1982d4002b98SHong Zhang   C->assembled = PETSC_TRUE;
1983d4002b98SHong Zhang 
19849566063dSJacob Faibussowitsch   PetscCall(PetscLayoutReference(A->rmap,&C->rmap));
19859566063dSJacob Faibussowitsch   PetscCall(PetscLayoutReference(A->cmap,&C->cmap));
1986d4002b98SHong Zhang 
19879566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(8*totalslices,&c->rlen));
19889566063dSJacob Faibussowitsch   PetscCall(PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt)));
19899566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(totalslices+1,&c->sliidx));
19909566063dSJacob Faibussowitsch   PetscCall(PetscLogObjectMemory((PetscObject)C, (totalslices+1)*sizeof(PetscInt)));
1991d4002b98SHong Zhang 
1992d4002b98SHong Zhang   for (i=0; i<m; i++) c->rlen[i] = a->rlen[i];
1993d4002b98SHong Zhang   for (i=0; i<totalslices+1; i++) c->sliidx[i] = a->sliidx[i];
1994d4002b98SHong Zhang 
1995d4002b98SHong Zhang   /* allocate the matrix space */
1996d4002b98SHong Zhang   if (mallocmatspace) {
19979566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(a->maxallocmat,&c->val,a->maxallocmat,&c->colidx));
19989566063dSJacob Faibussowitsch     PetscCall(PetscLogObjectMemory((PetscObject)C,a->maxallocmat*(sizeof(PetscScalar)+sizeof(PetscInt))));
1999d4002b98SHong Zhang 
2000d4002b98SHong Zhang     c->singlemalloc = PETSC_TRUE;
2001d4002b98SHong Zhang 
2002d4002b98SHong Zhang     if (m > 0) {
20039566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(c->colidx,a->colidx,a->maxallocmat));
2004d4002b98SHong Zhang       if (cpvalues == MAT_COPY_VALUES) {
20059566063dSJacob Faibussowitsch         PetscCall(PetscArraycpy(c->val,a->val,a->maxallocmat));
2006d4002b98SHong Zhang       } else {
20079566063dSJacob Faibussowitsch         PetscCall(PetscArrayzero(c->val,a->maxallocmat));
2008d4002b98SHong Zhang       }
2009d4002b98SHong Zhang     }
2010d4002b98SHong Zhang   }
2011d4002b98SHong Zhang 
2012d4002b98SHong Zhang   c->ignorezeroentries = a->ignorezeroentries;
2013d4002b98SHong Zhang   c->roworiented       = a->roworiented;
2014d4002b98SHong Zhang   c->nonew             = a->nonew;
2015d4002b98SHong Zhang   if (a->diag) {
20169566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(m,&c->diag));
20179566063dSJacob Faibussowitsch     PetscCall(PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt)));
2018d4002b98SHong Zhang     for (i=0; i<m; i++) {
2019d4002b98SHong Zhang       c->diag[i] = a->diag[i];
2020d4002b98SHong Zhang     }
2021f4259b30SLisandro Dalcin   } else c->diag = NULL;
2022d4002b98SHong Zhang 
2023f4259b30SLisandro Dalcin   c->solve_work         = NULL;
2024f4259b30SLisandro Dalcin   c->saved_values       = NULL;
2025f4259b30SLisandro Dalcin   c->idiag              = NULL;
2026f4259b30SLisandro Dalcin   c->ssor_work          = NULL;
2027d4002b98SHong Zhang   c->keepnonzeropattern = a->keepnonzeropattern;
2028d4002b98SHong Zhang   c->free_val           = PETSC_TRUE;
2029d4002b98SHong Zhang   c->free_colidx        = PETSC_TRUE;
2030d4002b98SHong Zhang 
2031d4002b98SHong Zhang   c->maxallocmat  = a->maxallocmat;
2032d4002b98SHong Zhang   c->maxallocrow  = a->maxallocrow;
2033d4002b98SHong Zhang   c->rlenmax      = a->rlenmax;
2034d4002b98SHong Zhang   c->nz           = a->nz;
2035d4002b98SHong Zhang   C->preallocated = PETSC_TRUE;
2036d4002b98SHong Zhang 
2037d4002b98SHong Zhang   c->nonzerorowcnt = a->nonzerorowcnt;
2038d4002b98SHong Zhang   C->nonzerostate  = A->nonzerostate;
2039d4002b98SHong Zhang 
20409566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist));
2041d4002b98SHong Zhang   PetscFunctionReturn(0);
2042d4002b98SHong Zhang }
2043d4002b98SHong Zhang 
2044d4002b98SHong Zhang PetscErrorCode MatDuplicate_SeqSELL(Mat A,MatDuplicateOption cpvalues,Mat *B)
2045d4002b98SHong Zhang {
2046d4002b98SHong Zhang   PetscFunctionBegin;
20479566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)A),B));
20489566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n));
2049d4002b98SHong Zhang   if (!(A->rmap->n % A->rmap->bs) && !(A->cmap->n % A->cmap->bs)) {
20509566063dSJacob Faibussowitsch     PetscCall(MatSetBlockSizesFromMats(*B,A,A));
2051d4002b98SHong Zhang   }
20529566063dSJacob Faibussowitsch   PetscCall(MatSetType(*B,((PetscObject)A)->type_name));
20539566063dSJacob Faibussowitsch   PetscCall(MatDuplicateNoCreate_SeqSELL(*B,A,cpvalues,PETSC_TRUE));
2054d4002b98SHong Zhang   PetscFunctionReturn(0);
2055d4002b98SHong Zhang }
2056d4002b98SHong Zhang 
2057ed73aabaSBarry Smith /*MC
2058ed73aabaSBarry Smith    MATSEQSELL - MATSEQSELL = "seqsell" - A matrix type to be used for sequential sparse matrices,
2059ed73aabaSBarry Smith    based on the sliced Ellpack format
2060ed73aabaSBarry Smith 
2061ed73aabaSBarry Smith    Options Database Keys:
2062ed73aabaSBarry Smith . -mat_type seqsell - sets the matrix type to "seqsell" during a call to MatSetFromOptions()
2063ed73aabaSBarry Smith 
2064ed73aabaSBarry Smith    Level: beginner
2065ed73aabaSBarry Smith 
2066db781477SPatrick Sanan .seealso: `MatCreateSeqSell()`, `MATSELL`, `MATMPISELL`, `MATSEQAIJ`, `MATAIJ`, `MATMPIAIJ`
2067ed73aabaSBarry Smith M*/
2068ed73aabaSBarry Smith 
2069ed73aabaSBarry Smith /*MC
2070ed73aabaSBarry Smith    MATSELL - MATSELL = "sell" - A matrix type to be used for sparse matrices.
2071ed73aabaSBarry Smith 
2072ed73aabaSBarry Smith    This matrix type is identical to MATSEQSELL when constructed with a single process communicator,
2073ed73aabaSBarry Smith    and MATMPISELL otherwise.  As a result, for single process communicators,
2074ed73aabaSBarry Smith   MatSeqSELLSetPreallocation() is supported, and similarly MatMPISELLSetPreallocation() is supported
2075ed73aabaSBarry Smith   for communicators controlling multiple processes.  It is recommended that you call both of
2076ed73aabaSBarry Smith   the above preallocation routines for simplicity.
2077ed73aabaSBarry Smith 
2078ed73aabaSBarry Smith    Options Database Keys:
2079ed73aabaSBarry Smith . -mat_type sell - sets the matrix type to "sell" during a call to MatSetFromOptions()
2080ed73aabaSBarry Smith 
2081ed73aabaSBarry Smith   Level: beginner
2082ed73aabaSBarry Smith 
2083ed73aabaSBarry Smith   Notes:
2084ed73aabaSBarry Smith    This format is only supported for real scalars, double precision, and 32 bit indices (the defaults).
2085ed73aabaSBarry Smith 
2086ed73aabaSBarry Smith    It can provide better performance on Intel and AMD processes with AVX2 or AVX512 support for matrices that have a similar number of
2087ed73aabaSBarry Smith    non-zeros in contiguous groups of rows. However if the computation is memory bandwidth limited it may not provide much improvement.
2088ed73aabaSBarry Smith 
2089ed73aabaSBarry Smith   Developer Notes:
2090ed73aabaSBarry Smith    On Intel (and AMD) systems some of the matrix operations use SIMD (AVX) instructions to achieve higher performance.
2091ed73aabaSBarry Smith 
2092ed73aabaSBarry Smith    The sparse matrix format is as follows. For simplicity we assume a slice size of 2, it is actually 8
2093ed73aabaSBarry Smith .vb
2094ed73aabaSBarry Smith                             (2 0  3 4)
2095ed73aabaSBarry Smith    Consider the matrix A =  (5 0  6 0)
2096ed73aabaSBarry Smith                             (0 0  7 8)
2097ed73aabaSBarry Smith                             (0 0  9 9)
2098ed73aabaSBarry Smith 
2099ed73aabaSBarry Smith    symbolically the Ellpack format can be written as
2100ed73aabaSBarry Smith 
2101ed73aabaSBarry Smith         (2 3 4 |)           (0 2 3 |)
2102ed73aabaSBarry Smith    v =  (5 6 0 |)  colidx = (0 2 2 |)
2103ed73aabaSBarry Smith         --------            ---------
2104ed73aabaSBarry Smith         (7 8 |)             (2 3 |)
2105ed73aabaSBarry Smith         (9 9 |)             (2 3 |)
2106ed73aabaSBarry Smith 
2107ed73aabaSBarry 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).
2108ed73aabaSBarry 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
2109ed73aabaSBarry 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.
2110ed73aabaSBarry Smith 
2111ed73aabaSBarry 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)
2112ed73aabaSBarry Smith 
2113ed73aabaSBarry Smith .ve
2114ed73aabaSBarry Smith 
2115ed73aabaSBarry Smith       See MatMult_SeqSELL() for how this format is used with the SIMD operations to achieve high performance.
2116ed73aabaSBarry Smith 
2117ed73aabaSBarry Smith  References:
2118606c0280SSatish Balay . * - Hong Zhang, Richard T. Mills, Karl Rupp, and Barry F. Smith, Vectorized Parallel Sparse Matrix-Vector Multiplication in {PETSc} Using {AVX-512},
2119ed73aabaSBarry Smith    Proceedings of the 47th International Conference on Parallel Processing, 2018.
2120ed73aabaSBarry Smith 
2121db781477SPatrick Sanan .seealso: `MatCreateSeqSELL()`, `MatCreateSeqAIJ()`, `MatCreateSell()`, `MATSEQSELL`, `MATMPISELL`, `MATSEQAIJ`, `MATMPIAIJ`, `MATAIJ`
2122ed73aabaSBarry Smith M*/
2123ed73aabaSBarry Smith 
2124d4002b98SHong Zhang /*@C
2125d4002b98SHong Zhang        MatCreateSeqSELL - Creates a sparse matrix in SELL format.
2126d4002b98SHong Zhang 
2127ed73aabaSBarry Smith  Collective on comm
2128d4002b98SHong Zhang 
2129d4002b98SHong Zhang  Input Parameters:
2130d4002b98SHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2131d4002b98SHong Zhang .  m - number of rows
2132d4002b98SHong Zhang .  n - number of columns
2133d4002b98SHong Zhang .  rlenmax - maximum number of nonzeros in a row
2134d4002b98SHong Zhang -  rlen - array containing the number of nonzeros in the various rows
2135d4002b98SHong Zhang  (possibly different for each row) or NULL
2136d4002b98SHong Zhang 
2137d4002b98SHong Zhang  Output Parameter:
2138d4002b98SHong Zhang .  A - the matrix
2139d4002b98SHong Zhang 
2140d4002b98SHong Zhang  It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2141f6f02116SRichard Tran Mills  MatXXXXSetPreallocation() paradigm instead of this routine directly.
2142d4002b98SHong Zhang  [MatXXXXSetPreallocation() is, for example, MatSeqSELLSetPreallocation]
2143d4002b98SHong Zhang 
2144d4002b98SHong Zhang  Notes:
2145d4002b98SHong Zhang  If nnz is given then nz is ignored
2146d4002b98SHong Zhang 
2147d4002b98SHong Zhang  Specify the preallocated storage with either rlenmax or rlen (not both).
2148d4002b98SHong Zhang  Set rlenmax=PETSC_DEFAULT and rlen=NULL for PETSc to control dynamic memory
2149d4002b98SHong Zhang  allocation.  For large problems you MUST preallocate memory or you
2150d4002b98SHong Zhang  will get TERRIBLE performance, see the users' manual chapter on matrices.
2151d4002b98SHong Zhang 
2152d4002b98SHong Zhang  Level: intermediate
2153d4002b98SHong Zhang 
2154db781477SPatrick Sanan  .seealso: `MatCreate()`, `MatCreateSELL()`, `MatSetValues()`, `MatSeqSELLSetPreallocation()`, `MATSELL`, `MATSEQSELL`, `MATMPISELL`
2155d4002b98SHong Zhang 
2156d4002b98SHong Zhang  @*/
2157d4002b98SHong Zhang PetscErrorCode MatCreateSeqSELL(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt maxallocrow,const PetscInt rlen[],Mat *A)
2158d4002b98SHong Zhang {
2159d4002b98SHong Zhang   PetscFunctionBegin;
21609566063dSJacob Faibussowitsch   PetscCall(MatCreate(comm,A));
21619566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(*A,m,n,m,n));
21629566063dSJacob Faibussowitsch   PetscCall(MatSetType(*A,MATSEQSELL));
21639566063dSJacob Faibussowitsch   PetscCall(MatSeqSELLSetPreallocation_SeqSELL(*A,maxallocrow,rlen));
2164d4002b98SHong Zhang   PetscFunctionReturn(0);
2165d4002b98SHong Zhang }
2166d4002b98SHong Zhang 
2167d4002b98SHong Zhang PetscErrorCode MatEqual_SeqSELL(Mat A,Mat B,PetscBool * flg)
2168d4002b98SHong Zhang {
2169d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data,*b=(Mat_SeqSELL*)B->data;
2170d4002b98SHong Zhang   PetscInt       totalslices=a->totalslices;
2171d4002b98SHong Zhang 
2172d4002b98SHong Zhang   PetscFunctionBegin;
2173d4002b98SHong Zhang   /* If the  matrix dimensions are not equal,or no of nonzeros */
2174d4002b98SHong Zhang   if ((A->rmap->n != B->rmap->n) || (A->cmap->n != B->cmap->n) ||(a->nz != b->nz) || (a->rlenmax != b->rlenmax)) {
2175d4002b98SHong Zhang     *flg = PETSC_FALSE;
2176d4002b98SHong Zhang     PetscFunctionReturn(0);
2177d4002b98SHong Zhang   }
2178d4002b98SHong Zhang   /* if the a->colidx are the same */
21799566063dSJacob Faibussowitsch   PetscCall(PetscArraycmp(a->colidx,b->colidx,a->sliidx[totalslices],flg));
2180d4002b98SHong Zhang   if (!*flg) PetscFunctionReturn(0);
2181d4002b98SHong Zhang   /* if a->val are the same */
21829566063dSJacob Faibussowitsch   PetscCall(PetscArraycmp(a->val,b->val,a->sliidx[totalslices],flg));
2183d4002b98SHong Zhang   PetscFunctionReturn(0);
2184d4002b98SHong Zhang }
2185d4002b98SHong Zhang 
2186d4002b98SHong Zhang PetscErrorCode MatSeqSELLInvalidateDiagonal(Mat A)
2187d4002b98SHong Zhang {
2188d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
2189d4002b98SHong Zhang 
2190d4002b98SHong Zhang   PetscFunctionBegin;
2191d4002b98SHong Zhang   a->idiagvalid  = PETSC_FALSE;
2192d4002b98SHong Zhang   PetscFunctionReturn(0);
2193d4002b98SHong Zhang }
2194d4002b98SHong Zhang 
2195d4002b98SHong Zhang PetscErrorCode MatConjugate_SeqSELL(Mat A)
2196d4002b98SHong Zhang {
2197d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
2198d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
2199d4002b98SHong Zhang   PetscInt    i;
2200d4002b98SHong Zhang   PetscScalar *val = a->val;
2201d4002b98SHong Zhang 
2202d4002b98SHong Zhang   PetscFunctionBegin;
2203d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) {
2204d4002b98SHong Zhang     val[i] = PetscConj(val[i]);
2205d4002b98SHong Zhang   }
2206d4002b98SHong Zhang #else
2207d4002b98SHong Zhang   PetscFunctionBegin;
2208d4002b98SHong Zhang #endif
2209d4002b98SHong Zhang   PetscFunctionReturn(0);
2210d4002b98SHong Zhang }
2211