xref: /petsc/src/mat/impls/sell/seq/sell.c (revision 2c71b3e237ead271e4f3aa1505f92bf476e3413d)
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 
82d4002b98SHong Zhang  .seealso: MatCreate(), MatCreateSELL(), MatSetValues(), MatGetInfo()
83d4002b98SHong Zhang 
84d4002b98SHong Zhang  @*/
85d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation(Mat B,PetscInt rlenmax,const PetscInt rlen[])
86d4002b98SHong Zhang {
87d4002b98SHong Zhang   PetscErrorCode ierr;
88d4002b98SHong Zhang 
89d4002b98SHong Zhang   PetscFunctionBegin;
90d4002b98SHong Zhang   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
91d4002b98SHong Zhang   PetscValidType(B,1);
92d4002b98SHong Zhang   ierr = PetscTryMethod(B,"MatSeqSELLSetPreallocation_C",(Mat,PetscInt,const PetscInt[]),(B,rlenmax,rlen));CHKERRQ(ierr);
93d4002b98SHong Zhang   PetscFunctionReturn(0);
94d4002b98SHong Zhang }
95d4002b98SHong Zhang 
96d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation_SeqSELL(Mat B,PetscInt maxallocrow,const PetscInt rlen[])
97d4002b98SHong Zhang {
98d4002b98SHong Zhang   Mat_SeqSELL    *b;
99d4002b98SHong Zhang   PetscInt       i,j,totalslices;
100d4002b98SHong Zhang   PetscBool      skipallocation=PETSC_FALSE,realalloc=PETSC_FALSE;
101d4002b98SHong Zhang   PetscErrorCode ierr;
102d4002b98SHong Zhang 
103d4002b98SHong Zhang   PetscFunctionBegin;
104d4002b98SHong Zhang   if (maxallocrow >= 0 || rlen) realalloc = PETSC_TRUE;
105d4002b98SHong Zhang   if (maxallocrow == MAT_SKIP_ALLOCATION) {
106d4002b98SHong Zhang     skipallocation = PETSC_TRUE;
107d4002b98SHong Zhang     maxallocrow    = 0;
108d4002b98SHong Zhang   }
109d4002b98SHong Zhang 
110d4002b98SHong Zhang   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
111d4002b98SHong Zhang   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
112d4002b98SHong Zhang 
113d4002b98SHong Zhang   /* FIXME: if one preallocates more space than needed, the matrix does not shrink automatically, but for best performance it should */
114d4002b98SHong Zhang   if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 5;
115*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(maxallocrow < 0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"maxallocrow cannot be less than 0: value %" PetscInt_FMT,maxallocrow);
116d4002b98SHong Zhang   if (rlen) {
117d4002b98SHong Zhang     for (i=0; i<B->rmap->n; i++) {
118*2c71b3e2SJacob Faibussowitsch       PetscCheckFalse(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]);
119*2c71b3e2SJacob Faibussowitsch       PetscCheckFalse(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);
120d4002b98SHong Zhang     }
121d4002b98SHong Zhang   }
122d4002b98SHong Zhang 
123d4002b98SHong Zhang   B->preallocated = PETSC_TRUE;
124d4002b98SHong Zhang 
125d4002b98SHong Zhang   b = (Mat_SeqSELL*)B->data;
126d4002b98SHong Zhang 
127faa75363SBarry Smith   totalslices = PetscCeilInt(B->rmap->n,8);
128d4002b98SHong Zhang   b->totalslices = totalslices;
129d4002b98SHong Zhang   if (!skipallocation) {
1307d3de750SJacob Faibussowitsch     if (B->rmap->n & 0x07) {ierr = 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);CHKERRQ(ierr);}
131d4002b98SHong Zhang 
132d4002b98SHong Zhang     if (!b->sliidx) { /* sliidx gives the starting index of each slice, the last element is the total space allocated */
133d4002b98SHong Zhang       ierr = PetscMalloc1(totalslices+1,&b->sliidx);CHKERRQ(ierr);
134d4002b98SHong Zhang       ierr = PetscLogObjectMemory((PetscObject)B,(totalslices+1)*sizeof(PetscInt));CHKERRQ(ierr);
135d4002b98SHong Zhang     }
136d4002b98SHong Zhang     if (!rlen) { /* if rlen is not provided, allocate same space for all the slices */
137d4002b98SHong Zhang       if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 10;
138d4002b98SHong Zhang       else if (maxallocrow < 0) maxallocrow = 1;
139d4002b98SHong Zhang       for (i=0; i<=totalslices; i++) b->sliidx[i] = i*8*maxallocrow;
140d4002b98SHong Zhang     } else {
141d4002b98SHong Zhang       maxallocrow = 0;
142d4002b98SHong Zhang       b->sliidx[0] = 0;
143d4002b98SHong Zhang       for (i=1; i<totalslices; i++) {
144d4002b98SHong Zhang         b->sliidx[i] = 0;
145d4002b98SHong Zhang         for (j=0;j<8;j++) {
146d4002b98SHong Zhang           b->sliidx[i] = PetscMax(b->sliidx[i],rlen[8*(i-1)+j]);
147d4002b98SHong Zhang         }
148d4002b98SHong Zhang         maxallocrow = PetscMax(b->sliidx[i],maxallocrow);
149c73702f5SBarry Smith         ierr = PetscIntSumError(b->sliidx[i-1],8*b->sliidx[i],&b->sliidx[i]);CHKERRQ(ierr);
150d4002b98SHong Zhang       }
151d4002b98SHong Zhang       /* last slice */
152d4002b98SHong Zhang       b->sliidx[totalslices] = 0;
153d4002b98SHong Zhang       for (j=(totalslices-1)*8;j<B->rmap->n;j++) b->sliidx[totalslices] = PetscMax(b->sliidx[totalslices],rlen[j]);
154d4002b98SHong Zhang       maxallocrow = PetscMax(b->sliidx[totalslices],maxallocrow);
155d4002b98SHong Zhang       b->sliidx[totalslices] = b->sliidx[totalslices-1] + 8*b->sliidx[totalslices];
156d4002b98SHong Zhang     }
157d4002b98SHong Zhang 
158d4002b98SHong Zhang     /* allocate space for val, colidx, rlen */
159d4002b98SHong Zhang     /* FIXME: should B's old memory be unlogged? */
160d4002b98SHong Zhang     ierr = MatSeqXSELLFreeSELL(B,&b->val,&b->colidx);CHKERRQ(ierr);
161d4002b98SHong Zhang     /* FIXME: assuming an element of the bit array takes 8 bits */
162d4002b98SHong Zhang     ierr = PetscMalloc2(b->sliidx[totalslices],&b->val,b->sliidx[totalslices],&b->colidx);CHKERRQ(ierr);
163d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)B,b->sliidx[totalslices]*(sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
164d4002b98SHong 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. */
165d4002b98SHong Zhang     ierr = PetscCalloc1(8*totalslices,&b->rlen);CHKERRQ(ierr);
166d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)B,8*totalslices*sizeof(PetscInt));CHKERRQ(ierr);
167d4002b98SHong Zhang 
168d4002b98SHong Zhang     b->singlemalloc = PETSC_TRUE;
169d4002b98SHong Zhang     b->free_val     = PETSC_TRUE;
170d4002b98SHong Zhang     b->free_colidx  = PETSC_TRUE;
171d4002b98SHong Zhang   } else {
172d4002b98SHong Zhang     b->free_val    = PETSC_FALSE;
173d4002b98SHong Zhang     b->free_colidx = PETSC_FALSE;
174d4002b98SHong Zhang   }
175d4002b98SHong Zhang 
176d4002b98SHong Zhang   b->nz               = 0;
177d4002b98SHong Zhang   b->maxallocrow      = maxallocrow;
178d4002b98SHong Zhang   b->rlenmax          = maxallocrow;
179d4002b98SHong Zhang   b->maxallocmat      = b->sliidx[totalslices];
180d4002b98SHong Zhang   B->info.nz_unneeded = (double)b->maxallocmat;
181d4002b98SHong Zhang   if (realalloc) {
182d4002b98SHong Zhang     ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
183d4002b98SHong Zhang   }
184d4002b98SHong Zhang   PetscFunctionReturn(0);
185d4002b98SHong Zhang }
186d4002b98SHong Zhang 
1876108893eSStefano Zampini PetscErrorCode MatGetRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
1886108893eSStefano Zampini {
1896108893eSStefano Zampini   Mat_SeqSELL *a = (Mat_SeqSELL*)A->data;
1906108893eSStefano Zampini   PetscInt    shift;
1916108893eSStefano Zampini 
1926108893eSStefano Zampini   PetscFunctionBegin;
193*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(row < 0 || row >= A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %" PetscInt_FMT " out of range",row);
1946108893eSStefano Zampini   if (nz) *nz = a->rlen[row];
1956108893eSStefano Zampini   shift = a->sliidx[row>>3]+(row&0x07);
1966108893eSStefano Zampini   if (!a->getrowcols) {
1976108893eSStefano Zampini     PetscErrorCode ierr;
1986108893eSStefano Zampini 
1996108893eSStefano Zampini     ierr = PetscMalloc2(a->rlenmax,&a->getrowcols,a->rlenmax,&a->getrowvals);CHKERRQ(ierr);
2006108893eSStefano Zampini   }
2016108893eSStefano Zampini   if (idx) {
2026108893eSStefano Zampini     PetscInt j;
2036108893eSStefano Zampini     for (j=0; j<a->rlen[row]; j++) a->getrowcols[j] = a->colidx[shift+8*j];
2046108893eSStefano Zampini     *idx = a->getrowcols;
2056108893eSStefano Zampini   }
2066108893eSStefano Zampini   if (v) {
2076108893eSStefano Zampini     PetscInt j;
2086108893eSStefano Zampini     for (j=0; j<a->rlen[row]; j++) a->getrowvals[j] = a->val[shift+8*j];
2096108893eSStefano Zampini     *v = a->getrowvals;
2106108893eSStefano Zampini   }
2116108893eSStefano Zampini   PetscFunctionReturn(0);
2126108893eSStefano Zampini }
2136108893eSStefano Zampini 
2146108893eSStefano Zampini PetscErrorCode MatRestoreRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
2156108893eSStefano Zampini {
2166108893eSStefano Zampini   PetscFunctionBegin;
2176108893eSStefano Zampini   PetscFunctionReturn(0);
2186108893eSStefano Zampini }
2196108893eSStefano Zampini 
220d4002b98SHong Zhang PetscErrorCode MatConvert_SeqSELL_SeqAIJ(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
221d4002b98SHong Zhang {
222d4002b98SHong Zhang   Mat            B;
223d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
224e3f1f374SStefano Zampini   PetscInt       i;
225d4002b98SHong Zhang   PetscErrorCode ierr;
226d4002b98SHong Zhang 
227d4002b98SHong Zhang   PetscFunctionBegin;
228ad013a7bSRichard Tran Mills   if (reuse == MAT_REUSE_MATRIX) {
229ad013a7bSRichard Tran Mills     B    = *newmat;
230e3f1f374SStefano Zampini     ierr = MatZeroEntries(B);CHKERRQ(ierr);
231ad013a7bSRichard Tran Mills   } else {
232d4002b98SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
233d4002b98SHong Zhang     ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
234d4002b98SHong Zhang     ierr = MatSetType(B,MATSEQAIJ);CHKERRQ(ierr);
235d4002b98SHong Zhang     ierr = MatSeqAIJSetPreallocation(B,0,a->rlen);CHKERRQ(ierr);
236ad013a7bSRichard Tran Mills   }
237d4002b98SHong Zhang 
238e3f1f374SStefano Zampini   for (i=0; i<A->rmap->n; i++) {
239e108cb99SStefano Zampini     PetscInt    nz = 0,*cols = NULL;
240e108cb99SStefano Zampini     PetscScalar *vals = NULL;
241e3f1f374SStefano Zampini 
242e3f1f374SStefano Zampini     ierr = MatGetRow_SeqSELL(A,i,&nz,&cols,&vals);CHKERRQ(ierr);
243e3f1f374SStefano Zampini     ierr = MatSetValues(B,1,&i,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
244e3f1f374SStefano Zampini     ierr = MatRestoreRow_SeqSELL(A,i,&nz,&cols,&vals);CHKERRQ(ierr);
245d4002b98SHong Zhang   }
246e3f1f374SStefano Zampini 
247d4002b98SHong Zhang   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
248d4002b98SHong Zhang   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
249d4002b98SHong Zhang   B->rmap->bs = A->rmap->bs;
250d4002b98SHong Zhang 
251d4002b98SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
252d4002b98SHong Zhang     ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr);
253d4002b98SHong Zhang   } else {
254d4002b98SHong Zhang     *newmat = B;
255d4002b98SHong Zhang   }
256d4002b98SHong Zhang   PetscFunctionReturn(0);
257d4002b98SHong Zhang }
258d4002b98SHong Zhang 
259d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/aij.h>
260d4002b98SHong Zhang 
261d4002b98SHong Zhang PetscErrorCode MatConvert_SeqAIJ_SeqSELL(Mat A,MatType newtype,MatReuse reuse,Mat *newmat)
262d4002b98SHong Zhang {
263d4002b98SHong Zhang   Mat               B;
264d4002b98SHong Zhang   Mat_SeqAIJ        *a=(Mat_SeqAIJ*)A->data;
265d4002b98SHong Zhang   PetscInt          *ai=a->i,m=A->rmap->N,n=A->cmap->N,i,*rowlengths,row,ncols;
266d4002b98SHong Zhang   const PetscInt    *cols;
267d4002b98SHong Zhang   const PetscScalar *vals;
268d4002b98SHong Zhang   PetscErrorCode    ierr;
269d4002b98SHong Zhang 
270d4002b98SHong Zhang   PetscFunctionBegin;
271ad013a7bSRichard Tran Mills 
272ad013a7bSRichard Tran Mills   if (reuse == MAT_REUSE_MATRIX) {
273ad013a7bSRichard Tran Mills     B = *newmat;
274ad013a7bSRichard Tran Mills   } else {
275d5e5b2e5SBarry Smith     if (PetscDefined(USE_DEBUG) || !a->ilen) {
276d4002b98SHong Zhang       ierr = PetscMalloc1(m,&rowlengths);CHKERRQ(ierr);
277d4002b98SHong Zhang       for (i=0; i<m; i++) {
278d4002b98SHong Zhang         rowlengths[i] = ai[i+1] - ai[i];
279d4002b98SHong Zhang       }
280d5e5b2e5SBarry Smith     }
281d5e5b2e5SBarry Smith     if (PetscDefined(USE_DEBUG) && a->ilen) {
282d5e5b2e5SBarry Smith       PetscBool eq;
283d5e5b2e5SBarry Smith       ierr = PetscMemcmp(rowlengths,a->ilen,m*sizeof(PetscInt),&eq);CHKERRQ(ierr);
284*2c71b3e2SJacob Faibussowitsch       PetscCheckFalse(!eq,PETSC_COMM_SELF,PETSC_ERR_PLIB,"SeqAIJ ilen array incorrect");
285d5e5b2e5SBarry Smith       ierr = PetscFree(rowlengths);CHKERRQ(ierr);
286d5e5b2e5SBarry Smith       rowlengths = a->ilen;
287d5e5b2e5SBarry Smith     } else if (a->ilen) rowlengths = a->ilen;
288d4002b98SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
289d4002b98SHong Zhang     ierr = MatSetSizes(B,m,n,m,n);CHKERRQ(ierr);
290d4002b98SHong Zhang     ierr = MatSetType(B,MATSEQSELL);CHKERRQ(ierr);
291d4002b98SHong Zhang     ierr = MatSeqSELLSetPreallocation(B,0,rowlengths);CHKERRQ(ierr);
292d5e5b2e5SBarry Smith     if (rowlengths != a->ilen) {ierr = PetscFree(rowlengths);CHKERRQ(ierr);}
293ad013a7bSRichard Tran Mills   }
294d4002b98SHong Zhang 
295d4002b98SHong Zhang   for (row=0; row<m; row++) {
296d5e5b2e5SBarry Smith     ierr = MatGetRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals);CHKERRQ(ierr);
297d5e5b2e5SBarry Smith     ierr = MatSetValues_SeqSELL(B,1,&row,ncols,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
298d5e5b2e5SBarry Smith     ierr = MatRestoreRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals);CHKERRQ(ierr);
299d4002b98SHong Zhang   }
300d4002b98SHong Zhang   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
301d4002b98SHong Zhang   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
302d4002b98SHong Zhang   B->rmap->bs = A->rmap->bs;
303d4002b98SHong Zhang 
304d4002b98SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
305d4002b98SHong Zhang     ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr);
306d4002b98SHong Zhang   } else {
307d4002b98SHong Zhang     *newmat = B;
308d4002b98SHong Zhang   }
309d4002b98SHong Zhang   PetscFunctionReturn(0);
310d4002b98SHong Zhang }
311d4002b98SHong Zhang 
312d4002b98SHong Zhang PetscErrorCode MatMult_SeqSELL(Mat A,Vec xx,Vec yy)
313d4002b98SHong Zhang {
314d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
315d4002b98SHong Zhang   PetscScalar       *y;
316d4002b98SHong Zhang   const PetscScalar *x;
317d4002b98SHong Zhang   const MatScalar   *aval=a->val;
318d4002b98SHong Zhang   PetscInt          totalslices=a->totalslices;
319d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
3207285fed1SHong Zhang   PetscInt          i,j;
321d4002b98SHong Zhang   PetscErrorCode    ierr;
322d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
323d4002b98SHong Zhang   __m512d           vec_x,vec_y,vec_vals;
324d4002b98SHong Zhang   __m256i           vec_idx;
325d4002b98SHong Zhang   __mmask8          mask;
326d4002b98SHong Zhang   __m512d           vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4;
327d4002b98SHong Zhang   __m256i           vec_idx2,vec_idx3,vec_idx4;
3285f70456aSHong 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)
329a48a6482SHong Zhang   __m128i           vec_idx;
330a48a6482SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
331a48a6482SHong Zhang   MatScalar         yval;
332a48a6482SHong Zhang   PetscInt          r,rows_left,row,nnz_in_row;
33321cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
334d4002b98SHong Zhang   __m128d           vec_x_tmp;
335d4002b98SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
336d4002b98SHong Zhang   MatScalar         yval;
337d4002b98SHong Zhang   PetscInt          r,rows_left,row,nnz_in_row;
338d4002b98SHong Zhang #else
339d4002b98SHong Zhang   PetscScalar       sum[8];
340d4002b98SHong Zhang #endif
341d4002b98SHong Zhang 
342d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
343d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
344d4002b98SHong Zhang #endif
345d4002b98SHong Zhang 
346d4002b98SHong Zhang   PetscFunctionBegin;
347d4002b98SHong Zhang   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
348d4002b98SHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
349d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
350d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
351d4002b98SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
352d4002b98SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
353d4002b98SHong Zhang 
354d4002b98SHong Zhang     vec_y  = _mm512_setzero_pd();
355d4002b98SHong Zhang     vec_y2 = _mm512_setzero_pd();
356d4002b98SHong Zhang     vec_y3 = _mm512_setzero_pd();
357d4002b98SHong Zhang     vec_y4 = _mm512_setzero_pd();
358d4002b98SHong Zhang 
35938efe8efSHong Zhang     j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */
360d4002b98SHong Zhang     switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) {
361d4002b98SHong Zhang     case 3:
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       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
367d4002b98SHong Zhang       acolidx += 8; aval += 8;
368d4002b98SHong Zhang       j += 3;
369d4002b98SHong Zhang       break;
370d4002b98SHong Zhang     case 2:
371d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
372d4002b98SHong Zhang       acolidx += 8; aval += 8;
373d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
374d4002b98SHong Zhang       acolidx += 8; aval += 8;
375d4002b98SHong Zhang       j += 2;
376d4002b98SHong Zhang       break;
377d4002b98SHong Zhang     case 1:
378d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
379d4002b98SHong Zhang       acolidx += 8; aval += 8;
380d4002b98SHong Zhang       j += 1;
381d4002b98SHong Zhang       break;
382d4002b98SHong Zhang     }
383d4002b98SHong Zhang     #pragma novector
384d4002b98SHong Zhang     for (; j<(a->sliidx[i+1]>>3); j+=4) {
385d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
386d4002b98SHong Zhang       acolidx += 8; aval += 8;
387d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
388d4002b98SHong Zhang       acolidx += 8; aval += 8;
389d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
390d4002b98SHong Zhang       acolidx += 8; aval += 8;
391d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4);
392d4002b98SHong Zhang       acolidx += 8; aval += 8;
393d4002b98SHong Zhang     }
394d4002b98SHong Zhang 
395d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y2);
396d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y3);
397d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y4);
398d4002b98SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
399d4002b98SHong Zhang       mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07)));
400ef588d5cSRichard Tran Mills       _mm512_mask_storeu_pd(&y[8*i],mask,vec_y);
401d4002b98SHong Zhang     } else {
402ef588d5cSRichard Tran Mills       _mm512_storeu_pd(&y[8*i],vec_y);
403d4002b98SHong Zhang     }
404d4002b98SHong Zhang   }
4055f70456aSHong 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)
406a48a6482SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
407a48a6482SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
408a48a6482SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
409a48a6482SHong Zhang 
410a48a6482SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
411a48a6482SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
412a48a6482SHong Zhang       rows_left = A->rmap->n - 8*i;
413a48a6482SHong Zhang       for (r=0; r<rows_left; ++r) {
414a48a6482SHong Zhang         yval = (MatScalar)0;
415a48a6482SHong Zhang         row = 8*i + r;
416a48a6482SHong Zhang         nnz_in_row = a->rlen[row];
417a48a6482SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]];
418a48a6482SHong Zhang         y[row] = yval;
419a48a6482SHong Zhang       }
420a48a6482SHong Zhang       break;
421a48a6482SHong Zhang     }
422a48a6482SHong Zhang 
423a48a6482SHong Zhang     vec_y  = _mm256_setzero_pd();
424a48a6482SHong Zhang     vec_y2 = _mm256_setzero_pd();
425a48a6482SHong Zhang 
426a48a6482SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
427a48a6482SHong Zhang     #pragma novector
428a48a6482SHong Zhang     #pragma unroll(2)
429a48a6482SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
430a48a6482SHong Zhang       AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
431a48a6482SHong Zhang       aval += 4; acolidx += 4;
432a48a6482SHong Zhang       AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y2);
433a48a6482SHong Zhang       aval += 4; acolidx += 4;
434a48a6482SHong Zhang     }
435a48a6482SHong Zhang 
436ef588d5cSRichard Tran Mills     _mm256_storeu_pd(y+i*8,vec_y);
437ef588d5cSRichard Tran Mills     _mm256_storeu_pd(y+i*8+4,vec_y2);
438a48a6482SHong Zhang   }
43921cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
440d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
441d4002b98SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
442d4002b98SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
443d4002b98SHong Zhang 
444d4002b98SHong Zhang     vec_y  = _mm256_setzero_pd();
445d4002b98SHong Zhang     vec_y2 = _mm256_setzero_pd();
446d4002b98SHong Zhang 
447d4002b98SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
448d4002b98SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
449d4002b98SHong Zhang       rows_left = A->rmap->n - 8*i;
450d4002b98SHong Zhang       for (r=0; r<rows_left; ++r) {
451d4002b98SHong Zhang         yval = (MatScalar)0;
452d4002b98SHong Zhang         row = 8*i + r;
453d4002b98SHong Zhang         nnz_in_row = a->rlen[row];
454d4002b98SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j + r] * x[acolidx[8*j + r]];
455d4002b98SHong Zhang         y[row] = yval;
456d4002b98SHong Zhang       }
457d4002b98SHong Zhang       break;
458d4002b98SHong Zhang     }
459d4002b98SHong Zhang 
460d4002b98SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
461a48a6482SHong Zhang     #pragma novector
462a48a6482SHong Zhang     #pragma unroll(2)
4637285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
464d4002b98SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
465d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
466d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
467d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
468d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
469d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
470d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
471d4002b98SHong Zhang       vec_y     = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y);
472d4002b98SHong Zhang       aval     += 4;
473d4002b98SHong Zhang 
474d4002b98SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
475d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
476d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
477d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
478d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
479d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
480d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
481d4002b98SHong Zhang       vec_y2    = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2);
482d4002b98SHong Zhang       aval     += 4;
483d4002b98SHong Zhang     }
484d4002b98SHong Zhang 
485d4002b98SHong Zhang     _mm256_storeu_pd(y + i*8,     vec_y);
486d4002b98SHong Zhang     _mm256_storeu_pd(y + i*8 + 4, vec_y2);
487d4002b98SHong Zhang   }
488d4002b98SHong Zhang #else
489d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
490d4002b98SHong Zhang     for (j=0; j<8; j++) sum[j] = 0.0;
491d4002b98SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
492d4002b98SHong Zhang       sum[0] += aval[j] * x[acolidx[j]];
493d4002b98SHong Zhang       sum[1] += aval[j+1] * x[acolidx[j+1]];
494d4002b98SHong Zhang       sum[2] += aval[j+2] * x[acolidx[j+2]];
495d4002b98SHong Zhang       sum[3] += aval[j+3] * x[acolidx[j+3]];
496d4002b98SHong Zhang       sum[4] += aval[j+4] * x[acolidx[j+4]];
497d4002b98SHong Zhang       sum[5] += aval[j+5] * x[acolidx[j+5]];
498d4002b98SHong Zhang       sum[6] += aval[j+6] * x[acolidx[j+6]];
499d4002b98SHong Zhang       sum[7] += aval[j+7] * x[acolidx[j+7]];
500d4002b98SHong Zhang     }
501d4002b98SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
502d4002b98SHong Zhang       for (j=0; j<(A->rmap->n & 0x07); j++) y[8*i+j] = sum[j];
503d4002b98SHong Zhang     } else {
5047285fed1SHong Zhang       for (j=0; j<8; j++) y[8*i+j] = sum[j];
505d4002b98SHong Zhang     }
506d4002b98SHong Zhang   }
507d4002b98SHong Zhang #endif
508d4002b98SHong Zhang 
509d4002b98SHong Zhang   ierr = PetscLogFlops(2.0*a->nz-a->nonzerorowcnt);CHKERRQ(ierr); /* theoretical minimal FLOPs */
510d4002b98SHong Zhang   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
511d4002b98SHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
512d4002b98SHong Zhang   PetscFunctionReturn(0);
513d4002b98SHong Zhang }
514d4002b98SHong Zhang 
515d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/ftn-kernels/fmultadd.h>
516d4002b98SHong Zhang PetscErrorCode MatMultAdd_SeqSELL(Mat A,Vec xx,Vec yy,Vec zz)
517d4002b98SHong Zhang {
518d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
519d4002b98SHong Zhang   PetscScalar       *y,*z;
520d4002b98SHong Zhang   const PetscScalar *x;
521d4002b98SHong Zhang   const MatScalar   *aval=a->val;
522d4002b98SHong Zhang   PetscInt          totalslices=a->totalslices;
523d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
524d4002b98SHong Zhang   PetscInt          i,j;
525d4002b98SHong Zhang   PetscErrorCode    ierr;
526d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5277285fed1SHong Zhang   __m512d           vec_x,vec_y,vec_vals;
528d4002b98SHong Zhang   __m256i           vec_idx;
529d4002b98SHong Zhang   __mmask8          mask;
5307285fed1SHong Zhang   __m512d           vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4;
5317285fed1SHong Zhang   __m256i           vec_idx2,vec_idx3,vec_idx4;
53221cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5337285fed1SHong Zhang   __m128d           vec_x_tmp;
5347285fed1SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
5357285fed1SHong Zhang   MatScalar         yval;
5367285fed1SHong Zhang   PetscInt          r,row,nnz_in_row;
537d4002b98SHong Zhang #else
538d4002b98SHong Zhang   PetscScalar       sum[8];
539d4002b98SHong Zhang #endif
540d4002b98SHong Zhang 
541d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
542d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
543d4002b98SHong Zhang #endif
544d4002b98SHong Zhang 
545d4002b98SHong Zhang   PetscFunctionBegin;
546d4002b98SHong Zhang   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
547d4002b98SHong Zhang   ierr = VecGetArrayPair(yy,zz,&y,&z);CHKERRQ(ierr);
548d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5497285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
5507285fed1SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
5517285fed1SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
5527285fed1SHong Zhang 
553d4002b98SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
554d4002b98SHong Zhang       mask   = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07)));
555ef588d5cSRichard Tran Mills       vec_y  = _mm512_mask_loadu_pd(vec_y,mask,&y[8*i]);
5567285fed1SHong Zhang     } else {
557ef588d5cSRichard Tran Mills       vec_y  = _mm512_loadu_pd(&y[8*i]);
5587285fed1SHong Zhang     }
5597285fed1SHong Zhang     vec_y2 = _mm512_setzero_pd();
5607285fed1SHong Zhang     vec_y3 = _mm512_setzero_pd();
5617285fed1SHong Zhang     vec_y4 = _mm512_setzero_pd();
5627285fed1SHong Zhang 
5637285fed1SHong Zhang     j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */
5647285fed1SHong Zhang     switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) {
5657285fed1SHong Zhang     case 3:
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       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
5717285fed1SHong Zhang       acolidx += 8; aval += 8;
5727285fed1SHong Zhang       j += 3;
5737285fed1SHong Zhang       break;
5747285fed1SHong Zhang     case 2:
5757285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5767285fed1SHong Zhang       acolidx += 8; aval += 8;
5777285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5787285fed1SHong Zhang       acolidx += 8; aval += 8;
5797285fed1SHong Zhang       j += 2;
5807285fed1SHong Zhang       break;
5817285fed1SHong Zhang     case 1:
5827285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5837285fed1SHong Zhang       acolidx += 8; aval += 8;
5847285fed1SHong Zhang       j += 1;
5857285fed1SHong Zhang       break;
5867285fed1SHong Zhang     }
5877285fed1SHong Zhang     #pragma novector
5887285fed1SHong Zhang     for (; j<(a->sliidx[i+1]>>3); j+=4) {
5897285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5907285fed1SHong Zhang       acolidx += 8; aval += 8;
5917285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5927285fed1SHong Zhang       acolidx += 8; aval += 8;
5937285fed1SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
5947285fed1SHong Zhang       acolidx += 8; aval += 8;
5957285fed1SHong Zhang       AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4);
5967285fed1SHong Zhang       acolidx += 8; aval += 8;
5977285fed1SHong Zhang     }
5987285fed1SHong Zhang 
5997285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y2);
6007285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y3);
6017285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y4);
6027285fed1SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
603ef588d5cSRichard Tran Mills       _mm512_mask_storeu_pd(&z[8*i],mask,vec_y);
604d4002b98SHong Zhang     } else {
605ef588d5cSRichard Tran Mills       _mm512_storeu_pd(&z[8*i],vec_y);
606d4002b98SHong Zhang     }
6077285fed1SHong Zhang   }
60821cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
6097285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
6107285fed1SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
6117285fed1SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
6127285fed1SHong Zhang 
6137285fed1SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
6147285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6157285fed1SHong Zhang       for (r=0; r<(A->rmap->n & 0x07); ++r) {
6167285fed1SHong Zhang         row        = 8*i + r;
6177285fed1SHong Zhang         yval       = (MatScalar)0.0;
6187285fed1SHong Zhang         nnz_in_row = a->rlen[row];
6197285fed1SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]];
6207285fed1SHong Zhang         z[row] = y[row] + yval;
6217285fed1SHong Zhang       }
6227285fed1SHong Zhang       break;
6237285fed1SHong Zhang     }
6247285fed1SHong Zhang 
6257285fed1SHong Zhang     vec_y  = _mm256_loadu_pd(y+8*i);
6267285fed1SHong Zhang     vec_y2 = _mm256_loadu_pd(y+8*i+4);
6277285fed1SHong Zhang 
6287285fed1SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
6297285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
6307285fed1SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
6317285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6327285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6337285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
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,1);
6377285fed1SHong Zhang       vec_y     = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y);
6387285fed1SHong Zhang       aval     += 4;
6397285fed1SHong Zhang 
6407285fed1SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
6417285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6427285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6437285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
6447285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6457285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6467285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
6477285fed1SHong Zhang       vec_y2    = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2);
6487285fed1SHong Zhang       aval     += 4;
6497285fed1SHong Zhang     }
6507285fed1SHong Zhang 
6517285fed1SHong Zhang     _mm256_storeu_pd(z+i*8,vec_y);
6527285fed1SHong Zhang     _mm256_storeu_pd(z+i*8+4,vec_y2);
6537285fed1SHong Zhang   }
654d4002b98SHong Zhang #else
6557285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
6567285fed1SHong Zhang     for (j=0; j<8; j++) sum[j] = 0.0;
657d4002b98SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
658d4002b98SHong Zhang       sum[0] += aval[j] * x[acolidx[j]];
659d4002b98SHong Zhang       sum[1] += aval[j+1] * x[acolidx[j+1]];
660d4002b98SHong Zhang       sum[2] += aval[j+2] * x[acolidx[j+2]];
661d4002b98SHong Zhang       sum[3] += aval[j+3] * x[acolidx[j+3]];
662d4002b98SHong Zhang       sum[4] += aval[j+4] * x[acolidx[j+4]];
663d4002b98SHong Zhang       sum[5] += aval[j+5] * x[acolidx[j+5]];
664d4002b98SHong Zhang       sum[6] += aval[j+6] * x[acolidx[j+6]];
665d4002b98SHong Zhang       sum[7] += aval[j+7] * x[acolidx[j+7]];
666d4002b98SHong Zhang     }
6677285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6687285fed1SHong Zhang       for (j=0; j<(A->rmap->n & 0x07); j++) z[8*i+j] = y[8*i+j] + sum[j];
669d4002b98SHong Zhang     } else {
6707285fed1SHong Zhang       for (j=0; j<8; j++) z[8*i+j] = y[8*i+j] + sum[j];
6717285fed1SHong Zhang     }
672d4002b98SHong Zhang   }
673d4002b98SHong Zhang #endif
674d4002b98SHong Zhang 
675d4002b98SHong Zhang   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
676d4002b98SHong Zhang   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
677d4002b98SHong Zhang   ierr = VecRestoreArrayPair(yy,zz,&y,&z);CHKERRQ(ierr);
678d4002b98SHong Zhang   PetscFunctionReturn(0);
679d4002b98SHong Zhang }
680d4002b98SHong Zhang 
681d4002b98SHong Zhang PetscErrorCode MatMultTransposeAdd_SeqSELL(Mat A,Vec xx,Vec zz,Vec yy)
682d4002b98SHong Zhang {
683d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
684d4002b98SHong Zhang   PetscScalar       *y;
685d4002b98SHong Zhang   const PetscScalar *x;
686d4002b98SHong Zhang   const MatScalar   *aval=a->val;
687d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
6887285fed1SHong Zhang   PetscInt          i,j,r,row,nnz_in_row,totalslices=a->totalslices;
689d4002b98SHong Zhang   PetscErrorCode    ierr;
690d4002b98SHong Zhang 
691d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
692d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
693d4002b98SHong Zhang #endif
694d4002b98SHong Zhang 
695d4002b98SHong Zhang   PetscFunctionBegin;
6969fc32365SStefano Zampini   if (A->symmetric) {
6979fc32365SStefano Zampini     ierr = MatMultAdd_SeqSELL(A,xx,zz,yy);CHKERRQ(ierr);
6989fc32365SStefano Zampini     PetscFunctionReturn(0);
6999fc32365SStefano Zampini   }
700d4002b98SHong Zhang   if (zz != yy) { ierr = VecCopy(zz,yy);CHKERRQ(ierr); }
701d4002b98SHong Zhang   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
702d4002b98SHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
703d4002b98SHong Zhang   for (i=0; i<a->totalslices; i++) { /* loop over slices */
7047285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
7057285fed1SHong Zhang       for (r=0; r<(A->rmap->n & 0x07); ++r) {
7067285fed1SHong Zhang         row        = 8*i + r;
7077285fed1SHong Zhang         nnz_in_row = a->rlen[row];
7087285fed1SHong Zhang         for (j=0; j<nnz_in_row; ++j) y[acolidx[8*j+r]] += aval[8*j+r] * x[row];
7097285fed1SHong Zhang       }
7107285fed1SHong Zhang       break;
7117285fed1SHong Zhang     }
7127285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
7137285fed1SHong Zhang       y[acolidx[j]]   += aval[j] * x[8*i];
7147285fed1SHong Zhang       y[acolidx[j+1]] += aval[j+1] * x[8*i+1];
7157285fed1SHong Zhang       y[acolidx[j+2]] += aval[j+2] * x[8*i+2];
7167285fed1SHong Zhang       y[acolidx[j+3]] += aval[j+3] * x[8*i+3];
7177285fed1SHong Zhang       y[acolidx[j+4]] += aval[j+4] * x[8*i+4];
7187285fed1SHong Zhang       y[acolidx[j+5]] += aval[j+5] * x[8*i+5];
7197285fed1SHong Zhang       y[acolidx[j+6]] += aval[j+6] * x[8*i+6];
7207285fed1SHong Zhang       y[acolidx[j+7]] += aval[j+7] * x[8*i+7];
721d4002b98SHong Zhang     }
722d4002b98SHong Zhang   }
723d4002b98SHong Zhang   ierr = PetscLogFlops(2.0*a->sliidx[a->totalslices]);CHKERRQ(ierr);
724d4002b98SHong Zhang   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
725d4002b98SHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
726d4002b98SHong Zhang   PetscFunctionReturn(0);
727d4002b98SHong Zhang }
728d4002b98SHong Zhang 
729d4002b98SHong Zhang PetscErrorCode MatMultTranspose_SeqSELL(Mat A,Vec xx,Vec yy)
730d4002b98SHong Zhang {
731d4002b98SHong Zhang   PetscErrorCode ierr;
732d4002b98SHong Zhang 
733d4002b98SHong Zhang   PetscFunctionBegin;
7349fc32365SStefano Zampini   if (A->symmetric) {
7359fc32365SStefano Zampini     ierr = MatMult_SeqSELL(A,xx,yy);CHKERRQ(ierr);
7369fc32365SStefano Zampini   } else {
737d4002b98SHong Zhang     ierr = VecSet(yy,0.0);CHKERRQ(ierr);
738d4002b98SHong Zhang     ierr = MatMultTransposeAdd_SeqSELL(A,xx,yy,yy);CHKERRQ(ierr);
7399fc32365SStefano Zampini   }
740d4002b98SHong Zhang   PetscFunctionReturn(0);
741d4002b98SHong Zhang }
742d4002b98SHong Zhang 
743d4002b98SHong Zhang /*
744d4002b98SHong Zhang      Checks for missing diagonals
745d4002b98SHong Zhang */
746d4002b98SHong Zhang PetscErrorCode MatMissingDiagonal_SeqSELL(Mat A,PetscBool  *missing,PetscInt *d)
747d4002b98SHong Zhang {
748d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
749d4002b98SHong Zhang   PetscInt       *diag,i;
750994fe344SLisandro Dalcin   PetscErrorCode ierr;
751d4002b98SHong Zhang 
752d4002b98SHong Zhang   PetscFunctionBegin;
753d4002b98SHong Zhang   *missing = PETSC_FALSE;
754d4002b98SHong Zhang   if (A->rmap->n > 0 && !(a->colidx)) {
755d4002b98SHong Zhang     *missing = PETSC_TRUE;
756d4002b98SHong Zhang     if (d) *d = 0;
757994fe344SLisandro Dalcin     ierr = PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n");CHKERRQ(ierr);
758d4002b98SHong Zhang   } else {
759d4002b98SHong Zhang     diag = a->diag;
760d4002b98SHong Zhang     for (i=0; i<A->rmap->n; i++) {
761d4002b98SHong Zhang       if (diag[i] == -1) {
762d4002b98SHong Zhang         *missing = PETSC_TRUE;
763d4002b98SHong Zhang         if (d) *d = i;
7647d3de750SJacob Faibussowitsch         ierr = PetscInfo(A,"Matrix is missing diagonal number %" PetscInt_FMT "\n",i);CHKERRQ(ierr);
765d4002b98SHong Zhang         break;
766d4002b98SHong Zhang       }
767d4002b98SHong Zhang     }
768d4002b98SHong Zhang   }
769d4002b98SHong Zhang   PetscFunctionReturn(0);
770d4002b98SHong Zhang }
771d4002b98SHong Zhang 
772d4002b98SHong Zhang PetscErrorCode MatMarkDiagonal_SeqSELL(Mat A)
773d4002b98SHong Zhang {
774d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
775d4002b98SHong Zhang   PetscInt       i,j,m=A->rmap->n,shift;
776d4002b98SHong Zhang   PetscErrorCode ierr;
777d4002b98SHong Zhang 
778d4002b98SHong Zhang   PetscFunctionBegin;
779d4002b98SHong Zhang   if (!a->diag) {
780d4002b98SHong Zhang     ierr         = PetscMalloc1(m,&a->diag);CHKERRQ(ierr);
781d4002b98SHong Zhang     ierr         = PetscLogObjectMemory((PetscObject)A,m*sizeof(PetscInt));CHKERRQ(ierr);
782d4002b98SHong Zhang     a->free_diag = PETSC_TRUE;
783d4002b98SHong Zhang   }
784d4002b98SHong Zhang   for (i=0; i<m; i++) { /* loop over rows */
785d4002b98SHong Zhang     shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
786d4002b98SHong Zhang     a->diag[i] = -1;
787d4002b98SHong Zhang     for (j=0; j<a->rlen[i]; j++) {
788d4002b98SHong Zhang       if (a->colidx[shift+j*8] == i) {
789d4002b98SHong Zhang         a->diag[i] = shift+j*8;
790d4002b98SHong Zhang         break;
791d4002b98SHong Zhang       }
792d4002b98SHong Zhang     }
793d4002b98SHong Zhang   }
794d4002b98SHong Zhang   PetscFunctionReturn(0);
795d4002b98SHong Zhang }
796d4002b98SHong Zhang 
797d4002b98SHong Zhang /*
798d4002b98SHong Zhang   Negative shift indicates do not generate an error if there is a zero diagonal, just invert it anyways
799d4002b98SHong Zhang */
800d4002b98SHong Zhang PetscErrorCode MatInvertDiagonal_SeqSELL(Mat A,PetscScalar omega,PetscScalar fshift)
801d4002b98SHong Zhang {
802d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*) A->data;
803d4002b98SHong Zhang   PetscInt       i,*diag,m = A->rmap->n;
804d4002b98SHong Zhang   MatScalar      *val = a->val;
805d4002b98SHong Zhang   PetscScalar    *idiag,*mdiag;
806d4002b98SHong Zhang   PetscErrorCode ierr;
807d4002b98SHong Zhang 
808d4002b98SHong Zhang   PetscFunctionBegin;
809d4002b98SHong Zhang   if (a->idiagvalid) PetscFunctionReturn(0);
810d4002b98SHong Zhang   ierr = MatMarkDiagonal_SeqSELL(A);CHKERRQ(ierr);
811d4002b98SHong Zhang   diag = a->diag;
812d4002b98SHong Zhang   if (!a->idiag) {
813d4002b98SHong Zhang     ierr = PetscMalloc3(m,&a->idiag,m,&a->mdiag,m,&a->ssor_work);CHKERRQ(ierr);
814d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)A, 3*m*sizeof(PetscScalar));CHKERRQ(ierr);
815d4002b98SHong Zhang     val  = a->val;
816d4002b98SHong Zhang   }
817d4002b98SHong Zhang   mdiag = a->mdiag;
818d4002b98SHong Zhang   idiag = a->idiag;
819d4002b98SHong Zhang 
820d4002b98SHong Zhang   if (omega == 1.0 && PetscRealPart(fshift) <= 0.0) {
821d4002b98SHong Zhang     for (i=0; i<m; i++) {
822d4002b98SHong Zhang       mdiag[i] = val[diag[i]];
823d4002b98SHong Zhang       if (!PetscAbsScalar(mdiag[i])) { /* zero diagonal */
824d4002b98SHong Zhang         if (PetscRealPart(fshift)) {
8257d3de750SJacob Faibussowitsch           ierr = PetscInfo(A,"Zero diagonal on row %" PetscInt_FMT "\n",i);CHKERRQ(ierr);
826d4002b98SHong Zhang           A->factorerrortype             = MAT_FACTOR_NUMERIC_ZEROPIVOT;
827d4002b98SHong Zhang           A->factorerror_zeropivot_value = 0.0;
828d4002b98SHong Zhang           A->factorerror_zeropivot_row   = i;
82998921bdaSJacob Faibussowitsch         } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Zero diagonal on row %" PetscInt_FMT,i);
830d4002b98SHong Zhang       }
831d4002b98SHong Zhang       idiag[i] = 1.0/val[diag[i]];
832d4002b98SHong Zhang     }
833d4002b98SHong Zhang     ierr = PetscLogFlops(m);CHKERRQ(ierr);
834d4002b98SHong Zhang   } else {
835d4002b98SHong Zhang     for (i=0; i<m; i++) {
836d4002b98SHong Zhang       mdiag[i] = val[diag[i]];
837d4002b98SHong Zhang       idiag[i] = omega/(fshift + val[diag[i]]);
838d4002b98SHong Zhang     }
839d4002b98SHong Zhang     ierr = PetscLogFlops(2.0*m);CHKERRQ(ierr);
840d4002b98SHong Zhang   }
841d4002b98SHong Zhang   a->idiagvalid = PETSC_TRUE;
842d4002b98SHong Zhang   PetscFunctionReturn(0);
843d4002b98SHong Zhang }
844d4002b98SHong Zhang 
845d4002b98SHong Zhang PetscErrorCode MatZeroEntries_SeqSELL(Mat A)
846d4002b98SHong Zhang {
847d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
848d4002b98SHong Zhang   PetscErrorCode ierr;
849d4002b98SHong Zhang 
850d4002b98SHong Zhang   PetscFunctionBegin;
851580bdb30SBarry Smith   ierr = PetscArrayzero(a->val,a->sliidx[a->totalslices]);CHKERRQ(ierr);
852d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr);
853d4002b98SHong Zhang   PetscFunctionReturn(0);
854d4002b98SHong Zhang }
855d4002b98SHong Zhang 
856d4002b98SHong Zhang PetscErrorCode MatDestroy_SeqSELL(Mat A)
857d4002b98SHong Zhang {
858d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
859d4002b98SHong Zhang   PetscErrorCode ierr;
860d4002b98SHong Zhang 
861d4002b98SHong Zhang   PetscFunctionBegin;
862d4002b98SHong Zhang #if defined(PETSC_USE_LOG)
863c0aa6a63SJacob Faibussowitsch   PetscLogObjectState((PetscObject)A,"Rows=%" PetscInt_FMT ", Cols=%" PetscInt_FMT ", NZ=%" PetscInt_FMT,A->rmap->n,A->cmap->n,a->nz);
864d4002b98SHong Zhang #endif
865d4002b98SHong Zhang   ierr = MatSeqXSELLFreeSELL(A,&a->val,&a->colidx);CHKERRQ(ierr);
866d4002b98SHong Zhang   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
867d4002b98SHong Zhang   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
868d4002b98SHong Zhang   ierr = PetscFree(a->diag);CHKERRQ(ierr);
869d4002b98SHong Zhang   ierr = PetscFree(a->rlen);CHKERRQ(ierr);
870d4002b98SHong Zhang   ierr = PetscFree(a->sliidx);CHKERRQ(ierr);
871d4002b98SHong Zhang   ierr = PetscFree3(a->idiag,a->mdiag,a->ssor_work);CHKERRQ(ierr);
872d4002b98SHong Zhang   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
873d4002b98SHong Zhang   ierr = ISDestroy(&a->icol);CHKERRQ(ierr);
874d4002b98SHong Zhang   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
8756108893eSStefano Zampini   ierr = PetscFree2(a->getrowcols,a->getrowvals);CHKERRQ(ierr);
876d4002b98SHong Zhang 
877d4002b98SHong Zhang   ierr = PetscFree(A->data);CHKERRQ(ierr);
878d4002b98SHong Zhang 
879f4259b30SLisandro Dalcin   ierr = PetscObjectChangeTypeName((PetscObject)A,NULL);CHKERRQ(ierr);
880d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr);
881d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr);
882d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSELLSetPreallocation_C",NULL);CHKERRQ(ierr);
883d4002b98SHong Zhang   PetscFunctionReturn(0);
884d4002b98SHong Zhang }
885d4002b98SHong Zhang 
886d4002b98SHong Zhang PetscErrorCode MatSetOption_SeqSELL(Mat A,MatOption op,PetscBool flg)
887d4002b98SHong Zhang {
888d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
889d4002b98SHong Zhang   PetscErrorCode ierr;
890d4002b98SHong Zhang 
891d4002b98SHong Zhang   PetscFunctionBegin;
892d4002b98SHong Zhang   switch (op) {
893d4002b98SHong Zhang   case MAT_ROW_ORIENTED:
894d4002b98SHong Zhang     a->roworiented = flg;
895d4002b98SHong Zhang     break;
896d4002b98SHong Zhang   case MAT_KEEP_NONZERO_PATTERN:
897d4002b98SHong Zhang     a->keepnonzeropattern = flg;
898d4002b98SHong Zhang     break;
899d4002b98SHong Zhang   case MAT_NEW_NONZERO_LOCATIONS:
900d4002b98SHong Zhang     a->nonew = (flg ? 0 : 1);
901d4002b98SHong Zhang     break;
902d4002b98SHong Zhang   case MAT_NEW_NONZERO_LOCATION_ERR:
903d4002b98SHong Zhang     a->nonew = (flg ? -1 : 0);
904d4002b98SHong Zhang     break;
905d4002b98SHong Zhang   case MAT_NEW_NONZERO_ALLOCATION_ERR:
906d4002b98SHong Zhang     a->nonew = (flg ? -2 : 0);
907d4002b98SHong Zhang     break;
908d4002b98SHong Zhang   case MAT_UNUSED_NONZERO_LOCATION_ERR:
909d4002b98SHong Zhang     a->nounused = (flg ? -1 : 0);
910d4002b98SHong Zhang     break;
9118c78258cSHong Zhang   case MAT_FORCE_DIAGONAL_ENTRIES:
912d4002b98SHong Zhang   case MAT_IGNORE_OFF_PROC_ENTRIES:
913d4002b98SHong Zhang   case MAT_USE_HASH_TABLE:
914071fcb05SBarry Smith   case MAT_SORTED_FULL:
9157d3de750SJacob Faibussowitsch     ierr = PetscInfo(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
916d4002b98SHong Zhang     break;
917d4002b98SHong Zhang   case MAT_SPD:
918d4002b98SHong Zhang   case MAT_SYMMETRIC:
919d4002b98SHong Zhang   case MAT_STRUCTURALLY_SYMMETRIC:
920d4002b98SHong Zhang   case MAT_HERMITIAN:
921d4002b98SHong Zhang   case MAT_SYMMETRY_ETERNAL:
922d4002b98SHong Zhang     /* These options are handled directly by MatSetOption() */
923d4002b98SHong Zhang     break;
924d4002b98SHong Zhang   default:
92598921bdaSJacob Faibussowitsch     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
926d4002b98SHong Zhang   }
927d4002b98SHong Zhang   PetscFunctionReturn(0);
928d4002b98SHong Zhang }
929d4002b98SHong Zhang 
930d4002b98SHong Zhang PetscErrorCode MatGetDiagonal_SeqSELL(Mat A,Vec v)
931d4002b98SHong Zhang {
932d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
933d4002b98SHong Zhang   PetscInt       i,j,n,shift;
934d4002b98SHong Zhang   PetscScalar    *x,zero=0.0;
935d4002b98SHong Zhang   PetscErrorCode ierr;
936d4002b98SHong Zhang 
937d4002b98SHong Zhang   PetscFunctionBegin;
938d4002b98SHong Zhang   ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr);
939*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(n != A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
940d4002b98SHong Zhang 
941d4002b98SHong Zhang   if (A->factortype == MAT_FACTOR_ILU || A->factortype == MAT_FACTOR_LU) {
942d4002b98SHong Zhang     PetscInt *diag=a->diag;
943d4002b98SHong Zhang     ierr = VecGetArray(v,&x);CHKERRQ(ierr);
944d4002b98SHong Zhang     for (i=0; i<n; i++) x[i] = 1.0/a->val[diag[i]];
945d4002b98SHong Zhang     ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
946d4002b98SHong Zhang     PetscFunctionReturn(0);
947d4002b98SHong Zhang   }
948d4002b98SHong Zhang 
949d4002b98SHong Zhang   ierr = VecSet(v,zero);CHKERRQ(ierr);
950d4002b98SHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
951d4002b98SHong Zhang   for (i=0; i<n; i++) { /* loop over rows */
952d4002b98SHong Zhang     shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
953d4002b98SHong Zhang     x[i] = 0;
954d4002b98SHong Zhang     for (j=0; j<a->rlen[i]; j++) {
955d4002b98SHong Zhang       if (a->colidx[shift+j*8] == i) {
956d4002b98SHong Zhang         x[i] = a->val[shift+j*8];
957d4002b98SHong Zhang         break;
958d4002b98SHong Zhang       }
959d4002b98SHong Zhang     }
960d4002b98SHong Zhang   }
961d4002b98SHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
962d4002b98SHong Zhang   PetscFunctionReturn(0);
963d4002b98SHong Zhang }
964d4002b98SHong Zhang 
965d4002b98SHong Zhang PetscErrorCode MatDiagonalScale_SeqSELL(Mat A,Vec ll,Vec rr)
966d4002b98SHong Zhang {
967d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
968d4002b98SHong Zhang   const PetscScalar *l,*r;
969d4002b98SHong Zhang   PetscInt          i,j,m,n,row;
970d4002b98SHong Zhang   PetscErrorCode    ierr;
971d4002b98SHong Zhang 
972d4002b98SHong Zhang   PetscFunctionBegin;
973d4002b98SHong Zhang   if (ll) {
974d4002b98SHong Zhang     /* The local size is used so that VecMPI can be passed to this routine
975d4002b98SHong Zhang        by MatDiagonalScale_MPISELL */
976d4002b98SHong Zhang     ierr = VecGetLocalSize(ll,&m);CHKERRQ(ierr);
977*2c71b3e2SJacob Faibussowitsch     PetscCheckFalse(m != A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length");
978d4002b98SHong Zhang     ierr = VecGetArrayRead(ll,&l);CHKERRQ(ierr);
979d4002b98SHong Zhang     for (i=0; i<a->totalslices; i++) { /* loop over slices */
980dab86139SHong Zhang       if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
981dab86139SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
982dab86139SHong Zhang           if (row < (A->rmap->n & 0x07)) a->val[j] *= l[8*i+row];
983dab86139SHong Zhang         }
984dab86139SHong Zhang       } else {
985d4002b98SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
986d4002b98SHong Zhang           a->val[j] *= l[8*i+row];
987d4002b98SHong Zhang         }
988d4002b98SHong Zhang       }
989dab86139SHong Zhang     }
990d4002b98SHong Zhang     ierr = VecRestoreArrayRead(ll,&l);CHKERRQ(ierr);
991d4002b98SHong Zhang     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
992d4002b98SHong Zhang   }
993d4002b98SHong Zhang   if (rr) {
994d4002b98SHong Zhang     ierr = VecGetLocalSize(rr,&n);CHKERRQ(ierr);
995*2c71b3e2SJacob Faibussowitsch     PetscCheckFalse(n != A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length");
996d4002b98SHong Zhang     ierr = VecGetArrayRead(rr,&r);CHKERRQ(ierr);
997d4002b98SHong Zhang     for (i=0; i<a->totalslices; i++) { /* loop over slices */
998dab86139SHong Zhang       if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
999dab86139SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
1000dab86139SHong Zhang           if (row < (A->rmap->n & 0x07)) a->val[j] *= r[a->colidx[j]];
1001dab86139SHong Zhang         }
1002dab86139SHong Zhang       } else {
1003d4002b98SHong Zhang         for (j=a->sliidx[i]; j<a->sliidx[i+1]; j++) {
1004d4002b98SHong Zhang           a->val[j] *= r[a->colidx[j]];
1005d4002b98SHong Zhang         }
1006d4002b98SHong Zhang       }
1007dab86139SHong Zhang     }
1008d4002b98SHong Zhang     ierr = VecRestoreArrayRead(rr,&r);CHKERRQ(ierr);
1009d4002b98SHong Zhang     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
1010d4002b98SHong Zhang   }
1011d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr);
1012d4002b98SHong Zhang   PetscFunctionReturn(0);
1013d4002b98SHong Zhang }
1014d4002b98SHong Zhang 
1015d4002b98SHong Zhang PetscErrorCode MatGetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
1016d4002b98SHong Zhang {
1017d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1018d4002b98SHong Zhang   PetscInt    *cp,i,k,low,high,t,row,col,l;
1019d4002b98SHong Zhang   PetscInt    shift;
1020d4002b98SHong Zhang   MatScalar   *vp;
1021d4002b98SHong Zhang 
1022d4002b98SHong Zhang   PetscFunctionBegin;
102368aafef3SStefano Zampini   for (k=0; k<m; k++) { /* loop over requested rows */
1024d4002b98SHong Zhang     row = im[k];
1025d4002b98SHong Zhang     if (row<0) continue;
1026*2c71b3e2SJacob Faibussowitsch     PetscAssertFalse(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);
1027d4002b98SHong Zhang     shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */
1028d4002b98SHong Zhang     cp = a->colidx+shift; /* pointer to the row */
1029d4002b98SHong Zhang     vp = a->val+shift; /* pointer to the row */
103068aafef3SStefano Zampini     for (l=0; l<n; l++) { /* loop over requested columns */
1031d4002b98SHong Zhang       col = in[l];
1032d4002b98SHong Zhang       if (col<0) continue;
1033*2c71b3e2SJacob Faibussowitsch       PetscAssertFalse(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);
1034d4002b98SHong Zhang       high = a->rlen[row]; low = 0; /* assume unsorted */
1035d4002b98SHong Zhang       while (high-low > 5) {
1036d4002b98SHong Zhang         t = (low+high)/2;
1037d4002b98SHong Zhang         if (*(cp+t*8) > col) high = t;
1038d4002b98SHong Zhang         else low = t;
1039d4002b98SHong Zhang       }
1040d4002b98SHong Zhang       for (i=low; i<high; i++) {
1041d4002b98SHong Zhang         if (*(cp+8*i) > col) break;
1042d4002b98SHong Zhang         if (*(cp+8*i) == col) {
1043d4002b98SHong Zhang           *v++ = *(vp+8*i);
1044d4002b98SHong Zhang           goto finished;
1045d4002b98SHong Zhang         }
1046d4002b98SHong Zhang       }
1047d4002b98SHong Zhang       *v++ = 0.0;
1048d4002b98SHong Zhang     finished:;
1049d4002b98SHong Zhang     }
1050d4002b98SHong Zhang   }
1051d4002b98SHong Zhang   PetscFunctionReturn(0);
1052d4002b98SHong Zhang }
1053d4002b98SHong Zhang 
1054d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_ASCII(Mat A,PetscViewer viewer)
1055d4002b98SHong Zhang {
1056d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1057d4002b98SHong Zhang   PetscInt          i,j,m=A->rmap->n,shift;
1058d4002b98SHong Zhang   const char        *name;
1059d4002b98SHong Zhang   PetscViewerFormat format;
1060d4002b98SHong Zhang   PetscErrorCode    ierr;
1061d4002b98SHong Zhang 
1062d4002b98SHong Zhang   PetscFunctionBegin;
1063d4002b98SHong Zhang   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
1064d4002b98SHong Zhang   if (format == PETSC_VIEWER_ASCII_MATLAB) {
1065d4002b98SHong Zhang     PetscInt nofinalvalue = 0;
1066d4002b98SHong Zhang     /*
1067d4002b98SHong Zhang     if (m && ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != A->cmap->n-1))) {
1068d4002b98SHong Zhang       nofinalvalue = 1;
1069d4002b98SHong Zhang     }
1070d4002b98SHong Zhang     */
1071d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1072c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %" PetscInt_FMT " %" PetscInt_FMT " \n",m,A->cmap->n);CHKERRQ(ierr);
1073c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"%% Nonzeros = %" PetscInt_FMT " \n",a->nz);CHKERRQ(ierr);
1074d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1075c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",4);\n",a->nz+nofinalvalue);CHKERRQ(ierr);
1076d4002b98SHong Zhang #else
1077c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",3);\n",a->nz+nofinalvalue);CHKERRQ(ierr);
1078d4002b98SHong Zhang #endif
1079d4002b98SHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"zzz = [\n");CHKERRQ(ierr);
1080d4002b98SHong Zhang 
1081d4002b98SHong Zhang     for (i=0; i<m; i++) {
1082d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1083d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1084d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1085c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1086d4002b98SHong Zhang #else
1087c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)a->val[shift+8*j]);CHKERRQ(ierr);
1088d4002b98SHong Zhang #endif
1089d4002b98SHong Zhang       }
1090d4002b98SHong Zhang     }
1091d4002b98SHong Zhang     /*
1092d4002b98SHong Zhang     if (nofinalvalue) {
1093d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1094c0aa6a63SJacob Faibussowitsch       ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e %18.16e\n",m,A->cmap->n,0.,0.);CHKERRQ(ierr);
1095d4002b98SHong Zhang #else
1096c0aa6a63SJacob Faibussowitsch       ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e\n",m,A->cmap->n,0.0);CHKERRQ(ierr);
1097d4002b98SHong Zhang #endif
1098d4002b98SHong Zhang     }
1099d4002b98SHong Zhang     */
1100d4002b98SHong Zhang     ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
1101d4002b98SHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"];\n %s = spconvert(zzz);\n",name);CHKERRQ(ierr);
1102d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
1103d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO || format == PETSC_VIEWER_ASCII_INFO) {
1104d4002b98SHong Zhang     PetscFunctionReturn(0);
1105d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
1106d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1107d4002b98SHong Zhang     for (i=0; i<m; i++) {
1108c0aa6a63SJacob Faibussowitsch       ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr);
1109d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1110d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1111d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1112d4002b98SHong Zhang         if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) {
1113c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1114d4002b98SHong Zhang         } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) {
1115c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1116d4002b98SHong Zhang         } else if (PetscRealPart(a->val[shift+8*j]) != 0.0) {
1117c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]));CHKERRQ(ierr);
1118d4002b98SHong Zhang         }
1119d4002b98SHong Zhang #else
1120c0aa6a63SJacob Faibussowitsch         if (a->val[shift+8*j] != 0.0) {ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]);CHKERRQ(ierr);}
1121d4002b98SHong Zhang #endif
1122d4002b98SHong Zhang       }
1123d4002b98SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
1124d4002b98SHong Zhang     }
1125d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
1126d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_DENSE) {
1127d4002b98SHong Zhang     PetscInt    cnt=0,jcnt;
1128d4002b98SHong Zhang     PetscScalar value;
1129d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1130d4002b98SHong Zhang     PetscBool   realonly=PETSC_TRUE;
1131d4002b98SHong Zhang     for (i=0; i<a->sliidx[a->totalslices]; i++) {
1132d4002b98SHong Zhang       if (PetscImaginaryPart(a->val[i]) != 0.0) {
1133d4002b98SHong Zhang         realonly = PETSC_FALSE;
1134d4002b98SHong Zhang         break;
1135d4002b98SHong Zhang       }
1136d4002b98SHong Zhang     }
1137d4002b98SHong Zhang #endif
1138d4002b98SHong Zhang 
1139d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1140d4002b98SHong Zhang     for (i=0; i<m; i++) {
1141d4002b98SHong Zhang       jcnt = 0;
1142d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1143d4002b98SHong Zhang       for (j=0; j<A->cmap->n; j++) {
1144d4002b98SHong Zhang         if (jcnt < a->rlen[i] && j == a->colidx[shift+8*j]) {
1145d4002b98SHong Zhang           value = a->val[cnt++];
1146d4002b98SHong Zhang           jcnt++;
1147d4002b98SHong Zhang         } else {
1148d4002b98SHong Zhang           value = 0.0;
1149d4002b98SHong Zhang         }
1150d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1151d4002b98SHong Zhang         if (realonly) {
1152d4002b98SHong Zhang           ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)PetscRealPart(value));CHKERRQ(ierr);
1153d4002b98SHong Zhang         } else {
1154d4002b98SHong Zhang           ierr = PetscViewerASCIIPrintf(viewer," %7.5e+%7.5e i ",(double)PetscRealPart(value),(double)PetscImaginaryPart(value));CHKERRQ(ierr);
1155d4002b98SHong Zhang         }
1156d4002b98SHong Zhang #else
1157d4002b98SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)value);CHKERRQ(ierr);
1158d4002b98SHong Zhang #endif
1159d4002b98SHong Zhang       }
1160d4002b98SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
1161d4002b98SHong Zhang     }
1162d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
1163d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_MATRIXMARKET) {
1164d4002b98SHong Zhang     PetscInt fshift=1;
1165d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1166d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1167d4002b98SHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate complex general\n");CHKERRQ(ierr);
1168d4002b98SHong Zhang #else
1169d4002b98SHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate real general\n");CHKERRQ(ierr);
1170d4002b98SHong Zhang #endif
1171c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %" PetscInt_FMT "\n", m, A->cmap->n, a->nz);CHKERRQ(ierr);
1172d4002b98SHong Zhang     for (i=0; i<m; i++) {
1173d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1174d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1175d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1176c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1177d4002b98SHong Zhang #else
1178c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)a->val[shift+8*j]);CHKERRQ(ierr);
1179d4002b98SHong Zhang #endif
1180d4002b98SHong Zhang       }
1181d4002b98SHong Zhang     }
1182d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
118368aafef3SStefano Zampini   } else if (format == PETSC_VIEWER_NATIVE) {
118468aafef3SStefano Zampini     for (i=0; i<a->totalslices; i++) { /* loop over slices */
118568aafef3SStefano Zampini       PetscInt row;
1186c0aa6a63SJacob Faibussowitsch       ierr = PetscViewerASCIIPrintf(viewer,"slice %" PetscInt_FMT ": %" PetscInt_FMT " %" PetscInt_FMT "\n",i,a->sliidx[i],a->sliidx[i+1]);CHKERRQ(ierr);
118768aafef3SStefano Zampini       for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
118868aafef3SStefano Zampini #if defined(PETSC_USE_COMPLEX)
118968aafef3SStefano Zampini         if (PetscImaginaryPart(a->val[j]) > 0.0) {
1190c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g + %g i\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr);
119168aafef3SStefano Zampini         } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
1192c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g - %g i\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]),-(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr);
119368aafef3SStefano Zampini         } else {
1194c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr);
119568aafef3SStefano Zampini         }
119668aafef3SStefano Zampini #else
1197c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)a->val[j]);CHKERRQ(ierr);
119868aafef3SStefano Zampini #endif
119968aafef3SStefano Zampini       }
120068aafef3SStefano Zampini     }
1201d4002b98SHong Zhang   } else {
1202d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1203d4002b98SHong Zhang     if (A->factortype) {
1204d4002b98SHong Zhang       for (i=0; i<m; i++) {
1205d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07);
1206c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr);
1207d4002b98SHong Zhang         /* L part */
1208d4002b98SHong Zhang         for (j=shift; j<a->diag[i]; j+=8) {
1209d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1210d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0) {
1211c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr);
1212d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0) {
1213c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])));CHKERRQ(ierr);
1214d4002b98SHong Zhang           } else {
1215c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr);
1216d4002b98SHong Zhang           }
1217d4002b98SHong Zhang #else
1218c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]);CHKERRQ(ierr);
1219d4002b98SHong Zhang #endif
1220d4002b98SHong Zhang         }
1221d4002b98SHong Zhang         /* diagonal */
1222d4002b98SHong Zhang         j = a->diag[i];
1223d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1224d4002b98SHong Zhang         if (PetscImaginaryPart(a->val[j]) > 0.0) {
1225c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)PetscImaginaryPart(1.0/a->val[j]));CHKERRQ(ierr);
1226d4002b98SHong Zhang         } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
1227c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)(-PetscImaginaryPart(1.0/a->val[j])));CHKERRQ(ierr);
1228d4002b98SHong Zhang         } else {
1229c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]));CHKERRQ(ierr);
1230d4002b98SHong Zhang         }
1231d4002b98SHong Zhang #else
1232c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)(1.0/a->val[j]));CHKERRQ(ierr);
1233d4002b98SHong Zhang #endif
1234d4002b98SHong Zhang 
1235d4002b98SHong Zhang         /* U part */
1236d4002b98SHong Zhang         for (j=a->diag[i]+1; j<shift+8*a->rlen[i]; j+=8) {
1237d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1238d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[j]) > 0.0) {
1239c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr);
1240d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
1241c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])));CHKERRQ(ierr);
1242d4002b98SHong Zhang           } else {
1243c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr);
1244d4002b98SHong Zhang           }
1245d4002b98SHong Zhang #else
1246c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]);CHKERRQ(ierr);
1247d4002b98SHong Zhang #endif
1248d4002b98SHong Zhang         }
1249d4002b98SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
1250d4002b98SHong Zhang       }
1251d4002b98SHong Zhang     } else {
1252d4002b98SHong Zhang       for (i=0; i<m; i++) {
1253d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07);
1254c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr);
1255d4002b98SHong Zhang         for (j=0; j<a->rlen[i]; j++) {
1256d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1257d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[j]) > 0.0) {
1258c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1259d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
1260c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1261d4002b98SHong Zhang           } else {
1262c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]));CHKERRQ(ierr);
1263d4002b98SHong Zhang           }
1264d4002b98SHong Zhang #else
1265c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]);CHKERRQ(ierr);
1266d4002b98SHong Zhang #endif
1267d4002b98SHong Zhang         }
1268d4002b98SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
1269d4002b98SHong Zhang       }
1270d4002b98SHong Zhang     }
1271d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
1272d4002b98SHong Zhang   }
1273d4002b98SHong Zhang   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
1274d4002b98SHong Zhang   PetscFunctionReturn(0);
1275d4002b98SHong Zhang }
1276d4002b98SHong Zhang 
1277d4002b98SHong Zhang #include <petscdraw.h>
1278d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw_Zoom(PetscDraw draw,void *Aa)
1279d4002b98SHong Zhang {
1280d4002b98SHong Zhang   Mat               A=(Mat)Aa;
1281d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1282d4002b98SHong Zhang   PetscInt          i,j,m=A->rmap->n,shift;
1283d4002b98SHong Zhang   int               color;
1284d4002b98SHong Zhang   PetscReal         xl,yl,xr,yr,x_l,x_r,y_l,y_r;
1285d4002b98SHong Zhang   PetscViewer       viewer;
1286d4002b98SHong Zhang   PetscViewerFormat format;
1287d4002b98SHong Zhang   PetscErrorCode    ierr;
1288d4002b98SHong Zhang 
1289d4002b98SHong Zhang   PetscFunctionBegin;
1290d4002b98SHong Zhang   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
1291d4002b98SHong Zhang   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
1292d4002b98SHong Zhang   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
1293d4002b98SHong Zhang 
1294d4002b98SHong Zhang   /* loop over matrix elements drawing boxes */
1295d4002b98SHong Zhang 
1296d4002b98SHong Zhang   if (format != PETSC_VIEWER_DRAW_CONTOUR) {
1297d4002b98SHong Zhang     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
1298d4002b98SHong Zhang     /* Blue for negative, Cyan for zero and  Red for positive */
1299d4002b98SHong Zhang     color = PETSC_DRAW_BLUE;
1300d4002b98SHong Zhang     for (i=0; i<m; i++) {
1301d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1302d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1303d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1304d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1305d4002b98SHong Zhang         if (PetscRealPart(a->val[shift+8*j]) >=  0.) continue;
1306d4002b98SHong Zhang         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1307d4002b98SHong Zhang       }
1308d4002b98SHong Zhang     }
1309d4002b98SHong Zhang     color = PETSC_DRAW_CYAN;
1310d4002b98SHong Zhang     for (i=0; i<m; i++) {
1311d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1312d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1313d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1314d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1315d4002b98SHong Zhang         if (a->val[shift+8*j] !=  0.) continue;
1316d4002b98SHong Zhang         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1317d4002b98SHong Zhang       }
1318d4002b98SHong Zhang     }
1319d4002b98SHong Zhang     color = PETSC_DRAW_RED;
1320d4002b98SHong Zhang     for (i=0; i<m; i++) {
1321d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1322d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1323d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1324d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1325d4002b98SHong Zhang         if (PetscRealPart(a->val[shift+8*j]) <=  0.) continue;
1326d4002b98SHong Zhang         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1327d4002b98SHong Zhang       }
1328d4002b98SHong Zhang     }
1329d4002b98SHong Zhang     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1330d4002b98SHong Zhang   } else {
1331d4002b98SHong Zhang     /* use contour shading to indicate magnitude of values */
1332d4002b98SHong Zhang     /* first determine max of all nonzero values */
1333d4002b98SHong Zhang     PetscReal minv=0.0,maxv=0.0;
1334d4002b98SHong Zhang     PetscInt  count=0;
1335d4002b98SHong Zhang     PetscDraw popup;
1336d4002b98SHong Zhang     for (i=0; i<a->sliidx[a->totalslices]; i++) {
1337d4002b98SHong Zhang       if (PetscAbsScalar(a->val[i]) > maxv) maxv = PetscAbsScalar(a->val[i]);
1338d4002b98SHong Zhang     }
1339d4002b98SHong Zhang     if (minv >= maxv) maxv = minv + PETSC_SMALL;
1340d4002b98SHong Zhang     ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr);
1341d4002b98SHong Zhang     ierr = PetscDrawScalePopup(popup,minv,maxv);CHKERRQ(ierr);
1342d4002b98SHong Zhang 
1343d4002b98SHong Zhang     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
1344d4002b98SHong Zhang     for (i=0; i<m; i++) {
1345d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1346d4002b98SHong Zhang       y_l = m - i - 1.0;
1347d4002b98SHong Zhang       y_r = y_l + 1.0;
1348d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1349d4002b98SHong Zhang         x_l = a->colidx[shift+j*8];
1350d4002b98SHong Zhang         x_r = x_l + 1.0;
1351d4002b98SHong Zhang         color = PetscDrawRealToColor(PetscAbsScalar(a->val[count]),minv,maxv);
1352d4002b98SHong Zhang         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1353d4002b98SHong Zhang         count++;
1354d4002b98SHong Zhang       }
1355d4002b98SHong Zhang     }
1356d4002b98SHong Zhang     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1357d4002b98SHong Zhang   }
1358d4002b98SHong Zhang   PetscFunctionReturn(0);
1359d4002b98SHong Zhang }
1360d4002b98SHong Zhang 
1361d4002b98SHong Zhang #include <petscdraw.h>
1362d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw(Mat A,PetscViewer viewer)
1363d4002b98SHong Zhang {
1364d4002b98SHong Zhang   PetscDraw      draw;
1365d4002b98SHong Zhang   PetscReal      xr,yr,xl,yl,h,w;
1366d4002b98SHong Zhang   PetscBool      isnull;
1367d4002b98SHong Zhang   PetscErrorCode ierr;
1368d4002b98SHong Zhang 
1369d4002b98SHong Zhang   PetscFunctionBegin;
1370d4002b98SHong Zhang   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
1371d4002b98SHong Zhang   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
1372d4002b98SHong Zhang   if (isnull) PetscFunctionReturn(0);
1373d4002b98SHong Zhang 
1374d4002b98SHong Zhang   xr   = A->cmap->n; yr  = A->rmap->n; h = yr/10.0; w = xr/10.0;
1375d4002b98SHong Zhang   xr  += w;          yr += h;         xl = -w;     yl = -h;
1376d4002b98SHong Zhang   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
1377d4002b98SHong Zhang   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
1378d4002b98SHong Zhang   ierr = PetscDrawZoom(draw,MatView_SeqSELL_Draw_Zoom,A);CHKERRQ(ierr);
1379d4002b98SHong Zhang   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
1380d4002b98SHong Zhang   ierr = PetscDrawSave(draw);CHKERRQ(ierr);
1381d4002b98SHong Zhang   PetscFunctionReturn(0);
1382d4002b98SHong Zhang }
1383d4002b98SHong Zhang 
1384d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL(Mat A,PetscViewer viewer)
1385d4002b98SHong Zhang {
1386d4002b98SHong Zhang   PetscBool      iascii,isbinary,isdraw;
1387d4002b98SHong Zhang   PetscErrorCode ierr;
1388d4002b98SHong Zhang 
1389d4002b98SHong Zhang   PetscFunctionBegin;
1390d4002b98SHong Zhang   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
1391d4002b98SHong Zhang   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
1392d4002b98SHong Zhang   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
1393d4002b98SHong Zhang   if (iascii) {
1394d4002b98SHong Zhang     ierr = MatView_SeqSELL_ASCII(A,viewer);CHKERRQ(ierr);
1395d4002b98SHong Zhang   } else if (isbinary) {
1396d4002b98SHong Zhang     /* ierr = MatView_SeqSELL_Binary(A,viewer);CHKERRQ(ierr); */
1397d4002b98SHong Zhang   } else if (isdraw) {
1398d4002b98SHong Zhang     ierr = MatView_SeqSELL_Draw(A,viewer);CHKERRQ(ierr);
1399d4002b98SHong Zhang   }
1400d4002b98SHong Zhang   PetscFunctionReturn(0);
1401d4002b98SHong Zhang }
1402d4002b98SHong Zhang 
1403d4002b98SHong Zhang PetscErrorCode MatAssemblyEnd_SeqSELL(Mat A,MatAssemblyType mode)
1404d4002b98SHong Zhang {
1405d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1406d4002b98SHong Zhang   PetscInt       i,shift,row_in_slice,row,nrow,*cp,lastcol,j,k;
1407d4002b98SHong Zhang   MatScalar      *vp;
1408d4002b98SHong Zhang   PetscErrorCode ierr;
1409d4002b98SHong Zhang 
1410d4002b98SHong Zhang   PetscFunctionBegin;
1411d4002b98SHong Zhang   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
1412d4002b98SHong Zhang   /* To do: compress out the unused elements */
1413d4002b98SHong Zhang   ierr = MatMarkDiagonal_SeqSELL(A);CHKERRQ(ierr);
14147d3de750SJacob Faibussowitsch   ierr = 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);CHKERRQ(ierr);
14157d3de750SJacob Faibussowitsch   ierr = PetscInfo(A,"Number of mallocs during MatSetValues() is %" PetscInt_FMT "\n",a->reallocs);CHKERRQ(ierr);
14167d3de750SJacob Faibussowitsch   ierr = PetscInfo(A,"Maximum nonzeros in any row is %" PetscInt_FMT "\n",a->rlenmax);CHKERRQ(ierr);
1417d4002b98SHong 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 */
1418d4002b98SHong Zhang   for (i=0; i<a->totalslices; ++i) {
1419d4002b98SHong Zhang     shift = a->sliidx[i];    /* starting index of the slice */
1420d4002b98SHong Zhang     cp    = a->colidx+shift; /* pointer to the column indices of the slice */
1421d4002b98SHong Zhang     vp    = a->val+shift;    /* pointer to the nonzero values of the slice */
1422d4002b98SHong Zhang     for (row_in_slice=0; row_in_slice<8; ++row_in_slice) { /* loop over rows in the slice */
1423d4002b98SHong Zhang       row  = 8*i + row_in_slice;
1424d4002b98SHong Zhang       nrow = a->rlen[row]; /* number of nonzeros in row */
1425d4002b98SHong Zhang       /*
1426d4002b98SHong Zhang         Search for the nearest nonzero. Normally setting the index to zero may cause extra communication.
1427d4002b98SHong Zhang         But if the entire slice are empty, it is fine to use 0 since the index will not be loaded.
1428d4002b98SHong Zhang       */
1429d4002b98SHong Zhang       lastcol = 0;
1430d4002b98SHong Zhang       if (nrow>0) { /* nonempty row */
1431d4002b98SHong Zhang         lastcol = cp[8*(nrow-1)+row_in_slice]; /* use the index from the last nonzero at current row */
1432d4002b98SHong Zhang       } else if (!row_in_slice) { /* first row of the currect slice is empty */
1433d4002b98SHong Zhang         for (j=1;j<8;j++) {
1434d4002b98SHong Zhang           if (a->rlen[8*i+j]) {
1435d4002b98SHong Zhang             lastcol = cp[j];
1436d4002b98SHong Zhang             break;
1437d4002b98SHong Zhang           }
1438d4002b98SHong Zhang         }
1439d4002b98SHong Zhang       } else {
1440d4002b98SHong Zhang         if (a->sliidx[i+1] != shift) lastcol = cp[row_in_slice-1]; /* use the index from the previous row */
1441d4002b98SHong Zhang       }
1442d4002b98SHong Zhang 
1443d4002b98SHong Zhang       for (k=nrow; k<(a->sliidx[i+1]-shift)/8; ++k) {
1444d4002b98SHong Zhang         cp[8*k+row_in_slice] = lastcol;
1445d4002b98SHong Zhang         vp[8*k+row_in_slice] = (MatScalar)0;
1446d4002b98SHong Zhang       }
1447d4002b98SHong Zhang     }
1448d4002b98SHong Zhang   }
1449d4002b98SHong Zhang 
1450d4002b98SHong Zhang   A->info.mallocs += a->reallocs;
1451d4002b98SHong Zhang   a->reallocs      = 0;
1452d4002b98SHong Zhang 
1453d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr);
1454d4002b98SHong Zhang   PetscFunctionReturn(0);
1455d4002b98SHong Zhang }
1456d4002b98SHong Zhang 
1457d4002b98SHong Zhang PetscErrorCode MatGetInfo_SeqSELL(Mat A,MatInfoType flag,MatInfo *info)
1458d4002b98SHong Zhang {
1459d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1460d4002b98SHong Zhang 
1461d4002b98SHong Zhang   PetscFunctionBegin;
1462d4002b98SHong Zhang   info->block_size   = 1.0;
14633966268fSBarry Smith   info->nz_allocated = a->maxallocmat;
14643966268fSBarry Smith   info->nz_used      = a->sliidx[a->totalslices]; /* include padding zeros */
14653966268fSBarry Smith   info->nz_unneeded  = (a->maxallocmat-a->sliidx[a->totalslices]);
14663966268fSBarry Smith   info->assemblies   = A->num_ass;
14673966268fSBarry Smith   info->mallocs      = A->info.mallocs;
1468d4002b98SHong Zhang   info->memory       = ((PetscObject)A)->mem;
1469d4002b98SHong Zhang   if (A->factortype) {
1470d4002b98SHong Zhang     info->fill_ratio_given  = A->info.fill_ratio_given;
1471d4002b98SHong Zhang     info->fill_ratio_needed = A->info.fill_ratio_needed;
1472d4002b98SHong Zhang     info->factor_mallocs    = A->info.factor_mallocs;
1473d4002b98SHong Zhang   } else {
1474d4002b98SHong Zhang     info->fill_ratio_given  = 0;
1475d4002b98SHong Zhang     info->fill_ratio_needed = 0;
1476d4002b98SHong Zhang     info->factor_mallocs    = 0;
1477d4002b98SHong Zhang   }
1478d4002b98SHong Zhang   PetscFunctionReturn(0);
1479d4002b98SHong Zhang }
1480d4002b98SHong Zhang 
1481d4002b98SHong Zhang PetscErrorCode MatSetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
1482d4002b98SHong Zhang {
1483d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1484d4002b98SHong Zhang   PetscInt       shift,i,k,l,low,high,t,ii,row,col,nrow;
1485d4002b98SHong Zhang   PetscInt       *cp,nonew=a->nonew,lastcol=-1;
1486d4002b98SHong Zhang   MatScalar      *vp,value;
1487d4002b98SHong Zhang   PetscErrorCode ierr;
1488d4002b98SHong Zhang 
1489d4002b98SHong Zhang   PetscFunctionBegin;
1490d4002b98SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
1491d4002b98SHong Zhang     row = im[k];
1492d4002b98SHong Zhang     if (row < 0) continue;
1493*2c71b3e2SJacob Faibussowitsch     PetscAssertFalse(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);
1494d4002b98SHong Zhang     shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */
1495d4002b98SHong Zhang     cp    = a->colidx+shift; /* pointer to the row */
1496d4002b98SHong Zhang     vp    = a->val+shift; /* pointer to the row */
1497d4002b98SHong Zhang     nrow  = a->rlen[row];
1498d4002b98SHong Zhang     low   = 0;
1499d4002b98SHong Zhang     high  = nrow;
1500d4002b98SHong Zhang 
1501d4002b98SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
1502d4002b98SHong Zhang       col = in[l];
1503d4002b98SHong Zhang       if (col<0) continue;
1504*2c71b3e2SJacob Faibussowitsch       PetscAssertFalse(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);
1505d4002b98SHong Zhang       if (a->roworiented) {
1506d4002b98SHong Zhang         value = v[l+k*n];
1507d4002b98SHong Zhang       } else {
1508d4002b98SHong Zhang         value = v[k+l*m];
1509d4002b98SHong Zhang       }
1510d4002b98SHong Zhang       if ((value == 0.0 && a->ignorezeroentries) && (is == ADD_VALUES)) continue;
1511d4002b98SHong Zhang 
1512ed73aabaSBarry Smith       /* search in this row for the specified column, i indicates the column to be set */
1513d4002b98SHong Zhang       if (col <= lastcol) low = 0;
1514d4002b98SHong Zhang       else high = nrow;
1515d4002b98SHong Zhang       lastcol = col;
1516d4002b98SHong Zhang       while (high-low > 5) {
1517d4002b98SHong Zhang         t = (low+high)/2;
1518d4002b98SHong Zhang         if (*(cp+t*8) > col) high = t;
1519d4002b98SHong Zhang         else low = t;
1520d4002b98SHong Zhang       }
1521d4002b98SHong Zhang       for (i=low; i<high; i++) {
1522d4002b98SHong Zhang         if (*(cp+i*8) > col) break;
1523d4002b98SHong Zhang         if (*(cp+i*8) == col) {
1524d4002b98SHong Zhang           if (is == ADD_VALUES) *(vp+i*8) += value;
1525d4002b98SHong Zhang           else *(vp+i*8) = value;
1526d4002b98SHong Zhang           low = i + 1;
1527d4002b98SHong Zhang           goto noinsert;
1528d4002b98SHong Zhang         }
1529d4002b98SHong Zhang       }
1530d4002b98SHong Zhang       if (value == 0.0 && a->ignorezeroentries) goto noinsert;
1531d4002b98SHong Zhang       if (nonew == 1) goto noinsert;
1532*2c71b3e2SJacob Faibussowitsch       PetscCheckFalse(nonew == -1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%" PetscInt_FMT ", %" PetscInt_FMT ") in the matrix", row, col);
1533d4002b98SHong Zhang       /* If the current row length exceeds the slice width (e.g. nrow==slice_width), allocate a new space, otherwise do nothing */
1534d4002b98SHong Zhang       MatSeqXSELLReallocateSELL(A,A->rmap->n,1,nrow,a->sliidx,row/8,row,col,a->colidx,a->val,cp,vp,nonew,MatScalar);
1535d4002b98SHong Zhang       /* add the new nonzero to the high position, shift the remaining elements in current row to the right by one slot */
1536d4002b98SHong Zhang       for (ii=nrow-1; ii>=i; ii--) {
1537d4002b98SHong Zhang         *(cp+(ii+1)*8) = *(cp+ii*8);
1538d4002b98SHong Zhang         *(vp+(ii+1)*8) = *(vp+ii*8);
1539d4002b98SHong Zhang       }
1540d4002b98SHong Zhang       a->rlen[row]++;
1541d4002b98SHong Zhang       *(cp+i*8) = col;
1542d4002b98SHong Zhang       *(vp+i*8) = value;
1543d4002b98SHong Zhang       a->nz++;
1544d4002b98SHong Zhang       A->nonzerostate++;
1545d4002b98SHong Zhang       low = i+1; high++; nrow++;
1546d4002b98SHong Zhang noinsert:;
1547d4002b98SHong Zhang     }
1548d4002b98SHong Zhang     a->rlen[row] = nrow;
1549d4002b98SHong Zhang   }
1550d4002b98SHong Zhang   PetscFunctionReturn(0);
1551d4002b98SHong Zhang }
1552d4002b98SHong Zhang 
1553d4002b98SHong Zhang PetscErrorCode MatCopy_SeqSELL(Mat A,Mat B,MatStructure str)
1554d4002b98SHong Zhang {
1555d4002b98SHong Zhang   PetscErrorCode ierr;
1556d4002b98SHong Zhang 
1557d4002b98SHong Zhang   PetscFunctionBegin;
1558d4002b98SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
1559d4002b98SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
1560d4002b98SHong Zhang     Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1561d4002b98SHong Zhang     Mat_SeqSELL *b=(Mat_SeqSELL*)B->data;
1562d4002b98SHong Zhang 
1563*2c71b3e2SJacob Faibussowitsch     PetscCheckFalse(a->sliidx[a->totalslices] != b->sliidx[b->totalslices],PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different");
1564580bdb30SBarry Smith     ierr = PetscArraycpy(b->val,a->val,a->sliidx[a->totalslices]);CHKERRQ(ierr);
1565d4002b98SHong Zhang   } else {
1566d4002b98SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1567d4002b98SHong Zhang   }
1568d4002b98SHong Zhang   PetscFunctionReturn(0);
1569d4002b98SHong Zhang }
1570d4002b98SHong Zhang 
1571d4002b98SHong Zhang PetscErrorCode MatSetUp_SeqSELL(Mat A)
1572d4002b98SHong Zhang {
1573d4002b98SHong Zhang   PetscErrorCode ierr;
1574d4002b98SHong Zhang 
1575d4002b98SHong Zhang   PetscFunctionBegin;
1576f4259b30SLisandro Dalcin   ierr = MatSeqSELLSetPreallocation(A,PETSC_DEFAULT,NULL);CHKERRQ(ierr);
1577d4002b98SHong Zhang   PetscFunctionReturn(0);
1578d4002b98SHong Zhang }
1579d4002b98SHong Zhang 
1580d4002b98SHong Zhang PetscErrorCode MatSeqSELLGetArray_SeqSELL(Mat A,PetscScalar *array[])
1581d4002b98SHong Zhang {
1582d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1583d4002b98SHong Zhang 
1584d4002b98SHong Zhang   PetscFunctionBegin;
1585d4002b98SHong Zhang   *array = a->val;
1586d4002b98SHong Zhang   PetscFunctionReturn(0);
1587d4002b98SHong Zhang }
1588d4002b98SHong Zhang 
1589d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray_SeqSELL(Mat A,PetscScalar *array[])
1590d4002b98SHong Zhang {
1591d4002b98SHong Zhang   PetscFunctionBegin;
1592d4002b98SHong Zhang   PetscFunctionReturn(0);
1593d4002b98SHong Zhang }
1594d4002b98SHong Zhang 
1595d4002b98SHong Zhang PetscErrorCode MatRealPart_SeqSELL(Mat A)
1596d4002b98SHong Zhang {
1597d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1598d4002b98SHong Zhang   PetscInt    i;
1599d4002b98SHong Zhang   MatScalar   *aval=a->val;
1600d4002b98SHong Zhang 
1601d4002b98SHong Zhang   PetscFunctionBegin;
1602d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i]=PetscRealPart(aval[i]);
1603d4002b98SHong Zhang   PetscFunctionReturn(0);
1604d4002b98SHong Zhang }
1605d4002b98SHong Zhang 
1606d4002b98SHong Zhang PetscErrorCode MatImaginaryPart_SeqSELL(Mat A)
1607d4002b98SHong Zhang {
1608d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1609d4002b98SHong Zhang   PetscInt       i;
1610d4002b98SHong Zhang   MatScalar      *aval=a->val;
1611d4002b98SHong Zhang   PetscErrorCode ierr;
1612d4002b98SHong Zhang 
1613d4002b98SHong Zhang   PetscFunctionBegin;
1614d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i] = PetscImaginaryPart(aval[i]);
1615d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr);
1616d4002b98SHong Zhang   PetscFunctionReturn(0);
1617d4002b98SHong Zhang }
1618d4002b98SHong Zhang 
1619d4002b98SHong Zhang PetscErrorCode MatScale_SeqSELL(Mat inA,PetscScalar alpha)
1620d4002b98SHong Zhang {
1621d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)inA->data;
1622d4002b98SHong Zhang   MatScalar      *aval=a->val;
1623d4002b98SHong Zhang   PetscScalar    oalpha=alpha;
1624d4002b98SHong Zhang   PetscBLASInt   one=1,size;
1625d4002b98SHong Zhang   PetscErrorCode ierr;
1626d4002b98SHong Zhang 
1627d4002b98SHong Zhang   PetscFunctionBegin;
1628d4002b98SHong Zhang   ierr = PetscBLASIntCast(a->sliidx[a->totalslices],&size);CHKERRQ(ierr);
1629d4002b98SHong Zhang   PetscStackCallBLAS("BLASscal",BLASscal_(&size,&oalpha,aval,&one));
1630d4002b98SHong Zhang   ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
1631d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(inA);CHKERRQ(ierr);
1632d4002b98SHong Zhang   PetscFunctionReturn(0);
1633d4002b98SHong Zhang }
1634d4002b98SHong Zhang 
1635d4002b98SHong Zhang PetscErrorCode MatShift_SeqSELL(Mat Y,PetscScalar a)
1636d4002b98SHong Zhang {
1637d4002b98SHong Zhang   Mat_SeqSELL    *y=(Mat_SeqSELL*)Y->data;
1638d4002b98SHong Zhang   PetscErrorCode ierr;
1639d4002b98SHong Zhang 
1640d4002b98SHong Zhang   PetscFunctionBegin;
1641d4002b98SHong Zhang   if (!Y->preallocated || !y->nz) {
1642d4002b98SHong Zhang     ierr = MatSeqSELLSetPreallocation(Y,1,NULL);CHKERRQ(ierr);
1643d4002b98SHong Zhang   }
1644d4002b98SHong Zhang   ierr = MatShift_Basic(Y,a);CHKERRQ(ierr);
1645d4002b98SHong Zhang   PetscFunctionReturn(0);
1646d4002b98SHong Zhang }
1647d4002b98SHong Zhang 
1648d4002b98SHong Zhang PetscErrorCode MatSOR_SeqSELL(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
1649d4002b98SHong Zhang {
1650d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1651d4002b98SHong Zhang   PetscScalar       *x,sum,*t;
1652f4259b30SLisandro Dalcin   const MatScalar   *idiag=NULL,*mdiag;
1653d4002b98SHong Zhang   const PetscScalar *b,*xb;
1654d4002b98SHong Zhang   PetscInt          n,m=A->rmap->n,i,j,shift;
1655d4002b98SHong Zhang   const PetscInt    *diag;
1656d4002b98SHong Zhang   PetscErrorCode    ierr;
1657d4002b98SHong Zhang 
1658d4002b98SHong Zhang   PetscFunctionBegin;
1659d4002b98SHong Zhang   its = its*lits;
1660d4002b98SHong Zhang 
1661d4002b98SHong Zhang   if (fshift != a->fshift || omega != a->omega) a->idiagvalid = PETSC_FALSE; /* must recompute idiag[] */
1662d4002b98SHong Zhang   if (!a->idiagvalid) {ierr = MatInvertDiagonal_SeqSELL(A,omega,fshift);CHKERRQ(ierr);}
1663d4002b98SHong Zhang   a->fshift = fshift;
1664d4002b98SHong Zhang   a->omega  = omega;
1665d4002b98SHong Zhang 
1666d4002b98SHong Zhang   diag  = a->diag;
1667d4002b98SHong Zhang   t     = a->ssor_work;
1668d4002b98SHong Zhang   idiag = a->idiag;
1669d4002b98SHong Zhang   mdiag = a->mdiag;
1670d4002b98SHong Zhang 
1671d4002b98SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1672d4002b98SHong Zhang   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1673d4002b98SHong Zhang   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
1674*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(flag == SOR_APPLY_UPPER,PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_UPPER is not implemented");
1675*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(flag == SOR_APPLY_LOWER,PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_LOWER is not implemented");
1676*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(flag & SOR_EISENSTAT,PETSC_COMM_SELF,PETSC_ERR_SUP,"No support yet for Eisenstat");
1677d4002b98SHong Zhang 
1678d4002b98SHong Zhang   if (flag & SOR_ZERO_INITIAL_GUESS) {
1679d4002b98SHong Zhang     if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) {
1680d4002b98SHong Zhang       for (i=0; i<m; i++) {
1681d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1682d4002b98SHong Zhang         sum   = b[i];
1683d4002b98SHong Zhang         n     = (diag[i]-shift)/8;
1684d4002b98SHong Zhang         for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1685d4002b98SHong Zhang         t[i]  = sum;
1686d4002b98SHong Zhang         x[i]  = sum*idiag[i];
1687d4002b98SHong Zhang       }
1688d4002b98SHong Zhang       xb   = t;
1689d4002b98SHong Zhang       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
1690d4002b98SHong Zhang     } else xb = b;
1691d4002b98SHong Zhang     if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) {
1692d4002b98SHong Zhang       for (i=m-1; i>=0; i--) {
1693d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1694d4002b98SHong Zhang         sum   = xb[i];
1695d4002b98SHong Zhang         n     = a->rlen[i]-(diag[i]-shift)/8-1;
1696d4002b98SHong Zhang         for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1697d4002b98SHong Zhang         if (xb == b) {
1698d4002b98SHong Zhang           x[i] = sum*idiag[i];
1699d4002b98SHong Zhang         } else {
1700d4002b98SHong Zhang           x[i] = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1701d4002b98SHong Zhang         }
1702d4002b98SHong Zhang       }
1703d4002b98SHong Zhang       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */
1704d4002b98SHong Zhang     }
1705d4002b98SHong Zhang     its--;
1706d4002b98SHong Zhang   }
1707d4002b98SHong Zhang   while (its--) {
1708d4002b98SHong Zhang     if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) {
1709d4002b98SHong Zhang       for (i=0; i<m; i++) {
1710d4002b98SHong Zhang         /* lower */
1711d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1712d4002b98SHong Zhang         sum   = b[i];
1713d4002b98SHong Zhang         n     = (diag[i]-shift)/8;
1714d4002b98SHong Zhang         for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1715d4002b98SHong Zhang         t[i]  = sum;             /* save application of the lower-triangular part */
1716d4002b98SHong Zhang         /* upper */
1717d4002b98SHong Zhang         n     = a->rlen[i]-(diag[i]-shift)/8-1;
1718d4002b98SHong Zhang         for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1719d4002b98SHong Zhang         x[i]  = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1720d4002b98SHong Zhang       }
1721d4002b98SHong Zhang       xb   = t;
1722d4002b98SHong Zhang       ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1723d4002b98SHong Zhang     } else xb = b;
1724d4002b98SHong Zhang     if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) {
1725d4002b98SHong Zhang       for (i=m-1; i>=0; i--) {
1726d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1727d4002b98SHong Zhang         sum = xb[i];
1728d4002b98SHong Zhang         if (xb == b) {
1729d4002b98SHong Zhang           /* whole matrix (no checkpointing available) */
1730d4002b98SHong Zhang           n     = a->rlen[i];
1731d4002b98SHong Zhang           for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1732d4002b98SHong Zhang           x[i] = (1.-omega)*x[i]+(sum+mdiag[i]*x[i])*idiag[i];
1733d4002b98SHong Zhang         } else { /* lower-triangular part has been saved, so only apply upper-triangular */
1734d4002b98SHong Zhang           n     = a->rlen[i]-(diag[i]-shift)/8-1;
1735d4002b98SHong Zhang           for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1736d4002b98SHong Zhang           x[i]  = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1737d4002b98SHong Zhang         }
1738d4002b98SHong Zhang       }
1739d4002b98SHong Zhang       if (xb == b) {
1740d4002b98SHong Zhang         ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1741d4002b98SHong Zhang       } else {
1742d4002b98SHong Zhang         ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */
1743d4002b98SHong Zhang       }
1744d4002b98SHong Zhang     }
1745d4002b98SHong Zhang   }
1746d4002b98SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1747d4002b98SHong Zhang   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1748d4002b98SHong Zhang   PetscFunctionReturn(0);
1749d4002b98SHong Zhang }
1750d4002b98SHong Zhang 
1751d4002b98SHong Zhang /* -------------------------------------------------------------------*/
1752d4002b98SHong Zhang static struct _MatOps MatOps_Values = {MatSetValues_SeqSELL,
17536108893eSStefano Zampini                                        MatGetRow_SeqSELL,
17546108893eSStefano Zampini                                        MatRestoreRow_SeqSELL,
1755d4002b98SHong Zhang                                        MatMult_SeqSELL,
1756d4002b98SHong Zhang                                /* 4*/  MatMultAdd_SeqSELL,
1757d4002b98SHong Zhang                                        MatMultTranspose_SeqSELL,
1758d4002b98SHong Zhang                                        MatMultTransposeAdd_SeqSELL,
1759f4259b30SLisandro Dalcin                                        NULL,
1760f4259b30SLisandro Dalcin                                        NULL,
1761f4259b30SLisandro Dalcin                                        NULL,
1762f4259b30SLisandro Dalcin                                /* 10*/ NULL,
1763f4259b30SLisandro Dalcin                                        NULL,
1764f4259b30SLisandro Dalcin                                        NULL,
1765d4002b98SHong Zhang                                        MatSOR_SeqSELL,
1766f4259b30SLisandro Dalcin                                        NULL,
1767d4002b98SHong Zhang                                /* 15*/ MatGetInfo_SeqSELL,
1768d4002b98SHong Zhang                                        MatEqual_SeqSELL,
1769d4002b98SHong Zhang                                        MatGetDiagonal_SeqSELL,
1770d4002b98SHong Zhang                                        MatDiagonalScale_SeqSELL,
1771f4259b30SLisandro Dalcin                                        NULL,
1772f4259b30SLisandro Dalcin                                /* 20*/ NULL,
1773d4002b98SHong Zhang                                        MatAssemblyEnd_SeqSELL,
1774d4002b98SHong Zhang                                        MatSetOption_SeqSELL,
1775d4002b98SHong Zhang                                        MatZeroEntries_SeqSELL,
1776f4259b30SLisandro Dalcin                                /* 24*/ NULL,
1777f4259b30SLisandro Dalcin                                        NULL,
1778f4259b30SLisandro Dalcin                                        NULL,
1779f4259b30SLisandro Dalcin                                        NULL,
1780f4259b30SLisandro Dalcin                                        NULL,
1781d4002b98SHong Zhang                                /* 29*/ MatSetUp_SeqSELL,
1782f4259b30SLisandro Dalcin                                        NULL,
1783f4259b30SLisandro Dalcin                                        NULL,
1784f4259b30SLisandro Dalcin                                        NULL,
1785f4259b30SLisandro Dalcin                                        NULL,
1786d4002b98SHong Zhang                                /* 34*/ MatDuplicate_SeqSELL,
1787f4259b30SLisandro Dalcin                                        NULL,
1788f4259b30SLisandro Dalcin                                        NULL,
1789f4259b30SLisandro Dalcin                                        NULL,
1790f4259b30SLisandro Dalcin                                        NULL,
1791f4259b30SLisandro Dalcin                                /* 39*/ NULL,
1792f4259b30SLisandro Dalcin                                        NULL,
1793f4259b30SLisandro Dalcin                                        NULL,
1794d4002b98SHong Zhang                                        MatGetValues_SeqSELL,
1795d4002b98SHong Zhang                                        MatCopy_SeqSELL,
1796f4259b30SLisandro Dalcin                                /* 44*/ NULL,
1797d4002b98SHong Zhang                                        MatScale_SeqSELL,
1798d4002b98SHong Zhang                                        MatShift_SeqSELL,
1799f4259b30SLisandro Dalcin                                        NULL,
1800f4259b30SLisandro Dalcin                                        NULL,
1801f4259b30SLisandro Dalcin                                /* 49*/ NULL,
1802f4259b30SLisandro Dalcin                                        NULL,
1803f4259b30SLisandro Dalcin                                        NULL,
1804f4259b30SLisandro Dalcin                                        NULL,
1805f4259b30SLisandro Dalcin                                        NULL,
1806d4002b98SHong Zhang                                /* 54*/ MatFDColoringCreate_SeqXAIJ,
1807f4259b30SLisandro Dalcin                                        NULL,
1808f4259b30SLisandro Dalcin                                        NULL,
1809f4259b30SLisandro Dalcin                                        NULL,
1810f4259b30SLisandro Dalcin                                        NULL,
1811f4259b30SLisandro Dalcin                                /* 59*/ NULL,
1812d4002b98SHong Zhang                                        MatDestroy_SeqSELL,
1813d4002b98SHong Zhang                                        MatView_SeqSELL,
1814f4259b30SLisandro Dalcin                                        NULL,
1815f4259b30SLisandro Dalcin                                        NULL,
1816f4259b30SLisandro Dalcin                                /* 64*/ NULL,
1817f4259b30SLisandro Dalcin                                        NULL,
1818f4259b30SLisandro Dalcin                                        NULL,
1819f4259b30SLisandro Dalcin                                        NULL,
1820f4259b30SLisandro Dalcin                                        NULL,
1821f4259b30SLisandro Dalcin                                /* 69*/ NULL,
1822f4259b30SLisandro Dalcin                                        NULL,
1823f4259b30SLisandro Dalcin                                        NULL,
1824f4259b30SLisandro Dalcin                                        NULL,
1825f4259b30SLisandro Dalcin                                        NULL,
1826f4259b30SLisandro Dalcin                                /* 74*/ NULL,
1827d4002b98SHong Zhang                                        MatFDColoringApply_AIJ, /* reuse the FDColoring function for AIJ */
1828f4259b30SLisandro Dalcin                                        NULL,
1829f4259b30SLisandro Dalcin                                        NULL,
1830f4259b30SLisandro Dalcin                                        NULL,
1831f4259b30SLisandro Dalcin                                /* 79*/ NULL,
1832f4259b30SLisandro Dalcin                                        NULL,
1833f4259b30SLisandro Dalcin                                        NULL,
1834f4259b30SLisandro Dalcin                                        NULL,
1835f4259b30SLisandro Dalcin                                        NULL,
1836f4259b30SLisandro Dalcin                                /* 84*/ NULL,
1837f4259b30SLisandro Dalcin                                        NULL,
1838f4259b30SLisandro Dalcin                                        NULL,
1839f4259b30SLisandro Dalcin                                        NULL,
1840f4259b30SLisandro Dalcin                                        NULL,
1841f4259b30SLisandro Dalcin                                /* 89*/ NULL,
1842f4259b30SLisandro Dalcin                                        NULL,
1843f4259b30SLisandro Dalcin                                        NULL,
1844f4259b30SLisandro Dalcin                                        NULL,
1845f4259b30SLisandro Dalcin                                        NULL,
1846f4259b30SLisandro Dalcin                                /* 94*/ NULL,
1847f4259b30SLisandro Dalcin                                        NULL,
1848f4259b30SLisandro Dalcin                                        NULL,
1849f4259b30SLisandro Dalcin                                        NULL,
1850f4259b30SLisandro Dalcin                                        NULL,
1851f4259b30SLisandro Dalcin                                /* 99*/ NULL,
1852f4259b30SLisandro Dalcin                                        NULL,
1853f4259b30SLisandro Dalcin                                        NULL,
1854d4002b98SHong Zhang                                        MatConjugate_SeqSELL,
1855f4259b30SLisandro Dalcin                                        NULL,
1856f4259b30SLisandro Dalcin                                /*104*/ NULL,
1857f4259b30SLisandro Dalcin                                        NULL,
1858f4259b30SLisandro Dalcin                                        NULL,
1859f4259b30SLisandro Dalcin                                        NULL,
1860f4259b30SLisandro Dalcin                                        NULL,
1861f4259b30SLisandro Dalcin                                /*109*/ NULL,
1862f4259b30SLisandro Dalcin                                        NULL,
1863f4259b30SLisandro Dalcin                                        NULL,
1864f4259b30SLisandro Dalcin                                        NULL,
1865d4002b98SHong Zhang                                        MatMissingDiagonal_SeqSELL,
1866f4259b30SLisandro Dalcin                                /*114*/ NULL,
1867f4259b30SLisandro Dalcin                                        NULL,
1868f4259b30SLisandro Dalcin                                        NULL,
1869f4259b30SLisandro Dalcin                                        NULL,
1870f4259b30SLisandro Dalcin                                        NULL,
1871f4259b30SLisandro Dalcin                                /*119*/ NULL,
1872f4259b30SLisandro Dalcin                                        NULL,
1873f4259b30SLisandro Dalcin                                        NULL,
1874f4259b30SLisandro Dalcin                                        NULL,
1875f4259b30SLisandro Dalcin                                        NULL,
1876f4259b30SLisandro Dalcin                                /*124*/ NULL,
1877f4259b30SLisandro Dalcin                                        NULL,
1878f4259b30SLisandro Dalcin                                        NULL,
1879f4259b30SLisandro Dalcin                                        NULL,
1880f4259b30SLisandro Dalcin                                        NULL,
1881f4259b30SLisandro Dalcin                                /*129*/ NULL,
1882f4259b30SLisandro Dalcin                                        NULL,
1883f4259b30SLisandro Dalcin                                        NULL,
1884f4259b30SLisandro Dalcin                                        NULL,
1885f4259b30SLisandro Dalcin                                        NULL,
1886f4259b30SLisandro Dalcin                                /*134*/ NULL,
1887f4259b30SLisandro Dalcin                                        NULL,
1888f4259b30SLisandro Dalcin                                        NULL,
1889f4259b30SLisandro Dalcin                                        NULL,
1890f4259b30SLisandro Dalcin                                        NULL,
1891f4259b30SLisandro Dalcin                                /*139*/ NULL,
1892f4259b30SLisandro Dalcin                                        NULL,
1893f4259b30SLisandro Dalcin                                        NULL,
1894d4002b98SHong Zhang                                        MatFDColoringSetUp_SeqXAIJ,
1895f4259b30SLisandro Dalcin                                        NULL,
1896f4259b30SLisandro Dalcin                                /*144*/ NULL
1897d4002b98SHong Zhang };
1898d4002b98SHong Zhang 
1899d4002b98SHong Zhang PetscErrorCode MatStoreValues_SeqSELL(Mat mat)
1900d4002b98SHong Zhang {
1901d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)mat->data;
1902d4002b98SHong Zhang   PetscErrorCode ierr;
1903d4002b98SHong Zhang 
1904d4002b98SHong Zhang   PetscFunctionBegin;
1905*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(!a->nonew,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1906d4002b98SHong Zhang 
1907d4002b98SHong Zhang   /* allocate space for values if not already there */
1908d4002b98SHong Zhang   if (!a->saved_values) {
1909d4002b98SHong Zhang     ierr = PetscMalloc1(a->sliidx[a->totalslices]+1,&a->saved_values);CHKERRQ(ierr);
1910d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)mat,(a->sliidx[a->totalslices]+1)*sizeof(PetscScalar));CHKERRQ(ierr);
1911d4002b98SHong Zhang   }
1912d4002b98SHong Zhang 
1913d4002b98SHong Zhang   /* copy values over */
1914580bdb30SBarry Smith   ierr = PetscArraycpy(a->saved_values,a->val,a->sliidx[a->totalslices]);CHKERRQ(ierr);
1915d4002b98SHong Zhang   PetscFunctionReturn(0);
1916d4002b98SHong Zhang }
1917d4002b98SHong Zhang 
1918d4002b98SHong Zhang PetscErrorCode MatRetrieveValues_SeqSELL(Mat mat)
1919d4002b98SHong Zhang {
1920d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)mat->data;
1921d4002b98SHong Zhang   PetscErrorCode ierr;
1922d4002b98SHong Zhang 
1923d4002b98SHong Zhang   PetscFunctionBegin;
1924*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(!a->nonew,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1925*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(!a->saved_values,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
1926580bdb30SBarry Smith   ierr = PetscArraycpy(a->val,a->saved_values,a->sliidx[a->totalslices]);CHKERRQ(ierr);
1927d4002b98SHong Zhang   PetscFunctionReturn(0);
1928d4002b98SHong Zhang }
1929d4002b98SHong Zhang 
1930d4002b98SHong Zhang /*@C
1931d4002b98SHong Zhang  MatSeqSELLRestoreArray - returns access to the array where the data for a MATSEQSELL matrix is stored obtained by MatSeqSELLGetArray()
1932d4002b98SHong Zhang 
1933d4002b98SHong Zhang  Not Collective
1934d4002b98SHong Zhang 
1935d4002b98SHong Zhang  Input Parameters:
1936d4002b98SHong Zhang  .  mat - a MATSEQSELL matrix
1937d4002b98SHong Zhang  .  array - pointer to the data
1938d4002b98SHong Zhang 
1939d4002b98SHong Zhang  Level: intermediate
1940d4002b98SHong Zhang 
1941d4002b98SHong Zhang  .seealso: MatSeqSELLGetArray(), MatSeqSELLRestoreArrayF90()
1942d4002b98SHong Zhang  @*/
1943d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray(Mat A,PetscScalar **array)
1944d4002b98SHong Zhang {
1945d4002b98SHong Zhang   PetscErrorCode ierr;
1946d4002b98SHong Zhang 
1947d4002b98SHong Zhang   PetscFunctionBegin;
1948d4002b98SHong Zhang   ierr = PetscUseMethod(A,"MatSeqSELLRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
1949d4002b98SHong Zhang   PetscFunctionReturn(0);
1950d4002b98SHong Zhang }
1951d4002b98SHong Zhang 
1952d4002b98SHong Zhang PETSC_EXTERN PetscErrorCode MatCreate_SeqSELL(Mat B)
1953d4002b98SHong Zhang {
1954d4002b98SHong Zhang   Mat_SeqSELL    *b;
1955d4002b98SHong Zhang   PetscMPIInt    size;
1956d4002b98SHong Zhang   PetscErrorCode ierr;
1957d4002b98SHong Zhang 
1958d4002b98SHong Zhang   PetscFunctionBegin;
1959ed73aabaSBarry Smith   ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr);
1960ffc4695bSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRMPI(ierr);
1961*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(size > 1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Comm must be of size 1");
1962d4002b98SHong Zhang 
1963d4002b98SHong Zhang   ierr = PetscNewLog(B,&b);CHKERRQ(ierr);
1964d4002b98SHong Zhang 
1965d4002b98SHong Zhang   B->data = (void*)b;
1966d4002b98SHong Zhang 
1967d4002b98SHong Zhang   ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1968d4002b98SHong Zhang 
1969f4259b30SLisandro Dalcin   b->row                = NULL;
1970f4259b30SLisandro Dalcin   b->col                = NULL;
1971f4259b30SLisandro Dalcin   b->icol               = NULL;
1972d4002b98SHong Zhang   b->reallocs           = 0;
1973d4002b98SHong Zhang   b->ignorezeroentries  = PETSC_FALSE;
1974d4002b98SHong Zhang   b->roworiented        = PETSC_TRUE;
1975d4002b98SHong Zhang   b->nonew              = 0;
1976f4259b30SLisandro Dalcin   b->diag               = NULL;
1977f4259b30SLisandro Dalcin   b->solve_work         = NULL;
1978f4259b30SLisandro Dalcin   B->spptr              = NULL;
1979f4259b30SLisandro Dalcin   b->saved_values       = NULL;
1980f4259b30SLisandro Dalcin   b->idiag              = NULL;
1981f4259b30SLisandro Dalcin   b->mdiag              = NULL;
1982f4259b30SLisandro Dalcin   b->ssor_work          = NULL;
1983d4002b98SHong Zhang   b->omega              = 1.0;
1984d4002b98SHong Zhang   b->fshift             = 0.0;
1985d4002b98SHong Zhang   b->idiagvalid         = PETSC_FALSE;
1986d4002b98SHong Zhang   b->keepnonzeropattern = PETSC_FALSE;
1987d4002b98SHong Zhang 
1988d4002b98SHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSELL);CHKERRQ(ierr);
1989d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLGetArray_C",MatSeqSELLGetArray_SeqSELL);CHKERRQ(ierr);
1990d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLRestoreArray_C",MatSeqSELLRestoreArray_SeqSELL);CHKERRQ(ierr);
1991d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSELL);CHKERRQ(ierr);
1992d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSELL);CHKERRQ(ierr);
1993d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLSetPreallocation_C",MatSeqSELLSetPreallocation_SeqSELL);CHKERRQ(ierr);
1994d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsell_seqaij_C",MatConvert_SeqSELL_SeqAIJ);CHKERRQ(ierr);
1995d4002b98SHong Zhang   PetscFunctionReturn(0);
1996d4002b98SHong Zhang }
1997d4002b98SHong Zhang 
1998d4002b98SHong Zhang /*
1999d4002b98SHong Zhang  Given a matrix generated with MatGetFactor() duplicates all the information in A into B
2000d4002b98SHong Zhang  */
2001d4002b98SHong Zhang PetscErrorCode MatDuplicateNoCreate_SeqSELL(Mat C,Mat A,MatDuplicateOption cpvalues,PetscBool mallocmatspace)
2002d4002b98SHong Zhang {
2003ed73aabaSBarry Smith   Mat_SeqSELL    *c = (Mat_SeqSELL*)C->data,*a = (Mat_SeqSELL*)A->data;
2004d4002b98SHong Zhang   PetscInt       i,m=A->rmap->n;
2005d4002b98SHong Zhang   PetscInt       totalslices=a->totalslices;
2006d4002b98SHong Zhang   PetscErrorCode ierr;
2007d4002b98SHong Zhang 
2008d4002b98SHong Zhang   PetscFunctionBegin;
2009d4002b98SHong Zhang   C->factortype = A->factortype;
2010f4259b30SLisandro Dalcin   c->row        = NULL;
2011f4259b30SLisandro Dalcin   c->col        = NULL;
2012f4259b30SLisandro Dalcin   c->icol       = NULL;
2013d4002b98SHong Zhang   c->reallocs   = 0;
2014d4002b98SHong Zhang   C->assembled = PETSC_TRUE;
2015d4002b98SHong Zhang 
2016d4002b98SHong Zhang   ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
2017d4002b98SHong Zhang   ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
2018d4002b98SHong Zhang 
2019d4002b98SHong Zhang   ierr = PetscMalloc1(8*totalslices,&c->rlen);CHKERRQ(ierr);
2020d4002b98SHong Zhang   ierr = PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt));CHKERRQ(ierr);
2021d4002b98SHong Zhang   ierr = PetscMalloc1(totalslices+1,&c->sliidx);CHKERRQ(ierr);
2022d4002b98SHong Zhang   ierr = PetscLogObjectMemory((PetscObject)C, (totalslices+1)*sizeof(PetscInt));CHKERRQ(ierr);
2023d4002b98SHong Zhang 
2024d4002b98SHong Zhang   for (i=0; i<m; i++) c->rlen[i] = a->rlen[i];
2025d4002b98SHong Zhang   for (i=0; i<totalslices+1; i++) c->sliidx[i] = a->sliidx[i];
2026d4002b98SHong Zhang 
2027d4002b98SHong Zhang   /* allocate the matrix space */
2028d4002b98SHong Zhang   if (mallocmatspace) {
2029d4002b98SHong Zhang     ierr = PetscMalloc2(a->maxallocmat,&c->val,a->maxallocmat,&c->colidx);CHKERRQ(ierr);
2030d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)C,a->maxallocmat*(sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
2031d4002b98SHong Zhang 
2032d4002b98SHong Zhang     c->singlemalloc = PETSC_TRUE;
2033d4002b98SHong Zhang 
2034d4002b98SHong Zhang     if (m > 0) {
2035580bdb30SBarry Smith       ierr = PetscArraycpy(c->colidx,a->colidx,a->maxallocmat);CHKERRQ(ierr);
2036d4002b98SHong Zhang       if (cpvalues == MAT_COPY_VALUES) {
2037580bdb30SBarry Smith         ierr = PetscArraycpy(c->val,a->val,a->maxallocmat);CHKERRQ(ierr);
2038d4002b98SHong Zhang       } else {
2039580bdb30SBarry Smith         ierr = PetscArrayzero(c->val,a->maxallocmat);CHKERRQ(ierr);
2040d4002b98SHong Zhang       }
2041d4002b98SHong Zhang     }
2042d4002b98SHong Zhang   }
2043d4002b98SHong Zhang 
2044d4002b98SHong Zhang   c->ignorezeroentries = a->ignorezeroentries;
2045d4002b98SHong Zhang   c->roworiented       = a->roworiented;
2046d4002b98SHong Zhang   c->nonew             = a->nonew;
2047d4002b98SHong Zhang   if (a->diag) {
2048d4002b98SHong Zhang     ierr = PetscMalloc1(m,&c->diag);CHKERRQ(ierr);
2049d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt));CHKERRQ(ierr);
2050d4002b98SHong Zhang     for (i=0; i<m; i++) {
2051d4002b98SHong Zhang       c->diag[i] = a->diag[i];
2052d4002b98SHong Zhang     }
2053f4259b30SLisandro Dalcin   } else c->diag = NULL;
2054d4002b98SHong Zhang 
2055f4259b30SLisandro Dalcin   c->solve_work         = NULL;
2056f4259b30SLisandro Dalcin   c->saved_values       = NULL;
2057f4259b30SLisandro Dalcin   c->idiag              = NULL;
2058f4259b30SLisandro Dalcin   c->ssor_work          = NULL;
2059d4002b98SHong Zhang   c->keepnonzeropattern = a->keepnonzeropattern;
2060d4002b98SHong Zhang   c->free_val           = PETSC_TRUE;
2061d4002b98SHong Zhang   c->free_colidx        = PETSC_TRUE;
2062d4002b98SHong Zhang 
2063d4002b98SHong Zhang   c->maxallocmat  = a->maxallocmat;
2064d4002b98SHong Zhang   c->maxallocrow  = a->maxallocrow;
2065d4002b98SHong Zhang   c->rlenmax      = a->rlenmax;
2066d4002b98SHong Zhang   c->nz           = a->nz;
2067d4002b98SHong Zhang   C->preallocated = PETSC_TRUE;
2068d4002b98SHong Zhang 
2069d4002b98SHong Zhang   c->nonzerorowcnt = a->nonzerorowcnt;
2070d4002b98SHong Zhang   C->nonzerostate  = A->nonzerostate;
2071d4002b98SHong Zhang 
2072d4002b98SHong Zhang   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
2073d4002b98SHong Zhang   PetscFunctionReturn(0);
2074d4002b98SHong Zhang }
2075d4002b98SHong Zhang 
2076d4002b98SHong Zhang PetscErrorCode MatDuplicate_SeqSELL(Mat A,MatDuplicateOption cpvalues,Mat *B)
2077d4002b98SHong Zhang {
2078d4002b98SHong Zhang   PetscErrorCode ierr;
2079d4002b98SHong Zhang 
2080d4002b98SHong Zhang   PetscFunctionBegin;
2081d4002b98SHong Zhang   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
2082d4002b98SHong Zhang   ierr = MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr);
2083d4002b98SHong Zhang   if (!(A->rmap->n % A->rmap->bs) && !(A->cmap->n % A->cmap->bs)) {
2084d4002b98SHong Zhang     ierr = MatSetBlockSizesFromMats(*B,A,A);CHKERRQ(ierr);
2085d4002b98SHong Zhang   }
2086d4002b98SHong Zhang   ierr = MatSetType(*B,((PetscObject)A)->type_name);CHKERRQ(ierr);
2087d4002b98SHong Zhang   ierr = MatDuplicateNoCreate_SeqSELL(*B,A,cpvalues,PETSC_TRUE);CHKERRQ(ierr);
2088d4002b98SHong Zhang   PetscFunctionReturn(0);
2089d4002b98SHong Zhang }
2090d4002b98SHong Zhang 
2091ed73aabaSBarry Smith /*MC
2092ed73aabaSBarry Smith    MATSEQSELL - MATSEQSELL = "seqsell" - A matrix type to be used for sequential sparse matrices,
2093ed73aabaSBarry Smith    based on the sliced Ellpack format
2094ed73aabaSBarry Smith 
2095ed73aabaSBarry Smith    Options Database Keys:
2096ed73aabaSBarry Smith . -mat_type seqsell - sets the matrix type to "seqsell" during a call to MatSetFromOptions()
2097ed73aabaSBarry Smith 
2098ed73aabaSBarry Smith    Level: beginner
2099ed73aabaSBarry Smith 
2100ed73aabaSBarry Smith .seealso: MatCreateSeqSell(), MATSELL, MATMPISELL, MATSEQAIJ, MATAIJ, MATMPIAIJ
2101ed73aabaSBarry Smith M*/
2102ed73aabaSBarry Smith 
2103ed73aabaSBarry Smith /*MC
2104ed73aabaSBarry Smith    MATSELL - MATSELL = "sell" - A matrix type to be used for sparse matrices.
2105ed73aabaSBarry Smith 
2106ed73aabaSBarry Smith    This matrix type is identical to MATSEQSELL when constructed with a single process communicator,
2107ed73aabaSBarry Smith    and MATMPISELL otherwise.  As a result, for single process communicators,
2108ed73aabaSBarry Smith   MatSeqSELLSetPreallocation() is supported, and similarly MatMPISELLSetPreallocation() is supported
2109ed73aabaSBarry Smith   for communicators controlling multiple processes.  It is recommended that you call both of
2110ed73aabaSBarry Smith   the above preallocation routines for simplicity.
2111ed73aabaSBarry Smith 
2112ed73aabaSBarry Smith    Options Database Keys:
2113ed73aabaSBarry Smith . -mat_type sell - sets the matrix type to "sell" during a call to MatSetFromOptions()
2114ed73aabaSBarry Smith 
2115ed73aabaSBarry Smith   Level: beginner
2116ed73aabaSBarry Smith 
2117ed73aabaSBarry Smith   Notes:
2118ed73aabaSBarry Smith    This format is only supported for real scalars, double precision, and 32 bit indices (the defaults).
2119ed73aabaSBarry Smith 
2120ed73aabaSBarry Smith    It can provide better performance on Intel and AMD processes with AVX2 or AVX512 support for matrices that have a similar number of
2121ed73aabaSBarry Smith    non-zeros in contiguous groups of rows. However if the computation is memory bandwidth limited it may not provide much improvement.
2122ed73aabaSBarry Smith 
2123ed73aabaSBarry Smith   Developer Notes:
2124ed73aabaSBarry Smith    On Intel (and AMD) systems some of the matrix operations use SIMD (AVX) instructions to achieve higher performance.
2125ed73aabaSBarry Smith 
2126ed73aabaSBarry Smith    The sparse matrix format is as follows. For simplicity we assume a slice size of 2, it is actually 8
2127ed73aabaSBarry Smith .vb
2128ed73aabaSBarry Smith                             (2 0  3 4)
2129ed73aabaSBarry Smith    Consider the matrix A =  (5 0  6 0)
2130ed73aabaSBarry Smith                             (0 0  7 8)
2131ed73aabaSBarry Smith                             (0 0  9 9)
2132ed73aabaSBarry Smith 
2133ed73aabaSBarry Smith    symbolically the Ellpack format can be written as
2134ed73aabaSBarry Smith 
2135ed73aabaSBarry Smith         (2 3 4 |)           (0 2 3 |)
2136ed73aabaSBarry Smith    v =  (5 6 0 |)  colidx = (0 2 2 |)
2137ed73aabaSBarry Smith         --------            ---------
2138ed73aabaSBarry Smith         (7 8 |)             (2 3 |)
2139ed73aabaSBarry Smith         (9 9 |)             (2 3 |)
2140ed73aabaSBarry Smith 
2141ed73aabaSBarry 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).
2142ed73aabaSBarry 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
2143ed73aabaSBarry 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.
2144ed73aabaSBarry Smith 
2145ed73aabaSBarry 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)
2146ed73aabaSBarry Smith 
2147ed73aabaSBarry Smith .ve
2148ed73aabaSBarry Smith 
2149ed73aabaSBarry Smith       See MatMult_SeqSELL() for how this format is used with the SIMD operations to achieve high performance.
2150ed73aabaSBarry Smith 
2151ed73aabaSBarry Smith  References:
2152ed73aabaSBarry Smith    Hong Zhang, Richard T. Mills, Karl Rupp, and Barry F. Smith, Vectorized Parallel Sparse Matrix-Vector Multiplication in {PETSc} Using {AVX-512},
2153ed73aabaSBarry Smith    Proceedings of the 47th International Conference on Parallel Processing, 2018.
2154ed73aabaSBarry Smith 
2155ed73aabaSBarry Smith .seealso: MatCreateSeqSELL(), MatCreateSeqAIJ(), MatCreateSell(), MATSEQSELL, MATMPISELL, MATSEQAIJ, MATMPIAIJ, MATAIJ
2156ed73aabaSBarry Smith M*/
2157ed73aabaSBarry Smith 
2158d4002b98SHong Zhang /*@C
2159d4002b98SHong Zhang        MatCreateSeqSELL - Creates a sparse matrix in SELL format.
2160d4002b98SHong Zhang 
2161ed73aabaSBarry Smith  Collective on comm
2162d4002b98SHong Zhang 
2163d4002b98SHong Zhang  Input Parameters:
2164d4002b98SHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2165d4002b98SHong Zhang .  m - number of rows
2166d4002b98SHong Zhang .  n - number of columns
2167d4002b98SHong Zhang .  rlenmax - maximum number of nonzeros in a row
2168d4002b98SHong Zhang -  rlen - array containing the number of nonzeros in the various rows
2169d4002b98SHong Zhang  (possibly different for each row) or NULL
2170d4002b98SHong Zhang 
2171d4002b98SHong Zhang  Output Parameter:
2172d4002b98SHong Zhang .  A - the matrix
2173d4002b98SHong Zhang 
2174d4002b98SHong Zhang  It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2175f6f02116SRichard Tran Mills  MatXXXXSetPreallocation() paradigm instead of this routine directly.
2176d4002b98SHong Zhang  [MatXXXXSetPreallocation() is, for example, MatSeqSELLSetPreallocation]
2177d4002b98SHong Zhang 
2178d4002b98SHong Zhang  Notes:
2179d4002b98SHong Zhang  If nnz is given then nz is ignored
2180d4002b98SHong Zhang 
2181d4002b98SHong Zhang  Specify the preallocated storage with either rlenmax or rlen (not both).
2182d4002b98SHong Zhang  Set rlenmax=PETSC_DEFAULT and rlen=NULL for PETSc to control dynamic memory
2183d4002b98SHong Zhang  allocation.  For large problems you MUST preallocate memory or you
2184d4002b98SHong Zhang  will get TERRIBLE performance, see the users' manual chapter on matrices.
2185d4002b98SHong Zhang 
2186d4002b98SHong Zhang  Level: intermediate
2187d4002b98SHong Zhang 
2188ed73aabaSBarry Smith  .seealso: MatCreate(), MatCreateSELL(), MatSetValues(), MatSeqSELLSetPreallocation(), MATSELL, MATSEQSELL, MATMPISELL
2189d4002b98SHong Zhang 
2190d4002b98SHong Zhang  @*/
2191d4002b98SHong Zhang PetscErrorCode MatCreateSeqSELL(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt maxallocrow,const PetscInt rlen[],Mat *A)
2192d4002b98SHong Zhang {
2193d4002b98SHong Zhang   PetscErrorCode ierr;
2194d4002b98SHong Zhang 
2195d4002b98SHong Zhang   PetscFunctionBegin;
2196d4002b98SHong Zhang   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2197d4002b98SHong Zhang   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2198d4002b98SHong Zhang   ierr = MatSetType(*A,MATSEQSELL);CHKERRQ(ierr);
2199d4002b98SHong Zhang   ierr = MatSeqSELLSetPreallocation_SeqSELL(*A,maxallocrow,rlen);CHKERRQ(ierr);
2200d4002b98SHong Zhang   PetscFunctionReturn(0);
2201d4002b98SHong Zhang }
2202d4002b98SHong Zhang 
2203d4002b98SHong Zhang PetscErrorCode MatEqual_SeqSELL(Mat A,Mat B,PetscBool * flg)
2204d4002b98SHong Zhang {
2205d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data,*b=(Mat_SeqSELL*)B->data;
2206d4002b98SHong Zhang   PetscInt       totalslices=a->totalslices;
2207d4002b98SHong Zhang   PetscErrorCode ierr;
2208d4002b98SHong Zhang 
2209d4002b98SHong Zhang   PetscFunctionBegin;
2210d4002b98SHong Zhang   /* If the  matrix dimensions are not equal,or no of nonzeros */
2211d4002b98SHong Zhang   if ((A->rmap->n != B->rmap->n) || (A->cmap->n != B->cmap->n) ||(a->nz != b->nz) || (a->rlenmax != b->rlenmax)) {
2212d4002b98SHong Zhang     *flg = PETSC_FALSE;
2213d4002b98SHong Zhang     PetscFunctionReturn(0);
2214d4002b98SHong Zhang   }
2215d4002b98SHong Zhang   /* if the a->colidx are the same */
2216580bdb30SBarry Smith   ierr = PetscArraycmp(a->colidx,b->colidx,a->sliidx[totalslices],flg);CHKERRQ(ierr);
2217d4002b98SHong Zhang   if (!*flg) PetscFunctionReturn(0);
2218d4002b98SHong Zhang   /* if a->val are the same */
2219580bdb30SBarry Smith   ierr = PetscArraycmp(a->val,b->val,a->sliidx[totalslices],flg);CHKERRQ(ierr);
2220d4002b98SHong Zhang   PetscFunctionReturn(0);
2221d4002b98SHong Zhang }
2222d4002b98SHong Zhang 
2223d4002b98SHong Zhang PetscErrorCode MatSeqSELLInvalidateDiagonal(Mat A)
2224d4002b98SHong Zhang {
2225d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
2226d4002b98SHong Zhang 
2227d4002b98SHong Zhang   PetscFunctionBegin;
2228d4002b98SHong Zhang   a->idiagvalid  = PETSC_FALSE;
2229d4002b98SHong Zhang   PetscFunctionReturn(0);
2230d4002b98SHong Zhang }
2231d4002b98SHong Zhang 
2232d4002b98SHong Zhang PetscErrorCode MatConjugate_SeqSELL(Mat A)
2233d4002b98SHong Zhang {
2234d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
2235d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
2236d4002b98SHong Zhang   PetscInt    i;
2237d4002b98SHong Zhang   PetscScalar *val = a->val;
2238d4002b98SHong Zhang 
2239d4002b98SHong Zhang   PetscFunctionBegin;
2240d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) {
2241d4002b98SHong Zhang     val[i] = PetscConj(val[i]);
2242d4002b98SHong Zhang   }
2243d4002b98SHong Zhang #else
2244d4002b98SHong Zhang   PetscFunctionBegin;
2245d4002b98SHong Zhang #endif
2246d4002b98SHong Zhang   PetscFunctionReturn(0);
2247d4002b98SHong Zhang }
2248