xref: /petsc/src/mat/impls/sell/seq/sell.c (revision 28b400f66ebc7ae0049166a2294dfcd3df27e64b)
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   PetscFunctionBegin;
88d4002b98SHong Zhang   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
89d4002b98SHong Zhang   PetscValidType(B,1);
905f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscTryMethod(B,"MatSeqSELLSetPreallocation_C",(Mat,PetscInt,const PetscInt[]),(B,rlenmax,rlen)));
91d4002b98SHong Zhang   PetscFunctionReturn(0);
92d4002b98SHong Zhang }
93d4002b98SHong Zhang 
94d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation_SeqSELL(Mat B,PetscInt maxallocrow,const PetscInt rlen[])
95d4002b98SHong Zhang {
96d4002b98SHong Zhang   Mat_SeqSELL    *b;
97d4002b98SHong Zhang   PetscInt       i,j,totalslices;
98d4002b98SHong Zhang   PetscBool      skipallocation=PETSC_FALSE,realalloc=PETSC_FALSE;
99d4002b98SHong Zhang 
100d4002b98SHong Zhang   PetscFunctionBegin;
101d4002b98SHong Zhang   if (maxallocrow >= 0 || rlen) realalloc = PETSC_TRUE;
102d4002b98SHong Zhang   if (maxallocrow == MAT_SKIP_ALLOCATION) {
103d4002b98SHong Zhang     skipallocation = PETSC_TRUE;
104d4002b98SHong Zhang     maxallocrow    = 0;
105d4002b98SHong Zhang   }
106d4002b98SHong Zhang 
1075f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLayoutSetUp(B->rmap));
1085f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLayoutSetUp(B->cmap));
109d4002b98SHong Zhang 
110d4002b98SHong Zhang   /* FIXME: if one preallocates more space than needed, the matrix does not shrink automatically, but for best performance it should */
111d4002b98SHong Zhang   if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 5;
1122c71b3e2SJacob Faibussowitsch   PetscCheckFalse(maxallocrow < 0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"maxallocrow cannot be less than 0: value %" PetscInt_FMT,maxallocrow);
113d4002b98SHong Zhang   if (rlen) {
114d4002b98SHong Zhang     for (i=0; i<B->rmap->n; i++) {
1152c71b3e2SJacob 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]);
1162c71b3e2SJacob 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);
117d4002b98SHong Zhang     }
118d4002b98SHong Zhang   }
119d4002b98SHong Zhang 
120d4002b98SHong Zhang   B->preallocated = PETSC_TRUE;
121d4002b98SHong Zhang 
122d4002b98SHong Zhang   b = (Mat_SeqSELL*)B->data;
123d4002b98SHong Zhang 
124faa75363SBarry Smith   totalslices = PetscCeilInt(B->rmap->n,8);
125d4002b98SHong Zhang   b->totalslices = totalslices;
126d4002b98SHong Zhang   if (!skipallocation) {
1275f80ce2aSJacob Faibussowitsch     if (B->rmap->n & 0x07) CHKERRQ(PetscInfo(B,"Padding rows to the SEQSELL matrix because the number of rows is not the multiple of 8 (value %" PetscInt_FMT ")\n",B->rmap->n));
128d4002b98SHong Zhang 
129d4002b98SHong Zhang     if (!b->sliidx) { /* sliidx gives the starting index of each slice, the last element is the total space allocated */
1305f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscMalloc1(totalslices+1,&b->sliidx));
1315f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscLogObjectMemory((PetscObject)B,(totalslices+1)*sizeof(PetscInt)));
132d4002b98SHong Zhang     }
133d4002b98SHong Zhang     if (!rlen) { /* if rlen is not provided, allocate same space for all the slices */
134d4002b98SHong Zhang       if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 10;
135d4002b98SHong Zhang       else if (maxallocrow < 0) maxallocrow = 1;
136d4002b98SHong Zhang       for (i=0; i<=totalslices; i++) b->sliidx[i] = i*8*maxallocrow;
137d4002b98SHong Zhang     } else {
138d4002b98SHong Zhang       maxallocrow = 0;
139d4002b98SHong Zhang       b->sliidx[0] = 0;
140d4002b98SHong Zhang       for (i=1; i<totalslices; i++) {
141d4002b98SHong Zhang         b->sliidx[i] = 0;
142d4002b98SHong Zhang         for (j=0;j<8;j++) {
143d4002b98SHong Zhang           b->sliidx[i] = PetscMax(b->sliidx[i],rlen[8*(i-1)+j]);
144d4002b98SHong Zhang         }
145d4002b98SHong Zhang         maxallocrow = PetscMax(b->sliidx[i],maxallocrow);
1465f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscIntSumError(b->sliidx[i-1],8*b->sliidx[i],&b->sliidx[i]));
147d4002b98SHong Zhang       }
148d4002b98SHong Zhang       /* last slice */
149d4002b98SHong Zhang       b->sliidx[totalslices] = 0;
150d4002b98SHong Zhang       for (j=(totalslices-1)*8;j<B->rmap->n;j++) b->sliidx[totalslices] = PetscMax(b->sliidx[totalslices],rlen[j]);
151d4002b98SHong Zhang       maxallocrow = PetscMax(b->sliidx[totalslices],maxallocrow);
152d4002b98SHong Zhang       b->sliidx[totalslices] = b->sliidx[totalslices-1] + 8*b->sliidx[totalslices];
153d4002b98SHong Zhang     }
154d4002b98SHong Zhang 
155d4002b98SHong Zhang     /* allocate space for val, colidx, rlen */
156d4002b98SHong Zhang     /* FIXME: should B's old memory be unlogged? */
1575f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSeqXSELLFreeSELL(B,&b->val,&b->colidx));
158d4002b98SHong Zhang     /* FIXME: assuming an element of the bit array takes 8 bits */
1595f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc2(b->sliidx[totalslices],&b->val,b->sliidx[totalslices],&b->colidx));
1605f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogObjectMemory((PetscObject)B,b->sliidx[totalslices]*(sizeof(PetscScalar)+sizeof(PetscInt))));
161d4002b98SHong Zhang     /* b->rlen will count nonzeros in each row so far. We dont copy rlen to b->rlen because the matrix has not been set. */
1625f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscCalloc1(8*totalslices,&b->rlen));
1635f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogObjectMemory((PetscObject)B,8*totalslices*sizeof(PetscInt)));
164d4002b98SHong Zhang 
165d4002b98SHong Zhang     b->singlemalloc = PETSC_TRUE;
166d4002b98SHong Zhang     b->free_val     = PETSC_TRUE;
167d4002b98SHong Zhang     b->free_colidx  = PETSC_TRUE;
168d4002b98SHong Zhang   } else {
169d4002b98SHong Zhang     b->free_val    = PETSC_FALSE;
170d4002b98SHong Zhang     b->free_colidx = PETSC_FALSE;
171d4002b98SHong Zhang   }
172d4002b98SHong Zhang 
173d4002b98SHong Zhang   b->nz               = 0;
174d4002b98SHong Zhang   b->maxallocrow      = maxallocrow;
175d4002b98SHong Zhang   b->rlenmax          = maxallocrow;
176d4002b98SHong Zhang   b->maxallocmat      = b->sliidx[totalslices];
177d4002b98SHong Zhang   B->info.nz_unneeded = (double)b->maxallocmat;
178d4002b98SHong Zhang   if (realalloc) {
1795f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE));
180d4002b98SHong Zhang   }
181d4002b98SHong Zhang   PetscFunctionReturn(0);
182d4002b98SHong Zhang }
183d4002b98SHong Zhang 
1846108893eSStefano Zampini PetscErrorCode MatGetRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
1856108893eSStefano Zampini {
1866108893eSStefano Zampini   Mat_SeqSELL *a = (Mat_SeqSELL*)A->data;
1876108893eSStefano Zampini   PetscInt    shift;
1886108893eSStefano Zampini 
1896108893eSStefano Zampini   PetscFunctionBegin;
1902c71b3e2SJacob Faibussowitsch   PetscCheckFalse(row < 0 || row >= A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %" PetscInt_FMT " out of range",row);
1916108893eSStefano Zampini   if (nz) *nz = a->rlen[row];
1926108893eSStefano Zampini   shift = a->sliidx[row>>3]+(row&0x07);
1936108893eSStefano Zampini   if (!a->getrowcols) {
1946108893eSStefano Zampini 
1955f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc2(a->rlenmax,&a->getrowcols,a->rlenmax,&a->getrowvals));
1966108893eSStefano Zampini   }
1976108893eSStefano Zampini   if (idx) {
1986108893eSStefano Zampini     PetscInt j;
1996108893eSStefano Zampini     for (j=0; j<a->rlen[row]; j++) a->getrowcols[j] = a->colidx[shift+8*j];
2006108893eSStefano Zampini     *idx = a->getrowcols;
2016108893eSStefano Zampini   }
2026108893eSStefano Zampini   if (v) {
2036108893eSStefano Zampini     PetscInt j;
2046108893eSStefano Zampini     for (j=0; j<a->rlen[row]; j++) a->getrowvals[j] = a->val[shift+8*j];
2056108893eSStefano Zampini     *v = a->getrowvals;
2066108893eSStefano Zampini   }
2076108893eSStefano Zampini   PetscFunctionReturn(0);
2086108893eSStefano Zampini }
2096108893eSStefano Zampini 
2106108893eSStefano Zampini PetscErrorCode MatRestoreRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
2116108893eSStefano Zampini {
2126108893eSStefano Zampini   PetscFunctionBegin;
2136108893eSStefano Zampini   PetscFunctionReturn(0);
2146108893eSStefano Zampini }
2156108893eSStefano Zampini 
216d4002b98SHong Zhang PetscErrorCode MatConvert_SeqSELL_SeqAIJ(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
217d4002b98SHong Zhang {
218d4002b98SHong Zhang   Mat            B;
219d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
220e3f1f374SStefano Zampini   PetscInt       i;
221d4002b98SHong Zhang 
222d4002b98SHong Zhang   PetscFunctionBegin;
223ad013a7bSRichard Tran Mills   if (reuse == MAT_REUSE_MATRIX) {
224ad013a7bSRichard Tran Mills     B    = *newmat;
2255f80ce2aSJacob Faibussowitsch     CHKERRQ(MatZeroEntries(B));
226ad013a7bSRichard Tran Mills   } else {
2275f80ce2aSJacob Faibussowitsch     CHKERRQ(MatCreate(PetscObjectComm((PetscObject)A),&B));
2285f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N));
2295f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetType(B,MATSEQAIJ));
2305f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSeqAIJSetPreallocation(B,0,a->rlen));
231ad013a7bSRichard Tran Mills   }
232d4002b98SHong Zhang 
233e3f1f374SStefano Zampini   for (i=0; i<A->rmap->n; i++) {
234e108cb99SStefano Zampini     PetscInt    nz = 0,*cols = NULL;
235e108cb99SStefano Zampini     PetscScalar *vals = NULL;
236e3f1f374SStefano Zampini 
2375f80ce2aSJacob Faibussowitsch     CHKERRQ(MatGetRow_SeqSELL(A,i,&nz,&cols,&vals));
2385f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetValues(B,1,&i,nz,cols,vals,INSERT_VALUES));
2395f80ce2aSJacob Faibussowitsch     CHKERRQ(MatRestoreRow_SeqSELL(A,i,&nz,&cols,&vals));
240d4002b98SHong Zhang   }
241e3f1f374SStefano Zampini 
2425f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY));
2435f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY));
244d4002b98SHong Zhang   B->rmap->bs = A->rmap->bs;
245d4002b98SHong Zhang 
246d4002b98SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
2475f80ce2aSJacob Faibussowitsch     CHKERRQ(MatHeaderReplace(A,&B));
248d4002b98SHong Zhang   } else {
249d4002b98SHong Zhang     *newmat = B;
250d4002b98SHong Zhang   }
251d4002b98SHong Zhang   PetscFunctionReturn(0);
252d4002b98SHong Zhang }
253d4002b98SHong Zhang 
254d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/aij.h>
255d4002b98SHong Zhang 
256d4002b98SHong Zhang PetscErrorCode MatConvert_SeqAIJ_SeqSELL(Mat A,MatType newtype,MatReuse reuse,Mat *newmat)
257d4002b98SHong Zhang {
258d4002b98SHong Zhang   Mat               B;
259d4002b98SHong Zhang   Mat_SeqAIJ        *a=(Mat_SeqAIJ*)A->data;
260d4002b98SHong Zhang   PetscInt          *ai=a->i,m=A->rmap->N,n=A->cmap->N,i,*rowlengths,row,ncols;
261d4002b98SHong Zhang   const PetscInt    *cols;
262d4002b98SHong Zhang   const PetscScalar *vals;
263d4002b98SHong Zhang 
264d4002b98SHong Zhang   PetscFunctionBegin;
265ad013a7bSRichard Tran Mills 
266ad013a7bSRichard Tran Mills   if (reuse == MAT_REUSE_MATRIX) {
267ad013a7bSRichard Tran Mills     B = *newmat;
268ad013a7bSRichard Tran Mills   } else {
269d5e5b2e5SBarry Smith     if (PetscDefined(USE_DEBUG) || !a->ilen) {
2705f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscMalloc1(m,&rowlengths));
271d4002b98SHong Zhang       for (i=0; i<m; i++) {
272d4002b98SHong Zhang         rowlengths[i] = ai[i+1] - ai[i];
273d4002b98SHong Zhang       }
274d5e5b2e5SBarry Smith     }
275d5e5b2e5SBarry Smith     if (PetscDefined(USE_DEBUG) && a->ilen) {
276d5e5b2e5SBarry Smith       PetscBool eq;
2775f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscMemcmp(rowlengths,a->ilen,m*sizeof(PetscInt),&eq));
278*28b400f6SJacob Faibussowitsch       PetscCheck(eq,PETSC_COMM_SELF,PETSC_ERR_PLIB,"SeqAIJ ilen array incorrect");
2795f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFree(rowlengths));
280d5e5b2e5SBarry Smith       rowlengths = a->ilen;
281d5e5b2e5SBarry Smith     } else if (a->ilen) rowlengths = a->ilen;
2825f80ce2aSJacob Faibussowitsch     CHKERRQ(MatCreate(PetscObjectComm((PetscObject)A),&B));
2835f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetSizes(B,m,n,m,n));
2845f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetType(B,MATSEQSELL));
2855f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSeqSELLSetPreallocation(B,0,rowlengths));
2865f80ce2aSJacob Faibussowitsch     if (rowlengths != a->ilen) CHKERRQ(PetscFree(rowlengths));
287ad013a7bSRichard Tran Mills   }
288d4002b98SHong Zhang 
289d4002b98SHong Zhang   for (row=0; row<m; row++) {
2905f80ce2aSJacob Faibussowitsch     CHKERRQ(MatGetRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals));
2915f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetValues_SeqSELL(B,1,&row,ncols,cols,vals,INSERT_VALUES));
2925f80ce2aSJacob Faibussowitsch     CHKERRQ(MatRestoreRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals));
293d4002b98SHong Zhang   }
2945f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY));
2955f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY));
296d4002b98SHong Zhang   B->rmap->bs = A->rmap->bs;
297d4002b98SHong Zhang 
298d4002b98SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
2995f80ce2aSJacob Faibussowitsch     CHKERRQ(MatHeaderReplace(A,&B));
300d4002b98SHong Zhang   } else {
301d4002b98SHong Zhang     *newmat = B;
302d4002b98SHong Zhang   }
303d4002b98SHong Zhang   PetscFunctionReturn(0);
304d4002b98SHong Zhang }
305d4002b98SHong Zhang 
306d4002b98SHong Zhang PetscErrorCode MatMult_SeqSELL(Mat A,Vec xx,Vec yy)
307d4002b98SHong Zhang {
308d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
309d4002b98SHong Zhang   PetscScalar       *y;
310d4002b98SHong Zhang   const PetscScalar *x;
311d4002b98SHong Zhang   const MatScalar   *aval=a->val;
312d4002b98SHong Zhang   PetscInt          totalslices=a->totalslices;
313d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
3147285fed1SHong Zhang   PetscInt          i,j;
315d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
316d4002b98SHong Zhang   __m512d           vec_x,vec_y,vec_vals;
317d4002b98SHong Zhang   __m256i           vec_idx;
318d4002b98SHong Zhang   __mmask8          mask;
319d4002b98SHong Zhang   __m512d           vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4;
320d4002b98SHong Zhang   __m256i           vec_idx2,vec_idx3,vec_idx4;
3215f70456aSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX2__) && defined(__FMA__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
322a48a6482SHong Zhang   __m128i           vec_idx;
323a48a6482SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
324a48a6482SHong Zhang   MatScalar         yval;
325a48a6482SHong Zhang   PetscInt          r,rows_left,row,nnz_in_row;
32621cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
327d4002b98SHong Zhang   __m128d           vec_x_tmp;
328d4002b98SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
329d4002b98SHong Zhang   MatScalar         yval;
330d4002b98SHong Zhang   PetscInt          r,rows_left,row,nnz_in_row;
331d4002b98SHong Zhang #else
332d4002b98SHong Zhang   PetscScalar       sum[8];
333d4002b98SHong Zhang #endif
334d4002b98SHong Zhang 
335d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
336d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
337d4002b98SHong Zhang #endif
338d4002b98SHong Zhang 
339d4002b98SHong Zhang   PetscFunctionBegin;
3405f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(xx,&x));
3415f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArray(yy,&y));
342d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
343d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
344d4002b98SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
345d4002b98SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
346d4002b98SHong Zhang 
347d4002b98SHong Zhang     vec_y  = _mm512_setzero_pd();
348d4002b98SHong Zhang     vec_y2 = _mm512_setzero_pd();
349d4002b98SHong Zhang     vec_y3 = _mm512_setzero_pd();
350d4002b98SHong Zhang     vec_y4 = _mm512_setzero_pd();
351d4002b98SHong Zhang 
35238efe8efSHong Zhang     j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */
353d4002b98SHong Zhang     switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) {
354d4002b98SHong Zhang     case 3:
355d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
356d4002b98SHong Zhang       acolidx += 8; aval += 8;
357d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
358d4002b98SHong Zhang       acolidx += 8; aval += 8;
359d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
360d4002b98SHong Zhang       acolidx += 8; aval += 8;
361d4002b98SHong Zhang       j += 3;
362d4002b98SHong Zhang       break;
363d4002b98SHong Zhang     case 2:
364d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
365d4002b98SHong Zhang       acolidx += 8; aval += 8;
366d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
367d4002b98SHong Zhang       acolidx += 8; aval += 8;
368d4002b98SHong Zhang       j += 2;
369d4002b98SHong Zhang       break;
370d4002b98SHong Zhang     case 1:
371d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
372d4002b98SHong Zhang       acolidx += 8; aval += 8;
373d4002b98SHong Zhang       j += 1;
374d4002b98SHong Zhang       break;
375d4002b98SHong Zhang     }
376d4002b98SHong Zhang     #pragma novector
377d4002b98SHong Zhang     for (; j<(a->sliidx[i+1]>>3); j+=4) {
378d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
379d4002b98SHong Zhang       acolidx += 8; aval += 8;
380d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
381d4002b98SHong Zhang       acolidx += 8; aval += 8;
382d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
383d4002b98SHong Zhang       acolidx += 8; aval += 8;
384d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4);
385d4002b98SHong Zhang       acolidx += 8; aval += 8;
386d4002b98SHong Zhang     }
387d4002b98SHong Zhang 
388d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y2);
389d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y3);
390d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y4);
391d4002b98SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
392d4002b98SHong Zhang       mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07)));
393ef588d5cSRichard Tran Mills       _mm512_mask_storeu_pd(&y[8*i],mask,vec_y);
394d4002b98SHong Zhang     } else {
395ef588d5cSRichard Tran Mills       _mm512_storeu_pd(&y[8*i],vec_y);
396d4002b98SHong Zhang     }
397d4002b98SHong Zhang   }
3985f70456aSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX2__) && defined(__FMA__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
399a48a6482SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
400a48a6482SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
401a48a6482SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
402a48a6482SHong Zhang 
403a48a6482SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
404a48a6482SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
405a48a6482SHong Zhang       rows_left = A->rmap->n - 8*i;
406a48a6482SHong Zhang       for (r=0; r<rows_left; ++r) {
407a48a6482SHong Zhang         yval = (MatScalar)0;
408a48a6482SHong Zhang         row = 8*i + r;
409a48a6482SHong Zhang         nnz_in_row = a->rlen[row];
410a48a6482SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]];
411a48a6482SHong Zhang         y[row] = yval;
412a48a6482SHong Zhang       }
413a48a6482SHong Zhang       break;
414a48a6482SHong Zhang     }
415a48a6482SHong Zhang 
416a48a6482SHong Zhang     vec_y  = _mm256_setzero_pd();
417a48a6482SHong Zhang     vec_y2 = _mm256_setzero_pd();
418a48a6482SHong Zhang 
419a48a6482SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
420a48a6482SHong Zhang     #pragma novector
421a48a6482SHong Zhang     #pragma unroll(2)
422a48a6482SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
423a48a6482SHong Zhang       AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
424a48a6482SHong Zhang       aval += 4; acolidx += 4;
425a48a6482SHong Zhang       AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y2);
426a48a6482SHong Zhang       aval += 4; acolidx += 4;
427a48a6482SHong Zhang     }
428a48a6482SHong Zhang 
429ef588d5cSRichard Tran Mills     _mm256_storeu_pd(y+i*8,vec_y);
430ef588d5cSRichard Tran Mills     _mm256_storeu_pd(y+i*8+4,vec_y2);
431a48a6482SHong Zhang   }
43221cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
433d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
434d4002b98SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
435d4002b98SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
436d4002b98SHong Zhang 
437d4002b98SHong Zhang     vec_y  = _mm256_setzero_pd();
438d4002b98SHong Zhang     vec_y2 = _mm256_setzero_pd();
439d4002b98SHong Zhang 
440d4002b98SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
441d4002b98SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
442d4002b98SHong Zhang       rows_left = A->rmap->n - 8*i;
443d4002b98SHong Zhang       for (r=0; r<rows_left; ++r) {
444d4002b98SHong Zhang         yval = (MatScalar)0;
445d4002b98SHong Zhang         row = 8*i + r;
446d4002b98SHong Zhang         nnz_in_row = a->rlen[row];
447d4002b98SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j + r] * x[acolidx[8*j + r]];
448d4002b98SHong Zhang         y[row] = yval;
449d4002b98SHong Zhang       }
450d4002b98SHong Zhang       break;
451d4002b98SHong Zhang     }
452d4002b98SHong Zhang 
453d4002b98SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
454a48a6482SHong Zhang     #pragma novector
455a48a6482SHong Zhang     #pragma unroll(2)
4567285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
457d4002b98SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
458d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
459d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
460d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
461d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
462d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
463d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
464d4002b98SHong Zhang       vec_y     = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y);
465d4002b98SHong Zhang       aval     += 4;
466d4002b98SHong Zhang 
467d4002b98SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
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,0);
471d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
472d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
473d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
474d4002b98SHong Zhang       vec_y2    = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2);
475d4002b98SHong Zhang       aval     += 4;
476d4002b98SHong Zhang     }
477d4002b98SHong Zhang 
478d4002b98SHong Zhang     _mm256_storeu_pd(y + i*8,     vec_y);
479d4002b98SHong Zhang     _mm256_storeu_pd(y + i*8 + 4, vec_y2);
480d4002b98SHong Zhang   }
481d4002b98SHong Zhang #else
482d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
483d4002b98SHong Zhang     for (j=0; j<8; j++) sum[j] = 0.0;
484d4002b98SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
485d4002b98SHong Zhang       sum[0] += aval[j] * x[acolidx[j]];
486d4002b98SHong Zhang       sum[1] += aval[j+1] * x[acolidx[j+1]];
487d4002b98SHong Zhang       sum[2] += aval[j+2] * x[acolidx[j+2]];
488d4002b98SHong Zhang       sum[3] += aval[j+3] * x[acolidx[j+3]];
489d4002b98SHong Zhang       sum[4] += aval[j+4] * x[acolidx[j+4]];
490d4002b98SHong Zhang       sum[5] += aval[j+5] * x[acolidx[j+5]];
491d4002b98SHong Zhang       sum[6] += aval[j+6] * x[acolidx[j+6]];
492d4002b98SHong Zhang       sum[7] += aval[j+7] * x[acolidx[j+7]];
493d4002b98SHong Zhang     }
494d4002b98SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
495d4002b98SHong Zhang       for (j=0; j<(A->rmap->n & 0x07); j++) y[8*i+j] = sum[j];
496d4002b98SHong Zhang     } else {
4977285fed1SHong Zhang       for (j=0; j<8; j++) y[8*i+j] = sum[j];
498d4002b98SHong Zhang     }
499d4002b98SHong Zhang   }
500d4002b98SHong Zhang #endif
501d4002b98SHong Zhang 
5025f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz-a->nonzerorowcnt)); /* theoretical minimal FLOPs */
5035f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(xx,&x));
5045f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArray(yy,&y));
505d4002b98SHong Zhang   PetscFunctionReturn(0);
506d4002b98SHong Zhang }
507d4002b98SHong Zhang 
508d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/ftn-kernels/fmultadd.h>
509d4002b98SHong Zhang PetscErrorCode MatMultAdd_SeqSELL(Mat A,Vec xx,Vec yy,Vec zz)
510d4002b98SHong Zhang {
511d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
512d4002b98SHong Zhang   PetscScalar       *y,*z;
513d4002b98SHong Zhang   const PetscScalar *x;
514d4002b98SHong Zhang   const MatScalar   *aval=a->val;
515d4002b98SHong Zhang   PetscInt          totalslices=a->totalslices;
516d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
517d4002b98SHong Zhang   PetscInt          i,j;
518d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5197285fed1SHong Zhang   __m512d           vec_x,vec_y,vec_vals;
520d4002b98SHong Zhang   __m256i           vec_idx;
521d4002b98SHong Zhang   __mmask8          mask;
5227285fed1SHong Zhang   __m512d           vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4;
5237285fed1SHong Zhang   __m256i           vec_idx2,vec_idx3,vec_idx4;
52421cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5257285fed1SHong Zhang   __m128d           vec_x_tmp;
5267285fed1SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
5277285fed1SHong Zhang   MatScalar         yval;
5287285fed1SHong Zhang   PetscInt          r,row,nnz_in_row;
529d4002b98SHong Zhang #else
530d4002b98SHong Zhang   PetscScalar       sum[8];
531d4002b98SHong Zhang #endif
532d4002b98SHong Zhang 
533d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
534d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
535d4002b98SHong Zhang #endif
536d4002b98SHong Zhang 
537d4002b98SHong Zhang   PetscFunctionBegin;
5385f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(xx,&x));
5395f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayPair(yy,zz,&y,&z));
540d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5417285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
5427285fed1SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
5437285fed1SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
5447285fed1SHong Zhang 
545d4002b98SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
546d4002b98SHong Zhang       mask   = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07)));
547ef588d5cSRichard Tran Mills       vec_y  = _mm512_mask_loadu_pd(vec_y,mask,&y[8*i]);
5487285fed1SHong Zhang     } else {
549ef588d5cSRichard Tran Mills       vec_y  = _mm512_loadu_pd(&y[8*i]);
5507285fed1SHong Zhang     }
5517285fed1SHong Zhang     vec_y2 = _mm512_setzero_pd();
5527285fed1SHong Zhang     vec_y3 = _mm512_setzero_pd();
5537285fed1SHong Zhang     vec_y4 = _mm512_setzero_pd();
5547285fed1SHong Zhang 
5557285fed1SHong Zhang     j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */
5567285fed1SHong Zhang     switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) {
5577285fed1SHong Zhang     case 3:
5587285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5597285fed1SHong Zhang       acolidx += 8; aval += 8;
5607285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5617285fed1SHong Zhang       acolidx += 8; aval += 8;
5627285fed1SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
5637285fed1SHong Zhang       acolidx += 8; aval += 8;
5647285fed1SHong Zhang       j += 3;
5657285fed1SHong Zhang       break;
5667285fed1SHong Zhang     case 2:
5677285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5687285fed1SHong Zhang       acolidx += 8; aval += 8;
5697285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5707285fed1SHong Zhang       acolidx += 8; aval += 8;
5717285fed1SHong Zhang       j += 2;
5727285fed1SHong Zhang       break;
5737285fed1SHong Zhang     case 1:
5747285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5757285fed1SHong Zhang       acolidx += 8; aval += 8;
5767285fed1SHong Zhang       j += 1;
5777285fed1SHong Zhang       break;
5787285fed1SHong Zhang     }
5797285fed1SHong Zhang     #pragma novector
5807285fed1SHong Zhang     for (; j<(a->sliidx[i+1]>>3); j+=4) {
5817285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5827285fed1SHong Zhang       acolidx += 8; aval += 8;
5837285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5847285fed1SHong Zhang       acolidx += 8; aval += 8;
5857285fed1SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
5867285fed1SHong Zhang       acolidx += 8; aval += 8;
5877285fed1SHong Zhang       AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4);
5887285fed1SHong Zhang       acolidx += 8; aval += 8;
5897285fed1SHong Zhang     }
5907285fed1SHong Zhang 
5917285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y2);
5927285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y3);
5937285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y4);
5947285fed1SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
595ef588d5cSRichard Tran Mills       _mm512_mask_storeu_pd(&z[8*i],mask,vec_y);
596d4002b98SHong Zhang     } else {
597ef588d5cSRichard Tran Mills       _mm512_storeu_pd(&z[8*i],vec_y);
598d4002b98SHong Zhang     }
5997285fed1SHong Zhang   }
60021cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
6017285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
6027285fed1SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
6037285fed1SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
6047285fed1SHong Zhang 
6057285fed1SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
6067285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6077285fed1SHong Zhang       for (r=0; r<(A->rmap->n & 0x07); ++r) {
6087285fed1SHong Zhang         row        = 8*i + r;
6097285fed1SHong Zhang         yval       = (MatScalar)0.0;
6107285fed1SHong Zhang         nnz_in_row = a->rlen[row];
6117285fed1SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]];
6127285fed1SHong Zhang         z[row] = y[row] + yval;
6137285fed1SHong Zhang       }
6147285fed1SHong Zhang       break;
6157285fed1SHong Zhang     }
6167285fed1SHong Zhang 
6177285fed1SHong Zhang     vec_y  = _mm256_loadu_pd(y+8*i);
6187285fed1SHong Zhang     vec_y2 = _mm256_loadu_pd(y+8*i+4);
6197285fed1SHong Zhang 
6207285fed1SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
6217285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
6227285fed1SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
6237285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6247285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6257285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
6267285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6277285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6287285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
6297285fed1SHong Zhang       vec_y     = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y);
6307285fed1SHong Zhang       aval     += 4;
6317285fed1SHong Zhang 
6327285fed1SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
6337285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6347285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6357285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
6367285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6377285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6387285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
6397285fed1SHong Zhang       vec_y2    = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2);
6407285fed1SHong Zhang       aval     += 4;
6417285fed1SHong Zhang     }
6427285fed1SHong Zhang 
6437285fed1SHong Zhang     _mm256_storeu_pd(z+i*8,vec_y);
6447285fed1SHong Zhang     _mm256_storeu_pd(z+i*8+4,vec_y2);
6457285fed1SHong Zhang   }
646d4002b98SHong Zhang #else
6477285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
6487285fed1SHong Zhang     for (j=0; j<8; j++) sum[j] = 0.0;
649d4002b98SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
650d4002b98SHong Zhang       sum[0] += aval[j] * x[acolidx[j]];
651d4002b98SHong Zhang       sum[1] += aval[j+1] * x[acolidx[j+1]];
652d4002b98SHong Zhang       sum[2] += aval[j+2] * x[acolidx[j+2]];
653d4002b98SHong Zhang       sum[3] += aval[j+3] * x[acolidx[j+3]];
654d4002b98SHong Zhang       sum[4] += aval[j+4] * x[acolidx[j+4]];
655d4002b98SHong Zhang       sum[5] += aval[j+5] * x[acolidx[j+5]];
656d4002b98SHong Zhang       sum[6] += aval[j+6] * x[acolidx[j+6]];
657d4002b98SHong Zhang       sum[7] += aval[j+7] * x[acolidx[j+7]];
658d4002b98SHong Zhang     }
6597285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6607285fed1SHong Zhang       for (j=0; j<(A->rmap->n & 0x07); j++) z[8*i+j] = y[8*i+j] + sum[j];
661d4002b98SHong Zhang     } else {
6627285fed1SHong Zhang       for (j=0; j<8; j++) z[8*i+j] = y[8*i+j] + sum[j];
6637285fed1SHong Zhang     }
664d4002b98SHong Zhang   }
665d4002b98SHong Zhang #endif
666d4002b98SHong Zhang 
6675f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz));
6685f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(xx,&x));
6695f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayPair(yy,zz,&y,&z));
670d4002b98SHong Zhang   PetscFunctionReturn(0);
671d4002b98SHong Zhang }
672d4002b98SHong Zhang 
673d4002b98SHong Zhang PetscErrorCode MatMultTransposeAdd_SeqSELL(Mat A,Vec xx,Vec zz,Vec yy)
674d4002b98SHong Zhang {
675d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
676d4002b98SHong Zhang   PetscScalar       *y;
677d4002b98SHong Zhang   const PetscScalar *x;
678d4002b98SHong Zhang   const MatScalar   *aval=a->val;
679d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
6807285fed1SHong Zhang   PetscInt          i,j,r,row,nnz_in_row,totalslices=a->totalslices;
681d4002b98SHong Zhang 
682d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
683d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
684d4002b98SHong Zhang #endif
685d4002b98SHong Zhang 
686d4002b98SHong Zhang   PetscFunctionBegin;
6879fc32365SStefano Zampini   if (A->symmetric) {
6885f80ce2aSJacob Faibussowitsch     CHKERRQ(MatMultAdd_SeqSELL(A,xx,zz,yy));
6899fc32365SStefano Zampini     PetscFunctionReturn(0);
6909fc32365SStefano Zampini   }
6915f80ce2aSJacob Faibussowitsch   if (zz != yy) CHKERRQ(VecCopy(zz,yy));
6925f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(xx,&x));
6935f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArray(yy,&y));
694d4002b98SHong Zhang   for (i=0; i<a->totalslices; i++) { /* loop over slices */
6957285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6967285fed1SHong Zhang       for (r=0; r<(A->rmap->n & 0x07); ++r) {
6977285fed1SHong Zhang         row        = 8*i + r;
6987285fed1SHong Zhang         nnz_in_row = a->rlen[row];
6997285fed1SHong Zhang         for (j=0; j<nnz_in_row; ++j) y[acolidx[8*j+r]] += aval[8*j+r] * x[row];
7007285fed1SHong Zhang       }
7017285fed1SHong Zhang       break;
7027285fed1SHong Zhang     }
7037285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
7047285fed1SHong Zhang       y[acolidx[j]]   += aval[j] * x[8*i];
7057285fed1SHong Zhang       y[acolidx[j+1]] += aval[j+1] * x[8*i+1];
7067285fed1SHong Zhang       y[acolidx[j+2]] += aval[j+2] * x[8*i+2];
7077285fed1SHong Zhang       y[acolidx[j+3]] += aval[j+3] * x[8*i+3];
7087285fed1SHong Zhang       y[acolidx[j+4]] += aval[j+4] * x[8*i+4];
7097285fed1SHong Zhang       y[acolidx[j+5]] += aval[j+5] * x[8*i+5];
7107285fed1SHong Zhang       y[acolidx[j+6]] += aval[j+6] * x[8*i+6];
7117285fed1SHong Zhang       y[acolidx[j+7]] += aval[j+7] * x[8*i+7];
712d4002b98SHong Zhang     }
713d4002b98SHong Zhang   }
7145f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->sliidx[a->totalslices]));
7155f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(xx,&x));
7165f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArray(yy,&y));
717d4002b98SHong Zhang   PetscFunctionReturn(0);
718d4002b98SHong Zhang }
719d4002b98SHong Zhang 
720d4002b98SHong Zhang PetscErrorCode MatMultTranspose_SeqSELL(Mat A,Vec xx,Vec yy)
721d4002b98SHong Zhang {
722d4002b98SHong Zhang   PetscFunctionBegin;
7239fc32365SStefano Zampini   if (A->symmetric) {
7245f80ce2aSJacob Faibussowitsch     CHKERRQ(MatMult_SeqSELL(A,xx,yy));
7259fc32365SStefano Zampini   } else {
7265f80ce2aSJacob Faibussowitsch     CHKERRQ(VecSet(yy,0.0));
7275f80ce2aSJacob Faibussowitsch     CHKERRQ(MatMultTransposeAdd_SeqSELL(A,xx,yy,yy));
7289fc32365SStefano Zampini   }
729d4002b98SHong Zhang   PetscFunctionReturn(0);
730d4002b98SHong Zhang }
731d4002b98SHong Zhang 
732d4002b98SHong Zhang /*
733d4002b98SHong Zhang      Checks for missing diagonals
734d4002b98SHong Zhang */
735d4002b98SHong Zhang PetscErrorCode MatMissingDiagonal_SeqSELL(Mat A,PetscBool  *missing,PetscInt *d)
736d4002b98SHong Zhang {
737d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
738d4002b98SHong Zhang   PetscInt       *diag,i;
739d4002b98SHong Zhang 
740d4002b98SHong Zhang   PetscFunctionBegin;
741d4002b98SHong Zhang   *missing = PETSC_FALSE;
742d4002b98SHong Zhang   if (A->rmap->n > 0 && !(a->colidx)) {
743d4002b98SHong Zhang     *missing = PETSC_TRUE;
744d4002b98SHong Zhang     if (d) *d = 0;
7455f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n"));
746d4002b98SHong Zhang   } else {
747d4002b98SHong Zhang     diag = a->diag;
748d4002b98SHong Zhang     for (i=0; i<A->rmap->n; i++) {
749d4002b98SHong Zhang       if (diag[i] == -1) {
750d4002b98SHong Zhang         *missing = PETSC_TRUE;
751d4002b98SHong Zhang         if (d) *d = i;
7525f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscInfo(A,"Matrix is missing diagonal number %" PetscInt_FMT "\n",i));
753d4002b98SHong Zhang         break;
754d4002b98SHong Zhang       }
755d4002b98SHong Zhang     }
756d4002b98SHong Zhang   }
757d4002b98SHong Zhang   PetscFunctionReturn(0);
758d4002b98SHong Zhang }
759d4002b98SHong Zhang 
760d4002b98SHong Zhang PetscErrorCode MatMarkDiagonal_SeqSELL(Mat A)
761d4002b98SHong Zhang {
762d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
763d4002b98SHong Zhang   PetscInt       i,j,m=A->rmap->n,shift;
764d4002b98SHong Zhang 
765d4002b98SHong Zhang   PetscFunctionBegin;
766d4002b98SHong Zhang   if (!a->diag) {
7675f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(m,&a->diag));
7685f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogObjectMemory((PetscObject)A,m*sizeof(PetscInt)));
769d4002b98SHong Zhang     a->free_diag = PETSC_TRUE;
770d4002b98SHong Zhang   }
771d4002b98SHong Zhang   for (i=0; i<m; i++) { /* loop over rows */
772d4002b98SHong Zhang     shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
773d4002b98SHong Zhang     a->diag[i] = -1;
774d4002b98SHong Zhang     for (j=0; j<a->rlen[i]; j++) {
775d4002b98SHong Zhang       if (a->colidx[shift+j*8] == i) {
776d4002b98SHong Zhang         a->diag[i] = shift+j*8;
777d4002b98SHong Zhang         break;
778d4002b98SHong Zhang       }
779d4002b98SHong Zhang     }
780d4002b98SHong Zhang   }
781d4002b98SHong Zhang   PetscFunctionReturn(0);
782d4002b98SHong Zhang }
783d4002b98SHong Zhang 
784d4002b98SHong Zhang /*
785d4002b98SHong Zhang   Negative shift indicates do not generate an error if there is a zero diagonal, just invert it anyways
786d4002b98SHong Zhang */
787d4002b98SHong Zhang PetscErrorCode MatInvertDiagonal_SeqSELL(Mat A,PetscScalar omega,PetscScalar fshift)
788d4002b98SHong Zhang {
789d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*) A->data;
790d4002b98SHong Zhang   PetscInt       i,*diag,m = A->rmap->n;
791d4002b98SHong Zhang   MatScalar      *val = a->val;
792d4002b98SHong Zhang   PetscScalar    *idiag,*mdiag;
793d4002b98SHong Zhang 
794d4002b98SHong Zhang   PetscFunctionBegin;
795d4002b98SHong Zhang   if (a->idiagvalid) PetscFunctionReturn(0);
7965f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMarkDiagonal_SeqSELL(A));
797d4002b98SHong Zhang   diag = a->diag;
798d4002b98SHong Zhang   if (!a->idiag) {
7995f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc3(m,&a->idiag,m,&a->mdiag,m,&a->ssor_work));
8005f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogObjectMemory((PetscObject)A, 3*m*sizeof(PetscScalar)));
801d4002b98SHong Zhang     val  = a->val;
802d4002b98SHong Zhang   }
803d4002b98SHong Zhang   mdiag = a->mdiag;
804d4002b98SHong Zhang   idiag = a->idiag;
805d4002b98SHong Zhang 
806d4002b98SHong Zhang   if (omega == 1.0 && PetscRealPart(fshift) <= 0.0) {
807d4002b98SHong Zhang     for (i=0; i<m; i++) {
808d4002b98SHong Zhang       mdiag[i] = val[diag[i]];
809d4002b98SHong Zhang       if (!PetscAbsScalar(mdiag[i])) { /* zero diagonal */
810d4002b98SHong Zhang         if (PetscRealPart(fshift)) {
8115f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscInfo(A,"Zero diagonal on row %" PetscInt_FMT "\n",i));
812d4002b98SHong Zhang           A->factorerrortype             = MAT_FACTOR_NUMERIC_ZEROPIVOT;
813d4002b98SHong Zhang           A->factorerror_zeropivot_value = 0.0;
814d4002b98SHong Zhang           A->factorerror_zeropivot_row   = i;
81598921bdaSJacob Faibussowitsch         } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Zero diagonal on row %" PetscInt_FMT,i);
816d4002b98SHong Zhang       }
817d4002b98SHong Zhang       idiag[i] = 1.0/val[diag[i]];
818d4002b98SHong Zhang     }
8195f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogFlops(m));
820d4002b98SHong Zhang   } else {
821d4002b98SHong Zhang     for (i=0; i<m; i++) {
822d4002b98SHong Zhang       mdiag[i] = val[diag[i]];
823d4002b98SHong Zhang       idiag[i] = omega/(fshift + val[diag[i]]);
824d4002b98SHong Zhang     }
8255f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogFlops(2.0*m));
826d4002b98SHong Zhang   }
827d4002b98SHong Zhang   a->idiagvalid = PETSC_TRUE;
828d4002b98SHong Zhang   PetscFunctionReturn(0);
829d4002b98SHong Zhang }
830d4002b98SHong Zhang 
831d4002b98SHong Zhang PetscErrorCode MatZeroEntries_SeqSELL(Mat A)
832d4002b98SHong Zhang {
833d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
834d4002b98SHong Zhang 
835d4002b98SHong Zhang   PetscFunctionBegin;
8365f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscArrayzero(a->val,a->sliidx[a->totalslices]));
8375f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqSELLInvalidateDiagonal(A));
838d4002b98SHong Zhang   PetscFunctionReturn(0);
839d4002b98SHong Zhang }
840d4002b98SHong Zhang 
841d4002b98SHong Zhang PetscErrorCode MatDestroy_SeqSELL(Mat A)
842d4002b98SHong Zhang {
843d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
844d4002b98SHong Zhang 
845d4002b98SHong Zhang   PetscFunctionBegin;
846d4002b98SHong Zhang #if defined(PETSC_USE_LOG)
847c0aa6a63SJacob Faibussowitsch   PetscLogObjectState((PetscObject)A,"Rows=%" PetscInt_FMT ", Cols=%" PetscInt_FMT ", NZ=%" PetscInt_FMT,A->rmap->n,A->cmap->n,a->nz);
848d4002b98SHong Zhang #endif
8495f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqXSELLFreeSELL(A,&a->val,&a->colidx));
8505f80ce2aSJacob Faibussowitsch   CHKERRQ(ISDestroy(&a->row));
8515f80ce2aSJacob Faibussowitsch   CHKERRQ(ISDestroy(&a->col));
8525f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(a->diag));
8535f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(a->rlen));
8545f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(a->sliidx));
8555f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree3(a->idiag,a->mdiag,a->ssor_work));
8565f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(a->solve_work));
8575f80ce2aSJacob Faibussowitsch   CHKERRQ(ISDestroy(&a->icol));
8585f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(a->saved_values));
8595f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(a->getrowcols,a->getrowvals));
860d4002b98SHong Zhang 
8615f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(A->data));
862d4002b98SHong Zhang 
8635f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectChangeTypeName((PetscObject)A,NULL));
8645f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL));
8655f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL));
8665f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)A,"MatSeqSELLSetPreallocation_C",NULL));
867d4002b98SHong Zhang   PetscFunctionReturn(0);
868d4002b98SHong Zhang }
869d4002b98SHong Zhang 
870d4002b98SHong Zhang PetscErrorCode MatSetOption_SeqSELL(Mat A,MatOption op,PetscBool flg)
871d4002b98SHong Zhang {
872d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
873d4002b98SHong Zhang 
874d4002b98SHong Zhang   PetscFunctionBegin;
875d4002b98SHong Zhang   switch (op) {
876d4002b98SHong Zhang   case MAT_ROW_ORIENTED:
877d4002b98SHong Zhang     a->roworiented = flg;
878d4002b98SHong Zhang     break;
879d4002b98SHong Zhang   case MAT_KEEP_NONZERO_PATTERN:
880d4002b98SHong Zhang     a->keepnonzeropattern = flg;
881d4002b98SHong Zhang     break;
882d4002b98SHong Zhang   case MAT_NEW_NONZERO_LOCATIONS:
883d4002b98SHong Zhang     a->nonew = (flg ? 0 : 1);
884d4002b98SHong Zhang     break;
885d4002b98SHong Zhang   case MAT_NEW_NONZERO_LOCATION_ERR:
886d4002b98SHong Zhang     a->nonew = (flg ? -1 : 0);
887d4002b98SHong Zhang     break;
888d4002b98SHong Zhang   case MAT_NEW_NONZERO_ALLOCATION_ERR:
889d4002b98SHong Zhang     a->nonew = (flg ? -2 : 0);
890d4002b98SHong Zhang     break;
891d4002b98SHong Zhang   case MAT_UNUSED_NONZERO_LOCATION_ERR:
892d4002b98SHong Zhang     a->nounused = (flg ? -1 : 0);
893d4002b98SHong Zhang     break;
8948c78258cSHong Zhang   case MAT_FORCE_DIAGONAL_ENTRIES:
895d4002b98SHong Zhang   case MAT_IGNORE_OFF_PROC_ENTRIES:
896d4002b98SHong Zhang   case MAT_USE_HASH_TABLE:
897071fcb05SBarry Smith   case MAT_SORTED_FULL:
8985f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Option %s ignored\n",MatOptions[op]));
899d4002b98SHong Zhang     break;
900d4002b98SHong Zhang   case MAT_SPD:
901d4002b98SHong Zhang   case MAT_SYMMETRIC:
902d4002b98SHong Zhang   case MAT_STRUCTURALLY_SYMMETRIC:
903d4002b98SHong Zhang   case MAT_HERMITIAN:
904d4002b98SHong Zhang   case MAT_SYMMETRY_ETERNAL:
905d4002b98SHong Zhang     /* These options are handled directly by MatSetOption() */
906d4002b98SHong Zhang     break;
907d4002b98SHong Zhang   default:
90898921bdaSJacob Faibussowitsch     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
909d4002b98SHong Zhang   }
910d4002b98SHong Zhang   PetscFunctionReturn(0);
911d4002b98SHong Zhang }
912d4002b98SHong Zhang 
913d4002b98SHong Zhang PetscErrorCode MatGetDiagonal_SeqSELL(Mat A,Vec v)
914d4002b98SHong Zhang {
915d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
916d4002b98SHong Zhang   PetscInt       i,j,n,shift;
917d4002b98SHong Zhang   PetscScalar    *x,zero=0.0;
918d4002b98SHong Zhang 
919d4002b98SHong Zhang   PetscFunctionBegin;
9205f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetLocalSize(v,&n));
9212c71b3e2SJacob Faibussowitsch   PetscCheckFalse(n != A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
922d4002b98SHong Zhang 
923d4002b98SHong Zhang   if (A->factortype == MAT_FACTOR_ILU || A->factortype == MAT_FACTOR_LU) {
924d4002b98SHong Zhang     PetscInt *diag=a->diag;
9255f80ce2aSJacob Faibussowitsch     CHKERRQ(VecGetArray(v,&x));
926d4002b98SHong Zhang     for (i=0; i<n; i++) x[i] = 1.0/a->val[diag[i]];
9275f80ce2aSJacob Faibussowitsch     CHKERRQ(VecRestoreArray(v,&x));
928d4002b98SHong Zhang     PetscFunctionReturn(0);
929d4002b98SHong Zhang   }
930d4002b98SHong Zhang 
9315f80ce2aSJacob Faibussowitsch   CHKERRQ(VecSet(v,zero));
9325f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArray(v,&x));
933d4002b98SHong Zhang   for (i=0; i<n; i++) { /* loop over rows */
934d4002b98SHong Zhang     shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
935d4002b98SHong Zhang     x[i] = 0;
936d4002b98SHong Zhang     for (j=0; j<a->rlen[i]; j++) {
937d4002b98SHong Zhang       if (a->colidx[shift+j*8] == i) {
938d4002b98SHong Zhang         x[i] = a->val[shift+j*8];
939d4002b98SHong Zhang         break;
940d4002b98SHong Zhang       }
941d4002b98SHong Zhang     }
942d4002b98SHong Zhang   }
9435f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArray(v,&x));
944d4002b98SHong Zhang   PetscFunctionReturn(0);
945d4002b98SHong Zhang }
946d4002b98SHong Zhang 
947d4002b98SHong Zhang PetscErrorCode MatDiagonalScale_SeqSELL(Mat A,Vec ll,Vec rr)
948d4002b98SHong Zhang {
949d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
950d4002b98SHong Zhang   const PetscScalar *l,*r;
951d4002b98SHong Zhang   PetscInt          i,j,m,n,row;
952d4002b98SHong Zhang 
953d4002b98SHong Zhang   PetscFunctionBegin;
954d4002b98SHong Zhang   if (ll) {
955d4002b98SHong Zhang     /* The local size is used so that VecMPI can be passed to this routine
956d4002b98SHong Zhang        by MatDiagonalScale_MPISELL */
9575f80ce2aSJacob Faibussowitsch     CHKERRQ(VecGetLocalSize(ll,&m));
9582c71b3e2SJacob Faibussowitsch     PetscCheckFalse(m != A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length");
9595f80ce2aSJacob Faibussowitsch     CHKERRQ(VecGetArrayRead(ll,&l));
960d4002b98SHong Zhang     for (i=0; i<a->totalslices; i++) { /* loop over slices */
961dab86139SHong Zhang       if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
962dab86139SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
963dab86139SHong Zhang           if (row < (A->rmap->n & 0x07)) a->val[j] *= l[8*i+row];
964dab86139SHong Zhang         }
965dab86139SHong Zhang       } else {
966d4002b98SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
967d4002b98SHong Zhang           a->val[j] *= l[8*i+row];
968d4002b98SHong Zhang         }
969d4002b98SHong Zhang       }
970dab86139SHong Zhang     }
9715f80ce2aSJacob Faibussowitsch     CHKERRQ(VecRestoreArrayRead(ll,&l));
9725f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogFlops(a->nz));
973d4002b98SHong Zhang   }
974d4002b98SHong Zhang   if (rr) {
9755f80ce2aSJacob Faibussowitsch     CHKERRQ(VecGetLocalSize(rr,&n));
9762c71b3e2SJacob Faibussowitsch     PetscCheckFalse(n != A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length");
9775f80ce2aSJacob Faibussowitsch     CHKERRQ(VecGetArrayRead(rr,&r));
978d4002b98SHong Zhang     for (i=0; i<a->totalslices; i++) { /* loop over slices */
979dab86139SHong Zhang       if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
980dab86139SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
981dab86139SHong Zhang           if (row < (A->rmap->n & 0x07)) a->val[j] *= r[a->colidx[j]];
982dab86139SHong Zhang         }
983dab86139SHong Zhang       } else {
984d4002b98SHong Zhang         for (j=a->sliidx[i]; j<a->sliidx[i+1]; j++) {
985d4002b98SHong Zhang           a->val[j] *= r[a->colidx[j]];
986d4002b98SHong Zhang         }
987d4002b98SHong Zhang       }
988dab86139SHong Zhang     }
9895f80ce2aSJacob Faibussowitsch     CHKERRQ(VecRestoreArrayRead(rr,&r));
9905f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogFlops(a->nz));
991d4002b98SHong Zhang   }
9925f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqSELLInvalidateDiagonal(A));
993d4002b98SHong Zhang   PetscFunctionReturn(0);
994d4002b98SHong Zhang }
995d4002b98SHong Zhang 
996d4002b98SHong Zhang PetscErrorCode MatGetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
997d4002b98SHong Zhang {
998d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
999d4002b98SHong Zhang   PetscInt    *cp,i,k,low,high,t,row,col,l;
1000d4002b98SHong Zhang   PetscInt    shift;
1001d4002b98SHong Zhang   MatScalar   *vp;
1002d4002b98SHong Zhang 
1003d4002b98SHong Zhang   PetscFunctionBegin;
100468aafef3SStefano Zampini   for (k=0; k<m; k++) { /* loop over requested rows */
1005d4002b98SHong Zhang     row = im[k];
1006d4002b98SHong Zhang     if (row<0) continue;
10076bdcaf15SBarry Smith     PetscCheck(row < A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->n-1);
1008d4002b98SHong Zhang     shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */
1009d4002b98SHong Zhang     cp = a->colidx+shift; /* pointer to the row */
1010d4002b98SHong Zhang     vp = a->val+shift; /* pointer to the row */
101168aafef3SStefano Zampini     for (l=0; l<n; l++) { /* loop over requested columns */
1012d4002b98SHong Zhang       col = in[l];
1013d4002b98SHong Zhang       if (col<0) continue;
10146bdcaf15SBarry Smith       PetscCheck(col < A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: row %" PetscInt_FMT " max %" PetscInt_FMT,col,A->cmap->n-1);
1015d4002b98SHong Zhang       high = a->rlen[row]; low = 0; /* assume unsorted */
1016d4002b98SHong Zhang       while (high-low > 5) {
1017d4002b98SHong Zhang         t = (low+high)/2;
1018d4002b98SHong Zhang         if (*(cp+t*8) > col) high = t;
1019d4002b98SHong Zhang         else low = t;
1020d4002b98SHong Zhang       }
1021d4002b98SHong Zhang       for (i=low; i<high; i++) {
1022d4002b98SHong Zhang         if (*(cp+8*i) > col) break;
1023d4002b98SHong Zhang         if (*(cp+8*i) == col) {
1024d4002b98SHong Zhang           *v++ = *(vp+8*i);
1025d4002b98SHong Zhang           goto finished;
1026d4002b98SHong Zhang         }
1027d4002b98SHong Zhang       }
1028d4002b98SHong Zhang       *v++ = 0.0;
1029d4002b98SHong Zhang     finished:;
1030d4002b98SHong Zhang     }
1031d4002b98SHong Zhang   }
1032d4002b98SHong Zhang   PetscFunctionReturn(0);
1033d4002b98SHong Zhang }
1034d4002b98SHong Zhang 
1035d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_ASCII(Mat A,PetscViewer viewer)
1036d4002b98SHong Zhang {
1037d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1038d4002b98SHong Zhang   PetscInt          i,j,m=A->rmap->n,shift;
1039d4002b98SHong Zhang   const char        *name;
1040d4002b98SHong Zhang   PetscViewerFormat format;
1041d4002b98SHong Zhang 
1042d4002b98SHong Zhang   PetscFunctionBegin;
10435f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerGetFormat(viewer,&format));
1044d4002b98SHong Zhang   if (format == PETSC_VIEWER_ASCII_MATLAB) {
1045d4002b98SHong Zhang     PetscInt nofinalvalue = 0;
1046d4002b98SHong Zhang     /*
1047d4002b98SHong Zhang     if (m && ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != A->cmap->n-1))) {
1048d4002b98SHong Zhang       nofinalvalue = 1;
1049d4002b98SHong Zhang     }
1050d4002b98SHong Zhang     */
10515f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
10525f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIPrintf(viewer,"%% Size = %" PetscInt_FMT " %" PetscInt_FMT " \n",m,A->cmap->n));
10535f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIPrintf(viewer,"%% Nonzeros = %" PetscInt_FMT " \n",a->nz));
1054d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
10555f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",4);\n",a->nz+nofinalvalue));
1056d4002b98SHong Zhang #else
10575f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",3);\n",a->nz+nofinalvalue));
1058d4002b98SHong Zhang #endif
10595f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIPrintf(viewer,"zzz = [\n"));
1060d4002b98SHong Zhang 
1061d4002b98SHong Zhang     for (i=0; i<m; i++) {
1062d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1063d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1064d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
10655f80ce2aSJacob Faibussowitsch         CHKERRQ(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])));
1066d4002b98SHong Zhang #else
10675f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)a->val[shift+8*j]));
1068d4002b98SHong Zhang #endif
1069d4002b98SHong Zhang       }
1070d4002b98SHong Zhang     }
1071d4002b98SHong Zhang     /*
1072d4002b98SHong Zhang     if (nofinalvalue) {
1073d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
10745f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e %18.16e\n",m,A->cmap->n,0.,0.));
1075d4002b98SHong Zhang #else
10765f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e\n",m,A->cmap->n,0.0));
1077d4002b98SHong Zhang #endif
1078d4002b98SHong Zhang     }
1079d4002b98SHong Zhang     */
10805f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscObjectGetName((PetscObject)A,&name));
10815f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIPrintf(viewer,"];\n %s = spconvert(zzz);\n",name));
10825f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
1083d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO || format == PETSC_VIEWER_ASCII_INFO) {
1084d4002b98SHong Zhang     PetscFunctionReturn(0);
1085d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
10865f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
1087d4002b98SHong Zhang     for (i=0; i<m; i++) {
10885f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i));
1089d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1090d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1091d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1092d4002b98SHong Zhang         if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) {
10935f80ce2aSJacob Faibussowitsch           CHKERRQ(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])));
1094d4002b98SHong Zhang         } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) {
10955f80ce2aSJacob Faibussowitsch           CHKERRQ(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])));
1096d4002b98SHong Zhang         } else if (PetscRealPart(a->val[shift+8*j]) != 0.0) {
10975f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j])));
1098d4002b98SHong Zhang         }
1099d4002b98SHong Zhang #else
11005f80ce2aSJacob Faibussowitsch         if (a->val[shift+8*j] != 0.0) CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]));
1101d4002b98SHong Zhang #endif
1102d4002b98SHong Zhang       }
11035f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscViewerASCIIPrintf(viewer,"\n"));
1104d4002b98SHong Zhang     }
11055f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
1106d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_DENSE) {
1107d4002b98SHong Zhang     PetscInt    cnt=0,jcnt;
1108d4002b98SHong Zhang     PetscScalar value;
1109d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1110d4002b98SHong Zhang     PetscBool   realonly=PETSC_TRUE;
1111d4002b98SHong Zhang     for (i=0; i<a->sliidx[a->totalslices]; i++) {
1112d4002b98SHong Zhang       if (PetscImaginaryPart(a->val[i]) != 0.0) {
1113d4002b98SHong Zhang         realonly = PETSC_FALSE;
1114d4002b98SHong Zhang         break;
1115d4002b98SHong Zhang       }
1116d4002b98SHong Zhang     }
1117d4002b98SHong Zhang #endif
1118d4002b98SHong Zhang 
11195f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
1120d4002b98SHong Zhang     for (i=0; i<m; i++) {
1121d4002b98SHong Zhang       jcnt = 0;
1122d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1123d4002b98SHong Zhang       for (j=0; j<A->cmap->n; j++) {
1124d4002b98SHong Zhang         if (jcnt < a->rlen[i] && j == a->colidx[shift+8*j]) {
1125d4002b98SHong Zhang           value = a->val[cnt++];
1126d4002b98SHong Zhang           jcnt++;
1127d4002b98SHong Zhang         } else {
1128d4002b98SHong Zhang           value = 0.0;
1129d4002b98SHong Zhang         }
1130d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1131d4002b98SHong Zhang         if (realonly) {
11325f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer," %7.5e ",(double)PetscRealPart(value)));
1133d4002b98SHong Zhang         } else {
11345f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer," %7.5e+%7.5e i ",(double)PetscRealPart(value),(double)PetscImaginaryPart(value)));
1135d4002b98SHong Zhang         }
1136d4002b98SHong Zhang #else
11375f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscViewerASCIIPrintf(viewer," %7.5e ",(double)value));
1138d4002b98SHong Zhang #endif
1139d4002b98SHong Zhang       }
11405f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscViewerASCIIPrintf(viewer,"\n"));
1141d4002b98SHong Zhang     }
11425f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
1143d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_MATRIXMARKET) {
1144d4002b98SHong Zhang     PetscInt fshift=1;
11455f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
1146d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
11475f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate complex general\n"));
1148d4002b98SHong Zhang #else
11495f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate real general\n"));
1150d4002b98SHong Zhang #endif
11515f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %" PetscInt_FMT "\n", m, A->cmap->n, a->nz));
1152d4002b98SHong Zhang     for (i=0; i<m; i++) {
1153d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1154d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1155d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
11565f80ce2aSJacob Faibussowitsch         CHKERRQ(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])));
1157d4002b98SHong Zhang #else
11585f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)a->val[shift+8*j]));
1159d4002b98SHong Zhang #endif
1160d4002b98SHong Zhang       }
1161d4002b98SHong Zhang     }
11625f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
116368aafef3SStefano Zampini   } else if (format == PETSC_VIEWER_NATIVE) {
116468aafef3SStefano Zampini     for (i=0; i<a->totalslices; i++) { /* loop over slices */
116568aafef3SStefano Zampini       PetscInt row;
11665f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscViewerASCIIPrintf(viewer,"slice %" PetscInt_FMT ": %" PetscInt_FMT " %" PetscInt_FMT "\n",i,a->sliidx[i],a->sliidx[i+1]));
116768aafef3SStefano Zampini       for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
116868aafef3SStefano Zampini #if defined(PETSC_USE_COMPLEX)
116968aafef3SStefano Zampini         if (PetscImaginaryPart(a->val[j]) > 0.0) {
11705f80ce2aSJacob Faibussowitsch           CHKERRQ(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])));
117168aafef3SStefano Zampini         } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
11725f80ce2aSJacob Faibussowitsch           CHKERRQ(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])));
117368aafef3SStefano Zampini         } else {
11745f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j])));
117568aafef3SStefano Zampini         }
117668aafef3SStefano Zampini #else
11775f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)a->val[j]));
117868aafef3SStefano Zampini #endif
117968aafef3SStefano Zampini       }
118068aafef3SStefano Zampini     }
1181d4002b98SHong Zhang   } else {
11825f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_FALSE));
1183d4002b98SHong Zhang     if (A->factortype) {
1184d4002b98SHong Zhang       for (i=0; i<m; i++) {
1185d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07);
11865f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i));
1187d4002b98SHong Zhang         /* L part */
1188d4002b98SHong Zhang         for (j=shift; j<a->diag[i]; j+=8) {
1189d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1190d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0) {
11915f80ce2aSJacob Faibussowitsch             CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j])));
1192d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0) {
11935f80ce2aSJacob Faibussowitsch             CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j]))));
1194d4002b98SHong Zhang           } else {
11955f80ce2aSJacob Faibussowitsch             CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j])));
1196d4002b98SHong Zhang           }
1197d4002b98SHong Zhang #else
11985f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]));
1199d4002b98SHong Zhang #endif
1200d4002b98SHong Zhang         }
1201d4002b98SHong Zhang         /* diagonal */
1202d4002b98SHong Zhang         j = a->diag[i];
1203d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1204d4002b98SHong Zhang         if (PetscImaginaryPart(a->val[j]) > 0.0) {
12055f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)PetscImaginaryPart(1.0/a->val[j])));
1206d4002b98SHong Zhang         } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
12075f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)(-PetscImaginaryPart(1.0/a->val[j]))));
1208d4002b98SHong Zhang         } else {
12095f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(1.0/a->val[j])));
1210d4002b98SHong Zhang         }
1211d4002b98SHong Zhang #else
12125f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)(1.0/a->val[j])));
1213d4002b98SHong Zhang #endif
1214d4002b98SHong Zhang 
1215d4002b98SHong Zhang         /* U part */
1216d4002b98SHong Zhang         for (j=a->diag[i]+1; j<shift+8*a->rlen[i]; j+=8) {
1217d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1218d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[j]) > 0.0) {
12195f80ce2aSJacob Faibussowitsch             CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j])));
1220d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
12215f80ce2aSJacob Faibussowitsch             CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j]))));
1222d4002b98SHong Zhang           } else {
12235f80ce2aSJacob Faibussowitsch             CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j])));
1224d4002b98SHong Zhang           }
1225d4002b98SHong Zhang #else
12265f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]));
1227d4002b98SHong Zhang #endif
1228d4002b98SHong Zhang         }
12295f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscViewerASCIIPrintf(viewer,"\n"));
1230d4002b98SHong Zhang       }
1231d4002b98SHong Zhang     } else {
1232d4002b98SHong Zhang       for (i=0; i<m; i++) {
1233d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07);
12345f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i));
1235d4002b98SHong Zhang         for (j=0; j<a->rlen[i]; j++) {
1236d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1237d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[j]) > 0.0) {
12385f80ce2aSJacob Faibussowitsch             CHKERRQ(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])));
1239d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
12405f80ce2aSJacob Faibussowitsch             CHKERRQ(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])));
1241d4002b98SHong Zhang           } else {
12425f80ce2aSJacob Faibussowitsch             CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j])));
1243d4002b98SHong Zhang           }
1244d4002b98SHong Zhang #else
12455f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]));
1246d4002b98SHong Zhang #endif
1247d4002b98SHong Zhang         }
12485f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscViewerASCIIPrintf(viewer,"\n"));
1249d4002b98SHong Zhang       }
1250d4002b98SHong Zhang     }
12515f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerASCIIUseTabs(viewer,PETSC_TRUE));
1252d4002b98SHong Zhang   }
12535f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerFlush(viewer));
1254d4002b98SHong Zhang   PetscFunctionReturn(0);
1255d4002b98SHong Zhang }
1256d4002b98SHong Zhang 
1257d4002b98SHong Zhang #include <petscdraw.h>
1258d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw_Zoom(PetscDraw draw,void *Aa)
1259d4002b98SHong Zhang {
1260d4002b98SHong Zhang   Mat               A=(Mat)Aa;
1261d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1262d4002b98SHong Zhang   PetscInt          i,j,m=A->rmap->n,shift;
1263d4002b98SHong Zhang   int               color;
1264d4002b98SHong Zhang   PetscReal         xl,yl,xr,yr,x_l,x_r,y_l,y_r;
1265d4002b98SHong Zhang   PetscViewer       viewer;
1266d4002b98SHong Zhang   PetscViewerFormat format;
1267d4002b98SHong Zhang   PetscErrorCode    ierr;
1268d4002b98SHong Zhang 
1269d4002b98SHong Zhang   PetscFunctionBegin;
12705f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer));
12715f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerGetFormat(viewer,&format));
12725f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr));
1273d4002b98SHong Zhang 
1274d4002b98SHong Zhang   /* loop over matrix elements drawing boxes */
1275d4002b98SHong Zhang 
1276d4002b98SHong Zhang   if (format != PETSC_VIEWER_DRAW_CONTOUR) {
1277d4002b98SHong Zhang     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
1278d4002b98SHong Zhang     /* Blue for negative, Cyan for zero and  Red for positive */
1279d4002b98SHong Zhang     color = PETSC_DRAW_BLUE;
1280d4002b98SHong Zhang     for (i=0; i<m; i++) {
1281d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1282d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1283d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1284d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1285d4002b98SHong Zhang         if (PetscRealPart(a->val[shift+8*j]) >=  0.) continue;
12865f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color));
1287d4002b98SHong Zhang       }
1288d4002b98SHong Zhang     }
1289d4002b98SHong Zhang     color = PETSC_DRAW_CYAN;
1290d4002b98SHong Zhang     for (i=0; i<m; i++) {
1291d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1292d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1293d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1294d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1295d4002b98SHong Zhang         if (a->val[shift+8*j] !=  0.) continue;
12965f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color));
1297d4002b98SHong Zhang       }
1298d4002b98SHong Zhang     }
1299d4002b98SHong Zhang     color = PETSC_DRAW_RED;
1300d4002b98SHong Zhang     for (i=0; i<m; i++) {
1301d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
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;
13065f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color));
1307d4002b98SHong Zhang       }
1308d4002b98SHong Zhang     }
1309d4002b98SHong Zhang     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1310d4002b98SHong Zhang   } else {
1311d4002b98SHong Zhang     /* use contour shading to indicate magnitude of values */
1312d4002b98SHong Zhang     /* first determine max of all nonzero values */
1313d4002b98SHong Zhang     PetscReal minv=0.0,maxv=0.0;
1314d4002b98SHong Zhang     PetscInt  count=0;
1315d4002b98SHong Zhang     PetscDraw popup;
1316d4002b98SHong Zhang     for (i=0; i<a->sliidx[a->totalslices]; i++) {
1317d4002b98SHong Zhang       if (PetscAbsScalar(a->val[i]) > maxv) maxv = PetscAbsScalar(a->val[i]);
1318d4002b98SHong Zhang     }
1319d4002b98SHong Zhang     if (minv >= maxv) maxv = minv + PETSC_SMALL;
13205f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDrawGetPopup(draw,&popup));
13215f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDrawScalePopup(popup,minv,maxv));
1322d4002b98SHong Zhang 
1323d4002b98SHong Zhang     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
1324d4002b98SHong Zhang     for (i=0; i<m; i++) {
1325d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1326d4002b98SHong Zhang       y_l = m - i - 1.0;
1327d4002b98SHong Zhang       y_r = y_l + 1.0;
1328d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1329d4002b98SHong Zhang         x_l = a->colidx[shift+j*8];
1330d4002b98SHong Zhang         x_r = x_l + 1.0;
1331d4002b98SHong Zhang         color = PetscDrawRealToColor(PetscAbsScalar(a->val[count]),minv,maxv);
13325f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color));
1333d4002b98SHong Zhang         count++;
1334d4002b98SHong Zhang       }
1335d4002b98SHong Zhang     }
1336d4002b98SHong Zhang     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1337d4002b98SHong Zhang   }
1338d4002b98SHong Zhang   PetscFunctionReturn(0);
1339d4002b98SHong Zhang }
1340d4002b98SHong Zhang 
1341d4002b98SHong Zhang #include <petscdraw.h>
1342d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw(Mat A,PetscViewer viewer)
1343d4002b98SHong Zhang {
1344d4002b98SHong Zhang   PetscDraw      draw;
1345d4002b98SHong Zhang   PetscReal      xr,yr,xl,yl,h,w;
1346d4002b98SHong Zhang   PetscBool      isnull;
1347d4002b98SHong Zhang 
1348d4002b98SHong Zhang   PetscFunctionBegin;
13495f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerDrawGetDraw(viewer,0,&draw));
13505f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawIsNull(draw,&isnull));
1351d4002b98SHong Zhang   if (isnull) PetscFunctionReturn(0);
1352d4002b98SHong Zhang 
1353d4002b98SHong Zhang   xr   = A->cmap->n; yr  = A->rmap->n; h = yr/10.0; w = xr/10.0;
1354d4002b98SHong Zhang   xr  += w;          yr += h;         xl = -w;     yl = -h;
13555f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawSetCoordinates(draw,xl,yl,xr,yr));
13565f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer));
13575f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawZoom(draw,MatView_SeqSELL_Draw_Zoom,A));
13585f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL));
13595f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawSave(draw));
1360d4002b98SHong Zhang   PetscFunctionReturn(0);
1361d4002b98SHong Zhang }
1362d4002b98SHong Zhang 
1363d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL(Mat A,PetscViewer viewer)
1364d4002b98SHong Zhang {
1365d4002b98SHong Zhang   PetscBool      iascii,isbinary,isdraw;
1366d4002b98SHong Zhang 
1367d4002b98SHong Zhang   PetscFunctionBegin;
13685f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii));
13695f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary));
13705f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw));
1371d4002b98SHong Zhang   if (iascii) {
13725f80ce2aSJacob Faibussowitsch     CHKERRQ(MatView_SeqSELL_ASCII(A,viewer));
1373d4002b98SHong Zhang   } else if (isbinary) {
13745f80ce2aSJacob Faibussowitsch     /* CHKERRQ(MatView_SeqSELL_Binary(A,viewer)); */
1375d4002b98SHong Zhang   } else if (isdraw) {
13765f80ce2aSJacob Faibussowitsch     CHKERRQ(MatView_SeqSELL_Draw(A,viewer));
1377d4002b98SHong Zhang   }
1378d4002b98SHong Zhang   PetscFunctionReturn(0);
1379d4002b98SHong Zhang }
1380d4002b98SHong Zhang 
1381d4002b98SHong Zhang PetscErrorCode MatAssemblyEnd_SeqSELL(Mat A,MatAssemblyType mode)
1382d4002b98SHong Zhang {
1383d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1384d4002b98SHong Zhang   PetscInt       i,shift,row_in_slice,row,nrow,*cp,lastcol,j,k;
1385d4002b98SHong Zhang   MatScalar      *vp;
1386d4002b98SHong Zhang 
1387d4002b98SHong Zhang   PetscFunctionBegin;
1388d4002b98SHong Zhang   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
1389d4002b98SHong Zhang   /* To do: compress out the unused elements */
13905f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMarkDiagonal_SeqSELL(A));
13915f80ce2aSJacob Faibussowitsch   CHKERRQ(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));
13925f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(A,"Number of mallocs during MatSetValues() is %" PetscInt_FMT "\n",a->reallocs));
13935f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(A,"Maximum nonzeros in any row is %" PetscInt_FMT "\n",a->rlenmax));
1394d4002b98SHong 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 */
1395d4002b98SHong Zhang   for (i=0; i<a->totalslices; ++i) {
1396d4002b98SHong Zhang     shift = a->sliidx[i];    /* starting index of the slice */
1397d4002b98SHong Zhang     cp    = a->colidx+shift; /* pointer to the column indices of the slice */
1398d4002b98SHong Zhang     vp    = a->val+shift;    /* pointer to the nonzero values of the slice */
1399d4002b98SHong Zhang     for (row_in_slice=0; row_in_slice<8; ++row_in_slice) { /* loop over rows in the slice */
1400d4002b98SHong Zhang       row  = 8*i + row_in_slice;
1401d4002b98SHong Zhang       nrow = a->rlen[row]; /* number of nonzeros in row */
1402d4002b98SHong Zhang       /*
1403d4002b98SHong Zhang         Search for the nearest nonzero. Normally setting the index to zero may cause extra communication.
1404d4002b98SHong Zhang         But if the entire slice are empty, it is fine to use 0 since the index will not be loaded.
1405d4002b98SHong Zhang       */
1406d4002b98SHong Zhang       lastcol = 0;
1407d4002b98SHong Zhang       if (nrow>0) { /* nonempty row */
1408d4002b98SHong Zhang         lastcol = cp[8*(nrow-1)+row_in_slice]; /* use the index from the last nonzero at current row */
1409d4002b98SHong Zhang       } else if (!row_in_slice) { /* first row of the currect slice is empty */
1410d4002b98SHong Zhang         for (j=1;j<8;j++) {
1411d4002b98SHong Zhang           if (a->rlen[8*i+j]) {
1412d4002b98SHong Zhang             lastcol = cp[j];
1413d4002b98SHong Zhang             break;
1414d4002b98SHong Zhang           }
1415d4002b98SHong Zhang         }
1416d4002b98SHong Zhang       } else {
1417d4002b98SHong Zhang         if (a->sliidx[i+1] != shift) lastcol = cp[row_in_slice-1]; /* use the index from the previous row */
1418d4002b98SHong Zhang       }
1419d4002b98SHong Zhang 
1420d4002b98SHong Zhang       for (k=nrow; k<(a->sliidx[i+1]-shift)/8; ++k) {
1421d4002b98SHong Zhang         cp[8*k+row_in_slice] = lastcol;
1422d4002b98SHong Zhang         vp[8*k+row_in_slice] = (MatScalar)0;
1423d4002b98SHong Zhang       }
1424d4002b98SHong Zhang     }
1425d4002b98SHong Zhang   }
1426d4002b98SHong Zhang 
1427d4002b98SHong Zhang   A->info.mallocs += a->reallocs;
1428d4002b98SHong Zhang   a->reallocs      = 0;
1429d4002b98SHong Zhang 
14305f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqSELLInvalidateDiagonal(A));
1431d4002b98SHong Zhang   PetscFunctionReturn(0);
1432d4002b98SHong Zhang }
1433d4002b98SHong Zhang 
1434d4002b98SHong Zhang PetscErrorCode MatGetInfo_SeqSELL(Mat A,MatInfoType flag,MatInfo *info)
1435d4002b98SHong Zhang {
1436d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1437d4002b98SHong Zhang 
1438d4002b98SHong Zhang   PetscFunctionBegin;
1439d4002b98SHong Zhang   info->block_size   = 1.0;
14403966268fSBarry Smith   info->nz_allocated = a->maxallocmat;
14413966268fSBarry Smith   info->nz_used      = a->sliidx[a->totalslices]; /* include padding zeros */
14423966268fSBarry Smith   info->nz_unneeded  = (a->maxallocmat-a->sliidx[a->totalslices]);
14433966268fSBarry Smith   info->assemblies   = A->num_ass;
14443966268fSBarry Smith   info->mallocs      = A->info.mallocs;
1445d4002b98SHong Zhang   info->memory       = ((PetscObject)A)->mem;
1446d4002b98SHong Zhang   if (A->factortype) {
1447d4002b98SHong Zhang     info->fill_ratio_given  = A->info.fill_ratio_given;
1448d4002b98SHong Zhang     info->fill_ratio_needed = A->info.fill_ratio_needed;
1449d4002b98SHong Zhang     info->factor_mallocs    = A->info.factor_mallocs;
1450d4002b98SHong Zhang   } else {
1451d4002b98SHong Zhang     info->fill_ratio_given  = 0;
1452d4002b98SHong Zhang     info->fill_ratio_needed = 0;
1453d4002b98SHong Zhang     info->factor_mallocs    = 0;
1454d4002b98SHong Zhang   }
1455d4002b98SHong Zhang   PetscFunctionReturn(0);
1456d4002b98SHong Zhang }
1457d4002b98SHong Zhang 
1458d4002b98SHong Zhang PetscErrorCode MatSetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
1459d4002b98SHong Zhang {
1460d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1461d4002b98SHong Zhang   PetscInt       shift,i,k,l,low,high,t,ii,row,col,nrow;
1462d4002b98SHong Zhang   PetscInt       *cp,nonew=a->nonew,lastcol=-1;
1463d4002b98SHong Zhang   MatScalar      *vp,value;
1464d4002b98SHong Zhang 
1465d4002b98SHong Zhang   PetscFunctionBegin;
1466d4002b98SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
1467d4002b98SHong Zhang     row = im[k];
1468d4002b98SHong Zhang     if (row < 0) continue;
14696bdcaf15SBarry Smith     PetscCheck(row < A->rmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->n-1);
1470d4002b98SHong Zhang     shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */
1471d4002b98SHong Zhang     cp    = a->colidx+shift; /* pointer to the row */
1472d4002b98SHong Zhang     vp    = a->val+shift; /* pointer to the row */
1473d4002b98SHong Zhang     nrow  = a->rlen[row];
1474d4002b98SHong Zhang     low   = 0;
1475d4002b98SHong Zhang     high  = nrow;
1476d4002b98SHong Zhang 
1477d4002b98SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
1478d4002b98SHong Zhang       col = in[l];
1479d4002b98SHong Zhang       if (col<0) continue;
14806bdcaf15SBarry Smith       PetscCheck(col < A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Col too large: row %" PetscInt_FMT " max %" PetscInt_FMT,col,A->cmap->n-1);
1481d4002b98SHong Zhang       if (a->roworiented) {
1482d4002b98SHong Zhang         value = v[l+k*n];
1483d4002b98SHong Zhang       } else {
1484d4002b98SHong Zhang         value = v[k+l*m];
1485d4002b98SHong Zhang       }
1486d4002b98SHong Zhang       if ((value == 0.0 && a->ignorezeroentries) && (is == ADD_VALUES)) continue;
1487d4002b98SHong Zhang 
1488ed73aabaSBarry Smith       /* search in this row for the specified column, i indicates the column to be set */
1489d4002b98SHong Zhang       if (col <= lastcol) low = 0;
1490d4002b98SHong Zhang       else high = nrow;
1491d4002b98SHong Zhang       lastcol = col;
1492d4002b98SHong Zhang       while (high-low > 5) {
1493d4002b98SHong Zhang         t = (low+high)/2;
1494d4002b98SHong Zhang         if (*(cp+t*8) > col) high = t;
1495d4002b98SHong Zhang         else low = t;
1496d4002b98SHong Zhang       }
1497d4002b98SHong Zhang       for (i=low; i<high; i++) {
1498d4002b98SHong Zhang         if (*(cp+i*8) > col) break;
1499d4002b98SHong Zhang         if (*(cp+i*8) == col) {
1500d4002b98SHong Zhang           if (is == ADD_VALUES) *(vp+i*8) += value;
1501d4002b98SHong Zhang           else *(vp+i*8) = value;
1502d4002b98SHong Zhang           low = i + 1;
1503d4002b98SHong Zhang           goto noinsert;
1504d4002b98SHong Zhang         }
1505d4002b98SHong Zhang       }
1506d4002b98SHong Zhang       if (value == 0.0 && a->ignorezeroentries) goto noinsert;
1507d4002b98SHong Zhang       if (nonew == 1) goto noinsert;
15082c71b3e2SJacob Faibussowitsch       PetscCheckFalse(nonew == -1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%" PetscInt_FMT ", %" PetscInt_FMT ") in the matrix", row, col);
1509d4002b98SHong Zhang       /* If the current row length exceeds the slice width (e.g. nrow==slice_width), allocate a new space, otherwise do nothing */
1510d4002b98SHong Zhang       MatSeqXSELLReallocateSELL(A,A->rmap->n,1,nrow,a->sliidx,row/8,row,col,a->colidx,a->val,cp,vp,nonew,MatScalar);
1511d4002b98SHong Zhang       /* add the new nonzero to the high position, shift the remaining elements in current row to the right by one slot */
1512d4002b98SHong Zhang       for (ii=nrow-1; ii>=i; ii--) {
1513d4002b98SHong Zhang         *(cp+(ii+1)*8) = *(cp+ii*8);
1514d4002b98SHong Zhang         *(vp+(ii+1)*8) = *(vp+ii*8);
1515d4002b98SHong Zhang       }
1516d4002b98SHong Zhang       a->rlen[row]++;
1517d4002b98SHong Zhang       *(cp+i*8) = col;
1518d4002b98SHong Zhang       *(vp+i*8) = value;
1519d4002b98SHong Zhang       a->nz++;
1520d4002b98SHong Zhang       A->nonzerostate++;
1521d4002b98SHong Zhang       low = i+1; high++; nrow++;
1522d4002b98SHong Zhang noinsert:;
1523d4002b98SHong Zhang     }
1524d4002b98SHong Zhang     a->rlen[row] = nrow;
1525d4002b98SHong Zhang   }
1526d4002b98SHong Zhang   PetscFunctionReturn(0);
1527d4002b98SHong Zhang }
1528d4002b98SHong Zhang 
1529d4002b98SHong Zhang PetscErrorCode MatCopy_SeqSELL(Mat A,Mat B,MatStructure str)
1530d4002b98SHong Zhang {
1531d4002b98SHong Zhang   PetscFunctionBegin;
1532d4002b98SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
1533d4002b98SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
1534d4002b98SHong Zhang     Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1535d4002b98SHong Zhang     Mat_SeqSELL *b=(Mat_SeqSELL*)B->data;
1536d4002b98SHong Zhang 
15372c71b3e2SJacob 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");
15385f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscArraycpy(b->val,a->val,a->sliidx[a->totalslices]));
1539d4002b98SHong Zhang   } else {
15405f80ce2aSJacob Faibussowitsch     CHKERRQ(MatCopy_Basic(A,B,str));
1541d4002b98SHong Zhang   }
1542d4002b98SHong Zhang   PetscFunctionReturn(0);
1543d4002b98SHong Zhang }
1544d4002b98SHong Zhang 
1545d4002b98SHong Zhang PetscErrorCode MatSetUp_SeqSELL(Mat A)
1546d4002b98SHong Zhang {
1547d4002b98SHong Zhang   PetscFunctionBegin;
15485f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqSELLSetPreallocation(A,PETSC_DEFAULT,NULL));
1549d4002b98SHong Zhang   PetscFunctionReturn(0);
1550d4002b98SHong Zhang }
1551d4002b98SHong Zhang 
1552d4002b98SHong Zhang PetscErrorCode MatSeqSELLGetArray_SeqSELL(Mat A,PetscScalar *array[])
1553d4002b98SHong Zhang {
1554d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1555d4002b98SHong Zhang 
1556d4002b98SHong Zhang   PetscFunctionBegin;
1557d4002b98SHong Zhang   *array = a->val;
1558d4002b98SHong Zhang   PetscFunctionReturn(0);
1559d4002b98SHong Zhang }
1560d4002b98SHong Zhang 
1561d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray_SeqSELL(Mat A,PetscScalar *array[])
1562d4002b98SHong Zhang {
1563d4002b98SHong Zhang   PetscFunctionBegin;
1564d4002b98SHong Zhang   PetscFunctionReturn(0);
1565d4002b98SHong Zhang }
1566d4002b98SHong Zhang 
1567d4002b98SHong Zhang PetscErrorCode MatRealPart_SeqSELL(Mat A)
1568d4002b98SHong Zhang {
1569d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1570d4002b98SHong Zhang   PetscInt    i;
1571d4002b98SHong Zhang   MatScalar   *aval=a->val;
1572d4002b98SHong Zhang 
1573d4002b98SHong Zhang   PetscFunctionBegin;
1574d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i]=PetscRealPart(aval[i]);
1575d4002b98SHong Zhang   PetscFunctionReturn(0);
1576d4002b98SHong Zhang }
1577d4002b98SHong Zhang 
1578d4002b98SHong Zhang PetscErrorCode MatImaginaryPart_SeqSELL(Mat A)
1579d4002b98SHong Zhang {
1580d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1581d4002b98SHong Zhang   PetscInt       i;
1582d4002b98SHong Zhang   MatScalar      *aval=a->val;
1583d4002b98SHong Zhang 
1584d4002b98SHong Zhang   PetscFunctionBegin;
1585d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i] = PetscImaginaryPart(aval[i]);
15865f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqSELLInvalidateDiagonal(A));
1587d4002b98SHong Zhang   PetscFunctionReturn(0);
1588d4002b98SHong Zhang }
1589d4002b98SHong Zhang 
1590d4002b98SHong Zhang PetscErrorCode MatScale_SeqSELL(Mat inA,PetscScalar alpha)
1591d4002b98SHong Zhang {
1592d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)inA->data;
1593d4002b98SHong Zhang   MatScalar      *aval=a->val;
1594d4002b98SHong Zhang   PetscScalar    oalpha=alpha;
1595d4002b98SHong Zhang   PetscBLASInt   one=1,size;
1596d4002b98SHong Zhang 
1597d4002b98SHong Zhang   PetscFunctionBegin;
15985f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscBLASIntCast(a->sliidx[a->totalslices],&size));
1599d4002b98SHong Zhang   PetscStackCallBLAS("BLASscal",BLASscal_(&size,&oalpha,aval,&one));
16005f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(a->nz));
16015f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqSELLInvalidateDiagonal(inA));
1602d4002b98SHong Zhang   PetscFunctionReturn(0);
1603d4002b98SHong Zhang }
1604d4002b98SHong Zhang 
1605d4002b98SHong Zhang PetscErrorCode MatShift_SeqSELL(Mat Y,PetscScalar a)
1606d4002b98SHong Zhang {
1607d4002b98SHong Zhang   Mat_SeqSELL    *y=(Mat_SeqSELL*)Y->data;
1608d4002b98SHong Zhang 
1609d4002b98SHong Zhang   PetscFunctionBegin;
1610d4002b98SHong Zhang   if (!Y->preallocated || !y->nz) {
16115f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSeqSELLSetPreallocation(Y,1,NULL));
1612d4002b98SHong Zhang   }
16135f80ce2aSJacob Faibussowitsch   CHKERRQ(MatShift_Basic(Y,a));
1614d4002b98SHong Zhang   PetscFunctionReturn(0);
1615d4002b98SHong Zhang }
1616d4002b98SHong Zhang 
1617d4002b98SHong Zhang PetscErrorCode MatSOR_SeqSELL(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
1618d4002b98SHong Zhang {
1619d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1620d4002b98SHong Zhang   PetscScalar       *x,sum,*t;
1621f4259b30SLisandro Dalcin   const MatScalar   *idiag=NULL,*mdiag;
1622d4002b98SHong Zhang   const PetscScalar *b,*xb;
1623d4002b98SHong Zhang   PetscInt          n,m=A->rmap->n,i,j,shift;
1624d4002b98SHong Zhang   const PetscInt    *diag;
1625d4002b98SHong Zhang 
1626d4002b98SHong Zhang   PetscFunctionBegin;
1627d4002b98SHong Zhang   its = its*lits;
1628d4002b98SHong Zhang 
1629d4002b98SHong Zhang   if (fshift != a->fshift || omega != a->omega) a->idiagvalid = PETSC_FALSE; /* must recompute idiag[] */
16305f80ce2aSJacob Faibussowitsch   if (!a->idiagvalid) CHKERRQ(MatInvertDiagonal_SeqSELL(A,omega,fshift));
1631d4002b98SHong Zhang   a->fshift = fshift;
1632d4002b98SHong Zhang   a->omega  = omega;
1633d4002b98SHong Zhang 
1634d4002b98SHong Zhang   diag  = a->diag;
1635d4002b98SHong Zhang   t     = a->ssor_work;
1636d4002b98SHong Zhang   idiag = a->idiag;
1637d4002b98SHong Zhang   mdiag = a->mdiag;
1638d4002b98SHong Zhang 
16395f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArray(xx,&x));
16405f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
1641d4002b98SHong Zhang   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
16422c71b3e2SJacob Faibussowitsch   PetscCheckFalse(flag == SOR_APPLY_UPPER,PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_UPPER is not implemented");
16432c71b3e2SJacob Faibussowitsch   PetscCheckFalse(flag == SOR_APPLY_LOWER,PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_LOWER is not implemented");
16442c71b3e2SJacob Faibussowitsch   PetscCheckFalse(flag & SOR_EISENSTAT,PETSC_COMM_SELF,PETSC_ERR_SUP,"No support yet for Eisenstat");
1645d4002b98SHong Zhang 
1646d4002b98SHong Zhang   if (flag & SOR_ZERO_INITIAL_GUESS) {
1647d4002b98SHong Zhang     if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) {
1648d4002b98SHong Zhang       for (i=0; i<m; i++) {
1649d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1650d4002b98SHong Zhang         sum   = b[i];
1651d4002b98SHong Zhang         n     = (diag[i]-shift)/8;
1652d4002b98SHong Zhang         for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1653d4002b98SHong Zhang         t[i]  = sum;
1654d4002b98SHong Zhang         x[i]  = sum*idiag[i];
1655d4002b98SHong Zhang       }
1656d4002b98SHong Zhang       xb   = t;
16575f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscLogFlops(a->nz));
1658d4002b98SHong Zhang     } else xb = b;
1659d4002b98SHong Zhang     if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) {
1660d4002b98SHong Zhang       for (i=m-1; i>=0; i--) {
1661d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1662d4002b98SHong Zhang         sum   = xb[i];
1663d4002b98SHong Zhang         n     = a->rlen[i]-(diag[i]-shift)/8-1;
1664d4002b98SHong Zhang         for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1665d4002b98SHong Zhang         if (xb == b) {
1666d4002b98SHong Zhang           x[i] = sum*idiag[i];
1667d4002b98SHong Zhang         } else {
1668d4002b98SHong Zhang           x[i] = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1669d4002b98SHong Zhang         }
1670d4002b98SHong Zhang       }
16715f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscLogFlops(a->nz)); /* assumes 1/2 in upper */
1672d4002b98SHong Zhang     }
1673d4002b98SHong Zhang     its--;
1674d4002b98SHong Zhang   }
1675d4002b98SHong Zhang   while (its--) {
1676d4002b98SHong Zhang     if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) {
1677d4002b98SHong Zhang       for (i=0; i<m; i++) {
1678d4002b98SHong Zhang         /* lower */
1679d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1680d4002b98SHong Zhang         sum   = b[i];
1681d4002b98SHong Zhang         n     = (diag[i]-shift)/8;
1682d4002b98SHong Zhang         for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1683d4002b98SHong Zhang         t[i]  = sum;             /* save application of the lower-triangular part */
1684d4002b98SHong Zhang         /* upper */
1685d4002b98SHong Zhang         n     = a->rlen[i]-(diag[i]-shift)/8-1;
1686d4002b98SHong Zhang         for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1687d4002b98SHong Zhang         x[i]  = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1688d4002b98SHong Zhang       }
1689d4002b98SHong Zhang       xb   = t;
16905f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscLogFlops(2.0*a->nz));
1691d4002b98SHong Zhang     } else xb = b;
1692d4002b98SHong Zhang     if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) {
1693d4002b98SHong Zhang       for (i=m-1; i>=0; i--) {
1694d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1695d4002b98SHong Zhang         sum = xb[i];
1696d4002b98SHong Zhang         if (xb == b) {
1697d4002b98SHong Zhang           /* whole matrix (no checkpointing available) */
1698d4002b98SHong Zhang           n     = a->rlen[i];
1699d4002b98SHong Zhang           for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1700d4002b98SHong Zhang           x[i] = (1.-omega)*x[i]+(sum+mdiag[i]*x[i])*idiag[i];
1701d4002b98SHong Zhang         } else { /* lower-triangular part has been saved, so only apply upper-triangular */
1702d4002b98SHong Zhang           n     = a->rlen[i]-(diag[i]-shift)/8-1;
1703d4002b98SHong Zhang           for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1704d4002b98SHong Zhang           x[i]  = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1705d4002b98SHong Zhang         }
1706d4002b98SHong Zhang       }
1707d4002b98SHong Zhang       if (xb == b) {
17085f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscLogFlops(2.0*a->nz));
1709d4002b98SHong Zhang       } else {
17105f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscLogFlops(a->nz)); /* assumes 1/2 in upper */
1711d4002b98SHong Zhang       }
1712d4002b98SHong Zhang     }
1713d4002b98SHong Zhang   }
17145f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArray(xx,&x));
17155f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
1716d4002b98SHong Zhang   PetscFunctionReturn(0);
1717d4002b98SHong Zhang }
1718d4002b98SHong Zhang 
1719d4002b98SHong Zhang /* -------------------------------------------------------------------*/
1720d4002b98SHong Zhang static struct _MatOps MatOps_Values = {MatSetValues_SeqSELL,
17216108893eSStefano Zampini                                        MatGetRow_SeqSELL,
17226108893eSStefano Zampini                                        MatRestoreRow_SeqSELL,
1723d4002b98SHong Zhang                                        MatMult_SeqSELL,
1724d4002b98SHong Zhang                                /* 4*/  MatMultAdd_SeqSELL,
1725d4002b98SHong Zhang                                        MatMultTranspose_SeqSELL,
1726d4002b98SHong Zhang                                        MatMultTransposeAdd_SeqSELL,
1727f4259b30SLisandro Dalcin                                        NULL,
1728f4259b30SLisandro Dalcin                                        NULL,
1729f4259b30SLisandro Dalcin                                        NULL,
1730f4259b30SLisandro Dalcin                                /* 10*/ NULL,
1731f4259b30SLisandro Dalcin                                        NULL,
1732f4259b30SLisandro Dalcin                                        NULL,
1733d4002b98SHong Zhang                                        MatSOR_SeqSELL,
1734f4259b30SLisandro Dalcin                                        NULL,
1735d4002b98SHong Zhang                                /* 15*/ MatGetInfo_SeqSELL,
1736d4002b98SHong Zhang                                        MatEqual_SeqSELL,
1737d4002b98SHong Zhang                                        MatGetDiagonal_SeqSELL,
1738d4002b98SHong Zhang                                        MatDiagonalScale_SeqSELL,
1739f4259b30SLisandro Dalcin                                        NULL,
1740f4259b30SLisandro Dalcin                                /* 20*/ NULL,
1741d4002b98SHong Zhang                                        MatAssemblyEnd_SeqSELL,
1742d4002b98SHong Zhang                                        MatSetOption_SeqSELL,
1743d4002b98SHong Zhang                                        MatZeroEntries_SeqSELL,
1744f4259b30SLisandro Dalcin                                /* 24*/ NULL,
1745f4259b30SLisandro Dalcin                                        NULL,
1746f4259b30SLisandro Dalcin                                        NULL,
1747f4259b30SLisandro Dalcin                                        NULL,
1748f4259b30SLisandro Dalcin                                        NULL,
1749d4002b98SHong Zhang                                /* 29*/ MatSetUp_SeqSELL,
1750f4259b30SLisandro Dalcin                                        NULL,
1751f4259b30SLisandro Dalcin                                        NULL,
1752f4259b30SLisandro Dalcin                                        NULL,
1753f4259b30SLisandro Dalcin                                        NULL,
1754d4002b98SHong Zhang                                /* 34*/ MatDuplicate_SeqSELL,
1755f4259b30SLisandro Dalcin                                        NULL,
1756f4259b30SLisandro Dalcin                                        NULL,
1757f4259b30SLisandro Dalcin                                        NULL,
1758f4259b30SLisandro Dalcin                                        NULL,
1759f4259b30SLisandro Dalcin                                /* 39*/ NULL,
1760f4259b30SLisandro Dalcin                                        NULL,
1761f4259b30SLisandro Dalcin                                        NULL,
1762d4002b98SHong Zhang                                        MatGetValues_SeqSELL,
1763d4002b98SHong Zhang                                        MatCopy_SeqSELL,
1764f4259b30SLisandro Dalcin                                /* 44*/ NULL,
1765d4002b98SHong Zhang                                        MatScale_SeqSELL,
1766d4002b98SHong Zhang                                        MatShift_SeqSELL,
1767f4259b30SLisandro Dalcin                                        NULL,
1768f4259b30SLisandro Dalcin                                        NULL,
1769f4259b30SLisandro Dalcin                                /* 49*/ NULL,
1770f4259b30SLisandro Dalcin                                        NULL,
1771f4259b30SLisandro Dalcin                                        NULL,
1772f4259b30SLisandro Dalcin                                        NULL,
1773f4259b30SLisandro Dalcin                                        NULL,
1774d4002b98SHong Zhang                                /* 54*/ MatFDColoringCreate_SeqXAIJ,
1775f4259b30SLisandro Dalcin                                        NULL,
1776f4259b30SLisandro Dalcin                                        NULL,
1777f4259b30SLisandro Dalcin                                        NULL,
1778f4259b30SLisandro Dalcin                                        NULL,
1779f4259b30SLisandro Dalcin                                /* 59*/ NULL,
1780d4002b98SHong Zhang                                        MatDestroy_SeqSELL,
1781d4002b98SHong Zhang                                        MatView_SeqSELL,
1782f4259b30SLisandro Dalcin                                        NULL,
1783f4259b30SLisandro Dalcin                                        NULL,
1784f4259b30SLisandro Dalcin                                /* 64*/ NULL,
1785f4259b30SLisandro Dalcin                                        NULL,
1786f4259b30SLisandro Dalcin                                        NULL,
1787f4259b30SLisandro Dalcin                                        NULL,
1788f4259b30SLisandro Dalcin                                        NULL,
1789f4259b30SLisandro Dalcin                                /* 69*/ NULL,
1790f4259b30SLisandro Dalcin                                        NULL,
1791f4259b30SLisandro Dalcin                                        NULL,
1792f4259b30SLisandro Dalcin                                        NULL,
1793f4259b30SLisandro Dalcin                                        NULL,
1794f4259b30SLisandro Dalcin                                /* 74*/ NULL,
1795d4002b98SHong Zhang                                        MatFDColoringApply_AIJ, /* reuse the FDColoring function for AIJ */
1796f4259b30SLisandro Dalcin                                        NULL,
1797f4259b30SLisandro Dalcin                                        NULL,
1798f4259b30SLisandro Dalcin                                        NULL,
1799f4259b30SLisandro Dalcin                                /* 79*/ NULL,
1800f4259b30SLisandro Dalcin                                        NULL,
1801f4259b30SLisandro Dalcin                                        NULL,
1802f4259b30SLisandro Dalcin                                        NULL,
1803f4259b30SLisandro Dalcin                                        NULL,
1804f4259b30SLisandro Dalcin                                /* 84*/ NULL,
1805f4259b30SLisandro Dalcin                                        NULL,
1806f4259b30SLisandro Dalcin                                        NULL,
1807f4259b30SLisandro Dalcin                                        NULL,
1808f4259b30SLisandro Dalcin                                        NULL,
1809f4259b30SLisandro Dalcin                                /* 89*/ NULL,
1810f4259b30SLisandro Dalcin                                        NULL,
1811f4259b30SLisandro Dalcin                                        NULL,
1812f4259b30SLisandro Dalcin                                        NULL,
1813f4259b30SLisandro Dalcin                                        NULL,
1814f4259b30SLisandro Dalcin                                /* 94*/ NULL,
1815f4259b30SLisandro Dalcin                                        NULL,
1816f4259b30SLisandro Dalcin                                        NULL,
1817f4259b30SLisandro Dalcin                                        NULL,
1818f4259b30SLisandro Dalcin                                        NULL,
1819f4259b30SLisandro Dalcin                                /* 99*/ NULL,
1820f4259b30SLisandro Dalcin                                        NULL,
1821f4259b30SLisandro Dalcin                                        NULL,
1822d4002b98SHong Zhang                                        MatConjugate_SeqSELL,
1823f4259b30SLisandro Dalcin                                        NULL,
1824f4259b30SLisandro Dalcin                                /*104*/ NULL,
1825f4259b30SLisandro Dalcin                                        NULL,
1826f4259b30SLisandro Dalcin                                        NULL,
1827f4259b30SLisandro Dalcin                                        NULL,
1828f4259b30SLisandro Dalcin                                        NULL,
1829f4259b30SLisandro Dalcin                                /*109*/ NULL,
1830f4259b30SLisandro Dalcin                                        NULL,
1831f4259b30SLisandro Dalcin                                        NULL,
1832f4259b30SLisandro Dalcin                                        NULL,
1833d4002b98SHong Zhang                                        MatMissingDiagonal_SeqSELL,
1834f4259b30SLisandro Dalcin                                /*114*/ NULL,
1835f4259b30SLisandro Dalcin                                        NULL,
1836f4259b30SLisandro Dalcin                                        NULL,
1837f4259b30SLisandro Dalcin                                        NULL,
1838f4259b30SLisandro Dalcin                                        NULL,
1839f4259b30SLisandro Dalcin                                /*119*/ NULL,
1840f4259b30SLisandro Dalcin                                        NULL,
1841f4259b30SLisandro Dalcin                                        NULL,
1842f4259b30SLisandro Dalcin                                        NULL,
1843f4259b30SLisandro Dalcin                                        NULL,
1844f4259b30SLisandro Dalcin                                /*124*/ NULL,
1845f4259b30SLisandro Dalcin                                        NULL,
1846f4259b30SLisandro Dalcin                                        NULL,
1847f4259b30SLisandro Dalcin                                        NULL,
1848f4259b30SLisandro Dalcin                                        NULL,
1849f4259b30SLisandro Dalcin                                /*129*/ NULL,
1850f4259b30SLisandro Dalcin                                        NULL,
1851f4259b30SLisandro Dalcin                                        NULL,
1852f4259b30SLisandro Dalcin                                        NULL,
1853f4259b30SLisandro Dalcin                                        NULL,
1854f4259b30SLisandro Dalcin                                /*134*/ NULL,
1855f4259b30SLisandro Dalcin                                        NULL,
1856f4259b30SLisandro Dalcin                                        NULL,
1857f4259b30SLisandro Dalcin                                        NULL,
1858f4259b30SLisandro Dalcin                                        NULL,
1859f4259b30SLisandro Dalcin                                /*139*/ NULL,
1860f4259b30SLisandro Dalcin                                        NULL,
1861f4259b30SLisandro Dalcin                                        NULL,
1862d4002b98SHong Zhang                                        MatFDColoringSetUp_SeqXAIJ,
1863f4259b30SLisandro Dalcin                                        NULL,
1864d70f29a3SPierre Jolivet                                /*144*/ NULL,
1865d70f29a3SPierre Jolivet                                        NULL,
1866d70f29a3SPierre Jolivet                                        NULL,
1867d70f29a3SPierre Jolivet                                        NULL
1868d4002b98SHong Zhang };
1869d4002b98SHong Zhang 
1870d4002b98SHong Zhang PetscErrorCode MatStoreValues_SeqSELL(Mat mat)
1871d4002b98SHong Zhang {
1872d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)mat->data;
1873d4002b98SHong Zhang 
1874d4002b98SHong Zhang   PetscFunctionBegin;
1875*28b400f6SJacob Faibussowitsch   PetscCheck(a->nonew,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1876d4002b98SHong Zhang 
1877d4002b98SHong Zhang   /* allocate space for values if not already there */
1878d4002b98SHong Zhang   if (!a->saved_values) {
18795f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(a->sliidx[a->totalslices]+1,&a->saved_values));
18805f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogObjectMemory((PetscObject)mat,(a->sliidx[a->totalslices]+1)*sizeof(PetscScalar)));
1881d4002b98SHong Zhang   }
1882d4002b98SHong Zhang 
1883d4002b98SHong Zhang   /* copy values over */
18845f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscArraycpy(a->saved_values,a->val,a->sliidx[a->totalslices]));
1885d4002b98SHong Zhang   PetscFunctionReturn(0);
1886d4002b98SHong Zhang }
1887d4002b98SHong Zhang 
1888d4002b98SHong Zhang PetscErrorCode MatRetrieveValues_SeqSELL(Mat mat)
1889d4002b98SHong Zhang {
1890d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)mat->data;
1891d4002b98SHong Zhang 
1892d4002b98SHong Zhang   PetscFunctionBegin;
1893*28b400f6SJacob Faibussowitsch   PetscCheck(a->nonew,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1894*28b400f6SJacob Faibussowitsch   PetscCheck(a->saved_values,PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
18955f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscArraycpy(a->val,a->saved_values,a->sliidx[a->totalslices]));
1896d4002b98SHong Zhang   PetscFunctionReturn(0);
1897d4002b98SHong Zhang }
1898d4002b98SHong Zhang 
1899d4002b98SHong Zhang /*@C
1900d4002b98SHong Zhang  MatSeqSELLRestoreArray - returns access to the array where the data for a MATSEQSELL matrix is stored obtained by MatSeqSELLGetArray()
1901d4002b98SHong Zhang 
1902d4002b98SHong Zhang  Not Collective
1903d4002b98SHong Zhang 
1904d4002b98SHong Zhang  Input Parameters:
1905d4002b98SHong Zhang  .  mat - a MATSEQSELL matrix
1906d4002b98SHong Zhang  .  array - pointer to the data
1907d4002b98SHong Zhang 
1908d4002b98SHong Zhang  Level: intermediate
1909d4002b98SHong Zhang 
1910d4002b98SHong Zhang  .seealso: MatSeqSELLGetArray(), MatSeqSELLRestoreArrayF90()
1911d4002b98SHong Zhang  @*/
1912d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray(Mat A,PetscScalar **array)
1913d4002b98SHong Zhang {
1914d4002b98SHong Zhang   PetscFunctionBegin;
19155f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscUseMethod(A,"MatSeqSELLRestoreArray_C",(Mat,PetscScalar**),(A,array)));
1916d4002b98SHong Zhang   PetscFunctionReturn(0);
1917d4002b98SHong Zhang }
1918d4002b98SHong Zhang 
1919d4002b98SHong Zhang PETSC_EXTERN PetscErrorCode MatCreate_SeqSELL(Mat B)
1920d4002b98SHong Zhang {
1921d4002b98SHong Zhang   Mat_SeqSELL    *b;
1922d4002b98SHong Zhang   PetscMPIInt    size;
1923d4002b98SHong Zhang 
1924d4002b98SHong Zhang   PetscFunctionBegin;
19255f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscCitationsRegister(citation,&cited));
19265f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_size(PetscObjectComm((PetscObject)B),&size));
19272c71b3e2SJacob Faibussowitsch   PetscCheckFalse(size > 1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Comm must be of size 1");
1928d4002b98SHong Zhang 
19295f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscNewLog(B,&b));
1930d4002b98SHong Zhang 
1931d4002b98SHong Zhang   B->data = (void*)b;
1932d4002b98SHong Zhang 
19335f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps)));
1934d4002b98SHong Zhang 
1935f4259b30SLisandro Dalcin   b->row                = NULL;
1936f4259b30SLisandro Dalcin   b->col                = NULL;
1937f4259b30SLisandro Dalcin   b->icol               = NULL;
1938d4002b98SHong Zhang   b->reallocs           = 0;
1939d4002b98SHong Zhang   b->ignorezeroentries  = PETSC_FALSE;
1940d4002b98SHong Zhang   b->roworiented        = PETSC_TRUE;
1941d4002b98SHong Zhang   b->nonew              = 0;
1942f4259b30SLisandro Dalcin   b->diag               = NULL;
1943f4259b30SLisandro Dalcin   b->solve_work         = NULL;
1944f4259b30SLisandro Dalcin   B->spptr              = NULL;
1945f4259b30SLisandro Dalcin   b->saved_values       = NULL;
1946f4259b30SLisandro Dalcin   b->idiag              = NULL;
1947f4259b30SLisandro Dalcin   b->mdiag              = NULL;
1948f4259b30SLisandro Dalcin   b->ssor_work          = NULL;
1949d4002b98SHong Zhang   b->omega              = 1.0;
1950d4002b98SHong Zhang   b->fshift             = 0.0;
1951d4002b98SHong Zhang   b->idiagvalid         = PETSC_FALSE;
1952d4002b98SHong Zhang   b->keepnonzeropattern = PETSC_FALSE;
1953d4002b98SHong Zhang 
19545f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectChangeTypeName((PetscObject)B,MATSEQSELL));
19555f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLGetArray_C",MatSeqSELLGetArray_SeqSELL));
19565f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLRestoreArray_C",MatSeqSELLRestoreArray_SeqSELL));
19575f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSELL));
19585f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSELL));
19595f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLSetPreallocation_C",MatSeqSELLSetPreallocation_SeqSELL));
19605f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsell_seqaij_C",MatConvert_SeqSELL_SeqAIJ));
1961d4002b98SHong Zhang   PetscFunctionReturn(0);
1962d4002b98SHong Zhang }
1963d4002b98SHong Zhang 
1964d4002b98SHong Zhang /*
1965d4002b98SHong Zhang  Given a matrix generated with MatGetFactor() duplicates all the information in A into B
1966d4002b98SHong Zhang  */
1967d4002b98SHong Zhang PetscErrorCode MatDuplicateNoCreate_SeqSELL(Mat C,Mat A,MatDuplicateOption cpvalues,PetscBool mallocmatspace)
1968d4002b98SHong Zhang {
1969ed73aabaSBarry Smith   Mat_SeqSELL    *c = (Mat_SeqSELL*)C->data,*a = (Mat_SeqSELL*)A->data;
1970d4002b98SHong Zhang   PetscInt       i,m=A->rmap->n;
1971d4002b98SHong Zhang   PetscInt       totalslices=a->totalslices;
1972d4002b98SHong Zhang 
1973d4002b98SHong Zhang   PetscFunctionBegin;
1974d4002b98SHong Zhang   C->factortype = A->factortype;
1975f4259b30SLisandro Dalcin   c->row        = NULL;
1976f4259b30SLisandro Dalcin   c->col        = NULL;
1977f4259b30SLisandro Dalcin   c->icol       = NULL;
1978d4002b98SHong Zhang   c->reallocs   = 0;
1979d4002b98SHong Zhang   C->assembled = PETSC_TRUE;
1980d4002b98SHong Zhang 
19815f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLayoutReference(A->rmap,&C->rmap));
19825f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLayoutReference(A->cmap,&C->cmap));
1983d4002b98SHong Zhang 
19845f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(8*totalslices,&c->rlen));
19855f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt)));
19865f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(totalslices+1,&c->sliidx));
19875f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)C, (totalslices+1)*sizeof(PetscInt)));
1988d4002b98SHong Zhang 
1989d4002b98SHong Zhang   for (i=0; i<m; i++) c->rlen[i] = a->rlen[i];
1990d4002b98SHong Zhang   for (i=0; i<totalslices+1; i++) c->sliidx[i] = a->sliidx[i];
1991d4002b98SHong Zhang 
1992d4002b98SHong Zhang   /* allocate the matrix space */
1993d4002b98SHong Zhang   if (mallocmatspace) {
19945f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc2(a->maxallocmat,&c->val,a->maxallocmat,&c->colidx));
19955f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogObjectMemory((PetscObject)C,a->maxallocmat*(sizeof(PetscScalar)+sizeof(PetscInt))));
1996d4002b98SHong Zhang 
1997d4002b98SHong Zhang     c->singlemalloc = PETSC_TRUE;
1998d4002b98SHong Zhang 
1999d4002b98SHong Zhang     if (m > 0) {
20005f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscArraycpy(c->colidx,a->colidx,a->maxallocmat));
2001d4002b98SHong Zhang       if (cpvalues == MAT_COPY_VALUES) {
20025f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscArraycpy(c->val,a->val,a->maxallocmat));
2003d4002b98SHong Zhang       } else {
20045f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscArrayzero(c->val,a->maxallocmat));
2005d4002b98SHong Zhang       }
2006d4002b98SHong Zhang     }
2007d4002b98SHong Zhang   }
2008d4002b98SHong Zhang 
2009d4002b98SHong Zhang   c->ignorezeroentries = a->ignorezeroentries;
2010d4002b98SHong Zhang   c->roworiented       = a->roworiented;
2011d4002b98SHong Zhang   c->nonew             = a->nonew;
2012d4002b98SHong Zhang   if (a->diag) {
20135f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(m,&c->diag));
20145f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt)));
2015d4002b98SHong Zhang     for (i=0; i<m; i++) {
2016d4002b98SHong Zhang       c->diag[i] = a->diag[i];
2017d4002b98SHong Zhang     }
2018f4259b30SLisandro Dalcin   } else c->diag = NULL;
2019d4002b98SHong Zhang 
2020f4259b30SLisandro Dalcin   c->solve_work         = NULL;
2021f4259b30SLisandro Dalcin   c->saved_values       = NULL;
2022f4259b30SLisandro Dalcin   c->idiag              = NULL;
2023f4259b30SLisandro Dalcin   c->ssor_work          = NULL;
2024d4002b98SHong Zhang   c->keepnonzeropattern = a->keepnonzeropattern;
2025d4002b98SHong Zhang   c->free_val           = PETSC_TRUE;
2026d4002b98SHong Zhang   c->free_colidx        = PETSC_TRUE;
2027d4002b98SHong Zhang 
2028d4002b98SHong Zhang   c->maxallocmat  = a->maxallocmat;
2029d4002b98SHong Zhang   c->maxallocrow  = a->maxallocrow;
2030d4002b98SHong Zhang   c->rlenmax      = a->rlenmax;
2031d4002b98SHong Zhang   c->nz           = a->nz;
2032d4002b98SHong Zhang   C->preallocated = PETSC_TRUE;
2033d4002b98SHong Zhang 
2034d4002b98SHong Zhang   c->nonzerorowcnt = a->nonzerorowcnt;
2035d4002b98SHong Zhang   C->nonzerostate  = A->nonzerostate;
2036d4002b98SHong Zhang 
20375f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist));
2038d4002b98SHong Zhang   PetscFunctionReturn(0);
2039d4002b98SHong Zhang }
2040d4002b98SHong Zhang 
2041d4002b98SHong Zhang PetscErrorCode MatDuplicate_SeqSELL(Mat A,MatDuplicateOption cpvalues,Mat *B)
2042d4002b98SHong Zhang {
2043d4002b98SHong Zhang   PetscFunctionBegin;
20445f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreate(PetscObjectComm((PetscObject)A),B));
20455f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n));
2046d4002b98SHong Zhang   if (!(A->rmap->n % A->rmap->bs) && !(A->cmap->n % A->cmap->bs)) {
20475f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetBlockSizesFromMats(*B,A,A));
2048d4002b98SHong Zhang   }
20495f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetType(*B,((PetscObject)A)->type_name));
20505f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDuplicateNoCreate_SeqSELL(*B,A,cpvalues,PETSC_TRUE));
2051d4002b98SHong Zhang   PetscFunctionReturn(0);
2052d4002b98SHong Zhang }
2053d4002b98SHong Zhang 
2054ed73aabaSBarry Smith /*MC
2055ed73aabaSBarry Smith    MATSEQSELL - MATSEQSELL = "seqsell" - A matrix type to be used for sequential sparse matrices,
2056ed73aabaSBarry Smith    based on the sliced Ellpack format
2057ed73aabaSBarry Smith 
2058ed73aabaSBarry Smith    Options Database Keys:
2059ed73aabaSBarry Smith . -mat_type seqsell - sets the matrix type to "seqsell" during a call to MatSetFromOptions()
2060ed73aabaSBarry Smith 
2061ed73aabaSBarry Smith    Level: beginner
2062ed73aabaSBarry Smith 
2063ed73aabaSBarry Smith .seealso: MatCreateSeqSell(), MATSELL, MATMPISELL, MATSEQAIJ, MATAIJ, MATMPIAIJ
2064ed73aabaSBarry Smith M*/
2065ed73aabaSBarry Smith 
2066ed73aabaSBarry Smith /*MC
2067ed73aabaSBarry Smith    MATSELL - MATSELL = "sell" - A matrix type to be used for sparse matrices.
2068ed73aabaSBarry Smith 
2069ed73aabaSBarry Smith    This matrix type is identical to MATSEQSELL when constructed with a single process communicator,
2070ed73aabaSBarry Smith    and MATMPISELL otherwise.  As a result, for single process communicators,
2071ed73aabaSBarry Smith   MatSeqSELLSetPreallocation() is supported, and similarly MatMPISELLSetPreallocation() is supported
2072ed73aabaSBarry Smith   for communicators controlling multiple processes.  It is recommended that you call both of
2073ed73aabaSBarry Smith   the above preallocation routines for simplicity.
2074ed73aabaSBarry Smith 
2075ed73aabaSBarry Smith    Options Database Keys:
2076ed73aabaSBarry Smith . -mat_type sell - sets the matrix type to "sell" during a call to MatSetFromOptions()
2077ed73aabaSBarry Smith 
2078ed73aabaSBarry Smith   Level: beginner
2079ed73aabaSBarry Smith 
2080ed73aabaSBarry Smith   Notes:
2081ed73aabaSBarry Smith    This format is only supported for real scalars, double precision, and 32 bit indices (the defaults).
2082ed73aabaSBarry Smith 
2083ed73aabaSBarry Smith    It can provide better performance on Intel and AMD processes with AVX2 or AVX512 support for matrices that have a similar number of
2084ed73aabaSBarry Smith    non-zeros in contiguous groups of rows. However if the computation is memory bandwidth limited it may not provide much improvement.
2085ed73aabaSBarry Smith 
2086ed73aabaSBarry Smith   Developer Notes:
2087ed73aabaSBarry Smith    On Intel (and AMD) systems some of the matrix operations use SIMD (AVX) instructions to achieve higher performance.
2088ed73aabaSBarry Smith 
2089ed73aabaSBarry Smith    The sparse matrix format is as follows. For simplicity we assume a slice size of 2, it is actually 8
2090ed73aabaSBarry Smith .vb
2091ed73aabaSBarry Smith                             (2 0  3 4)
2092ed73aabaSBarry Smith    Consider the matrix A =  (5 0  6 0)
2093ed73aabaSBarry Smith                             (0 0  7 8)
2094ed73aabaSBarry Smith                             (0 0  9 9)
2095ed73aabaSBarry Smith 
2096ed73aabaSBarry Smith    symbolically the Ellpack format can be written as
2097ed73aabaSBarry Smith 
2098ed73aabaSBarry Smith         (2 3 4 |)           (0 2 3 |)
2099ed73aabaSBarry Smith    v =  (5 6 0 |)  colidx = (0 2 2 |)
2100ed73aabaSBarry Smith         --------            ---------
2101ed73aabaSBarry Smith         (7 8 |)             (2 3 |)
2102ed73aabaSBarry Smith         (9 9 |)             (2 3 |)
2103ed73aabaSBarry Smith 
2104ed73aabaSBarry 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).
2105ed73aabaSBarry 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
2106ed73aabaSBarry 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.
2107ed73aabaSBarry Smith 
2108ed73aabaSBarry 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)
2109ed73aabaSBarry Smith 
2110ed73aabaSBarry Smith .ve
2111ed73aabaSBarry Smith 
2112ed73aabaSBarry Smith       See MatMult_SeqSELL() for how this format is used with the SIMD operations to achieve high performance.
2113ed73aabaSBarry Smith 
2114ed73aabaSBarry Smith  References:
2115606c0280SSatish Balay . * - Hong Zhang, Richard T. Mills, Karl Rupp, and Barry F. Smith, Vectorized Parallel Sparse Matrix-Vector Multiplication in {PETSc} Using {AVX-512},
2116ed73aabaSBarry Smith    Proceedings of the 47th International Conference on Parallel Processing, 2018.
2117ed73aabaSBarry Smith 
2118ed73aabaSBarry Smith .seealso: MatCreateSeqSELL(), MatCreateSeqAIJ(), MatCreateSell(), MATSEQSELL, MATMPISELL, MATSEQAIJ, MATMPIAIJ, MATAIJ
2119ed73aabaSBarry Smith M*/
2120ed73aabaSBarry Smith 
2121d4002b98SHong Zhang /*@C
2122d4002b98SHong Zhang        MatCreateSeqSELL - Creates a sparse matrix in SELL format.
2123d4002b98SHong Zhang 
2124ed73aabaSBarry Smith  Collective on comm
2125d4002b98SHong Zhang 
2126d4002b98SHong Zhang  Input Parameters:
2127d4002b98SHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2128d4002b98SHong Zhang .  m - number of rows
2129d4002b98SHong Zhang .  n - number of columns
2130d4002b98SHong Zhang .  rlenmax - maximum number of nonzeros in a row
2131d4002b98SHong Zhang -  rlen - array containing the number of nonzeros in the various rows
2132d4002b98SHong Zhang  (possibly different for each row) or NULL
2133d4002b98SHong Zhang 
2134d4002b98SHong Zhang  Output Parameter:
2135d4002b98SHong Zhang .  A - the matrix
2136d4002b98SHong Zhang 
2137d4002b98SHong Zhang  It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2138f6f02116SRichard Tran Mills  MatXXXXSetPreallocation() paradigm instead of this routine directly.
2139d4002b98SHong Zhang  [MatXXXXSetPreallocation() is, for example, MatSeqSELLSetPreallocation]
2140d4002b98SHong Zhang 
2141d4002b98SHong Zhang  Notes:
2142d4002b98SHong Zhang  If nnz is given then nz is ignored
2143d4002b98SHong Zhang 
2144d4002b98SHong Zhang  Specify the preallocated storage with either rlenmax or rlen (not both).
2145d4002b98SHong Zhang  Set rlenmax=PETSC_DEFAULT and rlen=NULL for PETSc to control dynamic memory
2146d4002b98SHong Zhang  allocation.  For large problems you MUST preallocate memory or you
2147d4002b98SHong Zhang  will get TERRIBLE performance, see the users' manual chapter on matrices.
2148d4002b98SHong Zhang 
2149d4002b98SHong Zhang  Level: intermediate
2150d4002b98SHong Zhang 
2151ed73aabaSBarry Smith  .seealso: MatCreate(), MatCreateSELL(), MatSetValues(), MatSeqSELLSetPreallocation(), MATSELL, MATSEQSELL, MATMPISELL
2152d4002b98SHong Zhang 
2153d4002b98SHong Zhang  @*/
2154d4002b98SHong Zhang PetscErrorCode MatCreateSeqSELL(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt maxallocrow,const PetscInt rlen[],Mat *A)
2155d4002b98SHong Zhang {
2156d4002b98SHong Zhang   PetscFunctionBegin;
21575f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreate(comm,A));
21585f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetSizes(*A,m,n,m,n));
21595f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetType(*A,MATSEQSELL));
21605f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqSELLSetPreallocation_SeqSELL(*A,maxallocrow,rlen));
2161d4002b98SHong Zhang   PetscFunctionReturn(0);
2162d4002b98SHong Zhang }
2163d4002b98SHong Zhang 
2164d4002b98SHong Zhang PetscErrorCode MatEqual_SeqSELL(Mat A,Mat B,PetscBool * flg)
2165d4002b98SHong Zhang {
2166d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data,*b=(Mat_SeqSELL*)B->data;
2167d4002b98SHong Zhang   PetscInt       totalslices=a->totalslices;
2168d4002b98SHong Zhang 
2169d4002b98SHong Zhang   PetscFunctionBegin;
2170d4002b98SHong Zhang   /* If the  matrix dimensions are not equal,or no of nonzeros */
2171d4002b98SHong Zhang   if ((A->rmap->n != B->rmap->n) || (A->cmap->n != B->cmap->n) ||(a->nz != b->nz) || (a->rlenmax != b->rlenmax)) {
2172d4002b98SHong Zhang     *flg = PETSC_FALSE;
2173d4002b98SHong Zhang     PetscFunctionReturn(0);
2174d4002b98SHong Zhang   }
2175d4002b98SHong Zhang   /* if the a->colidx are the same */
21765f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscArraycmp(a->colidx,b->colidx,a->sliidx[totalslices],flg));
2177d4002b98SHong Zhang   if (!*flg) PetscFunctionReturn(0);
2178d4002b98SHong Zhang   /* if a->val are the same */
21795f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscArraycmp(a->val,b->val,a->sliidx[totalslices],flg));
2180d4002b98SHong Zhang   PetscFunctionReturn(0);
2181d4002b98SHong Zhang }
2182d4002b98SHong Zhang 
2183d4002b98SHong Zhang PetscErrorCode MatSeqSELLInvalidateDiagonal(Mat A)
2184d4002b98SHong Zhang {
2185d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
2186d4002b98SHong Zhang 
2187d4002b98SHong Zhang   PetscFunctionBegin;
2188d4002b98SHong Zhang   a->idiagvalid  = PETSC_FALSE;
2189d4002b98SHong Zhang   PetscFunctionReturn(0);
2190d4002b98SHong Zhang }
2191d4002b98SHong Zhang 
2192d4002b98SHong Zhang PetscErrorCode MatConjugate_SeqSELL(Mat A)
2193d4002b98SHong Zhang {
2194d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
2195d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
2196d4002b98SHong Zhang   PetscInt    i;
2197d4002b98SHong Zhang   PetscScalar *val = a->val;
2198d4002b98SHong Zhang 
2199d4002b98SHong Zhang   PetscFunctionBegin;
2200d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) {
2201d4002b98SHong Zhang     val[i] = PetscConj(val[i]);
2202d4002b98SHong Zhang   }
2203d4002b98SHong Zhang #else
2204d4002b98SHong Zhang   PetscFunctionBegin;
2205d4002b98SHong Zhang #endif
2206d4002b98SHong Zhang   PetscFunctionReturn(0);
2207d4002b98SHong Zhang }
2208